xref: /freebsd/lib/libfetch/common.c (revision 5e53a4f90f82c4345f277dd87cc9292f26e04a29)
1842a95ccSDag-Erling Smørgrav /*-
2*5e53a4f9SPedro F. Giffuni  * SPDX-License-Identifier: BSD-3-Clause
3*5e53a4f9SPedro F. Giffuni  *
4792ef1aeSDag-Erling Smørgrav  * Copyright (c) 1998-2016 Dag-Erling Smørgrav
5dcd47379SDag-Erling Smørgrav  * Copyright (c) 2013 Michael Gmelin <freebsd@grem.de>
6842a95ccSDag-Erling Smørgrav  * All rights reserved.
7842a95ccSDag-Erling Smørgrav  *
8842a95ccSDag-Erling Smørgrav  * Redistribution and use in source and binary forms, with or without
9842a95ccSDag-Erling Smørgrav  * modification, are permitted provided that the following conditions
10842a95ccSDag-Erling Smørgrav  * are met:
11842a95ccSDag-Erling Smørgrav  * 1. Redistributions of source code must retain the above copyright
12842a95ccSDag-Erling Smørgrav  *    notice, this list of conditions and the following disclaimer
13842a95ccSDag-Erling Smørgrav  *    in this position and unchanged.
14842a95ccSDag-Erling Smørgrav  * 2. Redistributions in binary form must reproduce the above copyright
15842a95ccSDag-Erling Smørgrav  *    notice, this list of conditions and the following disclaimer in the
16842a95ccSDag-Erling Smørgrav  *    documentation and/or other materials provided with the distribution.
17842a95ccSDag-Erling Smørgrav  * 3. The name of the author may not be used to endorse or promote products
18842a95ccSDag-Erling Smørgrav  *    derived from this software without specific prior written permission
19842a95ccSDag-Erling Smørgrav  *
20842a95ccSDag-Erling Smørgrav  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
21842a95ccSDag-Erling Smørgrav  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
22842a95ccSDag-Erling Smørgrav  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
23842a95ccSDag-Erling Smørgrav  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
24842a95ccSDag-Erling Smørgrav  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
25842a95ccSDag-Erling Smørgrav  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26842a95ccSDag-Erling Smørgrav  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27842a95ccSDag-Erling Smørgrav  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28842a95ccSDag-Erling Smørgrav  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
29842a95ccSDag-Erling Smørgrav  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30842a95ccSDag-Erling Smørgrav  */
31842a95ccSDag-Erling Smørgrav 
32cecb889fSMatthew Dillon #include <sys/cdefs.h>
33cecb889fSMatthew Dillon __FBSDID("$FreeBSD$");
34cecb889fSMatthew Dillon 
350fba3a00SDag-Erling Smørgrav #include <sys/param.h>
36842a95ccSDag-Erling Smørgrav #include <sys/socket.h>
37fc6e9e65SDag-Erling Smørgrav #include <sys/time.h>
38a1763027SDag-Erling Smørgrav #include <sys/uio.h>
3962a2681cSDag-Erling Smørgrav 
40842a95ccSDag-Erling Smørgrav #include <netinet/in.h>
41842a95ccSDag-Erling Smørgrav 
4262a2681cSDag-Erling Smørgrav #include <ctype.h>
43842a95ccSDag-Erling Smørgrav #include <errno.h>
44caaffed8SDag-Erling Smørgrav #include <fcntl.h>
45842a95ccSDag-Erling Smørgrav #include <netdb.h>
46215a27f1SDag-Erling Smørgrav #include <poll.h>
4707350d12SDag-Erling Smørgrav #include <pwd.h>
48ba101983SDag-Erling Smørgrav #include <stdarg.h>
490fba3a00SDag-Erling Smørgrav #include <stdlib.h>
500fba3a00SDag-Erling Smørgrav #include <stdio.h>
51842a95ccSDag-Erling Smørgrav #include <string.h>
52842a95ccSDag-Erling Smørgrav #include <unistd.h>
53842a95ccSDag-Erling Smørgrav 
54dcd47379SDag-Erling Smørgrav #ifdef WITH_SSL
55dcd47379SDag-Erling Smørgrav #include <openssl/x509v3.h>
56dcd47379SDag-Erling Smørgrav #endif
57dcd47379SDag-Erling Smørgrav 
58842a95ccSDag-Erling Smørgrav #include "fetch.h"
59842a95ccSDag-Erling Smørgrav #include "common.h"
60842a95ccSDag-Erling Smørgrav 
61d8acd8dcSDag-Erling Smørgrav 
62842a95ccSDag-Erling Smørgrav /*** Local data **************************************************************/
63842a95ccSDag-Erling Smørgrav 
64842a95ccSDag-Erling Smørgrav /*
65842a95ccSDag-Erling Smørgrav  * Error messages for resolver errors
66842a95ccSDag-Erling Smørgrav  */
67a1b37df2SDag-Erling Smørgrav static struct fetcherr netdb_errlist[] = {
683d82ba43SHajimu UMEMOTO #ifdef EAI_NODATA
6928c645cfSHajimu UMEMOTO 	{ EAI_NODATA,	FETCH_RESOLV,	"Host not found" },
703d82ba43SHajimu UMEMOTO #endif
7128c645cfSHajimu UMEMOTO 	{ EAI_AGAIN,	FETCH_TEMP,	"Transient resolver failure" },
7228c645cfSHajimu UMEMOTO 	{ EAI_FAIL,	FETCH_RESOLV,	"Non-recoverable resolver failure" },
7328c645cfSHajimu UMEMOTO 	{ EAI_NONAME,	FETCH_RESOLV,	"No address record" },
74d8acd8dcSDag-Erling Smørgrav 	{ -1,		FETCH_UNKNOWN,	"Unknown resolver error" }
75842a95ccSDag-Erling Smørgrav };
76842a95ccSDag-Erling Smørgrav 
77a1763027SDag-Erling Smørgrav /* End-of-Line */
7838c7e4a6SArchie Cobbs static const char ENDL[2] = "\r\n";
79a1763027SDag-Erling Smørgrav 
80842a95ccSDag-Erling Smørgrav 
81842a95ccSDag-Erling Smørgrav /*** Error-reporting functions ***********************************************/
82842a95ccSDag-Erling Smørgrav 
83842a95ccSDag-Erling Smørgrav /*
84842a95ccSDag-Erling Smørgrav  * Map error code to string
85842a95ccSDag-Erling Smørgrav  */
86ba101983SDag-Erling Smørgrav static struct fetcherr *
87a1b37df2SDag-Erling Smørgrav fetch_finderr(struct fetcherr *p, int e)
88842a95ccSDag-Erling Smørgrav {
89ba101983SDag-Erling Smørgrav 	while (p->num != -1 && p->num != e)
90ba101983SDag-Erling Smørgrav 		p++;
91e19e6098SDag-Erling Smørgrav 	return (p);
92842a95ccSDag-Erling Smørgrav }
93842a95ccSDag-Erling Smørgrav 
94842a95ccSDag-Erling Smørgrav /*
95842a95ccSDag-Erling Smørgrav  * Set error code
96842a95ccSDag-Erling Smørgrav  */
97842a95ccSDag-Erling Smørgrav void
98a1b37df2SDag-Erling Smørgrav fetch_seterr(struct fetcherr *p, int e)
99842a95ccSDag-Erling Smørgrav {
100a1b37df2SDag-Erling Smørgrav 	p = fetch_finderr(p, e);
101ba101983SDag-Erling Smørgrav 	fetchLastErrCode = p->cat;
102ba101983SDag-Erling Smørgrav 	snprintf(fetchLastErrString, MAXERRSTRING, "%s", p->string);
103842a95ccSDag-Erling Smørgrav }
104842a95ccSDag-Erling Smørgrav 
105842a95ccSDag-Erling Smørgrav /*
106842a95ccSDag-Erling Smørgrav  * Set error code according to errno
107842a95ccSDag-Erling Smørgrav  */
108842a95ccSDag-Erling Smørgrav void
109a1b37df2SDag-Erling Smørgrav fetch_syserr(void)
110842a95ccSDag-Erling Smørgrav {
111d8acd8dcSDag-Erling Smørgrav 	switch (errno) {
112d8acd8dcSDag-Erling Smørgrav 	case 0:
1130fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_OK;
114d8acd8dcSDag-Erling Smørgrav 		break;
115d8acd8dcSDag-Erling Smørgrav 	case EPERM:
116d8acd8dcSDag-Erling Smørgrav 	case EACCES:
117d8acd8dcSDag-Erling Smørgrav 	case EROFS:
118d8acd8dcSDag-Erling Smørgrav 	case EAUTH:
119d8acd8dcSDag-Erling Smørgrav 	case ENEEDAUTH:
1200fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_AUTH;
121d8acd8dcSDag-Erling Smørgrav 		break;
122d8acd8dcSDag-Erling Smørgrav 	case ENOENT:
123d8acd8dcSDag-Erling Smørgrav 	case EISDIR: /* XXX */
1240fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_UNAVAIL;
125d8acd8dcSDag-Erling Smørgrav 		break;
126d8acd8dcSDag-Erling Smørgrav 	case ENOMEM:
1270fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_MEMORY;
128d8acd8dcSDag-Erling Smørgrav 		break;
129d8acd8dcSDag-Erling Smørgrav 	case EBUSY:
130d8acd8dcSDag-Erling Smørgrav 	case EAGAIN:
1310fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_TEMP;
132d8acd8dcSDag-Erling Smørgrav 		break;
133d8acd8dcSDag-Erling Smørgrav 	case EEXIST:
1340fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_EXISTS;
135d8acd8dcSDag-Erling Smørgrav 		break;
136d8acd8dcSDag-Erling Smørgrav 	case ENOSPC:
1370fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_FULL;
138d8acd8dcSDag-Erling Smørgrav 		break;
139d8acd8dcSDag-Erling Smørgrav 	case EADDRINUSE:
140d8acd8dcSDag-Erling Smørgrav 	case EADDRNOTAVAIL:
141d8acd8dcSDag-Erling Smørgrav 	case ENETDOWN:
142d8acd8dcSDag-Erling Smørgrav 	case ENETUNREACH:
143d8acd8dcSDag-Erling Smørgrav 	case ENETRESET:
144d8acd8dcSDag-Erling Smørgrav 	case EHOSTUNREACH:
1450fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_NETWORK;
146d8acd8dcSDag-Erling Smørgrav 		break;
147d8acd8dcSDag-Erling Smørgrav 	case ECONNABORTED:
148d8acd8dcSDag-Erling Smørgrav 	case ECONNRESET:
1490fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_ABORT;
150d8acd8dcSDag-Erling Smørgrav 		break;
151d8acd8dcSDag-Erling Smørgrav 	case ETIMEDOUT:
1520fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_TIMEOUT;
153d8acd8dcSDag-Erling Smørgrav 		break;
154d8acd8dcSDag-Erling Smørgrav 	case ECONNREFUSED:
155d8acd8dcSDag-Erling Smørgrav 	case EHOSTDOWN:
1560fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_DOWN;
157d8acd8dcSDag-Erling Smørgrav 		break;
158d8acd8dcSDag-Erling Smørgrav 	default:
1590fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_UNKNOWN;
160d8acd8dcSDag-Erling Smørgrav 	}
161e19e6098SDag-Erling Smørgrav 	snprintf(fetchLastErrString, MAXERRSTRING, "%s", strerror(errno));
1620fba3a00SDag-Erling Smørgrav }
1630fba3a00SDag-Erling Smørgrav 
1640fba3a00SDag-Erling Smørgrav 
1650fba3a00SDag-Erling Smørgrav /*
1660fba3a00SDag-Erling Smørgrav  * Emit status message
1670fba3a00SDag-Erling Smørgrav  */
168ba101983SDag-Erling Smørgrav void
169a1b37df2SDag-Erling Smørgrav fetch_info(const char *fmt, ...)
1700fba3a00SDag-Erling Smørgrav {
1710fba3a00SDag-Erling Smørgrav 	va_list ap;
1720fba3a00SDag-Erling Smørgrav 
1730fba3a00SDag-Erling Smørgrav 	va_start(ap, fmt);
174ba101983SDag-Erling Smørgrav 	vfprintf(stderr, fmt, ap);
1750fba3a00SDag-Erling Smørgrav 	va_end(ap);
176a68f8b58SDag-Erling Smørgrav 	fputc('\n', stderr);
177842a95ccSDag-Erling Smørgrav }
178842a95ccSDag-Erling Smørgrav 
179842a95ccSDag-Erling Smørgrav 
180842a95ccSDag-Erling Smørgrav /*** Network-related utility functions ***************************************/
181842a95ccSDag-Erling Smørgrav 
182842a95ccSDag-Erling Smørgrav /*
183e828ada7SDag-Erling Smørgrav  * Return the default port for a scheme
184e828ada7SDag-Erling Smørgrav  */
185e828ada7SDag-Erling Smørgrav int
186a1b37df2SDag-Erling Smørgrav fetch_default_port(const char *scheme)
187e828ada7SDag-Erling Smørgrav {
188e828ada7SDag-Erling Smørgrav 	struct servent *se;
189e828ada7SDag-Erling Smørgrav 
190e828ada7SDag-Erling Smørgrav 	if ((se = getservbyname(scheme, "tcp")) != NULL)
191e19e6098SDag-Erling Smørgrav 		return (ntohs(se->s_port));
192e828ada7SDag-Erling Smørgrav 	if (strcasecmp(scheme, SCHEME_FTP) == 0)
193e19e6098SDag-Erling Smørgrav 		return (FTP_DEFAULT_PORT);
194e828ada7SDag-Erling Smørgrav 	if (strcasecmp(scheme, SCHEME_HTTP) == 0)
195e19e6098SDag-Erling Smørgrav 		return (HTTP_DEFAULT_PORT);
196e19e6098SDag-Erling Smørgrav 	return (0);
197e828ada7SDag-Erling Smørgrav }
198e828ada7SDag-Erling Smørgrav 
199e828ada7SDag-Erling Smørgrav /*
200e828ada7SDag-Erling Smørgrav  * Return the default proxy port for a scheme
201e828ada7SDag-Erling Smørgrav  */
202e828ada7SDag-Erling Smørgrav int
203a1b37df2SDag-Erling Smørgrav fetch_default_proxy_port(const char *scheme)
204e828ada7SDag-Erling Smørgrav {
205e828ada7SDag-Erling Smørgrav 	if (strcasecmp(scheme, SCHEME_FTP) == 0)
206e19e6098SDag-Erling Smørgrav 		return (FTP_DEFAULT_PROXY_PORT);
207e828ada7SDag-Erling Smørgrav 	if (strcasecmp(scheme, SCHEME_HTTP) == 0)
208e19e6098SDag-Erling Smørgrav 		return (HTTP_DEFAULT_PROXY_PORT);
209e19e6098SDag-Erling Smørgrav 	return (0);
210e828ada7SDag-Erling Smørgrav }
211e828ada7SDag-Erling Smørgrav 
212f606d589SDag-Erling Smørgrav 
213e828ada7SDag-Erling Smørgrav /*
2149601e333SDag-Erling Smørgrav  * Create a connection for an existing descriptor.
2159601e333SDag-Erling Smørgrav  */
2169601e333SDag-Erling Smørgrav conn_t *
217a1b37df2SDag-Erling Smørgrav fetch_reopen(int sd)
2189601e333SDag-Erling Smørgrav {
2199601e333SDag-Erling Smørgrav 	conn_t *conn;
22028fd9307SJilles Tjoelker 	int opt = 1;
2219601e333SDag-Erling Smørgrav 
2229601e333SDag-Erling Smørgrav 	/* allocate and fill connection structure */
223930105c1SDag-Erling Smørgrav 	if ((conn = calloc(1, sizeof(*conn))) == NULL)
2249601e333SDag-Erling Smørgrav 		return (NULL);
22515b68c63SDag-Erling Smørgrav 	fcntl(sd, F_SETFD, FD_CLOEXEC);
22628fd9307SJilles Tjoelker 	setsockopt(sd, SOL_SOCKET, SO_NOSIGPIPE, &opt, sizeof opt);
2279601e333SDag-Erling Smørgrav 	conn->sd = sd;
228f606d589SDag-Erling Smørgrav 	++conn->ref;
229f606d589SDag-Erling Smørgrav 	return (conn);
230f606d589SDag-Erling Smørgrav }
231f606d589SDag-Erling Smørgrav 
232f606d589SDag-Erling Smørgrav 
233f606d589SDag-Erling Smørgrav /*
234f606d589SDag-Erling Smørgrav  * Bump a connection's reference count.
235f606d589SDag-Erling Smørgrav  */
236f606d589SDag-Erling Smørgrav conn_t *
237a1b37df2SDag-Erling Smørgrav fetch_ref(conn_t *conn)
238f606d589SDag-Erling Smørgrav {
239f606d589SDag-Erling Smørgrav 
240f606d589SDag-Erling Smørgrav 	++conn->ref;
2419601e333SDag-Erling Smørgrav 	return (conn);
2429601e333SDag-Erling Smørgrav }
2439601e333SDag-Erling Smørgrav 
2449601e333SDag-Erling Smørgrav 
2459601e333SDag-Erling Smørgrav /*
246792ef1aeSDag-Erling Smørgrav  * Resolve an address
247792ef1aeSDag-Erling Smørgrav  */
248792ef1aeSDag-Erling Smørgrav struct addrinfo *
249792ef1aeSDag-Erling Smørgrav fetch_resolve(const char *addr, int port, int af)
250792ef1aeSDag-Erling Smørgrav {
251792ef1aeSDag-Erling Smørgrav 	char hbuf[256], sbuf[8];
252792ef1aeSDag-Erling Smørgrav 	struct addrinfo hints, *res;
25308a49957SDag-Erling Smørgrav 	const char *hb, *he, *sep;
25408a49957SDag-Erling Smørgrav 	const char *host, *service;
255792ef1aeSDag-Erling Smørgrav 	int err, len;
256792ef1aeSDag-Erling Smørgrav 
25708a49957SDag-Erling Smørgrav 	/* first, check for a bracketed IPv6 address */
25808a49957SDag-Erling Smørgrav 	if (*addr == '[') {
25908a49957SDag-Erling Smørgrav 		hb = addr + 1;
26008a49957SDag-Erling Smørgrav 		if ((sep = strchr(hb, ']')) == NULL) {
26108a49957SDag-Erling Smørgrav 			errno = EINVAL;
26208a49957SDag-Erling Smørgrav 			goto syserr;
26308a49957SDag-Erling Smørgrav 		}
26408a49957SDag-Erling Smørgrav 		he = sep++;
26508a49957SDag-Erling Smørgrav 	} else {
26608a49957SDag-Erling Smørgrav 		hb = addr;
26708a49957SDag-Erling Smørgrav 		sep = strchrnul(hb, ':');
26808a49957SDag-Erling Smørgrav 		he = sep;
26908a49957SDag-Erling Smørgrav 	}
27008a49957SDag-Erling Smørgrav 
27108a49957SDag-Erling Smørgrav 	/* see if we need to copy the host name */
27208a49957SDag-Erling Smørgrav 	if (*he != '\0') {
273792ef1aeSDag-Erling Smørgrav 		len = snprintf(hbuf, sizeof(hbuf),
27408a49957SDag-Erling Smørgrav 		    "%.*s", (int)(he - hb), hb);
275792ef1aeSDag-Erling Smørgrav 		if (len < 0)
27608a49957SDag-Erling Smørgrav 			goto syserr;
277792ef1aeSDag-Erling Smørgrav 		if (len >= (int)sizeof(hbuf)) {
278792ef1aeSDag-Erling Smørgrav 			errno = ENAMETOOLONG;
27908a49957SDag-Erling Smørgrav 			goto syserr;
280792ef1aeSDag-Erling Smørgrav 		}
281792ef1aeSDag-Erling Smørgrav 		host = hbuf;
28208a49957SDag-Erling Smørgrav 	} else {
28308a49957SDag-Erling Smørgrav 		host = hb;
28408a49957SDag-Erling Smørgrav 	}
28508a49957SDag-Erling Smørgrav 
28608a49957SDag-Erling Smørgrav 	/* was it followed by a service name? */
28708a49957SDag-Erling Smørgrav 	if (*sep == '\0' && port != 0) {
288792ef1aeSDag-Erling Smørgrav 		if (port < 1 || port > 65535) {
289792ef1aeSDag-Erling Smørgrav 			errno = EINVAL;
29008a49957SDag-Erling Smørgrav 			goto syserr;
291792ef1aeSDag-Erling Smørgrav 		}
29208a49957SDag-Erling Smørgrav 		if (snprintf(sbuf, sizeof(sbuf), "%d", port) < 0)
29308a49957SDag-Erling Smørgrav 			goto syserr;
294792ef1aeSDag-Erling Smørgrav 		service = sbuf;
29508a49957SDag-Erling Smørgrav 	} else if (*sep != '\0') {
2963954b52fSDag-Erling Smørgrav 		service = sep + 1;
297792ef1aeSDag-Erling Smørgrav 	} else {
298792ef1aeSDag-Erling Smørgrav 		service = NULL;
299792ef1aeSDag-Erling Smørgrav 	}
300792ef1aeSDag-Erling Smørgrav 
301792ef1aeSDag-Erling Smørgrav 	/* resolve */
302792ef1aeSDag-Erling Smørgrav 	memset(&hints, 0, sizeof(hints));
303792ef1aeSDag-Erling Smørgrav 	hints.ai_family = af;
304792ef1aeSDag-Erling Smørgrav 	hints.ai_socktype = SOCK_STREAM;
305792ef1aeSDag-Erling Smørgrav 	hints.ai_flags = AI_ADDRCONFIG;
306792ef1aeSDag-Erling Smørgrav 	if ((err = getaddrinfo(host, service, &hints, &res)) != 0) {
307792ef1aeSDag-Erling Smørgrav 		netdb_seterr(err);
308792ef1aeSDag-Erling Smørgrav 		return (NULL);
309792ef1aeSDag-Erling Smørgrav 	}
310792ef1aeSDag-Erling Smørgrav 	return (res);
31108a49957SDag-Erling Smørgrav syserr:
31208a49957SDag-Erling Smørgrav 	fetch_syserr();
31308a49957SDag-Erling Smørgrav 	return (NULL);
314792ef1aeSDag-Erling Smørgrav }
315792ef1aeSDag-Erling Smørgrav 
316792ef1aeSDag-Erling Smørgrav 
317792ef1aeSDag-Erling Smørgrav 
318792ef1aeSDag-Erling Smørgrav /*
319c42cb9d9SDag-Erling Smørgrav  * Bind a socket to a specific local address
320c42cb9d9SDag-Erling Smørgrav  */
321c42cb9d9SDag-Erling Smørgrav int
322a1b37df2SDag-Erling Smørgrav fetch_bind(int sd, int af, const char *addr)
323c42cb9d9SDag-Erling Smørgrav {
324792ef1aeSDag-Erling Smørgrav 	struct addrinfo *cliai, *ai;
325c42cb9d9SDag-Erling Smørgrav 	int err;
326c42cb9d9SDag-Erling Smørgrav 
327792ef1aeSDag-Erling Smørgrav 	if ((cliai = fetch_resolve(addr, 0, af)) == NULL)
328c42cb9d9SDag-Erling Smørgrav 		return (-1);
329792ef1aeSDag-Erling Smørgrav 	for (ai = cliai; ai != NULL; ai = ai->ai_next)
330792ef1aeSDag-Erling Smørgrav 		if ((err = bind(sd, ai->ai_addr, ai->ai_addrlen)) == 0)
331792ef1aeSDag-Erling Smørgrav 			break;
332792ef1aeSDag-Erling Smørgrav 	if (err != 0)
333792ef1aeSDag-Erling Smørgrav 		fetch_syserr();
334792ef1aeSDag-Erling Smørgrav 	freeaddrinfo(cliai);
335792ef1aeSDag-Erling Smørgrav 	return (err == 0 ? 0 : -1);
336c42cb9d9SDag-Erling Smørgrav }
337c42cb9d9SDag-Erling Smørgrav 
338c42cb9d9SDag-Erling Smørgrav 
339c42cb9d9SDag-Erling Smørgrav /*
340842a95ccSDag-Erling Smørgrav  * Establish a TCP connection to the specified port on the specified host.
341842a95ccSDag-Erling Smørgrav  */
342dea29ca1SDag-Erling Smørgrav conn_t *
343a1b37df2SDag-Erling Smørgrav fetch_connect(const char *host, int port, int af, int verbose)
344842a95ccSDag-Erling Smørgrav {
345792ef1aeSDag-Erling Smørgrav 	struct addrinfo *cais = NULL, *sais = NULL, *cai, *sai;
346c42cb9d9SDag-Erling Smørgrav 	const char *bindaddr;
347792ef1aeSDag-Erling Smørgrav 	conn_t *conn = NULL;
348792ef1aeSDag-Erling Smørgrav 	int err = 0, sd = -1;
349842a95ccSDag-Erling Smørgrav 
350f67efa37SDag-Erling Smørgrav 	DEBUG(fprintf(stderr, "---> %s:%d\n", host, port));
351842a95ccSDag-Erling Smørgrav 
352792ef1aeSDag-Erling Smørgrav 	/* resolve server address */
3530fba3a00SDag-Erling Smørgrav 	if (verbose)
354792ef1aeSDag-Erling Smørgrav 		fetch_info("resolving server address: %s:%d", host, port);
355792ef1aeSDag-Erling Smørgrav 	if ((sais = fetch_resolve(host, port, af)) == NULL)
356792ef1aeSDag-Erling Smørgrav 		goto fail;
3570fba3a00SDag-Erling Smørgrav 
358792ef1aeSDag-Erling Smørgrav 	/* resolve client address */
359c42cb9d9SDag-Erling Smørgrav 	bindaddr = getenv("FETCH_BIND_ADDRESS");
360792ef1aeSDag-Erling Smørgrav 	if (bindaddr != NULL && *bindaddr != '\0') {
3610fba3a00SDag-Erling Smørgrav 		if (verbose)
362792ef1aeSDag-Erling Smørgrav 			fetch_info("resolving client address: %s", bindaddr);
363792ef1aeSDag-Erling Smørgrav 		if ((cais = fetch_resolve(bindaddr, 0, af)) == NULL)
364792ef1aeSDag-Erling Smørgrav 			goto fail;
365c42cb9d9SDag-Erling Smørgrav 	}
366792ef1aeSDag-Erling Smørgrav 
367792ef1aeSDag-Erling Smørgrav 	/* try each server address in turn */
368792ef1aeSDag-Erling Smørgrav 	for (err = 0, sai = sais; sai != NULL; sai = sai->ai_next) {
369792ef1aeSDag-Erling Smørgrav 		/* open socket */
370792ef1aeSDag-Erling Smørgrav 		if ((sd = socket(sai->ai_family, SOCK_STREAM, 0)) < 0)
371792ef1aeSDag-Erling Smørgrav 			goto syserr;
372792ef1aeSDag-Erling Smørgrav 		/* attempt to bind to client address */
373792ef1aeSDag-Erling Smørgrav 		for (err = 0, cai = cais; cai != NULL; cai = cai->ai_next) {
374792ef1aeSDag-Erling Smørgrav 			if (cai->ai_family != sai->ai_family)
375792ef1aeSDag-Erling Smørgrav 				continue;
376792ef1aeSDag-Erling Smørgrav 			if ((err = bind(sd, cai->ai_addr, cai->ai_addrlen)) == 0)
37728c645cfSHajimu UMEMOTO 				break;
37828c645cfSHajimu UMEMOTO 		}
379792ef1aeSDag-Erling Smørgrav 		if (err != 0) {
380792ef1aeSDag-Erling Smørgrav 			if (verbose)
381792ef1aeSDag-Erling Smørgrav 				fetch_info("failed to bind to %s", bindaddr);
382792ef1aeSDag-Erling Smørgrav 			goto syserr;
383792ef1aeSDag-Erling Smørgrav 		}
384792ef1aeSDag-Erling Smørgrav 		/* attempt to connect to server address */
385792ef1aeSDag-Erling Smørgrav 		if ((err = connect(sd, sai->ai_addr, sai->ai_addrlen)) == 0)
386792ef1aeSDag-Erling Smørgrav 			break;
387792ef1aeSDag-Erling Smørgrav 		/* clean up before next attempt */
388792ef1aeSDag-Erling Smørgrav 		close(sd);
389792ef1aeSDag-Erling Smørgrav 		sd = -1;
390792ef1aeSDag-Erling Smørgrav 	}
391792ef1aeSDag-Erling Smørgrav 	if (err != 0) {
392792ef1aeSDag-Erling Smørgrav 		if (verbose)
3936d916040SBjoern A. Zeeb 			fetch_info("failed to connect to %s:%d", host, port);
394792ef1aeSDag-Erling Smørgrav 		goto syserr;
395842a95ccSDag-Erling Smørgrav 	}
396842a95ccSDag-Erling Smørgrav 
397792ef1aeSDag-Erling Smørgrav 	if ((conn = fetch_reopen(sd)) == NULL)
398792ef1aeSDag-Erling Smørgrav 		goto syserr;
399792ef1aeSDag-Erling Smørgrav 	if (cais != NULL)
400792ef1aeSDag-Erling Smørgrav 		freeaddrinfo(cais);
401792ef1aeSDag-Erling Smørgrav 	if (sais != NULL)
402792ef1aeSDag-Erling Smørgrav 		freeaddrinfo(sais);
403dea29ca1SDag-Erling Smørgrav 	return (conn);
404792ef1aeSDag-Erling Smørgrav syserr:
405792ef1aeSDag-Erling Smørgrav 	fetch_syserr();
406792ef1aeSDag-Erling Smørgrav 	goto fail;
407792ef1aeSDag-Erling Smørgrav fail:
408792ef1aeSDag-Erling Smørgrav 	if (sd >= 0)
409792ef1aeSDag-Erling Smørgrav 		close(sd);
410792ef1aeSDag-Erling Smørgrav 	if (cais != NULL)
411792ef1aeSDag-Erling Smørgrav 		freeaddrinfo(cais);
412792ef1aeSDag-Erling Smørgrav 	if (sais != NULL)
413792ef1aeSDag-Erling Smørgrav 		freeaddrinfo(sais);
414792ef1aeSDag-Erling Smørgrav 	return (NULL);
415842a95ccSDag-Erling Smørgrav }
416ce71b736SDag-Erling Smørgrav 
417dcd47379SDag-Erling Smørgrav #ifdef WITH_SSL
418dcd47379SDag-Erling Smørgrav /*
419dcd47379SDag-Erling Smørgrav  * Convert characters A-Z to lowercase (intentionally avoid any locale
420dcd47379SDag-Erling Smørgrav  * specific conversions).
421dcd47379SDag-Erling Smørgrav  */
422dcd47379SDag-Erling Smørgrav static char
423dcd47379SDag-Erling Smørgrav fetch_ssl_tolower(char in)
424dcd47379SDag-Erling Smørgrav {
425dcd47379SDag-Erling Smørgrav 	if (in >= 'A' && in <= 'Z')
426dcd47379SDag-Erling Smørgrav 		return (in + 32);
427dcd47379SDag-Erling Smørgrav 	else
428dcd47379SDag-Erling Smørgrav 		return (in);
429dcd47379SDag-Erling Smørgrav }
430dcd47379SDag-Erling Smørgrav 
431dcd47379SDag-Erling Smørgrav /*
432dcd47379SDag-Erling Smørgrav  * isalpha implementation that intentionally avoids any locale specific
433dcd47379SDag-Erling Smørgrav  * conversions.
434dcd47379SDag-Erling Smørgrav  */
435dcd47379SDag-Erling Smørgrav static int
436dcd47379SDag-Erling Smørgrav fetch_ssl_isalpha(char in)
437dcd47379SDag-Erling Smørgrav {
438dcd47379SDag-Erling Smørgrav 	return ((in >= 'A' && in <= 'Z') || (in >= 'a' && in <= 'z'));
439dcd47379SDag-Erling Smørgrav }
440dcd47379SDag-Erling Smørgrav 
441dcd47379SDag-Erling Smørgrav /*
442dcd47379SDag-Erling Smørgrav  * Check if passed hostnames a and b are equal.
443dcd47379SDag-Erling Smørgrav  */
444dcd47379SDag-Erling Smørgrav static int
445dcd47379SDag-Erling Smørgrav fetch_ssl_hname_equal(const char *a, size_t alen, const char *b,
446dcd47379SDag-Erling Smørgrav     size_t blen)
447dcd47379SDag-Erling Smørgrav {
448dcd47379SDag-Erling Smørgrav 	size_t i;
449dcd47379SDag-Erling Smørgrav 
450dcd47379SDag-Erling Smørgrav 	if (alen != blen)
451dcd47379SDag-Erling Smørgrav 		return (0);
452dcd47379SDag-Erling Smørgrav 	for (i = 0; i < alen; ++i) {
453dcd47379SDag-Erling Smørgrav 		if (fetch_ssl_tolower(a[i]) != fetch_ssl_tolower(b[i]))
454dcd47379SDag-Erling Smørgrav 			return (0);
455dcd47379SDag-Erling Smørgrav 	}
456dcd47379SDag-Erling Smørgrav 	return (1);
457dcd47379SDag-Erling Smørgrav }
458dcd47379SDag-Erling Smørgrav 
459dcd47379SDag-Erling Smørgrav /*
460dcd47379SDag-Erling Smørgrav  * Check if domain label is traditional, meaning that only A-Z, a-z, 0-9
461dcd47379SDag-Erling Smørgrav  * and '-' (hyphen) are allowed. Hyphens have to be surrounded by alpha-
462dcd47379SDag-Erling Smørgrav  * numeric characters. Double hyphens (like they're found in IDN a-labels
463dcd47379SDag-Erling Smørgrav  * 'xn--') are not allowed. Empty labels are invalid.
464dcd47379SDag-Erling Smørgrav  */
465dcd47379SDag-Erling Smørgrav static int
466dcd47379SDag-Erling Smørgrav fetch_ssl_is_trad_domain_label(const char *l, size_t len, int wcok)
467dcd47379SDag-Erling Smørgrav {
468dcd47379SDag-Erling Smørgrav 	size_t i;
469dcd47379SDag-Erling Smørgrav 
470dcd47379SDag-Erling Smørgrav 	if (!len || l[0] == '-' || l[len-1] == '-')
471dcd47379SDag-Erling Smørgrav 		return (0);
472dcd47379SDag-Erling Smørgrav 	for (i = 0; i < len; ++i) {
473dcd47379SDag-Erling Smørgrav 		if (!isdigit(l[i]) &&
474dcd47379SDag-Erling Smørgrav 		    !fetch_ssl_isalpha(l[i]) &&
475dcd47379SDag-Erling Smørgrav 		    !(l[i] == '*' && wcok) &&
476dcd47379SDag-Erling Smørgrav 		    !(l[i] == '-' && l[i - 1] != '-'))
477dcd47379SDag-Erling Smørgrav 			return (0);
478dcd47379SDag-Erling Smørgrav 	}
479dcd47379SDag-Erling Smørgrav 	return (1);
480dcd47379SDag-Erling Smørgrav }
481dcd47379SDag-Erling Smørgrav 
482dcd47379SDag-Erling Smørgrav /*
483dcd47379SDag-Erling Smørgrav  * Check if host name consists only of numbers. This might indicate an IP
484dcd47379SDag-Erling Smørgrav  * address, which is not a good idea for CN wildcard comparison.
485dcd47379SDag-Erling Smørgrav  */
486dcd47379SDag-Erling Smørgrav static int
487dcd47379SDag-Erling Smørgrav fetch_ssl_hname_is_only_numbers(const char *hostname, size_t len)
488dcd47379SDag-Erling Smørgrav {
489dcd47379SDag-Erling Smørgrav 	size_t i;
490dcd47379SDag-Erling Smørgrav 
491dcd47379SDag-Erling Smørgrav 	for (i = 0; i < len; ++i) {
492dcd47379SDag-Erling Smørgrav 		if (!((hostname[i] >= '0' && hostname[i] <= '9') ||
493dcd47379SDag-Erling Smørgrav 		    hostname[i] == '.'))
494dcd47379SDag-Erling Smørgrav 			return (0);
495dcd47379SDag-Erling Smørgrav 	}
496dcd47379SDag-Erling Smørgrav 	return (1);
497dcd47379SDag-Erling Smørgrav }
498dcd47379SDag-Erling Smørgrav 
499dcd47379SDag-Erling Smørgrav /*
500dcd47379SDag-Erling Smørgrav  * Check if the host name h passed matches the pattern passed in m which
501dcd47379SDag-Erling Smørgrav  * is usually part of subjectAltName or CN of a certificate presented to
502dcd47379SDag-Erling Smørgrav  * the client. This includes wildcard matching. The algorithm is based on
503dcd47379SDag-Erling Smørgrav  * RFC6125, sections 6.4.3 and 7.2, which clarifies RFC2818 and RFC3280.
504dcd47379SDag-Erling Smørgrav  */
505dcd47379SDag-Erling Smørgrav static int
506dcd47379SDag-Erling Smørgrav fetch_ssl_hname_match(const char *h, size_t hlen, const char *m,
507dcd47379SDag-Erling Smørgrav     size_t mlen)
508dcd47379SDag-Erling Smørgrav {
509dcd47379SDag-Erling Smørgrav 	int delta, hdotidx, mdot1idx, wcidx;
510dcd47379SDag-Erling Smørgrav 	const char *hdot, *mdot1, *mdot2;
511dcd47379SDag-Erling Smørgrav 	const char *wc; /* wildcard */
512dcd47379SDag-Erling Smørgrav 
513dcd47379SDag-Erling Smørgrav 	if (!(h && *h && m && *m))
514dcd47379SDag-Erling Smørgrav 		return (0);
515dcd47379SDag-Erling Smørgrav 	if ((wc = strnstr(m, "*", mlen)) == NULL)
516dcd47379SDag-Erling Smørgrav 		return (fetch_ssl_hname_equal(h, hlen, m, mlen));
517dcd47379SDag-Erling Smørgrav 	wcidx = wc - m;
518dcd47379SDag-Erling Smørgrav 	/* hostname should not be just dots and numbers */
519dcd47379SDag-Erling Smørgrav 	if (fetch_ssl_hname_is_only_numbers(h, hlen))
520dcd47379SDag-Erling Smørgrav 		return (0);
521dcd47379SDag-Erling Smørgrav 	/* only one wildcard allowed in pattern */
522dcd47379SDag-Erling Smørgrav 	if (strnstr(wc + 1, "*", mlen - wcidx - 1) != NULL)
523dcd47379SDag-Erling Smørgrav 		return (0);
524dcd47379SDag-Erling Smørgrav 	/*
525dcd47379SDag-Erling Smørgrav 	 * there must be at least two more domain labels and
526dcd47379SDag-Erling Smørgrav 	 * wildcard has to be in the leftmost label (RFC6125)
527dcd47379SDag-Erling Smørgrav 	 */
528dcd47379SDag-Erling Smørgrav 	mdot1 = strnstr(m, ".", mlen);
529dcd47379SDag-Erling Smørgrav 	if (mdot1 == NULL || mdot1 < wc || (mlen - (mdot1 - m)) < 4)
530dcd47379SDag-Erling Smørgrav 		return (0);
531dcd47379SDag-Erling Smørgrav 	mdot1idx = mdot1 - m;
532dcd47379SDag-Erling Smørgrav 	mdot2 = strnstr(mdot1 + 1, ".", mlen - mdot1idx - 1);
533dcd47379SDag-Erling Smørgrav 	if (mdot2 == NULL || (mlen - (mdot2 - m)) < 2)
534dcd47379SDag-Erling Smørgrav 		return (0);
535dcd47379SDag-Erling Smørgrav 	/* hostname must contain a dot and not be the 1st char */
536dcd47379SDag-Erling Smørgrav 	hdot = strnstr(h, ".", hlen);
537dcd47379SDag-Erling Smørgrav 	if (hdot == NULL || hdot == h)
538dcd47379SDag-Erling Smørgrav 		return (0);
539dcd47379SDag-Erling Smørgrav 	hdotidx = hdot - h;
540dcd47379SDag-Erling Smørgrav 	/*
541dcd47379SDag-Erling Smørgrav 	 * host part of hostname must be at least as long as
542dcd47379SDag-Erling Smørgrav 	 * pattern it's supposed to match
543dcd47379SDag-Erling Smørgrav 	 */
544dcd47379SDag-Erling Smørgrav 	if (hdotidx < mdot1idx)
545dcd47379SDag-Erling Smørgrav 		return (0);
546dcd47379SDag-Erling Smørgrav 	/*
547dcd47379SDag-Erling Smørgrav 	 * don't allow wildcards in non-traditional domain names
548dcd47379SDag-Erling Smørgrav 	 * (IDN, A-label, U-label...)
549dcd47379SDag-Erling Smørgrav 	 */
550dcd47379SDag-Erling Smørgrav 	if (!fetch_ssl_is_trad_domain_label(h, hdotidx, 0) ||
551dcd47379SDag-Erling Smørgrav 	    !fetch_ssl_is_trad_domain_label(m, mdot1idx, 1))
552dcd47379SDag-Erling Smørgrav 		return (0);
553dcd47379SDag-Erling Smørgrav 	/* match domain part (part after first dot) */
554dcd47379SDag-Erling Smørgrav 	if (!fetch_ssl_hname_equal(hdot, hlen - hdotidx, mdot1,
555dcd47379SDag-Erling Smørgrav 	    mlen - mdot1idx))
556dcd47379SDag-Erling Smørgrav 		return (0);
557dcd47379SDag-Erling Smørgrav 	/* match part left of wildcard */
558dcd47379SDag-Erling Smørgrav 	if (!fetch_ssl_hname_equal(h, wcidx, m, wcidx))
559dcd47379SDag-Erling Smørgrav 		return (0);
560dcd47379SDag-Erling Smørgrav 	/* match part right of wildcard */
561dcd47379SDag-Erling Smørgrav 	delta = mdot1idx - wcidx - 1;
562dcd47379SDag-Erling Smørgrav 	if (!fetch_ssl_hname_equal(hdot - delta, delta,
563dcd47379SDag-Erling Smørgrav 	    mdot1 - delta, delta))
564dcd47379SDag-Erling Smørgrav 		return (0);
56575f46cf6SPedro F. Giffuni 	/* all tests succeeded, it's a match */
566dcd47379SDag-Erling Smørgrav 	return (1);
567dcd47379SDag-Erling Smørgrav }
568dcd47379SDag-Erling Smørgrav 
569dcd47379SDag-Erling Smørgrav /*
570dcd47379SDag-Erling Smørgrav  * Get numeric host address info - returns NULL if host was not an IP
571dcd47379SDag-Erling Smørgrav  * address. The caller is responsible for deallocation using
572dcd47379SDag-Erling Smørgrav  * freeaddrinfo(3).
573dcd47379SDag-Erling Smørgrav  */
574dcd47379SDag-Erling Smørgrav static struct addrinfo *
575dcd47379SDag-Erling Smørgrav fetch_ssl_get_numeric_addrinfo(const char *hostname, size_t len)
576dcd47379SDag-Erling Smørgrav {
577dcd47379SDag-Erling Smørgrav 	struct addrinfo hints, *res;
578dcd47379SDag-Erling Smørgrav 	char *host;
579dcd47379SDag-Erling Smørgrav 
580dcd47379SDag-Erling Smørgrav 	host = (char *)malloc(len + 1);
581dcd47379SDag-Erling Smørgrav 	memcpy(host, hostname, len);
582dcd47379SDag-Erling Smørgrav 	host[len] = '\0';
583dcd47379SDag-Erling Smørgrav 	memset(&hints, 0, sizeof(hints));
584dcd47379SDag-Erling Smørgrav 	hints.ai_family = PF_UNSPEC;
585dcd47379SDag-Erling Smørgrav 	hints.ai_socktype = SOCK_STREAM;
586dcd47379SDag-Erling Smørgrav 	hints.ai_protocol = 0;
587dcd47379SDag-Erling Smørgrav 	hints.ai_flags = AI_NUMERICHOST;
588dcd47379SDag-Erling Smørgrav 	/* port is not relevant for this purpose */
589034fb271SMichael Gmelin 	if (getaddrinfo(host, "443", &hints, &res) != 0)
590034fb271SMichael Gmelin 		res = NULL;
591dcd47379SDag-Erling Smørgrav 	free(host);
592dcd47379SDag-Erling Smørgrav 	return res;
593dcd47379SDag-Erling Smørgrav }
594dcd47379SDag-Erling Smørgrav 
595dcd47379SDag-Erling Smørgrav /*
596dcd47379SDag-Erling Smørgrav  * Compare ip address in addrinfo with address passes.
597dcd47379SDag-Erling Smørgrav  */
598dcd47379SDag-Erling Smørgrav static int
599dcd47379SDag-Erling Smørgrav fetch_ssl_ipaddr_match_bin(const struct addrinfo *lhost, const char *rhost,
600dcd47379SDag-Erling Smørgrav     size_t rhostlen)
601dcd47379SDag-Erling Smørgrav {
602dcd47379SDag-Erling Smørgrav 	const void *left;
603dcd47379SDag-Erling Smørgrav 
604dcd47379SDag-Erling Smørgrav 	if (lhost->ai_family == AF_INET && rhostlen == 4) {
605dcd47379SDag-Erling Smørgrav 		left = (void *)&((struct sockaddr_in*)(void *)
606dcd47379SDag-Erling Smørgrav 		    lhost->ai_addr)->sin_addr.s_addr;
607dcd47379SDag-Erling Smørgrav #ifdef INET6
608dcd47379SDag-Erling Smørgrav 	} else if (lhost->ai_family == AF_INET6 && rhostlen == 16) {
609dcd47379SDag-Erling Smørgrav 		left = (void *)&((struct sockaddr_in6 *)(void *)
610dcd47379SDag-Erling Smørgrav 		    lhost->ai_addr)->sin6_addr;
611dcd47379SDag-Erling Smørgrav #endif
612dcd47379SDag-Erling Smørgrav 	} else
613dcd47379SDag-Erling Smørgrav 		return (0);
614dcd47379SDag-Erling Smørgrav 	return (!memcmp(left, (const void *)rhost, rhostlen) ? 1 : 0);
615dcd47379SDag-Erling Smørgrav }
616dcd47379SDag-Erling Smørgrav 
617dcd47379SDag-Erling Smørgrav /*
618dcd47379SDag-Erling Smørgrav  * Compare ip address in addrinfo with host passed. If host is not an IP
619dcd47379SDag-Erling Smørgrav  * address, comparison will fail.
620dcd47379SDag-Erling Smørgrav  */
621dcd47379SDag-Erling Smørgrav static int
622dcd47379SDag-Erling Smørgrav fetch_ssl_ipaddr_match(const struct addrinfo *laddr, const char *r,
623dcd47379SDag-Erling Smørgrav     size_t rlen)
624dcd47379SDag-Erling Smørgrav {
625dcd47379SDag-Erling Smørgrav 	struct addrinfo *raddr;
626dcd47379SDag-Erling Smørgrav 	int ret;
627dcd47379SDag-Erling Smørgrav 	char *rip;
628dcd47379SDag-Erling Smørgrav 
629dcd47379SDag-Erling Smørgrav 	ret = 0;
630dcd47379SDag-Erling Smørgrav 	if ((raddr = fetch_ssl_get_numeric_addrinfo(r, rlen)) == NULL)
631dcd47379SDag-Erling Smørgrav 		return 0; /* not a numeric host */
632dcd47379SDag-Erling Smørgrav 
633dcd47379SDag-Erling Smørgrav 	if (laddr->ai_family == raddr->ai_family) {
634dcd47379SDag-Erling Smørgrav 		if (laddr->ai_family == AF_INET) {
635dcd47379SDag-Erling Smørgrav 			rip = (char *)&((struct sockaddr_in *)(void *)
636dcd47379SDag-Erling Smørgrav 			    raddr->ai_addr)->sin_addr.s_addr;
637dcd47379SDag-Erling Smørgrav 			ret = fetch_ssl_ipaddr_match_bin(laddr, rip, 4);
638dcd47379SDag-Erling Smørgrav #ifdef INET6
639dcd47379SDag-Erling Smørgrav 		} else if (laddr->ai_family == AF_INET6) {
640dcd47379SDag-Erling Smørgrav 			rip = (char *)&((struct sockaddr_in6 *)(void *)
641dcd47379SDag-Erling Smørgrav 			    raddr->ai_addr)->sin6_addr;
642dcd47379SDag-Erling Smørgrav 			ret = fetch_ssl_ipaddr_match_bin(laddr, rip, 16);
643dcd47379SDag-Erling Smørgrav #endif
644dcd47379SDag-Erling Smørgrav 		}
645dcd47379SDag-Erling Smørgrav 
646dcd47379SDag-Erling Smørgrav 	}
647dcd47379SDag-Erling Smørgrav 	freeaddrinfo(raddr);
648dcd47379SDag-Erling Smørgrav 	return (ret);
649dcd47379SDag-Erling Smørgrav }
650dcd47379SDag-Erling Smørgrav 
651dcd47379SDag-Erling Smørgrav /*
652dcd47379SDag-Erling Smørgrav  * Verify server certificate by subjectAltName.
653dcd47379SDag-Erling Smørgrav  */
654dcd47379SDag-Erling Smørgrav static int
655dcd47379SDag-Erling Smørgrav fetch_ssl_verify_altname(STACK_OF(GENERAL_NAME) *altnames,
656dcd47379SDag-Erling Smørgrav     const char *host, struct addrinfo *ip)
657dcd47379SDag-Erling Smørgrav {
658dcd47379SDag-Erling Smørgrav 	const GENERAL_NAME *name;
659dcd47379SDag-Erling Smørgrav 	size_t nslen;
660dcd47379SDag-Erling Smørgrav 	int i;
661dcd47379SDag-Erling Smørgrav 	const char *ns;
662dcd47379SDag-Erling Smørgrav 
663dcd47379SDag-Erling Smørgrav 	for (i = 0; i < sk_GENERAL_NAME_num(altnames); ++i) {
664dcd47379SDag-Erling Smørgrav #if OPENSSL_VERSION_NUMBER < 0x10000000L
665dcd47379SDag-Erling Smørgrav 		/*
666dcd47379SDag-Erling Smørgrav 		 * This is a workaround, since the following line causes
667dcd47379SDag-Erling Smørgrav 		 * alignment issues in clang:
668dcd47379SDag-Erling Smørgrav 		 * name = sk_GENERAL_NAME_value(altnames, i);
669dcd47379SDag-Erling Smørgrav 		 * OpenSSL explicitly warns not to use those macros
670dcd47379SDag-Erling Smørgrav 		 * directly, but there isn't much choice (and there
671dcd47379SDag-Erling Smørgrav 		 * shouldn't be any ill side effects)
672dcd47379SDag-Erling Smørgrav 		 */
673dcd47379SDag-Erling Smørgrav 		name = (GENERAL_NAME *)SKM_sk_value(void, altnames, i);
674dcd47379SDag-Erling Smørgrav #else
675dcd47379SDag-Erling Smørgrav 		name = sk_GENERAL_NAME_value(altnames, i);
676dcd47379SDag-Erling Smørgrav #endif
677dcd47379SDag-Erling Smørgrav 		ns = (const char *)ASN1_STRING_data(name->d.ia5);
678dcd47379SDag-Erling Smørgrav 		nslen = (size_t)ASN1_STRING_length(name->d.ia5);
679dcd47379SDag-Erling Smørgrav 
680dcd47379SDag-Erling Smørgrav 		if (name->type == GEN_DNS && ip == NULL &&
681dcd47379SDag-Erling Smørgrav 		    fetch_ssl_hname_match(host, strlen(host), ns, nslen))
682dcd47379SDag-Erling Smørgrav 			return (1);
683dcd47379SDag-Erling Smørgrav 		else if (name->type == GEN_IPADD && ip != NULL &&
684dcd47379SDag-Erling Smørgrav 		    fetch_ssl_ipaddr_match_bin(ip, ns, nslen))
685dcd47379SDag-Erling Smørgrav 			return (1);
686dcd47379SDag-Erling Smørgrav 	}
687dcd47379SDag-Erling Smørgrav 	return (0);
688dcd47379SDag-Erling Smørgrav }
689dcd47379SDag-Erling Smørgrav 
690dcd47379SDag-Erling Smørgrav /*
691dcd47379SDag-Erling Smørgrav  * Verify server certificate by CN.
692dcd47379SDag-Erling Smørgrav  */
693dcd47379SDag-Erling Smørgrav static int
694dcd47379SDag-Erling Smørgrav fetch_ssl_verify_cn(X509_NAME *subject, const char *host,
695dcd47379SDag-Erling Smørgrav     struct addrinfo *ip)
696dcd47379SDag-Erling Smørgrav {
697dcd47379SDag-Erling Smørgrav 	ASN1_STRING *namedata;
698dcd47379SDag-Erling Smørgrav 	X509_NAME_ENTRY *nameentry;
699dcd47379SDag-Erling Smørgrav 	int cnlen, lastpos, loc, ret;
700dcd47379SDag-Erling Smørgrav 	unsigned char *cn;
701dcd47379SDag-Erling Smørgrav 
702dcd47379SDag-Erling Smørgrav 	ret = 0;
703dcd47379SDag-Erling Smørgrav 	lastpos = -1;
704dcd47379SDag-Erling Smørgrav 	loc = -1;
705dcd47379SDag-Erling Smørgrav 	cn = NULL;
706dcd47379SDag-Erling Smørgrav 	/* get most specific CN (last entry in list) and compare */
707dcd47379SDag-Erling Smørgrav 	while ((lastpos = X509_NAME_get_index_by_NID(subject,
708dcd47379SDag-Erling Smørgrav 	    NID_commonName, lastpos)) != -1)
709dcd47379SDag-Erling Smørgrav 		loc = lastpos;
710dcd47379SDag-Erling Smørgrav 
711dcd47379SDag-Erling Smørgrav 	if (loc > -1) {
712dcd47379SDag-Erling Smørgrav 		nameentry = X509_NAME_get_entry(subject, loc);
713dcd47379SDag-Erling Smørgrav 		namedata = X509_NAME_ENTRY_get_data(nameentry);
714dcd47379SDag-Erling Smørgrav 		cnlen = ASN1_STRING_to_UTF8(&cn, namedata);
715dcd47379SDag-Erling Smørgrav 		if (ip == NULL &&
716dcd47379SDag-Erling Smørgrav 		    fetch_ssl_hname_match(host, strlen(host), cn, cnlen))
717dcd47379SDag-Erling Smørgrav 			ret = 1;
718dcd47379SDag-Erling Smørgrav 		else if (ip != NULL && fetch_ssl_ipaddr_match(ip, cn, cnlen))
719dcd47379SDag-Erling Smørgrav 			ret = 1;
720dcd47379SDag-Erling Smørgrav 		OPENSSL_free(cn);
721dcd47379SDag-Erling Smørgrav 	}
722dcd47379SDag-Erling Smørgrav 	return (ret);
723dcd47379SDag-Erling Smørgrav }
724dcd47379SDag-Erling Smørgrav 
725dcd47379SDag-Erling Smørgrav /*
726dcd47379SDag-Erling Smørgrav  * Verify that server certificate subjectAltName/CN matches
727dcd47379SDag-Erling Smørgrav  * hostname. First check, if there are alternative subject names. If yes,
728dcd47379SDag-Erling Smørgrav  * those have to match. Only if those don't exist it falls back to
729dcd47379SDag-Erling Smørgrav  * checking the subject's CN.
730dcd47379SDag-Erling Smørgrav  */
731dcd47379SDag-Erling Smørgrav static int
732dcd47379SDag-Erling Smørgrav fetch_ssl_verify_hname(X509 *cert, const char *host)
733dcd47379SDag-Erling Smørgrav {
734dcd47379SDag-Erling Smørgrav 	struct addrinfo *ip;
735dcd47379SDag-Erling Smørgrav 	STACK_OF(GENERAL_NAME) *altnames;
736dcd47379SDag-Erling Smørgrav 	X509_NAME *subject;
737dcd47379SDag-Erling Smørgrav 	int ret;
738dcd47379SDag-Erling Smørgrav 
739dcd47379SDag-Erling Smørgrav 	ret = 0;
740dcd47379SDag-Erling Smørgrav 	ip = fetch_ssl_get_numeric_addrinfo(host, strlen(host));
741dcd47379SDag-Erling Smørgrav 	altnames = X509_get_ext_d2i(cert, NID_subject_alt_name,
742dcd47379SDag-Erling Smørgrav 	    NULL, NULL);
743dcd47379SDag-Erling Smørgrav 
744dcd47379SDag-Erling Smørgrav 	if (altnames != NULL) {
745dcd47379SDag-Erling Smørgrav 		ret = fetch_ssl_verify_altname(altnames, host, ip);
746dcd47379SDag-Erling Smørgrav 	} else {
747dcd47379SDag-Erling Smørgrav 		subject = X509_get_subject_name(cert);
748dcd47379SDag-Erling Smørgrav 		if (subject != NULL)
749dcd47379SDag-Erling Smørgrav 			ret = fetch_ssl_verify_cn(subject, host, ip);
750dcd47379SDag-Erling Smørgrav 	}
751dcd47379SDag-Erling Smørgrav 
752dcd47379SDag-Erling Smørgrav 	if (ip != NULL)
753dcd47379SDag-Erling Smørgrav 		freeaddrinfo(ip);
754dcd47379SDag-Erling Smørgrav 	if (altnames != NULL)
755dcd47379SDag-Erling Smørgrav 		GENERAL_NAMES_free(altnames);
756dcd47379SDag-Erling Smørgrav 	return (ret);
757dcd47379SDag-Erling Smørgrav }
758dcd47379SDag-Erling Smørgrav 
759dcd47379SDag-Erling Smørgrav /*
760dcd47379SDag-Erling Smørgrav  * Configure transport security layer based on environment.
761dcd47379SDag-Erling Smørgrav  */
762dcd47379SDag-Erling Smørgrav static void
763dcd47379SDag-Erling Smørgrav fetch_ssl_setup_transport_layer(SSL_CTX *ctx, int verbose)
764dcd47379SDag-Erling Smørgrav {
765dcd47379SDag-Erling Smørgrav 	long ssl_ctx_options;
766dcd47379SDag-Erling Smørgrav 
76701ed3ca3SJung-uk Kim 	ssl_ctx_options = SSL_OP_ALL | SSL_OP_NO_SSLv2 | SSL_OP_NO_TICKET;
76876720765SDag-Erling Smørgrav 	if (getenv("SSL_ALLOW_SSL3") == NULL)
769dcd47379SDag-Erling Smørgrav 		ssl_ctx_options |= SSL_OP_NO_SSLv3;
770dcd47379SDag-Erling Smørgrav 	if (getenv("SSL_NO_TLS1") != NULL)
771dcd47379SDag-Erling Smørgrav 		ssl_ctx_options |= SSL_OP_NO_TLSv1;
77276720765SDag-Erling Smørgrav 	if (getenv("SSL_NO_TLS1_1") != NULL)
77376720765SDag-Erling Smørgrav 		ssl_ctx_options |= SSL_OP_NO_TLSv1_1;
77476720765SDag-Erling Smørgrav 	if (getenv("SSL_NO_TLS1_2") != NULL)
77576720765SDag-Erling Smørgrav 		ssl_ctx_options |= SSL_OP_NO_TLSv1_2;
776dcd47379SDag-Erling Smørgrav 	if (verbose)
7773443b2bcSDag-Erling Smørgrav 		fetch_info("SSL options: %lx", ssl_ctx_options);
778dcd47379SDag-Erling Smørgrav 	SSL_CTX_set_options(ctx, ssl_ctx_options);
779dcd47379SDag-Erling Smørgrav }
780dcd47379SDag-Erling Smørgrav 
781dcd47379SDag-Erling Smørgrav 
782dcd47379SDag-Erling Smørgrav /*
783dcd47379SDag-Erling Smørgrav  * Configure peer verification based on environment.
784dcd47379SDag-Erling Smørgrav  */
785e9de92adSDag-Erling Smørgrav #define LOCAL_CERT_FILE	"/usr/local/etc/ssl/cert.pem"
786e9de92adSDag-Erling Smørgrav #define BASE_CERT_FILE	"/etc/ssl/cert.pem"
787dcd47379SDag-Erling Smørgrav static int
788dcd47379SDag-Erling Smørgrav fetch_ssl_setup_peer_verification(SSL_CTX *ctx, int verbose)
789dcd47379SDag-Erling Smørgrav {
790dcd47379SDag-Erling Smørgrav 	X509_LOOKUP *crl_lookup;
791dcd47379SDag-Erling Smørgrav 	X509_STORE *crl_store;
792dcd47379SDag-Erling Smørgrav 	const char *ca_cert_file, *ca_cert_path, *crl_file;
793dcd47379SDag-Erling Smørgrav 
794dcd47379SDag-Erling Smørgrav 	if (getenv("SSL_NO_VERIFY_PEER") == NULL) {
795e9de92adSDag-Erling Smørgrav 		ca_cert_file = getenv("SSL_CA_CERT_FILE");
796e9de92adSDag-Erling Smørgrav 		if (ca_cert_file == NULL &&
797e9de92adSDag-Erling Smørgrav 		    access(LOCAL_CERT_FILE, R_OK) == 0)
798e9de92adSDag-Erling Smørgrav 			ca_cert_file = LOCAL_CERT_FILE;
799f16550adSBaptiste Daroussin 		if (ca_cert_file == NULL &&
800f16550adSBaptiste Daroussin 		    access(BASE_CERT_FILE, R_OK) == 0)
801e9de92adSDag-Erling Smørgrav 			ca_cert_file = BASE_CERT_FILE;
802dcd47379SDag-Erling Smørgrav 		ca_cert_path = getenv("SSL_CA_CERT_PATH");
803dcd47379SDag-Erling Smørgrav 		if (verbose) {
804dcd47379SDag-Erling Smørgrav 			fetch_info("Peer verification enabled");
805dcd47379SDag-Erling Smørgrav 			if (ca_cert_file != NULL)
806dcd47379SDag-Erling Smørgrav 				fetch_info("Using CA cert file: %s",
807dcd47379SDag-Erling Smørgrav 				    ca_cert_file);
808dcd47379SDag-Erling Smørgrav 			if (ca_cert_path != NULL)
809dcd47379SDag-Erling Smørgrav 				fetch_info("Using CA cert path: %s",
810dcd47379SDag-Erling Smørgrav 				    ca_cert_path);
811f16550adSBaptiste Daroussin 			if (ca_cert_file == NULL && ca_cert_path == NULL)
812f16550adSBaptiste Daroussin 				fetch_info("Using OpenSSL default "
813f16550adSBaptiste Daroussin 				    "CA cert file and path");
814dcd47379SDag-Erling Smørgrav 		}
815dcd47379SDag-Erling Smørgrav 		SSL_CTX_set_verify(ctx, SSL_VERIFY_PEER,
816dcd47379SDag-Erling Smørgrav 		    fetch_ssl_cb_verify_crt);
817f16550adSBaptiste Daroussin 		if (ca_cert_file != NULL || ca_cert_path != NULL)
818dcd47379SDag-Erling Smørgrav 			SSL_CTX_load_verify_locations(ctx, ca_cert_file,
819dcd47379SDag-Erling Smørgrav 			    ca_cert_path);
820f16550adSBaptiste Daroussin 		else
821f16550adSBaptiste Daroussin 			SSL_CTX_set_default_verify_paths(ctx);
822dcd47379SDag-Erling Smørgrav 		if ((crl_file = getenv("SSL_CRL_FILE")) != NULL) {
823dcd47379SDag-Erling Smørgrav 			if (verbose)
824dcd47379SDag-Erling Smørgrav 				fetch_info("Using CRL file: %s", crl_file);
825dcd47379SDag-Erling Smørgrav 			crl_store = SSL_CTX_get_cert_store(ctx);
826dcd47379SDag-Erling Smørgrav 			crl_lookup = X509_STORE_add_lookup(crl_store,
827dcd47379SDag-Erling Smørgrav 			    X509_LOOKUP_file());
828dcd47379SDag-Erling Smørgrav 			if (crl_lookup == NULL ||
829dcd47379SDag-Erling Smørgrav 			    !X509_load_crl_file(crl_lookup, crl_file,
830dcd47379SDag-Erling Smørgrav 				X509_FILETYPE_PEM)) {
831dcd47379SDag-Erling Smørgrav 				fprintf(stderr,
832dcd47379SDag-Erling Smørgrav 				    "Could not load CRL file %s\n",
833dcd47379SDag-Erling Smørgrav 				    crl_file);
834dcd47379SDag-Erling Smørgrav 				return (0);
835dcd47379SDag-Erling Smørgrav 			}
836dcd47379SDag-Erling Smørgrav 			X509_STORE_set_flags(crl_store,
837dcd47379SDag-Erling Smørgrav 			    X509_V_FLAG_CRL_CHECK |
838dcd47379SDag-Erling Smørgrav 			    X509_V_FLAG_CRL_CHECK_ALL);
839dcd47379SDag-Erling Smørgrav 		}
840dcd47379SDag-Erling Smørgrav 	}
841dcd47379SDag-Erling Smørgrav 	return (1);
842dcd47379SDag-Erling Smørgrav }
843dcd47379SDag-Erling Smørgrav 
844dcd47379SDag-Erling Smørgrav /*
845dcd47379SDag-Erling Smørgrav  * Configure client certificate based on environment.
846dcd47379SDag-Erling Smørgrav  */
847dcd47379SDag-Erling Smørgrav static int
848dcd47379SDag-Erling Smørgrav fetch_ssl_setup_client_certificate(SSL_CTX *ctx, int verbose)
849dcd47379SDag-Erling Smørgrav {
850dcd47379SDag-Erling Smørgrav 	const char *client_cert_file, *client_key_file;
851dcd47379SDag-Erling Smørgrav 
852dcd47379SDag-Erling Smørgrav 	if ((client_cert_file = getenv("SSL_CLIENT_CERT_FILE")) != NULL) {
853dcd47379SDag-Erling Smørgrav 		client_key_file = getenv("SSL_CLIENT_KEY_FILE") != NULL ?
854dcd47379SDag-Erling Smørgrav 		    getenv("SSL_CLIENT_KEY_FILE") : client_cert_file;
855dcd47379SDag-Erling Smørgrav 		if (verbose) {
856dcd47379SDag-Erling Smørgrav 			fetch_info("Using client cert file: %s",
857dcd47379SDag-Erling Smørgrav 			    client_cert_file);
858dcd47379SDag-Erling Smørgrav 			fetch_info("Using client key file: %s",
859dcd47379SDag-Erling Smørgrav 			    client_key_file);
860dcd47379SDag-Erling Smørgrav 		}
861dcd47379SDag-Erling Smørgrav 		if (SSL_CTX_use_certificate_chain_file(ctx,
862dcd47379SDag-Erling Smørgrav 			client_cert_file) != 1) {
863dcd47379SDag-Erling Smørgrav 			fprintf(stderr,
864dcd47379SDag-Erling Smørgrav 			    "Could not load client certificate %s\n",
865dcd47379SDag-Erling Smørgrav 			    client_cert_file);
866dcd47379SDag-Erling Smørgrav 			return (0);
867dcd47379SDag-Erling Smørgrav 		}
868dcd47379SDag-Erling Smørgrav 		if (SSL_CTX_use_PrivateKey_file(ctx, client_key_file,
869dcd47379SDag-Erling Smørgrav 			SSL_FILETYPE_PEM) != 1) {
870dcd47379SDag-Erling Smørgrav 			fprintf(stderr,
871dcd47379SDag-Erling Smørgrav 			    "Could not load client key %s\n",
872dcd47379SDag-Erling Smørgrav 			    client_key_file);
873dcd47379SDag-Erling Smørgrav 			return (0);
874dcd47379SDag-Erling Smørgrav 		}
875dcd47379SDag-Erling Smørgrav 	}
876dcd47379SDag-Erling Smørgrav 	return (1);
877dcd47379SDag-Erling Smørgrav }
878dcd47379SDag-Erling Smørgrav 
879dcd47379SDag-Erling Smørgrav /*
880dcd47379SDag-Erling Smørgrav  * Callback for SSL certificate verification, this is called on server
881dcd47379SDag-Erling Smørgrav  * cert verification. It takes no decision, but informs the user in case
882dcd47379SDag-Erling Smørgrav  * verification failed.
883dcd47379SDag-Erling Smørgrav  */
884dcd47379SDag-Erling Smørgrav int
885dcd47379SDag-Erling Smørgrav fetch_ssl_cb_verify_crt(int verified, X509_STORE_CTX *ctx)
886dcd47379SDag-Erling Smørgrav {
887dcd47379SDag-Erling Smørgrav 	X509 *crt;
888dcd47379SDag-Erling Smørgrav 	X509_NAME *name;
889dcd47379SDag-Erling Smørgrav 	char *str;
890dcd47379SDag-Erling Smørgrav 
891dcd47379SDag-Erling Smørgrav 	str = NULL;
892dcd47379SDag-Erling Smørgrav 	if (!verified) {
893dcd47379SDag-Erling Smørgrav 		if ((crt = X509_STORE_CTX_get_current_cert(ctx)) != NULL &&
894dcd47379SDag-Erling Smørgrav 		    (name = X509_get_subject_name(crt)) != NULL)
895dcd47379SDag-Erling Smørgrav 			str = X509_NAME_oneline(name, 0, 0);
896dcd47379SDag-Erling Smørgrav 		fprintf(stderr, "Certificate verification failed for %s\n",
897dcd47379SDag-Erling Smørgrav 		    str != NULL ? str : "no relevant certificate");
898dcd47379SDag-Erling Smørgrav 		OPENSSL_free(str);
899dcd47379SDag-Erling Smørgrav 	}
900dcd47379SDag-Erling Smørgrav 	return (verified);
901dcd47379SDag-Erling Smørgrav }
902dcd47379SDag-Erling Smørgrav 
903dcd47379SDag-Erling Smørgrav #endif
904ce71b736SDag-Erling Smørgrav 
905fc6e9e65SDag-Erling Smørgrav /*
906111e2510SDag-Erling Smørgrav  * Enable SSL on a connection.
907111e2510SDag-Erling Smørgrav  */
908111e2510SDag-Erling Smørgrav int
909dcd47379SDag-Erling Smørgrav fetch_ssl(conn_t *conn, const struct url *URL, int verbose)
910111e2510SDag-Erling Smørgrav {
911a9d0c849SEd Maste #ifdef WITH_SSL
912caaffed8SDag-Erling Smørgrav 	int ret, ssl_err;
913dcd47379SDag-Erling Smørgrav 	X509_NAME *name;
914dcd47379SDag-Erling Smørgrav 	char *str;
915111e2510SDag-Erling Smørgrav 
916111e2510SDag-Erling Smørgrav 	/* Init the SSL library and context */
917111e2510SDag-Erling Smørgrav 	if (!SSL_library_init()){
918111e2510SDag-Erling Smørgrav 		fprintf(stderr, "SSL library init failed\n");
919111e2510SDag-Erling Smørgrav 		return (-1);
920111e2510SDag-Erling Smørgrav 	}
921111e2510SDag-Erling Smørgrav 
922111e2510SDag-Erling Smørgrav 	SSL_load_error_strings();
923111e2510SDag-Erling Smørgrav 
92476720765SDag-Erling Smørgrav 	conn->ssl_meth = SSLv23_client_method();
925111e2510SDag-Erling Smørgrav 	conn->ssl_ctx = SSL_CTX_new(conn->ssl_meth);
926f8020ddeSDag-Erling Smørgrav 	SSL_CTX_set_mode(conn->ssl_ctx, SSL_MODE_AUTO_RETRY);
927111e2510SDag-Erling Smørgrav 
928dcd47379SDag-Erling Smørgrav 	fetch_ssl_setup_transport_layer(conn->ssl_ctx, verbose);
929dcd47379SDag-Erling Smørgrav 	if (!fetch_ssl_setup_peer_verification(conn->ssl_ctx, verbose))
930dcd47379SDag-Erling Smørgrav 		return (-1);
931dcd47379SDag-Erling Smørgrav 	if (!fetch_ssl_setup_client_certificate(conn->ssl_ctx, verbose))
932dcd47379SDag-Erling Smørgrav 		return (-1);
933dcd47379SDag-Erling Smørgrav 
934111e2510SDag-Erling Smørgrav 	conn->ssl = SSL_new(conn->ssl_ctx);
935111e2510SDag-Erling Smørgrav 	if (conn->ssl == NULL) {
936111e2510SDag-Erling Smørgrav 		fprintf(stderr, "SSL context creation failed\n");
937111e2510SDag-Erling Smørgrav 		return (-1);
938111e2510SDag-Erling Smørgrav 	}
939111e2510SDag-Erling Smørgrav 	SSL_set_fd(conn->ssl, conn->sd);
9404cb272a0SBryan Drewery 
9414cb272a0SBryan Drewery #if OPENSSL_VERSION_NUMBER >= 0x0090806fL && !defined(OPENSSL_NO_TLSEXT)
9425313966eSBryan Drewery 	if (!SSL_set_tlsext_host_name(conn->ssl,
9435313966eSBryan Drewery 	    __DECONST(struct url *, URL)->host)) {
9444cb272a0SBryan Drewery 		fprintf(stderr,
9454cb272a0SBryan Drewery 		    "TLS server name indication extension failed for host %s\n",
9464cb272a0SBryan Drewery 		    URL->host);
9474cb272a0SBryan Drewery 		return (-1);
9484cb272a0SBryan Drewery 	}
9494cb272a0SBryan Drewery #endif
950caaffed8SDag-Erling Smørgrav 	while ((ret = SSL_connect(conn->ssl)) == -1) {
951caaffed8SDag-Erling Smørgrav 		ssl_err = SSL_get_error(conn->ssl, ret);
952caaffed8SDag-Erling Smørgrav 		if (ssl_err != SSL_ERROR_WANT_READ &&
953caaffed8SDag-Erling Smørgrav 		    ssl_err != SSL_ERROR_WANT_WRITE) {
954111e2510SDag-Erling Smørgrav 			ERR_print_errors_fp(stderr);
955111e2510SDag-Erling Smørgrav 			return (-1);
956111e2510SDag-Erling Smørgrav 		}
957caaffed8SDag-Erling Smørgrav 	}
958dcd47379SDag-Erling Smørgrav 	conn->ssl_cert = SSL_get_peer_certificate(conn->ssl);
959dcd47379SDag-Erling Smørgrav 
960dcd47379SDag-Erling Smørgrav 	if (conn->ssl_cert == NULL) {
961dcd47379SDag-Erling Smørgrav 		fprintf(stderr, "No server SSL certificate\n");
962dcd47379SDag-Erling Smørgrav 		return (-1);
963dcd47379SDag-Erling Smørgrav 	}
964dcd47379SDag-Erling Smørgrav 
965dcd47379SDag-Erling Smørgrav 	if (getenv("SSL_NO_VERIFY_HOSTNAME") == NULL) {
966dcd47379SDag-Erling Smørgrav 		if (verbose)
967dcd47379SDag-Erling Smørgrav 			fetch_info("Verify hostname");
968dcd47379SDag-Erling Smørgrav 		if (!fetch_ssl_verify_hname(conn->ssl_cert, URL->host)) {
969dcd47379SDag-Erling Smørgrav 			fprintf(stderr,
970dcd47379SDag-Erling Smørgrav 			    "SSL certificate subject doesn't match host %s\n",
971dcd47379SDag-Erling Smørgrav 			    URL->host);
972dcd47379SDag-Erling Smørgrav 			return (-1);
973dcd47379SDag-Erling Smørgrav 		}
974dcd47379SDag-Erling Smørgrav 	}
975111e2510SDag-Erling Smørgrav 
976111e2510SDag-Erling Smørgrav 	if (verbose) {
97776720765SDag-Erling Smørgrav 		fetch_info("%s connection established using %s",
97876720765SDag-Erling Smørgrav 		    SSL_get_version(conn->ssl), SSL_get_cipher(conn->ssl));
979111e2510SDag-Erling Smørgrav 		name = X509_get_subject_name(conn->ssl_cert);
980111e2510SDag-Erling Smørgrav 		str = X509_NAME_oneline(name, 0, 0);
981dcd47379SDag-Erling Smørgrav 		fetch_info("Certificate subject: %s", str);
982dcd47379SDag-Erling Smørgrav 		OPENSSL_free(str);
983111e2510SDag-Erling Smørgrav 		name = X509_get_issuer_name(conn->ssl_cert);
984111e2510SDag-Erling Smørgrav 		str = X509_NAME_oneline(name, 0, 0);
985dcd47379SDag-Erling Smørgrav 		fetch_info("Certificate issuer: %s", str);
986dcd47379SDag-Erling Smørgrav 		OPENSSL_free(str);
987111e2510SDag-Erling Smørgrav 	}
988111e2510SDag-Erling Smørgrav 
989111e2510SDag-Erling Smørgrav 	return (0);
9903070f6cbSDag-Erling Smørgrav #else
9913070f6cbSDag-Erling Smørgrav 	(void)conn;
9923070f6cbSDag-Erling Smørgrav 	(void)verbose;
9933070f6cbSDag-Erling Smørgrav 	fprintf(stderr, "SSL support disabled\n");
9943070f6cbSDag-Erling Smørgrav 	return (-1);
9953070f6cbSDag-Erling Smørgrav #endif
996111e2510SDag-Erling Smørgrav }
997111e2510SDag-Erling Smørgrav 
998caaffed8SDag-Erling Smørgrav #define FETCH_READ_WAIT		-2
999caaffed8SDag-Erling Smørgrav #define FETCH_READ_ERROR	-1
1000caaffed8SDag-Erling Smørgrav #define FETCH_READ_DONE		 0
1001caaffed8SDag-Erling Smørgrav 
1002caaffed8SDag-Erling Smørgrav #ifdef WITH_SSL
1003caaffed8SDag-Erling Smørgrav static ssize_t
1004caaffed8SDag-Erling Smørgrav fetch_ssl_read(SSL *ssl, char *buf, size_t len)
1005caaffed8SDag-Erling Smørgrav {
1006caaffed8SDag-Erling Smørgrav 	ssize_t rlen;
1007caaffed8SDag-Erling Smørgrav 	int ssl_err;
1008caaffed8SDag-Erling Smørgrav 
1009caaffed8SDag-Erling Smørgrav 	rlen = SSL_read(ssl, buf, len);
1010caaffed8SDag-Erling Smørgrav 	if (rlen < 0) {
1011caaffed8SDag-Erling Smørgrav 		ssl_err = SSL_get_error(ssl, rlen);
1012caaffed8SDag-Erling Smørgrav 		if (ssl_err == SSL_ERROR_WANT_READ ||
1013caaffed8SDag-Erling Smørgrav 		    ssl_err == SSL_ERROR_WANT_WRITE) {
1014caaffed8SDag-Erling Smørgrav 			return (FETCH_READ_WAIT);
1015caaffed8SDag-Erling Smørgrav 		} else {
1016caaffed8SDag-Erling Smørgrav 			ERR_print_errors_fp(stderr);
1017caaffed8SDag-Erling Smørgrav 			return (FETCH_READ_ERROR);
1018caaffed8SDag-Erling Smørgrav 		}
1019caaffed8SDag-Erling Smørgrav 	}
1020caaffed8SDag-Erling Smørgrav 	return (rlen);
1021caaffed8SDag-Erling Smørgrav }
1022caaffed8SDag-Erling Smørgrav #endif
1023caaffed8SDag-Erling Smørgrav 
1024caaffed8SDag-Erling Smørgrav static ssize_t
1025caaffed8SDag-Erling Smørgrav fetch_socket_read(int sd, char *buf, size_t len)
1026caaffed8SDag-Erling Smørgrav {
1027caaffed8SDag-Erling Smørgrav 	ssize_t rlen;
1028caaffed8SDag-Erling Smørgrav 
1029caaffed8SDag-Erling Smørgrav 	rlen = read(sd, buf, len);
1030caaffed8SDag-Erling Smørgrav 	if (rlen < 0) {
1031caaffed8SDag-Erling Smørgrav 		if (errno == EAGAIN || (errno == EINTR && fetchRestartCalls))
1032caaffed8SDag-Erling Smørgrav 			return (FETCH_READ_WAIT);
1033caaffed8SDag-Erling Smørgrav 		else
1034caaffed8SDag-Erling Smørgrav 			return (FETCH_READ_ERROR);
1035caaffed8SDag-Erling Smørgrav 	}
1036caaffed8SDag-Erling Smørgrav 	return (rlen);
1037caaffed8SDag-Erling Smørgrav }
1038f606d589SDag-Erling Smørgrav 
1039111e2510SDag-Erling Smørgrav /*
10409601e333SDag-Erling Smørgrav  * Read a character from a connection w/ timeout
10419601e333SDag-Erling Smørgrav  */
10429601e333SDag-Erling Smørgrav ssize_t
1043a1b37df2SDag-Erling Smørgrav fetch_read(conn_t *conn, char *buf, size_t len)
10449601e333SDag-Erling Smørgrav {
10455092cf05SDag-Erling Smørgrav 	struct timeval now, timeout, delta;
1046215a27f1SDag-Erling Smørgrav 	struct pollfd pfd;
1047215a27f1SDag-Erling Smørgrav 	ssize_t rlen;
1048215a27f1SDag-Erling Smørgrav 	int deltams;
10499601e333SDag-Erling Smørgrav 
105030787285SDag-Erling Smørgrav 	if (fetchTimeout > 0) {
10519601e333SDag-Erling Smørgrav 		gettimeofday(&timeout, NULL);
10529601e333SDag-Erling Smørgrav 		timeout.tv_sec += fetchTimeout;
10539601e333SDag-Erling Smørgrav 	}
10549601e333SDag-Erling Smørgrav 
1055215a27f1SDag-Erling Smørgrav 	deltams = INFTIM;
1056215a27f1SDag-Erling Smørgrav 	memset(&pfd, 0, sizeof pfd);
1057215a27f1SDag-Erling Smørgrav 	pfd.fd = conn->sd;
1058215a27f1SDag-Erling Smørgrav 	pfd.events = POLLIN | POLLERR;
10592a7daafeSDag-Erling Smørgrav 
1060215a27f1SDag-Erling Smørgrav 	for (;;) {
1061caaffed8SDag-Erling Smørgrav 		/*
1062caaffed8SDag-Erling Smørgrav 		 * The socket is non-blocking.  Instead of the canonical
1063215a27f1SDag-Erling Smørgrav 		 * poll() -> read(), we do the following:
1064caaffed8SDag-Erling Smørgrav 		 *
1065caaffed8SDag-Erling Smørgrav 		 * 1) call read() or SSL_read().
1066215a27f1SDag-Erling Smørgrav 		 * 2) if we received some data, return it.
1067215a27f1SDag-Erling Smørgrav 		 * 3) if an error occurred, return -1.
1068caaffed8SDag-Erling Smørgrav 		 * 4) if read() or SSL_read() signaled EOF, return.
1069caaffed8SDag-Erling Smørgrav 		 * 5) if we did not receive any data but we're not at EOF,
1070215a27f1SDag-Erling Smørgrav 		 *    call poll().
1071caaffed8SDag-Erling Smørgrav 		 *
1072caaffed8SDag-Erling Smørgrav 		 * In the SSL case, this is necessary because if we
1073caaffed8SDag-Erling Smørgrav 		 * receive a close notification, we have to call
1074caaffed8SDag-Erling Smørgrav 		 * SSL_read() one additional time after we've read
1075caaffed8SDag-Erling Smørgrav 		 * everything we received.
1076caaffed8SDag-Erling Smørgrav 		 *
1077caaffed8SDag-Erling Smørgrav 		 * In the non-SSL case, it may improve performance (very
1078caaffed8SDag-Erling Smørgrav 		 * slightly) when reading small amounts of data.
1079caaffed8SDag-Erling Smørgrav 		 */
1080caaffed8SDag-Erling Smørgrav #ifdef WITH_SSL
1081caaffed8SDag-Erling Smørgrav 		if (conn->ssl != NULL)
1082caaffed8SDag-Erling Smørgrav 			rlen = fetch_ssl_read(conn->ssl, buf, len);
1083caaffed8SDag-Erling Smørgrav 		else
1084caaffed8SDag-Erling Smørgrav #endif
1085caaffed8SDag-Erling Smørgrav 			rlen = fetch_socket_read(conn->sd, buf, len);
10869c1ca3a1SDag-Erling Smørgrav 		if (rlen >= 0) {
1087caaffed8SDag-Erling Smørgrav 			break;
1088caaffed8SDag-Erling Smørgrav 		} else if (rlen == FETCH_READ_ERROR) {
10899c1ca3a1SDag-Erling Smørgrav 			fetch_syserr();
1090caaffed8SDag-Erling Smørgrav 			return (-1);
1091caaffed8SDag-Erling Smørgrav 		}
10929c1ca3a1SDag-Erling Smørgrav 		// assert(rlen == FETCH_READ_WAIT);
109330787285SDag-Erling Smørgrav 		if (fetchTimeout > 0) {
10949601e333SDag-Erling Smørgrav 			gettimeofday(&now, NULL);
109530787285SDag-Erling Smørgrav 			if (!timercmp(&timeout, &now, >)) {
1096bb13d0afSDag-Erling Smørgrav 				errno = ETIMEDOUT;
1097a1b37df2SDag-Erling Smørgrav 				fetch_syserr();
1098bb13d0afSDag-Erling Smørgrav 				return (-1);
1099bb13d0afSDag-Erling Smørgrav 			}
110030787285SDag-Erling Smørgrav 			timersub(&timeout, &now, &delta);
1101215a27f1SDag-Erling Smørgrav 			deltams = delta.tv_sec * 1000 +
1102215a27f1SDag-Erling Smørgrav 			    delta.tv_usec / 1000;;
110330787285SDag-Erling Smørgrav 		}
11049601e333SDag-Erling Smørgrav 		errno = 0;
1105215a27f1SDag-Erling Smørgrav 		pfd.revents = 0;
1106215a27f1SDag-Erling Smørgrav 		if (poll(&pfd, 1, deltams) < 0) {
1107215a27f1SDag-Erling Smørgrav 			if (errno == EINTR && fetchRestartCalls)
11089601e333SDag-Erling Smørgrav 				continue;
1109a1b37df2SDag-Erling Smørgrav 			fetch_syserr();
11109601e333SDag-Erling Smørgrav 			return (-1);
11119601e333SDag-Erling Smørgrav 		}
11129601e333SDag-Erling Smørgrav 	}
1113215a27f1SDag-Erling Smørgrav 	return (rlen);
11149601e333SDag-Erling Smørgrav }
11159601e333SDag-Erling Smørgrav 
1116f606d589SDag-Erling Smørgrav 
11179601e333SDag-Erling Smørgrav /*
11189601e333SDag-Erling Smørgrav  * Read a line of text from a connection w/ timeout
1119fc6e9e65SDag-Erling Smørgrav  */
1120fc6e9e65SDag-Erling Smørgrav #define MIN_BUF_SIZE 1024
1121fc6e9e65SDag-Erling Smørgrav 
1122fc6e9e65SDag-Erling Smørgrav int
1123a1b37df2SDag-Erling Smørgrav fetch_getln(conn_t *conn)
1124fc6e9e65SDag-Erling Smørgrav {
11259601e333SDag-Erling Smørgrav 	char *tmp;
11269601e333SDag-Erling Smørgrav 	size_t tmpsize;
1127bb13d0afSDag-Erling Smørgrav 	ssize_t len;
1128fc6e9e65SDag-Erling Smørgrav 	char c;
1129fc6e9e65SDag-Erling Smørgrav 
1130dea29ca1SDag-Erling Smørgrav 	if (conn->buf == NULL) {
1131dea29ca1SDag-Erling Smørgrav 		if ((conn->buf = malloc(MIN_BUF_SIZE)) == NULL) {
1132fc6e9e65SDag-Erling Smørgrav 			errno = ENOMEM;
1133e19e6098SDag-Erling Smørgrav 			return (-1);
1134fc6e9e65SDag-Erling Smørgrav 		}
1135dea29ca1SDag-Erling Smørgrav 		conn->bufsize = MIN_BUF_SIZE;
1136fc6e9e65SDag-Erling Smørgrav 	}
1137fc6e9e65SDag-Erling Smørgrav 
1138dea29ca1SDag-Erling Smørgrav 	conn->buf[0] = '\0';
1139dea29ca1SDag-Erling Smørgrav 	conn->buflen = 0;
1140fc6e9e65SDag-Erling Smørgrav 
1141fc6e9e65SDag-Erling Smørgrav 	do {
1142a1b37df2SDag-Erling Smørgrav 		len = fetch_read(conn, &c, 1);
1143bb13d0afSDag-Erling Smørgrav 		if (len == -1)
1144e19e6098SDag-Erling Smørgrav 			return (-1);
1145bb13d0afSDag-Erling Smørgrav 		if (len == 0)
1146b68fbebdSDavid E. O'Brien 			break;
1147dea29ca1SDag-Erling Smørgrav 		conn->buf[conn->buflen++] = c;
1148dea29ca1SDag-Erling Smørgrav 		if (conn->buflen == conn->bufsize) {
1149dea29ca1SDag-Erling Smørgrav 			tmp = conn->buf;
1150dea29ca1SDag-Erling Smørgrav 			tmpsize = conn->bufsize * 2 + 1;
1151dea29ca1SDag-Erling Smørgrav 			if ((tmp = realloc(tmp, tmpsize)) == NULL) {
1152fc6e9e65SDag-Erling Smørgrav 				errno = ENOMEM;
1153e19e6098SDag-Erling Smørgrav 				return (-1);
1154fc6e9e65SDag-Erling Smørgrav 			}
1155dea29ca1SDag-Erling Smørgrav 			conn->buf = tmp;
1156dea29ca1SDag-Erling Smørgrav 			conn->bufsize = tmpsize;
1157fc6e9e65SDag-Erling Smørgrav 		}
1158fc6e9e65SDag-Erling Smørgrav 	} while (c != '\n');
1159fc6e9e65SDag-Erling Smørgrav 
1160dea29ca1SDag-Erling Smørgrav 	conn->buf[conn->buflen] = '\0';
1161dea29ca1SDag-Erling Smørgrav 	DEBUG(fprintf(stderr, "<<< %s", conn->buf));
1162e19e6098SDag-Erling Smørgrav 	return (0);
1163fc6e9e65SDag-Erling Smørgrav }
1164fc6e9e65SDag-Erling Smørgrav 
1165fc6e9e65SDag-Erling Smørgrav 
1166a1763027SDag-Erling Smørgrav /*
11679601e333SDag-Erling Smørgrav  * Write to a connection w/ timeout
11689601e333SDag-Erling Smørgrav  */
11699601e333SDag-Erling Smørgrav ssize_t
1170a1b37df2SDag-Erling Smørgrav fetch_write(conn_t *conn, const char *buf, size_t len)
11719601e333SDag-Erling Smørgrav {
1172a4a37038SWarner Losh 	struct iovec iov;
1173a4a37038SWarner Losh 
1174a4a37038SWarner Losh 	iov.iov_base = __DECONST(char *, buf);
1175a4a37038SWarner Losh 	iov.iov_len = len;
1176a1b37df2SDag-Erling Smørgrav 	return fetch_writev(conn, &iov, 1);
1177a4a37038SWarner Losh }
1178a4a37038SWarner Losh 
1179a4a37038SWarner Losh /*
1180a4a37038SWarner Losh  * Write a vector to a connection w/ timeout
1181a4a37038SWarner Losh  * Note: can modify the iovec.
1182a4a37038SWarner Losh  */
1183a4a37038SWarner Losh ssize_t
1184a1b37df2SDag-Erling Smørgrav fetch_writev(conn_t *conn, struct iovec *iov, int iovcnt)
1185a4a37038SWarner Losh {
11865092cf05SDag-Erling Smørgrav 	struct timeval now, timeout, delta;
1187215a27f1SDag-Erling Smørgrav 	struct pollfd pfd;
11889601e333SDag-Erling Smørgrav 	ssize_t wlen, total;
11899c1ca3a1SDag-Erling Smørgrav 	int deltams;
11909601e333SDag-Erling Smørgrav 
1191215a27f1SDag-Erling Smørgrav 	memset(&pfd, 0, sizeof pfd);
11929601e333SDag-Erling Smørgrav 	if (fetchTimeout) {
1193215a27f1SDag-Erling Smørgrav 		pfd.fd = conn->sd;
1194215a27f1SDag-Erling Smørgrav 		pfd.events = POLLOUT | POLLERR;
11959601e333SDag-Erling Smørgrav 		gettimeofday(&timeout, NULL);
11969601e333SDag-Erling Smørgrav 		timeout.tv_sec += fetchTimeout;
11979601e333SDag-Erling Smørgrav 	}
11989601e333SDag-Erling Smørgrav 
1199a4a37038SWarner Losh 	total = 0;
1200a4a37038SWarner Losh 	while (iovcnt > 0) {
1201215a27f1SDag-Erling Smørgrav 		while (fetchTimeout && pfd.revents == 0) {
12029601e333SDag-Erling Smørgrav 			gettimeofday(&now, NULL);
12039c1ca3a1SDag-Erling Smørgrav 			if (!timercmp(&timeout, &now, >)) {
12049601e333SDag-Erling Smørgrav 				errno = ETIMEDOUT;
1205a1b37df2SDag-Erling Smørgrav 				fetch_syserr();
12069601e333SDag-Erling Smørgrav 				return (-1);
12079601e333SDag-Erling Smørgrav 			}
12089c1ca3a1SDag-Erling Smørgrav 			timersub(&timeout, &now, &delta);
12099c1ca3a1SDag-Erling Smørgrav 			deltams = delta.tv_sec * 1000 +
12109c1ca3a1SDag-Erling Smørgrav 			    delta.tv_usec / 1000;
12119601e333SDag-Erling Smørgrav 			errno = 0;
12129c1ca3a1SDag-Erling Smørgrav 			pfd.revents = 0;
12139c1ca3a1SDag-Erling Smørgrav 			if (poll(&pfd, 1, deltams) < 0) {
12143674911cSBaptiste Daroussin 				/* POSIX compliance */
12154472d6e1SBaptiste Daroussin 				if (errno == EAGAIN)
12164472d6e1SBaptiste Daroussin 					continue;
12179601e333SDag-Erling Smørgrav 				if (errno == EINTR && fetchRestartCalls)
12189601e333SDag-Erling Smørgrav 					continue;
12199601e333SDag-Erling Smørgrav 				return (-1);
12209601e333SDag-Erling Smørgrav 			}
12219601e333SDag-Erling Smørgrav 		}
12229601e333SDag-Erling Smørgrav 		errno = 0;
12233070f6cbSDag-Erling Smørgrav #ifdef WITH_SSL
12249601e333SDag-Erling Smørgrav 		if (conn->ssl != NULL)
1225a4a37038SWarner Losh 			wlen = SSL_write(conn->ssl,
1226a4a37038SWarner Losh 			    iov->iov_base, iov->iov_len);
12279601e333SDag-Erling Smørgrav 		else
12283070f6cbSDag-Erling Smørgrav #endif
1229a4a37038SWarner Losh 			wlen = writev(conn->sd, iov, iovcnt);
1230a4a37038SWarner Losh 		if (wlen == 0) {
12319601e333SDag-Erling Smørgrav 			/* we consider a short write a failure */
1232caaffed8SDag-Erling Smørgrav 			/* XXX perhaps we shouldn't in the SSL case */
1233a4a37038SWarner Losh 			errno = EPIPE;
1234a1b37df2SDag-Erling Smørgrav 			fetch_syserr();
12359601e333SDag-Erling Smørgrav 			return (-1);
1236a4a37038SWarner Losh 		}
12379601e333SDag-Erling Smørgrav 		if (wlen < 0) {
12389601e333SDag-Erling Smørgrav 			if (errno == EINTR && fetchRestartCalls)
12399601e333SDag-Erling Smørgrav 				continue;
12409601e333SDag-Erling Smørgrav 			return (-1);
12419601e333SDag-Erling Smørgrav 		}
12429601e333SDag-Erling Smørgrav 		total += wlen;
1243a4a37038SWarner Losh 		while (iovcnt > 0 && wlen >= (ssize_t)iov->iov_len) {
1244a4a37038SWarner Losh 			wlen -= iov->iov_len;
1245a4a37038SWarner Losh 			iov++;
1246a4a37038SWarner Losh 			iovcnt--;
1247a4a37038SWarner Losh 		}
1248a4a37038SWarner Losh 		if (iovcnt > 0) {
1249a4a37038SWarner Losh 			iov->iov_len -= wlen;
1250a4a37038SWarner Losh 			iov->iov_base = __DECONST(char *, iov->iov_base) + wlen;
1251a4a37038SWarner Losh 		}
12529601e333SDag-Erling Smørgrav 	}
12539601e333SDag-Erling Smørgrav 	return (total);
12549601e333SDag-Erling Smørgrav }
12559601e333SDag-Erling Smørgrav 
1256f606d589SDag-Erling Smørgrav 
12579601e333SDag-Erling Smørgrav /*
12589601e333SDag-Erling Smørgrav  * Write a line of text to a connection w/ timeout
1259a1763027SDag-Erling Smørgrav  */
1260a1763027SDag-Erling Smørgrav int
1261a1b37df2SDag-Erling Smørgrav fetch_putln(conn_t *conn, const char *str, size_t len)
1262a1763027SDag-Erling Smørgrav {
1263a4a37038SWarner Losh 	struct iovec iov[2];
12647504527eSDag-Erling Smørgrav 	int ret;
126566ffb8a3SDag-Erling Smørgrav 
126666ffb8a3SDag-Erling Smørgrav 	DEBUG(fprintf(stderr, ">>> %s\n", str));
1267a4a37038SWarner Losh 	iov[0].iov_base = __DECONST(char *, str);
1268a4a37038SWarner Losh 	iov[0].iov_len = len;
1269a4a37038SWarner Losh 	iov[1].iov_base = __DECONST(char *, ENDL);
1270930105c1SDag-Erling Smørgrav 	iov[1].iov_len = sizeof(ENDL);
12717504527eSDag-Erling Smørgrav 	if (len == 0)
1272a1b37df2SDag-Erling Smørgrav 		ret = fetch_writev(conn, &iov[1], 1);
12737504527eSDag-Erling Smørgrav 	else
1274a1b37df2SDag-Erling Smørgrav 		ret = fetch_writev(conn, iov, 2);
12757504527eSDag-Erling Smørgrav 	if (ret == -1)
1276e19e6098SDag-Erling Smørgrav 		return (-1);
1277e19e6098SDag-Erling Smørgrav 	return (0);
1278a1763027SDag-Erling Smørgrav }
1279a1763027SDag-Erling Smørgrav 
1280a1763027SDag-Erling Smørgrav 
1281dea29ca1SDag-Erling Smørgrav /*
1282dea29ca1SDag-Erling Smørgrav  * Close connection
1283dea29ca1SDag-Erling Smørgrav  */
1284dea29ca1SDag-Erling Smørgrav int
1285a1b37df2SDag-Erling Smørgrav fetch_close(conn_t *conn)
1286dea29ca1SDag-Erling Smørgrav {
1287dea29ca1SDag-Erling Smørgrav 	int ret;
1288dea29ca1SDag-Erling Smørgrav 
1289f606d589SDag-Erling Smørgrav 	if (--conn->ref > 0)
1290f606d589SDag-Erling Smørgrav 		return (0);
1291dcd47379SDag-Erling Smørgrav #ifdef WITH_SSL
1292dcd47379SDag-Erling Smørgrav 	if (conn->ssl) {
1293dcd47379SDag-Erling Smørgrav 		SSL_shutdown(conn->ssl);
1294dcd47379SDag-Erling Smørgrav 		SSL_set_connect_state(conn->ssl);
1295dcd47379SDag-Erling Smørgrav 		SSL_free(conn->ssl);
1296dcd47379SDag-Erling Smørgrav 		conn->ssl = NULL;
1297dcd47379SDag-Erling Smørgrav 	}
1298dcd47379SDag-Erling Smørgrav 	if (conn->ssl_ctx) {
1299dcd47379SDag-Erling Smørgrav 		SSL_CTX_free(conn->ssl_ctx);
1300dcd47379SDag-Erling Smørgrav 		conn->ssl_ctx = NULL;
1301dcd47379SDag-Erling Smørgrav 	}
1302dcd47379SDag-Erling Smørgrav 	if (conn->ssl_cert) {
1303dcd47379SDag-Erling Smørgrav 		X509_free(conn->ssl_cert);
1304dcd47379SDag-Erling Smørgrav 		conn->ssl_cert = NULL;
1305dcd47379SDag-Erling Smørgrav 	}
1306dcd47379SDag-Erling Smørgrav #endif
1307dea29ca1SDag-Erling Smørgrav 	ret = close(conn->sd);
130855cf7be1SDag-Erling Smørgrav 	free(conn->buf);
1309dea29ca1SDag-Erling Smørgrav 	free(conn);
1310dea29ca1SDag-Erling Smørgrav 	return (ret);
1311dea29ca1SDag-Erling Smørgrav }
1312dea29ca1SDag-Erling Smørgrav 
1313dea29ca1SDag-Erling Smørgrav 
1314ce71b736SDag-Erling Smørgrav /*** Directory-related utility functions *************************************/
1315ce71b736SDag-Erling Smørgrav 
1316ce71b736SDag-Erling Smørgrav int
1317a1b37df2SDag-Erling Smørgrav fetch_add_entry(struct url_ent **p, int *size, int *len,
1318f573a5fcSDag-Erling Smørgrav     const char *name, struct url_stat *us)
1319ce71b736SDag-Erling Smørgrav {
1320ce71b736SDag-Erling Smørgrav 	struct url_ent *tmp;
1321ce71b736SDag-Erling Smørgrav 
1322ce71b736SDag-Erling Smørgrav 	if (*p == NULL) {
13235a51c23bSDag-Erling Smørgrav 		*size = 0;
1324ce71b736SDag-Erling Smørgrav 		*len = 0;
1325ce71b736SDag-Erling Smørgrav 	}
1326ce71b736SDag-Erling Smørgrav 
1327ce71b736SDag-Erling Smørgrav 	if (*len >= *size - 1) {
13280012b66bSPedro F. Giffuni 		tmp = reallocarray(*p, *size * 2 + 1, sizeof(**p));
1329ce71b736SDag-Erling Smørgrav 		if (tmp == NULL) {
1330ce71b736SDag-Erling Smørgrav 			errno = ENOMEM;
1331a1b37df2SDag-Erling Smørgrav 			fetch_syserr();
1332e19e6098SDag-Erling Smørgrav 			return (-1);
1333ce71b736SDag-Erling Smørgrav 		}
13345a51c23bSDag-Erling Smørgrav 		*size = (*size * 2 + 1);
1335ce71b736SDag-Erling Smørgrav 		*p = tmp;
1336ce71b736SDag-Erling Smørgrav 	}
1337ce71b736SDag-Erling Smørgrav 
1338ce71b736SDag-Erling Smørgrav 	tmp = *p + *len;
13392b26f942SDag-Erling Smørgrav 	snprintf(tmp->name, PATH_MAX, "%s", name);
1340340b079bSDag-Erling Smørgrav 	memcpy(&tmp->stat, us, sizeof(*us));
1341ce71b736SDag-Erling Smørgrav 
1342ce71b736SDag-Erling Smørgrav 	(*len)++;
1343ce71b736SDag-Erling Smørgrav 	(++tmp)->name[0] = 0;
1344ce71b736SDag-Erling Smørgrav 
1345e19e6098SDag-Erling Smørgrav 	return (0);
1346ce71b736SDag-Erling Smørgrav }
134707350d12SDag-Erling Smørgrav 
134807350d12SDag-Erling Smørgrav 
134907350d12SDag-Erling Smørgrav /*** Authentication-related utility functions ********************************/
135007350d12SDag-Erling Smørgrav 
135107350d12SDag-Erling Smørgrav static const char *
1352a1b37df2SDag-Erling Smørgrav fetch_read_word(FILE *f)
135307350d12SDag-Erling Smørgrav {
135407350d12SDag-Erling Smørgrav 	static char word[1024];
135507350d12SDag-Erling Smørgrav 
1356fc2841a9SColin Percival 	if (fscanf(f, " %1023s ", word) != 1)
135707350d12SDag-Erling Smørgrav 		return (NULL);
135807350d12SDag-Erling Smørgrav 	return (word);
135907350d12SDag-Erling Smørgrav }
136007350d12SDag-Erling Smørgrav 
1361d8713bf3SBaptiste Daroussin static int
1362d8713bf3SBaptiste Daroussin fetch_netrc_open(void)
136307350d12SDag-Erling Smørgrav {
1364d8713bf3SBaptiste Daroussin 	const char *p;
136507350d12SDag-Erling Smørgrav 	char fn[PATH_MAX];
136607350d12SDag-Erling Smørgrav 
136707350d12SDag-Erling Smørgrav 	if ((p = getenv("NETRC")) != NULL) {
1368930105c1SDag-Erling Smørgrav 		if (snprintf(fn, sizeof(fn), "%s", p) >= (int)sizeof(fn)) {
1369a1b37df2SDag-Erling Smørgrav 			fetch_info("$NETRC specifies a file name "
137007350d12SDag-Erling Smørgrav 			    "longer than PATH_MAX");
137107350d12SDag-Erling Smørgrav 			return (-1);
137207350d12SDag-Erling Smørgrav 		}
137307350d12SDag-Erling Smørgrav 	} else {
137407350d12SDag-Erling Smørgrav 		if ((p = getenv("HOME")) != NULL) {
137507350d12SDag-Erling Smørgrav 			struct passwd *pwd;
137607350d12SDag-Erling Smørgrav 
137707350d12SDag-Erling Smørgrav 			if ((pwd = getpwuid(getuid())) == NULL ||
137807350d12SDag-Erling Smørgrav 			    (p = pwd->pw_dir) == NULL)
137907350d12SDag-Erling Smørgrav 				return (-1);
138007350d12SDag-Erling Smørgrav 		}
1381930105c1SDag-Erling Smørgrav 		if (snprintf(fn, sizeof(fn), "%s/.netrc", p) >= (int)sizeof(fn))
138207350d12SDag-Erling Smørgrav 			return (-1);
138307350d12SDag-Erling Smørgrav 	}
138407350d12SDag-Erling Smørgrav 
1385d8713bf3SBaptiste Daroussin 	return (open(fn, O_RDONLY));
1386d8713bf3SBaptiste Daroussin }
1387d8713bf3SBaptiste Daroussin 
1388d8713bf3SBaptiste Daroussin /*
1389d8713bf3SBaptiste Daroussin  * Get authentication data for a URL from .netrc
1390d8713bf3SBaptiste Daroussin  */
1391d8713bf3SBaptiste Daroussin int
1392d8713bf3SBaptiste Daroussin fetch_netrc_auth(struct url *url)
1393d8713bf3SBaptiste Daroussin {
1394d8713bf3SBaptiste Daroussin 	const char *word;
1395d8713bf3SBaptiste Daroussin 	FILE *f;
1396d8713bf3SBaptiste Daroussin 
1397d8713bf3SBaptiste Daroussin 	if (url->netrcfd == -2)
1398d8713bf3SBaptiste Daroussin 		url->netrcfd = fetch_netrc_open();
1399d8713bf3SBaptiste Daroussin 	if (url->netrcfd < 0)
140007350d12SDag-Erling Smørgrav 		return (-1);
1401d8713bf3SBaptiste Daroussin 	if ((f = fdopen(url->netrcfd, "r")) == NULL)
1402d8713bf3SBaptiste Daroussin 		return (-1);
1403d8713bf3SBaptiste Daroussin 	rewind(f);
1404a1b37df2SDag-Erling Smørgrav 	while ((word = fetch_read_word(f)) != NULL) {
140507350d12SDag-Erling Smørgrav 		if (strcmp(word, "default") == 0) {
1406a1b37df2SDag-Erling Smørgrav 			DEBUG(fetch_info("Using default .netrc settings"));
140707350d12SDag-Erling Smørgrav 			break;
140807350d12SDag-Erling Smørgrav 		}
140907350d12SDag-Erling Smørgrav 		if (strcmp(word, "machine") == 0 &&
1410a1b37df2SDag-Erling Smørgrav 		    (word = fetch_read_word(f)) != NULL &&
141107350d12SDag-Erling Smørgrav 		    strcasecmp(word, url->host) == 0) {
1412a1b37df2SDag-Erling Smørgrav 			DEBUG(fetch_info("Using .netrc settings for %s", word));
141307350d12SDag-Erling Smørgrav 			break;
141407350d12SDag-Erling Smørgrav 		}
141507350d12SDag-Erling Smørgrav 	}
141607350d12SDag-Erling Smørgrav 	if (word == NULL)
141707350d12SDag-Erling Smørgrav 		goto ferr;
1418a1b37df2SDag-Erling Smørgrav 	while ((word = fetch_read_word(f)) != NULL) {
141907350d12SDag-Erling Smørgrav 		if (strcmp(word, "login") == 0) {
1420a1b37df2SDag-Erling Smørgrav 			if ((word = fetch_read_word(f)) == NULL)
142107350d12SDag-Erling Smørgrav 				goto ferr;
1422930105c1SDag-Erling Smørgrav 			if (snprintf(url->user, sizeof(url->user),
14239015b953SJohn W. De Boskey 				"%s", word) > (int)sizeof(url->user)) {
1424a1b37df2SDag-Erling Smørgrav 				fetch_info("login name in .netrc is too long");
142507350d12SDag-Erling Smørgrav 				url->user[0] = '\0';
142607350d12SDag-Erling Smørgrav 			}
142707350d12SDag-Erling Smørgrav 		} else if (strcmp(word, "password") == 0) {
1428a1b37df2SDag-Erling Smørgrav 			if ((word = fetch_read_word(f)) == NULL)
142907350d12SDag-Erling Smørgrav 				goto ferr;
1430930105c1SDag-Erling Smørgrav 			if (snprintf(url->pwd, sizeof(url->pwd),
14319015b953SJohn W. De Boskey 				"%s", word) > (int)sizeof(url->pwd)) {
1432a1b37df2SDag-Erling Smørgrav 				fetch_info("password in .netrc is too long");
143307350d12SDag-Erling Smørgrav 				url->pwd[0] = '\0';
143407350d12SDag-Erling Smørgrav 			}
143507350d12SDag-Erling Smørgrav 		} else if (strcmp(word, "account") == 0) {
1436a1b37df2SDag-Erling Smørgrav 			if ((word = fetch_read_word(f)) == NULL)
143707350d12SDag-Erling Smørgrav 				goto ferr;
143807350d12SDag-Erling Smørgrav 			/* XXX not supported! */
143907350d12SDag-Erling Smørgrav 		} else {
144007350d12SDag-Erling Smørgrav 			break;
144107350d12SDag-Erling Smørgrav 		}
144207350d12SDag-Erling Smørgrav 	}
144307350d12SDag-Erling Smørgrav 	fclose(f);
144407350d12SDag-Erling Smørgrav 	return (0);
144507350d12SDag-Erling Smørgrav  ferr:
144607350d12SDag-Erling Smørgrav 	fclose(f);
144707350d12SDag-Erling Smørgrav 	return (-1);
144807350d12SDag-Erling Smørgrav }
144962a2681cSDag-Erling Smørgrav 
145062a2681cSDag-Erling Smørgrav /*
145162a2681cSDag-Erling Smørgrav  * The no_proxy environment variable specifies a set of domains for
145262a2681cSDag-Erling Smørgrav  * which the proxy should not be consulted; the contents is a comma-,
145362a2681cSDag-Erling Smørgrav  * or space-separated list of domain names.  A single asterisk will
145462a2681cSDag-Erling Smørgrav  * override all proxy variables and no transactions will be proxied
145575f46cf6SPedro F. Giffuni  * (for compatibility with lynx and curl, see the discussion at
145662a2681cSDag-Erling Smørgrav  * <http://curl.haxx.se/mail/archive_pre_oct_99/0009.html>).
145762a2681cSDag-Erling Smørgrav  */
145862a2681cSDag-Erling Smørgrav int
145962a2681cSDag-Erling Smørgrav fetch_no_proxy_match(const char *host)
146062a2681cSDag-Erling Smørgrav {
146162a2681cSDag-Erling Smørgrav 	const char *no_proxy, *p, *q;
146262a2681cSDag-Erling Smørgrav 	size_t h_len, d_len;
146362a2681cSDag-Erling Smørgrav 
146462a2681cSDag-Erling Smørgrav 	if ((no_proxy = getenv("NO_PROXY")) == NULL &&
146562a2681cSDag-Erling Smørgrav 	    (no_proxy = getenv("no_proxy")) == NULL)
146662a2681cSDag-Erling Smørgrav 		return (0);
146762a2681cSDag-Erling Smørgrav 
146862a2681cSDag-Erling Smørgrav 	/* asterisk matches any hostname */
146962a2681cSDag-Erling Smørgrav 	if (strcmp(no_proxy, "*") == 0)
147062a2681cSDag-Erling Smørgrav 		return (1);
147162a2681cSDag-Erling Smørgrav 
147262a2681cSDag-Erling Smørgrav 	h_len = strlen(host);
147362a2681cSDag-Erling Smørgrav 	p = no_proxy;
147462a2681cSDag-Erling Smørgrav 	do {
147562a2681cSDag-Erling Smørgrav 		/* position p at the beginning of a domain suffix */
1476facd9827SDag-Erling Smørgrav 		while (*p == ',' || isspace((unsigned char)*p))
147762a2681cSDag-Erling Smørgrav 			p++;
147862a2681cSDag-Erling Smørgrav 
147962a2681cSDag-Erling Smørgrav 		/* position q at the first separator character */
148062a2681cSDag-Erling Smørgrav 		for (q = p; *q; ++q)
1481facd9827SDag-Erling Smørgrav 			if (*q == ',' || isspace((unsigned char)*q))
148262a2681cSDag-Erling Smørgrav 				break;
148362a2681cSDag-Erling Smørgrav 
148462a2681cSDag-Erling Smørgrav 		d_len = q - p;
1485c0d2581bSFabien Thomas 		if (d_len > 0 && h_len >= d_len &&
148662a2681cSDag-Erling Smørgrav 		    strncasecmp(host + h_len - d_len,
148762a2681cSDag-Erling Smørgrav 			p, d_len) == 0) {
148862a2681cSDag-Erling Smørgrav 			/* domain name matches */
148962a2681cSDag-Erling Smørgrav 			return (1);
149062a2681cSDag-Erling Smørgrav 		}
149162a2681cSDag-Erling Smørgrav 
149262a2681cSDag-Erling Smørgrav 		p = q + 1;
149362a2681cSDag-Erling Smørgrav 	} while (*q);
149462a2681cSDag-Erling Smørgrav 
149562a2681cSDag-Erling Smørgrav 	return (0);
149662a2681cSDag-Erling Smørgrav }
1497