system_updater_config/
config_lib_rust_config_lib_source.rs

1use fidl::unpersist;
2use fidl_cf_sc_internal_systemupdaterconfig::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    0x88, 0x7d, 0xa9, 0x90, 0x74, 0x02, 0xde, 0xbf, 0x37, 0xfa, 0x5f, 0xb3, 0xb5, 0xd2, 0xa0, 0xdc,
8    0x2b, 0x16, 0xf2, 0xf7, 0x28, 0x86, 0x67, 0xbd, 0xde, 0x4d, 0xc1, 0xe8, 0xc0, 0x78, 0x3d, 0x81,
9];
10const EXPECTED_CHECKSUM_LENGTH: [u8; 2] = (EXPECTED_CHECKSUM.len() as u16).to_le_bytes();
11#[derive(Debug)]
12pub struct Config {
13    pub concurrent_blob_fetches: u16,
14    pub concurrent_package_resolves: u16,
15    pub excessive_update_duration_seconds: u32,
16    pub manifest_public_keys: Vec<String>,
17    pub verify_existing_blobs: bool,
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            concurrent_blob_fetches: fidl_config.concurrent_blob_fetches,
58            concurrent_package_resolves: fidl_config.concurrent_package_resolves,
59            excessive_update_duration_seconds: fidl_config.excessive_update_duration_seconds,
60            manifest_public_keys: fidl_config.manifest_public_keys,
61            verify_existing_blobs: fidl_config.verify_existing_blobs,
62        })
63    }
64    fn to_bytes(&self) -> Result<Vec<u8>, Error> {
65        let fidl_config = FidlConfig {
66            concurrent_blob_fetches: self.concurrent_blob_fetches.clone(),
67            concurrent_package_resolves: self.concurrent_package_resolves.clone(),
68            excessive_update_duration_seconds: self.excessive_update_duration_seconds.clone(),
69            manifest_public_keys: self.manifest_public_keys.clone(),
70            verify_existing_blobs: self.verify_existing_blobs.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("concurrent_blob_fetches", self.concurrent_blob_fetches as u64);
83        inspector_node
84            .record_uint("concurrent_package_resolves", self.concurrent_package_resolves as u64);
85        inspector_node.record_uint(
86            "excessive_update_duration_seconds",
87            self.excessive_update_duration_seconds as u64,
88        );
89        let arr = inspector_node
90            .create_string_array("manifest_public_keys", self.manifest_public_keys.len());
91        for i in 0..self.manifest_public_keys.len() {
92            arr.set(i, &self.manifest_public_keys[i]);
93        }
94        inspector_node.record(arr);
95        inspector_node.record_bool("verify_existing_blobs", self.verify_existing_blobs);
96    }
97}
system_updater_config/config_lib_rust_config_lib_source.rs

system_updater_config/
config_lib_rust_config_lib_source.rs
