contrib/lukemftp/src/fetch.c

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