1 | /* |
2 | * QEMU network listener |
3 | * |
4 | * Copyright (c) 2016-2017 Red Hat, Inc. |
5 | * |
6 | * This program is free software; you can redistribute it and/or modify |
7 | * it under the terms of the GNU General Public License as published by |
8 | * the Free Software Foundation; either version 2 of the License, or |
9 | * (at your option) any later version. |
10 | * |
11 | * This program is distributed in the hope that it will be useful, |
12 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
13 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
14 | * GNU General Public License for more details. |
15 | * |
16 | * You should have received a copy of the GNU General Public License along |
17 | * with this program; if not, see <http://www.gnu.org/licenses/>. |
18 | * |
19 | */ |
20 | |
21 | #include "qemu/osdep.h" |
22 | #include "io/net-listener.h" |
23 | #include "io/dns-resolver.h" |
24 | #include "qapi/error.h" |
25 | #include "qemu/module.h" |
26 | |
27 | QIONetListener *qio_net_listener_new(void) |
28 | { |
29 | return QIO_NET_LISTENER(object_new(TYPE_QIO_NET_LISTENER)); |
30 | } |
31 | |
32 | void qio_net_listener_set_name(QIONetListener *listener, |
33 | const char *name) |
34 | { |
35 | g_free(listener->name); |
36 | listener->name = g_strdup(name); |
37 | } |
38 | |
39 | |
40 | static gboolean qio_net_listener_channel_func(QIOChannel *ioc, |
41 | GIOCondition condition, |
42 | gpointer opaque) |
43 | { |
44 | QIONetListener *listener = QIO_NET_LISTENER(opaque); |
45 | QIOChannelSocket *sioc; |
46 | |
47 | sioc = qio_channel_socket_accept(QIO_CHANNEL_SOCKET(ioc), |
48 | NULL); |
49 | if (!sioc) { |
50 | return TRUE; |
51 | } |
52 | |
53 | if (listener->io_func) { |
54 | listener->io_func(listener, sioc, listener->io_data); |
55 | } |
56 | |
57 | object_unref(OBJECT(sioc)); |
58 | |
59 | return TRUE; |
60 | } |
61 | |
62 | |
63 | int qio_net_listener_open_sync(QIONetListener *listener, |
64 | SocketAddress *addr, |
65 | int num, |
66 | Error **errp) |
67 | { |
68 | QIODNSResolver *resolver = qio_dns_resolver_get_instance(); |
69 | SocketAddress **resaddrs; |
70 | size_t nresaddrs; |
71 | size_t i; |
72 | Error *err = NULL; |
73 | bool success = false; |
74 | |
75 | if (qio_dns_resolver_lookup_sync(resolver, |
76 | addr, |
77 | &nresaddrs, |
78 | &resaddrs, |
79 | errp) < 0) { |
80 | return -1; |
81 | } |
82 | |
83 | for (i = 0; i < nresaddrs; i++) { |
84 | QIOChannelSocket *sioc = qio_channel_socket_new(); |
85 | |
86 | if (qio_channel_socket_listen_sync(sioc, resaddrs[i], num, |
87 | err ? NULL : &err) == 0) { |
88 | success = true; |
89 | |
90 | qio_net_listener_add(listener, sioc); |
91 | } |
92 | |
93 | qapi_free_SocketAddress(resaddrs[i]); |
94 | object_unref(OBJECT(sioc)); |
95 | } |
96 | g_free(resaddrs); |
97 | |
98 | if (success) { |
99 | error_free(err); |
100 | return 0; |
101 | } else { |
102 | error_propagate(errp, err); |
103 | return -1; |
104 | } |
105 | } |
106 | |
107 | |
108 | void qio_net_listener_add(QIONetListener *listener, |
109 | QIOChannelSocket *sioc) |
110 | { |
111 | if (listener->name) { |
112 | char *name = g_strdup_printf("%s-listen" , listener->name); |
113 | qio_channel_set_name(QIO_CHANNEL(sioc), name); |
114 | g_free(name); |
115 | } |
116 | |
117 | listener->sioc = g_renew(QIOChannelSocket *, listener->sioc, |
118 | listener->nsioc + 1); |
119 | listener->io_source = g_renew(typeof(listener->io_source[0]), |
120 | listener->io_source, |
121 | listener->nsioc + 1); |
122 | listener->sioc[listener->nsioc] = sioc; |
123 | listener->io_source[listener->nsioc] = NULL; |
124 | |
125 | object_ref(OBJECT(sioc)); |
126 | listener->connected = true; |
127 | |
128 | if (listener->io_func != NULL) { |
129 | object_ref(OBJECT(listener)); |
130 | listener->io_source[listener->nsioc] = qio_channel_add_watch_source( |
131 | QIO_CHANNEL(listener->sioc[listener->nsioc]), G_IO_IN, |
132 | qio_net_listener_channel_func, |
133 | listener, (GDestroyNotify)object_unref, NULL); |
134 | } |
135 | |
136 | listener->nsioc++; |
137 | } |
138 | |
139 | |
140 | void qio_net_listener_set_client_func_full(QIONetListener *listener, |
141 | QIONetListenerClientFunc func, |
142 | gpointer data, |
143 | GDestroyNotify notify, |
144 | GMainContext *context) |
145 | { |
146 | size_t i; |
147 | |
148 | if (listener->io_notify) { |
149 | listener->io_notify(listener->io_data); |
150 | } |
151 | listener->io_func = func; |
152 | listener->io_data = data; |
153 | listener->io_notify = notify; |
154 | |
155 | for (i = 0; i < listener->nsioc; i++) { |
156 | if (listener->io_source[i]) { |
157 | g_source_destroy(listener->io_source[i]); |
158 | g_source_unref(listener->io_source[i]); |
159 | listener->io_source[i] = NULL; |
160 | } |
161 | } |
162 | |
163 | if (listener->io_func != NULL) { |
164 | for (i = 0; i < listener->nsioc; i++) { |
165 | object_ref(OBJECT(listener)); |
166 | listener->io_source[i] = qio_channel_add_watch_source( |
167 | QIO_CHANNEL(listener->sioc[i]), G_IO_IN, |
168 | qio_net_listener_channel_func, |
169 | listener, (GDestroyNotify)object_unref, context); |
170 | } |
171 | } |
172 | } |
173 | |
174 | void qio_net_listener_set_client_func(QIONetListener *listener, |
175 | QIONetListenerClientFunc func, |
176 | gpointer data, |
177 | GDestroyNotify notify) |
178 | { |
179 | qio_net_listener_set_client_func_full(listener, func, data, |
180 | notify, NULL); |
181 | } |
182 | |
183 | struct QIONetListenerClientWaitData { |
184 | QIOChannelSocket *sioc; |
185 | GMainLoop *loop; |
186 | }; |
187 | |
188 | |
189 | static gboolean qio_net_listener_wait_client_func(QIOChannel *ioc, |
190 | GIOCondition condition, |
191 | gpointer opaque) |
192 | { |
193 | struct QIONetListenerClientWaitData *data = opaque; |
194 | QIOChannelSocket *sioc; |
195 | |
196 | sioc = qio_channel_socket_accept(QIO_CHANNEL_SOCKET(ioc), |
197 | NULL); |
198 | if (!sioc) { |
199 | return TRUE; |
200 | } |
201 | |
202 | if (data->sioc) { |
203 | object_unref(OBJECT(sioc)); |
204 | } else { |
205 | data->sioc = sioc; |
206 | g_main_loop_quit(data->loop); |
207 | } |
208 | |
209 | return TRUE; |
210 | } |
211 | |
212 | QIOChannelSocket *qio_net_listener_wait_client(QIONetListener *listener) |
213 | { |
214 | GMainContext *ctxt = g_main_context_new(); |
215 | GMainLoop *loop = g_main_loop_new(ctxt, TRUE); |
216 | GSource **sources; |
217 | struct QIONetListenerClientWaitData data = { |
218 | .sioc = NULL, |
219 | .loop = loop |
220 | }; |
221 | size_t i; |
222 | |
223 | for (i = 0; i < listener->nsioc; i++) { |
224 | if (listener->io_source[i]) { |
225 | g_source_destroy(listener->io_source[i]); |
226 | g_source_unref(listener->io_source[i]); |
227 | listener->io_source[i] = NULL; |
228 | } |
229 | } |
230 | |
231 | sources = g_new0(GSource *, listener->nsioc); |
232 | for (i = 0; i < listener->nsioc; i++) { |
233 | sources[i] = qio_channel_create_watch(QIO_CHANNEL(listener->sioc[i]), |
234 | G_IO_IN); |
235 | |
236 | g_source_set_callback(sources[i], |
237 | (GSourceFunc)qio_net_listener_wait_client_func, |
238 | &data, |
239 | NULL); |
240 | g_source_attach(sources[i], ctxt); |
241 | } |
242 | |
243 | g_main_loop_run(loop); |
244 | |
245 | for (i = 0; i < listener->nsioc; i++) { |
246 | g_source_unref(sources[i]); |
247 | } |
248 | g_free(sources); |
249 | g_main_loop_unref(loop); |
250 | g_main_context_unref(ctxt); |
251 | |
252 | if (listener->io_func != NULL) { |
253 | for (i = 0; i < listener->nsioc; i++) { |
254 | object_ref(OBJECT(listener)); |
255 | listener->io_source[i] = qio_channel_add_watch_source( |
256 | QIO_CHANNEL(listener->sioc[i]), G_IO_IN, |
257 | qio_net_listener_channel_func, |
258 | listener, (GDestroyNotify)object_unref, NULL); |
259 | } |
260 | } |
261 | |
262 | return data.sioc; |
263 | } |
264 | |
265 | void qio_net_listener_disconnect(QIONetListener *listener) |
266 | { |
267 | size_t i; |
268 | |
269 | if (!listener->connected) { |
270 | return; |
271 | } |
272 | |
273 | for (i = 0; i < listener->nsioc; i++) { |
274 | if (listener->io_source[i]) { |
275 | g_source_destroy(listener->io_source[i]); |
276 | g_source_unref(listener->io_source[i]); |
277 | listener->io_source[i] = NULL; |
278 | } |
279 | qio_channel_close(QIO_CHANNEL(listener->sioc[i]), NULL); |
280 | } |
281 | listener->connected = false; |
282 | } |
283 | |
284 | |
285 | bool qio_net_listener_is_connected(QIONetListener *listener) |
286 | { |
287 | return listener->connected; |
288 | } |
289 | |
290 | static void qio_net_listener_finalize(Object *obj) |
291 | { |
292 | QIONetListener *listener = QIO_NET_LISTENER(obj); |
293 | size_t i; |
294 | |
295 | qio_net_listener_disconnect(listener); |
296 | |
297 | for (i = 0; i < listener->nsioc; i++) { |
298 | object_unref(OBJECT(listener->sioc[i])); |
299 | } |
300 | g_free(listener->io_source); |
301 | g_free(listener->sioc); |
302 | g_free(listener->name); |
303 | } |
304 | |
305 | static const TypeInfo qio_net_listener_info = { |
306 | .parent = TYPE_OBJECT, |
307 | .name = TYPE_QIO_NET_LISTENER, |
308 | .instance_size = sizeof(QIONetListener), |
309 | .instance_finalize = qio_net_listener_finalize, |
310 | .class_size = sizeof(QIONetListenerClass), |
311 | }; |
312 | |
313 | |
314 | static void qio_net_listener_register_types(void) |
315 | { |
316 | type_register_static(&qio_net_listener_info); |
317 | } |
318 | |
319 | |
320 | type_init(qio_net_listener_register_types); |
321 | |