system-linux: free netlink messages after sending
[project/netifd.git] / system-linux.c
1 #include <sys/socket.h>
2 #include <sys/ioctl.h>
3
4 #include <linux/rtnetlink.h>
5 #include <linux/sockios.h>
6 #include <linux/if_vlan.h>
7
8 #include <stddef.h>
9 #include <string.h>
10 #include <fcntl.h>
11 #include <errno.h>
12
13 #include <netlink/msg.h>
14
15 #include "netifd.h"
16 #include "device.h"
17 #include "system.h"
18
19 static int sock_ioctl = -1;
20 static struct nl_sock *sock_rtnl = NULL;
21
22 int system_init(void)
23 {
24         sock_ioctl = socket(AF_LOCAL, SOCK_DGRAM, 0);
25         fcntl(sock_ioctl, F_SETFD, fcntl(sock_ioctl, F_GETFD) | FD_CLOEXEC);
26
27         if ((sock_rtnl = nl_socket_alloc())) {
28                 if (nl_connect(sock_rtnl, NETLINK_ROUTE)) {
29                         nl_socket_free(sock_rtnl);
30                         sock_rtnl = NULL;
31                 }
32         }
33
34         return -(sock_ioctl < 0 || !sock_rtnl);
35 }
36
37 static int system_rtnl_call(struct nl_msg *msg)
38 {
39         int s = -(nl_send_auto_complete(sock_rtnl, msg)
40                         || nl_wait_for_ack(sock_rtnl));
41         nlmsg_free(msg);
42         return s;
43 }
44
45 int system_bridge_addbr(struct device *bridge)
46 {
47         return ioctl(sock_ioctl, SIOCBRADDBR, bridge->ifname);
48 }
49
50 int system_bridge_delbr(struct device *bridge)
51 {
52         return ioctl(sock_ioctl, SIOCBRDELBR, bridge->ifname);
53 }
54
55 static int system_bridge_if(struct device *bridge, struct device *dev, int cmd)
56 {
57         struct ifreq ifr;
58         ifr.ifr_ifindex = dev->ifindex;
59         strncpy(ifr.ifr_name, bridge->ifname, sizeof(ifr.ifr_name));
60         return ioctl(sock_ioctl, cmd, &ifr);
61 }
62
63 int system_bridge_addif(struct device *bridge, struct device *dev)
64 {
65         return system_bridge_if(bridge, dev, SIOCBRADDIF);
66 }
67
68 int system_bridge_delif(struct device *bridge, struct device *dev)
69 {
70         return system_bridge_if(bridge, dev, SIOCBRDELIF);
71 }
72
73 static int system_vlan(struct device *dev, int id)
74 {
75         struct vlan_ioctl_args ifr = {
76                 .cmd = (id == 0) ? DEL_VLAN_CMD : ADD_VLAN_CMD,
77                 .u = {.VID = id},
78         };
79         strncpy(ifr.device1, dev->ifname, sizeof(ifr.device1));
80         return ioctl(sock_ioctl, SIOCSIFVLAN, &ifr);
81 }
82
83 int system_vlan_add(struct device *dev, int id)
84 {
85         return system_vlan(dev, id);
86 }
87
88 int system_vlan_del(struct device *dev)
89 {
90         return system_vlan(dev, 0);
91 }
92
93 static int system_if_flags(struct device *dev, unsigned add, unsigned rem)
94 {
95         struct ifreq ifr;
96         strncpy(ifr.ifr_name, dev->ifname, sizeof(ifr.ifr_name));
97         ioctl(sock_ioctl, SIOCGIFFLAGS, &ifr);
98         ifr.ifr_flags |= add;
99         ifr.ifr_flags &= ~rem;
100         return ioctl(sock_ioctl, SIOCSIFFLAGS, &ifr);
101 }
102
103 int system_if_up(struct device *dev)
104 {
105         return system_if_flags(dev, IFF_UP, 0);
106 }
107
108 int system_if_down(struct device *dev)
109 {
110         return system_if_flags(dev, 0, IFF_UP);
111 }
112
113 int system_if_check(struct device *dev)
114 {
115         struct ifreq ifr;
116         strncpy(ifr.ifr_name, dev->ifname, sizeof(ifr.ifr_name));
117         if (ioctl(sock_ioctl, SIOCGIFINDEX, &ifr))
118                 return -1;
119
120         dev->ifindex = ifr.ifr_ifindex;
121
122         /* if (!strcmp(dev->ifname, "eth0"))
123                 device_set_present(dev, true); */
124         return 0;
125 }
126
127 static int system_addr(struct device *dev, struct device_addr *addr, int cmd)
128 {
129         int alen = ((addr->flags & DEVADDR_FAMILY) == DEVADDR_INET4) ? 4 : 16;
130         struct ifaddrmsg ifa = {
131                 .ifa_family = (alen == 4) ? AF_INET : AF_INET6,
132                 .ifa_prefixlen = addr->mask,
133                 .ifa_index = dev->ifindex,
134         };
135
136         struct nl_msg *msg = nlmsg_alloc_simple(cmd, 0);
137         if (!msg)
138                 return -1;
139
140         nlmsg_append(msg, &ifa, sizeof(ifa), 0);
141         nla_put(msg, IFA_ADDRESS, alen, &addr->addr);
142         return system_rtnl_call(msg);
143 }
144
145 int system_add_address(struct device *dev, struct device_addr *addr)
146 {
147         return system_addr(dev, addr, RTM_NEWADDR);
148 }
149
150 int system_del_address(struct device *dev, struct device_addr *addr)
151 {
152         return system_addr(dev, addr, RTM_DELADDR);
153 }
154
155 static int system_rt(struct device *dev, struct device_route *route, int cmd)
156 {
157         int alen = ((route->flags & DEVADDR_FAMILY) == DEVADDR_INET4) ? 4 : 16;
158         bool have_gw;
159
160         if (alen == 4)
161                 have_gw = !!route->nexthop.in.s_addr;
162         else
163                 have_gw = route->nexthop.in6.s6_addr32[0] ||
164                         route->nexthop.in6.s6_addr32[1] ||
165                         route->nexthop.in6.s6_addr32[2] ||
166                         route->nexthop.in6.s6_addr32[3];
167
168         unsigned char scope = (cmd == RTM_DELROUTE) ? RT_SCOPE_NOWHERE :
169                         (have_gw) ? RT_SCOPE_UNIVERSE : RT_SCOPE_LINK;
170
171         struct rtmsg rtm = {
172                 .rtm_family = (alen == 4) ? AF_INET : AF_INET6,
173                 .rtm_dst_len = route->mask,
174                 .rtm_table = RT_TABLE_MAIN,
175                 .rtm_protocol = RTPROT_BOOT,
176                 .rtm_scope = scope,
177                 .rtm_type = (cmd == RTM_DELROUTE) ? 0: RTN_UNICAST,
178         };
179
180         struct nl_msg *msg = nlmsg_alloc_simple(cmd, 0);
181         if (!msg)
182                 return -1;
183
184         nlmsg_append(msg, &rtm, sizeof(rtm), 0);
185
186         if (route->mask)
187                 nla_put(msg, RTA_DST, alen, &route->addr);
188
189         if (have_gw)
190                 nla_put(msg, RTA_GATEWAY, alen, &route->nexthop);
191
192         if (route->flags & DEVADDR_DEVICE)
193                 nla_put_u32(msg, RTA_OIF, dev->ifindex);
194
195         return system_rtnl_call(msg);
196 }
197
198 int system_add_route(struct device *dev, struct device_route *route)
199 {
200         return system_rt(dev, route, RTM_NEWROUTE);
201 }
202
203 int system_del_route(struct device *dev, struct device_route *route)
204 {
205         return system_rt(dev, route, RTM_DELROUTE);
206 }