]> sjero.net Git - wget/blob - src/http.c
The prefered way is to avoid #ifdef flooding, so take it that way. Introduce iri...
[wget] / src / http.c
1 /* HTTP support.
2    Copyright (C) 1996, 1997, 1998, 1999, 2000, 2001, 2002, 2003,
3    2004, 2005, 2006, 2007, 2008 Free Software Foundation, Inc.
4
5 This file is part of GNU Wget.
6
7 GNU Wget is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or
10  (at your option) any later version.
11
12 GNU Wget is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 GNU General Public License for more details.
16
17 You should have received a copy of the GNU General Public License
18 along with Wget.  If not, see <http://www.gnu.org/licenses/>.
19
20 Additional permission under GNU GPL version 3 section 7
21
22 If you modify this program, or any covered work, by linking or
23 combining it with the OpenSSL project's OpenSSL library (or a
24 modified version of that library), containing parts covered by the
25 terms of the OpenSSL or SSLeay licenses, the Free Software Foundation
26 grants you additional permission to convey the resulting work.
27 Corresponding Source for a non-source form of such a combination
28 shall include the source code for the parts of OpenSSL used as well
29 as that of the covered work.  */
30
31 #include "wget.h"
32
33 #include <stdio.h>
34 #include <stdlib.h>
35 #include <string.h>
36 #ifdef HAVE_UNISTD_H
37 # include <unistd.h>
38 #endif
39 #include <assert.h>
40 #include <errno.h>
41 #include <time.h>
42 #include <locale.h>
43
44 #include "hash.h"
45 #include "http.h"
46 #include "utils.h"
47 #include "url.h"
48 #include "host.h"
49 #include "retr.h"
50 #include "connect.h"
51 #include "netrc.h"
52 #include "iri.h"
53 #ifdef HAVE_SSL
54 # include "ssl.h"
55 #endif
56 #ifdef ENABLE_NTLM
57 # include "http-ntlm.h"
58 #endif
59 #include "cookies.h"
60 #ifdef ENABLE_DIGEST
61 # include "gen-md5.h"
62 #endif
63 #include "convert.h"
64 #include "spider.h"
65
66 #ifdef TESTING
67 #include "test.h"
68 #endif
69
70 extern char *version_string;
71
72 /* Forward decls. */
73 static char *create_authorization_line (const char *, const char *,
74                                         const char *, const char *,
75                                         const char *, bool *);
76 static char *basic_authentication_encode (const char *, const char *);
77 static bool known_authentication_scheme_p (const char *, const char *);
78 static void load_cookies (void);
79
80 #ifndef MIN
81 # define MIN(x, y) ((x) > (y) ? (y) : (x))
82 #endif
83
84 \f
85 static bool cookies_loaded_p;
86 static struct cookie_jar *wget_cookie_jar;
87
88 #define TEXTHTML_S "text/html"
89 #define TEXTXHTML_S "application/xhtml+xml"
90
91 /* Some status code validation macros: */
92 #define H_20X(x)        (((x) >= 200) && ((x) < 300))
93 #define H_PARTIAL(x)    ((x) == HTTP_STATUS_PARTIAL_CONTENTS)
94 #define H_REDIRECTED(x) ((x) == HTTP_STATUS_MOVED_PERMANENTLY          \
95                          || (x) == HTTP_STATUS_MOVED_TEMPORARILY       \
96                          || (x) == HTTP_STATUS_SEE_OTHER               \
97                          || (x) == HTTP_STATUS_TEMPORARY_REDIRECT)
98
99 /* HTTP/1.0 status codes from RFC1945, provided for reference.  */
100 /* Successful 2xx.  */
101 #define HTTP_STATUS_OK                    200
102 #define HTTP_STATUS_CREATED               201
103 #define HTTP_STATUS_ACCEPTED              202
104 #define HTTP_STATUS_NO_CONTENT            204
105 #define HTTP_STATUS_PARTIAL_CONTENTS      206
106
107 /* Redirection 3xx.  */
108 #define HTTP_STATUS_MULTIPLE_CHOICES      300
109 #define HTTP_STATUS_MOVED_PERMANENTLY     301
110 #define HTTP_STATUS_MOVED_TEMPORARILY     302
111 #define HTTP_STATUS_SEE_OTHER             303 /* from HTTP/1.1 */
112 #define HTTP_STATUS_NOT_MODIFIED          304
113 #define HTTP_STATUS_TEMPORARY_REDIRECT    307 /* from HTTP/1.1 */
114
115 /* Client error 4xx.  */
116 #define HTTP_STATUS_BAD_REQUEST           400
117 #define HTTP_STATUS_UNAUTHORIZED          401
118 #define HTTP_STATUS_FORBIDDEN             403
119 #define HTTP_STATUS_NOT_FOUND             404
120 #define HTTP_STATUS_RANGE_NOT_SATISFIABLE 416
121
122 /* Server errors 5xx.  */
123 #define HTTP_STATUS_INTERNAL              500
124 #define HTTP_STATUS_NOT_IMPLEMENTED       501
125 #define HTTP_STATUS_BAD_GATEWAY           502
126 #define HTTP_STATUS_UNAVAILABLE           503
127 \f
128 enum rp {
129   rel_none, rel_name, rel_value, rel_both
130 };
131
132 struct request {
133   const char *method;
134   char *arg;
135
136   struct request_header {
137     char *name, *value;
138     enum rp release_policy;
139   } *headers;
140   int hcount, hcapacity;
141 };
142
143 /* Create a new, empty request.  At least request_set_method must be
144    called before the request can be used.  */
145
146 static struct request *
147 request_new (void)
148 {
149   struct request *req = xnew0 (struct request);
150   req->hcapacity = 8;
151   req->headers = xnew_array (struct request_header, req->hcapacity);
152   return req;
153 }
154
155 /* Set the request's method and its arguments.  METH should be a
156    literal string (or it should outlive the request) because it will
157    not be freed.  ARG will be freed by request_free.  */
158
159 static void
160 request_set_method (struct request *req, const char *meth, char *arg)
161 {
162   req->method = meth;
163   req->arg = arg;
164 }
165
166 /* Return the method string passed with the last call to
167    request_set_method.  */
168
169 static const char *
170 request_method (const struct request *req)
171 {
172   return req->method;
173 }
174
175 /* Free one header according to the release policy specified with
176    request_set_header.  */
177
178 static void
179 release_header (struct request_header *hdr)
180 {
181   switch (hdr->release_policy)
182     {
183     case rel_none:
184       break;
185     case rel_name:
186       xfree (hdr->name);
187       break;
188     case rel_value:
189       xfree (hdr->value);
190       break;
191     case rel_both:
192       xfree (hdr->name);
193       xfree (hdr->value);
194       break;
195     }
196 }
197
198 /* Set the request named NAME to VALUE.  Specifically, this means that
199    a "NAME: VALUE\r\n" header line will be used in the request.  If a
200    header with the same name previously existed in the request, its
201    value will be replaced by this one.  A NULL value means do nothing.
202
203    RELEASE_POLICY determines whether NAME and VALUE should be released
204    (freed) with request_free.  Allowed values are:
205
206     - rel_none     - don't free NAME or VALUE
207     - rel_name     - free NAME when done
208     - rel_value    - free VALUE when done
209     - rel_both     - free both NAME and VALUE when done
210
211    Setting release policy is useful when arguments come from different
212    sources.  For example:
213
214      // Don't free literal strings!
215      request_set_header (req, "Pragma", "no-cache", rel_none);
216
217      // Don't free a global variable, we'll need it later.
218      request_set_header (req, "Referer", opt.referer, rel_none);
219
220      // Value freshly allocated, free it when done.
221      request_set_header (req, "Range",
222                          aprintf ("bytes=%s-", number_to_static_string (hs->restval)),
223                          rel_value);
224    */
225
226 static void
227 request_set_header (struct request *req, char *name, char *value,
228                     enum rp release_policy)
229 {
230   struct request_header *hdr;
231   int i;
232
233   if (!value)
234     {
235       /* A NULL value is a no-op; if freeing the name is requested,
236          free it now to avoid leaks.  */
237       if (release_policy == rel_name || release_policy == rel_both)
238         xfree (name);
239       return;
240     }
241
242   for (i = 0; i < req->hcount; i++)
243     {
244       hdr = &req->headers[i];
245       if (0 == strcasecmp (name, hdr->name))
246         {
247           /* Replace existing header. */
248           release_header (hdr);
249           hdr->name = name;
250           hdr->value = value;
251           hdr->release_policy = release_policy;
252           return;
253         }
254     }
255
256   /* Install new header. */
257
258   if (req->hcount >= req->hcapacity)
259     {
260       req->hcapacity <<= 1;
261       req->headers = xrealloc (req->headers, req->hcapacity * sizeof (*hdr));
262     }
263   hdr = &req->headers[req->hcount++];
264   hdr->name = name;
265   hdr->value = value;
266   hdr->release_policy = release_policy;
267 }
268
269 /* Like request_set_header, but sets the whole header line, as
270    provided by the user using the `--header' option.  For example,
271    request_set_user_header (req, "Foo: bar") works just like
272    request_set_header (req, "Foo", "bar").  */
273
274 static void
275 request_set_user_header (struct request *req, const char *header)
276 {
277   char *name;
278   const char *p = strchr (header, ':');
279   if (!p)
280     return;
281   BOUNDED_TO_ALLOCA (header, p, name);
282   ++p;
283   while (c_isspace (*p))
284     ++p;
285   request_set_header (req, xstrdup (name), (char *) p, rel_name);
286 }
287
288 /* Remove the header with specified name from REQ.  Returns true if
289    the header was actually removed, false otherwise.  */
290
291 static bool
292 request_remove_header (struct request *req, char *name)
293 {
294   int i;
295   for (i = 0; i < req->hcount; i++)
296     {
297       struct request_header *hdr = &req->headers[i];
298       if (0 == strcasecmp (name, hdr->name))
299         {
300           release_header (hdr);
301           /* Move the remaining headers by one. */
302           if (i < req->hcount - 1)
303             memmove (hdr, hdr + 1, (req->hcount - i - 1) * sizeof (*hdr));
304           --req->hcount;
305           return true;
306         }
307     }
308   return false;
309 }
310
311 #define APPEND(p, str) do {                     \
312   int A_len = strlen (str);                     \
313   memcpy (p, str, A_len);                       \
314   p += A_len;                                   \
315 } while (0)
316
317 /* Construct the request and write it to FD using fd_write.  */
318
319 static int
320 request_send (const struct request *req, int fd)
321 {
322   char *request_string, *p;
323   int i, size, write_error;
324
325   /* Count the request size. */
326   size = 0;
327
328   /* METHOD " " ARG " " "HTTP/1.0" "\r\n" */
329   size += strlen (req->method) + 1 + strlen (req->arg) + 1 + 8 + 2;
330
331   for (i = 0; i < req->hcount; i++)
332     {
333       struct request_header *hdr = &req->headers[i];
334       /* NAME ": " VALUE "\r\n" */
335       size += strlen (hdr->name) + 2 + strlen (hdr->value) + 2;
336     }
337
338   /* "\r\n\0" */
339   size += 3;
340
341   p = request_string = alloca_array (char, size);
342
343   /* Generate the request. */
344
345   APPEND (p, req->method); *p++ = ' ';
346   APPEND (p, req->arg);    *p++ = ' ';
347   memcpy (p, "HTTP/1.0\r\n", 10); p += 10;
348
349   for (i = 0; i < req->hcount; i++)
350     {
351       struct request_header *hdr = &req->headers[i];
352       APPEND (p, hdr->name);
353       *p++ = ':', *p++ = ' ';
354       APPEND (p, hdr->value);
355       *p++ = '\r', *p++ = '\n';
356     }
357
358   *p++ = '\r', *p++ = '\n', *p++ = '\0';
359   assert (p - request_string == size);
360
361 #undef APPEND
362
363   DEBUGP (("\n---request begin---\n%s---request end---\n", request_string));
364
365   /* Send the request to the server. */
366
367   write_error = fd_write (fd, request_string, size - 1, -1);
368   if (write_error < 0)
369     logprintf (LOG_VERBOSE, _("Failed writing HTTP request: %s.\n"),
370                fd_errstr (fd));
371   return write_error;
372 }
373
374 /* Release the resources used by REQ. */
375
376 static void
377 request_free (struct request *req)
378 {
379   int i;
380   xfree_null (req->arg);
381   for (i = 0; i < req->hcount; i++)
382     release_header (&req->headers[i]);
383   xfree_null (req->headers);
384   xfree (req);
385 }
386
387 static struct hash_table *basic_authed_hosts;
388
389 /* Find out if this host has issued a Basic challenge yet; if so, give
390  * it the username, password. A temporary measure until we can get
391  * proper authentication in place. */
392
393 static bool
394 maybe_send_basic_creds (const char *hostname, const char *user,
395                         const char *passwd, struct request *req)
396 {
397   bool do_challenge = false;
398
399   if (opt.auth_without_challenge)
400     {
401       DEBUGP(("Auth-without-challenge set, sending Basic credentials.\n"));
402       do_challenge = true;
403     }
404   else if (basic_authed_hosts
405       && hash_table_contains(basic_authed_hosts, hostname))
406     {
407       DEBUGP(("Found %s in basic_authed_hosts.\n", quote (hostname)));
408       do_challenge = true;
409     }
410   else
411     {
412       DEBUGP(("Host %s has not issued a general basic challenge.\n",
413               quote (hostname)));
414     }
415   if (do_challenge)
416     {
417       request_set_header (req, "Authorization",
418                           basic_authentication_encode (user, passwd),
419                           rel_value);
420     }
421   return do_challenge;
422 }
423
424 static void
425 register_basic_auth_host (const char *hostname)
426 {
427   if (!basic_authed_hosts)
428     {
429       basic_authed_hosts = make_nocase_string_hash_table (1);
430     }
431   if (!hash_table_contains(basic_authed_hosts, hostname))
432     {
433       hash_table_put (basic_authed_hosts, xstrdup(hostname), NULL);
434       DEBUGP(("Inserted %s into basic_authed_hosts\n", quote (hostname)));
435     }
436 }
437
438
439 /* Send the contents of FILE_NAME to SOCK.  Make sure that exactly
440    PROMISED_SIZE bytes are sent over the wire -- if the file is
441    longer, read only that much; if the file is shorter, report an error.  */
442
443 static int
444 post_file (int sock, const char *file_name, wgint promised_size)
445 {
446   static char chunk[8192];
447   wgint written = 0;
448   int write_error;
449   FILE *fp;
450
451   DEBUGP (("[writing POST file %s ... ", file_name));
452
453   fp = fopen (file_name, "rb");
454   if (!fp)
455     return -1;
456   while (!feof (fp) && written < promised_size)
457     {
458       int towrite;
459       int length = fread (chunk, 1, sizeof (chunk), fp);
460       if (length == 0)
461         break;
462       towrite = MIN (promised_size - written, length);
463       write_error = fd_write (sock, chunk, towrite, -1);
464       if (write_error < 0)
465         {
466           fclose (fp);
467           return -1;
468         }
469       written += towrite;
470     }
471   fclose (fp);
472
473   /* If we've written less than was promised, report a (probably
474      nonsensical) error rather than break the promise.  */
475   if (written < promised_size)
476     {
477       errno = EINVAL;
478       return -1;
479     }
480
481   assert (written == promised_size);
482   DEBUGP (("done]\n"));
483   return 0;
484 }
485 \f
486 /* Determine whether [START, PEEKED + PEEKLEN) contains an empty line.
487    If so, return the pointer to the position after the line, otherwise
488    return NULL.  This is used as callback to fd_read_hunk.  The data
489    between START and PEEKED has been read and cannot be "unread"; the
490    data after PEEKED has only been peeked.  */
491
492 static const char *
493 response_head_terminator (const char *start, const char *peeked, int peeklen)
494 {
495   const char *p, *end;
496
497   /* If at first peek, verify whether HUNK starts with "HTTP".  If
498      not, this is a HTTP/0.9 request and we must bail out without
499      reading anything.  */
500   if (start == peeked && 0 != memcmp (start, "HTTP", MIN (peeklen, 4)))
501     return start;
502
503   /* Look for "\n[\r]\n", and return the following position if found.
504      Start two chars before the current to cover the possibility that
505      part of the terminator (e.g. "\n\r") arrived in the previous
506      batch.  */
507   p = peeked - start < 2 ? start : peeked - 2;
508   end = peeked + peeklen;
509
510   /* Check for \n\r\n or \n\n anywhere in [p, end-2). */
511   for (; p < end - 2; p++)
512     if (*p == '\n')
513       {
514         if (p[1] == '\r' && p[2] == '\n')
515           return p + 3;
516         else if (p[1] == '\n')
517           return p + 2;
518       }
519   /* p==end-2: check for \n\n directly preceding END. */
520   if (p[0] == '\n' && p[1] == '\n')
521     return p + 2;
522
523   return NULL;
524 }
525
526 /* The maximum size of a single HTTP response we care to read.  Rather
527    than being a limit of the reader implementation, this limit
528    prevents Wget from slurping all available memory upon encountering
529    malicious or buggy server output, thus protecting the user.  Define
530    it to 0 to remove the limit.  */
531
532 #define HTTP_RESPONSE_MAX_SIZE 65536
533
534 /* Read the HTTP request head from FD and return it.  The error
535    conditions are the same as with fd_read_hunk.
536
537    To support HTTP/0.9 responses, this function tries to make sure
538    that the data begins with "HTTP".  If this is not the case, no data
539    is read and an empty request is returned, so that the remaining
540    data can be treated as body.  */
541
542 static char *
543 read_http_response_head (int fd)
544 {
545   return fd_read_hunk (fd, response_head_terminator, 512,
546                        HTTP_RESPONSE_MAX_SIZE);
547 }
548
549 struct response {
550   /* The response data. */
551   const char *data;
552
553   /* The array of pointers that indicate where each header starts.
554      For example, given this HTTP response:
555
556        HTTP/1.0 200 Ok
557        Description: some
558         text
559        Etag: x
560
561      The headers are located like this:
562
563      "HTTP/1.0 200 Ok\r\nDescription: some\r\n text\r\nEtag: x\r\n\r\n"
564      ^                   ^                             ^          ^
565      headers[0]          headers[1]                    headers[2] headers[3]
566
567      I.e. headers[0] points to the beginning of the request,
568      headers[1] points to the end of the first header and the
569      beginning of the second one, etc.  */
570
571   const char **headers;
572 };
573
574 /* Create a new response object from the text of the HTTP response,
575    available in HEAD.  That text is automatically split into
576    constituent header lines for fast retrieval using
577    resp_header_*.  */
578
579 static struct response *
580 resp_new (const char *head)
581 {
582   const char *hdr;
583   int count, size;
584
585   struct response *resp = xnew0 (struct response);
586   resp->data = head;
587
588   if (*head == '\0')
589     {
590       /* Empty head means that we're dealing with a headerless
591          (HTTP/0.9) response.  In that case, don't set HEADERS at
592          all.  */
593       return resp;
594     }
595
596   /* Split HEAD into header lines, so that resp_header_* functions
597      don't need to do this over and over again.  */
598
599   size = count = 0;
600   hdr = head;
601   while (1)
602     {
603       DO_REALLOC (resp->headers, size, count + 1, const char *);
604       resp->headers[count++] = hdr;
605
606       /* Break upon encountering an empty line. */
607       if (!hdr[0] || (hdr[0] == '\r' && hdr[1] == '\n') || hdr[0] == '\n')
608         break;
609
610       /* Find the end of HDR, including continuations. */
611       do
612         {
613           const char *end = strchr (hdr, '\n');
614           if (end)
615             hdr = end + 1;
616           else
617             hdr += strlen (hdr);
618         }
619       while (*hdr == ' ' || *hdr == '\t');
620     }
621   DO_REALLOC (resp->headers, size, count + 1, const char *);
622   resp->headers[count] = NULL;
623
624   return resp;
625 }
626
627 /* Locate the header named NAME in the request data, starting with
628    position START.  This allows the code to loop through the request
629    data, filtering for all requests of a given name.  Returns the
630    found position, or -1 for failure.  The code that uses this
631    function typically looks like this:
632
633      for (pos = 0; (pos = resp_header_locate (...)) != -1; pos++)
634        ... do something with header ...
635
636    If you only care about one header, use resp_header_get instead of
637    this function.  */
638
639 static int
640 resp_header_locate (const struct response *resp, const char *name, int start,
641                     const char **begptr, const char **endptr)
642 {
643   int i;
644   const char **headers = resp->headers;
645   int name_len;
646
647   if (!headers || !headers[1])
648     return -1;
649
650   name_len = strlen (name);
651   if (start > 0)
652     i = start;
653   else
654     i = 1;
655
656   for (; headers[i + 1]; i++)
657     {
658       const char *b = headers[i];
659       const char *e = headers[i + 1];
660       if (e - b > name_len
661           && b[name_len] == ':'
662           && 0 == strncasecmp (b, name, name_len))
663         {
664           b += name_len + 1;
665           while (b < e && c_isspace (*b))
666             ++b;
667           while (b < e && c_isspace (e[-1]))
668             --e;
669           *begptr = b;
670           *endptr = e;
671           return i;
672         }
673     }
674   return -1;
675 }
676
677 /* Find and retrieve the header named NAME in the request data.  If
678    found, set *BEGPTR to its starting, and *ENDPTR to its ending
679    position, and return true.  Otherwise return false.
680
681    This function is used as a building block for resp_header_copy
682    and resp_header_strdup.  */
683
684 static bool
685 resp_header_get (const struct response *resp, const char *name,
686                  const char **begptr, const char **endptr)
687 {
688   int pos = resp_header_locate (resp, name, 0, begptr, endptr);
689   return pos != -1;
690 }
691
692 /* Copy the response header named NAME to buffer BUF, no longer than
693    BUFSIZE (BUFSIZE includes the terminating 0).  If the header
694    exists, true is returned, false otherwise.  If there should be no
695    limit on the size of the header, use resp_header_strdup instead.
696
697    If BUFSIZE is 0, no data is copied, but the boolean indication of
698    whether the header is present is still returned.  */
699
700 static bool
701 resp_header_copy (const struct response *resp, const char *name,
702                   char *buf, int bufsize)
703 {
704   const char *b, *e;
705   if (!resp_header_get (resp, name, &b, &e))
706     return false;
707   if (bufsize)
708     {
709       int len = MIN (e - b, bufsize - 1);
710       memcpy (buf, b, len);
711       buf[len] = '\0';
712     }
713   return true;
714 }
715
716 /* Return the value of header named NAME in RESP, allocated with
717    malloc.  If such a header does not exist in RESP, return NULL.  */
718
719 static char *
720 resp_header_strdup (const struct response *resp, const char *name)
721 {
722   const char *b, *e;
723   if (!resp_header_get (resp, name, &b, &e))
724     return NULL;
725   return strdupdelim (b, e);
726 }
727
728 /* Parse the HTTP status line, which is of format:
729
730    HTTP-Version SP Status-Code SP Reason-Phrase
731
732    The function returns the status-code, or -1 if the status line
733    appears malformed.  The pointer to "reason-phrase" message is
734    returned in *MESSAGE.  */
735
736 static int
737 resp_status (const struct response *resp, char **message)
738 {
739   int status;
740   const char *p, *end;
741
742   if (!resp->headers)
743     {
744       /* For a HTTP/0.9 response, assume status 200. */
745       if (message)
746         *message = xstrdup (_("No headers, assuming HTTP/0.9"));
747       return 200;
748     }
749
750   p = resp->headers[0];
751   end = resp->headers[1];
752
753   if (!end)
754     return -1;
755
756   /* "HTTP" */
757   if (end - p < 4 || 0 != strncmp (p, "HTTP", 4))
758     return -1;
759   p += 4;
760
761   /* Match the HTTP version.  This is optional because Gnutella
762      servers have been reported to not specify HTTP version.  */
763   if (p < end && *p == '/')
764     {
765       ++p;
766       while (p < end && c_isdigit (*p))
767         ++p;
768       if (p < end && *p == '.')
769         ++p; 
770       while (p < end && c_isdigit (*p))
771         ++p;
772     }
773
774   while (p < end && c_isspace (*p))
775     ++p;
776   if (end - p < 3 || !c_isdigit (p[0]) || !c_isdigit (p[1]) || !c_isdigit (p[2]))
777     return -1;
778
779   status = 100 * (p[0] - '0') + 10 * (p[1] - '0') + (p[2] - '0');
780   p += 3;
781
782   if (message)
783     {
784       while (p < end && c_isspace (*p))
785         ++p;
786       while (p < end && c_isspace (end[-1]))
787         --end;
788       *message = strdupdelim (p, end);
789     }
790
791   return status;
792 }
793
794 /* Release the resources used by RESP.  */
795
796 static void
797 resp_free (struct response *resp)
798 {
799   xfree_null (resp->headers);
800   xfree (resp);
801 }
802
803 /* Print a single line of response, the characters [b, e).  We tried
804    getting away with
805       logprintf (LOG_VERBOSE, "%s%.*s\n", prefix, (int) (e - b), b);
806    but that failed to escape the non-printable characters and, in fact,
807    caused crashes in UTF-8 locales.  */
808
809 static void
810 print_response_line(const char *prefix, const char *b, const char *e)
811 {
812   char *copy;
813   BOUNDED_TO_ALLOCA(b, e, copy);
814   logprintf (LOG_ALWAYS, "%s%s\n", prefix, 
815              quotearg_style (escape_quoting_style, copy));
816 }
817
818 /* Print the server response, line by line, omitting the trailing CRLF
819    from individual header lines, and prefixed with PREFIX.  */
820
821 static void
822 print_server_response (const struct response *resp, const char *prefix)
823 {
824   int i;
825   if (!resp->headers)
826     return;
827   for (i = 0; resp->headers[i + 1]; i++)
828     {
829       const char *b = resp->headers[i];
830       const char *e = resp->headers[i + 1];
831       /* Skip CRLF */
832       if (b < e && e[-1] == '\n')
833         --e;
834       if (b < e && e[-1] == '\r')
835         --e;
836       print_response_line(prefix, b, e);
837     }
838 }
839
840 /* Parse the `Content-Range' header and extract the information it
841    contains.  Returns true if successful, false otherwise.  */
842 static bool
843 parse_content_range (const char *hdr, wgint *first_byte_ptr,
844                      wgint *last_byte_ptr, wgint *entity_length_ptr)
845 {
846   wgint num;
847
848   /* Ancient versions of Netscape proxy server, presumably predating
849      rfc2068, sent out `Content-Range' without the "bytes"
850      specifier.  */
851   if (0 == strncasecmp (hdr, "bytes", 5))
852     {
853       hdr += 5;
854       /* "JavaWebServer/1.1.1" sends "bytes: x-y/z", contrary to the
855          HTTP spec. */
856       if (*hdr == ':')
857         ++hdr;
858       while (c_isspace (*hdr))
859         ++hdr;
860       if (!*hdr)
861         return false;
862     }
863   if (!c_isdigit (*hdr))
864     return false;
865   for (num = 0; c_isdigit (*hdr); hdr++)
866     num = 10 * num + (*hdr - '0');
867   if (*hdr != '-' || !c_isdigit (*(hdr + 1)))
868     return false;
869   *first_byte_ptr = num;
870   ++hdr;
871   for (num = 0; c_isdigit (*hdr); hdr++)
872     num = 10 * num + (*hdr - '0');
873   if (*hdr != '/' || !c_isdigit (*(hdr + 1)))
874     return false;
875   *last_byte_ptr = num;
876   ++hdr;
877   if (*hdr == '*')
878     num = -1;
879   else
880     for (num = 0; c_isdigit (*hdr); hdr++)
881       num = 10 * num + (*hdr - '0');
882   *entity_length_ptr = num;
883   return true;
884 }
885
886 /* Read the body of the request, but don't store it anywhere and don't
887    display a progress gauge.  This is useful for reading the bodies of
888    administrative responses to which we will soon issue another
889    request.  The response is not useful to the user, but reading it
890    allows us to continue using the same connection to the server.
891
892    If reading fails, false is returned, true otherwise.  In debug
893    mode, the body is displayed for debugging purposes.  */
894
895 static bool
896 skip_short_body (int fd, wgint contlen)
897 {
898   enum {
899     SKIP_SIZE = 512,                /* size of the download buffer */
900     SKIP_THRESHOLD = 4096        /* the largest size we read */
901   };
902   char dlbuf[SKIP_SIZE + 1];
903   dlbuf[SKIP_SIZE] = '\0';        /* so DEBUGP can safely print it */
904
905   /* We shouldn't get here with unknown contlen.  (This will change
906      with HTTP/1.1, which supports "chunked" transfer.)  */
907   assert (contlen != -1);
908
909   /* If the body is too large, it makes more sense to simply close the
910      connection than to try to read the body.  */
911   if (contlen > SKIP_THRESHOLD)
912     return false;
913
914   DEBUGP (("Skipping %s bytes of body: [", number_to_static_string (contlen)));
915
916   while (contlen > 0)
917     {
918       int ret = fd_read (fd, dlbuf, MIN (contlen, SKIP_SIZE), -1);
919       if (ret <= 0)
920         {
921           /* Don't normally report the error since this is an
922              optimization that should be invisible to the user.  */
923           DEBUGP (("] aborting (%s).\n",
924                    ret < 0 ? fd_errstr (fd) : "EOF received"));
925           return false;
926         }
927       contlen -= ret;
928       /* Safe even if %.*s bogusly expects terminating \0 because
929          we've zero-terminated dlbuf above.  */
930       DEBUGP (("%.*s", ret, dlbuf));
931     }
932
933   DEBUGP (("] done.\n"));
934   return true;
935 }
936
937 /* Extract a parameter from the string (typically an HTTP header) at
938    **SOURCE and advance SOURCE to the next parameter.  Return false
939    when there are no more parameters to extract.  The name of the
940    parameter is returned in NAME, and the value in VALUE.  If the
941    parameter has no value, the token's value is zeroed out.
942
943    For example, if *SOURCE points to the string "attachment;
944    filename=\"foo bar\"", the first call to this function will return
945    the token named "attachment" and no value, and the second call will
946    return the token named "filename" and value "foo bar".  The third
947    call will return false, indicating no more valid tokens.  */
948
949 bool
950 extract_param (const char **source, param_token *name, param_token *value,
951                char separator)
952 {
953   const char *p = *source;
954
955   while (c_isspace (*p)) ++p;
956   if (!*p)
957     {
958       *source = p;
959       return false;             /* no error; nothing more to extract */
960     }
961
962   /* Extract name. */
963   name->b = p;
964   while (*p && !c_isspace (*p) && *p != '=' && *p != separator) ++p;
965   name->e = p;
966   if (name->b == name->e)
967     return false;               /* empty name: error */
968   while (c_isspace (*p)) ++p;
969   if (*p == separator || !*p)           /* no value */
970     {
971       xzero (*value);
972       if (*p == separator) ++p;
973       *source = p;
974       return true;
975     }
976   if (*p != '=')
977     return false;               /* error */
978
979   /* *p is '=', extract value */
980   ++p;
981   while (c_isspace (*p)) ++p;
982   if (*p == '"')                /* quoted */
983     {
984       value->b = ++p;
985       while (*p && *p != '"') ++p;
986       if (!*p)
987         return false;
988       value->e = p++;
989       /* Currently at closing quote; find the end of param. */
990       while (c_isspace (*p)) ++p;
991       while (*p && *p != separator) ++p;
992       if (*p == separator)
993         ++p;
994       else if (*p)
995         /* garbage after closed quote, e.g. foo="bar"baz */
996         return false;
997     }
998   else                          /* unquoted */
999     {
1000       value->b = p;
1001       while (*p && *p != separator) ++p;
1002       value->e = p;
1003       while (value->e != value->b && c_isspace (value->e[-1]))
1004         --value->e;
1005       if (*p == separator) ++p;
1006     }
1007   *source = p;
1008   return true;
1009 }
1010
1011 #undef MAX
1012 #define MAX(p, q) ((p) > (q) ? (p) : (q))
1013
1014 /* Parse the contents of the `Content-Disposition' header, extracting
1015    the information useful to Wget.  Content-Disposition is a header
1016    borrowed from MIME; when used in HTTP, it typically serves for
1017    specifying the desired file name of the resource.  For example:
1018
1019        Content-Disposition: attachment; filename="flora.jpg"
1020
1021    Wget will skip the tokens it doesn't care about, such as
1022    "attachment" in the previous example; it will also skip other
1023    unrecognized params.  If the header is syntactically correct and
1024    contains a file name, a copy of the file name is stored in
1025    *filename and true is returned.  Otherwise, the function returns
1026    false.
1027
1028    The file name is stripped of directory components and must not be
1029    empty.  */
1030
1031 static bool
1032 parse_content_disposition (const char *hdr, char **filename)
1033 {
1034   param_token name, value;
1035   while (extract_param (&hdr, &name, &value, ';'))
1036     if (BOUNDED_EQUAL_NO_CASE (name.b, name.e, "filename") && value.b != NULL)
1037       {
1038         /* Make the file name begin at the last slash or backslash. */
1039         const char *last_slash = memrchr (value.b, '/', value.e - value.b);
1040         const char *last_bs = memrchr (value.b, '\\', value.e - value.b);
1041         if (last_slash && last_bs)
1042           value.b = 1 + MAX (last_slash, last_bs);
1043         else if (last_slash || last_bs)
1044           value.b = 1 + (last_slash ? last_slash : last_bs);
1045         if (value.b == value.e)
1046           continue;
1047         /* Start with the directory prefix, if specified. */
1048         if (opt.dir_prefix)
1049           {
1050             int prefix_length = strlen (opt.dir_prefix);
1051             bool add_slash = (opt.dir_prefix[prefix_length - 1] != '/');
1052             int total_length;
1053
1054             if (add_slash) 
1055               ++prefix_length;
1056             total_length = prefix_length + (value.e - value.b);            
1057             *filename = xmalloc (total_length + 1);
1058             strcpy (*filename, opt.dir_prefix);
1059             if (add_slash) 
1060               (*filename)[prefix_length - 1] = '/';
1061             memcpy (*filename + prefix_length, value.b, (value.e - value.b));
1062             (*filename)[total_length] = '\0';
1063           }
1064         else
1065           *filename = strdupdelim (value.b, value.e);
1066         return true;
1067       }
1068   return false;
1069 }
1070 \f
1071 /* Persistent connections.  Currently, we cache the most recently used
1072    connection as persistent, provided that the HTTP server agrees to
1073    make it such.  The persistence data is stored in the variables
1074    below.  Ideally, it should be possible to cache an arbitrary fixed
1075    number of these connections.  */
1076
1077 /* Whether a persistent connection is active. */
1078 static bool pconn_active;
1079
1080 static struct {
1081   /* The socket of the connection.  */
1082   int socket;
1083
1084   /* Host and port of the currently active persistent connection. */
1085   char *host;
1086   int port;
1087
1088   /* Whether a ssl handshake has occoured on this connection.  */
1089   bool ssl;
1090
1091   /* Whether the connection was authorized.  This is only done by
1092      NTLM, which authorizes *connections* rather than individual
1093      requests.  (That practice is peculiar for HTTP, but it is a
1094      useful optimization.)  */
1095   bool authorized;
1096
1097 #ifdef ENABLE_NTLM
1098   /* NTLM data of the current connection.  */
1099   struct ntlmdata ntlm;
1100 #endif
1101 } pconn;
1102
1103 /* Mark the persistent connection as invalid and free the resources it
1104    uses.  This is used by the CLOSE_* macros after they forcefully
1105    close a registered persistent connection.  */
1106
1107 static void
1108 invalidate_persistent (void)
1109 {
1110   DEBUGP (("Disabling further reuse of socket %d.\n", pconn.socket));
1111   pconn_active = false;
1112   fd_close (pconn.socket);
1113   xfree (pconn.host);
1114   xzero (pconn);
1115 }
1116
1117 /* Register FD, which should be a TCP/IP connection to HOST:PORT, as
1118    persistent.  This will enable someone to use the same connection
1119    later.  In the context of HTTP, this must be called only AFTER the
1120    response has been received and the server has promised that the
1121    connection will remain alive.
1122
1123    If a previous connection was persistent, it is closed. */
1124
1125 static void
1126 register_persistent (const char *host, int port, int fd, bool ssl)
1127 {
1128   if (pconn_active)
1129     {
1130       if (pconn.socket == fd)
1131         {
1132           /* The connection FD is already registered. */
1133           return;
1134         }
1135       else
1136         {
1137           /* The old persistent connection is still active; close it
1138              first.  This situation arises whenever a persistent
1139              connection exists, but we then connect to a different
1140              host, and try to register a persistent connection to that
1141              one.  */
1142           invalidate_persistent ();
1143         }
1144     }
1145
1146   pconn_active = true;
1147   pconn.socket = fd;
1148   pconn.host = xstrdup (host);
1149   pconn.port = port;
1150   pconn.ssl = ssl;
1151   pconn.authorized = false;
1152
1153   DEBUGP (("Registered socket %d for persistent reuse.\n", fd));
1154 }
1155
1156 /* Return true if a persistent connection is available for connecting
1157    to HOST:PORT.  */
1158
1159 static bool
1160 persistent_available_p (const char *host, int port, bool ssl,
1161                         bool *host_lookup_failed)
1162 {
1163   /* First, check whether a persistent connection is active at all.  */
1164   if (!pconn_active)
1165     return false;
1166
1167   /* If we want SSL and the last connection wasn't or vice versa,
1168      don't use it.  Checking for host and port is not enough because
1169      HTTP and HTTPS can apparently coexist on the same port.  */
1170   if (ssl != pconn.ssl)
1171     return false;
1172
1173   /* If we're not connecting to the same port, we're not interested. */
1174   if (port != pconn.port)
1175     return false;
1176
1177   /* If the host is the same, we're in business.  If not, there is
1178      still hope -- read below.  */
1179   if (0 != strcasecmp (host, pconn.host))
1180     {
1181       /* Check if pconn.socket is talking to HOST under another name.
1182          This happens often when both sites are virtual hosts
1183          distinguished only by name and served by the same network
1184          interface, and hence the same web server (possibly set up by
1185          the ISP and serving many different web sites).  This
1186          admittedly unconventional optimization does not contradict
1187          HTTP and works well with popular server software.  */
1188
1189       bool found;
1190       ip_address ip;
1191       struct address_list *al;
1192
1193       if (ssl)
1194         /* Don't try to talk to two different SSL sites over the same
1195            secure connection!  (Besides, it's not clear that
1196            name-based virtual hosting is even possible with SSL.)  */
1197         return false;
1198
1199       /* If pconn.socket's peer is one of the IP addresses HOST
1200          resolves to, pconn.socket is for all intents and purposes
1201          already talking to HOST.  */
1202
1203       if (!socket_ip_address (pconn.socket, &ip, ENDPOINT_PEER))
1204         {
1205           /* Can't get the peer's address -- something must be very
1206              wrong with the connection.  */
1207           invalidate_persistent ();
1208           return false;
1209         }
1210       al = lookup_host (host, 0);
1211       if (!al)
1212         {
1213           *host_lookup_failed = true;
1214           return false;
1215         }
1216
1217       found = address_list_contains (al, &ip);
1218       address_list_release (al);
1219
1220       if (!found)
1221         return false;
1222
1223       /* The persistent connection's peer address was found among the
1224          addresses HOST resolved to; therefore, pconn.sock is in fact
1225          already talking to HOST -- no need to reconnect.  */
1226     }
1227
1228   /* Finally, check whether the connection is still open.  This is
1229      important because most servers implement liberal (short) timeout
1230      on persistent connections.  Wget can of course always reconnect
1231      if the connection doesn't work out, but it's nicer to know in
1232      advance.  This test is a logical followup of the first test, but
1233      is "expensive" and therefore placed at the end of the list.
1234
1235      (Current implementation of test_socket_open has a nice side
1236      effect that it treats sockets with pending data as "closed".
1237      This is exactly what we want: if a broken server sends message
1238      body in response to HEAD, or if it sends more than conent-length
1239      data, we won't reuse the corrupted connection.)  */
1240
1241   if (!test_socket_open (pconn.socket))
1242     {
1243       /* Oops, the socket is no longer open.  Now that we know that,
1244          let's invalidate the persistent connection before returning
1245          0.  */
1246       invalidate_persistent ();
1247       return false;
1248     }
1249
1250   return true;
1251 }
1252
1253 /* The idea behind these two CLOSE macros is to distinguish between
1254    two cases: one when the job we've been doing is finished, and we
1255    want to close the connection and leave, and two when something is
1256    seriously wrong and we're closing the connection as part of
1257    cleanup.
1258
1259    In case of keep_alive, CLOSE_FINISH should leave the connection
1260    open, while CLOSE_INVALIDATE should still close it.
1261
1262    Note that the semantics of the flag `keep_alive' is "this
1263    connection *will* be reused (the server has promised not to close
1264    the connection once we're done)", while the semantics of
1265    `pc_active_p && (fd) == pc_last_fd' is "we're *now* using an
1266    active, registered connection".  */
1267
1268 #define CLOSE_FINISH(fd) do {                   \
1269   if (!keep_alive)                              \
1270     {                                           \
1271       if (pconn_active && (fd) == pconn.socket) \
1272         invalidate_persistent ();               \
1273       else                                      \
1274         {                                       \
1275           fd_close (fd);                        \
1276           fd = -1;                              \
1277         }                                       \
1278     }                                           \
1279 } while (0)
1280
1281 #define CLOSE_INVALIDATE(fd) do {               \
1282   if (pconn_active && (fd) == pconn.socket)     \
1283     invalidate_persistent ();                   \
1284   else                                          \
1285     fd_close (fd);                              \
1286   fd = -1;                                      \
1287 } while (0)
1288 \f
1289 struct http_stat
1290 {
1291   wgint len;                    /* received length */
1292   wgint contlen;                /* expected length */
1293   wgint restval;                /* the restart value */
1294   int res;                      /* the result of last read */
1295   char *rderrmsg;               /* error message from read error */
1296   char *newloc;                 /* new location (redirection) */
1297   char *remote_time;            /* remote time-stamp string */
1298   char *error;                  /* textual HTTP error */
1299   int statcode;                 /* status code */
1300   wgint rd_size;                /* amount of data read from socket */
1301   double dltime;                /* time it took to download the data */
1302   const char *referer;          /* value of the referer header. */
1303   char *local_file;             /* local file name. */
1304   bool existence_checked;       /* true if we already checked for a file's
1305                                    existence after having begun to download
1306                                    (needed in gethttp for when connection is
1307                                    interrupted/restarted. */
1308   bool timestamp_checked;       /* true if pre-download time-stamping checks 
1309                                  * have already been performed */
1310   char *orig_file_name;         /* name of file to compare for time-stamping
1311                                  * (might be != local_file if -K is set) */
1312   wgint orig_file_size;         /* size of file to compare for time-stamping */
1313   time_t orig_file_tstamp;      /* time-stamp of file to compare for 
1314                                  * time-stamping */
1315 };
1316
1317 static void
1318 free_hstat (struct http_stat *hs)
1319 {
1320   xfree_null (hs->newloc);
1321   xfree_null (hs->remote_time);
1322   xfree_null (hs->error);
1323   xfree_null (hs->rderrmsg);
1324   xfree_null (hs->local_file);
1325   xfree_null (hs->orig_file_name);
1326
1327   /* Guard against being called twice. */
1328   hs->newloc = NULL;
1329   hs->remote_time = NULL;
1330   hs->error = NULL;
1331 }
1332
1333 #define BEGINS_WITH(line, string_constant)                               \
1334   (!strncasecmp (line, string_constant, sizeof (string_constant) - 1)    \
1335    && (c_isspace (line[sizeof (string_constant) - 1])                      \
1336        || !line[sizeof (string_constant) - 1]))
1337
1338 #define SET_USER_AGENT(req) do {                                         \
1339   if (!opt.useragent)                                                    \
1340     request_set_header (req, "User-Agent",                               \
1341                         aprintf ("Wget/%s", version_string), rel_value); \
1342   else if (*opt.useragent)                                               \
1343     request_set_header (req, "User-Agent", opt.useragent, rel_none);     \
1344 } while (0)
1345
1346 /* The flags that allow clobbering the file (opening with "wb").
1347    Defined here to avoid repetition later.  #### This will require
1348    rework.  */
1349 #define ALLOW_CLOBBER (opt.noclobber || opt.always_rest || opt.timestamping \
1350                        || opt.dirstruct || opt.output_document)
1351
1352 /* Retrieve a document through HTTP protocol.  It recognizes status
1353    code, and correctly handles redirections.  It closes the network
1354    socket.  If it receives an error from the functions below it, it
1355    will print it if there is enough information to do so (almost
1356    always), returning the error to the caller (i.e. http_loop).
1357
1358    Various HTTP parameters are stored to hs.
1359
1360    If PROXY is non-NULL, the connection will be made to the proxy
1361    server, and u->url will be requested.  */
1362 static uerr_t
1363 gethttp (struct url *u, struct http_stat *hs, int *dt, struct url *proxy)
1364 {
1365   struct request *req;
1366
1367   char *type;
1368   char *user, *passwd;
1369   char *proxyauth;
1370   int statcode;
1371   int write_error;
1372   wgint contlen, contrange;
1373   struct url *conn;
1374   FILE *fp;
1375
1376   int sock = -1;
1377   int flags;
1378
1379   /* Set to 1 when the authorization has already been sent and should
1380      not be tried again. */
1381   bool auth_finished = false;
1382
1383   /* Set to 1 when just globally-set Basic authorization has been sent;
1384    * should prevent further Basic negotiations, but not other
1385    * mechanisms. */
1386   bool basic_auth_finished = false;
1387
1388   /* Whether NTLM authentication is used for this request. */
1389   bool ntlm_seen = false;
1390
1391   /* Whether our connection to the remote host is through SSL.  */
1392   bool using_ssl = false;
1393
1394   /* Whether a HEAD request will be issued (as opposed to GET or
1395      POST). */
1396   bool head_only = !!(*dt & HEAD_ONLY);
1397
1398   char *head;
1399   struct response *resp;
1400   char hdrval[256];
1401   char *message;
1402
1403   /* Whether this connection will be kept alive after the HTTP request
1404      is done. */
1405   bool keep_alive;
1406
1407   /* Whether keep-alive should be inhibited.
1408
1409      RFC 2068 requests that 1.0 clients not send keep-alive requests
1410      to proxies.  This is because many 1.0 proxies do not interpret
1411      the Connection header and transfer it to the remote server,
1412      causing it to not close the connection and leave both the proxy
1413      and the client hanging.  */
1414   bool inhibit_keep_alive =
1415     !opt.http_keep_alive || opt.ignore_length || proxy != NULL;
1416
1417   /* Headers sent when using POST. */
1418   wgint post_data_size = 0;
1419
1420   bool host_lookup_failed = false;
1421
1422 #ifdef HAVE_SSL
1423   if (u->scheme == SCHEME_HTTPS)
1424     {
1425       /* Initialize the SSL context.  After this has once been done,
1426          it becomes a no-op.  */
1427       if (!ssl_init ())
1428         {
1429           scheme_disable (SCHEME_HTTPS);
1430           logprintf (LOG_NOTQUIET,
1431                      _("Disabling SSL due to encountered errors.\n"));
1432           return SSLINITFAILED;
1433         }
1434     }
1435 #endif /* HAVE_SSL */
1436
1437   /* Initialize certain elements of struct http_stat.  */
1438   hs->len = 0;
1439   hs->contlen = -1;
1440   hs->res = -1;
1441   hs->rderrmsg = NULL;
1442   hs->newloc = NULL;
1443   hs->remote_time = NULL;
1444   hs->error = NULL;
1445
1446   conn = u;
1447
1448   /* Prepare the request to send. */
1449
1450   req = request_new ();
1451   {
1452     char *meth_arg;
1453     const char *meth = "GET";
1454     if (head_only)
1455       meth = "HEAD";
1456     else if (opt.post_file_name || opt.post_data)
1457       meth = "POST";
1458     /* Use the full path, i.e. one that includes the leading slash and
1459        the query string.  E.g. if u->path is "foo/bar" and u->query is
1460        "param=value", full_path will be "/foo/bar?param=value".  */
1461     if (proxy
1462 #ifdef HAVE_SSL
1463         /* When using SSL over proxy, CONNECT establishes a direct
1464            connection to the HTTPS server.  Therefore use the same
1465            argument as when talking to the server directly. */
1466         && u->scheme != SCHEME_HTTPS
1467 #endif
1468         )
1469       meth_arg = xstrdup (u->url);
1470     else
1471       meth_arg = url_full_path (u);
1472     request_set_method (req, meth, meth_arg);
1473   }
1474
1475   request_set_header (req, "Referer", (char *) hs->referer, rel_none);
1476   if (*dt & SEND_NOCACHE)
1477     request_set_header (req, "Pragma", "no-cache", rel_none);
1478   if (hs->restval)
1479     request_set_header (req, "Range",
1480                         aprintf ("bytes=%s-",
1481                                  number_to_static_string (hs->restval)),
1482                         rel_value);
1483   SET_USER_AGENT (req);
1484   request_set_header (req, "Accept", "*/*", rel_none);
1485
1486   /* Find the username and password for authentication. */
1487   user = u->user;
1488   passwd = u->passwd;
1489   search_netrc (u->host, (const char **)&user, (const char **)&passwd, 0);
1490   user = user ? user : (opt.http_user ? opt.http_user : opt.user);
1491   passwd = passwd ? passwd : (opt.http_passwd ? opt.http_passwd : opt.passwd);
1492
1493   if (user && passwd
1494       && !u->user) /* We only do "site-wide" authentication with "global"
1495                       user/password values; URL user/password info overrides. */
1496     {
1497       /* If this is a host for which we've already received a Basic
1498        * challenge, we'll go ahead and send Basic authentication creds. */
1499       basic_auth_finished = maybe_send_basic_creds(u->host, user, passwd, req);
1500     }
1501
1502   /* Generate the Host header, HOST:PORT.  Take into account that:
1503
1504      - Broken server-side software often doesn't recognize the PORT
1505        argument, so we must generate "Host: www.server.com" instead of
1506        "Host: www.server.com:80" (and likewise for https port).
1507
1508      - IPv6 addresses contain ":", so "Host: 3ffe:8100:200:2::2:1234"
1509        becomes ambiguous and needs to be rewritten as "Host:
1510        [3ffe:8100:200:2::2]:1234".  */
1511   {
1512     /* Formats arranged for hfmt[add_port][add_squares].  */
1513     static const char *hfmt[][2] = {
1514       { "%s", "[%s]" }, { "%s:%d", "[%s]:%d" }
1515     };
1516     int add_port = u->port != scheme_default_port (u->scheme);
1517     int add_squares = strchr (u->host, ':') != NULL;
1518     request_set_header (req, "Host",
1519                         aprintf (hfmt[add_port][add_squares], u->host, u->port),
1520                         rel_value);
1521   }
1522
1523   if (!inhibit_keep_alive)
1524     request_set_header (req, "Connection", "Keep-Alive", rel_none);
1525
1526   if (opt.cookies)
1527     request_set_header (req, "Cookie",
1528                         cookie_header (wget_cookie_jar,
1529                                        u->host, u->port, u->path,
1530 #ifdef HAVE_SSL
1531                                        u->scheme == SCHEME_HTTPS
1532 #else
1533                                        0
1534 #endif
1535                                        ),
1536                         rel_value);
1537
1538   if (opt.post_data || opt.post_file_name)
1539     {
1540       request_set_header (req, "Content-Type",
1541                           "application/x-www-form-urlencoded", rel_none);
1542       if (opt.post_data)
1543         post_data_size = strlen (opt.post_data);
1544       else
1545         {
1546           post_data_size = file_size (opt.post_file_name);
1547           if (post_data_size == -1)
1548             {
1549               logprintf (LOG_NOTQUIET, _("POST data file %s missing: %s\n"),
1550                          quote (opt.post_file_name), strerror (errno));
1551               post_data_size = 0;
1552             }
1553         }
1554       request_set_header (req, "Content-Length",
1555                           xstrdup (number_to_static_string (post_data_size)),
1556                           rel_value);
1557     }
1558
1559   /* Add the user headers. */
1560   if (opt.user_headers)
1561     {
1562       int i;
1563       for (i = 0; opt.user_headers[i]; i++)
1564         request_set_user_header (req, opt.user_headers[i]);
1565     }
1566
1567  retry_with_auth:
1568   /* We need to come back here when the initial attempt to retrieve
1569      without authorization header fails.  (Expected to happen at least
1570      for the Digest authorization scheme.)  */
1571
1572   proxyauth = NULL;
1573   if (proxy)
1574     {
1575       char *proxy_user, *proxy_passwd;
1576       /* For normal username and password, URL components override
1577          command-line/wgetrc parameters.  With proxy
1578          authentication, it's the reverse, because proxy URLs are
1579          normally the "permanent" ones, so command-line args
1580          should take precedence.  */
1581       if (opt.proxy_user && opt.proxy_passwd)
1582         {
1583           proxy_user = opt.proxy_user;
1584           proxy_passwd = opt.proxy_passwd;
1585         }
1586       else
1587         {
1588           proxy_user = proxy->user;
1589           proxy_passwd = proxy->passwd;
1590         }
1591       /* #### This does not appear right.  Can't the proxy request,
1592          say, `Digest' authentication?  */
1593       if (proxy_user && proxy_passwd)
1594         proxyauth = basic_authentication_encode (proxy_user, proxy_passwd);
1595
1596       /* If we're using a proxy, we will be connecting to the proxy
1597          server.  */
1598       conn = proxy;
1599
1600       /* Proxy authorization over SSL is handled below. */
1601 #ifdef HAVE_SSL
1602       if (u->scheme != SCHEME_HTTPS)
1603 #endif
1604         request_set_header (req, "Proxy-Authorization", proxyauth, rel_value);
1605     }
1606
1607   keep_alive = false;
1608
1609   /* Establish the connection.  */
1610
1611   if (!inhibit_keep_alive)
1612     {
1613       /* Look for a persistent connection to target host, unless a
1614          proxy is used.  The exception is when SSL is in use, in which
1615          case the proxy is nothing but a passthrough to the target
1616          host, registered as a connection to the latter.  */
1617       struct url *relevant = conn;
1618 #ifdef HAVE_SSL
1619       if (u->scheme == SCHEME_HTTPS)
1620         relevant = u;
1621 #endif
1622
1623       if (persistent_available_p (relevant->host, relevant->port,
1624 #ifdef HAVE_SSL
1625                                   relevant->scheme == SCHEME_HTTPS,
1626 #else
1627                                   0,
1628 #endif
1629                                   &host_lookup_failed))
1630         {
1631           sock = pconn.socket;
1632           using_ssl = pconn.ssl;
1633           logprintf (LOG_VERBOSE, _("Reusing existing connection to %s:%d.\n"),
1634                      quotearg_style (escape_quoting_style, pconn.host), 
1635                      pconn.port);
1636           DEBUGP (("Reusing fd %d.\n", sock));
1637           if (pconn.authorized)
1638             /* If the connection is already authorized, the "Basic"
1639                authorization added by code above is unnecessary and
1640                only hurts us.  */
1641             request_remove_header (req, "Authorization");
1642         }
1643       else if (host_lookup_failed)
1644         {
1645           request_free (req);
1646           logprintf(LOG_NOTQUIET,
1647                     _("%s: unable to resolve host address %s\n"),
1648                     exec_name, quote (relevant->host));
1649           return HOSTERR;
1650         }
1651     }
1652
1653   if (sock < 0)
1654     {
1655       sock = connect_to_host (conn->host, conn->port);
1656       if (sock == E_HOST)
1657         {
1658           request_free (req);
1659           return HOSTERR;
1660         }
1661       else if (sock < 0)
1662         {
1663           request_free (req);
1664           return (retryable_socket_connect_error (errno)
1665                   ? CONERROR : CONIMPOSSIBLE);
1666         }
1667
1668 #ifdef HAVE_SSL
1669       if (proxy && u->scheme == SCHEME_HTTPS)
1670         {
1671           /* When requesting SSL URLs through proxies, use the
1672              CONNECT method to request passthrough.  */
1673           struct request *connreq = request_new ();
1674           request_set_method (connreq, "CONNECT",
1675                               aprintf ("%s:%d", u->host, u->port));
1676           SET_USER_AGENT (connreq);
1677           if (proxyauth)
1678             {
1679               request_set_header (connreq, "Proxy-Authorization",
1680                                   proxyauth, rel_value);
1681               /* Now that PROXYAUTH is part of the CONNECT request,
1682                  zero it out so we don't send proxy authorization with
1683                  the regular request below.  */
1684               proxyauth = NULL;
1685             }
1686           /* Examples in rfc2817 use the Host header in CONNECT
1687              requests.  I don't see how that gains anything, given
1688              that the contents of Host would be exactly the same as
1689              the contents of CONNECT.  */
1690
1691           write_error = request_send (connreq, sock);
1692           request_free (connreq);
1693           if (write_error < 0)
1694             {
1695               CLOSE_INVALIDATE (sock);
1696               return WRITEFAILED;
1697             }
1698
1699           head = read_http_response_head (sock);
1700           if (!head)
1701             {
1702               logprintf (LOG_VERBOSE, _("Failed reading proxy response: %s\n"),
1703                          fd_errstr (sock));
1704               CLOSE_INVALIDATE (sock);
1705               return HERR;
1706             }
1707           message = NULL;
1708           if (!*head)
1709             {
1710               xfree (head);
1711               goto failed_tunnel;
1712             }
1713           DEBUGP (("proxy responded with: [%s]\n", head));
1714
1715           resp = resp_new (head);
1716           statcode = resp_status (resp, &message);
1717           resp_free (resp);
1718           xfree (head);
1719           if (statcode != 200)
1720             {
1721             failed_tunnel:
1722               logprintf (LOG_NOTQUIET, _("Proxy tunneling failed: %s"),
1723                          message ? quotearg_style (escape_quoting_style, message) : "?");
1724               xfree_null (message);
1725               return CONSSLERR;
1726             }
1727           xfree_null (message);
1728
1729           /* SOCK is now *really* connected to u->host, so update CONN
1730              to reflect this.  That way register_persistent will
1731              register SOCK as being connected to u->host:u->port.  */
1732           conn = u;
1733         }
1734
1735       if (conn->scheme == SCHEME_HTTPS)
1736         {
1737           if (!ssl_connect (sock) || !ssl_check_certificate (sock, u->host))
1738             {
1739               fd_close (sock);
1740               return CONSSLERR;
1741             }
1742           using_ssl = true;
1743         }
1744 #endif /* HAVE_SSL */
1745     }
1746
1747   /* Send the request to server.  */
1748   write_error = request_send (req, sock);
1749
1750   if (write_error >= 0)
1751     {
1752       if (opt.post_data)
1753         {
1754           DEBUGP (("[POST data: %s]\n", opt.post_data));
1755           write_error = fd_write (sock, opt.post_data, post_data_size, -1);
1756         }
1757       else if (opt.post_file_name && post_data_size != 0)
1758         write_error = post_file (sock, opt.post_file_name, post_data_size);
1759     }
1760
1761   if (write_error < 0)
1762     {
1763       CLOSE_INVALIDATE (sock);
1764       request_free (req);
1765       return WRITEFAILED;
1766     }
1767   logprintf (LOG_VERBOSE, _("%s request sent, awaiting response... "),
1768              proxy ? "Proxy" : "HTTP");
1769   contlen = -1;
1770   contrange = 0;
1771   *dt &= ~RETROKF;
1772
1773   head = read_http_response_head (sock);
1774   if (!head)
1775     {
1776       if (errno == 0)
1777         {
1778           logputs (LOG_NOTQUIET, _("No data received.\n"));
1779           CLOSE_INVALIDATE (sock);
1780           request_free (req);
1781           return HEOF;
1782         }
1783       else
1784         {
1785           logprintf (LOG_NOTQUIET, _("Read error (%s) in headers.\n"),
1786                      fd_errstr (sock));
1787           CLOSE_INVALIDATE (sock);
1788           request_free (req);
1789           return HERR;
1790         }
1791     }
1792   DEBUGP (("\n---response begin---\n%s---response end---\n", head));
1793
1794   resp = resp_new (head);
1795
1796   /* Check for status line.  */
1797   message = NULL;
1798   statcode = resp_status (resp, &message);
1799   if (!opt.server_response)
1800     logprintf (LOG_VERBOSE, "%2d %s\n", statcode,
1801                message ? quotearg_style (escape_quoting_style, message) : "");
1802   else
1803     {
1804       logprintf (LOG_VERBOSE, "\n");
1805       print_server_response (resp, "  ");
1806     }
1807
1808   /* Determine the local filename if needed. Notice that if -O is used 
1809    * hstat.local_file is set by http_loop to the argument of -O. */
1810   if (!hs->local_file)
1811     {
1812       /* Honor Content-Disposition whether possible. */
1813       if (!opt.content_disposition
1814           || !resp_header_copy (resp, "Content-Disposition", 
1815                                 hdrval, sizeof (hdrval))
1816           || !parse_content_disposition (hdrval, &hs->local_file))
1817         {
1818           /* The Content-Disposition header is missing or broken. 
1819            * Choose unique file name according to given URL. */
1820           hs->local_file = url_file_name (u);
1821         }
1822     }
1823   
1824   /* TODO: perform this check only once. */
1825   if (!hs->existence_checked && file_exists_p (hs->local_file))
1826     {
1827       if (opt.noclobber && !opt.output_document)
1828         {
1829           /* If opt.noclobber is turned on and file already exists, do not
1830              retrieve the file. But if the output_document was given, then this
1831              test was already done and the file didn't exist. Hence the !opt.output_document */
1832           logprintf (LOG_VERBOSE, _("\
1833 File %s already there; not retrieving.\n\n"), quote (hs->local_file));
1834           /* If the file is there, we suppose it's retrieved OK.  */
1835           *dt |= RETROKF;
1836
1837           /* #### Bogusness alert.  */
1838           /* If its suffix is "html" or "htm" or similar, assume text/html.  */
1839           if (has_html_suffix_p (hs->local_file))
1840             *dt |= TEXTHTML;
1841
1842           return RETRUNNEEDED;
1843         }
1844       else if (!ALLOW_CLOBBER)
1845         {
1846           char *unique = unique_name (hs->local_file, true);
1847           if (unique != hs->local_file)
1848             xfree (hs->local_file);
1849           hs->local_file = unique;
1850         }
1851     }
1852   hs->existence_checked = true;
1853
1854   /* Support timestamping */
1855   /* TODO: move this code out of gethttp. */
1856   if (opt.timestamping && !hs->timestamp_checked)
1857     {
1858       size_t filename_len = strlen (hs->local_file);
1859       char *filename_plus_orig_suffix = alloca (filename_len + sizeof (".orig"));
1860       bool local_dot_orig_file_exists = false;
1861       char *local_filename = NULL;
1862       struct_stat st;
1863
1864       if (opt.backup_converted)
1865         /* If -K is specified, we'll act on the assumption that it was specified
1866            last time these files were downloaded as well, and instead of just
1867            comparing local file X against server file X, we'll compare local
1868            file X.orig (if extant, else X) against server file X.  If -K
1869            _wasn't_ specified last time, or the server contains files called
1870            *.orig, -N will be back to not operating correctly with -k. */
1871         {
1872           /* Would a single s[n]printf() call be faster?  --dan
1873
1874              Definitely not.  sprintf() is horribly slow.  It's a
1875              different question whether the difference between the two
1876              affects a program.  Usually I'd say "no", but at one
1877              point I profiled Wget, and found that a measurable and
1878              non-negligible amount of time was lost calling sprintf()
1879              in url.c.  Replacing sprintf with inline calls to
1880              strcpy() and number_to_string() made a difference.
1881              --hniksic */
1882           memcpy (filename_plus_orig_suffix, hs->local_file, filename_len);
1883           memcpy (filename_plus_orig_suffix + filename_len,
1884                   ".orig", sizeof (".orig"));
1885
1886           /* Try to stat() the .orig file. */
1887           if (stat (filename_plus_orig_suffix, &st) == 0)
1888             {
1889               local_dot_orig_file_exists = true;
1890               local_filename = filename_plus_orig_suffix;
1891             }
1892         }      
1893
1894       if (!local_dot_orig_file_exists)
1895         /* Couldn't stat() <file>.orig, so try to stat() <file>. */
1896         if (stat (hs->local_file, &st) == 0)
1897           local_filename = hs->local_file;
1898
1899       if (local_filename != NULL)
1900         /* There was a local file, so we'll check later to see if the version
1901            the server has is the same version we already have, allowing us to
1902            skip a download. */
1903         {
1904           hs->orig_file_name = xstrdup (local_filename);
1905           hs->orig_file_size = st.st_size;
1906           hs->orig_file_tstamp = st.st_mtime;
1907 #ifdef WINDOWS
1908           /* Modification time granularity is 2 seconds for Windows, so
1909              increase local time by 1 second for later comparison. */
1910           ++hs->orig_file_tstamp;
1911 #endif
1912         }
1913     }
1914
1915   if (!opt.ignore_length
1916       && resp_header_copy (resp, "Content-Length", hdrval, sizeof (hdrval)))
1917     {
1918       wgint parsed;
1919       errno = 0;
1920       parsed = str_to_wgint (hdrval, NULL, 10);
1921       if (parsed == WGINT_MAX && errno == ERANGE)
1922         {
1923           /* Out of range.
1924              #### If Content-Length is out of range, it most likely
1925              means that the file is larger than 2G and that we're
1926              compiled without LFS.  In that case we should probably
1927              refuse to even attempt to download the file.  */
1928           contlen = -1;
1929         }
1930       else if (parsed < 0)
1931         {
1932           /* Negative Content-Length; nonsensical, so we can't
1933              assume any information about the content to receive. */
1934           contlen = -1;
1935         }
1936       else
1937         contlen = parsed;
1938     }
1939
1940   /* Check for keep-alive related responses. */
1941   if (!inhibit_keep_alive && contlen != -1)
1942     {
1943       if (resp_header_copy (resp, "Keep-Alive", NULL, 0))
1944         keep_alive = true;
1945       else if (resp_header_copy (resp, "Connection", hdrval, sizeof (hdrval)))
1946         {
1947           if (0 == strcasecmp (hdrval, "Keep-Alive"))
1948             keep_alive = true;
1949         }
1950     }
1951   if (keep_alive)
1952     /* The server has promised that it will not close the connection
1953        when we're done.  This means that we can register it.  */
1954     register_persistent (conn->host, conn->port, sock, using_ssl);
1955
1956   if (statcode == HTTP_STATUS_UNAUTHORIZED)
1957     {
1958       /* Authorization is required.  */
1959       if (keep_alive && !head_only && skip_short_body (sock, contlen))
1960         CLOSE_FINISH (sock);
1961       else
1962         CLOSE_INVALIDATE (sock);
1963       pconn.authorized = false;
1964       if (!auth_finished && (user && passwd))
1965         {
1966           /* IIS sends multiple copies of WWW-Authenticate, one with
1967              the value "negotiate", and other(s) with data.  Loop over
1968              all the occurrences and pick the one we recognize.  */
1969           int wapos;
1970           const char *wabeg, *waend;
1971           char *www_authenticate = NULL;
1972           for (wapos = 0;
1973                (wapos = resp_header_locate (resp, "WWW-Authenticate", wapos,
1974                                             &wabeg, &waend)) != -1;
1975                ++wapos)
1976             if (known_authentication_scheme_p (wabeg, waend))
1977               {
1978                 BOUNDED_TO_ALLOCA (wabeg, waend, www_authenticate);
1979                 break;
1980               }
1981
1982           if (!www_authenticate)
1983             {
1984               /* If the authentication header is missing or
1985                  unrecognized, there's no sense in retrying.  */
1986               logputs (LOG_NOTQUIET, _("Unknown authentication scheme.\n"));
1987             }
1988           else if (!basic_auth_finished
1989                    || !BEGINS_WITH (www_authenticate, "Basic"))
1990             {
1991               char *pth;
1992               pth = url_full_path (u);
1993               request_set_header (req, "Authorization",
1994                                   create_authorization_line (www_authenticate,
1995                                                              user, passwd,
1996                                                              request_method (req),
1997                                                              pth,
1998                                                              &auth_finished),
1999                                   rel_value);
2000               if (BEGINS_WITH (www_authenticate, "NTLM"))
2001                 ntlm_seen = true;
2002               else if (!u->user && BEGINS_WITH (www_authenticate, "Basic"))
2003                 {
2004                   /* Need to register this host as using basic auth,
2005                    * so we automatically send creds next time. */
2006                   register_basic_auth_host (u->host);
2007                 }
2008               xfree (pth);
2009               goto retry_with_auth;
2010             }
2011           else
2012             {
2013               /* We already did Basic auth, and it failed. Gotta
2014                * give up. */
2015             }
2016         }
2017       logputs (LOG_NOTQUIET, _("Authorization failed.\n"));
2018       request_free (req);
2019       return AUTHFAILED;
2020     }
2021   else /* statcode != HTTP_STATUS_UNAUTHORIZED */
2022     {
2023       /* Kludge: if NTLM is used, mark the TCP connection as authorized. */
2024       if (ntlm_seen)
2025         pconn.authorized = true;
2026     }
2027   request_free (req);
2028
2029   hs->statcode = statcode;
2030   if (statcode == -1)
2031     hs->error = xstrdup (_("Malformed status line"));
2032   else if (!*message)
2033     hs->error = xstrdup (_("(no description)"));
2034   else
2035     hs->error = xstrdup (message);
2036   xfree_null (message);
2037
2038   type = resp_header_strdup (resp, "Content-Type");
2039   if (type)
2040     {
2041       char *tmp = strchr (type, ';');
2042       if (tmp)
2043         {
2044           /* sXXXav: only needed if IRI support is enabled */
2045           char *tmp2 = tmp + 1;
2046
2047           while (tmp > type && c_isspace (tmp[-1]))
2048             --tmp;
2049           *tmp = '\0';
2050
2051           /* Try to get remote encoding if needed */
2052           if (opt.enable_iri && !opt.encoding_remote)
2053             /* xxx = */ parse_charset (tmp2);
2054         }
2055     }
2056   hs->newloc = resp_header_strdup (resp, "Location");
2057   hs->remote_time = resp_header_strdup (resp, "Last-Modified");
2058
2059   /* Handle (possibly multiple instances of) the Set-Cookie header. */
2060   if (opt.cookies)
2061     {
2062       int scpos;
2063       const char *scbeg, *scend;
2064       /* The jar should have been created by now. */
2065       assert (wget_cookie_jar != NULL);
2066       for (scpos = 0;
2067            (scpos = resp_header_locate (resp, "Set-Cookie", scpos,
2068                                         &scbeg, &scend)) != -1;
2069            ++scpos)
2070         {
2071           char *set_cookie; BOUNDED_TO_ALLOCA (scbeg, scend, set_cookie);
2072           cookie_handle_set_cookie (wget_cookie_jar, u->host, u->port,
2073                                     u->path, set_cookie);
2074         }
2075     }
2076
2077   if (resp_header_copy (resp, "Content-Range", hdrval, sizeof (hdrval)))
2078     {
2079       wgint first_byte_pos, last_byte_pos, entity_length;
2080       if (parse_content_range (hdrval, &first_byte_pos, &last_byte_pos,
2081                                &entity_length))
2082         {
2083           contrange = first_byte_pos;
2084           contlen = last_byte_pos - first_byte_pos + 1;
2085         }
2086     }
2087   resp_free (resp);
2088
2089   /* 20x responses are counted among successful by default.  */
2090   if (H_20X (statcode))
2091     *dt |= RETROKF;
2092
2093   /* Return if redirected.  */
2094   if (H_REDIRECTED (statcode) || statcode == HTTP_STATUS_MULTIPLE_CHOICES)
2095     {
2096       /* RFC2068 says that in case of the 300 (multiple choices)
2097          response, the server can output a preferred URL through
2098          `Location' header; otherwise, the request should be treated
2099          like GET.  So, if the location is set, it will be a
2100          redirection; otherwise, just proceed normally.  */
2101       if (statcode == HTTP_STATUS_MULTIPLE_CHOICES && !hs->newloc)
2102         *dt |= RETROKF;
2103       else
2104         {
2105           logprintf (LOG_VERBOSE,
2106                      _("Location: %s%s\n"),
2107                      hs->newloc ? escnonprint_uri (hs->newloc) : _("unspecified"),
2108                      hs->newloc ? _(" [following]") : "");
2109           if (keep_alive && !head_only && skip_short_body (sock, contlen))
2110             CLOSE_FINISH (sock);
2111           else
2112             CLOSE_INVALIDATE (sock);
2113           xfree_null (type);
2114           return NEWLOCATION;
2115         }
2116     }
2117
2118   /* If content-type is not given, assume text/html.  This is because
2119      of the multitude of broken CGI's that "forget" to generate the
2120      content-type.  */
2121   if (!type ||
2122         0 == strncasecmp (type, TEXTHTML_S, strlen (TEXTHTML_S)) ||
2123         0 == strncasecmp (type, TEXTXHTML_S, strlen (TEXTXHTML_S)))    
2124     *dt |= TEXTHTML;
2125   else
2126     *dt &= ~TEXTHTML;
2127
2128   if (opt.html_extension && (*dt & TEXTHTML))
2129     /* -E / --html-extension / html_extension = on was specified, and this is a
2130        text/html file.  If some case-insensitive variation on ".htm[l]" isn't
2131        already the file's suffix, tack on ".html". */
2132     {
2133       char *last_period_in_local_filename = strrchr (hs->local_file, '.');
2134
2135       if (last_period_in_local_filename == NULL
2136           || !(0 == strcasecmp (last_period_in_local_filename, ".htm")
2137                || 0 == strcasecmp (last_period_in_local_filename, ".html")))
2138         {
2139           int local_filename_len = strlen (hs->local_file);
2140           /* Resize the local file, allowing for ".html" preceded by
2141              optional ".NUMBER".  */
2142           hs->local_file = xrealloc (hs->local_file,
2143                                      local_filename_len + 24 + sizeof (".html"));
2144           strcpy(hs->local_file + local_filename_len, ".html");
2145           /* If clobbering is not allowed and the file, as named,
2146              exists, tack on ".NUMBER.html" instead. */
2147           if (!ALLOW_CLOBBER && file_exists_p (hs->local_file))
2148             {
2149               int ext_num = 1;
2150               do
2151                 sprintf (hs->local_file + local_filename_len,
2152                          ".%d.html", ext_num++);
2153               while (file_exists_p (hs->local_file));
2154             }
2155           *dt |= ADDED_HTML_EXTENSION;
2156         }
2157     }
2158
2159   if (statcode == HTTP_STATUS_RANGE_NOT_SATISFIABLE)
2160     {
2161       /* If `-c' is in use and the file has been fully downloaded (or
2162          the remote file has shrunk), Wget effectively requests bytes
2163          after the end of file and the server response with 416.  */
2164       logputs (LOG_VERBOSE, _("\
2165 \n    The file is already fully retrieved; nothing to do.\n\n"));
2166       /* In case the caller inspects. */
2167       hs->len = contlen;
2168       hs->res = 0;
2169       /* Mark as successfully retrieved. */
2170       *dt |= RETROKF;
2171       xfree_null (type);
2172       CLOSE_INVALIDATE (sock);        /* would be CLOSE_FINISH, but there
2173                                    might be more bytes in the body. */
2174       return RETRUNNEEDED;
2175     }
2176   if ((contrange != 0 && contrange != hs->restval)
2177       || (H_PARTIAL (statcode) && !contrange))
2178     {
2179       /* The Range request was somehow misunderstood by the server.
2180          Bail out.  */
2181       xfree_null (type);
2182       CLOSE_INVALIDATE (sock);
2183       return RANGEERR;
2184     }
2185   if (contlen == -1)
2186     hs->contlen = -1;
2187   else
2188     hs->contlen = contlen + contrange;
2189
2190   if (opt.verbose)
2191     {
2192       if (*dt & RETROKF)
2193         {
2194           /* No need to print this output if the body won't be
2195              downloaded at all, or if the original server response is
2196              printed.  */
2197           logputs (LOG_VERBOSE, _("Length: "));
2198           if (contlen != -1)
2199             {
2200               logputs (LOG_VERBOSE, number_to_static_string (contlen + contrange));
2201               if (contlen + contrange >= 1024)
2202                 logprintf (LOG_VERBOSE, " (%s)",
2203                            human_readable (contlen + contrange));
2204               if (contrange)
2205                 {
2206                   if (contlen >= 1024)
2207                     logprintf (LOG_VERBOSE, _(", %s (%s) remaining"),
2208                                number_to_static_string (contlen),
2209                                human_readable (contlen));
2210                   else
2211                     logprintf (LOG_VERBOSE, _(", %s remaining"),
2212                                number_to_static_string (contlen));
2213                 }
2214             }
2215           else
2216             logputs (LOG_VERBOSE,
2217                      opt.ignore_length ? _("ignored") : _("unspecified"));
2218           if (type)
2219             logprintf (LOG_VERBOSE, " [%s]\n", quotearg_style (escape_quoting_style, type));
2220           else
2221             logputs (LOG_VERBOSE, "\n");
2222         }
2223     }
2224   xfree_null (type);
2225   type = NULL;                        /* We don't need it any more.  */
2226
2227   /* Return if we have no intention of further downloading.  */
2228   if (!(*dt & RETROKF) || head_only)
2229     {
2230       /* In case the caller cares to look...  */
2231       hs->len = 0;
2232       hs->res = 0;
2233       xfree_null (type);
2234       if (head_only)
2235         /* Pre-1.10 Wget used CLOSE_INVALIDATE here.  Now we trust the
2236            servers not to send body in response to a HEAD request, and
2237            those that do will likely be caught by test_socket_open.
2238            If not, they can be worked around using
2239            `--no-http-keep-alive'.  */
2240         CLOSE_FINISH (sock);
2241       else if (keep_alive && skip_short_body (sock, contlen))
2242         /* Successfully skipped the body; also keep using the socket. */
2243         CLOSE_FINISH (sock);
2244       else
2245         CLOSE_INVALIDATE (sock);
2246       return RETRFINISHED;
2247     }
2248
2249   /* Open the local file.  */
2250   if (!output_stream)
2251     {
2252       mkalldirs (hs->local_file);
2253       if (opt.backups)
2254         rotate_backups (hs->local_file);
2255       if (hs->restval)
2256         fp = fopen (hs->local_file, "ab");
2257       else if (ALLOW_CLOBBER)
2258         fp = fopen (hs->local_file, "wb");
2259       else
2260         {
2261           fp = fopen_excl (hs->local_file, true);
2262           if (!fp && errno == EEXIST)
2263             {
2264               /* We cannot just invent a new name and use it (which is
2265                  what functions like unique_create typically do)
2266                  because we told the user we'd use this name.
2267                  Instead, return and retry the download.  */
2268               logprintf (LOG_NOTQUIET,
2269                          _("%s has sprung into existence.\n"),
2270                          hs->local_file);
2271               CLOSE_INVALIDATE (sock);
2272               return FOPEN_EXCL_ERR;
2273             }
2274         }
2275       if (!fp)
2276         {
2277           logprintf (LOG_NOTQUIET, "%s: %s\n", hs->local_file, strerror (errno));
2278           CLOSE_INVALIDATE (sock);
2279           return FOPENERR;
2280         }
2281     }
2282   else
2283     fp = output_stream;
2284
2285   /* Print fetch message, if opt.verbose.  */
2286   if (opt.verbose)
2287     {
2288       logprintf (LOG_NOTQUIET, _("Saving to: %s\n"), 
2289                  HYPHENP (hs->local_file) ? quote ("STDOUT") : quote (hs->local_file));
2290     }
2291     
2292   /* This confuses the timestamping code that checks for file size.
2293      #### The timestamping code should be smarter about file size.  */
2294   if (opt.save_headers && hs->restval == 0)
2295     fwrite (head, 1, strlen (head), fp);
2296
2297   /* Now we no longer need to store the response header. */
2298   xfree (head);
2299
2300   /* Download the request body.  */
2301   flags = 0;
2302   if (contlen != -1)
2303     /* If content-length is present, read that much; otherwise, read
2304        until EOF.  The HTTP spec doesn't require the server to
2305        actually close the connection when it's done sending data. */
2306     flags |= rb_read_exactly;
2307   if (hs->restval > 0 && contrange == 0)
2308     /* If the server ignored our range request, instruct fd_read_body
2309        to skip the first RESTVAL bytes of body.  */
2310     flags |= rb_skip_startpos;
2311   hs->len = hs->restval;
2312   hs->rd_size = 0;
2313   hs->res = fd_read_body (sock, fp, contlen != -1 ? contlen : 0,
2314                           hs->restval, &hs->rd_size, &hs->len, &hs->dltime,
2315                           flags);
2316
2317   if (hs->res >= 0)
2318     CLOSE_FINISH (sock);
2319   else
2320     {
2321       if (hs->res < 0)
2322         hs->rderrmsg = xstrdup (fd_errstr (sock));
2323       CLOSE_INVALIDATE (sock);
2324     }
2325
2326   if (!output_stream)
2327     fclose (fp);
2328   if (hs->res == -2)
2329     return FWRITEERR;
2330   return RETRFINISHED;
2331 }
2332
2333 /* The genuine HTTP loop!  This is the part where the retrieval is
2334    retried, and retried, and retried, and...  */
2335 uerr_t
2336 http_loop (struct url *u, char **newloc, char **local_file, const char *referer,
2337            int *dt, struct url *proxy)
2338 {
2339   int count;
2340   bool got_head = false;         /* used for time-stamping and filename detection */
2341   bool time_came_from_head = false;
2342   bool got_name = false;
2343   char *tms;
2344   const char *tmrate;
2345   uerr_t err, ret = TRYLIMEXC;
2346   time_t tmr = -1;               /* remote time-stamp */
2347   struct http_stat hstat;        /* HTTP status */
2348   struct_stat st;  
2349   bool send_head_first = true;
2350
2351   /* Assert that no value for *LOCAL_FILE was passed. */
2352   assert (local_file == NULL || *local_file == NULL);
2353   
2354   /* Set LOCAL_FILE parameter. */
2355   if (local_file && opt.output_document)
2356     *local_file = HYPHENP (opt.output_document) ? NULL : xstrdup (opt.output_document);
2357   
2358   /* Reset NEWLOC parameter. */
2359   *newloc = NULL;
2360
2361   /* This used to be done in main(), but it's a better idea to do it
2362      here so that we don't go through the hoops if we're just using
2363      FTP or whatever. */
2364   if (opt.cookies)
2365     load_cookies();
2366
2367   /* Warn on (likely bogus) wildcard usage in HTTP. */
2368   if (opt.ftp_glob && has_wildcards_p (u->path))
2369     logputs (LOG_VERBOSE, _("Warning: wildcards not supported in HTTP.\n"));
2370
2371   /* Setup hstat struct. */
2372   xzero (hstat);
2373   hstat.referer = referer;
2374
2375   if (opt.output_document)
2376     {
2377       hstat.local_file = xstrdup (opt.output_document);
2378       got_name = true;
2379     }
2380   else if (!opt.content_disposition)
2381     {
2382       hstat.local_file = url_file_name (u);
2383       got_name = true;
2384     }
2385
2386   /* TODO: Ick! This code is now in both gethttp and http_loop, and is
2387    * screaming for some refactoring. */
2388   if (got_name && file_exists_p (hstat.local_file) && opt.noclobber && !opt.output_document)
2389     {
2390       /* If opt.noclobber is turned on and file already exists, do not
2391          retrieve the file. But if the output_document was given, then this
2392          test was already done and the file didn't exist. Hence the !opt.output_document */
2393       logprintf (LOG_VERBOSE, _("\
2394 File %s already there; not retrieving.\n\n"), 
2395                  quote (hstat.local_file));
2396       /* If the file is there, we suppose it's retrieved OK.  */
2397       *dt |= RETROKF;
2398
2399       /* #### Bogusness alert.  */
2400       /* If its suffix is "html" or "htm" or similar, assume text/html.  */
2401       if (has_html_suffix_p (hstat.local_file))
2402         *dt |= TEXTHTML;
2403
2404       ret = RETROK;
2405       goto exit;
2406     }
2407
2408   /* Reset the counter. */
2409   count = 0;
2410   
2411   /* Reset the document type. */
2412   *dt = 0;
2413   
2414   /* Skip preliminary HEAD request if we're not in spider mode AND
2415    * if -O was given or HTTP Content-Disposition support is disabled. */
2416   if (!opt.spider
2417       && (got_name || !opt.content_disposition))
2418     send_head_first = false;
2419
2420   /* Send preliminary HEAD request if -N is given and we have an existing 
2421    * destination file. */
2422   if (opt.timestamping 
2423       && !opt.content_disposition
2424       && file_exists_p (url_file_name (u)))
2425     send_head_first = true;
2426   
2427   /* THE loop */
2428   do
2429     {
2430       /* Increment the pass counter.  */
2431       ++count;
2432       sleep_between_retrievals (count);
2433       
2434       /* Get the current time string.  */
2435       tms = datetime_str (time (NULL));
2436       
2437       if (opt.spider && !got_head)
2438         logprintf (LOG_VERBOSE, _("\
2439 Spider mode enabled. Check if remote file exists.\n"));
2440
2441       /* Print fetch message, if opt.verbose.  */
2442       if (opt.verbose)
2443         {
2444           char *hurl = url_string (u, URL_AUTH_HIDE_PASSWD);
2445           
2446           if (count > 1) 
2447             {
2448               char tmp[256];
2449               sprintf (tmp, _("(try:%2d)"), count);
2450               logprintf (LOG_NOTQUIET, "--%s--  %s  %s\n",
2451                          tms, tmp, hurl);
2452             }
2453           else 
2454             {
2455               logprintf (LOG_NOTQUIET, "--%s--  %s\n",
2456                          tms, hurl);
2457             }
2458           
2459 #ifdef WINDOWS
2460           ws_changetitle (hurl);
2461 #endif
2462           xfree (hurl);
2463         }
2464
2465       /* Default document type is empty.  However, if spider mode is
2466          on or time-stamping is employed, HEAD_ONLY commands is
2467          encoded within *dt.  */
2468       if (send_head_first && !got_head) 
2469         *dt |= HEAD_ONLY;
2470       else
2471         *dt &= ~HEAD_ONLY;
2472
2473       /* Decide whether or not to restart.  */
2474       if (opt.always_rest
2475           && got_name
2476           && stat (hstat.local_file, &st) == 0
2477           && S_ISREG (st.st_mode))
2478         /* When -c is used, continue from on-disk size.  (Can't use
2479            hstat.len even if count>1 because we don't want a failed
2480            first attempt to clobber existing data.)  */
2481         hstat.restval = st.st_size;
2482       else if (count > 1)
2483         /* otherwise, continue where the previous try left off */
2484         hstat.restval = hstat.len;
2485       else
2486         hstat.restval = 0;
2487
2488       /* Decide whether to send the no-cache directive.  We send it in
2489          two cases:
2490            a) we're using a proxy, and we're past our first retrieval.
2491               Some proxies are notorious for caching incomplete data, so
2492               we require a fresh get.
2493            b) caching is explicitly inhibited. */
2494       if ((proxy && count > 1)        /* a */
2495           || !opt.allow_cache)        /* b */
2496         *dt |= SEND_NOCACHE;
2497       else
2498         *dt &= ~SEND_NOCACHE;
2499
2500       /* Try fetching the document, or at least its head.  */
2501       err = gethttp (u, &hstat, dt, proxy);
2502
2503       /* Time?  */
2504       tms = datetime_str (time (NULL));
2505       
2506       /* Get the new location (with or without the redirection).  */
2507       if (hstat.newloc)
2508         *newloc = xstrdup (hstat.newloc);
2509
2510       switch (err)
2511         {
2512         case HERR: case HEOF: case CONSOCKERR: case CONCLOSED:
2513         case CONERROR: case READERR: case WRITEFAILED:
2514         case RANGEERR: case FOPEN_EXCL_ERR:
2515           /* Non-fatal errors continue executing the loop, which will
2516              bring them to "while" statement at the end, to judge
2517              whether the number of tries was exceeded.  */
2518           printwhat (count, opt.ntry);
2519           continue;
2520         case FWRITEERR: case FOPENERR:
2521           /* Another fatal error.  */
2522           logputs (LOG_VERBOSE, "\n");
2523           logprintf (LOG_NOTQUIET, _("Cannot write to %s (%s).\n"),
2524                      quote (hstat.local_file), strerror (errno));
2525         case HOSTERR: case CONIMPOSSIBLE: case PROXERR: case AUTHFAILED: 
2526         case SSLINITFAILED: case CONTNOTSUPPORTED:
2527           /* Fatal errors just return from the function.  */
2528           ret = err;
2529           goto exit;
2530         case CONSSLERR:
2531           /* Another fatal error.  */
2532           logprintf (LOG_NOTQUIET, _("Unable to establish SSL connection.\n"));
2533           ret = err;
2534           goto exit;
2535         case NEWLOCATION:
2536           /* Return the new location to the caller.  */
2537           if (!*newloc)
2538             {
2539               logprintf (LOG_NOTQUIET,
2540                          _("ERROR: Redirection (%d) without location.\n"),
2541                          hstat.statcode);
2542               ret = WRONGCODE;
2543             }
2544           else 
2545             {
2546               ret = NEWLOCATION;
2547             }
2548           goto exit;
2549         case RETRUNNEEDED:
2550           /* The file was already fully retrieved. */
2551           ret = RETROK;
2552           goto exit;
2553         case RETRFINISHED:
2554           /* Deal with you later.  */
2555           break;
2556         default:
2557           /* All possibilities should have been exhausted.  */
2558           abort ();
2559         }
2560       
2561       if (!(*dt & RETROKF))
2562         {
2563           char *hurl = NULL;
2564           if (!opt.verbose)
2565             {
2566               /* #### Ugly ugly ugly! */
2567               hurl = url_string (u, URL_AUTH_HIDE_PASSWD);
2568               logprintf (LOG_NONVERBOSE, "%s:\n", hurl);
2569             }
2570
2571           /* Fall back to GET if HEAD fails with a 500 or 501 error code. */
2572           if (*dt & HEAD_ONLY
2573               && (hstat.statcode == 500 || hstat.statcode == 501))
2574             {
2575               got_head = true;
2576               continue;
2577             }
2578           /* Maybe we should always keep track of broken links, not just in
2579            * spider mode.  */
2580           else if (opt.spider)
2581             {
2582               /* #### Again: ugly ugly ugly! */
2583               if (!hurl) 
2584                 hurl = url_string (u, URL_AUTH_HIDE_PASSWD);
2585               nonexisting_url (hurl);
2586               logprintf (LOG_NOTQUIET, _("\
2587 Remote file does not exist -- broken link!!!\n"));
2588             }
2589           else
2590             {
2591               logprintf (LOG_NOTQUIET, _("%s ERROR %d: %s.\n"),
2592                          tms, hstat.statcode, 
2593                          quotearg_style (escape_quoting_style, hstat.error));
2594             }
2595           logputs (LOG_VERBOSE, "\n");
2596           ret = WRONGCODE;
2597           xfree_null (hurl);
2598           goto exit;
2599         }
2600
2601       /* Did we get the time-stamp? */
2602       if (!got_head)
2603         {
2604           got_head = true;    /* no more time-stamping */
2605
2606           if (opt.timestamping && !hstat.remote_time)
2607             {
2608               logputs (LOG_NOTQUIET, _("\
2609 Last-modified header missing -- time-stamps turned off.\n"));
2610             }
2611           else if (hstat.remote_time)
2612             {
2613               /* Convert the date-string into struct tm.  */
2614               tmr = http_atotm (hstat.remote_time);
2615               if (tmr == (time_t) (-1))
2616                 logputs (LOG_VERBOSE, _("\
2617 Last-modified header invalid -- time-stamp ignored.\n"));
2618               if (*dt & HEAD_ONLY)
2619                 time_came_from_head = true;
2620             }
2621       
2622           if (send_head_first)
2623             {
2624               /* The time-stamping section.  */
2625               if (opt.timestamping)
2626                 {
2627                   if (hstat.orig_file_name) /* Perform the following
2628                                                checks only if the file
2629                                                we're supposed to
2630                                                download already exists.  */
2631                     {
2632                       if (hstat.remote_time && 
2633                           tmr != (time_t) (-1))
2634                         {
2635                           /* Now time-stamping can be used validly.
2636                              Time-stamping means that if the sizes of
2637                              the local and remote file match, and local
2638                              file is newer than the remote file, it will
2639                              not be retrieved.  Otherwise, the normal
2640                              download procedure is resumed.  */
2641                           if (hstat.orig_file_tstamp >= tmr)
2642                             {
2643                               if (hstat.contlen == -1 
2644                                   || hstat.orig_file_size == hstat.contlen)
2645                                 {
2646                                   logprintf (LOG_VERBOSE, _("\
2647 Server file no newer than local file %s -- not retrieving.\n\n"),
2648                                              quote (hstat.orig_file_name));
2649                                   ret = RETROK;
2650                                   goto exit;
2651                                 }
2652                               else
2653                                 {
2654                                   logprintf (LOG_VERBOSE, _("\
2655 The sizes do not match (local %s) -- retrieving.\n"),
2656                                              number_to_static_string (hstat.orig_file_size));
2657                                 }
2658                             }
2659                           else
2660                             logputs (LOG_VERBOSE,
2661                                      _("Remote file is newer, retrieving.\n"));
2662
2663                           logputs (LOG_VERBOSE, "\n");
2664                         }
2665                     }
2666                   
2667                   /* free_hstat (&hstat); */
2668                   hstat.timestamp_checked = true;
2669                 }
2670               
2671               if (opt.spider)
2672                 {
2673                   if (opt.recursive)
2674                     {
2675                       if (*dt & TEXTHTML)
2676                         {
2677                           logputs (LOG_VERBOSE, _("\
2678 Remote file exists and could contain links to other resources -- retrieving.\n\n"));
2679                         }
2680                       else 
2681                         {
2682                           logprintf (LOG_VERBOSE, _("\
2683 Remote file exists but does not contain any link -- not retrieving.\n\n"));
2684                           ret = RETROK; /* RETRUNNEEDED is not for caller. */
2685                           goto exit;
2686                         }
2687                     }
2688                   else
2689                     {
2690                       if (*dt & TEXTHTML)
2691                         {
2692                           logprintf (LOG_VERBOSE, _("\
2693 Remote file exists and could contain further links,\n\
2694 but recursion is disabled -- not retrieving.\n\n"));
2695                         }
2696                       else 
2697                         {
2698                           logprintf (LOG_VERBOSE, _("\
2699 Remote file exists.\n\n"));
2700                         }
2701                       ret = RETROK; /* RETRUNNEEDED is not for caller. */
2702                       goto exit;
2703                     }
2704                 }
2705
2706               got_name = true;
2707               *dt &= ~HEAD_ONLY;
2708               count = 0;          /* the retrieve count for HEAD is reset */
2709               continue;
2710             } /* send_head_first */
2711         } /* !got_head */
2712           
2713       if ((tmr != (time_t) (-1))
2714           && ((hstat.len == hstat.contlen) ||
2715               ((hstat.res == 0) && (hstat.contlen == -1))))
2716         {
2717           /* #### This code repeats in http.c and ftp.c.  Move it to a
2718              function!  */
2719           const char *fl = NULL;
2720           if (opt.output_document)
2721             {
2722               if (output_stream_regular)
2723                 fl = opt.output_document;
2724             }
2725           else
2726             fl = hstat.local_file;
2727           if (fl)
2728             {
2729               time_t newtmr = -1;
2730               /* Reparse time header, in case it's changed. */
2731               if (time_came_from_head
2732                   && hstat.remote_time && hstat.remote_time[0])
2733                 {
2734                   newtmr = http_atotm (hstat.remote_time);
2735                   if (newtmr != -1)
2736                     tmr = newtmr;
2737                 }
2738               touch (fl, tmr);
2739             }
2740         }
2741       /* End of time-stamping section. */
2742
2743       tmrate = retr_rate (hstat.rd_size, hstat.dltime);
2744       total_download_time += hstat.dltime;
2745
2746       if (hstat.len == hstat.contlen)
2747         {
2748           if (*dt & RETROKF)
2749             {
2750               logprintf (LOG_VERBOSE,
2751                          _("%s (%s) - %s saved [%s/%s]\n\n"),
2752                          tms, tmrate, quote (hstat.local_file),
2753                          number_to_static_string (hstat.len),
2754                          number_to_static_string (hstat.contlen));
2755               logprintf (LOG_NONVERBOSE,
2756                          "%s URL:%s [%s/%s] -> \"%s\" [%d]\n",
2757                          tms, u->url,
2758                          number_to_static_string (hstat.len),
2759                          number_to_static_string (hstat.contlen),
2760                          hstat.local_file, count);
2761             }
2762           ++opt.numurls;
2763           total_downloaded_bytes += hstat.len;
2764
2765           /* Remember that we downloaded the file for later ".orig" code. */
2766           if (*dt & ADDED_HTML_EXTENSION)
2767             downloaded_file(FILE_DOWNLOADED_AND_HTML_EXTENSION_ADDED, hstat.local_file);
2768           else
2769             downloaded_file(FILE_DOWNLOADED_NORMALLY, hstat.local_file);
2770
2771           ret = RETROK;
2772           goto exit;
2773         }
2774       else if (hstat.res == 0) /* No read error */
2775         {
2776           if (hstat.contlen == -1)  /* We don't know how much we were supposed
2777                                        to get, so assume we succeeded. */ 
2778             {
2779               if (*dt & RETROKF)
2780                 {
2781                   logprintf (LOG_VERBOSE,
2782                              _("%s (%s) - %s saved [%s]\n\n"),
2783                              tms, tmrate, quote (hstat.local_file),
2784                              number_to_static_string (hstat.len));
2785                   logprintf (LOG_NONVERBOSE,
2786                              "%s URL:%s [%s] -> \"%s\" [%d]\n",
2787                              tms, u->url, number_to_static_string (hstat.len),
2788                              hstat.local_file, count);
2789                 }
2790               ++opt.numurls;
2791               total_downloaded_bytes += hstat.len;
2792
2793               /* Remember that we downloaded the file for later ".orig" code. */
2794               if (*dt & ADDED_HTML_EXTENSION)
2795                 downloaded_file(FILE_DOWNLOADED_AND_HTML_EXTENSION_ADDED, hstat.local_file);
2796               else
2797                 downloaded_file(FILE_DOWNLOADED_NORMALLY, hstat.local_file);
2798               
2799               ret = RETROK;
2800               goto exit;
2801             }
2802           else if (hstat.len < hstat.contlen) /* meaning we lost the
2803                                                  connection too soon */
2804             {
2805               logprintf (LOG_VERBOSE,
2806                          _("%s (%s) - Connection closed at byte %s. "),
2807                          tms, tmrate, number_to_static_string (hstat.len));
2808               printwhat (count, opt.ntry);
2809               continue;
2810             }
2811           else
2812             /* Getting here would mean reading more data than
2813                requested with content-length, which we never do.  */
2814             abort ();
2815         }
2816       else /* from now on hstat.res can only be -1 */
2817         {
2818           if (hstat.contlen == -1)
2819             {
2820               logprintf (LOG_VERBOSE,
2821                          _("%s (%s) - Read error at byte %s (%s)."),
2822                          tms, tmrate, number_to_static_string (hstat.len),
2823                          hstat.rderrmsg);
2824               printwhat (count, opt.ntry);
2825               continue;
2826             }
2827           else /* hstat.res == -1 and contlen is given */
2828             {
2829               logprintf (LOG_VERBOSE,
2830                          _("%s (%s) - Read error at byte %s/%s (%s). "),
2831                          tms, tmrate,
2832                          number_to_static_string (hstat.len),
2833                          number_to_static_string (hstat.contlen),
2834                          hstat.rderrmsg);
2835               printwhat (count, opt.ntry);
2836               continue;
2837             }
2838         }
2839       /* not reached */
2840     }
2841   while (!opt.ntry || (count < opt.ntry));
2842
2843 exit:
2844   if (ret == RETROK) 
2845     *local_file = xstrdup (hstat.local_file);
2846   free_hstat (&hstat);
2847   
2848   return ret;
2849 }
2850 \f
2851 /* Check whether the result of strptime() indicates success.
2852    strptime() returns the pointer to how far it got to in the string.
2853    The processing has been successful if the string is at `GMT' or
2854    `+X', or at the end of the string.
2855
2856    In extended regexp parlance, the function returns 1 if P matches
2857    "^ *(GMT|[+-][0-9]|$)", 0 otherwise.  P being NULL (which strptime
2858    can return) is considered a failure and 0 is returned.  */
2859 static bool
2860 check_end (const char *p)
2861 {
2862   if (!p)
2863     return false;
2864   while (c_isspace (*p))
2865     ++p;
2866   if (!*p
2867       || (p[0] == 'G' && p[1] == 'M' && p[2] == 'T')
2868       || ((p[0] == '+' || p[0] == '-') && c_isdigit (p[1])))
2869     return true;
2870   else
2871     return false;
2872 }
2873
2874 /* Convert the textual specification of time in TIME_STRING to the
2875    number of seconds since the Epoch.
2876
2877    TIME_STRING can be in any of the three formats RFC2616 allows the
2878    HTTP servers to emit -- RFC1123-date, RFC850-date or asctime-date,
2879    as well as the time format used in the Set-Cookie header.
2880    Timezones are ignored, and should be GMT.
2881
2882    Return the computed time_t representation, or -1 if the conversion
2883    fails.
2884
2885    This function uses strptime with various string formats for parsing
2886    TIME_STRING.  This results in a parser that is not as lenient in
2887    interpreting TIME_STRING as I would like it to be.  Being based on
2888    strptime, it always allows shortened months, one-digit days, etc.,
2889    but due to the multitude of formats in which time can be
2890    represented, an ideal HTTP time parser would be even more
2891    forgiving.  It should completely ignore things like week days and
2892    concentrate only on the various forms of representing years,
2893    months, days, hours, minutes, and seconds.  For example, it would
2894    be nice if it accepted ISO 8601 out of the box.
2895
2896    I've investigated free and PD code for this purpose, but none was
2897    usable.  getdate was big and unwieldy, and had potential copyright
2898    issues, or so I was informed.  Dr. Marcus Hennecke's atotm(),
2899    distributed with phttpd, is excellent, but we cannot use it because
2900    it is not assigned to the FSF.  So I stuck it with strptime.  */
2901
2902 time_t
2903 http_atotm (const char *time_string)
2904 {
2905   /* NOTE: Solaris strptime man page claims that %n and %t match white
2906      space, but that's not universally available.  Instead, we simply
2907      use ` ' to mean "skip all WS", which works under all strptime
2908      implementations I've tested.  */
2909
2910   static const char *time_formats[] = {
2911     "%a, %d %b %Y %T",          /* rfc1123: Thu, 29 Jan 1998 22:12:57 */
2912     "%A, %d-%b-%y %T",          /* rfc850:  Thursday, 29-Jan-98 22:12:57 */
2913     "%a %b %d %T %Y",           /* asctime: Thu Jan 29 22:12:57 1998 */
2914     "%a, %d-%b-%Y %T"           /* cookies: Thu, 29-Jan-1998 22:12:57
2915                                    (used in Set-Cookie, defined in the
2916                                    Netscape cookie specification.) */
2917   };
2918   const char *oldlocale;
2919   size_t i;
2920   time_t ret = (time_t) -1;
2921
2922   /* Solaris strptime fails to recognize English month names in
2923      non-English locales, which we work around by temporarily setting
2924      locale to C before invoking strptime.  */
2925   oldlocale = setlocale (LC_TIME, NULL);
2926   setlocale (LC_TIME, "C");
2927
2928   for (i = 0; i < countof (time_formats); i++)
2929     {
2930       struct tm t;
2931
2932       /* Some versions of strptime use the existing contents of struct
2933          tm to recalculate the date according to format.  Zero it out
2934          to prevent stack garbage from influencing strptime.  */
2935       xzero (t);
2936
2937       if (check_end (strptime (time_string, time_formats[i], &t)))
2938         {
2939           ret = timegm (&t);
2940           break;
2941         }
2942     }
2943
2944   /* Restore the previous locale. */
2945   setlocale (LC_TIME, oldlocale);
2946
2947   return ret;
2948 }
2949 \f
2950 /* Authorization support: We support three authorization schemes:
2951
2952    * `Basic' scheme, consisting of base64-ing USER:PASSWORD string;
2953
2954    * `Digest' scheme, added by Junio Hamano <junio@twinsun.com>,
2955    consisting of answering to the server's challenge with the proper
2956    MD5 digests.
2957
2958    * `NTLM' ("NT Lan Manager") scheme, based on code written by Daniel
2959    Stenberg for libcurl.  Like digest, NTLM is based on a
2960    challenge-response mechanism, but unlike digest, it is non-standard
2961    (authenticates TCP connections rather than requests), undocumented
2962    and Microsoft-specific.  */
2963
2964 /* Create the authentication header contents for the `Basic' scheme.
2965    This is done by encoding the string "USER:PASS" to base64 and
2966    prepending the string "Basic " in front of it.  */
2967
2968 static char *
2969 basic_authentication_encode (const char *user, const char *passwd)
2970 {
2971   char *t1, *t2;
2972   int len1 = strlen (user) + 1 + strlen (passwd);
2973
2974   t1 = (char *)alloca (len1 + 1);
2975   sprintf (t1, "%s:%s", user, passwd);
2976
2977   t2 = (char *)alloca (BASE64_LENGTH (len1) + 1);
2978   base64_encode (t1, len1, t2);
2979
2980   return concat_strings ("Basic ", t2, (char *) 0);
2981 }
2982
2983 #define SKIP_WS(x) do {                         \
2984   while (c_isspace (*(x)))                        \
2985     ++(x);                                      \
2986 } while (0)
2987
2988 #ifdef ENABLE_DIGEST
2989 /* Dump the hexadecimal representation of HASH to BUF.  HASH should be
2990    an array of 16 bytes containing the hash keys, and BUF should be a
2991    buffer of 33 writable characters (32 for hex digits plus one for
2992    zero termination).  */
2993 static void
2994 dump_hash (char *buf, const unsigned char *hash)
2995 {
2996   int i;
2997
2998   for (i = 0; i < MD5_HASHLEN; i++, hash++)
2999     {
3000       *buf++ = XNUM_TO_digit (*hash >> 4);
3001       *buf++ = XNUM_TO_digit (*hash & 0xf);
3002     }
3003   *buf = '\0';
3004 }
3005
3006 /* Take the line apart to find the challenge, and compose a digest
3007    authorization header.  See RFC2069 section 2.1.2.  */
3008 static char *
3009 digest_authentication_encode (const char *au, const char *user,
3010                               const char *passwd, const char *method,
3011                               const char *path)
3012 {
3013   static char *realm, *opaque, *nonce;
3014   static struct {
3015     const char *name;
3016     char **variable;
3017   } options[] = {
3018     { "realm", &realm },
3019     { "opaque", &opaque },
3020     { "nonce", &nonce }
3021   };
3022   char *res;
3023   param_token name, value;
3024
3025   realm = opaque = nonce = NULL;
3026
3027   au += 6;                      /* skip over `Digest' */
3028   while (extract_param (&au, &name, &value, ','))
3029     {
3030       size_t i;
3031       size_t namelen = name.e - name.b;
3032       for (i = 0; i < countof (options); i++)
3033         if (namelen == strlen (options[i].name)
3034             && 0 == strncmp (name.b, options[i].name,
3035                              namelen))
3036           {
3037             *options[i].variable = strdupdelim (value.b, value.e);
3038             break;
3039           }
3040     }
3041   if (!realm || !nonce || !user || !passwd || !path || !method)
3042     {
3043       xfree_null (realm);
3044       xfree_null (opaque);
3045       xfree_null (nonce);
3046       return NULL;
3047     }
3048
3049   /* Calculate the digest value.  */
3050   {
3051     ALLOCA_MD5_CONTEXT (ctx);
3052     unsigned char hash[MD5_HASHLEN];
3053     char a1buf[MD5_HASHLEN * 2 + 1], a2buf[MD5_HASHLEN * 2 + 1];
3054     char response_digest[MD5_HASHLEN * 2 + 1];
3055
3056     /* A1BUF = H(user ":" realm ":" password) */
3057     gen_md5_init (ctx);
3058     gen_md5_update ((unsigned char *)user, strlen (user), ctx);
3059     gen_md5_update ((unsigned char *)":", 1, ctx);
3060     gen_md5_update ((unsigned char *)realm, strlen (realm), ctx);
3061     gen_md5_update ((unsigned char *)":", 1, ctx);
3062     gen_md5_update ((unsigned char *)passwd, strlen (passwd), ctx);
3063     gen_md5_finish (ctx, hash);
3064     dump_hash (a1buf, hash);
3065
3066     /* A2BUF = H(method ":" path) */
3067     gen_md5_init (ctx);
3068     gen_md5_update ((unsigned char *)method, strlen (method), ctx);
3069     gen_md5_update ((unsigned char *)":", 1, ctx);
3070     gen_md5_update ((unsigned char *)path, strlen (path), ctx);
3071     gen_md5_finish (ctx, hash);
3072     dump_hash (a2buf, hash);
3073
3074     /* RESPONSE_DIGEST = H(A1BUF ":" nonce ":" A2BUF) */
3075     gen_md5_init (ctx);
3076     gen_md5_update ((unsigned char *)a1buf, MD5_HASHLEN * 2, ctx);
3077     gen_md5_update ((unsigned char *)":", 1, ctx);
3078     gen_md5_update ((unsigned char *)nonce, strlen (nonce), ctx);
3079     gen_md5_update ((unsigned char *)":", 1, ctx);
3080     gen_md5_update ((unsigned char *)a2buf, MD5_HASHLEN * 2, ctx);
3081     gen_md5_finish (ctx, hash);
3082     dump_hash (response_digest, hash);
3083
3084     res = xmalloc (strlen (user)
3085                    + strlen (user)
3086                    + strlen (realm)
3087                    + strlen (nonce)
3088                    + strlen (path)
3089                    + 2 * MD5_HASHLEN /*strlen (response_digest)*/
3090                    + (opaque ? strlen (opaque) : 0)
3091                    + 128);
3092     sprintf (res, "Digest \
3093 username=\"%s\", realm=\"%s\", nonce=\"%s\", uri=\"%s\", response=\"%s\"",
3094              user, realm, nonce, path, response_digest);
3095     if (opaque)
3096       {
3097         char *p = res + strlen (res);
3098         strcat (p, ", opaque=\"");
3099         strcat (p, opaque);
3100         strcat (p, "\"");
3101       }
3102   }
3103   return res;
3104 }
3105 #endif /* ENABLE_DIGEST */
3106
3107 /* Computing the size of a string literal must take into account that
3108    value returned by sizeof includes the terminating \0.  */
3109 #define STRSIZE(literal) (sizeof (literal) - 1)
3110
3111 /* Whether chars in [b, e) begin with the literal string provided as
3112    first argument and are followed by whitespace or terminating \0.
3113    The comparison is case-insensitive.  */
3114 #define STARTS(literal, b, e)                           \
3115   ((e > b) \
3116    && ((size_t) ((e) - (b))) >= STRSIZE (literal)   \
3117    && 0 == strncasecmp (b, literal, STRSIZE (literal))  \
3118    && ((size_t) ((e) - (b)) == STRSIZE (literal)          \
3119        || c_isspace (b[STRSIZE (literal)])))
3120
3121 static bool
3122 known_authentication_scheme_p (const char *hdrbeg, const char *hdrend)
3123 {
3124   return STARTS ("Basic", hdrbeg, hdrend)
3125 #ifdef ENABLE_DIGEST
3126     || STARTS ("Digest", hdrbeg, hdrend)
3127 #endif
3128 #ifdef ENABLE_NTLM
3129     || STARTS ("NTLM", hdrbeg, hdrend)
3130 #endif
3131     ;
3132 }
3133
3134 #undef STARTS
3135
3136 /* Create the HTTP authorization request header.  When the
3137    `WWW-Authenticate' response header is seen, according to the
3138    authorization scheme specified in that header (`Basic' and `Digest'
3139    are supported by the current implementation), produce an
3140    appropriate HTTP authorization request header.  */
3141 static char *
3142 create_authorization_line (const char *au, const char *user,
3143                            const char *passwd, const char *method,
3144                            const char *path, bool *finished)
3145 {
3146   /* We are called only with known schemes, so we can dispatch on the
3147      first letter. */
3148   switch (c_toupper (*au))
3149     {
3150     case 'B':                   /* Basic */
3151       *finished = true;
3152       return basic_authentication_encode (user, passwd);
3153 #ifdef ENABLE_DIGEST
3154     case 'D':                   /* Digest */
3155       *finished = true;
3156       return digest_authentication_encode (au, user, passwd, method, path);
3157 #endif
3158 #ifdef ENABLE_NTLM
3159     case 'N':                   /* NTLM */
3160       if (!ntlm_input (&pconn.ntlm, au))
3161         {
3162           *finished = true;
3163           return NULL;
3164         }
3165       return ntlm_output (&pconn.ntlm, user, passwd, finished);
3166 #endif
3167     default:
3168       /* We shouldn't get here -- this function should be only called
3169          with values approved by known_authentication_scheme_p.  */
3170       abort ();
3171     }
3172 }
3173 \f
3174 static void
3175 load_cookies (void)
3176 {
3177   if (!wget_cookie_jar)
3178     wget_cookie_jar = cookie_jar_new ();
3179   if (opt.cookies_input && !cookies_loaded_p)
3180     {
3181       cookie_jar_load (wget_cookie_jar, opt.cookies_input);
3182       cookies_loaded_p = true;
3183     }
3184 }
3185
3186 void
3187 save_cookies (void)
3188 {
3189   if (wget_cookie_jar)
3190     cookie_jar_save (wget_cookie_jar, opt.cookies_output);
3191 }
3192
3193 void
3194 http_cleanup (void)
3195 {
3196   xfree_null (pconn.host);
3197   if (wget_cookie_jar)
3198     cookie_jar_delete (wget_cookie_jar);
3199 }
3200
3201
3202 #ifdef TESTING
3203
3204 const char *
3205 test_parse_content_disposition()
3206 {
3207   int i;
3208   struct {
3209     char *hdrval;    
3210     char *opt_dir_prefix;
3211     char *filename;
3212     bool result;
3213   } test_array[] = {
3214     { "filename=\"file.ext\"", NULL, "file.ext", true },
3215     { "filename=\"file.ext\"", "somedir", "somedir/file.ext", true },
3216     { "attachment; filename=\"file.ext\"", NULL, "file.ext", true },
3217     { "attachment; filename=\"file.ext\"", "somedir", "somedir/file.ext", true },
3218     { "attachment; filename=\"file.ext\"; dummy", NULL, "file.ext", true },
3219     { "attachment; filename=\"file.ext\"; dummy", "somedir", "somedir/file.ext", true },
3220     { "attachment", NULL, NULL, false },
3221     { "attachment", "somedir", NULL, false },
3222   };
3223   
3224   for (i = 0; i < sizeof(test_array)/sizeof(test_array[0]); ++i) 
3225     {
3226       char *filename;
3227       bool res;
3228
3229       opt.dir_prefix = test_array[i].opt_dir_prefix;
3230       res = parse_content_disposition (test_array[i].hdrval, &filename);
3231
3232       mu_assert ("test_parse_content_disposition: wrong result", 
3233                  res == test_array[i].result
3234                  && (res == false 
3235                      || 0 == strcmp (test_array[i].filename, filename)));
3236     }
3237
3238   return NULL;
3239 }
3240
3241 #endif /* TESTING */
3242
3243 /*
3244  * vim: et sts=2 sw=2 cino+={s
3245  */
3246