]> git.proxmox.com Git - mirror_qemu.git/blame - tests/libqtest.c
Include qapi/qmp/qobject.h exactly where needed
[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.
15 *
16 */
681c28a3 17#include "qemu/osdep.h"
49ee3590
AL
18#include "libqtest.h"
19
49ee3590
AL
20#include <sys/socket.h>
21#include <sys/wait.h>
22#include <sys/un.h>
49ee3590 23
53f99152 24#include "qapi/error.h"
0c460dac 25#include "qapi/qmp/json-parser.h"
563890c7
PB
26#include "qapi/qmp/json-streamer.h"
27#include "qapi/qmp/qjson.h"
49ee3590
AL
28
29#define MAX_IRQ 256
f5d45791 30#define SOCKET_TIMEOUT 50
49ee3590
AL
31
32QTestState *global_qtest;
33
34struct QTestState
35{
36 int fd;
a3ca163c 37 int qmp_fd;
49ee3590
AL
38 bool irq_level[MAX_IRQ];
39 GString *rx;
cef60c92 40 pid_t qemu_pid; /* our child QEMU process */
54ce6f22 41 bool big_endian;
49ee3590
AL
42};
43
063c23d9 44static GHookList abrt_hooks;
d7668251
SH
45static struct sigaction sigact_old;
46
49ee3590
AL
47#define g_assert_no_errno(ret) do { \
48 g_assert_cmpint(ret, !=, -1); \
49} while (0)
50
54ce6f22
LV
51static int qtest_query_target_endianness(QTestState *s);
52
a3ca163c
KW
53static int init_socket(const char *socket_path)
54{
55 struct sockaddr_un addr;
56 int sock;
57 int ret;
58
59 sock = socket(PF_UNIX, SOCK_STREAM, 0);
60 g_assert_no_errno(sock);
61
62 addr.sun_family = AF_UNIX;
63 snprintf(addr.sun_path, sizeof(addr.sun_path), "%s", socket_path);
64 qemu_set_cloexec(sock);
65
66 do {
67 ret = bind(sock, (struct sockaddr *)&addr, sizeof(addr));
68 } while (ret == -1 && errno == EINTR);
69 g_assert_no_errno(ret);
a7ded163
AF
70 ret = listen(sock, 1);
71 g_assert_no_errno(ret);
a3ca163c
KW
72
73 return sock;
74}
75
76static int socket_accept(int sock)
77{
78 struct sockaddr_un addr;
79 socklen_t addrlen;
80 int ret;
f8762027
MA
81 struct timeval timeout = { .tv_sec = SOCKET_TIMEOUT,
82 .tv_usec = 0 };
83
84 setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, (void *)&timeout,
85 sizeof(timeout));
a3ca163c
KW
86
87 do {
535b4563 88 addrlen = sizeof(addr);
a3ca163c
KW
89 ret = accept(sock, (struct sockaddr *)&addr, &addrlen);
90 } while (ret == -1 && errno == EINTR);
a7d915f3
AF
91 if (ret == -1) {
92 fprintf(stderr, "%s failed: %s\n", __func__, strerror(errno));
93 }
a3ca163c
KW
94 close(sock);
95
96 return ret;
97}
98
b15d422a
SH
99static void kill_qemu(QTestState *s)
100{
101 if (s->qemu_pid != -1) {
102 kill(s->qemu_pid, SIGTERM);
103 waitpid(s->qemu_pid, NULL, 0);
104 }
105}
106
041088c7
MA
107static void kill_qemu_hook_func(void *s)
108{
109 kill_qemu(s);
110}
111
b15d422a
SH
112static void sigabrt_handler(int signo)
113{
063c23d9 114 g_hook_list_invoke(&abrt_hooks, FALSE);
d7668251
SH
115}
116
117static void setup_sigabrt_handler(void)
118{
119 struct sigaction sigact;
120
121 /* Catch SIGABRT to clean up on g_assert() failure */
122 sigact = (struct sigaction){
123 .sa_handler = sigabrt_handler,
124 .sa_flags = SA_RESETHAND,
125 };
126 sigemptyset(&sigact.sa_mask);
127 sigaction(SIGABRT, &sigact, &sigact_old);
128}
129
130static void cleanup_sigabrt_handler(void)
131{
132 sigaction(SIGABRT, &sigact_old, NULL);
b15d422a
SH
133}
134
041088c7 135void qtest_add_abrt_handler(GHookFunc fn, const void *data)
063c23d9
MAL
136{
137 GHook *hook;
138
139 /* Only install SIGABRT handler once */
140 if (!abrt_hooks.is_setup) {
141 g_hook_list_init(&abrt_hooks, sizeof(GHook));
063c23d9 142 }
24dd1e17 143 setup_sigabrt_handler();
063c23d9
MAL
144
145 hook = g_hook_alloc(&abrt_hooks);
146 hook->func = fn;
147 hook->data = (void *)data;
148
149 g_hook_prepend(&abrt_hooks, hook);
150}
151
db221e66
TH
152static const char *qtest_qemu_binary(void)
153{
154 const char *qemu_bin;
155
156 qemu_bin = getenv("QTEST_QEMU_BINARY");
157 if (!qemu_bin) {
158 fprintf(stderr, "Environment variable QTEST_QEMU_BINARY required\n");
159 exit(1);
160 }
161
162 return qemu_bin;
163}
164
f66e7ac8 165QTestState *qtest_init_without_qmp_handshake(const char *extra_args)
49ee3590
AL
166{
167 QTestState *s;
8a8fd637 168 int sock, qmpsock, i;
56db2e58
SH
169 gchar *socket_path;
170 gchar *qmp_socket_path;
49ee3590 171 gchar *command;
db221e66 172 const char *qemu_binary = qtest_qemu_binary();
49ee3590 173
790bbb97 174 s = g_new(QTestState, 1);
49ee3590 175
56db2e58
SH
176 socket_path = g_strdup_printf("/tmp/qtest-%d.sock", getpid());
177 qmp_socket_path = g_strdup_printf("/tmp/qtest-%d.qmp", getpid());
bb494a50 178
d9123d09
PM
179 /* It's possible that if an earlier test run crashed it might
180 * have left a stale unix socket lying around. Delete any
181 * stale old socket to avoid spurious test failures with
182 * tests/libqtest.c:70:init_socket: assertion failed (ret != -1): (-1 != -1)
183 */
184 unlink(socket_path);
185 unlink(qmp_socket_path);
186
56db2e58
SH
187 sock = init_socket(socket_path);
188 qmpsock = init_socket(qmp_socket_path);
49ee3590 189
041088c7 190 qtest_add_abrt_handler(kill_qemu_hook_func, s);
b15d422a 191
cef60c92
SH
192 s->qemu_pid = fork();
193 if (s->qemu_pid == 0) {
6b029216 194 setenv("QEMU_AUDIO_DRV", "none", true);
cef60c92 195 command = g_strdup_printf("exec %s "
49ee3590 196 "-qtest unix:%s,nowait "
7c282cb4 197 "-qtest-log %s "
a3ca163c 198 "-qmp unix:%s,nowait "
49ee3590 199 "-machine accel=qtest "
2ad645d2 200 "-display none "
56db2e58 201 "%s", qemu_binary, socket_path,
7c282cb4 202 getenv("QTEST_LOG") ? "/dev/fd/2" : "/dev/null",
cef60c92 203 qmp_socket_path,
49ee3590 204 extra_args ?: "");
8a8fd637
AL
205 execlp("/bin/sh", "sh", "-c", command, NULL);
206 exit(1);
49ee3590
AL
207 }
208
a3ca163c 209 s->fd = socket_accept(sock);
f8762027
MA
210 if (s->fd >= 0) {
211 s->qmp_fd = socket_accept(qmpsock);
212 }
56db2e58
SH
213 unlink(socket_path);
214 unlink(qmp_socket_path);
215 g_free(socket_path);
216 g_free(qmp_socket_path);
49ee3590 217
f8762027
MA
218 g_assert(s->fd >= 0 && s->qmp_fd >= 0);
219
49ee3590 220 s->rx = g_string_new("");
49ee3590
AL
221 for (i = 0; i < MAX_IRQ; i++) {
222 s->irq_level[i] = false;
223 }
224
e0fea6b1 225 if (getenv("QTEST_STOP")) {
1ad3c6ab 226 kill(s->qemu_pid, SIGSTOP);
e0fea6b1
PB
227 }
228
54ce6f22
LV
229 /* ask endianness of the target */
230
231 s->big_endian = qtest_query_target_endianness(s);
232
49ee3590
AL
233 return s;
234}
235
f66e7ac8
MA
236QTestState *qtest_init(const char *extra_args)
237{
238 QTestState *s = qtest_init_without_qmp_handshake(extra_args);
239
240 /* Read the QMP greeting and then do the handshake */
241 qtest_qmp_discard_response(s, "");
242 qtest_qmp_discard_response(s, "{ 'execute': 'qmp_capabilities' }");
243
244 return s;
245}
246
78b27bad
EB
247QTestState *qtest_vstartf(const char *fmt, va_list ap)
248{
249 char *args = g_strdup_vprintf(fmt, ap);
250 QTestState *s;
251
252 s = qtest_start(args);
253 g_free(args);
254 global_qtest = NULL;
255 return s;
256}
257
258QTestState *qtest_startf(const char *fmt, ...)
259{
260 va_list ap;
261 QTestState *s;
262
263 va_start(ap, fmt);
264 s = qtest_vstartf(fmt, ap);
265 va_end(ap);
266 return s;
267}
268
49ee3590
AL
269void qtest_quit(QTestState *s)
270{
063c23d9
MAL
271 g_hook_destroy_link(&abrt_hooks, g_hook_find_data(&abrt_hooks, TRUE, s));
272
d7668251 273 /* Uninstall SIGABRT handler on last instance */
4fb609ad 274 cleanup_sigabrt_handler();
d7668251 275
b15d422a 276 kill_qemu(s);
fdd26fca
MA
277 close(s->fd);
278 close(s->qmp_fd);
279 g_string_free(s->rx, true);
fdd26fca 280 g_free(s);
49ee3590
AL
281}
282
563890c7 283static void socket_send(int fd, const char *buf, size_t size)
49ee3590 284{
563890c7 285 size_t offset;
49ee3590
AL
286
287 offset = 0;
288 while (offset < size) {
289 ssize_t len;
290
563890c7 291 len = write(fd, buf + offset, size - offset);
49ee3590
AL
292 if (len == -1 && errno == EINTR) {
293 continue;
294 }
295
296 g_assert_no_errno(len);
297 g_assert_cmpint(len, >, 0);
298
299 offset += len;
300 }
301}
302
563890c7
PB
303static void socket_sendf(int fd, const char *fmt, va_list ap)
304{
305 gchar *str = g_strdup_vprintf(fmt, ap);
306 size_t size = strlen(str);
307
308 socket_send(fd, str, size);
309 g_free(str);
310}
311
a3ca163c
KW
312static void GCC_FMT_ATTR(2, 3) qtest_sendf(QTestState *s, const char *fmt, ...)
313{
314 va_list ap;
315
316 va_start(ap, fmt);
317 socket_sendf(s->fd, fmt, ap);
318 va_end(ap);
319}
320
49ee3590
AL
321static GString *qtest_recv_line(QTestState *s)
322{
323 GString *line;
324 size_t offset;
325 char *eol;
326
327 while ((eol = strchr(s->rx->str, '\n')) == NULL) {
328 ssize_t len;
329 char buffer[1024];
330
331 len = read(s->fd, buffer, sizeof(buffer));
332 if (len == -1 && errno == EINTR) {
333 continue;
334 }
335
336 if (len == -1 || len == 0) {
337 fprintf(stderr, "Broken pipe\n");
338 exit(1);
339 }
340
341 g_string_append_len(s->rx, buffer, len);
342 }
343
344 offset = eol - s->rx->str;
345 line = g_string_new_len(s->rx->str, offset);
346 g_string_erase(s->rx, 0, offset + 1);
347
348 return line;
349}
350
351static gchar **qtest_rsp(QTestState *s, int expected_args)
352{
353 GString *line;
354 gchar **words;
355 int i;
356
357redo:
358 line = qtest_recv_line(s);
359 words = g_strsplit(line->str, " ", 0);
360 g_string_free(line, TRUE);
361
362 if (strcmp(words[0], "IRQ") == 0) {
363 int irq;
364
365 g_assert(words[1] != NULL);
366 g_assert(words[2] != NULL);
367
368 irq = strtoul(words[2], NULL, 0);
369 g_assert_cmpint(irq, >=, 0);
370 g_assert_cmpint(irq, <, MAX_IRQ);
371
372 if (strcmp(words[1], "raise") == 0) {
373 s->irq_level[irq] = true;
374 } else {
375 s->irq_level[irq] = false;
376 }
377
378 g_strfreev(words);
379 goto redo;
380 }
381
382 g_assert(words[0] != NULL);
383 g_assert_cmpstr(words[0], ==, "OK");
384
385 if (expected_args) {
386 for (i = 0; i < expected_args; i++) {
387 g_assert(words[i] != NULL);
388 }
389 } else {
390 g_strfreev(words);
391 }
392
393 return words;
394}
395
54ce6f22
LV
396static int qtest_query_target_endianness(QTestState *s)
397{
398 gchar **args;
399 int big_endian;
400
401 qtest_sendf(s, "endianness\n");
402 args = qtest_rsp(s, 1);
403 g_assert(strcmp(args[1], "big") == 0 || strcmp(args[1], "little") == 0);
404 big_endian = strcmp(args[1], "big") == 0;
405 g_strfreev(args);
406
407 return big_endian;
408}
409
0c460dac
SH
410typedef struct {
411 JSONMessageParser parser;
412 QDict *response;
413} QMPResponseParser;
414
95385fe9 415static void qmp_response(JSONMessageParser *parser, GQueue *tokens)
a3ca163c 416{
0c460dac
SH
417 QMPResponseParser *qmp = container_of(parser, QMPResponseParser, parser);
418 QObject *obj;
419
420 obj = json_parser_parse(tokens, NULL);
421 if (!obj) {
422 fprintf(stderr, "QMP JSON response parsing failed\n");
423 exit(1);
424 }
425
0c460dac 426 g_assert(!qmp->response);
4d96f329
MA
427 qmp->response = qobject_to_qdict(obj);
428 g_assert(qmp->response);
0c460dac
SH
429}
430
dc47995e 431QDict *qmp_fd_receive(int fd)
0c460dac
SH
432{
433 QMPResponseParser qmp;
7c282cb4 434 bool log = getenv("QTEST_LOG") != NULL;
a3ca163c 435
0c460dac
SH
436 qmp.response = NULL;
437 json_message_parser_init(&qmp.parser, qmp_response);
438 while (!qmp.response) {
a3ca163c
KW
439 ssize_t len;
440 char c;
441
dc47995e 442 len = read(fd, &c, 1);
a3ca163c
KW
443 if (len == -1 && errno == EINTR) {
444 continue;
445 }
446
039380a8
AL
447 if (len == -1 || len == 0) {
448 fprintf(stderr, "Broken pipe\n");
449 exit(1);
450 }
451
7c282cb4
PB
452 if (log) {
453 len = write(2, &c, 1);
454 }
0c460dac 455 json_message_parser_feed(&qmp.parser, &c, 1);
a3ca163c 456 }
0c460dac
SH
457 json_message_parser_destroy(&qmp.parser);
458
459 return qmp.response;
460}
461
dc47995e
MAL
462QDict *qtest_qmp_receive(QTestState *s)
463{
464 return qmp_fd_receive(s->qmp_fd);
465}
466
ba4ed393
JS
467/**
468 * Allow users to send a message without waiting for the reply,
469 * in the case that they choose to discard all replies up until
470 * a particular EVENT is received.
471 */
dc47995e 472void qmp_fd_sendv(int fd, const char *fmt, va_list ap)
66e0c7b1 473{
563890c7
PB
474 va_list ap_copy;
475 QObject *qobj;
476
5229564b
EB
477 /* qobject_from_jsonv() silently eats leading 0xff as invalid
478 * JSON, but we want to test sending them over the wire to force
479 * resyncs */
480 if (*fmt == '\377') {
481 socket_send(fd, fmt, 1);
482 fmt++;
483 }
484
563890c7
PB
485 /* Going through qobject ensures we escape strings properly.
486 * This seemingly unnecessary copy is required in case va_list
487 * is an array type.
488 */
489 va_copy(ap_copy, ap);
53f99152 490 qobj = qobject_from_jsonv(fmt, &ap_copy, &error_abort);
563890c7
PB
491 va_end(ap_copy);
492
493 /* No need to send anything for an empty QObject. */
494 if (qobj) {
ae74f187 495 int log = getenv("QTEST_LOG") != NULL;
563890c7 496 QString *qstr = qobject_to_json(qobj);
13420ef8
MA
497 const char *str;
498
499 /*
500 * BUG: QMP doesn't react to input until it sees a newline, an
501 * object, or an array. Work-around: give it a newline.
502 */
503 qstring_append_chr(qstr, '\n');
504 str = qstring_get_str(qstr);
563890c7 505
ae74f187
MM
506 if (log) {
507 fprintf(stderr, "%s", str);
508 }
563890c7 509 /* Send QMP request */
13420ef8 510 socket_send(fd, str, qstring_get_length(qstr));
563890c7
PB
511
512 QDECREF(qstr);
513 qobject_decref(qobj);
514 }
ba4ed393
JS
515}
516
dc47995e
MAL
517void qtest_async_qmpv(QTestState *s, const char *fmt, va_list ap)
518{
519 qmp_fd_sendv(s->qmp_fd, fmt, ap);
520}
521
522QDict *qmp_fdv(int fd, const char *fmt, va_list ap)
523{
524 qmp_fd_sendv(fd, fmt, ap);
525
526 return qmp_fd_receive(fd);
527}
528
ba4ed393
JS
529QDict *qtest_qmpv(QTestState *s, const char *fmt, va_list ap)
530{
531 qtest_async_qmpv(s, fmt, ap);
66e0c7b1
AF
532
533 /* Receive reply */
534 return qtest_qmp_receive(s);
535}
536
dc47995e
MAL
537QDict *qmp_fd(int fd, const char *fmt, ...)
538{
539 va_list ap;
540 QDict *response;
541
542 va_start(ap, fmt);
543 response = qmp_fdv(fd, fmt, ap);
544 va_end(ap);
545 return response;
546}
547
548void qmp_fd_send(int fd, const char *fmt, ...)
549{
550 va_list ap;
551
552 va_start(ap, fmt);
553 qmp_fd_sendv(fd, fmt, ap);
554 va_end(ap);
555}
556
0c460dac
SH
557QDict *qtest_qmp(QTestState *s, const char *fmt, ...)
558{
559 va_list ap;
560 QDict *response;
561
562 va_start(ap, fmt);
563 response = qtest_qmpv(s, fmt, ap);
564 va_end(ap);
565 return response;
566}
567
ba4ed393
JS
568void qtest_async_qmp(QTestState *s, const char *fmt, ...)
569{
570 va_list ap;
571
572 va_start(ap, fmt);
573 qtest_async_qmpv(s, fmt, ap);
574 va_end(ap);
575}
576
0c460dac
SH
577void qtest_qmpv_discard_response(QTestState *s, const char *fmt, va_list ap)
578{
579 QDict *response = qtest_qmpv(s, fmt, ap);
580 QDECREF(response);
a3ca163c
KW
581}
582
0d1aa05e 583void qtest_qmp_discard_response(QTestState *s, const char *fmt, ...)
b73cf9e9
AF
584{
585 va_list ap;
0c460dac 586 QDict *response;
b73cf9e9
AF
587
588 va_start(ap, fmt);
0c460dac 589 response = qtest_qmpv(s, fmt, ap);
b73cf9e9 590 va_end(ap);
0c460dac 591 QDECREF(response);
b73cf9e9
AF
592}
593
7ffe3124 594QDict *qtest_qmp_eventwait_ref(QTestState *s, const char *event)
8fe941f7
JS
595{
596 QDict *response;
597
598 for (;;) {
599 response = qtest_qmp_receive(s);
600 if ((qdict_haskey(response, "event")) &&
601 (strcmp(qdict_get_str(response, "event"), event) == 0)) {
7ffe3124 602 return response;
8fe941f7
JS
603 }
604 QDECREF(response);
605 }
606}
607
7ffe3124
JS
608void qtest_qmp_eventwait(QTestState *s, const char *event)
609{
610 QDict *response;
611
612 response = qtest_qmp_eventwait_ref(s, event);
613 QDECREF(response);
614}
615
5fb48d96
MA
616char *qtest_hmpv(QTestState *s, const char *fmt, va_list ap)
617{
618 char *cmd;
619 QDict *resp;
620 char *ret;
621
622 cmd = g_strdup_vprintf(fmt, ap);
623 resp = qtest_qmp(s, "{'execute': 'human-monitor-command',"
624 " 'arguments': {'command-line': %s}}",
625 cmd);
626 ret = g_strdup(qdict_get_try_str(resp, "return"));
6bb87be8
TH
627 while (ret == NULL && qdict_get_try_str(resp, "event")) {
628 /* Ignore asynchronous QMP events */
629 QDECREF(resp);
630 resp = qtest_qmp_receive(s);
631 ret = g_strdup(qdict_get_try_str(resp, "return"));
632 }
5fb48d96
MA
633 g_assert(ret);
634 QDECREF(resp);
635 g_free(cmd);
636 return ret;
637}
638
639char *qtest_hmp(QTestState *s, const char *fmt, ...)
640{
641 va_list ap;
642 char *ret;
643
644 va_start(ap, fmt);
645 ret = qtest_hmpv(s, fmt, ap);
646 va_end(ap);
647 return ret;
648}
8fe941f7 649
49ee3590
AL
650const char *qtest_get_arch(void)
651{
db221e66 652 const char *qemu = qtest_qemu_binary();
49ee3590
AL
653 const char *end = strrchr(qemu, '/');
654
655 return end + strlen("/qemu-system-");
656}
657
658bool qtest_get_irq(QTestState *s, int num)
659{
660 /* dummy operation in order to make sure irq is up to date */
661 qtest_inb(s, 0);
662
663 return s->irq_level[num];
664}
665
666static int64_t qtest_clock_rsp(QTestState *s)
667{
668 gchar **words;
669 int64_t clock;
670 words = qtest_rsp(s, 2);
671 clock = g_ascii_strtoll(words[1], NULL, 0);
672 g_strfreev(words);
673 return clock;
674}
675
676int64_t qtest_clock_step_next(QTestState *s)
677{
678 qtest_sendf(s, "clock_step\n");
679 return qtest_clock_rsp(s);
680}
681
682int64_t qtest_clock_step(QTestState *s, int64_t step)
683{
684 qtest_sendf(s, "clock_step %"PRIi64"\n", step);
685 return qtest_clock_rsp(s);
686}
687
688int64_t qtest_clock_set(QTestState *s, int64_t val)
689{
690 qtest_sendf(s, "clock_set %"PRIi64"\n", val);
691 return qtest_clock_rsp(s);
692}
693
694void qtest_irq_intercept_out(QTestState *s, const char *qom_path)
695{
696 qtest_sendf(s, "irq_intercept_out %s\n", qom_path);
697 qtest_rsp(s, 0);
698}
699
700void qtest_irq_intercept_in(QTestState *s, const char *qom_path)
701{
702 qtest_sendf(s, "irq_intercept_in %s\n", qom_path);
703 qtest_rsp(s, 0);
704}
705
706static void qtest_out(QTestState *s, const char *cmd, uint16_t addr, uint32_t value)
707{
708 qtest_sendf(s, "%s 0x%x 0x%x\n", cmd, addr, value);
709 qtest_rsp(s, 0);
710}
711
712void qtest_outb(QTestState *s, uint16_t addr, uint8_t value)
713{
714 qtest_out(s, "outb", addr, value);
715}
716
717void qtest_outw(QTestState *s, uint16_t addr, uint16_t value)
718{
719 qtest_out(s, "outw", addr, value);
720}
721
722void qtest_outl(QTestState *s, uint16_t addr, uint32_t value)
723{
724 qtest_out(s, "outl", addr, value);
725}
726
727static uint32_t qtest_in(QTestState *s, const char *cmd, uint16_t addr)
728{
729 gchar **args;
730 uint32_t value;
731
732 qtest_sendf(s, "%s 0x%x\n", cmd, addr);
733 args = qtest_rsp(s, 2);
734 value = strtoul(args[1], NULL, 0);
735 g_strfreev(args);
736
737 return value;
738}
739
740uint8_t qtest_inb(QTestState *s, uint16_t addr)
741{
742 return qtest_in(s, "inb", addr);
743}
744
745uint16_t qtest_inw(QTestState *s, uint16_t addr)
746{
747 return qtest_in(s, "inw", addr);
748}
749
750uint32_t qtest_inl(QTestState *s, uint16_t addr)
751{
752 return qtest_in(s, "inl", addr);
753}
754
872536bf
AF
755static void qtest_write(QTestState *s, const char *cmd, uint64_t addr,
756 uint64_t value)
757{
758 qtest_sendf(s, "%s 0x%" PRIx64 " 0x%" PRIx64 "\n", cmd, addr, value);
759 qtest_rsp(s, 0);
760}
761
762void qtest_writeb(QTestState *s, uint64_t addr, uint8_t value)
763{
764 qtest_write(s, "writeb", addr, value);
765}
766
767void qtest_writew(QTestState *s, uint64_t addr, uint16_t value)
768{
769 qtest_write(s, "writew", addr, value);
770}
771
772void qtest_writel(QTestState *s, uint64_t addr, uint32_t value)
773{
774 qtest_write(s, "writel", addr, value);
775}
776
777void qtest_writeq(QTestState *s, uint64_t addr, uint64_t value)
778{
779 qtest_write(s, "writeq", addr, value);
780}
781
782static uint64_t qtest_read(QTestState *s, const char *cmd, uint64_t addr)
783{
784 gchar **args;
785 uint64_t value;
786
787 qtest_sendf(s, "%s 0x%" PRIx64 "\n", cmd, addr);
788 args = qtest_rsp(s, 2);
789 value = strtoull(args[1], NULL, 0);
790 g_strfreev(args);
791
792 return value;
793}
794
795uint8_t qtest_readb(QTestState *s, uint64_t addr)
796{
797 return qtest_read(s, "readb", addr);
798}
799
800uint16_t qtest_readw(QTestState *s, uint64_t addr)
801{
802 return qtest_read(s, "readw", addr);
803}
804
805uint32_t qtest_readl(QTestState *s, uint64_t addr)
806{
807 return qtest_read(s, "readl", addr);
808}
809
810uint64_t qtest_readq(QTestState *s, uint64_t addr)
811{
812 return qtest_read(s, "readq", addr);
813}
814
49ee3590
AL
815static int hex2nib(char ch)
816{
817 if (ch >= '0' && ch <= '9') {
818 return ch - '0';
819 } else if (ch >= 'a' && ch <= 'f') {
820 return 10 + (ch - 'a');
821 } else if (ch >= 'A' && ch <= 'F') {
822 return 10 + (ch - 'a');
823 } else {
824 return -1;
825 }
826}
827
828void qtest_memread(QTestState *s, uint64_t addr, void *data, size_t size)
829{
830 uint8_t *ptr = data;
831 gchar **args;
832 size_t i;
833
204febd1
GK
834 if (!size) {
835 return;
836 }
837
ea7697fc 838 qtest_sendf(s, "read 0x%" PRIx64 " 0x%zx\n", addr, size);
49ee3590
AL
839 args = qtest_rsp(s, 2);
840
841 for (i = 0; i < size; i++) {
842 ptr[i] = hex2nib(args[1][2 + (i * 2)]) << 4;
843 ptr[i] |= hex2nib(args[1][2 + (i * 2) + 1]);
844 }
845
846 g_strfreev(args);
847}
848
eeddd59f
LV
849uint64_t qtest_rtas_call(QTestState *s, const char *name,
850 uint32_t nargs, uint64_t args,
851 uint32_t nret, uint64_t ret)
852{
853 qtest_sendf(s, "rtas %s %u 0x%"PRIx64" %u 0x%"PRIx64"\n",
854 name, nargs, args, nret, ret);
855 qtest_rsp(s, 0);
856 return 0;
857}
858
041088c7 859void qtest_add_func(const char *str, void (*fn)(void))
49ee3590
AL
860{
861 gchar *path = g_strdup_printf("/%s/%s", qtest_get_arch(), str);
862 g_test_add_func(path, fn);
a7afc6b8 863 g_free(path);
49ee3590
AL
864}
865
822e36ca
MAL
866void qtest_add_data_func_full(const char *str, void *data,
867 void (*fn)(const void *),
868 GDestroyNotify data_free_func)
869{
870 gchar *path = g_strdup_printf("/%s/%s", qtest_get_arch(), str);
822e36ca 871 g_test_add_data_func_full(path, data, fn, data_free_func);
822e36ca
MAL
872 g_free(path);
873}
874
041088c7
MA
875void qtest_add_data_func(const char *str, const void *data,
876 void (*fn)(const void *))
7949c0e3
AF
877{
878 gchar *path = g_strdup_printf("/%s/%s", qtest_get_arch(), str);
879 g_test_add_data_func(path, data, fn);
880 g_free(path);
881}
882
7a6a740d
JS
883void qtest_bufwrite(QTestState *s, uint64_t addr, const void *data, size_t size)
884{
885 gchar *bdata;
886
887 bdata = g_base64_encode(data, size);
888 qtest_sendf(s, "b64write 0x%" PRIx64 " 0x%zx ", addr, size);
889 socket_send(s->fd, bdata, strlen(bdata));
890 socket_send(s->fd, "\n", 1);
891 qtest_rsp(s, 0);
892 g_free(bdata);
893}
894
895void qtest_bufread(QTestState *s, uint64_t addr, void *data, size_t size)
896{
897 gchar **args;
898 size_t len;
899
900 qtest_sendf(s, "b64read 0x%" PRIx64 " 0x%zx\n", addr, size);
901 args = qtest_rsp(s, 2);
902
903 g_base64_decode_inplace(args[1], &len);
904 if (size != len) {
905 fprintf(stderr, "bufread: asked for %zu bytes but decoded %zu\n",
906 size, len);
907 len = MIN(len, size);
908 }
909
910 memcpy(data, args[1], len);
911 g_strfreev(args);
912}
913
49ee3590
AL
914void qtest_memwrite(QTestState *s, uint64_t addr, const void *data, size_t size)
915{
916 const uint8_t *ptr = data;
917 size_t i;
204febd1
GK
918 char *enc;
919
920 if (!size) {
921 return;
922 }
923
924 enc = g_malloc(2 * size + 1);
49ee3590 925
49ee3590 926 for (i = 0; i < size; i++) {
5560b85a 927 sprintf(&enc[i * 2], "%02x", ptr[i]);
49ee3590 928 }
5560b85a
JS
929
930 qtest_sendf(s, "write 0x%" PRIx64 " 0x%zx 0x%s\n", addr, size, enc);
49ee3590 931 qtest_rsp(s, 0);
5560b85a 932 g_free(enc);
49ee3590 933}
0100f425 934
86298845
JS
935void qtest_memset(QTestState *s, uint64_t addr, uint8_t pattern, size_t size)
936{
4d007963 937 qtest_sendf(s, "memset 0x%" PRIx64 " 0x%zx 0x%02x\n", addr, size, pattern);
86298845
JS
938 qtest_rsp(s, 0);
939}
940
0100f425
PM
941QDict *qmp(const char *fmt, ...)
942{
943 va_list ap;
944 QDict *response;
945
946 va_start(ap, fmt);
947 response = qtest_qmpv(global_qtest, fmt, ap);
948 va_end(ap);
949 return response;
950}
951
ba4ed393
JS
952void qmp_async(const char *fmt, ...)
953{
954 va_list ap;
955
956 va_start(ap, fmt);
957 qtest_async_qmpv(global_qtest, fmt, ap);
958 va_end(ap);
959}
960
0100f425
PM
961void qmp_discard_response(const char *fmt, ...)
962{
963 va_list ap;
964
965 va_start(ap, fmt);
966 qtest_qmpv_discard_response(global_qtest, fmt, ap);
967 va_end(ap);
968}
5fb48d96
MA
969char *hmp(const char *fmt, ...)
970{
971 va_list ap;
972 char *ret;
973
974 va_start(ap, fmt);
975 ret = qtest_hmpv(global_qtest, fmt, ap);
976 va_end(ap);
977 return ret;
978}
46e0cf76 979
54ce6f22 980bool qtest_big_endian(QTestState *s)
46e0cf76 981{
54ce6f22 982 return s->big_endian;
46e0cf76 983}
02ef6e87
TH
984
985void qtest_cb_for_every_machine(void (*cb)(const char *machine))
986{
987 QDict *response, *minfo;
988 QList *list;
989 const QListEntry *p;
990 QObject *qobj;
991 QString *qstr;
992 const char *mname;
993
994 qtest_start("-machine none");
995 response = qmp("{ 'execute': 'query-machines' }");
996 g_assert(response);
997 list = qdict_get_qlist(response, "return");
998 g_assert(list);
999
1000 for (p = qlist_first(list); p; p = qlist_next(p)) {
1001 minfo = qobject_to_qdict(qlist_entry_obj(p));
1002 g_assert(minfo);
1003 qobj = qdict_get(minfo, "name");
1004 g_assert(qobj);
1005 qstr = qobject_to_qstring(qobj);
1006 g_assert(qstr);
1007 mname = qstring_get_str(qstr);
1008 cb(mname);
1009 }
1010
1011 qtest_end();
1012 QDECREF(response);
1013}
acd80015
TH
1014
1015/*
1016 * Generic hot-plugging test via the device_add QMP command.
1017 */
1018void qtest_qmp_device_add(const char *driver, const char *id, const char *fmt,
1019 ...)
1020{
1021 QDict *response;
1022 char *cmd, *opts = NULL;
1023 va_list va;
1024
1025 if (fmt) {
1026 va_start(va, fmt);
1027 opts = g_strdup_vprintf(fmt, va);
1028 va_end(va);
1029 }
1030
1031 cmd = g_strdup_printf("{'execute': 'device_add',"
1032 " 'arguments': { 'driver': '%s', 'id': '%s'%s%s }}",
1033 driver, id, opts ? ", " : "", opts ? opts : "");
1034 g_free(opts);
1035
1036 response = qmp(cmd);
1037 g_free(cmd);
1038 g_assert(response);
1039 g_assert(!qdict_haskey(response, "event")); /* We don't expect any events */
1040 g_assert(!qdict_haskey(response, "error"));
1041 QDECREF(response);
1042}
1043
1044/*
1045 * Generic hot-unplugging test via the device_del QMP command.
1046 * Device deletion will get one response and one event. For example:
1047 *
1048 * {'execute': 'device_del','arguments': { 'id': 'scsi-hd'}}
1049 *
1050 * will get this one:
1051 *
1052 * {"timestamp": {"seconds": 1505289667, "microseconds": 569862},
1053 * "event": "DEVICE_DELETED", "data": {"device": "scsi-hd",
1054 * "path": "/machine/peripheral/scsi-hd"}}
1055 *
1056 * and this one:
1057 *
1058 * {"return": {}}
1059 *
1060 * But the order of arrival may vary - so we've got to detect both.
1061 */
1062void qtest_qmp_device_del(const char *id)
1063{
1064 QDict *response1, *response2, *event = NULL;
1065 char *cmd;
1066
1067 cmd = g_strdup_printf("{'execute': 'device_del',"
1068 " 'arguments': { 'id': '%s' }}", id);
1069 response1 = qmp(cmd);
1070 g_free(cmd);
1071 g_assert(response1);
1072 g_assert(!qdict_haskey(response1, "error"));
1073
1074 response2 = qmp("");
1075 g_assert(response2);
1076 g_assert(!qdict_haskey(response2, "error"));
1077
1078 if (qdict_haskey(response1, "event")) {
1079 event = response1;
1080 } else if (qdict_haskey(response2, "event")) {
1081 event = response2;
1082 }
1083 g_assert(event);
1084 g_assert_cmpstr(qdict_get_str(event, "event"), ==, "DEVICE_DELETED");
1085
1086 QDECREF(response1);
1087 QDECREF(response2);
1088}