com_zerotierone_sdk_Node.cpp 34 KB

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