cpu_manager_config_lib/
cpu-manager-config-lib_rust_config_lib_source.rs

1use fidl::unpersist;
2use fidl_cf_sc_internal_cpumanagerconfiglib::Config as FidlConfig;
3use fuchsia_inspect::Node;
4use fuchsia_runtime::{take_startup_handle, HandleInfo, HandleType};
5#[derive(Debug)]
6pub struct Config {
7    pub node_config_path: String,
8}
9impl Config {
10    pub fn take_from_startup_handle() -> Self {
11        let config_vmo: zx::Vmo =
12            take_startup_handle(HandleInfo::new(HandleType::ComponentConfigVmo, 0))
13                .expect("Config VMO handle must be provided and cannot already have been taken.")
14                .into();
15        let config_size =
16            config_vmo.get_content_size().expect("must be able to read config vmo content size");
17        assert_ne!(config_size, 0, "config vmo must be non-empty");
18        let config_bytes =
19            config_vmo.read_to_vec(0, config_size).expect("must be able to read config vmo");
20        let checksum_length = u16::from_le_bytes([config_bytes[0], config_bytes[1]]) as usize;
21        let fidl_start = 2 + checksum_length;
22        let observed_checksum = &config_bytes[2..fidl_start];
23        let expected_checksum = vec![
24            0x67, 0x48, 0x7c, 0xe4, 0x07, 0x42, 0xcf, 0x13, 0x2f, 0xbc, 0x67, 0x17, 0xda, 0x8e,
25            0x57, 0xc8, 0x2e, 0x8f, 0x91, 0x93, 0xb6, 0xaf, 0x03, 0xaa, 0x6d, 0x2c, 0x02, 0x0b,
26            0xda, 0xa3, 0xa3, 0x27,
27        ];
28        assert_eq!(
29            observed_checksum, expected_checksum,
30            "checksum from config VMO does not match expected checksum"
31        );
32        let fidl_config: FidlConfig = unpersist(&config_bytes[fidl_start..])
33            .expect("must be able to parse bytes as config FIDL");
34        Self { node_config_path: fidl_config.node_config_path }
35    }
36    pub fn record_inspect(&self, inspector_node: &Node) {
37        inspector_node.record_string("node_config_path", &self.node_config_path);
38    }
39}