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