xref: /openbmc/qemu/tests/unit/test-qga.c (revision 5e437d3c)
1 #include "qemu/osdep.h"
2 #include <locale.h>
3 #include <glib/gstdio.h>
4 #include <sys/socket.h>
5 #include <sys/un.h>
6 
7 #include "../qtest/libqos/libqtest.h"
8 #include "qapi/qmp/qdict.h"
9 #include "qapi/qmp/qlist.h"
10 
11 typedef struct {
12     char *test_dir;
13     GMainLoop *loop;
14     int fd;
15     GPid pid;
16 } TestFixture;
17 
18 static 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 
42 static 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 
50 static void
51 fixture_setup(TestFixture *fixture, gconstpointer data, gchar **envp)
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();
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,
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 
72     g_spawn_async(fixture->test_dir, argv, envp,
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 
88 static void
89 fixture_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);
114     close(fixture->fd);
115 }
116 
117 static 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 
144 static void test_qga_sync_delimited(gconstpointer fix)
145 {
146     const TestFixture *fixture = fix;
147     guint32 v, r = g_test_rand_int();
148     unsigned char c;
149     QDict *ret;
150 
151     qmp_fd_send_raw(fixture->fd, "\xff");
152     qmp_fd_send(fixture->fd,
153                 "{'execute': 'guest-sync-delimited',"
154                 " 'arguments': {'id': %u } }",
155                 r);
156 
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);
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 
184     qobject_unref(ret);
185 }
186 
187 static void test_qga_sync(gconstpointer fix)
188 {
189     const TestFixture *fixture = fix;
190     guint32 v, r = g_test_rand_int();
191     QDict *ret;
192 
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      */
204     ret = qmp_fd(fixture->fd,
205                  "{'execute': 'guest-sync', 'arguments': {'id': %u } }",
206                  r);
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 
214     qobject_unref(ret);
215 }
216 
217 static 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 
226     qobject_unref(ret);
227 }
228 
229 static void test_qga_id(gconstpointer fix)
230 {
231     const TestFixture *fixture = fix;
232     QDict *ret;
233 
234     ret = qmp_fd(fixture->fd, "{'execute': 'guest-ping', 'id': 1}");
235     g_assert_nonnull(ret);
236     qmp_assert_no_error(ret);
237     g_assert_cmpint(qdict_get_int(ret, "id"), ==, 1);
238 
239     qobject_unref(ret);
240 }
241 
242 static void test_qga_invalid_oob(gconstpointer fix)
243 {
244     const TestFixture *fixture = fix;
245     QDict *ret;
246 
247     ret = qmp_fd(fixture->fd, "{'exec-oob': 'guest-ping'}");
248     g_assert_nonnull(ret);
249 
250     qmp_expect_error_and_unref(ret, "GenericError");
251 }
252 
253 static 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");
268     g_assert_cmpstr(desc, ==, "Parameter 'foo' is unexpected");
269 
270     qobject_unref(ret);
271 }
272 
273 static 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 
289     qobject_unref(ret);
290 }
291 
292 static 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 
306     qobject_unref(ret);
307 }
308 
309 static 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);
323     g_assert(qdict_haskey(qobject_to(QDict, entry->value), "online"));
324     g_assert(qdict_haskey(qobject_to(QDict, entry->value), "logical-id"));
325 
326     qobject_unref(ret);
327 }
328 
329 static 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 
340     /* sanity-check the response if there are any filesystems */
341     list = qdict_get_qlist(ret, "return");
342     entry = qlist_first(list);
343     if (entry) {
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"));
348     }
349 
350     qobject_unref(ret);
351 }
352 
353 static 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 
370     qobject_unref(ret);
371 }
372 
373 static 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) {
389             g_assert(qdict_haskey(qobject_to(QDict, entry->value),
390                                   "phys-index"));
391             g_assert(qdict_haskey(qobject_to(QDict, entry->value), "online"));
392         }
393     }
394 
395     qobject_unref(ret);
396 }
397 
398 static 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);
412     g_assert(qdict_haskey(qobject_to(QDict, entry->value), "name"));
413 
414     qobject_unref(ret);
415 }
416 
417 static void test_qga_file_ops(gconstpointer fix)
418 {
419     const TestFixture *fixture = fix;
420     const unsigned char helloworld[] = "Hello World!\n";
421     const char *b64;
422     gchar *path, *enc;
423     unsigned char *dec;
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");
436     qobject_unref(ret);
437 
438     enc = g_base64_encode(helloworld, sizeof(helloworld));
439     /* write */
440     ret = qmp_fd(fixture->fd,
441                  "{'execute': 'guest-file-write',"
442                  " 'arguments': { 'handle': %" PRId64 ", 'buf-b64': %s } }",
443                  id, enc);
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);
452     qobject_unref(ret);
453 
454     /* flush */
455     ret = qmp_fd(fixture->fd,
456                  "{'execute': 'guest-file-flush',"
457                  " 'arguments': {'handle': %" PRId64 "} }",
458                  id);
459     qobject_unref(ret);
460 
461     /* close */
462     ret = qmp_fd(fixture->fd,
463                  "{'execute': 'guest-file-close',"
464                  " 'arguments': {'handle': %" PRId64 "} }",
465                  id);
466     qobject_unref(ret);
467 
468     /* check content */
469     path = g_build_filename(fixture->test_dir, "foo", NULL);
470     f = fopen(path, "r");
471     g_free(path);
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");
485     qobject_unref(ret);
486 
487     /* read */
488     ret = qmp_fd(fixture->fd,
489                  "{'execute': 'guest-file-read',"
490                  " 'arguments': { 'handle': %" PRId64 "} }",
491                  id);
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 
500     qobject_unref(ret);
501     g_free(enc);
502 
503     /* read eof */
504     ret = qmp_fd(fixture->fd,
505                  "{'execute': 'guest-file-read',"
506                  " 'arguments': { 'handle': %" PRId64 "} }",
507                  id);
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, ==, "");
515     qobject_unref(ret);
516 
517     /* seek */
518     ret = qmp_fd(fixture->fd,
519                  "{'execute': 'guest-file-seek',"
520                  " 'arguments': { 'handle': %" PRId64 ", "
521                  " 'offset': %d, 'whence': %s } }",
522                  id, 6, "set");
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);
529     qobject_unref(ret);
530 
531     /* partial read */
532     ret = qmp_fd(fixture->fd,
533                  "{'execute': 'guest-file-read',"
534                  " 'arguments': { 'handle': %" PRId64 "} }",
535                  id);
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 
547     qobject_unref(ret);
548 
549     /* close */
550     ret = qmp_fd(fixture->fd,
551                  "{'execute': 'guest-file-close',"
552                  " 'arguments': {'handle': %" PRId64 "} }",
553                  id);
554     qobject_unref(ret);
555 }
556 
557 static 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;
562     gchar *enc;
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");
573     qobject_unref(ret);
574 
575     enc = g_base64_encode(helloworld, sizeof(helloworld));
576     /* write */
577     ret = qmp_fd(fixture->fd,
578                  "{'execute': 'guest-file-write',"
579                  " 'arguments': { 'handle': %" PRId64 ","
580                  " 'buf-b64': %s } }", id, enc);
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);
589     qobject_unref(ret);
590 
591     /* read (check implicit flush) */
592     ret = qmp_fd(fixture->fd,
593                  "{'execute': 'guest-file-read',"
594                  " 'arguments': { 'handle': %" PRId64 "} }",
595                  id);
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, ==, "");
603     qobject_unref(ret);
604 
605     /* seek to 0 */
606     ret = qmp_fd(fixture->fd,
607                  "{'execute': 'guest-file-seek',"
608                  " 'arguments': { 'handle': %" PRId64 ", "
609                  " 'offset': %d, 'whence': %s } }",
610                  id, 0, "set");
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);
617     qobject_unref(ret);
618 
619     /* read */
620     ret = qmp_fd(fixture->fd,
621                  "{'execute': 'guest-file-read',"
622                  " 'arguments': { 'handle': %" PRId64 "} }",
623                  id);
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);
631     qobject_unref(ret);
632     g_free(enc);
633 
634     /* close */
635     ret = qmp_fd(fixture->fd,
636                  "{'execute': 'guest-file-close',"
637                  " 'arguments': {'handle': %" PRId64 "} }",
638                  id);
639     qobject_unref(ret);
640 }
641 
642 static 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 
655     qobject_unref(ret);
656 }
657 
658 static void test_qga_blacklist(gconstpointer data)
659 {
660     TestFixture fix;
661     QDict *ret, *error;
662     const gchar *class, *desc;
663 
664     fixture_setup(&fix, "-b guest-ping,guest-get-time", NULL);
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");
672     g_assert_cmpstr(class, ==, "CommandNotFound");
673     g_assert_nonnull(g_strstr_len(desc, -1, "has been disabled"));
674     qobject_unref(ret);
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");
681     g_assert_cmpstr(class, ==, "CommandNotFound");
682     g_assert_nonnull(g_strstr_len(desc, -1, "has been disabled"));
683     qobject_unref(ret);
684 
685     /* check something work */
686     ret = qmp_fd(fix.fd, "{'execute': 'guest-get-fsinfo'}");
687     qmp_assert_no_error(ret);
688     qobject_unref(ret);
689 
690     fixture_tear_down(&fix, NULL);
691 }
692 
693 static void test_qga_config(gconstpointer data)
694 {
695     GError *error = NULL;
696     char *cwd, *cmd, *out, *err, *str, **strv, **argv = NULL;
697     char *env[2];
698     int status;
699     gsize n;
700     GKeyFile *kf;
701 
702     cwd = g_get_current_dir();
703     cmd = g_strdup_printf("%s%cqga%cqemu-ga -D",
704                           cwd, G_DIR_SEPARATOR, G_DIR_SEPARATOR);
705     g_free(cwd);
706     g_shell_parse_argv(cmd, NULL, &argv, &error);
707     g_free(cmd);
708     g_assert_no_error(error);
709 
710     env[0] = g_strdup_printf("QGA_CONF=tests%cdata%ctest-qga-config",
711                              G_DIR_SEPARATOR, G_DIR_SEPARATOR);
712     env[1] = NULL;
713     g_spawn_sync(NULL, argv, env, 0,
714                  NULL, NULL, &out, &err, &status, &error);
715     g_strfreev(argv);
716 
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);
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"));
761     g_assert_no_error(error);
762     g_strfreev(strv);
763 
764     g_free(out);
765     g_free(err);
766     g_free(env[0]);
767     g_key_file_free(kf);
768 }
769 
770 static 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 
783     qobject_unref(ret);
784 }
785 
786 static 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);
805     qobject_unref(ret);
806 
807     /* wait for completion */
808     now = g_get_monotonic_time();
809     do {
810         ret = qmp_fd(fixture->fd,
811                      "{'execute': 'guest-exec-status',"
812                      " 'arguments': { 'pid': %" PRId64 " } }", pid);
813         g_assert_nonnull(ret);
814         val = qdict_get_qdict(ret, "return");
815         exited = qdict_get_bool(val, "exited");
816         if (!exited) {
817             qobject_unref(ret);
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);
831     qobject_unref(ret);
832 }
833 
834 static 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);
850     qobject_unref(ret);
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);
862     qobject_unref(ret);
863 }
864 
865 static 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 
880 static 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 
896 static 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 
913 static 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 
962     qobject_unref(ret);
963     g_free(env[0]);
964     fixture_tear_down(&fixture, NULL);
965 }
966 
967 int main(int argc, char **argv)
968 {
969     TestFixture fix;
970     int ret;
971 
972     setlocale (LC_ALL, "");
973     g_test_init(&argc, &argv, NULL);
974     fixture_setup(&fix, NULL, NULL);
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);
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     }
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);
991     g_test_add_data_func("/qga/file-write-read", &fix, test_qga_file_write_read);
992     g_test_add_data_func("/qga/get-time", &fix, test_qga_get_time);
993     g_test_add_data_func("/qga/id", &fix, test_qga_id);
994     g_test_add_data_func("/qga/invalid-oob", &fix, test_qga_invalid_oob);
995     g_test_add_data_func("/qga/invalid-cmd", &fix, test_qga_invalid_cmd);
996     g_test_add_data_func("/qga/invalid-args", &fix, test_qga_invalid_args);
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);
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);
1005     g_test_add_data_func("/qga/guest-get-osinfo", &fix,
1006                          test_qga_guest_get_osinfo);
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);
1013 
1014     ret = g_test_run();
1015 
1016     fixture_tear_down(&fix, NULL);
1017 
1018     return ret;
1019 }
1020