GCOV - Varnish Cache


/*-
 * Copyright (c) 2006 Verdens Gang AS
 * Copyright (c) 2006-2015 Varnish Software AS
 * All rights reserved.
 *
 * Author: Poul-Henning Kamp <phk@phk.freebsd.dk>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include "config.h"

#include <sys/types.h>
#include <sys/socket.h>
#include <sys/time.h>           // for NetBSD
#include <sys/ioctl.h>
#ifdef HAVE_SYS_FILIO_H
#  include <sys/filio.h>
#endif

#include <netinet/in.h>
#include <netinet/tcp.h>

#include <math.h>
#include <netdb.h>
#include <poll.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>

#include "vdef.h"
#include "miniobj.h"
#include "vas.h"
#include "vsa.h"
#include "vss.h"
#include "vtcp.h"
#include "vtim.h"

/*--------------------------------------------------------------------*/
static void
vtcp_sa_to_ascii(const void *sa, socklen_t l, char *abuf, unsigned alen,
    char *pbuf, unsigned plen)
{
        int i;

        assert(abuf == NULL || alen > 0);
        assert(pbuf == NULL || plen > 0);
        i = getnameinfo(sa, l, abuf, alen, pbuf, plen,
           NI_NUMERICHOST | NI_NUMERICSERV);
        if (i) {
                /*
                 * XXX this printf is shitty, but we may not have space
                 * for the gai_strerror in the buffer :-(
                 */
                fprintf(stderr, "getnameinfo = %d %s\n", i, gai_strerror(i));
                if (i == EAI_SYSTEM)
                        fprintf(stderr, "errno = %d %s\n", errno, VAS_errtxt(errno));
                if (abuf != NULL)
                        (void)snprintf(abuf, alen, "Conversion");
                if (pbuf != NULL)
                        (void)snprintf(pbuf, plen, "Failed");
                return;
        }
        /* XXX dirty hack for v4-to-v6 mapped addresses */
        if (abuf != NULL && strncmp(abuf, "::ffff:", 7) == 0) {
                for (i = 0; abuf[i + 7]; ++i)
                        abuf[i] = abuf[i + 7];
                abuf[i] = '\0';
        }
}

/*--------------------------------------------------------------------*/

void
VTCP_name(const struct suckaddr *addr, char *abuf, unsigned alen,
    char *pbuf, unsigned plen)
{
        const struct sockaddr *sa;
        socklen_t sl;

        sa = VSA_Get_Sockaddr(addr, &sl);
        AN(sa);
        vtcp_sa_to_ascii(sa, sl, abuf, alen, pbuf, plen);
}

/*--------------------------------------------------------------------*/

struct suckaddr *
VTCP_my_suckaddr(int sock)
{
        struct suckaddr *r;

        r = malloc(vsa_suckaddr_len);
        AN(VSA_getsockname(sock, r, vsa_suckaddr_len));
        return (r);
}

/*--------------------------------------------------------------------*/

void
VTCP_myname(int sock, char *abuf, unsigned alen, char *pbuf, unsigned plen)
{
        char buf[vsa_suckaddr_len];

        VTCP_name(VSA_getsockname(sock, buf, sizeof buf),
                  abuf, alen, pbuf, plen);
}

/*--------------------------------------------------------------------*/

void
VTCP_hisname(int sock, char *abuf, unsigned alen, char *pbuf, unsigned plen)
{
        char buf[vsa_suckaddr_len];
        const struct suckaddr *sua;

        sua = VSA_getpeername(sock, buf, sizeof buf);
        if (sua != NULL)
                VTCP_name(sua, abuf, alen, pbuf, plen);
        else {
                (void)snprintf(abuf, alen, "<none>");
                (void)snprintf(pbuf, plen, "<none>");
        }
}

/*--------------------------------------------------------------------*/

#ifdef HAVE_ACCEPT_FILTERS

int
VTCP_filter_http(int sock)
{
        int retval;
        struct accept_filter_arg afa;

        memset(&afa, 0, sizeof afa);
        bprintf(afa.af_name, "%s", "httpready");
        retval = setsockopt(sock, SOL_SOCKET, SO_ACCEPTFILTER,
            &afa, sizeof afa);
        return (retval);
}

#elif defined(__linux)

int
VTCP_filter_http(int sock)
{
        int retval;
        int defer = 1;

        retval = setsockopt(sock, SOL_TCP, TCP_DEFER_ACCEPT,
            &defer, sizeof defer);
        return (retval);
}

#else

int
VTCP_filter_http(int sock)
{
        errno = EOPNOTSUPP;
        (void)sock;
        return (-1);
}

#endif

/*--------------------------------------------------------------------*/


int
VTCP_fastopen(int sock, int depth)
{
#ifdef HAVE_TCP_FASTOPEN
#  ifndef SOL_TCP
#    define SOL_TCP IPPROTO_TCP
#  endif
        return (setsockopt(sock, SOL_TCP, TCP_FASTOPEN, &depth, sizeof depth));
#else
        errno = EOPNOTSUPP;
        (void)sock;
        (void)depth;
        return (-1);
#endif
}


/*--------------------------------------------------------------------
 * Functions for controlling NONBLOCK mode.
 *
 * We use FIONBIO because it is cheaper than fcntl(2), which requires
 * us to do two syscalls, one to get and one to set, the latter of
 * which mucks about a bit before it ends up calling ioctl(FIONBIO),
 * at least on FreeBSD.
 * On Solaris ioctl(FIONBIO) can fail with connection related errnos,
 * but as long as that is how they fail, we're fine.
 */

void
VTCP_blocking(int sock)
{
        int i, j;

        i = 0;
        j = ioctl(sock, FIONBIO, &i);
        VTCP_Assert(j);
}

void
VTCP_nonblocking(int sock)
{
        int i, j;

        i = 1;
        j = ioctl(sock, FIONBIO, &i);
        VTCP_Assert(j);
}

/*--------------------------------------------------------------------
 * On TCP a connect(2) can block for a looong time, and we don't want that.
 * Unfortunately, the SocketWizards back in those days were happy to wait
 * any amount of time for a connection, so the connect(2) syscall does not
 * take an argument for patience.
 *
 * There is a little used work-around, and we employ it at our peril.
 *
 */

int
VTCP_connected(int s)
{
        int k;
        socklen_t l;

        /* Find out if we got a connection */
        l = sizeof k;
        AZ(getsockopt(s, SOL_SOCKET, SO_ERROR, &k, &l));

        /* An error means no connection established */
        errno = k;
        if (k) {
                closefd(&s);
                return (-1);
        }

        VTCP_blocking(s);
        return (s);
}

int
VTCP_connect(const struct suckaddr *name, int msec)
{
        int s, i;
        struct pollfd fds[1];
        const struct sockaddr *sa;
        socklen_t sl;
        int val;

        if (name == NULL)
                return (-1);
        /* Attempt the connect */
        AN(VSA_Sane(name));
        sa = VSA_Get_Sockaddr(name, &sl);
        AN(sa);
        AN(sl);

        s = socket(sa->sa_family, SOCK_STREAM, 0);
        if (s < 0)
                return (s);

        /* Set the socket non-blocking */
        if (msec != 0)
                VTCP_nonblocking(s);

        val = 1;
        AZ(setsockopt(s, IPPROTO_TCP, TCP_NODELAY, &val, sizeof val));

        i = connect(s, sa, sl);
        if (i == 0)
                return (s);
        if (errno != EINPROGRESS) {
                closefd(&s);
                return (-1);
        }

        if (msec < 0) {
                /*
                 * Caller is responsible for waiting and
                 * calling VTCP_connected
                 */
                return (s);
        }

        assert(msec > 0);
        /* Exercise our patience, polling for write */
        fds[0].fd = s;
        fds[0].events = POLLWRNORM;
        fds[0].revents = 0;
        i = poll(fds, 1, msec);

        if (i == 0) {
                /* Timeout, close and give up */
                closefd(&s);
                errno = ETIMEDOUT;
                return (-1);
        }

        return (VTCP_connected(s));
}

/*--------------------------------------------------------------------
 * When closing a TCP connection, a couple of errno's are legit, we
 * can't be held responsible for the other end wanting to talk to us.
 */

void
VTCP_close(int *s)
{
        int i;

        i = close(*s);

        VTCP_Assert(i);
        *s = -1;
}

void
VTCP_set_read_timeout(int s, vtim_dur seconds)
{
        struct timeval timeout = VTIM_timeval_sock(seconds);

        /*
         * Solaris bug (present at least in snv_151 and older): If this fails
         * with EINVAL, the socket is half-closed (SS_CANTSENDMORE) and the
         * timeout does not get set. Needs to be fixed in Solaris, there is
         * nothing we can do about this.
         */
        VTCP_Assert(setsockopt(s, SOL_SOCKET, SO_RCVTIMEO,
            &timeout, sizeof timeout));
}

/*--------------------------------------------------------------------
 */

struct vto_priv {
        unsigned                magic;
#define VTO_PRIV_MAGIC          0xca70b0e7
        int                     latest_errno;
        int                     fd;
        double                  timeout;
};

static int v_matchproto_(vss_resolved_f)
vtcp_open_callback(void *priv, const struct suckaddr *sa)
{
        struct vto_priv *vto;
        int fd;

        CAST_OBJ_NOTNULL(vto, priv, VTO_PRIV_MAGIC);

        errno = 0;
        fd = VTCP_connect(sa, (int)floor(vto->timeout * 1e3));
        if (fd >= 0) {
                vto->fd = fd;
                vto->latest_errno = 0;
                return (1);
        }
        vto->latest_errno = errno;
        return (0);
}

int
VTCP_open(const char *addr, const char *def_port, vtim_dur timeout,
    const char **errp)
{
        struct vto_priv vto[1];

        AN(errp);
        assert(timeout >= 0);
        INIT_OBJ(vto, VTO_PRIV_MAGIC);
        vto->fd = -1;
        vto->timeout = timeout;

        if (VSS_resolver(addr, def_port, vtcp_open_callback, vto, errp) < 0)
                return (-1);
        if (vto->fd < 0)
                *errp = strerror(vto->latest_errno);
        return (vto->fd);
}

/*--------------------------------------------------------------------
 * Given a struct suckaddr, open a socket of the appropriate type, and bind
 * it to the requested address.
 *
 * If the address is an IPv6 address, the IPV6_V6ONLY option is set to
 * avoid conflicts between INADDR_ANY and IN6ADDR_ANY.
 */

int
VTCP_bind(const struct suckaddr *sa, const char **errp)
{
        int sd, val, e;
        socklen_t sl;
        const struct sockaddr *so;
        int proto;

        if (errp != NULL)
                *errp = NULL;

        proto = VSA_Get_Proto(sa);
        sd = socket(proto, SOCK_STREAM, 0);
        if (sd < 0) {
                if (errp != NULL)
                        *errp = "socket(2)";
                return (-1);
        }
        val = 1;
        if (setsockopt(sd, SOL_SOCKET, SO_REUSEADDR, &val, sizeof val) != 0) {
                if (errp != NULL)
                        *errp = "setsockopt(SO_REUSEADDR, 1)";
                e = errno;
                closefd(&sd);
                errno = e;
                return (-1);
        }
#ifdef IPV6_V6ONLY
        /* forcibly use separate sockets for IPv4 and IPv6 */
        val = 1;
        if (proto == AF_INET6 &&
            setsockopt(sd, IPPROTO_IPV6, IPV6_V6ONLY, &val, sizeof val) != 0) {
                if (errp != NULL)
                        *errp = "setsockopt(IPV6_V6ONLY, 1)";
                e = errno;
                closefd(&sd);
                errno = e;
                return (-1);
        }
#endif
        so = VSA_Get_Sockaddr(sa, &sl);
        if (bind(sd, so, sl) != 0) {
                if (errp != NULL)
                        *errp = "bind(2)";
                e = errno;
                closefd(&sd);
                errno = e;
                return (-1);
        }
        return (sd);
}

/*--------------------------------------------------------------------
 * Given a struct suckaddr, open a socket of the appropriate type, bind it
 * to the requested address, and start listening.
 */

int
VTCP_listen(const struct suckaddr *sa, int depth, const char **errp)
{
        int sd;
        int e;

        if (errp != NULL)
                *errp = NULL;
        sd = VTCP_bind(sa, errp);
        if (sd >= 0)  {
                if (listen(sd, depth) != 0) {
                        e = errno;
                        closefd(&sd);
                        errno = e;
                        if (errp != NULL)
                                *errp = "listen(2)";
                        return (-1);
                }
        }
        return (sd);
}

/*--------------------------------------------------------------------*/

struct helper {
        int             depth;
        const char      **errp;
};

static int v_matchproto_(vss_resolved_f)
vtcp_lo_cb(void *priv, const struct suckaddr *sa)
{
        int sock;
        struct helper *hp = priv;

        sock = VTCP_listen(sa, hp->depth, hp->errp);
        if (sock >= 0) {
                *hp->errp = NULL;
                return (sock);
        }
        AN(*hp->errp);
        return (0);
}

int
VTCP_listen_on(const char *addr, const char *def_port, int depth,
    const char **errp)
{
        struct helper h;
        int sock;

        AN(errp);
        h.depth = depth;
        h.errp = errp;

        sock = VSS_resolver(addr, def_port, vtcp_lo_cb, &h, errp);
        if (*errp != NULL)
                return (-1);
        return (sock);
}

/*--------------------------------------------------------------------
 * Set or reset SO_LINGER flag
 */

int
VTCP_linger(int sock, int linger)
{
        struct linger lin;
        int i;

        memset(&lin, 0, sizeof lin);
        lin.l_onoff = linger;
        i = setsockopt(sock, SOL_SOCKET, SO_LINGER, &lin, sizeof lin);
        VTCP_Assert(i);
        return (i);
}

/*--------------------------------------------------------------------
 * Do a poll to check for remote HUP
 */

int
VTCP_check_hup(int sock)
{
        struct pollfd pfd;

        assert(sock > 0);
        pfd.fd = sock;
        pfd.events = POLLOUT;
        pfd.revents = 0;

        if (poll(&pfd, 1, 0) == 1 && pfd.revents & POLLHUP)
                return (1);
        return (0);
}

/*--------------------------------------------------------------------
 * Check if a TCP syscall return value is fatal
 */

int
VTCP_Check(ssize_t a)
{
        if (a == 0)
                return (1);
        if (a > 0)
                return (1);
        if (errno == ECONNRESET || errno == ENOTCONN || errno == EPIPE)
                return (1);
        /* Accept EAGAIN (and EWOULDBLOCK in case they are not the same)
         * as errno values. Even though our sockets are all non-blocking,
         * when a SO_{SND|RCV}TIMEO expires, read() or write() on the
         * socket will return (-1) and errno set to EAGAIN. (This is not
         * documented in the read(2) and write(2) manpages, but is
         * described in the socket(7) manpage.) */
        if (errno == EAGAIN || errno == EWOULDBLOCK)
                return (1);
        /* tcp(7): The other end didn't acknowledge retransmitted data after
         * some time. */
        if (errno == ETIMEDOUT)
                return (1);
        /* #3539 various errnos documented on linux as POSIX.1 */
        if (errno == ENETDOWN || errno == ENETUNREACH || errno == ENETRESET ||
            errno == ECONNABORTED || /* ECONNRESET see above */
            errno == EHOSTUNREACH || errno == EHOSTDOWN) {
                return (1);
        }

#if (defined (__SVR4) && defined (__sun))
        if (errno == ECONNREFUSED)      // in r02702.vtc
                return (1);
        if (errno == EPROTO)
                return (1);
#endif
#if (defined (__SVR4) && defined (__sun)) ||            \
    defined (__NetBSD__) ||                             \
    defined (__APPLE__)
        /*
         * Solaris and macOS returns EINVAL if the other end unexpectedly reset
         * the connection.
         *
         * On NetBSD it is documented behaviour.
         */
        if (errno == EINVAL)
                return (1);
#endif
#if defined(ENABLE_SANITIZER)
        if (errno == EINTR)
                return (1);
#endif
        return (0);
}

/*--------------------------------------------------------------------
 *
 */

int
VTCP_read(int fd, void *ptr, size_t len, vtim_dur tmo)
{
        struct pollfd pfd[1];
        int i, j;

        if (tmo > 0.0) {
                pfd[0].fd = fd;
                pfd[0].events = POLLIN;
                pfd[0].revents = 0;
                j = poll(pfd, 1, VTIM_poll_tmo(tmo));
                if (j == 0)
                        return (-2);
        }
        i = read(fd, ptr, len);
        VTCP_Assert(i);
        return (i < 0 ? -1 : i);
}

		varnish-cache/lib/libvarnish/vtcp.c
0		/*-
1		* Copyright (c) 2006 Verdens Gang AS
2		* Copyright (c) 2006-2015 Varnish Software AS
3		* All rights reserved.
4		*
5		* Author: Poul-Henning Kamp <phk@phk.freebsd.dk>
6		*
7		* SPDX-License-Identifier: BSD-2-Clause
8		*
9		* Redistribution and use in source and binary forms, with or without
10		* modification, are permitted provided that the following conditions
11		* are met:
12		* 1. Redistributions of source code must retain the above copyright
13		* notice, this list of conditions and the following disclaimer.
14		* 2. Redistributions in binary form must reproduce the above copyright
15		* notice, this list of conditions and the following disclaimer in the
16		* documentation and/or other materials provided with the distribution.
17		*
18		* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
19		* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20		* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21		* ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
22		* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23		* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
24		* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25		* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26		* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27		* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28		* SUCH DAMAGE.
29		*/
30
31		#include "config.h"
32
33		#include <sys/types.h>
34		#include <sys/socket.h>
35		#include <sys/time.h> // for NetBSD
36		#include <sys/ioctl.h>
37		#ifdef HAVE_SYS_FILIO_H
38		# include <sys/filio.h>
39		#endif
40
41		#include <netinet/in.h>
42		#include <netinet/tcp.h>
43
44		#include <math.h>
45		#include <netdb.h>
46		#include <poll.h>
47		#include <stdio.h>
48		#include <string.h>
49		#include <unistd.h>
50		#include <stdlib.h>
51
52		#include "vdef.h"
53		#include "miniobj.h"
54		#include "vas.h"
55		#include "vsa.h"
56		#include "vss.h"
57		#include "vtcp.h"
58		#include "vtim.h"
59
60		/--------------------------------------------------------------------/
61		static void
62	1125327	vtcp_sa_to_ascii(const void sa, socklen_t l, char abuf, unsigned alen,
63		char *pbuf, unsigned plen)
64		{
65		int i;
66
67	1125327	assert(abuf == NULL \|\| alen > 0);
68	1125305	assert(pbuf == NULL \|\| plen > 0);
69	1125305	i = getnameinfo(sa, l, abuf, alen, pbuf, plen,
70		NI_NUMERICHOST \| NI_NUMERICSERV);
71	1125305	if (i) {
72		/*
73		* XXX this printf is shitty, but we may not have space
74		* for the gai_strerror in the buffer :-(
75		*/
76	0	fprintf(stderr, "getnameinfo = %d %s\n", i, gai_strerror(i));
77	0	if (i == EAI_SYSTEM)
78	0	fprintf(stderr, "errno = %d %s\n", errno, VAS_errtxt(errno));
79	0	if (abuf != NULL)
80	0	(void)snprintf(abuf, alen, "Conversion");
81	0	if (pbuf != NULL)
82	0	(void)snprintf(pbuf, plen, "Failed");
83	0	return;
84		}
85		/* XXX dirty hack for v4-to-v6 mapped addresses */
86	1125305	if (abuf != NULL && strncmp(abuf, "::ffff:", 7) == 0) {
87	0	for (i = 0; abuf[i + 7]; ++i)
88	0	abuf[i] = abuf[i + 7];
89	0	abuf[i] = '\0';
90	0	}
91	1125325	}
92
93		/--------------------------------------------------------------------/
94
95		void
96	1125286	VTCP_name(const struct suckaddr addr, char abuf, unsigned alen,
97		char *pbuf, unsigned plen)
98		{
99		const struct sockaddr *sa;
100		socklen_t sl;
101
102	1125286	sa = VSA_Get_Sockaddr(addr, &sl);
103	1125286	AN(sa);
104	1125286	vtcp_sa_to_ascii(sa, sl, abuf, alen, pbuf, plen);
105	1125286	}
106
107		/--------------------------------------------------------------------/
108
109		struct suckaddr *
110	38280	VTCP_my_suckaddr(int sock)
111		{
112		struct suckaddr *r;
113
114	38280	r = malloc(vsa_suckaddr_len);
115	38280	AN(VSA_getsockname(sock, r, vsa_suckaddr_len));
116	38280	return (r);
117		}
118
119		/--------------------------------------------------------------------/
120
121		void
122	496593	VTCP_myname(int sock, char abuf, unsigned alen, char pbuf, unsigned plen)
123		{
124	496593	char buf[vsa_suckaddr_len];
125
126	993186	VTCP_name(VSA_getsockname(sock, buf, sizeof buf),
127	496593	abuf, alen, pbuf, plen);
128	496593	}
129
130		/--------------------------------------------------------------------/
131
132		void
133	311002	VTCP_hisname(int sock, char abuf, unsigned alen, char pbuf, unsigned plen)
134		{
135	311002	char buf[vsa_suckaddr_len];
136		const struct suckaddr *sua;
137
138	311002	sua = VSA_getpeername(sock, buf, sizeof buf);
139	311002	if (sua != NULL)
140	311002	VTCP_name(sua, abuf, alen, pbuf, plen);
141		else {
142	0	(void)snprintf(abuf, alen, "<none>");
143	0	(void)snprintf(pbuf, plen, "<none>");
144		}
145	311002	}
146
147		/--------------------------------------------------------------------/
148
149		#ifdef HAVE_ACCEPT_FILTERS
150
151		int
152	36480	VTCP_filter_http(int sock)
153		{
154		int retval;
155		struct accept_filter_arg afa;
156
157	36480	memset(&afa, 0, sizeof afa);
158	36480	bprintf(afa.af_name, "%s", "httpready");
159	36480	retval = setsockopt(sock, SOL_SOCKET, SO_ACCEPTFILTER,
160		&afa, sizeof afa);
161	36480	return (retval);
162		}
163
164		#elif defined(__linux)
165
166		int
167		VTCP_filter_http(int sock)
168		{
169		int retval;
170		int defer = 1;
171
172		retval = setsockopt(sock, SOL_TCP, TCP_DEFER_ACCEPT,
173		&defer, sizeof defer);
174		return (retval);
175		}
176
177		#else
178
179		int
180		VTCP_filter_http(int sock)
181		{
182		errno = EOPNOTSUPP;
183		(void)sock;
184		return (-1);
185		}
186
187		#endif
188
189		/--------------------------------------------------------------------/
190
191
192		int
193	0	VTCP_fastopen(int sock, int depth)
194		{
195		#ifdef HAVE_TCP_FASTOPEN
196		# ifndef SOL_TCP
197		# define SOL_TCP IPPROTO_TCP
198		# endif
199	0	return (setsockopt(sock, SOL_TCP, TCP_FASTOPEN, &depth, sizeof depth));
200		#else
201		errno = EOPNOTSUPP;
202		(void)sock;
203		(void)depth;
204		return (-1);
205		#endif
206		}
207
208
209		/*--------------------------------------------------------------------
210		* Functions for controlling NONBLOCK mode.
211		*
212		* We use FIONBIO because it is cheaper than fcntl(2), which requires
213		* us to do two syscalls, one to get and one to set, the latter of
214		* which mucks about a bit before it ends up calling ioctl(FIONBIO),
215		* at least on FreeBSD.
216		* On Solaris ioctl(FIONBIO) can fail with connection related errnos,
217		* but as long as that is how they fail, we're fine.
218		*/
219
220		void
221	557642	VTCP_blocking(int sock)
222		{
223		int i, j;
224
225	557642	i = 0;
226	557642	j = ioctl(sock, FIONBIO, &i);
227	557642	VTCP_Assert(j);
228	557642	}
229
230		void
231	197002	VTCP_nonblocking(int sock)
232		{
233		int i, j;
234
235	197002	i = 1;
236	197002	j = ioctl(sock, FIONBIO, &i);
237	197002	VTCP_Assert(j);
238	197002	}
239
240		/*--------------------------------------------------------------------
241		* On TCP a connect(2) can block for a looong time, and we don't want that.
242		* Unfortunately, the SocketWizards back in those days were happy to wait
243		* any amount of time for a connection, so the connect(2) syscall does not
244		* take an argument for patience.
245		*
246		* There is a little used work-around, and we employ it at our peril.
247		*
248		*/
249
250		int
251	172984	VTCP_connected(int s)
252		{
253		int k;
254		socklen_t l;
255
256		/* Find out if we got a connection */
257	172984	l = sizeof k;
258	172984	AZ(getsockopt(s, SOL_SOCKET, SO_ERROR, &k, &l));
259
260		/* An error means no connection established */
261	172984	errno = k;
262	172984	if (k) {
263	960	closefd(&s);
264	960	return (-1);
265		}
266
267	172024	VTCP_blocking(s);
268	172024	return (s);
269	172984	}
270
271		int
272	180320	VTCP_connect(const struct suckaddr *name, int msec)
273		{
274		int s, i;
275		struct pollfd fds[1];
276		const struct sockaddr *sa;
277		socklen_t sl;
278		int val;
279
280	180320	if (name == NULL)
281	1080	return (-1);
282		/* Attempt the connect */
283	179240	AN(VSA_Sane(name));
284	179240	sa = VSA_Get_Sockaddr(name, &sl);
285	179240	AN(sa);
286	179240	AN(sl);
287
288	179240	s = socket(sa->sa_family, SOCK_STREAM, 0);
289	179240	if (s < 0)
290	0	return (s);
291
292		/* Set the socket non-blocking */
293	179240	if (msec != 0)
294	173117	VTCP_nonblocking(s);
295
296	179240	val = 1;
297	179240	AZ(setsockopt(s, IPPROTO_TCP, TCP_NODELAY, &val, sizeof val));
298
299	179240	i = connect(s, sa, sl);
300	179240	if (i == 0)
301	6222	return (s);
302	173018	if (errno != EINPROGRESS) {
303	0	closefd(&s);
304	0	return (-1);
305		}
306
307	173018	if (msec < 0) {
308		/*
309		* Caller is responsible for waiting and
310		* calling VTCP_connected
311		*/
312	37187	return (s);
313		}
314
315	135831	assert(msec > 0);
316		/* Exercise our patience, polling for write */
317	135831	fds[0].fd = s;
318	135831	fds[0].events = POLLWRNORM;
319	135831	fds[0].revents = 0;
320	135831	i = poll(fds, 1, msec);
321
322	135831	if (i == 0) {
323		/* Timeout, close and give up */
324	40	closefd(&s);
325	40	errno = ETIMEDOUT;
326	40	return (-1);
327		}
328
329	135791	return (VTCP_connected(s));
330	180320	}
331
332		/*--------------------------------------------------------------------
333		* When closing a TCP connection, a couple of errno's are legit, we
334		* can't be held responsible for the other end wanting to talk to us.
335		*/
336
337		void
338	252595	VTCP_close(int *s)
339		{
340		int i;
341
342	252595	i = close(*s);
343
344	252595	VTCP_Assert(i);
345	252595	*s = -1;
346	252595	}
347
348		void
349	79960	VTCP_set_read_timeout(int s, vtim_dur seconds)
350		{
351	79960	struct timeval timeout = VTIM_timeval_sock(seconds);
352
353		/*
354		* Solaris bug (present at least in snv_151 and older): If this fails
355		* with EINVAL, the socket is half-closed (SS_CANTSENDMORE) and the
356		* timeout does not get set. Needs to be fixed in Solaris, there is
357		* nothing we can do about this.
358		*/
359	79960	VTCP_Assert(setsockopt(s, SOL_SOCKET, SO_RCVTIMEO,
360		&timeout, sizeof timeout));
361	79960	}
362
363		/*--------------------------------------------------------------------
364		*/
365
366		struct vto_priv {
367		unsigned magic;
368		#define VTO_PRIV_MAGIC 0xca70b0e7
369		int latest_errno;
370		int fd;
371		double timeout;
372		};
373
374		static int v_matchproto_(vss_resolved_f)
375	85796	vtcp_open_callback(void priv, const struct suckaddr sa)
376		{
377		struct vto_priv *vto;
378		int fd;
379
380	85796	CAST_OBJ_NOTNULL(vto, priv, VTO_PRIV_MAGIC);
381
382	85796	errno = 0;
383	85796	fd = VTCP_connect(sa, (int)floor(vto->timeout * 1e3));
384	85796	if (fd >= 0) {
385	85756	vto->fd = fd;
386	85756	vto->latest_errno = 0;
387	85756	return (1);
388		}
389	40	vto->latest_errno = errno;
390	40	return (0);
391	85796	}
392
393		int
394	85786	VTCP_open(const char addr, const char def_port, vtim_dur timeout,
395		const char **errp)
396		{
397		struct vto_priv vto[1];
398
399	85786	AN(errp);
400	85786	assert(timeout >= 0);
401	85786	INIT_OBJ(vto, VTO_PRIV_MAGIC);
402	85786	vto->fd = -1;
403	85786	vto->timeout = timeout;
404
405	85786	if (VSS_resolver(addr, def_port, vtcp_open_callback, vto, errp) < 0)
406	0	return (-1);
407	85786	if (vto->fd < 0)
408	40	*errp = strerror(vto->latest_errno);
409	85786	return (vto->fd);
410	85786	}
411
412		/*--------------------------------------------------------------------
413		* Given a struct suckaddr, open a socket of the appropriate type, and bind
414		* it to the requested address.
415		*
416		* If the address is an IPv6 address, the IPV6_V6ONLY option is set to
417		* avoid conflicts between INADDR_ANY and IN6ADDR_ANY.
418		*/
419
420		int
421	352000	VTCP_bind(const struct suckaddr sa, const char *errp)
422		{
423		int sd, val, e;
424		socklen_t sl;
425		const struct sockaddr *so;
426		int proto;
427
428	352000	if (errp != NULL)
429	197160	*errp = NULL;
430
431	352000	proto = VSA_Get_Proto(sa);
432	352000	sd = socket(proto, SOCK_STREAM, 0);
433	352000	if (sd < 0) {
434	0	if (errp != NULL)
435	0	*errp = "socket(2)";
436	0	return (-1);
437		}
438	352000	val = 1;
439	352000	if (setsockopt(sd, SOL_SOCKET, SO_REUSEADDR, &val, sizeof val) != 0) {
440	0	if (errp != NULL)
441	0	*errp = "setsockopt(SO_REUSEADDR, 1)";
442	0	e = errno;
443	0	closefd(&sd);
444	0	errno = e;
445	0	return (-1);
446		}
447		#ifdef IPV6_V6ONLY
448		/* forcibly use separate sockets for IPv4 and IPv6 */
449	352000	val = 1;
450	352000	if (proto == AF_INET6 &&
451	80120	setsockopt(sd, IPPROTO_IPV6, IPV6_V6ONLY, &val, sizeof val) != 0) {
452	0	if (errp != NULL)
453	0	*errp = "setsockopt(IPV6_V6ONLY, 1)";
454	0	e = errno;
455	0	closefd(&sd);
456	0	errno = e;
457	0	return (-1);
458		}
459		#endif
460	352000	so = VSA_Get_Sockaddr(sa, &sl);
461	352000	if (bind(sd, so, sl) != 0) {
462	40	if (errp != NULL)
463	0	*errp = "bind(2)";
464	40	e = errno;
465	40	closefd(&sd);
466	40	errno = e;
467	40	return (-1);
468		}
469	351960	return (sd);
470	352000	}
471
472		/*--------------------------------------------------------------------
473		* Given a struct suckaddr, open a socket of the appropriate type, bind it
474		* to the requested address, and start listening.
475		*/
476
477		int
478	197160	VTCP_listen(const struct suckaddr sa, int depth, const char *errp)
479		{
480		int sd;
481		int e;
482
483	197160	if (errp != NULL)
484	197160	*errp = NULL;
485	197160	sd = VTCP_bind(sa, errp);
486	197160	if (sd >= 0) {
487	197160	if (listen(sd, depth) != 0) {
488	0	e = errno;
489	0	closefd(&sd);
490	0	errno = e;
491	0	if (errp != NULL)
492	0	*errp = "listen(2)";
493	0	return (-1);
494		}
495	197160	}
496	197160	return (sd);
497	197160	}
498
499		/--------------------------------------------------------------------/