]>
Commit | Line | Data |
---|---|---|
53e1b683 | 1 | /* SPDX-License-Identifier: LGPL-2.1+ */ |
1c4baffc | 2 | |
1c4baffc | 3 | #include <net/if.h> |
07630cea | 4 | #include <netinet/ether.h> |
1c4baffc | 5 | |
1c4baffc | 6 | #include "sd-netlink.h" |
07630cea | 7 | |
81a56d6f | 8 | #include "ether-addr-util.h" |
07630cea | 9 | #include "macro.h" |
1c4baffc | 10 | #include "missing.h" |
07630cea LP |
11 | #include "netlink-util.h" |
12 | #include "socket-util.h" | |
13 | #include "string-util.h" | |
14 | #include "util.h" | |
1c4baffc TG |
15 | |
16 | static void test_message_link_bridge(sd_netlink *rtnl) { | |
4afd3348 | 17 | _cleanup_(sd_netlink_message_unrefp) sd_netlink_message *message = NULL; |
1c4baffc TG |
18 | uint32_t cost; |
19 | ||
20 | assert_se(sd_rtnl_message_new_link(rtnl, &message, RTM_NEWLINK, 1) >= 0); | |
21 | assert_se(sd_rtnl_message_link_set_family(message, PF_BRIDGE) >= 0); | |
22 | assert_se(sd_netlink_message_open_container(message, IFLA_PROTINFO) >= 0); | |
23 | assert_se(sd_netlink_message_append_u32(message, IFLA_BRPORT_COST, 10) >= 0); | |
24 | assert_se(sd_netlink_message_close_container(message) >= 0); | |
25 | ||
26 | assert_se(sd_netlink_message_rewind(message) >= 0); | |
27 | ||
28 | assert_se(sd_netlink_message_enter_container(message, IFLA_PROTINFO) >= 0); | |
29 | assert_se(sd_netlink_message_read_u32(message, IFLA_BRPORT_COST, &cost) >= 0); | |
30 | assert_se(cost == 10); | |
31 | assert_se(sd_netlink_message_exit_container(message) >= 0); | |
32 | } | |
33 | ||
34 | static void test_link_configure(sd_netlink *rtnl, int ifindex) { | |
4afd3348 | 35 | _cleanup_(sd_netlink_message_unrefp) sd_netlink_message *message = NULL; |
1c4baffc TG |
36 | const char *mac = "98:fe:94:3f:c6:18", *name = "test"; |
37 | char buffer[ETHER_ADDR_TO_STRING_MAX]; | |
4e964aa0 | 38 | uint32_t mtu = 1450, mtu_out; |
1c4baffc TG |
39 | const char *name_out; |
40 | struct ether_addr mac_out; | |
41 | ||
42 | /* we'd really like to test NEWLINK, but let's not mess with the running kernel */ | |
43 | assert_se(sd_rtnl_message_new_link(rtnl, &message, RTM_GETLINK, ifindex) >= 0); | |
44 | assert_se(sd_netlink_message_append_string(message, IFLA_IFNAME, name) >= 0); | |
45 | assert_se(sd_netlink_message_append_ether_addr(message, IFLA_ADDRESS, ether_aton(mac)) >= 0); | |
46 | assert_se(sd_netlink_message_append_u32(message, IFLA_MTU, mtu) >= 0); | |
47 | ||
48 | assert_se(sd_netlink_call(rtnl, message, 0, NULL) == 1); | |
49 | assert_se(sd_netlink_message_rewind(message) >= 0); | |
50 | ||
51 | assert_se(sd_netlink_message_read_string(message, IFLA_IFNAME, &name_out) >= 0); | |
52 | assert_se(streq(name, name_out)); | |
53 | ||
54 | assert_se(sd_netlink_message_read_ether_addr(message, IFLA_ADDRESS, &mac_out) >= 0); | |
55 | assert_se(streq(mac, ether_addr_to_string(&mac_out, buffer))); | |
56 | ||
57 | assert_se(sd_netlink_message_read_u32(message, IFLA_MTU, &mtu_out) >= 0); | |
58 | assert_se(mtu == mtu_out); | |
59 | } | |
60 | ||
61 | static void test_link_get(sd_netlink *rtnl, int ifindex) { | |
62 | sd_netlink_message *m; | |
63 | sd_netlink_message *r; | |
4e964aa0 | 64 | uint32_t mtu = 1500; |
1c4baffc TG |
65 | const char *str_data; |
66 | uint8_t u8_data; | |
67 | uint32_t u32_data; | |
68 | struct ether_addr eth_data; | |
69 | ||
70 | assert_se(sd_rtnl_message_new_link(rtnl, &m, RTM_GETLINK, ifindex) >= 0); | |
71 | assert_se(m); | |
72 | ||
73 | /* u8 test cases */ | |
74 | assert_se(sd_netlink_message_append_u8(m, IFLA_CARRIER, 0) >= 0); | |
75 | assert_se(sd_netlink_message_append_u8(m, IFLA_OPERSTATE, 0) >= 0); | |
76 | assert_se(sd_netlink_message_append_u8(m, IFLA_LINKMODE, 0) >= 0); | |
77 | ||
78 | /* u32 test cases */ | |
79 | assert_se(sd_netlink_message_append_u32(m, IFLA_MTU, mtu) >= 0); | |
80 | assert_se(sd_netlink_message_append_u32(m, IFLA_GROUP, 0) >= 0); | |
81 | assert_se(sd_netlink_message_append_u32(m, IFLA_TXQLEN, 0) >= 0); | |
82 | assert_se(sd_netlink_message_append_u32(m, IFLA_NUM_TX_QUEUES, 0) >= 0); | |
83 | assert_se(sd_netlink_message_append_u32(m, IFLA_NUM_RX_QUEUES, 0) >= 0); | |
84 | ||
85 | assert_se(sd_netlink_call(rtnl, m, -1, &r) == 1); | |
86 | ||
87 | assert_se(sd_netlink_message_read_string(r, IFLA_IFNAME, &str_data) == 0); | |
88 | ||
89 | assert_se(sd_netlink_message_read_u8(r, IFLA_CARRIER, &u8_data) == 0); | |
90 | assert_se(sd_netlink_message_read_u8(r, IFLA_OPERSTATE, &u8_data) == 0); | |
91 | assert_se(sd_netlink_message_read_u8(r, IFLA_LINKMODE, &u8_data) == 0); | |
92 | ||
93 | assert_se(sd_netlink_message_read_u32(r, IFLA_MTU, &u32_data) == 0); | |
94 | assert_se(sd_netlink_message_read_u32(r, IFLA_GROUP, &u32_data) == 0); | |
95 | assert_se(sd_netlink_message_read_u32(r, IFLA_TXQLEN, &u32_data) == 0); | |
96 | assert_se(sd_netlink_message_read_u32(r, IFLA_NUM_TX_QUEUES, &u32_data) == 0); | |
97 | assert_se(sd_netlink_message_read_u32(r, IFLA_NUM_RX_QUEUES, &u32_data) == 0); | |
98 | ||
99 | assert_se(sd_netlink_message_read_ether_addr(r, IFLA_ADDRESS, ð_data) == 0); | |
100 | ||
1c4baffc TG |
101 | assert_se((m = sd_netlink_message_unref(m)) == NULL); |
102 | assert_se((r = sd_netlink_message_unref(r)) == NULL); | |
103 | } | |
104 | ||
1c4baffc TG |
105 | static void test_address_get(sd_netlink *rtnl, int ifindex) { |
106 | sd_netlink_message *m; | |
107 | sd_netlink_message *r; | |
108 | struct in_addr in_data; | |
109 | struct ifa_cacheinfo cache; | |
110 | const char *label; | |
111 | ||
112 | assert_se(sd_rtnl_message_new_addr(rtnl, &m, RTM_GETADDR, ifindex, AF_INET) >= 0); | |
113 | assert_se(m); | |
114 | ||
115 | assert_se(sd_netlink_call(rtnl, m, -1, &r) == 1); | |
116 | ||
117 | assert_se(sd_netlink_message_read_in_addr(r, IFA_LOCAL, &in_data) == 0); | |
118 | assert_se(sd_netlink_message_read_in_addr(r, IFA_ADDRESS, &in_data) == 0); | |
119 | assert_se(sd_netlink_message_read_string(r, IFA_LABEL, &label) == 0); | |
120 | assert_se(sd_netlink_message_read_cache_info(r, IFA_CACHEINFO, &cache) == 0); | |
121 | ||
1c4baffc TG |
122 | assert_se((m = sd_netlink_message_unref(m)) == NULL); |
123 | assert_se((r = sd_netlink_message_unref(r)) == NULL); | |
124 | ||
125 | } | |
126 | ||
05d0c2e3 | 127 | static void test_route(sd_netlink *rtnl) { |
4afd3348 | 128 | _cleanup_(sd_netlink_message_unrefp) sd_netlink_message *req; |
1c4baffc TG |
129 | struct in_addr addr, addr_data; |
130 | uint32_t index = 2, u32_data; | |
131 | int r; | |
132 | ||
05d0c2e3 | 133 | r = sd_rtnl_message_new_route(rtnl, &req, RTM_NEWROUTE, AF_INET, RTPROT_STATIC); |
1c4baffc TG |
134 | if (r < 0) { |
135 | log_error_errno(r, "Could not create RTM_NEWROUTE message: %m"); | |
136 | return; | |
137 | } | |
138 | ||
139 | addr.s_addr = htonl(INADDR_LOOPBACK); | |
140 | ||
141 | r = sd_netlink_message_append_in_addr(req, RTA_GATEWAY, &addr); | |
142 | if (r < 0) { | |
143 | log_error_errno(r, "Could not append RTA_GATEWAY attribute: %m"); | |
144 | return; | |
145 | } | |
146 | ||
147 | r = sd_netlink_message_append_u32(req, RTA_OIF, index); | |
148 | if (r < 0) { | |
149 | log_error_errno(r, "Could not append RTA_OIF attribute: %m"); | |
150 | return; | |
151 | } | |
152 | ||
153 | assert_se(sd_netlink_message_rewind(req) >= 0); | |
154 | ||
155 | assert_se(sd_netlink_message_read_in_addr(req, RTA_GATEWAY, &addr_data) >= 0); | |
156 | assert_se(addr_data.s_addr == addr.s_addr); | |
157 | ||
158 | assert_se(sd_netlink_message_read_u32(req, RTA_OIF, &u32_data) >= 0); | |
159 | assert_se(u32_data == index); | |
160 | ||
161 | assert_se((req = sd_netlink_message_unref(req)) == NULL); | |
162 | } | |
163 | ||
164 | static void test_multiple(void) { | |
165 | sd_netlink *rtnl1, *rtnl2; | |
166 | ||
167 | assert_se(sd_netlink_open(&rtnl1) >= 0); | |
168 | assert_se(sd_netlink_open(&rtnl2) >= 0); | |
169 | ||
170 | rtnl1 = sd_netlink_unref(rtnl1); | |
171 | rtnl2 = sd_netlink_unref(rtnl2); | |
172 | } | |
173 | ||
174 | static int link_handler(sd_netlink *rtnl, sd_netlink_message *m, void *userdata) { | |
175 | char *ifname = userdata; | |
176 | const char *data; | |
177 | ||
178 | assert_se(rtnl); | |
179 | assert_se(m); | |
180 | ||
181 | log_info("got link info about %s", ifname); | |
182 | free(ifname); | |
183 | ||
184 | assert_se(sd_netlink_message_read_string(m, IFLA_IFNAME, &data) >= 0); | |
185 | assert_se(streq(data, "lo")); | |
186 | ||
187 | return 1; | |
188 | } | |
189 | ||
190 | static void test_event_loop(int ifindex) { | |
4afd3348 LP |
191 | _cleanup_(sd_event_unrefp) sd_event *event = NULL; |
192 | _cleanup_(sd_netlink_unrefp) sd_netlink *rtnl = NULL; | |
193 | _cleanup_(sd_netlink_message_unrefp) sd_netlink_message *m = NULL; | |
1c4baffc TG |
194 | char *ifname; |
195 | ||
196 | ifname = strdup("lo2"); | |
197 | assert_se(ifname); | |
198 | ||
199 | assert_se(sd_netlink_open(&rtnl) >= 0); | |
200 | assert_se(sd_rtnl_message_new_link(rtnl, &m, RTM_GETLINK, ifindex) >= 0); | |
201 | ||
545bab1f | 202 | assert_se(sd_netlink_call_async(rtnl, m, link_handler, NULL, ifname, 0, NULL) >= 0); |
1c4baffc TG |
203 | |
204 | assert_se(sd_event_default(&event) >= 0); | |
205 | ||
206 | assert_se(sd_netlink_attach_event(rtnl, event, 0) >= 0); | |
207 | ||
208 | assert_se(sd_event_run(event, 0) >= 0); | |
209 | ||
210 | assert_se(sd_netlink_detach_event(rtnl) >= 0); | |
211 | ||
212 | assert_se((rtnl = sd_netlink_unref(rtnl)) == NULL); | |
213 | } | |
214 | ||
215 | static int pipe_handler(sd_netlink *rtnl, sd_netlink_message *m, void *userdata) { | |
216 | int *counter = userdata; | |
217 | int r; | |
218 | ||
313cefa1 | 219 | (*counter)--; |
1c4baffc TG |
220 | |
221 | r = sd_netlink_message_get_errno(m); | |
222 | ||
223 | log_info_errno(r, "%d left in pipe. got reply: %m", *counter); | |
224 | ||
225 | assert_se(r >= 0); | |
226 | ||
227 | return 1; | |
228 | } | |
229 | ||
230 | static void test_async(int ifindex) { | |
4afd3348 LP |
231 | _cleanup_(sd_netlink_unrefp) sd_netlink *rtnl = NULL; |
232 | _cleanup_(sd_netlink_message_unrefp) sd_netlink_message *m = NULL, *r = NULL; | |
1c4baffc TG |
233 | uint32_t serial; |
234 | char *ifname; | |
235 | ||
236 | ifname = strdup("lo"); | |
237 | assert_se(ifname); | |
238 | ||
239 | assert_se(sd_netlink_open(&rtnl) >= 0); | |
240 | ||
241 | assert_se(sd_rtnl_message_new_link(rtnl, &m, RTM_GETLINK, ifindex) >= 0); | |
242 | ||
545bab1f | 243 | assert_se(sd_netlink_call_async(rtnl, m, link_handler, NULL, ifname, 0, &serial) >= 0); |
1c4baffc TG |
244 | |
245 | assert_se(sd_netlink_wait(rtnl, 0) >= 0); | |
246 | assert_se(sd_netlink_process(rtnl, &r) >= 0); | |
247 | ||
248 | assert_se((rtnl = sd_netlink_unref(rtnl)) == NULL); | |
249 | } | |
250 | ||
251 | static void test_pipe(int ifindex) { | |
4afd3348 LP |
252 | _cleanup_(sd_netlink_unrefp) sd_netlink *rtnl = NULL; |
253 | _cleanup_(sd_netlink_message_unrefp) sd_netlink_message *m1 = NULL, *m2 = NULL; | |
1c4baffc TG |
254 | int counter = 0; |
255 | ||
256 | assert_se(sd_netlink_open(&rtnl) >= 0); | |
257 | ||
258 | assert_se(sd_rtnl_message_new_link(rtnl, &m1, RTM_GETLINK, ifindex) >= 0); | |
259 | assert_se(sd_rtnl_message_new_link(rtnl, &m2, RTM_GETLINK, ifindex) >= 0); | |
260 | ||
313cefa1 | 261 | counter++; |
545bab1f | 262 | assert_se(sd_netlink_call_async(rtnl, m1, pipe_handler, NULL, &counter, 0, NULL) >= 0); |
1c4baffc | 263 | |
313cefa1 | 264 | counter++; |
545bab1f | 265 | assert_se(sd_netlink_call_async(rtnl, m2, pipe_handler, NULL, &counter, 0, NULL) >= 0); |
1c4baffc TG |
266 | |
267 | while (counter > 0) { | |
268 | assert_se(sd_netlink_wait(rtnl, 0) >= 0); | |
269 | assert_se(sd_netlink_process(rtnl, NULL) >= 0); | |
270 | } | |
271 | ||
272 | assert_se((rtnl = sd_netlink_unref(rtnl)) == NULL); | |
273 | } | |
274 | ||
05d0c2e3 | 275 | static void test_container(sd_netlink *rtnl) { |
4afd3348 | 276 | _cleanup_(sd_netlink_message_unrefp) sd_netlink_message *m = NULL; |
1c4baffc TG |
277 | uint16_t u16_data; |
278 | uint32_t u32_data; | |
279 | const char *string_data; | |
280 | ||
05d0c2e3 | 281 | assert_se(sd_rtnl_message_new_link(rtnl, &m, RTM_NEWLINK, 0) >= 0); |
1c4baffc TG |
282 | |
283 | assert_se(sd_netlink_message_open_container(m, IFLA_LINKINFO) >= 0); | |
284 | assert_se(sd_netlink_message_open_container_union(m, IFLA_INFO_DATA, "vlan") >= 0); | |
285 | assert_se(sd_netlink_message_append_u16(m, IFLA_VLAN_ID, 100) >= 0); | |
286 | assert_se(sd_netlink_message_close_container(m) >= 0); | |
287 | assert_se(sd_netlink_message_append_string(m, IFLA_INFO_KIND, "vlan") >= 0); | |
288 | assert_se(sd_netlink_message_close_container(m) >= 0); | |
289 | assert_se(sd_netlink_message_close_container(m) == -EINVAL); | |
290 | ||
291 | assert_se(sd_netlink_message_rewind(m) >= 0); | |
292 | ||
293 | assert_se(sd_netlink_message_enter_container(m, IFLA_LINKINFO) >= 0); | |
294 | assert_se(sd_netlink_message_read_string(m, IFLA_INFO_KIND, &string_data) >= 0); | |
295 | assert_se(streq("vlan", string_data)); | |
296 | ||
297 | assert_se(sd_netlink_message_enter_container(m, IFLA_INFO_DATA) >= 0); | |
298 | assert_se(sd_netlink_message_read_u16(m, IFLA_VLAN_ID, &u16_data) >= 0); | |
299 | assert_se(sd_netlink_message_exit_container(m) >= 0); | |
300 | ||
301 | assert_se(sd_netlink_message_read_string(m, IFLA_INFO_KIND, &string_data) >= 0); | |
302 | assert_se(streq("vlan", string_data)); | |
303 | assert_se(sd_netlink_message_exit_container(m) >= 0); | |
304 | ||
305 | assert_se(sd_netlink_message_read_u32(m, IFLA_LINKINFO, &u32_data) < 0); | |
306 | ||
307 | assert_se(sd_netlink_message_exit_container(m) == -EINVAL); | |
308 | } | |
309 | ||
310 | static void test_match(void) { | |
4afd3348 | 311 | _cleanup_(sd_netlink_unrefp) sd_netlink *rtnl = NULL; |
1c4baffc TG |
312 | |
313 | assert_se(sd_netlink_open(&rtnl) >= 0); | |
314 | ||
483d099e ZJS |
315 | assert_se(sd_netlink_add_match(rtnl, RTM_NEWLINK, link_handler, NULL) >= 0); |
316 | assert_se(sd_netlink_add_match(rtnl, RTM_NEWLINK, link_handler, NULL) >= 0); | |
1c4baffc | 317 | |
483d099e ZJS |
318 | assert_se(sd_netlink_remove_match(rtnl, RTM_NEWLINK, link_handler, NULL) == 1); |
319 | assert_se(sd_netlink_remove_match(rtnl, RTM_NEWLINK, link_handler, NULL) == 1); | |
320 | assert_se(sd_netlink_remove_match(rtnl, RTM_NEWLINK, link_handler, NULL) == 0); | |
1c4baffc TG |
321 | |
322 | assert_se((rtnl = sd_netlink_unref(rtnl)) == NULL); | |
323 | } | |
324 | ||
325 | static void test_get_addresses(sd_netlink *rtnl) { | |
4afd3348 | 326 | _cleanup_(sd_netlink_message_unrefp) sd_netlink_message *req = NULL, *reply = NULL; |
1c4baffc TG |
327 | sd_netlink_message *m; |
328 | ||
329 | assert_se(sd_rtnl_message_new_addr(rtnl, &req, RTM_GETADDR, 0, AF_UNSPEC) >= 0); | |
330 | ||
331 | assert_se(sd_netlink_call(rtnl, req, 0, &reply) >= 0); | |
332 | ||
333 | for (m = reply; m; m = sd_netlink_message_next(m)) { | |
334 | uint16_t type; | |
335 | unsigned char scope, flags; | |
336 | int family, ifindex; | |
337 | ||
338 | assert_se(sd_netlink_message_get_type(m, &type) >= 0); | |
339 | assert_se(type == RTM_NEWADDR); | |
340 | ||
341 | assert_se(sd_rtnl_message_addr_get_ifindex(m, &ifindex) >= 0); | |
342 | assert_se(sd_rtnl_message_addr_get_family(m, &family) >= 0); | |
343 | assert_se(sd_rtnl_message_addr_get_scope(m, &scope) >= 0); | |
344 | assert_se(sd_rtnl_message_addr_get_flags(m, &flags) >= 0); | |
345 | ||
346 | assert_se(ifindex > 0); | |
945c2931 | 347 | assert_se(IN_SET(family, AF_INET, AF_INET6)); |
1c4baffc TG |
348 | |
349 | log_info("got IPv%u address on ifindex %i", family == AF_INET ? 4: 6, ifindex); | |
350 | } | |
351 | } | |
352 | ||
05d0c2e3 | 353 | static void test_message(sd_netlink *rtnl) { |
4afd3348 | 354 | _cleanup_(sd_netlink_message_unrefp) sd_netlink_message *m = NULL; |
1c4baffc | 355 | |
05d0c2e3 | 356 | assert_se(rtnl_message_new_synthetic_error(rtnl, -ETIMEDOUT, 1, &m) >= 0); |
1c4baffc TG |
357 | assert_se(sd_netlink_message_get_errno(m) == -ETIMEDOUT); |
358 | } | |
359 | ||
360 | int main(void) { | |
361 | sd_netlink *rtnl; | |
362 | sd_netlink_message *m; | |
363 | sd_netlink_message *r; | |
364 | const char *string_data; | |
365 | int if_loopback; | |
366 | uint16_t type; | |
367 | ||
1c4baffc TG |
368 | test_match(); |
369 | ||
370 | test_multiple(); | |
371 | ||
1c4baffc TG |
372 | assert_se(sd_netlink_open(&rtnl) >= 0); |
373 | assert_se(rtnl); | |
374 | ||
05d0c2e3 JT |
375 | test_route(rtnl); |
376 | ||
377 | test_message(rtnl); | |
378 | ||
379 | test_container(rtnl); | |
380 | ||
1c4baffc TG |
381 | if_loopback = (int) if_nametoindex("lo"); |
382 | assert_se(if_loopback > 0); | |
383 | ||
384 | test_async(if_loopback); | |
385 | ||
386 | test_pipe(if_loopback); | |
387 | ||
388 | test_event_loop(if_loopback); | |
389 | ||
390 | test_link_configure(rtnl, if_loopback); | |
391 | ||
392 | test_get_addresses(rtnl); | |
393 | ||
394 | test_message_link_bridge(rtnl); | |
395 | ||
396 | assert_se(sd_rtnl_message_new_link(rtnl, &m, RTM_GETLINK, if_loopback) >= 0); | |
397 | assert_se(m); | |
398 | ||
399 | assert_se(sd_netlink_message_get_type(m, &type) >= 0); | |
400 | assert_se(type == RTM_GETLINK); | |
401 | ||
402 | assert_se(sd_netlink_message_read_string(m, IFLA_IFNAME, &string_data) == -EPERM); | |
403 | ||
404 | assert_se(sd_netlink_call(rtnl, m, 0, &r) == 1); | |
405 | assert_se(sd_netlink_message_get_type(r, &type) >= 0); | |
406 | assert_se(type == RTM_NEWLINK); | |
407 | ||
408 | assert_se((r = sd_netlink_message_unref(r)) == NULL); | |
409 | ||
410 | assert_se(sd_netlink_call(rtnl, m, -1, &r) == -EPERM); | |
411 | assert_se((m = sd_netlink_message_unref(m)) == NULL); | |
412 | assert_se((r = sd_netlink_message_unref(r)) == NULL); | |
413 | ||
414 | test_link_get(rtnl, if_loopback); | |
415 | test_address_get(rtnl, if_loopback); | |
416 | ||
1c4baffc TG |
417 | assert_se((m = sd_netlink_message_unref(m)) == NULL); |
418 | assert_se((r = sd_netlink_message_unref(r)) == NULL); | |
419 | assert_se((rtnl = sd_netlink_unref(rtnl)) == NULL); | |
420 | ||
421 | return EXIT_SUCCESS; | |
422 | } |