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