2 * Copyright (C) 2017 Hans Dedecker <dedeckeh@gmail.com>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License v2 as published by
6 * the Free Software Foundation.
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
19 #include <linux/netlink.h>
20 #include <linux/if_addr.h>
21 #include <linux/neighbour.h>
22 #include <linux/rtnetlink.h>
24 #include <netlink/msg.h>
25 #include <netlink/socket.h>
26 #include <netlink/attr.h>
28 #include <arpa/inet.h>
29 #include <libubox/list.h>
34 struct odhcpd_event ev;
39 static void handle_rtnl_event(struct odhcpd_event *ev);
40 static int cb_rtnl_valid(struct nl_msg *msg, void *arg);
41 static void catch_rtnl_err(struct odhcpd_event *e, int error);
42 static struct nl_sock *create_socket(int protocol);
44 static struct nl_sock *rtnl_socket = NULL;
45 struct list_head netevent_handler_list = LIST_HEAD_INIT(netevent_handler_list);
46 static struct event_socket rtnl_event = {
48 .uloop = {.fd = - 1, },
50 .handle_error = catch_rtnl_err,
51 .recv_msgs = handle_rtnl_event,
54 .sock_bufsize = 133120,
57 int netlink_init(void)
59 rtnl_socket = create_socket(NETLINK_ROUTE);
61 syslog(LOG_ERR, "Unable to open nl socket: %m");
65 rtnl_event.sock = create_socket(NETLINK_ROUTE);
66 if (!rtnl_event.sock) {
67 syslog(LOG_ERR, "Unable to open nl event socket: %m");
71 rtnl_event.ev.uloop.fd = nl_socket_get_fd(rtnl_event.sock);
73 if (nl_socket_set_buffer_size(rtnl_event.sock, rtnl_event.sock_bufsize, 0))
76 nl_socket_disable_seq_check(rtnl_event.sock);
78 nl_socket_modify_cb(rtnl_event.sock, NL_CB_VALID, NL_CB_CUSTOM,
81 /* Receive IPv4 address, IPv6 address, IPv6 routes and neighbor events */
82 if (nl_socket_add_memberships(rtnl_event.sock, RTNLGRP_IPV4_IFADDR,
83 RTNLGRP_IPV6_IFADDR, RTNLGRP_IPV6_ROUTE,
84 RTNLGRP_NEIGH, RTNLGRP_LINK, 0))
87 odhcpd_register(&rtnl_event.ev);
93 nl_socket_free(rtnl_socket);
97 if (rtnl_event.sock) {
98 nl_socket_free(rtnl_event.sock);
99 rtnl_event.sock = NULL;
100 rtnl_event.ev.uloop.fd = -1;
107 int netlink_add_netevent_handler(struct netevent_handler *handler)
112 list_add(&handler->head, &netevent_handler_list);
117 static void call_netevent_handler_list(unsigned long event, struct netevent_handler_info *info)
119 struct netevent_handler *handler;
121 list_for_each_entry(handler, &netevent_handler_list, head)
122 handler->cb(event, info);
125 static void handle_rtnl_event(struct odhcpd_event *e)
127 struct event_socket *ev_sock = container_of(e, struct event_socket, ev);
129 nl_recvmsgs_default(ev_sock->sock);
132 static void refresh_iface_addr4(int ifindex)
134 struct odhcpd_ipaddr *addr = NULL;
135 struct interface *iface;
136 ssize_t len = netlink_get_interface_addrs(ifindex, false, &addr);
142 avl_for_each_element(&interfaces, iface, avl) {
143 if (iface->ifindex != ifindex)
146 change = len != (ssize_t)iface->addr4_len;
147 for (ssize_t i = 0; !change && i < len; ++i) {
148 if (addr[i].addr.in.s_addr != iface->addr4[i].addr.in.s_addr)
159 avl_for_element_range(iface, avl_last_element(&interfaces, iface, avl), iface, avl) {
160 struct netevent_handler_info event_info;
162 if (iface->ifindex != ifindex)
165 memset(&event_info, 0, sizeof(event_info));
166 event_info.iface = iface;
167 event_info.addrs_old.addrs = iface->addr4;
168 event_info.addrs_old.len = iface->addr4_len;
171 iface->addr4_len = len;
173 call_netevent_handler_list(NETEV_ADDRLIST_CHANGE, &event_info);
175 free(event_info.addrs_old.addrs);
178 addr = malloc(len * sizeof(*addr));
182 memcpy(addr, iface->addr4, len * sizeof(*addr));
189 static void refresh_iface_addr6(int ifindex)
191 struct odhcpd_ipaddr *addr = NULL;
192 struct interface *iface;
193 ssize_t len = netlink_get_interface_addrs(ifindex, true, &addr);
194 time_t now = odhcpd_time();
200 avl_for_each_element(&interfaces, iface, avl) {
201 if (iface->ifindex != ifindex)
204 change = len != (ssize_t)iface->addr6_len;
205 for (ssize_t i = 0; !change && i < len; ++i) {
206 if (!IN6_ARE_ADDR_EQUAL(&addr[i].addr.in6, &iface->addr6[i].addr.in6) ||
207 (addr[i].preferred > (uint32_t)now) != (iface->addr6[i].preferred > (uint32_t)now) ||
208 addr[i].valid < iface->addr6[i].valid || addr[i].preferred < iface->addr6[i].preferred)
219 avl_for_element_range(iface, avl_last_element(&interfaces, iface, avl), iface, avl) {
220 struct netevent_handler_info event_info;
222 if (iface->ifindex != ifindex)
225 memset(&event_info, 0, sizeof(event_info));
226 event_info.iface = iface;
227 event_info.addrs_old.addrs = iface->addr6;
228 event_info.addrs_old.len = iface->addr6_len;
231 iface->addr6_len = len;
233 call_netevent_handler_list(NETEV_ADDR6LIST_CHANGE, &event_info);
235 free(event_info.addrs_old.addrs);
238 addr = malloc(len * sizeof(*addr));
242 memcpy(addr, iface->addr6, len * sizeof(*addr));
249 static int handle_rtm_link(struct nlmsghdr *hdr)
251 struct ifinfomsg *ifi = nlmsg_data(hdr);
252 struct nlattr *nla[__IFLA_MAX];
253 struct interface *iface;
254 struct netevent_handler_info event_info;
257 memset(&event_info, 0, sizeof(event_info));
259 if (!nlmsg_valid_hdr(hdr, sizeof(*ifi)) || ifi->ifi_family != AF_UNSPEC)
262 nlmsg_parse(hdr, sizeof(*ifi), nla, __IFLA_MAX - 1, NULL);
263 if (!nla[IFLA_IFNAME])
266 ifname = nla_get_string(nla[IFLA_IFNAME]);
268 avl_for_each_element(&interfaces, iface, avl) {
269 if (strcmp(iface->ifname, ifname) || iface->ifindex == ifi->ifi_index)
272 iface->ifindex = ifi->ifi_index;
273 event_info.iface = iface;
274 call_netevent_handler_list(NETEV_IFINDEX_CHANGE, &event_info);
280 static int handle_rtm_route(struct nlmsghdr *hdr, bool add)
282 struct rtmsg *rtm = nlmsg_data(hdr);
283 struct nlattr *nla[__RTA_MAX];
284 struct interface *iface;
285 struct netevent_handler_info event_info;
288 if (!nlmsg_valid_hdr(hdr, sizeof(*rtm)) || rtm->rtm_family != AF_INET6)
291 nlmsg_parse(hdr, sizeof(*rtm), nla, __RTA_MAX - 1, NULL);
293 memset(&event_info, 0, sizeof(event_info));
294 event_info.rt.dst_len = rtm->rtm_dst_len;
297 nla_memcpy(&event_info.rt.dst, nla[RTA_DST],
298 sizeof(event_info.rt.dst));
301 ifindex = nla_get_u32(nla[RTA_OIF]);
303 if (nla[RTA_GATEWAY])
304 nla_memcpy(&event_info.rt.gateway, nla[RTA_GATEWAY],
305 sizeof(event_info.rt.gateway));
307 avl_for_each_element(&interfaces, iface, avl) {
308 if (ifindex && iface->ifindex != ifindex)
311 event_info.iface = ifindex ? iface : NULL;
312 call_netevent_handler_list(add ? NETEV_ROUTE6_ADD : NETEV_ROUTE6_DEL,
319 static int handle_rtm_addr(struct nlmsghdr *hdr, bool add)
321 struct ifaddrmsg *ifa = nlmsg_data(hdr);
322 struct nlattr *nla[__IFA_MAX];
323 struct interface *iface;
324 struct netevent_handler_info event_info;
325 char buf[INET6_ADDRSTRLEN];
327 if (!nlmsg_valid_hdr(hdr, sizeof(*ifa)) ||
328 (ifa->ifa_family != AF_INET6 &&
329 ifa->ifa_family != AF_INET))
332 memset(&event_info, 0, sizeof(event_info));
334 nlmsg_parse(hdr, sizeof(*ifa), nla, __IFA_MAX - 1, NULL);
336 if (ifa->ifa_family == AF_INET6) {
337 if (!nla[IFA_ADDRESS])
340 nla_memcpy(&event_info.addr, nla[IFA_ADDRESS], sizeof(event_info.addr));
342 if (IN6_IS_ADDR_LINKLOCAL(&event_info.addr) || IN6_IS_ADDR_MULTICAST(&event_info.addr))
345 inet_ntop(AF_INET6, &event_info.addr, buf, sizeof(buf));
347 avl_for_each_element(&interfaces, iface, avl) {
348 if (iface->ifindex != (int)ifa->ifa_index)
351 syslog(LOG_DEBUG, "Netlink %s %s on %s", add ? "newaddr" : "deladdr",
354 event_info.iface = iface;
355 call_netevent_handler_list(add ? NETEV_ADDR6_ADD : NETEV_ADDR6_DEL,
359 refresh_iface_addr6(ifa->ifa_index);
364 nla_memcpy(&event_info.addr, nla[IFA_LOCAL], sizeof(event_info.addr));
366 inet_ntop(AF_INET, &event_info.addr, buf, sizeof(buf));
368 avl_for_each_element(&interfaces, iface, avl) {
369 if (iface->ifindex != (int)ifa->ifa_index)
372 syslog(LOG_DEBUG, "Netlink %s %s on %s", add ? "newaddr" : "deladdr",
375 event_info.iface = iface;
376 call_netevent_handler_list(add ? NETEV_ADDR_ADD : NETEV_ADDR_DEL,
380 refresh_iface_addr4(ifa->ifa_index);
386 static int handle_rtm_neigh(struct nlmsghdr *hdr, bool add)
388 struct ndmsg *ndm = nlmsg_data(hdr);
389 struct nlattr *nla[__NDA_MAX];
390 struct interface *iface;
391 struct netevent_handler_info event_info;
392 char buf[INET6_ADDRSTRLEN];
394 if (!nlmsg_valid_hdr(hdr, sizeof(*ndm)) ||
395 ndm->ndm_family != AF_INET6)
398 nlmsg_parse(hdr, sizeof(*ndm), nla, __NDA_MAX - 1, NULL);
402 memset(&event_info, 0, sizeof(event_info));
404 nla_memcpy(&event_info.neigh.dst, nla[NDA_DST], sizeof(event_info.neigh.dst));
406 if (IN6_IS_ADDR_LINKLOCAL(&event_info.neigh.dst) ||
407 IN6_IS_ADDR_MULTICAST(&event_info.neigh.dst))
410 inet_ntop(AF_INET6, &event_info.neigh.dst, buf, sizeof(buf));
412 avl_for_each_element(&interfaces, iface, avl) {
413 if (iface->ifindex != ndm->ndm_ifindex)
416 syslog(LOG_DEBUG, "Netlink %s %s on %s", true ? "newneigh" : "delneigh",
419 event_info.neigh.state = ndm->ndm_state;
420 event_info.neigh.flags = ndm->ndm_flags;
422 call_netevent_handler_list(add ? NETEV_NEIGH6_ADD : NETEV_NEIGH6_DEL,
429 /* Handler for neighbor cache entries from the kernel. This is our source
430 * to learn and unlearn hosts on interfaces. */
431 static int cb_rtnl_valid(struct nl_msg *msg, _unused void *arg)
433 struct nlmsghdr *hdr = nlmsg_hdr(msg);
437 switch (hdr->nlmsg_type) {
439 ret = handle_rtm_link(hdr);
446 ret = handle_rtm_route(hdr, add);
453 ret = handle_rtm_addr(hdr, add);
460 ret = handle_rtm_neigh(hdr, add);
470 static void catch_rtnl_err(struct odhcpd_event *e, int error)
472 struct event_socket *ev_sock = container_of(e, struct event_socket, ev);
474 if (error != ENOBUFS)
477 /* Double netlink event buffer size */
478 ev_sock->sock_bufsize *= 2;
480 if (nl_socket_set_buffer_size(ev_sock->sock, ev_sock->sock_bufsize, 0))
483 netlink_dump_addr_table(true);
487 odhcpd_deregister(e);
490 static struct nl_sock *create_socket(int protocol)
492 struct nl_sock *nl_sock;
494 nl_sock = nl_socket_alloc();
498 if (nl_connect(nl_sock, protocol) < 0)
505 nl_socket_free(nl_sock);
514 struct odhcpd_ipaddr **addrs;
520 static int cb_valid_handler(struct nl_msg *msg, void *arg)
522 struct addr_info *ctxt = (struct addr_info *)arg;
523 struct odhcpd_ipaddr *addrs = *(ctxt->addrs);
524 struct nlmsghdr *hdr = nlmsg_hdr(msg);
525 struct ifaddrmsg *ifa;
526 struct nlattr *nla[__IFA_MAX], *nla_addr = NULL;
528 if (hdr->nlmsg_type != RTM_NEWADDR)
531 ifa = NLMSG_DATA(hdr);
532 if (ifa->ifa_scope != RT_SCOPE_UNIVERSE ||
533 (ctxt->af != ifa->ifa_family) ||
534 (ctxt->ifindex && ifa->ifa_index != (unsigned)ctxt->ifindex))
537 nlmsg_parse(hdr, sizeof(*ifa), nla, __IFA_MAX - 1, NULL);
539 switch (ifa->ifa_family) {
541 if (nla[IFA_ADDRESS])
542 nla_addr = nla[IFA_ADDRESS];
547 nla_addr = nla[IFA_LOCAL];
556 addrs = realloc(addrs, sizeof(*addrs)*(ctxt->ret + 1));
560 memset(&addrs[ctxt->ret], 0, sizeof(addrs[ctxt->ret]));
561 addrs[ctxt->ret].prefix = ifa->ifa_prefixlen;
563 nla_memcpy(&addrs[ctxt->ret].addr, nla_addr,
564 sizeof(addrs[ctxt->ret].addr));
566 if (nla[IFA_BROADCAST])
567 nla_memcpy(&addrs[ctxt->ret].broadcast, nla[IFA_BROADCAST],
568 sizeof(addrs[ctxt->ret].broadcast));
570 if (nla[IFA_CACHEINFO]) {
571 struct ifa_cacheinfo *ifc = nla_data(nla[IFA_CACHEINFO]);
573 addrs[ctxt->ret].preferred = ifc->ifa_prefered;
574 addrs[ctxt->ret].valid = ifc->ifa_valid;
577 if (ifa->ifa_flags & IFA_F_DEPRECATED)
578 addrs[ctxt->ret].preferred = 0;
581 *(ctxt->addrs) = addrs;
587 static int cb_finish_handler(_unused struct nl_msg *msg, void *arg)
589 struct addr_info *ctxt = (struct addr_info *)arg;
597 static int cb_error_handler(_unused struct sockaddr_nl *nla, struct nlmsgerr *err,
600 struct addr_info *ctxt = (struct addr_info *)arg;
603 ctxt->ret = err->error;
609 static int prefix_cmp(const void *va, const void *vb)
611 const struct odhcpd_ipaddr *a = va, *b = vb;
614 if (a->prefix == b->prefix) {
615 ret = (ntohl(a->addr.in.s_addr) < ntohl(b->addr.in.s_addr)) ? 1 :
616 (ntohl(a->addr.in.s_addr) > ntohl(b->addr.in.s_addr)) ? -1 : 0;
618 ret = a->prefix < b->prefix ? 1 : -1;
624 /* compare IPv6 prefixes */
625 static int prefix6_cmp(const void *va, const void *vb)
627 const struct odhcpd_ipaddr *a = va, *b = vb;
628 uint32_t a_pref = IN6_IS_ADDR_ULA(&a->addr.in6) ? 1 : a->preferred;
629 uint32_t b_pref = IN6_IS_ADDR_ULA(&b->addr.in6) ? 1 : b->preferred;
630 return (a_pref < b_pref) ? 1 : (a_pref > b_pref) ? -1 : 0;
634 /* Detect an IPV6-address currently assigned to the given interface */
635 ssize_t netlink_get_interface_addrs(int ifindex, bool v6, struct odhcpd_ipaddr **addrs)
638 struct ifaddrmsg ifa = {
639 .ifa_family = v6? AF_INET6: AF_INET,
643 .ifa_index = ifindex, };
644 struct nl_cb *cb = nl_cb_alloc(NL_CB_DEFAULT);
645 struct addr_info ctxt = {
647 .af = v6? AF_INET6: AF_INET,
658 msg = nlmsg_alloc_simple(RTM_GETADDR, NLM_F_REQUEST | NLM_F_DUMP);
665 nlmsg_append(msg, &ifa, sizeof(ifa), 0);
667 nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, cb_valid_handler, &ctxt);
668 nl_cb_set(cb, NL_CB_FINISH, NL_CB_CUSTOM, cb_finish_handler, &ctxt);
669 nl_cb_err(cb, NL_CB_CUSTOM, cb_error_handler, &ctxt);
671 nl_send_auto_complete(rtnl_socket, msg);
672 while (ctxt.pending > 0)
673 nl_recvmsgs(rtnl_socket, cb);
680 time_t now = odhcpd_time();
681 struct odhcpd_ipaddr *addr = *addrs;
683 qsort(addr, ctxt.ret, sizeof(*addr), v6 ? prefix6_cmp : prefix_cmp);
685 for (ssize_t i = 0; i < ctxt.ret; ++i) {
686 if (addr[i].preferred < UINT32_MAX - now)
687 addr[i].preferred += now;
689 if (addr[i].valid < UINT32_MAX - now)
690 addr[i].valid += now;
700 int netlink_setup_route(const struct in6_addr *addr, const int prefixlen,
701 const int ifindex, const struct in6_addr *gw,
702 const uint32_t metric, const bool add)
706 .rtm_family = AF_INET6,
707 .rtm_dst_len = prefixlen,
709 .rtm_table = RT_TABLE_MAIN,
710 .rtm_protocol = (add ? RTPROT_STATIC : RTPROT_UNSPEC),
711 .rtm_scope = (add ? (gw ? RT_SCOPE_UNIVERSE : RT_SCOPE_LINK) : RT_SCOPE_NOWHERE),
712 .rtm_type = (add ? RTN_UNICAST : RTN_UNSPEC),
716 msg = nlmsg_alloc_simple(add ? RTM_NEWROUTE : RTM_DELROUTE,
717 add ? NLM_F_CREATE | NLM_F_REPLACE : 0);
721 nlmsg_append(msg, &rtm, sizeof(rtm), 0);
723 nla_put(msg, RTA_DST, sizeof(*addr), addr);
724 nla_put_u32(msg, RTA_OIF, ifindex);
725 nla_put_u32(msg, RTA_PRIORITY, metric);
728 nla_put(msg, RTA_GATEWAY, sizeof(*gw), gw);
730 ret = nl_send_auto_complete(rtnl_socket, msg);
736 return nl_wait_for_ack(rtnl_socket);
740 int netlink_setup_proxy_neigh(const struct in6_addr *addr,
741 const int ifindex, const bool add)
745 .ndm_family = AF_INET6,
746 .ndm_flags = NTF_PROXY,
747 .ndm_ifindex = ifindex,
749 int ret = 0, flags = NLM_F_REQUEST;
752 flags |= NLM_F_REPLACE | NLM_F_CREATE;
754 msg = nlmsg_alloc_simple(add ? RTM_NEWNEIGH : RTM_DELNEIGH, flags);
758 nlmsg_append(msg, &ndm, sizeof(ndm), 0);
760 nla_put(msg, NDA_DST, sizeof(*addr), addr);
762 ret = nl_send_auto_complete(rtnl_socket, msg);
768 return nl_wait_for_ack(rtnl_socket);
772 int netlink_setup_addr(struct odhcpd_ipaddr *addr,
773 const int ifindex, const bool v6, const bool add)
776 struct ifaddrmsg ifa = {
777 .ifa_family = v6 ? AF_INET6 : AF_INET,
778 .ifa_prefixlen = addr->prefix,
781 .ifa_index = ifindex, };
782 int ret = 0, flags = NLM_F_REQUEST;
785 flags |= NLM_F_REPLACE | NLM_F_CREATE;
787 msg = nlmsg_alloc_simple(add ? RTM_NEWADDR : RTM_DELADDR, 0);
791 nlmsg_append(msg, &ifa, sizeof(ifa), flags);
792 nla_put(msg, IFA_LOCAL, v6 ? 16 : 4, &addr->addr);
794 struct ifa_cacheinfo cinfo = { .ifa_prefered = 0xffffffffU,
795 .ifa_valid = 0xffffffffU,
798 time_t now = odhcpd_time();
800 if (addr->preferred) {
801 int64_t preferred = addr->preferred - now;
804 else if (preferred > UINT32_MAX)
805 preferred = UINT32_MAX;
807 cinfo.ifa_prefered = preferred;
811 int64_t valid = addr->valid - now;
816 else if (valid > UINT32_MAX)
819 cinfo.ifa_valid = valid;
822 nla_put(msg, IFA_CACHEINFO, sizeof(cinfo), &cinfo);
824 nla_put_u32(msg, IFA_FLAGS, IFA_F_NOPREFIXROUTE);
826 if (addr->broadcast.s_addr)
827 nla_put_u32(msg, IFA_BROADCAST, addr->broadcast.s_addr);
830 ret = nl_send_auto_complete(rtnl_socket, msg);
836 return nl_wait_for_ack(rtnl_socket);
839 void netlink_dump_neigh_table(const bool proxy)
843 .ndm_family = AF_INET6,
844 .ndm_flags = proxy ? NTF_PROXY : 0,
847 msg = nlmsg_alloc_simple(RTM_GETNEIGH, NLM_F_REQUEST | NLM_F_DUMP);
851 nlmsg_append(msg, &ndm, sizeof(ndm), 0);
853 nl_send_auto_complete(rtnl_event.sock, msg);
858 void netlink_dump_addr_table(const bool v6)
861 struct ifaddrmsg ifa = {
862 .ifa_family = v6 ? AF_INET6 : AF_INET,
865 msg = nlmsg_alloc_simple(RTM_GETADDR, NLM_F_REQUEST | NLM_F_DUMP);
869 nlmsg_append(msg, &ifa, sizeof(ifa), 0);
871 nl_send_auto_complete(rtnl_event.sock, msg);