1 #include "qemu/osdep.h"
6 #include "libqos/libqos.h"
7 #include "libqos/pci.h"
9 /*** Test Setup & Teardown ***/
12 * Launch QEMU with the given command line,
13 * and then set up interrupts and our guest malloc interface.
15 QOSState *qtest_vboot(QOSOps *ops, const char *cmdline_fmt, va_list ap)
19 struct QOSState *qs = g_malloc(sizeof(QOSState));
21 cmdline = g_strdup_vprintf(cmdline_fmt, ap);
22 qs->qts = qtest_start(cmdline);
24 qtest_irq_intercept_in(global_qtest, "ioapic");
25 if (ops && ops->init_allocator) {
26 qs->alloc = ops->init_allocator(ALLOC_NO_FLAGS);
34 * Launch QEMU with the given command line,
35 * and then set up interrupts and our guest malloc interface.
37 QOSState *qtest_boot(QOSOps *ops, const char *cmdline_fmt, ...)
42 va_start(ap, cmdline_fmt);
43 qs = qtest_vboot(ops, cmdline_fmt, ap);
50 * Tear down the QEMU instance.
52 void qtest_shutdown(QOSState *qs)
54 if (qs->alloc && qs->ops && qs->ops->uninit_allocator) {
55 qs->ops->uninit_allocator(qs->alloc);
62 void set_context(QOSState *s)
64 global_qtest = s->qts;
67 static QDict *qmp_execute(const char *command)
72 fmt = g_strdup_printf("{ 'execute': '%s' }", command);
79 void migrate(QOSState *from, QOSState *to, const char *uri)
88 /* Is the machine currently running? */
89 rsp = qmp_execute("query-status");
90 g_assert(qdict_haskey(rsp, "return"));
91 sub = qdict_get_qdict(rsp, "return");
92 g_assert(qdict_haskey(sub, "running"));
93 running = qdict_get_bool(sub, "running");
96 /* Issue the migrate command. */
97 s = g_strdup_printf("{ 'execute': 'migrate',"
98 "'arguments': { 'uri': '%s' } }",
102 g_assert(qdict_haskey(rsp, "return"));
105 /* Wait for STOP event, but only if we were running: */
107 qmp_eventwait("STOP");
110 /* If we were running, we can wait for an event. */
112 migrate_allocator(from->alloc, to->alloc);
114 qmp_eventwait("RESUME");
118 /* Otherwise, we need to wait: poll until migration is completed. */
120 rsp = qmp_execute("query-migrate");
121 g_assert(qdict_haskey(rsp, "return"));
122 sub = qdict_get_qdict(rsp, "return");
123 g_assert(qdict_haskey(sub, "status"));
124 st = qdict_get_str(sub, "status");
126 /* "setup", "active", "completed", "failed", "cancelled" */
127 if (strcmp(st, "completed") == 0) {
132 if ((strcmp(st, "setup") == 0) || (strcmp(st, "active") == 0)) {
138 fprintf(stderr, "Migration did not complete, status: %s\n", st);
139 g_assert_not_reached();
142 migrate_allocator(from->alloc, to->alloc);
146 bool have_qemu_img(void)
149 const char *path = getenv("QTEST_QEMU_IMG");
154 rpath = realpath(path, NULL);
163 void mkimg(const char *file, const char *fmt, unsigned size_mb)
171 char *qemu_img_abs_path;
173 qemu_img_path = getenv("QTEST_QEMU_IMG");
174 g_assert(qemu_img_path);
175 qemu_img_abs_path = realpath(qemu_img_path, NULL);
176 g_assert(qemu_img_abs_path);
178 cli = g_strdup_printf("%s create -f %s %s %uM", qemu_img_abs_path,
180 ret = g_spawn_command_line_sync(cli, &out, &out2, &rc, &err);
182 fprintf(stderr, "%s\n", err->message);
185 g_assert(ret && !err);
187 /* In glib 2.34, we have g_spawn_check_exit_status. in 2.12, we don't.
188 * glib 2.43.91 implementation assumes that any non-zero is an error for
189 * windows, but uses extra precautions for Linux. However,
190 * 0 is only possible if the program exited normally, so that should be
191 * sufficient for our purposes on all platforms, here. */
193 fprintf(stderr, "qemu-img returned status code %d\n", rc);
200 free(qemu_img_abs_path);
203 void mkqcow2(const char *file, unsigned size_mb)
205 return mkimg(file, "qcow2", size_mb);
208 void prepare_blkdebug_script(const char *debug_fn, const char *event)
210 FILE *debug_file = fopen(debug_fn, "w");
213 fprintf(debug_file, "[inject-error]\n");
214 fprintf(debug_file, "event = \"%s\"\n", event);
215 fprintf(debug_file, "errno = \"5\"\n");
216 fprintf(debug_file, "state = \"1\"\n");
217 fprintf(debug_file, "immediately = \"off\"\n");
218 fprintf(debug_file, "once = \"on\"\n");
220 fprintf(debug_file, "[set-state]\n");
221 fprintf(debug_file, "event = \"%s\"\n", event);
222 fprintf(debug_file, "new_state = \"2\"\n");
224 g_assert(!ferror(debug_file));
226 ret = fclose(debug_file);
230 void generate_pattern(void *buffer, size_t len, size_t cycle_len)
233 unsigned char *tx = (unsigned char *)buffer;
237 /* Write an indicative pattern that varies and is unique per-cycle */
239 for (i = 0; i < len; i++) {
241 if (i % cycle_len == 0) {
246 /* force uniqueness by writing an id per-cycle */
247 for (i = 0; i < len / cycle_len; i++) {
249 if (j + sizeof(*sx) <= len) {
250 sx = (size_t *)&tx[j];