]> sjero.net Git - wget/blob - src/http.c
34f97ce7cd348f5d2343b9389ba7cd435af3e666
[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 (hs->newloc);
913           FREE_MAYBE (all_headers);
914           CLOSE_INVALIDATE (sock);
915           return HEOF;
916         }
917       else if (status == HG_ERROR)
918         {
919           logputs (LOG_VERBOSE, "\n");
920           logprintf (LOG_NOTQUIET, _("Read error (%s) in headers.\n"),
921                      strerror (errno));
922           xfree (hdr);
923           FREE_MAYBE (type);
924           FREE_MAYBE (hs->newloc);
925           FREE_MAYBE (all_headers);
926           CLOSE_INVALIDATE (sock);
927           return HERR;
928         }
929
930       /* If the headers are to be saved to a file later, save them to
931          memory now.  */
932       if (opt.save_headers)
933         {
934           int lh = strlen (hdr);
935           all_headers = (char *)xrealloc (all_headers, all_length + lh + 2);
936           memcpy (all_headers + all_length, hdr, lh);
937           all_length += lh;
938           all_headers[all_length++] = '\n';
939           all_headers[all_length] = '\0';
940         }
941
942       /* Print the header if requested.  */
943       if (opt.server_response && hcount != 1)
944         logprintf (LOG_VERBOSE, "\n%d %s", hcount, hdr);
945
946       /* Check for status line.  */
947       if (hcount == 1)
948         {
949           const char *error;
950           /* Parse the first line of server response.  */
951           statcode = parse_http_status_line (hdr, &error);
952           hs->statcode = statcode;
953           /* Store the descriptive response.  */
954           if (statcode == -1) /* malformed response */
955             {
956               /* A common reason for "malformed response" error is the
957                  case when no data was actually received.  Handle this
958                  special case.  */
959               if (!*hdr)
960                 hs->error = xstrdup (_("No data received"));
961               else
962                 hs->error = xstrdup (_("Malformed status line"));
963               xfree (hdr);
964               break;
965             }
966           else if (!*error)
967             hs->error = xstrdup (_("(no description)"));
968           else
969             hs->error = xstrdup (error);
970
971           if ((statcode != -1)
972 #ifdef DEBUG
973               && !opt.debug
974 #endif
975               )
976             logprintf (LOG_VERBOSE, "%d %s", statcode, error);
977
978           goto done_header;
979         }
980
981       /* Exit on empty header.  */
982       if (!*hdr)
983         {
984           xfree (hdr);
985           break;
986         }
987
988       /* Try getting content-length.  */
989       if (contlen == -1 && !opt.ignore_length)
990         if (header_process (hdr, "Content-Length", header_extract_number,
991                             &contlen))
992           goto done_header;
993       /* Try getting content-type.  */
994       if (!type)
995         if (header_process (hdr, "Content-Type", http_process_type, &type))
996           goto done_header;
997       /* Try getting location.  */
998       if (!hs->newloc)
999         if (header_process (hdr, "Location", header_strdup, &hs->newloc))
1000           goto done_header;
1001       /* Try getting last-modified.  */
1002       if (!hs->remote_time)
1003         if (header_process (hdr, "Last-Modified", header_strdup,
1004                             &hs->remote_time))
1005           goto done_header;
1006       /* Try getting cookies. */
1007       if (opt.cookies)
1008         if (header_process (hdr, "Set-Cookie", set_cookie_header_cb, u))
1009           goto done_header;
1010       /* Try getting www-authentication.  */
1011       if (!authenticate_h)
1012         if (header_process (hdr, "WWW-Authenticate", header_strdup,
1013                             &authenticate_h))
1014           goto done_header;
1015       /* Check for accept-ranges header.  If it contains the word
1016          `none', disable the ranges.  */
1017       if (*dt & ACCEPTRANGES)
1018         {
1019           int nonep;
1020           if (header_process (hdr, "Accept-Ranges", http_process_none, &nonep))
1021             {
1022               if (nonep)
1023                 *dt &= ~ACCEPTRANGES;
1024               goto done_header;
1025             }
1026         }
1027       /* Try getting content-range.  */
1028       if (contrange == -1)
1029         {
1030           struct http_process_range_closure closure;
1031           if (header_process (hdr, "Content-Range", http_process_range, &closure))
1032             {
1033               contrange = closure.first_byte_pos;
1034               goto done_header;
1035             }
1036         }
1037       /* Check for keep-alive related responses. */
1038       if (!inhibit_keep_alive)
1039         {
1040           /* Check for the `Keep-Alive' header. */
1041           if (!http_keep_alive_1)
1042             {
1043               if (header_process (hdr, "Keep-Alive", header_exists,
1044                                   &http_keep_alive_1))
1045                 goto done_header;
1046             }
1047           /* Check for `Connection: Keep-Alive'. */
1048           if (!http_keep_alive_2)
1049             {
1050               if (header_process (hdr, "Connection", http_process_connection,
1051                                   &http_keep_alive_2))
1052                 goto done_header;
1053             }
1054         }
1055     done_header:
1056       xfree (hdr);
1057     }
1058
1059   logputs (LOG_VERBOSE, "\n");
1060
1061   if (contlen != -1
1062       && (http_keep_alive_1 || http_keep_alive_2))
1063     {
1064       assert (inhibit_keep_alive == 0);
1065       keep_alive = 1;
1066     }
1067   if (keep_alive)
1068     /* The server has promised that it will not close the connection
1069        when we're done.  This means that we can register it.  */
1070 #ifndef HAVE_SSL
1071     register_persistent (u->host, u->port, sock);
1072 #else
1073     register_persistent (u->host, u->port, sock, ssl);
1074 #endif /* HAVE_SSL */
1075
1076   if ((statcode == HTTP_STATUS_UNAUTHORIZED)
1077       && authenticate_h)
1078     {
1079       /* Authorization is required.  */
1080       FREE_MAYBE (type);
1081       type = NULL;
1082       FREEHSTAT (*hs);
1083       CLOSE_INVALIDATE (sock);  /* would be CLOSE_FINISH, but there
1084                                    might be more bytes in the body. */
1085       if (auth_tried_already)
1086         {
1087           /* If we have tried it already, then there is not point
1088              retrying it.  */
1089         failed:
1090           logputs (LOG_NOTQUIET, _("Authorization failed.\n"));
1091           xfree (authenticate_h);
1092           return AUTHFAILED;
1093         }
1094       else if (!known_authentication_scheme_p (authenticate_h))
1095         {
1096           xfree (authenticate_h);
1097           logputs (LOG_NOTQUIET, _("Unknown authentication scheme.\n"));
1098           return AUTHFAILED;
1099         }
1100       else if (BEGINS_WITH (authenticate_h, "Basic"))
1101         {
1102           /* The authentication scheme is basic, the one we try by
1103              default, and it failed.  There's no sense in trying
1104              again.  */
1105           goto failed;
1106         }
1107       else
1108         {
1109           auth_tried_already = 1;
1110           goto again;
1111         }
1112     }
1113   /* We do not need this anymore.  */
1114   if (authenticate_h)
1115     {
1116       xfree (authenticate_h);
1117       authenticate_h = NULL;
1118     }
1119
1120   /* 20x responses are counted among successful by default.  */
1121   if (H_20X (statcode))
1122     *dt |= RETROKF;
1123
1124   if (type && !strncasecmp (type, TEXTHTML_S, strlen (TEXTHTML_S)))
1125     *dt |= TEXTHTML;
1126   else
1127     /* We don't assume text/html by default.  */
1128     *dt &= ~TEXTHTML;
1129
1130   if (opt.html_extension && (*dt & TEXTHTML))
1131     /* -E / --html-extension / html_extension = on was specified, and this is a
1132        text/html file.  If some case-insensitive variation on ".htm[l]" isn't
1133        already the file's suffix, tack on ".html". */
1134     {
1135       char*  last_period_in_local_filename = strrchr(u->local, '.');
1136
1137       if (last_period_in_local_filename == NULL ||
1138           !(strcasecmp(last_period_in_local_filename, ".htm") == EQ ||
1139             strcasecmp(last_period_in_local_filename, ".html") == EQ))
1140         {
1141           size_t  local_filename_len = strlen(u->local);
1142           
1143           u->local = xrealloc(u->local, local_filename_len + sizeof(".html"));
1144           strcpy(u->local + local_filename_len, ".html");
1145
1146           *dt |= ADDED_HTML_EXTENSION;
1147         }
1148     }
1149
1150   if (contrange == -1)
1151     {
1152       /* We did not get a content-range header.  This means that the
1153          server did not honor our `Range' request.  Normally, this
1154          means we should reset hs->restval and continue normally.  */
1155
1156       /* However, if `-c' is used, we need to be a bit more careful:
1157
1158          1. If `-c' is specified and the file already existed when
1159          Wget was started, it would be a bad idea for us to start
1160          downloading it from scratch, effectively truncating it.  I
1161          believe this cannot happen unless `-c' was specified.
1162
1163          2. If `-c' is used on a file that is already fully
1164          downloaded, we're requesting bytes after the end of file,
1165          which can result in server not honoring `Range'.  If this is
1166          the case, `Content-Length' will be equal to the length of the
1167          file.  */
1168       if (opt.always_rest)
1169         {
1170           /* Check for condition #2. */
1171           if (hs->restval == contlen)
1172             {
1173               logputs (LOG_VERBOSE, _("\
1174 \n    The file is already fully retrieved; nothing to do.\n\n"));
1175               /* In case the caller inspects. */
1176               hs->len = contlen;
1177               hs->res = 0;
1178               FREE_MAYBE (type);
1179               FREE_MAYBE (hs->newloc);
1180               FREE_MAYBE (all_headers);
1181               CLOSE_INVALIDATE (sock);  /* would be CLOSE_FINISH, but there
1182                                            might be more bytes in the body. */
1183               return RETRFINISHED;
1184             }
1185
1186           /* Check for condition #1. */
1187           if (hs->no_truncate)
1188             {
1189               logprintf (LOG_NOTQUIET,
1190                          _("\
1191 \n\
1192     The server does not support continued download;\n\
1193     refusing to truncate `%s'.\n\n"), u->local);
1194               return CONTNOTSUPPORTED;
1195             }
1196
1197           /* Fallthrough */
1198         }
1199
1200       hs->restval = 0;
1201     }
1202
1203   else if (contrange != hs->restval ||
1204            (H_PARTIAL (statcode) && contrange == -1))
1205     {
1206       /* This means the whole request was somehow misunderstood by the
1207          server.  Bail out.  */
1208       FREE_MAYBE (type);
1209       FREE_MAYBE (hs->newloc);
1210       FREE_MAYBE (all_headers);
1211       CLOSE_INVALIDATE (sock);
1212       return RANGEERR;
1213     }
1214
1215   if (hs->restval)
1216     {
1217       if (contlen != -1)
1218         contlen += contrange;
1219       else
1220         contrange = -1;        /* If conent-length was not sent,
1221                                   content-range will be ignored.  */
1222     }
1223   hs->contlen = contlen;
1224
1225   /* Return if redirected.  */
1226   if (H_REDIRECTED (statcode) || statcode == HTTP_STATUS_MULTIPLE_CHOICES)
1227     {
1228       /* RFC2068 says that in case of the 300 (multiple choices)
1229          response, the server can output a preferred URL through
1230          `Location' header; otherwise, the request should be treated
1231          like GET.  So, if the location is set, it will be a
1232          redirection; otherwise, just proceed normally.  */
1233       if (statcode == HTTP_STATUS_MULTIPLE_CHOICES && !hs->newloc)
1234         *dt |= RETROKF;
1235       else
1236         {
1237           logprintf (LOG_VERBOSE,
1238                      _("Location: %s%s\n"),
1239                      hs->newloc ? hs->newloc : _("unspecified"),
1240                      hs->newloc ? _(" [following]") : "");
1241           CLOSE_INVALIDATE (sock);      /* would be CLOSE_FINISH, but there
1242                                            might be more bytes in the body. */
1243           FREE_MAYBE (type);
1244           FREE_MAYBE (all_headers);
1245           return NEWLOCATION;
1246         }
1247     }
1248   if (opt.verbose)
1249     {
1250       if ((*dt & RETROKF) && !opt.server_response)
1251         {
1252           /* No need to print this output if the body won't be
1253              downloaded at all, or if the original server response is
1254              printed.  */
1255           logputs (LOG_VERBOSE, _("Length: "));
1256           if (contlen != -1)
1257             {
1258               logputs (LOG_VERBOSE, legible (contlen));
1259               if (contrange != -1)
1260                 logprintf (LOG_VERBOSE, _(" (%s to go)"),
1261                            legible (contlen - contrange));
1262             }
1263           else
1264             logputs (LOG_VERBOSE,
1265                      opt.ignore_length ? _("ignored") : _("unspecified"));
1266           if (type)
1267             logprintf (LOG_VERBOSE, " [%s]\n", type);
1268           else
1269             logputs (LOG_VERBOSE, "\n");
1270         }
1271     }
1272   FREE_MAYBE (type);
1273   type = NULL;                  /* We don't need it any more.  */
1274
1275   /* Return if we have no intention of further downloading.  */
1276   if (!(*dt & RETROKF) || (*dt & HEAD_ONLY))
1277     {
1278       /* In case the caller cares to look...  */
1279       hs->len = 0L;
1280       hs->res = 0;
1281       FREE_MAYBE (type);
1282       FREE_MAYBE (all_headers);
1283       CLOSE_INVALIDATE (sock);  /* would be CLOSE_FINISH, but there
1284                                    might be more bytes in the body. */
1285       return RETRFINISHED;
1286     }
1287
1288   /* Open the local file.  */
1289   if (!opt.dfp)
1290     {
1291       mkalldirs (u->local);
1292       if (opt.backups)
1293         rotate_backups (u->local);
1294       fp = fopen (u->local, hs->restval ? "ab" : "wb");
1295       if (!fp)
1296         {
1297           logprintf (LOG_NOTQUIET, "%s: %s\n", u->local, strerror (errno));
1298           CLOSE_INVALIDATE (sock); /* would be CLOSE_FINISH, but there
1299                                       might be more bytes in the body. */
1300           FREE_MAYBE (all_headers);
1301           return FOPENERR;
1302         }
1303     }
1304   else                          /* opt.dfp */
1305     {
1306       extern int global_download_count;
1307       fp = opt.dfp;
1308       /* To ensure that repeated "from scratch" downloads work for -O
1309          files, we rewind the file pointer, unless restval is
1310          non-zero.  (This works only when -O is used on regular files,
1311          but it's still a valuable feature.)
1312
1313          However, this loses when more than one URL is specified on
1314          the command line the second rewinds eradicates the contents
1315          of the first download.  Thus we disable the above trick for
1316          all the downloads except the very first one.
1317
1318          #### A possible solution to this would be to remember the
1319          file position in the output document and to seek to that
1320          position, instead of rewinding.  */
1321       if (!hs->restval && global_download_count == 0)
1322         {
1323           /* This will silently fail for streams that don't correspond
1324              to regular files, but that's OK.  */
1325           rewind (fp);
1326           /* ftruncate is needed because opt.dfp is opened in append
1327              mode if opt.always_rest is set.  */
1328           ftruncate (fileno (fp), 0);
1329           clearerr (fp);
1330         }
1331     }
1332
1333   /* #### This confuses the code that checks for file size.  There
1334      should be some overhead information.  */
1335   if (opt.save_headers)
1336     fwrite (all_headers, 1, all_length, fp);
1337   timer = wtimer_new ();
1338   /* Get the contents of the document.  */
1339   hs->res = get_contents (sock, fp, &hs->len, hs->restval,
1340                           (contlen != -1 ? contlen : 0),
1341                           &rbuf, keep_alive);
1342   hs->dltime = wtimer_elapsed (timer);
1343   wtimer_delete (timer);
1344   {
1345     /* Close or flush the file.  We have to be careful to check for
1346        error here.  Checking the result of fwrite() is not enough --
1347        errors could go unnoticed!  */
1348     int flush_res;
1349     if (!opt.dfp)
1350       flush_res = fclose (fp);
1351     else
1352       flush_res = fflush (fp);
1353     if (flush_res == EOF)
1354       hs->res = -2;
1355   }
1356   FREE_MAYBE (all_headers);
1357   CLOSE_FINISH (sock);
1358   if (hs->res == -2)
1359     return FWRITEERR;
1360   return RETRFINISHED;
1361 }
1362
1363 /* The genuine HTTP loop!  This is the part where the retrieval is
1364    retried, and retried, and retried, and...  */
1365 uerr_t
1366 http_loop (struct urlinfo *u, char **newloc, int *dt)
1367 {
1368   int count;
1369   int use_ts, got_head = 0;     /* time-stamping info */
1370   char *filename_plus_orig_suffix;
1371   char *local_filename = NULL;
1372   char *tms, *suf, *locf, *tmrate;
1373   uerr_t err;
1374   time_t tml = -1, tmr = -1;    /* local and remote time-stamps */
1375   long local_size = 0;          /* the size of the local file */
1376   size_t filename_len;
1377   struct http_stat hstat;       /* HTTP status */
1378   struct stat st;
1379
1380   /* This used to be done in main(), but it's a better idea to do it
1381      here so that we don't go through the hoops if we're just using
1382      FTP or whatever. */
1383   if (opt.cookies && opt.cookies_input && !cookies_loaded_p)
1384     {
1385       load_cookies (opt.cookies_input);
1386       cookies_loaded_p = 1;
1387     }
1388
1389   *newloc = NULL;
1390
1391   /* Warn on (likely bogus) wildcard usage in HTTP.  Don't use
1392      has_wildcards_p because it would also warn on `?', and we know that
1393      shows up in CGI paths a *lot*.  */
1394   if (strchr (u->url, '*'))
1395     logputs (LOG_VERBOSE, _("Warning: wildcards not supported in HTTP.\n"));
1396
1397   /* Determine the local filename.  */
1398   if (!u->local)
1399     u->local = url_filename (u->proxy ? u->proxy : u);
1400
1401   if (!opt.output_document)
1402     locf = u->local;
1403   else
1404     locf = opt.output_document;
1405
1406   /* Yuck.  Multiple returns suck.  We need to remember to free() the space we
1407      xmalloc() here before EACH return.  This is one reason it's better to set
1408      flags that influence flow control and then return once at the end. */
1409   filename_len = strlen(u->local);
1410   filename_plus_orig_suffix = xmalloc(filename_len + sizeof(".orig"));
1411
1412   if (opt.noclobber && file_exists_p (u->local))
1413     {
1414       /* If opt.noclobber is turned on and file already exists, do not
1415          retrieve the file */
1416       logprintf (LOG_VERBOSE, _("\
1417 File `%s' already there, will not retrieve.\n"), u->local);
1418       /* If the file is there, we suppose it's retrieved OK.  */
1419       *dt |= RETROKF;
1420
1421       /* #### Bogusness alert.  */
1422       /* If its suffix is "html" or (yuck!) "htm", we suppose it's
1423          text/html, a harmless lie.  */
1424       if (((suf = suffix (u->local)) != NULL)
1425           && (!strcmp (suf, "html") || !strcmp (suf, "htm")))
1426         *dt |= TEXTHTML;
1427       xfree (suf);
1428       xfree (filename_plus_orig_suffix); /* must precede every return! */
1429       /* Another harmless lie: */
1430       return RETROK;
1431     }
1432
1433   use_ts = 0;
1434   if (opt.timestamping)
1435     {
1436       boolean  local_dot_orig_file_exists = FALSE;
1437
1438       if (opt.backup_converted)
1439         /* If -K is specified, we'll act on the assumption that it was specified
1440            last time these files were downloaded as well, and instead of just
1441            comparing local file X against server file X, we'll compare local
1442            file X.orig (if extant, else X) against server file X.  If -K
1443            _wasn't_ specified last time, or the server contains files called
1444            *.orig, -N will be back to not operating correctly with -k. */
1445         {
1446           /* Would a single s[n]printf() call be faster?  --dan
1447
1448              Definitely not.  sprintf() is horribly slow.  It's a
1449              different question whether the difference between the two
1450              affects a program.  Usually I'd say "no", but at one
1451              point I profiled Wget, and found that a measurable and
1452              non-negligible amount of time was lost calling sprintf()
1453              in url.c.  Replacing sprintf with inline calls to
1454              strcpy() and long_to_string() made a difference.
1455              --hniksic */
1456           strcpy(filename_plus_orig_suffix, u->local);
1457           strcpy(filename_plus_orig_suffix + filename_len, ".orig");
1458
1459           /* Try to stat() the .orig file. */
1460           if (stat(filename_plus_orig_suffix, &st) == 0)
1461             {
1462               local_dot_orig_file_exists = TRUE;
1463               local_filename = filename_plus_orig_suffix;
1464             }
1465         }      
1466
1467       if (!local_dot_orig_file_exists)
1468         /* Couldn't stat() <file>.orig, so try to stat() <file>. */
1469         if (stat (u->local, &st) == 0)
1470           local_filename = u->local;
1471
1472       if (local_filename != NULL)
1473         /* There was a local file, so we'll check later to see if the version
1474            the server has is the same version we already have, allowing us to
1475            skip a download. */
1476         {
1477           use_ts = 1;
1478           tml = st.st_mtime;
1479           local_size = st.st_size;
1480           got_head = 0;
1481         }
1482     }
1483   /* Reset the counter.  */
1484   count = 0;
1485   *dt = 0 | ACCEPTRANGES;
1486   /* THE loop */
1487   do
1488     {
1489       /* Increment the pass counter.  */
1490       ++count;
1491       sleep_between_retrievals (count);
1492       /* Get the current time string.  */
1493       tms = time_str (NULL);
1494       /* Print fetch message, if opt.verbose.  */
1495       if (opt.verbose)
1496         {
1497           char *hurl = str_url (u->proxy ? u->proxy : u, 1);
1498           char tmp[15];
1499           strcpy (tmp, "        ");
1500           if (count > 1)
1501             sprintf (tmp, _("(try:%2d)"), count);
1502           logprintf (LOG_VERBOSE, "--%s--  %s\n  %s => `%s'\n",
1503                      tms, hurl, tmp, locf);
1504 #ifdef WINDOWS
1505           ws_changetitle (hurl, 1);
1506 #endif
1507           xfree (hurl);
1508         }
1509
1510       /* Default document type is empty.  However, if spider mode is
1511          on or time-stamping is employed, HEAD_ONLY commands is
1512          encoded within *dt.  */
1513       if (opt.spider || (use_ts && !got_head))
1514         *dt |= HEAD_ONLY;
1515       else
1516         *dt &= ~HEAD_ONLY;
1517       /* Assume no restarting.  */
1518       hstat.restval = 0L;
1519       /* Decide whether or not to restart.  */
1520       if (((count > 1 && (*dt & ACCEPTRANGES)) || opt.always_rest)
1521           && file_exists_p (locf))
1522         if (stat (locf, &st) == 0 && S_ISREG (st.st_mode))
1523           hstat.restval = st.st_size;
1524       /* Decide whether to send the no-cache directive.  */
1525       if (u->proxy && (count > 1 || (opt.proxy_cache == 0)))
1526         *dt |= SEND_NOCACHE;
1527       else
1528         *dt &= ~SEND_NOCACHE;
1529
1530       /* Try fetching the document, or at least its head.  :-) */
1531       err = gethttp (u, &hstat, dt);
1532
1533       /* It's unfortunate that wget determines the local filename before finding
1534          out the Content-Type of the file.  Barring a major restructuring of the
1535          code, we need to re-set locf here, since gethttp() may have xrealloc()d
1536          u->local to tack on ".html". */
1537       if (!opt.output_document)
1538         locf = u->local;
1539       else
1540         locf = opt.output_document;
1541
1542       /* In `-c' is used, check whether the file we're writing to
1543          exists before we've done anything.  If so, we'll refuse to
1544          truncate it if the server doesn't support continued
1545          downloads.  */
1546       if (opt.always_rest)
1547         hstat.no_truncate = file_exists_p (locf);
1548
1549       /* Time?  */
1550       tms = time_str (NULL);
1551       /* Get the new location (with or without the redirection).  */
1552       if (hstat.newloc)
1553         *newloc = xstrdup (hstat.newloc);
1554       switch (err)
1555         {
1556         case HERR: case HEOF: case CONSOCKERR: case CONCLOSED:
1557         case CONERROR: case READERR: case WRITEFAILED:
1558         case RANGEERR:
1559           /* Non-fatal errors continue executing the loop, which will
1560              bring them to "while" statement at the end, to judge
1561              whether the number of tries was exceeded.  */
1562           FREEHSTAT (hstat);
1563           printwhat (count, opt.ntry);
1564           continue;
1565           break;
1566         case HOSTERR: case CONREFUSED: case PROXERR: case AUTHFAILED: 
1567         case SSLERRCTXCREATE: case CONTNOTSUPPORTED:
1568           /* Fatal errors just return from the function.  */
1569           FREEHSTAT (hstat);
1570           xfree (filename_plus_orig_suffix); /* must precede every return! */
1571           return err;
1572           break;
1573         case FWRITEERR: case FOPENERR:
1574           /* Another fatal error.  */
1575           logputs (LOG_VERBOSE, "\n");
1576           logprintf (LOG_NOTQUIET, _("Cannot write to `%s' (%s).\n"),
1577                      u->local, strerror (errno));
1578           FREEHSTAT (hstat);
1579           return err;
1580           break;
1581         case CONSSLERR:
1582           /* Another fatal error.  */
1583           logputs (LOG_VERBOSE, "\n");
1584           logprintf (LOG_NOTQUIET, _("Unable to establish SSL connection.\n"));
1585           FREEHSTAT (hstat);
1586           xfree (filename_plus_orig_suffix); /* must precede every return! */
1587           return err;
1588           break;
1589         case NEWLOCATION:
1590           /* Return the new location to the caller.  */
1591           if (!hstat.newloc)
1592             {
1593               logprintf (LOG_NOTQUIET,
1594                          _("ERROR: Redirection (%d) without location.\n"),
1595                          hstat.statcode);
1596               xfree (filename_plus_orig_suffix); /* must precede every return! */
1597               return WRONGCODE;
1598             }
1599           FREEHSTAT (hstat);
1600           xfree (filename_plus_orig_suffix); /* must precede every return! */
1601           return NEWLOCATION;
1602           break;
1603         case RETRFINISHED:
1604           /* Deal with you later.  */
1605           break;
1606         default:
1607           /* All possibilities should have been exhausted.  */
1608           abort ();
1609         }
1610       if (!(*dt & RETROKF))
1611         {
1612           if (!opt.verbose)
1613             {
1614               /* #### Ugly ugly ugly! */
1615               char *hurl = str_url (u->proxy ? u->proxy : u, 1);
1616               logprintf (LOG_NONVERBOSE, "%s:\n", hurl);
1617               xfree (hurl);
1618             }
1619           logprintf (LOG_NOTQUIET, _("%s ERROR %d: %s.\n"),
1620                      tms, hstat.statcode, hstat.error);
1621           logputs (LOG_VERBOSE, "\n");
1622           FREEHSTAT (hstat);
1623           xfree (filename_plus_orig_suffix); /* must precede every return! */
1624           return WRONGCODE;
1625         }
1626
1627       /* Did we get the time-stamp?  */
1628       if (!got_head)
1629         {
1630           if (opt.timestamping && !hstat.remote_time)
1631             {
1632               logputs (LOG_NOTQUIET, _("\
1633 Last-modified header missing -- time-stamps turned off.\n"));
1634             }
1635           else if (hstat.remote_time)
1636             {
1637               /* Convert the date-string into struct tm.  */
1638               tmr = http_atotm (hstat.remote_time);
1639               if (tmr == (time_t) (-1))
1640                 logputs (LOG_VERBOSE, _("\
1641 Last-modified header invalid -- time-stamp ignored.\n"));
1642             }
1643         }
1644
1645       /* The time-stamping section.  */
1646       if (use_ts)
1647         {
1648           got_head = 1;
1649           *dt &= ~HEAD_ONLY;
1650           use_ts = 0;           /* no more time-stamping */
1651           count = 0;            /* the retrieve count for HEAD is
1652                                    reset */
1653           if (hstat.remote_time && tmr != (time_t) (-1))
1654             {
1655               /* Now time-stamping can be used validly.  Time-stamping
1656                  means that if the sizes of the local and remote file
1657                  match, and local file is newer than the remote file,
1658                  it will not be retrieved.  Otherwise, the normal
1659                  download procedure is resumed.  */
1660               if (tml >= tmr &&
1661                   (hstat.contlen == -1 || local_size == hstat.contlen))
1662                 {
1663                   logprintf (LOG_VERBOSE, _("\
1664 Server file no newer than local file `%s' -- not retrieving.\n\n"),
1665                              local_filename);
1666                   FREEHSTAT (hstat);
1667                   xfree (filename_plus_orig_suffix); /*must precede every return!*/
1668                   return RETROK;
1669                 }
1670               else if (tml >= tmr)
1671                 logprintf (LOG_VERBOSE, _("\
1672 The sizes do not match (local %ld) -- retrieving.\n"), local_size);
1673               else
1674                 logputs (LOG_VERBOSE,
1675                          _("Remote file is newer, retrieving.\n"));
1676             }
1677           FREEHSTAT (hstat);
1678           continue;
1679         }
1680       if ((tmr != (time_t) (-1))
1681           && !opt.spider
1682           && ((hstat.len == hstat.contlen) ||
1683               ((hstat.res == 0) &&
1684                ((hstat.contlen == -1) ||
1685                 (hstat.len >= hstat.contlen && !opt.kill_longer)))))
1686         {
1687           /* #### This code repeats in http.c and ftp.c.  Move it to a
1688              function!  */
1689           const char *fl = NULL;
1690           if (opt.output_document)
1691             {
1692               if (opt.od_known_regular)
1693                 fl = opt.output_document;
1694             }
1695           else
1696             fl = u->local;
1697           if (fl)
1698             touch (fl, tmr);
1699         }
1700       /* End of time-stamping section.  */
1701
1702       if (opt.spider)
1703         {
1704           logprintf (LOG_NOTQUIET, "%d %s\n\n", hstat.statcode, hstat.error);
1705           xfree (filename_plus_orig_suffix); /* must precede every return! */
1706           return RETROK;
1707         }
1708
1709       /* It is now safe to free the remainder of hstat, since the
1710          strings within it will no longer be used.  */
1711       FREEHSTAT (hstat);
1712
1713       tmrate = rate (hstat.len - hstat.restval, hstat.dltime, 0);
1714
1715       if (hstat.len == hstat.contlen)
1716         {
1717           if (*dt & RETROKF)
1718             {
1719               logprintf (LOG_VERBOSE,
1720                          _("%s (%s) - `%s' saved [%ld/%ld]\n\n"),
1721                          tms, tmrate, locf, hstat.len, hstat.contlen);
1722               logprintf (LOG_NONVERBOSE,
1723                          "%s URL:%s [%ld/%ld] -> \"%s\" [%d]\n",
1724                          tms, u->url, hstat.len, hstat.contlen, locf, count);
1725             }
1726           ++opt.numurls;
1727           downloaded_increase (hstat.len);
1728
1729           /* Remember that we downloaded the file for later ".orig" code. */
1730           if (*dt & ADDED_HTML_EXTENSION)
1731             downloaded_file(FILE_DOWNLOADED_AND_HTML_EXTENSION_ADDED, locf);
1732           else
1733             downloaded_file(FILE_DOWNLOADED_NORMALLY, locf);
1734
1735           xfree(filename_plus_orig_suffix); /* must precede every return! */
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               xfree (filename_plus_orig_suffix); /* must precede every return! */
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               xfree (filename_plus_orig_suffix); /* must precede every return! */
1791               return RETROK;
1792             }
1793           else                  /* the same, but not accepted */
1794             {
1795               logprintf (LOG_VERBOSE,
1796                          _("%s (%s) - Connection closed at byte %ld/%ld. "),
1797                          tms, tmrate, hstat.len, hstat.contlen);
1798               printwhat (count, opt.ntry);
1799               continue;
1800             }
1801         }
1802       else                      /* now hstat.res can only be -1 */
1803         {
1804           if (hstat.contlen == -1)
1805             {
1806               logprintf (LOG_VERBOSE,
1807                          _("%s (%s) - Read error at byte %ld (%s)."),
1808                          tms, tmrate, hstat.len, strerror (errno));
1809               printwhat (count, opt.ntry);
1810               continue;
1811             }
1812           else                  /* hstat.res == -1 and contlen is given */
1813             {
1814               logprintf (LOG_VERBOSE,
1815                          _("%s (%s) - Read error at byte %ld/%ld (%s). "),
1816                          tms, tmrate, hstat.len, hstat.contlen,
1817                          strerror (errno));
1818               printwhat (count, opt.ntry);
1819               continue;
1820             }
1821         }
1822       /* not reached */
1823       break;
1824     }
1825   while (!opt.ntry || (count < opt.ntry));
1826   xfree (filename_plus_orig_suffix); /* must precede every return! */
1827   return TRYLIMEXC;
1828 }
1829 \f
1830 /* Converts struct tm to time_t, assuming the data in tm is UTC rather
1831    than local timezone.
1832
1833    mktime is similar but assumes struct tm, also known as the
1834    "broken-down" form of time, is in local time zone.  mktime_from_utc
1835    uses mktime to make the conversion understanding that an offset
1836    will be introduced by the local time assumption.
1837
1838    mktime_from_utc then measures the introduced offset by applying
1839    gmtime to the initial result and applying mktime to the resulting
1840    "broken-down" form.  The difference between the two mktime results
1841    is the measured offset which is then subtracted from the initial
1842    mktime result to yield a calendar time which is the value returned.
1843
1844    tm_isdst in struct tm is set to 0 to force mktime to introduce a
1845    consistent offset (the non DST offset) since tm and tm+o might be
1846    on opposite sides of a DST change.
1847
1848    Some implementations of mktime return -1 for the nonexistent
1849    localtime hour at the beginning of DST.  In this event, use
1850    mktime(tm - 1hr) + 3600.
1851
1852    Schematically
1853      mktime(tm)   --> t+o
1854      gmtime(t+o)  --> tm+o
1855      mktime(tm+o) --> t+2o
1856      t+o - (t+2o - t+o) = t
1857
1858    Note that glibc contains a function of the same purpose named
1859    `timegm' (reverse of gmtime).  But obviously, it is not universally
1860    available, and unfortunately it is not straightforwardly
1861    extractable for use here.  Perhaps configure should detect timegm
1862    and use it where available.
1863
1864    Contributed by Roger Beeman <beeman@cisco.com>, with the help of
1865    Mark Baushke <mdb@cisco.com> and the rest of the Gurus at CISCO.
1866    Further improved by Roger with assistance from Edward J. Sabol
1867    based on input by Jamie Zawinski.  */
1868
1869 static time_t
1870 mktime_from_utc (struct tm *t)
1871 {
1872   time_t tl, tb;
1873   struct tm *tg;
1874
1875   tl = mktime (t);
1876   if (tl == -1)
1877     {
1878       t->tm_hour--;
1879       tl = mktime (t);
1880       if (tl == -1)
1881         return -1; /* can't deal with output from strptime */
1882       tl += 3600;
1883     }
1884   tg = gmtime (&tl);
1885   tg->tm_isdst = 0;
1886   tb = mktime (tg);
1887   if (tb == -1)
1888     {
1889       tg->tm_hour--;
1890       tb = mktime (tg);
1891       if (tb == -1)
1892         return -1; /* can't deal with output from gmtime */
1893       tb += 3600;
1894     }
1895   return (tl - (tb - tl));
1896 }
1897
1898 /* Check whether the result of strptime() indicates success.
1899    strptime() returns the pointer to how far it got to in the string.
1900    The processing has been successful if the string is at `GMT' or
1901    `+X', or at the end of the string.
1902
1903    In extended regexp parlance, the function returns 1 if P matches
1904    "^ *(GMT|[+-][0-9]|$)", 0 otherwise.  P being NULL (which strptime
1905    can return) is considered a failure and 0 is returned.  */
1906 static int
1907 check_end (const char *p)
1908 {
1909   if (!p)
1910     return 0;
1911   while (ISSPACE (*p))
1912     ++p;
1913   if (!*p
1914       || (p[0] == 'G' && p[1] == 'M' && p[2] == 'T')
1915       || ((p[0] == '+' || p[0] == '-') && ISDIGIT (p[1])))
1916     return 1;
1917   else
1918     return 0;
1919 }
1920
1921 /* Convert the textual specification of time in TIME_STRING to the
1922    number of seconds since the Epoch.
1923
1924    TIME_STRING can be in any of the three formats RFC2068 allows the
1925    HTTP servers to emit -- RFC1123-date, RFC850-date or asctime-date.
1926    Timezones are ignored, and should be GMT.
1927
1928    Return the computed time_t representation, or -1 if the conversion
1929    fails.
1930
1931    This function uses strptime with various string formats for parsing
1932    TIME_STRING.  This results in a parser that is not as lenient in
1933    interpreting TIME_STRING as I would like it to be.  Being based on
1934    strptime, it always allows shortened months, one-digit days, etc.,
1935    but due to the multitude of formats in which time can be
1936    represented, an ideal HTTP time parser would be even more
1937    forgiving.  It should completely ignore things like week days and
1938    concentrate only on the various forms of representing years,
1939    months, days, hours, minutes, and seconds.  For example, it would
1940    be nice if it accepted ISO 8601 out of the box.
1941
1942    I've investigated free and PD code for this purpose, but none was
1943    usable.  getdate was big and unwieldy, and had potential copyright
1944    issues, or so I was informed.  Dr. Marcus Hennecke's atotm(),
1945    distributed with phttpd, is excellent, but we cannot use it because
1946    it is not assigned to the FSF.  So I stuck it with strptime.  */
1947
1948 time_t
1949 http_atotm (char *time_string)
1950 {
1951   /* NOTE: Solaris strptime man page claims that %n and %t match white
1952      space, but that's not universally available.  Instead, we simply
1953      use ` ' to mean "skip all WS", which works under all strptime
1954      implementations I've tested.  */
1955
1956   static const char *time_formats[] = {
1957     "%a, %d %b %Y %T",          /* RFC1123: Thu, 29 Jan 1998 22:12:57 */
1958     "%A, %d-%b-%y %T",          /* RFC850:  Thursday, 29-Jan-98 22:12:57 */
1959     "%a, %d-%b-%Y %T",          /* pseudo-RFC850:  Thu, 29-Jan-1998 22:12:57
1960                                    (google.com uses this for their cookies.) */
1961     "%a %b %d %T %Y"            /* asctime: Thu Jan 29 22:12:57 1998 */
1962   };
1963
1964   int i;
1965   struct tm t;
1966
1967   /* According to Roger Beeman, we need to initialize tm_isdst, since
1968      strptime won't do it.  */
1969   t.tm_isdst = 0;
1970
1971   /* Note that under foreign locales Solaris strptime() fails to
1972      recognize English dates, which renders this function useless.  We
1973      solve this by being careful not to affect LC_TIME when
1974      initializing locale.
1975
1976      Another solution would be to temporarily set locale to C, invoke
1977      strptime(), and restore it back.  This is slow and dirty,
1978      however, and locale support other than LC_MESSAGES can mess other
1979      things, so I rather chose to stick with just setting LC_MESSAGES.
1980
1981      GNU strptime does not have this problem because it recognizes
1982      both international and local dates.  */
1983
1984   for (i = 0; i < ARRAY_SIZE (time_formats); i++)
1985     if (check_end (strptime (time_string, time_formats[i], &t)))
1986       return mktime_from_utc (&t);
1987
1988   /* All formats have failed.  */
1989   return -1;
1990 }
1991 \f
1992 /* Authorization support: We support two authorization schemes:
1993
1994    * `Basic' scheme, consisting of base64-ing USER:PASSWORD string;
1995
1996    * `Digest' scheme, added by Junio Hamano <junio@twinsun.com>,
1997    consisting of answering to the server's challenge with the proper
1998    MD5 digests.  */
1999
2000 /* How many bytes it will take to store LEN bytes in base64.  */
2001 #define BASE64_LENGTH(len) (4 * (((len) + 2) / 3))
2002
2003 /* Encode the string S of length LENGTH to base64 format and place it
2004    to STORE.  STORE will be 0-terminated, and must point to a writable
2005    buffer of at least 1+BASE64_LENGTH(length) bytes.  */
2006 static void
2007 base64_encode (const char *s, char *store, int length)
2008 {
2009   /* Conversion table.  */
2010   static char tbl[64] = {
2011     'A','B','C','D','E','F','G','H',
2012     'I','J','K','L','M','N','O','P',
2013     'Q','R','S','T','U','V','W','X',
2014     'Y','Z','a','b','c','d','e','f',
2015     'g','h','i','j','k','l','m','n',
2016     'o','p','q','r','s','t','u','v',
2017     'w','x','y','z','0','1','2','3',
2018     '4','5','6','7','8','9','+','/'
2019   };
2020   int i;
2021   unsigned char *p = (unsigned char *)store;
2022
2023   /* Transform the 3x8 bits to 4x6 bits, as required by base64.  */
2024   for (i = 0; i < length; i += 3)
2025     {
2026       *p++ = tbl[s[0] >> 2];
2027       *p++ = tbl[((s[0] & 3) << 4) + (s[1] >> 4)];
2028       *p++ = tbl[((s[1] & 0xf) << 2) + (s[2] >> 6)];
2029       *p++ = tbl[s[2] & 0x3f];
2030       s += 3;
2031     }
2032   /* Pad the result if necessary...  */
2033   if (i == length + 1)
2034     *(p - 1) = '=';
2035   else if (i == length + 2)
2036     *(p - 1) = *(p - 2) = '=';
2037   /* ...and zero-terminate it.  */
2038   *p = '\0';
2039 }
2040
2041 /* Create the authentication header contents for the `Basic' scheme.
2042    This is done by encoding the string `USER:PASS' in base64 and
2043    prepending `HEADER: Basic ' to it.  */
2044 static char *
2045 basic_authentication_encode (const char *user, const char *passwd,
2046                              const char *header)
2047 {
2048   char *t1, *t2, *res;
2049   int len1 = strlen (user) + 1 + strlen (passwd);
2050   int len2 = BASE64_LENGTH (len1);
2051
2052   t1 = (char *)alloca (len1 + 1);
2053   sprintf (t1, "%s:%s", user, passwd);
2054   t2 = (char *)alloca (1 + len2);
2055   base64_encode (t1, t2, len1);
2056   res = (char *)xmalloc (len2 + 11 + strlen (header));
2057   sprintf (res, "%s: Basic %s\r\n", header, t2);
2058
2059   return res;
2060 }
2061
2062 #ifdef USE_DIGEST
2063 /* Parse HTTP `WWW-Authenticate:' header.  AU points to the beginning
2064    of a field in such a header.  If the field is the one specified by
2065    ATTR_NAME ("realm", "opaque", and "nonce" are used by the current
2066    digest authorization code), extract its value in the (char*)
2067    variable pointed by RET.  Returns negative on a malformed header,
2068    or number of bytes that have been parsed by this call.  */
2069 static int
2070 extract_header_attr (const char *au, const char *attr_name, char **ret)
2071 {
2072   const char *cp, *ep;
2073
2074   ep = cp = au;
2075
2076   if (strncmp (cp, attr_name, strlen (attr_name)) == 0)
2077     {
2078       cp += strlen (attr_name);
2079       if (!*cp)
2080         return -1;
2081       cp += skip_lws (cp);
2082       if (*cp != '=')
2083         return -1;
2084       if (!*++cp)
2085         return -1;
2086       cp += skip_lws (cp);
2087       if (*cp != '\"')
2088         return -1;
2089       if (!*++cp)
2090         return -1;
2091       for (ep = cp; *ep && *ep != '\"'; ep++)
2092         ;
2093       if (!*ep)
2094         return -1;
2095       FREE_MAYBE (*ret);
2096       *ret = strdupdelim (cp, ep);
2097       return ep - au + 1;
2098     }
2099   else
2100     return 0;
2101 }
2102
2103 /* Dump the hexadecimal representation of HASH to BUF.  HASH should be
2104    an array of 16 bytes containing the hash keys, and BUF should be a
2105    buffer of 33 writable characters (32 for hex digits plus one for
2106    zero termination).  */
2107 static void
2108 dump_hash (unsigned char *buf, const unsigned char *hash)
2109 {
2110   int i;
2111
2112   for (i = 0; i < MD5_HASHLEN; i++, hash++)
2113     {
2114       *buf++ = XDIGIT_TO_xchar (*hash >> 4);
2115       *buf++ = XDIGIT_TO_xchar (*hash & 0xf);
2116     }
2117   *buf = '\0';
2118 }
2119
2120 /* Take the line apart to find the challenge, and compose a digest
2121    authorization header.  See RFC2069 section 2.1.2.  */
2122 char *
2123 digest_authentication_encode (const char *au, const char *user,
2124                               const char *passwd, const char *method,
2125                               const char *path)
2126 {
2127   static char *realm, *opaque, *nonce;
2128   static struct {
2129     const char *name;
2130     char **variable;
2131   } options[] = {
2132     { "realm", &realm },
2133     { "opaque", &opaque },
2134     { "nonce", &nonce }
2135   };
2136   char *res;
2137
2138   realm = opaque = nonce = NULL;
2139
2140   au += 6;                      /* skip over `Digest' */
2141   while (*au)
2142     {
2143       int i;
2144
2145       au += skip_lws (au);
2146       for (i = 0; i < ARRAY_SIZE (options); i++)
2147         {
2148           int skip = extract_header_attr (au, options[i].name,
2149                                           options[i].variable);
2150           if (skip < 0)
2151             {
2152               FREE_MAYBE (realm);
2153               FREE_MAYBE (opaque);
2154               FREE_MAYBE (nonce);
2155               return NULL;
2156             }
2157           else if (skip)
2158             {
2159               au += skip;
2160               break;
2161             }
2162         }
2163       if (i == ARRAY_SIZE (options))
2164         {
2165           while (*au && *au != '=')
2166             au++;
2167           if (*au && *++au)
2168             {
2169               au += skip_lws (au);
2170               if (*au == '\"')
2171                 {
2172                   au++;
2173                   while (*au && *au != '\"')
2174                     au++;
2175                   if (*au)
2176                     au++;
2177                 }
2178             }
2179         }
2180       while (*au && *au != ',')
2181         au++;
2182       if (*au)
2183         au++;
2184     }
2185   if (!realm || !nonce || !user || !passwd || !path || !method)
2186     {
2187       FREE_MAYBE (realm);
2188       FREE_MAYBE (opaque);
2189       FREE_MAYBE (nonce);
2190       return NULL;
2191     }
2192
2193   /* Calculate the digest value.  */
2194   {
2195     struct md5_ctx ctx;
2196     unsigned char hash[MD5_HASHLEN];
2197     unsigned char a1buf[MD5_HASHLEN * 2 + 1], a2buf[MD5_HASHLEN * 2 + 1];
2198     unsigned char response_digest[MD5_HASHLEN * 2 + 1];
2199
2200     /* A1BUF = H(user ":" realm ":" password) */
2201     md5_init_ctx (&ctx);
2202     md5_process_bytes (user, strlen (user), &ctx);
2203     md5_process_bytes (":", 1, &ctx);
2204     md5_process_bytes (realm, strlen (realm), &ctx);
2205     md5_process_bytes (":", 1, &ctx);
2206     md5_process_bytes (passwd, strlen (passwd), &ctx);
2207     md5_finish_ctx (&ctx, hash);
2208     dump_hash (a1buf, hash);
2209
2210     /* A2BUF = H(method ":" path) */
2211     md5_init_ctx (&ctx);
2212     md5_process_bytes (method, strlen (method), &ctx);
2213     md5_process_bytes (":", 1, &ctx);
2214     md5_process_bytes (path, strlen (path), &ctx);
2215     md5_finish_ctx (&ctx, hash);
2216     dump_hash (a2buf, hash);
2217
2218     /* RESPONSE_DIGEST = H(A1BUF ":" nonce ":" A2BUF) */
2219     md5_init_ctx (&ctx);
2220     md5_process_bytes (a1buf, MD5_HASHLEN * 2, &ctx);
2221     md5_process_bytes (":", 1, &ctx);
2222     md5_process_bytes (nonce, strlen (nonce), &ctx);
2223     md5_process_bytes (":", 1, &ctx);
2224     md5_process_bytes (a2buf, MD5_HASHLEN * 2, &ctx);
2225     md5_finish_ctx (&ctx, hash);
2226     dump_hash (response_digest, hash);
2227
2228     res = (char*) xmalloc (strlen (user)
2229                            + strlen (user)
2230                            + strlen (realm)
2231                            + strlen (nonce)
2232                            + strlen (path)
2233                            + 2 * MD5_HASHLEN /*strlen (response_digest)*/
2234                            + (opaque ? strlen (opaque) : 0)
2235                            + 128);
2236     sprintf (res, "Authorization: Digest \
2237 username=\"%s\", realm=\"%s\", nonce=\"%s\", uri=\"%s\", response=\"%s\"",
2238              user, realm, nonce, path, response_digest);
2239     if (opaque)
2240       {
2241         char *p = res + strlen (res);
2242         strcat (p, ", opaque=\"");
2243         strcat (p, opaque);
2244         strcat (p, "\"");
2245       }
2246     strcat (res, "\r\n");
2247   }
2248   return res;
2249 }
2250 #endif /* USE_DIGEST */
2251
2252
2253 #define BEGINS_WITH(line, string_constant)                              \
2254   (!strncasecmp (line, string_constant, sizeof (string_constant) - 1)   \
2255    && (ISSPACE (line[sizeof (string_constant) - 1])                     \
2256        || !line[sizeof (string_constant) - 1]))
2257
2258 static int
2259 known_authentication_scheme_p (const char *au)
2260 {
2261   return BEGINS_WITH (au, "Basic")
2262     || BEGINS_WITH (au, "Digest")
2263     || BEGINS_WITH (au, "NTLM");
2264 }
2265
2266 #undef BEGINS_WITH
2267
2268 /* Create the HTTP authorization request header.  When the
2269    `WWW-Authenticate' response header is seen, according to the
2270    authorization scheme specified in that header (`Basic' and `Digest'
2271    are supported by the current implementation), produce an
2272    appropriate HTTP authorization request header.  */
2273 static char *
2274 create_authorization_line (const char *au, const char *user,
2275                            const char *passwd, const char *method,
2276                            const char *path)
2277 {
2278   char *wwwauth = NULL;
2279
2280   if (!strncasecmp (au, "Basic", 5))
2281     wwwauth = basic_authentication_encode (user, passwd, "Authorization");
2282   if (!strncasecmp (au, "NTLM", 4))
2283     wwwauth = basic_authentication_encode (user, passwd, "Authorization");
2284 #ifdef USE_DIGEST
2285   else if (!strncasecmp (au, "Digest", 6))
2286     wwwauth = digest_authentication_encode (au, user, passwd, method, path);
2287 #endif /* USE_DIGEST */
2288   return wwwauth;
2289 }