zhangmeng
2021-12-10 5990bac28af438a914165441b3c33a370b320d16
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
111
112
113
114
115
116
117
118
//
// Copyright 2021 Staysail Systems, Inc. <info@staysail.tech>
// Copyright 2018 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 <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
 
static nni_mtx  nni_init_mtx;
static nni_list nni_init_list;
static bool     nni_inited = false;
 
extern int nni_tls_sys_init(void);
extern void nni_tls_sys_fini(void);
 
static int
nni_init_helper(void)
{
    int rv;
 
    nni_mtx_init(&nni_init_mtx);
    NNI_LIST_INIT(&nni_init_list, nni_initializer, i_node);
    nni_inited = true;
#ifdef NNG_TEST_LIB
    static bool cleanup = false;
    if (!cleanup) {
        atexit(nng_fini);
        cleanup = true;
    }
#endif
 
    if (((rv = nni_stat_sys_init()) != 0) ||
        ((rv = nni_taskq_sys_init()) != 0) ||
        ((rv = nni_reap_sys_init()) != 0) ||
        ((rv = nni_timer_sys_init()) != 0) ||
        ((rv = nni_aio_sys_init()) != 0) ||
        ((rv = nni_sock_sys_init()) != 0) ||
        ((rv = nni_listener_sys_init()) != 0) ||
        ((rv = nni_dialer_sys_init()) != 0) ||
        ((rv = nni_pipe_sys_init()) != 0) ||
        ((rv = nni_tls_sys_init()) != 0) ||
        ((rv = nni_sp_tran_sys_init()) != 0)) {
        nni_fini();
    }
 
    return (rv);
}
 
int
nni_init(void)
{
    return (nni_plat_init(nni_init_helper));
}
 
void
nni_fini(void)
{
    if (!nni_inited) {
        return;
    }
    if (!nni_list_empty(&nni_init_list)) {
        nni_initializer *init;
 
        nni_mtx_lock(&nni_init_mtx);
        while ((init = nni_list_first(&nni_init_list)) != NULL) {
            if (init->i_fini != NULL) {
                init->i_fini();
            }
            init->i_once = 0;
            nni_list_remove(&nni_init_list, init);
        }
        nni_mtx_unlock(&nni_init_mtx);
    }
    nni_sp_tran_sys_fini();
    nni_tls_sys_fini();
    nni_pipe_sys_fini();
    nni_dialer_sys_fini();
    nni_listener_sys_fini();
    nni_sock_sys_fini();
    nni_reap_drain();
    nni_aio_sys_fini();
    nni_timer_sys_fini();
    nni_taskq_sys_fini();
    nni_reap_sys_fini(); // must be before timer and aio (expire)
    nni_stat_sys_fini();
 
    nni_mtx_fini(&nni_init_mtx);
    nni_plat_fini();
    nni_inited = false;
}
 
int
nni_initialize(nni_initializer *init)
{
    int rv;
    if (init->i_once) {
        return (0);
    }
    nni_mtx_lock(&nni_init_mtx);
    if (init->i_once) {
        nni_mtx_unlock(&nni_init_mtx);
        return (0);
    }
    if ((rv = init->i_init()) == 0) {
        init->i_once = 1;
        nni_list_append(&nni_init_list, init);
    }
    nni_mtx_unlock(&nni_init_mtx);
    return (rv);
}