libuv-svace-build
160 строк · 4.3 Кб
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#include <stdio.h>
26#include <stdlib.h>
27#include <string.h>
28
29
30static int udp_options_test(const struct sockaddr* addr) {
31static int invalid_ttls[] = { -1, 0, 256 };
32uv_loop_t* loop;
33uv_udp_t h;
34int i, r;
35
36loop = uv_default_loop();
37
38r = uv_udp_init(loop, &h);
39ASSERT_OK(r);
40
41uv_unref((uv_handle_t*)&h); /* don't keep the loop alive */
42
43r = uv_udp_bind(&h, addr, 0);
44ASSERT_OK(r);
45
46r = uv_udp_set_broadcast(&h, 1);
47r |= uv_udp_set_broadcast(&h, 1);
48r |= uv_udp_set_broadcast(&h, 0);
49r |= uv_udp_set_broadcast(&h, 0);
50ASSERT_OK(r);
51
52/* values 1-255 should work */
53for (i = 1; i <= 255; i++) {
54r = uv_udp_set_ttl(&h, i);
55#if defined(__MVS__)
56if (addr->sa_family == AF_INET6)
57ASSERT_OK(r);
58else
59ASSERT_EQ(r, UV_ENOTSUP);
60#else
61ASSERT_OK(r);
62#endif
63}
64
65for (i = 0; i < (int) ARRAY_SIZE(invalid_ttls); i++) {
66r = uv_udp_set_ttl(&h, invalid_ttls[i]);
67ASSERT_EQ(r, UV_EINVAL);
68}
69
70r = uv_udp_set_multicast_loop(&h, 1);
71r |= uv_udp_set_multicast_loop(&h, 1);
72r |= uv_udp_set_multicast_loop(&h, 0);
73r |= uv_udp_set_multicast_loop(&h, 0);
74ASSERT_OK(r);
75
76/* values 0-255 should work */
77for (i = 0; i <= 255; i++) {
78r = uv_udp_set_multicast_ttl(&h, i);
79ASSERT_OK(r);
80}
81
82/* anything >255 should fail */
83r = uv_udp_set_multicast_ttl(&h, 256);
84ASSERT_EQ(r, UV_EINVAL);
85/* don't test ttl=-1, it's a valid value on some platforms */
86
87r = uv_run(loop, UV_RUN_DEFAULT);
88ASSERT_OK(r);
89
90MAKE_VALGRIND_HAPPY(loop);
91return 0;
92}
93
94
95TEST_IMPL(udp_options) {
96struct sockaddr_in addr;
97
98ASSERT_OK(uv_ip4_addr("0.0.0.0", TEST_PORT, &addr));
99return udp_options_test((const struct sockaddr*) &addr);
100}
101
102
103TEST_IMPL(udp_options6) {
104struct sockaddr_in6 addr;
105
106if (!can_ipv6())
107RETURN_SKIP("IPv6 not supported");
108
109ASSERT_OK(uv_ip6_addr("::", TEST_PORT, &addr));
110return udp_options_test((const struct sockaddr*) &addr);
111}
112
113
114TEST_IMPL(udp_no_autobind) {
115uv_loop_t* loop;
116uv_udp_t h;
117uv_udp_t h2;
118
119loop = uv_default_loop();
120
121/* Test a lazy initialized socket. */
122ASSERT_OK(uv_udp_init(loop, &h));
123ASSERT_EQ(UV_EBADF, uv_udp_set_multicast_ttl(&h, 32));
124ASSERT_EQ(UV_EBADF, uv_udp_set_broadcast(&h, 1));
125#if defined(__MVS__)
126ASSERT_EQ(UV_ENOTSUP, uv_udp_set_ttl(&h, 1));
127#else
128ASSERT_EQ(UV_EBADF, uv_udp_set_ttl(&h, 1));
129#endif
130ASSERT_EQ(UV_EBADF, uv_udp_set_multicast_loop(&h, 1));
131/* TODO(gengjiawen): Fix test on QEMU. */
132#if defined(__QEMU__)
133RETURN_SKIP("Test does not currently work in QEMU");
134#endif
135ASSERT_EQ(UV_EBADF, uv_udp_set_multicast_interface(&h, "0.0.0.0"));
136
137uv_close((uv_handle_t*) &h, NULL);
138
139/* Test a non-lazily initialized socket. */
140ASSERT_OK(uv_udp_init_ex(loop, &h2, AF_INET | UV_UDP_RECVMMSG));
141ASSERT_OK(uv_udp_set_multicast_ttl(&h2, 32));
142ASSERT_OK(uv_udp_set_broadcast(&h2, 1));
143
144#if defined(__MVS__)
145/* zOS only supports setting ttl for IPv6 sockets. */
146ASSERT_EQ(UV_ENOTSUP, uv_udp_set_ttl(&h2, 1));
147#else
148ASSERT_OK(uv_udp_set_ttl(&h2, 1));
149#endif
150
151ASSERT_OK(uv_udp_set_multicast_loop(&h2, 1));
152ASSERT_OK(uv_udp_set_multicast_interface(&h2, "0.0.0.0"));
153
154uv_close((uv_handle_t*) &h2, NULL);
155
156ASSERT_OK(uv_run(loop, UV_RUN_DEFAULT));
157
158MAKE_VALGRIND_HAPPY(loop);
159return 0;
160}
161