]>
Commit | Line | Data |
---|---|---|
aba5acdf SH |
1 | /* |
2 | * libnetlink.c RTnetlink service routines. | |
3 | * | |
4 | * This program is free software; you can redistribute it and/or | |
5 | * modify it under the terms of the GNU General Public License | |
6 | * as published by the Free Software Foundation; either version | |
7 | * 2 of the License, or (at your option) any later version. | |
8 | * | |
9 | * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru> | |
10 | * | |
11 | */ | |
12 | ||
13 | #include <stdio.h> | |
14 | #include <stdlib.h> | |
463d9efa | 15 | #include <stdbool.h> |
aba5acdf | 16 | #include <unistd.h> |
aba5acdf SH |
17 | #include <fcntl.h> |
18 | #include <net/if_arp.h> | |
19 | #include <sys/socket.h> | |
20 | #include <netinet/in.h> | |
21 | #include <string.h> | |
22 | #include <errno.h> | |
23 | #include <time.h> | |
24 | #include <sys/uio.h> | |
39360023 | 25 | #include <linux/if_addrlabel.h> |
aba5acdf SH |
26 | |
27 | #include "libnetlink.h" | |
28 | ||
449b824a ND |
29 | #ifndef SOL_NETLINK |
30 | #define SOL_NETLINK 270 | |
31 | #endif | |
32 | ||
c079e121 SH |
33 | #ifndef MIN |
34 | #define MIN(a, b) ((a) < (b) ? (a) : (b)) | |
35 | #endif | |
36 | ||
7f03191f PM |
37 | int rcvbuf = 1024 * 1024; |
38 | ||
b6432e68 SH |
39 | #ifdef HAVE_LIBMNL |
40 | #include <libmnl/libmnl.h> | |
41 | ||
42 | static const enum mnl_attr_data_type extack_policy[NLMSGERR_ATTR_MAX + 1] = { | |
43 | [NLMSGERR_ATTR_MSG] = MNL_TYPE_NUL_STRING, | |
44 | [NLMSGERR_ATTR_OFFS] = MNL_TYPE_U32, | |
45 | }; | |
46 | ||
47 | static int err_attr_cb(const struct nlattr *attr, void *data) | |
48 | { | |
49 | const struct nlattr **tb = data; | |
50 | uint16_t type; | |
51 | ||
e5fa0e6f DA |
52 | if (mnl_attr_type_valid(attr, NLMSGERR_ATTR_MAX) < 0) { |
53 | fprintf(stderr, "Invalid extack attribute\n"); | |
b6432e68 | 54 | return MNL_CB_ERROR; |
e5fa0e6f | 55 | } |
b6432e68 SH |
56 | |
57 | type = mnl_attr_get_type(attr); | |
e5fa0e6f DA |
58 | if (mnl_attr_validate(attr, extack_policy[type]) < 0) { |
59 | fprintf(stderr, "extack attribute %d failed validation\n", | |
60 | type); | |
b6432e68 | 61 | return MNL_CB_ERROR; |
e5fa0e6f | 62 | } |
b6432e68 SH |
63 | |
64 | tb[type] = attr; | |
65 | return MNL_CB_OK; | |
66 | } | |
67 | ||
b6432e68 | 68 | /* dump netlink extended ack error message */ |
049c5853 | 69 | int nl_dump_ext_ack(const struct nlmsghdr *nlh, nl_ext_ack_fn_t errfn) |
b6432e68 | 70 | { |
e5fa0e6f | 71 | struct nlattr *tb[NLMSGERR_ATTR_MAX + 1] = {}; |
b6432e68 SH |
72 | const struct nlmsgerr *err = mnl_nlmsg_get_payload(nlh); |
73 | const struct nlmsghdr *err_nlh = NULL; | |
74 | unsigned int hlen = sizeof(*err); | |
844c37b4 | 75 | const char *msg = NULL; |
b6432e68 SH |
76 | uint32_t off = 0; |
77 | ||
b6432e68 SH |
78 | /* no TLVs, nothing to do here */ |
79 | if (!(nlh->nlmsg_flags & NLM_F_ACK_TLVS)) | |
80 | return 0; | |
81 | ||
82 | /* if NLM_F_CAPPED is set then the inner err msg was capped */ | |
83 | if (!(nlh->nlmsg_flags & NLM_F_CAPPED)) | |
84 | hlen += mnl_nlmsg_get_payload_len(&err->msg); | |
85 | ||
e5fa0e6f DA |
86 | if (mnl_attr_parse(nlh, hlen, err_attr_cb, tb) != MNL_CB_OK) |
87 | return 0; | |
b6432e68 SH |
88 | |
89 | if (tb[NLMSGERR_ATTR_MSG]) | |
844c37b4 | 90 | msg = mnl_attr_get_str(tb[NLMSGERR_ATTR_MSG]); |
b6432e68 SH |
91 | |
92 | if (tb[NLMSGERR_ATTR_OFFS]) { | |
93 | off = mnl_attr_get_u32(tb[NLMSGERR_ATTR_OFFS]); | |
94 | ||
95 | if (off > nlh->nlmsg_len) { | |
96 | fprintf(stderr, | |
97 | "Invalid offset for NLMSGERR_ATTR_OFFS\n"); | |
98 | off = 0; | |
99 | } else if (!(nlh->nlmsg_flags & NLM_F_CAPPED)) | |
100 | err_nlh = &err->msg; | |
101 | } | |
102 | ||
fb6cb307 | 103 | if (errfn) |
844c37b4 | 104 | return errfn(msg, off, err_nlh); |
fb6cb307 | 105 | |
844c37b4 DA |
106 | if (msg && *msg != '\0') { |
107 | bool is_err = !!err->error; | |
108 | ||
109 | fprintf(stderr, "%s: %s", | |
110 | is_err ? "Error" : "Warning", msg); | |
111 | if (msg[strlen(msg) - 1] != '.') | |
fb6cb307 DA |
112 | fprintf(stderr, "."); |
113 | fprintf(stderr, "\n"); | |
114 | ||
844c37b4 | 115 | return is_err ? 1 : 0; |
fb6cb307 DA |
116 | } |
117 | ||
118 | return 0; | |
b6432e68 SH |
119 | } |
120 | #else | |
7d23fa55 SH |
121 | #warning "libmnl required for error support" |
122 | ||
b6432e68 | 123 | /* No extended error ack without libmnl */ |
049c5853 | 124 | int nl_dump_ext_ack(const struct nlmsghdr *nlh, nl_ext_ack_fn_t errfn) |
b6432e68 SH |
125 | { |
126 | return 0; | |
127 | } | |
128 | #endif | |
129 | ||
aba5acdf SH |
130 | void rtnl_close(struct rtnl_handle *rth) |
131 | { | |
3bfa73ff SH |
132 | if (rth->fd >= 0) { |
133 | close(rth->fd); | |
134 | rth->fd = -1; | |
135 | } | |
aba5acdf SH |
136 | } |
137 | ||
2c500a4d | 138 | int rtnl_open_byproto(struct rtnl_handle *rth, unsigned int subscriptions, |
8ed63ab1 | 139 | int protocol) |
aba5acdf | 140 | { |
f332d169 | 141 | socklen_t addr_len; |
007d3a3e | 142 | int sndbuf = 32768; |
b6432e68 | 143 | int one = 1; |
aba5acdf | 144 | |
b16621ca | 145 | memset(rth, 0, sizeof(*rth)); |
aba5acdf | 146 | |
8a4025f6 | 147 | rth->proto = protocol; |
bcb9d403 | 148 | rth->fd = socket(AF_NETLINK, SOCK_RAW | SOCK_CLOEXEC, protocol); |
aba5acdf SH |
149 | if (rth->fd < 0) { |
150 | perror("Cannot open netlink socket"); | |
151 | return -1; | |
152 | } | |
153 | ||
2c500a4d SH |
154 | if (setsockopt(rth->fd, SOL_SOCKET, SO_SNDBUF, |
155 | &sndbuf, sizeof(sndbuf)) < 0) { | |
007d3a3e SH |
156 | perror("SO_SNDBUF"); |
157 | return -1; | |
158 | } | |
159 | ||
2c500a4d SH |
160 | if (setsockopt(rth->fd, SOL_SOCKET, SO_RCVBUF, |
161 | &rcvbuf, sizeof(rcvbuf)) < 0) { | |
007d3a3e SH |
162 | perror("SO_RCVBUF"); |
163 | return -1; | |
164 | } | |
165 | ||
b6432e68 SH |
166 | /* Older kernels may no support extended ACK reporting */ |
167 | setsockopt(rth->fd, SOL_NETLINK, NETLINK_EXT_ACK, | |
168 | &one, sizeof(one)); | |
169 | ||
aba5acdf SH |
170 | memset(&rth->local, 0, sizeof(rth->local)); |
171 | rth->local.nl_family = AF_NETLINK; | |
172 | rth->local.nl_groups = subscriptions; | |
173 | ||
2c500a4d SH |
174 | if (bind(rth->fd, (struct sockaddr *)&rth->local, |
175 | sizeof(rth->local)) < 0) { | |
aba5acdf SH |
176 | perror("Cannot bind netlink socket"); |
177 | return -1; | |
178 | } | |
179 | addr_len = sizeof(rth->local); | |
2c500a4d SH |
180 | if (getsockname(rth->fd, (struct sockaddr *)&rth->local, |
181 | &addr_len) < 0) { | |
aba5acdf SH |
182 | perror("Cannot getsockname"); |
183 | return -1; | |
184 | } | |
185 | if (addr_len != sizeof(rth->local)) { | |
186 | fprintf(stderr, "Wrong address length %d\n", addr_len); | |
187 | return -1; | |
188 | } | |
189 | if (rth->local.nl_family != AF_NETLINK) { | |
2c500a4d SH |
190 | fprintf(stderr, "Wrong address family %d\n", |
191 | rth->local.nl_family); | |
aba5acdf SH |
192 | return -1; |
193 | } | |
194 | rth->seq = time(NULL); | |
195 | return 0; | |
196 | } | |
197 | ||
2c500a4d | 198 | int rtnl_open(struct rtnl_handle *rth, unsigned int subscriptions) |
c7699875 | 199 | { |
200 | return rtnl_open_byproto(rth, subscriptions, NETLINK_ROUTE); | |
201 | } | |
202 | ||
46917d08 DA |
203 | int rtnl_addrdump_req(struct rtnl_handle *rth, int family) |
204 | { | |
205 | struct { | |
206 | struct nlmsghdr nlh; | |
207 | struct ifaddrmsg ifm; | |
208 | } req = { | |
209 | .nlh.nlmsg_len = sizeof(req), | |
210 | .nlh.nlmsg_type = RTM_GETADDR, | |
211 | .nlh.nlmsg_flags = NLM_F_DUMP | NLM_F_REQUEST, | |
212 | .nlh.nlmsg_seq = rth->dump = ++rth->seq, | |
213 | .ifm.ifa_family = family, | |
214 | }; | |
215 | ||
216 | return send(rth->fd, &req, sizeof(req), 0); | |
217 | } | |
218 | ||
39360023 DA |
219 | int rtnl_addrlbldump_req(struct rtnl_handle *rth, int family) |
220 | { | |
221 | struct { | |
222 | struct nlmsghdr nlh; | |
223 | struct ifaddrlblmsg ifal; | |
224 | } req = { | |
225 | .nlh.nlmsg_len = sizeof(req), | |
226 | .nlh.nlmsg_type = RTM_GETADDRLABEL, | |
227 | .nlh.nlmsg_flags = NLM_F_DUMP | NLM_F_REQUEST, | |
228 | .nlh.nlmsg_seq = rth->dump = ++rth->seq, | |
229 | .ifal.ifal_family = family, | |
230 | }; | |
231 | ||
232 | return send(rth->fd, &req, sizeof(req), 0); | |
233 | } | |
234 | ||
bfb27dfa DA |
235 | int rtnl_routedump_req(struct rtnl_handle *rth, int family) |
236 | { | |
237 | struct { | |
238 | struct nlmsghdr nlh; | |
239 | struct rtmsg rtm; | |
240 | } req = { | |
241 | .nlh.nlmsg_len = sizeof(req), | |
242 | .nlh.nlmsg_type = RTM_GETROUTE, | |
243 | .nlh.nlmsg_flags = NLM_F_DUMP | NLM_F_REQUEST, | |
244 | .nlh.nlmsg_seq = rth->dump = ++rth->seq, | |
245 | .rtm.rtm_family = family, | |
246 | }; | |
247 | ||
248 | return send(rth->fd, &req, sizeof(req), 0); | |
249 | } | |
250 | ||
aba5acdf | 251 | int rtnl_wilddump_request(struct rtnl_handle *rth, int family, int type) |
9eff0e5c VY |
252 | { |
253 | return rtnl_wilddump_req_filter(rth, family, type, RTEXT_FILTER_VF); | |
254 | } | |
255 | ||
256 | int rtnl_wilddump_req_filter(struct rtnl_handle *rth, int family, int type, | |
257 | __u32 filt_mask) | |
aba5acdf SH |
258 | { |
259 | struct { | |
260 | struct nlmsghdr nlh; | |
63338dca | 261 | struct ifinfomsg ifm; |
257422f7 LJ |
262 | /* attribute has to be NLMSG aligned */ |
263 | struct rtattr ext_req __attribute__ ((aligned(NLMSG_ALIGNTO))); | |
bd886ebb | 264 | __u32 ext_filter_mask; |
d17b136f PS |
265 | } req = { |
266 | .nlh.nlmsg_len = sizeof(req), | |
267 | .nlh.nlmsg_type = type, | |
268 | .nlh.nlmsg_flags = NLM_F_DUMP | NLM_F_REQUEST, | |
269 | .nlh.nlmsg_seq = rth->dump = ++rth->seq, | |
270 | .ifm.ifi_family = family, | |
271 | .ext_req.rta_type = IFLA_EXT_MASK, | |
272 | .ext_req.rta_len = RTA_LENGTH(sizeof(__u32)), | |
273 | .ext_filter_mask = filt_mask, | |
274 | }; | |
bd886ebb | 275 | |
2c500a4d | 276 | return send(rth->fd, &req, sizeof(req), 0); |
aba5acdf SH |
277 | } |
278 | ||
b0a4ce62 DA |
279 | int rtnl_wilddump_req_filter_fn(struct rtnl_handle *rth, int family, int type, |
280 | req_filter_fn_t filter_fn) | |
281 | { | |
282 | struct { | |
283 | struct nlmsghdr nlh; | |
284 | struct ifinfomsg ifm; | |
285 | char buf[1024]; | |
d17b136f PS |
286 | } req = { |
287 | .nlh.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifinfomsg)), | |
288 | .nlh.nlmsg_type = type, | |
289 | .nlh.nlmsg_flags = NLM_F_DUMP | NLM_F_REQUEST, | |
290 | .nlh.nlmsg_seq = rth->dump = ++rth->seq, | |
291 | .ifm.ifi_family = family, | |
292 | }; | |
b0a4ce62 DA |
293 | int err; |
294 | ||
295 | if (!filter_fn) | |
296 | return -EINVAL; | |
297 | ||
b0a4ce62 DA |
298 | err = filter_fn(&req.nlh, sizeof(req)); |
299 | if (err) | |
300 | return err; | |
301 | ||
1b109a30 | 302 | return send(rth->fd, &req, req.nlh.nlmsg_len, 0); |
b0a4ce62 DA |
303 | } |
304 | ||
7abf5de6 NA |
305 | int rtnl_wilddump_stats_req_filter(struct rtnl_handle *rth, int fam, int type, |
306 | __u32 filt_mask) | |
307 | { | |
308 | struct { | |
309 | struct nlmsghdr nlh; | |
310 | struct if_stats_msg ifsm; | |
311 | } req; | |
312 | ||
313 | memset(&req, 0, sizeof(req)); | |
314 | req.nlh.nlmsg_len = NLMSG_LENGTH(sizeof(struct if_stats_msg)); | |
315 | req.nlh.nlmsg_type = type; | |
316 | req.nlh.nlmsg_flags = NLM_F_DUMP|NLM_F_REQUEST; | |
317 | req.nlh.nlmsg_pid = 0; | |
318 | req.nlh.nlmsg_seq = rth->dump = ++rth->seq; | |
319 | req.ifsm.family = fam; | |
320 | req.ifsm.filter_mask = filt_mask; | |
321 | ||
2c500a4d | 322 | return send(rth->fd, &req, sizeof(req), 0); |
7abf5de6 NA |
323 | } |
324 | ||
6cf8398f | 325 | int rtnl_send(struct rtnl_handle *rth, const void *buf, int len) |
aba5acdf | 326 | { |
f31a37f7 SH |
327 | return send(rth->fd, buf, len, 0); |
328 | } | |
329 | ||
6cf8398f | 330 | int rtnl_send_check(struct rtnl_handle *rth, const void *buf, int len) |
f31a37f7 | 331 | { |
54bb35c6 SH |
332 | struct nlmsghdr *h; |
333 | int status; | |
334 | char resp[1024]; | |
aba5acdf | 335 | |
f31a37f7 | 336 | status = send(rth->fd, buf, len, 0); |
54bb35c6 SH |
337 | if (status < 0) |
338 | return status; | |
339 | ||
2d8240f8 SH |
340 | /* Check for immediate errors */ |
341 | status = recv(rth->fd, resp, sizeof(resp), MSG_DONTWAIT|MSG_PEEK); | |
54bb35c6 SH |
342 | if (status < 0) { |
343 | if (errno == EAGAIN) | |
344 | return 0; | |
345 | return -1; | |
346 | } | |
347 | ||
348 | for (h = (struct nlmsghdr *)resp; NLMSG_OK(h, status); | |
349 | h = NLMSG_NEXT(h, status)) { | |
350 | if (h->nlmsg_type == NLMSG_ERROR) { | |
2c500a4d SH |
351 | struct nlmsgerr *err = (struct nlmsgerr *)NLMSG_DATA(h); |
352 | ||
54bb35c6 SH |
353 | if (h->nlmsg_len < NLMSG_LENGTH(sizeof(struct nlmsgerr))) |
354 | fprintf(stderr, "ERROR truncated\n"); | |
e9e9365b | 355 | else |
54bb35c6 | 356 | errno = -err->error; |
24f38182 | 357 | return -1; |
54bb35c6 | 358 | } |
54bb35c6 SH |
359 | } |
360 | ||
361 | return 0; | |
aba5acdf SH |
362 | } |
363 | ||
364 | int rtnl_dump_request(struct rtnl_handle *rth, int type, void *req, int len) | |
365 | { | |
d17b136f PS |
366 | struct nlmsghdr nlh = { |
367 | .nlmsg_len = NLMSG_LENGTH(len), | |
368 | .nlmsg_type = type, | |
369 | .nlmsg_flags = NLM_F_DUMP | NLM_F_REQUEST, | |
370 | .nlmsg_seq = rth->dump = ++rth->seq, | |
371 | }; | |
6cf8398f | 372 | struct sockaddr_nl nladdr = { .nl_family = AF_NETLINK }; |
8ed63ab1 SH |
373 | struct iovec iov[2] = { |
374 | { .iov_base = &nlh, .iov_len = sizeof(nlh) }, | |
375 | { .iov_base = req, .iov_len = len } | |
376 | }; | |
aba5acdf | 377 | struct msghdr msg = { |
8ed63ab1 | 378 | .msg_name = &nladdr, |
d17b136f | 379 | .msg_namelen = sizeof(nladdr), |
8ed63ab1 SH |
380 | .msg_iov = iov, |
381 | .msg_iovlen = 2, | |
aba5acdf SH |
382 | }; |
383 | ||
aba5acdf SH |
384 | return sendmsg(rth->fd, &msg, 0); |
385 | } | |
386 | ||
0d238ca2 DA |
387 | int rtnl_dump_request_n(struct rtnl_handle *rth, struct nlmsghdr *n) |
388 | { | |
389 | struct sockaddr_nl nladdr = { .nl_family = AF_NETLINK }; | |
390 | struct iovec iov = { | |
2c500a4d | 391 | .iov_base = n, |
0d238ca2 DA |
392 | .iov_len = n->nlmsg_len |
393 | }; | |
394 | struct msghdr msg = { | |
395 | .msg_name = &nladdr, | |
396 | .msg_namelen = sizeof(nladdr), | |
397 | .msg_iov = &iov, | |
398 | .msg_iovlen = 1, | |
399 | }; | |
400 | ||
401 | n->nlmsg_flags = NLM_F_DUMP|NLM_F_REQUEST; | |
402 | n->nlmsg_pid = 0; | |
403 | n->nlmsg_seq = rth->dump = ++rth->seq; | |
404 | ||
405 | return sendmsg(rth->fd, &msg, 0); | |
406 | } | |
407 | ||
0efa6257 | 408 | static int rtnl_dump_done(struct nlmsghdr *h) |
892a25e2 SH |
409 | { |
410 | int len = *(int *)NLMSG_DATA(h); | |
411 | ||
05a14fc1 DA |
412 | if (h->nlmsg_len < NLMSG_LENGTH(sizeof(int))) { |
413 | fprintf(stderr, "DONE truncated\n"); | |
414 | return -1; | |
415 | } | |
892a25e2 | 416 | |
05a14fc1 DA |
417 | if (len < 0) { |
418 | errno = -len; | |
419 | switch (errno) { | |
420 | case ENOENT: | |
421 | case EOPNOTSUPP: | |
422 | return -1; | |
423 | case EMSGSIZE: | |
424 | fprintf(stderr, | |
425 | "Error: Buffer too small for object.\n"); | |
426 | break; | |
427 | default: | |
892a25e2 | 428 | perror("RTNETLINK answers"); |
892a25e2 | 429 | } |
05a14fc1 | 430 | return len; |
892a25e2 | 431 | } |
05a14fc1 | 432 | |
844c37b4 DA |
433 | /* check for any messages returned from kernel */ |
434 | nl_dump_ext_ack(h, NULL); | |
435 | ||
892a25e2 SH |
436 | return 0; |
437 | } | |
438 | ||
439 | static void rtnl_dump_error(const struct rtnl_handle *rth, | |
440 | struct nlmsghdr *h) | |
441 | { | |
442 | ||
443 | if (h->nlmsg_len < NLMSG_LENGTH(sizeof(struct nlmsgerr))) { | |
444 | fprintf(stderr, "ERROR truncated\n"); | |
445 | } else { | |
446 | const struct nlmsgerr *err = (struct nlmsgerr *)NLMSG_DATA(h); | |
447 | ||
448 | errno = -err->error; | |
449 | if (rth->proto == NETLINK_SOCK_DIAG && | |
450 | (errno == ENOENT || | |
451 | errno == EOPNOTSUPP)) | |
452 | return; | |
453 | ||
3ad6d176 DA |
454 | if (!(rth->flags & RTNL_HANDLE_F_SUPPRESS_NLERR)) |
455 | perror("RTNETLINK answers"); | |
892a25e2 SH |
456 | } |
457 | } | |
458 | ||
2d34851c HL |
459 | static int __rtnl_recvmsg(int fd, struct msghdr *msg, int flags) |
460 | { | |
461 | int len; | |
462 | ||
463 | do { | |
464 | len = recvmsg(fd, msg, flags); | |
465 | } while (len < 0 && (errno == EINTR || errno == EAGAIN)); | |
466 | ||
467 | if (len < 0) { | |
468 | fprintf(stderr, "netlink receive error %s (%d)\n", | |
469 | strerror(errno), errno); | |
470 | return -errno; | |
471 | } | |
472 | ||
473 | if (len == 0) { | |
474 | fprintf(stderr, "EOF on netlink\n"); | |
475 | return -ENODATA; | |
476 | } | |
477 | ||
478 | return len; | |
479 | } | |
480 | ||
481 | static int rtnl_recvmsg(int fd, struct msghdr *msg, char **answer) | |
482 | { | |
483 | struct iovec *iov = msg->msg_iov; | |
484 | char *buf; | |
485 | int len; | |
486 | ||
487 | iov->iov_base = NULL; | |
488 | iov->iov_len = 0; | |
489 | ||
490 | len = __rtnl_recvmsg(fd, msg, MSG_PEEK | MSG_TRUNC); | |
491 | if (len < 0) | |
492 | return len; | |
493 | ||
494 | buf = malloc(len); | |
495 | if (!buf) { | |
496 | fprintf(stderr, "malloc error: not enough buffer\n"); | |
497 | return -ENOMEM; | |
498 | } | |
499 | ||
500 | iov->iov_base = buf; | |
501 | iov->iov_len = len; | |
502 | ||
503 | len = __rtnl_recvmsg(fd, msg, 0); | |
504 | if (len < 0) { | |
505 | free(buf); | |
506 | return len; | |
507 | } | |
508 | ||
509 | if (answer) | |
510 | *answer = buf; | |
511 | else | |
512 | free(buf); | |
513 | ||
514 | return len; | |
515 | } | |
516 | ||
b49240ec SH |
517 | int rtnl_dump_filter_l(struct rtnl_handle *rth, |
518 | const struct rtnl_dump_filter_arg *arg) | |
aba5acdf | 519 | { |
aba5acdf | 520 | struct sockaddr_nl nladdr; |
8ed63ab1 SH |
521 | struct iovec iov; |
522 | struct msghdr msg = { | |
523 | .msg_name = &nladdr, | |
524 | .msg_namelen = sizeof(nladdr), | |
525 | .msg_iov = &iov, | |
526 | .msg_iovlen = 1, | |
527 | }; | |
2d34851c | 528 | char *buf; |
16f02e14 | 529 | int dump_intr = 0; |
aba5acdf SH |
530 | |
531 | while (1) { | |
532 | int status; | |
b49240ec | 533 | const struct rtnl_dump_filter_arg *a; |
3bc1c4f2 BG |
534 | int found_done = 0; |
535 | int msglen = 0; | |
aba5acdf | 536 | |
2d34851c HL |
537 | status = rtnl_recvmsg(rth->fd, &msg, &buf); |
538 | if (status < 0) | |
539 | return status; | |
aba5acdf | 540 | |
486ccd99 VK |
541 | if (rth->dump_fp) |
542 | fwrite(buf, 1, NLMSG_ALIGN(status), rth->dump_fp); | |
543 | ||
b49240ec | 544 | for (a = arg; a->filter; a++) { |
2c500a4d SH |
545 | struct nlmsghdr *h = (struct nlmsghdr *)buf; |
546 | ||
3bc1c4f2 | 547 | msglen = status; |
b49240ec | 548 | |
3bc1c4f2 | 549 | while (NLMSG_OK(h, msglen)) { |
486ccd99 | 550 | int err = 0; |
b49240ec | 551 | |
8e72880f PS |
552 | h->nlmsg_flags &= ~a->nc_flags; |
553 | ||
b49240ec SH |
554 | if (nladdr.nl_pid != 0 || |
555 | h->nlmsg_pid != rth->local.nl_pid || | |
cd70f3f5 | 556 | h->nlmsg_seq != rth->dump) |
b49240ec | 557 | goto skip_it; |
aba5acdf | 558 | |
16f02e14 ND |
559 | if (h->nlmsg_flags & NLM_F_DUMP_INTR) |
560 | dump_intr = 1; | |
561 | ||
3bc1c4f2 | 562 | if (h->nlmsg_type == NLMSG_DONE) { |
0efa6257 | 563 | err = rtnl_dump_done(h); |
2d34851c HL |
564 | if (err < 0) { |
565 | free(buf); | |
892a25e2 | 566 | return -1; |
2d34851c | 567 | } |
892a25e2 | 568 | |
3bc1c4f2 BG |
569 | found_done = 1; |
570 | break; /* process next filter */ | |
571 | } | |
892a25e2 | 572 | |
b49240ec | 573 | if (h->nlmsg_type == NLMSG_ERROR) { |
892a25e2 | 574 | rtnl_dump_error(rth, h); |
2d34851c | 575 | free(buf); |
b49240ec | 576 | return -1; |
aba5acdf | 577 | } |
486ccd99 VK |
578 | |
579 | if (!rth->dump_fp) { | |
580 | err = a->filter(&nladdr, h, a->arg1); | |
2d34851c HL |
581 | if (err < 0) { |
582 | free(buf); | |
486ccd99 | 583 | return err; |
2d34851c | 584 | } |
486ccd99 | 585 | } |
aba5acdf SH |
586 | |
587 | skip_it: | |
3bc1c4f2 | 588 | h = NLMSG_NEXT(h, msglen); |
b49240ec | 589 | } |
3bc1c4f2 | 590 | } |
2d34851c | 591 | free(buf); |
3bc1c4f2 | 592 | |
16f02e14 ND |
593 | if (found_done) { |
594 | if (dump_intr) | |
595 | fprintf(stderr, | |
596 | "Dump was interrupted and may be inconsistent.\n"); | |
3bc1c4f2 | 597 | return 0; |
16f02e14 | 598 | } |
3bc1c4f2 | 599 | |
aba5acdf SH |
600 | if (msg.msg_flags & MSG_TRUNC) { |
601 | fprintf(stderr, "Message truncated\n"); | |
602 | continue; | |
603 | } | |
3bc1c4f2 BG |
604 | if (msglen) { |
605 | fprintf(stderr, "!!!Remnant of size %d\n", msglen); | |
aba5acdf SH |
606 | exit(1); |
607 | } | |
608 | } | |
609 | } | |
610 | ||
8e72880f | 611 | int rtnl_dump_filter_nc(struct rtnl_handle *rth, |
b49240ec | 612 | rtnl_filter_t filter, |
8e72880f | 613 | void *arg1, __u16 nc_flags) |
b49240ec SH |
614 | { |
615 | const struct rtnl_dump_filter_arg a[2] = { | |
8e72880f PS |
616 | { .filter = filter, .arg1 = arg1, .nc_flags = nc_flags, }, |
617 | { .filter = NULL, .arg1 = NULL, .nc_flags = 0, }, | |
b49240ec SH |
618 | }; |
619 | ||
620 | return rtnl_dump_filter_l(rth, a); | |
621 | } | |
622 | ||
b6432e68 SH |
623 | static void rtnl_talk_error(struct nlmsghdr *h, struct nlmsgerr *err, |
624 | nl_ext_ack_fn_t errfn) | |
625 | { | |
844c37b4 | 626 | if (nl_dump_ext_ack(h, errfn)) |
b6432e68 SH |
627 | return; |
628 | ||
629 | fprintf(stderr, "RTNETLINK answers: %s\n", | |
630 | strerror(-err->error)); | |
631 | } | |
632 | ||
72a2ff39 CM |
633 | |
634 | static int __rtnl_talk_iov(struct rtnl_handle *rtnl, struct iovec *iov, | |
635 | size_t iovlen, struct nlmsghdr **answer, | |
636 | bool show_rtnl_err, nl_ext_ack_fn_t errfn) | |
aba5acdf | 637 | { |
d17b136f | 638 | struct sockaddr_nl nladdr = { .nl_family = AF_NETLINK }; |
72a2ff39 | 639 | struct iovec riov; |
aba5acdf | 640 | struct msghdr msg = { |
8ed63ab1 SH |
641 | .msg_name = &nladdr, |
642 | .msg_namelen = sizeof(nladdr), | |
72a2ff39 CM |
643 | .msg_iov = iov, |
644 | .msg_iovlen = iovlen, | |
aba5acdf | 645 | }; |
72a2ff39 CM |
646 | unsigned int seq = 0; |
647 | struct nlmsghdr *h; | |
648 | int i, status; | |
2d34851c | 649 | char *buf; |
aba5acdf | 650 | |
72a2ff39 CM |
651 | for (i = 0; i < iovlen; i++) { |
652 | h = iov[i].iov_base; | |
653 | h->nlmsg_seq = seq = ++rtnl->seq; | |
654 | if (answer == NULL) | |
655 | h->nlmsg_flags |= NLM_F_ACK; | |
656 | } | |
aba5acdf SH |
657 | |
658 | status = sendmsg(rtnl->fd, &msg, 0); | |
aba5acdf SH |
659 | if (status < 0) { |
660 | perror("Cannot talk to rtnetlink"); | |
661 | return -1; | |
662 | } | |
663 | ||
72a2ff39 CM |
664 | /* change msg to use the response iov */ |
665 | msg.msg_iov = &riov; | |
666 | msg.msg_iovlen = 1; | |
667 | i = 0; | |
aba5acdf | 668 | while (1) { |
2d34851c | 669 | status = rtnl_recvmsg(rtnl->fd, &msg, &buf); |
72a2ff39 | 670 | ++i; |
2d34851c HL |
671 | |
672 | if (status < 0) | |
673 | return status; | |
aba5acdf | 674 | |
aba5acdf | 675 | if (msg.msg_namelen != sizeof(nladdr)) { |
2c500a4d SH |
676 | fprintf(stderr, |
677 | "sender address length == %d\n", | |
678 | msg.msg_namelen); | |
aba5acdf SH |
679 | exit(1); |
680 | } | |
2c500a4d | 681 | for (h = (struct nlmsghdr *)buf; status >= sizeof(*h); ) { |
aba5acdf SH |
682 | int len = h->nlmsg_len; |
683 | int l = len - sizeof(*h); | |
684 | ||
2c500a4d | 685 | if (l < 0 || len > status) { |
aba5acdf SH |
686 | if (msg.msg_flags & MSG_TRUNC) { |
687 | fprintf(stderr, "Truncated message\n"); | |
2d34851c | 688 | free(buf); |
aba5acdf SH |
689 | return -1; |
690 | } | |
2c500a4d SH |
691 | fprintf(stderr, |
692 | "!!!malformed message: len=%d\n", | |
693 | len); | |
aba5acdf SH |
694 | exit(1); |
695 | } | |
696 | ||
c079e121 | 697 | if (nladdr.nl_pid != 0 || |
10f57ef1 | 698 | h->nlmsg_pid != rtnl->local.nl_pid || |
72a2ff39 | 699 | h->nlmsg_seq > seq || h->nlmsg_seq < seq - iovlen) { |
4cca16f2 SH |
700 | /* Don't forget to skip that message. */ |
701 | status -= NLMSG_ALIGN(len); | |
2c500a4d | 702 | h = (struct nlmsghdr *)((char *)h + NLMSG_ALIGN(len)); |
aba5acdf SH |
703 | continue; |
704 | } | |
705 | ||
706 | if (h->nlmsg_type == NLMSG_ERROR) { | |
2c500a4d SH |
707 | struct nlmsgerr *err = (struct nlmsgerr *)NLMSG_DATA(h); |
708 | ||
aba5acdf SH |
709 | if (l < sizeof(struct nlmsgerr)) { |
710 | fprintf(stderr, "ERROR truncated\n"); | |
c60389e4 SH |
711 | free(buf); |
712 | return -1; | |
713 | } | |
714 | ||
715 | if (!err->error) | |
844c37b4 DA |
716 | /* check messages from kernel */ |
717 | nl_dump_ext_ack(h, errfn); | |
b45e3000 SH |
718 | else { |
719 | errno = -err->error; | |
844c37b4 | 720 | |
b45e3000 SH |
721 | if (rtnl->proto != NETLINK_SOCK_DIAG && |
722 | show_rtnl_err) | |
723 | rtnl_talk_error(h, err, errfn); | |
724 | } | |
b6432e68 | 725 | |
c60389e4 SH |
726 | if (answer) |
727 | *answer = (struct nlmsghdr *)buf; | |
728 | else | |
729 | free(buf); | |
b45e3000 SH |
730 | |
731 | return err->error ? -i : 0; | |
aba5acdf | 732 | } |
c079e121 | 733 | |
aba5acdf | 734 | if (answer) { |
86bf43c7 | 735 | *answer = (struct nlmsghdr *)buf; |
aba5acdf SH |
736 | return 0; |
737 | } | |
738 | ||
739 | fprintf(stderr, "Unexpected reply!!!\n"); | |
740 | ||
741 | status -= NLMSG_ALIGN(len); | |
2c500a4d | 742 | h = (struct nlmsghdr *)((char *)h + NLMSG_ALIGN(len)); |
aba5acdf | 743 | } |
2d34851c | 744 | free(buf); |
c079e121 | 745 | |
aba5acdf SH |
746 | if (msg.msg_flags & MSG_TRUNC) { |
747 | fprintf(stderr, "Message truncated\n"); | |
748 | continue; | |
749 | } | |
c079e121 | 750 | |
aba5acdf SH |
751 | if (status) { |
752 | fprintf(stderr, "!!!Remnant of size %d\n", status); | |
753 | exit(1); | |
754 | } | |
755 | } | |
756 | } | |
757 | ||
72a2ff39 CM |
758 | static int __rtnl_talk(struct rtnl_handle *rtnl, struct nlmsghdr *n, |
759 | struct nlmsghdr **answer, | |
760 | bool show_rtnl_err, nl_ext_ack_fn_t errfn) | |
761 | { | |
762 | struct iovec iov = { | |
763 | .iov_base = n, | |
764 | .iov_len = n->nlmsg_len | |
765 | }; | |
766 | ||
767 | return __rtnl_talk_iov(rtnl, &iov, 1, answer, show_rtnl_err, errfn); | |
768 | } | |
769 | ||
463d9efa | 770 | int rtnl_talk(struct rtnl_handle *rtnl, struct nlmsghdr *n, |
86bf43c7 | 771 | struct nlmsghdr **answer) |
463d9efa | 772 | { |
86bf43c7 | 773 | return __rtnl_talk(rtnl, n, answer, true, NULL); |
b6432e68 SH |
774 | } |
775 | ||
72a2ff39 CM |
776 | int rtnl_talk_iov(struct rtnl_handle *rtnl, struct iovec *iovec, size_t iovlen, |
777 | struct nlmsghdr **answer) | |
778 | { | |
779 | return __rtnl_talk_iov(rtnl, iovec, iovlen, answer, true, NULL); | |
780 | } | |
781 | ||
b6432e68 | 782 | int rtnl_talk_extack(struct rtnl_handle *rtnl, struct nlmsghdr *n, |
86bf43c7 | 783 | struct nlmsghdr **answer, |
b6432e68 SH |
784 | nl_ext_ack_fn_t errfn) |
785 | { | |
86bf43c7 | 786 | return __rtnl_talk(rtnl, n, answer, true, errfn); |
463d9efa DA |
787 | } |
788 | ||
789 | int rtnl_talk_suppress_rtnl_errmsg(struct rtnl_handle *rtnl, struct nlmsghdr *n, | |
86bf43c7 | 790 | struct nlmsghdr **answer) |
463d9efa | 791 | { |
86bf43c7 | 792 | return __rtnl_talk(rtnl, n, answer, false, NULL); |
463d9efa DA |
793 | } |
794 | ||
449b824a ND |
795 | int rtnl_listen_all_nsid(struct rtnl_handle *rth) |
796 | { | |
797 | unsigned int on = 1; | |
798 | ||
799 | if (setsockopt(rth->fd, SOL_NETLINK, NETLINK_LISTEN_ALL_NSID, &on, | |
800 | sizeof(on)) < 0) { | |
801 | perror("NETLINK_LISTEN_ALL_NSID"); | |
802 | return -1; | |
803 | } | |
804 | rth->flags |= RTNL_HANDLE_F_LISTEN_ALL_NSID; | |
805 | return 0; | |
806 | } | |
807 | ||
8ed63ab1 | 808 | int rtnl_listen(struct rtnl_handle *rtnl, |
0628cddd | 809 | rtnl_listen_filter_t handler, |
6dc9f016 | 810 | void *jarg) |
aba5acdf SH |
811 | { |
812 | int status; | |
813 | struct nlmsghdr *h; | |
d17b136f | 814 | struct sockaddr_nl nladdr = { .nl_family = AF_NETLINK }; |
aba5acdf | 815 | struct iovec iov; |
aba5acdf | 816 | struct msghdr msg = { |
8ed63ab1 SH |
817 | .msg_name = &nladdr, |
818 | .msg_namelen = sizeof(nladdr), | |
819 | .msg_iov = &iov, | |
820 | .msg_iovlen = 1, | |
aba5acdf | 821 | }; |
e557212e | 822 | char buf[16384]; |
449b824a ND |
823 | char cmsgbuf[BUFSIZ]; |
824 | ||
825 | if (rtnl->flags & RTNL_HANDLE_F_LISTEN_ALL_NSID) { | |
826 | msg.msg_control = &cmsgbuf; | |
827 | msg.msg_controllen = sizeof(cmsgbuf); | |
828 | } | |
aba5acdf | 829 | |
aba5acdf | 830 | iov.iov_base = buf; |
aba5acdf | 831 | while (1) { |
449b824a ND |
832 | struct rtnl_ctrl_data ctrl; |
833 | struct cmsghdr *cmsg; | |
834 | ||
aba5acdf SH |
835 | iov.iov_len = sizeof(buf); |
836 | status = recvmsg(rtnl->fd, &msg, 0); | |
837 | ||
838 | if (status < 0) { | |
aa8032e6 | 839 | if (errno == EINTR || errno == EAGAIN) |
aba5acdf | 840 | continue; |
aa8032e6 SH |
841 | fprintf(stderr, "netlink receive error %s (%d)\n", |
842 | strerror(errno), errno); | |
7f03191f PM |
843 | if (errno == ENOBUFS) |
844 | continue; | |
aa8032e6 | 845 | return -1; |
aba5acdf SH |
846 | } |
847 | if (status == 0) { | |
848 | fprintf(stderr, "EOF on netlink\n"); | |
849 | return -1; | |
850 | } | |
851 | if (msg.msg_namelen != sizeof(nladdr)) { | |
2c500a4d SH |
852 | fprintf(stderr, |
853 | "Sender address length == %d\n", | |
854 | msg.msg_namelen); | |
aba5acdf SH |
855 | exit(1); |
856 | } | |
449b824a ND |
857 | |
858 | if (rtnl->flags & RTNL_HANDLE_F_LISTEN_ALL_NSID) { | |
859 | memset(&ctrl, 0, sizeof(ctrl)); | |
860 | ctrl.nsid = -1; | |
861 | for (cmsg = CMSG_FIRSTHDR(&msg); cmsg; | |
862 | cmsg = CMSG_NXTHDR(&msg, cmsg)) | |
863 | if (cmsg->cmsg_level == SOL_NETLINK && | |
864 | cmsg->cmsg_type == NETLINK_LISTEN_ALL_NSID && | |
865 | cmsg->cmsg_len == CMSG_LEN(sizeof(int))) { | |
866 | int *data = (int *)CMSG_DATA(cmsg); | |
867 | ||
868 | ctrl.nsid = *data; | |
869 | } | |
870 | } | |
871 | ||
2c500a4d | 872 | for (h = (struct nlmsghdr *)buf; status >= sizeof(*h); ) { |
aba5acdf SH |
873 | int err; |
874 | int len = h->nlmsg_len; | |
875 | int l = len - sizeof(*h); | |
876 | ||
2c500a4d | 877 | if (l < 0 || len > status) { |
aba5acdf SH |
878 | if (msg.msg_flags & MSG_TRUNC) { |
879 | fprintf(stderr, "Truncated message\n"); | |
880 | return -1; | |
881 | } | |
2c500a4d SH |
882 | fprintf(stderr, |
883 | "!!!malformed message: len=%d\n", | |
884 | len); | |
aba5acdf SH |
885 | exit(1); |
886 | } | |
887 | ||
449b824a | 888 | err = handler(&nladdr, &ctrl, h, jarg); |
aba5acdf SH |
889 | if (err < 0) |
890 | return err; | |
891 | ||
892 | status -= NLMSG_ALIGN(len); | |
2c500a4d | 893 | h = (struct nlmsghdr *)((char *)h + NLMSG_ALIGN(len)); |
aba5acdf SH |
894 | } |
895 | if (msg.msg_flags & MSG_TRUNC) { | |
896 | fprintf(stderr, "Message truncated\n"); | |
897 | continue; | |
898 | } | |
899 | if (status) { | |
900 | fprintf(stderr, "!!!Remnant of size %d\n", status); | |
901 | exit(1); | |
902 | } | |
903 | } | |
904 | } | |
905 | ||
0628cddd | 906 | int rtnl_from_file(FILE *rtnl, rtnl_listen_filter_t handler, |
6dc9f016 | 907 | void *jarg) |
aba5acdf SH |
908 | { |
909 | int status; | |
d17b136f | 910 | struct sockaddr_nl nladdr = { .nl_family = AF_NETLINK }; |
2c500a4d SH |
911 | char buf[16384]; |
912 | struct nlmsghdr *h = (struct nlmsghdr *)buf; | |
aba5acdf | 913 | |
aba5acdf | 914 | while (1) { |
2dd9f8e0 | 915 | int err, len; |
aba5acdf SH |
916 | int l; |
917 | ||
918 | status = fread(&buf, 1, sizeof(*h), rtnl); | |
919 | ||
920 | if (status < 0) { | |
921 | if (errno == EINTR) | |
922 | continue; | |
923 | perror("rtnl_from_file: fread"); | |
924 | return -1; | |
925 | } | |
926 | if (status == 0) | |
927 | return 0; | |
928 | ||
929 | len = h->nlmsg_len; | |
aba5acdf SH |
930 | l = len - sizeof(*h); |
931 | ||
2c500a4d | 932 | if (l < 0 || len > sizeof(buf)) { |
aba5acdf SH |
933 | fprintf(stderr, "!!!malformed message: len=%d @%lu\n", |
934 | len, ftell(rtnl)); | |
935 | return -1; | |
936 | } | |
937 | ||
938 | status = fread(NLMSG_DATA(h), 1, NLMSG_ALIGN(l), rtnl); | |
939 | ||
940 | if (status < 0) { | |
941 | perror("rtnl_from_file: fread"); | |
942 | return -1; | |
943 | } | |
944 | if (status < l) { | |
945 | fprintf(stderr, "rtnl-from_file: truncated message\n"); | |
946 | return -1; | |
947 | } | |
948 | ||
0628cddd | 949 | err = handler(&nladdr, NULL, h, jarg); |
aba5acdf SH |
950 | if (err < 0) |
951 | return err; | |
952 | } | |
953 | } | |
954 | ||
2aa3dd29 SH |
955 | int addattr(struct nlmsghdr *n, int maxlen, int type) |
956 | { | |
957 | return addattr_l(n, maxlen, type, NULL, 0); | |
958 | } | |
959 | ||
960 | int addattr8(struct nlmsghdr *n, int maxlen, int type, __u8 data) | |
961 | { | |
962 | return addattr_l(n, maxlen, type, &data, sizeof(__u8)); | |
963 | } | |
964 | ||
965 | int addattr16(struct nlmsghdr *n, int maxlen, int type, __u16 data) | |
966 | { | |
967 | return addattr_l(n, maxlen, type, &data, sizeof(__u16)); | |
968 | } | |
969 | ||
aba5acdf SH |
970 | int addattr32(struct nlmsghdr *n, int maxlen, int type, __u32 data) |
971 | { | |
2aa3dd29 SH |
972 | return addattr_l(n, maxlen, type, &data, sizeof(__u32)); |
973 | } | |
974 | ||
975 | int addattr64(struct nlmsghdr *n, int maxlen, int type, __u64 data) | |
976 | { | |
977 | return addattr_l(n, maxlen, type, &data, sizeof(__u64)); | |
978 | } | |
979 | ||
980 | int addattrstrz(struct nlmsghdr *n, int maxlen, int type, const char *str) | |
981 | { | |
982 | return addattr_l(n, maxlen, type, str, strlen(str)+1); | |
aba5acdf SH |
983 | } |
984 | ||
8ed63ab1 | 985 | int addattr_l(struct nlmsghdr *n, int maxlen, int type, const void *data, |
6dc9f016 | 986 | int alen) |
aba5acdf SH |
987 | { |
988 | int len = RTA_LENGTH(alen); | |
989 | struct rtattr *rta; | |
990 | ||
3dabdbb3 | 991 | if (NLMSG_ALIGN(n->nlmsg_len) + RTA_ALIGN(len) > maxlen) { |
2c500a4d SH |
992 | fprintf(stderr, |
993 | "addattr_l ERROR: message exceeded bound of %d\n", | |
994 | maxlen); | |
aba5acdf | 995 | return -1; |
007d3a3e | 996 | } |
07f94362 | 997 | rta = NLMSG_TAIL(n); |
aba5acdf SH |
998 | rta->rta_type = type; |
999 | rta->rta_len = len; | |
893deac4 PS |
1000 | if (alen) |
1001 | memcpy(RTA_DATA(rta), data, alen); | |
3dabdbb3 | 1002 | n->nlmsg_len = NLMSG_ALIGN(n->nlmsg_len) + RTA_ALIGN(len); |
aba5acdf SH |
1003 | return 0; |
1004 | } | |
1005 | ||
07f94362 | 1006 | int addraw_l(struct nlmsghdr *n, int maxlen, const void *data, int len) |
1007 | { | |
1008 | if (NLMSG_ALIGN(n->nlmsg_len) + NLMSG_ALIGN(len) > maxlen) { | |
2c500a4d SH |
1009 | fprintf(stderr, |
1010 | "addraw_l ERROR: message exceeded bound of %d\n", | |
1011 | maxlen); | |
07f94362 | 1012 | return -1; |
1013 | } | |
1014 | ||
1015 | memcpy(NLMSG_TAIL(n), data, len); | |
1016 | memset((void *) NLMSG_TAIL(n) + len, 0, NLMSG_ALIGN(len) - len); | |
1017 | n->nlmsg_len = NLMSG_ALIGN(n->nlmsg_len) + NLMSG_ALIGN(len); | |
1018 | return 0; | |
1019 | } | |
1020 | ||
2f90c9c0 PM |
1021 | struct rtattr *addattr_nest(struct nlmsghdr *n, int maxlen, int type) |
1022 | { | |
1023 | struct rtattr *nest = NLMSG_TAIL(n); | |
1024 | ||
1025 | addattr_l(n, maxlen, type, NULL, 0); | |
1026 | return nest; | |
1027 | } | |
1028 | ||
1029 | int addattr_nest_end(struct nlmsghdr *n, struct rtattr *nest) | |
1030 | { | |
1031 | nest->rta_len = (void *)NLMSG_TAIL(n) - (void *)nest; | |
1032 | return n->nlmsg_len; | |
1033 | } | |
1034 | ||
1035 | struct rtattr *addattr_nest_compat(struct nlmsghdr *n, int maxlen, int type, | |
1036 | const void *data, int len) | |
1037 | { | |
1038 | struct rtattr *start = NLMSG_TAIL(n); | |
1039 | ||
1040 | addattr_l(n, maxlen, type, data, len); | |
1041 | addattr_nest(n, maxlen, type); | |
1042 | return start; | |
1043 | } | |
1044 | ||
1045 | int addattr_nest_compat_end(struct nlmsghdr *n, struct rtattr *start) | |
1046 | { | |
1047 | struct rtattr *nest = (void *)start + NLMSG_ALIGN(start->rta_len); | |
1048 | ||
1049 | start->rta_len = (void *)NLMSG_TAIL(n) - (void *)start; | |
1050 | addattr_nest_end(n, nest); | |
1051 | return n->nlmsg_len; | |
1052 | } | |
1053 | ||
aba5acdf SH |
1054 | int rta_addattr32(struct rtattr *rta, int maxlen, int type, __u32 data) |
1055 | { | |
1056 | int len = RTA_LENGTH(4); | |
1057 | struct rtattr *subrta; | |
1058 | ||
007d3a3e | 1059 | if (RTA_ALIGN(rta->rta_len) + len > maxlen) { |
2c500a4d SH |
1060 | fprintf(stderr, |
1061 | "rta_addattr32: Error! max allowed bound %d exceeded\n", | |
1062 | maxlen); | |
aba5acdf | 1063 | return -1; |
007d3a3e | 1064 | } |
2c500a4d | 1065 | subrta = (struct rtattr *)(((char *)rta) + RTA_ALIGN(rta->rta_len)); |
aba5acdf SH |
1066 | subrta->rta_type = type; |
1067 | subrta->rta_len = len; | |
1068 | memcpy(RTA_DATA(subrta), &data, 4); | |
1069 | rta->rta_len = NLMSG_ALIGN(rta->rta_len) + len; | |
1070 | return 0; | |
1071 | } | |
1072 | ||
8ed63ab1 | 1073 | int rta_addattr_l(struct rtattr *rta, int maxlen, int type, |
6dc9f016 | 1074 | const void *data, int alen) |
aba5acdf SH |
1075 | { |
1076 | struct rtattr *subrta; | |
1077 | int len = RTA_LENGTH(alen); | |
1078 | ||
3dabdbb3 | 1079 | if (RTA_ALIGN(rta->rta_len) + RTA_ALIGN(len) > maxlen) { |
2c500a4d SH |
1080 | fprintf(stderr, |
1081 | "rta_addattr_l: Error! max allowed bound %d exceeded\n", | |
1082 | maxlen); | |
aba5acdf | 1083 | return -1; |
007d3a3e | 1084 | } |
2c500a4d | 1085 | subrta = (struct rtattr *)(((char *)rta) + RTA_ALIGN(rta->rta_len)); |
aba5acdf SH |
1086 | subrta->rta_type = type; |
1087 | subrta->rta_len = len; | |
893deac4 PS |
1088 | if (alen) |
1089 | memcpy(RTA_DATA(subrta), data, alen); | |
3dabdbb3 | 1090 | rta->rta_len = NLMSG_ALIGN(rta->rta_len) + RTA_ALIGN(len); |
aba5acdf SH |
1091 | return 0; |
1092 | } | |
1093 | ||
303cc9cb RP |
1094 | int rta_addattr8(struct rtattr *rta, int maxlen, int type, __u8 data) |
1095 | { | |
1096 | return rta_addattr_l(rta, maxlen, type, &data, sizeof(__u8)); | |
1097 | } | |
1098 | ||
1099 | int rta_addattr16(struct rtattr *rta, int maxlen, int type, __u16 data) | |
1100 | { | |
1101 | return rta_addattr_l(rta, maxlen, type, &data, sizeof(__u16)); | |
1102 | } | |
1103 | ||
1104 | int rta_addattr64(struct rtattr *rta, int maxlen, int type, __u64 data) | |
1105 | { | |
1106 | return rta_addattr_l(rta, maxlen, type, &data, sizeof(__u64)); | |
1107 | } | |
1108 | ||
1109 | struct rtattr *rta_nest(struct rtattr *rta, int maxlen, int type) | |
1110 | { | |
1111 | struct rtattr *nest = RTA_TAIL(rta); | |
1112 | ||
1113 | rta_addattr_l(rta, maxlen, type, NULL, 0); | |
1114 | ||
1115 | return nest; | |
1116 | } | |
1117 | ||
1118 | int rta_nest_end(struct rtattr *rta, struct rtattr *nest) | |
1119 | { | |
1120 | nest->rta_len = (void *)RTA_TAIL(rta) - (void *)nest; | |
1121 | ||
1122 | return rta->rta_len; | |
1123 | } | |
1124 | ||
aba5acdf SH |
1125 | int parse_rtattr(struct rtattr *tb[], int max, struct rtattr *rta, int len) |
1126 | { | |
b1b7ce0f VY |
1127 | return parse_rtattr_flags(tb, max, rta, len, 0); |
1128 | } | |
1129 | ||
1130 | int parse_rtattr_flags(struct rtattr *tb[], int max, struct rtattr *rta, | |
1131 | int len, unsigned short flags) | |
1132 | { | |
1133 | unsigned short type; | |
1134 | ||
175e2440 | 1135 | memset(tb, 0, sizeof(struct rtattr *) * (max + 1)); |
aba5acdf | 1136 | while (RTA_OK(rta, len)) { |
b1b7ce0f VY |
1137 | type = rta->rta_type & ~flags; |
1138 | if ((type <= max) && (!tb[type])) | |
1139 | tb[type] = rta; | |
2c500a4d | 1140 | rta = RTA_NEXT(rta, len); |
aba5acdf SH |
1141 | } |
1142 | if (len) | |
2c500a4d SH |
1143 | fprintf(stderr, "!!!Deficit %d, rta_len=%d\n", |
1144 | len, rta->rta_len); | |
aba5acdf SH |
1145 | return 0; |
1146 | } | |
c7699875 | 1147 | |
2c500a4d SH |
1148 | int parse_rtattr_byindex(struct rtattr *tb[], int max, |
1149 | struct rtattr *rta, int len) | |
c7699875 | 1150 | { |
1151 | int i = 0; | |
175e2440 SH |
1152 | |
1153 | memset(tb, 0, sizeof(struct rtattr *) * max); | |
c7699875 | 1154 | while (RTA_OK(rta, len)) { |
175e2440 | 1155 | if (rta->rta_type <= max && i < max) |
c7699875 | 1156 | tb[i++] = rta; |
2c500a4d | 1157 | rta = RTA_NEXT(rta, len); |
c7699875 | 1158 | } |
1159 | if (len) | |
2c500a4d SH |
1160 | fprintf(stderr, "!!!Deficit %d, rta_len=%d\n", |
1161 | len, rta->rta_len); | |
c7699875 | 1162 | return i; |
1163 | } | |
2f90c9c0 | 1164 | |
decbb437 JP |
1165 | struct rtattr *parse_rtattr_one(int type, struct rtattr *rta, int len) |
1166 | { | |
1167 | while (RTA_OK(rta, len)) { | |
1168 | if (rta->rta_type == type) | |
1169 | return rta; | |
1170 | rta = RTA_NEXT(rta, len); | |
1171 | } | |
2c500a4d | 1172 | |
decbb437 | 1173 | if (len) |
2c500a4d SH |
1174 | fprintf(stderr, "!!!Deficit %d, rta_len=%d\n", |
1175 | len, rta->rta_len); | |
decbb437 JP |
1176 | return NULL; |
1177 | } | |
1178 | ||
2c500a4d SH |
1179 | int __parse_rtattr_nested_compat(struct rtattr *tb[], int max, |
1180 | struct rtattr *rta, | |
1181 | int len) | |
2f90c9c0 PM |
1182 | { |
1183 | if (RTA_PAYLOAD(rta) < len) | |
1184 | return -1; | |
1185 | if (RTA_PAYLOAD(rta) >= RTA_ALIGN(len) + sizeof(struct rtattr)) { | |
1186 | rta = RTA_DATA(rta) + RTA_ALIGN(len); | |
1187 | return parse_rtattr_nested(tb, max, rta); | |
1188 | } | |
037c635e | 1189 | memset(tb, 0, sizeof(struct rtattr *) * (max + 1)); |
2f90c9c0 PM |
1190 | return 0; |
1191 | } |