mirror of
https://github.com/zerotier/ZeroTierOne.git
synced 2025-06-05 03:53:44 +02:00
rtnetlink integrated. no more callouts to iproute2
This commit is contained in:
parent
62210e57f1
commit
b22405b64b
7 changed files with 566 additions and 243 deletions
|
@ -19,7 +19,7 @@ include objects.mk
|
||||||
ONE_OBJS+=osdep/LinuxEthernetTap.o
|
ONE_OBJS+=osdep/LinuxEthernetTap.o
|
||||||
ONE_OBJS+=osdep/LinuxNetLink.o
|
ONE_OBJS+=osdep/LinuxNetLink.o
|
||||||
|
|
||||||
NLTEST_OBJS+=osdep/LinuxNetLink.o
|
NLTEST_OBJS+=osdep/LinuxNetLink.o node/InetAddress.o node/Utils.o node/Salsa20.o
|
||||||
NLTEST_OBJS+=nltest.o
|
NLTEST_OBJS+=nltest.o
|
||||||
|
|
||||||
# Auto-detect miniupnpc and nat-pmp as well and use system libs if present,
|
# Auto-detect miniupnpc and nat-pmp as well and use system libs if present,
|
||||||
|
|
|
@ -56,6 +56,7 @@
|
||||||
#include "../node/Dictionary.hpp"
|
#include "../node/Dictionary.hpp"
|
||||||
#include "OSUtils.hpp"
|
#include "OSUtils.hpp"
|
||||||
#include "LinuxEthernetTap.hpp"
|
#include "LinuxEthernetTap.hpp"
|
||||||
|
#include "LinuxNetLink.hpp"
|
||||||
|
|
||||||
// ff:ff:ff:ff:ff:ff with no ADI
|
// ff:ff:ff:ff:ff:ff with no ADI
|
||||||
static const ZeroTier::MulticastGroup _blindWildcardMulticastGroup(ZeroTier::MAC(0xff),0);
|
static const ZeroTier::MulticastGroup _blindWildcardMulticastGroup(ZeroTier::MAC(0xff),0);
|
||||||
|
@ -97,6 +98,9 @@ LinuxEthernetTap::LinuxEthernetTap(
|
||||||
char procpath[128],nwids[32];
|
char procpath[128],nwids[32];
|
||||||
struct stat sbuf;
|
struct stat sbuf;
|
||||||
|
|
||||||
|
// ensure netlink connection is started
|
||||||
|
(void)LinuxNetLink::getInstance();
|
||||||
|
|
||||||
OSUtils::ztsnprintf(nwids,sizeof(nwids),"%.16llx",nwid);
|
OSUtils::ztsnprintf(nwids,sizeof(nwids),"%.16llx",nwid);
|
||||||
|
|
||||||
Mutex::Lock _l(__tapCreateLock); // create only one tap at a time, globally
|
Mutex::Lock _l(__tapCreateLock); // create only one tap at a time, globally
|
||||||
|
@ -263,18 +267,8 @@ bool LinuxEthernetTap::enabled() const
|
||||||
|
|
||||||
static bool ___removeIp(const std::string &_dev,const InetAddress &ip)
|
static bool ___removeIp(const std::string &_dev,const InetAddress &ip)
|
||||||
{
|
{
|
||||||
long cpid = (long)vfork();
|
LinuxNetLink::getInstance().removeAddress(ip, _dev.c_str());
|
||||||
if (cpid == 0) {
|
return true;
|
||||||
OSUtils::redirectUnixOutputs("/dev/null",(const char *)0);
|
|
||||||
setenv("PATH", "/sbin:/bin:/usr/sbin:/usr/bin", 1);
|
|
||||||
char iptmp[128];
|
|
||||||
::execlp("ip","ip","addr","del",ip.toString(iptmp),"dev",_dev.c_str(),(const char *)0);
|
|
||||||
::_exit(-1);
|
|
||||||
} else {
|
|
||||||
int exitcode = -1;
|
|
||||||
::waitpid(cpid,&exitcode,0);
|
|
||||||
return (exitcode == 0);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#ifdef __SYNOLOGY__
|
#ifdef __SYNOLOGY__
|
||||||
|
@ -285,49 +279,32 @@ bool LinuxEthernetTap::addIpSyn(std::vector<InetAddress> ips)
|
||||||
std::string cfg_contents = "DEVICE="+_dev+"\nBOOTPROTO=static";
|
std::string cfg_contents = "DEVICE="+_dev+"\nBOOTPROTO=static";
|
||||||
int ip4=0,ip6=0,ip4_tot=0,ip6_tot=0;
|
int ip4=0,ip6=0,ip4_tot=0,ip6_tot=0;
|
||||||
|
|
||||||
long cpid = (long)vfork();
|
for(int i=0; i<(int)ips.size(); i++) {
|
||||||
if (cpid == 0) {
|
if (ips[i].isV4())
|
||||||
OSUtils::redirectUnixOutputs("/dev/null",(const char *)0);
|
ip4_tot++;
|
||||||
setenv("PATH", "/sbin:/bin:/usr/sbin:/usr/bin", 1);
|
else
|
||||||
// We must know if there is at least (one) of each protocol version so we
|
ip6_tot++;
|
||||||
// can properly enumerate address/netmask combinations in the ifcfg-dev file
|
}
|
||||||
for(int i=0; i<(int)ips.size(); i++) {
|
// Assemble and write contents of ifcfg-dev file
|
||||||
if (ips[i].isV4())
|
for(int i=0; i<(int)ips.size(); i++) {
|
||||||
ip4_tot++;
|
if (ips[i].isV4()) {
|
||||||
else
|
char iptmp[64],iptmp2[64];
|
||||||
ip6_tot++;
|
std::string numstr4 = ip4_tot > 1 ? std::to_string(ip4) : "";
|
||||||
|
cfg_contents += "\nIPADDR"+numstr4+"="+ips[i].toIpString(iptmp)
|
||||||
|
+ "\nNETMASK"+numstr4+"="+ips[i].netmask().toIpString(iptmp2)+"\n";
|
||||||
|
ip4++;
|
||||||
|
} else {
|
||||||
|
char iptmp[64],iptmp2[64];
|
||||||
|
std::string numstr6 = ip6_tot > 1 ? std::to_string(ip6) : "";
|
||||||
|
cfg_contents += "\nIPV6ADDR"+numstr6+"="+ips[i].toIpString(iptmp)
|
||||||
|
+ "\nNETMASK"+numstr6+"="+ips[i].netmask().toIpString(iptmp2)+"\n";
|
||||||
|
ip6++;
|
||||||
}
|
}
|
||||||
// Assemble and write contents of ifcfg-dev file
|
}
|
||||||
for(int i=0; i<(int)ips.size(); i++) {
|
OSUtils::writeFile(filepath.c_str(), cfg_contents.c_str(), cfg_contents.length());
|
||||||
if (ips[i].isV4()) {
|
// Finaly, add IPs
|
||||||
char iptmp[64],iptmp2[64];
|
for(int i=0; i<(int)ips.size(); i++){
|
||||||
std::string numstr4 = ip4_tot > 1 ? std::to_string(ip4) : "";
|
LinuxNetLink::getInstance().addAddress(ips[i], _dev.c_str());
|
||||||
cfg_contents += "\nIPADDR"+numstr4+"="+ips[i].toIpString(iptmp)
|
|
||||||
+ "\nNETMASK"+numstr4+"="+ips[i].netmask().toIpString(iptmp2)+"\n";
|
|
||||||
ip4++;
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
char iptmp[64],iptmp2[64];
|
|
||||||
std::string numstr6 = ip6_tot > 1 ? std::to_string(ip6) : "";
|
|
||||||
cfg_contents += "\nIPV6ADDR"+numstr6+"="+ips[i].toIpString(iptmp)
|
|
||||||
+ "\nNETMASK"+numstr6+"="+ips[i].netmask().toIpString(iptmp2)+"\n";
|
|
||||||
ip6++;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
OSUtils::writeFile(filepath.c_str(), cfg_contents.c_str(), cfg_contents.length());
|
|
||||||
// Finaly, add IPs
|
|
||||||
for(int i=0; i<(int)ips.size(); i++){
|
|
||||||
char iptmp[128],iptmp2[128];
|
|
||||||
if (ips[i].isV4())
|
|
||||||
::execlp("ip","ip","addr","add",ips[i].toString(iptmp),"broadcast",ips[i].broadcast().toIpString(iptmp2),"dev",_dev.c_str(),(const char *)0);
|
|
||||||
else
|
|
||||||
::execlp("ip","ip","addr","add",ips[i].toString(iptmp),"dev",_dev.c_str(),(const char *)0);
|
|
||||||
}
|
|
||||||
::_exit(-1);
|
|
||||||
} else if (cpid > 0) {
|
|
||||||
int exitcode = -1;
|
|
||||||
::waitpid(cpid,&exitcode,0);
|
|
||||||
return (exitcode == 0);
|
|
||||||
}
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
@ -348,24 +325,9 @@ bool LinuxEthernetTap::addIp(const InetAddress &ip)
|
||||||
___removeIp(_dev,*i);
|
___removeIp(_dev,*i);
|
||||||
}
|
}
|
||||||
|
|
||||||
long cpid = (long)vfork();
|
LinuxNetLink::getInstance().addAddress(ip, _dev.c_str());
|
||||||
if (cpid == 0) {
|
|
||||||
OSUtils::redirectUnixOutputs("/dev/null",(const char *)0);
|
|
||||||
setenv("PATH", "/sbin:/bin:/usr/sbin:/usr/bin", 1);
|
|
||||||
char iptmp[128],iptmp2[128];
|
|
||||||
if (ip.isV4()) {
|
|
||||||
::execlp("ip","ip","addr","add",ip.toString(iptmp),"broadcast",ip.broadcast().toIpString(iptmp2),"dev",_dev.c_str(),(const char *)0);
|
|
||||||
} else {
|
|
||||||
::execlp("ip","ip","addr","add",ip.toString(iptmp),"dev",_dev.c_str(),(const char *)0);
|
|
||||||
}
|
|
||||||
::_exit(-1);
|
|
||||||
} else if (cpid > 0) {
|
|
||||||
int exitcode = -1;
|
|
||||||
::waitpid(cpid,&exitcode,0);
|
|
||||||
return (exitcode == 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
return false;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool LinuxEthernetTap::removeIp(const InetAddress &ip)
|
bool LinuxEthernetTap::removeIp(const InetAddress &ip)
|
||||||
|
|
|
@ -54,39 +54,36 @@ LinuxNetLink::LinuxNetLink()
|
||||||
: _t()
|
: _t()
|
||||||
, _running(false)
|
, _running(false)
|
||||||
, _routes_ipv4()
|
, _routes_ipv4()
|
||||||
|
, _rv4_m()
|
||||||
, _routes_ipv6()
|
, _routes_ipv6()
|
||||||
|
, _rv6_m()
|
||||||
, _seq(0)
|
, _seq(0)
|
||||||
|
, _interfaces()
|
||||||
|
, _if_m()
|
||||||
, _fd(socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE))
|
, _fd(socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE))
|
||||||
, _la({0})
|
, _la({0})
|
||||||
{
|
{
|
||||||
|
|
||||||
// set socket timeout to 1 sec so we're not permablocking recv() calls
|
// set socket timeout to 1 sec so we're not permablocking recv() calls
|
||||||
struct timeval tv;
|
_setSocketTimeout(_fd, 1);
|
||||||
tv.tv_sec = 1;
|
|
||||||
tv.tv_usec = 0;
|
|
||||||
if(setsockopt(_fd, SOL_SOCKET, SO_RCVTIMEO, (const char*)&tv, sizeof(tv)) != 0) {
|
|
||||||
fprintf(stderr, "setsockopt failed: %s\n", strerror(errno));
|
|
||||||
}
|
|
||||||
|
|
||||||
_la.nl_family = AF_NETLINK;
|
_la.nl_family = AF_NETLINK;
|
||||||
_la.nl_pid = getpid();
|
_la.nl_pid = getpid()+1;
|
||||||
_la.nl_groups = RTMGRP_LINK|RTMGRP_IPV4_IFADDR|RTMGRP_IPV6_IFADDR|RTMGRP_IPV4_ROUTE|RTMGRP_IPV6_ROUTE|RTMGRP_NOTIFY;
|
_la.nl_groups = RTMGRP_LINK|RTMGRP_IPV4_IFADDR|RTMGRP_IPV6_IFADDR|RTMGRP_IPV4_ROUTE|RTMGRP_IPV6_ROUTE|RTMGRP_NOTIFY;
|
||||||
if (bind(_fd, (struct sockaddr*)&_la, sizeof(_la))) {
|
if (bind(_fd, (struct sockaddr*)&_la, sizeof(_la))) {
|
||||||
fprintf(stderr, "Error connecting to RTNETLINK: %s\n", strerror(errno));
|
fprintf(stderr, "Error connecting to RTNETLINK: %s\n", strerror(errno));
|
||||||
::exit(1);
|
::exit(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
_running = true;
|
|
||||||
_t = Thread::start(this);
|
|
||||||
|
|
||||||
fprintf(stderr, "Requesting IPV4 Routes\n");
|
fprintf(stderr, "Requesting IPV4 Routes\n");
|
||||||
_requestIPv4Routes();
|
_requestIPv4Routes();
|
||||||
Thread::sleep(10);
|
|
||||||
fprintf(stderr, "Requesting IPV6 Routes\n");
|
fprintf(stderr, "Requesting IPV6 Routes\n");
|
||||||
_requestIPv6Routes();
|
_requestIPv6Routes();
|
||||||
Thread::sleep(10);
|
|
||||||
fprintf(stderr, "Requesting Interface List\n");
|
fprintf(stderr, "Requesting Interface List\n");
|
||||||
_requestInterfaceList();
|
_requestInterfaceList();
|
||||||
|
|
||||||
|
_running = true;
|
||||||
|
_t = Thread::start(this);
|
||||||
}
|
}
|
||||||
|
|
||||||
LinuxNetLink::~LinuxNetLink()
|
LinuxNetLink::~LinuxNetLink()
|
||||||
|
@ -97,7 +94,17 @@ LinuxNetLink::~LinuxNetLink()
|
||||||
::close(_fd);
|
::close(_fd);
|
||||||
}
|
}
|
||||||
|
|
||||||
void LinuxNetLink::threadMain() throw()
|
void LinuxNetLink::_setSocketTimeout(int fd, int seconds)
|
||||||
|
{
|
||||||
|
struct timeval tv;
|
||||||
|
tv.tv_sec = seconds;
|
||||||
|
tv.tv_usec = 0;
|
||||||
|
if(setsockopt(fd, SOL_SOCKET, SO_RCVTIMEO, (const char*)&tv, sizeof(tv)) != 0) {
|
||||||
|
fprintf(stderr, "setsockopt failed: %s\n", strerror(errno));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int LinuxNetLink::_doRecv(int fd)
|
||||||
{
|
{
|
||||||
char buf[8192];
|
char buf[8192];
|
||||||
char *p = NULL;
|
char *p = NULL;
|
||||||
|
@ -106,26 +113,24 @@ void LinuxNetLink::threadMain() throw()
|
||||||
int rtn = 0;
|
int rtn = 0;
|
||||||
p = buf;
|
p = buf;
|
||||||
|
|
||||||
while(_running) {
|
while(true) {
|
||||||
rtn = recv(_fd, p, sizeof(buf) - nll, 0);
|
rtn = recv(fd, p, sizeof(buf) - nll, 0);
|
||||||
|
|
||||||
if (rtn > 0) {
|
if (rtn > 0) {
|
||||||
nlp = (struct nlmsghdr *)p;
|
nlp = (struct nlmsghdr *)p;
|
||||||
|
|
||||||
if(nlp->nlmsg_type == NLMSG_ERROR && (nlp->nlmsg_flags & NLM_F_ACK) != NLM_F_ACK) {
|
if(nlp->nlmsg_type == NLMSG_ERROR && (nlp->nlmsg_flags & NLM_F_ACK) != NLM_F_ACK) {
|
||||||
fprintf(stderr, "NLMSG_ERROR\n");
|
|
||||||
struct nlmsgerr *err = (struct nlmsgerr*)NLMSG_DATA(nlp);
|
struct nlmsgerr *err = (struct nlmsgerr*)NLMSG_DATA(nlp);
|
||||||
if (err->error != 0) {
|
if (err->error != 0) {
|
||||||
fprintf(stderr, "rtnetlink error: %s\n", strerror(-(err->error)));
|
fprintf(stderr, "rtnetlink error: %s\n", strerror(-(err->error)));
|
||||||
}
|
}
|
||||||
p = buf;
|
p = buf;
|
||||||
nll = 0;
|
nll = 0;
|
||||||
continue;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (nlp->nlmsg_type == NLMSG_NOOP) {
|
if (nlp->nlmsg_type == NLMSG_NOOP) {
|
||||||
fprintf(stderr, "noop\n");
|
break;
|
||||||
continue;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if( (nlp->nlmsg_flags & NLM_F_MULTI) == NLM_F_MULTI || (nlp->nlmsg_type == NLMSG_DONE))
|
if( (nlp->nlmsg_flags & NLM_F_MULTI) == NLM_F_MULTI || (nlp->nlmsg_type == NLMSG_DONE))
|
||||||
|
@ -134,7 +139,7 @@ void LinuxNetLink::threadMain() throw()
|
||||||
_processMessage(nlp, nll);
|
_processMessage(nlp, nll);
|
||||||
p = buf;
|
p = buf;
|
||||||
nll = 0;
|
nll = 0;
|
||||||
continue;
|
break;
|
||||||
}
|
}
|
||||||
p += rtn;
|
p += rtn;
|
||||||
nll += rtn;
|
nll += rtn;
|
||||||
|
@ -144,7 +149,7 @@ void LinuxNetLink::threadMain() throw()
|
||||||
fprintf(stderr, "NLMSG_OVERRUN: Data lost\n");
|
fprintf(stderr, "NLMSG_OVERRUN: Data lost\n");
|
||||||
p = buf;
|
p = buf;
|
||||||
nll = 0;
|
nll = 0;
|
||||||
continue;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
nll += rtn;
|
nll += rtn;
|
||||||
|
@ -152,8 +157,21 @@ void LinuxNetLink::threadMain() throw()
|
||||||
_processMessage(nlp, nll);
|
_processMessage(nlp, nll);
|
||||||
p = buf;
|
p = buf;
|
||||||
nll = 0;
|
nll = 0;
|
||||||
|
break;
|
||||||
|
} else {
|
||||||
|
break;
|
||||||
}
|
}
|
||||||
else {
|
}
|
||||||
|
return rtn;
|
||||||
|
}
|
||||||
|
|
||||||
|
void LinuxNetLink::threadMain() throw()
|
||||||
|
{
|
||||||
|
int rtn = 0;
|
||||||
|
|
||||||
|
while(_running) {
|
||||||
|
rtn = _doRecv(_fd);
|
||||||
|
if (rtn <= 0) {
|
||||||
Thread::sleep(100);
|
Thread::sleep(100);
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
@ -258,6 +276,7 @@ void LinuxNetLink::_routeAdded(struct nlmsghdr *nlp)
|
||||||
{
|
{
|
||||||
char dsts[40] = {0};
|
char dsts[40] = {0};
|
||||||
char gws[40] = {0};
|
char gws[40] = {0};
|
||||||
|
char srcs[40] = {0};
|
||||||
char ifs[16] = {0};
|
char ifs[16] = {0};
|
||||||
char ms[24] = {0};
|
char ms[24] = {0};
|
||||||
|
|
||||||
|
@ -272,6 +291,9 @@ void LinuxNetLink::_routeAdded(struct nlmsghdr *nlp)
|
||||||
case RTA_DST:
|
case RTA_DST:
|
||||||
inet_ntop(rtp->rtm_family, RTA_DATA(rtap), dsts, rtp->rtm_family == AF_INET ? 24 : 40);
|
inet_ntop(rtp->rtm_family, RTA_DATA(rtap), dsts, rtp->rtm_family == AF_INET ? 24 : 40);
|
||||||
break;
|
break;
|
||||||
|
case RTA_SRC:
|
||||||
|
inet_ntop(rtp->rtm_family, RTA_DATA(rtap), srcs, rtp->rtm_family == AF_INET ? 24: 40);
|
||||||
|
break;
|
||||||
case RTA_GATEWAY:
|
case RTA_GATEWAY:
|
||||||
inet_ntop(rtp->rtm_family, RTA_DATA(rtap), gws, rtp->rtm_family == AF_INET ? 24 : 40);
|
inet_ntop(rtp->rtm_family, RTA_DATA(rtap), gws, rtp->rtm_family == AF_INET ? 24 : 40);
|
||||||
break;
|
break;
|
||||||
|
@ -282,13 +304,14 @@ void LinuxNetLink::_routeAdded(struct nlmsghdr *nlp)
|
||||||
}
|
}
|
||||||
sprintf(ms, "%d", rtp->rtm_dst_len);
|
sprintf(ms, "%d", rtp->rtm_dst_len);
|
||||||
|
|
||||||
fprintf(stderr, "Route Added: dst %s/%s gw %s if %s\n", dsts, ms, gws, ifs);
|
fprintf(stderr, "Route Added: dst %s/%s gw %s src %s if %s\n", dsts, ms, gws, srcs, ifs);
|
||||||
}
|
}
|
||||||
|
|
||||||
void LinuxNetLink::_routeDeleted(struct nlmsghdr *nlp)
|
void LinuxNetLink::_routeDeleted(struct nlmsghdr *nlp)
|
||||||
{
|
{
|
||||||
char dsts[40] = {0};
|
char dsts[40] = {0};
|
||||||
char gws[40] = {0};
|
char gws[40] = {0};
|
||||||
|
char srcs[40] = {0};
|
||||||
char ifs[16] = {0};
|
char ifs[16] = {0};
|
||||||
char ms[24] = {0};
|
char ms[24] = {0};
|
||||||
|
|
||||||
|
@ -303,6 +326,9 @@ void LinuxNetLink::_routeDeleted(struct nlmsghdr *nlp)
|
||||||
case RTA_DST:
|
case RTA_DST:
|
||||||
inet_ntop(rtp->rtm_family, RTA_DATA(rtap), dsts, rtp->rtm_family == AF_INET ? 24 : 40);
|
inet_ntop(rtp->rtm_family, RTA_DATA(rtap), dsts, rtp->rtm_family == AF_INET ? 24 : 40);
|
||||||
break;
|
break;
|
||||||
|
case RTA_SRC:
|
||||||
|
inet_ntop(rtp->rtm_family, RTA_DATA(rtap), srcs, rtp->rtm_family == AF_INET ? 24 : 40);
|
||||||
|
break;
|
||||||
case RTA_GATEWAY:
|
case RTA_GATEWAY:
|
||||||
inet_ntop(rtp->rtm_family, RTA_DATA(rtap), gws, rtp->rtm_family == AF_INET ? 24 : 40);
|
inet_ntop(rtp->rtm_family, RTA_DATA(rtap), gws, rtp->rtm_family == AF_INET ? 24 : 40);
|
||||||
break;
|
break;
|
||||||
|
@ -313,7 +339,7 @@ void LinuxNetLink::_routeDeleted(struct nlmsghdr *nlp)
|
||||||
}
|
}
|
||||||
sprintf(ms, "%d", rtp->rtm_dst_len);
|
sprintf(ms, "%d", rtp->rtm_dst_len);
|
||||||
|
|
||||||
fprintf(stderr, "Route Deleted: dst %s/%s gw %s if %s\n", dsts, ms, gws, ifs);
|
fprintf(stderr, "Route Deleted: dst %s/%s gw %s src %s if %s\n", dsts, ms, gws, srcs, ifs);
|
||||||
}
|
}
|
||||||
|
|
||||||
void LinuxNetLink::_linkAdded(struct nlmsghdr *nlp)
|
void LinuxNetLink::_linkAdded(struct nlmsghdr *nlp)
|
||||||
|
@ -348,12 +374,15 @@ void LinuxNetLink::_linkAdded(struct nlmsghdr *nlp)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
struct iface_entry &entry = _interfaces[ifip->ifi_index];
|
{
|
||||||
entry.index = ifip->ifi_index;
|
Mutex::Lock l(_if_m);
|
||||||
memcpy(entry.ifacename, ifname, sizeof(ifname));
|
struct iface_entry &entry = _interfaces[ifip->ifi_index];
|
||||||
memcpy(entry.mac, mac, sizeof(mac));
|
entry.index = ifip->ifi_index;
|
||||||
memcpy(entry.mac_bin, mac_bin, 6);
|
memcpy(entry.ifacename, ifname, sizeof(ifname));
|
||||||
entry.mtu = mtu;
|
memcpy(entry.mac, mac, sizeof(mac));
|
||||||
|
memcpy(entry.mac_bin, mac_bin, 6);
|
||||||
|
entry.mtu = mtu;
|
||||||
|
}
|
||||||
|
|
||||||
fprintf(stderr, "Link Added: %s mac: %s, mtu: %d\n", ifname, mac, mtu);
|
fprintf(stderr, "Link Added: %s mac: %s, mtu: %d\n", ifname, mac, mtu);
|
||||||
}
|
}
|
||||||
|
@ -389,13 +418,33 @@ void LinuxNetLink::_linkDeleted(struct nlmsghdr *nlp)
|
||||||
}
|
}
|
||||||
|
|
||||||
fprintf(stderr, "Link Deleted: %s mac: %s, mtu: %d\n", ifname, mac, mtu);
|
fprintf(stderr, "Link Deleted: %s mac: %s, mtu: %d\n", ifname, mac, mtu);
|
||||||
if(_interfaces.contains(ifip->ifi_index)) {
|
{
|
||||||
_interfaces.erase(ifip->ifi_index);
|
Mutex::Lock l(_if_m);
|
||||||
|
if(_interfaces.contains(ifip->ifi_index)) {
|
||||||
|
_interfaces.erase(ifip->ifi_index);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void LinuxNetLink::_requestIPv4Routes()
|
void LinuxNetLink::_requestIPv4Routes()
|
||||||
{
|
{
|
||||||
|
int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
|
||||||
|
if (fd == -1) {
|
||||||
|
fprintf(stderr, "Error opening RTNETLINK socket: %s\n", strerror(errno));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
_setSocketTimeout(fd);
|
||||||
|
|
||||||
|
struct sockaddr_nl la;
|
||||||
|
la.nl_family = AF_NETLINK;
|
||||||
|
la.nl_pid = getpid();
|
||||||
|
la.nl_groups = RTMGRP_IPV4_ROUTE;
|
||||||
|
if(bind(fd, (struct sockaddr*)&la, sizeof(la))) {
|
||||||
|
fprintf(stderr, "Error binding RTNETLINK: %s\n", strerror(errno));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
struct nl_route_req req;
|
struct nl_route_req req;
|
||||||
bzero(&req, sizeof(req));
|
bzero(&req, sizeof(req));
|
||||||
req.nl.nlmsg_len = NLMSG_LENGTH(sizeof(struct rtmsg));
|
req.nl.nlmsg_len = NLMSG_LENGTH(sizeof(struct rtmsg));
|
||||||
|
@ -422,11 +471,32 @@ void LinuxNetLink::_requestIPv4Routes()
|
||||||
msg.msg_iov = &iov;
|
msg.msg_iov = &iov;
|
||||||
msg.msg_iovlen = 1;
|
msg.msg_iovlen = 1;
|
||||||
|
|
||||||
sendmsg(_fd, &msg, 0);
|
sendmsg(fd, &msg, 0);
|
||||||
|
|
||||||
|
_doRecv(fd);
|
||||||
|
|
||||||
|
close(fd);
|
||||||
}
|
}
|
||||||
|
|
||||||
void LinuxNetLink::_requestIPv6Routes()
|
void LinuxNetLink::_requestIPv6Routes()
|
||||||
{
|
{
|
||||||
|
int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
|
||||||
|
if (fd == -1) {
|
||||||
|
fprintf(stderr, "Error opening RTNETLINK socket: %s\n", strerror(errno));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
_setSocketTimeout(fd);
|
||||||
|
|
||||||
|
struct sockaddr_nl la;
|
||||||
|
la.nl_family = AF_NETLINK;
|
||||||
|
la.nl_pid = getpid();
|
||||||
|
la.nl_groups = RTMGRP_IPV6_ROUTE;
|
||||||
|
if(bind(fd, (struct sockaddr*)&la, sizeof(struct sockaddr_nl))) {
|
||||||
|
fprintf(stderr, "Error binding RTNETLINK: %s\n", strerror(errno));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
struct nl_route_req req;
|
struct nl_route_req req;
|
||||||
bzero(&req, sizeof(req));
|
bzero(&req, sizeof(req));
|
||||||
req.nl.nlmsg_len = NLMSG_LENGTH(sizeof(struct rtmsg));
|
req.nl.nlmsg_len = NLMSG_LENGTH(sizeof(struct rtmsg));
|
||||||
|
@ -453,11 +523,32 @@ void LinuxNetLink::_requestIPv6Routes()
|
||||||
msg.msg_iov = &iov;
|
msg.msg_iov = &iov;
|
||||||
msg.msg_iovlen = 1;
|
msg.msg_iovlen = 1;
|
||||||
|
|
||||||
sendmsg(_fd, &msg, 0);
|
sendmsg(fd, &msg, 0);
|
||||||
|
|
||||||
|
_doRecv(fd);
|
||||||
|
|
||||||
|
close(fd);
|
||||||
}
|
}
|
||||||
|
|
||||||
void LinuxNetLink::_requestInterfaceList()
|
void LinuxNetLink::_requestInterfaceList()
|
||||||
{
|
{
|
||||||
|
int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
|
||||||
|
if (fd == -1) {
|
||||||
|
fprintf(stderr, "Error opening RTNETLINK socket: %s\n", strerror(errno));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
_setSocketTimeout(fd);
|
||||||
|
|
||||||
|
struct sockaddr_nl la;
|
||||||
|
la.nl_family = AF_NETLINK;
|
||||||
|
la.nl_pid = getpid();
|
||||||
|
la.nl_groups = RTMGRP_LINK;
|
||||||
|
if(bind(fd, (struct sockaddr*)&la, sizeof(struct sockaddr_nl))) {
|
||||||
|
fprintf(stderr, "Error binding RTNETLINK: %s\n", strerror(errno));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
struct nl_if_req req;
|
struct nl_if_req req;
|
||||||
bzero(&req, sizeof(req));
|
bzero(&req, sizeof(req));
|
||||||
req.nl.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifinfomsg));
|
req.nl.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifinfomsg));
|
||||||
|
@ -482,11 +573,43 @@ void LinuxNetLink::_requestInterfaceList()
|
||||||
iov.iov_len = req.nl.nlmsg_len;
|
iov.iov_len = req.nl.nlmsg_len;
|
||||||
msg.msg_iov = &iov;
|
msg.msg_iov = &iov;
|
||||||
msg.msg_iovlen = 1;
|
msg.msg_iovlen = 1;
|
||||||
sendmsg(_fd, &msg, 0);
|
sendmsg(fd, &msg, 0);
|
||||||
|
|
||||||
|
_doRecv(fd);
|
||||||
|
|
||||||
|
close(fd);
|
||||||
}
|
}
|
||||||
|
|
||||||
void LinuxNetLink::addRoute(const InetAddress &target, const InetAddress &via, const char *ifaceName)
|
void LinuxNetLink::addRoute(const InetAddress &target, const InetAddress &via, const InetAddress &src, const char *ifaceName)
|
||||||
{
|
{
|
||||||
|
int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
|
||||||
|
if (fd == -1) {
|
||||||
|
fprintf(stderr, "Error opening RTNETLINK socket: %s\n", strerror(errno));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
_setSocketTimeout(fd);
|
||||||
|
|
||||||
|
struct sockaddr_nl la;
|
||||||
|
bzero(&la, sizeof(la));
|
||||||
|
la.nl_family = AF_NETLINK;
|
||||||
|
la.nl_pid = getpid();
|
||||||
|
|
||||||
|
if(bind(fd, (struct sockaddr*)&la, sizeof(struct sockaddr_nl))) {
|
||||||
|
fprintf(stderr, "Error binding RTNETLINK: %s\n", strerror(errno));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
char tmp[64];
|
||||||
|
char tmp2[64];
|
||||||
|
char tmp3[64];
|
||||||
|
fprintf(stderr, "Adding Route. target: %s via: %s src: %s iface: %s\n", target.toString(tmp), via.toString(tmp2), src.toString(tmp3), ifaceName);
|
||||||
|
|
||||||
|
if(!target) {
|
||||||
|
fprintf(stderr, "Uhhhh adding an empty route?!?!?");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
int rtl = sizeof(struct rtmsg);
|
int rtl = sizeof(struct rtmsg);
|
||||||
struct nl_route_req req;
|
struct nl_route_req req;
|
||||||
bzero(&req, sizeof(req));
|
bzero(&req, sizeof(req));
|
||||||
|
@ -494,49 +617,54 @@ void LinuxNetLink::addRoute(const InetAddress &target, const InetAddress &via, c
|
||||||
struct rtattr *rtap = (struct rtattr *)req.buf;
|
struct rtattr *rtap = (struct rtattr *)req.buf;
|
||||||
rtap->rta_type = RTA_DST;
|
rtap->rta_type = RTA_DST;
|
||||||
if (target.isV4()) {
|
if (target.isV4()) {
|
||||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(struct in_addr);
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
|
||||||
memcpy((void*)((char*)rtap+sizeof(struct rtattr)), &((struct sockaddr_in*)&target)->sin_addr, sizeof(struct in_addr));
|
memcpy(RTA_DATA(rtap), &((struct sockaddr_in*)&target)->sin_addr, sizeof(struct in_addr));
|
||||||
} else {
|
} else {
|
||||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(struct in6_addr);
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in6_addr));
|
||||||
memcpy((void*)((char*)rtap+sizeof(struct rtattr)), &((struct sockaddr_in6*)&target)->sin6_addr, sizeof(struct in6_addr));
|
memcpy(RTA_DATA(rtap), &((struct sockaddr_in6*)&target)->sin6_addr, sizeof(struct in6_addr));
|
||||||
}
|
}
|
||||||
rtl += rtap->rta_len;
|
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) {
|
if(via) {
|
||||||
rtap = (struct rtattr *)(((char*)rtap)+rtap->rta_len);
|
rtap = (struct rtattr *)(((char*)rtap)+rtap->rta_len);
|
||||||
rtap->rta_type = RTA_GATEWAY;
|
rtap->rta_type = RTA_GATEWAY;
|
||||||
if(via.isV4()) {
|
if(via.isV4()) {
|
||||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(struct in_addr);
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
|
||||||
memcpy((char*)rtap+sizeof(struct rtattr), &((struct sockaddr_in*)&via)->sin_addr, sizeof(struct in_addr));
|
memcpy(RTA_DATA(rtap), &((struct sockaddr_in*)&via)->sin_addr, sizeof(struct in_addr));
|
||||||
} else {
|
} else {
|
||||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(struct in6_addr);
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in6_addr));
|
||||||
memcpy((char*)rtap+sizeof(struct rtattr), &((struct sockaddr_in6*)&via)->sin6_addr, sizeof(struct in6_addr));
|
memcpy(RTA_DATA(rtap), &((struct sockaddr_in6*)&via)->sin6_addr, sizeof(struct in6_addr));
|
||||||
}
|
}
|
||||||
rtl += rtap->rta_len;
|
rtl += rtap->rta_len;
|
||||||
|
} else if (src) {
|
||||||
|
rtap = (struct rtattr *)(((char*)rtap)+rtap->rta_len);
|
||||||
|
rtap->rta_type = RTA_SRC;
|
||||||
|
if(src.isV4()) {
|
||||||
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
|
||||||
|
memcpy(RTA_DATA(rtap), &((struct sockaddr_in*)&src)->sin_addr, sizeof(struct in_addr));
|
||||||
|
|
||||||
|
} else {
|
||||||
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in6_addr));
|
||||||
|
memcpy(RTA_DATA(rtap), &((struct sockaddr_in6*)&src)->sin6_addr, sizeof(struct in6_addr));
|
||||||
|
}
|
||||||
|
req.rt.rtm_src_len = src.netmaskBits();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (ifaceName != NULL) {
|
||||||
|
int interface_index = _indexForInterface(ifaceName);
|
||||||
|
if (interface_index != -1) {
|
||||||
|
rtap = (struct rtattr *) (((char*)rtap) + rtap->rta_len);
|
||||||
|
rtap->rta_type = RTA_OIF;
|
||||||
|
rtap->rta_len = RTA_LENGTH(sizeof(int));
|
||||||
|
memcpy(RTA_DATA(rtap), &interface_index, sizeof(int));
|
||||||
|
rtl += rtap->rta_len;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
req.nl.nlmsg_len = NLMSG_LENGTH(rtl);
|
req.nl.nlmsg_len = NLMSG_LENGTH(rtl);
|
||||||
req.nl.nlmsg_flags = NLM_F_REQUEST | NLM_F_CREATE | NLM_F_EXCL;
|
req.nl.nlmsg_flags = NLM_F_REQUEST | NLM_F_EXCL | NLM_F_CREATE | NLM_F_ACK;
|
||||||
req.nl.nlmsg_type = RTM_NEWROUTE;
|
req.nl.nlmsg_type = RTM_NEWROUTE;
|
||||||
req.nl.nlmsg_pid = 0;
|
req.nl.nlmsg_pid = 0;
|
||||||
req.nl.nlmsg_seq = ++_seq;
|
req.nl.nlmsg_seq = ++_seq;
|
||||||
|
@ -546,6 +674,7 @@ void LinuxNetLink::addRoute(const InetAddress &target, const InetAddress &via, c
|
||||||
req.rt.rtm_scope = RT_SCOPE_UNIVERSE;
|
req.rt.rtm_scope = RT_SCOPE_UNIVERSE;
|
||||||
req.rt.rtm_type = RTN_UNICAST;
|
req.rt.rtm_type = RTN_UNICAST;
|
||||||
req.rt.rtm_dst_len = target.netmaskBits();
|
req.rt.rtm_dst_len = target.netmaskBits();
|
||||||
|
req.rt.rtm_flags = 0;
|
||||||
|
|
||||||
struct sockaddr_nl pa;
|
struct sockaddr_nl pa;
|
||||||
bzero(&pa, sizeof(pa));
|
bzero(&pa, sizeof(pa));
|
||||||
|
@ -562,11 +691,41 @@ void LinuxNetLink::addRoute(const InetAddress &target, const InetAddress &via, c
|
||||||
iov.iov_len = req.nl.nlmsg_len;
|
iov.iov_len = req.nl.nlmsg_len;
|
||||||
msg.msg_iov = &iov;
|
msg.msg_iov = &iov;
|
||||||
msg.msg_iovlen = 1;
|
msg.msg_iovlen = 1;
|
||||||
sendmsg(_fd, &msg, 0);
|
sendmsg(fd, &msg, 0);
|
||||||
|
|
||||||
|
_doRecv(fd);
|
||||||
|
|
||||||
|
close(fd);
|
||||||
}
|
}
|
||||||
|
|
||||||
void LinuxNetLink::delRoute(const InetAddress &target, const InetAddress &via, const char *ifaceName)
|
void LinuxNetLink::delRoute(const InetAddress &target, const InetAddress &via, const InetAddress &src, const char *ifaceName)
|
||||||
{
|
{
|
||||||
|
int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
|
||||||
|
if (fd == -1) {
|
||||||
|
fprintf(stderr, "Error opening RTNETLINK socket: %s\n", strerror(errno));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
_setSocketTimeout(fd);
|
||||||
|
|
||||||
|
struct sockaddr_nl la;
|
||||||
|
la.nl_family = AF_NETLINK;
|
||||||
|
la.nl_pid = getpid();
|
||||||
|
|
||||||
|
if(bind(fd, (struct sockaddr*)&la, sizeof(struct sockaddr_nl))) {
|
||||||
|
fprintf(stderr, "Error binding RTNETLINK: %s\n", strerror(errno));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
char tmp[64];
|
||||||
|
char tmp2[64];
|
||||||
|
char tmp3[64];
|
||||||
|
fprintf(stderr, "Removing Route. target: %s via: %s src: %s iface: %s\n", target.toString(tmp), via.toString(tmp2), src.toString(tmp3), ifaceName);
|
||||||
|
|
||||||
|
if(!target) {
|
||||||
|
fprintf(stderr, "Uhhhh deleting an empty route?!?!?");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
int rtl = sizeof(struct rtmsg);
|
int rtl = sizeof(struct rtmsg);
|
||||||
struct nl_route_req req;
|
struct nl_route_req req;
|
||||||
bzero(&req, sizeof(req));
|
bzero(&req, sizeof(req));
|
||||||
|
@ -574,47 +733,52 @@ void LinuxNetLink::delRoute(const InetAddress &target, const InetAddress &via, c
|
||||||
struct rtattr *rtap = (struct rtattr *)req.buf;
|
struct rtattr *rtap = (struct rtattr *)req.buf;
|
||||||
rtap->rta_type = RTA_DST;
|
rtap->rta_type = RTA_DST;
|
||||||
if (target.isV4()) {
|
if (target.isV4()) {
|
||||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(struct in_addr);
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
|
||||||
memcpy((void*)((char*)rtap+sizeof(struct rtattr)), &((struct sockaddr_in*)&target)->sin_addr, sizeof(struct in_addr));
|
memcpy(RTA_DATA(rtap), &((struct sockaddr_in*)&target)->sin_addr, sizeof(struct in_addr));
|
||||||
} else {
|
} else {
|
||||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(struct in6_addr);
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in6_addr));
|
||||||
memcpy((void*)((char*)rtap+sizeof(struct rtattr)), &((struct sockaddr_in6*)&target)->sin6_addr, sizeof(struct in6_addr));
|
memcpy(RTA_DATA(rtap), &((struct sockaddr_in6*)&target)->sin6_addr, sizeof(struct in6_addr));
|
||||||
}
|
}
|
||||||
rtl += rtap->rta_len;
|
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) {
|
if(via) {
|
||||||
rtap = (struct rtattr *)(((char*)rtap)+rtap->rta_len);
|
rtap = (struct rtattr *)(((char*)rtap)+rtap->rta_len);
|
||||||
rtap->rta_type = RTA_GATEWAY;
|
rtap->rta_type = RTA_GATEWAY;
|
||||||
if(via.isV4()) {
|
if(via.isV4()) {
|
||||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(struct in_addr);
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
|
||||||
memcpy((char*)rtap+sizeof(struct rtattr), &((struct sockaddr_in*)&via)->sin_addr, sizeof(struct in_addr));
|
memcpy(RTA_DATA(rtap), &((struct sockaddr_in*)&via)->sin_addr, sizeof(struct in_addr));
|
||||||
} else {
|
} else {
|
||||||
rtap->rta_len = sizeof(struct rtattr)+sizeof(struct in6_addr);
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in6_addr));
|
||||||
memcpy((char*)rtap+sizeof(struct rtattr), &((struct sockaddr_in6*)&via)->sin6_addr, sizeof(struct in6_addr));
|
memcpy(RTA_DATA(rtap), &((struct sockaddr_in6*)&via)->sin6_addr, sizeof(struct in6_addr));
|
||||||
}
|
}
|
||||||
rtl += rtap->rta_len;
|
rtl += rtap->rta_len;
|
||||||
|
} else if (src) {
|
||||||
|
rtap = (struct rtattr *)(((char*)rtap)+rtap->rta_len);
|
||||||
|
rtap->rta_type = RTA_SRC;
|
||||||
|
if(src.isV4()) {
|
||||||
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
|
||||||
|
memcpy(RTA_DATA(rtap), &((struct sockaddr_in*)&src)->sin_addr, sizeof(struct in_addr));
|
||||||
|
|
||||||
|
} else {
|
||||||
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in6_addr));
|
||||||
|
memcpy(RTA_DATA(rtap), &((struct sockaddr_in6*)&src)->sin6_addr, sizeof(struct in6_addr));
|
||||||
|
}
|
||||||
|
req.rt.rtm_src_len = src.netmaskBits();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (ifaceName != NULL) {
|
||||||
|
int interface_index = _indexForInterface(ifaceName);
|
||||||
|
if (interface_index != -1) {
|
||||||
|
rtap = (struct rtattr *) (((char*)rtap) + rtap->rta_len);
|
||||||
|
rtap->rta_type = RTA_OIF;
|
||||||
|
rtap->rta_len = RTA_LENGTH(sizeof(int));
|
||||||
|
memcpy(RTA_DATA(rtap), &interface_index, sizeof(int));
|
||||||
|
rtl += rtap->rta_len;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
req.nl.nlmsg_len = NLMSG_LENGTH(rtl);
|
req.nl.nlmsg_len = NLMSG_LENGTH(rtl);
|
||||||
req.nl.nlmsg_flags = NLM_F_REQUEST;
|
req.nl.nlmsg_flags = NLM_F_REQUEST;
|
||||||
req.nl.nlmsg_type = RTM_DELROUTE;
|
req.nl.nlmsg_type = RTM_DELROUTE;
|
||||||
|
@ -626,6 +790,7 @@ void LinuxNetLink::delRoute(const InetAddress &target, const InetAddress &via, c
|
||||||
req.rt.rtm_scope = RT_SCOPE_UNIVERSE;
|
req.rt.rtm_scope = RT_SCOPE_UNIVERSE;
|
||||||
req.rt.rtm_type = RTN_UNICAST;
|
req.rt.rtm_type = RTN_UNICAST;
|
||||||
req.rt.rtm_dst_len = target.netmaskBits();
|
req.rt.rtm_dst_len = target.netmaskBits();
|
||||||
|
req.rt.rtm_flags = 0;
|
||||||
|
|
||||||
struct sockaddr_nl pa;
|
struct sockaddr_nl pa;
|
||||||
bzero(&pa, sizeof(pa));
|
bzero(&pa, sizeof(pa));
|
||||||
|
@ -642,73 +807,226 @@ void LinuxNetLink::delRoute(const InetAddress &target, const InetAddress &via, c
|
||||||
iov.iov_len = req.nl.nlmsg_len;
|
iov.iov_len = req.nl.nlmsg_len;
|
||||||
msg.msg_iov = &iov;
|
msg.msg_iov = &iov;
|
||||||
msg.msg_iovlen = 1;
|
msg.msg_iovlen = 1;
|
||||||
sendmsg(_fd, &msg, 0);
|
sendmsg(fd, &msg, 0);
|
||||||
|
|
||||||
|
_doRecv(fd);
|
||||||
|
|
||||||
|
close(fd);
|
||||||
}
|
}
|
||||||
|
|
||||||
// void LinuxNetLink::addInterface(const char *iface, unsigned int mtu, const MAC &mac)
|
|
||||||
// {
|
|
||||||
// int rtl = sizeof(struct ifinfomsg);
|
|
||||||
// struct nl_if_req req;
|
|
||||||
// bzero(&req, sizeof(nl_if_req));
|
|
||||||
|
|
||||||
// struct rtattr *rtap = (struct rtattr *)req.buf;
|
|
||||||
// rtap->rta_type = IFLA_IFNAME;
|
|
||||||
// rtap->rta_len = sizeof(struct rtattr)+strlen(iface)+1;
|
|
||||||
// rtl += rtap->rta_len;
|
|
||||||
|
|
||||||
// rtap = (struct rtattr*)(((char*)rtap)+rtap->rta_len);
|
|
||||||
// rtap->rta_type = IFLA_MTU;
|
|
||||||
// rtap->rta_len = sizeof(struct rtattr)+sizeof(unsigned int);
|
|
||||||
// rtl += rtap->rta_len;
|
|
||||||
|
|
||||||
// rtap = (struct rtattr*)(((char*)rtap)+rtap->rta_len);
|
|
||||||
// rtap->rta_type = IFLA_ADDRESS;
|
|
||||||
// rtap->rta_len = sizeof(struct rtattr)+6;
|
|
||||||
// mac.copyTo(((char*)rtap)+sizeof(struct rtattr), 6);
|
|
||||||
// rtl += rtap->rta_len;
|
|
||||||
|
|
||||||
// IFLA_LINKINFO;
|
|
||||||
// req.nl.nlmsg_len = NLMSG_LENGTH(rtl);
|
|
||||||
// req.nl.nlmsg_flags = NLM_F_REQUEST | NLM_F_CREATE | NLM_F_EXCL;
|
|
||||||
// req.nl.nlmsg_type = RTM_NEWLINK;
|
|
||||||
// req.nl.nlmsg_pid = 0;
|
|
||||||
// req.nl.nlmsg_seq = ++_seq;
|
|
||||||
|
|
||||||
// req.ifa.ifi_family = AF_UNSPEC;
|
|
||||||
// req.ifa.ifi_type = 0; // TODO figure this one out
|
|
||||||
// req.ifa.ifi_index = 0;
|
|
||||||
// req.ifa.ifi_flags = IFF_UP;
|
|
||||||
|
|
||||||
// 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;
|
|
||||||
// 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::removeInterface(const char *iface)
|
|
||||||
// {
|
|
||||||
|
|
||||||
// }
|
|
||||||
|
|
||||||
void LinuxNetLink::addAddress(const InetAddress &addr, const char *iface)
|
void LinuxNetLink::addAddress(const InetAddress &addr, const char *iface)
|
||||||
{
|
{
|
||||||
|
int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
|
||||||
|
if (fd == -1) {
|
||||||
|
fprintf(stderr, "Error opening RTNETLINK socket: %s\n", strerror(errno));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
_setSocketTimeout(fd);
|
||||||
|
|
||||||
|
struct sockaddr_nl la;
|
||||||
|
la.nl_family = AF_NETLINK;
|
||||||
|
la.nl_pid = getpid();
|
||||||
|
if (addr.isV4()) {
|
||||||
|
la.nl_groups = RTMGRP_IPV4_IFADDR;
|
||||||
|
} else {
|
||||||
|
la.nl_groups = RTMGRP_IPV6_IFADDR;
|
||||||
|
}
|
||||||
|
if(bind(fd, (struct sockaddr*)&la, sizeof(struct sockaddr_nl))) {
|
||||||
|
fprintf(stderr, "Error binding RTNETLINK: %s\n", strerror(errno));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
char tmp[128];
|
||||||
|
fprintf(stderr, "Adding IP address %s to interface %s", addr.toString(tmp), iface);
|
||||||
|
int interface_index = _indexForInterface(iface);
|
||||||
|
|
||||||
|
if (interface_index == -1) {
|
||||||
|
fprintf(stderr, "Unable to find index for interface %s\n", iface);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
int rtl = sizeof(struct ifaddrmsg);
|
||||||
|
struct nl_adr_req req;
|
||||||
|
bzero(&req, sizeof(struct nl_adr_req));
|
||||||
|
|
||||||
|
struct rtattr *rtap = (struct rtattr *)req.buf;;
|
||||||
|
if(addr.isV4()) {
|
||||||
|
struct sockaddr_in *addr_v4 = (struct sockaddr_in*)&addr;
|
||||||
|
rtap->rta_type = IFA_ADDRESS;
|
||||||
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
|
||||||
|
memcpy(RTA_DATA(rtap), &addr_v4->sin_addr, sizeof(struct in_addr));
|
||||||
|
rtl += rtap->rta_len;
|
||||||
|
|
||||||
|
rtap = (struct rtattr*)(((char*)rtap) + rtap->rta_len);
|
||||||
|
rtap->rta_type = IFA_LOCAL;
|
||||||
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
|
||||||
|
memcpy(RTA_DATA(rtap), &addr_v4->sin_addr, sizeof(struct in_addr));
|
||||||
|
rtl += rtap->rta_len;
|
||||||
|
|
||||||
|
InetAddress broadcast = addr.broadcast();
|
||||||
|
if(broadcast) {
|
||||||
|
rtap = (struct rtattr*)(((char*)rtap)+rtap->rta_len);
|
||||||
|
struct sockaddr_in *bcast = (struct sockaddr_in*)&broadcast;
|
||||||
|
rtap->rta_type = IFA_BROADCAST;
|
||||||
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
|
||||||
|
memcpy(RTA_DATA(rtap), &bcast->sin_addr, sizeof(struct in_addr));
|
||||||
|
rtl += rtap->rta_len;
|
||||||
|
}
|
||||||
|
} else { //V6
|
||||||
|
rtap->rta_type = IFA_ADDRESS;
|
||||||
|
struct sockaddr_in6 *addr_v6 = (struct sockaddr_in6*)&addr;
|
||||||
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in6_addr));
|
||||||
|
memcpy(RTA_DATA(rtap), &addr_v6->sin6_addr, sizeof(struct in6_addr));
|
||||||
|
rtl += rtap->rta_len;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (iface) {
|
||||||
|
rtap = (struct rtattr*)(((char*)rtap)+rtap->rta_len);
|
||||||
|
rtap->rta_type = IFA_LABEL;
|
||||||
|
rtap->rta_len = RTA_LENGTH(strlen(iface));
|
||||||
|
memcpy(RTA_DATA(rtap), iface, strlen(iface));
|
||||||
|
rtl += rtap->rta_len;
|
||||||
|
}
|
||||||
|
|
||||||
|
req.nl.nlmsg_len = NLMSG_LENGTH(rtl);
|
||||||
|
req.nl.nlmsg_flags = NLM_F_REQUEST | NLM_F_CREATE | NLM_F_EXCL;
|
||||||
|
req.nl.nlmsg_type = RTM_NEWADDR;
|
||||||
|
req.nl.nlmsg_pid = 0;
|
||||||
|
req.nl.nlmsg_seq = ++_seq;
|
||||||
|
req.ifa.ifa_family = addr.ss_family;
|
||||||
|
req.ifa.ifa_prefixlen = addr.port();
|
||||||
|
req.ifa.ifa_flags = IFA_F_PERMANENT;
|
||||||
|
req.ifa.ifa_scope = 0;
|
||||||
|
req.ifa.ifa_index = interface_index;
|
||||||
|
|
||||||
|
struct sockaddr_nl pa;
|
||||||
|
bzero(&pa, sizeof(sockaddr_nl));
|
||||||
|
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;
|
||||||
|
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);
|
||||||
|
|
||||||
|
_doRecv(fd);
|
||||||
|
|
||||||
|
close(fd);
|
||||||
}
|
}
|
||||||
|
|
||||||
void LinuxNetLink::removeAddress(const InetAddress &addr, const char *iface)
|
void LinuxNetLink::removeAddress(const InetAddress &addr, const char *iface)
|
||||||
{
|
{
|
||||||
|
int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
|
||||||
|
if (fd == -1) {
|
||||||
|
fprintf(stderr, "Error opening RTNETLINK socket: %s\n", strerror(errno));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
_setSocketTimeout(fd);
|
||||||
|
|
||||||
|
struct sockaddr_nl la;
|
||||||
|
la.nl_family = AF_NETLINK;
|
||||||
|
la.nl_pid = getpid();
|
||||||
|
if (addr.isV4()) {
|
||||||
|
la.nl_groups = RTMGRP_IPV4_IFADDR;
|
||||||
|
} else {
|
||||||
|
la.nl_groups = RTMGRP_IPV6_IFADDR;
|
||||||
|
}
|
||||||
|
if(bind(fd, (struct sockaddr*)&la, sizeof(struct sockaddr_nl))) {
|
||||||
|
fprintf(stderr, "Error binding RTNETLINK: %s\n", strerror(errno));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
char tmp[128];
|
||||||
|
fprintf(stderr, "Removing IP address %s from interface %s", addr.toString(tmp), iface);
|
||||||
|
|
||||||
|
int interface_index = _indexForInterface(iface);
|
||||||
|
|
||||||
|
if (interface_index == -1) {
|
||||||
|
fprintf(stderr, "Unable to find index for interface %s\n", iface);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
int rtl = sizeof(struct ifaddrmsg);
|
||||||
|
struct nl_adr_req req;
|
||||||
|
bzero(&req, sizeof(struct nl_adr_req));
|
||||||
|
|
||||||
|
struct rtattr *rtap = (struct rtattr *)req.buf;
|
||||||
|
if(addr.isV4()) {
|
||||||
|
struct sockaddr_in *addr_v4 = (struct sockaddr_in*)&addr;
|
||||||
|
rtap->rta_type = IFA_ADDRESS;
|
||||||
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
|
||||||
|
memcpy(RTA_DATA(rtap), &addr_v4->sin_addr, sizeof(struct in_addr));
|
||||||
|
rtl += rtap->rta_len;
|
||||||
|
|
||||||
|
rtap = (struct rtattr*)(((char*)rtap) + rtap->rta_len);
|
||||||
|
rtap->rta_type = IFA_LOCAL;
|
||||||
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
|
||||||
|
memcpy(RTA_DATA(rtap), &addr_v4->sin_addr, sizeof(struct in_addr));
|
||||||
|
rtl += rtap->rta_len;
|
||||||
|
|
||||||
|
InetAddress broadcast = addr.broadcast();
|
||||||
|
if(broadcast) {
|
||||||
|
rtap = (struct rtattr*)(((char*)rtap)+rtap->rta_len);
|
||||||
|
struct sockaddr_in *bcast = (struct sockaddr_in*)&broadcast;
|
||||||
|
rtap->rta_type = IFA_BROADCAST;
|
||||||
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in_addr));
|
||||||
|
memcpy(RTA_DATA(rtap), &bcast->sin_addr, sizeof(struct in_addr));
|
||||||
|
rtl += rtap->rta_len;
|
||||||
|
}
|
||||||
|
} else { //V6
|
||||||
|
rtap->rta_type = IFA_ADDRESS;
|
||||||
|
struct sockaddr_in6 *addr_v6 = (struct sockaddr_in6*)&addr;
|
||||||
|
rtap->rta_len = RTA_LENGTH(sizeof(struct in6_addr));
|
||||||
|
memcpy(RTA_DATA(rtap), &addr_v6->sin6_addr, sizeof(struct in6_addr));
|
||||||
|
rtl += rtap->rta_len;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (iface) {
|
||||||
|
rtap = (struct rtattr*)(((char*)rtap)+rtap->rta_len);
|
||||||
|
rtap->rta_type = IFA_LABEL;
|
||||||
|
rtap->rta_len = RTA_LENGTH(strlen(iface));
|
||||||
|
memcpy(RTA_DATA(rtap), iface, strlen(iface));
|
||||||
|
rtl += rtap->rta_len;
|
||||||
|
}
|
||||||
|
|
||||||
|
req.nl.nlmsg_len = NLMSG_LENGTH(rtl);
|
||||||
|
req.nl.nlmsg_flags = NLM_F_REQUEST;
|
||||||
|
req.nl.nlmsg_type = RTM_DELADDR;
|
||||||
|
req.nl.nlmsg_pid = 0;
|
||||||
|
req.nl.nlmsg_seq = ++_seq;
|
||||||
|
req.ifa.ifa_family = addr.ss_family;
|
||||||
|
req.ifa.ifa_prefixlen = addr.port();
|
||||||
|
req.ifa.ifa_flags = IFA_F_PERMANENT;
|
||||||
|
req.ifa.ifa_scope = 0;
|
||||||
|
req.ifa.ifa_index = interface_index;
|
||||||
|
|
||||||
|
struct sockaddr_nl pa;
|
||||||
|
bzero(&pa, sizeof(sockaddr_nl));
|
||||||
|
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;
|
||||||
|
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);
|
||||||
|
|
||||||
|
_doRecv(fd);
|
||||||
|
|
||||||
|
close(fd);
|
||||||
}
|
}
|
||||||
|
|
||||||
RouteList LinuxNetLink::getIPV4Routes() const
|
RouteList LinuxNetLink::getIPV4Routes() const
|
||||||
|
@ -721,4 +1039,20 @@ RouteList LinuxNetLink::getIPV6Routes() const
|
||||||
return _routes_ipv6;
|
return _routes_ipv6;
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace ZeroTier
|
int LinuxNetLink::_indexForInterface(const char *iface)
|
||||||
|
{
|
||||||
|
Mutex::Lock l(_if_m);
|
||||||
|
int interface_index = -1;
|
||||||
|
Hashtable<int, iface_entry>::Iterator iter(_interfaces);
|
||||||
|
int *k = NULL;
|
||||||
|
iface_entry *v = NULL;
|
||||||
|
while(iter.next(k,v)) {
|
||||||
|
if(strcmp(iface, v->ifacename) == 0) {
|
||||||
|
interface_index = v->index;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return interface_index;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace ZeroTier
|
||||||
|
|
|
@ -39,6 +39,7 @@
|
||||||
#include "../node/MAC.hpp"
|
#include "../node/MAC.hpp"
|
||||||
#include "Thread.hpp"
|
#include "Thread.hpp"
|
||||||
#include "../node/Hashtable.hpp"
|
#include "../node/Hashtable.hpp"
|
||||||
|
#include "../node/Mutex.hpp"
|
||||||
|
|
||||||
|
|
||||||
namespace ZeroTier {
|
namespace ZeroTier {
|
||||||
|
@ -70,19 +71,18 @@ public:
|
||||||
LinuxNetLink(LinuxNetLink const&) = delete;
|
LinuxNetLink(LinuxNetLink const&) = delete;
|
||||||
void operator=(LinuxNetLink const&) = delete;
|
void operator=(LinuxNetLink const&) = delete;
|
||||||
|
|
||||||
void addRoute(const InetAddress &target, const InetAddress &via, const char *ifaceName);
|
void addRoute(const InetAddress &target, const InetAddress &via, const InetAddress &src, const char *ifaceName);
|
||||||
void delRoute(const InetAddress &target, const InetAddress &via, const char *ifaceName);
|
void delRoute(const InetAddress &target, const InetAddress &via, const InetAddress &src, const char *ifaceName);
|
||||||
RouteList getIPV4Routes() const;
|
RouteList getIPV4Routes() const;
|
||||||
RouteList getIPV6Routes() const;
|
RouteList getIPV6Routes() const;
|
||||||
|
|
||||||
// void addInterface(const char *iface, unsigned int mtu, const MAC &mac);
|
|
||||||
// void removeInterface(const char *iface);
|
|
||||||
|
|
||||||
void addAddress(const InetAddress &addr, const char *iface);
|
void addAddress(const InetAddress &addr, const char *iface);
|
||||||
void removeAddress(const InetAddress &addr, const char *iface);
|
void removeAddress(const InetAddress &addr, const char *iface);
|
||||||
|
|
||||||
void threadMain() throw();
|
void threadMain() throw();
|
||||||
private:
|
private:
|
||||||
|
int _doRecv(int fd);
|
||||||
|
|
||||||
void _processMessage(struct nlmsghdr *nlp, int nll);
|
void _processMessage(struct nlmsghdr *nlp, int nll);
|
||||||
void _routeAdded(struct nlmsghdr *nlp);
|
void _routeAdded(struct nlmsghdr *nlp);
|
||||||
void _routeDeleted(struct nlmsghdr *nlp);
|
void _routeDeleted(struct nlmsghdr *nlp);
|
||||||
|
@ -95,12 +95,17 @@ private:
|
||||||
void _requestIPv4Routes();
|
void _requestIPv4Routes();
|
||||||
void _requestIPv6Routes();
|
void _requestIPv6Routes();
|
||||||
|
|
||||||
|
int _indexForInterface(const char *iface);
|
||||||
|
|
||||||
|
void _setSocketTimeout(int fd, int seconds = 1);
|
||||||
|
|
||||||
Thread _t;
|
Thread _t;
|
||||||
bool _running;
|
bool _running;
|
||||||
|
|
||||||
RouteList _routes_ipv4;
|
RouteList _routes_ipv4;
|
||||||
|
Mutex _rv4_m;
|
||||||
RouteList _routes_ipv6;
|
RouteList _routes_ipv6;
|
||||||
|
Mutex _rv6_m;
|
||||||
|
|
||||||
uint32_t _seq;
|
uint32_t _seq;
|
||||||
|
|
||||||
|
@ -112,6 +117,7 @@ private:
|
||||||
unsigned int mtu;
|
unsigned int mtu;
|
||||||
};
|
};
|
||||||
Hashtable<int, iface_entry> _interfaces;
|
Hashtable<int, iface_entry> _interfaces;
|
||||||
|
Mutex _if_m;
|
||||||
|
|
||||||
// socket communication vars;
|
// socket communication vars;
|
||||||
int _fd;
|
int _fd;
|
||||||
|
|
|
@ -285,21 +285,23 @@ static void _routeCmd(const char *op,const InetAddress &target,const InetAddress
|
||||||
#ifdef __LINUX__ // ----------------------------------------------------------
|
#ifdef __LINUX__ // ----------------------------------------------------------
|
||||||
#define ZT_ROUTING_SUPPORT_FOUND 1
|
#define ZT_ROUTING_SUPPORT_FOUND 1
|
||||||
|
|
||||||
static void _routeCmd(const char *op, const InetAddress &target, const InetAddress &via, const char *localInterface)
|
static void _routeCmd(const char *op, const InetAddress &target, const InetAddress &via, const InetAddress &src, const char *localInterface)
|
||||||
{
|
{
|
||||||
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 targetStr[64] = {0};
|
||||||
char viaStr[64] = {0};
|
char viaStr[64] = {0};
|
||||||
InetAddress nmsk = target.netmask();
|
InetAddress nmsk = target.netmask();
|
||||||
char nmskStr[64] = {0};
|
char nmskStr[64] = {0};
|
||||||
fprintf(stderr, "Received Route Cmd: %s target: %s via: %s netmask: %s localInterface: %s\n", op, target.toString(targetStr), via.toString(viaStr), nmsk.toString(nmskStr), localInterface);
|
fprintf(stderr, "Received Route Cmd: %s target: %s via: %s netmask: %s localInterface: %s\n", op, target.toString(targetStr), via.toString(viaStr), nmsk.toString(nmskStr), localInterface);
|
||||||
|
|
||||||
|
|
||||||
|
if ((strcmp(op, "add") == 0 || strcmp(op, "replace") == 0)) {
|
||||||
|
LinuxNetLink::getInstance().addRoute(target, via, src, localInterface);
|
||||||
|
} else if ((strcmp(op, "remove") == 0 || strcmp(op, "del") == 0)) {
|
||||||
|
LinuxNetLink::getInstance().delRoute(target, via, src, localInterface);
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
|
||||||
|
|
||||||
int fd = socket(PF_INET, SOCK_DGRAM, IPPROTO_IP);;
|
int fd = socket(PF_INET, SOCK_DGRAM, IPPROTO_IP);;
|
||||||
struct rtentry route = {0};
|
struct rtentry route = {0};
|
||||||
|
|
||||||
|
@ -600,11 +602,11 @@ bool ManagedRoute::sync()
|
||||||
|
|
||||||
if (!_applied.count(leftt)) {
|
if (!_applied.count(leftt)) {
|
||||||
_applied[leftt] = false; // boolean unused
|
_applied[leftt] = false; // boolean unused
|
||||||
_routeCmd("replace",leftt,_via,(_via) ? (const char *)0 : _device);
|
_routeCmd("replace",leftt,_via,_src,_device);
|
||||||
}
|
}
|
||||||
if ((rightt)&&(!_applied.count(rightt))) {
|
if ((rightt)&&(!_applied.count(rightt))) {
|
||||||
_applied[rightt] = false; // boolean unused
|
_applied[rightt] = false; // boolean unused
|
||||||
_routeCmd("replace",rightt,_via,(_via) ? (const char *)0 : _device);
|
_routeCmd("replace",rightt,_via,_src,_device);
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif // __LINUX__ ----------------------------------------------------------
|
#endif // __LINUX__ ----------------------------------------------------------
|
||||||
|
@ -651,7 +653,7 @@ void ManagedRoute::remove()
|
||||||
#endif // __BSD__ ------------------------------------------------------------
|
#endif // __BSD__ ------------------------------------------------------------
|
||||||
|
|
||||||
#ifdef __LINUX__ // ----------------------------------------------------------
|
#ifdef __LINUX__ // ----------------------------------------------------------
|
||||||
_routeCmd("del",r->first,_via,(_via) ? (const char *)0 : _device);
|
_routeCmd("del",r->first,_via,_src,_device);
|
||||||
#endif // __LINUX__ ----------------------------------------------------------
|
#endif // __LINUX__ ----------------------------------------------------------
|
||||||
|
|
||||||
#ifdef __WINDOWS__ // --------------------------------------------------------
|
#ifdef __WINDOWS__ // --------------------------------------------------------
|
||||||
|
|
|
@ -49,14 +49,20 @@ class ManagedRoute
|
||||||
friend class SharedPtr<ManagedRoute>;
|
friend class SharedPtr<ManagedRoute>;
|
||||||
|
|
||||||
public:
|
public:
|
||||||
ManagedRoute(const InetAddress &target,const InetAddress &via,const char *device)
|
ManagedRoute(const InetAddress &target,const InetAddress &via,const InetAddress &src,const char *device)
|
||||||
{
|
{
|
||||||
_target = target;
|
_target = target;
|
||||||
_via = via;
|
_via = via;
|
||||||
|
_src = src;
|
||||||
if (via.ss_family == AF_INET)
|
if (via.ss_family == AF_INET)
|
||||||
_via.setPort(32);
|
_via.setPort(32);
|
||||||
else if (via.ss_family == AF_INET6)
|
else if (via.ss_family == AF_INET6)
|
||||||
_via.setPort(128);
|
_via.setPort(128);
|
||||||
|
if (src.ss_family == AF_INET) {
|
||||||
|
_src.setPort(32);
|
||||||
|
} else if (src.ss_family == AF_INET6) {
|
||||||
|
_src.setPort(128);
|
||||||
|
}
|
||||||
Utils::scopy(_device,sizeof(_device),device);
|
Utils::scopy(_device,sizeof(_device),device);
|
||||||
_systemDevice[0] = (char)0;
|
_systemDevice[0] = (char)0;
|
||||||
}
|
}
|
||||||
|
@ -87,6 +93,7 @@ public:
|
||||||
|
|
||||||
inline const InetAddress &target() const { return _target; }
|
inline const InetAddress &target() const { return _target; }
|
||||||
inline const InetAddress &via() const { return _via; }
|
inline const InetAddress &via() const { return _via; }
|
||||||
|
inline const InetAddress &src() const { return _src; }
|
||||||
inline const char *device() const { return _device; }
|
inline const char *device() const { return _device; }
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
@ -95,6 +102,7 @@ private:
|
||||||
|
|
||||||
InetAddress _target;
|
InetAddress _target;
|
||||||
InetAddress _via;
|
InetAddress _via;
|
||||||
|
InetAddress _src;
|
||||||
InetAddress _systemVia; // for route overrides
|
InetAddress _systemVia; // for route overrides
|
||||||
std::map<InetAddress,bool> _applied; // routes currently applied
|
std::map<InetAddress,bool> _applied; // routes currently applied
|
||||||
char _device[128];
|
char _device[128];
|
||||||
|
|
|
@ -1631,6 +1631,17 @@ public:
|
||||||
for(unsigned int i=0;i<n.config.routeCount;++i) {
|
for(unsigned int i=0;i<n.config.routeCount;++i) {
|
||||||
const InetAddress *const target = reinterpret_cast<const InetAddress *>(&(n.config.routes[i].target));
|
const InetAddress *const target = reinterpret_cast<const InetAddress *>(&(n.config.routes[i].target));
|
||||||
const InetAddress *const via = reinterpret_cast<const InetAddress *>(&(n.config.routes[i].via));
|
const InetAddress *const via = reinterpret_cast<const InetAddress *>(&(n.config.routes[i].via));
|
||||||
|
InetAddress *src = NULL;
|
||||||
|
for (unsigned int j=0; j<n.config.assignedAddressCount; ++j) {
|
||||||
|
const InetAddress *const tmp = reinterpret_cast<const InetAddress *>(&(n.config.assignedAddresses[j]));
|
||||||
|
if (target->isV4() && tmp->isV4()) {
|
||||||
|
src = reinterpret_cast<InetAddress *>(&(n.config.assignedAddresses[j]));
|
||||||
|
break;
|
||||||
|
} else if (target->isV6() && tmp->isV6()) {
|
||||||
|
src = reinterpret_cast<InetAddress *>(&(n.config.assignedAddresses[j]));
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if ( (!checkIfManagedIsAllowed(n,*target)) || ((via->ss_family == target->ss_family)&&(matchIpOnly(myIps,*via))) )
|
if ( (!checkIfManagedIsAllowed(n,*target)) || ((via->ss_family == target->ss_family)&&(matchIpOnly(myIps,*via))) )
|
||||||
continue;
|
continue;
|
||||||
|
@ -1662,7 +1673,7 @@ public:
|
||||||
continue;
|
continue;
|
||||||
|
|
||||||
// Add and apply new routes
|
// Add and apply new routes
|
||||||
n.managedRoutes.push_back(SharedPtr<ManagedRoute>(new ManagedRoute(*target,*via,tapdev)));
|
n.managedRoutes.push_back(SharedPtr<ManagedRoute>(new ManagedRoute(*target,*via,*src,tapdev)));
|
||||||
if (!n.managedRoutes.back()->sync())
|
if (!n.managedRoutes.back()->sync())
|
||||||
n.managedRoutes.pop_back();
|
n.managedRoutes.pop_back();
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Reference in a new issue