/* * 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 . * * -- * * 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 "NodeControlClient.hpp" #include "../node/Constants.hpp" #include "../node/Utils.hpp" #include "../node/Defaults.hpp" #include "IpcConnection.hpp" #include "IpcListener.hpp" #include "NodeControlService.hpp" namespace ZeroTier { struct _NodeControlClientImpl { void (*resultHandler)(void *,const char *); void *arg; bool ignoreNextBreak; IpcConnection *ipcc; std::string err; }; static void _CBipcResultHandler(void *arg,IpcConnection *ipcc,IpcConnection::EventType event,const char *result) { if ((event == IpcConnection::IPC_EVENT_COMMAND)&&(result)) { if (!strcmp(result,"200 auth OK")) { ((_NodeControlClientImpl *)arg)->ignoreNextBreak = true; } else if ((((_NodeControlClientImpl *)arg)->ignoreNextBreak)&&(!strcmp(result,"."))) { ((_NodeControlClientImpl *)arg)->ignoreNextBreak = false; } else ((_NodeControlClientImpl *)arg)->resultHandler(((_NodeControlClientImpl *)arg)->arg,result); } } NodeControlClient::NodeControlClient(const char *ep,const char *authToken,void (*resultHandler)(void *,const char *),void *arg) throw() : _impl((void *)new _NodeControlClientImpl) { _NodeControlClientImpl *impl = (_NodeControlClientImpl *)_impl; impl->resultHandler = resultHandler; impl->arg = arg; impl->ignoreNextBreak = false; try { impl->ipcc = new IpcConnection(ep,ZT_IPC_TIMEOUT,&_CBipcResultHandler,_impl); impl->ipcc->printf("auth %s"ZT_EOL_S,authToken); } catch ( ... ) { impl->ipcc = (IpcConnection *)0; impl->err = "failure connecting to running ZeroTier One service"; } } NodeControlClient::~NodeControlClient() { if (_impl) { delete ((_NodeControlClientImpl *)_impl)->ipcc; delete (_NodeControlClientImpl *)_impl; } } const char *NodeControlClient::error() const throw() { if (((_NodeControlClientImpl *)_impl)->err.length()) return ((_NodeControlClientImpl *)_impl)->err.c_str(); return (const char *)0; } void NodeControlClient::send(const char *command) throw() { try { if (((_NodeControlClientImpl *)_impl)->ipcc) ((_NodeControlClientImpl *)_impl)->ipcc->printf("%s"ZT_EOL_S,command); } catch ( ... ) {} } std::vector NodeControlClient::splitLine(const char *line) { return Utils::split(line," ","\\","\""); } const char *NodeControlClient::authTokenDefaultUserPath() { static std::string dlp; static Mutex dlp_m; Mutex::Lock _l(dlp_m); #ifdef __WINDOWS__ if (!dlp.length()) { char buf[16384]; if (SUCCEEDED(SHGetFolderPathA(NULL,CSIDL_APPDATA,NULL,0,buf))) dlp = (std::string(buf) + "\\ZeroTier\\One\\authtoken.secret"); } #else // not __WINDOWS__ if (!dlp.length()) { const char *home = getenv("HOME"); if (home) { #ifdef __APPLE__ dlp = (std::string(home) + "/Library/Application Support/ZeroTier/One/authtoken.secret"); #else dlp = (std::string(home) + "/.zeroTierOneAuthToken"); #endif } } #endif // __WINDOWS__ or not __WINDOWS__ return dlp.c_str(); } std::string NodeControlClient::getAuthToken(const char *path,bool generateIfNotFound) { unsigned char randbuf[24]; std::string token; if (Utils::readFile(path,token)) return Utils::trim(token); else token = ""; if (generateIfNotFound) { Utils::getSecureRandom(randbuf,sizeof(randbuf)); for(unsigned int i=0;i