Sync with NetBSD:
[dragonfly.git] / contrib / tnftp / fetch.c
1 /*      $NetBSD: fetch.c,v 1.180 2007/06/05 00:31:20 lukem Exp $        */
2
3 /*-
4  * Copyright (c) 1997-2007 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * This code is derived from software contributed to The NetBSD Foundation
8  * by Luke Mewburn.
9  *
10  * This code is derived from software contributed to The NetBSD Foundation
11  * by Scott Aaron Bamford.
12  *
13  * Redistribution and use in source and binary forms, with or without
14  * modification, are permitted provided that the following conditions
15  * are met:
16  * 1. Redistributions of source code must retain the above copyright
17  *    notice, this list of conditions and the following disclaimer.
18  * 2. Redistributions in binary form must reproduce the above copyright
19  *    notice, this list of conditions and the following disclaimer in the
20  *    documentation and/or other materials provided with the distribution.
21  * 3. All advertising materials mentioning features or use of this software
22  *    must display the following acknowledgement:
23  *      This product includes software developed by the NetBSD
24  *      Foundation, Inc. and its contributors.
25  * 4. Neither the name of The NetBSD Foundation nor the names of its
26  *    contributors may be used to endorse or promote products derived
27  *    from this software without specific prior written permission.
28  *
29  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
30  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
31  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
32  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
33  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
34  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
35  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
36  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
37  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
38  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
39  * POSSIBILITY OF SUCH DAMAGE.
40  */
41
42 #include <sys/cdefs.h>
43 #ifndef lint
44 __RCSID("$NetBSD: fetch.c,v 1.180 2007/06/05 00:31:20 lukem Exp $");
45 #endif /* not lint */
46
47 /*
48  * FTP User Program -- Command line file retrieval
49  */
50
51 #include <sys/types.h>
52 #include <sys/param.h>
53 #include <sys/socket.h>
54 #include <sys/stat.h>
55 #include <sys/time.h>
56
57 #include <netinet/in.h>
58
59 #include <arpa/ftp.h>
60 #include <arpa/inet.h>
61
62 #include <ctype.h>
63 #include <err.h>
64 #include <errno.h>
65 #include <netdb.h>
66 #include <fcntl.h>
67 #include <stdio.h>
68 #include <stdlib.h>
69 #include <string.h>
70 #include <unistd.h>
71 #include <time.h>
72
73 #include "ftp_var.h"
74 #include "version.h"
75
76 typedef enum {
77         UNKNOWN_URL_T=-1,
78         HTTP_URL_T,
79         FTP_URL_T,
80         FILE_URL_T,
81         CLASSIC_URL_T
82 } url_t;
83
84 void            aborthttp(int);
85 #ifndef NO_AUTH
86 static int      auth_url(const char *, char **, const char *, const char *);
87 static void     base64_encode(const unsigned char *, size_t, unsigned char *);
88 #endif
89 static int      go_fetch(const char *);
90 static int      fetch_ftp(const char *);
91 static int      fetch_url(const char *, const char *, char *, char *);
92 static const char *match_token(const char **, const char *);
93 static int      parse_url(const char *, const char *, url_t *, char **,
94                             char **, char **, char **, in_port_t *, char **);
95 static void     url_decode(char *);
96
97 static int      redirect_loop;
98
99
100 #define STRNEQUAL(a,b)  (strncasecmp((a), (b), sizeof((b))-1) == 0)
101 #define ISLWS(x)        ((x)=='\r' || (x)=='\n' || (x)==' ' || (x)=='\t')
102 #define SKIPLWS(x)      do { while (ISLWS((*x))) x++; } while (0)
103
104
105 #define ABOUT_URL       "about:"        /* propaganda */
106 #define FILE_URL        "file://"       /* file URL prefix */
107 #define FTP_URL         "ftp://"        /* ftp URL prefix */
108 #define HTTP_URL        "http://"       /* http URL prefix */
109
110
111 /*
112  * Determine if token is the next word in buf (case insensitive).
113  * If so, advance buf past the token and any trailing LWS, and
114  * return a pointer to the token (in buf).  Otherwise, return NULL.
115  * token may be preceded by LWS.
116  * token must be followed by LWS or NUL.  (I.e, don't partial match).
117  */
118 static const char *
119 match_token(const char **buf, const char *token)
120 {
121         const char      *p, *orig;
122         size_t          tlen;
123
124         tlen = strlen(token);
125         p = *buf;
126         SKIPLWS(p);
127         orig = p;
128         if (strncasecmp(p, token, tlen) != 0)
129                 return NULL;
130         p += tlen;
131         if (*p != '\0' && !ISLWS(*p))
132                 return NULL;
133         SKIPLWS(p);
134         orig = *buf;
135         *buf = p;
136         return orig;
137 }
138
139 #ifndef NO_AUTH
140 /*
141  * Generate authorization response based on given authentication challenge.
142  * Returns -1 if an error occurred, otherwise 0.
143  * Sets response to a malloc(3)ed string; caller should free.
144  */
145 static int
146 auth_url(const char *challenge, char **response, const char *guser,
147         const char *gpass)
148 {
149         const char      *cp, *scheme, *errormsg;
150         char            *ep, *clear, *realm;
151         char             user[BUFSIZ], *pass;
152         int              rval;
153         size_t           len, clen, rlen;
154
155         *response = NULL;
156         clear = realm = NULL;
157         rval = -1;
158         cp = challenge;
159         scheme = "Basic";       /* only support Basic authentication */
160
161         DPRINTF("auth_url: challenge `%s'\n", challenge);
162
163         if (! match_token(&cp, scheme)) {
164                 warnx("Unsupported authentication challenge `%s'",
165                     challenge);
166                 goto cleanup_auth_url;
167         }
168
169 #define REALM "realm=\""
170         if (STRNEQUAL(cp, REALM))
171                 cp += sizeof(REALM) - 1;
172         else {
173                 warnx("Unsupported authentication challenge `%s'",
174                     challenge);
175                 goto cleanup_auth_url;
176         }
177 /* XXX: need to improve quoted-string parsing to support \ quoting, etc. */
178         if ((ep = strchr(cp, '\"')) != NULL) {
179                 size_t len = ep - cp;
180
181                 realm = (char *)ftp_malloc(len + 1);
182                 (void)strlcpy(realm, cp, len + 1);
183         } else {
184                 warnx("Unsupported authentication challenge `%s'",
185                     challenge);
186                 goto cleanup_auth_url;
187         }
188
189         fprintf(ttyout, "Username for `%s': ", realm);
190         if (guser != NULL) {
191                 (void)strlcpy(user, guser, sizeof(user));
192                 fprintf(ttyout, "%s\n", user);
193         } else {
194                 (void)fflush(ttyout);
195                 if (getline(stdin, user, sizeof(user), &errormsg) < 0) {
196                         warnx("%s; can't authenticate", errormsg);
197                         goto cleanup_auth_url;
198                 }
199         }
200         if (gpass != NULL)
201                 pass = (char *)gpass;
202         else {
203                 pass = getpass("Password: ");
204                 if (pass == NULL) {
205                         warnx("Can't read password");
206                         goto cleanup_auth_url;
207                 }
208         }
209
210         clen = strlen(user) + strlen(pass) + 2; /* user + ":" + pass + "\0" */
211         clear = (char *)ftp_malloc(clen);
212         (void)strlcpy(clear, user, clen);
213         (void)strlcat(clear, ":", clen);
214         (void)strlcat(clear, pass, clen);
215         if (gpass == NULL)
216                 memset(pass, 0, strlen(pass));
217
218                                                 /* scheme + " " + enc + "\0" */
219         rlen = strlen(scheme) + 1 + (clen + 2) * 4 / 3 + 1;
220         *response = (char *)ftp_malloc(rlen);
221         (void)strlcpy(*response, scheme, rlen);
222         len = strlcat(*response, " ", rlen);
223                         /* use  `clen - 1'  to not encode the trailing NUL */
224         base64_encode((unsigned char *)clear, clen - 1,
225             (unsigned char *)*response + len);
226         memset(clear, 0, clen);
227         rval = 0;
228
229  cleanup_auth_url:
230         FREEPTR(clear);
231         FREEPTR(realm);
232         return (rval);
233 }
234
235 /*
236  * Encode len bytes starting at clear using base64 encoding into encoded,
237  * which should be at least ((len + 2) * 4 / 3 + 1) in size.
238  */
239 static void
240 base64_encode(const unsigned char *clear, size_t len, unsigned char *encoded)
241 {
242         static const unsigned char enc[] =
243             "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
244         unsigned char   *cp;
245         int      i;
246
247         cp = encoded;
248         for (i = 0; i < len; i += 3) {
249                 *(cp++) = enc[((clear[i + 0] >> 2))];
250                 *(cp++) = enc[((clear[i + 0] << 4) & 0x30)
251                             | ((clear[i + 1] >> 4) & 0x0f)];
252                 *(cp++) = enc[((clear[i + 1] << 2) & 0x3c)
253                             | ((clear[i + 2] >> 6) & 0x03)];
254                 *(cp++) = enc[((clear[i + 2]     ) & 0x3f)];
255         }
256         *cp = '\0';
257         while (i-- > len)
258                 *(--cp) = '=';
259 }
260 #endif
261
262 /*
263  * Decode %xx escapes in given string, `in-place'.
264  */
265 static void
266 url_decode(char *url)
267 {
268         unsigned char *p, *q;
269
270         if (EMPTYSTRING(url))
271                 return;
272         p = q = (unsigned char *)url;
273
274 #define HEXTOINT(x) (x - (isdigit(x) ? '0' : (islower(x) ? 'a' : 'A') - 10))
275         while (*p) {
276                 if (p[0] == '%'
277                     && p[1] && isxdigit((unsigned char)p[1])
278                     && p[2] && isxdigit((unsigned char)p[2])) {
279                         *q++ = HEXTOINT(p[1]) * 16 + HEXTOINT(p[2]);
280                         p+=3;
281                 } else
282                         *q++ = *p++;
283         }
284         *q = '\0';
285 }
286
287
288 /*
289  * Parse URL of form (per RFC3986):
290  *      <type>://[<user>[:<password>]@]<host>[:<port>][/<path>]
291  * Returns -1 if a parse error occurred, otherwise 0.
292  * It's the caller's responsibility to url_decode() the returned
293  * user, pass and path.
294  *
295  * Sets type to url_t, each of the given char ** pointers to a
296  * malloc(3)ed strings of the relevant section, and port to
297  * the number given, or ftpport if ftp://, or httpport if http://.
298  *
299  * XXX: this is not totally RFC3986 compliant; <path> will have the
300  * leading `/' unless it's an ftp:// URL, as this makes things easier
301  * for file:// and http:// URLs.  ftp:// URLs have the `/' between the
302  * host and the URL-path removed, but any additional leading slashes
303  * in the URL-path are retained (because they imply that we should
304  * later do "CWD" with a null argument).
305  *
306  * Examples:
307  *       input URL                       output path
308  *       ---------                       -----------
309  *      "http://host"                   "/"
310  *      "http://host/"                  "/"
311  *      "http://host/path"              "/path"
312  *      "file://host/dir/file"          "dir/file"
313  *      "ftp://host"                    ""
314  *      "ftp://host/"                   ""
315  *      "ftp://host//"                  "/"
316  *      "ftp://host/dir/file"           "dir/file"
317  *      "ftp://host//dir/file"          "/dir/file"
318  */
319 static int
320 parse_url(const char *url, const char *desc, url_t *type,
321                 char **user, char **pass, char **host, char **port,
322                 in_port_t *portnum, char **path)
323 {
324         const char      *origurl;
325         char            *cp, *ep, *thost, *tport;
326         size_t           len;
327
328         if (url == NULL || desc == NULL || type == NULL || user == NULL
329             || pass == NULL || host == NULL || port == NULL || portnum == NULL
330             || path == NULL)
331                 errx(1, "parse_url: invoked with NULL argument!");
332
333         origurl = url;
334         *type = UNKNOWN_URL_T;
335         *user = *pass = *host = *port = *path = NULL;
336         *portnum = 0;
337         tport = NULL;
338
339         if (STRNEQUAL(url, HTTP_URL)) {
340                 url += sizeof(HTTP_URL) - 1;
341                 *type = HTTP_URL_T;
342                 *portnum = HTTP_PORT;
343                 tport = httpport;
344         } else if (STRNEQUAL(url, FTP_URL)) {
345                 url += sizeof(FTP_URL) - 1;
346                 *type = FTP_URL_T;
347                 *portnum = FTP_PORT;
348                 tport = ftpport;
349         } else if (STRNEQUAL(url, FILE_URL)) {
350                 url += sizeof(FILE_URL) - 1;
351                 *type = FILE_URL_T;
352         } else {
353                 warnx("Invalid %s `%s'", desc, url);
354  cleanup_parse_url:
355                 FREEPTR(*user);
356                 if (*pass != NULL)
357                         memset(*pass, 0, strlen(*pass));
358                 FREEPTR(*pass);
359                 FREEPTR(*host);
360                 FREEPTR(*port);
361                 FREEPTR(*path);
362                 return (-1);
363         }
364
365         if (*url == '\0')
366                 return (0);
367
368                         /* find [user[:pass]@]host[:port] */
369         ep = strchr(url, '/');
370         if (ep == NULL)
371                 thost = ftp_strdup(url);
372         else {
373                 len = ep - url;
374                 thost = (char *)ftp_malloc(len + 1);
375                 (void)strlcpy(thost, url, len + 1);
376                 if (*type == FTP_URL_T) /* skip first / for ftp URLs */
377                         ep++;
378                 *path = ftp_strdup(ep);
379         }
380
381         cp = strchr(thost, '@');        /* look for user[:pass]@ in URLs */
382         if (cp != NULL) {
383                 if (*type == FTP_URL_T)
384                         anonftp = 0;    /* disable anonftp */
385                 *user = thost;
386                 *cp = '\0';
387                 thost = ftp_strdup(cp + 1);
388                 cp = strchr(*user, ':');
389                 if (cp != NULL) {
390                         *cp = '\0';
391                         *pass = ftp_strdup(cp + 1);
392                 }
393                 url_decode(*user);
394                 if (*pass)
395                         url_decode(*pass);
396         }
397
398 #ifdef INET6
399                         /*
400                          * Check if thost is an encoded IPv6 address, as per
401                          * RFC3986:
402                          *      `[' ipv6-address ']'
403                          */
404         if (*thost == '[') {
405                 cp = thost + 1;
406                 if ((ep = strchr(cp, ']')) == NULL ||
407                     (ep[1] != '\0' && ep[1] != ':')) {
408                         warnx("Invalid address `%s' in %s `%s'",
409                             thost, desc, origurl);
410                         goto cleanup_parse_url;
411                 }
412                 len = ep - cp;          /* change `[xyz]' -> `xyz' */
413                 memmove(thost, thost + 1, len);
414                 thost[len] = '\0';
415                 if (! isipv6addr(thost)) {
416                         warnx("Invalid IPv6 address `%s' in %s `%s'",
417                             thost, desc, origurl);
418                         goto cleanup_parse_url;
419                 }
420                 cp = ep + 1;
421                 if (*cp == ':')
422                         cp++;
423                 else
424                         cp = NULL;
425         } else
426 #endif /* INET6 */
427                 if ((cp = strchr(thost, ':')) != NULL)
428                         *cp++ =  '\0';
429         *host = thost;
430
431                         /* look for [:port] */
432         if (cp != NULL) {
433                 long    nport;
434
435                 nport = parseport(cp, -1);
436                 if (nport == -1) {
437                         warnx("Unknown port `%s' in %s `%s'",
438                             cp, desc, origurl);
439                         goto cleanup_parse_url;
440                 }
441                 *portnum = nport;
442                 tport = cp;
443         }
444
445         if (tport != NULL)
446                 *port = ftp_strdup(tport);
447         if (*path == NULL) {
448                 const char *emptypath = "/";
449                 if (*type == FTP_URL_T) /* skip first / for ftp URLs */
450                         emptypath++;
451                 *path = ftp_strdup(emptypath);
452         }
453
454         DPRINTF("parse_url: user `%s' pass `%s' host %s port %s(%d) "
455             "path `%s'\n",
456             *user ? *user : "<null>", *pass ? *pass : "<null>",
457             *host ? *host : "<null>", *port ? *port : "<null>",
458             *portnum ? *portnum : -1, *path ? *path : "<null>");
459
460         return (0);
461 }
462
463 sigjmp_buf      httpabort;
464
465 /*
466  * Retrieve URL, via a proxy if necessary, using HTTP.
467  * If proxyenv is set, use that for the proxy, otherwise try ftp_proxy or
468  * http_proxy as appropriate.
469  * Supports HTTP redirects.
470  * Returns 1 on failure, 0 on completed xfer, -1 if ftp connection
471  * is still open (e.g, ftp xfer with trailing /)
472  */
473 static int
474 fetch_url(const char *url, const char *proxyenv, char *proxyauth, char *wwwauth)
475 {
476         struct addrinfo         hints, *res, *res0 = NULL;
477         int                     error;
478         char                    hbuf[NI_MAXHOST];
479         sigfunc volatile        oldintr;
480         sigfunc volatile        oldintp;
481         int volatile            s;
482         struct stat             sb;
483         int volatile            ischunked;
484         int volatile            isproxy;
485         int volatile            rval;
486         int volatile            hcode;
487         size_t                  len;
488         static size_t           bufsize;
489         static char             *xferbuf;
490         const char              *cp, *token;
491         char                    *ep;
492         char                    buf[FTPBUFLEN];
493         const char              *errormsg;
494         char                    *volatile savefile;
495         char                    *volatile auth;
496         char                    *volatile location;
497         char                    *volatile message;
498         char                    *user, *pass, *host, *port, *path;
499         char                    *volatile decodedpath;
500         char                    *puser, *ppass, *useragent;
501         off_t                   hashbytes, rangestart, rangeend, entitylen;
502         int                     (*volatile closefunc)(FILE *);
503         FILE                    *volatile fin;
504         FILE                    *volatile fout;
505         time_t                  mtime;
506         url_t                   urltype;
507         in_port_t               portnum;
508
509         oldintr = oldintp = NULL;
510         closefunc = NULL;
511         fin = fout = NULL;
512         s = -1;
513         savefile = NULL;
514         auth = location = message = NULL;
515         ischunked = isproxy = hcode = 0;
516         rval = 1;
517         user = pass = host = path = decodedpath = puser = ppass = NULL;
518
519         if (parse_url(url, "URL", &urltype, &user, &pass, &host, &port,
520             &portnum, &path) == -1)
521                 goto cleanup_fetch_url;
522
523         if (urltype == FILE_URL_T && ! EMPTYSTRING(host)
524             && strcasecmp(host, "localhost") != 0) {
525                 warnx("No support for non local file URL `%s'", url);
526                 goto cleanup_fetch_url;
527         }
528
529         if (EMPTYSTRING(path)) {
530                 if (urltype == FTP_URL_T) {
531                         rval = fetch_ftp(url);
532                         goto cleanup_fetch_url;
533                 }
534                 if (urltype != HTTP_URL_T || outfile == NULL)  {
535                         warnx("Invalid URL (no file after host) `%s'", url);
536                         goto cleanup_fetch_url;
537                 }
538         }
539
540         decodedpath = ftp_strdup(path);
541         url_decode(decodedpath);
542
543         if (outfile)
544                 savefile = ftp_strdup(outfile);
545         else {
546                 cp = strrchr(decodedpath, '/');         /* find savefile */
547                 if (cp != NULL)
548                         savefile = ftp_strdup(cp + 1);
549                 else
550                         savefile = ftp_strdup(decodedpath);
551         }
552         if (EMPTYSTRING(savefile)) {
553                 if (urltype == FTP_URL_T) {
554                         rval = fetch_ftp(url);
555                         goto cleanup_fetch_url;
556                 }
557                 warnx("No file after directory (you must specify an "
558                     "output file) `%s'", url);
559                 goto cleanup_fetch_url;
560         } else {
561                 DPRINTF("savefile `%s'\n", savefile);
562         }
563
564         restart_point = 0;
565         filesize = -1;
566         rangestart = rangeend = entitylen = -1;
567         mtime = -1;
568         if (restartautofetch) {
569                 if (strcmp(savefile, "-") != 0 && *savefile != '|' &&
570                     stat(savefile, &sb) == 0)
571                         restart_point = sb.st_size;
572         }
573         if (urltype == FILE_URL_T) {            /* file:// URLs */
574                 direction = "copied";
575                 fin = fopen(decodedpath, "r");
576                 if (fin == NULL) {
577                         warn("Can't open `%s'", decodedpath);
578                         goto cleanup_fetch_url;
579                 }
580                 if (fstat(fileno(fin), &sb) == 0) {
581                         mtime = sb.st_mtime;
582                         filesize = sb.st_size;
583                 }
584                 if (restart_point) {
585                         if (lseek(fileno(fin), restart_point, SEEK_SET) < 0) {
586                                 warn("Can't seek to restart `%s'",
587                                     decodedpath);
588                                 goto cleanup_fetch_url;
589                         }
590                 }
591                 if (verbose) {
592                         fprintf(ttyout, "Copying %s", decodedpath);
593                         if (restart_point)
594                                 fprintf(ttyout, " (restarting at " LLF ")",
595                                     (LLT)restart_point);
596                         fputs("\n", ttyout);
597                 }
598         } else {                                /* ftp:// or http:// URLs */
599                 char *leading;
600                 int hasleading;
601
602                 if (proxyenv == NULL) {
603                         if (urltype == HTTP_URL_T)
604                                 proxyenv = getoptionvalue("http_proxy");
605                         else if (urltype == FTP_URL_T)
606                                 proxyenv = getoptionvalue("ftp_proxy");
607                 }
608                 direction = "retrieved";
609                 if (! EMPTYSTRING(proxyenv)) {                  /* use proxy */
610                         url_t purltype;
611                         char *phost, *ppath;
612                         char *pport, *no_proxy;
613
614                         isproxy = 1;
615
616                                 /* check URL against list of no_proxied sites */
617                         no_proxy = getoptionvalue("no_proxy");
618                         if (! EMPTYSTRING(no_proxy)) {
619                                 char *np, *np_copy, *np_iter;
620                                 long np_port;
621                                 size_t hlen, plen;
622
623                                 np_iter = np_copy = ftp_strdup(no_proxy);
624                                 hlen = strlen(host);
625                                 while ((cp = strsep(&np_iter, " ,")) != NULL) {
626                                         if (*cp == '\0')
627                                                 continue;
628                                         if ((np = strrchr(cp, ':')) != NULL) {
629                                                 *np = '\0';
630                                                 np_port =
631                                                     strtol(np + 1, &ep, 10);
632                                                 if (*ep != '\0')
633                                                         continue;
634                                                 if (np_port != portnum)
635                                                         continue;
636                                         }
637                                         plen = strlen(cp);
638                                         if (hlen < plen)
639                                                 continue;
640                                         if (strncasecmp(host + hlen - plen,
641                                             cp, plen) == 0) {
642                                                 isproxy = 0;
643                                                 break;
644                                         }
645                                 }
646                                 FREEPTR(np_copy);
647                                 if (isproxy == 0 && urltype == FTP_URL_T) {
648                                         rval = fetch_ftp(url);
649                                         goto cleanup_fetch_url;
650                                 }
651                         }
652
653                         if (isproxy) {
654                                 if (restart_point) {
655                                         warnx("Can't restart via proxy URL `%s'",
656                                             proxyenv);
657                                         goto cleanup_fetch_url;
658                                 }
659                                 if (parse_url(proxyenv, "proxy URL", &purltype,
660                                     &puser, &ppass, &phost, &pport, &portnum,
661                                     &ppath) == -1)
662                                         goto cleanup_fetch_url;
663
664                                 if ((purltype != HTTP_URL_T
665                                      && purltype != FTP_URL_T) ||
666                                     EMPTYSTRING(phost) ||
667                                     (! EMPTYSTRING(ppath)
668                                      && strcmp(ppath, "/") != 0)) {
669                                         warnx("Malformed proxy URL `%s'",
670                                             proxyenv);
671                                         FREEPTR(phost);
672                                         FREEPTR(pport);
673                                         FREEPTR(ppath);
674                                         goto cleanup_fetch_url;
675                                 }
676                                 if (isipv6addr(host) &&
677                                     strchr(host, '%') != NULL) {
678                                         warnx(
679 "Scoped address notation `%s' disallowed via web proxy",
680                                             host);
681                                         FREEPTR(phost);
682                                         FREEPTR(pport);
683                                         FREEPTR(ppath);
684                                         goto cleanup_fetch_url;
685                                 }
686
687                                 FREEPTR(host);
688                                 host = phost;
689                                 FREEPTR(port);
690                                 port = pport;
691                                 FREEPTR(path);
692                                 path = ftp_strdup(url);
693                                 FREEPTR(ppath);
694                         }
695                 } /* ! EMPTYSTRING(proxyenv) */
696
697                 memset(&hints, 0, sizeof(hints));
698                 hints.ai_flags = 0;
699                 hints.ai_family = family;
700                 hints.ai_socktype = SOCK_STREAM;
701                 hints.ai_protocol = 0;
702                 error = getaddrinfo(host, NULL, &hints, &res0);
703                 if (error) {
704                         warnx("Can't lookup `%s': %s", host,
705                             (error == EAI_SYSTEM) ? strerror(errno)
706                                                   : gai_strerror(error));
707                         goto cleanup_fetch_url;
708                 }
709                 if (res0->ai_canonname)
710                         host = res0->ai_canonname;
711
712                 s = -1;
713                 for (res = res0; res; res = res->ai_next) {
714                         ai_unmapped(res);
715                         if (getnameinfo(res->ai_addr, res->ai_addrlen,
716                             hbuf, sizeof(hbuf), NULL, 0, NI_NUMERICHOST) != 0)
717                                 strlcpy(hbuf, "?", sizeof(hbuf));
718
719                         if (verbose && res0->ai_next) {
720                                 fprintf(ttyout, "Trying %s...\n", hbuf);
721                         }
722
723                         ((struct sockaddr_in *)res->ai_addr)->sin_port =
724                             htons(portnum);
725                         s = socket(res->ai_family, SOCK_STREAM,
726                             res->ai_protocol);
727                         if (s < 0) {
728                                 warn(
729                                   "Can't create socket for connection to `%s'",
730                                     hbuf);
731                                 continue;
732                         }
733
734                         if (ftp_connect(s, res->ai_addr, res->ai_addrlen) < 0) {
735                                 close(s);
736                                 s = -1;
737                                 continue;
738                         }
739
740                         /* success */
741                         break;
742                 }
743
744                 if (s < 0) {
745                         warnx("Can't connect to `%s'", host);
746                         goto cleanup_fetch_url;
747                 }
748
749                 fin = fdopen(s, "r+");
750                 /*
751                  * Construct and send the request.
752                  */
753                 if (verbose)
754                         fprintf(ttyout, "Requesting %s\n", url);
755                 leading = "  (";
756                 hasleading = 0;
757                 if (isproxy) {
758                         if (verbose) {
759                                 fprintf(ttyout, "%svia %s:%s", leading,
760                                     host, port);
761                                 leading = ", ";
762                                 hasleading++;
763                         }
764                         fprintf(fin, "GET %s HTTP/1.0\r\n", path);
765                         if (flushcache)
766                                 fprintf(fin, "Pragma: no-cache\r\n");
767                 } else {
768                         fprintf(fin, "GET %s HTTP/1.1\r\n", path);
769                         if (strchr(host, ':')) {
770                                 char *h, *p;
771
772                                 /*
773                                  * strip off IPv6 scope identifier, since it is
774                                  * local to the node
775                                  */
776                                 h = ftp_strdup(host);
777                                 if (isipv6addr(h) &&
778                                     (p = strchr(h, '%')) != NULL) {
779                                         *p = '\0';
780                                 }
781                                 fprintf(fin, "Host: [%s]", h);
782                                 free(h);
783                         } else
784                                 fprintf(fin, "Host: %s", host);
785                         if (portnum != HTTP_PORT)
786                                 fprintf(fin, ":%u", portnum);
787                         fprintf(fin, "\r\n");
788                         fprintf(fin, "Accept: */*\r\n");
789                         fprintf(fin, "Connection: close\r\n");
790                         if (restart_point) {
791                                 fputs(leading, ttyout);
792                                 fprintf(fin, "Range: bytes=" LLF "-\r\n",
793                                     (LLT)restart_point);
794                                 fprintf(ttyout, "restarting at " LLF,
795                                     (LLT)restart_point);
796                                 leading = ", ";
797                                 hasleading++;
798                         }
799                         if (flushcache)
800                                 fprintf(fin, "Cache-Control: no-cache\r\n");
801                 }
802                 if ((useragent=getenv("FTPUSERAGENT")) != NULL) {
803                         fprintf(fin, "User-Agent: %s\r\n", useragent);
804                 } else {
805                         fprintf(fin, "User-Agent: %s/%s\r\n",
806                             FTP_PRODUCT, FTP_VERSION);
807                 }
808                 if (wwwauth) {
809                         if (verbose) {
810                                 fprintf(ttyout, "%swith authorization",
811                                     leading);
812                                 leading = ", ";
813                                 hasleading++;
814                         }
815                         fprintf(fin, "Authorization: %s\r\n", wwwauth);
816                 }
817                 if (proxyauth) {
818                         if (verbose) {
819                                 fprintf(ttyout,
820                                     "%swith proxy authorization", leading);
821                                 leading = ", ";
822                                 hasleading++;
823                         }
824                         fprintf(fin, "Proxy-Authorization: %s\r\n", proxyauth);
825                 }
826                 if (verbose && hasleading)
827                         fputs(")\n", ttyout);
828                 fprintf(fin, "\r\n");
829                 if (fflush(fin) == EOF) {
830                         warn("Writing HTTP request");
831                         goto cleanup_fetch_url;
832                 }
833
834                                 /* Read the response */
835                 len = getline(fin, buf, sizeof(buf), &errormsg);
836                 if (len < 0) {
837                         if (*errormsg == '\n')
838                                 errormsg++;
839                         warnx("Receiving HTTP reply: %s", errormsg);
840                         goto cleanup_fetch_url;
841                 }
842                 while (len > 0 && (ISLWS(buf[len-1])))
843                         buf[--len] = '\0';
844                 DPRINTF("received `%s'\n", buf);
845
846                                 /* Determine HTTP response code */
847                 cp = strchr(buf, ' ');
848                 if (cp == NULL)
849                         goto improper;
850                 else
851                         cp++;
852                 hcode = strtol(cp, &ep, 10);
853                 if (*ep != '\0' && !isspace((unsigned char)*ep))
854                         goto improper;
855                 message = ftp_strdup(cp);
856
857                                 /* Read the rest of the header. */
858                 while (1) {
859                         len = getline(fin, buf, sizeof(buf), &errormsg);
860                         if (len < 0) {
861                                 if (*errormsg == '\n')
862                                         errormsg++;
863                                 warnx("Receiving HTTP reply: %s", errormsg);
864                                 goto cleanup_fetch_url;
865                         }
866                         while (len > 0 && (ISLWS(buf[len-1])))
867                                 buf[--len] = '\0';
868                         if (len == 0)
869                                 break;
870                         DPRINTF("received `%s'\n", buf);
871
872                 /*
873                  * Look for some headers
874                  */
875
876                         cp = buf;
877
878                         if (match_token(&cp, "Content-Length:")) {
879                                 filesize = STRTOLL(cp, &ep, 10);
880                                 if (filesize < 0 || *ep != '\0')
881                                         goto improper;
882                                 DPRINTF("parsed len as: " LLF "\n",
883                                     (LLT)filesize);
884
885                         } else if (match_token(&cp, "Content-Range:")) {
886                                 if (! match_token(&cp, "bytes"))
887                                         goto improper;
888
889                                 if (*cp == '*')
890                                         cp++;
891                                 else {
892                                         rangestart = STRTOLL(cp, &ep, 10);
893                                         if (rangestart < 0 || *ep != '-')
894                                                 goto improper;
895                                         cp = ep + 1;
896                                         rangeend = STRTOLL(cp, &ep, 10);
897                                         if (rangeend < 0 || rangeend < rangestart)
898                                                 goto improper;
899                                         cp = ep;
900                                 }
901                                 if (*cp != '/')
902                                         goto improper;
903                                 cp++;
904                                 if (*cp == '*')
905                                         cp++;
906                                 else {
907                                         entitylen = STRTOLL(cp, &ep, 10);
908                                         if (entitylen < 0)
909                                                 goto improper;
910                                         cp = ep;
911                                 }
912                                 if (*cp != '\0')
913                                         goto improper;
914
915 #ifndef NO_DEBUG
916                                 if (ftp_debug) {
917                                         fprintf(ttyout, "parsed range as: ");
918                                         if (rangestart == -1)
919                                                 fprintf(ttyout, "*");
920                                         else
921                                                 fprintf(ttyout, LLF "-" LLF,
922                                                     (LLT)rangestart,
923                                                     (LLT)rangeend);
924                                         fprintf(ttyout, "/" LLF "\n", (LLT)entitylen);
925                                 }
926 #endif
927                                 if (! restart_point) {
928                                         warnx(
929                                     "Received unexpected Content-Range header");
930                                         goto cleanup_fetch_url;
931                                 }
932
933                         } else if (match_token(&cp, "Last-Modified:")) {
934                                 struct tm parsed;
935                                 char *t;
936
937                                 memset(&parsed, 0, sizeof(parsed));
938                                                         /* RFC1123 */
939                                 if ((t = strptime(cp,
940                                                 "%a, %d %b %Y %H:%M:%S GMT",
941                                                 &parsed))
942                                                         /* RFC0850 */
943                                     || (t = strptime(cp,
944                                                 "%a, %d-%b-%y %H:%M:%S GMT",
945                                                 &parsed))
946                                                         /* asctime */
947                                     || (t = strptime(cp,
948                                                 "%a, %b %d %H:%M:%S %Y",
949                                                 &parsed))) {
950                                         parsed.tm_isdst = -1;
951                                         if (*t == '\0')
952                                                 mtime = timegm(&parsed);
953 #ifndef NO_DEBUG
954                                         if (ftp_debug && mtime != -1) {
955                                                 fprintf(ttyout,
956                                                     "parsed date as: %s",
957                                                 rfc2822time(localtime(&mtime)));
958                                         }
959 #endif
960                                 }
961
962                         } else if (match_token(&cp, "Location:")) {
963                                 location = ftp_strdup(cp);
964                                 DPRINTF("parsed location as `%s'\n", cp);
965
966                         } else if (match_token(&cp, "Transfer-Encoding:")) {
967                                 if (match_token(&cp, "binary")) {
968                                         warnx(
969                         "Bogus transfer encoding `binary' (fetching anyway)");
970                                         continue;
971                                 }
972                                 if (! (token = match_token(&cp, "chunked"))) {
973                                         warnx(
974                                     "Unsupported transfer encoding `%s'",
975                                             token);
976                                         goto cleanup_fetch_url;
977                                 }
978                                 ischunked++;
979                                 DPRINTF("using chunked encoding\n");
980
981                         } else if (match_token(&cp, "Proxy-Authenticate:")
982                                 || match_token(&cp, "WWW-Authenticate:")) {
983                                 if (! (token = match_token(&cp, "Basic"))) {
984                                         DPRINTF(
985                                 "skipping unknown auth scheme `%s'\n",
986                                                     token);
987                                         continue;
988                                 }
989                                 FREEPTR(auth);
990                                 auth = ftp_strdup(token);
991                                 DPRINTF("parsed auth as `%s'\n", cp);
992                         }
993
994                 }
995                                 /* finished parsing header */
996
997                 switch (hcode) {
998                 case 200:
999                         break;
1000                 case 206:
1001                         if (! restart_point) {
1002                                 warnx("Not expecting partial content header");
1003                                 goto cleanup_fetch_url;
1004                         }
1005                         break;
1006                 case 300:
1007                 case 301:
1008                 case 302:
1009                 case 303:
1010                 case 305:
1011                 case 307:
1012                         if (EMPTYSTRING(location)) {
1013                                 warnx(
1014                                 "No redirection Location provided by server");
1015                                 goto cleanup_fetch_url;
1016                         }
1017                         if (redirect_loop++ > 5) {
1018                                 warnx("Too many redirections requested");
1019                                 goto cleanup_fetch_url;
1020                         }
1021                         if (hcode == 305) {
1022                                 if (verbose)
1023                                         fprintf(ttyout, "Redirected via %s\n",
1024                                             location);
1025                                 rval = fetch_url(url, location,
1026                                     proxyauth, wwwauth);
1027                         } else {
1028                                 if (verbose)
1029                                         fprintf(ttyout, "Redirected to %s\n",
1030                                             location);
1031                                 rval = go_fetch(location);
1032                         }
1033                         goto cleanup_fetch_url;
1034 #ifndef NO_AUTH
1035                 case 401:
1036                 case 407:
1037                     {
1038                         char **authp;
1039                         char *auser, *apass;
1040
1041                         if (hcode == 401) {
1042                                 authp = &wwwauth;
1043                                 auser = user;
1044                                 apass = pass;
1045                         } else {
1046                                 authp = &proxyauth;
1047                                 auser = puser;
1048                                 apass = ppass;
1049                         }
1050                         if (verbose || *authp == NULL ||
1051                             auser == NULL || apass == NULL)
1052                                 fprintf(ttyout, "%s\n", message);
1053                         if (EMPTYSTRING(auth)) {
1054                                 warnx(
1055                             "No authentication challenge provided by server");
1056                                 goto cleanup_fetch_url;
1057                         }
1058                         if (*authp != NULL) {
1059                                 char reply[10];
1060
1061                                 fprintf(ttyout,
1062                                     "Authorization failed. Retry (y/n)? ");
1063                                 if (getline(stdin, reply, sizeof(reply), NULL)
1064                                     < 0) {
1065                                         goto cleanup_fetch_url;
1066                                 }
1067                                 if (tolower((unsigned char)reply[0]) != 'y')
1068                                         goto cleanup_fetch_url;
1069                                 auser = NULL;
1070                                 apass = NULL;
1071                         }
1072                         if (auth_url(auth, authp, auser, apass) == 0) {
1073                                 rval = fetch_url(url, proxyenv,
1074                                     proxyauth, wwwauth);
1075                                 memset(*authp, 0, strlen(*authp));
1076                                 FREEPTR(*authp);
1077                         }
1078                         goto cleanup_fetch_url;
1079                     }
1080 #endif
1081                 default:
1082                         if (message)
1083                                 warnx("Error retrieving file `%s'", message);
1084                         else
1085                                 warnx("Unknown error retrieving file");
1086                         goto cleanup_fetch_url;
1087                 }
1088         }               /* end of ftp:// or http:// specific setup */
1089
1090                         /* Open the output file. */
1091         if (strcmp(savefile, "-") == 0) {
1092                 fout = stdout;
1093         } else if (*savefile == '|') {
1094                 oldintp = xsignal(SIGPIPE, SIG_IGN);
1095                 fout = popen(savefile + 1, "w");
1096                 if (fout == NULL) {
1097                         warn("Can't execute `%s'", savefile + 1);
1098                         goto cleanup_fetch_url;
1099                 }
1100                 closefunc = pclose;
1101         } else {
1102                 if ((rangeend != -1 && rangeend <= restart_point) ||
1103                     (rangestart == -1 && filesize != -1 && filesize <= restart_point)) {
1104                         /* already done */
1105                         if (verbose)
1106                                 fprintf(ttyout, "already done\n");
1107                         rval = 0;
1108                         goto cleanup_fetch_url;
1109                 }
1110                 if (restart_point && rangestart != -1) {
1111                         if (entitylen != -1)
1112                                 filesize = entitylen;
1113                         if (rangestart != restart_point) {
1114                                 warnx(
1115                                     "Size of `%s' differs from save file `%s'",
1116                                     url, savefile);
1117                                 goto cleanup_fetch_url;
1118                         }
1119                         fout = fopen(savefile, "a");
1120                 } else
1121                         fout = fopen(savefile, "w");
1122                 if (fout == NULL) {
1123                         warn("Can't open `%s'", savefile);
1124                         goto cleanup_fetch_url;
1125                 }
1126                 closefunc = fclose;
1127         }
1128
1129                         /* Trap signals */
1130         if (sigsetjmp(httpabort, 1))
1131                 goto cleanup_fetch_url;
1132         (void)xsignal(SIGQUIT, psummary);
1133         oldintr = xsignal(SIGINT, aborthttp);
1134
1135         if (rcvbuf_size > bufsize) {
1136                 if (xferbuf)
1137                         (void)free(xferbuf);
1138                 bufsize = rcvbuf_size;
1139                 xferbuf = ftp_malloc(bufsize);
1140         }
1141
1142         bytes = 0;
1143         hashbytes = mark;
1144         progressmeter(-1);
1145
1146                         /* Finally, suck down the file. */
1147         do {
1148                 long chunksize;
1149
1150                 chunksize = 0;
1151                                         /* read chunksize */
1152                 if (ischunked) {
1153                         if (fgets(xferbuf, bufsize, fin) == NULL) {
1154                                 warnx("Unexpected EOF reading chunksize");
1155                                 goto cleanup_fetch_url;
1156                         }
1157                         chunksize = strtol(xferbuf, &ep, 16);
1158
1159                                 /*
1160                                  * XXX: Work around bug in Apache 1.3.9 and
1161                                  *      1.3.11, which incorrectly put trailing
1162                                  *      space after the chunksize.
1163                                  */
1164                         while (*ep == ' ')
1165                                 ep++;
1166
1167                         if (strcmp(ep, "\r\n") != 0) {
1168                                 warnx("Unexpected data following chunksize");
1169                                 goto cleanup_fetch_url;
1170                         }
1171                         DPRINTF("got chunksize of " LLF "\n", (LLT)chunksize);
1172                         if (chunksize == 0)
1173                                 break;
1174                 }
1175                                         /* transfer file or chunk */
1176                 while (1) {
1177                         struct timeval then, now, td;
1178                         off_t bufrem;
1179
1180                         if (rate_get)
1181                                 (void)gettimeofday(&then, NULL);
1182                         bufrem = rate_get ? rate_get : bufsize;
1183                         if (ischunked)
1184                                 bufrem = MIN(chunksize, bufrem);
1185                         while (bufrem > 0) {
1186                                 len = fread(xferbuf, sizeof(char),
1187                                     MIN(bufsize, bufrem), fin);
1188                                 if (len <= 0)
1189                                         goto chunkdone;
1190                                 bytes += len;
1191                                 bufrem -= len;
1192                                 if (fwrite(xferbuf, sizeof(char), len, fout)
1193                                     != len) {
1194                                         warn("Writing `%s'", savefile);
1195                                         goto cleanup_fetch_url;
1196                                 }
1197                                 if (hash && !progress) {
1198                                         while (bytes >= hashbytes) {
1199                                                 (void)putc('#', ttyout);
1200                                                 hashbytes += mark;
1201                                         }
1202                                         (void)fflush(ttyout);
1203                                 }
1204                                 if (ischunked) {
1205                                         chunksize -= len;
1206                                         if (chunksize <= 0)
1207                                                 break;
1208                                 }
1209                         }
1210                         if (rate_get) {
1211                                 while (1) {
1212                                         (void)gettimeofday(&now, NULL);
1213                                         timersub(&now, &then, &td);
1214                                         if (td.tv_sec > 0)
1215                                                 break;
1216                                         usleep(1000000 - td.tv_usec);
1217                                 }
1218                         }
1219                         if (ischunked && chunksize <= 0)
1220                                 break;
1221                 }
1222                                         /* read CRLF after chunk*/
1223  chunkdone:
1224                 if (ischunked) {
1225                         if (fgets(xferbuf, bufsize, fin) == NULL)
1226                                 break;
1227                         if (strcmp(xferbuf, "\r\n") != 0) {
1228                                 warnx("Unexpected data following chunk");
1229                                 goto cleanup_fetch_url;
1230                         }
1231                 }
1232         } while (ischunked);
1233         if (hash && !progress && bytes > 0) {
1234                 if (bytes < mark)
1235                         (void)putc('#', ttyout);
1236                 (void)putc('\n', ttyout);
1237         }
1238         if (ferror(fin)) {
1239                 warn("Reading file");
1240                 goto cleanup_fetch_url;
1241         }
1242         progressmeter(1);
1243         (void)fflush(fout);
1244         if (closefunc == fclose && mtime != -1) {
1245                 struct timeval tval[2];
1246
1247                 (void)gettimeofday(&tval[0], NULL);
1248                 tval[1].tv_sec = mtime;
1249                 tval[1].tv_usec = 0;
1250                 (*closefunc)(fout);
1251                 fout = NULL;
1252
1253                 if (utimes(savefile, tval) == -1) {
1254                         fprintf(ttyout,
1255                             "Can't change modification time to %s",
1256                             rfc2822time(localtime(&mtime)));
1257                 }
1258         }
1259         if (bytes > 0)
1260                 ptransfer(0);
1261         bytes = 0;
1262
1263         rval = 0;
1264         goto cleanup_fetch_url;
1265
1266  improper:
1267         warnx("Improper response from `%s'", host);
1268
1269  cleanup_fetch_url:
1270         if (oldintr)
1271                 (void)xsignal(SIGINT, oldintr);
1272         if (oldintp)
1273                 (void)xsignal(SIGPIPE, oldintp);
1274         if (fin != NULL)
1275                 fclose(fin);
1276         else if (s != -1)
1277                 close(s);
1278         if (closefunc != NULL && fout != NULL)
1279                 (*closefunc)(fout);
1280         if (res0)
1281                 freeaddrinfo(res0);
1282         FREEPTR(savefile);
1283         FREEPTR(user);
1284         if (pass != NULL)
1285                 memset(pass, 0, strlen(pass));
1286         FREEPTR(pass);
1287         FREEPTR(host);
1288         FREEPTR(port);
1289         FREEPTR(path);
1290         FREEPTR(decodedpath);
1291         FREEPTR(puser);
1292         if (ppass != NULL)
1293                 memset(ppass, 0, strlen(ppass));
1294         FREEPTR(ppass);
1295         FREEPTR(auth);
1296         FREEPTR(location);
1297         FREEPTR(message);
1298         return (rval);
1299 }
1300
1301 /*
1302  * Abort a HTTP retrieval
1303  */
1304 void
1305 aborthttp(int notused)
1306 {
1307         char msgbuf[100];
1308         size_t len;
1309
1310         sigint_raised = 1;
1311         alarmtimer(0);
1312         len = strlcpy(msgbuf, "\nHTTP fetch aborted.\n", sizeof(msgbuf));
1313         write(fileno(ttyout), msgbuf, len);
1314         siglongjmp(httpabort, 1);
1315 }
1316
1317 /*
1318  * Retrieve ftp URL or classic ftp argument using FTP.
1319  * Returns 1 on failure, 0 on completed xfer, -1 if ftp connection
1320  * is still open (e.g, ftp xfer with trailing /)
1321  */
1322 static int
1323 fetch_ftp(const char *url)
1324 {
1325         char            *cp, *xargv[5], rempath[MAXPATHLEN];
1326         char            *host, *path, *dir, *file, *user, *pass;
1327         char            *port;
1328         int              dirhasglob, filehasglob, rval, type, xargc;
1329         int              oanonftp, oautologin;
1330         in_port_t        portnum;
1331         url_t            urltype;
1332
1333         host = path = dir = file = user = pass = NULL;
1334         port = NULL;
1335         rval = 1;
1336         type = TYPE_I;
1337
1338         if (STRNEQUAL(url, FTP_URL)) {
1339                 if ((parse_url(url, "URL", &urltype, &user, &pass,
1340                     &host, &port, &portnum, &path) == -1) ||
1341                     (user != NULL && *user == '\0') ||
1342                     EMPTYSTRING(host)) {
1343                         warnx("Invalid URL `%s'", url);
1344                         goto cleanup_fetch_ftp;
1345                 }
1346                 /*
1347                  * Note: Don't url_decode(path) here.  We need to keep the
1348                  * distinction between "/" and "%2F" until later.
1349                  */
1350
1351                                         /* check for trailing ';type=[aid]' */
1352                 if (! EMPTYSTRING(path) && (cp = strrchr(path, ';')) != NULL) {
1353                         if (strcasecmp(cp, ";type=a") == 0)
1354                                 type = TYPE_A;
1355                         else if (strcasecmp(cp, ";type=i") == 0)
1356                                 type = TYPE_I;
1357                         else if (strcasecmp(cp, ";type=d") == 0) {
1358                                 warnx(
1359                             "Directory listing via a URL is not supported");
1360                                 goto cleanup_fetch_ftp;
1361                         } else {
1362                                 warnx("Invalid suffix `%s' in URL `%s'", cp,
1363                                     url);
1364                                 goto cleanup_fetch_ftp;
1365                         }
1366                         *cp = 0;
1367                 }
1368         } else {                        /* classic style `[user@]host:[file]' */
1369                 urltype = CLASSIC_URL_T;
1370                 host = ftp_strdup(url);
1371                 cp = strchr(host, '@');
1372                 if (cp != NULL) {
1373                         *cp = '\0';
1374                         user = host;
1375                         anonftp = 0;    /* disable anonftp */
1376                         host = ftp_strdup(cp + 1);
1377                 }
1378                 cp = strchr(host, ':');
1379                 if (cp != NULL) {
1380                         *cp = '\0';
1381                         path = ftp_strdup(cp + 1);
1382                 }
1383         }
1384         if (EMPTYSTRING(host))
1385                 goto cleanup_fetch_ftp;
1386
1387                         /* Extract the file and (if present) directory name. */
1388         dir = path;
1389         if (! EMPTYSTRING(dir)) {
1390                 /*
1391                  * If we are dealing with classic `[user@]host:[path]' syntax,
1392                  * then a path of the form `/file' (resulting from input of the
1393                  * form `host:/file') means that we should do "CWD /" before
1394                  * retrieving the file.  So we set dir="/" and file="file".
1395                  *
1396                  * But if we are dealing with URLs like `ftp://host/path' then
1397                  * a path of the form `/file' (resulting from a URL of the form
1398                  * `ftp://host//file') means that we should do `CWD ' (with an
1399                  * empty argument) before retrieving the file.  So we set
1400                  * dir="" and file="file".
1401                  *
1402                  * If the path does not contain / at all, we set dir=NULL.
1403                  * (We get a path without any slashes if we are dealing with
1404                  * classic `[user@]host:[file]' or URL `ftp://host/file'.)
1405                  *
1406                  * In all other cases, we set dir to a string that does not
1407                  * include the final '/' that separates the dir part from the
1408                  * file part of the path.  (This will be the empty string if
1409                  * and only if we are dealing with a path of the form `/file'
1410                  * resulting from an URL of the form `ftp://host//file'.)
1411                  */
1412                 cp = strrchr(dir, '/');
1413                 if (cp == dir && urltype == CLASSIC_URL_T) {
1414                         file = cp + 1;
1415                         dir = "/";
1416                 } else if (cp != NULL) {
1417                         *cp++ = '\0';
1418                         file = cp;
1419                 } else {
1420                         file = dir;
1421                         dir = NULL;
1422                 }
1423         } else
1424                 dir = NULL;
1425         if (urltype == FTP_URL_T && file != NULL) {
1426                 url_decode(file);
1427                 /* but still don't url_decode(dir) */
1428         }
1429         DPRINTF("fetch_ftp: user `%s' pass `%s' host %s port %s "
1430             "path `%s' dir `%s' file `%s'\n",
1431             user ? user : "<null>", pass ? pass : "<null>",
1432             host ? host : "<null>", port ? port : "<null>",
1433             path ? path : "<null>",
1434             dir ? dir : "<null>", file ? file : "<null>");
1435
1436         dirhasglob = filehasglob = 0;
1437         if (doglob && urltype == CLASSIC_URL_T) {
1438                 if (! EMPTYSTRING(dir) && strpbrk(dir, "*?[]{}") != NULL)
1439                         dirhasglob = 1;
1440                 if (! EMPTYSTRING(file) && strpbrk(file, "*?[]{}") != NULL)
1441                         filehasglob = 1;
1442         }
1443
1444                         /* Set up the connection */
1445         oanonftp = anonftp;
1446         if (connected)
1447                 disconnect(0, NULL);
1448         anonftp = oanonftp;
1449         xargv[0] = (char *)getprogname();       /* XXX discards const */
1450         xargv[1] = host;
1451         xargv[2] = NULL;
1452         xargc = 2;
1453         if (port) {
1454                 xargv[2] = port;
1455                 xargv[3] = NULL;
1456                 xargc = 3;
1457         }
1458         oautologin = autologin;
1459                 /* don't autologin in setpeer(), use ftp_login() below */
1460         autologin = 0;
1461         setpeer(xargc, xargv);
1462         autologin = oautologin;
1463         if ((connected == 0) ||
1464             (connected == 1 && !ftp_login(host, user, pass))) {
1465                 warnx("Can't connect or login to host `%s'", host);
1466                 goto cleanup_fetch_ftp;
1467         }
1468
1469         switch (type) {
1470         case TYPE_A:
1471                 setascii(1, xargv);
1472                 break;
1473         case TYPE_I:
1474                 setbinary(1, xargv);
1475                 break;
1476         default:
1477                 errx(1, "fetch_ftp: unknown transfer type %d", type);
1478         }
1479
1480                 /*
1481                  * Change directories, if necessary.
1482                  *
1483                  * Note: don't use EMPTYSTRING(dir) below, because
1484                  * dir=="" means something different from dir==NULL.
1485                  */
1486         if (dir != NULL && !dirhasglob) {
1487                 char *nextpart;
1488
1489                 /*
1490                  * If we are dealing with a classic `[user@]host:[path]'
1491                  * (urltype is CLASSIC_URL_T) then we have a raw directory
1492                  * name (not encoded in any way) and we can change
1493                  * directories in one step.
1494                  *
1495                  * If we are dealing with an `ftp://host/path' URL
1496                  * (urltype is FTP_URL_T), then RFC3986 says we need to
1497                  * send a separate CWD command for each unescaped "/"
1498                  * in the path, and we have to interpret %hex escaping
1499                  * *after* we find the slashes.  It's possible to get
1500                  * empty components here, (from multiple adjacent
1501                  * slashes in the path) and RFC3986 says that we should
1502                  * still do `CWD ' (with a null argument) in such cases.
1503                  *
1504                  * Many ftp servers don't support `CWD ', so if there's an
1505                  * error performing that command, bail out with a descriptive
1506                  * message.
1507                  *
1508                  * Examples:
1509                  *
1510                  * host:                        dir="", urltype=CLASSIC_URL_T
1511                  *              logged in (to default directory)
1512                  * host:file                    dir=NULL, urltype=CLASSIC_URL_T
1513                  *              "RETR file"
1514                  * host:dir/                    dir="dir", urltype=CLASSIC_URL_T
1515                  *              "CWD dir", logged in
1516                  * ftp://host/                  dir="", urltype=FTP_URL_T
1517                  *              logged in (to default directory)
1518                  * ftp://host/dir/              dir="dir", urltype=FTP_URL_T
1519                  *              "CWD dir", logged in
1520                  * ftp://host/file              dir=NULL, urltype=FTP_URL_T
1521                  *              "RETR file"
1522                  * ftp://host//file             dir="", urltype=FTP_URL_T
1523                  *              "CWD ", "RETR file"
1524                  * host:/file                   dir="/", urltype=CLASSIC_URL_T
1525                  *              "CWD /", "RETR file"
1526                  * ftp://host///file            dir="/", urltype=FTP_URL_T
1527                  *              "CWD ", "CWD ", "RETR file"
1528                  * ftp://host/%2F/file          dir="%2F", urltype=FTP_URL_T
1529                  *              "CWD /", "RETR file"
1530                  * ftp://host/foo/file          dir="foo", urltype=FTP_URL_T
1531                  *              "CWD foo", "RETR file"
1532                  * ftp://host/foo/bar/file      dir="foo/bar"
1533                  *              "CWD foo", "CWD bar", "RETR file"
1534                  * ftp://host//foo/bar/file     dir="/foo/bar"
1535                  *              "CWD ", "CWD foo", "CWD bar", "RETR file"
1536                  * ftp://host/foo//bar/file     dir="foo//bar"
1537                  *              "CWD foo", "CWD ", "CWD bar", "RETR file"
1538                  * ftp://host/%2F/foo/bar/file  dir="%2F/foo/bar"
1539                  *              "CWD /", "CWD foo", "CWD bar", "RETR file"
1540                  * ftp://host/%2Ffoo/bar/file   dir="%2Ffoo/bar"
1541                  *              "CWD /foo", "CWD bar", "RETR file"
1542                  * ftp://host/%2Ffoo%2Fbar/file dir="%2Ffoo%2Fbar"
1543                  *              "CWD /foo/bar", "RETR file"
1544                  * ftp://host/%2Ffoo%2Fbar%2Ffile       dir=NULL
1545                  *              "RETR /foo/bar/file"
1546                  *
1547                  * Note that we don't need `dir' after this point.
1548                  */
1549                 do {
1550                         if (urltype == FTP_URL_T) {
1551                                 nextpart = strchr(dir, '/');
1552                                 if (nextpart) {
1553                                         *nextpart = '\0';
1554                                         nextpart++;
1555                                 }
1556                                 url_decode(dir);
1557                         } else
1558                                 nextpart = NULL;
1559                         DPRINTF("dir `%s', nextpart `%s'\n",
1560                             dir ? dir : "<null>",
1561                             nextpart ? nextpart : "<null>");
1562                         if (urltype == FTP_URL_T || *dir != '\0') {
1563                                 xargv[0] = "cd";
1564                                 xargv[1] = dir;
1565                                 xargv[2] = NULL;
1566                                 dirchange = 0;
1567                                 cd(2, xargv);
1568                                 if (! dirchange) {
1569                                         if (*dir == '\0' && code == 500)
1570                                                 fprintf(stderr,
1571 "\n"
1572 "ftp: The `CWD ' command (without a directory), which is required by\n"
1573 "     RFC3986 to support the empty directory in the URL pathname (`//'),\n"
1574 "     conflicts with the server's conformance to RFC0959.\n"
1575 "     Try the same URL without the `//' in the URL pathname.\n"
1576 "\n");
1577                                         goto cleanup_fetch_ftp;
1578                                 }
1579                         }
1580                         dir = nextpart;
1581                 } while (dir != NULL);
1582         }
1583
1584         if (EMPTYSTRING(file)) {
1585                 rval = -1;
1586                 goto cleanup_fetch_ftp;
1587         }
1588
1589         if (dirhasglob) {
1590                 (void)strlcpy(rempath, dir,     sizeof(rempath));
1591                 (void)strlcat(rempath, "/",     sizeof(rempath));
1592                 (void)strlcat(rempath, file,    sizeof(rempath));
1593                 file = rempath;
1594         }
1595
1596                         /* Fetch the file(s). */
1597         xargc = 2;
1598         xargv[0] = "get";
1599         xargv[1] = file;
1600         xargv[2] = NULL;
1601         if (dirhasglob || filehasglob) {
1602                 int ointeractive;
1603
1604                 ointeractive = interactive;
1605                 interactive = 0;
1606                 if (restartautofetch)
1607                         xargv[0] = "mreget";
1608                 else
1609                         xargv[0] = "mget";
1610                 mget(xargc, xargv);
1611                 interactive = ointeractive;
1612         } else {
1613                 if (outfile == NULL) {
1614                         cp = strrchr(file, '/');        /* find savefile */
1615                         if (cp != NULL)
1616                                 outfile = cp + 1;
1617                         else
1618                                 outfile = file;
1619                 }
1620                 xargv[2] = (char *)outfile;
1621                 xargv[3] = NULL;
1622                 xargc++;
1623                 if (restartautofetch)
1624                         reget(xargc, xargv);
1625                 else
1626                         get(xargc, xargv);
1627         }
1628
1629         if ((code / 100) == COMPLETE)
1630                 rval = 0;
1631
1632  cleanup_fetch_ftp:
1633         FREEPTR(port);
1634         FREEPTR(host);
1635         FREEPTR(path);
1636         FREEPTR(user);
1637         if (pass)
1638                 memset(pass, 0, strlen(pass));
1639         FREEPTR(pass);
1640         return (rval);
1641 }
1642
1643 /*
1644  * Retrieve the given file to outfile.
1645  * Supports arguments of the form:
1646  *      "host:path", "ftp://host/path"  if $ftpproxy, call fetch_url() else
1647  *                                      call fetch_ftp()
1648  *      "http://host/path"              call fetch_url() to use HTTP
1649  *      "file:///path"                  call fetch_url() to copy
1650  *      "about:..."                     print a message
1651  *
1652  * Returns 1 on failure, 0 on completed xfer, -1 if ftp connection
1653  * is still open (e.g, ftp xfer with trailing /)
1654  */
1655 static int
1656 go_fetch(const char *url)
1657 {
1658         char *proxy;
1659
1660 #ifndef NO_ABOUT
1661         /*
1662          * Check for about:*
1663          */
1664         if (STRNEQUAL(url, ABOUT_URL)) {
1665                 url += sizeof(ABOUT_URL) -1;
1666                 if (strcasecmp(url, "ftp") == 0 ||
1667                     strcasecmp(url, "tnftp") == 0) {
1668                         fputs(
1669 "This version of ftp has been enhanced by Luke Mewburn <lukem@NetBSD.org>\n"
1670 "for the NetBSD project.  Execute `man ftp' for more details.\n", ttyout);
1671                 } else if (strcasecmp(url, "lukem") == 0) {
1672                         fputs(
1673 "Luke Mewburn is the author of most of the enhancements in this ftp client.\n"
1674 "Please email feedback to <lukem@NetBSD.org>.\n", ttyout);
1675                 } else if (strcasecmp(url, "netbsd") == 0) {
1676                         fputs(
1677 "NetBSD is a freely available and redistributable UNIX-like operating system.\n"
1678 "For more information, see http://www.NetBSD.org/\n", ttyout);
1679                 } else if (strcasecmp(url, "version") == 0) {
1680                         fprintf(ttyout, "Version: %s %s%s\n",
1681                             FTP_PRODUCT, FTP_VERSION,
1682 #ifdef INET6
1683                             ""
1684 #else
1685                             " (-IPv6)"
1686 #endif
1687                         );
1688                 } else {
1689                         fprintf(ttyout, "`%s' is an interesting topic.\n", url);
1690                 }
1691                 fputs("\n", ttyout);
1692                 return (0);
1693         }
1694 #endif
1695
1696         /*
1697          * Check for file:// and http:// URLs.
1698          */
1699         if (STRNEQUAL(url, HTTP_URL) || STRNEQUAL(url, FILE_URL))
1700                 return (fetch_url(url, NULL, NULL, NULL));
1701
1702         /*
1703          * Try FTP URL-style and host:file arguments next.
1704          * If ftpproxy is set with an FTP URL, use fetch_url()
1705          * Othewise, use fetch_ftp().
1706          */
1707         proxy = getoptionvalue("ftp_proxy");
1708         if (!EMPTYSTRING(proxy) && STRNEQUAL(url, FTP_URL))
1709                 return (fetch_url(url, NULL, NULL, NULL));
1710
1711         return (fetch_ftp(url));
1712 }
1713
1714 /*
1715  * Retrieve multiple files from the command line,
1716  * calling go_fetch() for each file.
1717  *
1718  * If an ftp path has a trailing "/", the path will be cd-ed into and
1719  * the connection remains open, and the function will return -1
1720  * (to indicate the connection is alive).
1721  * If an error occurs the return value will be the offset+1 in
1722  * argv[] of the file that caused a problem (i.e, argv[x]
1723  * returns x+1)
1724  * Otherwise, 0 is returned if all files retrieved successfully.
1725  */
1726 int
1727 auto_fetch(int argc, char *argv[])
1728 {
1729         volatile int    argpos, rval;
1730
1731         argpos = rval = 0;
1732
1733         if (sigsetjmp(toplevel, 1)) {
1734                 if (connected)
1735                         disconnect(0, NULL);
1736                 if (rval > 0)
1737                         rval = argpos + 1;
1738                 return (rval);
1739         }
1740         (void)xsignal(SIGINT, intr);
1741         (void)xsignal(SIGPIPE, lostpeer);
1742
1743         /*
1744          * Loop through as long as there's files to fetch.
1745          */
1746         for (; (rval == 0) && (argpos < argc); argpos++) {
1747                 if (strchr(argv[argpos], ':') == NULL)
1748                         break;
1749                 redirect_loop = 0;
1750                 if (!anonftp)
1751                         anonftp = 2;    /* Handle "automatic" transfers. */
1752                 rval = go_fetch(argv[argpos]);
1753                 if (outfile != NULL && strcmp(outfile, "-") != 0
1754                     && outfile[0] != '|')
1755                         outfile = NULL;
1756                 if (rval > 0)
1757                         rval = argpos + 1;
1758         }
1759
1760         if (connected && rval != -1)
1761                 disconnect(0, NULL);
1762         return (rval);
1763 }
1764
1765
1766 /*
1767  * Upload multiple files from the command line.
1768  *
1769  * If an error occurs the return value will be the offset+1 in
1770  * argv[] of the file that caused a problem (i.e, argv[x]
1771  * returns x+1)
1772  * Otherwise, 0 is returned if all files uploaded successfully.
1773  */
1774 int
1775 auto_put(int argc, char **argv, const char *uploadserver)
1776 {
1777         char    *uargv[4], *path, *pathsep;
1778         int      uargc, rval, argpos;
1779         size_t   len;
1780
1781         uargc = 0;
1782         uargv[uargc++] = "mput";
1783         uargv[uargc++] = argv[0];
1784         uargv[2] = uargv[3] = NULL;
1785         pathsep = NULL;
1786         rval = 1;
1787
1788         DPRINTF("auto_put: target `%s'\n", uploadserver);
1789
1790         path = ftp_strdup(uploadserver);
1791         len = strlen(path);
1792         if (path[len - 1] != '/' && path[len - 1] != ':') {
1793                         /*
1794                          * make sure we always pass a directory to auto_fetch
1795                          */
1796                 if (argc > 1) {         /* more than one file to upload */
1797                         len = strlen(uploadserver) + 2; /* path + "/" + "\0" */
1798                         free(path);
1799                         path = (char *)ftp_malloc(len);
1800                         (void)strlcpy(path, uploadserver, len);
1801                         (void)strlcat(path, "/", len);
1802                 } else {                /* single file to upload */
1803                         uargv[0] = "put";
1804                         pathsep = strrchr(path, '/');
1805                         if (pathsep == NULL) {
1806                                 pathsep = strrchr(path, ':');
1807                                 if (pathsep == NULL) {
1808                                         warnx("Invalid URL `%s'", path);
1809                                         goto cleanup_auto_put;
1810                                 }
1811                                 pathsep++;
1812                                 uargv[2] = ftp_strdup(pathsep);
1813                                 pathsep[0] = '/';
1814                         } else
1815                                 uargv[2] = ftp_strdup(pathsep + 1);
1816                         pathsep[1] = '\0';
1817                         uargc++;
1818                 }
1819         }
1820         DPRINTF("auto_put: URL `%s' argv[2] `%s'\n",
1821             path, uargv[2] ? uargv[2] : "<null>");
1822
1823                         /* connect and cwd */
1824         rval = auto_fetch(1, &path);
1825         if(rval >= 0)
1826                 goto cleanup_auto_put;
1827
1828         rval = 0;
1829
1830                         /* target filename provided; upload 1 file */
1831                         /* XXX : is this the best way? */
1832         if (uargc == 3) {
1833                 uargv[1] = argv[0];
1834                 put(uargc, uargv);
1835                 if ((code / 100) != COMPLETE)
1836                         rval = 1;
1837         } else {        /* otherwise a target dir: upload all files to it */
1838                 for(argpos = 0; argv[argpos] != NULL; argpos++) {
1839                         uargv[1] = argv[argpos];
1840                         mput(uargc, uargv);
1841                         if ((code / 100) != COMPLETE) {
1842                                 rval = argpos + 1;
1843                                 break;
1844                         }
1845                 }
1846         }
1847
1848  cleanup_auto_put:
1849         free(path);
1850         FREEPTR(uargv[2]);
1851         return (rval);
1852 }