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