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