diff options
Diffstat (limited to 'networking/wget.c')
-rw-r--r-- | networking/wget.c | 105 |
1 files changed, 49 insertions, 56 deletions
diff --git a/networking/wget.c b/networking/wget.c index 1da18ff77..026722afc 100644 --- a/networking/wget.c +++ b/networking/wget.c @@ -34,7 +34,7 @@ static off_t beg_range; /* Range at which continue begins */ #if ENABLE_FEATURE_WGET_STATUSBAR static off_t transferred; /* Number of bytes transferred so far */ #endif -static int chunked; /* chunked transfer encoding */ +static bool chunked; /* chunked transfer encoding */ #if ENABLE_FEATURE_WGET_STATUSBAR static void progressmeter(int flag); static const char *curfile; /* Name of current file being transferred */ @@ -76,9 +76,7 @@ static char *safe_fgets(char *s, int size, FILE *stream) } #if ENABLE_FEATURE_WGET_AUTHENTICATION -/* - * Base64-encode character string and return the string. - */ +/* Base64-encode character string and return the string. */ static char *base64enc(unsigned char *p, char *buf, int len) { bb_uuencode(p, buf, len, bb_uuenc_tbl_base64); @@ -96,7 +94,7 @@ int wget_main(int argc, char **argv) int port; int try = 5; unsigned opt; - char *s; + char *str; char *proxy = 0; char *dir_prefix = NULL; #if ENABLE_FEATURE_WGET_LONG_OPTIONS @@ -104,20 +102,20 @@ int wget_main(int argc, char **argv) llist_t *headers_llist = NULL; #endif - /* server.allocated = target.allocated = NULL; */ - FILE *sfp = NULL; /* socket to web/ftp server */ FILE *dfp = NULL; /* socket to ftp server (data) */ char *fname_out = NULL; /* where to direct output (-O) */ - int got_clen = 0; /* got content-length: from server */ + bool got_clen = 0; /* got content-length: from server */ int output_fd = -1; - int use_proxy = 1; /* Use proxies if env vars are set */ + bool use_proxy = 1; /* Use proxies if env vars are set */ const char *proxy_flag = "on"; /* Use proxies if env vars are set */ - const char *user_agent = "Wget";/* Content of the "User-Agent" header field */ - - /* - * Crack command line. - */ + const char *user_agent = "Wget";/* "User-Agent" header field */ + static const char * const keywords[] = { + "content-length", "transfer-encoding", "chunked", "location", NULL + }; + enum { + KEY_content_length = 1, KEY_transfer_encoding, KEY_chunked, KEY_location + }; enum { WGET_OPT_CONTINUE = 0x1, WGET_OPT_SPIDER = 0x2, @@ -131,7 +129,7 @@ int wget_main(int argc, char **argv) }; #if ENABLE_FEATURE_WGET_LONG_OPTIONS static const struct option wget_long_options[] = { - // name, has_arg, flag, val + /* name, has_arg, flag, val */ { "continue", no_argument, NULL, 'c' }, { "spider", no_argument, NULL, 's' }, { "quiet", no_argument, NULL, 'q' }, @@ -145,6 +143,7 @@ int wget_main(int argc, char **argv) }; applet_long_options = wget_long_options; #endif + /* server.allocated = target.allocated = NULL; */ opt_complementary = "-1" USE_FEATURE_WGET_LONG_OPTIONS(":\xfe::"); opt = getopt32(argc, argv, "csqO:P:Y:U:", &fname_out, &dir_prefix, @@ -152,7 +151,7 @@ int wget_main(int argc, char **argv) USE_FEATURE_WGET_LONG_OPTIONS(, &headers_llist) ); if (strcmp(proxy_flag, "off") == 0) { - /* Use the proxy if necessary. */ + /* Use the proxy if necessary */ use_proxy = 0; } #if ENABLE_FEATURE_WGET_LONG_OPTIONS @@ -176,9 +175,7 @@ int wget_main(int argc, char **argv) server.host = target.host; server.port = target.port; - /* - * Use the proxy if necessary. - */ + /* Use the proxy if necessary */ if (use_proxy) { proxy = getenv(target.is_ftp ? "ftp_proxy" : "http_proxy"); if (proxy && *proxy) { @@ -217,9 +214,7 @@ int wget_main(int argc, char **argv) if ((opt & WGET_OPT_CONTINUE) && !fname_out) bb_error_msg_and_die("cannot specify continue (-c) without a filename (-O)"); */ - /* - * Determine where to start transfer. - */ + /* Determine where to start transfer */ if (LONE_DASH(fname_out)) { output_fd = 1; opt &= ~WGET_OPT_CONTINUE; @@ -253,15 +248,11 @@ int wget_main(int argc, char **argv) if (!--try) bb_error_msg_and_die("too many redirections"); - /* - * Open socket to http server - */ + /* Open socket to http server */ if (sfp) fclose(sfp); sfp = open_socket(lsa); - /* - * Send HTTP request. - */ + /* Send HTTP request. */ if (use_proxy) { fprintf(sfp, "GET %stp://%s/%s HTTP/1.1\r\n", target.is_ftp ? "f" : "ht", target.host, @@ -299,12 +290,12 @@ int wget_main(int argc, char **argv) if (fgets(buf, sizeof(buf), sfp) == NULL) bb_error_msg_and_die("no response from server"); - s = buf; - while (*s != '\0' && !isspace(*s)) ++s; - s = skip_whitespace(s); + str = buf; + str = skip_non_whitespace(str); + str = skip_whitespace(str); // FIXME: no error check // xatou wouldn't work: "200 OK" - status = atoi(s); + status = atoi(str); switch (status) { case 0: case 100: @@ -331,26 +322,28 @@ int wget_main(int argc, char **argv) /* * Retrieve HTTP headers. */ - while ((s = gethdr(buf, sizeof(buf), sfp, &n)) != NULL) { - if (strcasecmp(buf, "content-length") == 0) { - content_len = BB_STRTOOFF(s, NULL, 10); + while ((str = gethdr(buf, sizeof(buf), sfp, &n)) != NULL) { + /* gethdr did already convert the "FOO:" string to lowercase */ + smalluint key = index_in_str_array(keywords, *&buf) + 1; + if (key == KEY_content_length) { + content_len = BB_STRTOOFF(str, NULL, 10); if (errno || content_len < 0) { - bb_error_msg_and_die("content-length %s is garbage", s); + bb_error_msg_and_die("content-length %s is garbage", str); } got_clen = 1; continue; } - if (strcasecmp(buf, "transfer-encoding") == 0) { - if (strcasecmp(s, "chunked") != 0) - bb_error_msg_and_die("server wants to do %s transfer encoding", s); + if (key == KEY_transfer_encoding) { + if (index_in_str_array(keywords, str_tolower(str)) + 1 != KEY_chunked) + bb_error_msg_and_die("server wants to do %s transfer encoding", str); chunked = got_clen = 1; } - if (strcasecmp(buf, "location") == 0) { - if (s[0] == '/') + if (key == KEY_location) { + if (str[0] == '/') /* free(target.allocated); */ - target.path = /* target.allocated = */ xstrdup(s+1); + target.path = /* target.allocated = */ xstrdup(str+1); else { - parse_url(s, &target); + parse_url(str, &target); if (use_proxy == 0) { server.host = target.host; server.port = target.port; @@ -381,14 +374,14 @@ int wget_main(int argc, char **argv) * Splitting username:password pair, * trying to log in */ - s = strchr(target.user, ':'); - if (s) - *(s++) = '\0'; + str = strchr(target.user, ':'); + if (str) + *(str++) = '\0'; switch (ftpcmd("USER ", target.user, sfp, buf)) { case 230: break; case 331: - if (ftpcmd("PASS ", s, sfp, buf) == 230) + if (ftpcmd("PASS ", str, sfp, buf) == 230) break; /* FALLTHRU (failed login) */ default: @@ -418,15 +411,15 @@ int wget_main(int argc, char **argv) // Response is "227 garbageN1,N2,N3,N4,P1,P2[)garbage] // Server's IP is N1.N2.N3.N4 (we ignore it) // Server's port for data connection is P1*256+P2 - s = strrchr(buf, ')'); - if (s) s[0] = '\0'; - s = strrchr(buf, ','); - if (!s) goto pasv_error; - port = xatou_range(s+1, 0, 255); - *s = '\0'; - s = strrchr(buf, ','); - if (!s) goto pasv_error; - port += xatou_range(s+1, 0, 255) * 256; + str = strrchr(buf, ')'); + if (str) str[0] = '\0'; + str = strrchr(buf, ','); + if (!str) goto pasv_error; + port = xatou_range(str+1, 0, 255); + *str = '\0'; + str = strrchr(buf, ','); + if (!str) goto pasv_error; + port += xatou_range(str+1, 0, 255) * 256; set_nport(lsa, htons(port)); dfp = open_socket(lsa); |