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