diff options
Diffstat (limited to 'src/sp/transport.c')
| -rw-r--r-- | src/sp/transport.c | 179 |
1 files changed, 179 insertions, 0 deletions
diff --git a/src/sp/transport.c b/src/sp/transport.c new file mode 100644 index 00000000..78859078 --- /dev/null +++ b/src/sp/transport.c @@ -0,0 +1,179 @@ +// +// Copyright 2019 Staysail Systems, Inc. <info@staysail.tech> +// Copyright 2018 Capitar IT Group BV <info@capitar.com> +// Copyright 2019 Devolutions <info@devolutions.net> +// +// This software is supplied under the terms of the MIT License, a +// copy of which should be located in the distribution where this +// file was obtained (LICENSE.txt). A copy of the license may also be +// found online at https://opensource.org/licenses/MIT. +// + +#include "core/nng_impl.h" + +#ifdef NNG_TRANSPORT_INPROC +#include "nng/transport/inproc/inproc.h" +#endif +#ifdef NNG_TRANSPORT_IPC +#include "nng/transport/ipc/ipc.h" +#endif +#ifdef NNG_TRANSPORT_TCP +#include "nng/transport/tcp/tcp.h" +#endif +#ifdef NNG_TRANSPORT_TLS +#include "nng/transport/tls/tls.h" +#endif +#ifdef NNG_TRANSPORT_WS +#include "nng/transport/ws/websocket.h" +#endif +#ifdef NNG_TRANSPORT_ZEROTIER +#include "nng/transport/zerotier/zerotier.h" +#endif + +#include <stdio.h> +#include <string.h> + +// For now the list of transports is hard-wired. Adding new transports +// to the system dynamically is something that might be considered later. +extern nni_tran nni_tcp_tran; +extern nni_tran nni_ipc_tran; + +typedef struct nni_transport { + nni_tran t_tran; + nni_list_node t_node; +} nni_transport; + +static nni_list nni_tran_list; +static nni_mtx nni_tran_lk; +static int nni_tran_inited; + +int +nni_tran_register(const nni_tran *tran) +{ + nni_transport *t; + int rv; + + // Its entirely possible that we are called before any sockets + // are opened. Make sure we are initialized. This has to be + // protected by a guard to prevent infinite recursion, since + // nni_init also winds up calling us. + if (!nni_tran_inited) { + nni_init(); + } + + if (tran->tran_version != NNI_TRANSPORT_VERSION) { + return (NNG_ENOTSUP); + } + + nni_mtx_lock(&nni_tran_lk); + // Check to see if the transport is already registered... + NNI_LIST_FOREACH (&nni_tran_list, t) { + if (strcmp(tran->tran_scheme, t->t_tran.tran_scheme) == 0) { + if (tran->tran_init == t->t_tran.tran_init) { + // duplicate. + nni_mtx_unlock(&nni_tran_lk); + return (0); + } + nni_mtx_unlock(&nni_tran_lk); + return (NNG_ESTATE); + } + } + if ((t = NNI_ALLOC_STRUCT(t)) == NULL) { + nni_mtx_unlock(&nni_tran_lk); + return (NNG_ENOMEM); + } + + t->t_tran = *tran; + if ((rv = t->t_tran.tran_init()) != 0) { + nni_mtx_unlock(&nni_tran_lk); + NNI_FREE_STRUCT(t); + return (rv); + } + nni_list_append(&nni_tran_list, t); + nni_mtx_unlock(&nni_tran_lk); + return (0); +} + +nni_tran * +nni_tran_find(nni_url *url) +{ + // address is of the form "<scheme>://blah..." + nni_transport *t; + + nni_mtx_lock(&nni_tran_lk); + NNI_LIST_FOREACH (&nni_tran_list, t) { + if (strcmp(url->u_scheme, t->t_tran.tran_scheme) == 0) { + nni_mtx_unlock(&nni_tran_lk); + return (&t->t_tran); + } + } + nni_mtx_unlock(&nni_tran_lk); + return (NULL); +} + +// nni_tran_sys_init initializes the entire transport subsystem, including +// each individual transport. + +typedef int (*nni_tran_ctor)(void); + +// These are just the statically compiled in constructors. +// In the future we might want to support dynamic additions. +static nni_tran_ctor nni_tran_ctors[] = { +#ifdef NNG_TRANSPORT_INPROC + nng_inproc_register, +#endif +#ifdef NNG_TRANSPORT_IPC + nng_ipc_register, +#endif +#ifdef NNG_TRANSPORT_TCP + nng_tcp_register, +#endif +#ifdef NNG_TRANSPORT_TLS + nng_tls_register, +#endif +#ifdef NNG_TRANSPORT_WS + nng_ws_register, +#endif +#ifdef NNG_TRANSPORT_WSS + nng_wss_register, +#endif +#ifdef NNG_TRANSPORT_ZEROTIER + nng_zt_register, +#endif + NULL, +}; + +int +nni_tran_sys_init(void) +{ + int i; + + nni_tran_inited = 1; + NNI_LIST_INIT(&nni_tran_list, nni_transport, t_node); + nni_mtx_init(&nni_tran_lk); + + for (i = 0; nni_tran_ctors[i] != NULL; i++) { + int rv; + if ((rv = (nni_tran_ctors[i])()) != 0) { + nni_tran_sys_fini(); + return (rv); + } + } + return (0); +} + +// nni_tran_sys_fini finalizes the entire transport system, including all +// transports. +void +nni_tran_sys_fini(void) +{ + nni_transport *t; + + while ((t = nni_list_first(&nni_tran_list)) != NULL) { + nni_list_remove(&nni_tran_list, t); + t->t_tran.tran_fini(); + NNI_FREE_STRUCT(t); + } + nni_mtx_fini(&nni_tran_lk); + nni_tran_inited = 0; +} |