| /* |
| *------------------------------------------------------------------ |
| * Copyright (c) 2017 Cisco and/or its affiliates. |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at: |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| *------------------------------------------------------------------ |
| */ |
| |
| #include <sys/types.h> |
| #include <sys/stat.h> |
| #include <fcntl.h> |
| #include <net/if.h> |
| |
| #include <linux/netlink.h> |
| #include <linux/rtnetlink.h> |
| |
| #include <vlib/vlib.h> |
| #include <vlib/unix/unix.h> |
| #include <vnet/devices/netlink.h> |
| |
| typedef struct |
| { |
| u8 *data; |
| } vnet_netlink_msg_t; |
| |
| static void |
| vnet_netlink_msg_init (vnet_netlink_msg_t * m, u16 type, u16 flags, |
| void *msg_data, int msg_len) |
| { |
| struct nlmsghdr *nh; |
| u8 *p; |
| clib_memset (m, 0, sizeof (vnet_netlink_msg_t)); |
| vec_add2 (m->data, p, NLMSG_SPACE (msg_len)); |
| ASSERT (m->data == p); |
| |
| nh = (struct nlmsghdr *) p; |
| nh->nlmsg_flags = flags | NLM_F_ACK; |
| nh->nlmsg_type = type; |
| clib_memcpy (m->data + sizeof (struct nlmsghdr), msg_data, msg_len); |
| } |
| |
| static void |
| vnet_netlink_msg_add_rtattr (vnet_netlink_msg_t * m, u16 rta_type, |
| void *rta_data, int rta_data_len) |
| { |
| struct rtattr *rta; |
| u8 *p; |
| |
| vec_add2 (m->data, p, RTA_SPACE (rta_data_len)); |
| rta = (struct rtattr *) p; |
| rta->rta_type = rta_type; |
| rta->rta_len = RTA_LENGTH (rta_data_len); |
| clib_memcpy (RTA_DATA (rta), rta_data, rta_data_len); |
| } |
| |
| static clib_error_t * |
| vnet_netlink_msg_send (vnet_netlink_msg_t *m, vnet_netlink_msg_t **replies) |
| { |
| clib_error_t *err = 0; |
| struct sockaddr_nl ra = { 0 }; |
| int len, sock; |
| struct nlmsghdr *nh = (struct nlmsghdr *) m->data; |
| nh->nlmsg_len = vec_len (m->data); |
| char buf[4096]; |
| |
| if ((sock = socket (AF_NETLINK, SOCK_RAW, NETLINK_ROUTE)) == -1) |
| return clib_error_return_unix (0, "socket(AF_NETLINK)"); |
| |
| ra.nl_family = AF_NETLINK; |
| ra.nl_pid = 0; |
| |
| if ((bind (sock, (struct sockaddr *) &ra, sizeof (ra))) == -1) |
| { |
| err = clib_error_return_unix (0, "bind"); |
| goto done; |
| } |
| |
| if ((send (sock, m->data, vec_len (m->data), 0)) == -1) |
| err = clib_error_return_unix (0, "send"); |
| |
| if ((len = recv (sock, buf, sizeof (buf), 0)) == -1) |
| err = clib_error_return_unix (0, "recv"); |
| |
| for (nh = (struct nlmsghdr *) buf; NLMSG_OK (nh, len); |
| nh = NLMSG_NEXT (nh, len)) |
| { |
| if (nh->nlmsg_type == NLMSG_DONE) |
| goto done; |
| |
| if (nh->nlmsg_type == NLMSG_ERROR) |
| { |
| struct nlmsgerr *e = (struct nlmsgerr *) NLMSG_DATA (nh); |
| if (e->error) |
| err = clib_error_return (0, "netlink error %d", e->error); |
| goto done; |
| } |
| |
| if (replies) |
| { |
| vnet_netlink_msg_t msg = { NULL }; |
| u8 *p; |
| vec_add2 (msg.data, p, nh->nlmsg_len); |
| clib_memcpy (p, nh, nh->nlmsg_len); |
| vec_add1 (*replies, msg); |
| } |
| } |
| |
| done: |
| close (sock); |
| vec_free (m->data); |
| return err; |
| } |
| |
| clib_error_t * |
| vnet_netlink_set_link_name (int ifindex, char *new_ifname) |
| { |
| vnet_netlink_msg_t m; |
| struct ifinfomsg ifmsg = { 0 }; |
| clib_error_t *err = 0; |
| |
| ifmsg.ifi_index = ifindex; |
| vnet_netlink_msg_init (&m, RTM_SETLINK, NLM_F_REQUEST, |
| &ifmsg, sizeof (struct ifinfomsg)); |
| |
| vnet_netlink_msg_add_rtattr (&m, IFLA_IFNAME, new_ifname, |
| strlen (new_ifname) + 1); |
| |
| err = vnet_netlink_msg_send (&m, NULL); |
| if (err) |
| err = clib_error_return (0, "set link name %U", format_clib_error, err); |
| return err; |
| } |
| |
| clib_error_t * |
| vnet_netlink_set_link_netns (int ifindex, int netns_fd, char *new_ifname) |
| { |
| vnet_netlink_msg_t m; |
| struct ifinfomsg ifmsg = { 0 }; |
| clib_error_t *err = 0; |
| |
| ifmsg.ifi_index = ifindex; |
| vnet_netlink_msg_init (&m, RTM_SETLINK, NLM_F_REQUEST, |
| &ifmsg, sizeof (struct ifinfomsg)); |
| |
| vnet_netlink_msg_add_rtattr (&m, IFLA_NET_NS_FD, &netns_fd, sizeof (int)); |
| if (new_ifname) |
| vnet_netlink_msg_add_rtattr (&m, IFLA_IFNAME, new_ifname, |
| strlen (new_ifname) + 1); |
| |
| err = vnet_netlink_msg_send (&m, NULL); |
| if (err) |
| err = clib_error_return (0, "set link netns %U", format_clib_error, err); |
| return err; |
| } |
| |
| clib_error_t * |
| vnet_netlink_set_link_master (int ifindex, char *master_ifname) |
| { |
| vnet_netlink_msg_t m; |
| struct ifinfomsg ifmsg = { 0 }; |
| int i; |
| clib_error_t *err = 0; |
| |
| ifmsg.ifi_index = ifindex; |
| |
| if ((i = if_nametoindex (master_ifname)) == 0) |
| clib_error_return_unix (0, "unknown master interface '%s'", |
| master_ifname); |
| |
| vnet_netlink_msg_init (&m, RTM_SETLINK, NLM_F_REQUEST, |
| &ifmsg, sizeof (struct ifinfomsg)); |
| vnet_netlink_msg_add_rtattr (&m, IFLA_MASTER, &i, sizeof (int)); |
| err = vnet_netlink_msg_send (&m, NULL); |
| if (err) |
| err = clib_error_return (0, "set link master %U", format_clib_error, err); |
| return err; |
| } |
| |
| clib_error_t * |
| vnet_netlink_set_link_addr (int ifindex, u8 * mac) |
| { |
| vnet_netlink_msg_t m; |
| struct ifinfomsg ifmsg = { 0 }; |
| clib_error_t *err = 0; |
| |
| ifmsg.ifi_index = ifindex; |
| |
| vnet_netlink_msg_init (&m, RTM_SETLINK, NLM_F_REQUEST, |
| &ifmsg, sizeof (struct ifinfomsg)); |
| vnet_netlink_msg_add_rtattr (&m, IFLA_ADDRESS, mac, 6); |
| err = vnet_netlink_msg_send (&m, NULL); |
| if (err) |
| err = clib_error_return (0, "set link addr %U", format_clib_error, err); |
| return err; |
| } |
| |
| clib_error_t * |
| vnet_netlink_set_link_state (int ifindex, int up) |
| { |
| vnet_netlink_msg_t m; |
| struct ifinfomsg ifmsg = { 0 }; |
| clib_error_t *err = 0; |
| |
| ifmsg.ifi_flags = ((up) ? IFF_UP : 0); |
| ifmsg.ifi_change = IFF_UP; |
| ifmsg.ifi_index = ifindex; |
| |
| vnet_netlink_msg_init (&m, RTM_SETLINK, NLM_F_REQUEST, |
| &ifmsg, sizeof (struct ifinfomsg)); |
| err = vnet_netlink_msg_send (&m, NULL); |
| if (err) |
| err = clib_error_return (0, "set link state %U", format_clib_error, err); |
| return err; |
| } |
| |
| clib_error_t * |
| vnet_netlink_get_link_mtu (int ifindex, u32 *mtu) |
| { |
| vnet_netlink_msg_t m, *msg; |
| struct ifinfomsg ifmsg = { 0 }; |
| struct nlattr *attr; |
| clib_error_t *err = 0; |
| vnet_netlink_msg_t *replies = NULL; |
| int len = 0, offset = 0; |
| u32 msg_mtu; |
| |
| ifmsg.ifi_index = ifindex; |
| |
| vnet_netlink_msg_init (&m, RTM_GETLINK, NLM_F_REQUEST, &ifmsg, |
| sizeof (struct ifinfomsg)); |
| // vnet_netlink_msg_add_rtattr (&m, IFLA_MTU, &mtu, sizeof (int)); |
| err = vnet_netlink_msg_send (&m, &replies); |
| if (err) |
| { |
| err = clib_error_return (0, "get link mtu %U", format_clib_error, err); |
| goto done; |
| } |
| |
| if (vec_len (replies) != 1) |
| { |
| err = clib_error_return (0, "got %d != 1 netlink reply msg", |
| vec_len (replies)); |
| goto done; |
| } |
| |
| struct nlmsghdr *nh = (struct nlmsghdr *) replies[0].data; |
| if (nh->nlmsg_type != RTM_NEWLINK) |
| { |
| err = clib_error_return ( |
| 0, "netlink reply has wrong type: %d != RTM_NEWLINK", nh->nlmsg_type); |
| goto done; |
| } |
| |
| offset = NLMSG_HDRLEN + NLMSG_ALIGN (sizeof (struct ifinfomsg)); |
| attr = (struct nlattr *) ((u8 *) nh + offset); |
| len = nh->nlmsg_len - offset; |
| |
| do |
| { |
| if ((attr->nla_type & NLA_TYPE_MASK) == IFLA_MTU) |
| { |
| msg_mtu = *(u32 *) ((u8 *) attr + NLA_HDRLEN); |
| if (attr->nla_type & NLA_F_NET_BYTEORDER) |
| *mtu = clib_net_to_host_u32 (msg_mtu); |
| else |
| *mtu = msg_mtu; |
| goto done; |
| } |
| offset = NLA_ALIGN (attr->nla_len); |
| len -= offset; |
| attr = (struct nlattr *) ((u8 *) attr + offset); |
| } |
| while (len > sizeof (struct nlattr)); |
| |
| /* not found */ |
| err = clib_error_return (0, "mtu not found in netlink message"); |
| |
| done: |
| vec_foreach (msg, replies) |
| { |
| vec_free (msg->data); |
| } |
| vec_free (replies); |
| |
| return err; |
| } |
| |
| clib_error_t * |
| vnet_netlink_set_link_mtu (int ifindex, int mtu) |
| { |
| vnet_netlink_msg_t m; |
| struct ifinfomsg ifmsg = { 0 }; |
| clib_error_t *err = 0; |
| |
| ifmsg.ifi_index = ifindex; |
| |
| vnet_netlink_msg_init (&m, RTM_SETLINK, NLM_F_REQUEST, |
| &ifmsg, sizeof (struct ifinfomsg)); |
| vnet_netlink_msg_add_rtattr (&m, IFLA_MTU, &mtu, sizeof (int)); |
| err = vnet_netlink_msg_send (&m, NULL); |
| if (err) |
| err = clib_error_return (0, "set link mtu %U", format_clib_error, err); |
| return err; |
| } |
| |
| clib_error_t * |
| vnet_netlink_add_ip4_addr (int ifindex, void *addr, int pfx_len) |
| { |
| vnet_netlink_msg_t m; |
| struct ifaddrmsg ifa = { 0 }; |
| clib_error_t *err = 0; |
| |
| ifa.ifa_family = AF_INET; |
| ifa.ifa_prefixlen = pfx_len; |
| ifa.ifa_index = ifindex; |
| |
| vnet_netlink_msg_init (&m, RTM_NEWADDR, |
| NLM_F_REQUEST | NLM_F_CREATE | NLM_F_REPLACE, |
| &ifa, sizeof (struct ifaddrmsg)); |
| |
| vnet_netlink_msg_add_rtattr (&m, IFA_LOCAL, addr, 4); |
| vnet_netlink_msg_add_rtattr (&m, IFA_ADDRESS, addr, 4); |
| err = vnet_netlink_msg_send (&m, NULL); |
| if (err) |
| err = clib_error_return (0, "add ip4 addr %U", format_clib_error, err); |
| return err; |
| } |
| |
| clib_error_t * |
| vnet_netlink_add_ip6_addr (int ifindex, void *addr, int pfx_len) |
| { |
| vnet_netlink_msg_t m; |
| struct ifaddrmsg ifa = { 0 }; |
| clib_error_t *err = 0; |
| |
| ifa.ifa_family = AF_INET6; |
| ifa.ifa_prefixlen = pfx_len; |
| ifa.ifa_index = ifindex; |
| |
| vnet_netlink_msg_init (&m, RTM_NEWADDR, |
| NLM_F_REQUEST | NLM_F_CREATE | NLM_F_REPLACE, |
| &ifa, sizeof (struct ifaddrmsg)); |
| |
| vnet_netlink_msg_add_rtattr (&m, IFA_LOCAL, addr, 16); |
| vnet_netlink_msg_add_rtattr (&m, IFA_ADDRESS, addr, 16); |
| err = vnet_netlink_msg_send (&m, NULL); |
| if (err) |
| err = clib_error_return (0, "add ip6 addr %U", format_clib_error, err); |
| return err; |
| } |
| |
| clib_error_t * |
| vnet_netlink_add_ip4_route (void *dst, u8 dst_len, void *gw) |
| { |
| vnet_netlink_msg_t m; |
| struct rtmsg rtm = { 0 }; |
| u8 dflt[4] = { 0 }; |
| clib_error_t *err = 0; |
| |
| rtm.rtm_family = AF_INET; |
| rtm.rtm_table = RT_TABLE_MAIN; |
| rtm.rtm_type = RTN_UNICAST; |
| rtm.rtm_dst_len = dst_len; |
| |
| vnet_netlink_msg_init (&m, RTM_NEWROUTE, |
| NLM_F_REQUEST | NLM_F_CREATE | NLM_F_REPLACE, |
| &rtm, sizeof (struct rtmsg)); |
| |
| vnet_netlink_msg_add_rtattr (&m, RTA_GATEWAY, gw, 4); |
| vnet_netlink_msg_add_rtattr (&m, RTA_DST, dst ? dst : dflt, 4); |
| err = vnet_netlink_msg_send (&m, NULL); |
| if (err) |
| err = clib_error_return (0, "add ip4 route %U", format_clib_error, err); |
| return err; |
| } |
| |
| clib_error_t * |
| vnet_netlink_add_ip6_route (void *dst, u8 dst_len, void *gw) |
| { |
| vnet_netlink_msg_t m; |
| struct rtmsg rtm = { 0 }; |
| u8 dflt[16] = { 0 }; |
| clib_error_t *err = 0; |
| |
| rtm.rtm_family = AF_INET6; |
| rtm.rtm_table = RT_TABLE_MAIN; |
| rtm.rtm_type = RTN_UNICAST; |
| rtm.rtm_dst_len = dst_len; |
| |
| vnet_netlink_msg_init (&m, RTM_NEWROUTE, |
| NLM_F_REQUEST | NLM_F_CREATE | NLM_F_REPLACE, |
| &rtm, sizeof (struct rtmsg)); |
| |
| vnet_netlink_msg_add_rtattr (&m, RTA_GATEWAY, gw, 16); |
| vnet_netlink_msg_add_rtattr (&m, RTA_DST, dst ? dst : dflt, 16); |
| err = vnet_netlink_msg_send (&m, NULL); |
| if (err) |
| err = clib_error_return (0, "add ip6 route %U", format_clib_error, err); |
| return err; |
| } |
| |
| clib_error_t * |
| vnet_netlink_del_ip4_addr (int ifindex, void *addr, int pfx_len) |
| { |
| vnet_netlink_msg_t m; |
| struct ifaddrmsg ifa = { 0 }; |
| clib_error_t *err = 0; |
| |
| ifa.ifa_family = AF_INET; |
| ifa.ifa_prefixlen = pfx_len; |
| ifa.ifa_index = ifindex; |
| |
| vnet_netlink_msg_init (&m, RTM_DELADDR, NLM_F_REQUEST, &ifa, |
| sizeof (struct ifaddrmsg)); |
| |
| vnet_netlink_msg_add_rtattr (&m, IFA_LOCAL, addr, 4); |
| vnet_netlink_msg_add_rtattr (&m, IFA_ADDRESS, addr, 4); |
| err = vnet_netlink_msg_send (&m, NULL); |
| if (err) |
| err = clib_error_return (0, "del ip4 addr %U", format_clib_error, err); |
| return err; |
| } |
| |
| clib_error_t * |
| vnet_netlink_del_ip6_addr (int ifindex, void *addr, int pfx_len) |
| { |
| vnet_netlink_msg_t m; |
| struct ifaddrmsg ifa = { 0 }; |
| clib_error_t *err = 0; |
| |
| ifa.ifa_family = AF_INET6; |
| ifa.ifa_prefixlen = pfx_len; |
| ifa.ifa_index = ifindex; |
| |
| vnet_netlink_msg_init (&m, RTM_DELADDR, NLM_F_REQUEST, &ifa, |
| sizeof (struct ifaddrmsg)); |
| |
| vnet_netlink_msg_add_rtattr (&m, IFA_LOCAL, addr, 16); |
| vnet_netlink_msg_add_rtattr (&m, IFA_ADDRESS, addr, 16); |
| err = vnet_netlink_msg_send (&m, NULL); |
| if (err) |
| err = clib_error_return (0, "del ip6 addr %U", format_clib_error, err); |
| return err; |
| } |
| |
| /* |
| * fd.io coding-style-patch-verification: ON |
| * |
| * Local Variables: |
| * eval: (c-set-style "gnu") |
| * End: |
| */ |