xref: /libuv/test/test-tcp-read-stop.c (revision 011a1ac1)
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 
25 static uv_timer_t timer_handle;
26 static uv_tcp_t tcp_handle;
27 static uv_write_t write_req;
28 
29 
fail_cb(void)30 static void fail_cb(void) {
31   ASSERT(0 && "fail_cb called");
32 }
33 
34 
write_cb(uv_write_t * req,int status)35 static void write_cb(uv_write_t* req, int status) {
36   uv_close((uv_handle_t*) &timer_handle, NULL);
37   uv_close((uv_handle_t*) &tcp_handle, NULL);
38 }
39 
40 
timer_cb(uv_timer_t * handle)41 static void timer_cb(uv_timer_t* handle) {
42   uv_buf_t buf = uv_buf_init("PING", 4);
43   ASSERT_OK(uv_write(&write_req,
44                      (uv_stream_t*) &tcp_handle,
45                      &buf,
46                      1,
47                      write_cb));
48   ASSERT_OK(uv_read_stop((uv_stream_t*) &tcp_handle));
49 }
50 
51 
connect_cb(uv_connect_t * req,int status)52 static void connect_cb(uv_connect_t* req, int status) {
53   ASSERT_OK(status);
54   ASSERT_OK(uv_timer_start(&timer_handle, timer_cb, 50, 0));
55   ASSERT_OK(uv_read_start((uv_stream_t*) &tcp_handle,
56                           (uv_alloc_cb) fail_cb,
57                           (uv_read_cb) fail_cb));
58 }
59 
60 
TEST_IMPL(tcp_read_stop)61 TEST_IMPL(tcp_read_stop) {
62   uv_connect_t connect_req;
63   struct sockaddr_in addr;
64 
65   ASSERT_OK(uv_ip4_addr("127.0.0.1", TEST_PORT, &addr));
66   ASSERT_OK(uv_timer_init(uv_default_loop(), &timer_handle));
67   ASSERT_OK(uv_tcp_init(uv_default_loop(), &tcp_handle));
68   ASSERT_OK(uv_tcp_connect(&connect_req,
69                            &tcp_handle,
70                            (const struct sockaddr*) &addr,
71                            connect_cb));
72   ASSERT_OK(uv_run(uv_default_loop(), UV_RUN_DEFAULT));
73   MAKE_VALGRIND_HAPPY(uv_default_loop());
74 
75   return 0;
76 }
77