http.c revision 270460
1340084Svangyzen/*- 2340084Svangyzen * Copyright (c) 2000-2014 Dag-Erling Sm��rgrav 3340084Svangyzen * All rights reserved. 4340084Svangyzen * 5340084Svangyzen * Redistribution and use in source and binary forms, with or without 6340084Svangyzen * modification, are permitted provided that the following conditions 7340084Svangyzen * are met: 8340084Svangyzen * 1. Redistributions of source code must retain the above copyright 9340084Svangyzen * notice, this list of conditions and the following disclaimer 10340084Svangyzen * in this position and unchanged. 11340084Svangyzen * 2. Redistributions in binary form must reproduce the above copyright 12340084Svangyzen * notice, this list of conditions and the following disclaimer in the 13340084Svangyzen * documentation and/or other materials provided with the distribution. 14340084Svangyzen * 3. The name of the author may not be used to endorse or promote products 15340084Svangyzen * derived from this software without specific prior written permission. 16340084Svangyzen * 17340084Svangyzen * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 18340084Svangyzen * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 19340084Svangyzen * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 20340084Svangyzen * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 21340084Svangyzen * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 22340084Svangyzen * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 23340084Svangyzen * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 24340084Svangyzen * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25340084Svangyzen * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 26340084Svangyzen * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27340084Svangyzen */ 28340084Svangyzen 29340084Svangyzen#include <sys/cdefs.h> 30340084Svangyzen__FBSDID("$FreeBSD: stable/10/lib/libfetch/http.c 270460 2014-08-24 14:04:20Z des $"); 31340084Svangyzen 32340084Svangyzen/* 33340084Svangyzen * The following copyright applies to the base64 code: 34340084Svangyzen * 35340084Svangyzen *- 36340084Svangyzen * Copyright 1997 Massachusetts Institute of Technology 37340084Svangyzen * 38340084Svangyzen * Permission to use, copy, modify, and distribute this software and 39340084Svangyzen * its documentation for any purpose and without fee is hereby 40340084Svangyzen * granted, provided that both the above copyright notice and this 41340084Svangyzen * permission notice appear in all copies, that both the above 42340084Svangyzen * copyright notice and this permission notice appear in all 43340084Svangyzen * supporting documentation, and that the name of M.I.T. not be used 44340084Svangyzen * in advertising or publicity pertaining to distribution of the 45340084Svangyzen * software without specific, written prior permission. M.I.T. makes 46340084Svangyzen * no representations about the suitability of this software for any 47340084Svangyzen * purpose. It is provided "as is" without express or implied 48340084Svangyzen * warranty. 49340084Svangyzen * 50340084Svangyzen * THIS SOFTWARE IS PROVIDED BY M.I.T. ``AS IS''. M.I.T. DISCLAIMS 51340084Svangyzen * ALL EXPRESS OR IMPLIED WARRANTIES WITH REGARD TO THIS SOFTWARE, 52340084Svangyzen * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 53340084Svangyzen * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. IN NO EVENT 54340084Svangyzen * SHALL M.I.T. BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 55340084Svangyzen * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 56340084Svangyzen * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF 57340084Svangyzen * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 58340084Svangyzen * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 59340084Svangyzen * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT 60340084Svangyzen * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 61340084Svangyzen * SUCH DAMAGE. 62340084Svangyzen */ 63340084Svangyzen 64340084Svangyzen#include <sys/param.h> 65340084Svangyzen#include <sys/socket.h> 66340084Svangyzen#include <sys/time.h> 67340084Svangyzen 68340084Svangyzen#include <ctype.h> 69340084Svangyzen#include <err.h> 70340084Svangyzen#include <errno.h> 71340084Svangyzen#include <locale.h> 72340084Svangyzen#include <netdb.h> 73340084Svangyzen#include <stdarg.h> 74340084Svangyzen#include <stdio.h> 75340084Svangyzen#include <stdlib.h> 76340084Svangyzen#include <string.h> 77340084Svangyzen#include <time.h> 78340084Svangyzen#include <unistd.h> 79340084Svangyzen 80340084Svangyzen#ifdef WITH_SSL 81340084Svangyzen#include <openssl/md5.h> 82340084Svangyzen#define MD5Init(c) MD5_Init(c) 83340084Svangyzen#define MD5Update(c, data, len) MD5_Update(c, data, len) 84340084Svangyzen#define MD5Final(md, c) MD5_Final(md, c) 85340084Svangyzen#else 86340084Svangyzen#include <md5.h> 87340084Svangyzen#endif 88340084Svangyzen 89340084Svangyzen#include <netinet/in.h> 90340084Svangyzen#include <netinet/tcp.h> 91340084Svangyzen 92340084Svangyzen#include "fetch.h" 93340084Svangyzen#include "common.h" 94340084Svangyzen#include "httperr.h" 95340084Svangyzen 96340084Svangyzen/* Maximum number of redirects to follow */ 97340084Svangyzen#define MAX_REDIRECT 20 98340084Svangyzen 99340084Svangyzen/* Symbolic names for reply codes we care about */ 100340084Svangyzen#define HTTP_OK 200 101340084Svangyzen#define HTTP_PARTIAL 206 102340084Svangyzen#define HTTP_MOVED_PERM 301 103340084Svangyzen#define HTTP_MOVED_TEMP 302 104340084Svangyzen#define HTTP_SEE_OTHER 303 105340084Svangyzen#define HTTP_NOT_MODIFIED 304 106340084Svangyzen#define HTTP_USE_PROXY 305 107340084Svangyzen#define HTTP_TEMP_REDIRECT 307 108340084Svangyzen#define HTTP_PERM_REDIRECT 308 109340084Svangyzen#define HTTP_NEED_AUTH 401 110340084Svangyzen#define HTTP_NEED_PROXY_AUTH 407 111340084Svangyzen#define HTTP_BAD_RANGE 416 112340084Svangyzen#define HTTP_PROTOCOL_ERROR 999 113340084Svangyzen 114340084Svangyzen#define HTTP_REDIRECT(xyz) ((xyz) == HTTP_MOVED_PERM \ 115340084Svangyzen || (xyz) == HTTP_MOVED_TEMP \ 116340084Svangyzen || (xyz) == HTTP_TEMP_REDIRECT \ 117340084Svangyzen || (xyz) == HTTP_USE_PROXY \ 118340084Svangyzen || (xyz) == HTTP_SEE_OTHER) 119340084Svangyzen 120340084Svangyzen#define HTTP_ERROR(xyz) ((xyz) > 400 && (xyz) < 599) 121340084Svangyzen 122340084Svangyzen 123340084Svangyzen/***************************************************************************** 124340084Svangyzen * I/O functions for decoding chunked streams 125340084Svangyzen */ 126340084Svangyzen 127340084Svangyzenstruct httpio 128340084Svangyzen{ 129340084Svangyzen conn_t *conn; /* connection */ 130340084Svangyzen int chunked; /* chunked mode */ 131340084Svangyzen char *buf; /* chunk buffer */ 132340084Svangyzen size_t bufsize; /* size of chunk buffer */ 133340084Svangyzen ssize_t buflen; /* amount of data currently in buffer */ 134340084Svangyzen int bufpos; /* current read offset in buffer */ 135340084Svangyzen int eof; /* end-of-file flag */ 136340084Svangyzen int error; /* error flag */ 137340084Svangyzen size_t chunksize; /* remaining size of current chunk */ 138340084Svangyzen#ifndef NDEBUG 139340084Svangyzen size_t total; 140340084Svangyzen#endif 141340084Svangyzen}; 142340084Svangyzen 143340084Svangyzen/* 144340084Svangyzen * Get next chunk header 145340084Svangyzen */ 146340084Svangyzenstatic int 147340084Svangyzenhttp_new_chunk(struct httpio *io) 148340084Svangyzen{ 149340084Svangyzen char *p; 150340084Svangyzen 151340084Svangyzen if (fetch_getln(io->conn) == -1) 152340084Svangyzen return (-1); 153340084Svangyzen 154340084Svangyzen if (io->conn->buflen < 2 || !isxdigit((unsigned char)*io->conn->buf)) 155340084Svangyzen return (-1); 156340084Svangyzen 157340084Svangyzen for (p = io->conn->buf; *p && !isspace((unsigned char)*p); ++p) { 158340084Svangyzen if (*p == ';') 159340084Svangyzen break; 160340084Svangyzen if (!isxdigit((unsigned char)*p)) 161340084Svangyzen return (-1); 162340084Svangyzen if (isdigit((unsigned char)*p)) { 163340084Svangyzen io->chunksize = io->chunksize * 16 + 164340084Svangyzen *p - '0'; 165340084Svangyzen } else { 166340084Svangyzen io->chunksize = io->chunksize * 16 + 167340084Svangyzen 10 + tolower((unsigned char)*p) - 'a'; 168340084Svangyzen } 169340084Svangyzen } 170340084Svangyzen 171340084Svangyzen#ifndef NDEBUG 172340084Svangyzen if (fetchDebug) { 173340084Svangyzen io->total += io->chunksize; 174340084Svangyzen if (io->chunksize == 0) 175340084Svangyzen fprintf(stderr, "%s(): end of last chunk\n", __func__); 176340084Svangyzen else 177340084Svangyzen fprintf(stderr, "%s(): new chunk: %lu (%lu)\n", 178340084Svangyzen __func__, (unsigned long)io->chunksize, 179340084Svangyzen (unsigned long)io->total); 180340084Svangyzen } 181340084Svangyzen#endif 182340084Svangyzen 183340084Svangyzen return (io->chunksize); 184340084Svangyzen} 185340084Svangyzen 186340084Svangyzen/* 187340084Svangyzen * Grow the input buffer to at least len bytes 188340084Svangyzen */ 189340084Svangyzenstatic inline int 190340084Svangyzenhttp_growbuf(struct httpio *io, size_t len) 191340084Svangyzen{ 192340084Svangyzen char *tmp; 193340084Svangyzen 194340084Svangyzen if (io->bufsize >= len) 195340084Svangyzen return (0); 196340084Svangyzen 197340084Svangyzen if ((tmp = realloc(io->buf, len)) == NULL) 198340084Svangyzen return (-1); 199340084Svangyzen io->buf = tmp; 200340084Svangyzen io->bufsize = len; 201340084Svangyzen return (0); 202340084Svangyzen} 203340084Svangyzen 204340084Svangyzen/* 205340084Svangyzen * Fill the input buffer, do chunk decoding on the fly 206340084Svangyzen */ 207340084Svangyzenstatic ssize_t 208340084Svangyzenhttp_fillbuf(struct httpio *io, size_t len) 209340084Svangyzen{ 210340084Svangyzen ssize_t nbytes; 211340084Svangyzen char ch; 212340084Svangyzen 213340084Svangyzen if (io->error) 214340084Svangyzen return (-1); 215340084Svangyzen if (io->eof) 216340084Svangyzen return (0); 217340084Svangyzen 218340084Svangyzen if (io->chunked == 0) { 219340084Svangyzen if (http_growbuf(io, len) == -1) 220340084Svangyzen return (-1); 221340084Svangyzen if ((nbytes = fetch_read(io->conn, io->buf, len)) == -1) { 222340084Svangyzen io->error = errno; 223340084Svangyzen return (-1); 224340084Svangyzen } 225340084Svangyzen io->buflen = nbytes; 226340084Svangyzen io->bufpos = 0; 227340084Svangyzen return (io->buflen); 228340084Svangyzen } 229340084Svangyzen 230340084Svangyzen if (io->chunksize == 0) { 231340084Svangyzen switch (http_new_chunk(io)) { 232340084Svangyzen case -1: 233340084Svangyzen io->error = EPROTO; 234340084Svangyzen return (-1); 235340084Svangyzen case 0: 236340084Svangyzen io->eof = 1; 237340084Svangyzen return (0); 238340084Svangyzen } 239340084Svangyzen } 240340084Svangyzen 241340084Svangyzen if (len > io->chunksize) 242340084Svangyzen len = io->chunksize; 243340084Svangyzen if (http_growbuf(io, len) == -1) 244340084Svangyzen return (-1); 245340084Svangyzen if ((nbytes = fetch_read(io->conn, io->buf, len)) == -1) { 246340084Svangyzen io->error = errno; 247340084Svangyzen return (-1); 248340084Svangyzen } 249340084Svangyzen io->buflen = nbytes; 250340084Svangyzen io->chunksize -= io->buflen; 251340084Svangyzen 252340084Svangyzen if (io->chunksize == 0) { 253340084Svangyzen if (fetch_read(io->conn, &ch, 1) != 1 || ch != '\r' || 254340084Svangyzen fetch_read(io->conn, &ch, 1) != 1 || ch != '\n') 255340084Svangyzen return (-1); 256340084Svangyzen } 257340084Svangyzen 258340084Svangyzen io->bufpos = 0; 259340084Svangyzen 260340084Svangyzen return (io->buflen); 261340084Svangyzen} 262340084Svangyzen 263340084Svangyzen/* 264340084Svangyzen * Read function 265340084Svangyzen */ 266340084Svangyzenstatic int 267340084Svangyzenhttp_readfn(void *v, char *buf, int len) 268340084Svangyzen{ 269340084Svangyzen struct httpio *io = (struct httpio *)v; 270340084Svangyzen int rlen; 271340084Svangyzen 272340084Svangyzen if (io->error) 273340084Svangyzen return (-1); 274340084Svangyzen if (io->eof) 275340084Svangyzen return (0); 276340084Svangyzen 277340084Svangyzen /* empty buffer */ 278340084Svangyzen if (!io->buf || io->bufpos == io->buflen) { 279340084Svangyzen if ((rlen = http_fillbuf(io, len)) < 0) { 280340084Svangyzen if ((errno = io->error) == EINTR) 281340084Svangyzen io->error = 0; 282340084Svangyzen return (-1); 283340084Svangyzen } else if (rlen == 0) { 284340084Svangyzen return (0); 285340084Svangyzen } 286340084Svangyzen } 287340084Svangyzen 288340084Svangyzen rlen = io->buflen - io->bufpos; 289340084Svangyzen if (len < rlen) 290340084Svangyzen rlen = len; 291340084Svangyzen memcpy(buf, io->buf + io->bufpos, rlen); 292340084Svangyzen io->bufpos += rlen; 293340084Svangyzen return (rlen); 294340084Svangyzen} 295340084Svangyzen 296340084Svangyzen/* 297340084Svangyzen * Write function 298340084Svangyzen */ 299340084Svangyzenstatic int 300340084Svangyzenhttp_writefn(void *v, const char *buf, int len) 301340084Svangyzen{ 302340084Svangyzen struct httpio *io = (struct httpio *)v; 303340084Svangyzen 304340084Svangyzen return (fetch_write(io->conn, buf, len)); 305340084Svangyzen} 306340084Svangyzen 307340084Svangyzen/* 308340084Svangyzen * Close function 309340084Svangyzen */ 310340084Svangyzenstatic int 311340084Svangyzenhttp_closefn(void *v) 312340084Svangyzen{ 313340084Svangyzen struct httpio *io = (struct httpio *)v; 314340084Svangyzen int r; 315340084Svangyzen 316340084Svangyzen r = fetch_close(io->conn); 317340084Svangyzen if (io->buf) 318340084Svangyzen free(io->buf); 319340084Svangyzen free(io); 320340084Svangyzen return (r); 321340084Svangyzen} 322340084Svangyzen 323340084Svangyzen/* 324340084Svangyzen * Wrap a file descriptor up 325340084Svangyzen */ 326340084Svangyzenstatic FILE * 327340084Svangyzenhttp_funopen(conn_t *conn, int chunked) 328340084Svangyzen{ 329340084Svangyzen struct httpio *io; 330340084Svangyzen FILE *f; 331340084Svangyzen 332340084Svangyzen if ((io = calloc(1, sizeof(*io))) == NULL) { 333340084Svangyzen fetch_syserr(); 334340084Svangyzen return (NULL); 335340084Svangyzen } 336340084Svangyzen io->conn = conn; 337340084Svangyzen io->chunked = chunked; 338340084Svangyzen f = funopen(io, http_readfn, http_writefn, NULL, http_closefn); 339340084Svangyzen if (f == NULL) { 340340084Svangyzen fetch_syserr(); 341340084Svangyzen free(io); 342340084Svangyzen return (NULL); 343340084Svangyzen } 344340084Svangyzen return (f); 345340084Svangyzen} 346340084Svangyzen 347340084Svangyzen 348340084Svangyzen/***************************************************************************** 349340084Svangyzen * Helper functions for talking to the server and parsing its replies 350340084Svangyzen */ 351340084Svangyzen 352340084Svangyzen/* Header types */ 353340084Svangyzentypedef enum { 354340084Svangyzen hdr_syserror = -2, 355340084Svangyzen hdr_error = -1, 356340084Svangyzen hdr_end = 0, 357340084Svangyzen hdr_unknown = 1, 358340084Svangyzen hdr_content_length, 359340084Svangyzen hdr_content_range, 360340084Svangyzen hdr_last_modified, 361340084Svangyzen hdr_location, 362340084Svangyzen hdr_transfer_encoding, 363340084Svangyzen hdr_www_authenticate, 364340084Svangyzen hdr_proxy_authenticate, 365340084Svangyzen} hdr_t; 366340084Svangyzen 367340084Svangyzen/* Names of interesting headers */ 368340084Svangyzenstatic struct { 369340084Svangyzen hdr_t num; 370340084Svangyzen const char *name; 371340084Svangyzen} hdr_names[] = { 372340084Svangyzen { hdr_content_length, "Content-Length" }, 373340084Svangyzen { hdr_content_range, "Content-Range" }, 374340084Svangyzen { hdr_last_modified, "Last-Modified" }, 375340084Svangyzen { hdr_location, "Location" }, 376340084Svangyzen { hdr_transfer_encoding, "Transfer-Encoding" }, 377340084Svangyzen { hdr_www_authenticate, "WWW-Authenticate" }, 378340084Svangyzen { hdr_proxy_authenticate, "Proxy-Authenticate" }, 379340084Svangyzen { hdr_unknown, NULL }, 380340084Svangyzen}; 381340084Svangyzen 382340084Svangyzen/* 383340084Svangyzen * Send a formatted line; optionally echo to terminal 384340084Svangyzen */ 385340084Svangyzenstatic int 386340084Svangyzenhttp_cmd(conn_t *conn, const char *fmt, ...) 387340084Svangyzen{ 388340084Svangyzen va_list ap; 389340084Svangyzen size_t len; 390340084Svangyzen char *msg; 391340084Svangyzen int r; 392340084Svangyzen 393340084Svangyzen va_start(ap, fmt); 394340084Svangyzen len = vasprintf(&msg, fmt, ap); 395340084Svangyzen va_end(ap); 396340084Svangyzen 397340084Svangyzen if (msg == NULL) { 398340084Svangyzen errno = ENOMEM; 399340084Svangyzen fetch_syserr(); 400340084Svangyzen return (-1); 401340084Svangyzen } 402340084Svangyzen 403340084Svangyzen r = fetch_putln(conn, msg, len); 404340084Svangyzen free(msg); 405340084Svangyzen 406340084Svangyzen if (r == -1) { 407340084Svangyzen fetch_syserr(); 408340084Svangyzen return (-1); 409340084Svangyzen } 410340084Svangyzen 411340084Svangyzen return (0); 412340084Svangyzen} 413340084Svangyzen 414340084Svangyzen/* 415340084Svangyzen * Get and parse status line 416340084Svangyzen */ 417340084Svangyzenstatic int 418340084Svangyzenhttp_get_reply(conn_t *conn) 419340084Svangyzen{ 420340084Svangyzen char *p; 421340084Svangyzen 422340084Svangyzen if (fetch_getln(conn) == -1) 423340084Svangyzen return (-1); 424340084Svangyzen /* 425340084Svangyzen * A valid status line looks like "HTTP/m.n xyz reason" where m 426340084Svangyzen * and n are the major and minor protocol version numbers and xyz 427340084Svangyzen * is the reply code. 428340084Svangyzen * Unfortunately, there are servers out there (NCSA 1.5.1, to name 429340084Svangyzen * just one) that do not send a version number, so we can't rely 430340084Svangyzen * on finding one, but if we do, insist on it being 1.0 or 1.1. 431340084Svangyzen * We don't care about the reason phrase. 432340084Svangyzen */ 433340084Svangyzen if (strncmp(conn->buf, "HTTP", 4) != 0) 434340084Svangyzen return (HTTP_PROTOCOL_ERROR); 435340084Svangyzen p = conn->buf + 4; 436340084Svangyzen if (*p == '/') { 437340084Svangyzen if (p[1] != '1' || p[2] != '.' || (p[3] != '0' && p[3] != '1')) 438340084Svangyzen return (HTTP_PROTOCOL_ERROR); 439340084Svangyzen p += 4; 440340084Svangyzen } 441340084Svangyzen if (*p != ' ' || 442340084Svangyzen !isdigit((unsigned char)p[1]) || 443340084Svangyzen !isdigit((unsigned char)p[2]) || 444340084Svangyzen !isdigit((unsigned char)p[3])) 445340084Svangyzen return (HTTP_PROTOCOL_ERROR); 446340084Svangyzen 447340084Svangyzen conn->err = (p[1] - '0') * 100 + (p[2] - '0') * 10 + (p[3] - '0'); 448340084Svangyzen return (conn->err); 449340084Svangyzen} 450340084Svangyzen 451340084Svangyzen/* 452340084Svangyzen * Check a header; if the type matches the given string, return a pointer 453340084Svangyzen * to the beginning of the value. 454340084Svangyzen */ 455340084Svangyzenstatic const char * 456340084Svangyzenhttp_match(const char *str, const char *hdr) 457340084Svangyzen{ 458340084Svangyzen while (*str && *hdr && 459340084Svangyzen tolower((unsigned char)*str++) == tolower((unsigned char)*hdr++)) 460340084Svangyzen /* nothing */; 461340084Svangyzen if (*str || *hdr != ':') 462340084Svangyzen return (NULL); 463340084Svangyzen while (*hdr && isspace((unsigned char)*++hdr)) 464340084Svangyzen /* nothing */; 465340084Svangyzen return (hdr); 466340084Svangyzen} 467340084Svangyzen 468340084Svangyzen 469340084Svangyzen/* 470340084Svangyzen * Get the next header and return the appropriate symbolic code. We 471340084Svangyzen * need to read one line ahead for checking for a continuation line 472340084Svangyzen * belonging to the current header (continuation lines start with 473340084Svangyzen * white space). 474340084Svangyzen * 475340084Svangyzen * We get called with a fresh line already in the conn buffer, either 476340084Svangyzen * from the previous http_next_header() invocation, or, the first 477340084Svangyzen * time, from a fetch_getln() performed by our caller. 478340084Svangyzen * 479340084Svangyzen * This stops when we encounter an empty line (we dont read beyond the header 480340084Svangyzen * area). 481340084Svangyzen * 482340084Svangyzen * Note that the "headerbuf" is just a place to return the result. Its 483340084Svangyzen * contents are not used for the next call. This means that no cleanup 484340084Svangyzen * is needed when ie doing another connection, just call the cleanup when 485340084Svangyzen * fully done to deallocate memory. 486340084Svangyzen */ 487340084Svangyzen 488340084Svangyzen/* Limit the max number of continuation lines to some reasonable value */ 489340084Svangyzen#define HTTP_MAX_CONT_LINES 10 490340084Svangyzen 491340084Svangyzen/* Place into which to build a header from one or several lines */ 492340084Svangyzentypedef struct { 493340084Svangyzen char *buf; /* buffer */ 494340084Svangyzen size_t bufsize; /* buffer size */ 495340084Svangyzen size_t buflen; /* length of buffer contents */ 496340084Svangyzen} http_headerbuf_t; 497340084Svangyzen 498340084Svangyzenstatic void 499340084Svangyzeninit_http_headerbuf(http_headerbuf_t *buf) 500340084Svangyzen{ 501340084Svangyzen buf->buf = NULL; 502340084Svangyzen buf->bufsize = 0; 503340084Svangyzen buf->buflen = 0; 504340084Svangyzen} 505340084Svangyzen 506340084Svangyzenstatic void 507340084Svangyzenclean_http_headerbuf(http_headerbuf_t *buf) 508340084Svangyzen{ 509340084Svangyzen if (buf->buf) 510340084Svangyzen free(buf->buf); 511340084Svangyzen init_http_headerbuf(buf); 512340084Svangyzen} 513340084Svangyzen 514340084Svangyzen/* Remove whitespace at the end of the buffer */ 515340084Svangyzenstatic void 516340084Svangyzenhttp_conn_trimright(conn_t *conn) 517340084Svangyzen{ 518340084Svangyzen while (conn->buflen && 519340084Svangyzen isspace((unsigned char)conn->buf[conn->buflen - 1])) 520340084Svangyzen conn->buflen--; 521340084Svangyzen conn->buf[conn->buflen] = '\0'; 522340084Svangyzen} 523340084Svangyzen 524340084Svangyzenstatic hdr_t 525340084Svangyzenhttp_next_header(conn_t *conn, http_headerbuf_t *hbuf, const char **p) 526340084Svangyzen{ 527340084Svangyzen unsigned int i, len; 528340084Svangyzen 529340084Svangyzen /* 530340084Svangyzen * Have to do the stripping here because of the first line. So 531340084Svangyzen * it's done twice for the subsequent lines. No big deal 532340084Svangyzen */ 533340084Svangyzen http_conn_trimright(conn); 534340084Svangyzen if (conn->buflen == 0) 535340084Svangyzen return (hdr_end); 536340084Svangyzen 537340084Svangyzen /* Copy the line to the headerbuf */ 538340084Svangyzen if (hbuf->bufsize < conn->buflen + 1) { 539340084Svangyzen if ((hbuf->buf = realloc(hbuf->buf, conn->buflen + 1)) == NULL) 540340084Svangyzen return (hdr_syserror); 541340084Svangyzen hbuf->bufsize = conn->buflen + 1; 542340084Svangyzen } 543340084Svangyzen strcpy(hbuf->buf, conn->buf); 544340084Svangyzen hbuf->buflen = conn->buflen; 545340084Svangyzen 546340084Svangyzen /* 547340084Svangyzen * Fetch possible continuation lines. Stop at 1st non-continuation 548340084Svangyzen * and leave it in the conn buffer 549340084Svangyzen */ 550340084Svangyzen for (i = 0; i < HTTP_MAX_CONT_LINES; i++) { 551340084Svangyzen if (fetch_getln(conn) == -1) 552340084Svangyzen return (hdr_syserror); 553340084Svangyzen 554340084Svangyzen /* 555340084Svangyzen * Note: we carry on the idea from the previous version 556340084Svangyzen * that a pure whitespace line is equivalent to an empty 557340084Svangyzen * one (so it's not continuation and will be handled when 558340084Svangyzen * we are called next) 559340084Svangyzen */ 560340084Svangyzen http_conn_trimright(conn); 561340084Svangyzen if (conn->buf[0] != ' ' && conn->buf[0] != "\t"[0]) 562340084Svangyzen break; 563340084Svangyzen 564340084Svangyzen /* Got a continuation line. Concatenate to previous */ 565340084Svangyzen len = hbuf->buflen + conn->buflen; 566340084Svangyzen if (hbuf->bufsize < len + 1) { 567340084Svangyzen len *= 2; 568340084Svangyzen if ((hbuf->buf = realloc(hbuf->buf, len + 1)) == NULL) 569340084Svangyzen return (hdr_syserror); 570340084Svangyzen hbuf->bufsize = len + 1; 571340084Svangyzen } 572340084Svangyzen strcpy(hbuf->buf + hbuf->buflen, conn->buf); 573340084Svangyzen hbuf->buflen += conn->buflen; 574340084Svangyzen } 575340084Svangyzen 576340084Svangyzen /* 577340084Svangyzen * We could check for malformed headers but we don't really care. 578340084Svangyzen * A valid header starts with a token immediately followed by a 579340084Svangyzen * colon; a token is any sequence of non-control, non-whitespace 580340084Svangyzen * characters except "()<>@,;:\\\"{}". 581340084Svangyzen */ 582340084Svangyzen for (i = 0; hdr_names[i].num != hdr_unknown; i++) 583340084Svangyzen if ((*p = http_match(hdr_names[i].name, hbuf->buf)) != NULL) 584340084Svangyzen return (hdr_names[i].num); 585340084Svangyzen 586340084Svangyzen return (hdr_unknown); 587340084Svangyzen} 588340084Svangyzen 589340084Svangyzen/************************** 590340084Svangyzen * [Proxy-]Authenticate header parsing 591340084Svangyzen */ 592340084Svangyzen 593340084Svangyzen/* 594340084Svangyzen * Read doublequote-delimited string into output buffer obuf (allocated 595340084Svangyzen * by caller, whose responsibility it is to ensure that it's big enough) 596340084Svangyzen * cp points to the first char after the initial '"' 597340084Svangyzen * Handles \ quoting 598340084Svangyzen * Returns pointer to the first char after the terminating double quote, or 599340084Svangyzen * NULL for error. 600340084Svangyzen */ 601340084Svangyzenstatic const char * 602340084Svangyzenhttp_parse_headerstring(const char *cp, char *obuf) 603340084Svangyzen{ 604340084Svangyzen for (;;) { 605340084Svangyzen switch (*cp) { 606340084Svangyzen case 0: /* Unterminated string */ 607340084Svangyzen *obuf = 0; 608340084Svangyzen return (NULL); 609340084Svangyzen case '"': /* Ending quote */ 610340084Svangyzen *obuf = 0; 611340084Svangyzen return (++cp); 612340084Svangyzen case '\\': 613340084Svangyzen if (*++cp == 0) { 614340084Svangyzen *obuf = 0; 615340084Svangyzen return (NULL); 616340084Svangyzen } 617340084Svangyzen /* FALLTHROUGH */ 618340084Svangyzen default: 619340084Svangyzen *obuf++ = *cp++; 620340084Svangyzen } 621340084Svangyzen } 622340084Svangyzen} 623340084Svangyzen 624340084Svangyzen/* Http auth challenge schemes */ 625340084Svangyzentypedef enum {HTTPAS_UNKNOWN, HTTPAS_BASIC,HTTPAS_DIGEST} http_auth_schemes_t; 626340084Svangyzen 627340084Svangyzen/* Data holder for a Basic or Digest challenge. */ 628340084Svangyzentypedef struct { 629340084Svangyzen http_auth_schemes_t scheme; 630340084Svangyzen char *realm; 631340084Svangyzen char *qop; 632340084Svangyzen char *nonce; 633340084Svangyzen char *opaque; 634340084Svangyzen char *algo; 635340084Svangyzen int stale; 636340084Svangyzen int nc; /* Nonce count */ 637340084Svangyzen} http_auth_challenge_t; 638340084Svangyzen 639340084Svangyzenstatic void 640340084Svangyzeninit_http_auth_challenge(http_auth_challenge_t *b) 641340084Svangyzen{ 642340084Svangyzen b->scheme = HTTPAS_UNKNOWN; 643340084Svangyzen b->realm = b->qop = b->nonce = b->opaque = b->algo = NULL; 644340084Svangyzen b->stale = b->nc = 0; 645340084Svangyzen} 646340084Svangyzen 647340084Svangyzenstatic void 648340084Svangyzenclean_http_auth_challenge(http_auth_challenge_t *b) 649340084Svangyzen{ 650340084Svangyzen if (b->realm) 651340084Svangyzen free(b->realm); 652340084Svangyzen if (b->qop) 653340084Svangyzen free(b->qop); 654340084Svangyzen if (b->nonce) 655340084Svangyzen free(b->nonce); 656340084Svangyzen if (b->opaque) 657340084Svangyzen free(b->opaque); 658340084Svangyzen if (b->algo) 659340084Svangyzen free(b->algo); 660340084Svangyzen init_http_auth_challenge(b); 661340084Svangyzen} 662340084Svangyzen 663340084Svangyzen/* Data holder for an array of challenges offered in an http response. */ 664340084Svangyzen#define MAX_CHALLENGES 10 665340084Svangyzentypedef struct { 666340084Svangyzen http_auth_challenge_t *challenges[MAX_CHALLENGES]; 667340084Svangyzen int count; /* Number of parsed challenges in the array */ 668340084Svangyzen int valid; /* We did parse an authenticate header */ 669340084Svangyzen} http_auth_challenges_t; 670340084Svangyzen 671340084Svangyzenstatic void 672340084Svangyzeninit_http_auth_challenges(http_auth_challenges_t *cs) 673340084Svangyzen{ 674340084Svangyzen int i; 675340084Svangyzen for (i = 0; i < MAX_CHALLENGES; i++) 676340084Svangyzen cs->challenges[i] = NULL; 677340084Svangyzen cs->count = cs->valid = 0; 678340084Svangyzen} 679340084Svangyzen 680340084Svangyzenstatic void 681340084Svangyzenclean_http_auth_challenges(http_auth_challenges_t *cs) 682340084Svangyzen{ 683340084Svangyzen int i; 684340084Svangyzen /* We rely on non-zero pointers being allocated, not on the count */ 685340084Svangyzen for (i = 0; i < MAX_CHALLENGES; i++) { 686340084Svangyzen if (cs->challenges[i] != NULL) { 687340084Svangyzen clean_http_auth_challenge(cs->challenges[i]); 688340084Svangyzen free(cs->challenges[i]); 689340084Svangyzen } 690340084Svangyzen } 691340084Svangyzen init_http_auth_challenges(cs); 692340084Svangyzen} 693340084Svangyzen 694340084Svangyzen/* 695340084Svangyzen * Enumeration for lexical elements. Separators will be returned as their own 696340084Svangyzen * ascii value 697340084Svangyzen */ 698340084Svangyzentypedef enum {HTTPHL_WORD=256, HTTPHL_STRING=257, HTTPHL_END=258, 699340084Svangyzen HTTPHL_ERROR = 259} http_header_lex_t; 700340084Svangyzen 701340084Svangyzen/* 702340084Svangyzen * Determine what kind of token comes next and return possible value 703340084Svangyzen * in buf, which is supposed to have been allocated big enough by 704340084Svangyzen * caller. Advance input pointer and return element type. 705340084Svangyzen */ 706340084Svangyzenstatic int 707340084Svangyzenhttp_header_lex(const char **cpp, char *buf) 708340084Svangyzen{ 709340084Svangyzen size_t l; 710340084Svangyzen /* Eat initial whitespace */ 711340084Svangyzen *cpp += strspn(*cpp, " \t"); 712340084Svangyzen if (**cpp == 0) 713340084Svangyzen return (HTTPHL_END); 714340084Svangyzen 715340084Svangyzen /* Separator ? */ 716340084Svangyzen if (**cpp == ',' || **cpp == '=') 717340084Svangyzen return (*((*cpp)++)); 718340084Svangyzen 719340084Svangyzen /* String ? */ 720340084Svangyzen if (**cpp == '"') { 721340084Svangyzen *cpp = http_parse_headerstring(++*cpp, buf); 722340084Svangyzen if (*cpp == NULL) 723340084Svangyzen return (HTTPHL_ERROR); 724340084Svangyzen return (HTTPHL_STRING); 725340084Svangyzen } 726340084Svangyzen 727340084Svangyzen /* Read other token, until separator or whitespace */ 728340084Svangyzen l = strcspn(*cpp, " \t,="); 729340084Svangyzen memcpy(buf, *cpp, l); 730340084Svangyzen buf[l] = 0; 731340084Svangyzen *cpp += l; 732340084Svangyzen return (HTTPHL_WORD); 733340084Svangyzen} 734340084Svangyzen 735340084Svangyzen/* 736340084Svangyzen * Read challenges from http xxx-authenticate header and accumulate them 737340084Svangyzen * in the challenges list structure. 738340084Svangyzen * 739340084Svangyzen * Headers with multiple challenges are specified by rfc2617, but 740340084Svangyzen * servers (ie: squid) often send them in separate headers instead, 741340084Svangyzen * which in turn is forbidden by the http spec (multiple headers with 742340084Svangyzen * the same name are only allowed for pure comma-separated lists, see 743340084Svangyzen * rfc2616 sec 4.2). 744340084Svangyzen * 745340084Svangyzen * We support both approaches anyway 746340084Svangyzen */ 747340084Svangyzenstatic int 748340084Svangyzenhttp_parse_authenticate(const char *cp, http_auth_challenges_t *cs) 749340084Svangyzen{ 750340084Svangyzen int ret = -1; 751340084Svangyzen http_header_lex_t lex; 752340084Svangyzen char *key = malloc(strlen(cp) + 1); 753340084Svangyzen char *value = malloc(strlen(cp) + 1); 754340084Svangyzen char *buf = malloc(strlen(cp) + 1); 755340084Svangyzen 756340084Svangyzen if (key == NULL || value == NULL || buf == NULL) { 757340084Svangyzen fetch_syserr(); 758340084Svangyzen goto out; 759340084Svangyzen } 760340084Svangyzen 761340084Svangyzen /* In any case we've seen the header and we set the valid bit */ 762340084Svangyzen cs->valid = 1; 763340084Svangyzen 764340084Svangyzen /* Need word first */ 765340084Svangyzen lex = http_header_lex(&cp, key); 766340084Svangyzen if (lex != HTTPHL_WORD) 767340084Svangyzen goto out; 768340084Svangyzen 769340084Svangyzen /* Loop on challenges */ 770340084Svangyzen for (; cs->count < MAX_CHALLENGES; cs->count++) { 771340084Svangyzen cs->challenges[cs->count] = 772340084Svangyzen malloc(sizeof(http_auth_challenge_t)); 773340084Svangyzen if (cs->challenges[cs->count] == NULL) { 774340084Svangyzen fetch_syserr(); 775340084Svangyzen goto out; 776340084Svangyzen } 777340084Svangyzen init_http_auth_challenge(cs->challenges[cs->count]); 778340084Svangyzen if (!strcasecmp(key, "basic")) { 779340084Svangyzen cs->challenges[cs->count]->scheme = HTTPAS_BASIC; 780340084Svangyzen } else if (!strcasecmp(key, "digest")) { 781340084Svangyzen cs->challenges[cs->count]->scheme = HTTPAS_DIGEST; 782340084Svangyzen } else { 783340084Svangyzen cs->challenges[cs->count]->scheme = HTTPAS_UNKNOWN; 784340084Svangyzen /* 785340084Svangyzen * Continue parsing as basic or digest may 786340084Svangyzen * follow, and the syntax is the same for 787340084Svangyzen * all. We'll just ignore this one when 788340084Svangyzen * looking at the list 789340084Svangyzen */ 790340084Svangyzen } 791340084Svangyzen 792340084Svangyzen /* Loop on attributes */ 793340084Svangyzen for (;;) { 794340084Svangyzen /* Key */ 795340084Svangyzen lex = http_header_lex(&cp, key); 796340084Svangyzen if (lex != HTTPHL_WORD) 797340084Svangyzen goto out; 798340084Svangyzen 799340084Svangyzen /* Equal sign */ 800340084Svangyzen lex = http_header_lex(&cp, buf); 801340084Svangyzen if (lex != '=') 802340084Svangyzen goto out; 803340084Svangyzen 804340084Svangyzen /* Value */ 805340084Svangyzen lex = http_header_lex(&cp, value); 806340084Svangyzen if (lex != HTTPHL_WORD && lex != HTTPHL_STRING) 807340084Svangyzen goto out; 808340084Svangyzen 809340084Svangyzen if (!strcasecmp(key, "realm")) 810340084Svangyzen cs->challenges[cs->count]->realm = 811340084Svangyzen strdup(value); 812340084Svangyzen else if (!strcasecmp(key, "qop")) 813340084Svangyzen cs->challenges[cs->count]->qop = 814340084Svangyzen strdup(value); 815340084Svangyzen else if (!strcasecmp(key, "nonce")) 816340084Svangyzen cs->challenges[cs->count]->nonce = 817340084Svangyzen strdup(value); 818340084Svangyzen else if (!strcasecmp(key, "opaque")) 819340084Svangyzen cs->challenges[cs->count]->opaque = 820340084Svangyzen strdup(value); 821340084Svangyzen else if (!strcasecmp(key, "algorithm")) 822340084Svangyzen cs->challenges[cs->count]->algo = 823340084Svangyzen strdup(value); 824340084Svangyzen else if (!strcasecmp(key, "stale")) 825340084Svangyzen cs->challenges[cs->count]->stale = 826340084Svangyzen strcasecmp(value, "no"); 827340084Svangyzen /* Else ignore unknown attributes */ 828340084Svangyzen 829340084Svangyzen /* Comma or Next challenge or End */ 830340084Svangyzen lex = http_header_lex(&cp, key); 831340084Svangyzen /* 832340084Svangyzen * If we get a word here, this is the beginning of the 833340084Svangyzen * next challenge. Break the attributes loop 834340084Svangyzen */ 835340084Svangyzen if (lex == HTTPHL_WORD) 836340084Svangyzen break; 837340084Svangyzen 838340084Svangyzen if (lex == HTTPHL_END) { 839340084Svangyzen /* End while looking for ',' is normal exit */ 840340084Svangyzen cs->count++; 841340084Svangyzen ret = 0; 842340084Svangyzen goto out; 843340084Svangyzen } 844340084Svangyzen /* Anything else is an error */ 845340084Svangyzen if (lex != ',') 846340084Svangyzen goto out; 847340084Svangyzen 848340084Svangyzen } /* End attributes loop */ 849340084Svangyzen } /* End challenge loop */ 850340084Svangyzen 851340084Svangyzen /* 852340084Svangyzen * Challenges max count exceeded. This really can't happen 853340084Svangyzen * with normal data, something's fishy -> error 854340084Svangyzen */ 855340084Svangyzen 856340084Svangyzenout: 857340084Svangyzen if (key) 858340084Svangyzen free(key); 859340084Svangyzen if (value) 860340084Svangyzen free(value); 861340084Svangyzen if (buf) 862340084Svangyzen free(buf); 863340084Svangyzen return (ret); 864340084Svangyzen} 865340084Svangyzen 866340084Svangyzen 867340084Svangyzen/* 868340084Svangyzen * Parse a last-modified header 869340084Svangyzen */ 870340084Svangyzenstatic int 871340084Svangyzenhttp_parse_mtime(const char *p, time_t *mtime) 872340084Svangyzen{ 873340084Svangyzen char locale[64], *r; 874340084Svangyzen struct tm tm; 875340084Svangyzen 876340084Svangyzen strncpy(locale, setlocale(LC_TIME, NULL), sizeof(locale)); 877340084Svangyzen setlocale(LC_TIME, "C"); 878340084Svangyzen r = strptime(p, "%a, %d %b %Y %H:%M:%S GMT", &tm); 879340084Svangyzen /* 880340084Svangyzen * Some proxies use UTC in response, but it should still be 881340084Svangyzen * parsed. RFC2616 states GMT and UTC are exactly equal for HTTP. 882340084Svangyzen */ 883340084Svangyzen if (r == NULL) 884340084Svangyzen r = strptime(p, "%a, %d %b %Y %H:%M:%S UTC", &tm); 885340084Svangyzen /* XXX should add support for date-2 and date-3 */ 886340084Svangyzen setlocale(LC_TIME, locale); 887340084Svangyzen if (r == NULL) 888340084Svangyzen return (-1); 889340084Svangyzen DEBUG(fprintf(stderr, "last modified: [%04d-%02d-%02d " 890340084Svangyzen "%02d:%02d:%02d]\n", 891340084Svangyzen tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday, 892340084Svangyzen tm.tm_hour, tm.tm_min, tm.tm_sec)); 893340084Svangyzen *mtime = timegm(&tm); 894340084Svangyzen return (0); 895340084Svangyzen} 896340084Svangyzen 897340084Svangyzen/* 898340084Svangyzen * Parse a content-length header 899340084Svangyzen */ 900340084Svangyzenstatic int 901340084Svangyzenhttp_parse_length(const char *p, off_t *length) 902340084Svangyzen{ 903340084Svangyzen off_t len; 904340084Svangyzen 905340084Svangyzen for (len = 0; *p && isdigit((unsigned char)*p); ++p) 906340084Svangyzen len = len * 10 + (*p - '0'); 907340084Svangyzen if (*p) 908340084Svangyzen return (-1); 909340084Svangyzen DEBUG(fprintf(stderr, "content length: [%lld]\n", 910340084Svangyzen (long long)len)); 911340084Svangyzen *length = len; 912340084Svangyzen return (0); 913340084Svangyzen} 914340084Svangyzen 915340084Svangyzen/* 916340084Svangyzen * Parse a content-range header 917340084Svangyzen */ 918340084Svangyzenstatic int 919340084Svangyzenhttp_parse_range(const char *p, off_t *offset, off_t *length, off_t *size) 920340084Svangyzen{ 921340084Svangyzen off_t first, last, len; 922340084Svangyzen 923340084Svangyzen if (strncasecmp(p, "bytes ", 6) != 0) 924340084Svangyzen return (-1); 925340084Svangyzen p += 6; 926340084Svangyzen if (*p == '*') { 927340084Svangyzen first = last = -1; 928340084Svangyzen ++p; 929340084Svangyzen } else { 930340084Svangyzen for (first = 0; *p && isdigit((unsigned char)*p); ++p) 931340084Svangyzen first = first * 10 + *p - '0'; 932340084Svangyzen if (*p != '-') 933340084Svangyzen return (-1); 934340084Svangyzen for (last = 0, ++p; *p && isdigit((unsigned char)*p); ++p) 935340084Svangyzen last = last * 10 + *p - '0'; 936340084Svangyzen } 937340084Svangyzen if (first > last || *p != '/') 938340084Svangyzen return (-1); 939340084Svangyzen for (len = 0, ++p; *p && isdigit((unsigned char)*p); ++p) 940340084Svangyzen len = len * 10 + *p - '0'; 941340084Svangyzen if (*p || len < last - first + 1) 942340084Svangyzen return (-1); 943340084Svangyzen if (first == -1) { 944340084Svangyzen DEBUG(fprintf(stderr, "content range: [*/%lld]\n", 945340084Svangyzen (long long)len)); 946340084Svangyzen *length = 0; 947340084Svangyzen } else { 948340084Svangyzen DEBUG(fprintf(stderr, "content range: [%lld-%lld/%lld]\n", 949340084Svangyzen (long long)first, (long long)last, (long long)len)); 950340084Svangyzen *length = last - first + 1; 951340084Svangyzen } 952340084Svangyzen *offset = first; 953340084Svangyzen *size = len; 954340084Svangyzen return (0); 955340084Svangyzen} 956340084Svangyzen 957340084Svangyzen 958340084Svangyzen/***************************************************************************** 959340084Svangyzen * Helper functions for authorization 960340084Svangyzen */ 961340084Svangyzen 962340084Svangyzen/* 963340084Svangyzen * Base64 encoding 964340084Svangyzen */ 965340084Svangyzenstatic char * 966340084Svangyzenhttp_base64(const char *src) 967340084Svangyzen{ 968340084Svangyzen static const char base64[] = 969340084Svangyzen "ABCDEFGHIJKLMNOPQRSTUVWXYZ" 970340084Svangyzen "abcdefghijklmnopqrstuvwxyz" 971340084Svangyzen "0123456789+/"; 972340084Svangyzen char *str, *dst; 973340084Svangyzen size_t l; 974340084Svangyzen int t, r; 975340084Svangyzen 976340084Svangyzen l = strlen(src); 977340084Svangyzen if ((str = malloc(((l + 2) / 3) * 4 + 1)) == NULL) 978340084Svangyzen return (NULL); 979340084Svangyzen dst = str; 980340084Svangyzen r = 0; 981340084Svangyzen 982340084Svangyzen while (l >= 3) { 983340084Svangyzen t = (src[0] << 16) | (src[1] << 8) | src[2]; 984340084Svangyzen dst[0] = base64[(t >> 18) & 0x3f]; 985340084Svangyzen dst[1] = base64[(t >> 12) & 0x3f]; 986340084Svangyzen dst[2] = base64[(t >> 6) & 0x3f]; 987340084Svangyzen dst[3] = base64[(t >> 0) & 0x3f]; 988340084Svangyzen src += 3; l -= 3; 989340084Svangyzen dst += 4; r += 4; 990340084Svangyzen } 991340084Svangyzen 992340084Svangyzen switch (l) { 993340084Svangyzen case 2: 994340084Svangyzen t = (src[0] << 16) | (src[1] << 8); 995340084Svangyzen dst[0] = base64[(t >> 18) & 0x3f]; 996340084Svangyzen dst[1] = base64[(t >> 12) & 0x3f]; 997340084Svangyzen dst[2] = base64[(t >> 6) & 0x3f]; 998340084Svangyzen dst[3] = '='; 999340084Svangyzen dst += 4; 1000340084Svangyzen r += 4; 1001340084Svangyzen break; 1002340084Svangyzen case 1: 1003340084Svangyzen t = src[0] << 16; 1004340084Svangyzen dst[0] = base64[(t >> 18) & 0x3f]; 1005340084Svangyzen dst[1] = base64[(t >> 12) & 0x3f]; 1006340084Svangyzen dst[2] = dst[3] = '='; 1007340084Svangyzen dst += 4; 1008340084Svangyzen r += 4; 1009340084Svangyzen break; 1010340084Svangyzen case 0: 1011340084Svangyzen break; 1012340084Svangyzen } 1013340084Svangyzen 1014340084Svangyzen *dst = 0; 1015340084Svangyzen return (str); 1016340084Svangyzen} 1017340084Svangyzen 1018340084Svangyzen 1019340084Svangyzen/* 1020340084Svangyzen * Extract authorization parameters from environment value. 1021340084Svangyzen * The value is like scheme:realm:user:pass 1022340084Svangyzen */ 1023340084Svangyzentypedef struct { 1024340084Svangyzen char *scheme; 1025340084Svangyzen char *realm; 1026340084Svangyzen char *user; 1027340084Svangyzen char *password; 1028340084Svangyzen} http_auth_params_t; 1029340084Svangyzen 1030340084Svangyzenstatic void 1031340084Svangyzeninit_http_auth_params(http_auth_params_t *s) 1032340084Svangyzen{ 1033340084Svangyzen s->scheme = s->realm = s->user = s->password = NULL; 1034340084Svangyzen} 1035340084Svangyzen 1036340084Svangyzenstatic void 1037340084Svangyzenclean_http_auth_params(http_auth_params_t *s) 1038340084Svangyzen{ 1039340084Svangyzen if (s->scheme) 1040340084Svangyzen free(s->scheme); 1041340084Svangyzen if (s->realm) 1042340084Svangyzen free(s->realm); 1043340084Svangyzen if (s->user) 1044340084Svangyzen free(s->user); 1045340084Svangyzen if (s->password) 1046340084Svangyzen free(s->password); 1047340084Svangyzen init_http_auth_params(s); 1048340084Svangyzen} 1049340084Svangyzen 1050340084Svangyzenstatic int 1051340084Svangyzenhttp_authfromenv(const char *p, http_auth_params_t *parms) 1052340084Svangyzen{ 1053340084Svangyzen int ret = -1; 1054340084Svangyzen char *v, *ve; 1055340084Svangyzen char *str = strdup(p); 1056340084Svangyzen 1057340084Svangyzen if (str == NULL) { 1058340084Svangyzen fetch_syserr(); 1059340084Svangyzen return (-1); 1060340084Svangyzen } 1061340084Svangyzen v = str; 1062340084Svangyzen 1063340084Svangyzen if ((ve = strchr(v, ':')) == NULL) 1064340084Svangyzen goto out; 1065340084Svangyzen 1066340084Svangyzen *ve = 0; 1067340084Svangyzen if ((parms->scheme = strdup(v)) == NULL) { 1068340084Svangyzen fetch_syserr(); 1069340084Svangyzen goto out; 1070340084Svangyzen } 1071340084Svangyzen v = ve + 1; 1072340084Svangyzen 1073340084Svangyzen if ((ve = strchr(v, ':')) == NULL) 1074340084Svangyzen goto out; 1075340084Svangyzen 1076340084Svangyzen *ve = 0; 1077340084Svangyzen if ((parms->realm = strdup(v)) == NULL) { 1078340084Svangyzen fetch_syserr(); 1079340084Svangyzen goto out; 1080340084Svangyzen } 1081340084Svangyzen v = ve + 1; 1082340084Svangyzen 1083340084Svangyzen if ((ve = strchr(v, ':')) == NULL) 1084340084Svangyzen goto out; 1085340084Svangyzen 1086340084Svangyzen *ve = 0; 1087340084Svangyzen if ((parms->user = strdup(v)) == NULL) { 1088340084Svangyzen fetch_syserr(); 1089340084Svangyzen goto out; 1090340084Svangyzen } 1091340084Svangyzen v = ve + 1; 1092340084Svangyzen 1093340084Svangyzen 1094340084Svangyzen if ((parms->password = strdup(v)) == NULL) { 1095340084Svangyzen fetch_syserr(); 1096340084Svangyzen goto out; 1097340084Svangyzen } 1098340084Svangyzen ret = 0; 1099340084Svangyzenout: 1100340084Svangyzen if (ret == -1) 1101340084Svangyzen clean_http_auth_params(parms); 1102340084Svangyzen if (str) 1103340084Svangyzen free(str); 1104340084Svangyzen return (ret); 1105340084Svangyzen} 1106340084Svangyzen 1107340084Svangyzen 1108340084Svangyzen/* 1109340084Svangyzen * Digest response: the code to compute the digest is taken from the 1110340084Svangyzen * sample implementation in RFC2616 1111340084Svangyzen */ 1112340084Svangyzen#define IN const 1113340084Svangyzen#define OUT 1114340084Svangyzen 1115340084Svangyzen#define HASHLEN 16 1116340084Svangyzentypedef char HASH[HASHLEN]; 1117340084Svangyzen#define HASHHEXLEN 32 1118340084Svangyzentypedef char HASHHEX[HASHHEXLEN+1]; 1119340084Svangyzen 1120340084Svangyzenstatic const char *hexchars = "0123456789abcdef"; 1121340084Svangyzenstatic void 1122340084SvangyzenCvtHex(IN HASH Bin, OUT HASHHEX Hex) 1123340084Svangyzen{ 1124340084Svangyzen unsigned short i; 1125340084Svangyzen unsigned char j; 1126340084Svangyzen 1127340084Svangyzen for (i = 0; i < HASHLEN; i++) { 1128340084Svangyzen j = (Bin[i] >> 4) & 0xf; 1129340084Svangyzen Hex[i*2] = hexchars[j]; 1130340084Svangyzen j = Bin[i] & 0xf; 1131340084Svangyzen Hex[i*2+1] = hexchars[j]; 1132340084Svangyzen } 1133340084Svangyzen Hex[HASHHEXLEN] = '\0'; 1134340084Svangyzen}; 1135340084Svangyzen 1136340084Svangyzen/* calculate H(A1) as per spec */ 1137340084Svangyzenstatic void 1138340084SvangyzenDigestCalcHA1( 1139340084Svangyzen IN char * pszAlg, 1140340084Svangyzen IN char * pszUserName, 1141340084Svangyzen IN char * pszRealm, 1142340084Svangyzen IN char * pszPassword, 1143340084Svangyzen IN char * pszNonce, 1144340084Svangyzen IN char * pszCNonce, 1145340084Svangyzen OUT HASHHEX SessionKey 1146340084Svangyzen ) 1147340084Svangyzen{ 1148340084Svangyzen MD5_CTX Md5Ctx; 1149340084Svangyzen HASH HA1; 1150340084Svangyzen 1151340084Svangyzen MD5Init(&Md5Ctx); 1152340084Svangyzen MD5Update(&Md5Ctx, pszUserName, strlen(pszUserName)); 1153340084Svangyzen MD5Update(&Md5Ctx, ":", 1); 1154340084Svangyzen MD5Update(&Md5Ctx, pszRealm, strlen(pszRealm)); 1155340084Svangyzen MD5Update(&Md5Ctx, ":", 1); 1156340084Svangyzen MD5Update(&Md5Ctx, pszPassword, strlen(pszPassword)); 1157340084Svangyzen MD5Final(HA1, &Md5Ctx); 1158340084Svangyzen if (strcasecmp(pszAlg, "md5-sess") == 0) { 1159340084Svangyzen 1160340084Svangyzen MD5Init(&Md5Ctx); 1161340084Svangyzen MD5Update(&Md5Ctx, HA1, HASHLEN); 1162340084Svangyzen MD5Update(&Md5Ctx, ":", 1); 1163340084Svangyzen MD5Update(&Md5Ctx, pszNonce, strlen(pszNonce)); 1164340084Svangyzen MD5Update(&Md5Ctx, ":", 1); 1165340084Svangyzen MD5Update(&Md5Ctx, pszCNonce, strlen(pszCNonce)); 1166340084Svangyzen MD5Final(HA1, &Md5Ctx); 1167340084Svangyzen } 1168340084Svangyzen CvtHex(HA1, SessionKey); 1169340084Svangyzen} 1170340084Svangyzen 1171340084Svangyzen/* calculate request-digest/response-digest as per HTTP Digest spec */ 1172340084Svangyzenstatic void 1173340084SvangyzenDigestCalcResponse( 1174340084Svangyzen IN HASHHEX HA1, /* H(A1) */ 1175340084Svangyzen IN char * pszNonce, /* nonce from server */ 1176340084Svangyzen IN char * pszNonceCount, /* 8 hex digits */ 1177340084Svangyzen IN char * pszCNonce, /* client nonce */ 1178340084Svangyzen IN char * pszQop, /* qop-value: "", "auth", "auth-int" */ 1179340084Svangyzen IN char * pszMethod, /* method from the request */ 1180340084Svangyzen IN char * pszDigestUri, /* requested URL */ 1181340084Svangyzen IN HASHHEX HEntity, /* H(entity body) if qop="auth-int" */ 1182340084Svangyzen OUT HASHHEX Response /* request-digest or response-digest */ 1183340084Svangyzen ) 1184340084Svangyzen{ 1185340084Svangyzen/* DEBUG(fprintf(stderr, 1186340084Svangyzen "Calc: HA1[%s] Nonce[%s] qop[%s] method[%s] URI[%s]\n", 1187340084Svangyzen HA1, pszNonce, pszQop, pszMethod, pszDigestUri));*/ 1188340084Svangyzen MD5_CTX Md5Ctx; 1189340084Svangyzen HASH HA2; 1190340084Svangyzen HASH RespHash; 1191340084Svangyzen HASHHEX HA2Hex; 1192340084Svangyzen 1193340084Svangyzen // calculate H(A2) 1194340084Svangyzen MD5Init(&Md5Ctx); 1195340084Svangyzen MD5Update(&Md5Ctx, pszMethod, strlen(pszMethod)); 1196340084Svangyzen MD5Update(&Md5Ctx, ":", 1); 1197340084Svangyzen MD5Update(&Md5Ctx, pszDigestUri, strlen(pszDigestUri)); 1198340084Svangyzen if (strcasecmp(pszQop, "auth-int") == 0) { 1199340084Svangyzen MD5Update(&Md5Ctx, ":", 1); 1200340084Svangyzen MD5Update(&Md5Ctx, HEntity, HASHHEXLEN); 1201340084Svangyzen } 1202340084Svangyzen MD5Final(HA2, &Md5Ctx); 1203340084Svangyzen CvtHex(HA2, HA2Hex); 1204340084Svangyzen 1205340084Svangyzen // calculate response 1206340084Svangyzen MD5Init(&Md5Ctx); 1207340084Svangyzen MD5Update(&Md5Ctx, HA1, HASHHEXLEN); 1208340084Svangyzen MD5Update(&Md5Ctx, ":", 1); 1209340084Svangyzen MD5Update(&Md5Ctx, pszNonce, strlen(pszNonce)); 1210340084Svangyzen MD5Update(&Md5Ctx, ":", 1); 1211340084Svangyzen if (*pszQop) { 1212340084Svangyzen MD5Update(&Md5Ctx, pszNonceCount, strlen(pszNonceCount)); 1213340084Svangyzen MD5Update(&Md5Ctx, ":", 1); 1214340084Svangyzen MD5Update(&Md5Ctx, pszCNonce, strlen(pszCNonce)); 1215340084Svangyzen MD5Update(&Md5Ctx, ":", 1); 1216340084Svangyzen MD5Update(&Md5Ctx, pszQop, strlen(pszQop)); 1217340084Svangyzen MD5Update(&Md5Ctx, ":", 1); 1218340084Svangyzen } 1219340084Svangyzen MD5Update(&Md5Ctx, HA2Hex, HASHHEXLEN); 1220 MD5Final(RespHash, &Md5Ctx); 1221 CvtHex(RespHash, Response); 1222} 1223 1224/* 1225 * Generate/Send a Digest authorization header 1226 * This looks like: [Proxy-]Authorization: credentials 1227 * 1228 * credentials = "Digest" digest-response 1229 * digest-response = 1#( username | realm | nonce | digest-uri 1230 * | response | [ algorithm ] | [cnonce] | 1231 * [opaque] | [message-qop] | 1232 * [nonce-count] | [auth-param] ) 1233 * username = "username" "=" username-value 1234 * username-value = quoted-string 1235 * digest-uri = "uri" "=" digest-uri-value 1236 * digest-uri-value = request-uri ; As specified by HTTP/1.1 1237 * message-qop = "qop" "=" qop-value 1238 * cnonce = "cnonce" "=" cnonce-value 1239 * cnonce-value = nonce-value 1240 * nonce-count = "nc" "=" nc-value 1241 * nc-value = 8LHEX 1242 * response = "response" "=" request-digest 1243 * request-digest = <"> 32LHEX <"> 1244 */ 1245static int 1246http_digest_auth(conn_t *conn, const char *hdr, http_auth_challenge_t *c, 1247 http_auth_params_t *parms, struct url *url) 1248{ 1249 int r; 1250 char noncecount[10]; 1251 char cnonce[40]; 1252 char *options = NULL; 1253 1254 if (!c->realm || !c->nonce) { 1255 DEBUG(fprintf(stderr, "realm/nonce not set in challenge\n")); 1256 return(-1); 1257 } 1258 if (!c->algo) 1259 c->algo = strdup(""); 1260 1261 if (asprintf(&options, "%s%s%s%s", 1262 *c->algo? ",algorithm=" : "", c->algo, 1263 c->opaque? ",opaque=" : "", c->opaque?c->opaque:"")== -1) 1264 return (-1); 1265 1266 if (!c->qop) { 1267 c->qop = strdup(""); 1268 *noncecount = 0; 1269 *cnonce = 0; 1270 } else { 1271 c->nc++; 1272 sprintf(noncecount, "%08x", c->nc); 1273 /* We don't try very hard with the cnonce ... */ 1274 sprintf(cnonce, "%x%lx", getpid(), (unsigned long)time(0)); 1275 } 1276 1277 HASHHEX HA1; 1278 DigestCalcHA1(c->algo, parms->user, c->realm, 1279 parms->password, c->nonce, cnonce, HA1); 1280 DEBUG(fprintf(stderr, "HA1: [%s]\n", HA1)); 1281 HASHHEX digest; 1282 DigestCalcResponse(HA1, c->nonce, noncecount, cnonce, c->qop, 1283 "GET", url->doc, "", digest); 1284 1285 if (c->qop[0]) { 1286 r = http_cmd(conn, "%s: Digest username=\"%s\",realm=\"%s\"," 1287 "nonce=\"%s\",uri=\"%s\",response=\"%s\"," 1288 "qop=\"auth\", cnonce=\"%s\", nc=%s%s", 1289 hdr, parms->user, c->realm, 1290 c->nonce, url->doc, digest, 1291 cnonce, noncecount, options); 1292 } else { 1293 r = http_cmd(conn, "%s: Digest username=\"%s\",realm=\"%s\"," 1294 "nonce=\"%s\",uri=\"%s\",response=\"%s\"%s", 1295 hdr, parms->user, c->realm, 1296 c->nonce, url->doc, digest, options); 1297 } 1298 if (options) 1299 free(options); 1300 return (r); 1301} 1302 1303/* 1304 * Encode username and password 1305 */ 1306static int 1307http_basic_auth(conn_t *conn, const char *hdr, const char *usr, const char *pwd) 1308{ 1309 char *upw, *auth; 1310 int r; 1311 1312 DEBUG(fprintf(stderr, "basic: usr: [%s]\n", usr)); 1313 DEBUG(fprintf(stderr, "basic: pwd: [%s]\n", pwd)); 1314 if (asprintf(&upw, "%s:%s", usr, pwd) == -1) 1315 return (-1); 1316 auth = http_base64(upw); 1317 free(upw); 1318 if (auth == NULL) 1319 return (-1); 1320 r = http_cmd(conn, "%s: Basic %s", hdr, auth); 1321 free(auth); 1322 return (r); 1323} 1324 1325/* 1326 * Chose the challenge to answer and call the appropriate routine to 1327 * produce the header. 1328 */ 1329static int 1330http_authorize(conn_t *conn, const char *hdr, http_auth_challenges_t *cs, 1331 http_auth_params_t *parms, struct url *url) 1332{ 1333 http_auth_challenge_t *basic = NULL; 1334 http_auth_challenge_t *digest = NULL; 1335 int i; 1336 1337 /* If user or pass are null we're not happy */ 1338 if (!parms->user || !parms->password) { 1339 DEBUG(fprintf(stderr, "NULL usr or pass\n")); 1340 return (-1); 1341 } 1342 1343 /* Look for a Digest and a Basic challenge */ 1344 for (i = 0; i < cs->count; i++) { 1345 if (cs->challenges[i]->scheme == HTTPAS_BASIC) 1346 basic = cs->challenges[i]; 1347 if (cs->challenges[i]->scheme == HTTPAS_DIGEST) 1348 digest = cs->challenges[i]; 1349 } 1350 1351 /* Error if "Digest" was specified and there is no Digest challenge */ 1352 if (!digest && (parms->scheme && 1353 !strcasecmp(parms->scheme, "digest"))) { 1354 DEBUG(fprintf(stderr, 1355 "Digest auth in env, not supported by peer\n")); 1356 return (-1); 1357 } 1358 /* 1359 * If "basic" was specified in the environment, or there is no Digest 1360 * challenge, do the basic thing. Don't need a challenge for this, 1361 * so no need to check basic!=NULL 1362 */ 1363 if (!digest || (parms->scheme && !strcasecmp(parms->scheme,"basic"))) 1364 return (http_basic_auth(conn,hdr,parms->user,parms->password)); 1365 1366 /* Else, prefer digest. We just checked that it's not NULL */ 1367 return (http_digest_auth(conn, hdr, digest, parms, url)); 1368} 1369 1370/***************************************************************************** 1371 * Helper functions for connecting to a server or proxy 1372 */ 1373 1374/* 1375 * Connect to the correct HTTP server or proxy. 1376 */ 1377static conn_t * 1378http_connect(struct url *URL, struct url *purl, const char *flags) 1379{ 1380 struct url *curl; 1381 conn_t *conn; 1382 int verbose; 1383 int af, val; 1384 1385#ifdef INET6 1386 af = AF_UNSPEC; 1387#else 1388 af = AF_INET; 1389#endif 1390 1391 verbose = CHECK_FLAG('v'); 1392 if (CHECK_FLAG('4')) 1393 af = AF_INET; 1394#ifdef INET6 1395 else if (CHECK_FLAG('6')) 1396 af = AF_INET6; 1397#endif 1398 1399 curl = (purl != NULL) ? purl : URL; 1400 1401 if ((conn = fetch_connect(curl->host, curl->port, af, verbose)) == NULL) 1402 /* fetch_connect() has already set an error code */ 1403 return (NULL); 1404 if (strcasecmp(URL->scheme, SCHEME_HTTPS) == 0 && purl) { 1405 http_cmd(conn, "CONNECT %s:%d HTTP/1.1", 1406 URL->host, URL->port); 1407 http_cmd(conn, "Host: %s:%d", 1408 URL->host, URL->port); 1409 http_cmd(conn, ""); 1410 if (http_get_reply(conn) != HTTP_OK) { 1411 fetch_close(conn); 1412 return (NULL); 1413 } 1414 http_get_reply(conn); 1415 } 1416 if (strcasecmp(URL->scheme, SCHEME_HTTPS) == 0 && 1417 fetch_ssl(conn, URL, verbose) == -1) { 1418 fetch_close(conn); 1419 /* grrr */ 1420 errno = EAUTH; 1421 fetch_syserr(); 1422 return (NULL); 1423 } 1424 1425 val = 1; 1426 setsockopt(conn->sd, IPPROTO_TCP, TCP_NOPUSH, &val, sizeof(val)); 1427 1428 return (conn); 1429} 1430 1431static struct url * 1432http_get_proxy(struct url * url, const char *flags) 1433{ 1434 struct url *purl; 1435 char *p; 1436 1437 if (flags != NULL && strchr(flags, 'd') != NULL) 1438 return (NULL); 1439 if (fetch_no_proxy_match(url->host)) 1440 return (NULL); 1441 if (((p = getenv("HTTP_PROXY")) || (p = getenv("http_proxy"))) && 1442 *p && (purl = fetchParseURL(p))) { 1443 if (!*purl->scheme) 1444 strcpy(purl->scheme, SCHEME_HTTP); 1445 if (!purl->port) 1446 purl->port = fetch_default_proxy_port(purl->scheme); 1447 if (strcasecmp(purl->scheme, SCHEME_HTTP) == 0) 1448 return (purl); 1449 fetchFreeURL(purl); 1450 } 1451 return (NULL); 1452} 1453 1454static void 1455http_print_html(FILE *out, FILE *in) 1456{ 1457 size_t len; 1458 char *line, *p, *q; 1459 int comment, tag; 1460 1461 comment = tag = 0; 1462 while ((line = fgetln(in, &len)) != NULL) { 1463 while (len && isspace((unsigned char)line[len - 1])) 1464 --len; 1465 for (p = q = line; q < line + len; ++q) { 1466 if (comment && *q == '-') { 1467 if (q + 2 < line + len && 1468 strcmp(q, "-->") == 0) { 1469 tag = comment = 0; 1470 q += 2; 1471 } 1472 } else if (tag && !comment && *q == '>') { 1473 p = q + 1; 1474 tag = 0; 1475 } else if (!tag && *q == '<') { 1476 if (q > p) 1477 fwrite(p, q - p, 1, out); 1478 tag = 1; 1479 if (q + 3 < line + len && 1480 strcmp(q, "<!--") == 0) { 1481 comment = 1; 1482 q += 3; 1483 } 1484 } 1485 } 1486 if (!tag && q > p) 1487 fwrite(p, q - p, 1, out); 1488 fputc('\n', out); 1489 } 1490} 1491 1492 1493/***************************************************************************** 1494 * Core 1495 */ 1496 1497FILE * 1498http_request(struct url *URL, const char *op, struct url_stat *us, 1499 struct url *purl, const char *flags) 1500{ 1501 1502 return (http_request_body(URL, op, us, purl, flags, NULL, NULL)); 1503} 1504 1505/* 1506 * Send a request and process the reply 1507 * 1508 * XXX This function is way too long, the do..while loop should be split 1509 * XXX off into a separate function. 1510 */ 1511FILE * 1512http_request_body(struct url *URL, const char *op, struct url_stat *us, 1513 struct url *purl, const char *flags, const char *content_type, 1514 const char *body) 1515{ 1516 char timebuf[80]; 1517 char hbuf[MAXHOSTNAMELEN + 7], *host; 1518 conn_t *conn; 1519 struct url *url, *new; 1520 int chunked, direct, ims, noredirect, verbose; 1521 int e, i, n, val; 1522 off_t offset, clength, length, size; 1523 time_t mtime; 1524 const char *p; 1525 FILE *f; 1526 hdr_t h; 1527 struct tm *timestruct; 1528 http_headerbuf_t headerbuf; 1529 http_auth_challenges_t server_challenges; 1530 http_auth_challenges_t proxy_challenges; 1531 size_t body_len; 1532 1533 /* The following calls don't allocate anything */ 1534 init_http_headerbuf(&headerbuf); 1535 init_http_auth_challenges(&server_challenges); 1536 init_http_auth_challenges(&proxy_challenges); 1537 1538 direct = CHECK_FLAG('d'); 1539 noredirect = CHECK_FLAG('A'); 1540 verbose = CHECK_FLAG('v'); 1541 ims = CHECK_FLAG('i'); 1542 1543 if (direct && purl) { 1544 fetchFreeURL(purl); 1545 purl = NULL; 1546 } 1547 1548 /* try the provided URL first */ 1549 url = URL; 1550 1551 n = MAX_REDIRECT; 1552 i = 0; 1553 1554 e = HTTP_PROTOCOL_ERROR; 1555 do { 1556 new = NULL; 1557 chunked = 0; 1558 offset = 0; 1559 clength = -1; 1560 length = -1; 1561 size = -1; 1562 mtime = 0; 1563 1564 /* check port */ 1565 if (!url->port) 1566 url->port = fetch_default_port(url->scheme); 1567 1568 /* were we redirected to an FTP URL? */ 1569 if (purl == NULL && strcmp(url->scheme, SCHEME_FTP) == 0) { 1570 if (strcmp(op, "GET") == 0) 1571 return (ftp_request(url, "RETR", us, purl, flags)); 1572 else if (strcmp(op, "HEAD") == 0) 1573 return (ftp_request(url, "STAT", us, purl, flags)); 1574 } 1575 1576 /* connect to server or proxy */ 1577 if ((conn = http_connect(url, purl, flags)) == NULL) 1578 goto ouch; 1579 1580 host = url->host; 1581#ifdef INET6 1582 if (strchr(url->host, ':')) { 1583 snprintf(hbuf, sizeof(hbuf), "[%s]", url->host); 1584 host = hbuf; 1585 } 1586#endif 1587 if (url->port != fetch_default_port(url->scheme)) { 1588 if (host != hbuf) { 1589 strcpy(hbuf, host); 1590 host = hbuf; 1591 } 1592 snprintf(hbuf + strlen(hbuf), 1593 sizeof(hbuf) - strlen(hbuf), ":%d", url->port); 1594 } 1595 1596 /* send request */ 1597 if (verbose) 1598 fetch_info("requesting %s://%s%s", 1599 url->scheme, host, url->doc); 1600 if (purl && strcasecmp(URL->scheme, SCHEME_HTTPS) != 0) { 1601 http_cmd(conn, "%s %s://%s%s HTTP/1.1", 1602 op, url->scheme, host, url->doc); 1603 } else { 1604 http_cmd(conn, "%s %s HTTP/1.1", 1605 op, url->doc); 1606 } 1607 1608 if (ims && url->ims_time) { 1609 timestruct = gmtime((time_t *)&url->ims_time); 1610 (void)strftime(timebuf, 80, "%a, %d %b %Y %T GMT", 1611 timestruct); 1612 if (verbose) 1613 fetch_info("If-Modified-Since: %s", timebuf); 1614 http_cmd(conn, "If-Modified-Since: %s", timebuf); 1615 } 1616 /* virtual host */ 1617 http_cmd(conn, "Host: %s", host); 1618 1619 /* 1620 * Proxy authorization: we only send auth after we received 1621 * a 407 error. We do not first try basic anyway (changed 1622 * when support was added for digest-auth) 1623 */ 1624 if (purl && proxy_challenges.valid) { 1625 http_auth_params_t aparams; 1626 init_http_auth_params(&aparams); 1627 if (*purl->user || *purl->pwd) { 1628 aparams.user = purl->user ? 1629 strdup(purl->user) : strdup(""); 1630 aparams.password = purl->pwd? 1631 strdup(purl->pwd) : strdup(""); 1632 } else if ((p = getenv("HTTP_PROXY_AUTH")) != NULL && 1633 *p != '\0') { 1634 if (http_authfromenv(p, &aparams) < 0) { 1635 http_seterr(HTTP_NEED_PROXY_AUTH); 1636 goto ouch; 1637 } 1638 } 1639 http_authorize(conn, "Proxy-Authorization", 1640 &proxy_challenges, &aparams, url); 1641 clean_http_auth_params(&aparams); 1642 } 1643 1644 /* 1645 * Server authorization: we never send "a priori" 1646 * Basic auth, which used to be done if user/pass were 1647 * set in the url. This would be weird because we'd send the 1648 * password in the clear even if Digest is finally to be 1649 * used (it would have made more sense for the 1650 * pre-digest version to do this when Basic was specified 1651 * in the environment) 1652 */ 1653 if (server_challenges.valid) { 1654 http_auth_params_t aparams; 1655 init_http_auth_params(&aparams); 1656 if (*url->user || *url->pwd) { 1657 aparams.user = url->user ? 1658 strdup(url->user) : strdup(""); 1659 aparams.password = url->pwd ? 1660 strdup(url->pwd) : strdup(""); 1661 } else if ((p = getenv("HTTP_AUTH")) != NULL && 1662 *p != '\0') { 1663 if (http_authfromenv(p, &aparams) < 0) { 1664 http_seterr(HTTP_NEED_AUTH); 1665 goto ouch; 1666 } 1667 } else if (fetchAuthMethod && 1668 fetchAuthMethod(url) == 0) { 1669 aparams.user = url->user ? 1670 strdup(url->user) : strdup(""); 1671 aparams.password = url->pwd ? 1672 strdup(url->pwd) : strdup(""); 1673 } else { 1674 http_seterr(HTTP_NEED_AUTH); 1675 goto ouch; 1676 } 1677 http_authorize(conn, "Authorization", 1678 &server_challenges, &aparams, url); 1679 clean_http_auth_params(&aparams); 1680 } 1681 1682 /* other headers */ 1683 if ((p = getenv("HTTP_ACCEPT")) != NULL) { 1684 if (*p != '\0') 1685 http_cmd(conn, "Accept: %s", p); 1686 } else { 1687 http_cmd(conn, "Accept: */*"); 1688 } 1689 if ((p = getenv("HTTP_REFERER")) != NULL && *p != '\0') { 1690 if (strcasecmp(p, "auto") == 0) 1691 http_cmd(conn, "Referer: %s://%s%s", 1692 url->scheme, host, url->doc); 1693 else 1694 http_cmd(conn, "Referer: %s", p); 1695 } 1696 if ((p = getenv("HTTP_USER_AGENT")) != NULL) { 1697 /* no User-Agent if defined but empty */ 1698 if (*p != '\0') 1699 http_cmd(conn, "User-Agent: %s", p); 1700 } else { 1701 /* default User-Agent */ 1702 http_cmd(conn, "User-Agent: %s " _LIBFETCH_VER, 1703 getprogname()); 1704 } 1705 if (url->offset > 0) 1706 http_cmd(conn, "Range: bytes=%lld-", (long long)url->offset); 1707 http_cmd(conn, "Connection: close"); 1708 1709 if (body) { 1710 body_len = strlen(body); 1711 http_cmd(conn, "Content-Length: %zu", body_len); 1712 if (content_type != NULL) 1713 http_cmd(conn, "Content-Type: %s", content_type); 1714 } 1715 1716 http_cmd(conn, ""); 1717 1718 if (body) 1719 fetch_write(conn, body, body_len); 1720 1721 /* 1722 * Force the queued request to be dispatched. Normally, one 1723 * would do this with shutdown(2) but squid proxies can be 1724 * configured to disallow such half-closed connections. To 1725 * be compatible with such configurations, fiddle with socket 1726 * options to force the pending data to be written. 1727 */ 1728 val = 0; 1729 setsockopt(conn->sd, IPPROTO_TCP, TCP_NOPUSH, &val, 1730 sizeof(val)); 1731 val = 1; 1732 setsockopt(conn->sd, IPPROTO_TCP, TCP_NODELAY, &val, 1733 sizeof(val)); 1734 1735 /* get reply */ 1736 switch (http_get_reply(conn)) { 1737 case HTTP_OK: 1738 case HTTP_PARTIAL: 1739 case HTTP_NOT_MODIFIED: 1740 /* fine */ 1741 break; 1742 case HTTP_MOVED_PERM: 1743 case HTTP_MOVED_TEMP: 1744 case HTTP_SEE_OTHER: 1745 case HTTP_USE_PROXY: 1746 /* 1747 * Not so fine, but we still have to read the 1748 * headers to get the new location. 1749 */ 1750 break; 1751 case HTTP_NEED_AUTH: 1752 if (server_challenges.valid) { 1753 /* 1754 * We already sent out authorization code, 1755 * so there's nothing more we can do. 1756 */ 1757 http_seterr(conn->err); 1758 goto ouch; 1759 } 1760 /* try again, but send the password this time */ 1761 if (verbose) 1762 fetch_info("server requires authorization"); 1763 break; 1764 case HTTP_NEED_PROXY_AUTH: 1765 if (proxy_challenges.valid) { 1766 /* 1767 * We already sent our proxy 1768 * authorization code, so there's 1769 * nothing more we can do. */ 1770 http_seterr(conn->err); 1771 goto ouch; 1772 } 1773 /* try again, but send the password this time */ 1774 if (verbose) 1775 fetch_info("proxy requires authorization"); 1776 break; 1777 case HTTP_BAD_RANGE: 1778 /* 1779 * This can happen if we ask for 0 bytes because 1780 * we already have the whole file. Consider this 1781 * a success for now, and check sizes later. 1782 */ 1783 break; 1784 case HTTP_PROTOCOL_ERROR: 1785 /* fall through */ 1786 case -1: 1787 fetch_syserr(); 1788 goto ouch; 1789 default: 1790 http_seterr(conn->err); 1791 if (!verbose) 1792 goto ouch; 1793 /* fall through so we can get the full error message */ 1794 } 1795 1796 /* get headers. http_next_header expects one line readahead */ 1797 if (fetch_getln(conn) == -1) { 1798 fetch_syserr(); 1799 goto ouch; 1800 } 1801 do { 1802 switch ((h = http_next_header(conn, &headerbuf, &p))) { 1803 case hdr_syserror: 1804 fetch_syserr(); 1805 goto ouch; 1806 case hdr_error: 1807 http_seterr(HTTP_PROTOCOL_ERROR); 1808 goto ouch; 1809 case hdr_content_length: 1810 http_parse_length(p, &clength); 1811 break; 1812 case hdr_content_range: 1813 http_parse_range(p, &offset, &length, &size); 1814 break; 1815 case hdr_last_modified: 1816 http_parse_mtime(p, &mtime); 1817 break; 1818 case hdr_location: 1819 if (!HTTP_REDIRECT(conn->err)) 1820 break; 1821 /* 1822 * if the A flag is set, we don't follow 1823 * temporary redirects. 1824 */ 1825 if (noredirect && 1826 conn->err != HTTP_MOVED_PERM && 1827 conn->err != HTTP_PERM_REDIRECT && 1828 conn->err != HTTP_USE_PROXY) { 1829 n = 1; 1830 break; 1831 } 1832 if (new) 1833 free(new); 1834 if (verbose) 1835 fetch_info("%d redirect to %s", conn->err, p); 1836 if (*p == '/') 1837 /* absolute path */ 1838 new = fetchMakeURL(url->scheme, url->host, url->port, p, 1839 url->user, url->pwd); 1840 else 1841 new = fetchParseURL(p); 1842 if (new == NULL) { 1843 /* XXX should set an error code */ 1844 DEBUG(fprintf(stderr, "failed to parse new URL\n")); 1845 goto ouch; 1846 } 1847 1848 /* Only copy credentials if the host matches */ 1849 if (!strcmp(new->host, url->host) && !*new->user && !*new->pwd) { 1850 strcpy(new->user, url->user); 1851 strcpy(new->pwd, url->pwd); 1852 } 1853 new->offset = url->offset; 1854 new->length = url->length; 1855 break; 1856 case hdr_transfer_encoding: 1857 /* XXX weak test*/ 1858 chunked = (strcasecmp(p, "chunked") == 0); 1859 break; 1860 case hdr_www_authenticate: 1861 if (conn->err != HTTP_NEED_AUTH) 1862 break; 1863 if (http_parse_authenticate(p, &server_challenges) == 0) 1864 ++n; 1865 break; 1866 case hdr_proxy_authenticate: 1867 if (conn->err != HTTP_NEED_PROXY_AUTH) 1868 break; 1869 if (http_parse_authenticate(p, &proxy_challenges) == 0) 1870 ++n; 1871 break; 1872 case hdr_end: 1873 /* fall through */ 1874 case hdr_unknown: 1875 /* ignore */ 1876 break; 1877 } 1878 } while (h > hdr_end); 1879 1880 /* we need to provide authentication */ 1881 if (conn->err == HTTP_NEED_AUTH || 1882 conn->err == HTTP_NEED_PROXY_AUTH) { 1883 e = conn->err; 1884 if ((conn->err == HTTP_NEED_AUTH && 1885 !server_challenges.valid) || 1886 (conn->err == HTTP_NEED_PROXY_AUTH && 1887 !proxy_challenges.valid)) { 1888 /* 401/7 but no www/proxy-authenticate ?? */ 1889 DEBUG(fprintf(stderr, "401/7 and no auth header\n")); 1890 goto ouch; 1891 } 1892 fetch_close(conn); 1893 conn = NULL; 1894 continue; 1895 } 1896 1897 /* requested range not satisfiable */ 1898 if (conn->err == HTTP_BAD_RANGE) { 1899 if (url->offset == size && url->length == 0) { 1900 /* asked for 0 bytes; fake it */ 1901 offset = url->offset; 1902 clength = -1; 1903 conn->err = HTTP_OK; 1904 break; 1905 } else { 1906 http_seterr(conn->err); 1907 goto ouch; 1908 } 1909 } 1910 1911 /* we have a hit or an error */ 1912 if (conn->err == HTTP_OK 1913 || conn->err == HTTP_NOT_MODIFIED 1914 || conn->err == HTTP_PARTIAL 1915 || HTTP_ERROR(conn->err)) 1916 break; 1917 1918 /* all other cases: we got a redirect */ 1919 e = conn->err; 1920 clean_http_auth_challenges(&server_challenges); 1921 fetch_close(conn); 1922 conn = NULL; 1923 if (!new) { 1924 DEBUG(fprintf(stderr, "redirect with no new location\n")); 1925 break; 1926 } 1927 if (url != URL) 1928 fetchFreeURL(url); 1929 url = new; 1930 } while (++i < n); 1931 1932 /* we failed, or ran out of retries */ 1933 if (conn == NULL) { 1934 http_seterr(e); 1935 goto ouch; 1936 } 1937 1938 DEBUG(fprintf(stderr, "offset %lld, length %lld," 1939 " size %lld, clength %lld\n", 1940 (long long)offset, (long long)length, 1941 (long long)size, (long long)clength)); 1942 1943 if (conn->err == HTTP_NOT_MODIFIED) { 1944 http_seterr(HTTP_NOT_MODIFIED); 1945 return (NULL); 1946 } 1947 1948 /* check for inconsistencies */ 1949 if (clength != -1 && length != -1 && clength != length) { 1950 http_seterr(HTTP_PROTOCOL_ERROR); 1951 goto ouch; 1952 } 1953 if (clength == -1) 1954 clength = length; 1955 if (clength != -1) 1956 length = offset + clength; 1957 if (length != -1 && size != -1 && length != size) { 1958 http_seterr(HTTP_PROTOCOL_ERROR); 1959 goto ouch; 1960 } 1961 if (size == -1) 1962 size = length; 1963 1964 /* fill in stats */ 1965 if (us) { 1966 us->size = size; 1967 us->atime = us->mtime = mtime; 1968 } 1969 1970 /* too far? */ 1971 if (URL->offset > 0 && offset > URL->offset) { 1972 http_seterr(HTTP_PROTOCOL_ERROR); 1973 goto ouch; 1974 } 1975 1976 /* report back real offset and size */ 1977 URL->offset = offset; 1978 URL->length = clength; 1979 1980 /* wrap it up in a FILE */ 1981 if ((f = http_funopen(conn, chunked)) == NULL) { 1982 fetch_syserr(); 1983 goto ouch; 1984 } 1985 1986 if (url != URL) 1987 fetchFreeURL(url); 1988 if (purl) 1989 fetchFreeURL(purl); 1990 1991 if (HTTP_ERROR(conn->err)) { 1992 http_print_html(stderr, f); 1993 fclose(f); 1994 f = NULL; 1995 } 1996 clean_http_headerbuf(&headerbuf); 1997 clean_http_auth_challenges(&server_challenges); 1998 clean_http_auth_challenges(&proxy_challenges); 1999 return (f); 2000 2001ouch: 2002 if (url != URL) 2003 fetchFreeURL(url); 2004 if (purl) 2005 fetchFreeURL(purl); 2006 if (conn != NULL) 2007 fetch_close(conn); 2008 clean_http_headerbuf(&headerbuf); 2009 clean_http_auth_challenges(&server_challenges); 2010 clean_http_auth_challenges(&proxy_challenges); 2011 return (NULL); 2012} 2013 2014 2015/***************************************************************************** 2016 * Entry points 2017 */ 2018 2019/* 2020 * Retrieve and stat a file by HTTP 2021 */ 2022FILE * 2023fetchXGetHTTP(struct url *URL, struct url_stat *us, const char *flags) 2024{ 2025 return (http_request(URL, "GET", us, http_get_proxy(URL, flags), flags)); 2026} 2027 2028/* 2029 * Retrieve a file by HTTP 2030 */ 2031FILE * 2032fetchGetHTTP(struct url *URL, const char *flags) 2033{ 2034 return (fetchXGetHTTP(URL, NULL, flags)); 2035} 2036 2037/* 2038 * Store a file by HTTP 2039 */ 2040FILE * 2041fetchPutHTTP(struct url *URL __unused, const char *flags __unused) 2042{ 2043 warnx("fetchPutHTTP(): not implemented"); 2044 return (NULL); 2045} 2046 2047/* 2048 * Get an HTTP document's metadata 2049 */ 2050int 2051fetchStatHTTP(struct url *URL, struct url_stat *us, const char *flags) 2052{ 2053 FILE *f; 2054 2055 f = http_request(URL, "HEAD", us, http_get_proxy(URL, flags), flags); 2056 if (f == NULL) 2057 return (-1); 2058 fclose(f); 2059 return (0); 2060} 2061 2062/* 2063 * List a directory 2064 */ 2065struct url_ent * 2066fetchListHTTP(struct url *url __unused, const char *flags __unused) 2067{ 2068 warnx("fetchListHTTP(): not implemented"); 2069 return (NULL); 2070} 2071 2072FILE * 2073fetchReqHTTP(struct url *URL, const char *method, const char *flags, 2074 const char *content_type, const char *body) 2075{ 2076 2077 return (http_request_body(URL, method, NULL, http_get_proxy(URL, flags), 2078 flags, content_type, body)); 2079} 2080