xref: /qemu/tests/qtest/libqos/virtio-rng.c (revision d3d400d3df0146575a8af44382978f404d57def9)
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 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 "libqos/qgraph.h"
22 #include "libqos/virtio-rng.h"
23 
24 /* virtio-rng-device */
25 static void *qvirtio_rng_get_driver(QVirtioRng *v_rng,
26                                     const char *interface)
27 {
28     if (!g_strcmp0(interface, "virtio-rng")) {
29         return v_rng;
30     }
31     if (!g_strcmp0(interface, "virtio")) {
32         return v_rng->vdev;
33     }
34 
35     fprintf(stderr, "%s not present in virtio-rng-device\n", interface);
36     g_assert_not_reached();
37 }
38 
39 static void *qvirtio_rng_device_get_driver(void *object,
40                                            const char *interface)
41 {
42     QVirtioRngDevice *v_rng = object;
43     return qvirtio_rng_get_driver(&v_rng->rng, interface);
44 }
45 
46 static void *virtio_rng_device_create(void *virtio_dev,
47                                       QGuestAllocator *t_alloc,
48                                       void *addr)
49 {
50     QVirtioRngDevice *virtio_rdevice = g_new0(QVirtioRngDevice, 1);
51     QVirtioRng *interface = &virtio_rdevice->rng;
52 
53     interface->vdev = virtio_dev;
54 
55     virtio_rdevice->obj.get_driver = qvirtio_rng_device_get_driver;
56 
57     return &virtio_rdevice->obj;
58 }
59 
60 /* virtio-rng-pci */
61 static void *qvirtio_rng_pci_get_driver(void *object, const char *interface)
62 {
63     QVirtioRngPCI *v_rng = object;
64     if (!g_strcmp0(interface, "pci-device")) {
65         return v_rng->pci_vdev.pdev;
66     }
67     return qvirtio_rng_get_driver(&v_rng->rng, interface);
68 }
69 
70 static void *virtio_rng_pci_create(void *pci_bus, QGuestAllocator *t_alloc,
71                                    void *addr)
72 {
73     QVirtioRngPCI *virtio_rpci = g_new0(QVirtioRngPCI, 1);
74     QVirtioRng *interface = &virtio_rpci->rng;
75     QOSGraphObject *obj = &virtio_rpci->pci_vdev.obj;
76 
77     virtio_pci_init(&virtio_rpci->pci_vdev, pci_bus, addr);
78     interface->vdev = &virtio_rpci->pci_vdev.vdev;
79 
80     obj->get_driver = qvirtio_rng_pci_get_driver;
81 
82     return obj;
83 }
84 
85 static void virtio_rng_register_nodes(void)
86 {
87     QPCIAddress addr = {
88         .devfn = QPCI_DEVFN(4, 0),
89     };
90 
91     QOSGraphEdgeOptions opts = {
92         .extra_device_opts = "addr=04.0",
93     };
94 
95     /* virtio-rng-device */
96     qos_node_create_driver("virtio-rng-device", virtio_rng_device_create);
97     qos_node_consumes("virtio-rng-device", "virtio-bus", NULL);
98     qos_node_produces("virtio-rng-device", "virtio");
99     qos_node_produces("virtio-rng-device", "virtio-rng");
100 
101     /* virtio-rng-pci */
102     add_qpci_address(&opts, &addr);
103     qos_node_create_driver("virtio-rng-pci", virtio_rng_pci_create);
104     qos_node_consumes("virtio-rng-pci", "pci-bus", &opts);
105     qos_node_produces("virtio-rng-pci", "pci-device");
106     qos_node_produces("virtio-rng-pci", "virtio");
107     qos_node_produces("virtio-rng-pci", "virtio-rng");
108 }
109 
110 libqos_init(virtio_rng_register_nodes);
111