#[cfg(target_os = "macos")] use crate::core::tray::Tray; use crate::{ config::*, core::{handle, service}, log_err, module::mihomo::MihomoManager, utils::{dirs, help}, }; use anyhow::{bail, Result}; use fs2::FileExt; use once_cell::sync::OnceCell; use std::{path::PathBuf, sync::Arc, time::Duration}; use tauri_plugin_shell::ShellExt; use tokio::{sync::Mutex, time::sleep}; use super::service::is_service_running; #[derive(Debug)] pub struct CoreManager { running: Arc>, } /// 内核运行模式 #[derive(Debug, Clone, serde::Serialize)] pub enum RunningMode { /// 服务模式运行 Service, /// Sidecar模式运行 Sidecar, /// 未运行 NotRunning, } impl CoreManager { pub fn global() -> &'static CoreManager { static CORE_MANAGER: OnceCell = OnceCell::new(); CORE_MANAGER.get_or_init(|| CoreManager { running: Arc::new(Mutex::new(false)), }) } pub async fn init(&self) -> Result<()> { log::trace!("run core start"); // 启动clash log_err!(Self::global().start_core().await); log::trace!("run core end"); Ok(()) } /// 停止核心运行 pub async fn stop_core(&self) -> Result<()> { let mut running = self.running.lock().await; if !*running { println!("[停止内核] 内核未运行"); log::debug!("core is not running"); return Ok(()); } println!("[停止内核] 开始停止内核"); // 关闭tun模式 // Create a JSON object to disable TUN mode let disable = serde_json::json!({ "tun": { "enable": false } }); println!("[停止内核] 禁用TUN模式"); log::debug!(target: "app", "disable tun mode"); log_err!(MihomoManager::global().patch_configs(disable).await); // 服务模式 if service::check_service().await.is_ok() { println!("[停止内核] 尝试通过服务停止内核"); log::info!(target: "app", "stop the core by service"); match service::stop_core_by_service().await { Ok(_) => { println!("[停止内核] 服务模式下内核停止成功"); log::info!(target: "app", "core stopped successfully by service"); } Err(err) => { println!("[停止内核] 服务模式下停止内核失败: {}", err); println!("[停止内核] 尝试停止可能的sidecar进程"); log::warn!(target: "app", "failed to stop core by service: {}", err); // 服务停止失败,尝试停止可能的sidecar进程 self.stop_sidecar_process(); } } } else { // 如果没有使用服务,尝试停止sidecar进程 println!("[停止内核] 服务不可用,尝试停止sidecar进程"); self.stop_sidecar_process(); } // 释放文件锁 println!("[停止内核] 尝试释放文件锁"); if let Some(_) = handle::Handle::global().release_core_lock() { println!("[停止内核] 文件锁释放成功"); log::info!(target: "app", "released core lock file"); } else { println!("[停止内核] 没有文件锁需要释放"); } *running = false; println!("[停止内核] 内核停止完成"); Ok(()) } /// 停止通过sidecar启动的进程 fn stop_sidecar_process(&self) { if let Some(process) = handle::Handle::global().take_core_process() { println!("[停止sidecar] 发现sidecar进程,准备停止"); log::info!(target: "app", "stopping core process in sidecar mode"); // 尝试获取进程ID let pid = process.pid(); println!("[停止sidecar] 进程PID: {}", pid); // 尝试终止进程 if let Err(e) = process.kill() { println!("[停止sidecar] 终止sidecar进程失败: {}", e); log::warn!(target: "app", "failed to kill core process: {}", e); } else { println!("[停止sidecar] sidecar进程已成功终止"); log::info!(target: "app", "core process stopped successfully"); } } else { println!("[停止sidecar] 没有找到sidecar进程"); } } /// 启动核心 pub async fn start_core(&self) -> Result<()> { let mut running = self.running.lock().await; if *running { log::info!("core is running"); return Ok(()); } let config_path = Config::generate_file(ConfigType::Run)?; // 先尝试服务模式 if service::check_service().await.is_ok() { log::info!(target: "app", "try to run core in service mode"); match service::run_core_by_service(&config_path).await { Ok(_) => { log::info!(target: "app", "core started successfully in service mode"); } Err(err) => { // 服务启动失败,尝试sidecar模式 log::warn!(target: "app", "failed to start core in service mode: {}", err); log::info!(target: "app", "trying to run core in sidecar mode"); self.run_core_by_sidecar(&config_path).await?; } } } else { // 服务不可用,直接使用sidecar模式 log::info!(target: "app", "service not available, running core in sidecar mode"); self.run_core_by_sidecar(&config_path).await?; } // 流量订阅 #[cfg(target_os = "macos")] log_err!(Tray::global().subscribe_traffic().await); *running = true; Ok(()) } /// 通过sidecar启动内核 async fn run_core_by_sidecar(&self, config_path: &PathBuf) -> Result<()> { let clash_core = { Config::verge().latest().clash_core.clone() }; let clash_core = clash_core.unwrap_or("verge-mihomo".into()); log::info!(target: "app", "starting core {} in sidecar mode", clash_core); println!("[sidecar启动] 开始以sidecar模式启动内核: {}", clash_core); // 检查系统中是否存在同名进程 if let Ok(pids) = self.check_existing_processes(&clash_core).await { if !pids.is_empty() { println!("[sidecar启动] 警告:系统中已存在同名进程"); // 尝试检查端口占用 if let Ok(config_content) = std::fs::read_to_string(config_path) { if let Ok(config) = serde_yaml::from_str::(&config_content) { // 获取配置中定义的端口 let mixed_port = config .get("mixed-port") .and_then(|v| v.as_u64()) .unwrap_or(7890); let http_port = config.get("port").and_then(|v| v.as_u64()).unwrap_or(7890); println!( "[sidecar启动] 检查端口占用: HTTP端口={}, 混合端口={}", http_port, mixed_port ); // 检查端口是否被占用 if self.is_port_in_use(mixed_port as u16).await || self.is_port_in_use(http_port as u16).await { println!("[sidecar启动] 端口已被占用,尝试终止已存在的进程"); // 尝试终止已存在的进程 for pid in pids { println!("[sidecar启动] 尝试终止进程 PID: {}", pid); self.terminate_process(pid).await; } // 等待短暂时间让资源释放 println!("[sidecar启动] 等待500ms让资源释放"); sleep(Duration::from_millis(500)).await; } } } } } else { println!("[sidecar启动] 无法检查系统进程,继续尝试启动"); } // 创建锁文件路径 let lock_file = dirs::app_home_dir()?.join(format!("{}.lock", clash_core)); println!("[sidecar启动] 锁文件路径: {:?}", lock_file); // 尝试获取文件锁 println!("[sidecar启动] 尝试获取文件锁"); let file = std::fs::OpenOptions::new() .write(true) .create(true) .open(&lock_file)?; match file.try_lock_exclusive() { Ok(_) => { // 成功获取锁,说明没有其他实例运行 println!("[sidecar启动] 成功获取文件锁,没有检测到其他运行的实例"); log::info!(target: "app", "acquired lock for core process"); // 保存锁对象到全局,防止被Drop handle::Handle::global().set_core_lock(file); } Err(err) => { // 无法获取锁,说明已有实例运行 println!("[sidecar启动] 无法获取文件锁,检测到其他实例可能正在运行"); println!("[sidecar启动] 错误信息: {:?}", err); log::warn!(target: "app", "another core process appears to be running"); // 尝试强制获取锁(可能会导致其他进程崩溃) println!("[sidecar启动] 尝试强制删除并重新创建锁文件"); std::fs::remove_file(&lock_file)?; let file = std::fs::OpenOptions::new() .write(true) .create(true) .open(&lock_file)?; println!("[sidecar启动] 尝试强制获取锁"); match file.lock_exclusive() { Ok(_) => println!("[sidecar启动] 成功强制获取锁"), Err(e) => println!("[sidecar启动] 强制获取锁失败: {:?}", e), } file.lock_exclusive()?; // 保存新锁 handle::Handle::global().set_core_lock(file); // 等待可能的其他进程退出 println!("[sidecar启动] 等待500ms,让可能的其他进程退出"); sleep(Duration::from_millis(500)).await; } } let app_handle = handle::Handle::global() .app_handle() .ok_or(anyhow::anyhow!("failed to get app handle"))?; // 获取配置目录 let config_dir = dirs::app_home_dir()?; let config_path_str = dirs::path_to_str(config_path)?; // 启动核心进程并转入后台运行 println!("[sidecar启动] 开始启动核心进程"); let (_, child) = app_handle .shell() .sidecar(clash_core)? .args(["-d", dirs::path_to_str(&config_dir)?, "-f", config_path_str]) .spawn()?; // 保存进程ID以便后续管理 println!("[sidecar启动] 核心进程启动成功,PID: {:?}", child.pid()); handle::Handle::global().set_core_process(child); // 等待短暂时间确保启动成功 sleep(Duration::from_millis(300)).await; println!("[sidecar启动] 内核启动完成"); log::info!(target: "app", "core started in sidecar mode"); Ok(()) } /// 重启内核 pub async fn restart_core(&self) -> Result<()> { // 重新启动app log::info!(target: "app", "restarting core"); self.stop_core().await?; self.start_core().await?; log::info!(target: "app", "core restarted successfully"); Ok(()) } /// 使用默认配置 pub async fn use_default_config(&self, msg_type: &str, msg_content: &str) -> Result<()> { let runtime_path = dirs::app_home_dir()?.join(RUNTIME_CONFIG); *Config::runtime().draft() = IRuntime { config: Some(Config::clash().latest().0.clone()), exists_keys: vec![], chain_logs: Default::default(), }; help::save_yaml( &runtime_path, &Config::clash().latest().0, Some("# Clash Verge Runtime"), )?; handle::Handle::notice_message(msg_type, msg_content); Ok(()) } /// 切换核心 pub async fn change_core(&self, clash_core: Option) -> Result<()> { let clash_core = clash_core.ok_or(anyhow::anyhow!("clash core is null"))?; const CLASH_CORES: [&str; 2] = ["verge-mihomo", "verge-mihomo-alpha"]; if !CLASH_CORES.contains(&clash_core.as_str()) { bail!("invalid clash core name \"{clash_core}\""); } log::info!(target: "app", "change core to `{clash_core}`"); // 1. 先更新内核配置(但不应用) Config::verge().draft().clash_core = Some(clash_core); // 2. 使用新内核验证配置 println!("[切换内核] 使用新内核验证配置"); match self.validate_config().await { Ok((true, _)) => { println!("[切换内核] 配置验证通过,开始切换内核"); // 3. 验证通过后,应用内核配置并重启 Config::verge().apply(); log_err!(Config::verge().latest().save_file()); match self.restart_core().await { Ok(_) => { println!("[切换内核] 内核切换成功"); Config::runtime().apply(); Ok(()) } Err(err) => { println!("[切换内核] 内核切换失败: {}", err); // 即使使用服务失败,我们也尝试使用sidecar模式启动 log::info!(target: "app", "trying sidecar mode after service failure"); self.start_core().await?; Config::runtime().apply(); Ok(()) } } } Ok((false, error_msg)) => { println!("[切换内核] 配置验证失败: {}", error_msg); // 使用默认配置并继续切换内核 self.use_default_config("config_validate::core_change", &error_msg) .await?; Config::verge().apply(); log_err!(Config::verge().latest().save_file()); match self.restart_core().await { Ok(_) => { println!("[切换内核] 内核切换成功(使用默认配置)"); Ok(()) } Err(err) => { println!("[切换内核] 内核切换失败: {}", err); // 即使使用服务失败,我们也尝试使用sidecar模式启动 log::info!(target: "app", "trying sidecar mode after service failure with default config"); self.start_core().await?; Ok(()) } } } Err(err) => { println!("[切换内核] 验证过程发生错误: {}", err); Config::verge().discard(); Err(err) } } } /// 内部验证配置文件的实现 async fn validate_config_internal(&self, config_path: &str) -> Result<(bool, String)> { // 检查程序是否正在退出,如果是则跳过验证 if handle::Handle::global().is_exiting() { println!("[core配置验证] 应用正在退出,跳过验证"); return Ok((true, String::new())); } println!("[core配置验证] 开始验证配置文件: {}", config_path); let clash_core = { Config::verge().latest().clash_core.clone() }; let clash_core = clash_core.unwrap_or("verge-mihomo".into()); println!("[core配置验证] 使用内核: {}", clash_core); let app_handle = handle::Handle::global().app_handle().unwrap(); let test_dir = dirs::app_home_dir()?.join("test"); let test_dir = dirs::path_to_str(&test_dir)?; println!("[core配置验证] 测试目录: {}", test_dir); // 使用子进程运行clash验证配置 println!("[core配置验证] 运行子进程验证配置"); let output = app_handle .shell() .sidecar(clash_core)? .args(["-t", "-d", test_dir, "-f", config_path]) .output() .await?; let stderr = String::from_utf8_lossy(&output.stderr); let stdout = String::from_utf8_lossy(&output.stdout); // 检查进程退出状态和错误输出 let error_keywords = ["FATA", "fatal", "Parse config error", "level=fatal"]; let has_error = !output.status.success() || error_keywords.iter().any(|&kw| stderr.contains(kw)); println!("\n[core配置验证] -------- 验证结果 --------"); println!("[core配置验证] 进程退出状态: {:?}", output.status); if !stderr.is_empty() { println!("[core配置验证] stderr输出:\n{}", stderr); } if !stdout.is_empty() { println!("[core配置验证] stdout输出:\n{}", stdout); } if has_error { println!("[core配置验证] 发现错误,开始处理错误信息"); let error_msg = if !stdout.is_empty() { stdout.to_string() } else if !stderr.is_empty() { stderr.to_string() } else if let Some(code) = output.status.code() { format!("验证进程异常退出,退出码: {}", code) } else { "验证进程被终止".to_string() }; println!("[core配置验证] -------- 验证结束 --------\n"); Ok((false, error_msg)) // 返回错误消息给调用者处理 } else { println!("[core配置验证] 验证成功"); println!("[core配置验证] -------- 验证结束 --------\n"); Ok((true, String::new())) } } /// 验证运行时配置 pub async fn validate_config(&self) -> Result<(bool, String)> { let config_path = Config::generate_file(ConfigType::Check)?; let config_path = dirs::path_to_str(&config_path)?; self.validate_config_internal(config_path).await } /// 验证指定的配置文件 pub async fn validate_config_file( &self, config_path: &str, is_merge_file: Option, ) -> Result<(bool, String)> { // 检查程序是否正在退出,如果是则跳过验证 if handle::Handle::global().is_exiting() { println!("[core配置验证] 应用正在退出,跳过验证"); return Ok((true, String::new())); } // 检查文件是否存在 if !std::path::Path::new(config_path).exists() { let error_msg = format!("File not found: {}", config_path); //handle::Handle::notice_message("config_validate::file_not_found", &error_msg); return Ok((false, error_msg)); } // 如果是合并文件且不是强制验证,执行语法检查但不进行完整验证 if is_merge_file.unwrap_or(false) { println!( "[core配置验证] 检测到Merge文件,仅进行语法检查: {}", config_path ); return self.validate_file_syntax(config_path).await; } // 检查是否为脚本文件 let is_script = if config_path.ends_with(".js") { true } else { match self.is_script_file(config_path) { Ok(result) => result, Err(err) => { // 如果无法确定文件类型,尝试使用Clash内核验证 log::warn!(target: "app", "无法确定文件类型: {}, 错误: {}", config_path, err); return self.validate_config_internal(config_path).await; } } }; if is_script { log::info!(target: "app", "检测到脚本文件,使用JavaScript验证: {}", config_path); return self.validate_script_file(config_path).await; } // 对YAML配置文件使用Clash内核验证 log::info!(target: "app", "使用Clash内核验证配置文件: {}", config_path); self.validate_config_internal(config_path).await } /// 检查文件是否为脚本文件 fn is_script_file(&self, path: &str) -> Result { // 1. 先通过扩展名快速判断 if path.ends_with(".yaml") || path.ends_with(".yml") { return Ok(false); // YAML文件不是脚本文件 } else if path.ends_with(".js") { return Ok(true); // JS文件是脚本文件 } // 2. 读取文件内容 let content = match std::fs::read_to_string(path) { Ok(content) => content, Err(err) => { log::warn!(target: "app", "无法读取文件以检测类型: {}, 错误: {}", path, err); return Err(anyhow::anyhow!( "Failed to read file to detect type: {}", err )); } }; // 3. 检查是否存在明显的YAML特征 let has_yaml_features = content.contains(": ") || content.contains("#") || content.contains("---") || content.lines().any(|line| line.trim().starts_with("- ")); // 4. 检查是否存在明显的JS特征 let has_js_features = content.contains("function ") || content.contains("const ") || content.contains("let ") || content.contains("var ") || content.contains("//") || content.contains("/*") || content.contains("*/") || content.contains("export ") || content.contains("import "); // 5. 决策逻辑 if has_yaml_features && !has_js_features { // 只有YAML特征,没有JS特征 return Ok(false); } else if has_js_features && !has_yaml_features { // 只有JS特征,没有YAML特征 return Ok(true); } else if has_yaml_features && has_js_features { // 两种特征都有,需要更精细判断 // 优先检查是否有明确的JS结构特征 if content.contains("function main") || content.contains("module.exports") || content.contains("export default") { return Ok(true); } // 检查冒号后是否有空格(YAML的典型特征) let yaml_pattern_count = content.lines().filter(|line| line.contains(": ")).count(); if yaml_pattern_count > 2 { return Ok(false); // 多个键值对格式,更可能是YAML } } // 默认情况:无法确定时,假设为非脚本文件(更安全) log::debug!(target: "app", "无法确定文件类型,默认当作YAML处理: {}", path); Ok(false) } /// 验证脚本文件语法 async fn validate_script_file(&self, path: &str) -> Result<(bool, String)> { // 读取脚本内容 let content = match std::fs::read_to_string(path) { Ok(content) => content, Err(err) => { let error_msg = format!("Failed to read script file: {}", err); log::warn!(target: "app", "脚本语法错误: {}", err); //handle::Handle::notice_message("config_validate::script_syntax_error", &error_msg); return Ok((false, error_msg)); } }; log::debug!(target: "app", "验证脚本文件: {}", path); // 使用boa引擎进行基本语法检查 use boa_engine::{Context, Source}; let mut context = Context::default(); let result = context.eval(Source::from_bytes(&content)); match result { Ok(_) => { log::debug!(target: "app", "脚本语法验证通过: {}", path); // 检查脚本是否包含main函数 if !content.contains("function main") && !content.contains("const main") && !content.contains("let main") { let error_msg = "Script must contain a main function"; log::warn!(target: "app", "脚本缺少main函数: {}", path); //handle::Handle::notice_message("config_validate::script_missing_main", error_msg); return Ok((false, error_msg.to_string())); } Ok((true, String::new())) } Err(err) => { let error_msg = format!("Script syntax error: {}", err); log::warn!(target: "app", "脚本语法错误: {}", err); //handle::Handle::notice_message("config_validate::script_syntax_error", &error_msg); Ok((false, error_msg)) } } } /// 更新proxies等配置 pub async fn update_config(&self) -> Result<(bool, String)> { // 检查程序是否正在退出,如果是则跳过完整验证流程 if handle::Handle::global().is_exiting() { println!("[core配置更新] 应用正在退出,跳过验证"); return Ok((true, String::new())); } println!("[core配置更新] 开始更新配置"); // 1. 先生成新的配置内容 println!("[core配置更新] 生成新的配置内容"); Config::generate().await?; // 2. 生成临时文件并进行验证 println!("[core配置更新] 生成临时配置文件用于验证"); let temp_config = Config::generate_file(ConfigType::Check)?; let temp_config = dirs::path_to_str(&temp_config)?; println!("[core配置更新] 临时配置文件路径: {}", temp_config); // 3. 验证配置 match self.validate_config().await { Ok((true, _)) => { println!("[core配置更新] 配置验证通过"); // 4. 验证通过后,生成正式的运行时配置 println!("[core配置更新] 生成运行时配置"); let run_path = Config::generate_file(ConfigType::Run)?; let run_path = dirs::path_to_str(&run_path)?; // 5. 应用新配置 println!("[core配置更新] 应用新配置"); for i in 0..3 { CoreManager::global().ensure_running_core().await; match MihomoManager::global().put_configs_force(run_path).await { Ok(_) => { println!("[core配置更新] 配置应用成功"); Config::runtime().apply(); return Ok((true, String::new())); } Err(err) => { if i < 2 { println!("[core配置更新] 第{}次重试应用配置", i + 1); log::info!(target: "app", "{err}"); sleep(Duration::from_millis(100)).await; } else { println!("[core配置更新] 配置应用失败: {}", err); Config::runtime().discard(); return Ok((false, err.to_string())); } } } } Ok((true, String::new())) } Ok((false, error_msg)) => { println!("[core配置更新] 配置验证失败: {}", error_msg); Config::runtime().discard(); Ok((false, error_msg)) } Err(e) => { println!("[core配置更新] 验证过程发生错误: {}", e); Config::runtime().discard(); Err(e) } } } /// 只进行文件语法检查,不进行完整验证 async fn validate_file_syntax(&self, config_path: &str) -> Result<(bool, String)> { println!("[core配置语法检查] 开始检查文件: {}", config_path); // 读取文件内容 let content = match std::fs::read_to_string(config_path) { Ok(content) => content, Err(err) => { let error_msg = format!("Failed to read file: {}", err); println!("[core配置语法检查] 无法读取文件: {}", error_msg); return Ok((false, error_msg)); } }; // 对YAML文件尝试解析,只检查语法正确性 println!("[core配置语法检查] 进行YAML语法检查"); match serde_yaml::from_str::(&content) { Ok(_) => { println!("[core配置语法检查] YAML语法检查通过"); Ok((true, String::new())) } Err(err) => { // 使用标准化的前缀,以便错误处理函数能正确识别 let error_msg = format!("YAML syntax error: {}", err); println!("[core配置语法检查] YAML语法错误: {}", error_msg); Ok((false, error_msg)) } } } /// 获取当前内核运行模式 pub async fn get_running_mode(&self) -> RunningMode { let running = self.running.lock().await; if !*running { return RunningMode::NotRunning; } // 检查服务状态 match service::check_service().await { Ok(_) => { // 检查服务是否实际运行核心 match service::is_service_running().await { Ok(true) => RunningMode::Service, _ => { // 服务存在但可能没有运行,检查是否有sidecar进程 if handle::Handle::global().has_core_process() { // 检查是否持有文件锁,确保是由我们启动的进程 if handle::Handle::global().has_core_lock() { RunningMode::Sidecar } else { // 有进程但没有文件锁,可能是外部启动的进程 log::warn!(target: "app", "core process exists but no lock file"); RunningMode::Sidecar // 仍返回Sidecar模式,但记录了警告 } } else { RunningMode::NotRunning } } } } Err(_) => { // 服务不可用,检查是否有sidecar进程 if handle::Handle::global().has_core_process() { // 检查是否持有文件锁,确保是由我们启动的进程 if handle::Handle::global().has_core_lock() { RunningMode::Sidecar } else { // 有进程但没有文件锁,可能是外部启动的进程 log::warn!(target: "app", "core process exists but no lock file"); RunningMode::Sidecar // 仍返回Sidecar模式,但记录了警告 } } else { RunningMode::NotRunning } } } } /// 检查系统中是否存在同名进程 async fn check_existing_processes(&self, process_name: &str) -> Result> { println!("[进程检查] 检查系统中是否存在进程: {}", process_name); #[cfg(target_os = "windows")] { use std::process::Command; println!("[进程检查] Windows系统,使用tasklist命令"); let output = Command::new("tasklist") .args(["/FO", "CSV", "/NH"]) .output()?; let output = String::from_utf8_lossy(&output.stdout); let pids: Vec = output .lines() .filter(|line| line.contains(process_name)) .filter_map(|line| { println!("[进程检查] 发现匹配行: {}", line); let parts: Vec<&str> = line.split(',').collect(); if parts.len() >= 2 { let pid_str = parts[1].trim_matches('"'); pid_str.parse::().ok().map(|pid| { println!("[进程检查] 发现进程 PID: {}", pid); pid }) } else { None } }) .collect(); println!("[进程检查] 共发现 {} 个相关进程", pids.len()); Ok(pids) } #[cfg(target_os = "linux")] { use std::process::Command; println!("[进程检查] Linux系统,使用pgrep命令"); let output = Command::new("pgrep") .arg("-f") .arg(process_name) .output()?; let output = String::from_utf8_lossy(&output.stdout); let pids: Vec = output .lines() .filter_map(|line| { line.trim().parse::().ok().map(|pid| { println!("[进程检查] 发现进程 PID: {}", pid); pid }) }) .collect(); println!("[进程检查] 共发现 {} 个相关进程", pids.len()); Ok(pids) } #[cfg(target_os = "macos")] { use std::process::Command; println!("[进程检查] macOS系统,使用ps命令"); let output = Command::new("ps") .args(["-ax", "-o", "pid,command"]) .output()?; let output = String::from_utf8_lossy(&output.stdout); let pids: Vec = output .lines() .filter(|line| line.contains(process_name)) .filter_map(|line| { println!("[进程检查] 发现匹配行: {}", line); let parts: Vec<&str> = line.split_whitespace().collect(); if !parts.is_empty() { parts[0].parse::().ok().map(|pid| { println!("[进程检查] 发现进程 PID: {}", pid); pid }) } else { None } }) .collect(); println!("[进程检查] 共发现 {} 个相关进程", pids.len()); Ok(pids) } } /// 检查端口是否被占用 async fn is_port_in_use(&self, port: u16) -> bool { println!("[端口检查] 检查端口 {} 是否被占用", port); use tokio::net::TcpSocket; match TcpSocket::new_v4() { Ok(socket) => { let addr = format!("127.0.0.1:{}", port).parse().unwrap(); match socket.bind(addr) { Ok(_) => { // 如果能绑定成功,说明端口未被占用 println!("[端口检查] 端口 {} 未被占用", port); false } Err(_) => { // 绑定失败,端口已被占用 println!("[端口检查] 端口 {} 已被占用", port); true } } } Err(err) => { // 创建socket失败,保守返回端口被占用 println!("[端口检查] 创建Socket失败: {:?}, 假设端口已被占用", err); true } } } /// 终止进程 async fn terminate_process(&self, pid: u32) { println!("[进程终止] 尝试终止进程 PID: {}", pid); #[cfg(target_os = "windows")] { use std::process::Command; let output = Command::new("taskkill") .args(["/F", "/PID", &pid.to_string()]) .output(); match output { Ok(output) => { if output.status.success() { println!("[进程终止] 成功终止进程 PID: {}", pid); } else { let stderr = String::from_utf8_lossy(&output.stderr); println!("[进程终止] 终止进程失败: {}", stderr); } } Err(err) => { println!("[进程终止] 执行终止命令失败: {:?}", err); } } } #[cfg(target_os = "linux")] { use std::process::Command; let output = Command::new("kill").args(["-9", &pid.to_string()]).output(); match output { Ok(output) => { if output.status.success() { println!("[进程终止] 成功终止进程 PID: {}", pid); } else { let stderr = String::from_utf8_lossy(&output.stderr); println!("[进程终止] 终止进程失败: {}", stderr); } } Err(err) => { println!("[进程终止] 执行终止命令失败: {:?}", err); } } } #[cfg(target_os = "macos")] { use std::process::Command; let output = Command::new("kill").args(["-9", &pid.to_string()]).output(); match output { Ok(output) => { if output.status.success() { println!("[进程终止] 成功终止进程 PID: {}", pid); } else { let stderr = String::from_utf8_lossy(&output.stderr); println!("[进程终止] 终止进程失败: {}", stderr); } } Err(err) => { println!("[进程终止] 执行终止命令失败: {:?}", err); } } } } /// 确保 Mihomo 和 Verge service 都在运行 pub async fn ensure_running_core(&self) { if MihomoManager::global().is_mihomo_running().await.is_err() { log_err!(self.restart_core().await); } match is_service_running().await { Ok(false) => log_err!(self.restart_core().await), Ok(true) => { if MihomoManager::global().is_mihomo_running().await.is_err() { log_err!(self.restart_core().await); } } _ => {} } } }