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