unix: introduce uv_socket_get_tos and uv_socket_set_tos
Signed-off-by: Juan José Arboleda <soyjuanarbol@gmail.com>
This commit is contained in:
Juan José Arboleda
parent
4839e28d50
commit
09499f580c
@ -631,6 +631,7 @@ if(LIBUV_BUILD_TESTS)
|
||||
test/test-signal-pending-on-close.c
|
||||
test/test-signal.c
|
||||
test/test-socket-buffer-size.c
|
||||
test/test-socket-tos.c
|
||||
test/test-spawn.c
|
||||
test/test-stdio-over-pipes.c
|
||||
test/test-strscpy.c
|
||||
|
||||
@ -256,6 +256,7 @@ test_run_tests_SOURCES = test/blackhole-server.c \
|
||||
test/test-signal-pending-on-close.c \
|
||||
test/test-signal.c \
|
||||
test/test-socket-buffer-size.c \
|
||||
test/test-socket-tos.c \
|
||||
test/test-spawn.c \
|
||||
test/test-stdio-over-pipes.c \
|
||||
test/test-strscpy.c \
|
||||
|
||||
@ -519,6 +519,9 @@ UV_EXTERN int uv_socketpair(int type,
|
||||
int flags0,
|
||||
int flags1);
|
||||
|
||||
UV_EXTERN int uv_socket_set_tos(const uv_handle_t* handle, int tos);
|
||||
UV_EXTERN int uv_socket_get_tos(const uv_handle_t* handle, int* tos);
|
||||
|
||||
#define UV_STREAM_FIELDS \
|
||||
/* number of bytes queued for writing */ \
|
||||
size_t write_queue_size; \
|
||||
|
||||
101
src/uv-common.c
101
src/uv-common.c
@ -1062,3 +1062,104 @@ void uv_free_interface_addresses(uv_interface_address_t* addresses,
|
||||
uv__free(addresses);
|
||||
}
|
||||
#endif /* !__MVS__ */
|
||||
|
||||
int uv_socket_get_tos(const uv_handle_t* handle, int* tos) {
|
||||
int addrlen;
|
||||
int fd;
|
||||
int level;
|
||||
int option;
|
||||
int r;
|
||||
int tos_;
|
||||
socklen_t optlen;
|
||||
struct sockaddr_storage storage;
|
||||
|
||||
addrlen = sizeof(storage);
|
||||
tos_ = 0;
|
||||
optlen = sizeof(tos_);
|
||||
|
||||
switch (handle->type) {
|
||||
case UV_TCP:
|
||||
r = uv_tcp_getsockname((uv_tcp_t*)handle, (struct sockaddr*)&storage, &addrlen);
|
||||
break;
|
||||
case UV_UDP:
|
||||
r = uv_udp_getsockname((uv_udp_t*)handle, (struct sockaddr*)&storage, &addrlen);
|
||||
break;
|
||||
default:
|
||||
return UV_EINVAL;
|
||||
}
|
||||
|
||||
if (r)
|
||||
return r;
|
||||
|
||||
r = uv_fileno(handle, &fd);
|
||||
if (r)
|
||||
return r;
|
||||
|
||||
switch (storage.ss_family) {
|
||||
case AF_INET:
|
||||
level = IPPROTO_IP;
|
||||
option = IP_TOS;
|
||||
break;
|
||||
case AF_INET6:
|
||||
level = IPPROTO_IPV6;
|
||||
option = IPV6_TCLASS;
|
||||
break;
|
||||
default:
|
||||
return UV_EAFNOSUPPORT;
|
||||
}
|
||||
|
||||
r = getsockopt(fd, level, option, &tos_, &optlen);
|
||||
if (r)
|
||||
return r;
|
||||
|
||||
*tos = tos_;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int uv_socket_set_tos(const uv_handle_t* handle, int tos) {
|
||||
int fd;
|
||||
int r;
|
||||
struct sockaddr_storage storage;
|
||||
int addrlen;
|
||||
int level;
|
||||
int option;
|
||||
|
||||
addrlen = sizeof(storage);
|
||||
|
||||
if (tos < 0 || tos > 255)
|
||||
return UV_EINVAL;
|
||||
|
||||
switch (handle->type) {
|
||||
case UV_TCP:
|
||||
r = uv_tcp_getsockname((uv_tcp_t*)handle, (struct sockaddr*)&storage, &addrlen);
|
||||
break;
|
||||
case UV_UDP:
|
||||
r = uv_udp_getsockname((uv_udp_t*)handle, (struct sockaddr*)&storage, &addrlen);
|
||||
break;
|
||||
default:
|
||||
return UV_EINVAL;
|
||||
}
|
||||
|
||||
if (r)
|
||||
return r;
|
||||
r = uv_fileno(handle, &fd);
|
||||
if (r)
|
||||
return r;
|
||||
|
||||
switch (storage.ss_family) {
|
||||
case AF_INET:
|
||||
level = IPPROTO_IP;
|
||||
option = IP_TOS;
|
||||
break;
|
||||
case AF_INET6:
|
||||
level = IPPROTO_IPV6;
|
||||
option = IPV6_TCLASS;
|
||||
break;
|
||||
default:
|
||||
return UV_EAFNOSUPPORT;
|
||||
}
|
||||
|
||||
r = setsockopt(fd, level, option, &tos, sizeof(tos));
|
||||
return UV__ERR(r);
|
||||
}
|
||||
|
||||
@ -190,6 +190,12 @@ TEST_DECLARE (udp_ipv6_only)
|
||||
TEST_DECLARE (udp_options)
|
||||
TEST_DECLARE (udp_options6)
|
||||
TEST_DECLARE (udp_no_autobind)
|
||||
TEST_DECLARE (tcp_socket_set_get_tos)
|
||||
TEST_DECLARE (tcp6_socket_set_get_tos)
|
||||
TEST_DECLARE (udp_socket_set_get_tos)
|
||||
TEST_DECLARE (udp6_socket_set_get_tos)
|
||||
TEST_DECLARE (socket_tos_invalid_handle)
|
||||
TEST_DECLARE (socket_tos_unbound_tcp)
|
||||
TEST_DECLARE (udp_open)
|
||||
TEST_DECLARE (udp_open_twice)
|
||||
TEST_DECLARE (udp_open_bound)
|
||||
@ -818,6 +824,12 @@ TASK_LIST_START
|
||||
TEST_ENTRY (udp_options)
|
||||
TEST_ENTRY (udp_options6)
|
||||
TEST_ENTRY (udp_no_autobind)
|
||||
TEST_ENTRY (tcp_socket_set_get_tos)
|
||||
TEST_ENTRY (tcp6_socket_set_get_tos)
|
||||
TEST_ENTRY (udp_socket_set_get_tos)
|
||||
TEST_ENTRY (udp6_socket_set_get_tos)
|
||||
TEST_ENTRY (socket_tos_invalid_handle)
|
||||
TEST_ENTRY (socket_tos_unbound_tcp)
|
||||
TEST_ENTRY (udp_mmsg)
|
||||
TEST_ENTRY (udp_multicast_interface)
|
||||
TEST_ENTRY (udp_multicast_interface6)
|
||||
|
||||
325
test/test-socket-tos.c
Normal file
325
test/test-socket-tos.c
Normal file
@ -0,0 +1,325 @@
|
||||
/* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
|
||||
*
|
||||
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
* of this software and associated documentation files (the "Software"), to
|
||||
* deal in the Software without restriction, including without limitation the
|
||||
* rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
|
||||
* sell copies of the Software, and to permit persons to whom the Software is
|
||||
* furnished to do so, subject to the following conditions:
|
||||
*
|
||||
* The above copyright notice and this permission notice shall be included in
|
||||
* all copies or substantial portions of the Software.
|
||||
*
|
||||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
||||
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
|
||||
* IN THE SOFTWARE.
|
||||
*/
|
||||
|
||||
#include "uv.h"
|
||||
#include "task.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
|
||||
TEST_IMPL(tcp_socket_set_get_tos) {
|
||||
uv_loop_t* loop;
|
||||
uv_tcp_t tcp;
|
||||
struct sockaddr_in addr;
|
||||
int r;
|
||||
int tos_in;
|
||||
int tos_out;
|
||||
|
||||
loop = uv_default_loop();
|
||||
|
||||
r = uv_tcp_init(loop, &tcp);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_ip4_addr("0.0.0.0", TEST_PORT, &addr);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_tcp_bind(&tcp, (const struct sockaddr*) &addr, 0);
|
||||
ASSERT_OK(r);
|
||||
|
||||
/* Test valid TOS values */
|
||||
tos_in = 0;
|
||||
r = uv_socket_set_tos((uv_handle_t*) &tcp, tos_in);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_socket_get_tos((uv_handle_t*) &tcp, &tos_out);
|
||||
ASSERT_OK(r);
|
||||
ASSERT_EQ(tos_out, tos_in);
|
||||
|
||||
tos_in = 128;
|
||||
r = uv_socket_set_tos((uv_handle_t*) &tcp, tos_in);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_socket_get_tos((uv_handle_t*) &tcp, &tos_out);
|
||||
ASSERT_OK(r);
|
||||
ASSERT_EQ(tos_out, tos_in);
|
||||
|
||||
tos_in = 255;
|
||||
r = uv_socket_set_tos((uv_handle_t*) &tcp, tos_in);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_socket_get_tos((uv_handle_t*) &tcp, &tos_out);
|
||||
ASSERT_OK(r);
|
||||
ASSERT_EQ(tos_out, tos_in);
|
||||
|
||||
/* Test invalid TOS values */
|
||||
r = uv_socket_set_tos((uv_handle_t*) &tcp, -1);
|
||||
ASSERT_EQ(r, UV_EINVAL);
|
||||
|
||||
r = uv_socket_set_tos((uv_handle_t*) &tcp, 256);
|
||||
ASSERT_EQ(r, UV_EINVAL);
|
||||
|
||||
r = uv_socket_set_tos((uv_handle_t*) &tcp, 1000);
|
||||
ASSERT_EQ(r, UV_EINVAL);
|
||||
|
||||
uv_close((uv_handle_t*) &tcp, NULL);
|
||||
r = uv_run(loop, UV_RUN_DEFAULT);
|
||||
ASSERT_OK(r);
|
||||
|
||||
MAKE_VALGRIND_HAPPY(loop);
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
TEST_IMPL(tcp6_socket_set_get_tos) {
|
||||
uv_loop_t* loop;
|
||||
uv_tcp_t tcp;
|
||||
struct sockaddr_in6 addr;
|
||||
int r;
|
||||
int tos_in;
|
||||
int tos_out;
|
||||
|
||||
loop = uv_default_loop();
|
||||
|
||||
r = uv_tcp_init(loop, &tcp);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_ip6_addr("::1", TEST_PORT, &addr);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_tcp_bind(&tcp, (const struct sockaddr*) &addr, 0);
|
||||
ASSERT_OK(r);
|
||||
|
||||
/* Test valid TOS values */
|
||||
tos_in = 0;
|
||||
r = uv_socket_set_tos((uv_handle_t*) &tcp, tos_in);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_socket_get_tos((uv_handle_t*) &tcp, &tos_out);
|
||||
ASSERT_OK(r);
|
||||
ASSERT_EQ(tos_out, tos_in);
|
||||
|
||||
tos_in = 64;
|
||||
r = uv_socket_set_tos((uv_handle_t*) &tcp, tos_in);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_socket_get_tos((uv_handle_t*) &tcp, &tos_out);
|
||||
ASSERT_OK(r);
|
||||
ASSERT_EQ(tos_out, tos_in);
|
||||
|
||||
tos_in = 255;
|
||||
r = uv_socket_set_tos((uv_handle_t*) &tcp, tos_in);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_socket_get_tos((uv_handle_t*) &tcp, &tos_out);
|
||||
ASSERT_OK(r);
|
||||
ASSERT_EQ(tos_out, tos_in);
|
||||
|
||||
uv_close((uv_handle_t*) &tcp, NULL);
|
||||
r = uv_run(loop, UV_RUN_DEFAULT);
|
||||
ASSERT_OK(r);
|
||||
|
||||
MAKE_VALGRIND_HAPPY(loop);
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
TEST_IMPL(udp_socket_set_get_tos) {
|
||||
uv_loop_t* loop;
|
||||
uv_udp_t udp;
|
||||
struct sockaddr_in addr;
|
||||
int r;
|
||||
int tos_in;
|
||||
int tos_out;
|
||||
|
||||
loop = uv_default_loop();
|
||||
|
||||
r = uv_udp_init(loop, &udp);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_ip4_addr("0.0.0.0", TEST_PORT, &addr);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_udp_bind(&udp, (const struct sockaddr*) &addr, 0);
|
||||
ASSERT_OK(r);
|
||||
|
||||
/* Test valid TOS values */
|
||||
tos_in = 0;
|
||||
r = uv_socket_set_tos((uv_handle_t*) &udp, tos_in);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_socket_get_tos((uv_handle_t*) &udp, &tos_out);
|
||||
ASSERT_OK(r);
|
||||
ASSERT_EQ(tos_out, tos_in);
|
||||
|
||||
tos_in = 32;
|
||||
r = uv_socket_set_tos((uv_handle_t*) &udp, tos_in);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_socket_get_tos((uv_handle_t*) &udp, &tos_out);
|
||||
ASSERT_OK(r);
|
||||
ASSERT_EQ(tos_out, tos_in);
|
||||
|
||||
tos_in = 255;
|
||||
r = uv_socket_set_tos((uv_handle_t*) &udp, tos_in);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_socket_get_tos((uv_handle_t*) &udp, &tos_out);
|
||||
ASSERT_OK(r);
|
||||
ASSERT_EQ(tos_out, tos_in);
|
||||
|
||||
/* Test invalid TOS values */
|
||||
r = uv_socket_set_tos((uv_handle_t*) &udp, -1);
|
||||
ASSERT_EQ(r, UV_EINVAL);
|
||||
|
||||
r = uv_socket_set_tos((uv_handle_t*) &udp, 256);
|
||||
ASSERT_EQ(r, UV_EINVAL);
|
||||
|
||||
uv_close((uv_handle_t*) &udp, NULL);
|
||||
r = uv_run(loop, UV_RUN_DEFAULT);
|
||||
ASSERT_OK(r);
|
||||
|
||||
MAKE_VALGRIND_HAPPY(loop);
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
TEST_IMPL(udp6_socket_set_get_tos) {
|
||||
uv_loop_t* loop;
|
||||
uv_udp_t udp;
|
||||
struct sockaddr_in6 addr;
|
||||
int r;
|
||||
int tos_in;
|
||||
int tos_out;
|
||||
|
||||
loop = uv_default_loop();
|
||||
|
||||
r = uv_udp_init(loop, &udp);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_ip6_addr("::1", TEST_PORT, &addr);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_udp_bind(&udp, (const struct sockaddr*) &addr, 0);
|
||||
ASSERT_OK(r);
|
||||
|
||||
/* Test valid TOS values */
|
||||
tos_in = 0;
|
||||
r = uv_socket_set_tos((uv_handle_t*) &udp, tos_in);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_socket_get_tos((uv_handle_t*) &udp, &tos_out);
|
||||
ASSERT_OK(r);
|
||||
ASSERT_EQ(tos_out, tos_in);
|
||||
|
||||
tos_in = 96;
|
||||
r = uv_socket_set_tos((uv_handle_t*) &udp, tos_in);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_socket_get_tos((uv_handle_t*) &udp, &tos_out);
|
||||
ASSERT_OK(r);
|
||||
ASSERT_EQ(tos_out, tos_in);
|
||||
|
||||
tos_in = 255;
|
||||
r = uv_socket_set_tos((uv_handle_t*) &udp, tos_in);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_socket_get_tos((uv_handle_t*) &udp, &tos_out);
|
||||
ASSERT_OK(r);
|
||||
ASSERT_EQ(tos_out, tos_in);
|
||||
|
||||
uv_close((uv_handle_t*) &udp, NULL);
|
||||
r = uv_run(loop, UV_RUN_DEFAULT);
|
||||
ASSERT_OK(r);
|
||||
|
||||
MAKE_VALGRIND_HAPPY(loop);
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
TEST_IMPL(socket_tos_invalid_handle) {
|
||||
uv_loop_t* loop;
|
||||
uv_timer_t timer;
|
||||
uv_pipe_t pipe;
|
||||
int r;
|
||||
int tos;
|
||||
|
||||
loop = uv_default_loop();
|
||||
|
||||
/* Test with invalid handle types */
|
||||
r = uv_timer_init(loop, &timer);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_socket_set_tos((uv_handle_t*) &timer, 64);
|
||||
ASSERT_EQ(r, UV_EINVAL);
|
||||
|
||||
r = uv_socket_get_tos((uv_handle_t*) &timer, &tos);
|
||||
ASSERT_EQ(r, UV_EINVAL);
|
||||
|
||||
uv_close((uv_handle_t*) &timer, NULL);
|
||||
|
||||
/* Test with pipe handle */
|
||||
r = uv_pipe_init(loop, &pipe, 0);
|
||||
ASSERT_OK(r);
|
||||
|
||||
r = uv_socket_set_tos((uv_handle_t*) &pipe, 64);
|
||||
ASSERT_EQ(r, UV_EINVAL);
|
||||
|
||||
r = uv_socket_get_tos((uv_handle_t*) &pipe, &tos);
|
||||
ASSERT_EQ(r, UV_EINVAL);
|
||||
|
||||
uv_close((uv_handle_t*) &pipe, NULL);
|
||||
|
||||
r = uv_run(loop, UV_RUN_DEFAULT);
|
||||
ASSERT_OK(r);
|
||||
|
||||
MAKE_VALGRIND_HAPPY(loop);
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
TEST_IMPL(socket_tos_unbound_tcp) {
|
||||
uv_loop_t* loop;
|
||||
uv_tcp_t tcp;
|
||||
int r;
|
||||
int tos;
|
||||
|
||||
loop = uv_default_loop();
|
||||
|
||||
/* Test with unbound TCP socket - should fail since no address family */
|
||||
r = uv_tcp_init(loop, &tcp);
|
||||
ASSERT_OK(r);
|
||||
|
||||
/* Getting/setting TOS on unbound socket should fail */
|
||||
r = uv_socket_set_tos((uv_handle_t*) &tcp, 64);
|
||||
ASSERT_NE(r, 0);
|
||||
|
||||
r = uv_socket_get_tos((uv_handle_t*) &tcp, &tos);
|
||||
ASSERT_NE(r, 0);
|
||||
|
||||
uv_close((uv_handle_t*) &tcp, NULL);
|
||||
r = uv_run(loop, UV_RUN_DEFAULT);
|
||||
ASSERT_OK(r);
|
||||
|
||||
MAKE_VALGRIND_HAPPY(loop);
|
||||
return 0;
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user