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