You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
329 lines
11 KiB
329 lines
11 KiB
/* |
|
* ZeroTier One - Global Peer to Peer Ethernet |
|
* Copyright (C) 2011-2014 ZeroTier Networks 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/ |
|
*/ |
|
|
|
#include <stdio.h> |
|
#include <string.h> |
|
#include <stdlib.h> |
|
#include <stdint.h> |
|
|
|
#include <memory> |
|
#include <string> |
|
#include <map> |
|
#include <set> |
|
|
|
#include "Constants.hpp" |
|
|
|
#include "NodeConfig.hpp" |
|
#include "RuntimeEnvironment.hpp" |
|
#include "Defaults.hpp" |
|
#include "Utils.hpp" |
|
#include "Logger.hpp" |
|
#include "Topology.hpp" |
|
#include "Packet.hpp" |
|
#include "InetAddress.hpp" |
|
#include "Peer.hpp" |
|
#include "Node.hpp" |
|
#include "SoftwareUpdater.hpp" |
|
|
|
namespace ZeroTier { |
|
|
|
NodeConfig::NodeConfig(const RuntimeEnvironment *renv,const char *authToken) : |
|
_r(renv), |
|
_ipcListener((std::string(ZT_IPC_ENDPOINT_BASE) + renv->identity.address().toString()).c_str(),&_CBcommandHandler,this), |
|
_authToken(authToken) |
|
{ |
|
{ |
|
Mutex::Lock _l(_localConfig_m); |
|
_readLocalConfig(); |
|
} |
|
|
|
std::string networksFolder(_r->homePath + ZT_PATH_SEPARATOR_S + "networks.d"); |
|
std::map<std::string,bool> networksDotD(Utils::listDirectory(networksFolder.c_str())); |
|
std::vector<uint64_t> configuredNets; |
|
for(std::map<std::string,bool>::iterator d(networksDotD.begin());d!=networksDotD.end();++d) { |
|
if (!d->second) { |
|
std::string::size_type dot = d->first.rfind(".conf"); |
|
if (dot != std::string::npos) { |
|
uint64_t nwid = Utils::hexStrToU64(d->first.substr(0,dot).c_str()); |
|
if ((nwid > 0)&&(std::find(configuredNets.begin(),configuredNets.end(),nwid) == configuredNets.end())) |
|
configuredNets.push_back(nwid); |
|
} |
|
} |
|
} |
|
|
|
for(std::vector<uint64_t>::iterator n(configuredNets.begin());n!=configuredNets.end();++n) { |
|
try { |
|
_networks[*n] = Network::newInstance(_r,this,*n); |
|
} catch (std::exception &exc) { |
|
LOG("unable to create network %.16llx: %s",(unsigned long long)*n,exc.what()); |
|
} catch ( ... ) { |
|
LOG("unable to create network %.16llx: (unknown exception)",(unsigned long long)*n); |
|
} |
|
} |
|
} |
|
|
|
NodeConfig::~NodeConfig() |
|
{ |
|
_writeLocalConfig(); |
|
|
|
// Close any open IPC connections |
|
Mutex::Lock _l(_connections_m); |
|
for(std::map< IpcConnection *,bool >::iterator c(_connections.begin());c!=_connections.end();++c) |
|
delete c->first; |
|
_connections.clear(); |
|
} |
|
|
|
void NodeConfig::putLocalConfig(const std::string &key,const char *value) |
|
{ |
|
Mutex::Lock _l(_localConfig_m); |
|
_localConfig[key] = value; |
|
_writeLocalConfig(); |
|
} |
|
|
|
void NodeConfig::putLocalConfig(const std::string &key,const std::string &value) |
|
{ |
|
Mutex::Lock _l(_localConfig_m); |
|
_localConfig[key] = value; |
|
_writeLocalConfig(); |
|
} |
|
|
|
std::string NodeConfig::getLocalConfig(const std::string &key) const |
|
{ |
|
Mutex::Lock _l(_localConfig_m); |
|
Dictionary::const_iterator i(_localConfig.find(key)); |
|
if (i == _localConfig.end()) |
|
return std::string(); |
|
return i->second; |
|
} |
|
|
|
void NodeConfig::clean() |
|
{ |
|
Mutex::Lock _l(_networks_m); |
|
for(std::map< uint64_t,SharedPtr<Network> >::const_iterator n(_networks.begin());n!=_networks.end();++n) |
|
n->second->clean(); |
|
} |
|
|
|
void NodeConfig::_CBcommandHandler(void *arg,IpcConnection *ipcc,IpcConnection::EventType event,const char *commandLine) |
|
{ |
|
switch(event) { |
|
case IpcConnection::IPC_EVENT_COMMAND: |
|
((NodeConfig *)arg)->_doCommand(ipcc,commandLine); |
|
break; |
|
case IpcConnection::IPC_EVENT_NEW_CONNECTION: { |
|
Mutex::Lock _l(((NodeConfig *)arg)->_connections_m); |
|
((NodeConfig *)arg)->_connections[ipcc] = false; // not yet authenticated |
|
} break; |
|
case IpcConnection::IPC_EVENT_CONNECTION_CLOSED: { |
|
Mutex::Lock _l(((NodeConfig *)arg)->_connections_m); |
|
((NodeConfig *)arg)->_connections.erase(ipcc); |
|
delete ipcc; |
|
} break; |
|
} |
|
} |
|
|
|
// Used with Topology::eachPeer to dump peer stats |
|
class _DumpPeerStatistics |
|
{ |
|
public: |
|
_DumpPeerStatistics(IpcConnection *i) : |
|
ipcc(i), |
|
now(Utils::now()) |
|
{ |
|
} |
|
inline void operator()(Topology &t,const SharedPtr<Peer> &p) |
|
{ |
|
std::vector<Path> pp(p->paths()); |
|
std::string pathsStr; |
|
for(std::vector<Path>::const_iterator ppp(pp.begin());ppp!=pp.end();++ppp) { |
|
if (pathsStr.length()) |
|
pathsStr.push_back(','); |
|
pathsStr.append(ppp->toString()); |
|
} |
|
ipcc->printf("200 listpeers %s %s %u %s"ZT_EOL_S, |
|
p->address().toString().c_str(), |
|
((pathsStr.length() > 0) ? pathsStr.c_str() : "-"), |
|
p->latency(), |
|
p->remoteVersion().c_str()); |
|
} |
|
IpcConnection *ipcc; |
|
uint64_t now; |
|
}; |
|
|
|
void NodeConfig::_doCommand(IpcConnection *ipcc,const char *commandLine) |
|
{ |
|
if ((!commandLine)||(!commandLine[0])) |
|
return; |
|
std::vector<std::string> r; |
|
std::vector<std::string> cmd(Utils::split(commandLine,"\r\n \t","\\","'")); |
|
|
|
if ((cmd.empty())||(cmd[0] == "help")) { |
|
ipcc->printf("200 help help"ZT_EOL_S); |
|
ipcc->printf("200 help auth <token>"ZT_EOL_S); |
|
ipcc->printf("200 help info"ZT_EOL_S); |
|
ipcc->printf("200 help listpeers"ZT_EOL_S); |
|
ipcc->printf("200 help listnetworks"ZT_EOL_S); |
|
ipcc->printf("200 help join <network ID>"ZT_EOL_S); |
|
ipcc->printf("200 help leave <network ID>"ZT_EOL_S); |
|
ipcc->printf("200 help terminate [<reason>]"ZT_EOL_S); |
|
ipcc->printf("200 help updatecheck"ZT_EOL_S); |
|
} else if (cmd[0] == "auth") { |
|
if ((cmd.size() > 1)&&(_authToken == cmd[1])) { |
|
Mutex::Lock _l(_connections_m); |
|
_connections[ipcc] = true; |
|
ipcc->printf("200 auth OK"ZT_EOL_S); |
|
} else ipcc->printf("403 auth failed"ZT_EOL_S); |
|
} else { |
|
{ |
|
Mutex::Lock _l(_connections_m); |
|
if (!_connections[ipcc]) { |
|
ipcc->printf("403 %s unauthorized"ZT_EOL_S"."ZT_EOL_S,cmd[0].c_str()); |
|
return; |
|
} |
|
} |
|
|
|
if (cmd[0] == "info") { |
|
// We are online if at least one supernode has spoken to us since the last time our |
|
// network environment changed and also less than ZT_PEER_LINK_ACTIVITY_TIMEOUT ago. |
|
bool isOnline = false; |
|
uint64_t now = Utils::now(); |
|
uint64_t since = _r->timeOfLastResynchronize; |
|
std::vector< SharedPtr<Peer> > snp(_r->topology->supernodePeers()); |
|
for(std::vector< SharedPtr<Peer> >::const_iterator sn(snp.begin());sn!=snp.end();++sn) { |
|
uint64_t lastRec = (*sn)->lastDirectReceive(); |
|
if ((lastRec)&&(lastRec > since)&&((now - lastRec) < ZT_PEER_PATH_ACTIVITY_TIMEOUT)) { |
|
isOnline = true; |
|
break; |
|
} |
|
} |
|
|
|
ipcc->printf("200 info %s %s %s"ZT_EOL_S,_r->identity.address().toString().c_str(),(isOnline ? "ONLINE" : "OFFLINE"),Node::versionString()); |
|
} else if (cmd[0] == "listpeers") { |
|
ipcc->printf("200 listpeers <ztaddr> <paths> <latency> <version>"ZT_EOL_S); |
|
_r->topology->eachPeer(_DumpPeerStatistics(ipcc)); |
|
} else if (cmd[0] == "listnetworks") { |
|
Mutex::Lock _l(_networks_m); |
|
ipcc->printf("200 listnetworks <nwid> <name> <mac> <status> <config age> <type> <dev> <ips>"ZT_EOL_S); |
|
for(std::map< uint64_t,SharedPtr<Network> >::const_iterator nw(_networks.begin());nw!=_networks.end();++nw) { |
|
std::string tmp; |
|
std::set<InetAddress> ips(nw->second->ips()); |
|
for(std::set<InetAddress>::iterator i(ips.begin());i!=ips.end();++i) { |
|
if (tmp.length()) |
|
tmp.push_back(','); |
|
tmp.append(i->toString()); |
|
} |
|
|
|
SharedPtr<NetworkConfig> nconf(nw->second->config2()); |
|
|
|
long long age = (nconf) ? ((long long)Utils::now() - (long long)nconf->timestamp()) : (long long)0; |
|
if (age < 0) |
|
age = 0; |
|
age /= 1000; |
|
|
|
std::string dn(nw->second->tapDeviceName()); |
|
ipcc->printf("200 listnetworks %.16llx %s %s %s %lld %s %s %s"ZT_EOL_S, |
|
(unsigned long long)nw->first, |
|
((nconf) ? nconf->name().c_str() : "?"), |
|
nw->second->mac().toString().c_str(), |
|
Network::statusString(nw->second->status()), |
|
age, |
|
((nconf) ? (nconf->isPublic() ? "public" : "private") : "?"), |
|
(dn.length() > 0) ? dn.c_str() : "?", |
|
((tmp.length() > 0) ? tmp.c_str() : "-")); |
|
} |
|
} else if (cmd[0] == "join") { |
|
if (cmd.size() > 1) { |
|
uint64_t nwid = Utils::hexStrToU64(cmd[1].c_str()); |
|
if (nwid > 0) { |
|
Mutex::Lock _l(_networks_m); |
|
if (_networks.count(nwid)) { |
|
ipcc->printf("409 already a member of %.16llx"ZT_EOL_S,(unsigned long long)nwid); |
|
} else { |
|
try { |
|
SharedPtr<Network> nw(Network::newInstance(_r,this,nwid)); |
|
_networks[nwid] = nw; |
|
ipcc->printf("200 join %.16llx OK"ZT_EOL_S,(unsigned long long)nwid); |
|
} catch (std::exception &exc) { |
|
ipcc->printf("500 join %.16llx ERROR: %s"ZT_EOL_S,(unsigned long long)nwid,exc.what()); |
|
} catch ( ... ) { |
|
ipcc->printf("500 join %.16llx ERROR: (unknown exception)"ZT_EOL_S,(unsigned long long)nwid); |
|
} |
|
} |
|
} else { |
|
ipcc->printf("400 join requires a network ID (>0) in hexadecimal format"ZT_EOL_S); |
|
} |
|
} else { |
|
ipcc->printf("400 join requires a network ID (>0) in hexadecimal format"ZT_EOL_S); |
|
} |
|
} else if (cmd[0] == "leave") { |
|
if (cmd.size() > 1) { |
|
Mutex::Lock _l(_networks_m); |
|
uint64_t nwid = Utils::hexStrToU64(cmd[1].c_str()); |
|
std::map< uint64_t,SharedPtr<Network> >::iterator nw(_networks.find(nwid)); |
|
if (nw == _networks.end()) { |
|
ipcc->printf("404 leave %.16llx ERROR: not a member of that network"ZT_EOL_S,(unsigned long long)nwid); |
|
} else { |
|
nw->second->destroyOnDelete(); |
|
_networks.erase(nw); |
|
} |
|
} else { |
|
ipcc->printf("400 leave requires a network ID (>0) in hexadecimal format"ZT_EOL_S); |
|
} |
|
} else if (cmd[0] == "terminate") { |
|
if (cmd.size() > 1) |
|
_r->node->terminate(Node::NODE_NORMAL_TERMINATION,cmd[1].c_str()); |
|
else _r->node->terminate(Node::NODE_NORMAL_TERMINATION,"terminate via IPC command"); |
|
} else if (cmd[0] == "updatecheck") { |
|
if (_r->updater) { |
|
ipcc->printf("200 checking for software updates now at: %s"ZT_EOL_S,ZT_DEFAULTS.updateLatestNfoURL.c_str()); |
|
_r->updater->checkNow(); |
|
} else { |
|
ipcc->printf("500 software updates are not enabled"ZT_EOL_S); |
|
} |
|
} else { |
|
ipcc->printf("404 %s No such command. Use 'help' for help."ZT_EOL_S,cmd[0].c_str()); |
|
} |
|
} |
|
|
|
ipcc->printf("."ZT_EOL_S); // blank line ends response |
|
} |
|
|
|
void NodeConfig::_readLocalConfig() |
|
{ |
|
// assumes _localConfig_m is locked |
|
std::string localDotConf(_r->homePath + ZT_PATH_SEPARATOR_S + "local.conf"); |
|
std::string buf; |
|
if (Utils::readFile(localDotConf.c_str(),buf)) |
|
_localConfig.fromString(buf.c_str()); |
|
} |
|
|
|
void NodeConfig::_writeLocalConfig() |
|
{ |
|
// assumes _localConfig_m is locked |
|
Utils::writeFile(((_r->homePath + ZT_PATH_SEPARATOR_S + "local.conf")).c_str(),_localConfig.toString()); |
|
} |
|
|
|
} // namespace ZeroTier
|
|
|