Basic relay link connection

libs/evt-tls/sample/libuv-tls/uv_tls.c

@@ -0,0 +1,176 @@
+//%LICENSE////////////////////////////////////////////////////////////////////
+//
+// Copyright (c) 2015 Devchandra M. Leishangthem (dlmeetei at gmail dot com)
+//
+// Distributed under the MIT License (See accompanying file LICENSE)
+//
+//////////////////////////////////////////////////////////////////////////
+//
+//%///////////////////////////////////////////////////////////////////////////
+
+#include "uv_tls.h"
+#include <assert.h>
+
+static void alloc_cb(uv_handle_t *handle, size_t size, uv_buf_t *buf)
+{
+    buf->base = (char*)malloc(size);
+    memset(buf->base, 0, size);
+    buf->len = size;
+    assert(buf->base != NULL && "Memory allocation failed");
+}
+
+int uv_tls_writer(evt_tls_t *t, void *bfr, int sz) {
+    int rv = 0;
+    uv_buf_t b;
+    b.base = bfr;
+    b.len = sz;
+    uv_tls_t *uvt = t->data;
+    if(uv_is_writable((uv_stream_t*)&(uvt->skt)) ) {
+        rv = uv_try_write((uv_stream_t*)&(uvt->skt), &b, 1);
+    }
+    return rv;
+}
+
+int uv_tls_init(uv_loop_t *loop, evt_ctx_t *ctx, uv_tls_t *endpt)
+{
+    int r = 0;
+    memset( endpt, 0, sizeof *endpt);
+
+    r = uv_tcp_init(loop, &(endpt->skt));
+
+    evt_tls_t *t = evt_ctx_get_tls(ctx);
+    assert( t != NULL );
+
+    t->data = endpt;
+
+    endpt->tls = t;
+    endpt->tls_rd_cb = NULL;
+    endpt->tls_cls_cb = NULL;
+    endpt->tls_hsk_cb = NULL;
+    endpt->tls_wr_cb = NULL;
+    return r;
+}
+
+void on_tcp_read(uv_stream_t *stream, ssize_t nrd, const uv_buf_t *data)
+{
+    uv_tls_t *parent = CONTAINER_OF(stream, uv_tls_t, skt);
+    assert( parent != NULL);
+    if ( nrd <= 0 ) {
+        if( nrd == UV_EOF) {
+            uv_tls_close((uv_handle_t*)stream, (uv_close_cb)free);
+        }
+        free(data->base);
+        return;
+    }
+
+    static int is_tls = 0;
+    static int is_first = 1;
+    if (is_first) {
+        if (is_tls_stream(data->base, nrd)) {
+            evt_tls_feed_data(parent->tls, data->base, nrd);
+            is_tls = 1;
+        }
+        is_first = 0;
+    }
+    else {
+        if(is_tls) {
+            evt_tls_feed_data(parent->tls, data->base, nrd);
+        }
+    }
+    free(data->base);
+}
+
+static void on_hd_complete( evt_tls_t *t, int status)
+{
+    uv_tls_t *ut = (uv_tls_t*)t->data;
+    assert( ut != NULL && ut->tls_hsk_cb != NULL);
+    ut->tls_hsk_cb(ut, status -1);
+}
+
+
+int uv_tls_accept(uv_tls_t *t, uv_handshake_cb cb)
+{
+    int rv = 0;
+    assert( t != NULL);
+    t->tls_hsk_cb = cb;
+    evt_tls_t *tls = t->tls;
+    rv = evt_tls_accept(tls, on_hd_complete);
+    uv_read_start((uv_stream_t*)&(t->skt), alloc_cb, on_tcp_read);
+    return rv;
+}
+
+static void evt_on_rd(evt_tls_t *t, char *bfr, int sz)
+{
+    uv_buf_t data;
+    uv_tls_t *tls = (uv_tls_t*)t->data;
+
+    data.base = bfr;
+    data.len = sz;
+
+    assert(tls->tls_rd_cb != NULL);
+    tls->tls_rd_cb((uv_stream_t*)tls, sz, &data);
+}
+
+void on_close(evt_tls_t *tls, int status)
+{
+    assert(1 == status);
+    uv_tls_t *ut = (uv_tls_t*)tls->data;
+    assert( ut->tls_cls_cb != NULL);
+
+    evt_tls_free(tls);
+    if ( !uv_is_closing((uv_handle_t*)&(ut->skt)))
+        uv_close( (uv_handle_t*)&(ut->skt), ut->tls_cls_cb);
+}
+
+int uv_tls_close(uv_handle_t *strm,  uv_close_cb cb)
+{
+    uv_tls_t *t = (uv_tls_t*)strm;
+    t->tls_cls_cb = cb;
+
+    return evt_tls_close(t->tls, on_close);
+}
+
+//uv_alloc_cb is unused. This is here to mimick API similarity with libuv
+// XXX remove?
+int uv_tls_read(uv_stream_t *tls, uv_alloc_cb alloc_cb, uv_read_cb cb)
+{
+    uv_tls_t *ptr = (uv_tls_t*)tls;
+    ptr->tls_rd_cb = cb;
+    return evt_tls_read(ptr->tls, evt_on_rd);
+}
+
+static void on_hshake(evt_tls_t *etls, int status)
+{
+    assert(etls != NULL);
+    uv_tls_t *ut = (uv_tls_t*)etls->data;
+    assert(ut != NULL && ut->tls_hsk_cb != NULL);
+    ut->tls_hsk_cb(ut, status - 1);
+}
+
+int uv_tls_connect(uv_tls_t *t, uv_handshake_cb cb)
+{
+    assert( t != NULL);
+    t->tls_hsk_cb = cb;
+    evt_tls_t *evt = t->tls;
+    assert( evt != NULL);
+
+    evt_tls_connect(evt, on_hshake);
+    return uv_read_start((uv_stream_t*)&(t->skt), alloc_cb, on_tcp_read);
+}
+
+void on_evt_write(evt_tls_t *tls, int status) {
+    assert( tls != NULL);
+    uv_tls_t *ut = (uv_tls_t*)tls->data;
+    assert( ut != NULL && ut->tls_wr_cb != NULL);
+    ut->tls_wr_cb(ut, status);
+}
+
+int uv_tls_write(uv_tls_t *stream, uv_buf_t *buf, uv_tls_write_cb cb)
+{
+    assert( stream != NULL);
+    stream->tls_wr_cb = cb;
+    evt_tls_t *evt = stream->tls;
+    assert( evt != NULL);
+
+    return evt_tls_write(evt, buf->base, buf->len, on_evt_write);
+}