]>
git.proxmox.com Git - mirror_frr.git/blob - pimd/mtracebis_netlink.c
2 * libnetlink.c RTnetlink service routines.
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.
9 * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
24 #include <net/if_arp.h>
25 #include <sys/socket.h>
26 #include <netinet/in.h>
33 #include "mtracebis_netlink.h"
35 int rcvbuf
= 1024 * 1024;
37 void rtnl_close(struct rtnl_handle
*rth
)
45 int rtnl_open_byproto(struct rtnl_handle
*rth
, unsigned subscriptions
,
51 memset(rth
, 0, sizeof(*rth
));
53 rth
->fd
= socket(AF_NETLINK
, SOCK_RAW
, protocol
);
55 perror("Cannot open netlink socket");
59 if (setsockopt(rth
->fd
, SOL_SOCKET
, SO_SNDBUF
, &sndbuf
, sizeof(sndbuf
))
65 if (setsockopt(rth
->fd
, SOL_SOCKET
, SO_RCVBUF
, &rcvbuf
, sizeof(rcvbuf
))
71 memset(&rth
->local
, 0, sizeof(rth
->local
));
72 rth
->local
.nl_family
= AF_NETLINK
;
73 rth
->local
.nl_groups
= subscriptions
;
75 if (bind(rth
->fd
, (struct sockaddr
*)&rth
->local
, sizeof(rth
->local
))
77 perror("Cannot bind netlink socket");
80 addr_len
= sizeof(rth
->local
);
81 if (getsockname(rth
->fd
, (struct sockaddr
*)&rth
->local
, &addr_len
)
83 perror("Cannot getsockname");
86 if (addr_len
!= sizeof(rth
->local
)) {
87 fprintf(stderr
, "Wrong address length %d\n", addr_len
);
90 if (rth
->local
.nl_family
!= AF_NETLINK
) {
91 fprintf(stderr
, "Wrong address family %d\n",
92 rth
->local
.nl_family
);
95 rth
->seq
= time(NULL
);
99 int rtnl_open(struct rtnl_handle
*rth
, unsigned subscriptions
)
101 return rtnl_open_byproto(rth
, subscriptions
, NETLINK_ROUTE
);
104 int rtnl_wilddump_request(struct rtnl_handle
*rth
, int family
, int type
)
111 memset(&req
, 0, sizeof(req
));
112 req
.nlh
.nlmsg_len
= sizeof(req
);
113 req
.nlh
.nlmsg_type
= type
;
114 req
.nlh
.nlmsg_flags
= NLM_F_ROOT
| NLM_F_MATCH
| NLM_F_REQUEST
;
115 req
.nlh
.nlmsg_pid
= 0;
116 req
.nlh
.nlmsg_seq
= rth
->dump
= ++rth
->seq
;
117 req
.g
.rtgen_family
= family
;
119 return send(rth
->fd
, (void *)&req
, sizeof(req
), 0);
122 int rtnl_send(struct rtnl_handle
*rth
, const char *buf
, int len
)
124 return send(rth
->fd
, buf
, len
, 0);
127 int rtnl_send_check(struct rtnl_handle
*rth
, const char *buf
, int len
)
133 status
= send(rth
->fd
, buf
, len
, 0);
137 /* Check for immediate errors */
138 status
= recv(rth
->fd
, resp
, sizeof(resp
), MSG_DONTWAIT
| MSG_PEEK
);
145 for (h
= (struct nlmsghdr
*)resp
; NLMSG_OK(h
, (uint32_t)status
);
146 h
= NLMSG_NEXT(h
, status
)) {
147 if (h
->nlmsg_type
== NLMSG_ERROR
) {
148 struct nlmsgerr
*err
= (struct nlmsgerr
*)NLMSG_DATA(h
);
150 < NLMSG_LENGTH(sizeof(struct nlmsgerr
)))
151 fprintf(stderr
, "ERROR truncated\n");
161 int rtnl_dump_request(struct rtnl_handle
*rth
, int type
, void *req
, int len
)
164 struct sockaddr_nl nladdr
;
165 struct iovec iov
[2] = {{.iov_base
= &nlh
, .iov_len
= sizeof(nlh
)},
166 {.iov_base
= req
, .iov_len
= len
}};
167 struct msghdr msg
= {
169 .msg_namelen
= sizeof(nladdr
),
174 memset(&nladdr
, 0, sizeof(nladdr
));
175 nladdr
.nl_family
= AF_NETLINK
;
177 nlh
.nlmsg_len
= NLMSG_LENGTH(len
);
178 nlh
.nlmsg_type
= type
;
179 nlh
.nlmsg_flags
= NLM_F_ROOT
| NLM_F_MATCH
| NLM_F_REQUEST
;
181 nlh
.nlmsg_seq
= rth
->dump
= ++rth
->seq
;
183 return sendmsg(rth
->fd
, &msg
, 0);
186 int rtnl_dump_filter_l(struct rtnl_handle
*rth
,
187 const struct rtnl_dump_filter_arg
*arg
)
189 struct sockaddr_nl nladdr
;
191 struct msghdr msg
= {
193 .msg_namelen
= sizeof(nladdr
),
202 const struct rtnl_dump_filter_arg
*a
;
206 iov
.iov_len
= sizeof(buf
);
207 status
= recvmsg(rth
->fd
, &msg
, 0);
210 if (errno
== EINTR
|| errno
== EAGAIN
)
212 fprintf(stderr
, "netlink receive error %s (%d)\n",
213 strerror(errno
), errno
);
218 fprintf(stderr
, "EOF on netlink\n");
222 for (a
= arg
; a
->filter
; a
++) {
223 struct nlmsghdr
*h
= (struct nlmsghdr
*)buf
;
226 while (NLMSG_OK(h
, (uint32_t)msglen
)) {
229 if (nladdr
.nl_pid
!= 0
230 || h
->nlmsg_pid
!= rth
->local
.nl_pid
231 || h
->nlmsg_seq
!= rth
->dump
) {
233 err
= a
->junk(&nladdr
, h
,
241 if (h
->nlmsg_type
== NLMSG_DONE
) {
243 break; /* process next filter */
245 if (h
->nlmsg_type
== NLMSG_ERROR
) {
246 struct nlmsgerr
*merr
=
247 (struct nlmsgerr
*)NLMSG_DATA(
250 < NLMSG_LENGTH(sizeof(
253 "ERROR truncated\n");
255 errno
= -merr
->error
;
256 perror("RTNETLINK answers");
260 err
= a
->filter(&nladdr
, h
, a
->arg1
);
265 h
= NLMSG_NEXT(h
, msglen
);
272 if (msg
.msg_flags
& MSG_TRUNC
) {
273 fprintf(stderr
, "Message truncated\n");
277 fprintf(stderr
, "!!!Remnant of size %d\n", msglen
);
283 int rtnl_dump_filter(struct rtnl_handle
*rth
, rtnl_filter_t filter
, void *arg1
,
284 rtnl_filter_t junk
, void *arg2
)
286 const struct rtnl_dump_filter_arg a
[2] = {
287 {.filter
= filter
, .arg1
= arg1
, .junk
= junk
, .arg2
= arg2
},
288 {.filter
= NULL
, .arg1
= NULL
, .junk
= NULL
, .arg2
= NULL
}};
290 return rtnl_dump_filter_l(rth
, a
);
293 int rtnl_talk(struct rtnl_handle
*rtnl
, struct nlmsghdr
*n
, pid_t peer
,
294 unsigned groups
, struct nlmsghdr
*answer
, rtnl_filter_t junk
,
300 struct sockaddr_nl nladdr
;
301 struct iovec iov
= {.iov_base
= (void *)n
, .iov_len
= n
->nlmsg_len
};
302 struct msghdr msg
= {
304 .msg_namelen
= sizeof(nladdr
),
310 memset(&nladdr
, 0, sizeof(nladdr
));
311 nladdr
.nl_family
= AF_NETLINK
;
312 nladdr
.nl_pid
= peer
;
313 nladdr
.nl_groups
= groups
;
315 n
->nlmsg_seq
= seq
= ++rtnl
->seq
;
318 n
->nlmsg_flags
|= NLM_F_ACK
;
320 status
= sendmsg(rtnl
->fd
, &msg
, 0);
323 perror("Cannot talk to rtnetlink");
327 memset(buf
, 0, sizeof(buf
));
332 iov
.iov_len
= sizeof(buf
);
333 status
= recvmsg(rtnl
->fd
, &msg
, 0);
336 if (errno
== EINTR
|| errno
== EAGAIN
)
338 fprintf(stderr
, "netlink receive error %s (%d)\n",
339 strerror(errno
), errno
);
343 fprintf(stderr
, "EOF on netlink\n");
346 if (msg
.msg_namelen
!= sizeof(nladdr
)) {
347 fprintf(stderr
, "sender address length == %d\n",
351 for (h
= (struct nlmsghdr
*)buf
; status
>= (int)sizeof(*h
);) {
353 int len
= h
->nlmsg_len
;
354 int l
= len
- sizeof(*h
);
356 if (l
< 0 || len
> status
) {
357 if (msg
.msg_flags
& MSG_TRUNC
) {
358 fprintf(stderr
, "Truncated message\n");
362 "!!!malformed message: len=%d\n", len
);
366 if ((int)nladdr
.nl_pid
!= peer
367 || h
->nlmsg_pid
!= rtnl
->local
.nl_pid
368 || h
->nlmsg_seq
!= seq
) {
370 err
= junk(&nladdr
, h
, jarg
);
374 /* Don't forget to skip that message. */
375 status
-= NLMSG_ALIGN(len
);
376 h
= (struct nlmsghdr
*)((char *)h
381 if (h
->nlmsg_type
== NLMSG_ERROR
) {
382 struct nlmsgerr
*merr
=
383 (struct nlmsgerr
*)NLMSG_DATA(h
);
384 if (l
< (int)sizeof(struct nlmsgerr
)) {
385 fprintf(stderr
, "ERROR truncated\n");
387 errno
= -merr
->error
;
394 perror("RTNETLINK answers");
399 memcpy(answer
, h
, h
->nlmsg_len
);
403 fprintf(stderr
, "Unexpected reply!!!\n");
405 status
-= NLMSG_ALIGN(len
);
406 h
= (struct nlmsghdr
*)((char *)h
+ NLMSG_ALIGN(len
));
408 if (msg
.msg_flags
& MSG_TRUNC
) {
409 fprintf(stderr
, "Message truncated\n");
413 fprintf(stderr
, "!!!Remnant of size %d\n", status
);
419 int rtnl_listen(struct rtnl_handle
*rtnl
, rtnl_filter_t handler
, void *jarg
)
423 struct sockaddr_nl nladdr
;
425 struct msghdr msg
= {
427 .msg_namelen
= sizeof(nladdr
),
433 memset(&nladdr
, 0, sizeof(nladdr
));
434 nladdr
.nl_family
= AF_NETLINK
;
436 nladdr
.nl_groups
= 0;
440 iov
.iov_len
= sizeof(buf
);
441 status
= recvmsg(rtnl
->fd
, &msg
, 0);
444 if (errno
== EINTR
|| errno
== EAGAIN
)
446 fprintf(stderr
, "netlink receive error %s (%d)\n",
447 strerror(errno
), errno
);
448 if (errno
== ENOBUFS
)
453 fprintf(stderr
, "EOF on netlink\n");
456 if (msg
.msg_namelen
!= sizeof(nladdr
)) {
457 fprintf(stderr
, "Sender address length == %d\n",
461 for (h
= (struct nlmsghdr
*)buf
; status
>= (int)sizeof(*h
);) {
463 int len
= h
->nlmsg_len
;
464 int l
= len
- sizeof(*h
);
466 if (l
< 0 || len
> status
) {
467 if (msg
.msg_flags
& MSG_TRUNC
) {
468 fprintf(stderr
, "Truncated message\n");
472 "!!!malformed message: len=%d\n", len
);
476 err
= handler(&nladdr
, h
, jarg
);
480 status
-= NLMSG_ALIGN(len
);
481 h
= (struct nlmsghdr
*)((char *)h
+ NLMSG_ALIGN(len
));
483 if (msg
.msg_flags
& MSG_TRUNC
) {
484 fprintf(stderr
, "Message truncated\n");
488 fprintf(stderr
, "!!!Remnant of size %d\n", status
);
494 int rtnl_from_file(FILE *rtnl
, rtnl_filter_t handler
, void *jarg
)
496 struct sockaddr_nl nladdr
;
498 struct nlmsghdr
*h
= (void *)buf
;
500 memset(&nladdr
, 0, sizeof(nladdr
));
501 nladdr
.nl_family
= AF_NETLINK
;
503 nladdr
.nl_groups
= 0;
510 arl
= fread(&buf
, 1, rl
, rtnl
);
517 fprintf(stderr
, "%s: header read failed\n",
520 fprintf(stderr
, "%s: truncated header\n",
525 l
= h
->nlmsg_len
> rl
? h
->nlmsg_len
- rl
: 0;
527 if (l
== 0 || (l
+ (size_t)NLMSG_HDRLEN
) > sizeof(buf
)) {
528 fprintf(stderr
, "%s: malformed message: len=%zu @%lu\n",
529 __func__
, (size_t)h
->nlmsg_len
, ftell(rtnl
));
534 arl
= fread(NLMSG_DATA(h
), 1, rl
, rtnl
);
538 fprintf(stderr
, "%s: msg read failed\n",
541 fprintf(stderr
, "%s: truncated message\n",
546 err
= handler(&nladdr
, h
, jarg
);
552 int addattr32(struct nlmsghdr
*n
, int maxlen
, int type
, __u32 data
)
554 int len
= RTA_LENGTH(4);
556 if ((int)(NLMSG_ALIGN(n
->nlmsg_len
) + len
) > maxlen
) {
558 "addattr32: Error! max allowed bound %d exceeded\n",
563 rta
->rta_type
= type
;
565 memcpy(RTA_DATA(rta
), &data
, 4);
566 n
->nlmsg_len
= NLMSG_ALIGN(n
->nlmsg_len
) + len
;
570 int addattr_l(struct nlmsghdr
*n
, int maxlen
, int type
, const void *data
,
573 int len
= RTA_LENGTH(alen
);
576 if ((int)(NLMSG_ALIGN(n
->nlmsg_len
) + RTA_ALIGN(len
)) > maxlen
) {
578 "addattr_l ERROR: message exceeded bound of %d\n",
583 rta
->rta_type
= type
;
587 memcpy(RTA_DATA(rta
), data
, alen
);
591 n
->nlmsg_len
= NLMSG_ALIGN(n
->nlmsg_len
) + RTA_ALIGN(len
);
595 int addraw_l(struct nlmsghdr
*n
, int maxlen
, const void *data
, int len
)
597 if ((int)(NLMSG_ALIGN(n
->nlmsg_len
) + NLMSG_ALIGN(len
)) > maxlen
) {
599 "addraw_l ERROR: message exceeded bound of %d\n",
604 memcpy(NLMSG_TAIL(n
), data
, len
);
605 memset((uint8_t *)NLMSG_TAIL(n
) + len
, 0, NLMSG_ALIGN(len
) - len
);
606 n
->nlmsg_len
= NLMSG_ALIGN(n
->nlmsg_len
) + NLMSG_ALIGN(len
);
610 struct rtattr
*addattr_nest(struct nlmsghdr
*n
, int maxlen
, int type
)
612 struct rtattr
*nest
= NLMSG_TAIL(n
);
614 addattr_l(n
, maxlen
, type
, NULL
, 0);
618 int addattr_nest_end(struct nlmsghdr
*n
, struct rtattr
*nest
)
620 nest
->rta_len
= (uint8_t *)NLMSG_TAIL(n
) - (uint8_t *)nest
;
624 struct rtattr
*addattr_nest_compat(struct nlmsghdr
*n
, int maxlen
, int type
,
625 const void *data
, int len
)
627 struct rtattr
*start
= NLMSG_TAIL(n
);
629 addattr_l(n
, maxlen
, type
, data
, len
);
630 addattr_nest(n
, maxlen
, type
);
634 int addattr_nest_compat_end(struct nlmsghdr
*n
, struct rtattr
*start
)
636 struct rtattr
*nest
= start
+ NLMSG_ALIGN(start
->rta_len
);
638 start
->rta_len
= (uint8_t *)NLMSG_TAIL(n
) - (uint8_t *)start
;
639 addattr_nest_end(n
, nest
);
643 int rta_addattr32(struct rtattr
*rta
, int maxlen
, int type
, __u32 data
)
645 int len
= RTA_LENGTH(4);
646 struct rtattr
*subrta
;
648 if ((int)(RTA_ALIGN(rta
->rta_len
) + len
) > maxlen
) {
650 "rta_addattr32: Error! max allowed bound %d exceeded\n",
654 subrta
= (struct rtattr
*)(((char *)rta
) + RTA_ALIGN(rta
->rta_len
));
655 subrta
->rta_type
= type
;
656 subrta
->rta_len
= len
;
657 memcpy(RTA_DATA(subrta
), &data
, 4);
658 rta
->rta_len
= NLMSG_ALIGN(rta
->rta_len
) + len
;
662 int rta_addattr_l(struct rtattr
*rta
, int maxlen
, int type
, const void *data
,
665 struct rtattr
*subrta
;
666 int len
= RTA_LENGTH(alen
);
668 if ((int)(RTA_ALIGN(rta
->rta_len
) + RTA_ALIGN(len
)) > maxlen
) {
670 "rta_addattr_l: Error! max allowed bound %d exceeded\n",
674 subrta
= (struct rtattr
*)(((char *)rta
) + RTA_ALIGN(rta
->rta_len
));
675 subrta
->rta_type
= type
;
676 subrta
->rta_len
= len
;
677 memcpy(RTA_DATA(subrta
), data
, alen
);
678 rta
->rta_len
= NLMSG_ALIGN(rta
->rta_len
) + RTA_ALIGN(len
);
682 int parse_rtattr(struct rtattr
*tb
[], int max
, struct rtattr
*rta
, int len
)
684 memset(tb
, 0, sizeof(struct rtattr
*) * (max
+ 1));
685 while (RTA_OK(rta
, len
)) {
686 if ((rta
->rta_type
<= max
) && (!tb
[rta
->rta_type
]))
687 tb
[rta
->rta_type
] = rta
;
688 rta
= RTA_NEXT(rta
, len
);
691 fprintf(stderr
, "!!!Deficit %d, rta_len=%d\n", len
,
696 int parse_rtattr_byindex(struct rtattr
*tb
[], int max
, struct rtattr
*rta
,
701 memset(tb
, 0, sizeof(struct rtattr
*) * max
);
702 while (RTA_OK(rta
, len
)) {
703 if (rta
->rta_type
<= max
&& i
< max
)
705 rta
= RTA_NEXT(rta
, len
);
708 fprintf(stderr
, "!!!Deficit %d, rta_len=%d\n", len
,
713 int __parse_rtattr_nested_compat(struct rtattr
*tb
[], int max
,
714 struct rtattr
*rta
, int len
)
716 if ((int)RTA_PAYLOAD(rta
) < len
)
718 if (RTA_PAYLOAD(rta
) >= RTA_ALIGN(len
) + sizeof(struct rtattr
)) {
719 rta
= (struct rtattr
*)(uint8_t *)RTA_DATA(rta
)
721 return parse_rtattr_nested(tb
, max
, rta
);
723 memset(tb
, 0, sizeof(struct rtattr
*) * (max
+ 1));
727 #endif /* __linux__ */