Adam Ierymenko
11 years ago
8 changed files with 272 additions and 23 deletions
@ -0,0 +1,125 @@ |
|||||||
|
/*
|
||||||
|
* 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_CLUSTERDEFINITION_HPP |
||||||
|
#define ZT_CLUSTERDEFINITION_HPP |
||||||
|
|
||||||
|
#ifdef ZT_ENABLE_CLUSTER |
||||||
|
|
||||||
|
#include <vector> |
||||||
|
#include <algorithm> |
||||||
|
|
||||||
|
#include "../node/Constants.hpp" |
||||||
|
#include "../node/Utils.hpp" |
||||||
|
#include "../osdep/OSUtils.hpp" |
||||||
|
|
||||||
|
namespace ZeroTier { |
||||||
|
|
||||||
|
/**
|
||||||
|
* Parser for cluster definition file |
||||||
|
*/ |
||||||
|
class ClusterDefinition |
||||||
|
{ |
||||||
|
public: |
||||||
|
struct MemberDefinition |
||||||
|
{ |
||||||
|
MemberDefinition() : id(0),x(0),y(0),z(0) { name[0] = (char)0; } |
||||||
|
|
||||||
|
unsigned int id; |
||||||
|
int x,y,z; |
||||||
|
char name[256]; |
||||||
|
InetAddress clusterEndpoint; |
||||||
|
std::vector<InetAddress> zeroTierEndpoints; |
||||||
|
}; |
||||||
|
|
||||||
|
ClusterDefinition(uint64_t myAddress,const char *pathToClusterFile) |
||||||
|
{ |
||||||
|
std::string cf; |
||||||
|
if (!OSUtils::readFile(pathToClusterFile,cf)) |
||||||
|
return; |
||||||
|
|
||||||
|
char myAddressStr[64]; |
||||||
|
Utils::snprintf(myAddressStr,sizeof(myAddressStr),"%.10llx",myAddress); |
||||||
|
|
||||||
|
std::vector<std::string> lines(Utils::split(cf.c_str(),"\r\n","","")); |
||||||
|
for(std::vector<std::string>::iterator l(lines.begin());l!=lines.end();++l) { |
||||||
|
std::vector<std::string> fields(Utils::split(l->c_str()," \t","","")); |
||||||
|
if ((fields.size() < 5)||(fields[0][0] == '#')||(fields[0] != myAddressStr)) |
||||||
|
continue; |
||||||
|
|
||||||
|
int id = Utils::strToUInt(fields[1].c_str()); |
||||||
|
if ((id < 0)||(id > ZT_CLUSTER_MAX_MEMBERS)) |
||||||
|
continue; |
||||||
|
MemberDefinition &md = _md[id]; |
||||||
|
|
||||||
|
md.id = (unsigned int)id; |
||||||
|
if (fields.size() >= 6) { |
||||||
|
std::vector<std::string> xyz(Utils::split(fields[5].c_str(),",","","")); |
||||||
|
md.x = (xyz.size() > 0) ? Utils::strToInt(xyz[0].c_str()) : 0; |
||||||
|
md.y = (xyz.size() > 1) ? Utils::strToInt(xyz[1].c_str()) : 0; |
||||||
|
md.z = (xyz.size() > 2) ? Utils::strToInt(xyz[2].c_str()) : 0; |
||||||
|
} |
||||||
|
Utils::scopy(md.name,sizeof(md.name),fields[2].c_str()); |
||||||
|
md.clusterEndpoint.fromString(fields[3]); |
||||||
|
if (!md.clusterEndpoint) |
||||||
|
continue; |
||||||
|
std::vector<std::string> zips(Utils::split(fields[4].c_str(),",","","")); |
||||||
|
for(std::vector<std::string>::iterator zip(zips.begin());zip!=zips.end();++zip) { |
||||||
|
InetAddress i; |
||||||
|
i.fromString(*zip); |
||||||
|
if (i) |
||||||
|
md.zeroTierEndpoints.push_back(i); |
||||||
|
} |
||||||
|
|
||||||
|
_ids.push_back((unsigned int)id); |
||||||
|
} |
||||||
|
|
||||||
|
std::sort(_ids.begin(),_ids.end()); |
||||||
|
} |
||||||
|
|
||||||
|
inline const MemberDefinition &operator[](unsigned int id) const throw() { return _md[id]; } |
||||||
|
inline unsigned int size() const throw() { return (unsigned int)_ids.size(); } |
||||||
|
inline const std::vector<unsigned int> &ids() const throw() { return _ids; } |
||||||
|
|
||||||
|
inline std::vector<MemberDefinition> members() const |
||||||
|
{ |
||||||
|
std::vector<MemberDefinition> m; |
||||||
|
for(std::vector<unsigned int>::const_iterator i(_ids.begin());i!=_ids.end();++i) |
||||||
|
m.push_back(_md[*i]); |
||||||
|
return m; |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
MemberDefinition _md[ZT_CLUSTER_MAX_MEMBERS]; |
||||||
|
std::vector<unsigned int> _ids; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace ZeroTier
|
||||||
|
|
||||||
|
#endif // ZT_ENABLE_CLUSTER
|
||||||
|
|
||||||
|
#endif |
||||||
Loading…
Reference in new issue