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