3 #include <sys/socket.h>
6 #include <sys/syscall.h>
8 #include <linux/rtnetlink.h>
9 #include <linux/sockios.h>
10 #include <linux/if_vlan.h>
11 #include <linux/if_bridge.h>
18 #include <netlink/msg.h>
19 #include <netlink/attr.h>
20 #include <netlink/socket.h>
21 #include <libubox/uloop.h>
28 struct uloop_fd uloop;
33 static int sock_ioctl = -1;
34 static struct nl_sock *sock_rtnl = NULL;
36 static int cb_rtnl_event(struct nl_msg *msg, void *arg);
37 static void handle_hotplug_event(struct uloop_fd *u, unsigned int events);
40 handler_nl_event(struct uloop_fd *u, unsigned int events)
42 struct event_socket *ev = container_of(u, struct event_socket, uloop);
43 nl_recvmsgs(ev->sock, ev->cb);
46 static struct nl_sock *
47 create_socket(int protocol, int groups)
51 sock = nl_socket_alloc();
56 nl_join_groups(sock, groups);
58 if (nl_connect(sock, protocol))
65 create_raw_event_socket(struct event_socket *ev, int protocol, int groups,
68 ev->sock = create_socket(protocol, groups);
72 ev->uloop.fd = nl_socket_get_fd(ev->sock);
74 uloop_fd_add(&ev->uloop, ULOOP_READ | ULOOP_EDGE_TRIGGER);
79 create_event_socket(struct event_socket *ev, int protocol,
80 int (*cb)(struct nl_msg *msg, void *arg))
82 // Prepare socket for link events
83 ev->cb = nl_cb_alloc(NL_CB_DEFAULT);
87 nl_cb_set(ev->cb, NL_CB_VALID, NL_CB_CUSTOM, cb, NULL);
89 return create_raw_event_socket(ev, protocol, 0, handler_nl_event);
94 static struct event_socket rtnl_event;
95 static struct event_socket hotplug_event;
97 sock_ioctl = socket(AF_LOCAL, SOCK_DGRAM, 0);
98 fcntl(sock_ioctl, F_SETFD, fcntl(sock_ioctl, F_GETFD) | FD_CLOEXEC);
100 // Prepare socket for routing / address control
101 sock_rtnl = create_socket(NETLINK_ROUTE, 0);
105 if (!create_event_socket(&rtnl_event, NETLINK_ROUTE, cb_rtnl_event))
108 if (!create_raw_event_socket(&hotplug_event, NETLINK_KOBJECT_UEVENT, 1,
109 handle_hotplug_event))
112 // Receive network link events form kernel
113 nl_socket_add_membership(rtnl_event.sock, RTNLGRP_LINK);
118 static void system_set_sysctl(const char *path, const char *val)
122 fd = open(path, O_WRONLY);
126 write(fd, val, strlen(val));
130 static void system_set_dev_sysctl(const char *path, const char *device, const char *val)
132 static char buf[256];
134 snprintf(buf, sizeof(buf), path, val);
135 system_set_sysctl(buf, val);
138 static void system_set_disable_ipv6(struct device *dev, const char *val)
140 system_set_dev_sysctl("/proc/sys/net/ipv6/conf/%s/disable_ipv6", dev->ifname, val);
143 // Evaluate netlink messages
144 static int cb_rtnl_event(struct nl_msg *msg, void *arg)
146 struct nlmsghdr *nh = nlmsg_hdr(msg);
147 struct ifinfomsg *ifi = NLMSG_DATA(nh);
148 struct nlattr *nla[__IFLA_MAX];
150 if (nh->nlmsg_type != RTM_DELLINK && nh->nlmsg_type != RTM_NEWLINK)
153 nlmsg_parse(nh, sizeof(*ifi), nla, __IFLA_MAX - 1, NULL);
154 if (!nla[IFLA_IFNAME])
157 struct device *dev = device_get(RTA_DATA(nla[IFLA_IFNAME]), false);
161 dev->ifindex = ifi->ifi_index;
162 /* TODO: parse link status */
169 handle_hotplug_msg(char *data, int size)
171 const char *subsystem = NULL, *interface = NULL;
172 char *cur, *end, *sep;
177 if (!strncmp(data, "add@", 4))
179 else if (!strncmp(data, "remove@", 7))
184 skip = strlen(data) + 1;
187 for (cur = data + skip; cur < end; cur += skip) {
188 skip = strlen(cur) + 1;
190 sep = strchr(cur, '=');
195 if (!strcmp(cur, "INTERFACE"))
197 else if (!strcmp(cur, "SUBSYSTEM")) {
199 if (strcmp(subsystem, "net") != 0)
202 if (subsystem && interface)
208 dev = device_get(interface, false);
212 if (dev->type != &simple_device_type)
215 device_set_present(dev, add);
219 handle_hotplug_event(struct uloop_fd *u, unsigned int events)
221 struct event_socket *ev = container_of(u, struct event_socket, uloop);
222 struct sockaddr_nl nla;
223 unsigned char *buf = NULL;
226 while ((size = nl_recv(ev->sock, &nla, &buf, NULL)) > 0) {
228 handle_hotplug_msg((char *) buf, size);
234 static int system_rtnl_call(struct nl_msg *msg)
236 int s = -(nl_send_auto_complete(sock_rtnl, msg)
237 || nl_wait_for_ack(sock_rtnl));
242 int system_bridge_delbr(struct device *bridge)
244 return ioctl(sock_ioctl, SIOCBRDELBR, bridge->ifname);
247 static int system_bridge_if(const char *bridge, struct device *dev, int cmd, void *data)
251 ifr.ifr_ifindex = dev->ifindex;
254 strncpy(ifr.ifr_name, bridge, sizeof(ifr.ifr_name));
255 return ioctl(sock_ioctl, cmd, &ifr);
258 int system_bridge_addif(struct device *bridge, struct device *dev)
260 system_set_disable_ipv6(dev, "1");
261 return system_bridge_if(bridge->ifname, dev, SIOCBRADDIF, NULL);
264 int system_bridge_delif(struct device *bridge, struct device *dev)
266 system_set_disable_ipv6(dev, "0");
267 return system_bridge_if(bridge->ifname, dev, SIOCBRDELIF, NULL);
270 static bool system_is_bridge(const char *name, char *buf, int buflen)
274 snprintf(buf, buflen, "/sys/devices/virtual/net/%s/bridge", name);
275 if (stat(buf, &st) < 0)
281 static char *system_get_bridge(const char *name, char *buf, int buflen)
287 snprintf(buf, buflen, "/sys/devices/virtual/net/*/brif/%s/bridge", name);
288 if (glob(buf, GLOB_NOSORT, NULL, &gl) < 0)
291 if (gl.gl_pathc == 0)
294 len = readlink(gl.gl_pathv[0], buf, buflen);
299 path = strrchr(buf, '/');
306 static int system_if_resolve(struct device *dev)
309 strncpy(ifr.ifr_name, dev->ifname, sizeof(ifr.ifr_name));
310 if (!ioctl(sock_ioctl, SIOCGIFINDEX, &ifr))
311 return ifr.ifr_ifindex;
316 static int system_if_flags(const char *ifname, unsigned add, unsigned rem)
319 strncpy(ifr.ifr_name, ifname, sizeof(ifr.ifr_name));
320 ioctl(sock_ioctl, SIOCGIFFLAGS, &ifr);
321 ifr.ifr_flags |= add;
322 ifr.ifr_flags &= ~rem;
323 return ioctl(sock_ioctl, SIOCSIFFLAGS, &ifr);
335 static bool check_ifaddr(struct nlmsghdr *hdr, int ifindex)
337 struct ifaddrmsg *ifa = NLMSG_DATA(hdr);
339 return ifa->ifa_index == ifindex;
342 static bool check_route(struct nlmsghdr *hdr, int ifindex)
344 struct nlattr *tb[__RTA_MAX];
346 nlmsg_parse(hdr, sizeof(struct rtmsg), tb, __RTA_MAX - 1, NULL);
350 return *(int *)RTA_DATA(tb[RTA_OIF]) == ifindex;
353 static int cb_clear_event(struct nl_msg *msg, void *arg)
355 struct clear_data *clr = arg;
356 struct nlmsghdr *hdr = nlmsg_hdr(msg);
357 bool (*cb)(struct nlmsghdr *, int ifindex);
363 if (hdr->nlmsg_type != RTM_NEWADDR)
370 if (hdr->nlmsg_type != RTM_NEWROUTE)
379 if (!cb(hdr, clr->dev->ifindex))
382 D(SYSTEM, "Remove %s from device %s\n",
383 type == RTM_DELADDR ? "an address" : "a route",
385 memcpy(nlmsg_hdr(clr->msg), hdr, hdr->nlmsg_len);
386 hdr = nlmsg_hdr(clr->msg);
387 hdr->nlmsg_type = type;
388 hdr->nlmsg_flags = NLM_F_REQUEST;
390 if (!nl_send_auto_complete(sock_rtnl, clr->msg))
391 nl_wait_for_ack(sock_rtnl);
397 cb_finish_event(struct nl_msg *msg, void *arg)
405 error_handler(struct sockaddr_nl *nla, struct nlmsgerr *err, void *arg)
408 *pending = err->error;
413 system_if_clear_entries(struct device *dev, int type, int af)
415 struct clear_data clr;
416 struct nl_cb *cb = nl_cb_alloc(NL_CB_DEFAULT);
419 .rtm_flags = RTM_F_CLONED,
421 int flags = NLM_F_DUMP;
429 clr.size = sizeof(struct rtgenmsg);
432 clr.size = sizeof(struct rtmsg);
441 clr.msg = nlmsg_alloc_simple(type, flags);
445 nlmsg_append(clr.msg, &rtm, clr.size, 0);
446 nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, cb_clear_event, &clr);
447 nl_cb_set(cb, NL_CB_FINISH, NL_CB_CUSTOM, cb_finish_event, &pending);
448 nl_cb_err(cb, NL_CB_CUSTOM, error_handler, &pending);
450 nl_send_auto_complete(sock_rtnl, clr.msg);
452 nl_recvmsgs(sock_rtnl, cb);
460 * Clear bridge (membership) state and bring down device
462 void system_if_clear_state(struct device *dev)
464 static char buf[256];
470 dev->ifindex = system_if_resolve(dev);
474 system_if_flags(dev->ifname, 0, IFF_UP);
476 if (system_is_bridge(dev->ifname, buf, sizeof(buf))) {
477 D(SYSTEM, "Delete existing bridge named '%s'\n", dev->ifname);
478 system_bridge_delbr(dev);
482 bridge = system_get_bridge(dev->ifname, buf, sizeof(buf));
484 D(SYSTEM, "Remove device '%s' from bridge '%s'\n", dev->ifname, bridge);
485 system_bridge_if(bridge, dev, SIOCBRDELIF, NULL);
488 system_if_clear_entries(dev, RTM_GETROUTE, AF_INET);
489 system_if_clear_entries(dev, RTM_GETADDR, AF_INET);
490 system_if_clear_entries(dev, RTM_GETROUTE, AF_INET6);
491 system_if_clear_entries(dev, RTM_GETADDR, AF_INET6);
492 system_set_disable_ipv6(dev, "0");
495 static inline unsigned long
496 sec_to_jiffies(int val)
498 return (unsigned long) val * 100;
501 int system_bridge_addbr(struct device *bridge, struct bridge_config *cfg)
503 unsigned long args[4] = {};
505 if (ioctl(sock_ioctl, SIOCBRADDBR, bridge->ifname) < 0)
508 args[0] = BRCTL_SET_BRIDGE_STP_STATE;
509 args[1] = !!cfg->stp;
510 system_bridge_if(bridge->ifname, NULL, SIOCDEVPRIVATE, &args);
512 args[0] = BRCTL_SET_BRIDGE_FORWARD_DELAY;
513 args[1] = sec_to_jiffies(cfg->forward_delay);
514 system_bridge_if(bridge->ifname, NULL, SIOCDEVPRIVATE, &args);
516 system_set_dev_sysctl("/sys/devices/virtual/net/%s/bridge/multicast_snooping",
517 bridge->ifname, cfg->igmp_snoop ? "1" : "0");
519 if (cfg->flags & BRIDGE_OPT_AGEING_TIME) {
520 args[0] = BRCTL_SET_AGEING_TIME;
521 args[1] = sec_to_jiffies(cfg->ageing_time);
522 system_bridge_if(bridge->ifname, NULL, SIOCDEVPRIVATE, &args);
525 if (cfg->flags & BRIDGE_OPT_HELLO_TIME) {
526 args[0] = BRCTL_SET_BRIDGE_HELLO_TIME;
527 args[1] = sec_to_jiffies(cfg->hello_time);
528 system_bridge_if(bridge->ifname, NULL, SIOCDEVPRIVATE, &args);
531 if (cfg->flags & BRIDGE_OPT_MAX_AGE) {
532 args[0] = BRCTL_SET_BRIDGE_MAX_AGE;
533 args[1] = sec_to_jiffies(cfg->max_age);
534 system_bridge_if(bridge->ifname, NULL, SIOCDEVPRIVATE, &args);
540 static int system_vlan(struct device *dev, int id)
542 struct vlan_ioctl_args ifr = {
543 .cmd = SET_VLAN_NAME_TYPE_CMD,
544 .u.name_type = VLAN_NAME_TYPE_RAW_PLUS_VID_NO_PAD,
547 ioctl(sock_ioctl, SIOCSIFVLAN, &ifr);
550 ifr.cmd = DEL_VLAN_CMD;
553 ifr.cmd = ADD_VLAN_CMD;
556 strncpy(ifr.device1, dev->ifname, sizeof(ifr.device1));
557 return ioctl(sock_ioctl, SIOCSIFVLAN, &ifr);
560 int system_vlan_add(struct device *dev, int id)
562 return system_vlan(dev, id);
565 int system_vlan_del(struct device *dev)
567 return system_vlan(dev, -1);
571 system_if_apply_settings(struct device *dev)
575 memset(&ifr, 0, sizeof(ifr));
576 strncpy(ifr.ifr_name, dev->ifname, sizeof(ifr.ifr_name));
577 if (dev->flags & DEV_OPT_MTU) {
578 ifr.ifr_mtu = dev->mtu;
579 ioctl(sock_ioctl, SIOCSIFMTU, &ifr);
581 if (dev->flags & DEV_OPT_TXQUEUELEN) {
582 ifr.ifr_qlen = dev->txqueuelen;
583 ioctl(sock_ioctl, SIOCSIFTXQLEN, &ifr);
585 if (dev->flags & DEV_OPT_MACADDR) {
586 memcpy(&ifr.ifr_hwaddr, dev->macaddr, sizeof(dev->macaddr));
587 ioctl(sock_ioctl, SIOCSIFHWADDR, &ifr);
590 dev->ifindex = system_if_resolve(dev);
593 int system_if_up(struct device *dev)
595 system_if_apply_settings(dev);
596 return system_if_flags(dev->ifname, IFF_UP, 0);
599 int system_if_down(struct device *dev)
601 return system_if_flags(dev->ifname, 0, IFF_UP);
604 int system_if_check(struct device *dev)
606 device_set_present(dev, (system_if_resolve(dev) > 0));
611 system_if_get_parent(struct device *dev)
613 char buf[64], *devname;
614 int ifindex, iflink, len;
617 snprintf(buf, sizeof(buf), "/sys/class/net/%s/iflink", dev->ifname);
622 len = fread(buf, 1, sizeof(buf) - 1, f);
629 iflink = strtoul(buf, NULL, 0);
630 ifindex = system_if_resolve(dev);
631 if (!iflink || iflink == ifindex)
634 devname = if_indextoname(iflink, buf);
638 return device_get(devname, true);
641 int system_if_dump_stats(struct device *dev, struct blob_buf *b)
643 const char *const counters[] = {
644 "collisions", "rx_frame_errors", "tx_compressed",
645 "multicast", "rx_length_errors", "tx_dropped",
646 "rx_bytes", "rx_missed_errors", "tx_errors",
647 "rx_compressed", "rx_over_errors", "tx_fifo_errors",
648 "rx_crc_errors", "rx_packets", "tx_heartbeat_errors",
649 "rx_dropped", "tx_aborted_errors", "tx_packets",
650 "rx_errors", "tx_bytes", "tx_window_errors",
651 "rx_fifo_errors", "tx_carrier_errors",
657 snprintf(buf, sizeof(buf), "/sys/class/net/%s/statistics", dev->ifname);
658 stats_dir = open(buf, O_DIRECTORY);
662 for (i = 0; i < ARRAY_SIZE(counters); i++) {
663 fd = openat(stats_dir, counters[i], O_RDONLY);
668 len = read(fd, buf, sizeof(buf));
676 blobmsg_add_u32(b, counters[i], strtoul(buf, NULL, 0));
684 static int system_addr(struct device *dev, struct device_addr *addr, int cmd)
686 int alen = ((addr->flags & DEVADDR_FAMILY) == DEVADDR_INET4) ? 4 : 16;
687 struct ifaddrmsg ifa = {
688 .ifa_family = (alen == 4) ? AF_INET : AF_INET6,
689 .ifa_prefixlen = addr->mask,
690 .ifa_index = dev->ifindex,
700 ifa.ifa_index = dev->ifindex;
703 msg = nlmsg_alloc_simple(cmd, 0);
707 nlmsg_append(msg, &ifa, sizeof(ifa), 0);
708 nla_put(msg, IFA_LOCAL, alen, &addr->addr);
709 return system_rtnl_call(msg);
712 int system_add_address(struct device *dev, struct device_addr *addr)
714 return system_addr(dev, addr, RTM_NEWADDR);
717 int system_del_address(struct device *dev, struct device_addr *addr)
719 return system_addr(dev, addr, RTM_DELADDR);
722 static int system_rt(struct device *dev, struct device_route *route, int cmd)
724 int alen = ((route->flags & DEVADDR_FAMILY) == DEVADDR_INET4) ? 4 : 16;
726 unsigned int flags = 0;
727 int ifindex = dev->ifindex;
730 have_gw = !!route->nexthop.in.s_addr;
732 have_gw = route->nexthop.in6.s6_addr32[0] ||
733 route->nexthop.in6.s6_addr32[1] ||
734 route->nexthop.in6.s6_addr32[2] ||
735 route->nexthop.in6.s6_addr32[3];
737 unsigned char scope = (cmd == RTM_DELROUTE) ? RT_SCOPE_NOWHERE :
738 (have_gw) ? RT_SCOPE_UNIVERSE : RT_SCOPE_LINK;
741 .rtm_family = (alen == 4) ? AF_INET : AF_INET6,
742 .rtm_dst_len = route->mask,
743 .rtm_table = RT_TABLE_MAIN,
744 .rtm_protocol = RTPROT_BOOT,
746 .rtm_type = (cmd == RTM_DELROUTE) ? 0: RTN_UNICAST,
750 if (cmd == RTM_NEWROUTE)
751 flags |= NLM_F_CREATE | NLM_F_REPLACE;
758 ifindex = dev->ifindex;
761 msg = nlmsg_alloc_simple(cmd, flags);
765 nlmsg_append(msg, &rtm, sizeof(rtm), 0);
768 nla_put(msg, RTA_DST, alen, &route->addr);
771 nla_put(msg, RTA_GATEWAY, alen, &route->nexthop);
773 if (route->flags & DEVADDR_DEVICE)
774 nla_put_u32(msg, RTA_OIF, ifindex);
776 return system_rtnl_call(msg);
779 int system_add_route(struct device *dev, struct device_route *route)
781 return system_rt(dev, route, RTM_NEWROUTE);
784 int system_del_route(struct device *dev, struct device_route *route)
786 return system_rt(dev, route, RTM_DELROUTE);
789 int system_flush_routes(void)
791 const char *names[] = {
792 "/proc/sys/net/ipv4/route/flush",
793 "/proc/sys/net/ipv6/route/flush"
797 for (i = 0; i < ARRAY_SIZE(names); i++) {
798 fd = open(names[i], O_WRONLY);
808 time_t system_get_rtime(void)
813 if (syscall(__NR_clock_gettime, CLOCK_MONOTONIC, &ts) == 0)
816 if (gettimeofday(&tv, NULL) == 0)