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