]> git.proxmox.com Git - qemu.git/blame - qemu-sockets.c
tests: do not include tools-obj-y
[qemu.git] / qemu-sockets.c
CommitLineData
305b0eb2
AL
1/*
2 * inet and unix socket functions for qemu
3 *
4 * (c) 2008 Gerd Hoffmann <kraxel@redhat.com>
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; under version 2 of the License.
9 *
10 * This program 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
13 * GNU General Public License for more details.
e6d91ab6
PB
14 *
15 * Contributions after 2012-01-13 are licensed under the terms of the
16 * GNU GPL, version 2 or (at your option) any later version.
305b0eb2 17 */
d247d25f
AL
18#include <stdio.h>
19#include <stdlib.h>
20#include <string.h>
21#include <ctype.h>
22#include <errno.h>
23#include <unistd.h>
24
25#include "qemu_socket.h"
47398b9c 26#include "qemu-common.h" /* for qemu_isdigit */
233aa5c2 27#include "main-loop.h"
d247d25f
AL
28
29#ifndef AI_ADDRCONFIG
30# define AI_ADDRCONFIG 0
31#endif
32
d247d25f
AL
33static const int on=1, off=0;
34
2af2bf67 35/* used temporarely until all users are converted to QemuOpts */
c1390903 36static QemuOptsList dummy_opts = {
2af2bf67 37 .name = "dummy",
72cf2d4f 38 .head = QTAILQ_HEAD_INITIALIZER(dummy_opts.head),
2af2bf67
GH
39 .desc = {
40 {
41 .name = "path",
42 .type = QEMU_OPT_STRING,
f4c94c7c
GH
43 },{
44 .name = "host",
45 .type = QEMU_OPT_STRING,
46 },{
47 .name = "port",
48 .type = QEMU_OPT_STRING,
49 },{
50 .name = "to",
51 .type = QEMU_OPT_NUMBER,
52 },{
53 .name = "ipv4",
54 .type = QEMU_OPT_BOOL,
55 },{
56 .name = "ipv6",
57 .type = QEMU_OPT_BOOL,
2af2bf67
GH
58 },
59 { /* end if list */ }
60 },
61};
62
d247d25f
AL
63static int inet_getport(struct addrinfo *e)
64{
65 struct sockaddr_in *i4;
66 struct sockaddr_in6 *i6;
67
68 switch (e->ai_family) {
69 case PF_INET6:
70 i6 = (void*)e->ai_addr;
71 return ntohs(i6->sin6_port);
72 case PF_INET:
73 i4 = (void*)e->ai_addr;
74 return ntohs(i4->sin_port);
75 default:
76 return 0;
77 }
78}
79
80static void inet_setport(struct addrinfo *e, int port)
81{
82 struct sockaddr_in *i4;
83 struct sockaddr_in6 *i6;
84
85 switch (e->ai_family) {
86 case PF_INET6:
87 i6 = (void*)e->ai_addr;
88 i6->sin6_port = htons(port);
89 break;
90 case PF_INET:
91 i4 = (void*)e->ai_addr;
92 i4->sin_port = htons(port);
93 break;
94 }
95}
96
c9c4b34e 97const char *inet_strfamily(int family)
d247d25f
AL
98{
99 switch (family) {
100 case PF_INET6: return "ipv6";
101 case PF_INET: return "ipv4";
102 case PF_UNIX: return "unix";
103 }
c445321e 104 return "unknown";
d247d25f
AL
105}
106
029409e5 107int inet_listen_opts(QemuOpts *opts, int port_offset, Error **errp)
d247d25f
AL
108{
109 struct addrinfo ai,*res,*e;
e5bc776f 110 const char *addr;
d247d25f
AL
111 char port[33];
112 char uaddr[INET6_ADDRSTRLEN+1];
113 char uport[33];
877691f9 114 int slisten, rc, to, port_min, port_max, p;
d247d25f
AL
115
116 memset(&ai,0, sizeof(ai));
117 ai.ai_flags = AI_PASSIVE | AI_ADDRCONFIG;
118 ai.ai_family = PF_UNSPEC;
e5bc776f 119 ai.ai_socktype = SOCK_STREAM;
d247d25f 120
e23a22e6
JO
121 if ((qemu_opt_get(opts, "host") == NULL) ||
122 (qemu_opt_get(opts, "port") == NULL)) {
a12fb8ad 123 error_setg(errp, "host and/or port not specified");
e5bc776f 124 return -1;
d247d25f 125 }
e5bc776f
GH
126 pstrcpy(port, sizeof(port), qemu_opt_get(opts, "port"));
127 addr = qemu_opt_get(opts, "host");
d247d25f 128
e5bc776f
GH
129 to = qemu_opt_get_number(opts, "to", 0);
130 if (qemu_opt_get_bool(opts, "ipv4", 0))
d247d25f 131 ai.ai_family = PF_INET;
e5bc776f 132 if (qemu_opt_get_bool(opts, "ipv6", 0))
d247d25f
AL
133 ai.ai_family = PF_INET6;
134
135 /* lookup */
136 if (port_offset)
137 snprintf(port, sizeof(port), "%d", atoi(port) + port_offset);
138 rc = getaddrinfo(strlen(addr) ? addr : NULL, port, &ai, &res);
139 if (rc != 0) {
a12fb8ad
PB
140 error_setg(errp, "address resolution failed for %s:%s: %s", addr, port,
141 gai_strerror(rc));
d247d25f
AL
142 return -1;
143 }
d247d25f
AL
144
145 /* create socket + bind */
146 for (e = res; e != NULL; e = e->ai_next) {
39b6efc8 147 getnameinfo((struct sockaddr*)e->ai_addr,e->ai_addrlen,
148 uaddr,INET6_ADDRSTRLEN,uport,32,
149 NI_NUMERICHOST | NI_NUMERICSERV);
40ff6d7e 150 slisten = qemu_socket(e->ai_family, e->ai_socktype, e->ai_protocol);
39b6efc8 151 if (slisten < 0) {
029409e5 152 if (!e->ai_next) {
a12fb8ad 153 error_set_errno(errp, errno, QERR_SOCKET_CREATE_FAILED);
029409e5 154 }
39b6efc8 155 continue;
156 }
d247d25f
AL
157
158 setsockopt(slisten,SOL_SOCKET,SO_REUSEADDR,(void*)&on,sizeof(on));
159#ifdef IPV6_V6ONLY
160 if (e->ai_family == PF_INET6) {
161 /* listen on both ipv4 and ipv6 */
39b6efc8 162 setsockopt(slisten,IPPROTO_IPV6,IPV6_V6ONLY,(void*)&off,
163 sizeof(off));
d247d25f
AL
164 }
165#endif
166
877691f9
MA
167 port_min = inet_getport(e);
168 port_max = to ? to + port_offset : port_min;
169 for (p = port_min; p <= port_max; p++) {
170 inet_setport(e, p);
d247d25f 171 if (bind(slisten, e->ai_addr, e->ai_addrlen) == 0) {
d247d25f
AL
172 goto listen;
173 }
877691f9 174 if (p == port_max) {
029409e5 175 if (!e->ai_next) {
a12fb8ad 176 error_set_errno(errp, errno, QERR_SOCKET_BIND_FAILED);
029409e5 177 }
d247d25f 178 }
d247d25f
AL
179 }
180 closesocket(slisten);
181 }
d247d25f
AL
182 freeaddrinfo(res);
183 return -1;
184
185listen:
186 if (listen(slisten,1) != 0) {
a12fb8ad 187 error_set_errno(errp, errno, QERR_SOCKET_LISTEN_FAILED);
d247d25f 188 closesocket(slisten);
39b6efc8 189 freeaddrinfo(res);
d247d25f
AL
190 return -1;
191 }
e5bc776f
GH
192 snprintf(uport, sizeof(uport), "%d", inet_getport(e) - port_offset);
193 qemu_opt_set(opts, "host", uaddr);
194 qemu_opt_set(opts, "port", uport);
195 qemu_opt_set(opts, "ipv6", (e->ai_family == PF_INET6) ? "on" : "off");
196 qemu_opt_set(opts, "ipv4", (e->ai_family != PF_INET6) ? "on" : "off");
d247d25f
AL
197 freeaddrinfo(res);
198 return slisten;
199}
200
05bc1d8a
MT
201#ifdef _WIN32
202#define QEMU_SOCKET_RC_INPROGRESS(rc) \
203 ((rc) == -EINPROGRESS || (rc) == -EWOULDBLOCK || (rc) == -WSAEALREADY)
204#else
205#define QEMU_SOCKET_RC_INPROGRESS(rc) \
206 ((rc) == -EINPROGRESS)
207#endif
208
233aa5c2
OW
209/* Struct to store connect state for non blocking connect */
210typedef struct ConnectState {
211 int fd;
212 struct addrinfo *addr_list;
213 struct addrinfo *current_addr;
214 NonBlockingConnectHandler *callback;
215 void *opaque;
216} ConnectState;
217
218static int inet_connect_addr(struct addrinfo *addr, bool *in_progress,
11663b55 219 ConnectState *connect_state, Error **errp);
233aa5c2
OW
220
221static void wait_for_connect(void *opaque)
d247d25f 222{
233aa5c2
OW
223 ConnectState *s = opaque;
224 int val = 0, rc = 0;
225 socklen_t valsize = sizeof(val);
226 bool in_progress;
227
228 qemu_set_fd_handler2(s->fd, NULL, NULL, NULL, NULL);
229
230 do {
231 rc = getsockopt(s->fd, SOL_SOCKET, SO_ERROR, (void *) &val, &valsize);
232 } while (rc == -1 && socket_error() == EINTR);
233
234 /* update rc to contain error */
235 if (!rc && val) {
236 rc = -1;
237 }
238
239 /* connect error */
240 if (rc < 0) {
241 closesocket(s->fd);
242 s->fd = rc;
243 }
244
245 /* try to connect to the next address on the list */
1fc05adf
PB
246 if (s->current_addr) {
247 while (s->current_addr->ai_next != NULL && s->fd < 0) {
248 s->current_addr = s->current_addr->ai_next;
11663b55 249 s->fd = inet_connect_addr(s->current_addr, &in_progress, s, NULL);
1fc05adf
PB
250 /* connect in progress */
251 if (in_progress) {
252 return;
253 }
233aa5c2 254 }
1fc05adf
PB
255
256 freeaddrinfo(s->addr_list);
233aa5c2 257 }
05bc1d8a 258
233aa5c2
OW
259 if (s->callback) {
260 s->callback(s->fd, s->opaque);
05bc1d8a 261 }
233aa5c2 262 g_free(s);
233aa5c2
OW
263}
264
265static int inet_connect_addr(struct addrinfo *addr, bool *in_progress,
11663b55 266 ConnectState *connect_state, Error **errp)
233aa5c2
OW
267{
268 int sock, rc;
269
270 *in_progress = false;
05bc1d8a
MT
271
272 sock = qemu_socket(addr->ai_family, addr->ai_socktype, addr->ai_protocol);
273 if (sock < 0) {
11663b55 274 error_set_errno(errp, errno, QERR_SOCKET_CREATE_FAILED);
05bc1d8a
MT
275 return -1;
276 }
58455eb9 277 qemu_setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
233aa5c2 278 if (connect_state != NULL) {
05bc1d8a
MT
279 socket_set_nonblock(sock);
280 }
281 /* connect to peer */
282 do {
283 rc = 0;
284 if (connect(sock, addr->ai_addr, addr->ai_addrlen) < 0) {
285 rc = -socket_error();
286 }
287 } while (rc == -EINTR);
288
233aa5c2
OW
289 if (connect_state != NULL && QEMU_SOCKET_RC_INPROGRESS(rc)) {
290 connect_state->fd = sock;
291 qemu_set_fd_handler2(sock, NULL, NULL, wait_for_connect,
292 connect_state);
293 *in_progress = true;
05bc1d8a 294 } else if (rc < 0) {
11663b55 295 error_set_errno(errp, errno, QERR_SOCKET_CONNECT_FAILED);
05bc1d8a
MT
296 closesocket(sock);
297 return -1;
298 }
299 return sock;
300}
301
302static struct addrinfo *inet_parse_connect_opts(QemuOpts *opts, Error **errp)
303{
304 struct addrinfo ai, *res;
305 int rc;
f4c94c7c
GH
306 const char *addr;
307 const char *port;
d247d25f 308
05bc1d8a 309 memset(&ai, 0, sizeof(ai));
233aa5c2 310
d247d25f
AL
311 ai.ai_flags = AI_CANONNAME | AI_ADDRCONFIG;
312 ai.ai_family = PF_UNSPEC;
f4c94c7c 313 ai.ai_socktype = SOCK_STREAM;
d247d25f 314
f4c94c7c
GH
315 addr = qemu_opt_get(opts, "host");
316 port = qemu_opt_get(opts, "port");
317 if (addr == NULL || port == NULL) {
a12fb8ad 318 error_setg(errp, "host and/or port not specified");
05bc1d8a 319 return NULL;
d247d25f
AL
320 }
321
05bc1d8a 322 if (qemu_opt_get_bool(opts, "ipv4", 0)) {
d247d25f 323 ai.ai_family = PF_INET;
05bc1d8a
MT
324 }
325 if (qemu_opt_get_bool(opts, "ipv6", 0)) {
d247d25f 326 ai.ai_family = PF_INET6;
05bc1d8a 327 }
d247d25f
AL
328
329 /* lookup */
05bc1d8a
MT
330 rc = getaddrinfo(addr, port, &ai, &res);
331 if (rc != 0) {
a12fb8ad
PB
332 error_setg(errp, "address resolution failed for %s:%s: %s", addr, port,
333 gai_strerror(rc));
05bc1d8a
MT
334 return NULL;
335 }
336 return res;
337}
338
5db5f44c
OW
339/**
340 * Create a socket and connect it to an address.
341 *
342 * @opts: QEMU options, recognized parameters strings "host" and "port",
343 * bools "ipv4" and "ipv6".
5db5f44c 344 * @errp: set on error
233aa5c2
OW
345 * @callback: callback function for non-blocking connect
346 * @opaque: opaque for callback function
5db5f44c
OW
347 *
348 * Returns: -1 on error, file descriptor on success.
233aa5c2
OW
349 *
350 * If @callback is non-null, the connect is non-blocking. If this
351 * function succeeds, callback will be called when the connection
352 * completes, with the file descriptor on success, or -1 on error.
5db5f44c 353 */
233aa5c2
OW
354int inet_connect_opts(QemuOpts *opts, Error **errp,
355 NonBlockingConnectHandler *callback, void *opaque)
05bc1d8a
MT
356{
357 struct addrinfo *res, *e;
358 int sock = -1;
233aa5c2
OW
359 bool in_progress;
360 ConnectState *connect_state = NULL;
05bc1d8a
MT
361
362 res = inet_parse_connect_opts(opts, errp);
363 if (!res) {
364 return -1;
365 }
366
233aa5c2
OW
367 if (callback != NULL) {
368 connect_state = g_malloc0(sizeof(*connect_state));
369 connect_state->addr_list = res;
370 connect_state->callback = callback;
371 connect_state->opaque = opaque;
d247d25f 372 }
d247d25f
AL
373
374 for (e = res; e != NULL; e = e->ai_next) {
233aa5c2
OW
375 if (connect_state != NULL) {
376 connect_state->current_addr = e;
377 }
11663b55 378 sock = inet_connect_addr(e, &in_progress, connect_state, errp);
233aa5c2
OW
379 if (in_progress) {
380 return sock;
381 } else if (sock >= 0) {
382 /* non blocking socket immediate success, call callback */
383 if (callback != NULL) {
384 callback(sock, opaque);
385 }
05bc1d8a 386 break;
a6ba35b3 387 }
d247d25f 388 }
233aa5c2 389 g_free(connect_state);
d247d25f 390 freeaddrinfo(res);
05bc1d8a 391 return sock;
d247d25f
AL
392}
393
7fc4e63e 394int inet_dgram_opts(QemuOpts *opts, Error **errp)
7e1b35b4
GH
395{
396 struct addrinfo ai, *peer = NULL, *local = NULL;
397 const char *addr;
398 const char *port;
7e1b35b4
GH
399 int sock = -1, rc;
400
401 /* lookup peer addr */
402 memset(&ai,0, sizeof(ai));
403 ai.ai_flags = AI_CANONNAME | AI_ADDRCONFIG;
404 ai.ai_family = PF_UNSPEC;
405 ai.ai_socktype = SOCK_DGRAM;
406
407 addr = qemu_opt_get(opts, "host");
408 port = qemu_opt_get(opts, "port");
409 if (addr == NULL || strlen(addr) == 0) {
410 addr = "localhost";
411 }
412 if (port == NULL || strlen(port) == 0) {
4f085c82 413 error_setg(errp, "remote port not specified");
7e1b35b4
GH
414 return -1;
415 }
416
417 if (qemu_opt_get_bool(opts, "ipv4", 0))
418 ai.ai_family = PF_INET;
419 if (qemu_opt_get_bool(opts, "ipv6", 0))
420 ai.ai_family = PF_INET6;
421
422 if (0 != (rc = getaddrinfo(addr, port, &ai, &peer))) {
4f085c82
PB
423 error_setg(errp, "address resolution failed for %s:%s: %s", addr, port,
424 gai_strerror(rc));
7e1b35b4
GH
425 return -1;
426 }
7e1b35b4
GH
427
428 /* lookup local addr */
429 memset(&ai,0, sizeof(ai));
430 ai.ai_flags = AI_PASSIVE;
431 ai.ai_family = peer->ai_family;
432 ai.ai_socktype = SOCK_DGRAM;
433
434 addr = qemu_opt_get(opts, "localaddr");
435 port = qemu_opt_get(opts, "localport");
436 if (addr == NULL || strlen(addr) == 0) {
437 addr = NULL;
438 }
439 if (!port || strlen(port) == 0)
440 port = "0";
441
442 if (0 != (rc = getaddrinfo(addr, port, &ai, &local))) {
4f085c82
PB
443 error_setg(errp, "address resolution failed for %s:%s: %s", addr, port,
444 gai_strerror(rc));
39b38459 445 goto err;
7e1b35b4 446 }
7e1b35b4
GH
447
448 /* create socket */
40ff6d7e 449 sock = qemu_socket(peer->ai_family, peer->ai_socktype, peer->ai_protocol);
7e1b35b4 450 if (sock < 0) {
4f085c82 451 error_set_errno(errp, errno, QERR_SOCKET_CREATE_FAILED);
7e1b35b4
GH
452 goto err;
453 }
454 setsockopt(sock,SOL_SOCKET,SO_REUSEADDR,(void*)&on,sizeof(on));
455
456 /* bind socket */
7e1b35b4 457 if (bind(sock, local->ai_addr, local->ai_addrlen) < 0) {
4f085c82 458 error_set_errno(errp, errno, QERR_SOCKET_BIND_FAILED);
7e1b35b4
GH
459 goto err;
460 }
461
462 /* connect to peer */
7e1b35b4 463 if (connect(sock,peer->ai_addr,peer->ai_addrlen) < 0) {
4f085c82 464 error_set_errno(errp, errno, QERR_SOCKET_CONNECT_FAILED);
7e1b35b4
GH
465 goto err;
466 }
467
468 freeaddrinfo(local);
469 freeaddrinfo(peer);
470 return sock;
471
472err:
473 if (-1 != sock)
474 closesocket(sock);
475 if (local)
476 freeaddrinfo(local);
477 if (peer)
478 freeaddrinfo(peer);
479 return -1;
480}
481
f4c94c7c 482/* compatibility wrapper */
879e45c7 483static InetSocketAddress *inet_parse(const char *str, Error **errp)
f4c94c7c 484{
879e45c7 485 InetSocketAddress *addr;
f4c94c7c 486 const char *optstr, *h;
879e45c7 487 char host[64];
f4c94c7c 488 char port[33];
879e45c7 489 int to;
f4c94c7c
GH
490 int pos;
491
879e45c7
PB
492 addr = g_new0(InetSocketAddress, 1);
493
f4c94c7c
GH
494 /* parse address */
495 if (str[0] == ':') {
496 /* no host given */
879e45c7
PB
497 host[0] = '\0';
498 if (1 != sscanf(str, ":%32[^,]%n", port, &pos)) {
2f002c43 499 error_setg(errp, "error parsing port in address '%s'", str);
879e45c7 500 goto fail;
f4c94c7c
GH
501 }
502 } else if (str[0] == '[') {
503 /* IPv6 addr */
879e45c7 504 if (2 != sscanf(str, "[%64[^]]]:%32[^,]%n", host, port, &pos)) {
2f002c43 505 error_setg(errp, "error parsing IPv6 address '%s'", str);
879e45c7 506 goto fail;
f4c94c7c 507 }
879e45c7 508 addr->ipv6 = addr->has_ipv6 = true;
f4c94c7c
GH
509 } else if (qemu_isdigit(str[0])) {
510 /* IPv4 addr */
879e45c7 511 if (2 != sscanf(str, "%64[0-9.]:%32[^,]%n", host, port, &pos)) {
2f002c43 512 error_setg(errp, "error parsing IPv4 address '%s'", str);
879e45c7 513 goto fail;
f4c94c7c 514 }
879e45c7 515 addr->ipv4 = addr->has_ipv4 = true;
f4c94c7c
GH
516 } else {
517 /* hostname */
879e45c7 518 if (2 != sscanf(str, "%64[^:]:%32[^,]%n", host, port, &pos)) {
2f002c43 519 error_setg(errp, "error parsing address '%s'", str);
879e45c7 520 goto fail;
f4c94c7c
GH
521 }
522 }
879e45c7
PB
523
524 addr->host = g_strdup(host);
525 addr->port = g_strdup(port);
f4c94c7c
GH
526
527 /* parse options */
528 optstr = str + pos;
529 h = strstr(optstr, ",to=");
879e45c7
PB
530 if (h) {
531 if (1 != sscanf(str, "%d%n", &to, &pos) ||
532 (str[pos] != '\0' && str[pos] != ',')) {
533 error_setg(errp, "error parsing to= argument");
534 goto fail;
535 }
536 addr->has_to = true;
537 addr->to = to;
538 }
539 if (strstr(optstr, ",ipv4")) {
540 addr->ipv4 = addr->has_ipv4 = true;
541 }
542 if (strstr(optstr, ",ipv6")) {
543 addr->ipv6 = addr->has_ipv6 = true;
544 }
545 return addr;
546
547fail:
548 qapi_free_InetSocketAddress(addr);
549 return NULL;
550}
551
552static void inet_addr_to_opts(QemuOpts *opts, InetSocketAddress *addr)
553{
554 bool ipv4 = addr->ipv4 || !addr->has_ipv4;
555 bool ipv6 = addr->ipv6 || !addr->has_ipv6;
556
557 if (!ipv4 || !ipv6) {
558 qemu_opt_set_bool(opts, "ipv4", ipv4);
559 qemu_opt_set_bool(opts, "ipv6", ipv6);
560 }
561 if (addr->has_to) {
562 char to[20];
563 snprintf(to, sizeof(to), "%d", addr->to);
564 qemu_opt_set(opts, "to", to);
565 }
566 qemu_opt_set(opts, "host", addr->host);
567 qemu_opt_set(opts, "port", addr->port);
f4c94c7c
GH
568}
569
e5bc776f 570int inet_listen(const char *str, char *ostr, int olen,
029409e5 571 int socktype, int port_offset, Error **errp)
e5bc776f
GH
572{
573 QemuOpts *opts;
574 char *optstr;
575 int sock = -1;
879e45c7 576 InetSocketAddress *addr;
e5bc776f 577
879e45c7
PB
578 addr = inet_parse(str, errp);
579 if (addr != NULL) {
580 opts = qemu_opts_create(&dummy_opts, NULL, 0, NULL);
581 inet_addr_to_opts(opts, addr);
582 qapi_free_InetSocketAddress(addr);
029409e5 583 sock = inet_listen_opts(opts, port_offset, errp);
e5bc776f
GH
584 if (sock != -1 && ostr) {
585 optstr = strchr(str, ',');
586 if (qemu_opt_get_bool(opts, "ipv6", 0)) {
587 snprintf(ostr, olen, "[%s]:%s%s",
588 qemu_opt_get(opts, "host"),
589 qemu_opt_get(opts, "port"),
590 optstr ? optstr : "");
591 } else {
592 snprintf(ostr, olen, "%s:%s%s",
593 qemu_opt_get(opts, "host"),
594 qemu_opt_get(opts, "port"),
595 optstr ? optstr : "");
596 }
597 }
879e45c7 598 qemu_opts_del(opts);
e5bc776f 599 }
e5bc776f
GH
600 return sock;
601}
602
5db5f44c
OW
603/**
604 * Create a blocking socket and connect it to an address.
605 *
606 * @str: address string
607 * @errp: set in case of an error
608 *
609 * Returns -1 in case of error, file descriptor on success
610 **/
611int inet_connect(const char *str, Error **errp)
f4c94c7c
GH
612{
613 QemuOpts *opts;
614 int sock = -1;
879e45c7 615 InetSocketAddress *addr;
f4c94c7c 616
879e45c7
PB
617 addr = inet_parse(str, errp);
618 if (addr != NULL) {
619 opts = qemu_opts_create(&dummy_opts, NULL, 0, NULL);
620 inet_addr_to_opts(opts, addr);
621 qapi_free_InetSocketAddress(addr);
233aa5c2 622 sock = inet_connect_opts(opts, errp, NULL, NULL);
879e45c7 623 qemu_opts_del(opts);
5db5f44c 624 }
5db5f44c
OW
625 return sock;
626}
627
628/**
629 * Create a non-blocking socket and connect it to an address.
233aa5c2
OW
630 * Calls the callback function with fd in case of success or -1 in case of
631 * error.
5db5f44c
OW
632 *
633 * @str: address string
233aa5c2
OW
634 * @callback: callback function that is called when connect completes,
635 * cannot be NULL.
636 * @opaque: opaque for callback function
5db5f44c
OW
637 * @errp: set in case of an error
638 *
233aa5c2 639 * Returns: -1 on immediate error, file descriptor on success.
5db5f44c 640 **/
233aa5c2
OW
641int inet_nonblocking_connect(const char *str,
642 NonBlockingConnectHandler *callback,
643 void *opaque, Error **errp)
5db5f44c
OW
644{
645 QemuOpts *opts;
646 int sock = -1;
879e45c7 647 InetSocketAddress *addr;
5db5f44c 648
233aa5c2
OW
649 g_assert(callback != NULL);
650
879e45c7
PB
651 addr = inet_parse(str, errp);
652 if (addr != NULL) {
653 opts = qemu_opts_create(&dummy_opts, NULL, 0, NULL);
654 inet_addr_to_opts(opts, addr);
655 qapi_free_InetSocketAddress(addr);
233aa5c2 656 sock = inet_connect_opts(opts, errp, callback, opaque);
879e45c7 657 qemu_opts_del(opts);
a6ba35b3 658 }
f4c94c7c
GH
659 return sock;
660}
661
d247d25f
AL
662#ifndef _WIN32
663
7fc4e63e 664int unix_listen_opts(QemuOpts *opts, Error **errp)
d247d25f
AL
665{
666 struct sockaddr_un un;
62b6adfb
GH
667 const char *path = qemu_opt_get(opts, "path");
668 int sock, fd;
d247d25f 669
40ff6d7e 670 sock = qemu_socket(PF_UNIX, SOCK_STREAM, 0);
d247d25f 671 if (sock < 0) {
58899664 672 error_set_errno(errp, errno, QERR_SOCKET_CREATE_FAILED);
39b6efc8 673 return -1;
d247d25f
AL
674 }
675
d247d25f
AL
676 memset(&un, 0, sizeof(un));
677 un.sun_family = AF_UNIX;
678 if (path && strlen(path)) {
679 snprintf(un.sun_path, sizeof(un.sun_path), "%s", path);
680 } else {
681 char *tmpdir = getenv("TMPDIR");
682 snprintf(un.sun_path, sizeof(un.sun_path), "%s/qemu-socket-XXXXXX",
683 tmpdir ? tmpdir : "/tmp");
684 /*
685 * This dummy fd usage silences the mktemp() unsecure warning.
686 * Using mkstemp() doesn't make things more secure here
687 * though. bind() complains about existing files, so we have
688 * to unlink first and thus re-open the race window. The
689 * worst case possible is bind() failing, i.e. a DoS attack.
690 */
691 fd = mkstemp(un.sun_path); close(fd);
62b6adfb 692 qemu_opt_set(opts, "path", un.sun_path);
d247d25f 693 }
d247d25f
AL
694
695 unlink(un.sun_path);
696 if (bind(sock, (struct sockaddr*) &un, sizeof(un)) < 0) {
58899664 697 error_set_errno(errp, errno, QERR_SOCKET_BIND_FAILED);
d247d25f
AL
698 goto err;
699 }
700 if (listen(sock, 1) < 0) {
58899664 701 error_set_errno(errp, errno, QERR_SOCKET_LISTEN_FAILED);
d247d25f
AL
702 goto err;
703 }
704
d247d25f
AL
705 return sock;
706
707err:
d247d25f
AL
708 closesocket(sock);
709 return -1;
710}
711
1fc05adf
PB
712int unix_connect_opts(QemuOpts *opts, Error **errp,
713 NonBlockingConnectHandler *callback, void *opaque)
d247d25f
AL
714{
715 struct sockaddr_un un;
2af2bf67 716 const char *path = qemu_opt_get(opts, "path");
1fc05adf
PB
717 ConnectState *connect_state = NULL;
718 int sock, rc;
d247d25f 719
2af2bf67 720 if (NULL == path) {
58899664 721 error_setg(errp, "unix connect: no path specified\n");
2af2bf67
GH
722 return -1;
723 }
724
40ff6d7e 725 sock = qemu_socket(PF_UNIX, SOCK_STREAM, 0);
d247d25f 726 if (sock < 0) {
58899664 727 error_set_errno(errp, errno, QERR_SOCKET_CREATE_FAILED);
39b6efc8 728 return -1;
d247d25f 729 }
1fc05adf
PB
730 if (callback != NULL) {
731 connect_state = g_malloc0(sizeof(*connect_state));
732 connect_state->callback = callback;
733 connect_state->opaque = opaque;
734 socket_set_nonblock(sock);
735 }
d247d25f
AL
736
737 memset(&un, 0, sizeof(un));
738 un.sun_family = AF_UNIX;
739 snprintf(un.sun_path, sizeof(un.sun_path), "%s", path);
1fc05adf
PB
740
741 /* connect to peer */
742 do {
743 rc = 0;
744 if (connect(sock, (struct sockaddr *) &un, sizeof(un)) < 0) {
745 rc = -socket_error();
746 }
747 } while (rc == -EINTR);
748
749 if (connect_state != NULL && QEMU_SOCKET_RC_INPROGRESS(rc)) {
750 connect_state->fd = sock;
751 qemu_set_fd_handler2(sock, NULL, NULL, wait_for_connect,
752 connect_state);
753 return sock;
754 } else if (rc >= 0) {
755 /* non blocking socket immediate success, call callback */
756 if (callback != NULL) {
757 callback(sock, opaque);
758 }
759 }
760
761 if (rc < 0) {
58899664 762 error_set_errno(errp, -rc, QERR_SOCKET_CONNECT_FAILED);
9d947472 763 close(sock);
1fc05adf 764 sock = -1;
d247d25f
AL
765 }
766
1fc05adf 767 g_free(connect_state);
d247d25f
AL
768 return sock;
769}
770
0c814709
PB
771#else
772
773int unix_listen_opts(QemuOpts *opts, Error **errp)
774{
58899664 775 error_setg(errp, "unix sockets are not available on windows");
0c814709
PB
776 errno = ENOTSUP;
777 return -1;
778}
779
1fc05adf
PB
780int unix_connect_opts(QemuOpts *opts, Error **errp,
781 NonBlockingConnectHandler *callback, void *opaque)
0c814709 782{
58899664 783 error_setg(errp, "unix sockets are not available on windows");
0c814709
PB
784 errno = ENOTSUP;
785 return -1;
786}
787#endif
788
2af2bf67 789/* compatibility wrapper */
7fc4e63e 790int unix_listen(const char *str, char *ostr, int olen, Error **errp)
62b6adfb
GH
791{
792 QemuOpts *opts;
793 char *path, *optstr;
794 int sock, len;
795
8be7e7e4 796 opts = qemu_opts_create(&dummy_opts, NULL, 0, NULL);
62b6adfb
GH
797
798 optstr = strchr(str, ',');
799 if (optstr) {
800 len = optstr - str;
801 if (len) {
7267c094 802 path = g_malloc(len+1);
62b6adfb
GH
803 snprintf(path, len+1, "%.*s", len, str);
804 qemu_opt_set(opts, "path", path);
7267c094 805 g_free(path);
62b6adfb
GH
806 }
807 } else {
808 qemu_opt_set(opts, "path", str);
809 }
810
7fc4e63e 811 sock = unix_listen_opts(opts, errp);
62b6adfb
GH
812
813 if (sock != -1 && ostr)
814 snprintf(ostr, olen, "%s%s", qemu_opt_get(opts, "path"), optstr ? optstr : "");
815 qemu_opts_del(opts);
816 return sock;
817}
818
7fc4e63e 819int unix_connect(const char *path, Error **errp)
2af2bf67
GH
820{
821 QemuOpts *opts;
822 int sock;
823
8be7e7e4 824 opts = qemu_opts_create(&dummy_opts, NULL, 0, NULL);
2af2bf67 825 qemu_opt_set(opts, "path", path);
1fc05adf
PB
826 sock = unix_connect_opts(opts, errp, NULL, NULL);
827 qemu_opts_del(opts);
828 return sock;
829}
830
831
832int unix_nonblocking_connect(const char *path,
833 NonBlockingConnectHandler *callback,
834 void *opaque, Error **errp)
835{
836 QemuOpts *opts;
837 int sock = -1;
838
839 g_assert(callback != NULL);
840
841 opts = qemu_opts_create(&dummy_opts, NULL, 0, NULL);
842 qemu_opt_set(opts, "path", path);
843 sock = unix_connect_opts(opts, errp, callback, opaque);
2af2bf67
GH
844 qemu_opts_del(opts);
845 return sock;
846}
847
0706a4dc
PB
848#ifdef _WIN32
849static void socket_cleanup(void)
850{
851 WSACleanup();
852}
853#endif
854
855int socket_init(void)
856{
857#ifdef _WIN32
858 WSADATA Data;
859 int ret, err;
860
861 ret = WSAStartup(MAKEWORD(2,2), &Data);
862 if (ret != 0) {
863 err = WSAGetLastError();
864 fprintf(stderr, "WSAStartup: %d\n", err);
865 return -1;
866 }
867 atexit(socket_cleanup);
868#endif
869 return 0;
870}