1use fidl::unpersist;
2use fidl_cf_sc_internal_fshostconfig::Config as FidlConfig;
3use fuchsia_inspect::Node;
4use fuchsia_runtime::{take_startup_handle, HandleInfo, HandleType};
5use std::convert::TryInto;
6const EXPECTED_CHECKSUM: &[u8] = &[
7 0x63, 0x1f, 0xe3, 0xc7, 0xee, 0x84, 0x00, 0xd4, 0x1d, 0x76, 0xf6, 0x82, 0x0f, 0x9d, 0x3d, 0x86,
8 0xdc, 0x1b, 0x45, 0x0c, 0x06, 0xc7, 0x6b, 0x70, 0x53, 0xd4, 0x82, 0xe8, 0x66, 0x7c, 0xf5, 0x44,
9];
10#[derive(Debug)]
11pub struct Config {
12 pub blobfs: bool,
13 pub blobfs_cache_eviction_policy: String,
14 pub blobfs_initial_inodes: u64,
15 pub blobfs_max_bytes: u64,
16 pub blobfs_use_deprecated_padded_format: bool,
17 pub blobfs_write_compression_algorithm: String,
18 pub bootpart: bool,
19 pub check_filesystems: bool,
20 pub data: bool,
21 pub data_filesystem_format: String,
22 pub data_max_bytes: u64,
23 pub disable_automount: bool,
24 pub disable_block_watcher: bool,
25 pub factory: bool,
26 pub format_data_on_corruption: bool,
27 pub fvm: bool,
28 pub fvm_slice_size: u64,
29 pub fxfs_blob: bool,
30 pub fxfs_crypt_url: String,
31 pub gpt: bool,
32 pub gpt_all: bool,
33 pub mbr: bool,
34 pub nand: bool,
35 pub netboot: bool,
36 pub no_zxcrypt: bool,
37 pub ramdisk_image: bool,
38 pub starnix_volume_name: String,
39 pub storage_host: bool,
40 pub use_disk_migration: bool,
41}
42impl Config {
43 #[doc = r" Take the config startup handle and parse its contents."]
44 #[doc = r""]
45 #[doc = r" # Panics"]
46 #[doc = r""]
47 #[doc = r" If the config startup handle was already taken or if it is not valid."]
48 pub fn take_from_startup_handle() -> Self {
49 let handle_info = HandleInfo::new(HandleType::ComponentConfigVmo, 0);
50 let config_vmo: zx::Vmo =
51 take_startup_handle(handle_info).expect("Config VMO handle must be present.").into();
52 Self::from_vmo(&config_vmo).expect("Config VMO handle must be valid.")
53 }
54 #[doc = r" Parse `Self` from `vmo`."]
55 pub fn from_vmo(vmo: &zx::Vmo) -> Result<Self, Error> {
56 let config_size = vmo.get_content_size().map_err(Error::GettingContentSize)?;
57 let config_bytes = vmo.read_to_vec(0, config_size).map_err(Error::ReadingConfigBytes)?;
58 Self::from_bytes(&config_bytes)
59 }
60 #[doc = r" Parse `Self` from `bytes`."]
61 pub fn from_bytes(bytes: &[u8]) -> Result<Self, Error> {
62 let (checksum_len_bytes, bytes) = bytes.split_at_checked(2).ok_or(Error::TooFewBytes)?;
63 let checksum_len_bytes: [u8; 2] =
64 checksum_len_bytes.try_into().expect("previous call guaranteed 2 element slice");
65 let checksum_length = u16::from_le_bytes(checksum_len_bytes) as usize;
66 let (observed_checksum, bytes) =
67 bytes.split_at_checked(checksum_length).ok_or(Error::TooFewBytes)?;
68 if observed_checksum != EXPECTED_CHECKSUM {
69 return Err(Error::ChecksumMismatch { observed_checksum: observed_checksum.to_vec() });
70 }
71 let fidl_config: FidlConfig = unpersist(bytes).map_err(Error::Unpersist)?;
72 Ok(Self {
73 blobfs: fidl_config.blobfs,
74 blobfs_cache_eviction_policy: fidl_config.blobfs_cache_eviction_policy,
75 blobfs_initial_inodes: fidl_config.blobfs_initial_inodes,
76 blobfs_max_bytes: fidl_config.blobfs_max_bytes,
77 blobfs_use_deprecated_padded_format: fidl_config.blobfs_use_deprecated_padded_format,
78 blobfs_write_compression_algorithm: fidl_config.blobfs_write_compression_algorithm,
79 bootpart: fidl_config.bootpart,
80 check_filesystems: fidl_config.check_filesystems,
81 data: fidl_config.data,
82 data_filesystem_format: fidl_config.data_filesystem_format,
83 data_max_bytes: fidl_config.data_max_bytes,
84 disable_automount: fidl_config.disable_automount,
85 disable_block_watcher: fidl_config.disable_block_watcher,
86 factory: fidl_config.factory,
87 format_data_on_corruption: fidl_config.format_data_on_corruption,
88 fvm: fidl_config.fvm,
89 fvm_slice_size: fidl_config.fvm_slice_size,
90 fxfs_blob: fidl_config.fxfs_blob,
91 fxfs_crypt_url: fidl_config.fxfs_crypt_url,
92 gpt: fidl_config.gpt,
93 gpt_all: fidl_config.gpt_all,
94 mbr: fidl_config.mbr,
95 nand: fidl_config.nand,
96 netboot: fidl_config.netboot,
97 no_zxcrypt: fidl_config.no_zxcrypt,
98 ramdisk_image: fidl_config.ramdisk_image,
99 starnix_volume_name: fidl_config.starnix_volume_name,
100 storage_host: fidl_config.storage_host,
101 use_disk_migration: fidl_config.use_disk_migration,
102 })
103 }
104 pub fn record_inspect(&self, inspector_node: &Node) {
105 inspector_node.record_bool("blobfs", self.blobfs);
106 inspector_node
107 .record_string("blobfs_cache_eviction_policy", &self.blobfs_cache_eviction_policy);
108 inspector_node.record_uint("blobfs_initial_inodes", self.blobfs_initial_inodes);
109 inspector_node.record_uint("blobfs_max_bytes", self.blobfs_max_bytes);
110 inspector_node.record_bool(
111 "blobfs_use_deprecated_padded_format",
112 self.blobfs_use_deprecated_padded_format,
113 );
114 inspector_node.record_string(
115 "blobfs_write_compression_algorithm",
116 &self.blobfs_write_compression_algorithm,
117 );
118 inspector_node.record_bool("bootpart", self.bootpart);
119 inspector_node.record_bool("check_filesystems", self.check_filesystems);
120 inspector_node.record_bool("data", self.data);
121 inspector_node.record_string("data_filesystem_format", &self.data_filesystem_format);
122 inspector_node.record_uint("data_max_bytes", self.data_max_bytes);
123 inspector_node.record_bool("disable_automount", self.disable_automount);
124 inspector_node.record_bool("disable_block_watcher", self.disable_block_watcher);
125 inspector_node.record_bool("factory", self.factory);
126 inspector_node.record_bool("format_data_on_corruption", self.format_data_on_corruption);
127 inspector_node.record_bool("fvm", self.fvm);
128 inspector_node.record_uint("fvm_slice_size", self.fvm_slice_size);
129 inspector_node.record_bool("fxfs_blob", self.fxfs_blob);
130 inspector_node.record_string("fxfs_crypt_url", &self.fxfs_crypt_url);
131 inspector_node.record_bool("gpt", self.gpt);
132 inspector_node.record_bool("gpt_all", self.gpt_all);
133 inspector_node.record_bool("mbr", self.mbr);
134 inspector_node.record_bool("nand", self.nand);
135 inspector_node.record_bool("netboot", self.netboot);
136 inspector_node.record_bool("no_zxcrypt", self.no_zxcrypt);
137 inspector_node.record_bool("ramdisk_image", self.ramdisk_image);
138 inspector_node.record_string("starnix_volume_name", &self.starnix_volume_name);
139 inspector_node.record_bool("storage_host", self.storage_host);
140 inspector_node.record_bool("use_disk_migration", self.use_disk_migration);
141 }
142}
143#[derive(Debug)]
144pub enum Error {
145 #[doc = r" Failed to read the content size of the VMO."]
146 GettingContentSize(zx::Status),
147 #[doc = r" Failed to read the content of the VMO."]
148 ReadingConfigBytes(zx::Status),
149 #[doc = r" The VMO was too small for this config library."]
150 TooFewBytes,
151 #[doc = r" The VMO's config ABI checksum did not match this library's."]
152 ChecksumMismatch { observed_checksum: Vec<u8> },
153 #[doc = r" Failed to parse the non-checksum bytes of the VMO as this library's FIDL type."]
154 Unpersist(fidl::Error),
155}
156impl std::fmt::Display for Error {
157 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
158 match self {
159 Self::GettingContentSize(status) => {
160 write!(f, "Failed to get content size: {status}")
161 }
162 Self::ReadingConfigBytes(status) => {
163 write!(f, "Failed to read VMO content: {status}")
164 }
165 Self::TooFewBytes => {
166 write!(f, "VMO content is not large enough for this config library.")
167 }
168 Self::ChecksumMismatch { observed_checksum } => {
169 write!(
170 f,
171 "ABI checksum mismatch, expected {:?}, got {:?}",
172 EXPECTED_CHECKSUM, observed_checksum,
173 )
174 }
175 Self::Unpersist(fidl_error) => {
176 write!(f, "Failed to parse contents of config VMO: {fidl_error}")
177 }
178 }
179 }
180}
181impl std::error::Error for Error {
182 #[allow(unused_parens, reason = "rustfmt errors without parens here")]
183 fn source(&self) -> Option<(&'_ (dyn std::error::Error + 'static))> {
184 match self {
185 Self::GettingContentSize(ref status) | Self::ReadingConfigBytes(ref status) => {
186 Some(status)
187 }
188 Self::TooFewBytes => None,
189 Self::ChecksumMismatch { .. } => None,
190 Self::Unpersist(ref fidl_error) => Some(fidl_error),
191 }
192 }
193 fn description(&self) -> &str {
194 match self {
195 Self::GettingContentSize(_) => "getting content size",
196 Self::ReadingConfigBytes(_) => "reading VMO contents",
197 Self::TooFewBytes => "VMO contents too small",
198 Self::ChecksumMismatch { .. } => "ABI checksum mismatch",
199 Self::Unpersist(_) => "FIDL parsing error",
200 }
201 }
202}