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