1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
/*
* Copyright (C) 2016 Canonical Ltd
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 3 as
* published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
#include "test-utils.h"
#include "string-utils.h"
#include "error.h"
#include "utils.h"
#if !GLIB_CHECK_VERSION(2, 69, 0)
// g_spawn_check_exit_status is considered deprecated since 2.69
#define g_spawn_check_wait_status(x, y) (g_spawn_check_exit_status(x, y))
#endif
void rm_rf_tmp(const char *dir) {
// Sanity check, don't remove anything that's not in the temporary
// directory. This is here to prevent unintended data loss.
if (!g_str_has_prefix(dir, "/tmp/")) die("refusing to remove: %s", dir);
const gchar *working_directory = NULL;
gchar **argv = NULL;
gchar **envp = NULL;
GSpawnFlags flags = G_SPAWN_SEARCH_PATH;
GSpawnChildSetupFunc child_setup = NULL;
gpointer user_data = NULL;
gchar **standard_output = NULL;
gchar **standard_error = NULL;
gint exit_status = 0;
GError *error = NULL;
argv = calloc(5, sizeof *argv);
if (argv == NULL) die("cannot allocate command argument array");
argv[0] = g_strdup("rm");
if (argv[0] == NULL) die("cannot allocate memory");
argv[1] = g_strdup("-rf");
if (argv[1] == NULL) die("cannot allocate memory");
argv[2] = g_strdup("--");
if (argv[2] == NULL) die("cannot allocate memory");
argv[3] = g_strdup(dir);
if (argv[3] == NULL) die("cannot allocate memory");
argv[4] = NULL;
g_assert_true(g_spawn_sync(working_directory, argv, envp, flags, child_setup, user_data, standard_output,
standard_error, &exit_status, &error));
g_assert_true(g_spawn_check_wait_status(exit_status, NULL));
if (error != NULL) {
g_test_message("cannot remove temporary directory: %s\n", error->message);
g_error_free(error);
}
g_free(argv[0]);
g_free(argv[1]);
g_free(argv[2]);
g_free(argv[3]);
g_free(argv);
}
void __attribute__((sentinel)) test_argc_argv(int *argcp, char ***argvp, ...) {
int argc = 0;
char **argv = NULL;
va_list ap;
/* find out how many elements there are */
va_start(ap, argvp);
while (NULL != va_arg(ap, const char *)) {
argc += 1;
}
va_end(ap);
/* argc + terminating NULL entry */
argv = calloc(argc + 1, sizeof argv[0]);
g_assert_nonnull(argv);
va_start(ap, argvp);
for (int i = 0; i < argc; i++) {
const char *arg = va_arg(ap, const char *);
char *arg_copy = sc_strdup(arg);
g_test_queue_free(arg_copy);
argv[i] = arg_copy;
}
va_end(ap);
/* free argv last, so that entries do not leak */
g_test_queue_free(argv);
*argcp = argc;
*argvp = argv;
}
extern void sc_set_snap_mount_dir(const char *dir);
void snap_mount_dir_fixture_setup(snap_mount_dir_fixture *fix, gconstpointer user_data) {
sc_set_snap_mount_dir((const char *)user_data);
}
void snap_mount_dir_fixture_teardown(snap_mount_dir_fixture *fix, gconstpointer user_data) {
sc_set_snap_mount_dir(NULL);
}
|
