mirror of
https://github.com/zerotier/ZeroTierOne.git
synced 2025-06-05 03:53:44 +02:00
add/remove routes via rtnetlink
This commit is contained in:
Grant Limberg
parent
7da4b0af15
commit
be469f4dd0
4 changed files with 538 additions and 359 deletions
|
|
@ -399,6 +399,10 @@ private:
|
|||
{
|
||||
return ((unsigned long)i * (unsigned long)0x9e3779b1);
|
||||
}
|
||||
static inline unsigned long _hc(const int i)
|
||||
{
|
||||
return ((unsigned long)i * (unsigned long)0x9e3379b1);
|
||||
}
|
||||
|
||||
inline void _grow()
|
||||
{
|
||||
|
|
|
|||
|
|
@ -30,6 +30,18 @@
|
|||
|
||||
namespace ZeroTier {
|
||||
|
||||
struct nl_route_req {
|
||||
struct nlmsghdr nl;
|
||||
struct rtmsg rt;
|
||||
char buf[8192];
|
||||
};
|
||||
|
||||
struct nl_if_req {
|
||||
struct nlmsghdr nl;
|
||||
struct ifinfomsg ifa;
|
||||
char buf[8192];
|
||||
};
|
||||
|
||||
LinuxNetLink::LinuxNetLink()
|
||||
: _t()
|
||||
, _running(false)
|
||||
|
|
@ -64,6 +76,9 @@ LinuxNetLink::LinuxNetLink()
|
|||
Thread::sleep(10);
|
||||
fprintf(stderr, "Requesting IPV6 Routes\n");
|
||||
_requestIPv6Routes();
|
||||
Thread::sleep(10);
|
||||
fprintf(stderr, "Requesting Interface List\n");
|
||||
_requestInterfaceList();
|
||||
}
|
||||
|
||||
LinuxNetLink::~LinuxNetLink()
|
||||
|
|
@ -295,9 +310,9 @@ void LinuxNetLink::_routeDeleted(struct nlmsghdr *nlp)
|
|||
|
||||
void LinuxNetLink::_linkAdded(struct nlmsghdr *nlp)
|
||||
{
|
||||
char mac[20] = {0};
|
||||
char mac[18] = {0};
|
||||
unsigned int mtu = 0;
|
||||
char ifname[40] = {0};
|
||||
char ifname[IFNAMSIZ] = {0};
|
||||
|
||||
struct ifinfomsg *ifip = (struct ifinfomsg *)NLMSG_DATA(nlp);
|
||||
struct rtattr *rtap = (struct rtattr *)IFLA_RTA(ifip);
|
||||
|
|
@ -323,12 +338,18 @@ void LinuxNetLink::_linkAdded(struct nlmsghdr *nlp)
|
|||
}
|
||||
}
|
||||
|
||||
struct iface_entry &entry = _interfaces[ifip->ifi_index];
|
||||
entry.index = ifip->ifi_index;
|
||||
memcpy(entry.ifacename, ifname, sizeof(ifname));
|
||||
memcpy(entry.mac, mac, sizeof(mac));
|
||||
entry.mtu = mtu;
|
||||
|
||||
fprintf(stderr, "Link Added: %s mac: %s, mtu: %d\n", ifname, mac, mtu);
|
||||
}
|
||||
|
||||
void LinuxNetLink::_linkDeleted(struct nlmsghdr *nlp)
|
||||
{
|
||||
char mac[20] = {0};
|
||||
char mac[18] = {0};
|
||||
unsigned int mtu = 0;
|
||||
char ifname[40] = {0};
|
||||
|
||||
|
|
@ -357,11 +378,14 @@ void LinuxNetLink::_linkDeleted(struct nlmsghdr *nlp)
|
|||
}
|
||||
|
||||
fprintf(stderr, "Link Deleted: %s mac: %s, mtu: %d\n", ifname, mac, mtu);
|
||||
if(_interfaces.contains(ifip->ifi_index)) {
|
||||
_interfaces.erase(ifip->ifi_index);
|
||||
}
|
||||
}
|
||||
|
||||
void LinuxNetLink::_requestIPv4Routes()
|
||||
{
|
||||
struct nl_req req;
|
||||
struct nl_route_req req;
|
||||
bzero(&req, sizeof(req));
|
||||
req.nl.nlmsg_len = NLMSG_LENGTH(sizeof(struct rtmsg));
|
||||
req.nl.nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP;
|
||||
|
|
@ -392,7 +416,7 @@ void LinuxNetLink::_requestIPv4Routes()
|
|||
|
||||
void LinuxNetLink::_requestIPv6Routes()
|
||||
{
|
||||
struct nl_req req;
|
||||
struct nl_route_req req;
|
||||
bzero(&req, sizeof(req));
|
||||
req.nl.nlmsg_len = NLMSG_LENGTH(sizeof(struct rtmsg));
|
||||
req.nl.nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP;
|
||||
|
|
@ -418,20 +442,196 @@ void LinuxNetLink::_requestIPv6Routes()
|
|||
msg.msg_iov = &iov;
|
||||
msg.msg_iovlen = 1;
|
||||
|
||||
while((sendmsg(_fd, &msg, 0)) == -1) {
|
||||
fprintf(stderr, "ipv6 waiting...");
|
||||
Thread::sleep(100);
|
||||
}
|
||||
sendmsg(_fd, &msg, 0);
|
||||
}
|
||||
|
||||
void LinuxNetLink::_requestInterfaceList()
|
||||
{
|
||||
struct nl_if_req req;
|
||||
bzero(&req, sizeof(req));
|
||||
req.nl.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifinfomsg));
|
||||
req.nl.nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP;
|
||||
req.nl.nlmsg_type = RTM_GETLINK;
|
||||
req.nl.nlmsg_pid = 0;
|
||||
req.nl.nlmsg_seq = ++_seq;
|
||||
req.ifa.ifi_family = AF_UNSPEC;
|
||||
|
||||
struct sockaddr_nl pa;
|
||||
bzero(&pa, sizeof(pa));
|
||||
pa.nl_family = AF_NETLINK;
|
||||
|
||||
struct msghdr msg;
|
||||
bzero(&msg, sizeof(msg));
|
||||
msg.msg_name = (void*)&pa;
|
||||
msg.msg_namelen = sizeof(pa);
|
||||
|
||||
struct iovec iov;
|
||||
bzero(&iov, sizeof(iov));
|
||||
iov.iov_base = (void*)&req.nl;
|
||||
iov.iov_len = req.nl.nlmsg_len;
|
||||
msg.msg_iov = &iov;
|
||||
msg.msg_iovlen = 1;
|
||||
sendmsg(_fd, &msg, 0);
|
||||
}
|
||||
|
||||
void LinuxNetLink::addRoute(const InetAddress &target, const InetAddress &via, const char *ifaceName)
|
||||
{
|
||||
int rtl = sizeof(struct rtmsg);
|
||||
struct nl_route_req req;
|
||||
bzero(&req, sizeof(req));
|
||||
|
||||
struct rtattr *rtap = (struct rtattr *)req.buf;
|
||||
rtap->rta_type = RTA_DST;
|
||||
if (target.isV4()) {
|
||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(struct in_addr);
|
||||
memcpy((void*)((char*)rtap+sizeof(struct rtattr)), &((struct sockaddr_in*)&target)->sin_addr, sizeof(struct in_addr));
|
||||
} else {
|
||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(struct in6_addr);
|
||||
memcpy((void*)((char*)rtap+sizeof(struct rtattr)), &((struct sockaddr_in6*)&target)->sin6_addr, sizeof(struct in6_addr));
|
||||
}
|
||||
rtl += rtap->rta_len;
|
||||
|
||||
int interface_index = -1;
|
||||
if (ifaceName != NULL) {
|
||||
Hashtable<int, iface_entry>::Iterator iter(_interfaces);
|
||||
int *k = NULL;
|
||||
iface_entry *v = NULL;
|
||||
while(iter.next(k, v)) {
|
||||
if(strcmp(ifaceName, v->ifacename) == 0) {
|
||||
interface_index = v->index;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (interface_index != -1) {
|
||||
rtap = (struct rtattr *) (((char*)rtap) + rtap->rta_len);
|
||||
rtap->rta_type = RTA_OIF;
|
||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(int);
|
||||
memcpy(((char*)rtap)+sizeof(rtattr), &interface_index, sizeof(int));
|
||||
rtl += rtap->rta_len;
|
||||
}
|
||||
}
|
||||
|
||||
if(via) {
|
||||
rtap = (struct rtattr *)(((char*)rtap)+rtap->rta_len);
|
||||
rtap->rta_type = RTA_GATEWAY;
|
||||
if(via.isV4()) {
|
||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(struct in_addr);
|
||||
memcpy((char*)rtap+sizeof(struct rtattr), &((struct sockaddr_in*)&via)->sin_addr, sizeof(struct in_addr));
|
||||
} else {
|
||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(struct in6_addr);
|
||||
memcpy((char*)rtap+sizeof(struct rtattr), &((struct sockaddr_in6*)&via)->sin6_addr, sizeof(struct in6_addr));
|
||||
}
|
||||
rtl += rtap->rta_len;
|
||||
}
|
||||
|
||||
req.nl.nlmsg_len = NLMSG_LENGTH(rtl);
|
||||
req.nl.nlmsg_flags = NLM_F_REQUEST | NLM_F_CREATE;
|
||||
req.nl.nlmsg_type = RTM_NEWROUTE;
|
||||
req.nl.nlmsg_pid = 0;
|
||||
req.nl.nlmsg_seq = ++_seq;
|
||||
req.rt.rtm_family = target.ss_family;
|
||||
req.rt.rtm_table = RT_TABLE_MAIN;
|
||||
req.rt.rtm_protocol = RTPROT_STATIC;
|
||||
req.rt.rtm_scope = RT_SCOPE_UNIVERSE;
|
||||
req.rt.rtm_type = RTN_UNICAST;
|
||||
req.rt.rtm_dst_len = target.netmaskBits();
|
||||
|
||||
struct sockaddr_nl pa;
|
||||
bzero(&pa, sizeof(pa));
|
||||
pa.nl_family = AF_NETLINK;
|
||||
|
||||
struct msghdr msg;
|
||||
bzero(&msg, sizeof(msg));
|
||||
msg.msg_name = (void*)&pa;
|
||||
msg.msg_namelen = sizeof(pa);
|
||||
|
||||
struct iovec iov;
|
||||
bzero(&iov, sizeof(iov));
|
||||
iov.iov_base = (void*)&req.nl;
|
||||
iov.iov_len = req.nl.nlmsg_len;
|
||||
msg.msg_iov = &iov;
|
||||
msg.msg_iovlen = 1;
|
||||
sendmsg(_fd, &msg, 0);
|
||||
}
|
||||
|
||||
void LinuxNetLink::delRoute(const InetAddress &target, const InetAddress &via, const char *ifaceName)
|
||||
{
|
||||
int rtl = sizeof(struct rtmsg);
|
||||
struct nl_route_req req;
|
||||
bzero(&req, sizeof(req));
|
||||
|
||||
struct rtattr *rtap = (struct rtattr *)req.buf;
|
||||
rtap->rta_type = RTA_DST;
|
||||
if (target.isV4()) {
|
||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(struct in_addr);
|
||||
memcpy((void*)((char*)rtap+sizeof(struct rtattr)), &((struct sockaddr_in*)&target)->sin_addr, sizeof(struct in_addr));
|
||||
} else {
|
||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(struct in6_addr);
|
||||
memcpy((void*)((char*)rtap+sizeof(struct rtattr)), &((struct sockaddr_in6*)&target)->sin6_addr, sizeof(struct in6_addr));
|
||||
}
|
||||
rtl += rtap->rta_len;
|
||||
|
||||
int interface_index = -1;
|
||||
if (ifaceName != NULL) {
|
||||
Hashtable<int, iface_entry>::Iterator iter(_interfaces);
|
||||
int *k = NULL;
|
||||
iface_entry *v = NULL;
|
||||
while(iter.next(k, v)) {
|
||||
if(strcmp(ifaceName, v->ifacename) == 0) {
|
||||
interface_index = v->index;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (interface_index != -1) {
|
||||
rtap = (struct rtattr *) (((char*)rtap) + rtap->rta_len);
|
||||
rtap->rta_type = RTA_OIF;
|
||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(int);
|
||||
memcpy(((char*)rtap)+sizeof(rtattr), &interface_index, sizeof(int));
|
||||
rtl += rtap->rta_len;
|
||||
}
|
||||
}
|
||||
|
||||
if(via) {
|
||||
rtap = (struct rtattr *)(((char*)rtap)+rtap->rta_len);
|
||||
rtap->rta_type = RTA_GATEWAY;
|
||||
if(via.isV4()) {
|
||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(struct in_addr);
|
||||
memcpy((char*)rtap+sizeof(struct rtattr), &((struct sockaddr_in*)&via)->sin_addr, sizeof(struct in_addr));
|
||||
} else {
|
||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(struct in6_addr);
|
||||
memcpy((char*)rtap+sizeof(struct rtattr), &((struct sockaddr_in6*)&via)->sin6_addr, sizeof(struct in6_addr));
|
||||
}
|
||||
rtl += rtap->rta_len;
|
||||
}
|
||||
|
||||
req.nl.nlmsg_len = NLMSG_LENGTH(rtl);
|
||||
req.nl.nlmsg_flags = NLM_F_REQUEST;
|
||||
req.nl.nlmsg_type = RTM_DELROUTE;
|
||||
req.nl.nlmsg_pid = 0;
|
||||
req.nl.nlmsg_seq = ++_seq;
|
||||
req.rt.rtm_family = target.ss_family;
|
||||
req.rt.rtm_table = RT_TABLE_MAIN;
|
||||
req.rt.rtm_protocol = RTPROT_STATIC;
|
||||
req.rt.rtm_scope = RT_SCOPE_UNIVERSE;
|
||||
req.rt.rtm_type = RTN_UNICAST;
|
||||
req.rt.rtm_dst_len = target.netmaskBits();
|
||||
|
||||
struct sockaddr_nl pa;
|
||||
bzero(&pa, sizeof(pa));
|
||||
pa.nl_family = AF_NETLINK;
|
||||
|
||||
struct msghdr msg;
|
||||
bzero(&msg, sizeof(msg));
|
||||
msg.msg_name = (void*)&pa;
|
||||
msg.msg_namelen = sizeof(pa);
|
||||
|
||||
struct iovec iov;
|
||||
bzero(&iov, sizeof(iov));
|
||||
iov.iov_base = (void*)&req.nl;
|
||||
iov.iov_len = req.nl.nlmsg_len;
|
||||
msg.msg_iov = &iov;
|
||||
msg.msg_iovlen = 1;
|
||||
sendmsg(_fd, &msg, 0);
|
||||
}
|
||||
|
||||
void LinuxNetLink::addInterface(const char *iface, unsigned int mtu)
|
||||
|
|
|
|||
|
|
@ -33,10 +33,11 @@
|
|||
#include <asm/types.h>
|
||||
#include <linux/rtnetlink.h>
|
||||
#include <sys/socket.h>
|
||||
|
||||
#include <linux/if.h>
|
||||
|
||||
#include "../node/InetAddress.hpp"
|
||||
#include "Thread.hpp"
|
||||
#include "../node/Hashtable.hpp"
|
||||
|
||||
|
||||
namespace ZeroTier {
|
||||
|
|
@ -44,17 +45,11 @@ namespace ZeroTier {
|
|||
struct route_entry {
|
||||
InetAddress target;
|
||||
InetAddress via;
|
||||
const char *iface;
|
||||
int if_index;
|
||||
char iface[IFNAMSIZ];
|
||||
};
|
||||
|
||||
typedef std::vector<route_entry> RouteList;
|
||||
|
||||
struct nl_req {
|
||||
struct nlmsghdr nl;
|
||||
struct rtmsg rt;
|
||||
char buf[8192];
|
||||
};
|
||||
|
||||
/**
|
||||
* Interface with Linux's RTNETLINK
|
||||
*/
|
||||
|
|
@ -93,18 +88,27 @@ private:
|
|||
void _ipAddressAdded(struct nlmsghdr *nlp);
|
||||
void _ipAddressDeleted(struct nlmsghdr *nlp);
|
||||
|
||||
|
||||
void _requestInterfaceList();
|
||||
void _requestIPv4Routes();
|
||||
void _requestIPv6Routes();
|
||||
|
||||
|
||||
Thread _t;
|
||||
bool _running;
|
||||
|
||||
RouteList _routes_ipv4;
|
||||
RouteList _routes_ipv6;
|
||||
|
||||
uint32_t _seq;
|
||||
|
||||
struct iface_entry {
|
||||
int index;
|
||||
char ifacename[IFNAMSIZ];
|
||||
char mac[18];
|
||||
unsigned int mtu;
|
||||
};
|
||||
Hashtable<int, iface_entry> _interfaces;
|
||||
|
||||
// socket communication vars;
|
||||
int _fd;
|
||||
struct sockaddr_nl _la;
|
||||
|
|
|
|||
|
|
@ -54,6 +54,7 @@
|
|||
#include <asm/types.h>
|
||||
#include <linux/rtnetlink.h>
|
||||
#include <sys/socket.h>
|
||||
#include "../osdep/LinuxNetLink.hpp"
|
||||
#endif
|
||||
#ifdef __BSD__
|
||||
#include <net/if_dl.h>
|
||||
|
|
@ -284,44 +285,14 @@ static void _routeCmd(const char *op,const InetAddress &target,const InetAddress
|
|||
#ifdef __LINUX__ // ----------------------------------------------------------
|
||||
#define ZT_ROUTING_SUPPORT_FOUND 1
|
||||
|
||||
static bool _hasRoute(const InetAddress &target, const InetAddress &via, const char *localInterface)
|
||||
{
|
||||
if (target.ss_family == AF_INET) {
|
||||
int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
|
||||
|
||||
char *buf;
|
||||
int nll;
|
||||
struct rtmsg *rtp;
|
||||
int rtl;
|
||||
struct rtattr *rtap;
|
||||
|
||||
struct sockaddr_nl la;
|
||||
bzero(&la, sizeof(la));
|
||||
la.nl_family = AF_NETLINK;
|
||||
la.nl_pad = 0;
|
||||
la.nl_pid = (uint32_t)((ptrdiff_t)&target % getpid());
|
||||
la.nl_groups = 0;
|
||||
int rtn = bind(fd, (struct sockaddr*)&la, sizeof(la));
|
||||
|
||||
|
||||
|
||||
close(fd);
|
||||
return false;
|
||||
} else {
|
||||
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
static void _routeCmd(const char *op, const InetAddress &target, const InetAddress &via, const char *localInterface)
|
||||
{
|
||||
bool hasRoute = _hasRoute(target, via, localInterface);
|
||||
if (hasRoute && (strcmp(op, "add") == 0 || strcmp(op, "replace") == 0)) {
|
||||
return;
|
||||
} else if (!hasRoute && (strcmp(op, "remove") == 0 || strcmp(op, "del") == 0)) {
|
||||
return;
|
||||
if ((strcmp(op, "add") == 0 || strcmp(op, "replace") == 0)) {
|
||||
LinuxNetLink::getInstance().addRoute(target, via, localInterface);
|
||||
} else if ((strcmp(op, "remove") == 0 || strcmp(op, "del") == 0)) {
|
||||
LinuxNetLink::getInstance().delRoute(target, via, localInterface);
|
||||
}
|
||||
return;
|
||||
|
||||
char targetStr[64] = {0};
|
||||
char viaStr[64] = {0};
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue