]>
Commit | Line | Data |
---|---|---|
681c28a3 | 1 | #include "qemu/osdep.h" |
62c39b30 | 2 | #include <locale.h> |
62c39b30 | 3 | #include <glib/gstdio.h> |
62c39b30 MAL |
4 | #include <sys/socket.h> |
5 | #include <sys/un.h> | |
62c39b30 MAL |
6 | |
7 | #include "libqtest.h" | |
62c39b30 MAL |
8 | |
9 | typedef struct { | |
10 | char *test_dir; | |
11 | GMainLoop *loop; | |
12 | int fd; | |
13 | GPid pid; | |
14 | } TestFixture; | |
15 | ||
16 | static int connect_qga(char *path) | |
17 | { | |
18 | int s, ret, len, i = 0; | |
19 | struct sockaddr_un remote; | |
20 | ||
21 | s = socket(AF_UNIX, SOCK_STREAM, 0); | |
22 | g_assert(s != -1); | |
23 | ||
24 | remote.sun_family = AF_UNIX; | |
25 | do { | |
26 | strcpy(remote.sun_path, path); | |
27 | len = strlen(remote.sun_path) + sizeof(remote.sun_family); | |
28 | ret = connect(s, (struct sockaddr *)&remote, len); | |
29 | if (ret == -1) { | |
30 | g_usleep(G_USEC_PER_SEC); | |
31 | } | |
32 | if (i++ == 10) { | |
33 | return -1; | |
34 | } | |
35 | } while (ret == -1); | |
36 | ||
37 | return s; | |
38 | } | |
39 | ||
40 | static void qga_watch(GPid pid, gint status, gpointer user_data) | |
41 | { | |
42 | TestFixture *fixture = user_data; | |
43 | ||
44 | g_assert_cmpint(status, ==, 0); | |
45 | g_main_loop_quit(fixture->loop); | |
46 | } | |
47 | ||
48 | static void | |
49 | fixture_setup(TestFixture *fixture, gconstpointer data) | |
50 | { | |
51 | const gchar *extra_arg = data; | |
52 | GError *error = NULL; | |
53 | gchar *cwd, *path, *cmd, **argv = NULL; | |
54 | ||
55 | fixture->loop = g_main_loop_new(NULL, FALSE); | |
56 | ||
57 | fixture->test_dir = g_strdup("/tmp/qgatest.XXXXXX"); | |
58 | g_assert_nonnull(mkdtemp(fixture->test_dir)); | |
59 | ||
60 | path = g_build_filename(fixture->test_dir, "sock", NULL); | |
61 | cwd = g_get_current_dir(); | |
62 | cmd = g_strdup_printf("%s%cqemu-ga -m unix-listen -t %s -p %s %s %s", | |
63 | cwd, G_DIR_SEPARATOR, | |
64 | fixture->test_dir, path, | |
65 | getenv("QTEST_LOG") ? "-v" : "", | |
66 | extra_arg ?: ""); | |
67 | g_shell_parse_argv(cmd, NULL, &argv, &error); | |
68 | g_assert_no_error(error); | |
69 | ||
70 | g_spawn_async(fixture->test_dir, argv, NULL, | |
71 | G_SPAWN_SEARCH_PATH|G_SPAWN_DO_NOT_REAP_CHILD, | |
72 | NULL, NULL, &fixture->pid, &error); | |
73 | g_assert_no_error(error); | |
74 | ||
75 | g_child_watch_add(fixture->pid, qga_watch, fixture); | |
76 | ||
77 | fixture->fd = connect_qga(path); | |
78 | g_assert_cmpint(fixture->fd, !=, -1); | |
79 | ||
80 | g_strfreev(argv); | |
81 | g_free(cmd); | |
82 | g_free(cwd); | |
83 | g_free(path); | |
84 | } | |
85 | ||
86 | static void | |
87 | fixture_tear_down(TestFixture *fixture, gconstpointer data) | |
88 | { | |
89 | gchar *tmp; | |
90 | ||
91 | kill(fixture->pid, SIGTERM); | |
92 | ||
93 | g_main_loop_run(fixture->loop); | |
94 | g_main_loop_unref(fixture->loop); | |
95 | ||
96 | g_spawn_close_pid(fixture->pid); | |
97 | ||
98 | tmp = g_build_filename(fixture->test_dir, "foo", NULL); | |
99 | g_unlink(tmp); | |
100 | g_free(tmp); | |
101 | ||
102 | tmp = g_build_filename(fixture->test_dir, "qga.state", NULL); | |
103 | g_unlink(tmp); | |
104 | g_free(tmp); | |
105 | ||
106 | tmp = g_build_filename(fixture->test_dir, "sock", NULL); | |
107 | g_unlink(tmp); | |
108 | g_free(tmp); | |
109 | ||
110 | g_rmdir(fixture->test_dir); | |
111 | g_free(fixture->test_dir); | |
112 | } | |
113 | ||
114 | static void qmp_assertion_message_error(const char *domain, | |
115 | const char *file, | |
116 | int line, | |
117 | const char *func, | |
118 | const char *expr, | |
119 | QDict *dict) | |
120 | { | |
121 | const char *class, *desc; | |
122 | char *s; | |
123 | QDict *error; | |
124 | ||
125 | error = qdict_get_qdict(dict, "error"); | |
126 | class = qdict_get_try_str(error, "class"); | |
127 | desc = qdict_get_try_str(error, "desc"); | |
128 | ||
129 | s = g_strdup_printf("assertion failed %s: %s %s", expr, class, desc); | |
130 | g_assertion_message(domain, file, line, func, s); | |
131 | g_free(s); | |
132 | } | |
133 | ||
134 | #define qmp_assert_no_error(err) do { \ | |
135 | if (qdict_haskey(err, "error")) { \ | |
136 | qmp_assertion_message_error(G_LOG_DOMAIN, __FILE__, __LINE__, \ | |
137 | G_STRFUNC, #err, err); \ | |
138 | } \ | |
139 | } while (0) | |
140 | ||
141 | static void test_qga_sync_delimited(gconstpointer fix) | |
142 | { | |
143 | const TestFixture *fixture = fix; | |
144 | guint32 v, r = g_random_int(); | |
145 | unsigned char c; | |
146 | QDict *ret; | |
147 | gchar *cmd; | |
148 | ||
149 | cmd = g_strdup_printf("%c{'execute': 'guest-sync-delimited'," | |
150 | " 'arguments': {'id': %u } }", 0xff, r); | |
151 | qmp_fd_send(fixture->fd, cmd); | |
152 | g_free(cmd); | |
153 | ||
154 | v = read(fixture->fd, &c, 1); | |
155 | g_assert_cmpint(v, ==, 1); | |
156 | g_assert_cmpint(c, ==, 0xff); | |
157 | ||
158 | ret = qmp_fd_receive(fixture->fd); | |
159 | g_assert_nonnull(ret); | |
160 | qmp_assert_no_error(ret); | |
161 | ||
162 | v = qdict_get_int(ret, "return"); | |
163 | g_assert_cmpint(r, ==, v); | |
164 | ||
165 | QDECREF(ret); | |
166 | } | |
167 | ||
168 | static void test_qga_sync(gconstpointer fix) | |
169 | { | |
170 | const TestFixture *fixture = fix; | |
171 | guint32 v, r = g_random_int(); | |
172 | QDict *ret; | |
173 | gchar *cmd; | |
174 | ||
175 | cmd = g_strdup_printf("%c{'execute': 'guest-sync'," | |
176 | " 'arguments': {'id': %u } }", 0xff, r); | |
177 | ret = qmp_fd(fixture->fd, cmd); | |
178 | g_free(cmd); | |
179 | ||
180 | g_assert_nonnull(ret); | |
181 | qmp_assert_no_error(ret); | |
182 | ||
183 | v = qdict_get_int(ret, "return"); | |
184 | g_assert_cmpint(r, ==, v); | |
185 | ||
186 | QDECREF(ret); | |
187 | } | |
188 | ||
189 | static void test_qga_ping(gconstpointer fix) | |
190 | { | |
191 | const TestFixture *fixture = fix; | |
192 | QDict *ret; | |
193 | ||
194 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-ping'}"); | |
195 | g_assert_nonnull(ret); | |
196 | qmp_assert_no_error(ret); | |
197 | ||
198 | QDECREF(ret); | |
199 | } | |
200 | ||
201 | static void test_qga_invalid_cmd(gconstpointer fix) | |
202 | { | |
203 | const TestFixture *fixture = fix; | |
204 | QDict *ret, *error; | |
205 | const gchar *class, *desc; | |
206 | ||
207 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-invalid-cmd'}"); | |
208 | g_assert_nonnull(ret); | |
209 | ||
210 | error = qdict_get_qdict(ret, "error"); | |
211 | class = qdict_get_try_str(error, "class"); | |
212 | desc = qdict_get_try_str(error, "desc"); | |
213 | ||
214 | g_assert_cmpstr(class, ==, "CommandNotFound"); | |
215 | g_assert_cmpint(strlen(desc), >, 0); | |
216 | ||
217 | QDECREF(ret); | |
218 | } | |
219 | ||
220 | static void test_qga_info(gconstpointer fix) | |
221 | { | |
222 | const TestFixture *fixture = fix; | |
223 | QDict *ret, *val; | |
224 | const gchar *version; | |
225 | ||
226 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-info'}"); | |
227 | g_assert_nonnull(ret); | |
228 | qmp_assert_no_error(ret); | |
229 | ||
230 | val = qdict_get_qdict(ret, "return"); | |
231 | version = qdict_get_try_str(val, "version"); | |
232 | g_assert_cmpstr(version, ==, QEMU_VERSION); | |
233 | ||
234 | QDECREF(ret); | |
235 | } | |
236 | ||
237 | static void test_qga_get_vcpus(gconstpointer fix) | |
238 | { | |
239 | const TestFixture *fixture = fix; | |
240 | QDict *ret; | |
241 | QList *list; | |
242 | const QListEntry *entry; | |
243 | ||
244 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-get-vcpus'}"); | |
245 | g_assert_nonnull(ret); | |
246 | qmp_assert_no_error(ret); | |
247 | ||
248 | /* check there is at least a cpu */ | |
249 | list = qdict_get_qlist(ret, "return"); | |
250 | entry = qlist_first(list); | |
251 | g_assert(qdict_haskey(qobject_to_qdict(entry->value), "online")); | |
252 | g_assert(qdict_haskey(qobject_to_qdict(entry->value), "logical-id")); | |
253 | ||
254 | QDECREF(ret); | |
255 | } | |
256 | ||
257 | static void test_qga_get_fsinfo(gconstpointer fix) | |
258 | { | |
259 | const TestFixture *fixture = fix; | |
260 | QDict *ret; | |
261 | QList *list; | |
262 | const QListEntry *entry; | |
263 | ||
264 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-get-fsinfo'}"); | |
265 | g_assert_nonnull(ret); | |
266 | qmp_assert_no_error(ret); | |
267 | ||
b3e9e584 | 268 | /* sanity-check the response if there are any filesystems */ |
62c39b30 MAL |
269 | list = qdict_get_qlist(ret, "return"); |
270 | entry = qlist_first(list); | |
b3e9e584 MR |
271 | if (entry) { |
272 | g_assert(qdict_haskey(qobject_to_qdict(entry->value), "name")); | |
273 | g_assert(qdict_haskey(qobject_to_qdict(entry->value), "mountpoint")); | |
274 | g_assert(qdict_haskey(qobject_to_qdict(entry->value), "type")); | |
275 | g_assert(qdict_haskey(qobject_to_qdict(entry->value), "disk")); | |
276 | } | |
62c39b30 MAL |
277 | |
278 | QDECREF(ret); | |
279 | } | |
280 | ||
281 | static void test_qga_get_memory_block_info(gconstpointer fix) | |
282 | { | |
283 | const TestFixture *fixture = fix; | |
284 | QDict *ret, *val; | |
285 | int64_t size; | |
286 | ||
287 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-get-memory-block-info'}"); | |
288 | g_assert_nonnull(ret); | |
289 | ||
290 | /* some systems might not expose memory block info in sysfs */ | |
291 | if (!qdict_haskey(ret, "error")) { | |
292 | /* check there is at least some memory */ | |
293 | val = qdict_get_qdict(ret, "return"); | |
294 | size = qdict_get_int(val, "size"); | |
295 | g_assert_cmpint(size, >, 0); | |
296 | } | |
297 | ||
298 | QDECREF(ret); | |
299 | } | |
300 | ||
301 | static void test_qga_get_memory_blocks(gconstpointer fix) | |
302 | { | |
303 | const TestFixture *fixture = fix; | |
304 | QDict *ret; | |
305 | QList *list; | |
306 | const QListEntry *entry; | |
307 | ||
308 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-get-memory-blocks'}"); | |
309 | g_assert_nonnull(ret); | |
310 | ||
311 | /* some systems might not expose memory block info in sysfs */ | |
312 | if (!qdict_haskey(ret, "error")) { | |
313 | list = qdict_get_qlist(ret, "return"); | |
314 | entry = qlist_first(list); | |
315 | /* newer versions of qga may return empty list without error */ | |
316 | if (entry) { | |
317 | g_assert(qdict_haskey(qobject_to_qdict(entry->value), "phys-index")); | |
318 | g_assert(qdict_haskey(qobject_to_qdict(entry->value), "online")); | |
319 | } | |
320 | } | |
321 | ||
322 | QDECREF(ret); | |
323 | } | |
324 | ||
325 | static void test_qga_network_get_interfaces(gconstpointer fix) | |
326 | { | |
327 | const TestFixture *fixture = fix; | |
328 | QDict *ret; | |
329 | QList *list; | |
330 | const QListEntry *entry; | |
331 | ||
332 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-network-get-interfaces'}"); | |
333 | g_assert_nonnull(ret); | |
334 | qmp_assert_no_error(ret); | |
335 | ||
336 | /* check there is at least an interface */ | |
337 | list = qdict_get_qlist(ret, "return"); | |
338 | entry = qlist_first(list); | |
339 | g_assert(qdict_haskey(qobject_to_qdict(entry->value), "name")); | |
340 | ||
341 | QDECREF(ret); | |
342 | } | |
343 | ||
344 | static void test_qga_file_ops(gconstpointer fix) | |
345 | { | |
346 | const TestFixture *fixture = fix; | |
4eaab85c | 347 | const unsigned char helloworld[] = "Hello World!\n"; |
62c39b30 MAL |
348 | const char *b64; |
349 | gchar *cmd, *path, *enc; | |
4eaab85c | 350 | unsigned char *dec; |
62c39b30 MAL |
351 | QDict *ret, *val; |
352 | int64_t id, eof; | |
353 | gsize count; | |
354 | FILE *f; | |
355 | char tmp[100]; | |
356 | ||
357 | /* open */ | |
358 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-file-open'," | |
359 | " 'arguments': { 'path': 'foo', 'mode': 'w+' } }"); | |
360 | g_assert_nonnull(ret); | |
361 | qmp_assert_no_error(ret); | |
362 | id = qdict_get_int(ret, "return"); | |
363 | QDECREF(ret); | |
364 | ||
365 | enc = g_base64_encode(helloworld, sizeof(helloworld)); | |
366 | /* write */ | |
367 | cmd = g_strdup_printf("{'execute': 'guest-file-write'," | |
368 | " 'arguments': { 'handle': %" PRId64 "," | |
369 | " 'buf-b64': '%s' } }", id, enc); | |
370 | ret = qmp_fd(fixture->fd, cmd); | |
371 | g_assert_nonnull(ret); | |
372 | qmp_assert_no_error(ret); | |
373 | ||
374 | val = qdict_get_qdict(ret, "return"); | |
375 | count = qdict_get_int(val, "count"); | |
376 | eof = qdict_get_bool(val, "eof"); | |
377 | g_assert_cmpint(count, ==, sizeof(helloworld)); | |
378 | g_assert_cmpint(eof, ==, 0); | |
379 | QDECREF(ret); | |
380 | g_free(cmd); | |
381 | ||
382 | /* flush */ | |
383 | cmd = g_strdup_printf("{'execute': 'guest-file-flush'," | |
384 | " 'arguments': {'handle': %" PRId64 "} }", | |
385 | id); | |
386 | ret = qmp_fd(fixture->fd, cmd); | |
387 | QDECREF(ret); | |
388 | g_free(cmd); | |
389 | ||
390 | /* close */ | |
391 | cmd = g_strdup_printf("{'execute': 'guest-file-close'," | |
392 | " 'arguments': {'handle': %" PRId64 "} }", | |
393 | id); | |
394 | ret = qmp_fd(fixture->fd, cmd); | |
395 | QDECREF(ret); | |
396 | g_free(cmd); | |
397 | ||
398 | /* check content */ | |
399 | path = g_build_filename(fixture->test_dir, "foo", NULL); | |
400 | f = fopen(path, "r"); | |
401 | g_assert_nonnull(f); | |
402 | count = fread(tmp, 1, sizeof(tmp), f); | |
403 | g_assert_cmpint(count, ==, sizeof(helloworld)); | |
404 | tmp[count] = 0; | |
405 | g_assert_cmpstr(tmp, ==, (char *)helloworld); | |
406 | fclose(f); | |
407 | ||
408 | /* open */ | |
409 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-file-open'," | |
410 | " 'arguments': { 'path': 'foo', 'mode': 'r' } }"); | |
411 | g_assert_nonnull(ret); | |
412 | qmp_assert_no_error(ret); | |
413 | id = qdict_get_int(ret, "return"); | |
414 | QDECREF(ret); | |
415 | ||
416 | /* read */ | |
417 | cmd = g_strdup_printf("{'execute': 'guest-file-read'," | |
418 | " 'arguments': { 'handle': %" PRId64 "} }", | |
419 | id); | |
420 | ret = qmp_fd(fixture->fd, cmd); | |
421 | val = qdict_get_qdict(ret, "return"); | |
422 | count = qdict_get_int(val, "count"); | |
423 | eof = qdict_get_bool(val, "eof"); | |
424 | b64 = qdict_get_str(val, "buf-b64"); | |
425 | g_assert_cmpint(count, ==, sizeof(helloworld)); | |
426 | g_assert(eof); | |
427 | g_assert_cmpstr(b64, ==, enc); | |
428 | ||
429 | QDECREF(ret); | |
430 | g_free(cmd); | |
431 | g_free(enc); | |
432 | ||
433 | /* read eof */ | |
434 | cmd = g_strdup_printf("{'execute': 'guest-file-read'," | |
435 | " 'arguments': { 'handle': %" PRId64 "} }", | |
436 | id); | |
437 | ret = qmp_fd(fixture->fd, cmd); | |
438 | val = qdict_get_qdict(ret, "return"); | |
439 | count = qdict_get_int(val, "count"); | |
440 | eof = qdict_get_bool(val, "eof"); | |
441 | b64 = qdict_get_str(val, "buf-b64"); | |
442 | g_assert_cmpint(count, ==, 0); | |
443 | g_assert(eof); | |
444 | g_assert_cmpstr(b64, ==, ""); | |
445 | QDECREF(ret); | |
446 | g_free(cmd); | |
447 | ||
448 | /* seek */ | |
449 | cmd = g_strdup_printf("{'execute': 'guest-file-seek'," | |
450 | " 'arguments': { 'handle': %" PRId64 ", " | |
0b4b4938 EB |
451 | " 'offset': %d, 'whence': '%s' } }", |
452 | id, 6, "set"); | |
62c39b30 MAL |
453 | ret = qmp_fd(fixture->fd, cmd); |
454 | qmp_assert_no_error(ret); | |
455 | val = qdict_get_qdict(ret, "return"); | |
456 | count = qdict_get_int(val, "position"); | |
457 | eof = qdict_get_bool(val, "eof"); | |
458 | g_assert_cmpint(count, ==, 6); | |
459 | g_assert(!eof); | |
460 | QDECREF(ret); | |
461 | g_free(cmd); | |
462 | ||
463 | /* partial read */ | |
464 | cmd = g_strdup_printf("{'execute': 'guest-file-read'," | |
465 | " 'arguments': { 'handle': %" PRId64 "} }", | |
466 | id); | |
467 | ret = qmp_fd(fixture->fd, cmd); | |
468 | val = qdict_get_qdict(ret, "return"); | |
469 | count = qdict_get_int(val, "count"); | |
470 | eof = qdict_get_bool(val, "eof"); | |
471 | b64 = qdict_get_str(val, "buf-b64"); | |
472 | g_assert_cmpint(count, ==, sizeof(helloworld) - 6); | |
473 | g_assert(eof); | |
474 | dec = g_base64_decode(b64, &count); | |
475 | g_assert_cmpint(count, ==, sizeof(helloworld) - 6); | |
476 | g_assert_cmpmem(dec, count, helloworld + 6, sizeof(helloworld) - 6); | |
477 | g_free(dec); | |
478 | ||
479 | QDECREF(ret); | |
480 | g_free(cmd); | |
481 | ||
482 | /* close */ | |
483 | cmd = g_strdup_printf("{'execute': 'guest-file-close'," | |
484 | " 'arguments': {'handle': %" PRId64 "} }", | |
485 | id); | |
486 | ret = qmp_fd(fixture->fd, cmd); | |
487 | QDECREF(ret); | |
488 | g_free(cmd); | |
489 | } | |
490 | ||
4eaab85c MAL |
491 | static void test_qga_file_write_read(gconstpointer fix) |
492 | { | |
493 | const TestFixture *fixture = fix; | |
494 | const unsigned char helloworld[] = "Hello World!\n"; | |
495 | const char *b64; | |
496 | gchar *cmd, *enc; | |
497 | QDict *ret, *val; | |
498 | int64_t id, eof; | |
499 | gsize count; | |
500 | ||
501 | /* open */ | |
502 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-file-open'," | |
503 | " 'arguments': { 'path': 'foo', 'mode': 'w+' } }"); | |
504 | g_assert_nonnull(ret); | |
505 | qmp_assert_no_error(ret); | |
506 | id = qdict_get_int(ret, "return"); | |
507 | QDECREF(ret); | |
508 | ||
509 | enc = g_base64_encode(helloworld, sizeof(helloworld)); | |
510 | /* write */ | |
511 | cmd = g_strdup_printf("{'execute': 'guest-file-write'," | |
512 | " 'arguments': { 'handle': %" PRId64 "," | |
513 | " 'buf-b64': '%s' } }", id, enc); | |
514 | ret = qmp_fd(fixture->fd, cmd); | |
515 | g_assert_nonnull(ret); | |
516 | qmp_assert_no_error(ret); | |
517 | ||
518 | val = qdict_get_qdict(ret, "return"); | |
519 | count = qdict_get_int(val, "count"); | |
520 | eof = qdict_get_bool(val, "eof"); | |
521 | g_assert_cmpint(count, ==, sizeof(helloworld)); | |
522 | g_assert_cmpint(eof, ==, 0); | |
523 | QDECREF(ret); | |
524 | g_free(cmd); | |
525 | ||
526 | /* read (check implicit flush) */ | |
527 | cmd = g_strdup_printf("{'execute': 'guest-file-read'," | |
528 | " 'arguments': { 'handle': %" PRId64 "} }", | |
529 | id); | |
530 | ret = qmp_fd(fixture->fd, cmd); | |
531 | val = qdict_get_qdict(ret, "return"); | |
532 | count = qdict_get_int(val, "count"); | |
533 | eof = qdict_get_bool(val, "eof"); | |
534 | b64 = qdict_get_str(val, "buf-b64"); | |
535 | g_assert_cmpint(count, ==, 0); | |
536 | g_assert(eof); | |
537 | g_assert_cmpstr(b64, ==, ""); | |
538 | QDECREF(ret); | |
539 | g_free(cmd); | |
540 | ||
541 | /* seek to 0 */ | |
542 | cmd = g_strdup_printf("{'execute': 'guest-file-seek'," | |
543 | " 'arguments': { 'handle': %" PRId64 ", " | |
0b4b4938 EB |
544 | " 'offset': %d, 'whence': '%s' } }", |
545 | id, 0, "set"); | |
4eaab85c MAL |
546 | ret = qmp_fd(fixture->fd, cmd); |
547 | qmp_assert_no_error(ret); | |
548 | val = qdict_get_qdict(ret, "return"); | |
549 | count = qdict_get_int(val, "position"); | |
550 | eof = qdict_get_bool(val, "eof"); | |
551 | g_assert_cmpint(count, ==, 0); | |
552 | g_assert(!eof); | |
553 | QDECREF(ret); | |
554 | g_free(cmd); | |
555 | ||
556 | /* read */ | |
557 | cmd = g_strdup_printf("{'execute': 'guest-file-read'," | |
558 | " 'arguments': { 'handle': %" PRId64 "} }", | |
559 | id); | |
560 | ret = qmp_fd(fixture->fd, cmd); | |
561 | val = qdict_get_qdict(ret, "return"); | |
562 | count = qdict_get_int(val, "count"); | |
563 | eof = qdict_get_bool(val, "eof"); | |
564 | b64 = qdict_get_str(val, "buf-b64"); | |
565 | g_assert_cmpint(count, ==, sizeof(helloworld)); | |
566 | g_assert(eof); | |
567 | g_assert_cmpstr(b64, ==, enc); | |
568 | QDECREF(ret); | |
569 | g_free(cmd); | |
570 | g_free(enc); | |
571 | ||
572 | /* close */ | |
573 | cmd = g_strdup_printf("{'execute': 'guest-file-close'," | |
574 | " 'arguments': {'handle': %" PRId64 "} }", | |
575 | id); | |
576 | ret = qmp_fd(fixture->fd, cmd); | |
577 | QDECREF(ret); | |
578 | g_free(cmd); | |
579 | } | |
580 | ||
62c39b30 MAL |
581 | static void test_qga_get_time(gconstpointer fix) |
582 | { | |
583 | const TestFixture *fixture = fix; | |
584 | QDict *ret; | |
585 | int64_t time; | |
586 | ||
587 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-get-time'}"); | |
588 | g_assert_nonnull(ret); | |
589 | qmp_assert_no_error(ret); | |
590 | ||
591 | time = qdict_get_int(ret, "return"); | |
592 | g_assert_cmpint(time, >, 0); | |
593 | ||
594 | QDECREF(ret); | |
595 | } | |
596 | ||
597 | static void test_qga_set_time(gconstpointer fix) | |
598 | { | |
599 | const TestFixture *fixture = fix; | |
600 | QDict *ret; | |
601 | int64_t current, time; | |
602 | gchar *cmd; | |
603 | ||
604 | /* get current time */ | |
605 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-get-time'}"); | |
606 | g_assert_nonnull(ret); | |
607 | qmp_assert_no_error(ret); | |
608 | current = qdict_get_int(ret, "return"); | |
609 | g_assert_cmpint(current, >, 0); | |
610 | QDECREF(ret); | |
611 | ||
612 | /* set some old time */ | |
613 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-set-time'," | |
614 | " 'arguments': { 'time': 1000 } }"); | |
615 | g_assert_nonnull(ret); | |
616 | qmp_assert_no_error(ret); | |
617 | QDECREF(ret); | |
618 | ||
619 | /* check old time */ | |
620 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-get-time'}"); | |
621 | g_assert_nonnull(ret); | |
622 | qmp_assert_no_error(ret); | |
623 | time = qdict_get_int(ret, "return"); | |
624 | g_assert_cmpint(time / 1000, <, G_USEC_PER_SEC * 10); | |
625 | QDECREF(ret); | |
626 | ||
627 | /* set back current time */ | |
628 | cmd = g_strdup_printf("{'execute': 'guest-set-time'," | |
629 | " 'arguments': { 'time': %" PRId64 " } }", | |
630 | current + time * 1000); | |
631 | ret = qmp_fd(fixture->fd, cmd); | |
632 | g_free(cmd); | |
633 | g_assert_nonnull(ret); | |
634 | qmp_assert_no_error(ret); | |
635 | QDECREF(ret); | |
636 | } | |
637 | ||
638 | static void test_qga_fstrim(gconstpointer fix) | |
639 | { | |
640 | const TestFixture *fixture = fix; | |
641 | QDict *ret; | |
642 | QList *list; | |
643 | const QListEntry *entry; | |
644 | ||
645 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-fstrim'," | |
646 | " arguments: { minimum: 4194304 } }"); | |
647 | g_assert_nonnull(ret); | |
648 | qmp_assert_no_error(ret); | |
649 | list = qdict_get_qlist(ret, "return"); | |
650 | entry = qlist_first(list); | |
651 | g_assert(qdict_haskey(qobject_to_qdict(entry->value), "paths")); | |
652 | ||
653 | QDECREF(ret); | |
654 | } | |
655 | ||
656 | static void test_qga_blacklist(gconstpointer data) | |
657 | { | |
658 | TestFixture fix; | |
659 | QDict *ret, *error; | |
660 | const gchar *class, *desc; | |
661 | ||
662 | fixture_setup(&fix, "-b guest-ping,guest-get-time"); | |
663 | ||
664 | /* check blacklist */ | |
665 | ret = qmp_fd(fix.fd, "{'execute': 'guest-ping'}"); | |
666 | g_assert_nonnull(ret); | |
667 | error = qdict_get_qdict(ret, "error"); | |
668 | class = qdict_get_try_str(error, "class"); | |
669 | desc = qdict_get_try_str(error, "desc"); | |
670 | g_assert_cmpstr(class, ==, "GenericError"); | |
671 | g_assert_nonnull(g_strstr_len(desc, -1, "has been disabled")); | |
672 | QDECREF(ret); | |
673 | ||
674 | ret = qmp_fd(fix.fd, "{'execute': 'guest-get-time'}"); | |
675 | g_assert_nonnull(ret); | |
676 | error = qdict_get_qdict(ret, "error"); | |
677 | class = qdict_get_try_str(error, "class"); | |
678 | desc = qdict_get_try_str(error, "desc"); | |
679 | g_assert_cmpstr(class, ==, "GenericError"); | |
680 | g_assert_nonnull(g_strstr_len(desc, -1, "has been disabled")); | |
681 | QDECREF(ret); | |
682 | ||
683 | /* check something work */ | |
684 | ret = qmp_fd(fix.fd, "{'execute': 'guest-get-fsinfo'}"); | |
685 | qmp_assert_no_error(ret); | |
686 | QDECREF(ret); | |
687 | ||
688 | fixture_tear_down(&fix, NULL); | |
689 | } | |
690 | ||
691 | static void test_qga_config(gconstpointer data) | |
692 | { | |
693 | GError *error = NULL; | |
694 | char *cwd, *cmd, *out, *err, *str, **strv, *conf, **argv = NULL; | |
695 | char *env[2]; | |
696 | int status, tmp; | |
697 | gsize n; | |
698 | GKeyFile *kf; | |
699 | const char *qga_config = | |
700 | "[general]\n" | |
701 | "daemon=false\n" | |
702 | "method=virtio-serial\n" | |
703 | "path=/path/to/org.qemu.guest_agent.0\n" | |
704 | "pidfile=/var/foo/qemu-ga.pid\n" | |
705 | "statedir=/var/state\n" | |
706 | "verbose=true\n" | |
707 | "blacklist=guest-ping;guest-get-time\n"; | |
708 | ||
709 | tmp = g_file_open_tmp(NULL, &conf, &error); | |
710 | g_assert_no_error(error); | |
711 | g_assert_cmpint(tmp, >=, 0); | |
712 | g_assert_cmpstr(conf, !=, ""); | |
713 | ||
714 | g_file_set_contents(conf, qga_config, -1, &error); | |
715 | g_assert_no_error(error); | |
716 | ||
717 | cwd = g_get_current_dir(); | |
718 | cmd = g_strdup_printf("%s%cqemu-ga -D", | |
719 | cwd, G_DIR_SEPARATOR); | |
720 | g_shell_parse_argv(cmd, NULL, &argv, &error); | |
721 | g_assert_no_error(error); | |
722 | ||
723 | env[0] = g_strdup_printf("QGA_CONF=%s", conf); | |
724 | env[1] = NULL; | |
725 | g_spawn_sync(NULL, argv, env, 0, | |
726 | NULL, NULL, &out, &err, &status, &error); | |
727 | g_assert_no_error(error); | |
728 | g_assert_cmpstr(err, ==, ""); | |
729 | g_assert_cmpint(status, ==, 0); | |
730 | ||
731 | kf = g_key_file_new(); | |
732 | g_key_file_load_from_data(kf, out, -1, G_KEY_FILE_NONE, &error); | |
733 | g_assert_no_error(error); | |
734 | ||
735 | str = g_key_file_get_start_group(kf); | |
736 | g_assert_cmpstr(str, ==, "general"); | |
737 | g_free(str); | |
738 | ||
739 | g_assert_false(g_key_file_get_boolean(kf, "general", "daemon", &error)); | |
740 | g_assert_no_error(error); | |
741 | ||
742 | str = g_key_file_get_string(kf, "general", "method", &error); | |
743 | g_assert_no_error(error); | |
744 | g_assert_cmpstr(str, ==, "virtio-serial"); | |
745 | g_free(str); | |
746 | ||
747 | str = g_key_file_get_string(kf, "general", "path", &error); | |
748 | g_assert_no_error(error); | |
749 | g_assert_cmpstr(str, ==, "/path/to/org.qemu.guest_agent.0"); | |
750 | g_free(str); | |
751 | ||
752 | str = g_key_file_get_string(kf, "general", "pidfile", &error); | |
753 | g_assert_no_error(error); | |
754 | g_assert_cmpstr(str, ==, "/var/foo/qemu-ga.pid"); | |
755 | g_free(str); | |
756 | ||
757 | str = g_key_file_get_string(kf, "general", "statedir", &error); | |
758 | g_assert_no_error(error); | |
759 | g_assert_cmpstr(str, ==, "/var/state"); | |
760 | g_free(str); | |
761 | ||
762 | g_assert_true(g_key_file_get_boolean(kf, "general", "verbose", &error)); | |
763 | g_assert_no_error(error); | |
764 | ||
765 | strv = g_key_file_get_string_list(kf, "general", "blacklist", &n, &error); | |
766 | g_assert_cmpint(n, ==, 2); | |
767 | #if GLIB_CHECK_VERSION(2, 44, 0) | |
768 | g_assert_true(g_strv_contains((const char * const *)strv, | |
769 | "guest-ping")); | |
770 | g_assert_true(g_strv_contains((const char * const *)strv, | |
771 | "guest-get-time")); | |
772 | #endif | |
773 | g_assert_no_error(error); | |
774 | g_strfreev(strv); | |
775 | ||
776 | g_free(out); | |
777 | g_free(err); | |
778 | g_free(conf); | |
779 | g_free(env[0]); | |
780 | g_key_file_free(kf); | |
781 | ||
782 | close(tmp); | |
783 | } | |
784 | ||
785 | static void test_qga_fsfreeze_status(gconstpointer fix) | |
786 | { | |
787 | const TestFixture *fixture = fix; | |
788 | QDict *ret; | |
789 | const gchar *status; | |
790 | ||
791 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-fsfreeze-status'}"); | |
792 | g_assert_nonnull(ret); | |
793 | qmp_assert_no_error(ret); | |
794 | ||
795 | status = qdict_get_try_str(ret, "return"); | |
796 | g_assert_cmpstr(status, ==, "thawed"); | |
797 | ||
798 | QDECREF(ret); | |
799 | } | |
800 | ||
801 | static void test_qga_fsfreeze_and_thaw(gconstpointer fix) | |
802 | { | |
803 | const TestFixture *fixture = fix; | |
804 | QDict *ret; | |
805 | const gchar *status; | |
806 | ||
807 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-fsfreeze-freeze'}"); | |
808 | g_assert_nonnull(ret); | |
809 | qmp_assert_no_error(ret); | |
810 | QDECREF(ret); | |
811 | ||
812 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-fsfreeze-status'}"); | |
813 | g_assert_nonnull(ret); | |
814 | qmp_assert_no_error(ret); | |
815 | status = qdict_get_try_str(ret, "return"); | |
816 | g_assert_cmpstr(status, ==, "frozen"); | |
817 | QDECREF(ret); | |
818 | ||
819 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-fsfreeze-thaw'}"); | |
820 | g_assert_nonnull(ret); | |
821 | qmp_assert_no_error(ret); | |
822 | QDECREF(ret); | |
823 | } | |
824 | ||
3dab9fa1 MAL |
825 | static void test_qga_guest_exec(gconstpointer fix) |
826 | { | |
827 | const TestFixture *fixture = fix; | |
828 | QDict *ret, *val; | |
829 | const gchar *out; | |
830 | guchar *decoded; | |
831 | int64_t pid, now, exitcode; | |
832 | gsize len; | |
833 | bool exited; | |
834 | ||
835 | /* exec 'echo foo bar' */ | |
836 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-exec', 'arguments': {" | |
837 | " 'path': '/bin/echo', 'arg': [ '-n', '\" test_str \"' ]," | |
838 | " 'capture-output': true } }"); | |
839 | g_assert_nonnull(ret); | |
840 | qmp_assert_no_error(ret); | |
841 | val = qdict_get_qdict(ret, "return"); | |
842 | pid = qdict_get_int(val, "pid"); | |
843 | g_assert_cmpint(pid, >, 0); | |
844 | QDECREF(ret); | |
845 | ||
846 | /* wait for completion */ | |
847 | now = g_get_monotonic_time(); | |
848 | do { | |
849 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-exec-status'," | |
850 | " 'arguments': { 'pid': %" PRId64 " } }", pid); | |
851 | g_assert_nonnull(ret); | |
852 | val = qdict_get_qdict(ret, "return"); | |
853 | exited = qdict_get_bool(val, "exited"); | |
854 | if (!exited) { | |
855 | QDECREF(ret); | |
856 | } | |
857 | } while (!exited && | |
858 | g_get_monotonic_time() < now + 5 * G_TIME_SPAN_SECOND); | |
859 | g_assert(exited); | |
860 | ||
861 | /* check stdout */ | |
862 | exitcode = qdict_get_int(val, "exitcode"); | |
863 | g_assert_cmpint(exitcode, ==, 0); | |
864 | out = qdict_get_str(val, "out-data"); | |
865 | decoded = g_base64_decode(out, &len); | |
866 | g_assert_cmpint(len, ==, 12); | |
867 | g_assert_cmpstr((char *)decoded, ==, "\" test_str \""); | |
868 | g_free(decoded); | |
869 | QDECREF(ret); | |
870 | } | |
871 | ||
872 | static void test_qga_guest_exec_invalid(gconstpointer fix) | |
873 | { | |
874 | const TestFixture *fixture = fix; | |
875 | QDict *ret, *error; | |
876 | const gchar *class, *desc; | |
877 | ||
878 | /* invalid command */ | |
879 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-exec', 'arguments': {" | |
880 | " 'path': '/bin/invalid-cmd42' } }"); | |
881 | g_assert_nonnull(ret); | |
882 | error = qdict_get_qdict(ret, "error"); | |
883 | g_assert_nonnull(error); | |
884 | class = qdict_get_str(error, "class"); | |
885 | desc = qdict_get_str(error, "desc"); | |
886 | g_assert_cmpstr(class, ==, "GenericError"); | |
887 | g_assert_cmpint(strlen(desc), >, 0); | |
888 | QDECREF(ret); | |
889 | ||
890 | /* invalid pid */ | |
891 | ret = qmp_fd(fixture->fd, "{'execute': 'guest-exec-status'," | |
892 | " 'arguments': { 'pid': 0 } }"); | |
893 | g_assert_nonnull(ret); | |
894 | error = qdict_get_qdict(ret, "error"); | |
895 | g_assert_nonnull(error); | |
896 | class = qdict_get_str(error, "class"); | |
897 | desc = qdict_get_str(error, "desc"); | |
898 | g_assert_cmpstr(class, ==, "GenericError"); | |
899 | g_assert_cmpint(strlen(desc), >, 0); | |
900 | QDECREF(ret); | |
901 | } | |
902 | ||
62c39b30 MAL |
903 | int main(int argc, char **argv) |
904 | { | |
905 | TestFixture fix; | |
906 | int ret; | |
907 | ||
908 | setlocale (LC_ALL, ""); | |
909 | g_test_init(&argc, &argv, NULL); | |
910 | fixture_setup(&fix, NULL); | |
911 | ||
912 | g_test_add_data_func("/qga/sync-delimited", &fix, test_qga_sync_delimited); | |
913 | g_test_add_data_func("/qga/sync", &fix, test_qga_sync); | |
914 | g_test_add_data_func("/qga/ping", &fix, test_qga_ping); | |
915 | g_test_add_data_func("/qga/info", &fix, test_qga_info); | |
916 | g_test_add_data_func("/qga/network-get-interfaces", &fix, | |
917 | test_qga_network_get_interfaces); | |
918 | g_test_add_data_func("/qga/get-vcpus", &fix, test_qga_get_vcpus); | |
919 | g_test_add_data_func("/qga/get-fsinfo", &fix, test_qga_get_fsinfo); | |
920 | g_test_add_data_func("/qga/get-memory-block-info", &fix, | |
921 | test_qga_get_memory_block_info); | |
922 | g_test_add_data_func("/qga/get-memory-blocks", &fix, | |
923 | test_qga_get_memory_blocks); | |
924 | g_test_add_data_func("/qga/file-ops", &fix, test_qga_file_ops); | |
4eaab85c | 925 | g_test_add_data_func("/qga/file-write-read", &fix, test_qga_file_write_read); |
62c39b30 MAL |
926 | g_test_add_data_func("/qga/get-time", &fix, test_qga_get_time); |
927 | g_test_add_data_func("/qga/invalid-cmd", &fix, test_qga_invalid_cmd); | |
928 | g_test_add_data_func("/qga/fsfreeze-status", &fix, | |
929 | test_qga_fsfreeze_status); | |
930 | ||
931 | g_test_add_data_func("/qga/blacklist", NULL, test_qga_blacklist); | |
932 | g_test_add_data_func("/qga/config", NULL, test_qga_config); | |
3dab9fa1 MAL |
933 | g_test_add_data_func("/qga/guest-exec", &fix, test_qga_guest_exec); |
934 | g_test_add_data_func("/qga/guest-exec-invalid", &fix, | |
935 | test_qga_guest_exec_invalid); | |
62c39b30 MAL |
936 | |
937 | if (g_getenv("QGA_TEST_SIDE_EFFECTING")) { | |
938 | g_test_add_data_func("/qga/fsfreeze-and-thaw", &fix, | |
939 | test_qga_fsfreeze_and_thaw); | |
940 | g_test_add_data_func("/qga/set-time", &fix, test_qga_set_time); | |
941 | g_test_add_data_func("/qga/fstrim", &fix, test_qga_fstrim); | |
942 | } | |
943 | ||
944 | ret = g_test_run(); | |
945 | ||
946 | fixture_tear_down(&fix, NULL); | |
947 | ||
948 | return ret; | |
949 | } |