Skip to main content

starnix_kernel_structured_config/
starnix_kernel_structured_config_rust_config_lib_source.rs

1use fidl::unpersist;
2use fidl_cf_sc_internal_starnixkernelstructuredconfig::Config as FidlConfig;
3use fuchsia_component_config::{Config as ComponentConfig, Error};
4use fuchsia_inspect::{ArrayProperty, Node};
5use std::convert::TryInto;
6const EXPECTED_CHECKSUM: &[u8] = &[
7    0x7a, 0x25, 0x78, 0xf8, 0x90, 0xa6, 0x8d, 0xfc, 0xdd, 0x6d, 0x2f, 0x23, 0xba, 0x41, 0xa6, 0x20,
8    0x7f, 0x47, 0xe6, 0xf4, 0x0b, 0xe6, 0xcc, 0x6e, 0x2c, 0x85, 0x1e, 0x1c, 0x23, 0x27, 0x83, 0xdb,
9];
10const EXPECTED_CHECKSUM_LENGTH: [u8; 2] = (EXPECTED_CHECKSUM.len() as u16).to_le_bytes();
11#[derive(Debug)]
12pub struct Config {
13    pub critical_mb_threshold: u32,
14    pub extra_features: Vec<String>,
15    pub out_of_memory_mb_threshold: u32,
16    pub prefetch: bool,
17    pub warning_mb_threshold: u32,
18}
19impl Config {
20    #[doc = r" Take the config startup handle and parse its contents."]
21    #[doc = r""]
22    #[doc = r" # Panics"]
23    #[doc = r""]
24    #[doc = r" If the config startup handle was already taken or if it is not valid."]
25    pub fn take_from_startup_handle() -> Self {
26        <Self as ComponentConfig>::take_from_startup_handle()
27    }
28    #[doc = r" Parse `Self` from `vmo`."]
29    pub fn from_vmo(vmo: &zx::Vmo) -> Result<Self, Error> {
30        <Self as ComponentConfig>::from_vmo(vmo)
31    }
32    #[doc = r" Parse `Self` from `bytes`."]
33    pub fn from_bytes(bytes: &[u8]) -> Result<Self, Error> {
34        <Self as ComponentConfig>::from_bytes(bytes)
35    }
36    pub fn record_inspect(&self, inspector_node: &Node) {
37        <Self as ComponentConfig>::record_inspect(self, inspector_node)
38    }
39}
40impl ComponentConfig for Config {
41    #[doc = r" Parse `Self` from `bytes`."]
42    fn from_bytes(bytes: &[u8]) -> Result<Self, Error> {
43        let (checksum_len_bytes, bytes) = bytes.split_at_checked(2).ok_or(Error::TooFewBytes)?;
44        let checksum_len_bytes: [u8; 2] =
45            checksum_len_bytes.try_into().expect("previous call guaranteed 2 element slice");
46        let checksum_length = u16::from_le_bytes(checksum_len_bytes) as usize;
47        let (observed_checksum, bytes) =
48            bytes.split_at_checked(checksum_length).ok_or(Error::TooFewBytes)?;
49        if observed_checksum != EXPECTED_CHECKSUM {
50            return Err(Error::ChecksumMismatch {
51                expected_checksum: EXPECTED_CHECKSUM.to_vec(),
52                observed_checksum: observed_checksum.to_vec(),
53            });
54        }
55        let fidl_config: FidlConfig = unpersist(bytes).map_err(Error::Unpersist)?;
56        Ok(Self {
57            critical_mb_threshold: fidl_config.critical_mb_threshold,
58            extra_features: fidl_config.extra_features,
59            out_of_memory_mb_threshold: fidl_config.out_of_memory_mb_threshold,
60            prefetch: fidl_config.prefetch,
61            warning_mb_threshold: fidl_config.warning_mb_threshold,
62        })
63    }
64    fn to_bytes(&self) -> Result<Vec<u8>, Error> {
65        let fidl_config = FidlConfig {
66            critical_mb_threshold: self.critical_mb_threshold.clone(),
67            extra_features: self.extra_features.clone(),
68            out_of_memory_mb_threshold: self.out_of_memory_mb_threshold.clone(),
69            prefetch: self.prefetch.clone(),
70            warning_mb_threshold: self.warning_mb_threshold.clone(),
71        };
72        let mut fidl_bytes = fidl::persist(&fidl_config).map_err(Error::Persist)?;
73        let mut bytes = Vec::with_capacity(
74            EXPECTED_CHECKSUM_LENGTH.len() + EXPECTED_CHECKSUM.len() + fidl_bytes.len(),
75        );
76        bytes.extend_from_slice(&EXPECTED_CHECKSUM_LENGTH);
77        bytes.extend_from_slice(EXPECTED_CHECKSUM);
78        bytes.append(&mut fidl_bytes);
79        Ok(bytes)
80    }
81    fn record_inspect(&self, inspector_node: &Node) {
82        inspector_node.record_uint("critical_mb_threshold", self.critical_mb_threshold as u64);
83        let arr = inspector_node.create_string_array("extra_features", self.extra_features.len());
84        for i in 0..self.extra_features.len() {
85            arr.set(i, &self.extra_features[i]);
86        }
87        inspector_node.record(arr);
88        inspector_node
89            .record_uint("out_of_memory_mb_threshold", self.out_of_memory_mb_threshold as u64);
90        inspector_node.record_bool("prefetch", self.prefetch);
91        inspector_node.record_uint("warning_mb_threshold", self.warning_mb_threshold as u64);
92    }
93}