1 #include <sys/socket.h>
5 #include <linux/rtnetlink.h>
6 #include <linux/sockios.h>
7 #include <linux/if_vlan.h>
14 #include <netlink/msg.h>
15 #include <netlink/attr.h>
16 #include <netlink/socket.h>
17 #include <libubox/uloop.h>
23 static int sock_ioctl = -1;
24 static struct nl_sock *sock_rtnl = NULL;
25 static struct nl_sock *sock_rtnl_event = NULL;
27 static void handler_rtnl_event(struct uloop_fd *u, unsigned int events);
28 static int cb_rtnl_event(struct nl_msg *msg, void *arg);
29 static struct uloop_fd rtnl_event = {.cb = handler_rtnl_event};
30 static struct nl_cb *nl_cb_rtnl_event;
34 sock_ioctl = socket(AF_LOCAL, SOCK_DGRAM, 0);
35 fcntl(sock_ioctl, F_SETFD, fcntl(sock_ioctl, F_GETFD) | FD_CLOEXEC);
37 // Prepare socket for routing / address control
38 sock_rtnl = nl_socket_alloc();
42 if (nl_connect(sock_rtnl, NETLINK_ROUTE))
45 // Prepare socket for link events
46 nl_cb_rtnl_event = nl_cb_alloc(NL_CB_DEFAULT);
47 if (!nl_cb_rtnl_event)
50 nl_cb_set(nl_cb_rtnl_event, NL_CB_VALID, NL_CB_CUSTOM,
53 sock_rtnl_event = nl_socket_alloc();
57 if (nl_connect(sock_rtnl_event, NETLINK_ROUTE))
58 goto error_free_event;
60 // Receive network link events form kernel
61 nl_socket_add_membership(sock_rtnl_event, RTNLGRP_LINK);
63 rtnl_event.fd = nl_socket_get_fd(sock_rtnl_event);
64 uloop_fd_add(&rtnl_event, ULOOP_READ | ULOOP_EDGE_TRIGGER);
69 nl_socket_free(sock_rtnl_event);
70 sock_rtnl_event = NULL;
72 nl_cb_put(nl_cb_rtnl_event);
73 nl_cb_rtnl_event = NULL;
75 nl_socket_free(sock_rtnl);
80 // If socket is ready for reading parse netlink events
81 static void handler_rtnl_event(struct uloop_fd *u, unsigned int events)
83 nl_recvmsgs(sock_rtnl_event, nl_cb_rtnl_event);
86 // Evaluate netlink messages
87 static int cb_rtnl_event(struct nl_msg *msg, void *arg)
89 struct nlmsghdr *nh = nlmsg_hdr(msg);
90 struct ifinfomsg *ifi = NLMSG_DATA(nh);
91 struct nlattr *nla[__IFLA_MAX];
93 if (nh->nlmsg_type != RTM_DELLINK && nh->nlmsg_type != RTM_NEWLINK)
96 nlmsg_parse(nh, sizeof(*ifi), nla, __IFLA_MAX - 1, NULL);
97 if (!nla[IFLA_IFNAME])
100 struct device *dev = device_get(RTA_DATA(nla[IFLA_IFNAME]), false);
104 dev->ifindex = ifi->ifi_index;
105 device_set_present(dev, (nh->nlmsg_type == RTM_NEWLINK));
111 static int system_rtnl_call(struct nl_msg *msg)
113 int s = -(nl_send_auto_complete(sock_rtnl, msg)
114 || nl_wait_for_ack(sock_rtnl));
119 int system_bridge_delbr(struct device *bridge)
121 return ioctl(sock_ioctl, SIOCBRDELBR, bridge->ifname);
124 static int system_bridge_if(const char *bridge, struct device *dev, int cmd)
127 ifr.ifr_ifindex = dev->ifindex;
128 strncpy(ifr.ifr_name, bridge, sizeof(ifr.ifr_name));
129 return ioctl(sock_ioctl, cmd, &ifr);
132 int system_bridge_addif(struct device *bridge, struct device *dev)
134 return system_bridge_if(bridge->ifname, dev, SIOCBRADDIF);
137 int system_bridge_delif(struct device *bridge, struct device *dev)
139 return system_bridge_if(bridge->ifname, dev, SIOCBRDELIF);
142 static bool system_is_bridge(const char *name, char *buf, int buflen)
146 snprintf(buf, buflen, "/sys/devices/virtual/net/%s/bridge", name);
147 if (stat(buf, &st) < 0)
153 static char *system_get_bridge(const char *name, char *buf, int buflen)
159 snprintf(buf, buflen, "/sys/devices/virtual/net/*/brif/%s/bridge", name);
160 if (glob(buf, GLOB_NOSORT, NULL, &gl) < 0)
163 if (gl.gl_pathc == 0)
166 len = readlink(gl.gl_pathv[0], buf, buflen);
171 path = strrchr(buf, '/');
178 static int system_if_resolve(struct device *dev)
181 strncpy(ifr.ifr_name, dev->ifname, sizeof(ifr.ifr_name));
182 if (!ioctl(sock_ioctl, SIOCGIFINDEX, &ifr))
183 return ifr.ifr_ifindex;
188 static int system_if_flags(const char *ifname, unsigned add, unsigned rem)
191 strncpy(ifr.ifr_name, ifname, sizeof(ifr.ifr_name));
192 ioctl(sock_ioctl, SIOCGIFFLAGS, &ifr);
193 ifr.ifr_flags |= add;
194 ifr.ifr_flags &= ~rem;
195 return ioctl(sock_ioctl, SIOCSIFFLAGS, &ifr);
199 * Clear bridge (membership) state and bring down device
201 static void system_if_clear_state(struct device *dev)
206 dev->ifindex = system_if_resolve(dev);
210 system_if_flags(dev->ifname, 0, IFF_UP);
212 if (system_is_bridge(dev->ifname, buf, sizeof(buf))) {
213 D(SYSTEM, "Delete existing bridge named '%s'\n", dev->ifname);
214 system_bridge_delbr(dev);
218 bridge = system_get_bridge(dev->ifname, buf, sizeof(buf));
220 D(SYSTEM, "Remove device '%s' from bridge '%s'\n", dev->ifname, bridge);
221 system_bridge_if(bridge, dev, SIOCBRDELIF);
225 int system_bridge_addbr(struct device *bridge)
227 system_if_clear_state(bridge);
228 return ioctl(sock_ioctl, SIOCBRADDBR, bridge->ifname);
231 static int system_vlan(struct device *dev, int id)
233 struct vlan_ioctl_args ifr = {
234 .cmd = SET_VLAN_NAME_TYPE_CMD,
235 .u.name_type = VLAN_NAME_TYPE_RAW_PLUS_VID_NO_PAD,
238 ioctl(sock_ioctl, SIOCSIFVLAN, &ifr);
241 ifr.cmd = DEL_VLAN_CMD;
244 ifr.cmd = ADD_VLAN_CMD;
247 strncpy(ifr.device1, dev->ifname, sizeof(ifr.device1));
248 return ioctl(sock_ioctl, SIOCSIFVLAN, &ifr);
251 int system_vlan_add(struct device *dev, int id)
253 system_if_clear_state(dev);
254 return system_vlan(dev, id);
257 int system_vlan_del(struct device *dev)
259 return system_vlan(dev, -1);
262 int system_if_up(struct device *dev)
264 dev->ifindex = system_if_resolve(dev);
265 return system_if_flags(dev->ifname, IFF_UP, 0);
268 int system_if_down(struct device *dev)
270 return system_if_flags(dev->ifname, 0, IFF_UP);
273 int system_if_check(struct device *dev)
275 system_if_clear_state(dev);
276 device_set_present(dev, (system_if_resolve(dev) >= 0));
280 static int system_addr(struct device *dev, struct device_addr *addr, int cmd)
282 int alen = ((addr->flags & DEVADDR_FAMILY) == DEVADDR_INET4) ? 4 : 16;
283 struct ifaddrmsg ifa = {
284 .ifa_family = (alen == 4) ? AF_INET : AF_INET6,
285 .ifa_prefixlen = addr->mask,
286 .ifa_index = dev->ifindex,
289 struct nl_msg *msg = nlmsg_alloc_simple(cmd, 0);
293 nlmsg_append(msg, &ifa, sizeof(ifa), 0);
294 nla_put(msg, IFA_LOCAL, alen, &addr->addr);
295 return system_rtnl_call(msg);
298 int system_add_address(struct device *dev, struct device_addr *addr)
300 return system_addr(dev, addr, RTM_NEWADDR);
303 int system_del_address(struct device *dev, struct device_addr *addr)
305 return system_addr(dev, addr, RTM_DELADDR);
308 static int system_rt(struct device *dev, struct device_route *route, int cmd)
310 int alen = ((route->flags & DEVADDR_FAMILY) == DEVADDR_INET4) ? 4 : 16;
314 have_gw = !!route->nexthop.in.s_addr;
316 have_gw = route->nexthop.in6.s6_addr32[0] ||
317 route->nexthop.in6.s6_addr32[1] ||
318 route->nexthop.in6.s6_addr32[2] ||
319 route->nexthop.in6.s6_addr32[3];
321 unsigned char scope = (cmd == RTM_DELROUTE) ? RT_SCOPE_NOWHERE :
322 (have_gw) ? RT_SCOPE_UNIVERSE : RT_SCOPE_LINK;
325 .rtm_family = (alen == 4) ? AF_INET : AF_INET6,
326 .rtm_dst_len = route->mask,
327 .rtm_table = RT_TABLE_MAIN,
328 .rtm_protocol = RTPROT_BOOT,
330 .rtm_type = (cmd == RTM_DELROUTE) ? 0: RTN_UNICAST,
333 struct nl_msg *msg = nlmsg_alloc_simple(cmd, 0);
337 nlmsg_append(msg, &rtm, sizeof(rtm), 0);
340 nla_put(msg, RTA_DST, alen, &route->addr);
343 nla_put(msg, RTA_GATEWAY, alen, &route->nexthop);
345 if (route->flags & DEVADDR_DEVICE)
346 nla_put_u32(msg, RTA_OIF, dev->ifindex);
348 return system_rtnl_call(msg);
351 int system_add_route(struct device *dev, struct device_route *route)
353 return system_rt(dev, route, RTM_NEWROUTE);
356 int system_del_route(struct device *dev, struct device_route *route)
358 return system_rt(dev, route, RTM_DELROUTE);