mirror of
https://github.com/zerotier/ZeroTierOne.git
synced 2024-12-21 05:53:09 +00:00
Kill old testnet -- was useful in its time, but Docker make running real test networks locally stupid easy and a more realistic simulation.
This commit is contained in:
parent
9a34fde8a5
commit
3ac23165c7
@ -1,765 +0,0 @@
|
||||
/*
|
||||
* ZeroTier One - Network Virtualization Everywhere
|
||||
* Copyright (C) 2011-2015 ZeroTier, Inc.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* --
|
||||
*
|
||||
* ZeroTier may be used and distributed under the terms of the GPLv3, which
|
||||
* are available at: http://www.gnu.org/licenses/gpl-3.0.html
|
||||
*
|
||||
* If you would like to embed ZeroTier into a commercial application or
|
||||
* redistribute it in a modified binary form, please contact ZeroTier Networks
|
||||
* LLC. Start here: http://www.zerotier.com/
|
||||
*/
|
||||
|
||||
/* SEE: testnet/README.md */
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <stdint.h>
|
||||
#include <string.h>
|
||||
|
||||
#include <string>
|
||||
#include <map>
|
||||
#include <vector>
|
||||
#include <set>
|
||||
|
||||
#include "node/Constants.hpp"
|
||||
#include "node/Node.hpp"
|
||||
#include "node/Utils.hpp"
|
||||
#include "node/Address.hpp"
|
||||
#include "node/Identity.hpp"
|
||||
#include "node/Thread.hpp"
|
||||
#include "node/CMWC4096.hpp"
|
||||
#include "node/Dictionary.hpp"
|
||||
|
||||
#include "testnet/SimNet.hpp"
|
||||
#include "testnet/SimNetSocketManager.hpp"
|
||||
#include "testnet/TestEthernetTap.hpp"
|
||||
#include "testnet/TestEthernetTapFactory.hpp"
|
||||
|
||||
#ifdef __WINDOWS__
|
||||
#include <windows.h>
|
||||
#else
|
||||
#include <unistd.h>
|
||||
#include <sys/stat.h>
|
||||
#endif
|
||||
|
||||
using namespace ZeroTier;
|
||||
|
||||
class SimNode
|
||||
{
|
||||
public:
|
||||
SimNode(SimNet &net,const std::string &hp,const char *rootTopology,bool issn,const InetAddress &addr) :
|
||||
home(hp),
|
||||
tapFactory(),
|
||||
socketManager(net.newEndpoint(addr)),
|
||||
node(home.c_str(),&tapFactory,socketManager,false,rootTopology),
|
||||
reasonForTermination(Node::NODE_RUNNING),
|
||||
supernode(issn)
|
||||
{
|
||||
thread = Thread::start(this);
|
||||
}
|
||||
|
||||
~SimNode()
|
||||
{
|
||||
node.terminate(Node::NODE_NORMAL_TERMINATION,"SimNode shutdown");
|
||||
Thread::join(thread);
|
||||
}
|
||||
|
||||
void threadMain()
|
||||
throw()
|
||||
{
|
||||
reasonForTermination = node.run();
|
||||
}
|
||||
|
||||
std::string home;
|
||||
TestEthernetTapFactory tapFactory;
|
||||
SimNetSocketManager *socketManager;
|
||||
Node node;
|
||||
Node::ReasonForTermination reasonForTermination;
|
||||
bool supernode;
|
||||
Thread thread;
|
||||
};
|
||||
|
||||
static std::string basePath;
|
||||
static SimNet net;
|
||||
static std::map< Address,SimNode * > nodes;
|
||||
static std::map< InetAddress,Address > usedIps;
|
||||
static CMWC4096 prng;
|
||||
static std::string rootTopology;
|
||||
|
||||
// Converts an address into a fake IP not already claimed.
|
||||
// Be sure to call only once, as this claims the IP before returning it.
|
||||
static InetAddress inetAddressFromZeroTierAddress(const Address &addr)
|
||||
{
|
||||
uint32_t ip = (uint32_t)(addr.toInt() & 0xffffffff);
|
||||
for(;;) {
|
||||
if (((ip >> 24) & 0xff) >= 240) {
|
||||
ip &= 0x00ffffff;
|
||||
ip |= (((ip >> 24) & 0xff) % 240) << 24;
|
||||
}
|
||||
if (((ip >> 24) & 0xff) == 0)
|
||||
ip |= 0x01000000;
|
||||
if (((ip & 0xff) == 0)||((ip & 0xff) == 255))
|
||||
ip ^= 0x00000001;
|
||||
InetAddress inaddr(Utils::hton(ip),9993);
|
||||
if (usedIps.find(inaddr) == usedIps.end()) {
|
||||
usedIps[inaddr] = addr;
|
||||
return inaddr;
|
||||
}
|
||||
++ip; // keep looking sequentially for an unclaimed IP
|
||||
}
|
||||
}
|
||||
|
||||
static Identity makeNodeHome(bool super)
|
||||
{
|
||||
Identity id;
|
||||
id.generate();
|
||||
|
||||
std::string path(basePath + ZT_PATH_SEPARATOR_S + (super ? "S" : "N") + id.address().toString());
|
||||
|
||||
#ifdef __WINDOWS__
|
||||
CreateDirectoryA(path.c_str(),NULL);
|
||||
#else
|
||||
mkdir(path.c_str(),0700);
|
||||
#endif
|
||||
|
||||
if (!Utils::writeFile((path + ZT_PATH_SEPARATOR_S + "identity.secret").c_str(),id.toString(true)))
|
||||
return Identity();
|
||||
if (!Utils::writeFile((path + ZT_PATH_SEPARATOR_S + "identity.public").c_str(),id.toString(false)))
|
||||
return Identity();
|
||||
|
||||
return id;
|
||||
}
|
||||
|
||||
// Instantiates supernodes by scanning for S########## subdirectories
|
||||
static std::vector<Address> initSupernodes()
|
||||
{
|
||||
Dictionary supernodes;
|
||||
std::vector< std::pair<Identity,InetAddress> > snids;
|
||||
std::map<std::string,bool> dir(Utils::listDirectory(basePath.c_str()));
|
||||
|
||||
for(std::map<std::string,bool>::iterator d(dir.begin());d!=dir.end();++d) {
|
||||
if ((d->first.length() == 11)&&(d->second)&&(d->first[0] == 'S')) {
|
||||
std::string idbuf;
|
||||
if (Utils::readFile((basePath + ZT_PATH_SEPARATOR_S + d->first + ZT_PATH_SEPARATOR_S + "identity.public").c_str(),idbuf)) {
|
||||
Identity id(idbuf);
|
||||
if (id) {
|
||||
InetAddress inaddr(inetAddressFromZeroTierAddress(id.address()));
|
||||
snids.push_back(std::pair<Identity,InetAddress>(id,inaddr));
|
||||
|
||||
Dictionary snd;
|
||||
snd["id"] = id.toString(false);
|
||||
snd["udp"] = inaddr.toString();
|
||||
snd["desc"] = id.address().toString();
|
||||
snd["dns"] = inaddr.toIpString();
|
||||
supernodes[id.address().toString()] = snd.toString();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Dictionary rtd;
|
||||
rtd["supernodes"] = supernodes.toString();
|
||||
rtd["noupdate"] = "1";
|
||||
rootTopology = rtd.toString();
|
||||
|
||||
std::vector<Address> newNodes;
|
||||
|
||||
for(std::vector< std::pair<Identity,InetAddress> >::iterator i(snids.begin());i!=snids.end();++i) {
|
||||
SimNode *n = new SimNode(net,(basePath + ZT_PATH_SEPARATOR_S + "S" + i->first.address().toString()),rootTopology.c_str(),true,i->second);
|
||||
nodes[i->first.address()] = n;
|
||||
newNodes.push_back(i->first.address());
|
||||
}
|
||||
|
||||
return newNodes;
|
||||
}
|
||||
|
||||
// Instantiates any not-already-instantiated regular nodes
|
||||
static std::vector<Address> scanForNewNodes()
|
||||
{
|
||||
std::vector<Address> newNodes;
|
||||
std::map<std::string,bool> dir(Utils::listDirectory(basePath.c_str()));
|
||||
|
||||
for(std::map<std::string,bool>::iterator d(dir.begin());d!=dir.end();++d) {
|
||||
if ((d->first.length() == 11)&&(d->second)&&(d->first[0] == 'N')) {
|
||||
Address na(d->first.c_str() + 1);
|
||||
if (nodes.find(na) == nodes.end()) {
|
||||
InetAddress inaddr(inetAddressFromZeroTierAddress(na));
|
||||
|
||||
SimNode *n = new SimNode(net,(basePath + ZT_PATH_SEPARATOR_S + d->first),rootTopology.c_str(),false,inaddr);
|
||||
nodes[na] = n;
|
||||
|
||||
newNodes.push_back(na);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return newNodes;
|
||||
}
|
||||
|
||||
static void doHelp(const std::vector<std::string> &cmd)
|
||||
{
|
||||
printf("---------- help"ZT_EOL_S);
|
||||
printf("---------- mksn <number of supernodes>"ZT_EOL_S);
|
||||
printf("---------- mkn <number of normal nodes>"ZT_EOL_S);
|
||||
printf("---------- list"ZT_EOL_S);
|
||||
printf("---------- join <address/*/**> <network ID>"ZT_EOL_S);
|
||||
printf("---------- leave <address/*/**> <network ID>"ZT_EOL_S);
|
||||
printf("---------- listnetworks <address/*/**>"ZT_EOL_S);
|
||||
printf("---------- listpeers <address/*/**>"ZT_EOL_S);
|
||||
printf("---------- unicast <address/*/**> <address/*/**> <network ID> <frame length, min: 16> [<timeout (sec)>]"ZT_EOL_S);
|
||||
printf("---------- multicast <address/*/**> <MAC/* for bcast> <network ID> <frame length, min: 16> [<timeout (sec)>]"ZT_EOL_S);
|
||||
printf("---------- quit"ZT_EOL_S);
|
||||
printf("---------- ( * means all regular nodes, ** means including supernodes )"ZT_EOL_S);
|
||||
printf("---------- ( . runs previous command again )"ZT_EOL_S);
|
||||
}
|
||||
|
||||
static void doMKSN(const std::vector<std::string> &cmd)
|
||||
{
|
||||
if (cmd.size() < 2) {
|
||||
doHelp(cmd);
|
||||
return;
|
||||
}
|
||||
if (nodes.size() > 0) {
|
||||
printf("---------- mksn error: mksn can only be called once (network already exists)"ZT_EOL_S);
|
||||
return;
|
||||
}
|
||||
|
||||
int count = Utils::strToInt(cmd[1].c_str());
|
||||
for(int i=0;i<count;++i) {
|
||||
Identity id(makeNodeHome(true));
|
||||
printf("%s identity created"ZT_EOL_S,id.address().toString().c_str());
|
||||
}
|
||||
|
||||
std::vector<Address> nodes(initSupernodes());
|
||||
for(std::vector<Address>::iterator a(nodes.begin());a!=nodes.end();++a)
|
||||
printf("%s started (supernode)"ZT_EOL_S,a->toString().c_str());
|
||||
|
||||
//printf("---------- root topology is: %s"ZT_EOL_S,rootTopology.c_str());
|
||||
}
|
||||
|
||||
static void doMKN(const std::vector<std::string> &cmd)
|
||||
{
|
||||
if (cmd.size() < 2) {
|
||||
doHelp(cmd);
|
||||
return;
|
||||
}
|
||||
if (nodes.size() == 0) {
|
||||
printf("---------- mkn error: use mksn to create supernodes first."ZT_EOL_S);
|
||||
return;
|
||||
}
|
||||
|
||||
int count = Utils::strToInt(cmd[1].c_str());
|
||||
for(int i=0;i<count;++i) {
|
||||
Identity id(makeNodeHome(false));
|
||||
printf("%s identity created"ZT_EOL_S,id.address().toString().c_str());
|
||||
}
|
||||
|
||||
std::vector<Address> nodes(scanForNewNodes());
|
||||
for(std::vector<Address>::iterator a(nodes.begin());a!=nodes.end();++a)
|
||||
printf("%s started (regular node)"ZT_EOL_S,a->toString().c_str());
|
||||
}
|
||||
|
||||
static void doList(const std::vector<std::string> &cmd)
|
||||
{
|
||||
unsigned int peers = 0,supernodes = 0;
|
||||
ZT1_Node_Status status;
|
||||
for(std::map< Address,SimNode * >::iterator n(nodes.begin());n!=nodes.end();++n) {
|
||||
n->second->node.status(&status);
|
||||
if (status.initialized) {
|
||||
printf("%s %c %s (%u peers, %u direct links)"ZT_EOL_S,
|
||||
n->first.toString().c_str(),
|
||||
n->second->supernode ? 'S' : 'N',
|
||||
(status.online ? "ONLINE" : "OFFLINE"),
|
||||
status.knownPeers,
|
||||
status.directlyConnectedPeers);
|
||||
if (n->second->supernode)
|
||||
++supernodes;
|
||||
else ++peers;
|
||||
} else printf("%s ? INITIALIZING (0 peers, 0 direct links)"ZT_EOL_S,n->first.toString().c_str());
|
||||
}
|
||||
printf("---------- %u regular peers, %u supernodes"ZT_EOL_S,peers,supernodes);
|
||||
}
|
||||
|
||||
static void doJoin(const std::vector<std::string> &cmd)
|
||||
{
|
||||
if (cmd.size() < 3) {
|
||||
doHelp(cmd);
|
||||
return;
|
||||
}
|
||||
|
||||
std::vector<Address> addrs;
|
||||
if ((cmd[1] == "*")||(cmd[1] == "**")) {
|
||||
bool includeSuper = (cmd[1] == "**");
|
||||
for(std::map< Address,SimNode * >::iterator n(nodes.begin());n!=nodes.end();++n) {
|
||||
if ((includeSuper)||(!n->second->supernode))
|
||||
addrs.push_back(n->first);
|
||||
}
|
||||
} else addrs.push_back(Address(cmd[1]));
|
||||
|
||||
uint64_t nwid = Utils::hexStrToU64(cmd[2].c_str());
|
||||
|
||||
for(std::vector<Address>::iterator a(addrs.begin());a!=addrs.end();++a) {
|
||||
std::map< Address,SimNode * >::iterator n(nodes.find(*a));
|
||||
if (n != nodes.end()) {
|
||||
n->second->node.join(nwid);
|
||||
printf("%s join %.16llx"ZT_EOL_S,n->first.toString().c_str(),(unsigned long long)nwid);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void doLeave(const std::vector<std::string> &cmd)
|
||||
{
|
||||
if (cmd.size() < 3) {
|
||||
doHelp(cmd);
|
||||
return;
|
||||
}
|
||||
|
||||
std::vector<Address> addrs;
|
||||
if ((cmd[1] == "*")||(cmd[1] == "**")) {
|
||||
bool includeSuper = (cmd[1] == "**");
|
||||
for(std::map< Address,SimNode * >::iterator n(nodes.begin());n!=nodes.end();++n) {
|
||||
if ((includeSuper)||(!n->second->supernode))
|
||||
addrs.push_back(n->first);
|
||||
}
|
||||
} else addrs.push_back(Address(cmd[1]));
|
||||
|
||||
uint64_t nwid = Utils::hexStrToU64(cmd[2].c_str());
|
||||
|
||||
for(std::vector<Address>::iterator a(addrs.begin());a!=addrs.end();++a) {
|
||||
std::map< Address,SimNode * >::iterator n(nodes.find(*a));
|
||||
if (n != nodes.end()) {
|
||||
n->second->node.leave(nwid);
|
||||
printf("%s leave %.16llx"ZT_EOL_S,n->first.toString().c_str(),(unsigned long long)nwid);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void doListNetworks(const std::vector<std::string> &cmd)
|
||||
{
|
||||
if (cmd.size() < 2) {
|
||||
doHelp(cmd);
|
||||
return;
|
||||
}
|
||||
|
||||
std::vector<Address> addrs;
|
||||
if ((cmd[1] == "*")||(cmd[1] == "**")) {
|
||||
bool includeSuper = (cmd[1] == "**");
|
||||
for(std::map< Address,SimNode * >::iterator n(nodes.begin());n!=nodes.end();++n) {
|
||||
if ((includeSuper)||(!n->second->supernode))
|
||||
addrs.push_back(n->first);
|
||||
}
|
||||
} else addrs.push_back(Address(cmd[1]));
|
||||
|
||||
printf("---------- <nwid> <name> <mac> <status> <config age> <type> <dev> <ips>"ZT_EOL_S);
|
||||
|
||||
for(std::vector<Address>::iterator a(addrs.begin());a!=addrs.end();++a) {
|
||||
std::string astr(a->toString());
|
||||
std::map< Address,SimNode * >::iterator n(nodes.find(*a));
|
||||
if (n != nodes.end()) {
|
||||
ZT1_Node_NetworkList *nl = n->second->node.listNetworks();
|
||||
if (nl) {
|
||||
for(unsigned int i=0;i<nl->numNetworks;++i) {
|
||||
printf("%s %s %s %s %s %ld %s %s ",
|
||||
astr.c_str(),
|
||||
nl->networks[i].nwidHex,
|
||||
nl->networks[i].name,
|
||||
nl->networks[i].macStr,
|
||||
nl->networks[i].statusStr,
|
||||
nl->networks[i].configAge,
|
||||
(nl->networks[i].isPrivate ? "private" : "public"),
|
||||
nl->networks[i].device);
|
||||
if (nl->networks[i].numIps > 0) {
|
||||
for(unsigned int j=0;j<nl->networks[i].numIps;++j) {
|
||||
if (j > 0)
|
||||
printf(",");
|
||||
printf("%s/%d",nl->networks[i].ips[j].ascii,(int)nl->networks[i].ips[j].port);
|
||||
}
|
||||
} else printf("-");
|
||||
printf(ZT_EOL_S);
|
||||
}
|
||||
n->second->node.freeQueryResult(nl);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void doListPeers(const std::vector<std::string> &cmd)
|
||||
{
|
||||
if (cmd.size() < 2) {
|
||||
doHelp(cmd);
|
||||
return;
|
||||
}
|
||||
|
||||
std::vector<Address> addrs;
|
||||
if ((cmd[1] == "*")||(cmd[1] == "**")) {
|
||||
bool includeSuper = (cmd[1] == "**");
|
||||
for(std::map< Address,SimNode * >::iterator n(nodes.begin());n!=nodes.end();++n) {
|
||||
if ((includeSuper)||(!n->second->supernode))
|
||||
addrs.push_back(n->first);
|
||||
}
|
||||
} else addrs.push_back(Address(cmd[1]));
|
||||
|
||||
printf("---------- <ztaddr> <paths> <latency> <version> <role>"ZT_EOL_S);
|
||||
|
||||
for(std::vector<Address>::iterator a(addrs.begin());a!=addrs.end();++a) {
|
||||
std::string astr(a->toString());
|
||||
std::map< Address,SimNode * >::iterator n(nodes.find(*a));
|
||||
if (n != nodes.end()) {
|
||||
ZT1_Node_PeerList *pl = n->second->node.listPeers();
|
||||
if (pl) {
|
||||
for(unsigned int i=0;i<pl->numPeers;++i) {
|
||||
printf("%s %.10llx ",astr.c_str(),(unsigned long long)pl->peers[i].rawAddress);
|
||||
if (pl->peers[i].numPaths == 0)
|
||||
printf("-");
|
||||
else {
|
||||
for(unsigned int j=0;j<pl->peers[i].numPaths;++j) {
|
||||
if (j > 0)
|
||||
printf(",");
|
||||
switch(pl->peers[i].paths[j].type) {
|
||||
default:
|
||||
printf("unknown;");
|
||||
break;
|
||||
case ZT1_Node_PhysicalPath_TYPE_UDP:
|
||||
printf("udp;");
|
||||
break;
|
||||
case ZT1_Node_PhysicalPath_TYPE_TCP_OUT:
|
||||
printf("tcp_out;");
|
||||
break;
|
||||
case ZT1_Node_PhysicalPath_TYPE_TCP_IN:
|
||||
printf("tcp_in;");
|
||||
break;
|
||||
case ZT1_Node_PhysicalPath_TYPE_ETHERNET:
|
||||
printf("eth;");
|
||||
break;
|
||||
}
|
||||
printf("%s/%d;%ld;%ld;%ld;%s",
|
||||
pl->peers[i].paths[j].address.ascii,
|
||||
(int)pl->peers[i].paths[j].address.port,
|
||||
pl->peers[i].paths[j].lastSend,
|
||||
pl->peers[i].paths[j].lastReceive,
|
||||
pl->peers[i].paths[j].lastPing,
|
||||
(pl->peers[i].paths[j].fixed ? "fixed" : (pl->peers[i].paths[j].active ? "active" : "inactive")));
|
||||
}
|
||||
}
|
||||
const char *rolestr;
|
||||
switch(pl->peers[i].role) {
|
||||
case ZT1_Node_Peer_SUPERNODE: rolestr = "SUPERNODE"; break;
|
||||
case ZT1_Node_Peer_HUB: rolestr = "HUB"; break;
|
||||
case ZT1_Node_Peer_NODE: rolestr = "NODE"; break;
|
||||
default: rolestr = "?"; break;
|
||||
}
|
||||
printf(" %u %s %s"ZT_EOL_S,
|
||||
pl->peers[i].latency,
|
||||
((pl->peers[i].remoteVersion[0]) ? pl->peers[i].remoteVersion : "-"),
|
||||
rolestr);
|
||||
}
|
||||
n->second->node.freeQueryResult(pl);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void doUnicast(const std::vector<std::string> &cmd)
|
||||
{
|
||||
union {
|
||||
uint64_t i[2];
|
||||
unsigned char data[2800];
|
||||
} pkt;
|
||||
|
||||
if (cmd.size() < 5) {
|
||||
doHelp(cmd);
|
||||
return;
|
||||
}
|
||||
|
||||
uint64_t nwid = Utils::hexStrToU64(cmd[3].c_str());
|
||||
unsigned int frameLen = Utils::strToUInt(cmd[4].c_str());
|
||||
uint64_t tout = 2000;
|
||||
if (cmd.size() >= 6)
|
||||
tout = Utils::strToU64(cmd[5].c_str()) * 1000ULL;
|
||||
|
||||
if (frameLen < 16)
|
||||
frameLen = 16;
|
||||
if (frameLen > 2800)
|
||||
frameLen = 2800;
|
||||
|
||||
std::vector<Address> senders;
|
||||
if ((cmd[1] == "*")||(cmd[1] == "**")) {
|
||||
bool includeSuper = (cmd[1] == "**");
|
||||
for(std::map< Address,SimNode * >::iterator n(nodes.begin());n!=nodes.end();++n) {
|
||||
if ((includeSuper)||(!n->second->supernode))
|
||||
senders.push_back(n->first);
|
||||
}
|
||||
} else senders.push_back(Address(cmd[1]));
|
||||
|
||||
std::vector<Address> receivers;
|
||||
if ((cmd[2] == "*")||(cmd[2] == "**")) {
|
||||
bool includeSuper = (cmd[2] == "**");
|
||||
for(std::map< Address,SimNode * >::iterator n(nodes.begin());n!=nodes.end();++n) {
|
||||
if ((includeSuper)||(!n->second->supernode))
|
||||
receivers.push_back(n->first);
|
||||
}
|
||||
} else receivers.push_back(Address(cmd[2]));
|
||||
|
||||
for(unsigned int i=0;i<frameLen;++i)
|
||||
pkt.data[i] = (unsigned char)prng.next32();
|
||||
|
||||
std::set< std::pair<Address,Address> > sentPairs;
|
||||
for(std::vector<Address>::iterator s(senders.begin());s!=senders.end();++s) {
|
||||
for(std::vector<Address>::iterator r(receivers.begin());r!=receivers.end();++r) {
|
||||
if (*s == *r)
|
||||
continue;
|
||||
|
||||
SimNode *sender = nodes[*s];
|
||||
SimNode *receiver = nodes[*r];
|
||||
TestEthernetTap *stap = sender->tapFactory.getByNwid(nwid);
|
||||
TestEthernetTap *rtap = receiver->tapFactory.getByNwid(nwid);
|
||||
|
||||
if ((stap)&&(rtap)) {
|
||||
pkt.i[0] = s->toInt();
|
||||
pkt.i[1] = Utils::now();
|
||||
stap->injectPacketFromHost(stap->mac(),rtap->mac(),0xdead,pkt.data,frameLen);
|
||||
printf("%s -> %s etherType 0xdead network %.16llx length %u"ZT_EOL_S,s->toString().c_str(),r->toString().c_str(),(unsigned long long)nwid,frameLen);
|
||||
sentPairs.insert(std::pair<Address,Address>(*s,*r));
|
||||
} else if (stap) {
|
||||
printf("%s -> !%s (receiver not a member of %.16llx)"ZT_EOL_S,s->toString().c_str(),r->toString().c_str(),(unsigned long long)nwid);
|
||||
} else if (rtap) {
|
||||
printf("%s -> !%s (sender not a member of %.16llx)"ZT_EOL_S,s->toString().c_str(),r->toString().c_str(),(unsigned long long)nwid);
|
||||
} else {
|
||||
printf("%s -> !%s (neither party is a member of %.16llx)"ZT_EOL_S,s->toString().c_str(),r->toString().c_str(),(unsigned long long)nwid);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
printf("---------- waiting up to %llu seconds..."ZT_EOL_S,tout / 1000ULL);
|
||||
|
||||
std::set< std::pair<Address,Address> > receivedPairs;
|
||||
TestEthernetTap::TestFrame frame;
|
||||
uint64_t toutend = Utils::now() + tout;
|
||||
do {
|
||||
for(std::vector<Address>::iterator r(receivers.begin());r!=receivers.end();++r) {
|
||||
SimNode *receiver = nodes[*r];
|
||||
TestEthernetTap *rtap = receiver->tapFactory.getByNwid(nwid);
|
||||
|
||||
if ((rtap)&&(rtap->getNextReceivedFrame(frame,5))) {
|
||||
if ((frame.len == frameLen)&&(!memcmp(frame.data + 16,pkt.data + 16,frameLen - 16))) {
|
||||
uint64_t ints[2];
|
||||
memcpy(ints,frame.data,16);
|
||||
printf("%s <- %.10llx received test packet, length == %u, latency == %llums"ZT_EOL_S,r->toString().c_str(),(unsigned long long)ints[0],frame.len,(unsigned long long)(frame.timestamp - ints[1]));
|
||||
receivedPairs.insert(std::pair<Address,Address>(Address(ints[0]),*r));
|
||||
} else {
|
||||
printf("%s !! got spurious packet, length == %u, etherType == 0x%.4x"ZT_EOL_S,r->toString().c_str(),frame.len,frame.etherType);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Thread::sleep(100);
|
||||
} while ((receivedPairs.size() < sentPairs.size())&&(Utils::now() < toutend));
|
||||
|
||||
for(std::vector<Address>::iterator s(senders.begin());s!=senders.end();++s) {
|
||||
for(std::vector<Address>::iterator r(receivers.begin());r!=receivers.end();++r) {
|
||||
if (*s == *r)
|
||||
continue;
|
||||
if ((sentPairs.count(std::pair<Address,Address>(*s,*r)))&&(!receivedPairs.count(std::pair<Address,Address>(*s,*r)))) {
|
||||
printf("%s <- %s was never received (timed out)"ZT_EOL_S,r->toString().c_str(),s->toString().c_str());
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
printf("---------- sent %u, received %u"ZT_EOL_S,(unsigned int)sentPairs.size(),(unsigned int)receivedPairs.size());
|
||||
}
|
||||
|
||||
static void doMulticast(const std::vector<std::string> &cmd)
|
||||
{
|
||||
union {
|
||||
uint64_t i[2];
|
||||
unsigned char data[2800];
|
||||
} pkt;
|
||||
|
||||
if (cmd.size() < 5) {
|
||||
doHelp(cmd);
|
||||
return;
|
||||
}
|
||||
|
||||
uint64_t nwid = Utils::hexStrToU64(cmd[3].c_str());
|
||||
unsigned int frameLen = Utils::strToUInt(cmd[4].c_str());
|
||||
uint64_t tout = 2000;
|
||||
if (cmd.size() >= 6)
|
||||
tout = Utils::strToU64(cmd[5].c_str()) * 1000ULL;
|
||||
|
||||
if (frameLen < 16)
|
||||
frameLen = 16;
|
||||
if (frameLen > 2800)
|
||||
frameLen = 2800;
|
||||
|
||||
std::vector<Address> senders;
|
||||
if ((cmd[1] == "*")||(cmd[1] == "**")) {
|
||||
bool includeSuper = (cmd[1] == "**");
|
||||
for(std::map< Address,SimNode * >::iterator n(nodes.begin());n!=nodes.end();++n) {
|
||||
if ((includeSuper)||(!n->second->supernode))
|
||||
senders.push_back(n->first);
|
||||
}
|
||||
} else senders.push_back(Address(cmd[1]));
|
||||
|
||||
MAC mcaddr;
|
||||
if (cmd[2] == "*")
|
||||
mcaddr = MAC(0xff,0xff,0xff,0xff,0xff,0xff);
|
||||
else mcaddr.fromString(cmd[2].c_str());
|
||||
|
||||
if (!mcaddr.isMulticast()) {
|
||||
printf("---------- %s is not a multicast MAC address"ZT_EOL_S,mcaddr.toString().c_str());
|
||||
return;
|
||||
}
|
||||
|
||||
for(unsigned int i=0;i<frameLen;++i)
|
||||
pkt.data[i] = (unsigned char)prng.next32();
|
||||
|
||||
for(std::vector<Address>::iterator s(senders.begin());s!=senders.end();++s) {
|
||||
SimNode *sender = nodes[*s];
|
||||
TestEthernetTap *stap = sender->tapFactory.getByNwid(nwid);
|
||||
|
||||
if (stap) {
|
||||
pkt.i[0] = s->toInt();
|
||||
pkt.i[1] = Utils::now();
|
||||
stap->injectPacketFromHost(stap->mac(),mcaddr,0xdead,pkt.data,frameLen);
|
||||
printf("%s -> %s etherType 0xdead network %.16llx length %u"ZT_EOL_S,s->toString().c_str(),mcaddr.toString().c_str(),(unsigned long long)nwid,frameLen);
|
||||
} else {
|
||||
printf("%s -> !%s (sender is not a member of %.16llx)"ZT_EOL_S,s->toString().c_str(),mcaddr.toString().c_str(),(unsigned long long)nwid);
|
||||
}
|
||||
}
|
||||
|
||||
printf("---------- waiting %llu seconds..."ZT_EOL_S,tout / 1000ULL);
|
||||
|
||||
unsigned int receiveCount = 0;
|
||||
TestEthernetTap::TestFrame frame;
|
||||
uint64_t toutend = Utils::now() + tout;
|
||||
do {
|
||||
for(std::map< Address,SimNode * >::iterator nn(nodes.begin());nn!=nodes.end();++nn) {
|
||||
SimNode *receiver = nn->second;
|
||||
TestEthernetTap *rtap = receiver->tapFactory.getByNwid(nwid);
|
||||
|
||||
if ((rtap)&&(rtap->getNextReceivedFrame(frame,5))) {
|
||||
if ((frame.len == frameLen)&&(!memcmp(frame.data + 16,pkt.data + 16,frameLen - 16))) {
|
||||
uint64_t ints[2];
|
||||
memcpy(ints,frame.data,16);
|
||||
printf("%s <- %.10llx received test packet, length == %u, latency == %llums"ZT_EOL_S,nn->first.toString().c_str(),(unsigned long long)ints[0],frame.len,(unsigned long long)(frame.timestamp - ints[1]));
|
||||
++receiveCount;
|
||||
} else {
|
||||
printf("%s !! got spurious packet, length == %u, etherType == 0x%.4x"ZT_EOL_S,nn->first.toString().c_str(),frame.len,frame.etherType);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Thread::sleep(100);
|
||||
} while (Utils::now() < toutend);
|
||||
|
||||
printf("---------- test multicast received by %u peers"ZT_EOL_S,receiveCount);
|
||||
}
|
||||
|
||||
int main(int argc,char **argv)
|
||||
{
|
||||
char linebuf[1024];
|
||||
|
||||
if (argc <= 1) {
|
||||
fprintf(stderr,"Usage: %s <base path for temporary node home directories>"ZT_EOL_S,argv[0]);
|
||||
return 1;
|
||||
}
|
||||
|
||||
basePath = argv[1];
|
||||
#ifdef __WINDOWS__
|
||||
CreateDirectoryA(basePath.c_str(),NULL);
|
||||
#else
|
||||
mkdir(basePath.c_str(),0700);
|
||||
#endif
|
||||
|
||||
printf("*** ZeroTier One Version %s -- Headless Network Simulator ***"ZT_EOL_S,Node::versionString());
|
||||
printf(ZT_EOL_S);
|
||||
|
||||
{
|
||||
printf("---------- scanning '%s' for existing network..."ZT_EOL_S,basePath.c_str());
|
||||
std::vector<Address> snodes(initSupernodes());
|
||||
if (snodes.empty()) {
|
||||
printf("---------- no existing network found; use 'mksn' to create one."ZT_EOL_S);
|
||||
} else {
|
||||
for(std::vector<Address>::iterator a(snodes.begin());a!=snodes.end();++a)
|
||||
printf("%s started (supernode)"ZT_EOL_S,a->toString().c_str());
|
||||
//printf("---------- root topology is: %s"ZT_EOL_S,rootTopology.c_str());
|
||||
std::vector<Address> nodes(scanForNewNodes());
|
||||
for(std::vector<Address>::iterator a(nodes.begin());a!=nodes.end();++a)
|
||||
printf("%s started (normal peer)"ZT_EOL_S,a->toString().c_str());
|
||||
printf("---------- %u peers and %u supernodes loaded!"ZT_EOL_S,(unsigned int)nodes.size(),(unsigned int)snodes.size());
|
||||
}
|
||||
}
|
||||
printf(ZT_EOL_S);
|
||||
|
||||
printf("Type 'help' for help."ZT_EOL_S);
|
||||
printf(ZT_EOL_S);
|
||||
|
||||
std::vector<std::string> cmd,prevCmd;
|
||||
bool run = true;
|
||||
while (run) {
|
||||
printf(">> ");
|
||||
fflush(stdout);
|
||||
if (!fgets(linebuf,sizeof(linebuf),stdin))
|
||||
break;
|
||||
|
||||
cmd = Utils::split(linebuf," \r\n\t","\\","\"");
|
||||
|
||||
for(;;) {
|
||||
if (cmd.size() == 0)
|
||||
break;
|
||||
else if (cmd[0] == "quit")
|
||||
run = false;
|
||||
else if (cmd[0] == "help")
|
||||
doHelp(cmd);
|
||||
else if (cmd[0] == "mksn")
|
||||
doMKSN(cmd);
|
||||
else if (cmd[0] == "mkn")
|
||||
doMKN(cmd);
|
||||
else if (cmd[0] == "list")
|
||||
doList(cmd);
|
||||
else if (cmd[0] == "join")
|
||||
doJoin(cmd);
|
||||
else if (cmd[0] == "leave")
|
||||
doLeave(cmd);
|
||||
else if (cmd[0] == "listnetworks")
|
||||
doListNetworks(cmd);
|
||||
else if (cmd[0] == "listpeers")
|
||||
doListPeers(cmd);
|
||||
else if (cmd[0] == "unicast")
|
||||
doUnicast(cmd);
|
||||
else if (cmd[0] == "multicast")
|
||||
doMulticast(cmd);
|
||||
else if ((cmd[0] == ".")&&(prevCmd.size() > 0)) {
|
||||
cmd = prevCmd;
|
||||
continue;
|
||||
} else doHelp(cmd);
|
||||
break;
|
||||
}
|
||||
|
||||
if ((cmd.size() > 0)&&(cmd[0] != "."))
|
||||
prevCmd = cmd;
|
||||
}
|
||||
|
||||
for(std::map< Address,SimNode * >::iterator n(nodes.begin());n!=nodes.end();++n) {
|
||||
printf("%s shutting down..."ZT_EOL_S,n->first.toString().c_str());
|
||||
delete n->second;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
@ -1,181 +0,0 @@
|
||||
/*
|
||||
* ZeroTier One - Network Virtualization Everywhere
|
||||
* Copyright (C) 2011-2015 ZeroTier, Inc. LLC
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* --
|
||||
*
|
||||
* ZeroTier may be used and distributed under the terms of the GPLv3, which
|
||||
* are available at: http://www.gnu.org/licenses/gpl-3.0.html
|
||||
*
|
||||
* If you would like to embed ZeroTier into a commercial application or
|
||||
* redistribute it in a modified binary form, please contact ZeroTier Networks
|
||||
* LLC. Start here: http://www.zerotier.com/
|
||||
*/
|
||||
|
||||
#ifndef ZT_MTQ_HPP
|
||||
#define ZT_MTQ_HPP
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <stdint.h>
|
||||
|
||||
#include <queue>
|
||||
|
||||
#include "../node/Constants.hpp"
|
||||
#include "../node/NonCopyable.hpp"
|
||||
#include "../node/Utils.hpp"
|
||||
|
||||
#ifdef __WINDOWS__
|
||||
#include <Windows.h>
|
||||
#else
|
||||
#include <time.h>
|
||||
#include <pthread.h>
|
||||
#endif
|
||||
|
||||
namespace ZeroTier {
|
||||
|
||||
/**
|
||||
* A synchronized multithreaded FIFO queue
|
||||
*
|
||||
* This is designed for a use case where one thread pushes, the
|
||||
* other pops.
|
||||
*/
|
||||
template<typename T>
|
||||
class MTQ : NonCopyable
|
||||
{
|
||||
public:
|
||||
MTQ()
|
||||
{
|
||||
#ifdef __WINDOWS__
|
||||
_sem = CreateSemaphore(NULL,0,0x7fffffff,NULL);
|
||||
InitializeCriticalSection(&_cs);
|
||||
#else
|
||||
pthread_mutex_init(&_mh,(const pthread_mutexattr_t *)0);
|
||||
pthread_cond_init(&_cond,(const pthread_condattr_t *)0);
|
||||
#endif
|
||||
}
|
||||
|
||||
~MTQ()
|
||||
{
|
||||
#ifdef __WINDOWS__
|
||||
CloseHandle(_sem);
|
||||
DeleteCriticalSection(&_cs);
|
||||
#else
|
||||
pthread_cond_destroy(&_cond);
|
||||
pthread_mutex_destroy(&_mh);
|
||||
#endif
|
||||
}
|
||||
|
||||
/**
|
||||
* Push something onto the end of the FIFO and signal waiting thread(s)
|
||||
*
|
||||
* @param v Value to push
|
||||
*/
|
||||
inline void push(const T &v)
|
||||
{
|
||||
#ifdef __WINDOWS__
|
||||
EnterCriticalSection(&_cs);
|
||||
try {
|
||||
_q.push(v);
|
||||
LeaveCriticalSection(&_cs);
|
||||
ReleaseSemaphore(_sem,1,NULL);
|
||||
} catch ( ... ) {
|
||||
LeaveCriticalSection(&_cs);
|
||||
throw;
|
||||
}
|
||||
#else
|
||||
pthread_mutex_lock(const_cast <pthread_mutex_t *>(&_mh));
|
||||
try {
|
||||
_q.push(v);
|
||||
pthread_mutex_unlock(const_cast <pthread_mutex_t *>(&_mh));
|
||||
pthread_cond_signal(const_cast <pthread_cond_t *>(&_cond));
|
||||
} catch ( ... ) {
|
||||
pthread_mutex_unlock(const_cast <pthread_mutex_t *>(&_mh));
|
||||
throw;
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
/**
|
||||
* Pop fron queue with optional timeout
|
||||
*
|
||||
* @param v Result parameter to set to next value
|
||||
* @param ms Milliseconds timeout or 0 for none
|
||||
* @return True if v was set to something, false on timeout
|
||||
*/
|
||||
inline bool pop(T &v,unsigned long ms = 0)
|
||||
{
|
||||
#ifdef __WINDOWS__
|
||||
if (ms > 0)
|
||||
WaitForSingleObject(_sem,(DWORD)ms);
|
||||
else WaitForSingleObject(_sem,INFINITE);
|
||||
EnterCriticalSection(&_cs);
|
||||
try {
|
||||
if (_q.empty()) {
|
||||
LeaveCriticalSection(&_cs);
|
||||
return false;
|
||||
} else {
|
||||
v = _q.front();
|
||||
_q.pop();
|
||||
LeaveCriticalSection(&_cs);
|
||||
return true;
|
||||
}
|
||||
} catch ( ... ) {
|
||||
LeaveCriticalSection(&_cs);
|
||||
throw;
|
||||
}
|
||||
#else
|
||||
pthread_mutex_lock(const_cast <pthread_mutex_t *>(&_mh));
|
||||
try {
|
||||
if (_q.empty()) {
|
||||
if (ms > 0) {
|
||||
uint64_t when = Utils::now() + (uint64_t)ms;
|
||||
struct timespec ts;
|
||||
ts.tv_sec = (unsigned long)(when / 1000);
|
||||
ts.tv_nsec = (unsigned long)(when % 1000) * (unsigned long)1000000;
|
||||
pthread_cond_timedwait(const_cast <pthread_cond_t *>(&_cond),const_cast <pthread_mutex_t *>(&_mh),&ts);
|
||||
} else {
|
||||
pthread_cond_wait(const_cast <pthread_cond_t *>(&_cond),const_cast <pthread_mutex_t *>(&_mh));
|
||||
}
|
||||
if (_q.empty()) {
|
||||
pthread_mutex_unlock(const_cast <pthread_mutex_t *>(&_mh));
|
||||
return false;
|
||||
}
|
||||
}
|
||||
v = _q.front();
|
||||
_q.pop();
|
||||
pthread_mutex_unlock(const_cast <pthread_mutex_t *>(&_mh));
|
||||
return true;
|
||||
} catch ( ... ) {
|
||||
pthread_mutex_unlock(const_cast <pthread_mutex_t *>(&_mh));
|
||||
throw;
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
private:
|
||||
std::queue<T> _q;
|
||||
#ifdef __WINDOWS__
|
||||
HANDLE _sem;
|
||||
CRITICAL_SECTION _cs;
|
||||
#else
|
||||
pthread_cond_t _cond;
|
||||
pthread_mutex_t _mh;
|
||||
#endif
|
||||
};
|
||||
|
||||
} // namespace ZeroTier
|
||||
|
||||
#endif
|
@ -1,36 +0,0 @@
|
||||
Headless Test Network
|
||||
======
|
||||
|
||||
The code in testnet.cpp (in base) and here in testnet/ is for running headless ZeroTier One test networks.
|
||||
|
||||
To build, type (from base) "make testnet". This will build the *zerotier-testnet* binary. Then run it with a directory to use for temporary node home directory storage as a parameter, e.g. "./zerotier-testnet /tmp/zttestnet".
|
||||
|
||||
Type **help** for help.
|
||||
|
||||
Right now the testnet simulates a perfect IP network and allows you to perform unicast and multicast tests. This is useful for verifying the basic correctness of everything under ideal conditions, and for smoke testing. In the future support for NAT emulation, packet loss, and other test features will be added to make this a more full-blown test suite.
|
||||
|
||||
When you start the testnet for the first time, no nodes will exist. You have to create some. First, create supernodes with **mksn**. Create as many as you want. Once you've created supernodes (you can only do this once per testnet) you can create regular nodes with **mkn**.
|
||||
|
||||
Once everything is created use **list** to check the status.
|
||||
|
||||
Each node will create a couple threads, so if your OS imposes a limit this might cause some of your virtual nodes to stick in *INITIALIZING* status as shown in the **list** command. If this happens you might want to blow away the contents of your temp directory and try again with fewer nodes.
|
||||
|
||||
Each node will get a home at the test path you specified, so quitting with **quit** and re-entering will reload the same test network.
|
||||
|
||||
Next you'll need to join your nodes to a virtual ZeroTier network. ZeroTier supports a built-in "fake" public network with the ID **ffffffffffffffff**. This network is for testing and is convenient to use here. It's also possible to set up the netconf-master within one of your test nodes, but doing so is beyond the scope of this doc (for now, but if your clever you can probably figure it out). Verify by doing **listnetworks**.
|
||||
|
||||
Now you can send some packets. Try:
|
||||
|
||||
unicast * * ffffffffffffffff 24 60
|
||||
|
||||
That will do a unicast all-to-all test and report results. At first latencies might seem high, especially for a headless fake IP network. If you try it again you'll see them drop to zero or nearly so, since everyone will have executed a peer to peer connection.
|
||||
|
||||
multicast <some node's 10-digit ZT address> * ffffffffffffffff 24 60
|
||||
|
||||
This will send a multicast packet to ff:ff:ff:ff:ff:ff (broadcast) and report back who receives it. You should see multicast propagation limited to 32 nodes, since this is the setting for multicast limit on the fake test network (and the default if not overridden in netconf). Multicast will show the same "warm up" behavior as unicast.
|
||||
|
||||
Typing just "." will execute the same testnet command again.
|
||||
|
||||
The first 10-digit field of each response is the ZeroTier node doing the sending or receiving. A prefix of "----------" is used for general responses to make everything line up neatly on the screen. We recommend using a wide terminal emulator.
|
||||
|
||||
Enjoy!
|
@ -1,68 +0,0 @@
|
||||
/*
|
||||
* ZeroTier One - Network Virtualization Everywhere
|
||||
* Copyright (C) 2011-2015 ZeroTier, Inc.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* --
|
||||
*
|
||||
* ZeroTier may be used and distributed under the terms of the GPLv3, which
|
||||
* are available at: http://www.gnu.org/licenses/gpl-3.0.html
|
||||
*
|
||||
* If you would like to embed ZeroTier into a commercial application or
|
||||
* redistribute it in a modified binary form, please contact ZeroTier Networks
|
||||
* LLC. Start here: http://www.zerotier.com/
|
||||
*/
|
||||
|
||||
#include "SimNet.hpp"
|
||||
|
||||
#include "../node/Constants.hpp"
|
||||
#include "../node/Utils.hpp"
|
||||
|
||||
namespace ZeroTier {
|
||||
|
||||
SimNet::SimNet()
|
||||
{
|
||||
}
|
||||
|
||||
SimNet::~SimNet()
|
||||
{
|
||||
}
|
||||
|
||||
SimNetSocketManager *SimNet::newEndpoint(const InetAddress &addr)
|
||||
{
|
||||
Mutex::Lock _l(_lock);
|
||||
|
||||
if (_endpoints.size() >= ZT_SIMNET_MAX_TESTNET_SIZE)
|
||||
return (SimNetSocketManager *)0;
|
||||
if (_endpoints.find(addr) != _endpoints.end())
|
||||
return (SimNetSocketManager *)0;
|
||||
|
||||
SimNetSocketManager *sm = new SimNetSocketManager();
|
||||
sm->_sn = this;
|
||||
sm->_address = addr;
|
||||
_endpoints[addr] = sm;
|
||||
return sm;
|
||||
}
|
||||
|
||||
SimNetSocketManager *SimNet::get(const InetAddress &addr)
|
||||
{
|
||||
Mutex::Lock _l(_lock);
|
||||
std::map< InetAddress,SimNetSocketManager * >::iterator ep(_endpoints.find(addr));
|
||||
if (ep == _endpoints.end())
|
||||
return (SimNetSocketManager *)0;
|
||||
return ep->second;
|
||||
}
|
||||
|
||||
} // namespace ZeroTier
|
@ -1,71 +0,0 @@
|
||||
/*
|
||||
* ZeroTier One - Network Virtualization Everywhere
|
||||
* Copyright (C) 2011-2015 ZeroTier, Inc.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* --
|
||||
*
|
||||
* ZeroTier may be used and distributed under the terms of the GPLv3, which
|
||||
* are available at: http://www.gnu.org/licenses/gpl-3.0.html
|
||||
*
|
||||
* If you would like to embed ZeroTier into a commercial application or
|
||||
* redistribute it in a modified binary form, please contact ZeroTier Networks
|
||||
* LLC. Start here: http://www.zerotier.com/
|
||||
*/
|
||||
|
||||
#ifndef ZT_SIMNET_HPP
|
||||
#define ZT_SIMNET_HPP
|
||||
|
||||
#include <map>
|
||||
#include <vector>
|
||||
|
||||
#include "../node/Constants.hpp"
|
||||
#include "../node/InetAddress.hpp"
|
||||
#include "../node/Mutex.hpp"
|
||||
|
||||
#include "SimNetSocketManager.hpp"
|
||||
|
||||
#define ZT_SIMNET_MAX_TESTNET_SIZE 1048576
|
||||
|
||||
namespace ZeroTier {
|
||||
|
||||
/**
|
||||
* A simulated headless IP network for testing
|
||||
*/
|
||||
class SimNet
|
||||
{
|
||||
public:
|
||||
SimNet();
|
||||
~SimNet();
|
||||
|
||||
/**
|
||||
* @return New endpoint or NULL on failure
|
||||
*/
|
||||
SimNetSocketManager *newEndpoint(const InetAddress &addr);
|
||||
|
||||
/**
|
||||
* @param addr Address to look up
|
||||
* @return Endpoint or NULL if none
|
||||
*/
|
||||
SimNetSocketManager *get(const InetAddress &addr);
|
||||
|
||||
private:
|
||||
std::map< InetAddress,SimNetSocketManager * > _endpoints;
|
||||
Mutex _lock;
|
||||
};
|
||||
|
||||
} // namespace ZeroTier
|
||||
|
||||
#endif
|
@ -1,90 +0,0 @@
|
||||
/*
|
||||
* ZeroTier One - Network Virtualization Everywhere
|
||||
* Copyright (C) 2011-2015 ZeroTier, Inc.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* --
|
||||
*
|
||||
* ZeroTier may be used and distributed under the terms of the GPLv3, which
|
||||
* are available at: http://www.gnu.org/licenses/gpl-3.0.html
|
||||
*
|
||||
* If you would like to embed ZeroTier into a commercial application or
|
||||
* redistribute it in a modified binary form, please contact ZeroTier Networks
|
||||
* LLC. Start here: http://www.zerotier.com/
|
||||
*/
|
||||
|
||||
#include "SimNetSocketManager.hpp"
|
||||
#include "SimNet.hpp"
|
||||
|
||||
#include "../node/Constants.hpp"
|
||||
#include "../node/Socket.hpp"
|
||||
|
||||
namespace ZeroTier {
|
||||
|
||||
class SimNetSocket : public Socket
|
||||
{
|
||||
public:
|
||||
SimNetSocket(SimNetSocketManager *sm) :
|
||||
Socket(ZT_SOCKET_TYPE_UDP_V4),
|
||||
_parent(sm) {}
|
||||
|
||||
virtual bool send(const InetAddress &to,const void *msg,unsigned int msglen)
|
||||
{
|
||||
SimNetSocketManager *dest = _parent->net()->get(to);
|
||||
if (dest)
|
||||
dest->enqueue(_parent->address(),msg,msglen);
|
||||
return true; // we emulate UDP, which has no delivery guarantee semantics
|
||||
}
|
||||
|
||||
SimNetSocketManager *_parent;
|
||||
};
|
||||
|
||||
SimNetSocketManager::SimNetSocketManager() :
|
||||
_sn((SimNet *)0), // initialized by SimNet
|
||||
_mySocket(new SimNetSocket(this))
|
||||
{
|
||||
}
|
||||
|
||||
SimNetSocketManager::~SimNetSocketManager()
|
||||
{
|
||||
}
|
||||
|
||||
bool SimNetSocketManager::send(const InetAddress &to,bool tcp,bool autoConnectTcp,const void *msg,unsigned int msglen)
|
||||
{
|
||||
if (tcp)
|
||||
return false; // we emulate UDP
|
||||
SimNetSocketManager *dest = _sn->get(to);
|
||||
if (dest)
|
||||
dest->enqueue(_address,msg,msglen);
|
||||
return true; // we emulate UDP, which has no delivery guarantee semantics
|
||||
}
|
||||
|
||||
void SimNetSocketManager::poll(unsigned long timeout,void (*handler)(const SharedPtr<Socket> &,void *,const InetAddress &,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN> &),void *arg)
|
||||
{
|
||||
std::pair< InetAddress,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN> > msg;
|
||||
if ((_inbox.pop(msg,timeout))&&(msg.second.size()))
|
||||
handler(_mySocket,arg,msg.first,msg.second);
|
||||
}
|
||||
|
||||
void SimNetSocketManager::whack()
|
||||
{
|
||||
_inbox.push(std::pair< InetAddress,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN> >());
|
||||
}
|
||||
|
||||
void SimNetSocketManager::closeTcpSockets()
|
||||
{
|
||||
}
|
||||
|
||||
} // namespace ZeroTier
|
@ -1,124 +0,0 @@
|
||||
/*
|
||||
* ZeroTier One - Network Virtualization Everywhere
|
||||
* Copyright (C) 2011-2015 ZeroTier, Inc.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* --
|
||||
*
|
||||
* ZeroTier may be used and distributed under the terms of the GPLv3, which
|
||||
* are available at: http://www.gnu.org/licenses/gpl-3.0.html
|
||||
*
|
||||
* If you would like to embed ZeroTier into a commercial application or
|
||||
* redistribute it in a modified binary form, please contact ZeroTier Networks
|
||||
* LLC. Start here: http://www.zerotier.com/
|
||||
*/
|
||||
|
||||
#ifndef ZT_SIMNETSOCKETMANAGER_HPP
|
||||
#define ZT_SIMNETSOCKETMANAGER_HPP
|
||||
|
||||
#include <map>
|
||||
#include <utility>
|
||||
#include <vector>
|
||||
|
||||
#include "../node/Constants.hpp"
|
||||
#include "../node/SocketManager.hpp"
|
||||
#include "../node/Mutex.hpp"
|
||||
|
||||
#include "MTQ.hpp"
|
||||
|
||||
namespace ZeroTier {
|
||||
|
||||
class SimNet;
|
||||
|
||||
/**
|
||||
* Socket manager for an IP endpoint in a simulated network
|
||||
*/
|
||||
class SimNetSocketManager : public SocketManager
|
||||
{
|
||||
friend class SimNet;
|
||||
|
||||
public:
|
||||
struct TransferStats
|
||||
{
|
||||
TransferStats() : received(0),sent(0) {}
|
||||
unsigned long long received;
|
||||
unsigned long long sent;
|
||||
};
|
||||
|
||||
SimNetSocketManager();
|
||||
virtual ~SimNetSocketManager();
|
||||
|
||||
/**
|
||||
* @return IP address of this simulated endpoint
|
||||
*/
|
||||
inline const InetAddress &address() const { return _address; }
|
||||
|
||||
/**
|
||||
* @return Local endpoint stats
|
||||
*/
|
||||
inline const TransferStats &totals() const { return _totals; }
|
||||
|
||||
/**
|
||||
* @param peer Peer IP address
|
||||
* @return Transfer stats for this peer
|
||||
*/
|
||||
inline TransferStats stats(const InetAddress &peer) const
|
||||
{
|
||||
Mutex::Lock _l(_stats_m);
|
||||
std::map< InetAddress,TransferStats >::const_iterator s(_stats.find(peer));
|
||||
if (s == _stats.end())
|
||||
return TransferStats();
|
||||
return s->second;
|
||||
}
|
||||
|
||||
/**
|
||||
* @return Network to which this endpoint belongs
|
||||
*/
|
||||
inline SimNet *net() const { return _sn; }
|
||||
|
||||
/**
|
||||
* Enqueue data from another endpoint to be picked up on next poll()
|
||||
*
|
||||
* @param from Originating endpoint address
|
||||
* @param data Data
|
||||
* @param len Length of data in bytes
|
||||
*/
|
||||
inline void enqueue(const InetAddress &from,const void *data,unsigned int len)
|
||||
{
|
||||
_inbox.push(std::pair< InetAddress,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN> >(from,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN>(data,len)));
|
||||
}
|
||||
|
||||
virtual bool send(const InetAddress &to,bool tcp,bool autoConnectTcp,const void *msg,unsigned int msglen);
|
||||
virtual void poll(unsigned long timeout,void (*handler)(const SharedPtr<Socket> &,void *,const InetAddress &,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN> &),void *arg);
|
||||
virtual void whack();
|
||||
virtual void closeTcpSockets();
|
||||
|
||||
private:
|
||||
// These are set by SimNet after object creation
|
||||
SimNet *_sn;
|
||||
InetAddress _address;
|
||||
|
||||
SharedPtr<Socket> _mySocket;
|
||||
TransferStats _totals;
|
||||
|
||||
MTQ< std::pair< InetAddress,Buffer<ZT_SOCKET_MAX_MESSAGE_LEN> > > _inbox;
|
||||
|
||||
std::map< InetAddress,TransferStats > _stats;
|
||||
Mutex _stats_m;
|
||||
};
|
||||
|
||||
} // namespace ZeroTier
|
||||
|
||||
#endif
|
@ -1,150 +0,0 @@
|
||||
/*
|
||||
* ZeroTier One - Network Virtualization Everywhere
|
||||
* Copyright (C) 2011-2015 ZeroTier, Inc.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* --
|
||||
*
|
||||
* ZeroTier may be used and distributed under the terms of the GPLv3, which
|
||||
* are available at: http://www.gnu.org/licenses/gpl-3.0.html
|
||||
*
|
||||
* If you would like to embed ZeroTier into a commercial application or
|
||||
* redistribute it in a modified binary form, please contact ZeroTier Networks
|
||||
* LLC. Start here: http://www.zerotier.com/
|
||||
*/
|
||||
|
||||
#include "TestEthernetTap.hpp"
|
||||
#include "TestEthernetTapFactory.hpp"
|
||||
|
||||
#include "../node/Constants.hpp"
|
||||
#include "../node/Utils.hpp"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
#ifdef __WINDOWS__
|
||||
#include <process.h>
|
||||
#else
|
||||
#include <unistd.h>
|
||||
#endif
|
||||
|
||||
namespace ZeroTier {
|
||||
|
||||
TestEthernetTap::TestEthernetTap(
|
||||
const MAC &mac,
|
||||
unsigned int mtu,
|
||||
unsigned int metric,
|
||||
uint64_t nwid,
|
||||
const char *desiredDevice,
|
||||
const char *friendlyName,
|
||||
void (*handler)(void *,const MAC &,const MAC &,unsigned int,const Buffer<4096> &),
|
||||
void *arg) :
|
||||
EthernetTap("TestEthernetTap",mac,mtu,metric),
|
||||
_nwid(nwid),
|
||||
_handler(handler),
|
||||
_arg(arg),
|
||||
_enabled(true)
|
||||
{
|
||||
static volatile unsigned int testTapCounter = 0;
|
||||
|
||||
char tmp[64];
|
||||
int pid = 0;
|
||||
#ifdef __UNIX_LIKE__
|
||||
pid = (int)getpid();
|
||||
#endif
|
||||
#ifdef __WINDOWS__
|
||||
pid = (int)_getpid();
|
||||
#endif
|
||||
Utils::snprintf(tmp,sizeof(tmp),"test%dtap%d",pid,testTapCounter++);
|
||||
_dev = tmp;
|
||||
|
||||
_thread = Thread::start(this);
|
||||
}
|
||||
|
||||
TestEthernetTap::~TestEthernetTap()
|
||||
{
|
||||
static const TestFrame zf; // use a static empty frame because of weirdo G++ warning bug...
|
||||
_pq.push(zf); // empty frame terminates thread
|
||||
Thread::join(_thread);
|
||||
}
|
||||
|
||||
void TestEthernetTap::setEnabled(bool en)
|
||||
{
|
||||
_enabled = en;
|
||||
}
|
||||
|
||||
bool TestEthernetTap::enabled() const
|
||||
{
|
||||
return _enabled;
|
||||
}
|
||||
|
||||
bool TestEthernetTap::addIP(const InetAddress &ip)
|
||||
{
|
||||
return true;
|
||||
}
|
||||
|
||||
bool TestEthernetTap::removeIP(const InetAddress &ip)
|
||||
{
|
||||
return true;
|
||||
}
|
||||
|
||||
std::set<InetAddress> TestEthernetTap::ips() const
|
||||
{
|
||||
return std::set<InetAddress>();
|
||||
}
|
||||
|
||||
void TestEthernetTap::put(const MAC &from,const MAC &to,unsigned int etherType,const void *data,unsigned int len)
|
||||
{
|
||||
_gq.push(TestFrame(from,to,data,etherType,len));
|
||||
}
|
||||
|
||||
std::string TestEthernetTap::deviceName() const
|
||||
{
|
||||
return _dev;
|
||||
}
|
||||
|
||||
void TestEthernetTap::setFriendlyName(const char *friendlyName)
|
||||
{
|
||||
}
|
||||
|
||||
bool TestEthernetTap::updateMulticastGroups(std::set<MulticastGroup> &groups)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
bool TestEthernetTap::injectPacketFromHost(const MAC &from,const MAC &to,unsigned int etherType,const void *data,unsigned int len)
|
||||
{
|
||||
if ((len == 0)||(len > _mtu))
|
||||
return false;
|
||||
_pq.push(TestFrame(from,to,data,etherType & 0xffff,len));
|
||||
return true;
|
||||
}
|
||||
|
||||
void TestEthernetTap::threadMain()
|
||||
throw()
|
||||
{
|
||||
TestFrame f;
|
||||
for(;;) {
|
||||
if (_pq.pop(f,0)) {
|
||||
if (f.len) {
|
||||
try {
|
||||
_handler(_arg,f.from,f.to,f.etherType,Buffer<4096>(f.data,f.len));
|
||||
} catch ( ... ) {}
|
||||
} else break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
} // namespace ZeroTier
|
@ -1,124 +0,0 @@
|
||||
/*
|
||||
* ZeroTier One - Network Virtualization Everywhere
|
||||
* Copyright (C) 2011-2015 ZeroTier, Inc.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* --
|
||||
*
|
||||
* ZeroTier may be used and distributed under the terms of the GPLv3, which
|
||||
* are available at: http://www.gnu.org/licenses/gpl-3.0.html
|
||||
*
|
||||
* If you would like to embed ZeroTier into a commercial application or
|
||||
* redistribute it in a modified binary form, please contact ZeroTier Networks
|
||||
* LLC. Start here: http://www.zerotier.com/
|
||||
*/
|
||||
|
||||
#ifndef ZT_TESTETHERNETTAP_HPP
|
||||
#define ZT_TESTETHERNETTAP_HPP
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#include <string>
|
||||
|
||||
#include "../node/Constants.hpp"
|
||||
#include "../node/EthernetTap.hpp"
|
||||
#include "../node/Thread.hpp"
|
||||
#include "../node/Mutex.hpp"
|
||||
|
||||
#include "MTQ.hpp"
|
||||
|
||||
namespace ZeroTier {
|
||||
|
||||
class TestEthernetTapFactory;
|
||||
|
||||
/**
|
||||
* Dummy Ethernet tap
|
||||
*
|
||||
* This tap device prints the contents of packets it receives on stdout
|
||||
* and also prints outgoing packets when they are injected. It does not
|
||||
* connect to any real tap or other interface. It's useful for running
|
||||
* test networks.
|
||||
*/
|
||||
class TestEthernetTap : public EthernetTap
|
||||
{
|
||||
public:
|
||||
struct TestFrame
|
||||
{
|
||||
TestFrame() : from(),to(),timestamp(0),etherType(0),len(0) {}
|
||||
TestFrame(const MAC &f,const MAC &t,const void *d,unsigned int et,unsigned int l) :
|
||||
from(f),
|
||||
to(t),
|
||||
timestamp(Utils::now()),
|
||||
etherType(et),
|
||||
len(l)
|
||||
{
|
||||
memcpy(data,d,l);
|
||||
}
|
||||
|
||||
MAC from;
|
||||
MAC to;
|
||||
uint64_t timestamp;
|
||||
unsigned int etherType;
|
||||
unsigned int len;
|
||||
char data[4096];
|
||||
};
|
||||
|
||||
TestEthernetTap(
|
||||
const MAC &mac,
|
||||
unsigned int mtu,
|
||||
unsigned int metric,
|
||||
uint64_t nwid,
|
||||
const char *desiredDevice,
|
||||
const char *friendlyName,
|
||||
void (*handler)(void *,const MAC &,const MAC &,unsigned int,const Buffer<4096> &),
|
||||
void *arg);
|
||||
|
||||
virtual ~TestEthernetTap();
|
||||
|
||||
virtual void setEnabled(bool en);
|
||||
virtual bool enabled() const;
|
||||
virtual bool addIP(const InetAddress &ip);
|
||||
virtual bool removeIP(const InetAddress &ip);
|
||||
virtual std::set<InetAddress> ips() const;
|
||||
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 bool updateMulticastGroups(std::set<MulticastGroup> &groups);
|
||||
virtual bool injectPacketFromHost(const MAC &from,const MAC &to,unsigned int etherType,const void *data,unsigned int len);
|
||||
|
||||
inline uint64_t nwid() const { return _nwid; }
|
||||
inline bool getNextReceivedFrame(TestFrame &v,unsigned long timeout) { return _gq.pop(v,timeout); }
|
||||
|
||||
void threadMain()
|
||||
throw();
|
||||
|
||||
private:
|
||||
uint64_t _nwid;
|
||||
|
||||
void (*_handler)(void *,const MAC &,const MAC &,unsigned int,const Buffer<4096> &);
|
||||
void *_arg;
|
||||
Thread _thread;
|
||||
std::string _dev;
|
||||
volatile bool _enabled;
|
||||
|
||||
MTQ<TestFrame> _pq;
|
||||
MTQ<TestFrame> _gq;
|
||||
};
|
||||
|
||||
} // namespace ZeroTier
|
||||
|
||||
#endif
|
@ -1,79 +0,0 @@
|
||||
/*
|
||||
* ZeroTier One - Network Virtualization Everywhere
|
||||
* Copyright (C) 2011-2015 ZeroTier, Inc.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* --
|
||||
*
|
||||
* ZeroTier may be used and distributed under the terms of the GPLv3, which
|
||||
* are available at: http://www.gnu.org/licenses/gpl-3.0.html
|
||||
*
|
||||
* If you would like to embed ZeroTier into a commercial application or
|
||||
* redistribute it in a modified binary form, please contact ZeroTier Networks
|
||||
* LLC. Start here: http://www.zerotier.com/
|
||||
*/
|
||||
|
||||
#include "TestEthernetTapFactory.hpp"
|
||||
#include "TestEthernetTap.hpp"
|
||||
|
||||
namespace ZeroTier {
|
||||
|
||||
TestEthernetTapFactory::TestEthernetTapFactory()
|
||||
{
|
||||
}
|
||||
|
||||
TestEthernetTapFactory::~TestEthernetTapFactory()
|
||||
{
|
||||
Mutex::Lock _l1(_taps_m);
|
||||
Mutex::Lock _l2(_tapsByMac_m);
|
||||
Mutex::Lock _l3(_tapsByNwid_m);
|
||||
for(std::set<EthernetTap *>::iterator t(_taps.begin());t!=_taps.end();++t)
|
||||
delete *t;
|
||||
}
|
||||
|
||||
EthernetTap *TestEthernetTapFactory::open(
|
||||
const MAC &mac,
|
||||
unsigned int mtu,
|
||||
unsigned int metric,
|
||||
uint64_t nwid,
|
||||
const char *desiredDevice,
|
||||
const char *friendlyName,
|
||||
void (*handler)(void *,const MAC &,const MAC &,unsigned int,const Buffer<4096> &),
|
||||
void *arg)
|
||||
{
|
||||
TestEthernetTap *tap = new TestEthernetTap(mac,mtu,metric,nwid,desiredDevice,friendlyName,handler,arg);
|
||||
Mutex::Lock _l1(_taps_m);
|
||||
Mutex::Lock _l2(_tapsByMac_m);
|
||||
Mutex::Lock _l3(_tapsByNwid_m);
|
||||
_taps.insert(tap);
|
||||
_tapsByMac[mac] = tap;
|
||||
_tapsByNwid[nwid] = tap;
|
||||
return tap;
|
||||
}
|
||||
|
||||
void TestEthernetTapFactory::close(EthernetTap *tap,bool destroyPersistentDevices)
|
||||
{
|
||||
Mutex::Lock _l1(_taps_m);
|
||||
Mutex::Lock _l2(_tapsByMac_m);
|
||||
Mutex::Lock _l3(_tapsByNwid_m);
|
||||
if (!tap)
|
||||
return;
|
||||
_taps.erase(tap);
|
||||
_tapsByMac.erase(tap->mac());
|
||||
_tapsByNwid.erase(((TestEthernetTap *)tap)->nwid());
|
||||
delete tap;
|
||||
}
|
||||
|
||||
} // namespace ZeroTier
|
@ -1,91 +0,0 @@
|
||||
/*
|
||||
* ZeroTier One - Network Virtualization Everywhere
|
||||
* Copyright (C) 2011-2015 ZeroTier, Inc.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* --
|
||||
*
|
||||
* ZeroTier may be used and distributed under the terms of the GPLv3, which
|
||||
* are available at: http://www.gnu.org/licenses/gpl-3.0.html
|
||||
*
|
||||
* If you would like to embed ZeroTier into a commercial application or
|
||||
* redistribute it in a modified binary form, please contact ZeroTier Networks
|
||||
* LLC. Start here: http://www.zerotier.com/
|
||||
*/
|
||||
|
||||
#ifndef ZT_TESTETHERNETTAPFACTORY_HPP
|
||||
#define ZT_TESTETHERNETTAPFACTORY_HPP
|
||||
|
||||
#include <vector>
|
||||
#include <string>
|
||||
#include <set>
|
||||
|
||||
#include "../node/EthernetTapFactory.hpp"
|
||||
#include "../node/Mutex.hpp"
|
||||
#include "../node/MAC.hpp"
|
||||
#include "TestEthernetTap.hpp"
|
||||
|
||||
namespace ZeroTier {
|
||||
|
||||
class TestEthernetTapFactory : public EthernetTapFactory
|
||||
{
|
||||
public:
|
||||
TestEthernetTapFactory();
|
||||
virtual ~TestEthernetTapFactory();
|
||||
|
||||
virtual EthernetTap *open(
|
||||
const MAC &mac,
|
||||
unsigned int mtu,
|
||||
unsigned int metric,
|
||||
uint64_t nwid,
|
||||
const char *desiredDevice,
|
||||
const char *friendlyName,
|
||||
void (*handler)(void *,const MAC &,const MAC &,unsigned int,const Buffer<4096> &),
|
||||
void *arg);
|
||||
|
||||
virtual void close(EthernetTap *tap,bool destroyPersistentDevices);
|
||||
|
||||
inline TestEthernetTap *getByMac(const MAC &mac) const
|
||||
{
|
||||
Mutex::Lock _l(_tapsByMac_m);
|
||||
std::map< MAC,TestEthernetTap * >::const_iterator t(_tapsByMac.find(mac));
|
||||
if (t == _tapsByMac.end())
|
||||
return (TestEthernetTap *)0;
|
||||
return t->second;
|
||||
}
|
||||
|
||||
inline TestEthernetTap *getByNwid(uint64_t nwid) const
|
||||
{
|
||||
Mutex::Lock _l(_tapsByNwid_m);
|
||||
std::map< uint64_t,TestEthernetTap * >::const_iterator t(_tapsByNwid.find(nwid));
|
||||
if (t == _tapsByNwid.end())
|
||||
return (TestEthernetTap *)0;
|
||||
return t->second;
|
||||
}
|
||||
|
||||
private:
|
||||
std::set< EthernetTap * > _taps;
|
||||
Mutex _taps_m;
|
||||
|
||||
std::map< MAC,TestEthernetTap * > _tapsByMac;
|
||||
Mutex _tapsByMac_m;
|
||||
|
||||
std::map< uint64_t,TestEthernetTap * > _tapsByNwid;
|
||||
Mutex _tapsByNwid_m;
|
||||
};
|
||||
|
||||
} // namespace ZeroTier
|
||||
|
||||
#endif
|
Loading…
Reference in New Issue
Block a user