-#include <libubox/blobmsg.h>
+/*
+ * uclient - ustream based protocol client library
+ *
+ * Copyright (C) 2014 Felix Fietkau <nbd@openwrt.org>
+ *
+ * Permission to use, copy, modify, and/or distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+
#include <unistd.h>
#include <stdio.h>
+#include <dlfcn.h>
+
+#include <libubox/blobmsg.h>
#include "uclient.h"
+#ifdef __APPLE__
+#define LIB_EXT "dylib"
+#else
+#define LIB_EXT "so"
+#endif
+
+static struct ustream_ssl_ctx *ssl_ctx;
+static const struct ustream_ssl_ops *ssl_ops;
+
static void example_header_done(struct uclient *cl)
{
struct blob_attr *cur;
+ char local[INET6_ADDRSTRLEN], remote[INET6_ADDRSTRLEN];
+ int local_port, remote_port;
int rem;
+ uclient_get_addr(local, &local_port, &cl->local_addr);
+ uclient_get_addr(remote, &remote_port, &cl->remote_addr);
+
+ fprintf(stderr, "Connected: %s:%d -> %s:%d\n",
+ local, local_port, remote, remote_port);
+
printf("Headers (%d): \n", cl->status_code);
blobmsg_for_each_attr(cur, cl->meta, rem) {
printf("%s=%s\n", blobmsg_name(cur), (char *) blobmsg_data(cur));
uclient_request(cl);
break;
default:
- uclient_free(cl);
uloop_end();
break;
};
static void example_eof(struct uclient *cl)
{
+ static int retries;
+
+ if (retries < 10 && uclient_http_redirect(cl)) {
+ retries++;
+ return;
+ }
+
+ retries = 0;
+ example_request_sm(cl);
+}
+
+static void example_error(struct uclient *cl, int code)
+{
+ fprintf(stderr, "Error %d!\n", code);
example_request_sm(cl);
}
.header_done = example_header_done,
.data_read = example_read_data,
.data_eof = example_eof,
+ .error = example_error,
};
+static int usage(const char *progname)
+{
+ fprintf(stderr,
+ "Usage: %s [options] <hostname> <port>\n"
+ "Options:\n"
+ " -c <cert>: Load CA certificates from file <cert>\n"
+ " -C: Skip certificate CN verification against hostname\n"
+ "\n", progname);
+ return 1;
+}
+
+
+static void init_ustream_ssl(void)
+{
+ void *dlh;
+
+ dlh = dlopen("libustream-ssl." LIB_EXT, RTLD_LAZY | RTLD_LOCAL);
+ if (!dlh)
+ return;
+
+ ssl_ops = dlsym(dlh, "ustream_ssl_ops");
+ if (!ssl_ops)
+ return;
+
+ ssl_ctx = ssl_ops->context_new(false);
+}
+
+static int no_ssl(const char *progname)
+{
+ fprintf(stderr, "%s: SSL support not available, please install ustream-ssl\n", progname);
+ return 1;
+}
+
+
int main(int argc, char **argv)
{
+ const char *progname = argv[0];
struct uclient *cl;
+ bool verify = true;
+ int ch;
- if (argc != 2) {
- fprintf(stderr, "Usage: %s <URL>\n", argv[0]);
- return 1;
+ init_ustream_ssl();
+
+ while ((ch = getopt(argc, argv, "Cc:")) != -1) {
+ switch(ch) {
+ case 'c':
+ if (ssl_ctx)
+ ssl_ops->context_add_ca_crt_file(ssl_ctx, optarg);
+ break;
+ case 'C':
+ verify = false;
+ break;
+ default:
+ return usage(progname);
+ }
}
+ argv += optind;
+ argc -= optind;
+
+ if (argc != 1)
+ return usage(progname);
+
+ if (!strncmp(argv[0], "https", 5) && !ssl_ctx)
+ return no_ssl(progname);
+
uloop_init();
- cl = uclient_new(argv[1], &cb);
+
+ cl = uclient_new(argv[0], NULL, &cb);
if (!cl) {
fprintf(stderr, "Failed to allocate uclient context\n");
return 1;
}
+
+ if (ssl_ctx)
+ uclient_http_set_ssl_ctx(cl, ssl_ops, ssl_ctx, verify);
+
example_request_sm(cl);
uloop_run();
uloop_done();
+ uclient_free(cl);
+
+ if (ssl_ctx)
+ ssl_ops->context_free(ssl_ctx);
+
return 0;
}