com_zerotierone_sdk_Node.cpp 47 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616
  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. LOGE("no path checker?");
  676. return NULL;
  677. }
  678. tmp = env->GetObjectField(obj, fid);
  679. if(tmp != NULL)
  680. {
  681. ref->pathChecker = env->NewGlobalRef(tmp);
  682. }
  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: orbit
  1106. * Signature: (JJJ)Lcom/zerotier/sdk/ResultCode;
  1107. */
  1108. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_orbit(
  1109. JNIEnv *env, jobject obj,
  1110. jlong id,
  1111. jlong in_moonWorldId,
  1112. jlong in_moonSeed)
  1113. {
  1114. uint64_t nodeId = (uint64_t)id;
  1115. ZT_Node *node = findNode(nodeId);
  1116. if(node == NULL)
  1117. {
  1118. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  1119. }
  1120. uint64_t moonWorldId = (uint64_t)in_moonWorldId;
  1121. uint64_t moonSeed = (uint64_t)in_moonSeed;
  1122. ZT_ResultCode rc = ZT_Node_orbit(node, NULL, moonWorldId, moonSeed);
  1123. return createResultObject(env, rc);
  1124. }
  1125. /*
  1126. * Class: com_zerotier_sdk_Node
  1127. * Method: deorbit
  1128. * Signature: (JJ)L/com/zerotier/sdk/ResultCode;
  1129. */
  1130. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_deorbit(
  1131. JNIEnv *env, jobject obj,
  1132. jlong id,
  1133. jlong in_moonWorldId)
  1134. {
  1135. uint64_t nodeId = (uint64_t)id;
  1136. ZT_Node *node = findNode(nodeId);
  1137. if(node == NULL)
  1138. {
  1139. return createResultObject(env, ZT_RESULT_FATAL_ERROR_INTERNAL);
  1140. }
  1141. uint64_t moonWorldId = (uint64_t)in_moonWorldId;
  1142. ZT_ResultCode rc = ZT_Node_deorbit(node, NULL, moonWorldId);
  1143. return createResultObject(env, rc);
  1144. }
  1145. /*
  1146. * Class: com_zerotier_sdk_Node
  1147. * Method: address
  1148. * Signature: (J)J
  1149. */
  1150. JNIEXPORT jlong JNICALL Java_com_zerotier_sdk_Node_address(
  1151. JNIEnv *env , jobject obj, jlong id)
  1152. {
  1153. uint64_t nodeId = (uint64_t) id;
  1154. ZT_Node *node = findNode(nodeId);
  1155. if(node == NULL)
  1156. {
  1157. // cannot find valid node. We should never get here.
  1158. return 0;
  1159. }
  1160. uint64_t address = ZT_Node_address(node);
  1161. return (jlong)address;
  1162. }
  1163. /*
  1164. * Class: com_zerotier_sdk_Node
  1165. * Method: status
  1166. * Signature: (J)Lcom/zerotier/sdk/NodeStatus;
  1167. */
  1168. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_status
  1169. (JNIEnv *env, jobject obj, jlong id)
  1170. {
  1171. uint64_t nodeId = (uint64_t) id;
  1172. ZT_Node *node = findNode(nodeId);
  1173. if(node == NULL)
  1174. {
  1175. // cannot find valid node. We should never get here.
  1176. return 0;
  1177. }
  1178. jclass nodeStatusClass = NULL;
  1179. jmethodID nodeStatusConstructor = NULL;
  1180. // create a com.zerotier.sdk.NodeStatus object
  1181. nodeStatusClass = lookup.findClass("com/zerotier/sdk/NodeStatus");
  1182. if(nodeStatusClass == NULL)
  1183. {
  1184. return NULL;
  1185. }
  1186. nodeStatusConstructor = lookup.findMethod(
  1187. nodeStatusClass, "<init>", "()V");
  1188. if(nodeStatusConstructor == NULL)
  1189. {
  1190. return NULL;
  1191. }
  1192. jobject nodeStatusObj = env->NewObject(nodeStatusClass, nodeStatusConstructor);
  1193. if(nodeStatusObj == NULL)
  1194. {
  1195. return NULL;
  1196. }
  1197. ZT_NodeStatus nodeStatus;
  1198. ZT_Node_status(node, &nodeStatus);
  1199. jfieldID addressField = NULL;
  1200. jfieldID publicIdentityField = NULL;
  1201. jfieldID secretIdentityField = NULL;
  1202. jfieldID onlineField = NULL;
  1203. addressField = lookup.findField(nodeStatusClass, "address", "J");
  1204. if(addressField == NULL)
  1205. {
  1206. return NULL;
  1207. }
  1208. publicIdentityField = lookup.findField(nodeStatusClass, "publicIdentity", "Ljava/lang/String;");
  1209. if(publicIdentityField == NULL)
  1210. {
  1211. return NULL;
  1212. }
  1213. secretIdentityField = lookup.findField(nodeStatusClass, "secretIdentity", "Ljava/lang/String;");
  1214. if(secretIdentityField == NULL)
  1215. {
  1216. return NULL;
  1217. }
  1218. onlineField = lookup.findField(nodeStatusClass, "online", "Z");
  1219. if(onlineField == NULL)
  1220. {
  1221. return NULL;
  1222. }
  1223. env->SetLongField(nodeStatusObj, addressField, nodeStatus.address);
  1224. jstring pubIdentStr = env->NewStringUTF(nodeStatus.publicIdentity);
  1225. if(pubIdentStr == NULL)
  1226. {
  1227. return NULL; // out of memory
  1228. }
  1229. env->SetObjectField(nodeStatusObj, publicIdentityField, pubIdentStr);
  1230. jstring secIdentStr = env->NewStringUTF(nodeStatus.secretIdentity);
  1231. if(secIdentStr == NULL)
  1232. {
  1233. return NULL; // out of memory
  1234. }
  1235. env->SetObjectField(nodeStatusObj, secretIdentityField, secIdentStr);
  1236. env->SetBooleanField(nodeStatusObj, onlineField, nodeStatus.online);
  1237. return nodeStatusObj;
  1238. }
  1239. /*
  1240. * Class: com_zerotier_sdk_Node
  1241. * Method: networkConfig
  1242. * Signature: (J)Lcom/zerotier/sdk/VirtualNetworkConfig;
  1243. */
  1244. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_networkConfig(
  1245. JNIEnv *env, jobject obj, jlong id, jlong nwid)
  1246. {
  1247. uint64_t nodeId = (uint64_t) id;
  1248. ZT_Node *node = findNode(nodeId);
  1249. if(node == NULL)
  1250. {
  1251. // cannot find valid node. We should never get here.
  1252. return 0;
  1253. }
  1254. ZT_VirtualNetworkConfig *vnetConfig = ZT_Node_networkConfig(node, nwid);
  1255. jobject vnetConfigObject = newNetworkConfig(env, *vnetConfig);
  1256. ZT_Node_freeQueryResult(node, vnetConfig);
  1257. return vnetConfigObject;
  1258. }
  1259. /*
  1260. * Class: com_zerotier_sdk_Node
  1261. * Method: version
  1262. * Signature: (J)Lcom/zerotier/sdk/Version;
  1263. */
  1264. JNIEXPORT jobject JNICALL Java_com_zerotier_sdk_Node_version(
  1265. JNIEnv *env, jobject obj)
  1266. {
  1267. int major = 0;
  1268. int minor = 0;
  1269. int revision = 0;
  1270. ZT_version(&major, &minor, &revision);
  1271. return newVersion(env, major, minor, revision);
  1272. }
  1273. /*
  1274. * Class: com_zerotier_sdk_Node
  1275. * Method: peers
  1276. * Signature: (J)[Lcom/zerotier/sdk/Peer;
  1277. */
  1278. JNIEXPORT jobjectArray JNICALL Java_com_zerotier_sdk_Node_peers(
  1279. JNIEnv *env, jobject obj, jlong id)
  1280. {
  1281. uint64_t nodeId = (uint64_t) id;
  1282. ZT_Node *node = findNode(nodeId);
  1283. if(node == NULL)
  1284. {
  1285. // cannot find valid node. We should never get here.
  1286. return 0;
  1287. }
  1288. ZT_PeerList *peerList = ZT_Node_peers(node);
  1289. if(peerList == NULL)
  1290. {
  1291. LOGE("ZT_Node_peers returned NULL");
  1292. return NULL;
  1293. }
  1294. int peerCount = peerList->peerCount * 100;
  1295. LOGV("Ensure Local Capacity: %d", peerCount);
  1296. if(env->EnsureLocalCapacity(peerCount))
  1297. {
  1298. LOGE("EnsureLocalCapacity failed!!");
  1299. ZT_Node_freeQueryResult(node, peerList);
  1300. return NULL;
  1301. }
  1302. jclass peerClass = lookup.findClass("com/zerotier/sdk/Peer");
  1303. if(env->ExceptionCheck() || peerClass == NULL)
  1304. {
  1305. LOGE("Error finding Peer class");
  1306. ZT_Node_freeQueryResult(node, peerList);
  1307. return NULL;
  1308. }
  1309. jobjectArray peerArrayObj = env->NewObjectArray(
  1310. peerList->peerCount, peerClass, NULL);
  1311. if(env->ExceptionCheck() || peerArrayObj == NULL)
  1312. {
  1313. LOGE("Error creating Peer[] array");
  1314. ZT_Node_freeQueryResult(node, peerList);
  1315. return NULL;
  1316. }
  1317. for(unsigned int i = 0; i < peerList->peerCount; ++i)
  1318. {
  1319. jobject peerObj = newPeer(env, peerList->peers[i]);
  1320. env->SetObjectArrayElement(peerArrayObj, i, peerObj);
  1321. if(env->ExceptionCheck())
  1322. {
  1323. LOGE("Error assigning Peer object to array");
  1324. break;
  1325. }
  1326. }
  1327. ZT_Node_freeQueryResult(node, peerList);
  1328. peerList = NULL;
  1329. return peerArrayObj;
  1330. }
  1331. /*
  1332. * Class: com_zerotier_sdk_Node
  1333. * Method: networks
  1334. * Signature: (J)[Lcom/zerotier/sdk/VirtualNetworkConfig;
  1335. */
  1336. JNIEXPORT jobjectArray JNICALL Java_com_zerotier_sdk_Node_networks(
  1337. JNIEnv *env, jobject obj, jlong id)
  1338. {
  1339. uint64_t nodeId = (uint64_t) id;
  1340. ZT_Node *node = findNode(nodeId);
  1341. if(node == NULL)
  1342. {
  1343. // cannot find valid node. We should never get here.
  1344. return 0;
  1345. }
  1346. ZT_VirtualNetworkList *networkList = ZT_Node_networks(node);
  1347. if(networkList == NULL)
  1348. {
  1349. return NULL;
  1350. }
  1351. jclass vnetConfigClass = lookup.findClass("com/zerotier/sdk/VirtualNetworkConfig");
  1352. if(env->ExceptionCheck() || vnetConfigClass == NULL)
  1353. {
  1354. LOGE("Error finding VirtualNetworkConfig class");
  1355. ZT_Node_freeQueryResult(node, networkList);
  1356. return NULL;
  1357. }
  1358. jobjectArray networkListObject = env->NewObjectArray(
  1359. networkList->networkCount, vnetConfigClass, NULL);
  1360. if(env->ExceptionCheck() || networkListObject == NULL)
  1361. {
  1362. LOGE("Error creating VirtualNetworkConfig[] array");
  1363. ZT_Node_freeQueryResult(node, networkList);
  1364. return NULL;
  1365. }
  1366. for(unsigned int i = 0; i < networkList->networkCount; ++i)
  1367. {
  1368. jobject networkObject = newNetworkConfig(env, networkList->networks[i]);
  1369. env->SetObjectArrayElement(networkListObject, i, networkObject);
  1370. if(env->ExceptionCheck())
  1371. {
  1372. LOGE("Error assigning VirtualNetworkConfig object to array");
  1373. break;
  1374. }
  1375. }
  1376. ZT_Node_freeQueryResult(node, networkList);
  1377. return networkListObject;
  1378. }
  1379. #ifdef __cplusplus
  1380. } // extern "C"
  1381. #endif