]>
Commit | Line | Data |
---|---|---|
49ee3590 AL |
1 | /* |
2 | * QTest | |
3 | * | |
4 | * Copyright IBM, Corp. 2012 | |
5 | * Copyright Red Hat, Inc. 2012 | |
872536bf | 6 | * Copyright SUSE LINUX Products GmbH 2013 |
49ee3590 AL |
7 | * |
8 | * Authors: | |
9 | * Anthony Liguori <aliguori@us.ibm.com> | |
10 | * Paolo Bonzini <pbonzini@redhat.com> | |
872536bf | 11 | * Andreas Färber <afaerber@suse.de> |
49ee3590 AL |
12 | * |
13 | * This work is licensed under the terms of the GNU GPL, version 2 or later. | |
14 | * See the COPYING file in the top-level directory. | |
49ee3590 | 15 | */ |
47e6b297 | 16 | |
681c28a3 | 17 | #include "qemu/osdep.h" |
49ee3590 | 18 | |
49ee3590 AL |
19 | #include <sys/socket.h> |
20 | #include <sys/wait.h> | |
21 | #include <sys/un.h> | |
49ee3590 | 22 | |
50990b16 | 23 | #include "libqtest.h" |
62815d85 | 24 | #include "qemu-common.h" |
856dfd8a | 25 | #include "qemu/ctype.h" |
50990b16 | 26 | #include "qemu/cutils.h" |
53f99152 | 27 | #include "qapi/error.h" |
86cdf9ec | 28 | #include "qapi/qmp/json-parser.h" |
452fcdbc | 29 | #include "qapi/qmp/qdict.h" |
563890c7 | 30 | #include "qapi/qmp/qjson.h" |
47e6b297 | 31 | #include "qapi/qmp/qlist.h" |
fc81fa1e | 32 | #include "qapi/qmp/qstring.h" |
49ee3590 AL |
33 | |
34 | #define MAX_IRQ 256 | |
f5d45791 | 35 | #define SOCKET_TIMEOUT 50 |
24d5588c | 36 | #define SOCKET_MAX_FDS 16 |
49ee3590 | 37 | |
49ee3590 AL |
38 | struct QTestState |
39 | { | |
40 | int fd; | |
a3ca163c | 41 | int qmp_fd; |
cef60c92 | 42 | pid_t qemu_pid; /* our child QEMU process */ |
21f80286 | 43 | int wstatus; |
54ce6f22 | 44 | bool big_endian; |
21f80286 RH |
45 | bool irq_level[MAX_IRQ]; |
46 | GString *rx; | |
49ee3590 AL |
47 | }; |
48 | ||
063c23d9 | 49 | static GHookList abrt_hooks; |
d7668251 SH |
50 | static struct sigaction sigact_old; |
51 | ||
54ce6f22 LV |
52 | static int qtest_query_target_endianness(QTestState *s); |
53 | ||
a3ca163c KW |
54 | static int init_socket(const char *socket_path) |
55 | { | |
56 | struct sockaddr_un addr; | |
57 | int sock; | |
58 | int ret; | |
59 | ||
60 | sock = socket(PF_UNIX, SOCK_STREAM, 0); | |
70b11059 | 61 | g_assert_cmpint(sock, !=, -1); |
a3ca163c KW |
62 | |
63 | addr.sun_family = AF_UNIX; | |
64 | snprintf(addr.sun_path, sizeof(addr.sun_path), "%s", socket_path); | |
65 | qemu_set_cloexec(sock); | |
66 | ||
67 | do { | |
68 | ret = bind(sock, (struct sockaddr *)&addr, sizeof(addr)); | |
69 | } while (ret == -1 && errno == EINTR); | |
70b11059 | 70 | g_assert_cmpint(ret, !=, -1); |
a7ded163 | 71 | ret = listen(sock, 1); |
70b11059 | 72 | g_assert_cmpint(ret, !=, -1); |
a3ca163c KW |
73 | |
74 | return sock; | |
75 | } | |
76 | ||
77 | static int socket_accept(int sock) | |
78 | { | |
79 | struct sockaddr_un addr; | |
80 | socklen_t addrlen; | |
81 | int ret; | |
f8762027 MA |
82 | struct timeval timeout = { .tv_sec = SOCKET_TIMEOUT, |
83 | .tv_usec = 0 }; | |
84 | ||
85 | setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, (void *)&timeout, | |
86 | sizeof(timeout)); | |
a3ca163c KW |
87 | |
88 | do { | |
535b4563 | 89 | addrlen = sizeof(addr); |
a3ca163c KW |
90 | ret = accept(sock, (struct sockaddr *)&addr, &addrlen); |
91 | } while (ret == -1 && errno == EINTR); | |
a7d915f3 AF |
92 | if (ret == -1) { |
93 | fprintf(stderr, "%s failed: %s\n", __func__, strerror(errno)); | |
94 | } | |
a3ca163c KW |
95 | close(sock); |
96 | ||
97 | return ret; | |
98 | } | |
99 | ||
21f80286 | 100 | bool qtest_probe_child(QTestState *s) |
b15d422a | 101 | { |
21f80286 RH |
102 | pid_t pid = s->qemu_pid; |
103 | ||
104 | if (pid != -1) { | |
105 | pid = waitpid(pid, &s->wstatus, WNOHANG); | |
106 | if (pid == 0) { | |
107 | return true; | |
108 | } | |
109 | s->qemu_pid = -1; | |
110 | } | |
111 | return false; | |
112 | } | |
fa0d421f | 113 | |
21f80286 RH |
114 | static void kill_qemu(QTestState *s) |
115 | { | |
116 | pid_t pid = s->qemu_pid; | |
117 | int wstatus; | |
71a268a5 | 118 | |
21f80286 RH |
119 | /* Skip wait if qtest_probe_child already reaped. */ |
120 | if (pid != -1) { | |
121 | kill(pid, SIGTERM); | |
122 | TFR(pid = waitpid(s->qemu_pid, &s->wstatus, 0)); | |
71a268a5 | 123 | assert(pid == s->qemu_pid); |
21f80286 RH |
124 | } |
125 | ||
126 | /* | |
127 | * We expect qemu to exit with status 0; anything else is | |
128 | * fishy and should be logged with as much detail as possible. | |
129 | */ | |
130 | wstatus = s->wstatus; | |
131 | if (wstatus) { | |
132 | if (WIFEXITED(wstatus)) { | |
133 | fprintf(stderr, "%s:%d: kill_qemu() tried to terminate QEMU " | |
134 | "process but encountered exit status %d\n", | |
135 | __FILE__, __LINE__, WEXITSTATUS(wstatus)); | |
136 | } else if (WIFSIGNALED(wstatus)) { | |
137 | int sig = WTERMSIG(wstatus); | |
138 | const char *signame = strsignal(sig) ?: "unknown ???"; | |
139 | const char *dump = WCOREDUMP(wstatus) ? " (core dumped)" : ""; | |
140 | ||
141 | fprintf(stderr, "%s:%d: kill_qemu() detected QEMU death " | |
142 | "from signal %d (%s)%s\n", | |
143 | __FILE__, __LINE__, sig, signame, dump); | |
fa0d421f | 144 | } |
21f80286 | 145 | abort(); |
b15d422a SH |
146 | } |
147 | } | |
148 | ||
041088c7 MA |
149 | static void kill_qemu_hook_func(void *s) |
150 | { | |
151 | kill_qemu(s); | |
152 | } | |
153 | ||
b15d422a SH |
154 | static void sigabrt_handler(int signo) |
155 | { | |
063c23d9 | 156 | g_hook_list_invoke(&abrt_hooks, FALSE); |
d7668251 SH |
157 | } |
158 | ||
159 | static void setup_sigabrt_handler(void) | |
160 | { | |
161 | struct sigaction sigact; | |
162 | ||
163 | /* Catch SIGABRT to clean up on g_assert() failure */ | |
164 | sigact = (struct sigaction){ | |
165 | .sa_handler = sigabrt_handler, | |
166 | .sa_flags = SA_RESETHAND, | |
167 | }; | |
168 | sigemptyset(&sigact.sa_mask); | |
169 | sigaction(SIGABRT, &sigact, &sigact_old); | |
170 | } | |
171 | ||
172 | static void cleanup_sigabrt_handler(void) | |
173 | { | |
174 | sigaction(SIGABRT, &sigact_old, NULL); | |
b15d422a SH |
175 | } |
176 | ||
041088c7 | 177 | void qtest_add_abrt_handler(GHookFunc fn, const void *data) |
063c23d9 MAL |
178 | { |
179 | GHook *hook; | |
180 | ||
181 | /* Only install SIGABRT handler once */ | |
182 | if (!abrt_hooks.is_setup) { | |
183 | g_hook_list_init(&abrt_hooks, sizeof(GHook)); | |
063c23d9 | 184 | } |
24dd1e17 | 185 | setup_sigabrt_handler(); |
063c23d9 MAL |
186 | |
187 | hook = g_hook_alloc(&abrt_hooks); | |
188 | hook->func = fn; | |
189 | hook->data = (void *)data; | |
190 | ||
191 | g_hook_prepend(&abrt_hooks, hook); | |
192 | } | |
193 | ||
db221e66 TH |
194 | static const char *qtest_qemu_binary(void) |
195 | { | |
196 | const char *qemu_bin; | |
197 | ||
198 | qemu_bin = getenv("QTEST_QEMU_BINARY"); | |
199 | if (!qemu_bin) { | |
200 | fprintf(stderr, "Environment variable QTEST_QEMU_BINARY required\n"); | |
201 | exit(1); | |
202 | } | |
203 | ||
204 | return qemu_bin; | |
205 | } | |
206 | ||
192f26a7 | 207 | QTestState *qtest_init_without_qmp_handshake(const char *extra_args) |
49ee3590 AL |
208 | { |
209 | QTestState *s; | |
8a8fd637 | 210 | int sock, qmpsock, i; |
56db2e58 SH |
211 | gchar *socket_path; |
212 | gchar *qmp_socket_path; | |
49ee3590 | 213 | gchar *command; |
db221e66 | 214 | const char *qemu_binary = qtest_qemu_binary(); |
49ee3590 | 215 | |
790bbb97 | 216 | s = g_new(QTestState, 1); |
49ee3590 | 217 | |
56db2e58 SH |
218 | socket_path = g_strdup_printf("/tmp/qtest-%d.sock", getpid()); |
219 | qmp_socket_path = g_strdup_printf("/tmp/qtest-%d.qmp", getpid()); | |
bb494a50 | 220 | |
d9123d09 PM |
221 | /* It's possible that if an earlier test run crashed it might |
222 | * have left a stale unix socket lying around. Delete any | |
223 | * stale old socket to avoid spurious test failures with | |
224 | * tests/libqtest.c:70:init_socket: assertion failed (ret != -1): (-1 != -1) | |
225 | */ | |
226 | unlink(socket_path); | |
227 | unlink(qmp_socket_path); | |
228 | ||
56db2e58 SH |
229 | sock = init_socket(socket_path); |
230 | qmpsock = init_socket(qmp_socket_path); | |
49ee3590 | 231 | |
041088c7 | 232 | qtest_add_abrt_handler(kill_qemu_hook_func, s); |
b15d422a | 233 | |
ced09f9b | 234 | command = g_strdup_printf("exec %s " |
767abe7f | 235 | "-qtest unix:%s " |
ced09f9b | 236 | "-qtest-log %s " |
767abe7f | 237 | "-chardev socket,path=%s,id=char0 " |
192f26a7 | 238 | "-mon chardev=char0,mode=control " |
ced09f9b PB |
239 | "-machine accel=qtest " |
240 | "-display none " | |
241 | "%s", qemu_binary, socket_path, | |
242 | getenv("QTEST_LOG") ? "/dev/fd/2" : "/dev/null", | |
192f26a7 | 243 | qmp_socket_path, |
ced09f9b PB |
244 | extra_args ?: ""); |
245 | ||
246 | g_test_message("starting QEMU: %s", command); | |
247 | ||
21f80286 | 248 | s->wstatus = 0; |
cef60c92 SH |
249 | s->qemu_pid = fork(); |
250 | if (s->qemu_pid == 0) { | |
6b029216 | 251 | setenv("QEMU_AUDIO_DRV", "none", true); |
8a8fd637 AL |
252 | execlp("/bin/sh", "sh", "-c", command, NULL); |
253 | exit(1); | |
49ee3590 AL |
254 | } |
255 | ||
ced09f9b | 256 | g_free(command); |
a3ca163c | 257 | s->fd = socket_accept(sock); |
f8762027 MA |
258 | if (s->fd >= 0) { |
259 | s->qmp_fd = socket_accept(qmpsock); | |
260 | } | |
56db2e58 SH |
261 | unlink(socket_path); |
262 | unlink(qmp_socket_path); | |
263 | g_free(socket_path); | |
264 | g_free(qmp_socket_path); | |
49ee3590 | 265 | |
f8762027 MA |
266 | g_assert(s->fd >= 0 && s->qmp_fd >= 0); |
267 | ||
49ee3590 | 268 | s->rx = g_string_new(""); |
49ee3590 AL |
269 | for (i = 0; i < MAX_IRQ; i++) { |
270 | s->irq_level[i] = false; | |
271 | } | |
272 | ||
e0fea6b1 | 273 | if (getenv("QTEST_STOP")) { |
1ad3c6ab | 274 | kill(s->qemu_pid, SIGSTOP); |
e0fea6b1 PB |
275 | } |
276 | ||
54ce6f22 LV |
277 | /* ask endianness of the target */ |
278 | ||
279 | s->big_endian = qtest_query_target_endianness(s); | |
280 | ||
49ee3590 AL |
281 | return s; |
282 | } | |
283 | ||
f66e7ac8 MA |
284 | QTestState *qtest_init(const char *extra_args) |
285 | { | |
192f26a7 | 286 | QTestState *s = qtest_init_without_qmp_handshake(extra_args); |
be62e172 | 287 | QDict *greeting; |
f66e7ac8 MA |
288 | |
289 | /* Read the QMP greeting and then do the handshake */ | |
be62e172 MA |
290 | greeting = qtest_qmp_receive(s); |
291 | qobject_unref(greeting); | |
055a1efc | 292 | qobject_unref(qtest_qmp(s, "{ 'execute': 'qmp_capabilities' }")); |
f66e7ac8 MA |
293 | |
294 | return s; | |
295 | } | |
296 | ||
88b988c8 | 297 | QTestState *qtest_vinitf(const char *fmt, va_list ap) |
78b27bad EB |
298 | { |
299 | char *args = g_strdup_vprintf(fmt, ap); | |
300 | QTestState *s; | |
301 | ||
88b988c8 | 302 | s = qtest_init(args); |
78b27bad | 303 | g_free(args); |
78b27bad EB |
304 | return s; |
305 | } | |
306 | ||
88b988c8 | 307 | QTestState *qtest_initf(const char *fmt, ...) |
78b27bad EB |
308 | { |
309 | va_list ap; | |
310 | QTestState *s; | |
311 | ||
312 | va_start(ap, fmt); | |
88b988c8 | 313 | s = qtest_vinitf(fmt, ap); |
78b27bad EB |
314 | va_end(ap); |
315 | return s; | |
6c90a82c JS |
316 | } |
317 | ||
318 | QTestState *qtest_init_with_serial(const char *extra_args, int *sock_fd) | |
319 | { | |
320 | int sock_fd_init; | |
321 | char *sock_path, sock_dir[] = "/tmp/qtest-serial-XXXXXX"; | |
322 | QTestState *qts; | |
323 | ||
324 | g_assert_true(mkdtemp(sock_dir) != NULL); | |
325 | sock_path = g_strdup_printf("%s/sock", sock_dir); | |
326 | ||
327 | sock_fd_init = init_socket(sock_path); | |
328 | ||
329 | qts = qtest_initf("-chardev socket,id=s0,path=%s -serial chardev:s0 %s", | |
330 | sock_path, extra_args); | |
331 | ||
332 | *sock_fd = socket_accept(sock_fd_init); | |
333 | ||
334 | unlink(sock_path); | |
335 | g_free(sock_path); | |
336 | rmdir(sock_dir); | |
337 | ||
338 | g_assert_true(*sock_fd >= 0); | |
339 | ||
340 | return qts; | |
78b27bad EB |
341 | } |
342 | ||
49ee3590 AL |
343 | void qtest_quit(QTestState *s) |
344 | { | |
063c23d9 MAL |
345 | g_hook_destroy_link(&abrt_hooks, g_hook_find_data(&abrt_hooks, TRUE, s)); |
346 | ||
d7668251 | 347 | /* Uninstall SIGABRT handler on last instance */ |
4fb609ad | 348 | cleanup_sigabrt_handler(); |
d7668251 | 349 | |
b15d422a | 350 | kill_qemu(s); |
fdd26fca MA |
351 | close(s->fd); |
352 | close(s->qmp_fd); | |
353 | g_string_free(s->rx, true); | |
fdd26fca | 354 | g_free(s); |
49ee3590 AL |
355 | } |
356 | ||
563890c7 | 357 | static void socket_send(int fd, const char *buf, size_t size) |
49ee3590 | 358 | { |
563890c7 | 359 | size_t offset; |
49ee3590 AL |
360 | |
361 | offset = 0; | |
362 | while (offset < size) { | |
363 | ssize_t len; | |
364 | ||
563890c7 | 365 | len = write(fd, buf + offset, size - offset); |
49ee3590 AL |
366 | if (len == -1 && errno == EINTR) { |
367 | continue; | |
368 | } | |
369 | ||
49ee3590 AL |
370 | g_assert_cmpint(len, >, 0); |
371 | ||
372 | offset += len; | |
373 | } | |
374 | } | |
375 | ||
563890c7 PB |
376 | static void socket_sendf(int fd, const char *fmt, va_list ap) |
377 | { | |
378 | gchar *str = g_strdup_vprintf(fmt, ap); | |
379 | size_t size = strlen(str); | |
380 | ||
381 | socket_send(fd, str, size); | |
382 | g_free(str); | |
383 | } | |
384 | ||
a3ca163c KW |
385 | static void GCC_FMT_ATTR(2, 3) qtest_sendf(QTestState *s, const char *fmt, ...) |
386 | { | |
387 | va_list ap; | |
388 | ||
389 | va_start(ap, fmt); | |
390 | socket_sendf(s->fd, fmt, ap); | |
391 | va_end(ap); | |
392 | } | |
393 | ||
24d5588c YK |
394 | /* Sends a message and file descriptors to the socket. |
395 | * It's needed for qmp-commands like getfd/add-fd */ | |
396 | static void socket_send_fds(int socket_fd, int *fds, size_t fds_num, | |
397 | const char *buf, size_t buf_size) | |
398 | { | |
399 | ssize_t ret; | |
400 | struct msghdr msg = { 0 }; | |
401 | char control[CMSG_SPACE(sizeof(int) * SOCKET_MAX_FDS)] = { 0 }; | |
402 | size_t fdsize = sizeof(int) * fds_num; | |
403 | struct cmsghdr *cmsg; | |
404 | struct iovec iov = { .iov_base = (char *)buf, .iov_len = buf_size }; | |
405 | ||
406 | msg.msg_iov = &iov; | |
407 | msg.msg_iovlen = 1; | |
408 | ||
409 | if (fds && fds_num > 0) { | |
410 | g_assert_cmpuint(fds_num, <, SOCKET_MAX_FDS); | |
411 | ||
412 | msg.msg_control = control; | |
413 | msg.msg_controllen = CMSG_SPACE(fdsize); | |
414 | ||
415 | cmsg = CMSG_FIRSTHDR(&msg); | |
416 | cmsg->cmsg_len = CMSG_LEN(fdsize); | |
417 | cmsg->cmsg_level = SOL_SOCKET; | |
418 | cmsg->cmsg_type = SCM_RIGHTS; | |
419 | memcpy(CMSG_DATA(cmsg), fds, fdsize); | |
420 | } | |
421 | ||
422 | do { | |
423 | ret = sendmsg(socket_fd, &msg, 0); | |
424 | } while (ret < 0 && errno == EINTR); | |
425 | g_assert_cmpint(ret, >, 0); | |
426 | } | |
427 | ||
49ee3590 AL |
428 | static GString *qtest_recv_line(QTestState *s) |
429 | { | |
430 | GString *line; | |
431 | size_t offset; | |
432 | char *eol; | |
433 | ||
434 | while ((eol = strchr(s->rx->str, '\n')) == NULL) { | |
435 | ssize_t len; | |
436 | char buffer[1024]; | |
437 | ||
438 | len = read(s->fd, buffer, sizeof(buffer)); | |
439 | if (len == -1 && errno == EINTR) { | |
440 | continue; | |
441 | } | |
442 | ||
443 | if (len == -1 || len == 0) { | |
444 | fprintf(stderr, "Broken pipe\n"); | |
f9e986d2 | 445 | abort(); |
49ee3590 AL |
446 | } |
447 | ||
448 | g_string_append_len(s->rx, buffer, len); | |
449 | } | |
450 | ||
451 | offset = eol - s->rx->str; | |
452 | line = g_string_new_len(s->rx->str, offset); | |
453 | g_string_erase(s->rx, 0, offset + 1); | |
454 | ||
455 | return line; | |
456 | } | |
457 | ||
458 | static gchar **qtest_rsp(QTestState *s, int expected_args) | |
459 | { | |
460 | GString *line; | |
461 | gchar **words; | |
462 | int i; | |
463 | ||
464 | redo: | |
465 | line = qtest_recv_line(s); | |
466 | words = g_strsplit(line->str, " ", 0); | |
467 | g_string_free(line, TRUE); | |
468 | ||
469 | if (strcmp(words[0], "IRQ") == 0) { | |
50990b16 EB |
470 | long irq; |
471 | int ret; | |
49ee3590 AL |
472 | |
473 | g_assert(words[1] != NULL); | |
474 | g_assert(words[2] != NULL); | |
475 | ||
50990b16 EB |
476 | ret = qemu_strtol(words[2], NULL, 0, &irq); |
477 | g_assert(!ret); | |
49ee3590 AL |
478 | g_assert_cmpint(irq, >=, 0); |
479 | g_assert_cmpint(irq, <, MAX_IRQ); | |
480 | ||
481 | if (strcmp(words[1], "raise") == 0) { | |
482 | s->irq_level[irq] = true; | |
483 | } else { | |
484 | s->irq_level[irq] = false; | |
485 | } | |
486 | ||
487 | g_strfreev(words); | |
488 | goto redo; | |
489 | } | |
490 | ||
491 | g_assert(words[0] != NULL); | |
492 | g_assert_cmpstr(words[0], ==, "OK"); | |
493 | ||
494 | if (expected_args) { | |
495 | for (i = 0; i < expected_args; i++) { | |
496 | g_assert(words[i] != NULL); | |
497 | } | |
498 | } else { | |
499 | g_strfreev(words); | |
500 | } | |
501 | ||
502 | return words; | |
503 | } | |
504 | ||
54ce6f22 LV |
505 | static int qtest_query_target_endianness(QTestState *s) |
506 | { | |
507 | gchar **args; | |
508 | int big_endian; | |
509 | ||
510 | qtest_sendf(s, "endianness\n"); | |
511 | args = qtest_rsp(s, 1); | |
512 | g_assert(strcmp(args[1], "big") == 0 || strcmp(args[1], "little") == 0); | |
513 | big_endian = strcmp(args[1], "big") == 0; | |
514 | g_strfreev(args); | |
515 | ||
516 | return big_endian; | |
517 | } | |
518 | ||
0c460dac SH |
519 | typedef struct { |
520 | JSONMessageParser parser; | |
521 | QDict *response; | |
522 | } QMPResponseParser; | |
523 | ||
62815d85 | 524 | static void qmp_response(void *opaque, QObject *obj, Error *err) |
a3ca163c | 525 | { |
62815d85 | 526 | QMPResponseParser *qmp = opaque; |
0c460dac | 527 | |
84a56f38 MA |
528 | assert(!obj != !err); |
529 | ||
530 | if (err) { | |
531 | error_prepend(&err, "QMP JSON response parsing failed: "); | |
532 | error_report_err(err); | |
f9e986d2 | 533 | abort(); |
0c460dac SH |
534 | } |
535 | ||
0c460dac | 536 | g_assert(!qmp->response); |
7dc847eb | 537 | qmp->response = qobject_to(QDict, obj); |
4d96f329 | 538 | g_assert(qmp->response); |
0c460dac SH |
539 | } |
540 | ||
dc47995e | 541 | QDict *qmp_fd_receive(int fd) |
0c460dac SH |
542 | { |
543 | QMPResponseParser qmp; | |
7c282cb4 | 544 | bool log = getenv("QTEST_LOG") != NULL; |
a3ca163c | 545 | |
0c460dac | 546 | qmp.response = NULL; |
62815d85 | 547 | json_message_parser_init(&qmp.parser, qmp_response, &qmp, NULL); |
0c460dac | 548 | while (!qmp.response) { |
a3ca163c KW |
549 | ssize_t len; |
550 | char c; | |
551 | ||
dc47995e | 552 | len = read(fd, &c, 1); |
a3ca163c KW |
553 | if (len == -1 && errno == EINTR) { |
554 | continue; | |
555 | } | |
556 | ||
039380a8 AL |
557 | if (len == -1 || len == 0) { |
558 | fprintf(stderr, "Broken pipe\n"); | |
f9e986d2 | 559 | abort(); |
039380a8 AL |
560 | } |
561 | ||
7c282cb4 PB |
562 | if (log) { |
563 | len = write(2, &c, 1); | |
564 | } | |
0c460dac | 565 | json_message_parser_feed(&qmp.parser, &c, 1); |
a3ca163c | 566 | } |
0c460dac SH |
567 | json_message_parser_destroy(&qmp.parser); |
568 | ||
569 | return qmp.response; | |
570 | } | |
571 | ||
dc47995e MAL |
572 | QDict *qtest_qmp_receive(QTestState *s) |
573 | { | |
574 | return qmp_fd_receive(s->qmp_fd); | |
575 | } | |
576 | ||
ba4ed393 JS |
577 | /** |
578 | * Allow users to send a message without waiting for the reply, | |
579 | * in the case that they choose to discard all replies up until | |
580 | * a particular EVENT is received. | |
581 | */ | |
24d5588c YK |
582 | void qmp_fd_vsend_fds(int fd, int *fds, size_t fds_num, |
583 | const char *fmt, va_list ap) | |
66e0c7b1 | 584 | { |
563890c7 PB |
585 | QObject *qobj; |
586 | ||
69f0cb66 MA |
587 | /* Going through qobject ensures we escape strings properly */ |
588 | qobj = qobject_from_vjsonf_nofail(fmt, ap); | |
563890c7 PB |
589 | |
590 | /* No need to send anything for an empty QObject. */ | |
591 | if (qobj) { | |
ae74f187 | 592 | int log = getenv("QTEST_LOG") != NULL; |
563890c7 | 593 | QString *qstr = qobject_to_json(qobj); |
13420ef8 MA |
594 | const char *str; |
595 | ||
596 | /* | |
597 | * BUG: QMP doesn't react to input until it sees a newline, an | |
598 | * object, or an array. Work-around: give it a newline. | |
599 | */ | |
600 | qstring_append_chr(qstr, '\n'); | |
601 | str = qstring_get_str(qstr); | |
563890c7 | 602 | |
ae74f187 MM |
603 | if (log) { |
604 | fprintf(stderr, "%s", str); | |
605 | } | |
563890c7 | 606 | /* Send QMP request */ |
24d5588c YK |
607 | if (fds && fds_num > 0) { |
608 | socket_send_fds(fd, fds, fds_num, str, qstring_get_length(qstr)); | |
609 | } else { | |
610 | socket_send(fd, str, qstring_get_length(qstr)); | |
611 | } | |
563890c7 | 612 | |
cb3e7f08 MAL |
613 | qobject_unref(qstr); |
614 | qobject_unref(qobj); | |
563890c7 | 615 | } |
ba4ed393 JS |
616 | } |
617 | ||
24d5588c YK |
618 | void qmp_fd_vsend(int fd, const char *fmt, va_list ap) |
619 | { | |
620 | qmp_fd_vsend_fds(fd, NULL, 0, fmt, ap); | |
621 | } | |
622 | ||
623 | void qtest_qmp_vsend_fds(QTestState *s, int *fds, size_t fds_num, | |
624 | const char *fmt, va_list ap) | |
625 | { | |
626 | qmp_fd_vsend_fds(s->qmp_fd, fds, fds_num, fmt, ap); | |
627 | } | |
628 | ||
4277f1eb | 629 | void qtest_qmp_vsend(QTestState *s, const char *fmt, va_list ap) |
dc47995e | 630 | { |
24d5588c | 631 | qmp_fd_vsend_fds(s->qmp_fd, NULL, 0, fmt, ap); |
dc47995e MAL |
632 | } |
633 | ||
634 | QDict *qmp_fdv(int fd, const char *fmt, va_list ap) | |
635 | { | |
24d5588c | 636 | qmp_fd_vsend_fds(fd, NULL, 0, fmt, ap); |
dc47995e MAL |
637 | |
638 | return qmp_fd_receive(fd); | |
639 | } | |
640 | ||
24d5588c YK |
641 | QDict *qtest_vqmp_fds(QTestState *s, int *fds, size_t fds_num, |
642 | const char *fmt, va_list ap) | |
643 | { | |
644 | qtest_qmp_vsend_fds(s, fds, fds_num, fmt, ap); | |
645 | ||
646 | /* Receive reply */ | |
647 | return qtest_qmp_receive(s); | |
648 | } | |
649 | ||
248eef02 | 650 | QDict *qtest_vqmp(QTestState *s, const char *fmt, va_list ap) |
ba4ed393 | 651 | { |
4277f1eb | 652 | qtest_qmp_vsend(s, fmt, ap); |
66e0c7b1 AF |
653 | |
654 | /* Receive reply */ | |
655 | return qtest_qmp_receive(s); | |
656 | } | |
657 | ||
dc47995e MAL |
658 | QDict *qmp_fd(int fd, const char *fmt, ...) |
659 | { | |
660 | va_list ap; | |
661 | QDict *response; | |
662 | ||
663 | va_start(ap, fmt); | |
664 | response = qmp_fdv(fd, fmt, ap); | |
665 | va_end(ap); | |
666 | return response; | |
667 | } | |
668 | ||
669 | void qmp_fd_send(int fd, const char *fmt, ...) | |
670 | { | |
671 | va_list ap; | |
672 | ||
673 | va_start(ap, fmt); | |
4277f1eb | 674 | qmp_fd_vsend(fd, fmt, ap); |
dc47995e MAL |
675 | va_end(ap); |
676 | } | |
677 | ||
24d5588c YK |
678 | QDict *qtest_qmp_fds(QTestState *s, int *fds, size_t fds_num, |
679 | const char *fmt, ...) | |
680 | { | |
681 | va_list ap; | |
682 | QDict *response; | |
683 | ||
684 | va_start(ap, fmt); | |
685 | response = qtest_vqmp_fds(s, fds, fds_num, fmt, ap); | |
686 | va_end(ap); | |
687 | return response; | |
688 | } | |
689 | ||
0c460dac SH |
690 | QDict *qtest_qmp(QTestState *s, const char *fmt, ...) |
691 | { | |
692 | va_list ap; | |
693 | QDict *response; | |
694 | ||
695 | va_start(ap, fmt); | |
248eef02 | 696 | response = qtest_vqmp(s, fmt, ap); |
0c460dac SH |
697 | va_end(ap); |
698 | return response; | |
699 | } | |
700 | ||
4277f1eb | 701 | void qtest_qmp_send(QTestState *s, const char *fmt, ...) |
ba4ed393 JS |
702 | { |
703 | va_list ap; | |
704 | ||
705 | va_start(ap, fmt); | |
4277f1eb | 706 | qtest_qmp_vsend(s, fmt, ap); |
ba4ed393 JS |
707 | va_end(ap); |
708 | } | |
709 | ||
e2f64a68 | 710 | void qmp_fd_vsend_raw(int fd, const char *fmt, va_list ap) |
aed877c5 MA |
711 | { |
712 | bool log = getenv("QTEST_LOG") != NULL; | |
e2f64a68 | 713 | char *str = g_strdup_vprintf(fmt, ap); |
aed877c5 MA |
714 | |
715 | if (log) { | |
716 | fprintf(stderr, "%s", str); | |
717 | } | |
e2f64a68 | 718 | socket_send(fd, str, strlen(str)); |
aed877c5 MA |
719 | g_free(str); |
720 | } | |
721 | ||
e2f64a68 MA |
722 | void qmp_fd_send_raw(int fd, const char *fmt, ...) |
723 | { | |
724 | va_list ap; | |
725 | ||
726 | va_start(ap, fmt); | |
727 | qmp_fd_vsend_raw(fd, fmt, ap); | |
728 | va_end(ap); | |
729 | } | |
730 | ||
731 | void qtest_qmp_send_raw(QTestState *s, const char *fmt, ...) | |
732 | { | |
733 | va_list ap; | |
734 | ||
735 | va_start(ap, fmt); | |
736 | qmp_fd_vsend_raw(s->qmp_fd, fmt, ap); | |
737 | va_end(ap); | |
738 | } | |
739 | ||
7ffe3124 | 740 | QDict *qtest_qmp_eventwait_ref(QTestState *s, const char *event) |
8fe941f7 JS |
741 | { |
742 | QDict *response; | |
743 | ||
744 | for (;;) { | |
745 | response = qtest_qmp_receive(s); | |
746 | if ((qdict_haskey(response, "event")) && | |
747 | (strcmp(qdict_get_str(response, "event"), event) == 0)) { | |
7ffe3124 | 748 | return response; |
8fe941f7 | 749 | } |
cb3e7f08 | 750 | qobject_unref(response); |
8fe941f7 JS |
751 | } |
752 | } | |
753 | ||
7ffe3124 JS |
754 | void qtest_qmp_eventwait(QTestState *s, const char *event) |
755 | { | |
756 | QDict *response; | |
757 | ||
758 | response = qtest_qmp_eventwait_ref(s, event); | |
cb3e7f08 | 759 | qobject_unref(response); |
7ffe3124 JS |
760 | } |
761 | ||
248eef02 | 762 | char *qtest_vhmp(QTestState *s, const char *fmt, va_list ap) |
5fb48d96 MA |
763 | { |
764 | char *cmd; | |
765 | QDict *resp; | |
766 | char *ret; | |
767 | ||
768 | cmd = g_strdup_vprintf(fmt, ap); | |
769 | resp = qtest_qmp(s, "{'execute': 'human-monitor-command'," | |
770 | " 'arguments': {'command-line': %s}}", | |
771 | cmd); | |
772 | ret = g_strdup(qdict_get_try_str(resp, "return")); | |
6bb87be8 TH |
773 | while (ret == NULL && qdict_get_try_str(resp, "event")) { |
774 | /* Ignore asynchronous QMP events */ | |
cb3e7f08 | 775 | qobject_unref(resp); |
6bb87be8 TH |
776 | resp = qtest_qmp_receive(s); |
777 | ret = g_strdup(qdict_get_try_str(resp, "return")); | |
778 | } | |
5fb48d96 | 779 | g_assert(ret); |
cb3e7f08 | 780 | qobject_unref(resp); |
5fb48d96 MA |
781 | g_free(cmd); |
782 | return ret; | |
783 | } | |
784 | ||
785 | char *qtest_hmp(QTestState *s, const char *fmt, ...) | |
786 | { | |
787 | va_list ap; | |
788 | char *ret; | |
789 | ||
790 | va_start(ap, fmt); | |
248eef02 | 791 | ret = qtest_vhmp(s, fmt, ap); |
5fb48d96 MA |
792 | va_end(ap); |
793 | return ret; | |
794 | } | |
8fe941f7 | 795 | |
49ee3590 AL |
796 | const char *qtest_get_arch(void) |
797 | { | |
db221e66 | 798 | const char *qemu = qtest_qemu_binary(); |
49ee3590 AL |
799 | const char *end = strrchr(qemu, '/'); |
800 | ||
801 | return end + strlen("/qemu-system-"); | |
802 | } | |
803 | ||
804 | bool qtest_get_irq(QTestState *s, int num) | |
805 | { | |
806 | /* dummy operation in order to make sure irq is up to date */ | |
807 | qtest_inb(s, 0); | |
808 | ||
809 | return s->irq_level[num]; | |
810 | } | |
811 | ||
eb062cfa MAL |
812 | void qtest_module_load(QTestState *s, const char *prefix, const char *libname) |
813 | { | |
814 | qtest_sendf(s, "module_load %s %s\n", prefix, libname); | |
815 | qtest_rsp(s, 0); | |
816 | } | |
817 | ||
49ee3590 AL |
818 | static int64_t qtest_clock_rsp(QTestState *s) |
819 | { | |
820 | gchar **words; | |
821 | int64_t clock; | |
822 | words = qtest_rsp(s, 2); | |
823 | clock = g_ascii_strtoll(words[1], NULL, 0); | |
824 | g_strfreev(words); | |
825 | return clock; | |
826 | } | |
827 | ||
828 | int64_t qtest_clock_step_next(QTestState *s) | |
829 | { | |
830 | qtest_sendf(s, "clock_step\n"); | |
831 | return qtest_clock_rsp(s); | |
832 | } | |
833 | ||
834 | int64_t qtest_clock_step(QTestState *s, int64_t step) | |
835 | { | |
836 | qtest_sendf(s, "clock_step %"PRIi64"\n", step); | |
837 | return qtest_clock_rsp(s); | |
838 | } | |
839 | ||
840 | int64_t qtest_clock_set(QTestState *s, int64_t val) | |
841 | { | |
842 | qtest_sendf(s, "clock_set %"PRIi64"\n", val); | |
843 | return qtest_clock_rsp(s); | |
844 | } | |
845 | ||
846 | void qtest_irq_intercept_out(QTestState *s, const char *qom_path) | |
847 | { | |
848 | qtest_sendf(s, "irq_intercept_out %s\n", qom_path); | |
849 | qtest_rsp(s, 0); | |
850 | } | |
851 | ||
852 | void qtest_irq_intercept_in(QTestState *s, const char *qom_path) | |
853 | { | |
854 | qtest_sendf(s, "irq_intercept_in %s\n", qom_path); | |
855 | qtest_rsp(s, 0); | |
856 | } | |
857 | ||
9813dc6a SG |
858 | void qtest_set_irq_in(QTestState *s, const char *qom_path, const char *name, |
859 | int num, int level) | |
860 | { | |
861 | if (!name) { | |
862 | name = "unnamed-gpio-in"; | |
863 | } | |
864 | qtest_sendf(s, "set_irq_in %s %s %d %d\n", qom_path, name, num, level); | |
865 | qtest_rsp(s, 0); | |
866 | } | |
867 | ||
49ee3590 AL |
868 | static void qtest_out(QTestState *s, const char *cmd, uint16_t addr, uint32_t value) |
869 | { | |
870 | qtest_sendf(s, "%s 0x%x 0x%x\n", cmd, addr, value); | |
871 | qtest_rsp(s, 0); | |
872 | } | |
873 | ||
874 | void qtest_outb(QTestState *s, uint16_t addr, uint8_t value) | |
875 | { | |
876 | qtest_out(s, "outb", addr, value); | |
877 | } | |
878 | ||
879 | void qtest_outw(QTestState *s, uint16_t addr, uint16_t value) | |
880 | { | |
881 | qtest_out(s, "outw", addr, value); | |
882 | } | |
883 | ||
884 | void qtest_outl(QTestState *s, uint16_t addr, uint32_t value) | |
885 | { | |
886 | qtest_out(s, "outl", addr, value); | |
887 | } | |
888 | ||
889 | static uint32_t qtest_in(QTestState *s, const char *cmd, uint16_t addr) | |
890 | { | |
891 | gchar **args; | |
50990b16 EB |
892 | int ret; |
893 | unsigned long value; | |
49ee3590 AL |
894 | |
895 | qtest_sendf(s, "%s 0x%x\n", cmd, addr); | |
896 | args = qtest_rsp(s, 2); | |
50990b16 EB |
897 | ret = qemu_strtoul(args[1], NULL, 0, &value); |
898 | g_assert(!ret && value <= UINT32_MAX); | |
49ee3590 AL |
899 | g_strfreev(args); |
900 | ||
901 | return value; | |
902 | } | |
903 | ||
904 | uint8_t qtest_inb(QTestState *s, uint16_t addr) | |
905 | { | |
906 | return qtest_in(s, "inb", addr); | |
907 | } | |
908 | ||
909 | uint16_t qtest_inw(QTestState *s, uint16_t addr) | |
910 | { | |
911 | return qtest_in(s, "inw", addr); | |
912 | } | |
913 | ||
914 | uint32_t qtest_inl(QTestState *s, uint16_t addr) | |
915 | { | |
916 | return qtest_in(s, "inl", addr); | |
917 | } | |
918 | ||
872536bf AF |
919 | static void qtest_write(QTestState *s, const char *cmd, uint64_t addr, |
920 | uint64_t value) | |
921 | { | |
922 | qtest_sendf(s, "%s 0x%" PRIx64 " 0x%" PRIx64 "\n", cmd, addr, value); | |
923 | qtest_rsp(s, 0); | |
924 | } | |
925 | ||
926 | void qtest_writeb(QTestState *s, uint64_t addr, uint8_t value) | |
927 | { | |
928 | qtest_write(s, "writeb", addr, value); | |
929 | } | |
930 | ||
931 | void qtest_writew(QTestState *s, uint64_t addr, uint16_t value) | |
932 | { | |
933 | qtest_write(s, "writew", addr, value); | |
934 | } | |
935 | ||
936 | void qtest_writel(QTestState *s, uint64_t addr, uint32_t value) | |
937 | { | |
938 | qtest_write(s, "writel", addr, value); | |
939 | } | |
940 | ||
941 | void qtest_writeq(QTestState *s, uint64_t addr, uint64_t value) | |
942 | { | |
943 | qtest_write(s, "writeq", addr, value); | |
944 | } | |
945 | ||
946 | static uint64_t qtest_read(QTestState *s, const char *cmd, uint64_t addr) | |
947 | { | |
948 | gchar **args; | |
50990b16 | 949 | int ret; |
872536bf AF |
950 | uint64_t value; |
951 | ||
952 | qtest_sendf(s, "%s 0x%" PRIx64 "\n", cmd, addr); | |
953 | args = qtest_rsp(s, 2); | |
50990b16 EB |
954 | ret = qemu_strtou64(args[1], NULL, 0, &value); |
955 | g_assert(!ret); | |
872536bf AF |
956 | g_strfreev(args); |
957 | ||
958 | return value; | |
959 | } | |
960 | ||
961 | uint8_t qtest_readb(QTestState *s, uint64_t addr) | |
962 | { | |
963 | return qtest_read(s, "readb", addr); | |
964 | } | |
965 | ||
966 | uint16_t qtest_readw(QTestState *s, uint64_t addr) | |
967 | { | |
968 | return qtest_read(s, "readw", addr); | |
969 | } | |
970 | ||
971 | uint32_t qtest_readl(QTestState *s, uint64_t addr) | |
972 | { | |
973 | return qtest_read(s, "readl", addr); | |
974 | } | |
975 | ||
976 | uint64_t qtest_readq(QTestState *s, uint64_t addr) | |
977 | { | |
978 | return qtest_read(s, "readq", addr); | |
979 | } | |
980 | ||
49ee3590 AL |
981 | static int hex2nib(char ch) |
982 | { | |
983 | if (ch >= '0' && ch <= '9') { | |
984 | return ch - '0'; | |
985 | } else if (ch >= 'a' && ch <= 'f') { | |
986 | return 10 + (ch - 'a'); | |
987 | } else if (ch >= 'A' && ch <= 'F') { | |
988 | return 10 + (ch - 'a'); | |
989 | } else { | |
990 | return -1; | |
991 | } | |
992 | } | |
993 | ||
994 | void qtest_memread(QTestState *s, uint64_t addr, void *data, size_t size) | |
995 | { | |
996 | uint8_t *ptr = data; | |
997 | gchar **args; | |
998 | size_t i; | |
999 | ||
204febd1 GK |
1000 | if (!size) { |
1001 | return; | |
1002 | } | |
1003 | ||
ea7697fc | 1004 | qtest_sendf(s, "read 0x%" PRIx64 " 0x%zx\n", addr, size); |
49ee3590 AL |
1005 | args = qtest_rsp(s, 2); |
1006 | ||
1007 | for (i = 0; i < size; i++) { | |
1008 | ptr[i] = hex2nib(args[1][2 + (i * 2)]) << 4; | |
1009 | ptr[i] |= hex2nib(args[1][2 + (i * 2) + 1]); | |
1010 | } | |
1011 | ||
1012 | g_strfreev(args); | |
1013 | } | |
1014 | ||
eeddd59f LV |
1015 | uint64_t qtest_rtas_call(QTestState *s, const char *name, |
1016 | uint32_t nargs, uint64_t args, | |
1017 | uint32_t nret, uint64_t ret) | |
1018 | { | |
1019 | qtest_sendf(s, "rtas %s %u 0x%"PRIx64" %u 0x%"PRIx64"\n", | |
1020 | name, nargs, args, nret, ret); | |
1021 | qtest_rsp(s, 0); | |
1022 | return 0; | |
1023 | } | |
1024 | ||
041088c7 | 1025 | void qtest_add_func(const char *str, void (*fn)(void)) |
49ee3590 AL |
1026 | { |
1027 | gchar *path = g_strdup_printf("/%s/%s", qtest_get_arch(), str); | |
1028 | g_test_add_func(path, fn); | |
a7afc6b8 | 1029 | g_free(path); |
49ee3590 AL |
1030 | } |
1031 | ||
822e36ca MAL |
1032 | void qtest_add_data_func_full(const char *str, void *data, |
1033 | void (*fn)(const void *), | |
1034 | GDestroyNotify data_free_func) | |
1035 | { | |
1036 | gchar *path = g_strdup_printf("/%s/%s", qtest_get_arch(), str); | |
822e36ca | 1037 | g_test_add_data_func_full(path, data, fn, data_free_func); |
822e36ca MAL |
1038 | g_free(path); |
1039 | } | |
1040 | ||
041088c7 MA |
1041 | void qtest_add_data_func(const char *str, const void *data, |
1042 | void (*fn)(const void *)) | |
7949c0e3 AF |
1043 | { |
1044 | gchar *path = g_strdup_printf("/%s/%s", qtest_get_arch(), str); | |
1045 | g_test_add_data_func(path, data, fn); | |
1046 | g_free(path); | |
1047 | } | |
1048 | ||
7a6a740d JS |
1049 | void qtest_bufwrite(QTestState *s, uint64_t addr, const void *data, size_t size) |
1050 | { | |
1051 | gchar *bdata; | |
1052 | ||
1053 | bdata = g_base64_encode(data, size); | |
1054 | qtest_sendf(s, "b64write 0x%" PRIx64 " 0x%zx ", addr, size); | |
1055 | socket_send(s->fd, bdata, strlen(bdata)); | |
1056 | socket_send(s->fd, "\n", 1); | |
1057 | qtest_rsp(s, 0); | |
1058 | g_free(bdata); | |
1059 | } | |
1060 | ||
1061 | void qtest_bufread(QTestState *s, uint64_t addr, void *data, size_t size) | |
1062 | { | |
1063 | gchar **args; | |
1064 | size_t len; | |
1065 | ||
1066 | qtest_sendf(s, "b64read 0x%" PRIx64 " 0x%zx\n", addr, size); | |
1067 | args = qtest_rsp(s, 2); | |
1068 | ||
1069 | g_base64_decode_inplace(args[1], &len); | |
1070 | if (size != len) { | |
1071 | fprintf(stderr, "bufread: asked for %zu bytes but decoded %zu\n", | |
1072 | size, len); | |
1073 | len = MIN(len, size); | |
1074 | } | |
1075 | ||
1076 | memcpy(data, args[1], len); | |
1077 | g_strfreev(args); | |
1078 | } | |
1079 | ||
49ee3590 AL |
1080 | void qtest_memwrite(QTestState *s, uint64_t addr, const void *data, size_t size) |
1081 | { | |
1082 | const uint8_t *ptr = data; | |
1083 | size_t i; | |
204febd1 GK |
1084 | char *enc; |
1085 | ||
1086 | if (!size) { | |
1087 | return; | |
1088 | } | |
1089 | ||
1090 | enc = g_malloc(2 * size + 1); | |
49ee3590 | 1091 | |
49ee3590 | 1092 | for (i = 0; i < size; i++) { |
5560b85a | 1093 | sprintf(&enc[i * 2], "%02x", ptr[i]); |
49ee3590 | 1094 | } |
5560b85a JS |
1095 | |
1096 | qtest_sendf(s, "write 0x%" PRIx64 " 0x%zx 0x%s\n", addr, size, enc); | |
49ee3590 | 1097 | qtest_rsp(s, 0); |
5560b85a | 1098 | g_free(enc); |
49ee3590 | 1099 | } |
0100f425 | 1100 | |
86298845 JS |
1101 | void qtest_memset(QTestState *s, uint64_t addr, uint8_t pattern, size_t size) |
1102 | { | |
4d007963 | 1103 | qtest_sendf(s, "memset 0x%" PRIx64 " 0x%zx 0x%02x\n", addr, size, pattern); |
86298845 | 1104 | qtest_rsp(s, 0); |
0100f425 PM |
1105 | } |
1106 | ||
6fc9f3d3 | 1107 | void qtest_qmp_assert_success(QTestState *qts, const char *fmt, ...) |
edbe36ad KW |
1108 | { |
1109 | va_list ap; | |
1110 | QDict *response; | |
1111 | ||
1112 | va_start(ap, fmt); | |
6fc9f3d3 | 1113 | response = qtest_vqmp(qts, fmt, ap); |
edbe36ad KW |
1114 | va_end(ap); |
1115 | ||
1116 | g_assert(response); | |
1117 | if (!qdict_haskey(response, "return")) { | |
1118 | QString *s = qobject_to_json_pretty(QOBJECT(response)); | |
1119 | g_test_message("%s", qstring_get_str(s)); | |
1120 | qobject_unref(s); | |
1121 | } | |
1122 | g_assert(qdict_haskey(response, "return")); | |
1123 | qobject_unref(response); | |
1124 | } | |
1125 | ||
54ce6f22 | 1126 | bool qtest_big_endian(QTestState *s) |
46e0cf76 | 1127 | { |
54ce6f22 | 1128 | return s->big_endian; |
46e0cf76 | 1129 | } |
02ef6e87 | 1130 | |
1f4a0d81 TH |
1131 | static bool qtest_check_machine_version(const char *mname, const char *basename, |
1132 | int major, int minor) | |
1133 | { | |
1134 | char *newname; | |
1135 | bool is_equal; | |
1136 | ||
1137 | newname = g_strdup_printf("%s-%i.%i", basename, major, minor); | |
1138 | is_equal = g_str_equal(mname, newname); | |
1139 | g_free(newname); | |
1140 | ||
1141 | return is_equal; | |
1142 | } | |
1143 | ||
1144 | static bool qtest_is_old_versioned_machine(const char *mname) | |
1145 | { | |
1146 | const char *dash = strrchr(mname, '-'); | |
1147 | const char *dot = strrchr(mname, '.'); | |
1148 | const char *chr; | |
1149 | char *bname; | |
1150 | const int major = QEMU_VERSION_MAJOR; | |
1151 | const int minor = QEMU_VERSION_MINOR; | |
1152 | bool res = false; | |
1153 | ||
1154 | if (dash && dot && dot > dash) { | |
1155 | for (chr = dash + 1; *chr; chr++) { | |
1156 | if (!qemu_isdigit(*chr) && *chr != '.') { | |
1157 | return false; | |
1158 | } | |
1159 | } | |
1160 | /* | |
1161 | * Now check if it is one of the latest versions. Check major + 1 | |
1162 | * and minor + 1 versions as well, since they might already exist | |
1163 | * in the development branch. | |
1164 | */ | |
1165 | bname = g_strdup(mname); | |
1166 | bname[dash - mname] = 0; | |
1167 | res = !qtest_check_machine_version(mname, bname, major + 1, 0) && | |
1168 | !qtest_check_machine_version(mname, bname, major, minor + 1) && | |
1169 | !qtest_check_machine_version(mname, bname, major, minor); | |
1170 | g_free(bname); | |
1171 | } | |
1172 | ||
1173 | return res; | |
1174 | } | |
1175 | ||
1176 | void qtest_cb_for_every_machine(void (*cb)(const char *machine), | |
1177 | bool skip_old_versioned) | |
02ef6e87 TH |
1178 | { |
1179 | QDict *response, *minfo; | |
1180 | QList *list; | |
1181 | const QListEntry *p; | |
1182 | QObject *qobj; | |
1183 | QString *qstr; | |
1184 | const char *mname; | |
17de4741 | 1185 | QTestState *qts; |
02ef6e87 | 1186 | |
17de4741 TH |
1187 | qts = qtest_init("-machine none"); |
1188 | response = qtest_qmp(qts, "{ 'execute': 'query-machines' }"); | |
02ef6e87 TH |
1189 | g_assert(response); |
1190 | list = qdict_get_qlist(response, "return"); | |
1191 | g_assert(list); | |
1192 | ||
1193 | for (p = qlist_first(list); p; p = qlist_next(p)) { | |
7dc847eb | 1194 | minfo = qobject_to(QDict, qlist_entry_obj(p)); |
02ef6e87 TH |
1195 | g_assert(minfo); |
1196 | qobj = qdict_get(minfo, "name"); | |
1197 | g_assert(qobj); | |
7dc847eb | 1198 | qstr = qobject_to(QString, qobj); |
02ef6e87 TH |
1199 | g_assert(qstr); |
1200 | mname = qstring_get_str(qstr); | |
1f4a0d81 TH |
1201 | if (!skip_old_versioned || !qtest_is_old_versioned_machine(mname)) { |
1202 | cb(mname); | |
1203 | } | |
02ef6e87 TH |
1204 | } |
1205 | ||
17de4741 | 1206 | qtest_quit(qts); |
cb3e7f08 | 1207 | qobject_unref(response); |
02ef6e87 | 1208 | } |
acd80015 | 1209 | |
3cd46d42 MA |
1210 | QDict *qtest_qmp_receive_success(QTestState *s, |
1211 | void (*event_cb)(void *opaque, | |
1212 | const char *event, | |
1213 | QDict *data), | |
1214 | void *opaque) | |
1215 | { | |
1216 | QDict *response, *ret, *data; | |
1217 | const char *event; | |
1218 | ||
1219 | for (;;) { | |
1220 | response = qtest_qmp_receive(s); | |
1221 | g_assert(!qdict_haskey(response, "error")); | |
1222 | ret = qdict_get_qdict(response, "return"); | |
1223 | if (ret) { | |
1224 | break; | |
1225 | } | |
1226 | event = qdict_get_str(response, "event"); | |
1227 | data = qdict_get_qdict(response, "data"); | |
1228 | if (event_cb) { | |
1229 | event_cb(opaque, event, data); | |
1230 | } | |
1231 | qobject_unref(response); | |
1232 | } | |
1233 | ||
1234 | qobject_ref(ret); | |
1235 | qobject_unref(response); | |
1236 | return ret; | |
1237 | } | |
1238 | ||
acd80015 TH |
1239 | /* |
1240 | * Generic hot-plugging test via the device_add QMP command. | |
1241 | */ | |
e5758de4 | 1242 | void qtest_qmp_device_add(QTestState *qts, const char *driver, const char *id, |
82cab70b | 1243 | const char *fmt, ...) |
acd80015 | 1244 | { |
82cab70b MA |
1245 | QDict *args, *response; |
1246 | va_list ap; | |
acd80015 | 1247 | |
82cab70b MA |
1248 | va_start(ap, fmt); |
1249 | args = qdict_from_vjsonf_nofail(fmt, ap); | |
1250 | va_end(ap); | |
acd80015 | 1251 | |
82cab70b MA |
1252 | g_assert(!qdict_haskey(args, "driver") && !qdict_haskey(args, "id")); |
1253 | qdict_put_str(args, "driver", driver); | |
1254 | qdict_put_str(args, "id", id); | |
acd80015 | 1255 | |
e5758de4 TH |
1256 | response = qtest_qmp(qts, "{'execute': 'device_add', 'arguments': %p}", |
1257 | args); | |
acd80015 TH |
1258 | g_assert(response); |
1259 | g_assert(!qdict_haskey(response, "event")); /* We don't expect any events */ | |
1260 | g_assert(!qdict_haskey(response, "error")); | |
cb3e7f08 | 1261 | qobject_unref(response); |
acd80015 TH |
1262 | } |
1263 | ||
3cd46d42 MA |
1264 | static void device_deleted_cb(void *opaque, const char *name, QDict *data) |
1265 | { | |
1266 | bool *got_event = opaque; | |
1267 | ||
1268 | g_assert_cmpstr(name, ==, "DEVICE_DELETED"); | |
1269 | *got_event = true; | |
1270 | } | |
1271 | ||
acd80015 TH |
1272 | /* |
1273 | * Generic hot-unplugging test via the device_del QMP command. | |
1274 | * Device deletion will get one response and one event. For example: | |
1275 | * | |
1276 | * {'execute': 'device_del','arguments': { 'id': 'scsi-hd'}} | |
1277 | * | |
1278 | * will get this one: | |
1279 | * | |
1280 | * {"timestamp": {"seconds": 1505289667, "microseconds": 569862}, | |
1281 | * "event": "DEVICE_DELETED", "data": {"device": "scsi-hd", | |
1282 | * "path": "/machine/peripheral/scsi-hd"}} | |
1283 | * | |
1284 | * and this one: | |
1285 | * | |
1286 | * {"return": {}} | |
1287 | * | |
1288 | * But the order of arrival may vary - so we've got to detect both. | |
1289 | */ | |
e5758de4 | 1290 | void qtest_qmp_device_del(QTestState *qts, const char *id) |
acd80015 | 1291 | { |
3cd46d42 MA |
1292 | bool got_event = false; |
1293 | QDict *rsp; | |
acd80015 | 1294 | |
e5758de4 | 1295 | qtest_qmp_send(qts, "{'execute': 'device_del', 'arguments': {'id': %s}}", |
3cd46d42 | 1296 | id); |
e5758de4 | 1297 | rsp = qtest_qmp_receive_success(qts, device_deleted_cb, &got_event); |
3cd46d42 MA |
1298 | qobject_unref(rsp); |
1299 | if (!got_event) { | |
e5758de4 | 1300 | rsp = qtest_qmp_receive(qts); |
3cd46d42 MA |
1301 | g_assert_cmpstr(qdict_get_try_str(rsp, "event"), |
1302 | ==, "DEVICE_DELETED"); | |
1303 | qobject_unref(rsp); | |
acd80015 | 1304 | } |
acd80015 | 1305 | } |
c35665e1 IM |
1306 | |
1307 | bool qmp_rsp_is_err(QDict *rsp) | |
1308 | { | |
1309 | QDict *error = qdict_get_qdict(rsp, "error"); | |
1310 | qobject_unref(rsp); | |
1311 | return !!error; | |
1312 | } | |
ebb4d82d MAL |
1313 | |
1314 | void qmp_assert_error_class(QDict *rsp, const char *class) | |
1315 | { | |
1316 | QDict *error = qdict_get_qdict(rsp, "error"); | |
1317 | ||
1318 | g_assert_cmpstr(qdict_get_try_str(error, "class"), ==, class); | |
1319 | g_assert_nonnull(qdict_get_try_str(error, "desc")); | |
1320 | g_assert(!qdict_haskey(rsp, "return")); | |
1321 | ||
1322 | qobject_unref(rsp); | |
1323 | } |