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