786 lines
		
	
	
		
			22 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			786 lines
		
	
	
		
			22 KiB
		
	
	
	
		
			C
		
	
	
	
| #include <locale.h>
 | |
| #include <glib.h>
 | |
| #include <glib/gstdio.h>
 | |
| #include <stdio.h>
 | |
| #include <stdlib.h>
 | |
| #include <errno.h>
 | |
| #include <string.h>
 | |
| #include <sys/types.h>
 | |
| #include <sys/socket.h>
 | |
| #include <sys/un.h>
 | |
| #include <unistd.h>
 | |
| #include <inttypes.h>
 | |
| 
 | |
| #include "libqtest.h"
 | |
| #include "config-host.h"
 | |
| 
 | |
| typedef struct {
 | |
|     char *test_dir;
 | |
|     GMainLoop *loop;
 | |
|     int fd;
 | |
|     GPid pid;
 | |
| } TestFixture;
 | |
| 
 | |
| static int connect_qga(char *path)
 | |
| {
 | |
|     int s, ret, len, i = 0;
 | |
|     struct sockaddr_un remote;
 | |
| 
 | |
|     s = socket(AF_UNIX, SOCK_STREAM, 0);
 | |
|     g_assert(s != -1);
 | |
| 
 | |
|     remote.sun_family = AF_UNIX;
 | |
|     do {
 | |
|         strcpy(remote.sun_path, path);
 | |
|         len = strlen(remote.sun_path) + sizeof(remote.sun_family);
 | |
|         ret = connect(s, (struct sockaddr *)&remote, len);
 | |
|         if (ret == -1) {
 | |
|             g_usleep(G_USEC_PER_SEC);
 | |
|         }
 | |
|         if (i++ == 10) {
 | |
|             return -1;
 | |
|         }
 | |
|     } while (ret == -1);
 | |
| 
 | |
|     return s;
 | |
| }
 | |
| 
 | |
| static void qga_watch(GPid pid, gint status, gpointer user_data)
 | |
| {
 | |
|     TestFixture *fixture = user_data;
 | |
| 
 | |
|     g_assert_cmpint(status, ==, 0);
 | |
|     g_main_loop_quit(fixture->loop);
 | |
| }
 | |
| 
 | |
| static void
 | |
| fixture_setup(TestFixture *fixture, gconstpointer data)
 | |
| {
 | |
|     const gchar *extra_arg = data;
 | |
|     GError *error = NULL;
 | |
|     gchar *cwd, *path, *cmd, **argv = NULL;
 | |
| 
 | |
|     fixture->loop = g_main_loop_new(NULL, FALSE);
 | |
| 
 | |
|     fixture->test_dir = g_strdup("/tmp/qgatest.XXXXXX");
 | |
|     g_assert_nonnull(mkdtemp(fixture->test_dir));
 | |
| 
 | |
|     path = g_build_filename(fixture->test_dir, "sock", NULL);
 | |
|     cwd = g_get_current_dir();
 | |
|     cmd = g_strdup_printf("%s%cqemu-ga -m unix-listen -t %s -p %s %s %s",
 | |
|                           cwd, G_DIR_SEPARATOR,
 | |
|                           fixture->test_dir, path,
 | |
|                           getenv("QTEST_LOG") ? "-v" : "",
 | |
|                           extra_arg ?: "");
 | |
|     g_shell_parse_argv(cmd, NULL, &argv, &error);
 | |
|     g_assert_no_error(error);
 | |
| 
 | |
|     g_spawn_async(fixture->test_dir, argv, NULL,
 | |
|                   G_SPAWN_SEARCH_PATH|G_SPAWN_DO_NOT_REAP_CHILD,
 | |
|                   NULL, NULL, &fixture->pid, &error);
 | |
|     g_assert_no_error(error);
 | |
| 
 | |
|     g_child_watch_add(fixture->pid, qga_watch, fixture);
 | |
| 
 | |
|     fixture->fd = connect_qga(path);
 | |
|     g_assert_cmpint(fixture->fd, !=, -1);
 | |
| 
 | |
|     g_strfreev(argv);
 | |
|     g_free(cmd);
 | |
|     g_free(cwd);
 | |
|     g_free(path);
 | |
| }
 | |
| 
 | |
| static void
 | |
| fixture_tear_down(TestFixture *fixture, gconstpointer data)
 | |
| {
 | |
|     gchar *tmp;
 | |
| 
 | |
|     kill(fixture->pid, SIGTERM);
 | |
| 
 | |
|     g_main_loop_run(fixture->loop);
 | |
|     g_main_loop_unref(fixture->loop);
 | |
| 
 | |
|     g_spawn_close_pid(fixture->pid);
 | |
| 
 | |
|     tmp = g_build_filename(fixture->test_dir, "foo", NULL);
 | |
|     g_unlink(tmp);
 | |
|     g_free(tmp);
 | |
| 
 | |
|     tmp = g_build_filename(fixture->test_dir, "qga.state", NULL);
 | |
|     g_unlink(tmp);
 | |
|     g_free(tmp);
 | |
| 
 | |
|     tmp = g_build_filename(fixture->test_dir, "sock", NULL);
 | |
|     g_unlink(tmp);
 | |
|     g_free(tmp);
 | |
| 
 | |
|     g_rmdir(fixture->test_dir);
 | |
|     g_free(fixture->test_dir);
 | |
| }
 | |
| 
 | |
| static void qmp_assertion_message_error(const char     *domain,
 | |
|                                         const char     *file,
 | |
|                                         int             line,
 | |
|                                         const char     *func,
 | |
|                                         const char     *expr,
 | |
|                                         QDict          *dict)
 | |
| {
 | |
|     const char *class, *desc;
 | |
|     char *s;
 | |
|     QDict *error;
 | |
| 
 | |
|     error = qdict_get_qdict(dict, "error");
 | |
|     class = qdict_get_try_str(error, "class");
 | |
|     desc = qdict_get_try_str(error, "desc");
 | |
| 
 | |
|     s = g_strdup_printf("assertion failed %s: %s %s", expr, class, desc);
 | |
|     g_assertion_message(domain, file, line, func, s);
 | |
|     g_free(s);
 | |
| }
 | |
| 
 | |
| #define qmp_assert_no_error(err) do {                                   \
 | |
|     if (qdict_haskey(err, "error")) {                                   \
 | |
|         qmp_assertion_message_error(G_LOG_DOMAIN, __FILE__, __LINE__,   \
 | |
|                                     G_STRFUNC, #err, err);              \
 | |
|     }                                                                   \
 | |
| } while (0)
 | |
| 
 | |
| static void test_qga_sync_delimited(gconstpointer fix)
 | |
| {
 | |
|     const TestFixture *fixture = fix;
 | |
|     guint32 v, r = g_random_int();
 | |
|     unsigned char c;
 | |
|     QDict *ret;
 | |
|     gchar *cmd;
 | |
| 
 | |
|     cmd = g_strdup_printf("%c{'execute': 'guest-sync-delimited',"
 | |
|                           " 'arguments': {'id': %u } }", 0xff, r);
 | |
|     qmp_fd_send(fixture->fd, cmd);
 | |
|     g_free(cmd);
 | |
| 
 | |
|     v = read(fixture->fd, &c, 1);
 | |
|     g_assert_cmpint(v, ==, 1);
 | |
|     g_assert_cmpint(c, ==, 0xff);
 | |
| 
 | |
|     ret = qmp_fd_receive(fixture->fd);
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
| 
 | |
|     v = qdict_get_int(ret, "return");
 | |
|     g_assert_cmpint(r, ==, v);
 | |
| 
 | |
|     QDECREF(ret);
 | |
| }
 | |
| 
 | |
| static void test_qga_sync(gconstpointer fix)
 | |
| {
 | |
|     const TestFixture *fixture = fix;
 | |
|     guint32 v, r = g_random_int();
 | |
|     QDict *ret;
 | |
|     gchar *cmd;
 | |
| 
 | |
|     cmd = g_strdup_printf("%c{'execute': 'guest-sync',"
 | |
|                           " 'arguments': {'id': %u } }", 0xff, r);
 | |
|     ret = qmp_fd(fixture->fd, cmd);
 | |
|     g_free(cmd);
 | |
| 
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
| 
 | |
|     v = qdict_get_int(ret, "return");
 | |
|     g_assert_cmpint(r, ==, v);
 | |
| 
 | |
|     QDECREF(ret);
 | |
| }
 | |
| 
 | |
| static void test_qga_ping(gconstpointer fix)
 | |
| {
 | |
|     const TestFixture *fixture = fix;
 | |
|     QDict *ret;
 | |
| 
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-ping'}");
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
| 
 | |
|     QDECREF(ret);
 | |
| }
 | |
| 
 | |
| static void test_qga_invalid_cmd(gconstpointer fix)
 | |
| {
 | |
|     const TestFixture *fixture = fix;
 | |
|     QDict *ret, *error;
 | |
|     const gchar *class, *desc;
 | |
| 
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-invalid-cmd'}");
 | |
|     g_assert_nonnull(ret);
 | |
| 
 | |
|     error = qdict_get_qdict(ret, "error");
 | |
|     class = qdict_get_try_str(error, "class");
 | |
|     desc = qdict_get_try_str(error, "desc");
 | |
| 
 | |
|     g_assert_cmpstr(class, ==, "CommandNotFound");
 | |
|     g_assert_cmpint(strlen(desc), >, 0);
 | |
| 
 | |
|     QDECREF(ret);
 | |
| }
 | |
| 
 | |
| static void test_qga_info(gconstpointer fix)
 | |
| {
 | |
|     const TestFixture *fixture = fix;
 | |
|     QDict *ret, *val;
 | |
|     const gchar *version;
 | |
| 
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-info'}");
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
| 
 | |
|     val = qdict_get_qdict(ret, "return");
 | |
|     version = qdict_get_try_str(val, "version");
 | |
|     g_assert_cmpstr(version, ==, QEMU_VERSION);
 | |
| 
 | |
|     QDECREF(ret);
 | |
| }
 | |
| 
 | |
| static void test_qga_get_vcpus(gconstpointer fix)
 | |
| {
 | |
|     const TestFixture *fixture = fix;
 | |
|     QDict *ret;
 | |
|     QList *list;
 | |
|     const QListEntry *entry;
 | |
| 
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-get-vcpus'}");
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
| 
 | |
|     /* check there is at least a cpu */
 | |
|     list = qdict_get_qlist(ret, "return");
 | |
|     entry = qlist_first(list);
 | |
|     g_assert(qdict_haskey(qobject_to_qdict(entry->value), "online"));
 | |
|     g_assert(qdict_haskey(qobject_to_qdict(entry->value), "logical-id"));
 | |
| 
 | |
|     QDECREF(ret);
 | |
| }
 | |
| 
 | |
| static void test_qga_get_fsinfo(gconstpointer fix)
 | |
| {
 | |
|     const TestFixture *fixture = fix;
 | |
|     QDict *ret;
 | |
|     QList *list;
 | |
|     const QListEntry *entry;
 | |
| 
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-get-fsinfo'}");
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
| 
 | |
|     /* sanity-check the response if there are any filesystems */
 | |
|     list = qdict_get_qlist(ret, "return");
 | |
|     entry = qlist_first(list);
 | |
|     if (entry) {
 | |
|         g_assert(qdict_haskey(qobject_to_qdict(entry->value), "name"));
 | |
|         g_assert(qdict_haskey(qobject_to_qdict(entry->value), "mountpoint"));
 | |
|         g_assert(qdict_haskey(qobject_to_qdict(entry->value), "type"));
 | |
|         g_assert(qdict_haskey(qobject_to_qdict(entry->value), "disk"));
 | |
|     }
 | |
| 
 | |
|     QDECREF(ret);
 | |
| }
 | |
| 
 | |
| static void test_qga_get_memory_block_info(gconstpointer fix)
 | |
| {
 | |
|     const TestFixture *fixture = fix;
 | |
|     QDict *ret, *val;
 | |
|     int64_t size;
 | |
| 
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-get-memory-block-info'}");
 | |
|     g_assert_nonnull(ret);
 | |
| 
 | |
|     /* some systems might not expose memory block info in sysfs */
 | |
|     if (!qdict_haskey(ret, "error")) {
 | |
|         /* check there is at least some memory */
 | |
|         val = qdict_get_qdict(ret, "return");
 | |
|         size = qdict_get_int(val, "size");
 | |
|         g_assert_cmpint(size, >, 0);
 | |
|     }
 | |
| 
 | |
|     QDECREF(ret);
 | |
| }
 | |
| 
 | |
| static void test_qga_get_memory_blocks(gconstpointer fix)
 | |
| {
 | |
|     const TestFixture *fixture = fix;
 | |
|     QDict *ret;
 | |
|     QList *list;
 | |
|     const QListEntry *entry;
 | |
| 
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-get-memory-blocks'}");
 | |
|     g_assert_nonnull(ret);
 | |
| 
 | |
|     /* some systems might not expose memory block info in sysfs */
 | |
|     if (!qdict_haskey(ret, "error")) {
 | |
|         list = qdict_get_qlist(ret, "return");
 | |
|         entry = qlist_first(list);
 | |
|         /* newer versions of qga may return empty list without error */
 | |
|         if (entry) {
 | |
|             g_assert(qdict_haskey(qobject_to_qdict(entry->value), "phys-index"));
 | |
|             g_assert(qdict_haskey(qobject_to_qdict(entry->value), "online"));
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     QDECREF(ret);
 | |
| }
 | |
| 
 | |
| static void test_qga_network_get_interfaces(gconstpointer fix)
 | |
| {
 | |
|     const TestFixture *fixture = fix;
 | |
|     QDict *ret;
 | |
|     QList *list;
 | |
|     const QListEntry *entry;
 | |
| 
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-network-get-interfaces'}");
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
| 
 | |
|     /* check there is at least an interface */
 | |
|     list = qdict_get_qlist(ret, "return");
 | |
|     entry = qlist_first(list);
 | |
|     g_assert(qdict_haskey(qobject_to_qdict(entry->value), "name"));
 | |
| 
 | |
|     QDECREF(ret);
 | |
| }
 | |
| 
 | |
| static void test_qga_file_ops(gconstpointer fix)
 | |
| {
 | |
|     const TestFixture *fixture = fix;
 | |
|     const guchar helloworld[] = "Hello World!\n";
 | |
|     const char *b64;
 | |
|     gchar *cmd, *path, *enc;
 | |
|     guchar *dec;
 | |
|     QDict *ret, *val;
 | |
|     int64_t id, eof;
 | |
|     gsize count;
 | |
|     FILE *f;
 | |
|     char tmp[100];
 | |
| 
 | |
|     /* open */
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-file-open',"
 | |
|                  " 'arguments': { 'path': 'foo', 'mode': 'w+' } }");
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
|     id = qdict_get_int(ret, "return");
 | |
|     QDECREF(ret);
 | |
| 
 | |
|     enc = g_base64_encode(helloworld, sizeof(helloworld));
 | |
|     /* write */
 | |
|     cmd = g_strdup_printf("{'execute': 'guest-file-write',"
 | |
|                           " 'arguments': { 'handle': %" PRId64 ","
 | |
|                           " 'buf-b64': '%s' } }", id, enc);
 | |
|     ret = qmp_fd(fixture->fd, cmd);
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
| 
 | |
|     val = qdict_get_qdict(ret, "return");
 | |
|     count = qdict_get_int(val, "count");
 | |
|     eof = qdict_get_bool(val, "eof");
 | |
|     g_assert_cmpint(count, ==, sizeof(helloworld));
 | |
|     g_assert_cmpint(eof, ==, 0);
 | |
|     QDECREF(ret);
 | |
|     g_free(cmd);
 | |
| 
 | |
|     /* flush */
 | |
|     cmd = g_strdup_printf("{'execute': 'guest-file-flush',"
 | |
|                           " 'arguments': {'handle': %" PRId64 "} }",
 | |
|                           id);
 | |
|     ret = qmp_fd(fixture->fd, cmd);
 | |
|     QDECREF(ret);
 | |
|     g_free(cmd);
 | |
| 
 | |
|     /* close */
 | |
|     cmd = g_strdup_printf("{'execute': 'guest-file-close',"
 | |
|                           " 'arguments': {'handle': %" PRId64 "} }",
 | |
|                           id);
 | |
|     ret = qmp_fd(fixture->fd, cmd);
 | |
|     QDECREF(ret);
 | |
|     g_free(cmd);
 | |
| 
 | |
|     /* check content */
 | |
|     path = g_build_filename(fixture->test_dir, "foo", NULL);
 | |
|     f = fopen(path, "r");
 | |
|     g_assert_nonnull(f);
 | |
|     count = fread(tmp, 1, sizeof(tmp), f);
 | |
|     g_assert_cmpint(count, ==, sizeof(helloworld));
 | |
|     tmp[count] = 0;
 | |
|     g_assert_cmpstr(tmp, ==, (char *)helloworld);
 | |
|     fclose(f);
 | |
| 
 | |
|     /* open */
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-file-open',"
 | |
|                  " 'arguments': { 'path': 'foo', 'mode': 'r' } }");
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
|     id = qdict_get_int(ret, "return");
 | |
|     QDECREF(ret);
 | |
| 
 | |
|     /* read */
 | |
|     cmd = g_strdup_printf("{'execute': 'guest-file-read',"
 | |
|                           " 'arguments': { 'handle': %" PRId64 "} }",
 | |
|                           id);
 | |
|     ret = qmp_fd(fixture->fd, cmd);
 | |
|     val = qdict_get_qdict(ret, "return");
 | |
|     count = qdict_get_int(val, "count");
 | |
|     eof = qdict_get_bool(val, "eof");
 | |
|     b64 = qdict_get_str(val, "buf-b64");
 | |
|     g_assert_cmpint(count, ==, sizeof(helloworld));
 | |
|     g_assert(eof);
 | |
|     g_assert_cmpstr(b64, ==, enc);
 | |
| 
 | |
|     QDECREF(ret);
 | |
|     g_free(cmd);
 | |
|     g_free(enc);
 | |
| 
 | |
|     /* read eof */
 | |
|     cmd = g_strdup_printf("{'execute': 'guest-file-read',"
 | |
|                           " 'arguments': { 'handle': %" PRId64 "} }",
 | |
|                           id);
 | |
|     ret = qmp_fd(fixture->fd, cmd);
 | |
|     val = qdict_get_qdict(ret, "return");
 | |
|     count = qdict_get_int(val, "count");
 | |
|     eof = qdict_get_bool(val, "eof");
 | |
|     b64 = qdict_get_str(val, "buf-b64");
 | |
|     g_assert_cmpint(count, ==, 0);
 | |
|     g_assert(eof);
 | |
|     g_assert_cmpstr(b64, ==, "");
 | |
|     QDECREF(ret);
 | |
|     g_free(cmd);
 | |
| 
 | |
|     /* seek */
 | |
|     cmd = g_strdup_printf("{'execute': 'guest-file-seek',"
 | |
|                           " 'arguments': { 'handle': %" PRId64 ", "
 | |
|                           " 'offset': %d, 'whence': %d } }",
 | |
|                           id, 6, SEEK_SET);
 | |
|     ret = qmp_fd(fixture->fd, cmd);
 | |
|     qmp_assert_no_error(ret);
 | |
|     val = qdict_get_qdict(ret, "return");
 | |
|     count = qdict_get_int(val, "position");
 | |
|     eof = qdict_get_bool(val, "eof");
 | |
|     g_assert_cmpint(count, ==, 6);
 | |
|     g_assert(!eof);
 | |
|     QDECREF(ret);
 | |
|     g_free(cmd);
 | |
| 
 | |
|     /* partial read */
 | |
|     cmd = g_strdup_printf("{'execute': 'guest-file-read',"
 | |
|                           " 'arguments': { 'handle': %" PRId64 "} }",
 | |
|                           id);
 | |
|     ret = qmp_fd(fixture->fd, cmd);
 | |
|     val = qdict_get_qdict(ret, "return");
 | |
|     count = qdict_get_int(val, "count");
 | |
|     eof = qdict_get_bool(val, "eof");
 | |
|     b64 = qdict_get_str(val, "buf-b64");
 | |
|     g_assert_cmpint(count, ==, sizeof(helloworld) - 6);
 | |
|     g_assert(eof);
 | |
|     dec = g_base64_decode(b64, &count);
 | |
|     g_assert_cmpint(count, ==, sizeof(helloworld) - 6);
 | |
|     g_assert_cmpmem(dec, count, helloworld + 6, sizeof(helloworld) - 6);
 | |
|     g_free(dec);
 | |
| 
 | |
|     QDECREF(ret);
 | |
|     g_free(cmd);
 | |
| 
 | |
|     /* close */
 | |
|     cmd = g_strdup_printf("{'execute': 'guest-file-close',"
 | |
|                           " 'arguments': {'handle': %" PRId64 "} }",
 | |
|                           id);
 | |
|     ret = qmp_fd(fixture->fd, cmd);
 | |
|     QDECREF(ret);
 | |
|     g_free(cmd);
 | |
| }
 | |
| 
 | |
| static void test_qga_get_time(gconstpointer fix)
 | |
| {
 | |
|     const TestFixture *fixture = fix;
 | |
|     QDict *ret;
 | |
|     int64_t time;
 | |
| 
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-get-time'}");
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
| 
 | |
|     time = qdict_get_int(ret, "return");
 | |
|     g_assert_cmpint(time, >, 0);
 | |
| 
 | |
|     QDECREF(ret);
 | |
| }
 | |
| 
 | |
| static void test_qga_set_time(gconstpointer fix)
 | |
| {
 | |
|     const TestFixture *fixture = fix;
 | |
|     QDict *ret;
 | |
|     int64_t current, time;
 | |
|     gchar *cmd;
 | |
| 
 | |
|     /* get current time */
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-get-time'}");
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
|     current = qdict_get_int(ret, "return");
 | |
|     g_assert_cmpint(current, >, 0);
 | |
|     QDECREF(ret);
 | |
| 
 | |
|     /* set some old time */
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-set-time',"
 | |
|                  " 'arguments': { 'time': 1000 } }");
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
|     QDECREF(ret);
 | |
| 
 | |
|     /* check old time */
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-get-time'}");
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
|     time = qdict_get_int(ret, "return");
 | |
|     g_assert_cmpint(time / 1000, <, G_USEC_PER_SEC * 10);
 | |
|     QDECREF(ret);
 | |
| 
 | |
|     /* set back current time */
 | |
|     cmd = g_strdup_printf("{'execute': 'guest-set-time',"
 | |
|                           " 'arguments': { 'time': %" PRId64 " } }",
 | |
|                           current + time * 1000);
 | |
|     ret = qmp_fd(fixture->fd, cmd);
 | |
|     g_free(cmd);
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
|     QDECREF(ret);
 | |
| }
 | |
| 
 | |
| static void test_qga_fstrim(gconstpointer fix)
 | |
| {
 | |
|     const TestFixture *fixture = fix;
 | |
|     QDict *ret;
 | |
|     QList *list;
 | |
|     const QListEntry *entry;
 | |
| 
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-fstrim',"
 | |
|                  " arguments: { minimum: 4194304 } }");
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
|     list = qdict_get_qlist(ret, "return");
 | |
|     entry = qlist_first(list);
 | |
|     g_assert(qdict_haskey(qobject_to_qdict(entry->value), "paths"));
 | |
| 
 | |
|     QDECREF(ret);
 | |
| }
 | |
| 
 | |
| static void test_qga_blacklist(gconstpointer data)
 | |
| {
 | |
|     TestFixture fix;
 | |
|     QDict *ret, *error;
 | |
|     const gchar *class, *desc;
 | |
| 
 | |
|     fixture_setup(&fix, "-b guest-ping,guest-get-time");
 | |
| 
 | |
|     /* check blacklist */
 | |
|     ret = qmp_fd(fix.fd, "{'execute': 'guest-ping'}");
 | |
|     g_assert_nonnull(ret);
 | |
|     error = qdict_get_qdict(ret, "error");
 | |
|     class = qdict_get_try_str(error, "class");
 | |
|     desc = qdict_get_try_str(error, "desc");
 | |
|     g_assert_cmpstr(class, ==, "GenericError");
 | |
|     g_assert_nonnull(g_strstr_len(desc, -1, "has been disabled"));
 | |
|     QDECREF(ret);
 | |
| 
 | |
|     ret = qmp_fd(fix.fd, "{'execute': 'guest-get-time'}");
 | |
|     g_assert_nonnull(ret);
 | |
|     error = qdict_get_qdict(ret, "error");
 | |
|     class = qdict_get_try_str(error, "class");
 | |
|     desc = qdict_get_try_str(error, "desc");
 | |
|     g_assert_cmpstr(class, ==, "GenericError");
 | |
|     g_assert_nonnull(g_strstr_len(desc, -1, "has been disabled"));
 | |
|     QDECREF(ret);
 | |
| 
 | |
|     /* check something work */
 | |
|     ret = qmp_fd(fix.fd, "{'execute': 'guest-get-fsinfo'}");
 | |
|     qmp_assert_no_error(ret);
 | |
|     QDECREF(ret);
 | |
| 
 | |
|     fixture_tear_down(&fix, NULL);
 | |
| }
 | |
| 
 | |
| static void test_qga_config(gconstpointer data)
 | |
| {
 | |
|     GError *error = NULL;
 | |
|     char *cwd, *cmd, *out, *err, *str, **strv, *conf, **argv = NULL;
 | |
|     char *env[2];
 | |
|     int status, tmp;
 | |
|     gsize n;
 | |
|     GKeyFile *kf;
 | |
|     const char *qga_config =
 | |
|         "[general]\n"
 | |
|         "daemon=false\n"
 | |
|         "method=virtio-serial\n"
 | |
|         "path=/path/to/org.qemu.guest_agent.0\n"
 | |
|         "pidfile=/var/foo/qemu-ga.pid\n"
 | |
|         "statedir=/var/state\n"
 | |
|         "verbose=true\n"
 | |
|         "blacklist=guest-ping;guest-get-time\n";
 | |
| 
 | |
|     tmp = g_file_open_tmp(NULL, &conf, &error);
 | |
|     g_assert_no_error(error);
 | |
|     g_assert_cmpint(tmp, >=, 0);
 | |
|     g_assert_cmpstr(conf, !=, "");
 | |
| 
 | |
|     g_file_set_contents(conf, qga_config, -1, &error);
 | |
|     g_assert_no_error(error);
 | |
| 
 | |
|     cwd = g_get_current_dir();
 | |
|     cmd = g_strdup_printf("%s%cqemu-ga -D",
 | |
|                           cwd, G_DIR_SEPARATOR);
 | |
|     g_shell_parse_argv(cmd, NULL, &argv, &error);
 | |
|     g_assert_no_error(error);
 | |
| 
 | |
|     env[0] = g_strdup_printf("QGA_CONF=%s", conf);
 | |
|     env[1] = NULL;
 | |
|     g_spawn_sync(NULL, argv, env, 0,
 | |
|                  NULL, NULL, &out, &err, &status, &error);
 | |
|     g_assert_no_error(error);
 | |
|     g_assert_cmpstr(err, ==, "");
 | |
|     g_assert_cmpint(status, ==, 0);
 | |
| 
 | |
|     kf = g_key_file_new();
 | |
|     g_key_file_load_from_data(kf, out, -1, G_KEY_FILE_NONE, &error);
 | |
|     g_assert_no_error(error);
 | |
| 
 | |
|     str = g_key_file_get_start_group(kf);
 | |
|     g_assert_cmpstr(str, ==, "general");
 | |
|     g_free(str);
 | |
| 
 | |
|     g_assert_false(g_key_file_get_boolean(kf, "general", "daemon", &error));
 | |
|     g_assert_no_error(error);
 | |
| 
 | |
|     str = g_key_file_get_string(kf, "general", "method", &error);
 | |
|     g_assert_no_error(error);
 | |
|     g_assert_cmpstr(str, ==, "virtio-serial");
 | |
|     g_free(str);
 | |
| 
 | |
|     str = g_key_file_get_string(kf, "general", "path", &error);
 | |
|     g_assert_no_error(error);
 | |
|     g_assert_cmpstr(str, ==, "/path/to/org.qemu.guest_agent.0");
 | |
|     g_free(str);
 | |
| 
 | |
|     str = g_key_file_get_string(kf, "general", "pidfile", &error);
 | |
|     g_assert_no_error(error);
 | |
|     g_assert_cmpstr(str, ==, "/var/foo/qemu-ga.pid");
 | |
|     g_free(str);
 | |
| 
 | |
|     str = g_key_file_get_string(kf, "general", "statedir", &error);
 | |
|     g_assert_no_error(error);
 | |
|     g_assert_cmpstr(str, ==, "/var/state");
 | |
|     g_free(str);
 | |
| 
 | |
|     g_assert_true(g_key_file_get_boolean(kf, "general", "verbose", &error));
 | |
|     g_assert_no_error(error);
 | |
| 
 | |
|     strv = g_key_file_get_string_list(kf, "general", "blacklist", &n, &error);
 | |
|     g_assert_cmpint(n, ==, 2);
 | |
| #if GLIB_CHECK_VERSION(2, 44, 0)
 | |
|     g_assert_true(g_strv_contains((const char * const *)strv,
 | |
|                                   "guest-ping"));
 | |
|     g_assert_true(g_strv_contains((const char * const *)strv,
 | |
|                                   "guest-get-time"));
 | |
| #endif
 | |
|     g_assert_no_error(error);
 | |
|     g_strfreev(strv);
 | |
| 
 | |
|     g_free(out);
 | |
|     g_free(err);
 | |
|     g_free(conf);
 | |
|     g_free(env[0]);
 | |
|     g_key_file_free(kf);
 | |
| 
 | |
|     close(tmp);
 | |
| }
 | |
| 
 | |
| static void test_qga_fsfreeze_status(gconstpointer fix)
 | |
| {
 | |
|     const TestFixture *fixture = fix;
 | |
|     QDict *ret;
 | |
|     const gchar *status;
 | |
| 
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-fsfreeze-status'}");
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
| 
 | |
|     status = qdict_get_try_str(ret, "return");
 | |
|     g_assert_cmpstr(status, ==, "thawed");
 | |
| 
 | |
|     QDECREF(ret);
 | |
| }
 | |
| 
 | |
| static void test_qga_fsfreeze_and_thaw(gconstpointer fix)
 | |
| {
 | |
|     const TestFixture *fixture = fix;
 | |
|     QDict *ret;
 | |
|     const gchar *status;
 | |
| 
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-fsfreeze-freeze'}");
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
|     QDECREF(ret);
 | |
| 
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-fsfreeze-status'}");
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
|     status = qdict_get_try_str(ret, "return");
 | |
|     g_assert_cmpstr(status, ==, "frozen");
 | |
|     QDECREF(ret);
 | |
| 
 | |
|     ret = qmp_fd(fixture->fd, "{'execute': 'guest-fsfreeze-thaw'}");
 | |
|     g_assert_nonnull(ret);
 | |
|     qmp_assert_no_error(ret);
 | |
|     QDECREF(ret);
 | |
| }
 | |
| 
 | |
| int main(int argc, char **argv)
 | |
| {
 | |
|     TestFixture fix;
 | |
|     int ret;
 | |
| 
 | |
|     setlocale (LC_ALL, "");
 | |
|     g_test_init(&argc, &argv, NULL);
 | |
|     fixture_setup(&fix, NULL);
 | |
| 
 | |
|     g_test_add_data_func("/qga/sync-delimited", &fix, test_qga_sync_delimited);
 | |
|     g_test_add_data_func("/qga/sync", &fix, test_qga_sync);
 | |
|     g_test_add_data_func("/qga/ping", &fix, test_qga_ping);
 | |
|     g_test_add_data_func("/qga/info", &fix, test_qga_info);
 | |
|     g_test_add_data_func("/qga/network-get-interfaces", &fix,
 | |
|                          test_qga_network_get_interfaces);
 | |
|     g_test_add_data_func("/qga/get-vcpus", &fix, test_qga_get_vcpus);
 | |
|     g_test_add_data_func("/qga/get-fsinfo", &fix, test_qga_get_fsinfo);
 | |
|     g_test_add_data_func("/qga/get-memory-block-info", &fix,
 | |
|                          test_qga_get_memory_block_info);
 | |
|     g_test_add_data_func("/qga/get-memory-blocks", &fix,
 | |
|                          test_qga_get_memory_blocks);
 | |
|     g_test_add_data_func("/qga/file-ops", &fix, test_qga_file_ops);
 | |
|     g_test_add_data_func("/qga/get-time", &fix, test_qga_get_time);
 | |
|     g_test_add_data_func("/qga/invalid-cmd", &fix, test_qga_invalid_cmd);
 | |
|     g_test_add_data_func("/qga/fsfreeze-status", &fix,
 | |
|                          test_qga_fsfreeze_status);
 | |
| 
 | |
|     g_test_add_data_func("/qga/blacklist", NULL, test_qga_blacklist);
 | |
|     g_test_add_data_func("/qga/config", NULL, test_qga_config);
 | |
| 
 | |
|     if (g_getenv("QGA_TEST_SIDE_EFFECTING")) {
 | |
|         g_test_add_data_func("/qga/fsfreeze-and-thaw", &fix,
 | |
|                              test_qga_fsfreeze_and_thaw);
 | |
|         g_test_add_data_func("/qga/set-time", &fix, test_qga_set_time);
 | |
|         g_test_add_data_func("/qga/fstrim", &fix, test_qga_fstrim);
 | |
|     }
 | |
| 
 | |
|     ret = g_test_run();
 | |
| 
 | |
|     fixture_tear_down(&fix, NULL);
 | |
| 
 | |
|     return ret;
 | |
| }
 |