com_zerotierone_sdk_Node.cpp 48 KB

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