1 // Copyright 2021, The Android Open Source Project
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 //     http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14 
15 //! Routines for handling payload
16 
17 use crate::instance::ApexData;
18 use crate::ioutil::wait_for_file;
19 use anyhow::{Context, Result};
20 use log::{info, warn};
21 use microdroid_metadata::{read_metadata, ApexPayload, Metadata};
22 use std::time::Duration;
23 
24 const PAYLOAD_METADATA_PATH: &str = "/dev/block/by-name/payload-metadata";
25 const WAIT_TIMEOUT: Duration = Duration::from_secs(10);
26 
27 /// Loads payload metadata from /dev/block/by-name/payload-metadata
load_metadata() -> Result<Metadata>28 pub fn load_metadata() -> Result<Metadata> {
29     info!("loading payload metadata...");
30     let file = wait_for_file(PAYLOAD_METADATA_PATH, WAIT_TIMEOUT)?;
31     read_metadata(file)
32 }
33 
34 /// Loads (name, public_key, root_digest) from payload APEXes
get_apex_data_from_payload(metadata: &Metadata) -> Result<Vec<ApexData>>35 pub fn get_apex_data_from_payload(metadata: &Metadata) -> Result<Vec<ApexData>> {
36     metadata
37         .apexes
38         .iter()
39         .map(|apex| {
40             let apex_path = format!("/dev/block/by-name/{}", apex.partition_name);
41             let extracted =
42                 apexutil::verify(&apex_path).context(format!("Failed to parse {}", &apex_path))?;
43             if let Some(manifest_name) = &extracted.name {
44                 if &apex.name != manifest_name {
45                     warn!("Apex named {} is named {} in its manifest", apex.name, manifest_name);
46                 }
47             };
48             Ok(ApexData {
49                 name: apex.name.clone(),
50                 manifest_name: extracted.name,
51                 manifest_version: extracted.version,
52                 public_key: extracted.public_key,
53                 root_digest: extracted.root_digest,
54                 last_update_seconds: apex.last_update_seconds,
55                 is_factory: apex.is_factory,
56             })
57         })
58         .collect()
59 }
60 
61 /// Convert vector of ApexData into Metadata
to_metadata(apex_data: &[ApexData]) -> Metadata62 pub fn to_metadata(apex_data: &[ApexData]) -> Metadata {
63     Metadata {
64         apexes: apex_data
65             .iter()
66             .map(|data| ApexPayload {
67                 name: data.name.clone(),
68                 public_key: data.public_key.clone(),
69                 root_digest: data.root_digest.clone(),
70                 manifest_name: data.manifest_name.clone().unwrap_or_default(),
71                 manifest_version: data.manifest_version.unwrap_or_default(),
72                 last_update_seconds: data.last_update_seconds,
73                 is_factory: data.is_factory,
74                 ..Default::default()
75             })
76             .collect(),
77         ..Default::default()
78     }
79 }
80