Joseph Henry
9 years ago
4 changed files with 340 additions and 298 deletions
@ -0,0 +1,321 @@
|
||||
/*
|
||||
* ZeroTier SDK - Network Virtualization Everywhere |
||||
* Copyright (C) 2011-2017 ZeroTier, Inc. https://www.zerotier.com/
|
||||
* |
||||
* This program is free software: you can redistribute it and/or modify |
||||
* it under the terms of the GNU General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
* -- |
||||
* |
||||
* You can be released from the requirements of the license by purchasing |
||||
* a commercial license. Buying such a license is mandatory as soon as you |
||||
* develop commercial closed-source software that incorporates or links |
||||
* directly against ZeroTier software without disclosing the source code |
||||
* of your own application. |
||||
*/ |
||||
|
||||
/**
|
||||
* @file |
||||
* |
||||
* Javs JNI wrapper for partially-POSIX-compliant socket API |
||||
* JNI naming convention: Java_PACKAGENAME_CLASSNAME_METHODNAME
|
||||
*/ |
||||
|
||||
#if defined(SDK_JNI) |
||||
|
||||
#include <sys/socket.h> |
||||
|
||||
#include "libzt.h" |
||||
#include "ZT1Service.h" |
||||
|
||||
#ifdef __cplusplus |
||||
extern "C" { |
||||
#endif |
||||
|
||||
namespace ZeroTier { |
||||
|
||||
#include <jni.h> |
||||
|
||||
/****************************************************************************/ |
||||
/* Java JNI Wrapper for ZeroTier Socket API */ |
||||
/****************************************************************************/ |
||||
|
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1sendto( |
||||
JNIEnv *env, jobject thisObj, jint fd, jarray buf, jint len, jint flags, jobject ztaddr) |
||||
{ |
||||
struct sockaddr_in addr; |
||||
jclass cls = (*env).GetObjectClass( ztaddr); |
||||
jfieldID f = (*env).GetFieldID( cls, "port", "I"); |
||||
addr.sin_port = htons((*env).GetIntField( ztaddr, f)); |
||||
f = (*env).GetFieldID( cls, "_rawAddr", "J"); |
||||
addr.sin_addr.s_addr = (*env).GetLongField( ztaddr, f); |
||||
addr.sin_family = AF_INET; |
||||
//LOGV("zt_sendto(): fd = %d\naddr = %s\nport=%d", fd, inet_ntoa(addr.sin_addr), ntohs(addr.sin_port));
|
||||
// TODO: Optimize this
|
||||
jbyte *body = (*env).GetByteArrayElements((_jbyteArray *)buf, 0); |
||||
char * bufp = (char *)malloc(sizeof(char)*len); |
||||
memcpy(bufp, body, len); |
||||
(*env).ReleaseByteArrayElements((_jbyteArray *)buf, body, 0); |
||||
// "connect" and send buffer contents
|
||||
int sent_bytes = zts_sendto(fd, body, len, flags, (struct sockaddr *)&addr, sizeof(addr)); |
||||
return sent_bytes; |
||||
} |
||||
|
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1recvfrom( |
||||
JNIEnv *env, jobject thisObj, jint fd, jbyteArray buf, jint len, jint flags, jobject ztaddr) |
||||
{ |
||||
struct sockaddr_in addr; |
||||
jbyte *body = (*env).GetByteArrayElements( buf, 0); |
||||
unsigned char buffer[ZT_SDK_MTU]; |
||||
int payload_offset = sizeof(int32_t) + sizeof(struct sockaddr_storage); |
||||
int rxbytes = zts_recvfrom(fd, &buffer, len, flags, (struct sockaddr *)&addr, (socklen_t *)sizeof(struct sockaddr_storage)); |
||||
if (rxbytes > 0) |
||||
memcpy(body, (jbyte*)buffer + payload_offset, rxbytes); |
||||
(*env).ReleaseByteArrayElements( buf, body, 0); |
||||
// Update fields of Java ZTAddress object
|
||||
jfieldID fid; |
||||
jclass cls = (*env).GetObjectClass( ztaddr); |
||||
fid = (*env).GetFieldID( cls, "port", "I"); |
||||
(*env).SetIntField( ztaddr, fid, addr.sin_port); |
||||
fid = (*env).GetFieldID( cls,"_rawAddr", "J"); |
||||
(*env).SetLongField( ztaddr, fid,addr.sin_addr.s_addr); |
||||
return rxbytes; |
||||
} |
||||
|
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1send(JNIEnv *env, jobject thisObj, jint fd, jarray buf, jint len, int flags) |
||||
{ |
||||
jbyte *body = (*env).GetByteArrayElements((_jbyteArray *)buf, 0); |
||||
char * bufp = (char *)malloc(sizeof(char)*len); |
||||
memcpy(bufp, body, len); |
||||
(*env).ReleaseByteArrayElements((_jbyteArray *)buf, body, 0); |
||||
int written_bytes = zts_write(fd, body, len); |
||||
return written_bytes; |
||||
} |
||||
|
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1write(JNIEnv *env, jobject thisObj,
|
||||
jint fd, jarray buf, jint len) |
||||
{ |
||||
jbyte *body = (*env).GetByteArrayElements((_jbyteArray *)buf, 0); |
||||
char * bufp = (char *)malloc(sizeof(char)*len); |
||||
memcpy(bufp, body, len); |
||||
(*env).ReleaseByteArrayElements((_jbyteArray *)buf, body, 0); |
||||
int written_bytes = zts_write(fd, body, len); |
||||
return written_bytes; |
||||
} |
||||
|
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1read(JNIEnv *env, jobject thisObj,
|
||||
jint fd, jarray buf, jint len) |
||||
{ |
||||
jbyte *body = (*env).GetByteArrayElements((_jbyteArray *)buf, 0); |
||||
int read_bytes = read(fd, body, len); |
||||
(*env).ReleaseByteArrayElements((_jbyteArray *)buf, body, 0); |
||||
return read_bytes; |
||||
} |
||||
|
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1setsockopt( |
||||
JNIEnv *env, jobject thisObj,
|
||||
jint fd, jint level, jint optname, jint optval, jint optlen) { |
||||
return zts_setsockopt(fd, level, optname, (const void*)optval, optlen); |
||||
} |
||||
|
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1getsockopt(JNIEnv *env, jobject thisObj,
|
||||
jint fd, jint level, jint optname, jint optval, jint optlen) { |
||||
return zts_getsockopt(fd, level, optname, (void*)optval, (socklen_t *)optlen); |
||||
} |
||||
|
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1socket(JNIEnv *env, jobject thisObj,
|
||||
jint family, jint type, jint protocol) { |
||||
return zts_socket(family, type, protocol); |
||||
} |
||||
|
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1connect(JNIEnv *env, jobject thisObj,
|
||||
jint fd, jstring addrstr, jint port) { |
||||
struct sockaddr_in addr; |
||||
const char *str = (*env).GetStringUTFChars( addrstr, 0); |
||||
addr.sin_addr.s_addr = inet_addr(str); |
||||
addr.sin_family = AF_INET; |
||||
addr.sin_port = htons( port ); |
||||
(*env).ReleaseStringUTFChars( addrstr, str); |
||||
return zts_connect(fd, (struct sockaddr *)&addr, sizeof(addr)); |
||||
} |
||||
|
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1bind(JNIEnv *env, jobject thisObj,
|
||||
jint fd, jstring addrstr, jint port) { |
||||
struct sockaddr_in addr; |
||||
const char *str = (*env).GetStringUTFChars( addrstr, 0); |
||||
DEBUG_INFO("fd=%d, addr=%s, port=%d", fd, str, port); |
||||
addr.sin_addr.s_addr = inet_addr(str); |
||||
addr.sin_family = AF_INET; |
||||
addr.sin_port = htons( port ); |
||||
(*env).ReleaseStringUTFChars( addrstr, str); |
||||
return zts_bind(fd, (struct sockaddr *)&addr, sizeof(addr)); |
||||
} |
||||
|
||||
#if defined(__linux__) |
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1accept4(JNIEnv *env, jobject thisObj,
|
||||
jint fd, jstring addrstr, jint port, jint flags) { |
||||
struct sockaddr_in addr; |
||||
char *str; |
||||
// = env->GetStringUTFChars(addrstr, NULL);
|
||||
(*env).ReleaseStringUTFChars( addrstr, str); |
||||
addr.sin_addr.s_addr = inet_addr(str); |
||||
addr.sin_family = AF_INET; |
||||
addr.sin_port = htons( port ); |
||||
return zts_accept4(fd, (struct sockaddr *)&addr, sizeof(addr), flags); |
||||
} |
||||
#endif |
||||
|
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1accept(JNIEnv *env, jobject thisObj,
|
||||
jint fd, jstring addrstr, jint port) { |
||||
struct sockaddr_in addr; |
||||
// TODO: Send addr info back to Javaland
|
||||
addr.sin_addr.s_addr = inet_addr(""); |
||||
addr.sin_family = AF_INET; |
||||
addr.sin_port = htons( port ); |
||||
return zts_accept(fd, (struct sockaddr *)&addr, (socklen_t *)sizeof(addr)); |
||||
} |
||||
|
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1listen(JNIEnv *env, jobject thisObj,
|
||||
jint fd, int backlog) { |
||||
return zts_listen(fd, backlog); |
||||
} |
||||
|
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1close(JNIEnv *env, jobject thisObj,
|
||||
jint fd) { |
||||
return zts_close(fd); |
||||
} |
||||
|
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1getsockname(JNIEnv *env, jobject thisObj,
|
||||
jint fd, jobject ztaddr) { |
||||
struct sockaddr_in addr; |
||||
int err = zts_getsockname(fd, (struct sockaddr *)&addr, (socklen_t *)sizeof(struct sockaddr)); |
||||
jfieldID fid; |
||||
jclass cls = (*env).GetObjectClass(ztaddr); |
||||
fid = (*env).GetFieldID( cls, "port", "I"); |
||||
(*env).SetIntField( ztaddr, fid, addr.sin_port); |
||||
fid = (*env).GetFieldID( cls,"_rawAddr", "J"); |
||||
(*env).SetLongField( ztaddr, fid,addr.sin_addr.s_addr); |
||||
return err; |
||||
} |
||||
|
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1getpeername(JNIEnv *env, jobject thisObj,
|
||||
jint fd, jobject ztaddr) { |
||||
struct sockaddr_in addr; |
||||
int err = zts_getpeername(fd, (struct sockaddr *)&addr, (socklen_t *)sizeof(struct sockaddr)); |
||||
jfieldID fid; |
||||
jclass cls = (*env).GetObjectClass( ztaddr); |
||||
fid = (*env).GetFieldID( cls, "port", "I"); |
||||
(*env).SetIntField( ztaddr, fid, addr.sin_port); |
||||
fid = (*env).GetFieldID( cls,"_rawAddr", "J"); |
||||
(*env).SetLongField( ztaddr, fid,addr.sin_addr.s_addr); |
||||
return err; |
||||
} |
||||
|
||||
JNIEXPORT jint JNICALL Java_zerotier_ZeroTier_ztjni_1fcntl(JNIEnv *env, jobject thisObj,
|
||||
jint fd, jint cmd, jint flags) { |
||||
return zts_fcntl(fd,cmd,flags); |
||||
} |
||||
|
||||
/****************************************************************************/ |
||||
/* Java JNI Wrapper for ZeroTier service controls */ |
||||
/****************************************************************************/ |
||||
|
||||
JNIEXPORT void JNICALL Java_zerotier_ZeroTier_ztjni_1start(JNIEnv *env, jobject thisObj, jstring path) { |
||||
if (path) { |
||||
zts_start(env->GetStringUTFChars(path, NULL)); |
||||
} |
||||
} |
||||
|
||||
JNIEXPORT void JNICALL Java_zerotier_ZeroTier_ztjni_1startjoin(JNIEnv *env, jobject thisObj, jstring path, jstring nwid) { |
||||
if (path) { |
||||
zts_startjoin(env->GetStringUTFChars(path, NULL), env->GetStringUTFChars(nwid, NULL)); |
||||
} |
||||
} |
||||
|
||||
JNIEXPORT void JNICALL Java_zerotier_ZeroTier_ztjni_1stop(JNIEnv *env, jobject thisObj) { |
||||
zts_stop(); |
||||
} |
||||
|
||||
JNIEXPORT jboolean JNICALL Java_zerotier_ZeroTier_ztjni_1running( |
||||
JNIEnv *env, jobject thisObj) |
||||
{ |
||||
return zts_running(); |
||||
} |
||||
|
||||
JNIEXPORT jstring JNICALL Java_zerotier_ZeroTier_ztjni_1homepath( |
||||
JNIEnv *env, jobject thisObj) |
||||
{ |
||||
// TODO: fix, should copy into given arg
|
||||
// return (*env).NewStringUTF(zts_get_homepath());
|
||||
return (*env).NewStringUTF(""); |
||||
} |
||||
|
||||
JNIEXPORT void JNICALL Java_zerotier_ZeroTier_ztjni_1join( |
||||
JNIEnv *env, jobject thisObj, jstring nwid) |
||||
{ |
||||
const char *nwidstr; |
||||
if (nwid) { |
||||
nwidstr = env->GetStringUTFChars(nwid, NULL); |
||||
zts_join(nwidstr); |
||||
} |
||||
} |
||||
|
||||
JNIEXPORT void JNICALL Java_zerotier_ZeroTier_ztjni_1leave( |
||||
JNIEnv *env, jobject thisObj, jstring nwid) |
||||
{ |
||||
const char *nwidstr; |
||||
if (nwid) { |
||||
nwidstr = env->GetStringUTFChars(nwid, NULL); |
||||
zts_leave(nwidstr); |
||||
} |
||||
} |
||||
|
||||
JNIEXPORT jobject JNICALL Java_zerotier_ZeroTier_ztjni_1get_1ipv4_1address( |
||||
JNIEnv *env, jobject thisObj, jstring nwid) |
||||
{ |
||||
const char *nwid_str = env->GetStringUTFChars(nwid, NULL); |
||||
char address_string[INET_ADDRSTRLEN]; |
||||
memset(address_string, 0, INET_ADDRSTRLEN); |
||||
zts_get_ipv4_address(nwid_str, address_string, INET_ADDRSTRLEN); |
||||
jclass clazz = (*env).FindClass("java/util/ArrayList"); |
||||
jobject addresses = (*env).NewObject(clazz, (*env).GetMethodID(clazz, "<init>", "()V")); |
||||
jstring _str = (*env).NewStringUTF(address_string); |
||||
env->CallBooleanMethod(addresses, env->GetMethodID(clazz, "add", "(Ljava/lang/Object;)Z"), _str); |
||||
return addresses; |
||||
} |
||||
|
||||
JNIEXPORT jobject JNICALL Java_zerotier_ZeroTier_ztjni_1get_1ipv6_1address( |
||||
JNIEnv *env, jobject thisObj, jstring nwid) |
||||
{ |
||||
const char *nwid_str = env->GetStringUTFChars(nwid, NULL); |
||||
char address_string[INET6_ADDRSTRLEN]; |
||||
memset(address_string, 0, INET6_ADDRSTRLEN); |
||||
zts_get_ipv6_address(nwid_str, address_string, INET6_ADDRSTRLEN); |
||||
jclass clazz = (*env).FindClass("java/util/ArrayList"); |
||||
jobject addresses = (*env).NewObject(clazz, (*env).GetMethodID(clazz, "<init>", "()V")); |
||||
jstring _str = (*env).NewStringUTF(address_string); |
||||
env->CallBooleanMethod(addresses, env->GetMethodID(clazz, "add", "(Ljava/lang/Object;)Z"), _str); |
||||
return addresses; |
||||
} |
||||
|
||||
JNIEXPORT jint Java_zerotier_ZeroTier_ztjni_1get_1device_1id() |
||||
{ |
||||
return zts_get_device_id(NULL); // TODO
|
||||
} |
||||
} |
||||
|
||||
#endif // SDK_JNI
|
||||
|
||||
#ifdef __cplusplus |
||||
} |
||||
#endif |
||||
Loading…
Reference in new issue