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