mirror of
https://github.com/clash-verge-rev/clash-verge-rev.git
synced 2026-01-29 08:45:41 +08:00
feat(signal): add clash-verge-signal crate and integrate signal handling (#5500)
* feat(signal): add clash-verge-signal crate and integrate signal handling * fix: clippy error on type complexity
This commit is contained in:
Tunglies
@@ -1,4 +1,3 @@
|
||||
pub mod auto_backup;
|
||||
pub mod lightweight;
|
||||
pub mod signal;
|
||||
pub mod sysinfo;
|
||||
|
||||
@@ -1,12 +0,0 @@
|
||||
#[cfg(unix)]
|
||||
mod unix;
|
||||
#[cfg(windows)]
|
||||
mod windows;
|
||||
|
||||
pub fn register() {
|
||||
#[cfg(windows)]
|
||||
windows::register();
|
||||
|
||||
#[cfg(unix)]
|
||||
unix::register();
|
||||
}
|
||||
@@ -1,37 +0,0 @@
|
||||
use signal_hook::{
|
||||
consts::{SIGHUP, SIGINT, SIGTERM},
|
||||
iterator::Signals,
|
||||
low_level,
|
||||
};
|
||||
|
||||
use crate::feat;
|
||||
use clash_verge_logging::{Type, logging, logging_error};
|
||||
|
||||
pub fn register() {
|
||||
tauri::async_runtime::spawn(async {
|
||||
let signals = [SIGTERM, SIGINT, SIGHUP];
|
||||
match Signals::new(signals) {
|
||||
Ok(mut sigs) => {
|
||||
for signal in &mut sigs {
|
||||
let signal_to_str = |signal: i32| match signal {
|
||||
SIGTERM => "SIGTERM",
|
||||
SIGINT => "SIGINT",
|
||||
SIGHUP => "SIGHUP",
|
||||
_ => "UNKNOWN",
|
||||
};
|
||||
logging!(info, Type::System, "捕获到信号 {}", signal_to_str(signal));
|
||||
feat::clean_async().await;
|
||||
// After printing it, do whatever the signal was supposed to do in the first place
|
||||
logging_error!(
|
||||
Type::System,
|
||||
"信号 {:?} 默认处理失败",
|
||||
low_level::emulate_default_handler(signal)
|
||||
);
|
||||
}
|
||||
}
|
||||
Err(e) => {
|
||||
logging!(error, Type::System, "注册信号处理器失败: {}", e);
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
@@ -1,127 +0,0 @@
|
||||
use tauri::Manager as _;
|
||||
use windows_sys::Win32::{
|
||||
Foundation::{HWND, LPARAM, LRESULT, WPARAM},
|
||||
UI::WindowsAndMessaging::{
|
||||
CW_USEDEFAULT, CreateWindowExW, DefWindowProcW, DestroyWindow, RegisterClassW,
|
||||
WM_ENDSESSION, WM_QUERYENDSESSION, WNDCLASSW, WS_EX_LAYERED, WS_EX_NOACTIVATE,
|
||||
WS_EX_TOOLWINDOW, WS_EX_TRANSPARENT, WS_OVERLAPPED,
|
||||
},
|
||||
};
|
||||
|
||||
use crate::{core::handle, feat};
|
||||
use clash_verge_logging::{Type, logging};
|
||||
|
||||
// code refer to:
|
||||
// global-hotkey (https://github.com/tauri-apps/global-hotkey)
|
||||
// Global Shortcut (https://github.com/tauri-apps/plugins-workspace/tree/v2/plugins/global-shortcut)
|
||||
|
||||
struct ShutdownState {
|
||||
hwnd: HWND,
|
||||
}
|
||||
|
||||
unsafe impl Send for ShutdownState {}
|
||||
unsafe impl Sync for ShutdownState {}
|
||||
|
||||
impl Drop for ShutdownState {
|
||||
fn drop(&mut self) {
|
||||
// this log not be printed, I don't know why.
|
||||
logging!(info, Type::System, "正在销毁系统关闭监听窗口");
|
||||
unsafe {
|
||||
DestroyWindow(self.hwnd);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
unsafe extern "system" fn shutdown_proc(
|
||||
hwnd: HWND,
|
||||
msg: u32,
|
||||
wparam: WPARAM,
|
||||
lparam: LPARAM,
|
||||
) -> LRESULT {
|
||||
// refer: https://learn.microsoft.com/zh-cn/windows/win32/shutdown/shutting-down#shutdown-notifications
|
||||
// only perform reset operations in `WM_ENDSESSION`
|
||||
match msg {
|
||||
WM_QUERYENDSESSION => {
|
||||
logging!(
|
||||
info,
|
||||
Type::System,
|
||||
"System is shutting down or user is logging off."
|
||||
);
|
||||
}
|
||||
WM_ENDSESSION => {
|
||||
tauri::async_runtime::block_on(async move {
|
||||
logging!(info, Type::System, "Session ended, system shutting down.");
|
||||
feat::clean_async().await;
|
||||
logging!(info, Type::System, "resolved reset finished");
|
||||
});
|
||||
}
|
||||
_ => {}
|
||||
};
|
||||
unsafe { DefWindowProcW(hwnd, msg, wparam, lparam) }
|
||||
}
|
||||
|
||||
fn encode_wide<S: AsRef<std::ffi::OsStr>>(string: S) -> Vec<u16> {
|
||||
std::os::windows::prelude::OsStrExt::encode_wide(string.as_ref())
|
||||
.chain(std::iter::once(0))
|
||||
.collect::<Vec<u16>>()
|
||||
}
|
||||
|
||||
fn get_instance_handle() -> windows_sys::Win32::Foundation::HMODULE {
|
||||
// Gets the instance handle by taking the address of the
|
||||
// pseudo-variable created by the microsoft linker:
|
||||
// https://devblogs.microsoft.com/oldnewthing/20041025-00/?p=37483
|
||||
|
||||
// This is preferred over GetModuleHandle(NULL) because it also works in DLLs:
|
||||
// https://stackoverflow.com/questions/21718027/getmodulehandlenull-vs-hinstance
|
||||
|
||||
unsafe extern "C" {
|
||||
static __ImageBase: windows_sys::Win32::System::SystemServices::IMAGE_DOS_HEADER;
|
||||
}
|
||||
|
||||
unsafe { &__ImageBase as *const _ as _ }
|
||||
}
|
||||
|
||||
pub fn register() {
|
||||
let app_handle = handle::Handle::app_handle();
|
||||
let class_name = encode_wide("global_shutdown_app");
|
||||
unsafe {
|
||||
let hinstance = get_instance_handle();
|
||||
|
||||
let wnd_class = WNDCLASSW {
|
||||
lpfnWndProc: Some(shutdown_proc),
|
||||
lpszClassName: class_name.as_ptr(),
|
||||
hInstance: hinstance,
|
||||
..std::mem::zeroed()
|
||||
};
|
||||
|
||||
RegisterClassW(&wnd_class);
|
||||
|
||||
let hwnd = CreateWindowExW(
|
||||
WS_EX_NOACTIVATE | WS_EX_TRANSPARENT | WS_EX_LAYERED |
|
||||
// WS_EX_TOOLWINDOW prevents this window from ever showing up in the taskbar, which
|
||||
// we want to avoid. If you remove this style, this window won't show up in the
|
||||
// taskbar *initially*, but it can show up at some later point. This can sometimes
|
||||
// happen on its own after several hours have passed, although this has proven
|
||||
// difficult to reproduce. Alternatively, it can be manually triggered by killing
|
||||
// `explorer.exe` and then starting the process back up.
|
||||
// It is unclear why the bug is triggered by waiting for several hours.
|
||||
WS_EX_TOOLWINDOW,
|
||||
class_name.as_ptr(),
|
||||
std::ptr::null(),
|
||||
WS_OVERLAPPED,
|
||||
CW_USEDEFAULT,
|
||||
0,
|
||||
CW_USEDEFAULT,
|
||||
0,
|
||||
std::ptr::null_mut(),
|
||||
std::ptr::null_mut(),
|
||||
hinstance,
|
||||
std::ptr::null_mut(),
|
||||
);
|
||||
if hwnd.is_null() {
|
||||
logging!(error, Type::System, "failed to create shutdown window");
|
||||
} else {
|
||||
app_handle.manage(ShutdownState { hwnd });
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -9,11 +9,13 @@ use crate::{
|
||||
sysopt,
|
||||
tray::Tray,
|
||||
},
|
||||
module::{auto_backup::AutoBackupManager, lightweight::auto_lightweight_boot, signal},
|
||||
feat,
|
||||
module::{auto_backup::AutoBackupManager, lightweight::auto_lightweight_boot},
|
||||
process::AsyncHandler,
|
||||
utils::{init, server, window_manager::WindowManager},
|
||||
};
|
||||
use clash_verge_logging::{Type, logging, logging_error};
|
||||
use clash_verge_signal;
|
||||
|
||||
pub mod dns;
|
||||
pub mod scheme;
|
||||
@@ -134,7 +136,11 @@ pub(super) async fn init_auto_backup() {
|
||||
|
||||
pub(super) fn init_signal() {
|
||||
logging!(info, Type::Setup, "Initializing signal handlers...");
|
||||
signal::register();
|
||||
clash_verge_signal::register(
|
||||
#[cfg(windows)]
|
||||
handle::Handle::app_handle(),
|
||||
feat::clean_async,
|
||||
);
|
||||
}
|
||||
|
||||
pub async fn init_work_config() {
|
||||
|
||||
Reference in New Issue
Block a user