src/main.rs


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

mod bpf_skel;
pub use bpf_skel::*;
pub mod bpf_intf;

mod e_core_selector;
mod energy;
mod scheduler;

#[rustfmt::skip]
mod bpf;

use anyhow::Result;
use clap::{Arg, ArgAction, Command};
use scheduler::Scheduler;
use std::mem::MaybeUninit;

type Pid = i32;

fn main() -> Result<()> {
    let matches = Command::new("Energy User Space Scheduler")
        .arg(
            Arg::new("mock")
                .short('m')
                .long("mock")
                .help("Use this flag if to activate kernel module mocking")
                .action(ArgAction::SetTrue)
                .required(false),
        )
        .arg(
            Arg::new("power_cap")
                .long("energy_cap")
                .help("Set a power cap for the processor")
                .required(false)
                .value_name("power in watts"),
        )
        .get_matches();

    let power_cap = *matches.get_one::<u64>("power_cap").unwrap_or(&u64::MAX);
    let use_mocking = matches.get_flag("mock");

    // Initialize and load the scheduler.
    let mut open_object = MaybeUninit::uninit();
    loop {
        let mut sched = Scheduler::init(&mut open_object, use_mocking, power_cap)?;
        if !sched.run()?.should_restart() {
            break;
        }
    }

    Ok(())
}