Improved reliability and performance, better startup and shutdown semantics, HTTP control plane is now disabled by default
This commit is contained in:
841
src/ServiceControls.cpp
Normal file
841
src/ServiceControls.cpp
Normal file
@@ -0,0 +1,841 @@
|
||||
/*
|
||||
* 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 <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* --
|
||||
*
|
||||
* 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<void*> 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 <jni.h>
|
||||
#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 <Windows.h>
|
||||
#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<std::string> 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<std::string>::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<tap->_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; i<numJoined; i++) {
|
||||
zts_leave(nds[i].nwid);
|
||||
}
|
||||
}
|
||||
return retval;
|
||||
}
|
||||
#ifdef SDK_JNI
|
||||
#endif
|
||||
|
||||
zts_err_t zts_orbit(uint64_t moonWorldId, uint64_t moonSeed)
|
||||
{
|
||||
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()->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;i<pl->peerCount;++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
|
||||
Reference in New Issue
Block a user