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