1 /*
2 * Copyright (c) 2017 JingPiao Chen <chenjingpiao@gmail.com>
3 * Copyright (c) 2017-2021 The strace developers.
4 * All rights reserved.
5 *
6 * SPDX-License-Identifier: GPL-2.0-or-later
7 */
8
9 #include "tests.h"
10
11 #include <stdio.h>
12 #include <netinet/in.h>
13 #include <arpa/inet.h>
14 #include "test_nlattr.h"
15 #include <linux/ip.h>
16 #include <linux/rtnetlink.h>
17
18 #if !defined HAVE_MEMPCPY
19 # undef mempcpy
20 # define mempcpy strace_mempcpy
21 static void *
22 mempcpy(void *dest, const void *src, size_t n)
23 {
24 memcpy(dest, src, n);
25
26 return dest + n;
27 }
28 #endif
29
30 static void
31 print_quoted_hex_ellipsis(const void *const instr, const size_t len)
32 {
33 const unsigned char *str = instr;
34
35 printf("\"");
36 for (size_t i = 0; i < MIN(len, DEFAULT_STRLEN); ++i)
37 printf("\\x%02x", str[i]);
38 printf("\"");
39 if (len > DEFAULT_STRLEN)
40 printf("...");
41 }
42
43 #define LWTUNNEL_ENCAP_NONE 0
44
45 #define DEF_NLATTR_RTMSG_FUNCS(sfx_, af_) \
46 static void \
47 init_##sfx_(struct nlmsghdr *const nlh, const unsigned int msg_len) \
48 { \
49 SET_STRUCT(struct nlmsghdr, nlh, \
50 .nlmsg_len = msg_len, \
51 .nlmsg_type = RTM_GETROUTE, \
52 .nlmsg_flags = NLM_F_DUMP \
53 ); \
54 \
55 struct rtmsg *const msg = NLMSG_DATA(nlh); \
56 SET_STRUCT(struct rtmsg, msg, \
57 .rtm_family = (af_), \
58 .rtm_tos = IPTOS_LOWDELAY, \
59 .rtm_table = RT_TABLE_DEFAULT, \
60 .rtm_protocol = RTPROT_KERNEL, \
61 .rtm_scope = RT_SCOPE_UNIVERSE, \
62 .rtm_type = RTN_LOCAL, \
63 .rtm_flags = RTM_F_NOTIFY \
64 ); \
65 } \
66 \
67 static void \
68 print_##sfx_(const unsigned int msg_len) \
69 { \
70 printf("{nlmsg_len=%u, nlmsg_type=RTM_GETROUTE" \
71 ", nlmsg_flags=NLM_F_DUMP" \
72 ", nlmsg_seq=0, nlmsg_pid=0}, {rtm_family=" #af_ \
73 ", rtm_dst_len=0, rtm_src_len=0" \
74 ", rtm_tos=IPTOS_LOWDELAY" \
75 ", rtm_table=RT_TABLE_DEFAULT" \
76 ", rtm_protocol=RTPROT_KERNEL" \
77 ", rtm_scope=RT_SCOPE_UNIVERSE" \
78 ", rtm_type=RTN_LOCAL" \
79 ", rtm_flags=RTM_F_NOTIFY}", \
80 msg_len); \
81 } \
82 /* End of DEF_NLATTR_RTMSG_FUNCS */
83
84 DEF_NLATTR_RTMSG_FUNCS(rtmsg, AF_UNIX)
85 DEF_NLATTR_RTMSG_FUNCS(rtmsg_inet, AF_INET)
86 DEF_NLATTR_RTMSG_FUNCS(rtmsg_inet6, AF_INET6)
87
88 int
89 main(void)
90 {
91 skip_if_unavailable("/proc/self/fd/");
92
93 const int fd = create_nl_socket(NETLINK_ROUTE);
94 const unsigned int hdrlen = sizeof(struct rtmsg);
95 const unsigned int nla_type = 0xffff & NLA_TYPE_MASK;
96 char nla_type_str[256];
97 void *nlh0 = midtail_alloc(NLMSG_SPACE(hdrlen),
98 NLA_HDRLEN + sizeof(nla_type_str));
99
100 static char pattern[4096];
101 fill_memory_ex(pattern, sizeof(pattern), 'a', 'z' - 'a' + 1);
102
103 sprintf(nla_type_str, "%#x /* RTA_??? */", nla_type);
104 TEST_NLATTR_(fd, nlh0, hdrlen,
105 init_rtmsg, print_rtmsg,
106 nla_type, nla_type_str,
107 4, pattern, 4,
108 print_quoted_hex(pattern, 4));
109
110 TEST_NLATTR(fd, nlh0, hdrlen,
111 init_rtmsg, print_rtmsg,
112 RTA_DST, 4, pattern, 4,
113 print_quoted_hex(pattern, 4));
114
115 #define MAX_ADDR_SZ 35
116 static const struct {
117 uint8_t af;
118 uint8_t addr[MAX_ADDR_SZ];
119 const char *str;
120 void (* init_fn)(struct nlmsghdr *, unsigned int);
121 void (* print_fn)(unsigned int);
122 uint32_t len;
123 } addrs[] = {
124 { AF_UNIX, { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 },
125 "\"\\x00\\x01\\x02\\x03\\x04\\x05\\x06\\x07\\x08\\x09\"",
126 init_rtmsg, print_rtmsg, 10 },
127 { AF_UNIX,
128 { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, [MAX_ADDR_SZ - 1] = 0xea },
129 "\"\\x00\\x01\\x02\\x03\\x04\\x05\\x06\\x07\\x08\\x09"
130 "\\x00\\x00\\x00\\x00\\x00\\x00\\x00\\x00\\x00\\x00"
131 "\\x00\\x00\\x00\\x00\\x00\\x00\\x00\\x00\\x00\\x00"
132 "\\x00\\x00"
133 #if DEFAULT_STRLEN == 32
134 "\"...",
135 #else
136 "\\x00\\x00\\xea\"",
137 #endif
138 init_rtmsg, print_rtmsg, MAX_ADDR_SZ },
139 { AF_INET, { 0xde, 0xca, 0xff, 0xed },
140 "inet_addr(\"222.202.255.237\")",
141 init_rtmsg_inet, print_rtmsg_inet, 4 },
142 { AF_INET6, { 0xfa, 0xce, 0xbe, 0xef, [15] = 0xda },
143 "inet_pton(AF_INET6, \"face:beef::da\")",
144 init_rtmsg_inet6, print_rtmsg_inet6, 16 },
145 };
146 static const struct strval32 addr_attrs[] = {
147 { ARG_STR(RTA_DST) },
148 { ARG_STR(RTA_SRC) },
149 { ARG_STR(RTA_GATEWAY) },
150 { ARG_STR(RTA_PREFSRC) },
151 { ARG_STR(RTA_NEWDST) },
152 };
153 for (size_t i = 0; i < ARRAY_SIZE(addrs); i++) {
154 for (size_t j = 0; j < ARRAY_SIZE(addr_attrs); j++) {
155 TEST_NLATTR_(fd, nlh0, hdrlen,
156 addrs[i].init_fn, addrs[i].print_fn,
157 addr_attrs[j].val, addr_attrs[j].str,
158 addrs[i].len - 1, addrs[i].addr,
159 addrs[i].len - 1,
160 print_quoted_hex_ellipsis(addrs[i].addr,
161 addrs[i].len - 1)
162 );
163 TEST_NLATTR_(fd, nlh0, hdrlen,
164 addrs[i].init_fn, addrs[i].print_fn,
165 addr_attrs[j].val, addr_attrs[j].str,
166 addrs[i].len, addrs[i].addr, addrs[i].len,
167 printf("%s", addrs[i].str));
168 }
169 }
170
171 const uint32_t ifindex = ifindex_lo();
172 TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
173 init_rtmsg, print_rtmsg,
174 RTA_OIF, pattern, ifindex,
175 printf(IFINDEX_LO_STR));
176
177 const uint32_t rt_class_id = RT_TABLE_DEFAULT;
178 TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
179 init_rtmsg, print_rtmsg,
180 RTA_TABLE, pattern, rt_class_id,
181 printf("RT_TABLE_DEFAULT"));
182
183 struct nlattr nla = {
184 .nla_type = RTAX_LOCK,
185 .nla_len = sizeof(nla)
186 };
187 TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
188 init_rtmsg, print_rtmsg,
189 RTA_METRICS, pattern, nla,
190 printf("{nla_len=%u, nla_type=RTAX_LOCK}",
191 nla.nla_len));
192 struct rtnexthop nh = {
193 .rtnh_len = sizeof(nh) - 1,
194 .rtnh_flags = RTNH_F_DEAD,
195 .rtnh_hops = 0xab,
196 .rtnh_ifindex = ifindex_lo()
197 };
198 TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
199 init_rtmsg, print_rtmsg,
200 RTA_MULTIPATH, pattern, nh,
201 printf("[{rtnh_len=%u, rtnh_flags=RTNH_F_DEAD"
202 ", rtnh_hops=%u"
203 ", rtnh_ifindex=" IFINDEX_LO_STR "}]",
204 nh.rtnh_len, nh.rtnh_hops));
205
206 char buf[RTNH_ALIGN(sizeof(nh)) + sizeof(nla)];
207 nh.rtnh_len = sizeof(buf);
208 nla.nla_type = RTA_DST;
209 memcpy(buf, &nh, sizeof(nh));
210 memcpy(buf + RTNH_ALIGN(sizeof(nh)), &nla, sizeof(nla));
211 TEST_NLATTR(fd, nlh0, hdrlen,
212 init_rtmsg, print_rtmsg,
213 RTA_MULTIPATH, sizeof(buf), buf, sizeof(buf),
214 printf("[{rtnh_len=%u, rtnh_flags=RTNH_F_DEAD"
215 ", rtnh_hops=%u, rtnh_ifindex=" IFINDEX_LO_STR "}"
216 ", {nla_len=%u, nla_type=RTA_DST}]",
217 nh.rtnh_len, nh.rtnh_hops, nla.nla_len));
218
219 static const struct in_addr gw_inet_addr = { .s_addr = BE32(0xdeadbeef) };
220 static const uint8_t via_inet6_addr[16] = {
221 0xde, 0xad, 0xfa, 0xce, 0xbe, 0xef, 0xca, 0xfe,
222 0xfe, 0xed, 0xba, 0x5e, 0x00, 0x00, 0xfa, 0xde };
223 static const struct rtvia rtvia = { .rtvia_family = AF_INET6 };
224 char buf2[2 * (RTNH_ALIGN(sizeof(nh)) + NLMSG_ALIGN(sizeof(nla))) +
225 + NLMSG_ALIGN(sizeof(gw_inet_addr))
226 + NLMSG_ALIGN(offsetof(struct rtvia, rtvia_addr)
227 + sizeof(via_inet6_addr))];
228 char *pos = buf2;
229
230 nh.rtnh_len = RTNH_ALIGN(sizeof(nh)) + NLMSG_ALIGN(sizeof(nla))
231 + NLMSG_ALIGN(sizeof(gw_inet_addr));
232 nh.rtnh_flags = 0xc0;
233 nla.nla_type = RTA_GATEWAY;
234 nla.nla_len = NLMSG_ALIGN(sizeof(nla))
235 + NLMSG_ALIGN(sizeof(gw_inet_addr));
236 pos = mempcpy(pos, &nh, sizeof(nh));
237 pos = mempcpy(pos, &nla, sizeof(nla));
238 pos = mempcpy(pos, &gw_inet_addr, sizeof(gw_inet_addr));
239
240 nh.rtnh_len = RTNH_ALIGN(sizeof(nh)) + NLMSG_ALIGN(sizeof(nla))
241 + NLMSG_ALIGN(offsetof(struct rtvia, rtvia_addr)
242 + sizeof(via_inet6_addr));
243 nla.nla_type = RTA_VIA;
244 nla.nla_len = NLMSG_ALIGN(sizeof(nla))
245 + NLMSG_ALIGN(offsetof(struct rtvia, rtvia_addr)
246 + sizeof(via_inet6_addr));
247 pos = mempcpy(pos, &nh, sizeof(nh));
248 pos = mempcpy(pos, &nla, sizeof(nla));
249 pos = mempcpy(pos, &rtvia, sizeof(rtvia));
250 pos = mempcpy(pos, &via_inet6_addr, sizeof(via_inet6_addr));
251 TEST_NLATTR(fd, nlh0, hdrlen,
252 init_rtmsg_inet, print_rtmsg_inet,
253 RTA_MULTIPATH, sizeof(buf2), buf2, sizeof(buf2),
254 printf("[[{rtnh_len=%u, rtnh_flags=RTNH_F_TRAP|0x80"
255 ", rtnh_hops=%u, rtnh_ifindex=" IFINDEX_LO_STR "}"
256 ", [{nla_len=%u, nla_type=RTA_GATEWAY}"
257 ", inet_addr(\"222.173.190.239\")]]"
258 ", [{rtnh_len=%u, rtnh_flags=RTNH_F_TRAP|0x80"
259 ", rtnh_hops=%u, rtnh_ifindex=" IFINDEX_LO_STR "}"
260 ", [{nla_len=%u, nla_type=RTA_VIA}"
261 ", {rtvia_family=AF_INET6"
262 ", inet_pton(AF_INET6"
263 ", \"dead:face:beef:cafe:feed:ba5e:0:fade\""
264 ", &rtvia_addr)}]]]",
265 (uint32_t) (RTNH_ALIGN(sizeof(nh))
266 + NLMSG_ALIGN(sizeof(nla))
267 + NLMSG_ALIGN(sizeof(gw_inet_addr))),
268 nh.rtnh_hops,
269 (uint32_t) (NLMSG_ALIGN(sizeof(nla))
270 + NLMSG_ALIGN(sizeof(gw_inet_addr))),
271 (uint32_t) (RTNH_ALIGN(sizeof(nh))
272 + NLMSG_ALIGN(sizeof(nla))
273 + NLMSG_ALIGN(offsetof(struct rtvia, rtvia_addr)
274 + sizeof(via_inet6_addr))),
275 nh.rtnh_hops,
276 (uint32_t) (NLMSG_ALIGN(sizeof(nla))
277 + NLMSG_ALIGN(offsetof(struct rtvia, rtvia_addr)
278 + sizeof(via_inet6_addr)))
279 ));
280
281 static const struct rta_cacheinfo ci = {
282 .rta_clntref = 0xabcdefab,
283 .rta_lastuse = 0xbdadaedc,
284 .rta_expires = 0xcdadebad,
285 .rta_error = 0xdaedadeb,
286 .rta_used = 0xedfabdad,
287 .rta_id = 0xfeadbcda,
288 .rta_ts = 0xacdbaded,
289 .rta_tsage = 0xbadeadef
290 };
291 TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
292 init_rtmsg, print_rtmsg,
293 RTA_CACHEINFO, pattern, ci,
294 printf("{");
295 PRINT_FIELD_U(ci, rta_clntref);
296 printf(", ");
297 PRINT_FIELD_U(ci, rta_lastuse);
298 printf(", ");
299 PRINT_FIELD_U(ci, rta_expires);
300 printf(", ");
301 PRINT_FIELD_U(ci, rta_error);
302 printf(", ");
303 PRINT_FIELD_U(ci, rta_used);
304 printf(", ");
305 PRINT_FIELD_X(ci, rta_id);
306 printf(", ");
307 PRINT_FIELD_U(ci, rta_ts);
308 printf(", ");
309 PRINT_FIELD_U(ci, rta_tsage);
310 printf("}"));
311
312 static const struct rta_mfc_stats mfcs = {
313 .mfcs_packets = 0xadcdedfdadefadcd,
314 .mfcs_bytes = 0xbaedadedcdedadbd,
315 .mfcs_wrong_if = 0xcddeabeedaedabfa
316 };
317 TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
318 init_rtmsg, print_rtmsg,
319 RTA_MFC_STATS, pattern, mfcs,
320 printf("{");
321 PRINT_FIELD_U(mfcs, mfcs_packets);
322 printf(", ");
323 PRINT_FIELD_U(mfcs, mfcs_bytes);
324 printf(", ");
325 PRINT_FIELD_U(mfcs, mfcs_wrong_if);
326 printf("}"));
327
328 static const struct rtvia via = {
329 .rtvia_family = AF_INET
330 };
331
332 TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
333 init_rtmsg, print_rtmsg,
334 RTA_VIA, pattern, via,
335 printf("{rtvia_family=AF_INET}"));
336
337 static const char address4[] = "12.34.56.78";
338 struct in_addr a4 = {
339 .s_addr = inet_addr(address4)
340 };
341 char rtviabuf[sizeof(via) + sizeof(a4)];
342 memcpy(rtviabuf, &via, sizeof(via));
343 memcpy(rtviabuf + sizeof(via), &a4, sizeof(a4));
344
345 TEST_NLATTR(fd, nlh0, hdrlen,
346 init_rtmsg, print_rtmsg,
347 RTA_VIA, sizeof(rtviabuf), rtviabuf, sizeof(rtviabuf),
348 printf("{rtvia_family=AF_INET"
349 ", rtvia_addr=inet_addr(\"%s\")}", address4));
350
351 const uint16_t encap_type = LWTUNNEL_ENCAP_NONE;
352 TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
353 init_rtmsg, print_rtmsg,
354 RTA_ENCAP_TYPE, pattern, encap_type,
355 printf("LWTUNNEL_ENCAP_NONE"));
356
357 puts("+++ exited with 0 +++");
358 return 0;
359 }