1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
//
// Copyright 2017 Garrett D'Amore <garrett@damore.org>
// Copyright 2017 Capitar IT Group BV <info@capitar.com>
//
// 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"
#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_inproc_tran;
extern nni_tran nni_tcp_tran;
extern nni_tran nni_ipc_tran;
typedef struct nni_transport {
nni_tran t_tran;
char t_prefix[16]; // e.g. "tcp://" or "tls+tcp://"
nni_list_node t_node;
} nni_transport;
static nni_list nni_tran_list;
static nni_mtx nni_tran_lk;
int
nni_tran_register(const nni_tran *tran)
{
nni_transport *t;
int rv;
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) {
nni_mtx_unlock(&nni_tran_lk);
return (NNG_ESTATE);
}
}
if ((t = NNI_ALLOC_STRUCT(t)) == NULL) {
return (NNG_ENOMEM);
}
t->t_tran = *tran;
(void) snprintf(
t->t_prefix, sizeof(t->t_prefix), "%s://", tran->tran_scheme);
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(const char *addr)
{
// address is of the form "<scheme>://blah..."
nni_transport *t;
nni_mtx_lock(&nni_tran_lk);
NNI_LIST_FOREACH (&nni_tran_list, t) {
if (strncmp(addr, t->t_prefix, strlen(t->t_prefix)) == 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.
int
nni_tran_sys_init(void)
{
int rv;
NNI_LIST_INIT(&nni_tran_list, nni_transport, t_node);
if (((rv = nni_mtx_init(&nni_tran_lk)) != 0) ||
((rv = nni_tran_register(&nni_inproc_tran)) != 0) ||
((rv = nni_tran_register(&nni_ipc_tran)) != 0) ||
((rv = nni_tran_register(&nni_tcp_tran)) != 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);
}
|
