com_zerotierone_sdk_Node.cpp 46 KB

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