]> git.proxmox.com Git - mirror_qemu.git/blame - tests/libqtest.c
bios-tables-test: ignore identical binaries
[mirror_qemu.git] / tests / libqtest.c
CommitLineData
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"
50990b16 25#include "qemu/cutils.h"
53f99152 26#include "qapi/error.h"
86cdf9ec 27#include "qapi/qmp/json-parser.h"
452fcdbc 28#include "qapi/qmp/qdict.h"
563890c7 29#include "qapi/qmp/qjson.h"
47e6b297 30#include "qapi/qmp/qlist.h"
fc81fa1e 31#include "qapi/qmp/qstring.h"
49ee3590
AL
32
33#define MAX_IRQ 256
f5d45791 34#define SOCKET_TIMEOUT 50
49ee3590
AL
35
36QTestState *global_qtest;
37
38struct 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 49static GHookList abrt_hooks;
d7668251
SH
50static struct sigaction sigact_old;
51
54ce6f22
LV
52static int qtest_query_target_endianness(QTestState *s);
53
a3ca163c
KW
54static 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
77static 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 100bool 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
114static 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
149static void kill_qemu_hook_func(void *s)
150{
151 kill_qemu(s);
152}
153
b15d422a
SH
154static void sigabrt_handler(int signo)
155{
063c23d9 156 g_hook_list_invoke(&abrt_hooks, FALSE);
d7668251
SH
157}
158
159static 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
172static void cleanup_sigabrt_handler(void)
173{
174 sigaction(SIGABRT, &sigact_old, NULL);
b15d422a
SH
175}
176
041088c7 177void 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
194static 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 207QTestState *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
284QTestState *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 297QTestState *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 307QTestState *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
318QTestState *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
343void 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 357static 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
376static 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
385static 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
49ee3590
AL
394static GString *qtest_recv_line(QTestState *s)
395{
396 GString *line;
397 size_t offset;
398 char *eol;
399
400 while ((eol = strchr(s->rx->str, '\n')) == NULL) {
401 ssize_t len;
402 char buffer[1024];
403
404 len = read(s->fd, buffer, sizeof(buffer));
405 if (len == -1 && errno == EINTR) {
406 continue;
407 }
408
409 if (len == -1 || len == 0) {
410 fprintf(stderr, "Broken pipe\n");
f9e986d2 411 abort();
49ee3590
AL
412 }
413
414 g_string_append_len(s->rx, buffer, len);
415 }
416
417 offset = eol - s->rx->str;
418 line = g_string_new_len(s->rx->str, offset);
419 g_string_erase(s->rx, 0, offset + 1);
420
421 return line;
422}
423
424static gchar **qtest_rsp(QTestState *s, int expected_args)
425{
426 GString *line;
427 gchar **words;
428 int i;
429
430redo:
431 line = qtest_recv_line(s);
432 words = g_strsplit(line->str, " ", 0);
433 g_string_free(line, TRUE);
434
435 if (strcmp(words[0], "IRQ") == 0) {
50990b16
EB
436 long irq;
437 int ret;
49ee3590
AL
438
439 g_assert(words[1] != NULL);
440 g_assert(words[2] != NULL);
441
50990b16
EB
442 ret = qemu_strtol(words[2], NULL, 0, &irq);
443 g_assert(!ret);
49ee3590
AL
444 g_assert_cmpint(irq, >=, 0);
445 g_assert_cmpint(irq, <, MAX_IRQ);
446
447 if (strcmp(words[1], "raise") == 0) {
448 s->irq_level[irq] = true;
449 } else {
450 s->irq_level[irq] = false;
451 }
452
453 g_strfreev(words);
454 goto redo;
455 }
456
457 g_assert(words[0] != NULL);
458 g_assert_cmpstr(words[0], ==, "OK");
459
460 if (expected_args) {
461 for (i = 0; i < expected_args; i++) {
462 g_assert(words[i] != NULL);
463 }
464 } else {
465 g_strfreev(words);
466 }
467
468 return words;
469}
470
54ce6f22
LV
471static int qtest_query_target_endianness(QTestState *s)
472{
473 gchar **args;
474 int big_endian;
475
476 qtest_sendf(s, "endianness\n");
477 args = qtest_rsp(s, 1);
478 g_assert(strcmp(args[1], "big") == 0 || strcmp(args[1], "little") == 0);
479 big_endian = strcmp(args[1], "big") == 0;
480 g_strfreev(args);
481
482 return big_endian;
483}
484
0c460dac
SH
485typedef struct {
486 JSONMessageParser parser;
487 QDict *response;
488} QMPResponseParser;
489
62815d85 490static void qmp_response(void *opaque, QObject *obj, Error *err)
a3ca163c 491{
62815d85 492 QMPResponseParser *qmp = opaque;
0c460dac 493
84a56f38
MA
494 assert(!obj != !err);
495
496 if (err) {
497 error_prepend(&err, "QMP JSON response parsing failed: ");
498 error_report_err(err);
f9e986d2 499 abort();
0c460dac
SH
500 }
501
0c460dac 502 g_assert(!qmp->response);
7dc847eb 503 qmp->response = qobject_to(QDict, obj);
4d96f329 504 g_assert(qmp->response);
0c460dac
SH
505}
506
dc47995e 507QDict *qmp_fd_receive(int fd)
0c460dac
SH
508{
509 QMPResponseParser qmp;
7c282cb4 510 bool log = getenv("QTEST_LOG") != NULL;
a3ca163c 511
0c460dac 512 qmp.response = NULL;
62815d85 513 json_message_parser_init(&qmp.parser, qmp_response, &qmp, NULL);
0c460dac 514 while (!qmp.response) {
a3ca163c
KW
515 ssize_t len;
516 char c;
517
dc47995e 518 len = read(fd, &c, 1);
a3ca163c
KW
519 if (len == -1 && errno == EINTR) {
520 continue;
521 }
522
039380a8
AL
523 if (len == -1 || len == 0) {
524 fprintf(stderr, "Broken pipe\n");
f9e986d2 525 abort();
039380a8
AL
526 }
527
7c282cb4
PB
528 if (log) {
529 len = write(2, &c, 1);
530 }
0c460dac 531 json_message_parser_feed(&qmp.parser, &c, 1);
a3ca163c 532 }
0c460dac
SH
533 json_message_parser_destroy(&qmp.parser);
534
535 return qmp.response;
536}
537
dc47995e
MAL
538QDict *qtest_qmp_receive(QTestState *s)
539{
540 return qmp_fd_receive(s->qmp_fd);
541}
542
ba4ed393
JS
543/**
544 * Allow users to send a message without waiting for the reply,
545 * in the case that they choose to discard all replies up until
546 * a particular EVENT is received.
547 */
4277f1eb 548void qmp_fd_vsend(int fd, const char *fmt, va_list ap)
66e0c7b1 549{
563890c7
PB
550 QObject *qobj;
551
69f0cb66
MA
552 /* Going through qobject ensures we escape strings properly */
553 qobj = qobject_from_vjsonf_nofail(fmt, ap);
563890c7
PB
554
555 /* No need to send anything for an empty QObject. */
556 if (qobj) {
ae74f187 557 int log = getenv("QTEST_LOG") != NULL;
563890c7 558 QString *qstr = qobject_to_json(qobj);
13420ef8
MA
559 const char *str;
560
561 /*
562 * BUG: QMP doesn't react to input until it sees a newline, an
563 * object, or an array. Work-around: give it a newline.
564 */
565 qstring_append_chr(qstr, '\n');
566 str = qstring_get_str(qstr);
563890c7 567
ae74f187
MM
568 if (log) {
569 fprintf(stderr, "%s", str);
570 }
563890c7 571 /* Send QMP request */
13420ef8 572 socket_send(fd, str, qstring_get_length(qstr));
563890c7 573
cb3e7f08
MAL
574 qobject_unref(qstr);
575 qobject_unref(qobj);
563890c7 576 }
ba4ed393
JS
577}
578
4277f1eb 579void qtest_qmp_vsend(QTestState *s, const char *fmt, va_list ap)
dc47995e 580{
4277f1eb 581 qmp_fd_vsend(s->qmp_fd, fmt, ap);
dc47995e
MAL
582}
583
584QDict *qmp_fdv(int fd, const char *fmt, va_list ap)
585{
4277f1eb 586 qmp_fd_vsend(fd, fmt, ap);
dc47995e
MAL
587
588 return qmp_fd_receive(fd);
589}
590
248eef02 591QDict *qtest_vqmp(QTestState *s, const char *fmt, va_list ap)
ba4ed393 592{
4277f1eb 593 qtest_qmp_vsend(s, fmt, ap);
66e0c7b1
AF
594
595 /* Receive reply */
596 return qtest_qmp_receive(s);
597}
598
dc47995e
MAL
599QDict *qmp_fd(int fd, const char *fmt, ...)
600{
601 va_list ap;
602 QDict *response;
603
604 va_start(ap, fmt);
605 response = qmp_fdv(fd, fmt, ap);
606 va_end(ap);
607 return response;
608}
609
610void qmp_fd_send(int fd, const char *fmt, ...)
611{
612 va_list ap;
613
614 va_start(ap, fmt);
4277f1eb 615 qmp_fd_vsend(fd, fmt, ap);
dc47995e
MAL
616 va_end(ap);
617}
618
0c460dac
SH
619QDict *qtest_qmp(QTestState *s, const char *fmt, ...)
620{
621 va_list ap;
622 QDict *response;
623
624 va_start(ap, fmt);
248eef02 625 response = qtest_vqmp(s, fmt, ap);
0c460dac
SH
626 va_end(ap);
627 return response;
628}
629
4277f1eb 630void qtest_qmp_send(QTestState *s, const char *fmt, ...)
ba4ed393
JS
631{
632 va_list ap;
633
634 va_start(ap, fmt);
4277f1eb 635 qtest_qmp_vsend(s, fmt, ap);
ba4ed393
JS
636 va_end(ap);
637}
638
e2f64a68 639void qmp_fd_vsend_raw(int fd, const char *fmt, va_list ap)
aed877c5
MA
640{
641 bool log = getenv("QTEST_LOG") != NULL;
e2f64a68 642 char *str = g_strdup_vprintf(fmt, ap);
aed877c5
MA
643
644 if (log) {
645 fprintf(stderr, "%s", str);
646 }
e2f64a68 647 socket_send(fd, str, strlen(str));
aed877c5
MA
648 g_free(str);
649}
650
e2f64a68
MA
651void qmp_fd_send_raw(int fd, const char *fmt, ...)
652{
653 va_list ap;
654
655 va_start(ap, fmt);
656 qmp_fd_vsend_raw(fd, fmt, ap);
657 va_end(ap);
658}
659
660void qtest_qmp_send_raw(QTestState *s, const char *fmt, ...)
661{
662 va_list ap;
663
664 va_start(ap, fmt);
665 qmp_fd_vsend_raw(s->qmp_fd, fmt, ap);
666 va_end(ap);
667}
668
7ffe3124 669QDict *qtest_qmp_eventwait_ref(QTestState *s, const char *event)
8fe941f7
JS
670{
671 QDict *response;
672
673 for (;;) {
674 response = qtest_qmp_receive(s);
675 if ((qdict_haskey(response, "event")) &&
676 (strcmp(qdict_get_str(response, "event"), event) == 0)) {
7ffe3124 677 return response;
8fe941f7 678 }
cb3e7f08 679 qobject_unref(response);
8fe941f7
JS
680 }
681}
682
7ffe3124
JS
683void qtest_qmp_eventwait(QTestState *s, const char *event)
684{
685 QDict *response;
686
687 response = qtest_qmp_eventwait_ref(s, event);
cb3e7f08 688 qobject_unref(response);
7ffe3124
JS
689}
690
248eef02 691char *qtest_vhmp(QTestState *s, const char *fmt, va_list ap)
5fb48d96
MA
692{
693 char *cmd;
694 QDict *resp;
695 char *ret;
696
697 cmd = g_strdup_vprintf(fmt, ap);
698 resp = qtest_qmp(s, "{'execute': 'human-monitor-command',"
699 " 'arguments': {'command-line': %s}}",
700 cmd);
701 ret = g_strdup(qdict_get_try_str(resp, "return"));
6bb87be8
TH
702 while (ret == NULL && qdict_get_try_str(resp, "event")) {
703 /* Ignore asynchronous QMP events */
cb3e7f08 704 qobject_unref(resp);
6bb87be8
TH
705 resp = qtest_qmp_receive(s);
706 ret = g_strdup(qdict_get_try_str(resp, "return"));
707 }
5fb48d96 708 g_assert(ret);
cb3e7f08 709 qobject_unref(resp);
5fb48d96
MA
710 g_free(cmd);
711 return ret;
712}
713
714char *qtest_hmp(QTestState *s, const char *fmt, ...)
715{
716 va_list ap;
717 char *ret;
718
719 va_start(ap, fmt);
248eef02 720 ret = qtest_vhmp(s, fmt, ap);
5fb48d96
MA
721 va_end(ap);
722 return ret;
723}
8fe941f7 724
49ee3590
AL
725const char *qtest_get_arch(void)
726{
db221e66 727 const char *qemu = qtest_qemu_binary();
49ee3590
AL
728 const char *end = strrchr(qemu, '/');
729
730 return end + strlen("/qemu-system-");
731}
732
733bool qtest_get_irq(QTestState *s, int num)
734{
735 /* dummy operation in order to make sure irq is up to date */
736 qtest_inb(s, 0);
737
738 return s->irq_level[num];
739}
740
741static int64_t qtest_clock_rsp(QTestState *s)
742{
743 gchar **words;
744 int64_t clock;
745 words = qtest_rsp(s, 2);
746 clock = g_ascii_strtoll(words[1], NULL, 0);
747 g_strfreev(words);
748 return clock;
749}
750
751int64_t qtest_clock_step_next(QTestState *s)
752{
753 qtest_sendf(s, "clock_step\n");
754 return qtest_clock_rsp(s);
755}
756
757int64_t qtest_clock_step(QTestState *s, int64_t step)
758{
759 qtest_sendf(s, "clock_step %"PRIi64"\n", step);
760 return qtest_clock_rsp(s);
761}
762
763int64_t qtest_clock_set(QTestState *s, int64_t val)
764{
765 qtest_sendf(s, "clock_set %"PRIi64"\n", val);
766 return qtest_clock_rsp(s);
767}
768
769void qtest_irq_intercept_out(QTestState *s, const char *qom_path)
770{
771 qtest_sendf(s, "irq_intercept_out %s\n", qom_path);
772 qtest_rsp(s, 0);
773}
774
775void qtest_irq_intercept_in(QTestState *s, const char *qom_path)
776{
777 qtest_sendf(s, "irq_intercept_in %s\n", qom_path);
778 qtest_rsp(s, 0);
779}
780
9813dc6a
SG
781void qtest_set_irq_in(QTestState *s, const char *qom_path, const char *name,
782 int num, int level)
783{
784 if (!name) {
785 name = "unnamed-gpio-in";
786 }
787 qtest_sendf(s, "set_irq_in %s %s %d %d\n", qom_path, name, num, level);
788 qtest_rsp(s, 0);
789}
790
49ee3590
AL
791static void qtest_out(QTestState *s, const char *cmd, uint16_t addr, uint32_t value)
792{
793 qtest_sendf(s, "%s 0x%x 0x%x\n", cmd, addr, value);
794 qtest_rsp(s, 0);
795}
796
797void qtest_outb(QTestState *s, uint16_t addr, uint8_t value)
798{
799 qtest_out(s, "outb", addr, value);
800}
801
802void qtest_outw(QTestState *s, uint16_t addr, uint16_t value)
803{
804 qtest_out(s, "outw", addr, value);
805}
806
807void qtest_outl(QTestState *s, uint16_t addr, uint32_t value)
808{
809 qtest_out(s, "outl", addr, value);
810}
811
812static uint32_t qtest_in(QTestState *s, const char *cmd, uint16_t addr)
813{
814 gchar **args;
50990b16
EB
815 int ret;
816 unsigned long value;
49ee3590
AL
817
818 qtest_sendf(s, "%s 0x%x\n", cmd, addr);
819 args = qtest_rsp(s, 2);
50990b16
EB
820 ret = qemu_strtoul(args[1], NULL, 0, &value);
821 g_assert(!ret && value <= UINT32_MAX);
49ee3590
AL
822 g_strfreev(args);
823
824 return value;
825}
826
827uint8_t qtest_inb(QTestState *s, uint16_t addr)
828{
829 return qtest_in(s, "inb", addr);
830}
831
832uint16_t qtest_inw(QTestState *s, uint16_t addr)
833{
834 return qtest_in(s, "inw", addr);
835}
836
837uint32_t qtest_inl(QTestState *s, uint16_t addr)
838{
839 return qtest_in(s, "inl", addr);
840}
841
872536bf
AF
842static void qtest_write(QTestState *s, const char *cmd, uint64_t addr,
843 uint64_t value)
844{
845 qtest_sendf(s, "%s 0x%" PRIx64 " 0x%" PRIx64 "\n", cmd, addr, value);
846 qtest_rsp(s, 0);
847}
848
849void qtest_writeb(QTestState *s, uint64_t addr, uint8_t value)
850{
851 qtest_write(s, "writeb", addr, value);
852}
853
854void qtest_writew(QTestState *s, uint64_t addr, uint16_t value)
855{
856 qtest_write(s, "writew", addr, value);
857}
858
859void qtest_writel(QTestState *s, uint64_t addr, uint32_t value)
860{
861 qtest_write(s, "writel", addr, value);
862}
863
864void qtest_writeq(QTestState *s, uint64_t addr, uint64_t value)
865{
866 qtest_write(s, "writeq", addr, value);
867}
868
869static uint64_t qtest_read(QTestState *s, const char *cmd, uint64_t addr)
870{
871 gchar **args;
50990b16 872 int ret;
872536bf
AF
873 uint64_t value;
874
875 qtest_sendf(s, "%s 0x%" PRIx64 "\n", cmd, addr);
876 args = qtest_rsp(s, 2);
50990b16
EB
877 ret = qemu_strtou64(args[1], NULL, 0, &value);
878 g_assert(!ret);
872536bf
AF
879 g_strfreev(args);
880
881 return value;
882}
883
884uint8_t qtest_readb(QTestState *s, uint64_t addr)
885{
886 return qtest_read(s, "readb", addr);
887}
888
889uint16_t qtest_readw(QTestState *s, uint64_t addr)
890{
891 return qtest_read(s, "readw", addr);
892}
893
894uint32_t qtest_readl(QTestState *s, uint64_t addr)
895{
896 return qtest_read(s, "readl", addr);
897}
898
899uint64_t qtest_readq(QTestState *s, uint64_t addr)
900{
901 return qtest_read(s, "readq", addr);
902}
903
49ee3590
AL
904static int hex2nib(char ch)
905{
906 if (ch >= '0' && ch <= '9') {
907 return ch - '0';
908 } else if (ch >= 'a' && ch <= 'f') {
909 return 10 + (ch - 'a');
910 } else if (ch >= 'A' && ch <= 'F') {
911 return 10 + (ch - 'a');
912 } else {
913 return -1;
914 }
915}
916
917void qtest_memread(QTestState *s, uint64_t addr, void *data, size_t size)
918{
919 uint8_t *ptr = data;
920 gchar **args;
921 size_t i;
922
204febd1
GK
923 if (!size) {
924 return;
925 }
926
ea7697fc 927 qtest_sendf(s, "read 0x%" PRIx64 " 0x%zx\n", addr, size);
49ee3590
AL
928 args = qtest_rsp(s, 2);
929
930 for (i = 0; i < size; i++) {
931 ptr[i] = hex2nib(args[1][2 + (i * 2)]) << 4;
932 ptr[i] |= hex2nib(args[1][2 + (i * 2) + 1]);
933 }
934
935 g_strfreev(args);
936}
937
eeddd59f
LV
938uint64_t qtest_rtas_call(QTestState *s, const char *name,
939 uint32_t nargs, uint64_t args,
940 uint32_t nret, uint64_t ret)
941{
942 qtest_sendf(s, "rtas %s %u 0x%"PRIx64" %u 0x%"PRIx64"\n",
943 name, nargs, args, nret, ret);
944 qtest_rsp(s, 0);
945 return 0;
946}
947
041088c7 948void qtest_add_func(const char *str, void (*fn)(void))
49ee3590
AL
949{
950 gchar *path = g_strdup_printf("/%s/%s", qtest_get_arch(), str);
951 g_test_add_func(path, fn);
a7afc6b8 952 g_free(path);
49ee3590
AL
953}
954
822e36ca
MAL
955void qtest_add_data_func_full(const char *str, void *data,
956 void (*fn)(const void *),
957 GDestroyNotify data_free_func)
958{
959 gchar *path = g_strdup_printf("/%s/%s", qtest_get_arch(), str);
822e36ca 960 g_test_add_data_func_full(path, data, fn, data_free_func);
822e36ca
MAL
961 g_free(path);
962}
963
041088c7
MA
964void qtest_add_data_func(const char *str, const void *data,
965 void (*fn)(const void *))
7949c0e3
AF
966{
967 gchar *path = g_strdup_printf("/%s/%s", qtest_get_arch(), str);
968 g_test_add_data_func(path, data, fn);
969 g_free(path);
970}
971
7a6a740d
JS
972void qtest_bufwrite(QTestState *s, uint64_t addr, const void *data, size_t size)
973{
974 gchar *bdata;
975
976 bdata = g_base64_encode(data, size);
977 qtest_sendf(s, "b64write 0x%" PRIx64 " 0x%zx ", addr, size);
978 socket_send(s->fd, bdata, strlen(bdata));
979 socket_send(s->fd, "\n", 1);
980 qtest_rsp(s, 0);
981 g_free(bdata);
982}
983
984void qtest_bufread(QTestState *s, uint64_t addr, void *data, size_t size)
985{
986 gchar **args;
987 size_t len;
988
989 qtest_sendf(s, "b64read 0x%" PRIx64 " 0x%zx\n", addr, size);
990 args = qtest_rsp(s, 2);
991
992 g_base64_decode_inplace(args[1], &len);
993 if (size != len) {
994 fprintf(stderr, "bufread: asked for %zu bytes but decoded %zu\n",
995 size, len);
996 len = MIN(len, size);
997 }
998
999 memcpy(data, args[1], len);
1000 g_strfreev(args);
1001}
1002
49ee3590
AL
1003void qtest_memwrite(QTestState *s, uint64_t addr, const void *data, size_t size)
1004{
1005 const uint8_t *ptr = data;
1006 size_t i;
204febd1
GK
1007 char *enc;
1008
1009 if (!size) {
1010 return;
1011 }
1012
1013 enc = g_malloc(2 * size + 1);
49ee3590 1014
49ee3590 1015 for (i = 0; i < size; i++) {
5560b85a 1016 sprintf(&enc[i * 2], "%02x", ptr[i]);
49ee3590 1017 }
5560b85a
JS
1018
1019 qtest_sendf(s, "write 0x%" PRIx64 " 0x%zx 0x%s\n", addr, size, enc);
49ee3590 1020 qtest_rsp(s, 0);
5560b85a 1021 g_free(enc);
49ee3590 1022}
0100f425 1023
86298845
JS
1024void qtest_memset(QTestState *s, uint64_t addr, uint8_t pattern, size_t size)
1025{
4d007963 1026 qtest_sendf(s, "memset 0x%" PRIx64 " 0x%zx 0x%02x\n", addr, size, pattern);
86298845
JS
1027 qtest_rsp(s, 0);
1028}
1029
0100f425
PM
1030QDict *qmp(const char *fmt, ...)
1031{
1032 va_list ap;
1033 QDict *response;
1034
1035 va_start(ap, fmt);
248eef02 1036 response = qtest_vqmp(global_qtest, fmt, ap);
0100f425
PM
1037 va_end(ap);
1038 return response;
1039}
1040
5fb48d96
MA
1041char *hmp(const char *fmt, ...)
1042{
1043 va_list ap;
1044 char *ret;
1045
1046 va_start(ap, fmt);
248eef02 1047 ret = qtest_vhmp(global_qtest, fmt, ap);
5fb48d96
MA
1048 va_end(ap);
1049 return ret;
1050}
46e0cf76 1051
54ce6f22 1052bool qtest_big_endian(QTestState *s)
46e0cf76 1053{
54ce6f22 1054 return s->big_endian;
46e0cf76 1055}
02ef6e87 1056
1f4a0d81
TH
1057static bool qtest_check_machine_version(const char *mname, const char *basename,
1058 int major, int minor)
1059{
1060 char *newname;
1061 bool is_equal;
1062
1063 newname = g_strdup_printf("%s-%i.%i", basename, major, minor);
1064 is_equal = g_str_equal(mname, newname);
1065 g_free(newname);
1066
1067 return is_equal;
1068}
1069
1070static bool qtest_is_old_versioned_machine(const char *mname)
1071{
1072 const char *dash = strrchr(mname, '-');
1073 const char *dot = strrchr(mname, '.');
1074 const char *chr;
1075 char *bname;
1076 const int major = QEMU_VERSION_MAJOR;
1077 const int minor = QEMU_VERSION_MINOR;
1078 bool res = false;
1079
1080 if (dash && dot && dot > dash) {
1081 for (chr = dash + 1; *chr; chr++) {
1082 if (!qemu_isdigit(*chr) && *chr != '.') {
1083 return false;
1084 }
1085 }
1086 /*
1087 * Now check if it is one of the latest versions. Check major + 1
1088 * and minor + 1 versions as well, since they might already exist
1089 * in the development branch.
1090 */
1091 bname = g_strdup(mname);
1092 bname[dash - mname] = 0;
1093 res = !qtest_check_machine_version(mname, bname, major + 1, 0) &&
1094 !qtest_check_machine_version(mname, bname, major, minor + 1) &&
1095 !qtest_check_machine_version(mname, bname, major, minor);
1096 g_free(bname);
1097 }
1098
1099 return res;
1100}
1101
1102void qtest_cb_for_every_machine(void (*cb)(const char *machine),
1103 bool skip_old_versioned)
02ef6e87
TH
1104{
1105 QDict *response, *minfo;
1106 QList *list;
1107 const QListEntry *p;
1108 QObject *qobj;
1109 QString *qstr;
1110 const char *mname;
1111
1112 qtest_start("-machine none");
1113 response = qmp("{ 'execute': 'query-machines' }");
1114 g_assert(response);
1115 list = qdict_get_qlist(response, "return");
1116 g_assert(list);
1117
1118 for (p = qlist_first(list); p; p = qlist_next(p)) {
7dc847eb 1119 minfo = qobject_to(QDict, qlist_entry_obj(p));
02ef6e87
TH
1120 g_assert(minfo);
1121 qobj = qdict_get(minfo, "name");
1122 g_assert(qobj);
7dc847eb 1123 qstr = qobject_to(QString, qobj);
02ef6e87
TH
1124 g_assert(qstr);
1125 mname = qstring_get_str(qstr);
1f4a0d81
TH
1126 if (!skip_old_versioned || !qtest_is_old_versioned_machine(mname)) {
1127 cb(mname);
1128 }
02ef6e87
TH
1129 }
1130
1131 qtest_end();
cb3e7f08 1132 qobject_unref(response);
02ef6e87 1133}
acd80015 1134
3cd46d42
MA
1135QDict *qtest_qmp_receive_success(QTestState *s,
1136 void (*event_cb)(void *opaque,
1137 const char *event,
1138 QDict *data),
1139 void *opaque)
1140{
1141 QDict *response, *ret, *data;
1142 const char *event;
1143
1144 for (;;) {
1145 response = qtest_qmp_receive(s);
1146 g_assert(!qdict_haskey(response, "error"));
1147 ret = qdict_get_qdict(response, "return");
1148 if (ret) {
1149 break;
1150 }
1151 event = qdict_get_str(response, "event");
1152 data = qdict_get_qdict(response, "data");
1153 if (event_cb) {
1154 event_cb(opaque, event, data);
1155 }
1156 qobject_unref(response);
1157 }
1158
1159 qobject_ref(ret);
1160 qobject_unref(response);
1161 return ret;
1162}
1163
acd80015
TH
1164/*
1165 * Generic hot-plugging test via the device_add QMP command.
1166 */
82cab70b
MA
1167void qtest_qmp_device_add(const char *driver, const char *id,
1168 const char *fmt, ...)
acd80015 1169{
82cab70b
MA
1170 QDict *args, *response;
1171 va_list ap;
acd80015 1172
82cab70b
MA
1173 va_start(ap, fmt);
1174 args = qdict_from_vjsonf_nofail(fmt, ap);
1175 va_end(ap);
acd80015 1176
82cab70b
MA
1177 g_assert(!qdict_haskey(args, "driver") && !qdict_haskey(args, "id"));
1178 qdict_put_str(args, "driver", driver);
1179 qdict_put_str(args, "id", id);
acd80015 1180
82cab70b 1181 response = qmp("{'execute': 'device_add', 'arguments': %p}", args);
acd80015
TH
1182 g_assert(response);
1183 g_assert(!qdict_haskey(response, "event")); /* We don't expect any events */
1184 g_assert(!qdict_haskey(response, "error"));
cb3e7f08 1185 qobject_unref(response);
acd80015
TH
1186}
1187
3cd46d42
MA
1188static void device_deleted_cb(void *opaque, const char *name, QDict *data)
1189{
1190 bool *got_event = opaque;
1191
1192 g_assert_cmpstr(name, ==, "DEVICE_DELETED");
1193 *got_event = true;
1194}
1195
acd80015
TH
1196/*
1197 * Generic hot-unplugging test via the device_del QMP command.
1198 * Device deletion will get one response and one event. For example:
1199 *
1200 * {'execute': 'device_del','arguments': { 'id': 'scsi-hd'}}
1201 *
1202 * will get this one:
1203 *
1204 * {"timestamp": {"seconds": 1505289667, "microseconds": 569862},
1205 * "event": "DEVICE_DELETED", "data": {"device": "scsi-hd",
1206 * "path": "/machine/peripheral/scsi-hd"}}
1207 *
1208 * and this one:
1209 *
1210 * {"return": {}}
1211 *
1212 * But the order of arrival may vary - so we've got to detect both.
1213 */
1214void qtest_qmp_device_del(const char *id)
1215{
3cd46d42
MA
1216 bool got_event = false;
1217 QDict *rsp;
acd80015 1218
3cd46d42
MA
1219 qtest_qmp_send(global_qtest,
1220 "{'execute': 'device_del', 'arguments': {'id': %s}}",
1221 id);
1222 rsp = qtest_qmp_receive_success(global_qtest, device_deleted_cb,
1223 &got_event);
1224 qobject_unref(rsp);
1225 if (!got_event) {
1f0396db 1226 rsp = qtest_qmp_receive(global_qtest);
3cd46d42
MA
1227 g_assert_cmpstr(qdict_get_try_str(rsp, "event"),
1228 ==, "DEVICE_DELETED");
1229 qobject_unref(rsp);
acd80015 1230 }
acd80015 1231}
c35665e1
IM
1232
1233bool qmp_rsp_is_err(QDict *rsp)
1234{
1235 QDict *error = qdict_get_qdict(rsp, "error");
1236 qobject_unref(rsp);
1237 return !!error;
1238}
ebb4d82d
MAL
1239
1240void qmp_assert_error_class(QDict *rsp, const char *class)
1241{
1242 QDict *error = qdict_get_qdict(rsp, "error");
1243
1244 g_assert_cmpstr(qdict_get_try_str(error, "class"), ==, class);
1245 g_assert_nonnull(qdict_get_try_str(error, "desc"));
1246 g_assert(!qdict_haskey(rsp, "return"));
1247
1248 qobject_unref(rsp);
1249}