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