180695202SCoiby Xu /*
280695202SCoiby Xu * libqos driver framework
380695202SCoiby Xu *
480695202SCoiby Xu * Based on tests/qtest/libqos/virtio-blk.c
580695202SCoiby Xu *
680695202SCoiby Xu * Copyright (c) 2020 Coiby Xu <coiby.xu@gmail.com>
780695202SCoiby Xu *
880695202SCoiby Xu * Copyright (c) 2018 Emanuele Giuseppe Esposito <e.emanuelegiuseppe@gmail.com>
980695202SCoiby Xu *
1080695202SCoiby Xu * This library is free software; you can redistribute it and/or
1180695202SCoiby Xu * modify it under the terms of the GNU Lesser General Public
1280695202SCoiby Xu * License version 2.1 as published by the Free Software Foundation.
1380695202SCoiby Xu *
1480695202SCoiby Xu * This library is distributed in the hope that it will be useful,
1580695202SCoiby Xu * but WITHOUT ANY WARRANTY; without even the implied warranty of
1680695202SCoiby Xu * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
1780695202SCoiby Xu * Lesser General Public License for more details.
1880695202SCoiby Xu *
1980695202SCoiby Xu * You should have received a copy of the GNU Lesser General Public
2080695202SCoiby Xu * License along with this library; if not, see <http://www.gnu.org/licenses/>
2180695202SCoiby Xu */
2280695202SCoiby Xu
2380695202SCoiby Xu #include "qemu/osdep.h"
24*907b5105SMarc-André Lureau #include "../libqtest.h"
2580695202SCoiby Xu #include "qemu/module.h"
2680695202SCoiby Xu #include "standard-headers/linux/virtio_blk.h"
2780695202SCoiby Xu #include "vhost-user-blk.h"
2880695202SCoiby Xu
2980695202SCoiby Xu #define PCI_SLOT 0x04
3080695202SCoiby Xu #define PCI_FN 0x00
3180695202SCoiby Xu
3280695202SCoiby Xu /* virtio-blk-device */
qvhost_user_blk_get_driver(QVhostUserBlk * v_blk,const char * interface)3380695202SCoiby Xu static void *qvhost_user_blk_get_driver(QVhostUserBlk *v_blk,
3480695202SCoiby Xu const char *interface)
3580695202SCoiby Xu {
3680695202SCoiby Xu if (!g_strcmp0(interface, "vhost-user-blk")) {
3780695202SCoiby Xu return v_blk;
3880695202SCoiby Xu }
3980695202SCoiby Xu if (!g_strcmp0(interface, "virtio")) {
4080695202SCoiby Xu return v_blk->vdev;
4180695202SCoiby Xu }
4280695202SCoiby Xu
4380695202SCoiby Xu fprintf(stderr, "%s not present in vhost-user-blk-device\n", interface);
4480695202SCoiby Xu g_assert_not_reached();
4580695202SCoiby Xu }
4680695202SCoiby Xu
qvhost_user_blk_device_get_driver(void * object,const char * interface)4780695202SCoiby Xu static void *qvhost_user_blk_device_get_driver(void *object,
4880695202SCoiby Xu const char *interface)
4980695202SCoiby Xu {
5080695202SCoiby Xu QVhostUserBlkDevice *v_blk = object;
5180695202SCoiby Xu return qvhost_user_blk_get_driver(&v_blk->blk, interface);
5280695202SCoiby Xu }
5380695202SCoiby Xu
vhost_user_blk_device_create(void * virtio_dev,QGuestAllocator * t_alloc,void * addr)5480695202SCoiby Xu static void *vhost_user_blk_device_create(void *virtio_dev,
5580695202SCoiby Xu QGuestAllocator *t_alloc,
5680695202SCoiby Xu void *addr)
5780695202SCoiby Xu {
5880695202SCoiby Xu QVhostUserBlkDevice *vhost_user_blk = g_new0(QVhostUserBlkDevice, 1);
5980695202SCoiby Xu QVhostUserBlk *interface = &vhost_user_blk->blk;
6080695202SCoiby Xu
6180695202SCoiby Xu interface->vdev = virtio_dev;
6280695202SCoiby Xu
6380695202SCoiby Xu vhost_user_blk->obj.get_driver = qvhost_user_blk_device_get_driver;
6480695202SCoiby Xu
6580695202SCoiby Xu return &vhost_user_blk->obj;
6680695202SCoiby Xu }
6780695202SCoiby Xu
6880695202SCoiby Xu /* virtio-blk-pci */
qvhost_user_blk_pci_get_driver(void * object,const char * interface)6980695202SCoiby Xu static void *qvhost_user_blk_pci_get_driver(void *object, const char *interface)
7080695202SCoiby Xu {
7180695202SCoiby Xu QVhostUserBlkPCI *v_blk = object;
7280695202SCoiby Xu if (!g_strcmp0(interface, "pci-device")) {
7380695202SCoiby Xu return v_blk->pci_vdev.pdev;
7480695202SCoiby Xu }
7580695202SCoiby Xu return qvhost_user_blk_get_driver(&v_blk->blk, interface);
7680695202SCoiby Xu }
7780695202SCoiby Xu
vhost_user_blk_pci_create(void * pci_bus,QGuestAllocator * t_alloc,void * addr)7880695202SCoiby Xu static void *vhost_user_blk_pci_create(void *pci_bus, QGuestAllocator *t_alloc,
7980695202SCoiby Xu void *addr)
8080695202SCoiby Xu {
8180695202SCoiby Xu QVhostUserBlkPCI *vhost_user_blk = g_new0(QVhostUserBlkPCI, 1);
8280695202SCoiby Xu QVhostUserBlk *interface = &vhost_user_blk->blk;
8380695202SCoiby Xu QOSGraphObject *obj = &vhost_user_blk->pci_vdev.obj;
8480695202SCoiby Xu
8580695202SCoiby Xu virtio_pci_init(&vhost_user_blk->pci_vdev, pci_bus, addr);
8680695202SCoiby Xu interface->vdev = &vhost_user_blk->pci_vdev.vdev;
8780695202SCoiby Xu
8880695202SCoiby Xu g_assert_cmphex(interface->vdev->device_type, ==, VIRTIO_ID_BLOCK);
8980695202SCoiby Xu
9080695202SCoiby Xu obj->get_driver = qvhost_user_blk_pci_get_driver;
9180695202SCoiby Xu
9280695202SCoiby Xu return obj;
9380695202SCoiby Xu }
9480695202SCoiby Xu
vhost_user_blk_register_nodes(void)9580695202SCoiby Xu static void vhost_user_blk_register_nodes(void)
9680695202SCoiby Xu {
9780695202SCoiby Xu /*
9880695202SCoiby Xu * FIXME: every test using these two nodes needs to setup a
9980695202SCoiby Xu * -drive,id=drive0 otherwise QEMU is not going to start.
10080695202SCoiby Xu * Therefore, we do not include "produces" edge for virtio
10180695202SCoiby Xu * and pci-device yet.
10280695202SCoiby Xu */
10380695202SCoiby Xu
10480695202SCoiby Xu char *arg = g_strdup_printf("id=drv0,chardev=char1,addr=%x.%x",
10580695202SCoiby Xu PCI_SLOT, PCI_FN);
10680695202SCoiby Xu
10780695202SCoiby Xu QPCIAddress addr = {
10880695202SCoiby Xu .devfn = QPCI_DEVFN(PCI_SLOT, PCI_FN),
10980695202SCoiby Xu };
11080695202SCoiby Xu
11180695202SCoiby Xu QOSGraphEdgeOptions opts = { };
11280695202SCoiby Xu
11380695202SCoiby Xu /* virtio-blk-device */
11480695202SCoiby Xu /** opts.extra_device_opts = "drive=drive0"; */
11580695202SCoiby Xu qos_node_create_driver("vhost-user-blk-device",
11680695202SCoiby Xu vhost_user_blk_device_create);
11780695202SCoiby Xu qos_node_consumes("vhost-user-blk-device", "virtio-bus", &opts);
11880695202SCoiby Xu qos_node_produces("vhost-user-blk-device", "vhost-user-blk");
11980695202SCoiby Xu
12080695202SCoiby Xu /* virtio-blk-pci */
12180695202SCoiby Xu opts.extra_device_opts = arg;
12280695202SCoiby Xu add_qpci_address(&opts, &addr);
12380695202SCoiby Xu qos_node_create_driver("vhost-user-blk-pci", vhost_user_blk_pci_create);
12480695202SCoiby Xu qos_node_consumes("vhost-user-blk-pci", "pci-bus", &opts);
12580695202SCoiby Xu qos_node_produces("vhost-user-blk-pci", "vhost-user-blk");
12680695202SCoiby Xu
12780695202SCoiby Xu g_free(arg);
12880695202SCoiby Xu }
12980695202SCoiby Xu
13080695202SCoiby Xu libqos_init(vhost_user_blk_register_nodes);
131