tests/vmgenid-test.c

   1 /*
   2  * QTest testcase for VM Generation ID
   3  *
   4  * Copyright (c) 2016 Red Hat, Inc.
   5  * Copyright (c) 2017 Skyport Systems
   6  *
   7  * This work is licensed under the terms of the GNU GPL, version 2 or later.
   8  * See the COPYING file in the top-level directory.
   9  */
  10
  11 #include "qemu/osdep.h"
  12 #include "qemu/bitmap.h"
  13 #include "qemu/uuid.h"
  14 #include "hw/acpi/acpi-defs.h"
  15 #include "boot-sector.h"
  16 #include "acpi-utils.h"
  17 #include "libqtest.h"
  18 #include "qapi/qmp/qdict.h"
  19
  20 #define VGID_GUID "324e6eaf-d1d1-4bf6-bf41-b9bb6c91fb87"
  21 #define VMGENID_GUID_OFFSET 40   /* allow space for
  22                                   * OVMF SDT Header Probe Supressor
  23                                   */
  24 #define RSDP_ADDR_INVALID 0x100000 /* RSDP must be below this address */
  25
  26 typedef struct {
  27     AcpiTableHeader header;
  28     gchar name_op;
  29     gchar vgia[4];
  30     gchar val_op;
  31     uint32_t vgia_val;
  32 } QEMU_PACKED VgidTable;
  33
  34 static uint32_t acpi_find_vgia(QTestState *qts)
  35 {
  36     uint32_t rsdp_offset;
  37     uint32_t guid_offset = 0;
  38     uint8_t rsdp_table[36 /* ACPI 2.0+ RSDP size */];
  39     uint32_t rsdt, rsdt_table_length;
  40     AcpiRsdtDescriptorRev1 rsdt_table;
  41     size_t tables_nr;
  42     uint32_t *tables;
  43     AcpiTableHeader ssdt_table;
  44     VgidTable vgid_table;
  45     int i;
  46
  47     /* Wait for guest firmware to finish and start the payload. */
  48     boot_sector_test(qts);
  49
  50     /* Tables should be initialized now. */
  51     rsdp_offset = acpi_find_rsdp_address(qts);
  52
  53     g_assert_cmphex(rsdp_offset, <, RSDP_ADDR_INVALID);
  54
  55     acpi_parse_rsdp_table(qts, rsdp_offset, rsdp_table);
  56
  57     rsdt = acpi_get_rsdt_address(rsdp_table);
  58     g_assert(rsdt);
  59
  60     /* read the header */
  61     ACPI_READ_TABLE_HEADER(qts, &rsdt_table, rsdt);
  62     ACPI_ASSERT_CMP(rsdt_table.signature, "RSDT");
  63     rsdt_table_length = le32_to_cpu(rsdt_table.length);
  64
  65     /* compute the table entries in rsdt */
  66     g_assert_cmpint(rsdt_table_length, >, sizeof(AcpiRsdtDescriptorRev1));
  67     tables_nr = (rsdt_table_length - sizeof(AcpiRsdtDescriptorRev1)) /
  68                 sizeof(uint32_t);
  69
  70     /* get the addresses of the tables pointed by rsdt */
  71     tables = g_new0(uint32_t, tables_nr);
  72     ACPI_READ_ARRAY_PTR(qts, tables, tables_nr, rsdt);
  73
  74     for (i = 0; i < tables_nr; i++) {
  75         uint32_t addr = le32_to_cpu(tables[i]);
  76         ACPI_READ_TABLE_HEADER(qts, &ssdt_table, addr);
  77         if (!strncmp((char *)ssdt_table.oem_table_id, "VMGENID", 7)) {
  78             /* the first entry in the table should be VGIA
  79              * That's all we need
  80              */
  81             ACPI_READ_FIELD(qts, vgid_table.name_op, addr);
  82             g_assert(vgid_table.name_op == 0x08);  /* name */
  83             ACPI_READ_ARRAY(qts, vgid_table.vgia, addr);
  84             g_assert(memcmp(vgid_table.vgia, "VGIA", 4) == 0);
  85             ACPI_READ_FIELD(qts, vgid_table.val_op, addr);
  86             g_assert(vgid_table.val_op == 0x0C);  /* dword */
  87             ACPI_READ_FIELD(qts, vgid_table.vgia_val, addr);
  88             /* The GUID is written at a fixed offset into the fw_cfg file
  89              * in order to implement the "OVMF SDT Header probe suppressor"
  90              * see docs/specs/vmgenid.txt for more details
  91              */
  92             guid_offset = le32_to_cpu(vgid_table.vgia_val) + VMGENID_GUID_OFFSET;
  93             break;
  94         }
  95     }
  96     g_free(tables);
  97     return guid_offset;
  98 }
  99
 100 static void read_guid_from_memory(QTestState *qts, QemuUUID *guid)
 101 {
 102     uint32_t vmgenid_addr;
 103     int i;
 104
 105     vmgenid_addr = acpi_find_vgia(qts);
 106     g_assert(vmgenid_addr);
 107
 108     /* Read the GUID directly from guest memory */
 109     for (i = 0; i < 16; i++) {
 110         guid->data[i] = qtest_readb(qts, vmgenid_addr + i);
 111     }
 112     /* The GUID is in little-endian format in the guest, while QEMU
 113      * uses big-endian.  Swap after reading.
 114      */
 115     qemu_uuid_bswap(guid);
 116 }
 117
 118 static void read_guid_from_monitor(QTestState *qts, QemuUUID *guid)
 119 {
 120     QDict *rsp, *rsp_ret;
 121     const char *guid_str;
 122
 123     rsp = qtest_qmp(qts, "{ 'execute': 'query-vm-generation-id' }");
 124     if (qdict_haskey(rsp, "return")) {
 125         rsp_ret = qdict_get_qdict(rsp, "return");
 126         g_assert(qdict_haskey(rsp_ret, "guid"));
 127         guid_str = qdict_get_str(rsp_ret, "guid");
 128         g_assert(qemu_uuid_parse(guid_str, guid) == 0);
 129     }
 130     qobject_unref(rsp);
 131 }
 132
 133 static char disk[] = "tests/vmgenid-test-disk-XXXXXX";
 134
 135 #define GUID_CMD(guid)                          \
 136     "-machine accel=kvm:tcg "                   \
 137     "-device vmgenid,id=testvgid,guid=%s "      \
 138     "-drive id=hd0,if=none,file=%s,format=raw " \
 139     "-device ide-hd,drive=hd0 ", guid, disk
 140
 141 static void vmgenid_set_guid_test(void)
 142 {
 143     QemuUUID expected, measured;
 144     QTestState *qts;
 145
 146     g_assert(qemu_uuid_parse(VGID_GUID, &expected) == 0);
 147
 148     qts = qtest_initf(GUID_CMD(VGID_GUID));
 149
 150     /* Read the GUID from accessing guest memory */
 151     read_guid_from_memory(qts, &measured);
 152     g_assert(memcmp(measured.data, expected.data, sizeof(measured.data)) == 0);
 153
 154     qtest_quit(qts);
 155 }
 156
 157 static void vmgenid_set_guid_auto_test(void)
 158 {
 159     QemuUUID measured;
 160     QTestState *qts;
 161
 162     qts = qtest_initf(GUID_CMD("auto"));
 163
 164     read_guid_from_memory(qts, &measured);
 165
 166     /* Just check that the GUID is non-null */
 167     g_assert(!qemu_uuid_is_null(&measured));
 168
 169     qtest_quit(qts);
 170 }
 171
 172 static void vmgenid_query_monitor_test(void)
 173 {
 174     QemuUUID expected, measured;
 175     QTestState *qts;
 176
 177     g_assert(qemu_uuid_parse(VGID_GUID, &expected) == 0);
 178
 179     qts = qtest_initf(GUID_CMD(VGID_GUID));
 180
 181     /* Read the GUID via the monitor */
 182     read_guid_from_monitor(qts, &measured);
 183     g_assert(memcmp(measured.data, expected.data, sizeof(measured.data)) == 0);
 184
 185     qtest_quit(qts);
 186 }
 187
 188 int main(int argc, char **argv)
 189 {
 190     int ret;
 191
 192     ret = boot_sector_init(disk);
 193     if (ret) {
 194         return ret;
 195     }
 196
 197     g_test_init(&argc, &argv, NULL);
 198
 199     qtest_add_func("/vmgenid/vmgenid/set-guid",
 200                    vmgenid_set_guid_test);
 201     qtest_add_func("/vmgenid/vmgenid/set-guid-auto",
 202                    vmgenid_set_guid_auto_test);
 203     qtest_add_func("/vmgenid/vmgenid/query-monitor",
 204                    vmgenid_query_monitor_test);
 205     ret = g_test_run();
 206     boot_sector_cleanup(disk);
 207
 208     return ret;
 209 }