10e552da7Schristos /* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
20e552da7Schristos *
30e552da7Schristos * Permission is hereby granted, free of charge, to any person obtaining a copy
40e552da7Schristos * of this software and associated documentation files (the "Software"), to
50e552da7Schristos * deal in the Software without restriction, including without limitation the
60e552da7Schristos * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
70e552da7Schristos * sell copies of the Software, and to permit persons to whom the Software is
80e552da7Schristos * furnished to do so, subject to the following conditions:
90e552da7Schristos *
100e552da7Schristos * The above copyright notice and this permission notice shall be included in
110e552da7Schristos * all copies or substantial portions of the Software.
120e552da7Schristos *
130e552da7Schristos * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
140e552da7Schristos * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
150e552da7Schristos * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
160e552da7Schristos * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
170e552da7Schristos * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
180e552da7Schristos * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
190e552da7Schristos * IN THE SOFTWARE.
200e552da7Schristos */
210e552da7Schristos
220e552da7Schristos #include "uv.h"
230e552da7Schristos #include "task.h"
240e552da7Schristos #include <stdio.h>
250e552da7Schristos #include <stdlib.h>
260e552da7Schristos
270e552da7Schristos
28*5f2f4271Schristos static int connect_cb_called = 0;
290e552da7Schristos static int close_cb_called = 0;
300e552da7Schristos
310e552da7Schristos
close_cb(uv_handle_t * handle)320e552da7Schristos static void close_cb(uv_handle_t* handle) {
33*5f2f4271Schristos ASSERT_NOT_NULL(handle);
340e552da7Schristos close_cb_called++;
350e552da7Schristos }
360e552da7Schristos
370e552da7Schristos
connect_cb(uv_connect_t * req,int status)38*5f2f4271Schristos static void connect_cb(uv_connect_t* req, int status) {
39*5f2f4271Schristos ASSERT(status == UV_EADDRINUSE);
40*5f2f4271Schristos uv_close((uv_handle_t*) req->handle, close_cb);
41*5f2f4271Schristos connect_cb_called++;
42*5f2f4271Schristos }
43*5f2f4271Schristos
44*5f2f4271Schristos
TEST_IMPL(tcp_bind_error_addrinuse_connect)45*5f2f4271Schristos TEST_IMPL(tcp_bind_error_addrinuse_connect) {
46*5f2f4271Schristos struct sockaddr_in addr;
47*5f2f4271Schristos int addrlen;
48*5f2f4271Schristos uv_connect_t req;
49*5f2f4271Schristos uv_tcp_t conn;
50*5f2f4271Schristos
51*5f2f4271Schristos /* 127.0.0.1:<TEST_PORT> is already taken by tcp4_echo_server running in
52*5f2f4271Schristos * another process. uv_tcp_bind() and uv_tcp_connect() should still succeed
53*5f2f4271Schristos * (greatest common denominator across platforms) but the connect callback
54*5f2f4271Schristos * should receive an UV_EADDRINUSE error.
55*5f2f4271Schristos */
56*5f2f4271Schristos ASSERT(0 == uv_tcp_init(uv_default_loop(), &conn));
57*5f2f4271Schristos ASSERT(0 == uv_ip4_addr("127.0.0.1", TEST_PORT, &addr));
58*5f2f4271Schristos ASSERT(0 == uv_tcp_bind(&conn, (const struct sockaddr*) &addr, 0));
59*5f2f4271Schristos
60*5f2f4271Schristos ASSERT(0 == uv_ip4_addr("127.0.0.1", TEST_PORT + 1, &addr));
61*5f2f4271Schristos ASSERT(0 == uv_tcp_connect(&req,
62*5f2f4271Schristos &conn,
63*5f2f4271Schristos (const struct sockaddr*) &addr,
64*5f2f4271Schristos connect_cb));
65*5f2f4271Schristos
66*5f2f4271Schristos addrlen = sizeof(addr);
67*5f2f4271Schristos ASSERT(UV_EADDRINUSE == uv_tcp_getsockname(&conn,
68*5f2f4271Schristos (struct sockaddr*) &addr,
69*5f2f4271Schristos &addrlen));
70*5f2f4271Schristos
71*5f2f4271Schristos ASSERT(0 == uv_run(uv_default_loop(), UV_RUN_DEFAULT));
72*5f2f4271Schristos ASSERT(connect_cb_called == 1);
73*5f2f4271Schristos ASSERT(close_cb_called == 1);
74*5f2f4271Schristos
75*5f2f4271Schristos MAKE_VALGRIND_HAPPY();
76*5f2f4271Schristos return 0;
77*5f2f4271Schristos }
78*5f2f4271Schristos
79*5f2f4271Schristos
TEST_IMPL(tcp_bind_error_addrinuse_listen)80*5f2f4271Schristos TEST_IMPL(tcp_bind_error_addrinuse_listen) {
810e552da7Schristos struct sockaddr_in addr;
820e552da7Schristos uv_tcp_t server1, server2;
830e552da7Schristos int r;
840e552da7Schristos
850e552da7Schristos ASSERT(0 == uv_ip4_addr("0.0.0.0", TEST_PORT, &addr));
860e552da7Schristos r = uv_tcp_init(uv_default_loop(), &server1);
870e552da7Schristos ASSERT(r == 0);
880e552da7Schristos r = uv_tcp_bind(&server1, (const struct sockaddr*) &addr, 0);
890e552da7Schristos ASSERT(r == 0);
900e552da7Schristos
910e552da7Schristos r = uv_tcp_init(uv_default_loop(), &server2);
920e552da7Schristos ASSERT(r == 0);
930e552da7Schristos r = uv_tcp_bind(&server2, (const struct sockaddr*) &addr, 0);
940e552da7Schristos ASSERT(r == 0);
950e552da7Schristos
960e552da7Schristos r = uv_listen((uv_stream_t*)&server1, 128, NULL);
970e552da7Schristos ASSERT(r == 0);
980e552da7Schristos r = uv_listen((uv_stream_t*)&server2, 128, NULL);
990e552da7Schristos ASSERT(r == UV_EADDRINUSE);
1000e552da7Schristos
1010e552da7Schristos uv_close((uv_handle_t*)&server1, close_cb);
1020e552da7Schristos uv_close((uv_handle_t*)&server2, close_cb);
1030e552da7Schristos
1040e552da7Schristos uv_run(uv_default_loop(), UV_RUN_DEFAULT);
1050e552da7Schristos
1060e552da7Schristos ASSERT(close_cb_called == 2);
1070e552da7Schristos
1080e552da7Schristos MAKE_VALGRIND_HAPPY();
1090e552da7Schristos return 0;
1100e552da7Schristos }
1110e552da7Schristos
1120e552da7Schristos
TEST_IMPL(tcp_bind_error_addrnotavail_1)1130e552da7Schristos TEST_IMPL(tcp_bind_error_addrnotavail_1) {
1140e552da7Schristos struct sockaddr_in addr;
1150e552da7Schristos uv_tcp_t server;
1160e552da7Schristos int r;
1170e552da7Schristos
1180e552da7Schristos ASSERT(0 == uv_ip4_addr("127.255.255.255", TEST_PORT, &addr));
1190e552da7Schristos
1200e552da7Schristos r = uv_tcp_init(uv_default_loop(), &server);
1210e552da7Schristos ASSERT(r == 0);
1220e552da7Schristos
1230e552da7Schristos /* It seems that Linux is broken here - bind succeeds. */
1240e552da7Schristos r = uv_tcp_bind(&server, (const struct sockaddr*) &addr, 0);
1250e552da7Schristos ASSERT(r == 0 || r == UV_EADDRNOTAVAIL);
1260e552da7Schristos
1270e552da7Schristos uv_close((uv_handle_t*)&server, close_cb);
1280e552da7Schristos
1290e552da7Schristos uv_run(uv_default_loop(), UV_RUN_DEFAULT);
1300e552da7Schristos
1310e552da7Schristos ASSERT(close_cb_called == 1);
1320e552da7Schristos
1330e552da7Schristos MAKE_VALGRIND_HAPPY();
1340e552da7Schristos return 0;
1350e552da7Schristos }
1360e552da7Schristos
1370e552da7Schristos
TEST_IMPL(tcp_bind_error_addrnotavail_2)1380e552da7Schristos TEST_IMPL(tcp_bind_error_addrnotavail_2) {
1390e552da7Schristos struct sockaddr_in addr;
1400e552da7Schristos uv_tcp_t server;
1410e552da7Schristos int r;
1420e552da7Schristos
1430e552da7Schristos ASSERT(0 == uv_ip4_addr("4.4.4.4", TEST_PORT, &addr));
1440e552da7Schristos
1450e552da7Schristos r = uv_tcp_init(uv_default_loop(), &server);
1460e552da7Schristos ASSERT(r == 0);
1470e552da7Schristos r = uv_tcp_bind(&server, (const struct sockaddr*) &addr, 0);
1480e552da7Schristos ASSERT(r == UV_EADDRNOTAVAIL);
1490e552da7Schristos
1500e552da7Schristos uv_close((uv_handle_t*)&server, close_cb);
1510e552da7Schristos
1520e552da7Schristos uv_run(uv_default_loop(), UV_RUN_DEFAULT);
1530e552da7Schristos
1540e552da7Schristos ASSERT(close_cb_called == 1);
1550e552da7Schristos
1560e552da7Schristos MAKE_VALGRIND_HAPPY();
1570e552da7Schristos return 0;
1580e552da7Schristos }
1590e552da7Schristos
1600e552da7Schristos
TEST_IMPL(tcp_bind_error_fault)1610e552da7Schristos TEST_IMPL(tcp_bind_error_fault) {
1620e552da7Schristos char garbage[] =
1630e552da7Schristos "blah blah blah blah blah blah blah blah blah blah blah blah";
1640e552da7Schristos struct sockaddr_in* garbage_addr;
1650e552da7Schristos uv_tcp_t server;
1660e552da7Schristos int r;
1670e552da7Schristos
1680e552da7Schristos garbage_addr = (struct sockaddr_in*) &garbage;
1690e552da7Schristos
1700e552da7Schristos r = uv_tcp_init(uv_default_loop(), &server);
1710e552da7Schristos ASSERT(r == 0);
1720e552da7Schristos r = uv_tcp_bind(&server, (const struct sockaddr*) garbage_addr, 0);
1730e552da7Schristos ASSERT(r == UV_EINVAL);
1740e552da7Schristos
1750e552da7Schristos uv_close((uv_handle_t*)&server, close_cb);
1760e552da7Schristos
1770e552da7Schristos uv_run(uv_default_loop(), UV_RUN_DEFAULT);
1780e552da7Schristos
1790e552da7Schristos ASSERT(close_cb_called == 1);
1800e552da7Schristos
1810e552da7Schristos MAKE_VALGRIND_HAPPY();
1820e552da7Schristos return 0;
1830e552da7Schristos }
1840e552da7Schristos
1850e552da7Schristos /* Notes: On Linux uv_bind(server, NULL) will segfault the program. */
1860e552da7Schristos
TEST_IMPL(tcp_bind_error_inval)1870e552da7Schristos TEST_IMPL(tcp_bind_error_inval) {
1880e552da7Schristos struct sockaddr_in addr1;
1890e552da7Schristos struct sockaddr_in addr2;
1900e552da7Schristos uv_tcp_t server;
1910e552da7Schristos int r;
1920e552da7Schristos
1930e552da7Schristos ASSERT(0 == uv_ip4_addr("0.0.0.0", TEST_PORT, &addr1));
1940e552da7Schristos ASSERT(0 == uv_ip4_addr("0.0.0.0", TEST_PORT_2, &addr2));
1950e552da7Schristos
1960e552da7Schristos r = uv_tcp_init(uv_default_loop(), &server);
1970e552da7Schristos ASSERT(r == 0);
1980e552da7Schristos r = uv_tcp_bind(&server, (const struct sockaddr*) &addr1, 0);
1990e552da7Schristos ASSERT(r == 0);
2000e552da7Schristos r = uv_tcp_bind(&server, (const struct sockaddr*) &addr2, 0);
2010e552da7Schristos ASSERT(r == UV_EINVAL);
2020e552da7Schristos
2030e552da7Schristos uv_close((uv_handle_t*)&server, close_cb);
2040e552da7Schristos
2050e552da7Schristos uv_run(uv_default_loop(), UV_RUN_DEFAULT);
2060e552da7Schristos
2070e552da7Schristos ASSERT(close_cb_called == 1);
2080e552da7Schristos
2090e552da7Schristos MAKE_VALGRIND_HAPPY();
2100e552da7Schristos return 0;
2110e552da7Schristos }
2120e552da7Schristos
2130e552da7Schristos
TEST_IMPL(tcp_bind_localhost_ok)2140e552da7Schristos TEST_IMPL(tcp_bind_localhost_ok) {
2150e552da7Schristos struct sockaddr_in addr;
2160e552da7Schristos uv_tcp_t server;
2170e552da7Schristos int r;
2180e552da7Schristos
2190e552da7Schristos ASSERT(0 == uv_ip4_addr("127.0.0.1", TEST_PORT, &addr));
2200e552da7Schristos
2210e552da7Schristos r = uv_tcp_init(uv_default_loop(), &server);
2220e552da7Schristos ASSERT(r == 0);
2230e552da7Schristos r = uv_tcp_bind(&server, (const struct sockaddr*) &addr, 0);
2240e552da7Schristos ASSERT(r == 0);
2250e552da7Schristos
2260e552da7Schristos MAKE_VALGRIND_HAPPY();
2270e552da7Schristos return 0;
2280e552da7Schristos }
2290e552da7Schristos
2300e552da7Schristos
TEST_IMPL(tcp_bind_invalid_flags)2310e552da7Schristos TEST_IMPL(tcp_bind_invalid_flags) {
2320e552da7Schristos struct sockaddr_in addr;
2330e552da7Schristos uv_tcp_t server;
2340e552da7Schristos int r;
2350e552da7Schristos
2360e552da7Schristos ASSERT(0 == uv_ip4_addr("127.0.0.1", TEST_PORT, &addr));
2370e552da7Schristos
2380e552da7Schristos r = uv_tcp_init(uv_default_loop(), &server);
2390e552da7Schristos ASSERT(r == 0);
2400e552da7Schristos r = uv_tcp_bind(&server, (const struct sockaddr*) &addr, UV_TCP_IPV6ONLY);
2410e552da7Schristos ASSERT(r == UV_EINVAL);
2420e552da7Schristos
2430e552da7Schristos MAKE_VALGRIND_HAPPY();
2440e552da7Schristos return 0;
2450e552da7Schristos }
2460e552da7Schristos
2470e552da7Schristos
TEST_IMPL(tcp_listen_without_bind)2480e552da7Schristos TEST_IMPL(tcp_listen_without_bind) {
2490e552da7Schristos int r;
2500e552da7Schristos uv_tcp_t server;
2510e552da7Schristos
2520e552da7Schristos r = uv_tcp_init(uv_default_loop(), &server);
2530e552da7Schristos ASSERT(r == 0);
2540e552da7Schristos r = uv_listen((uv_stream_t*)&server, 128, NULL);
2550e552da7Schristos ASSERT(r == 0);
2560e552da7Schristos
2570e552da7Schristos MAKE_VALGRIND_HAPPY();
2580e552da7Schristos return 0;
2590e552da7Schristos }
2600e552da7Schristos
2610e552da7Schristos
TEST_IMPL(tcp_bind_writable_flags)2620e552da7Schristos TEST_IMPL(tcp_bind_writable_flags) {
2630e552da7Schristos struct sockaddr_in addr;
2640e552da7Schristos uv_tcp_t server;
2650e552da7Schristos uv_buf_t buf;
2660e552da7Schristos uv_write_t write_req;
2670e552da7Schristos uv_shutdown_t shutdown_req;
2680e552da7Schristos int r;
2690e552da7Schristos
2700e552da7Schristos ASSERT(0 == uv_ip4_addr("0.0.0.0", TEST_PORT, &addr));
2710e552da7Schristos r = uv_tcp_init(uv_default_loop(), &server);
2720e552da7Schristos ASSERT(r == 0);
2730e552da7Schristos r = uv_tcp_bind(&server, (const struct sockaddr*) &addr, 0);
2740e552da7Schristos ASSERT(r == 0);
2750e552da7Schristos r = uv_listen((uv_stream_t*)&server, 128, NULL);
2760e552da7Schristos ASSERT(r == 0);
2770e552da7Schristos
2780e552da7Schristos ASSERT(0 == uv_is_writable((uv_stream_t*) &server));
2790e552da7Schristos ASSERT(0 == uv_is_readable((uv_stream_t*) &server));
2800e552da7Schristos
2810e552da7Schristos buf = uv_buf_init("PING", 4);
2820e552da7Schristos r = uv_write(&write_req, (uv_stream_t*) &server, &buf, 1, NULL);
2830e552da7Schristos ASSERT(r == UV_EPIPE);
2840e552da7Schristos r = uv_shutdown(&shutdown_req, (uv_stream_t*) &server, NULL);
2850e552da7Schristos ASSERT(r == UV_ENOTCONN);
286*5f2f4271Schristos r = uv_read_start((uv_stream_t*) &server,
287*5f2f4271Schristos (uv_alloc_cb) abort,
288*5f2f4271Schristos (uv_read_cb) abort);
2890e552da7Schristos ASSERT(r == UV_ENOTCONN);
2900e552da7Schristos
2910e552da7Schristos uv_close((uv_handle_t*)&server, close_cb);
2920e552da7Schristos
2930e552da7Schristos uv_run(uv_default_loop(), UV_RUN_DEFAULT);
2940e552da7Schristos
2950e552da7Schristos ASSERT(close_cb_called == 1);
2960e552da7Schristos
2970e552da7Schristos MAKE_VALGRIND_HAPPY();
2980e552da7Schristos return 0;
2990e552da7Schristos }
300*5f2f4271Schristos
TEST_IMPL(tcp_bind_or_listen_error_after_close)301*5f2f4271Schristos TEST_IMPL(tcp_bind_or_listen_error_after_close) {
302*5f2f4271Schristos uv_tcp_t tcp;
303*5f2f4271Schristos struct sockaddr_in addr;
304*5f2f4271Schristos
305*5f2f4271Schristos memset(&addr, 0, sizeof(addr));
306*5f2f4271Schristos addr.sin_addr.s_addr = htonl(INADDR_ANY);
307*5f2f4271Schristos addr.sin_port = htons(9999);
308*5f2f4271Schristos addr.sin_family = AF_INET;
309*5f2f4271Schristos
310*5f2f4271Schristos ASSERT_EQ(uv_tcp_init(uv_default_loop(), &tcp), 0);
311*5f2f4271Schristos uv_close((uv_handle_t*) &tcp, NULL);
312*5f2f4271Schristos ASSERT_EQ(uv_tcp_bind(&tcp, (struct sockaddr*) &addr, 0), UV_EINVAL);
313*5f2f4271Schristos ASSERT_EQ(uv_listen((uv_stream_t*) &tcp, 5, NULL), UV_EINVAL);
314*5f2f4271Schristos ASSERT_EQ(uv_run(uv_default_loop(), UV_RUN_DEFAULT), 0);
315*5f2f4271Schristos MAKE_VALGRIND_HAPPY();
316*5f2f4271Schristos return 0;
317*5f2f4271Schristos }
318