1 #include <sys/socket.h>
4 #include <linux/rtnetlink.h>
5 #include <linux/sockios.h>
6 #include <linux/if_vlan.h>
11 #include <netlink/msg.h>
12 #include <libubox/uloop.h>
18 static int sock_ioctl
= -1;
19 static struct nl_sock
*sock_rtnl
= NULL
;
20 static struct nl_sock
*sock_rtnl_event
= NULL
;
22 static void handler_rtnl_event(struct uloop_fd
*u
, unsigned int events
);
23 static int cb_rtnl_event(struct nl_msg
*msg
, void *arg
);
24 static struct uloop_fd rtnl_event
= {.cb
= handler_rtnl_event
};
28 sock_ioctl
= socket(AF_LOCAL
, SOCK_DGRAM
, 0);
29 fcntl(sock_ioctl
, F_SETFD
, fcntl(sock_ioctl
, F_GETFD
) | FD_CLOEXEC
);
31 // Prepare socket for routing / address control
32 if ((sock_rtnl
= nl_socket_alloc())) {
33 if (nl_connect(sock_rtnl
, NETLINK_ROUTE
)) {
34 nl_socket_free(sock_rtnl
);
39 // Prepare socket for link events
40 struct nl_cb
*cb
= nl_cb_alloc(NL_CB_DEFAULT
);
42 nl_cb_set(cb
, NL_CB_VALID
, NL_CB_CUSTOM
, cb_rtnl_event
, NULL
);
44 if (cb
&& (sock_rtnl_event
= nl_socket_alloc_cb(cb
))) {
45 if (nl_connect(sock_rtnl_event
, NETLINK_ROUTE
)) {
46 nl_socket_free(sock_rtnl_event
);
47 sock_rtnl_event
= NULL
;
49 // Receive network link events form kernel
50 nl_socket_add_membership(sock_rtnl_event
, RTNLGRP_LINK
);
52 // Synthesize initial link messages
53 struct nl_msg
*m
= nlmsg_alloc_simple(RTM_GETLINK
, NLM_F_DUMP
);
54 if (m
&& nlmsg_reserve(m
, sizeof(struct ifinfomsg
), 0)) {
55 nl_send_auto_complete(sock_rtnl_event
, m
);
60 rtnl_event
.fd
= nl_socket_get_fd(sock_rtnl_event
);
62 rtnl_event
.fd
= sock_rtnl_event
->s_fd
; // libnl-tiny hack...
64 uloop_fd_add(&rtnl_event
, ULOOP_READ
| ULOOP_EDGE_TRIGGER
);
67 return -(sock_ioctl
< 0 || !sock_rtnl
);
70 // If socket is ready for reading parse netlink events
71 static void handler_rtnl_event(struct uloop_fd
*u
, unsigned int events
)
73 nl_recvmsgs(sock_rtnl_event
, NULL
);
76 // Evaluate netlink messages
77 static int cb_rtnl_event(struct nl_msg
*msg
, void *arg
)
79 struct nlmsghdr
*nh
= nlmsg_hdr(msg
);
80 struct ifinfomsg
*ifi
= NLMSG_DATA(nh
);
81 struct nlattr
*nla
[__IFLA_MAX
];
83 if (nh
->nlmsg_type
!= RTM_DELLINK
&& nh
->nlmsg_type
!= RTM_NEWLINK
)
86 nlmsg_parse(nh
, sizeof(*ifi
), nla
, __IFLA_MAX
- 1, NULL
);
87 if (!nla
[IFLA_IFNAME
])
90 struct device
*dev
= device_get(RTA_DATA(nla
[IFLA_IFNAME
]), false);
94 dev
->ifindex
= ifi
->ifi_index
;
95 device_set_present(dev
, (nh
->nlmsg_type
== RTM_NEWLINK
));
101 static int system_rtnl_call(struct nl_msg
*msg
)
103 int s
= -(nl_send_auto_complete(sock_rtnl
, msg
)
104 || nl_wait_for_ack(sock_rtnl
));
109 int system_bridge_addbr(struct device
*bridge
)
111 return ioctl(sock_ioctl
, SIOCBRADDBR
, bridge
->ifname
);
114 int system_bridge_delbr(struct device
*bridge
)
116 return ioctl(sock_ioctl
, SIOCBRDELBR
, bridge
->ifname
);
119 static int system_bridge_if(struct device
*bridge
, struct device
*dev
, int cmd
)
122 ifr
.ifr_ifindex
= dev
->ifindex
;
123 strncpy(ifr
.ifr_name
, bridge
->ifname
, sizeof(ifr
.ifr_name
));
124 return ioctl(sock_ioctl
, cmd
, &ifr
);
127 int system_bridge_addif(struct device
*bridge
, struct device
*dev
)
129 return system_bridge_if(bridge
, dev
, SIOCBRADDIF
);
132 int system_bridge_delif(struct device
*bridge
, struct device
*dev
)
134 return system_bridge_if(bridge
, dev
, SIOCBRDELIF
);
137 static int system_vlan(struct device
*dev
, int id
)
139 struct vlan_ioctl_args ifr
= {
140 .cmd
= (id
== 0) ? DEL_VLAN_CMD
: ADD_VLAN_CMD
,
143 strncpy(ifr
.device1
, dev
->ifname
, sizeof(ifr
.device1
));
144 return ioctl(sock_ioctl
, SIOCSIFVLAN
, &ifr
);
147 int system_vlan_add(struct device
*dev
, int id
)
149 return system_vlan(dev
, id
);
152 int system_vlan_del(struct device
*dev
)
154 return system_vlan(dev
, 0);
157 static int system_if_flags(struct device
*dev
, unsigned add
, unsigned rem
)
160 strncpy(ifr
.ifr_name
, dev
->ifname
, sizeof(ifr
.ifr_name
));
161 ioctl(sock_ioctl
, SIOCGIFFLAGS
, &ifr
);
162 ifr
.ifr_flags
|= add
;
163 ifr
.ifr_flags
&= ~rem
;
164 return ioctl(sock_ioctl
, SIOCSIFFLAGS
, &ifr
);
167 int system_if_up(struct device
*dev
)
169 return system_if_flags(dev
, IFF_UP
, 0);
172 int system_if_down(struct device
*dev
)
174 return system_if_flags(dev
, 0, IFF_UP
);
177 int system_if_check(struct device
*dev
)
180 strncpy(ifr
.ifr_name
, dev
->ifname
, sizeof(ifr
.ifr_name
));
181 if (ioctl(sock_ioctl
, SIOCGIFINDEX
, &ifr
))
184 dev
->ifindex
= ifr
.ifr_ifindex
;
186 /* if (!strcmp(dev->ifname, "eth0"))
187 device_set_present(dev, true); */
191 static int system_addr(struct device
*dev
, struct device_addr
*addr
, int cmd
)
193 int alen
= ((addr
->flags
& DEVADDR_FAMILY
) == DEVADDR_INET4
) ? 4 : 16;
194 struct ifaddrmsg ifa
= {
195 .ifa_family
= (alen
== 4) ? AF_INET
: AF_INET6
,
196 .ifa_prefixlen
= addr
->mask
,
197 .ifa_index
= dev
->ifindex
,
200 struct nl_msg
*msg
= nlmsg_alloc_simple(cmd
, 0);
204 nlmsg_append(msg
, &ifa
, sizeof(ifa
), 0);
205 nla_put(msg
, IFA_ADDRESS
, alen
, &addr
->addr
);
206 return system_rtnl_call(msg
);
209 int system_add_address(struct device
*dev
, struct device_addr
*addr
)
211 return system_addr(dev
, addr
, RTM_NEWADDR
);
214 int system_del_address(struct device
*dev
, struct device_addr
*addr
)
216 return system_addr(dev
, addr
, RTM_DELADDR
);
219 static int system_rt(struct device
*dev
, struct device_route
*route
, int cmd
)
221 int alen
= ((route
->flags
& DEVADDR_FAMILY
) == DEVADDR_INET4
) ? 4 : 16;
225 have_gw
= !!route
->nexthop
.in
.s_addr
;
227 have_gw
= route
->nexthop
.in6
.s6_addr32
[0] ||
228 route
->nexthop
.in6
.s6_addr32
[1] ||
229 route
->nexthop
.in6
.s6_addr32
[2] ||
230 route
->nexthop
.in6
.s6_addr32
[3];
232 unsigned char scope
= (cmd
== RTM_DELROUTE
) ? RT_SCOPE_NOWHERE
:
233 (have_gw
) ? RT_SCOPE_UNIVERSE
: RT_SCOPE_LINK
;
236 .rtm_family
= (alen
== 4) ? AF_INET
: AF_INET6
,
237 .rtm_dst_len
= route
->mask
,
238 .rtm_table
= RT_TABLE_MAIN
,
239 .rtm_protocol
= RTPROT_BOOT
,
241 .rtm_type
= (cmd
== RTM_DELROUTE
) ? 0: RTN_UNICAST
,
244 struct nl_msg
*msg
= nlmsg_alloc_simple(cmd
, 0);
248 nlmsg_append(msg
, &rtm
, sizeof(rtm
), 0);
251 nla_put(msg
, RTA_DST
, alen
, &route
->addr
);
254 nla_put(msg
, RTA_GATEWAY
, alen
, &route
->nexthop
);
256 if (route
->flags
& DEVADDR_DEVICE
)
257 nla_put_u32(msg
, RTA_OIF
, dev
->ifindex
);
259 return system_rtnl_call(msg
);
262 int system_add_route(struct device
*dev
, struct device_route
*route
)
264 return system_rt(dev
, route
, RTM_NEWROUTE
);
267 int system_del_route(struct device
*dev
, struct device_route
*route
)
269 return system_rt(dev
, route
, RTM_DELROUTE
);