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