rdsh 0.5.2

Rust port/shim for dsh: fast launcher, token saver, zero behavior breakage via passthrough
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

//! Small signal bridge so local servers can clean up per-process handoff files.

use std::sync::atomic::{AtomicBool, Ordering};

static REQUESTED: AtomicBool = AtomicBool::new(false);

pub fn requested() -> bool {
    REQUESTED.load(Ordering::Relaxed)
}

#[cfg(unix)]
pub fn install(locale: crate::locale::Locale) -> anyhow::Result<()> {
    extern "C" fn request_shutdown(_: libc::c_int) {
        REQUESTED.store(true, Ordering::Relaxed);
    }

    unsafe {
        for signal in [libc::SIGINT, libc::SIGTERM] {
            if libc::signal(signal, request_shutdown as *const () as libc::sighandler_t)
                == libc::SIG_ERR
            {
                anyhow::bail!(match locale {
                    crate::locale::Locale::English =>
                        "could not install server shutdown signal handler",
                    crate::locale::Locale::Chinese => "无法安装服务器关闭信号处理程序",
                    crate::locale::Locale::Japanese =>
                        "サーバー終了シグナルハンドラーを設定できませんでした",
                });
            }
        }
    }
    Ok(())
}

#[cfg(windows)]
pub fn install(locale: crate::locale::Locale) -> anyhow::Result<()> {
    unsafe extern "system" fn request_shutdown(event: u32) -> i32 {
        const CTRL_C_EVENT: u32 = 0;
        const CTRL_BREAK_EVENT: u32 = 1;
        if event == CTRL_C_EVENT || event == CTRL_BREAK_EVENT {
            REQUESTED.store(true, Ordering::Relaxed);
            1
        } else {
            0
        }
    }

    #[link(name = "Kernel32")]
    unsafe extern "system" {
        fn SetConsoleCtrlHandler(
            handler: Option<unsafe extern "system" fn(u32) -> i32>,
            add: i32,
        ) -> i32;
    }

    if unsafe { SetConsoleCtrlHandler(Some(request_shutdown), 1) } == 0 {
        anyhow::bail!(match locale {
            crate::locale::Locale::English => "could not install server shutdown signal handler",
            crate::locale::Locale::Chinese => "无法安装服务器关闭信号处理程序",
            crate::locale::Locale::Japanese =>
                "サーバー終了シグナルハンドラーを設定できませんでした",
        });
    }
    Ok(())
}

#[cfg(not(any(unix, windows)))]
pub fn install(_locale: crate::locale::Locale) -> anyhow::Result<()> {
    Ok(())
}