This repository has been archived on 2025-09-14. You can view files and clone it, but cannot push or open issues or pull requests.
Files
zhangyang-libzt/src/ZeroTierSDK.cpp

785 lines
26 KiB
C++
Raw Normal View History

/*
2017-04-14 17:23:28 -07:00
* ZeroTier SDK - Network Virtualization Everywhere
* Copyright (C) 2011-2016 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/>.
*/
#include <dlfcn.h>
#include <sys/socket.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <stdarg.h>
#include <sys/stat.h>
#include <unistd.h>
#include <errno.h>
#include <fcntl.h>
#include <sys/types.h>
#include <pthread.h>
#include "OneService.hpp"
#include "Utils.hpp"
#include "OSUtils.hpp"
#include "InetAddress.hpp"
#include "ZeroTierOne.h"
#include "SocketTap.hpp"
#include "ZeroTierSDK.h"
#ifdef __cplusplus
extern "C" {
#endif
static ZeroTier::OneService *zt1Service;
2017-04-14 17:23:28 -07:00
namespace ZeroTier {
std::string homeDir; // The resultant platform-specific dir we *must* use internally
std::string netDir; // Where network .conf files are to be written
/*
* Global reference to stack
*/
2017-04-14 17:23:28 -07:00
picoTCP *picostack = NULL;
/*
* "sockets" that have been created but not bound to a SocketTap interface yet
*/
2017-04-14 17:23:28 -07:00
std::map<int, Connection*> UnassignedConnections;
// FIXME: make sure these are properly deleted
/*
* For fast lookup of Connections and SocketTaps via given file descriptor
*/
std::map<int, std::pair<Connection*,SocketTap*>*> AssignedConnections;
ZeroTier::Mutex _multiplexer_lock;
ZeroTier::Mutex _accepted_connection_lock;
2017-04-14 17:23:28 -07:00
}
/****************************************************************************/
2017-04-07 17:56:05 -07:00
/* SDK Socket API - Language Bindings are written in terms of these */
/****************************************************************************/
void zts_start(const char *path)
{
2017-04-07 17:56:05 -07:00
if(zt1Service)
return;
if(ZeroTier::picostack)
return;
ZeroTier::picostack = new ZeroTier::picoTCP();
pico_stack_init();
DEBUG_INFO("path=%s", path);
if(path)
2017-04-14 17:23:28 -07:00
ZeroTier::homeDir = path;
2017-04-07 17:56:05 -07:00
pthread_t service_thread;
pthread_create(&service_thread, NULL, _start_service, (void *)(path));
}
void zts_stop() {
2017-04-07 17:56:05 -07:00
if(zt1Service) {
zt1Service->terminate();
zt1Service->removeNets();
}
}
void zts_join_network(const char * nwid) {
2017-04-07 17:56:05 -07:00
if(zt1Service) {
std::string confFile = zt1Service->givenHomePath() + "/networks.d/" + nwid + ".conf";
2017-04-14 17:23:28 -07:00
if(!ZeroTier::OSUtils::mkdir(ZeroTier::netDir))
DEBUG_ERROR("unable to create: %s", ZeroTier::netDir.c_str());
2017-04-07 17:56:05 -07:00
if(!ZeroTier::OSUtils::writeFile(confFile.c_str(), ""))
DEBUG_ERROR("unable to write network conf file: %s", confFile.c_str());
zt1Service->join(nwid);
}
}
2017-04-07 17:56:05 -07:00
void zts_join_network_soft(const char * filepath, const char * nwid) {
std::string net_dir = std::string(filepath) + "/networks.d/";
std::string confFile = net_dir + std::string(nwid) + ".conf";
if(!ZeroTier::OSUtils::mkdir(net_dir)) {
DEBUG_ERROR("unable to create: %s", net_dir.c_str());
}
if(!ZeroTier::OSUtils::fileExists(confFile.c_str(),false)) {
if(!ZeroTier::OSUtils::writeFile(confFile.c_str(), "")) {
DEBUG_ERROR("unable to write network conf file: %s", confFile.c_str());
}
}
}
2017-04-07 17:56:05 -07:00
void zts_leave_network(const char * nwid) {
if(zt1Service)
zt1Service->leave(nwid);
}
void zts_leave_network_soft(const char * filepath, const char * nwid) {
std::string net_dir = std::string(filepath) + "/networks.d/";
ZeroTier::OSUtils::rm((net_dir + nwid + ".conf").c_str());
}
2017-04-07 17:56:05 -07:00
void zts_get_homepath(char *homePath, int len) {
2017-04-14 17:23:28 -07:00
if(ZeroTier::homeDir.length()) {
memset(homePath, 0, len);
memcpy(homePath, ZeroTier::homeDir.c_str(), len < ZeroTier::homeDir.length() ? len : ZeroTier::homeDir.length());
}
}
2017-04-07 17:56:05 -07:00
void zts_core_version(char *ver) {
int major, minor, revision;
ZT_version(&major, &minor, &revision);
sprintf(ver, "%d.%d.%d", major, minor, revision);
}
void zts_sdk_version(char *ver) {
sprintf(ver, "%d.%d.%d", ZT_SDK_VERSION_MAJOR, ZT_SDK_VERSION_MINOR, ZT_SDK_VERSION_REVISION);
}
int zts_get_device_id(char *devID) {
if(zt1Service) {
char id[ZT_ID_LEN+1];
sprintf(id, "%lx",zt1Service->getNode()->address());
memcpy(devID, id, ZT_ID_LEN+1);
return 0;
}
else // Service isn't online, try to read ID from file
{
std::string fname("identity.public");
2017-04-14 17:23:28 -07:00
std::string fpath(ZeroTier::homeDir);
2017-04-07 17:56:05 -07:00
if(ZeroTier::OSUtils::fileExists((fpath + ZT_PATH_SEPARATOR_S + fname).c_str(),false)) {
std::string oldid;
ZeroTier::OSUtils::readFile((fpath + ZT_PATH_SEPARATOR_S + fname).c_str(),oldid);
memcpy(devID, oldid.c_str(), ZT_ID_LEN); // first 10 bytes of file
return 0;
}
}
return -1;
}
2017-04-07 17:56:05 -07:00
int zts_service_running() {
return !zt1Service ? false : zt1Service->isRunning();
}
2017-04-07 17:56:05 -07:00
int zts_has_ipv4_address(const char *nwid)
{
char ipv4_addr[ZT_MAX_IPADDR_LEN];
memset(ipv4_addr, 0, ZT_MAX_IPADDR_LEN);
zts_get_ipv4_address(nwid, ipv4_addr, ZT_MAX_IPADDR_LEN);
return strcmp(ipv4_addr, "\0");
}
2017-04-07 17:56:05 -07:00
int zts_has_ipv6_address(const char *nwid)
{
char ipv6_addr[ZT_MAX_IPADDR_LEN];
memset(ipv6_addr, 0, ZT_MAX_IPADDR_LEN);
zts_get_ipv6_address(nwid, ipv6_addr, ZT_MAX_IPADDR_LEN);
return strcmp(ipv6_addr, "\0");
}
int zts_has_address(const char *nwid)
{
2017-04-07 17:56:05 -07:00
return zts_has_ipv4_address(nwid) || zts_has_ipv6_address(nwid);
}
2017-04-07 17:56:05 -07:00
void zts_get_ipv4_address(const char *nwid, char *addrstr, const int addrlen)
{
2017-04-07 17:56:05 -07:00
if(zt1Service) {
uint64_t nwid_int = strtoull(nwid, NULL, 16);
ZeroTier::SocketTap *tap = zt1Service->getTap(nwid_int);
if(tap && tap->_ips.size()){
for(int i=0; i<tap->_ips.size(); i++) {
if(tap->_ips[i].isV4()) {
std::string addr = tap->_ips[i].toString();
int len = addrlen < addr.length() ? addrlen : addr.length();
memset(addrstr, 0, len);
memcpy(addrstr, addr.c_str(), len);
return;
}
}
}
}
2017-04-07 17:56:05 -07:00
else
memcpy(addrstr, "\0", 1);
}
2017-04-07 17:56:05 -07:00
void zts_get_ipv6_address(const char *nwid, char *addrstr, const int addrlen)
{
2017-04-07 17:56:05 -07:00
if(zt1Service) {
uint64_t nwid_int = strtoull(nwid, NULL, 16);
ZeroTier::SocketTap *tap = zt1Service->getTap(nwid_int);
if(tap && tap->_ips.size()){
for(int i=0; i<tap->_ips.size(); i++) {
if(tap->_ips[i].isV6()) {
std::string addr = tap->_ips[i].toString();
int len = addrlen < addr.length() ? addrlen : addr.length();
memset(addrstr, 0, len);
memcpy(addrstr, addr.c_str(), len);
return;
}
}
}
}
else
2017-04-07 17:56:05 -07:00
memcpy(addrstr, "\0", 1);
}
2017-04-07 17:56:05 -07:00
void zts_get_6plane_addr(char *addr, const char *nwid, const char *devID)
{
ZeroTier::InetAddress _6planeAddr = ZeroTier::InetAddress::makeIpv66plane(
ZeroTier::Utils::hexStrToU64(nwid),ZeroTier::Utils::hexStrToU64(devID));
memcpy(addr, _6planeAddr.toIpString().c_str(), 40);
}
void zts_get_rfc4193_addr(char *addr, const char *nwid, const char *devID)
{
ZeroTier::InetAddress _6planeAddr = ZeroTier::InetAddress::makeIpv6rfc4193(
ZeroTier::Utils::hexStrToU64(nwid),ZeroTier::Utils::hexStrToU64(devID));
memcpy(addr, _6planeAddr.toIpString().c_str(), 40);
}
unsigned long zts_get_peer_count() {
if(zt1Service)
return zt1Service->getNode()->peers()->peerCount;
else
return 0;
}
2017-04-07 17:56:05 -07:00
int zts_get_peer_address(char *peer, const char *devID) {
if(zt1Service) {
ZT_PeerList *pl = zt1Service->getNode()->peers();
// uint64_t addr;
for(int i=0; i<pl->peerCount; i++) {
// ZT_Peer *p = &(pl->peers[i]);
// DEBUG_INFO("peer[%d] = %lx", i, p->address);
}
return pl->peerCount;
}
else
return -1;
}
2017-04-07 17:56:05 -07:00
void zts_enable_http_control_plane()
{
// TODO
}
2017-04-07 17:56:05 -07:00
void zts_disable_http_control_plane()
{
// TODO
}
2017-04-07 17:56:05 -07:00
/****************************************************************************/
2017-04-14 17:23:28 -07:00
/* SocketTap Multiplexer Functionality */
2017-04-07 17:56:05 -07:00
/* - This section of the API is used to implement the general socket */
/* controls. Basically this is designed to handle socket provisioning */
/* requests when no SocketTap is yet initialized, and as a way to */
/* determine which SocketTap is to be used for a particular connect() or */
2017-04-14 17:23:28 -07:00
/* bind() call. This enables multi-network support */
2017-04-07 17:56:05 -07:00
/****************************************************************************/
2017-04-14 17:23:28 -07:00
int zts_socket(ZT_SOCKET_SIG) {
DEBUG_ERROR("UnassConn=%d, AssigFDs=%d", ZeroTier::UnassignedConnections.size(), ZeroTier::AssignedConnections.size());
2017-04-07 17:56:05 -07:00
DEBUG_INFO();
2017-04-14 17:23:28 -07:00
ZeroTier::_multiplexer_lock.lock();
2017-04-07 17:56:05 -07:00
ZeroTier::Connection *conn = new ZeroTier::Connection();
2017-04-14 17:23:28 -07:00
int err, protocol_version;
2017-04-07 17:56:05 -07:00
// set up pico_socket
struct pico_socket * psock;
#if defined(SDK_IPV4)
protocol_version = PICO_PROTO_IPV4;
#elif defined(SDK_IPV6)
protocol_version = PICO_PROTO_IPV6;
#endif
if(socket_type == SOCK_DGRAM) {
2017-04-14 17:23:28 -07:00
psock = pico_socket_open(
protocol_version, PICO_PROTO_UDP, &ZeroTier::picoTCP::pico_cb_socket_activity);
2017-04-07 17:56:05 -07:00
}
if(socket_type == SOCK_STREAM) {
2017-04-14 17:23:28 -07:00
psock = pico_socket_open(
protocol_version, PICO_PROTO_TCP, &ZeroTier::picoTCP::pico_cb_socket_activity);
}
// set up Unix Domain socketpair (used for data later on)
if(psock) {
conn->socket_family = socket_family;
conn->socket_type = socket_type;
conn->picosock = psock;
memset(conn->rxbuf, 0, ZT_UDP_RX_BUF_SZ);
ZeroTier::UnassignedConnections[conn->app_fd] = conn;
err = conn->app_fd; // return one end of the socketpair
}
else {
DEBUG_ERROR("failed to create pico_socket");
err = -1;
}
ZeroTier::_multiplexer_lock.unlock();
return err;
}
int zts_connect(ZT_CONNECT_SIG) {
if(!zt1Service) {
DEBUG_ERROR("Service not started. Call zts_start(path) first");
2017-04-14 17:23:28 -07:00
// errno = ?
return -1;
}
ZeroTier::_multiplexer_lock.lock();
int err;
ZeroTier::Connection *conn = ZeroTier::UnassignedConnections[fd];
ZeroTier::SocketTap *tap;
if(conn) {
char ipstr[INET6_ADDRSTRLEN];//, nm_str[INET6_ADDRSTRLEN];
memset(ipstr, 0, INET6_ADDRSTRLEN);
ZeroTier::InetAddress iaddr;
if(conn->socket_family == AF_INET) {
// FIXME: Fix this typecast mess
inet_ntop(AF_INET,
(const void *)&((struct sockaddr_in *)addr)->sin_addr.s_addr, ipstr, INET_ADDRSTRLEN);
}
if(conn->socket_family == AF_INET6) {
// FIXME: Fix this typecast mess
inet_ntop(AF_INET6,
(const void *)&((struct sockaddr_in6 *)addr)->sin6_addr.s6_addr, ipstr, INET6_ADDRSTRLEN);
}
iaddr.fromString(ipstr);
DEBUG_INFO("ipstr= %s", ipstr);
DEBUG_INFO("iaddr= %s", iaddr.toString().c_str());
tap = zt1Service->getTap(iaddr);
if(!tap) {
// TODO: More canonical error?
DEBUG_ERROR("no route to host");
2017-04-07 17:56:05 -07:00
// errno = ?
err = -1;
}
else {
// pointer to tap we use in callbacks from the stack
conn->picosock->priv = new ZeroTier::ConnectionPair(tap, conn);
2017-04-14 17:23:28 -07:00
DEBUG_INFO("found appropriate SocketTap");
// TODO: Perhaps move this connect call outside of the lock
// Semantically: tap->stack->connect
err = tap->Connect(conn, fd, addr, addrlen);
if(err == 0) {
tap->_Connections.push_back(conn); // Give this Connection to the tap we decided on
conn->tap = tap;
}
// Wrap the socketpair we created earlier
// For I/O loop participation and referencing the PhySocket's parent Connection in callbacks
conn->sock = tap->_phy.wrapSocket(conn->sdk_fd, conn);
2017-04-07 17:56:05 -07:00
}
}
else {
2017-04-14 17:23:28 -07:00
DEBUG_ERROR("unable to locate connection");
// errno = ?
2017-04-07 17:56:05 -07:00
err = -1;
}
ZeroTier::UnassignedConnections.erase(fd);
ZeroTier::AssignedConnections[fd] = new std::pair<ZeroTier::Connection*,ZeroTier::SocketTap*>(conn, tap);
2017-04-14 17:23:28 -07:00
ZeroTier::_multiplexer_lock.unlock();
// NOTE: pico_socket_connect() will return 0 if no error happens immediately, but that doesn't indicate
// the connection was completed, for that we must wait for a callback from the stack. During that
// callback we will place the Connection in a ZT_SOCK_STATE_UNHANDLED_CONNECTED state to signal
// to the multiplexer logic that this connection is complete and a success value can be sent to the
// user application
// non-blocking
if(err == 0 && false)
{
errno = EINPROGRESS;
return -1;
}
// blocking
if(err == 0 && true) {
while(true)
{
usleep(ZT_CONNECT_RECHECK_DELAY * 1000);
tap->_tcpconns_m.lock();
for(int i=0; i<tap->_Connections.size(); i++)
{
if(tap->_Connections[i]->state == PICO_ERR_ECONNRESET) {
errno = ECONNRESET;
return -1;
}
if(tap->_Connections[i]->state == ZT_SOCK_STATE_UNHANDLED_CONNECTED) {
tap->_Connections[i]->state = ZT_SOCK_STATE_CONNECTED;
errno = 0;
return 0; // complete
}
}
tap->_tcpconns_m.unlock();
}
}
2017-04-07 17:56:05 -07:00
return err;
}
2017-04-14 17:23:28 -07:00
int zts_bind(ZT_BIND_SIG) {
2017-04-07 17:56:05 -07:00
if(!zt1Service) {
DEBUG_ERROR("Service not started. Call zts_start(path) first");
2017-04-07 17:56:05 -07:00
// errno = ?
return -1;
}
2017-04-14 17:23:28 -07:00
ZeroTier::_multiplexer_lock.lock();
2017-04-07 17:56:05 -07:00
int err;
ZeroTier::Connection *conn = ZeroTier::UnassignedConnections[fd];
2017-04-14 17:23:28 -07:00
ZeroTier::SocketTap *tap;
if(conn) {
2017-04-07 17:56:05 -07:00
char ipstr[INET6_ADDRSTRLEN];//, nm_str[INET6_ADDRSTRLEN];
memset(ipstr, 0, INET6_ADDRSTRLEN);
ZeroTier::InetAddress iaddr;
if(conn->socket_family == AF_INET) {
// FIXME: Fix this typecast mess
2017-04-14 17:23:28 -07:00
inet_ntop(AF_INET,
(const void *)&((struct sockaddr_in *)addr)->sin_addr.s_addr, ipstr, INET_ADDRSTRLEN);
2017-04-07 17:56:05 -07:00
}
if(conn->socket_family == AF_INET6) {
// FIXME: Fix this typecast mess
2017-04-14 17:23:28 -07:00
inet_ntop(AF_INET6,
(const void *)&((struct sockaddr_in6 *)addr)->sin6_addr.s6_addr, ipstr, INET6_ADDRSTRLEN);
2017-04-07 17:56:05 -07:00
}
iaddr.fromString(ipstr);
DEBUG_INFO("ipstr= %s", ipstr);
DEBUG_INFO("iaddr= %s", iaddr.toString().c_str());
2017-04-14 17:23:28 -07:00
tap = zt1Service->getTap(iaddr);
2017-04-07 17:56:05 -07:00
if(!tap) {
2017-04-14 17:23:28 -07:00
// TODO: More canonical error?
DEBUG_ERROR("no appropriate interface to bind to");
2017-04-07 17:56:05 -07:00
// errno = ?
err = -1;
}
else {
DEBUG_INFO("found appropriate SocketTap");
2017-04-14 17:23:28 -07:00
DEBUG_INFO("conn->picosock = %p", conn->picosock);
conn->picosock->priv = new ZeroTier::ConnectionPair(tap, conn);
2017-04-14 17:23:28 -07:00
// TODO: Perhaps move this connect call outside of the lock
tap->_Connections.push_back(conn); // Give this Connection to the tap we decided on
err = tap->Bind(conn, fd, addr, addrlen); // Semantically: tap->stack->connect
conn->tap = tap;
2017-04-07 17:56:05 -07:00
}
}
else {
DEBUG_ERROR("unable to locate connection");
// errno = ?
err = -1;
}
ZeroTier::AssignedConnections[fd] = new std::pair<ZeroTier::Connection*,ZeroTier::SocketTap*>(conn, tap);
2017-04-14 17:23:28 -07:00
ZeroTier::_multiplexer_lock.unlock();
2017-04-07 17:56:05 -07:00
return err;
}
2017-04-14 17:23:28 -07:00
int zts_listen(ZT_LISTEN_SIG) {
if(!zt1Service) {
DEBUG_ERROR("Service not started. Call zts_start(path) first");
2017-04-14 17:23:28 -07:00
// errno = ?
return -1;
}
2017-04-07 17:56:05 -07:00
int err;
2017-04-14 17:23:28 -07:00
ZeroTier::_multiplexer_lock.lock();
std::pair<ZeroTier::Connection*, ZeroTier::SocketTap*> *p = ZeroTier::AssignedConnections[fd];
if(!p) {
DEBUG_ERROR("unable to locate connection pair (did you zbind()?");
return -1;
}
ZeroTier::Connection *conn = p->first;
ZeroTier::SocketTap *tap = p->second;
2017-04-14 17:23:28 -07:00
if(!tap || !conn) {
DEBUG_ERROR("unable to locate tap interface for file descriptor");
return -1;
2017-04-14 17:23:28 -07:00
}
tap->Listen(conn, fd, backlog);
err = 0;
DEBUG_INFO("put conn=%p into LISTENING state", conn);
2017-04-14 17:23:28 -07:00
ZeroTier::_multiplexer_lock.unlock();
return err;
}
2017-04-07 17:56:05 -07:00
2017-04-14 17:23:28 -07:00
int zts_accept(ZT_ACCEPT_SIG) {
int err;
ZeroTier::Connection *conn = ZeroTier::AssignedConnections[fd]->first;
ZeroTier::SocketTap *tap = ZeroTier::AssignedConnections[fd]->second;
2017-04-14 17:23:28 -07:00
// BLOCKING: loop and keep checking until we find a newly accepted connection
if(true) {
while(true) {
usleep(ZT_ACCEPT_RECHECK_DELAY * 1000);
err = tap->Accept(conn);
if(err >= 0)
return err;
}
}
// NON-BLOCKING: only check for a new connection once
else
err = tap->Accept(conn);
2017-04-07 17:56:05 -07:00
return err;
}
2017-04-14 17:23:28 -07:00
#if defined(__linux__)
int zts_accept4(ZT_ACCEPT4_SIG)
{
// TODO
2017-04-14 17:23:28 -07:00
return 0;
}
#endif
int zts_setsockopt(ZT_SETSOCKOPT_SIG)
{
// TODO
2017-04-14 17:23:28 -07:00
return 0;
}
int zts_getsockopt(ZT_GETSOCKOPT_SIG)
{
// TODO
2017-04-14 17:23:28 -07:00
return 0;
}
int zts_getsockname(ZT_GETSOCKNAME_SIG)
{
// TODO
2017-04-14 17:23:28 -07:00
return 0;
}
int zts_getpeername(ZT_GETPEERNAME_SIG)
{
// TODO
2017-04-14 17:23:28 -07:00
return 0;
}
int zts_close(ZT_CLOSE_SIG)
{
ZeroTier::_multiplexer_lock.lock();
ZeroTier::Connection *conn = ZeroTier::AssignedConnections[fd]->first;
ZeroTier::SocketTap *tap = ZeroTier::AssignedConnections[fd]->second;
// Tell the tap to stop monitoring this PhySocket
tap->Close(conn);
// Tell the userspace stack to close this pico_socket
ZeroTier::picostack->pico_Close(conn);
ZeroTier::_multiplexer_lock.unlock();
2017-04-14 17:23:28 -07:00
return 0;
}
int zts_fcntl(ZT_FCNTL_SIG)
{
// TODO
2017-04-14 17:23:28 -07:00
return 0;
}
ssize_t zts_sendto(ZT_SENDTO_SIG)
{
// TODO
2017-04-14 17:23:28 -07:00
return 0;
}
ssize_t zts_sendmsg(ZT_SENDMSG_SIG)
{
// TODO
2017-04-14 17:23:28 -07:00
return 0;
}
ssize_t zts_recvfrom(ZT_RECVFROM_SIG)
{
// TODO
2017-04-14 17:23:28 -07:00
return 0;
}
ssize_t zts_recvmsg(ZT_RECVMSG_SIG)
{
// TODO
2017-04-14 17:23:28 -07:00
return 0;
}
int zts_read(ZT_READ_SIG) {
return read(fd, buf, len);
}
int zts_write(ZT_WRITE_SIG) {
return write(fd, buf, len);
}
2017-04-07 17:56:05 -07:00
/****************************************************************************/
/* SDK Socket API (Java Native Interface JNI) */
/* JNI naming convention: Java_PACKAGENAME_CLASSNAME_METHODNAME */
/****************************************************************************/
#if defined(__ANDROID__) || defined(__JNI_LIB__)
// Returns whether the ZeroTier service is running
2017-04-07 17:56:05 -07:00
JNIEXPORT jboolean JNICALL Java_zerotier_ZeroTier_zt_1service_1is_1running(
JNIEnv *env, jobject thisObj)
{
return zts_service_is_running();
}
// Returns path for ZT config/data files
2017-04-07 17:56:05 -07:00
JNIEXPORT jstring JNICALL Java_zerotier_ZeroTier_zt_1get_1homepath(
JNIEnv *env, jobject thisObj)
{
return (*env).NewStringUTF(zts_get_homepath());
}
// Join a network
2017-04-07 17:56:05 -07:00
JNIEXPORT void JNICALL Java_zerotier_ZeroTier_zt_1join_1network(
JNIEnv *env, jobject thisObj, jstring nwid)
{
const char *nwidstr;
if(nwid) {
nwidstr = env->GetStringUTFChars(nwid, NULL);
zts_join_network(nwidstr);
}
}
// Leave a network
2017-04-07 17:56:05 -07:00
JNIEXPORT void JNICALL Java_zerotier_ZeroTier_zt_1leave_1network(
JNIEnv *env, jobject thisObj, jstring nwid)
{
const char *nwidstr;
if(nwid) {
nwidstr = env->GetStringUTFChars(nwid, NULL);
zts_leave_network(nwidstr);
}
}
// FIXME: Re-implemented to make it play nicer with the C-linkage required for Xcode integrations
// Now only returns first assigned address per network. Shouldn't normally be a problem
2017-04-07 17:56:05 -07:00
JNIEXPORT jobject JNICALL Java_zerotier_ZeroTier_zt_1get_1ipv4_1address(
JNIEnv *env, jobject thisObj, jstring nwid)
{
const char *nwid_str = env->GetStringUTFChars(nwid, NULL);
char address_string[32];
memset(address_string, 0, 32);
zts_get_ipv4_address(nwid_str, address_string);
jclass clazz = (*env).FindClass("java/util/ArrayList");
jobject addresses = (*env).NewObject(clazz, (*env).GetMethodID(clazz, "<init>", "()V"));
jstring _str = (*env).NewStringUTF(address_string);
env->CallBooleanMethod(addresses, env->GetMethodID(clazz, "add", "(Ljava/lang/Object;)Z"), _str);
return addresses;
}
2017-04-07 17:56:05 -07:00
JNIEXPORT jobject JNICALL Java_zerotier_ZeroTier_zt_1get_1ipv6_1address(
JNIEnv *env, jobject thisObj, jstring nwid)
{
const char *nwid_str = env->GetStringUTFChars(nwid, NULL);
char address_string[32];
memset(address_string, 0, 32);
zts_get_ipv6_address(nwid_str, address_string);
jclass clazz = (*env).FindClass("java/util/ArrayList");
jobject addresses = (*env).NewObject(clazz, (*env).GetMethodID(clazz, "<init>", "()V"));
jstring _str = (*env).NewStringUTF(address_string);
env->CallBooleanMethod(addresses, env->GetMethodID(clazz, "add", "(Ljava/lang/Object;)Z"), _str);
return addresses;
}
// Returns the device is in integer form
2017-04-07 17:56:05 -07:00
JNIEXPORT jint Java_zerotier_ZeroTier_zt_1get_1device_1id()
{
return zts_get_device_id(NULL); // TODO
}
// Returns whether the path to an endpoint is currently relayed by a root server
2017-04-07 17:56:05 -07:00
JNIEXPORT jboolean JNICALL Java_zerotier_ZeroTier_zt_1is_1relayed()
{
return 0;
// TODO
// zts_is_relayed();
}
#endif
2017-04-07 17:56:05 -07:00
/****************************************************************************/
/* SDK Socket API Helper functions --- DONT CALL THESE DIRECTLY */
/****************************************************************************/
// Starts a ZeroTier service in the background
2017-04-07 17:56:05 -07:00
void *_start_service(void *thread_id) {
2017-04-14 17:23:28 -07:00
DEBUG_INFO("homeDir=%s", ZeroTier::homeDir.c_str());
// Where network .conf files will be stored
2017-04-14 17:23:28 -07:00
ZeroTier::netDir = ZeroTier::homeDir + "/networks.d";
zt1Service = (ZeroTier::OneService *)0;
// Construct path for network config and supporting service files
2017-04-14 17:23:28 -07:00
if (ZeroTier::homeDir.length()) {
std::vector<std::string> hpsp(ZeroTier::OSUtils::split(ZeroTier::homeDir.c_str(),
2017-04-07 17:56:05 -07:00
ZT_PATH_SEPARATOR_S,"",""));
std::string ptmp;
2017-04-14 17:23:28 -07:00
if (ZeroTier::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 (!ZeroTier::OSUtils::mkdir(ptmp)) {
DEBUG_ERROR("home path does not exist, and could not create");
perror("error\n");
}
}
}
}
else {
DEBUG_ERROR("homeDir is empty, could not construct path");
return NULL;
}
2017-04-07 17:56:05 -07:00
// rpc dir
2017-04-14 17:23:28 -07:00
if(!ZeroTier::OSUtils::mkdir(ZeroTier::homeDir + "/" + ZT_SDK_RPC_DIR_PREFIX)) {
2017-04-07 17:56:05 -07:00
DEBUG_ERROR("unable to create dir: " ZT_SDK_RPC_DIR_PREFIX);
return NULL;
}
// Generate random port for new service instance
unsigned int randp = 0;
ZeroTier::Utils::getSecureRandom(&randp,sizeof(randp));
int servicePort = 9000 + (randp % 1000);
2017-04-07 17:56:05 -07:00
DEBUG_ERROR("servicePort = %d", servicePort);
for(;;) {
2017-04-14 17:23:28 -07:00
zt1Service = ZeroTier::OneService::newInstance(ZeroTier::homeDir.c_str(),servicePort);
switch(zt1Service->run()) {
2017-04-07 17:56:05 -07:00
case ZeroTier::OneService::ONE_STILL_RUNNING:
case ZeroTier::OneService::ONE_NORMAL_TERMINATION:
break;
case ZeroTier::OneService::ONE_UNRECOVERABLE_ERROR:
DEBUG_ERROR("fatal error: %s",zt1Service->fatalErrorMessage().c_str());
break;
case ZeroTier::OneService::ONE_IDENTITY_COLLISION: {
delete zt1Service;
zt1Service = (ZeroTier::OneService *)0;
std::string oldid;
2017-04-14 17:23:28 -07:00
ZeroTier::OSUtils::readFile((ZeroTier::homeDir + ZT_PATH_SEPARATOR_S
2017-04-07 17:56:05 -07:00
+ "identity.secret").c_str(),oldid);
if (oldid.length()) {
2017-04-14 17:23:28 -07:00
ZeroTier::OSUtils::writeFile((ZeroTier::homeDir + ZT_PATH_SEPARATOR_S
2017-04-07 17:56:05 -07:00
+ "identity.secret.saved_after_collision").c_str(),oldid);
2017-04-14 17:23:28 -07:00
ZeroTier::OSUtils::rm((ZeroTier::homeDir + ZT_PATH_SEPARATOR_S
2017-04-07 17:56:05 -07:00
+ "identity.secret").c_str());
2017-04-14 17:23:28 -07:00
ZeroTier::OSUtils::rm((ZeroTier::homeDir + ZT_PATH_SEPARATOR_S
2017-04-07 17:56:05 -07:00
+ "identity.public").c_str());
}
}
continue; // restart!
}
break; // terminate loop -- normally we don't keep restarting
}
delete zt1Service;
zt1Service = (ZeroTier::OneService *)0;
return NULL;
}
#ifdef __cplusplus
}
#endif