[qemu.git] / qemu-sockets.c

/*
 *  inet and unix socket functions for qemu
 *
 *  (c) 2008 Gerd Hoffmann <[email protected]>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; under version 2 of the License.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <errno.h>
#include <unistd.h>

#include "qemu_socket.h"
#include "qemu-common.h" /* for qemu_isdigit */

#ifndef AI_ADDRCONFIG
# define AI_ADDRCONFIG 0
#endif

static int sockets_debug = 0;
static const int on=1, off=0;

static int inet_getport(struct addrinfo *e)
{
    struct sockaddr_in *i4;
    struct sockaddr_in6 *i6;

    switch (e->ai_family) {
    case PF_INET6:
        i6 = (void*)e->ai_addr;
        return ntohs(i6->sin6_port);
    case PF_INET:
        i4 = (void*)e->ai_addr;
        return ntohs(i4->sin_port);
    default:
        return 0;
    }
}

static void inet_setport(struct addrinfo *e, int port)
{
    struct sockaddr_in *i4;
    struct sockaddr_in6 *i6;

    switch (e->ai_family) {
    case PF_INET6:
        i6 = (void*)e->ai_addr;
        i6->sin6_port = htons(port);
        break;
    case PF_INET:
        i4 = (void*)e->ai_addr;
        i4->sin_port = htons(port);
        break;
    }
}

static const char *inet_strfamily(int family)
{
    switch (family) {
    case PF_INET6: return "ipv6";
    case PF_INET:  return "ipv4";
    case PF_UNIX:  return "unix";
    }
    return "????";
}

static void inet_print_addrinfo(const char *tag, struct addrinfo *res)
{
    struct addrinfo *e;
    char uaddr[INET6_ADDRSTRLEN+1];
    char uport[33];

    for (e = res; e != NULL; e = e->ai_next) {
        getnameinfo((struct sockaddr*)e->ai_addr,e->ai_addrlen,
                    uaddr,INET6_ADDRSTRLEN,uport,32,
                    NI_NUMERICHOST | NI_NUMERICSERV);
        fprintf(stderr,"%s: getaddrinfo: family %s, host %s, port %s\n",
                tag, inet_strfamily(e->ai_family), uaddr, uport);
    }
}

int inet_listen(const char *str, char *ostr, int olen,
                int socktype, int port_offset)
{
    struct addrinfo ai,*res,*e;
    char addr[64];
    char port[33];
    char uaddr[INET6_ADDRSTRLEN+1];
    char uport[33];
    const char *opts, *h;
    int slisten,rc,pos,to,try_next;

    memset(&ai,0, sizeof(ai));
    ai.ai_flags = AI_PASSIVE | AI_ADDRCONFIG;
    ai.ai_family = PF_UNSPEC;
    ai.ai_socktype = socktype;

    /* parse address */
    if (str[0] == ':') {
        /* no host given */
        addr[0] = '\0';
        if (1 != sscanf(str,":%32[^,]%n",port,&pos)) {
            fprintf(stderr, "%s: portonly parse error (%s)\n",
                    __FUNCTION__, str);
            return -1;
        }
    } else if (str[0] == '[') {
        /* IPv6 addr */
        if (2 != sscanf(str,"[%64[^]]]:%32[^,]%n",addr,port,&pos)) {
            fprintf(stderr, "%s: ipv6 parse error (%s)\n",
                    __FUNCTION__, str);
            return -1;
        }
        ai.ai_family = PF_INET6;
    } else if (qemu_isdigit(str[0])) {
        /* IPv4 addr */
        if (2 != sscanf(str,"%64[0-9.]:%32[^,]%n",addr,port,&pos)) {
            fprintf(stderr, "%s: ipv4 parse error (%s)\n",
                    __FUNCTION__, str);
            return -1;
        }
        ai.ai_family = PF_INET;
    } else {
        /* hostname */
        if (2 != sscanf(str,"%64[^:]:%32[^,]%n",addr,port,&pos)) {
            fprintf(stderr, "%s: hostname parse error (%s)\n",
                    __FUNCTION__, str);
            return -1;
        }
    }

    /* parse options */
    opts = str + pos;
    h = strstr(opts, ",to=");
    to = h ? atoi(h+4) : 0;
    if (strstr(opts, ",ipv4"))
        ai.ai_family = PF_INET;
    if (strstr(opts, ",ipv6"))
        ai.ai_family = PF_INET6;

    /* lookup */
    if (port_offset)
        snprintf(port, sizeof(port), "%d", atoi(port) + port_offset);
    rc = getaddrinfo(strlen(addr) ? addr : NULL, port, &ai, &res);
    if (rc != 0) {
        fprintf(stderr,"%s: getaddrinfo(%s,%s): %s\n", __FUNCTION__,
                addr, port, gai_strerror(rc));
        return -1;
    }
    if (sockets_debug)
        inet_print_addrinfo(__FUNCTION__, res);

    /* create socket + bind */
    for (e = res; e != NULL; e = e->ai_next) {
        getnameinfo((struct sockaddr*)e->ai_addr,e->ai_addrlen,
		        uaddr,INET6_ADDRSTRLEN,uport,32,
		        NI_NUMERICHOST | NI_NUMERICSERV);
        slisten = socket(e->ai_family, e->ai_socktype, e->ai_protocol);
        if (slisten < 0) {
            fprintf(stderr,"%s: socket(%s): %s\n", __FUNCTION__,
                    inet_strfamily(e->ai_family), strerror(errno));
            continue;
        }

        setsockopt(slisten,SOL_SOCKET,SO_REUSEADDR,(void*)&on,sizeof(on));
#ifdef IPV6_V6ONLY
        if (e->ai_family == PF_INET6) {
            /* listen on both ipv4 and ipv6 */
            setsockopt(slisten,IPPROTO_IPV6,IPV6_V6ONLY,(void*)&off,
                sizeof(off));
        }
#endif

        for (;;) {
            if (bind(slisten, e->ai_addr, e->ai_addrlen) == 0) {
                if (sockets_debug)
                    fprintf(stderr,"%s: bind(%s,%s,%d): OK\n", __FUNCTION__,
                        inet_strfamily(e->ai_family), uaddr, inet_getport(e));
                goto listen;
            }
            try_next = to && (inet_getport(e) <= to + port_offset);
            if (!try_next || sockets_debug)
                fprintf(stderr,"%s: bind(%s,%s,%d): %s\n", __FUNCTION__,
                        inet_strfamily(e->ai_family), uaddr, inet_getport(e),
                        strerror(errno));
            if (try_next) {
                inet_setport(e, inet_getport(e) + 1);
                continue;
            }
            break;
        }
        closesocket(slisten);
    }
    fprintf(stderr, "%s: FAILED\n", __FUNCTION__);
    freeaddrinfo(res);
    return -1;

listen:
    if (listen(slisten,1) != 0) {
        perror("listen");
        closesocket(slisten);
        freeaddrinfo(res);
        return -1;
    }
    if (ostr) {
        if (e->ai_family == PF_INET6) {
            snprintf(ostr, olen, "[%s]:%d%s", uaddr,
                     inet_getport(e) - port_offset, opts);
        } else {
            snprintf(ostr, olen, "%s:%d%s", uaddr,
                     inet_getport(e) - port_offset, opts);
        }
    }
    freeaddrinfo(res);
    return slisten;
}

int inet_connect(const char *str, int socktype)
{
    struct addrinfo ai,*res,*e;
    char addr[64];
    char port[33];
    char uaddr[INET6_ADDRSTRLEN+1];
    char uport[33];
    int sock,rc;

    memset(&ai,0, sizeof(ai));
    ai.ai_flags = AI_CANONNAME | AI_ADDRCONFIG;
    ai.ai_family = PF_UNSPEC;
    ai.ai_socktype = socktype;

    /* parse address */
    if (str[0] == '[') {
        /* IPv6 addr */
        if (2 != sscanf(str,"[%64[^]]]:%32[^,]",addr,port)) {
            fprintf(stderr, "%s: ipv6 parse error (%s)\n",
                    __FUNCTION__, str);
            return -1;
        }
        ai.ai_family = PF_INET6;
    } else if (qemu_isdigit(str[0])) {
        /* IPv4 addr */
        if (2 != sscanf(str,"%64[0-9.]:%32[^,]",addr,port)) {
            fprintf(stderr, "%s: ipv4 parse error (%s)\n",
                    __FUNCTION__, str);
            return -1;
        }
        ai.ai_family = PF_INET;
    } else {
        /* hostname */
        if (2 != sscanf(str,"%64[^:]:%32[^,]",addr,port)) {
            fprintf(stderr, "%s: hostname parse error (%s)\n",
                    __FUNCTION__, str);
            return -1;
        }
    }

    /* parse options */
    if (strstr(str, ",ipv4"))
        ai.ai_family = PF_INET;
    if (strstr(str, ",ipv6"))
        ai.ai_family = PF_INET6;

    /* lookup */
    if (0 != (rc = getaddrinfo(addr, port, &ai, &res))) {
        fprintf(stderr,"getaddrinfo(%s,%s): %s\n", gai_strerror(rc),
                addr, port);
	return -1;
    }
    if (sockets_debug)
        inet_print_addrinfo(__FUNCTION__, res);

    for (e = res; e != NULL; e = e->ai_next) {
        if (getnameinfo((struct sockaddr*)e->ai_addr,e->ai_addrlen,
                            uaddr,INET6_ADDRSTRLEN,uport,32,
                            NI_NUMERICHOST | NI_NUMERICSERV) != 0) {
            fprintf(stderr,"%s: getnameinfo: oops\n", __FUNCTION__);
            continue;
        }
        sock = socket(e->ai_family, e->ai_socktype, e->ai_protocol);
        if (sock < 0) {
            fprintf(stderr,"%s: socket(%s): %s\n", __FUNCTION__,
            inet_strfamily(e->ai_family), strerror(errno));
            continue;
        }
        setsockopt(sock,SOL_SOCKET,SO_REUSEADDR,(void*)&on,sizeof(on));

        /* connect to peer */
        if (connect(sock,e->ai_addr,e->ai_addrlen) < 0) {
            if (sockets_debug || NULL == e->ai_next)
                fprintf(stderr, "%s: connect(%s,%s,%s,%s): %s\n", __FUNCTION__,
                        inet_strfamily(e->ai_family),
                        e->ai_canonname, uaddr, uport, strerror(errno));
            closesocket(sock);
            continue;
        }
        if (sockets_debug)
            fprintf(stderr, "%s: connect(%s,%s,%s,%s): OK\n", __FUNCTION__,
                    inet_strfamily(e->ai_family),
                    e->ai_canonname, uaddr, uport);
        freeaddrinfo(res);
        return sock;
    }
    freeaddrinfo(res);
    return -1;
}

#ifndef _WIN32

int unix_listen(const char *str, char *ostr, int olen)
{
    struct sockaddr_un un;
    char *path, *opts;
    int sock, fd, len;

    sock = socket(PF_UNIX, SOCK_STREAM, 0);
    if (sock < 0) {
        perror("socket(unix)");
        return -1;
    }

    opts = strchr(str, ',');
    if (opts) {
        len = opts - str;
        path = qemu_malloc(len+1);
        snprintf(path, len+1, "%.*s", len, str);
    } else
        path = qemu_strdup(str);

    memset(&un, 0, sizeof(un));
    un.sun_family = AF_UNIX;
    if (path && strlen(path)) {
        snprintf(un.sun_path, sizeof(un.sun_path), "%s", path);
    } else {
        char *tmpdir = getenv("TMPDIR");
        snprintf(un.sun_path, sizeof(un.sun_path), "%s/qemu-socket-XXXXXX",
                 tmpdir ? tmpdir : "/tmp");
        /*
         * This dummy fd usage silences the mktemp() unsecure warning.
         * Using mkstemp() doesn't make things more secure here
         * though.  bind() complains about existing files, so we have
         * to unlink first and thus re-open the race window.  The
         * worst case possible is bind() failing, i.e. a DoS attack.
         */
        fd = mkstemp(un.sun_path); close(fd);
    }
    snprintf(ostr, olen, "%s%s", un.sun_path, opts ? opts : "");

    unlink(un.sun_path);
    if (bind(sock, (struct sockaddr*) &un, sizeof(un)) < 0) {
        fprintf(stderr, "bind(unix:%s): %s\n", un.sun_path, strerror(errno));
        goto err;
    }
    if (listen(sock, 1) < 0) {
        fprintf(stderr, "listen(unix:%s): %s\n", un.sun_path, strerror(errno));
        goto err;
    }

    if (sockets_debug)
        fprintf(stderr, "bind(unix:%s): OK\n", un.sun_path);
    qemu_free(path);
    return sock;

err:
    qemu_free(path);
    closesocket(sock);
    return -1;
}

int unix_connect(const char *path)
{
    struct sockaddr_un un;
    int sock;

    sock = socket(PF_UNIX, SOCK_STREAM, 0);
    if (sock < 0) {
        perror("socket(unix)");
        return -1;
    }

    memset(&un, 0, sizeof(un));
    un.sun_family = AF_UNIX;
    snprintf(un.sun_path, sizeof(un.sun_path), "%s", path);
    if (connect(sock, (struct sockaddr*) &un, sizeof(un)) < 0) {
        fprintf(stderr, "connect(unix:%s): %s\n", path, strerror(errno));
	return -1;
    }

    if (sockets_debug)
        fprintf(stderr, "connect(unix:%s): OK\n", path);
    return sock;
}

#else

int unix_listen(const char *path, char *ostr, int olen)
{
    fprintf(stderr, "unix sockets are not available on windows\n");
    return -1;
}

int unix_connect(const char *path)
{
    fprintf(stderr, "unix sockets are not available on windows\n");
    return -1;
}

#endif
Commit	Line	Data
305b0eb2 AL	1	/*
	2	* inet and unix socket functions for qemu
	3	*
	4	* (c) 2008 Gerd Hoffmann <[email protected]>
	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.
	14	*/
d247d25f AL	15	#include <stdio.h>
	16	#include <stdlib.h>
	17	#include <string.h>
	18	#include <ctype.h>
	19	#include <errno.h>
	20	#include <unistd.h>
	21
	22	#include "qemu_socket.h"
47398b9c	23	#include "qemu-common.h" /* for qemu_isdigit */
d247d25f AL	24
	25	#ifndef AI_ADDRCONFIG
	26	# define AI_ADDRCONFIG 0
	27	#endif
	28
	29	static int sockets_debug = 0;
	30	static const int on=1, off=0;
	31
	32	static int inet_getport(struct addrinfo *e)
	33	{
	34	struct sockaddr_in *i4;
	35	struct sockaddr_in6 *i6;
	36
	37	switch (e->ai_family) {
	38	case PF_INET6:
	39	i6 = (void*)e->ai_addr;
	40	return ntohs(i6->sin6_port);
	41	case PF_INET:
	42	i4 = (void*)e->ai_addr;
	43	return ntohs(i4->sin_port);
	44	default:
	45	return 0;
	46	}
	47	}
	48
	49	static void inet_setport(struct addrinfo *e, int port)
	50	{
	51	struct sockaddr_in *i4;
	52	struct sockaddr_in6 *i6;
	53
	54	switch (e->ai_family) {
	55	case PF_INET6:
	56	i6 = (void*)e->ai_addr;
	57	i6->sin6_port = htons(port);
	58	break;
	59	case PF_INET:
	60	i4 = (void*)e->ai_addr;
	61	i4->sin_port = htons(port);
	62	break;
	63	}
	64	}
	65
	66	static const char *inet_strfamily(int family)
	67	{
	68	switch (family) {
	69	case PF_INET6: return "ipv6";
	70	case PF_INET: return "ipv4";
	71	case PF_UNIX: return "unix";
	72	}
	73	return "????";
	74	}
	75
	76	static void inet_print_addrinfo(const char tag, struct addrinfo res)
	77	{
	78	struct addrinfo *e;
	79	char uaddr[INET6_ADDRSTRLEN+1];
	80	char uport[33];
	81
	82	for (e = res; e != NULL; e = e->ai_next) {
	83	getnameinfo((struct sockaddr*)e->ai_addr,e->ai_addrlen,
	84	uaddr,INET6_ADDRSTRLEN,uport,32,
	85	NI_NUMERICHOST \| NI_NUMERICSERV);
	86	fprintf(stderr,"%s: getaddrinfo: family %s, host %s, port %s\n",
	87	tag, inet_strfamily(e->ai_family), uaddr, uport);
88	}
89	}
90
91	int inet_listen(const char str, char ostr, int olen,
92	int socktype, int port_offset)
93	{
94	struct addrinfo ai,res,e;
95	char addr[64];
96	char port[33];
97	char uaddr[INET6_ADDRSTRLEN+1];
98	char uport[33];
99	const char opts, h;
100	int slisten,rc,pos,to,try_next;
101
102	memset(&ai,0, sizeof(ai));
103	ai.ai_flags = AI_PASSIVE \| AI_ADDRCONFIG;
104	ai.ai_family = PF_UNSPEC;
105	ai.ai_socktype = socktype;
106
107	/* parse address */
108	if (str[0] == ':') {
109	/* no host given */
bc575e95	110	addr[0] = '\0';
d247d25f AL	111	if (1 != sscanf(str,":%32[^,]%n",port,&pos)) {
	112	fprintf(stderr, "%s: portonly parse error (%s)\n",
	113	__FUNCTION__, str);
	114	return -1;
	115	}
	116	} else if (str[0] == '[') {
	117	/* IPv6 addr */
	118	if (2 != sscanf(str,"[%64[^]]]:%32[^,]%n",addr,port,&pos)) {
	119	fprintf(stderr, "%s: ipv6 parse error (%s)\n",
	120	__FUNCTION__, str);
	121	return -1;
	122	}
	123	ai.ai_family = PF_INET6;
47398b9c	124	} else if (qemu_isdigit(str[0])) {
d247d25f AL	125	/* IPv4 addr */
	126	if (2 != sscanf(str,"%64[0-9.]:%32[^,]%n",addr,port,&pos)) {
	127	fprintf(stderr, "%s: ipv4 parse error (%s)\n",
	128	__FUNCTION__, str);
	129	return -1;
	130	}
	131	ai.ai_family = PF_INET;
	132	} else {
	133	/* hostname */
	134	if (2 != sscanf(str,"%64[^:]:%32[^,]%n",addr,port,&pos)) {
	135	fprintf(stderr, "%s: hostname parse error (%s)\n",
	136	__FUNCTION__, str);
	137	return -1;
	138	}
	139	}
	140
	141	/* parse options */
	142	opts = str + pos;
	143	h = strstr(opts, ",to=");
	144	to = h ? atoi(h+4) : 0;
	145	if (strstr(opts, ",ipv4"))
	146	ai.ai_family = PF_INET;
	147	if (strstr(opts, ",ipv6"))
	148	ai.ai_family = PF_INET6;
	149
	150	/* lookup */
	151	if (port_offset)
	152	snprintf(port, sizeof(port), "%d", atoi(port) + port_offset);
	153	rc = getaddrinfo(strlen(addr) ? addr : NULL, port, &ai, &res);
	154	if (rc != 0) {
	155	fprintf(stderr,"%s: getaddrinfo(%s,%s): %s\n", __FUNCTION__,
	156	addr, port, gai_strerror(rc));
	157	return -1;
	158	}
	159	if (sockets_debug)
	160	inet_print_addrinfo(__FUNCTION__, res);
	161
	162	/* create socket + bind */
	163	for (e = res; e != NULL; e = e->ai_next) {
39b6efc8 VS	164	getnameinfo((struct sockaddr*)e->ai_addr,e->ai_addrlen,
	165	uaddr,INET6_ADDRSTRLEN,uport,32,
	166	NI_NUMERICHOST \| NI_NUMERICSERV);
d247d25f	167	slisten = socket(e->ai_family, e->ai_socktype, e->ai_protocol);
39b6efc8	168	if (slisten < 0) {
d247d25f AL	169	fprintf(stderr,"%s: socket(%s): %s\n", __FUNCTION__,
d247d25f AL	170	inet_strfamily(e->ai_family), strerror(errno));
39b6efc8 VS	171	continue;
39b6efc8 VS	172	}
d247d25f AL	173
	174	setsockopt(slisten,SOL_SOCKET,SO_REUSEADDR,(void*)&on,sizeof(on));
	175	#ifdef IPV6_V6ONLY
	176	if (e->ai_family == PF_INET6) {
	177	/* listen on both ipv4 and ipv6 */
39b6efc8 VS	178	setsockopt(slisten,IPPROTO_IPV6,IPV6_V6ONLY,(void*)&off,
39b6efc8 VS	179	sizeof(off));
d247d25f AL	180	}
	181	#endif
	182
	183	for (;;) {
	184	if (bind(slisten, e->ai_addr, e->ai_addrlen) == 0) {
	185	if (sockets_debug)
	186	fprintf(stderr,"%s: bind(%s,%s,%d): OK\n", __FUNCTION__,
39b6efc8	187	inet_strfamily(e->ai_family), uaddr, inet_getport(e));
d247d25f AL	188	goto listen;
	189	}
	190	try_next = to && (inet_getport(e) <= to + port_offset);
	191	if (!try_next \|\| sockets_debug)
	192	fprintf(stderr,"%s: bind(%s,%s,%d): %s\n", __FUNCTION__,
	193	inet_strfamily(e->ai_family), uaddr, inet_getport(e),
	194	strerror(errno));
	195	if (try_next) {
	196	inet_setport(e, inet_getport(e) + 1);
	197	continue;
	198	}
	199	break;
	200	}
	201	closesocket(slisten);
	202	}
	203	fprintf(stderr, "%s: FAILED\n", __FUNCTION__);
	204	freeaddrinfo(res);
	205	return -1;
	206
	207	listen:
	208	if (listen(slisten,1) != 0) {
	209	perror("listen");
	210	closesocket(slisten);
39b6efc8	211	freeaddrinfo(res);
d247d25f AL	212	return -1;
	213	}
	214	if (ostr) {
	215	if (e->ai_family == PF_INET6) {
	216	snprintf(ostr, olen, "[%s]:%d%s", uaddr,
	217	inet_getport(e) - port_offset, opts);
	218	} else {
	219	snprintf(ostr, olen, "%s:%d%s", uaddr,
	220	inet_getport(e) - port_offset, opts);
	221	}
	222	}
	223	freeaddrinfo(res);
	224	return slisten;
	225	}
	226
	227	int inet_connect(const char *str, int socktype)
	228	{
	229	struct addrinfo ai,res,e;
	230	char addr[64];
	231	char port[33];
	232	char uaddr[INET6_ADDRSTRLEN+1];
	233	char uport[33];
	234	int sock,rc;
	235
	236	memset(&ai,0, sizeof(ai));
	237	ai.ai_flags = AI_CANONNAME \| AI_ADDRCONFIG;
	238	ai.ai_family = PF_UNSPEC;
	239	ai.ai_socktype = socktype;
	240
	241	/* parse address */
	242	if (str[0] == '[') {
	243	/* IPv6 addr */
	244	if (2 != sscanf(str,"[%64[^]]]:%32[^,]",addr,port)) {
	245	fprintf(stderr, "%s: ipv6 parse error (%s)\n",
	246	__FUNCTION__, str);
	247	return -1;
	248	}
	249	ai.ai_family = PF_INET6;
47398b9c	250	} else if (qemu_isdigit(str[0])) {
d247d25f AL	251	/* IPv4 addr */
	252	if (2 != sscanf(str,"%64[0-9.]:%32[^,]",addr,port)) {
	253	fprintf(stderr, "%s: ipv4 parse error (%s)\n",
	254	__FUNCTION__, str);
	255	return -1;
	256	}
	257	ai.ai_family = PF_INET;
	258	} else {
	259	/* hostname */
	260	if (2 != sscanf(str,"%64[^:]:%32[^,]",addr,port)) {
	261	fprintf(stderr, "%s: hostname parse error (%s)\n",
	262	__FUNCTION__, str);
	263	return -1;
	264	}
	265	}
	266
	267	/* parse options */
	268	if (strstr(str, ",ipv4"))
	269	ai.ai_family = PF_INET;
	270	if (strstr(str, ",ipv6"))
	271	ai.ai_family = PF_INET6;
	272
	273	/* lookup */
	274	if (0 != (rc = getaddrinfo(addr, port, &ai, &res))) {
	275	fprintf(stderr,"getaddrinfo(%s,%s): %s\n", gai_strerror(rc),
	276	addr, port);
	277	return -1;
	278	}
	279	if (sockets_debug)
	280	inet_print_addrinfo(__FUNCTION__, res);
	281
	282	for (e = res; e != NULL; e = e->ai_next) {
39b6efc8 VS	283	if (getnameinfo((struct sockaddr*)e->ai_addr,e->ai_addrlen,
	284	uaddr,INET6_ADDRSTRLEN,uport,32,
	285	NI_NUMERICHOST \| NI_NUMERICSERV) != 0) {
d247d25f	286	fprintf(stderr,"%s: getnameinfo: oops\n", __FUNCTION__);
39b6efc8 VS	287	continue;
39b6efc8 VS	288	}
d247d25f	289	sock = socket(e->ai_family, e->ai_socktype, e->ai_protocol);
39b6efc8	290	if (sock < 0) {
d247d25f	291	fprintf(stderr,"%s: socket(%s): %s\n", __FUNCTION__,
39b6efc8 VS	292	inet_strfamily(e->ai_family), strerror(errno));
	293	continue;
	294	}
d247d25f AL	295	setsockopt(sock,SOL_SOCKET,SO_REUSEADDR,(void*)&on,sizeof(on));
d247d25f AL	296
39b6efc8 VS	297	/* connect to peer */
39b6efc8 VS	298	if (connect(sock,e->ai_addr,e->ai_addrlen) < 0) {
d247d25f AL	299	if (sockets_debug \|\| NULL == e->ai_next)
	300	fprintf(stderr, "%s: connect(%s,%s,%s,%s): %s\n", __FUNCTION__,
	301	inet_strfamily(e->ai_family),
	302	e->ai_canonname, uaddr, uport, strerror(errno));
	303	closesocket(sock);
39b6efc8 VS	304	continue;
39b6efc8 VS	305	}
d247d25f AL	306	if (sockets_debug)
	307	fprintf(stderr, "%s: connect(%s,%s,%s,%s): OK\n", __FUNCTION__,
	308	inet_strfamily(e->ai_family),
	309	e->ai_canonname, uaddr, uport);
	310	freeaddrinfo(res);
39b6efc8	311	return sock;
d247d25f AL	312	}
	313	freeaddrinfo(res);
	314	return -1;
	315	}
	316
	317	#ifndef _WIN32
	318
	319	int unix_listen(const char str, char ostr, int olen)
	320	{
	321	struct sockaddr_un un;
	322	char path, opts;
	323	int sock, fd, len;
	324
	325	sock = socket(PF_UNIX, SOCK_STREAM, 0);
	326	if (sock < 0) {
39b6efc8 VS	327	perror("socket(unix)");
39b6efc8 VS	328	return -1;
d247d25f AL	329	}
	330
	331	opts = strchr(str, ',');
	332	if (opts) {
	333	len = opts - str;
39b6efc8	334	path = qemu_malloc(len+1);
d247d25f AL	335	snprintf(path, len+1, "%.*s", len, str);
d247d25f AL	336	} else
39b6efc8	337	path = qemu_strdup(str);
d247d25f AL	338
	339	memset(&un, 0, sizeof(un));
	340	un.sun_family = AF_UNIX;
	341	if (path && strlen(path)) {
	342	snprintf(un.sun_path, sizeof(un.sun_path), "%s", path);
	343	} else {
	344	char *tmpdir = getenv("TMPDIR");
	345	snprintf(un.sun_path, sizeof(un.sun_path), "%s/qemu-socket-XXXXXX",
	346	tmpdir ? tmpdir : "/tmp");
	347	/*
	348	* This dummy fd usage silences the mktemp() unsecure warning.
	349	* Using mkstemp() doesn't make things more secure here
	350	* though. bind() complains about existing files, so we have
	351	* to unlink first and thus re-open the race window. The
	352	* worst case possible is bind() failing, i.e. a DoS attack.
	353	*/
	354	fd = mkstemp(un.sun_path); close(fd);
	355	}
	356	snprintf(ostr, olen, "%s%s", un.sun_path, opts ? opts : "");
	357
	358	unlink(un.sun_path);
	359	if (bind(sock, (struct sockaddr*) &un, sizeof(un)) < 0) {
	360	fprintf(stderr, "bind(unix:%s): %s\n", un.sun_path, strerror(errno));
	361	goto err;
	362	}
	363	if (listen(sock, 1) < 0) {
	364	fprintf(stderr, "listen(unix:%s): %s\n", un.sun_path, strerror(errno));
	365	goto err;
	366	}
	367
	368	if (sockets_debug)
	369	fprintf(stderr, "bind(unix:%s): OK\n", un.sun_path);
39b6efc8	370	qemu_free(path);
d247d25f AL	371	return sock;
	372
	373	err:
39b6efc8	374	qemu_free(path);
d247d25f AL	375	closesocket(sock);
	376	return -1;
	377	}
	378
	379	int unix_connect(const char *path)
	380	{
	381	struct sockaddr_un un;
	382	int sock;
	383
	384	sock = socket(PF_UNIX, SOCK_STREAM, 0);
	385	if (sock < 0) {
39b6efc8 VS	386	perror("socket(unix)");
39b6efc8 VS	387	return -1;
d247d25f AL	388	}
	389
	390	memset(&un, 0, sizeof(un));
	391	un.sun_family = AF_UNIX;
	392	snprintf(un.sun_path, sizeof(un.sun_path), "%s", path);
	393	if (connect(sock, (struct sockaddr*) &un, sizeof(un)) < 0) {
	394	fprintf(stderr, "connect(unix:%s): %s\n", path, strerror(errno));
	395	return -1;
	396	}
	397
	398	if (sockets_debug)
	399	fprintf(stderr, "connect(unix:%s): OK\n", path);
	400	return sock;
	401	}
	402
	403	#else
	404
	405	int unix_listen(const char path, char ostr, int olen)
	406	{
	407	fprintf(stderr, "unix sockets are not available on windows\n");
	408	return -1;
	409	}
	410
	411	int unix_connect(const char *path)
	412	{
	413	fprintf(stderr, "unix sockets are not available on windows\n");
	414	return -1;
	415	}
	416
	417	#endif