add client example code

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

1 files changed, 118 insertions, 0 deletions

diff --git a/ustream-example-client.c b/ustream-example-client.c
new file mode 100644
index 0000000..a02815b
--- /dev/null
+++ b/ustream-example-client.c
@@ -0,0 +1,118 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+
+#include <libubox/usock.h>
+#include <libubox/uloop.h>
+#include "ustream-ssl.h"
+
+static struct uloop_fd fd;
+
+static struct ustream_fd stream, s_input;
+static struct ustream_ssl ssl;
+
+static void *ctx;
+
+static void client_teardown(void)
+{
+	if (s_input.fd.registered)
+		ustream_free(&s_input.stream);
+
+	ustream_free(&ssl.stream);
+	ustream_free(&stream.stream);
+	close(stream.fd.fd);
+	uloop_end();
+}
+
+static void client_input_notify_read(struct ustream *s, int bytes)
+{
+	char *buf;
+	int len;
+
+	buf = ustream_get_read_buf(s, &len);
+	ustream_write(&ssl.stream, buf, len, false);
+	ustream_consume(s, len);
+}
+
+static void client_ssl_notify_read(struct ustream *s, int bytes)
+{
+	char *buf;
+	int len;
+
+	buf = ustream_get_read_buf(s, &len);
+	fwrite(buf, len, 1, stdout);
+	fflush(stdout);
+	ustream_consume(s, len);
+}
+
+static void client_notify_connected(struct ustream_ssl *ssl)
+{
+	fprintf(stderr, "SSL connection established\n");
+	s_input.stream.notify_read = client_input_notify_read;
+	ustream_fd_init(&s_input, 0);
+}
+
+static void client_notify_error(struct ustream_ssl *ssl, int error, const char *str)
+{
+	fprintf(stderr, "SSL connection error(%d): %s\n", error, str);
+	client_teardown();
+}
+
+static void client_notify_state(struct ustream *us)
+{
+	if (!us->write_error && !us->eof)
+		return;
+
+	fprintf(stderr, "Connection closed\n");
+	client_teardown();
+}
+
+static void example_connect_ssl(int fd)
+{
+	fprintf(stderr, "Starting SSL negnotiation\n");
+
+	ssl.notify_error = client_notify_error;
+	ssl.notify_connected = client_notify_connected;
+	ssl.stream.notify_read = client_ssl_notify_read;
+	ssl.stream.notify_state = client_notify_state;
+
+	ustream_fd_init(&stream, fd);
+	ustream_ssl_init(&ssl, &stream.stream, ctx, false);
+}
+
+static void example_connect_cb(struct uloop_fd *f, unsigned int events)
+{
+	if (fd.eof || fd.error) {
+		fprintf(stderr, "Connection failed\n");
+		uloop_end();
+		return;
+	}
+
+	fprintf(stderr, "Connection established\n");
+	uloop_fd_delete(&fd);
+	example_connect_ssl(fd.fd);
+}
+
+static void connect_client(const char *host, const char *port)
+{
+	fd.fd = usock(USOCK_TCP | USOCK_NONBLOCK, host, port);
+	fd.cb = example_connect_cb;
+	uloop_fd_add(&fd, ULOOP_WRITE | ULOOP_EDGE_TRIGGER);
+}
+
+int main(int argc, char **argv)
+{
+	if (argc != 3) {
+		fprintf(stderr, "Usage: %s <hostname> <port>\n", argv[0]);
+		return 1;
+	}
+
+	ctx = ustream_ssl_context_new(false);
+	uloop_init();
+	connect_client(argv[1], argv[2]);
+	uloop_run();
+
+	close(fd.fd);
+	uloop_done();
+	return 0;
+}