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
119
120
121
|
#include <sys/socket.h>
#include <unistd.h>
#include "test-common.h"
static int fds[2];
static uv_loop_t* loop;
static uv_pipe_t pair_right;
static uv_link_source_t source;
static int write_cb_called;
static int alloc_cb_called;
static int read_cb_called;
static void read_one() {
char buf[1024];
int err;
do
err = read(fds[0], buf, sizeof(buf));
while (err == -1 && errno == EINTR);
CHECK_EQ(err, 1, "read() == 1");
CHECK_EQ(buf[0], 'x', "data should match");
}
static void source_write_cb(uv_link_t* link, int status) {
CHECK_EQ(link, &source.link, "link == &source.link");
write_cb_called++;
read_one();
}
static void test_writes() {
uv_buf_t buf;
/* .write() should work */
buf = uv_buf_init("x", 1);
CHECK_EQ(uv_link_write(&source.link, &buf, 1, NULL, source_write_cb), 0,
"source.link.write() should return 0");
CHECK_EQ(uv_run(loop, UV_RUN_DEFAULT), 0, "uv_run()");
CHECK_EQ(write_cb_called, 1, "source_write_cb() must be called");
/* .try_write() should work too */
CHECK_EQ(uv_link_try_write(&source.link, &buf, 1), 1,
"source.link.try_write() should return 1");
read_one();
}
static void source_alloc_cb(uv_link_t* link,
size_t suggested_size,
uv_buf_t* buf) {
static char storage[1024];
CHECK_EQ(link, &source.link, "link == &source.link");
alloc_cb_called++;
buf->base = storage;
buf->len = sizeof(storage);
}
static void source_read_cb(uv_link_t* link,
ssize_t nread,
const uv_buf_t* buf) {
CHECK_EQ(link, &source.link, "link == &source.link");
read_cb_called++;
CHECK_EQ(nread, 1, "source_read_cb must read one byte");
CHECK_EQ(buf->base[0], 'x', "source_read_cb must read correct one byte");
CHECK_EQ(uv_link_read_stop(&source.link), 0, "source.link.read_stop()");
}
static void test_reads() {
int err;
/* alloc_cb/read_cb */
source.link.alloc_cb = source_alloc_cb;
source.link.read_cb = source_read_cb;
do
err = write(fds[0], "x", 1);
while (err == -1 && errno == EINTR);
CHECK_EQ(err, 1, "write() == 1");
CHECK_EQ(uv_link_read_start(&source.link), 0, "source.link.read_start()");
CHECK_EQ(uv_run(loop, UV_RUN_DEFAULT), 0, "uv_run()");
CHECK_EQ(alloc_cb_called, 1, "alloc_cb must be called once");
CHECK_EQ(read_cb_called, 1, "read_cb must be called once");
}
TEST_IMPL(uv_link_source_t) {
loop = uv_default_loop();
CHECK_NE(loop, NULL, "uv_default_loop()");
CHECK_EQ(socketpair(AF_UNIX, SOCK_STREAM, 0, fds), 0, "socketpair()");
CHECK_EQ(uv_pipe_init(loop, &pair_right, 0), 0, "uv_pipe_init(pair_right)");
CHECK_EQ(uv_pipe_open(&pair_right, fds[1]), 0, "uv_pipe_open(pair_right)");
CHECK_EQ(uv_link_source_init(&source, (uv_stream_t*) &pair_right), 0,
"uv_link_source_init()");
test_writes();
test_reads();
uv_link_source_close(&source);
uv_close((uv_handle_t*) &pair_right, NULL);
CHECK_EQ(close(fds[0]), 0, "close(fds[0])");
}
|