]> sjero.net Git - wget/blob - src/http.c
Improve output in case of --post-{file,body} commands.
[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 body_file_send (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 BODY 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 body_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.method)
1769       {
1770         char *q;
1771         for (q = opt.method; *q; ++q)
1772           *q = c_toupper (*q);
1773         meth = opt.method;
1774       }
1775     /* Use the full path, i.e. one that includes the leading slash and
1776        the query string.  E.g. if u->path is "foo/bar" and u->query is
1777        "param=value", full_path will be "/foo/bar?param=value".  */
1778     if (proxy
1779 #ifdef HAVE_SSL
1780         /* When using SSL over proxy, CONNECT establishes a direct
1781            connection to the HTTPS server.  Therefore use the same
1782            argument as when talking to the server directly. */
1783         && u->scheme != SCHEME_HTTPS
1784 #endif
1785         )
1786       meth_arg = xstrdup (u->url);
1787     else
1788       meth_arg = url_full_path (u);
1789     request_set_method (req, meth, meth_arg);
1790   }
1791
1792   request_set_header (req, "Referer", (char *) hs->referer, rel_none);
1793   if (*dt & SEND_NOCACHE)
1794     {
1795       /* Cache-Control MUST be obeyed by all HTTP/1.1 caching mechanisms...  */
1796       request_set_header (req, "Cache-Control", "no-cache, must-revalidate", rel_none);
1797
1798       /* ... but some HTTP/1.0 caches doesn't implement Cache-Control.  */
1799       request_set_header (req, "Pragma", "no-cache", rel_none);
1800     }
1801   if (hs->restval)
1802     request_set_header (req, "Range",
1803                         aprintf ("bytes=%s-",
1804                                  number_to_static_string (hs->restval)),
1805                         rel_value);
1806   SET_USER_AGENT (req);
1807   request_set_header (req, "Accept", "*/*", rel_none);
1808
1809   /* Find the username and password for authentication. */
1810   user = u->user;
1811   passwd = u->passwd;
1812   search_netrc (u->host, (const char **)&user, (const char **)&passwd, 0);
1813   user = user ? user : (opt.http_user ? opt.http_user : opt.user);
1814   passwd = passwd ? passwd : (opt.http_passwd ? opt.http_passwd : opt.passwd);
1815
1816   /* We only do "site-wide" authentication with "global" user/password
1817    * values unless --auth-no-challange has been requested; URL user/password
1818    * info overrides. */
1819   if (user && passwd && (!u->user || opt.auth_without_challenge))
1820     {
1821       /* If this is a host for which we've already received a Basic
1822        * challenge, we'll go ahead and send Basic authentication creds. */
1823       basic_auth_finished = maybe_send_basic_creds(u->host, user, passwd, req);
1824     }
1825
1826   /* Generate the Host header, HOST:PORT.  Take into account that:
1827
1828      - Broken server-side software often doesn't recognize the PORT
1829        argument, so we must generate "Host: www.server.com" instead of
1830        "Host: www.server.com:80" (and likewise for https port).
1831
1832      - IPv6 addresses contain ":", so "Host: 3ffe:8100:200:2::2:1234"
1833        becomes ambiguous and needs to be rewritten as "Host:
1834        [3ffe:8100:200:2::2]:1234".  */
1835   {
1836     /* Formats arranged for hfmt[add_port][add_squares].  */
1837     static const char *hfmt[][2] = {
1838       { "%s", "[%s]" }, { "%s:%d", "[%s]:%d" }
1839     };
1840     int add_port = u->port != scheme_default_port (u->scheme);
1841     int add_squares = strchr (u->host, ':') != NULL;
1842     request_set_header (req, "Host",
1843                         aprintf (hfmt[add_port][add_squares], u->host, u->port),
1844                         rel_value);
1845   }
1846
1847   if (inhibit_keep_alive)
1848     request_set_header (req, "Connection", "Close", rel_none);
1849   else
1850     {
1851       if (proxy == NULL)
1852         request_set_header (req, "Connection", "Keep-Alive", rel_none);
1853       else
1854         {
1855           request_set_header (req, "Connection", "Close", rel_none);
1856           request_set_header (req, "Proxy-Connection", "Keep-Alive", rel_none);
1857         }
1858     }
1859
1860   if (opt.method)
1861     {
1862
1863       if (opt.body_data || opt.body_file)
1864         {
1865           request_set_header (req, "Content-Type",
1866                               "application/x-www-form-urlencoded", rel_none);
1867
1868           if (opt.body_data)
1869             body_data_size = strlen (opt.body_data);
1870           else
1871             {
1872               body_data_size = file_size (opt.body_file);
1873               if (body_data_size == -1)
1874                 {
1875                   logprintf (LOG_NOTQUIET, _("BODY data file %s missing: %s\n"),
1876                              quote (opt.body_file), strerror (errno));
1877                   return FILEBADFILE;
1878                 }
1879             }
1880           request_set_header (req, "Content-Length",
1881                               xstrdup (number_to_static_string (body_data_size)),
1882                               rel_value);
1883         }
1884     }
1885
1886  retry_with_auth:
1887   /* We need to come back here when the initial attempt to retrieve
1888      without authorization header fails.  (Expected to happen at least
1889      for the Digest authorization scheme.)  */
1890
1891   if (opt.cookies)
1892     request_set_header (req, "Cookie",
1893                         cookie_header (wget_cookie_jar,
1894                                        u->host, u->port, u->path,
1895 #ifdef HAVE_SSL
1896                                        u->scheme == SCHEME_HTTPS
1897 #else
1898                                        0
1899 #endif
1900                                        ),
1901                         rel_value);
1902
1903   /* Add the user headers. */
1904   if (opt.user_headers)
1905     {
1906       int i;
1907       for (i = 0; opt.user_headers[i]; i++)
1908         request_set_user_header (req, opt.user_headers[i]);
1909     }
1910
1911   proxyauth = NULL;
1912   if (proxy)
1913     {
1914       char *proxy_user, *proxy_passwd;
1915       /* For normal username and password, URL components override
1916          command-line/wgetrc parameters.  With proxy
1917          authentication, it's the reverse, because proxy URLs are
1918          normally the "permanent" ones, so command-line args
1919          should take precedence.  */
1920       if (opt.proxy_user && opt.proxy_passwd)
1921         {
1922           proxy_user = opt.proxy_user;
1923           proxy_passwd = opt.proxy_passwd;
1924         }
1925       else
1926         {
1927           proxy_user = proxy->user;
1928           proxy_passwd = proxy->passwd;
1929         }
1930       /* #### This does not appear right.  Can't the proxy request,
1931          say, `Digest' authentication?  */
1932       if (proxy_user && proxy_passwd)
1933         proxyauth = basic_authentication_encode (proxy_user, proxy_passwd);
1934
1935       /* If we're using a proxy, we will be connecting to the proxy
1936          server.  */
1937       conn = proxy;
1938
1939       /* Proxy authorization over SSL is handled below. */
1940 #ifdef HAVE_SSL
1941       if (u->scheme != SCHEME_HTTPS)
1942 #endif
1943         request_set_header (req, "Proxy-Authorization", proxyauth, rel_value);
1944     }
1945
1946   keep_alive = true;
1947
1948   /* Establish the connection.  */
1949
1950   if (inhibit_keep_alive)
1951     keep_alive = false;
1952   else
1953     {
1954       /* Look for a persistent connection to target host, unless a
1955          proxy is used.  The exception is when SSL is in use, in which
1956          case the proxy is nothing but a passthrough to the target
1957          host, registered as a connection to the latter.  */
1958       struct url *relevant = conn;
1959 #ifdef HAVE_SSL
1960       if (u->scheme == SCHEME_HTTPS)
1961         relevant = u;
1962 #endif
1963
1964       if (persistent_available_p (relevant->host, relevant->port,
1965 #ifdef HAVE_SSL
1966                                   relevant->scheme == SCHEME_HTTPS,
1967 #else
1968                                   0,
1969 #endif
1970                                   &host_lookup_failed))
1971         {
1972           int family = socket_family (pconn.socket, ENDPOINT_PEER);
1973           sock = pconn.socket;
1974           using_ssl = pconn.ssl;
1975 #if ENABLE_IPV6
1976           if (family == AF_INET6)
1977              logprintf (LOG_VERBOSE, _("Reusing existing connection to [%s]:%d.\n"),
1978                         quotearg_style (escape_quoting_style, pconn.host),
1979                          pconn.port);
1980           else
1981 #endif
1982              logprintf (LOG_VERBOSE, _("Reusing existing connection to %s:%d.\n"),
1983                         quotearg_style (escape_quoting_style, pconn.host),
1984                         pconn.port);
1985           DEBUGP (("Reusing fd %d.\n", sock));
1986           if (pconn.authorized)
1987             /* If the connection is already authorized, the "Basic"
1988                authorization added by code above is unnecessary and
1989                only hurts us.  */
1990             request_remove_header (req, "Authorization");
1991         }
1992       else if (host_lookup_failed)
1993         {
1994           request_free (req);
1995           logprintf(LOG_NOTQUIET,
1996                     _("%s: unable to resolve host address %s\n"),
1997                     exec_name, quote (relevant->host));
1998           return HOSTERR;
1999         }
2000     }
2001
2002   if (sock < 0)
2003     {
2004       sock = connect_to_host (conn->host, conn->port);
2005       if (sock == E_HOST)
2006         {
2007           request_free (req);
2008           return HOSTERR;
2009         }
2010       else if (sock < 0)
2011         {
2012           request_free (req);
2013           return (retryable_socket_connect_error (errno)
2014                   ? CONERROR : CONIMPOSSIBLE);
2015         }
2016
2017 #ifdef HAVE_SSL
2018       if (proxy && u->scheme == SCHEME_HTTPS)
2019         {
2020           /* When requesting SSL URLs through proxies, use the
2021              CONNECT method to request passthrough.  */
2022           struct request *connreq = request_new ();
2023           request_set_method (connreq, "CONNECT",
2024                               aprintf ("%s:%d", u->host, u->port));
2025           SET_USER_AGENT (connreq);
2026           if (proxyauth)
2027             {
2028               request_set_header (connreq, "Proxy-Authorization",
2029                                   proxyauth, rel_value);
2030               /* Now that PROXYAUTH is part of the CONNECT request,
2031                  zero it out so we don't send proxy authorization with
2032                  the regular request below.  */
2033               proxyauth = NULL;
2034             }
2035           /* Examples in rfc2817 use the Host header in CONNECT
2036              requests.  I don't see how that gains anything, given
2037              that the contents of Host would be exactly the same as
2038              the contents of CONNECT.  */
2039
2040           write_error = request_send (connreq, sock, 0);
2041           request_free (connreq);
2042           if (write_error < 0)
2043             {
2044               CLOSE_INVALIDATE (sock);
2045               request_free (req);
2046               return WRITEFAILED;
2047             }
2048
2049           head = read_http_response_head (sock);
2050           if (!head)
2051             {
2052               logprintf (LOG_VERBOSE, _("Failed reading proxy response: %s\n"),
2053                          fd_errstr (sock));
2054               CLOSE_INVALIDATE (sock);
2055               request_free (req);
2056               return HERR;
2057             }
2058           message = NULL;
2059           if (!*head)
2060             {
2061               xfree (head);
2062               goto failed_tunnel;
2063             }
2064           DEBUGP (("proxy responded with: [%s]\n", head));
2065
2066           resp = resp_new (head);
2067           statcode = resp_status (resp, &message);
2068           if (statcode < 0)
2069             {
2070               char *tms = datetime_str (time (NULL));
2071               logprintf (LOG_VERBOSE, "%d\n", statcode);
2072               logprintf (LOG_NOTQUIET, _("%s ERROR %d: %s.\n"), tms, statcode,
2073                          quotearg_style (escape_quoting_style,
2074                                          _("Malformed status line")));
2075               xfree (head);
2076               request_free (req);
2077               return HERR;
2078             }
2079           hs->message = xstrdup (message);
2080           resp_free (resp);
2081           xfree (head);
2082           if (statcode != 200)
2083             {
2084             failed_tunnel:
2085               logprintf (LOG_NOTQUIET, _("Proxy tunneling failed: %s"),
2086                          message ? quotearg_style (escape_quoting_style, message) : "?");
2087               xfree_null (message);
2088               request_free (req);
2089               return CONSSLERR;
2090             }
2091           xfree_null (message);
2092
2093           /* SOCK is now *really* connected to u->host, so update CONN
2094              to reflect this.  That way register_persistent will
2095              register SOCK as being connected to u->host:u->port.  */
2096           conn = u;
2097         }
2098
2099       if (conn->scheme == SCHEME_HTTPS)
2100         {
2101           if (!ssl_connect_wget (sock, u->host))
2102             {
2103               fd_close (sock);
2104               request_free (req);
2105               return CONSSLERR;
2106             }
2107           else if (!ssl_check_certificate (sock, u->host))
2108             {
2109               fd_close (sock);
2110               request_free (req);
2111               return VERIFCERTERR;
2112             }
2113           using_ssl = true;
2114         }
2115 #endif /* HAVE_SSL */
2116     }
2117
2118   /* Open the temporary file where we will write the request. */
2119   if (warc_enabled)
2120     {
2121       warc_tmp = warc_tempfile ();
2122       if (warc_tmp == NULL)
2123         {
2124           CLOSE_INVALIDATE (sock);
2125           request_free (req);
2126           return WARC_TMP_FOPENERR;
2127         }
2128
2129       if (! proxy)
2130         {
2131           warc_ip = (ip_address *) alloca (sizeof (ip_address));
2132           socket_ip_address (sock, warc_ip, ENDPOINT_PEER);
2133         }
2134     }
2135
2136   /* Send the request to server.  */
2137   write_error = request_send (req, sock, warc_tmp);
2138
2139   if (write_error >= 0)
2140     {
2141       if (opt.body_data)
2142         {
2143           DEBUGP (("[BODY data: %s]\n", opt.body_data));
2144           write_error = fd_write (sock, opt.body_data, body_data_size, -1);
2145           if (write_error >= 0 && warc_tmp != NULL)
2146             {
2147               /* Remember end of headers / start of payload. */
2148               warc_payload_offset = ftello (warc_tmp);
2149
2150               /* Write a copy of the data to the WARC record. */
2151               int warc_tmp_written = fwrite (opt.body_data, 1, body_data_size, warc_tmp);
2152               if (warc_tmp_written != body_data_size)
2153                 write_error = -2;
2154             }
2155          }
2156       else if (opt.body_file && body_data_size != 0)
2157         {
2158           if (warc_tmp != NULL)
2159             /* Remember end of headers / start of payload */
2160             warc_payload_offset = ftello (warc_tmp);
2161
2162           write_error = body_file_send (sock, opt.body_file, body_data_size, warc_tmp);
2163         }
2164     }
2165
2166   if (write_error < 0)
2167     {
2168       CLOSE_INVALIDATE (sock);
2169       request_free (req);
2170
2171       if (warc_tmp != NULL)
2172         fclose (warc_tmp);
2173
2174       if (write_error == -2)
2175         return WARC_TMP_FWRITEERR;
2176       else
2177         return WRITEFAILED;
2178     }
2179   logprintf (LOG_VERBOSE, _("%s request sent, awaiting response... "),
2180              proxy ? "Proxy" : "HTTP");
2181   contlen = -1;
2182   contrange = 0;
2183   *dt &= ~RETROKF;
2184
2185
2186   if (warc_enabled)
2187     {
2188       bool warc_result;
2189       /* Generate a timestamp and uuid for this request. */
2190       warc_timestamp (warc_timestamp_str);
2191       warc_uuid_str (warc_request_uuid);
2192
2193       /* Create a request record and store it in the WARC file. */
2194       warc_result = warc_write_request_record (u->url, warc_timestamp_str,
2195                                                warc_request_uuid, warc_ip,
2196                                                warc_tmp, warc_payload_offset);
2197       if (! warc_result)
2198         {
2199           CLOSE_INVALIDATE (sock);
2200           request_free (req);
2201           return WARC_ERR;
2202         }
2203
2204       /* warc_write_request_record has also closed warc_tmp. */
2205     }
2206
2207
2208 read_header:
2209   head = read_http_response_head (sock);
2210   if (!head)
2211     {
2212       if (errno == 0)
2213         {
2214           logputs (LOG_NOTQUIET, _("No data received.\n"));
2215           CLOSE_INVALIDATE (sock);
2216           request_free (req);
2217           return HEOF;
2218         }
2219       else
2220         {
2221           logprintf (LOG_NOTQUIET, _("Read error (%s) in headers.\n"),
2222                      fd_errstr (sock));
2223           CLOSE_INVALIDATE (sock);
2224           request_free (req);
2225           return HERR;
2226         }
2227     }
2228   DEBUGP (("\n---response begin---\n%s---response end---\n", head));
2229
2230   resp = resp_new (head);
2231
2232   /* Check for status line.  */
2233   message = NULL;
2234   statcode = resp_status (resp, &message);
2235   if (statcode < 0)
2236     {
2237       char *tms = datetime_str (time (NULL));
2238       logprintf (LOG_VERBOSE, "%d\n", statcode);
2239       logprintf (LOG_NOTQUIET, _("%s ERROR %d: %s.\n"), tms, statcode,
2240                  quotearg_style (escape_quoting_style,
2241                                  _("Malformed status line")));
2242       CLOSE_INVALIDATE (sock);
2243       resp_free (resp);
2244       request_free (req);
2245       xfree (head);
2246       return HERR;
2247     }
2248
2249   if (H_10X (statcode))
2250     {
2251       DEBUGP (("Ignoring response\n"));
2252       resp_free (resp);
2253       xfree (head);
2254       goto read_header;
2255     }
2256
2257   hs->message = xstrdup (message);
2258   if (!opt.server_response)
2259     logprintf (LOG_VERBOSE, "%2d %s\n", statcode,
2260                message ? quotearg_style (escape_quoting_style, message) : "");
2261   else
2262     {
2263       logprintf (LOG_VERBOSE, "\n");
2264       print_server_response (resp, "  ");
2265     }
2266
2267   if (!opt.ignore_length
2268       && resp_header_copy (resp, "Content-Length", hdrval, sizeof (hdrval)))
2269     {
2270       wgint parsed;
2271       errno = 0;
2272       parsed = str_to_wgint (hdrval, NULL, 10);
2273       if (parsed == WGINT_MAX && errno == ERANGE)
2274         {
2275           /* Out of range.
2276              #### If Content-Length is out of range, it most likely
2277              means that the file is larger than 2G and that we're
2278              compiled without LFS.  In that case we should probably
2279              refuse to even attempt to download the file.  */
2280           contlen = -1;
2281         }
2282       else if (parsed < 0)
2283         {
2284           /* Negative Content-Length; nonsensical, so we can't
2285              assume any information about the content to receive. */
2286           contlen = -1;
2287         }
2288       else
2289         contlen = parsed;
2290     }
2291
2292   /* Check for keep-alive related responses. */
2293   if (!inhibit_keep_alive && contlen != -1)
2294     {
2295       if (resp_header_copy (resp, "Connection", hdrval, sizeof (hdrval)))
2296         {
2297           if (0 == strcasecmp (hdrval, "Close"))
2298             keep_alive = false;
2299         }
2300     }
2301
2302   chunked_transfer_encoding = false;
2303   if (resp_header_copy (resp, "Transfer-Encoding", hdrval, sizeof (hdrval))
2304       && 0 == strcasecmp (hdrval, "chunked"))
2305     chunked_transfer_encoding = true;
2306
2307   /* Handle (possibly multiple instances of) the Set-Cookie header. */
2308   if (opt.cookies)
2309     {
2310       int scpos;
2311       const char *scbeg, *scend;
2312       /* The jar should have been created by now. */
2313       assert (wget_cookie_jar != NULL);
2314       for (scpos = 0;
2315            (scpos = resp_header_locate (resp, "Set-Cookie", scpos,
2316                                         &scbeg, &scend)) != -1;
2317            ++scpos)
2318         {
2319           char *set_cookie; BOUNDED_TO_ALLOCA (scbeg, scend, set_cookie);
2320           cookie_handle_set_cookie (wget_cookie_jar, u->host, u->port,
2321                                     u->path, set_cookie);
2322         }
2323     }
2324
2325   if (keep_alive)
2326     /* The server has promised that it will not close the connection
2327        when we're done.  This means that we can register it.  */
2328     register_persistent (conn->host, conn->port, sock, using_ssl);
2329
2330   if (statcode == HTTP_STATUS_UNAUTHORIZED)
2331     {
2332       /* Authorization is required.  */
2333
2334       /* Normally we are not interested in the response body.
2335          But if we are writing a WARC file we are: we like to keep everyting.  */
2336       if (warc_enabled)
2337         {
2338           int err;
2339           type = resp_header_strdup (resp, "Content-Type");
2340           err = read_response_body (hs, sock, NULL, contlen, 0,
2341                                     chunked_transfer_encoding,
2342                                     u->url, warc_timestamp_str,
2343                                     warc_request_uuid, warc_ip, type,
2344                                     statcode, head);
2345           xfree_null (type);
2346
2347           if (err != RETRFINISHED || hs->res < 0)
2348             {
2349               CLOSE_INVALIDATE (sock);
2350               request_free (req);
2351               xfree_null (message);
2352               resp_free (resp);
2353               xfree (head);
2354               return err;
2355             }
2356           else
2357             CLOSE_FINISH (sock);
2358         }
2359       else
2360         {
2361           /* Since WARC is disabled, we are not interested in the response body.  */
2362           if (keep_alive && !head_only
2363               && skip_short_body (sock, contlen, chunked_transfer_encoding))
2364             CLOSE_FINISH (sock);
2365           else
2366             CLOSE_INVALIDATE (sock);
2367         }
2368
2369       pconn.authorized = false;
2370       if (!auth_finished && (user && passwd))
2371         {
2372           /* IIS sends multiple copies of WWW-Authenticate, one with
2373              the value "negotiate", and other(s) with data.  Loop over
2374              all the occurrences and pick the one we recognize.  */
2375           int wapos;
2376           const char *wabeg, *waend;
2377           char *www_authenticate = NULL;
2378           for (wapos = 0;
2379                (wapos = resp_header_locate (resp, "WWW-Authenticate", wapos,
2380                                             &wabeg, &waend)) != -1;
2381                ++wapos)
2382             if (known_authentication_scheme_p (wabeg, waend))
2383               {
2384                 BOUNDED_TO_ALLOCA (wabeg, waend, www_authenticate);
2385                 break;
2386               }
2387
2388           if (!www_authenticate)
2389             {
2390               /* If the authentication header is missing or
2391                  unrecognized, there's no sense in retrying.  */
2392               logputs (LOG_NOTQUIET, _("Unknown authentication scheme.\n"));
2393             }
2394           else if (!basic_auth_finished
2395                    || !BEGINS_WITH (www_authenticate, "Basic"))
2396             {
2397               char *pth;
2398               pth = url_full_path (u);
2399               request_set_header (req, "Authorization",
2400                                   create_authorization_line (www_authenticate,
2401                                                              user, passwd,
2402                                                              request_method (req),
2403                                                              pth,
2404                                                              &auth_finished),
2405                                   rel_value);
2406               if (BEGINS_WITH (www_authenticate, "NTLM"))
2407                 ntlm_seen = true;
2408               else if (!u->user && BEGINS_WITH (www_authenticate, "Basic"))
2409                 {
2410                   /* Need to register this host as using basic auth,
2411                    * so we automatically send creds next time. */
2412                   register_basic_auth_host (u->host);
2413                 }
2414               xfree (pth);
2415               xfree_null (message);
2416               resp_free (resp);
2417               xfree (head);
2418               goto retry_with_auth;
2419             }
2420           else
2421             {
2422               /* We already did Basic auth, and it failed. Gotta
2423                * give up. */
2424             }
2425         }
2426       logputs (LOG_NOTQUIET, _("Authorization failed.\n"));
2427       request_free (req);
2428       xfree_null (message);
2429       resp_free (resp);
2430       xfree (head);
2431       return AUTHFAILED;
2432     }
2433   else /* statcode != HTTP_STATUS_UNAUTHORIZED */
2434     {
2435       /* Kludge: if NTLM is used, mark the TCP connection as authorized. */
2436       if (ntlm_seen)
2437         pconn.authorized = true;
2438     }
2439
2440   /* Determine the local filename if needed. Notice that if -O is used
2441    * hstat.local_file is set by http_loop to the argument of -O. */
2442   if (!hs->local_file)
2443     {
2444       char *local_file = NULL;
2445
2446       /* Honor Content-Disposition whether possible. */
2447       if (!opt.content_disposition
2448           || !resp_header_copy (resp, "Content-Disposition",
2449                                 hdrval, sizeof (hdrval))
2450           || !parse_content_disposition (hdrval, &local_file))
2451         {
2452           /* The Content-Disposition header is missing or broken.
2453            * Choose unique file name according to given URL. */
2454           hs->local_file = url_file_name (u, NULL);
2455         }
2456       else
2457         {
2458           DEBUGP (("Parsed filename from Content-Disposition: %s\n",
2459                   local_file));
2460           hs->local_file = url_file_name (u, local_file);
2461         }
2462     }
2463
2464   /* TODO: perform this check only once. */
2465   if (!hs->existence_checked && file_exists_p (hs->local_file))
2466     {
2467       if (opt.noclobber && !opt.output_document)
2468         {
2469           /* If opt.noclobber is turned on and file already exists, do not
2470              retrieve the file. But if the output_document was given, then this
2471              test was already done and the file didn't exist. Hence the !opt.output_document */
2472           get_file_flags (hs->local_file, dt);
2473           request_free (req);
2474           resp_free (resp);
2475           xfree (head);
2476           xfree_null (message);
2477           return RETRUNNEEDED;
2478         }
2479       else if (!ALLOW_CLOBBER)
2480         {
2481           char *unique = unique_name (hs->local_file, true);
2482           if (unique != hs->local_file)
2483             xfree (hs->local_file);
2484           hs->local_file = unique;
2485         }
2486     }
2487   hs->existence_checked = true;
2488
2489   /* Support timestamping */
2490   /* TODO: move this code out of gethttp. */
2491   if (opt.timestamping && !hs->timestamp_checked)
2492     {
2493       size_t filename_len = strlen (hs->local_file);
2494       char *filename_plus_orig_suffix = alloca (filename_len + sizeof (ORIG_SFX));
2495       bool local_dot_orig_file_exists = false;
2496       char *local_filename = NULL;
2497       struct_stat st;
2498
2499       if (opt.backup_converted)
2500         /* If -K is specified, we'll act on the assumption that it was specified
2501            last time these files were downloaded as well, and instead of just
2502            comparing local file X against server file X, we'll compare local
2503            file X.orig (if extant, else X) against server file X.  If -K
2504            _wasn't_ specified last time, or the server contains files called
2505            *.orig, -N will be back to not operating correctly with -k. */
2506         {
2507           /* Would a single s[n]printf() call be faster?  --dan
2508
2509              Definitely not.  sprintf() is horribly slow.  It's a
2510              different question whether the difference between the two
2511              affects a program.  Usually I'd say "no", but at one
2512              point I profiled Wget, and found that a measurable and
2513              non-negligible amount of time was lost calling sprintf()
2514              in url.c.  Replacing sprintf with inline calls to
2515              strcpy() and number_to_string() made a difference.
2516              --hniksic */
2517           memcpy (filename_plus_orig_suffix, hs->local_file, filename_len);
2518           memcpy (filename_plus_orig_suffix + filename_len,
2519                   ORIG_SFX, sizeof (ORIG_SFX));
2520
2521           /* Try to stat() the .orig file. */
2522           if (stat (filename_plus_orig_suffix, &st) == 0)
2523             {
2524               local_dot_orig_file_exists = true;
2525               local_filename = filename_plus_orig_suffix;
2526             }
2527         }
2528
2529       if (!local_dot_orig_file_exists)
2530         /* Couldn't stat() <file>.orig, so try to stat() <file>. */
2531         if (stat (hs->local_file, &st) == 0)
2532           local_filename = hs->local_file;
2533
2534       if (local_filename != NULL)
2535         /* There was a local file, so we'll check later to see if the version
2536            the server has is the same version we already have, allowing us to
2537            skip a download. */
2538         {
2539           hs->orig_file_name = xstrdup (local_filename);
2540           hs->orig_file_size = st.st_size;
2541           hs->orig_file_tstamp = st.st_mtime;
2542 #ifdef WINDOWS
2543           /* Modification time granularity is 2 seconds for Windows, so
2544              increase local time by 1 second for later comparison. */
2545           ++hs->orig_file_tstamp;
2546 #endif
2547         }
2548     }
2549
2550   request_free (req);
2551
2552   hs->statcode = statcode;
2553   if (statcode == -1)
2554     hs->error = xstrdup (_("Malformed status line"));
2555   else if (!*message)
2556     hs->error = xstrdup (_("(no description)"));
2557   else
2558     hs->error = xstrdup (message);
2559   xfree_null (message);
2560
2561   type = resp_header_strdup (resp, "Content-Type");
2562   if (type)
2563     {
2564       char *tmp = strchr (type, ';');
2565       if (tmp)
2566         {
2567           /* sXXXav: only needed if IRI support is enabled */
2568           char *tmp2 = tmp + 1;
2569
2570           while (tmp > type && c_isspace (tmp[-1]))
2571             --tmp;
2572           *tmp = '\0';
2573
2574           /* Try to get remote encoding if needed */
2575           if (opt.enable_iri && !opt.encoding_remote)
2576             {
2577               tmp = parse_charset (tmp2);
2578               if (tmp)
2579                 set_content_encoding (iri, tmp);
2580             }
2581         }
2582     }
2583   hs->newloc = resp_header_strdup (resp, "Location");
2584   hs->remote_time = resp_header_strdup (resp, "Last-Modified");
2585
2586   if (resp_header_copy (resp, "Content-Range", hdrval, sizeof (hdrval)))
2587     {
2588       wgint first_byte_pos, last_byte_pos, entity_length;
2589       if (parse_content_range (hdrval, &first_byte_pos, &last_byte_pos,
2590                                &entity_length))
2591         {
2592           contrange = first_byte_pos;
2593           contlen = last_byte_pos - first_byte_pos + 1;
2594         }
2595     }
2596   resp_free (resp);
2597
2598   /* 20x responses are counted among successful by default.  */
2599   if (H_20X (statcode))
2600     *dt |= RETROKF;
2601
2602   /* Return if redirected.  */
2603   if (H_REDIRECTED (statcode) || statcode == HTTP_STATUS_MULTIPLE_CHOICES)
2604     {
2605       /* RFC2068 says that in case of the 300 (multiple choices)
2606          response, the server can output a preferred URL through
2607          `Location' header; otherwise, the request should be treated
2608          like GET.  So, if the location is set, it will be a
2609          redirection; otherwise, just proceed normally.  */
2610       if (statcode == HTTP_STATUS_MULTIPLE_CHOICES && !hs->newloc)
2611         *dt |= RETROKF;
2612       else
2613         {
2614           logprintf (LOG_VERBOSE,
2615                      _("Location: %s%s\n"),
2616                      hs->newloc ? escnonprint_uri (hs->newloc) : _("unspecified"),
2617                      hs->newloc ? _(" [following]") : "");
2618  
2619           /* In case the caller cares to look...  */
2620           hs->len = 0;
2621           hs->res = 0;
2622           hs->restval = 0;
2623
2624           /* Normally we are not interested in the response body of a redirect.
2625              But if we are writing a WARC file we are: we like to keep everyting.  */
2626           if (warc_enabled)
2627             {
2628               int err = read_response_body (hs, sock, NULL, contlen, 0,
2629                                             chunked_transfer_encoding,
2630                                             u->url, warc_timestamp_str,
2631                                             warc_request_uuid, warc_ip, type,
2632                                             statcode, head);
2633
2634               if (err != RETRFINISHED || hs->res < 0)
2635                 {
2636                   CLOSE_INVALIDATE (sock);
2637                   xfree_null (type);
2638                   xfree (head);
2639                   return err;
2640                 }
2641               else
2642                 CLOSE_FINISH (sock);
2643             }
2644           else
2645             {
2646               /* Since WARC is disabled, we are not interested in the response body.  */
2647               if (keep_alive && !head_only
2648                   && skip_short_body (sock, contlen, chunked_transfer_encoding))
2649                 CLOSE_FINISH (sock);
2650               else
2651                 CLOSE_INVALIDATE (sock);
2652             }
2653
2654           xfree_null (type);
2655           xfree (head);
2656           /* From RFC2616: The status codes 303 and 307 have
2657              been added for servers that wish to make unambiguously
2658              clear which kind of reaction is expected of the client.
2659              
2660              A 307 should be redirected using the same method,
2661              in other words, a POST should be preserved and not
2662              converted to a GET in that case. */
2663           if (statcode == HTTP_STATUS_TEMPORARY_REDIRECT)
2664             return NEWLOCATION_KEEP_POST;
2665           return NEWLOCATION;
2666         }
2667     }
2668
2669   /* If content-type is not given, assume text/html.  This is because
2670      of the multitude of broken CGI's that "forget" to generate the
2671      content-type.  */
2672   if (!type ||
2673         0 == strncasecmp (type, TEXTHTML_S, strlen (TEXTHTML_S)) ||
2674         0 == strncasecmp (type, TEXTXHTML_S, strlen (TEXTXHTML_S)))
2675     *dt |= TEXTHTML;
2676   else
2677     *dt &= ~TEXTHTML;
2678
2679   if (type &&
2680       0 == strncasecmp (type, TEXTCSS_S, strlen (TEXTCSS_S)))
2681     *dt |= TEXTCSS;
2682   else
2683     *dt &= ~TEXTCSS;
2684
2685   if (opt.adjust_extension)
2686     {
2687       if (*dt & TEXTHTML)
2688         /* -E / --adjust-extension / adjust_extension = on was specified,
2689            and this is a text/html file.  If some case-insensitive
2690            variation on ".htm[l]" isn't already the file's suffix,
2691            tack on ".html". */
2692         {
2693           ensure_extension (hs, ".html", dt);
2694         }
2695       else if (*dt & TEXTCSS)
2696         {
2697           ensure_extension (hs, ".css", dt);
2698         }
2699     }
2700
2701   if (statcode == HTTP_STATUS_RANGE_NOT_SATISFIABLE
2702       || (!opt.timestamping && hs->restval > 0 && statcode == HTTP_STATUS_OK
2703           && contrange == 0 && contlen >= 0 && hs->restval >= contlen))
2704     {
2705       /* If `-c' is in use and the file has been fully downloaded (or
2706          the remote file has shrunk), Wget effectively requests bytes
2707          after the end of file and the server response with 416
2708          (or 200 with a <= Content-Length.  */
2709       logputs (LOG_VERBOSE, _("\
2710 \n    The file is already fully retrieved; nothing to do.\n\n"));
2711       /* In case the caller inspects. */
2712       hs->len = contlen;
2713       hs->res = 0;
2714       /* Mark as successfully retrieved. */
2715       *dt |= RETROKF;
2716       xfree_null (type);
2717       CLOSE_INVALIDATE (sock);        /* would be CLOSE_FINISH, but there
2718                                    might be more bytes in the body. */
2719       xfree (head);
2720       return RETRUNNEEDED;
2721     }
2722   if ((contrange != 0 && contrange != hs->restval)
2723       || (H_PARTIAL (statcode) && !contrange))
2724     {
2725       /* The Range request was somehow misunderstood by the server.
2726          Bail out.  */
2727       xfree_null (type);
2728       CLOSE_INVALIDATE (sock);
2729       xfree (head);
2730       return RANGEERR;
2731     }
2732   if (contlen == -1)
2733     hs->contlen = -1;
2734   else
2735     hs->contlen = contlen + contrange;
2736
2737   if (opt.verbose)
2738     {
2739       if (*dt & RETROKF)
2740         {
2741           /* No need to print this output if the body won't be
2742              downloaded at all, or if the original server response is
2743              printed.  */
2744           logputs (LOG_VERBOSE, _("Length: "));
2745           if (contlen != -1)
2746             {
2747               logputs (LOG_VERBOSE, number_to_static_string (contlen + contrange));
2748               if (contlen + contrange >= 1024)
2749                 logprintf (LOG_VERBOSE, " (%s)",
2750                            human_readable (contlen + contrange));
2751               if (contrange)
2752                 {
2753                   if (contlen >= 1024)
2754                     logprintf (LOG_VERBOSE, _(", %s (%s) remaining"),
2755                                number_to_static_string (contlen),
2756                                human_readable (contlen));
2757                   else
2758                     logprintf (LOG_VERBOSE, _(", %s remaining"),
2759                                number_to_static_string (contlen));
2760                 }
2761             }
2762           else
2763             logputs (LOG_VERBOSE,
2764                      opt.ignore_length ? _("ignored") : _("unspecified"));
2765           if (type)
2766             logprintf (LOG_VERBOSE, " [%s]\n", quotearg_style (escape_quoting_style, type));
2767           else
2768             logputs (LOG_VERBOSE, "\n");
2769         }
2770     }
2771
2772   /* Return if we have no intention of further downloading.  */
2773   if ((!(*dt & RETROKF) && !opt.content_on_error) || head_only)
2774     {
2775       /* In case the caller cares to look...  */
2776       hs->len = 0;
2777       hs->res = 0;
2778       hs->restval = 0;
2779
2780       /* Normally we are not interested in the response body of a error responses.
2781          But if we are writing a WARC file we are: we like to keep everyting.  */
2782       if (warc_enabled)
2783         {
2784           int err = read_response_body (hs, sock, NULL, contlen, 0,
2785                                         chunked_transfer_encoding,
2786                                         u->url, warc_timestamp_str,
2787                                         warc_request_uuid, warc_ip, type,
2788                                         statcode, head);
2789
2790           if (err != RETRFINISHED || hs->res < 0)
2791             {
2792               CLOSE_INVALIDATE (sock);
2793               xfree (head);
2794               xfree_null (type);
2795               return err;
2796             }
2797           else
2798             CLOSE_FINISH (sock);
2799         }
2800       else
2801         {
2802           /* Since WARC is disabled, we are not interested in the response body.  */
2803           if (head_only)
2804             /* Pre-1.10 Wget used CLOSE_INVALIDATE here.  Now we trust the
2805                servers not to send body in response to a HEAD request, and
2806                those that do will likely be caught by test_socket_open.
2807                If not, they can be worked around using
2808                `--no-http-keep-alive'.  */
2809             CLOSE_FINISH (sock);
2810           else if (keep_alive
2811                    && skip_short_body (sock, contlen, chunked_transfer_encoding))
2812             /* Successfully skipped the body; also keep using the socket. */
2813             CLOSE_FINISH (sock);
2814           else
2815             CLOSE_INVALIDATE (sock);
2816         }
2817
2818       xfree (head);
2819       xfree_null (type);
2820       return RETRFINISHED;
2821     }
2822
2823 /* 2005-06-17 SMS.
2824    For VMS, define common fopen() optional arguments.
2825 */
2826 #ifdef __VMS
2827 # define FOPEN_OPT_ARGS "fop=sqo", "acc", acc_cb, &open_id
2828 # define FOPEN_BIN_FLAG 3
2829 #else /* def __VMS */
2830 # define FOPEN_BIN_FLAG true
2831 #endif /* def __VMS [else] */
2832
2833   /* Open the local file.  */
2834   if (!output_stream)
2835     {
2836       mkalldirs (hs->local_file);
2837       if (opt.backups)
2838         rotate_backups (hs->local_file);
2839       if (hs->restval)
2840         {
2841 #ifdef __VMS
2842           int open_id;
2843
2844           open_id = 21;
2845           fp = fopen (hs->local_file, "ab", FOPEN_OPT_ARGS);
2846 #else /* def __VMS */
2847           fp = fopen (hs->local_file, "ab");
2848 #endif /* def __VMS [else] */
2849         }
2850       else if (ALLOW_CLOBBER || count > 0)
2851         {
2852           if (opt.unlink && file_exists_p (hs->local_file))
2853             {
2854               int res = unlink (hs->local_file);
2855               if (res < 0)
2856                 {
2857                   logprintf (LOG_NOTQUIET, "%s: %s\n", hs->local_file,
2858                              strerror (errno));
2859                   CLOSE_INVALIDATE (sock);
2860                   xfree (head);
2861       xfree_null (type);
2862                   return UNLINKERR;
2863                 }
2864             }
2865
2866 #ifdef __VMS
2867           int open_id;
2868
2869           open_id = 22;
2870           fp = fopen (hs->local_file, "wb", FOPEN_OPT_ARGS);
2871 #else /* def __VMS */
2872           fp = fopen (hs->local_file, "wb");
2873 #endif /* def __VMS [else] */
2874         }
2875       else
2876         {
2877           fp = fopen_excl (hs->local_file, FOPEN_BIN_FLAG);
2878           if (!fp && errno == EEXIST)
2879             {
2880               /* We cannot just invent a new name and use it (which is
2881                  what functions like unique_create typically do)
2882                  because we told the user we'd use this name.
2883                  Instead, return and retry the download.  */
2884               logprintf (LOG_NOTQUIET,
2885                          _("%s has sprung into existence.\n"),
2886                          hs->local_file);
2887               CLOSE_INVALIDATE (sock);
2888               xfree (head);
2889               xfree_null (type);
2890               return FOPEN_EXCL_ERR;
2891             }
2892         }
2893       if (!fp)
2894         {
2895           logprintf (LOG_NOTQUIET, "%s: %s\n", hs->local_file, strerror (errno));
2896           CLOSE_INVALIDATE (sock);
2897           xfree (head);
2898           xfree_null (type);
2899           return FOPENERR;
2900         }
2901     }
2902   else
2903     fp = output_stream;
2904
2905   /* Print fetch message, if opt.verbose.  */
2906   if (opt.verbose)
2907     {
2908       logprintf (LOG_NOTQUIET, _("Saving to: %s\n"),
2909                  HYPHENP (hs->local_file) ? quote ("STDOUT") : quote (hs->local_file));
2910     }
2911
2912
2913   err = read_response_body (hs, sock, fp, contlen, contrange,
2914                             chunked_transfer_encoding,
2915                             u->url, warc_timestamp_str,
2916                             warc_request_uuid, warc_ip, type,
2917                             statcode, head);
2918
2919   /* Now we no longer need to store the response header. */
2920   xfree (head);
2921   xfree_null (type);
2922
2923   if (hs->res >= 0)
2924     CLOSE_FINISH (sock);
2925   else
2926     CLOSE_INVALIDATE (sock);
2927
2928   if (!output_stream)
2929     fclose (fp);
2930
2931   return err;
2932 }
2933
2934 /* The genuine HTTP loop!  This is the part where the retrieval is
2935    retried, and retried, and retried, and...  */
2936 uerr_t
2937 http_loop (struct url *u, struct url *original_url, char **newloc,
2938            char **local_file, const char *referer, int *dt, struct url *proxy,
2939            struct iri *iri)
2940 {
2941   int count;
2942   bool got_head = false;         /* used for time-stamping and filename detection */
2943   bool time_came_from_head = false;
2944   bool got_name = false;
2945   char *tms;
2946   const char *tmrate;
2947   uerr_t err, ret = TRYLIMEXC;
2948   time_t tmr = -1;               /* remote time-stamp */
2949   struct http_stat hstat;        /* HTTP status */
2950   struct_stat st;
2951   bool send_head_first = true;
2952   char *file_name;
2953   bool force_full_retrieve = false;
2954
2955
2956   /* If we are writing to a WARC file: always retrieve the whole file. */
2957   if (opt.warc_filename != NULL)
2958     force_full_retrieve = true;
2959
2960
2961   /* Assert that no value for *LOCAL_FILE was passed. */
2962   assert (local_file == NULL || *local_file == NULL);
2963
2964   /* Set LOCAL_FILE parameter. */
2965   if (local_file && opt.output_document)
2966     *local_file = HYPHENP (opt.output_document) ? NULL : xstrdup (opt.output_document);
2967
2968   /* Reset NEWLOC parameter. */
2969   *newloc = NULL;
2970
2971   /* This used to be done in main(), but it's a better idea to do it
2972      here so that we don't go through the hoops if we're just using
2973      FTP or whatever. */
2974   if (opt.cookies)
2975     load_cookies ();
2976
2977   /* Warn on (likely bogus) wildcard usage in HTTP. */
2978   if (opt.ftp_glob && has_wildcards_p (u->path))
2979     logputs (LOG_VERBOSE, _("Warning: wildcards not supported in HTTP.\n"));
2980
2981   /* Setup hstat struct. */
2982   xzero (hstat);
2983   hstat.referer = referer;
2984
2985   if (opt.output_document)
2986     {
2987       hstat.local_file = xstrdup (opt.output_document);
2988       got_name = true;
2989     }
2990   else if (!opt.content_disposition)
2991     {
2992       hstat.local_file =
2993         url_file_name (opt.trustservernames ? u : original_url, NULL);
2994       got_name = true;
2995     }
2996
2997   if (got_name && file_exists_p (hstat.local_file) && opt.noclobber && !opt.output_document)
2998     {
2999       /* If opt.noclobber is turned on and file already exists, do not
3000          retrieve the file. But if the output_document was given, then this
3001          test was already done and the file didn't exist. Hence the !opt.output_document */
3002       get_file_flags (hstat.local_file, dt);
3003       ret = RETROK;
3004       goto exit;
3005     }
3006
3007   /* Reset the counter. */
3008   count = 0;
3009
3010   /* Reset the document type. */
3011   *dt = 0;
3012
3013   /* Skip preliminary HEAD request if we're not in spider mode.  */
3014   if (!opt.spider)
3015     send_head_first = false;
3016
3017   /* Send preliminary HEAD request if --content-disposition and -c are used
3018      together.  */
3019   if (opt.content_disposition && opt.always_rest)
3020     send_head_first = true;
3021
3022   /* Send preliminary HEAD request if -N is given and we have an existing
3023    * destination file. */
3024   file_name = url_file_name (opt.trustservernames ? u : original_url, NULL);
3025   if (opt.timestamping && (file_exists_p (file_name)
3026                            || opt.content_disposition))
3027     send_head_first = true;
3028   xfree (file_name);
3029
3030   /* THE loop */
3031   do
3032     {
3033       /* Increment the pass counter.  */
3034       ++count;
3035       sleep_between_retrievals (count);
3036
3037       /* Get the current time string.  */
3038       tms = datetime_str (time (NULL));
3039
3040       if (opt.spider && !got_head)
3041         logprintf (LOG_VERBOSE, _("\
3042 Spider mode enabled. Check if remote file exists.\n"));
3043
3044       /* Print fetch message, if opt.verbose.  */
3045       if (opt.verbose)
3046         {
3047           char *hurl = url_string (u, URL_AUTH_HIDE_PASSWD);
3048
3049           if (count > 1)
3050             {
3051               char tmp[256];
3052               sprintf (tmp, _("(try:%2d)"), count);
3053               logprintf (LOG_NOTQUIET, "--%s--  %s  %s\n",
3054                          tms, tmp, hurl);
3055             }
3056           else
3057             {
3058               logprintf (LOG_NOTQUIET, "--%s--  %s\n",
3059                          tms, hurl);
3060             }
3061
3062 #ifdef WINDOWS
3063           ws_changetitle (hurl);
3064 #endif
3065           xfree (hurl);
3066         }
3067
3068       /* Default document type is empty.  However, if spider mode is
3069          on or time-stamping is employed, HEAD_ONLY commands is
3070          encoded within *dt.  */
3071       if (send_head_first && !got_head)
3072         *dt |= HEAD_ONLY;
3073       else
3074         *dt &= ~HEAD_ONLY;
3075
3076       /* Decide whether or not to restart.  */
3077       if (force_full_retrieve)
3078         hstat.restval = hstat.len;
3079       else if (opt.always_rest
3080           && got_name
3081           && stat (hstat.local_file, &st) == 0
3082           && S_ISREG (st.st_mode))
3083         /* When -c is used, continue from on-disk size.  (Can't use
3084            hstat.len even if count>1 because we don't want a failed
3085            first attempt to clobber existing data.)  */
3086         hstat.restval = st.st_size;
3087       else if (count > 1)
3088         /* otherwise, continue where the previous try left off */
3089         hstat.restval = hstat.len;
3090       else
3091         hstat.restval = 0;
3092
3093       /* Decide whether to send the no-cache directive.  We send it in
3094          two cases:
3095            a) we're using a proxy, and we're past our first retrieval.
3096               Some proxies are notorious for caching incomplete data, so
3097               we require a fresh get.
3098            b) caching is explicitly inhibited. */
3099       if ((proxy && count > 1)        /* a */
3100           || !opt.allow_cache)        /* b */
3101         *dt |= SEND_NOCACHE;
3102       else
3103         *dt &= ~SEND_NOCACHE;
3104
3105       /* Try fetching the document, or at least its head.  */
3106       err = gethttp (u, &hstat, dt, proxy, iri, count);
3107
3108       /* Time?  */
3109       tms = datetime_str (time (NULL));
3110
3111       /* Get the new location (with or without the redirection).  */
3112       if (hstat.newloc)
3113         *newloc = xstrdup (hstat.newloc);
3114
3115       switch (err)
3116         {
3117         case HERR: case HEOF: case CONSOCKERR: case CONCLOSED:
3118         case CONERROR: case READERR: case WRITEFAILED:
3119         case RANGEERR: case FOPEN_EXCL_ERR:
3120           /* Non-fatal errors continue executing the loop, which will
3121              bring them to "while" statement at the end, to judge
3122              whether the number of tries was exceeded.  */
3123           printwhat (count, opt.ntry);
3124           continue;
3125         case FWRITEERR: case FOPENERR:
3126           /* Another fatal error.  */
3127           logputs (LOG_VERBOSE, "\n");
3128           logprintf (LOG_NOTQUIET, _("Cannot write to %s (%s).\n"),
3129                      quote (hstat.local_file), strerror (errno));
3130         case HOSTERR: case CONIMPOSSIBLE: case PROXERR: case AUTHFAILED:
3131         case SSLINITFAILED: case CONTNOTSUPPORTED: case VERIFCERTERR:
3132         case FILEBADFILE:
3133           /* Fatal errors just return from the function.  */
3134           ret = err;
3135           goto exit;
3136         case WARC_ERR:
3137           /* A fatal WARC error. */
3138           logputs (LOG_VERBOSE, "\n");
3139           logprintf (LOG_NOTQUIET, _("Cannot write to WARC file.\n"));
3140           ret = err;
3141           goto exit;
3142         case WARC_TMP_FOPENERR: case WARC_TMP_FWRITEERR:
3143           /* A fatal WARC error. */
3144           logputs (LOG_VERBOSE, "\n");
3145           logprintf (LOG_NOTQUIET, _("Cannot write to temporary WARC file.\n"));
3146           ret = err;
3147           goto exit;
3148         case CONSSLERR:
3149           /* Another fatal error.  */
3150           logprintf (LOG_NOTQUIET, _("Unable to establish SSL connection.\n"));
3151           ret = err;
3152           goto exit;
3153         case UNLINKERR:
3154           /* Another fatal error.  */
3155           logputs (LOG_VERBOSE, "\n");
3156           logprintf (LOG_NOTQUIET, _("Cannot unlink %s (%s).\n"),
3157                      quote (hstat.local_file), strerror (errno));
3158           ret = err;
3159           goto exit;
3160         case NEWLOCATION:
3161         case NEWLOCATION_KEEP_POST:
3162           /* Return the new location to the caller.  */
3163           if (!*newloc)
3164             {
3165               logprintf (LOG_NOTQUIET,
3166                          _("ERROR: Redirection (%d) without location.\n"),
3167                          hstat.statcode);
3168               ret = WRONGCODE;
3169             }
3170           else
3171             {
3172               ret = err;
3173             }
3174           goto exit;
3175         case RETRUNNEEDED:
3176           /* The file was already fully retrieved. */
3177           ret = RETROK;
3178           goto exit;
3179         case RETRFINISHED:
3180           /* Deal with you later.  */
3181           break;
3182         default:
3183           /* All possibilities should have been exhausted.  */
3184           abort ();
3185         }
3186
3187       if (!(*dt & RETROKF))
3188         {
3189           char *hurl = NULL;
3190           if (!opt.verbose)
3191             {
3192               /* #### Ugly ugly ugly! */
3193               hurl = url_string (u, URL_AUTH_HIDE_PASSWD);
3194               logprintf (LOG_NONVERBOSE, "%s:\n", hurl);
3195             }
3196
3197           /* Fall back to GET if HEAD fails with a 500 or 501 error code. */
3198           if (*dt & HEAD_ONLY
3199               && (hstat.statcode == 500 || hstat.statcode == 501))
3200             {
3201               got_head = true;
3202               continue;
3203             }
3204           /* Maybe we should always keep track of broken links, not just in
3205            * spider mode.
3206            * Don't log error if it was UTF-8 encoded because we will try
3207            * once unencoded. */
3208           else if (opt.spider && !iri->utf8_encode)
3209             {
3210               /* #### Again: ugly ugly ugly! */
3211               if (!hurl)
3212                 hurl = url_string (u, URL_AUTH_HIDE_PASSWD);
3213               nonexisting_url (hurl);
3214               logprintf (LOG_NOTQUIET, _("\
3215 Remote file does not exist -- broken link!!!\n"));
3216             }
3217           else
3218             {
3219               logprintf (LOG_NOTQUIET, _("%s ERROR %d: %s.\n"),
3220                          tms, hstat.statcode,
3221                          quotearg_style (escape_quoting_style, hstat.error));
3222             }
3223           logputs (LOG_VERBOSE, "\n");
3224           ret = WRONGCODE;
3225           xfree_null (hurl);
3226           goto exit;
3227         }
3228
3229       /* Did we get the time-stamp? */
3230       if (!got_head)
3231         {
3232           got_head = true;    /* no more time-stamping */
3233
3234           if (opt.timestamping && !hstat.remote_time)
3235             {
3236               logputs (LOG_NOTQUIET, _("\
3237 Last-modified header missing -- time-stamps turned off.\n"));
3238             }
3239           else if (hstat.remote_time)
3240             {
3241               /* Convert the date-string into struct tm.  */
3242               tmr = http_atotm (hstat.remote_time);
3243               if (tmr == (time_t) (-1))
3244                 logputs (LOG_VERBOSE, _("\
3245 Last-modified header invalid -- time-stamp ignored.\n"));
3246               if (*dt & HEAD_ONLY)
3247                 time_came_from_head = true;
3248             }
3249
3250           if (send_head_first)
3251             {
3252               /* The time-stamping section.  */
3253               if (opt.timestamping)
3254                 {
3255                   if (hstat.orig_file_name) /* Perform the following
3256                                                checks only if the file
3257                                                we're supposed to
3258                                                download already exists.  */
3259                     {
3260                       if (hstat.remote_time &&
3261                           tmr != (time_t) (-1))
3262                         {
3263                           /* Now time-stamping can be used validly.
3264                              Time-stamping means that if the sizes of
3265                              the local and remote file match, and local
3266                              file is newer than the remote file, it will
3267                              not be retrieved.  Otherwise, the normal
3268                              download procedure is resumed.  */
3269                           if (hstat.orig_file_tstamp >= tmr)
3270                             {
3271                               if (hstat.contlen == -1
3272                                   || hstat.orig_file_size == hstat.contlen)
3273                                 {
3274                                   logprintf (LOG_VERBOSE, _("\
3275 Server file no newer than local file %s -- not retrieving.\n\n"),
3276                                              quote (hstat.orig_file_name));
3277                                   ret = RETROK;
3278                                   goto exit;
3279                                 }
3280                               else
3281                                 {
3282                                   logprintf (LOG_VERBOSE, _("\
3283 The sizes do not match (local %s) -- retrieving.\n"),
3284                                              number_to_static_string (hstat.orig_file_size));
3285                                 }
3286                             }
3287                           else
3288                             {
3289                               force_full_retrieve = true;
3290                               logputs (LOG_VERBOSE,
3291                                        _("Remote file is newer, retrieving.\n"));
3292                             }
3293
3294                           logputs (LOG_VERBOSE, "\n");
3295                         }
3296                     }
3297
3298                   /* free_hstat (&hstat); */
3299                   hstat.timestamp_checked = true;
3300                 }
3301
3302               if (opt.spider)
3303                 {
3304                   bool finished = true;
3305                   if (opt.recursive)
3306                     {
3307                       if (*dt & TEXTHTML)
3308                         {
3309                           logputs (LOG_VERBOSE, _("\
3310 Remote file exists and could contain links to other resources -- retrieving.\n\n"));
3311                           finished = false;
3312                         }
3313                       else
3314                         {
3315                           logprintf (LOG_VERBOSE, _("\
3316 Remote file exists but does not contain any link -- not retrieving.\n\n"));
3317                           ret = RETROK; /* RETRUNNEEDED is not for caller. */
3318                         }
3319                     }
3320                   else
3321                     {
3322                       if (*dt & TEXTHTML)
3323                         {
3324                           logprintf (LOG_VERBOSE, _("\
3325 Remote file exists and could contain further links,\n\
3326 but recursion is disabled -- not retrieving.\n\n"));
3327                         }
3328                       else
3329                         {
3330                           logprintf (LOG_VERBOSE, _("\
3331 Remote file exists.\n\n"));
3332                         }
3333                       ret = RETROK; /* RETRUNNEEDED is not for caller. */
3334                     }
3335
3336                   if (finished)
3337                     {
3338                       logprintf (LOG_NONVERBOSE,
3339                                  _("%s URL: %s %2d %s\n"),
3340                                  tms, u->url, hstat.statcode,
3341                                  hstat.message ? quotearg_style (escape_quoting_style, hstat.message) : "");
3342                       goto exit;
3343                     }
3344                 }
3345
3346               got_name = true;
3347               *dt &= ~HEAD_ONLY;
3348               count = 0;          /* the retrieve count for HEAD is reset */
3349               continue;
3350             } /* send_head_first */
3351         } /* !got_head */
3352
3353       if (opt.useservertimestamps
3354           && (tmr != (time_t) (-1))
3355           && ((hstat.len == hstat.contlen) ||
3356               ((hstat.res == 0) && (hstat.contlen == -1))))
3357         {
3358           const char *fl = NULL;
3359           set_local_file (&fl, hstat.local_file);
3360           if (fl)
3361             {
3362               time_t newtmr = -1;
3363               /* Reparse time header, in case it's changed. */
3364               if (time_came_from_head
3365                   && hstat.remote_time && hstat.remote_time[0])
3366                 {
3367                   newtmr = http_atotm (hstat.remote_time);
3368                   if (newtmr != (time_t)-1)
3369                     tmr = newtmr;
3370                 }
3371               touch (fl, tmr);
3372             }
3373         }
3374       /* End of time-stamping section. */
3375
3376       tmrate = retr_rate (hstat.rd_size, hstat.dltime);
3377       total_download_time += hstat.dltime;
3378
3379       if (hstat.len == hstat.contlen)
3380         {
3381           if (*dt & RETROKF)
3382             {
3383               bool write_to_stdout = (opt.output_document && HYPHENP (opt.output_document));
3384
3385               logprintf (LOG_VERBOSE,
3386                          write_to_stdout
3387                          ? _("%s (%s) - written to stdout %s[%s/%s]\n\n")
3388                          : _("%s (%s) - %s saved [%s/%s]\n\n"),
3389                          tms, tmrate,
3390                          write_to_stdout ? "" : quote (hstat.local_file),
3391                          number_to_static_string (hstat.len),
3392                          number_to_static_string (hstat.contlen));
3393               logprintf (LOG_NONVERBOSE,
3394                          "%s URL:%s [%s/%s] -> \"%s\" [%d]\n",
3395                          tms, u->url,
3396                          number_to_static_string (hstat.len),
3397                          number_to_static_string (hstat.contlen),
3398                          hstat.local_file, count);
3399             }
3400           ++numurls;
3401           total_downloaded_bytes += hstat.rd_size;
3402
3403           /* Remember that we downloaded the file for later ".orig" code. */
3404           if (*dt & ADDED_HTML_EXTENSION)
3405             downloaded_file (FILE_DOWNLOADED_AND_HTML_EXTENSION_ADDED, hstat.local_file);
3406           else
3407             downloaded_file (FILE_DOWNLOADED_NORMALLY, hstat.local_file);
3408
3409           ret = RETROK;
3410           goto exit;
3411         }
3412       else if (hstat.res == 0) /* No read error */
3413         {
3414           if (hstat.contlen == -1)  /* We don't know how much we were supposed
3415                                        to get, so assume we succeeded. */
3416             {
3417               if (*dt & RETROKF)
3418                 {
3419                   bool write_to_stdout = (opt.output_document && HYPHENP (opt.output_document));
3420
3421                   logprintf (LOG_VERBOSE,
3422                              write_to_stdout
3423                              ? _("%s (%s) - written to stdout %s[%s]\n\n")
3424                              : _("%s (%s) - %s saved [%s]\n\n"),
3425                              tms, tmrate,
3426                              write_to_stdout ? "" : quote (hstat.local_file),
3427                              number_to_static_string (hstat.len));
3428                   logprintf (LOG_NONVERBOSE,
3429                              "%s URL:%s [%s] -> \"%s\" [%d]\n",
3430                              tms, u->url, number_to_static_string (hstat.len),
3431                              hstat.local_file, count);
3432                 }
3433               ++numurls;
3434               total_downloaded_bytes += hstat.rd_size;
3435
3436               /* Remember that we downloaded the file for later ".orig" code. */
3437               if (*dt & ADDED_HTML_EXTENSION)
3438                 downloaded_file (FILE_DOWNLOADED_AND_HTML_EXTENSION_ADDED, hstat.local_file);
3439               else
3440                 downloaded_file (FILE_DOWNLOADED_NORMALLY, hstat.local_file);
3441
3442               ret = RETROK;
3443               goto exit;
3444             }
3445           else if (hstat.len < hstat.contlen) /* meaning we lost the
3446                                                  connection too soon */
3447             {
3448               logprintf (LOG_VERBOSE,
3449                          _("%s (%s) - Connection closed at byte %s. "),
3450                          tms, tmrate, number_to_static_string (hstat.len));
3451               printwhat (count, opt.ntry);
3452               continue;
3453             }
3454           else if (hstat.len != hstat.restval)
3455             /* Getting here would mean reading more data than
3456                requested with content-length, which we never do.  */
3457             abort ();
3458           else
3459             {
3460               /* Getting here probably means that the content-length was
3461                * _less_ than the original, local size. We should probably
3462                * truncate or re-read, or something. FIXME */
3463               ret = RETROK;
3464               goto exit;
3465             }
3466         }
3467       else /* from now on hstat.res can only be -1 */
3468         {
3469           if (hstat.contlen == -1)
3470             {
3471               logprintf (LOG_VERBOSE,
3472                          _("%s (%s) - Read error at byte %s (%s)."),
3473                          tms, tmrate, number_to_static_string (hstat.len),
3474                          hstat.rderrmsg);
3475               printwhat (count, opt.ntry);
3476               continue;
3477             }
3478           else /* hstat.res == -1 and contlen is given */
3479             {
3480               logprintf (LOG_VERBOSE,
3481                          _("%s (%s) - Read error at byte %s/%s (%s). "),
3482                          tms, tmrate,
3483                          number_to_static_string (hstat.len),
3484                          number_to_static_string (hstat.contlen),
3485                          hstat.rderrmsg);
3486               printwhat (count, opt.ntry);
3487               continue;
3488             }
3489         }
3490       /* not reached */
3491     }
3492   while (!opt.ntry || (count < opt.ntry));
3493
3494 exit:
3495   if (ret == RETROK && local_file)
3496     *local_file = xstrdup (hstat.local_file);
3497   free_hstat (&hstat);
3498
3499   return ret;
3500 }
3501 \f
3502 /* Check whether the result of strptime() indicates success.
3503    strptime() returns the pointer to how far it got to in the string.
3504    The processing has been successful if the string is at `GMT' or
3505    `+X', or at the end of the string.
3506
3507    In extended regexp parlance, the function returns 1 if P matches
3508    "^ *(GMT|[+-][0-9]|$)", 0 otherwise.  P being NULL (which strptime
3509    can return) is considered a failure and 0 is returned.  */
3510 static bool
3511 check_end (const char *p)
3512 {
3513   if (!p)
3514     return false;
3515   while (c_isspace (*p))
3516     ++p;
3517   if (!*p
3518       || (p[0] == 'G' && p[1] == 'M' && p[2] == 'T')
3519       || ((p[0] == '+' || p[0] == '-') && c_isdigit (p[1])))
3520     return true;
3521   else
3522     return false;
3523 }
3524
3525 /* Convert the textual specification of time in TIME_STRING to the
3526    number of seconds since the Epoch.
3527
3528    TIME_STRING can be in any of the three formats RFC2616 allows the
3529    HTTP servers to emit -- RFC1123-date, RFC850-date or asctime-date,
3530    as well as the time format used in the Set-Cookie header.
3531    Timezones are ignored, and should be GMT.
3532
3533    Return the computed time_t representation, or -1 if the conversion
3534    fails.
3535
3536    This function uses strptime with various string formats for parsing
3537    TIME_STRING.  This results in a parser that is not as lenient in
3538    interpreting TIME_STRING as I would like it to be.  Being based on
3539    strptime, it always allows shortened months, one-digit days, etc.,
3540    but due to the multitude of formats in which time can be
3541    represented, an ideal HTTP time parser would be even more
3542    forgiving.  It should completely ignore things like week days and
3543    concentrate only on the various forms of representing years,
3544    months, days, hours, minutes, and seconds.  For example, it would
3545    be nice if it accepted ISO 8601 out of the box.
3546
3547    I've investigated free and PD code for this purpose, but none was
3548    usable.  getdate was big and unwieldy, and had potential copyright
3549    issues, or so I was informed.  Dr. Marcus Hennecke's atotm(),
3550    distributed with phttpd, is excellent, but we cannot use it because
3551    it is not assigned to the FSF.  So I stuck it with strptime.  */
3552
3553 time_t
3554 http_atotm (const char *time_string)
3555 {
3556   /* NOTE: Solaris strptime man page claims that %n and %t match white
3557      space, but that's not universally available.  Instead, we simply
3558      use ` ' to mean "skip all WS", which works under all strptime
3559      implementations I've tested.  */
3560
3561   static const char *time_formats[] = {
3562     "%a, %d %b %Y %T",          /* rfc1123: Thu, 29 Jan 1998 22:12:57 */
3563     "%A, %d-%b-%y %T",          /* rfc850:  Thursday, 29-Jan-98 22:12:57 */
3564     "%a %b %d %T %Y",           /* asctime: Thu Jan 29 22:12:57 1998 */
3565     "%a, %d-%b-%Y %T"           /* cookies: Thu, 29-Jan-1998 22:12:57
3566                                    (used in Set-Cookie, defined in the
3567                                    Netscape cookie specification.) */
3568   };
3569   const char *oldlocale;
3570   char savedlocale[256];
3571   size_t i;
3572   time_t ret = (time_t) -1;
3573
3574   /* Solaris strptime fails to recognize English month names in
3575      non-English locales, which we work around by temporarily setting
3576      locale to C before invoking strptime.  */
3577   oldlocale = setlocale (LC_TIME, NULL);
3578   if (oldlocale)
3579     {
3580       size_t l = strlen (oldlocale) + 1;
3581       if (l >= sizeof savedlocale)
3582         savedlocale[0] = '\0';
3583       else
3584         memcpy (savedlocale, oldlocale, l);
3585     }
3586   else savedlocale[0] = '\0';
3587
3588   setlocale (LC_TIME, "C");
3589
3590   for (i = 0; i < countof (time_formats); i++)
3591     {
3592       struct tm t;
3593
3594       /* Some versions of strptime use the existing contents of struct
3595          tm to recalculate the date according to format.  Zero it out
3596          to prevent stack garbage from influencing strptime.  */
3597       xzero (t);
3598
3599       if (check_end (strptime (time_string, time_formats[i], &t)))
3600         {
3601           ret = timegm (&t);
3602           break;
3603         }
3604     }
3605
3606   /* Restore the previous locale. */
3607   if (savedlocale[0])
3608     setlocale (LC_TIME, savedlocale);
3609
3610   return ret;
3611 }
3612 \f
3613 /* Authorization support: We support three authorization schemes:
3614
3615    * `Basic' scheme, consisting of base64-ing USER:PASSWORD string;
3616
3617    * `Digest' scheme, added by Junio Hamano <junio@twinsun.com>,
3618    consisting of answering to the server's challenge with the proper
3619    MD5 digests.
3620
3621    * `NTLM' ("NT Lan Manager") scheme, based on code written by Daniel
3622    Stenberg for libcurl.  Like digest, NTLM is based on a
3623    challenge-response mechanism, but unlike digest, it is non-standard
3624    (authenticates TCP connections rather than requests), undocumented
3625    and Microsoft-specific.  */
3626
3627 /* Create the authentication header contents for the `Basic' scheme.
3628    This is done by encoding the string "USER:PASS" to base64 and
3629    prepending the string "Basic " in front of it.  */
3630
3631 static char *
3632 basic_authentication_encode (const char *user, const char *passwd)
3633 {
3634   char *t1, *t2;
3635   int len1 = strlen (user) + 1 + strlen (passwd);
3636
3637   t1 = (char *)alloca (len1 + 1);
3638   sprintf (t1, "%s:%s", user, passwd);
3639
3640   t2 = (char *)alloca (BASE64_LENGTH (len1) + 1);
3641   base64_encode (t1, len1, t2);
3642
3643   return concat_strings ("Basic ", t2, (char *) 0);
3644 }
3645
3646 #define SKIP_WS(x) do {                         \
3647   while (c_isspace (*(x)))                        \
3648     ++(x);                                      \
3649 } while (0)
3650
3651 #ifdef ENABLE_DIGEST
3652 /* Dump the hexadecimal representation of HASH to BUF.  HASH should be
3653    an array of 16 bytes containing the hash keys, and BUF should be a
3654    buffer of 33 writable characters (32 for hex digits plus one for
3655    zero termination).  */
3656 static void
3657 dump_hash (char *buf, const unsigned char *hash)
3658 {
3659   int i;
3660
3661   for (i = 0; i < MD5_DIGEST_SIZE; i++, hash++)
3662     {
3663       *buf++ = XNUM_TO_digit (*hash >> 4);
3664       *buf++ = XNUM_TO_digit (*hash & 0xf);
3665     }
3666   *buf = '\0';
3667 }
3668
3669 /* Take the line apart to find the challenge, and compose a digest
3670    authorization header.  See RFC2069 section 2.1.2.  */
3671 static char *
3672 digest_authentication_encode (const char *au, const char *user,
3673                               const char *passwd, const char *method,
3674                               const char *path)
3675 {
3676   static char *realm, *opaque, *nonce, *qop, *algorithm;
3677   static struct {
3678     const char *name;
3679     char **variable;
3680   } options[] = {
3681     { "realm", &realm },
3682     { "opaque", &opaque },
3683     { "nonce", &nonce },
3684     { "qop", &qop },
3685     { "algorithm", &algorithm }
3686   };
3687   char cnonce[16] = "";
3688   char *res;
3689   int res_len;
3690   size_t res_size;
3691   param_token name, value;
3692
3693
3694   realm = opaque = nonce = qop = algorithm = NULL;
3695
3696   au += 6;                      /* skip over `Digest' */
3697   while (extract_param (&au, &name, &value, ','))
3698     {
3699       size_t i;
3700       size_t namelen = name.e - name.b;
3701       for (i = 0; i < countof (options); i++)
3702         if (namelen == strlen (options[i].name)
3703             && 0 == strncmp (name.b, options[i].name,
3704                              namelen))
3705           {
3706             *options[i].variable = strdupdelim (value.b, value.e);
3707             break;
3708           }
3709     }
3710
3711   if (qop != NULL && strcmp(qop,"auth"))
3712     {
3713       logprintf (LOG_NOTQUIET, _("Unsupported quality of protection '%s'.\n"), qop);
3714       user = NULL; /* force freeing mem and return */
3715     }
3716
3717   if (algorithm != NULL && strcmp (algorithm,"MD5") && strcmp (algorithm,"MD5-sess"))
3718     {
3719       logprintf (LOG_NOTQUIET, _("Unsupported algorithm '%s'.\n"), algorithm);
3720       user = NULL; /* force freeing mem and return */
3721     }
3722
3723   if (!realm || !nonce || !user || !passwd || !path || !method)
3724     {
3725       xfree_null (realm);
3726       xfree_null (opaque);
3727       xfree_null (nonce);
3728       xfree_null (qop);
3729       xfree_null (algorithm);
3730       return NULL;
3731     }
3732
3733   /* Calculate the digest value.  */
3734   {
3735     struct md5_ctx ctx;
3736     unsigned char hash[MD5_DIGEST_SIZE];
3737     char a1buf[MD5_DIGEST_SIZE * 2 + 1], a2buf[MD5_DIGEST_SIZE * 2 + 1];
3738     char response_digest[MD5_DIGEST_SIZE * 2 + 1];
3739
3740     /* A1BUF = H(user ":" realm ":" password) */
3741     md5_init_ctx (&ctx);
3742     md5_process_bytes ((unsigned char *)user, strlen (user), &ctx);
3743     md5_process_bytes ((unsigned char *)":", 1, &ctx);
3744     md5_process_bytes ((unsigned char *)realm, strlen (realm), &ctx);
3745     md5_process_bytes ((unsigned char *)":", 1, &ctx);
3746     md5_process_bytes ((unsigned char *)passwd, strlen (passwd), &ctx);
3747     md5_finish_ctx (&ctx, hash);
3748
3749     dump_hash (a1buf, hash);
3750
3751     if (! strcmp (algorithm, "MD5-sess"))
3752       {
3753         /* A1BUF = H( H(user ":" realm ":" password) ":" nonce ":" cnonce ) */
3754         snprintf (cnonce, sizeof (cnonce), "%08x", random_number(INT_MAX));
3755
3756         md5_init_ctx (&ctx);
3757         // md5_process_bytes (hash, MD5_DIGEST_SIZE, &ctx);
3758         md5_process_bytes (a1buf, MD5_DIGEST_SIZE * 2, &ctx);
3759         md5_process_bytes ((unsigned char *)":", 1, &ctx);
3760         md5_process_bytes ((unsigned char *)nonce, strlen (nonce), &ctx);
3761         md5_process_bytes ((unsigned char *)":", 1, &ctx);
3762         md5_process_bytes ((unsigned char *)cnonce, strlen (cnonce), &ctx);
3763         md5_finish_ctx (&ctx, hash);
3764
3765         dump_hash (a1buf, hash);
3766       }
3767
3768     /* A2BUF = H(method ":" path) */
3769     md5_init_ctx (&ctx);
3770     md5_process_bytes ((unsigned char *)method, strlen (method), &ctx);
3771     md5_process_bytes ((unsigned char *)":", 1, &ctx);
3772     md5_process_bytes ((unsigned char *)path, strlen (path), &ctx);
3773     md5_finish_ctx (&ctx, hash);
3774     dump_hash (a2buf, hash);
3775
3776     if (!strcmp(qop, "auth") || !strcmp (qop, "auth-int"))
3777       {
3778         /* RFC 2617 Digest Access Authentication */
3779         /* generate random hex string */
3780         if (!*cnonce)
3781           snprintf(cnonce, sizeof(cnonce), "%08x", random_number(INT_MAX));
3782
3783         /* RESPONSE_DIGEST = H(A1BUF ":" nonce ":" noncecount ":" clientnonce ":" qop ": " A2BUF) */
3784         md5_init_ctx (&ctx);
3785         md5_process_bytes ((unsigned char *)a1buf, MD5_DIGEST_SIZE * 2, &ctx);
3786         md5_process_bytes ((unsigned char *)":", 1, &ctx);
3787         md5_process_bytes ((unsigned char *)nonce, strlen (nonce), &ctx);
3788         md5_process_bytes ((unsigned char *)":", 1, &ctx);
3789         md5_process_bytes ((unsigned char *)"00000001", 8, &ctx); /* TODO: keep track of server nonce values */
3790         md5_process_bytes ((unsigned char *)":", 1, &ctx);
3791         md5_process_bytes ((unsigned char *)cnonce, strlen(cnonce), &ctx);
3792         md5_process_bytes ((unsigned char *)":", 1, &ctx);
3793         md5_process_bytes ((unsigned char *)qop, strlen(qop), &ctx);
3794         md5_process_bytes ((unsigned char *)":", 1, &ctx);
3795         md5_process_bytes ((unsigned char *)a2buf, MD5_DIGEST_SIZE * 2, &ctx);
3796         md5_finish_ctx (&ctx, hash);
3797       }
3798     else
3799       {
3800         /* RFC 2069 Digest Access Authentication */
3801         /* RESPONSE_DIGEST = H(A1BUF ":" nonce ":" A2BUF) */
3802         md5_init_ctx (&ctx);
3803         md5_process_bytes ((unsigned char *)a1buf, MD5_DIGEST_SIZE * 2, &ctx);
3804         md5_process_bytes ((unsigned char *)":", 1, &ctx);
3805         md5_process_bytes ((unsigned char *)nonce, strlen (nonce), &ctx);
3806         md5_process_bytes ((unsigned char *)":", 1, &ctx);
3807         md5_process_bytes ((unsigned char *)a2buf, MD5_DIGEST_SIZE * 2, &ctx);
3808         md5_finish_ctx (&ctx, hash);
3809       }
3810
3811     dump_hash (response_digest, hash);
3812
3813     res_size = strlen (user)
3814              + strlen (realm)
3815              + strlen (nonce)
3816              + strlen (path)
3817              + 2 * MD5_DIGEST_SIZE /*strlen (response_digest)*/
3818              + (opaque ? strlen (opaque) : 0)
3819              + (algorithm ? strlen (algorithm) : 0)
3820              + (qop ? 128: 0)
3821              + strlen (cnonce)
3822              + 128;
3823
3824     res = xmalloc (res_size);
3825
3826     if (!strcmp(qop,"auth"))
3827       {
3828         res_len = snprintf (res, res_size, "Digest "\
3829                 "username=\"%s\", realm=\"%s\", nonce=\"%s\", uri=\"%s\", response=\"%s\""\
3830                 ", qop=auth, nc=00000001, cnonce=\"%s\"",
3831                   user, realm, nonce, path, response_digest, cnonce);
3832
3833       }
3834     else
3835       {
3836         res_len = snprintf (res, res_size, "Digest "\
3837                 "username=\"%s\", realm=\"%s\", nonce=\"%s\", uri=\"%s\", response=\"%s\"",
3838                   user, realm, nonce, path, response_digest);
3839       }
3840
3841     if (opaque)
3842       {
3843         res_len += snprintf(res + res_len, res_size - res_len, ", opaque=\"%s\"", opaque);
3844       }
3845
3846     if (algorithm)
3847       {
3848         snprintf(res + res_len, res_size - res_len, ", algorithm=\"%s\"", algorithm);
3849       }
3850   }
3851   return res;
3852 }
3853 #endif /* ENABLE_DIGEST */
3854
3855 /* Computing the size of a string literal must take into account that
3856    value returned by sizeof includes the terminating \0.  */
3857 #define STRSIZE(literal) (sizeof (literal) - 1)
3858
3859 /* Whether chars in [b, e) begin with the literal string provided as
3860    first argument and are followed by whitespace or terminating \0.
3861    The comparison is case-insensitive.  */
3862 #define STARTS(literal, b, e)                           \
3863   ((e > b) \
3864    && ((size_t) ((e) - (b))) >= STRSIZE (literal)   \
3865    && 0 == strncasecmp (b, literal, STRSIZE (literal))  \
3866    && ((size_t) ((e) - (b)) == STRSIZE (literal)          \
3867        || c_isspace (b[STRSIZE (literal)])))
3868
3869 static bool
3870 known_authentication_scheme_p (const char *hdrbeg, const char *hdrend)
3871 {
3872   return STARTS ("Basic", hdrbeg, hdrend)
3873 #ifdef ENABLE_DIGEST
3874     || STARTS ("Digest", hdrbeg, hdrend)
3875 #endif
3876 #ifdef ENABLE_NTLM
3877     || STARTS ("NTLM", hdrbeg, hdrend)
3878 #endif
3879     ;
3880 }
3881
3882 #undef STARTS
3883
3884 /* Create the HTTP authorization request header.  When the
3885    `WWW-Authenticate' response header is seen, according to the
3886    authorization scheme specified in that header (`Basic' and `Digest'
3887    are supported by the current implementation), produce an
3888    appropriate HTTP authorization request header.  */
3889 static char *
3890 create_authorization_line (const char *au, const char *user,
3891                            const char *passwd, const char *method,
3892                            const char *path, bool *finished)
3893 {
3894   /* We are called only with known schemes, so we can dispatch on the
3895      first letter. */
3896   switch (c_toupper (*au))
3897     {
3898     case 'B':                   /* Basic */
3899       *finished = true;
3900       return basic_authentication_encode (user, passwd);
3901 #ifdef ENABLE_DIGEST
3902     case 'D':                   /* Digest */
3903       *finished = true;
3904       return digest_authentication_encode (au, user, passwd, method, path);
3905 #endif
3906 #ifdef ENABLE_NTLM
3907     case 'N':                   /* NTLM */
3908       if (!ntlm_input (&pconn.ntlm, au))
3909         {
3910           *finished = true;
3911           return NULL;
3912         }
3913       return ntlm_output (&pconn.ntlm, user, passwd, finished);
3914 #endif
3915     default:
3916       /* We shouldn't get here -- this function should be only called
3917          with values approved by known_authentication_scheme_p.  */
3918       abort ();
3919     }
3920 }
3921 \f
3922 static void
3923 load_cookies (void)
3924 {
3925   if (!wget_cookie_jar)
3926     wget_cookie_jar = cookie_jar_new ();
3927   if (opt.cookies_input && !cookies_loaded_p)
3928     {
3929       cookie_jar_load (wget_cookie_jar, opt.cookies_input);
3930       cookies_loaded_p = true;
3931     }
3932 }
3933
3934 void
3935 save_cookies (void)
3936 {
3937   if (wget_cookie_jar)
3938     cookie_jar_save (wget_cookie_jar, opt.cookies_output);
3939 }
3940
3941 void
3942 http_cleanup (void)
3943 {
3944   xfree_null (pconn.host);
3945   if (wget_cookie_jar)
3946     cookie_jar_delete (wget_cookie_jar);
3947 }
3948
3949 void
3950 ensure_extension (struct http_stat *hs, const char *ext, int *dt)
3951 {
3952   char *last_period_in_local_filename = strrchr (hs->local_file, '.');
3953   char shortext[8];
3954   int len = strlen (ext);
3955   if (len == 5)
3956     {
3957       strncpy (shortext, ext, len - 1);
3958       shortext[len - 1] = '\0';
3959     }
3960
3961   if (last_period_in_local_filename == NULL
3962       || !(0 == strcasecmp (last_period_in_local_filename, shortext)
3963            || 0 == strcasecmp (last_period_in_local_filename, ext)))
3964     {
3965       int local_filename_len = strlen (hs->local_file);
3966       /* Resize the local file, allowing for ".html" preceded by
3967          optional ".NUMBER".  */
3968       hs->local_file = xrealloc (hs->local_file,
3969                                  local_filename_len + 24 + len);
3970       strcpy (hs->local_file + local_filename_len, ext);
3971       /* If clobbering is not allowed and the file, as named,
3972          exists, tack on ".NUMBER.html" instead. */
3973       if (!ALLOW_CLOBBER && file_exists_p (hs->local_file))
3974         {
3975           int ext_num = 1;
3976           do
3977             sprintf (hs->local_file + local_filename_len,
3978                      ".%d%s", ext_num++, ext);
3979           while (file_exists_p (hs->local_file));
3980         }
3981       *dt |= ADDED_HTML_EXTENSION;
3982     }
3983 }
3984
3985
3986 #ifdef TESTING
3987
3988 const char *
3989 test_parse_content_disposition()
3990 {
3991   int i;
3992   struct {
3993     char *hdrval;
3994     char *filename;
3995     bool result;
3996   } test_array[] = {
3997     { "filename=\"file.ext\"", "file.ext", true },
3998     { "attachment; filename=\"file.ext\"", "file.ext", true },
3999     { "attachment; filename=\"file.ext\"; dummy", "file.ext", true },
4000     { "attachment", NULL, false },
4001     { "attachement; filename*=UTF-8'en-US'hello.txt", "hello.txt", true },
4002     { "attachement; filename*0=\"hello\"; filename*1=\"world.txt\"", "helloworld.txt", true },
4003   };
4004
4005   for (i = 0; i < sizeof(test_array)/sizeof(test_array[0]); ++i)
4006     {
4007       char *filename;
4008       bool res;
4009
4010       res = parse_content_disposition (test_array[i].hdrval, &filename);
4011
4012       mu_assert ("test_parse_content_disposition: wrong result",
4013                  res == test_array[i].result
4014                  && (res == false
4015                      || 0 == strcmp (test_array[i].filename, filename)));
4016     }
4017
4018   return NULL;
4019 }
4020
4021 #endif /* TESTING */
4022
4023 /*
4024  * vim: et sts=2 sw=2 cino+={s
4025  */
4026