com_zerotierone_sdk_Node.cpp 46 KB

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