#include "busybox.h"
#include <stdio.h>
+#include <errno.h>
#include <stdlib.h>
#include <unistd.h>
#include <ctype.h>
/* Globals (can be accessed from signal handlers */
static off_t filesize = 0; /* content-length of the file */
-#ifdef BB_FEATURE_STATUSBAR
+#ifdef BB_FEATURE_WGET_STATUSBAR
static char *curfile; /* Name of current file being transferred. */
static struct timeval start; /* Time a transfer started. */
volatile unsigned long statbytes; /* Number of bytes transferred so far. */
/* For progressmeter() -- number of seconds before xfer considered "stalled" */
-#define STALLTIME 5
+static const int STALLTIME = 5;
#endif
+
+void close_and_delete_outfile(FILE* output, char *fname_out, int do_continue)
+{
+ if (output != stdout && do_continue==0) {
+ fclose(output);
+ unlink(fname_out);
+ }
+}
int wget_main(int argc, char **argv)
{
- FILE *sfp; /* socket to web server */
- char *uri_host, *uri_path; /* parsed from command line url */
- int uri_port;
- char *s, buf[512];
int n;
+ char *proxy, *proxy_host;
+ int uri_port, proxy_port;
+ char *s, buf[512];
+ struct stat sbuf;
+ FILE *sfp; /* socket to web server */
+ char *uri_host, *uri_path; /* parsed from command line url */
char *fname_out = NULL; /* where to direct output (-O) */
int do_continue = 0; /* continue a prev transfer (-c) */
long beg_range = 0L; /* range at which continue begins */
int got_clen = 0; /* got content-length: from server */
- FILE *output; /* socket to web server */
+ FILE *output; /* socket to web server */
+ int quiet_flag = FALSE; /* Be verry, verry quiet... */
/*
* Crack command line.
*/
- while ((n = getopt(argc, argv, "cO:")) != EOF) {
+ while ((n = getopt(argc, argv, "cqO:")) != EOF) {
switch (n) {
case 'c':
++do_continue;
break;
+ case 'q':
+ quiet_flag = TRUE;
+ break;
case 'O':
/* can't set fname_out to NULL if outputting to stdout, because
* this gets interpreted as the auto-gen output filename
if (argc - optind != 1)
usage(wget_usage);
+ /*
+ * Use the proxy if necessary.
+ */
+ if ((proxy = getenv("http_proxy")) != NULL) {
+ proxy = xstrdup(proxy);
+ parse_url(proxy, &proxy_host, &proxy_port, &uri_path);
+ parse_url(argv[optind], &uri_host, &uri_port, &uri_path);
+ } else {
+ /*
+ * Parse url into components.
+ */
+ parse_url(argv[optind], &uri_host, &uri_port, &uri_path);
+ proxy_host=uri_host;
+ proxy_port=uri_port;
+ }
+
/* Guess an output filename */
if (!fname_out) {
fname_out =
-#ifdef BB_FEATURE_STATUSBAR
+#ifdef BB_FEATURE_WGET_STATUSBAR
curfile =
#endif
- get_last_path_component(argv[optind]);
-#ifdef BB_FEATURE_STATUSBAR
+ get_last_path_component(uri_path);
+ if (fname_out==NULL || strlen(fname_out)<1) {
+ fname_out =
+#ifdef BB_FEATURE_WGET_STATUSBAR
+ curfile =
+#endif
+ "index.html";
+ }
+#ifdef BB_FEATURE_WGET_STATUSBAR
} else {
curfile=argv[optind];
#endif
}
-
-
if (do_continue && !fname_out)
error_msg_and_die("cannot specify continue (-c) without a filename (-O)\n");
- /*
- * Parse url into components.
- */
- parse_url(argv[optind], &uri_host, &uri_port, &uri_path);
+
/*
* Open socket to server.
*/
- sfp = open_socket(uri_host, uri_port);
+ sfp = open_socket(proxy_host, proxy_port);
+
+ /* Make the assumption that if the file already exists
+ * on disk that the intention is to continue downloading
+ * a previously aborted download -Erik */
+ if (stat(fname_out, &sbuf) == 0) {
+ ++do_continue;
+ }
/*
- * Open the output stream.
+ * Open the output file stream.
*/
if (fname_out != (char *)1) {
- if ( (output=fopen(fname_out, (do_continue ? "a" : "w")))
- == NULL)
+ if ( (output=fopen(fname_out, (do_continue ? "a" : "w"))) == NULL)
perror_msg_and_die("fopen(%s)", fname_out);
} else {
output = stdout;
* Determine where to start transfer.
*/
if (do_continue) {
- struct stat sbuf;
if (fstat(fileno(output), &sbuf) < 0)
- error_msg_and_die("fstat()");
+ perror_msg_and_die("fstat()");
if (sbuf.st_size > 0)
beg_range = sbuf.st_size;
else
/*
* Send HTTP request.
*/
- fprintf(sfp, "GET %s HTTP/1.1\r\nHost: %s\r\n", uri_path, uri_host);
+ fprintf(sfp, "GET http://%s:%d/%s HTTP/1.1\r\n",
+ uri_host, uri_port, uri_path);
+ fprintf(sfp, "Host: %s\r\nUser-Agent: Wget\r\n", uri_host);
+
if (do_continue)
fprintf(sfp, "Range: bytes=%ld-\r\n", beg_range);
- fputs("Connection: close\r\n\r\n", sfp);
+ fprintf(sfp,"Connection: close\r\n\r\n");
/*
* Retrieve HTTP response line and check for "200" status code.
*/
- if (fgets(buf, sizeof(buf), sfp) == NULL)
+ if (fgets(buf, sizeof(buf), sfp) == NULL) {
+ close_and_delete_outfile(output, fname_out, do_continue);
error_msg_and_die("no response from server\n");
+ }
for (s = buf ; *s != '\0' && !isspace(*s) ; ++s)
;
for ( ; isspace(*s) ; ++s)
;
switch (atoi(s)) {
+ case 0:
case 200:
- if (!do_continue)
- break;
- error_msg_and_die("server does not support ranges\n");
+ break;
case 206:
if (do_continue)
break;
/*FALLTHRU*/
default:
- error_msg_and_die("server returned error: %s", buf);
+ close_and_delete_outfile(output, fname_out, do_continue);
+ error_msg_and_die("server returned error %d: %s", atoi(s), buf);
}
/*
* Retrieve HTTP headers.
*/
while ((s = gethdr(buf, sizeof(buf), sfp, &n)) != NULL) {
- if (strcmp(buf, "content-length") == 0) {
+ if (strcasecmp(buf, "content-length") == 0) {
filesize = atol(s);
got_clen = 1;
continue;
}
- if (strcmp(buf, "transfer-encoding") == 0) {
+ if (strcasecmp(buf, "transfer-encoding") == 0) {
+ close_and_delete_outfile(output, fname_out, do_continue);
error_msg_and_die("server wants to do %s transfer encoding\n", s);
continue;
}
/*
* Retrieve HTTP body.
*/
-#ifdef BB_FEATURE_STATUSBAR
+#ifdef BB_FEATURE_WGET_STATUSBAR
statbytes=0;
- progressmeter(-1);
+ if (quiet_flag==FALSE)
+ progressmeter(-1);
#endif
while (filesize > 0 && (n = fread(buf, 1, sizeof(buf), sfp)) > 0) {
fwrite(buf, 1, n, output);
-#ifdef BB_FEATURE_STATUSBAR
+#ifdef BB_FEATURE_WGET_STATUSBAR
statbytes+=n;
- progressmeter(1);
+ if (quiet_flag==FALSE)
+ progressmeter(1);
#endif
if (got_clen)
filesize -= n;
void parse_url(char *url, char **uri_host, int *uri_port, char **uri_path)
{
- char *s, *h;
- static char *defaultpath = "/";
+ char *cp, *sp;
*uri_port = 80;
if (strncmp(url, "http://", 7) != 0)
error_msg_and_die("not an http url: %s\n", url);
- /* pull the host portion to the front of the buffer */
- for (s = url, h = url+7 ; *h != '/' && *h != 0; ++h) {
- if (*h == ':') {
- *uri_port = atoi(h+1);
- *h = '\0';
- }
- *s++ = *h;
- }
- *s = '\0';
+ *uri_host = url + 7;
+
+ cp = strchr(*uri_host, ':');
+ sp = strchr(*uri_host, '/');
- if (*h == 0) h = defaultpath;
+ if (cp != NULL && (sp == NULL || cp < sp)) {
+ *cp++ = '\0';
+ *uri_port = atoi(cp);
+ }
- *uri_host = url;
- *uri_path = h;
+ if (sp != NULL) {
+ *sp++ = '\0';
+ *uri_path = sp;
+ } else
+ *uri_path = "";
}
return hdrval;
}
-#ifdef BB_FEATURE_STATUSBAR
+#ifdef BB_FEATURE_WGET_STATUSBAR
/* Stuff below is from BSD rcp util.c, as added to openshh.
* Original copyright notice is retained at the end of this file.
*
}
#endif
-/* Original copyright notice which applies to the BB_FEATURE_STATUSBAR stuff,
+/* Original copyright notice which applies to the BB_FEATURE_WGET_STATUSBAR stuff,
* much of which was blatently stolen from openssh. */
/*-
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*
- * $Id: wget.c,v 1.10 2000/12/07 19:56:48 markw Exp $
+ * $Id: wget.c,v 1.22 2001/01/26 02:04:49 andersen Exp $
*/
projects / oweals / busybox.git / blobdiff