diff options
author | Luka Perkov <luka@openwrt.org> | 2014-07-07 10:06:24 +0200 |
---|---|---|
committer | Felix Fietkau <nbd@openwrt.org> | 2014-07-07 10:25:10 +0200 |
commit | 5152b7147e7a2fbb4543160e69ea75dbb3b5821a (patch) | |
tree | 382de2713aad71b4cbb42643c6814afa8d7e4bf6 | |
parent | 0fa104b6042913eb3eae9bdddad1cf341c85e614 (diff) | |
download | uclient-5152b7147e7a2fbb4543160e69ea75dbb3b5821a.tar.gz |
fetch: fix segfault after destination was not reached
Signed-off-by: Luka Perkov <luka@openwrt.org>
-rw-r--r-- | uclient-fetch.c | 34 |
1 files changed, 28 insertions, 6 deletions
diff --git a/uclient-fetch.c b/uclient-fetch.c index 881b4ea..22f15c6 100644 --- a/uclient-fetch.c +++ b/uclient-fetch.c @@ -149,13 +149,27 @@ static void msg_connecting(struct uclient *cl) 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) @@ -269,6 +283,7 @@ int main(int argc, char **argv) struct uclient *cl; int ch; int longopt_idx = 0; + int rc; init_ustream_ssl(); @@ -340,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); |