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