sampler_component_config/
sampler-component-config_rust_config_lib_source.rs

1use fidl::unpersist;
2use fidl_cf_sc_internal_samplercomponentconfig::Config as FidlConfig;
3use fuchsia_inspect::{ArrayProperty, Node};
4use fuchsia_runtime::{take_startup_handle, HandleInfo, HandleType};
5#[derive(Debug)]
6pub struct Config {
7    pub minimum_sample_rate_sec: i64,
8    pub project_configs: Vec<String>,
9}
10impl Config {
11    pub fn take_from_startup_handle() -> Self {
12        let config_vmo: zx::Vmo =
13            take_startup_handle(HandleInfo::new(HandleType::ComponentConfigVmo, 0))
14                .expect("Config VMO handle must be provided and cannot already have been taken.")
15                .into();
16        let config_size =
17            config_vmo.get_content_size().expect("must be able to read config vmo content size");
18        assert_ne!(config_size, 0, "config vmo must be non-empty");
19        let config_bytes =
20            config_vmo.read_to_vec(0, config_size).expect("must be able to read config vmo");
21        let checksum_length = u16::from_le_bytes([config_bytes[0], config_bytes[1]]) as usize;
22        let fidl_start = 2 + checksum_length;
23        let observed_checksum = &config_bytes[2..fidl_start];
24        let expected_checksum = vec![
25            0xa1, 0xe4, 0xf3, 0xf9, 0xd9, 0x6e, 0x93, 0x5b, 0x56, 0xa0, 0x04, 0xb7, 0xef, 0xdd,
26            0xfc, 0x93, 0x61, 0x8d, 0x11, 0xee, 0x9c, 0xbb, 0x5f, 0x69, 0x31, 0x08, 0x93, 0xfc,
27            0x8c, 0xfd, 0x84, 0x38,
28        ];
29        assert_eq!(
30            observed_checksum, expected_checksum,
31            "checksum from config VMO does not match expected checksum"
32        );
33        let fidl_config: FidlConfig = unpersist(&config_bytes[fidl_start..])
34            .expect("must be able to parse bytes as config FIDL");
35        Self {
36            minimum_sample_rate_sec: fidl_config.minimum_sample_rate_sec,
37            project_configs: fidl_config.project_configs,
38        }
39    }
40    pub fn record_inspect(&self, inspector_node: &Node) {
41        inspector_node.record_int("minimum_sample_rate_sec", self.minimum_sample_rate_sec);
42        let arr = inspector_node.create_string_array("project_configs", self.project_configs.len());
43        for i in 0..self.project_configs.len() {
44            arr.set(i, &self.project_configs[i]);
45        }
46        inspector_node.record(arr);
47    }
48}