]> git.ipfire.org Git - thirdparty/systemd.git/blob - src/libsystemd/sd-netlink/test-netlink.c
sd-netlink: rename from sd-rtnl
[thirdparty/systemd.git] / src / libsystemd / sd-netlink / test-netlink.c
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
22 #include <netinet/ether.h>
23 #include <net/if.h>
24
25 #include "util.h"
26 #include "macro.h"
27 #include "sd-netlink.h"
28 #include "socket-util.h"
29 #include "netlink-util.h"
30 #include "event-util.h"
31 #include "missing.h"
32
33 static void test_message_link_bridge(sd_netlink *rtnl) {
34 _cleanup_netlink_message_unref_ sd_netlink_message *message = NULL;
35 uint32_t cost;
36
37 assert_se(sd_rtnl_message_new_link(rtnl, &message, RTM_NEWLINK, 1) >= 0);
38 assert_se(sd_rtnl_message_link_set_family(message, PF_BRIDGE) >= 0);
39 assert_se(sd_netlink_message_open_container(message, IFLA_PROTINFO) >= 0);
40 assert_se(sd_netlink_message_append_u32(message, IFLA_BRPORT_COST, 10) >= 0);
41 assert_se(sd_netlink_message_close_container(message) >= 0);
42
43 assert_se(sd_netlink_message_rewind(message) >= 0);
44
45 assert_se(sd_netlink_message_enter_container(message, IFLA_PROTINFO) >= 0);
46 assert_se(sd_netlink_message_read_u32(message, IFLA_BRPORT_COST, &cost) >= 0);
47 assert_se(cost == 10);
48 assert_se(sd_netlink_message_exit_container(message) >= 0);
49 }
50
51 static void test_link_configure(sd_netlink *rtnl, int ifindex) {
52 _cleanup_netlink_message_unref_ sd_netlink_message *message = NULL;
53 const char *mac = "98:fe:94:3f:c6:18", *name = "test";
54 char buffer[ETHER_ADDR_TO_STRING_MAX];
55 unsigned int mtu = 1450, mtu_out;
56 const char *name_out;
57 struct ether_addr mac_out;
58
59 /* we'd really like to test NEWLINK, but let's not mess with the running kernel */
60 assert_se(sd_rtnl_message_new_link(rtnl, &message, RTM_GETLINK, ifindex) >= 0);
61 assert_se(sd_netlink_message_append_string(message, IFLA_IFNAME, name) >= 0);
62 assert_se(sd_netlink_message_append_ether_addr(message, IFLA_ADDRESS, ether_aton(mac)) >= 0);
63 assert_se(sd_netlink_message_append_u32(message, IFLA_MTU, mtu) >= 0);
64
65 assert_se(sd_netlink_call(rtnl, message, 0, NULL) == 1);
66 assert_se(sd_netlink_message_rewind(message) >= 0);
67
68 assert_se(sd_netlink_message_read_string(message, IFLA_IFNAME, &name_out) >= 0);
69 assert_se(streq(name, name_out));
70
71 assert_se(sd_netlink_message_read_ether_addr(message, IFLA_ADDRESS, &mac_out) >= 0);
72 assert_se(streq(mac, ether_addr_to_string(&mac_out, buffer)));
73
74 assert_se(sd_netlink_message_read_u32(message, IFLA_MTU, &mtu_out) >= 0);
75 assert_se(mtu == mtu_out);
76 }
77
78 static void test_link_get(sd_netlink *rtnl, int ifindex) {
79 sd_netlink_message *m;
80 sd_netlink_message *r;
81 unsigned int mtu = 1500;
82 const char *str_data;
83 uint8_t u8_data;
84 uint32_t u32_data;
85 struct ether_addr eth_data;
86
87 assert_se(sd_rtnl_message_new_link(rtnl, &m, RTM_GETLINK, ifindex) >= 0);
88 assert_se(m);
89
90 /* u8 test cases */
91 assert_se(sd_netlink_message_append_u8(m, IFLA_CARRIER, 0) >= 0);
92 assert_se(sd_netlink_message_append_u8(m, IFLA_OPERSTATE, 0) >= 0);
93 assert_se(sd_netlink_message_append_u8(m, IFLA_LINKMODE, 0) >= 0);
94
95 /* u32 test cases */
96 assert_se(sd_netlink_message_append_u32(m, IFLA_MTU, mtu) >= 0);
97 assert_se(sd_netlink_message_append_u32(m, IFLA_GROUP, 0) >= 0);
98 assert_se(sd_netlink_message_append_u32(m, IFLA_TXQLEN, 0) >= 0);
99 assert_se(sd_netlink_message_append_u32(m, IFLA_NUM_TX_QUEUES, 0) >= 0);
100 assert_se(sd_netlink_message_append_u32(m, IFLA_NUM_RX_QUEUES, 0) >= 0);
101
102 assert_se(sd_netlink_call(rtnl, m, -1, &r) == 1);
103
104 assert_se(sd_netlink_message_read_string(r, IFLA_IFNAME, &str_data) == 0);
105
106 assert_se(sd_netlink_message_read_u8(r, IFLA_CARRIER, &u8_data) == 0);
107 assert_se(sd_netlink_message_read_u8(r, IFLA_OPERSTATE, &u8_data) == 0);
108 assert_se(sd_netlink_message_read_u8(r, IFLA_LINKMODE, &u8_data) == 0);
109
110 assert_se(sd_netlink_message_read_u32(r, IFLA_MTU, &u32_data) == 0);
111 assert_se(sd_netlink_message_read_u32(r, IFLA_GROUP, &u32_data) == 0);
112 assert_se(sd_netlink_message_read_u32(r, IFLA_TXQLEN, &u32_data) == 0);
113 assert_se(sd_netlink_message_read_u32(r, IFLA_NUM_TX_QUEUES, &u32_data) == 0);
114 assert_se(sd_netlink_message_read_u32(r, IFLA_NUM_RX_QUEUES, &u32_data) == 0);
115
116 assert_se(sd_netlink_message_read_ether_addr(r, IFLA_ADDRESS, &eth_data) == 0);
117
118 assert_se(sd_netlink_flush(rtnl) >= 0);
119 assert_se((m = sd_netlink_message_unref(m)) == NULL);
120 assert_se((r = sd_netlink_message_unref(r)) == NULL);
121 }
122
123
124 static void test_address_get(sd_netlink *rtnl, int ifindex) {
125 sd_netlink_message *m;
126 sd_netlink_message *r;
127 struct in_addr in_data;
128 struct ifa_cacheinfo cache;
129 const char *label;
130
131 assert_se(sd_rtnl_message_new_addr(rtnl, &m, RTM_GETADDR, ifindex, AF_INET) >= 0);
132 assert_se(m);
133
134 assert_se(sd_netlink_call(rtnl, m, -1, &r) == 1);
135
136 assert_se(sd_netlink_message_read_in_addr(r, IFA_LOCAL, &in_data) == 0);
137 assert_se(sd_netlink_message_read_in_addr(r, IFA_ADDRESS, &in_data) == 0);
138 assert_se(sd_netlink_message_read_string(r, IFA_LABEL, &label) == 0);
139 assert_se(sd_netlink_message_read_cache_info(r, IFA_CACHEINFO, &cache) == 0);
140
141 assert_se(sd_netlink_flush(rtnl) >= 0);
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) {
148 _cleanup_netlink_message_unref_ sd_netlink_message *req;
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) {
211 _cleanup_event_unref_ sd_event *event = NULL;
212 _cleanup_netlink_unref_ sd_netlink *rtnl = NULL;
213 _cleanup_netlink_message_unref_ sd_netlink_message *m = NULL;
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) {
251 _cleanup_netlink_unref_ sd_netlink *rtnl = NULL;
252 _cleanup_netlink_message_unref_ sd_netlink_message *m = NULL, *r = NULL;
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) {
272 _cleanup_netlink_unref_ sd_netlink *rtnl = NULL;
273 _cleanup_netlink_message_unref_ sd_netlink_message *m1 = NULL, *m2 = NULL;
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) {
296 _cleanup_netlink_message_unref_ sd_netlink_message *m = NULL;
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) {
331 _cleanup_netlink_unref_ sd_netlink *rtnl = NULL;
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) {
346 _cleanup_netlink_message_unref_ sd_netlink_message *req = NULL, *reply = NULL;
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) {
374 _cleanup_netlink_message_unref_ sd_netlink_message *m = NULL;
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
437 assert_se(sd_netlink_flush(rtnl) >= 0);
438 assert_se((m = sd_netlink_message_unref(m)) == NULL);
439 assert_se((r = sd_netlink_message_unref(r)) == NULL);
440 assert_se((rtnl = sd_netlink_unref(rtnl)) == NULL);
441
442 return EXIT_SUCCESS;
443 }