]> sjero.net Git - wget/blob - src/http.c
[svn] Use alloca to allocate filename_plus_orig_suffix.
[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 `%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           && file_exists_p (locf))
1518         if (stat (locf, &st) == 0 && S_ISREG (st.st_mode))
1519           hstat.restval = st.st_size;
1520
1521       /* In `-c' is used and the file is existing and non-empty,
1522          refuse to truncate it if the server doesn't support continued
1523          downloads.  */
1524       if (opt.always_rest && hstat.restval)
1525         hstat.no_truncate = file_exists_p (locf);
1526
1527       /* Decide whether to send the no-cache directive.  We send it in
1528          two cases:
1529            a) we're using a proxy, and we're past our first retrieval.
1530               Some proxies are notorious for caching incomplete data, so
1531               we require a fresh get.
1532            b) caching is explicitly inhibited. */
1533       if ((u->proxy && count > 1) /* a */
1534           || !opt.allow_cache     /* b */
1535           )
1536         *dt |= SEND_NOCACHE;
1537       else
1538         *dt &= ~SEND_NOCACHE;
1539
1540       /* Try fetching the document, or at least its head.  :-) */
1541       err = gethttp (u, &hstat, dt);
1542
1543       /* It's unfortunate that wget determines the local filename before finding
1544          out the Content-Type of the file.  Barring a major restructuring of the
1545          code, we need to re-set locf here, since gethttp() may have xrealloc()d
1546          u->local to tack on ".html". */
1547       if (!opt.output_document)
1548         locf = u->local;
1549       else
1550         locf = opt.output_document;
1551
1552       /* Time?  */
1553       tms = time_str (NULL);
1554       /* Get the new location (with or without the redirection).  */
1555       if (hstat.newloc)
1556         *newloc = xstrdup (hstat.newloc);
1557       switch (err)
1558         {
1559         case HERR: case HEOF: case CONSOCKERR: case CONCLOSED:
1560         case CONERROR: case READERR: case WRITEFAILED:
1561         case RANGEERR:
1562           /* Non-fatal errors continue executing the loop, which will
1563              bring them to "while" statement at the end, to judge
1564              whether the number of tries was exceeded.  */
1565           FREEHSTAT (hstat);
1566           printwhat (count, opt.ntry);
1567           continue;
1568           break;
1569         case HOSTERR: case CONREFUSED: case PROXERR: case AUTHFAILED: 
1570         case SSLERRCTXCREATE: case CONTNOTSUPPORTED:
1571           /* Fatal errors just return from the function.  */
1572           FREEHSTAT (hstat);
1573           return err;
1574           break;
1575         case FWRITEERR: case FOPENERR:
1576           /* Another fatal error.  */
1577           logputs (LOG_VERBOSE, "\n");
1578           logprintf (LOG_NOTQUIET, _("Cannot write to `%s' (%s).\n"),
1579                      u->local, strerror (errno));
1580           FREEHSTAT (hstat);
1581           return err;
1582           break;
1583         case CONSSLERR:
1584           /* Another fatal error.  */
1585           logputs (LOG_VERBOSE, "\n");
1586           logprintf (LOG_NOTQUIET, _("Unable to establish SSL connection.\n"));
1587           FREEHSTAT (hstat);
1588           return err;
1589           break;
1590         case NEWLOCATION:
1591           /* Return the new location to the caller.  */
1592           if (!hstat.newloc)
1593             {
1594               logprintf (LOG_NOTQUIET,
1595                          _("ERROR: Redirection (%d) without location.\n"),
1596                          hstat.statcode);
1597               return WRONGCODE;
1598             }
1599           FREEHSTAT (hstat);
1600           return NEWLOCATION;
1601           break;
1602         case RETRUNNEEDED:
1603           /* The file was already fully retrieved. */
1604           FREEHSTAT (hstat);
1605           return RETROK;
1606           break;
1607         case RETRFINISHED:
1608           /* Deal with you later.  */
1609           break;
1610         default:
1611           /* All possibilities should have been exhausted.  */
1612           abort ();
1613         }
1614       if (!(*dt & RETROKF))
1615         {
1616           if (!opt.verbose)
1617             {
1618               /* #### Ugly ugly ugly! */
1619               char *hurl = str_url (u->proxy ? u->proxy : u, 1);
1620               logprintf (LOG_NONVERBOSE, "%s:\n", hurl);
1621               xfree (hurl);
1622             }
1623           logprintf (LOG_NOTQUIET, _("%s ERROR %d: %s.\n"),
1624                      tms, hstat.statcode, hstat.error);
1625           logputs (LOG_VERBOSE, "\n");
1626           FREEHSTAT (hstat);
1627           return WRONGCODE;
1628         }
1629
1630       /* Did we get the time-stamp?  */
1631       if (!got_head)
1632         {
1633           if (opt.timestamping && !hstat.remote_time)
1634             {
1635               logputs (LOG_NOTQUIET, _("\
1636 Last-modified header missing -- time-stamps turned off.\n"));
1637             }
1638           else if (hstat.remote_time)
1639             {
1640               /* Convert the date-string into struct tm.  */
1641               tmr = http_atotm (hstat.remote_time);
1642               if (tmr == (time_t) (-1))
1643                 logputs (LOG_VERBOSE, _("\
1644 Last-modified header invalid -- time-stamp ignored.\n"));
1645             }
1646         }
1647
1648       /* The time-stamping section.  */
1649       if (use_ts)
1650         {
1651           got_head = 1;
1652           *dt &= ~HEAD_ONLY;
1653           use_ts = 0;           /* no more time-stamping */
1654           count = 0;            /* the retrieve count for HEAD is
1655                                    reset */
1656           if (hstat.remote_time && tmr != (time_t) (-1))
1657             {
1658               /* Now time-stamping can be used validly.  Time-stamping
1659                  means that if the sizes of the local and remote file
1660                  match, and local file is newer than the remote file,
1661                  it will not be retrieved.  Otherwise, the normal
1662                  download procedure is resumed.  */
1663               if (tml >= tmr &&
1664                   (hstat.contlen == -1 || local_size == hstat.contlen))
1665                 {
1666                   logprintf (LOG_VERBOSE, _("\
1667 Server file no newer than local file `%s' -- not retrieving.\n\n"),
1668                              local_filename);
1669                   FREEHSTAT (hstat);
1670                   return RETROK;
1671                 }
1672               else if (tml >= tmr)
1673                 logprintf (LOG_VERBOSE, _("\
1674 The sizes do not match (local %ld) -- retrieving.\n"), local_size);
1675               else
1676                 logputs (LOG_VERBOSE,
1677                          _("Remote file is newer, retrieving.\n"));
1678             }
1679           FREEHSTAT (hstat);
1680           continue;
1681         }
1682       if ((tmr != (time_t) (-1))
1683           && !opt.spider
1684           && ((hstat.len == hstat.contlen) ||
1685               ((hstat.res == 0) &&
1686                ((hstat.contlen == -1) ||
1687                 (hstat.len >= hstat.contlen && !opt.kill_longer)))))
1688         {
1689           /* #### This code repeats in http.c and ftp.c.  Move it to a
1690              function!  */
1691           const char *fl = NULL;
1692           if (opt.output_document)
1693             {
1694               if (opt.od_known_regular)
1695                 fl = opt.output_document;
1696             }
1697           else
1698             fl = u->local;
1699           if (fl)
1700             touch (fl, tmr);
1701         }
1702       /* End of time-stamping section.  */
1703
1704       if (opt.spider)
1705         {
1706           logprintf (LOG_NOTQUIET, "%d %s\n\n", hstat.statcode, hstat.error);
1707           return RETROK;
1708         }
1709
1710       /* It is now safe to free the remainder of hstat, since the
1711          strings within it will no longer be used.  */
1712       FREEHSTAT (hstat);
1713
1714       tmrate = rate (hstat.len - hstat.restval, hstat.dltime, 0);
1715
1716       if (hstat.len == hstat.contlen)
1717         {
1718           if (*dt & RETROKF)
1719             {
1720               logprintf (LOG_VERBOSE,
1721                          _("%s (%s) - `%s' saved [%ld/%ld]\n\n"),
1722                          tms, tmrate, locf, hstat.len, hstat.contlen);
1723               logprintf (LOG_NONVERBOSE,
1724                          "%s URL:%s [%ld/%ld] -> \"%s\" [%d]\n",
1725                          tms, u->url, hstat.len, hstat.contlen, locf, count);
1726             }
1727           ++opt.numurls;
1728           downloaded_increase (hstat.len);
1729
1730           /* Remember that we downloaded the file for later ".orig" code. */
1731           if (*dt & ADDED_HTML_EXTENSION)
1732             downloaded_file(FILE_DOWNLOADED_AND_HTML_EXTENSION_ADDED, locf);
1733           else
1734             downloaded_file(FILE_DOWNLOADED_NORMALLY, locf);
1735
1736           return RETROK;
1737         }
1738       else if (hstat.res == 0) /* No read error */
1739         {
1740           if (hstat.contlen == -1)  /* We don't know how much we were supposed
1741                                        to get, so assume we succeeded. */ 
1742             {
1743               if (*dt & RETROKF)
1744                 {
1745                   logprintf (LOG_VERBOSE,
1746                              _("%s (%s) - `%s' saved [%ld]\n\n"),
1747                              tms, tmrate, locf, hstat.len);
1748                   logprintf (LOG_NONVERBOSE,
1749                              "%s URL:%s [%ld] -> \"%s\" [%d]\n",
1750                              tms, u->url, hstat.len, locf, count);
1751                 }
1752               ++opt.numurls;
1753               downloaded_increase (hstat.len);
1754
1755               /* Remember that we downloaded the file for later ".orig" code. */
1756               if (*dt & ADDED_HTML_EXTENSION)
1757                 downloaded_file(FILE_DOWNLOADED_AND_HTML_EXTENSION_ADDED, locf);
1758               else
1759                 downloaded_file(FILE_DOWNLOADED_NORMALLY, locf);
1760               
1761               return RETROK;
1762             }
1763           else if (hstat.len < hstat.contlen) /* meaning we lost the
1764                                                  connection too soon */
1765             {
1766               logprintf (LOG_VERBOSE,
1767                          _("%s (%s) - Connection closed at byte %ld. "),
1768                          tms, tmrate, hstat.len);
1769               printwhat (count, opt.ntry);
1770               continue;
1771             }
1772           else if (!opt.kill_longer) /* meaning we got more than expected */
1773             {
1774               logprintf (LOG_VERBOSE,
1775                          _("%s (%s) - `%s' saved [%ld/%ld])\n\n"),
1776                          tms, tmrate, locf, hstat.len, hstat.contlen);
1777               logprintf (LOG_NONVERBOSE,
1778                          "%s URL:%s [%ld/%ld] -> \"%s\" [%d]\n",
1779                          tms, u->url, hstat.len, hstat.contlen, locf, count);
1780               ++opt.numurls;
1781               downloaded_increase (hstat.len);
1782
1783               /* Remember that we downloaded the file for later ".orig" code. */
1784               if (*dt & ADDED_HTML_EXTENSION)
1785                 downloaded_file(FILE_DOWNLOADED_AND_HTML_EXTENSION_ADDED, locf);
1786               else
1787                 downloaded_file(FILE_DOWNLOADED_NORMALLY, locf);
1788               
1789               return RETROK;
1790             }
1791           else                  /* the same, but not accepted */
1792             {
1793               logprintf (LOG_VERBOSE,
1794                          _("%s (%s) - Connection closed at byte %ld/%ld. "),
1795                          tms, tmrate, hstat.len, hstat.contlen);
1796               printwhat (count, opt.ntry);
1797               continue;
1798             }
1799         }
1800       else                      /* now hstat.res can only be -1 */
1801         {
1802           if (hstat.contlen == -1)
1803             {
1804               logprintf (LOG_VERBOSE,
1805                          _("%s (%s) - Read error at byte %ld (%s)."),
1806                          tms, tmrate, hstat.len, strerror (errno));
1807               printwhat (count, opt.ntry);
1808               continue;
1809             }
1810           else                  /* hstat.res == -1 and contlen is given */
1811             {
1812               logprintf (LOG_VERBOSE,
1813                          _("%s (%s) - Read error at byte %ld/%ld (%s). "),
1814                          tms, tmrate, hstat.len, hstat.contlen,
1815                          strerror (errno));
1816               printwhat (count, opt.ntry);
1817               continue;
1818             }
1819         }
1820       /* not reached */
1821       break;
1822     }
1823   while (!opt.ntry || (count < opt.ntry));
1824   return TRYLIMEXC;
1825 }
1826 \f
1827 /* Converts struct tm to time_t, assuming the data in tm is UTC rather
1828    than local timezone.
1829
1830    mktime is similar but assumes struct tm, also known as the
1831    "broken-down" form of time, is in local time zone.  mktime_from_utc
1832    uses mktime to make the conversion understanding that an offset
1833    will be introduced by the local time assumption.
1834
1835    mktime_from_utc then measures the introduced offset by applying
1836    gmtime to the initial result and applying mktime to the resulting
1837    "broken-down" form.  The difference between the two mktime results
1838    is the measured offset which is then subtracted from the initial
1839    mktime result to yield a calendar time which is the value returned.
1840
1841    tm_isdst in struct tm is set to 0 to force mktime to introduce a
1842    consistent offset (the non DST offset) since tm and tm+o might be
1843    on opposite sides of a DST change.
1844
1845    Some implementations of mktime return -1 for the nonexistent
1846    localtime hour at the beginning of DST.  In this event, use
1847    mktime(tm - 1hr) + 3600.
1848
1849    Schematically
1850      mktime(tm)   --> t+o
1851      gmtime(t+o)  --> tm+o
1852      mktime(tm+o) --> t+2o
1853      t+o - (t+2o - t+o) = t
1854
1855    Note that glibc contains a function of the same purpose named
1856    `timegm' (reverse of gmtime).  But obviously, it is not universally
1857    available, and unfortunately it is not straightforwardly
1858    extractable for use here.  Perhaps configure should detect timegm
1859    and use it where available.
1860
1861    Contributed by Roger Beeman <beeman@cisco.com>, with the help of
1862    Mark Baushke <mdb@cisco.com> and the rest of the Gurus at CISCO.
1863    Further improved by Roger with assistance from Edward J. Sabol
1864    based on input by Jamie Zawinski.  */
1865
1866 static time_t
1867 mktime_from_utc (struct tm *t)
1868 {
1869   time_t tl, tb;
1870   struct tm *tg;
1871
1872   tl = mktime (t);
1873   if (tl == -1)
1874     {
1875       t->tm_hour--;
1876       tl = mktime (t);
1877       if (tl == -1)
1878         return -1; /* can't deal with output from strptime */
1879       tl += 3600;
1880     }
1881   tg = gmtime (&tl);
1882   tg->tm_isdst = 0;
1883   tb = mktime (tg);
1884   if (tb == -1)
1885     {
1886       tg->tm_hour--;
1887       tb = mktime (tg);
1888       if (tb == -1)
1889         return -1; /* can't deal with output from gmtime */
1890       tb += 3600;
1891     }
1892   return (tl - (tb - tl));
1893 }
1894
1895 /* Check whether the result of strptime() indicates success.
1896    strptime() returns the pointer to how far it got to in the string.
1897    The processing has been successful if the string is at `GMT' or
1898    `+X', or at the end of the string.
1899
1900    In extended regexp parlance, the function returns 1 if P matches
1901    "^ *(GMT|[+-][0-9]|$)", 0 otherwise.  P being NULL (which strptime
1902    can return) is considered a failure and 0 is returned.  */
1903 static int
1904 check_end (const char *p)
1905 {
1906   if (!p)
1907     return 0;
1908   while (ISSPACE (*p))
1909     ++p;
1910   if (!*p
1911       || (p[0] == 'G' && p[1] == 'M' && p[2] == 'T')
1912       || ((p[0] == '+' || p[0] == '-') && ISDIGIT (p[1])))
1913     return 1;
1914   else
1915     return 0;
1916 }
1917
1918 /* Convert the textual specification of time in TIME_STRING to the
1919    number of seconds since the Epoch.
1920
1921    TIME_STRING can be in any of the three formats RFC2068 allows the
1922    HTTP servers to emit -- RFC1123-date, RFC850-date or asctime-date.
1923    Timezones are ignored, and should be GMT.
1924
1925    Return the computed time_t representation, or -1 if the conversion
1926    fails.
1927
1928    This function uses strptime with various string formats for parsing
1929    TIME_STRING.  This results in a parser that is not as lenient in
1930    interpreting TIME_STRING as I would like it to be.  Being based on
1931    strptime, it always allows shortened months, one-digit days, etc.,
1932    but due to the multitude of formats in which time can be
1933    represented, an ideal HTTP time parser would be even more
1934    forgiving.  It should completely ignore things like week days and
1935    concentrate only on the various forms of representing years,
1936    months, days, hours, minutes, and seconds.  For example, it would
1937    be nice if it accepted ISO 8601 out of the box.
1938
1939    I've investigated free and PD code for this purpose, but none was
1940    usable.  getdate was big and unwieldy, and had potential copyright
1941    issues, or so I was informed.  Dr. Marcus Hennecke's atotm(),
1942    distributed with phttpd, is excellent, but we cannot use it because
1943    it is not assigned to the FSF.  So I stuck it with strptime.  */
1944
1945 time_t
1946 http_atotm (char *time_string)
1947 {
1948   /* NOTE: Solaris strptime man page claims that %n and %t match white
1949      space, but that's not universally available.  Instead, we simply
1950      use ` ' to mean "skip all WS", which works under all strptime
1951      implementations I've tested.  */
1952
1953   static const char *time_formats[] = {
1954     "%a, %d %b %Y %T",          /* RFC1123: Thu, 29 Jan 1998 22:12:57 */
1955     "%A, %d-%b-%y %T",          /* RFC850:  Thursday, 29-Jan-98 22:12:57 */
1956     "%a, %d-%b-%Y %T",          /* pseudo-RFC850:  Thu, 29-Jan-1998 22:12:57
1957                                    (google.com uses this for their cookies.) */
1958     "%a %b %d %T %Y"            /* asctime: Thu Jan 29 22:12:57 1998 */
1959   };
1960
1961   int i;
1962   struct tm t;
1963
1964   /* According to Roger Beeman, we need to initialize tm_isdst, since
1965      strptime won't do it.  */
1966   t.tm_isdst = 0;
1967
1968   /* Note that under foreign locales Solaris strptime() fails to
1969      recognize English dates, which renders this function useless.  We
1970      solve this by being careful not to affect LC_TIME when
1971      initializing locale.
1972
1973      Another solution would be to temporarily set locale to C, invoke
1974      strptime(), and restore it back.  This is slow and dirty,
1975      however, and locale support other than LC_MESSAGES can mess other
1976      things, so I rather chose to stick with just setting LC_MESSAGES.
1977
1978      GNU strptime does not have this problem because it recognizes
1979      both international and local dates.  */
1980
1981   for (i = 0; i < ARRAY_SIZE (time_formats); i++)
1982     if (check_end (strptime (time_string, time_formats[i], &t)))
1983       return mktime_from_utc (&t);
1984
1985   /* All formats have failed.  */
1986   return -1;
1987 }
1988 \f
1989 /* Authorization support: We support two authorization schemes:
1990
1991    * `Basic' scheme, consisting of base64-ing USER:PASSWORD string;
1992
1993    * `Digest' scheme, added by Junio Hamano <junio@twinsun.com>,
1994    consisting of answering to the server's challenge with the proper
1995    MD5 digests.  */
1996
1997 /* How many bytes it will take to store LEN bytes in base64.  */
1998 #define BASE64_LENGTH(len) (4 * (((len) + 2) / 3))
1999
2000 /* Encode the string S of length LENGTH to base64 format and place it
2001    to STORE.  STORE will be 0-terminated, and must point to a writable
2002    buffer of at least 1+BASE64_LENGTH(length) bytes.  */
2003 static void
2004 base64_encode (const char *s, char *store, int length)
2005 {
2006   /* Conversion table.  */
2007   static char tbl[64] = {
2008     'A','B','C','D','E','F','G','H',
2009     'I','J','K','L','M','N','O','P',
2010     'Q','R','S','T','U','V','W','X',
2011     'Y','Z','a','b','c','d','e','f',
2012     'g','h','i','j','k','l','m','n',
2013     'o','p','q','r','s','t','u','v',
2014     'w','x','y','z','0','1','2','3',
2015     '4','5','6','7','8','9','+','/'
2016   };
2017   int i;
2018   unsigned char *p = (unsigned char *)store;
2019
2020   /* Transform the 3x8 bits to 4x6 bits, as required by base64.  */
2021   for (i = 0; i < length; i += 3)
2022     {
2023       *p++ = tbl[s[0] >> 2];
2024       *p++ = tbl[((s[0] & 3) << 4) + (s[1] >> 4)];
2025       *p++ = tbl[((s[1] & 0xf) << 2) + (s[2] >> 6)];
2026       *p++ = tbl[s[2] & 0x3f];
2027       s += 3;
2028     }
2029   /* Pad the result if necessary...  */
2030   if (i == length + 1)
2031     *(p - 1) = '=';
2032   else if (i == length + 2)
2033     *(p - 1) = *(p - 2) = '=';
2034   /* ...and zero-terminate it.  */
2035   *p = '\0';
2036 }
2037
2038 /* Create the authentication header contents for the `Basic' scheme.
2039    This is done by encoding the string `USER:PASS' in base64 and
2040    prepending `HEADER: Basic ' to it.  */
2041 static char *
2042 basic_authentication_encode (const char *user, const char *passwd,
2043                              const char *header)
2044 {
2045   char *t1, *t2, *res;
2046   int len1 = strlen (user) + 1 + strlen (passwd);
2047   int len2 = BASE64_LENGTH (len1);
2048
2049   t1 = (char *)alloca (len1 + 1);
2050   sprintf (t1, "%s:%s", user, passwd);
2051   t2 = (char *)alloca (1 + len2);
2052   base64_encode (t1, t2, len1);
2053   res = (char *)xmalloc (len2 + 11 + strlen (header));
2054   sprintf (res, "%s: Basic %s\r\n", header, t2);
2055
2056   return res;
2057 }
2058
2059 #ifdef USE_DIGEST
2060 /* Parse HTTP `WWW-Authenticate:' header.  AU points to the beginning
2061    of a field in such a header.  If the field is the one specified by
2062    ATTR_NAME ("realm", "opaque", and "nonce" are used by the current
2063    digest authorization code), extract its value in the (char*)
2064    variable pointed by RET.  Returns negative on a malformed header,
2065    or number of bytes that have been parsed by this call.  */
2066 static int
2067 extract_header_attr (const char *au, const char *attr_name, char **ret)
2068 {
2069   const char *cp, *ep;
2070
2071   ep = cp = au;
2072
2073   if (strncmp (cp, attr_name, strlen (attr_name)) == 0)
2074     {
2075       cp += strlen (attr_name);
2076       if (!*cp)
2077         return -1;
2078       cp += skip_lws (cp);
2079       if (*cp != '=')
2080         return -1;
2081       if (!*++cp)
2082         return -1;
2083       cp += skip_lws (cp);
2084       if (*cp != '\"')
2085         return -1;
2086       if (!*++cp)
2087         return -1;
2088       for (ep = cp; *ep && *ep != '\"'; ep++)
2089         ;
2090       if (!*ep)
2091         return -1;
2092       FREE_MAYBE (*ret);
2093       *ret = strdupdelim (cp, ep);
2094       return ep - au + 1;
2095     }
2096   else
2097     return 0;
2098 }
2099
2100 /* Dump the hexadecimal representation of HASH to BUF.  HASH should be
2101    an array of 16 bytes containing the hash keys, and BUF should be a
2102    buffer of 33 writable characters (32 for hex digits plus one for
2103    zero termination).  */
2104 static void
2105 dump_hash (unsigned char *buf, const unsigned char *hash)
2106 {
2107   int i;
2108
2109   for (i = 0; i < MD5_HASHLEN; i++, hash++)
2110     {
2111       *buf++ = XDIGIT_TO_xchar (*hash >> 4);
2112       *buf++ = XDIGIT_TO_xchar (*hash & 0xf);
2113     }
2114   *buf = '\0';
2115 }
2116
2117 /* Take the line apart to find the challenge, and compose a digest
2118    authorization header.  See RFC2069 section 2.1.2.  */
2119 char *
2120 digest_authentication_encode (const char *au, const char *user,
2121                               const char *passwd, const char *method,
2122                               const char *path)
2123 {
2124   static char *realm, *opaque, *nonce;
2125   static struct {
2126     const char *name;
2127     char **variable;
2128   } options[] = {
2129     { "realm", &realm },
2130     { "opaque", &opaque },
2131     { "nonce", &nonce }
2132   };
2133   char *res;
2134
2135   realm = opaque = nonce = NULL;
2136
2137   au += 6;                      /* skip over `Digest' */
2138   while (*au)
2139     {
2140       int i;
2141
2142       au += skip_lws (au);
2143       for (i = 0; i < ARRAY_SIZE (options); i++)
2144         {
2145           int skip = extract_header_attr (au, options[i].name,
2146                                           options[i].variable);
2147           if (skip < 0)
2148             {
2149               FREE_MAYBE (realm);
2150               FREE_MAYBE (opaque);
2151               FREE_MAYBE (nonce);
2152               return NULL;
2153             }
2154           else if (skip)
2155             {
2156               au += skip;
2157               break;
2158             }
2159         }
2160       if (i == ARRAY_SIZE (options))
2161         {
2162           while (*au && *au != '=')
2163             au++;
2164           if (*au && *++au)
2165             {
2166               au += skip_lws (au);
2167               if (*au == '\"')
2168                 {
2169                   au++;
2170                   while (*au && *au != '\"')
2171                     au++;
2172                   if (*au)
2173                     au++;
2174                 }
2175             }
2176         }
2177       while (*au && *au != ',')
2178         au++;
2179       if (*au)
2180         au++;
2181     }
2182   if (!realm || !nonce || !user || !passwd || !path || !method)
2183     {
2184       FREE_MAYBE (realm);
2185       FREE_MAYBE (opaque);
2186       FREE_MAYBE (nonce);
2187       return NULL;
2188     }
2189
2190   /* Calculate the digest value.  */
2191   {
2192     struct md5_ctx ctx;
2193     unsigned char hash[MD5_HASHLEN];
2194     unsigned char a1buf[MD5_HASHLEN * 2 + 1], a2buf[MD5_HASHLEN * 2 + 1];
2195     unsigned char response_digest[MD5_HASHLEN * 2 + 1];
2196
2197     /* A1BUF = H(user ":" realm ":" password) */
2198     md5_init_ctx (&ctx);
2199     md5_process_bytes (user, strlen (user), &ctx);
2200     md5_process_bytes (":", 1, &ctx);
2201     md5_process_bytes (realm, strlen (realm), &ctx);
2202     md5_process_bytes (":", 1, &ctx);
2203     md5_process_bytes (passwd, strlen (passwd), &ctx);
2204     md5_finish_ctx (&ctx, hash);
2205     dump_hash (a1buf, hash);
2206
2207     /* A2BUF = H(method ":" path) */
2208     md5_init_ctx (&ctx);
2209     md5_process_bytes (method, strlen (method), &ctx);
2210     md5_process_bytes (":", 1, &ctx);
2211     md5_process_bytes (path, strlen (path), &ctx);
2212     md5_finish_ctx (&ctx, hash);
2213     dump_hash (a2buf, hash);
2214
2215     /* RESPONSE_DIGEST = H(A1BUF ":" nonce ":" A2BUF) */
2216     md5_init_ctx (&ctx);
2217     md5_process_bytes (a1buf, MD5_HASHLEN * 2, &ctx);
2218     md5_process_bytes (":", 1, &ctx);
2219     md5_process_bytes (nonce, strlen (nonce), &ctx);
2220     md5_process_bytes (":", 1, &ctx);
2221     md5_process_bytes (a2buf, MD5_HASHLEN * 2, &ctx);
2222     md5_finish_ctx (&ctx, hash);
2223     dump_hash (response_digest, hash);
2224
2225     res = (char*) xmalloc (strlen (user)
2226                            + strlen (user)
2227                            + strlen (realm)
2228                            + strlen (nonce)
2229                            + strlen (path)
2230                            + 2 * MD5_HASHLEN /*strlen (response_digest)*/
2231                            + (opaque ? strlen (opaque) : 0)
2232                            + 128);
2233     sprintf (res, "Authorization: Digest \
2234 username=\"%s\", realm=\"%s\", nonce=\"%s\", uri=\"%s\", response=\"%s\"",
2235              user, realm, nonce, path, response_digest);
2236     if (opaque)
2237       {
2238         char *p = res + strlen (res);
2239         strcat (p, ", opaque=\"");
2240         strcat (p, opaque);
2241         strcat (p, "\"");
2242       }
2243     strcat (res, "\r\n");
2244   }
2245   return res;
2246 }
2247 #endif /* USE_DIGEST */
2248
2249
2250 #define BEGINS_WITH(line, string_constant)                              \
2251   (!strncasecmp (line, string_constant, sizeof (string_constant) - 1)   \
2252    && (ISSPACE (line[sizeof (string_constant) - 1])                     \
2253        || !line[sizeof (string_constant) - 1]))
2254
2255 static int
2256 known_authentication_scheme_p (const char *au)
2257 {
2258   return BEGINS_WITH (au, "Basic")
2259     || BEGINS_WITH (au, "Digest")
2260     || BEGINS_WITH (au, "NTLM");
2261 }
2262
2263 #undef BEGINS_WITH
2264
2265 /* Create the HTTP authorization request header.  When the
2266    `WWW-Authenticate' response header is seen, according to the
2267    authorization scheme specified in that header (`Basic' and `Digest'
2268    are supported by the current implementation), produce an
2269    appropriate HTTP authorization request header.  */
2270 static char *
2271 create_authorization_line (const char *au, const char *user,
2272                            const char *passwd, const char *method,
2273                            const char *path)
2274 {
2275   char *wwwauth = NULL;
2276
2277   if (!strncasecmp (au, "Basic", 5))
2278     wwwauth = basic_authentication_encode (user, passwd, "Authorization");
2279   if (!strncasecmp (au, "NTLM", 4))
2280     wwwauth = basic_authentication_encode (user, passwd, "Authorization");
2281 #ifdef USE_DIGEST
2282   else if (!strncasecmp (au, "Digest", 6))
2283     wwwauth = digest_authentication_encode (au, user, passwd, method, path);
2284 #endif /* USE_DIGEST */
2285   return wwwauth;
2286 }