com_zerotierone_sdk_Node.cpp 47 KB

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