71 lines
2.2 KiB
Rust
71 lines
2.2 KiB
Rust
|
use std::fs;
|
|||
|
|
use std::process::Command;
|
|||
|
|
use std::thread::sleep;
|
|||
|
|
use std::time::Duration;
|
|||
|
|
use windows::Win32::System::Console;
|
|||
|
|
use windows::Win32::Foundation::HWND;
|
|||
|
|
|
|||
|
|
fn hide_console() {
|
|||
|
|
unsafe {
|
|||
|
|
let console = Console::GetConsoleWindow();
|
|||
|
|
if console != HWND::default() {
|
|||
|
|
let _ = Console::FreeConsole();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fn main() {
|
|||
|
|
// 隐藏控制台窗口
|
|||
|
|
hide_console();
|
|||
|
|
|
|||
|
|
// 获取 BootLoader 启动目录所在盘的 AppData 目录
|
|||
|
|
let appdata_dir = get_appdata_dir();
|
|||
|
|
|
|||
|
|
// 构建 Updater 相关文件路径
|
|||
|
|
let updater_exe = appdata_dir.join("Updater.exe");
|
|||
|
|
let updater_new_exe = appdata_dir.join("Updater.new.exe");
|
|||
|
|
|
|||
|
|
// 等待 Updater 进程退出
|
|||
|
|
wait_for_process_exit("Updater.exe");
|
|||
|
|
|
|||
|
|
// 检查 Updater.new.exe 是否存在
|
|||
|
|
if updater_new_exe.exists() {
|
|||
|
|
// 直接覆盖 Updater.exe（fs::rename 在 Windows 上会自动替换已存在的文件）
|
|||
|
|
fs::rename(&updater_new_exe, &updater_exe).expect("Failed to rename Updater.new.exe to Updater.exe");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 启动 Updater.exe
|
|||
|
|
Command::new(updater_exe)
|
|||
|
|
.spawn()
|
|||
|
|
.expect("Failed to start Updater.exe");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fn get_appdata_dir() -> std::path::PathBuf {
|
|||
|
|
let exe_path = std::env::current_exe().expect("Failed to get executable path");
|
|||
|
|
let drive = if let Some(parent) = exe_path.parent() {
|
|||
|
|
let drive = parent.as_os_str().to_str().unwrap_or("C:/").split('\\').next().unwrap_or("C:");
|
|||
|
|
std::path::Path::new(&format!("{}/", drive)).to_path_buf()
|
|||
|
|
} else {
|
|||
|
|
std::path::Path::new("C:/").to_path_buf()
|
|||
|
|
};
|
|||
|
|
drive.join("AppData").join("Updater")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fn wait_for_process_exit(process_name: &str) {
|
|||
|
|
loop {
|
|||
|
|
// 使用 tasklist 命令检查进程是否存在
|
|||
|
|
let output = Command::new("tasklist")
|
|||
|
|
.args(["/FI", &format!("IMAGENAME eq {}", process_name)])
|
|||
|
|
.output()
|
|||
|
|
.expect("Failed to execute tasklist");
|
|||
|
|
|
|||
|
|
let output_str = String::from_utf8_lossy(&output.stdout);
|
|||
|
|
if !output_str.contains(process_name) {
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 等待 100 毫秒后再次检查
|
|||
|
|
sleep(Duration::from_millis(100));
|
|||
|
|
}
|
|||
|
|
}
|