X-Git-Url: http://git.archive.openwrt.org/?p=project%2Fuclient.git;a=blobdiff_plain;f=uclient-fetch.c;h=733a5b955612ad5534ad97f3f9911f330ae1d1b1;hp=9672e07ecb0b447ef6ef93a1daec5dcd5e1c253a;hb=c8e85b4714709aebaa055bea6cc5a26b8eda4071;hpb=43be3a3e32f9eb32358beaa3934623c664ae7c12;ds=sidebyside

diff --git a/uclient-fetch.c b/uclient-fetch.c
index 9672e07..733a5b9 100644
--- a/uclient-fetch.c
+++ b/uclient-fetch.c
@@ -25,6 +25,7 @@
 #include <libubox/blobmsg.h>
 
 #include "uclient.h"
+#include "uclient-utils.h"
 
 #ifdef __APPLE__
 #define LIB_EXT "dylib"
@@ -39,49 +40,38 @@ static bool verify = true;
 static const char *output_file;
 static int output_fd = -1;
 static int error_ret;
+static int out_bytes;
+static char *username;
+static char *password;
+static char *auth_str;
 
 static int open_output_file(const char *path, bool create)
 {
-	const char *str;
 	char *filename;
-	int len;
 	int flags = O_WRONLY;
+	int ret;
 
 	if (create)
-		flags |= O_CREAT;
+		flags |= O_CREAT | O_EXCL;
 
 	if (output_file) {
 		if (!strcmp(output_file, "-"))
 			return STDOUT_FILENO;
 
-		return open(output_file, flags, 0644);
-	}
-
-	/* Don't automatically overwrite files if the name is derived from the URL */
-	if (create)
-		flags |= O_EXCL;
-
-	len = strcspn(path, ";&");
-	while (len > 0 && path[len - 1] == '/')
-		len--;
+		if (!quiet)
+			fprintf(stderr, "Writing to stdout\n");
 
-	for (str = path + len - 1; str >= path; str--) {
-		if (*str == '/')
-			break;
+		unlink(output_file);
+		return open(output_file, flags, 0644);
 	}
 
-	str++;
-	len -= str - path;
-
-	if (len > 0) {
-		filename = alloca(len + 1);
-		strncpy(filename, str, len);
-		filename[len] = 0;
-	} else {
-		filename = "index.html";
-	}
+	filename = uclient_get_url_filename(path, "index.html");
+	if (!quiet)
+		fprintf(stderr, "Writing to '%s'\n", filename);
+	ret = open(filename, flags, 0644);
+	free(filename);
 
-	return open(filename, flags, 0644);
+	return ret;
 }
 
 static void request_done(struct uclient *cl)
@@ -141,6 +131,7 @@ static void read_data_cb(struct uclient *cl)
 		if (!len)
 			return;
 
+		out_bytes += len;
 		write(output_fd, buf, len);
 	}
 }
@@ -159,6 +150,7 @@ static void msg_connecting(struct uclient *cl)
 
 static void init_request(struct uclient *cl)
 {
+	out_bytes = 0;
 	uclient_connect(cl);
 	msg_connecting(cl);
 	uclient_http_set_request_type(cl, "GET");
@@ -167,6 +159,13 @@ static void init_request(struct uclient *cl)
 
 static void eof_cb(struct uclient *cl)
 {
+	if (!cl->data_eof) {
+		if (!quiet)
+			fprintf(stderr, "Connection reset prematurely\n");
+		error_ret = 4;
+	} else if (!quiet) {
+		fprintf(stderr, "Download completed (%d bytes)\n", out_bytes);
+	}
 	request_done(cl);
 }
 
@@ -251,11 +250,15 @@ static int no_ssl(const char *progname)
 enum {
 	L_NO_CHECK_CERTIFICATE,
 	L_CA_CERTIFICATE,
+	L_USER,
+	L_PASSWORD,
 };
 
 static const struct option longopts[] = {
 	[L_NO_CHECK_CERTIFICATE] = { "no-check-certificate", no_argument },
 	[L_CA_CERTIFICATE] = { "ca-certificate", required_argument },
+	[L_USER] = { "user", required_argument },
+	[L_PASSWORD] = { "password", required_argument },
 	{}
 };
 
@@ -279,6 +282,18 @@ int main(int argc, char **argv)
 				if (ssl_ctx)
 					ssl_ops->context_add_ca_crt_file(ssl_ctx, optarg);
 				break;
+			case L_USER:
+				if (!strlen(optarg))
+					break;
+				username = strdup(optarg);
+				memset(optarg, '*', strlen(optarg));
+				break;
+			case L_PASSWORD:
+				if (!strlen(optarg))
+					break;
+				password = strdup(optarg);
+				memset(optarg, '*', strlen(optarg));
+				break;
 			default:
 				return usage(progname);
 			}
@@ -305,7 +320,14 @@ int main(int argc, char **argv)
 
 	uloop_init();
 
-	cl = uclient_new(argv[0], NULL, &cb);
+	if (username) {
+		if (password)
+			asprintf(&auth_str, "%s:%s", username, password);
+		else
+			auth_str = username;
+	}
+
+	cl = uclient_new(argv[0], auth_str, &cb);
 	if (!cl) {
 		fprintf(stderr, "Failed to allocate uclient context\n");
 		return 1;