|
@@ -0,0 +1,218 @@
|
|
|
+#![cfg(target_os = "windows")]
|
|
|
+
|
|
|
+use crate::config::Config;
|
|
|
+use crate::utils::dirs;
|
|
|
+use anyhow::{bail, Context, Result};
|
|
|
+use deelevate::{PrivilegeLevel, Token};
|
|
|
+use runas::Command as RunasCommand;
|
|
|
+use serde::{Deserialize, Serialize};
|
|
|
+use std::collections::HashMap;
|
|
|
+use std::os::windows::process::CommandExt;
|
|
|
+use std::time::Duration;
|
|
|
+use std::{env::current_exe, process::Command as StdCommand};
|
|
|
+use tokio::time::sleep;
|
|
|
+
|
|
|
+const SERVICE_NAME: &str = "clash_verge_service";
|
|
|
+
|
|
|
+const SERVICE_URL: &str = "http://127.0.0.1:33211";
|
|
|
+
|
|
|
+#[derive(Debug, Deserialize, Serialize, Clone)]
|
|
|
+pub struct ResponseBody {
|
|
|
+ pub core_type: Option<String>,
|
|
|
+ pub bin_path: String,
|
|
|
+ pub config_dir: String,
|
|
|
+ pub log_file: String,
|
|
|
+}
|
|
|
+
|
|
|
+#[derive(Debug, Deserialize, Serialize, Clone)]
|
|
|
+pub struct JsonResponse {
|
|
|
+ pub code: u64,
|
|
|
+ pub msg: String,
|
|
|
+ pub data: Option<ResponseBody>,
|
|
|
+}
|
|
|
+
|
|
|
+/// Install the Clash Verge Service
|
|
|
+/// 该函数应该在协程或者线程中执行,避免UAC弹窗阻塞主线程
|
|
|
+pub async fn install_service() -> Result<()> {
|
|
|
+ let binary_path = dirs::service_path();
|
|
|
+ let install_path = binary_path.with_file_name("install-service.exe");
|
|
|
+
|
|
|
+ if !install_path.exists() {
|
|
|
+ bail!("installer exe not found");
|
|
|
+ }
|
|
|
+
|
|
|
+ let token = Token::with_current_process()?;
|
|
|
+ let level = token.privilege_level()?;
|
|
|
+
|
|
|
+ let status = match level {
|
|
|
+ PrivilegeLevel::NotPrivileged => RunasCommand::new(install_path).status()?,
|
|
|
+ _ => StdCommand::new(install_path)
|
|
|
+ .creation_flags(0x08000000)
|
|
|
+ .status()?,
|
|
|
+ };
|
|
|
+
|
|
|
+ if !status.success() {
|
|
|
+ bail!(
|
|
|
+ "failed to install service with status {}",
|
|
|
+ status.code().unwrap()
|
|
|
+ );
|
|
|
+ }
|
|
|
+
|
|
|
+ Ok(())
|
|
|
+}
|
|
|
+
|
|
|
+/// Uninstall the Clash Verge Service
|
|
|
+/// 该函数应该在协程或者线程中执行,避免UAC弹窗阻塞主线程
|
|
|
+pub async fn uninstall_service() -> Result<()> {
|
|
|
+ let binary_path = dirs::service_path();
|
|
|
+ let uninstall_path = binary_path.with_file_name("uninstall-service.exe");
|
|
|
+
|
|
|
+ if !uninstall_path.exists() {
|
|
|
+ bail!("uninstaller exe not found");
|
|
|
+ }
|
|
|
+
|
|
|
+ let token = Token::with_current_process()?;
|
|
|
+ let level = token.privilege_level()?;
|
|
|
+
|
|
|
+ let status = match level {
|
|
|
+ PrivilegeLevel::NotPrivileged => RunasCommand::new(uninstall_path).status()?,
|
|
|
+ _ => StdCommand::new(uninstall_path)
|
|
|
+ .creation_flags(0x08000000)
|
|
|
+ .status()?,
|
|
|
+ };
|
|
|
+
|
|
|
+ if !status.success() {
|
|
|
+ bail!(
|
|
|
+ "failed to uninstall service with status {}",
|
|
|
+ status.code().unwrap()
|
|
|
+ );
|
|
|
+ }
|
|
|
+
|
|
|
+ Ok(())
|
|
|
+}
|
|
|
+
|
|
|
+/// start service
|
|
|
+/// 该函数应该在协程或者线程中执行,避免UAC弹窗阻塞主线程
|
|
|
+#[deprecated]
|
|
|
+pub async fn start_service() -> Result<()> {
|
|
|
+ let token = Token::with_current_process()?;
|
|
|
+ let level = token.privilege_level()?;
|
|
|
+
|
|
|
+ let args = ["start", SERVICE_NAME];
|
|
|
+
|
|
|
+ let status = match level {
|
|
|
+ PrivilegeLevel::NotPrivileged => RunasCommand::new("sc").args(&args).status()?,
|
|
|
+ _ => StdCommand::new("sc").args(&args).status()?,
|
|
|
+ };
|
|
|
+
|
|
|
+ match status.success() {
|
|
|
+ true => Ok(()),
|
|
|
+ false => bail!(
|
|
|
+ "failed to start service with status {}",
|
|
|
+ status.code().unwrap()
|
|
|
+ ),
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+/// stop service
|
|
|
+pub async fn stop_service() -> Result<()> {
|
|
|
+ let url = format!("{SERVICE_URL}/stop_service");
|
|
|
+ let res = reqwest::ClientBuilder::new()
|
|
|
+ .no_proxy()
|
|
|
+ .build()?
|
|
|
+ .post(url)
|
|
|
+ .send()
|
|
|
+ .await?
|
|
|
+ .json::<JsonResponse>()
|
|
|
+ .await
|
|
|
+ .context("failed to connect to the Clash Verge Service")?;
|
|
|
+
|
|
|
+ if res.code != 0 {
|
|
|
+ bail!(res.msg);
|
|
|
+ }
|
|
|
+
|
|
|
+ Ok(())
|
|
|
+}
|
|
|
+
|
|
|
+/// check the windows service status
|
|
|
+pub async fn check_service() -> Result<JsonResponse> {
|
|
|
+ let url = format!("{SERVICE_URL}/get_clash");
|
|
|
+ let response = reqwest::ClientBuilder::new()
|
|
|
+ .no_proxy()
|
|
|
+ .build()?
|
|
|
+ .get(url)
|
|
|
+ .send()
|
|
|
+ .await?
|
|
|
+ .json::<JsonResponse>()
|
|
|
+ .await
|
|
|
+ .context("failed to connect to the Clash Verge Service")?;
|
|
|
+
|
|
|
+ Ok(response)
|
|
|
+}
|
|
|
+
|
|
|
+/// start the clash by service
|
|
|
+pub(super) async fn run_core_by_service() -> Result<()> {
|
|
|
+ let status = check_service().await?;
|
|
|
+
|
|
|
+ if status.code == 0 {
|
|
|
+ stop_core_by_service().await?;
|
|
|
+ sleep(Duration::from_secs(1)).await;
|
|
|
+ }
|
|
|
+
|
|
|
+ let clash_core = { Config::verge().latest().clash_core.clone() };
|
|
|
+ let clash_core = clash_core.unwrap_or("clash".into());
|
|
|
+
|
|
|
+ let clash_bin = format!("{clash_core}.exe");
|
|
|
+ let bin_path = current_exe()?.with_file_name(clash_bin);
|
|
|
+ let bin_path = dirs::path_to_str(&bin_path)?;
|
|
|
+
|
|
|
+ let config_dir = dirs::app_home_dir();
|
|
|
+ let config_dir = dirs::path_to_str(&config_dir)?;
|
|
|
+
|
|
|
+ let log_path = dirs::service_log_file();
|
|
|
+ let log_path = dirs::path_to_str(&log_path)?;
|
|
|
+
|
|
|
+ let mut map = HashMap::new();
|
|
|
+ map.insert("core_type", clash_core.as_str());
|
|
|
+ map.insert("bin_path", bin_path);
|
|
|
+ map.insert("config_dir", config_dir);
|
|
|
+ map.insert("log_file", log_path);
|
|
|
+
|
|
|
+ let url = format!("{SERVICE_URL}/start_clash");
|
|
|
+ let res = reqwest::ClientBuilder::new()
|
|
|
+ .no_proxy()
|
|
|
+ .build()?
|
|
|
+ .post(url)
|
|
|
+ .json(&map)
|
|
|
+ .send()
|
|
|
+ .await?
|
|
|
+ .json::<JsonResponse>()
|
|
|
+ .await
|
|
|
+ .context("failed to connect to the Clash Verge Service")?;
|
|
|
+
|
|
|
+ if res.code != 0 {
|
|
|
+ bail!(res.msg);
|
|
|
+ }
|
|
|
+
|
|
|
+ Ok(())
|
|
|
+}
|
|
|
+
|
|
|
+/// stop the clash by service
|
|
|
+pub(super) async fn stop_core_by_service() -> Result<()> {
|
|
|
+ let url = format!("{SERVICE_URL}/stop_clash");
|
|
|
+ let res = reqwest::ClientBuilder::new()
|
|
|
+ .no_proxy()
|
|
|
+ .build()?
|
|
|
+ .post(url)
|
|
|
+ .send()
|
|
|
+ .await?
|
|
|
+ .json::<JsonResponse>()
|
|
|
+ .await
|
|
|
+ .context("failed to connect to the Clash Verge Service")?;
|
|
|
+
|
|
|
+ if res.code != 0 {
|
|
|
+ bail!(res.msg);
|
|
|
+ }
|
|
|
+
|
|
|
+ Ok(())
|
|
|
+}
|