blob: 2f871f7379dd809b56541fb617050658c4d398fd (
plain)
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
|
//
// Copyright 2017 Garrett D'Amore <garrett@damore.org>
//
// 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 <string.h>
#include "core/nng_impl.h"
#define NNI_AIO_WAKE (1<<0)
int
nni_aio_init(nni_aio *aio, nni_cb cb, void *arg)
{
int rv;
if (cb == NULL) {
cb = (nni_cb) nni_aio_wake;
arg = aio;
}
memset(aio, 0, sizeof (*aio));
if ((rv = nni_mtx_init(&aio->a_lk)) != 0) {
return (rv);
}
if ((rv = nni_cv_init(&aio->a_cv, &aio->a_lk)) != 0) {
nni_mtx_fini(&aio->a_lk);
return (rv);
}
aio->a_cb = cb;
aio->a_cbarg = arg;
aio->a_expire = NNI_TIME_NEVER;
nni_taskq_ent_init(&aio->a_tqe, cb, arg);
return (0);
}
void
nni_aio_fini(nni_aio *aio)
{
nni_taskq_cancel(&aio->a_tqe);
nni_cv_fini(&aio->a_cv);
nni_mtx_fini(&aio->a_lk);
}
int
nni_aio_result(nni_aio *aio)
{
return (aio->a_result);
}
size_t
nni_aio_count(nni_aio *aio)
{
return (aio->a_count);
}
void
nni_aio_wake(nni_aio *aio)
{
nni_mtx_lock(&aio->a_lk);
aio->a_flags |= NNI_AIO_WAKE;
nni_cv_wake(&aio->a_cv);
nni_mtx_unlock(&aio->a_lk);
}
void
nni_aio_wait(nni_aio *aio)
{
nni_mtx_lock(&aio->a_lk);
while ((aio->a_flags & NNI_AIO_WAKE) == 0) {
nni_cv_wait(&aio->a_cv);
}
nni_mtx_unlock(&aio->a_lk);
}
// I/O provider related functions.
void
nni_aio_finish(nni_aio *aio, int result, size_t count)
{
nni_cb cb;
void *arg;
nni_mtx_lock(&aio->a_lk);
aio->a_result = result;
aio->a_count = count;
cb = aio->a_cb;
arg = aio->a_cbarg;
nni_cv_wake(&aio->a_cv);
nni_mtx_unlock(&aio->a_lk);
nni_taskq_dispatch(NULL, &aio->a_tqe);
}
|
