1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19#include "qemu/osdep.h"
20#include <getopt.h>
21#include "libqtest-single.h"
22#include "qapi/error.h"
23#include "qapi/qmp/qdict.h"
24#include "qemu/module.h"
25#include "qapi/qobject-input-visitor.h"
26#include "qapi/qapi-visit-machine.h"
27#include "qapi/qapi-visit-qom.h"
28#include "libqos/malloc.h"
29#include "libqos/qgraph.h"
30#include "libqos/qgraph_internal.h"
31#include "libqos/qos_external.h"
32
33static char *old_path;
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51static void qos_set_machines_devices_available(void)
52{
53 QDict *response;
54 QDict *args = qdict_new();
55 QObject *ret;
56 Visitor *v;
57 MachineInfoList *mach_info;
58 ObjectTypeInfoList *type_info;
59
60 qtest_start("-machine none");
61 response = qmp("{ 'execute': 'query-machines' }");
62 ret = qdict_get(response, "return");
63
64 v = qobject_input_visitor_new(ret);
65 visit_type_MachineInfoList(v, NULL, &mach_info, &error_abort);
66 visit_free(v);
67 machines_apply_to_node(mach_info);
68 qapi_free_MachineInfoList(mach_info);
69
70 qobject_unref(response);
71
72 qdict_put_bool(args, "abstract", true);
73 qdict_put_str(args, "implements", "device");
74
75 response = qmp("{'execute': 'qom-list-types',"
76 " 'arguments': %p }", args);
77 ret = qdict_get(response, "return");
78
79 v = qobject_input_visitor_new(ret);
80 visit_type_ObjectTypeInfoList(v, NULL, &type_info, &error_abort);
81 visit_free(v);
82 types_apply_to_node(type_info);
83 qapi_free_ObjectTypeInfoList(type_info);
84
85 qtest_end();
86 qobject_unref(response);
87}
88
89
90static void restart_qemu_or_continue(char *path)
91{
92 if (g_test_verbose()) {
93 qos_printf("Run QEMU with: '%s'\n", path);
94 }
95
96
97
98
99
100
101 if (g_strcmp0(old_path, path)) {
102 qtest_end();
103 qos_invalidate_command_line();
104 old_path = g_strdup(path);
105 qtest_start(path);
106 } else {
107 qobject_unref(qmp("{ 'execute': 'system_reset' }"));
108 qmp_eventwait("RESET");
109 }
110}
111
112void qos_invalidate_command_line(void)
113{
114 g_free(old_path);
115 old_path = NULL;
116}
117
118
119
120
121
122
123
124static char **current_path;
125
126const char *qos_get_current_command_line(void)
127{
128 return current_path[0];
129}
130
131void *qos_allocate_objects(QTestState *qts, QGuestAllocator **p_alloc)
132{
133 return allocate_objects(qts, current_path + 1, p_alloc);
134}
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161static void run_one_test(const void *arg)
162{
163 QOSGraphNode *test_node;
164 QGuestAllocator *alloc = NULL;
165 void *obj;
166 char **path = (char **) arg;
167 GString *cmd_line = g_string_new(path[0]);
168 void *test_arg;
169
170
171 current_path = path;
172 test_node = qos_graph_get_node(path[(g_strv_length(path) - 1)]);
173 test_arg = test_node->u.test.arg;
174 if (test_node->u.test.before) {
175 test_arg = test_node->u.test.before(cmd_line, test_arg);
176 }
177
178 restart_qemu_or_continue(cmd_line->str);
179 g_string_free(cmd_line, true);
180
181 obj = qos_allocate_objects(global_qtest, &alloc);
182 test_node->u.test.function(obj, test_arg, alloc);
183}
184
185static void subprocess_run_one_test(const void *arg)
186{
187 const gchar *path = arg;
188 g_test_trap_subprocess(path, 0, 0);
189 g_test_trap_assert_passed();
190}
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216static void walk_path(QOSGraphNode *orig_path, int len)
217{
218 QOSGraphNode *path;
219 QOSGraphEdge *edge;
220
221
222 QOSEdgeType etype = QEDGE_CONSUMED_BY;
223
224
225 char **path_vec = g_new0(char *, (QOS_PATH_MAX_ELEMENT_SIZE * 2));
226 int path_vec_size = 0;
227
228 char *after_cmd, *before_cmd, *after_device;
229 GString *after_device_str = g_string_new("");
230 char *node_name = orig_path->name, *path_str;
231
232 GString *cmd_line = g_string_new("");
233 GString *cmd_line2 = g_string_new("");
234
235 path = qos_graph_get_node(node_name);
236 node_name = qos_graph_edge_get_dest(path->path_edge);
237
238 path_vec[path_vec_size++] = node_name;
239 path_vec[path_vec_size++] = qos_get_machine_type(node_name);
240
241 for (;;) {
242 path = qos_graph_get_node(node_name);
243 if (!path->path_edge) {
244 break;
245 }
246
247 node_name = qos_graph_edge_get_dest(path->path_edge);
248
249
250 if (path->command_line && etype == QEDGE_CONSUMED_BY) {
251 g_string_append(cmd_line, path->command_line);
252 g_string_append(cmd_line, after_device_str->str);
253 g_string_truncate(after_device_str, 0);
254 }
255
256 path_vec[path_vec_size++] = qos_graph_edge_get_name(path->path_edge);
257
258 after_cmd = qos_graph_edge_get_after_cmd_line(path->path_edge);
259 after_device = qos_graph_edge_get_extra_device_opts(path->path_edge);
260 before_cmd = qos_graph_edge_get_before_cmd_line(path->path_edge);
261 edge = qos_graph_get_edge(path->name, node_name);
262 etype = qos_graph_edge_get_type(edge);
263
264 if (before_cmd) {
265 g_string_append(cmd_line, before_cmd);
266 }
267 if (after_cmd) {
268 g_string_append(cmd_line2, after_cmd);
269 }
270 if (after_device) {
271 g_string_append(after_device_str, after_device);
272 }
273 }
274
275 path_vec[path_vec_size++] = NULL;
276 g_string_append(cmd_line, after_device_str->str);
277 g_string_free(after_device_str, true);
278
279 g_string_append(cmd_line, cmd_line2->str);
280 g_string_free(cmd_line2, true);
281
282
283
284
285 path_str = g_strjoinv("/", path_vec + 1);
286
287
288
289
290 path_vec[1] = path_vec[0];
291 path_vec[0] = g_string_free(cmd_line, false);
292
293 if (path->u.test.subprocess) {
294 gchar *subprocess_path = g_strdup_printf("/%s/%s/subprocess",
295 qtest_get_arch(), path_str);
296 qtest_add_data_func(path_str, subprocess_path, subprocess_run_one_test);
297 g_test_add_data_func(subprocess_path, path_vec, run_one_test);
298 } else {
299 qtest_add_data_func(path_str, path_vec, run_one_test);
300 }
301
302 g_free(path_str);
303}
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319int main(int argc, char **argv, char** envp)
320{
321 g_test_init(&argc, &argv, NULL);
322 if (g_test_verbose()) {
323 qos_printf("ENVIRONMENT VARIABLES: {\n");
324 for (char **env = envp; *env != 0; env++) {
325 qos_printf("\t%s\n", *env);
326 }
327 qos_printf("}\n");
328 }
329 qos_graph_init();
330 module_call_init(MODULE_INIT_QOM);
331 module_call_init(MODULE_INIT_LIBQOS);
332 qos_set_machines_devices_available();
333
334 qos_graph_foreach_test_path(walk_path);
335 if (g_test_verbose()) {
336 qos_dump_graph();
337 }
338 g_test_run();
339 qtest_end();
340 qos_graph_destroy();
341 g_free(old_path);
342 return 0;
343}
344