feat: disable running with admin permission and check service mode

This commit is contained in:
MystiPanda
2024-06-14 22:35:43 +08:00
parent 1b8c4cb832
commit a0f9fb90ee
4 changed files with 94 additions and 79 deletions

View File

@@ -88,16 +88,37 @@ pub fn toggle_system_proxy() {
pub fn toggle_tun_mode() {
let enable = Config::verge().data().enable_tun_mode;
let enable = enable.unwrap_or(false);
tauri::async_runtime::spawn(async move {
match patch_verge(IVerge {
enable_tun_mode: Some(!enable),
..IVerge::default()
})
.await
{
Ok(_) => handle::Handle::refresh_verge(),
Err(err) => log::error!(target: "app", "{err}"),
if !enable {
if let Ok(res) = service::check_service().await {
if res.code == 0 {
match patch_verge(IVerge {
enable_tun_mode: Some(!enable),
..IVerge::default()
})
.await
{
Ok(_) => handle::Handle::refresh_verge(),
Err(err) => log::error!(target: "app", "{err}"),
}
return;
}
}
tauri::api::dialog::message(
None::<&tauri::Window>,
"Please install and enable service mode",
"Service mode is required for Tun mode",
);
} else {
match patch_verge(IVerge {
enable_tun_mode: Some(!enable),
..IVerge::default()
})
.await
{
Ok(_) => handle::Handle::refresh_verge(),
Err(err) => log::error!(target: "app", "{err}"),
}
}
});
}
@@ -403,3 +424,19 @@ pub async fn test_delay(url: String) -> Result<u32> {
}
}
}
pub fn check_permission() -> Result<()> {
#[cfg(target_os = "windows")]
{
let hklm = winreg::RegKey::predef(winreg::enums::HKEY_LOCAL_MACHINE);
if let Ok(reg) = hklm.open_subkey_with_flags(
"SOFTWARE\\WOW6432Node\\Microsoft\\Windows\\CurrentVersion\\Run",
winreg::enums::KEY_SET_VALUE,
) {
reg.delete_value("Clash Verge").unwrap_or_default();
return Ok(());
}
}
Err(anyhow::anyhow!("permission denied"))
}

View File

@@ -19,6 +19,16 @@ fn main() -> std::io::Result<()> {
println!("app exists");
return Ok(());
}
// 权限检测
if feat::check_permission().is_ok() {
println!("please do not run with admin permission");
tauri::api::dialog::blocking::message(
None::<&tauri::Window>,
"Please do not run with admin permission",
"If you want to use Tun mode, please enable service mode instead",
);
return Ok(());
}
#[cfg(target_os = "linux")]
std::env::set_var("WEBKIT_DISABLE_DMABUF_RENDERER", "1");