com_zerotierone_sdk_Node.cpp 32 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138
  1. /*
  2. * ZeroTier One - Network Virtualization Everywhere
  3. * Copyright (C) 2011-2015 ZeroTier, Inc.
  4. *
  5. * This program is free software: you can redistribute it and/or modify
  6. * it under the terms of the GNU General Public License as published by
  7. * the Free Software Foundation, either version 3 of the License, or
  8. * (at your option) any later version.
  9. *
  10. * This program is distributed in the hope that it will be useful,
  11. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13. * GNU General Public License for more details.
  14. *
  15. * You should have received a copy of the GNU General Public License
  16. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  17. *
  18. * --
  19. *
  20. * ZeroTier may be used and distributed under the terms of the GPLv3, which
  21. * are available at: http://www.gnu.org/licenses/gpl-3.0.html
  22. *
  23. * If you would like to embed ZeroTier into a commercial application or
  24. * redistribute it in a modified binary form, please contact ZeroTier Networks
  25. * LLC. Start here: http://www.zerotier.com/
  26. */
  27. #include "com_zerotierone_sdk_Node.h"
  28. #include "ZT_jnicache.h"
  29. #include "ZT_jniutils.h"
  30. #include <ZeroTierOne.h>
  31. #include "Mutex.hpp"
  32. #include <map>
  33. #include <string>
  34. #include <cassert>
  35. #include <cstring>
  36. #include <cinttypes> // for PRId64
  37. #define LOG_TAG "Node"
  38. namespace {
  39. struct JniRef
  40. {
  41. JniRef(
  42. int64_t id,
  43. JavaVM *jvm,
  44. jobject dataStoreGetListenerLocalIn,
  45. jobject dataStorePutListenerLocalIn,
  46. jobject packetSenderLocalIn,
  47. jobject eventListenerLocalIn,
  48. jobject frameListenerLocalIn,
  49. jobject configListenerLocalIn,
  50. jobject pathCheckerLocalIn)
  51. : id(id)
  52. , jvm(jvm)
  53. , node()
  54. , dataStoreGetListener()
  55. , dataStorePutListener()
  56. , packetSender()
  57. , eventListener()
  58. , frameListener()
  59. , configListener()
  60. , pathChecker()
  61. , inited() {
  62. JNIEnv *env;
  63. GETENV(env, jvm);
  64. dataStoreGetListener = env->NewGlobalRef(dataStoreGetListenerLocalIn);
  65. dataStorePutListener = env->NewGlobalRef(dataStorePutListenerLocalIn);
  66. packetSender = env->NewGlobalRef(packetSenderLocalIn);
  67. eventListener = env->NewGlobalRef(eventListenerLocalIn);
  68. frameListener = env->NewGlobalRef(frameListenerLocalIn);
  69. configListener = env->NewGlobalRef(configListenerLocalIn);
  70. pathChecker = env->NewGlobalRef(pathCheckerLocalIn);
  71. };
  72. ~JniRef()
  73. {
  74. JNIEnv *env;
  75. GETENV(env, jvm);
  76. env->DeleteGlobalRef(dataStoreGetListener);
  77. env->DeleteGlobalRef(dataStorePutListener);
  78. env->DeleteGlobalRef(packetSender);
  79. env->DeleteGlobalRef(eventListener);
  80. env->DeleteGlobalRef(frameListener);
  81. env->DeleteGlobalRef(configListener);
  82. env->DeleteGlobalRef(pathChecker);
  83. }
  84. int64_t id;
  85. JavaVM *jvm;
  86. ZT_Node *node;
  87. jobject dataStoreGetListener;
  88. jobject dataStorePutListener;
  89. jobject packetSender;
  90. jobject eventListener;
  91. jobject frameListener;
  92. jobject configListener;
  93. jobject pathChecker;
  94. bool inited;
  95. bool finishInitializing();
  96. };
  97. int VirtualNetworkConfigFunctionCallback(
  98. ZT_Node *node,
  99. void *userData,
  100. void *threadData,
  101. uint64_t nwid,
  102. void **,
  103. enum ZT_VirtualNetworkConfigOperation operation,
  104. const ZT_VirtualNetworkConfig *config)
  105. {
  106. LOGV("VirtualNetworkConfigFunctionCallback");
  107. JniRef *ref = (JniRef*)userData;
  108. JNIEnv *env;
  109. GETENV(env, ref->jvm);
  110. if (ref->configListener == NULL) {
  111. LOGE("configListener is NULL");
  112. return -100;
  113. }
  114. jobject operationObject = createVirtualNetworkConfigOperation(env, operation);
  115. if(env->ExceptionCheck() || operationObject == NULL)
  116. {
  117. return -101;
  118. }
  119. if (config == NULL) {
  120. LOGE("Config is NULL");
  121. return -102;
  122. }
  123. jobject networkConfigObject = newNetworkConfig(env, *config);
  124. if(env->ExceptionCheck() || networkConfigObject == NULL)
  125. {
  126. return -103;
  127. }
  128. return env->CallIntMethod(
  129. ref->configListener,
  130. VirtualNetworkConfigListener_onNetworkConfigurationUpdated_method,
  131. (jlong)nwid, operationObject, networkConfigObject);
  132. }
  133. void VirtualNetworkFrameFunctionCallback(ZT_Node *node,
  134. void *userData,
  135. void *threadData,
  136. uint64_t nwid,
  137. void**,
  138. uint64_t sourceMac,
  139. uint64_t destMac,
  140. unsigned int etherType,
  141. unsigned int vlanid,
  142. const void *frameData,
  143. unsigned int frameLength)
  144. {
  145. LOGV("VirtualNetworkFrameFunctionCallback");
  146. #ifndef NDEBUG
  147. unsigned char* local = (unsigned char*)frameData;
  148. LOGV("Type Bytes: 0x%02x%02x", local[12], local[13]);
  149. #endif
  150. JniRef *ref = (JniRef*)userData;
  151. assert(ref->node == node);
  152. JNIEnv *env;
  153. GETENV(env, ref->jvm);
  154. if (ref->frameListener == NULL) {
  155. LOGE("frameListener is NULL");
  156. return;
  157. }
  158. const unsigned char *bytes = static_cast<const unsigned char*>(frameData);
  159. jbyteArray dataArray = newByteArray(env, bytes, frameLength);
  160. if(env->ExceptionCheck() || dataArray == NULL)
  161. {
  162. return;
  163. }
  164. env->CallVoidMethod(ref->frameListener, VirtualNetworkFrameListener_onVirtualNetworkFrame_method, (jlong)nwid, (jlong)sourceMac, (jlong)destMac, (jlong)etherType, (jlong)vlanid, dataArray);
  165. }
  166. void EventCallback(ZT_Node *node,
  167. void *userData,
  168. void *threadData,
  169. enum ZT_Event event,
  170. const void *data) {
  171. LOGV("EventCallback");
  172. JniRef *ref = (JniRef *) userData;
  173. if (ref->node != node && event != ZT_EVENT_UP) {
  174. LOGE("Nodes not equal. ref->node %p, node %p. Event: %d", ref->node, node, event);
  175. return;
  176. }
  177. JNIEnv *env;
  178. GETENV(env, ref->jvm);
  179. if (ref->eventListener == NULL) {
  180. LOGE("eventListener is NULL");
  181. return;
  182. }
  183. jobject eventObject = createEvent(env, event);
  184. if (env->ExceptionCheck() || eventObject == NULL) {
  185. return;
  186. }
  187. switch (event) {
  188. case ZT_EVENT_UP: {
  189. LOGD("Event Up");
  190. env->CallVoidMethod(ref->eventListener, EventListener_onEvent_method, eventObject);
  191. break;
  192. }
  193. case ZT_EVENT_OFFLINE: {
  194. LOGD("Event Offline");
  195. env->CallVoidMethod(ref->eventListener, EventListener_onEvent_method, eventObject);
  196. break;
  197. }
  198. case ZT_EVENT_ONLINE: {
  199. LOGD("Event Online");
  200. env->CallVoidMethod(ref->eventListener, EventListener_onEvent_method, eventObject);
  201. break;
  202. }
  203. case ZT_EVENT_DOWN: {
  204. LOGD("Event Down");
  205. env->CallVoidMethod(ref->eventListener, EventListener_onEvent_method, eventObject);
  206. break;
  207. }
  208. case ZT_EVENT_FATAL_ERROR_IDENTITY_COLLISION: {
  209. LOGV("Identity Collision");
  210. // call onEvent()
  211. env->CallVoidMethod(ref->eventListener, EventListener_onEvent_method, eventObject);
  212. }
  213. break;
  214. case ZT_EVENT_TRACE: {
  215. LOGV("Trace Event");
  216. // call onTrace()
  217. if (data == NULL) {
  218. break;
  219. }
  220. const char *message = (const char *) data;
  221. jstring messageStr = env->NewStringUTF(message);
  222. env->CallVoidMethod(ref->eventListener, EventListener_onTrace_method, messageStr);
  223. }
  224. break;
  225. case ZT_EVENT_USER_MESSAGE:
  226. case ZT_EVENT_REMOTE_TRACE:
  227. break;
  228. }
  229. }
  230. void StatePutFunction(
  231. ZT_Node *node,
  232. void *userData,
  233. void *threadData,
  234. enum ZT_StateObjectType type,
  235. const uint64_t id[2],
  236. const void *buffer,
  237. int bufferLength) {
  238. char p[4096] = {0};
  239. bool secure = false;
  240. switch (type) {
  241. case ZT_STATE_OBJECT_IDENTITY_PUBLIC:
  242. snprintf(p, sizeof(p), "identity.public");
  243. break;
  244. case ZT_STATE_OBJECT_IDENTITY_SECRET:
  245. snprintf(p, sizeof(p), "identity.secret");
  246. secure = true;
  247. break;
  248. case ZT_STATE_OBJECT_PLANET:
  249. snprintf(p, sizeof(p), "planet");
  250. break;
  251. case ZT_STATE_OBJECT_MOON:
  252. snprintf(p, sizeof(p), "moons.d/%.16" PRIx64 ".moon", id[0]);
  253. break;
  254. case ZT_STATE_OBJECT_NETWORK_CONFIG:
  255. snprintf(p, sizeof(p), "networks.d/%.16" PRIx64 ".conf", id[0]);
  256. break;
  257. case ZT_STATE_OBJECT_PEER:
  258. snprintf(p, sizeof(p), "peers.d/%.10" PRIx64, id[0]);
  259. break;
  260. case ZT_STATE_OBJECT_NULL:
  261. return;
  262. }
  263. if (strlen(p) < 1) {
  264. return;
  265. }
  266. JniRef *ref = (JniRef*)userData;
  267. JNIEnv *env;
  268. GETENV(env, ref->jvm);
  269. if (ref->dataStorePutListener == NULL) {
  270. LOGE("dataStorePutListener is NULL");
  271. return;
  272. }
  273. jstring nameStr = env->NewStringUTF(p);
  274. if (bufferLength >= 0) {
  275. LOGD("JNI: Write file: %s", p);
  276. const unsigned char *bytes = static_cast<const unsigned char *>(buffer);
  277. jbyteArray bufferObj = newByteArray(env, bytes, bufferLength);
  278. if(env->ExceptionCheck() || bufferObj == NULL)
  279. {
  280. return;
  281. }
  282. env->CallIntMethod(ref->dataStorePutListener,
  283. DataStorePutListener_onDataStorePut_method,
  284. nameStr, bufferObj, secure);
  285. } else {
  286. LOGD("JNI: Delete file: %s", p);
  287. env->CallIntMethod(ref->dataStorePutListener, DataStorePutListener_onDelete_method, nameStr);
  288. }
  289. }
  290. int StateGetFunction(
  291. ZT_Node *node,
  292. void *userData,
  293. void *threadData,
  294. ZT_StateObjectType type,
  295. const uint64_t id[2],
  296. void *buffer,
  297. unsigned int bufferLength) {
  298. char p[4096] = {0};
  299. switch (type) {
  300. case ZT_STATE_OBJECT_IDENTITY_PUBLIC:
  301. snprintf(p, sizeof(p), "identity.public");
  302. break;
  303. case ZT_STATE_OBJECT_IDENTITY_SECRET:
  304. snprintf(p, sizeof(p), "identity.secret");
  305. break;
  306. case ZT_STATE_OBJECT_PLANET:
  307. snprintf(p, sizeof(p), "planet");
  308. break;
  309. case ZT_STATE_OBJECT_MOON:
  310. snprintf(p, sizeof(p), "moons.d/%.16" PRIx64 ".moon", id[0]);
  311. break;
  312. case ZT_STATE_OBJECT_NETWORK_CONFIG:
  313. snprintf(p, sizeof(p), "networks.d/%.16" PRIx64 ".conf", id[0]);
  314. break;
  315. case ZT_STATE_OBJECT_PEER:
  316. snprintf(p, sizeof(p), "peers.d/%.10" PRIx64, id[0]);
  317. break;
  318. case ZT_STATE_OBJECT_NULL:
  319. return -100;
  320. }
  321. if (strlen(p) < 1) {
  322. return -101;
  323. }
  324. JniRef *ref = (JniRef*)userData;
  325. JNIEnv *env;
  326. GETENV(env, ref->jvm);
  327. if (ref->dataStoreGetListener == NULL) {
  328. LOGE("dataStoreGetListener is NULL");
  329. return -102;
  330. }
  331. jstring nameStr = env->NewStringUTF(p);
  332. if(nameStr == NULL)
  333. {
  334. LOGE("Error creating name string object");
  335. return -103; // out of memory
  336. }
  337. jbyteArray bufferObj = newByteArray(env, bufferLength);
  338. if(env->ExceptionCheck() || bufferObj == NULL)
  339. {
  340. return -104;
  341. }
  342. LOGV("Calling onDataStoreGet(%s, %p)", p, buffer);
  343. int retval = (int)env->CallLongMethod(
  344. ref->dataStoreGetListener,
  345. DataStoreGetListener_onDataStoreGet_method,
  346. nameStr,
  347. bufferObj);
  348. LOGV("onDataStoreGet returned %d", retval);
  349. if(retval > 0)
  350. {
  351. void *data = env->GetPrimitiveArrayCritical(bufferObj, NULL);
  352. memcpy(buffer, data, retval);
  353. env->ReleasePrimitiveArrayCritical(bufferObj, data, 0);
  354. }
  355. return retval;
  356. }
  357. int WirePacketSendFunction(ZT_Node *node,
  358. void *userData,
  359. void *threadData,
  360. int64_t localSocket,
  361. const struct sockaddr_storage *remoteAddress,
  362. const void *buffer,
  363. unsigned int bufferSize,
  364. unsigned int ttl)
  365. {
  366. LOGV("WirePacketSendFunction(%" PRId64 ", %p, %p, %d, %u)", localSocket, remoteAddress, buffer, bufferSize, ttl);
  367. JniRef *ref = (JniRef*)userData;
  368. assert(ref->node == node);
  369. JNIEnv *env;
  370. GETENV(env, ref->jvm);
  371. if (ref->packetSender == NULL) {
  372. LOGE("packetSender is NULL");
  373. return -100;
  374. }
  375. //
  376. // may be NULL
  377. //
  378. jobject remoteAddressObj = newInetSocketAddress(env, *remoteAddress);
  379. if (env->ExceptionCheck()) {
  380. return -101;
  381. }
  382. const unsigned char *bytes = static_cast<const unsigned char *>(buffer);
  383. jbyteArray bufferObj = newByteArray(env, bytes, bufferSize);
  384. if (env->ExceptionCheck() || bufferObj == NULL)
  385. {
  386. return -102;
  387. }
  388. int retval = env->CallIntMethod(ref->packetSender, PacketSender_onSendPacketRequested_method, localSocket, remoteAddressObj, bufferObj, 0);
  389. LOGV("JNI Packet Sender returned: %d", retval);
  390. return retval;
  391. }
  392. int PathCheckFunction(ZT_Node *node,
  393. void *userPtr,
  394. void *threadPtr,
  395. uint64_t address,
  396. int64_t localSocket,
  397. const struct sockaddr_storage *remoteAddress)
  398. {
  399. JniRef *ref = (JniRef*)userPtr;
  400. assert(ref->node == node);
  401. if(ref->pathChecker == NULL) {
  402. return true;
  403. }
  404. JNIEnv *env;
  405. GETENV(env, ref->jvm);
  406. //
  407. // may be NULL
  408. //
  409. jobject remoteAddressObj = newInetSocketAddress(env, *remoteAddress);
  410. if (env->ExceptionCheck()) {
  411. return true;
  412. }
  413. return env->CallBooleanMethod(ref->pathChecker, PathChecker_onPathCheck_method, address, localSocket, remoteAddressObj);
  414. }
  415. int PathLookupFunction(ZT_Node *node,
  416. void *userPtr,
  417. void *threadPtr,
  418. uint64_t address,
  419. int ss_family,
  420. struct sockaddr_storage *result)
  421. {
  422. JniRef *ref = (JniRef*)userPtr;
  423. assert(ref->node == node);
  424. if(ref->pathChecker == NULL) {
  425. return false;
  426. }
  427. JNIEnv *env;
  428. GETENV(env, ref->jvm);
  429. jobject sockAddressObject = env->CallObjectMethod(ref->pathChecker, PathChecker_onPathLookup_method, address, ss_family);
  430. if(sockAddressObject == NULL)
  431. {
  432. LOGE("Unable to call onPathLookup implementation");
  433. return false;
  434. }
  435. *result = fromSocketAddressObject(env, sockAddressObject);
  436. if (env->ExceptionCheck() || isSocketAddressEmpty(*result)) {
  437. return false;
  438. }
  439. return true;
  440. }
  441. typedef std::map<int64_t, JniRef*> NodeMap;
  442. NodeMap nodeMap;
  443. ZeroTier::Mutex nodeMapMutex;
  444. bool isInited(int64_t nodeId) {
  445. ZeroTier::Mutex::Lock lock(nodeMapMutex);
  446. NodeMap::iterator found = nodeMap.find(nodeId);
  447. if (found == nodeMap.end()) {
  448. //
  449. // not in map yet, or has been removed from map
  450. //
  451. return false;
  452. }
  453. JniRef *ref = found->second;
  454. assert(ref);
  455. return ref->inited;
  456. }
  457. bool JniRef::finishInitializing() {
  458. ZeroTier::Mutex::Lock lock(nodeMapMutex);
  459. NodeMap::iterator found = nodeMap.find(id);
  460. if (found != nodeMap.end()) {
  461. //
  462. // already in map
  463. //
  464. LOGE("Cannot finish initializing; node is already in map");
  465. return false;
  466. }
  467. nodeMap.insert(std::make_pair(id, this));
  468. assert(!inited);
  469. inited = true;
  470. return true;
  471. }
  472. ZT_Node* findNode(int64_t nodeId)
  473. {
  474. ZeroTier::Mutex::Lock lock(nodeMapMutex);
  475. NodeMap::iterator found = nodeMap.find(nodeId);
  476. assert(found != nodeMap.end());
  477. JniRef *ref = found->second;
  478. assert(ref);
  479. return ref->node;
  480. }
  481. JniRef *removeRef(int64_t nodeId) {
  482. ZeroTier::Mutex::Lock lock(nodeMapMutex);
  483. NodeMap::iterator found = nodeMap.find(nodeId);
  484. if (found == nodeMap.end()) {
  485. return nullptr;
  486. }
  487. JniRef *ref = found->second;
  488. assert(ref);
  489. nodeMap.erase(nodeId);
  490. return ref;
  491. }
  492. }
  493. #ifdef __cplusplus
  494. extern "C" {
  495. #endif
  496. JNIEXPORT jint JNICALL JNI_OnLoad(JavaVM *vm, void *reserved)
  497. {
  498. setupJNICache(vm);
  499. return JNI_VERSION_1_6;
  500. }
  501. JNIEXPORT void JNICALL JNI_OnUnload(JavaVM *vm, void *reserved)
  502. {
  503. teardownJNICache(vm);
  504. }
  505. /*
  506. * Class: com_zerotier_sdk_Node
  507. * Method: node_init
  508. * Signature: (JLcom/zerotier/sdk/DataStoreGetListener;Lcom/zerotier/sdk/DataStorePutListener;Lcom/zerotier/sdk/PacketSender;Lcom/zerotier/sdk/EventListener;Lcom/zerotier/sdk/VirtualNetworkFrameListener;Lcom/zerotier/sdk/VirtualNetworkConfigListener;Lcom/zerotier/sdk/PathChecker;)Lcom/zerotier/sdk/ResultCode;
  509. */
  510. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_node_1init(
  511. JNIEnv *env, jobject obj, jlong now, jobject dataStoreGetListener,
  512. jobject dataStorePutListener, jobject packetSender, jobject eventListener,
  513. jobject frameListener, jobject configListener,
  514. jobject pathChecker)
  515. {
  516. LOGV("Creating ZT_Node struct");
  517. jobject resultObject = ResultCode_RESULT_OK_enum;
  518. JavaVM *vm;
  519. GETJAVAVM(env, vm);
  520. assert(dataStoreGetListener != NULL);
  521. assert(dataStorePutListener != NULL);
  522. assert(packetSender != NULL);
  523. assert(frameListener != NULL);
  524. assert(configListener != NULL);
  525. assert(eventListener != NULL);
  526. //
  527. // OPTIONAL, pathChecker may be NULL
  528. //
  529. // assert(pathChecker != NULL);
  530. ZT_Node_Callbacks callbacks{};
  531. callbacks.stateGetFunction = &StateGetFunction;
  532. callbacks.statePutFunction = &StatePutFunction;
  533. callbacks.wirePacketSendFunction = &WirePacketSendFunction;
  534. callbacks.virtualNetworkFrameFunction = &VirtualNetworkFrameFunctionCallback;
  535. callbacks.virtualNetworkConfigFunction = &VirtualNetworkConfigFunctionCallback;
  536. callbacks.eventCallback = &EventCallback;
  537. callbacks.pathCheckFunction = &PathCheckFunction;
  538. callbacks.pathLookupFunction = &PathLookupFunction;
  539. //
  540. // a bit of a confusing dance here where ref and node both know about each other
  541. //
  542. JniRef *ref = new JniRef(
  543. now,
  544. vm,
  545. dataStoreGetListener,
  546. dataStorePutListener,
  547. packetSender,
  548. eventListener,
  549. frameListener,
  550. configListener,
  551. pathChecker);
  552. ZT_Node *node;
  553. ZT_ResultCode rc = ZT_Node_new(
  554. &node,
  555. ref,
  556. NULL,
  557. &callbacks,
  558. (int64_t)now);
  559. if(rc != ZT_RESULT_OK)
  560. {
  561. LOGE("Error creating Node: %d", rc);
  562. resultObject = createResultObject(env, rc);
  563. if (env->ExceptionCheck() || resultObject == NULL) {
  564. return NULL;
  565. }
  566. if(node)
  567. {
  568. ZT_Node_delete(node);
  569. node = NULL;
  570. }
  571. delete ref;
  572. ref = NULL;
  573. return resultObject;
  574. }
  575. //
  576. // node is now updated
  577. //
  578. ref->node = node;
  579. if (!ref->finishInitializing()) {
  580. LOGE("finishInitializing() failed");
  581. return ResultCode_RESULT_FATAL_ERROR_INTERNAL_enum;
  582. }
  583. return resultObject;
  584. }
  585. /*
  586. * Class: com_zerotier_sdk_Node
  587. * Method: node_isInited
  588. * Signature: (J)Z
  589. */
  590. JNIEXPORT jboolean JNICALL Java_com_zerotier_sdk_Node_node_1isInited
  591. (JNIEnv *env, jobject obj, jlong nodeId) {
  592. return isInited(nodeId);
  593. }
  594. /*
  595. * Class: com_zerotier_sdk_Node
  596. * Method: node_delete
  597. * Signature: (J)V
  598. */
  599. JNIEXPORT void JNICALL Java_com_zerotier_sdk_Node_node_1delete(
  600. JNIEnv *env, jobject obj, jlong id)
  601. {
  602. LOGV("Destroying ZT_Node struct");
  603. int64_t nodeId = (int64_t)id;
  604. JniRef *ref = removeRef(nodeId);
  605. if (!ref) {
  606. return;
  607. }
  608. ZT_Node_delete(ref->node);
  609. delete ref;
  610. }
  611. /*
  612. * Class: com_zerotier_sdk_Node
  613. * Method: processVirtualNetworkFrame
  614. * Signature: (JJJJJII[B[J)Lcom/zerotier/sdk/ResultCode;
  615. */
  616. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_processVirtualNetworkFrame(
  617. JNIEnv *env, jobject obj,
  618. jlong id,
  619. jlong in_now,
  620. jlong in_nwid,
  621. jlong in_sourceMac,
  622. jlong in_destMac,
  623. jint in_etherType,
  624. jint in_vlanId,
  625. jbyteArray in_frameData,
  626. jlongArray out_nextBackgroundTaskDeadline)
  627. {
  628. int64_t nodeId = (int64_t) id;
  629. ZT_Node *node = findNode(nodeId);
  630. unsigned int nbtd_len = env->GetArrayLength(out_nextBackgroundTaskDeadline);
  631. if(nbtd_len < 1)
  632. {
  633. // array for next background task length has 0 elements!
  634. return ResultCode_RESULT_FATAL_ERROR_INTERNAL_enum;
  635. }
  636. int64_t now = (int64_t)in_now;
  637. uint64_t nwid = (uint64_t)in_nwid;
  638. uint64_t sourceMac = (uint64_t)in_sourceMac;
  639. uint64_t destMac = (uint64_t)in_destMac;
  640. unsigned int etherType = (unsigned int)in_etherType;
  641. unsigned int vlanId = (unsigned int)in_vlanId;
  642. unsigned int frameLength = env->GetArrayLength(in_frameData);
  643. void *frameData = env->GetPrimitiveArrayCritical(in_frameData, NULL);
  644. void *localData = malloc(frameLength);
  645. memcpy(localData, frameData, frameLength);
  646. env->ReleasePrimitiveArrayCritical(in_frameData, frameData, 0);
  647. int64_t nextBackgroundTaskDeadline = 0;
  648. ZT_ResultCode rc = ZT_Node_processVirtualNetworkFrame(
  649. node,
  650. NULL,
  651. now,
  652. nwid,
  653. sourceMac,
  654. destMac,
  655. etherType,
  656. vlanId,
  657. (const void*)localData,
  658. frameLength,
  659. &nextBackgroundTaskDeadline);
  660. free(localData);
  661. jlong *outDeadline = (jlong*)env->GetPrimitiveArrayCritical(out_nextBackgroundTaskDeadline, NULL);
  662. outDeadline[0] = (jlong)nextBackgroundTaskDeadline;
  663. env->ReleasePrimitiveArrayCritical(out_nextBackgroundTaskDeadline, outDeadline, 0);
  664. return createResultObject(env, rc);
  665. }
  666. /*
  667. * Class: com_zerotier_sdk_Node
  668. * Method: processWirePacket
  669. * Signature: (JJJLjava/net/InetSocketAddress;[B[J)Lcom/zerotier/sdk/ResultCode;
  670. */
  671. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_processWirePacket(
  672. JNIEnv *env, jobject obj,
  673. jlong id,
  674. jlong in_now,
  675. jlong in_localSocket,
  676. jobject in_remoteAddress,
  677. jbyteArray in_packetData,
  678. jlongArray out_nextBackgroundTaskDeadline)
  679. {
  680. int64_t nodeId = (int64_t) id;
  681. ZT_Node *node = findNode(nodeId);
  682. unsigned int nbtd_len = (unsigned int)env->GetArrayLength(out_nextBackgroundTaskDeadline);
  683. if(nbtd_len < 1)
  684. {
  685. LOGE("nbtd_len < 1");
  686. return ResultCode_RESULT_FATAL_ERROR_INTERNAL_enum;
  687. }
  688. int64_t now = (int64_t)in_now;
  689. sockaddr_storage remoteAddress = fromSocketAddressObject(env, in_remoteAddress);
  690. if (env->ExceptionCheck() || isSocketAddressEmpty(remoteAddress)) {
  691. return NULL;
  692. }
  693. unsigned int packetLength = (unsigned int)env->GetArrayLength(in_packetData);
  694. if(packetLength == 0)
  695. {
  696. LOGE("Empty packet?!?");
  697. return ResultCode_RESULT_FATAL_ERROR_INTERNAL_enum;
  698. }
  699. void *packetData = env->GetPrimitiveArrayCritical(in_packetData, NULL);
  700. void *localData = malloc(packetLength);
  701. memcpy(localData, packetData, packetLength);
  702. env->ReleasePrimitiveArrayCritical(in_packetData, packetData, 0);
  703. int64_t nextBackgroundTaskDeadline = 0;
  704. ZT_ResultCode rc = ZT_Node_processWirePacket(
  705. node,
  706. NULL,
  707. now,
  708. in_localSocket,
  709. &remoteAddress,
  710. localData,
  711. packetLength,
  712. &nextBackgroundTaskDeadline);
  713. if(rc != ZT_RESULT_OK)
  714. {
  715. LOGE("ZT_Node_processWirePacket returned: %d", rc);
  716. }
  717. free(localData);
  718. jlong *outDeadline = (jlong*)env->GetPrimitiveArrayCritical(out_nextBackgroundTaskDeadline, NULL);
  719. outDeadline[0] = (jlong)nextBackgroundTaskDeadline;
  720. env->ReleasePrimitiveArrayCritical(out_nextBackgroundTaskDeadline, outDeadline, 0);
  721. return createResultObject(env, rc);
  722. }
  723. /*
  724. * Class: com_zerotier_sdk_Node
  725. * Method: processBackgroundTasks
  726. * Signature: (JJ[J)Lcom/zerotier/sdk/ResultCode;
  727. */
  728. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_processBackgroundTasks(
  729. JNIEnv *env, jobject obj,
  730. jlong id,
  731. jlong in_now,
  732. jlongArray out_nextBackgroundTaskDeadline)
  733. {
  734. int64_t nodeId = (int64_t) id;
  735. ZT_Node *node = findNode(nodeId);
  736. unsigned int nbtd_len = env->GetArrayLength(out_nextBackgroundTaskDeadline);
  737. if(nbtd_len < 1)
  738. {
  739. return ResultCode_RESULT_FATAL_ERROR_INTERNAL_enum;
  740. }
  741. int64_t now = (int64_t)in_now;
  742. int64_t nextBackgroundTaskDeadline = 0;
  743. ZT_ResultCode rc = ZT_Node_processBackgroundTasks(node, NULL, now, &nextBackgroundTaskDeadline);
  744. jlong *outDeadline = (jlong*)env->GetPrimitiveArrayCritical(out_nextBackgroundTaskDeadline, NULL);
  745. outDeadline[0] = (jlong)nextBackgroundTaskDeadline;
  746. env->ReleasePrimitiveArrayCritical(out_nextBackgroundTaskDeadline, outDeadline, 0);
  747. return createResultObject(env, rc);
  748. }
  749. /*
  750. * Class: com_zerotier_sdk_Node
  751. * Method: join
  752. * Signature: (JJ)Lcom/zerotier/sdk/ResultCode;
  753. */
  754. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_join(
  755. JNIEnv *env, jobject obj, jlong id, jlong in_nwid)
  756. {
  757. int64_t nodeId = (int64_t) id;
  758. ZT_Node *node = findNode(nodeId);
  759. uint64_t nwid = (uint64_t)in_nwid;
  760. ZT_ResultCode rc = ZT_Node_join(node, nwid, NULL, NULL);
  761. return createResultObject(env, rc);
  762. }
  763. /*
  764. * Class: com_zerotier_sdk_Node
  765. * Method: leave
  766. * Signature: (JJ)Lcom/zerotier/sdk/ResultCode;
  767. */
  768. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_leave(
  769. JNIEnv *env, jobject obj, jlong id, jlong in_nwid)
  770. {
  771. int64_t nodeId = (int64_t) id;
  772. ZT_Node *node = findNode(nodeId);
  773. uint64_t nwid = (uint64_t)in_nwid;
  774. ZT_ResultCode rc = ZT_Node_leave(node, nwid, NULL, NULL);
  775. return createResultObject(env, rc);
  776. }
  777. /*
  778. * Class: com_zerotier_sdk_Node
  779. * Method: multicastSubscribe
  780. * Signature: (JJJJ)Lcom/zerotier/sdk/ResultCode;
  781. */
  782. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_multicastSubscribe(
  783. JNIEnv *env, jobject obj,
  784. jlong id,
  785. jlong in_nwid,
  786. jlong in_multicastGroup,
  787. jlong in_multicastAdi)
  788. {
  789. int64_t nodeId = (int64_t) id;
  790. ZT_Node *node = findNode(nodeId);
  791. uint64_t nwid = (uint64_t)in_nwid;
  792. uint64_t multicastGroup = (uint64_t)in_multicastGroup;
  793. unsigned long multicastAdi = (unsigned long)in_multicastAdi;
  794. ZT_ResultCode rc = ZT_Node_multicastSubscribe(
  795. node, NULL, nwid, multicastGroup, multicastAdi);
  796. return createResultObject(env, rc);
  797. }
  798. /*
  799. * Class: com_zerotier_sdk_Node
  800. * Method: multicastUnsubscribe
  801. * Signature: (JJJJ)Lcom/zerotier/sdk/ResultCode;
  802. */
  803. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_multicastUnsubscribe(
  804. JNIEnv *env, jobject obj,
  805. jlong id,
  806. jlong in_nwid,
  807. jlong in_multicastGroup,
  808. jlong in_multicastAdi)
  809. {
  810. int64_t nodeId = (int64_t) id;
  811. ZT_Node *node = findNode(nodeId);
  812. uint64_t nwid = (uint64_t)in_nwid;
  813. uint64_t multicastGroup = (uint64_t)in_multicastGroup;
  814. unsigned long multicastAdi = (unsigned long)in_multicastAdi;
  815. ZT_ResultCode rc = ZT_Node_multicastUnsubscribe(
  816. node, nwid, multicastGroup, multicastAdi);
  817. return createResultObject(env, rc);
  818. }
  819. /*
  820. * Class: com_zerotier_sdk_Node
  821. * Method: orbit
  822. * Signature: (JJJ)Lcom/zerotier/sdk/ResultCode;
  823. */
  824. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_orbit(
  825. JNIEnv *env, jobject obj,
  826. jlong id,
  827. jlong in_moonWorldId,
  828. jlong in_moonSeed)
  829. {
  830. int64_t nodeId = (int64_t)id;
  831. ZT_Node *node = findNode(nodeId);
  832. uint64_t moonWorldId = (uint64_t)in_moonWorldId;
  833. uint64_t moonSeed = (uint64_t)in_moonSeed;
  834. ZT_ResultCode rc = ZT_Node_orbit(node, NULL, moonWorldId, moonSeed);
  835. return createResultObject(env, rc);
  836. }
  837. /*
  838. * Class: com_zerotier_sdk_Node
  839. * Method: deorbit
  840. * Signature: (JJ)L/com/zerotier/sdk/ResultCode;
  841. */
  842. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_deorbit(
  843. JNIEnv *env, jobject obj,
  844. jlong id,
  845. jlong in_moonWorldId)
  846. {
  847. int64_t nodeId = (int64_t)id;
  848. ZT_Node *node = findNode(nodeId);
  849. uint64_t moonWorldId = (uint64_t)in_moonWorldId;
  850. ZT_ResultCode rc = ZT_Node_deorbit(node, NULL, moonWorldId);
  851. return createResultObject(env, rc);
  852. }
  853. /*
  854. * Class: com_zerotier_sdk_Node
  855. * Method: address
  856. * Signature: (J)J
  857. */
  858. JNIEXPORT jlong JNICALL Java_com_zerotier_sdk_Node_address(
  859. JNIEnv *env , jobject obj, jlong id)
  860. {
  861. int64_t nodeId = (int64_t) id;
  862. ZT_Node *node = findNode(nodeId);
  863. uint64_t address = ZT_Node_address(node);
  864. return (jlong)address;
  865. }
  866. /*
  867. * Class: com_zerotier_sdk_Node
  868. * Method: status
  869. * Signature: (J)Lcom/zerotier/sdk/NodeStatus;
  870. */
  871. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_status
  872. (JNIEnv *env, jobject obj, jlong id)
  873. {
  874. int64_t nodeId = (int64_t) id;
  875. ZT_Node *node = findNode(nodeId);
  876. ZT_NodeStatus nodeStatus;
  877. ZT_Node_status(node, &nodeStatus);
  878. return newNodeStatus(env, nodeStatus);
  879. }
  880. /*
  881. * Class: com_zerotier_sdk_Node
  882. * Method: networkConfig
  883. * Signature: (JJ)Lcom/zerotier/sdk/VirtualNetworkConfig;
  884. */
  885. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_networkConfig(
  886. JNIEnv *env, jobject obj, jlong id, jlong nwid)
  887. {
  888. int64_t nodeId = (int64_t) id;
  889. ZT_Node *node = findNode(nodeId);
  890. ZT_VirtualNetworkConfig *vnetConfig = ZT_Node_networkConfig(node, nwid);
  891. jobject vnetConfigObject = newNetworkConfig(env, *vnetConfig);
  892. ZT_Node_freeQueryResult(node, vnetConfig);
  893. return vnetConfigObject;
  894. }
  895. /*
  896. * Class: com_zerotier_sdk_Node
  897. * Method: version
  898. * Signature: ()Lcom/zerotier/sdk/Version;
  899. */
  900. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_version(
  901. JNIEnv *env, jobject obj)
  902. {
  903. int major = 0;
  904. int minor = 0;
  905. int revision = 0;
  906. ZT_version(&major, &minor, &revision);
  907. return newVersion(env, major, minor, revision);
  908. }
  909. /*
  910. * Class: com_zerotier_sdk_Node
  911. * Method: peers
  912. * Signature: (J)[Lcom/zerotier/sdk/Peer;
  913. */
  914. JNIEXPORT jobjectArray JNICALL Java_com_zerotier_sdk_Node_peers(
  915. JNIEnv *env, jobject obj, jlong id)
  916. {
  917. int64_t nodeId = (int64_t) id;
  918. ZT_Node *node = findNode(nodeId);
  919. ZT_PeerList *peerList = ZT_Node_peers(node);
  920. if(peerList == NULL)
  921. {
  922. LOGE("ZT_Node_peers returned NULL");
  923. return NULL;
  924. }
  925. jobjectArray peerArrayObj = newPeerArray(env, peerList->peers, peerList->peerCount);
  926. ZT_Node_freeQueryResult(node, peerList);
  927. peerList = NULL;
  928. return peerArrayObj;
  929. }
  930. /*
  931. * Class: com_zerotier_sdk_Node
  932. * Method: networks
  933. * Signature: (J)[Lcom/zerotier/sdk/VirtualNetworkConfig;
  934. */
  935. JNIEXPORT jobjectArray JNICALL Java_com_zerotier_sdk_Node_networks(
  936. JNIEnv *env, jobject obj, jlong id)
  937. {
  938. int64_t nodeId = (int64_t) id;
  939. ZT_Node *node = findNode(nodeId);
  940. ZT_VirtualNetworkList *networkList = ZT_Node_networks(node);
  941. if(networkList == NULL)
  942. {
  943. return NULL;
  944. }
  945. jobjectArray networkListObject = newVirtualNetworkConfigArray(env, networkList->networks, networkList->networkCount);
  946. ZT_Node_freeQueryResult(node, networkList);
  947. return networkListObject;
  948. }
  949. #ifdef __cplusplus
  950. } // extern "C"
  951. #endif