]> sjero.net Git - wget/blob - src/http.c
8373133b23617f036504afa50b0ca3305de74fe0
[wget] / src / http.c
1 /* HTTP support.
2    Copyright (C) 1995, 1996, 1997, 1998 Free Software Foundation, Inc.
3
4 This file is part of Wget.
5
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 2 of the License, or
9 (at your option) any later version.
10
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 GNU General Public License for more details.
15
16 You should have received a copy of the GNU General Public License
17 along with this program; if not, write to the Free Software
18 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
19
20 #include <config.h>
21
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <sys/types.h>
25 #ifdef HAVE_STRING_H
26 # include <string.h>
27 #else
28 # include <strings.h>
29 #endif
30 #include <ctype.h>
31 #ifdef HAVE_UNISTD_H
32 # include <unistd.h>
33 #endif
34 #include <assert.h>
35 #include <errno.h>
36 #if TIME_WITH_SYS_TIME
37 # include <sys/time.h>
38 # include <time.h>
39 #else
40 # if HAVE_SYS_TIME_H
41 #  include <sys/time.h>
42 # else
43 #  include <time.h>
44 # endif
45 #endif
46
47 #ifdef WINDOWS
48 # include <winsock.h>
49 #endif
50
51 #include "wget.h"
52 #include "utils.h"
53 #include "url.h"
54 #include "host.h"
55 #include "rbuf.h"
56 #include "retr.h"
57 #include "headers.h"
58 #include "connect.h"
59 #include "fnmatch.h"
60 #include "netrc.h"
61 #if USE_DIGEST
62 # include "md5.h"
63 #endif
64
65 extern char *version_string;
66
67 #ifndef errno
68 extern int errno;
69 #endif
70 #ifndef h_errno
71 extern int h_errno;
72 #endif
73 \f
74
75 #define TEXTHTML_S "text/html"
76 #define HTTP_ACCEPT "*/*"
77
78 /* Some status code validation macros: */
79 #define H_20X(x)        (((x) >= 200) && ((x) < 300))
80 #define H_PARTIAL(x)    ((x) == HTTP_STATUS_PARTIAL_CONTENTS)
81 #define H_REDIRECTED(x) (((x) == HTTP_STATUS_MOVED_PERMANENTLY) \
82                          || ((x) == HTTP_STATUS_MOVED_TEMPORARILY))
83
84 /* HTTP/1.0 status codes from RFC1945, provided for reference.  */
85 /* Successful 2xx.  */
86 #define HTTP_STATUS_OK                  200
87 #define HTTP_STATUS_CREATED             201
88 #define HTTP_STATUS_ACCEPTED            202
89 #define HTTP_STATUS_NO_CONTENT          204
90 #define HTTP_STATUS_PARTIAL_CONTENTS    206
91
92 /* Redirection 3xx.  */
93 #define HTTP_STATUS_MULTIPLE_CHOICES    300
94 #define HTTP_STATUS_MOVED_PERMANENTLY   301
95 #define HTTP_STATUS_MOVED_TEMPORARILY   302
96 #define HTTP_STATUS_NOT_MODIFIED        304
97
98 /* Client error 4xx.  */
99 #define HTTP_STATUS_BAD_REQUEST         400
100 #define HTTP_STATUS_UNAUTHORIZED        401
101 #define HTTP_STATUS_FORBIDDEN           403
102 #define HTTP_STATUS_NOT_FOUND           404
103
104 /* Server errors 5xx.  */
105 #define HTTP_STATUS_INTERNAL            500
106 #define HTTP_STATUS_NOT_IMPLEMENTED     501
107 #define HTTP_STATUS_BAD_GATEWAY         502
108 #define HTTP_STATUS_UNAVAILABLE         503
109
110 \f
111 /* Parse the HTTP status line, which is of format:
112
113    HTTP-Version SP Status-Code SP Reason-Phrase
114
115    The function returns the status-code, or -1 if the status line is
116    malformed.  The pointer to reason-phrase is returned in RP.  */
117 static int
118 parse_http_status_line (const char *line, const char **reason_phrase_ptr)
119 {
120   /* (the variables must not be named `major' and `minor', because
121      that breaks compilation with SunOS4 cc.)  */
122   int mjr, mnr, statcode;
123   const char *p;
124
125   *reason_phrase_ptr = NULL;
126
127   /* The standard format of HTTP-Version is: `HTTP/X.Y', where X is
128      major version, and Y is minor version.  */
129   if (strncmp (line, "HTTP/", 5) != 0)
130     return -1;
131   line += 5;
132
133   /* Calculate major HTTP version.  */
134   p = line;
135   for (mjr = 0; ISDIGIT (*line); line++)
136     mjr = 10 * mjr + (*line - '0');
137   if (*line != '.' || p == line)
138     return -1;
139   ++line;
140
141   /* Calculate minor HTTP version.  */
142   p = line;
143   for (mnr = 0; ISDIGIT (*line); line++)
144     mnr = 10 * mnr + (*line - '0');
145   if (*line != ' ' || p == line)
146     return -1;
147   /* Wget will accept only 1.0 and higher HTTP-versions.  The value of
148      minor version can be safely ignored.  */
149   if (mjr < 1)
150     return -1;
151   ++line;
152
153   /* Calculate status code.  */
154   if (!(ISDIGIT (*line) && ISDIGIT (line[1]) && ISDIGIT (line[2])))
155     return -1;
156   statcode = 100 * (*line - '0') + 10 * (line[1] - '0') + (line[2] - '0');
157
158   /* Set up the reason phrase pointer.  */
159   line += 3;
160   /* RFC2068 requires SPC here, but we allow the string to finish
161      here, in case no reason-phrase is present.  */
162   if (*line != ' ')
163     {
164       if (!*line)
165         *reason_phrase_ptr = line;
166       else
167         return -1;
168     }
169   else
170     *reason_phrase_ptr = line + 1;
171
172   return statcode;
173 }
174 \f
175 /* Functions to be used as arguments to header_process(): */
176
177 struct http_process_range_closure {
178   long first_byte_pos;
179   long last_byte_pos;
180   long entity_length;
181 };
182
183 /* Parse the `Content-Range' header and extract the information it
184    contains.  Returns 1 if successful, -1 otherwise.  */
185 static int
186 http_process_range (const char *hdr, void *arg)
187 {
188   struct http_process_range_closure *closure
189     = (struct http_process_range_closure *)arg;
190   long num;
191
192   /* Certain versions of Nutscape proxy server send out
193      `Content-Length' without "bytes" specifier, which is a breach of
194      RFC2068 (as well as the HTTP/1.1 draft which was current at the
195      time).  But hell, I must support it...  */
196   if (!strncasecmp (hdr, "bytes", 5))
197     {
198       hdr += 5;
199       hdr += skip_lws (hdr);
200       if (!*hdr)
201         return 0;
202     }
203   if (!ISDIGIT (*hdr))
204     return 0;
205   for (num = 0; ISDIGIT (*hdr); hdr++)
206     num = 10 * num + (*hdr - '0');
207   if (*hdr != '-' || !ISDIGIT (*(hdr + 1)))
208     return 0;
209   closure->first_byte_pos = num;
210   ++hdr;
211   for (num = 0; ISDIGIT (*hdr); hdr++)
212     num = 10 * num + (*hdr - '0');
213   if (*hdr != '/' || !ISDIGIT (*(hdr + 1)))
214     return 0;
215   closure->last_byte_pos = num;
216   ++hdr;
217   for (num = 0; ISDIGIT (*hdr); hdr++)
218     num = 10 * num + (*hdr - '0');
219   closure->entity_length = num;
220   return 1;
221 }
222
223 /* Place 1 to ARG if the HDR contains the word "none", 0 otherwise.
224    Used for `Accept-Ranges'.  */
225 static int
226 http_process_none (const char *hdr, void *arg)
227 {
228   int *where = (int *)arg;
229
230   if (strstr (hdr, "none"))
231     *where = 1;
232   else
233     *where = 0;
234   return 1;
235 }
236
237 /* Place the malloc-ed copy of HDR hdr, to the first `;' to ARG.  */
238 static int
239 http_process_type (const char *hdr, void *arg)
240 {
241   char **result = (char **)arg;
242   char *p;
243
244   *result = xstrdup (hdr);
245   p = strrchr (hdr, ';');
246   if (p)
247     {
248       int len = p - hdr;
249       *result = (char *)xmalloc (len + 1);
250       memcpy (*result, hdr, len);
251       (*result)[len] = '\0';
252     }
253   else
254     *result = xstrdup (hdr);
255   return 1;
256 }
257
258 \f
259 struct http_stat
260 {
261   long len;                     /* received length */
262   long contlen;                 /* expected length */
263   long restval;                 /* the restart value */
264   int res;                      /* the result of last read */
265   char *newloc;                 /* new location (redirection) */
266   char *remote_time;            /* remote time-stamp string */
267   char *error;                  /* textual HTTP error */
268   int statcode;                 /* status code */
269   long dltime;                  /* time of the download */
270 };
271
272 /* Free the elements of hstat X.  */
273 #define FREEHSTAT(x) do                                 \
274 {                                                       \
275   FREE_MAYBE ((x).newloc);                              \
276   FREE_MAYBE ((x).remote_time);                         \
277   FREE_MAYBE ((x).error);                               \
278   (x).newloc = (x).remote_time = (x).error = NULL;      \
279 } while (0)
280
281 static char *create_authorization_line PARAMS ((const char *, const char *,
282                                                 const char *, const char *,
283                                                 const char *));
284 static char *basic_authentication_encode PARAMS ((const char *, const char *,
285                                                   const char *));
286 static int known_authentication_scheme_p PARAMS ((const char *));
287
288 static time_t http_atotm PARAMS ((char *));
289
290 /* Retrieve a document through HTTP protocol.  It recognizes status
291    code, and correctly handles redirections.  It closes the network
292    socket.  If it receives an error from the functions below it, it
293    will print it if there is enough information to do so (almost
294    always), returning the error to the caller (i.e. http_loop).
295
296    Various HTTP parameters are stored to hs.  Although it parses the
297    response code correctly, it is not used in a sane way.  The caller
298    can do that, though.
299
300    If u->proxy is non-NULL, the URL u will be taken as a proxy URL,
301    and u->proxy->url will be given to the proxy server (bad naming,
302    I'm afraid).  */
303 static uerr_t
304 gethttp (struct urlinfo *u, struct http_stat *hs, int *dt)
305 {
306   char *request, *type, *command, *path, *qstring;
307   char *user, *passwd;
308   char *pragma_h, *referer, *useragent, *range, *wwwauth, *remhost;
309   char *authenticate_h;
310   char *proxyauth;
311   char *all_headers;
312   char *host_port;
313   int host_port_len;
314   int sock, hcount, num_written, all_length, remport, statcode;
315   long contlen, contrange;
316   struct urlinfo *ou;
317   uerr_t err;
318   FILE *fp;
319   int auth_tried_already;
320   struct rbuf rbuf;
321
322   /* Let the others worry about local filename...  */
323   if (!(*dt & HEAD_ONLY))
324     assert (u->local != NULL);
325
326   authenticate_h = 0;
327   auth_tried_already = 0;
328
329  again:
330   /* We need to come back here when the initial attempt to retrieve
331      without authorization header fails.  */
332
333   /* Initialize certain elements of struct hstat.  */
334   hs->len = 0L;
335   hs->contlen = -1;
336   hs->res = -1;
337   hs->newloc = NULL;
338   hs->remote_time = NULL;
339   hs->error = NULL;
340
341   /* Which structure to use to retrieve the original URL data.  */
342   if (u->proxy)
343     ou = u->proxy;
344   else
345     ou = u;
346
347   /* First: establish the connection.  */
348   logprintf (LOG_VERBOSE, _("Connecting to %s:%hu... "), u->host, u->port);
349   err = make_connection (&sock, u->host, u->port);
350   switch (err)
351     {
352     case HOSTERR:
353       logputs (LOG_VERBOSE, "\n");
354       logprintf (LOG_NOTQUIET, "%s: %s.\n", u->host, herrmsg (h_errno));
355       return HOSTERR;
356       break;
357     case CONSOCKERR:
358       logputs (LOG_VERBOSE, "\n");
359       logprintf (LOG_NOTQUIET, "socket: %s\n", strerror (errno));
360       return CONSOCKERR;
361       break;
362     case CONREFUSED:
363       logputs (LOG_VERBOSE, "\n");
364       logprintf (LOG_NOTQUIET,
365                  _("Connection to %s:%hu refused.\n"), u->host, u->port);
366       CLOSE (sock);
367       return CONREFUSED;
368     case CONERROR:
369       logputs (LOG_VERBOSE, "\n");
370       logprintf (LOG_NOTQUIET, "connect: %s\n", strerror (errno));
371       CLOSE (sock);
372       return CONERROR;
373       break;
374     case NOCONERROR:
375       /* Everything is fine!  */
376       logputs (LOG_VERBOSE, _("connected!\n"));
377       break;
378     default:
379       abort ();
380       break;
381     } /* switch */
382
383   if (u->proxy)
384     path = u->proxy->url;
385   else
386     path = u->path;
387   
388   qstring = u->qstring;
389
390   command = (*dt & HEAD_ONLY) ? "HEAD" : "GET";
391   referer = NULL;
392   if (ou->referer)
393     {
394       referer = (char *)alloca (9 + strlen (ou->referer) + 3);
395       sprintf (referer, "Referer: %s\r\n", ou->referer);
396     }
397   if (*dt & SEND_NOCACHE)
398     pragma_h = "Pragma: no-cache\r\n";
399   else
400     pragma_h = "";
401   if (hs->restval)
402     {
403       range = (char *)alloca (13 + numdigit (hs->restval) + 4);
404       /* #### Gag me!  Some servers (e.g. WebSitePro) have been known
405          to misinterpret the following `Range' format, and return the
406          document as multipart/x-byte-ranges MIME type!
407
408          #### TODO: Interpret MIME types, recognize bullshits similar
409          the one described above, and deal with them!  */
410       sprintf (range, "Range: bytes=%ld-\r\n", hs->restval);
411     }
412   else
413     range = NULL;
414   if (opt.useragent)
415     STRDUP_ALLOCA (useragent, opt.useragent);
416   else
417     {
418       useragent = (char *)alloca (10 + strlen (version_string));
419       sprintf (useragent, "Wget/%s", version_string);
420     }
421   /* Construct the authentication, if userid is present.  */
422   user = ou->user;
423   passwd = ou->passwd;
424   search_netrc (u->host, (const char **)&user, (const char **)&passwd, 0);
425   user = user ? user : opt.http_user;
426   passwd = passwd ? passwd : opt.http_passwd;
427
428   wwwauth = NULL;
429   if (authenticate_h && user && passwd)
430     {
431       wwwauth = create_authorization_line (authenticate_h, user, passwd,
432                                            command, ou->path);
433     }
434
435   proxyauth = NULL;
436   if (u->proxy)
437     {
438       char *proxy_user, *proxy_passwd;
439       /* For normal username and password, URL components override
440          command-line/wgetrc parameters.  With proxy authentication,
441          it's the reverse, because proxy URLs are normally the
442          "permanent" ones, so command-line args should take
443          precedence.  */
444       if (opt.proxy_user && opt.proxy_passwd)
445         {
446           proxy_user = opt.proxy_user;
447           proxy_passwd = opt.proxy_passwd;
448         }
449       else
450         {
451           proxy_user = u->user;
452           proxy_passwd = u->passwd;
453         }
454       /* #### This is junky.  Can't the proxy request, say, `Digest'
455          authentication?  */
456       if (proxy_user && proxy_passwd)
457         proxyauth = basic_authentication_encode (proxy_user, proxy_passwd,
458                                                  "Proxy-Authorization");
459     }
460   remhost = ou->host;
461   remport = ou->port;
462
463   if (remport == 80) {
464       host_port = NULL; host_port_len = 0;
465   }
466   else {
467       host_port = (char *)alloca (numdigit (remport) + 2);
468       host_port_len = sprintf (host_port, ":%d", remport);
469   }
470
471   /* Allocate the memory for the request.  */
472   request = (char *)alloca (strlen (command) + strlen (path)
473                             + (qstring ? strlen (qstring) : 0)
474                             + strlen (useragent)
475                             + strlen (remhost) + host_port_len
476                             + strlen (HTTP_ACCEPT)
477                             + (referer ? strlen (referer) : 0)
478                             + (wwwauth ? strlen (wwwauth) : 0)
479                             + (proxyauth ? strlen (proxyauth) : 0)
480                             + (range ? strlen (range) : 0)
481                             + strlen (pragma_h)
482                             + (opt.user_header ? strlen (opt.user_header) : 0)
483                             + 64);
484   /* Construct the request.  */
485   sprintf (request, "\
486 %s %s%s HTTP/1.0\r\n\
487 User-Agent: %s\r\n\
488 Host: %s%s\r\n\
489 Accept: %s\r\n\
490 %s%s%s%s%s%s\r\n",
491           command, path, qstring, useragent, remhost, host_port ? host_port : "",
492           HTTP_ACCEPT, referer ? referer : "",
493           wwwauth ? wwwauth : "", 
494           proxyauth ? proxyauth : "", 
495           range ? range : "",
496           pragma_h, 
497           opt.user_header ? opt.user_header : "");
498   DEBUGP (("---request begin---\n%s---request end---\n", request));
499    /* Free the temporary memory.  */
500   FREE_MAYBE (wwwauth);
501   FREE_MAYBE (proxyauth);
502
503   /* Send the request to server.  */
504   num_written = iwrite (sock, request, strlen (request));
505   if (num_written < 0)
506     {
507       logputs (LOG_VERBOSE, _("Failed writing HTTP request.\n"));
508       free (request);
509       CLOSE (sock);
510       return WRITEFAILED;
511     }
512   logprintf (LOG_VERBOSE, _("%s request sent, awaiting response... "),
513              u->proxy ? "Proxy" : "HTTP");
514   contlen = contrange = -1;
515   type = NULL;
516   statcode = -1;
517   *dt &= ~RETROKF;
518
519   /* Before reading anything, initialize the rbuf.  */
520   rbuf_initialize (&rbuf, sock);
521
522   all_headers = NULL;
523   all_length = 0;
524   /* Header-fetching loop.  */
525   hcount = 0;
526   while (1)
527     {
528       char *hdr;
529       int status;
530
531       ++hcount;
532       /* Get the header.  */
533       status = header_get (&rbuf, &hdr,
534                            /* Disallow continuations for status line.  */
535                            (hcount == 1 ? HG_NO_CONTINUATIONS : HG_NONE));
536
537       /* Check for errors.  */
538       if (status == HG_EOF && *hdr)
539         {
540           /* This used to be an unconditional error, but that was
541              somewhat controversial, because of a large number of
542              broken CGI's that happily "forget" to send the second EOL
543              before closing the connection of a HEAD request.
544
545              So, the deal is to check whether the header is empty
546              (*hdr is zero if it is); if yes, it means that the
547              previous header was fully retrieved, and that -- most
548              probably -- the request is complete.  "...be liberal in
549              what you accept."  Oh boy.  */
550           logputs (LOG_VERBOSE, "\n");
551           logputs (LOG_NOTQUIET, _("End of file while parsing headers.\n"));
552           free (hdr);
553           FREE_MAYBE (type);
554           FREE_MAYBE (hs->newloc);
555           FREE_MAYBE (all_headers);
556           CLOSE (sock);
557           return HEOF;
558         }
559       else if (status == HG_ERROR)
560         {
561           logputs (LOG_VERBOSE, "\n");
562           logprintf (LOG_NOTQUIET, _("Read error (%s) in headers.\n"),
563                      strerror (errno));
564           free (hdr);
565           FREE_MAYBE (type);
566           FREE_MAYBE (hs->newloc);
567           FREE_MAYBE (all_headers);
568           CLOSE (sock);
569           return HERR;
570         }
571
572       /* If the headers are to be saved to a file later, save them to
573          memory now.  */
574       if (opt.save_headers)
575         {
576           int lh = strlen (hdr);
577           all_headers = (char *)xrealloc (all_headers, all_length + lh + 2);
578           memcpy (all_headers + all_length, hdr, lh);
579           all_length += lh;
580           all_headers[all_length++] = '\n';
581           all_headers[all_length] = '\0';
582         }
583
584       /* Print the header if requested.  */
585       if (opt.server_response && hcount != 1)
586         logprintf (LOG_VERBOSE, "\n%d %s", hcount, hdr);
587
588       /* Check for status line.  */
589       if (hcount == 1)
590         {
591           const char *error;
592           /* Parse the first line of server response.  */
593           statcode = parse_http_status_line (hdr, &error);
594           hs->statcode = statcode;
595           /* Store the descriptive response.  */
596           if (statcode == -1) /* malformed response */
597             {
598               /* A common reason for "malformed response" error is the
599                  case when no data was actually received.  Handle this
600                  special case.  */
601               if (!*hdr)
602                 hs->error = xstrdup (_("No data received"));
603               else
604                 hs->error = xstrdup (_("Malformed status line"));
605               free (hdr);
606               break;
607             }
608           else if (!*error)
609             hs->error = xstrdup (_("(no description)"));
610           else
611             hs->error = xstrdup (error);
612
613           if ((statcode != -1)
614 #ifdef DEBUG
615               && !opt.debug
616 #endif
617               )
618             logprintf (LOG_VERBOSE, "%d %s", statcode, error);
619
620           goto done_header;
621         }
622
623       /* Exit on empty header.  */
624       if (!*hdr)
625         {
626           free (hdr);
627           break;
628         }
629
630       /* Try getting content-length.  */
631       if (contlen == -1 && !opt.ignore_length)
632         if (header_process (hdr, "Content-Length", header_extract_number,
633                             &contlen))
634           goto done_header;
635       /* Try getting content-type.  */
636       if (!type)
637         if (header_process (hdr, "Content-Type", http_process_type, &type))
638           goto done_header;
639       /* Try getting location.  */
640       if (!hs->newloc)
641         if (header_process (hdr, "Location", header_strdup, &hs->newloc))
642           goto done_header;
643       /* Try getting last-modified.  */
644       if (!hs->remote_time)
645         if (header_process (hdr, "Last-Modified", header_strdup,
646                             &hs->remote_time))
647           goto done_header;
648       /* Try getting www-authentication.  */
649       if (!authenticate_h)
650         if (header_process (hdr, "WWW-Authenticate", header_strdup,
651                             &authenticate_h))
652           goto done_header;
653       /* Check for accept-ranges header.  If it contains the word
654          `none', disable the ranges.  */
655       if (*dt & ACCEPTRANGES)
656         {
657           int nonep;
658           if (header_process (hdr, "Accept-Ranges", http_process_none, &nonep))
659             {
660               if (nonep)
661                 *dt &= ~ACCEPTRANGES;
662               goto done_header;
663             }
664         }
665       /* Try getting content-range.  */
666       if (contrange == -1)
667         {
668           struct http_process_range_closure closure;
669           if (header_process (hdr, "Content-Range", http_process_range, &closure))
670             {
671               contrange = closure.first_byte_pos;
672               goto done_header;
673             }
674         }
675     done_header:
676       free (hdr);
677     }
678
679   logputs (LOG_VERBOSE, "\n");
680
681   if ((statcode == HTTP_STATUS_UNAUTHORIZED)
682       && authenticate_h)
683     {
684       /* Authorization is required.  */
685       FREE_MAYBE (type);
686       type = NULL;
687       FREEHSTAT (*hs);
688       CLOSE (sock);
689       if (auth_tried_already)
690         {
691           /* If we have tried it already, then there is not point
692              retrying it.  */
693           logputs (LOG_NOTQUIET, _("Authorization failed.\n"));
694           free (authenticate_h);
695           return AUTHFAILED;
696         }
697       else if (!known_authentication_scheme_p (authenticate_h))
698         {
699           free (authenticate_h);
700           logputs (LOG_NOTQUIET, _("Unknown authentication scheme.\n"));
701           return AUTHFAILED;
702         }
703       else
704         {
705           auth_tried_already = 1;
706           goto again;
707         }
708     }
709   /* We do not need this anymore.  */
710   if (authenticate_h)
711     {
712       free (authenticate_h);
713       authenticate_h = NULL;
714     }
715
716   /* 20x responses are counted among successful by default.  */
717   if (H_20X (statcode))
718     *dt |= RETROKF;
719
720   if (type && !strncasecmp (type, TEXTHTML_S, strlen (TEXTHTML_S)))
721     *dt |= TEXTHTML;
722   else
723     /* We don't assume text/html by default.  */
724     *dt &= ~TEXTHTML;
725
726   if (contrange == -1)
727     hs->restval = 0;
728   else if (contrange != hs->restval ||
729            (H_PARTIAL (statcode) && contrange == -1))
730     {
731       /* This means the whole request was somehow misunderstood by the
732          server.  Bail out.  */
733       FREE_MAYBE (type);
734       FREE_MAYBE (hs->newloc);
735       FREE_MAYBE (all_headers);
736       CLOSE (sock);
737       return RANGEERR;
738     }
739
740   if (hs->restval)
741     {
742       if (contlen != -1)
743         contlen += contrange;
744       else
745         contrange = -1;        /* If conent-length was not sent,
746                                   content-range will be ignored.  */
747     }
748   hs->contlen = contlen;
749
750   /* Return if redirected.  */
751   if (H_REDIRECTED (statcode) || statcode == HTTP_STATUS_MULTIPLE_CHOICES)
752     {
753       /* RFC2068 says that in case of the 300 (multiple choices)
754          response, the server can output a preferred URL through
755          `Location' header; otherwise, the request should be treated
756          like GET.  So, if the location is set, it will be a
757          redirection; otherwise, just proceed normally.  */
758       if (statcode == HTTP_STATUS_MULTIPLE_CHOICES && !hs->newloc)
759         *dt |= RETROKF;
760       else
761         {
762           logprintf (LOG_VERBOSE,
763                      _("Location: %s%s\n"),
764                      hs->newloc ? hs->newloc : _("unspecified"),
765                      hs->newloc ? _(" [following]") : "");
766           CLOSE (sock);
767           FREE_MAYBE (type);
768           FREE_MAYBE (all_headers);
769           return NEWLOCATION;
770         }
771     }
772   if (opt.verbose)
773     {
774       if ((*dt & RETROKF) && !opt.server_response)
775         {
776           /* No need to print this output if the body won't be
777              downloaded at all, or if the original server response is
778              printed.  */
779           logputs (LOG_VERBOSE, _("Length: "));
780           if (contlen != -1)
781             {
782               logputs (LOG_VERBOSE, legible (contlen));
783               if (contrange != -1)
784                 logprintf (LOG_VERBOSE, _(" (%s to go)"),
785                            legible (contlen - contrange));
786             }
787           else
788             logputs (LOG_VERBOSE,
789                      opt.ignore_length ? _("ignored") : _("unspecified"));
790           if (type)
791             logprintf (LOG_VERBOSE, " [%s]\n", type);
792           else
793             logputs (LOG_VERBOSE, "\n");
794         }
795     }
796   FREE_MAYBE (type);
797   type = NULL;                  /* We don't need it any more.  */
798
799   /* Return if we have no intention of further downloading.  */
800   if (!(*dt & RETROKF) || (*dt & HEAD_ONLY))
801     {
802       /* In case someone cares to look...  */
803       hs->len = 0L;
804       hs->res = 0;
805       FREE_MAYBE (type);
806       FREE_MAYBE (all_headers);
807       CLOSE (sock);
808       return RETRFINISHED;
809     }
810
811   /* Open the local file.  */
812   if (!opt.dfp)
813     {
814       mkalldirs (u->local);
815       if (opt.backups)
816         rotate_backups (u->local);
817       fp = fopen (u->local, hs->restval ? "ab" : "wb");
818       if (!fp)
819         {
820           logprintf (LOG_NOTQUIET, "%s: %s\n", u->local, strerror (errno));
821           CLOSE (sock);
822           FREE_MAYBE (all_headers);
823           return FOPENERR;
824         }
825     }
826   else                      /* opt.dfp */
827     fp = opt.dfp;
828
829   /* #### This confuses the code that checks for file size.  There
830      should be some overhead information.  */
831   if (opt.save_headers)
832     fwrite (all_headers, 1, all_length, fp);
833   reset_timer ();
834   /* Get the contents of the document.  */
835   hs->res = get_contents (sock, fp, &hs->len, hs->restval,
836                           (contlen != -1 ? contlen : 0),
837                           &rbuf);
838   hs->dltime = elapsed_time ();
839   if (!opt.dfp)
840     fclose (fp);
841   else
842     fflush (fp);
843   FREE_MAYBE (all_headers);
844   CLOSE (sock);
845   if (hs->res == -2)
846     return FWRITEERR;
847   return RETRFINISHED;
848 }
849
850 /* The genuine HTTP loop!  This is the part where the retrieval is
851    retried, and retried, and retried, and...  */
852 uerr_t
853 http_loop (struct urlinfo *u, char **newloc, int *dt)
854 {
855   static int first_retrieval = 1;
856
857   int count;
858   int local_dot_orig_file_exists = FALSE;
859   int use_ts, got_head = 0;     /* time-stamping info */
860   char *tms, *suf, *locf, *tmrate;
861   uerr_t err;
862   time_t tml = -1, tmr = -1;    /* local and remote time-stamps */
863   long local_size = 0;          /* the size of the local file */
864   struct http_stat hstat;       /* HTTP status */
865   struct stat st;
866
867   *newloc = NULL;
868
869   /* Warn on (likely bogus) wildcard usage in HTTP.  Don't use
870      has_wildcards_p because it would also warn on `?', and we know that
871      shows up in CGI paths a *lot*.  */
872   if (strchr (u->url, '*'))
873     logputs (LOG_VERBOSE, _("Warning: wildcards not supported in HTTP.\n"));
874
875   /* Determine the local filename.  */
876   if (!u->local)
877     u->local = url_filename (u->proxy ? u->proxy : u);
878
879   if (!opt.output_document)
880     locf = u->local;
881   else
882     locf = opt.output_document;
883
884   if (opt.noclobber && file_exists_p (u->local))
885     {
886       /* If opt.noclobber is turned on and file already exists, do not
887          retrieve the file */
888       logprintf (LOG_VERBOSE, _("\
889 File `%s' already there, will not retrieve.\n"), u->local);
890       /* If the file is there, we suppose it's retrieved OK.  */
891       *dt |= RETROKF;
892
893       /* #### Bogusness alert.  */
894       /* If its suffix is "html" or (yuck!) "htm", we suppose it's
895          text/html, a harmless lie.  */
896       if (((suf = suffix (u->local)) != NULL)
897           && (!strcmp (suf, "html") || !strcmp (suf, "htm")))
898         *dt |= TEXTHTML;
899       free (suf);
900       /* Another harmless lie: */
901       return RETROK;
902     }
903
904   use_ts = 0;
905   if (opt.timestamping)
906     {
907       boolean  local_file_exists = FALSE;
908
909       if (opt.backup_converted)
910         /* If -K is specified, we'll act on the assumption that it was specified
911            last time these files were downloaded as well, and instead of just
912            comparing local file X against server file X, we'll compare local
913            file X.orig (if extant, else X) against server file X.  If -K
914            _wasn't_ specified last time, or the server contains files called
915            *.orig, -N will be back to not operating correctly with -k. */
916         {
917           size_t filename_len = strlen(u->local);
918           char*  filename_plus_orig_suffix = malloc(filename_len +
919                                                     sizeof(".orig"));
920
921           /* Would a single s[n]printf() call be faster? */
922           strcpy(filename_plus_orig_suffix, u->local);
923           strcpy(filename_plus_orig_suffix + filename_len, ".orig");
924
925           /* Try to stat() the .orig file. */
926           if (stat(filename_plus_orig_suffix, &st) == 0)
927             {
928               local_file_exists = TRUE;
929               local_dot_orig_file_exists = TRUE;
930             }
931
932           free(filename_plus_orig_suffix);
933         }      
934
935       if (!local_dot_orig_file_exists)
936         /* Couldn't stat() <file>.orig, so try to stat() <file>. */
937         if (stat (u->local, &st) == 0)
938           local_file_exists = TRUE;
939
940       if (local_file_exists)
941         /* There was a local file, so we'll check later to see if the version
942            the server has is the same version we already have, allowing us to
943            skip a download. */
944         {
945           use_ts = 1;
946           tml = st.st_mtime;
947           local_size = st.st_size;
948           got_head = 0;
949         }
950     }
951   /* Reset the counter.  */
952   count = 0;
953   *dt = 0 | ACCEPTRANGES;
954   /* THE loop */
955   do
956     {
957       /* Increment the pass counter.  */
958       ++count;
959       /* Wait before the retrieval (unless this is the very first
960          retrieval).  */
961       if (!first_retrieval && opt.wait)
962         sleep (opt.wait);
963       if (first_retrieval)
964         first_retrieval = 0;
965       /* Get the current time string.  */
966       tms = time_str (NULL);
967       /* Print fetch message, if opt.verbose.  */
968       if (opt.verbose)
969         {
970           char *hurl = str_url (u->proxy ? u->proxy : u, 1);
971           char tmp[15];
972           strcpy (tmp, "        ");
973           if (count > 1)
974             sprintf (tmp, _("(try:%2d)"), count);
975           logprintf (LOG_VERBOSE, "--%s--  %s\n  %s => `%s'\n",
976                      tms, hurl, tmp, locf);
977 #ifdef WINDOWS
978           ws_changetitle (hurl, 1);
979 #endif
980           free (hurl);
981         }
982
983       /* Default document type is empty.  However, if spider mode is
984          on or time-stamping is employed, HEAD_ONLY commands is
985          encoded within *dt.  */
986       if (opt.spider || (use_ts && !got_head))
987         *dt |= HEAD_ONLY;
988       else
989         *dt &= ~HEAD_ONLY;
990       /* Assume no restarting.  */
991       hstat.restval = 0L;
992       /* Decide whether or not to restart.  */
993       if (((count > 1 && (*dt & ACCEPTRANGES)) || opt.always_rest)
994           && file_exists_p (u->local))
995         if (stat (u->local, &st) == 0)
996           hstat.restval = st.st_size;
997       /* Decide whether to send the no-cache directive.  */
998       if (u->proxy && (count > 1 || (opt.proxy_cache == 0)))
999         *dt |= SEND_NOCACHE;
1000       else
1001         *dt &= ~SEND_NOCACHE;
1002
1003       /* Try fetching the document, or at least its head.  :-) */
1004       err = gethttp (u, &hstat, dt);
1005       /* Time?  */
1006       tms = time_str (NULL);
1007       /* Get the new location (with or without the redirection).  */
1008       if (hstat.newloc)
1009         *newloc = xstrdup (hstat.newloc);
1010       switch (err)
1011         {
1012         case HERR: case HEOF: case CONSOCKERR: case CONCLOSED:
1013         case CONERROR: case READERR: case WRITEFAILED:
1014         case RANGEERR:
1015           /* Non-fatal errors continue executing the loop, which will
1016              bring them to "while" statement at the end, to judge
1017              whether the number of tries was exceeded.  */
1018           FREEHSTAT (hstat);
1019           printwhat (count, opt.ntry);
1020           continue;
1021           break;
1022         case HOSTERR: case CONREFUSED: case PROXERR: case AUTHFAILED:
1023           /* Fatal errors just return from the function.  */
1024           FREEHSTAT (hstat);
1025           return err;
1026           break;
1027         case FWRITEERR: case FOPENERR:
1028           /* Another fatal error.  */
1029           logputs (LOG_VERBOSE, "\n");
1030           logprintf (LOG_NOTQUIET, _("Cannot write to `%s' (%s).\n"),
1031                      u->local, strerror (errno));
1032           FREEHSTAT (hstat);
1033           return err;
1034           break;
1035         case NEWLOCATION:
1036           /* Return the new location to the caller.  */
1037           if (!hstat.newloc)
1038             {
1039               logprintf (LOG_NOTQUIET,
1040                          _("ERROR: Redirection (%d) without location.\n"),
1041                          hstat.statcode);
1042               return WRONGCODE;
1043             }
1044           FREEHSTAT (hstat);
1045           return NEWLOCATION;
1046           break;
1047         case RETRFINISHED:
1048           /* Deal with you later.  */
1049           break;
1050         default:
1051           /* All possibilities should have been exhausted.  */
1052           abort ();
1053         }
1054       if (!(*dt & RETROKF))
1055         {
1056           if (!opt.verbose)
1057             {
1058               /* #### Ugly ugly ugly! */
1059               char *hurl = str_url (u->proxy ? u->proxy : u, 1);
1060               logprintf (LOG_NONVERBOSE, "%s:\n", hurl);
1061               free (hurl);
1062             }
1063           logprintf (LOG_NOTQUIET, _("%s ERROR %d: %s.\n"),
1064                      tms, hstat.statcode, hstat.error);
1065           logputs (LOG_VERBOSE, "\n");
1066           FREEHSTAT (hstat);
1067           return WRONGCODE;
1068         }
1069
1070       /* Did we get the time-stamp?  */
1071       if (!got_head)
1072         {
1073           if (opt.timestamping && !hstat.remote_time)
1074             {
1075               logputs (LOG_NOTQUIET, _("\
1076 Last-modified header missing -- time-stamps turned off.\n"));
1077             }
1078           else if (hstat.remote_time)
1079             {
1080               /* Convert the date-string into struct tm.  */
1081               tmr = http_atotm (hstat.remote_time);
1082               if (tmr == (time_t) (-1))
1083                 logputs (LOG_VERBOSE, _("\
1084 Last-modified header invalid -- time-stamp ignored.\n"));
1085             }
1086         }
1087
1088       /* The time-stamping section.  */
1089       if (use_ts)
1090         {
1091           got_head = 1;
1092           *dt &= ~HEAD_ONLY;
1093           use_ts = 0;           /* no more time-stamping */
1094           count = 0;            /* the retrieve count for HEAD is
1095                                    reset */
1096           if (hstat.remote_time && tmr != (time_t) (-1))
1097             {
1098               /* Now time-stamping can be used validly.  Time-stamping
1099                  means that if the sizes of the local and remote file
1100                  match, and local file is newer than the remote file,
1101                  it will not be retrieved.  Otherwise, the normal
1102                  download procedure is resumed.  */
1103               if (tml >= tmr &&
1104                   (hstat.contlen == -1 || local_size == hstat.contlen))
1105                 {
1106                   if (local_dot_orig_file_exists)
1107                     /* We can't collapse this down into just one logprintf()
1108                        call with a variable set to u->local or the .orig
1109                        filename because we have to malloc() space for the
1110                        latter, and because there are multiple returns above (a
1111                        coding style no-no by many measures, for reasons such as
1112                        this) we'd have to remember to free() the string at each
1113                        one to avoid a memory leak. */
1114                     logprintf (LOG_VERBOSE, _("\
1115 Server file no newer than local file `%s.orig' -- not retrieving.\n\n"),
1116                                u->local);
1117                   else
1118                     logprintf (LOG_VERBOSE, _("\
1119 Server file no newer than local file `%s' -- not retrieving.\n\n"), u->local);
1120                   FREEHSTAT (hstat);
1121                   return RETROK;
1122                 }
1123               else if (tml >= tmr)
1124                 logprintf (LOG_VERBOSE, _("\
1125 The sizes do not match (local %ld) -- retrieving.\n"), local_size);
1126               else
1127                 logputs (LOG_VERBOSE,
1128                          _("Remote file is newer, retrieving.\n"));
1129             }
1130           FREEHSTAT (hstat);
1131           continue;
1132         }
1133       if (!opt.dfp
1134           && (tmr != (time_t) (-1))
1135           && !opt.spider
1136           && ((hstat.len == hstat.contlen) ||
1137               ((hstat.res == 0) &&
1138                ((hstat.contlen == -1) ||
1139                 (hstat.len >= hstat.contlen && !opt.kill_longer)))))
1140         {
1141           touch (u->local, tmr);
1142         }
1143       /* End of time-stamping section.  */
1144
1145       if (opt.spider)
1146         {
1147           logprintf (LOG_NOTQUIET, "%d %s\n\n", hstat.statcode, hstat.error);
1148           return RETROK;
1149         }
1150
1151       /* It is now safe to free the remainder of hstat, since the
1152          strings within it will no longer be used.  */
1153       FREEHSTAT (hstat);
1154
1155       tmrate = rate (hstat.len - hstat.restval, hstat.dltime);
1156
1157       if (hstat.len == hstat.contlen)
1158         {
1159           if (*dt & RETROKF)
1160             {
1161               logprintf (LOG_VERBOSE,
1162                          _("%s (%s) - `%s' saved [%ld/%ld]\n\n"),
1163                          tms, tmrate, locf, hstat.len, hstat.contlen);
1164               logprintf (LOG_NONVERBOSE,
1165                          "%s URL:%s [%ld/%ld] -> \"%s\" [%d]\n",
1166                          tms, u->url, hstat.len, hstat.contlen, locf, count);
1167             }
1168           ++opt.numurls;
1169           opt.downloaded += hstat.len;
1170           downloaded_file(ADD_FILE, locf);
1171           return RETROK;
1172         }
1173       else if (hstat.res == 0) /* No read error */
1174         {
1175           if (hstat.contlen == -1)  /* We don't know how much we were supposed
1176                                        to get, so assume we succeeded. */ 
1177             {
1178               if (*dt & RETROKF)
1179                 {
1180                   logprintf (LOG_VERBOSE,
1181                              _("%s (%s) - `%s' saved [%ld]\n\n"),
1182                              tms, tmrate, locf, hstat.len);
1183                   logprintf (LOG_NONVERBOSE,
1184                              "%s URL:%s [%ld] -> \"%s\" [%d]\n",
1185                              tms, u->url, hstat.len, locf, count);
1186                 }
1187               ++opt.numurls;
1188               opt.downloaded += hstat.len;
1189               downloaded_file(ADD_FILE, locf);
1190               return RETROK;
1191             }
1192           else if (hstat.len < hstat.contlen) /* meaning we lost the
1193                                                  connection too soon */
1194             {
1195               logprintf (LOG_VERBOSE,
1196                          _("%s (%s) - Connection closed at byte %ld. "),
1197                          tms, tmrate, hstat.len);
1198               printwhat (count, opt.ntry);
1199               continue;
1200             }
1201           else if (!opt.kill_longer) /* meaning we got more than expected */
1202             {
1203               logprintf (LOG_VERBOSE,
1204                          _("%s (%s) - `%s' saved [%ld/%ld])\n\n"),
1205                          tms, tmrate, locf, hstat.len, hstat.contlen);
1206               logprintf (LOG_NONVERBOSE,
1207                          "%s URL:%s [%ld/%ld] -> \"%s\" [%d]\n",
1208                          tms, u->url, hstat.len, hstat.contlen, locf, count);
1209               ++opt.numurls;
1210               opt.downloaded += hstat.len;
1211               downloaded_file(ADD_FILE, locf);
1212               return RETROK;
1213             }
1214           else                  /* the same, but not accepted */
1215             {
1216               logprintf (LOG_VERBOSE,
1217                          _("%s (%s) - Connection closed at byte %ld/%ld. "),
1218                          tms, tmrate, hstat.len, hstat.contlen);
1219               printwhat (count, opt.ntry);
1220               continue;
1221             }
1222         }
1223       else                      /* now hstat.res can only be -1 */
1224         {
1225           if (hstat.contlen == -1)
1226             {
1227               logprintf (LOG_VERBOSE,
1228                          _("%s (%s) - Read error at byte %ld (%s)."),
1229                          tms, tmrate, hstat.len, strerror (errno));
1230               printwhat (count, opt.ntry);
1231               continue;
1232             }
1233           else                  /* hstat.res == -1 and contlen is given */
1234             {
1235               logprintf (LOG_VERBOSE,
1236                          _("%s (%s) - Read error at byte %ld/%ld (%s). "),
1237                          tms, tmrate, hstat.len, hstat.contlen,
1238                          strerror (errno));
1239               printwhat (count, opt.ntry);
1240               continue;
1241             }
1242         }
1243       /* not reached */
1244       break;
1245     }
1246   while (!opt.ntry || (count < opt.ntry));
1247   return TRYLIMEXC;
1248 }
1249 \f
1250 /* Converts struct tm to time_t, assuming the data in tm is UTC rather
1251    than local timezone (mktime assumes the latter).
1252
1253    Contributed by Roger Beeman <beeman@cisco.com>, with the help of
1254    Mark Baushke <mdb@cisco.com> and the rest of the Gurus at CISCO.  */
1255 static time_t
1256 mktime_from_utc (struct tm *t)
1257 {
1258   time_t tl, tb;
1259
1260   tl = mktime (t);
1261   if (tl == -1)
1262     return -1;
1263   tb = mktime (gmtime (&tl));
1264   return (tl <= tb ? (tl + (tl - tb)) : (tl - (tb - tl)));
1265 }
1266
1267 /* Check whether the result of strptime() indicates success.
1268    strptime() returns the pointer to how far it got to in the string.
1269    The processing has been successful if the string is at `GMT' or
1270    `+X', or at the end of the string.
1271
1272    In extended regexp parlance, the function returns 1 if P matches
1273    "^ *(GMT|[+-][0-9]|$)", 0 otherwise.  P being NULL (a valid result of
1274    strptime()) is considered a failure and 0 is returned.  */
1275 static int
1276 check_end (char *p)
1277 {
1278   if (!p)
1279     return 0;
1280   while (ISSPACE (*p))
1281     ++p;
1282   if (!*p
1283       || (p[0] == 'G' && p[1] == 'M' && p[2] == 'T')
1284       || ((p[0] == '+' || p[1] == '-') && ISDIGIT (p[1])))
1285     return 1;
1286   else
1287     return 0;
1288 }
1289
1290 /* Convert TIME_STRING time to time_t.  TIME_STRING can be in any of
1291    the three formats RFC2068 allows the HTTP servers to emit --
1292    RFC1123-date, RFC850-date or asctime-date.  Timezones are ignored,
1293    and should be GMT.
1294
1295    We use strptime() to recognize various dates, which makes it a
1296    little bit slacker than the RFC1123/RFC850/asctime (e.g. it always
1297    allows shortened dates and months, one-digit days, etc.).  It also
1298    allows more than one space anywhere where the specs require one SP.
1299    The routine should probably be even more forgiving (as recommended
1300    by RFC2068), but I do not have the time to write one.
1301
1302    Return the computed time_t representation, or -1 if all the
1303    schemes fail.
1304
1305    Needless to say, what we *really* need here is something like
1306    Marcus Hennecke's atotm(), which is forgiving, fast, to-the-point,
1307    and does not use strptime().  atotm() is to be found in the sources
1308    of `phttpd', a little-known HTTP server written by Peter Erikson.  */
1309 static time_t
1310 http_atotm (char *time_string)
1311 {
1312   struct tm t;
1313
1314   /* Roger Beeman says: "This function dynamically allocates struct tm
1315      t, but does no initialization.  The only field that actually
1316      needs initialization is tm_isdst, since the others will be set by
1317      strptime.  Since strptime does not set tm_isdst, it will return
1318      the data structure with whatever data was in tm_isdst to begin
1319      with.  For those of us in timezones where DST can occur, there
1320      can be a one hour shift depending on the previous contents of the
1321      data area where the data structure is allocated."  */
1322   t.tm_isdst = -1;
1323
1324   /* Note that under foreign locales Solaris strptime() fails to
1325      recognize English dates, which renders this function useless.  I
1326      assume that other non-GNU strptime's are plagued by the same
1327      disease.  We solve this by setting only LC_MESSAGES in
1328      i18n_initialize(), instead of LC_ALL.
1329
1330      Another solution could be to temporarily set locale to C, invoke
1331      strptime(), and restore it back.  This is slow and dirty,
1332      however, and locale support other than LC_MESSAGES can mess other
1333      things, so I rather chose to stick with just setting LC_MESSAGES.
1334
1335      Also note that none of this is necessary under GNU strptime(),
1336      because it recognizes both international and local dates.  */
1337
1338   /* NOTE: We don't use `%n' for white space, as OSF's strptime uses
1339      it to eat all white space up to (and including) a newline, and
1340      the function fails if there is no newline (!).
1341
1342      Let's hope all strptime() implementations use ` ' to skip *all*
1343      whitespace instead of just one (it works that way on all the
1344      systems I've tested it on).  */
1345
1346   /* RFC1123: Thu, 29 Jan 1998 22:12:57 */
1347   if (check_end (strptime (time_string, "%a, %d %b %Y %T", &t)))
1348     return mktime_from_utc (&t);
1349   /* RFC850:  Thu, 29-Jan-98 22:12:57 */
1350   if (check_end (strptime (time_string, "%a, %d-%b-%y %T", &t)))
1351     return mktime_from_utc (&t);
1352   /* asctime: Thu Jan 29 22:12:57 1998 */
1353   if (check_end (strptime (time_string, "%a %b %d %T %Y", &t)))
1354     return mktime_from_utc (&t);
1355   /* Failure.  */
1356   return -1;
1357 }
1358 \f
1359 /* Authorization support: We support two authorization schemes:
1360
1361    * `Basic' scheme, consisting of base64-ing USER:PASSWORD string;
1362
1363    * `Digest' scheme, added by Junio Hamano <junio@twinsun.com>,
1364    consisting of answering to the server's challenge with the proper
1365    MD5 digests.  */
1366
1367 /* How many bytes it will take to store LEN bytes in base64.  */
1368 #define BASE64_LENGTH(len) (4 * (((len) + 2) / 3))
1369
1370 /* Encode the string S of length LENGTH to base64 format and place it
1371    to STORE.  STORE will be 0-terminated, and must point to a writable
1372    buffer of at least 1+BASE64_LENGTH(length) bytes.  */
1373 static void
1374 base64_encode (const char *s, char *store, int length)
1375 {
1376   /* Conversion table.  */
1377   static char tbl[64] = {
1378     'A','B','C','D','E','F','G','H',
1379     'I','J','K','L','M','N','O','P',
1380     'Q','R','S','T','U','V','W','X',
1381     'Y','Z','a','b','c','d','e','f',
1382     'g','h','i','j','k','l','m','n',
1383     'o','p','q','r','s','t','u','v',
1384     'w','x','y','z','0','1','2','3',
1385     '4','5','6','7','8','9','+','/'
1386   };
1387   int i;
1388   unsigned char *p = (unsigned char *)store;
1389
1390   /* Transform the 3x8 bits to 4x6 bits, as required by base64.  */
1391   for (i = 0; i < length; i += 3)
1392     {
1393       *p++ = tbl[s[0] >> 2];
1394       *p++ = tbl[((s[0] & 3) << 4) + (s[1] >> 4)];
1395       *p++ = tbl[((s[1] & 0xf) << 2) + (s[2] >> 6)];
1396       *p++ = tbl[s[2] & 0x3f];
1397       s += 3;
1398     }
1399   /* Pad the result if necessary...  */
1400   if (i == length + 1)
1401     *(p - 1) = '=';
1402   else if (i == length + 2)
1403     *(p - 1) = *(p - 2) = '=';
1404   /* ...and zero-terminate it.  */
1405   *p = '\0';
1406 }
1407
1408 /* Create the authentication header contents for the `Basic' scheme.
1409    This is done by encoding the string `USER:PASS' in base64 and
1410    prepending `HEADER: Basic ' to it.  */
1411 static char *
1412 basic_authentication_encode (const char *user, const char *passwd,
1413                              const char *header)
1414 {
1415   char *t1, *t2, *res;
1416   int len1 = strlen (user) + 1 + strlen (passwd);
1417   int len2 = BASE64_LENGTH (len1);
1418
1419   t1 = (char *)alloca (len1 + 1);
1420   sprintf (t1, "%s:%s", user, passwd);
1421   t2 = (char *)alloca (1 + len2);
1422   base64_encode (t1, t2, len1);
1423   res = (char *)malloc (len2 + 11 + strlen (header));
1424   sprintf (res, "%s: Basic %s\r\n", header, t2);
1425
1426   return res;
1427 }
1428
1429 #ifdef USE_DIGEST
1430 /* Parse HTTP `WWW-Authenticate:' header.  AU points to the beginning
1431    of a field in such a header.  If the field is the one specified by
1432    ATTR_NAME ("realm", "opaque", and "nonce" are used by the current
1433    digest authorization code), extract its value in the (char*)
1434    variable pointed by RET.  Returns negative on a malformed header,
1435    or number of bytes that have been parsed by this call.  */
1436 static int
1437 extract_header_attr (const char *au, const char *attr_name, char **ret)
1438 {
1439   const char *cp, *ep;
1440
1441   ep = cp = au;
1442
1443   if (strncmp (cp, attr_name, strlen (attr_name)) == 0)
1444     {
1445       cp += strlen (attr_name);
1446       if (!*cp)
1447         return -1;
1448       cp += skip_lws (cp);
1449       if (*cp != '=')
1450         return -1;
1451       if (!*++cp)
1452         return -1;
1453       cp += skip_lws (cp);
1454       if (*cp != '\"')
1455         return -1;
1456       if (!*++cp)
1457         return -1;
1458       for (ep = cp; *ep && *ep != '\"'; ep++)
1459         ;
1460       if (!*ep)
1461         return -1;
1462       FREE_MAYBE (*ret);
1463       *ret = strdupdelim (cp, ep);
1464       return ep - au + 1;
1465     }
1466   else
1467     return 0;
1468 }
1469
1470 /* Response value needs to be in lowercase, so we cannot use HEXD2ASC
1471    from url.h.  See RFC 2069 2.1.2 for the syntax of response-digest.  */
1472 #define HEXD2asc(x) (((x) < 10) ? ((x) + '0') : ((x) - 10 + 'a'))
1473
1474 /* Dump the hexadecimal representation of HASH to BUF.  HASH should be
1475    an array of 16 bytes containing the hash keys, and BUF should be a
1476    buffer of 33 writable characters (32 for hex digits plus one for
1477    zero termination).  */
1478 static void
1479 dump_hash (unsigned char *buf, const unsigned char *hash)
1480 {
1481   int i;
1482
1483   for (i = 0; i < MD5_HASHLEN; i++, hash++)
1484     {
1485       *buf++ = HEXD2asc (*hash >> 4);
1486       *buf++ = HEXD2asc (*hash & 0xf);
1487     }
1488   *buf = '\0';
1489 }
1490
1491 /* Take the line apart to find the challenge, and compose a digest
1492    authorization header.  See RFC2069 section 2.1.2.  */
1493 char *
1494 digest_authentication_encode (const char *au, const char *user,
1495                               const char *passwd, const char *method,
1496                               const char *path)
1497 {
1498   static char *realm, *opaque, *nonce;
1499   static struct {
1500     const char *name;
1501     char **variable;
1502   } options[] = {
1503     { "realm", &realm },
1504     { "opaque", &opaque },
1505     { "nonce", &nonce }
1506   };
1507   char *res;
1508
1509   realm = opaque = nonce = NULL;
1510
1511   au += 6;                      /* skip over `Digest' */
1512   while (*au)
1513     {
1514       int i;
1515
1516       au += skip_lws (au);
1517       for (i = 0; i < ARRAY_SIZE (options); i++)
1518         {
1519           int skip = extract_header_attr (au, options[i].name,
1520                                           options[i].variable);
1521           if (skip < 0)
1522             {
1523               FREE_MAYBE (realm);
1524               FREE_MAYBE (opaque);
1525               FREE_MAYBE (nonce);
1526               return NULL;
1527             }
1528           else if (skip)
1529             {
1530               au += skip;
1531               break;
1532             }
1533         }
1534       if (i == ARRAY_SIZE (options))
1535         {
1536           while (*au && *au != '=')
1537             au++;
1538           if (*au && *++au)
1539             {
1540               au += skip_lws (au);
1541               if (*au == '\"')
1542                 {
1543                   au++;
1544                   while (*au && *au != '\"')
1545                     au++;
1546                   if (*au)
1547                     au++;
1548                 }
1549             }
1550         }
1551       while (*au && *au != ',')
1552         au++;
1553       if (*au)
1554         au++;
1555     }
1556   if (!realm || !nonce || !user || !passwd || !path || !method)
1557     {
1558       FREE_MAYBE (realm);
1559       FREE_MAYBE (opaque);
1560       FREE_MAYBE (nonce);
1561       return NULL;
1562     }
1563
1564   /* Calculate the digest value.  */
1565   {
1566     struct md5_ctx ctx;
1567     unsigned char hash[MD5_HASHLEN];
1568     unsigned char a1buf[MD5_HASHLEN * 2 + 1], a2buf[MD5_HASHLEN * 2 + 1];
1569     unsigned char response_digest[MD5_HASHLEN * 2 + 1];
1570
1571     /* A1BUF = H(user ":" realm ":" password) */
1572     md5_init_ctx (&ctx);
1573     md5_process_bytes (user, strlen (user), &ctx);
1574     md5_process_bytes (":", 1, &ctx);
1575     md5_process_bytes (realm, strlen (realm), &ctx);
1576     md5_process_bytes (":", 1, &ctx);
1577     md5_process_bytes (passwd, strlen (passwd), &ctx);
1578     md5_finish_ctx (&ctx, hash);
1579     dump_hash (a1buf, hash);
1580
1581     /* A2BUF = H(method ":" path) */
1582     md5_init_ctx (&ctx);
1583     md5_process_bytes (method, strlen (method), &ctx);
1584     md5_process_bytes (":", 1, &ctx);
1585     md5_process_bytes (path, strlen (path), &ctx);
1586     md5_finish_ctx (&ctx, hash);
1587     dump_hash (a2buf, hash);
1588
1589     /* RESPONSE_DIGEST = H(A1BUF ":" nonce ":" A2BUF) */
1590     md5_init_ctx (&ctx);
1591     md5_process_bytes (a1buf, MD5_HASHLEN * 2, &ctx);
1592     md5_process_bytes (":", 1, &ctx);
1593     md5_process_bytes (nonce, strlen (nonce), &ctx);
1594     md5_process_bytes (":", 1, &ctx);
1595     md5_process_bytes (a2buf, MD5_HASHLEN * 2, &ctx);
1596     md5_finish_ctx (&ctx, hash);
1597     dump_hash (response_digest, hash);
1598
1599     res = (char*) xmalloc (strlen (user)
1600                            + strlen (user)
1601                            + strlen (realm)
1602                            + strlen (nonce)
1603                            + strlen (path)
1604                            + 2 * MD5_HASHLEN /*strlen (response_digest)*/
1605                            + (opaque ? strlen (opaque) : 0)
1606                            + 128);
1607     sprintf (res, "Authorization: Digest \
1608 username=\"%s\", realm=\"%s\", nonce=\"%s\", uri=\"%s\", response=\"%s\"",
1609              user, realm, nonce, path, response_digest);
1610     if (opaque)
1611       {
1612         char *p = res + strlen (res);
1613         strcat (p, ", opaque=\"");
1614         strcat (p, opaque);
1615         strcat (p, "\"");
1616       }
1617     strcat (res, "\r\n");
1618   }
1619   return res;
1620 }
1621 #endif /* USE_DIGEST */
1622
1623
1624 #define HACK_O_MATIC(line, string_constant)                             \
1625   (!strncasecmp (line, string_constant, sizeof (string_constant) - 1)   \
1626    && (ISSPACE (line[sizeof (string_constant) - 1])                     \
1627        || !line[sizeof (string_constant) - 1]))
1628
1629 static int
1630 known_authentication_scheme_p (const char *au)
1631 {
1632   return HACK_O_MATIC (au, "Basic") || HACK_O_MATIC (au, "Digest");
1633 }
1634
1635 #undef HACK_O_MATIC
1636
1637 /* Create the HTTP authorization request header.  When the
1638    `WWW-Authenticate' response header is seen, according to the
1639    authorization scheme specified in that header (`Basic' and `Digest'
1640    are supported by the current implementation), produce an
1641    appropriate HTTP authorization request header.  */
1642 static char *
1643 create_authorization_line (const char *au, const char *user,
1644                            const char *passwd, const char *method,
1645                            const char *path)
1646 {
1647   char *wwwauth = NULL;
1648
1649   if (!strncasecmp (au, "Basic", 5))
1650     wwwauth = basic_authentication_encode (user, passwd, "Authorization");
1651 #ifdef USE_DIGEST
1652   else if (!strncasecmp (au, "Digest", 6))
1653     wwwauth = digest_authentication_encode (au, user, passwd, method, path);
1654 #endif /* USE_DIGEST */
1655   return wwwauth;
1656 }