networking/wget.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * wget - retrieve a file using HTTP or FTP
   4  *
   5  * Chip Rosenthal Covad Communications <chip@laserlink.net>
   6  * Licensed under GPLv2, see file LICENSE in this source tree.
   7  *
   8  * Copyright (C) 2010 Bradley M. Kuhn <bkuhn@ebb.org>
   9  * Kuhn's copyrights are licensed GPLv2-or-later.  File as a whole remains GPLv2.
  10  */
  11 //config:config WGET
  12 //config:       bool "wget (35 kb)"
  13 //config:       default y
  14 //config:       help
  15 //config:       wget is a utility for non-interactive download of files from HTTP
  16 //config:       and FTP servers.
  17 //config:
  18 //config:config FEATURE_WGET_LONG_OPTIONS
  19 //config:       bool "Enable long options"
  20 //config:       default y
  21 //config:       depends on WGET && LONG_OPTS
  22 //config:
  23 //config:config FEATURE_WGET_STATUSBAR
  24 //config:       bool "Enable progress bar (+2k)"
  25 //config:       default y
  26 //config:       depends on WGET
  27 //config:
  28 //config:config FEATURE_WGET_AUTHENTICATION
  29 //config:       bool "Enable HTTP authentication"
  30 //config:       default y
  31 //config:       depends on WGET
  32 //config:       help
  33 //config:       Support authenticated HTTP transfers.
  34 //config:
  35 //config:config FEATURE_WGET_TIMEOUT
  36 //config:       bool "Enable timeout option -T SEC"
  37 //config:       default y
  38 //config:       depends on WGET
  39 //config:       help
  40 //config:       Supports network read and connect timeouts for wget,
  41 //config:       so that wget will give up and timeout, through the -T
  42 //config:       command line option.
  43 //config:
  44 //config:       Currently only connect and network data read timeout are
  45 //config:       supported (i.e., timeout is not applied to the DNS query). When
  46 //config:       FEATURE_WGET_LONG_OPTIONS is also enabled, the --timeout option
  47 //config:       will work in addition to -T.
  48 //config:
  49 //config:config FEATURE_WGET_HTTPS
  50 //config:       bool "Support HTTPS using internal TLS code"
  51 //it also enables FTPS support, but it's not well tested yet
  52 //config:       default y
  53 //config:       depends on WGET
  54 //config:       select TLS
  55 //config:       help
  56 //config:       wget will use internal TLS code to connect to https:// URLs.
  57 //config:       Note:
  58 //config:       On NOMMU machines, ssl_helper applet should be available
  59 //config:       in the $PATH for this to work. Make sure to select that applet.
  60 //config:
  61 //config:       Note: currently, TLS code only makes TLS I/O work, it
  62 //config:       does *not* check that the peer is who it claims to be, etc.
  63 //config:       IOW: it uses peer-supplied public keys to establish encryption
  64 //config:       and signing keys, then encrypts and signs outgoing data and
  65 //config:       decrypts incoming data.
  66 //config:       It does not check signature hashes on the incoming data:
  67 //config:       this means that attackers manipulating TCP packets can
  68 //config:       send altered data and we unknowingly receive garbage.
  69 //config:       (This check might be relatively easy to add).
  70 //config:       It does not check public key's certificate:
  71 //config:       this means that the peer may be an attacker impersonating
  72 //config:       the server we think we are talking to.
  73 //config:
  74 //config:       If you think this is unacceptable, consider this. As more and more
  75 //config:       servers switch to HTTPS-only operation, without such "crippled"
  76 //config:       TLS code it is *impossible* to simply download a kernel source
  77 //config:       from kernel.org. Which can in real world translate into
  78 //config:       "my small automatic tooling to build cross-compilers from sources
  79 //config:       no longer works, I need to additionally keep a local copy
  80 //config:       of ~4 megabyte source tarball of a SSL library and ~2 megabyte
  81 //config:       source of wget, need to compile and built both before I can
  82 //config:       download anything. All this despite the fact that the build
  83 //config:       is done in a QEMU sandbox on a machine with absolutely nothing
  84 //config:       worth stealing, so I don't care if someone would go to a lot
  85 //config:       of trouble to intercept my HTTPS download to send me an altered
  86 //config:       kernel tarball".
  87 //config:
  88 //config:       If you still think this is unacceptable, send patches.
  89 //config:
  90 //config:       If you still think this is unacceptable, do not want to send
  91 //config:       patches, but do want to waste bandwidth expaining how wrong
  92 //config:       it is, you will be ignored.
  93 //config:
  94 //config:config FEATURE_WGET_OPENSSL
  95 //config:       bool "Try to connect to HTTPS using openssl"
  96 //config:       default y
  97 //config:       depends on WGET
  98 //config:       help
  99 //config:       Try to use openssl to handle HTTPS.
 100 //config:
 101 //config:       OpenSSL has a simple SSL client for debug purposes.
 102 //config:       If you select this option, wget will effectively run:
 103 //config:       "openssl s_client -quiet -connect hostname:443
 104 //config:       -servername hostname 2>/dev/null" and pipe its data
 105 //config:       through it. -servername is not used if hostname is numeric.
 106 //config:       Note inconvenient API: host resolution is done twice,
 107 //config:       and there is no guarantee openssl's idea of IPv6 address
 108 //config:       format is the same as ours.
 109 //config:       Another problem is that s_client prints debug information
 110 //config:       to stderr, and it needs to be suppressed. This means
 111 //config:       all error messages get suppressed too.
 112 //config:       openssl is also a big binary, often dynamically linked
 113 //config:       against ~15 libraries.
 114 //config:
 115 //config:       If openssl can't be executed, internal TLS code will be used
 116 //config:       (if you enabled it); if openssl can be executed but fails later,
 117 //config:       wget can't detect this, and download will fail.
 118
 119 //applet:IF_WGET(APPLET(wget, BB_DIR_USR_BIN, BB_SUID_DROP))
 120
 121 //kbuild:lib-$(CONFIG_WGET) += wget.o
 122
 123 //usage:#define wget_trivial_usage
 124 //usage:        IF_FEATURE_WGET_LONG_OPTIONS(
 125 //usage:       "[-c|--continue] [--spider] [-q|--quiet] [-O|--output-document FILE]\n"
 126 //usage:       "        [--header 'header: value'] [-Y|--proxy on/off] [-P DIR]\n"
 127 /* Since we ignore these opts, we don't show them in --help */
 128 /* //usage:    "        [--no-check-certificate] [--no-cache] [--passive-ftp] [-t TRIES]" */
 129 /* //usage:    "        [-nv] [-nc] [-nH] [-np]" */
 130 //usage:       "        [-S|--server-response] [-U|--user-agent AGENT]" IF_FEATURE_WGET_TIMEOUT(" [-T SEC]") " URL..."
 131 //usage:        )
 132 //usage:        IF_NOT_FEATURE_WGET_LONG_OPTIONS(
 133 //usage:       "[-cq] [-O FILE] [-Y on/off] [-P DIR] [-S] [-U AGENT]"
 134 //usage:                        IF_FEATURE_WGET_TIMEOUT(" [-T SEC]") " URL..."
 135 //usage:        )
 136 //usage:#define wget_full_usage "\n\n"
 137 //usage:       "Retrieve files via HTTP or FTP\n"
 138 //usage:        IF_FEATURE_WGET_LONG_OPTIONS(
 139 //usage:     "\n        --spider        Only check URL existence: $? is 0 if exists"
 140 //usage:        )
 141 //usage:     "\n        -c              Continue retrieval of aborted transfer"
 142 //usage:     "\n        -q              Quiet"
 143 //usage:     "\n        -P DIR          Save to DIR (default .)"
 144 //usage:     "\n        -S              Show server response"
 145 //usage:        IF_FEATURE_WGET_TIMEOUT(
 146 //usage:     "\n        -T SEC          Network read timeout is SEC seconds"
 147 //usage:        )
 148 //usage:     "\n        -O FILE         Save to FILE ('-' for stdout)"
 149 //usage:     "\n        -U STR          Use STR for User-Agent header"
 150 //usage:     "\n        -Y on/off       Use proxy"
 151
 152 #include "libbb.h"
 153
 154 #if 0
 155 # define log_io(...) bb_error_msg(__VA_ARGS__)
 156 # define SENDFMT(fp, fmt, ...) \
 157         do { \
 158                 log_io("> " fmt, ##__VA_ARGS__); \
 159                 fprintf(fp, fmt, ##__VA_ARGS__); \
 160         } while (0);
 161 #else
 162 # define log_io(...) ((void)0)
 163 # define SENDFMT(fp, fmt, ...) fprintf(fp, fmt, ##__VA_ARGS__)
 164 #endif
 165
 166
 167 #define SSL_SUPPORTED (ENABLE_FEATURE_WGET_OPENSSL || ENABLE_FEATURE_WGET_HTTPS)
 168
 169 struct host_info {
 170         char *allocated;
 171         const char *path;
 172         char       *user;
 173         const char *protocol;
 174         char       *host;
 175         int         port;
 176 };
 177 static const char P_FTP[] ALIGN1 = "ftp";
 178 static const char P_HTTP[] ALIGN1 = "http";
 179 #if SSL_SUPPORTED
 180 # if ENABLE_FEATURE_WGET_HTTPS
 181 static const char P_FTPS[] ALIGN1 = "ftps";
 182 # endif
 183 static const char P_HTTPS[] ALIGN1 = "https";
 184 #endif
 185
 186 #if ENABLE_FEATURE_WGET_LONG_OPTIONS
 187 /* User-specified headers prevent using our corresponding built-in headers.  */
 188 enum {
 189         HDR_HOST          = (1<<0),
 190         HDR_USER_AGENT    = (1<<1),
 191         HDR_RANGE         = (1<<2),
 192         HDR_AUTH          = (1<<3) * ENABLE_FEATURE_WGET_AUTHENTICATION,
 193         HDR_PROXY_AUTH    = (1<<4) * ENABLE_FEATURE_WGET_AUTHENTICATION,
 194 };
 195 static const char wget_user_headers[] ALIGN1 =
 196         "Host:\0"
 197         "User-Agent:\0"
 198         "Range:\0"
 199 # if ENABLE_FEATURE_WGET_AUTHENTICATION
 200         "Authorization:\0"
 201         "Proxy-Authorization:\0"
 202 # endif
 203         ;
 204 # define USR_HEADER_HOST       (G.user_headers & HDR_HOST)
 205 # define USR_HEADER_USER_AGENT (G.user_headers & HDR_USER_AGENT)
 206 # define USR_HEADER_RANGE      (G.user_headers & HDR_RANGE)
 207 # define USR_HEADER_AUTH       (G.user_headers & HDR_AUTH)
 208 # define USR_HEADER_PROXY_AUTH (G.user_headers & HDR_PROXY_AUTH)
 209 #else /* No long options, no user-headers :( */
 210 # define USR_HEADER_HOST       0
 211 # define USR_HEADER_USER_AGENT 0
 212 # define USR_HEADER_RANGE      0
 213 # define USR_HEADER_AUTH       0
 214 # define USR_HEADER_PROXY_AUTH 0
 215 #endif
 216
 217 /* Globals */
 218 struct globals {
 219         off_t content_len;        /* Content-length of the file */
 220         off_t beg_range;          /* Range at which continue begins */
 221 #if ENABLE_FEATURE_WGET_STATUSBAR
 222         off_t transferred;        /* Number of bytes transferred so far */
 223         const char *curfile;      /* Name of current file being transferred */
 224         bb_progress_t pmt;
 225 #endif
 226         char *dir_prefix;
 227 #if ENABLE_FEATURE_WGET_LONG_OPTIONS
 228         char *post_data;
 229         char *extra_headers;
 230         unsigned char user_headers; /* Headers mentioned by the user */
 231 #endif
 232         char *fname_out;        /* where to direct output (-O) */
 233         const char *proxy_flag; /* Use proxies if env vars are set */
 234         const char *user_agent; /* "User-Agent" header field */
 235 #if ENABLE_FEATURE_WGET_TIMEOUT
 236         unsigned timeout_seconds;
 237         bool die_if_timed_out;
 238 #endif
 239         int output_fd;
 240         int o_flags;
 241         smallint chunked;         /* chunked transfer encoding */
 242         smallint got_clen;        /* got content-length: from server  */
 243         /* Local downloads do benefit from big buffer.
 244          * With 512 byte buffer, it was measured to be
 245          * an order of magnitude slower than with big one.
 246          */
 247         uint64_t just_to_align_next_member;
 248         char wget_buf[CONFIG_FEATURE_COPYBUF_KB*1024];
 249 } FIX_ALIASING;
 250 #define G (*ptr_to_globals)
 251 #define INIT_G() do { \
 252         SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
 253 } while (0)
 254 #define FINI_G() do { \
 255         FREE_PTR_TO_GLOBALS(); \
 256 } while (0)
 257
 258
 259 /* Must match option string! */
 260 enum {
 261         WGET_OPT_CONTINUE   = (1 << 0),
 262         WGET_OPT_QUIET      = (1 << 1),
 263         WGET_OPT_SERVER_RESPONSE = (1 << 2),
 264         WGET_OPT_OUTNAME    = (1 << 3),
 265         WGET_OPT_PREFIX     = (1 << 4),
 266         WGET_OPT_PROXY      = (1 << 5),
 267         WGET_OPT_USER_AGENT = (1 << 6),
 268         WGET_OPT_NETWORK_READ_TIMEOUT = (1 << 7),
 269         WGET_OPT_RETRIES    = (1 << 8),
 270         WGET_OPT_nsomething = (1 << 9),
 271         WGET_OPT_HEADER     = (1 << 10) * ENABLE_FEATURE_WGET_LONG_OPTIONS,
 272         WGET_OPT_POST_DATA  = (1 << 11) * ENABLE_FEATURE_WGET_LONG_OPTIONS,
 273         WGET_OPT_SPIDER     = (1 << 12) * ENABLE_FEATURE_WGET_LONG_OPTIONS,
 274 };
 275
 276 enum {
 277         PROGRESS_START = -1,
 278         PROGRESS_END   = 0,
 279         PROGRESS_BUMP  = 1,
 280 };
 281 #if ENABLE_FEATURE_WGET_STATUSBAR
 282 static void progress_meter(int flag)
 283 {
 284         if (option_mask32 & WGET_OPT_QUIET)
 285                 return;
 286
 287         if (flag == PROGRESS_START)
 288                 bb_progress_init(&G.pmt, G.curfile);
 289
 290         bb_progress_update(&G.pmt,
 291                         G.beg_range,
 292                         G.transferred,
 293                         (G.chunked || !G.got_clen) ? 0 : G.beg_range + G.transferred + G.content_len
 294         );
 295
 296         if (flag == PROGRESS_END) {
 297                 bb_progress_free(&G.pmt);
 298                 bb_putchar_stderr('\n');
 299                 G.transferred = 0;
 300         }
 301 }
 302 #else
 303 static ALWAYS_INLINE void progress_meter(int flag UNUSED_PARAM) { }
 304 #endif
 305
 306
 307 /* IPv6 knows scoped address types i.e. link and site local addresses. Link
 308  * local addresses can have a scope identifier to specify the
 309  * interface/link an address is valid on (e.g. fe80::1%eth0). This scope
 310  * identifier is only valid on a single node.
 311  *
 312  * RFC 4007 says that the scope identifier MUST NOT be sent across the wire,
 313  * unless all nodes agree on the semantic. Apache e.g. regards zone identifiers
 314  * in the Host header as invalid requests, see
 315  * https://issues.apache.org/bugzilla/show_bug.cgi?id=35122
 316  */
 317 static void strip_ipv6_scope_id(char *host)
 318 {
 319         char *scope, *cp;
 320
 321         /* bbox wget actually handles IPv6 addresses without [], like
 322          * wget "http://::1/xxx", but this is not standard.
 323          * To save code, _here_ we do not support it. */
 324
 325         if (host[0] != '[')
 326                 return; /* not IPv6 */
 327
 328         scope = strchr(host, '%');
 329         if (!scope)
 330                 return;
 331
 332         /* Remove the IPv6 zone identifier from the host address */
 333         cp = strchr(host, ']');
 334         if (!cp || (cp[1] != ':' && cp[1] != '\0')) {
 335                 /* malformed address (not "[xx]:nn" or "[xx]") */
 336                 return;
 337         }
 338
 339         /* cp points to "]...", scope points to "%eth0]..." */
 340         overlapping_strcpy(scope, cp);
 341 }
 342
 343 #if ENABLE_FEATURE_WGET_AUTHENTICATION
 344 /* Base64-encode character string. */
 345 static char *base64enc(const char *str)
 346 {
 347         unsigned len = strlen(str);
 348         if (len > sizeof(G.wget_buf)/4*3 - 10) /* paranoia */
 349                 len = sizeof(G.wget_buf)/4*3 - 10;
 350         bb_uuencode(G.wget_buf, str, len, bb_uuenc_tbl_base64);
 351         return G.wget_buf;
 352 }
 353 #endif
 354
 355 #if ENABLE_FEATURE_WGET_TIMEOUT
 356 static void alarm_handler(int sig UNUSED_PARAM)
 357 {
 358         /* This is theoretically unsafe (uses stdio and malloc in signal handler) */
 359         if (G.die_if_timed_out)
 360                 bb_error_msg_and_die("download timed out");
 361 }
 362 static void set_alarm(void)
 363 {
 364         if (G.timeout_seconds) {
 365                 alarm(G.timeout_seconds);
 366                 G.die_if_timed_out = 1;
 367         }
 368 }
 369 # define clear_alarm() ((void)(G.die_if_timed_out = 0))
 370 #else
 371 # define set_alarm()   ((void)0)
 372 # define clear_alarm() ((void)0)
 373 #endif
 374
 375 #if ENABLE_FEATURE_WGET_OPENSSL
 376 /*
 377  * is_ip_address() attempts to verify whether or not a string
 378  * contains an IPv4 or IPv6 address (vs. an FQDN).  The result
 379  * of inet_pton() can be used to determine this.
 380  *
 381  * TODO add proper error checking when inet_pton() returns -1
 382  * (some form of system error has occurred, and errno is set)
 383  */
 384 static int is_ip_address(const char *string)
 385 {
 386         struct sockaddr_in sa;
 387
 388         int result = inet_pton(AF_INET, string, &(sa.sin_addr));
 389 # if ENABLE_FEATURE_IPV6
 390         if (result == 0) {
 391                 struct sockaddr_in6 sa6;
 392                 result = inet_pton(AF_INET6, string, &(sa6.sin6_addr));
 393         }
 394 # endif
 395         return (result == 1);
 396 }
 397 #endif
 398
 399 static FILE *open_socket(len_and_sockaddr *lsa)
 400 {
 401         int fd;
 402         FILE *fp;
 403
 404         set_alarm();
 405         fd = xconnect_stream(lsa);
 406         clear_alarm();
 407
 408         /* glibc 2.4 seems to try seeking on it - ??! */
 409         /* hopefully it understands what ESPIPE means... */
 410         fp = fdopen(fd, "r+");
 411         if (!fp)
 412                 bb_die_memory_exhausted();
 413
 414         return fp;
 415 }
 416
 417 /* We balk at any control chars in other side's messages.
 418  * This prevents nasty surprises (e.g. ESC sequences) in "Location:" URLs
 419  * and error messages.
 420  *
 421  * The only exception is tabs, which are converted to (one) space:
 422  * HTTP's "headers: <whitespace> values" may have those.
 423  */
 424 static char* sanitize_string(char *s)
 425 {
 426         unsigned char *p = (void *) s;
 427         while (*p) {
 428                 if (*p < ' ') {
 429                         if (*p != '\t')
 430                                 break;
 431                         *p = ' ';
 432                 }
 433                 p++;
 434         }
 435         *p = '\0';
 436         return s;
 437 }
 438
 439 /* Returns '\n' if it was seen, else '\0'. Trims at first '\r' or '\n' */
 440 static char fgets_trim_sanitize(FILE *fp, const char *fmt)
 441 {
 442         char c;
 443         char *buf_ptr;
 444
 445         set_alarm();
 446         if (fgets(G.wget_buf, sizeof(G.wget_buf), fp) == NULL)
 447                 bb_perror_msg_and_die("error getting response");
 448         clear_alarm();
 449
 450         buf_ptr = strchrnul(G.wget_buf, '\n');
 451         c = *buf_ptr;
 452 #if 1
 453         /* Disallow any control chars: trim at first char < 0x20 */
 454         sanitize_string(G.wget_buf);
 455 #else
 456         *buf_ptr = '\0';
 457         buf_ptr = strchrnul(G.wget_buf, '\r');
 458         *buf_ptr = '\0';
 459 #endif
 460
 461         log_io("< %s", G.wget_buf);
 462
 463         if (fmt && (option_mask32 & WGET_OPT_SERVER_RESPONSE))
 464                 fprintf(stderr, fmt, G.wget_buf);
 465
 466         return c;
 467 }
 468
 469 static int ftpcmd(const char *s1, const char *s2, FILE *fp)
 470 {
 471         int result;
 472         if (s1) {
 473                 if (!s2)
 474                         s2 = "";
 475                 fprintf(fp, "%s%s\r\n", s1, s2);
 476                 /* With --server-response, wget also shows its ftp commands */
 477                 if (option_mask32 & WGET_OPT_SERVER_RESPONSE)
 478                         fprintf(stderr, "--> %s%s\n\n", s1, s2);
 479                 fflush(fp);
 480                 log_io("> %s%s", s1, s2);
 481         }
 482
 483         /* Read until "Nxx something" is received */
 484         G.wget_buf[3] = 0;
 485         do {
 486                 fgets_trim_sanitize(fp, "%s\n");
 487         } while (!isdigit(G.wget_buf[0]) || G.wget_buf[3] != ' ');
 488
 489         G.wget_buf[3] = '\0';
 490         result = xatoi_positive(G.wget_buf);
 491         G.wget_buf[3] = ' ';
 492         return result;
 493 }
 494
 495 static void parse_url(const char *src_url, struct host_info *h)
 496 {
 497         char *url, *p, *sp;
 498
 499         free(h->allocated);
 500         h->allocated = url = xstrdup(src_url);
 501
 502         h->protocol = P_FTP;
 503         p = strstr(url, "://");
 504         if (p) {
 505                 *p = '\0';
 506                 h->host = p + 3;
 507                 if (strcmp(url, P_FTP) == 0) {
 508                         h->port = bb_lookup_port(P_FTP, "tcp", 21);
 509                 } else
 510 #if SSL_SUPPORTED
 511 # if ENABLE_FEATURE_WGET_HTTPS
 512                 if (strcmp(url, P_FTPS) == 0) {
 513                         h->port = bb_lookup_port(P_FTPS, "tcp", 990);
 514                         h->protocol = P_FTPS;
 515                 } else
 516 # endif
 517                 if (strcmp(url, P_HTTPS) == 0) {
 518                         h->port = bb_lookup_port(P_HTTPS, "tcp", 443);
 519                         h->protocol = P_HTTPS;
 520                 } else
 521 #endif
 522                 if (strcmp(url, P_HTTP) == 0) {
 523  http:
 524                         h->port = bb_lookup_port(P_HTTP, "tcp", 80);
 525                         h->protocol = P_HTTP;
 526                 } else {
 527                         *p = ':';
 528                         bb_error_msg_and_die("not an http or ftp url: %s", url);
 529                 }
 530         } else {
 531                 // GNU wget is user-friendly and falls back to http://
 532                 h->host = url;
 533                 goto http;
 534         }
 535
 536         // FYI:
 537         // "Real" wget 'http://busybox.net?var=a/b' sends this request:
 538         //   'GET /?var=a/b HTTP/1.0'
 539         //   and saves 'index.html?var=a%2Fb' (we save 'b')
 540         // wget 'http://busybox.net?login=john@doe':
 541         //   request: 'GET /?login=john@doe HTTP/1.0'
 542         //   saves: 'index.html?login=john@doe' (we save '?login=john@doe')
 543         // wget 'http://busybox.net#test/test':
 544         //   request: 'GET / HTTP/1.0'
 545         //   saves: 'index.html' (we save 'test')
 546         //
 547         // We also don't add unique .N suffix if file exists...
 548         sp = strchr(h->host, '/');
 549         p = strchr(h->host, '?'); if (!sp || (p && sp > p)) sp = p;
 550         p = strchr(h->host, '#'); if (!sp || (p && sp > p)) sp = p;
 551         if (!sp) {
 552                 h->path = "";
 553         } else if (*sp == '/') {
 554                 *sp = '\0';
 555                 h->path = sp + 1;
 556         } else { // '#' or '?'
 557                 // http://busybox.net?login=john@doe is a valid URL
 558                 // memmove converts to:
 559                 // http:/busybox.nett?login=john@doe...
 560                 memmove(h->host - 1, h->host, sp - h->host);
 561                 h->host--;
 562                 sp[-1] = '\0';
 563                 h->path = sp;
 564         }
 565
 566         sp = strrchr(h->host, '@');
 567         if (sp != NULL) {
 568                 // URL-decode "user:password" string before base64-encoding:
 569                 // wget http://test:my%20pass@example.com should send
 570                 // Authorization: Basic dGVzdDpteSBwYXNz
 571                 // which decodes to "test:my pass".
 572                 // Standard wget and curl do this too.
 573                 *sp = '\0';
 574                 free(h->user);
 575                 h->user = xstrdup(percent_decode_in_place(h->host, /*strict:*/ 0));
 576                 h->host = sp + 1;
 577         }
 578         /* else: h->user remains NULL, or as set by original request
 579          * before redirect (if we are here after a redirect).
 580          */
 581 }
 582
 583 static char *get_sanitized_hdr(FILE *fp)
 584 {
 585         char *s, *hdrval;
 586         int c;
 587
 588         /* retrieve header line */
 589         c = fgets_trim_sanitize(fp, "  %s\n");
 590
 591         /* end of the headers? */
 592         if (G.wget_buf[0] == '\0')
 593                 return NULL;
 594
 595         /* convert the header name to lower case */
 596         for (s = G.wget_buf; isalnum(*s) || *s == '-' || *s == '.' || *s == '_'; ++s) {
 597                 /*
 598                  * No-op for 20-3f and 60-7f. "0-9a-z-." are in these ranges.
 599                  * 40-5f range ("@A-Z[\]^_") maps to 60-7f.
 600                  * "A-Z" maps to "a-z".
 601                  * "@[\]" can't occur in header names.
 602                  * "^_" maps to "~,DEL" (which is wrong).
 603                  * "^" was never seen yet, "_" was seen from web.archive.org
 604                  * (x-archive-orig-x_commoncrawl_Signature: HEXSTRING).
 605                  */
 606                 *s |= 0x20;
 607         }
 608
 609         /* verify we are at the end of the header name */
 610         if (*s != ':')
 611                 bb_error_msg_and_die("bad header line: %s", G.wget_buf);
 612
 613         /* locate the start of the header value */
 614         *s++ = '\0';
 615         hdrval = skip_whitespace(s);
 616
 617         if (c != '\n') {
 618                 /* Rats! The buffer isn't big enough to hold the entire header value */
 619                 while (c = getc(fp), c != EOF && c != '\n')
 620                         continue;
 621         }
 622
 623         return hdrval;
 624 }
 625
 626 static void reset_beg_range_to_zero(void)
 627 {
 628         bb_error_msg("restart failed");
 629         G.beg_range = 0;
 630         xlseek(G.output_fd, 0, SEEK_SET);
 631         /* Done at the end instead: */
 632         /* ftruncate(G.output_fd, 0); */
 633 }
 634
 635 #if ENABLE_FEATURE_WGET_OPENSSL
 636 static int spawn_https_helper_openssl(const char *host, unsigned port)
 637 {
 638         char *allocated = NULL;
 639         char *servername;
 640         int sp[2];
 641         int pid;
 642         IF_FEATURE_WGET_HTTPS(volatile int child_failed = 0;)
 643
 644         if (socketpair(AF_UNIX, SOCK_STREAM, 0, sp) != 0)
 645                 /* Kernel can have AF_UNIX support disabled */
 646                 bb_perror_msg_and_die("socketpair");
 647
 648         if (!strchr(host, ':'))
 649                 host = allocated = xasprintf("%s:%u", host, port);
 650         servername = xstrdup(host);
 651         strrchr(servername, ':')[0] = '\0';
 652
 653         fflush_all();
 654         pid = xvfork();
 655         if (pid == 0) {
 656                 /* Child */
 657                 char *argv[8];
 658
 659                 close(sp[0]);
 660                 xmove_fd(sp[1], 0);
 661                 xdup2(0, 1);
 662                 /*
 663                  * openssl s_client -quiet -connect www.kernel.org:443 2>/dev/null
 664                  * It prints some debug stuff on stderr, don't know how to suppress it.
 665                  * Work around by dev-nulling stderr. We lose all error messages :(
 666                  */
 667                 xmove_fd(2, 3);
 668                 xopen("/dev/null", O_RDWR);
 669                 memset(&argv, 0, sizeof(argv));
 670                 argv[0] = (char*)"openssl";
 671                 argv[1] = (char*)"s_client";
 672                 argv[2] = (char*)"-quiet";
 673                 argv[3] = (char*)"-connect";
 674                 argv[4] = (char*)host;
 675                 /*
 676                  * Per RFC 6066 Section 3, the only permitted values in the
 677                  * TLS server_name (SNI) field are FQDNs (DNS hostnames).
 678                  * IPv4 and IPv6 addresses, port numbers are not allowed.
 679                  */
 680                 if (!is_ip_address(servername)) {
 681                         argv[5] = (char*)"-servername";
 682                         argv[6] = (char*)servername;
 683                 }
 684
 685                 BB_EXECVP(argv[0], argv);
 686                 xmove_fd(3, 2);
 687 # if ENABLE_FEATURE_WGET_HTTPS
 688                 child_failed = 1;
 689                 xfunc_die();
 690 # else
 691                 bb_perror_msg_and_die("can't execute '%s'", argv[0]);
 692 # endif
 693                 /* notreached */
 694         }
 695
 696         /* Parent */
 697         free(servername);
 698         free(allocated);
 699         close(sp[1]);
 700 # if ENABLE_FEATURE_WGET_HTTPS
 701         if (child_failed) {
 702                 close(sp[0]);
 703                 return -1;
 704         }
 705 # endif
 706         return sp[0];
 707 }
 708 #endif
 709
 710 #if ENABLE_FEATURE_WGET_HTTPS
 711 static void spawn_ssl_client(const char *host, int network_fd, int flags)
 712 {
 713         int sp[2];
 714         int pid;
 715         char *servername, *p;
 716
 717         servername = xstrdup(host);
 718         p = strrchr(servername, ':');
 719         if (p) *p = '\0';
 720
 721         if (socketpair(AF_UNIX, SOCK_STREAM, 0, sp) != 0)
 722                 /* Kernel can have AF_UNIX support disabled */
 723                 bb_perror_msg_and_die("socketpair");
 724
 725         fflush_all();
 726         pid = BB_MMU ? xfork() : xvfork();
 727         if (pid == 0) {
 728                 /* Child */
 729                 close(sp[0]);
 730                 xmove_fd(sp[1], 0);
 731                 xdup2(0, 1);
 732                 if (BB_MMU) {
 733                         tls_state_t *tls = new_tls_state();
 734                         tls->ifd = tls->ofd = network_fd;
 735                         tls_handshake(tls, servername);
 736                         tls_run_copy_loop(tls, flags);
 737                         exit(0);
 738                 } else {
 739                         char *argv[6];
 740
 741                         xmove_fd(network_fd, 3);
 742                         argv[0] = (char*)"ssl_client";
 743                         argv[1] = (char*)"-s3";
 744                         //TODO: if (!is_ip_address(servername))...
 745                         argv[2] = (char*)"-n";
 746                         argv[3] = servername;
 747                         argv[4] = (flags & TLSLOOP_EXIT_ON_LOCAL_EOF ? (char*)"-e" : NULL);
 748                         argv[5] = NULL;
 749                         BB_EXECVP(argv[0], argv);
 750                         bb_perror_msg_and_die("can't execute '%s'", argv[0]);
 751                 }
 752                 /* notreached */
 753         }
 754
 755         /* Parent */
 756         free(servername);
 757         close(sp[1]);
 758         xmove_fd(sp[0], network_fd);
 759 }
 760 #endif
 761
 762 static FILE* prepare_ftp_session(FILE **dfpp, struct host_info *target, len_and_sockaddr *lsa)
 763 {
 764         FILE *sfp;
 765         char *pass;
 766         int port;
 767
 768         sfp = open_socket(lsa);
 769 #if ENABLE_FEATURE_WGET_HTTPS
 770         if (target->protocol == P_FTPS)
 771                 spawn_ssl_client(target->host, fileno(sfp), TLSLOOP_EXIT_ON_LOCAL_EOF);
 772 #endif
 773
 774         if (ftpcmd(NULL, NULL, sfp) != 220)
 775                 bb_error_msg_and_die("%s", G.wget_buf);
 776                 /* note: ftpcmd() sanitizes G.wget_buf, ok to print */
 777
 778         /* Split username:password pair */
 779         pass = (char*)"busybox"; /* password for "anonymous" */
 780         if (target->user) {
 781                 pass = strchr(target->user, ':');
 782                 if (pass)
 783                         *pass++ = '\0';
 784         }
 785
 786         /* Log in */
 787         switch (ftpcmd("USER ", target->user ?: "anonymous", sfp)) {
 788         case 230:
 789                 break;
 790         case 331:
 791                 if (ftpcmd("PASS ", pass, sfp) == 230)
 792                         break;
 793                 /* fall through (failed login) */
 794         default:
 795                 bb_error_msg_and_die("ftp login: %s", G.wget_buf);
 796         }
 797
 798         ftpcmd("TYPE I", NULL, sfp);
 799
 800         /* Query file size */
 801         if (ftpcmd("SIZE ", target->path, sfp) == 213) {
 802                 G.content_len = BB_STRTOOFF(G.wget_buf + 4, NULL, 10);
 803                 if (G.content_len < 0 || errno) {
 804                         bb_error_msg_and_die("SIZE value is garbage");
 805                 }
 806                 G.got_clen = 1;
 807         }
 808
 809         /* Enter passive mode */
 810         if (ENABLE_FEATURE_IPV6 && ftpcmd("EPSV", NULL, sfp) == 229) {
 811                 /* good */
 812         } else
 813         if (ftpcmd("PASV", NULL, sfp) != 227) {
 814  pasv_error:
 815                 bb_error_msg_and_die("bad response to %s: %s", "PASV", G.wget_buf);
 816         }
 817         port = parse_pasv_epsv(G.wget_buf);
 818         if (port < 0)
 819                 goto pasv_error;
 820
 821         set_nport(&lsa->u.sa, htons(port));
 822
 823         *dfpp = open_socket(lsa);
 824
 825 #if ENABLE_FEATURE_WGET_HTTPS
 826         if (target->protocol == P_FTPS) {
 827                 /* "PROT P" enables encryption of data stream.
 828                  * Without it (or with "PROT C"), data is sent unencrypted.
 829                  */
 830                 if (ftpcmd("PROT P", NULL, sfp) == 200)
 831                         spawn_ssl_client(target->host, fileno(*dfpp), /*flags*/ 0);
 832         }
 833 #endif
 834
 835         if (G.beg_range != 0) {
 836                 sprintf(G.wget_buf, "REST %"OFF_FMT"u", G.beg_range);
 837                 if (ftpcmd(G.wget_buf, NULL, sfp) == 350)
 838                         G.content_len -= G.beg_range;
 839                 else
 840                         reset_beg_range_to_zero();
 841         }
 842
 843 //TODO: needs ftp-escaping 0xff and '\n' bytes here.
 844 //Or disallow '\n' altogether via sanitize_string() in parse_url().
 845 //But 0xff's are possible in valid utf8 filenames.
 846         if (ftpcmd("RETR ", target->path, sfp) > 150)
 847                 bb_error_msg_and_die("bad response to %s: %s", "RETR", G.wget_buf);
 848
 849         return sfp;
 850 }
 851
 852 static void NOINLINE retrieve_file_data(FILE *dfp)
 853 {
 854 #if ENABLE_FEATURE_WGET_STATUSBAR || ENABLE_FEATURE_WGET_TIMEOUT
 855 # if ENABLE_FEATURE_WGET_TIMEOUT
 856         unsigned second_cnt = G.timeout_seconds;
 857 # endif
 858         struct pollfd polldata;
 859
 860         polldata.fd = fileno(dfp);
 861         polldata.events = POLLIN | POLLPRI;
 862 #endif
 863         progress_meter(PROGRESS_START);
 864
 865         if (G.chunked)
 866                 goto get_clen;
 867
 868         /* Loops only if chunked */
 869         while (1) {
 870
 871 #if ENABLE_FEATURE_WGET_STATUSBAR || ENABLE_FEATURE_WGET_TIMEOUT
 872                 /* Must use nonblocking I/O, otherwise fread will loop
 873                  * and *block* until it reads full buffer,
 874                  * which messes up progress bar and/or timeout logic.
 875                  * Because of nonblocking I/O, we need to dance
 876                  * very carefully around EAGAIN. See explanation at
 877                  * clearerr() calls.
 878                  */
 879                 ndelay_on(polldata.fd);
 880 #endif
 881                 while (1) {
 882                         int n;
 883                         unsigned rdsz;
 884
 885 #if ENABLE_FEATURE_WGET_STATUSBAR || ENABLE_FEATURE_WGET_TIMEOUT
 886                         /* fread internally uses read loop, which in our case
 887                          * is usually exited when we get EAGAIN.
 888                          * In this case, libc sets error marker on the stream.
 889                          * Need to clear it before next fread to avoid possible
 890                          * rare false positive ferror below. Rare because usually
 891                          * fread gets more than zero bytes, and we don't fall
 892                          * into if (n <= 0) ...
 893                          */
 894                         clearerr(dfp);
 895 #endif
 896                         errno = 0;
 897                         rdsz = sizeof(G.wget_buf);
 898                         if (G.got_clen) {
 899                                 if (G.content_len < (off_t)sizeof(G.wget_buf)) {
 900                                         if ((int)G.content_len <= 0)
 901                                                 break;
 902                                         rdsz = (unsigned)G.content_len;
 903                                 }
 904                         }
 905                         n = fread(G.wget_buf, 1, rdsz, dfp);
 906
 907                         if (n > 0) {
 908                                 xwrite(G.output_fd, G.wget_buf, n);
 909 #if ENABLE_FEATURE_WGET_STATUSBAR
 910                                 G.transferred += n;
 911 #endif
 912                                 if (G.got_clen) {
 913                                         G.content_len -= n;
 914                                         if (G.content_len == 0)
 915                                                 break;
 916                                 }
 917 #if ENABLE_FEATURE_WGET_TIMEOUT
 918                                 second_cnt = G.timeout_seconds;
 919 #endif
 920                                 goto bump;
 921                         }
 922
 923                         /* n <= 0.
 924                          * man fread:
 925                          * If error occurs, or EOF is reached, the return value
 926                          * is a short item count (or zero).
 927                          * fread does not distinguish between EOF and error.
 928                          */
 929                         if (errno != EAGAIN) {
 930                                 if (ferror(dfp)) {
 931                                         progress_meter(PROGRESS_END);
 932                                         bb_perror_msg_and_die(bb_msg_read_error);
 933                                 }
 934                                 break; /* EOF, not error */
 935                         }
 936
 937 #if ENABLE_FEATURE_WGET_STATUSBAR || ENABLE_FEATURE_WGET_TIMEOUT
 938                         /* It was EAGAIN. There is no data. Wait up to one second
 939                          * then abort if timed out, or update the bar and try reading again.
 940                          */
 941                         if (safe_poll(&polldata, 1, 1000) == 0) {
 942 # if ENABLE_FEATURE_WGET_TIMEOUT
 943                                 if (second_cnt != 0 && --second_cnt == 0) {
 944                                         progress_meter(PROGRESS_END);
 945                                         bb_error_msg_and_die("download timed out");
 946                                 }
 947 # endif
 948                                 /* We used to loop back to poll here,
 949                                  * but there is no great harm in letting fread
 950                                  * to try reading anyway.
 951                                  */
 952                         }
 953 #endif
 954  bump:
 955                         /* Need to do it _every_ second for "stalled" indicator
 956                          * to be shown properly.
 957                          */
 958                         progress_meter(PROGRESS_BUMP);
 959                 } /* while (reading data) */
 960
 961 #if ENABLE_FEATURE_WGET_STATUSBAR || ENABLE_FEATURE_WGET_TIMEOUT
 962                 clearerr(dfp);
 963                 ndelay_off(polldata.fd); /* else fgets can get very unhappy */
 964 #endif
 965                 if (!G.chunked)
 966                         break;
 967
 968                 fgets_trim_sanitize(dfp, NULL); /* Eat empty line */
 969  get_clen:
 970                 fgets_trim_sanitize(dfp, NULL);
 971                 G.content_len = STRTOOFF(G.wget_buf, NULL, 16);
 972                 /* FIXME: error check? */
 973                 if (G.content_len == 0)
 974                         break; /* all done! */
 975                 G.got_clen = 1;
 976                 /*
 977                  * Note that fgets may result in some data being buffered in dfp.
 978                  * We loop back to fread, which will retrieve this data.
 979                  * Also note that code has to be arranged so that fread
 980                  * is done _before_ one-second poll wait - poll doesn't know
 981                  * about stdio buffering and can result in spurious one second waits!
 982                  */
 983         }
 984
 985         /* If -c failed, we restart from the beginning,
 986          * but we do not truncate file then, we do it only now, at the end.
 987          * This lets user to ^C if his 99% complete 10 GB file download
 988          * failed to restart *without* losing the almost complete file.
 989          */
 990         {
 991                 off_t pos = lseek(G.output_fd, 0, SEEK_CUR);
 992                 if (pos != (off_t)-1)
 993                         ftruncate(G.output_fd, pos);
 994         }
 995
 996         /* Draw full bar and free its resources */
 997         G.chunked = 0;  /* makes it show 100% even for chunked download */
 998         G.got_clen = 1; /* makes it show 100% even for download of (formerly) unknown size */
 999         progress_meter(PROGRESS_END);
1000 }
1001
1002 static void download_one_url(const char *url)
1003 {
1004         bool use_proxy;                 /* Use proxies if env vars are set  */
1005         int redir_limit;
1006         len_and_sockaddr *lsa;
1007         FILE *sfp;                      /* socket to web/ftp server         */
1008         FILE *dfp;                      /* socket to ftp server (data)      */
1009         char *fname_out_alloc;
1010         char *redirected_path = NULL;
1011         struct host_info server;
1012         struct host_info target;
1013
1014         server.allocated = NULL;
1015         target.allocated = NULL;
1016         server.user = NULL;
1017         target.user = NULL;
1018
1019         parse_url(url, &target);
1020
1021         /* Use the proxy if necessary */
1022         use_proxy = (strcmp(G.proxy_flag, "off") != 0);
1023         if (use_proxy) {
1024                 char *proxy = getenv(target.protocol[0] == 'f' ? "ftp_proxy" : "http_proxy");
1025 //FIXME: what if protocol is https? Ok to use http_proxy?
1026                 use_proxy = (proxy && proxy[0]);
1027                 if (use_proxy)
1028                         parse_url(proxy, &server);
1029         }
1030         if (!use_proxy) {
1031                 server.protocol = target.protocol;
1032                 server.port = target.port;
1033                 if (ENABLE_FEATURE_IPV6) {
1034                         //free(server.allocated); - can't be non-NULL
1035                         server.host = server.allocated = xstrdup(target.host);
1036                 } else {
1037                         server.host = target.host;
1038                 }
1039         }
1040
1041         if (ENABLE_FEATURE_IPV6)
1042                 strip_ipv6_scope_id(target.host);
1043
1044         /* If there was no -O FILE, guess output filename */
1045         fname_out_alloc = NULL;
1046         if (!(option_mask32 & WGET_OPT_OUTNAME)) {
1047                 G.fname_out = bb_get_last_path_component_nostrip(target.path);
1048                 /* handle "wget http://kernel.org//" */
1049                 if (G.fname_out[0] == '/' || !G.fname_out[0])
1050                         G.fname_out = (char*)"index.html";
1051                 /* -P DIR is considered only if there was no -O FILE */
1052                 if (G.dir_prefix)
1053                         G.fname_out = fname_out_alloc = concat_path_file(G.dir_prefix, G.fname_out);
1054                 else {
1055                         /* redirects may free target.path later, need to make a copy */
1056                         G.fname_out = fname_out_alloc = xstrdup(G.fname_out);
1057                 }
1058         }
1059 #if ENABLE_FEATURE_WGET_STATUSBAR
1060         G.curfile = bb_get_last_path_component_nostrip(G.fname_out);
1061 #endif
1062
1063         /* Determine where to start transfer */
1064         G.beg_range = 0;
1065         if (option_mask32 & WGET_OPT_CONTINUE) {
1066                 G.output_fd = open(G.fname_out, O_WRONLY);
1067                 if (G.output_fd >= 0) {
1068                         G.beg_range = xlseek(G.output_fd, 0, SEEK_END);
1069                 }
1070                 /* File doesn't exist. We do not create file here yet.
1071                  * We are not sure it exists on remote side */
1072         }
1073
1074         redir_limit = 5;
1075  resolve_lsa:
1076         lsa = xhost2sockaddr(server.host, server.port);
1077         if (!(option_mask32 & WGET_OPT_QUIET)) {
1078                 char *s = xmalloc_sockaddr2dotted(&lsa->u.sa);
1079                 fprintf(stderr, "Connecting to %s (%s)\n", server.host, s);
1080                 free(s);
1081         }
1082  establish_session:
1083         /*G.content_len = 0; - redundant, got_clen = 0 is enough */
1084         G.got_clen = 0;
1085         G.chunked = 0;
1086         if (use_proxy || target.protocol[0] != 'f' /*not ftp[s]*/) {
1087                 /*
1088                  *  HTTP session
1089                  */
1090                 char *str;
1091                 int status;
1092
1093                 /* Open socket to http(s) server */
1094 #if ENABLE_FEATURE_WGET_OPENSSL
1095                 /* openssl (and maybe internal TLS) support is configured */
1096                 if (server.protocol == P_HTTPS) {
1097                         /* openssl-based helper
1098                          * Inconvenient API since we can't give it an open fd
1099                          */
1100                         int fd = spawn_https_helper_openssl(server.host, server.port);
1101 # if ENABLE_FEATURE_WGET_HTTPS
1102                         if (fd < 0) { /* no openssl? try internal */
1103                                 sfp = open_socket(lsa);
1104                                 spawn_ssl_client(server.host, fileno(sfp), /*flags*/ 0);
1105                                 goto socket_opened;
1106                         }
1107 # else
1108                         /* We don't check for exec("openssl") failure in this case */
1109 # endif
1110                         sfp = fdopen(fd, "r+");
1111                         if (!sfp)
1112                                 bb_die_memory_exhausted();
1113                         goto socket_opened;
1114                 }
1115                 sfp = open_socket(lsa);
1116  socket_opened:
1117 #elif ENABLE_FEATURE_WGET_HTTPS
1118                 /* Only internal TLS support is configured */
1119                 sfp = open_socket(lsa);
1120                 if (server.protocol == P_HTTPS)
1121                         spawn_ssl_client(server.host, fileno(sfp), /*flags*/ 0);
1122 #else
1123                 /* ssl (https) support is not configured */
1124                 sfp = open_socket(lsa);
1125 #endif
1126                 /* Send HTTP request */
1127                 if (use_proxy) {
1128                         SENDFMT(sfp, "GET %s://%s/%s HTTP/1.1\r\n",
1129                                 target.protocol, target.host,
1130                                 target.path);
1131                 } else {
1132                         SENDFMT(sfp, "%s /%s HTTP/1.1\r\n",
1133                                 (option_mask32 & WGET_OPT_POST_DATA) ? "POST" : "GET",
1134                                 target.path);
1135                 }
1136                 if (!USR_HEADER_HOST)
1137                         SENDFMT(sfp, "Host: %s\r\n", target.host);
1138                 if (!USR_HEADER_USER_AGENT)
1139                         SENDFMT(sfp, "User-Agent: %s\r\n", G.user_agent);
1140
1141                 /* Ask server to close the connection as soon as we are done
1142                  * (IOW: we do not intend to send more requests)
1143                  */
1144                 SENDFMT(sfp, "Connection: close\r\n");
1145
1146 #if ENABLE_FEATURE_WGET_AUTHENTICATION
1147                 if (target.user && !USR_HEADER_AUTH) {
1148                         SENDFMT(sfp, "Proxy-Authorization: Basic %s\r\n"+6,
1149                                 base64enc(target.user));
1150                 }
1151                 if (use_proxy && server.user && !USR_HEADER_PROXY_AUTH) {
1152                         SENDFMT(sfp, "Proxy-Authorization: Basic %s\r\n",
1153                                 base64enc(server.user));
1154                 }
1155 #endif
1156
1157                 if (G.beg_range != 0 && !USR_HEADER_RANGE)
1158                         SENDFMT(sfp, "Range: bytes=%"OFF_FMT"u-\r\n", G.beg_range);
1159
1160 #if ENABLE_FEATURE_WGET_LONG_OPTIONS
1161                 if (G.extra_headers) {
1162                         log_io(G.extra_headers);
1163                         fputs(G.extra_headers, sfp);
1164                 }
1165
1166                 if (option_mask32 & WGET_OPT_POST_DATA) {
1167                         SENDFMT(sfp,
1168                                 "Content-Type: application/x-www-form-urlencoded\r\n"
1169                                 "Content-Length: %u\r\n"
1170                                 "\r\n"
1171                                 "%s",
1172                                 (int) strlen(G.post_data), G.post_data
1173                         );
1174                 } else
1175 #endif
1176                 {
1177                         SENDFMT(sfp, "\r\n");
1178                 }
1179
1180                 fflush(sfp);
1181
1182 /* Tried doing this unconditionally.
1183  * Cloudflare and nginx/1.11.5 are shocked to see SHUT_WR on non-HTTPS.
1184  */
1185 #if SSL_SUPPORTED
1186                 if (target.protocol == P_HTTPS) {
1187                         /* If we use SSL helper, keeping our end of the socket open for writing
1188                          * makes our end (i.e. the same fd!) readable (EAGAIN instead of EOF)
1189                          * even after child closes its copy of the fd.
1190                          * This helps:
1191                          */
1192                         shutdown(fileno(sfp), SHUT_WR);
1193                 }
1194 #endif
1195
1196                 /*
1197                  * Retrieve HTTP response line and check for "200" status code.
1198                  */
1199  read_response:
1200                 fgets_trim_sanitize(sfp, "  %s\n");
1201
1202                 str = G.wget_buf;
1203                 str = skip_non_whitespace(str);
1204                 str = skip_whitespace(str);
1205                 // FIXME: no error check
1206                 // xatou wouldn't work: "200 OK"
1207                 status = atoi(str);
1208                 switch (status) {
1209                 case 0:
1210                 case 100:
1211                         while (get_sanitized_hdr(sfp) != NULL)
1212                                 /* eat all remaining headers */;
1213                         goto read_response;
1214
1215                 /* Success responses */
1216                 case 200:
1217                         /* fall through */
1218                 case 201: /* 201 Created */
1219 /* "The request has been fulfilled and resulted in a new resource being created" */
1220                         /* Standard wget is reported to treat this as success */
1221                         /* fall through */
1222                 case 202: /* 202 Accepted */
1223 /* "The request has been accepted for processing, but the processing has not been completed" */
1224                         /* Treat as success: fall through */
1225                 case 203: /* 203 Non-Authoritative Information */
1226 /* "Use of this response code is not required and is only appropriate when the response would otherwise be 200 (OK)" */
1227                         /* fall through */
1228                 case 204: /* 204 No Content */
1229 /*
1230 Response 204 doesn't say "null file", it says "metadata
1231 has changed but data didn't":
1232
1233 "10.2.5 204 No Content
1234 The server has fulfilled the request but does not need to return
1235 an entity-body, and might want to return updated metainformation.
1236 The response MAY include new or updated metainformation in the form
1237 of entity-headers, which if present SHOULD be associated with
1238 the requested variant.
1239
1240 If the client is a user agent, it SHOULD NOT change its document
1241 view from that which caused the request to be sent. This response
1242 is primarily intended to allow input for actions to take place
1243 without causing a change to the user agent's active document view,
1244 although any new or updated metainformation SHOULD be applied
1245 to the document currently in the user agent's active view.
1246
1247 The 204 response MUST NOT include a message-body, and thus
1248 is always terminated by the first empty line after the header fields."
1249
1250 However, in real world it was observed that some web servers
1251 (e.g. Boa/0.94.14rc21) simply use code 204 when file size is zero.
1252 */
1253                         if (G.beg_range != 0) {
1254                                 /* "Range:..." was not honored by the server.
1255                                  * Restart download from the beginning.
1256                                  */
1257                                 reset_beg_range_to_zero();
1258                         }
1259                         break;
1260                 /* 205 Reset Content ?? what to do on this ??   */
1261
1262                 case 300:  /* redirection */
1263                 case 301:
1264                 case 302:
1265                 case 303:
1266                         break;
1267
1268                 case 206: /* Partial Content */
1269                         if (G.beg_range != 0)
1270                                 /* "Range:..." worked. Good. */
1271                                 break;
1272                         /* Partial Content even though we did not ask for it??? */
1273                         /* fall through */
1274                 default:
1275                         bb_error_msg_and_die("server returned error: %s", G.wget_buf);
1276                 }
1277
1278                 /*
1279                  * Retrieve HTTP headers.
1280                  */
1281                 while ((str = get_sanitized_hdr(sfp)) != NULL) {
1282                         static const char keywords[] ALIGN1 =
1283                                 "content-length\0""transfer-encoding\0""location\0";
1284                         enum {
1285                                 KEY_content_length = 1, KEY_transfer_encoding, KEY_location
1286                         };
1287                         smalluint key;
1288
1289                         /* get_sanitized_hdr converted "FOO:" string to lowercase */
1290
1291                         /* strip trailing whitespace */
1292                         char *s = strchrnul(str, '\0') - 1;
1293                         while (s >= str && (*s == ' ' || *s == '\t')) {
1294                                 *s = '\0';
1295                                 s--;
1296                         }
1297                         key = index_in_strings(keywords, G.wget_buf) + 1;
1298                         if (key == KEY_content_length) {
1299                                 G.content_len = BB_STRTOOFF(str, NULL, 10);
1300                                 if (G.content_len < 0 || errno) {
1301                                         bb_error_msg_and_die("content-length %s is garbage", str);
1302                                 }
1303                                 G.got_clen = 1;
1304                                 continue;
1305                         }
1306                         if (key == KEY_transfer_encoding) {
1307                                 if (strcmp(str_tolower(str), "chunked") != 0)
1308                                         bb_error_msg_and_die("transfer encoding '%s' is not supported", str);
1309                                 G.chunked = 1;
1310                         }
1311                         if (key == KEY_location && status >= 300) {
1312                                 if (--redir_limit == 0)
1313                                         bb_error_msg_and_die("too many redirections");
1314                                 fclose(sfp);
1315                                 if (str[0] == '/') {
1316                                         free(redirected_path);
1317                                         target.path = redirected_path = xstrdup(str + 1);
1318                                         /* lsa stays the same: it's on the same server */
1319                                 } else {
1320                                         parse_url(str, &target);
1321                                         if (!use_proxy) {
1322                                                 /* server.user remains untouched */
1323                                                 free(server.allocated);
1324                                                 server.allocated = NULL;
1325                                                 server.host = target.host;
1326                                                 /* strip_ipv6_scope_id(target.host); - no! */
1327                                                 /* we assume remote never gives us IPv6 addr with scope id */
1328                                                 server.port = target.port;
1329                                                 free(lsa);
1330                                                 goto resolve_lsa;
1331                                         } /* else: lsa stays the same: we use proxy */
1332                                 }
1333                                 goto establish_session;
1334                         }
1335                 }
1336 //              if (status >= 300)
1337 //                      bb_error_msg_and_die("bad redirection (no Location: header from server)");
1338
1339                 /* For HTTP, data is pumped over the same connection */
1340                 dfp = sfp;
1341         } else {
1342                 /*
1343                  *  FTP session
1344                  */
1345                 sfp = prepare_ftp_session(&dfp, &target, lsa);
1346         }
1347
1348         free(lsa);
1349
1350         if (!(option_mask32 & WGET_OPT_SPIDER)) {
1351                 if (G.output_fd < 0)
1352                         G.output_fd = xopen(G.fname_out, G.o_flags);
1353                 retrieve_file_data(dfp);
1354                 if (!(option_mask32 & WGET_OPT_OUTNAME)) {
1355                         xclose(G.output_fd);
1356                         G.output_fd = -1;
1357                 }
1358         }
1359
1360         if (dfp != sfp) {
1361                 /* It's ftp. Close data connection properly */
1362                 fclose(dfp);
1363                 if (ftpcmd(NULL, NULL, sfp) != 226)
1364                         bb_error_msg_and_die("ftp error: %s", G.wget_buf);
1365                 /* ftpcmd("QUIT", NULL, sfp); - why bother? */
1366         }
1367         fclose(sfp);
1368
1369         free(server.allocated);
1370         free(target.allocated);
1371         free(server.user);
1372         free(target.user);
1373         free(fname_out_alloc);
1374         free(redirected_path);
1375 }
1376
1377 int wget_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
1378 int wget_main(int argc UNUSED_PARAM, char **argv)
1379 {
1380 #if ENABLE_FEATURE_WGET_LONG_OPTIONS
1381         static const char wget_longopts[] ALIGN1 =
1382                 /* name, has_arg, val */
1383                 "continue\0"         No_argument       "c"
1384                 "quiet\0"            No_argument       "q"
1385                 "server-response\0"  No_argument       "S"
1386                 "output-document\0"  Required_argument "O"
1387                 "directory-prefix\0" Required_argument "P"
1388                 "proxy\0"            Required_argument "Y"
1389                 "user-agent\0"       Required_argument "U"
1390 IF_FEATURE_WGET_TIMEOUT(
1391                 "timeout\0"          Required_argument "T")
1392                 /* Ignored: */
1393 IF_DESKTOP(     "tries\0"            Required_argument "t")
1394                 "header\0"           Required_argument "\xff"
1395                 "post-data\0"        Required_argument "\xfe"
1396                 "spider\0"           No_argument       "\xfd"
1397                 /* Ignored (we always use PASV): */
1398 IF_DESKTOP(     "passive-ftp\0"      No_argument       "\xf0")
1399                 /* Ignored (we don't do ssl) */
1400 IF_DESKTOP(     "no-check-certificate\0" No_argument   "\xf0")
1401                 /* Ignored (we don't support caching) */
1402 IF_DESKTOP(     "no-cache\0"         No_argument       "\xf0")
1403 IF_DESKTOP(     "no-verbose\0"       No_argument       "\xf0")
1404 IF_DESKTOP(     "no-clobber\0"       No_argument       "\xf0")
1405 IF_DESKTOP(     "no-host-directories\0" No_argument    "\xf0")
1406 IF_DESKTOP(     "no-parent\0"        No_argument       "\xf0")
1407                 ;
1408 # define GETOPT32 getopt32long
1409 # define LONGOPTS ,wget_longopts
1410 #else
1411 # define GETOPT32 getopt32
1412 # define LONGOPTS
1413 #endif
1414
1415 #if ENABLE_FEATURE_WGET_LONG_OPTIONS
1416         llist_t *headers_llist = NULL;
1417 #endif
1418
1419         INIT_G();
1420
1421 #if ENABLE_FEATURE_WGET_TIMEOUT
1422         G.timeout_seconds = 900;
1423         signal(SIGALRM, alarm_handler);
1424 #endif
1425         G.proxy_flag = "on";   /* use proxies if env vars are set */
1426         G.user_agent = "Wget"; /* "User-Agent" header field */
1427
1428 #if ENABLE_FEATURE_WGET_LONG_OPTIONS
1429 #endif
1430         GETOPT32(argv, "^"
1431                 "cqSO:P:Y:U:T:+"
1432                 /*ignored:*/ "t:"
1433                 /*ignored:*/ "n::"
1434                 /* wget has exactly four -n<letter> opts, all of which we can ignore:
1435                  * -nv --no-verbose: be moderately quiet (-q is full quiet)
1436                  * -nc --no-clobber: abort if exists, neither download to FILE.n nor overwrite FILE
1437                  * -nH --no-host-directories: wget -r http://host/ won't create host/
1438                  * -np --no-parent
1439                  * "n::" above says that we accept -n[ARG].
1440                  * Specifying "n:" would be a bug: "-n ARG" would eat ARG!
1441                  */
1442                 "\0"
1443                 "-1" /* at least one URL */
1444                 IF_FEATURE_WGET_LONG_OPTIONS(":\xff::") /* --header is a list */
1445                 LONGOPTS
1446                 , &G.fname_out, &G.dir_prefix,
1447                 &G.proxy_flag, &G.user_agent,
1448                 IF_FEATURE_WGET_TIMEOUT(&G.timeout_seconds) IF_NOT_FEATURE_WGET_TIMEOUT(NULL),
1449                 NULL, /* -t RETRIES */
1450                 NULL  /* -n[ARG] */
1451                 IF_FEATURE_WGET_LONG_OPTIONS(, &headers_llist)
1452                 IF_FEATURE_WGET_LONG_OPTIONS(, &G.post_data)
1453         );
1454 #if 0 /* option bits debug */
1455         if (option_mask32 & WGET_OPT_RETRIES) bb_error_msg("-t NUM");
1456         if (option_mask32 & WGET_OPT_nsomething) bb_error_msg("-nsomething");
1457         if (option_mask32 & WGET_OPT_HEADER) bb_error_msg("--header");
1458         if (option_mask32 & WGET_OPT_POST_DATA) bb_error_msg("--post-data");
1459         if (option_mask32 & WGET_OPT_SPIDER) bb_error_msg("--spider");
1460         exit(0);
1461 #endif
1462         argv += optind;
1463
1464 #if ENABLE_FEATURE_WGET_LONG_OPTIONS
1465         if (headers_llist) {
1466                 int size = 0;
1467                 char *hdr;
1468                 llist_t *ll = headers_llist;
1469                 while (ll) {
1470                         size += strlen(ll->data) + 2;
1471                         ll = ll->link;
1472                 }
1473                 G.extra_headers = hdr = xmalloc(size + 1);
1474                 while (headers_llist) {
1475                         int bit;
1476                         const char *words;
1477
1478                         size = sprintf(hdr, "%s\r\n",
1479                                         (char*)llist_pop(&headers_llist));
1480                         /* a bit like index_in_substrings but don't match full key */
1481                         bit = 1;
1482                         words = wget_user_headers;
1483                         while (*words) {
1484                                 if (strstr(hdr, words) == hdr) {
1485                                         G.user_headers |= bit;
1486                                         break;
1487                                 }
1488                                 bit <<= 1;
1489                                 words += strlen(words) + 1;
1490                         }
1491                         hdr += size;
1492                 }
1493         }
1494 #endif
1495
1496         G.output_fd = -1;
1497         G.o_flags = O_WRONLY | O_CREAT | O_TRUNC | O_EXCL;
1498         if (G.fname_out) { /* -O FILE ? */
1499                 if (LONE_DASH(G.fname_out)) { /* -O - ? */
1500                         G.output_fd = 1;
1501                         option_mask32 &= ~WGET_OPT_CONTINUE;
1502                 }
1503                 /* compat with wget: -O FILE can overwrite */
1504                 G.o_flags = O_WRONLY | O_CREAT | O_TRUNC;
1505         }
1506
1507         while (*argv)
1508                 download_one_url(*argv++);
1509
1510         if (G.output_fd >= 0)
1511                 xclose(G.output_fd);
1512
1513 #if ENABLE_FEATURE_CLEAN_UP && ENABLE_FEATURE_WGET_LONG_OPTIONS
1514         free(G.extra_headers);
1515 #endif
1516         FINI_G();
1517
1518         return EXIT_SUCCESS;
1519 }