contrib/tnftp/fetch.c

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