1 /*
2 * Copyright (c) 2023-2024 Apple Inc. All rights reserved.
3 *
4 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
5 *
6 * This file contains Original Code and/or Modifications of Original Code
7 * as defined in and that are subject to the Apple Public Source License
8 * Version 2.0 (the 'License'). You may not use this file except in
9 * compliance with the License. The rights granted to you under the License
10 * may not be used to create, or enable the creation or redistribution of,
11 * unlawful or unlicensed copies of an Apple operating system, or to
12 * circumvent, violate, or enable the circumvention or violation of, any
13 * terms of an Apple operating system software license agreement.
14 *
15 * Please obtain a copy of the License at
16 * http://www.opensource.apple.com/apsl/ and read it before using this file.
17 *
18 * The Original Code and all software distributed under the License are
19 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
20 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
22 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23 * Please see the License for the specific language governing rights and
24 * limitations under the License.
25 *
26 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
27 */
28
29 #include <darwintest.h>
30
31 #include <sys/ioctl.h>
32
33 #include <stdlib.h>
34 #include <string.h>
35 #include <strings.h>
36
37 #include "net_test_lib.h"
38
39 T_GLOBAL_META(
40 T_META_NAMESPACE("xnu.net"),
41 T_META_ASROOT(true),
42 T_META_RADAR_COMPONENT_NAME("xnu"),
43 T_META_RADAR_COMPONENT_VERSION("networking"),
44 T_META_CHECK_LEAKS(false));
45
46 static char ifname1[IF_NAMESIZE];
47
48 /**
49 ** stolen from bootp/bootplib/util.c
50 **
51 **/
52
53 #define ROUNDUP(a) \
54 ((a) > 0 ? (1 + (((a) - 1) | (sizeof(u_int32_t) - 1))) : sizeof(u_int32_t))
55
56 static int
rt_xaddrs(char * cp,const char * cplim,struct rt_addrinfo * rtinfo)57 rt_xaddrs(char * cp, const char * cplim, struct rt_addrinfo * rtinfo)
58 {
59 int i;
60 struct sockaddr * sa;
61
62 bzero(rtinfo->rti_info, sizeof(rtinfo->rti_info));
63 for (i = 0; (i < RTAX_MAX) && (cp < cplim); i++) {
64 if ((rtinfo->rti_addrs & (1 << i)) == 0) {
65 continue;
66 }
67 sa = (struct sockaddr *)cp;
68 if ((cp + sa->sa_len) > cplim) {
69 return EINVAL;
70 }
71 rtinfo->rti_info[i] = sa;
72 cp += ROUNDUP(sa->sa_len);
73 }
74 return 0;
75 }
76
77 /**
78 ** stolen from bootp/IPConfiguration.bproj/iputil.c
79 **
80 ** inet6_addrlist_*
81 **/
82
83 #define s6_addr16 __u6_addr.__u6_addr16
84
85 static char *
copy_if_info(unsigned int if_index,int af,int * ret_len_p)86 copy_if_info(unsigned int if_index, int af, int *ret_len_p)
87 {
88 char * buf = NULL;
89 size_t buf_len = 0;
90 int mib[6];
91
92 mib[0] = CTL_NET;
93 mib[1] = PF_ROUTE;
94 mib[2] = 0;
95 mib[3] = af;
96 mib[4] = NET_RT_IFLIST;
97 mib[5] = (int)if_index;
98
99 *ret_len_p = 0;
100 if (sysctl(mib, 6, NULL, &buf_len, NULL, 0) < 0) {
101 fprintf(stderr, "sysctl() size failed: %s", strerror(errno));
102 goto failed;
103 }
104 buf_len *= 2; /* just in case something changes */
105 buf = malloc(buf_len);
106 if (sysctl(mib, 6, buf, &buf_len, NULL, 0) < 0) {
107 free(buf);
108 buf = NULL;
109 fprintf(stderr, "sysctl() failed: %s", strerror(errno));
110 goto failed;
111 }
112 *ret_len_p = (int)buf_len;
113
114 failed:
115 return buf;
116 }
117
118 static bool
inet6_get_linklocal_address(unsigned int if_index,struct in6_addr * ret_addr)119 inet6_get_linklocal_address(unsigned int if_index, struct in6_addr *ret_addr)
120 {
121 char * buf = NULL;
122 char * buf_end;
123 int buf_len;
124 bool found = FALSE;
125 char *scan;
126 struct rt_msghdr *rtm;
127
128 bzero(ret_addr, sizeof(*ret_addr));
129 buf = copy_if_info(if_index, AF_INET6, &buf_len);
130 if (buf == NULL) {
131 goto done;
132 }
133 buf_end = buf + buf_len;
134 for (scan = buf; scan < buf_end; scan += rtm->rtm_msglen) {
135 struct ifa_msghdr * ifam;
136 struct rt_addrinfo info;
137
138 /* ALIGN: buf aligned (from calling copy_if_info), scan aligned,
139 * cast ok. */
140 rtm = (struct rt_msghdr *)(void *)scan;
141 if (rtm->rtm_version != RTM_VERSION) {
142 continue;
143 }
144 if (rtm->rtm_type == RTM_NEWADDR) {
145 errno_t error;
146 struct sockaddr_in6 *sin6_p;
147
148 ifam = (struct ifa_msghdr *)rtm;
149 info.rti_addrs = ifam->ifam_addrs;
150 error = rt_xaddrs((char *)(ifam + 1),
151 ((char *)ifam) + ifam->ifam_msglen,
152 &info);
153 if (error) {
154 fprintf(stderr, "couldn't extract rt_addrinfo %s (%d)\n",
155 strerror(error), error);
156 goto done;
157 }
158 /* ALIGN: info.rti_info aligned (sockaddr), cast ok. */
159 sin6_p = (struct sockaddr_in6 *)(void *)info.rti_info[RTAX_IFA];
160 if (sin6_p == NULL
161 || sin6_p->sin6_len < sizeof(struct sockaddr_in6)) {
162 continue;
163 }
164 if (IN6_IS_ADDR_LINKLOCAL(&sin6_p->sin6_addr)) {
165 *ret_addr = sin6_p->sin6_addr;
166 ret_addr->s6_addr16[1] = 0; /* mask scope id */
167 found = TRUE;
168 break;
169 }
170 }
171 }
172
173 done:
174 if (buf != NULL) {
175 free(buf);
176 }
177 return found;
178 }
179
180 static void
cleanup(void)181 cleanup(void)
182 {
183 if (ifname1[0] != '\0') {
184 T_LOG("ifnet_destroy %s", ifname1);
185 (void)ifnet_destroy(ifname1, false);
186 }
187 }
188
189 static void
set_sockaddr_in6(struct sockaddr_in6 * sin6_p,const struct in6_addr * addr)190 set_sockaddr_in6(struct sockaddr_in6 *sin6_p, const struct in6_addr *addr)
191 {
192 sin6_p->sin6_family = AF_INET6;
193 sin6_p->sin6_len = sizeof(struct sockaddr_in6);
194 sin6_p->sin6_addr = *addr;
195 return;
196 }
197
198 static int
inet6_difaddr(const char * name,const struct in6_addr * addr)199 inet6_difaddr(const char *name, const struct in6_addr *addr)
200 {
201 struct in6_ifreq ifr;
202 int s6 = inet6_dgram_socket_get();
203
204 bzero(&ifr, sizeof(ifr));
205 strncpy(ifr.ifr_name, name, sizeof(ifr.ifr_name));
206 if (addr != NULL) {
207 set_sockaddr_in6(&ifr.ifr_ifru.ifru_addr, addr);
208 }
209 return ioctl(s6, SIOCDIFADDR_IN6, &ifr);
210 }
211
212 static void
in6_len2mask(struct in6_addr * mask,int len)213 in6_len2mask(struct in6_addr *mask, int len)
214 {
215 int i;
216
217 bzero(mask, sizeof(*mask));
218 for (i = 0; i < len / 8; i++) {
219 mask->s6_addr[i] = 0xff;
220 }
221 if (len % 8) {
222 mask->s6_addr[i] = (0xff00 >> (len % 8)) & 0xff;
223 }
224 }
225
226 static int
inet6_aifaddr(const char * name,const struct in6_addr * addr,const struct in6_addr * dstaddr,int prefix_length,int flags,u_int32_t valid_lifetime,u_int32_t preferred_lifetime)227 inet6_aifaddr(const char *name, const struct in6_addr *addr,
228 const struct in6_addr *dstaddr, int prefix_length,
229 int flags,
230 u_int32_t valid_lifetime,
231 u_int32_t preferred_lifetime)
232 {
233 struct in6_aliasreq ifra_in6;
234 int s6 = inet6_dgram_socket_get();
235
236 bzero(&ifra_in6, sizeof(ifra_in6));
237 strncpy(ifra_in6.ifra_name, name, sizeof(ifra_in6.ifra_name));
238 ifra_in6.ifra_lifetime.ia6t_vltime = valid_lifetime;
239 ifra_in6.ifra_lifetime.ia6t_pltime = preferred_lifetime;
240 ifra_in6.ifra_flags = flags;
241 if (addr != NULL) {
242 set_sockaddr_in6(&ifra_in6.ifra_addr, addr);
243 }
244
245 if (dstaddr != NULL) {
246 set_sockaddr_in6(&ifra_in6.ifra_dstaddr, dstaddr);
247 }
248
249 if (prefix_length != 0) {
250 struct in6_addr prefixmask;
251
252 in6_len2mask(&prefixmask, prefix_length);
253 set_sockaddr_in6(&ifra_in6.ifra_prefixmask, &prefixmask);
254 }
255
256 return ioctl(s6, SIOCAIFADDR_IN6, &ifra_in6);
257 }
258
259 static void
create_fake_interface(void)260 create_fake_interface(void)
261 {
262 int error;
263
264 strlcpy(ifname1, FETH_NAME, sizeof(ifname1));
265 error = ifnet_create_2(ifname1, sizeof(ifname1));
266 if (error != 0) {
267 ifname1[0] = '\0';
268 T_ASSERT_POSIX_SUCCESS(error, "ifnet_create_2");
269 }
270 T_LOG("created %s", ifname1);
271 }
272
273 T_DECL(inet6_addr_mode_auto_to_manual, "inet6 address mode-switching (auto -> manual)")
274 {
275 struct in6_addr lladdr;
276 struct in6_addr newaddr;
277 unsigned int if_index;
278
279 T_ATEND(cleanup);
280
281 create_fake_interface();
282 ifnet_start_ipv6(ifname1);
283
284 if_index = if_nametoindex(ifname1);
285 T_EXPECT_GT(if_index, 0, NULL);
286 T_ASSERT_EQ(inet6_get_linklocal_address(if_index, &lladdr), 1, NULL);
287
288 T_ASSERT_POSIX_SUCCESS(inet6_aifaddr(ifname1, &lladdr, NULL, 64, 123, ND6_INFINITE_LIFETIME, ND6_INFINITE_LIFETIME), NULL);
289
290 /* Create address as an autoconfed address */
291 T_ASSERT_EQ(inet_pton(AF_INET6, "2001:db8::3", &newaddr), 1, NULL);
292 T_ASSERT_POSIX_SUCCESS(inet6_aifaddr(ifname1, &newaddr, NULL, 64, (IN6_IFF_AUTOCONF | IN6_IFF_TEMPORARY), ND6_INFINITE_LIFETIME, ND6_INFINITE_LIFETIME), NULL);
293
294 /* Now mark it as manual */
295 T_ASSERT_POSIX_SUCCESS(inet6_aifaddr(ifname1, &newaddr, NULL, 64, 0, ND6_INFINITE_LIFETIME, ND6_INFINITE_LIFETIME), NULL);
296
297 /* Deleting address should NOT result in panic */
298 T_ASSERT_POSIX_SUCCESS(inet6_difaddr(ifname1, &newaddr), NULL);
299 }
300
301 T_DECL(inet6_addr_mode_manual_to_auto, "inet6 address mode-switching (manual -> auto)")
302 {
303 struct in6_addr lladdr;
304 struct in6_addr newaddr;
305 unsigned int if_index;
306
307 T_ATEND(cleanup);
308 create_fake_interface();
309
310 T_LOG("created %s", ifname1);
311
312 ifnet_start_ipv6(ifname1);
313
314 if_index = if_nametoindex(ifname1);
315 T_EXPECT_GT(if_index, 0, NULL);
316 T_ASSERT_EQ(inet6_get_linklocal_address(if_index, &lladdr), 1, NULL);
317
318 T_ASSERT_POSIX_SUCCESS(inet6_aifaddr(ifname1, &lladdr, NULL, 64, 123, ND6_INFINITE_LIFETIME, ND6_INFINITE_LIFETIME), NULL);
319
320 /* Create address as a manual address */
321 T_ASSERT_EQ(inet_pton(AF_INET6, "2001:db8::1", &newaddr), 1, NULL);
322 T_ASSERT_POSIX_SUCCESS(inet6_aifaddr(ifname1, &newaddr, NULL, 64, 0, ND6_INFINITE_LIFETIME, ND6_INFINITE_LIFETIME), NULL);
323
324 /* Now make it autoconfed */
325 T_ASSERT_POSIX_SUCCESS(inet6_aifaddr(ifname1, &newaddr, NULL, 64, (IN6_IFF_AUTOCONF | IN6_IFF_TEMPORARY), ND6_INFINITE_LIFETIME, ND6_INFINITE_LIFETIME), NULL);
326
327 /* Deleting address should NOT result in panic */
328 T_ASSERT_POSIX_SUCCESS(inet6_difaddr(ifname1, &newaddr), NULL);
329 }
330