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