vnt/vnt-jni/src/vnt_util.rs

use std::net::ToSocketAddrs;
use std::ptr;
use std::str::FromStr;

use jni::errors::Error;
use jni::objects::{JClass, JObject, JString, JValue};
#[cfg(not(target_os = "android"))]
use jni::sys::jboolean;
use jni::sys::{jint, jlong, jobject};
use jni::JNIEnv;
use vnt::channel::punch::PunchModel;
use vnt::cipher::CipherModel;
use vnt::core::sync::VntUtilSync;
use vnt::core::Config;
use vnt::handle::registration_handler::{RegResponse, ReqEnum};
#[cfg(not(target_os = "android"))]
use vnt::tun_tap_device::DriverInfo;

fn to_string_not_null(
    env: &mut JNIEnv,
    config: &JObject,
    name: &'static str,
) -> Result<String, Error> {
    let value = env.get_field(config, name, "Ljava/lang/String;")?.l()?;
    if value.is_null() {
        env.throw_new("java/lang/NullPointerException", name)
            .expect("throw");
        return Err(Error::NullPtr(name));
    }
    let binding = JString::from(value);
    let value = env.get_string(binding.as_ref())?;
    match value.to_str() {
        Ok(value) => Ok(value.to_string()),
        Err(_) => {
            env.throw_new("java/lang/RuntimeException", "not utf-8")
                .expect("throw");
            return Err(Error::JavaException);
        }
    }
}

fn to_string(env: &mut JNIEnv, config: &JObject, name: &str) -> Result<Option<String>, Error> {
    let value = env.get_field(config, name, "Ljava/lang/String;")?.l()?;
    if value.is_null() {
        return Ok(None);
    }
    let tmp = JString::from(value);
    let value = env.get_string(tmp.as_ref())?;
    match value.to_str() {
        Ok(value) => Ok(Some(value.to_string())),
        Err(_) => {
            env.throw_new("java/lang/RuntimeException", "not utf-8")
                .expect("throw");
            return Err(Error::JavaException);
        }
    }
}

fn new_sync(env: &mut JNIEnv, config: JObject) -> Result<VntUtilSync, Error> {
    let token = to_string_not_null(env, &config, "token")?;
    let name = to_string_not_null(env, &config, "name")?;
    let device_id = to_string_not_null(env, &config, "deviceId")?;
    let password = to_string(env, &config, "password")?;
    let server_address_str = to_string_not_null(env, &config, "server")?;
    let stun_server_str = to_string_not_null(env, &config, "stunServer")?;
    let cipher_model = to_string_not_null(env, &config, "cipherModel")?;
    let tcp = env.get_field(&config, "tcp", "Z")?.z()?;
    let finger = env.get_field(&config, "finger", "Z")?.z()?;
    let in_ips = to_string(env, &config, "inIps")?;
    let out_ips = to_string(env, &config, "outIps")?;
    let port = env.get_field(&config, "port", "I")?.i()? as u16;
    let in_ips = if let Some(in_ips) = in_ips {
        let in_ips: Vec<&str> = in_ips.split("\n").collect();
        let in_ips = in_ips.iter().map(|v| v.to_string()).collect();
        match common::args_parse::ips_parse(&in_ips) {
            Ok(in_ips) => in_ips,
            Err(e) => {
                env.throw_new("java/lang/RuntimeException", format!("in_ips {}", e))
                    .expect("throw");
                return Err(Error::JavaException);
            }
        }
    } else {
        vec![]
    };
    let out_ips = if let Some(out_ips) = out_ips {
        let out_ips: Vec<&str> = out_ips.split("\n").collect();
        let out_ips = out_ips.iter().map(|v| v.to_string()).collect();
        match common::args_parse::out_ips_parse(&out_ips) {
            Ok(out_ips) => out_ips,
            Err(e) => {
                env.throw_new("java/lang/RuntimeException", format!("out_ips {}", e))
                    .expect("throw");
                return Err(Error::JavaException);
            }
        }
    } else {
        vec![]
    };

    let server_address = match server_address_str.to_socket_addrs() {
        Ok(mut rs) => {
            if let Some(addr) = rs.next() {
                addr
            } else {
                env.throw_new("java/lang/RuntimeException", "server address err")
                    .expect("throw");
                return Err(Error::JavaException);
            }
        }
        Err(e) => {
            env.throw_new(
                "java/lang/RuntimeException",
                format!("server address {}", e),
            )
            .expect("throw");
            return Err(Error::JavaException);
        }
    };
    let cipher_model = match CipherModel::from_str(&cipher_model) {
        Ok(cipher_model) => cipher_model,
        Err(e) => {
            env.throw_new("java/lang/RuntimeException", format!("cipher_model {}", e))
                .expect("throw");
            return Err(Error::JavaException);
        }
    };
    let mut stun_server = Vec::new();
    for addr in stun_server_str.split(",") {
        stun_server.push(addr.trim().to_string());
    }
    let config = Config::new(
        false,
        token,
        device_id,
        name,
        server_address,
        server_address_str,
        stun_server,
        in_ips,
        out_ips,
        password,
        false,
        None,
        tcp,
        None,
        false,
        false,
        false,
        1,
        cipher_model,
        finger,
        PunchModel::All,
        port,
    );
    match VntUtilSync::new(config) {
        Ok(vnt_util) => Ok(vnt_util),
        Err(e) => {
            env.throw_new(
                "java/lang/RuntimeException",
                format!("vnt start error {}", e),
            )
            .expect("throw");
            return Err(Error::JavaException);
        }
    }
}

#[no_mangle]
pub unsafe extern "C" fn Java_top_wherewego_vnt_jni_VntUtil_new0(
    mut env: JNIEnv,
    _class: JClass,
    config: JObject,
) -> jlong {
    match new_sync(&mut env, config) {
        Ok(vnt_util) => {
            let ptr = Box::into_raw(Box::new(vnt_util));
            return ptr as jlong;
        }
        Err(_) => {}
    }
    return 0;
}

#[no_mangle]
pub unsafe extern "C" fn Java_top_wherewego_vnt_jni_VntUtil_connect0(
    mut env: JNIEnv,
    _class: JClass,
    raw_vnt_util: jlong,
) {
    let raw_vnt_util = raw_vnt_util as *mut VntUtilSync;
    match (&mut *raw_vnt_util).connect() {
        Ok(_) => {}
        Err(e) => {
            env.throw_new(
                "java/lang/RuntimeException",
                format!("vnt connect error {}", e),
            )
            .expect("throw");
        }
    }
}

#[no_mangle]
pub unsafe extern "C" fn Java_top_wherewego_vnt_jni_VntUtil_register0(
    mut env: JNIEnv,
    _class: JClass,
    raw_vnt_util: jlong,
) -> jobject {
    let raw_vnt_util = raw_vnt_util as *mut VntUtilSync;
    match (&mut *raw_vnt_util).register() {
        Ok(response) => match reg_response(&mut env, response) {
            Ok(res) => {
                return res;
            }
            Err(e) => {
                env.throw(format!("vnt register error {}", e))
                    .expect("throw");
            }
        },
        Err(e) => match e {
            ReqEnum::TokenError => {
                env.throw_new(
                    "top/wherewego/vnt/jni/exception/TokenErrorException",
                    "TokenError",
                )
                .expect("throw");
            }
            ReqEnum::AddressExhausted => {
                env.throw_new(
                    "top/wherewego/vnt/jni/exception/AddressExhaustedException",
                    "AddressExhausted",
                )
                .expect("throw");
            }
            ReqEnum::Timeout => {
                env.throw_new(
                    "top/wherewego/vnt/jni/exception/TimeoutException",
                    "Timeout",
                )
                .expect("throw");
            }
            ReqEnum::ServerError(str) => {
                env.throw_new(
                    "java/lang/RuntimeException",
                    format!("vnt register error {}", str),
                )
                .expect("throw");
            }
            ReqEnum::Other(str) => {
                env.throw_new(
                    "java/lang/RuntimeException",
                    format!("vnt register error {}", str),
                )
                .expect("throw");
            }
            ReqEnum::IpAlreadyExists => {
                env.throw_new(
                    "top/wherewego/vnt/jni/exception/IpAlreadyExistsException",
                    "IpAlreadyExists",
                )
                .expect("throw");
            }
            ReqEnum::InvalidIp => {
                env.throw_new(
                    "top/wherewego/vnt/jni/exception/InvalidIpException",
                    "InvalidIp",
                )
                .expect("throw");
            }
        },
    }
    return ptr::null_mut();
}

#[cfg(target_os = "android")]
#[no_mangle]
pub unsafe extern "C" fn Java_top_wherewego_vnt_jni_VntUtil_createIface0(
    _env: JNIEnv,
    _class: JClass,
    raw_vnt_util: jlong,
    fd: jint,
) {
    let raw_vnt_util = raw_vnt_util as *mut VntUtilSync;

    (&mut *raw_vnt_util).create_iface(fd as i32);
}

#[cfg(not(target_os = "android"))]
#[no_mangle]
pub unsafe extern "C" fn Java_top_wherewego_vnt_jni_VntUtil_createIface0(
    mut env: JNIEnv,
    _class: JClass,
    raw_vnt_util: jlong,
) -> jobject {
    let raw_vnt_util = raw_vnt_util as *mut VntUtilSync;
    let rs = (&mut *raw_vnt_util).create_iface();
    match rs {
        Ok(driver_info) => match driver_info_e(&mut env, driver_info) {
            Ok(res) => {
                return res;
            }
            Err(e) => {
                env.throw(format!("vnt create iface  error {}", e))
                    .expect("throw");
            }
        },
        Err(e) => {
            env.throw_new(
                "java/lang/RuntimeException",
                format!("vnt create iface error {}", e),
            )
            .expect("throw");
        }
    }
    return ptr::null_mut();
}

#[no_mangle]
pub unsafe extern "C" fn Java_top_wherewego_vnt_jni_VntUtil_build0(
    mut env: JNIEnv,
    _class: JClass,
    raw_vnt_util: jlong,
) -> jlong {
    let raw_vnt_util = Box::from_raw(raw_vnt_util as *mut VntUtilSync);
    match raw_vnt_util.build() {
        Ok(rs) => {
            return Box::into_raw(Box::new(rs)) as jlong;
        }
        Err(e) => {
            env.throw_new(
                "java/lang/RuntimeException",
                format!("vnt start error:{:?}", e),
            )
            .expect("throw");
        }
    }
    return 0;
}

fn reg_response(env: &mut JNIEnv, response: RegResponse) -> Result<jobject, Error> {
    let virtual_ip = u32::from(response.virtual_ip);
    let virtual_gateway = u32::from(response.virtual_gateway);
    let virtual_netmask = u32::from(response.virtual_netmask);
    let response = env.new_object(
        "top/wherewego/vnt/jni/RegResponse",
        "(III)V",
        &[
            JValue::Int(virtual_ip as jint),
            JValue::Int(virtual_gateway as jint),
            JValue::Int(virtual_netmask as jint),
        ],
    )?;
    Ok(response.into_raw())
}

#[cfg(not(target_os = "android"))]
fn driver_info_e(env: &mut JNIEnv, driver_info: DriverInfo) -> Result<jobject, Error> {
    let is_tun = driver_info.device_type.is_tun();
    let name = driver_info.name;
    let version = driver_info.version;
    let mac = driver_info.mac.unwrap_or(String::new());
    let response = env.new_object(
        "top/wherewego/vnt/jni/DriverInfo",
        "(ZLjava/lang/String;Ljava/lang/String;Ljava/lang/String;)V",
        &[
            JValue::Bool(is_tun as jboolean),
            JValue::Object(&env.new_string(name)?.into()),
            JValue::Object(&env.new_string(version)?.into()),
            JValue::Object(&env.new_string(mac)?.into()),
        ],
    )?;
    Ok(response.into_raw())
}