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