qapi: Improve reporting of invalid flags
[qemu/ar7.git] / tests / libqos / libqos.c
blobd71557c5cb629da5b5540ada368e1068c76588ca
1 #include "qemu/osdep.h"
2 #include <sys/wait.h>
4 #include "libqtest.h"
5 #include "libqos/libqos.h"
6 #include "libqos/pci.h"
7 #include "qapi/qmp/qdict.h"
9 /*** Test Setup & Teardown ***/
11 /**
12 * Launch QEMU with the given command line,
13 * and then set up interrupts and our guest malloc interface.
14 * Never returns NULL:
15 * Terminates the application in case an error is encountered.
17 QOSState *qtest_vboot(QOSOps *ops, const char *cmdline_fmt, va_list ap)
19 char *cmdline;
21 QOSState *qs = g_new0(QOSState, 1);
23 cmdline = g_strdup_vprintf(cmdline_fmt, ap);
24 qs->qts = qtest_init(cmdline);
25 qs->ops = ops;
26 if (ops) {
27 ops->alloc_init(&qs->alloc, qs->qts, ALLOC_NO_FLAGS);
28 qs->pcibus = ops->qpci_new(qs->qts, &qs->alloc);
31 g_free(cmdline);
32 return qs;
35 /**
36 * Launch QEMU with the given command line,
37 * and then set up interrupts and our guest malloc interface.
39 QOSState *qtest_boot(QOSOps *ops, const char *cmdline_fmt, ...)
41 QOSState *qs;
42 va_list ap;
44 va_start(ap, cmdline_fmt);
45 qs = qtest_vboot(ops, cmdline_fmt, ap);
46 va_end(ap);
48 return qs;
51 /**
52 * Tear down the QEMU instance.
54 void qtest_common_shutdown(QOSState *qs)
56 if (qs->ops) {
57 if (qs->pcibus && qs->ops->qpci_free) {
58 qs->ops->qpci_free(qs->pcibus);
59 qs->pcibus = NULL;
62 alloc_destroy(&qs->alloc);
63 qtest_quit(qs->qts);
64 g_free(qs);
67 void qtest_shutdown(QOSState *qs)
69 if (qs->ops && qs->ops->shutdown) {
70 qs->ops->shutdown(qs);
71 } else {
72 qtest_common_shutdown(qs);
76 static QDict *qmp_execute(QTestState *qts, const char *command)
78 return qtest_qmp(qts, "{ 'execute': %s }", command);
81 void migrate(QOSState *from, QOSState *to, const char *uri)
83 const char *st;
84 QDict *rsp, *sub;
85 bool running;
87 /* Is the machine currently running? */
88 rsp = qmp_execute(from->qts, "query-status");
89 g_assert(qdict_haskey(rsp, "return"));
90 sub = qdict_get_qdict(rsp, "return");
91 g_assert(qdict_haskey(sub, "running"));
92 running = qdict_get_bool(sub, "running");
93 qobject_unref(rsp);
95 /* Issue the migrate command. */
96 rsp = qtest_qmp(from->qts,
97 "{ 'execute': 'migrate', 'arguments': { 'uri': %s }}",
98 uri);
99 g_assert(qdict_haskey(rsp, "return"));
100 qobject_unref(rsp);
102 /* Wait for STOP event, but only if we were running: */
103 if (running) {
104 qtest_qmp_eventwait(from->qts, "STOP");
107 /* If we were running, we can wait for an event. */
108 if (running) {
109 migrate_allocator(&from->alloc, &to->alloc);
110 qtest_qmp_eventwait(to->qts, "RESUME");
111 return;
114 /* Otherwise, we need to wait: poll until migration is completed. */
115 while (1) {
116 rsp = qmp_execute(from->qts, "query-migrate");
117 g_assert(qdict_haskey(rsp, "return"));
118 sub = qdict_get_qdict(rsp, "return");
119 g_assert(qdict_haskey(sub, "status"));
120 st = qdict_get_str(sub, "status");
122 /* "setup", "active", "completed", "failed", "cancelled" */
123 if (strcmp(st, "completed") == 0) {
124 qobject_unref(rsp);
125 break;
128 if ((strcmp(st, "setup") == 0) || (strcmp(st, "active") == 0)) {
129 qobject_unref(rsp);
130 g_usleep(5000);
131 continue;
134 fprintf(stderr, "Migration did not complete, status: %s\n", st);
135 g_assert_not_reached();
138 migrate_allocator(&from->alloc, &to->alloc);
141 bool have_qemu_img(void)
143 char *rpath;
144 const char *path = getenv("QTEST_QEMU_IMG");
145 if (!path) {
146 return false;
149 rpath = realpath(path, NULL);
150 if (!rpath) {
151 return false;
152 } else {
153 free(rpath);
154 return true;
158 void mkimg(const char *file, const char *fmt, unsigned size_mb)
160 gchar *cli;
161 bool ret;
162 int rc;
163 GError *err = NULL;
164 char *qemu_img_path;
165 gchar *out, *out2;
166 char *qemu_img_abs_path;
168 qemu_img_path = getenv("QTEST_QEMU_IMG");
169 g_assert(qemu_img_path);
170 qemu_img_abs_path = realpath(qemu_img_path, NULL);
171 g_assert(qemu_img_abs_path);
173 cli = g_strdup_printf("%s create -f %s %s %uM", qemu_img_abs_path,
174 fmt, file, size_mb);
175 ret = g_spawn_command_line_sync(cli, &out, &out2, &rc, &err);
176 if (err || !g_spawn_check_exit_status(rc, &err)) {
177 fprintf(stderr, "%s\n", err->message);
178 g_error_free(err);
180 g_assert(ret && !err);
182 g_free(out);
183 g_free(out2);
184 g_free(cli);
185 free(qemu_img_abs_path);
188 void mkqcow2(const char *file, unsigned size_mb)
190 return mkimg(file, "qcow2", size_mb);
193 void prepare_blkdebug_script(const char *debug_fn, const char *event)
195 FILE *debug_file = fopen(debug_fn, "w");
196 int ret;
198 fprintf(debug_file, "[inject-error]\n");
199 fprintf(debug_file, "event = \"%s\"\n", event);
200 fprintf(debug_file, "errno = \"5\"\n");
201 fprintf(debug_file, "state = \"1\"\n");
202 fprintf(debug_file, "immediately = \"off\"\n");
203 fprintf(debug_file, "once = \"on\"\n");
205 fprintf(debug_file, "[set-state]\n");
206 fprintf(debug_file, "event = \"%s\"\n", event);
207 fprintf(debug_file, "new_state = \"2\"\n");
208 fflush(debug_file);
209 g_assert(!ferror(debug_file));
211 ret = fclose(debug_file);
212 g_assert(ret == 0);
215 void generate_pattern(void *buffer, size_t len, size_t cycle_len)
217 int i, j;
218 unsigned char *tx = (unsigned char *)buffer;
219 unsigned char p;
220 size_t *sx;
222 /* Write an indicative pattern that varies and is unique per-cycle */
223 p = rand() % 256;
224 for (i = 0; i < len; i++) {
225 tx[i] = p++ % 256;
226 if (i % cycle_len == 0) {
227 p = rand() % 256;
231 /* force uniqueness by writing an id per-cycle */
232 for (i = 0; i < len / cycle_len; i++) {
233 j = i * cycle_len;
234 if (j + sizeof(*sx) <= len) {
235 sx = (size_t *)&tx[j];
236 *sx = i;