cachepc-qemu

Fork of AMDESE/qemu with changes for cachepc side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-qemu
Log | Files | Refs | Submodules | LICENSE | sfeed.txt

vhost-user-blk.c (4225B)

      1/*
      2 * libqos driver framework
      3 *
      4 * Based on tests/qtest/libqos/virtio-blk.c
      5 *
      6 * Copyright (c) 2020 Coiby Xu <coiby.xu@gmail.com>
      7 *
      8 * Copyright (c) 2018 Emanuele Giuseppe Esposito <e.emanuelegiuseppe@gmail.com>
      9 *
     10 * This library is free software; you can redistribute it and/or
     11 * modify it under the terms of the GNU Lesser General Public
     12 * License version 2.1 as published by the Free Software Foundation.
     13 *
     14 * This library is distributed in the hope that it will be useful,
     15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
     16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
     17 * Lesser General Public License for more details.
     18 *
     19 * You should have received a copy of the GNU Lesser General Public
     20 * License along with this library; if not, see <http://www.gnu.org/licenses/>
     21 */
     22
     23#include "qemu/osdep.h"
     24#include "libqtest.h"
     25#include "qemu/module.h"
     26#include "standard-headers/linux/virtio_blk.h"
     27#include "vhost-user-blk.h"
     28
     29#define PCI_SLOT                0x04
     30#define PCI_FN                  0x00
     31
     32/* virtio-blk-device */
     33static void *qvhost_user_blk_get_driver(QVhostUserBlk *v_blk,
     34                                    const char *interface)
     35{
     36    if (!g_strcmp0(interface, "vhost-user-blk")) {
     37        return v_blk;
     38    }
     39    if (!g_strcmp0(interface, "virtio")) {
     40        return v_blk->vdev;
     41    }
     42
     43    fprintf(stderr, "%s not present in vhost-user-blk-device\n", interface);
     44    g_assert_not_reached();
     45}
     46
     47static void *qvhost_user_blk_device_get_driver(void *object,
     48                                           const char *interface)
     49{
     50    QVhostUserBlkDevice *v_blk = object;
     51    return qvhost_user_blk_get_driver(&v_blk->blk, interface);
     52}
     53
     54static void *vhost_user_blk_device_create(void *virtio_dev,
     55                                      QGuestAllocator *t_alloc,
     56                                      void *addr)
     57{
     58    QVhostUserBlkDevice *vhost_user_blk = g_new0(QVhostUserBlkDevice, 1);
     59    QVhostUserBlk *interface = &vhost_user_blk->blk;
     60
     61    interface->vdev = virtio_dev;
     62
     63    vhost_user_blk->obj.get_driver = qvhost_user_blk_device_get_driver;
     64
     65    return &vhost_user_blk->obj;
     66}
     67
     68/* virtio-blk-pci */
     69static void *qvhost_user_blk_pci_get_driver(void *object, const char *interface)
     70{
     71    QVhostUserBlkPCI *v_blk = object;
     72    if (!g_strcmp0(interface, "pci-device")) {
     73        return v_blk->pci_vdev.pdev;
     74    }
     75    return qvhost_user_blk_get_driver(&v_blk->blk, interface);
     76}
     77
     78static void *vhost_user_blk_pci_create(void *pci_bus, QGuestAllocator *t_alloc,
     79                                      void *addr)
     80{
     81    QVhostUserBlkPCI *vhost_user_blk = g_new0(QVhostUserBlkPCI, 1);
     82    QVhostUserBlk *interface = &vhost_user_blk->blk;
     83    QOSGraphObject *obj = &vhost_user_blk->pci_vdev.obj;
     84
     85    virtio_pci_init(&vhost_user_blk->pci_vdev, pci_bus, addr);
     86    interface->vdev = &vhost_user_blk->pci_vdev.vdev;
     87
     88    g_assert_cmphex(interface->vdev->device_type, ==, VIRTIO_ID_BLOCK);
     89
     90    obj->get_driver = qvhost_user_blk_pci_get_driver;
     91
     92    return obj;
     93}
     94
     95static void vhost_user_blk_register_nodes(void)
     96{
     97    /*
     98     * FIXME: every test using these two nodes needs to setup a
     99     * -drive,id=drive0 otherwise QEMU is not going to start.
    100     * Therefore, we do not include "produces" edge for virtio
    101     * and pci-device yet.
    102     */
    103
    104    char *arg = g_strdup_printf("id=drv0,chardev=char1,addr=%x.%x",
    105                                PCI_SLOT, PCI_FN);
    106
    107    QPCIAddress addr = {
    108        .devfn = QPCI_DEVFN(PCI_SLOT, PCI_FN),
    109    };
    110
    111    QOSGraphEdgeOptions opts = { };
    112
    113    /* virtio-blk-device */
    114    /** opts.extra_device_opts = "drive=drive0"; */
    115    qos_node_create_driver("vhost-user-blk-device",
    116                           vhost_user_blk_device_create);
    117    qos_node_consumes("vhost-user-blk-device", "virtio-bus", &opts);
    118    qos_node_produces("vhost-user-blk-device", "vhost-user-blk");
    119
    120    /* virtio-blk-pci */
    121    opts.extra_device_opts = arg;
    122    add_qpci_address(&opts, &addr);
    123    qos_node_create_driver("vhost-user-blk-pci", vhost_user_blk_pci_create);
    124    qos_node_consumes("vhost-user-blk-pci", "pci-bus", &opts);
    125    qos_node_produces("vhost-user-blk-pci", "vhost-user-blk");
    126
    127    g_free(arg);
    128}
    129
    130libqos_init(vhost_user_blk_register_nodes);