use std::{os::raw::c_void, sync::LazyLock};
use std::path::MAIN_SEPARATOR;
use java_locator::{get_jvm_dyn_lib_file_name, locate_jvm_dyn_library};
use jni_sys::{jclass, jint, jsize, JNIEnv, JavaVM};
use crate::{errors, utils};
#[allow(non_snake_case)]
type JNIGetCreatedJavaVMs =
unsafe extern "system" fn(vmBuf: *mut *mut JavaVM, bufLen: jsize, nVMs: *mut jsize) -> jint;
type JNICreateJavaVM = unsafe extern "system" fn(
pvm: *mut *mut JavaVM,
penv: *mut *mut c_void,
args: *mut c_void,
) -> jint;
static JVM_LIB: LazyLock<libloading::Library> = LazyLock::new(|| {
let full_path = format!(
"{}{}{}",
locate_jvm_dyn_library().expect("Could not find the jvm dynamic library"),
MAIN_SEPARATOR,
get_jvm_dyn_lib_file_name()
);
unsafe {
libloading::Library::new(full_path).expect("Could not load the jvm dynamic library")
}
});
static GET_CREATED_JVMS: LazyLock<libloading::Symbol<'static, JNIGetCreatedJavaVMs>> = LazyLock::new(|| {
unsafe {
JVM_LIB
.get(b"JNI_GetCreatedJavaVMs")
.expect("Could not find symbol: JNI_GetCreatedJavaVMs")
}
});
static CREATE_JVM: LazyLock<libloading::Symbol<'static, JNICreateJavaVM>> = LazyLock::new(|| {
unsafe {
JVM_LIB
.get(b"JNI_CreateJavaVM")
.expect("Could not find symbol: JNI_CreateJavaVM")
}
});
pub(crate) fn get_created_java_vms(
vm_buf: &mut Vec<*mut JavaVM>,
buf_len: jsize,
n_vms: *mut jsize,
) -> jint {
unsafe { GET_CREATED_JVMS(vm_buf.as_mut_ptr(), buf_len, n_vms) }
}
pub(crate) fn create_java_vm(
jvm: *mut *mut JavaVM,
penv: *mut *mut c_void,
args: *mut c_void,
) -> jint {
unsafe { CREATE_JVM(jvm, penv, args) }
}
pub(crate) fn find_class(env: *mut JNIEnv, classname: &str) -> errors::Result<jclass> {
unsafe {
let cstr = utils::to_c_string(classname);
let fc = (**env).v1_6.FindClass;
let jc = (fc)(env, cstr);
utils::drop_c_string(cstr);
Ok(jc)
}
}