1
2
3
4
5
6
7
8
9
10
11
12
13
14
15#include "qemu/osdep.h"
16
17#include "libvhost-user-glib.h"
18
19
20
21G_STATIC_ASSERT((int)G_IO_IN == (int)VU_WATCH_IN);
22G_STATIC_ASSERT((int)G_IO_OUT == (int)VU_WATCH_OUT);
23G_STATIC_ASSERT((int)G_IO_PRI == (int)VU_WATCH_PRI);
24G_STATIC_ASSERT((int)G_IO_ERR == (int)VU_WATCH_ERR);
25G_STATIC_ASSERT((int)G_IO_HUP == (int)VU_WATCH_HUP);
26
27typedef struct VugSrc {
28 GSource parent;
29 VuDev *dev;
30 GPollFD gfd;
31} VugSrc;
32
33static gboolean
34vug_src_prepare(GSource *gsrc, gint *timeout)
35{
36 g_assert(timeout);
37
38 *timeout = -1;
39 return FALSE;
40}
41
42static gboolean
43vug_src_check(GSource *gsrc)
44{
45 VugSrc *src = (VugSrc *)gsrc;
46
47 g_assert(src);
48
49 return src->gfd.revents & src->gfd.events;
50}
51
52static gboolean
53vug_src_dispatch(GSource *gsrc, GSourceFunc cb, gpointer data)
54{
55 VugSrc *src = (VugSrc *)gsrc;
56
57 g_assert(src);
58
59 ((vu_watch_cb)cb)(src->dev, src->gfd.revents, data);
60
61 return G_SOURCE_CONTINUE;
62}
63
64static GSourceFuncs vug_src_funcs = {
65 vug_src_prepare,
66 vug_src_check,
67 vug_src_dispatch,
68 NULL
69};
70
71GSource *
72vug_source_new(VugDev *gdev, int fd, GIOCondition cond,
73 vu_watch_cb vu_cb, gpointer data)
74{
75 VuDev *dev = &gdev->parent;
76 GSource *gsrc;
77 VugSrc *src;
78 guint id;
79
80 g_assert(gdev);
81 g_assert(fd >= 0);
82 g_assert(vu_cb);
83
84 gsrc = g_source_new(&vug_src_funcs, sizeof(VugSrc));
85 g_source_set_callback(gsrc, (GSourceFunc)vu_cb, data, NULL);
86 src = (VugSrc *)gsrc;
87 src->dev = dev;
88 src->gfd.fd = fd;
89 src->gfd.events = cond;
90
91 g_source_add_poll(gsrc, &src->gfd);
92 id = g_source_attach(gsrc, NULL);
93 g_assert(id);
94 g_source_unref(gsrc);
95
96 return gsrc;
97}
98
99static void
100set_watch(VuDev *vu_dev, int fd, int vu_evt, vu_watch_cb cb, void *pvt)
101{
102 GSource *src;
103 VugDev *dev;
104
105 g_assert(vu_dev);
106 g_assert(fd >= 0);
107 g_assert(cb);
108
109 dev = container_of(vu_dev, VugDev, parent);
110 src = vug_source_new(dev, fd, vu_evt, cb, pvt);
111 g_hash_table_replace(dev->fdmap, GINT_TO_POINTER(fd), src);
112}
113
114static void
115remove_watch(VuDev *vu_dev, int fd)
116{
117 VugDev *dev;
118
119 g_assert(vu_dev);
120 g_assert(fd >= 0);
121
122 dev = container_of(vu_dev, VugDev, parent);
123 g_hash_table_remove(dev->fdmap, GINT_TO_POINTER(fd));
124}
125
126
127static void vug_watch(VuDev *dev, int condition, void *data)
128{
129 if (!vu_dispatch(dev) != 0) {
130 dev->panic(dev, "Error processing vhost message");
131 }
132}
133
134void
135vug_init(VugDev *dev, int socket,
136 vu_panic_cb panic, const VuDevIface *iface)
137{
138 g_assert(dev);
139 g_assert(iface);
140
141 vu_init(&dev->parent, socket, panic, set_watch, remove_watch, iface);
142 dev->fdmap = g_hash_table_new_full(NULL, NULL, NULL,
143 (GDestroyNotify) g_source_destroy);
144
145 dev->src = vug_source_new(dev, socket, G_IO_IN, vug_watch, NULL);
146}
147
148void
149vug_deinit(VugDev *dev)
150{
151 g_assert(dev);
152
153 g_hash_table_unref(dev->fdmap);
154 g_source_unref(dev->src);
155}
156