LXR qemu/tests/nvme-test.c

   1/*
   2 * QTest testcase for NVMe
   3 *
   4 * Copyright (c) 2014 SUSE LINUX Products GmbH
   5 *
   6 * This work is licensed under the terms of the GNU GPL, version 2 or later.
   7 * See the COPYING file in the top-level directory.
   8 */
   9
  10#include "qemu/osdep.h"
  11#include "qemu/module.h"
  12#include "qemu/units.h"
  13#include "libqtest.h"
  14#include "libqos/qgraph.h"
  15#include "libqos/pci.h"
  16
  17typedef struct QNvme QNvme;
  18
  19struct QNvme {
  20    QOSGraphObject obj;
  21    QPCIDevice dev;
  22};
  23
  24static void *nvme_get_driver(void *obj, const char *interface)
  25{
  26    QNvme *nvme = obj;
  27
  28    if (!g_strcmp0(interface, "pci-device")) {
  29        return &nvme->dev;
  30    }
  31
  32    fprintf(stderr, "%s not present in nvme\n", interface);
  33    g_assert_not_reached();
  34}
  35
  36static void *nvme_create(void *pci_bus, QGuestAllocator *alloc, void *addr)
  37{
  38    QNvme *nvme = g_new0(QNvme, 1);
  39    QPCIBus *bus = pci_bus;
  40
  41    qpci_device_init(&nvme->dev, bus, addr);
  42    nvme->obj.get_driver = nvme_get_driver;
  43
  44    return &nvme->obj;
  45}
  46
  47/* This used to cause a NULL pointer dereference.  */
  48static void nvmetest_oob_cmb_test(void *obj, void *data, QGuestAllocator *alloc)
  49{
  50    const int cmb_bar_size = 2 * MiB;
  51    QNvme *nvme = obj;
  52    QPCIDevice *pdev = &nvme->dev;
  53    QPCIBar bar;
  54
  55    qpci_device_enable(pdev);
  56    bar = qpci_iomap(pdev, 2, NULL);
  57
  58    qpci_io_writel(pdev, bar, 0, 0xccbbaa99);
  59    g_assert_cmpint(qpci_io_readb(pdev, bar, 0), ==, 0x99);
  60    g_assert_cmpint(qpci_io_readw(pdev, bar, 0), ==, 0xaa99);
  61
  62    /* Test partially out-of-bounds accesses.  */
  63    qpci_io_writel(pdev, bar, cmb_bar_size - 1, 0x44332211);
  64    g_assert_cmpint(qpci_io_readb(pdev, bar, cmb_bar_size - 1), ==, 0x11);
  65    g_assert_cmpint(qpci_io_readw(pdev, bar, cmb_bar_size - 1), !=, 0x2211);
  66    g_assert_cmpint(qpci_io_readl(pdev, bar, cmb_bar_size - 1), !=, 0x44332211);
  67}
  68
  69static void nvme_register_nodes(void)
  70{
  71    QOSGraphEdgeOptions opts = {
  72        .extra_device_opts = "addr=04.0,drive=drv0,serial=foo",
  73        .before_cmd_line = "-drive id=drv0,if=none,file=null-co://,format=raw",
  74    };
  75
  76    add_qpci_address(&opts, &(QPCIAddress) { .devfn = QPCI_DEVFN(4, 0) });
  77
  78    qos_node_create_driver("nvme", nvme_create);
  79    qos_node_consumes("nvme", "pci-bus", &opts);
  80    qos_node_produces("nvme", "pci-device");
  81
  82    qos_add_test("oob-cmb-access", "nvme", nvmetest_oob_cmb_test, &(QOSGraphTestOptions) {
  83        .edge.extra_device_opts = "cmb_size_mb=2"
  84    });
  85}
  86
  87libqos_init(nvme_register_nodes);
  88