]> sjero.net Git - wget/blob - src/http.c
[svn] Applied contributed patches (see ChangeLog for details.)
[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 ? qstring : "", useragent, remhost,
492            host_port ? host_port : "",
493            HTTP_ACCEPT, referer ? referer : "",
494            wwwauth ? wwwauth : "", 
495            proxyauth ? proxyauth : "", 
496            range ? range : "",
497            pragma_h, 
498            opt.user_header ? opt.user_header : "");
499   DEBUGP (("---request begin---\n%s---request end---\n", request));
500    /* Free the temporary memory.  */
501   FREE_MAYBE (wwwauth);
502   FREE_MAYBE (proxyauth);
503
504   /* Send the request to server.  */
505   num_written = iwrite (sock, request, strlen (request));
506   if (num_written < 0)
507     {
508       logputs (LOG_VERBOSE, _("Failed writing HTTP request.\n"));
509       free (request);
510       CLOSE (sock);
511       return WRITEFAILED;
512     }
513   logprintf (LOG_VERBOSE, _("%s request sent, awaiting response... "),
514              u->proxy ? "Proxy" : "HTTP");
515   contlen = contrange = -1;
516   type = NULL;
517   statcode = -1;
518   *dt &= ~RETROKF;
519
520   /* Before reading anything, initialize the rbuf.  */
521   rbuf_initialize (&rbuf, sock);
522
523   all_headers = NULL;
524   all_length = 0;
525   /* Header-fetching loop.  */
526   hcount = 0;
527   while (1)
528     {
529       char *hdr;
530       int status;
531
532       ++hcount;
533       /* Get the header.  */
534       status = header_get (&rbuf, &hdr,
535                            /* Disallow continuations for status line.  */
536                            (hcount == 1 ? HG_NO_CONTINUATIONS : HG_NONE));
537
538       /* Check for errors.  */
539       if (status == HG_EOF && *hdr)
540         {
541           /* This used to be an unconditional error, but that was
542              somewhat controversial, because of a large number of
543              broken CGI's that happily "forget" to send the second EOL
544              before closing the connection of a HEAD request.
545
546              So, the deal is to check whether the header is empty
547              (*hdr is zero if it is); if yes, it means that the
548              previous header was fully retrieved, and that -- most
549              probably -- the request is complete.  "...be liberal in
550              what you accept."  Oh boy.  */
551           logputs (LOG_VERBOSE, "\n");
552           logputs (LOG_NOTQUIET, _("End of file while parsing headers.\n"));
553           free (hdr);
554           FREE_MAYBE (type);
555           FREE_MAYBE (hs->newloc);
556           FREE_MAYBE (all_headers);
557           CLOSE (sock);
558           return HEOF;
559         }
560       else if (status == HG_ERROR)
561         {
562           logputs (LOG_VERBOSE, "\n");
563           logprintf (LOG_NOTQUIET, _("Read error (%s) in headers.\n"),
564                      strerror (errno));
565           free (hdr);
566           FREE_MAYBE (type);
567           FREE_MAYBE (hs->newloc);
568           FREE_MAYBE (all_headers);
569           CLOSE (sock);
570           return HERR;
571         }
572
573       /* If the headers are to be saved to a file later, save them to
574          memory now.  */
575       if (opt.save_headers)
576         {
577           int lh = strlen (hdr);
578           all_headers = (char *)xrealloc (all_headers, all_length + lh + 2);
579           memcpy (all_headers + all_length, hdr, lh);
580           all_length += lh;
581           all_headers[all_length++] = '\n';
582           all_headers[all_length] = '\0';
583         }
584
585       /* Print the header if requested.  */
586       if (opt.server_response && hcount != 1)
587         logprintf (LOG_VERBOSE, "\n%d %s", hcount, hdr);
588
589       /* Check for status line.  */
590       if (hcount == 1)
591         {
592           const char *error;
593           /* Parse the first line of server response.  */
594           statcode = parse_http_status_line (hdr, &error);
595           hs->statcode = statcode;
596           /* Store the descriptive response.  */
597           if (statcode == -1) /* malformed response */
598             {
599               /* A common reason for "malformed response" error is the
600                  case when no data was actually received.  Handle this
601                  special case.  */
602               if (!*hdr)
603                 hs->error = xstrdup (_("No data received"));
604               else
605                 hs->error = xstrdup (_("Malformed status line"));
606               free (hdr);
607               break;
608             }
609           else if (!*error)
610             hs->error = xstrdup (_("(no description)"));
611           else
612             hs->error = xstrdup (error);
613
614           if ((statcode != -1)
615 #ifdef DEBUG
616               && !opt.debug
617 #endif
618               )
619             logprintf (LOG_VERBOSE, "%d %s", statcode, error);
620
621           goto done_header;
622         }
623
624       /* Exit on empty header.  */
625       if (!*hdr)
626         {
627           free (hdr);
628           break;
629         }
630
631       /* Try getting content-length.  */
632       if (contlen == -1 && !opt.ignore_length)
633         if (header_process (hdr, "Content-Length", header_extract_number,
634                             &contlen))
635           goto done_header;
636       /* Try getting content-type.  */
637       if (!type)
638         if (header_process (hdr, "Content-Type", http_process_type, &type))
639           goto done_header;
640       /* Try getting location.  */
641       if (!hs->newloc)
642         if (header_process (hdr, "Location", header_strdup, &hs->newloc))
643           goto done_header;
644       /* Try getting last-modified.  */
645       if (!hs->remote_time)
646         if (header_process (hdr, "Last-Modified", header_strdup,
647                             &hs->remote_time))
648           goto done_header;
649       /* Try getting www-authentication.  */
650       if (!authenticate_h)
651         if (header_process (hdr, "WWW-Authenticate", header_strdup,
652                             &authenticate_h))
653           goto done_header;
654       /* Check for accept-ranges header.  If it contains the word
655          `none', disable the ranges.  */
656       if (*dt & ACCEPTRANGES)
657         {
658           int nonep;
659           if (header_process (hdr, "Accept-Ranges", http_process_none, &nonep))
660             {
661               if (nonep)
662                 *dt &= ~ACCEPTRANGES;
663               goto done_header;
664             }
665         }
666       /* Try getting content-range.  */
667       if (contrange == -1)
668         {
669           struct http_process_range_closure closure;
670           if (header_process (hdr, "Content-Range", http_process_range, &closure))
671             {
672               contrange = closure.first_byte_pos;
673               goto done_header;
674             }
675         }
676     done_header:
677       free (hdr);
678     }
679
680   logputs (LOG_VERBOSE, "\n");
681
682   if ((statcode == HTTP_STATUS_UNAUTHORIZED)
683       && authenticate_h)
684     {
685       /* Authorization is required.  */
686       FREE_MAYBE (type);
687       type = NULL;
688       FREEHSTAT (*hs);
689       CLOSE (sock);
690       if (auth_tried_already)
691         {
692           /* If we have tried it already, then there is not point
693              retrying it.  */
694           logputs (LOG_NOTQUIET, _("Authorization failed.\n"));
695           free (authenticate_h);
696           return AUTHFAILED;
697         }
698       else if (!known_authentication_scheme_p (authenticate_h))
699         {
700           free (authenticate_h);
701           logputs (LOG_NOTQUIET, _("Unknown authentication scheme.\n"));
702           return AUTHFAILED;
703         }
704       else
705         {
706           auth_tried_already = 1;
707           goto again;
708         }
709     }
710   /* We do not need this anymore.  */
711   if (authenticate_h)
712     {
713       free (authenticate_h);
714       authenticate_h = NULL;
715     }
716
717   /* 20x responses are counted among successful by default.  */
718   if (H_20X (statcode))
719     *dt |= RETROKF;
720
721   if (type && !strncasecmp (type, TEXTHTML_S, strlen (TEXTHTML_S)))
722     *dt |= TEXTHTML;
723   else
724     /* We don't assume text/html by default.  */
725     *dt &= ~TEXTHTML;
726
727   if (contrange == -1)
728     hs->restval = 0;
729   else if (contrange != hs->restval ||
730            (H_PARTIAL (statcode) && contrange == -1))
731     {
732       /* This means the whole request was somehow misunderstood by the
733          server.  Bail out.  */
734       FREE_MAYBE (type);
735       FREE_MAYBE (hs->newloc);
736       FREE_MAYBE (all_headers);
737       CLOSE (sock);
738       return RANGEERR;
739     }
740
741   if (hs->restval)
742     {
743       if (contlen != -1)
744         contlen += contrange;
745       else
746         contrange = -1;        /* If conent-length was not sent,
747                                   content-range will be ignored.  */
748     }
749   hs->contlen = contlen;
750
751   /* Return if redirected.  */
752   if (H_REDIRECTED (statcode) || statcode == HTTP_STATUS_MULTIPLE_CHOICES)
753     {
754       /* RFC2068 says that in case of the 300 (multiple choices)
755          response, the server can output a preferred URL through
756          `Location' header; otherwise, the request should be treated
757          like GET.  So, if the location is set, it will be a
758          redirection; otherwise, just proceed normally.  */
759       if (statcode == HTTP_STATUS_MULTIPLE_CHOICES && !hs->newloc)
760         *dt |= RETROKF;
761       else
762         {
763           logprintf (LOG_VERBOSE,
764                      _("Location: %s%s\n"),
765                      hs->newloc ? hs->newloc : _("unspecified"),
766                      hs->newloc ? _(" [following]") : "");
767           CLOSE (sock);
768           FREE_MAYBE (type);
769           FREE_MAYBE (all_headers);
770           return NEWLOCATION;
771         }
772     }
773   if (opt.verbose)
774     {
775       if ((*dt & RETROKF) && !opt.server_response)
776         {
777           /* No need to print this output if the body won't be
778              downloaded at all, or if the original server response is
779              printed.  */
780           logputs (LOG_VERBOSE, _("Length: "));
781           if (contlen != -1)
782             {
783               logputs (LOG_VERBOSE, legible (contlen));
784               if (contrange != -1)
785                 logprintf (LOG_VERBOSE, _(" (%s to go)"),
786                            legible (contlen - contrange));
787             }
788           else
789             logputs (LOG_VERBOSE,
790                      opt.ignore_length ? _("ignored") : _("unspecified"));
791           if (type)
792             logprintf (LOG_VERBOSE, " [%s]\n", type);
793           else
794             logputs (LOG_VERBOSE, "\n");
795         }
796     }
797   FREE_MAYBE (type);
798   type = NULL;                  /* We don't need it any more.  */
799
800   /* Return if we have no intention of further downloading.  */
801   if (!(*dt & RETROKF) || (*dt & HEAD_ONLY))
802     {
803       /* In case someone cares to look...  */
804       hs->len = 0L;
805       hs->res = 0;
806       FREE_MAYBE (type);
807       FREE_MAYBE (all_headers);
808       CLOSE (sock);
809       return RETRFINISHED;
810     }
811
812   /* Open the local file.  */
813   if (!opt.dfp)
814     {
815       mkalldirs (u->local);
816       if (opt.backups)
817         rotate_backups (u->local);
818       fp = fopen (u->local, hs->restval ? "ab" : "wb");
819       if (!fp)
820         {
821           logprintf (LOG_NOTQUIET, "%s: %s\n", u->local, strerror (errno));
822           CLOSE (sock);
823           FREE_MAYBE (all_headers);
824           return FOPENERR;
825         }
826     }
827   else                      /* opt.dfp */
828     fp = opt.dfp;
829
830   /* #### This confuses the code that checks for file size.  There
831      should be some overhead information.  */
832   if (opt.save_headers)
833     fwrite (all_headers, 1, all_length, fp);
834   reset_timer ();
835   /* Get the contents of the document.  */
836   hs->res = get_contents (sock, fp, &hs->len, hs->restval,
837                           (contlen != -1 ? contlen : 0),
838                           &rbuf);
839   hs->dltime = elapsed_time ();
840   if (!opt.dfp)
841     fclose (fp);
842   else
843     fflush (fp);
844   FREE_MAYBE (all_headers);
845   CLOSE (sock);
846   if (hs->res == -2)
847     return FWRITEERR;
848   return RETRFINISHED;
849 }
850
851 /* The genuine HTTP loop!  This is the part where the retrieval is
852    retried, and retried, and retried, and...  */
853 uerr_t
854 http_loop (struct urlinfo *u, char **newloc, int *dt)
855 {
856   static int first_retrieval = 1;
857
858   int count;
859   int local_dot_orig_file_exists = FALSE;
860   int use_ts, got_head = 0;     /* time-stamping info */
861   char *tms, *suf, *locf, *tmrate;
862   uerr_t err;
863   time_t tml = -1, tmr = -1;    /* local and remote time-stamps */
864   long local_size = 0;          /* the size of the local file */
865   struct http_stat hstat;       /* HTTP status */
866   struct stat st;
867
868   *newloc = NULL;
869
870   /* Warn on (likely bogus) wildcard usage in HTTP.  Don't use
871      has_wildcards_p because it would also warn on `?', and we know that
872      shows up in CGI paths a *lot*.  */
873   if (strchr (u->url, '*'))
874     logputs (LOG_VERBOSE, _("Warning: wildcards not supported in HTTP.\n"));
875
876   /* Determine the local filename.  */
877   if (!u->local)
878     u->local = url_filename (u->proxy ? u->proxy : u);
879
880   if (!opt.output_document)
881     locf = u->local;
882   else
883     locf = opt.output_document;
884
885   if (opt.noclobber && file_exists_p (u->local))
886     {
887       /* If opt.noclobber is turned on and file already exists, do not
888          retrieve the file */
889       logprintf (LOG_VERBOSE, _("\
890 File `%s' already there, will not retrieve.\n"), u->local);
891       /* If the file is there, we suppose it's retrieved OK.  */
892       *dt |= RETROKF;
893
894       /* #### Bogusness alert.  */
895       /* If its suffix is "html" or (yuck!) "htm", we suppose it's
896          text/html, a harmless lie.  */
897       if (((suf = suffix (u->local)) != NULL)
898           && (!strcmp (suf, "html") || !strcmp (suf, "htm")))
899         *dt |= TEXTHTML;
900       free (suf);
901       /* Another harmless lie: */
902       return RETROK;
903     }
904
905   use_ts = 0;
906   if (opt.timestamping)
907     {
908       boolean  local_file_exists = FALSE;
909
910       if (opt.backup_converted)
911         /* If -K is specified, we'll act on the assumption that it was specified
912            last time these files were downloaded as well, and instead of just
913            comparing local file X against server file X, we'll compare local
914            file X.orig (if extant, else X) against server file X.  If -K
915            _wasn't_ specified last time, or the server contains files called
916            *.orig, -N will be back to not operating correctly with -k. */
917         {
918           size_t filename_len = strlen(u->local);
919           char*  filename_plus_orig_suffix = malloc(filename_len +
920                                                     sizeof(".orig"));
921
922           /* Would a single s[n]printf() call be faster? */
923           strcpy(filename_plus_orig_suffix, u->local);
924           strcpy(filename_plus_orig_suffix + filename_len, ".orig");
925
926           /* Try to stat() the .orig file. */
927           if (stat(filename_plus_orig_suffix, &st) == 0)
928             {
929               local_file_exists = TRUE;
930               local_dot_orig_file_exists = TRUE;
931             }
932
933           free(filename_plus_orig_suffix);
934         }      
935
936       if (!local_dot_orig_file_exists)
937         /* Couldn't stat() <file>.orig, so try to stat() <file>. */
938         if (stat (u->local, &st) == 0)
939           local_file_exists = TRUE;
940
941       if (local_file_exists)
942         /* There was a local file, so we'll check later to see if the version
943            the server has is the same version we already have, allowing us to
944            skip a download. */
945         {
946           use_ts = 1;
947           tml = st.st_mtime;
948           local_size = st.st_size;
949           got_head = 0;
950         }
951     }
952   /* Reset the counter.  */
953   count = 0;
954   *dt = 0 | ACCEPTRANGES;
955   /* THE loop */
956   do
957     {
958       /* Increment the pass counter.  */
959       ++count;
960       /* Wait before the retrieval (unless this is the very first
961          retrieval).
962          Check if we are retrying or not, wait accordingly - HEH */
963       if (!first_retrieval && (opt.wait || (count && opt.waitretry)))
964         if (count)
965           if (count<opt.waitretry)
966             sleep(count);
967           else
968             sleep(opt.waitretry);
969         else
970           sleep (opt.wait);
971       if (first_retrieval)
972         first_retrieval = 0;
973       /* Get the current time string.  */
974       tms = time_str (NULL);
975       /* Print fetch message, if opt.verbose.  */
976       if (opt.verbose)
977         {
978           char *hurl = str_url (u->proxy ? u->proxy : u, 1);
979           char tmp[15];
980           strcpy (tmp, "        ");
981           if (count > 1)
982             sprintf (tmp, _("(try:%2d)"), count);
983           logprintf (LOG_VERBOSE, "--%s--  %s\n  %s => `%s'\n",
984                      tms, hurl, tmp, locf);
985 #ifdef WINDOWS
986           ws_changetitle (hurl, 1);
987 #endif
988           free (hurl);
989         }
990
991       /* Default document type is empty.  However, if spider mode is
992          on or time-stamping is employed, HEAD_ONLY commands is
993          encoded within *dt.  */
994       if (opt.spider || (use_ts && !got_head))
995         *dt |= HEAD_ONLY;
996       else
997         *dt &= ~HEAD_ONLY;
998       /* Assume no restarting.  */
999       hstat.restval = 0L;
1000       /* Decide whether or not to restart.  */
1001       if (((count > 1 && (*dt & ACCEPTRANGES)) || opt.always_rest)
1002           && file_exists_p (u->local))
1003         if (stat (u->local, &st) == 0)
1004           hstat.restval = st.st_size;
1005       /* Decide whether to send the no-cache directive.  */
1006       if (u->proxy && (count > 1 || (opt.proxy_cache == 0)))
1007         *dt |= SEND_NOCACHE;
1008       else
1009         *dt &= ~SEND_NOCACHE;
1010
1011       /* Try fetching the document, or at least its head.  :-) */
1012       err = gethttp (u, &hstat, dt);
1013       /* Time?  */
1014       tms = time_str (NULL);
1015       /* Get the new location (with or without the redirection).  */
1016       if (hstat.newloc)
1017         *newloc = xstrdup (hstat.newloc);
1018       switch (err)
1019         {
1020         case HERR: case HEOF: case CONSOCKERR: case CONCLOSED:
1021         case CONERROR: case READERR: case WRITEFAILED:
1022         case RANGEERR:
1023           /* Non-fatal errors continue executing the loop, which will
1024              bring them to "while" statement at the end, to judge
1025              whether the number of tries was exceeded.  */
1026           FREEHSTAT (hstat);
1027           printwhat (count, opt.ntry);
1028           continue;
1029           break;
1030         case HOSTERR: case CONREFUSED: case PROXERR: case AUTHFAILED:
1031           /* Fatal errors just return from the function.  */
1032           FREEHSTAT (hstat);
1033           return err;
1034           break;
1035         case FWRITEERR: case FOPENERR:
1036           /* Another fatal error.  */
1037           logputs (LOG_VERBOSE, "\n");
1038           logprintf (LOG_NOTQUIET, _("Cannot write to `%s' (%s).\n"),
1039                      u->local, strerror (errno));
1040           FREEHSTAT (hstat);
1041           return err;
1042           break;
1043         case NEWLOCATION:
1044           /* Return the new location to the caller.  */
1045           if (!hstat.newloc)
1046             {
1047               logprintf (LOG_NOTQUIET,
1048                          _("ERROR: Redirection (%d) without location.\n"),
1049                          hstat.statcode);
1050               return WRONGCODE;
1051             }
1052           FREEHSTAT (hstat);
1053           return NEWLOCATION;
1054           break;
1055         case RETRFINISHED:
1056           /* Deal with you later.  */
1057           break;
1058         default:
1059           /* All possibilities should have been exhausted.  */
1060           abort ();
1061         }
1062       if (!(*dt & RETROKF))
1063         {
1064           if (!opt.verbose)
1065             {
1066               /* #### Ugly ugly ugly! */
1067               char *hurl = str_url (u->proxy ? u->proxy : u, 1);
1068               logprintf (LOG_NONVERBOSE, "%s:\n", hurl);
1069               free (hurl);
1070             }
1071           logprintf (LOG_NOTQUIET, _("%s ERROR %d: %s.\n"),
1072                      tms, hstat.statcode, hstat.error);
1073           logputs (LOG_VERBOSE, "\n");
1074           FREEHSTAT (hstat);
1075           return WRONGCODE;
1076         }
1077
1078       /* Did we get the time-stamp?  */
1079       if (!got_head)
1080         {
1081           if (opt.timestamping && !hstat.remote_time)
1082             {
1083               logputs (LOG_NOTQUIET, _("\
1084 Last-modified header missing -- time-stamps turned off.\n"));
1085             }
1086           else if (hstat.remote_time)
1087             {
1088               /* Convert the date-string into struct tm.  */
1089               tmr = http_atotm (hstat.remote_time);
1090               if (tmr == (time_t) (-1))
1091                 logputs (LOG_VERBOSE, _("\
1092 Last-modified header invalid -- time-stamp ignored.\n"));
1093             }
1094         }
1095
1096       /* The time-stamping section.  */
1097       if (use_ts)
1098         {
1099           got_head = 1;
1100           *dt &= ~HEAD_ONLY;
1101           use_ts = 0;           /* no more time-stamping */
1102           count = 0;            /* the retrieve count for HEAD is
1103                                    reset */
1104           if (hstat.remote_time && tmr != (time_t) (-1))
1105             {
1106               /* Now time-stamping can be used validly.  Time-stamping
1107                  means that if the sizes of the local and remote file
1108                  match, and local file is newer than the remote file,
1109                  it will not be retrieved.  Otherwise, the normal
1110                  download procedure is resumed.  */
1111               if (tml >= tmr &&
1112                   (hstat.contlen == -1 || local_size == hstat.contlen))
1113                 {
1114                   if (local_dot_orig_file_exists)
1115                     /* We can't collapse this down into just one logprintf()
1116                        call with a variable set to u->local or the .orig
1117                        filename because we have to malloc() space for the
1118                        latter, and because there are multiple returns above (a
1119                        coding style no-no by many measures, for reasons such as
1120                        this) we'd have to remember to free() the string at each
1121                        one to avoid a memory leak. */
1122                     logprintf (LOG_VERBOSE, _("\
1123 Server file no newer than local file `%s.orig' -- not retrieving.\n\n"),
1124                                u->local);
1125                   else
1126                     logprintf (LOG_VERBOSE, _("\
1127 Server file no newer than local file `%s' -- not retrieving.\n\n"), u->local);
1128                   FREEHSTAT (hstat);
1129                   return RETROK;
1130                 }
1131               else if (tml >= tmr)
1132                 logprintf (LOG_VERBOSE, _("\
1133 The sizes do not match (local %ld) -- retrieving.\n"), local_size);
1134               else
1135                 logputs (LOG_VERBOSE,
1136                          _("Remote file is newer, retrieving.\n"));
1137             }
1138           FREEHSTAT (hstat);
1139           continue;
1140         }
1141       if (!opt.dfp
1142           && (tmr != (time_t) (-1))
1143           && !opt.spider
1144           && ((hstat.len == hstat.contlen) ||
1145               ((hstat.res == 0) &&
1146                ((hstat.contlen == -1) ||
1147                 (hstat.len >= hstat.contlen && !opt.kill_longer)))))
1148         {
1149           touch (u->local, tmr);
1150         }
1151       /* End of time-stamping section.  */
1152
1153       if (opt.spider)
1154         {
1155           logprintf (LOG_NOTQUIET, "%d %s\n\n", hstat.statcode, hstat.error);
1156           return RETROK;
1157         }
1158
1159       /* It is now safe to free the remainder of hstat, since the
1160          strings within it will no longer be used.  */
1161       FREEHSTAT (hstat);
1162
1163       tmrate = rate (hstat.len - hstat.restval, hstat.dltime);
1164
1165       if (hstat.len == hstat.contlen)
1166         {
1167           if (*dt & RETROKF)
1168             {
1169               logprintf (LOG_VERBOSE,
1170                          _("%s (%s) - `%s' saved [%ld/%ld]\n\n"),
1171                          tms, tmrate, locf, hstat.len, hstat.contlen);
1172               logprintf (LOG_NONVERBOSE,
1173                          "%s URL:%s [%ld/%ld] -> \"%s\" [%d]\n",
1174                          tms, u->url, hstat.len, hstat.contlen, locf, count);
1175             }
1176           ++opt.numurls;
1177           opt.downloaded += hstat.len;
1178           downloaded_file(ADD_FILE, locf);
1179           return RETROK;
1180         }
1181       else if (hstat.res == 0) /* No read error */
1182         {
1183           if (hstat.contlen == -1)  /* We don't know how much we were supposed
1184                                        to get, so assume we succeeded. */ 
1185             {
1186               if (*dt & RETROKF)
1187                 {
1188                   logprintf (LOG_VERBOSE,
1189                              _("%s (%s) - `%s' saved [%ld]\n\n"),
1190                              tms, tmrate, locf, hstat.len);
1191                   logprintf (LOG_NONVERBOSE,
1192                              "%s URL:%s [%ld] -> \"%s\" [%d]\n",
1193                              tms, u->url, hstat.len, locf, count);
1194                 }
1195               ++opt.numurls;
1196               opt.downloaded += hstat.len;
1197               downloaded_file(ADD_FILE, locf);
1198               return RETROK;
1199             }
1200           else if (hstat.len < hstat.contlen) /* meaning we lost the
1201                                                  connection too soon */
1202             {
1203               logprintf (LOG_VERBOSE,
1204                          _("%s (%s) - Connection closed at byte %ld. "),
1205                          tms, tmrate, hstat.len);
1206               printwhat (count, opt.ntry);
1207               continue;
1208             }
1209           else if (!opt.kill_longer) /* meaning we got more than expected */
1210             {
1211               logprintf (LOG_VERBOSE,
1212                          _("%s (%s) - `%s' saved [%ld/%ld])\n\n"),
1213                          tms, tmrate, locf, hstat.len, hstat.contlen);
1214               logprintf (LOG_NONVERBOSE,
1215                          "%s URL:%s [%ld/%ld] -> \"%s\" [%d]\n",
1216                          tms, u->url, hstat.len, hstat.contlen, locf, count);
1217               ++opt.numurls;
1218               opt.downloaded += hstat.len;
1219               downloaded_file(ADD_FILE, locf);
1220               return RETROK;
1221             }
1222           else                  /* the same, but not accepted */
1223             {
1224               logprintf (LOG_VERBOSE,
1225                          _("%s (%s) - Connection closed at byte %ld/%ld. "),
1226                          tms, tmrate, hstat.len, hstat.contlen);
1227               printwhat (count, opt.ntry);
1228               continue;
1229             }
1230         }
1231       else                      /* now hstat.res can only be -1 */
1232         {
1233           if (hstat.contlen == -1)
1234             {
1235               logprintf (LOG_VERBOSE,
1236                          _("%s (%s) - Read error at byte %ld (%s)."),
1237                          tms, tmrate, hstat.len, strerror (errno));
1238               printwhat (count, opt.ntry);
1239               continue;
1240             }
1241           else                  /* hstat.res == -1 and contlen is given */
1242             {
1243               logprintf (LOG_VERBOSE,
1244                          _("%s (%s) - Read error at byte %ld/%ld (%s). "),
1245                          tms, tmrate, hstat.len, hstat.contlen,
1246                          strerror (errno));
1247               printwhat (count, opt.ntry);
1248               continue;
1249             }
1250         }
1251       /* not reached */
1252       break;
1253     }
1254   while (!opt.ntry || (count < opt.ntry));
1255   return TRYLIMEXC;
1256 }
1257 \f
1258 /* Converts struct tm to time_t, assuming the data in tm is UTC rather
1259    than local timezone (mktime assumes the latter).
1260
1261    Contributed by Roger Beeman <beeman@cisco.com>, with the help of
1262    Mark Baushke <mdb@cisco.com> and the rest of the Gurus at CISCO.  */
1263 static time_t
1264 mktime_from_utc (struct tm *t)
1265 {
1266   time_t tl, tb;
1267
1268   tl = mktime (t);
1269   if (tl == -1)
1270     return -1;
1271   tb = mktime (gmtime (&tl));
1272   return (tl <= tb ? (tl + (tl - tb)) : (tl - (tb - tl)));
1273 }
1274
1275 /* Check whether the result of strptime() indicates success.
1276    strptime() returns the pointer to how far it got to in the string.
1277    The processing has been successful if the string is at `GMT' or
1278    `+X', or at the end of the string.
1279
1280    In extended regexp parlance, the function returns 1 if P matches
1281    "^ *(GMT|[+-][0-9]|$)", 0 otherwise.  P being NULL (a valid result of
1282    strptime()) is considered a failure and 0 is returned.  */
1283 static int
1284 check_end (char *p)
1285 {
1286   if (!p)
1287     return 0;
1288   while (ISSPACE (*p))
1289     ++p;
1290   if (!*p
1291       || (p[0] == 'G' && p[1] == 'M' && p[2] == 'T')
1292       || ((p[0] == '+' || p[1] == '-') && ISDIGIT (p[1])))
1293     return 1;
1294   else
1295     return 0;
1296 }
1297
1298 /* Convert TIME_STRING time to time_t.  TIME_STRING can be in any of
1299    the three formats RFC2068 allows the HTTP servers to emit --
1300    RFC1123-date, RFC850-date or asctime-date.  Timezones are ignored,
1301    and should be GMT.
1302
1303    We use strptime() to recognize various dates, which makes it a
1304    little bit slacker than the RFC1123/RFC850/asctime (e.g. it always
1305    allows shortened dates and months, one-digit days, etc.).  It also
1306    allows more than one space anywhere where the specs require one SP.
1307    The routine should probably be even more forgiving (as recommended
1308    by RFC2068), but I do not have the time to write one.
1309
1310    Return the computed time_t representation, or -1 if all the
1311    schemes fail.
1312
1313    Needless to say, what we *really* need here is something like
1314    Marcus Hennecke's atotm(), which is forgiving, fast, to-the-point,
1315    and does not use strptime().  atotm() is to be found in the sources
1316    of `phttpd', a little-known HTTP server written by Peter Erikson.  */
1317 static time_t
1318 http_atotm (char *time_string)
1319 {
1320   struct tm t;
1321
1322   /* Roger Beeman says: "This function dynamically allocates struct tm
1323      t, but does no initialization.  The only field that actually
1324      needs initialization is tm_isdst, since the others will be set by
1325      strptime.  Since strptime does not set tm_isdst, it will return
1326      the data structure with whatever data was in tm_isdst to begin
1327      with.  For those of us in timezones where DST can occur, there
1328      can be a one hour shift depending on the previous contents of the
1329      data area where the data structure is allocated."  */
1330   t.tm_isdst = -1;
1331
1332   /* Note that under foreign locales Solaris strptime() fails to
1333      recognize English dates, which renders this function useless.  I
1334      assume that other non-GNU strptime's are plagued by the same
1335      disease.  We solve this by setting only LC_MESSAGES in
1336      i18n_initialize(), instead of LC_ALL.
1337
1338      Another solution could be to temporarily set locale to C, invoke
1339      strptime(), and restore it back.  This is slow and dirty,
1340      however, and locale support other than LC_MESSAGES can mess other
1341      things, so I rather chose to stick with just setting LC_MESSAGES.
1342
1343      Also note that none of this is necessary under GNU strptime(),
1344      because it recognizes both international and local dates.  */
1345
1346   /* NOTE: We don't use `%n' for white space, as OSF's strptime uses
1347      it to eat all white space up to (and including) a newline, and
1348      the function fails if there is no newline (!).
1349
1350      Let's hope all strptime() implementations use ` ' to skip *all*
1351      whitespace instead of just one (it works that way on all the
1352      systems I've tested it on).  */
1353
1354   /* RFC1123: Thu, 29 Jan 1998 22:12:57 */
1355   if (check_end (strptime (time_string, "%a, %d %b %Y %T", &t)))
1356     return mktime_from_utc (&t);
1357   /* RFC850:  Thu, 29-Jan-98 22:12:57 */
1358   if (check_end (strptime (time_string, "%a, %d-%b-%y %T", &t)))
1359     return mktime_from_utc (&t);
1360   /* asctime: Thu Jan 29 22:12:57 1998 */
1361   if (check_end (strptime (time_string, "%a %b %d %T %Y", &t)))
1362     return mktime_from_utc (&t);
1363   /* Failure.  */
1364   return -1;
1365 }
1366 \f
1367 /* Authorization support: We support two authorization schemes:
1368
1369    * `Basic' scheme, consisting of base64-ing USER:PASSWORD string;
1370
1371    * `Digest' scheme, added by Junio Hamano <junio@twinsun.com>,
1372    consisting of answering to the server's challenge with the proper
1373    MD5 digests.  */
1374
1375 /* How many bytes it will take to store LEN bytes in base64.  */
1376 #define BASE64_LENGTH(len) (4 * (((len) + 2) / 3))
1377
1378 /* Encode the string S of length LENGTH to base64 format and place it
1379    to STORE.  STORE will be 0-terminated, and must point to a writable
1380    buffer of at least 1+BASE64_LENGTH(length) bytes.  */
1381 static void
1382 base64_encode (const char *s, char *store, int length)
1383 {
1384   /* Conversion table.  */
1385   static char tbl[64] = {
1386     'A','B','C','D','E','F','G','H',
1387     'I','J','K','L','M','N','O','P',
1388     'Q','R','S','T','U','V','W','X',
1389     'Y','Z','a','b','c','d','e','f',
1390     'g','h','i','j','k','l','m','n',
1391     'o','p','q','r','s','t','u','v',
1392     'w','x','y','z','0','1','2','3',
1393     '4','5','6','7','8','9','+','/'
1394   };
1395   int i;
1396   unsigned char *p = (unsigned char *)store;
1397
1398   /* Transform the 3x8 bits to 4x6 bits, as required by base64.  */
1399   for (i = 0; i < length; i += 3)
1400     {
1401       *p++ = tbl[s[0] >> 2];
1402       *p++ = tbl[((s[0] & 3) << 4) + (s[1] >> 4)];
1403       *p++ = tbl[((s[1] & 0xf) << 2) + (s[2] >> 6)];
1404       *p++ = tbl[s[2] & 0x3f];
1405       s += 3;
1406     }
1407   /* Pad the result if necessary...  */
1408   if (i == length + 1)
1409     *(p - 1) = '=';
1410   else if (i == length + 2)
1411     *(p - 1) = *(p - 2) = '=';
1412   /* ...and zero-terminate it.  */
1413   *p = '\0';
1414 }
1415
1416 /* Create the authentication header contents for the `Basic' scheme.
1417    This is done by encoding the string `USER:PASS' in base64 and
1418    prepending `HEADER: Basic ' to it.  */
1419 static char *
1420 basic_authentication_encode (const char *user, const char *passwd,
1421                              const char *header)
1422 {
1423   char *t1, *t2, *res;
1424   int len1 = strlen (user) + 1 + strlen (passwd);
1425   int len2 = BASE64_LENGTH (len1);
1426
1427   t1 = (char *)alloca (len1 + 1);
1428   sprintf (t1, "%s:%s", user, passwd);
1429   t2 = (char *)alloca (1 + len2);
1430   base64_encode (t1, t2, len1);
1431   res = (char *)malloc (len2 + 11 + strlen (header));
1432   sprintf (res, "%s: Basic %s\r\n", header, t2);
1433
1434   return res;
1435 }
1436
1437 #ifdef USE_DIGEST
1438 /* Parse HTTP `WWW-Authenticate:' header.  AU points to the beginning
1439    of a field in such a header.  If the field is the one specified by
1440    ATTR_NAME ("realm", "opaque", and "nonce" are used by the current
1441    digest authorization code), extract its value in the (char*)
1442    variable pointed by RET.  Returns negative on a malformed header,
1443    or number of bytes that have been parsed by this call.  */
1444 static int
1445 extract_header_attr (const char *au, const char *attr_name, char **ret)
1446 {
1447   const char *cp, *ep;
1448
1449   ep = cp = au;
1450
1451   if (strncmp (cp, attr_name, strlen (attr_name)) == 0)
1452     {
1453       cp += strlen (attr_name);
1454       if (!*cp)
1455         return -1;
1456       cp += skip_lws (cp);
1457       if (*cp != '=')
1458         return -1;
1459       if (!*++cp)
1460         return -1;
1461       cp += skip_lws (cp);
1462       if (*cp != '\"')
1463         return -1;
1464       if (!*++cp)
1465         return -1;
1466       for (ep = cp; *ep && *ep != '\"'; ep++)
1467         ;
1468       if (!*ep)
1469         return -1;
1470       FREE_MAYBE (*ret);
1471       *ret = strdupdelim (cp, ep);
1472       return ep - au + 1;
1473     }
1474   else
1475     return 0;
1476 }
1477
1478 /* Response value needs to be in lowercase, so we cannot use HEXD2ASC
1479    from url.h.  See RFC 2069 2.1.2 for the syntax of response-digest.  */
1480 #define HEXD2asc(x) (((x) < 10) ? ((x) + '0') : ((x) - 10 + 'a'))
1481
1482 /* Dump the hexadecimal representation of HASH to BUF.  HASH should be
1483    an array of 16 bytes containing the hash keys, and BUF should be a
1484    buffer of 33 writable characters (32 for hex digits plus one for
1485    zero termination).  */
1486 static void
1487 dump_hash (unsigned char *buf, const unsigned char *hash)
1488 {
1489   int i;
1490
1491   for (i = 0; i < MD5_HASHLEN; i++, hash++)
1492     {
1493       *buf++ = HEXD2asc (*hash >> 4);
1494       *buf++ = HEXD2asc (*hash & 0xf);
1495     }
1496   *buf = '\0';
1497 }
1498
1499 /* Take the line apart to find the challenge, and compose a digest
1500    authorization header.  See RFC2069 section 2.1.2.  */
1501 char *
1502 digest_authentication_encode (const char *au, const char *user,
1503                               const char *passwd, const char *method,
1504                               const char *path)
1505 {
1506   static char *realm, *opaque, *nonce;
1507   static struct {
1508     const char *name;
1509     char **variable;
1510   } options[] = {
1511     { "realm", &realm },
1512     { "opaque", &opaque },
1513     { "nonce", &nonce }
1514   };
1515   char *res;
1516
1517   realm = opaque = nonce = NULL;
1518
1519   au += 6;                      /* skip over `Digest' */
1520   while (*au)
1521     {
1522       int i;
1523
1524       au += skip_lws (au);
1525       for (i = 0; i < ARRAY_SIZE (options); i++)
1526         {
1527           int skip = extract_header_attr (au, options[i].name,
1528                                           options[i].variable);
1529           if (skip < 0)
1530             {
1531               FREE_MAYBE (realm);
1532               FREE_MAYBE (opaque);
1533               FREE_MAYBE (nonce);
1534               return NULL;
1535             }
1536           else if (skip)
1537             {
1538               au += skip;
1539               break;
1540             }
1541         }
1542       if (i == ARRAY_SIZE (options))
1543         {
1544           while (*au && *au != '=')
1545             au++;
1546           if (*au && *++au)
1547             {
1548               au += skip_lws (au);
1549               if (*au == '\"')
1550                 {
1551                   au++;
1552                   while (*au && *au != '\"')
1553                     au++;
1554                   if (*au)
1555                     au++;
1556                 }
1557             }
1558         }
1559       while (*au && *au != ',')
1560         au++;
1561       if (*au)
1562         au++;
1563     }
1564   if (!realm || !nonce || !user || !passwd || !path || !method)
1565     {
1566       FREE_MAYBE (realm);
1567       FREE_MAYBE (opaque);
1568       FREE_MAYBE (nonce);
1569       return NULL;
1570     }
1571
1572   /* Calculate the digest value.  */
1573   {
1574     struct md5_ctx ctx;
1575     unsigned char hash[MD5_HASHLEN];
1576     unsigned char a1buf[MD5_HASHLEN * 2 + 1], a2buf[MD5_HASHLEN * 2 + 1];
1577     unsigned char response_digest[MD5_HASHLEN * 2 + 1];
1578
1579     /* A1BUF = H(user ":" realm ":" password) */
1580     md5_init_ctx (&ctx);
1581     md5_process_bytes (user, strlen (user), &ctx);
1582     md5_process_bytes (":", 1, &ctx);
1583     md5_process_bytes (realm, strlen (realm), &ctx);
1584     md5_process_bytes (":", 1, &ctx);
1585     md5_process_bytes (passwd, strlen (passwd), &ctx);
1586     md5_finish_ctx (&ctx, hash);
1587     dump_hash (a1buf, hash);
1588
1589     /* A2BUF = H(method ":" path) */
1590     md5_init_ctx (&ctx);
1591     md5_process_bytes (method, strlen (method), &ctx);
1592     md5_process_bytes (":", 1, &ctx);
1593     md5_process_bytes (path, strlen (path), &ctx);
1594     md5_finish_ctx (&ctx, hash);
1595     dump_hash (a2buf, hash);
1596
1597     /* RESPONSE_DIGEST = H(A1BUF ":" nonce ":" A2BUF) */
1598     md5_init_ctx (&ctx);
1599     md5_process_bytes (a1buf, MD5_HASHLEN * 2, &ctx);
1600     md5_process_bytes (":", 1, &ctx);
1601     md5_process_bytes (nonce, strlen (nonce), &ctx);
1602     md5_process_bytes (":", 1, &ctx);
1603     md5_process_bytes (a2buf, MD5_HASHLEN * 2, &ctx);
1604     md5_finish_ctx (&ctx, hash);
1605     dump_hash (response_digest, hash);
1606
1607     res = (char*) xmalloc (strlen (user)
1608                            + strlen (user)
1609                            + strlen (realm)
1610                            + strlen (nonce)
1611                            + strlen (path)
1612                            + 2 * MD5_HASHLEN /*strlen (response_digest)*/
1613                            + (opaque ? strlen (opaque) : 0)
1614                            + 128);
1615     sprintf (res, "Authorization: Digest \
1616 username=\"%s\", realm=\"%s\", nonce=\"%s\", uri=\"%s\", response=\"%s\"",
1617              user, realm, nonce, path, response_digest);
1618     if (opaque)
1619       {
1620         char *p = res + strlen (res);
1621         strcat (p, ", opaque=\"");
1622         strcat (p, opaque);
1623         strcat (p, "\"");
1624       }
1625     strcat (res, "\r\n");
1626   }
1627   return res;
1628 }
1629 #endif /* USE_DIGEST */
1630
1631
1632 #define HACK_O_MATIC(line, string_constant)                             \
1633   (!strncasecmp (line, string_constant, sizeof (string_constant) - 1)   \
1634    && (ISSPACE (line[sizeof (string_constant) - 1])                     \
1635        || !line[sizeof (string_constant) - 1]))
1636
1637 static int
1638 known_authentication_scheme_p (const char *au)
1639 {
1640   return HACK_O_MATIC (au, "Basic") || HACK_O_MATIC (au, "Digest");
1641 }
1642
1643 #undef HACK_O_MATIC
1644
1645 /* Create the HTTP authorization request header.  When the
1646    `WWW-Authenticate' response header is seen, according to the
1647    authorization scheme specified in that header (`Basic' and `Digest'
1648    are supported by the current implementation), produce an
1649    appropriate HTTP authorization request header.  */
1650 static char *
1651 create_authorization_line (const char *au, const char *user,
1652                            const char *passwd, const char *method,
1653                            const char *path)
1654 {
1655   char *wwwauth = NULL;
1656
1657   if (!strncasecmp (au, "Basic", 5))
1658     wwwauth = basic_authentication_encode (user, passwd, "Authorization");
1659 #ifdef USE_DIGEST
1660   else if (!strncasecmp (au, "Digest", 6))
1661     wwwauth = digest_authentication_encode (au, user, passwd, method, path);
1662 #endif /* USE_DIGEST */
1663   return wwwauth;
1664 }