/* * ZeroTier SDK - Network Virtualization Everywhere * Copyright (C) 2011-2019 ZeroTier, Inc. https://www.zerotier.com/ * * 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 . * * -- * * You can be released from the requirements of the license by purchasing * a commercial license. Buying such a license is mandatory as soon as you * develop commercial closed-source software that incorporates or links * directly against ZeroTier software without disclosing the source code * of your own application. */ /** * @file * * ZeroTier service controls */ #include "OneService.hpp" #include "Node.hpp" namespace ZeroTier { std::vector vtaps; Mutex _vtaps_lock; } #include "Constants.hpp" #include "VirtualTapManager.hpp" #include "lwIP.h" #include "OSUtils.hpp" #include "ServiceControls.hpp" //#define SDK_JNI 1 #ifdef SDK_JNI #include #endif ZeroTier::Mutex _service_lock; ZeroTier::Mutex _startup_lock; static ZeroTier::OneService *zt1Service; std::string homeDir; int servicePort = ZTS_DEFAULT_PORT; bool _freeHasBeenCalled = false; bool _serviceIsShuttingDown = false; bool _startupError = false; #if defined(_WIN32) WSADATA wsaData; #include #endif pthread_t service_thread; using namespace ZeroTier; ////////////////////////////////////////////////////////////////////////////// // Internal ZeroTier Service Controls (user application shall not use these)// ////////////////////////////////////////////////////////////////////////////// void api_sleep(int interval_ms) { #if defined(_WIN32) Sleep(interval_ms); #else struct timespec sleepValue = {0}; sleepValue.tv_nsec = interval_ms * 500000; nanosleep(&sleepValue, NULL); #endif } int _zts_node_online() { return zt1Service && zt1Service->getNode() && zt1Service->getNode()->online(); } int _zts_can_perform_service_operation() { return zt1Service && zt1Service->isRunning() && zt1Service->getNode() && zt1Service->getNode()->online() && !_serviceIsShuttingDown; } void _hibernate_if_needed() { if (VirtualTapManager::get_vtaps_size()) { lwip_wake_driver(); } else { lwip_hibernate_driver(); } } #ifdef SDK_JNI #endif // Starts a ZeroTier service in the background #if defined(_WIN32) DWORD WINAPI _zts_start_service(LPVOID thread_id) #else void *_zts_start_service(void *thread_id) #endif { void *retval; //DEBUG_INFO("identities are stored in path (%s)", homeDir.c_str()); zt1Service = (OneService *)0; if (!homeDir.length()) { DEBUG_ERROR("homeDir is empty, could not construct path"); _startupError = true; retval = NULL; } if (zt1Service) { DEBUG_INFO("service already started, doing nothing"); retval = NULL; } try { std::vector hpsp(OSUtils::split(homeDir.c_str(), ZT_PATH_SEPARATOR_S,"","")); std::string ptmp; if (homeDir[0] == ZT_PATH_SEPARATOR) { ptmp.push_back(ZT_PATH_SEPARATOR); } for (std::vector::iterator pi(hpsp.begin());pi!=hpsp.end();++pi) { if (ptmp.length() > 0) { ptmp.push_back(ZT_PATH_SEPARATOR); } ptmp.append(*pi); if ((*pi != ".")&&(*pi != "..")) { if (OSUtils::mkdir(ptmp) == false) { DEBUG_ERROR("home path does not exist, and could not create"); _startupError = true; retval = NULL; perror("error\n"); } } } if (!_startupError) { for(;;) { _service_lock.lock(); zt1Service = OneService::newInstance(homeDir.c_str(),servicePort); _service_lock.unlock(); switch(zt1Service->run()) { case OneService::ONE_STILL_RUNNING: case OneService::ONE_NORMAL_TERMINATION: break; case OneService::ONE_UNRECOVERABLE_ERROR: DEBUG_ERROR("fatal error: %s", zt1Service->fatalErrorMessage().c_str()); _startupError = true; break; case OneService::ONE_IDENTITY_COLLISION: { _startupError = true; delete zt1Service; zt1Service = (OneService *)0; std::string oldid; OSUtils::readFile((homeDir + ZT_PATH_SEPARATOR_S + "identity.secret").c_str(),oldid); if (oldid.length()) { OSUtils::writeFile((homeDir + ZT_PATH_SEPARATOR_S + "identity.secret.saved_after_collision").c_str(),oldid); OSUtils::rm((homeDir + ZT_PATH_SEPARATOR_S + "identity.secret").c_str()); OSUtils::rm((homeDir + ZT_PATH_SEPARATOR_S + "identity.public").c_str()); } } continue; // restart! } break; // terminate loop -- normally we don't keep restarting } } _serviceIsShuttingDown = true; _service_lock.lock(); delete zt1Service; zt1Service = (OneService *)0; _service_lock.unlock(); _serviceIsShuttingDown = false; } catch ( ... ) { DEBUG_ERROR("unexpected exception starting ZeroTier instance"); } pthread_exit(NULL); } #ifdef __cplusplus extern "C" { #endif ////////////////////////////////////////////////////////////////////////////// // ZeroTier Service Controls // ////////////////////////////////////////////////////////////////////////////// zts_err_t zts_set_service_port(int portno) { zts_err_t retval = ZTS_ERR_OK; _service_lock.lock(); if (zt1Service) { // Stop service before attempting to set a port retval = ZTS_ERR_SERVICE; } else { if (portno > -1 && portno < ZTS_MAX_PORT) { // 0 is allowed, signals to ZT service to bind to a random port servicePort = portno; retval = ZTS_ERR_OK; } } _service_lock.unlock(); return retval; } #ifdef SDK_JNI JNIEXPORT void JNICALL Java_com_zerotier_libzt_ZeroTier_set_1service_1port( JNIEnv *env, jobject thisObj, jint port) { zts_set_service_port(port); } #endif int zts_get_service_port() { return servicePort; } #ifdef SDK_JNI JNIEXPORT jint JNICALL Java_com_zerotier_libzt_ZeroTier_get_1service_1port( JNIEnv *env, jobject thisObj) { return zts_get_service_port(); } #endif /* int zts_get_address(const uint64_t nwid, struct sockaddr_storage *addr, const int address_family) { int err = -1; if (!zt1Service) { return ZTS_ERR_SERVICE; } VirtualTap *tap = getTapByNWID(nwid); if (!tap) { return -1; } _vtaps_lock.lock(); socklen_t addrlen = address_family == AF_INET ? sizeof(struct sockaddr_in) : sizeof(struct sockaddr_in6); for (size_t i=0; i_ips.size(); i++) { if (address_family == AF_INET) { if (tap->_ips[i].isV4()) { memcpy(addr, &(tap->_ips[i]), addrlen); addr->ss_family = AF_INET; err = 0; break; } } if (address_family == AF_INET6) { if (tap->_ips[i].isV6()) { memcpy(addr, &(tap->_ips[i]), addrlen); addr->ss_family = AF_INET6; err = 0; break; } } } _vtaps_lock.unlock(); return err; // nothing found } #ifdef SDK_JNI JNIEXPORT jboolean JNICALL Java_com_zerotier_libzt_ZeroTier_get_1address( JNIEnv *env, jobject thisObj, jlong nwid, jint address_family, jobject addr) { struct sockaddr_storage ss; int err = zts_get_address((uint64_t)nwid, &ss, address_family); ss2zta(env, &ss, addr); return err; } #endif int zts_has_address(const uint64_t nwid) { struct sockaddr_storage ss; memset(&ss, 0, sizeof(ss)); zts_get_address(nwid, &ss, AF_INET); if (ss.ss_family == AF_INET) { return true; } zts_get_address(nwid, &ss, AF_INET6); if (ss.ss_family == AF_INET6) { return true; } return false; } #ifdef SDK_JNI JNIEXPORT jboolean JNICALL Java_com_zerotier_libzt_ZeroTier_has_1address( JNIEnv *env, jobject thisObj, jlong nwid) { return zts_has_address(nwid); } #endif */ /* void zts_get_6plane_addr(struct sockaddr_storage *addr, const uint64_t nwid, const uint64_t nodeId) { ZeroTier::InetAddress _6planeAddr = ZeroTier::InetAddress::makeIpv66plane(nwid,nodeId); memcpy(addr, _6planeAddr.rawIpData(), sizeof(struct sockaddr_storage)); } #ifdef SDK_JNI JNIEXPORT void JNICALL Java_com_zerotier_libzt_ZeroTier_get_16plane_1addr( JNIEnv *env, jobject thisObj, jlong nwid, jlong nodeId, jobject addr) { struct sockaddr_storage ss; zts_get_6plane_addr(&ss, nwid, nodeId); ss2zta(env, &ss, addr); } #endif void zts_get_rfc4193_addr(struct sockaddr_storage *addr, const uint64_t nwid, const uint64_t nodeId) { ZeroTier::InetAddress _rfc4193Addr = ZeroTier::InetAddress::makeIpv6rfc4193(nwid,nodeId); memcpy(addr, _rfc4193Addr.rawIpData(), sizeof(struct sockaddr_storage)); } #ifdef SDK_JNI JNIEXPORT void JNICALL Java_com_zerotier_libzt_ZeroTier_get_1rfc4193_1addr( JNIEnv *env, jobject thisObj, jlong nwid, jlong nodeId, jobject addr) { struct sockaddr_storage ss; zts_get_rfc4193_addr(&ss, nwid, nodeId); ss2zta(env, &ss, addr); } #endif */ zts_err_t zts_join(const uint64_t nwid, int blocking) { zts_err_t retval = ZTS_ERR_OK; retval = VirtualTapManager::get_vtaps_size() >= ZTS_MAX_JOINED_NETWORKS ? ZTS_ERR_INVALID_OP : ZTS_ERR_OK; if (retval == ZTS_ERR_OK) { _service_lock.lock(); if (blocking) { if (!zt1Service) { retval = ZTS_ERR_SERVICE; } else { while (!_zts_node_online()) { if (_serviceIsShuttingDown) { retval = ZTS_ERR_SERVICE; break; } api_sleep(ZTS_WRAPPER_CHECK_INTERVAL); } } } else { if (!zt1Service || !_zts_node_online()) { retval = ZTS_ERR_SERVICE; } } if (retval == ZTS_ERR_OK) { if (nwid == 0) { retval = ZTS_ERR_INVALID_ARG; } if (zt1Service) { zt1Service->getNode()->join(nwid, NULL, NULL); } VirtualTapManager::update_service_references((void*)zt1Service); } _service_lock.unlock(); _hibernate_if_needed(); } return retval; } #ifdef SDK_JNI JNIEXPORT jint JNICALL Java_com_zerotier_libzt_ZeroTier_join( JNIEnv *env, jobject thisObj, jlong nwid) { return zts_join((uint64_t)nwid); } #endif zts_err_t zts_leave(const uint64_t nwid, int blocking) { zts_err_t retval = ZTS_ERR_OK; _service_lock.lock(); if (blocking) { if (!zt1Service) { retval = ZTS_ERR_SERVICE; } else { while (!_zts_node_online()) { if (_serviceIsShuttingDown) { retval = ZTS_ERR_SERVICE; break; } api_sleep(ZTS_WRAPPER_CHECK_INTERVAL); } } } else { if (!zt1Service || !_zts_node_online()) { retval = ZTS_ERR_SERVICE; } } if (retval == ZTS_ERR_OK) { if (nwid == 0) { retval = ZTS_ERR_INVALID_ARG; } if (zt1Service) { zt1Service->getNode()->leave(nwid, NULL, NULL); } } VirtualTapManager::remove_by_nwid(nwid); _hibernate_if_needed(); _service_lock.unlock(); return retval; } #ifdef SDK_JNI JNIEXPORT jint JNICALL Java_com_zerotier_libzt_ZeroTier_leave( JNIEnv *env, jobject thisObj, jlong nwid) { return zts_leave((uint64_t)nwid); } #endif zts_err_t zts_leave_all(int blocking) { zts_err_t retval = ZTS_ERR_OK; if (!zt1Service || _freeHasBeenCalled || _serviceIsShuttingDown) { retval = ZTS_ERR_SERVICE; } else { struct zts_network_details nds[ZTS_MAX_JOINED_NETWORKS]; int numJoined = ZTS_MAX_JOINED_NETWORKS; zts_get_all_network_details(nds, &numJoined); for (int i=0; igetNode()->orbit(tptr, moonWorldId, moonSeed); } _service_lock.unlock(); return retval; } #ifdef SDK_JNI #endif zts_err_t zts_deorbit(uint64_t moonWorldId) { zts_err_t retval = ZTS_ERR_OK; void *tptr = NULL; _service_lock.lock(); if (!zt1Service || _freeHasBeenCalled || _serviceIsShuttingDown) { retval = ZTS_ERR_SERVICE; } else if (_zts_can_perform_service_operation()) { zt1Service->getNode()->deorbit(tptr, moonWorldId); } _service_lock.unlock(); return retval; } #ifdef SDK_JNI #endif int zts_core_running() { _service_lock.lock(); int retval = zt1Service == NULL ? false : zt1Service->isRunning(); _service_lock.unlock(); return retval; } #ifdef SDK_JNI JNIEXPORT jboolean JNICALL Java_com_zerotier_libzt_ZeroTier_core_1running( JNIEnv *env, jobject thisObj) { return zts_core_running(); } #endif int zts_ready() { _service_lock.lock(); bool stackRunning = VirtualTapManager::get_vtaps_size() > 0 ? true : false; _service_lock.unlock(); return zts_core_running() && stackRunning; } #ifdef SDK_JNI JNIEXPORT jboolean JNICALL Java_com_zerotier_libzt_ZeroTier_ready( JNIEnv *env, jobject thisObj) { return zts_ready(); } #endif zts_err_t zts_start(const char *path, int blocking = false) { _startup_lock.lock(); zts_err_t retval = ZTS_ERR_OK; if (zt1Service) { // Service is already initialized retval = ZTS_ERR_SERVICE; } if (_freeHasBeenCalled) { // Stack (presumably lwIP) has been dismantled, an application restart is required now retval = ZTS_ERR_INVALID_OP; } if (!path) { retval = ZTS_ERR_INVALID_ARG; } if (retval == ZTS_ERR_OK) { homeDir = path; #if defined(_WIN32) // initialize WinSock. Used in Phy for loopback pipe WSAStartup(MAKEWORD(2, 2), &wsaData); HANDLE thr = CreateThread(NULL, 0, _zts_start_service, NULL, 0, NULL); #else _startupError = false; retval = pthread_create(&service_thread, NULL, _zts_start_service, NULL); // Wait for confirmation that the ZT service has been initialized, // this wait condition is so brief and so rarely used that it should be // acceptable even in a non-blocking context. while(!zt1Service) { if (_serviceIsShuttingDown || _startupError) { // ZT service startup/binding might have failed for some reason retval = ZTS_ERR_SERVICE; break; } api_sleep(10); } #endif if (blocking && retval == ZTS_ERR_OK) { // block to prevent service calls before we're ready // waiting for zerotier service thread to start while (zts_core_running() == false || zt1Service->getNode() == NULL) { if (_serviceIsShuttingDown || _startupError) { // ZT service startup/binding might have failed for some reason retval = ZTS_ERR_SERVICE; break; } api_sleep(ZTS_WRAPPER_CHECK_INTERVAL); } if (retval == ZTS_ERR_OK) { // waiting for node address assignment while (zt1Service->getNode()->address() <= 0) { if (_serviceIsShuttingDown || _startupError) { retval = ZTS_ERR_SERVICE; break; } api_sleep(ZTS_WRAPPER_CHECK_INTERVAL); } } if (retval == ZTS_ERR_OK) { // Waiting for node to come online. Ensure the node is authorized to join the network while (true) { _service_lock.lock(); if (_serviceIsShuttingDown || _startupError) { retval = ZTS_ERR_SERVICE; break; } if (zt1Service && zt1Service->getNode() && zt1Service->getNode()->online()) { // Node is fully online break; } api_sleep(ZTS_WRAPPER_CHECK_INTERVAL); _service_lock.unlock(); } _service_lock.unlock(); } } } _startup_lock.unlock(); // if (blocking && retval == ZTS_ERR_OK) { DEBUG_INFO("node=%llx online", (unsigned long long)zts_get_node_id());} _hibernate_if_needed(); return retval; } #ifdef SDK_JNI JNIEXPORT void JNICALL Java_com_zerotier_libzt_ZeroTier_start( JNIEnv *env, jobject thisObj, jstring path, jboolean blocking) { if (path) { const char* utf_string = env->GetStringUTFChars(path, NULL); zts_start(utf_string, blocking); env->ReleaseStringUTFChars(path, utf_string); } } #endif zts_err_t zts_startjoin(const char *path, const uint64_t nwid) { zts_err_t retval = ZTS_ERR_OK; if ((retval = zts_start(path, true)) < 0) { return retval; } while (true) { try { zts_join(nwid); break; } catch( ... ) { api_sleep(ZTS_WRAPPER_CHECK_INTERVAL); retval = ZTS_ERR_SERVICE; } } _hibernate_if_needed(); return retval; } #ifdef SDK_JNI JNIEXPORT void JNICALL Java_com_zerotier_libzt_ZeroTier_startjoin( JNIEnv *env, jobject thisObj, jstring path, jlong nwid) { if (path && nwid) { const char* utf_string = env->GetStringUTFChars(path, NULL); zts_startjoin(utf_string, (uint64_t)nwid); env->ReleaseStringUTFChars(path, utf_string); } } #endif zts_err_t zts_stop(int blocking) { zts_err_t retval = ZTS_ERR_OK; _service_lock.lock(); bool didStop = false; if (_zts_can_perform_service_operation()) { didStop = true; zt1Service->terminate(); VirtualTapManager::clear(); } else { // Nothing to do retval = ZTS_ERR_SERVICE; } #if defined(_WIN32) WSACleanup(); #endif _service_lock.unlock(); if (blocking && retval == ZTS_ERR_OK && didStop) { // Block until ZT service thread successfully exits pthread_join(service_thread, NULL); } _hibernate_if_needed(); return retval; } #ifdef SDK_JNI JNIEXPORT void JNICALL Java_com_zerotier_libzt_ZeroTier_stop( JNIEnv *env, jobject thisObj) { zts_stop(); } #endif zts_err_t zts_free() { zts_err_t retval = 0; _service_lock.lock(); if (_freeHasBeenCalled) { retval = ZTS_ERR_INVALID_OP; _service_lock.unlock(); } else { _freeHasBeenCalled = true; _service_lock.unlock(); retval = zts_stop(); } // PENDING: add stack shutdown logic return retval; } #ifdef SDK_JNI JNIEXPORT void JNICALL Java_com_zerotier_libzt_ZeroTier_free( JNIEnv *env, jobject thisObj) { zts_free(); } #endif uint64_t zts_get_node_id() { uint64_t nodeId = 0; _service_lock.lock(); if (_zts_can_perform_service_operation()) { nodeId = zt1Service->getNode()->address(); } _service_lock.unlock(); return nodeId; } #ifdef SDK_JNI JNIEXPORT jlong JNICALL Java_com_zerotier_libzt_ZeroTier_get_1node_1id( JNIEnv *env, jobject thisObj) { return zts_get_node_id(); } #endif int zts_get_peer_count() { unsigned int peerCount = 0; _service_lock.lock(); if (_zts_can_perform_service_operation()) { peerCount = zt1Service->getNode()->peers()->peerCount; } else { peerCount = ZTS_ERR_SERVICE; } _service_lock.unlock(); return peerCount; } #ifdef SDK_JNI JNIEXPORT jlong JNICALL Java_com_zerotier_libzt_ZeroTier_get_1peer_1count( JNIEnv *env, jobject thisObj) { return zts_get_peer_count(); } #endif int zts_get_peers(struct zts_peer_details *pds, int *num) { zts_err_t retval = ZTS_ERR_OK; if (!pds || !num) { retval = ZTS_ERR_INVALID_ARG; } if (retval == ZTS_ERR_OK) { _service_lock.lock(); if (_zts_can_perform_service_operation()) { ZT_PeerList *pl = zt1Service->getNode()->peers(); if (pl) { *num = pl->peerCount; for(unsigned long i=0;ipeerCount;++i) { memcpy(&(pds[i]), &(pl->peers[i]), sizeof(struct zts_peer_details)); } } } else { retval = ZTS_ERR_SERVICE; } _service_lock.unlock(); } return retval; } #ifdef SDK_JNI #endif zts_err_t zts_get_num_joined_networks() { zts_err_t retval = ZTS_ERR_OK; if (!zt1Service || _freeHasBeenCalled || _serviceIsShuttingDown) { retval = ZTS_ERR_SERVICE; } else { retval = VirtualTapManager::get_vtaps_size(); } return retval; } #ifdef SDK_JNI JNIEXPORT jint JNICALL Java_com_zerotier_libzt_ZeroTier_get_1num_1joined_1networks( JNIEnv *env, jobject thisObj) { return zts_get_num_joined_networks(); } #endif zts_err_t zts_get_network_details(uint64_t nwid, struct zts_network_details *nd) { zts_err_t retval = ZTS_ERR_OK; if (!nd || nwid == 0) { retval = ZTS_ERR_INVALID_ARG; } if (!zt1Service || _freeHasBeenCalled || _serviceIsShuttingDown) { retval = ZTS_ERR_SERVICE; } if (retval == ZTS_ERR_OK) { VirtualTapManager::get_network_details(nwid, nd); } return retval; } #ifdef SDK_JNI #endif zts_err_t zts_get_all_network_details(struct zts_network_details *nds, int *num) { zts_err_t retval = ZTS_ERR_OK; if (!nds || !num) { retval = ZTS_ERR_INVALID_ARG; } if (!zt1Service || _freeHasBeenCalled || _serviceIsShuttingDown) { retval = ZTS_ERR_SERVICE; } if (retval == ZTS_ERR_OK) { VirtualTapManager::get_all_network_details(nds, num); } return retval; } #ifdef SDK_JNI #endif zts_err_t zts_enable_http_backplane_mgmt() { zts_err_t retval = ZTS_ERR_OK; _service_lock.lock(); if (!zt1Service || _freeHasBeenCalled || _serviceIsShuttingDown) { retval = ZTS_ERR_SERVICE; } else { zt1Service->allowHttpBackplaneManagement = true; } _service_lock.unlock(); return retval; } #ifdef SDK_JNI #endif zts_err_t zts_disable_http_backplane_mgmt() { zts_err_t retval = ZTS_ERR_OK; _service_lock.lock(); if (!zt1Service || _freeHasBeenCalled || _serviceIsShuttingDown) { retval = ZTS_ERR_SERVICE; } else { zt1Service->allowHttpBackplaneManagement = false; } _service_lock.unlock(); return retval; } #ifdef SDK_JNI #endif #ifdef __cplusplus } #endif