com_zerotierone_sdk_Node.cpp 29 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088
  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 "ZT1_jniutils.h"
  29. #include <ZeroTierOne.h>
  30. #include <map>
  31. #include <string>
  32. #include <assert.h>
  33. #include <string.h>
  34. #ifdef __cplusplus
  35. extern "C" {
  36. #endif
  37. namespace {
  38. struct JniRef
  39. {
  40. JniRef()
  41. : env(NULL)
  42. , node(NULL)
  43. , dataStoreGetListener(NULL)
  44. , dataStorePutListener(NULL)
  45. , packetSender(NULL)
  46. , frameListener(NULL)
  47. , configListener(NULL)
  48. {}
  49. uint64_t id;
  50. JNIEnv *env;
  51. ZT1_Node *node;
  52. jobject dataStoreGetListener;
  53. jobject dataStorePutListener;
  54. jobject packetSender;
  55. jobject frameListener;
  56. jobject configListener;
  57. };
  58. int VirtualNetworkConfigFunctionCallback(
  59. ZT1_Node *node,
  60. void *userData,
  61. uint64_t nwid,
  62. enum ZT1_VirtualNetworkConfigOperation operation,
  63. const ZT1_VirtualNetworkConfig *config)
  64. {
  65. JniRef *ref = (JniRef*)userData;
  66. assert(ref->node == node);
  67. JNIEnv *env = ref->env;
  68. jclass configListenerClass = NULL;
  69. jmethodID callbackMethod = NULL;
  70. configListenerClass = env->GetObjectClass(ref->configListener);
  71. if(configListenerClass == NULL)
  72. {
  73. return -1;
  74. }
  75. callbackMethod = env->GetMethodID(configListenerClass,
  76. "onNetworkConfigurationUpdated",
  77. "(JLcom/zerotierone/sdk/VirtualNetworkConfigOperation;Lcom/zerotierone/sdk/VirtualNetworkConfig;)I");
  78. if(callbackMethod == NULL)
  79. {
  80. return -2;
  81. }
  82. jobject operationObject = createVirtualNetworkConfigOperation(env, operation);
  83. if(operationObject == NULL)
  84. {
  85. return -3;
  86. }
  87. jobject networkConfigObject = newNetworkConfig(env, *config);
  88. if(networkConfigObject == NULL)
  89. {
  90. return -4;
  91. }
  92. return env->CallIntMethod(ref->configListener, callbackMethod, (jlong)nwid, operationObject, networkConfigObject);
  93. }
  94. void VirtualNetworkFrameFunctionCallback(ZT1_Node *node,void *userData,
  95. uint64_t nwid,
  96. uint64_t sourceMac,
  97. uint64_t destMac,
  98. unsigned int etherType,
  99. unsigned int vlanid,
  100. const void *frameData,
  101. unsigned int frameLength)
  102. {
  103. JniRef *ref = (JniRef*)userData;
  104. assert(ref->node == node);
  105. JNIEnv *env = ref->env;
  106. jclass frameListenerClass = NULL;
  107. jmethodID callbackMethod = NULL;
  108. frameListenerClass = env->GetObjectClass(ref->frameListener);
  109. if(frameListenerClass == NULL)
  110. {
  111. return;
  112. }
  113. callbackMethod = env->GetMethodID(frameListenerClass,
  114. "onVirtualNetworkFrame", "(JJJJJ[B)V");
  115. if(callbackMethod == NULL)
  116. {
  117. return;
  118. }
  119. jbyteArray dataArray = env->NewByteArray(frameLength);
  120. env->SetByteArrayRegion(dataArray, 0, frameLength, (jbyte*)frameData);
  121. env->CallVoidMethod(ref->frameListener, callbackMethod, nwid, sourceMac, destMac, etherType, vlanid, dataArray);
  122. }
  123. void EventCallback(ZT1_Node *node,void *userData,enum ZT1_Event event, const void *data)
  124. {
  125. JniRef *ref = (JniRef*)userData;
  126. assert(ref->node == node);
  127. JNIEnv *env = ref->env;
  128. }
  129. long DataStoreGetFunction(ZT1_Node *node,void *userData,
  130. const char *objectName,
  131. void *buffer,
  132. unsigned long bufferSize,
  133. unsigned long bufferIndex,
  134. unsigned long *out_objectSize)
  135. {
  136. JniRef *ref = (JniRef*)userData;
  137. assert(ref->node == node);
  138. JNIEnv *env = ref->env;
  139. jclass dataStoreGetClass = NULL;
  140. jmethodID callbackMethod = NULL;
  141. dataStoreGetClass = env->GetObjectClass(ref->dataStoreGetListener);
  142. if(dataStoreGetClass == NULL)
  143. {
  144. return -2;
  145. }
  146. callbackMethod = env->GetMethodID(dataStoreGetClass,
  147. "onDataStoreGet",
  148. "(Ljava/lang/String;[BJ[J)J");
  149. if(callbackMethod == NULL)
  150. {
  151. return -2;
  152. }
  153. jstring nameStr = env->NewStringUTF(objectName);
  154. jbyteArray bufferObj = env->NewByteArray(bufferSize);
  155. jlongArray objectSizeObj = env->NewLongArray(1);
  156. long retval = env->CallLongMethod(
  157. dataStoreGetClass, callbackMethod, nameStr, bufferObj, bufferIndex, objectSizeObj);
  158. env->GetByteArrayRegion(bufferObj, 0, bufferSize, (jbyte*)buffer);
  159. env->GetLongArrayRegion(objectSizeObj, 0, 1, (jlong*)&out_objectSize);
  160. env->ReleaseByteArrayElements(bufferObj, (jbyte*)buffer, 0);
  161. env->ReleaseLongArrayElements(objectSizeObj, (jlong*)&out_objectSize, 0);
  162. return retval;
  163. }
  164. int DataStorePutFunction(ZT1_Node *node,void *userData,
  165. const char *objectName,
  166. const void *buffer,
  167. unsigned long bufferSize,
  168. int secure)
  169. {
  170. JniRef *ref = (JniRef*)userData;
  171. assert(ref->node == node);
  172. JNIEnv *env = ref->env;
  173. jclass dataStorePutClass = NULL;
  174. jmethodID callbackMethod = NULL;
  175. jmethodID deleteMethod = NULL;
  176. dataStorePutClass = env->GetObjectClass(ref->dataStorePutListener);
  177. if(dataStorePutClass == NULL)
  178. {
  179. return -1;
  180. }
  181. callbackMethod = env->GetMethodID(dataStorePutClass,
  182. "onDataStorePut",
  183. "(Ljava/lang/String;[BZ)I");
  184. if(callbackMethod == NULL)
  185. {
  186. return -2;
  187. }
  188. deleteMethod = env->GetMethodID(dataStorePutClass,
  189. "onDelete", "(Ljava/lang/String;)I");
  190. if(deleteMethod == NULL)
  191. {
  192. return -3;
  193. }
  194. jstring nameStr = env->NewStringUTF(objectName);
  195. if(buffer == NULL)
  196. {
  197. // delete operation
  198. return env->CallIntMethod(dataStorePutClass, deleteMethod, nameStr);
  199. }
  200. else
  201. {
  202. // set operation
  203. jbyteArray bufferObj = env->NewByteArray(bufferSize);
  204. env->SetByteArrayRegion(bufferObj, 0, bufferSize, (jbyte*)buffer);
  205. bool secure = secure != 0;
  206. return env->CallIntMethod(dataStorePutClass, callbackMethod,
  207. nameStr, bufferObj, secure);
  208. }
  209. }
  210. int WirePacketSendFunction(ZT1_Node *node,void *userData,\
  211. const struct sockaddr_storage *address,
  212. unsigned int linkDesparation,
  213. const void *buffer,
  214. unsigned int bufferSize)
  215. {
  216. JniRef *ref = (JniRef*)userData;
  217. assert(ref->node == node);
  218. JNIEnv *env = ref->env;
  219. jclass packetSenderClass = NULL;
  220. jmethodID callbackMethod = NULL;
  221. packetSenderClass = env->GetObjectClass(ref->packetSender);
  222. if(packetSenderClass == NULL)
  223. {
  224. return -1;
  225. }
  226. callbackMethod = env->GetMethodID(packetSenderClass,
  227. "onSendPacketRequested", "(Ljava/lang/String;I[B)I");
  228. if(callbackMethod == NULL)
  229. {
  230. return -2;
  231. }
  232. jobject addressObj = newInetAddress(env, *address);
  233. jbyteArray bufferObj = env->NewByteArray(bufferSize);
  234. env->SetByteArrayRegion(bufferObj, 0, bufferSize, (jbyte*)buffer);
  235. return env->CallIntMethod(packetSenderClass, callbackMethod, addressObj, linkDesparation, bufferObj);
  236. }
  237. typedef std::map<uint64_t, JniRef*> NodeMap;
  238. static NodeMap nodeMap;
  239. ZT1_Node* findNode(uint64_t nodeId)
  240. {
  241. NodeMap::iterator found = nodeMap.find(nodeId);
  242. if(found != nodeMap.end())
  243. {
  244. JniRef *ref = found->second;
  245. return ref->node;
  246. }
  247. return NULL;
  248. }
  249. }
  250. /*
  251. * Class: com_zerotierone_sdk_Node
  252. * Method: node_init
  253. * Signature: (J)Lcom/zerotierone/sdk/ResultCode;
  254. */
  255. JNIEXPORT jobject JNICALL Java_com_zerotierone_sdk_Node_node_1init(
  256. JNIEnv *env, jobject obj, jlong now)
  257. {
  258. jobject resultObject = createResultObject(env, ZT1_RESULT_OK);
  259. ZT1_Node *node;
  260. JniRef *ref = new JniRef;
  261. ZT1_ResultCode rc = ZT1_Node_new(
  262. &node,
  263. ref,
  264. (uint64_t)now,
  265. &DataStoreGetFunction,
  266. &DataStorePutFunction,
  267. &WirePacketSendFunction,
  268. &VirtualNetworkFrameFunctionCallback,
  269. &VirtualNetworkConfigFunctionCallback,
  270. &EventCallback);
  271. if(rc != ZT1_RESULT_OK)
  272. {
  273. resultObject = createResultObject(env, rc);
  274. if(node)
  275. {
  276. ZT1_Node_delete(node);
  277. node = NULL;
  278. }
  279. delete ref;
  280. ref = NULL;
  281. return resultObject;
  282. }
  283. ref->id = (uint64_t)now;
  284. ref->env = env;
  285. ref->node = node;
  286. jclass cls = env->GetObjectClass(obj);
  287. jfieldID fid = env->GetFieldID(
  288. cls, "getListener", "Lcom/zerotierone/sdk/DataStoreGetListener;");
  289. if(fid == NULL)
  290. {
  291. return NULL; // exception already thrown
  292. }
  293. ref->dataStoreGetListener = env->GetObjectField(obj, fid);
  294. if(ref->dataStoreGetListener == NULL)
  295. {
  296. return NULL;
  297. }
  298. fid = env->GetFieldID(
  299. cls, "putListener", "Lcom/zerotierone/sdk/DataStorePutLisetner;");
  300. if(fid == NULL)
  301. {
  302. return NULL; // exception already thrown
  303. }
  304. ref->dataStorePutListener = env->GetObjectField(obj, fid);
  305. if(ref->dataStorePutListener == NULL)
  306. {
  307. return NULL;
  308. }
  309. fid = env->GetFieldID(
  310. cls, "sender", "Lcom/zerotierone/sdk/PacketSender;");
  311. if(fid == NULL)
  312. {
  313. return NULL; // exception already thrown
  314. }
  315. ref->packetSender = env->GetObjectField(obj, fid);
  316. if(ref->packetSender == NULL)
  317. {
  318. return NULL;
  319. }
  320. fid = env->GetFieldID(
  321. cls, "frameListener", "Lcom/zerotierone/sdk/VirtualNetworkFrameListener;");
  322. if(fid == NULL)
  323. {
  324. return NULL; // exception already thrown
  325. }
  326. ref->frameListener = env->GetObjectField(obj, fid);
  327. if(ref->frameListener = NULL)
  328. {
  329. return NULL;
  330. }
  331. fid = env->GetFieldID(
  332. cls, "configListener", "Lcom/zerotierone/sdk/VirtualNetworkConfigListener;");
  333. if(fid == NULL)
  334. {
  335. return NULL; // exception already thrown
  336. }
  337. ref->configListener = env->GetObjectField(obj, fid);
  338. if(ref->configListener == NULL)
  339. {
  340. return NULL;
  341. }
  342. nodeMap.insert(std::make_pair(ref->id, ref));
  343. return resultObject;
  344. }
  345. /*
  346. * Class: com_zerotierone_sdk_Node
  347. * Method: node_delete
  348. * Signature: (J)V
  349. */
  350. JNIEXPORT void JNICALL Java_com_zerotierone_sdk_Node_node_1delete(
  351. JNIEnv *env, jobject obj, jlong id)
  352. {
  353. uint64_t nodeId = (uint64_t)id;
  354. NodeMap::iterator found = nodeMap.find(nodeId);
  355. if(found != nodeMap.end())
  356. {
  357. JniRef *ref = found->second;
  358. nodeMap.erase(found);
  359. ZT1_Node_delete(ref->node);
  360. delete ref;
  361. ref = NULL;
  362. }
  363. }
  364. /*
  365. * Class: com_zerotierone_sdk_Node
  366. * Method: processVirtualNetworkFrame
  367. * Signature: (JJJJJII[B[J)Lcom/zerotierone/sdk/ResultCode;
  368. */
  369. JNIEXPORT jobject JNICALL Java_com_zerotierone_sdk_Node_processVirtualNetworkFrame(
  370. JNIEnv *env, jobject obj,
  371. jlong id,
  372. jlong in_now,
  373. jlong in_nwid,
  374. jlong in_sourceMac,
  375. jlong in_destMac,
  376. jint in_etherType,
  377. jint in_vlanId,
  378. jbyteArray in_frameData,
  379. jlongArray out_nextBackgroundTaskDeadline)
  380. {
  381. uint64_t nodeId = (uint64_t) id;
  382. ZT1_Node *node = findNode(nodeId);
  383. if(node == NULL)
  384. {
  385. // cannot find valid node. We should never get here.
  386. return createResultObject(env, ZT1_RESULT_FATAL_ERROR_INTERNAL);
  387. }
  388. unsigned int nbtd_len = env->GetArrayLength(out_nextBackgroundTaskDeadline);
  389. if(nbtd_len < 1)
  390. {
  391. // array for next background task length has 0 elements!
  392. return createResultObject(env, ZT1_RESULT_FATAL_ERROR_INTERNAL);
  393. }
  394. uint64_t now = (uint64_t)in_now;
  395. uint64_t nwid = (uint64_t)in_nwid;
  396. uint64_t sourceMac = (uint64_t)in_sourceMac;
  397. uint64_t destMac = (uint64_t)in_destMac;
  398. unsigned int etherType = (unsigned int)in_etherType;
  399. unsigned int vlanId = (unsigned int)in_vlanId;
  400. unsigned int frameLength = env->GetArrayLength(in_frameData);
  401. jbyte *frameData =env->GetByteArrayElements(in_frameData, NULL);
  402. uint64_t nextBackgroundTaskDeadline = 0;
  403. ZT1_ResultCode rc = ZT1_Node_processVirtualNetworkFrame(
  404. node,
  405. now,
  406. nwid,
  407. sourceMac,
  408. destMac,
  409. etherType,
  410. vlanId,
  411. (const void*)frameData,
  412. frameLength,
  413. &nextBackgroundTaskDeadline);
  414. jlong *outDeadline = env->GetLongArrayElements(out_nextBackgroundTaskDeadline, NULL);
  415. outDeadline[0] = (jlong)nextBackgroundTaskDeadline;
  416. env->ReleaseLongArrayElements(out_nextBackgroundTaskDeadline, outDeadline, 0);
  417. env->ReleaseByteArrayElements(in_frameData, frameData, 0);
  418. return createResultObject(env, rc);
  419. }
  420. /*
  421. * Class: com_zerotierone_sdk_Node
  422. * Method: processWirePacket
  423. * Signature: (JJLjava/net/InetAddress;I[B[J)Lcom/zerotierone/sdk/ResultCode;
  424. */
  425. JNIEXPORT jobject JNICALL Java_com_zerotierone_sdk_Node_processWirePacket(
  426. JNIEnv *env, jobject obj,
  427. jlong id,
  428. jlong in_now,
  429. jobject in_remoteAddress,
  430. jint in_linkDesparation,
  431. jbyteArray in_packetData,
  432. jlongArray out_nextBackgroundTaskDeadline)
  433. {
  434. uint64_t nodeId = (uint64_t) id;
  435. ZT1_Node *node = findNode(nodeId);
  436. if(node == NULL)
  437. {
  438. // cannot find valid node. We should never get here.
  439. return createResultObject(env, ZT1_RESULT_FATAL_ERROR_INTERNAL);
  440. }
  441. unsigned int nbtd_len = env->GetArrayLength(out_nextBackgroundTaskDeadline);
  442. if(nbtd_len < 1)
  443. {
  444. return createResultObject(env, ZT1_RESULT_FATAL_ERROR_INTERNAL);
  445. }
  446. uint64_t now = (uint64_t)in_now;
  447. unsigned int linkDesparation = (unsigned int)in_linkDesparation;
  448. // get the java.net.InetAddress class and getAddress() method
  449. jclass inetAddressClass = env->FindClass("java/net/InetAddress");
  450. if(inetAddressClass == NULL)
  451. {
  452. // can't find java.net.InetAddress
  453. return createResultObject(env, ZT1_RESULT_FATAL_ERROR_INTERNAL);
  454. }
  455. jmethodID getAddressMethod = env->GetMethodID(
  456. inetAddressClass, "getAddress", "()[B");
  457. if(getAddressMethod == NULL)
  458. {
  459. // cant find InetAddress.getAddres()
  460. return createResultObject(env, ZT1_RESULT_FATAL_ERROR_INTERNAL);
  461. }
  462. // Call InetAddress.getAddress()
  463. jbyteArray addressArray = (jbyteArray)env->CallObjectMethod(in_remoteAddress, getAddressMethod);
  464. if(addressArray == NULL)
  465. {
  466. // unable to call getAddress()
  467. return createResultObject(env, ZT1_RESULT_FATAL_ERROR_INTERNAL);
  468. }
  469. unsigned int addrSize = env->GetArrayLength(addressArray);
  470. // get the address bytes
  471. jbyte *addr = env->GetByteArrayElements(addressArray, NULL);
  472. sockaddr_storage remoteAddress = {};
  473. if(addrSize == 16)
  474. {
  475. // IPV6 address
  476. sockaddr_in6 ipv6 = {};
  477. ipv6.sin6_family = AF_INET6;
  478. memcpy(ipv6.sin6_addr.s6_addr, addr, 16);
  479. memcpy(&remoteAddress, &ipv6, sizeof(sockaddr_in6));
  480. }
  481. else if(addrSize = 4)
  482. {
  483. // IPV4 address
  484. sockaddr_in ipv4 = {};
  485. ipv4.sin_family = AF_INET;
  486. memcpy(&ipv4.sin_addr, addr, 4);
  487. memcpy(&remoteAddress, &ipv4, sizeof(sockaddr_in));
  488. }
  489. else
  490. {
  491. // unknown address type
  492. env->ReleaseByteArrayElements(addressArray, addr, 0);
  493. return createResultObject(env, ZT1_RESULT_FATAL_ERROR_INTERNAL);
  494. }
  495. unsigned int packetLength = env->GetArrayLength(in_packetData);
  496. jbyte *packetData = env->GetByteArrayElements(in_packetData, NULL);
  497. uint64_t nextBackgroundTaskDeadline = 0;
  498. ZT1_ResultCode rc = ZT1_Node_processWirePacket(
  499. node,
  500. now,
  501. &remoteAddress,
  502. linkDesparation,
  503. packetData,
  504. packetLength,
  505. &nextBackgroundTaskDeadline);
  506. jlong *outDeadline = env->GetLongArrayElements(out_nextBackgroundTaskDeadline, NULL);
  507. outDeadline[0] = (jlong)nextBackgroundTaskDeadline;
  508. env->ReleaseLongArrayElements(out_nextBackgroundTaskDeadline, outDeadline, 0);
  509. env->ReleaseByteArrayElements(addressArray, addr, 0);
  510. env->ReleaseByteArrayElements(in_packetData, packetData, 0);
  511. return createResultObject(env, ZT1_RESULT_OK);
  512. }
  513. /*
  514. * Class: com_zerotierone_sdk_Node
  515. * Method: processBackgroundTasks
  516. * Signature: (JJ[J)Lcom/zerotierone/sdk/ResultCode;
  517. */
  518. JNIEXPORT jobject JNICALL Java_com_zerotierone_sdk_Node_processBackgroundTasks(
  519. JNIEnv *env, jobject obj,
  520. jlong id,
  521. jlong in_now,
  522. jlongArray out_nextBackgroundTaskDeadline)
  523. {
  524. uint64_t nodeId = (uint64_t) id;
  525. ZT1_Node *node = findNode(nodeId);
  526. if(node == NULL)
  527. {
  528. // cannot find valid node. We should never get here.
  529. return createResultObject(env, ZT1_RESULT_FATAL_ERROR_INTERNAL);
  530. }
  531. unsigned int nbtd_len = env->GetArrayLength(out_nextBackgroundTaskDeadline);
  532. if(nbtd_len < 1)
  533. {
  534. return createResultObject(env, ZT1_RESULT_FATAL_ERROR_INTERNAL);
  535. }
  536. uint64_t now = (uint64_t)in_now;
  537. uint64_t nextBackgroundTaskDeadline = 0;
  538. ZT1_ResultCode rc = ZT1_Node_processBackgroundTasks(node, now, &nextBackgroundTaskDeadline);
  539. jlong *outDeadline = env->GetLongArrayElements(out_nextBackgroundTaskDeadline, NULL);
  540. outDeadline[0] = (jlong)nextBackgroundTaskDeadline;
  541. env->ReleaseLongArrayElements(out_nextBackgroundTaskDeadline, outDeadline, 0);
  542. return createResultObject(env, rc);
  543. }
  544. /*
  545. * Class: com_zerotierone_sdk_Node
  546. * Method: join
  547. * Signature: (JJ)Lcom/zerotierone/sdk/ResultCode;
  548. */
  549. JNIEXPORT jobject JNICALL Java_com_zerotierone_sdk_Node_join(
  550. JNIEnv *env, jobject obj, jlong id, jlong in_nwid)
  551. {
  552. uint64_t nodeId = (uint64_t) id;
  553. ZT1_Node *node = findNode(nodeId);
  554. if(node == NULL)
  555. {
  556. // cannot find valid node. We should never get here.
  557. return createResultObject(env, ZT1_RESULT_FATAL_ERROR_INTERNAL);
  558. }
  559. uint64_t nwid = (uint64_t)in_nwid;
  560. ZT1_ResultCode rc = ZT1_Node_join(node, nwid);
  561. return createResultObject(env, rc);
  562. }
  563. /*
  564. * Class: com_zerotierone_sdk_Node
  565. * Method: leave
  566. * Signature: (JJ)Lcom/zerotierone/sdk/ResultCode;
  567. */
  568. JNIEXPORT jobject JNICALL Java_com_zerotierone_sdk_Node_leave(
  569. JNIEnv *env, jobject obj, jlong id, jlong in_nwid)
  570. {
  571. uint64_t nodeId = (uint64_t) id;
  572. ZT1_Node *node = findNode(nodeId);
  573. if(node == NULL)
  574. {
  575. // cannot find valid node. We should never get here.
  576. return createResultObject(env, ZT1_RESULT_FATAL_ERROR_INTERNAL);
  577. }
  578. uint64_t nwid = (uint64_t)in_nwid;
  579. ZT1_ResultCode rc = ZT1_Node_leave(node, nwid);
  580. return createResultObject(env, rc);
  581. }
  582. /*
  583. * Class: com_zerotierone_sdk_Node
  584. * Method: multicastSubscribe
  585. * Signature: (JJJJ)Lcom/zerotierone/sdk/ResultCode;
  586. */
  587. JNIEXPORT jobject JNICALL Java_com_zerotierone_sdk_Node_multicastSubscribe(
  588. JNIEnv *env, jobject obj,
  589. jlong id,
  590. jlong in_nwid,
  591. jlong in_multicastGroup,
  592. jlong in_multicastAdi)
  593. {
  594. uint64_t nodeId = (uint64_t) id;
  595. ZT1_Node *node = findNode(nodeId);
  596. if(node == NULL)
  597. {
  598. // cannot find valid node. We should never get here.
  599. return createResultObject(env, ZT1_RESULT_FATAL_ERROR_INTERNAL);
  600. }
  601. uint64_t nwid = (uint64_t)in_nwid;
  602. uint64_t multicastGroup = (uint64_t)in_multicastGroup;
  603. uint64_t multicastAdi = (uint64_t)in_multicastAdi;
  604. ZT1_ResultCode rc = ZT1_Node_multicastSubscribe(
  605. node, nwid, multicastGroup, multicastAdi);
  606. return createResultObject(env, rc);
  607. }
  608. /*
  609. * Class: com_zerotierone_sdk_Node
  610. * Method: multicastUnsubscribe
  611. * Signature: (JJJJ)Lcom/zerotierone/sdk/ResultCode;
  612. */
  613. JNIEXPORT jobject JNICALL Java_com_zerotierone_sdk_Node_multicastUnsubscribe(
  614. JNIEnv *env, jobject obj,
  615. jlong id,
  616. jlong in_nwid,
  617. jlong in_multicastGroup,
  618. jlong in_multicastAdi)
  619. {
  620. uint64_t nodeId = (uint64_t) id;
  621. ZT1_Node *node = findNode(nodeId);
  622. if(node == NULL)
  623. {
  624. // cannot find valid node. We should never get here.
  625. return createResultObject(env, ZT1_RESULT_FATAL_ERROR_INTERNAL);
  626. }
  627. uint64_t nwid = (uint64_t)in_nwid;
  628. uint64_t multicastGroup = (uint64_t)in_multicastGroup;
  629. uint64_t multicastAdi = (uint64_t)in_multicastAdi;
  630. ZT1_ResultCode rc = ZT1_Node_multicastUnsubscribe(
  631. node, nwid, multicastGroup, multicastAdi);
  632. return createResultObject(env, rc);
  633. }
  634. /*
  635. * Class: com_zerotierone_sdk_Node
  636. * Method: address
  637. * Signature: (J)J
  638. */
  639. JNIEXPORT jlong JNICALL Java_com_zerotierone_sdk_Node_address(
  640. JNIEnv *env , jobject obj, jlong id)
  641. {
  642. uint64_t nodeId = (uint64_t) id;
  643. ZT1_Node *node = findNode(nodeId);
  644. if(node == NULL)
  645. {
  646. // cannot find valid node. We should never get here.
  647. return 0;
  648. }
  649. uint64_t address = ZT1_Node_address(node);
  650. return (jlong)address;
  651. }
  652. /*
  653. * Class: com_zerotierone_sdk_Node
  654. * Method: status
  655. * Signature: (J)Lcom/zerotierone/sdk/NodeStatus;
  656. */
  657. JNIEXPORT jobject JNICALL Java_com_zerotierone_sdk_Node_status
  658. (JNIEnv *env, jobject obj, jlong id)
  659. {
  660. uint64_t nodeId = (uint64_t) id;
  661. ZT1_Node *node = findNode(nodeId);
  662. if(node == NULL)
  663. {
  664. // cannot find valid node. We should never get here.
  665. return 0;
  666. }
  667. // static so we only have to look these up once
  668. static jclass nodeStatusClass = NULL;
  669. static jmethodID nodeStatusConstructor = NULL;
  670. // create a com.zerotierone.sdk.NodeStatus object
  671. if(nodeStatusClass == NULL)
  672. {
  673. nodeStatusClass = env->FindClass("com/zerotierone/sdk/NodeStatus");
  674. if(nodeStatusClass == NULL)
  675. {
  676. return NULL;
  677. }
  678. }
  679. if(nodeStatusConstructor == NULL)
  680. {
  681. nodeStatusConstructor = env->GetMethodID(
  682. nodeStatusClass, "<init>", "()V");
  683. if(nodeStatusConstructor == NULL)
  684. {
  685. return NULL;
  686. }
  687. }
  688. jobject nodeStatusObj = env->NewObject(nodeStatusClass, nodeStatusConstructor);
  689. if(nodeStatusObj == NULL)
  690. {
  691. return NULL;
  692. }
  693. ZT1_NodeStatus nodeStatus;
  694. ZT1_Node_status(node, &nodeStatus);
  695. static jfieldID addressField = NULL;
  696. static jfieldID publicIdentityField = NULL;
  697. static jfieldID secretIdentityField = NULL;
  698. static jfieldID onlineField = NULL;
  699. if(addressField == NULL)
  700. {
  701. addressField = env->GetFieldID(nodeStatusClass, "address", "J");
  702. if(addressField == NULL)
  703. {
  704. return NULL;
  705. }
  706. }
  707. if(publicIdentityField == NULL)
  708. {
  709. publicIdentityField = env->GetFieldID(nodeStatusClass, "publicIdentity", "Ljava/lang/String;");
  710. if(publicIdentityField == NULL)
  711. {
  712. return NULL;
  713. }
  714. }
  715. if(secretIdentityField == NULL)
  716. {
  717. secretIdentityField = env->GetFieldID(nodeStatusClass, "secretIdentity", "Ljava/lang/String;");
  718. if(secretIdentityField == NULL)
  719. {
  720. return NULL;
  721. }
  722. }
  723. if(onlineField == NULL)
  724. {
  725. onlineField = env->GetFieldID(nodeStatusClass, "online", "Z");
  726. if(onlineField == NULL)
  727. {
  728. return NULL;
  729. }
  730. }
  731. env->SetLongField(nodeStatusObj, addressField, nodeStatus.address);
  732. jstring pubIdentStr = env->NewStringUTF(nodeStatus.publicIdentity);
  733. if(pubIdentStr == NULL)
  734. {
  735. return NULL; // out of memory
  736. }
  737. env->SetObjectField(nodeStatusObj, publicIdentityField, pubIdentStr);
  738. jstring secIdentStr = env->NewStringUTF(nodeStatus.secretIdentity);
  739. if(secIdentStr == NULL)
  740. {
  741. return NULL; // out of memory
  742. }
  743. env->SetObjectField(nodeStatusObj, secretIdentityField, secIdentStr);
  744. env->SetBooleanField(nodeStatusObj, onlineField, nodeStatus.online);
  745. return nodeStatusObj;
  746. }
  747. /*
  748. * Class: com_zerotierone_sdk_Node
  749. * Method: networkConfig
  750. * Signature: (J)Lcom/zerotierone/sdk/VirtualNetworkConfig;
  751. */
  752. JNIEXPORT jobject JNICALL Java_com_zerotierone_sdk_Node_networkConfig(
  753. JNIEnv *env, jobject obj, jlong id, jlong nwid)
  754. {
  755. uint64_t nodeId = (uint64_t) id;
  756. ZT1_Node *node = findNode(nodeId);
  757. if(node == NULL)
  758. {
  759. // cannot find valid node. We should never get here.
  760. return 0;
  761. }
  762. ZT1_VirtualNetworkConfig *vnetConfig = ZT1_Node_networkConfig(node, nwid);
  763. jobject vnetConfigObject = newNetworkConfig(env, *vnetConfig);
  764. ZT1_Node_freeQueryResult(node, vnetConfig);
  765. return vnetConfigObject;
  766. }
  767. /*
  768. * Class: com_zerotierone_sdk_Node
  769. * Method: version
  770. * Signature: (J)Lcom/zerotierone/sdk/Version;
  771. */
  772. JNIEXPORT jobject JNICALL Java_com_zerotierone_sdk_Node_version(
  773. JNIEnv *env, jobject obj)
  774. {
  775. // create a com.zerotierone.sdk.Version object
  776. jclass versionClass = env->FindClass("com/zerotierone/sdk/Version");
  777. if(versionClass == NULL)
  778. {
  779. return NULL;
  780. }
  781. jmethodID versionConstructor = env->GetMethodID(
  782. versionClass, "<init>", "()V");
  783. if(versionConstructor == NULL)
  784. {
  785. return NULL;
  786. }
  787. jobject versionObj = env->NewObject(versionClass, versionConstructor);
  788. if(versionObj == NULL)
  789. {
  790. return NULL;
  791. }
  792. int major = 0;
  793. int minor = 0;
  794. int revision = 0;
  795. unsigned long featureFlags = 0;
  796. ZT1_version(&major, &minor, &revision, &featureFlags);
  797. // copy data to Version object
  798. static jfieldID majorField = NULL;
  799. static jfieldID minorField = NULL;
  800. static jfieldID revisionField = NULL;
  801. static jfieldID featureFlagsField = NULL;
  802. if(majorField == NULL)
  803. {
  804. majorField = env->GetFieldID(versionClass, "major", "I");
  805. if(majorField = NULL)
  806. {
  807. return NULL;
  808. }
  809. }
  810. if(minorField == NULL)
  811. {
  812. minorField = env->GetFieldID(versionClass, "minor", "I");
  813. if(minorField == NULL)
  814. {
  815. return NULL;
  816. }
  817. }
  818. if(revisionField == NULL)
  819. {
  820. revisionField = env->GetFieldID(versionClass, "revision", "I");
  821. if(revisionField == NULL)
  822. {
  823. return NULL;
  824. }
  825. }
  826. if(featureFlagsField == NULL)
  827. {
  828. featureFlagsField = env->GetFieldID(versionClass, "featureFlags", "J");
  829. if(featureFlagsField == NULL)
  830. {
  831. return NULL;
  832. }
  833. }
  834. env->SetIntField(versionObj, majorField, (jint)major);
  835. env->SetIntField(versionObj, minorField, (jint)minor);
  836. env->SetIntField(versionObj, revisionField, (jint)revision);
  837. env->SetLongField(versionObj, featureFlagsField, (jlong)featureFlags);
  838. return versionObj;
  839. }
  840. /*
  841. * Class: com_zerotierone_sdk_Node
  842. * Method: peers
  843. * Signature: (J)Ljava/util/ArrayList;
  844. */
  845. JNIEXPORT jobject JNICALL Java_com_zerotierone_sdk_Node_peers(
  846. JNIEnv *env, jobject obj, jlong id)
  847. {
  848. uint64_t nodeId = (uint64_t) id;
  849. ZT1_Node *node = findNode(nodeId);
  850. if(node == NULL)
  851. {
  852. // cannot find valid node. We should never get here.
  853. return 0;
  854. }
  855. ZT1_PeerList *peerList = ZT1_Node_peers(node);
  856. if(peerList == NULL)
  857. {
  858. return NULL;
  859. }
  860. jobject peerListObject = newArrayList(env);
  861. if(peerListObject == NULL)
  862. {
  863. ZT1_Node_freeQueryResult(node, peerList);
  864. return NULL;
  865. }
  866. for(unsigned int i = 0; i < peerList->peerCount; ++i)
  867. {
  868. jobject peerObj = newPeer(env, peerList->peers[i]);
  869. appendItemToArrayList(env, peerListObject, peerObj);
  870. }
  871. ZT1_Node_freeQueryResult(node, peerList);
  872. peerList = NULL;
  873. return peerListObject;
  874. }
  875. /*
  876. * Class: com_zerotierone_sdk_Node
  877. * Method: networks
  878. * Signature: (J)Ljava/util/ArrayList;
  879. */
  880. JNIEXPORT jobject JNICALL Java_com_zerotierone_sdk_Node_networks(
  881. JNIEnv *env, jobject obj, jlong id)
  882. {
  883. uint64_t nodeId = (uint64_t) id;
  884. ZT1_Node *node = findNode(nodeId);
  885. if(node == NULL)
  886. {
  887. // cannot find valid node. We should never get here.
  888. return 0;
  889. }
  890. ZT1_VirtualNetworkList *networkList = ZT1_Node_networks(node);
  891. if(networkList == NULL)
  892. {
  893. return NULL;
  894. }
  895. jobject networkListObject = newArrayList(env);
  896. if(networkListObject == NULL)
  897. {
  898. ZT1_Node_freeQueryResult(node, networkList);
  899. return NULL;
  900. }
  901. for(unsigned int i = 0; i < networkList->networkCount; ++i)
  902. {
  903. jobject networkObject = newNetworkConfig(env, networkList->networks[i]);
  904. appendItemToArrayList(env, networkListObject, networkObject);
  905. }
  906. ZT1_Node_freeQueryResult(node, networkList);
  907. return networkListObject;
  908. }
  909. #ifdef __cplusplus
  910. } // extern "C"
  911. #endif