diff --git a/osdep/ExtOsdep.cpp b/osdep/ExtOsdep.cpp index 9ea03340e..47543ae4d 100644 --- a/osdep/ExtOsdep.cpp +++ b/osdep/ExtOsdep.cpp @@ -1,11 +1,26 @@ -#include -#include -#include -#include +/* + * Copyright (c)2019 ZeroTier, Inc. + * + * Use of this software is governed by the Business Source License included + * in the LICENSE.TXT file in the project's root directory. + * + * Change Date: 2026-01-01 + * + * On the date above, in accordance with the Business Source License, use + * of this software will be governed by version 2.0 of the Apache License. + */ +/****/ + #include "ExtOsdep.hpp" -#include + #include "../node/AtomicCounter.hpp" +#include +#include +#include +#include +#include + #define ZT_TAP_BUF_SIZE 16384 namespace ZeroTier { @@ -22,18 +37,24 @@ struct EodRoute { }; static std::list allRoutes; -template static void __eodSend(const T &t) { +template static void __eodSend(const T& t) +{ write(eodFd, &t, sizeof(t)); } -static void strncpyx(char *dest, const char *src, size_t n) { +static void strncpyx(char* dest, const char* src, size_t n) +{ strncpy(dest, src, n); - if (n > 1) dest[n - 1] = 0; + if (n > 1) { + dest[n - 1] = 0; + } } -static int __eodWait(unsigned char msg, unsigned char *d, unsigned l, - unsigned maxl = 0, int *recvfd = nullptr) { - if (!maxl) maxl = l; +static int __eodWait(unsigned char msg, unsigned char* d, unsigned l, unsigned maxl = 0, int* recvfd = nullptr) +{ + if (! maxl) { + maxl = l; + } auto start = times(NULL); while (1) { msghdr mh; @@ -56,10 +77,7 @@ static int __eodWait(unsigned char msg, unsigned char *d, unsigned l, int r = recvmsg(eodFd, &mh, MSG_TRUNC | MSG_CMSG_CLOEXEC); if (r > 0) { - if (recvfd && mh.msg_controllen >= sizeof(cmsg) - && cmsg.cmsg_len == sizeof(cmsg) - && cmsg.cmsg_level == SOL_SOCKET - && cmsg.cmsg_type == SCM_RIGHTS) { + if (recvfd && mh.msg_controllen >= sizeof(cmsg) && cmsg.cmsg_len == sizeof(cmsg) && cmsg.cmsg_level == SOL_SOCKET && cmsg.cmsg_type == SCM_RIGHTS) { *recvfd = cmsg.fd; fprintf(stderr, "eodWait: received fd %d\n", *recvfd); } @@ -82,29 +100,34 @@ static int __eodWait(unsigned char msg, unsigned char *d, unsigned l, } } -template static bool __eodWait(unsigned msg, T &t) { - return __eodWait(msg, (unsigned char *)&t, sizeof(T)) == (int)sizeof(T); +template static bool __eodWait(unsigned msg, T& t) +{ + return __eodWait(msg, (unsigned char*)&t, sizeof(T)) == (int)sizeof(T); } -template static bool __eodXchg(const M &m, unsigned rm, R &r) { +template static bool __eodXchg(const M& m, unsigned rm, R& r) +{ __eodSend(m); return __eodWait(rm, r); } -template static bool eodXchg(const M &m, unsigned rm, R &r) { +template static bool eodXchg(const M& m, unsigned rm, R& r) +{ Mutex::Lock l(eodMutex); return __eodXchg(m, rm, r); } -void ExtOsdep::init(int fd1, int fd2) { +void ExtOsdep::init(int fd1, int fd2) +{ eodFd = fd1; eodMgmtFd = fd2; - fcntl(eodMgmtFd,F_SETFL,O_NONBLOCK); + fcntl(eodMgmtFd, F_SETFL, O_NONBLOCK); } -void ExtOsdep::started(int *f, void **cp) { +void ExtOsdep::started(int* f, void** cp) +{ *f = eodMgmtFd; - *cp = (void *)eodMgmtFd; + *cp = (void*)eodMgmtFd; unsigned char msg = ZT_EOD_MSG_STARTED; Mutex::Lock l(eodMutex); @@ -114,66 +137,85 @@ void ExtOsdep::started(int *f, void **cp) { static std::string mgmtrd; static std::string mgmtwr; -bool ExtOsdep::mgmtWritable(void *cookie) { - if (cookie != (void *)eodMgmtFd) return false; - if (mgmtwr.size() == 0) return true; +bool ExtOsdep::mgmtWritable(void* cookie) +{ + if (cookie != (void*)eodMgmtFd) { + return false; + } + if (mgmtwr.size() == 0) { + return true; + } auto sz = write(eodMgmtFd, mgmtwr.data(), mgmtwr.size()); - if (sz <= 0) return false; + if (sz <= 0) { + return false; + } mgmtwr.erase(mgmtwr.begin(), mgmtwr.begin() + sz); return mgmtwr.empty(); } -bool ExtOsdep::mgmtRecv(void *cookie, void *data, unsigned long len, - std::function cb) { - if (cookie != (void *)eodMgmtFd) return false; - mgmtrd.append((char *)data, len); +bool ExtOsdep::mgmtRecv(void* cookie, void* data, unsigned long len, std::function cb) +{ + if (cookie != (void*)eodMgmtFd) { + return false; + } + mgmtrd.append((char*)data, len); while (1) { - auto req = (zt_eod_mgmt_req *)mgmtrd.data(); - if (mgmtrd.size() < sizeof(*req)) break; + auto req = (zt_eod_mgmt_req*)mgmtrd.data(); + if (mgmtrd.size() < sizeof(*req)) { + break; + } unsigned reqsz = sizeof(*req) + req->pathlen + req->datalen; - if (mgmtrd.size() < reqsz) break; + if (mgmtrd.size() < reqsz) { + break; + } std::string resp; - char *p = (char *)req->data; + char* p = (char*)req->data; zt_eod_mgmt_reply rep; - rep.scode = cb(req->method, std::string(p, p + req->pathlen), - std::string(p + req->pathlen, p + req->pathlen + req->datalen), resp); + rep.scode = cb(req->method, std::string(p, p + req->pathlen), std::string(p + req->pathlen, p + req->pathlen + req->datalen), resp); rep.datalen = resp.size(); mgmtrd.erase(mgmtrd.begin(), mgmtrd.begin() + reqsz); - mgmtwr.append((char *)&rep, sizeof(rep)); + mgmtwr.append((char*)&rep, sizeof(rep)); mgmtwr.append(resp); auto sz = write(eodMgmtFd, mgmtwr.data(), mgmtwr.size()); - if (sz > 0) mgmtwr.erase(mgmtwr.begin(), mgmtwr.begin() + sz); + if (sz > 0) { + mgmtwr.erase(mgmtwr.begin(), mgmtwr.begin() + sz); + } } - return !mgmtwr.empty(); + return ! mgmtwr.empty(); } -void ExtOsdep::routeAddDel(bool add, const InetAddress &target, const InetAddress &via, const InetAddress &src, const char *ifname) { +void ExtOsdep::routeAddDel(bool add, const InetAddress& target, const InetAddress& via, const InetAddress& src, const char* ifname) +{ Mutex::Lock l(eodMutex); std::string ifn; - if (ifname) ifn = ifname; + if (ifname) { + ifn = ifname; + } if (add) { for (auto x = allRoutes.begin(); x != allRoutes.end(); ++x) { - if (x->target == target && x->via == via - && x->src == src && x->ifname == ifn) return; + if (x->target == target && x->via == via && x->src == src && x->ifname == ifn) { + return; + } } - allRoutes.push_back({target, via, src, ifn}); + allRoutes.push_back({ target, via, src, ifn }); } else { bool found = false; for (auto x = allRoutes.begin(); x != allRoutes.end(); ++x) { - if (x->target == target && x->via == via - && x->src == src && x->ifname == ifn) { + if (x->target == target && x->via == via && x->src == src && x->ifname == ifn) { allRoutes.erase(x); found = true; break; } } - if (!found) return; + if (! found) { + return; + } } zt_eod_msg_route req; @@ -182,54 +224,62 @@ void ExtOsdep::routeAddDel(bool add, const InetAddress &target, const InetAddres req.afi = target.isV4() ? 1 : 2; req.dstlen = target.netmaskBits(); memcpy(req.dst, target.rawIpData(), target.isV4() ? 4 : 16); - if (ifname) strncpyx(req.dev, ifname, sizeof(req.dev)); - if (via) memcpy(req.gw, via.rawIpData(), target.isV4() ? 4 : 16); - if (src) memcpy(req.src, src.rawIpData(), target.isV4() ? 4 : 16); + if (ifname) { + strncpyx(req.dev, ifname, sizeof(req.dev)); + } + if (via) { + memcpy(req.gw, via.rawIpData(), target.isV4() ? 4 : 16); + } + if (src) { + memcpy(req.src, src.rawIpData(), target.isV4() ? 4 : 16); + } unsigned char resp; __eodXchg(req, add ? ZT_EOD_MSG_ADDROUTERESP : ZT_EOD_MSG_DELROUTERESP, resp); } -bool ExtOsdep::getBindAddrs(std::map &ret) { +bool ExtOsdep::getBindAddrs(std::map& ret) +{ Mutex::Lock l(eodMutex); unsigned char req = ZT_EOD_MSG_GETBINDADDRS; __eodSend(req); - zt_eod_msg_getbindaddrsresp *resp; + zt_eod_msg_getbindaddrsresp* resp; unsigned char buf[ZT_EOD_MAXMSGSIZE]; - int r = __eodWait(ZT_EOD_MSG_GETBINDADDRSRESP, (unsigned char *)buf, sizeof(*resp), sizeof(buf)); - if (r < (int)sizeof(*resp)) return false; + int r = __eodWait(ZT_EOD_MSG_GETBINDADDRSRESP, (unsigned char*)buf, sizeof(*resp), sizeof(buf)); + if (r < (int)sizeof(*resp)) { + return false; + } int c = (r - (int)sizeof(*resp)) / sizeof(resp->addrs[0]); - resp = (zt_eod_msg_getbindaddrsresp *)buf; + resp = (zt_eod_msg_getbindaddrsresp*)buf; for (int i = 0; i < c; ++i) { - ret[InetAddress(resp->addrs[i].data, resp->addrs[i].afi == 1 ? 4 : 16, resp->addrs[i].len)] - = resp->addrs[i].ifname; + ret[InetAddress(resp->addrs[i].data, resp->addrs[i].afi == 1 ? 4 : 16, resp->addrs[i].len)] = resp->addrs[i].ifname; } return resp->result; } ExtOsdepTap::ExtOsdepTap( - const char *homePath, - const MAC &mac, - unsigned int mtu, - unsigned int metric, - uint64_t nwid, - const char *friendlyName, - void (*handler)(void *,void *,uint64_t,const MAC &,const MAC &,unsigned int,unsigned int,const void *,unsigned int), - void *arg) : - _handler(handler), - _arg(arg), - _nwid(nwid), - _mac(mac), - _homePath(homePath), - _mtu(mtu), - _fd(0), - _enabled(true), - _run(true) - { + const char* homePath, + const MAC& mac, + unsigned int mtu, + unsigned int metric, + uint64_t nwid, + const char* friendlyName, + void (*handler)(void*, void*, uint64_t, const MAC&, const MAC&, unsigned int, unsigned int, const void*, unsigned int), + void* arg) + : _handler(handler) + , _arg(arg) + , _nwid(nwid) + , _mac(mac) + , _homePath(homePath) + , _mtu(mtu) + , _fd(0) + , _enabled(true) + , _run(true) +{ zt_eod_msg_addtap req; req.cmd = ZT_EOD_MSG_ADDTAP; req.nwid = nwid; @@ -243,131 +293,147 @@ ExtOsdepTap::ExtOsdepTap( Mutex::Lock l(eodMutex); __eodSend(req); _fd = -1; - if (__eodWait(ZT_EOD_MSG_ADDTAPRESP, (unsigned char *)&resp, sizeof(resp), sizeof(resp), &_fd) != sizeof(resp)) + if (__eodWait(ZT_EOD_MSG_ADDTAPRESP, (unsigned char*)&resp, sizeof(resp), sizeof(resp), &_fd) != sizeof(resp)) { throw std::runtime_error(std::string("could not create TAP")); + } _dev = resp.name; - if (_dev.empty() || _fd < 0) + if (_dev.empty() || _fd < 0) { throw std::runtime_error(std::string("could not create TAP")); + } - fcntl(_fd,F_SETFL,O_NONBLOCK); + fcntl(_fd, F_SETFL, O_NONBLOCK); - // processing shamelessly copied from LinuxEthernetTap - (void)::pipe(_shutdownSignalPipe); - for(unsigned int t=0;t<2;++t) { - _tapReaderThread[t] = std::thread([this, t]{ - fd_set readfds,nullfds; - int n,nfds,r; - void *buf = nullptr; - std::vector buffers; + (void)::pipe(_shutdownSignalPipe); + for (unsigned int t = 0; t < 2; ++t) { + _tapReaderThread[t] = std::thread([this, t] { + fd_set readfds, nullfds; + int n, nfds, r; + void* buf = nullptr; + std::vector buffers; - if (!_run) - return; + if (! _run) { + return; + } - FD_ZERO(&readfds); - FD_ZERO(&nullfds); - nfds = (int)std::max(_shutdownSignalPipe[0],_fd) + 1; + FD_ZERO(&readfds); + FD_ZERO(&nullfds); + nfds = (int)std::max(_shutdownSignalPipe[0], _fd) + 1; - r = 0; - for(;;) { - FD_SET(_shutdownSignalPipe[0],&readfds); - FD_SET(_fd,&readfds); - select(nfds,&readfds,&nullfds,&nullfds,(struct timeval *)0); + r = 0; + for (;;) { + FD_SET(_shutdownSignalPipe[0], &readfds); + FD_SET(_fd, &readfds); + select(nfds, &readfds, &nullfds, &nullfds, (struct timeval*)0); - if (FD_ISSET(_shutdownSignalPipe[0],&readfds)) // writes to shutdown pipe terminate thread - break; + if (FD_ISSET(_shutdownSignalPipe[0], &readfds)) { // writes to shutdown pipe terminate thread + break; + } - if (FD_ISSET(_fd,&readfds)) { - for(;;) { // read until there are no more packets, then return to outer select() loop - if (!buf) { - // To reduce use of the mutex, we keep a local buffer vector and - // swap (which is a pointer swap) with the global one when it's - // empty. This retrieves a batch of buffers to use. - if (buffers.empty()) { - std::lock_guard l(_buffers_l); - buffers.swap(_buffers); - } - if (buffers.empty()) { - buf = malloc(ZT_TAP_BUF_SIZE); - if (!buf) - break; - } else { - buf = buffers.back(); - buffers.pop_back(); - } - } + if (FD_ISSET(_fd, &readfds)) { + for (;;) { // read until there are no more packets, then return to outer select() loop + if (! buf) { + // To reduce use of the mutex, we keep a local buffer vector and + // swap (which is a pointer swap) with the global one when it's + // empty. This retrieves a batch of buffers to use. + if (buffers.empty()) { + std::lock_guard l(_buffers_l); + buffers.swap(_buffers); + } + if (buffers.empty()) { + buf = malloc(ZT_TAP_BUF_SIZE); + if (! buf) { + break; + } + } + else { + buf = buffers.back(); + buffers.pop_back(); + } + } - n = (int)::read(_fd,reinterpret_cast(buf) + r,ZT_TAP_BUF_SIZE - r); - if (n > 0) { - // Some tap drivers like to send the ethernet frame and the - // payload in two chunks, so handle that by accumulating - // data until we have at least a frame. - r += n; - if (r > 14) { - if (r > ((int)_mtu + 14)) // sanity check for weird TAP behavior on some platforms - r = _mtu + 14; + n = (int)::read(_fd, reinterpret_cast(buf) + r, ZT_TAP_BUF_SIZE - r); + if (n > 0) { + // Some tap drivers like to send the ethernet frame and the + // payload in two chunks, so handle that by accumulating + // data until we have at least a frame. + r += n; + if (r > 14) { + if (r > ((int)_mtu + 14)) { // sanity check for weird TAP behavior on some platforms + r = _mtu + 14; + } - if (_enabled && _tapqsize.load() < 1000) { + if (_enabled && _tapqsize.load() < 1000) { ++_tapqsize; - _tapq.post(std::pair(buf,r)); - buf = nullptr; - } + _tapq.post(std::pair(buf, r)); + buf = nullptr; + } - r = 0; - } - } else { - r = 0; - break; - } - } - } - } - }); - } + r = 0; + } + } + else { + r = 0; + break; + } + } + } + } + }); + } - _tapProcessorThread = std::thread([this] { - MAC to,from; - std::pair qi; - while (_tapq.get(qi)) { + _tapProcessorThread = std::thread([this] { + MAC to, from; + std::pair qi; + while (_tapq.get(qi)) { --_tapqsize; - uint8_t *const b = reinterpret_cast(qi.first); - if (b) { - to.setTo(b, 6); - from.setTo(b + 6, 6); - unsigned int etherType = Utils::ntoh(((const uint16_t *)b)[6]); - _handler(_arg, nullptr, _nwid, from, to, etherType, 0, (const void *)(b + 14),(unsigned int)(qi.second - 14)); - { - std::lock_guard l(_buffers_l); - if (_buffers.size() < 128) - _buffers.push_back(qi.first); - else free(qi.first); - } - } else break; - } - }); + uint8_t* const b = reinterpret_cast(qi.first); + if (b) { + to.setTo(b, 6); + from.setTo(b + 6, 6); + unsigned int etherType = Utils::ntoh(((const uint16_t*)b)[6]); + _handler(_arg, nullptr, _nwid, from, to, etherType, 0, (const void*)(b + 14), (unsigned int)(qi.second - 14)); + { + std::lock_guard l(_buffers_l); + if (_buffers.size() < 128) { + _buffers.push_back(qi.first); + } + else { + free(qi.first); + } + } + } + else { + break; + } + } + }); } -ExtOsdepTap::~ExtOsdepTap() { - _run = false; +ExtOsdepTap::~ExtOsdepTap() +{ + _run = false; - (void)::write(_shutdownSignalPipe[1],"\0",1); // causes reader thread(s) to exit - _tapq.post(std::pair(nullptr,0)); // causes processor thread to exit + (void)::write(_shutdownSignalPipe[1], "\0", 1); // causes reader thread(s) to exit + _tapq.post(std::pair(nullptr, 0)); // causes processor thread to exit - _tapReaderThread[0].join(); - _tapReaderThread[1].join(); - _tapProcessorThread.join(); + _tapReaderThread[0].join(); + _tapReaderThread[1].join(); + _tapProcessorThread.join(); - ::close(_fd); - ::close(_shutdownSignalPipe[0]); - ::close(_shutdownSignalPipe[1]); + ::close(_fd); + ::close(_shutdownSignalPipe[0]); + ::close(_shutdownSignalPipe[1]); - for(std::vector::iterator i(_buffers.begin());i!=_buffers.end();++i) - free(*i); - std::vector< std::pair > dv(_tapq.drain()); - for(std::vector< std::pair >::iterator i(dv.begin());i!=dv.end();++i) { - if (i->first) - free(i->first); - } + for (std::vector::iterator i(_buffers.begin()); i != _buffers.end(); ++i) { + free(*i); + } + std::vector > dv(_tapq.drain()); + for (std::vector >::iterator i(dv.begin()); i != dv.end(); ++i) { + if (i->first) { + free(i->first); + } + } zt_eod_msg_deltap req; req.cmd = ZT_EOD_MSG_DELTAP; @@ -377,15 +443,18 @@ ExtOsdepTap::~ExtOsdepTap() { eodXchg(req, ZT_EOD_MSG_DELTAPRESP, resp); } -void ExtOsdepTap::setEnabled(bool en) { - _enabled = en; +void ExtOsdepTap::setEnabled(bool en) +{ + _enabled = en; } -bool ExtOsdepTap::enabled() const { - return _enabled; +bool ExtOsdepTap::enabled() const +{ + return _enabled; } -void ExtOsdepTap::doRemoveIp(const InetAddress &ip) { +void ExtOsdepTap::doRemoveIp(const InetAddress& ip) +{ zt_eod_msg_ip req; req.cmd = ZT_EOD_MSG_DELIP; strcpy(req.name, _dev.c_str()); @@ -397,13 +466,18 @@ void ExtOsdepTap::doRemoveIp(const InetAddress &ip) { __eodXchg(req, ZT_EOD_MSG_DELIPRESP, resp); } -bool ExtOsdepTap::addIp(const InetAddress &ip) { +bool ExtOsdepTap::addIp(const InetAddress& ip) +{ Mutex::Lock l(eodMutex); - for(auto i = allIps.begin();i!=allIps.end();++i) { - if (*i == ip) return true; - if (i->ipsEqual(ip)) doRemoveIp(*i); - } + for (auto i = allIps.begin(); i != allIps.end(); ++i) { + if (*i == ip) { + return true; + } + if (i->ipsEqual(ip)) { + doRemoveIp(*i); + } + } zt_eod_msg_ip req; req.cmd = ZT_EOD_MSG_ADDIP; @@ -419,20 +493,24 @@ bool ExtOsdepTap::addIp(const InetAddress &ip) { return true; } -bool ExtOsdepTap::addIps(std::vector ips) { +bool ExtOsdepTap::addIps(std::vector ips) +{ return false; } -bool ExtOsdepTap::removeIp(const InetAddress &ip) { + +bool ExtOsdepTap::removeIp(const InetAddress& ip) +{ Mutex::Lock l(eodMutex); - for(auto i = allIps.begin();i!=allIps.end();++i) { + for (auto i = allIps.begin(); i != allIps.end(); ++i) { if (*i == ip) { doRemoveIp(*i); return true; } - } + } return false; } -std::vector ExtOsdepTap::ips() const { +std::vector ExtOsdepTap::ips() const +{ std::vector ret; Mutex::Lock l(eodMutex); @@ -442,84 +520,100 @@ std::vector ExtOsdepTap::ips() const { strcpy(req.name, _dev.c_str()); __eodSend(req); - zt_eod_msg_getipsresp *resp; + zt_eod_msg_getipsresp* resp; unsigned char buf[ZT_EOD_MAXMSGSIZE]; - int r = __eodWait(ZT_EOD_MSG_GETIPSRESP, (unsigned char *)buf, sizeof(*resp), sizeof(buf)); - if (r < (int)sizeof(*resp)) return ret; + int r = __eodWait(ZT_EOD_MSG_GETIPSRESP, (unsigned char*)buf, sizeof(*resp), sizeof(buf)); + if (r < (int)sizeof(*resp)) { + return ret; + } int c = (r - (int)sizeof(*resp)) / sizeof(resp->addrs[0]); - resp = (zt_eod_msg_getipsresp *)buf; + resp = (zt_eod_msg_getipsresp*)buf; for (int i = 0; i < c; ++i) { ret.push_back(InetAddress(resp->addrs[i].data, resp->addrs[i].afi == 1 ? 4 : 16, resp->addrs[i].len)); } return ret; } -void ExtOsdepTap::put(const MAC &from,const MAC &to,unsigned int etherType,const void *data,unsigned int len) { - char putBuf[ZT_MAX_MTU + 64]; - if ((_fd > 0)&&(len <= _mtu)&&(_enabled)) { - to.copyTo(putBuf,6); - from.copyTo(putBuf + 6,6); - *((uint16_t *)(putBuf + 12)) = htons((uint16_t)etherType); - memcpy(putBuf + 14,data,len); - len += 14; - (void)::write(_fd,putBuf,len); - } +void ExtOsdepTap::put(const MAC& from, const MAC& to, unsigned int etherType, const void* data, unsigned int len) +{ + char putBuf[ZT_MAX_MTU + 64]; + if ((_fd > 0) && (len <= _mtu) && (_enabled)) { + to.copyTo(putBuf, 6); + from.copyTo(putBuf + 6, 6); + *((uint16_t*)(putBuf + 12)) = htons((uint16_t)etherType); + memcpy(putBuf + 14, data, len); + len += 14; + (void)::write(_fd, putBuf, len); + } } -std::string ExtOsdepTap::deviceName() const { +std::string ExtOsdepTap::deviceName() const +{ return _dev; } -void ExtOsdepTap::setFriendlyName(const char *friendlyName) {} - -void ExtOsdepTap::scanMulticastGroups(std::vector &added,std::vector &removed) { - char *ptr,*ptr2; - unsigned char mac[6]; - std::vector newGroups; - - int fd = ::open("/proc/net/dev_mcast",O_RDONLY); - if (fd > 0) { - char buf[131072]; - int n = (int)::read(fd,buf,sizeof(buf)); - if ((n > 0)&&(n < (int)sizeof(buf))) { - buf[n] = (char)0; - for(char *l=strtok_r(buf,"\r\n",&ptr);(l);l=strtok_r((char *)0,"\r\n",&ptr)) { - int fno = 0; - char *devname = (char *)0; - char *mcastmac = (char *)0; - for(char *f=strtok_r(l," \t",&ptr2);(f);f=strtok_r((char *)0," \t",&ptr2)) { - if (fno == 1) - devname = f; - else if (fno == 4) - mcastmac = f; - ++fno; - } - if ((devname)&&(!strcmp(devname,_dev.c_str()))&&(mcastmac)&&(Utils::unhex(mcastmac,mac,6) == 6)) - newGroups.push_back(MulticastGroup(MAC(mac,6),0)); - } - } - ::close(fd); - } - - std::vector allIps(ips()); - for(std::vector::iterator ip(allIps.begin());ip!=allIps.end();++ip) - newGroups.push_back(MulticastGroup::deriveMulticastGroupForAddressResolution(*ip)); - - std::sort(newGroups.begin(),newGroups.end()); - newGroups.erase(std::unique(newGroups.begin(),newGroups.end()),newGroups.end()); - - for(std::vector::iterator m(newGroups.begin());m!=newGroups.end();++m) { - if (!std::binary_search(_multicastGroups.begin(),_multicastGroups.end(),*m)) - added.push_back(*m); - } - for(std::vector::iterator m(_multicastGroups.begin());m!=_multicastGroups.end();++m) { - if (!std::binary_search(newGroups.begin(),newGroups.end(),*m)) - removed.push_back(*m); - } - - _multicastGroups.swap(newGroups); +void ExtOsdepTap::setFriendlyName(const char* friendlyName) +{ } -void ExtOsdepTap::setMtu(unsigned int mtu) { - if (mtu == _mtu) return; + +void ExtOsdepTap::scanMulticastGroups(std::vector& added, std::vector& removed) +{ + char *ptr, *ptr2; + unsigned char mac[6]; + std::vector newGroups; + + int fd = ::open("/proc/net/dev_mcast", O_RDONLY); + if (fd > 0) { + char buf[131072]; + int n = (int)::read(fd, buf, sizeof(buf)); + if ((n > 0) && (n < (int)sizeof(buf))) { + buf[n] = (char)0; + for (char* l = strtok_r(buf, "\r\n", &ptr); (l); l = strtok_r((char*)0, "\r\n", &ptr)) { + int fno = 0; + char* devname = (char*)0; + char* mcastmac = (char*)0; + for (char* f = strtok_r(l, " \t", &ptr2); (f); f = strtok_r((char*)0, " \t", &ptr2)) { + if (fno == 1) { + devname = f; + } + else if (fno == 4) { + mcastmac = f; + } + ++fno; + } + if ((devname) && (! strcmp(devname, _dev.c_str())) && (mcastmac) && (Utils::unhex(mcastmac, mac, 6) == 6)) { + newGroups.push_back(MulticastGroup(MAC(mac, 6), 0)); + } + } + } + ::close(fd); + } + + std::vector allIps(ips()); + for (std::vector::iterator ip(allIps.begin()); ip != allIps.end(); ++ip) { + newGroups.push_back(MulticastGroup::deriveMulticastGroupForAddressResolution(*ip)); + } + + std::sort(newGroups.begin(), newGroups.end()); + newGroups.erase(std::unique(newGroups.begin(), newGroups.end()), newGroups.end()); + + for (std::vector::iterator m(newGroups.begin()); m != newGroups.end(); ++m) { + if (! std::binary_search(_multicastGroups.begin(), _multicastGroups.end(), *m)) { + added.push_back(*m); + } + } + for (std::vector::iterator m(_multicastGroups.begin()); m != _multicastGroups.end(); ++m) { + if (! std::binary_search(newGroups.begin(), newGroups.end(), *m)) { + removed.push_back(*m); + } + } + + _multicastGroups.swap(newGroups); +} +void ExtOsdepTap::setMtu(unsigned int mtu) +{ + if (mtu == _mtu) { + return; + } _mtu = mtu; zt_eod_msg_setmtu req; @@ -531,4 +625,4 @@ void ExtOsdepTap::setMtu(unsigned int mtu) { eodXchg(req, ZT_EOD_MSG_SETMTURESP, resp); } -} // namespace ZeroTier +} // namespace ZeroTier diff --git a/osdep/ExtOsdep.hpp b/osdep/ExtOsdep.hpp index a224c467d..328d00619 100644 --- a/osdep/ExtOsdep.hpp +++ b/osdep/ExtOsdep.hpp @@ -1,3 +1,16 @@ +/* + * Copyright (c)2019 ZeroTier, Inc. + * + * Use of this software is governed by the Business Source License included + * in the LICENSE.TXT file in the project's root directory. + * + * Change Date: 2026-01-01 + * + * On the date above, in accordance with the Business Source License, use + * of this software will be governed by version 2.0 of the Apache License. + */ +/****/ + #ifndef ZT_EXTOSDEP_HPP #define ZT_EXTOSDEP_HPP @@ -5,25 +18,25 @@ #define ZT_EOD_MAXMSGSIZE (64 * 1024) -#define ZT_EOD_MSG_STARTED 1 // no data -#define ZT_EOD_MSG_ADDTAP 2 -#define ZT_EOD_MSG_ADDTAPRESP 3 -#define ZT_EOD_MSG_DELTAP 4 -#define ZT_EOD_MSG_DELTAPRESP 5 // no data -#define ZT_EOD_MSG_SETMTU 6 -#define ZT_EOD_MSG_SETMTURESP 7 -#define ZT_EOD_MSG_ADDIP 8 -#define ZT_EOD_MSG_ADDIPRESP 9 -#define ZT_EOD_MSG_DELIP 10 -#define ZT_EOD_MSG_DELIPRESP 11 -#define ZT_EOD_MSG_GETIPS 12 -#define ZT_EOD_MSG_GETIPSRESP 13 -#define ZT_EOD_MSG_GETBINDADDRS 14 +#define ZT_EOD_MSG_STARTED 1 // no data +#define ZT_EOD_MSG_ADDTAP 2 +#define ZT_EOD_MSG_ADDTAPRESP 3 +#define ZT_EOD_MSG_DELTAP 4 +#define ZT_EOD_MSG_DELTAPRESP 5 // no data +#define ZT_EOD_MSG_SETMTU 6 +#define ZT_EOD_MSG_SETMTURESP 7 +#define ZT_EOD_MSG_ADDIP 8 +#define ZT_EOD_MSG_ADDIPRESP 9 +#define ZT_EOD_MSG_DELIP 10 +#define ZT_EOD_MSG_DELIPRESP 11 +#define ZT_EOD_MSG_GETIPS 12 +#define ZT_EOD_MSG_GETIPSRESP 13 +#define ZT_EOD_MSG_GETBINDADDRS 14 #define ZT_EOD_MSG_GETBINDADDRSRESP 15 -#define ZT_EOD_MSG_ADDROUTE 16 -#define ZT_EOD_MSG_ADDROUTERESP 17 -#define ZT_EOD_MSG_DELROUTE 18 -#define ZT_EOD_MSG_DELROUTERESP 19 +#define ZT_EOD_MSG_ADDROUTE 16 +#define ZT_EOD_MSG_ADDROUTERESP 17 +#define ZT_EOD_MSG_DELROUTE 18 +#define ZT_EOD_MSG_DELROUTERESP 19 struct zt_eod_msg_addtap { unsigned char cmd; @@ -53,8 +66,8 @@ struct zt_eod_msg_setmtu { struct zt_eod_msg_ip { unsigned char cmd; char name[16]; - unsigned char afi; // 1 ip, 2 ip6 - unsigned char len; // bits in mask + unsigned char afi; // 1 ip, 2 ip6 + unsigned char len; // bits in mask unsigned char data[16]; } __attribute__((packed)); @@ -87,7 +100,7 @@ struct zt_eod_msg_getbindaddrsresp { struct zt_eod_msg_route { unsigned char cmd; - unsigned char afi; // 1 ip, 2 ip6 + unsigned char afi; // 1 ip, 2 ip6 unsigned char dstlen; unsigned char dst[16]; unsigned char gw[16]; @@ -110,89 +123,91 @@ struct zt_eod_mgmt_reply { #ifndef ZT_EXTOSDEP_IFACEONLY +#include "../node/AtomicCounter.hpp" +#include "../node/Hashtable.hpp" #include "../node/InetAddress.hpp" #include "../node/MAC.hpp" -#include "Thread.hpp" -#include "../node/Hashtable.hpp" #include "../node/Mutex.hpp" -#include "../node/AtomicCounter.hpp" -#include "EthernetTap.hpp" #include "BlockingQueue.hpp" +#include "EthernetTap.hpp" +#include "Thread.hpp" + #include -#include -#include #include +#include #include +#include namespace ZeroTier { class ExtOsdep { -public: + public: static void init(int, int); - static void started(int *, void **); + static void started(int*, void**); - static void routeAddDel(bool, const InetAddress &target, const InetAddress &via, const InetAddress &src, const char *ifaceName); - static bool getBindAddrs(std::map &); + static void routeAddDel(bool, const InetAddress& target, const InetAddress& via, const InetAddress& src, const char* ifaceName); + static bool getBindAddrs(std::map&); - static bool mgmtRecv(void *cookie, void *data, unsigned long len, - std::function); - static bool mgmtWritable(void *); + static bool mgmtRecv(void* cookie, void* data, unsigned long len, std::function); + static bool mgmtWritable(void*); }; -class ExtOsdepTap : public EthernetTap -{ -public: +class ExtOsdepTap : public EthernetTap { + public: ExtOsdepTap( - const char *homePath, - const MAC &mac, + const char* homePath, + const MAC& mac, unsigned int mtu, unsigned int metric, uint64_t nwid, - const char *friendlyName, - void (*handler)(void *,void *,uint64_t,const MAC &,const MAC &,unsigned int,unsigned int,const void *,unsigned int), - void *arg); + const char* friendlyName, + void (*handler)(void*, void*, uint64_t, const MAC&, const MAC&, unsigned int, unsigned int, const void*, unsigned int), + void* arg); virtual ~ExtOsdepTap(); virtual void setEnabled(bool en); virtual bool enabled() const; - virtual bool addIp(const InetAddress &ip); + virtual bool addIp(const InetAddress& ip); virtual bool addIps(std::vector ips); - virtual bool removeIp(const InetAddress &ip); + virtual bool removeIp(const InetAddress& ip); virtual std::vector ips() const; - virtual void put(const MAC &from,const MAC &to,unsigned int etherType,const void *data,unsigned int len); + virtual void put(const MAC& from, const MAC& to, unsigned int etherType, const void* data, unsigned int len); virtual std::string deviceName() const; - virtual void setFriendlyName(const char *friendlyName); - virtual void scanMulticastGroups(std::vector &added,std::vector &removed); + virtual void setFriendlyName(const char* friendlyName); + virtual void scanMulticastGroups(std::vector& added, std::vector& removed); virtual void setMtu(unsigned int mtu); - virtual void setDns(const char *domain, const std::vector &servers) {} -private: - void (*_handler)(void *,void *,uint64_t,const MAC &,const MAC &,unsigned int,unsigned int,const void *,unsigned int); - void *_arg; - uint64_t _nwid; - MAC _mac; - std::string _homePath; - std::string _dev; - std::vector _multicastGroups; - unsigned int _mtu; + virtual void setDns(const char* domain, const std::vector& servers) + { + } + + private: + void (*_handler)(void*, void*, uint64_t, const MAC&, const MAC&, unsigned int, unsigned int, const void*, unsigned int); + void* _arg; + uint64_t _nwid; + MAC _mac; + std::string _homePath; + std::string _dev; + std::vector _multicastGroups; + unsigned int _mtu; int _fd; - int _shutdownSignalPipe[2]; - std::atomic_bool _enabled; - std::atomic_bool _run; - std::thread _tapReaderThread[2]; - std::thread _tapProcessorThread; - std::mutex _buffers_l; - std::vector _buffers; - BlockingQueue< std::pair > _tapq; + int _shutdownSignalPipe[2]; + std::atomic_bool _enabled; + std::atomic_bool _run; + std::thread _tapReaderThread[2]; + std::thread _tapProcessorThread; + std::mutex _buffers_l; + std::vector _buffers; + BlockingQueue > _tapq; AtomicCounter _tapqsize; std::vector allIps; - void doRemoveIp(const InetAddress &); + void doRemoveIp(const InetAddress&); }; -} // namespace ZeroTier +} // namespace ZeroTier -#endif // ZT_EXTOSDEP_IFACEONLY -#endif // ZT_EXTOSDEP +#endif // ZT_EXTOSDEP_IFACEONLY +#endif // ZT_EXTOSDEP #endif