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