LXR qemu/tests/qtest/libqos/virtio-balloon.c

   1/*
   2 * libqos driver framework
   3 *
   4 * Copyright (c) 2018 Emanuele Giuseppe Esposito <e.emanuelegiuseppe@gmail.com>
   5 *
   6 * This library is free software; you can redistribute it and/or
   7 * modify it under the terms of the GNU Lesser General Public
   8 * License version 2.1 as published by the Free Software Foundation.
   9 *
  10 * This library is distributed in the hope that it will be useful,
  11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13 * Lesser General Public License for more details.
  14 *
  15 * You should have received a copy of the GNU Lesser General Public
  16 * License along with this library; if not, see <http://www.gnu.org/licenses/>
  17 */
  18
  19#include "qemu/osdep.h"
  20#include "../libqtest.h"
  21#include "qemu/module.h"
  22#include "qgraph.h"
  23#include "virtio-balloon.h"
  24
  25/* virtio-balloon-device */
  26static void *qvirtio_balloon_get_driver(QVirtioBalloon *v_balloon,
  27                                        const char *interface)
  28{
  29    if (!g_strcmp0(interface, "virtio-balloon")) {
  30        return v_balloon;
  31    }
  32    if (!g_strcmp0(interface, "virtio")) {
  33        return v_balloon->vdev;
  34    }
  35
  36    fprintf(stderr, "%s not present in virtio-balloon-device\n", interface);
  37    g_assert_not_reached();
  38}
  39
  40static void *qvirtio_balloon_device_get_driver(void *object,
  41                                               const char *interface)
  42{
  43    QVirtioBalloonDevice *v_balloon = object;
  44    return qvirtio_balloon_get_driver(&v_balloon->balloon, interface);
  45}
  46
  47static void *virtio_balloon_device_create(void *virtio_dev,
  48                                          QGuestAllocator *t_alloc,
  49                                          void *addr)
  50{
  51    QVirtioBalloonDevice *virtio_bdevice = g_new0(QVirtioBalloonDevice, 1);
  52    QVirtioBalloon *interface = &virtio_bdevice->balloon;
  53
  54    interface->vdev = virtio_dev;
  55
  56    virtio_bdevice->obj.get_driver = qvirtio_balloon_device_get_driver;
  57
  58    return &virtio_bdevice->obj;
  59}
  60
  61/* virtio-balloon-pci */
  62static void *qvirtio_balloon_pci_get_driver(void *object,
  63                                            const char *interface)
  64{
  65    QVirtioBalloonPCI *v_balloon = object;
  66    if (!g_strcmp0(interface, "pci-device")) {
  67        return v_balloon->pci_vdev.pdev;
  68    }
  69    return qvirtio_balloon_get_driver(&v_balloon->balloon, interface);
  70}
  71
  72static void *virtio_balloon_pci_create(void *pci_bus, QGuestAllocator *t_alloc,
  73                                  void *addr)
  74{
  75    QVirtioBalloonPCI *virtio_bpci = g_new0(QVirtioBalloonPCI, 1);
  76    QVirtioBalloon *interface = &virtio_bpci->balloon;
  77    QOSGraphObject *obj = &virtio_bpci->pci_vdev.obj;
  78
  79
  80    virtio_pci_init(&virtio_bpci->pci_vdev, pci_bus, addr);
  81    interface->vdev = &virtio_bpci->pci_vdev.vdev;
  82
  83    obj->get_driver = qvirtio_balloon_pci_get_driver;
  84
  85    return obj;
  86}
  87
  88static void virtio_balloon_register_nodes(void)
  89{
  90    QPCIAddress addr = {
  91        .devfn = QPCI_DEVFN(4, 0),
  92    };
  93
  94    QOSGraphEdgeOptions opts = {
  95        .extra_device_opts = "addr=04.0",
  96    };
  97
  98    /* virtio-balloon-device */
  99    qos_node_create_driver("virtio-balloon-device",
 100                            virtio_balloon_device_create);
 101    qos_node_consumes("virtio-balloon-device", "virtio-bus", NULL);
 102    qos_node_produces("virtio-balloon-device", "virtio");
 103    qos_node_produces("virtio-balloon-device", "virtio-balloon");
 104
 105    /* virtio-balloon-pci */
 106    add_qpci_address(&opts, &addr);
 107    qos_node_create_driver("virtio-balloon-pci", virtio_balloon_pci_create);
 108    qos_node_consumes("virtio-balloon-pci", "pci-bus", &opts);
 109    qos_node_produces("virtio-balloon-pci", "pci-device");
 110    qos_node_produces("virtio-balloon-pci", "virtio");
 111    qos_node_produces("virtio-balloon-pci", "virtio-balloon");
 112}
 113
 114libqos_init(virtio_balloon_register_nodes);
 115