qtest: add memset to qtest protocol
[sdk/emulator/qemu.git] / qtest.c
1 /*
2  * Test Server
3  *
4  * Copyright IBM, Corp. 2011
5  *
6  * Authors:
7  *  Anthony Liguori   <aliguori@us.ibm.com>
8  *
9  * This work is licensed under the terms of the GNU GPL, version 2 or later.
10  * See the COPYING file in the top-level directory.
11  *
12  */
13
14 #include "sysemu/qtest.h"
15 #include "hw/qdev.h"
16 #include "sysemu/char.h"
17 #include "exec/ioport.h"
18 #include "exec/memory.h"
19 #include "hw/irq.h"
20 #include "sysemu/accel.h"
21 #include "sysemu/sysemu.h"
22 #include "sysemu/cpus.h"
23 #include "qemu/config-file.h"
24 #include "qemu/option.h"
25 #include "qemu/error-report.h"
26
27 #define MAX_IRQ 256
28
29 bool qtest_allowed;
30
31 static DeviceState *irq_intercept_dev;
32 static FILE *qtest_log_fp;
33 static CharDriverState *qtest_chr;
34 static GString *inbuf;
35 static int irq_levels[MAX_IRQ];
36 static qemu_timeval start_time;
37 static bool qtest_opened;
38
39 #define FMT_timeval "%ld.%06ld"
40
41 /**
42  * QTest Protocol
43  *
44  * Line based protocol, request/response based.  Server can send async messages
45  * so clients should always handle many async messages before the response
46  * comes in.
47  *
48  * Valid requests
49  *
50  * Clock management:
51  *
52  * The qtest client is completely in charge of the QEMU_CLOCK_VIRTUAL.  qtest commands
53  * let you adjust the value of the clock (monotonically).  All the commands
54  * return the current value of the clock in nanoseconds.
55  *
56  *  > clock_step
57  *  < OK VALUE
58  *
59  *     Advance the clock to the next deadline.  Useful when waiting for
60  *     asynchronous events.
61  *
62  *  > clock_step NS
63  *  < OK VALUE
64  *
65  *     Advance the clock by NS nanoseconds.
66  *
67  *  > clock_set NS
68  *  < OK VALUE
69  *
70  *     Advance the clock to NS nanoseconds (do nothing if it's already past).
71  *
72  * PIO and memory access:
73  *
74  *  > outb ADDR VALUE
75  *  < OK
76  *
77  *  > outw ADDR VALUE
78  *  < OK
79  *
80  *  > outl ADDR VALUE
81  *  < OK
82  *
83  *  > inb ADDR
84  *  < OK VALUE
85  *
86  *  > inw ADDR
87  *  < OK VALUE
88  *
89  *  > inl ADDR
90  *  < OK VALUE
91  *
92  *  > writeb ADDR VALUE
93  *  < OK
94  *
95  *  > writew ADDR VALUE
96  *  < OK
97  *
98  *  > writel ADDR VALUE
99  *  < OK
100  *
101  *  > writeq ADDR VALUE
102  *  < OK
103  *
104  *  > readb ADDR
105  *  < OK VALUE
106  *
107  *  > readw ADDR
108  *  < OK VALUE
109  *
110  *  > readl ADDR
111  *  < OK VALUE
112  *
113  *  > readq ADDR
114  *  < OK VALUE
115  *
116  *  > read ADDR SIZE
117  *  < OK DATA
118  *
119  *  > write ADDR SIZE DATA
120  *  < OK
121  *
122  *  > b64read ADDR SIZE
123  *  < OK B64_DATA
124  *
125  *  > b64write ADDR SIZE B64_DATA
126  *  < OK
127  *
128  *  > memset ADDR SIZE VALUE
129  *  < OK
130  *
131  * ADDR, SIZE, VALUE are all integers parsed with strtoul() with a base of 0.
132  *
133  * DATA is an arbitrarily long hex number prefixed with '0x'.  If it's smaller
134  * than the expected size, the value will be zero filled at the end of the data
135  * sequence.
136  *
137  * B64_DATA is an arbitrarily long base64 encoded string.
138  * If the sizes do not match, the data will be truncated.
139  *
140  * IRQ management:
141  *
142  *  > irq_intercept_in QOM-PATH
143  *  < OK
144  *
145  *  > irq_intercept_out QOM-PATH
146  *  < OK
147  *
148  * Attach to the gpio-in (resp. gpio-out) pins exported by the device at
149  * QOM-PATH.  When the pin is triggered, one of the following async messages
150  * will be printed to the qtest stream:
151  *
152  *  IRQ raise NUM
153  *  IRQ lower NUM
154  *
155  * where NUM is an IRQ number.  For the PC, interrupts can be intercepted
156  * simply with "irq_intercept_in ioapic" (note that IRQ0 comes out with
157  * NUM=0 even though it is remapped to GSI 2).
158  */
159
160 static int hex2nib(char ch)
161 {
162     if (ch >= '0' && ch <= '9') {
163         return ch - '0';
164     } else if (ch >= 'a' && ch <= 'f') {
165         return 10 + (ch - 'a');
166     } else if (ch >= 'A' && ch <= 'F') {
167         return 10 + (ch - 'A');
168     } else {
169         return -1;
170     }
171 }
172
173 static void qtest_get_time(qemu_timeval *tv)
174 {
175     qemu_gettimeofday(tv);
176     tv->tv_sec -= start_time.tv_sec;
177     tv->tv_usec -= start_time.tv_usec;
178     if (tv->tv_usec < 0) {
179         tv->tv_usec += 1000000;
180         tv->tv_sec -= 1;
181     }
182 }
183
184 static void qtest_send_prefix(CharDriverState *chr)
185 {
186     qemu_timeval tv;
187
188     if (!qtest_log_fp || !qtest_opened) {
189         return;
190     }
191
192     qtest_get_time(&tv);
193     fprintf(qtest_log_fp, "[S +" FMT_timeval "] ",
194             (long) tv.tv_sec, (long) tv.tv_usec);
195 }
196
197 static void GCC_FMT_ATTR(1, 2) qtest_log_send(const char *fmt, ...)
198 {
199     va_list ap;
200
201     if (!qtest_log_fp || !qtest_opened) {
202         return;
203     }
204
205     qtest_send_prefix(NULL);
206
207     va_start(ap, fmt);
208     vfprintf(qtest_log_fp, fmt, ap);
209     va_end(ap);
210 }
211
212 static void do_qtest_send(CharDriverState *chr, const char *str, size_t len)
213 {
214     qemu_chr_fe_write_all(chr, (uint8_t *)str, len);
215     if (qtest_log_fp && qtest_opened) {
216         fprintf(qtest_log_fp, "%s", str);
217     }
218 }
219
220 static void qtest_send(CharDriverState *chr, const char *str)
221 {
222     do_qtest_send(chr, str, strlen(str));
223 }
224
225 static void GCC_FMT_ATTR(2, 3) qtest_sendf(CharDriverState *chr,
226                                            const char *fmt, ...)
227 {
228     va_list ap;
229     gchar *buffer;
230
231     va_start(ap, fmt);
232     buffer = g_strdup_vprintf(fmt, ap);
233     qtest_send(chr, buffer);
234     va_end(ap);
235 }
236
237 static void qtest_irq_handler(void *opaque, int n, int level)
238 {
239     qemu_irq old_irq = *(qemu_irq *)opaque;
240     qemu_set_irq(old_irq, level);
241
242     if (irq_levels[n] != level) {
243         CharDriverState *chr = qtest_chr;
244         irq_levels[n] = level;
245         qtest_send_prefix(chr);
246         qtest_sendf(chr, "IRQ %s %d\n",
247                     level ? "raise" : "lower", n);
248     }
249 }
250
251 static void qtest_process_command(CharDriverState *chr, gchar **words)
252 {
253     const gchar *command;
254
255     g_assert(words);
256
257     command = words[0];
258
259     if (qtest_log_fp) {
260         qemu_timeval tv;
261         int i;
262
263         qtest_get_time(&tv);
264         fprintf(qtest_log_fp, "[R +" FMT_timeval "]",
265                 (long) tv.tv_sec, (long) tv.tv_usec);
266         for (i = 0; words[i]; i++) {
267             fprintf(qtest_log_fp, " %s", words[i]);
268         }
269         fprintf(qtest_log_fp, "\n");
270     }
271
272     g_assert(command);
273     if (strcmp(words[0], "irq_intercept_out") == 0
274         || strcmp(words[0], "irq_intercept_in") == 0) {
275         DeviceState *dev;
276         NamedGPIOList *ngl;
277
278         g_assert(words[1]);
279         dev = DEVICE(object_resolve_path(words[1], NULL));
280         if (!dev) {
281             qtest_send_prefix(chr);
282             qtest_send(chr, "FAIL Unknown device\n");
283             return;
284         }
285
286         if (irq_intercept_dev) {
287             qtest_send_prefix(chr);
288             if (irq_intercept_dev != dev) {
289                 qtest_send(chr, "FAIL IRQ intercept already enabled\n");
290             } else {
291                 qtest_send(chr, "OK\n");
292             }
293             return;
294         }
295
296         QLIST_FOREACH(ngl, &dev->gpios, node) {
297             /* We don't support intercept of named GPIOs yet */
298             if (ngl->name) {
299                 continue;
300             }
301             if (words[0][14] == 'o') {
302                 int i;
303                 for (i = 0; i < ngl->num_out; ++i) {
304                     qemu_irq *disconnected = g_new0(qemu_irq, 1);
305                     qemu_irq icpt = qemu_allocate_irq(qtest_irq_handler,
306                                                       disconnected, i);
307
308                     *disconnected = qdev_intercept_gpio_out(dev, icpt,
309                                                             ngl->name, i);
310                 }
311             } else {
312                 qemu_irq_intercept_in(ngl->in, qtest_irq_handler,
313                                       ngl->num_in);
314             }
315         }
316         irq_intercept_dev = dev;
317         qtest_send_prefix(chr);
318         qtest_send(chr, "OK\n");
319
320     } else if (strcmp(words[0], "outb") == 0 ||
321                strcmp(words[0], "outw") == 0 ||
322                strcmp(words[0], "outl") == 0) {
323         uint16_t addr;
324         uint32_t value;
325
326         g_assert(words[1] && words[2]);
327         addr = strtoul(words[1], NULL, 0);
328         value = strtoul(words[2], NULL, 0);
329
330         if (words[0][3] == 'b') {
331             cpu_outb(addr, value);
332         } else if (words[0][3] == 'w') {
333             cpu_outw(addr, value);
334         } else if (words[0][3] == 'l') {
335             cpu_outl(addr, value);
336         }
337         qtest_send_prefix(chr);
338         qtest_send(chr, "OK\n");
339     } else if (strcmp(words[0], "inb") == 0 ||
340         strcmp(words[0], "inw") == 0 ||
341         strcmp(words[0], "inl") == 0) {
342         uint16_t addr;
343         uint32_t value = -1U;
344
345         g_assert(words[1]);
346         addr = strtoul(words[1], NULL, 0);
347
348         if (words[0][2] == 'b') {
349             value = cpu_inb(addr);
350         } else if (words[0][2] == 'w') {
351             value = cpu_inw(addr);
352         } else if (words[0][2] == 'l') {
353             value = cpu_inl(addr);
354         }
355         qtest_send_prefix(chr);
356         qtest_sendf(chr, "OK 0x%04x\n", value);
357     } else if (strcmp(words[0], "writeb") == 0 ||
358                strcmp(words[0], "writew") == 0 ||
359                strcmp(words[0], "writel") == 0 ||
360                strcmp(words[0], "writeq") == 0) {
361         uint64_t addr;
362         uint64_t value;
363
364         g_assert(words[1] && words[2]);
365         addr = strtoull(words[1], NULL, 0);
366         value = strtoull(words[2], NULL, 0);
367
368         if (words[0][5] == 'b') {
369             uint8_t data = value;
370             cpu_physical_memory_write(addr, &data, 1);
371         } else if (words[0][5] == 'w') {
372             uint16_t data = value;
373             tswap16s(&data);
374             cpu_physical_memory_write(addr, &data, 2);
375         } else if (words[0][5] == 'l') {
376             uint32_t data = value;
377             tswap32s(&data);
378             cpu_physical_memory_write(addr, &data, 4);
379         } else if (words[0][5] == 'q') {
380             uint64_t data = value;
381             tswap64s(&data);
382             cpu_physical_memory_write(addr, &data, 8);
383         }
384         qtest_send_prefix(chr);
385         qtest_send(chr, "OK\n");
386     } else if (strcmp(words[0], "readb") == 0 ||
387                strcmp(words[0], "readw") == 0 ||
388                strcmp(words[0], "readl") == 0 ||
389                strcmp(words[0], "readq") == 0) {
390         uint64_t addr;
391         uint64_t value = UINT64_C(-1);
392
393         g_assert(words[1]);
394         addr = strtoull(words[1], NULL, 0);
395
396         if (words[0][4] == 'b') {
397             uint8_t data;
398             cpu_physical_memory_read(addr, &data, 1);
399             value = data;
400         } else if (words[0][4] == 'w') {
401             uint16_t data;
402             cpu_physical_memory_read(addr, &data, 2);
403             value = tswap16(data);
404         } else if (words[0][4] == 'l') {
405             uint32_t data;
406             cpu_physical_memory_read(addr, &data, 4);
407             value = tswap32(data);
408         } else if (words[0][4] == 'q') {
409             cpu_physical_memory_read(addr, &value, 8);
410             tswap64s(&value);
411         }
412         qtest_send_prefix(chr);
413         qtest_sendf(chr, "OK 0x%016" PRIx64 "\n", value);
414     } else if (strcmp(words[0], "read") == 0) {
415         uint64_t addr, len, i;
416         uint8_t *data;
417
418         g_assert(words[1] && words[2]);
419         addr = strtoull(words[1], NULL, 0);
420         len = strtoull(words[2], NULL, 0);
421
422         data = g_malloc(len);
423         cpu_physical_memory_read(addr, data, len);
424
425         qtest_send_prefix(chr);
426         qtest_send(chr, "OK 0x");
427         for (i = 0; i < len; i++) {
428             qtest_sendf(chr, "%02x", data[i]);
429         }
430         qtest_send(chr, "\n");
431
432         g_free(data);
433     } else if (strcmp(words[0], "b64read") == 0) {
434         uint64_t addr, len;
435         uint8_t *data;
436         gchar *b64_data;
437
438         g_assert(words[1] && words[2]);
439         addr = strtoull(words[1], NULL, 0);
440         len = strtoull(words[2], NULL, 0);
441
442         data = g_malloc(len);
443         cpu_physical_memory_read(addr, data, len);
444         b64_data = g_base64_encode(data, len);
445         qtest_send_prefix(chr);
446         qtest_sendf(chr, "OK %s\n", b64_data);
447
448         g_free(data);
449         g_free(b64_data);
450     } else if (strcmp(words[0], "write") == 0) {
451         uint64_t addr, len, i;
452         uint8_t *data;
453         size_t data_len;
454
455         g_assert(words[1] && words[2] && words[3]);
456         addr = strtoull(words[1], NULL, 0);
457         len = strtoull(words[2], NULL, 0);
458
459         data_len = strlen(words[3]);
460         if (data_len < 3) {
461             qtest_send(chr, "ERR invalid argument size\n");
462             return;
463         }
464
465         data = g_malloc(len);
466         for (i = 0; i < len; i++) {
467             if ((i * 2 + 4) <= data_len) {
468                 data[i] = hex2nib(words[3][i * 2 + 2]) << 4;
469                 data[i] |= hex2nib(words[3][i * 2 + 3]);
470             } else {
471                 data[i] = 0;
472             }
473         }
474         cpu_physical_memory_write(addr, data, len);
475         g_free(data);
476
477         qtest_send_prefix(chr);
478         qtest_send(chr, "OK\n");
479     } else if (strcmp(words[0], "memset") == 0) {
480         uint64_t addr, len;
481         uint8_t *data;
482         uint8_t pattern;
483
484         g_assert(words[1] && words[2] && words[3]);
485         addr = strtoull(words[1], NULL, 0);
486         len = strtoull(words[2], NULL, 0);
487         pattern = strtoull(words[3], NULL, 0);
488
489         data = g_malloc(len);
490         memset(data, pattern, len);
491         cpu_physical_memory_write(addr, data, len);
492         g_free(data);
493
494         qtest_send_prefix(chr);
495         qtest_send(chr, "OK\n");
496     }  else if (strcmp(words[0], "b64write") == 0) {
497         uint64_t addr, len;
498         uint8_t *data;
499         size_t data_len;
500         gsize out_len;
501
502         g_assert(words[1] && words[2] && words[3]);
503         addr = strtoull(words[1], NULL, 0);
504         len = strtoull(words[2], NULL, 0);
505
506         data_len = strlen(words[3]);
507         if (data_len < 3) {
508             qtest_send(chr, "ERR invalid argument size\n");
509             return;
510         }
511
512         data = g_base64_decode_inplace(words[3], &out_len);
513         if (out_len != len) {
514             qtest_log_send("b64write: data length mismatch (told %"PRIu64", "
515                            "found %zu)\n",
516                            len, out_len);
517             out_len = MIN(out_len, len);
518         }
519
520         cpu_physical_memory_write(addr, data, out_len);
521
522         qtest_send_prefix(chr);
523         qtest_send(chr, "OK\n");
524     } else if (qtest_enabled() && strcmp(words[0], "clock_step") == 0) {
525         int64_t ns;
526
527         if (words[1]) {
528             ns = strtoll(words[1], NULL, 0);
529         } else {
530             ns = qemu_clock_deadline_ns_all(QEMU_CLOCK_VIRTUAL);
531         }
532         qtest_clock_warp(qemu_clock_get_ns(QEMU_CLOCK_VIRTUAL) + ns);
533         qtest_send_prefix(chr);
534         qtest_sendf(chr, "OK %"PRIi64"\n",
535                     (int64_t)qemu_clock_get_ns(QEMU_CLOCK_VIRTUAL));
536     } else if (qtest_enabled() && strcmp(words[0], "clock_set") == 0) {
537         int64_t ns;
538
539         g_assert(words[1]);
540         ns = strtoll(words[1], NULL, 0);
541         qtest_clock_warp(ns);
542         qtest_send_prefix(chr);
543         qtest_sendf(chr, "OK %"PRIi64"\n",
544                     (int64_t)qemu_clock_get_ns(QEMU_CLOCK_VIRTUAL));
545     } else {
546         qtest_send_prefix(chr);
547         qtest_sendf(chr, "FAIL Unknown command '%s'\n", words[0]);
548     }
549 }
550
551 static void qtest_process_inbuf(CharDriverState *chr, GString *inbuf)
552 {
553     char *end;
554
555     while ((end = strchr(inbuf->str, '\n')) != NULL) {
556         size_t offset;
557         GString *cmd;
558         gchar **words;
559
560         offset = end - inbuf->str;
561
562         cmd = g_string_new_len(inbuf->str, offset);
563         g_string_erase(inbuf, 0, offset + 1);
564
565         words = g_strsplit(cmd->str, " ", 0);
566         qtest_process_command(chr, words);
567         g_strfreev(words);
568
569         g_string_free(cmd, TRUE);
570     }
571 }
572
573 static void qtest_read(void *opaque, const uint8_t *buf, int size)
574 {
575     CharDriverState *chr = opaque;
576
577     g_string_append_len(inbuf, (const gchar *)buf, size);
578     qtest_process_inbuf(chr, inbuf);
579 }
580
581 static int qtest_can_read(void *opaque)
582 {
583     return 1024;
584 }
585
586 static void qtest_event(void *opaque, int event)
587 {
588     int i;
589
590     switch (event) {
591     case CHR_EVENT_OPENED:
592         /*
593          * We used to call qemu_system_reset() here, hoping we could
594          * use the same process for multiple tests that way.  Never
595          * used.  Injects an extra reset even when it's not used, and
596          * that can mess up tests, e.g. -boot once.
597          */
598         for (i = 0; i < ARRAY_SIZE(irq_levels); i++) {
599             irq_levels[i] = 0;
600         }
601         qemu_gettimeofday(&start_time);
602         qtest_opened = true;
603         if (qtest_log_fp) {
604             fprintf(qtest_log_fp, "[I " FMT_timeval "] OPENED\n",
605                     (long) start_time.tv_sec, (long) start_time.tv_usec);
606         }
607         break;
608     case CHR_EVENT_CLOSED:
609         qtest_opened = false;
610         if (qtest_log_fp) {
611             qemu_timeval tv;
612             qtest_get_time(&tv);
613             fprintf(qtest_log_fp, "[I +" FMT_timeval "] CLOSED\n",
614                     (long) tv.tv_sec, (long) tv.tv_usec);
615         }
616         break;
617     default:
618         break;
619     }
620 }
621
622 static int qtest_init_accel(MachineState *ms)
623 {
624     QemuOpts *opts = qemu_opts_create(qemu_find_opts("icount"), NULL, 0,
625                                       &error_abort);
626     qemu_opt_set(opts, "shift", "0", &error_abort);
627     configure_icount(opts, &error_abort);
628     qemu_opts_del(opts);
629     return 0;
630 }
631
632 void qtest_init(const char *qtest_chrdev, const char *qtest_log, Error **errp)
633 {
634     CharDriverState *chr;
635
636     chr = qemu_chr_new("qtest", qtest_chrdev, NULL);
637
638     if (chr == NULL) {
639         error_setg(errp, "Failed to initialize device for qtest: \"%s\"",
640                    qtest_chrdev);
641         return;
642     }
643
644     if (qtest_log) {
645         if (strcmp(qtest_log, "none") != 0) {
646             qtest_log_fp = fopen(qtest_log, "w+");
647         }
648     } else {
649         qtest_log_fp = stderr;
650     }
651
652     qemu_chr_add_handlers(chr, qtest_can_read, qtest_read, qtest_event, chr);
653     qemu_chr_fe_set_echo(chr, true);
654
655     inbuf = g_string_new("");
656     qtest_chr = chr;
657 }
658
659 bool qtest_driver(void)
660 {
661     return qtest_chr;
662 }
663
664 static void qtest_accel_class_init(ObjectClass *oc, void *data)
665 {
666     AccelClass *ac = ACCEL_CLASS(oc);
667     ac->name = "QTest";
668     ac->available = qtest_available;
669     ac->init_machine = qtest_init_accel;
670     ac->allowed = &qtest_allowed;
671 }
672
673 #define TYPE_QTEST_ACCEL ACCEL_CLASS_NAME("qtest")
674
675 static const TypeInfo qtest_accel_type = {
676     .name = TYPE_QTEST_ACCEL,
677     .parent = TYPE_ACCEL,
678     .class_init = qtest_accel_class_init,
679 };
680
681 static void qtest_type_init(void)
682 {
683     type_register_static(&qtest_accel_type);
684 }
685
686 type_init(qtest_type_init);