remove some more code duplication
[project/netifd.git] / system-linux.c
index 9f1aa9e..0308e6f 100644 (file)
@@ -1,6 +1,9 @@
+#define _GNU_SOURCE
+
 #include <sys/socket.h>
 #include <sys/ioctl.h>
 #include <sys/stat.h>
+#include <sys/syscall.h>
 
 #include <linux/rtnetlink.h>
 #include <linux/sockios.h>
 #include "device.h"
 #include "system.h"
 
+struct event_socket {
+       struct uloop_fd uloop;
+       struct nl_sock *sock;
+       struct nl_cb *cb;
+};
+
 static int sock_ioctl = -1;
 static struct nl_sock *sock_rtnl = NULL;
-static struct nl_sock *sock_rtnl_event = NULL;
 
-static void handler_rtnl_event(struct uloop_fd *u, unsigned int events);
 static int cb_rtnl_event(struct nl_msg *msg, void *arg);
-static struct uloop_fd rtnl_event = {.cb = handler_rtnl_event};
-static struct nl_cb *nl_cb_rtnl_event;
+
+static void
+handler_nl_event(struct uloop_fd *u, unsigned int events)
+{
+       struct event_socket *ev = container_of(u, struct event_socket, uloop);
+       nl_recvmsgs(ev->sock, ev->cb);
+}
+
+static struct nl_sock *
+create_socket(int protocol)
+{
+       struct nl_sock *sock;
+
+       sock = nl_socket_alloc();
+       if (!sock)
+               return NULL;
+
+       if (nl_connect(sock, protocol))
+               return NULL;
+
+       return sock;
+}
+
+static bool
+create_event_socket(struct event_socket *ev, int protocol,
+                   int (*cb)(struct nl_msg *msg, void *arg))
+{
+       // Prepare socket for link events
+       ev->cb = nl_cb_alloc(NL_CB_DEFAULT);
+       if (!ev->cb)
+               return false;
+
+       nl_cb_set(ev->cb, NL_CB_VALID, NL_CB_CUSTOM, cb, NULL);
+
+       ev->sock = create_socket(protocol);
+       if (!ev->sock)
+               return false;
+
+       ev->uloop.fd = nl_socket_get_fd(ev->sock);
+       ev->uloop.cb = handler_nl_event;
+       uloop_fd_add(&ev->uloop, ULOOP_READ | ULOOP_EDGE_TRIGGER);
+       return true;
+}
 
 int system_init(void)
 {
+       static struct event_socket rtnl_event;
+
        sock_ioctl = socket(AF_LOCAL, SOCK_DGRAM, 0);
        fcntl(sock_ioctl, F_SETFD, fcntl(sock_ioctl, F_GETFD) | FD_CLOEXEC);
 
        // Prepare socket for routing / address control
-       sock_rtnl = nl_socket_alloc();
+       sock_rtnl = create_socket(NETLINK_ROUTE);
        if (!sock_rtnl)
                return -1;
 
-       if (nl_connect(sock_rtnl, NETLINK_ROUTE))
-               goto error_free_sock;
-
-       // Prepare socket for link events
-       nl_cb_rtnl_event = nl_cb_alloc(NL_CB_DEFAULT);
-       if (!nl_cb_rtnl_event)
-               goto error_free_sock;
+       if (!create_event_socket(&rtnl_event, NETLINK_ROUTE, cb_rtnl_event))
+               return -1;
 
-       nl_cb_set(nl_cb_rtnl_event, NL_CB_VALID, NL_CB_CUSTOM,
-                 cb_rtnl_event, NULL);
+       // Receive network link events form kernel
+       nl_socket_add_membership(rtnl_event.sock, RTNLGRP_LINK);
 
-       sock_rtnl_event = nl_socket_alloc();
-       if (!sock_rtnl_event)
-               goto error_free_cb;
+       return 0;
+}
 
-       if (nl_connect(sock_rtnl_event, NETLINK_ROUTE))
-               goto error_free_event;
+static void system_set_sysctl(const char *path, const char *val)
+{
+       int fd;
 
-       // Receive network link events form kernel
-       nl_socket_add_membership(sock_rtnl_event, RTNLGRP_LINK);
+       fd = open(path, O_WRONLY);
+       if (fd < 0)
+               return;
 
-       rtnl_event.fd = nl_socket_get_fd(sock_rtnl_event);
-       uloop_fd_add(&rtnl_event, ULOOP_READ | ULOOP_EDGE_TRIGGER);
+       write(fd, val, strlen(val));
+       close(fd);
+}
 
-       return 0;
+static void system_set_dev_sysctl(const char *path, const char *device, const char *val)
+{
+       static char buf[256];
 
-error_free_event:
-       nl_socket_free(sock_rtnl_event);
-       sock_rtnl_event = NULL;
-error_free_cb:
-       nl_cb_put(nl_cb_rtnl_event);
-       nl_cb_rtnl_event = NULL;
-error_free_sock:
-       nl_socket_free(sock_rtnl);
-       sock_rtnl = NULL;
-       return -1;
+       snprintf(buf, sizeof(buf), path, val);
+       system_set_sysctl(buf, val);
 }
 
-// If socket is ready for reading parse netlink events
-static void handler_rtnl_event(struct uloop_fd *u, unsigned int events)
+static void system_set_disable_ipv6(struct device *dev, const char *val)
 {
-       nl_recvmsgs(sock_rtnl_event, nl_cb_rtnl_event);
+       system_set_dev_sysctl("/proc/sys/net/ipv6/conf/%s/disable_ipv6", dev->ifname, val);
 }
 
 // Evaluate netlink messages
@@ -135,11 +175,13 @@ static int system_bridge_if(const char *bridge, struct device *dev, int cmd, voi
 
 int system_bridge_addif(struct device *bridge, struct device *dev)
 {
+       system_set_disable_ipv6(dev, "1");
        return system_bridge_if(bridge->ifname, dev, SIOCBRADDIF, NULL);
 }
 
 int system_bridge_delif(struct device *bridge, struct device *dev)
 {
+       system_set_disable_ipv6(dev, "0");
        return system_bridge_if(bridge->ifname, dev, SIOCBRDELIF, NULL);
 }
 
@@ -199,14 +241,150 @@ static int system_if_flags(const char *ifname, unsigned add, unsigned rem)
        return ioctl(sock_ioctl, SIOCSIFFLAGS, &ifr);
 }
 
+struct clear_data {
+       struct nl_msg *msg;
+       struct device *dev;
+       int type;
+       int size;
+       int af;
+};
+
+
+static bool check_ifaddr(struct nlmsghdr *hdr, int ifindex)
+{
+       struct ifaddrmsg *ifa = NLMSG_DATA(hdr);
+
+       return ifa->ifa_index == ifindex;
+}
+
+static bool check_route(struct nlmsghdr *hdr, int ifindex)
+{
+       struct nlattr *tb[__RTA_MAX];
+
+       nlmsg_parse(hdr, sizeof(struct rtmsg), tb, __RTA_MAX - 1, NULL);
+       if (!tb[RTA_OIF])
+               return false;
+
+       return *(int *)RTA_DATA(tb[RTA_OIF]) == ifindex;
+}
+
+static int cb_clear_event(struct nl_msg *msg, void *arg)
+{
+       struct clear_data *clr = arg;
+       struct nlmsghdr *hdr = nlmsg_hdr(msg);
+       bool (*cb)(struct nlmsghdr *, int ifindex);
+       int type;
+
+       switch(clr->type) {
+       case RTM_GETADDR:
+               type = RTM_DELADDR;
+               if (hdr->nlmsg_type != RTM_NEWADDR)
+                       return NL_SKIP;
+
+               cb = check_ifaddr;
+               break;
+       case RTM_GETROUTE:
+               type = RTM_DELROUTE;
+               if (hdr->nlmsg_type != RTM_NEWROUTE)
+                       return NL_SKIP;
+
+               cb = check_route;
+               break;
+       default:
+               return NL_SKIP;
+       }
+
+       if (!cb(hdr, clr->dev->ifindex))
+               return NL_SKIP;
+
+       D(SYSTEM, "Remove %s from device %s\n",
+         type == RTM_DELADDR ? "an address" : "a route",
+         clr->dev->ifname);
+       memcpy(nlmsg_hdr(clr->msg), hdr, hdr->nlmsg_len);
+       hdr = nlmsg_hdr(clr->msg);
+       hdr->nlmsg_type = type;
+       hdr->nlmsg_flags = NLM_F_REQUEST;
+
+       if (!nl_send_auto_complete(sock_rtnl, clr->msg))
+               nl_wait_for_ack(sock_rtnl);
+
+       return NL_SKIP;
+}
+
+static int
+cb_finish_event(struct nl_msg *msg, void *arg)
+{
+       int *pending = arg;
+       *pending = 0;
+       return NL_STOP;
+}
+
+static int
+error_handler(struct sockaddr_nl *nla, struct nlmsgerr *err, void *arg)
+{
+       int *pending = arg;
+       *pending = err->error;
+       return NL_STOP;
+}
+
+static void
+system_if_clear_entries(struct device *dev, int type, int af)
+{
+       struct clear_data clr;
+       struct nl_cb *cb = nl_cb_alloc(NL_CB_DEFAULT);
+       struct rtmsg rtm = {
+               .rtm_family = af,
+               .rtm_flags = RTM_F_CLONED,
+       };
+       int flags = NLM_F_DUMP;
+       int pending = 1;
+
+       clr.af = af;
+       clr.dev = dev;
+       clr.type = type;
+       switch (type) {
+       case RTM_GETADDR:
+               clr.size = sizeof(struct rtgenmsg);
+               break;
+       case RTM_GETROUTE:
+               clr.size = sizeof(struct rtmsg);
+               break;
+       default:
+               return;
+       }
+
+       if (!cb)
+               return;
+
+       clr.msg = nlmsg_alloc_simple(type, flags);
+       if (!clr.msg)
+               goto out;
+
+       nlmsg_append(clr.msg, &rtm, clr.size, 0);
+       nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, cb_clear_event, &clr);
+       nl_cb_set(cb, NL_CB_FINISH, NL_CB_CUSTOM, cb_finish_event, &pending);
+       nl_cb_err(cb, NL_CB_CUSTOM, error_handler, &pending);
+
+       nl_send_auto_complete(sock_rtnl, clr.msg);
+       while (pending > 0)
+               nl_recvmsgs(sock_rtnl, cb);
+
+       nlmsg_free(clr.msg);
+out:
+       nl_cb_put(cb);
+}
+
 /*
  * Clear bridge (membership) state and bring down device
  */
 void system_if_clear_state(struct device *dev)
 {
-       char buf[256];
+       static char buf[256];
        char *bridge;
 
+       if (dev->external)
+               return;
+
        dev->ifindex = system_if_resolve(dev);
        if (!dev->ifindex)
                return;
@@ -224,6 +402,12 @@ void system_if_clear_state(struct device *dev)
                D(SYSTEM, "Remove device '%s' from bridge '%s'\n", dev->ifname, bridge);
                system_bridge_if(bridge, dev, SIOCBRDELIF, NULL);
        }
+
+       system_if_clear_entries(dev, RTM_GETROUTE, AF_INET);
+       system_if_clear_entries(dev, RTM_GETADDR, AF_INET);
+       system_if_clear_entries(dev, RTM_GETROUTE, AF_INET6);
+       system_if_clear_entries(dev, RTM_GETADDR, AF_INET6);
+       system_set_disable_ipv6(dev, "0");
 }
 
 static inline unsigned long
@@ -247,6 +431,9 @@ int system_bridge_addbr(struct device *bridge, struct bridge_config *cfg)
        args[1] = sec_to_jiffies(cfg->forward_delay);
        system_bridge_if(bridge->ifname, NULL, SIOCDEVPRIVATE, &args);
 
+       system_set_dev_sysctl("/sys/devices/virtual/net/%s/bridge/multicast_snooping",
+               bridge->ifname, cfg->igmp_snoop ? "1" : "0");
+
        if (cfg->flags & BRIDGE_OPT_AGEING_TIME) {
                args[0] = BRCTL_SET_AGEING_TIME;
                args[1] = sec_to_jiffies(cfg->ageing_time);
@@ -298,9 +485,32 @@ int system_vlan_del(struct device *dev)
        return system_vlan(dev, -1);
 }
 
-int system_if_up(struct device *dev)
+static void
+system_if_apply_settings(struct device *dev)
 {
+       struct ifreq ifr;
+
+       memset(&ifr, 0, sizeof(ifr));
+       strncpy(ifr.ifr_name, dev->ifname, sizeof(ifr.ifr_name));
+       if (dev->flags & DEV_OPT_MTU) {
+               ifr.ifr_mtu = dev->mtu;
+               ioctl(sock_ioctl, SIOCSIFMTU, &ifr);
+       }
+       if (dev->flags & DEV_OPT_TXQUEUELEN) {
+               ifr.ifr_qlen = dev->txqueuelen;
+               ioctl(sock_ioctl, SIOCSIFTXQLEN, &ifr);
+       }
+       if (dev->flags & DEV_OPT_MACADDR) {
+               memcpy(&ifr.ifr_hwaddr, dev->macaddr, sizeof(dev->macaddr));
+               ioctl(sock_ioctl, SIOCSIFHWADDR, &ifr);
+       }
+
        dev->ifindex = system_if_resolve(dev);
+}
+
+int system_if_up(struct device *dev)
+{
+       system_if_apply_settings(dev);
        return system_if_flags(dev->ifname, IFF_UP, 0);
 }
 
@@ -315,6 +525,49 @@ int system_if_check(struct device *dev)
        return 0;
 }
 
+int system_if_dump_stats(struct device *dev, struct blob_buf *b)
+{
+       const char *const counters[] = {
+               "collisions",     "rx_frame_errors",   "tx_compressed",
+               "multicast",      "rx_length_errors",  "tx_dropped",
+               "rx_bytes",       "rx_missed_errors",  "tx_errors",
+               "rx_compressed",  "rx_over_errors",    "tx_fifo_errors",
+               "rx_crc_errors",  "rx_packets",        "tx_heartbeat_errors",
+               "rx_dropped",     "tx_aborted_errors", "tx_packets",
+               "rx_errors",      "tx_bytes",          "tx_window_errors",
+               "rx_fifo_errors", "tx_carrier_errors",
+       };
+       char buf[64];
+       int stats_dir;
+       int i, fd, len;
+
+       snprintf(buf, sizeof(buf), "/sys/class/net/%s/statistics", dev->ifname);
+       stats_dir = open(buf, O_DIRECTORY);
+       if (stats_dir < 0)
+               return -1;
+
+       for (i = 0; i < ARRAY_SIZE(counters); i++) {
+               fd = openat(stats_dir, counters[i], O_RDONLY);
+               if (fd < 0)
+                       continue;
+
+retry:
+               len = read(fd, buf, sizeof(buf));
+               if (len < 0) {
+                       if (errno == EINTR)
+                               goto retry;
+                       continue;
+               }
+
+               buf[len] = 0;
+               blobmsg_add_u32(b, counters[i], strtoul(buf, NULL, 0));
+               close(fd);
+       }
+
+       close(stats_dir);
+       return 0;
+}
+
 static int system_addr(struct device *dev, struct device_addr *addr, int cmd)
 {
        int alen = ((addr->flags & DEVADDR_FAMILY) == DEVADDR_INET4) ? 4 : 16;
@@ -324,7 +577,17 @@ static int system_addr(struct device *dev, struct device_addr *addr, int cmd)
                .ifa_index = dev->ifindex,
        };
 
-       struct nl_msg *msg = nlmsg_alloc_simple(cmd, 0);
+       struct nl_msg *msg;
+
+       dev = addr->device;
+       if (dev) {
+               if (!dev->ifindex)
+                       return -1;
+
+               ifa.ifa_index = dev->ifindex;
+       }
+
+       msg = nlmsg_alloc_simple(cmd, 0);
        if (!msg)
                return -1;
 
@@ -347,6 +610,8 @@ static int system_rt(struct device *dev, struct device_route *route, int cmd)
 {
        int alen = ((route->flags & DEVADDR_FAMILY) == DEVADDR_INET4) ? 4 : 16;
        bool have_gw;
+       unsigned int flags = 0;
+       int ifindex = dev->ifindex;
 
        if (alen == 4)
                have_gw = !!route->nexthop.in.s_addr;
@@ -367,8 +632,20 @@ static int system_rt(struct device *dev, struct device_route *route, int cmd)
                .rtm_scope = scope,
                .rtm_type = (cmd == RTM_DELROUTE) ? 0: RTN_UNICAST,
        };
+       struct nl_msg *msg;
+
+       if (cmd == RTM_NEWROUTE)
+               flags |= NLM_F_CREATE | NLM_F_REPLACE;
+
+       dev = route->device;
+       if (dev) {
+               if (!dev->ifindex)
+                       return -1;
+
+               ifindex = dev->ifindex;
+       }
 
-       struct nl_msg *msg = nlmsg_alloc_simple(cmd, 0);
+       msg = nlmsg_alloc_simple(cmd, flags);
        if (!msg)
                return -1;
 
@@ -381,7 +658,7 @@ static int system_rt(struct device *dev, struct device_route *route, int cmd)
                nla_put(msg, RTA_GATEWAY, alen, &route->nexthop);
 
        if (route->flags & DEVADDR_DEVICE)
-               nla_put_u32(msg, RTA_OIF, dev->ifindex);
+               nla_put_u32(msg, RTA_OIF, ifindex);
 
        return system_rtnl_call(msg);
 }
@@ -395,3 +672,36 @@ int system_del_route(struct device *dev, struct device_route *route)
 {
        return system_rt(dev, route, RTM_DELROUTE);
 }
+
+int system_flush_routes(void)
+{
+       const char *names[] = {
+               "/proc/sys/net/ipv4/route/flush",
+               "/proc/sys/net/ipv6/route/flush"
+       };
+       int fd, i;
+
+       for (i = 0; i < ARRAY_SIZE(names); i++) {
+               fd = open(names[i], O_WRONLY);
+               if (fd < 0)
+                       continue;
+
+               write(fd, "-1", 2);
+               close(fd);
+       }
+       return 0;
+}
+
+time_t system_get_rtime(void)
+{
+       struct timespec ts;
+       struct timeval tv;
+
+       if (syscall(__NR_clock_gettime, CLOCK_MONOTONIC, &ts) == 0)
+               return ts.tv_sec;
+
+       if (gettimeofday(&tv, NULL) == 0)
+               return tv.tv_sec;
+
+       return 0;
+}