example: handle uclient connection errors

Signed-off-by: Felix Fietkau <nbd@openwrt.org>

diff --git a/uclient-example.c b/uclient-example.c
index 57a8f821d9d0a81cdf1337934f9178930aeaa4f6..144696c00fd94a1c3f6486e07bdeb3197d9e3e58 100644 (file)
--- a/uclient-example.c
+++ b/uclient-example.c
@@ -35,6 +35,7 @@
 static struct ustream_ssl_ctx *ssl_ctx;
 static const struct ustream_ssl_ops *ssl_ops;
 static int quiet = false;
+static bool verify = true;
 static const char *output_file;
 static int output_fd = -1;
 static int error_ret;
@@ -177,19 +178,45 @@ static void example_eof(struct uclient *cl)
        request_done(cl);
 }
 
-static void example_error(struct uclient *cl, int code)
+static void handle_uclient_error(struct uclient *cl, int code)
 {
+       const char *type = "Unknown error";
+       bool ignore = false;
+
+       switch(code) {
+       case UCLIENT_ERROR_CONNECT:
+               type = "Connection failed";
+               error_ret = 4;
+               break;
+       case UCLIENT_ERROR_SSL_INVALID_CERT:
+               type = "Invalid SSL certificate";
+               ignore = !verify;
+               error_ret = 5;
+               break;
+       case UCLIENT_ERROR_SSL_CN_MISMATCH:
+               type = "Server hostname does not match SSL certificate";
+               ignore = !verify;
+               error_ret = 5;
+               break;
+       default:
+               error_ret = 1;
+               break;
+       }
+
        if (!quiet)
-               fprintf(stderr, "Error %d!\n", code);
+               fprintf(stderr, "Connection error: %s%s\n", type, ignore ? " (ignored)" : "");
 
-       request_done(cl);
+       if (ignore)
+               error_ret = 0;
+       else
+               request_done(cl);
 }
 
 static const struct uclient_cb cb = {
        .header_done = example_header_done,
        .data_read = example_read_data,
        .data_eof = example_eof,
-       .error = example_error,
+       .error = handle_uclient_error,
 };
 
 static int usage(const char *progname)
@@ -243,7 +270,6 @@ int main(int argc, char **argv)
 {
        const char *progname = argv[0];
        struct uclient *cl;
-       bool verify = true;
        int ch;
        int longopt_idx = 0;