123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852 |
- /*
- * 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: 2023-01-01
- *
- * On the date above, in accordance with the Business Source License, use
- * of this software will be governed by vergnn 2.0 of the Apache License.
- */
- /****/
- #include "GoGlue.h"
- #include "../../node/Constants.hpp"
- #include "../../node/InetAddress.hpp"
- #include "../../node/Node.hpp"
- #include "../../node/Utils.hpp"
- #include "../../node/MAC.hpp"
- #include "../../node/Address.hpp"
- #include "../../node/Locator.hpp"
- #include "../../osdep/OSUtils.hpp"
- #include "../../osdep/EthernetTap.hpp"
- #include "../../osdep/ManagedRoute.hpp"
- #include <string.h>
- #include <stdlib.h>
- #include <errno.h>
- #ifndef __WINDOWS__
- #include <errno.h>
- #include <signal.h>
- #include <unistd.h>
- #include <fcntl.h>
- #include <sys/time.h>
- #include <sys/types.h>
- #include <sys/select.h>
- #include <sys/socket.h>
- #include <sys/un.h>
- #include <arpa/inet.h>
- #include <netinet/in.h>
- #include <netinet/ip6.h>
- #include <netinet/tcp.h>
- #ifdef __BSD__
- #include <net/if.h>
- #endif
- #ifdef __LINUX__
- #ifndef IPV6_DONTFRAG
- #define IPV6_DONTFRAG 62
- #endif
- #endif
- #endif // !__WINDOWS__
- #include <thread>
- #include <mutex>
- #include <map>
- #include <vector>
- #include <array>
- #include <set>
- #include <memory>
- #include <atomic>
- #ifdef __WINDOWS__
- #define SETSOCKOPT_FLAG_TYPE BOOL
- #define SETSOCKOPT_FLAG_TRUE TRUE
- #define SETSOCKOPT_FLAG_FALSE FALSE
- #else
- #define SETSOCKOPT_FLAG_TYPE int
- #define SETSOCKOPT_FLAG_TRUE 1
- #define SETSOCKOPT_FLAG_FALSE 0
- #endif
- #ifndef MSG_DONTWAIT
- #define MSG_DONTWAIT 0
- #endif
- using namespace ZeroTier;
- struct ZT_GoNodeThread
- {
- std::string ip;
- int port;
- int af;
- std::atomic<bool> run;
- std::thread thr;
- };
- struct ZT_GoNode_Impl
- {
- Node *node;
- volatile int64_t nextBackgroundTaskDeadline;
- std::string path;
- std::atomic<bool> run;
- std::map< ZT_SOCKET,ZT_GoNodeThread > threads;
- std::mutex threads_l;
- std::map< uint64_t,std::shared_ptr<EthernetTap> > taps;
- std::mutex taps_l;
- std::thread backgroundTaskThread;
- };
- static const std::string defaultHomePath(OSUtils::platformDefaultHomePath());
- const char *ZT_PLATFORM_DEFAULT_HOMEPATH = defaultHomePath.c_str();
- /****************************************************************************/
- /* These functions are implemented in Go in pkg/ztnode/node-callbacks.go */
- extern "C" int goPathCheckFunc(ZT_GoNode *,uint64_t,int,const void *,int);
- extern "C" int goPathLookupFunc(ZT_GoNode *,uint64_t,int,int *,uint8_t [16],int *);
- extern "C" void goStateObjectPutFunc(ZT_GoNode *,int,const uint64_t [2],const void *,int);
- extern "C" int goStateObjectGetFunc(ZT_GoNode *,int,const uint64_t [2],void *,unsigned int);
- extern "C" void goDNSResolverFunc(ZT_GoNode *,const uint8_t *,int,const char *,uintptr_t);
- extern "C" void goVirtualNetworkConfigFunc(ZT_GoNode *,ZT_GoTap *,uint64_t,int,const ZT_VirtualNetworkConfig *);
- extern "C" void goZtEvent(ZT_GoNode *,int,const void *);
- extern "C" void goHandleTapAddedMulticastGroup(ZT_GoNode *,ZT_GoTap *,uint64_t,uint64_t,uint32_t);
- extern "C" void goHandleTapRemovedMulticastGroup(ZT_GoNode *,ZT_GoTap *,uint64_t,uint64_t,uint32_t);
- static void ZT_GoNode_VirtualNetworkConfigFunction(
- ZT_Node *node,
- void *uptr,
- void *tptr,
- uint64_t nwid,
- void **nptr,
- enum ZT_VirtualNetworkConfigOperation op,
- const ZT_VirtualNetworkConfig *cfg)
- {
- goVirtualNetworkConfigFunc(reinterpret_cast<ZT_GoNode *>(uptr),reinterpret_cast<ZT_GoTap *>(*nptr),nwid,op,cfg);
- }
- static void ZT_GoNode_VirtualNetworkFrameFunction(
- ZT_Node *node,
- void *uptr,
- void *tptr,
- uint64_t nwid,
- void **nptr,
- uint64_t srcMac,
- uint64_t destMac,
- unsigned int etherType,
- unsigned int vlanId,
- const void *data,
- unsigned int len)
- {
- if (*nptr)
- reinterpret_cast<EthernetTap *>(*nptr)->put(MAC(srcMac),MAC(destMac),etherType,data,len);
- }
- static void ZT_GoNode_EventCallback(
- ZT_Node *node,
- void *uptr,
- void *tptr,
- enum ZT_Event et,
- const void *data)
- {
- goZtEvent(reinterpret_cast<ZT_GoNode *>(uptr),et,data);
- }
- static void ZT_GoNode_StatePutFunction(
- ZT_Node *node,
- void *uptr,
- void *tptr,
- enum ZT_StateObjectType objType,
- const uint64_t id[2],
- const void *data,
- int len)
- {
- goStateObjectPutFunc(reinterpret_cast<ZT_GoNode *>(uptr),objType,id,data,len);
- }
- static int ZT_GoNode_StateGetFunction(
- ZT_Node *node,
- void *uptr,
- void *tptr,
- enum ZT_StateObjectType objType,
- const uint64_t id[2],
- void *buf,
- unsigned int buflen)
- {
- return goStateObjectGetFunc(
- reinterpret_cast<ZT_GoNode *>(uptr),
- (int)objType,
- id,
- buf,
- buflen);
- }
- static ZT_ALWAYS_INLINE void doUdpSend(ZT_SOCKET sock,const struct sockaddr_storage *addr,const void *data,const unsigned int len,const unsigned int ipTTL)
- {
- switch(addr->ss_family) {
- case AF_INET:
- if ((ipTTL > 0)&&(ipTTL < 255)) {
- #ifdef __WINDOWS__
- DWORD tmp = (DWORD)ipTTL;
- #else
- int tmp = (int)ipTTL;
- #endif
- setsockopt(sock,IPPROTO_IP,IP_TTL,&tmp,sizeof(tmp));
- sendto(sock,data,len,MSG_DONTWAIT,(const sockaddr *)addr,sizeof(struct sockaddr_in));
- tmp = 255;
- setsockopt(sock,IPPROTO_IP,IP_TTL,&tmp,sizeof(tmp));
- } else {
- sendto(sock,data,len,MSG_DONTWAIT,(const sockaddr *)addr,sizeof(struct sockaddr_in));
- }
- break;
- case AF_INET6:
- // The ipTTL option isn't currently used with IPv6. It's only used
- // with IPv4 "firewall opener" / "NAT buster" preamble packets as part
- // of IPv4 NAT traversal.
- sendto(sock,data,len,MSG_DONTWAIT,(const sockaddr *)addr,sizeof(struct sockaddr_in6));
- break;
- }
- }
- static int ZT_GoNode_WirePacketSendFunction(
- ZT_Node *node,
- void *uptr,
- void *tptr,
- int64_t localSocket,
- const struct sockaddr_storage *addr,
- const void *data,
- unsigned int len,
- unsigned int ipTTL)
- {
- if ((localSocket != -1)&&(localSocket != ZT_INVALID_SOCKET)) {
- doUdpSend((ZT_SOCKET)localSocket,addr,data,len,ipTTL);
- } else {
- ZT_GoNode *const gn = reinterpret_cast<ZT_GoNode *>(uptr);
- std::set<std::string> ipsSentFrom;
- std::lock_guard<std::mutex> l(gn->threads_l);
- for(auto t=gn->threads.begin();t!=gn->threads.end();++t) {
- if (t->second.af == addr->ss_family) {
- if (ipsSentFrom.insert(t->second.ip).second) {
- doUdpSend(t->first,addr,data,len,ipTTL);
- }
- }
- }
- }
- return 0;
- }
- static int ZT_GoNode_PathCheckFunction(
- ZT_Node *node,
- void *uptr,
- void *tptr,
- uint64_t ztAddress,
- int64_t localSocket,
- const struct sockaddr_storage *sa)
- {
- switch(sa->ss_family) {
- case AF_INET:
- return goPathCheckFunc(
- reinterpret_cast<ZT_GoNode *>(uptr),
- ztAddress,
- AF_INET,
- &(reinterpret_cast<const struct sockaddr_in *>(sa)->sin_addr.s_addr),
- Utils::ntoh((uint16_t)reinterpret_cast<const struct sockaddr_in *>(sa)->sin_port));
- case AF_INET6:
- return goPathCheckFunc(
- reinterpret_cast<ZT_GoNode *>(uptr),
- ztAddress,
- AF_INET6,
- reinterpret_cast<const struct sockaddr_in6 *>(sa)->sin6_addr.s6_addr,
- Utils::ntoh((uint16_t)reinterpret_cast<const struct sockaddr_in6 *>(sa)->sin6_port));
- }
- return 0;
- }
- static int ZT_GoNode_PathLookupFunction(
- ZT_Node *node,
- void *uptr,
- void *tptr,
- uint64_t ztAddress,
- int desiredAddressFamily,
- struct sockaddr_storage *sa)
- {
- int family = 0;
- uint8_t ip[16];
- int port = 0;
- const int result = goPathLookupFunc(
- reinterpret_cast<ZT_GoNode *>(uptr),
- ztAddress,
- desiredAddressFamily,
- &family,
- ip,
- &port
- );
- if (result != 0) {
- switch(family) {
- case AF_INET:
- reinterpret_cast<struct sockaddr_in *>(sa)->sin_family = AF_INET;
- memcpy(&(reinterpret_cast<struct sockaddr_in *>(sa)->sin_addr.s_addr),ip,4);
- reinterpret_cast<struct sockaddr_in *>(sa)->sin_port = Utils::hton((uint16_t)port);
- return 1;
- case AF_INET6:
- reinterpret_cast<struct sockaddr_in6 *>(sa)->sin6_family = AF_INET6;
- memcpy(reinterpret_cast<struct sockaddr_in6 *>(sa)->sin6_addr.s6_addr,ip,16);
- reinterpret_cast<struct sockaddr_in6 *>(sa)->sin6_port = Utils::hton((uint16_t)port);
- return 1;
- }
- }
- return 0;
- }
- static void ZT_GoNode_DNSResolver(
- ZT_Node *node,
- void *uptr,
- void *tptr,
- const enum ZT_DNSRecordType *types,
- unsigned int numTypes,
- const char *name,
- uintptr_t requestId)
- {
- uint8_t t[256];
- for(unsigned int i=0;(i<numTypes)&&(i<256);++i) t[i] = (uint8_t)types[i];
- goDNSResolverFunc(reinterpret_cast<ZT_GoNode *>(uptr),t,(int)numTypes,name,requestId);
- }
- /****************************************************************************/
- extern "C" ZT_GoNode *ZT_GoNode_new(const char *workingPath)
- {
- try {
- struct ZT_Node_Callbacks cb;
- cb.statePutFunction = &ZT_GoNode_StatePutFunction;
- cb.stateGetFunction = &ZT_GoNode_StateGetFunction;
- cb.wirePacketSendFunction = &ZT_GoNode_WirePacketSendFunction;
- cb.virtualNetworkFrameFunction = &ZT_GoNode_VirtualNetworkFrameFunction;
- cb.virtualNetworkConfigFunction = &ZT_GoNode_VirtualNetworkConfigFunction;
- cb.eventCallback = &ZT_GoNode_EventCallback;
- cb.dnsResolver = &ZT_GoNode_DNSResolver;
- cb.pathCheckFunction = &ZT_GoNode_PathCheckFunction;
- cb.pathLookupFunction = &ZT_GoNode_PathLookupFunction;
- ZT_GoNode_Impl *gn = new ZT_GoNode_Impl;
- const int64_t now = OSUtils::now();
- gn->node = new Node(reinterpret_cast<void *>(gn),nullptr,&cb,now);
- gn->nextBackgroundTaskDeadline = now;
- gn->path = workingPath;
- gn->run = true;
- gn->backgroundTaskThread = std::thread([gn] {
- int64_t lastCheckedTaps = 0;
- while (gn->run) {
- std::this_thread::sleep_for(std::chrono::milliseconds(500));
- const int64_t now = OSUtils::now();
- if (now >= gn->nextBackgroundTaskDeadline)
- gn->node->processBackgroundTasks(nullptr,now,&(gn->nextBackgroundTaskDeadline));
- if ((now - lastCheckedTaps) > 10000) {
- lastCheckedTaps = now;
- std::vector<MulticastGroup> added,removed;
- std::lock_guard<std::mutex> tl(gn->taps_l);
- for(auto t=gn->taps.begin();t!=gn->taps.end();++t) {
- added.clear();
- removed.clear();
- t->second->scanMulticastGroups(added,removed);
- for(auto g=added.begin();g!=added.end();++g)
- goHandleTapAddedMulticastGroup(gn,(ZT_GoTap *)t->second.get(),t->first,g->mac().toInt(),g->adi());
- for(auto g=removed.begin();g!=removed.end();++g)
- goHandleTapRemovedMulticastGroup(gn,(ZT_GoTap *)t->second.get(),t->first,g->mac().toInt(),g->adi());
- t->second->syncRoutes();
- }
- }
- }
- });
- return gn;
- } catch ( ... ) {
- fprintf(stderr,"FATAL: unable to create new instance of Node (out of memory?)" ZT_EOL_S);
- exit(1);
- }
- }
- extern "C" void ZT_GoNode_delete(ZT_GoNode *gn)
- {
- gn->run = false;
- gn->threads_l.lock();
- for(auto t=gn->threads.begin();t!=gn->threads.end();++t) {
- t->second.run = false;
- shutdown(t->first,SHUT_RDWR);
- close(t->first);
- t->second.thr.join();
- }
- gn->threads_l.unlock();
- gn->taps_l.lock();
- for(auto t=gn->taps.begin();t!=gn->taps.end();++t)
- gn->node->leave(t->first,nullptr,nullptr);
- gn->taps.clear();
- gn->taps_l.unlock();
- gn->backgroundTaskThread.join();
- delete gn->node;
- delete gn;
- }
- extern "C" ZT_Node *ZT_GoNode_getNode(ZT_GoNode *gn)
- {
- return gn->node;
- }
- // Sets flags and socket options common to both IPv4 and IPv6 UDP sockets
- static void setCommonUdpSocketSettings(ZT_SOCKET udpSock,const char *dev)
- {
- int bufSize = 1048576;
- while (bufSize > 131072) {
- if (setsockopt(udpSock,SOL_SOCKET,SO_RCVBUF,(const char *)&bufSize,sizeof(bufSize)) == 0)
- break;
- bufSize -= 131072;
- }
- bufSize = 1048576;
- while (bufSize > 131072) {
- if (setsockopt(udpSock,SOL_SOCKET,SO_SNDBUF,(const char *)&bufSize,sizeof(bufSize)) == 0)
- break;
- bufSize -= 131072;
- }
- SETSOCKOPT_FLAG_TYPE fl;
- #ifdef SO_REUSEPORT
- fl = SETSOCKOPT_FLAG_TRUE;
- setsockopt(udpSock,SOL_SOCKET,SO_REUSEPORT,(void *)&fl,sizeof(fl));
- #endif
- #ifndef __LINUX__ // linux wants just SO_REUSEPORT
- fl = SETSOCKOPT_FLAG_TRUE;
- setsockopt(udpSock,SOL_SOCKET,SO_REUSEADDR,(void *)&fl,sizeof(fl));
- #endif
- fl = SETSOCKOPT_FLAG_TRUE;
- setsockopt(udpSock,SOL_SOCKET,SO_BROADCAST,(void *)&fl,sizeof(fl));
- #ifdef IP_DONTFRAG
- fl = SETSOCKOPT_FLAG_FALSE;
- setsockopt(udpSock,IPPROTO_IP,IP_DONTFRAG,(void *)&fl,sizeof(fl));
- #endif
- #ifdef IP_MTU_DISCOVER
- fl = SETSOCKOPT_FLAG_FALSE;
- setsockopt(udpSock,IPPROTO_IP,IP_MTU_DISCOVER,(void *)&fl,sizeof(fl));
- #endif
- #ifdef SO_BINDTODEVICE
- if ((dev)&&(strlen(dev)))
- setsockopt(udpSock,SOL_SOCKET,SO_BINDTODEVICE,dev,strlen(dev));
- #endif
- #if defined(__BSD__) && defined(IP_BOUND_IF)
- if ((dev)&&(strlen(dev))) {
- int idx = if_nametoindex(dev);
- if (idx != 0)
- setsockopt(udpSock,IPPROTO_IP,IP_BOUND_IF,(void *)&idx,sizeof(idx));
- }
- #endif
- }
- extern "C" int ZT_GoNode_phyStartListen(ZT_GoNode *gn,const char *dev,const char *ip,const int port)
- {
- if (strchr(ip,':')) {
- struct sockaddr_in6 in6;
- memset(&in6,0,sizeof(in6));
- in6.sin6_family = AF_INET6;
- if (inet_pton(AF_INET6,ip,&(in6.sin6_addr)) <= 0)
- return errno;
- in6.sin6_port = htons((uint16_t)port);
- ZT_SOCKET udpSock = socket(AF_INET6,SOCK_DGRAM,0);
- if (udpSock == ZT_INVALID_SOCKET)
- return errno;
- setCommonUdpSocketSettings(udpSock,dev);
- SETSOCKOPT_FLAG_TYPE fl = SETSOCKOPT_FLAG_TRUE;
- setsockopt(udpSock,IPPROTO_IPV6,IPV6_V6ONLY,(const char *)&fl,sizeof(fl));
- #ifdef IPV6_DONTFRAG
- fl = SETSOCKOPT_FLAG_FALSE;
- setsockopt(udpSock,IPPROTO_IPV6,IPV6_DONTFRAG,&fl,sizeof(fl));
- #endif
- if (bind(udpSock,reinterpret_cast<const struct sockaddr *>(&in6),sizeof(in6)) != 0)
- return errno;
- {
- std::lock_guard<std::mutex> l(gn->threads_l);
- ZT_GoNodeThread &gnt = gn->threads[udpSock];
- gnt.ip = ip;
- gnt.port = port;
- gnt.af = AF_INET6;
- gnt.run = true;
- gnt.thr = std::thread([udpSock,gn,&gnt] {
- struct sockaddr_in6 in6;
- socklen_t salen;
- char buf[16384];
- while (gnt.run) {
- salen = sizeof(in6);
- int s = (int)recvfrom(udpSock,buf,sizeof(buf),0,reinterpret_cast<struct sockaddr *>(&in6),&salen);
- if (s > 0) {
- gn->node->processWirePacket(&gnt,OSUtils::now(),(int64_t)udpSock,reinterpret_cast<const struct sockaddr_storage *>(&in6),buf,(unsigned int)s,&(gn->nextBackgroundTaskDeadline));
- } else {
- // If something goes bad with this socket such as its interface vanishing, it
- // will eventually be closed by higher level (Go) code. Until then prevent the
- // system from consuming too much CPU.
- std::this_thread::sleep_for(std::chrono::milliseconds(10));
- }
- }
- });
- }
- } else {
- struct sockaddr_in in;
- memset(&in,0,sizeof(in));
- in.sin_family = AF_INET;
- if (inet_pton(AF_INET,ip,&(in.sin_addr)) <= 0)
- return errno;
- in.sin_port = htons((uint16_t)port);
- ZT_SOCKET udpSock = socket(AF_INET,SOCK_DGRAM,0);
- if (udpSock == ZT_INVALID_SOCKET)
- return errno;
- setCommonUdpSocketSettings(udpSock,dev);
- #ifdef SO_NO_CHECK
- SETSOCKOPT_FLAG_TYPE fl = SETSOCKOPT_FLAG_TRUE;
- setsockopt(udpSock,SOL_SOCKET,SO_NO_CHECK,&fl,sizeof(fl));
- #endif
- if (bind(udpSock,reinterpret_cast<const struct sockaddr *>(&in),sizeof(in)) != 0)
- return errno;
- {
- std::lock_guard<std::mutex> l(gn->threads_l);
- ZT_GoNodeThread &gnt = gn->threads[udpSock];
- gnt.ip = ip;
- gnt.port = port;
- gnt.af = AF_INET6;
- gnt.run = true;
- gnt.thr = std::thread([udpSock,gn,&gnt] {
- struct sockaddr_in in4;
- socklen_t salen;
- char buf[16384];
- while (gnt.run) {
- salen = sizeof(in4);
- int s = (int)recvfrom(udpSock,buf,sizeof(buf),0,reinterpret_cast<struct sockaddr *>(&in4),&salen);
- if (s > 0) {
- gn->node->processWirePacket(&gnt,OSUtils::now(),(int64_t)udpSock,reinterpret_cast<const struct sockaddr_storage *>(&in4),buf,(unsigned int)s,&(gn->nextBackgroundTaskDeadline));
- }
- }
- });
- }
- }
- return 0;
- }
- extern "C" int ZT_GoNode_phyStopListen(ZT_GoNode *gn,const char *dev,const char *ip,const int port)
- {
- {
- std::lock_guard<std::mutex> l(gn->threads_l);
- for(auto t=gn->threads.begin();t!=gn->threads.end();) {
- if ((t->second.ip == ip)&&(t->second.port == port)) {
- t->second.run = false;
- shutdown(t->first,SHUT_RDWR);
- close(t->first);
- t->second.thr.join();
- gn->threads.erase(t++);
- } else ++t;
- }
- }
- return 0;
- }
- static void tapFrameHandler(void *uptr,void *tptr,uint64_t nwid,const MAC &from,const MAC &to,unsigned int etherType,unsigned int vlanId,const void *data,unsigned int len)
- {
- ZT_GoNode *const gn = reinterpret_cast<ZT_GoNode *>(uptr);
- gn->node->processVirtualNetworkFrame(tptr,OSUtils::now(),nwid,from.toInt(),to.toInt(),etherType,vlanId,data,len,&(gn->nextBackgroundTaskDeadline));
- }
- extern "C" ZT_GoTap *ZT_GoNode_join(ZT_GoNode *gn,uint64_t nwid)
- {
- try {
- std::lock_guard<std::mutex> l(gn->taps_l);
- auto existingTap = gn->taps.find(nwid);
- if (existingTap != gn->taps.end())
- return (ZT_GoTap *)existingTap->second.get();
- char tmp[256];
- OSUtils::ztsnprintf(tmp,sizeof(tmp),"ZeroTier Network %.16llx",(unsigned long long)nwid);
- std::shared_ptr<EthernetTap> tap(EthernetTap::newInstance(nullptr,gn->path.c_str(),MAC(Address(gn->node->address()),nwid),ZT_DEFAULT_MTU,0,nwid,tmp,&tapFrameHandler,gn));
- if (!tap)
- return nullptr;
- gn->taps[nwid] = tap;
- gn->node->join(nwid,tap.get(),nullptr);
- return (ZT_GoTap *)tap.get();
- } catch ( ... ) {
- return nullptr;
- }
- }
- extern "C" void ZT_GoNode_leave(ZT_GoNode *gn,uint64_t nwid)
- {
- std::lock_guard<std::mutex> l(gn->taps_l);
- auto existingTap = gn->taps.find(nwid);
- if (existingTap != gn->taps.end()) {
- gn->node->leave(nwid,nullptr,nullptr);
- gn->taps.erase(existingTap);
- }
- }
- /****************************************************************************/
- extern "C" void ZT_GoTap_setEnabled(ZT_GoTap *tap,int enabled)
- {
- reinterpret_cast<EthernetTap *>(tap)->setEnabled(enabled != 0);
- }
- extern "C" int ZT_GoTap_addIp(ZT_GoTap *tap,int af,const void *ip,int netmaskBits)
- {
- switch(af) {
- case AF_INET:
- return (reinterpret_cast<EthernetTap *>(tap)->addIp(InetAddress(ip,4,(unsigned int)netmaskBits)) ? 1 : 0);
- case AF_INET6:
- return (reinterpret_cast<EthernetTap *>(tap)->addIp(InetAddress(ip,16,(unsigned int)netmaskBits)) ? 1 : 0);
- }
- return 0;
- }
- extern "C" int ZT_GoTap_removeIp(ZT_GoTap *tap,int af,const void *ip,int netmaskBits)
- {
- switch(af) {
- case AF_INET:
- return (reinterpret_cast<EthernetTap *>(tap)->removeIp(InetAddress(ip,4,(unsigned int)netmaskBits)) ? 1 : 0);
- case AF_INET6:
- return (reinterpret_cast<EthernetTap *>(tap)->removeIp(InetAddress(ip,16,(unsigned int)netmaskBits)) ? 1 : 0);
- }
- return 0;
- }
- extern "C" int ZT_GoTap_ips(ZT_GoTap *tap,void *buf,unsigned int bufSize)
- {
- auto ips = reinterpret_cast<EthernetTap *>(tap)->ips();
- unsigned int p = 0;
- uint8_t *const b = reinterpret_cast<uint8_t *>(buf);
- for(auto ip=ips.begin();ip!=ips.end();++ip) {
- if ((p + 6) > bufSize)
- break;
- const uint8_t *const ipd = reinterpret_cast<const uint8_t *>(ip->rawIpData());
- if (ip->isV4()) {
- b[p++] = AF_INET;
- b[p++] = ipd[0];
- b[p++] = ipd[1];
- b[p++] = ipd[2];
- b[p++] = ipd[3];
- b[p++] = (uint8_t)ip->netmaskBits();
- } else if (ip->isV6()) {
- if ((p + 18) <= bufSize) {
- b[p++] = AF_INET6;
- for(int j=0;j<16;++j)
- b[p++] = ipd[j];
- b[p++] = (uint8_t)ip->netmaskBits();
- }
- }
- }
- return (int)p;
- }
- extern "C" void ZT_GoTap_deviceName(ZT_GoTap *tap,char nbuf[256])
- {
- Utils::scopy(nbuf,256,reinterpret_cast<EthernetTap *>(tap)->deviceName().c_str());
- }
- extern "C" void ZT_GoTap_setFriendlyName(ZT_GoTap *tap,const char *friendlyName)
- {
- reinterpret_cast<EthernetTap *>(tap)->setFriendlyName(friendlyName);
- }
- extern "C" void ZT_GoTap_setMtu(ZT_GoTap *tap,unsigned int mtu)
- {
- reinterpret_cast<EthernetTap *>(tap)->setMtu(mtu);
- }
- extern "C" int ZT_GoTap_addRoute(ZT_GoTap *tap,int targetAf,const void *targetIp,int targetNetmaskBits,int viaAf,const void *viaIp,unsigned int metric)
- {
- InetAddress target,via;
- switch(targetAf) {
- case AF_INET:
- target.set(targetIp,4,(unsigned int)targetNetmaskBits);
- break;
- case AF_INET6:
- target.set(targetIp,16,(unsigned int)targetNetmaskBits);
- break;
- }
- switch(viaAf) {
- case AF_INET:
- via.set(viaIp,4,0);
- break;
- case AF_INET6:
- via.set(viaIp,16,0);
- break;
- }
- return reinterpret_cast<EthernetTap *>(tap)->addRoute(target,via,metric);
- }
- extern "C" int ZT_GoTap_removeRoute(ZT_GoTap *tap,int targetAf,const void *targetIp,int targetNetmaskBits,int viaAf,const void *viaIp,unsigned int metric)
- {
- InetAddress target,via;
- switch(targetAf) {
- case AF_INET:
- target.set(targetIp,4,(unsigned int)targetNetmaskBits);
- break;
- case AF_INET6:
- target.set(targetIp,16,(unsigned int)targetNetmaskBits);
- break;
- }
- switch(viaAf) {
- case AF_INET:
- via.set(viaIp,4,0);
- break;
- case AF_INET6:
- via.set(viaIp,16,0);
- break;
- }
- return reinterpret_cast<EthernetTap *>(tap)->removeRoute(target,via,metric);
- }
- /****************************************************************************/
- extern "C" const char *ZT_GoIdentity_generate(int type)
- {
- Identity id;
- id.generate((Identity::Type)type);
- char *tmp = (char *)malloc(ZT_IDENTITY_STRING_BUFFER_LENGTH);
- if (tmp)
- id.toString(true,tmp);
- return tmp;
- }
- extern "C" int ZT_GoIdentity_validate(const char *idStr)
- {
- Identity id;
- if (!id.fromString(idStr))
- return 0;
- if (!id.locallyValidate())
- return 0;
- return 1;
- }
- extern "C" int ZT_GoIdentity_sign(const char *idStr,const void *data,unsigned int len,void *sigbuf,unsigned int sigbuflen)
- {
- Identity id;
- if (!id.fromString(idStr))
- return 0;
- return (int)id.sign(data,len,sigbuf,sigbuflen);
- }
- extern "C" int ZT_GoIdentity_verify(const char *idStr,const void *data,unsigned int len,const void *sig,unsigned int siglen)
- {
- Identity id;
- if (!id.fromString(idStr))
- return 0;
- return id.verify(data,len,sig,siglen) ? 1 : 0;
- }
- /****************************************************************************/
- extern "C" int ZT_GoLocator_makeSecureDNSName(char *name,unsigned int nameBufSize,uint8_t *privateKey,unsigned int privateKeyBufSize)
- {
- if ((privateKeyBufSize < ZT_ECC384_PRIVATE_KEY_SIZE)||(nameBufSize < 256))
- return -1;
- uint8_t pub[ZT_ECC384_PUBLIC_KEY_SIZE];
- ECC384GenerateKey(pub,privateKey);
- const Str n(Locator::makeSecureDnsName(pub));
- if (n.length() >= nameBufSize)
- return -1;
- Utils::scopy(name,nameBufSize,n.c_str());
- return ZT_ECC384_PRIVATE_KEY_SIZE;
- }
- extern "C" int ZT_GoLocator_makeLocator(
- uint8_t *buf,
- unsigned int bufSize,
- int64_t ts,
- const char *id,
- const struct sockaddr_storage *physicalAddresses,
- unsigned int physicalAddressCount,
- const char **virtualAddresses,
- unsigned int virtualAddressCount)
- {
- Locator loc;
- for(unsigned int i=0;i<physicalAddressCount;++i) {
- loc.add(*reinterpret_cast<const InetAddress *>(physicalAddresses + i));
- }
- for(unsigned int i=0;i<virtualAddressCount;++i) {
- Identity id;
- if (!id.fromString(virtualAddresses[i]))
- return -1;
- loc.add(id);
- }
- Identity signingId;
- if (!signingId.fromString(id))
- return -1;
- if (!signingId.hasPrivate())
- return -1;
- if (!loc.finish(signingId,ts))
- return -1;
- Buffer<65536> *tmp = new Buffer<65536>();
- loc.serialize(*tmp);
- if (tmp->size() > bufSize) {
- delete tmp;
- return -1;
- }
- memcpy(buf,tmp->data(),tmp->size());
- int s = (int)tmp->size();
- delete tmp;
- return s;
- }
- extern "C" int ZT_GoLocator_decodeLocator(const uint8_t *locatorBytes,unsigned int locatorSize,struct ZT_GoLocator_Info *info)
- {
- Locator loc;
- if (!loc.deserialize(locatorBytes,locatorSize))
- return -1;
- if (!loc.verify())
- return -2;
- loc.id().toString(false,info->id);
- info->phyCount = 0;
- info->virtCount = 0;
- for(auto p=loc.phy().begin();p!=loc.phy().end();++p)
- memcpy(&(info->phy[info->phyCount++]),&(*p),sizeof(struct sockaddr_storage));
- for(auto v=loc.virt().begin();v!=loc.virt().end();++v)
- v->toString(false,info->virt[info->virtCount++]);
- return 1;
- }
- int ZT_GoLocator_makeSignedTxtRecords(
- const uint8_t *locator,
- unsigned int locatorSize,
- const char *name,
- const uint8_t *privateKey,
- unsigned int privateKeySize,
- char results[256][256])
- {
- if (privateKeySize != ZT_ECC384_PRIVATE_KEY_SIZE)
- return -1;
- Locator loc;
- if (!loc.deserialize(locator,locatorSize))
- return -1;
- std::vector<Str> r(loc.makeTxtRecords(privateKey));
- if (r.size() > 256)
- return -1;
- for(unsigned long i=0;i<r.size();++i)
- Utils::scopy(results[i],256,r[i].c_str());
- return (int)r.size();
- }
|