]> git.proxmox.com Git - mirror_qemu.git/blame - util/qemu-sockets.c
Merge remote-tracking branch 'remotes/armbru/tags/pull-qapi-2016-03-18' into staging
[mirror_qemu.git] / util / 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 */
aafd7584 18#include "qemu/osdep.h"
d247d25f 19
83c9089e 20#include "monitor/monitor.h"
1de7afc9 21#include "qemu/sockets.h"
1de7afc9 22#include "qemu/main-loop.h"
2a8e21c7
DB
23#include "qapi/qmp-input-visitor.h"
24#include "qapi/qmp-output-visitor.h"
25#include "qapi-visit.h"
d247d25f
AL
26
27#ifndef AI_ADDRCONFIG
28# define AI_ADDRCONFIG 0
29#endif
3de3d698
WB
30#ifndef AI_V4MAPPED
31# define AI_V4MAPPED 0
32#endif
d247d25f 33
2af2bf67 34
d247d25f
AL
35static int inet_getport(struct addrinfo *e)
36{
37 struct sockaddr_in *i4;
38 struct sockaddr_in6 *i6;
39
40 switch (e->ai_family) {
41 case PF_INET6:
42 i6 = (void*)e->ai_addr;
43 return ntohs(i6->sin6_port);
44 case PF_INET:
45 i4 = (void*)e->ai_addr;
46 return ntohs(i4->sin_port);
47 default:
48 return 0;
49 }
50}
51
52static void inet_setport(struct addrinfo *e, int port)
53{
54 struct sockaddr_in *i4;
55 struct sockaddr_in6 *i6;
56
57 switch (e->ai_family) {
58 case PF_INET6:
59 i6 = (void*)e->ai_addr;
60 i6->sin6_port = htons(port);
61 break;
62 case PF_INET:
63 i4 = (void*)e->ai_addr;
64 i4->sin_port = htons(port);
65 break;
66 }
67}
68
a589569f
WX
69NetworkAddressFamily inet_netfamily(int family)
70{
71 switch (family) {
72 case PF_INET6: return NETWORK_ADDRESS_FAMILY_IPV6;
73 case PF_INET: return NETWORK_ADDRESS_FAMILY_IPV4;
74 case PF_UNIX: return NETWORK_ADDRESS_FAMILY_UNIX;
75 }
76 return NETWORK_ADDRESS_FAMILY_UNKNOWN;
77}
78
1856835d
DB
79/*
80 * Matrix we're trying to apply
81 *
82 * ipv4 ipv6 family
83 * - - PF_UNSPEC
84 * - f PF_INET
85 * - t PF_INET6
86 * f - PF_INET6
87 * f f <error>
88 * f t PF_INET6
89 * t - PF_INET
90 * t f PF_INET
91 * t t PF_INET6
92 *
93 * NB, this matrix is only about getting the neccessary results
94 * from getaddrinfo(). Some of the cases require further work
95 * after reading results from getaddrinfo in order to fully
96 * apply the logic the end user wants. eg with the last case
97 * ipv4=t + ipv6=t + PF_INET6, getaddrinfo alone can only
98 * guarantee the ipv6=t part of the request - we need more
99 * checks to provide ipv4=t part of the guarantee. This is
100 * outside scope of this method and not currently handled by
101 * callers at all.
102 */
103static int inet_ai_family_from_address(InetSocketAddress *addr,
104 Error **errp)
105{
106 if (addr->has_ipv6 && addr->has_ipv4 &&
107 !addr->ipv6 && !addr->ipv4) {
108 error_setg(errp, "Cannot disable IPv4 and IPv6 at same time");
109 return PF_UNSPEC;
110 }
111 if ((addr->has_ipv6 && addr->ipv6) || (addr->has_ipv4 && !addr->ipv4)) {
112 return PF_INET6;
113 }
114 if ((addr->has_ipv4 && addr->ipv4) || (addr->has_ipv6 && !addr->ipv6)) {
115 return PF_INET;
116 }
117 return PF_UNSPEC;
118}
119
120static int inet_listen_saddr(InetSocketAddress *saddr,
121 int port_offset,
122 bool update_addr,
123 Error **errp)
d247d25f
AL
124{
125 struct addrinfo ai,*res,*e;
d247d25f
AL
126 char port[33];
127 char uaddr[INET6_ADDRSTRLEN+1];
128 char uport[33];
1856835d
DB
129 int slisten, rc, port_min, port_max, p;
130 Error *err = NULL;
d247d25f
AL
131
132 memset(&ai,0, sizeof(ai));
3de3d698 133 ai.ai_flags = AI_PASSIVE;
1856835d 134 ai.ai_family = inet_ai_family_from_address(saddr, &err);
e5bc776f 135 ai.ai_socktype = SOCK_STREAM;
d247d25f 136
1856835d
DB
137 if (err) {
138 error_propagate(errp, err);
139 return -1;
140 }
141
142 if (saddr->host == NULL) {
0983f5e6 143 error_setg(errp, "host not specified");
e5bc776f 144 return -1;
d247d25f 145 }
1856835d
DB
146 if (saddr->port != NULL) {
147 pstrcpy(port, sizeof(port), saddr->port);
0983f5e6
DB
148 } else {
149 port[0] = '\0';
150 }
d247d25f
AL
151
152 /* lookup */
8bc89127
GH
153 if (port_offset) {
154 unsigned long long baseport;
0983f5e6
DB
155 if (strlen(port) == 0) {
156 error_setg(errp, "port not specified");
157 return -1;
158 }
8bc89127
GH
159 if (parse_uint_full(port, &baseport, 10) < 0) {
160 error_setg(errp, "can't convert to a number: %s", port);
161 return -1;
162 }
163 if (baseport > 65535 ||
164 baseport + port_offset > 65535) {
165 error_setg(errp, "port %s out of range", port);
166 return -1;
167 }
168 snprintf(port, sizeof(port), "%d", (int)baseport + port_offset);
169 }
1856835d 170 rc = getaddrinfo(strlen(saddr->host) ? saddr->host : NULL,
0983f5e6 171 strlen(port) ? port : NULL, &ai, &res);
d247d25f 172 if (rc != 0) {
1856835d
DB
173 error_setg(errp, "address resolution failed for %s:%s: %s",
174 saddr->host, port, gai_strerror(rc));
d247d25f
AL
175 return -1;
176 }
d247d25f
AL
177
178 /* create socket + bind */
179 for (e = res; e != NULL; e = e->ai_next) {
39b6efc8
VS
180 getnameinfo((struct sockaddr*)e->ai_addr,e->ai_addrlen,
181 uaddr,INET6_ADDRSTRLEN,uport,32,
182 NI_NUMERICHOST | NI_NUMERICSERV);
40ff6d7e 183 slisten = qemu_socket(e->ai_family, e->ai_socktype, e->ai_protocol);
39b6efc8 184 if (slisten < 0) {
029409e5 185 if (!e->ai_next) {
235256a2 186 error_setg_errno(errp, errno, "Failed to create socket");
029409e5 187 }
39b6efc8
VS
188 continue;
189 }
d247d25f 190
04fd1c78 191 socket_set_fast_reuse(slisten);
d247d25f
AL
192#ifdef IPV6_V6ONLY
193 if (e->ai_family == PF_INET6) {
194 /* listen on both ipv4 and ipv6 */
f9b5426f 195 const int off = 0;
9957fc7f
SW
196 qemu_setsockopt(slisten, IPPROTO_IPV6, IPV6_V6ONLY, &off,
197 sizeof(off));
d247d25f
AL
198 }
199#endif
200
877691f9 201 port_min = inet_getport(e);
1856835d 202 port_max = saddr->has_to ? saddr->to + port_offset : port_min;
877691f9
MA
203 for (p = port_min; p <= port_max; p++) {
204 inet_setport(e, p);
d247d25f 205 if (bind(slisten, e->ai_addr, e->ai_addrlen) == 0) {
d247d25f
AL
206 goto listen;
207 }
877691f9 208 if (p == port_max) {
029409e5 209 if (!e->ai_next) {
235256a2 210 error_setg_errno(errp, errno, "Failed to bind socket");
029409e5 211 }
d247d25f 212 }
d247d25f
AL
213 }
214 closesocket(slisten);
215 }
d247d25f
AL
216 freeaddrinfo(res);
217 return -1;
218
219listen:
220 if (listen(slisten,1) != 0) {
235256a2 221 error_setg_errno(errp, errno, "Failed to listen on socket");
d247d25f 222 closesocket(slisten);
39b6efc8 223 freeaddrinfo(res);
d247d25f
AL
224 return -1;
225 }
1856835d
DB
226 if (update_addr) {
227 g_free(saddr->host);
228 saddr->host = g_strdup(uaddr);
229 g_free(saddr->port);
230 saddr->port = g_strdup_printf("%d",
231 inet_getport(e) - port_offset);
232 saddr->has_ipv6 = saddr->ipv6 = e->ai_family == PF_INET6;
233 saddr->has_ipv4 = saddr->ipv4 = e->ai_family != PF_INET6;
234 }
d247d25f
AL
235 freeaddrinfo(res);
236 return slisten;
237}
238
05bc1d8a
MT
239#ifdef _WIN32
240#define QEMU_SOCKET_RC_INPROGRESS(rc) \
241 ((rc) == -EINPROGRESS || (rc) == -EWOULDBLOCK || (rc) == -WSAEALREADY)
242#else
243#define QEMU_SOCKET_RC_INPROGRESS(rc) \
244 ((rc) == -EINPROGRESS)
245#endif
246
233aa5c2
OW
247/* Struct to store connect state for non blocking connect */
248typedef struct ConnectState {
249 int fd;
250 struct addrinfo *addr_list;
251 struct addrinfo *current_addr;
252 NonBlockingConnectHandler *callback;
253 void *opaque;
254} ConnectState;
255
256static int inet_connect_addr(struct addrinfo *addr, bool *in_progress,
11663b55 257 ConnectState *connect_state, Error **errp);
233aa5c2
OW
258
259static void wait_for_connect(void *opaque)
d247d25f 260{
233aa5c2
OW
261 ConnectState *s = opaque;
262 int val = 0, rc = 0;
263 socklen_t valsize = sizeof(val);
264 bool in_progress;
51795029 265 Error *err = NULL;
233aa5c2 266
82e1cc4b 267 qemu_set_fd_handler(s->fd, NULL, NULL, NULL);
233aa5c2
OW
268
269 do {
9957fc7f 270 rc = qemu_getsockopt(s->fd, SOL_SOCKET, SO_ERROR, &val, &valsize);
b16a44e1 271 } while (rc == -1 && errno == EINTR);
233aa5c2
OW
272
273 /* update rc to contain error */
274 if (!rc && val) {
275 rc = -1;
51795029 276 errno = val;
233aa5c2
OW
277 }
278
279 /* connect error */
280 if (rc < 0) {
51795029 281 error_setg_errno(&err, errno, "Error connecting to socket");
233aa5c2
OW
282 closesocket(s->fd);
283 s->fd = rc;
284 }
285
286 /* try to connect to the next address on the list */
1fc05adf
PB
287 if (s->current_addr) {
288 while (s->current_addr->ai_next != NULL && s->fd < 0) {
289 s->current_addr = s->current_addr->ai_next;
11663b55 290 s->fd = inet_connect_addr(s->current_addr, &in_progress, s, NULL);
51795029
CM
291 if (s->fd < 0) {
292 error_free(err);
293 err = NULL;
294 error_setg_errno(&err, errno, "Unable to start socket connect");
295 }
1fc05adf
PB
296 /* connect in progress */
297 if (in_progress) {
51795029 298 goto out;
1fc05adf 299 }
233aa5c2 300 }
1fc05adf
PB
301
302 freeaddrinfo(s->addr_list);
233aa5c2 303 }
05bc1d8a 304
233aa5c2 305 if (s->callback) {
51795029 306 s->callback(s->fd, err, s->opaque);
05bc1d8a 307 }
233aa5c2 308 g_free(s);
51795029
CM
309out:
310 error_free(err);
233aa5c2
OW
311}
312
313static int inet_connect_addr(struct addrinfo *addr, bool *in_progress,
11663b55 314 ConnectState *connect_state, Error **errp)
233aa5c2
OW
315{
316 int sock, rc;
317
318 *in_progress = false;
05bc1d8a
MT
319
320 sock = qemu_socket(addr->ai_family, addr->ai_socktype, addr->ai_protocol);
321 if (sock < 0) {
235256a2 322 error_setg_errno(errp, errno, "Failed to create socket");
05bc1d8a
MT
323 return -1;
324 }
04fd1c78 325 socket_set_fast_reuse(sock);
233aa5c2 326 if (connect_state != NULL) {
f9e8cacc 327 qemu_set_nonblock(sock);
05bc1d8a
MT
328 }
329 /* connect to peer */
330 do {
331 rc = 0;
332 if (connect(sock, addr->ai_addr, addr->ai_addrlen) < 0) {
b16a44e1 333 rc = -errno;
05bc1d8a
MT
334 }
335 } while (rc == -EINTR);
336
233aa5c2
OW
337 if (connect_state != NULL && QEMU_SOCKET_RC_INPROGRESS(rc)) {
338 connect_state->fd = sock;
82e1cc4b 339 qemu_set_fd_handler(sock, NULL, wait_for_connect, connect_state);
233aa5c2 340 *in_progress = true;
05bc1d8a 341 } else if (rc < 0) {
235256a2 342 error_setg_errno(errp, errno, "Failed to connect socket");
05bc1d8a
MT
343 closesocket(sock);
344 return -1;
345 }
346 return sock;
347}
348
2942e420
DB
349static struct addrinfo *inet_parse_connect_saddr(InetSocketAddress *saddr,
350 Error **errp)
05bc1d8a
MT
351{
352 struct addrinfo ai, *res;
353 int rc;
2942e420 354 Error *err = NULL;
d247d25f 355
05bc1d8a 356 memset(&ai, 0, sizeof(ai));
233aa5c2 357
3de3d698 358 ai.ai_flags = AI_CANONNAME | AI_V4MAPPED | AI_ADDRCONFIG;
2942e420 359 ai.ai_family = inet_ai_family_from_address(saddr, &err);
f4c94c7c 360 ai.ai_socktype = SOCK_STREAM;
d247d25f 361
2942e420
DB
362 if (err) {
363 error_propagate(errp, err);
05bc1d8a 364 return NULL;
d247d25f
AL
365 }
366
2942e420
DB
367 if (saddr->host == NULL || saddr->port == NULL) {
368 error_setg(errp, "host and/or port not specified");
369 return NULL;
05bc1d8a 370 }
d247d25f
AL
371
372 /* lookup */
2942e420 373 rc = getaddrinfo(saddr->host, saddr->port, &ai, &res);
05bc1d8a 374 if (rc != 0) {
2942e420
DB
375 error_setg(errp, "address resolution failed for %s:%s: %s",
376 saddr->host, saddr->port, gai_strerror(rc));
05bc1d8a
MT
377 return NULL;
378 }
379 return res;
380}
381
5db5f44c
OW
382/**
383 * Create a socket and connect it to an address.
384 *
2942e420 385 * @saddr: Inet socket address specification
5db5f44c 386 * @errp: set on error
233aa5c2
OW
387 * @callback: callback function for non-blocking connect
388 * @opaque: opaque for callback function
5db5f44c
OW
389 *
390 * Returns: -1 on error, file descriptor on success.
233aa5c2
OW
391 *
392 * If @callback is non-null, the connect is non-blocking. If this
393 * function succeeds, callback will be called when the connection
394 * completes, with the file descriptor on success, or -1 on error.
5db5f44c 395 */
2942e420
DB
396static int inet_connect_saddr(InetSocketAddress *saddr, Error **errp,
397 NonBlockingConnectHandler *callback, void *opaque)
05bc1d8a 398{
3f9286b7 399 Error *local_err = NULL;
05bc1d8a
MT
400 struct addrinfo *res, *e;
401 int sock = -1;
233aa5c2
OW
402 bool in_progress;
403 ConnectState *connect_state = NULL;
05bc1d8a 404
2942e420 405 res = inet_parse_connect_saddr(saddr, errp);
05bc1d8a
MT
406 if (!res) {
407 return -1;
408 }
409
233aa5c2
OW
410 if (callback != NULL) {
411 connect_state = g_malloc0(sizeof(*connect_state));
412 connect_state->addr_list = res;
413 connect_state->callback = callback;
414 connect_state->opaque = opaque;
d247d25f 415 }
d247d25f
AL
416
417 for (e = res; e != NULL; e = e->ai_next) {
3f9286b7
MA
418 error_free(local_err);
419 local_err = NULL;
233aa5c2
OW
420 if (connect_state != NULL) {
421 connect_state->current_addr = e;
422 }
3f9286b7
MA
423 sock = inet_connect_addr(e, &in_progress, connect_state, &local_err);
424 if (sock >= 0) {
05bc1d8a 425 break;
a6ba35b3 426 }
d247d25f 427 }
3f9286b7
MA
428
429 if (sock < 0) {
430 error_propagate(errp, local_err);
431 } else if (in_progress) {
432 /* wait_for_connect() will do the rest */
433 return sock;
434 } else {
435 if (callback) {
51795029 436 callback(sock, NULL, opaque);
3f9286b7
MA
437 }
438 }
233aa5c2 439 g_free(connect_state);
d247d25f 440 freeaddrinfo(res);
05bc1d8a 441 return sock;
d247d25f
AL
442}
443
8b39910e
DB
444static int inet_dgram_saddr(InetSocketAddress *sraddr,
445 InetSocketAddress *sladdr,
446 Error **errp)
7e1b35b4
GH
447{
448 struct addrinfo ai, *peer = NULL, *local = NULL;
449 const char *addr;
450 const char *port;
7e1b35b4 451 int sock = -1, rc;
8b39910e 452 Error *err = NULL;
7e1b35b4
GH
453
454 /* lookup peer addr */
455 memset(&ai,0, sizeof(ai));
3de3d698 456 ai.ai_flags = AI_CANONNAME | AI_V4MAPPED | AI_ADDRCONFIG;
8b39910e 457 ai.ai_family = inet_ai_family_from_address(sraddr, &err);
7e1b35b4
GH
458 ai.ai_socktype = SOCK_DGRAM;
459
8b39910e
DB
460 if (err) {
461 error_propagate(errp, err);
58c652c0 462 goto err;
8b39910e
DB
463 }
464
465 addr = sraddr->host;
466 port = sraddr->port;
7e1b35b4
GH
467 if (addr == NULL || strlen(addr) == 0) {
468 addr = "localhost";
469 }
470 if (port == NULL || strlen(port) == 0) {
4f085c82 471 error_setg(errp, "remote port not specified");
58c652c0 472 goto err;
7e1b35b4
GH
473 }
474
7e1b35b4 475 if (0 != (rc = getaddrinfo(addr, port, &ai, &peer))) {
4f085c82
PB
476 error_setg(errp, "address resolution failed for %s:%s: %s", addr, port,
477 gai_strerror(rc));
58c652c0 478 goto err;
7e1b35b4 479 }
7e1b35b4
GH
480
481 /* lookup local addr */
482 memset(&ai,0, sizeof(ai));
483 ai.ai_flags = AI_PASSIVE;
484 ai.ai_family = peer->ai_family;
485 ai.ai_socktype = SOCK_DGRAM;
486
8b39910e
DB
487 if (sladdr) {
488 addr = sladdr->host;
489 port = sladdr->port;
490 if (addr == NULL || strlen(addr) == 0) {
491 addr = NULL;
492 }
493 if (!port || strlen(port) == 0) {
494 port = "0";
495 }
496 } else {
7e1b35b4 497 addr = NULL;
7e1b35b4 498 port = "0";
8b39910e 499 }
7e1b35b4
GH
500
501 if (0 != (rc = getaddrinfo(addr, port, &ai, &local))) {
4f085c82
PB
502 error_setg(errp, "address resolution failed for %s:%s: %s", addr, port,
503 gai_strerror(rc));
39b38459 504 goto err;
7e1b35b4 505 }
7e1b35b4
GH
506
507 /* create socket */
40ff6d7e 508 sock = qemu_socket(peer->ai_family, peer->ai_socktype, peer->ai_protocol);
7e1b35b4 509 if (sock < 0) {
235256a2 510 error_setg_errno(errp, errno, "Failed to create socket");
7e1b35b4
GH
511 goto err;
512 }
04fd1c78 513 socket_set_fast_reuse(sock);
7e1b35b4
GH
514
515 /* bind socket */
7e1b35b4 516 if (bind(sock, local->ai_addr, local->ai_addrlen) < 0) {
235256a2 517 error_setg_errno(errp, errno, "Failed to bind socket");
7e1b35b4
GH
518 goto err;
519 }
520
521 /* connect to peer */
7e1b35b4 522 if (connect(sock,peer->ai_addr,peer->ai_addrlen) < 0) {
235256a2 523 error_setg_errno(errp, errno, "Failed to connect socket");
7e1b35b4
GH
524 goto err;
525 }
526
527 freeaddrinfo(local);
528 freeaddrinfo(peer);
529 return sock;
530
531err:
532 if (-1 != sock)
533 closesocket(sock);
534 if (local)
535 freeaddrinfo(local);
536 if (peer)
537 freeaddrinfo(peer);
538 return -1;
539}
540
f4c94c7c 541/* compatibility wrapper */
f17c90be 542InetSocketAddress *inet_parse(const char *str, Error **errp)
f4c94c7c 543{
879e45c7 544 InetSocketAddress *addr;
f4c94c7c 545 const char *optstr, *h;
55a10996 546 char host[65];
f4c94c7c 547 char port[33];
879e45c7 548 int to;
f4c94c7c
GH
549 int pos;
550
879e45c7
PB
551 addr = g_new0(InetSocketAddress, 1);
552
f4c94c7c
GH
553 /* parse address */
554 if (str[0] == ':') {
555 /* no host given */
879e45c7
PB
556 host[0] = '\0';
557 if (1 != sscanf(str, ":%32[^,]%n", port, &pos)) {
2f002c43 558 error_setg(errp, "error parsing port in address '%s'", str);
879e45c7 559 goto fail;
f4c94c7c
GH
560 }
561 } else if (str[0] == '[') {
562 /* IPv6 addr */
879e45c7 563 if (2 != sscanf(str, "[%64[^]]]:%32[^,]%n", host, port, &pos)) {
2f002c43 564 error_setg(errp, "error parsing IPv6 address '%s'", str);
879e45c7 565 goto fail;
f4c94c7c 566 }
879e45c7 567 addr->ipv6 = addr->has_ipv6 = true;
f4c94c7c 568 } else {
391b7b97 569 /* hostname or IPv4 addr */
879e45c7 570 if (2 != sscanf(str, "%64[^:]:%32[^,]%n", host, port, &pos)) {
2f002c43 571 error_setg(errp, "error parsing address '%s'", str);
879e45c7 572 goto fail;
f4c94c7c 573 }
391b7b97
JT
574 if (host[strspn(host, "0123456789.")] == '\0') {
575 addr->ipv4 = addr->has_ipv4 = true;
576 }
f4c94c7c 577 }
879e45c7
PB
578
579 addr->host = g_strdup(host);
580 addr->port = g_strdup(port);
f4c94c7c
GH
581
582 /* parse options */
583 optstr = str + pos;
584 h = strstr(optstr, ",to=");
879e45c7 585 if (h) {
1ccbc285
AP
586 h += 4;
587 if (sscanf(h, "%d%n", &to, &pos) != 1 ||
588 (h[pos] != '\0' && h[pos] != ',')) {
879e45c7
PB
589 error_setg(errp, "error parsing to= argument");
590 goto fail;
591 }
592 addr->has_to = true;
593 addr->to = to;
594 }
595 if (strstr(optstr, ",ipv4")) {
596 addr->ipv4 = addr->has_ipv4 = true;
597 }
598 if (strstr(optstr, ",ipv6")) {
599 addr->ipv6 = addr->has_ipv6 = true;
600 }
601 return addr;
602
603fail:
604 qapi_free_InetSocketAddress(addr);
605 return NULL;
606}
607
e5bc776f 608int inet_listen(const char *str, char *ostr, int olen,
029409e5 609 int socktype, int port_offset, Error **errp)
e5bc776f 610{
e5bc776f
GH
611 char *optstr;
612 int sock = -1;
879e45c7 613 InetSocketAddress *addr;
e5bc776f 614
879e45c7
PB
615 addr = inet_parse(str, errp);
616 if (addr != NULL) {
1856835d 617 sock = inet_listen_saddr(addr, port_offset, true, errp);
e5bc776f
GH
618 if (sock != -1 && ostr) {
619 optstr = strchr(str, ',');
1856835d 620 if (addr->ipv6) {
e5bc776f 621 snprintf(ostr, olen, "[%s]:%s%s",
1856835d
DB
622 addr->host,
623 addr->port,
e5bc776f
GH
624 optstr ? optstr : "");
625 } else {
626 snprintf(ostr, olen, "%s:%s%s",
1856835d
DB
627 addr->host,
628 addr->port,
e5bc776f
GH
629 optstr ? optstr : "");
630 }
631 }
1856835d 632 qapi_free_InetSocketAddress(addr);
e5bc776f 633 }
e5bc776f
GH
634 return sock;
635}
636
5db5f44c
OW
637/**
638 * Create a blocking socket and connect it to an address.
639 *
640 * @str: address string
641 * @errp: set in case of an error
642 *
643 * Returns -1 in case of error, file descriptor on success
644 **/
645int inet_connect(const char *str, Error **errp)
f4c94c7c 646{
f4c94c7c 647 int sock = -1;
879e45c7 648 InetSocketAddress *addr;
f4c94c7c 649
879e45c7
PB
650 addr = inet_parse(str, errp);
651 if (addr != NULL) {
2942e420 652 sock = inet_connect_saddr(addr, errp, NULL, NULL);
879e45c7 653 qapi_free_InetSocketAddress(addr);
5db5f44c 654 }
5db5f44c
OW
655 return sock;
656}
657
658/**
659 * Create a non-blocking socket and connect it to an address.
233aa5c2
OW
660 * Calls the callback function with fd in case of success or -1 in case of
661 * error.
5db5f44c
OW
662 *
663 * @str: address string
233aa5c2
OW
664 * @callback: callback function that is called when connect completes,
665 * cannot be NULL.
666 * @opaque: opaque for callback function
5db5f44c
OW
667 * @errp: set in case of an error
668 *
233aa5c2 669 * Returns: -1 on immediate error, file descriptor on success.
5db5f44c 670 **/
233aa5c2
OW
671int inet_nonblocking_connect(const char *str,
672 NonBlockingConnectHandler *callback,
673 void *opaque, Error **errp)
5db5f44c 674{
5db5f44c 675 int sock = -1;
879e45c7 676 InetSocketAddress *addr;
5db5f44c 677
233aa5c2
OW
678 g_assert(callback != NULL);
679
879e45c7
PB
680 addr = inet_parse(str, errp);
681 if (addr != NULL) {
2942e420 682 sock = inet_connect_saddr(addr, errp, callback, opaque);
879e45c7 683 qapi_free_InetSocketAddress(addr);
a6ba35b3 684 }
f4c94c7c
GH
685 return sock;
686}
687
d247d25f
AL
688#ifndef _WIN32
689
1856835d
DB
690static int unix_listen_saddr(UnixSocketAddress *saddr,
691 bool update_addr,
692 Error **errp)
d247d25f
AL
693{
694 struct sockaddr_un un;
62b6adfb 695 int sock, fd;
d247d25f 696
40ff6d7e 697 sock = qemu_socket(PF_UNIX, SOCK_STREAM, 0);
d247d25f 698 if (sock < 0) {
b658c53d 699 error_setg_errno(errp, errno, "Failed to create Unix socket");
39b6efc8 700 return -1;
d247d25f
AL
701 }
702
d247d25f
AL
703 memset(&un, 0, sizeof(un));
704 un.sun_family = AF_UNIX;
1856835d
DB
705 if (saddr->path && strlen(saddr->path)) {
706 snprintf(un.sun_path, sizeof(un.sun_path), "%s", saddr->path);
d247d25f 707 } else {
b658c53d
PB
708 const char *tmpdir = getenv("TMPDIR");
709 tmpdir = tmpdir ? tmpdir : "/tmp";
710 if (snprintf(un.sun_path, sizeof(un.sun_path), "%s/qemu-socket-XXXXXX",
711 tmpdir) >= sizeof(un.sun_path)) {
712 error_setg_errno(errp, errno,
713 "TMPDIR environment variable (%s) too large", tmpdir);
714 goto err;
715 }
716
d247d25f
AL
717 /*
718 * This dummy fd usage silences the mktemp() unsecure warning.
719 * Using mkstemp() doesn't make things more secure here
720 * though. bind() complains about existing files, so we have
721 * to unlink first and thus re-open the race window. The
722 * worst case possible is bind() failing, i.e. a DoS attack.
723 */
b658c53d
PB
724 fd = mkstemp(un.sun_path);
725 if (fd < 0) {
726 error_setg_errno(errp, errno,
727 "Failed to make a temporary socket name in %s", tmpdir);
728 goto err;
729 }
730 close(fd);
1856835d
DB
731 if (update_addr) {
732 g_free(saddr->path);
733 saddr->path = g_strdup(un.sun_path);
734 }
d247d25f 735 }
d247d25f 736
a2f31f18 737 if (unlink(un.sun_path) < 0 && errno != ENOENT) {
0ef705a2
CR
738 error_setg_errno(errp, errno,
739 "Failed to unlink socket %s", un.sun_path);
740 goto err;
741 }
d247d25f 742 if (bind(sock, (struct sockaddr*) &un, sizeof(un)) < 0) {
b658c53d 743 error_setg_errno(errp, errno, "Failed to bind socket to %s", un.sun_path);
d247d25f
AL
744 goto err;
745 }
746 if (listen(sock, 1) < 0) {
235256a2 747 error_setg_errno(errp, errno, "Failed to listen on socket");
d247d25f
AL
748 goto err;
749 }
750
d247d25f
AL
751 return sock;
752
753err:
d247d25f
AL
754 closesocket(sock);
755 return -1;
756}
757
2942e420
DB
758static int unix_connect_saddr(UnixSocketAddress *saddr, Error **errp,
759 NonBlockingConnectHandler *callback, void *opaque)
d247d25f
AL
760{
761 struct sockaddr_un un;
1fc05adf
PB
762 ConnectState *connect_state = NULL;
763 int sock, rc;
d247d25f 764
2942e420 765 if (saddr->path == NULL) {
312fd5f2 766 error_setg(errp, "unix connect: no path specified");
2af2bf67
GH
767 return -1;
768 }
769
40ff6d7e 770 sock = qemu_socket(PF_UNIX, SOCK_STREAM, 0);
d247d25f 771 if (sock < 0) {
235256a2 772 error_setg_errno(errp, errno, "Failed to create socket");
39b6efc8 773 return -1;
d247d25f 774 }
1fc05adf
PB
775 if (callback != NULL) {
776 connect_state = g_malloc0(sizeof(*connect_state));
777 connect_state->callback = callback;
778 connect_state->opaque = opaque;
f9e8cacc 779 qemu_set_nonblock(sock);
1fc05adf 780 }
d247d25f
AL
781
782 memset(&un, 0, sizeof(un));
783 un.sun_family = AF_UNIX;
2942e420 784 snprintf(un.sun_path, sizeof(un.sun_path), "%s", saddr->path);
1fc05adf
PB
785
786 /* connect to peer */
787 do {
788 rc = 0;
789 if (connect(sock, (struct sockaddr *) &un, sizeof(un)) < 0) {
b16a44e1 790 rc = -errno;
1fc05adf
PB
791 }
792 } while (rc == -EINTR);
793
794 if (connect_state != NULL && QEMU_SOCKET_RC_INPROGRESS(rc)) {
795 connect_state->fd = sock;
82e1cc4b 796 qemu_set_fd_handler(sock, NULL, wait_for_connect, connect_state);
1fc05adf
PB
797 return sock;
798 } else if (rc >= 0) {
799 /* non blocking socket immediate success, call callback */
800 if (callback != NULL) {
51795029 801 callback(sock, NULL, opaque);
1fc05adf
PB
802 }
803 }
804
805 if (rc < 0) {
235256a2 806 error_setg_errno(errp, -rc, "Failed to connect socket");
9d947472 807 close(sock);
1fc05adf 808 sock = -1;
d247d25f
AL
809 }
810
1fc05adf 811 g_free(connect_state);
d247d25f
AL
812 return sock;
813}
814
0c814709
PB
815#else
816
1856835d
DB
817static int unix_listen_saddr(UnixSocketAddress *saddr,
818 bool update_addr,
819 Error **errp)
0c814709 820{
58899664 821 error_setg(errp, "unix sockets are not available on windows");
0c814709
PB
822 errno = ENOTSUP;
823 return -1;
824}
825
2942e420
DB
826static int unix_connect_saddr(UnixSocketAddress *saddr, Error **errp,
827 NonBlockingConnectHandler *callback, void *opaque)
0c814709 828{
58899664 829 error_setg(errp, "unix sockets are not available on windows");
0c814709
PB
830 errno = ENOTSUP;
831 return -1;
832}
833#endif
834
2af2bf67 835/* compatibility wrapper */
7fc4e63e 836int unix_listen(const char *str, char *ostr, int olen, Error **errp)
62b6adfb 837{
62b6adfb
GH
838 char *path, *optstr;
839 int sock, len;
1856835d 840 UnixSocketAddress *saddr;
62b6adfb 841
1856835d 842 saddr = g_new0(UnixSocketAddress, 1);
62b6adfb
GH
843
844 optstr = strchr(str, ',');
845 if (optstr) {
846 len = optstr - str;
847 if (len) {
7267c094 848 path = g_malloc(len+1);
62b6adfb 849 snprintf(path, len+1, "%.*s", len, str);
1856835d 850 saddr->path = path;
62b6adfb
GH
851 }
852 } else {
1856835d 853 saddr->path = g_strdup(str);
62b6adfb
GH
854 }
855
1856835d 856 sock = unix_listen_saddr(saddr, true, errp);
62b6adfb
GH
857
858 if (sock != -1 && ostr)
1856835d
DB
859 snprintf(ostr, olen, "%s%s", saddr->path, optstr ? optstr : "");
860 qapi_free_UnixSocketAddress(saddr);
62b6adfb
GH
861 return sock;
862}
863
7fc4e63e 864int unix_connect(const char *path, Error **errp)
2af2bf67 865{
2942e420 866 UnixSocketAddress *saddr;
2af2bf67
GH
867 int sock;
868
2942e420
DB
869 saddr = g_new0(UnixSocketAddress, 1);
870 saddr->path = g_strdup(path);
871 sock = unix_connect_saddr(saddr, errp, NULL, NULL);
872 qapi_free_UnixSocketAddress(saddr);
1fc05adf
PB
873 return sock;
874}
875
876
877int unix_nonblocking_connect(const char *path,
878 NonBlockingConnectHandler *callback,
879 void *opaque, Error **errp)
880{
2942e420 881 UnixSocketAddress *saddr;
1fc05adf
PB
882 int sock = -1;
883
884 g_assert(callback != NULL);
885
2942e420
DB
886 saddr = g_new0(UnixSocketAddress, 1);
887 saddr->path = g_strdup(path);
888 sock = unix_connect_saddr(saddr, errp, callback, opaque);
889 qapi_free_UnixSocketAddress(saddr);
2af2bf67
GH
890 return sock;
891}
892
101f9cbc
PB
893SocketAddress *socket_parse(const char *str, Error **errp)
894{
afde3f8b 895 SocketAddress *addr;
101f9cbc 896
afde3f8b 897 addr = g_new0(SocketAddress, 1);
101f9cbc
PB
898 if (strstart(str, "unix:", NULL)) {
899 if (str[5] == '\0') {
312fd5f2 900 error_setg(errp, "invalid Unix socket address");
101f9cbc
PB
901 goto fail;
902 } else {
2d32adda 903 addr->type = SOCKET_ADDRESS_KIND_UNIX;
32bafa8f
EB
904 addr->u.q_unix.data = g_new(UnixSocketAddress, 1);
905 addr->u.q_unix.data->path = g_strdup(str + 5);
101f9cbc
PB
906 }
907 } else if (strstart(str, "fd:", NULL)) {
908 if (str[3] == '\0') {
312fd5f2 909 error_setg(errp, "invalid file descriptor address");
101f9cbc
PB
910 goto fail;
911 } else {
2d32adda 912 addr->type = SOCKET_ADDRESS_KIND_FD;
32bafa8f
EB
913 addr->u.fd.data = g_new(String, 1);
914 addr->u.fd.data->str = g_strdup(str + 3);
101f9cbc
PB
915 }
916 } else {
2d32adda 917 addr->type = SOCKET_ADDRESS_KIND_INET;
32bafa8f
EB
918 addr->u.inet.data = inet_parse(str, errp);
919 if (addr->u.inet.data == NULL) {
101f9cbc
PB
920 goto fail;
921 }
922 }
923 return addr;
924
925fail:
926 qapi_free_SocketAddress(addr);
927 return NULL;
928}
929
930int socket_connect(SocketAddress *addr, Error **errp,
931 NonBlockingConnectHandler *callback, void *opaque)
932{
101f9cbc
PB
933 int fd;
934
2d32adda 935 switch (addr->type) {
101f9cbc 936 case SOCKET_ADDRESS_KIND_INET:
32bafa8f 937 fd = inet_connect_saddr(addr->u.inet.data, errp, callback, opaque);
101f9cbc
PB
938 break;
939
940 case SOCKET_ADDRESS_KIND_UNIX:
32bafa8f 941 fd = unix_connect_saddr(addr->u.q_unix.data, errp, callback, opaque);
101f9cbc
PB
942 break;
943
944 case SOCKET_ADDRESS_KIND_FD:
32bafa8f 945 fd = monitor_get_fd(cur_mon, addr->u.fd.data->str, errp);
d1ec72a4 946 if (fd >= 0 && callback) {
1a751ebf 947 qemu_set_nonblock(fd);
51795029 948 callback(fd, NULL, opaque);
101f9cbc
PB
949 }
950 break;
951
952 default:
953 abort();
954 }
101f9cbc
PB
955 return fd;
956}
957
958int socket_listen(SocketAddress *addr, Error **errp)
959{
101f9cbc
PB
960 int fd;
961
2d32adda 962 switch (addr->type) {
101f9cbc 963 case SOCKET_ADDRESS_KIND_INET:
32bafa8f 964 fd = inet_listen_saddr(addr->u.inet.data, 0, false, errp);
101f9cbc
PB
965 break;
966
967 case SOCKET_ADDRESS_KIND_UNIX:
32bafa8f 968 fd = unix_listen_saddr(addr->u.q_unix.data, false, errp);
101f9cbc
PB
969 break;
970
971 case SOCKET_ADDRESS_KIND_FD:
32bafa8f 972 fd = monitor_get_fd(cur_mon, addr->u.fd.data->str, errp);
101f9cbc
PB
973 break;
974
975 default:
976 abort();
977 }
101f9cbc
PB
978 return fd;
979}
980
3ecc059d
GH
981int socket_dgram(SocketAddress *remote, SocketAddress *local, Error **errp)
982{
3ecc059d
GH
983 int fd;
984
2d32adda 985 switch (remote->type) {
3ecc059d 986 case SOCKET_ADDRESS_KIND_INET:
32bafa8f
EB
987 fd = inet_dgram_saddr(remote->u.inet.data,
988 local ? local->u.inet.data : NULL, errp);
3ecc059d
GH
989 break;
990
991 default:
992 error_setg(errp, "socket type unsupported for datagram");
7a5b6af1 993 fd = -1;
3ecc059d 994 }
3ecc059d
GH
995 return fd;
996}
17c55dec
DB
997
998
999static SocketAddress *
1000socket_sockaddr_to_address_inet(struct sockaddr_storage *sa,
1001 socklen_t salen,
1002 Error **errp)
1003{
1004 char host[NI_MAXHOST];
1005 char serv[NI_MAXSERV];
1006 SocketAddress *addr;
0399293e 1007 InetSocketAddress *inet;
17c55dec
DB
1008 int ret;
1009
1010 ret = getnameinfo((struct sockaddr *)sa, salen,
1011 host, sizeof(host),
1012 serv, sizeof(serv),
1013 NI_NUMERICHOST | NI_NUMERICSERV);
1014 if (ret != 0) {
1015 error_setg(errp, "Cannot format numeric socket address: %s",
1016 gai_strerror(ret));
1017 return NULL;
1018 }
1019
1020 addr = g_new0(SocketAddress, 1);
2d32adda 1021 addr->type = SOCKET_ADDRESS_KIND_INET;
32bafa8f 1022 inet = addr->u.inet.data = g_new0(InetSocketAddress, 1);
0399293e
EB
1023 inet->host = g_strdup(host);
1024 inet->port = g_strdup(serv);
17c55dec 1025 if (sa->ss_family == AF_INET) {
0399293e 1026 inet->has_ipv4 = inet->ipv4 = true;
17c55dec 1027 } else {
0399293e 1028 inet->has_ipv6 = inet->ipv6 = true;
17c55dec
DB
1029 }
1030
1031 return addr;
1032}
1033
1034
1035#ifndef WIN32
1036static SocketAddress *
1037socket_sockaddr_to_address_unix(struct sockaddr_storage *sa,
1038 socklen_t salen,
1039 Error **errp)
1040{
1041 SocketAddress *addr;
1042 struct sockaddr_un *su = (struct sockaddr_un *)sa;
1043
1044 addr = g_new0(SocketAddress, 1);
2d32adda 1045 addr->type = SOCKET_ADDRESS_KIND_UNIX;
32bafa8f 1046 addr->u.q_unix.data = g_new0(UnixSocketAddress, 1);
17c55dec 1047 if (su->sun_path[0]) {
32bafa8f
EB
1048 addr->u.q_unix.data->path = g_strndup(su->sun_path,
1049 sizeof(su->sun_path));
17c55dec
DB
1050 }
1051
1052 return addr;
1053}
1054#endif /* WIN32 */
1055
559607ea 1056SocketAddress *
17c55dec
DB
1057socket_sockaddr_to_address(struct sockaddr_storage *sa,
1058 socklen_t salen,
1059 Error **errp)
1060{
1061 switch (sa->ss_family) {
1062 case AF_INET:
1063 case AF_INET6:
1064 return socket_sockaddr_to_address_inet(sa, salen, errp);
1065
1066#ifndef WIN32
1067 case AF_UNIX:
1068 return socket_sockaddr_to_address_unix(sa, salen, errp);
1069#endif /* WIN32 */
1070
1071 default:
1072 error_setg(errp, "socket family %d unsupported",
1073 sa->ss_family);
1074 return NULL;
1075 }
1076 return 0;
1077}
1078
1079
1080SocketAddress *socket_local_address(int fd, Error **errp)
1081{
1082 struct sockaddr_storage ss;
1083 socklen_t sslen = sizeof(ss);
1084
1085 if (getsockname(fd, (struct sockaddr *)&ss, &sslen) < 0) {
b16a44e1 1086 error_setg_errno(errp, errno, "%s",
17c55dec
DB
1087 "Unable to query local socket address");
1088 return NULL;
1089 }
1090
1091 return socket_sockaddr_to_address(&ss, sslen, errp);
1092}
1093
1094
1095SocketAddress *socket_remote_address(int fd, Error **errp)
1096{
1097 struct sockaddr_storage ss;
1098 socklen_t sslen = sizeof(ss);
1099
1100 if (getpeername(fd, (struct sockaddr *)&ss, &sslen) < 0) {
b16a44e1 1101 error_setg_errno(errp, errno, "%s",
17c55dec
DB
1102 "Unable to query remote socket address");
1103 return NULL;
1104 }
1105
1106 return socket_sockaddr_to_address(&ss, sslen, errp);
1107}
2a8e21c7
DB
1108
1109
1110void qapi_copy_SocketAddress(SocketAddress **p_dest,
1111 SocketAddress *src)
1112{
1113 QmpOutputVisitor *qov;
1114 QmpInputVisitor *qiv;
1115 Visitor *ov, *iv;
1116 QObject *obj;
1117
1118 *p_dest = NULL;
1119
1120 qov = qmp_output_visitor_new();
1121 ov = qmp_output_get_visitor(qov);
51e72bc1 1122 visit_type_SocketAddress(ov, NULL, &src, &error_abort);
2a8e21c7
DB
1123 obj = qmp_output_get_qobject(qov);
1124 qmp_output_visitor_cleanup(qov);
1125 if (!obj) {
1126 return;
1127 }
1128
1129 qiv = qmp_input_visitor_new(obj);
1130 iv = qmp_input_get_visitor(qiv);
51e72bc1 1131 visit_type_SocketAddress(iv, NULL, p_dest, &error_abort);
2a8e21c7
DB
1132 qmp_input_visitor_cleanup(qiv);
1133 qobject_decref(obj);
1134}