2
0

com_zerotierone_sdk_Node.cpp 47 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615
  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_jniutils.h"
  29. #include "ZT_jnilookup.h"
  30. #include <ZeroTierOne.h>
  31. #include "Mutex.hpp"
  32. #include <map>
  33. #include <string>
  34. #include <assert.h>
  35. #include <string.h>
  36. // global static JNI Lookup Object
  37. JniLookup lookup;
  38. #ifdef __cplusplus
  39. extern "C" {
  40. #endif
  41. namespace {
  42. struct JniRef
  43. {
  44. JniRef()
  45. : jvm(NULL)
  46. , node(NULL)
  47. , dataStoreGetListener(NULL)
  48. , dataStorePutListener(NULL)
  49. , packetSender(NULL)
  50. , eventListener(NULL)
  51. , frameListener(NULL)
  52. , configListener(NULL)
  53. , pathChecker(NULL)
  54. , callbacks(NULL)
  55. {
  56. callbacks = (ZT_Node_Callbacks*)malloc(sizeof(ZT_Node_Callbacks));
  57. memset(callbacks, 0, sizeof(ZT_Node_Callbacks));
  58. }
  59. ~JniRef()
  60. {
  61. JNIEnv *env = NULL;
  62. jvm->GetEnv((void**)&env, JNI_VERSION_1_6);
  63. env->DeleteGlobalRef(dataStoreGetListener);
  64. env->DeleteGlobalRef(dataStorePutListener);
  65. env->DeleteGlobalRef(packetSender);
  66. env->DeleteGlobalRef(eventListener);
  67. env->DeleteGlobalRef(frameListener);
  68. env->DeleteGlobalRef(configListener);
  69. env->DeleteGlobalRef(pathChecker);
  70. free(callbacks);
  71. callbacks = NULL;
  72. }
  73. int64_t id;
  74. JavaVM *jvm;
  75. ZT_Node *node;
  76. jobject dataStoreGetListener;
  77. jobject dataStorePutListener;
  78. jobject packetSender;
  79. jobject eventListener;
  80. jobject frameListener;
  81. jobject configListener;
  82. jobject pathChecker;
  83. ZT_Node_Callbacks *callbacks;
  84. };
  85. int VirtualNetworkConfigFunctionCallback(
  86. ZT_Node *node,
  87. void *userData,
  88. void *threadData,
  89. uint64_t nwid,
  90. void **,
  91. enum ZT_VirtualNetworkConfigOperation operation,
  92. const ZT_VirtualNetworkConfig *config)
  93. {
  94. LOGV("VritualNetworkConfigFunctionCallback");
  95. JniRef *ref = (JniRef*)userData;
  96. JNIEnv *env = NULL;
  97. ref->jvm->GetEnv((void**)&env, JNI_VERSION_1_6);
  98. if (ref->configListener == NULL) {
  99. LOGE("configListener is NULL");
  100. return -1;
  101. }
  102. jclass configListenerClass = env->GetObjectClass(ref->configListener);
  103. if(configListenerClass == NULL)
  104. {
  105. LOGE("Couldn't find class for VirtualNetworkConfigListener instance");
  106. return -1;
  107. }
  108. jmethodID configListenerCallbackMethod = lookup.findMethod(configListenerClass,
  109. "onNetworkConfigurationUpdated",
  110. "(JLcom/zerotier/sdk/VirtualNetworkConfigOperation;Lcom/zerotier/sdk/VirtualNetworkConfig;)I");
  111. if(configListenerCallbackMethod == NULL)
  112. {
  113. LOGE("Couldn't find onVirtualNetworkFrame() method");
  114. return -2;
  115. }
  116. jobject operationObject = createVirtualNetworkConfigOperation(env, operation);
  117. if(operationObject == NULL)
  118. {
  119. LOGE("Error creating VirtualNetworkConfigOperation object");
  120. return -3;
  121. }
  122. jobject networkConfigObject = newNetworkConfig(env, *config);
  123. if(networkConfigObject == NULL)
  124. {
  125. LOGE("Error creating VirtualNetworkConfig object");
  126. return -4;
  127. }
  128. return env->CallIntMethod(
  129. ref->configListener,
  130. configListenerCallbackMethod,
  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 = NULL;
  153. ref->jvm->GetEnv((void**)&env, JNI_VERSION_1_6);
  154. if (ref->frameListener == NULL) {
  155. LOGE("frameListener is NULL");
  156. return;
  157. }
  158. jclass frameListenerClass = env->GetObjectClass(ref->frameListener);
  159. if(env->ExceptionCheck() || frameListenerClass == NULL)
  160. {
  161. LOGE("Couldn't find class for VirtualNetworkFrameListener instance");
  162. return;
  163. }
  164. jmethodID frameListenerCallbackMethod = lookup.findMethod(
  165. frameListenerClass,
  166. "onVirtualNetworkFrame", "(JJJJJ[B)V");
  167. if(env->ExceptionCheck() || frameListenerCallbackMethod == NULL)
  168. {
  169. LOGE("Couldn't find onVirtualNetworkFrame() method");
  170. return;
  171. }
  172. jbyteArray dataArray = env->NewByteArray(frameLength);
  173. if(env->ExceptionCheck() || dataArray == NULL)
  174. {
  175. LOGE("Couldn't create frame data array");
  176. return;
  177. }
  178. void *data = env->GetPrimitiveArrayCritical(dataArray, NULL);
  179. memcpy(data, frameData, frameLength);
  180. env->ReleasePrimitiveArrayCritical(dataArray, data, 0);
  181. if(env->ExceptionCheck())
  182. {
  183. LOGE("Error setting frame data to array");
  184. return;
  185. }
  186. env->CallVoidMethod(ref->frameListener, frameListenerCallbackMethod, (jlong)nwid, (jlong)sourceMac, (jlong)destMac, (jlong)etherType, (jlong)vlanid, dataArray);
  187. }
  188. void EventCallback(ZT_Node *node,
  189. void *userData,
  190. void *threadData,
  191. enum ZT_Event event,
  192. const void *data) {
  193. LOGV("EventCallback");
  194. JniRef *ref = (JniRef *) userData;
  195. if (ref->node != node && event != ZT_EVENT_UP) {
  196. LOGE("Nodes not equal. ref->node %p, node %p. Event: %d", ref->node, node, event);
  197. return;
  198. }
  199. JNIEnv *env = NULL;
  200. ref->jvm->GetEnv((void **) &env, JNI_VERSION_1_6);
  201. if (ref->eventListener == NULL) {
  202. LOGE("eventListener is NULL");
  203. return;
  204. }
  205. jclass eventListenerClass = env->GetObjectClass(ref->eventListener);
  206. if (eventListenerClass == NULL) {
  207. LOGE("Couldn't class for EventListener instance");
  208. return;
  209. }
  210. jmethodID onEventMethod = lookup.findMethod(eventListenerClass,
  211. "onEvent", "(Lcom/zerotier/sdk/Event;)V");
  212. if (onEventMethod == NULL) {
  213. LOGE("Couldn't find onEvent method");
  214. return;
  215. }
  216. jmethodID onTraceMethod = lookup.findMethod(eventListenerClass,
  217. "onTrace", "(Ljava/lang/String;)V");
  218. if (onTraceMethod == NULL) {
  219. LOGE("Couldn't find onTrace method");
  220. return;
  221. }
  222. jobject eventObject = createEvent(env, event);
  223. if (eventObject == NULL) {
  224. return;
  225. }
  226. switch (event) {
  227. case ZT_EVENT_UP: {
  228. LOGD("Event Up");
  229. env->CallVoidMethod(ref->eventListener, onEventMethod, eventObject);
  230. break;
  231. }
  232. case ZT_EVENT_OFFLINE: {
  233. LOGD("Event Offline");
  234. env->CallVoidMethod(ref->eventListener, onEventMethod, eventObject);
  235. break;
  236. }
  237. case ZT_EVENT_ONLINE: {
  238. LOGD("Event Online");
  239. env->CallVoidMethod(ref->eventListener, onEventMethod, eventObject);
  240. break;
  241. }
  242. case ZT_EVENT_DOWN: {
  243. LOGD("Event Down");
  244. env->CallVoidMethod(ref->eventListener, onEventMethod, eventObject);
  245. break;
  246. }
  247. case ZT_EVENT_FATAL_ERROR_IDENTITY_COLLISION: {
  248. LOGV("Identity Collision");
  249. // call onEvent()
  250. env->CallVoidMethod(ref->eventListener, onEventMethod, eventObject);
  251. }
  252. break;
  253. case ZT_EVENT_TRACE: {
  254. LOGV("Trace Event");
  255. // call onTrace()
  256. if (data != NULL) {
  257. const char *message = (const char *) data;
  258. jstring messageStr = env->NewStringUTF(message);
  259. env->CallVoidMethod(ref->eventListener, onTraceMethod, messageStr);
  260. }
  261. }
  262. break;
  263. case ZT_EVENT_USER_MESSAGE:
  264. case ZT_EVENT_REMOTE_TRACE:
  265. default:
  266. break;
  267. }
  268. }
  269. void StatePutFunction(
  270. ZT_Node *node,
  271. void *userData,
  272. void *threadData,
  273. enum ZT_StateObjectType type,
  274. const uint64_t id[2],
  275. const void *buffer,
  276. int bufferLength) {
  277. char p[4096] = {0};
  278. bool secure = false;
  279. switch (type) {
  280. case ZT_STATE_OBJECT_IDENTITY_PUBLIC:
  281. snprintf(p, sizeof(p), "identity.public");
  282. break;
  283. case ZT_STATE_OBJECT_IDENTITY_SECRET:
  284. snprintf(p, sizeof(p), "identity.secret");
  285. secure = true;
  286. break;
  287. case ZT_STATE_OBJECT_PLANET:
  288. snprintf(p, sizeof(p), "planet");
  289. break;
  290. case ZT_STATE_OBJECT_MOON:
  291. snprintf(p, sizeof(p), "moons.d/%.16llx.moon", (unsigned long long)id[0]);
  292. break;
  293. case ZT_STATE_OBJECT_NETWORK_CONFIG:
  294. snprintf(p, sizeof(p), "networks.d/%.16llx.conf", (unsigned long long)id[0]);
  295. break;
  296. case ZT_STATE_OBJECT_PEER:
  297. snprintf(p, sizeof(p), "peers.d/%.10llx", (unsigned long long)id[0]);
  298. break;
  299. default:
  300. return;
  301. }
  302. if (strlen(p) < 1) {
  303. return;
  304. }
  305. JniRef *ref = (JniRef*)userData;
  306. JNIEnv *env = NULL;
  307. ref->jvm->GetEnv((void**)&env, JNI_VERSION_1_6);
  308. if (ref->dataStorePutListener == NULL) {
  309. LOGE("dataStorePutListener is NULL");
  310. return;
  311. }
  312. jclass dataStorePutClass = env->GetObjectClass(ref->dataStorePutListener);
  313. if (dataStorePutClass == NULL)
  314. {
  315. LOGE("Couldn't find class for DataStorePutListener instance");
  316. return;
  317. }
  318. jmethodID dataStorePutCallbackMethod = lookup.findMethod(
  319. dataStorePutClass,
  320. "onDataStorePut",
  321. "(Ljava/lang/String;[BZ)I");
  322. if(dataStorePutCallbackMethod == NULL)
  323. {
  324. LOGE("Couldn't find onDataStorePut method");
  325. return;
  326. }
  327. jmethodID deleteMethod = lookup.findMethod(dataStorePutClass,
  328. "onDelete", "(Ljava/lang/String;)I");
  329. if(deleteMethod == NULL)
  330. {
  331. LOGE("Couldn't find onDelete method");
  332. return;
  333. }
  334. jstring nameStr = env->NewStringUTF(p);
  335. if (bufferLength >= 0) {
  336. LOGD("JNI: Write file: %s", p);
  337. // set operation
  338. jbyteArray bufferObj = env->NewByteArray(bufferLength);
  339. if(env->ExceptionCheck() || bufferObj == NULL)
  340. {
  341. LOGE("Error creating byte array buffer!");
  342. return;
  343. }
  344. env->SetByteArrayRegion(bufferObj, 0, bufferLength, (jbyte*)buffer);
  345. env->CallIntMethod(ref->dataStorePutListener,
  346. dataStorePutCallbackMethod,
  347. nameStr, bufferObj, secure);
  348. } else {
  349. LOGD("JNI: Delete file: %s", p);
  350. env->CallIntMethod(ref->dataStorePutListener, deleteMethod, nameStr);
  351. }
  352. }
  353. int StateGetFunction(
  354. ZT_Node *node,
  355. void *userData,
  356. void *threadData,
  357. ZT_StateObjectType type,
  358. const uint64_t id[2],
  359. void *buffer,
  360. unsigned int bufferLength) {
  361. char p[4096] = {0};
  362. switch (type) {
  363. case ZT_STATE_OBJECT_IDENTITY_PUBLIC:
  364. snprintf(p, sizeof(p), "identity.public");
  365. break;
  366. case ZT_STATE_OBJECT_IDENTITY_SECRET:
  367. snprintf(p, sizeof(p), "identity.secret");
  368. break;
  369. case ZT_STATE_OBJECT_PLANET:
  370. snprintf(p, sizeof(p), "planet");
  371. break;
  372. case ZT_STATE_OBJECT_MOON:
  373. snprintf(p, sizeof(p), "moons.d/%.16llx.moon", (unsigned long long)id[0]);
  374. break;
  375. case ZT_STATE_OBJECT_NETWORK_CONFIG:
  376. snprintf(p, sizeof(p), "networks.d/%.16llx.conf", (unsigned long long)id[0]);
  377. break;
  378. case ZT_STATE_OBJECT_PEER:
  379. snprintf(p, sizeof(p), "peers.d/%.10llx", (unsigned long long)id[0]);
  380. break;
  381. default:
  382. return -1;
  383. }
  384. if (strlen(p) < 1) {
  385. return -1;
  386. }
  387. JniRef *ref = (JniRef*)userData;
  388. JNIEnv *env = NULL;
  389. ref->jvm->GetEnv((void**)&env, JNI_VERSION_1_6);
  390. if (ref->dataStoreGetListener == NULL) {
  391. LOGE("dataStoreGetListener is NULL");
  392. return -2;
  393. }
  394. jclass dataStoreGetClass = env->GetObjectClass(ref->dataStoreGetListener);
  395. if(dataStoreGetClass == NULL)
  396. {
  397. LOGE("Couldn't find class for DataStoreGetListener instance");
  398. return -2;
  399. }
  400. jmethodID dataStoreGetCallbackMethod = lookup.findMethod(
  401. dataStoreGetClass,
  402. "onDataStoreGet",
  403. "(Ljava/lang/String;[B)J");
  404. if(dataStoreGetCallbackMethod == NULL)
  405. {
  406. LOGE("Couldn't find onDataStoreGet method");
  407. return -2;
  408. }
  409. jstring nameStr = env->NewStringUTF(p);
  410. if(nameStr == NULL)
  411. {
  412. LOGE("Error creating name string object");
  413. return -2; // out of memory
  414. }
  415. jbyteArray bufferObj = env->NewByteArray(bufferLength);
  416. if(bufferObj == NULL)
  417. {
  418. LOGE("Error creating byte[] buffer of size: %u", bufferLength);
  419. return -2;
  420. }
  421. LOGV("Calling onDataStoreGet(%s, %p)", p, buffer);
  422. int retval = (int)env->CallLongMethod(
  423. ref->dataStoreGetListener,
  424. dataStoreGetCallbackMethod,
  425. nameStr,
  426. bufferObj);
  427. LOGV("onDataStoreGet returned %d", retval);
  428. if(retval > 0)
  429. {
  430. void *data = env->GetPrimitiveArrayCritical(bufferObj, NULL);
  431. memcpy(buffer, data, retval);
  432. env->ReleasePrimitiveArrayCritical(bufferObj, data, 0);
  433. }
  434. return retval;
  435. }
  436. int WirePacketSendFunction(ZT_Node *node,
  437. void *userData,
  438. void *threadData,
  439. int64_t localSocket,
  440. const struct sockaddr_storage *remoteAddress,
  441. const void *buffer,
  442. unsigned int bufferSize,
  443. unsigned int ttl)
  444. {
  445. LOGV("WirePacketSendFunction(%lld, %p, %p, %d)", (long long)localSocket, remoteAddress, buffer, bufferSize);
  446. JniRef *ref = (JniRef*)userData;
  447. assert(ref->node == node);
  448. JNIEnv *env = NULL;
  449. ref->jvm->GetEnv((void**)&env, JNI_VERSION_1_6);
  450. if (ref->packetSender == NULL) {
  451. LOGE("packetSender is NULL");
  452. return -1;
  453. }
  454. jclass packetSenderClass = env->GetObjectClass(ref->packetSender);
  455. if(packetSenderClass == NULL)
  456. {
  457. LOGE("Couldn't find class for PacketSender instance");
  458. return -1;
  459. }
  460. jmethodID packetSenderCallbackMethod = lookup.findMethod(packetSenderClass,
  461. "onSendPacketRequested", "(JLjava/net/InetSocketAddress;[BI)I");
  462. if(packetSenderCallbackMethod == NULL)
  463. {
  464. LOGE("Couldn't find onSendPacketRequested method");
  465. return -2;
  466. }
  467. jobject remoteAddressObj = newInetSocketAddress(env, *remoteAddress);
  468. jbyteArray bufferObj = env->NewByteArray(bufferSize);
  469. env->SetByteArrayRegion(bufferObj, 0, bufferSize, (jbyte*)buffer);
  470. int retval = env->CallIntMethod(ref->packetSender, packetSenderCallbackMethod, localSocket, remoteAddressObj, bufferObj);
  471. LOGV("JNI Packet Sender returned: %d", retval);
  472. return retval;
  473. }
  474. int PathCheckFunction(ZT_Node *node,
  475. void *userPtr,
  476. void *threadPtr,
  477. uint64_t address,
  478. int64_t localSocket,
  479. const struct sockaddr_storage *remoteAddress)
  480. {
  481. JniRef *ref = (JniRef*)userPtr;
  482. assert(ref->node == node);
  483. if(ref->pathChecker == NULL) {
  484. return true;
  485. }
  486. JNIEnv *env = NULL;
  487. ref->jvm->GetEnv((void**)&env, JNI_VERSION_1_6);
  488. jclass pathCheckerClass = env->GetObjectClass(ref->pathChecker);
  489. if(pathCheckerClass == NULL)
  490. {
  491. LOGE("Couldn't find class for PathChecker instance");
  492. return true;
  493. }
  494. jmethodID pathCheckCallbackMethod = lookup.findMethod(pathCheckerClass,
  495. "onPathCheck", "(JJLjava/net/InetSocketAddress;)Z");
  496. if(pathCheckCallbackMethod == NULL)
  497. {
  498. LOGE("Couldn't find onPathCheck method implementation");
  499. return true;
  500. }
  501. struct sockaddr_storage nullAddress = {0};
  502. jobject remoteAddressObj = NULL;
  503. if(memcmp(remoteAddress, &nullAddress, sizeof(sockaddr_storage)) != 0)
  504. {
  505. remoteAddressObj = newInetSocketAddress(env, *remoteAddress);
  506. }
  507. return env->CallBooleanMethod(ref->pathChecker, pathCheckCallbackMethod, address, localSocket, remoteAddressObj);
  508. }
  509. int PathLookupFunction(ZT_Node *node,
  510. void *userPtr,
  511. void *threadPtr,
  512. uint64_t address,
  513. int ss_family,
  514. struct sockaddr_storage *result)
  515. {
  516. JniRef *ref = (JniRef*)userPtr;
  517. assert(ref->node == node);
  518. if(ref->pathChecker == NULL) {
  519. return false;
  520. }
  521. JNIEnv *env = NULL;
  522. ref->jvm->GetEnv((void**)&env, JNI_VERSION_1_6);
  523. jclass pathCheckerClass = env->GetObjectClass(ref->pathChecker);
  524. if(pathCheckerClass == NULL)
  525. {
  526. LOGE("Couldn't find class for PathChecker instance");
  527. return false;
  528. }
  529. jmethodID pathLookupMethod = lookup.findMethod(pathCheckerClass,
  530. "onPathLookup", "(JI)Ljava/net/InetSocketAddress;");
  531. if(pathLookupMethod == NULL) {
  532. return false;
  533. }
  534. jobject sockAddressObject = env->CallObjectMethod(ref->pathChecker, pathLookupMethod, address, ss_family);
  535. if(sockAddressObject == NULL)
  536. {
  537. LOGE("Unable to call onPathLookup implementation");
  538. return false;
  539. }
  540. jclass inetSockAddressClass = env->GetObjectClass(sockAddressObject);
  541. if(inetSockAddressClass == NULL)
  542. {
  543. LOGE("Unable to find InetSocketAddress class");
  544. return false;
  545. }
  546. jmethodID getAddressMethod = lookup.findMethod(inetSockAddressClass, "getAddress", "()Ljava/net/InetAddress;");
  547. if(getAddressMethod == NULL)
  548. {
  549. LOGE("Unable to find InetSocketAddress.getAddress() method");
  550. return false;
  551. }
  552. jmethodID getPortMethod = lookup.findMethod(inetSockAddressClass, "getPort", "()I");
  553. if(getPortMethod == NULL)
  554. {
  555. LOGE("Unable to find InetSocketAddress.getPort() method");
  556. return false;
  557. }
  558. jint port = env->CallIntMethod(sockAddressObject, getPortMethod);
  559. jobject addressObject = env->CallObjectMethod(sockAddressObject, getAddressMethod);
  560. jclass inetAddressClass = lookup.findClass("java/net/InetAddress");
  561. if(inetAddressClass == NULL)
  562. {
  563. LOGE("Unable to find InetAddress class");
  564. return false;
  565. }
  566. getAddressMethod = lookup.findMethod(inetAddressClass, "getAddress", "()[B");
  567. if(getAddressMethod == NULL)
  568. {
  569. LOGE("Unable to find InetAddress.getAddress() method");
  570. return false;
  571. }
  572. jbyteArray addressBytes = (jbyteArray)env->CallObjectMethod(addressObject, getAddressMethod);
  573. if(addressBytes == NULL)
  574. {
  575. LOGE("Unable to call InetAddress.getBytes()");
  576. return false;
  577. }
  578. int addressSize = env->GetArrayLength(addressBytes);
  579. if(addressSize == 4)
  580. {
  581. // IPV4
  582. sockaddr_in *addr = (sockaddr_in*)result;
  583. addr->sin_family = AF_INET;
  584. addr->sin_port = htons(port);
  585. void *data = env->GetPrimitiveArrayCritical(addressBytes, NULL);
  586. memcpy(&addr->sin_addr, data, 4);
  587. env->ReleasePrimitiveArrayCritical(addressBytes, data, 0);
  588. }
  589. else if (addressSize == 16)
  590. {
  591. // IPV6
  592. sockaddr_in6 *addr = (sockaddr_in6*)result;
  593. addr->sin6_family = AF_INET6;
  594. addr->sin6_port = htons(port);
  595. void *data = env->GetPrimitiveArrayCritical(addressBytes, NULL);
  596. memcpy(&addr->sin6_addr, data, 16);
  597. env->ReleasePrimitiveArrayCritical(addressBytes, data, 0);
  598. }
  599. else
  600. {
  601. return false;
  602. }
  603. return true;
  604. }
  605. typedef std::map<int64_t, JniRef*> NodeMap;
  606. static NodeMap nodeMap;
  607. ZeroTier::Mutex nodeMapMutex;
  608. ZT_Node* findNode(int64_t nodeId)
  609. {
  610. ZeroTier::Mutex::Lock lock(nodeMapMutex);
  611. NodeMap::iterator found = nodeMap.find(nodeId);
  612. if(found != nodeMap.end())
  613. {
  614. JniRef *ref = found->second;
  615. return ref->node;
  616. }
  617. return NULL;
  618. }
  619. }
  620. JNIEXPORT jint JNICALL JNI_OnLoad(JavaVM *vm, void *reserved)
  621. {
  622. lookup.setJavaVM(vm);
  623. return JNI_VERSION_1_6;
  624. }
  625. JNIEXPORT void JNICALL JNI_OnUnload(JavaVM *vm, void *reserved)
  626. {
  627. }
  628. /*
  629. * Class: com_zerotier_sdk_Node
  630. * Method: node_init
  631. * Signature: (J)Lcom/zerotier/sdk/ResultCode;
  632. */
  633. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_node_1init(
  634. JNIEnv *env, jobject obj, jlong now)
  635. {
  636. LOGV("Creating ZT_Node struct");
  637. jobject resultObject = createResultObject(env, ZT_RESULT_OK);
  638. ZT_Node *node;
  639. JniRef *ref = new JniRef;
  640. ref->id = (int64_t)now;
  641. env->GetJavaVM(&ref->jvm);
  642. jclass cls = env->GetObjectClass(obj);
  643. jfieldID fid = lookup.findField(
  644. cls, "getListener", "Lcom/zerotier/sdk/DataStoreGetListener;");
  645. if(fid == NULL)
  646. {
  647. return NULL; // exception already thrown
  648. }
  649. jobject tmp = env->GetObjectField(obj, fid);
  650. if(tmp == NULL)
  651. {
  652. return NULL;
  653. }
  654. ref->dataStoreGetListener = env->NewGlobalRef(tmp);
  655. fid = lookup.findField(
  656. cls, "putListener", "Lcom/zerotier/sdk/DataStorePutListener;");
  657. if(fid == NULL)
  658. {
  659. return NULL; // exception already thrown
  660. }
  661. tmp = env->GetObjectField(obj, fid);
  662. if(tmp == NULL)
  663. {
  664. return NULL;
  665. }
  666. ref->dataStorePutListener = env->NewGlobalRef(tmp);
  667. fid = lookup.findField(
  668. cls, "sender", "Lcom/zerotier/sdk/PacketSender;");
  669. if(fid == NULL)
  670. {
  671. return NULL; // exception already thrown
  672. }
  673. tmp = env->GetObjectField(obj, fid);
  674. if(tmp == NULL)
  675. {
  676. return NULL;
  677. }
  678. ref->packetSender = env->NewGlobalRef(tmp);
  679. fid = lookup.findField(
  680. cls, "frameListener", "Lcom/zerotier/sdk/VirtualNetworkFrameListener;");
  681. if(fid == NULL)
  682. {
  683. return NULL; // exception already thrown
  684. }
  685. tmp = env->GetObjectField(obj, fid);
  686. if(tmp == NULL)
  687. {
  688. return NULL;
  689. }
  690. ref->frameListener = env->NewGlobalRef(tmp);
  691. fid = lookup.findField(
  692. cls, "configListener", "Lcom/zerotier/sdk/VirtualNetworkConfigListener;");
  693. if(fid == NULL)
  694. {
  695. return NULL; // exception already thrown
  696. }
  697. tmp = env->GetObjectField(obj, fid);
  698. if(tmp == NULL)
  699. {
  700. return NULL;
  701. }
  702. ref->configListener = env->NewGlobalRef(tmp);
  703. fid = lookup.findField(
  704. cls, "eventListener", "Lcom/zerotier/sdk/EventListener;");
  705. if(fid == NULL)
  706. {
  707. return NULL;
  708. }
  709. tmp = env->GetObjectField(obj, fid);
  710. if(tmp == NULL)
  711. {
  712. return NULL;
  713. }
  714. ref->eventListener = env->NewGlobalRef(tmp);
  715. fid = lookup.findField(
  716. cls, "pathChecker", "Lcom/zerotier/sdk/PathChecker;");
  717. if(fid == NULL)
  718. {
  719. LOGE("no path checker?");
  720. return NULL;
  721. }
  722. tmp = env->GetObjectField(obj, fid);
  723. if(tmp != NULL)
  724. {
  725. ref->pathChecker = env->NewGlobalRef(tmp);
  726. }
  727. ref->callbacks->stateGetFunction = &StateGetFunction;
  728. ref->callbacks->statePutFunction = &StatePutFunction;
  729. ref->callbacks->wirePacketSendFunction = &WirePacketSendFunction;
  730. ref->callbacks->virtualNetworkFrameFunction = &VirtualNetworkFrameFunctionCallback;
  731. ref->callbacks->virtualNetworkConfigFunction = &VirtualNetworkConfigFunctionCallback;
  732. ref->callbacks->eventCallback = &EventCallback;
  733. ref->callbacks->pathCheckFunction = &PathCheckFunction;
  734. ref->callbacks->pathLookupFunction = &PathLookupFunction;
  735. ZT_ResultCode rc = ZT_Node_new(
  736. &node,
  737. ref,
  738. NULL,
  739. ref->callbacks,
  740. (int64_t)now);
  741. if(rc != ZT_RESULT_OK)
  742. {
  743. LOGE("Error creating Node: %d", rc);
  744. resultObject = createResultObject(env, rc);
  745. if(node)
  746. {
  747. ZT_Node_delete(node);
  748. node = NULL;
  749. }
  750. delete ref;
  751. ref = NULL;
  752. return resultObject;
  753. }
  754. ZeroTier::Mutex::Lock lock(nodeMapMutex);
  755. ref->node = node;
  756. nodeMap.insert(std::make_pair(ref->id, ref));
  757. return resultObject;
  758. }
  759. /*
  760. * Class: com_zerotier_sdk_Node
  761. * Method: node_delete
  762. * Signature: (J)V
  763. */
  764. JNIEXPORT void JNICALL Java_com_zerotier_sdk_Node_node_1delete(
  765. JNIEnv *env, jobject obj, jlong id)
  766. {
  767. LOGV("Destroying ZT_Node struct");
  768. int64_t nodeId = (int64_t)id;
  769. NodeMap::iterator found;
  770. {
  771. ZeroTier::Mutex::Lock lock(nodeMapMutex);
  772. found = nodeMap.find(nodeId);
  773. }
  774. if(found != nodeMap.end())
  775. {
  776. JniRef *ref = found->second;
  777. nodeMap.erase(found);
  778. ZT_Node_delete(ref->node);
  779. delete ref;
  780. ref = NULL;
  781. }
  782. else
  783. {
  784. LOGE("Attempted to delete a node that doesn't exist!");
  785. }
  786. }
  787. /*
  788. * Class: com_zerotier_sdk_Node
  789. * Method: processVirtualNetworkFrame
  790. * Signature: (JJJJJII[B[J)Lcom/zerotier/sdk/ResultCode;
  791. */
  792. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_processVirtualNetworkFrame(
  793. JNIEnv *env, jobject obj,
  794. jlong id,
  795. jlong in_now,
  796. jlong in_nwid,
  797. jlong in_sourceMac,
  798. jlong in_destMac,
  799. jint in_etherType,
  800. jint in_vlanId,
  801. jbyteArray in_frameData,
  802. jlongArray out_nextBackgroundTaskDeadline)
  803. {
  804. int64_t nodeId = (int64_t) id;
  805. ZT_Node *node = findNode(nodeId);
  806. if(node == NULL)
  807. {
  808. // cannot find valid node. We should never get here.
  809. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  810. }
  811. unsigned int nbtd_len = env->GetArrayLength(out_nextBackgroundTaskDeadline);
  812. if(nbtd_len < 1)
  813. {
  814. // array for next background task length has 0 elements!
  815. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  816. }
  817. int64_t now = (int64_t)in_now;
  818. uint64_t nwid = (uint64_t)in_nwid;
  819. uint64_t sourceMac = (uint64_t)in_sourceMac;
  820. uint64_t destMac = (uint64_t)in_destMac;
  821. unsigned int etherType = (unsigned int)in_etherType;
  822. unsigned int vlanId = (unsigned int)in_vlanId;
  823. unsigned int frameLength = env->GetArrayLength(in_frameData);
  824. void *frameData = env->GetPrimitiveArrayCritical(in_frameData, NULL);
  825. void *localData = malloc(frameLength);
  826. memcpy(localData, frameData, frameLength);
  827. env->ReleasePrimitiveArrayCritical(in_frameData, frameData, 0);
  828. int64_t nextBackgroundTaskDeadline = 0;
  829. ZT_ResultCode rc = ZT_Node_processVirtualNetworkFrame(
  830. node,
  831. NULL,
  832. now,
  833. nwid,
  834. sourceMac,
  835. destMac,
  836. etherType,
  837. vlanId,
  838. (const void*)localData,
  839. frameLength,
  840. &nextBackgroundTaskDeadline);
  841. jlong *outDeadline = (jlong*)env->GetPrimitiveArrayCritical(out_nextBackgroundTaskDeadline, NULL);
  842. outDeadline[0] = (jlong)nextBackgroundTaskDeadline;
  843. env->ReleasePrimitiveArrayCritical(out_nextBackgroundTaskDeadline, outDeadline, 0);
  844. return createResultObject(env, rc);
  845. }
  846. /*
  847. * Class: com_zerotier_sdk_Node
  848. * Method: processWirePacket
  849. * Signature: (JJJLjava/net/InetSocketAddress;I[B[J)Lcom/zerotier/sdk/ResultCode;
  850. */
  851. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_processWirePacket(
  852. JNIEnv *env, jobject obj,
  853. jlong id,
  854. jlong in_now,
  855. jlong in_localSocket,
  856. jobject in_remoteAddress,
  857. jbyteArray in_packetData,
  858. jlongArray out_nextBackgroundTaskDeadline)
  859. {
  860. int64_t nodeId = (int64_t) id;
  861. ZT_Node *node = findNode(nodeId);
  862. if(node == NULL)
  863. {
  864. // cannot find valid node. We should never get here.
  865. LOGE("Couldn't find a valid node!");
  866. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  867. }
  868. unsigned int nbtd_len = (unsigned int)env->GetArrayLength(out_nextBackgroundTaskDeadline);
  869. if(nbtd_len < 1)
  870. {
  871. LOGE("nbtd_len < 1");
  872. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  873. }
  874. int64_t now = (int64_t)in_now;
  875. // get the java.net.InetSocketAddress class and getAddress() method
  876. jclass inetAddressClass = lookup.findClass("java/net/InetAddress");
  877. if(inetAddressClass == NULL)
  878. {
  879. LOGE("Can't find InetAddress class");
  880. // can't find java.net.InetAddress
  881. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  882. }
  883. jmethodID getAddressMethod = lookup.findMethod(
  884. inetAddressClass, "getAddress", "()[B");
  885. if(getAddressMethod == NULL)
  886. {
  887. // cant find InetAddress.getAddres()
  888. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  889. }
  890. jclass InetSocketAddressClass = lookup.findClass("java/net/InetSocketAddress");
  891. if(InetSocketAddressClass == NULL)
  892. {
  893. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  894. }
  895. jmethodID inetSockGetAddressMethod = lookup.findMethod(
  896. InetSocketAddressClass, "getAddress", "()Ljava/net/InetAddress;");
  897. jobject remoteAddrObject = env->CallObjectMethod(in_remoteAddress, inetSockGetAddressMethod);
  898. if(remoteAddrObject == NULL)
  899. {
  900. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  901. }
  902. jmethodID inetSock_getPort = lookup.findMethod(
  903. InetSocketAddressClass, "getPort", "()I");
  904. if(env->ExceptionCheck() || inetSock_getPort == NULL)
  905. {
  906. LOGE("Couldn't find getPort method on InetSocketAddress");
  907. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  908. }
  909. // call InetSocketAddress.getPort()
  910. int remotePort = env->CallIntMethod(in_remoteAddress, inetSock_getPort);
  911. if(env->ExceptionCheck())
  912. {
  913. LOGE("Exception calling InetSocketAddress.getPort()");
  914. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  915. }
  916. // Call InetAddress.getAddress()
  917. jbyteArray remoteAddressArray = (jbyteArray)env->CallObjectMethod(remoteAddrObject, getAddressMethod);
  918. if(remoteAddressArray == NULL)
  919. {
  920. LOGE("Unable to call getAddress()");
  921. // unable to call getAddress()
  922. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  923. }
  924. unsigned int addrSize = env->GetArrayLength(remoteAddressArray);
  925. // get the address bytes
  926. jbyte *addr = (jbyte*)env->GetPrimitiveArrayCritical(remoteAddressArray, NULL);
  927. sockaddr_storage remoteAddress = {};
  928. if(addrSize == 16)
  929. {
  930. // IPV6 address
  931. sockaddr_in6 ipv6 = {};
  932. ipv6.sin6_family = AF_INET6;
  933. ipv6.sin6_port = htons(remotePort);
  934. memcpy(ipv6.sin6_addr.s6_addr, addr, 16);
  935. memcpy(&remoteAddress, &ipv6, sizeof(sockaddr_in6));
  936. }
  937. else if(addrSize == 4)
  938. {
  939. // IPV4 address
  940. sockaddr_in ipv4 = {};
  941. ipv4.sin_family = AF_INET;
  942. ipv4.sin_port = htons(remotePort);
  943. memcpy(&ipv4.sin_addr, addr, 4);
  944. memcpy(&remoteAddress, &ipv4, sizeof(sockaddr_in));
  945. }
  946. else
  947. {
  948. LOGE("Unknown IP version");
  949. // unknown address type
  950. env->ReleasePrimitiveArrayCritical(remoteAddressArray, addr, 0);
  951. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  952. }
  953. env->ReleasePrimitiveArrayCritical(remoteAddressArray, addr, 0);
  954. unsigned int packetLength = (unsigned int)env->GetArrayLength(in_packetData);
  955. if(packetLength == 0)
  956. {
  957. LOGE("Empty packet?!?");
  958. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  959. }
  960. void *packetData = env->GetPrimitiveArrayCritical(in_packetData, NULL);
  961. void *localData = malloc(packetLength);
  962. memcpy(localData, packetData, packetLength);
  963. env->ReleasePrimitiveArrayCritical(in_packetData, packetData, 0);
  964. int64_t nextBackgroundTaskDeadline = 0;
  965. ZT_ResultCode rc = ZT_Node_processWirePacket(
  966. node,
  967. NULL,
  968. now,
  969. in_localSocket,
  970. &remoteAddress,
  971. localData,
  972. packetLength,
  973. &nextBackgroundTaskDeadline);
  974. if(rc != ZT_RESULT_OK)
  975. {
  976. LOGE("ZT_Node_processWirePacket returned: %d", rc);
  977. }
  978. free(localData);
  979. jlong *outDeadline = (jlong*)env->GetPrimitiveArrayCritical(out_nextBackgroundTaskDeadline, NULL);
  980. outDeadline[0] = (jlong)nextBackgroundTaskDeadline;
  981. env->ReleasePrimitiveArrayCritical(out_nextBackgroundTaskDeadline, outDeadline, 0);
  982. return createResultObject(env, rc);
  983. }
  984. /*
  985. * Class: com_zerotier_sdk_Node
  986. * Method: processBackgroundTasks
  987. * Signature: (JJ[J)Lcom/zerotier/sdk/ResultCode;
  988. */
  989. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_processBackgroundTasks(
  990. JNIEnv *env, jobject obj,
  991. jlong id,
  992. jlong in_now,
  993. jlongArray out_nextBackgroundTaskDeadline)
  994. {
  995. int64_t nodeId = (int64_t) id;
  996. ZT_Node *node = findNode(nodeId);
  997. if(node == NULL)
  998. {
  999. // cannot find valid node. We should never get here.
  1000. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  1001. }
  1002. unsigned int nbtd_len = env->GetArrayLength(out_nextBackgroundTaskDeadline);
  1003. if(nbtd_len < 1)
  1004. {
  1005. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  1006. }
  1007. int64_t now = (int64_t)in_now;
  1008. int64_t nextBackgroundTaskDeadline = 0;
  1009. ZT_ResultCode rc = ZT_Node_processBackgroundTasks(node, NULL, now, &nextBackgroundTaskDeadline);
  1010. jlong *outDeadline = (jlong*)env->GetPrimitiveArrayCritical(out_nextBackgroundTaskDeadline, NULL);
  1011. outDeadline[0] = (jlong)nextBackgroundTaskDeadline;
  1012. env->ReleasePrimitiveArrayCritical(out_nextBackgroundTaskDeadline, outDeadline, 0);
  1013. return createResultObject(env, rc);
  1014. }
  1015. /*
  1016. * Class: com_zerotier_sdk_Node
  1017. * Method: join
  1018. * Signature: (JJ)Lcom/zerotier/sdk/ResultCode;
  1019. */
  1020. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_join(
  1021. JNIEnv *env, jobject obj, jlong id, jlong in_nwid)
  1022. {
  1023. int64_t nodeId = (int64_t) id;
  1024. ZT_Node *node = findNode(nodeId);
  1025. if(node == NULL)
  1026. {
  1027. // cannot find valid node. We should never get here.
  1028. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  1029. }
  1030. uint64_t nwid = (uint64_t)in_nwid;
  1031. ZT_ResultCode rc = ZT_Node_join(node, nwid, NULL, NULL);
  1032. return createResultObject(env, rc);
  1033. }
  1034. /*
  1035. * Class: com_zerotier_sdk_Node
  1036. * Method: leave
  1037. * Signature: (JJ)Lcom/zerotier/sdk/ResultCode;
  1038. */
  1039. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_leave(
  1040. JNIEnv *env, jobject obj, jlong id, jlong in_nwid)
  1041. {
  1042. int64_t nodeId = (int64_t) id;
  1043. ZT_Node *node = findNode(nodeId);
  1044. if(node == NULL)
  1045. {
  1046. // cannot find valid node. We should never get here.
  1047. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  1048. }
  1049. uint64_t nwid = (uint64_t)in_nwid;
  1050. ZT_ResultCode rc = ZT_Node_leave(node, nwid, NULL, NULL);
  1051. return createResultObject(env, rc);
  1052. }
  1053. /*
  1054. * Class: com_zerotier_sdk_Node
  1055. * Method: multicastSubscribe
  1056. * Signature: (JJJJ)Lcom/zerotier/sdk/ResultCode;
  1057. */
  1058. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_multicastSubscribe(
  1059. JNIEnv *env, jobject obj,
  1060. jlong id,
  1061. jlong in_nwid,
  1062. jlong in_multicastGroup,
  1063. jlong in_multicastAdi)
  1064. {
  1065. int64_t nodeId = (int64_t) id;
  1066. ZT_Node *node = findNode(nodeId);
  1067. if(node == NULL)
  1068. {
  1069. // cannot find valid node. We should never get here.
  1070. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  1071. }
  1072. uint64_t nwid = (uint64_t)in_nwid;
  1073. uint64_t multicastGroup = (uint64_t)in_multicastGroup;
  1074. unsigned long multicastAdi = (unsigned long)in_multicastAdi;
  1075. ZT_ResultCode rc = ZT_Node_multicastSubscribe(
  1076. node, NULL, nwid, multicastGroup, multicastAdi);
  1077. return createResultObject(env, rc);
  1078. }
  1079. /*
  1080. * Class: com_zerotier_sdk_Node
  1081. * Method: multicastUnsubscribe
  1082. * Signature: (JJJJ)Lcom/zerotier/sdk/ResultCode;
  1083. */
  1084. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_multicastUnsubscribe(
  1085. JNIEnv *env, jobject obj,
  1086. jlong id,
  1087. jlong in_nwid,
  1088. jlong in_multicastGroup,
  1089. jlong in_multicastAdi)
  1090. {
  1091. int64_t nodeId = (int64_t) id;
  1092. ZT_Node *node = findNode(nodeId);
  1093. if(node == NULL)
  1094. {
  1095. // cannot find valid node. We should never get here.
  1096. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  1097. }
  1098. uint64_t nwid = (uint64_t)in_nwid;
  1099. uint64_t multicastGroup = (uint64_t)in_multicastGroup;
  1100. unsigned long multicastAdi = (unsigned long)in_multicastAdi;
  1101. ZT_ResultCode rc = ZT_Node_multicastUnsubscribe(
  1102. node, nwid, multicastGroup, multicastAdi);
  1103. return createResultObject(env, rc);
  1104. }
  1105. /*
  1106. * Class: com_zerotier_sdk_Node
  1107. * Method: orbit
  1108. * Signature: (JJJ)Lcom/zerotier/sdk/ResultCode;
  1109. */
  1110. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_orbit(
  1111. JNIEnv *env, jobject obj,
  1112. jlong id,
  1113. jlong in_moonWorldId,
  1114. jlong in_moonSeed)
  1115. {
  1116. int64_t nodeId = (int64_t)id;
  1117. ZT_Node *node = findNode(nodeId);
  1118. if(node == NULL)
  1119. {
  1120. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  1121. }
  1122. uint64_t moonWorldId = (uint64_t)in_moonWorldId;
  1123. uint64_t moonSeed = (uint64_t)in_moonSeed;
  1124. ZT_ResultCode rc = ZT_Node_orbit(node, NULL, moonWorldId, moonSeed);
  1125. return createResultObject(env, rc);
  1126. }
  1127. /*
  1128. * Class: com_zerotier_sdk_Node
  1129. * Method: deorbit
  1130. * Signature: (JJ)L/com/zerotier/sdk/ResultCode;
  1131. */
  1132. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_deorbit(
  1133. JNIEnv *env, jobject obj,
  1134. jlong id,
  1135. jlong in_moonWorldId)
  1136. {
  1137. int64_t nodeId = (int64_t)id;
  1138. ZT_Node *node = findNode(nodeId);
  1139. if(node == NULL)
  1140. {
  1141. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  1142. }
  1143. uint64_t moonWorldId = (uint64_t)in_moonWorldId;
  1144. ZT_ResultCode rc = ZT_Node_deorbit(node, NULL, moonWorldId);
  1145. return createResultObject(env, rc);
  1146. }
  1147. /*
  1148. * Class: com_zerotier_sdk_Node
  1149. * Method: address
  1150. * Signature: (J)J
  1151. */
  1152. JNIEXPORT jlong JNICALL Java_com_zerotier_sdk_Node_address(
  1153. JNIEnv *env , jobject obj, jlong id)
  1154. {
  1155. int64_t nodeId = (int64_t) id;
  1156. ZT_Node *node = findNode(nodeId);
  1157. if(node == NULL)
  1158. {
  1159. // cannot find valid node. We should never get here.
  1160. return 0;
  1161. }
  1162. uint64_t address = ZT_Node_address(node);
  1163. return (jlong)address;
  1164. }
  1165. /*
  1166. * Class: com_zerotier_sdk_Node
  1167. * Method: status
  1168. * Signature: (J)Lcom/zerotier/sdk/NodeStatus;
  1169. */
  1170. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_status
  1171. (JNIEnv *env, jobject obj, jlong id)
  1172. {
  1173. int64_t nodeId = (int64_t) id;
  1174. ZT_Node *node = findNode(nodeId);
  1175. if(node == NULL)
  1176. {
  1177. // cannot find valid node. We should never get here.
  1178. return 0;
  1179. }
  1180. jclass nodeStatusClass = NULL;
  1181. jmethodID nodeStatusConstructor = NULL;
  1182. // create a com.zerotier.sdk.NodeStatus object
  1183. nodeStatusClass = lookup.findClass("com/zerotier/sdk/NodeStatus");
  1184. if(nodeStatusClass == NULL)
  1185. {
  1186. return NULL;
  1187. }
  1188. nodeStatusConstructor = lookup.findMethod(
  1189. nodeStatusClass, "<init>", "()V");
  1190. if(nodeStatusConstructor == NULL)
  1191. {
  1192. return NULL;
  1193. }
  1194. jobject nodeStatusObj = env->NewObject(nodeStatusClass, nodeStatusConstructor);
  1195. if(nodeStatusObj == NULL)
  1196. {
  1197. return NULL;
  1198. }
  1199. ZT_NodeStatus nodeStatus;
  1200. ZT_Node_status(node, &nodeStatus);
  1201. jfieldID addressField = NULL;
  1202. jfieldID publicIdentityField = NULL;
  1203. jfieldID secretIdentityField = NULL;
  1204. jfieldID onlineField = NULL;
  1205. addressField = lookup.findField(nodeStatusClass, "address", "J");
  1206. if(addressField == NULL)
  1207. {
  1208. return NULL;
  1209. }
  1210. publicIdentityField = lookup.findField(nodeStatusClass, "publicIdentity", "Ljava/lang/String;");
  1211. if(publicIdentityField == NULL)
  1212. {
  1213. return NULL;
  1214. }
  1215. secretIdentityField = lookup.findField(nodeStatusClass, "secretIdentity", "Ljava/lang/String;");
  1216. if(secretIdentityField == NULL)
  1217. {
  1218. return NULL;
  1219. }
  1220. onlineField = lookup.findField(nodeStatusClass, "online", "Z");
  1221. if(onlineField == NULL)
  1222. {
  1223. return NULL;
  1224. }
  1225. env->SetLongField(nodeStatusObj, addressField, nodeStatus.address);
  1226. jstring pubIdentStr = env->NewStringUTF(nodeStatus.publicIdentity);
  1227. if(pubIdentStr == NULL)
  1228. {
  1229. return NULL; // out of memory
  1230. }
  1231. env->SetObjectField(nodeStatusObj, publicIdentityField, pubIdentStr);
  1232. jstring secIdentStr = env->NewStringUTF(nodeStatus.secretIdentity);
  1233. if(secIdentStr == NULL)
  1234. {
  1235. return NULL; // out of memory
  1236. }
  1237. env->SetObjectField(nodeStatusObj, secretIdentityField, secIdentStr);
  1238. env->SetBooleanField(nodeStatusObj, onlineField, nodeStatus.online);
  1239. return nodeStatusObj;
  1240. }
  1241. /*
  1242. * Class: com_zerotier_sdk_Node
  1243. * Method: networkConfig
  1244. * Signature: (J)Lcom/zerotier/sdk/VirtualNetworkConfig;
  1245. */
  1246. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_networkConfig(
  1247. JNIEnv *env, jobject obj, jlong id, jlong nwid)
  1248. {
  1249. int64_t nodeId = (int64_t) id;
  1250. ZT_Node *node = findNode(nodeId);
  1251. if(node == NULL)
  1252. {
  1253. // cannot find valid node. We should never get here.
  1254. return 0;
  1255. }
  1256. ZT_VirtualNetworkConfig *vnetConfig = ZT_Node_networkConfig(node, nwid);
  1257. jobject vnetConfigObject = newNetworkConfig(env, *vnetConfig);
  1258. ZT_Node_freeQueryResult(node, vnetConfig);
  1259. return vnetConfigObject;
  1260. }
  1261. /*
  1262. * Class: com_zerotier_sdk_Node
  1263. * Method: version
  1264. * Signature: (J)Lcom/zerotier/sdk/Version;
  1265. */
  1266. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_version(
  1267. JNIEnv *env, jobject obj)
  1268. {
  1269. int major = 0;
  1270. int minor = 0;
  1271. int revision = 0;
  1272. ZT_version(&major, &minor, &revision);
  1273. return newVersion(env, major, minor, revision);
  1274. }
  1275. /*
  1276. * Class: com_zerotier_sdk_Node
  1277. * Method: peers
  1278. * Signature: (J)[Lcom/zerotier/sdk/Peer;
  1279. */
  1280. JNIEXPORT jobjectArray JNICALL Java_com_zerotier_sdk_Node_peers(
  1281. JNIEnv *env, jobject obj, jlong id)
  1282. {
  1283. int64_t nodeId = (int64_t) id;
  1284. ZT_Node *node = findNode(nodeId);
  1285. if(node == NULL)
  1286. {
  1287. // cannot find valid node. We should never get here.
  1288. return 0;
  1289. }
  1290. ZT_PeerList *peerList = ZT_Node_peers(node);
  1291. if(peerList == NULL)
  1292. {
  1293. LOGE("ZT_Node_peers returned NULL");
  1294. return NULL;
  1295. }
  1296. int peerCount = peerList->peerCount * 100;
  1297. LOGV("Ensure Local Capacity: %d", peerCount);
  1298. if(env->EnsureLocalCapacity(peerCount))
  1299. {
  1300. LOGE("EnsureLocalCapacity failed!!");
  1301. ZT_Node_freeQueryResult(node, peerList);
  1302. return NULL;
  1303. }
  1304. jclass peerClass = lookup.findClass("com/zerotier/sdk/Peer");
  1305. if(env->ExceptionCheck() || peerClass == NULL)
  1306. {
  1307. LOGE("Error finding Peer class");
  1308. ZT_Node_freeQueryResult(node, peerList);
  1309. return NULL;
  1310. }
  1311. jobjectArray peerArrayObj = env->NewObjectArray(
  1312. peerList->peerCount, peerClass, NULL);
  1313. if(env->ExceptionCheck() || peerArrayObj == NULL)
  1314. {
  1315. LOGE("Error creating Peer[] array");
  1316. ZT_Node_freeQueryResult(node, peerList);
  1317. return NULL;
  1318. }
  1319. for(unsigned int i = 0; i < peerList->peerCount; ++i)
  1320. {
  1321. jobject peerObj = newPeer(env, peerList->peers[i]);
  1322. env->SetObjectArrayElement(peerArrayObj, i, peerObj);
  1323. if(env->ExceptionCheck())
  1324. {
  1325. LOGE("Error assigning Peer object to array");
  1326. break;
  1327. }
  1328. }
  1329. ZT_Node_freeQueryResult(node, peerList);
  1330. peerList = NULL;
  1331. return peerArrayObj;
  1332. }
  1333. /*
  1334. * Class: com_zerotier_sdk_Node
  1335. * Method: networks
  1336. * Signature: (J)[Lcom/zerotier/sdk/VirtualNetworkConfig;
  1337. */
  1338. JNIEXPORT jobjectArray JNICALL Java_com_zerotier_sdk_Node_networks(
  1339. JNIEnv *env, jobject obj, jlong id)
  1340. {
  1341. int64_t nodeId = (int64_t) id;
  1342. ZT_Node *node = findNode(nodeId);
  1343. if(node == NULL)
  1344. {
  1345. // cannot find valid node. We should never get here.
  1346. return 0;
  1347. }
  1348. ZT_VirtualNetworkList *networkList = ZT_Node_networks(node);
  1349. if(networkList == NULL)
  1350. {
  1351. return NULL;
  1352. }
  1353. jclass vnetConfigClass = lookup.findClass("com/zerotier/sdk/VirtualNetworkConfig");
  1354. if(env->ExceptionCheck() || vnetConfigClass == NULL)
  1355. {
  1356. LOGE("Error finding VirtualNetworkConfig class");
  1357. ZT_Node_freeQueryResult(node, networkList);
  1358. return NULL;
  1359. }
  1360. jobjectArray networkListObject = env->NewObjectArray(
  1361. networkList->networkCount, vnetConfigClass, NULL);
  1362. if(env->ExceptionCheck() || networkListObject == NULL)
  1363. {
  1364. LOGE("Error creating VirtualNetworkConfig[] array");
  1365. ZT_Node_freeQueryResult(node, networkList);
  1366. return NULL;
  1367. }
  1368. for(unsigned int i = 0; i < networkList->networkCount; ++i)
  1369. {
  1370. jobject networkObject = newNetworkConfig(env, networkList->networks[i]);
  1371. env->SetObjectArrayElement(networkListObject, i, networkObject);
  1372. if(env->ExceptionCheck())
  1373. {
  1374. LOGE("Error assigning VirtualNetworkConfig object to array");
  1375. break;
  1376. }
  1377. }
  1378. ZT_Node_freeQueryResult(node, networkList);
  1379. return networkListObject;
  1380. }
  1381. #ifdef __cplusplus
  1382. } // extern "C"
  1383. #endif