bustd 0.1.1

Lightweight process killer daemon for out-of-memory scenarios
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

use argh::FromArgs;

#[derive(FromArgs)]
/// Lightweight process killer daemon for out-of-memory scenarios
pub struct CommandLineArgs {
    /// toggles on verbose output
    #[argh(switch, short = 'V')]
    pub verbose: bool,

    /// when set, the process will not be daemonized
    #[argh(switch, short = 'n')]
    pub no_daemon: bool,

    /// when set, the victim's entire process group will be killed
    #[argh(switch, short = 'g')]
    pub kill_pgroup: bool,

    /// sets the PSI value on which, if surpassed, a process will be killed
    #[argh(option, short = 'p', long = "psi", default = "25.0")]
    pub cutoff_psi: f32, // TODO: responsitivity multiplier?

    #[cfg(feature = "glob-ignore")]
    /// all processes whose names match any of the supplied vertical bar-separated glob patterns will never be chosen to be killed
    #[argh(
        option,
        short = 'u',
        long = "unkillables",
        from_str_fn(parse_unkillables)
    )]
    pub ignored: Option<Vec<glob::Pattern>>,
}

#[cfg(feature = "glob-ignore")]
fn parse_unkillables(arg: &str) -> Result<Vec<glob::Pattern>, String> {
    let unkillables: Result<Vec<_>, _> = arg.split('|').map(glob::Pattern::new).collect();
    
    unkillables.map_err(|err| err.to_string())
}