blob: df6d36358f69d833288aa8523938215c1878b543 [file] [log] [blame]
Eric Andersen96700832000-09-04 15:15:55 +00001/* vi: set sw=4 ts=4: */
2/*
Eric Andersen79757c92001-04-05 21:45:54 +00003 * wget - retrieve a file using HTTP or FTP
Eric Andersen96700832000-09-04 15:15:55 +00004 *
Eric Andersen4e573f42000-11-14 23:29:24 +00005 * Chip Rosenthal Covad Communications <chip@laserlink.net>
Denys Vlasenko0ef64bd2010-08-16 20:14:46 +02006 * Licensed under GPLv2, see file LICENSE in this source tree.
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +02007 *
8 * Copyright (C) 2010 Bradley M. Kuhn <bkuhn@ebb.org>
Denys Vlasenkofb132e42010-10-29 11:46:52 +02009 * Kuhn's copyrights are licensed GPLv2-or-later. File as a whole remains GPLv2.
Eric Andersen96700832000-09-04 15:15:55 +000010 */
Denys Vlasenkoe2e55b02011-03-21 00:37:05 +010011
12//usage:#define wget_trivial_usage
13//usage: IF_FEATURE_WGET_LONG_OPTIONS(
14//usage: "[-c|--continue] [-s|--spider] [-q|--quiet] [-O|--output-document FILE]\n"
15//usage: " [--header 'header: value'] [-Y|--proxy on/off] [-P DIR]\n"
16//usage: " [--no-check-certificate] [-U|--user-agent AGENT]"
17//usage: IF_FEATURE_WGET_TIMEOUT(" [-T SEC]") " URL..."
18//usage: )
19//usage: IF_NOT_FEATURE_WGET_LONG_OPTIONS(
20//usage: "[-csq] [-O FILE] [-Y on/off] [-P DIR] [-U AGENT]"
21//usage: IF_FEATURE_WGET_TIMEOUT(" [-T SEC]") " URL..."
22//usage: )
23//usage:#define wget_full_usage "\n\n"
24//usage: "Retrieve files via HTTP or FTP\n"
25//usage: "\nOptions:"
26//usage: "\n -s Spider mode - only check file existence"
27//usage: "\n -c Continue retrieval of aborted transfer"
28//usage: "\n -q Quiet"
29//usage: "\n -P DIR Save to DIR (default .)"
30//usage: IF_FEATURE_WGET_TIMEOUT(
31//usage: "\n -T SEC Network read timeout is SEC seconds"
32//usage: )
33//usage: "\n -O FILE Save to FILE ('-' for stdout)"
34//usage: "\n -U STR Use STR for User-Agent header"
35//usage: "\n -Y Use proxy ('on' or 'off')"
36
Denis Vlasenkob6adbf12007-05-26 19:00:18 +000037#include "libbb.h"
Denis Vlasenkoa552eeb2006-09-26 09:22:12 +000038
Denys Vlasenkof836f012011-02-10 23:02:28 +010039//#define log_io(...) bb_error_msg(__VA_ARGS__)
40#define log_io(...) ((void)0)
41
42
Eric Andersen79757c92001-04-05 21:45:54 +000043struct host_info {
Denys Vlasenkoa3661092011-02-13 02:33:11 +010044 char *allocated;
Denis Vlasenko818322b2007-09-24 18:27:04 +000045 const char *path;
46 const char *user;
47 char *host;
48 int port;
49 smallint is_ftp;
Eric Andersen79757c92001-04-05 21:45:54 +000050};
51
Denis Vlasenko77105632007-09-24 15:04:00 +000052
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +020053/* Globals */
Denis Vlasenko77105632007-09-24 15:04:00 +000054struct globals {
55 off_t content_len; /* Content-length of the file */
56 off_t beg_range; /* Range at which continue begins */
57#if ENABLE_FEATURE_WGET_STATUSBAR
Denis Vlasenko77105632007-09-24 15:04:00 +000058 off_t transferred; /* Number of bytes transferred so far */
59 const char *curfile; /* Name of current file being transferred */
Magnus Dammf5914992009-11-08 16:34:43 +010060 bb_progress_t pmt;
Denis Vlasenko77105632007-09-24 15:04:00 +000061#endif
Denys Vlasenkoa3661092011-02-13 02:33:11 +010062 char *dir_prefix;
63#if ENABLE_FEATURE_WGET_LONG_OPTIONS
64 char *post_data;
65 char *extra_headers;
66#endif
67 char *fname_out; /* where to direct output (-O) */
68 const char *proxy_flag; /* Use proxies if env vars are set */
69 const char *user_agent; /* "User-Agent" header field */
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +020070#if ENABLE_FEATURE_WGET_TIMEOUT
71 unsigned timeout_seconds;
72#endif
Denys Vlasenko2384a352011-02-15 00:58:36 +010073 int output_fd;
74 int o_flags;
Denys Vlasenko7f432802009-06-28 01:02:24 +020075 smallint chunked; /* chunked transfer encoding */
76 smallint got_clen; /* got content-length: from server */
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +010077 /* Local downloads do benefit from big buffer.
78 * With 512 byte buffer, it was measured to be
79 * an order of magnitude slower than with big one.
80 */
81 uint64_t just_to_align_next_member;
82 char wget_buf[CONFIG_FEATURE_COPYBUF_KB*1024];
Denys Vlasenko98a4c7c2010-02-04 15:00:15 +010083} FIX_ALIASING;
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +010084#define G (*ptr_to_globals)
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +020085#define INIT_G() do { \
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +010086 SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +020087 IF_FEATURE_WGET_TIMEOUT(G.timeout_seconds = 900;) \
88} while (0)
Denis Vlasenko77105632007-09-24 15:04:00 +000089
90
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +020091/* Must match option string! */
92enum {
93 WGET_OPT_CONTINUE = (1 << 0),
Denys Vlasenkofb132e42010-10-29 11:46:52 +020094 WGET_OPT_SPIDER = (1 << 1),
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +020095 WGET_OPT_QUIET = (1 << 2),
96 WGET_OPT_OUTNAME = (1 << 3),
97 WGET_OPT_PREFIX = (1 << 4),
98 WGET_OPT_PROXY = (1 << 5),
99 WGET_OPT_USER_AGENT = (1 << 6),
100 WGET_OPT_NETWORK_READ_TIMEOUT = (1 << 7),
101 WGET_OPT_RETRIES = (1 << 8),
102 WGET_OPT_PASSIVE = (1 << 9),
103 WGET_OPT_HEADER = (1 << 10) * ENABLE_FEATURE_WGET_LONG_OPTIONS,
104 WGET_OPT_POST_DATA = (1 << 11) * ENABLE_FEATURE_WGET_LONG_OPTIONS,
105};
106
107enum {
108 PROGRESS_START = -1,
109 PROGRESS_END = 0,
110 PROGRESS_BUMP = 1,
111};
Denis Vlasenko9cade082006-11-21 10:43:02 +0000112#if ENABLE_FEATURE_WGET_STATUSBAR
Denis Vlasenko00d84172008-11-24 07:34:42 +0000113static void progress_meter(int flag)
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000114{
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +0200115 if (option_mask32 & WGET_OPT_QUIET)
116 return;
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000117
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +0200118 if (flag == PROGRESS_START)
Denys Vlasenkod55e1392011-02-11 18:56:13 +0100119 bb_progress_init(&G.pmt, G.curfile);
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000120
Denys Vlasenko2384a352011-02-15 00:58:36 +0100121 bb_progress_update(&G.pmt,
122 G.beg_range,
123 G.transferred,
124 (G.chunked || !G.got_clen) ? 0 : G.beg_range + G.transferred + G.content_len
125 );
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000126
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +0200127 if (flag == PROGRESS_END) {
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100128 bb_progress_free(&G.pmt);
Denys Vlasenko19ced5c2010-06-06 21:53:09 +0200129 bb_putchar_stderr('\n');
Denys Vlasenkoa3aa3e32009-12-11 12:36:10 +0100130 G.transferred = 0;
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000131 }
132}
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +0200133#else
Denis Vlasenko00d84172008-11-24 07:34:42 +0000134static ALWAYS_INLINE void progress_meter(int flag UNUSED_PARAM) { }
Eric Andersenb520e082000-10-03 00:21:45 +0000135#endif
Eric Andersenc7bda1c2004-03-15 08:29:22 +0000136
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000137
Denys Vlasenko7d5ddf12009-06-30 20:36:27 +0200138/* IPv6 knows scoped address types i.e. link and site local addresses. Link
139 * local addresses can have a scope identifier to specify the
140 * interface/link an address is valid on (e.g. fe80::1%eth0). This scope
141 * identifier is only valid on a single node.
142 *
143 * RFC 4007 says that the scope identifier MUST NOT be sent across the wire,
144 * unless all nodes agree on the semantic. Apache e.g. regards zone identifiers
145 * in the Host header as invalid requests, see
146 * https://issues.apache.org/bugzilla/show_bug.cgi?id=35122
147 */
148static void strip_ipv6_scope_id(char *host)
149{
150 char *scope, *cp;
151
152 /* bbox wget actually handles IPv6 addresses without [], like
153 * wget "http://::1/xxx", but this is not standard.
154 * To save code, _here_ we do not support it. */
155
156 if (host[0] != '[')
157 return; /* not IPv6 */
158
159 scope = strchr(host, '%');
160 if (!scope)
161 return;
162
163 /* Remove the IPv6 zone identifier from the host address */
164 cp = strchr(host, ']');
165 if (!cp || (cp[1] != ':' && cp[1] != '\0')) {
166 /* malformed address (not "[xx]:nn" or "[xx]") */
167 return;
168 }
169
170 /* cp points to "]...", scope points to "%eth0]..." */
171 overlapping_strcpy(scope, cp);
172}
173
Denis Vlasenko9cade082006-11-21 10:43:02 +0000174#if ENABLE_FEATURE_WGET_AUTHENTICATION
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100175/* Base64-encode character string. */
176static char *base64enc(const char *str)
Denis Vlasenko3526a132006-09-09 12:20:57 +0000177{
Denis Vlasenko12d21292007-06-27 21:40:07 +0000178 unsigned len = strlen(str);
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100179 if (len > sizeof(G.wget_buf)/4*3 - 10) /* paranoia */
180 len = sizeof(G.wget_buf)/4*3 - 10;
181 bb_uuencode(G.wget_buf, str, len, bb_uuenc_tbl_base64);
182 return G.wget_buf;
Eric Andersen79757c92001-04-05 21:45:54 +0000183}
184#endif
185
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200186static char* sanitize_string(char *s)
187{
188 unsigned char *p = (void *) s;
189 while (*p >= ' ')
190 p++;
191 *p = '\0';
192 return s;
193}
194
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000195static FILE *open_socket(len_and_sockaddr *lsa)
196{
197 FILE *fp;
198
199 /* glibc 2.4 seems to try seeking on it - ??! */
200 /* hopefully it understands what ESPIPE means... */
201 fp = fdopen(xconnect_stream(lsa), "r+");
202 if (fp == NULL)
Denys Vlasenkodee0fc92011-02-10 10:01:49 +0100203 bb_perror_msg_and_die(bb_msg_memory_exhausted);
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000204
205 return fp;
206}
207
Denys Vlasenkof836f012011-02-10 23:02:28 +0100208/* Returns '\n' if it was seen, else '\0'. Trims at first '\r' or '\n' */
209static char fgets_and_trim(FILE *fp)
210{
211 char c;
212 char *buf_ptr;
213
214 if (fgets(G.wget_buf, sizeof(G.wget_buf) - 1, fp) == NULL)
215 bb_perror_msg_and_die("error getting response");
216
217 buf_ptr = strchrnul(G.wget_buf, '\n');
218 c = *buf_ptr;
219 *buf_ptr = '\0';
220 buf_ptr = strchrnul(G.wget_buf, '\r');
221 *buf_ptr = '\0';
222
223 log_io("< %s", G.wget_buf);
224
225 return c;
226}
227
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100228static int ftpcmd(const char *s1, const char *s2, FILE *fp)
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000229{
230 int result;
231 if (s1) {
Denys Vlasenkof836f012011-02-10 23:02:28 +0100232 if (!s2)
233 s2 = "";
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000234 fprintf(fp, "%s%s\r\n", s1, s2);
235 fflush(fp);
Denys Vlasenkof836f012011-02-10 23:02:28 +0100236 log_io("> %s%s", s1, s2);
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000237 }
238
239 do {
Denys Vlasenkof836f012011-02-10 23:02:28 +0100240 fgets_and_trim(fp);
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100241 } while (!isdigit(G.wget_buf[0]) || G.wget_buf[3] != ' ');
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000242
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100243 G.wget_buf[3] = '\0';
244 result = xatoi_positive(G.wget_buf);
245 G.wget_buf[3] = ' ';
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000246 return result;
247}
248
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100249static void parse_url(const char *src_url, struct host_info *h)
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000250{
251 char *url, *p, *sp;
252
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100253 free(h->allocated);
254 h->allocated = url = xstrdup(src_url);
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000255
256 if (strncmp(url, "http://", 7) == 0) {
257 h->port = bb_lookup_port("http", "tcp", 80);
258 h->host = url + 7;
259 h->is_ftp = 0;
260 } else if (strncmp(url, "ftp://", 6) == 0) {
261 h->port = bb_lookup_port("ftp", "tcp", 21);
262 h->host = url + 6;
263 h->is_ftp = 1;
264 } else
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200265 bb_error_msg_and_die("not an http or ftp url: %s", sanitize_string(url));
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000266
267 // FYI:
268 // "Real" wget 'http://busybox.net?var=a/b' sends this request:
269 // 'GET /?var=a/b HTTP 1.0'
270 // and saves 'index.html?var=a%2Fb' (we save 'b')
271 // wget 'http://busybox.net?login=john@doe':
272 // request: 'GET /?login=john@doe HTTP/1.0'
273 // saves: 'index.html?login=john@doe' (we save '?login=john@doe')
274 // wget 'http://busybox.net#test/test':
275 // request: 'GET / HTTP/1.0'
276 // saves: 'index.html' (we save 'test')
277 //
278 // We also don't add unique .N suffix if file exists...
279 sp = strchr(h->host, '/');
280 p = strchr(h->host, '?'); if (!sp || (p && sp > p)) sp = p;
281 p = strchr(h->host, '#'); if (!sp || (p && sp > p)) sp = p;
282 if (!sp) {
Denis Vlasenko818322b2007-09-24 18:27:04 +0000283 h->path = "";
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000284 } else if (*sp == '/') {
285 *sp = '\0';
286 h->path = sp + 1;
287 } else { // '#' or '?'
288 // http://busybox.net?login=john@doe is a valid URL
289 // memmove converts to:
290 // http:/busybox.nett?login=john@doe...
Denis Vlasenko818322b2007-09-24 18:27:04 +0000291 memmove(h->host - 1, h->host, sp - h->host);
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000292 h->host--;
293 sp[-1] = '\0';
294 h->path = sp;
295 }
296
Vladimir Dronnikovbe168b12009-10-05 02:18:01 +0200297 // We used to set h->user to NULL here, but this interferes
298 // with handling of code 302 ("object was moved")
299
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000300 sp = strrchr(h->host, '@');
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000301 if (sp != NULL) {
302 h->user = h->host;
303 *sp = '\0';
304 h->host = sp + 1;
305 }
306
307 sp = h->host;
308}
309
Denys Vlasenkof836f012011-02-10 23:02:28 +0100310static char *gethdr(FILE *fp)
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000311{
312 char *s, *hdrval;
313 int c;
314
315 /* *istrunc = 0; */
316
317 /* retrieve header line */
Denys Vlasenkof836f012011-02-10 23:02:28 +0100318 c = fgets_and_trim(fp);
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000319
Denys Vlasenkof836f012011-02-10 23:02:28 +0100320 /* end of the headers? */
321 if (G.wget_buf[0] == '\0')
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000322 return NULL;
323
324 /* convert the header name to lower case */
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100325 for (s = G.wget_buf; isalnum(*s) || *s == '-' || *s == '.'; ++s) {
Denys Vlasenko48363312010-04-04 15:29:32 +0200326 /* tolower for "A-Z", no-op for "0-9a-z-." */
Denys Vlasenkof836f012011-02-10 23:02:28 +0100327 *s |= 0x20;
Denys Vlasenko48363312010-04-04 15:29:32 +0200328 }
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000329
330 /* verify we are at the end of the header name */
331 if (*s != ':')
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100332 bb_error_msg_and_die("bad header line: %s", sanitize_string(G.wget_buf));
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000333
334 /* locate the start of the header value */
335 *s++ = '\0';
336 hdrval = skip_whitespace(s);
337
Denys Vlasenkof836f012011-02-10 23:02:28 +0100338 if (c != '\n') {
339 /* Rats! The buffer isn't big enough to hold the entire header value */
340 while (c = getc(fp), c != EOF && c != '\n')
341 continue;
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000342 }
343
Denis Vlasenko47ddd012007-09-24 18:24:17 +0000344 return hdrval;
345}
346
Denys Vlasenko7f432802009-06-28 01:02:24 +0200347static FILE* prepare_ftp_session(FILE **dfpp, struct host_info *target, len_and_sockaddr *lsa)
348{
Denys Vlasenko7f432802009-06-28 01:02:24 +0200349 FILE *sfp;
350 char *str;
351 int port;
352
353 if (!target->user)
354 target->user = xstrdup("anonymous:busybox@");
355
356 sfp = open_socket(lsa);
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100357 if (ftpcmd(NULL, NULL, sfp) != 220)
358 bb_error_msg_and_die("%s", sanitize_string(G.wget_buf + 4));
Denys Vlasenko7f432802009-06-28 01:02:24 +0200359
360 /*
361 * Splitting username:password pair,
362 * trying to log in
363 */
364 str = strchr(target->user, ':');
365 if (str)
366 *str++ = '\0';
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100367 switch (ftpcmd("USER ", target->user, sfp)) {
Denys Vlasenko7f432802009-06-28 01:02:24 +0200368 case 230:
369 break;
370 case 331:
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100371 if (ftpcmd("PASS ", str, sfp) == 230)
Denys Vlasenko7f432802009-06-28 01:02:24 +0200372 break;
373 /* fall through (failed login) */
374 default:
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100375 bb_error_msg_and_die("ftp login: %s", sanitize_string(G.wget_buf + 4));
Denys Vlasenko7f432802009-06-28 01:02:24 +0200376 }
377
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100378 ftpcmd("TYPE I", NULL, sfp);
Denys Vlasenko7f432802009-06-28 01:02:24 +0200379
380 /*
381 * Querying file size
382 */
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100383 if (ftpcmd("SIZE ", target->path, sfp) == 213) {
384 G.content_len = BB_STRTOOFF(G.wget_buf + 4, NULL, 10);
Denys Vlasenkoa3aa3e32009-12-11 12:36:10 +0100385 if (G.content_len < 0 || errno) {
Denys Vlasenko7f432802009-06-28 01:02:24 +0200386 bb_error_msg_and_die("SIZE value is garbage");
387 }
388 G.got_clen = 1;
389 }
390
391 /*
392 * Entering passive mode
393 */
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100394 if (ftpcmd("PASV", NULL, sfp) != 227) {
Denys Vlasenko7f432802009-06-28 01:02:24 +0200395 pasv_error:
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100396 bb_error_msg_and_die("bad response to %s: %s", "PASV", sanitize_string(G.wget_buf));
Denys Vlasenko7f432802009-06-28 01:02:24 +0200397 }
398 // Response is "227 garbageN1,N2,N3,N4,P1,P2[)garbage]
399 // Server's IP is N1.N2.N3.N4 (we ignore it)
400 // Server's port for data connection is P1*256+P2
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100401 str = strrchr(G.wget_buf, ')');
Denys Vlasenko7f432802009-06-28 01:02:24 +0200402 if (str) str[0] = '\0';
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100403 str = strrchr(G.wget_buf, ',');
Denys Vlasenko7f432802009-06-28 01:02:24 +0200404 if (!str) goto pasv_error;
405 port = xatou_range(str+1, 0, 255);
406 *str = '\0';
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100407 str = strrchr(G.wget_buf, ',');
Denys Vlasenko7f432802009-06-28 01:02:24 +0200408 if (!str) goto pasv_error;
409 port += xatou_range(str+1, 0, 255) * 256;
Denys Vlasenkoca183112011-04-07 17:52:20 +0200410 set_nport(&lsa->u.sa, htons(port));
Denys Vlasenko7f432802009-06-28 01:02:24 +0200411
412 *dfpp = open_socket(lsa);
413
Denys Vlasenkoa3aa3e32009-12-11 12:36:10 +0100414 if (G.beg_range) {
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100415 sprintf(G.wget_buf, "REST %"OFF_FMT"u", G.beg_range);
416 if (ftpcmd(G.wget_buf, NULL, sfp) == 350)
Denys Vlasenkoa3aa3e32009-12-11 12:36:10 +0100417 G.content_len -= G.beg_range;
Denys Vlasenko7f432802009-06-28 01:02:24 +0200418 }
419
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100420 if (ftpcmd("RETR ", target->path, sfp) > 150)
421 bb_error_msg_and_die("bad response to %s: %s", "RETR", sanitize_string(G.wget_buf));
Denys Vlasenko7f432802009-06-28 01:02:24 +0200422
423 return sfp;
424}
425
Denys Vlasenko2384a352011-02-15 00:58:36 +0100426static void NOINLINE retrieve_file_data(FILE *dfp)
Denys Vlasenko7f432802009-06-28 01:02:24 +0200427{
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +0200428#if ENABLE_FEATURE_WGET_STATUSBAR || ENABLE_FEATURE_WGET_TIMEOUT
429# if ENABLE_FEATURE_WGET_TIMEOUT
430 unsigned second_cnt;
431# endif
432 struct pollfd polldata;
Denys Vlasenko7f432802009-06-28 01:02:24 +0200433
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +0200434 polldata.fd = fileno(dfp);
435 polldata.events = POLLIN | POLLPRI;
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +0200436#endif
437 progress_meter(PROGRESS_START);
Denys Vlasenko7f432802009-06-28 01:02:24 +0200438
439 if (G.chunked)
440 goto get_clen;
441
442 /* Loops only if chunked */
443 while (1) {
Denys Vlasenkoc60f4462011-02-11 22:23:23 +0100444
445#if ENABLE_FEATURE_WGET_STATUSBAR || ENABLE_FEATURE_WGET_TIMEOUT
446 /* Must use nonblocking I/O, otherwise fread will loop
447 * and *block* until it reads full buffer,
448 * which messes up progress bar and/or timeout logic.
449 * Because of nonblocking I/O, we need to dance
450 * very carefully around EAGAIN. See explanation at
451 * clearerr() call.
452 */
453 ndelay_on(polldata.fd);
454#endif
Denys Vlasenkoa3aa3e32009-12-11 12:36:10 +0100455 while (1) {
Denys Vlasenko7f432802009-06-28 01:02:24 +0200456 int n;
Denys Vlasenkoa3aa3e32009-12-11 12:36:10 +0100457 unsigned rdsz;
Denys Vlasenko7f432802009-06-28 01:02:24 +0200458
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100459 rdsz = sizeof(G.wget_buf);
Denys Vlasenkoa3aa3e32009-12-11 12:36:10 +0100460 if (G.got_clen) {
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100461 if (G.content_len < (off_t)sizeof(G.wget_buf)) {
Denys Vlasenkoa3aa3e32009-12-11 12:36:10 +0100462 if ((int)G.content_len <= 0)
463 break;
464 rdsz = (unsigned)G.content_len;
465 }
466 }
Denys Vlasenko8766a792011-02-11 21:42:00 +0100467
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +0200468#if ENABLE_FEATURE_WGET_STATUSBAR || ENABLE_FEATURE_WGET_TIMEOUT
469# if ENABLE_FEATURE_WGET_TIMEOUT
470 second_cnt = G.timeout_seconds;
471# endif
472 while (1) {
473 if (safe_poll(&polldata, 1, 1000) != 0)
474 break; /* error, EOF, or data is available */
475# if ENABLE_FEATURE_WGET_TIMEOUT
476 if (second_cnt != 0 && --second_cnt == 0) {
477 progress_meter(PROGRESS_END);
Denys Vlasenko8766a792011-02-11 21:42:00 +0100478 bb_error_msg_and_die("download timed out");
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +0200479 }
480# endif
481 /* Needed for "stalled" indicator */
482 progress_meter(PROGRESS_BUMP);
483 }
Denys Vlasenkof9af3752011-02-11 22:01:33 +0100484
Denys Vlasenko8766a792011-02-11 21:42:00 +0100485 /* fread internally uses read loop, which in our case
486 * is usually exited when we get EAGAIN.
487 * In this case, libc sets error marker on the stream.
488 * Need to clear it before next fread to avoid possible
489 * rare false positive ferror below. Rare because usually
490 * fread gets more than zero bytes, and we don't fall
491 * into if (n <= 0) ...
492 */
493 clearerr(dfp);
494 errno = 0;
Denys Vlasenkof9af3752011-02-11 22:01:33 +0100495#endif
Denys Vlasenko0fac2f72011-02-10 09:55:05 +0100496 n = fread(G.wget_buf, 1, rdsz, dfp);
Denys Vlasenko8766a792011-02-11 21:42:00 +0100497 /* man fread:
498 * If error occurs, or EOF is reached, the return value
499 * is a short item count (or zero).
500 * fread does not distinguish between EOF and error.
501 */
Denys Vlasenko7f432802009-06-28 01:02:24 +0200502 if (n <= 0) {
Denys Vlasenko8766a792011-02-11 21:42:00 +0100503#if ENABLE_FEATURE_WGET_STATUSBAR || ENABLE_FEATURE_WGET_TIMEOUT
504 if (errno == EAGAIN) /* poll lied, there is no data? */
505 continue; /* yes */
506#endif
507 if (ferror(dfp))
508 bb_perror_msg_and_die(bb_msg_read_error);
509 break; /* EOF, not error */
Denys Vlasenko7f432802009-06-28 01:02:24 +0200510 }
Denys Vlasenko8766a792011-02-11 21:42:00 +0100511
Denys Vlasenko2384a352011-02-15 00:58:36 +0100512 xwrite(G.output_fd, G.wget_buf, n);
Denys Vlasenko8766a792011-02-11 21:42:00 +0100513
Denys Vlasenko7f432802009-06-28 01:02:24 +0200514#if ENABLE_FEATURE_WGET_STATUSBAR
Denys Vlasenkoa3aa3e32009-12-11 12:36:10 +0100515 G.transferred += n;
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +0200516 progress_meter(PROGRESS_BUMP);
Denys Vlasenko7f432802009-06-28 01:02:24 +0200517#endif
Denys Vlasenko9213a552011-02-10 13:23:45 +0100518 if (G.got_clen) {
Denys Vlasenkoa3aa3e32009-12-11 12:36:10 +0100519 G.content_len -= n;
Denys Vlasenko9213a552011-02-10 13:23:45 +0100520 if (G.content_len == 0)
521 break;
522 }
Denys Vlasenko7f432802009-06-28 01:02:24 +0200523 }
Denys Vlasenkoc60f4462011-02-11 22:23:23 +0100524#if ENABLE_FEATURE_WGET_STATUSBAR || ENABLE_FEATURE_WGET_TIMEOUT
525 clearerr(dfp);
Denys Vlasenko88ad9da2011-02-11 23:06:21 +0100526 ndelay_off(polldata.fd); /* else fgets can get very unhappy */
Denys Vlasenkoc60f4462011-02-11 22:23:23 +0100527#endif
Denys Vlasenko7f432802009-06-28 01:02:24 +0200528 if (!G.chunked)
529 break;
530
Denys Vlasenkoc60f4462011-02-11 22:23:23 +0100531 fgets_and_trim(dfp); /* Eat empty line */
Denys Vlasenko7f432802009-06-28 01:02:24 +0200532 get_clen:
Denys Vlasenkof836f012011-02-10 23:02:28 +0100533 fgets_and_trim(dfp);
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100534 G.content_len = STRTOOFF(G.wget_buf, NULL, 16);
Denys Vlasenko7f432802009-06-28 01:02:24 +0200535 /* FIXME: error check? */
Denys Vlasenkoa3aa3e32009-12-11 12:36:10 +0100536 if (G.content_len == 0)
Denys Vlasenko7f432802009-06-28 01:02:24 +0200537 break; /* all done! */
Denys Vlasenkoa3aa3e32009-12-11 12:36:10 +0100538 G.got_clen = 1;
Denys Vlasenko7f432802009-06-28 01:02:24 +0200539 }
540
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100541 /* Draw full bar and free its resources */
Denys Vlasenko2384a352011-02-15 00:58:36 +0100542 G.chunked = 0; /* makes it show 100% even for chunked download */
543 G.got_clen = 1; /* makes it show 100% even for download of (formerly) unknown size */
Bradley M. Kuhnc97131c2010-08-08 02:51:20 +0200544 progress_meter(PROGRESS_END);
Denys Vlasenko7f432802009-06-28 01:02:24 +0200545}
546
Pere Orga53695632011-02-16 20:09:36 +0100547static void download_one_url(const char *url)
Eric Andersen96700832000-09-04 15:15:55 +0000548{
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100549 bool use_proxy; /* Use proxies if env vars are set */
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200550 int redir_limit;
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100551 len_and_sockaddr *lsa;
Denys Vlasenko7f432802009-06-28 01:02:24 +0200552 FILE *sfp; /* socket to web/ftp server */
Denis Vlasenkoa36535b2007-09-27 15:07:23 +0000553 FILE *dfp; /* socket to ftp server (data) */
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100554 char *proxy = NULL;
555 char *fname_out_alloc;
556 struct host_info server;
557 struct host_info target;
Denis Vlasenko77105632007-09-24 15:04:00 +0000558
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100559 server.allocated = NULL;
560 target.allocated = NULL;
561 server.user = NULL;
Vladimir Dronnikovbe168b12009-10-05 02:18:01 +0200562 target.user = NULL;
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100563
564 parse_url(url, &target);
Eric Andersen79757c92001-04-05 21:45:54 +0000565
Bernhard Reutner-Fischer7e8a53a2007-04-10 09:37:29 +0000566 /* Use the proxy if necessary */
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100567 use_proxy = (strcmp(G.proxy_flag, "off") != 0);
Glenn L McGrathf1c4b112004-02-22 00:27:34 +0000568 if (use_proxy) {
Robert Griebld7760112002-05-14 23:36:45 +0000569 proxy = getenv(target.is_ftp ? "ftp_proxy" : "http_proxy");
Denys Vlasenko2384a352011-02-15 00:58:36 +0100570 use_proxy = (proxy && proxy[0]);
571 if (use_proxy)
Denis Vlasenko96e9d3c2006-10-07 14:28:55 +0000572 parse_url(proxy, &server);
Robert Griebld7760112002-05-14 23:36:45 +0000573 }
Denys Vlasenko7d5ddf12009-06-30 20:36:27 +0200574 if (!use_proxy) {
575 server.port = target.port;
576 if (ENABLE_FEATURE_IPV6) {
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100577 //free(server.allocated); - can't be non-NULL
578 server.host = server.allocated = xstrdup(target.host);
Denys Vlasenko7d5ddf12009-06-30 20:36:27 +0200579 } else {
580 server.host = target.host;
581 }
582 }
583
584 if (ENABLE_FEATURE_IPV6)
585 strip_ipv6_scope_id(target.host);
Eric Andersenc7bda1c2004-03-15 08:29:22 +0000586
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100587 /* If there was no -O FILE, guess output filename */
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100588 fname_out_alloc = NULL;
Denys Vlasenko9a5b7f62011-02-13 02:49:43 +0100589 if (!(option_mask32 & WGET_OPT_OUTNAME)) {
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100590 G.fname_out = bb_get_last_path_component_nostrip(target.path);
Denis Vlasenko818322b2007-09-24 18:27:04 +0000591 /* handle "wget http://kernel.org//" */
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100592 if (G.fname_out[0] == '/' || !G.fname_out[0])
593 G.fname_out = (char*)"index.html";
Denis Vlasenko818322b2007-09-24 18:27:04 +0000594 /* -P DIR is considered only if there was no -O FILE */
Denys Vlasenko625f2182011-03-21 00:29:37 +0100595 else {
596 if (G.dir_prefix)
597 G.fname_out = fname_out_alloc = concat_path_file(G.dir_prefix, G.fname_out);
598 else {
599 /* redirects may free target.path later, need to make a copy */
600 G.fname_out = fname_out_alloc = xstrdup(G.fname_out);
601 }
602 }
Eric Andersen29edd002000-12-09 16:55:35 +0000603 }
Denis Vlasenko818322b2007-09-24 18:27:04 +0000604#if ENABLE_FEATURE_WGET_STATUSBAR
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100605 G.curfile = bb_get_last_path_component_nostrip(G.fname_out);
Denis Vlasenko818322b2007-09-24 18:27:04 +0000606#endif
607
Bernhard Reutner-Fischer7e8a53a2007-04-10 09:37:29 +0000608 /* Determine where to start transfer */
Denys Vlasenko2384a352011-02-15 00:58:36 +0100609 G.beg_range = 0;
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100610 if (option_mask32 & WGET_OPT_CONTINUE) {
Denys Vlasenko2384a352011-02-15 00:58:36 +0100611 G.output_fd = open(G.fname_out, O_WRONLY);
612 if (G.output_fd >= 0) {
613 G.beg_range = xlseek(G.output_fd, 0, SEEK_END);
Denis Vlasenkoa94554d2006-09-23 17:49:09 +0000614 }
615 /* File doesn't exist. We do not create file here yet.
Denys Vlasenkoa84eadf2011-02-12 23:40:31 +0100616 * We are not sure it exists on remote side */
Eric Andersen96700832000-09-04 15:15:55 +0000617 }
618
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200619 redir_limit = 5;
620 resolve_lsa:
Denis Vlasenko42823d52007-02-04 02:39:08 +0000621 lsa = xhost2sockaddr(server.host, server.port);
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100622 if (!(option_mask32 & WGET_OPT_QUIET)) {
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200623 char *s = xmalloc_sockaddr2dotted(&lsa->u.sa);
624 fprintf(stderr, "Connecting to %s (%s)\n", server.host, s);
625 free(s);
Eric Andersene6dc4392003-10-31 09:31:46 +0000626 }
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200627 establish_session:
Denys Vlasenko2384a352011-02-15 00:58:36 +0100628 /*G.content_len = 0; - redundant, got_clen = 0 is enough */
629 G.got_clen = 0;
630 G.chunked = 0;
Glenn L McGrathf1c4b112004-02-22 00:27:34 +0000631 if (use_proxy || !target.is_ftp) {
Eric Andersen79757c92001-04-05 21:45:54 +0000632 /*
633 * HTTP session
634 */
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200635 char *str;
Denys Vlasenko7f432802009-06-28 01:02:24 +0200636 int status;
Denys Vlasenko7f432802009-06-28 01:02:24 +0200637
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100638
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200639 /* Open socket to http server */
640 sfp = open_socket(lsa);
Denys Vlasenko7f432802009-06-28 01:02:24 +0200641
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200642 /* Send HTTP request */
643 if (use_proxy) {
644 fprintf(sfp, "GET %stp://%s/%s HTTP/1.1\r\n",
645 target.is_ftp ? "f" : "ht", target.host,
646 target.path);
647 } else {
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100648 if (option_mask32 & WGET_OPT_POST_DATA)
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200649 fprintf(sfp, "POST /%s HTTP/1.1\r\n", target.path);
650 else
651 fprintf(sfp, "GET /%s HTTP/1.1\r\n", target.path);
652 }
Glenn L McGrathe7bdfcc2003-08-28 22:03:19 +0000653
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200654 fprintf(sfp, "Host: %s\r\nUser-Agent: %s\r\n",
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100655 target.host, G.user_agent);
Eric Andersen79757c92001-04-05 21:45:54 +0000656
Denys Vlasenko9213a552011-02-10 13:23:45 +0100657 /* Ask server to close the connection as soon as we are done
658 * (IOW: we do not intend to send more requests)
659 */
660 fprintf(sfp, "Connection: close\r\n");
661
Denis Vlasenko9cade082006-11-21 10:43:02 +0000662#if ENABLE_FEATURE_WGET_AUTHENTICATION
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200663 if (target.user) {
Denys Vlasenkod5068972011-04-16 19:31:08 +0200664//TODO: URL-decode "user:password" string before base64-encoding:
665//wget http://test:my%20pass@example.com should send
666// Authorization: Basic dGVzdDpteSBwYXNz
667//which decodes to "test:my pass", instead of what we send now:
668// Authorization: Basic dGVzdDpteSUyMHBhc3M=
669//Can reuse decodeString() from httpd.c
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200670 fprintf(sfp, "Proxy-Authorization: Basic %s\r\n"+6,
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100671 base64enc(target.user));
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200672 }
673 if (use_proxy && server.user) {
674 fprintf(sfp, "Proxy-Authorization: Basic %s\r\n",
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100675 base64enc(server.user));
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200676 }
Eric Andersen79757c92001-04-05 21:45:54 +0000677#endif
678
Denys Vlasenkoa3aa3e32009-12-11 12:36:10 +0100679 if (G.beg_range)
680 fprintf(sfp, "Range: bytes=%"OFF_FMT"u-\r\n", G.beg_range);
Denys Vlasenko9213a552011-02-10 13:23:45 +0100681
Denis Vlasenkoc8400a22006-10-25 00:33:44 +0000682#if ENABLE_FEATURE_WGET_LONG_OPTIONS
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100683 if (G.extra_headers)
684 fputs(G.extra_headers, sfp);
Denis Vlasenko5a2ad692009-03-04 14:13:37 +0000685
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100686 if (option_mask32 & WGET_OPT_POST_DATA) {
Denys Vlasenko9213a552011-02-10 13:23:45 +0100687 fprintf(sfp,
688 "Content-Type: application/x-www-form-urlencoded\r\n"
689 "Content-Length: %u\r\n"
690 "\r\n"
691 "%s",
Vitaly Magerya700fbc32011-03-27 22:33:13 +0200692 (int) strlen(G.post_data), G.post_data
Denys Vlasenko9213a552011-02-10 13:23:45 +0100693 );
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200694 } else
Denis Vlasenkoc8400a22006-10-25 00:33:44 +0000695#endif
Denys Vlasenko9213a552011-02-10 13:23:45 +0100696 {
697 fprintf(sfp, "\r\n");
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200698 }
Eric Andersen79757c92001-04-05 21:45:54 +0000699
Nguyễn Thái Ngọc Duyebec11d2010-09-23 15:18:41 +0200700 fflush(sfp);
701
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200702 /*
703 * Retrieve HTTP response line and check for "200" status code.
704 */
Denis Vlasenko023b57d2006-10-15 17:05:55 +0000705 read_response:
Denys Vlasenkof836f012011-02-10 23:02:28 +0100706 fgets_and_trim(sfp);
Eric Andersenc7bda1c2004-03-15 08:29:22 +0000707
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100708 str = G.wget_buf;
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200709 str = skip_non_whitespace(str);
710 str = skip_whitespace(str);
711 // FIXME: no error check
712 // xatou wouldn't work: "200 OK"
713 status = atoi(str);
714 switch (status) {
715 case 0:
716 case 100:
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100717 while (gethdr(sfp) != NULL)
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200718 /* eat all remaining headers */;
719 goto read_response;
720 case 200:
Denis Vlasenko50b5cac2008-06-22 16:28:02 +0000721/*
722Response 204 doesn't say "null file", it says "metadata
723has changed but data didn't":
724
725"10.2.5 204 No Content
726The server has fulfilled the request but does not need to return
727an entity-body, and might want to return updated metainformation.
728The response MAY include new or updated metainformation in the form
729of entity-headers, which if present SHOULD be associated with
730the requested variant.
731
732If the client is a user agent, it SHOULD NOT change its document
733view from that which caused the request to be sent. This response
734is primarily intended to allow input for actions to take place
735without causing a change to the user agent's active document view,
736although any new or updated metainformation SHOULD be applied
737to the document currently in the user agent's active view.
738
739The 204 response MUST NOT include a message-body, and thus
740is always terminated by the first empty line after the header fields."
741
742However, in real world it was observed that some web servers
743(e.g. Boa/0.94.14rc21) simply use code 204 when file size is zero.
744*/
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200745 case 204:
746 break;
Denys Vlasenkofb132e42010-10-29 11:46:52 +0200747 case 300: /* redirection */
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200748 case 301:
749 case 302:
750 case 303:
751 break;
752 case 206:
Denys Vlasenkoa3aa3e32009-12-11 12:36:10 +0100753 if (G.beg_range)
Denis Vlasenko023b57d2006-10-15 17:05:55 +0000754 break;
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200755 /* fall through */
756 default:
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100757 bb_error_msg_and_die("server returned error: %s", sanitize_string(G.wget_buf));
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200758 }
Eric Andersenc7bda1c2004-03-15 08:29:22 +0000759
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200760 /*
761 * Retrieve HTTP headers.
762 */
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100763 while ((str = gethdr(sfp)) != NULL) {
764 static const char keywords[] ALIGN1 =
765 "content-length\0""transfer-encoding\0""location\0";
766 enum {
767 KEY_content_length = 1, KEY_transfer_encoding, KEY_location
768 };
Matthijs van de Water0d586662009-08-22 20:19:48 +0200769 smalluint key;
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100770
771 /* gethdr converted "FOO:" string to lowercase */
772
Matthijs van de Water0d586662009-08-22 20:19:48 +0200773 /* strip trailing whitespace */
774 char *s = strchrnul(str, '\0') - 1;
775 while (s >= str && (*s == ' ' || *s == '\t')) {
776 *s = '\0';
777 s--;
778 }
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100779 key = index_in_strings(keywords, G.wget_buf) + 1;
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200780 if (key == KEY_content_length) {
Denys Vlasenkoa3aa3e32009-12-11 12:36:10 +0100781 G.content_len = BB_STRTOOFF(str, NULL, 10);
782 if (G.content_len < 0 || errno) {
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200783 bb_error_msg_and_die("content-length %s is garbage", sanitize_string(str));
Eric Andersen79757c92001-04-05 21:45:54 +0000784 }
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200785 G.got_clen = 1;
786 continue;
787 }
788 if (key == KEY_transfer_encoding) {
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100789 if (strcmp(str_tolower(str), "chunked") != 0)
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200790 bb_error_msg_and_die("transfer encoding '%s' is not supported", sanitize_string(str));
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100791 G.chunked = 1;
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200792 }
793 if (key == KEY_location && status >= 300) {
794 if (--redir_limit == 0)
795 bb_error_msg_and_die("too many redirections");
796 fclose(sfp);
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100797 if (str[0] == '/') {
798 free(target.allocated);
799 target.path = target.allocated = xstrdup(str+1);
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200800 /* lsa stays the same: it's on the same server */
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100801 } else {
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200802 parse_url(str, &target);
803 if (!use_proxy) {
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100804 free(server.allocated);
Pere Orga57b49092011-02-14 23:56:07 +0100805 server.allocated = NULL;
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200806 server.host = target.host;
Denys Vlasenko7d5ddf12009-06-30 20:36:27 +0200807 /* strip_ipv6_scope_id(target.host); - no! */
808 /* we assume remote never gives us IPv6 addr with scope id */
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200809 server.port = target.port;
Denis Vlasenko6536a9b2007-01-12 10:35:23 +0000810 free(lsa);
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200811 goto resolve_lsa;
812 } /* else: lsa stays the same: we use proxy */
Eric Andersen79757c92001-04-05 21:45:54 +0000813 }
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200814 goto establish_session;
Eric Andersen79757c92001-04-05 21:45:54 +0000815 }
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200816 }
817// if (status >= 300)
818// bb_error_msg_and_die("bad redirection (no Location: header from server)");
Eric Andersenc7bda1c2004-03-15 08:29:22 +0000819
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200820 /* For HTTP, data is pumped over the same connection */
Eric Andersen79757c92001-04-05 21:45:54 +0000821 dfp = sfp;
Denis Vlasenko96e9d3c2006-10-07 14:28:55 +0000822
823 } else {
Eric Andersen79757c92001-04-05 21:45:54 +0000824 /*
825 * FTP session
826 */
Denys Vlasenko7f432802009-06-28 01:02:24 +0200827 sfp = prepare_ftp_session(&dfp, &target, lsa);
Eric Andersen96700832000-09-04 15:15:55 +0000828 }
Denis Vlasenko77105632007-09-24 15:04:00 +0000829
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100830 free(lsa);
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100831
Denys Vlasenko9a5b7f62011-02-13 02:49:43 +0100832 if (!(option_mask32 & WGET_OPT_SPIDER)) {
Denys Vlasenko2384a352011-02-15 00:58:36 +0100833 if (G.output_fd < 0)
834 G.output_fd = xopen(G.fname_out, G.o_flags);
835 retrieve_file_data(dfp);
836 if (!(option_mask32 & WGET_OPT_OUTNAME)) {
837 xclose(G.output_fd);
838 G.output_fd = -1;
Denys Vlasenko9a5b7f62011-02-13 02:49:43 +0100839 }
Bernhard Reutner-Fischer2e75dcc2007-04-05 10:31:47 +0000840 }
Eric Andersen79757c92001-04-05 21:45:54 +0000841
Denys Vlasenkof1fab092009-06-28 03:33:57 +0200842 if (dfp != sfp) {
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100843 /* It's ftp. Close data connection properly */
Eric Andersen79757c92001-04-05 21:45:54 +0000844 fclose(dfp);
Denys Vlasenkodf4e16c2011-02-10 06:29:06 +0100845 if (ftpcmd(NULL, NULL, sfp) != 226)
846 bb_error_msg_and_die("ftp error: %s", sanitize_string(G.wget_buf + 4));
847 /* ftpcmd("QUIT", NULL, sfp); - why bother? */
Eric Andersen79757c92001-04-05 21:45:54 +0000848 }
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100849 fclose(sfp);
Denis Vlasenko77105632007-09-24 15:04:00 +0000850
Denys Vlasenko9a5b7f62011-02-13 02:49:43 +0100851 free(server.allocated);
852 free(target.allocated);
853 free(fname_out_alloc);
Eric Andersen96700832000-09-04 15:15:55 +0000854}
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100855
856int wget_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
857int wget_main(int argc UNUSED_PARAM, char **argv)
858{
859#if ENABLE_FEATURE_WGET_LONG_OPTIONS
860 static const char wget_longopts[] ALIGN1 =
861 /* name, has_arg, val */
862 "continue\0" No_argument "c"
863//FIXME: -s isn't --spider, it's --save-headers!
864 "spider\0" No_argument "s"
865 "quiet\0" No_argument "q"
866 "output-document\0" Required_argument "O"
867 "directory-prefix\0" Required_argument "P"
868 "proxy\0" Required_argument "Y"
869 "user-agent\0" Required_argument "U"
870#if ENABLE_FEATURE_WGET_TIMEOUT
871 "timeout\0" Required_argument "T"
872#endif
873 /* Ignored: */
874 // "tries\0" Required_argument "t"
875 /* Ignored (we always use PASV): */
876 "passive-ftp\0" No_argument "\xff"
877 "header\0" Required_argument "\xfe"
878 "post-data\0" Required_argument "\xfd"
879 /* Ignored (we don't do ssl) */
880 "no-check-certificate\0" No_argument "\xfc"
881 ;
882#endif
883
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100884#if ENABLE_FEATURE_WGET_LONG_OPTIONS
885 llist_t *headers_llist = NULL;
886#endif
887
888 INIT_G();
889
890 IF_FEATURE_WGET_TIMEOUT(G.timeout_seconds = 900;)
891 G.proxy_flag = "on"; /* use proxies if env vars are set */
892 G.user_agent = "Wget"; /* "User-Agent" header field */
893
894#if ENABLE_FEATURE_WGET_LONG_OPTIONS
895 applet_long_options = wget_longopts;
896#endif
897 opt_complementary = "-1" IF_FEATURE_WGET_TIMEOUT(":T+") IF_FEATURE_WGET_LONG_OPTIONS(":\xfe::");
898 getopt32(argv, "csqO:P:Y:U:T:" /*ignored:*/ "t:",
899 &G.fname_out, &G.dir_prefix,
900 &G.proxy_flag, &G.user_agent,
901 IF_FEATURE_WGET_TIMEOUT(&G.timeout_seconds) IF_NOT_FEATURE_WGET_TIMEOUT(NULL),
902 NULL /* -t RETRIES */
903 IF_FEATURE_WGET_LONG_OPTIONS(, &headers_llist)
904 IF_FEATURE_WGET_LONG_OPTIONS(, &G.post_data)
905 );
906 argv += optind;
907
908#if ENABLE_FEATURE_WGET_LONG_OPTIONS
909 if (headers_llist) {
910 int size = 1;
911 char *cp;
912 llist_t *ll = headers_llist;
913 while (ll) {
914 size += strlen(ll->data) + 2;
915 ll = ll->link;
916 }
917 G.extra_headers = cp = xmalloc(size);
918 while (headers_llist) {
919 cp += sprintf(cp, "%s\r\n", (char*)llist_pop(&headers_llist));
920 }
921 }
922#endif
923
Denys Vlasenko2384a352011-02-15 00:58:36 +0100924 G.output_fd = -1;
925 G.o_flags = O_WRONLY | O_CREAT | O_TRUNC | O_EXCL;
926 if (G.fname_out) { /* -O FILE ? */
927 if (LONE_DASH(G.fname_out)) { /* -O - ? */
928 G.output_fd = 1;
929 option_mask32 &= ~WGET_OPT_CONTINUE;
930 }
931 /* compat with wget: -O FILE can overwrite */
932 G.o_flags = O_WRONLY | O_CREAT | O_TRUNC;
933 }
934
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100935 while (*argv)
Pere Orga53695632011-02-16 20:09:36 +0100936 download_one_url(*argv++);
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100937
Denys Vlasenko28556b92011-02-15 11:03:53 +0100938 if (G.output_fd >= 0)
939 xclose(G.output_fd);
940
Pere Orga53695632011-02-16 20:09:36 +0100941 return EXIT_SUCCESS;
Denys Vlasenkoa3661092011-02-13 02:33:11 +0100942}