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>
27 static int sock_ioctl = -1;
28 static struct nl_sock *sock_rtnl = NULL;
29 static struct nl_sock *sock_rtnl_event = NULL;
31 static void handler_rtnl_event(struct uloop_fd *u, unsigned int events);
32 static int cb_rtnl_event(struct nl_msg *msg, void *arg);
33 static struct uloop_fd rtnl_event = {.cb = handler_rtnl_event};
34 static struct nl_cb *nl_cb_rtnl_event;
38 sock_ioctl = socket(AF_LOCAL, SOCK_DGRAM, 0);
39 fcntl(sock_ioctl, F_SETFD, fcntl(sock_ioctl, F_GETFD) | FD_CLOEXEC);
41 // Prepare socket for routing / address control
42 sock_rtnl = nl_socket_alloc();
46 if (nl_connect(sock_rtnl, NETLINK_ROUTE))
49 // Prepare socket for link events
50 nl_cb_rtnl_event = nl_cb_alloc(NL_CB_DEFAULT);
51 if (!nl_cb_rtnl_event)
54 nl_cb_set(nl_cb_rtnl_event, NL_CB_VALID, NL_CB_CUSTOM,
57 sock_rtnl_event = nl_socket_alloc();
61 if (nl_connect(sock_rtnl_event, NETLINK_ROUTE))
62 goto error_free_event;
64 // Receive network link events form kernel
65 nl_socket_add_membership(sock_rtnl_event, RTNLGRP_LINK);
67 rtnl_event.fd = nl_socket_get_fd(sock_rtnl_event);
68 uloop_fd_add(&rtnl_event, ULOOP_READ | ULOOP_EDGE_TRIGGER);
73 nl_socket_free(sock_rtnl_event);
74 sock_rtnl_event = NULL;
76 nl_cb_put(nl_cb_rtnl_event);
77 nl_cb_rtnl_event = NULL;
79 nl_socket_free(sock_rtnl);
84 // If socket is ready for reading parse netlink events
85 static void handler_rtnl_event(struct uloop_fd *u, unsigned int events)
87 nl_recvmsgs(sock_rtnl_event, nl_cb_rtnl_event);
90 // Evaluate netlink messages
91 static int cb_rtnl_event(struct nl_msg *msg, void *arg)
93 struct nlmsghdr *nh = nlmsg_hdr(msg);
94 struct ifinfomsg *ifi = NLMSG_DATA(nh);
95 struct nlattr *nla[__IFLA_MAX];
97 if (nh->nlmsg_type != RTM_DELLINK && nh->nlmsg_type != RTM_NEWLINK)
100 nlmsg_parse(nh, sizeof(*ifi), nla, __IFLA_MAX - 1, NULL);
101 if (!nla[IFLA_IFNAME])
104 struct device *dev = device_get(RTA_DATA(nla[IFLA_IFNAME]), false);
108 dev->ifindex = ifi->ifi_index;
109 device_set_present(dev, (nh->nlmsg_type == RTM_NEWLINK));
115 static int system_rtnl_call(struct nl_msg *msg)
117 int s = -(nl_send_auto_complete(sock_rtnl, msg)
118 || nl_wait_for_ack(sock_rtnl));
123 int system_bridge_delbr(struct device *bridge)
125 return ioctl(sock_ioctl, SIOCBRDELBR, bridge->ifname);
128 static int system_bridge_if(const char *bridge, struct device *dev, int cmd, void *data)
132 ifr.ifr_ifindex = dev->ifindex;
135 strncpy(ifr.ifr_name, bridge, sizeof(ifr.ifr_name));
136 return ioctl(sock_ioctl, cmd, &ifr);
139 int system_bridge_addif(struct device *bridge, struct device *dev)
141 return system_bridge_if(bridge->ifname, dev, SIOCBRADDIF, NULL);
144 int system_bridge_delif(struct device *bridge, struct device *dev)
146 return system_bridge_if(bridge->ifname, dev, SIOCBRDELIF, NULL);
149 static bool system_is_bridge(const char *name, char *buf, int buflen)
153 snprintf(buf, buflen, "/sys/devices/virtual/net/%s/bridge", name);
154 if (stat(buf, &st) < 0)
160 static char *system_get_bridge(const char *name, char *buf, int buflen)
166 snprintf(buf, buflen, "/sys/devices/virtual/net/*/brif/%s/bridge", name);
167 if (glob(buf, GLOB_NOSORT, NULL, &gl) < 0)
170 if (gl.gl_pathc == 0)
173 len = readlink(gl.gl_pathv[0], buf, buflen);
178 path = strrchr(buf, '/');
185 static int system_if_resolve(struct device *dev)
188 strncpy(ifr.ifr_name, dev->ifname, sizeof(ifr.ifr_name));
189 if (!ioctl(sock_ioctl, SIOCGIFINDEX, &ifr))
190 return ifr.ifr_ifindex;
195 static int system_if_flags(const char *ifname, unsigned add, unsigned rem)
198 strncpy(ifr.ifr_name, ifname, sizeof(ifr.ifr_name));
199 ioctl(sock_ioctl, SIOCGIFFLAGS, &ifr);
200 ifr.ifr_flags |= add;
201 ifr.ifr_flags &= ~rem;
202 return ioctl(sock_ioctl, SIOCSIFFLAGS, &ifr);
214 static bool check_ifaddr(struct nlmsghdr *hdr, int ifindex)
216 struct ifaddrmsg *ifa = NLMSG_DATA(hdr);
218 return ifa->ifa_index == ifindex;
221 static bool check_route(struct nlmsghdr *hdr, int ifindex)
223 struct nlattr *tb[__RTA_MAX];
225 nlmsg_parse(hdr, sizeof(struct rtmsg), tb, __RTA_MAX - 1, NULL);
229 return *(int *)RTA_DATA(tb[RTA_OIF]) == ifindex;
232 static int cb_clear_event(struct nl_msg *msg, void *arg)
234 struct clear_data *clr = arg;
235 struct nlmsghdr *hdr = nlmsg_hdr(msg);
236 bool (*cb)(struct nlmsghdr *, int ifindex);
242 if (hdr->nlmsg_type != RTM_NEWADDR)
249 if (hdr->nlmsg_type != RTM_NEWROUTE)
258 if (!cb(hdr, clr->dev->ifindex))
261 D(SYSTEM, "Remove %s from device %s\n",
262 type == RTM_DELADDR ? "an address" : "a route",
264 memcpy(nlmsg_hdr(clr->msg), hdr, hdr->nlmsg_len);
265 hdr = nlmsg_hdr(clr->msg);
266 hdr->nlmsg_type = type;
267 hdr->nlmsg_flags = NLM_F_REQUEST;
269 if (!nl_send_auto_complete(sock_rtnl, clr->msg))
270 nl_wait_for_ack(sock_rtnl);
276 cb_finish_event(struct nl_msg *msg, void *arg)
284 error_handler(struct sockaddr_nl *nla, struct nlmsgerr *err, void *arg)
287 *pending = err->error;
292 system_if_clear_entries(struct device *dev, int type, int af)
294 struct clear_data clr;
295 struct nl_cb *cb = nl_cb_alloc(NL_CB_DEFAULT);
298 .rtm_flags = RTM_F_CLONED,
300 int flags = NLM_F_DUMP;
308 clr.size = sizeof(struct rtgenmsg);
311 clr.size = sizeof(struct rtmsg);
320 clr.msg = nlmsg_alloc_simple(type, flags);
324 nlmsg_append(clr.msg, &rtm, clr.size, 0);
325 nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, cb_clear_event, &clr);
326 nl_cb_set(cb, NL_CB_FINISH, NL_CB_CUSTOM, cb_finish_event, &pending);
327 nl_cb_err(cb, NL_CB_CUSTOM, error_handler, &pending);
329 nl_send_auto_complete(sock_rtnl, clr.msg);
331 nl_recvmsgs(sock_rtnl, cb);
339 * Clear bridge (membership) state and bring down device
341 void system_if_clear_state(struct device *dev)
346 dev->ifindex = system_if_resolve(dev);
350 system_if_flags(dev->ifname, 0, IFF_UP);
352 if (system_is_bridge(dev->ifname, buf, sizeof(buf))) {
353 D(SYSTEM, "Delete existing bridge named '%s'\n", dev->ifname);
354 system_bridge_delbr(dev);
358 bridge = system_get_bridge(dev->ifname, buf, sizeof(buf));
360 D(SYSTEM, "Remove device '%s' from bridge '%s'\n", dev->ifname, bridge);
361 system_bridge_if(bridge, dev, SIOCBRDELIF, NULL);
364 system_if_clear_entries(dev, RTM_GETROUTE, AF_INET);
365 system_if_clear_entries(dev, RTM_GETADDR, AF_INET);
366 system_if_clear_entries(dev, RTM_GETROUTE, AF_INET6);
367 system_if_clear_entries(dev, RTM_GETADDR, AF_INET6);
370 static inline unsigned long
371 sec_to_jiffies(int val)
373 return (unsigned long) val * 100;
376 int system_bridge_addbr(struct device *bridge, struct bridge_config *cfg)
378 unsigned long args[4] = {};
380 if (ioctl(sock_ioctl, SIOCBRADDBR, bridge->ifname) < 0)
383 args[0] = BRCTL_SET_BRIDGE_STP_STATE;
384 args[1] = !!cfg->stp;
385 system_bridge_if(bridge->ifname, NULL, SIOCDEVPRIVATE, &args);
387 args[0] = BRCTL_SET_BRIDGE_FORWARD_DELAY;
388 args[1] = sec_to_jiffies(cfg->forward_delay);
389 system_bridge_if(bridge->ifname, NULL, SIOCDEVPRIVATE, &args);
391 if (cfg->flags & BRIDGE_OPT_AGEING_TIME) {
392 args[0] = BRCTL_SET_AGEING_TIME;
393 args[1] = sec_to_jiffies(cfg->ageing_time);
394 system_bridge_if(bridge->ifname, NULL, SIOCDEVPRIVATE, &args);
397 if (cfg->flags & BRIDGE_OPT_HELLO_TIME) {
398 args[0] = BRCTL_SET_BRIDGE_HELLO_TIME;
399 args[1] = sec_to_jiffies(cfg->hello_time);
400 system_bridge_if(bridge->ifname, NULL, SIOCDEVPRIVATE, &args);
403 if (cfg->flags & BRIDGE_OPT_MAX_AGE) {
404 args[0] = BRCTL_SET_BRIDGE_MAX_AGE;
405 args[1] = sec_to_jiffies(cfg->max_age);
406 system_bridge_if(bridge->ifname, NULL, SIOCDEVPRIVATE, &args);
412 static int system_vlan(struct device *dev, int id)
414 struct vlan_ioctl_args ifr = {
415 .cmd = SET_VLAN_NAME_TYPE_CMD,
416 .u.name_type = VLAN_NAME_TYPE_RAW_PLUS_VID_NO_PAD,
419 ioctl(sock_ioctl, SIOCSIFVLAN, &ifr);
422 ifr.cmd = DEL_VLAN_CMD;
425 ifr.cmd = ADD_VLAN_CMD;
428 strncpy(ifr.device1, dev->ifname, sizeof(ifr.device1));
429 return ioctl(sock_ioctl, SIOCSIFVLAN, &ifr);
432 int system_vlan_add(struct device *dev, int id)
434 return system_vlan(dev, id);
437 int system_vlan_del(struct device *dev)
439 return system_vlan(dev, -1);
442 int system_if_up(struct device *dev)
444 dev->ifindex = system_if_resolve(dev);
445 return system_if_flags(dev->ifname, IFF_UP, 0);
448 int system_if_down(struct device *dev)
450 return system_if_flags(dev->ifname, 0, IFF_UP);
453 int system_if_check(struct device *dev)
455 device_set_present(dev, (system_if_resolve(dev) >= 0));
459 int system_if_dump_stats(struct device *dev, struct blob_buf *b)
461 const char *const counters[] = {
462 "collisions", "rx_frame_errors", "tx_compressed",
463 "multicast", "rx_length_errors", "tx_dropped",
464 "rx_bytes", "rx_missed_errors", "tx_errors",
465 "rx_compressed", "rx_over_errors", "tx_fifo_errors",
466 "rx_crc_errors", "rx_packets", "tx_heartbeat_errors",
467 "rx_dropped", "tx_aborted_errors", "tx_packets",
468 "rx_errors", "tx_bytes", "tx_window_errors",
469 "rx_fifo_errors", "tx_carrier_errors",
475 snprintf(buf, sizeof(buf), "/sys/class/net/%s/statistics", dev->ifname);
476 stats_dir = open(buf, O_DIRECTORY);
480 for (i = 0; i < ARRAY_SIZE(counters); i++) {
481 fd = openat(stats_dir, counters[i], O_RDONLY);
486 len = read(fd, buf, sizeof(buf));
494 blobmsg_add_u32(b, counters[i], strtoul(buf, NULL, 0));
502 static int system_addr(struct device *dev, struct device_addr *addr, int cmd)
504 int alen = ((addr->flags & DEVADDR_FAMILY) == DEVADDR_INET4) ? 4 : 16;
505 struct ifaddrmsg ifa = {
506 .ifa_family = (alen == 4) ? AF_INET : AF_INET6,
507 .ifa_prefixlen = addr->mask,
508 .ifa_index = dev->ifindex,
511 struct nl_msg *msg = nlmsg_alloc_simple(cmd, 0);
515 nlmsg_append(msg, &ifa, sizeof(ifa), 0);
516 nla_put(msg, IFA_LOCAL, alen, &addr->addr);
517 return system_rtnl_call(msg);
520 int system_add_address(struct device *dev, struct device_addr *addr)
522 return system_addr(dev, addr, RTM_NEWADDR);
525 int system_del_address(struct device *dev, struct device_addr *addr)
527 return system_addr(dev, addr, RTM_DELADDR);
530 static int system_rt(struct device *dev, struct device_route *route, int cmd)
532 int alen = ((route->flags & DEVADDR_FAMILY) == DEVADDR_INET4) ? 4 : 16;
534 unsigned int flags = 0;
537 have_gw = !!route->nexthop.in.s_addr;
539 have_gw = route->nexthop.in6.s6_addr32[0] ||
540 route->nexthop.in6.s6_addr32[1] ||
541 route->nexthop.in6.s6_addr32[2] ||
542 route->nexthop.in6.s6_addr32[3];
544 unsigned char scope = (cmd == RTM_DELROUTE) ? RT_SCOPE_NOWHERE :
545 (have_gw) ? RT_SCOPE_UNIVERSE : RT_SCOPE_LINK;
548 .rtm_family = (alen == 4) ? AF_INET : AF_INET6,
549 .rtm_dst_len = route->mask,
550 .rtm_table = RT_TABLE_MAIN,
551 .rtm_protocol = RTPROT_BOOT,
553 .rtm_type = (cmd == RTM_DELROUTE) ? 0: RTN_UNICAST,
556 if (cmd == RTM_NEWROUTE)
557 flags |= NLM_F_CREATE | NLM_F_REPLACE;
559 struct nl_msg *msg = nlmsg_alloc_simple(cmd, flags);
563 nlmsg_append(msg, &rtm, sizeof(rtm), 0);
566 nla_put(msg, RTA_DST, alen, &route->addr);
569 nla_put(msg, RTA_GATEWAY, alen, &route->nexthop);
571 if (route->flags & DEVADDR_DEVICE)
572 nla_put_u32(msg, RTA_OIF, dev->ifindex);
574 return system_rtnl_call(msg);
577 int system_add_route(struct device *dev, struct device_route *route)
579 return system_rt(dev, route, RTM_NEWROUTE);
582 int system_del_route(struct device *dev, struct device_route *route)
584 return system_rt(dev, route, RTM_DELROUTE);
587 int system_flush_routes(void)
589 const char *names[] = {
590 "/proc/sys/net/ipv4/route/flush",
591 "/proc/sys/net/ipv6/route/flush"
595 for (i = 0; i < ARRAY_SIZE(names); i++) {
596 fd = open(names[i], O_WRONLY);
606 time_t system_get_rtime(void)
611 if (syscall(__NR_clock_gettime, CLOCK_MONOTONIC, &ts) == 0)
614 if (gettimeofday(&tv, NULL) == 0)