xref: /freebsd/lib/libfetch/http.c (revision 79257dd70a02290126060aa7876802653cb5badc)
14ca1ab94SDag-Erling Smørgrav /*-
22cbbf9daSDag-Erling Smørgrav  * Copyright (c) 2000-2004 Dag-Erling Co�dan Sm�rgrav
34ca1ab94SDag-Erling Smørgrav  * All rights reserved.
44ca1ab94SDag-Erling Smørgrav  *
54ca1ab94SDag-Erling Smørgrav  * Redistribution and use in source and binary forms, with or without
64ca1ab94SDag-Erling Smørgrav  * modification, are permitted provided that the following conditions
74ca1ab94SDag-Erling Smørgrav  * are met:
84ca1ab94SDag-Erling Smørgrav  * 1. Redistributions of source code must retain the above copyright
94ca1ab94SDag-Erling Smørgrav  *    notice, this list of conditions and the following disclaimer
104ca1ab94SDag-Erling Smørgrav  *    in this position and unchanged.
114ca1ab94SDag-Erling Smørgrav  * 2. Redistributions in binary form must reproduce the above copyright
124ca1ab94SDag-Erling Smørgrav  *    notice, this list of conditions and the following disclaimer in the
134ca1ab94SDag-Erling Smørgrav  *    documentation and/or other materials provided with the distribution.
144ca1ab94SDag-Erling Smørgrav  * 3. The name of the author may not be used to endorse or promote products
15e4878e39SDag-Erling Smørgrav  *    derived from this software without specific prior written permission.
164ca1ab94SDag-Erling Smørgrav  *
174ca1ab94SDag-Erling Smørgrav  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
184ca1ab94SDag-Erling Smørgrav  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
194ca1ab94SDag-Erling Smørgrav  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
204ca1ab94SDag-Erling Smørgrav  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
214ca1ab94SDag-Erling Smørgrav  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
224ca1ab94SDag-Erling Smørgrav  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
234ca1ab94SDag-Erling Smørgrav  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
244ca1ab94SDag-Erling Smørgrav  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
254ca1ab94SDag-Erling Smørgrav  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
264ca1ab94SDag-Erling Smørgrav  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
274ca1ab94SDag-Erling Smørgrav  */
284ca1ab94SDag-Erling Smørgrav 
29cecb889fSMatthew Dillon #include <sys/cdefs.h>
30cecb889fSMatthew Dillon __FBSDID("$FreeBSD$");
31cecb889fSMatthew Dillon 
326290ee73SDag-Erling Smørgrav /*
336290ee73SDag-Erling Smørgrav  * The following copyright applies to the base64 code:
346290ee73SDag-Erling Smørgrav  *
356290ee73SDag-Erling Smørgrav  *-
366290ee73SDag-Erling Smørgrav  * Copyright 1997 Massachusetts Institute of Technology
376290ee73SDag-Erling Smørgrav  *
386290ee73SDag-Erling Smørgrav  * Permission to use, copy, modify, and distribute this software and
396290ee73SDag-Erling Smørgrav  * its documentation for any purpose and without fee is hereby
406290ee73SDag-Erling Smørgrav  * granted, provided that both the above copyright notice and this
416290ee73SDag-Erling Smørgrav  * permission notice appear in all copies, that both the above
426290ee73SDag-Erling Smørgrav  * copyright notice and this permission notice appear in all
436290ee73SDag-Erling Smørgrav  * supporting documentation, and that the name of M.I.T. not be used
446290ee73SDag-Erling Smørgrav  * in advertising or publicity pertaining to distribution of the
456290ee73SDag-Erling Smørgrav  * software without specific, written prior permission.  M.I.T. makes
466290ee73SDag-Erling Smørgrav  * no representations about the suitability of this software for any
476290ee73SDag-Erling Smørgrav  * purpose.  It is provided "as is" without express or implied
486290ee73SDag-Erling Smørgrav  * warranty.
496290ee73SDag-Erling Smørgrav  *
506290ee73SDag-Erling Smørgrav  * THIS SOFTWARE IS PROVIDED BY M.I.T. ``AS IS''.  M.I.T. DISCLAIMS
516290ee73SDag-Erling Smørgrav  * ALL EXPRESS OR IMPLIED WARRANTIES WITH REGARD TO THIS SOFTWARE,
526290ee73SDag-Erling Smørgrav  * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
536290ee73SDag-Erling Smørgrav  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. IN NO EVENT
546290ee73SDag-Erling Smørgrav  * SHALL M.I.T. BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
556290ee73SDag-Erling Smørgrav  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
566290ee73SDag-Erling Smørgrav  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
576290ee73SDag-Erling Smørgrav  * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
586290ee73SDag-Erling Smørgrav  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
596290ee73SDag-Erling Smørgrav  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
606290ee73SDag-Erling Smørgrav  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
616290ee73SDag-Erling Smørgrav  * SUCH DAMAGE.
626290ee73SDag-Erling Smørgrav  */
636290ee73SDag-Erling Smørgrav 
644ca1ab94SDag-Erling Smørgrav #include <sys/param.h>
6528c645cfSHajimu UMEMOTO #include <sys/socket.h>
664ca1ab94SDag-Erling Smørgrav 
674ca1ab94SDag-Erling Smørgrav #include <ctype.h>
68e4878e39SDag-Erling Smørgrav #include <err.h>
69e4878e39SDag-Erling Smørgrav #include <errno.h>
7060245e42SDag-Erling Smørgrav #include <locale.h>
71e6182307SDag-Erling Smørgrav #include <netdb.h>
72f62e5228SDag-Erling Smørgrav #include <stdarg.h>
734ca1ab94SDag-Erling Smørgrav #include <stdio.h>
744ca1ab94SDag-Erling Smørgrav #include <stdlib.h>
754ca1ab94SDag-Erling Smørgrav #include <string.h>
7660245e42SDag-Erling Smørgrav #include <time.h>
774ca1ab94SDag-Erling Smørgrav #include <unistd.h>
784ca1ab94SDag-Erling Smørgrav 
7920a2a51bSKelly Yancey #include <netinet/in.h>
8020a2a51bSKelly Yancey #include <netinet/tcp.h>
8120a2a51bSKelly Yancey 
824ca1ab94SDag-Erling Smørgrav #include "fetch.h"
83842a95ccSDag-Erling Smørgrav #include "common.h"
840fba3a00SDag-Erling Smørgrav #include "httperr.h"
854ca1ab94SDag-Erling Smørgrav 
86e4878e39SDag-Erling Smørgrav /* Maximum number of redirects to follow */
87e4878e39SDag-Erling Smørgrav #define MAX_REDIRECT 5
884ca1ab94SDag-Erling Smørgrav 
89e4878e39SDag-Erling Smørgrav /* Symbolic names for reply codes we care about */
903d2a8471SDag-Erling Smørgrav #define HTTP_OK			200
913d2a8471SDag-Erling Smørgrav #define HTTP_PARTIAL		206
92e4878e39SDag-Erling Smørgrav #define HTTP_MOVED_PERM		301
93e4878e39SDag-Erling Smørgrav #define HTTP_MOVED_TEMP		302
94e4878e39SDag-Erling Smørgrav #define HTTP_SEE_OTHER		303
95cd842257SDag-Erling Smørgrav #define HTTP_TEMP_REDIRECT	307
96e4878e39SDag-Erling Smørgrav #define HTTP_NEED_AUTH		401
9765986545SDag-Erling Smørgrav #define HTTP_NEED_PROXY_AUTH	407
98b5e12145SDag-Erling Smørgrav #define HTTP_BAD_RANGE		416
99e4878e39SDag-Erling Smørgrav #define HTTP_PROTOCOL_ERROR	999
100e4878e39SDag-Erling Smørgrav 
101e4878e39SDag-Erling Smørgrav #define HTTP_REDIRECT(xyz) ((xyz) == HTTP_MOVED_PERM \
102e4878e39SDag-Erling Smørgrav 			    || (xyz) == HTTP_MOVED_TEMP \
103cd842257SDag-Erling Smørgrav 			    || (xyz) == HTTP_TEMP_REDIRECT \
104e4878e39SDag-Erling Smørgrav 			    || (xyz) == HTTP_SEE_OTHER)
105e4878e39SDag-Erling Smørgrav 
106a8e9bd87SDag-Erling Smørgrav #define HTTP_ERROR(xyz) ((xyz) > 400 && (xyz) < 599)
107e4878e39SDag-Erling Smørgrav 
108e19e6098SDag-Erling Smørgrav 
109e4878e39SDag-Erling Smørgrav /*****************************************************************************
110e4878e39SDag-Erling Smørgrav  * I/O functions for decoding chunked streams
111e4878e39SDag-Erling Smørgrav  */
1123d2a8471SDag-Erling Smørgrav 
113ccdd94bdSDag-Erling Smørgrav struct httpio
1144ca1ab94SDag-Erling Smørgrav {
115d3b03a90SDag-Erling Smørgrav 	conn_t		*conn;		/* connection */
1169601e333SDag-Erling Smørgrav 	int		 chunked;	/* chunked mode */
117d3b03a90SDag-Erling Smørgrav 	char		*buf;		/* chunk buffer */
1189601e333SDag-Erling Smørgrav 	size_t		 bufsize;	/* size of chunk buffer */
1199601e333SDag-Erling Smørgrav 	ssize_t		 buflen;	/* amount of data currently in buffer */
1209601e333SDag-Erling Smørgrav 	int		 bufpos;	/* current read offset in buffer */
121d3b03a90SDag-Erling Smørgrav 	int		 eof;		/* end-of-file flag */
122d3b03a90SDag-Erling Smørgrav 	int		 error;		/* error flag */
123d3b03a90SDag-Erling Smørgrav 	size_t		 chunksize;	/* remaining size of current chunk */
124e66b3802SDag-Erling Smørgrav #ifndef NDEBUG
125f573a5fcSDag-Erling Smørgrav 	size_t		 total;
126e4878e39SDag-Erling Smørgrav #endif
1274ca1ab94SDag-Erling Smørgrav };
1284ca1ab94SDag-Erling Smørgrav 
129f62e5228SDag-Erling Smørgrav /*
130e4878e39SDag-Erling Smørgrav  * Get next chunk header
131f62e5228SDag-Erling Smørgrav  */
132f62e5228SDag-Erling Smørgrav static int
133a1b37df2SDag-Erling Smørgrav http_new_chunk(struct httpio *io)
134f62e5228SDag-Erling Smørgrav {
135e4878e39SDag-Erling Smørgrav 	char *p;
136f62e5228SDag-Erling Smørgrav 
137a1b37df2SDag-Erling Smørgrav 	if (fetch_getln(io->conn) == -1)
138e19e6098SDag-Erling Smørgrav 		return (-1);
139f62e5228SDag-Erling Smørgrav 
140facd9827SDag-Erling Smørgrav 	if (io->conn->buflen < 2 || !isxdigit((unsigned char)*io->conn->buf))
141e19e6098SDag-Erling Smørgrav 		return (-1);
142e4878e39SDag-Erling Smørgrav 
143facd9827SDag-Erling Smørgrav 	for (p = io->conn->buf; *p && !isspace((unsigned char)*p); ++p) {
144e19e6098SDag-Erling Smørgrav 		if (*p == ';')
145e19e6098SDag-Erling Smørgrav 			break;
146facd9827SDag-Erling Smørgrav 		if (!isxdigit((unsigned char)*p))
147e19e6098SDag-Erling Smørgrav 			return (-1);
148facd9827SDag-Erling Smørgrav 		if (isdigit((unsigned char)*p)) {
149ccdd94bdSDag-Erling Smørgrav 			io->chunksize = io->chunksize * 16 +
150e19e6098SDag-Erling Smørgrav 			    *p - '0';
151e19e6098SDag-Erling Smørgrav 		} else {
152ccdd94bdSDag-Erling Smørgrav 			io->chunksize = io->chunksize * 16 +
15379257dd7SDag-Erling Smørgrav 			    10 + tolower((unsigned char)*p) - 'a';
154e19e6098SDag-Erling Smørgrav 		}
155e19e6098SDag-Erling Smørgrav 	}
156e4878e39SDag-Erling Smørgrav 
157e66b3802SDag-Erling Smørgrav #ifndef NDEBUG
1587f807cb8SDag-Erling Smørgrav 	if (fetchDebug) {
159ccdd94bdSDag-Erling Smørgrav 		io->total += io->chunksize;
160ccdd94bdSDag-Erling Smørgrav 		if (io->chunksize == 0)
161a49b853dSDag-Erling Smørgrav 			fprintf(stderr, "%s(): end of last chunk\n", __func__);
162e4878e39SDag-Erling Smørgrav 		else
163a49b853dSDag-Erling Smørgrav 			fprintf(stderr, "%s(): new chunk: %lu (%lu)\n",
164a49b853dSDag-Erling Smørgrav 			    __func__, (unsigned long)io->chunksize,
165a49b853dSDag-Erling Smørgrav 			    (unsigned long)io->total);
1667f807cb8SDag-Erling Smørgrav 	}
167e4878e39SDag-Erling Smørgrav #endif
168e4878e39SDag-Erling Smørgrav 
169ccdd94bdSDag-Erling Smørgrav 	return (io->chunksize);
170f62e5228SDag-Erling Smørgrav }
171f62e5228SDag-Erling Smørgrav 
172f62e5228SDag-Erling Smørgrav /*
1739601e333SDag-Erling Smørgrav  * Grow the input buffer to at least len bytes
1749601e333SDag-Erling Smørgrav  */
1759601e333SDag-Erling Smørgrav static inline int
176a1b37df2SDag-Erling Smørgrav http_growbuf(struct httpio *io, size_t len)
1779601e333SDag-Erling Smørgrav {
1789601e333SDag-Erling Smørgrav 	char *tmp;
1799601e333SDag-Erling Smørgrav 
1809601e333SDag-Erling Smørgrav 	if (io->bufsize >= len)
1819601e333SDag-Erling Smørgrav 		return (0);
1829601e333SDag-Erling Smørgrav 
1839601e333SDag-Erling Smørgrav 	if ((tmp = realloc(io->buf, len)) == NULL)
1849601e333SDag-Erling Smørgrav 		return (-1);
1859601e333SDag-Erling Smørgrav 	io->buf = tmp;
1869601e333SDag-Erling Smørgrav 	io->bufsize = len;
1879f788e9cSDag-Erling Smørgrav 	return (0);
1889601e333SDag-Erling Smørgrav }
1899601e333SDag-Erling Smørgrav 
1909601e333SDag-Erling Smørgrav /*
191f62e5228SDag-Erling Smørgrav  * Fill the input buffer, do chunk decoding on the fly
192f62e5228SDag-Erling Smørgrav  */
193e4878e39SDag-Erling Smørgrav static int
194a1b37df2SDag-Erling Smørgrav http_fillbuf(struct httpio *io, size_t len)
1954ca1ab94SDag-Erling Smørgrav {
196ccdd94bdSDag-Erling Smørgrav 	if (io->error)
197e19e6098SDag-Erling Smørgrav 		return (-1);
198ccdd94bdSDag-Erling Smørgrav 	if (io->eof)
199e19e6098SDag-Erling Smørgrav 		return (0);
2004ca1ab94SDag-Erling Smørgrav 
2019601e333SDag-Erling Smørgrav 	if (io->chunked == 0) {
202a1b37df2SDag-Erling Smørgrav 		if (http_growbuf(io, len) == -1)
2039601e333SDag-Erling Smørgrav 			return (-1);
204a1b37df2SDag-Erling Smørgrav 		if ((io->buflen = fetch_read(io->conn, io->buf, len)) == -1) {
205cab84950SDag-Erling Smørgrav 			io->error = 1;
2069601e333SDag-Erling Smørgrav 			return (-1);
207cab84950SDag-Erling Smørgrav 		}
2089601e333SDag-Erling Smørgrav 		io->bufpos = 0;
2099601e333SDag-Erling Smørgrav 		return (io->buflen);
2109601e333SDag-Erling Smørgrav 	}
2119601e333SDag-Erling Smørgrav 
212ccdd94bdSDag-Erling Smørgrav 	if (io->chunksize == 0) {
213a1b37df2SDag-Erling Smørgrav 		switch (http_new_chunk(io)) {
214e4878e39SDag-Erling Smørgrav 		case -1:
215ccdd94bdSDag-Erling Smørgrav 			io->error = 1;
216e19e6098SDag-Erling Smørgrav 			return (-1);
217e4878e39SDag-Erling Smørgrav 		case 0:
218ccdd94bdSDag-Erling Smørgrav 			io->eof = 1;
219e19e6098SDag-Erling Smørgrav 			return (0);
2204ca1ab94SDag-Erling Smørgrav 		}
2214ca1ab94SDag-Erling Smørgrav 	}
222e4878e39SDag-Erling Smørgrav 
2239601e333SDag-Erling Smørgrav 	if (len > io->chunksize)
2249601e333SDag-Erling Smørgrav 		len = io->chunksize;
225a1b37df2SDag-Erling Smørgrav 	if (http_growbuf(io, len) == -1)
226e19e6098SDag-Erling Smørgrav 		return (-1);
227a1b37df2SDag-Erling Smørgrav 	if ((io->buflen = fetch_read(io->conn, io->buf, len)) == -1) {
228cab84950SDag-Erling Smørgrav 		io->error = 1;
229e19e6098SDag-Erling Smørgrav 		return (-1);
230cab84950SDag-Erling Smørgrav 	}
2319601e333SDag-Erling Smørgrav 	io->chunksize -= io->buflen;
232e4878e39SDag-Erling Smørgrav 
233ccdd94bdSDag-Erling Smørgrav 	if (io->chunksize == 0) {
234dea29ca1SDag-Erling Smørgrav 		char endl[2];
235dea29ca1SDag-Erling Smørgrav 
236a1b37df2SDag-Erling Smørgrav 		if (fetch_read(io->conn, endl, 2) != 2 ||
237dea29ca1SDag-Erling Smørgrav 		    endl[0] != '\r' || endl[1] != '\n')
238e19e6098SDag-Erling Smørgrav 			return (-1);
2394ca1ab94SDag-Erling Smørgrav 	}
240e4878e39SDag-Erling Smørgrav 
2419601e333SDag-Erling Smørgrav 	io->bufpos = 0;
242e4878e39SDag-Erling Smørgrav 
2439601e333SDag-Erling Smørgrav 	return (io->buflen);
2444ca1ab94SDag-Erling Smørgrav }
2454ca1ab94SDag-Erling Smørgrav 
246f62e5228SDag-Erling Smørgrav /*
247f62e5228SDag-Erling Smørgrav  * Read function
248f62e5228SDag-Erling Smørgrav  */
2494ca1ab94SDag-Erling Smørgrav static int
250a1b37df2SDag-Erling Smørgrav http_readfn(void *v, char *buf, int len)
2514ca1ab94SDag-Erling Smørgrav {
252ccdd94bdSDag-Erling Smørgrav 	struct httpio *io = (struct httpio *)v;
253e4878e39SDag-Erling Smørgrav 	int l, pos;
2544ca1ab94SDag-Erling Smørgrav 
255ccdd94bdSDag-Erling Smørgrav 	if (io->error)
256e19e6098SDag-Erling Smørgrav 		return (-1);
257ccdd94bdSDag-Erling Smørgrav 	if (io->eof)
258e19e6098SDag-Erling Smørgrav 		return (0);
259e4878e39SDag-Erling Smørgrav 
260e4878e39SDag-Erling Smørgrav 	for (pos = 0; len > 0; pos += l, len -= l) {
261e4878e39SDag-Erling Smørgrav 		/* empty buffer */
2629601e333SDag-Erling Smørgrav 		if (!io->buf || io->bufpos == io->buflen)
263a1b37df2SDag-Erling Smørgrav 			if (http_fillbuf(io, len) < 1)
264e4878e39SDag-Erling Smørgrav 				break;
2659601e333SDag-Erling Smørgrav 		l = io->buflen - io->bufpos;
266e4878e39SDag-Erling Smørgrav 		if (len < l)
267e4878e39SDag-Erling Smørgrav 			l = len;
2689601e333SDag-Erling Smørgrav 		bcopy(io->buf + io->bufpos, buf + pos, l);
2699601e333SDag-Erling Smørgrav 		io->bufpos += l;
2704ca1ab94SDag-Erling Smørgrav 	}
2714ca1ab94SDag-Erling Smørgrav 
272ccdd94bdSDag-Erling Smørgrav 	if (!pos && io->error)
273e19e6098SDag-Erling Smørgrav 		return (-1);
274e19e6098SDag-Erling Smørgrav 	return (pos);
2754ca1ab94SDag-Erling Smørgrav }
2764ca1ab94SDag-Erling Smørgrav 
277f62e5228SDag-Erling Smørgrav /*
278f62e5228SDag-Erling Smørgrav  * Write function
279f62e5228SDag-Erling Smørgrav  */
2804ca1ab94SDag-Erling Smørgrav static int
281a1b37df2SDag-Erling Smørgrav http_writefn(void *v, const char *buf, int len)
2824ca1ab94SDag-Erling Smørgrav {
283ccdd94bdSDag-Erling Smørgrav 	struct httpio *io = (struct httpio *)v;
284e4878e39SDag-Erling Smørgrav 
285a1b37df2SDag-Erling Smørgrav 	return (fetch_write(io->conn, buf, len));
2864ca1ab94SDag-Erling Smørgrav }
2874ca1ab94SDag-Erling Smørgrav 
288f62e5228SDag-Erling Smørgrav /*
289f62e5228SDag-Erling Smørgrav  * Close function
290f62e5228SDag-Erling Smørgrav  */
2914ca1ab94SDag-Erling Smørgrav static int
292a1b37df2SDag-Erling Smørgrav http_closefn(void *v)
2934ca1ab94SDag-Erling Smørgrav {
294ccdd94bdSDag-Erling Smørgrav 	struct httpio *io = (struct httpio *)v;
295e4878e39SDag-Erling Smørgrav 	int r;
296e4878e39SDag-Erling Smørgrav 
297a1b37df2SDag-Erling Smørgrav 	r = fetch_close(io->conn);
298ccdd94bdSDag-Erling Smørgrav 	if (io->buf)
299ccdd94bdSDag-Erling Smørgrav 		free(io->buf);
300ccdd94bdSDag-Erling Smørgrav 	free(io);
301e19e6098SDag-Erling Smørgrav 	return (r);
3024ca1ab94SDag-Erling Smørgrav }
3034ca1ab94SDag-Erling Smørgrav 
304f62e5228SDag-Erling Smørgrav /*
305e4878e39SDag-Erling Smørgrav  * Wrap a file descriptor up
306f62e5228SDag-Erling Smørgrav  */
307e4878e39SDag-Erling Smørgrav static FILE *
308a1b37df2SDag-Erling Smørgrav http_funopen(conn_t *conn, int chunked)
3094ca1ab94SDag-Erling Smørgrav {
310ccdd94bdSDag-Erling Smørgrav 	struct httpio *io;
311e4878e39SDag-Erling Smørgrav 	FILE *f;
312e4878e39SDag-Erling Smørgrav 
313930105c1SDag-Erling Smørgrav 	if ((io = calloc(1, sizeof(*io))) == NULL) {
314a1b37df2SDag-Erling Smørgrav 		fetch_syserr();
315e19e6098SDag-Erling Smørgrav 		return (NULL);
3164ca1ab94SDag-Erling Smørgrav 	}
317ccdd94bdSDag-Erling Smørgrav 	io->conn = conn;
3189601e333SDag-Erling Smørgrav 	io->chunked = chunked;
319a1b37df2SDag-Erling Smørgrav 	f = funopen(io, http_readfn, http_writefn, NULL, http_closefn);
320e19e6098SDag-Erling Smørgrav 	if (f == NULL) {
321a1b37df2SDag-Erling Smørgrav 		fetch_syserr();
322ccdd94bdSDag-Erling Smørgrav 		free(io);
323e19e6098SDag-Erling Smørgrav 		return (NULL);
324e4878e39SDag-Erling Smørgrav 	}
325e19e6098SDag-Erling Smørgrav 	return (f);
326e4878e39SDag-Erling Smørgrav }
327e4878e39SDag-Erling Smørgrav 
328e19e6098SDag-Erling Smørgrav 
329e4878e39SDag-Erling Smørgrav /*****************************************************************************
330e4878e39SDag-Erling Smørgrav  * Helper functions for talking to the server and parsing its replies
331e4878e39SDag-Erling Smørgrav  */
332e4878e39SDag-Erling Smørgrav 
333e4878e39SDag-Erling Smørgrav /* Header types */
334e4878e39SDag-Erling Smørgrav typedef enum {
335e4878e39SDag-Erling Smørgrav 	hdr_syserror = -2,
336e4878e39SDag-Erling Smørgrav 	hdr_error = -1,
337e4878e39SDag-Erling Smørgrav 	hdr_end = 0,
338e4878e39SDag-Erling Smørgrav 	hdr_unknown = 1,
339e4878e39SDag-Erling Smørgrav 	hdr_content_length,
340e4878e39SDag-Erling Smørgrav 	hdr_content_range,
341e4878e39SDag-Erling Smørgrav 	hdr_last_modified,
342e4878e39SDag-Erling Smørgrav 	hdr_location,
3436490b215SDag-Erling Smørgrav 	hdr_transfer_encoding,
3446490b215SDag-Erling Smørgrav 	hdr_www_authenticate
345f573a5fcSDag-Erling Smørgrav } hdr_t;
346e4878e39SDag-Erling Smørgrav 
347e4878e39SDag-Erling Smørgrav /* Names of interesting headers */
348e4878e39SDag-Erling Smørgrav static struct {
349f573a5fcSDag-Erling Smørgrav 	hdr_t		 num;
35038c7e4a6SArchie Cobbs 	const char	*name;
351e4878e39SDag-Erling Smørgrav } hdr_names[] = {
352e4878e39SDag-Erling Smørgrav 	{ hdr_content_length,		"Content-Length" },
353e4878e39SDag-Erling Smørgrav 	{ hdr_content_range,		"Content-Range" },
354e4878e39SDag-Erling Smørgrav 	{ hdr_last_modified,		"Last-Modified" },
355e4878e39SDag-Erling Smørgrav 	{ hdr_location,			"Location" },
356e4878e39SDag-Erling Smørgrav 	{ hdr_transfer_encoding,	"Transfer-Encoding" },
3576490b215SDag-Erling Smørgrav 	{ hdr_www_authenticate,		"WWW-Authenticate" },
358e4878e39SDag-Erling Smørgrav 	{ hdr_unknown,			NULL },
359e4878e39SDag-Erling Smørgrav };
360e4878e39SDag-Erling Smørgrav 
361e4878e39SDag-Erling Smørgrav /*
362e4878e39SDag-Erling Smørgrav  * Send a formatted line; optionally echo to terminal
363e4878e39SDag-Erling Smørgrav  */
364e4878e39SDag-Erling Smørgrav static int
365a1b37df2SDag-Erling Smørgrav http_cmd(conn_t *conn, const char *fmt, ...)
366e4878e39SDag-Erling Smørgrav {
367e4878e39SDag-Erling Smørgrav 	va_list ap;
368e4878e39SDag-Erling Smørgrav 	size_t len;
369e4878e39SDag-Erling Smørgrav 	char *msg;
370e4878e39SDag-Erling Smørgrav 	int r;
371e4878e39SDag-Erling Smørgrav 
372e4878e39SDag-Erling Smørgrav 	va_start(ap, fmt);
373e4878e39SDag-Erling Smørgrav 	len = vasprintf(&msg, fmt, ap);
374e4878e39SDag-Erling Smørgrav 	va_end(ap);
375e4878e39SDag-Erling Smørgrav 
376e4878e39SDag-Erling Smørgrav 	if (msg == NULL) {
377e4878e39SDag-Erling Smørgrav 		errno = ENOMEM;
378a1b37df2SDag-Erling Smørgrav 		fetch_syserr();
379e19e6098SDag-Erling Smørgrav 		return (-1);
380e4878e39SDag-Erling Smørgrav 	}
381e4878e39SDag-Erling Smørgrav 
382a1b37df2SDag-Erling Smørgrav 	r = fetch_putln(conn, msg, len);
383e4878e39SDag-Erling Smørgrav 	free(msg);
384e4878e39SDag-Erling Smørgrav 
385e4878e39SDag-Erling Smørgrav 	if (r == -1) {
386a1b37df2SDag-Erling Smørgrav 		fetch_syserr();
387e19e6098SDag-Erling Smørgrav 		return (-1);
388e4878e39SDag-Erling Smørgrav 	}
389e4878e39SDag-Erling Smørgrav 
390e19e6098SDag-Erling Smørgrav 	return (0);
391e4878e39SDag-Erling Smørgrav }
392e4878e39SDag-Erling Smørgrav 
393e4878e39SDag-Erling Smørgrav /*
394e4878e39SDag-Erling Smørgrav  * Get and parse status line
395e4878e39SDag-Erling Smørgrav  */
396e4878e39SDag-Erling Smørgrav static int
397a1b37df2SDag-Erling Smørgrav http_get_reply(conn_t *conn)
398e4878e39SDag-Erling Smørgrav {
399a898bb8dSDag-Erling Smørgrav 	char *p;
400a898bb8dSDag-Erling Smørgrav 
401a1b37df2SDag-Erling Smørgrav 	if (fetch_getln(conn) == -1)
402e19e6098SDag-Erling Smørgrav 		return (-1);
403e4878e39SDag-Erling Smørgrav 	/*
404e4878e39SDag-Erling Smørgrav 	 * A valid status line looks like "HTTP/m.n xyz reason" where m
405e4878e39SDag-Erling Smørgrav 	 * and n are the major and minor protocol version numbers and xyz
406e4878e39SDag-Erling Smørgrav 	 * is the reply code.
407a898bb8dSDag-Erling Smørgrav 	 * Unfortunately, there are servers out there (NCSA 1.5.1, to name
408a898bb8dSDag-Erling Smørgrav 	 * just one) that do not send a version number, so we can't rely
409a898bb8dSDag-Erling Smørgrav 	 * on finding one, but if we do, insist on it being 1.0 or 1.1.
410e4878e39SDag-Erling Smørgrav 	 * We don't care about the reason phrase.
411e4878e39SDag-Erling Smørgrav 	 */
412dea29ca1SDag-Erling Smørgrav 	if (strncmp(conn->buf, "HTTP", 4) != 0)
413e19e6098SDag-Erling Smørgrav 		return (HTTP_PROTOCOL_ERROR);
414dea29ca1SDag-Erling Smørgrav 	p = conn->buf + 4;
415a898bb8dSDag-Erling Smørgrav 	if (*p == '/') {
416a898bb8dSDag-Erling Smørgrav 		if (p[1] != '1' || p[2] != '.' || (p[3] != '0' && p[3] != '1'))
417e19e6098SDag-Erling Smørgrav 			return (HTTP_PROTOCOL_ERROR);
418a898bb8dSDag-Erling Smørgrav 		p += 4;
419a898bb8dSDag-Erling Smørgrav 	}
420facd9827SDag-Erling Smørgrav 	if (*p != ' ' ||
421facd9827SDag-Erling Smørgrav 	    !isdigit((unsigned char)p[1]) ||
422facd9827SDag-Erling Smørgrav 	    !isdigit((unsigned char)p[2]) ||
423facd9827SDag-Erling Smørgrav 	    !isdigit((unsigned char)p[3]))
424e19e6098SDag-Erling Smørgrav 		return (HTTP_PROTOCOL_ERROR);
425e4878e39SDag-Erling Smørgrav 
426dea29ca1SDag-Erling Smørgrav 	conn->err = (p[1] - '0') * 100 + (p[2] - '0') * 10 + (p[3] - '0');
427dea29ca1SDag-Erling Smørgrav 	return (conn->err);
428e4878e39SDag-Erling Smørgrav }
429e4878e39SDag-Erling Smørgrav 
430e4878e39SDag-Erling Smørgrav /*
431e19e6098SDag-Erling Smørgrav  * Check a header; if the type matches the given string, return a pointer
432e19e6098SDag-Erling Smørgrav  * to the beginning of the value.
433e4878e39SDag-Erling Smørgrav  */
43438c7e4a6SArchie Cobbs static const char *
435a1b37df2SDag-Erling Smørgrav http_match(const char *str, const char *hdr)
436e4878e39SDag-Erling Smørgrav {
43779257dd7SDag-Erling Smørgrav 	while (*str && *hdr &&
43879257dd7SDag-Erling Smørgrav 	    tolower((unsigned char)*str++) == tolower((unsigned char)*hdr++))
439e4878e39SDag-Erling Smørgrav 		/* nothing */;
440e4878e39SDag-Erling Smørgrav 	if (*str || *hdr != ':')
441e19e6098SDag-Erling Smørgrav 		return (NULL);
442facd9827SDag-Erling Smørgrav 	while (*hdr && isspace((unsigned char)*++hdr))
443e4878e39SDag-Erling Smørgrav 		/* nothing */;
444e19e6098SDag-Erling Smørgrav 	return (hdr);
445e4878e39SDag-Erling Smørgrav }
446e4878e39SDag-Erling Smørgrav 
447e4878e39SDag-Erling Smørgrav /*
448e4878e39SDag-Erling Smørgrav  * Get the next header and return the appropriate symbolic code.
449e4878e39SDag-Erling Smørgrav  */
450f573a5fcSDag-Erling Smørgrav static hdr_t
451a1b37df2SDag-Erling Smørgrav http_next_header(conn_t *conn, const char **p)
452e4878e39SDag-Erling Smørgrav {
453e4878e39SDag-Erling Smørgrav 	int i;
454e4878e39SDag-Erling Smørgrav 
455a1b37df2SDag-Erling Smørgrav 	if (fetch_getln(conn) == -1)
456e19e6098SDag-Erling Smørgrav 		return (hdr_syserror);
457facd9827SDag-Erling Smørgrav 	while (conn->buflen && isspace((unsigned char)conn->buf[conn->buflen - 1]))
458dea29ca1SDag-Erling Smørgrav 		conn->buflen--;
459dea29ca1SDag-Erling Smørgrav 	conn->buf[conn->buflen] = '\0';
460dea29ca1SDag-Erling Smørgrav 	if (conn->buflen == 0)
461e19e6098SDag-Erling Smørgrav 		return (hdr_end);
462e4878e39SDag-Erling Smørgrav 	/*
463e4878e39SDag-Erling Smørgrav 	 * We could check for malformed headers but we don't really care.
464e4878e39SDag-Erling Smørgrav 	 * A valid header starts with a token immediately followed by a
465e4878e39SDag-Erling Smørgrav 	 * colon; a token is any sequence of non-control, non-whitespace
466e4878e39SDag-Erling Smørgrav 	 * characters except "()<>@,;:\\\"{}".
467e4878e39SDag-Erling Smørgrav 	 */
468e4878e39SDag-Erling Smørgrav 	for (i = 0; hdr_names[i].num != hdr_unknown; i++)
469a1b37df2SDag-Erling Smørgrav 		if ((*p = http_match(hdr_names[i].name, conn->buf)) != NULL)
470e19e6098SDag-Erling Smørgrav 			return (hdr_names[i].num);
471e19e6098SDag-Erling Smørgrav 	return (hdr_unknown);
472e4878e39SDag-Erling Smørgrav }
473e4878e39SDag-Erling Smørgrav 
474e4878e39SDag-Erling Smørgrav /*
475e4878e39SDag-Erling Smørgrav  * Parse a last-modified header
476e4878e39SDag-Erling Smørgrav  */
477c78f1cc9SDag-Erling Smørgrav static int
478a1b37df2SDag-Erling Smørgrav http_parse_mtime(const char *p, time_t *mtime)
479e4878e39SDag-Erling Smørgrav {
480c78f1cc9SDag-Erling Smørgrav 	char locale[64], *r;
481e4878e39SDag-Erling Smørgrav 	struct tm tm;
482e4878e39SDag-Erling Smørgrav 
483930105c1SDag-Erling Smørgrav 	strncpy(locale, setlocale(LC_TIME, NULL), sizeof(locale));
484e4878e39SDag-Erling Smørgrav 	setlocale(LC_TIME, "C");
485c78f1cc9SDag-Erling Smørgrav 	r = strptime(p, "%a, %d %b %Y %H:%M:%S GMT", &tm);
486e4878e39SDag-Erling Smørgrav 	/* XXX should add support for date-2 and date-3 */
487e4878e39SDag-Erling Smørgrav 	setlocale(LC_TIME, locale);
488c78f1cc9SDag-Erling Smørgrav 	if (r == NULL)
489e19e6098SDag-Erling Smørgrav 		return (-1);
490f67efa37SDag-Erling Smørgrav 	DEBUG(fprintf(stderr, "last modified: [%04d-%02d-%02d "
491f67efa37SDag-Erling Smørgrav 		  "%02d:%02d:%02d]\n",
492e4878e39SDag-Erling Smørgrav 		  tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday,
493e4878e39SDag-Erling Smørgrav 		  tm.tm_hour, tm.tm_min, tm.tm_sec));
494c78f1cc9SDag-Erling Smørgrav 	*mtime = timegm(&tm);
495e19e6098SDag-Erling Smørgrav 	return (0);
496e4878e39SDag-Erling Smørgrav }
497e4878e39SDag-Erling Smørgrav 
498e4878e39SDag-Erling Smørgrav /*
499e4878e39SDag-Erling Smørgrav  * Parse a content-length header
500e4878e39SDag-Erling Smørgrav  */
501c78f1cc9SDag-Erling Smørgrav static int
502a1b37df2SDag-Erling Smørgrav http_parse_length(const char *p, off_t *length)
503e4878e39SDag-Erling Smørgrav {
504be6aff99SDag-Erling Smørgrav 	off_t len;
505e4878e39SDag-Erling Smørgrav 
506facd9827SDag-Erling Smørgrav 	for (len = 0; *p && isdigit((unsigned char)*p); ++p)
507e4878e39SDag-Erling Smørgrav 		len = len * 10 + (*p - '0');
508f573a5fcSDag-Erling Smørgrav 	if (*p)
509e19e6098SDag-Erling Smørgrav 		return (-1);
510f67efa37SDag-Erling Smørgrav 	DEBUG(fprintf(stderr, "content length: [%lld]\n",
511f573a5fcSDag-Erling Smørgrav 	    (long long)len));
512c78f1cc9SDag-Erling Smørgrav 	*length = len;
513e19e6098SDag-Erling Smørgrav 	return (0);
514e4878e39SDag-Erling Smørgrav }
515e4878e39SDag-Erling Smørgrav 
516e4878e39SDag-Erling Smørgrav /*
517e4878e39SDag-Erling Smørgrav  * Parse a content-range header
518e4878e39SDag-Erling Smørgrav  */
519c78f1cc9SDag-Erling Smørgrav static int
520a1b37df2SDag-Erling Smørgrav http_parse_range(const char *p, off_t *offset, off_t *length, off_t *size)
521e4878e39SDag-Erling Smørgrav {
522f573a5fcSDag-Erling Smørgrav 	off_t first, last, len;
523e4878e39SDag-Erling Smørgrav 
524e4878e39SDag-Erling Smørgrav 	if (strncasecmp(p, "bytes ", 6) != 0)
525e19e6098SDag-Erling Smørgrav 		return (-1);
526b5e12145SDag-Erling Smørgrav 	p += 6;
527b5e12145SDag-Erling Smørgrav 	if (*p == '*') {
528b5e12145SDag-Erling Smørgrav 		first = last = -1;
529b5e12145SDag-Erling Smørgrav 		++p;
530b5e12145SDag-Erling Smørgrav 	} else {
531facd9827SDag-Erling Smørgrav 		for (first = 0; *p && isdigit((unsigned char)*p); ++p)
532c78f1cc9SDag-Erling Smørgrav 			first = first * 10 + *p - '0';
533e4878e39SDag-Erling Smørgrav 		if (*p != '-')
534e19e6098SDag-Erling Smørgrav 			return (-1);
535facd9827SDag-Erling Smørgrav 		for (last = 0, ++p; *p && isdigit((unsigned char)*p); ++p)
536c78f1cc9SDag-Erling Smørgrav 			last = last * 10 + *p - '0';
537b5e12145SDag-Erling Smørgrav 	}
538c78f1cc9SDag-Erling Smørgrav 	if (first > last || *p != '/')
539e19e6098SDag-Erling Smørgrav 		return (-1);
540facd9827SDag-Erling Smørgrav 	for (len = 0, ++p; *p && isdigit((unsigned char)*p); ++p)
541c78f1cc9SDag-Erling Smørgrav 		len = len * 10 + *p - '0';
542f573a5fcSDag-Erling Smørgrav 	if (*p || len < last - first + 1)
543e19e6098SDag-Erling Smørgrav 		return (-1);
544b5e12145SDag-Erling Smørgrav 	if (first == -1) {
545b5e12145SDag-Erling Smørgrav 		DEBUG(fprintf(stderr, "content range: [*/%lld]\n",
546b5e12145SDag-Erling Smørgrav 		    (long long)len));
547b5e12145SDag-Erling Smørgrav 		*length = 0;
548b5e12145SDag-Erling Smørgrav 	} else {
549f67efa37SDag-Erling Smørgrav 		DEBUG(fprintf(stderr, "content range: [%lld-%lld/%lld]\n",
550f573a5fcSDag-Erling Smørgrav 		    (long long)first, (long long)last, (long long)len));
551c78f1cc9SDag-Erling Smørgrav 		*length = last - first + 1;
552b5e12145SDag-Erling Smørgrav 	}
553b5e12145SDag-Erling Smørgrav 	*offset = first;
554c78f1cc9SDag-Erling Smørgrav 	*size = len;
555e19e6098SDag-Erling Smørgrav 	return (0);
556e4878e39SDag-Erling Smørgrav }
557e4878e39SDag-Erling Smørgrav 
558e19e6098SDag-Erling Smørgrav 
559e4878e39SDag-Erling Smørgrav /*****************************************************************************
560e4878e39SDag-Erling Smørgrav  * Helper functions for authorization
561e4878e39SDag-Erling Smørgrav  */
5624ca1ab94SDag-Erling Smørgrav 
563f62e5228SDag-Erling Smørgrav /*
564f62e5228SDag-Erling Smørgrav  * Base64 encoding
565f62e5228SDag-Erling Smørgrav  */
56635f723dbSDag-Erling Smørgrav static char *
567a1b37df2SDag-Erling Smørgrav http_base64(const char *src)
568f62e5228SDag-Erling Smørgrav {
569f62e5228SDag-Erling Smørgrav 	static const char base64[] =
570f62e5228SDag-Erling Smørgrav 	    "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
571f62e5228SDag-Erling Smørgrav 	    "abcdefghijklmnopqrstuvwxyz"
572f62e5228SDag-Erling Smørgrav 	    "0123456789+/";
57335f723dbSDag-Erling Smørgrav 	char *str, *dst;
57435f723dbSDag-Erling Smørgrav 	size_t l;
57535f723dbSDag-Erling Smørgrav 	int t, r;
57635f723dbSDag-Erling Smørgrav 
57735f723dbSDag-Erling Smørgrav 	l = strlen(src);
578264a4c03SDag-Erling Smørgrav 	if ((str = malloc(((l + 2) / 3) * 4 + 1)) == NULL)
579e19e6098SDag-Erling Smørgrav 		return (NULL);
58035f723dbSDag-Erling Smørgrav 	dst = str;
58135f723dbSDag-Erling Smørgrav 	r = 0;
582f62e5228SDag-Erling Smørgrav 
583f62e5228SDag-Erling Smørgrav 	while (l >= 3) {
584f62e5228SDag-Erling Smørgrav 		t = (src[0] << 16) | (src[1] << 8) | src[2];
585f62e5228SDag-Erling Smørgrav 		dst[0] = base64[(t >> 18) & 0x3f];
586f62e5228SDag-Erling Smørgrav 		dst[1] = base64[(t >> 12) & 0x3f];
587f62e5228SDag-Erling Smørgrav 		dst[2] = base64[(t >> 6) & 0x3f];
588f62e5228SDag-Erling Smørgrav 		dst[3] = base64[(t >> 0) & 0x3f];
589f62e5228SDag-Erling Smørgrav 		src += 3; l -= 3;
590f62e5228SDag-Erling Smørgrav 		dst += 4; r += 4;
591f62e5228SDag-Erling Smørgrav 	}
592f62e5228SDag-Erling Smørgrav 
593f62e5228SDag-Erling Smørgrav 	switch (l) {
594f62e5228SDag-Erling Smørgrav 	case 2:
595f62e5228SDag-Erling Smørgrav 		t = (src[0] << 16) | (src[1] << 8);
596f62e5228SDag-Erling Smørgrav 		dst[0] = base64[(t >> 18) & 0x3f];
597f62e5228SDag-Erling Smørgrav 		dst[1] = base64[(t >> 12) & 0x3f];
598f62e5228SDag-Erling Smørgrav 		dst[2] = base64[(t >> 6) & 0x3f];
599f62e5228SDag-Erling Smørgrav 		dst[3] = '=';
600f62e5228SDag-Erling Smørgrav 		dst += 4;
601f62e5228SDag-Erling Smørgrav 		r += 4;
602f62e5228SDag-Erling Smørgrav 		break;
603f62e5228SDag-Erling Smørgrav 	case 1:
604f62e5228SDag-Erling Smørgrav 		t = src[0] << 16;
605f62e5228SDag-Erling Smørgrav 		dst[0] = base64[(t >> 18) & 0x3f];
606f62e5228SDag-Erling Smørgrav 		dst[1] = base64[(t >> 12) & 0x3f];
607f62e5228SDag-Erling Smørgrav 		dst[2] = dst[3] = '=';
608f62e5228SDag-Erling Smørgrav 		dst += 4;
609f62e5228SDag-Erling Smørgrav 		r += 4;
610f62e5228SDag-Erling Smørgrav 		break;
611f62e5228SDag-Erling Smørgrav 	case 0:
612f62e5228SDag-Erling Smørgrav 		break;
613f62e5228SDag-Erling Smørgrav 	}
614f62e5228SDag-Erling Smørgrav 
615f62e5228SDag-Erling Smørgrav 	*dst = 0;
616e19e6098SDag-Erling Smørgrav 	return (str);
617f62e5228SDag-Erling Smørgrav }
618f62e5228SDag-Erling Smørgrav 
619f62e5228SDag-Erling Smørgrav /*
620f62e5228SDag-Erling Smørgrav  * Encode username and password
621f62e5228SDag-Erling Smørgrav  */
62235f723dbSDag-Erling Smørgrav static int
623a1b37df2SDag-Erling Smørgrav http_basic_auth(conn_t *conn, const char *hdr, const char *usr, const char *pwd)
624f62e5228SDag-Erling Smørgrav {
62535f723dbSDag-Erling Smørgrav 	char *upw, *auth;
62635f723dbSDag-Erling Smørgrav 	int r;
627f62e5228SDag-Erling Smørgrav 
628f67efa37SDag-Erling Smørgrav 	DEBUG(fprintf(stderr, "usr: [%s]\n", usr));
629f67efa37SDag-Erling Smørgrav 	DEBUG(fprintf(stderr, "pwd: [%s]\n", pwd));
63035f723dbSDag-Erling Smørgrav 	if (asprintf(&upw, "%s:%s", usr, pwd) == -1)
631e19e6098SDag-Erling Smørgrav 		return (-1);
632a1b37df2SDag-Erling Smørgrav 	auth = http_base64(upw);
63335f723dbSDag-Erling Smørgrav 	free(upw);
63435f723dbSDag-Erling Smørgrav 	if (auth == NULL)
635e19e6098SDag-Erling Smørgrav 		return (-1);
636a1b37df2SDag-Erling Smørgrav 	r = http_cmd(conn, "%s: Basic %s", hdr, auth);
63735f723dbSDag-Erling Smørgrav 	free(auth);
638e19e6098SDag-Erling Smørgrav 	return (r);
6394d029f13SDag-Erling Smørgrav }
64035f723dbSDag-Erling Smørgrav 
64135f723dbSDag-Erling Smørgrav /*
64235f723dbSDag-Erling Smørgrav  * Send an authorization header
64335f723dbSDag-Erling Smørgrav  */
64435f723dbSDag-Erling Smørgrav static int
645a1b37df2SDag-Erling Smørgrav http_authorize(conn_t *conn, const char *hdr, const char *p)
64635f723dbSDag-Erling Smørgrav {
64735f723dbSDag-Erling Smørgrav 	/* basic authorization */
64835f723dbSDag-Erling Smørgrav 	if (strncasecmp(p, "basic:", 6) == 0) {
64935f723dbSDag-Erling Smørgrav 		char *user, *pwd, *str;
65035f723dbSDag-Erling Smørgrav 		int r;
65135f723dbSDag-Erling Smørgrav 
65235f723dbSDag-Erling Smørgrav 		/* skip realm */
65335f723dbSDag-Erling Smørgrav 		for (p += 6; *p && *p != ':'; ++p)
65435f723dbSDag-Erling Smørgrav 			/* nothing */ ;
65535f723dbSDag-Erling Smørgrav 		if (!*p || strchr(++p, ':') == NULL)
656e19e6098SDag-Erling Smørgrav 			return (-1);
65735f723dbSDag-Erling Smørgrav 		if ((str = strdup(p)) == NULL)
658e19e6098SDag-Erling Smørgrav 			return (-1); /* XXX */
65935f723dbSDag-Erling Smørgrav 		user = str;
66035f723dbSDag-Erling Smørgrav 		pwd = strchr(str, ':');
66135f723dbSDag-Erling Smørgrav 		*pwd++ = '\0';
662a1b37df2SDag-Erling Smørgrav 		r = http_basic_auth(conn, hdr, user, pwd);
66335f723dbSDag-Erling Smørgrav 		free(str);
664e19e6098SDag-Erling Smørgrav 		return (r);
6654d029f13SDag-Erling Smørgrav 	}
666e19e6098SDag-Erling Smørgrav 	return (-1);
667f62e5228SDag-Erling Smørgrav }
668f62e5228SDag-Erling Smørgrav 
669e19e6098SDag-Erling Smørgrav 
670e4878e39SDag-Erling Smørgrav /*****************************************************************************
671e4878e39SDag-Erling Smørgrav  * Helper functions for connecting to a server or proxy
672f62e5228SDag-Erling Smørgrav  */
673e4878e39SDag-Erling Smørgrav 
674e4878e39SDag-Erling Smørgrav /*
675e4878e39SDag-Erling Smørgrav  * Connect to the correct HTTP server or proxy.
67660245e42SDag-Erling Smørgrav  */
677dea29ca1SDag-Erling Smørgrav static conn_t *
678a1b37df2SDag-Erling Smørgrav http_connect(struct url *URL, struct url *purl, const char *flags)
67960245e42SDag-Erling Smørgrav {
680dea29ca1SDag-Erling Smørgrav 	conn_t *conn;
6811a16ed4cSDag-Erling Smørgrav 	int verbose;
68220a2a51bSKelly Yancey 	int af, val;
683e4878e39SDag-Erling Smørgrav 
684e4878e39SDag-Erling Smørgrav #ifdef INET6
685e4878e39SDag-Erling Smørgrav 	af = AF_UNSPEC;
686e4878e39SDag-Erling Smørgrav #else
687e4878e39SDag-Erling Smørgrav 	af = AF_INET;
688e4878e39SDag-Erling Smørgrav #endif
689e4878e39SDag-Erling Smørgrav 
690d74a913bSDag-Erling Smørgrav 	verbose = CHECK_FLAG('v');
691d74a913bSDag-Erling Smørgrav 	if (CHECK_FLAG('4'))
692e4878e39SDag-Erling Smørgrav 		af = AF_INET;
6931a16ed4cSDag-Erling Smørgrav #ifdef INET6
694d74a913bSDag-Erling Smørgrav 	else if (CHECK_FLAG('6'))
695e4878e39SDag-Erling Smørgrav 		af = AF_INET6;
6961a16ed4cSDag-Erling Smørgrav #endif
697e4878e39SDag-Erling Smørgrav 
698111e2510SDag-Erling Smørgrav 	if (purl && strcasecmp(URL->scheme, SCHEME_HTTPS) != 0) {
6991a16ed4cSDag-Erling Smørgrav 		URL = purl;
7001a16ed4cSDag-Erling Smørgrav 	} else if (strcasecmp(URL->scheme, SCHEME_FTP) == 0) {
701e4878e39SDag-Erling Smørgrav 		/* can't talk http to an ftp server */
702e4878e39SDag-Erling Smørgrav 		/* XXX should set an error code */
703dea29ca1SDag-Erling Smørgrav 		return (NULL);
704e4878e39SDag-Erling Smørgrav 	}
7051a16ed4cSDag-Erling Smørgrav 
706a1b37df2SDag-Erling Smørgrav 	if ((conn = fetch_connect(URL->host, URL->port, af, verbose)) == NULL)
707a1b37df2SDag-Erling Smørgrav 		/* fetch_connect() has already set an error code */
708dea29ca1SDag-Erling Smørgrav 		return (NULL);
709111e2510SDag-Erling Smørgrav 	if (strcasecmp(URL->scheme, SCHEME_HTTPS) == 0 &&
710a1b37df2SDag-Erling Smørgrav 	    fetch_ssl(conn, verbose) == -1) {
711a1b37df2SDag-Erling Smørgrav 		fetch_close(conn);
7123070f6cbSDag-Erling Smørgrav 		/* grrr */
7133070f6cbSDag-Erling Smørgrav 		errno = EAUTH;
714a1b37df2SDag-Erling Smørgrav 		fetch_syserr();
715111e2510SDag-Erling Smørgrav 		return (NULL);
716111e2510SDag-Erling Smørgrav 	}
71720a2a51bSKelly Yancey 
71820a2a51bSKelly Yancey 	val = 1;
71920a2a51bSKelly Yancey 	setsockopt(conn->sd, IPPROTO_TCP, TCP_NOPUSH, &val, sizeof(val));
72020a2a51bSKelly Yancey 
721dea29ca1SDag-Erling Smørgrav 	return (conn);
722e4878e39SDag-Erling Smørgrav }
723e4878e39SDag-Erling Smørgrav 
7241a16ed4cSDag-Erling Smørgrav static struct url *
72562a2681cSDag-Erling Smørgrav http_get_proxy(struct url * url, const char *flags)
7261a16ed4cSDag-Erling Smørgrav {
7271a16ed4cSDag-Erling Smørgrav 	struct url *purl;
7281a16ed4cSDag-Erling Smørgrav 	char *p;
7291a16ed4cSDag-Erling Smørgrav 
730762892c0SDag-Erling Smørgrav 	if (flags != NULL && strchr(flags, 'd') != NULL)
731b91d6074SDag-Erling Smørgrav 		return (NULL);
73262a2681cSDag-Erling Smørgrav 	if (fetch_no_proxy_match(url->host))
73362a2681cSDag-Erling Smørgrav 		return (NULL);
7344cee73c8SDag-Erling Smørgrav 	if (((p = getenv("HTTP_PROXY")) || (p = getenv("http_proxy"))) &&
7352738229eSDag-Erling Smørgrav 	    *p && (purl = fetchParseURL(p))) {
7361a16ed4cSDag-Erling Smørgrav 		if (!*purl->scheme)
7371a16ed4cSDag-Erling Smørgrav 			strcpy(purl->scheme, SCHEME_HTTP);
7381a16ed4cSDag-Erling Smørgrav 		if (!purl->port)
739a1b37df2SDag-Erling Smørgrav 			purl->port = fetch_default_proxy_port(purl->scheme);
7401a16ed4cSDag-Erling Smørgrav 		if (strcasecmp(purl->scheme, SCHEME_HTTP) == 0)
741e19e6098SDag-Erling Smørgrav 			return (purl);
7421a16ed4cSDag-Erling Smørgrav 		fetchFreeURL(purl);
7431a16ed4cSDag-Erling Smørgrav 	}
744e19e6098SDag-Erling Smørgrav 	return (NULL);
745e4878e39SDag-Erling Smørgrav }
746e4878e39SDag-Erling Smørgrav 
747a8e9bd87SDag-Erling Smørgrav static void
748a1b37df2SDag-Erling Smørgrav http_print_html(FILE *out, FILE *in)
749a8e9bd87SDag-Erling Smørgrav {
750a8e9bd87SDag-Erling Smørgrav 	size_t len;
751a8e9bd87SDag-Erling Smørgrav 	char *line, *p, *q;
752a8e9bd87SDag-Erling Smørgrav 	int comment, tag;
753a8e9bd87SDag-Erling Smørgrav 
754a8e9bd87SDag-Erling Smørgrav 	comment = tag = 0;
755a8e9bd87SDag-Erling Smørgrav 	while ((line = fgetln(in, &len)) != NULL) {
756facd9827SDag-Erling Smørgrav 		while (len && isspace((unsigned char)line[len - 1]))
757a8e9bd87SDag-Erling Smørgrav 			--len;
758a8e9bd87SDag-Erling Smørgrav 		for (p = q = line; q < line + len; ++q) {
759a8e9bd87SDag-Erling Smørgrav 			if (comment && *q == '-') {
760e19e6098SDag-Erling Smørgrav 				if (q + 2 < line + len &&
761e19e6098SDag-Erling Smørgrav 				    strcmp(q, "-->") == 0) {
762a8e9bd87SDag-Erling Smørgrav 					tag = comment = 0;
763a8e9bd87SDag-Erling Smørgrav 					q += 2;
764a8e9bd87SDag-Erling Smørgrav 				}
765e19e6098SDag-Erling Smørgrav 			} else if (tag && !comment && *q == '>') {
766a8e9bd87SDag-Erling Smørgrav 				p = q + 1;
767a8e9bd87SDag-Erling Smørgrav 				tag = 0;
768a8e9bd87SDag-Erling Smørgrav 			} else if (!tag && *q == '<') {
769a8e9bd87SDag-Erling Smørgrav 				if (q > p)
770a8e9bd87SDag-Erling Smørgrav 					fwrite(p, q - p, 1, out);
771a8e9bd87SDag-Erling Smørgrav 				tag = 1;
772e19e6098SDag-Erling Smørgrav 				if (q + 3 < line + len &&
773e19e6098SDag-Erling Smørgrav 				    strcmp(q, "<!--") == 0) {
774a8e9bd87SDag-Erling Smørgrav 					comment = 1;
775a8e9bd87SDag-Erling Smørgrav 					q += 3;
776a8e9bd87SDag-Erling Smørgrav 				}
777a8e9bd87SDag-Erling Smørgrav 			}
778a8e9bd87SDag-Erling Smørgrav 		}
779a8e9bd87SDag-Erling Smørgrav 		if (!tag && q > p)
780a8e9bd87SDag-Erling Smørgrav 			fwrite(p, q - p, 1, out);
781a8e9bd87SDag-Erling Smørgrav 		fputc('\n', out);
782a8e9bd87SDag-Erling Smørgrav 	}
783a8e9bd87SDag-Erling Smørgrav }
784a8e9bd87SDag-Erling Smørgrav 
785e19e6098SDag-Erling Smørgrav 
786e4878e39SDag-Erling Smørgrav /*****************************************************************************
787e4878e39SDag-Erling Smørgrav  * Core
788e4878e39SDag-Erling Smørgrav  */
789e4878e39SDag-Erling Smørgrav 
790e4878e39SDag-Erling Smørgrav /*
791e4878e39SDag-Erling Smørgrav  * Send a request and process the reply
7929601e333SDag-Erling Smørgrav  *
7939601e333SDag-Erling Smørgrav  * XXX This function is way too long, the do..while loop should be split
7949601e333SDag-Erling Smørgrav  * XXX off into a separate function.
795e4878e39SDag-Erling Smørgrav  */
7961a16ed4cSDag-Erling Smørgrav FILE *
797a1b37df2SDag-Erling Smørgrav http_request(struct url *URL, const char *op, struct url_stat *us,
79838c7e4a6SArchie Cobbs     struct url *purl, const char *flags)
799e4878e39SDag-Erling Smørgrav {
800dea29ca1SDag-Erling Smørgrav 	conn_t *conn;
801e4878e39SDag-Erling Smørgrav 	struct url *url, *new;
8021a16ed4cSDag-Erling Smørgrav 	int chunked, direct, need_auth, noredirect, verbose;
803448980e7SKelly Yancey 	int e, i, n, val;
804be6aff99SDag-Erling Smørgrav 	off_t offset, clength, length, size;
805c78f1cc9SDag-Erling Smørgrav 	time_t mtime;
80638c7e4a6SArchie Cobbs 	const char *p;
807e4878e39SDag-Erling Smørgrav 	FILE *f;
808f573a5fcSDag-Erling Smørgrav 	hdr_t h;
809d5216a4fSDag-Erling Smørgrav 	char hbuf[MAXHOSTNAMELEN + 7], *host;
81060245e42SDag-Erling Smørgrav 
811d74a913bSDag-Erling Smørgrav 	direct = CHECK_FLAG('d');
812d74a913bSDag-Erling Smørgrav 	noredirect = CHECK_FLAG('A');
813d74a913bSDag-Erling Smørgrav 	verbose = CHECK_FLAG('v');
8144ca1ab94SDag-Erling Smørgrav 
8151a16ed4cSDag-Erling Smørgrav 	if (direct && purl) {
8161a16ed4cSDag-Erling Smørgrav 		fetchFreeURL(purl);
8171a16ed4cSDag-Erling Smørgrav 		purl = NULL;
8181a16ed4cSDag-Erling Smørgrav 	}
8191a16ed4cSDag-Erling Smørgrav 
820c78f1cc9SDag-Erling Smørgrav 	/* try the provided URL first */
821c78f1cc9SDag-Erling Smørgrav 	url = URL;
822c78f1cc9SDag-Erling Smørgrav 
823c78f1cc9SDag-Erling Smørgrav 	/* if the A flag is set, we only get one try */
824e4878e39SDag-Erling Smørgrav 	n = noredirect ? 1 : MAX_REDIRECT;
825c78f1cc9SDag-Erling Smørgrav 	i = 0;
826e4878e39SDag-Erling Smørgrav 
827f8bd33a0SDag-Erling Smørgrav 	e = HTTP_PROTOCOL_ERROR;
8286490b215SDag-Erling Smørgrav 	need_auth = 0;
829c78f1cc9SDag-Erling Smørgrav 	do {
830f4683775SDag-Erling Smørgrav 		new = NULL;
831f4683775SDag-Erling Smørgrav 		chunked = 0;
832f4683775SDag-Erling Smørgrav 		offset = 0;
833c78f1cc9SDag-Erling Smørgrav 		clength = -1;
834c78f1cc9SDag-Erling Smørgrav 		length = -1;
835c78f1cc9SDag-Erling Smørgrav 		size = -1;
836c78f1cc9SDag-Erling Smørgrav 		mtime = 0;
8376490b215SDag-Erling Smørgrav 
8381a16ed4cSDag-Erling Smørgrav 		/* check port */
8391a16ed4cSDag-Erling Smørgrav 		if (!url->port)
840a1b37df2SDag-Erling Smørgrav 			url->port = fetch_default_port(url->scheme);
8411a16ed4cSDag-Erling Smørgrav 
84265986545SDag-Erling Smørgrav 		/* were we redirected to an FTP URL? */
84365986545SDag-Erling Smørgrav 		if (purl == NULL && strcmp(url->scheme, SCHEME_FTP) == 0) {
84465986545SDag-Erling Smørgrav 			if (strcmp(op, "GET") == 0)
845a1b37df2SDag-Erling Smørgrav 				return (ftp_request(url, "RETR", us, purl, flags));
84665986545SDag-Erling Smørgrav 			else if (strcmp(op, "HEAD") == 0)
847a1b37df2SDag-Erling Smørgrav 				return (ftp_request(url, "STAT", us, purl, flags));
84865986545SDag-Erling Smørgrav 		}
84965986545SDag-Erling Smørgrav 
850e4878e39SDag-Erling Smørgrav 		/* connect to server or proxy */
851a1b37df2SDag-Erling Smørgrav 		if ((conn = http_connect(url, purl, flags)) == NULL)
852e4878e39SDag-Erling Smørgrav 			goto ouch;
853e4878e39SDag-Erling Smørgrav 
854e4878e39SDag-Erling Smørgrav 		host = url->host;
85528c645cfSHajimu UMEMOTO #ifdef INET6
856e4878e39SDag-Erling Smørgrav 		if (strchr(url->host, ':')) {
857e4878e39SDag-Erling Smørgrav 			snprintf(hbuf, sizeof(hbuf), "[%s]", url->host);
85828c645cfSHajimu UMEMOTO 			host = hbuf;
85928c645cfSHajimu UMEMOTO 		}
86028c645cfSHajimu UMEMOTO #endif
861a1b37df2SDag-Erling Smørgrav 		if (url->port != fetch_default_port(url->scheme)) {
862d5216a4fSDag-Erling Smørgrav 			if (host != hbuf) {
863d5216a4fSDag-Erling Smørgrav 				strcpy(hbuf, host);
864d5216a4fSDag-Erling Smørgrav 				host = hbuf;
865d5216a4fSDag-Erling Smørgrav 			}
866d5216a4fSDag-Erling Smørgrav 			snprintf(hbuf + strlen(hbuf),
867d5216a4fSDag-Erling Smørgrav 			    sizeof(hbuf) - strlen(hbuf), ":%d", url->port);
868d5216a4fSDag-Erling Smørgrav 		}
86928c645cfSHajimu UMEMOTO 
870e4878e39SDag-Erling Smørgrav 		/* send request */
8710fba3a00SDag-Erling Smørgrav 		if (verbose)
872a1b37df2SDag-Erling Smørgrav 			fetch_info("requesting %s://%s%s",
873d5216a4fSDag-Erling Smørgrav 			    url->scheme, host, url->doc);
8741a16ed4cSDag-Erling Smørgrav 		if (purl) {
875a1b37df2SDag-Erling Smørgrav 			http_cmd(conn, "%s %s://%s%s HTTP/1.1",
876d5216a4fSDag-Erling Smørgrav 			    op, url->scheme, host, url->doc);
877e4878e39SDag-Erling Smørgrav 		} else {
878a1b37df2SDag-Erling Smørgrav 			http_cmd(conn, "%s %s HTTP/1.1",
879e4878e39SDag-Erling Smørgrav 			    op, url->doc);
88060245e42SDag-Erling Smørgrav 		}
88160245e42SDag-Erling Smørgrav 
8826490b215SDag-Erling Smørgrav 		/* virtual host */
883a1b37df2SDag-Erling Smørgrav 		http_cmd(conn, "Host: %s", host);
8846490b215SDag-Erling Smørgrav 
885e4878e39SDag-Erling Smørgrav 		/* proxy authorization */
8861a16ed4cSDag-Erling Smørgrav 		if (purl) {
8871a16ed4cSDag-Erling Smørgrav 			if (*purl->user || *purl->pwd)
888a1b37df2SDag-Erling Smørgrav 				http_basic_auth(conn, "Proxy-Authorization",
8891a16ed4cSDag-Erling Smørgrav 				    purl->user, purl->pwd);
8901a16ed4cSDag-Erling Smørgrav 			else if ((p = getenv("HTTP_PROXY_AUTH")) != NULL && *p != '\0')
891a1b37df2SDag-Erling Smørgrav 				http_authorize(conn, "Proxy-Authorization", p);
8921a16ed4cSDag-Erling Smørgrav 		}
893e4878e39SDag-Erling Smørgrav 
894e4878e39SDag-Erling Smørgrav 		/* server authorization */
8956490b215SDag-Erling Smørgrav 		if (need_auth || *url->user || *url->pwd) {
896e4878e39SDag-Erling Smørgrav 			if (*url->user || *url->pwd)
897a1b37df2SDag-Erling Smørgrav 				http_basic_auth(conn, "Authorization", url->user, url->pwd);
898c78f1cc9SDag-Erling Smørgrav 			else if ((p = getenv("HTTP_AUTH")) != NULL && *p != '\0')
899a1b37df2SDag-Erling Smørgrav 				http_authorize(conn, "Authorization", p);
9006490b215SDag-Erling Smørgrav 			else if (fetchAuthMethod && fetchAuthMethod(url) == 0) {
901a1b37df2SDag-Erling Smørgrav 				http_basic_auth(conn, "Authorization", url->user, url->pwd);
9026490b215SDag-Erling Smørgrav 			} else {
903a1b37df2SDag-Erling Smørgrav 				http_seterr(HTTP_NEED_AUTH);
904e4878e39SDag-Erling Smørgrav 				goto ouch;
905e4878e39SDag-Erling Smørgrav 			}
906e4878e39SDag-Erling Smørgrav 		}
907e4878e39SDag-Erling Smørgrav 
908e4878e39SDag-Erling Smørgrav 		/* other headers */
909d5216a4fSDag-Erling Smørgrav 		if ((p = getenv("HTTP_REFERER")) != NULL && *p != '\0') {
910d5216a4fSDag-Erling Smørgrav 			if (strcasecmp(p, "auto") == 0)
911a1b37df2SDag-Erling Smørgrav 				http_cmd(conn, "Referer: %s://%s%s",
912d5216a4fSDag-Erling Smørgrav 				    url->scheme, host, url->doc);
913d5216a4fSDag-Erling Smørgrav 			else
914a1b37df2SDag-Erling Smørgrav 				http_cmd(conn, "Referer: %s", p);
915d5216a4fSDag-Erling Smørgrav 		}
9166a0cf64bSDag-Erling Smørgrav 		if ((p = getenv("HTTP_USER_AGENT")) != NULL && *p != '\0')
917a1b37df2SDag-Erling Smørgrav 			http_cmd(conn, "User-Agent: %s", p);
9186a0cf64bSDag-Erling Smørgrav 		else
919a1b37df2SDag-Erling Smørgrav 			http_cmd(conn, "User-Agent: %s " _LIBFETCH_VER, getprogname());
920591c0199SDag-Erling Smørgrav 		if (url->offset > 0)
921a1b37df2SDag-Erling Smørgrav 			http_cmd(conn, "Range: bytes=%lld-", (long long)url->offset);
922a1b37df2SDag-Erling Smørgrav 		http_cmd(conn, "Connection: close");
923a1b37df2SDag-Erling Smørgrav 		http_cmd(conn, "");
924448980e7SKelly Yancey 
925448980e7SKelly Yancey 		/*
926448980e7SKelly Yancey 		 * Force the queued request to be dispatched.  Normally, one
927448980e7SKelly Yancey 		 * would do this with shutdown(2) but squid proxies can be
928448980e7SKelly Yancey 		 * configured to disallow such half-closed connections.  To
929448980e7SKelly Yancey 		 * be compatible with such configurations, fiddle with socket
930448980e7SKelly Yancey 		 * options to force the pending data to be written.
931448980e7SKelly Yancey 		 */
932448980e7SKelly Yancey 		val = 0;
933448980e7SKelly Yancey 		setsockopt(conn->sd, IPPROTO_TCP, TCP_NOPUSH, &val,
934448980e7SKelly Yancey 			   sizeof(val));
935448980e7SKelly Yancey 		val = 1;
936448980e7SKelly Yancey 		setsockopt(conn->sd, IPPROTO_TCP, TCP_NODELAY, &val,
937448980e7SKelly Yancey 			   sizeof(val));
938e4878e39SDag-Erling Smørgrav 
939e4878e39SDag-Erling Smørgrav 		/* get reply */
940a1b37df2SDag-Erling Smørgrav 		switch (http_get_reply(conn)) {
941e4878e39SDag-Erling Smørgrav 		case HTTP_OK:
942e4878e39SDag-Erling Smørgrav 		case HTTP_PARTIAL:
943e4878e39SDag-Erling Smørgrav 			/* fine */
944e4878e39SDag-Erling Smørgrav 			break;
945e4878e39SDag-Erling Smørgrav 		case HTTP_MOVED_PERM:
946e4878e39SDag-Erling Smørgrav 		case HTTP_MOVED_TEMP:
94765986545SDag-Erling Smørgrav 		case HTTP_SEE_OTHER:
948e4878e39SDag-Erling Smørgrav 			/*
9492576e459SDag-Erling Smørgrav 			 * Not so fine, but we still have to read the
9502576e459SDag-Erling Smørgrav 			 * headers to get the new location.
951e4878e39SDag-Erling Smørgrav 			 */
952e4878e39SDag-Erling Smørgrav 			break;
953e4878e39SDag-Erling Smørgrav 		case HTTP_NEED_AUTH:
954e4878e39SDag-Erling Smørgrav 			if (need_auth) {
955e4878e39SDag-Erling Smørgrav 				/*
9562576e459SDag-Erling Smørgrav 				 * We already sent out authorization code,
9572576e459SDag-Erling Smørgrav 				 * so there's nothing more we can do.
958e4878e39SDag-Erling Smørgrav 				 */
959a1b37df2SDag-Erling Smørgrav 				http_seterr(conn->err);
960e4878e39SDag-Erling Smørgrav 				goto ouch;
961e4878e39SDag-Erling Smørgrav 			}
962e4878e39SDag-Erling Smørgrav 			/* try again, but send the password this time */
963e4878e39SDag-Erling Smørgrav 			if (verbose)
964a1b37df2SDag-Erling Smørgrav 				fetch_info("server requires authorization");
9656490b215SDag-Erling Smørgrav 			break;
966e4878e39SDag-Erling Smørgrav 		case HTTP_NEED_PROXY_AUTH:
967e4878e39SDag-Erling Smørgrav 			/*
9682576e459SDag-Erling Smørgrav 			 * If we're talking to a proxy, we already sent
9692576e459SDag-Erling Smørgrav 			 * our proxy authorization code, so there's
9702576e459SDag-Erling Smørgrav 			 * nothing more we can do.
971e4878e39SDag-Erling Smørgrav 			 */
972a1b37df2SDag-Erling Smørgrav 			http_seterr(conn->err);
973e4878e39SDag-Erling Smørgrav 			goto ouch;
974b5e12145SDag-Erling Smørgrav 		case HTTP_BAD_RANGE:
975b5e12145SDag-Erling Smørgrav 			/*
976b5e12145SDag-Erling Smørgrav 			 * This can happen if we ask for 0 bytes because
977b5e12145SDag-Erling Smørgrav 			 * we already have the whole file.  Consider this
978b5e12145SDag-Erling Smørgrav 			 * a success for now, and check sizes later.
979b5e12145SDag-Erling Smørgrav 			 */
980b5e12145SDag-Erling Smørgrav 			break;
981e4878e39SDag-Erling Smørgrav 		case HTTP_PROTOCOL_ERROR:
982e4878e39SDag-Erling Smørgrav 			/* fall through */
983e4878e39SDag-Erling Smørgrav 		case -1:
984a1b37df2SDag-Erling Smørgrav 			fetch_syserr();
985e4878e39SDag-Erling Smørgrav 			goto ouch;
986e4878e39SDag-Erling Smørgrav 		default:
987a1b37df2SDag-Erling Smørgrav 			http_seterr(conn->err);
988a8e9bd87SDag-Erling Smørgrav 			if (!verbose)
989e4878e39SDag-Erling Smørgrav 				goto ouch;
990a8e9bd87SDag-Erling Smørgrav 			/* fall through so we can get the full error message */
991e4878e39SDag-Erling Smørgrav 		}
992e4878e39SDag-Erling Smørgrav 
993e4878e39SDag-Erling Smørgrav 		/* get headers */
994e4878e39SDag-Erling Smørgrav 		do {
995a1b37df2SDag-Erling Smørgrav 			switch ((h = http_next_header(conn, &p))) {
996e4878e39SDag-Erling Smørgrav 			case hdr_syserror:
997a1b37df2SDag-Erling Smørgrav 				fetch_syserr();
998e4878e39SDag-Erling Smørgrav 				goto ouch;
999e4878e39SDag-Erling Smørgrav 			case hdr_error:
1000a1b37df2SDag-Erling Smørgrav 				http_seterr(HTTP_PROTOCOL_ERROR);
1001e4878e39SDag-Erling Smørgrav 				goto ouch;
1002e4878e39SDag-Erling Smørgrav 			case hdr_content_length:
1003a1b37df2SDag-Erling Smørgrav 				http_parse_length(p, &clength);
1004e4878e39SDag-Erling Smørgrav 				break;
1005e4878e39SDag-Erling Smørgrav 			case hdr_content_range:
1006a1b37df2SDag-Erling Smørgrav 				http_parse_range(p, &offset, &length, &size);
1007e4878e39SDag-Erling Smørgrav 				break;
1008e4878e39SDag-Erling Smørgrav 			case hdr_last_modified:
1009a1b37df2SDag-Erling Smørgrav 				http_parse_mtime(p, &mtime);
1010e4878e39SDag-Erling Smørgrav 				break;
1011e4878e39SDag-Erling Smørgrav 			case hdr_location:
1012dea29ca1SDag-Erling Smørgrav 				if (!HTTP_REDIRECT(conn->err))
1013e4878e39SDag-Erling Smørgrav 					break;
1014f4683775SDag-Erling Smørgrav 				if (new)
1015f4683775SDag-Erling Smørgrav 					free(new);
1016e4878e39SDag-Erling Smørgrav 				if (verbose)
1017a1b37df2SDag-Erling Smørgrav 					fetch_info("%d redirect to %s", conn->err, p);
1018f4683775SDag-Erling Smørgrav 				if (*p == '/')
1019f4683775SDag-Erling Smørgrav 					/* absolute path */
1020f4683775SDag-Erling Smørgrav 					new = fetchMakeURL(url->scheme, url->host, url->port, p,
1021f4683775SDag-Erling Smørgrav 					    url->user, url->pwd);
1022f4683775SDag-Erling Smørgrav 				else
1023f4683775SDag-Erling Smørgrav 					new = fetchParseURL(p);
1024f4683775SDag-Erling Smørgrav 				if (new == NULL) {
1025f4683775SDag-Erling Smørgrav 					/* XXX should set an error code */
1026f4683775SDag-Erling Smørgrav 					DEBUG(fprintf(stderr, "failed to parse new URL\n"));
1027e4878e39SDag-Erling Smørgrav 					goto ouch;
1028f4683775SDag-Erling Smørgrav 				}
1029e4878e39SDag-Erling Smørgrav 				if (!*new->user && !*new->pwd) {
1030e4878e39SDag-Erling Smørgrav 					strcpy(new->user, url->user);
1031e4878e39SDag-Erling Smørgrav 					strcpy(new->pwd, url->pwd);
1032e4878e39SDag-Erling Smørgrav 				}
1033e4878e39SDag-Erling Smørgrav 				new->offset = url->offset;
1034e4878e39SDag-Erling Smørgrav 				new->length = url->length;
1035f4683775SDag-Erling Smørgrav 				break;
1036e4878e39SDag-Erling Smørgrav 			case hdr_transfer_encoding:
1037e4878e39SDag-Erling Smørgrav 				/* XXX weak test*/
1038e4878e39SDag-Erling Smørgrav 				chunked = (strcasecmp(p, "chunked") == 0);
1039e4878e39SDag-Erling Smørgrav 				break;
10406490b215SDag-Erling Smørgrav 			case hdr_www_authenticate:
1041dea29ca1SDag-Erling Smørgrav 				if (conn->err != HTTP_NEED_AUTH)
10426490b215SDag-Erling Smørgrav 					break;
10436490b215SDag-Erling Smørgrav 				/* if we were smarter, we'd check the method and realm */
10446490b215SDag-Erling Smørgrav 				break;
1045e4878e39SDag-Erling Smørgrav 			case hdr_end:
1046e4878e39SDag-Erling Smørgrav 				/* fall through */
1047e4878e39SDag-Erling Smørgrav 			case hdr_unknown:
1048e4878e39SDag-Erling Smørgrav 				/* ignore */
1049e4878e39SDag-Erling Smørgrav 				break;
1050e4878e39SDag-Erling Smørgrav 			}
1051e4878e39SDag-Erling Smørgrav 		} while (h > hdr_end);
1052e4878e39SDag-Erling Smørgrav 
10536490b215SDag-Erling Smørgrav 		/* we need to provide authentication */
1054dea29ca1SDag-Erling Smørgrav 		if (conn->err == HTTP_NEED_AUTH) {
1055f8bd33a0SDag-Erling Smørgrav 			e = conn->err;
10566490b215SDag-Erling Smørgrav 			need_auth = 1;
1057a1b37df2SDag-Erling Smørgrav 			fetch_close(conn);
1058dea29ca1SDag-Erling Smørgrav 			conn = NULL;
10596490b215SDag-Erling Smørgrav 			continue;
10606490b215SDag-Erling Smørgrav 		}
10616490b215SDag-Erling Smørgrav 
1062b5e12145SDag-Erling Smørgrav 		/* requested range not satisfiable */
1063b5e12145SDag-Erling Smørgrav 		if (conn->err == HTTP_BAD_RANGE) {
1064b5e12145SDag-Erling Smørgrav 			if (url->offset == size && url->length == 0) {
1065b5e12145SDag-Erling Smørgrav 				/* asked for 0 bytes; fake it */
1066b5e12145SDag-Erling Smørgrav 				offset = url->offset;
1067b5e12145SDag-Erling Smørgrav 				conn->err = HTTP_OK;
1068b5e12145SDag-Erling Smørgrav 				break;
1069b5e12145SDag-Erling Smørgrav 			} else {
1070a1b37df2SDag-Erling Smørgrav 				http_seterr(conn->err);
1071b5e12145SDag-Erling Smørgrav 				goto ouch;
1072b5e12145SDag-Erling Smørgrav 			}
1073b5e12145SDag-Erling Smørgrav 		}
1074b5e12145SDag-Erling Smørgrav 
1075721d559cSRuslan Ermilov 		/* we have a hit or an error */
1076721d559cSRuslan Ermilov 		if (conn->err == HTTP_OK || conn->err == HTTP_PARTIAL || HTTP_ERROR(conn->err))
1077721d559cSRuslan Ermilov 			break;
1078721d559cSRuslan Ermilov 
10796490b215SDag-Erling Smørgrav 		/* all other cases: we got a redirect */
1080f8bd33a0SDag-Erling Smørgrav 		e = conn->err;
10816490b215SDag-Erling Smørgrav 		need_auth = 0;
1082a1b37df2SDag-Erling Smørgrav 		fetch_close(conn);
1083dea29ca1SDag-Erling Smørgrav 		conn = NULL;
10846490b215SDag-Erling Smørgrav 		if (!new) {
10856490b215SDag-Erling Smørgrav 			DEBUG(fprintf(stderr, "redirect with no new location\n"));
10866490b215SDag-Erling Smørgrav 			break;
10876490b215SDag-Erling Smørgrav 		}
1088f4683775SDag-Erling Smørgrav 		if (url != URL)
1089f4683775SDag-Erling Smørgrav 			fetchFreeURL(url);
1090f4683775SDag-Erling Smørgrav 		url = new;
1091c78f1cc9SDag-Erling Smørgrav 	} while (++i < n);
1092e4878e39SDag-Erling Smørgrav 
10936490b215SDag-Erling Smørgrav 	/* we failed, or ran out of retries */
1094dea29ca1SDag-Erling Smørgrav 	if (conn == NULL) {
1095a1b37df2SDag-Erling Smørgrav 		http_seterr(e);
1096e4878e39SDag-Erling Smørgrav 		goto ouch;
1097e4878e39SDag-Erling Smørgrav 	}
1098e4878e39SDag-Erling Smørgrav 
1099f573a5fcSDag-Erling Smørgrav 	DEBUG(fprintf(stderr, "offset %lld, length %lld,"
1100f573a5fcSDag-Erling Smørgrav 		  " size %lld, clength %lld\n",
1101f573a5fcSDag-Erling Smørgrav 		  (long long)offset, (long long)length,
1102f573a5fcSDag-Erling Smørgrav 		  (long long)size, (long long)clength));
1103c78f1cc9SDag-Erling Smørgrav 
1104c78f1cc9SDag-Erling Smørgrav 	/* check for inconsistencies */
1105c78f1cc9SDag-Erling Smørgrav 	if (clength != -1 && length != -1 && clength != length) {
1106a1b37df2SDag-Erling Smørgrav 		http_seterr(HTTP_PROTOCOL_ERROR);
1107c78f1cc9SDag-Erling Smørgrav 		goto ouch;
1108c78f1cc9SDag-Erling Smørgrav 	}
1109c78f1cc9SDag-Erling Smørgrav 	if (clength == -1)
1110c78f1cc9SDag-Erling Smørgrav 		clength = length;
1111c78f1cc9SDag-Erling Smørgrav 	if (clength != -1)
1112c78f1cc9SDag-Erling Smørgrav 		length = offset + clength;
1113c78f1cc9SDag-Erling Smørgrav 	if (length != -1 && size != -1 && length != size) {
1114a1b37df2SDag-Erling Smørgrav 		http_seterr(HTTP_PROTOCOL_ERROR);
1115c78f1cc9SDag-Erling Smørgrav 		goto ouch;
1116c78f1cc9SDag-Erling Smørgrav 	}
1117c78f1cc9SDag-Erling Smørgrav 	if (size == -1)
1118c78f1cc9SDag-Erling Smørgrav 		size = length;
1119c78f1cc9SDag-Erling Smørgrav 
1120c78f1cc9SDag-Erling Smørgrav 	/* fill in stats */
1121c78f1cc9SDag-Erling Smørgrav 	if (us) {
1122c78f1cc9SDag-Erling Smørgrav 		us->size = size;
1123c78f1cc9SDag-Erling Smørgrav 		us->atime = us->mtime = mtime;
1124c78f1cc9SDag-Erling Smørgrav 	}
1125c78f1cc9SDag-Erling Smørgrav 
112687b41116SDag-Erling Smørgrav 	/* too far? */
1127591c0199SDag-Erling Smørgrav 	if (URL->offset > 0 && offset > URL->offset) {
1128a1b37df2SDag-Erling Smørgrav 		http_seterr(HTTP_PROTOCOL_ERROR);
112987b41116SDag-Erling Smørgrav 		goto ouch;
113087b41116SDag-Erling Smørgrav 	}
113187b41116SDag-Erling Smørgrav 
1132c78f1cc9SDag-Erling Smørgrav 	/* report back real offset and size */
113387b41116SDag-Erling Smørgrav 	URL->offset = offset;
1134c78f1cc9SDag-Erling Smørgrav 	URL->length = clength;
113587b41116SDag-Erling Smørgrav 
1136e4878e39SDag-Erling Smørgrav 	/* wrap it up in a FILE */
1137a1b37df2SDag-Erling Smørgrav 	if ((f = http_funopen(conn, chunked)) == NULL) {
1138a1b37df2SDag-Erling Smørgrav 		fetch_syserr();
1139e4878e39SDag-Erling Smørgrav 		goto ouch;
1140e4878e39SDag-Erling Smørgrav 	}
1141e4878e39SDag-Erling Smørgrav 
1142e4878e39SDag-Erling Smørgrav 	if (url != URL)
1143e4878e39SDag-Erling Smørgrav 		fetchFreeURL(url);
11441a16ed4cSDag-Erling Smørgrav 	if (purl)
11451a16ed4cSDag-Erling Smørgrav 		fetchFreeURL(purl);
1146e4878e39SDag-Erling Smørgrav 
1147dea29ca1SDag-Erling Smørgrav 	if (HTTP_ERROR(conn->err)) {
1148a1b37df2SDag-Erling Smørgrav 		http_print_html(stderr, f);
1149a8e9bd87SDag-Erling Smørgrav 		fclose(f);
1150a8e9bd87SDag-Erling Smørgrav 		f = NULL;
1151a8e9bd87SDag-Erling Smørgrav 	}
1152a8e9bd87SDag-Erling Smørgrav 
1153e19e6098SDag-Erling Smørgrav 	return (f);
1154e4878e39SDag-Erling Smørgrav 
1155e4878e39SDag-Erling Smørgrav ouch:
1156e4878e39SDag-Erling Smørgrav 	if (url != URL)
1157e4878e39SDag-Erling Smørgrav 		fetchFreeURL(url);
11581a16ed4cSDag-Erling Smørgrav 	if (purl)
11591a16ed4cSDag-Erling Smørgrav 		fetchFreeURL(purl);
1160dea29ca1SDag-Erling Smørgrav 	if (conn != NULL)
1161a1b37df2SDag-Erling Smørgrav 		fetch_close(conn);
1162e19e6098SDag-Erling Smørgrav 	return (NULL);
1163e4878e39SDag-Erling Smørgrav }
1164e4878e39SDag-Erling Smørgrav 
1165e19e6098SDag-Erling Smørgrav 
1166e4878e39SDag-Erling Smørgrav /*****************************************************************************
1167e4878e39SDag-Erling Smørgrav  * Entry points
1168e4878e39SDag-Erling Smørgrav  */
1169e4878e39SDag-Erling Smørgrav 
117060245e42SDag-Erling Smørgrav /*
11711a5faa10SDag-Erling Smørgrav  * Retrieve and stat a file by HTTP
11721a5faa10SDag-Erling Smørgrav  */
11731a5faa10SDag-Erling Smørgrav FILE *
117438c7e4a6SArchie Cobbs fetchXGetHTTP(struct url *URL, struct url_stat *us, const char *flags)
11751a5faa10SDag-Erling Smørgrav {
117662a2681cSDag-Erling Smørgrav 	return (http_request(URL, "GET", us, http_get_proxy(URL, flags), flags));
11771a5faa10SDag-Erling Smørgrav }
11781a5faa10SDag-Erling Smørgrav 
11791a5faa10SDag-Erling Smørgrav /*
118060245e42SDag-Erling Smørgrav  * Retrieve a file by HTTP
118160245e42SDag-Erling Smørgrav  */
118260245e42SDag-Erling Smørgrav FILE *
118338c7e4a6SArchie Cobbs fetchGetHTTP(struct url *URL, const char *flags)
118460245e42SDag-Erling Smørgrav {
1185e19e6098SDag-Erling Smørgrav 	return (fetchXGetHTTP(URL, NULL, flags));
11864ca1ab94SDag-Erling Smørgrav }
11874ca1ab94SDag-Erling Smørgrav 
11881a5faa10SDag-Erling Smørgrav /*
11891a5faa10SDag-Erling Smørgrav  * Store a file by HTTP
11901a5faa10SDag-Erling Smørgrav  */
11914ca1ab94SDag-Erling Smørgrav FILE *
1192f573a5fcSDag-Erling Smørgrav fetchPutHTTP(struct url *URL __unused, const char *flags __unused)
11934ca1ab94SDag-Erling Smørgrav {
11944ca1ab94SDag-Erling Smørgrav 	warnx("fetchPutHTTP(): not implemented");
1195e19e6098SDag-Erling Smørgrav 	return (NULL);
11964ca1ab94SDag-Erling Smørgrav }
1197d8acd8dcSDag-Erling Smørgrav 
1198d8acd8dcSDag-Erling Smørgrav /*
1199d8acd8dcSDag-Erling Smørgrav  * Get an HTTP document's metadata
1200d8acd8dcSDag-Erling Smørgrav  */
1201d8acd8dcSDag-Erling Smørgrav int
120238c7e4a6SArchie Cobbs fetchStatHTTP(struct url *URL, struct url_stat *us, const char *flags)
1203d8acd8dcSDag-Erling Smørgrav {
120460245e42SDag-Erling Smørgrav 	FILE *f;
120560245e42SDag-Erling Smørgrav 
120662a2681cSDag-Erling Smørgrav 	f = http_request(URL, "HEAD", us, http_get_proxy(URL, flags), flags);
1207b91d6074SDag-Erling Smørgrav 	if (f == NULL)
1208e19e6098SDag-Erling Smørgrav 		return (-1);
1209f8f4130bSDag-Erling Smørgrav 	fclose(f);
1210e19e6098SDag-Erling Smørgrav 	return (0);
1211d8acd8dcSDag-Erling Smørgrav }
1212ce71b736SDag-Erling Smørgrav 
1213ce71b736SDag-Erling Smørgrav /*
1214ce71b736SDag-Erling Smørgrav  * List a directory
1215ce71b736SDag-Erling Smørgrav  */
1216ce71b736SDag-Erling Smørgrav struct url_ent *
1217f573a5fcSDag-Erling Smørgrav fetchListHTTP(struct url *url __unused, const char *flags __unused)
1218ce71b736SDag-Erling Smørgrav {
1219ce71b736SDag-Erling Smørgrav 	warnx("fetchListHTTP(): not implemented");
1220e19e6098SDag-Erling Smørgrav 	return (NULL);
1221ce71b736SDag-Erling Smørgrav }
1222