X-Git-Url: http://git.archive.openwrt.org/?p=project%2Fuclient.git;a=blobdiff_plain;f=uclient-fetch.c;h=22f15c6feb801cdafe05b811ce076e56d22f9ea4;hp=164492128319a7f8e966aa979f29707be56a7809;hb=5152b7147e7a2fbb4543160e69ea75dbb3b5821a;hpb=e350ca4e73771c5a23e08ecc25ea4d3a966e51f8

diff --git a/uclient-fetch.c b/uclient-fetch.c
index 1644921..22f15c6 100644
--- a/uclient-fetch.c
+++ b/uclient-fetch.c
@@ -16,6 +16,7 @@
  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  */
 
+#define _GNU_SOURCE
 #include <unistd.h>
 #include <stdio.h>
 #include <dlfcn.h>
@@ -41,6 +42,9 @@ 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)
 {
@@ -142,16 +146,30 @@ static void msg_connecting(struct uclient *cl)
 		return;
 
 	uclient_get_addr(addr, &port, &cl->remote_addr);
-	fprintf(stderr, "Connecting to %s %s:%d\n", cl->url->host, addr, port);
+	fprintf(stderr, "Connecting to %s:%d\n", addr, port);
 }
 
-static void init_request(struct uclient *cl)
+static int init_request(struct uclient *cl)
 {
+	int rc;
+
 	out_bytes = 0;
-	uclient_connect(cl);
+
+	rc = uclient_connect(cl);
+	if (rc)
+		return rc;
+
 	msg_connecting(cl);
-	uclient_http_set_request_type(cl, "GET");
-	uclient_request(cl);
+
+	rc = uclient_http_set_request_type(cl, "GET");
+	if (rc)
+		return rc;
+
+	rc = uclient_request(cl);
+	if (rc)
+		return rc;
+
+	return 0;
 }
 
 static void eof_cb(struct uclient *cl)
@@ -247,11 +265,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 },
 	{}
 };
 
@@ -261,6 +283,7 @@ int main(int argc, char **argv)
 	struct uclient *cl;
 	int ch;
 	int longopt_idx = 0;
+	int rc;
 
 	init_ustream_ssl();
 
@@ -275,6 +298,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);
 			}
@@ -301,7 +336,17 @@ 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;
+	}
+
+	if (!quiet)
+		fprintf(stderr, "Downloading '%s'\n", argv[0]);
+
+	cl = uclient_new(argv[0], auth_str, &cb);
 	if (!cl) {
 		fprintf(stderr, "Failed to allocate uclient context\n");
 		return 1;
@@ -310,8 +355,15 @@ int main(int argc, char **argv)
 	if (ssl_ctx)
 		uclient_http_set_ssl_ctx(cl, ssl_ops, ssl_ctx, verify);
 
-	init_request(cl);
-	uloop_run();
+	rc = init_request(cl);
+	if (!rc) {
+		/* no error received, we can enter main loop */
+		uloop_run();
+	} else {
+		fprintf(stderr, "Failed to establish connection\n");
+		error_ret = 4;
+	}
+
 	uloop_done();
 
 	uclient_free(cl);