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