1use fidl::unpersist;
2use fidl_cf_sc_internal_httpsdateconfig::Config as FidlConfig;
3use fuchsia_inspect::Node;
4use fuchsia_runtime::{take_startup_handle, HandleInfo, HandleType};
5use std::convert::TryInto;
6const EXPECTED_CHECKSUM: &[u8] = &[
7 0x47, 0x9d, 0xa0, 0xef, 0x6a, 0xf6, 0xb8, 0xac, 0x91, 0xd9, 0x20, 0xb6, 0xe9, 0x78, 0x94, 0xb6,
8 0x0a, 0xbf, 0x2c, 0x4f, 0x88, 0x4e, 0xa1, 0xdc, 0x15, 0x86, 0xae, 0x9e, 0x97, 0x2f, 0x1c, 0x6f,
9];
10#[derive(Debug)]
11pub struct Config {
12 pub first_rtt_time_factor: u16,
13 pub https_timeout_sec: u8,
14 pub max_attempts_urgency_high: u32,
15 pub max_attempts_urgency_low: u32,
16 pub max_attempts_urgency_medium: u32,
17 pub num_polls_urgency_high: u32,
18 pub num_polls_urgency_low: u32,
19 pub num_polls_urgency_medium: u32,
20 pub standard_deviation_bound_percentage: u8,
21 pub time_source_endpoint_url: String,
22 pub use_pull_api: bool,
23}
24impl Config {
25 #[doc = r" Take the config startup handle and parse its contents."]
26 #[doc = r""]
27 #[doc = r" # Panics"]
28 #[doc = r""]
29 #[doc = r" If the config startup handle was already taken or if it is not valid."]
30 pub fn take_from_startup_handle() -> Self {
31 let handle_info = HandleInfo::new(HandleType::ComponentConfigVmo, 0);
32 let config_vmo: zx::Vmo =
33 take_startup_handle(handle_info).expect("Config VMO handle must be present.").into();
34 Self::from_vmo(&config_vmo).expect("Config VMO handle must be valid.")
35 }
36 #[doc = r" Parse `Self` from `vmo`."]
37 pub fn from_vmo(vmo: &zx::Vmo) -> Result<Self, Error> {
38 let config_size = vmo.get_content_size().map_err(Error::GettingContentSize)?;
39 let config_bytes = vmo.read_to_vec(0, config_size).map_err(Error::ReadingConfigBytes)?;
40 Self::from_bytes(&config_bytes)
41 }
42 #[doc = r" Parse `Self` from `bytes`."]
43 pub fn from_bytes(bytes: &[u8]) -> Result<Self, Error> {
44 let (checksum_len_bytes, bytes) = bytes.split_at_checked(2).ok_or(Error::TooFewBytes)?;
45 let checksum_len_bytes: [u8; 2] =
46 checksum_len_bytes.try_into().expect("previous call guaranteed 2 element slice");
47 let checksum_length = u16::from_le_bytes(checksum_len_bytes) as usize;
48 let (observed_checksum, bytes) =
49 bytes.split_at_checked(checksum_length).ok_or(Error::TooFewBytes)?;
50 if observed_checksum != EXPECTED_CHECKSUM {
51 return Err(Error::ChecksumMismatch { observed_checksum: observed_checksum.to_vec() });
52 }
53 let fidl_config: FidlConfig = unpersist(bytes).map_err(Error::Unpersist)?;
54 Ok(Self {
55 first_rtt_time_factor: fidl_config.first_rtt_time_factor,
56 https_timeout_sec: fidl_config.https_timeout_sec,
57 max_attempts_urgency_high: fidl_config.max_attempts_urgency_high,
58 max_attempts_urgency_low: fidl_config.max_attempts_urgency_low,
59 max_attempts_urgency_medium: fidl_config.max_attempts_urgency_medium,
60 num_polls_urgency_high: fidl_config.num_polls_urgency_high,
61 num_polls_urgency_low: fidl_config.num_polls_urgency_low,
62 num_polls_urgency_medium: fidl_config.num_polls_urgency_medium,
63 standard_deviation_bound_percentage: fidl_config.standard_deviation_bound_percentage,
64 time_source_endpoint_url: fidl_config.time_source_endpoint_url,
65 use_pull_api: fidl_config.use_pull_api,
66 })
67 }
68 pub fn record_inspect(&self, inspector_node: &Node) {
69 inspector_node.record_uint("first_rtt_time_factor", self.first_rtt_time_factor as u64);
70 inspector_node.record_uint("https_timeout_sec", self.https_timeout_sec as u64);
71 inspector_node
72 .record_uint("max_attempts_urgency_high", self.max_attempts_urgency_high as u64);
73 inspector_node
74 .record_uint("max_attempts_urgency_low", self.max_attempts_urgency_low as u64);
75 inspector_node
76 .record_uint("max_attempts_urgency_medium", self.max_attempts_urgency_medium as u64);
77 inspector_node.record_uint("num_polls_urgency_high", self.num_polls_urgency_high as u64);
78 inspector_node.record_uint("num_polls_urgency_low", self.num_polls_urgency_low as u64);
79 inspector_node
80 .record_uint("num_polls_urgency_medium", self.num_polls_urgency_medium as u64);
81 inspector_node.record_uint(
82 "standard_deviation_bound_percentage",
83 self.standard_deviation_bound_percentage as u64,
84 );
85 inspector_node.record_string("time_source_endpoint_url", &self.time_source_endpoint_url);
86 inspector_node.record_bool("use_pull_api", self.use_pull_api);
87 }
88}
89#[derive(Debug)]
90pub enum Error {
91 #[doc = r" Failed to read the content size of the VMO."]
92 GettingContentSize(zx::Status),
93 #[doc = r" Failed to read the content of the VMO."]
94 ReadingConfigBytes(zx::Status),
95 #[doc = r" The VMO was too small for this config library."]
96 TooFewBytes,
97 #[doc = r" The VMO's config ABI checksum did not match this library's."]
98 ChecksumMismatch { observed_checksum: Vec<u8> },
99 #[doc = r" Failed to parse the non-checksum bytes of the VMO as this library's FIDL type."]
100 Unpersist(fidl::Error),
101}
102impl std::fmt::Display for Error {
103 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
104 match self {
105 Self::GettingContentSize(status) => {
106 write!(f, "Failed to get content size: {status}")
107 }
108 Self::ReadingConfigBytes(status) => {
109 write!(f, "Failed to read VMO content: {status}")
110 }
111 Self::TooFewBytes => {
112 write!(f, "VMO content is not large enough for this config library.")
113 }
114 Self::ChecksumMismatch { observed_checksum } => {
115 write!(
116 f,
117 "ABI checksum mismatch, expected {:?}, got {:?}",
118 EXPECTED_CHECKSUM, observed_checksum,
119 )
120 }
121 Self::Unpersist(fidl_error) => {
122 write!(f, "Failed to parse contents of config VMO: {fidl_error}")
123 }
124 }
125 }
126}
127impl std::error::Error for Error {
128 #[allow(unused_parens, reason = "rustfmt errors without parens here")]
129 fn source(&self) -> Option<(&'_ (dyn std::error::Error + 'static))> {
130 match self {
131 Self::GettingContentSize(ref status) | Self::ReadingConfigBytes(ref status) => {
132 Some(status)
133 }
134 Self::TooFewBytes => None,
135 Self::ChecksumMismatch { .. } => None,
136 Self::Unpersist(ref fidl_error) => Some(fidl_error),
137 }
138 }
139 fn description(&self) -> &str {
140 match self {
141 Self::GettingContentSize(_) => "getting content size",
142 Self::ReadingConfigBytes(_) => "reading VMO contents",
143 Self::TooFewBytes => "VMO contents too small",
144 Self::ChecksumMismatch { .. } => "ABI checksum mismatch",
145 Self::Unpersist(_) => "FIDL parsing error",
146 }
147 }
148}