/*
*------------------------------------------------------------------
* 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;
clib_warning ("mtu: %d", *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;
}
/*
* fd.io coding-style-patch-verification: ON
*
* Local Variables:
* eval: (c-set-style "gnu")
* End:
*/
