xref: /freebsd/lib/libfetch/common.c (revision 6d9160409317733fd7cf140d8f33fe408e69ee40)
1842a95ccSDag-Erling Smørgrav /*-
2792ef1aeSDag-Erling Smørgrav  * Copyright (c) 1998-2016 Dag-Erling Smørgrav
3dcd47379SDag-Erling Smørgrav  * Copyright (c) 2013 Michael Gmelin <freebsd@grem.de>
4842a95ccSDag-Erling Smørgrav  * All rights reserved.
5842a95ccSDag-Erling Smørgrav  *
6842a95ccSDag-Erling Smørgrav  * Redistribution and use in source and binary forms, with or without
7842a95ccSDag-Erling Smørgrav  * modification, are permitted provided that the following conditions
8842a95ccSDag-Erling Smørgrav  * are met:
9842a95ccSDag-Erling Smørgrav  * 1. Redistributions of source code must retain the above copyright
10842a95ccSDag-Erling Smørgrav  *    notice, this list of conditions and the following disclaimer
11842a95ccSDag-Erling Smørgrav  *    in this position and unchanged.
12842a95ccSDag-Erling Smørgrav  * 2. Redistributions in binary form must reproduce the above copyright
13842a95ccSDag-Erling Smørgrav  *    notice, this list of conditions and the following disclaimer in the
14842a95ccSDag-Erling Smørgrav  *    documentation and/or other materials provided with the distribution.
15842a95ccSDag-Erling Smørgrav  * 3. The name of the author may not be used to endorse or promote products
16842a95ccSDag-Erling Smørgrav  *    derived from this software without specific prior written permission
17842a95ccSDag-Erling Smørgrav  *
18842a95ccSDag-Erling Smørgrav  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
19842a95ccSDag-Erling Smørgrav  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
20842a95ccSDag-Erling Smørgrav  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
21842a95ccSDag-Erling Smørgrav  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
22842a95ccSDag-Erling Smørgrav  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
23842a95ccSDag-Erling Smørgrav  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24842a95ccSDag-Erling Smørgrav  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25842a95ccSDag-Erling Smørgrav  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26842a95ccSDag-Erling Smørgrav  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
27842a95ccSDag-Erling Smørgrav  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28842a95ccSDag-Erling Smørgrav  */
29842a95ccSDag-Erling Smørgrav 
30cecb889fSMatthew Dillon #include <sys/cdefs.h>
31cecb889fSMatthew Dillon __FBSDID("$FreeBSD$");
32cecb889fSMatthew Dillon 
330fba3a00SDag-Erling Smørgrav #include <sys/param.h>
34842a95ccSDag-Erling Smørgrav #include <sys/socket.h>
35fc6e9e65SDag-Erling Smørgrav #include <sys/time.h>
36a1763027SDag-Erling Smørgrav #include <sys/uio.h>
3762a2681cSDag-Erling Smørgrav 
38842a95ccSDag-Erling Smørgrav #include <netinet/in.h>
39842a95ccSDag-Erling Smørgrav 
4062a2681cSDag-Erling Smørgrav #include <ctype.h>
41842a95ccSDag-Erling Smørgrav #include <errno.h>
42caaffed8SDag-Erling Smørgrav #include <fcntl.h>
43842a95ccSDag-Erling Smørgrav #include <netdb.h>
44215a27f1SDag-Erling Smørgrav #include <poll.h>
4507350d12SDag-Erling Smørgrav #include <pwd.h>
46ba101983SDag-Erling Smørgrav #include <stdarg.h>
470fba3a00SDag-Erling Smørgrav #include <stdlib.h>
480fba3a00SDag-Erling Smørgrav #include <stdio.h>
49842a95ccSDag-Erling Smørgrav #include <string.h>
50842a95ccSDag-Erling Smørgrav #include <unistd.h>
51842a95ccSDag-Erling Smørgrav 
52dcd47379SDag-Erling Smørgrav #ifdef WITH_SSL
53dcd47379SDag-Erling Smørgrav #include <openssl/x509v3.h>
54dcd47379SDag-Erling Smørgrav #endif
55dcd47379SDag-Erling Smørgrav 
56842a95ccSDag-Erling Smørgrav #include "fetch.h"
57842a95ccSDag-Erling Smørgrav #include "common.h"
58842a95ccSDag-Erling Smørgrav 
59d8acd8dcSDag-Erling Smørgrav 
60842a95ccSDag-Erling Smørgrav /*** Local data **************************************************************/
61842a95ccSDag-Erling Smørgrav 
62842a95ccSDag-Erling Smørgrav /*
63842a95ccSDag-Erling Smørgrav  * Error messages for resolver errors
64842a95ccSDag-Erling Smørgrav  */
65a1b37df2SDag-Erling Smørgrav static struct fetcherr netdb_errlist[] = {
663d82ba43SHajimu UMEMOTO #ifdef EAI_NODATA
6728c645cfSHajimu UMEMOTO 	{ EAI_NODATA,	FETCH_RESOLV,	"Host not found" },
683d82ba43SHajimu UMEMOTO #endif
6928c645cfSHajimu UMEMOTO 	{ EAI_AGAIN,	FETCH_TEMP,	"Transient resolver failure" },
7028c645cfSHajimu UMEMOTO 	{ EAI_FAIL,	FETCH_RESOLV,	"Non-recoverable resolver failure" },
7128c645cfSHajimu UMEMOTO 	{ EAI_NONAME,	FETCH_RESOLV,	"No address record" },
72d8acd8dcSDag-Erling Smørgrav 	{ -1,		FETCH_UNKNOWN,	"Unknown resolver error" }
73842a95ccSDag-Erling Smørgrav };
74842a95ccSDag-Erling Smørgrav 
75a1763027SDag-Erling Smørgrav /* End-of-Line */
7638c7e4a6SArchie Cobbs static const char ENDL[2] = "\r\n";
77a1763027SDag-Erling Smørgrav 
78842a95ccSDag-Erling Smørgrav 
79842a95ccSDag-Erling Smørgrav /*** Error-reporting functions ***********************************************/
80842a95ccSDag-Erling Smørgrav 
81842a95ccSDag-Erling Smørgrav /*
82842a95ccSDag-Erling Smørgrav  * Map error code to string
83842a95ccSDag-Erling Smørgrav  */
84ba101983SDag-Erling Smørgrav static struct fetcherr *
85a1b37df2SDag-Erling Smørgrav fetch_finderr(struct fetcherr *p, int e)
86842a95ccSDag-Erling Smørgrav {
87ba101983SDag-Erling Smørgrav 	while (p->num != -1 && p->num != e)
88ba101983SDag-Erling Smørgrav 		p++;
89e19e6098SDag-Erling Smørgrav 	return (p);
90842a95ccSDag-Erling Smørgrav }
91842a95ccSDag-Erling Smørgrav 
92842a95ccSDag-Erling Smørgrav /*
93842a95ccSDag-Erling Smørgrav  * Set error code
94842a95ccSDag-Erling Smørgrav  */
95842a95ccSDag-Erling Smørgrav void
96a1b37df2SDag-Erling Smørgrav fetch_seterr(struct fetcherr *p, int e)
97842a95ccSDag-Erling Smørgrav {
98a1b37df2SDag-Erling Smørgrav 	p = fetch_finderr(p, e);
99ba101983SDag-Erling Smørgrav 	fetchLastErrCode = p->cat;
100ba101983SDag-Erling Smørgrav 	snprintf(fetchLastErrString, MAXERRSTRING, "%s", p->string);
101842a95ccSDag-Erling Smørgrav }
102842a95ccSDag-Erling Smørgrav 
103842a95ccSDag-Erling Smørgrav /*
104842a95ccSDag-Erling Smørgrav  * Set error code according to errno
105842a95ccSDag-Erling Smørgrav  */
106842a95ccSDag-Erling Smørgrav void
107a1b37df2SDag-Erling Smørgrav fetch_syserr(void)
108842a95ccSDag-Erling Smørgrav {
109d8acd8dcSDag-Erling Smørgrav 	switch (errno) {
110d8acd8dcSDag-Erling Smørgrav 	case 0:
1110fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_OK;
112d8acd8dcSDag-Erling Smørgrav 		break;
113d8acd8dcSDag-Erling Smørgrav 	case EPERM:
114d8acd8dcSDag-Erling Smørgrav 	case EACCES:
115d8acd8dcSDag-Erling Smørgrav 	case EROFS:
116d8acd8dcSDag-Erling Smørgrav 	case EAUTH:
117d8acd8dcSDag-Erling Smørgrav 	case ENEEDAUTH:
1180fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_AUTH;
119d8acd8dcSDag-Erling Smørgrav 		break;
120d8acd8dcSDag-Erling Smørgrav 	case ENOENT:
121d8acd8dcSDag-Erling Smørgrav 	case EISDIR: /* XXX */
1220fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_UNAVAIL;
123d8acd8dcSDag-Erling Smørgrav 		break;
124d8acd8dcSDag-Erling Smørgrav 	case ENOMEM:
1250fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_MEMORY;
126d8acd8dcSDag-Erling Smørgrav 		break;
127d8acd8dcSDag-Erling Smørgrav 	case EBUSY:
128d8acd8dcSDag-Erling Smørgrav 	case EAGAIN:
1290fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_TEMP;
130d8acd8dcSDag-Erling Smørgrav 		break;
131d8acd8dcSDag-Erling Smørgrav 	case EEXIST:
1320fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_EXISTS;
133d8acd8dcSDag-Erling Smørgrav 		break;
134d8acd8dcSDag-Erling Smørgrav 	case ENOSPC:
1350fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_FULL;
136d8acd8dcSDag-Erling Smørgrav 		break;
137d8acd8dcSDag-Erling Smørgrav 	case EADDRINUSE:
138d8acd8dcSDag-Erling Smørgrav 	case EADDRNOTAVAIL:
139d8acd8dcSDag-Erling Smørgrav 	case ENETDOWN:
140d8acd8dcSDag-Erling Smørgrav 	case ENETUNREACH:
141d8acd8dcSDag-Erling Smørgrav 	case ENETRESET:
142d8acd8dcSDag-Erling Smørgrav 	case EHOSTUNREACH:
1430fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_NETWORK;
144d8acd8dcSDag-Erling Smørgrav 		break;
145d8acd8dcSDag-Erling Smørgrav 	case ECONNABORTED:
146d8acd8dcSDag-Erling Smørgrav 	case ECONNRESET:
1470fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_ABORT;
148d8acd8dcSDag-Erling Smørgrav 		break;
149d8acd8dcSDag-Erling Smørgrav 	case ETIMEDOUT:
1500fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_TIMEOUT;
151d8acd8dcSDag-Erling Smørgrav 		break;
152d8acd8dcSDag-Erling Smørgrav 	case ECONNREFUSED:
153d8acd8dcSDag-Erling Smørgrav 	case EHOSTDOWN:
1540fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_DOWN;
155d8acd8dcSDag-Erling Smørgrav 		break;
156d8acd8dcSDag-Erling Smørgrav 	default:
1570fba3a00SDag-Erling Smørgrav 		fetchLastErrCode = FETCH_UNKNOWN;
158d8acd8dcSDag-Erling Smørgrav 	}
159e19e6098SDag-Erling Smørgrav 	snprintf(fetchLastErrString, MAXERRSTRING, "%s", strerror(errno));
1600fba3a00SDag-Erling Smørgrav }
1610fba3a00SDag-Erling Smørgrav 
1620fba3a00SDag-Erling Smørgrav 
1630fba3a00SDag-Erling Smørgrav /*
1640fba3a00SDag-Erling Smørgrav  * Emit status message
1650fba3a00SDag-Erling Smørgrav  */
166ba101983SDag-Erling Smørgrav void
167a1b37df2SDag-Erling Smørgrav fetch_info(const char *fmt, ...)
1680fba3a00SDag-Erling Smørgrav {
1690fba3a00SDag-Erling Smørgrav 	va_list ap;
1700fba3a00SDag-Erling Smørgrav 
1710fba3a00SDag-Erling Smørgrav 	va_start(ap, fmt);
172ba101983SDag-Erling Smørgrav 	vfprintf(stderr, fmt, ap);
1730fba3a00SDag-Erling Smørgrav 	va_end(ap);
174a68f8b58SDag-Erling Smørgrav 	fputc('\n', stderr);
175842a95ccSDag-Erling Smørgrav }
176842a95ccSDag-Erling Smørgrav 
177842a95ccSDag-Erling Smørgrav 
178842a95ccSDag-Erling Smørgrav /*** Network-related utility functions ***************************************/
179842a95ccSDag-Erling Smørgrav 
180842a95ccSDag-Erling Smørgrav /*
181e828ada7SDag-Erling Smørgrav  * Return the default port for a scheme
182e828ada7SDag-Erling Smørgrav  */
183e828ada7SDag-Erling Smørgrav int
184a1b37df2SDag-Erling Smørgrav fetch_default_port(const char *scheme)
185e828ada7SDag-Erling Smørgrav {
186e828ada7SDag-Erling Smørgrav 	struct servent *se;
187e828ada7SDag-Erling Smørgrav 
188e828ada7SDag-Erling Smørgrav 	if ((se = getservbyname(scheme, "tcp")) != NULL)
189e19e6098SDag-Erling Smørgrav 		return (ntohs(se->s_port));
190e828ada7SDag-Erling Smørgrav 	if (strcasecmp(scheme, SCHEME_FTP) == 0)
191e19e6098SDag-Erling Smørgrav 		return (FTP_DEFAULT_PORT);
192e828ada7SDag-Erling Smørgrav 	if (strcasecmp(scheme, SCHEME_HTTP) == 0)
193e19e6098SDag-Erling Smørgrav 		return (HTTP_DEFAULT_PORT);
194e19e6098SDag-Erling Smørgrav 	return (0);
195e828ada7SDag-Erling Smørgrav }
196e828ada7SDag-Erling Smørgrav 
197e828ada7SDag-Erling Smørgrav /*
198e828ada7SDag-Erling Smørgrav  * Return the default proxy port for a scheme
199e828ada7SDag-Erling Smørgrav  */
200e828ada7SDag-Erling Smørgrav int
201a1b37df2SDag-Erling Smørgrav fetch_default_proxy_port(const char *scheme)
202e828ada7SDag-Erling Smørgrav {
203e828ada7SDag-Erling Smørgrav 	if (strcasecmp(scheme, SCHEME_FTP) == 0)
204e19e6098SDag-Erling Smørgrav 		return (FTP_DEFAULT_PROXY_PORT);
205e828ada7SDag-Erling Smørgrav 	if (strcasecmp(scheme, SCHEME_HTTP) == 0)
206e19e6098SDag-Erling Smørgrav 		return (HTTP_DEFAULT_PROXY_PORT);
207e19e6098SDag-Erling Smørgrav 	return (0);
208e828ada7SDag-Erling Smørgrav }
209e828ada7SDag-Erling Smørgrav 
210f606d589SDag-Erling Smørgrav 
211e828ada7SDag-Erling Smørgrav /*
2129601e333SDag-Erling Smørgrav  * Create a connection for an existing descriptor.
2139601e333SDag-Erling Smørgrav  */
2149601e333SDag-Erling Smørgrav conn_t *
215a1b37df2SDag-Erling Smørgrav fetch_reopen(int sd)
2169601e333SDag-Erling Smørgrav {
2179601e333SDag-Erling Smørgrav 	conn_t *conn;
21828fd9307SJilles Tjoelker 	int opt = 1;
2199601e333SDag-Erling Smørgrav 
2209601e333SDag-Erling Smørgrav 	/* allocate and fill connection structure */
221930105c1SDag-Erling Smørgrav 	if ((conn = calloc(1, sizeof(*conn))) == NULL)
2229601e333SDag-Erling Smørgrav 		return (NULL);
22315b68c63SDag-Erling Smørgrav 	fcntl(sd, F_SETFD, FD_CLOEXEC);
22428fd9307SJilles Tjoelker 	setsockopt(sd, SOL_SOCKET, SO_NOSIGPIPE, &opt, sizeof opt);
2259601e333SDag-Erling Smørgrav 	conn->sd = sd;
226f606d589SDag-Erling Smørgrav 	++conn->ref;
227f606d589SDag-Erling Smørgrav 	return (conn);
228f606d589SDag-Erling Smørgrav }
229f606d589SDag-Erling Smørgrav 
230f606d589SDag-Erling Smørgrav 
231f606d589SDag-Erling Smørgrav /*
232f606d589SDag-Erling Smørgrav  * Bump a connection's reference count.
233f606d589SDag-Erling Smørgrav  */
234f606d589SDag-Erling Smørgrav conn_t *
235a1b37df2SDag-Erling Smørgrav fetch_ref(conn_t *conn)
236f606d589SDag-Erling Smørgrav {
237f606d589SDag-Erling Smørgrav 
238f606d589SDag-Erling Smørgrav 	++conn->ref;
2399601e333SDag-Erling Smørgrav 	return (conn);
2409601e333SDag-Erling Smørgrav }
2419601e333SDag-Erling Smørgrav 
2429601e333SDag-Erling Smørgrav 
2439601e333SDag-Erling Smørgrav /*
244792ef1aeSDag-Erling Smørgrav  * Resolve an address
245792ef1aeSDag-Erling Smørgrav  */
246792ef1aeSDag-Erling Smørgrav struct addrinfo *
247792ef1aeSDag-Erling Smørgrav fetch_resolve(const char *addr, int port, int af)
248792ef1aeSDag-Erling Smørgrav {
249792ef1aeSDag-Erling Smørgrav 	char hbuf[256], sbuf[8];
250792ef1aeSDag-Erling Smørgrav 	struct addrinfo hints, *res;
251792ef1aeSDag-Erling Smørgrav 	const char *sep, *host, *service;
252792ef1aeSDag-Erling Smørgrav 	int err, len;
253792ef1aeSDag-Erling Smørgrav 
254792ef1aeSDag-Erling Smørgrav 	/* split address if necessary */
255792ef1aeSDag-Erling Smørgrav 	err = EAI_SYSTEM;
256792ef1aeSDag-Erling Smørgrav 	if ((sep = strchr(addr, ':')) != NULL) {
257792ef1aeSDag-Erling Smørgrav 		len = snprintf(hbuf, sizeof(hbuf),
258792ef1aeSDag-Erling Smørgrav 		    "%.*s", (int)(sep - addr), addr);
259792ef1aeSDag-Erling Smørgrav 		if (len < 0)
260792ef1aeSDag-Erling Smørgrav 			return (NULL);
261792ef1aeSDag-Erling Smørgrav 		if (len >= (int)sizeof(hbuf)) {
262792ef1aeSDag-Erling Smørgrav 			errno = ENAMETOOLONG;
263792ef1aeSDag-Erling Smørgrav 			fetch_syserr();
264792ef1aeSDag-Erling Smørgrav 			return (NULL);
265792ef1aeSDag-Erling Smørgrav 		}
266792ef1aeSDag-Erling Smørgrav 		host = hbuf;
267792ef1aeSDag-Erling Smørgrav 		service = sep + 1;
268792ef1aeSDag-Erling Smørgrav 	} else if (port != 0) {
269792ef1aeSDag-Erling Smørgrav 		if (port < 1 || port > 65535) {
270792ef1aeSDag-Erling Smørgrav 			errno = EINVAL;
271792ef1aeSDag-Erling Smørgrav 			fetch_syserr();
272792ef1aeSDag-Erling Smørgrav 			return (NULL);
273792ef1aeSDag-Erling Smørgrav 		}
274792ef1aeSDag-Erling Smørgrav 		if (snprintf(sbuf, sizeof(sbuf), "%d", port) < 0) {
275792ef1aeSDag-Erling Smørgrav 			fetch_syserr();
276792ef1aeSDag-Erling Smørgrav 			return (NULL);
277792ef1aeSDag-Erling Smørgrav 		}
278792ef1aeSDag-Erling Smørgrav 		host = addr;
279792ef1aeSDag-Erling Smørgrav 		service = sbuf;
280792ef1aeSDag-Erling Smørgrav 	} else {
281792ef1aeSDag-Erling Smørgrav 		host = addr;
282792ef1aeSDag-Erling Smørgrav 		service = NULL;
283792ef1aeSDag-Erling Smørgrav 	}
284792ef1aeSDag-Erling Smørgrav 
285792ef1aeSDag-Erling Smørgrav 	/* resolve */
286792ef1aeSDag-Erling Smørgrav 	memset(&hints, 0, sizeof(hints));
287792ef1aeSDag-Erling Smørgrav 	hints.ai_family = af;
288792ef1aeSDag-Erling Smørgrav 	hints.ai_socktype = SOCK_STREAM;
289792ef1aeSDag-Erling Smørgrav 	hints.ai_flags = AI_ADDRCONFIG;
290792ef1aeSDag-Erling Smørgrav 	if ((err = getaddrinfo(host, service, &hints, &res)) != 0) {
291792ef1aeSDag-Erling Smørgrav 		netdb_seterr(err);
292792ef1aeSDag-Erling Smørgrav 		return (NULL);
293792ef1aeSDag-Erling Smørgrav 	}
294792ef1aeSDag-Erling Smørgrav 	return (res);
295792ef1aeSDag-Erling Smørgrav }
296792ef1aeSDag-Erling Smørgrav 
297792ef1aeSDag-Erling Smørgrav 
298792ef1aeSDag-Erling Smørgrav 
299792ef1aeSDag-Erling Smørgrav /*
300c42cb9d9SDag-Erling Smørgrav  * Bind a socket to a specific local address
301c42cb9d9SDag-Erling Smørgrav  */
302c42cb9d9SDag-Erling Smørgrav int
303a1b37df2SDag-Erling Smørgrav fetch_bind(int sd, int af, const char *addr)
304c42cb9d9SDag-Erling Smørgrav {
305792ef1aeSDag-Erling Smørgrav 	struct addrinfo *cliai, *ai;
306c42cb9d9SDag-Erling Smørgrav 	int err;
307c42cb9d9SDag-Erling Smørgrav 
308792ef1aeSDag-Erling Smørgrav 	if ((cliai = fetch_resolve(addr, 0, af)) == NULL)
309c42cb9d9SDag-Erling Smørgrav 		return (-1);
310792ef1aeSDag-Erling Smørgrav 	for (ai = cliai; ai != NULL; ai = ai->ai_next)
311792ef1aeSDag-Erling Smørgrav 		if ((err = bind(sd, ai->ai_addr, ai->ai_addrlen)) == 0)
312792ef1aeSDag-Erling Smørgrav 			break;
313792ef1aeSDag-Erling Smørgrav 	if (err != 0)
314792ef1aeSDag-Erling Smørgrav 		fetch_syserr();
315792ef1aeSDag-Erling Smørgrav 	freeaddrinfo(cliai);
316792ef1aeSDag-Erling Smørgrav 	return (err == 0 ? 0 : -1);
317c42cb9d9SDag-Erling Smørgrav }
318c42cb9d9SDag-Erling Smørgrav 
319c42cb9d9SDag-Erling Smørgrav 
320c42cb9d9SDag-Erling Smørgrav /*
321842a95ccSDag-Erling Smørgrav  * Establish a TCP connection to the specified port on the specified host.
322842a95ccSDag-Erling Smørgrav  */
323dea29ca1SDag-Erling Smørgrav conn_t *
324a1b37df2SDag-Erling Smørgrav fetch_connect(const char *host, int port, int af, int verbose)
325842a95ccSDag-Erling Smørgrav {
326792ef1aeSDag-Erling Smørgrav 	struct addrinfo *cais = NULL, *sais = NULL, *cai, *sai;
327c42cb9d9SDag-Erling Smørgrav 	const char *bindaddr;
328792ef1aeSDag-Erling Smørgrav 	conn_t *conn = NULL;
329792ef1aeSDag-Erling Smørgrav 	int err = 0, sd = -1;
330842a95ccSDag-Erling Smørgrav 
331f67efa37SDag-Erling Smørgrav 	DEBUG(fprintf(stderr, "---> %s:%d\n", host, port));
332842a95ccSDag-Erling Smørgrav 
333792ef1aeSDag-Erling Smørgrav 	/* resolve server address */
3340fba3a00SDag-Erling Smørgrav 	if (verbose)
335792ef1aeSDag-Erling Smørgrav 		fetch_info("resolving server address: %s:%d", host, port);
336792ef1aeSDag-Erling Smørgrav 	if ((sais = fetch_resolve(host, port, af)) == NULL)
337792ef1aeSDag-Erling Smørgrav 		goto fail;
3380fba3a00SDag-Erling Smørgrav 
339792ef1aeSDag-Erling Smørgrav 	/* resolve client address */
340c42cb9d9SDag-Erling Smørgrav 	bindaddr = getenv("FETCH_BIND_ADDRESS");
341792ef1aeSDag-Erling Smørgrav 	if (bindaddr != NULL && *bindaddr != '\0') {
3420fba3a00SDag-Erling Smørgrav 		if (verbose)
343792ef1aeSDag-Erling Smørgrav 			fetch_info("resolving client address: %s", bindaddr);
344792ef1aeSDag-Erling Smørgrav 		if ((cais = fetch_resolve(bindaddr, 0, af)) == NULL)
345792ef1aeSDag-Erling Smørgrav 			goto fail;
346c42cb9d9SDag-Erling Smørgrav 	}
347792ef1aeSDag-Erling Smørgrav 
348792ef1aeSDag-Erling Smørgrav 	/* try each server address in turn */
349792ef1aeSDag-Erling Smørgrav 	for (err = 0, sai = sais; sai != NULL; sai = sai->ai_next) {
350792ef1aeSDag-Erling Smørgrav 		/* open socket */
351792ef1aeSDag-Erling Smørgrav 		if ((sd = socket(sai->ai_family, SOCK_STREAM, 0)) < 0)
352792ef1aeSDag-Erling Smørgrav 			goto syserr;
353792ef1aeSDag-Erling Smørgrav 		/* attempt to bind to client address */
354792ef1aeSDag-Erling Smørgrav 		for (err = 0, cai = cais; cai != NULL; cai = cai->ai_next) {
355792ef1aeSDag-Erling Smørgrav 			if (cai->ai_family != sai->ai_family)
356792ef1aeSDag-Erling Smørgrav 				continue;
357792ef1aeSDag-Erling Smørgrav 			if ((err = bind(sd, cai->ai_addr, cai->ai_addrlen)) == 0)
35828c645cfSHajimu UMEMOTO 				break;
35928c645cfSHajimu UMEMOTO 		}
360792ef1aeSDag-Erling Smørgrav 		if (err != 0) {
361792ef1aeSDag-Erling Smørgrav 			if (verbose)
362792ef1aeSDag-Erling Smørgrav 				fetch_info("failed to bind to %s", bindaddr);
363792ef1aeSDag-Erling Smørgrav 			goto syserr;
364792ef1aeSDag-Erling Smørgrav 		}
365792ef1aeSDag-Erling Smørgrav 		/* attempt to connect to server address */
366792ef1aeSDag-Erling Smørgrav 		if ((err = connect(sd, sai->ai_addr, sai->ai_addrlen)) == 0)
367792ef1aeSDag-Erling Smørgrav 			break;
368792ef1aeSDag-Erling Smørgrav 		/* clean up before next attempt */
369792ef1aeSDag-Erling Smørgrav 		close(sd);
370792ef1aeSDag-Erling Smørgrav 		sd = -1;
371792ef1aeSDag-Erling Smørgrav 	}
372792ef1aeSDag-Erling Smørgrav 	if (err != 0) {
373792ef1aeSDag-Erling Smørgrav 		if (verbose)
374*6d916040SBjoern A. Zeeb 			fetch_info("failed to connect to %s:%d", host, port);
375792ef1aeSDag-Erling Smørgrav 		goto syserr;
376842a95ccSDag-Erling Smørgrav 	}
377842a95ccSDag-Erling Smørgrav 
378792ef1aeSDag-Erling Smørgrav 	if ((conn = fetch_reopen(sd)) == NULL)
379792ef1aeSDag-Erling Smørgrav 		goto syserr;
380792ef1aeSDag-Erling Smørgrav 	if (cais != NULL)
381792ef1aeSDag-Erling Smørgrav 		freeaddrinfo(cais);
382792ef1aeSDag-Erling Smørgrav 	if (sais != NULL)
383792ef1aeSDag-Erling Smørgrav 		freeaddrinfo(sais);
384dea29ca1SDag-Erling Smørgrav 	return (conn);
385792ef1aeSDag-Erling Smørgrav syserr:
386792ef1aeSDag-Erling Smørgrav 	fetch_syserr();
387792ef1aeSDag-Erling Smørgrav 	goto fail;
388792ef1aeSDag-Erling Smørgrav fail:
389792ef1aeSDag-Erling Smørgrav 	if (sd >= 0)
390792ef1aeSDag-Erling Smørgrav 		close(sd);
391792ef1aeSDag-Erling Smørgrav 	if (cais != NULL)
392792ef1aeSDag-Erling Smørgrav 		freeaddrinfo(cais);
393792ef1aeSDag-Erling Smørgrav 	if (sais != NULL)
394792ef1aeSDag-Erling Smørgrav 		freeaddrinfo(sais);
395792ef1aeSDag-Erling Smørgrav 	return (NULL);
396842a95ccSDag-Erling Smørgrav }
397ce71b736SDag-Erling Smørgrav 
398dcd47379SDag-Erling Smørgrav #ifdef WITH_SSL
399dcd47379SDag-Erling Smørgrav /*
400dcd47379SDag-Erling Smørgrav  * Convert characters A-Z to lowercase (intentionally avoid any locale
401dcd47379SDag-Erling Smørgrav  * specific conversions).
402dcd47379SDag-Erling Smørgrav  */
403dcd47379SDag-Erling Smørgrav static char
404dcd47379SDag-Erling Smørgrav fetch_ssl_tolower(char in)
405dcd47379SDag-Erling Smørgrav {
406dcd47379SDag-Erling Smørgrav 	if (in >= 'A' && in <= 'Z')
407dcd47379SDag-Erling Smørgrav 		return (in + 32);
408dcd47379SDag-Erling Smørgrav 	else
409dcd47379SDag-Erling Smørgrav 		return (in);
410dcd47379SDag-Erling Smørgrav }
411dcd47379SDag-Erling Smørgrav 
412dcd47379SDag-Erling Smørgrav /*
413dcd47379SDag-Erling Smørgrav  * isalpha implementation that intentionally avoids any locale specific
414dcd47379SDag-Erling Smørgrav  * conversions.
415dcd47379SDag-Erling Smørgrav  */
416dcd47379SDag-Erling Smørgrav static int
417dcd47379SDag-Erling Smørgrav fetch_ssl_isalpha(char in)
418dcd47379SDag-Erling Smørgrav {
419dcd47379SDag-Erling Smørgrav 	return ((in >= 'A' && in <= 'Z') || (in >= 'a' && in <= 'z'));
420dcd47379SDag-Erling Smørgrav }
421dcd47379SDag-Erling Smørgrav 
422dcd47379SDag-Erling Smørgrav /*
423dcd47379SDag-Erling Smørgrav  * Check if passed hostnames a and b are equal.
424dcd47379SDag-Erling Smørgrav  */
425dcd47379SDag-Erling Smørgrav static int
426dcd47379SDag-Erling Smørgrav fetch_ssl_hname_equal(const char *a, size_t alen, const char *b,
427dcd47379SDag-Erling Smørgrav     size_t blen)
428dcd47379SDag-Erling Smørgrav {
429dcd47379SDag-Erling Smørgrav 	size_t i;
430dcd47379SDag-Erling Smørgrav 
431dcd47379SDag-Erling Smørgrav 	if (alen != blen)
432dcd47379SDag-Erling Smørgrav 		return (0);
433dcd47379SDag-Erling Smørgrav 	for (i = 0; i < alen; ++i) {
434dcd47379SDag-Erling Smørgrav 		if (fetch_ssl_tolower(a[i]) != fetch_ssl_tolower(b[i]))
435dcd47379SDag-Erling Smørgrav 			return (0);
436dcd47379SDag-Erling Smørgrav 	}
437dcd47379SDag-Erling Smørgrav 	return (1);
438dcd47379SDag-Erling Smørgrav }
439dcd47379SDag-Erling Smørgrav 
440dcd47379SDag-Erling Smørgrav /*
441dcd47379SDag-Erling Smørgrav  * Check if domain label is traditional, meaning that only A-Z, a-z, 0-9
442dcd47379SDag-Erling Smørgrav  * and '-' (hyphen) are allowed. Hyphens have to be surrounded by alpha-
443dcd47379SDag-Erling Smørgrav  * numeric characters. Double hyphens (like they're found in IDN a-labels
444dcd47379SDag-Erling Smørgrav  * 'xn--') are not allowed. Empty labels are invalid.
445dcd47379SDag-Erling Smørgrav  */
446dcd47379SDag-Erling Smørgrav static int
447dcd47379SDag-Erling Smørgrav fetch_ssl_is_trad_domain_label(const char *l, size_t len, int wcok)
448dcd47379SDag-Erling Smørgrav {
449dcd47379SDag-Erling Smørgrav 	size_t i;
450dcd47379SDag-Erling Smørgrav 
451dcd47379SDag-Erling Smørgrav 	if (!len || l[0] == '-' || l[len-1] == '-')
452dcd47379SDag-Erling Smørgrav 		return (0);
453dcd47379SDag-Erling Smørgrav 	for (i = 0; i < len; ++i) {
454dcd47379SDag-Erling Smørgrav 		if (!isdigit(l[i]) &&
455dcd47379SDag-Erling Smørgrav 		    !fetch_ssl_isalpha(l[i]) &&
456dcd47379SDag-Erling Smørgrav 		    !(l[i] == '*' && wcok) &&
457dcd47379SDag-Erling Smørgrav 		    !(l[i] == '-' && l[i - 1] != '-'))
458dcd47379SDag-Erling Smørgrav 			return (0);
459dcd47379SDag-Erling Smørgrav 	}
460dcd47379SDag-Erling Smørgrav 	return (1);
461dcd47379SDag-Erling Smørgrav }
462dcd47379SDag-Erling Smørgrav 
463dcd47379SDag-Erling Smørgrav /*
464dcd47379SDag-Erling Smørgrav  * Check if host name consists only of numbers. This might indicate an IP
465dcd47379SDag-Erling Smørgrav  * address, which is not a good idea for CN wildcard comparison.
466dcd47379SDag-Erling Smørgrav  */
467dcd47379SDag-Erling Smørgrav static int
468dcd47379SDag-Erling Smørgrav fetch_ssl_hname_is_only_numbers(const char *hostname, size_t len)
469dcd47379SDag-Erling Smørgrav {
470dcd47379SDag-Erling Smørgrav 	size_t i;
471dcd47379SDag-Erling Smørgrav 
472dcd47379SDag-Erling Smørgrav 	for (i = 0; i < len; ++i) {
473dcd47379SDag-Erling Smørgrav 		if (!((hostname[i] >= '0' && hostname[i] <= '9') ||
474dcd47379SDag-Erling Smørgrav 		    hostname[i] == '.'))
475dcd47379SDag-Erling Smørgrav 			return (0);
476dcd47379SDag-Erling Smørgrav 	}
477dcd47379SDag-Erling Smørgrav 	return (1);
478dcd47379SDag-Erling Smørgrav }
479dcd47379SDag-Erling Smørgrav 
480dcd47379SDag-Erling Smørgrav /*
481dcd47379SDag-Erling Smørgrav  * Check if the host name h passed matches the pattern passed in m which
482dcd47379SDag-Erling Smørgrav  * is usually part of subjectAltName or CN of a certificate presented to
483dcd47379SDag-Erling Smørgrav  * the client. This includes wildcard matching. The algorithm is based on
484dcd47379SDag-Erling Smørgrav  * RFC6125, sections 6.4.3 and 7.2, which clarifies RFC2818 and RFC3280.
485dcd47379SDag-Erling Smørgrav  */
486dcd47379SDag-Erling Smørgrav static int
487dcd47379SDag-Erling Smørgrav fetch_ssl_hname_match(const char *h, size_t hlen, const char *m,
488dcd47379SDag-Erling Smørgrav     size_t mlen)
489dcd47379SDag-Erling Smørgrav {
490dcd47379SDag-Erling Smørgrav 	int delta, hdotidx, mdot1idx, wcidx;
491dcd47379SDag-Erling Smørgrav 	const char *hdot, *mdot1, *mdot2;
492dcd47379SDag-Erling Smørgrav 	const char *wc; /* wildcard */
493dcd47379SDag-Erling Smørgrav 
494dcd47379SDag-Erling Smørgrav 	if (!(h && *h && m && *m))
495dcd47379SDag-Erling Smørgrav 		return (0);
496dcd47379SDag-Erling Smørgrav 	if ((wc = strnstr(m, "*", mlen)) == NULL)
497dcd47379SDag-Erling Smørgrav 		return (fetch_ssl_hname_equal(h, hlen, m, mlen));
498dcd47379SDag-Erling Smørgrav 	wcidx = wc - m;
499dcd47379SDag-Erling Smørgrav 	/* hostname should not be just dots and numbers */
500dcd47379SDag-Erling Smørgrav 	if (fetch_ssl_hname_is_only_numbers(h, hlen))
501dcd47379SDag-Erling Smørgrav 		return (0);
502dcd47379SDag-Erling Smørgrav 	/* only one wildcard allowed in pattern */
503dcd47379SDag-Erling Smørgrav 	if (strnstr(wc + 1, "*", mlen - wcidx - 1) != NULL)
504dcd47379SDag-Erling Smørgrav 		return (0);
505dcd47379SDag-Erling Smørgrav 	/*
506dcd47379SDag-Erling Smørgrav 	 * there must be at least two more domain labels and
507dcd47379SDag-Erling Smørgrav 	 * wildcard has to be in the leftmost label (RFC6125)
508dcd47379SDag-Erling Smørgrav 	 */
509dcd47379SDag-Erling Smørgrav 	mdot1 = strnstr(m, ".", mlen);
510dcd47379SDag-Erling Smørgrav 	if (mdot1 == NULL || mdot1 < wc || (mlen - (mdot1 - m)) < 4)
511dcd47379SDag-Erling Smørgrav 		return (0);
512dcd47379SDag-Erling Smørgrav 	mdot1idx = mdot1 - m;
513dcd47379SDag-Erling Smørgrav 	mdot2 = strnstr(mdot1 + 1, ".", mlen - mdot1idx - 1);
514dcd47379SDag-Erling Smørgrav 	if (mdot2 == NULL || (mlen - (mdot2 - m)) < 2)
515dcd47379SDag-Erling Smørgrav 		return (0);
516dcd47379SDag-Erling Smørgrav 	/* hostname must contain a dot and not be the 1st char */
517dcd47379SDag-Erling Smørgrav 	hdot = strnstr(h, ".", hlen);
518dcd47379SDag-Erling Smørgrav 	if (hdot == NULL || hdot == h)
519dcd47379SDag-Erling Smørgrav 		return (0);
520dcd47379SDag-Erling Smørgrav 	hdotidx = hdot - h;
521dcd47379SDag-Erling Smørgrav 	/*
522dcd47379SDag-Erling Smørgrav 	 * host part of hostname must be at least as long as
523dcd47379SDag-Erling Smørgrav 	 * pattern it's supposed to match
524dcd47379SDag-Erling Smørgrav 	 */
525dcd47379SDag-Erling Smørgrav 	if (hdotidx < mdot1idx)
526dcd47379SDag-Erling Smørgrav 		return (0);
527dcd47379SDag-Erling Smørgrav 	/*
528dcd47379SDag-Erling Smørgrav 	 * don't allow wildcards in non-traditional domain names
529dcd47379SDag-Erling Smørgrav 	 * (IDN, A-label, U-label...)
530dcd47379SDag-Erling Smørgrav 	 */
531dcd47379SDag-Erling Smørgrav 	if (!fetch_ssl_is_trad_domain_label(h, hdotidx, 0) ||
532dcd47379SDag-Erling Smørgrav 	    !fetch_ssl_is_trad_domain_label(m, mdot1idx, 1))
533dcd47379SDag-Erling Smørgrav 		return (0);
534dcd47379SDag-Erling Smørgrav 	/* match domain part (part after first dot) */
535dcd47379SDag-Erling Smørgrav 	if (!fetch_ssl_hname_equal(hdot, hlen - hdotidx, mdot1,
536dcd47379SDag-Erling Smørgrav 	    mlen - mdot1idx))
537dcd47379SDag-Erling Smørgrav 		return (0);
538dcd47379SDag-Erling Smørgrav 	/* match part left of wildcard */
539dcd47379SDag-Erling Smørgrav 	if (!fetch_ssl_hname_equal(h, wcidx, m, wcidx))
540dcd47379SDag-Erling Smørgrav 		return (0);
541dcd47379SDag-Erling Smørgrav 	/* match part right of wildcard */
542dcd47379SDag-Erling Smørgrav 	delta = mdot1idx - wcidx - 1;
543dcd47379SDag-Erling Smørgrav 	if (!fetch_ssl_hname_equal(hdot - delta, delta,
544dcd47379SDag-Erling Smørgrav 	    mdot1 - delta, delta))
545dcd47379SDag-Erling Smørgrav 		return (0);
54675f46cf6SPedro F. Giffuni 	/* all tests succeeded, it's a match */
547dcd47379SDag-Erling Smørgrav 	return (1);
548dcd47379SDag-Erling Smørgrav }
549dcd47379SDag-Erling Smørgrav 
550dcd47379SDag-Erling Smørgrav /*
551dcd47379SDag-Erling Smørgrav  * Get numeric host address info - returns NULL if host was not an IP
552dcd47379SDag-Erling Smørgrav  * address. The caller is responsible for deallocation using
553dcd47379SDag-Erling Smørgrav  * freeaddrinfo(3).
554dcd47379SDag-Erling Smørgrav  */
555dcd47379SDag-Erling Smørgrav static struct addrinfo *
556dcd47379SDag-Erling Smørgrav fetch_ssl_get_numeric_addrinfo(const char *hostname, size_t len)
557dcd47379SDag-Erling Smørgrav {
558dcd47379SDag-Erling Smørgrav 	struct addrinfo hints, *res;
559dcd47379SDag-Erling Smørgrav 	char *host;
560dcd47379SDag-Erling Smørgrav 
561dcd47379SDag-Erling Smørgrav 	host = (char *)malloc(len + 1);
562dcd47379SDag-Erling Smørgrav 	memcpy(host, hostname, len);
563dcd47379SDag-Erling Smørgrav 	host[len] = '\0';
564dcd47379SDag-Erling Smørgrav 	memset(&hints, 0, sizeof(hints));
565dcd47379SDag-Erling Smørgrav 	hints.ai_family = PF_UNSPEC;
566dcd47379SDag-Erling Smørgrav 	hints.ai_socktype = SOCK_STREAM;
567dcd47379SDag-Erling Smørgrav 	hints.ai_protocol = 0;
568dcd47379SDag-Erling Smørgrav 	hints.ai_flags = AI_NUMERICHOST;
569dcd47379SDag-Erling Smørgrav 	/* port is not relevant for this purpose */
570034fb271SMichael Gmelin 	if (getaddrinfo(host, "443", &hints, &res) != 0)
571034fb271SMichael Gmelin 		res = NULL;
572dcd47379SDag-Erling Smørgrav 	free(host);
573dcd47379SDag-Erling Smørgrav 	return res;
574dcd47379SDag-Erling Smørgrav }
575dcd47379SDag-Erling Smørgrav 
576dcd47379SDag-Erling Smørgrav /*
577dcd47379SDag-Erling Smørgrav  * Compare ip address in addrinfo with address passes.
578dcd47379SDag-Erling Smørgrav  */
579dcd47379SDag-Erling Smørgrav static int
580dcd47379SDag-Erling Smørgrav fetch_ssl_ipaddr_match_bin(const struct addrinfo *lhost, const char *rhost,
581dcd47379SDag-Erling Smørgrav     size_t rhostlen)
582dcd47379SDag-Erling Smørgrav {
583dcd47379SDag-Erling Smørgrav 	const void *left;
584dcd47379SDag-Erling Smørgrav 
585dcd47379SDag-Erling Smørgrav 	if (lhost->ai_family == AF_INET && rhostlen == 4) {
586dcd47379SDag-Erling Smørgrav 		left = (void *)&((struct sockaddr_in*)(void *)
587dcd47379SDag-Erling Smørgrav 		    lhost->ai_addr)->sin_addr.s_addr;
588dcd47379SDag-Erling Smørgrav #ifdef INET6
589dcd47379SDag-Erling Smørgrav 	} else if (lhost->ai_family == AF_INET6 && rhostlen == 16) {
590dcd47379SDag-Erling Smørgrav 		left = (void *)&((struct sockaddr_in6 *)(void *)
591dcd47379SDag-Erling Smørgrav 		    lhost->ai_addr)->sin6_addr;
592dcd47379SDag-Erling Smørgrav #endif
593dcd47379SDag-Erling Smørgrav 	} else
594dcd47379SDag-Erling Smørgrav 		return (0);
595dcd47379SDag-Erling Smørgrav 	return (!memcmp(left, (const void *)rhost, rhostlen) ? 1 : 0);
596dcd47379SDag-Erling Smørgrav }
597dcd47379SDag-Erling Smørgrav 
598dcd47379SDag-Erling Smørgrav /*
599dcd47379SDag-Erling Smørgrav  * Compare ip address in addrinfo with host passed. If host is not an IP
600dcd47379SDag-Erling Smørgrav  * address, comparison will fail.
601dcd47379SDag-Erling Smørgrav  */
602dcd47379SDag-Erling Smørgrav static int
603dcd47379SDag-Erling Smørgrav fetch_ssl_ipaddr_match(const struct addrinfo *laddr, const char *r,
604dcd47379SDag-Erling Smørgrav     size_t rlen)
605dcd47379SDag-Erling Smørgrav {
606dcd47379SDag-Erling Smørgrav 	struct addrinfo *raddr;
607dcd47379SDag-Erling Smørgrav 	int ret;
608dcd47379SDag-Erling Smørgrav 	char *rip;
609dcd47379SDag-Erling Smørgrav 
610dcd47379SDag-Erling Smørgrav 	ret = 0;
611dcd47379SDag-Erling Smørgrav 	if ((raddr = fetch_ssl_get_numeric_addrinfo(r, rlen)) == NULL)
612dcd47379SDag-Erling Smørgrav 		return 0; /* not a numeric host */
613dcd47379SDag-Erling Smørgrav 
614dcd47379SDag-Erling Smørgrav 	if (laddr->ai_family == raddr->ai_family) {
615dcd47379SDag-Erling Smørgrav 		if (laddr->ai_family == AF_INET) {
616dcd47379SDag-Erling Smørgrav 			rip = (char *)&((struct sockaddr_in *)(void *)
617dcd47379SDag-Erling Smørgrav 			    raddr->ai_addr)->sin_addr.s_addr;
618dcd47379SDag-Erling Smørgrav 			ret = fetch_ssl_ipaddr_match_bin(laddr, rip, 4);
619dcd47379SDag-Erling Smørgrav #ifdef INET6
620dcd47379SDag-Erling Smørgrav 		} else if (laddr->ai_family == AF_INET6) {
621dcd47379SDag-Erling Smørgrav 			rip = (char *)&((struct sockaddr_in6 *)(void *)
622dcd47379SDag-Erling Smørgrav 			    raddr->ai_addr)->sin6_addr;
623dcd47379SDag-Erling Smørgrav 			ret = fetch_ssl_ipaddr_match_bin(laddr, rip, 16);
624dcd47379SDag-Erling Smørgrav #endif
625dcd47379SDag-Erling Smørgrav 		}
626dcd47379SDag-Erling Smørgrav 
627dcd47379SDag-Erling Smørgrav 	}
628dcd47379SDag-Erling Smørgrav 	freeaddrinfo(raddr);
629dcd47379SDag-Erling Smørgrav 	return (ret);
630dcd47379SDag-Erling Smørgrav }
631dcd47379SDag-Erling Smørgrav 
632dcd47379SDag-Erling Smørgrav /*
633dcd47379SDag-Erling Smørgrav  * Verify server certificate by subjectAltName.
634dcd47379SDag-Erling Smørgrav  */
635dcd47379SDag-Erling Smørgrav static int
636dcd47379SDag-Erling Smørgrav fetch_ssl_verify_altname(STACK_OF(GENERAL_NAME) *altnames,
637dcd47379SDag-Erling Smørgrav     const char *host, struct addrinfo *ip)
638dcd47379SDag-Erling Smørgrav {
639dcd47379SDag-Erling Smørgrav 	const GENERAL_NAME *name;
640dcd47379SDag-Erling Smørgrav 	size_t nslen;
641dcd47379SDag-Erling Smørgrav 	int i;
642dcd47379SDag-Erling Smørgrav 	const char *ns;
643dcd47379SDag-Erling Smørgrav 
644dcd47379SDag-Erling Smørgrav 	for (i = 0; i < sk_GENERAL_NAME_num(altnames); ++i) {
645dcd47379SDag-Erling Smørgrav #if OPENSSL_VERSION_NUMBER < 0x10000000L
646dcd47379SDag-Erling Smørgrav 		/*
647dcd47379SDag-Erling Smørgrav 		 * This is a workaround, since the following line causes
648dcd47379SDag-Erling Smørgrav 		 * alignment issues in clang:
649dcd47379SDag-Erling Smørgrav 		 * name = sk_GENERAL_NAME_value(altnames, i);
650dcd47379SDag-Erling Smørgrav 		 * OpenSSL explicitly warns not to use those macros
651dcd47379SDag-Erling Smørgrav 		 * directly, but there isn't much choice (and there
652dcd47379SDag-Erling Smørgrav 		 * shouldn't be any ill side effects)
653dcd47379SDag-Erling Smørgrav 		 */
654dcd47379SDag-Erling Smørgrav 		name = (GENERAL_NAME *)SKM_sk_value(void, altnames, i);
655dcd47379SDag-Erling Smørgrav #else
656dcd47379SDag-Erling Smørgrav 		name = sk_GENERAL_NAME_value(altnames, i);
657dcd47379SDag-Erling Smørgrav #endif
658dcd47379SDag-Erling Smørgrav 		ns = (const char *)ASN1_STRING_data(name->d.ia5);
659dcd47379SDag-Erling Smørgrav 		nslen = (size_t)ASN1_STRING_length(name->d.ia5);
660dcd47379SDag-Erling Smørgrav 
661dcd47379SDag-Erling Smørgrav 		if (name->type == GEN_DNS && ip == NULL &&
662dcd47379SDag-Erling Smørgrav 		    fetch_ssl_hname_match(host, strlen(host), ns, nslen))
663dcd47379SDag-Erling Smørgrav 			return (1);
664dcd47379SDag-Erling Smørgrav 		else if (name->type == GEN_IPADD && ip != NULL &&
665dcd47379SDag-Erling Smørgrav 		    fetch_ssl_ipaddr_match_bin(ip, ns, nslen))
666dcd47379SDag-Erling Smørgrav 			return (1);
667dcd47379SDag-Erling Smørgrav 	}
668dcd47379SDag-Erling Smørgrav 	return (0);
669dcd47379SDag-Erling Smørgrav }
670dcd47379SDag-Erling Smørgrav 
671dcd47379SDag-Erling Smørgrav /*
672dcd47379SDag-Erling Smørgrav  * Verify server certificate by CN.
673dcd47379SDag-Erling Smørgrav  */
674dcd47379SDag-Erling Smørgrav static int
675dcd47379SDag-Erling Smørgrav fetch_ssl_verify_cn(X509_NAME *subject, const char *host,
676dcd47379SDag-Erling Smørgrav     struct addrinfo *ip)
677dcd47379SDag-Erling Smørgrav {
678dcd47379SDag-Erling Smørgrav 	ASN1_STRING *namedata;
679dcd47379SDag-Erling Smørgrav 	X509_NAME_ENTRY *nameentry;
680dcd47379SDag-Erling Smørgrav 	int cnlen, lastpos, loc, ret;
681dcd47379SDag-Erling Smørgrav 	unsigned char *cn;
682dcd47379SDag-Erling Smørgrav 
683dcd47379SDag-Erling Smørgrav 	ret = 0;
684dcd47379SDag-Erling Smørgrav 	lastpos = -1;
685dcd47379SDag-Erling Smørgrav 	loc = -1;
686dcd47379SDag-Erling Smørgrav 	cn = NULL;
687dcd47379SDag-Erling Smørgrav 	/* get most specific CN (last entry in list) and compare */
688dcd47379SDag-Erling Smørgrav 	while ((lastpos = X509_NAME_get_index_by_NID(subject,
689dcd47379SDag-Erling Smørgrav 	    NID_commonName, lastpos)) != -1)
690dcd47379SDag-Erling Smørgrav 		loc = lastpos;
691dcd47379SDag-Erling Smørgrav 
692dcd47379SDag-Erling Smørgrav 	if (loc > -1) {
693dcd47379SDag-Erling Smørgrav 		nameentry = X509_NAME_get_entry(subject, loc);
694dcd47379SDag-Erling Smørgrav 		namedata = X509_NAME_ENTRY_get_data(nameentry);
695dcd47379SDag-Erling Smørgrav 		cnlen = ASN1_STRING_to_UTF8(&cn, namedata);
696dcd47379SDag-Erling Smørgrav 		if (ip == NULL &&
697dcd47379SDag-Erling Smørgrav 		    fetch_ssl_hname_match(host, strlen(host), cn, cnlen))
698dcd47379SDag-Erling Smørgrav 			ret = 1;
699dcd47379SDag-Erling Smørgrav 		else if (ip != NULL && fetch_ssl_ipaddr_match(ip, cn, cnlen))
700dcd47379SDag-Erling Smørgrav 			ret = 1;
701dcd47379SDag-Erling Smørgrav 		OPENSSL_free(cn);
702dcd47379SDag-Erling Smørgrav 	}
703dcd47379SDag-Erling Smørgrav 	return (ret);
704dcd47379SDag-Erling Smørgrav }
705dcd47379SDag-Erling Smørgrav 
706dcd47379SDag-Erling Smørgrav /*
707dcd47379SDag-Erling Smørgrav  * Verify that server certificate subjectAltName/CN matches
708dcd47379SDag-Erling Smørgrav  * hostname. First check, if there are alternative subject names. If yes,
709dcd47379SDag-Erling Smørgrav  * those have to match. Only if those don't exist it falls back to
710dcd47379SDag-Erling Smørgrav  * checking the subject's CN.
711dcd47379SDag-Erling Smørgrav  */
712dcd47379SDag-Erling Smørgrav static int
713dcd47379SDag-Erling Smørgrav fetch_ssl_verify_hname(X509 *cert, const char *host)
714dcd47379SDag-Erling Smørgrav {
715dcd47379SDag-Erling Smørgrav 	struct addrinfo *ip;
716dcd47379SDag-Erling Smørgrav 	STACK_OF(GENERAL_NAME) *altnames;
717dcd47379SDag-Erling Smørgrav 	X509_NAME *subject;
718dcd47379SDag-Erling Smørgrav 	int ret;
719dcd47379SDag-Erling Smørgrav 
720dcd47379SDag-Erling Smørgrav 	ret = 0;
721dcd47379SDag-Erling Smørgrav 	ip = fetch_ssl_get_numeric_addrinfo(host, strlen(host));
722dcd47379SDag-Erling Smørgrav 	altnames = X509_get_ext_d2i(cert, NID_subject_alt_name,
723dcd47379SDag-Erling Smørgrav 	    NULL, NULL);
724dcd47379SDag-Erling Smørgrav 
725dcd47379SDag-Erling Smørgrav 	if (altnames != NULL) {
726dcd47379SDag-Erling Smørgrav 		ret = fetch_ssl_verify_altname(altnames, host, ip);
727dcd47379SDag-Erling Smørgrav 	} else {
728dcd47379SDag-Erling Smørgrav 		subject = X509_get_subject_name(cert);
729dcd47379SDag-Erling Smørgrav 		if (subject != NULL)
730dcd47379SDag-Erling Smørgrav 			ret = fetch_ssl_verify_cn(subject, host, ip);
731dcd47379SDag-Erling Smørgrav 	}
732dcd47379SDag-Erling Smørgrav 
733dcd47379SDag-Erling Smørgrav 	if (ip != NULL)
734dcd47379SDag-Erling Smørgrav 		freeaddrinfo(ip);
735dcd47379SDag-Erling Smørgrav 	if (altnames != NULL)
736dcd47379SDag-Erling Smørgrav 		GENERAL_NAMES_free(altnames);
737dcd47379SDag-Erling Smørgrav 	return (ret);
738dcd47379SDag-Erling Smørgrav }
739dcd47379SDag-Erling Smørgrav 
740dcd47379SDag-Erling Smørgrav /*
741dcd47379SDag-Erling Smørgrav  * Configure transport security layer based on environment.
742dcd47379SDag-Erling Smørgrav  */
743dcd47379SDag-Erling Smørgrav static void
744dcd47379SDag-Erling Smørgrav fetch_ssl_setup_transport_layer(SSL_CTX *ctx, int verbose)
745dcd47379SDag-Erling Smørgrav {
746dcd47379SDag-Erling Smørgrav 	long ssl_ctx_options;
747dcd47379SDag-Erling Smørgrav 
74801ed3ca3SJung-uk Kim 	ssl_ctx_options = SSL_OP_ALL | SSL_OP_NO_SSLv2 | SSL_OP_NO_TICKET;
74976720765SDag-Erling Smørgrav 	if (getenv("SSL_ALLOW_SSL3") == NULL)
750dcd47379SDag-Erling Smørgrav 		ssl_ctx_options |= SSL_OP_NO_SSLv3;
751dcd47379SDag-Erling Smørgrav 	if (getenv("SSL_NO_TLS1") != NULL)
752dcd47379SDag-Erling Smørgrav 		ssl_ctx_options |= SSL_OP_NO_TLSv1;
75376720765SDag-Erling Smørgrav 	if (getenv("SSL_NO_TLS1_1") != NULL)
75476720765SDag-Erling Smørgrav 		ssl_ctx_options |= SSL_OP_NO_TLSv1_1;
75576720765SDag-Erling Smørgrav 	if (getenv("SSL_NO_TLS1_2") != NULL)
75676720765SDag-Erling Smørgrav 		ssl_ctx_options |= SSL_OP_NO_TLSv1_2;
757dcd47379SDag-Erling Smørgrav 	if (verbose)
7583443b2bcSDag-Erling Smørgrav 		fetch_info("SSL options: %lx", ssl_ctx_options);
759dcd47379SDag-Erling Smørgrav 	SSL_CTX_set_options(ctx, ssl_ctx_options);
760dcd47379SDag-Erling Smørgrav }
761dcd47379SDag-Erling Smørgrav 
762dcd47379SDag-Erling Smørgrav 
763dcd47379SDag-Erling Smørgrav /*
764dcd47379SDag-Erling Smørgrav  * Configure peer verification based on environment.
765dcd47379SDag-Erling Smørgrav  */
766e9de92adSDag-Erling Smørgrav #define LOCAL_CERT_FILE	"/usr/local/etc/ssl/cert.pem"
767e9de92adSDag-Erling Smørgrav #define BASE_CERT_FILE	"/etc/ssl/cert.pem"
768dcd47379SDag-Erling Smørgrav static int
769dcd47379SDag-Erling Smørgrav fetch_ssl_setup_peer_verification(SSL_CTX *ctx, int verbose)
770dcd47379SDag-Erling Smørgrav {
771dcd47379SDag-Erling Smørgrav 	X509_LOOKUP *crl_lookup;
772dcd47379SDag-Erling Smørgrav 	X509_STORE *crl_store;
773dcd47379SDag-Erling Smørgrav 	const char *ca_cert_file, *ca_cert_path, *crl_file;
774dcd47379SDag-Erling Smørgrav 
775dcd47379SDag-Erling Smørgrav 	if (getenv("SSL_NO_VERIFY_PEER") == NULL) {
776e9de92adSDag-Erling Smørgrav 		ca_cert_file = getenv("SSL_CA_CERT_FILE");
777e9de92adSDag-Erling Smørgrav 		if (ca_cert_file == NULL &&
778e9de92adSDag-Erling Smørgrav 		    access(LOCAL_CERT_FILE, R_OK) == 0)
779e9de92adSDag-Erling Smørgrav 			ca_cert_file = LOCAL_CERT_FILE;
780f16550adSBaptiste Daroussin 		if (ca_cert_file == NULL &&
781f16550adSBaptiste Daroussin 		    access(BASE_CERT_FILE, R_OK) == 0)
782e9de92adSDag-Erling Smørgrav 			ca_cert_file = BASE_CERT_FILE;
783dcd47379SDag-Erling Smørgrav 		ca_cert_path = getenv("SSL_CA_CERT_PATH");
784dcd47379SDag-Erling Smørgrav 		if (verbose) {
785dcd47379SDag-Erling Smørgrav 			fetch_info("Peer verification enabled");
786dcd47379SDag-Erling Smørgrav 			if (ca_cert_file != NULL)
787dcd47379SDag-Erling Smørgrav 				fetch_info("Using CA cert file: %s",
788dcd47379SDag-Erling Smørgrav 				    ca_cert_file);
789dcd47379SDag-Erling Smørgrav 			if (ca_cert_path != NULL)
790dcd47379SDag-Erling Smørgrav 				fetch_info("Using CA cert path: %s",
791dcd47379SDag-Erling Smørgrav 				    ca_cert_path);
792f16550adSBaptiste Daroussin 			if (ca_cert_file == NULL && ca_cert_path == NULL)
793f16550adSBaptiste Daroussin 				fetch_info("Using OpenSSL default "
794f16550adSBaptiste Daroussin 				    "CA cert file and path");
795dcd47379SDag-Erling Smørgrav 		}
796dcd47379SDag-Erling Smørgrav 		SSL_CTX_set_verify(ctx, SSL_VERIFY_PEER,
797dcd47379SDag-Erling Smørgrav 		    fetch_ssl_cb_verify_crt);
798f16550adSBaptiste Daroussin 		if (ca_cert_file != NULL || ca_cert_path != NULL)
799dcd47379SDag-Erling Smørgrav 			SSL_CTX_load_verify_locations(ctx, ca_cert_file,
800dcd47379SDag-Erling Smørgrav 			    ca_cert_path);
801f16550adSBaptiste Daroussin 		else
802f16550adSBaptiste Daroussin 			SSL_CTX_set_default_verify_paths(ctx);
803dcd47379SDag-Erling Smørgrav 		if ((crl_file = getenv("SSL_CRL_FILE")) != NULL) {
804dcd47379SDag-Erling Smørgrav 			if (verbose)
805dcd47379SDag-Erling Smørgrav 				fetch_info("Using CRL file: %s", crl_file);
806dcd47379SDag-Erling Smørgrav 			crl_store = SSL_CTX_get_cert_store(ctx);
807dcd47379SDag-Erling Smørgrav 			crl_lookup = X509_STORE_add_lookup(crl_store,
808dcd47379SDag-Erling Smørgrav 			    X509_LOOKUP_file());
809dcd47379SDag-Erling Smørgrav 			if (crl_lookup == NULL ||
810dcd47379SDag-Erling Smørgrav 			    !X509_load_crl_file(crl_lookup, crl_file,
811dcd47379SDag-Erling Smørgrav 				X509_FILETYPE_PEM)) {
812dcd47379SDag-Erling Smørgrav 				fprintf(stderr,
813dcd47379SDag-Erling Smørgrav 				    "Could not load CRL file %s\n",
814dcd47379SDag-Erling Smørgrav 				    crl_file);
815dcd47379SDag-Erling Smørgrav 				return (0);
816dcd47379SDag-Erling Smørgrav 			}
817dcd47379SDag-Erling Smørgrav 			X509_STORE_set_flags(crl_store,
818dcd47379SDag-Erling Smørgrav 			    X509_V_FLAG_CRL_CHECK |
819dcd47379SDag-Erling Smørgrav 			    X509_V_FLAG_CRL_CHECK_ALL);
820dcd47379SDag-Erling Smørgrav 		}
821dcd47379SDag-Erling Smørgrav 	}
822dcd47379SDag-Erling Smørgrav 	return (1);
823dcd47379SDag-Erling Smørgrav }
824dcd47379SDag-Erling Smørgrav 
825dcd47379SDag-Erling Smørgrav /*
826dcd47379SDag-Erling Smørgrav  * Configure client certificate based on environment.
827dcd47379SDag-Erling Smørgrav  */
828dcd47379SDag-Erling Smørgrav static int
829dcd47379SDag-Erling Smørgrav fetch_ssl_setup_client_certificate(SSL_CTX *ctx, int verbose)
830dcd47379SDag-Erling Smørgrav {
831dcd47379SDag-Erling Smørgrav 	const char *client_cert_file, *client_key_file;
832dcd47379SDag-Erling Smørgrav 
833dcd47379SDag-Erling Smørgrav 	if ((client_cert_file = getenv("SSL_CLIENT_CERT_FILE")) != NULL) {
834dcd47379SDag-Erling Smørgrav 		client_key_file = getenv("SSL_CLIENT_KEY_FILE") != NULL ?
835dcd47379SDag-Erling Smørgrav 		    getenv("SSL_CLIENT_KEY_FILE") : client_cert_file;
836dcd47379SDag-Erling Smørgrav 		if (verbose) {
837dcd47379SDag-Erling Smørgrav 			fetch_info("Using client cert file: %s",
838dcd47379SDag-Erling Smørgrav 			    client_cert_file);
839dcd47379SDag-Erling Smørgrav 			fetch_info("Using client key file: %s",
840dcd47379SDag-Erling Smørgrav 			    client_key_file);
841dcd47379SDag-Erling Smørgrav 		}
842dcd47379SDag-Erling Smørgrav 		if (SSL_CTX_use_certificate_chain_file(ctx,
843dcd47379SDag-Erling Smørgrav 			client_cert_file) != 1) {
844dcd47379SDag-Erling Smørgrav 			fprintf(stderr,
845dcd47379SDag-Erling Smørgrav 			    "Could not load client certificate %s\n",
846dcd47379SDag-Erling Smørgrav 			    client_cert_file);
847dcd47379SDag-Erling Smørgrav 			return (0);
848dcd47379SDag-Erling Smørgrav 		}
849dcd47379SDag-Erling Smørgrav 		if (SSL_CTX_use_PrivateKey_file(ctx, client_key_file,
850dcd47379SDag-Erling Smørgrav 			SSL_FILETYPE_PEM) != 1) {
851dcd47379SDag-Erling Smørgrav 			fprintf(stderr,
852dcd47379SDag-Erling Smørgrav 			    "Could not load client key %s\n",
853dcd47379SDag-Erling Smørgrav 			    client_key_file);
854dcd47379SDag-Erling Smørgrav 			return (0);
855dcd47379SDag-Erling Smørgrav 		}
856dcd47379SDag-Erling Smørgrav 	}
857dcd47379SDag-Erling Smørgrav 	return (1);
858dcd47379SDag-Erling Smørgrav }
859dcd47379SDag-Erling Smørgrav 
860dcd47379SDag-Erling Smørgrav /*
861dcd47379SDag-Erling Smørgrav  * Callback for SSL certificate verification, this is called on server
862dcd47379SDag-Erling Smørgrav  * cert verification. It takes no decision, but informs the user in case
863dcd47379SDag-Erling Smørgrav  * verification failed.
864dcd47379SDag-Erling Smørgrav  */
865dcd47379SDag-Erling Smørgrav int
866dcd47379SDag-Erling Smørgrav fetch_ssl_cb_verify_crt(int verified, X509_STORE_CTX *ctx)
867dcd47379SDag-Erling Smørgrav {
868dcd47379SDag-Erling Smørgrav 	X509 *crt;
869dcd47379SDag-Erling Smørgrav 	X509_NAME *name;
870dcd47379SDag-Erling Smørgrav 	char *str;
871dcd47379SDag-Erling Smørgrav 
872dcd47379SDag-Erling Smørgrav 	str = NULL;
873dcd47379SDag-Erling Smørgrav 	if (!verified) {
874dcd47379SDag-Erling Smørgrav 		if ((crt = X509_STORE_CTX_get_current_cert(ctx)) != NULL &&
875dcd47379SDag-Erling Smørgrav 		    (name = X509_get_subject_name(crt)) != NULL)
876dcd47379SDag-Erling Smørgrav 			str = X509_NAME_oneline(name, 0, 0);
877dcd47379SDag-Erling Smørgrav 		fprintf(stderr, "Certificate verification failed for %s\n",
878dcd47379SDag-Erling Smørgrav 		    str != NULL ? str : "no relevant certificate");
879dcd47379SDag-Erling Smørgrav 		OPENSSL_free(str);
880dcd47379SDag-Erling Smørgrav 	}
881dcd47379SDag-Erling Smørgrav 	return (verified);
882dcd47379SDag-Erling Smørgrav }
883dcd47379SDag-Erling Smørgrav 
884dcd47379SDag-Erling Smørgrav #endif
885ce71b736SDag-Erling Smørgrav 
886fc6e9e65SDag-Erling Smørgrav /*
887111e2510SDag-Erling Smørgrav  * Enable SSL on a connection.
888111e2510SDag-Erling Smørgrav  */
889111e2510SDag-Erling Smørgrav int
890dcd47379SDag-Erling Smørgrav fetch_ssl(conn_t *conn, const struct url *URL, int verbose)
891111e2510SDag-Erling Smørgrav {
892a9d0c849SEd Maste #ifdef WITH_SSL
893caaffed8SDag-Erling Smørgrav 	int ret, ssl_err;
894dcd47379SDag-Erling Smørgrav 	X509_NAME *name;
895dcd47379SDag-Erling Smørgrav 	char *str;
896111e2510SDag-Erling Smørgrav 
897111e2510SDag-Erling Smørgrav 	/* Init the SSL library and context */
898111e2510SDag-Erling Smørgrav 	if (!SSL_library_init()){
899111e2510SDag-Erling Smørgrav 		fprintf(stderr, "SSL library init failed\n");
900111e2510SDag-Erling Smørgrav 		return (-1);
901111e2510SDag-Erling Smørgrav 	}
902111e2510SDag-Erling Smørgrav 
903111e2510SDag-Erling Smørgrav 	SSL_load_error_strings();
904111e2510SDag-Erling Smørgrav 
90576720765SDag-Erling Smørgrav 	conn->ssl_meth = SSLv23_client_method();
906111e2510SDag-Erling Smørgrav 	conn->ssl_ctx = SSL_CTX_new(conn->ssl_meth);
907f8020ddeSDag-Erling Smørgrav 	SSL_CTX_set_mode(conn->ssl_ctx, SSL_MODE_AUTO_RETRY);
908111e2510SDag-Erling Smørgrav 
909dcd47379SDag-Erling Smørgrav 	fetch_ssl_setup_transport_layer(conn->ssl_ctx, verbose);
910dcd47379SDag-Erling Smørgrav 	if (!fetch_ssl_setup_peer_verification(conn->ssl_ctx, verbose))
911dcd47379SDag-Erling Smørgrav 		return (-1);
912dcd47379SDag-Erling Smørgrav 	if (!fetch_ssl_setup_client_certificate(conn->ssl_ctx, verbose))
913dcd47379SDag-Erling Smørgrav 		return (-1);
914dcd47379SDag-Erling Smørgrav 
915111e2510SDag-Erling Smørgrav 	conn->ssl = SSL_new(conn->ssl_ctx);
916111e2510SDag-Erling Smørgrav 	if (conn->ssl == NULL) {
917111e2510SDag-Erling Smørgrav 		fprintf(stderr, "SSL context creation failed\n");
918111e2510SDag-Erling Smørgrav 		return (-1);
919111e2510SDag-Erling Smørgrav 	}
920111e2510SDag-Erling Smørgrav 	SSL_set_fd(conn->ssl, conn->sd);
9214cb272a0SBryan Drewery 
9224cb272a0SBryan Drewery #if OPENSSL_VERSION_NUMBER >= 0x0090806fL && !defined(OPENSSL_NO_TLSEXT)
9235313966eSBryan Drewery 	if (!SSL_set_tlsext_host_name(conn->ssl,
9245313966eSBryan Drewery 	    __DECONST(struct url *, URL)->host)) {
9254cb272a0SBryan Drewery 		fprintf(stderr,
9264cb272a0SBryan Drewery 		    "TLS server name indication extension failed for host %s\n",
9274cb272a0SBryan Drewery 		    URL->host);
9284cb272a0SBryan Drewery 		return (-1);
9294cb272a0SBryan Drewery 	}
9304cb272a0SBryan Drewery #endif
931caaffed8SDag-Erling Smørgrav 	while ((ret = SSL_connect(conn->ssl)) == -1) {
932caaffed8SDag-Erling Smørgrav 		ssl_err = SSL_get_error(conn->ssl, ret);
933caaffed8SDag-Erling Smørgrav 		if (ssl_err != SSL_ERROR_WANT_READ &&
934caaffed8SDag-Erling Smørgrav 		    ssl_err != SSL_ERROR_WANT_WRITE) {
935111e2510SDag-Erling Smørgrav 			ERR_print_errors_fp(stderr);
936111e2510SDag-Erling Smørgrav 			return (-1);
937111e2510SDag-Erling Smørgrav 		}
938caaffed8SDag-Erling Smørgrav 	}
939dcd47379SDag-Erling Smørgrav 	conn->ssl_cert = SSL_get_peer_certificate(conn->ssl);
940dcd47379SDag-Erling Smørgrav 
941dcd47379SDag-Erling Smørgrav 	if (conn->ssl_cert == NULL) {
942dcd47379SDag-Erling Smørgrav 		fprintf(stderr, "No server SSL certificate\n");
943dcd47379SDag-Erling Smørgrav 		return (-1);
944dcd47379SDag-Erling Smørgrav 	}
945dcd47379SDag-Erling Smørgrav 
946dcd47379SDag-Erling Smørgrav 	if (getenv("SSL_NO_VERIFY_HOSTNAME") == NULL) {
947dcd47379SDag-Erling Smørgrav 		if (verbose)
948dcd47379SDag-Erling Smørgrav 			fetch_info("Verify hostname");
949dcd47379SDag-Erling Smørgrav 		if (!fetch_ssl_verify_hname(conn->ssl_cert, URL->host)) {
950dcd47379SDag-Erling Smørgrav 			fprintf(stderr,
951dcd47379SDag-Erling Smørgrav 			    "SSL certificate subject doesn't match host %s\n",
952dcd47379SDag-Erling Smørgrav 			    URL->host);
953dcd47379SDag-Erling Smørgrav 			return (-1);
954dcd47379SDag-Erling Smørgrav 		}
955dcd47379SDag-Erling Smørgrav 	}
956111e2510SDag-Erling Smørgrav 
957111e2510SDag-Erling Smørgrav 	if (verbose) {
95876720765SDag-Erling Smørgrav 		fetch_info("%s connection established using %s",
95976720765SDag-Erling Smørgrav 		    SSL_get_version(conn->ssl), SSL_get_cipher(conn->ssl));
960111e2510SDag-Erling Smørgrav 		name = X509_get_subject_name(conn->ssl_cert);
961111e2510SDag-Erling Smørgrav 		str = X509_NAME_oneline(name, 0, 0);
962dcd47379SDag-Erling Smørgrav 		fetch_info("Certificate subject: %s", str);
963dcd47379SDag-Erling Smørgrav 		OPENSSL_free(str);
964111e2510SDag-Erling Smørgrav 		name = X509_get_issuer_name(conn->ssl_cert);
965111e2510SDag-Erling Smørgrav 		str = X509_NAME_oneline(name, 0, 0);
966dcd47379SDag-Erling Smørgrav 		fetch_info("Certificate issuer: %s", str);
967dcd47379SDag-Erling Smørgrav 		OPENSSL_free(str);
968111e2510SDag-Erling Smørgrav 	}
969111e2510SDag-Erling Smørgrav 
970111e2510SDag-Erling Smørgrav 	return (0);
9713070f6cbSDag-Erling Smørgrav #else
9723070f6cbSDag-Erling Smørgrav 	(void)conn;
9733070f6cbSDag-Erling Smørgrav 	(void)verbose;
9743070f6cbSDag-Erling Smørgrav 	fprintf(stderr, "SSL support disabled\n");
9753070f6cbSDag-Erling Smørgrav 	return (-1);
9763070f6cbSDag-Erling Smørgrav #endif
977111e2510SDag-Erling Smørgrav }
978111e2510SDag-Erling Smørgrav 
979caaffed8SDag-Erling Smørgrav #define FETCH_READ_WAIT		-2
980caaffed8SDag-Erling Smørgrav #define FETCH_READ_ERROR	-1
981caaffed8SDag-Erling Smørgrav #define FETCH_READ_DONE		 0
982caaffed8SDag-Erling Smørgrav 
983caaffed8SDag-Erling Smørgrav #ifdef WITH_SSL
984caaffed8SDag-Erling Smørgrav static ssize_t
985caaffed8SDag-Erling Smørgrav fetch_ssl_read(SSL *ssl, char *buf, size_t len)
986caaffed8SDag-Erling Smørgrav {
987caaffed8SDag-Erling Smørgrav 	ssize_t rlen;
988caaffed8SDag-Erling Smørgrav 	int ssl_err;
989caaffed8SDag-Erling Smørgrav 
990caaffed8SDag-Erling Smørgrav 	rlen = SSL_read(ssl, buf, len);
991caaffed8SDag-Erling Smørgrav 	if (rlen < 0) {
992caaffed8SDag-Erling Smørgrav 		ssl_err = SSL_get_error(ssl, rlen);
993caaffed8SDag-Erling Smørgrav 		if (ssl_err == SSL_ERROR_WANT_READ ||
994caaffed8SDag-Erling Smørgrav 		    ssl_err == SSL_ERROR_WANT_WRITE) {
995caaffed8SDag-Erling Smørgrav 			return (FETCH_READ_WAIT);
996caaffed8SDag-Erling Smørgrav 		} else {
997caaffed8SDag-Erling Smørgrav 			ERR_print_errors_fp(stderr);
998caaffed8SDag-Erling Smørgrav 			return (FETCH_READ_ERROR);
999caaffed8SDag-Erling Smørgrav 		}
1000caaffed8SDag-Erling Smørgrav 	}
1001caaffed8SDag-Erling Smørgrav 	return (rlen);
1002caaffed8SDag-Erling Smørgrav }
1003caaffed8SDag-Erling Smørgrav #endif
1004caaffed8SDag-Erling Smørgrav 
1005caaffed8SDag-Erling Smørgrav static ssize_t
1006caaffed8SDag-Erling Smørgrav fetch_socket_read(int sd, char *buf, size_t len)
1007caaffed8SDag-Erling Smørgrav {
1008caaffed8SDag-Erling Smørgrav 	ssize_t rlen;
1009caaffed8SDag-Erling Smørgrav 
1010caaffed8SDag-Erling Smørgrav 	rlen = read(sd, buf, len);
1011caaffed8SDag-Erling Smørgrav 	if (rlen < 0) {
1012caaffed8SDag-Erling Smørgrav 		if (errno == EAGAIN || (errno == EINTR && fetchRestartCalls))
1013caaffed8SDag-Erling Smørgrav 			return (FETCH_READ_WAIT);
1014caaffed8SDag-Erling Smørgrav 		else
1015caaffed8SDag-Erling Smørgrav 			return (FETCH_READ_ERROR);
1016caaffed8SDag-Erling Smørgrav 	}
1017caaffed8SDag-Erling Smørgrav 	return (rlen);
1018caaffed8SDag-Erling Smørgrav }
1019f606d589SDag-Erling Smørgrav 
1020111e2510SDag-Erling Smørgrav /*
10219601e333SDag-Erling Smørgrav  * Read a character from a connection w/ timeout
10229601e333SDag-Erling Smørgrav  */
10239601e333SDag-Erling Smørgrav ssize_t
1024a1b37df2SDag-Erling Smørgrav fetch_read(conn_t *conn, char *buf, size_t len)
10259601e333SDag-Erling Smørgrav {
10265092cf05SDag-Erling Smørgrav 	struct timeval now, timeout, delta;
1027215a27f1SDag-Erling Smørgrav 	struct pollfd pfd;
1028215a27f1SDag-Erling Smørgrav 	ssize_t rlen;
1029215a27f1SDag-Erling Smørgrav 	int deltams;
10309601e333SDag-Erling Smørgrav 
103130787285SDag-Erling Smørgrav 	if (fetchTimeout > 0) {
10329601e333SDag-Erling Smørgrav 		gettimeofday(&timeout, NULL);
10339601e333SDag-Erling Smørgrav 		timeout.tv_sec += fetchTimeout;
10349601e333SDag-Erling Smørgrav 	}
10359601e333SDag-Erling Smørgrav 
1036215a27f1SDag-Erling Smørgrav 	deltams = INFTIM;
1037215a27f1SDag-Erling Smørgrav 	memset(&pfd, 0, sizeof pfd);
1038215a27f1SDag-Erling Smørgrav 	pfd.fd = conn->sd;
1039215a27f1SDag-Erling Smørgrav 	pfd.events = POLLIN | POLLERR;
10402a7daafeSDag-Erling Smørgrav 
1041215a27f1SDag-Erling Smørgrav 	for (;;) {
1042caaffed8SDag-Erling Smørgrav 		/*
1043caaffed8SDag-Erling Smørgrav 		 * The socket is non-blocking.  Instead of the canonical
1044215a27f1SDag-Erling Smørgrav 		 * poll() -> read(), we do the following:
1045caaffed8SDag-Erling Smørgrav 		 *
1046caaffed8SDag-Erling Smørgrav 		 * 1) call read() or SSL_read().
1047215a27f1SDag-Erling Smørgrav 		 * 2) if we received some data, return it.
1048215a27f1SDag-Erling Smørgrav 		 * 3) if an error occurred, return -1.
1049caaffed8SDag-Erling Smørgrav 		 * 4) if read() or SSL_read() signaled EOF, return.
1050caaffed8SDag-Erling Smørgrav 		 * 5) if we did not receive any data but we're not at EOF,
1051215a27f1SDag-Erling Smørgrav 		 *    call poll().
1052caaffed8SDag-Erling Smørgrav 		 *
1053caaffed8SDag-Erling Smørgrav 		 * In the SSL case, this is necessary because if we
1054caaffed8SDag-Erling Smørgrav 		 * receive a close notification, we have to call
1055caaffed8SDag-Erling Smørgrav 		 * SSL_read() one additional time after we've read
1056caaffed8SDag-Erling Smørgrav 		 * everything we received.
1057caaffed8SDag-Erling Smørgrav 		 *
1058caaffed8SDag-Erling Smørgrav 		 * In the non-SSL case, it may improve performance (very
1059caaffed8SDag-Erling Smørgrav 		 * slightly) when reading small amounts of data.
1060caaffed8SDag-Erling Smørgrav 		 */
1061caaffed8SDag-Erling Smørgrav #ifdef WITH_SSL
1062caaffed8SDag-Erling Smørgrav 		if (conn->ssl != NULL)
1063caaffed8SDag-Erling Smørgrav 			rlen = fetch_ssl_read(conn->ssl, buf, len);
1064caaffed8SDag-Erling Smørgrav 		else
1065caaffed8SDag-Erling Smørgrav #endif
1066caaffed8SDag-Erling Smørgrav 			rlen = fetch_socket_read(conn->sd, buf, len);
10679c1ca3a1SDag-Erling Smørgrav 		if (rlen >= 0) {
1068caaffed8SDag-Erling Smørgrav 			break;
1069caaffed8SDag-Erling Smørgrav 		} else if (rlen == FETCH_READ_ERROR) {
10709c1ca3a1SDag-Erling Smørgrav 			fetch_syserr();
1071caaffed8SDag-Erling Smørgrav 			return (-1);
1072caaffed8SDag-Erling Smørgrav 		}
10739c1ca3a1SDag-Erling Smørgrav 		// assert(rlen == FETCH_READ_WAIT);
107430787285SDag-Erling Smørgrav 		if (fetchTimeout > 0) {
10759601e333SDag-Erling Smørgrav 			gettimeofday(&now, NULL);
107630787285SDag-Erling Smørgrav 			if (!timercmp(&timeout, &now, >)) {
1077bb13d0afSDag-Erling Smørgrav 				errno = ETIMEDOUT;
1078a1b37df2SDag-Erling Smørgrav 				fetch_syserr();
1079bb13d0afSDag-Erling Smørgrav 				return (-1);
1080bb13d0afSDag-Erling Smørgrav 			}
108130787285SDag-Erling Smørgrav 			timersub(&timeout, &now, &delta);
1082215a27f1SDag-Erling Smørgrav 			deltams = delta.tv_sec * 1000 +
1083215a27f1SDag-Erling Smørgrav 			    delta.tv_usec / 1000;;
108430787285SDag-Erling Smørgrav 		}
10859601e333SDag-Erling Smørgrav 		errno = 0;
1086215a27f1SDag-Erling Smørgrav 		pfd.revents = 0;
1087215a27f1SDag-Erling Smørgrav 		if (poll(&pfd, 1, deltams) < 0) {
1088215a27f1SDag-Erling Smørgrav 			if (errno == EINTR && fetchRestartCalls)
10899601e333SDag-Erling Smørgrav 				continue;
1090a1b37df2SDag-Erling Smørgrav 			fetch_syserr();
10919601e333SDag-Erling Smørgrav 			return (-1);
10929601e333SDag-Erling Smørgrav 		}
10939601e333SDag-Erling Smørgrav 	}
1094215a27f1SDag-Erling Smørgrav 	return (rlen);
10959601e333SDag-Erling Smørgrav }
10969601e333SDag-Erling Smørgrav 
1097f606d589SDag-Erling Smørgrav 
10989601e333SDag-Erling Smørgrav /*
10999601e333SDag-Erling Smørgrav  * Read a line of text from a connection w/ timeout
1100fc6e9e65SDag-Erling Smørgrav  */
1101fc6e9e65SDag-Erling Smørgrav #define MIN_BUF_SIZE 1024
1102fc6e9e65SDag-Erling Smørgrav 
1103fc6e9e65SDag-Erling Smørgrav int
1104a1b37df2SDag-Erling Smørgrav fetch_getln(conn_t *conn)
1105fc6e9e65SDag-Erling Smørgrav {
11069601e333SDag-Erling Smørgrav 	char *tmp;
11079601e333SDag-Erling Smørgrav 	size_t tmpsize;
1108bb13d0afSDag-Erling Smørgrav 	ssize_t len;
1109fc6e9e65SDag-Erling Smørgrav 	char c;
1110fc6e9e65SDag-Erling Smørgrav 
1111dea29ca1SDag-Erling Smørgrav 	if (conn->buf == NULL) {
1112dea29ca1SDag-Erling Smørgrav 		if ((conn->buf = malloc(MIN_BUF_SIZE)) == NULL) {
1113fc6e9e65SDag-Erling Smørgrav 			errno = ENOMEM;
1114e19e6098SDag-Erling Smørgrav 			return (-1);
1115fc6e9e65SDag-Erling Smørgrav 		}
1116dea29ca1SDag-Erling Smørgrav 		conn->bufsize = MIN_BUF_SIZE;
1117fc6e9e65SDag-Erling Smørgrav 	}
1118fc6e9e65SDag-Erling Smørgrav 
1119dea29ca1SDag-Erling Smørgrav 	conn->buf[0] = '\0';
1120dea29ca1SDag-Erling Smørgrav 	conn->buflen = 0;
1121fc6e9e65SDag-Erling Smørgrav 
1122fc6e9e65SDag-Erling Smørgrav 	do {
1123a1b37df2SDag-Erling Smørgrav 		len = fetch_read(conn, &c, 1);
1124bb13d0afSDag-Erling Smørgrav 		if (len == -1)
1125e19e6098SDag-Erling Smørgrav 			return (-1);
1126bb13d0afSDag-Erling Smørgrav 		if (len == 0)
1127b68fbebdSDavid E. O'Brien 			break;
1128dea29ca1SDag-Erling Smørgrav 		conn->buf[conn->buflen++] = c;
1129dea29ca1SDag-Erling Smørgrav 		if (conn->buflen == conn->bufsize) {
1130dea29ca1SDag-Erling Smørgrav 			tmp = conn->buf;
1131dea29ca1SDag-Erling Smørgrav 			tmpsize = conn->bufsize * 2 + 1;
1132dea29ca1SDag-Erling Smørgrav 			if ((tmp = realloc(tmp, tmpsize)) == NULL) {
1133fc6e9e65SDag-Erling Smørgrav 				errno = ENOMEM;
1134e19e6098SDag-Erling Smørgrav 				return (-1);
1135fc6e9e65SDag-Erling Smørgrav 			}
1136dea29ca1SDag-Erling Smørgrav 			conn->buf = tmp;
1137dea29ca1SDag-Erling Smørgrav 			conn->bufsize = tmpsize;
1138fc6e9e65SDag-Erling Smørgrav 		}
1139fc6e9e65SDag-Erling Smørgrav 	} while (c != '\n');
1140fc6e9e65SDag-Erling Smørgrav 
1141dea29ca1SDag-Erling Smørgrav 	conn->buf[conn->buflen] = '\0';
1142dea29ca1SDag-Erling Smørgrav 	DEBUG(fprintf(stderr, "<<< %s", conn->buf));
1143e19e6098SDag-Erling Smørgrav 	return (0);
1144fc6e9e65SDag-Erling Smørgrav }
1145fc6e9e65SDag-Erling Smørgrav 
1146fc6e9e65SDag-Erling Smørgrav 
1147a1763027SDag-Erling Smørgrav /*
11489601e333SDag-Erling Smørgrav  * Write to a connection w/ timeout
11499601e333SDag-Erling Smørgrav  */
11509601e333SDag-Erling Smørgrav ssize_t
1151a1b37df2SDag-Erling Smørgrav fetch_write(conn_t *conn, const char *buf, size_t len)
11529601e333SDag-Erling Smørgrav {
1153a4a37038SWarner Losh 	struct iovec iov;
1154a4a37038SWarner Losh 
1155a4a37038SWarner Losh 	iov.iov_base = __DECONST(char *, buf);
1156a4a37038SWarner Losh 	iov.iov_len = len;
1157a1b37df2SDag-Erling Smørgrav 	return fetch_writev(conn, &iov, 1);
1158a4a37038SWarner Losh }
1159a4a37038SWarner Losh 
1160a4a37038SWarner Losh /*
1161a4a37038SWarner Losh  * Write a vector to a connection w/ timeout
1162a4a37038SWarner Losh  * Note: can modify the iovec.
1163a4a37038SWarner Losh  */
1164a4a37038SWarner Losh ssize_t
1165a1b37df2SDag-Erling Smørgrav fetch_writev(conn_t *conn, struct iovec *iov, int iovcnt)
1166a4a37038SWarner Losh {
11675092cf05SDag-Erling Smørgrav 	struct timeval now, timeout, delta;
1168215a27f1SDag-Erling Smørgrav 	struct pollfd pfd;
11699601e333SDag-Erling Smørgrav 	ssize_t wlen, total;
11709c1ca3a1SDag-Erling Smørgrav 	int deltams;
11719601e333SDag-Erling Smørgrav 
1172215a27f1SDag-Erling Smørgrav 	memset(&pfd, 0, sizeof pfd);
11739601e333SDag-Erling Smørgrav 	if (fetchTimeout) {
1174215a27f1SDag-Erling Smørgrav 		pfd.fd = conn->sd;
1175215a27f1SDag-Erling Smørgrav 		pfd.events = POLLOUT | POLLERR;
11769601e333SDag-Erling Smørgrav 		gettimeofday(&timeout, NULL);
11779601e333SDag-Erling Smørgrav 		timeout.tv_sec += fetchTimeout;
11789601e333SDag-Erling Smørgrav 	}
11799601e333SDag-Erling Smørgrav 
1180a4a37038SWarner Losh 	total = 0;
1181a4a37038SWarner Losh 	while (iovcnt > 0) {
1182215a27f1SDag-Erling Smørgrav 		while (fetchTimeout && pfd.revents == 0) {
11839601e333SDag-Erling Smørgrav 			gettimeofday(&now, NULL);
11849c1ca3a1SDag-Erling Smørgrav 			if (!timercmp(&timeout, &now, >)) {
11859601e333SDag-Erling Smørgrav 				errno = ETIMEDOUT;
1186a1b37df2SDag-Erling Smørgrav 				fetch_syserr();
11879601e333SDag-Erling Smørgrav 				return (-1);
11889601e333SDag-Erling Smørgrav 			}
11899c1ca3a1SDag-Erling Smørgrav 			timersub(&timeout, &now, &delta);
11909c1ca3a1SDag-Erling Smørgrav 			deltams = delta.tv_sec * 1000 +
11919c1ca3a1SDag-Erling Smørgrav 			    delta.tv_usec / 1000;
11929601e333SDag-Erling Smørgrav 			errno = 0;
11939c1ca3a1SDag-Erling Smørgrav 			pfd.revents = 0;
11949c1ca3a1SDag-Erling Smørgrav 			if (poll(&pfd, 1, deltams) < 0) {
11953674911cSBaptiste Daroussin 				/* POSIX compliance */
11964472d6e1SBaptiste Daroussin 				if (errno == EAGAIN)
11974472d6e1SBaptiste Daroussin 					continue;
11989601e333SDag-Erling Smørgrav 				if (errno == EINTR && fetchRestartCalls)
11999601e333SDag-Erling Smørgrav 					continue;
12009601e333SDag-Erling Smørgrav 				return (-1);
12019601e333SDag-Erling Smørgrav 			}
12029601e333SDag-Erling Smørgrav 		}
12039601e333SDag-Erling Smørgrav 		errno = 0;
12043070f6cbSDag-Erling Smørgrav #ifdef WITH_SSL
12059601e333SDag-Erling Smørgrav 		if (conn->ssl != NULL)
1206a4a37038SWarner Losh 			wlen = SSL_write(conn->ssl,
1207a4a37038SWarner Losh 			    iov->iov_base, iov->iov_len);
12089601e333SDag-Erling Smørgrav 		else
12093070f6cbSDag-Erling Smørgrav #endif
1210a4a37038SWarner Losh 			wlen = writev(conn->sd, iov, iovcnt);
1211a4a37038SWarner Losh 		if (wlen == 0) {
12129601e333SDag-Erling Smørgrav 			/* we consider a short write a failure */
1213caaffed8SDag-Erling Smørgrav 			/* XXX perhaps we shouldn't in the SSL case */
1214a4a37038SWarner Losh 			errno = EPIPE;
1215a1b37df2SDag-Erling Smørgrav 			fetch_syserr();
12169601e333SDag-Erling Smørgrav 			return (-1);
1217a4a37038SWarner Losh 		}
12189601e333SDag-Erling Smørgrav 		if (wlen < 0) {
12199601e333SDag-Erling Smørgrav 			if (errno == EINTR && fetchRestartCalls)
12209601e333SDag-Erling Smørgrav 				continue;
12219601e333SDag-Erling Smørgrav 			return (-1);
12229601e333SDag-Erling Smørgrav 		}
12239601e333SDag-Erling Smørgrav 		total += wlen;
1224a4a37038SWarner Losh 		while (iovcnt > 0 && wlen >= (ssize_t)iov->iov_len) {
1225a4a37038SWarner Losh 			wlen -= iov->iov_len;
1226a4a37038SWarner Losh 			iov++;
1227a4a37038SWarner Losh 			iovcnt--;
1228a4a37038SWarner Losh 		}
1229a4a37038SWarner Losh 		if (iovcnt > 0) {
1230a4a37038SWarner Losh 			iov->iov_len -= wlen;
1231a4a37038SWarner Losh 			iov->iov_base = __DECONST(char *, iov->iov_base) + wlen;
1232a4a37038SWarner Losh 		}
12339601e333SDag-Erling Smørgrav 	}
12349601e333SDag-Erling Smørgrav 	return (total);
12359601e333SDag-Erling Smørgrav }
12369601e333SDag-Erling Smørgrav 
1237f606d589SDag-Erling Smørgrav 
12389601e333SDag-Erling Smørgrav /*
12399601e333SDag-Erling Smørgrav  * Write a line of text to a connection w/ timeout
1240a1763027SDag-Erling Smørgrav  */
1241a1763027SDag-Erling Smørgrav int
1242a1b37df2SDag-Erling Smørgrav fetch_putln(conn_t *conn, const char *str, size_t len)
1243a1763027SDag-Erling Smørgrav {
1244a4a37038SWarner Losh 	struct iovec iov[2];
12457504527eSDag-Erling Smørgrav 	int ret;
124666ffb8a3SDag-Erling Smørgrav 
124766ffb8a3SDag-Erling Smørgrav 	DEBUG(fprintf(stderr, ">>> %s\n", str));
1248a4a37038SWarner Losh 	iov[0].iov_base = __DECONST(char *, str);
1249a4a37038SWarner Losh 	iov[0].iov_len = len;
1250a4a37038SWarner Losh 	iov[1].iov_base = __DECONST(char *, ENDL);
1251930105c1SDag-Erling Smørgrav 	iov[1].iov_len = sizeof(ENDL);
12527504527eSDag-Erling Smørgrav 	if (len == 0)
1253a1b37df2SDag-Erling Smørgrav 		ret = fetch_writev(conn, &iov[1], 1);
12547504527eSDag-Erling Smørgrav 	else
1255a1b37df2SDag-Erling Smørgrav 		ret = fetch_writev(conn, iov, 2);
12567504527eSDag-Erling Smørgrav 	if (ret == -1)
1257e19e6098SDag-Erling Smørgrav 		return (-1);
1258e19e6098SDag-Erling Smørgrav 	return (0);
1259a1763027SDag-Erling Smørgrav }
1260a1763027SDag-Erling Smørgrav 
1261a1763027SDag-Erling Smørgrav 
1262dea29ca1SDag-Erling Smørgrav /*
1263dea29ca1SDag-Erling Smørgrav  * Close connection
1264dea29ca1SDag-Erling Smørgrav  */
1265dea29ca1SDag-Erling Smørgrav int
1266a1b37df2SDag-Erling Smørgrav fetch_close(conn_t *conn)
1267dea29ca1SDag-Erling Smørgrav {
1268dea29ca1SDag-Erling Smørgrav 	int ret;
1269dea29ca1SDag-Erling Smørgrav 
1270f606d589SDag-Erling Smørgrav 	if (--conn->ref > 0)
1271f606d589SDag-Erling Smørgrav 		return (0);
1272dcd47379SDag-Erling Smørgrav #ifdef WITH_SSL
1273dcd47379SDag-Erling Smørgrav 	if (conn->ssl) {
1274dcd47379SDag-Erling Smørgrav 		SSL_shutdown(conn->ssl);
1275dcd47379SDag-Erling Smørgrav 		SSL_set_connect_state(conn->ssl);
1276dcd47379SDag-Erling Smørgrav 		SSL_free(conn->ssl);
1277dcd47379SDag-Erling Smørgrav 		conn->ssl = NULL;
1278dcd47379SDag-Erling Smørgrav 	}
1279dcd47379SDag-Erling Smørgrav 	if (conn->ssl_ctx) {
1280dcd47379SDag-Erling Smørgrav 		SSL_CTX_free(conn->ssl_ctx);
1281dcd47379SDag-Erling Smørgrav 		conn->ssl_ctx = NULL;
1282dcd47379SDag-Erling Smørgrav 	}
1283dcd47379SDag-Erling Smørgrav 	if (conn->ssl_cert) {
1284dcd47379SDag-Erling Smørgrav 		X509_free(conn->ssl_cert);
1285dcd47379SDag-Erling Smørgrav 		conn->ssl_cert = NULL;
1286dcd47379SDag-Erling Smørgrav 	}
1287dcd47379SDag-Erling Smørgrav #endif
1288dea29ca1SDag-Erling Smørgrav 	ret = close(conn->sd);
128955cf7be1SDag-Erling Smørgrav 	free(conn->buf);
1290dea29ca1SDag-Erling Smørgrav 	free(conn);
1291dea29ca1SDag-Erling Smørgrav 	return (ret);
1292dea29ca1SDag-Erling Smørgrav }
1293dea29ca1SDag-Erling Smørgrav 
1294dea29ca1SDag-Erling Smørgrav 
1295ce71b736SDag-Erling Smørgrav /*** Directory-related utility functions *************************************/
1296ce71b736SDag-Erling Smørgrav 
1297ce71b736SDag-Erling Smørgrav int
1298a1b37df2SDag-Erling Smørgrav fetch_add_entry(struct url_ent **p, int *size, int *len,
1299f573a5fcSDag-Erling Smørgrav     const char *name, struct url_stat *us)
1300ce71b736SDag-Erling Smørgrav {
1301ce71b736SDag-Erling Smørgrav 	struct url_ent *tmp;
1302ce71b736SDag-Erling Smørgrav 
1303ce71b736SDag-Erling Smørgrav 	if (*p == NULL) {
13045a51c23bSDag-Erling Smørgrav 		*size = 0;
1305ce71b736SDag-Erling Smørgrav 		*len = 0;
1306ce71b736SDag-Erling Smørgrav 	}
1307ce71b736SDag-Erling Smørgrav 
1308ce71b736SDag-Erling Smørgrav 	if (*len >= *size - 1) {
1309930105c1SDag-Erling Smørgrav 		tmp = realloc(*p, (*size * 2 + 1) * sizeof(**p));
1310ce71b736SDag-Erling Smørgrav 		if (tmp == NULL) {
1311ce71b736SDag-Erling Smørgrav 			errno = ENOMEM;
1312a1b37df2SDag-Erling Smørgrav 			fetch_syserr();
1313e19e6098SDag-Erling Smørgrav 			return (-1);
1314ce71b736SDag-Erling Smørgrav 		}
13155a51c23bSDag-Erling Smørgrav 		*size = (*size * 2 + 1);
1316ce71b736SDag-Erling Smørgrav 		*p = tmp;
1317ce71b736SDag-Erling Smørgrav 	}
1318ce71b736SDag-Erling Smørgrav 
1319ce71b736SDag-Erling Smørgrav 	tmp = *p + *len;
13202b26f942SDag-Erling Smørgrav 	snprintf(tmp->name, PATH_MAX, "%s", name);
1321340b079bSDag-Erling Smørgrav 	memcpy(&tmp->stat, us, sizeof(*us));
1322ce71b736SDag-Erling Smørgrav 
1323ce71b736SDag-Erling Smørgrav 	(*len)++;
1324ce71b736SDag-Erling Smørgrav 	(++tmp)->name[0] = 0;
1325ce71b736SDag-Erling Smørgrav 
1326e19e6098SDag-Erling Smørgrav 	return (0);
1327ce71b736SDag-Erling Smørgrav }
132807350d12SDag-Erling Smørgrav 
132907350d12SDag-Erling Smørgrav 
133007350d12SDag-Erling Smørgrav /*** Authentication-related utility functions ********************************/
133107350d12SDag-Erling Smørgrav 
133207350d12SDag-Erling Smørgrav static const char *
1333a1b37df2SDag-Erling Smørgrav fetch_read_word(FILE *f)
133407350d12SDag-Erling Smørgrav {
133507350d12SDag-Erling Smørgrav 	static char word[1024];
133607350d12SDag-Erling Smørgrav 
1337fc2841a9SColin Percival 	if (fscanf(f, " %1023s ", word) != 1)
133807350d12SDag-Erling Smørgrav 		return (NULL);
133907350d12SDag-Erling Smørgrav 	return (word);
134007350d12SDag-Erling Smørgrav }
134107350d12SDag-Erling Smørgrav 
1342d8713bf3SBaptiste Daroussin static int
1343d8713bf3SBaptiste Daroussin fetch_netrc_open(void)
134407350d12SDag-Erling Smørgrav {
1345d8713bf3SBaptiste Daroussin 	const char *p;
134607350d12SDag-Erling Smørgrav 	char fn[PATH_MAX];
134707350d12SDag-Erling Smørgrav 
134807350d12SDag-Erling Smørgrav 	if ((p = getenv("NETRC")) != NULL) {
1349930105c1SDag-Erling Smørgrav 		if (snprintf(fn, sizeof(fn), "%s", p) >= (int)sizeof(fn)) {
1350a1b37df2SDag-Erling Smørgrav 			fetch_info("$NETRC specifies a file name "
135107350d12SDag-Erling Smørgrav 			    "longer than PATH_MAX");
135207350d12SDag-Erling Smørgrav 			return (-1);
135307350d12SDag-Erling Smørgrav 		}
135407350d12SDag-Erling Smørgrav 	} else {
135507350d12SDag-Erling Smørgrav 		if ((p = getenv("HOME")) != NULL) {
135607350d12SDag-Erling Smørgrav 			struct passwd *pwd;
135707350d12SDag-Erling Smørgrav 
135807350d12SDag-Erling Smørgrav 			if ((pwd = getpwuid(getuid())) == NULL ||
135907350d12SDag-Erling Smørgrav 			    (p = pwd->pw_dir) == NULL)
136007350d12SDag-Erling Smørgrav 				return (-1);
136107350d12SDag-Erling Smørgrav 		}
1362930105c1SDag-Erling Smørgrav 		if (snprintf(fn, sizeof(fn), "%s/.netrc", p) >= (int)sizeof(fn))
136307350d12SDag-Erling Smørgrav 			return (-1);
136407350d12SDag-Erling Smørgrav 	}
136507350d12SDag-Erling Smørgrav 
1366d8713bf3SBaptiste Daroussin 	return (open(fn, O_RDONLY));
1367d8713bf3SBaptiste Daroussin }
1368d8713bf3SBaptiste Daroussin 
1369d8713bf3SBaptiste Daroussin /*
1370d8713bf3SBaptiste Daroussin  * Get authentication data for a URL from .netrc
1371d8713bf3SBaptiste Daroussin  */
1372d8713bf3SBaptiste Daroussin int
1373d8713bf3SBaptiste Daroussin fetch_netrc_auth(struct url *url)
1374d8713bf3SBaptiste Daroussin {
1375d8713bf3SBaptiste Daroussin 	const char *word;
1376d8713bf3SBaptiste Daroussin 	FILE *f;
1377d8713bf3SBaptiste Daroussin 
1378d8713bf3SBaptiste Daroussin 	if (url->netrcfd == -2)
1379d8713bf3SBaptiste Daroussin 		url->netrcfd = fetch_netrc_open();
1380d8713bf3SBaptiste Daroussin 	if (url->netrcfd < 0)
138107350d12SDag-Erling Smørgrav 		return (-1);
1382d8713bf3SBaptiste Daroussin 	if ((f = fdopen(url->netrcfd, "r")) == NULL)
1383d8713bf3SBaptiste Daroussin 		return (-1);
1384d8713bf3SBaptiste Daroussin 	rewind(f);
1385a1b37df2SDag-Erling Smørgrav 	while ((word = fetch_read_word(f)) != NULL) {
138607350d12SDag-Erling Smørgrav 		if (strcmp(word, "default") == 0) {
1387a1b37df2SDag-Erling Smørgrav 			DEBUG(fetch_info("Using default .netrc settings"));
138807350d12SDag-Erling Smørgrav 			break;
138907350d12SDag-Erling Smørgrav 		}
139007350d12SDag-Erling Smørgrav 		if (strcmp(word, "machine") == 0 &&
1391a1b37df2SDag-Erling Smørgrav 		    (word = fetch_read_word(f)) != NULL &&
139207350d12SDag-Erling Smørgrav 		    strcasecmp(word, url->host) == 0) {
1393a1b37df2SDag-Erling Smørgrav 			DEBUG(fetch_info("Using .netrc settings for %s", word));
139407350d12SDag-Erling Smørgrav 			break;
139507350d12SDag-Erling Smørgrav 		}
139607350d12SDag-Erling Smørgrav 	}
139707350d12SDag-Erling Smørgrav 	if (word == NULL)
139807350d12SDag-Erling Smørgrav 		goto ferr;
1399a1b37df2SDag-Erling Smørgrav 	while ((word = fetch_read_word(f)) != NULL) {
140007350d12SDag-Erling Smørgrav 		if (strcmp(word, "login") == 0) {
1401a1b37df2SDag-Erling Smørgrav 			if ((word = fetch_read_word(f)) == NULL)
140207350d12SDag-Erling Smørgrav 				goto ferr;
1403930105c1SDag-Erling Smørgrav 			if (snprintf(url->user, sizeof(url->user),
14049015b953SJohn W. De Boskey 				"%s", word) > (int)sizeof(url->user)) {
1405a1b37df2SDag-Erling Smørgrav 				fetch_info("login name in .netrc is too long");
140607350d12SDag-Erling Smørgrav 				url->user[0] = '\0';
140707350d12SDag-Erling Smørgrav 			}
140807350d12SDag-Erling Smørgrav 		} else if (strcmp(word, "password") == 0) {
1409a1b37df2SDag-Erling Smørgrav 			if ((word = fetch_read_word(f)) == NULL)
141007350d12SDag-Erling Smørgrav 				goto ferr;
1411930105c1SDag-Erling Smørgrav 			if (snprintf(url->pwd, sizeof(url->pwd),
14129015b953SJohn W. De Boskey 				"%s", word) > (int)sizeof(url->pwd)) {
1413a1b37df2SDag-Erling Smørgrav 				fetch_info("password in .netrc is too long");
141407350d12SDag-Erling Smørgrav 				url->pwd[0] = '\0';
141507350d12SDag-Erling Smørgrav 			}
141607350d12SDag-Erling Smørgrav 		} else if (strcmp(word, "account") == 0) {
1417a1b37df2SDag-Erling Smørgrav 			if ((word = fetch_read_word(f)) == NULL)
141807350d12SDag-Erling Smørgrav 				goto ferr;
141907350d12SDag-Erling Smørgrav 			/* XXX not supported! */
142007350d12SDag-Erling Smørgrav 		} else {
142107350d12SDag-Erling Smørgrav 			break;
142207350d12SDag-Erling Smørgrav 		}
142307350d12SDag-Erling Smørgrav 	}
142407350d12SDag-Erling Smørgrav 	fclose(f);
142507350d12SDag-Erling Smørgrav 	return (0);
142607350d12SDag-Erling Smørgrav  ferr:
142707350d12SDag-Erling Smørgrav 	fclose(f);
142807350d12SDag-Erling Smørgrav 	return (-1);
142907350d12SDag-Erling Smørgrav }
143062a2681cSDag-Erling Smørgrav 
143162a2681cSDag-Erling Smørgrav /*
143262a2681cSDag-Erling Smørgrav  * The no_proxy environment variable specifies a set of domains for
143362a2681cSDag-Erling Smørgrav  * which the proxy should not be consulted; the contents is a comma-,
143462a2681cSDag-Erling Smørgrav  * or space-separated list of domain names.  A single asterisk will
143562a2681cSDag-Erling Smørgrav  * override all proxy variables and no transactions will be proxied
143675f46cf6SPedro F. Giffuni  * (for compatibility with lynx and curl, see the discussion at
143762a2681cSDag-Erling Smørgrav  * <http://curl.haxx.se/mail/archive_pre_oct_99/0009.html>).
143862a2681cSDag-Erling Smørgrav  */
143962a2681cSDag-Erling Smørgrav int
144062a2681cSDag-Erling Smørgrav fetch_no_proxy_match(const char *host)
144162a2681cSDag-Erling Smørgrav {
144262a2681cSDag-Erling Smørgrav 	const char *no_proxy, *p, *q;
144362a2681cSDag-Erling Smørgrav 	size_t h_len, d_len;
144462a2681cSDag-Erling Smørgrav 
144562a2681cSDag-Erling Smørgrav 	if ((no_proxy = getenv("NO_PROXY")) == NULL &&
144662a2681cSDag-Erling Smørgrav 	    (no_proxy = getenv("no_proxy")) == NULL)
144762a2681cSDag-Erling Smørgrav 		return (0);
144862a2681cSDag-Erling Smørgrav 
144962a2681cSDag-Erling Smørgrav 	/* asterisk matches any hostname */
145062a2681cSDag-Erling Smørgrav 	if (strcmp(no_proxy, "*") == 0)
145162a2681cSDag-Erling Smørgrav 		return (1);
145262a2681cSDag-Erling Smørgrav 
145362a2681cSDag-Erling Smørgrav 	h_len = strlen(host);
145462a2681cSDag-Erling Smørgrav 	p = no_proxy;
145562a2681cSDag-Erling Smørgrav 	do {
145662a2681cSDag-Erling Smørgrav 		/* position p at the beginning of a domain suffix */
1457facd9827SDag-Erling Smørgrav 		while (*p == ',' || isspace((unsigned char)*p))
145862a2681cSDag-Erling Smørgrav 			p++;
145962a2681cSDag-Erling Smørgrav 
146062a2681cSDag-Erling Smørgrav 		/* position q at the first separator character */
146162a2681cSDag-Erling Smørgrav 		for (q = p; *q; ++q)
1462facd9827SDag-Erling Smørgrav 			if (*q == ',' || isspace((unsigned char)*q))
146362a2681cSDag-Erling Smørgrav 				break;
146462a2681cSDag-Erling Smørgrav 
146562a2681cSDag-Erling Smørgrav 		d_len = q - p;
1466c0d2581bSFabien Thomas 		if (d_len > 0 && h_len >= d_len &&
146762a2681cSDag-Erling Smørgrav 		    strncasecmp(host + h_len - d_len,
146862a2681cSDag-Erling Smørgrav 			p, d_len) == 0) {
146962a2681cSDag-Erling Smørgrav 			/* domain name matches */
147062a2681cSDag-Erling Smørgrav 			return (1);
147162a2681cSDag-Erling Smørgrav 		}
147262a2681cSDag-Erling Smørgrav 
147362a2681cSDag-Erling Smørgrav 		p = q + 1;
147462a2681cSDag-Erling Smørgrav 	} while (*q);
147562a2681cSDag-Erling Smørgrav 
147662a2681cSDag-Erling Smørgrav 	return (0);
147762a2681cSDag-Erling Smørgrav }
1478