com_zerotierone_sdk_Node.cpp 48 KB

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