pkg_resolver_config/
config_lib_rust_config_lib_source.rs

1use fidl::unpersist;
2use fidl_cf_sc_internal_pkgresolverconfig::Config as FidlConfig;
3use fuchsia_inspect::Node;
4use fuchsia_runtime::{take_startup_handle, HandleInfo, HandleType};
5#[derive(Debug)]
6pub struct Config {
7    pub blob_download_concurrency_limit: u16,
8    pub blob_download_resumption_attempts_limit: u32,
9    pub blob_network_body_timeout_seconds: u32,
10    pub blob_network_header_timeout_seconds: u32,
11    pub delivery_blob_type: u32,
12    pub tuf_metadata_timeout_seconds: u32,
13}
14impl Config {
15    pub fn take_from_startup_handle() -> Self {
16        let config_vmo: zx::Vmo =
17            take_startup_handle(HandleInfo::new(HandleType::ComponentConfigVmo, 0))
18                .expect("Config VMO handle must be provided and cannot already have been taken.")
19                .into();
20        let config_size =
21            config_vmo.get_content_size().expect("must be able to read config vmo content size");
22        assert_ne!(config_size, 0, "config vmo must be non-empty");
23        let config_bytes =
24            config_vmo.read_to_vec(0, config_size).expect("must be able to read config vmo");
25        let checksum_length = u16::from_le_bytes([config_bytes[0], config_bytes[1]]) as usize;
26        let fidl_start = 2 + checksum_length;
27        let observed_checksum = &config_bytes[2..fidl_start];
28        let expected_checksum = vec![
29            0x52, 0x0d, 0xab, 0xe2, 0xc1, 0xed, 0xd2, 0xa4, 0xf9, 0x22, 0xdb, 0xd9, 0xce, 0xf0,
30            0x14, 0x4b, 0x63, 0xd9, 0x3a, 0x26, 0x20, 0xe7, 0x55, 0x74, 0x96, 0x4f, 0x06, 0x07,
31            0x03, 0x48, 0xac, 0x28,
32        ];
33        assert_eq!(
34            observed_checksum, expected_checksum,
35            "checksum from config VMO does not match expected checksum"
36        );
37        let fidl_config: FidlConfig = unpersist(&config_bytes[fidl_start..])
38            .expect("must be able to parse bytes as config FIDL");
39        Self {
40            blob_download_concurrency_limit: fidl_config.blob_download_concurrency_limit,
41            blob_download_resumption_attempts_limit: fidl_config
42                .blob_download_resumption_attempts_limit,
43            blob_network_body_timeout_seconds: fidl_config.blob_network_body_timeout_seconds,
44            blob_network_header_timeout_seconds: fidl_config.blob_network_header_timeout_seconds,
45            delivery_blob_type: fidl_config.delivery_blob_type,
46            tuf_metadata_timeout_seconds: fidl_config.tuf_metadata_timeout_seconds,
47        }
48    }
49    pub fn record_inspect(&self, inspector_node: &Node) {
50        inspector_node.record_uint(
51            "blob_download_concurrency_limit",
52            self.blob_download_concurrency_limit as u64,
53        );
54        inspector_node.record_uint(
55            "blob_download_resumption_attempts_limit",
56            self.blob_download_resumption_attempts_limit as u64,
57        );
58        inspector_node.record_uint(
59            "blob_network_body_timeout_seconds",
60            self.blob_network_body_timeout_seconds as u64,
61        );
62        inspector_node.record_uint(
63            "blob_network_header_timeout_seconds",
64            self.blob_network_header_timeout_seconds as u64,
65        );
66        inspector_node.record_uint("delivery_blob_type", self.delivery_blob_type as u64);
67        inspector_node
68            .record_uint("tuf_metadata_timeout_seconds", self.tuf_metadata_timeout_seconds as u64);
69    }
70}