capsula-capture-machine 0.3.0

A Capsula hook that captures machine information.
Documentation 
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

mod config;
mod error;

use crate::error::MachineHookError;
use capsula_core::captured::Captured;
use capsula_core::error::CapsulaResult;
use capsula_core::hook::{Hook, HookFactory, RuntimeParams};
use config::{MachineHookConfig, MachineHookFactory};
use sysinfo::{CpuRefreshKind, MemoryRefreshKind, RefreshKind, System};

pub const KEY: &str = "capture-machine";

#[derive(Debug, Default)]
pub struct MachineHook;

#[derive(Debug)]
pub struct CpuInfo {
    name: String,
    vender_id: String,
    brand: String,
    frequency_mhz: u64,
}

#[derive(Debug)]
pub struct MachineCaptured {
    pub os: String,
    pub os_version: String,
    pub kernel_version: String,
    pub architecture: String,
    pub cpus: Vec<CpuInfo>,
    // pub cpu_cores: usize,
    pub total_memory: usize,
    // pub user: String,
    pub hostname: String,
}

impl Hook for MachineHook {
    type Config = MachineHookConfig;
    type Output = MachineCaptured;

    fn id(&self) -> String {
        KEY.to_string()
    }

    fn config(&self) -> &Self::Config {
        &MachineHookConfig {}
    }

    fn run(&self, _params: &RuntimeParams) -> CapsulaResult<Self::Output> {
        let os = System::name().ok_or(MachineHookError::OsInfoError)?;
        let os_version = System::os_version().ok_or(MachineHookError::OsInfoError)?;
        let kernel_version = System::kernel_version().ok_or(MachineHookError::OsInfoError)?;
        let architecture = std::env::consts::ARCH.to_string();

        let system = System::new_with_specifics(
            RefreshKind::nothing()
                .with_cpu(CpuRefreshKind::nothing().with_frequency())
                .with_memory(MemoryRefreshKind::nothing().with_ram()),
        );
        let cpus = system
            .cpus()
            .iter()
            .map(|cpu| CpuInfo {
                name: cpu.name().to_string(),
                vender_id: cpu.vendor_id().to_string(),
                brand: cpu.brand().to_string(),
                frequency_mhz: cpu.frequency(),
            })
            .collect::<Vec<_>>();

        let total_memory = system.total_memory();
        let hostname = System::host_name().ok_or(MachineHookError::HostnameError)?;

        Ok(MachineCaptured {
            os,
            os_version,
            kernel_version,
            architecture,
            cpus,
            total_memory: total_memory as usize,
            hostname,
        })
    }
}

impl Captured for MachineCaptured {
    fn to_json(&self) -> serde_json::Value {
        serde_json::json!({
            "os": self.os,
            "os_version": self.os_version,
            "kernel_version": self.kernel_version,
            "architecture": self.architecture,
            "cpus": self.cpus.iter().map(|cpu| {
                serde_json::json!({
                    "name": cpu.name,
                    "vender_id": cpu.vender_id,
                    "brand": cpu.brand,
                    "frequency_mhz": cpu.frequency_mhz,
                })
            }).collect::<Vec<_>>(),
            "total_memory": self.total_memory,
            "hostname": self.hostname,
        })
    }
}

pub fn create_factory() -> Box<dyn HookFactory> {
    Box::new(MachineHookFactory)
}