]> sjero.net Git - wget/blob - src/http.c
[svn] New option --keep-session-cookies.
[wget] / src / http.c
1 /* HTTP support.
2    Copyright (C) 1995, 1996, 1997, 1998, 2000, 2001, 2002
3    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 2 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, write to the Free Software
19 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20
21 In addition, as a special exception, the Free Software Foundation
22 gives permission to link the code of its release of Wget with the
23 OpenSSL project's "OpenSSL" library (or with modified versions of it
24 that use the same license as the "OpenSSL" library), and distribute
25 the linked executables.  You must obey the GNU General Public License
26 in all respects for all of the code used other than "OpenSSL".  If you
27 modify this file, you may extend this exception to your version of the
28 file, but you are not obligated to do so.  If you do not wish to do
29 so, delete this exception statement from your version.  */
30
31 #include <config.h>
32
33 #include <stdio.h>
34 #include <stdlib.h>
35 #include <sys/types.h>
36 #ifdef HAVE_STRING_H
37 # include <string.h>
38 #else
39 # include <strings.h>
40 #endif
41 #ifdef HAVE_UNISTD_H
42 # include <unistd.h>
43 #endif
44 #include <assert.h>
45 #include <errno.h>
46 #if TIME_WITH_SYS_TIME
47 # include <sys/time.h>
48 # include <time.h>
49 #else
50 # if HAVE_SYS_TIME_H
51 #  include <sys/time.h>
52 # else
53 #  include <time.h>
54 # endif
55 #endif
56 #ifndef errno
57 extern int errno;
58 #endif
59
60 #include "wget.h"
61 #include "utils.h"
62 #include "url.h"
63 #include "host.h"
64 #include "rbuf.h"
65 #include "retr.h"
66 #include "headers.h"
67 #include "connect.h"
68 #include "netrc.h"
69 #ifdef HAVE_SSL
70 # include "gen_sslfunc.h"
71 #endif /* HAVE_SSL */
72 #include "cookies.h"
73 #ifdef USE_DIGEST
74 # include "gen-md5.h"
75 #endif
76 #include "convert.h"
77
78 extern char *version_string;
79 extern LARGE_INT total_downloaded_bytes;
80
81 \f
82 static int cookies_loaded_p;
83 struct cookie_jar *wget_cookie_jar;
84
85 #define TEXTHTML_S "text/html"
86 #define TEXTXHTML_S "application/xhtml+xml"
87 #define HTTP_ACCEPT "*/*"
88
89 /* Some status code validation macros: */
90 #define H_20X(x)        (((x) >= 200) && ((x) < 300))
91 #define H_PARTIAL(x)    ((x) == HTTP_STATUS_PARTIAL_CONTENTS)
92 #define H_REDIRECTED(x) ((x) == HTTP_STATUS_MOVED_PERMANENTLY   \
93                          || (x) == HTTP_STATUS_MOVED_TEMPORARILY \
94                          || (x) == HTTP_STATUS_TEMPORARY_REDIRECT)
95
96 /* HTTP/1.0 status codes from RFC1945, provided for reference.  */
97 /* Successful 2xx.  */
98 #define HTTP_STATUS_OK                  200
99 #define HTTP_STATUS_CREATED             201
100 #define HTTP_STATUS_ACCEPTED            202
101 #define HTTP_STATUS_NO_CONTENT          204
102 #define HTTP_STATUS_PARTIAL_CONTENTS    206
103
104 /* Redirection 3xx.  */
105 #define HTTP_STATUS_MULTIPLE_CHOICES    300
106 #define HTTP_STATUS_MOVED_PERMANENTLY   301
107 #define HTTP_STATUS_MOVED_TEMPORARILY   302
108 #define HTTP_STATUS_NOT_MODIFIED        304
109 #define HTTP_STATUS_TEMPORARY_REDIRECT  307
110
111 /* Client error 4xx.  */
112 #define HTTP_STATUS_BAD_REQUEST         400
113 #define HTTP_STATUS_UNAUTHORIZED        401
114 #define HTTP_STATUS_FORBIDDEN           403
115 #define HTTP_STATUS_NOT_FOUND           404
116
117 /* Server errors 5xx.  */
118 #define HTTP_STATUS_INTERNAL            500
119 #define HTTP_STATUS_NOT_IMPLEMENTED     501
120 #define HTTP_STATUS_BAD_GATEWAY         502
121 #define HTTP_STATUS_UNAVAILABLE         503
122
123 \f
124 /* Parse the HTTP status line, which is of format:
125
126    HTTP-Version SP Status-Code SP Reason-Phrase
127
128    The function returns the status-code, or -1 if the status line is
129    malformed.  The pointer to reason-phrase is returned in RP.  */
130 static int
131 parse_http_status_line (const char *line, const char **reason_phrase_ptr)
132 {
133   /* (the variables must not be named `major' and `minor', because
134      that breaks compilation with SunOS4 cc.)  */
135   int mjr, mnr, statcode;
136   const char *p;
137
138   *reason_phrase_ptr = NULL;
139
140   /* The standard format of HTTP-Version is: `HTTP/X.Y', where X is
141      major version, and Y is minor version.  */
142   if (strncmp (line, "HTTP/", 5) != 0)
143     return -1;
144   line += 5;
145
146   /* Calculate major HTTP version.  */
147   p = line;
148   for (mjr = 0; ISDIGIT (*line); line++)
149     mjr = 10 * mjr + (*line - '0');
150   if (*line != '.' || p == line)
151     return -1;
152   ++line;
153
154   /* Calculate minor HTTP version.  */
155   p = line;
156   for (mnr = 0; ISDIGIT (*line); line++)
157     mnr = 10 * mnr + (*line - '0');
158   if (*line != ' ' || p == line)
159     return -1;
160   /* Wget will accept only 1.0 and higher HTTP-versions.  The value of
161      minor version can be safely ignored.  */
162   if (mjr < 1)
163     return -1;
164   ++line;
165
166   /* Calculate status code.  */
167   if (!(ISDIGIT (*line) && ISDIGIT (line[1]) && ISDIGIT (line[2])))
168     return -1;
169   statcode = 100 * (*line - '0') + 10 * (line[1] - '0') + (line[2] - '0');
170
171   /* Set up the reason phrase pointer.  */
172   line += 3;
173   /* RFC2068 requires SPC here, but we allow the string to finish
174      here, in case no reason-phrase is present.  */
175   if (*line != ' ')
176     {
177       if (!*line)
178         *reason_phrase_ptr = line;
179       else
180         return -1;
181     }
182   else
183     *reason_phrase_ptr = line + 1;
184
185   return statcode;
186 }
187 \f
188 #define WMIN(x, y) ((x) > (y) ? (y) : (x))
189
190 /* Send the contents of FILE_NAME to SOCK/SSL.  Make sure that exactly
191    PROMISED_SIZE bytes are sent over the wire -- if the file is
192    longer, read only that much; if the file is shorter, report an error.  */
193
194 static int
195 post_file (int sock, void *ssl, const char *file_name, long promised_size)
196 {
197   static char chunk[8192];
198   long written = 0;
199   int write_error;
200   FILE *fp;
201
202   /* Only one of SOCK and SSL may be active at the same time. */
203   assert (sock > -1 || ssl != NULL);
204   assert (sock == -1 || ssl == NULL);
205
206   DEBUGP (("[writing POST file %s ... ", file_name));
207
208   fp = fopen (file_name, "rb");
209   if (!fp)
210     return -1;
211   while (!feof (fp) && written < promised_size)
212     {
213       int towrite;
214       int length = fread (chunk, 1, sizeof (chunk), fp);
215       if (length == 0)
216         break;
217       towrite = WMIN (promised_size - written, length);
218 #ifdef HAVE_SSL
219       if (ssl)
220         write_error = ssl_iwrite (ssl, chunk, towrite);
221       else
222 #endif
223         write_error = iwrite (sock, chunk, towrite);
224       if (write_error < 0)
225         {
226           fclose (fp);
227           return -1;
228         }
229       written += towrite;
230     }
231   fclose (fp);
232
233   /* If we've written less than was promised, report a (probably
234      nonsensical) error rather than break the promise.  */
235   if (written < promised_size)
236     {
237       errno = EINVAL;
238       return -1;
239     }
240
241   assert (written == promised_size);
242   DEBUGP (("done]\n"));
243   return 0;
244 }
245 \f
246 /* Functions to be used as arguments to header_process(): */
247
248 struct http_process_range_closure {
249   long first_byte_pos;
250   long last_byte_pos;
251   long entity_length;
252 };
253
254 /* Parse the `Content-Range' header and extract the information it
255    contains.  Returns 1 if successful, -1 otherwise.  */
256 static int
257 http_process_range (const char *hdr, void *arg)
258 {
259   struct http_process_range_closure *closure
260     = (struct http_process_range_closure *)arg;
261   long num;
262
263   /* Certain versions of Nutscape proxy server send out
264      `Content-Length' without "bytes" specifier, which is a breach of
265      RFC2068 (as well as the HTTP/1.1 draft which was current at the
266      time).  But hell, I must support it...  */
267   if (!strncasecmp (hdr, "bytes", 5))
268     {
269       hdr += 5;
270       /* "JavaWebServer/1.1.1" sends "bytes: x-y/z", contrary to the
271          HTTP spec. */
272       if (*hdr == ':')
273         ++hdr;
274       hdr += skip_lws (hdr);
275       if (!*hdr)
276         return 0;
277     }
278   if (!ISDIGIT (*hdr))
279     return 0;
280   for (num = 0; ISDIGIT (*hdr); hdr++)
281     num = 10 * num + (*hdr - '0');
282   if (*hdr != '-' || !ISDIGIT (*(hdr + 1)))
283     return 0;
284   closure->first_byte_pos = num;
285   ++hdr;
286   for (num = 0; ISDIGIT (*hdr); hdr++)
287     num = 10 * num + (*hdr - '0');
288   if (*hdr != '/' || !ISDIGIT (*(hdr + 1)))
289     return 0;
290   closure->last_byte_pos = num;
291   ++hdr;
292   for (num = 0; ISDIGIT (*hdr); hdr++)
293     num = 10 * num + (*hdr - '0');
294   closure->entity_length = num;
295   return 1;
296 }
297
298 /* Place 1 to ARG if the HDR contains the word "none", 0 otherwise.
299    Used for `Accept-Ranges'.  */
300 static int
301 http_process_none (const char *hdr, void *arg)
302 {
303   int *where = (int *)arg;
304
305   if (strstr (hdr, "none"))
306     *where = 1;
307   else
308     *where = 0;
309   return 1;
310 }
311
312 /* Place the malloc-ed copy of HDR hdr, to the first `;' to ARG.  */
313 static int
314 http_process_type (const char *hdr, void *arg)
315 {
316   char **result = (char **)arg;
317   /* Locate P on `;' or the terminating zero, whichever comes first. */
318   const char *p = strchr (hdr, ';');
319   if (!p)
320     p = hdr + strlen (hdr);
321   while (p > hdr && ISSPACE (*(p - 1)))
322     --p;
323   *result = strdupdelim (hdr, p);
324   return 1;
325 }
326
327 /* Check whether the `Connection' header is set to "keep-alive". */
328 static int
329 http_process_connection (const char *hdr, void *arg)
330 {
331   int *flag = (int *)arg;
332   if (!strcasecmp (hdr, "Keep-Alive"))
333     *flag = 1;
334   return 1;
335 }
336
337 /* Commit the cookie to the cookie jar. */
338
339 int
340 http_process_set_cookie (const char *hdr, void *arg)
341 {
342   struct url *u = (struct url *)arg;
343
344   /* The jar should have been created by now. */
345   assert (wget_cookie_jar != NULL);
346
347   cookie_handle_set_cookie (wget_cookie_jar, u->host, u->port, u->path, hdr);
348   return 1;
349 }
350
351 \f
352 /* Persistent connections.  Currently, we cache the most recently used
353    connection as persistent, provided that the HTTP server agrees to
354    make it such.  The persistence data is stored in the variables
355    below.  Ideally, it would be in a structure, and it should be
356    possible to cache an arbitrary fixed number of these connections.
357
358    I think the code is quite easy to extend in that direction.  */
359
360 /* Whether a persistent connection is active. */
361 static int pc_active_p;
362 /* Host and port of currently active persistent connection. */
363 static struct address_list *pc_last_host_ip;
364 static unsigned short pc_last_port;
365
366 /* File descriptor of the currently active persistent connection. */
367 static int pc_last_fd;
368
369 #ifdef HAVE_SSL
370 /* Whether a ssl handshake has occoured on this connection */
371 static int pc_active_ssl;
372 /* SSL connection of the currently active persistent connection. */
373 static SSL *pc_last_ssl;
374 #endif /* HAVE_SSL */
375
376 /* Mark the persistent connection as invalid.  This is used by the
377    CLOSE_* macros after they forcefully close a registered persistent
378    connection.  This does not close the file descriptor -- it is left
379    to the caller to do that.  (Maybe it should, though.)  */
380
381 static void
382 invalidate_persistent (void)
383 {
384   pc_active_p = 0;
385 #ifdef HAVE_SSL
386   pc_active_ssl = 0;
387 #endif /* HAVE_SSL */
388   if (pc_last_host_ip != NULL)
389     {
390       address_list_release (pc_last_host_ip);
391       pc_last_host_ip = NULL;
392     }
393   DEBUGP (("Invalidating fd %d from further reuse.\n", pc_last_fd));
394 }
395
396 /* Register FD, which should be a TCP/IP connection to HOST:PORT, as
397    persistent.  This will enable someone to use the same connection
398    later.  In the context of HTTP, this must be called only AFTER the
399    response has been received and the server has promised that the
400    connection will remain alive.
401
402    If a previous connection was persistent, it is closed. */
403
404 #ifdef HAVE_SSL
405 static void
406 register_persistent (const char *host, unsigned short port, int fd, SSL *ssl)
407 {
408 #else
409 static void
410 register_persistent (const char *host, unsigned short port, int fd)
411 {
412 #endif
413   if (pc_active_p)
414     {
415       if (pc_last_fd == fd)
416         {
417           /* The connection FD is already registered.  Nothing to
418              do. */
419           return;
420         }
421       else
422         {
423           /* The old persistent connection is still active; let's
424              close it first.  This situation arises whenever a
425              persistent connection exists, but we then connect to a
426              different host, and try to register a persistent
427              connection to that one.  */
428 #ifdef HAVE_SSL
429           /* The ssl disconnect has to take place before the closing
430              of pc_last_fd.  */
431           if (pc_last_ssl)
432             shutdown_ssl(pc_last_ssl);
433 #endif
434           CLOSE (pc_last_fd);
435           invalidate_persistent ();
436         }
437     }
438
439   assert (pc_last_host_ip == NULL);
440
441   /* This lookup_host cannot fail, because it has the results in the
442      cache.  */
443   pc_last_host_ip = lookup_host (host, LH_SILENT);
444   assert (pc_last_host_ip != NULL);
445
446   pc_last_port = port;
447   pc_last_fd = fd;
448   pc_active_p = 1;
449 #ifdef HAVE_SSL
450   pc_last_ssl = ssl;
451   pc_active_ssl = ssl ? 1 : 0;
452 #endif
453   DEBUGP (("Registered fd %d for persistent reuse.\n", fd));
454 }
455
456 #ifdef HAVE_SSL
457 # define SHUTDOWN_SSL(ssl) do {         \
458   if (ssl)                              \
459     shutdown_ssl (ssl);                 \
460 } while (0)
461 #else
462 # define SHUTDOWN_SSL(ssl) 
463 #endif
464
465 /* Return non-zero if a persistent connection is available for
466    connecting to HOST:PORT.  */
467
468 #ifdef HAVE_SSL
469 static int
470 persistent_available_p (const char *host, unsigned short port, int ssl)
471 {
472 #else
473 static int
474 persistent_available_p (const char *host, unsigned short port)
475 {
476 #endif
477   int success;
478   struct address_list *this_host_ip;
479
480   /* First, check whether a persistent connection is active at all.  */
481   if (!pc_active_p)
482     return 0;
483   /* Second, check if the active connection pertains to the correct
484      (HOST, PORT) ordered pair.  */
485   if (port != pc_last_port)
486     return 0;
487
488 #ifdef HAVE_SSL
489   /* Second, a): check if current connection is (not) ssl, too.  This
490      test is unlikely to fail because HTTP and HTTPS typicaly use
491      different ports.  Yet it is possible, or so I [Christian
492      Fraenkel] have been told, to run HTTPS and HTTP simultaneus on
493      the same port.  */
494   if (ssl != pc_active_ssl)
495     return 0;
496 #endif /* HAVE_SSL */
497
498   this_host_ip = lookup_host (host, LH_SILENT);
499   if (!this_host_ip)
500     return 0;
501
502   /* To equate the two host names for the purposes of persistent
503      connections, they need to share all the IP addresses in the
504      list.  */
505   success = address_list_match_all (pc_last_host_ip, this_host_ip);
506   address_list_release (this_host_ip);
507   if (!success)
508     return 0;
509
510   /* Third: check whether the connection is still open.  This is
511      important because most server implement a liberal (short) timeout
512      on persistent connections.  Wget can of course always reconnect
513      if the connection doesn't work out, but it's nicer to know in
514      advance.  This test is a logical followup of the first test, but
515      is "expensive" and therefore placed at the end of the list.  */
516   if (!test_socket_open (pc_last_fd))
517     {
518       /* Oops, the socket is no longer open.  Now that we know that,
519          let's invalidate the persistent connection before returning
520          0.  */
521       CLOSE (pc_last_fd);
522 #ifdef HAVE_SSL
523       SHUTDOWN_SSL (pc_last_ssl);
524       pc_last_ssl = NULL;
525 #endif
526       invalidate_persistent ();
527       return 0;
528     }
529   return 1;
530 }
531
532 /* The idea behind these two CLOSE macros is to distinguish between
533    two cases: one when the job we've been doing is finished, and we
534    want to close the connection and leave, and two when something is
535    seriously wrong and we're closing the connection as part of
536    cleanup.
537
538    In case of keep_alive, CLOSE_FINISH should leave the connection
539    open, while CLOSE_INVALIDATE should still close it.
540
541    Note that the semantics of the flag `keep_alive' is "this
542    connection *will* be reused (the server has promised not to close
543    the connection once we're done)", while the semantics of
544    `pc_active_p && (fd) == pc_last_fd' is "we're *now* using an
545    active, registered connection".  */
546
547 #define CLOSE_FINISH(fd) do {                   \
548   if (!keep_alive)                              \
549     {                                           \
550       SHUTDOWN_SSL (ssl);                       \
551       CLOSE (fd);                               \
552       if (pc_active_p && (fd) == pc_last_fd)    \
553         invalidate_persistent ();               \
554     }                                           \
555 } while (0)
556
557 #define CLOSE_INVALIDATE(fd) do {               \
558   SHUTDOWN_SSL (ssl);                           \
559   CLOSE (fd);                                   \
560   if (pc_active_p && (fd) == pc_last_fd)        \
561     invalidate_persistent ();                   \
562 } while (0)
563 \f
564 struct http_stat
565 {
566   long len;                     /* received length */
567   long contlen;                 /* expected length */
568   long restval;                 /* the restart value */
569   int res;                      /* the result of last read */
570   char *newloc;                 /* new location (redirection) */
571   char *remote_time;            /* remote time-stamp string */
572   char *error;                  /* textual HTTP error */
573   int statcode;                 /* status code */
574   double dltime;                /* time of the download in msecs */
575   int no_truncate;              /* whether truncating the file is
576                                    forbidden. */
577   const char *referer;          /* value of the referer header. */
578   char **local_file;            /* local file. */
579 };
580
581 static void
582 free_hstat (struct http_stat *hs)
583 {
584   xfree_null (hs->newloc);
585   xfree_null (hs->remote_time);
586   xfree_null (hs->error);
587
588   /* Guard against being called twice. */
589   hs->newloc = NULL;
590   hs->remote_time = NULL;
591   hs->error = NULL;
592 }
593
594 static char *create_authorization_line PARAMS ((const char *, const char *,
595                                                 const char *, const char *,
596                                                 const char *));
597 static char *basic_authentication_encode PARAMS ((const char *, const char *,
598                                                   const char *));
599 static int known_authentication_scheme_p PARAMS ((const char *));
600
601 time_t http_atotm PARAMS ((const char *));
602
603 #define BEGINS_WITH(line, string_constant)                              \
604   (!strncasecmp (line, string_constant, sizeof (string_constant) - 1)   \
605    && (ISSPACE (line[sizeof (string_constant) - 1])                     \
606        || !line[sizeof (string_constant) - 1]))
607
608 /* Retrieve a document through HTTP protocol.  It recognizes status
609    code, and correctly handles redirections.  It closes the network
610    socket.  If it receives an error from the functions below it, it
611    will print it if there is enough information to do so (almost
612    always), returning the error to the caller (i.e. http_loop).
613
614    Various HTTP parameters are stored to hs.  Although it parses the
615    response code correctly, it is not used in a sane way.  The caller
616    can do that, though.
617
618    If PROXY is non-NULL, the connection will be made to the proxy
619    server, and u->url will be requested.  */
620 static uerr_t
621 gethttp (struct url *u, struct http_stat *hs, int *dt, struct url *proxy)
622 {
623   char *request, *type, *command, *full_path;
624   char *user, *passwd;
625   char *pragma_h, *referer, *useragent, *range, *wwwauth;
626   char *authenticate_h;
627   char *proxyauth;
628   char *all_headers;
629   char *port_maybe;
630   char *request_keep_alive;
631   int sock, hcount, all_length, statcode;
632   int write_error;
633   long contlen, contrange;
634   struct url *conn;
635   FILE *fp;
636   int auth_tried_already;
637   struct rbuf rbuf;
638 #ifdef HAVE_SSL
639   static SSL_CTX *ssl_ctx = NULL;
640   SSL *ssl = NULL;
641 #endif
642   char *cookies = NULL;
643
644   /* Whether this connection will be kept alive after the HTTP request
645      is done. */
646   int keep_alive;
647
648   /* Flags that detect the two ways of specifying HTTP keep-alive
649      response.  */
650   int http_keep_alive_1, http_keep_alive_2;
651
652   /* Whether keep-alive should be inhibited. */
653   int inhibit_keep_alive;
654
655   /* Whether we need to print the host header with braces around host,
656      e.g. "Host: [3ffe:8100:200:2::2]:1234" instead of the usual
657      "Host: symbolic-name:1234". */
658   int squares_around_host = 0;
659
660   /* Headers sent when using POST. */
661   char *post_content_type, *post_content_length;
662   long post_data_size = 0;
663
664 #ifdef HAVE_SSL
665   /* initialize ssl_ctx on first run */
666   if (!ssl_ctx)
667     {
668       uerr_t err = init_ssl (&ssl_ctx);
669       if (err != 0)
670         {
671           switch (err)
672             {
673             case SSLERRCTXCREATE:
674               /* this is fatal */
675               logprintf (LOG_NOTQUIET, _("Failed to set up an SSL context\n"));
676               ssl_printerrors ();
677               return err;
678             case SSLERRCERTFILE:
679               /* try without certfile */
680               logprintf (LOG_NOTQUIET,
681                          _("Failed to load certificates from %s\n"),
682                          opt.sslcertfile);
683               ssl_printerrors ();
684               logprintf (LOG_NOTQUIET,
685                          _("Trying without the specified certificate\n"));
686               break;
687             case SSLERRCERTKEY:
688               logprintf (LOG_NOTQUIET,
689                          _("Failed to get certificate key from %s\n"),
690                          opt.sslcertkey);
691               ssl_printerrors ();
692               logprintf (LOG_NOTQUIET,
693                          _("Trying without the specified certificate\n"));
694               break;
695             default:
696               break;
697             }
698         }
699     }
700 #endif /* HAVE_SSL */
701
702   if (!(*dt & HEAD_ONLY))
703     /* If we're doing a GET on the URL, as opposed to just a HEAD, we need to
704        know the local filename so we can save to it. */
705     assert (*hs->local_file != NULL);
706
707   authenticate_h = 0;
708   auth_tried_already = 0;
709
710   inhibit_keep_alive = !opt.http_keep_alive || proxy != NULL;
711
712  again:
713   /* We need to come back here when the initial attempt to retrieve
714      without authorization header fails.  (Expected to happen at least
715      for the Digest authorization scheme.)  */
716
717   keep_alive = 0;
718   http_keep_alive_1 = http_keep_alive_2 = 0;
719
720   post_content_type = NULL;
721   post_content_length = NULL;
722
723   /* Initialize certain elements of struct http_stat.  */
724   hs->len = 0L;
725   hs->contlen = -1;
726   hs->res = -1;
727   hs->newloc = NULL;
728   hs->remote_time = NULL;
729   hs->error = NULL;
730
731   /* If we're using a proxy, we will be connecting to the proxy
732      server. */
733   conn = proxy ? proxy : u;
734
735   /* First: establish the connection.  */
736   if (inhibit_keep_alive
737       ||
738 #ifndef HAVE_SSL
739       !persistent_available_p (conn->host, conn->port)
740 #else
741       !persistent_available_p (conn->host, conn->port,
742                                u->scheme == SCHEME_HTTPS)
743 #endif /* HAVE_SSL */
744       )
745     {
746       sock = connect_to_host (conn->host, conn->port);
747       if (sock == E_HOST)
748         return HOSTERR;
749       else if (sock < 0)
750         return CONNECT_ERROR (errno);
751
752 #ifdef HAVE_SSL
753      if (conn->scheme == SCHEME_HTTPS)
754        if (connect_ssl (&ssl, ssl_ctx,sock) != 0)
755          {
756            logputs (LOG_VERBOSE, "\n");
757            logprintf (LOG_NOTQUIET, _("Unable to establish SSL connection.\n"));
758            ssl_printerrors ();
759            CLOSE (sock);
760            return CONSSLERR;
761          }
762 #endif /* HAVE_SSL */
763     }
764   else
765     {
766       logprintf (LOG_VERBOSE, _("Reusing connection to %s:%hu.\n"),
767                  conn->host, conn->port);
768       /* #### pc_last_fd should be accessed through an accessor
769          function.  */
770       sock = pc_last_fd;
771 #ifdef HAVE_SSL
772       ssl = pc_last_ssl;
773 #endif /* HAVE_SSL */
774       DEBUGP (("Reusing fd %d.\n", sock));
775     }
776
777   if (*dt & HEAD_ONLY)
778     command = "HEAD";
779   else if (opt.post_file_name || opt.post_data)
780     command = "POST";
781   else
782     command = "GET";
783
784   referer = NULL;
785   if (hs->referer)
786     {
787       referer = (char *)alloca (9 + strlen (hs->referer) + 3);
788       sprintf (referer, "Referer: %s\r\n", hs->referer);
789     }
790
791   if (*dt & SEND_NOCACHE)
792     pragma_h = "Pragma: no-cache\r\n";
793   else
794     pragma_h = "";
795
796   if (hs->restval)
797     {
798       range = (char *)alloca (13 + numdigit (hs->restval) + 4);
799       /* Gag me!  Some servers (e.g. WebSitePro) have been known to
800          respond to the following `Range' format by generating a
801          multipart/x-byte-ranges MIME document!  This MIME type was
802          present in an old draft of the byteranges specification.
803          HTTP/1.1 specifies a multipart/byte-ranges MIME type, but
804          only if multiple non-overlapping ranges are requested --
805          which Wget never does.  */
806       sprintf (range, "Range: bytes=%ld-\r\n", hs->restval);
807     }
808   else
809     range = NULL;
810   if (opt.useragent)
811     STRDUP_ALLOCA (useragent, opt.useragent);
812   else
813     {
814       useragent = (char *)alloca (10 + strlen (version_string));
815       sprintf (useragent, "Wget/%s", version_string);
816     }
817   /* Construct the authentication, if userid is present.  */
818   user = u->user;
819   passwd = u->passwd;
820   search_netrc (u->host, (const char **)&user, (const char **)&passwd, 0);
821   user = user ? user : opt.http_user;
822   passwd = passwd ? passwd : opt.http_passwd;
823
824   wwwauth = NULL;
825   if (user && passwd)
826     {
827       if (!authenticate_h)
828         {
829           /* We have the username and the password, but haven't tried
830              any authorization yet.  Let's see if the "Basic" method
831              works.  If not, we'll come back here and construct a
832              proper authorization method with the right challenges.
833
834              If we didn't employ this kind of logic, every URL that
835              requires authorization would have to be processed twice,
836              which is very suboptimal and generates a bunch of false
837              "unauthorized" errors in the server log.
838
839              #### But this logic also has a serious problem when used
840              with stronger authentications: we *first* transmit the
841              username and the password in clear text, and *then*
842              attempt a stronger authentication scheme.  That cannot be
843              right!  We are only fortunate that almost everyone still
844              uses the `Basic' scheme anyway.
845
846              There should be an option to prevent this from happening,
847              for those who use strong authentication schemes and value
848              their passwords.  */
849           wwwauth = basic_authentication_encode (user, passwd, "Authorization");
850         }
851       else
852         {
853           /* Use the full path, i.e. one that includes the leading
854              slash and the query string, but is independent of proxy
855              setting.  */
856           char *pth = url_full_path (u);
857           wwwauth = create_authorization_line (authenticate_h, user, passwd,
858                                                command, pth);
859           xfree (pth);
860         }
861     }
862
863   proxyauth = NULL;
864   if (proxy)
865     {
866       char *proxy_user, *proxy_passwd;
867       /* For normal username and password, URL components override
868          command-line/wgetrc parameters.  With proxy authentication,
869          it's the reverse, because proxy URLs are normally the
870          "permanent" ones, so command-line args should take
871          precedence.  */
872       if (opt.proxy_user && opt.proxy_passwd)
873         {
874           proxy_user = opt.proxy_user;
875           proxy_passwd = opt.proxy_passwd;
876         }
877       else
878         {
879           proxy_user = proxy->user;
880           proxy_passwd = proxy->passwd;
881         }
882       /* #### This does not appear right.  Can't the proxy request,
883          say, `Digest' authentication?  */
884       if (proxy_user && proxy_passwd)
885         proxyauth = basic_authentication_encode (proxy_user, proxy_passwd,
886                                                  "Proxy-Authorization");
887     }
888
889   /* String of the form :PORT.  Used only for non-standard ports. */
890   port_maybe = NULL;
891   if (u->port != scheme_default_port (u->scheme))
892     {
893       port_maybe = (char *)alloca (numdigit (u->port) + 2);
894       sprintf (port_maybe, ":%d", u->port);
895     }
896
897   if (!inhibit_keep_alive)
898     request_keep_alive = "Connection: Keep-Alive\r\n";
899   else
900     request_keep_alive = NULL;
901
902   if (opt.cookies)
903     cookies = cookie_header (wget_cookie_jar, u->host, u->port, u->path,
904 #ifdef HAVE_SSL
905                              u->scheme == SCHEME_HTTPS
906 #else
907                              0
908 #endif
909                              );
910
911   if (opt.post_data || opt.post_file_name)
912     {
913       post_content_type = "Content-Type: application/x-www-form-urlencoded\r\n";
914       if (opt.post_data)
915         post_data_size = strlen (opt.post_data);
916       else
917         {
918           post_data_size = file_size (opt.post_file_name);
919           if (post_data_size == -1)
920             {
921               logprintf (LOG_NOTQUIET, "POST data file missing: %s\n",
922                          opt.post_file_name);
923               post_data_size = 0;
924             }
925         }
926       post_content_length = xmalloc (16 + numdigit (post_data_size) + 2 + 1);
927       sprintf (post_content_length,
928                "Content-Length: %ld\r\n", post_data_size);
929     }
930
931   if (proxy)
932     full_path = xstrdup (u->url);
933   else
934     /* Use the full path, i.e. one that includes the leading slash and
935        the query string.  E.g. if u->path is "foo/bar" and u->query is
936        "param=value", full_path will be "/foo/bar?param=value".  */
937     full_path = url_full_path (u);
938
939   if (strchr (u->host, ':'))
940     squares_around_host = 1;
941
942   /* Allocate the memory for the request.  */
943   request = (char *)alloca (strlen (command)
944                             + strlen (full_path)
945                             + strlen (useragent)
946                             + strlen (u->host)
947                             + (port_maybe ? strlen (port_maybe) : 0)
948                             + strlen (HTTP_ACCEPT)
949                             + (request_keep_alive
950                                ? strlen (request_keep_alive) : 0)
951                             + (referer ? strlen (referer) : 0)
952                             + (cookies ? strlen (cookies) : 0)
953                             + (wwwauth ? strlen (wwwauth) : 0)
954                             + (proxyauth ? strlen (proxyauth) : 0)
955                             + (range ? strlen (range) : 0)
956                             + strlen (pragma_h)
957                             + (post_content_type
958                                ? strlen (post_content_type) : 0)
959                             + (post_content_length
960                                ? strlen (post_content_length) : 0)
961                             + (opt.user_header ? strlen (opt.user_header) : 0)
962                             + 64);
963   /* Construct the request.  */
964   sprintf (request, "\
965 %s %s HTTP/1.0\r\n\
966 User-Agent: %s\r\n\
967 Host: %s%s%s%s\r\n\
968 Accept: %s\r\n\
969 %s%s%s%s%s%s%s%s%s%s\r\n",
970            command, full_path,
971            useragent,
972            squares_around_host ? "[" : "", u->host, squares_around_host ? "]" : "",
973            port_maybe ? port_maybe : "",
974            HTTP_ACCEPT,
975            request_keep_alive ? request_keep_alive : "",
976            referer ? referer : "",
977            cookies ? cookies : "", 
978            wwwauth ? wwwauth : "", 
979            proxyauth ? proxyauth : "", 
980            range ? range : "",
981            pragma_h,
982            post_content_type ? post_content_type : "",
983            post_content_length ? post_content_length : "",
984            opt.user_header ? opt.user_header : "");
985   DEBUGP (("---request begin---\n%s", request));
986
987   /* Free the temporary memory.  */
988   xfree_null (wwwauth);
989   xfree_null (proxyauth);
990   xfree_null (cookies);
991   xfree (full_path);
992
993   /* Send the request to server.  */
994 #ifdef HAVE_SSL
995   if (conn->scheme == SCHEME_HTTPS)
996     write_error = ssl_iwrite (ssl, request, strlen (request));
997   else
998 #endif
999     write_error = iwrite (sock, request, strlen (request));
1000
1001   if (write_error >= 0)
1002     {
1003       if (opt.post_data)
1004         {
1005           DEBUGP (("[POST data: %s]\n", opt.post_data));
1006 #ifdef HAVE_SSL
1007           if (conn->scheme == SCHEME_HTTPS)
1008             write_error = ssl_iwrite (ssl, opt.post_data, post_data_size);
1009           else
1010 #endif
1011             write_error = iwrite (sock, opt.post_data, post_data_size);
1012         }
1013       else if (opt.post_file_name && post_data_size != 0)
1014         {
1015 #ifdef HAVE_SSL
1016           if (conn->scheme == SCHEME_HTTPS)
1017             write_error = post_file (-1, ssl, opt.post_file_name,
1018                                      post_data_size);
1019           else
1020 #endif
1021             write_error = post_file (sock, NULL, opt.post_file_name,
1022                                      post_data_size);
1023         }
1024     }
1025   DEBUGP (("---request end---\n"));
1026
1027   if (write_error < 0)
1028     {
1029       logprintf (LOG_VERBOSE, _("Failed writing HTTP request: %s.\n"),
1030                  strerror (errno));
1031       CLOSE_INVALIDATE (sock);
1032       return WRITEFAILED;
1033     }
1034   logprintf (LOG_VERBOSE, _("%s request sent, awaiting response... "),
1035              proxy ? "Proxy" : "HTTP");
1036   contlen = contrange = -1;
1037   type = NULL;
1038   statcode = -1;
1039   *dt &= ~RETROKF;
1040
1041   /* Before reading anything, initialize the rbuf.  */
1042   rbuf_initialize (&rbuf, sock);
1043 #ifdef HAVE_SSL
1044   if (conn->scheme == SCHEME_HTTPS)
1045     rbuf.ssl = ssl;
1046   else
1047     rbuf.ssl = NULL;
1048 #endif /* HAVE_SSL */
1049   all_headers = NULL;
1050   all_length = 0;
1051   /* Header-fetching loop.  */
1052   hcount = 0;
1053   while (1)
1054     {
1055       char *hdr;
1056       int status;
1057
1058       ++hcount;
1059       /* Get the header.  */
1060       status = header_get (&rbuf, &hdr,
1061                            /* Disallow continuations for status line.  */
1062                            (hcount == 1 ? HG_NO_CONTINUATIONS : HG_NONE));
1063
1064       /* Check for errors.  */
1065       if (status == HG_EOF && *hdr)
1066         {
1067           /* This used to be an unconditional error, but that was
1068              somewhat controversial, because of a large number of
1069              broken CGI's that happily "forget" to send the second EOL
1070              before closing the connection of a HEAD request.
1071
1072              So, the deal is to check whether the header is empty
1073              (*hdr is zero if it is); if yes, it means that the
1074              previous header was fully retrieved, and that -- most
1075              probably -- the request is complete.  "...be liberal in
1076              what you accept."  Oh boy.  */
1077           logputs (LOG_VERBOSE, "\n");
1078           logputs (LOG_NOTQUIET, _("End of file while parsing headers.\n"));
1079           xfree (hdr);
1080           xfree_null (type);
1081           xfree_null (all_headers);
1082           CLOSE_INVALIDATE (sock);
1083           return HEOF;
1084         }
1085       else if (status == HG_ERROR)
1086         {
1087           logputs (LOG_VERBOSE, "\n");
1088           logprintf (LOG_NOTQUIET, _("Read error (%s) in headers.\n"),
1089                      strerror (errno));
1090           xfree (hdr);
1091           xfree_null (type);
1092           xfree_null (all_headers);
1093           CLOSE_INVALIDATE (sock);
1094           return HERR;
1095         }
1096
1097       /* If the headers are to be saved to a file later, save them to
1098          memory now.  */
1099       if (opt.save_headers)
1100         {
1101           int lh = strlen (hdr);
1102           all_headers = (char *)xrealloc (all_headers, all_length + lh + 2);
1103           memcpy (all_headers + all_length, hdr, lh);
1104           all_length += lh;
1105           all_headers[all_length++] = '\n';
1106           all_headers[all_length] = '\0';
1107         }
1108
1109       /* Check for status line.  */
1110       if (hcount == 1)
1111         {
1112           const char *error;
1113           /* Parse the first line of server response.  */
1114           statcode = parse_http_status_line (hdr, &error);
1115           hs->statcode = statcode;
1116           /* Store the descriptive response.  */
1117           if (statcode == -1) /* malformed response */
1118             {
1119               /* A common reason for "malformed response" error is the
1120                  case when no data was actually received.  Handle this
1121                  special case.  */
1122               if (!*hdr)
1123                 hs->error = xstrdup (_("No data received"));
1124               else
1125                 hs->error = xstrdup (_("Malformed status line"));
1126               xfree (hdr);
1127               break;
1128             }
1129           else if (!*error)
1130             hs->error = xstrdup (_("(no description)"));
1131           else
1132             hs->error = xstrdup (error);
1133
1134           if ((statcode != -1)
1135 #ifdef ENABLE_DEBUG
1136               && !opt.debug
1137 #endif
1138               )
1139             {
1140              if (opt.server_response)
1141                logprintf (LOG_VERBOSE, "\n%2d %s", hcount, hdr);
1142              else
1143                logprintf (LOG_VERBOSE, "%2d %s", statcode, error);
1144             }
1145
1146           goto done_header;
1147         }
1148
1149       /* Exit on empty header.  */
1150       if (!*hdr)
1151         {
1152           xfree (hdr);
1153           break;
1154         }
1155
1156       /* Print the header if requested.  */
1157       if (opt.server_response && hcount != 1)
1158         logprintf (LOG_VERBOSE, "\n%2d %s", hcount, hdr);
1159
1160       /* Try getting content-length.  */
1161       if (contlen == -1 && !opt.ignore_length)
1162         if (header_process (hdr, "Content-Length", header_extract_number,
1163                             &contlen))
1164           goto done_header;
1165       /* Try getting content-type.  */
1166       if (!type)
1167         if (header_process (hdr, "Content-Type", http_process_type, &type))
1168           goto done_header;
1169       /* Try getting location.  */
1170       if (!hs->newloc)
1171         if (header_process (hdr, "Location", header_strdup, &hs->newloc))
1172           goto done_header;
1173       /* Try getting last-modified.  */
1174       if (!hs->remote_time)
1175         if (header_process (hdr, "Last-Modified", header_strdup,
1176                             &hs->remote_time))
1177           goto done_header;
1178       /* Try getting cookies. */
1179       if (opt.cookies)
1180         if (header_process (hdr, "Set-Cookie", http_process_set_cookie, u))
1181           goto done_header;
1182       /* Try getting www-authentication.  */
1183       if (!authenticate_h)
1184         if (header_process (hdr, "WWW-Authenticate", header_strdup,
1185                             &authenticate_h))
1186           goto done_header;
1187       /* Check for accept-ranges header.  If it contains the word
1188          `none', disable the ranges.  */
1189       if (*dt & ACCEPTRANGES)
1190         {
1191           int nonep;
1192           if (header_process (hdr, "Accept-Ranges", http_process_none, &nonep))
1193             {
1194               if (nonep)
1195                 *dt &= ~ACCEPTRANGES;
1196               goto done_header;
1197             }
1198         }
1199       /* Try getting content-range.  */
1200       if (contrange == -1)
1201         {
1202           struct http_process_range_closure closure;
1203           if (header_process (hdr, "Content-Range", http_process_range, &closure))
1204             {
1205               contrange = closure.first_byte_pos;
1206               goto done_header;
1207             }
1208         }
1209       /* Check for keep-alive related responses. */
1210       if (!inhibit_keep_alive)
1211         {
1212           /* Check for the `Keep-Alive' header. */
1213           if (!http_keep_alive_1)
1214             {
1215               if (header_process (hdr, "Keep-Alive", header_exists,
1216                                   &http_keep_alive_1))
1217                 goto done_header;
1218             }
1219           /* Check for `Connection: Keep-Alive'. */
1220           if (!http_keep_alive_2)
1221             {
1222               if (header_process (hdr, "Connection", http_process_connection,
1223                                   &http_keep_alive_2))
1224                 goto done_header;
1225             }
1226         }
1227     done_header:
1228       xfree (hdr);
1229     }
1230
1231   logputs (LOG_VERBOSE, "\n");
1232
1233   if (contlen != -1
1234       && (http_keep_alive_1 || http_keep_alive_2))
1235     {
1236       assert (inhibit_keep_alive == 0);
1237       keep_alive = 1;
1238     }
1239   if (keep_alive)
1240     /* The server has promised that it will not close the connection
1241        when we're done.  This means that we can register it.  */
1242 #ifndef HAVE_SSL
1243     register_persistent (conn->host, conn->port, sock);
1244 #else
1245     register_persistent (conn->host, conn->port, sock, ssl);
1246 #endif /* HAVE_SSL */
1247
1248   if ((statcode == HTTP_STATUS_UNAUTHORIZED)
1249       && authenticate_h)
1250     {
1251       /* Authorization is required.  */
1252       xfree_null (type);
1253       type = NULL;
1254       free_hstat (hs);
1255       CLOSE_INVALIDATE (sock);  /* would be CLOSE_FINISH, but there
1256                                    might be more bytes in the body. */
1257       if (auth_tried_already)
1258         {
1259           /* If we have tried it already, then there is not point
1260              retrying it.  */
1261         failed:
1262           logputs (LOG_NOTQUIET, _("Authorization failed.\n"));
1263           xfree (authenticate_h);
1264           return AUTHFAILED;
1265         }
1266       else if (!known_authentication_scheme_p (authenticate_h))
1267         {
1268           xfree (authenticate_h);
1269           logputs (LOG_NOTQUIET, _("Unknown authentication scheme.\n"));
1270           return AUTHFAILED;
1271         }
1272       else if (BEGINS_WITH (authenticate_h, "Basic"))
1273         {
1274           /* The authentication scheme is basic, the one we try by
1275              default, and it failed.  There's no sense in trying
1276              again.  */
1277           goto failed;
1278         }
1279       else
1280         {
1281           auth_tried_already = 1;
1282           goto again;
1283         }
1284     }
1285   /* We do not need this anymore.  */
1286   if (authenticate_h)
1287     {
1288       xfree (authenticate_h);
1289       authenticate_h = NULL;
1290     }
1291
1292   /* 20x responses are counted among successful by default.  */
1293   if (H_20X (statcode))
1294     *dt |= RETROKF;
1295
1296   /* Return if redirected.  */
1297   if (H_REDIRECTED (statcode) || statcode == HTTP_STATUS_MULTIPLE_CHOICES)
1298     {
1299       /* RFC2068 says that in case of the 300 (multiple choices)
1300          response, the server can output a preferred URL through
1301          `Location' header; otherwise, the request should be treated
1302          like GET.  So, if the location is set, it will be a
1303          redirection; otherwise, just proceed normally.  */
1304       if (statcode == HTTP_STATUS_MULTIPLE_CHOICES && !hs->newloc)
1305         *dt |= RETROKF;
1306       else
1307         {
1308           logprintf (LOG_VERBOSE,
1309                      _("Location: %s%s\n"),
1310                      hs->newloc ? hs->newloc : _("unspecified"),
1311                      hs->newloc ? _(" [following]") : "");
1312           CLOSE_INVALIDATE (sock);      /* would be CLOSE_FINISH, but there
1313                                            might be more bytes in the body. */
1314           xfree_null (type);
1315           xfree_null (all_headers);
1316           return NEWLOCATION;
1317         }
1318     }
1319
1320   /* If content-type is not given, assume text/html.  This is because
1321      of the multitude of broken CGI's that "forget" to generate the
1322      content-type.  */
1323   if (!type ||
1324         0 == strncasecmp (type, TEXTHTML_S, strlen (TEXTHTML_S)) ||
1325         0 == strncasecmp (type, TEXTXHTML_S, strlen (TEXTXHTML_S)))
1326     *dt |= TEXTHTML;
1327   else
1328     *dt &= ~TEXTHTML;
1329
1330   if (opt.html_extension && (*dt & TEXTHTML))
1331     /* -E / --html-extension / html_extension = on was specified, and this is a
1332        text/html file.  If some case-insensitive variation on ".htm[l]" isn't
1333        already the file's suffix, tack on ".html". */
1334     {
1335       char*  last_period_in_local_filename = strrchr(*hs->local_file, '.');
1336
1337       if (last_period_in_local_filename == NULL
1338           || !(0 == strcasecmp (last_period_in_local_filename, ".htm")
1339                || 0 == strcasecmp (last_period_in_local_filename, ".html")))
1340         {
1341           size_t  local_filename_len = strlen(*hs->local_file);
1342           
1343           *hs->local_file = xrealloc(*hs->local_file,
1344                                      local_filename_len + sizeof(".html"));
1345           strcpy(*hs->local_file + local_filename_len, ".html");
1346
1347           *dt |= ADDED_HTML_EXTENSION;
1348         }
1349     }
1350
1351   if (contrange == -1)
1352     {
1353       /* We did not get a content-range header.  This means that the
1354          server did not honor our `Range' request.  Normally, this
1355          means we should reset hs->restval and continue normally.  */
1356
1357       /* However, if `-c' is used, we need to be a bit more careful:
1358
1359          1. If `-c' is specified and the file already existed when
1360          Wget was started, it would be a bad idea for us to start
1361          downloading it from scratch, effectively truncating it.  I
1362          believe this cannot happen unless `-c' was specified.
1363
1364          2. If `-c' is used on a file that is already fully
1365          downloaded, we're requesting bytes after the end of file,
1366          which can result in server not honoring `Range'.  If this is
1367          the case, `Content-Length' will be equal to the length of the
1368          file.  */
1369       if (opt.always_rest)
1370         {
1371           /* Check for condition #2. */
1372           if (hs->restval > 0               /* restart was requested. */
1373               && contlen != -1              /* we got content-length. */
1374               && hs->restval >= contlen     /* file fully downloaded
1375                                                or has shrunk.  */
1376               )
1377             {
1378               logputs (LOG_VERBOSE, _("\
1379 \n    The file is already fully retrieved; nothing to do.\n\n"));
1380               /* In case the caller inspects. */
1381               hs->len = contlen;
1382               hs->res = 0;
1383               /* Mark as successfully retrieved. */
1384               *dt |= RETROKF;
1385               xfree_null (type);
1386               xfree_null (all_headers);
1387               CLOSE_INVALIDATE (sock);  /* would be CLOSE_FINISH, but there
1388                                            might be more bytes in the body. */
1389               return RETRUNNEEDED;
1390             }
1391
1392           /* Check for condition #1. */
1393           if (hs->no_truncate)
1394             {
1395               logprintf (LOG_NOTQUIET,
1396                          _("\
1397 \n\
1398 Continued download failed on this file, which conflicts with `-c'.\n\
1399 Refusing to truncate existing file `%s'.\n\n"), *hs->local_file);
1400               xfree_null (type);
1401               xfree_null (all_headers);
1402               CLOSE_INVALIDATE (sock);
1403               return CONTNOTSUPPORTED;
1404             }
1405
1406           /* Fallthrough */
1407         }
1408
1409       hs->restval = 0;
1410     }
1411   else if (contrange != hs->restval ||
1412            (H_PARTIAL (statcode) && contrange == -1))
1413     {
1414       /* This means the whole request was somehow misunderstood by the
1415          server.  Bail out.  */
1416       xfree_null (type);
1417       xfree_null (all_headers);
1418       CLOSE_INVALIDATE (sock);
1419       return RANGEERR;
1420     }
1421
1422   if (hs->restval)
1423     {
1424       if (contlen != -1)
1425         contlen += contrange;
1426       else
1427         contrange = -1;        /* If conent-length was not sent,
1428                                   content-range will be ignored.  */
1429     }
1430   hs->contlen = contlen;
1431
1432   if (opt.verbose)
1433     {
1434       if ((*dt & RETROKF) && !opt.server_response)
1435         {
1436           /* No need to print this output if the body won't be
1437              downloaded at all, or if the original server response is
1438              printed.  */
1439           logputs (LOG_VERBOSE, _("Length: "));
1440           if (contlen != -1)
1441             {
1442               logputs (LOG_VERBOSE, legible (contlen));
1443               if (contrange != -1)
1444                 logprintf (LOG_VERBOSE, _(" (%s to go)"),
1445                            legible (contlen - contrange));
1446             }
1447           else
1448             logputs (LOG_VERBOSE,
1449                      opt.ignore_length ? _("ignored") : _("unspecified"));
1450           if (type)
1451             logprintf (LOG_VERBOSE, " [%s]\n", type);
1452           else
1453             logputs (LOG_VERBOSE, "\n");
1454         }
1455     }
1456   xfree_null (type);
1457   type = NULL;                  /* We don't need it any more.  */
1458
1459   /* Return if we have no intention of further downloading.  */
1460   if (!(*dt & RETROKF) || (*dt & HEAD_ONLY))
1461     {
1462       /* In case the caller cares to look...  */
1463       hs->len = 0L;
1464       hs->res = 0;
1465       xfree_null (type);
1466       xfree_null (all_headers);
1467       CLOSE_INVALIDATE (sock);  /* would be CLOSE_FINISH, but there
1468                                    might be more bytes in the body. */
1469       return RETRFINISHED;
1470     }
1471
1472   /* Open the local file.  */
1473   if (!opt.dfp)
1474     {
1475       mkalldirs (*hs->local_file);
1476       if (opt.backups)
1477         rotate_backups (*hs->local_file);
1478       fp = fopen (*hs->local_file, hs->restval ? "ab" : "wb");
1479       if (!fp)
1480         {
1481           logprintf (LOG_NOTQUIET, "%s: %s\n", *hs->local_file, strerror (errno));
1482           CLOSE_INVALIDATE (sock); /* would be CLOSE_FINISH, but there
1483                                       might be more bytes in the body. */
1484           xfree_null (all_headers);
1485           return FOPENERR;
1486         }
1487     }
1488   else                          /* opt.dfp */
1489     {
1490       extern int global_download_count;
1491       fp = opt.dfp;
1492       /* To ensure that repeated "from scratch" downloads work for -O
1493          files, we rewind the file pointer, unless restval is
1494          non-zero.  (This works only when -O is used on regular files,
1495          but it's still a valuable feature.)
1496
1497          However, this loses when more than one URL is specified on
1498          the command line the second rewinds eradicates the contents
1499          of the first download.  Thus we disable the above trick for
1500          all the downloads except the very first one.
1501
1502          #### A possible solution to this would be to remember the
1503          file position in the output document and to seek to that
1504          position, instead of rewinding.
1505
1506          We don't truncate stdout, since that breaks
1507          "wget -O - [...] >> foo".
1508       */
1509       if (!hs->restval && global_download_count == 0 && opt.dfp != stdout)
1510         {
1511           /* This will silently fail for streams that don't correspond
1512              to regular files, but that's OK.  */
1513           rewind (fp);
1514           /* ftruncate is needed because opt.dfp is opened in append
1515              mode if opt.always_rest is set.  */
1516           ftruncate (fileno (fp), 0);
1517           clearerr (fp);
1518         }
1519     }
1520
1521   /* #### This confuses the code that checks for file size.  There
1522      should be some overhead information.  */
1523   if (opt.save_headers)
1524     fwrite (all_headers, 1, all_length, fp);
1525
1526   /* Get the contents of the document.  */
1527   hs->res = get_contents (sock, fp, &hs->len, hs->restval,
1528                           (contlen != -1 ? contlen : 0),
1529                           &rbuf, keep_alive, &hs->dltime);
1530
1531   if (hs->res >= 0)
1532     CLOSE_FINISH (sock);
1533   else
1534     CLOSE_INVALIDATE (sock);
1535
1536   {
1537     /* Close or flush the file.  We have to be careful to check for
1538        error here.  Checking the result of fwrite() is not enough --
1539        errors could go unnoticed!  */
1540     int flush_res;
1541     if (!opt.dfp)
1542       flush_res = fclose (fp);
1543     else
1544       flush_res = fflush (fp);
1545     if (flush_res == EOF)
1546       hs->res = -2;
1547   }
1548   xfree_null (all_headers);
1549   if (hs->res == -2)
1550     return FWRITEERR;
1551   return RETRFINISHED;
1552 }
1553
1554 /* The genuine HTTP loop!  This is the part where the retrieval is
1555    retried, and retried, and retried, and...  */
1556 uerr_t
1557 http_loop (struct url *u, char **newloc, char **local_file, const char *referer,
1558            int *dt, struct url *proxy)
1559 {
1560   int count;
1561   int use_ts, got_head = 0;     /* time-stamping info */
1562   char *filename_plus_orig_suffix;
1563   char *local_filename = NULL;
1564   char *tms, *locf, *tmrate;
1565   uerr_t err;
1566   time_t tml = -1, tmr = -1;    /* local and remote time-stamps */
1567   long local_size = 0;          /* the size of the local file */
1568   size_t filename_len;
1569   struct http_stat hstat;       /* HTTP status */
1570   struct stat st;
1571   char *dummy = NULL;
1572
1573   /* This used to be done in main(), but it's a better idea to do it
1574      here so that we don't go through the hoops if we're just using
1575      FTP or whatever. */
1576   if (opt.cookies)
1577     {
1578       if (!wget_cookie_jar)
1579         wget_cookie_jar = cookie_jar_new ();
1580       if (opt.cookies_input && !cookies_loaded_p)
1581         {
1582           cookie_jar_load (wget_cookie_jar, opt.cookies_input);
1583           cookies_loaded_p = 1;
1584         }
1585     }
1586
1587   *newloc = NULL;
1588
1589   /* Warn on (likely bogus) wildcard usage in HTTP.  Don't use
1590      has_wildcards_p because it would also warn on `?', and we know that
1591      shows up in CGI paths a *lot*.  */
1592   if (strchr (u->url, '*'))
1593     logputs (LOG_VERBOSE, _("Warning: wildcards not supported in HTTP.\n"));
1594
1595   /* Determine the local filename.  */
1596   if (local_file && *local_file)
1597     hstat.local_file = local_file;
1598   else if (local_file)
1599     {
1600       *local_file = url_file_name (u);
1601       hstat.local_file = local_file;
1602     }
1603   else
1604     {
1605       dummy = url_file_name (u);
1606       hstat.local_file = &dummy;
1607     }
1608
1609   if (!opt.output_document)
1610     locf = *hstat.local_file;
1611   else
1612     locf = opt.output_document;
1613
1614   hstat.referer = referer;
1615
1616   filename_len = strlen (*hstat.local_file);
1617   filename_plus_orig_suffix = alloca (filename_len + sizeof (".orig"));
1618
1619   if (opt.noclobber && file_exists_p (*hstat.local_file))
1620     {
1621       /* If opt.noclobber is turned on and file already exists, do not
1622          retrieve the file */
1623       logprintf (LOG_VERBOSE, _("\
1624 File `%s' already there, will not retrieve.\n"), *hstat.local_file);
1625       /* If the file is there, we suppose it's retrieved OK.  */
1626       *dt |= RETROKF;
1627
1628       /* #### Bogusness alert.  */
1629       /* If its suffix is "html" or "htm" or similar, assume text/html.  */
1630       if (has_html_suffix_p (*hstat.local_file))
1631         *dt |= TEXTHTML;
1632
1633       xfree_null (dummy);
1634       return RETROK;
1635     }
1636
1637   use_ts = 0;
1638   if (opt.timestamping)
1639     {
1640       int local_dot_orig_file_exists = 0;
1641
1642       if (opt.backup_converted)
1643         /* If -K is specified, we'll act on the assumption that it was specified
1644            last time these files were downloaded as well, and instead of just
1645            comparing local file X against server file X, we'll compare local
1646            file X.orig (if extant, else X) against server file X.  If -K
1647            _wasn't_ specified last time, or the server contains files called
1648            *.orig, -N will be back to not operating correctly with -k. */
1649         {
1650           /* Would a single s[n]printf() call be faster?  --dan
1651
1652              Definitely not.  sprintf() is horribly slow.  It's a
1653              different question whether the difference between the two
1654              affects a program.  Usually I'd say "no", but at one
1655              point I profiled Wget, and found that a measurable and
1656              non-negligible amount of time was lost calling sprintf()
1657              in url.c.  Replacing sprintf with inline calls to
1658              strcpy() and long_to_string() made a difference.
1659              --hniksic */
1660           memcpy (filename_plus_orig_suffix, *hstat.local_file, filename_len);
1661           memcpy (filename_plus_orig_suffix + filename_len,
1662                   ".orig", sizeof (".orig"));
1663
1664           /* Try to stat() the .orig file. */
1665           if (stat (filename_plus_orig_suffix, &st) == 0)
1666             {
1667               local_dot_orig_file_exists = 1;
1668               local_filename = filename_plus_orig_suffix;
1669             }
1670         }      
1671
1672       if (!local_dot_orig_file_exists)
1673         /* Couldn't stat() <file>.orig, so try to stat() <file>. */
1674         if (stat (*hstat.local_file, &st) == 0)
1675           local_filename = *hstat.local_file;
1676
1677       if (local_filename != NULL)
1678         /* There was a local file, so we'll check later to see if the version
1679            the server has is the same version we already have, allowing us to
1680            skip a download. */
1681         {
1682           use_ts = 1;
1683           tml = st.st_mtime;
1684 #ifdef WINDOWS
1685           /* Modification time granularity is 2 seconds for Windows, so
1686              increase local time by 1 second for later comparison. */
1687           tml++;
1688 #endif
1689           local_size = st.st_size;
1690           got_head = 0;
1691         }
1692     }
1693   /* Reset the counter.  */
1694   count = 0;
1695   *dt = 0 | ACCEPTRANGES;
1696   /* THE loop */
1697   do
1698     {
1699       /* Increment the pass counter.  */
1700       ++count;
1701       sleep_between_retrievals (count);
1702       /* Get the current time string.  */
1703       tms = time_str (NULL);
1704       /* Print fetch message, if opt.verbose.  */
1705       if (opt.verbose)
1706         {
1707           char *hurl = url_string (u, 1);
1708           char tmp[15];
1709           strcpy (tmp, "        ");
1710           if (count > 1)
1711             sprintf (tmp, _("(try:%2d)"), count);
1712           logprintf (LOG_VERBOSE, "--%s--  %s\n  %s => `%s'\n",
1713                      tms, hurl, tmp, locf);
1714 #ifdef WINDOWS
1715           ws_changetitle (hurl, 1);
1716 #endif
1717           xfree (hurl);
1718         }
1719
1720       /* Default document type is empty.  However, if spider mode is
1721          on or time-stamping is employed, HEAD_ONLY commands is
1722          encoded within *dt.  */
1723       if (opt.spider || (use_ts && !got_head))
1724         *dt |= HEAD_ONLY;
1725       else
1726         *dt &= ~HEAD_ONLY;
1727       /* Assume no restarting.  */
1728       hstat.restval = 0L;
1729       /* Decide whether or not to restart.  */
1730       if (((count > 1 && (*dt & ACCEPTRANGES)) || opt.always_rest)
1731           /* #### this calls access() and then stat(); could be optimized. */
1732           && file_exists_p (locf))
1733         if (stat (locf, &st) == 0 && S_ISREG (st.st_mode))
1734           hstat.restval = st.st_size;
1735
1736       /* In `-c' is used and the file is existing and non-empty,
1737          refuse to truncate it if the server doesn't support continued
1738          downloads.  */
1739       hstat.no_truncate = 0;
1740       if (opt.always_rest && hstat.restval)
1741         hstat.no_truncate = 1;
1742
1743       /* Decide whether to send the no-cache directive.  We send it in
1744          two cases:
1745            a) we're using a proxy, and we're past our first retrieval.
1746               Some proxies are notorious for caching incomplete data, so
1747               we require a fresh get.
1748            b) caching is explicitly inhibited. */
1749       if ((proxy && count > 1)  /* a */
1750           || !opt.allow_cache   /* b */
1751           )
1752         *dt |= SEND_NOCACHE;
1753       else
1754         *dt &= ~SEND_NOCACHE;
1755
1756       /* Try fetching the document, or at least its head.  */
1757       err = gethttp (u, &hstat, dt, proxy);
1758
1759       /* It's unfortunate that wget determines the local filename before finding
1760          out the Content-Type of the file.  Barring a major restructuring of the
1761          code, we need to re-set locf here, since gethttp() may have xrealloc()d
1762          *hstat.local_file to tack on ".html". */
1763       if (!opt.output_document)
1764         locf = *hstat.local_file;
1765       else
1766         locf = opt.output_document;
1767
1768       /* Time?  */
1769       tms = time_str (NULL);
1770       /* Get the new location (with or without the redirection).  */
1771       if (hstat.newloc)
1772         *newloc = xstrdup (hstat.newloc);
1773       switch (err)
1774         {
1775         case HERR: case HEOF: case CONSOCKERR: case CONCLOSED:
1776         case CONERROR: case READERR: case WRITEFAILED:
1777         case RANGEERR:
1778           /* Non-fatal errors continue executing the loop, which will
1779              bring them to "while" statement at the end, to judge
1780              whether the number of tries was exceeded.  */
1781           free_hstat (&hstat);
1782           printwhat (count, opt.ntry);
1783           continue;
1784           break;
1785         case HOSTERR: case CONIMPOSSIBLE: case PROXERR: case AUTHFAILED: 
1786         case SSLERRCTXCREATE: case CONTNOTSUPPORTED:
1787           /* Fatal errors just return from the function.  */
1788           free_hstat (&hstat);
1789           xfree_null (dummy);
1790           return err;
1791           break;
1792         case FWRITEERR: case FOPENERR:
1793           /* Another fatal error.  */
1794           logputs (LOG_VERBOSE, "\n");
1795           logprintf (LOG_NOTQUIET, _("Cannot write to `%s' (%s).\n"),
1796                      *hstat.local_file, strerror (errno));
1797           free_hstat (&hstat);
1798           xfree_null (dummy);
1799           return err;
1800           break;
1801         case CONSSLERR:
1802           /* Another fatal error.  */
1803           logputs (LOG_VERBOSE, "\n");
1804           logprintf (LOG_NOTQUIET, _("Unable to establish SSL connection.\n"));
1805           free_hstat (&hstat);
1806           xfree_null (dummy);
1807           return err;
1808           break;
1809         case NEWLOCATION:
1810           /* Return the new location to the caller.  */
1811           if (!hstat.newloc)
1812             {
1813               logprintf (LOG_NOTQUIET,
1814                          _("ERROR: Redirection (%d) without location.\n"),
1815                          hstat.statcode);
1816               free_hstat (&hstat);
1817               xfree_null (dummy);
1818               return WRONGCODE;
1819             }
1820           free_hstat (&hstat);
1821           xfree_null (dummy);
1822           return NEWLOCATION;
1823           break;
1824         case RETRUNNEEDED:
1825           /* The file was already fully retrieved. */
1826           free_hstat (&hstat);
1827           xfree_null (dummy);
1828           return RETROK;
1829           break;
1830         case RETRFINISHED:
1831           /* Deal with you later.  */
1832           break;
1833         default:
1834           /* All possibilities should have been exhausted.  */
1835           abort ();
1836         }
1837       if (!(*dt & RETROKF))
1838         {
1839           if (!opt.verbose)
1840             {
1841               /* #### Ugly ugly ugly! */
1842               char *hurl = url_string (u, 1);
1843               logprintf (LOG_NONVERBOSE, "%s:\n", hurl);
1844               xfree (hurl);
1845             }
1846           logprintf (LOG_NOTQUIET, _("%s ERROR %d: %s.\n"),
1847                      tms, hstat.statcode, hstat.error);
1848           logputs (LOG_VERBOSE, "\n");
1849           free_hstat (&hstat);
1850           xfree_null (dummy);
1851           return WRONGCODE;
1852         }
1853
1854       /* Did we get the time-stamp?  */
1855       if (!got_head)
1856         {
1857           if (opt.timestamping && !hstat.remote_time)
1858             {
1859               logputs (LOG_NOTQUIET, _("\
1860 Last-modified header missing -- time-stamps turned off.\n"));
1861             }
1862           else if (hstat.remote_time)
1863             {
1864               /* Convert the date-string into struct tm.  */
1865               tmr = http_atotm (hstat.remote_time);
1866               if (tmr == (time_t) (-1))
1867                 logputs (LOG_VERBOSE, _("\
1868 Last-modified header invalid -- time-stamp ignored.\n"));
1869             }
1870         }
1871
1872       /* The time-stamping section.  */
1873       if (use_ts)
1874         {
1875           got_head = 1;
1876           *dt &= ~HEAD_ONLY;
1877           use_ts = 0;           /* no more time-stamping */
1878           count = 0;            /* the retrieve count for HEAD is
1879                                    reset */
1880           if (hstat.remote_time && tmr != (time_t) (-1))
1881             {
1882               /* Now time-stamping can be used validly.  Time-stamping
1883                  means that if the sizes of the local and remote file
1884                  match, and local file is newer than the remote file,
1885                  it will not be retrieved.  Otherwise, the normal
1886                  download procedure is resumed.  */
1887               if (tml >= tmr &&
1888                   (hstat.contlen == -1 || local_size == hstat.contlen))
1889                 {
1890                   logprintf (LOG_VERBOSE, _("\
1891 Server file no newer than local file `%s' -- not retrieving.\n\n"),
1892                              local_filename);
1893                   free_hstat (&hstat);
1894                   xfree_null (dummy);
1895                   return RETROK;
1896                 }
1897               else if (tml >= tmr)
1898                 logprintf (LOG_VERBOSE, _("\
1899 The sizes do not match (local %ld) -- retrieving.\n"), local_size);
1900               else
1901                 logputs (LOG_VERBOSE,
1902                          _("Remote file is newer, retrieving.\n"));
1903             }
1904           free_hstat (&hstat);
1905           continue;
1906         }
1907       if ((tmr != (time_t) (-1))
1908           && !opt.spider
1909           && ((hstat.len == hstat.contlen) ||
1910               ((hstat.res == 0) &&
1911                ((hstat.contlen == -1) ||
1912                 (hstat.len >= hstat.contlen && !opt.kill_longer)))))
1913         {
1914           /* #### This code repeats in http.c and ftp.c.  Move it to a
1915              function!  */
1916           const char *fl = NULL;
1917           if (opt.output_document)
1918             {
1919               if (opt.od_known_regular)
1920                 fl = opt.output_document;
1921             }
1922           else
1923             fl = *hstat.local_file;
1924           if (fl)
1925             touch (fl, tmr);
1926         }
1927       /* End of time-stamping section.  */
1928
1929       if (opt.spider)
1930         {
1931           logprintf (LOG_NOTQUIET, "%d %s\n\n", hstat.statcode, hstat.error);
1932           xfree_null (dummy);
1933           return RETROK;
1934         }
1935
1936       tmrate = retr_rate (hstat.len - hstat.restval, hstat.dltime, 0);
1937
1938       if (hstat.len == hstat.contlen)
1939         {
1940           if (*dt & RETROKF)
1941             {
1942               logprintf (LOG_VERBOSE,
1943                          _("%s (%s) - `%s' saved [%ld/%ld]\n\n"),
1944                          tms, tmrate, locf, hstat.len, hstat.contlen);
1945               logprintf (LOG_NONVERBOSE,
1946                          "%s URL:%s [%ld/%ld] -> \"%s\" [%d]\n",
1947                          tms, u->url, hstat.len, hstat.contlen, locf, count);
1948             }
1949           ++opt.numurls;
1950           total_downloaded_bytes += hstat.len;
1951
1952           /* Remember that we downloaded the file for later ".orig" code. */
1953           if (*dt & ADDED_HTML_EXTENSION)
1954             downloaded_file(FILE_DOWNLOADED_AND_HTML_EXTENSION_ADDED, locf);
1955           else
1956             downloaded_file(FILE_DOWNLOADED_NORMALLY, locf);
1957
1958           free_hstat (&hstat);
1959           xfree_null (dummy);
1960           return RETROK;
1961         }
1962       else if (hstat.res == 0) /* No read error */
1963         {
1964           if (hstat.contlen == -1)  /* We don't know how much we were supposed
1965                                        to get, so assume we succeeded. */ 
1966             {
1967               if (*dt & RETROKF)
1968                 {
1969                   logprintf (LOG_VERBOSE,
1970                              _("%s (%s) - `%s' saved [%ld]\n\n"),
1971                              tms, tmrate, locf, hstat.len);
1972                   logprintf (LOG_NONVERBOSE,
1973                              "%s URL:%s [%ld] -> \"%s\" [%d]\n",
1974                              tms, u->url, hstat.len, locf, count);
1975                 }
1976               ++opt.numurls;
1977               total_downloaded_bytes += hstat.len;
1978
1979               /* Remember that we downloaded the file for later ".orig" code. */
1980               if (*dt & ADDED_HTML_EXTENSION)
1981                 downloaded_file(FILE_DOWNLOADED_AND_HTML_EXTENSION_ADDED, locf);
1982               else
1983                 downloaded_file(FILE_DOWNLOADED_NORMALLY, locf);
1984               
1985               free_hstat (&hstat);
1986               xfree_null (dummy);
1987               return RETROK;
1988             }
1989           else if (hstat.len < hstat.contlen) /* meaning we lost the
1990                                                  connection too soon */
1991             {
1992               logprintf (LOG_VERBOSE,
1993                          _("%s (%s) - Connection closed at byte %ld. "),
1994                          tms, tmrate, hstat.len);
1995               printwhat (count, opt.ntry);
1996               free_hstat (&hstat);
1997               continue;
1998             }
1999           else if (!opt.kill_longer) /* meaning we got more than expected */
2000             {
2001               logprintf (LOG_VERBOSE,
2002                          _("%s (%s) - `%s' saved [%ld/%ld])\n\n"),
2003                          tms, tmrate, locf, hstat.len, hstat.contlen);
2004               logprintf (LOG_NONVERBOSE,
2005                          "%s URL:%s [%ld/%ld] -> \"%s\" [%d]\n",
2006                          tms, u->url, hstat.len, hstat.contlen, locf, count);
2007               ++opt.numurls;
2008               total_downloaded_bytes += hstat.len;
2009
2010               /* Remember that we downloaded the file for later ".orig" code. */
2011               if (*dt & ADDED_HTML_EXTENSION)
2012                 downloaded_file(FILE_DOWNLOADED_AND_HTML_EXTENSION_ADDED, locf);
2013               else
2014                 downloaded_file(FILE_DOWNLOADED_NORMALLY, locf);
2015               
2016               free_hstat (&hstat);
2017               xfree_null (dummy);
2018               return RETROK;
2019             }
2020           else                  /* the same, but not accepted */
2021             {
2022               logprintf (LOG_VERBOSE,
2023                          _("%s (%s) - Connection closed at byte %ld/%ld. "),
2024                          tms, tmrate, hstat.len, hstat.contlen);
2025               printwhat (count, opt.ntry);
2026               free_hstat (&hstat);
2027               continue;
2028             }
2029         }
2030       else                      /* now hstat.res can only be -1 */
2031         {
2032           if (hstat.contlen == -1)
2033             {
2034               logprintf (LOG_VERBOSE,
2035                          _("%s (%s) - Read error at byte %ld (%s)."),
2036                          tms, tmrate, hstat.len, strerror (errno));
2037               printwhat (count, opt.ntry);
2038               free_hstat (&hstat);
2039               continue;
2040             }
2041           else                  /* hstat.res == -1 and contlen is given */
2042             {
2043               logprintf (LOG_VERBOSE,
2044                          _("%s (%s) - Read error at byte %ld/%ld (%s). "),
2045                          tms, tmrate, hstat.len, hstat.contlen,
2046                          strerror (errno));
2047               printwhat (count, opt.ntry);
2048               free_hstat (&hstat);
2049               continue;
2050             }
2051         }
2052       /* not reached */
2053       break;
2054     }
2055   while (!opt.ntry || (count < opt.ntry));
2056   return TRYLIMEXC;
2057 }
2058 \f
2059 /* Converts struct tm to time_t, assuming the data in tm is UTC rather
2060    than local timezone.
2061
2062    mktime is similar but assumes struct tm, also known as the
2063    "broken-down" form of time, is in local time zone.  mktime_from_utc
2064    uses mktime to make the conversion understanding that an offset
2065    will be introduced by the local time assumption.
2066
2067    mktime_from_utc then measures the introduced offset by applying
2068    gmtime to the initial result and applying mktime to the resulting
2069    "broken-down" form.  The difference between the two mktime results
2070    is the measured offset which is then subtracted from the initial
2071    mktime result to yield a calendar time which is the value returned.
2072
2073    tm_isdst in struct tm is set to 0 to force mktime to introduce a
2074    consistent offset (the non DST offset) since tm and tm+o might be
2075    on opposite sides of a DST change.
2076
2077    Some implementations of mktime return -1 for the nonexistent
2078    localtime hour at the beginning of DST.  In this event, use
2079    mktime(tm - 1hr) + 3600.
2080
2081    Schematically
2082      mktime(tm)   --> t+o
2083      gmtime(t+o)  --> tm+o
2084      mktime(tm+o) --> t+2o
2085      t+o - (t+2o - t+o) = t
2086
2087    Note that glibc contains a function of the same purpose named
2088    `timegm' (reverse of gmtime).  But obviously, it is not universally
2089    available, and unfortunately it is not straightforwardly
2090    extractable for use here.  Perhaps configure should detect timegm
2091    and use it where available.
2092
2093    Contributed by Roger Beeman <beeman@cisco.com>, with the help of
2094    Mark Baushke <mdb@cisco.com> and the rest of the Gurus at CISCO.
2095    Further improved by Roger with assistance from Edward J. Sabol
2096    based on input by Jamie Zawinski.  */
2097
2098 static time_t
2099 mktime_from_utc (struct tm *t)
2100 {
2101   time_t tl, tb;
2102   struct tm *tg;
2103
2104   tl = mktime (t);
2105   if (tl == -1)
2106     {
2107       t->tm_hour--;
2108       tl = mktime (t);
2109       if (tl == -1)
2110         return -1; /* can't deal with output from strptime */
2111       tl += 3600;
2112     }
2113   tg = gmtime (&tl);
2114   tg->tm_isdst = 0;
2115   tb = mktime (tg);
2116   if (tb == -1)
2117     {
2118       tg->tm_hour--;
2119       tb = mktime (tg);
2120       if (tb == -1)
2121         return -1; /* can't deal with output from gmtime */
2122       tb += 3600;
2123     }
2124   return (tl - (tb - tl));
2125 }
2126
2127 /* Check whether the result of strptime() indicates success.
2128    strptime() returns the pointer to how far it got to in the string.
2129    The processing has been successful if the string is at `GMT' or
2130    `+X', or at the end of the string.
2131
2132    In extended regexp parlance, the function returns 1 if P matches
2133    "^ *(GMT|[+-][0-9]|$)", 0 otherwise.  P being NULL (which strptime
2134    can return) is considered a failure and 0 is returned.  */
2135 static int
2136 check_end (const char *p)
2137 {
2138   if (!p)
2139     return 0;
2140   while (ISSPACE (*p))
2141     ++p;
2142   if (!*p
2143       || (p[0] == 'G' && p[1] == 'M' && p[2] == 'T')
2144       || ((p[0] == '+' || p[0] == '-') && ISDIGIT (p[1])))
2145     return 1;
2146   else
2147     return 0;
2148 }
2149
2150 /* Convert the textual specification of time in TIME_STRING to the
2151    number of seconds since the Epoch.
2152
2153    TIME_STRING can be in any of the three formats RFC2068 allows the
2154    HTTP servers to emit -- RFC1123-date, RFC850-date or asctime-date.
2155    Timezones are ignored, and should be GMT.
2156
2157    Return the computed time_t representation, or -1 if the conversion
2158    fails.
2159
2160    This function uses strptime with various string formats for parsing
2161    TIME_STRING.  This results in a parser that is not as lenient in
2162    interpreting TIME_STRING as I would like it to be.  Being based on
2163    strptime, it always allows shortened months, one-digit days, etc.,
2164    but due to the multitude of formats in which time can be
2165    represented, an ideal HTTP time parser would be even more
2166    forgiving.  It should completely ignore things like week days and
2167    concentrate only on the various forms of representing years,
2168    months, days, hours, minutes, and seconds.  For example, it would
2169    be nice if it accepted ISO 8601 out of the box.
2170
2171    I've investigated free and PD code for this purpose, but none was
2172    usable.  getdate was big and unwieldy, and had potential copyright
2173    issues, or so I was informed.  Dr. Marcus Hennecke's atotm(),
2174    distributed with phttpd, is excellent, but we cannot use it because
2175    it is not assigned to the FSF.  So I stuck it with strptime.  */
2176
2177 time_t
2178 http_atotm (const char *time_string)
2179 {
2180   /* NOTE: Solaris strptime man page claims that %n and %t match white
2181      space, but that's not universally available.  Instead, we simply
2182      use ` ' to mean "skip all WS", which works under all strptime
2183      implementations I've tested.  */
2184
2185   static const char *time_formats[] = {
2186     "%a, %d %b %Y %T",          /* RFC1123: Thu, 29 Jan 1998 22:12:57 */
2187     "%A, %d-%b-%y %T",          /* RFC850:  Thursday, 29-Jan-98 22:12:57 */
2188     "%a, %d-%b-%Y %T",          /* pseudo-RFC850:  Thu, 29-Jan-1998 22:12:57
2189                                    (google.com uses this for their cookies.) */
2190     "%a %b %d %T %Y"            /* asctime: Thu Jan 29 22:12:57 1998 */
2191   };
2192
2193   int i;
2194   struct tm t;
2195
2196   /* According to Roger Beeman, we need to initialize tm_isdst, since
2197      strptime won't do it.  */
2198   t.tm_isdst = 0;
2199
2200   /* Note that under foreign locales Solaris strptime() fails to
2201      recognize English dates, which renders this function useless.  We
2202      solve this by being careful not to affect LC_TIME when
2203      initializing locale.
2204
2205      Another solution would be to temporarily set locale to C, invoke
2206      strptime(), and restore it back.  This is slow and dirty,
2207      however, and locale support other than LC_MESSAGES can mess other
2208      things, so I rather chose to stick with just setting LC_MESSAGES.
2209
2210      GNU strptime does not have this problem because it recognizes
2211      both international and local dates.  */
2212
2213   for (i = 0; i < countof (time_formats); i++)
2214     if (check_end (strptime (time_string, time_formats[i], &t)))
2215       return mktime_from_utc (&t);
2216
2217   /* All formats have failed.  */
2218   return -1;
2219 }
2220 \f
2221 /* Authorization support: We support two authorization schemes:
2222
2223    * `Basic' scheme, consisting of base64-ing USER:PASSWORD string;
2224
2225    * `Digest' scheme, added by Junio Hamano <junio@twinsun.com>,
2226    consisting of answering to the server's challenge with the proper
2227    MD5 digests.  */
2228
2229 /* How many bytes it will take to store LEN bytes in base64.  */
2230 #define BASE64_LENGTH(len) (4 * (((len) + 2) / 3))
2231
2232 /* Encode the string S of length LENGTH to base64 format and place it
2233    to STORE.  STORE will be 0-terminated, and must point to a writable
2234    buffer of at least 1+BASE64_LENGTH(length) bytes.  */
2235 static void
2236 base64_encode (const char *s, char *store, int length)
2237 {
2238   /* Conversion table.  */
2239   static char tbl[64] = {
2240     'A','B','C','D','E','F','G','H',
2241     'I','J','K','L','M','N','O','P',
2242     'Q','R','S','T','U','V','W','X',
2243     'Y','Z','a','b','c','d','e','f',
2244     'g','h','i','j','k','l','m','n',
2245     'o','p','q','r','s','t','u','v',
2246     'w','x','y','z','0','1','2','3',
2247     '4','5','6','7','8','9','+','/'
2248   };
2249   int i;
2250   unsigned char *p = (unsigned char *)store;
2251
2252   /* Transform the 3x8 bits to 4x6 bits, as required by base64.  */
2253   for (i = 0; i < length; i += 3)
2254     {
2255       *p++ = tbl[s[0] >> 2];
2256       *p++ = tbl[((s[0] & 3) << 4) + (s[1] >> 4)];
2257       *p++ = tbl[((s[1] & 0xf) << 2) + (s[2] >> 6)];
2258       *p++ = tbl[s[2] & 0x3f];
2259       s += 3;
2260     }
2261   /* Pad the result if necessary...  */
2262   if (i == length + 1)
2263     *(p - 1) = '=';
2264   else if (i == length + 2)
2265     *(p - 1) = *(p - 2) = '=';
2266   /* ...and zero-terminate it.  */
2267   *p = '\0';
2268 }
2269
2270 /* Create the authentication header contents for the `Basic' scheme.
2271    This is done by encoding the string `USER:PASS' in base64 and
2272    prepending `HEADER: Basic ' to it.  */
2273 static char *
2274 basic_authentication_encode (const char *user, const char *passwd,
2275                              const char *header)
2276 {
2277   char *t1, *t2, *res;
2278   int len1 = strlen (user) + 1 + strlen (passwd);
2279   int len2 = BASE64_LENGTH (len1);
2280
2281   t1 = (char *)alloca (len1 + 1);
2282   sprintf (t1, "%s:%s", user, passwd);
2283   t2 = (char *)alloca (1 + len2);
2284   base64_encode (t1, t2, len1);
2285   res = (char *)xmalloc (len2 + 11 + strlen (header));
2286   sprintf (res, "%s: Basic %s\r\n", header, t2);
2287
2288   return res;
2289 }
2290
2291 #ifdef USE_DIGEST
2292 /* Parse HTTP `WWW-Authenticate:' header.  AU points to the beginning
2293    of a field in such a header.  If the field is the one specified by
2294    ATTR_NAME ("realm", "opaque", and "nonce" are used by the current
2295    digest authorization code), extract its value in the (char*)
2296    variable pointed by RET.  Returns negative on a malformed header,
2297    or number of bytes that have been parsed by this call.  */
2298 static int
2299 extract_header_attr (const char *au, const char *attr_name, char **ret)
2300 {
2301   const char *cp, *ep;
2302
2303   ep = cp = au;
2304
2305   if (strncmp (cp, attr_name, strlen (attr_name)) == 0)
2306     {
2307       cp += strlen (attr_name);
2308       if (!*cp)
2309         return -1;
2310       cp += skip_lws (cp);
2311       if (*cp != '=')
2312         return -1;
2313       if (!*++cp)
2314         return -1;
2315       cp += skip_lws (cp);
2316       if (*cp != '\"')
2317         return -1;
2318       if (!*++cp)
2319         return -1;
2320       for (ep = cp; *ep && *ep != '\"'; ep++)
2321         ;
2322       if (!*ep)
2323         return -1;
2324       xfree_null (*ret);
2325       *ret = strdupdelim (cp, ep);
2326       return ep - au + 1;
2327     }
2328   else
2329     return 0;
2330 }
2331
2332 /* Dump the hexadecimal representation of HASH to BUF.  HASH should be
2333    an array of 16 bytes containing the hash keys, and BUF should be a
2334    buffer of 33 writable characters (32 for hex digits plus one for
2335    zero termination).  */
2336 static void
2337 dump_hash (unsigned char *buf, const unsigned char *hash)
2338 {
2339   int i;
2340
2341   for (i = 0; i < MD5_HASHLEN; i++, hash++)
2342     {
2343       *buf++ = XNUM_TO_digit (*hash >> 4);
2344       *buf++ = XNUM_TO_digit (*hash & 0xf);
2345     }
2346   *buf = '\0';
2347 }
2348
2349 /* Take the line apart to find the challenge, and compose a digest
2350    authorization header.  See RFC2069 section 2.1.2.  */
2351 static char *
2352 digest_authentication_encode (const char *au, const char *user,
2353                               const char *passwd, const char *method,
2354                               const char *path)
2355 {
2356   static char *realm, *opaque, *nonce;
2357   static struct {
2358     const char *name;
2359     char **variable;
2360   } options[] = {
2361     { "realm", &realm },
2362     { "opaque", &opaque },
2363     { "nonce", &nonce }
2364   };
2365   char *res;
2366
2367   realm = opaque = nonce = NULL;
2368
2369   au += 6;                      /* skip over `Digest' */
2370   while (*au)
2371     {
2372       int i;
2373
2374       au += skip_lws (au);
2375       for (i = 0; i < countof (options); i++)
2376         {
2377           int skip = extract_header_attr (au, options[i].name,
2378                                           options[i].variable);
2379           if (skip < 0)
2380             {
2381               xfree_null (realm);
2382               xfree_null (opaque);
2383               xfree_null (nonce);
2384               return NULL;
2385             }
2386           else if (skip)
2387             {
2388               au += skip;
2389               break;
2390             }
2391         }
2392       if (i == countof (options))
2393         {
2394           while (*au && *au != '=')
2395             au++;
2396           if (*au && *++au)
2397             {
2398               au += skip_lws (au);
2399               if (*au == '\"')
2400                 {
2401                   au++;
2402                   while (*au && *au != '\"')
2403                     au++;
2404                   if (*au)
2405                     au++;
2406                 }
2407             }
2408         }
2409       while (*au && *au != ',')
2410         au++;
2411       if (*au)
2412         au++;
2413     }
2414   if (!realm || !nonce || !user || !passwd || !path || !method)
2415     {
2416       xfree_null (realm);
2417       xfree_null (opaque);
2418       xfree_null (nonce);
2419       return NULL;
2420     }
2421
2422   /* Calculate the digest value.  */
2423   {
2424     ALLOCA_MD5_CONTEXT (ctx);
2425     unsigned char hash[MD5_HASHLEN];
2426     unsigned char a1buf[MD5_HASHLEN * 2 + 1], a2buf[MD5_HASHLEN * 2 + 1];
2427     unsigned char response_digest[MD5_HASHLEN * 2 + 1];
2428
2429     /* A1BUF = H(user ":" realm ":" password) */
2430     gen_md5_init (ctx);
2431     gen_md5_update ((unsigned char *)user, strlen (user), ctx);
2432     gen_md5_update ((unsigned char *)":", 1, ctx);
2433     gen_md5_update ((unsigned char *)realm, strlen (realm), ctx);
2434     gen_md5_update ((unsigned char *)":", 1, ctx);
2435     gen_md5_update ((unsigned char *)passwd, strlen (passwd), ctx);
2436     gen_md5_finish (ctx, hash);
2437     dump_hash (a1buf, hash);
2438
2439     /* A2BUF = H(method ":" path) */
2440     gen_md5_init (ctx);
2441     gen_md5_update ((unsigned char *)method, strlen (method), ctx);
2442     gen_md5_update ((unsigned char *)":", 1, ctx);
2443     gen_md5_update ((unsigned char *)path, strlen (path), ctx);
2444     gen_md5_finish (ctx, hash);
2445     dump_hash (a2buf, hash);
2446
2447     /* RESPONSE_DIGEST = H(A1BUF ":" nonce ":" A2BUF) */
2448     gen_md5_init (ctx);
2449     gen_md5_update (a1buf, MD5_HASHLEN * 2, ctx);
2450     gen_md5_update ((unsigned char *)":", 1, ctx);
2451     gen_md5_update ((unsigned char *)nonce, strlen (nonce), ctx);
2452     gen_md5_update ((unsigned char *)":", 1, ctx);
2453     gen_md5_update (a2buf, MD5_HASHLEN * 2, ctx);
2454     gen_md5_finish (ctx, hash);
2455     dump_hash (response_digest, hash);
2456
2457     res = (char*) xmalloc (strlen (user)
2458                            + strlen (user)
2459                            + strlen (realm)
2460                            + strlen (nonce)
2461                            + strlen (path)
2462                            + 2 * MD5_HASHLEN /*strlen (response_digest)*/
2463                            + (opaque ? strlen (opaque) : 0)
2464                            + 128);
2465     sprintf (res, "Authorization: Digest \
2466 username=\"%s\", realm=\"%s\", nonce=\"%s\", uri=\"%s\", response=\"%s\"",
2467              user, realm, nonce, path, response_digest);
2468     if (opaque)
2469       {
2470         char *p = res + strlen (res);
2471         strcat (p, ", opaque=\"");
2472         strcat (p, opaque);
2473         strcat (p, "\"");
2474       }
2475     strcat (res, "\r\n");
2476   }
2477   return res;
2478 }
2479 #endif /* USE_DIGEST */
2480
2481
2482 #define BEGINS_WITH(line, string_constant)                              \
2483   (!strncasecmp (line, string_constant, sizeof (string_constant) - 1)   \
2484    && (ISSPACE (line[sizeof (string_constant) - 1])                     \
2485        || !line[sizeof (string_constant) - 1]))
2486
2487 static int
2488 known_authentication_scheme_p (const char *au)
2489 {
2490   return BEGINS_WITH (au, "Basic")
2491     || BEGINS_WITH (au, "Digest")
2492     || BEGINS_WITH (au, "NTLM");
2493 }
2494
2495 #undef BEGINS_WITH
2496
2497 /* Create the HTTP authorization request header.  When the
2498    `WWW-Authenticate' response header is seen, according to the
2499    authorization scheme specified in that header (`Basic' and `Digest'
2500    are supported by the current implementation), produce an
2501    appropriate HTTP authorization request header.  */
2502 static char *
2503 create_authorization_line (const char *au, const char *user,
2504                            const char *passwd, const char *method,
2505                            const char *path)
2506 {
2507   char *wwwauth = NULL;
2508
2509   if (!strncasecmp (au, "Basic", 5))
2510     wwwauth = basic_authentication_encode (user, passwd, "Authorization");
2511   if (!strncasecmp (au, "NTLM", 4))
2512     wwwauth = basic_authentication_encode (user, passwd, "Authorization");
2513 #ifdef USE_DIGEST
2514   else if (!strncasecmp (au, "Digest", 6))
2515     wwwauth = digest_authentication_encode (au, user, passwd, method, path);
2516 #endif /* USE_DIGEST */
2517   return wwwauth;
2518 }
2519 \f
2520 void
2521 http_cleanup (void)
2522 {
2523   if (pc_last_host_ip)
2524     address_list_release (pc_last_host_ip);
2525 }