cargo-minimize-oom 0.1.0

A tool for minimizing rustc ICEs, specially OOMs
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

#[macro_use]
extern crate tracing;

use std::sync::{
    Arc,
    atomic::{AtomicBool, Ordering},
};
use std::{thread, time};

use anyhow::Result;
use cargo_minimize_oom::{Cargo, Parser};
use tracing::{Level, error};

fn main() -> Result<()> {
    let Cargo::Minimize_OOM(options) = Cargo::parse();

    cargo_minimize_oom::init_recommended_tracing_subscriber(Level::INFO);

    let cancel = Arc::new(AtomicBool::new(false));
    let cancel2 = Arc::clone(&cancel);

    let mut ctrl_c_pressed = false;
    let result = ctrlc::set_handler(move || {
        // If ctrl c was pressed already, kill it now.
        if ctrl_c_pressed {
            warn!("Process killed");
            std::process::exit(130);
        }

        warn!("Shutting down gracefully, press CTRL-C again to kill");
        cancel.store(true, Ordering::SeqCst);
        ctrl_c_pressed = true;
        cargo_minimize_oom::stop_keyboard_io();
    });

    if let Err(err) = result {
        error!("Failed to install CTRL-C handler: {err}");
    }

    let amount = time::Duration::from_secs(4);
    thread::sleep(amount);
    let res = cargo_minimize_oom::minimize(options, cancel2);
    cargo_minimize_oom::stop_keyboard_io();
    res
}