xref: /freebsd/lib/libfetch/http.c (revision facd9827942c7e735b3ecafb1e4719f022ab60c8)
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 +
153e19e6098SDag-Erling Smørgrav 			    10 + tolower(*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 {
437e4878e39SDag-Erling Smørgrav 	while (*str && *hdr && tolower(*str++) == tolower(*hdr++))
438e4878e39SDag-Erling Smørgrav 		/* nothing */;
439e4878e39SDag-Erling Smørgrav 	if (*str || *hdr != ':')
440e19e6098SDag-Erling Smørgrav 		return (NULL);
441facd9827SDag-Erling Smørgrav 	while (*hdr && isspace((unsigned char)*++hdr))
442e4878e39SDag-Erling Smørgrav 		/* nothing */;
443e19e6098SDag-Erling Smørgrav 	return (hdr);
444e4878e39SDag-Erling Smørgrav }
445e4878e39SDag-Erling Smørgrav 
446e4878e39SDag-Erling Smørgrav /*
447e4878e39SDag-Erling Smørgrav  * Get the next header and return the appropriate symbolic code.
448e4878e39SDag-Erling Smørgrav  */
449f573a5fcSDag-Erling Smørgrav static hdr_t
450a1b37df2SDag-Erling Smørgrav http_next_header(conn_t *conn, const char **p)
451e4878e39SDag-Erling Smørgrav {
452e4878e39SDag-Erling Smørgrav 	int i;
453e4878e39SDag-Erling Smørgrav 
454a1b37df2SDag-Erling Smørgrav 	if (fetch_getln(conn) == -1)
455e19e6098SDag-Erling Smørgrav 		return (hdr_syserror);
456facd9827SDag-Erling Smørgrav 	while (conn->buflen && isspace((unsigned char)conn->buf[conn->buflen - 1]))
457dea29ca1SDag-Erling Smørgrav 		conn->buflen--;
458dea29ca1SDag-Erling Smørgrav 	conn->buf[conn->buflen] = '\0';
459dea29ca1SDag-Erling Smørgrav 	if (conn->buflen == 0)
460e19e6098SDag-Erling Smørgrav 		return (hdr_end);
461e4878e39SDag-Erling Smørgrav 	/*
462e4878e39SDag-Erling Smørgrav 	 * We could check for malformed headers but we don't really care.
463e4878e39SDag-Erling Smørgrav 	 * A valid header starts with a token immediately followed by a
464e4878e39SDag-Erling Smørgrav 	 * colon; a token is any sequence of non-control, non-whitespace
465e4878e39SDag-Erling Smørgrav 	 * characters except "()<>@,;:\\\"{}".
466e4878e39SDag-Erling Smørgrav 	 */
467e4878e39SDag-Erling Smørgrav 	for (i = 0; hdr_names[i].num != hdr_unknown; i++)
468a1b37df2SDag-Erling Smørgrav 		if ((*p = http_match(hdr_names[i].name, conn->buf)) != NULL)
469e19e6098SDag-Erling Smørgrav 			return (hdr_names[i].num);
470e19e6098SDag-Erling Smørgrav 	return (hdr_unknown);
471e4878e39SDag-Erling Smørgrav }
472e4878e39SDag-Erling Smørgrav 
473e4878e39SDag-Erling Smørgrav /*
474e4878e39SDag-Erling Smørgrav  * Parse a last-modified header
475e4878e39SDag-Erling Smørgrav  */
476c78f1cc9SDag-Erling Smørgrav static int
477a1b37df2SDag-Erling Smørgrav http_parse_mtime(const char *p, time_t *mtime)
478e4878e39SDag-Erling Smørgrav {
479c78f1cc9SDag-Erling Smørgrav 	char locale[64], *r;
480e4878e39SDag-Erling Smørgrav 	struct tm tm;
481e4878e39SDag-Erling Smørgrav 
482930105c1SDag-Erling Smørgrav 	strncpy(locale, setlocale(LC_TIME, NULL), sizeof(locale));
483e4878e39SDag-Erling Smørgrav 	setlocale(LC_TIME, "C");
484c78f1cc9SDag-Erling Smørgrav 	r = strptime(p, "%a, %d %b %Y %H:%M:%S GMT", &tm);
485e4878e39SDag-Erling Smørgrav 	/* XXX should add support for date-2 and date-3 */
486e4878e39SDag-Erling Smørgrav 	setlocale(LC_TIME, locale);
487c78f1cc9SDag-Erling Smørgrav 	if (r == NULL)
488e19e6098SDag-Erling Smørgrav 		return (-1);
489f67efa37SDag-Erling Smørgrav 	DEBUG(fprintf(stderr, "last modified: [%04d-%02d-%02d "
490f67efa37SDag-Erling Smørgrav 		  "%02d:%02d:%02d]\n",
491e4878e39SDag-Erling Smørgrav 		  tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday,
492e4878e39SDag-Erling Smørgrav 		  tm.tm_hour, tm.tm_min, tm.tm_sec));
493c78f1cc9SDag-Erling Smørgrav 	*mtime = timegm(&tm);
494e19e6098SDag-Erling Smørgrav 	return (0);
495e4878e39SDag-Erling Smørgrav }
496e4878e39SDag-Erling Smørgrav 
497e4878e39SDag-Erling Smørgrav /*
498e4878e39SDag-Erling Smørgrav  * Parse a content-length header
499e4878e39SDag-Erling Smørgrav  */
500c78f1cc9SDag-Erling Smørgrav static int
501a1b37df2SDag-Erling Smørgrav http_parse_length(const char *p, off_t *length)
502e4878e39SDag-Erling Smørgrav {
503be6aff99SDag-Erling Smørgrav 	off_t len;
504e4878e39SDag-Erling Smørgrav 
505facd9827SDag-Erling Smørgrav 	for (len = 0; *p && isdigit((unsigned char)*p); ++p)
506e4878e39SDag-Erling Smørgrav 		len = len * 10 + (*p - '0');
507f573a5fcSDag-Erling Smørgrav 	if (*p)
508e19e6098SDag-Erling Smørgrav 		return (-1);
509f67efa37SDag-Erling Smørgrav 	DEBUG(fprintf(stderr, "content length: [%lld]\n",
510f573a5fcSDag-Erling Smørgrav 	    (long long)len));
511c78f1cc9SDag-Erling Smørgrav 	*length = len;
512e19e6098SDag-Erling Smørgrav 	return (0);
513e4878e39SDag-Erling Smørgrav }
514e4878e39SDag-Erling Smørgrav 
515e4878e39SDag-Erling Smørgrav /*
516e4878e39SDag-Erling Smørgrav  * Parse a content-range header
517e4878e39SDag-Erling Smørgrav  */
518c78f1cc9SDag-Erling Smørgrav static int
519a1b37df2SDag-Erling Smørgrav http_parse_range(const char *p, off_t *offset, off_t *length, off_t *size)
520e4878e39SDag-Erling Smørgrav {
521f573a5fcSDag-Erling Smørgrav 	off_t first, last, len;
522e4878e39SDag-Erling Smørgrav 
523e4878e39SDag-Erling Smørgrav 	if (strncasecmp(p, "bytes ", 6) != 0)
524e19e6098SDag-Erling Smørgrav 		return (-1);
525b5e12145SDag-Erling Smørgrav 	p += 6;
526b5e12145SDag-Erling Smørgrav 	if (*p == '*') {
527b5e12145SDag-Erling Smørgrav 		first = last = -1;
528b5e12145SDag-Erling Smørgrav 		++p;
529b5e12145SDag-Erling Smørgrav 	} else {
530facd9827SDag-Erling Smørgrav 		for (first = 0; *p && isdigit((unsigned char)*p); ++p)
531c78f1cc9SDag-Erling Smørgrav 			first = first * 10 + *p - '0';
532e4878e39SDag-Erling Smørgrav 		if (*p != '-')
533e19e6098SDag-Erling Smørgrav 			return (-1);
534facd9827SDag-Erling Smørgrav 		for (last = 0, ++p; *p && isdigit((unsigned char)*p); ++p)
535c78f1cc9SDag-Erling Smørgrav 			last = last * 10 + *p - '0';
536b5e12145SDag-Erling Smørgrav 	}
537c78f1cc9SDag-Erling Smørgrav 	if (first > last || *p != '/')
538e19e6098SDag-Erling Smørgrav 		return (-1);
539facd9827SDag-Erling Smørgrav 	for (len = 0, ++p; *p && isdigit((unsigned char)*p); ++p)
540c78f1cc9SDag-Erling Smørgrav 		len = len * 10 + *p - '0';
541f573a5fcSDag-Erling Smørgrav 	if (*p || len < last - first + 1)
542e19e6098SDag-Erling Smørgrav 		return (-1);
543b5e12145SDag-Erling Smørgrav 	if (first == -1) {
544b5e12145SDag-Erling Smørgrav 		DEBUG(fprintf(stderr, "content range: [*/%lld]\n",
545b5e12145SDag-Erling Smørgrav 		    (long long)len));
546b5e12145SDag-Erling Smørgrav 		*length = 0;
547b5e12145SDag-Erling Smørgrav 	} else {
548f67efa37SDag-Erling Smørgrav 		DEBUG(fprintf(stderr, "content range: [%lld-%lld/%lld]\n",
549f573a5fcSDag-Erling Smørgrav 		    (long long)first, (long long)last, (long long)len));
550c78f1cc9SDag-Erling Smørgrav 		*length = last - first + 1;
551b5e12145SDag-Erling Smørgrav 	}
552b5e12145SDag-Erling Smørgrav 	*offset = first;
553c78f1cc9SDag-Erling Smørgrav 	*size = len;
554e19e6098SDag-Erling Smørgrav 	return (0);
555e4878e39SDag-Erling Smørgrav }
556e4878e39SDag-Erling Smørgrav 
557e19e6098SDag-Erling Smørgrav 
558e4878e39SDag-Erling Smørgrav /*****************************************************************************
559e4878e39SDag-Erling Smørgrav  * Helper functions for authorization
560e4878e39SDag-Erling Smørgrav  */
5614ca1ab94SDag-Erling Smørgrav 
562f62e5228SDag-Erling Smørgrav /*
563f62e5228SDag-Erling Smørgrav  * Base64 encoding
564f62e5228SDag-Erling Smørgrav  */
56535f723dbSDag-Erling Smørgrav static char *
566a1b37df2SDag-Erling Smørgrav http_base64(const char *src)
567f62e5228SDag-Erling Smørgrav {
568f62e5228SDag-Erling Smørgrav 	static const char base64[] =
569f62e5228SDag-Erling Smørgrav 	    "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
570f62e5228SDag-Erling Smørgrav 	    "abcdefghijklmnopqrstuvwxyz"
571f62e5228SDag-Erling Smørgrav 	    "0123456789+/";
57235f723dbSDag-Erling Smørgrav 	char *str, *dst;
57335f723dbSDag-Erling Smørgrav 	size_t l;
57435f723dbSDag-Erling Smørgrav 	int t, r;
57535f723dbSDag-Erling Smørgrav 
57635f723dbSDag-Erling Smørgrav 	l = strlen(src);
577264a4c03SDag-Erling Smørgrav 	if ((str = malloc(((l + 2) / 3) * 4 + 1)) == NULL)
578e19e6098SDag-Erling Smørgrav 		return (NULL);
57935f723dbSDag-Erling Smørgrav 	dst = str;
58035f723dbSDag-Erling Smørgrav 	r = 0;
581f62e5228SDag-Erling Smørgrav 
582f62e5228SDag-Erling Smørgrav 	while (l >= 3) {
583f62e5228SDag-Erling Smørgrav 		t = (src[0] << 16) | (src[1] << 8) | src[2];
584f62e5228SDag-Erling Smørgrav 		dst[0] = base64[(t >> 18) & 0x3f];
585f62e5228SDag-Erling Smørgrav 		dst[1] = base64[(t >> 12) & 0x3f];
586f62e5228SDag-Erling Smørgrav 		dst[2] = base64[(t >> 6) & 0x3f];
587f62e5228SDag-Erling Smørgrav 		dst[3] = base64[(t >> 0) & 0x3f];
588f62e5228SDag-Erling Smørgrav 		src += 3; l -= 3;
589f62e5228SDag-Erling Smørgrav 		dst += 4; r += 4;
590f62e5228SDag-Erling Smørgrav 	}
591f62e5228SDag-Erling Smørgrav 
592f62e5228SDag-Erling Smørgrav 	switch (l) {
593f62e5228SDag-Erling Smørgrav 	case 2:
594f62e5228SDag-Erling Smørgrav 		t = (src[0] << 16) | (src[1] << 8);
595f62e5228SDag-Erling Smørgrav 		dst[0] = base64[(t >> 18) & 0x3f];
596f62e5228SDag-Erling Smørgrav 		dst[1] = base64[(t >> 12) & 0x3f];
597f62e5228SDag-Erling Smørgrav 		dst[2] = base64[(t >> 6) & 0x3f];
598f62e5228SDag-Erling Smørgrav 		dst[3] = '=';
599f62e5228SDag-Erling Smørgrav 		dst += 4;
600f62e5228SDag-Erling Smørgrav 		r += 4;
601f62e5228SDag-Erling Smørgrav 		break;
602f62e5228SDag-Erling Smørgrav 	case 1:
603f62e5228SDag-Erling Smørgrav 		t = src[0] << 16;
604f62e5228SDag-Erling Smørgrav 		dst[0] = base64[(t >> 18) & 0x3f];
605f62e5228SDag-Erling Smørgrav 		dst[1] = base64[(t >> 12) & 0x3f];
606f62e5228SDag-Erling Smørgrav 		dst[2] = dst[3] = '=';
607f62e5228SDag-Erling Smørgrav 		dst += 4;
608f62e5228SDag-Erling Smørgrav 		r += 4;
609f62e5228SDag-Erling Smørgrav 		break;
610f62e5228SDag-Erling Smørgrav 	case 0:
611f62e5228SDag-Erling Smørgrav 		break;
612f62e5228SDag-Erling Smørgrav 	}
613f62e5228SDag-Erling Smørgrav 
614f62e5228SDag-Erling Smørgrav 	*dst = 0;
615e19e6098SDag-Erling Smørgrav 	return (str);
616f62e5228SDag-Erling Smørgrav }
617f62e5228SDag-Erling Smørgrav 
618f62e5228SDag-Erling Smørgrav /*
619f62e5228SDag-Erling Smørgrav  * Encode username and password
620f62e5228SDag-Erling Smørgrav  */
62135f723dbSDag-Erling Smørgrav static int
622a1b37df2SDag-Erling Smørgrav http_basic_auth(conn_t *conn, const char *hdr, const char *usr, const char *pwd)
623f62e5228SDag-Erling Smørgrav {
62435f723dbSDag-Erling Smørgrav 	char *upw, *auth;
62535f723dbSDag-Erling Smørgrav 	int r;
626f62e5228SDag-Erling Smørgrav 
627f67efa37SDag-Erling Smørgrav 	DEBUG(fprintf(stderr, "usr: [%s]\n", usr));
628f67efa37SDag-Erling Smørgrav 	DEBUG(fprintf(stderr, "pwd: [%s]\n", pwd));
62935f723dbSDag-Erling Smørgrav 	if (asprintf(&upw, "%s:%s", usr, pwd) == -1)
630e19e6098SDag-Erling Smørgrav 		return (-1);
631a1b37df2SDag-Erling Smørgrav 	auth = http_base64(upw);
63235f723dbSDag-Erling Smørgrav 	free(upw);
63335f723dbSDag-Erling Smørgrav 	if (auth == NULL)
634e19e6098SDag-Erling Smørgrav 		return (-1);
635a1b37df2SDag-Erling Smørgrav 	r = http_cmd(conn, "%s: Basic %s", hdr, auth);
63635f723dbSDag-Erling Smørgrav 	free(auth);
637e19e6098SDag-Erling Smørgrav 	return (r);
6384d029f13SDag-Erling Smørgrav }
63935f723dbSDag-Erling Smørgrav 
64035f723dbSDag-Erling Smørgrav /*
64135f723dbSDag-Erling Smørgrav  * Send an authorization header
64235f723dbSDag-Erling Smørgrav  */
64335f723dbSDag-Erling Smørgrav static int
644a1b37df2SDag-Erling Smørgrav http_authorize(conn_t *conn, const char *hdr, const char *p)
64535f723dbSDag-Erling Smørgrav {
64635f723dbSDag-Erling Smørgrav 	/* basic authorization */
64735f723dbSDag-Erling Smørgrav 	if (strncasecmp(p, "basic:", 6) == 0) {
64835f723dbSDag-Erling Smørgrav 		char *user, *pwd, *str;
64935f723dbSDag-Erling Smørgrav 		int r;
65035f723dbSDag-Erling Smørgrav 
65135f723dbSDag-Erling Smørgrav 		/* skip realm */
65235f723dbSDag-Erling Smørgrav 		for (p += 6; *p && *p != ':'; ++p)
65335f723dbSDag-Erling Smørgrav 			/* nothing */ ;
65435f723dbSDag-Erling Smørgrav 		if (!*p || strchr(++p, ':') == NULL)
655e19e6098SDag-Erling Smørgrav 			return (-1);
65635f723dbSDag-Erling Smørgrav 		if ((str = strdup(p)) == NULL)
657e19e6098SDag-Erling Smørgrav 			return (-1); /* XXX */
65835f723dbSDag-Erling Smørgrav 		user = str;
65935f723dbSDag-Erling Smørgrav 		pwd = strchr(str, ':');
66035f723dbSDag-Erling Smørgrav 		*pwd++ = '\0';
661a1b37df2SDag-Erling Smørgrav 		r = http_basic_auth(conn, hdr, user, pwd);
66235f723dbSDag-Erling Smørgrav 		free(str);
663e19e6098SDag-Erling Smørgrav 		return (r);
6644d029f13SDag-Erling Smørgrav 	}
665e19e6098SDag-Erling Smørgrav 	return (-1);
666f62e5228SDag-Erling Smørgrav }
667f62e5228SDag-Erling Smørgrav 
668e19e6098SDag-Erling Smørgrav 
669e4878e39SDag-Erling Smørgrav /*****************************************************************************
670e4878e39SDag-Erling Smørgrav  * Helper functions for connecting to a server or proxy
671f62e5228SDag-Erling Smørgrav  */
672e4878e39SDag-Erling Smørgrav 
673e4878e39SDag-Erling Smørgrav /*
674e4878e39SDag-Erling Smørgrav  * Connect to the correct HTTP server or proxy.
67560245e42SDag-Erling Smørgrav  */
676dea29ca1SDag-Erling Smørgrav static conn_t *
677a1b37df2SDag-Erling Smørgrav http_connect(struct url *URL, struct url *purl, const char *flags)
67860245e42SDag-Erling Smørgrav {
679dea29ca1SDag-Erling Smørgrav 	conn_t *conn;
6801a16ed4cSDag-Erling Smørgrav 	int verbose;
68120a2a51bSKelly Yancey 	int af, val;
682e4878e39SDag-Erling Smørgrav 
683e4878e39SDag-Erling Smørgrav #ifdef INET6
684e4878e39SDag-Erling Smørgrav 	af = AF_UNSPEC;
685e4878e39SDag-Erling Smørgrav #else
686e4878e39SDag-Erling Smørgrav 	af = AF_INET;
687e4878e39SDag-Erling Smørgrav #endif
688e4878e39SDag-Erling Smørgrav 
689d74a913bSDag-Erling Smørgrav 	verbose = CHECK_FLAG('v');
690d74a913bSDag-Erling Smørgrav 	if (CHECK_FLAG('4'))
691e4878e39SDag-Erling Smørgrav 		af = AF_INET;
6921a16ed4cSDag-Erling Smørgrav #ifdef INET6
693d74a913bSDag-Erling Smørgrav 	else if (CHECK_FLAG('6'))
694e4878e39SDag-Erling Smørgrav 		af = AF_INET6;
6951a16ed4cSDag-Erling Smørgrav #endif
696e4878e39SDag-Erling Smørgrav 
697111e2510SDag-Erling Smørgrav 	if (purl && strcasecmp(URL->scheme, SCHEME_HTTPS) != 0) {
6981a16ed4cSDag-Erling Smørgrav 		URL = purl;
6991a16ed4cSDag-Erling Smørgrav 	} else if (strcasecmp(URL->scheme, SCHEME_FTP) == 0) {
700e4878e39SDag-Erling Smørgrav 		/* can't talk http to an ftp server */
701e4878e39SDag-Erling Smørgrav 		/* XXX should set an error code */
702dea29ca1SDag-Erling Smørgrav 		return (NULL);
703e4878e39SDag-Erling Smørgrav 	}
7041a16ed4cSDag-Erling Smørgrav 
705a1b37df2SDag-Erling Smørgrav 	if ((conn = fetch_connect(URL->host, URL->port, af, verbose)) == NULL)
706a1b37df2SDag-Erling Smørgrav 		/* fetch_connect() has already set an error code */
707dea29ca1SDag-Erling Smørgrav 		return (NULL);
708111e2510SDag-Erling Smørgrav 	if (strcasecmp(URL->scheme, SCHEME_HTTPS) == 0 &&
709a1b37df2SDag-Erling Smørgrav 	    fetch_ssl(conn, verbose) == -1) {
710a1b37df2SDag-Erling Smørgrav 		fetch_close(conn);
7113070f6cbSDag-Erling Smørgrav 		/* grrr */
7123070f6cbSDag-Erling Smørgrav 		errno = EAUTH;
713a1b37df2SDag-Erling Smørgrav 		fetch_syserr();
714111e2510SDag-Erling Smørgrav 		return (NULL);
715111e2510SDag-Erling Smørgrav 	}
71620a2a51bSKelly Yancey 
71720a2a51bSKelly Yancey 	val = 1;
71820a2a51bSKelly Yancey 	setsockopt(conn->sd, IPPROTO_TCP, TCP_NOPUSH, &val, sizeof(val));
71920a2a51bSKelly Yancey 
720dea29ca1SDag-Erling Smørgrav 	return (conn);
721e4878e39SDag-Erling Smørgrav }
722e4878e39SDag-Erling Smørgrav 
7231a16ed4cSDag-Erling Smørgrav static struct url *
72462a2681cSDag-Erling Smørgrav http_get_proxy(struct url * url, const char *flags)
7251a16ed4cSDag-Erling Smørgrav {
7261a16ed4cSDag-Erling Smørgrav 	struct url *purl;
7271a16ed4cSDag-Erling Smørgrav 	char *p;
7281a16ed4cSDag-Erling Smørgrav 
729762892c0SDag-Erling Smørgrav 	if (flags != NULL && strchr(flags, 'd') != NULL)
730b91d6074SDag-Erling Smørgrav 		return (NULL);
73162a2681cSDag-Erling Smørgrav 	if (fetch_no_proxy_match(url->host))
73262a2681cSDag-Erling Smørgrav 		return (NULL);
7334cee73c8SDag-Erling Smørgrav 	if (((p = getenv("HTTP_PROXY")) || (p = getenv("http_proxy"))) &&
7342738229eSDag-Erling Smørgrav 	    *p && (purl = fetchParseURL(p))) {
7351a16ed4cSDag-Erling Smørgrav 		if (!*purl->scheme)
7361a16ed4cSDag-Erling Smørgrav 			strcpy(purl->scheme, SCHEME_HTTP);
7371a16ed4cSDag-Erling Smørgrav 		if (!purl->port)
738a1b37df2SDag-Erling Smørgrav 			purl->port = fetch_default_proxy_port(purl->scheme);
7391a16ed4cSDag-Erling Smørgrav 		if (strcasecmp(purl->scheme, SCHEME_HTTP) == 0)
740e19e6098SDag-Erling Smørgrav 			return (purl);
7411a16ed4cSDag-Erling Smørgrav 		fetchFreeURL(purl);
7421a16ed4cSDag-Erling Smørgrav 	}
743e19e6098SDag-Erling Smørgrav 	return (NULL);
744e4878e39SDag-Erling Smørgrav }
745e4878e39SDag-Erling Smørgrav 
746a8e9bd87SDag-Erling Smørgrav static void
747a1b37df2SDag-Erling Smørgrav http_print_html(FILE *out, FILE *in)
748a8e9bd87SDag-Erling Smørgrav {
749a8e9bd87SDag-Erling Smørgrav 	size_t len;
750a8e9bd87SDag-Erling Smørgrav 	char *line, *p, *q;
751a8e9bd87SDag-Erling Smørgrav 	int comment, tag;
752a8e9bd87SDag-Erling Smørgrav 
753a8e9bd87SDag-Erling Smørgrav 	comment = tag = 0;
754a8e9bd87SDag-Erling Smørgrav 	while ((line = fgetln(in, &len)) != NULL) {
755facd9827SDag-Erling Smørgrav 		while (len && isspace((unsigned char)line[len - 1]))
756a8e9bd87SDag-Erling Smørgrav 			--len;
757a8e9bd87SDag-Erling Smørgrav 		for (p = q = line; q < line + len; ++q) {
758a8e9bd87SDag-Erling Smørgrav 			if (comment && *q == '-') {
759e19e6098SDag-Erling Smørgrav 				if (q + 2 < line + len &&
760e19e6098SDag-Erling Smørgrav 				    strcmp(q, "-->") == 0) {
761a8e9bd87SDag-Erling Smørgrav 					tag = comment = 0;
762a8e9bd87SDag-Erling Smørgrav 					q += 2;
763a8e9bd87SDag-Erling Smørgrav 				}
764e19e6098SDag-Erling Smørgrav 			} else if (tag && !comment && *q == '>') {
765a8e9bd87SDag-Erling Smørgrav 				p = q + 1;
766a8e9bd87SDag-Erling Smørgrav 				tag = 0;
767a8e9bd87SDag-Erling Smørgrav 			} else if (!tag && *q == '<') {
768a8e9bd87SDag-Erling Smørgrav 				if (q > p)
769a8e9bd87SDag-Erling Smørgrav 					fwrite(p, q - p, 1, out);
770a8e9bd87SDag-Erling Smørgrav 				tag = 1;
771e19e6098SDag-Erling Smørgrav 				if (q + 3 < line + len &&
772e19e6098SDag-Erling Smørgrav 				    strcmp(q, "<!--") == 0) {
773a8e9bd87SDag-Erling Smørgrav 					comment = 1;
774a8e9bd87SDag-Erling Smørgrav 					q += 3;
775a8e9bd87SDag-Erling Smørgrav 				}
776a8e9bd87SDag-Erling Smørgrav 			}
777a8e9bd87SDag-Erling Smørgrav 		}
778a8e9bd87SDag-Erling Smørgrav 		if (!tag && q > p)
779a8e9bd87SDag-Erling Smørgrav 			fwrite(p, q - p, 1, out);
780a8e9bd87SDag-Erling Smørgrav 		fputc('\n', out);
781a8e9bd87SDag-Erling Smørgrav 	}
782a8e9bd87SDag-Erling Smørgrav }
783a8e9bd87SDag-Erling Smørgrav 
784e19e6098SDag-Erling Smørgrav 
785e4878e39SDag-Erling Smørgrav /*****************************************************************************
786e4878e39SDag-Erling Smørgrav  * Core
787e4878e39SDag-Erling Smørgrav  */
788e4878e39SDag-Erling Smørgrav 
789e4878e39SDag-Erling Smørgrav /*
790e4878e39SDag-Erling Smørgrav  * Send a request and process the reply
7919601e333SDag-Erling Smørgrav  *
7929601e333SDag-Erling Smørgrav  * XXX This function is way too long, the do..while loop should be split
7939601e333SDag-Erling Smørgrav  * XXX off into a separate function.
794e4878e39SDag-Erling Smørgrav  */
7951a16ed4cSDag-Erling Smørgrav FILE *
796a1b37df2SDag-Erling Smørgrav http_request(struct url *URL, const char *op, struct url_stat *us,
79738c7e4a6SArchie Cobbs     struct url *purl, const char *flags)
798e4878e39SDag-Erling Smørgrav {
799dea29ca1SDag-Erling Smørgrav 	conn_t *conn;
800e4878e39SDag-Erling Smørgrav 	struct url *url, *new;
8011a16ed4cSDag-Erling Smørgrav 	int chunked, direct, need_auth, noredirect, verbose;
802448980e7SKelly Yancey 	int e, i, n, val;
803be6aff99SDag-Erling Smørgrav 	off_t offset, clength, length, size;
804c78f1cc9SDag-Erling Smørgrav 	time_t mtime;
80538c7e4a6SArchie Cobbs 	const char *p;
806e4878e39SDag-Erling Smørgrav 	FILE *f;
807f573a5fcSDag-Erling Smørgrav 	hdr_t h;
808d5216a4fSDag-Erling Smørgrav 	char hbuf[MAXHOSTNAMELEN + 7], *host;
80960245e42SDag-Erling Smørgrav 
810d74a913bSDag-Erling Smørgrav 	direct = CHECK_FLAG('d');
811d74a913bSDag-Erling Smørgrav 	noredirect = CHECK_FLAG('A');
812d74a913bSDag-Erling Smørgrav 	verbose = CHECK_FLAG('v');
8134ca1ab94SDag-Erling Smørgrav 
8141a16ed4cSDag-Erling Smørgrav 	if (direct && purl) {
8151a16ed4cSDag-Erling Smørgrav 		fetchFreeURL(purl);
8161a16ed4cSDag-Erling Smørgrav 		purl = NULL;
8171a16ed4cSDag-Erling Smørgrav 	}
8181a16ed4cSDag-Erling Smørgrav 
819c78f1cc9SDag-Erling Smørgrav 	/* try the provided URL first */
820c78f1cc9SDag-Erling Smørgrav 	url = URL;
821c78f1cc9SDag-Erling Smørgrav 
822c78f1cc9SDag-Erling Smørgrav 	/* if the A flag is set, we only get one try */
823e4878e39SDag-Erling Smørgrav 	n = noredirect ? 1 : MAX_REDIRECT;
824c78f1cc9SDag-Erling Smørgrav 	i = 0;
825e4878e39SDag-Erling Smørgrav 
826f8bd33a0SDag-Erling Smørgrav 	e = HTTP_PROTOCOL_ERROR;
8276490b215SDag-Erling Smørgrav 	need_auth = 0;
828c78f1cc9SDag-Erling Smørgrav 	do {
829f4683775SDag-Erling Smørgrav 		new = NULL;
830f4683775SDag-Erling Smørgrav 		chunked = 0;
831f4683775SDag-Erling Smørgrav 		offset = 0;
832c78f1cc9SDag-Erling Smørgrav 		clength = -1;
833c78f1cc9SDag-Erling Smørgrav 		length = -1;
834c78f1cc9SDag-Erling Smørgrav 		size = -1;
835c78f1cc9SDag-Erling Smørgrav 		mtime = 0;
8366490b215SDag-Erling Smørgrav 
8371a16ed4cSDag-Erling Smørgrav 		/* check port */
8381a16ed4cSDag-Erling Smørgrav 		if (!url->port)
839a1b37df2SDag-Erling Smørgrav 			url->port = fetch_default_port(url->scheme);
8401a16ed4cSDag-Erling Smørgrav 
84165986545SDag-Erling Smørgrav 		/* were we redirected to an FTP URL? */
84265986545SDag-Erling Smørgrav 		if (purl == NULL && strcmp(url->scheme, SCHEME_FTP) == 0) {
84365986545SDag-Erling Smørgrav 			if (strcmp(op, "GET") == 0)
844a1b37df2SDag-Erling Smørgrav 				return (ftp_request(url, "RETR", us, purl, flags));
84565986545SDag-Erling Smørgrav 			else if (strcmp(op, "HEAD") == 0)
846a1b37df2SDag-Erling Smørgrav 				return (ftp_request(url, "STAT", us, purl, flags));
84765986545SDag-Erling Smørgrav 		}
84865986545SDag-Erling Smørgrav 
849e4878e39SDag-Erling Smørgrav 		/* connect to server or proxy */
850a1b37df2SDag-Erling Smørgrav 		if ((conn = http_connect(url, purl, flags)) == NULL)
851e4878e39SDag-Erling Smørgrav 			goto ouch;
852e4878e39SDag-Erling Smørgrav 
853e4878e39SDag-Erling Smørgrav 		host = url->host;
85428c645cfSHajimu UMEMOTO #ifdef INET6
855e4878e39SDag-Erling Smørgrav 		if (strchr(url->host, ':')) {
856e4878e39SDag-Erling Smørgrav 			snprintf(hbuf, sizeof(hbuf), "[%s]", url->host);
85728c645cfSHajimu UMEMOTO 			host = hbuf;
85828c645cfSHajimu UMEMOTO 		}
85928c645cfSHajimu UMEMOTO #endif
860a1b37df2SDag-Erling Smørgrav 		if (url->port != fetch_default_port(url->scheme)) {
861d5216a4fSDag-Erling Smørgrav 			if (host != hbuf) {
862d5216a4fSDag-Erling Smørgrav 				strcpy(hbuf, host);
863d5216a4fSDag-Erling Smørgrav 				host = hbuf;
864d5216a4fSDag-Erling Smørgrav 			}
865d5216a4fSDag-Erling Smørgrav 			snprintf(hbuf + strlen(hbuf),
866d5216a4fSDag-Erling Smørgrav 			    sizeof(hbuf) - strlen(hbuf), ":%d", url->port);
867d5216a4fSDag-Erling Smørgrav 		}
86828c645cfSHajimu UMEMOTO 
869e4878e39SDag-Erling Smørgrav 		/* send request */
8700fba3a00SDag-Erling Smørgrav 		if (verbose)
871a1b37df2SDag-Erling Smørgrav 			fetch_info("requesting %s://%s%s",
872d5216a4fSDag-Erling Smørgrav 			    url->scheme, host, url->doc);
8731a16ed4cSDag-Erling Smørgrav 		if (purl) {
874a1b37df2SDag-Erling Smørgrav 			http_cmd(conn, "%s %s://%s%s HTTP/1.1",
875d5216a4fSDag-Erling Smørgrav 			    op, url->scheme, host, url->doc);
876e4878e39SDag-Erling Smørgrav 		} else {
877a1b37df2SDag-Erling Smørgrav 			http_cmd(conn, "%s %s HTTP/1.1",
878e4878e39SDag-Erling Smørgrav 			    op, url->doc);
87960245e42SDag-Erling Smørgrav 		}
88060245e42SDag-Erling Smørgrav 
8816490b215SDag-Erling Smørgrav 		/* virtual host */
882a1b37df2SDag-Erling Smørgrav 		http_cmd(conn, "Host: %s", host);
8836490b215SDag-Erling Smørgrav 
884e4878e39SDag-Erling Smørgrav 		/* proxy authorization */
8851a16ed4cSDag-Erling Smørgrav 		if (purl) {
8861a16ed4cSDag-Erling Smørgrav 			if (*purl->user || *purl->pwd)
887a1b37df2SDag-Erling Smørgrav 				http_basic_auth(conn, "Proxy-Authorization",
8881a16ed4cSDag-Erling Smørgrav 				    purl->user, purl->pwd);
8891a16ed4cSDag-Erling Smørgrav 			else if ((p = getenv("HTTP_PROXY_AUTH")) != NULL && *p != '\0')
890a1b37df2SDag-Erling Smørgrav 				http_authorize(conn, "Proxy-Authorization", p);
8911a16ed4cSDag-Erling Smørgrav 		}
892e4878e39SDag-Erling Smørgrav 
893e4878e39SDag-Erling Smørgrav 		/* server authorization */
8946490b215SDag-Erling Smørgrav 		if (need_auth || *url->user || *url->pwd) {
895e4878e39SDag-Erling Smørgrav 			if (*url->user || *url->pwd)
896a1b37df2SDag-Erling Smørgrav 				http_basic_auth(conn, "Authorization", url->user, url->pwd);
897c78f1cc9SDag-Erling Smørgrav 			else if ((p = getenv("HTTP_AUTH")) != NULL && *p != '\0')
898a1b37df2SDag-Erling Smørgrav 				http_authorize(conn, "Authorization", p);
8996490b215SDag-Erling Smørgrav 			else if (fetchAuthMethod && fetchAuthMethod(url) == 0) {
900a1b37df2SDag-Erling Smørgrav 				http_basic_auth(conn, "Authorization", url->user, url->pwd);
9016490b215SDag-Erling Smørgrav 			} else {
902a1b37df2SDag-Erling Smørgrav 				http_seterr(HTTP_NEED_AUTH);
903e4878e39SDag-Erling Smørgrav 				goto ouch;
904e4878e39SDag-Erling Smørgrav 			}
905e4878e39SDag-Erling Smørgrav 		}
906e4878e39SDag-Erling Smørgrav 
907e4878e39SDag-Erling Smørgrav 		/* other headers */
908d5216a4fSDag-Erling Smørgrav 		if ((p = getenv("HTTP_REFERER")) != NULL && *p != '\0') {
909d5216a4fSDag-Erling Smørgrav 			if (strcasecmp(p, "auto") == 0)
910a1b37df2SDag-Erling Smørgrav 				http_cmd(conn, "Referer: %s://%s%s",
911d5216a4fSDag-Erling Smørgrav 				    url->scheme, host, url->doc);
912d5216a4fSDag-Erling Smørgrav 			else
913a1b37df2SDag-Erling Smørgrav 				http_cmd(conn, "Referer: %s", p);
914d5216a4fSDag-Erling Smørgrav 		}
9156a0cf64bSDag-Erling Smørgrav 		if ((p = getenv("HTTP_USER_AGENT")) != NULL && *p != '\0')
916a1b37df2SDag-Erling Smørgrav 			http_cmd(conn, "User-Agent: %s", p);
9176a0cf64bSDag-Erling Smørgrav 		else
918a1b37df2SDag-Erling Smørgrav 			http_cmd(conn, "User-Agent: %s " _LIBFETCH_VER, getprogname());
919591c0199SDag-Erling Smørgrav 		if (url->offset > 0)
920a1b37df2SDag-Erling Smørgrav 			http_cmd(conn, "Range: bytes=%lld-", (long long)url->offset);
921a1b37df2SDag-Erling Smørgrav 		http_cmd(conn, "Connection: close");
922a1b37df2SDag-Erling Smørgrav 		http_cmd(conn, "");
923448980e7SKelly Yancey 
924448980e7SKelly Yancey 		/*
925448980e7SKelly Yancey 		 * Force the queued request to be dispatched.  Normally, one
926448980e7SKelly Yancey 		 * would do this with shutdown(2) but squid proxies can be
927448980e7SKelly Yancey 		 * configured to disallow such half-closed connections.  To
928448980e7SKelly Yancey 		 * be compatible with such configurations, fiddle with socket
929448980e7SKelly Yancey 		 * options to force the pending data to be written.
930448980e7SKelly Yancey 		 */
931448980e7SKelly Yancey 		val = 0;
932448980e7SKelly Yancey 		setsockopt(conn->sd, IPPROTO_TCP, TCP_NOPUSH, &val,
933448980e7SKelly Yancey 			   sizeof(val));
934448980e7SKelly Yancey 		val = 1;
935448980e7SKelly Yancey 		setsockopt(conn->sd, IPPROTO_TCP, TCP_NODELAY, &val,
936448980e7SKelly Yancey 			   sizeof(val));
937e4878e39SDag-Erling Smørgrav 
938e4878e39SDag-Erling Smørgrav 		/* get reply */
939a1b37df2SDag-Erling Smørgrav 		switch (http_get_reply(conn)) {
940e4878e39SDag-Erling Smørgrav 		case HTTP_OK:
941e4878e39SDag-Erling Smørgrav 		case HTTP_PARTIAL:
942e4878e39SDag-Erling Smørgrav 			/* fine */
943e4878e39SDag-Erling Smørgrav 			break;
944e4878e39SDag-Erling Smørgrav 		case HTTP_MOVED_PERM:
945e4878e39SDag-Erling Smørgrav 		case HTTP_MOVED_TEMP:
94665986545SDag-Erling Smørgrav 		case HTTP_SEE_OTHER:
947e4878e39SDag-Erling Smørgrav 			/*
9482576e459SDag-Erling Smørgrav 			 * Not so fine, but we still have to read the
9492576e459SDag-Erling Smørgrav 			 * headers to get the new location.
950e4878e39SDag-Erling Smørgrav 			 */
951e4878e39SDag-Erling Smørgrav 			break;
952e4878e39SDag-Erling Smørgrav 		case HTTP_NEED_AUTH:
953e4878e39SDag-Erling Smørgrav 			if (need_auth) {
954e4878e39SDag-Erling Smørgrav 				/*
9552576e459SDag-Erling Smørgrav 				 * We already sent out authorization code,
9562576e459SDag-Erling Smørgrav 				 * so there's nothing more we can do.
957e4878e39SDag-Erling Smørgrav 				 */
958a1b37df2SDag-Erling Smørgrav 				http_seterr(conn->err);
959e4878e39SDag-Erling Smørgrav 				goto ouch;
960e4878e39SDag-Erling Smørgrav 			}
961e4878e39SDag-Erling Smørgrav 			/* try again, but send the password this time */
962e4878e39SDag-Erling Smørgrav 			if (verbose)
963a1b37df2SDag-Erling Smørgrav 				fetch_info("server requires authorization");
9646490b215SDag-Erling Smørgrav 			break;
965e4878e39SDag-Erling Smørgrav 		case HTTP_NEED_PROXY_AUTH:
966e4878e39SDag-Erling Smørgrav 			/*
9672576e459SDag-Erling Smørgrav 			 * If we're talking to a proxy, we already sent
9682576e459SDag-Erling Smørgrav 			 * our proxy authorization code, so there's
9692576e459SDag-Erling Smørgrav 			 * nothing more we can do.
970e4878e39SDag-Erling Smørgrav 			 */
971a1b37df2SDag-Erling Smørgrav 			http_seterr(conn->err);
972e4878e39SDag-Erling Smørgrav 			goto ouch;
973b5e12145SDag-Erling Smørgrav 		case HTTP_BAD_RANGE:
974b5e12145SDag-Erling Smørgrav 			/*
975b5e12145SDag-Erling Smørgrav 			 * This can happen if we ask for 0 bytes because
976b5e12145SDag-Erling Smørgrav 			 * we already have the whole file.  Consider this
977b5e12145SDag-Erling Smørgrav 			 * a success for now, and check sizes later.
978b5e12145SDag-Erling Smørgrav 			 */
979b5e12145SDag-Erling Smørgrav 			break;
980e4878e39SDag-Erling Smørgrav 		case HTTP_PROTOCOL_ERROR:
981e4878e39SDag-Erling Smørgrav 			/* fall through */
982e4878e39SDag-Erling Smørgrav 		case -1:
983a1b37df2SDag-Erling Smørgrav 			fetch_syserr();
984e4878e39SDag-Erling Smørgrav 			goto ouch;
985e4878e39SDag-Erling Smørgrav 		default:
986a1b37df2SDag-Erling Smørgrav 			http_seterr(conn->err);
987a8e9bd87SDag-Erling Smørgrav 			if (!verbose)
988e4878e39SDag-Erling Smørgrav 				goto ouch;
989a8e9bd87SDag-Erling Smørgrav 			/* fall through so we can get the full error message */
990e4878e39SDag-Erling Smørgrav 		}
991e4878e39SDag-Erling Smørgrav 
992e4878e39SDag-Erling Smørgrav 		/* get headers */
993e4878e39SDag-Erling Smørgrav 		do {
994a1b37df2SDag-Erling Smørgrav 			switch ((h = http_next_header(conn, &p))) {
995e4878e39SDag-Erling Smørgrav 			case hdr_syserror:
996a1b37df2SDag-Erling Smørgrav 				fetch_syserr();
997e4878e39SDag-Erling Smørgrav 				goto ouch;
998e4878e39SDag-Erling Smørgrav 			case hdr_error:
999a1b37df2SDag-Erling Smørgrav 				http_seterr(HTTP_PROTOCOL_ERROR);
1000e4878e39SDag-Erling Smørgrav 				goto ouch;
1001e4878e39SDag-Erling Smørgrav 			case hdr_content_length:
1002a1b37df2SDag-Erling Smørgrav 				http_parse_length(p, &clength);
1003e4878e39SDag-Erling Smørgrav 				break;
1004e4878e39SDag-Erling Smørgrav 			case hdr_content_range:
1005a1b37df2SDag-Erling Smørgrav 				http_parse_range(p, &offset, &length, &size);
1006e4878e39SDag-Erling Smørgrav 				break;
1007e4878e39SDag-Erling Smørgrav 			case hdr_last_modified:
1008a1b37df2SDag-Erling Smørgrav 				http_parse_mtime(p, &mtime);
1009e4878e39SDag-Erling Smørgrav 				break;
1010e4878e39SDag-Erling Smørgrav 			case hdr_location:
1011dea29ca1SDag-Erling Smørgrav 				if (!HTTP_REDIRECT(conn->err))
1012e4878e39SDag-Erling Smørgrav 					break;
1013f4683775SDag-Erling Smørgrav 				if (new)
1014f4683775SDag-Erling Smørgrav 					free(new);
1015e4878e39SDag-Erling Smørgrav 				if (verbose)
1016a1b37df2SDag-Erling Smørgrav 					fetch_info("%d redirect to %s", conn->err, p);
1017f4683775SDag-Erling Smørgrav 				if (*p == '/')
1018f4683775SDag-Erling Smørgrav 					/* absolute path */
1019f4683775SDag-Erling Smørgrav 					new = fetchMakeURL(url->scheme, url->host, url->port, p,
1020f4683775SDag-Erling Smørgrav 					    url->user, url->pwd);
1021f4683775SDag-Erling Smørgrav 				else
1022f4683775SDag-Erling Smørgrav 					new = fetchParseURL(p);
1023f4683775SDag-Erling Smørgrav 				if (new == NULL) {
1024f4683775SDag-Erling Smørgrav 					/* XXX should set an error code */
1025f4683775SDag-Erling Smørgrav 					DEBUG(fprintf(stderr, "failed to parse new URL\n"));
1026e4878e39SDag-Erling Smørgrav 					goto ouch;
1027f4683775SDag-Erling Smørgrav 				}
1028e4878e39SDag-Erling Smørgrav 				if (!*new->user && !*new->pwd) {
1029e4878e39SDag-Erling Smørgrav 					strcpy(new->user, url->user);
1030e4878e39SDag-Erling Smørgrav 					strcpy(new->pwd, url->pwd);
1031e4878e39SDag-Erling Smørgrav 				}
1032e4878e39SDag-Erling Smørgrav 				new->offset = url->offset;
1033e4878e39SDag-Erling Smørgrav 				new->length = url->length;
1034f4683775SDag-Erling Smørgrav 				break;
1035e4878e39SDag-Erling Smørgrav 			case hdr_transfer_encoding:
1036e4878e39SDag-Erling Smørgrav 				/* XXX weak test*/
1037e4878e39SDag-Erling Smørgrav 				chunked = (strcasecmp(p, "chunked") == 0);
1038e4878e39SDag-Erling Smørgrav 				break;
10396490b215SDag-Erling Smørgrav 			case hdr_www_authenticate:
1040dea29ca1SDag-Erling Smørgrav 				if (conn->err != HTTP_NEED_AUTH)
10416490b215SDag-Erling Smørgrav 					break;
10426490b215SDag-Erling Smørgrav 				/* if we were smarter, we'd check the method and realm */
10436490b215SDag-Erling Smørgrav 				break;
1044e4878e39SDag-Erling Smørgrav 			case hdr_end:
1045e4878e39SDag-Erling Smørgrav 				/* fall through */
1046e4878e39SDag-Erling Smørgrav 			case hdr_unknown:
1047e4878e39SDag-Erling Smørgrav 				/* ignore */
1048e4878e39SDag-Erling Smørgrav 				break;
1049e4878e39SDag-Erling Smørgrav 			}
1050e4878e39SDag-Erling Smørgrav 		} while (h > hdr_end);
1051e4878e39SDag-Erling Smørgrav 
10526490b215SDag-Erling Smørgrav 		/* we need to provide authentication */
1053dea29ca1SDag-Erling Smørgrav 		if (conn->err == HTTP_NEED_AUTH) {
1054f8bd33a0SDag-Erling Smørgrav 			e = conn->err;
10556490b215SDag-Erling Smørgrav 			need_auth = 1;
1056a1b37df2SDag-Erling Smørgrav 			fetch_close(conn);
1057dea29ca1SDag-Erling Smørgrav 			conn = NULL;
10586490b215SDag-Erling Smørgrav 			continue;
10596490b215SDag-Erling Smørgrav 		}
10606490b215SDag-Erling Smørgrav 
1061b5e12145SDag-Erling Smørgrav 		/* requested range not satisfiable */
1062b5e12145SDag-Erling Smørgrav 		if (conn->err == HTTP_BAD_RANGE) {
1063b5e12145SDag-Erling Smørgrav 			if (url->offset == size && url->length == 0) {
1064b5e12145SDag-Erling Smørgrav 				/* asked for 0 bytes; fake it */
1065b5e12145SDag-Erling Smørgrav 				offset = url->offset;
1066b5e12145SDag-Erling Smørgrav 				conn->err = HTTP_OK;
1067b5e12145SDag-Erling Smørgrav 				break;
1068b5e12145SDag-Erling Smørgrav 			} else {
1069a1b37df2SDag-Erling Smørgrav 				http_seterr(conn->err);
1070b5e12145SDag-Erling Smørgrav 				goto ouch;
1071b5e12145SDag-Erling Smørgrav 			}
1072b5e12145SDag-Erling Smørgrav 		}
1073b5e12145SDag-Erling Smørgrav 
1074721d559cSRuslan Ermilov 		/* we have a hit or an error */
1075721d559cSRuslan Ermilov 		if (conn->err == HTTP_OK || conn->err == HTTP_PARTIAL || HTTP_ERROR(conn->err))
1076721d559cSRuslan Ermilov 			break;
1077721d559cSRuslan Ermilov 
10786490b215SDag-Erling Smørgrav 		/* all other cases: we got a redirect */
1079f8bd33a0SDag-Erling Smørgrav 		e = conn->err;
10806490b215SDag-Erling Smørgrav 		need_auth = 0;
1081a1b37df2SDag-Erling Smørgrav 		fetch_close(conn);
1082dea29ca1SDag-Erling Smørgrav 		conn = NULL;
10836490b215SDag-Erling Smørgrav 		if (!new) {
10846490b215SDag-Erling Smørgrav 			DEBUG(fprintf(stderr, "redirect with no new location\n"));
10856490b215SDag-Erling Smørgrav 			break;
10866490b215SDag-Erling Smørgrav 		}
1087f4683775SDag-Erling Smørgrav 		if (url != URL)
1088f4683775SDag-Erling Smørgrav 			fetchFreeURL(url);
1089f4683775SDag-Erling Smørgrav 		url = new;
1090c78f1cc9SDag-Erling Smørgrav 	} while (++i < n);
1091e4878e39SDag-Erling Smørgrav 
10926490b215SDag-Erling Smørgrav 	/* we failed, or ran out of retries */
1093dea29ca1SDag-Erling Smørgrav 	if (conn == NULL) {
1094a1b37df2SDag-Erling Smørgrav 		http_seterr(e);
1095e4878e39SDag-Erling Smørgrav 		goto ouch;
1096e4878e39SDag-Erling Smørgrav 	}
1097e4878e39SDag-Erling Smørgrav 
1098f573a5fcSDag-Erling Smørgrav 	DEBUG(fprintf(stderr, "offset %lld, length %lld,"
1099f573a5fcSDag-Erling Smørgrav 		  " size %lld, clength %lld\n",
1100f573a5fcSDag-Erling Smørgrav 		  (long long)offset, (long long)length,
1101f573a5fcSDag-Erling Smørgrav 		  (long long)size, (long long)clength));
1102c78f1cc9SDag-Erling Smørgrav 
1103c78f1cc9SDag-Erling Smørgrav 	/* check for inconsistencies */
1104c78f1cc9SDag-Erling Smørgrav 	if (clength != -1 && length != -1 && clength != length) {
1105a1b37df2SDag-Erling Smørgrav 		http_seterr(HTTP_PROTOCOL_ERROR);
1106c78f1cc9SDag-Erling Smørgrav 		goto ouch;
1107c78f1cc9SDag-Erling Smørgrav 	}
1108c78f1cc9SDag-Erling Smørgrav 	if (clength == -1)
1109c78f1cc9SDag-Erling Smørgrav 		clength = length;
1110c78f1cc9SDag-Erling Smørgrav 	if (clength != -1)
1111c78f1cc9SDag-Erling Smørgrav 		length = offset + clength;
1112c78f1cc9SDag-Erling Smørgrav 	if (length != -1 && size != -1 && length != size) {
1113a1b37df2SDag-Erling Smørgrav 		http_seterr(HTTP_PROTOCOL_ERROR);
1114c78f1cc9SDag-Erling Smørgrav 		goto ouch;
1115c78f1cc9SDag-Erling Smørgrav 	}
1116c78f1cc9SDag-Erling Smørgrav 	if (size == -1)
1117c78f1cc9SDag-Erling Smørgrav 		size = length;
1118c78f1cc9SDag-Erling Smørgrav 
1119c78f1cc9SDag-Erling Smørgrav 	/* fill in stats */
1120c78f1cc9SDag-Erling Smørgrav 	if (us) {
1121c78f1cc9SDag-Erling Smørgrav 		us->size = size;
1122c78f1cc9SDag-Erling Smørgrav 		us->atime = us->mtime = mtime;
1123c78f1cc9SDag-Erling Smørgrav 	}
1124c78f1cc9SDag-Erling Smørgrav 
112587b41116SDag-Erling Smørgrav 	/* too far? */
1126591c0199SDag-Erling Smørgrav 	if (URL->offset > 0 && offset > URL->offset) {
1127a1b37df2SDag-Erling Smørgrav 		http_seterr(HTTP_PROTOCOL_ERROR);
112887b41116SDag-Erling Smørgrav 		goto ouch;
112987b41116SDag-Erling Smørgrav 	}
113087b41116SDag-Erling Smørgrav 
1131c78f1cc9SDag-Erling Smørgrav 	/* report back real offset and size */
113287b41116SDag-Erling Smørgrav 	URL->offset = offset;
1133c78f1cc9SDag-Erling Smørgrav 	URL->length = clength;
113487b41116SDag-Erling Smørgrav 
1135e4878e39SDag-Erling Smørgrav 	/* wrap it up in a FILE */
1136a1b37df2SDag-Erling Smørgrav 	if ((f = http_funopen(conn, chunked)) == NULL) {
1137a1b37df2SDag-Erling Smørgrav 		fetch_syserr();
1138e4878e39SDag-Erling Smørgrav 		goto ouch;
1139e4878e39SDag-Erling Smørgrav 	}
1140e4878e39SDag-Erling Smørgrav 
1141e4878e39SDag-Erling Smørgrav 	if (url != URL)
1142e4878e39SDag-Erling Smørgrav 		fetchFreeURL(url);
11431a16ed4cSDag-Erling Smørgrav 	if (purl)
11441a16ed4cSDag-Erling Smørgrav 		fetchFreeURL(purl);
1145e4878e39SDag-Erling Smørgrav 
1146dea29ca1SDag-Erling Smørgrav 	if (HTTP_ERROR(conn->err)) {
1147a1b37df2SDag-Erling Smørgrav 		http_print_html(stderr, f);
1148a8e9bd87SDag-Erling Smørgrav 		fclose(f);
1149a8e9bd87SDag-Erling Smørgrav 		f = NULL;
1150a8e9bd87SDag-Erling Smørgrav 	}
1151a8e9bd87SDag-Erling Smørgrav 
1152e19e6098SDag-Erling Smørgrav 	return (f);
1153e4878e39SDag-Erling Smørgrav 
1154e4878e39SDag-Erling Smørgrav ouch:
1155e4878e39SDag-Erling Smørgrav 	if (url != URL)
1156e4878e39SDag-Erling Smørgrav 		fetchFreeURL(url);
11571a16ed4cSDag-Erling Smørgrav 	if (purl)
11581a16ed4cSDag-Erling Smørgrav 		fetchFreeURL(purl);
1159dea29ca1SDag-Erling Smørgrav 	if (conn != NULL)
1160a1b37df2SDag-Erling Smørgrav 		fetch_close(conn);
1161e19e6098SDag-Erling Smørgrav 	return (NULL);
1162e4878e39SDag-Erling Smørgrav }
1163e4878e39SDag-Erling Smørgrav 
1164e19e6098SDag-Erling Smørgrav 
1165e4878e39SDag-Erling Smørgrav /*****************************************************************************
1166e4878e39SDag-Erling Smørgrav  * Entry points
1167e4878e39SDag-Erling Smørgrav  */
1168e4878e39SDag-Erling Smørgrav 
116960245e42SDag-Erling Smørgrav /*
11701a5faa10SDag-Erling Smørgrav  * Retrieve and stat a file by HTTP
11711a5faa10SDag-Erling Smørgrav  */
11721a5faa10SDag-Erling Smørgrav FILE *
117338c7e4a6SArchie Cobbs fetchXGetHTTP(struct url *URL, struct url_stat *us, const char *flags)
11741a5faa10SDag-Erling Smørgrav {
117562a2681cSDag-Erling Smørgrav 	return (http_request(URL, "GET", us, http_get_proxy(URL, flags), flags));
11761a5faa10SDag-Erling Smørgrav }
11771a5faa10SDag-Erling Smørgrav 
11781a5faa10SDag-Erling Smørgrav /*
117960245e42SDag-Erling Smørgrav  * Retrieve a file by HTTP
118060245e42SDag-Erling Smørgrav  */
118160245e42SDag-Erling Smørgrav FILE *
118238c7e4a6SArchie Cobbs fetchGetHTTP(struct url *URL, const char *flags)
118360245e42SDag-Erling Smørgrav {
1184e19e6098SDag-Erling Smørgrav 	return (fetchXGetHTTP(URL, NULL, flags));
11854ca1ab94SDag-Erling Smørgrav }
11864ca1ab94SDag-Erling Smørgrav 
11871a5faa10SDag-Erling Smørgrav /*
11881a5faa10SDag-Erling Smørgrav  * Store a file by HTTP
11891a5faa10SDag-Erling Smørgrav  */
11904ca1ab94SDag-Erling Smørgrav FILE *
1191f573a5fcSDag-Erling Smørgrav fetchPutHTTP(struct url *URL __unused, const char *flags __unused)
11924ca1ab94SDag-Erling Smørgrav {
11934ca1ab94SDag-Erling Smørgrav 	warnx("fetchPutHTTP(): not implemented");
1194e19e6098SDag-Erling Smørgrav 	return (NULL);
11954ca1ab94SDag-Erling Smørgrav }
1196d8acd8dcSDag-Erling Smørgrav 
1197d8acd8dcSDag-Erling Smørgrav /*
1198d8acd8dcSDag-Erling Smørgrav  * Get an HTTP document's metadata
1199d8acd8dcSDag-Erling Smørgrav  */
1200d8acd8dcSDag-Erling Smørgrav int
120138c7e4a6SArchie Cobbs fetchStatHTTP(struct url *URL, struct url_stat *us, const char *flags)
1202d8acd8dcSDag-Erling Smørgrav {
120360245e42SDag-Erling Smørgrav 	FILE *f;
120460245e42SDag-Erling Smørgrav 
120562a2681cSDag-Erling Smørgrav 	f = http_request(URL, "HEAD", us, http_get_proxy(URL, flags), flags);
1206b91d6074SDag-Erling Smørgrav 	if (f == NULL)
1207e19e6098SDag-Erling Smørgrav 		return (-1);
1208f8f4130bSDag-Erling Smørgrav 	fclose(f);
1209e19e6098SDag-Erling Smørgrav 	return (0);
1210d8acd8dcSDag-Erling Smørgrav }
1211ce71b736SDag-Erling Smørgrav 
1212ce71b736SDag-Erling Smørgrav /*
1213ce71b736SDag-Erling Smørgrav  * List a directory
1214ce71b736SDag-Erling Smørgrav  */
1215ce71b736SDag-Erling Smørgrav struct url_ent *
1216f573a5fcSDag-Erling Smørgrav fetchListHTTP(struct url *url __unused, const char *flags __unused)
1217ce71b736SDag-Erling Smørgrav {
1218ce71b736SDag-Erling Smørgrav 	warnx("fetchListHTTP(): not implemented");
1219e19e6098SDag-Erling Smørgrav 	return (NULL);
1220ce71b736SDag-Erling Smørgrav }
1221