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