com_zerotierone_sdk_Node.cpp 46 KB

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