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
40
41
42
43
44
45
46
47
48
49
50
51

use std::fs::OpenOptions;

use daemonize::Daemonize;

use crate::{error::Result, utils};

pub fn daemonize() -> Result<()> {
    let running_as_sudo = utils::running_as_sudo();

    let username = if running_as_sudo {
        "root".into()
    } else {
        utils::get_username().unwrap_or_else(|| "nobody".into())
    };

    let open_opts = OpenOptions::new()
        .truncate(false)
        .create(true)
        .write(true)
        .to_owned();

    let (stdout_path, stderr_path, pidfile_path) = if running_as_sudo {
        (
            "/var/log/bustd.out",
            "/var/log/bustd.err",
            "/var/run/bustd.pid",
        )
    } else {
        ("/tmp/bustd.out", "/tmp/bustd.err", "/tmp/bustd.pid")
    };

    let stdout = open_opts.open(stdout_path)?;
    let stderr = open_opts.open(stderr_path)?;

    let daemonize = Daemonize::new()
        .user(&*username)
        .pid_file(pidfile_path)
        .chown_pid_file(false)
        .working_directory("/tmp")
        .stdout(stdout)
        .stderr(stderr);

    daemonize.start()?;

    println!(
        "[LOG] User {} has started the daemon successfully.",
        username
    );

    Ok(())
}