SDK_JNI

2017-09-29 17:05:21 -07:00
parent 6a5c73eec2
commit 0319f8a894
3 changed files with 288 additions and 4 deletions
--- a/src/libzt.cpp
+++ b/src/libzt.cpp
@@ -496,6 +496,196 @@ int zts_del_dns_nameserver(struct sockaddr *addr)
 	return err;
 }

+
+/****************************************************************************/
+/* SDK Socket API (Java Native Interface JNI)                               */
+/* JNI naming convention: Java_PACKAGENAME_CLASSNAME_METHODNAME             */
+/****************************************************************************/
+
+#if defined(SDK_JNI)
+
+namespace ZeroTier {
+
+	#include <jni.h>
+
+	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);
+	}
+}
+#endif // SDK_JNI
+
 #ifdef __cplusplus
 }
 #endif