]> sjero.net Git - wget/blob - src/url.c
[svn] Better document the workings of construct_relative().
[wget] / src / url.c
1 /* URL handling.
2    Copyright (C) 1995, 1996, 1997, 2000, 2001, 2003, 2003
3    Free Software Foundation, Inc.
4
5 This file is part of GNU Wget.
6
7 GNU Wget is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 2 of the License, or (at
10 your option) any later version.
11
12 GNU Wget is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 GNU General Public License for more details.
16
17 You should have received a copy of the GNU General Public License
18 along with Wget; if not, write to the Free Software
19 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20
21 In addition, as a special exception, the Free Software Foundation
22 gives permission to link the code of its release of Wget with the
23 OpenSSL project's "OpenSSL" library (or with modified versions of it
24 that use the same license as the "OpenSSL" library), and distribute
25 the linked executables.  You must obey the GNU General Public License
26 in all respects for all of the code used other than "OpenSSL".  If you
27 modify this file, you may extend this exception to your version of the
28 file, but you are not obligated to do so.  If you do not wish to do
29 so, delete this exception statement from your version.  */
30
31 #include <config.h>
32
33 #include <stdio.h>
34 #include <stdlib.h>
35 #ifdef HAVE_STRING_H
36 # include <string.h>
37 #else
38 # include <strings.h>
39 #endif
40 #include <sys/types.h>
41 #ifdef HAVE_UNISTD_H
42 # include <unistd.h>
43 #endif
44 #include <errno.h>
45 #include <assert.h>
46
47 #include "wget.h"
48 #include "utils.h"
49 #include "url.h"
50
51 #ifndef errno
52 extern int errno;
53 #endif
54
55 struct scheme_data
56 {
57   char *leading_string;
58   int default_port;
59   int enabled;
60 };
61
62 /* Supported schemes: */
63 static struct scheme_data supported_schemes[] =
64 {
65   { "http://",  DEFAULT_HTTP_PORT,  1 },
66 #ifdef HAVE_SSL
67   { "https://", DEFAULT_HTTPS_PORT, 1 },
68 #endif
69   { "ftp://",   DEFAULT_FTP_PORT,   1 },
70
71   /* SCHEME_INVALID */
72   { NULL,       -1,                 0 }
73 };
74
75 /* Forward declarations: */
76
77 static int path_simplify PARAMS ((char *));
78 \f
79 /* Support for encoding and decoding of URL strings.  We determine
80    whether a character is unsafe through static table lookup.  This
81    code assumes ASCII character set and 8-bit chars.  */
82
83 enum {
84   /* rfc1738 reserved chars, preserved from encoding.  */
85   urlchr_reserved = 1,
86
87   /* rfc1738 unsafe chars, plus some more.  */
88   urlchr_unsafe   = 2
89 };
90
91 #define urlchr_test(c, mask) (urlchr_table[(unsigned char)(c)] & (mask))
92 #define URL_RESERVED_CHAR(c) urlchr_test(c, urlchr_reserved)
93 #define URL_UNSAFE_CHAR(c) urlchr_test(c, urlchr_unsafe)
94
95 /* Shorthands for the table: */
96 #define R  urlchr_reserved
97 #define U  urlchr_unsafe
98 #define RU R|U
99
100 const static unsigned char urlchr_table[256] =
101 {
102   U,  U,  U,  U,   U,  U,  U,  U,   /* NUL SOH STX ETX  EOT ENQ ACK BEL */
103   U,  U,  U,  U,   U,  U,  U,  U,   /* BS  HT  LF  VT   FF  CR  SO  SI  */
104   U,  U,  U,  U,   U,  U,  U,  U,   /* DLE DC1 DC2 DC3  DC4 NAK SYN ETB */
105   U,  U,  U,  U,   U,  U,  U,  U,   /* CAN EM  SUB ESC  FS  GS  RS  US  */
106   U,  0,  U, RU,   0,  U,  R,  0,   /* SP  !   "   #    $   %   &   '   */
107   0,  0,  0,  R,   0,  0,  0,  R,   /* (   )   *   +    ,   -   .   /   */
108   0,  0,  0,  0,   0,  0,  0,  0,   /* 0   1   2   3    4   5   6   7   */
109   0,  0, RU,  R,   U,  R,  U,  R,   /* 8   9   :   ;    <   =   >   ?   */
110  RU,  0,  0,  0,   0,  0,  0,  0,   /* @   A   B   C    D   E   F   G   */
111   0,  0,  0,  0,   0,  0,  0,  0,   /* H   I   J   K    L   M   N   O   */
112   0,  0,  0,  0,   0,  0,  0,  0,   /* P   Q   R   S    T   U   V   W   */
113   0,  0,  0, RU,   U, RU,  U,  0,   /* X   Y   Z   [    \   ]   ^   _   */
114   U,  0,  0,  0,   0,  0,  0,  0,   /* `   a   b   c    d   e   f   g   */
115   0,  0,  0,  0,   0,  0,  0,  0,   /* h   i   j   k    l   m   n   o   */
116   0,  0,  0,  0,   0,  0,  0,  0,   /* p   q   r   s    t   u   v   w   */
117   0,  0,  0,  U,   U,  U,  U,  U,   /* x   y   z   {    |   }   ~   DEL */
118
119   U, U, U, U,  U, U, U, U,  U, U, U, U,  U, U, U, U,
120   U, U, U, U,  U, U, U, U,  U, U, U, U,  U, U, U, U,
121   U, U, U, U,  U, U, U, U,  U, U, U, U,  U, U, U, U,
122   U, U, U, U,  U, U, U, U,  U, U, U, U,  U, U, U, U,
123
124   U, U, U, U,  U, U, U, U,  U, U, U, U,  U, U, U, U,
125   U, U, U, U,  U, U, U, U,  U, U, U, U,  U, U, U, U,
126   U, U, U, U,  U, U, U, U,  U, U, U, U,  U, U, U, U,
127   U, U, U, U,  U, U, U, U,  U, U, U, U,  U, U, U, U,
128 };
129 #undef R
130 #undef U
131 #undef RU
132
133 /* URL-unescape the string S.
134
135    This is done by transforming the sequences "%HH" to the character
136    represented by the hexadecimal digits HH.  If % is not followed by
137    two hexadecimal digits, it is inserted literally.
138
139    The transformation is done in place.  If you need the original
140    string intact, make a copy before calling this function.  */
141
142 static void
143 url_unescape (char *s)
144 {
145   char *t = s;                  /* t - tortoise */
146   char *h = s;                  /* h - hare     */
147
148   for (; *h; h++, t++)
149     {
150       if (*h != '%')
151         {
152         copychar:
153           *t = *h;
154         }
155       else
156         {
157           /* Do nothing if '%' is not followed by two hex digits. */
158           if (!h[1] || !h[2] || !(ISXDIGIT (h[1]) && ISXDIGIT (h[2])))
159             goto copychar;
160           *t = X2DIGITS_TO_NUM (h[1], h[2]);
161           h += 2;
162         }
163     }
164   *t = '\0';
165 }
166
167 /* The core of url_escape_* functions.  Escapes the characters that
168    match the provided mask in urlchr_table.
169
170    If ALLOW_PASSTHROUGH is non-zero, a string with no unsafe chars
171    will be returned unchanged.  If ALLOW_PASSTHROUGH is zero, a
172    freshly allocated string will be returned in all cases.  */
173
174 static char *
175 url_escape_1 (const char *s, unsigned char mask, int allow_passthrough)
176 {
177   const char *p1;
178   char *p2, *newstr;
179   int newlen;
180   int addition = 0;
181
182   for (p1 = s; *p1; p1++)
183     if (urlchr_test (*p1, mask))
184       addition += 2;            /* Two more characters (hex digits) */
185
186   if (!addition)
187     return allow_passthrough ? (char *)s : xstrdup (s);
188
189   newlen = (p1 - s) + addition;
190   newstr = (char *)xmalloc (newlen + 1);
191
192   p1 = s;
193   p2 = newstr;
194   while (*p1)
195     {
196       /* Quote the characters that match the test mask. */
197       if (urlchr_test (*p1, mask))
198         {
199           unsigned char c = *p1++;
200           *p2++ = '%';
201           *p2++ = XNUM_TO_DIGIT (c >> 4);
202           *p2++ = XNUM_TO_DIGIT (c & 0xf);
203         }
204       else
205         *p2++ = *p1++;
206     }
207   assert (p2 - newstr == newlen);
208   *p2 = '\0';
209
210   return newstr;
211 }
212
213 /* URL-escape the unsafe characters (see urlchr_table) in a given
214    string, returning a freshly allocated string.  */
215
216 char *
217 url_escape (const char *s)
218 {
219   return url_escape_1 (s, urlchr_unsafe, 0);
220 }
221
222 /* URL-escape the unsafe characters (see urlchr_table) in a given
223    string.  If no characters are unsafe, S is returned.  */
224
225 static char *
226 url_escape_allow_passthrough (const char *s)
227 {
228   return url_escape_1 (s, urlchr_unsafe, 1);
229 }
230 \f
231 enum copy_method { CM_DECODE, CM_ENCODE, CM_PASSTHROUGH };
232
233 /* Decide whether to encode, decode, or pass through the char at P.
234    This used to be a macro, but it got a little too convoluted.  */
235 static inline enum copy_method
236 decide_copy_method (const char *p)
237 {
238   if (*p == '%')
239     {
240       if (ISXDIGIT (*(p + 1)) && ISXDIGIT (*(p + 2)))
241         {
242           /* %xx sequence: decode it, unless it would decode to an
243              unsafe or a reserved char; in that case, leave it as
244              is. */
245           char preempt = X2DIGITS_TO_NUM (*(p + 1), *(p + 2));
246           if (URL_UNSAFE_CHAR (preempt) || URL_RESERVED_CHAR (preempt))
247             return CM_PASSTHROUGH;
248           else
249             return CM_DECODE;
250         }
251       else
252         /* Garbled %.. sequence: encode `%'. */
253         return CM_ENCODE;
254     }
255   else if (URL_UNSAFE_CHAR (*p) && !URL_RESERVED_CHAR (*p))
256     return CM_ENCODE;
257   else
258     return CM_PASSTHROUGH;
259 }
260
261 /* Translate a %-escaped (but possibly non-conformant) input string S
262    into a %-escaped (and conformant) output string.  If no characters
263    are encoded or decoded, return the same string S; otherwise, return
264    a freshly allocated string with the new contents.
265
266    After a URL has been run through this function, the protocols that
267    use `%' as the quote character can use the resulting string as-is,
268    while those that don't call url_unescape() to get to the intended
269    data.  This function is also stable: after an input string is
270    transformed the first time, all further transformations of the
271    result yield the same result string.
272
273    Let's discuss why this function is needed.
274
275    Imagine Wget is to retrieve `http://abc.xyz/abc def'.  Since a raw
276    space character would mess up the HTTP request, it needs to be
277    quoted, like this:
278
279        GET /abc%20def HTTP/1.0
280
281    It appears that the unsafe chars need to be quoted, for example
282    with url_escape.  But what if we're requested to download
283    `abc%20def'?  url_escape transforms "%" to "%25", which would leave
284    us with `abc%2520def'.  This is incorrect -- since %-escapes are
285    part of URL syntax, "%20" is the correct way to denote a literal
286    space on the Wget command line.  This leaves us in the conclusion
287    that in that case Wget should not call url_escape, but leave the
288    `%20' as is.
289
290    And what if the requested URI is `abc%20 def'?  If we call
291    url_escape, we end up with `/abc%2520%20def', which is almost
292    certainly not intended.  If we don't call url_escape, we are left
293    with the embedded space and cannot complete the request.  What the
294    user meant was for Wget to request `/abc%20%20def', and this is
295    where reencode_escapes kicks in.
296
297    Wget used to solve this by first decoding %-quotes, and then
298    encoding all the "unsafe" characters found in the resulting string.
299    This was wrong because it didn't preserve certain URL special
300    (reserved) characters.  For instance, URI containing "a%2B+b" (0x2b
301    == '+') would get translated to "a%2B%2Bb" or "a++b" depending on
302    whether we considered `+' reserved (it is).  One of these results
303    is inevitable because by the second step we would lose information
304    on whether the `+' was originally encoded or not.  Both results
305    were wrong because in CGI parameters + means space, while %2B means
306    literal plus.  reencode_escapes correctly translates the above to
307    "a%2B+b", i.e. returns the original string.
308
309    This function uses an algorithm proposed by Anon Sricharoenchai:
310
311    1. Encode all URL_UNSAFE and the "%" that are not followed by 2
312       hexdigits.
313
314    2. Decode all "%XX" except URL_UNSAFE, URL_RESERVED (";/?:@=&") and
315       "+".
316
317    ...except that this code conflates the two steps, and decides
318    whether to encode, decode, or pass through each character in turn.
319    The function still uses two passes, but their logic is the same --
320    the first pass exists merely for the sake of allocation.  Another
321    small difference is that we include `+' to URL_RESERVED.
322
323    Anon's test case:
324
325    "http://abc.xyz/%20%3F%%36%31%25aa% a?a=%61+a%2Ba&b=b%26c%3Dc"
326    ->
327    "http://abc.xyz/%20%3F%2561%25aa%25%20a?a=a+a%2Ba&b=b%26c%3Dc"
328
329    Simpler test cases:
330
331    "foo bar"         -> "foo%20bar"
332    "foo%20bar"       -> "foo%20bar"
333    "foo %20bar"      -> "foo%20%20bar"
334    "foo%%20bar"      -> "foo%25%20bar"       (0x25 == '%')
335    "foo%25%20bar"    -> "foo%25%20bar"
336    "foo%2%20bar"     -> "foo%252%20bar"
337    "foo+bar"         -> "foo+bar"            (plus is reserved!)
338    "foo%2b+bar"      -> "foo%2b+bar"  */
339
340 static char *
341 reencode_escapes (const char *s)
342 {
343   const char *p1;
344   char *newstr, *p2;
345   int oldlen, newlen;
346
347   int encode_count = 0;
348   int decode_count = 0;
349
350   /* First, pass through the string to see if there's anything to do,
351      and to calculate the new length.  */
352   for (p1 = s; *p1; p1++)
353     {
354       switch (decide_copy_method (p1))
355         {
356         case CM_ENCODE:
357           ++encode_count;
358           break;
359         case CM_DECODE:
360           ++decode_count;
361           break;
362         case CM_PASSTHROUGH:
363           break;
364         }
365     }
366
367   if (!encode_count && !decode_count)
368     /* The string is good as it is. */
369     return (char *)s;           /* C const model sucks. */
370
371   oldlen = p1 - s;
372   /* Each encoding adds two characters (hex digits), while each
373      decoding removes two characters.  */
374   newlen = oldlen + 2 * (encode_count - decode_count);
375   newstr = xmalloc (newlen + 1);
376
377   p1 = s;
378   p2 = newstr;
379
380   while (*p1)
381     {
382       switch (decide_copy_method (p1))
383         {
384         case CM_ENCODE:
385           {
386             unsigned char c = *p1++;
387             *p2++ = '%';
388             *p2++ = XNUM_TO_DIGIT (c >> 4);
389             *p2++ = XNUM_TO_DIGIT (c & 0xf);
390           }
391           break;
392         case CM_DECODE:
393           *p2++ = X2DIGITS_TO_NUM (p1[1], p1[2]);
394           p1 += 3;              /* skip %xx */
395           break;
396         case CM_PASSTHROUGH:
397           *p2++ = *p1++;
398         }
399     }
400   *p2 = '\0';
401   assert (p2 - newstr == newlen);
402   return newstr;
403 }
404 \f
405 /* Returns the scheme type if the scheme is supported, or
406    SCHEME_INVALID if not.  */
407
408 enum url_scheme
409 url_scheme (const char *url)
410 {
411   int i;
412
413   for (i = 0; supported_schemes[i].leading_string; i++)
414     if (0 == strncasecmp (url, supported_schemes[i].leading_string,
415                           strlen (supported_schemes[i].leading_string)))
416       {
417         if (supported_schemes[i].enabled)
418           return (enum url_scheme) i;
419         else
420           return SCHEME_INVALID;
421       }
422
423   return SCHEME_INVALID;
424 }
425
426 #define SCHEME_CHAR(ch) (ISALNUM (ch) || (ch) == '-' || (ch) == '+')
427
428 /* Return 1 if the URL begins with any "scheme", 0 otherwise.  As
429    currently implemented, it returns true if URL begins with
430    [-+a-zA-Z0-9]+: .  */
431
432 int
433 url_has_scheme (const char *url)
434 {
435   const char *p = url;
436
437   /* The first char must be a scheme char. */
438   if (!*p || !SCHEME_CHAR (*p))
439     return 0;
440   ++p;
441   /* Followed by 0 or more scheme chars. */
442   while (*p && SCHEME_CHAR (*p))
443     ++p;
444   /* Terminated by ':'. */
445   return *p == ':';
446 }
447
448 int
449 scheme_default_port (enum url_scheme scheme)
450 {
451   return supported_schemes[scheme].default_port;
452 }
453
454 void
455 scheme_disable (enum url_scheme scheme)
456 {
457   supported_schemes[scheme].enabled = 0;
458 }
459
460 /* Skip the username and password, if present here.  The function
461    should *not* be called with the complete URL, but with the part
462    right after the scheme.
463
464    If no username and password are found, return 0.  */
465
466 static int
467 url_skip_credentials (const char *url)
468 {
469   /* Look for '@' that comes before terminators, such as '/', '?',
470      '#', or ';'.  */
471   const char *p = (const char *)strpbrk (url, "@/?#;");
472   if (!p || *p != '@')
473     return 0;
474   return p + 1 - url;
475 }
476
477 /* Parse credentials contained in [BEG, END).  The region is expected
478    to have come from a URL and is unescaped.  */
479
480 static int
481 parse_credentials (const char *beg, const char *end, char **user, char **passwd)
482 {
483   char *colon;
484   const char *userend;
485
486   if (beg == end)
487     return 0;                   /* empty user name */
488
489   colon = memchr (beg, ':', end - beg);
490   if (colon == beg)
491     return 0;                   /* again empty user name */
492
493   if (colon)
494     {
495       *passwd = strdupdelim (colon + 1, end);
496       userend = colon;
497       url_unescape (*passwd);
498     }
499   else
500     {
501       *passwd = NULL;
502       userend = end;
503     }
504   *user = strdupdelim (beg, userend);
505   url_unescape (*user);
506   return 1;
507 }
508
509 /* Used by main.c: detect URLs written using the "shorthand" URL forms
510    popularized by Netscape and NcFTP.  HTTP shorthands look like this:
511
512    www.foo.com[:port]/dir/file   -> http://www.foo.com[:port]/dir/file
513    www.foo.com[:port]            -> http://www.foo.com[:port]
514
515    FTP shorthands look like this:
516
517    foo.bar.com:dir/file          -> ftp://foo.bar.com/dir/file
518    foo.bar.com:/absdir/file      -> ftp://foo.bar.com//absdir/file
519
520    If the URL needs not or cannot be rewritten, return NULL.  */
521
522 char *
523 rewrite_shorthand_url (const char *url)
524 {
525   const char *p;
526
527   if (url_has_scheme (url))
528     return NULL;
529
530   /* Look for a ':' or '/'.  The former signifies NcFTP syntax, the
531      latter Netscape.  */
532   for (p = url; *p && *p != ':' && *p != '/'; p++)
533     ;
534
535   if (p == url)
536     return NULL;
537
538   if (*p == ':')
539     {
540       const char *pp;
541       char *res;
542       /* If the characters after the colon and before the next slash
543          or end of string are all digits, it's HTTP.  */
544       int digits = 0;
545       for (pp = p + 1; ISDIGIT (*pp); pp++)
546         ++digits;
547       if (digits > 0 && (*pp == '/' || *pp == '\0'))
548         goto http;
549
550       /* Prepend "ftp://" to the entire URL... */
551       res = xmalloc (6 + strlen (url) + 1);
552       sprintf (res, "ftp://%s", url);
553       /* ...and replace ':' with '/'. */
554       res[6 + (p - url)] = '/';
555       return res;
556     }
557   else
558     {
559       char *res;
560     http:
561       /* Just prepend "http://" to what we have. */
562       res = xmalloc (7 + strlen (url) + 1);
563       sprintf (res, "http://%s", url);
564       return res;
565     }
566 }
567 \f
568 static void split_path PARAMS ((const char *, char **, char **));
569
570 /* Like strpbrk, with the exception that it returns the pointer to the
571    terminating zero (end-of-string aka "eos") if no matching character
572    is found.
573
574    Although I normally balk at Gcc-specific optimizations, it probably
575    makes sense here: glibc has optimizations that detect strpbrk being
576    called with literal string as ACCEPT and inline the search.  That
577    optimization is defeated if strpbrk is hidden within the call to
578    another function.  (And no, making strpbrk_or_eos inline doesn't
579    help because the check for literal accept is in the
580    preprocessor.)  */
581
582 #ifdef __GNUC__
583
584 #define strpbrk_or_eos(s, accept) ({            \
585   char *SOE_p = strpbrk (s, accept);            \
586   if (!SOE_p)                                   \
587     SOE_p = (char *)s + strlen (s);             \
588   SOE_p;                                        \
589 })
590
591 #else  /* not __GNUC__ */
592
593 static char *
594 strpbrk_or_eos (const char *s, const char *accept)
595 {
596   char *p = strpbrk (s, accept);
597   if (!p)
598     p = (char *)s + strlen (s);
599   return p;
600 }
601 #endif
602
603 /* Turn STR into lowercase; return non-zero if a character was
604    actually changed. */
605
606 static int
607 lowercase_str (char *str)
608 {
609   int change = 0;
610   for (; *str; str++)
611     if (ISUPPER (*str))
612       {
613         change = 1;
614         *str = TOLOWER (*str);
615       }
616   return change;
617 }
618
619 static char *parse_errors[] = {
620 #define PE_NO_ERROR                     0
621   N_("No error"),
622 #define PE_UNSUPPORTED_SCHEME           1
623   N_("Unsupported scheme"),
624 #define PE_EMPTY_HOST                   2
625   N_("Empty host"),
626 #define PE_BAD_PORT_NUMBER              3
627   N_("Bad port number"),
628 #define PE_INVALID_USER_NAME            4
629   N_("Invalid user name"),
630 #define PE_UNTERMINATED_IPV6_ADDRESS    5
631   N_("Unterminated IPv6 numeric address"),
632 #define PE_IPV6_NOT_SUPPORTED           6
633   N_("IPv6 addresses not supported"),
634 #define PE_INVALID_IPV6_ADDRESS         7
635   N_("Invalid IPv6 numeric address")
636 };
637
638 #ifdef ENABLE_IPV6
639 /* The following two functions were adapted from glibc. */
640
641 static int
642 is_valid_ipv4_address (const char *str, const char *end)
643 {
644   int saw_digit = 0;
645   int octets = 0;
646   int val = 0;
647
648   while (str < end)
649     {
650       int ch = *str++;
651
652       if (ch >= '0' && ch <= '9')
653         {
654           val = val * 10 + (ch - '0');
655
656           if (val > 255)
657             return 0;
658           if (saw_digit == 0)
659             {
660               if (++octets > 4)
661                 return 0;
662               saw_digit = 1;
663             }
664         }
665       else if (ch == '.' && saw_digit == 1)
666         {
667           if (octets == 4)
668             return 0;
669           val = 0;
670           saw_digit = 0;
671         }
672       else
673         return 0;
674     }
675   if (octets < 4)
676     return 0;
677   
678   return 1;
679 }
680
681 static int
682 is_valid_ipv6_address (const char *str, const char *end)
683 {
684   enum {
685     NS_INADDRSZ  = 4,
686     NS_IN6ADDRSZ = 16,
687     NS_INT16SZ   = 2
688   };
689
690   const char *curtok;
691   int tp;
692   const char *colonp;
693   int saw_xdigit;
694   unsigned int val;
695
696   tp = 0;
697   colonp = NULL;
698
699   if (str == end)
700     return 0;
701   
702   /* Leading :: requires some special handling. */
703   if (*str == ':')
704     {
705       ++str;
706       if (str == end || *str != ':')
707         return 0;
708     }
709
710   curtok = str;
711   saw_xdigit = 0;
712   val = 0;
713
714   while (str < end)
715     {
716       int ch = *str++;
717
718       /* if ch is a number, add it to val. */
719       if (ISXDIGIT (ch))
720         {
721           val <<= 4;
722           val |= XDIGIT_TO_NUM (ch);
723           if (val > 0xffff)
724             return 0;
725           saw_xdigit = 1;
726           continue;
727         }
728
729       /* if ch is a colon ... */
730       if (ch == ':')
731         {
732           curtok = str;
733           if (saw_xdigit == 0)
734             {
735               if (colonp != NULL)
736                 return 0;
737               colonp = str + tp;
738               continue;
739             }
740           else if (str == end)
741             return 0;
742           if (tp > NS_IN6ADDRSZ - NS_INT16SZ)
743             return 0;
744           tp += NS_INT16SZ;
745           saw_xdigit = 0;
746           val = 0;
747           continue;
748         }
749
750       /* if ch is a dot ... */
751       if (ch == '.' && (tp <= NS_IN6ADDRSZ - NS_INADDRSZ)
752           && is_valid_ipv4_address (curtok, end) == 1)
753         {
754           tp += NS_INADDRSZ;
755           saw_xdigit = 0;
756           break;
757         }
758     
759       return 0;
760     }
761
762   if (saw_xdigit == 1)
763     {
764       if (tp > NS_IN6ADDRSZ - NS_INT16SZ) 
765         return 0;
766       tp += NS_INT16SZ;
767     }
768
769   if (colonp != NULL)
770     {
771       if (tp == NS_IN6ADDRSZ) 
772         return 0;
773       tp = NS_IN6ADDRSZ;
774     }
775
776   if (tp != NS_IN6ADDRSZ)
777     return 0;
778
779   return 1;
780 }
781 #endif
782
783 /* Parse a URL.
784
785    Return a new struct url if successful, NULL on error.  In case of
786    error, and if ERROR is not NULL, also set *ERROR to the appropriate
787    error code. */
788 struct url *
789 url_parse (const char *url, int *error)
790 {
791   struct url *u;
792   const char *p;
793   int path_modified, host_modified;
794
795   enum url_scheme scheme;
796
797   const char *uname_b,     *uname_e;
798   const char *host_b,      *host_e;
799   const char *path_b,      *path_e;
800   const char *params_b,    *params_e;
801   const char *query_b,     *query_e;
802   const char *fragment_b,  *fragment_e;
803
804   int port;
805   char *user = NULL, *passwd = NULL;
806
807   char *url_encoded = NULL;
808
809   int error_code;
810
811   scheme = url_scheme (url);
812   if (scheme == SCHEME_INVALID)
813     {
814       error_code = PE_UNSUPPORTED_SCHEME;
815       goto error;
816     }
817
818   url_encoded = reencode_escapes (url);
819   p = url_encoded;
820
821   p += strlen (supported_schemes[scheme].leading_string);
822   uname_b = p;
823   p += url_skip_credentials (p);
824   uname_e = p;
825
826   /* scheme://user:pass@host[:port]... */
827   /*                    ^              */
828
829   /* We attempt to break down the URL into the components path,
830      params, query, and fragment.  They are ordered like this:
831
832        scheme://host[:port][/path][;params][?query][#fragment]  */
833
834   params_b   = params_e   = NULL;
835   query_b    = query_e    = NULL;
836   fragment_b = fragment_e = NULL;
837
838   host_b = p;
839
840   if (*p == '[')
841     {
842       /* Handle IPv6 address inside square brackets.  Ideally we'd
843          just look for the terminating ']', but rfc2732 mandates
844          rejecting invalid IPv6 addresses.  */
845
846       /* The address begins after '['. */
847       host_b = p + 1;
848       host_e = strchr (host_b, ']');
849
850       if (!host_e)
851         {
852           error_code = PE_UNTERMINATED_IPV6_ADDRESS;
853           goto error;
854         }
855
856 #ifdef ENABLE_IPV6
857       /* Check if the IPv6 address is valid. */
858       if (!is_valid_ipv6_address(host_b, host_e))
859         {
860           error_code = PE_INVALID_IPV6_ADDRESS;
861           goto error;
862         }
863
864       /* Continue parsing after the closing ']'. */
865       p = host_e + 1;
866 #else
867       error_code = PE_IPV6_NOT_SUPPORTED;
868       goto error;
869 #endif
870     }
871   else
872     {
873       p = strpbrk_or_eos (p, ":/;?#");
874       host_e = p;
875     }
876
877   if (host_b == host_e)
878     {
879       error_code = PE_EMPTY_HOST;
880       goto error;
881     }
882
883   port = scheme_default_port (scheme);
884   if (*p == ':')
885     {
886       const char *port_b, *port_e, *pp;
887
888       /* scheme://host:port/tralala */
889       /*              ^             */
890       ++p;
891       port_b = p;
892       p = strpbrk_or_eos (p, "/;?#");
893       port_e = p;
894
895       if (port_b == port_e)
896         {
897           /* http://host:/whatever */
898           /*             ^         */
899           error_code = PE_BAD_PORT_NUMBER;
900           goto error;
901         }
902
903       for (port = 0, pp = port_b; pp < port_e; pp++)
904         {
905           if (!ISDIGIT (*pp))
906             {
907               /* http://host:12randomgarbage/blah */
908               /*               ^                  */
909               error_code = PE_BAD_PORT_NUMBER;
910               goto error;
911             }
912           
913           port = 10 * port + (*pp - '0');
914         }
915     }
916
917   if (*p == '/')
918     {
919       ++p;
920       path_b = p;
921       p = strpbrk_or_eos (p, ";?#");
922       path_e = p;
923     }
924   else
925     {
926       /* Path is not allowed not to exist. */
927       path_b = path_e = p;
928     }
929
930   if (*p == ';')
931     {
932       ++p;
933       params_b = p;
934       p = strpbrk_or_eos (p, "?#");
935       params_e = p;
936     }
937   if (*p == '?')
938     {
939       ++p;
940       query_b = p;
941       p = strpbrk_or_eos (p, "#");
942       query_e = p;
943
944       /* Hack that allows users to use '?' (a wildcard character) in
945          FTP URLs without it being interpreted as a query string
946          delimiter.  */
947       if (scheme == SCHEME_FTP)
948         {
949           query_b = query_e = NULL;
950           path_e = p;
951         }
952     }
953   if (*p == '#')
954     {
955       ++p;
956       fragment_b = p;
957       p += strlen (p);
958       fragment_e = p;
959     }
960   assert (*p == 0);
961
962   if (uname_b != uname_e)
963     {
964       /* http://user:pass@host */
965       /*        ^         ^    */
966       /*     uname_b   uname_e */
967       if (!parse_credentials (uname_b, uname_e - 1, &user, &passwd))
968         {
969           error_code = PE_INVALID_USER_NAME;
970           goto error;
971         }
972     }
973
974   u = (struct url *)xmalloc (sizeof (struct url));
975   memset (u, 0, sizeof (*u));
976
977   u->scheme = scheme;
978   u->host   = strdupdelim (host_b, host_e);
979   u->port   = port;
980   u->user   = user;
981   u->passwd = passwd;
982
983   u->path = strdupdelim (path_b, path_e);
984   path_modified = path_simplify (u->path);
985   split_path (u->path, &u->dir, &u->file);
986
987   host_modified = lowercase_str (u->host);
988
989   if (params_b)
990     u->params = strdupdelim (params_b, params_e);
991   if (query_b)
992     u->query = strdupdelim (query_b, query_e);
993   if (fragment_b)
994     u->fragment = strdupdelim (fragment_b, fragment_e);
995
996   if (path_modified || u->fragment || host_modified || path_b == path_e)
997     {
998       /* If we suspect that a transformation has rendered what
999          url_string might return different from URL_ENCODED, rebuild
1000          u->url using url_string.  */
1001       u->url = url_string (u, 0);
1002
1003       if (url_encoded != url)
1004         xfree ((char *) url_encoded);
1005     }
1006   else
1007     {
1008       if (url_encoded == url)
1009         u->url = xstrdup (url);
1010       else
1011         u->url = url_encoded;
1012     }
1013   url_encoded = NULL;
1014
1015   return u;
1016
1017  error:
1018   /* Cleanup in case of error: */
1019   if (url_encoded && url_encoded != url)
1020     xfree (url_encoded);
1021
1022   /* Transmit the error code to the caller, if the caller wants to
1023      know.  */
1024   if (error)
1025     *error = error_code;
1026   return NULL;
1027 }
1028
1029 /* Return the error message string from ERROR_CODE, which should have
1030    been retrieved from url_parse.  The error message is translated.  */
1031
1032 const char *
1033 url_error (int error_code)
1034 {
1035   assert (error_code >= 0 && error_code < countof (parse_errors));
1036   return _(parse_errors[error_code]);
1037 }
1038
1039 /* Split PATH into DIR and FILE.  PATH comes from the URL and is
1040    expected to be URL-escaped.
1041
1042    The path is split into directory (the part up to the last slash)
1043    and file (the part after the last slash), which are subsequently
1044    unescaped.  Examples:
1045
1046    PATH                 DIR           FILE
1047    "foo/bar/baz"        "foo/bar"     "baz"
1048    "foo/bar/"           "foo/bar"     ""
1049    "foo"                ""            "foo"
1050    "foo/bar/baz%2fqux"  "foo/bar"     "baz/qux" (!)
1051
1052    DIR and FILE are freshly allocated.  */
1053
1054 static void
1055 split_path (const char *path, char **dir, char **file)
1056 {
1057   char *last_slash = strrchr (path, '/');
1058   if (!last_slash)
1059     {
1060       *dir = xstrdup ("");
1061       *file = xstrdup (path);
1062     }
1063   else
1064     {
1065       *dir = strdupdelim (path, last_slash);
1066       *file = xstrdup (last_slash + 1);
1067     }
1068   url_unescape (*dir);
1069   url_unescape (*file);
1070 }
1071
1072 /* Note: URL's "full path" is the path with the query string and
1073    params appended.  The "fragment" (#foo) is intentionally ignored,
1074    but that might be changed.  For example, if the original URL was
1075    "http://host:port/foo/bar/baz;bullshit?querystring#uselessfragment",
1076    the full path will be "/foo/bar/baz;bullshit?querystring".  */
1077
1078 /* Return the length of the full path, without the terminating
1079    zero.  */
1080
1081 static int
1082 full_path_length (const struct url *url)
1083 {
1084   int len = 0;
1085
1086 #define FROB(el) if (url->el) len += 1 + strlen (url->el)
1087
1088   FROB (path);
1089   FROB (params);
1090   FROB (query);
1091
1092 #undef FROB
1093
1094   return len;
1095 }
1096
1097 /* Write out the full path. */
1098
1099 static void
1100 full_path_write (const struct url *url, char *where)
1101 {
1102 #define FROB(el, chr) do {                      \
1103   char *f_el = url->el;                         \
1104   if (f_el) {                                   \
1105     int l = strlen (f_el);                      \
1106     *where++ = chr;                             \
1107     memcpy (where, f_el, l);                    \
1108     where += l;                                 \
1109   }                                             \
1110 } while (0)
1111
1112   FROB (path, '/');
1113   FROB (params, ';');
1114   FROB (query, '?');
1115
1116 #undef FROB
1117 }
1118
1119 /* Public function for getting the "full path".  E.g. if u->path is
1120    "foo/bar" and u->query is "param=value", full_path will be
1121    "/foo/bar?param=value". */
1122
1123 char *
1124 url_full_path (const struct url *url)
1125 {
1126   int length = full_path_length (url);
1127   char *full_path = (char *)xmalloc(length + 1);
1128
1129   full_path_write (url, full_path);
1130   full_path[length] = '\0';
1131
1132   return full_path;
1133 }
1134
1135 /* Escape unsafe and reserved characters, except for the slash
1136    characters.  */
1137
1138 static char *
1139 url_escape_dir (const char *dir)
1140 {
1141   char *newdir = url_escape_1 (dir, urlchr_unsafe | urlchr_reserved, 1);
1142   char *h, *t;
1143   if (newdir == dir)
1144     return (char *)dir;
1145
1146   /* Unescape slashes in NEWDIR. */
1147
1148   h = newdir;                   /* hare */
1149   t = newdir;                   /* tortoise */
1150
1151   for (; *h; h++, t++)
1152     {
1153       /* url_escape_1 having converted '/' to "%2F" exactly. */
1154       if (*h == '%' && h[1] == '2' && h[2] == 'F')
1155         {
1156           *t = '/';
1157           h += 2;
1158         }
1159       else
1160         *t = *h;
1161     }
1162   *t = '\0';
1163
1164   return newdir;
1165 }
1166
1167 /* Sync u->path and u->url with u->dir and u->file.  Called after
1168    u->file or u->dir have been changed, typically by the FTP code.  */
1169
1170 static void
1171 sync_path (struct url *u)
1172 {
1173   char *newpath, *efile, *edir;
1174
1175   xfree (u->path);
1176
1177   /* u->dir and u->file are not escaped.  URL-escape them before
1178      reassembling them into u->path.  That way, if they contain
1179      separators like '?' or even if u->file contains slashes, the
1180      path will be correctly assembled.  (u->file can contain slashes
1181      if the URL specifies it with %2f, or if an FTP server returns
1182      it.)  */
1183   edir = url_escape_dir (u->dir);
1184   efile = url_escape_1 (u->file, urlchr_unsafe | urlchr_reserved, 1);
1185
1186   if (!*edir)
1187     newpath = xstrdup (efile);
1188   else
1189     {
1190       int dirlen = strlen (edir);
1191       int filelen = strlen (efile);
1192
1193       /* Copy "DIR/FILE" to newpath. */
1194       char *p = newpath = xmalloc (dirlen + 1 + filelen + 1);
1195       memcpy (p, edir, dirlen);
1196       p += dirlen;
1197       *p++ = '/';
1198       memcpy (p, efile, filelen);
1199       p += filelen;
1200       *p++ = '\0';
1201     }
1202
1203   u->path = newpath;
1204
1205   if (edir != u->dir)
1206     xfree (edir);
1207   if (efile != u->file)
1208     xfree (efile);
1209
1210   /* Regenerate u->url as well.  */
1211   xfree (u->url);
1212   u->url = url_string (u, 0);
1213 }
1214
1215 /* Mutators.  Code in ftp.c insists on changing u->dir and u->file.
1216    This way we can sync u->path and u->url when they get changed.  */
1217
1218 void
1219 url_set_dir (struct url *url, const char *newdir)
1220 {
1221   xfree (url->dir);
1222   url->dir = xstrdup (newdir);
1223   sync_path (url);
1224 }
1225
1226 void
1227 url_set_file (struct url *url, const char *newfile)
1228 {
1229   xfree (url->file);
1230   url->file = xstrdup (newfile);
1231   sync_path (url);
1232 }
1233
1234 void
1235 url_free (struct url *url)
1236 {
1237   xfree (url->host);
1238   xfree (url->path);
1239   xfree (url->url);
1240
1241   FREE_MAYBE (url->params);
1242   FREE_MAYBE (url->query);
1243   FREE_MAYBE (url->fragment);
1244   FREE_MAYBE (url->user);
1245   FREE_MAYBE (url->passwd);
1246
1247   xfree (url->dir);
1248   xfree (url->file);
1249
1250   xfree (url);
1251 }
1252 \f
1253 /* Create all the necessary directories for PATH (a file).  Calls
1254    mkdirhier() internally.  */
1255 int
1256 mkalldirs (const char *path)
1257 {
1258   const char *p;
1259   char *t;
1260   struct stat st;
1261   int res;
1262
1263   p = path + strlen (path);
1264   for (; *p != '/' && p != path; p--)
1265     ;
1266
1267   /* Don't create if it's just a file.  */
1268   if ((p == path) && (*p != '/'))
1269     return 0;
1270   t = strdupdelim (path, p);
1271
1272   /* Check whether the directory exists.  */
1273   if ((stat (t, &st) == 0))
1274     {
1275       if (S_ISDIR (st.st_mode))
1276         {
1277           xfree (t);
1278           return 0;
1279         }
1280       else
1281         {
1282           /* If the dir exists as a file name, remove it first.  This
1283              is *only* for Wget to work with buggy old CERN http
1284              servers.  Here is the scenario: When Wget tries to
1285              retrieve a directory without a slash, e.g.
1286              http://foo/bar (bar being a directory), CERN server will
1287              not redirect it too http://foo/bar/ -- it will generate a
1288              directory listing containing links to bar/file1,
1289              bar/file2, etc.  Wget will lose because it saves this
1290              HTML listing to a file `bar', so it cannot create the
1291              directory.  To work around this, if the file of the same
1292              name exists, we just remove it and create the directory
1293              anyway.  */
1294           DEBUGP (("Removing %s because of directory danger!\n", t));
1295           unlink (t);
1296         }
1297     }
1298   res = make_directory (t);
1299   if (res != 0)
1300     logprintf (LOG_NOTQUIET, "%s: %s", t, strerror (errno));
1301   xfree (t);
1302   return res;
1303 }
1304 \f
1305 /* Functions for constructing the file name out of URL components.  */
1306
1307 /* A growable string structure, used by url_file_name and friends.
1308    This should perhaps be moved to utils.c.
1309
1310    The idea is to have a convenient and efficient way to construct a
1311    string by having various functions append data to it.  Instead of
1312    passing the obligatory BASEVAR, SIZEVAR and TAILPOS to all the
1313    functions in questions, we pass the pointer to this struct.  */
1314
1315 struct growable {
1316   char *base;
1317   int size;
1318   int tail;
1319 };
1320
1321 /* Ensure that the string can accept APPEND_COUNT more characters past
1322    the current TAIL position.  If necessary, this will grow the string
1323    and update its allocated size.  If the string is already large
1324    enough to take TAIL+APPEND_COUNT characters, this does nothing.  */
1325 #define GROW(g, append_size) do {                                       \
1326   struct growable *G_ = g;                                              \
1327   DO_REALLOC (G_->base, G_->size, G_->tail + append_size, char);        \
1328 } while (0)
1329
1330 /* Return the tail position of the string. */
1331 #define TAIL(r) ((r)->base + (r)->tail)
1332
1333 /* Move the tail position by APPEND_COUNT characters. */
1334 #define TAIL_INCR(r, append_count) ((r)->tail += append_count)
1335
1336 /* Append the string STR to DEST.  NOTICE: the string in DEST is not
1337    terminated.  */
1338
1339 static void
1340 append_string (const char *str, struct growable *dest)
1341 {
1342   int l = strlen (str);
1343   GROW (dest, l);
1344   memcpy (TAIL (dest), str, l);
1345   TAIL_INCR (dest, l);
1346 }
1347
1348 /* Append CH to DEST.  For example, append_char (0, DEST)
1349    zero-terminates DEST.  */
1350
1351 static void
1352 append_char (char ch, struct growable *dest)
1353 {
1354   GROW (dest, 1);
1355   *TAIL (dest) = ch;
1356   TAIL_INCR (dest, 1);
1357 }
1358
1359 enum {
1360   filechr_not_unix    = 1,      /* unusable on Unix, / and \0 */
1361   filechr_not_windows = 2,      /* unusable on Windows, one of \|/<>?:*" */
1362   filechr_control     = 4       /* a control character, e.g. 0-31 */
1363 };
1364
1365 #define FILE_CHAR_TEST(c, mask) (filechr_table[(unsigned char)(c)] & (mask))
1366
1367 /* Shorthands for the table: */
1368 #define U filechr_not_unix
1369 #define W filechr_not_windows
1370 #define C filechr_control
1371
1372 #define UW U|W
1373 #define UWC U|W|C
1374
1375 /* Table of characters unsafe under various conditions (see above).
1376
1377    Arguably we could also claim `%' to be unsafe, since we use it as
1378    the escape character.  If we ever want to be able to reliably
1379    translate file name back to URL, this would become important
1380    crucial.  Right now, it's better to be minimal in escaping.  */
1381
1382 const static unsigned char filechr_table[256] =
1383 {
1384 UWC,  C,  C,  C,   C,  C,  C,  C,   /* NUL SOH STX ETX  EOT ENQ ACK BEL */
1385   C,  C,  C,  C,   C,  C,  C,  C,   /* BS  HT  LF  VT   FF  CR  SO  SI  */
1386   C,  C,  C,  C,   C,  C,  C,  C,   /* DLE DC1 DC2 DC3  DC4 NAK SYN ETB */
1387   C,  C,  C,  C,   C,  C,  C,  C,   /* CAN EM  SUB ESC  FS  GS  RS  US  */
1388   0,  0,  W,  0,   0,  0,  0,  0,   /* SP  !   "   #    $   %   &   '   */
1389   0,  0,  W,  0,   0,  0,  0, UW,   /* (   )   *   +    ,   -   .   /   */
1390   0,  0,  0,  0,   0,  0,  0,  0,   /* 0   1   2   3    4   5   6   7   */
1391   0,  0,  W,  0,   W,  0,  W,  W,   /* 8   9   :   ;    <   =   >   ?   */
1392   0,  0,  0,  0,   0,  0,  0,  0,   /* @   A   B   C    D   E   F   G   */
1393   0,  0,  0,  0,   0,  0,  0,  0,   /* H   I   J   K    L   M   N   O   */
1394   0,  0,  0,  0,   0,  0,  0,  0,   /* P   Q   R   S    T   U   V   W   */
1395   0,  0,  0,  0,   W,  0,  0,  0,   /* X   Y   Z   [    \   ]   ^   _   */
1396   0,  0,  0,  0,   0,  0,  0,  0,   /* `   a   b   c    d   e   f   g   */
1397   0,  0,  0,  0,   0,  0,  0,  0,   /* h   i   j   k    l   m   n   o   */
1398   0,  0,  0,  0,   0,  0,  0,  0,   /* p   q   r   s    t   u   v   w   */
1399   0,  0,  0,  0,   0,  0,  0,  0,   /* x   y   z   {    |   }   ~   DEL */
1400
1401   C, C, C, C,  C, C, C, C,  C, C, C, C,  C, C, C, C, /* 128-143 */
1402   C, C, C, C,  C, C, C, C,  C, C, C, C,  C, C, C, C, /* 144-159 */
1403   0, 0, 0, 0,  0, 0, 0, 0,  0, 0, 0, 0,  0, 0, 0, 0,
1404   0, 0, 0, 0,  0, 0, 0, 0,  0, 0, 0, 0,  0, 0, 0, 0,
1405
1406   0, 0, 0, 0,  0, 0, 0, 0,  0, 0, 0, 0,  0, 0, 0, 0,
1407   0, 0, 0, 0,  0, 0, 0, 0,  0, 0, 0, 0,  0, 0, 0, 0,
1408   0, 0, 0, 0,  0, 0, 0, 0,  0, 0, 0, 0,  0, 0, 0, 0,
1409   0, 0, 0, 0,  0, 0, 0, 0,  0, 0, 0, 0,  0, 0, 0, 0,
1410 };
1411 #undef U
1412 #undef W
1413 #undef C
1414 #undef UW
1415 #undef UWC
1416
1417 /* FN_PORT_SEP is the separator between host and port in file names
1418    for non-standard port numbers.  On Unix this is normally ':', as in
1419    "www.xemacs.org:4001/index.html".  Under Windows, we set it to +
1420    because Windows can't handle ':' in file names.  */
1421 #define FN_PORT_SEP  (opt.restrict_files_os != restrict_windows ? ':' : '+')
1422
1423 /* FN_QUERY_SEP is the separator between the file name and the URL
1424    query, normally '?'.  Since Windows cannot handle '?' as part of
1425    file name, we use '@' instead there.  */
1426 #define FN_QUERY_SEP (opt.restrict_files_os != restrict_windows ? '?' : '@')
1427
1428 /* Quote path element, characters in [b, e), as file name, and append
1429    the quoted string to DEST.  Each character is quoted as per
1430    file_unsafe_char and the corresponding table.  */
1431
1432 static void
1433 append_uri_pathel (const char *b, const char *e, struct growable *dest)
1434 {
1435   char *pathel;
1436   int pathlen;
1437
1438   const char *p;
1439   int quoted, outlen;
1440
1441   int mask;
1442   if (opt.restrict_files_os == restrict_unix)
1443     mask = filechr_not_unix;
1444   else
1445     mask = filechr_not_windows;
1446   if (opt.restrict_files_ctrl)
1447     mask |= filechr_control;
1448
1449   /* Copy [b, e) to PATHEL and URL-unescape it. */
1450   BOUNDED_TO_ALLOCA (b, e, pathel);
1451   url_unescape (pathel);
1452   pathlen = strlen (pathel);
1453
1454   /* Go through PATHEL and check how many characters we'll need to
1455      add for file quoting. */
1456   quoted = 0;
1457   for (p = pathel; *p; p++)
1458     if (FILE_CHAR_TEST (*p, mask))
1459       ++quoted;
1460
1461   /* p - pathel is the string length.  Each quoted char means two
1462      additional characters in the string, hence 2*quoted.  */
1463   outlen = (p - pathel) + (2 * quoted);
1464   GROW (dest, outlen);
1465
1466   if (!quoted)
1467     {
1468       /* If there's nothing to quote, we don't need to go through the
1469          string the second time.  */
1470       memcpy (TAIL (dest), pathel, outlen);
1471     }
1472   else
1473     {
1474       char *q = TAIL (dest);
1475       for (p = pathel; *p; p++)
1476         {
1477           if (!FILE_CHAR_TEST (*p, mask))
1478             *q++ = *p;
1479           else
1480             {
1481               unsigned char ch = *p;
1482               *q++ = '%';
1483               *q++ = XNUM_TO_DIGIT (ch >> 4);
1484               *q++ = XNUM_TO_DIGIT (ch & 0xf);
1485             }
1486         }
1487       assert (q - TAIL (dest) == outlen);
1488     }
1489   TAIL_INCR (dest, outlen);
1490 }
1491
1492 /* Append to DEST the directory structure that corresponds the
1493    directory part of URL's path.  For example, if the URL is
1494    http://server/dir1/dir2/file, this appends "/dir1/dir2".
1495
1496    Each path element ("dir1" and "dir2" in the above example) is
1497    examined, url-unescaped, and re-escaped as file name element.
1498
1499    Additionally, it cuts as many directories from the path as
1500    specified by opt.cut_dirs.  For example, if opt.cut_dirs is 1, it
1501    will produce "bar" for the above example.  For 2 or more, it will
1502    produce "".
1503
1504    Each component of the path is quoted for use as file name.  */
1505
1506 static void
1507 append_dir_structure (const struct url *u, struct growable *dest)
1508 {
1509   char *pathel, *next;
1510   int cut = opt.cut_dirs;
1511
1512   /* Go through the path components, de-URL-quote them, and quote them
1513      (if necessary) as file names.  */
1514
1515   pathel = u->path;
1516   for (; (next = strchr (pathel, '/')) != NULL; pathel = next + 1)
1517     {
1518       if (cut-- > 0)
1519         continue;
1520       if (pathel == next)
1521         /* Ignore empty pathels.  */
1522         continue;
1523
1524       if (dest->tail)
1525         append_char ('/', dest);
1526       append_uri_pathel (pathel, next, dest);
1527     }
1528 }
1529
1530 /* Return a unique file name that matches the given URL as good as
1531    possible.  Does not create directories on the file system.  */
1532
1533 char *
1534 url_file_name (const struct url *u)
1535 {
1536   struct growable fnres;
1537
1538   char *u_file, *u_query;
1539   char *fname, *unique;
1540
1541   fnres.base = NULL;
1542   fnres.size = 0;
1543   fnres.tail = 0;
1544
1545   /* Start with the directory prefix, if specified. */
1546   if (opt.dir_prefix)
1547     append_string (opt.dir_prefix, &fnres);
1548
1549   /* If "dirstruct" is turned on (typically the case with -r), add
1550      the host and port (unless those have been turned off) and
1551      directory structure.  */
1552   if (opt.dirstruct)
1553     {
1554       if (opt.add_hostdir)
1555         {
1556           if (fnres.tail)
1557             append_char ('/', &fnres);
1558           append_string (u->host, &fnres);
1559           if (u->port != scheme_default_port (u->scheme))
1560             {
1561               char portstr[24];
1562               number_to_string (portstr, u->port);
1563               append_char (FN_PORT_SEP, &fnres);
1564               append_string (portstr, &fnres);
1565             }
1566         }
1567
1568       append_dir_structure (u, &fnres);
1569     }
1570
1571   /* Add the file name. */
1572   if (fnres.tail)
1573     append_char ('/', &fnres);
1574   u_file = *u->file ? u->file : "index.html";
1575   append_uri_pathel (u_file, u_file + strlen (u_file), &fnres);
1576
1577   /* Append "?query" to the file name. */
1578   u_query = u->query && *u->query ? u->query : NULL;
1579   if (u_query)
1580     {
1581       append_char (FN_QUERY_SEP, &fnres);
1582       append_uri_pathel (u_query, u_query + strlen (u_query), &fnres);
1583     }
1584
1585   /* Zero-terminate the file name. */
1586   append_char ('\0', &fnres);
1587
1588   fname = fnres.base;
1589
1590   /* Check the cases in which the unique extensions are not used:
1591      1) Clobbering is turned off (-nc).
1592      2) Retrieval with regetting.
1593      3) Timestamping is used.
1594      4) Hierarchy is built.
1595
1596      The exception is the case when file does exist and is a
1597      directory (see `mkalldirs' for explanation).  */
1598
1599   if ((opt.noclobber || opt.always_rest || opt.timestamping || opt.dirstruct)
1600       && !(file_exists_p (fname) && !file_non_directory_p (fname)))
1601     return fname;
1602
1603   unique = unique_name (fname, 1);
1604   if (unique != fname)
1605     xfree (fname);
1606   return unique;
1607 }
1608
1609 /* Return the length of URL's path.  Path is considered to be
1610    terminated by one of '?', ';', '#', or by the end of the
1611    string.  */
1612 static int
1613 path_length (const char *url)
1614 {
1615   const char *q = strpbrk_or_eos (url, "?;#");
1616   return q - url;
1617 }
1618
1619 /* Find the last occurrence of character C in the range [b, e), or
1620    NULL, if none are present.  This is equivalent to strrchr(b, c),
1621    except that it accepts an END argument instead of requiring the
1622    string to be zero-terminated.  Why is there no memrchr()?  */
1623 static const char *
1624 find_last_char (const char *b, const char *e, char c)
1625 {
1626   for (; e > b; e--)
1627     if (*e == c)
1628       return e;
1629   return NULL;
1630 }
1631 \f
1632 /* Resolve "." and ".." elements of PATH by destructively modifying
1633    PATH and return non-zero if PATH has been modified, zero otherwise.
1634
1635    The algorithm is in spirit similar to the one described in rfc1808,
1636    although implemented differently, in one pass.  To recap, path
1637    elements containing only "." are removed, and ".." is taken to mean
1638    "back up one element".  Single leading and trailing slashes are
1639    preserved.
1640
1641    This function does not handle URL escapes explicitly.  If you're
1642    passing paths from URLs, make sure to unquote "%2e" and "%2E" to
1643    ".", so that this function can find the dots.  (Wget's URL parser
1644    calls reencode_escapes, which see.)
1645
1646    For example, "a/b/c/./../d/.." will yield "a/b/".  More exhaustive
1647    test examples are provided below.  If you change anything in this
1648    function, run test_path_simplify to make sure you haven't broken a
1649    test case.  */
1650
1651 static int
1652 path_simplify (char *path)
1653 {
1654   char *h, *t, *end;
1655
1656   /* Preserve the leading '/'. */
1657   if (path[0] == '/')
1658     ++path;
1659
1660   h = path;                     /* hare */
1661   t = path;                     /* tortoise */
1662   end = path + strlen (path);
1663
1664   while (h < end)
1665     {
1666       /* Hare should be at the beginning of a path element. */
1667
1668       if (h[0] == '.' && (h[1] == '/' || h[1] == '\0'))
1669         {
1670           /* Ignore "./". */
1671           h += 2;
1672         }
1673       else if (h[0] == '.' && h[1] == '.' && (h[2] == '/' || h[2] == '\0'))
1674         {
1675           /* Handle "../" by retreating the tortoise by one path
1676              element -- but not past beggining of PATH.  */
1677           if (t > path)
1678             {
1679               /* Move backwards until T hits the beginning of the
1680                  previous path element or the beginning of path. */
1681               for (--t; t > path && t[-1] != '/'; t--)
1682                 ;
1683             }
1684           h += 3;
1685         }
1686       else if (*h == '/')
1687         {
1688           /* Ignore empty path elements.  Supporting them well is hard
1689              (where do you save "http://x.com///y.html"?), and they
1690              don't bring any practical gain.  Plus, they break our
1691              filesystem-influenced assumptions: allowing them would
1692              make "x/y//../z" simplify to "x/y/z", whereas most people
1693              would expect "x/z".  */
1694           ++h;
1695         }
1696       else
1697         {
1698           /* A regular path element.  If H hasn't advanced past T,
1699              simply skip to the next path element.  Otherwise, copy
1700              the path element until the next slash.  */
1701           if (t == h)
1702             {
1703               /* Skip the path element, including the slash.  */
1704               while (h < end && *h != '/')
1705                 t++, h++;
1706               if (h < end)
1707                 t++, h++;
1708             }
1709           else
1710             {
1711               /* Copy the path element, including the final slash.  */
1712               while (h < end && *h != '/')
1713                 *t++ = *h++;
1714               if (h < end)
1715                 *t++ = *h++;
1716             }
1717         }
1718     }
1719
1720   if (t != h)
1721     *t = '\0';
1722
1723   return t != h;
1724 }
1725 \f
1726 /* Merge BASE with LINK and return the resulting URI.
1727
1728    Either of the URIs may be absolute or relative, complete with the
1729    host name, or path only.  This tries to reasonably handle all
1730    foreseeable cases.  It only employs minimal URL parsing, without
1731    knowledge of the specifics of schemes.
1732
1733    Perhaps this function should call path_simplify so that the callers
1734    don't have to call url_parse unconditionally.  */
1735
1736 char *
1737 uri_merge (const char *base, const char *link)
1738 {
1739   int linklength;
1740   const char *end;
1741   char *merge;
1742
1743   if (url_has_scheme (link))
1744     return xstrdup (link);
1745
1746   /* We may not examine BASE past END. */
1747   end = base + path_length (base);
1748   linklength = strlen (link);
1749
1750   if (!*link)
1751     {
1752       /* Empty LINK points back to BASE, query string and all. */
1753       return xstrdup (base);
1754     }
1755   else if (*link == '?')
1756     {
1757       /* LINK points to the same location, but changes the query
1758          string.  Examples: */
1759       /* uri_merge("path",         "?new") -> "path?new"     */
1760       /* uri_merge("path?foo",     "?new") -> "path?new"     */
1761       /* uri_merge("path?foo#bar", "?new") -> "path?new"     */
1762       /* uri_merge("path#foo",     "?new") -> "path?new"     */
1763       int baselength = end - base;
1764       merge = xmalloc (baselength + linklength + 1);
1765       memcpy (merge, base, baselength);
1766       memcpy (merge + baselength, link, linklength);
1767       merge[baselength + linklength] = '\0';
1768     }
1769   else if (*link == '#')
1770     {
1771       /* uri_merge("path",         "#new") -> "path#new"     */
1772       /* uri_merge("path#foo",     "#new") -> "path#new"     */
1773       /* uri_merge("path?foo",     "#new") -> "path?foo#new" */
1774       /* uri_merge("path?foo#bar", "#new") -> "path?foo#new" */
1775       int baselength;
1776       const char *end1 = strchr (base, '#');
1777       if (!end1)
1778         end1 = base + strlen (base);
1779       baselength = end1 - base;
1780       merge = xmalloc (baselength + linklength + 1);
1781       memcpy (merge, base, baselength);
1782       memcpy (merge + baselength, link, linklength);
1783       merge[baselength + linklength] = '\0';
1784     }
1785   else if (*link == '/' && *(link + 1) == '/')
1786     {
1787       /* LINK begins with "//" and so is a net path: we need to
1788          replace everything after (and including) the double slash
1789          with LINK. */
1790
1791       /* uri_merge("foo", "//new/bar")            -> "//new/bar"      */
1792       /* uri_merge("//old/foo", "//new/bar")      -> "//new/bar"      */
1793       /* uri_merge("http://old/foo", "//new/bar") -> "http://new/bar" */
1794
1795       int span;
1796       const char *slash;
1797       const char *start_insert;
1798
1799       /* Look for first slash. */
1800       slash = memchr (base, '/', end - base);
1801       /* If found slash and it is a double slash, then replace
1802          from this point, else default to replacing from the
1803          beginning.  */
1804       if (slash && *(slash + 1) == '/')
1805         start_insert = slash;
1806       else
1807         start_insert = base;
1808
1809       span = start_insert - base;
1810       merge = (char *)xmalloc (span + linklength + 1);
1811       if (span)
1812         memcpy (merge, base, span);
1813       memcpy (merge + span, link, linklength);
1814       merge[span + linklength] = '\0';
1815     }
1816   else if (*link == '/')
1817     {
1818       /* LINK is an absolute path: we need to replace everything
1819          after (and including) the FIRST slash with LINK.
1820
1821          So, if BASE is "http://host/whatever/foo/bar", and LINK is
1822          "/qux/xyzzy", our result should be
1823          "http://host/qux/xyzzy".  */
1824       int span;
1825       const char *slash;
1826       const char *start_insert = NULL; /* for gcc to shut up. */
1827       const char *pos = base;
1828       int seen_slash_slash = 0;
1829       /* We're looking for the first slash, but want to ignore
1830          double slash. */
1831     again:
1832       slash = memchr (pos, '/', end - pos);
1833       if (slash && !seen_slash_slash)
1834         if (*(slash + 1) == '/')
1835           {
1836             pos = slash + 2;
1837             seen_slash_slash = 1;
1838             goto again;
1839           }
1840
1841       /* At this point, SLASH is the location of the first / after
1842          "//", or the first slash altogether.  START_INSERT is the
1843          pointer to the location where LINK will be inserted.  When
1844          examining the last two examples, keep in mind that LINK
1845          begins with '/'. */
1846
1847       if (!slash && !seen_slash_slash)
1848         /* example: "foo" */
1849         /*           ^    */
1850         start_insert = base;
1851       else if (!slash && seen_slash_slash)
1852         /* example: "http://foo" */
1853         /*                     ^ */
1854         start_insert = end;
1855       else if (slash && !seen_slash_slash)
1856         /* example: "foo/bar" */
1857         /*           ^        */
1858         start_insert = base;
1859       else if (slash && seen_slash_slash)
1860         /* example: "http://something/" */
1861         /*                           ^  */
1862         start_insert = slash;
1863
1864       span = start_insert - base;
1865       merge = (char *)xmalloc (span + linklength + 1);
1866       if (span)
1867         memcpy (merge, base, span);
1868       memcpy (merge + span, link, linklength);
1869       merge[span + linklength] = '\0';
1870     }
1871   else
1872     {
1873       /* LINK is a relative URL: we need to replace everything
1874          after last slash (possibly empty) with LINK.
1875
1876          So, if BASE is "whatever/foo/bar", and LINK is "qux/xyzzy",
1877          our result should be "whatever/foo/qux/xyzzy".  */
1878       int need_explicit_slash = 0;
1879       int span;
1880       const char *start_insert;
1881       const char *last_slash = find_last_char (base, end, '/');
1882       if (!last_slash)
1883         {
1884           /* No slash found at all.  Append LINK to what we have,
1885              but we'll need a slash as a separator.
1886
1887              Example: if base == "foo" and link == "qux/xyzzy", then
1888              we cannot just append link to base, because we'd get
1889              "fooqux/xyzzy", whereas what we want is
1890              "foo/qux/xyzzy".
1891
1892              To make sure the / gets inserted, we set
1893              need_explicit_slash to 1.  We also set start_insert
1894              to end + 1, so that the length calculations work out
1895              correctly for one more (slash) character.  Accessing
1896              that character is fine, since it will be the
1897              delimiter, '\0' or '?'.  */
1898           /* example: "foo?..." */
1899           /*               ^    ('?' gets changed to '/') */
1900           start_insert = end + 1;
1901           need_explicit_slash = 1;
1902         }
1903       else if (last_slash && last_slash >= base + 2
1904                && last_slash[-2] == ':' && last_slash[-1] == '/')
1905         {
1906           /* example: http://host"  */
1907           /*                      ^ */
1908           start_insert = end + 1;
1909           need_explicit_slash = 1;
1910         }
1911       else
1912         {
1913           /* example: "whatever/foo/bar" */
1914           /*                        ^    */
1915           start_insert = last_slash + 1;
1916         }
1917
1918       span = start_insert - base;
1919       merge = (char *)xmalloc (span + linklength + 1);
1920       if (span)
1921         memcpy (merge, base, span);
1922       if (need_explicit_slash)
1923         merge[span - 1] = '/';
1924       memcpy (merge + span, link, linklength);
1925       merge[span + linklength] = '\0';
1926     }
1927
1928   return merge;
1929 }
1930 \f
1931 #define APPEND(p, s) do {                       \
1932   int len = strlen (s);                         \
1933   memcpy (p, s, len);                           \
1934   p += len;                                     \
1935 } while (0)
1936
1937 /* Use this instead of password when the actual password is supposed
1938    to be hidden.  We intentionally use a generic string without giving
1939    away the number of characters in the password, like previous
1940    versions did.  */
1941 #define HIDDEN_PASSWORD "*password*"
1942
1943 /* Recreate the URL string from the data in URL.
1944
1945    If HIDE is non-zero (as it is when we're calling this on a URL we
1946    plan to print, but not when calling it to canonicalize a URL for
1947    use within the program), password will be hidden.  Unsafe
1948    characters in the URL will be quoted.  */
1949
1950 char *
1951 url_string (const struct url *url, int hide_password)
1952 {
1953   int size;
1954   char *result, *p;
1955   char *quoted_user = NULL, *quoted_passwd = NULL;
1956
1957   int scheme_port  = supported_schemes[url->scheme].default_port;
1958   char *scheme_str = supported_schemes[url->scheme].leading_string;
1959   int fplen = full_path_length (url);
1960
1961   int brackets_around_host = 0;
1962
1963   assert (scheme_str != NULL);
1964
1965   /* Make sure the user name and password are quoted. */
1966   if (url->user)
1967     {
1968       quoted_user = url_escape_allow_passthrough (url->user);
1969       if (url->passwd)
1970         {
1971           if (hide_password)
1972             quoted_passwd = HIDDEN_PASSWORD;
1973           else
1974             quoted_passwd = url_escape_allow_passthrough (url->passwd);
1975         }
1976     }
1977
1978   if (strchr (url->host, ':'))
1979     brackets_around_host = 1;
1980
1981   size = (strlen (scheme_str)
1982           + strlen (url->host)
1983           + (brackets_around_host ? 2 : 0)
1984           + fplen
1985           + 1);
1986   if (url->port != scheme_port)
1987     size += 1 + numdigit (url->port);
1988   if (quoted_user)
1989     {
1990       size += 1 + strlen (quoted_user);
1991       if (quoted_passwd)
1992         size += 1 + strlen (quoted_passwd);
1993     }
1994
1995   p = result = xmalloc (size);
1996
1997   APPEND (p, scheme_str);
1998   if (quoted_user)
1999     {
2000       APPEND (p, quoted_user);
2001       if (quoted_passwd)
2002         {
2003           *p++ = ':';
2004           APPEND (p, quoted_passwd);
2005         }
2006       *p++ = '@';
2007     }
2008
2009   if (brackets_around_host)
2010     *p++ = '[';
2011   APPEND (p, url->host);
2012   if (brackets_around_host)
2013     *p++ = ']';
2014   if (url->port != scheme_port)
2015     {
2016       *p++ = ':';
2017       p = number_to_string (p, url->port);
2018     }
2019
2020   full_path_write (url, p);
2021   p += fplen;
2022   *p++ = '\0';
2023
2024   assert (p - result == size);
2025
2026   if (quoted_user && quoted_user != url->user)
2027     xfree (quoted_user);
2028   if (quoted_passwd && !hide_password
2029       && quoted_passwd != url->passwd)
2030     xfree (quoted_passwd);
2031
2032   return result;
2033 }
2034 \f
2035 /* Return non-zero if scheme a is similar to scheme b.
2036  
2037    Schemes are similar if they are equal.  If SSL is supported, schemes
2038    are also similar if one is http (SCHEME_HTTP) and the other is https
2039    (SCHEME_HTTPS).  */
2040 int
2041 schemes_are_similar_p (enum url_scheme a, enum url_scheme b)
2042 {
2043   if (a == b)
2044     return 1;
2045 #ifdef HAVE_SSL
2046   if ((a == SCHEME_HTTP && b == SCHEME_HTTPS)
2047       || (a == SCHEME_HTTPS && b == SCHEME_HTTP))
2048     return 1;
2049 #endif
2050   return 0;
2051 }
2052 \f
2053 #if 0
2054 /* Debugging and testing support for path_simplify. */
2055
2056 /* Debug: run path_simplify on PATH and return the result in a new
2057    string.  Useful for calling from the debugger.  */
2058 static char *
2059 ps (char *path)
2060 {
2061   char *copy = xstrdup (path);
2062   path_simplify (copy);
2063   return copy;
2064 }
2065
2066 static void
2067 run_test (char *test, char *expected_result, int expected_change)
2068 {
2069   char *test_copy = xstrdup (test);
2070   int modified = path_simplify (test_copy);
2071
2072   if (0 != strcmp (test_copy, expected_result))
2073     {
2074       printf ("Failed path_simplify(\"%s\"): expected \"%s\", got \"%s\".\n",
2075               test, expected_result, test_copy);
2076     }
2077   if (modified != expected_change)
2078     {
2079       if (expected_change == 1)
2080         printf ("Expected no modification with path_simplify(\"%s\").\n",
2081                 test);
2082       else
2083         printf ("Expected modification with path_simplify(\"%s\").\n",
2084                 test);
2085     }
2086   xfree (test_copy);
2087 }
2088
2089 static void
2090 test_path_simplify (void)
2091 {
2092   static struct {
2093     char *test, *result;
2094     int should_modify;
2095   } tests[] = {
2096     { "",               "",             0 },
2097     { ".",              "",             1 },
2098     { "..",             "",             1 },
2099     { "foo",            "foo",          0 },
2100     { "foo/bar",        "foo/bar",      0 },
2101     { "foo///bar",      "foo/bar",      1 },
2102     { "foo/.",          "foo/",         1 },
2103     { "foo/./",         "foo/",         1 },
2104     { "foo./",          "foo./",        0 },
2105     { "foo/../bar",     "bar",          1 },
2106     { "foo/../bar/",    "bar/",         1 },
2107     { "foo/bar/..",     "foo/",         1 },
2108     { "foo/bar/../x",   "foo/x",        1 },
2109     { "foo/bar/../x/",  "foo/x/",       1 },
2110     { "foo/..",         "",             1 },
2111     { "foo/../..",      "",             1 },
2112     { "a/b/../../c",    "c",            1 },
2113     { "./a/../b",       "b",            1 }
2114   };
2115   int i;
2116
2117   for (i = 0; i < countof (tests); i++)
2118     {
2119       char *test = tests[i].test;
2120       char *expected_result = tests[i].result;
2121       int   expected_change = tests[i].should_modify;
2122       run_test (test, expected_result, expected_change);
2123     }
2124
2125   /* Now run all the tests with a leading slash before the test case,
2126      to prove that the slash is being preserved.  */
2127   for (i = 0; i < countof (tests); i++)
2128     {
2129       char *test, *expected_result;
2130       int expected_change = tests[i].should_modify;
2131
2132       test = xmalloc (1 + strlen (tests[i].test) + 1);
2133       sprintf (test, "/%s", tests[i].test);
2134
2135       expected_result = xmalloc (1 + strlen (tests[i].result) + 1);
2136       sprintf (expected_result, "/%s", tests[i].result);
2137
2138       run_test (test, expected_result, expected_change);
2139
2140       xfree (test);
2141       xfree (expected_result);
2142     }
2143 }
2144 #endif