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
};
25 static struct nl_cb
*nl_cb_rtnl_event
;
29 sock_ioctl
= socket(AF_LOCAL
, SOCK_DGRAM
, 0);
30 fcntl(sock_ioctl
, F_SETFD
, fcntl(sock_ioctl
, F_GETFD
) | FD_CLOEXEC
);
32 // Prepare socket for routing / address control
33 if ((sock_rtnl
= nl_socket_alloc())) {
34 if (nl_connect(sock_rtnl
, NETLINK_ROUTE
)) {
35 nl_socket_free(sock_rtnl
);
40 // Prepare socket for link events
41 if ((nl_cb_rtnl_event
= nl_cb_alloc(NL_CB_DEFAULT
)))
42 nl_cb_set(nl_cb_rtnl_event
, NL_CB_VALID
, NL_CB_CUSTOM
,
45 if (nl_cb_rtnl_event
&& (sock_rtnl_event
= nl_socket_alloc())) {
46 if (nl_connect(sock_rtnl_event
, NETLINK_ROUTE
)) {
47 nl_socket_free(sock_rtnl_event
);
48 sock_rtnl_event
= NULL
;
50 // Receive network link events form kernel
51 nl_socket_add_membership(sock_rtnl_event
, RTNLGRP_LINK
);
53 // Synthesize initial link messages
54 struct nl_msg
*m
= nlmsg_alloc_simple(RTM_GETLINK
, NLM_F_DUMP
);
55 if (m
&& nlmsg_reserve(m
, sizeof(struct ifinfomsg
), 0)) {
56 nl_send_auto_complete(sock_rtnl_event
, m
);
61 rtnl_event
.fd
= nl_socket_get_fd(sock_rtnl_event
);
63 rtnl_event
.fd
= sock_rtnl_event
->s_fd
; // libnl-tiny hack...
65 uloop_fd_add(&rtnl_event
, ULOOP_READ
| ULOOP_EDGE_TRIGGER
);
68 return -(sock_ioctl
< 0 || !sock_rtnl
);
71 // If socket is ready for reading parse netlink events
72 static void handler_rtnl_event(struct uloop_fd
*u
, unsigned int events
)
74 nl_recvmsgs(sock_rtnl_event
, nl_cb_rtnl_event
);
77 // Evaluate netlink messages
78 static int cb_rtnl_event(struct nl_msg
*msg
, void *arg
)
80 struct nlmsghdr
*nh
= nlmsg_hdr(msg
);
81 struct ifinfomsg
*ifi
= NLMSG_DATA(nh
);
82 struct nlattr
*nla
[__IFLA_MAX
];
84 if (nh
->nlmsg_type
!= RTM_DELLINK
&& nh
->nlmsg_type
!= RTM_NEWLINK
)
87 nlmsg_parse(nh
, sizeof(*ifi
), nla
, __IFLA_MAX
- 1, NULL
);
88 if (!nla
[IFLA_IFNAME
])
91 struct device
*dev
= device_get(RTA_DATA(nla
[IFLA_IFNAME
]), false);
95 dev
->ifindex
= ifi
->ifi_index
;
96 device_set_present(dev
, (nh
->nlmsg_type
== RTM_NEWLINK
));
102 static int system_rtnl_call(struct nl_msg
*msg
)
104 int s
= -(nl_send_auto_complete(sock_rtnl
, msg
)
105 || nl_wait_for_ack(sock_rtnl
));
110 int system_bridge_addbr(struct device
*bridge
)
112 return ioctl(sock_ioctl
, SIOCBRADDBR
, bridge
->ifname
);
115 int system_bridge_delbr(struct device
*bridge
)
117 return ioctl(sock_ioctl
, SIOCBRDELBR
, bridge
->ifname
);
120 static int system_bridge_if(struct device
*bridge
, struct device
*dev
, int cmd
)
123 ifr
.ifr_ifindex
= dev
->ifindex
;
124 strncpy(ifr
.ifr_name
, bridge
->ifname
, sizeof(ifr
.ifr_name
));
125 return ioctl(sock_ioctl
, cmd
, &ifr
);
128 int system_bridge_addif(struct device
*bridge
, struct device
*dev
)
130 return system_bridge_if(bridge
, dev
, SIOCBRADDIF
);
133 int system_bridge_delif(struct device
*bridge
, struct device
*dev
)
135 return system_bridge_if(bridge
, dev
, SIOCBRDELIF
);
138 static int system_vlan(struct device
*dev
, int id
)
140 struct vlan_ioctl_args ifr
= {
141 .cmd
= (id
== 0) ? DEL_VLAN_CMD
: ADD_VLAN_CMD
,
144 strncpy(ifr
.device1
, dev
->ifname
, sizeof(ifr
.device1
));
145 return ioctl(sock_ioctl
, SIOCSIFVLAN
, &ifr
);
148 int system_vlan_add(struct device
*dev
, int id
)
150 return system_vlan(dev
, id
);
153 int system_vlan_del(struct device
*dev
)
155 return system_vlan(dev
, 0);
158 static int system_if_flags(struct device
*dev
, unsigned add
, unsigned rem
)
161 strncpy(ifr
.ifr_name
, dev
->ifname
, sizeof(ifr
.ifr_name
));
162 ioctl(sock_ioctl
, SIOCGIFFLAGS
, &ifr
);
163 ifr
.ifr_flags
|= add
;
164 ifr
.ifr_flags
&= ~rem
;
165 return ioctl(sock_ioctl
, SIOCSIFFLAGS
, &ifr
);
168 int system_if_up(struct device
*dev
)
170 return system_if_flags(dev
, IFF_UP
, 0);
173 int system_if_down(struct device
*dev
)
175 return system_if_flags(dev
, 0, IFF_UP
);
178 int system_if_check(struct device
*dev
)
181 strncpy(ifr
.ifr_name
, dev
->ifname
, sizeof(ifr
.ifr_name
));
182 if (ioctl(sock_ioctl
, SIOCGIFINDEX
, &ifr
))
185 dev
->ifindex
= ifr
.ifr_ifindex
;
187 /* if (!strcmp(dev->ifname, "eth0"))
188 device_set_present(dev, true); */
192 static int system_addr(struct device
*dev
, struct device_addr
*addr
, int cmd
)
194 int alen
= ((addr
->flags
& DEVADDR_FAMILY
) == DEVADDR_INET4
) ? 4 : 16;
195 struct ifaddrmsg ifa
= {
196 .ifa_family
= (alen
== 4) ? AF_INET
: AF_INET6
,
197 .ifa_prefixlen
= addr
->mask
,
198 .ifa_index
= dev
->ifindex
,
201 struct nl_msg
*msg
= nlmsg_alloc_simple(cmd
, 0);
205 nlmsg_append(msg
, &ifa
, sizeof(ifa
), 0);
206 nla_put(msg
, IFA_ADDRESS
, alen
, &addr
->addr
);
207 return system_rtnl_call(msg
);
210 int system_add_address(struct device
*dev
, struct device_addr
*addr
)
212 return system_addr(dev
, addr
, RTM_NEWADDR
);
215 int system_del_address(struct device
*dev
, struct device_addr
*addr
)
217 return system_addr(dev
, addr
, RTM_DELADDR
);
220 static int system_rt(struct device
*dev
, struct device_route
*route
, int cmd
)
222 int alen
= ((route
->flags
& DEVADDR_FAMILY
) == DEVADDR_INET4
) ? 4 : 16;
226 have_gw
= !!route
->nexthop
.in
.s_addr
;
228 have_gw
= route
->nexthop
.in6
.s6_addr32
[0] ||
229 route
->nexthop
.in6
.s6_addr32
[1] ||
230 route
->nexthop
.in6
.s6_addr32
[2] ||
231 route
->nexthop
.in6
.s6_addr32
[3];
233 unsigned char scope
= (cmd
== RTM_DELROUTE
) ? RT_SCOPE_NOWHERE
:
234 (have_gw
) ? RT_SCOPE_UNIVERSE
: RT_SCOPE_LINK
;
237 .rtm_family
= (alen
== 4) ? AF_INET
: AF_INET6
,
238 .rtm_dst_len
= route
->mask
,
239 .rtm_table
= RT_TABLE_MAIN
,
240 .rtm_protocol
= RTPROT_BOOT
,
242 .rtm_type
= (cmd
== RTM_DELROUTE
) ? 0: RTN_UNICAST
,
245 struct nl_msg
*msg
= nlmsg_alloc_simple(cmd
, 0);
249 nlmsg_append(msg
, &rtm
, sizeof(rtm
), 0);
252 nla_put(msg
, RTA_DST
, alen
, &route
->addr
);
255 nla_put(msg
, RTA_GATEWAY
, alen
, &route
->nexthop
);
257 if (route
->flags
& DEVADDR_DEVICE
)
258 nla_put_u32(msg
, RTA_OIF
, dev
->ifindex
);
260 return system_rtnl_call(msg
);
263 int system_add_route(struct device
*dev
, struct device_route
*route
)
265 return system_rt(dev
, route
, RTM_NEWROUTE
);
268 int system_del_route(struct device
*dev
, struct device_route
*route
)
270 return system_rt(dev
, route
, RTM_DELROUTE
);