126 lines
4.0 KiB
Rust
126 lines
4.0 KiB
Rust
use hbb_common::{bail, ResultType};
|
|
use std::sync::{Arc, Mutex};
|
|
|
|
const LIB_NAME_VIRTUAL_DISPLAY: &str = "dylib_virtual_display";
|
|
|
|
lazy_static::lazy_static! {
|
|
static ref LIB_VIRTUAL_DISPLAY: Arc<Mutex<Result<libloading::Library, libloading::Error>>> = {
|
|
Arc::new(Mutex::new(unsafe { libloading::Library::new(get_lib_name()) }))
|
|
};
|
|
}
|
|
|
|
#[cfg(target_os = "windows")]
|
|
fn get_lib_name() -> String {
|
|
format!("{}.dll", LIB_NAME_VIRTUAL_DISPLAY)
|
|
}
|
|
|
|
#[cfg(target_os = "linux")]
|
|
fn get_lib_name() -> String {
|
|
format!("lib{}.so", LIB_NAME_VIRTUAL_DISPLAY)
|
|
}
|
|
|
|
#[cfg(target_os = "macos")]
|
|
fn get_lib_name() -> String {
|
|
format!("lib{}.dylib", LIB_NAME_VIRTUAL_DISPLAY)
|
|
}
|
|
|
|
fn try_reload_lib() {
|
|
let mut lock = LIB_VIRTUAL_DISPLAY.lock().unwrap();
|
|
if lock.is_err() {
|
|
*lock = unsafe { libloading::Library::new(get_lib_name()) };
|
|
}
|
|
}
|
|
|
|
#[cfg(windows)]
|
|
pub fn get_dirver_install_path() -> ResultType<&'static str> {
|
|
try_reload_lib();
|
|
match &*LIB_VIRTUAL_DISPLAY.lock().unwrap() {
|
|
Ok(lib) => unsafe {
|
|
match lib.get::<libloading::Symbol<fn() -> &'static str>>(b"get_dirver_install_path") {
|
|
Ok(func) => Ok(func()),
|
|
Err(e) => bail!("Failed to load func get_dirver_install_path, {}", e),
|
|
}
|
|
},
|
|
Err(e) => bail!("Failed to load library {}, {}", LIB_NAME_VIRTUAL_DISPLAY, e),
|
|
}
|
|
}
|
|
|
|
pub fn is_device_created() -> bool {
|
|
try_reload_lib();
|
|
match &*LIB_VIRTUAL_DISPLAY.lock().unwrap() {
|
|
Ok(lib) => unsafe {
|
|
match lib.get::<libloading::Symbol<fn() -> bool>>(b"is_device_created") {
|
|
Ok(func) => func(),
|
|
Err(..) => false,
|
|
}
|
|
},
|
|
Err(..) => false,
|
|
}
|
|
}
|
|
|
|
pub fn close_device() {
|
|
try_reload_lib();
|
|
match &*LIB_VIRTUAL_DISPLAY.lock().unwrap() {
|
|
Ok(lib) => unsafe {
|
|
match lib.get::<libloading::Symbol<fn()>>(b"close_device") {
|
|
Ok(func) => func(),
|
|
Err(..) => {}
|
|
}
|
|
},
|
|
Err(..) => {}
|
|
}
|
|
}
|
|
|
|
macro_rules! def_func_result {
|
|
($func:ident, $name: tt) => {
|
|
pub fn $func() -> ResultType<()> {
|
|
try_reload_lib();
|
|
match &*LIB_VIRTUAL_DISPLAY.lock().unwrap() {
|
|
Ok(lib) => unsafe {
|
|
match lib.get::<libloading::Symbol<fn() -> ResultType<()>>>($name.as_bytes()) {
|
|
Ok(func) => func(),
|
|
Err(e) => bail!("Failed to load func {}, {}", $name, e),
|
|
}
|
|
},
|
|
Err(e) => bail!("Failed to load library {}, {}", LIB_NAME_VIRTUAL_DISPLAY, e),
|
|
}
|
|
}
|
|
};
|
|
}
|
|
|
|
pub fn install_update_driver(reboot_required: &mut bool) -> ResultType<()> {
|
|
try_reload_lib();
|
|
match &*LIB_VIRTUAL_DISPLAY.lock().unwrap() {
|
|
Ok(lib) => unsafe {
|
|
match lib.get::<libloading::Symbol<fn(&mut bool) -> ResultType<()>>>(
|
|
b"install_update_driver",
|
|
) {
|
|
Ok(func) => func(reboot_required),
|
|
Err(e) => bail!("Failed to load func install_update_driver, {}", e),
|
|
}
|
|
},
|
|
Err(e) => bail!("Failed to load library {}, {}", LIB_NAME_VIRTUAL_DISPLAY, e),
|
|
}
|
|
}
|
|
|
|
pub fn uninstall_driver(reboot_required: &mut bool) -> ResultType<()> {
|
|
try_reload_lib();
|
|
match &*LIB_VIRTUAL_DISPLAY.lock().unwrap() {
|
|
Ok(lib) => unsafe {
|
|
match lib
|
|
.get::<libloading::Symbol<fn(&mut bool) -> ResultType<()>>>(b"uninstall_driver")
|
|
{
|
|
Ok(func) => func(reboot_required),
|
|
Err(e) => bail!("Failed to load func uninstall_driver, {}", e),
|
|
}
|
|
},
|
|
Err(e) => bail!("Failed to load library {}, {}", LIB_NAME_VIRTUAL_DISPLAY, e),
|
|
}
|
|
}
|
|
|
|
def_func_result!(download_driver, "download_driver");
|
|
def_func_result!(create_device, "create_device");
|
|
def_func_result!(plug_in_monitor, "plug_in_monitor");
|
|
def_func_result!(plug_out_monitor, "plug_out_monitor");
|
|
def_func_result!(update_monitor_modes, "update_monitor_modes");
|