libuv-svace-build
407 строк · 8.7 Кб
1/* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
2*
3* Permission is hereby granted, free of charge, to any person obtaining a copy
4* of this software and associated documentation files (the "Software"), to
5* deal in the Software without restriction, including without limitation the
6* rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
7* sell copies of the Software, and to permit persons to whom the Software is
8* furnished to do so, subject to the following conditions:
9*
10* The above copyright notice and this permission notice shall be included in
11* all copies or substantial portions of the Software.
12*
13* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
17* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
18* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
19* IN THE SOFTWARE.
20*/
21
22#include "uv.h"
23#include "task.h"
24#include <stdio.h>
25#include <stdlib.h>
26#include <string.h>
27
28#ifndef _WIN32
29# include <unistd.h>
30#endif
31
32static int shutdown_cb_called = 0;
33static int shutdown_requested = 0;
34static int connect_cb_called = 0;
35static int write_cb_called = 0;
36static int close_cb_called = 0;
37
38static uv_connect_t connect_req;
39static uv_shutdown_t shutdown_req;
40static uv_write_t write_req;
41static uv_timer_t tm;
42static uv_tcp_t client;
43
44
45static void startup(void) {
46#ifdef _WIN32
47struct WSAData wsa_data;
48int r = WSAStartup(MAKEWORD(2, 2), &wsa_data);
49ASSERT_OK(r);
50#endif
51}
52
53
54static uv_os_sock_t create_tcp_socket(void) {
55uv_os_sock_t sock;
56
57sock = socket(AF_INET, SOCK_STREAM, IPPROTO_IP);
58#ifdef _WIN32
59ASSERT_NE(sock, INVALID_SOCKET);
60#else
61ASSERT_GE(sock, 0);
62#endif
63
64#ifndef _WIN32
65{
66/* Allow reuse of the port. */
67int yes = 1;
68int r = setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof yes);
69ASSERT_OK(r);
70}
71#endif
72
73return sock;
74}
75
76
77static void close_socket(uv_os_sock_t sock) {
78int r;
79#ifdef _WIN32
80r = closesocket(sock);
81#else
82r = close(sock);
83#endif
84ASSERT_OK(r);
85}
86
87
88static void alloc_cb(uv_handle_t* handle,
89size_t suggested_size,
90uv_buf_t* buf) {
91static char slab[65536];
92ASSERT_LE(suggested_size, sizeof(slab));
93buf->base = slab;
94buf->len = sizeof(slab);
95}
96
97
98static void close_cb(uv_handle_t* handle) {
99ASSERT_NOT_NULL(handle);
100close_cb_called++;
101}
102
103
104static void shutdown_cb(uv_shutdown_t* req, int status) {
105ASSERT_PTR_EQ(req, &shutdown_req);
106ASSERT_OK(status);
107
108/* Now we wait for the EOF */
109shutdown_cb_called++;
110}
111
112
113static void read_cb(uv_stream_t* tcp, ssize_t nread, const uv_buf_t* buf) {
114ASSERT_NOT_NULL(tcp);
115
116if (nread >= 0) {
117ASSERT_EQ(4, nread);
118ASSERT_OK(memcmp("PING", buf->base, nread));
119}
120else {
121ASSERT_EQ(nread, UV_EOF);
122uv_close((uv_handle_t*)tcp, close_cb);
123}
124}
125
126
127static void read1_cb(uv_stream_t* tcp, ssize_t nread, const uv_buf_t* buf) {
128int i;
129ASSERT_NOT_NULL(tcp);
130
131if (nread >= 0) {
132for (i = 0; i < nread; ++i)
133ASSERT_EQ(buf->base[i], 'P');
134} else {
135ASSERT_EQ(nread, UV_EOF);
136printf("GOT EOF\n");
137uv_close((uv_handle_t*)tcp, close_cb);
138}
139}
140
141
142static void write_cb(uv_write_t* req, int status) {
143ASSERT_NOT_NULL(req);
144
145if (status) {
146fprintf(stderr, "uv_write error: %s\n", uv_strerror(status));
147ASSERT(0);
148}
149
150write_cb_called++;
151}
152
153
154static void write1_cb(uv_write_t* req, int status) {
155uv_buf_t buf;
156int r;
157
158ASSERT_NOT_NULL(req);
159if (status) {
160ASSERT(shutdown_cb_called);
161return;
162}
163
164if (shutdown_requested)
165return;
166
167buf = uv_buf_init("P", 1);
168r = uv_write(&write_req, req->handle, &buf, 1, write1_cb);
169ASSERT_OK(r);
170
171write_cb_called++;
172}
173
174
175static void timer_cb(uv_timer_t* handle) {
176int r;
177
178/* Shutdown on drain. */
179r = uv_shutdown(&shutdown_req, (uv_stream_t*) &client, shutdown_cb);
180ASSERT_OK(r);
181shutdown_requested++;
182}
183
184
185static void connect_cb(uv_connect_t* req, int status) {
186uv_buf_t buf = uv_buf_init("PING", 4);
187uv_stream_t* stream;
188int r;
189
190ASSERT_PTR_EQ(req, &connect_req);
191ASSERT_OK(status);
192
193stream = req->handle;
194connect_cb_called++;
195
196r = uv_write(&write_req, stream, &buf, 1, write_cb);
197ASSERT_OK(r);
198
199/* Shutdown on drain. */
200r = uv_shutdown(&shutdown_req, stream, shutdown_cb);
201ASSERT_OK(r);
202
203/* Start reading */
204r = uv_read_start(stream, alloc_cb, read_cb);
205ASSERT_OK(r);
206}
207
208
209static void connect1_cb(uv_connect_t* req, int status) {
210uv_buf_t buf;
211uv_stream_t* stream;
212int r;
213
214ASSERT_PTR_EQ(req, &connect_req);
215ASSERT_OK(status);
216
217stream = req->handle;
218connect_cb_called++;
219
220r = uv_timer_init(uv_default_loop(), &tm);
221ASSERT_OK(r);
222
223r = uv_timer_start(&tm, timer_cb, 2000, 0);
224ASSERT_OK(r);
225
226buf = uv_buf_init("P", 1);
227r = uv_write(&write_req, stream, &buf, 1, write1_cb);
228ASSERT_OK(r);
229
230/* Start reading */
231r = uv_read_start(stream, alloc_cb, read1_cb);
232ASSERT_OK(r);
233}
234
235
236TEST_IMPL(tcp_open) {
237struct sockaddr_in addr;
238uv_os_sock_t sock;
239int r;
240uv_tcp_t client2;
241
242ASSERT_OK(uv_ip4_addr("127.0.0.1", TEST_PORT, &addr));
243
244startup();
245sock = create_tcp_socket();
246
247r = uv_tcp_init(uv_default_loop(), &client);
248ASSERT_OK(r);
249
250r = uv_tcp_open(&client, sock);
251ASSERT_OK(r);
252
253r = uv_tcp_connect(&connect_req,
254&client,
255(const struct sockaddr*) &addr,
256connect_cb);
257ASSERT_OK(r);
258
259#ifndef _WIN32
260{
261r = uv_tcp_init(uv_default_loop(), &client2);
262ASSERT_OK(r);
263
264r = uv_tcp_open(&client2, sock);
265ASSERT_EQ(r, UV_EEXIST);
266
267uv_close((uv_handle_t*) &client2, NULL);
268}
269#else /* _WIN32 */
270(void)client2;
271#endif
272
273uv_run(uv_default_loop(), UV_RUN_DEFAULT);
274
275ASSERT_EQ(1, shutdown_cb_called);
276ASSERT_EQ(1, connect_cb_called);
277ASSERT_EQ(1, write_cb_called);
278ASSERT_EQ(1, close_cb_called);
279
280MAKE_VALGRIND_HAPPY(uv_default_loop());
281return 0;
282}
283
284
285TEST_IMPL(tcp_open_twice) {
286uv_tcp_t client;
287uv_os_sock_t sock1, sock2;
288int r;
289
290startup();
291sock1 = create_tcp_socket();
292sock2 = create_tcp_socket();
293
294r = uv_tcp_init(uv_default_loop(), &client);
295ASSERT_OK(r);
296
297r = uv_tcp_open(&client, sock1);
298ASSERT_OK(r);
299
300r = uv_tcp_open(&client, sock2);
301ASSERT_EQ(r, UV_EBUSY);
302close_socket(sock2);
303
304uv_close((uv_handle_t*) &client, NULL);
305uv_run(uv_default_loop(), UV_RUN_DEFAULT);
306
307MAKE_VALGRIND_HAPPY(uv_default_loop());
308return 0;
309}
310
311
312TEST_IMPL(tcp_open_bound) {
313struct sockaddr_in addr;
314uv_tcp_t server;
315uv_os_sock_t sock;
316
317startup();
318sock = create_tcp_socket();
319
320ASSERT_OK(uv_ip4_addr("127.0.0.1", TEST_PORT, &addr));
321
322ASSERT_OK(uv_tcp_init(uv_default_loop(), &server));
323
324ASSERT_OK(bind(sock, (struct sockaddr*) &addr, sizeof(addr)));
325
326ASSERT_OK(uv_tcp_open(&server, sock));
327
328ASSERT_OK(uv_listen((uv_stream_t*) &server, 128, NULL));
329
330MAKE_VALGRIND_HAPPY(uv_default_loop());
331return 0;
332}
333
334
335TEST_IMPL(tcp_open_connected) {
336struct sockaddr_in addr;
337uv_tcp_t client;
338uv_os_sock_t sock;
339uv_buf_t buf = uv_buf_init("PING", 4);
340
341ASSERT_OK(uv_ip4_addr("127.0.0.1", TEST_PORT, &addr));
342
343startup();
344sock = create_tcp_socket();
345
346ASSERT_OK(connect(sock, (struct sockaddr*) &addr, sizeof(addr)));
347
348ASSERT_OK(uv_tcp_init(uv_default_loop(), &client));
349
350ASSERT_OK(uv_tcp_open(&client, sock));
351
352ASSERT_OK(uv_write(&write_req,
353(uv_stream_t*) &client,
354&buf,
3551,
356write_cb));
357
358ASSERT_OK(uv_shutdown(&shutdown_req,
359(uv_stream_t*) &client,
360shutdown_cb));
361
362ASSERT_OK(uv_read_start((uv_stream_t*) &client, alloc_cb, read_cb));
363
364uv_run(uv_default_loop(), UV_RUN_DEFAULT);
365
366ASSERT_EQ(1, shutdown_cb_called);
367ASSERT_EQ(1, write_cb_called);
368ASSERT_EQ(1, close_cb_called);
369
370MAKE_VALGRIND_HAPPY(uv_default_loop());
371return 0;
372}
373
374
375TEST_IMPL(tcp_write_ready) {
376struct sockaddr_in addr;
377uv_os_sock_t sock;
378int r;
379
380ASSERT_OK(uv_ip4_addr("127.0.0.1", TEST_PORT, &addr));
381
382startup();
383sock = create_tcp_socket();
384
385r = uv_tcp_init(uv_default_loop(), &client);
386ASSERT_OK(r);
387
388r = uv_tcp_open(&client, sock);
389ASSERT_OK(r);
390
391r = uv_tcp_connect(&connect_req,
392&client,
393(const struct sockaddr*) &addr,
394connect1_cb);
395ASSERT_OK(r);
396
397uv_run(uv_default_loop(), UV_RUN_DEFAULT);
398
399ASSERT_EQ(1, shutdown_cb_called);
400ASSERT_EQ(1, shutdown_requested);
401ASSERT_EQ(1, connect_cb_called);
402ASSERT_GT(write_cb_called, 0);
403ASSERT_EQ(1, close_cb_called);
404
405MAKE_VALGRIND_HAPPY(uv_default_loop());
406return 0;
407}
408