]> sjero.net Git - wget/blob - src/http.c
[svn] Remember that the NTLM-authorized TCP connections stay authorized.
[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 "gen_sslfunc.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 ()
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);
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);
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);
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 /* Retrieve a document through HTTP protocol.  It recognizes status
1117    code, and correctly handles redirections.  It closes the network
1118    socket.  If it receives an error from the functions below it, it
1119    will print it if there is enough information to do so (almost
1120    always), returning the error to the caller (i.e. http_loop).
1121
1122    Various HTTP parameters are stored to hs.
1123
1124    If PROXY is non-NULL, the connection will be made to the proxy
1125    server, and u->url will be requested.  */
1126 static uerr_t
1127 gethttp (struct url *u, struct http_stat *hs, int *dt, struct url *proxy)
1128 {
1129   struct request *req;
1130
1131   char *type;
1132   char *user, *passwd;
1133   char *proxyauth;
1134   int statcode;
1135   int write_error;
1136   wgint contlen, contrange;
1137   struct url *conn;
1138   FILE *fp;
1139
1140   int sock = -1;
1141   int flags;
1142
1143   /* Set to 1 when the authorization has failed permanently and should
1144      not be tried again. */
1145   int auth_finished = 0;
1146
1147   /* Whether NTLM authentication is used for this request. */
1148   int ntlm_seen = 0;
1149
1150   /* Whether our connection to the remote host is through SSL.  */
1151   int using_ssl = 0;
1152
1153   char *head;
1154   struct response *resp;
1155   char hdrval[256];
1156   char *message;
1157
1158   /* Whether this connection will be kept alive after the HTTP request
1159      is done. */
1160   int keep_alive;
1161
1162   /* Whether keep-alive should be inhibited.
1163
1164      RFC 2068 requests that 1.0 clients not send keep-alive requests
1165      to proxies.  This is because many 1.0 proxies do not interpret
1166      the Connection header and transfer it to the remote server,
1167      causing it to not close the connection and leave both the proxy
1168      and the client hanging.  */
1169   int inhibit_keep_alive =
1170     !opt.http_keep_alive || opt.ignore_length || proxy != NULL;
1171
1172   /* Headers sent when using POST. */
1173   wgint post_data_size = 0;
1174
1175   int host_lookup_failed = 0;
1176
1177 #ifdef HAVE_SSL
1178   if (u->scheme == SCHEME_HTTPS)
1179     {
1180       /* Initialize the SSL context.  After this has once been done,
1181          it becomes a no-op.  */
1182       switch (ssl_init ())
1183         {
1184         case SSLERRCTXCREATE:
1185           /* this is fatal */
1186           logprintf (LOG_NOTQUIET, _("Failed to set up an SSL context\n"));
1187           return SSLERRCTXCREATE;
1188         case SSLERRCERTFILE:
1189           /* try without certfile */
1190           logprintf (LOG_NOTQUIET,
1191                      _("Failed to load certificates from %s\n"),
1192                      opt.sslcertfile);
1193           logprintf (LOG_NOTQUIET,
1194                      _("Trying without the specified certificate\n"));
1195           break;
1196         case SSLERRCERTKEY:
1197           logprintf (LOG_NOTQUIET,
1198                      _("Failed to get certificate key from %s\n"),
1199                      opt.sslcertkey);
1200           logprintf (LOG_NOTQUIET,
1201                      _("Trying without the specified certificate\n"));
1202           break;
1203         default:
1204           break;
1205         }
1206     }
1207 #endif /* HAVE_SSL */
1208
1209   if (!(*dt & HEAD_ONLY))
1210     /* If we're doing a GET on the URL, as opposed to just a HEAD, we need to
1211        know the local filename so we can save to it. */
1212     assert (*hs->local_file != NULL);
1213
1214   /* Initialize certain elements of struct http_stat.  */
1215   hs->len = 0;
1216   hs->contlen = -1;
1217   hs->res = -1;
1218   hs->newloc = NULL;
1219   hs->remote_time = NULL;
1220   hs->error = NULL;
1221
1222   conn = u;
1223
1224   /* Prepare the request to send. */
1225
1226   req = request_new ();
1227   {
1228     const char *meth = "GET";
1229     if (*dt & HEAD_ONLY)
1230       meth = "HEAD";
1231     else if (opt.post_file_name || opt.post_data)
1232       meth = "POST";
1233     /* Use the full path, i.e. one that includes the leading slash and
1234        the query string.  E.g. if u->path is "foo/bar" and u->query is
1235        "param=value", full_path will be "/foo/bar?param=value".  */
1236     request_set_method (req, meth,
1237                         proxy ? xstrdup (u->url) : url_full_path (u));
1238   }
1239
1240   request_set_header (req, "Referer", (char *) hs->referer, rel_none);
1241   if (*dt & SEND_NOCACHE)
1242     request_set_header (req, "Pragma", "no-cache", rel_none);
1243   if (hs->restval)
1244     request_set_header (req, "Range",
1245                         aprintf ("bytes=%s-",
1246                                  number_to_static_string (hs->restval)),
1247                         rel_value);
1248   if (opt.useragent)
1249     request_set_header (req, "User-Agent", opt.useragent, rel_none);
1250   else
1251     request_set_header (req, "User-Agent",
1252                         aprintf ("Wget/%s", version_string), rel_value);
1253   request_set_header (req, "Accept", "*/*", rel_none);
1254
1255   /* Find the username and password for authentication. */
1256   user = u->user;
1257   passwd = u->passwd;
1258   search_netrc (u->host, (const char **)&user, (const char **)&passwd, 0);
1259   user = user ? user : opt.http_user;
1260   passwd = passwd ? passwd : opt.http_passwd;
1261
1262   if (user && passwd)
1263     {
1264       /* We have the username and the password, but haven't tried
1265          any authorization yet.  Let's see if the "Basic" method
1266          works.  If not, we'll come back here and construct a
1267          proper authorization method with the right challenges.
1268
1269          If we didn't employ this kind of logic, every URL that
1270          requires authorization would have to be processed twice,
1271          which is very suboptimal and generates a bunch of false
1272          "unauthorized" errors in the server log.
1273
1274          #### But this logic also has a serious problem when used
1275          with stronger authentications: we *first* transmit the
1276          username and the password in clear text, and *then* attempt a
1277          stronger authentication scheme.  That cannot be right!  We
1278          are only fortunate that almost everyone still uses the
1279          `Basic' scheme anyway.
1280
1281          There should be an option to prevent this from happening, for
1282          those who use strong authentication schemes and value their
1283          passwords.  */
1284       request_set_header (req, "Authorization",
1285                           basic_authentication_encode (user, passwd),
1286                           rel_value);
1287     }
1288
1289   proxyauth = NULL;
1290   if (proxy)
1291     {
1292       char *proxy_user, *proxy_passwd;
1293       /* For normal username and password, URL components override
1294          command-line/wgetrc parameters.  With proxy
1295          authentication, it's the reverse, because proxy URLs are
1296          normally the "permanent" ones, so command-line args
1297          should take precedence.  */
1298       if (opt.proxy_user && opt.proxy_passwd)
1299         {
1300           proxy_user = opt.proxy_user;
1301           proxy_passwd = opt.proxy_passwd;
1302         }
1303       else
1304         {
1305           proxy_user = proxy->user;
1306           proxy_passwd = proxy->passwd;
1307         }
1308       /* #### This does not appear right.  Can't the proxy request,
1309          say, `Digest' authentication?  */
1310       if (proxy_user && proxy_passwd)
1311         proxyauth = basic_authentication_encode (proxy_user, proxy_passwd);
1312
1313       /* If we're using a proxy, we will be connecting to the proxy
1314          server.  */
1315       conn = proxy;
1316
1317       /* Proxy authorization over SSL is handled below. */
1318 #ifdef HAVE_SSL
1319       if (u->scheme != SCHEME_HTTPS)
1320 #endif
1321         request_set_header (req, "Proxy-Authorization", proxyauth, rel_value);
1322     }
1323
1324   {
1325     /* Whether we need to print the host header with braces around
1326        host, e.g. "Host: [3ffe:8100:200:2::2]:1234" instead of the
1327        usual "Host: symbolic-name:1234". */
1328     int squares = strchr (u->host, ':') != NULL;
1329     if (u->port == scheme_default_port (u->scheme))
1330       request_set_header (req, "Host",
1331                           aprintf (squares ? "[%s]" : "%s", u->host),
1332                           rel_value);
1333     else
1334       request_set_header (req, "Host",
1335                           aprintf (squares ? "[%s]:%d" : "%s:%d",
1336                                    u->host, u->port),
1337                           rel_value);
1338   }
1339
1340   if (!inhibit_keep_alive)
1341     request_set_header (req, "Connection", "Keep-Alive", rel_none);
1342
1343   if (opt.cookies)
1344     request_set_header (req, "Cookie",
1345                         cookie_header (wget_cookie_jar,
1346                                        u->host, u->port, u->path,
1347 #ifdef HAVE_SSL
1348                                        u->scheme == SCHEME_HTTPS
1349 #else
1350                                        0
1351 #endif
1352                                        ),
1353                         rel_value);
1354
1355   if (opt.post_data || opt.post_file_name)
1356     {
1357       request_set_header (req, "Content-Type",
1358                           "application/x-www-form-urlencoded", rel_none);
1359       if (opt.post_data)
1360         post_data_size = strlen (opt.post_data);
1361       else
1362         {
1363           post_data_size = file_size (opt.post_file_name);
1364           if (post_data_size == -1)
1365             {
1366               logprintf (LOG_NOTQUIET, "POST data file missing: %s\n",
1367                          opt.post_file_name);
1368               post_data_size = 0;
1369             }
1370         }
1371       request_set_header (req, "Content-Length",
1372                           xstrdup (number_to_static_string (post_data_size)),
1373                           rel_value);
1374     }
1375
1376   /* Add the user headers. */
1377   if (opt.user_headers)
1378     {
1379       int i;
1380       for (i = 0; opt.user_headers[i]; i++)
1381         request_set_user_header (req, opt.user_headers[i]);
1382     }
1383
1384  retry_with_auth:
1385   /* We need to come back here when the initial attempt to retrieve
1386      without authorization header fails.  (Expected to happen at least
1387      for the Digest authorization scheme.)  */
1388
1389   keep_alive = 0;
1390
1391   /* Establish the connection.  */
1392
1393   if (!inhibit_keep_alive)
1394     {
1395       /* Look for a persistent connection to target host, unless a
1396          proxy is used.  The exception is when SSL is in use, in which
1397          case the proxy is nothing but a passthrough to the target
1398          host, registered as a connection to the latter.  */
1399       struct url *relevant = conn;
1400 #ifdef HAVE_SSL
1401       if (u->scheme == SCHEME_HTTPS)
1402         relevant = u;
1403 #endif
1404
1405       if (persistent_available_p (relevant->host, relevant->port,
1406 #ifdef HAVE_SSL
1407                                   relevant->scheme == SCHEME_HTTPS,
1408 #else
1409                                   0,
1410 #endif
1411                                   &host_lookup_failed))
1412         {
1413           sock = pconn.socket;
1414           using_ssl = pconn.ssl;
1415           logprintf (LOG_VERBOSE, _("Reusing existing connection to %s:%d.\n"),
1416                      escnonprint (pconn.host), pconn.port);
1417           DEBUGP (("Reusing fd %d.\n", sock));
1418           if (pconn.authorized)
1419             /* If the connection is already authorized, the "Basic"
1420                authorization added by code above is unnecessary and
1421                only hurts us.  */
1422             request_remove_header (req, "Authorization");
1423         }
1424     }
1425
1426   if (sock < 0)
1427     {
1428       /* In its current implementation, persistent_available_p will
1429          look up conn->host in some cases.  If that lookup failed, we
1430          don't need to bother with connect_to_host.  */
1431       if (host_lookup_failed)
1432         {
1433           request_free (req);
1434           return HOSTERR;
1435         }
1436
1437       sock = connect_to_host (conn->host, conn->port);
1438       if (sock == E_HOST)
1439         {
1440           request_free (req);
1441           return HOSTERR;
1442         }
1443       else if (sock < 0)
1444         {
1445           request_free (req);
1446           return (retryable_socket_connect_error (errno)
1447                   ? CONERROR : CONIMPOSSIBLE);
1448         }
1449
1450 #ifdef HAVE_SSL
1451       if (proxy && u->scheme == SCHEME_HTTPS)
1452         {
1453           /* When requesting SSL URLs through proxies, use the
1454              CONNECT method to request passthrough.  */
1455           struct request *connreq = request_new ();
1456           request_set_method (connreq, "CONNECT",
1457                               aprintf ("%s:%d", u->host, u->port));
1458           if (proxyauth)
1459             {
1460               request_set_header (connreq, "Proxy-Authorization",
1461                                   proxyauth, rel_value);
1462               /* Now that PROXYAUTH is part of the CONNECT request,
1463                  zero it out so we don't send proxy authorization with
1464                  the regular request below.  */
1465               proxyauth = NULL;
1466             }
1467
1468           write_error = request_send (connreq, sock);
1469           request_free (connreq);
1470           if (write_error < 0)
1471             {
1472               logprintf (LOG_VERBOSE, _("Failed writing to proxy: %s.\n"),
1473                          strerror (errno));
1474               CLOSE_INVALIDATE (sock);
1475               return WRITEFAILED;
1476             }
1477
1478           head = read_http_response_head (sock);
1479           if (!head)
1480             {
1481               logprintf (LOG_VERBOSE, _("Failed reading proxy response: %s\n"),
1482                          strerror (errno));
1483               CLOSE_INVALIDATE (sock);
1484               return HERR;
1485             }
1486           message = NULL;
1487           if (!*head)
1488             {
1489               xfree (head);
1490               goto failed_tunnel;
1491             }
1492           DEBUGP (("proxy responded with: [%s]\n", head));
1493
1494           resp = resp_new (head);
1495           statcode = resp_status (resp, &message);
1496           resp_free (resp);
1497           xfree (head);
1498           if (statcode != 200)
1499             {
1500             failed_tunnel:
1501               logprintf (LOG_NOTQUIET, _("Proxy tunneling failed: %s"),
1502                          message ? escnonprint (message) : "?");
1503               xfree_null (message);
1504               return CONSSLERR;
1505             }
1506           xfree_null (message);
1507
1508           /* SOCK is now *really* connected to u->host, so update CONN
1509              to reflect this.  That way register_persistent will
1510              register SOCK as being connected to u->host:u->port.  */
1511           conn = u;
1512         }
1513
1514       if (conn->scheme == SCHEME_HTTPS)
1515         {
1516           if (!ssl_connect (sock))
1517             {
1518               fd_close (sock);
1519               return CONSSLERR;
1520             }
1521           using_ssl = 1;
1522         }
1523 #endif /* HAVE_SSL */
1524     }
1525
1526   /* Send the request to server.  */
1527   write_error = request_send (req, sock);
1528
1529   if (write_error >= 0)
1530     {
1531       if (opt.post_data)
1532         {
1533           DEBUGP (("[POST data: %s]\n", opt.post_data));
1534           write_error = fd_write (sock, opt.post_data, post_data_size, -1);
1535         }
1536       else if (opt.post_file_name && post_data_size != 0)
1537         write_error = post_file (sock, opt.post_file_name, post_data_size);
1538     }
1539
1540   if (write_error < 0)
1541     {
1542       logprintf (LOG_VERBOSE, _("Failed writing HTTP request: %s.\n"),
1543                  strerror (errno));
1544       CLOSE_INVALIDATE (sock);
1545       request_free (req);
1546       return WRITEFAILED;
1547     }
1548   logprintf (LOG_VERBOSE, _("%s request sent, awaiting response... "),
1549              proxy ? "Proxy" : "HTTP");
1550   contlen = -1;
1551   contrange = 0;
1552   *dt &= ~RETROKF;
1553
1554   head = read_http_response_head (sock);
1555   if (!head)
1556     {
1557       if (errno == 0)
1558         {
1559           logputs (LOG_NOTQUIET, _("No data received.\n"));
1560           CLOSE_INVALIDATE (sock);
1561           request_free (req);
1562           return HEOF;
1563         }
1564       else
1565         {
1566           logprintf (LOG_NOTQUIET, _("Read error (%s) in headers.\n"),
1567                      strerror (errno));
1568           CLOSE_INVALIDATE (sock);
1569           request_free (req);
1570           return HERR;
1571         }
1572     }
1573   DEBUGP (("\n---response begin---\n%s---response end---\n", head));
1574
1575   resp = resp_new (head);
1576
1577   /* Check for status line.  */
1578   message = NULL;
1579   statcode = resp_status (resp, &message);
1580   if (!opt.server_response)
1581     logprintf (LOG_VERBOSE, "%2d %s\n", statcode,
1582                message ? escnonprint (message) : "");
1583   else
1584     {
1585       logprintf (LOG_VERBOSE, "\n");
1586       print_server_response (resp, "  ");
1587     }
1588
1589   if (!opt.ignore_length
1590       && resp_header_copy (resp, "Content-Length", hdrval, sizeof (hdrval)))
1591     {
1592       wgint parsed;
1593       errno = 0;
1594       parsed = str_to_wgint (hdrval, NULL, 10);
1595       if (parsed == WGINT_MAX && errno == ERANGE)
1596         /* Out of range.
1597            #### If Content-Length is out of range, it most likely
1598            means that the file is larger than 2G and that we're
1599            compiled without LFS.  In that case we should probably
1600            refuse to even attempt to download the file.  */
1601         contlen = -1;
1602       else
1603         contlen = parsed;
1604     }
1605
1606   /* Check for keep-alive related responses. */
1607   if (!inhibit_keep_alive && contlen != -1)
1608     {
1609       if (resp_header_copy (resp, "Keep-Alive", NULL, 0))
1610         keep_alive = 1;
1611       else if (resp_header_copy (resp, "Connection", hdrval, sizeof (hdrval)))
1612         {
1613           if (0 == strcasecmp (hdrval, "Keep-Alive"))
1614             keep_alive = 1;
1615         }
1616     }
1617   if (keep_alive)
1618     /* The server has promised that it will not close the connection
1619        when we're done.  This means that we can register it.  */
1620     register_persistent (conn->host, conn->port, sock, using_ssl);
1621
1622   if (statcode == HTTP_STATUS_UNAUTHORIZED)
1623     {
1624       /* Authorization is required.  */
1625       if (skip_short_body (sock, contlen))
1626         CLOSE_FINISH (sock);
1627       else
1628         CLOSE_INVALIDATE (sock);
1629       pconn.authorized = 0;
1630       if (auth_finished || !(user && passwd))
1631         {
1632           /* If we have tried it already, then there is not point
1633              retrying it.  */
1634           logputs (LOG_NOTQUIET, _("Authorization failed.\n"));
1635         }
1636       else
1637         {
1638           /* IIS sometimes sends two instances of WWW-Authenticate
1639              header, one with the keyword "negotiate", and other with
1640              useful data.  Loop over all occurrences of this header
1641              and use the one we recognize.  */
1642           int wapos;
1643           const char *wabeg, *waend;
1644           char *www_authenticate = NULL;
1645           for (wapos = 0;
1646                (wapos = resp_header_locate (resp, "WWW-Authenticate", wapos,
1647                                             &wabeg, &waend)) != -1;
1648                ++wapos)
1649             if (known_authentication_scheme_p (wabeg, waend))
1650               {
1651                 www_authenticate = strdupdelim (wabeg, waend);
1652                 break;
1653               }
1654           /* If the authentication header is missing or recognized, or
1655              if the authentication scheme is "Basic" (which we send by
1656              default), there's no sense in retrying.  */
1657           if (!www_authenticate
1658               || BEGINS_WITH (www_authenticate, "Basic"))
1659             {
1660               xfree_null (www_authenticate);
1661               logputs (LOG_NOTQUIET, _("Unknown authentication scheme.\n"));
1662             }
1663           else
1664             {
1665               char *pth;
1666               pth = url_full_path (u);
1667               request_set_header (req, "Authorization",
1668                                   create_authorization_line (www_authenticate,
1669                                                              user, passwd,
1670                                                              request_method (req),
1671                                                              pth,
1672                                                              &auth_finished),
1673                                   rel_value);
1674               if (BEGINS_WITH (www_authenticate, "NTLM"))
1675                 ntlm_seen = 1;
1676               xfree (pth);
1677               xfree (www_authenticate);
1678               goto retry_with_auth;
1679             }
1680         }
1681       request_free (req);
1682       return AUTHFAILED;
1683     }
1684   else /* statcode != HTTP_STATUS_UNAUTHORIZED */
1685     {
1686       /* Kludge: if NTLM is used, mark the TCP connection as authorized. */
1687       if (ntlm_seen)
1688         pconn.authorized = 1;
1689     }
1690   request_free (req);
1691
1692   hs->statcode = statcode;
1693   if (statcode == -1)
1694     hs->error = xstrdup (_("Malformed status line"));
1695   else if (!*message)
1696     hs->error = xstrdup (_("(no description)"));
1697   else
1698     hs->error = xstrdup (message);
1699   xfree (message);
1700
1701   type = resp_header_strdup (resp, "Content-Type");
1702   if (type)
1703     {
1704       char *tmp = strchr (type, ';');
1705       if (tmp)
1706         {
1707           while (tmp > type && ISSPACE (tmp[-1]))
1708             --tmp;
1709           *tmp = '\0';
1710         }
1711     }
1712   hs->newloc = resp_header_strdup (resp, "Location");
1713   hs->remote_time = resp_header_strdup (resp, "Last-Modified");
1714
1715   /* Handle (possibly multiple instances of) the Set-Cookie header. */
1716   {
1717     int scpos;
1718     const char *scbeg, *scend;
1719     /* The jar should have been created by now. */
1720     assert (wget_cookie_jar != NULL);
1721     for (scpos = 0;
1722          (scpos = resp_header_locate (resp, "Set-Cookie", scpos,
1723                                       &scbeg, &scend)) != -1;
1724          ++scpos)
1725       {
1726         char *set_cookie = strdupdelim (scbeg, scend);
1727         cookie_handle_set_cookie (wget_cookie_jar, u->host, u->port, u->path,
1728                                   set_cookie);
1729         xfree (set_cookie);
1730       }
1731   }
1732
1733   if (resp_header_copy (resp, "Content-Range", hdrval, sizeof (hdrval)))
1734     {
1735       wgint first_byte_pos, last_byte_pos, entity_length;
1736       if (parse_content_range (hdrval, &first_byte_pos, &last_byte_pos,
1737                                &entity_length))
1738         contrange = first_byte_pos;
1739     }
1740   resp_free (resp);
1741   xfree (head);
1742
1743   /* 20x responses are counted among successful by default.  */
1744   if (H_20X (statcode))
1745     *dt |= RETROKF;
1746
1747   /* Return if redirected.  */
1748   if (H_REDIRECTED (statcode) || statcode == HTTP_STATUS_MULTIPLE_CHOICES)
1749     {
1750       /* RFC2068 says that in case of the 300 (multiple choices)
1751          response, the server can output a preferred URL through
1752          `Location' header; otherwise, the request should be treated
1753          like GET.  So, if the location is set, it will be a
1754          redirection; otherwise, just proceed normally.  */
1755       if (statcode == HTTP_STATUS_MULTIPLE_CHOICES && !hs->newloc)
1756         *dt |= RETROKF;
1757       else
1758         {
1759           logprintf (LOG_VERBOSE,
1760                      _("Location: %s%s\n"),
1761                      hs->newloc ? escnonprint_uri (hs->newloc) : _("unspecified"),
1762                      hs->newloc ? _(" [following]") : "");
1763           if (keep_alive)
1764             {
1765               if (skip_short_body (sock, contlen))
1766                 CLOSE_FINISH (sock);
1767               else
1768                 CLOSE_INVALIDATE (sock);
1769             }
1770           xfree_null (type);
1771           return NEWLOCATION;
1772         }
1773     }
1774
1775   /* If content-type is not given, assume text/html.  This is because
1776      of the multitude of broken CGI's that "forget" to generate the
1777      content-type.  */
1778   if (!type ||
1779         0 == strncasecmp (type, TEXTHTML_S, strlen (TEXTHTML_S)) ||
1780         0 == strncasecmp (type, TEXTXHTML_S, strlen (TEXTXHTML_S)))
1781     *dt |= TEXTHTML;
1782   else
1783     *dt &= ~TEXTHTML;
1784
1785   if (opt.html_extension && (*dt & TEXTHTML))
1786     /* -E / --html-extension / html_extension = on was specified, and this is a
1787        text/html file.  If some case-insensitive variation on ".htm[l]" isn't
1788        already the file's suffix, tack on ".html". */
1789     {
1790       char*  last_period_in_local_filename = strrchr(*hs->local_file, '.');
1791
1792       if (last_period_in_local_filename == NULL
1793           || !(0 == strcasecmp (last_period_in_local_filename, ".htm")
1794                || 0 == strcasecmp (last_period_in_local_filename, ".html")))
1795         {
1796           size_t  local_filename_len = strlen(*hs->local_file);
1797           
1798           *hs->local_file = xrealloc(*hs->local_file,
1799                                      local_filename_len + sizeof(".html"));
1800           strcpy(*hs->local_file + local_filename_len, ".html");
1801
1802           *dt |= ADDED_HTML_EXTENSION;
1803         }
1804     }
1805
1806   if (statcode == HTTP_STATUS_RANGE_NOT_SATISFIABLE)
1807     {
1808       /* If `-c' is in use and the file has been fully downloaded (or
1809          the remote file has shrunk), Wget effectively requests bytes
1810          after the end of file and the server response with 416.  */
1811       logputs (LOG_VERBOSE, _("\
1812 \n    The file is already fully retrieved; nothing to do.\n\n"));
1813       /* In case the caller inspects. */
1814       hs->len = contlen;
1815       hs->res = 0;
1816       /* Mark as successfully retrieved. */
1817       *dt |= RETROKF;
1818       xfree_null (type);
1819       CLOSE_INVALIDATE (sock);  /* would be CLOSE_FINISH, but there
1820                                    might be more bytes in the body. */
1821       return RETRUNNEEDED;
1822     }
1823   if ((contrange != 0 && contrange != hs->restval)
1824       || (H_PARTIAL (statcode) && !contrange))
1825     {
1826       /* The Range request was somehow misunderstood by the server.
1827          Bail out.  */
1828       xfree_null (type);
1829       CLOSE_INVALIDATE (sock);
1830       return RANGEERR;
1831     }
1832   hs->contlen = contlen + contrange;
1833
1834   if (opt.verbose)
1835     {
1836       if (*dt & RETROKF)
1837         {
1838           /* No need to print this output if the body won't be
1839              downloaded at all, or if the original server response is
1840              printed.  */
1841           logputs (LOG_VERBOSE, _("Length: "));
1842           if (contlen != -1)
1843             {
1844               logputs (LOG_VERBOSE, with_thousand_seps (contlen + contrange));
1845               if (contlen + contrange >= 1024)
1846                 logprintf (LOG_VERBOSE, " (%s)",
1847                            human_readable (contlen + contrange));
1848               if (contrange)
1849                 {
1850                   if (contlen >= 1024)
1851                     logprintf (LOG_VERBOSE, _(", %s (%s) remaining"),
1852                                with_thousand_seps (contlen),
1853                                human_readable (contlen));
1854                   else
1855                     logprintf (LOG_VERBOSE, _(", %s remaining"),
1856                                with_thousand_seps (contlen));
1857                 }
1858             }
1859           else
1860             logputs (LOG_VERBOSE,
1861                      opt.ignore_length ? _("ignored") : _("unspecified"));
1862           if (type)
1863             logprintf (LOG_VERBOSE, " [%s]\n", escnonprint (type));
1864           else
1865             logputs (LOG_VERBOSE, "\n");
1866         }
1867     }
1868   xfree_null (type);
1869   type = NULL;                  /* We don't need it any more.  */
1870
1871   /* Return if we have no intention of further downloading.  */
1872   if (!(*dt & RETROKF) || (*dt & HEAD_ONLY))
1873     {
1874       /* In case the caller cares to look...  */
1875       hs->len = 0;
1876       hs->res = 0;
1877       xfree_null (type);
1878       /* Pre-1.10 Wget used CLOSE_INVALIDATE here.  Now we trust the
1879          servers not to send body in response to a HEAD request.  If
1880          you encounter such a server (more likely a broken CGI), use
1881          `--no-http-keep-alive'.  */
1882       CLOSE_FINISH (sock);
1883       return RETRFINISHED;
1884     }
1885
1886   /* Open the local file.  */
1887   if (!output_stream)
1888     {
1889       mkalldirs (*hs->local_file);
1890       if (opt.backups)
1891         rotate_backups (*hs->local_file);
1892       if (hs->restval)
1893         fp = fopen (*hs->local_file, "ab");
1894       else if (opt.noclobber || opt.always_rest || opt.timestamping || opt.dirstruct
1895                || opt.output_document)
1896         fp = fopen (*hs->local_file, "wb");
1897       else
1898         {
1899           fp = fopen_excl (*hs->local_file, 1);
1900           if (!fp && errno == EEXIST)
1901             {
1902               /* We cannot just invent a new name and use it (which is
1903                  what functions like unique_create typically do)
1904                  because we told the user we'd use this name.
1905                  Instead, return and retry the download.  */
1906               logprintf (LOG_NOTQUIET,
1907                          _("%s has sprung into existence.\n"),
1908                          *hs->local_file);
1909               CLOSE_INVALIDATE (sock);
1910               return FOPEN_EXCL_ERR;
1911             }
1912         }
1913       if (!fp)
1914         {
1915           logprintf (LOG_NOTQUIET, "%s: %s\n", *hs->local_file, strerror (errno));
1916           CLOSE_INVALIDATE (sock);
1917           return FOPENERR;
1918         }
1919     }
1920   else
1921     fp = output_stream;
1922
1923   /* #### This confuses the timestamping code that checks for file
1924      size.  Maybe we should save some additional information?  */
1925   if (opt.save_headers)
1926     fwrite (head, 1, strlen (head), fp);
1927
1928   /* Download the request body.  */
1929   flags = 0;
1930   if (keep_alive)
1931     flags |= rb_read_exactly;
1932   if (hs->restval > 0 && contrange == 0)
1933     /* If the server ignored our range request, instruct fd_read_body
1934        to skip the first RESTVAL bytes of body.  */
1935     flags |= rb_skip_startpos;
1936   hs->len = hs->restval;
1937   hs->rd_size = 0;
1938   hs->res = fd_read_body (sock, fp, contlen != -1 ? contlen : 0,
1939                           hs->restval, &hs->rd_size, &hs->len, &hs->dltime,
1940                           flags);
1941
1942   if (hs->res >= 0)
1943     CLOSE_FINISH (sock);
1944   else
1945     CLOSE_INVALIDATE (sock);
1946
1947   {
1948     /* Close or flush the file.  We have to be careful to check for
1949        error here.  Checking the result of fwrite() is not enough --
1950        errors could go unnoticed!  */
1951     int flush_res;
1952     if (!output_stream)
1953       flush_res = fclose (fp);
1954     else
1955       flush_res = fflush (fp);
1956     if (flush_res == EOF)
1957       hs->res = -2;
1958   }
1959   if (hs->res == -2)
1960     return FWRITEERR;
1961   return RETRFINISHED;
1962 }
1963
1964 /* The genuine HTTP loop!  This is the part where the retrieval is
1965    retried, and retried, and retried, and...  */
1966 uerr_t
1967 http_loop (struct url *u, char **newloc, char **local_file, const char *referer,
1968            int *dt, struct url *proxy)
1969 {
1970   int count;
1971   int use_ts, got_head = 0;     /* time-stamping info */
1972   char *filename_plus_orig_suffix;
1973   char *local_filename = NULL;
1974   char *tms, *locf, *tmrate;
1975   uerr_t err;
1976   time_t tml = -1, tmr = -1;    /* local and remote time-stamps */
1977   wgint local_size = 0;         /* the size of the local file */
1978   size_t filename_len;
1979   struct http_stat hstat;       /* HTTP status */
1980   struct_stat st;
1981   char *dummy = NULL;
1982
1983   /* This used to be done in main(), but it's a better idea to do it
1984      here so that we don't go through the hoops if we're just using
1985      FTP or whatever. */
1986   if (opt.cookies)
1987     {
1988       if (!wget_cookie_jar)
1989         wget_cookie_jar = cookie_jar_new ();
1990       if (opt.cookies_input && !cookies_loaded_p)
1991         {
1992           cookie_jar_load (wget_cookie_jar, opt.cookies_input);
1993           cookies_loaded_p = 1;
1994         }
1995     }
1996
1997   *newloc = NULL;
1998
1999   /* Warn on (likely bogus) wildcard usage in HTTP.  Don't use
2000      has_wildcards_p because it would also warn on `?', and we know that
2001      shows up in CGI paths a *lot*.  */
2002   if (strchr (u->url, '*'))
2003     logputs (LOG_VERBOSE, _("Warning: wildcards not supported in HTTP.\n"));
2004
2005   xzero (hstat);
2006
2007   /* Determine the local filename.  */
2008   if (local_file && *local_file)
2009     hstat.local_file = local_file;
2010   else if (local_file && !opt.output_document)
2011     {
2012       *local_file = url_file_name (u);
2013       hstat.local_file = local_file;
2014     }
2015   else
2016     {
2017       dummy = url_file_name (u);
2018       hstat.local_file = &dummy;
2019       /* be honest about where we will save the file */
2020       if (local_file && opt.output_document)
2021         *local_file = HYPHENP (opt.output_document) ? NULL : xstrdup (opt.output_document);
2022     }
2023
2024   if (!opt.output_document)
2025     locf = *hstat.local_file;
2026   else
2027     locf = opt.output_document;
2028
2029   hstat.referer = referer;
2030
2031   filename_len = strlen (*hstat.local_file);
2032   filename_plus_orig_suffix = alloca (filename_len + sizeof (".orig"));
2033
2034   if (opt.noclobber && file_exists_p (*hstat.local_file))
2035     {
2036       /* If opt.noclobber is turned on and file already exists, do not
2037          retrieve the file */
2038       logprintf (LOG_VERBOSE, _("\
2039 File `%s' already there, will not retrieve.\n"), *hstat.local_file);
2040       /* If the file is there, we suppose it's retrieved OK.  */
2041       *dt |= RETROKF;
2042
2043       /* #### Bogusness alert.  */
2044       /* If its suffix is "html" or "htm" or similar, assume text/html.  */
2045       if (has_html_suffix_p (*hstat.local_file))
2046         *dt |= TEXTHTML;
2047
2048       xfree_null (dummy);
2049       return RETROK;
2050     }
2051
2052   use_ts = 0;
2053   if (opt.timestamping)
2054     {
2055       int local_dot_orig_file_exists = 0;
2056
2057       if (opt.backup_converted)
2058         /* If -K is specified, we'll act on the assumption that it was specified
2059            last time these files were downloaded as well, and instead of just
2060            comparing local file X against server file X, we'll compare local
2061            file X.orig (if extant, else X) against server file X.  If -K
2062            _wasn't_ specified last time, or the server contains files called
2063            *.orig, -N will be back to not operating correctly with -k. */
2064         {
2065           /* Would a single s[n]printf() call be faster?  --dan
2066
2067              Definitely not.  sprintf() is horribly slow.  It's a
2068              different question whether the difference between the two
2069              affects a program.  Usually I'd say "no", but at one
2070              point I profiled Wget, and found that a measurable and
2071              non-negligible amount of time was lost calling sprintf()
2072              in url.c.  Replacing sprintf with inline calls to
2073              strcpy() and number_to_string() made a difference.
2074              --hniksic */
2075           memcpy (filename_plus_orig_suffix, *hstat.local_file, filename_len);
2076           memcpy (filename_plus_orig_suffix + filename_len,
2077                   ".orig", sizeof (".orig"));
2078
2079           /* Try to stat() the .orig file. */
2080           if (stat (filename_plus_orig_suffix, &st) == 0)
2081             {
2082               local_dot_orig_file_exists = 1;
2083               local_filename = filename_plus_orig_suffix;
2084             }
2085         }      
2086
2087       if (!local_dot_orig_file_exists)
2088         /* Couldn't stat() <file>.orig, so try to stat() <file>. */
2089         if (stat (*hstat.local_file, &st) == 0)
2090           local_filename = *hstat.local_file;
2091
2092       if (local_filename != NULL)
2093         /* There was a local file, so we'll check later to see if the version
2094            the server has is the same version we already have, allowing us to
2095            skip a download. */
2096         {
2097           use_ts = 1;
2098           tml = st.st_mtime;
2099 #ifdef WINDOWS
2100           /* Modification time granularity is 2 seconds for Windows, so
2101              increase local time by 1 second for later comparison. */
2102           tml++;
2103 #endif
2104           local_size = st.st_size;
2105           got_head = 0;
2106         }
2107     }
2108   /* Reset the counter.  */
2109   count = 0;
2110   *dt = 0;
2111   /* THE loop */
2112   do
2113     {
2114       /* Increment the pass counter.  */
2115       ++count;
2116       sleep_between_retrievals (count);
2117       /* Get the current time string.  */
2118       tms = time_str (NULL);
2119       /* Print fetch message, if opt.verbose.  */
2120       if (opt.verbose)
2121         {
2122           char *hurl = url_string (u, 1);
2123           char tmp[256];
2124           strcpy (tmp, "        ");
2125           if (count > 1)
2126             sprintf (tmp, _("(try:%2d)"), count);
2127           logprintf (LOG_VERBOSE, "--%s--  %s\n  %s => `%s'\n",
2128                      tms, hurl, tmp, locf);
2129 #ifdef WINDOWS
2130           ws_changetitle (hurl);
2131 #endif
2132           xfree (hurl);
2133         }
2134
2135       /* Default document type is empty.  However, if spider mode is
2136          on or time-stamping is employed, HEAD_ONLY commands is
2137          encoded within *dt.  */
2138       if (opt.spider || (use_ts && !got_head))
2139         *dt |= HEAD_ONLY;
2140       else
2141         *dt &= ~HEAD_ONLY;
2142
2143       /* Decide whether or not to restart.  */
2144       hstat.restval = 0;
2145       if (count > 1)
2146         hstat.restval = hstat.len; /* continue where we left off */
2147       else if (opt.always_rest
2148                && stat (locf, &st) == 0
2149                && S_ISREG (st.st_mode))
2150         hstat.restval = st.st_size;
2151
2152       /* Decide whether to send the no-cache directive.  We send it in
2153          two cases:
2154            a) we're using a proxy, and we're past our first retrieval.
2155               Some proxies are notorious for caching incomplete data, so
2156               we require a fresh get.
2157            b) caching is explicitly inhibited. */
2158       if ((proxy && count > 1)  /* a */
2159           || !opt.allow_cache   /* b */
2160           )
2161         *dt |= SEND_NOCACHE;
2162       else
2163         *dt &= ~SEND_NOCACHE;
2164
2165       /* Try fetching the document, or at least its head.  */
2166       err = gethttp (u, &hstat, dt, proxy);
2167
2168       /* It's unfortunate that wget determines the local filename before finding
2169          out the Content-Type of the file.  Barring a major restructuring of the
2170          code, we need to re-set locf here, since gethttp() may have xrealloc()d
2171          *hstat.local_file to tack on ".html". */
2172       if (!opt.output_document)
2173         locf = *hstat.local_file;
2174
2175       /* Time?  */
2176       tms = time_str (NULL);
2177       /* Get the new location (with or without the redirection).  */
2178       if (hstat.newloc)
2179         *newloc = xstrdup (hstat.newloc);
2180       switch (err)
2181         {
2182         case HERR: case HEOF: case CONSOCKERR: case CONCLOSED:
2183         case CONERROR: case READERR: case WRITEFAILED:
2184         case RANGEERR: case FOPEN_EXCL_ERR:
2185           /* Non-fatal errors continue executing the loop, which will
2186              bring them to "while" statement at the end, to judge
2187              whether the number of tries was exceeded.  */
2188           free_hstat (&hstat);
2189           printwhat (count, opt.ntry);
2190           if (err == FOPEN_EXCL_ERR)
2191             {
2192               /* Re-determine the file name. */
2193               if (local_file && *local_file)
2194                 {
2195                   xfree (*local_file);
2196                   *local_file = url_file_name (u);
2197                   hstat.local_file = local_file;
2198                 }
2199               else
2200                 {
2201                   xfree (dummy);
2202                   dummy = url_file_name (u);
2203                   hstat.local_file = &dummy;
2204                 }
2205               /* be honest about where we will save the file */
2206               if (local_file && opt.output_document)
2207                 *local_file = HYPHENP (opt.output_document) ? NULL : xstrdup (opt.output_document);
2208               if (!opt.output_document)
2209                 locf = *hstat.local_file;
2210               else
2211                 locf = opt.output_document;
2212             }
2213           continue;
2214           break;
2215         case HOSTERR: case CONIMPOSSIBLE: case PROXERR: case AUTHFAILED: 
2216         case SSLERRCTXCREATE: case CONTNOTSUPPORTED:
2217           /* Fatal errors just return from the function.  */
2218           free_hstat (&hstat);
2219           xfree_null (dummy);
2220           return err;
2221           break;
2222         case FWRITEERR: case FOPENERR:
2223           /* Another fatal error.  */
2224           logputs (LOG_VERBOSE, "\n");
2225           logprintf (LOG_NOTQUIET, _("Cannot write to `%s' (%s).\n"),
2226                      *hstat.local_file, strerror (errno));
2227           free_hstat (&hstat);
2228           xfree_null (dummy);
2229           return err;
2230           break;
2231         case CONSSLERR:
2232           /* Another fatal error.  */
2233           logputs (LOG_VERBOSE, "\n");
2234           logprintf (LOG_NOTQUIET, _("Unable to establish SSL connection.\n"));
2235           free_hstat (&hstat);
2236           xfree_null (dummy);
2237           return err;
2238           break;
2239         case NEWLOCATION:
2240           /* Return the new location to the caller.  */
2241           if (!hstat.newloc)
2242             {
2243               logprintf (LOG_NOTQUIET,
2244                          _("ERROR: Redirection (%d) without location.\n"),
2245                          hstat.statcode);
2246               free_hstat (&hstat);
2247               xfree_null (dummy);
2248               return WRONGCODE;
2249             }
2250           free_hstat (&hstat);
2251           xfree_null (dummy);
2252           return NEWLOCATION;
2253           break;
2254         case RETRUNNEEDED:
2255           /* The file was already fully retrieved. */
2256           free_hstat (&hstat);
2257           xfree_null (dummy);
2258           return RETROK;
2259           break;
2260         case RETRFINISHED:
2261           /* Deal with you later.  */
2262           break;
2263         default:
2264           /* All possibilities should have been exhausted.  */
2265           abort ();
2266         }
2267       if (!(*dt & RETROKF))
2268         {
2269           if (!opt.verbose)
2270             {
2271               /* #### Ugly ugly ugly! */
2272               char *hurl = url_string (u, 1);
2273               logprintf (LOG_NONVERBOSE, "%s:\n", hurl);
2274               xfree (hurl);
2275             }
2276           logprintf (LOG_NOTQUIET, _("%s ERROR %d: %s.\n"),
2277                      tms, hstat.statcode, escnonprint (hstat.error));
2278           logputs (LOG_VERBOSE, "\n");
2279           free_hstat (&hstat);
2280           xfree_null (dummy);
2281           return WRONGCODE;
2282         }
2283
2284       /* Did we get the time-stamp?  */
2285       if (!got_head)
2286         {
2287           if (opt.timestamping && !hstat.remote_time)
2288             {
2289               logputs (LOG_NOTQUIET, _("\
2290 Last-modified header missing -- time-stamps turned off.\n"));
2291             }
2292           else if (hstat.remote_time)
2293             {
2294               /* Convert the date-string into struct tm.  */
2295               tmr = http_atotm (hstat.remote_time);
2296               if (tmr == (time_t) (-1))
2297                 logputs (LOG_VERBOSE, _("\
2298 Last-modified header invalid -- time-stamp ignored.\n"));
2299             }
2300         }
2301
2302       /* The time-stamping section.  */
2303       if (use_ts)
2304         {
2305           got_head = 1;
2306           *dt &= ~HEAD_ONLY;
2307           use_ts = 0;           /* no more time-stamping */
2308           count = 0;            /* the retrieve count for HEAD is
2309                                    reset */
2310           if (hstat.remote_time && tmr != (time_t) (-1))
2311             {
2312               /* Now time-stamping can be used validly.  Time-stamping
2313                  means that if the sizes of the local and remote file
2314                  match, and local file is newer than the remote file,
2315                  it will not be retrieved.  Otherwise, the normal
2316                  download procedure is resumed.  */
2317               if (tml >= tmr &&
2318                   (hstat.contlen == -1 || local_size == hstat.contlen))
2319                 {
2320                   logprintf (LOG_VERBOSE, _("\
2321 Server file no newer than local file `%s' -- not retrieving.\n\n"),
2322                              local_filename);
2323                   free_hstat (&hstat);
2324                   xfree_null (dummy);
2325                   return RETROK;
2326                 }
2327               else if (tml >= tmr)
2328                 logprintf (LOG_VERBOSE, _("\
2329 The sizes do not match (local %s) -- retrieving.\n"),
2330                            number_to_static_string (local_size));
2331               else
2332                 logputs (LOG_VERBOSE,
2333                          _("Remote file is newer, retrieving.\n"));
2334             }
2335           free_hstat (&hstat);
2336           continue;
2337         }
2338       if ((tmr != (time_t) (-1))
2339           && !opt.spider
2340           && ((hstat.len == hstat.contlen) ||
2341               ((hstat.res == 0) &&
2342                ((hstat.contlen == -1) ||
2343                 (hstat.len >= hstat.contlen && !opt.kill_longer)))))
2344         {
2345           /* #### This code repeats in http.c and ftp.c.  Move it to a
2346              function!  */
2347           const char *fl = NULL;
2348           if (opt.output_document)
2349             {
2350               if (output_stream_regular)
2351                 fl = opt.output_document;
2352             }
2353           else
2354             fl = *hstat.local_file;
2355           if (fl)
2356             touch (fl, tmr);
2357         }
2358       /* End of time-stamping section.  */
2359
2360       if (opt.spider)
2361         {
2362           logprintf (LOG_NOTQUIET, "%d %s\n\n", hstat.statcode,
2363                      escnonprint (hstat.error));
2364           xfree_null (dummy);
2365           return RETROK;
2366         }
2367
2368       tmrate = retr_rate (hstat.rd_size, hstat.dltime, 0);
2369
2370       if (hstat.len == hstat.contlen)
2371         {
2372           if (*dt & RETROKF)
2373             {
2374               logprintf (LOG_VERBOSE,
2375                          _("%s (%s) - `%s' saved [%s/%s]\n\n"),
2376                          tms, tmrate, locf,
2377                          number_to_static_string (hstat.len),
2378                          number_to_static_string (hstat.contlen));
2379               logprintf (LOG_NONVERBOSE,
2380                          "%s URL:%s [%s/%s] -> \"%s\" [%d]\n",
2381                          tms, u->url,
2382                          number_to_static_string (hstat.len),
2383                          number_to_static_string (hstat.contlen),
2384                          locf, count);
2385             }
2386           ++opt.numurls;
2387           total_downloaded_bytes += hstat.len;
2388
2389           /* Remember that we downloaded the file for later ".orig" code. */
2390           if (*dt & ADDED_HTML_EXTENSION)
2391             downloaded_file(FILE_DOWNLOADED_AND_HTML_EXTENSION_ADDED, locf);
2392           else
2393             downloaded_file(FILE_DOWNLOADED_NORMALLY, locf);
2394
2395           free_hstat (&hstat);
2396           xfree_null (dummy);
2397           return RETROK;
2398         }
2399       else if (hstat.res == 0) /* No read error */
2400         {
2401           if (hstat.contlen == -1)  /* We don't know how much we were supposed
2402                                        to get, so assume we succeeded. */ 
2403             {
2404               if (*dt & RETROKF)
2405                 {
2406                   logprintf (LOG_VERBOSE,
2407                              _("%s (%s) - `%s' saved [%s]\n\n"),
2408                              tms, tmrate, locf,
2409                              number_to_static_string (hstat.len));
2410                   logprintf (LOG_NONVERBOSE,
2411                              "%s URL:%s [%s] -> \"%s\" [%d]\n",
2412                              tms, u->url, number_to_static_string (hstat.len),
2413                              locf, count);
2414                 }
2415               ++opt.numurls;
2416               total_downloaded_bytes += hstat.len;
2417
2418               /* Remember that we downloaded the file for later ".orig" code. */
2419               if (*dt & ADDED_HTML_EXTENSION)
2420                 downloaded_file(FILE_DOWNLOADED_AND_HTML_EXTENSION_ADDED, locf);
2421               else
2422                 downloaded_file(FILE_DOWNLOADED_NORMALLY, locf);
2423               
2424               free_hstat (&hstat);
2425               xfree_null (dummy);
2426               return RETROK;
2427             }
2428           else if (hstat.len < hstat.contlen) /* meaning we lost the
2429                                                  connection too soon */
2430             {
2431               logprintf (LOG_VERBOSE,
2432                          _("%s (%s) - Connection closed at byte %s. "),
2433                          tms, tmrate, number_to_static_string (hstat.len));
2434               printwhat (count, opt.ntry);
2435               free_hstat (&hstat);
2436               continue;
2437             }
2438           else if (!opt.kill_longer) /* meaning we got more than expected */
2439             {
2440               logprintf (LOG_VERBOSE,
2441                          _("%s (%s) - `%s' saved [%s/%s])\n\n"),
2442                          tms, tmrate, locf,
2443                          number_to_static_string (hstat.len),
2444                          number_to_static_string (hstat.contlen));
2445               logprintf (LOG_NONVERBOSE,
2446                          "%s URL:%s [%s/%s] -> \"%s\" [%d]\n",
2447                          tms, u->url,
2448                          number_to_static_string (hstat.len),
2449                          number_to_static_string (hstat.contlen),
2450                          locf, count);
2451               ++opt.numurls;
2452               total_downloaded_bytes += hstat.len;
2453
2454               /* Remember that we downloaded the file for later ".orig" code. */
2455               if (*dt & ADDED_HTML_EXTENSION)
2456                 downloaded_file(FILE_DOWNLOADED_AND_HTML_EXTENSION_ADDED, locf);
2457               else
2458                 downloaded_file(FILE_DOWNLOADED_NORMALLY, locf);
2459               
2460               free_hstat (&hstat);
2461               xfree_null (dummy);
2462               return RETROK;
2463             }
2464           else                  /* the same, but not accepted */
2465             {
2466               logprintf (LOG_VERBOSE,
2467                          _("%s (%s) - Connection closed at byte %s/%s. "),
2468                          tms, tmrate,
2469                          number_to_static_string (hstat.len),
2470                          number_to_static_string (hstat.contlen));
2471               printwhat (count, opt.ntry);
2472               free_hstat (&hstat);
2473               continue;
2474             }
2475         }
2476       else                      /* now hstat.res can only be -1 */
2477         {
2478           if (hstat.contlen == -1)
2479             {
2480               logprintf (LOG_VERBOSE,
2481                          _("%s (%s) - Read error at byte %s (%s)."),
2482                          tms, tmrate, number_to_static_string (hstat.len),
2483                          strerror (errno));
2484               printwhat (count, opt.ntry);
2485               free_hstat (&hstat);
2486               continue;
2487             }
2488           else                  /* hstat.res == -1 and contlen is given */
2489             {
2490               logprintf (LOG_VERBOSE,
2491                          _("%s (%s) - Read error at byte %s/%s (%s). "),
2492                          tms, tmrate,
2493                          number_to_static_string (hstat.len),
2494                          number_to_static_string (hstat.contlen),
2495                          strerror (errno));
2496               printwhat (count, opt.ntry);
2497               free_hstat (&hstat);
2498               continue;
2499             }
2500         }
2501       /* not reached */
2502       break;
2503     }
2504   while (!opt.ntry || (count < opt.ntry));
2505   return TRYLIMEXC;
2506 }
2507 \f
2508 /* Converts struct tm to time_t, assuming the data in tm is UTC rather
2509    than local timezone.
2510
2511    mktime is similar but assumes struct tm, also known as the
2512    "broken-down" form of time, is in local time zone.  mktime_from_utc
2513    uses mktime to make the conversion understanding that an offset
2514    will be introduced by the local time assumption.
2515
2516    mktime_from_utc then measures the introduced offset by applying
2517    gmtime to the initial result and applying mktime to the resulting
2518    "broken-down" form.  The difference between the two mktime results
2519    is the measured offset which is then subtracted from the initial
2520    mktime result to yield a calendar time which is the value returned.
2521
2522    tm_isdst in struct tm is set to 0 to force mktime to introduce a
2523    consistent offset (the non DST offset) since tm and tm+o might be
2524    on opposite sides of a DST change.
2525
2526    Some implementations of mktime return -1 for the nonexistent
2527    localtime hour at the beginning of DST.  In this event, use
2528    mktime(tm - 1hr) + 3600.
2529
2530    Schematically
2531      mktime(tm)   --> t+o
2532      gmtime(t+o)  --> tm+o
2533      mktime(tm+o) --> t+2o
2534      t+o - (t+2o - t+o) = t
2535
2536    Note that glibc contains a function of the same purpose named
2537    `timegm' (reverse of gmtime).  But obviously, it is not universally
2538    available, and unfortunately it is not straightforwardly
2539    extractable for use here.  Perhaps configure should detect timegm
2540    and use it where available.
2541
2542    Contributed by Roger Beeman <beeman@cisco.com>, with the help of
2543    Mark Baushke <mdb@cisco.com> and the rest of the Gurus at CISCO.
2544    Further improved by Roger with assistance from Edward J. Sabol
2545    based on input by Jamie Zawinski.  */
2546
2547 static time_t
2548 mktime_from_utc (struct tm *t)
2549 {
2550   time_t tl, tb;
2551   struct tm *tg;
2552
2553   tl = mktime (t);
2554   if (tl == -1)
2555     {
2556       t->tm_hour--;
2557       tl = mktime (t);
2558       if (tl == -1)
2559         return -1; /* can't deal with output from strptime */
2560       tl += 3600;
2561     }
2562   tg = gmtime (&tl);
2563   tg->tm_isdst = 0;
2564   tb = mktime (tg);
2565   if (tb == -1)
2566     {
2567       tg->tm_hour--;
2568       tb = mktime (tg);
2569       if (tb == -1)
2570         return -1; /* can't deal with output from gmtime */
2571       tb += 3600;
2572     }
2573   return (tl - (tb - tl));
2574 }
2575
2576 /* Check whether the result of strptime() indicates success.
2577    strptime() returns the pointer to how far it got to in the string.
2578    The processing has been successful if the string is at `GMT' or
2579    `+X', or at the end of the string.
2580
2581    In extended regexp parlance, the function returns 1 if P matches
2582    "^ *(GMT|[+-][0-9]|$)", 0 otherwise.  P being NULL (which strptime
2583    can return) is considered a failure and 0 is returned.  */
2584 static int
2585 check_end (const char *p)
2586 {
2587   if (!p)
2588     return 0;
2589   while (ISSPACE (*p))
2590     ++p;
2591   if (!*p
2592       || (p[0] == 'G' && p[1] == 'M' && p[2] == 'T')
2593       || ((p[0] == '+' || p[0] == '-') && ISDIGIT (p[1])))
2594     return 1;
2595   else
2596     return 0;
2597 }
2598
2599 /* Convert the textual specification of time in TIME_STRING to the
2600    number of seconds since the Epoch.
2601
2602    TIME_STRING can be in any of the three formats RFC2068 allows the
2603    HTTP servers to emit -- RFC1123-date, RFC850-date or asctime-date.
2604    Timezones are ignored, and should be GMT.
2605
2606    Return the computed time_t representation, or -1 if the conversion
2607    fails.
2608
2609    This function uses strptime with various string formats for parsing
2610    TIME_STRING.  This results in a parser that is not as lenient in
2611    interpreting TIME_STRING as I would like it to be.  Being based on
2612    strptime, it always allows shortened months, one-digit days, etc.,
2613    but due to the multitude of formats in which time can be
2614    represented, an ideal HTTP time parser would be even more
2615    forgiving.  It should completely ignore things like week days and
2616    concentrate only on the various forms of representing years,
2617    months, days, hours, minutes, and seconds.  For example, it would
2618    be nice if it accepted ISO 8601 out of the box.
2619
2620    I've investigated free and PD code for this purpose, but none was
2621    usable.  getdate was big and unwieldy, and had potential copyright
2622    issues, or so I was informed.  Dr. Marcus Hennecke's atotm(),
2623    distributed with phttpd, is excellent, but we cannot use it because
2624    it is not assigned to the FSF.  So I stuck it with strptime.  */
2625
2626 time_t
2627 http_atotm (const char *time_string)
2628 {
2629   /* NOTE: Solaris strptime man page claims that %n and %t match white
2630      space, but that's not universally available.  Instead, we simply
2631      use ` ' to mean "skip all WS", which works under all strptime
2632      implementations I've tested.  */
2633
2634   static const char *time_formats[] = {
2635     "%a, %d %b %Y %T",          /* RFC1123: Thu, 29 Jan 1998 22:12:57 */
2636     "%A, %d-%b-%y %T",          /* RFC850:  Thursday, 29-Jan-98 22:12:57 */
2637     "%a, %d-%b-%Y %T",          /* pseudo-RFC850:  Thu, 29-Jan-1998 22:12:57
2638                                    (google.com uses this for their cookies.) */
2639     "%a %b %d %T %Y"            /* asctime: Thu Jan 29 22:12:57 1998 */
2640   };
2641
2642   int i;
2643   struct tm t;
2644
2645   /* According to Roger Beeman, we need to initialize tm_isdst, since
2646      strptime won't do it.  */
2647   t.tm_isdst = 0;
2648
2649   /* Note that under foreign locales Solaris strptime() fails to
2650      recognize English dates, which renders this function useless.  We
2651      solve this by being careful not to affect LC_TIME when
2652      initializing locale.
2653
2654      Another solution would be to temporarily set locale to C, invoke
2655      strptime(), and restore it back.  This is slow and dirty,
2656      however, and locale support other than LC_MESSAGES can mess other
2657      things, so I rather chose to stick with just setting LC_MESSAGES.
2658
2659      GNU strptime does not have this problem because it recognizes
2660      both international and local dates.  */
2661
2662   for (i = 0; i < countof (time_formats); i++)
2663     if (check_end (strptime (time_string, time_formats[i], &t)))
2664       return mktime_from_utc (&t);
2665
2666   /* All formats have failed.  */
2667   return -1;
2668 }
2669 \f
2670 /* Authorization support: We support three authorization schemes:
2671
2672    * `Basic' scheme, consisting of base64-ing USER:PASSWORD string;
2673
2674    * `Digest' scheme, added by Junio Hamano <junio@twinsun.com>,
2675    consisting of answering to the server's challenge with the proper
2676    MD5 digests.
2677
2678    * `NTLM' ("NT Lan Manager") scheme, based on code written by Daniel
2679    Stenberg for libcurl.  Like digest, NTLM is based on a
2680    challenge-response mechanism, but unlike digest, it is non-standard
2681    (authenticates TCP connections rather than requests), undocumented
2682    and Microsoft-specific.  */
2683
2684 /* Create the authentication header contents for the `Basic' scheme.
2685    This is done by encoding the string `USER:PASS' in base64 and
2686    prepending `HEADER: Basic ' to it.  */
2687
2688 static char *
2689 basic_authentication_encode (const char *user, const char *passwd)
2690 {
2691   char *t1, *t2;
2692   int len1 = strlen (user) + 1 + strlen (passwd);
2693
2694   t1 = (char *)alloca (len1 + 1);
2695   sprintf (t1, "%s:%s", user, passwd);
2696
2697   t2 = (char *)alloca (BASE64_LENGTH (len1) + 1);
2698   base64_encode (t1, len1, t2);
2699
2700   return concat_strings ("Basic ", t2, (char *) 0);
2701 }
2702
2703 #define SKIP_WS(x) do {                         \
2704   while (ISSPACE (*(x)))                        \
2705     ++(x);                                      \
2706 } while (0)
2707
2708 #ifdef ENABLE_DIGEST
2709 /* Parse HTTP `WWW-Authenticate:' header.  AU points to the beginning
2710    of a field in such a header.  If the field is the one specified by
2711    ATTR_NAME ("realm", "opaque", and "nonce" are used by the current
2712    digest authorization code), extract its value in the (char*)
2713    variable pointed by RET.  Returns negative on a malformed header,
2714    or number of bytes that have been parsed by this call.  */
2715 static int
2716 extract_header_attr (const char *au, const char *attr_name, char **ret)
2717 {
2718   const char *ep;
2719   const char *cp = au;
2720
2721   if (strncmp (cp, attr_name, strlen (attr_name)) == 0)
2722     {
2723       cp += strlen (attr_name);
2724       if (!*cp)
2725         return -1;
2726       SKIP_WS (cp);
2727       if (*cp != '=')
2728         return -1;
2729       if (!*++cp)
2730         return -1;
2731       SKIP_WS (cp);
2732       if (*cp != '\"')
2733         return -1;
2734       if (!*++cp)
2735         return -1;
2736       for (ep = cp; *ep && *ep != '\"'; ep++)
2737         ;
2738       if (!*ep)
2739         return -1;
2740       xfree_null (*ret);
2741       *ret = strdupdelim (cp, ep);
2742       return ep - au + 1;
2743     }
2744   else
2745     return 0;
2746 }
2747
2748 /* Dump the hexadecimal representation of HASH to BUF.  HASH should be
2749    an array of 16 bytes containing the hash keys, and BUF should be a
2750    buffer of 33 writable characters (32 for hex digits plus one for
2751    zero termination).  */
2752 static void
2753 dump_hash (unsigned char *buf, const unsigned char *hash)
2754 {
2755   int i;
2756
2757   for (i = 0; i < MD5_HASHLEN; i++, hash++)
2758     {
2759       *buf++ = XNUM_TO_digit (*hash >> 4);
2760       *buf++ = XNUM_TO_digit (*hash & 0xf);
2761     }
2762   *buf = '\0';
2763 }
2764
2765 /* Take the line apart to find the challenge, and compose a digest
2766    authorization header.  See RFC2069 section 2.1.2.  */
2767 static char *
2768 digest_authentication_encode (const char *au, const char *user,
2769                               const char *passwd, const char *method,
2770                               const char *path)
2771 {
2772   static char *realm, *opaque, *nonce;
2773   static struct {
2774     const char *name;
2775     char **variable;
2776   } options[] = {
2777     { "realm", &realm },
2778     { "opaque", &opaque },
2779     { "nonce", &nonce }
2780   };
2781   char *res;
2782
2783   realm = opaque = nonce = NULL;
2784
2785   au += 6;                      /* skip over `Digest' */
2786   while (*au)
2787     {
2788       int i;
2789
2790       SKIP_WS (au);
2791       for (i = 0; i < countof (options); i++)
2792         {
2793           int skip = extract_header_attr (au, options[i].name,
2794                                           options[i].variable);
2795           if (skip < 0)
2796             {
2797               xfree_null (realm);
2798               xfree_null (opaque);
2799               xfree_null (nonce);
2800               return NULL;
2801             }
2802           else if (skip)
2803             {
2804               au += skip;
2805               break;
2806             }
2807         }
2808       if (i == countof (options))
2809         {
2810           while (*au && *au != '=')
2811             au++;
2812           if (*au && *++au)
2813             {
2814               SKIP_WS (au);
2815               if (*au == '\"')
2816                 {
2817                   au++;
2818                   while (*au && *au != '\"')
2819                     au++;
2820                   if (*au)
2821                     au++;
2822                 }
2823             }
2824         }
2825       while (*au && *au != ',')
2826         au++;
2827       if (*au)
2828         au++;
2829     }
2830   if (!realm || !nonce || !user || !passwd || !path || !method)
2831     {
2832       xfree_null (realm);
2833       xfree_null (opaque);
2834       xfree_null (nonce);
2835       return NULL;
2836     }
2837
2838   /* Calculate the digest value.  */
2839   {
2840     ALLOCA_MD5_CONTEXT (ctx);
2841     unsigned char hash[MD5_HASHLEN];
2842     unsigned char a1buf[MD5_HASHLEN * 2 + 1], a2buf[MD5_HASHLEN * 2 + 1];
2843     unsigned char response_digest[MD5_HASHLEN * 2 + 1];
2844
2845     /* A1BUF = H(user ":" realm ":" password) */
2846     gen_md5_init (ctx);
2847     gen_md5_update ((unsigned char *)user, strlen (user), ctx);
2848     gen_md5_update ((unsigned char *)":", 1, ctx);
2849     gen_md5_update ((unsigned char *)realm, strlen (realm), ctx);
2850     gen_md5_update ((unsigned char *)":", 1, ctx);
2851     gen_md5_update ((unsigned char *)passwd, strlen (passwd), ctx);
2852     gen_md5_finish (ctx, hash);
2853     dump_hash (a1buf, hash);
2854
2855     /* A2BUF = H(method ":" path) */
2856     gen_md5_init (ctx);
2857     gen_md5_update ((unsigned char *)method, strlen (method), ctx);
2858     gen_md5_update ((unsigned char *)":", 1, ctx);
2859     gen_md5_update ((unsigned char *)path, strlen (path), ctx);
2860     gen_md5_finish (ctx, hash);
2861     dump_hash (a2buf, hash);
2862
2863     /* RESPONSE_DIGEST = H(A1BUF ":" nonce ":" A2BUF) */
2864     gen_md5_init (ctx);
2865     gen_md5_update (a1buf, MD5_HASHLEN * 2, ctx);
2866     gen_md5_update ((unsigned char *)":", 1, ctx);
2867     gen_md5_update ((unsigned char *)nonce, strlen (nonce), ctx);
2868     gen_md5_update ((unsigned char *)":", 1, ctx);
2869     gen_md5_update (a2buf, MD5_HASHLEN * 2, ctx);
2870     gen_md5_finish (ctx, hash);
2871     dump_hash (response_digest, hash);
2872
2873     res = (char*) xmalloc (strlen (user)
2874                            + strlen (user)
2875                            + strlen (realm)
2876                            + strlen (nonce)
2877                            + strlen (path)
2878                            + 2 * MD5_HASHLEN /*strlen (response_digest)*/
2879                            + (opaque ? strlen (opaque) : 0)
2880                            + 128);
2881     sprintf (res, "Digest \
2882 username=\"%s\", realm=\"%s\", nonce=\"%s\", uri=\"%s\", response=\"%s\"",
2883              user, realm, nonce, path, response_digest);
2884     if (opaque)
2885       {
2886         char *p = res + strlen (res);
2887         strcat (p, ", opaque=\"");
2888         strcat (p, opaque);
2889         strcat (p, "\"");
2890       }
2891   }
2892   return res;
2893 }
2894 #endif /* ENABLE_DIGEST */
2895
2896 /* Computing the size of a string literal must take into account that
2897    value returned by sizeof includes the terminating \0.  */
2898 #define STRSIZE(literal) (sizeof (literal) - 1)
2899
2900 /* Whether chars in [b, e) begin with the literal string provided as
2901    first argument and are followed by whitespace or terminating \0.
2902    The comparison is case-insensitive.  */
2903 #define STARTS(literal, b, e)                           \
2904   ((e) - (b) >= STRSIZE (literal)                       \
2905    && 0 == strncasecmp (b, literal, STRSIZE (literal))  \
2906    && ((e) - (b) == STRSIZE (literal)                   \
2907        || ISSPACE (b[STRSIZE (literal)])))
2908
2909 static int
2910 known_authentication_scheme_p (const char *hdrbeg, const char *hdrend)
2911 {
2912   return STARTS ("Basic", hdrbeg, hdrend)
2913 #ifdef ENABLE_DIGEST
2914     || STARTS ("Digest", hdrbeg, hdrend)
2915 #endif
2916 #ifdef ENABLE_NTLM
2917     || STARTS ("NTLM", hdrbeg, hdrend)
2918 #endif
2919     ;
2920 }
2921
2922 #undef STARTS
2923
2924 /* Create the HTTP authorization request header.  When the
2925    `WWW-Authenticate' response header is seen, according to the
2926    authorization scheme specified in that header (`Basic' and `Digest'
2927    are supported by the current implementation), produce an
2928    appropriate HTTP authorization request header.  */
2929 static char *
2930 create_authorization_line (const char *au, const char *user,
2931                            const char *passwd, const char *method,
2932                            const char *path, int *finished)
2933 {
2934   /* We are called only with known schemes, so we can dispatch on the
2935      first letter. */
2936   switch (TOUPPER (*au))
2937     {
2938     case 'B':                   /* Basic */
2939       *finished = 1;
2940       return basic_authentication_encode (user, passwd);
2941 #ifdef ENABLE_DIGEST
2942     case 'D':                   /* Digest */
2943       *finished = 1;
2944       return digest_authentication_encode (au, user, passwd, method, path);
2945 #endif
2946 #ifdef ENABLE_NTLM
2947     case 'N':                   /* NTLM */
2948       if (!ntlm_input (&pconn.ntlm, au))
2949         {
2950           *finished = 1;
2951           return NULL;
2952         }
2953       return ntlm_output (&pconn.ntlm, user, passwd, finished);
2954 #endif
2955     default:
2956       /* We shouldn't get here -- this function should be only called
2957          with values approved by known_authentication_scheme_p.  */
2958       abort ();
2959     }
2960 }
2961 \f
2962 void
2963 save_cookies (void)
2964 {
2965   if (wget_cookie_jar)
2966     cookie_jar_save (wget_cookie_jar, opt.cookies_output);
2967 }
2968
2969 void
2970 http_cleanup (void)
2971 {
2972   xfree_null (pconn.host);
2973   if (wget_cookie_jar)
2974     cookie_jar_delete (wget_cookie_jar);
2975 }