blob: b6df500c2e9a5969e2bfeb524dcdec7c2a5dd91f [file] [log] [blame]
Jooyung Han21e9b922021-06-26 04:14:16 +09001// 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//! Payload disk image
16
Andrew Walbrancc0db522021-07-12 17:03:42 +000017use android_system_virtualizationservice::aidl::android::system::virtualizationservice::{
Alan Stokes0d1ef782022-09-27 13:46:35 +010018 DiskImage::DiskImage,
19 Partition::Partition,
20 VirtualMachineAppConfig::DebugLevel::DebugLevel,
21 VirtualMachineAppConfig::{Payload::Payload, VirtualMachineAppConfig},
Andrew Walbrancc0db522021-07-12 17:03:42 +000022 VirtualMachineRawConfig::VirtualMachineRawConfig,
23};
Inseob Kima5a262f2021-11-17 19:41:03 +090024use anyhow::{anyhow, bail, Context, Result};
Alan Stokes0e82b502022-08-08 14:44:48 +010025use binder::{wait_for_interface, ParcelFileDescriptor};
Alan Stokesbf20c6a2022-01-04 12:30:50 +000026use log::{info, warn};
Alan Stokes0d1ef782022-09-27 13:46:35 +010027use microdroid_metadata::{ApexPayload, ApkPayload, Metadata, PayloadConfig, PayloadMetadata};
Jooyung Han5dc42172021-10-05 16:43:47 +090028use microdroid_payload_config::{ApexConfig, VmPayloadConfig};
Jooyung Han9900f3d2021-07-06 10:27:54 +090029use once_cell::sync::OnceCell;
Jooyung Han5dc42172021-10-05 16:43:47 +090030use packagemanager_aidl::aidl::android::content::pm::IPackageManagerNative::IPackageManagerNative;
Alan Stokesbf20c6a2022-01-04 12:30:50 +000031use regex::Regex;
Jooyung Han44b02ab2021-07-16 03:19:13 +090032use serde::Deserialize;
33use serde_xml_rs::from_reader;
Alan Stokesbf20c6a2022-01-04 12:30:50 +000034use std::collections::HashSet;
Andrew Walbran40be9d52022-01-19 14:32:53 +000035use std::fs::{metadata, File, OpenOptions};
Jooyung Han21e9b922021-06-26 04:14:16 +090036use std::path::{Path, PathBuf};
Alan Stokesbf20c6a2022-01-04 12:30:50 +000037use std::process::Command;
Andrew Walbran40be9d52022-01-19 14:32:53 +000038use std::time::SystemTime;
Andrew Walbrancc0db522021-07-12 17:03:42 +000039use vmconfig::open_parcel_file;
40
41/// The list of APEXes which microdroid requires.
42// TODO(b/192200378) move this to microdroid.json?
Inseob Kimb4868e62021-11-09 17:28:23 +090043const MICRODROID_REQUIRED_APEXES: [&str; 1] = ["com.android.os.statsd"];
44const MICRODROID_REQUIRED_APEXES_DEBUG: [&str; 1] = ["com.android.adbd"];
Jooyung Han21e9b922021-06-26 04:14:16 +090045
Jooyung Han44b02ab2021-07-16 03:19:13 +090046const APEX_INFO_LIST_PATH: &str = "/apex/apex-info-list.xml";
47
Jooyung Han5dc42172021-10-05 16:43:47 +090048const PACKAGE_MANAGER_NATIVE_SERVICE: &str = "package_native";
49
Jooyung Han73bac242021-07-02 10:25:49 +090050/// Represents the list of APEXes
Jooyung Han9d7cd7d2021-10-12 17:44:14 +090051#[derive(Clone, Debug, Deserialize)]
Jooyung Han9900f3d2021-07-06 10:27:54 +090052struct ApexInfoList {
Jooyung Han44b02ab2021-07-16 03:19:13 +090053 #[serde(rename = "apex-info")]
Jooyung Han73bac242021-07-02 10:25:49 +090054 list: Vec<ApexInfo>,
55}
56
Jooyung Han5ce867a2022-01-28 03:18:38 +090057#[derive(Clone, Debug, Default, Deserialize, Eq, PartialEq)]
Jooyung Han73bac242021-07-02 10:25:49 +090058struct ApexInfo {
Jooyung Han44b02ab2021-07-16 03:19:13 +090059 #[serde(rename = "moduleName")]
Jooyung Han73bac242021-07-02 10:25:49 +090060 name: String,
Jooyung Han44b02ab2021-07-16 03:19:13 +090061 #[serde(rename = "modulePath")]
Jooyung Han73bac242021-07-02 10:25:49 +090062 path: PathBuf,
Jooyung Han9d7cd7d2021-10-12 17:44:14 +090063
64 #[serde(default)]
Alan Stokes46ac3862021-12-21 15:31:47 +000065 has_classpath_jar: bool,
Andrew Walbran40be9d52022-01-19 14:32:53 +000066
67 // The field claims to be milliseconds but is actually seconds.
68 #[serde(rename = "lastUpdateMillis")]
69 last_update_seconds: u64,
Jiyong Parkd6502352022-01-27 01:07:30 +090070
71 #[serde(rename = "isFactory")]
72 is_factory: bool,
Jooyung Hanec788042022-01-27 22:28:37 +090073
74 #[serde(rename = "isActive")]
75 is_active: bool,
Jooyung Han5ce867a2022-01-28 03:18:38 +090076
77 #[serde(rename = "provideSharedApexLibs")]
78 provide_shared_apex_libs: bool,
Jooyung Han73bac242021-07-02 10:25:49 +090079}
80
81impl ApexInfoList {
82 /// Loads ApexInfoList
Jooyung Han9900f3d2021-07-06 10:27:54 +090083 fn load() -> Result<&'static ApexInfoList> {
84 static INSTANCE: OnceCell<ApexInfoList> = OnceCell::new();
85 INSTANCE.get_or_try_init(|| {
Jooyung Han44b02ab2021-07-16 03:19:13 +090086 let apex_info_list = File::open(APEX_INFO_LIST_PATH)
87 .context(format!("Failed to open {}", APEX_INFO_LIST_PATH))?;
Jooyung Han9d7cd7d2021-10-12 17:44:14 +090088 let mut apex_info_list: ApexInfoList = from_reader(apex_info_list)
Jooyung Han44b02ab2021-07-16 03:19:13 +090089 .context(format!("Failed to parse {}", APEX_INFO_LIST_PATH))?;
Jooyung Han9d7cd7d2021-10-12 17:44:14 +090090
Alan Stokesbf20c6a2022-01-04 12:30:50 +000091 // For active APEXes, we run derive_classpath and parse its output to see if it
92 // contributes to the classpath(s). (This allows us to handle any new classpath env
93 // vars seamlessly.)
94 let classpath_vars = run_derive_classpath()?;
95 let classpath_apexes = find_apex_names_in_classpath(&classpath_vars)?;
96
Jooyung Han9d7cd7d2021-10-12 17:44:14 +090097 for apex_info in apex_info_list.list.iter_mut() {
Alan Stokesbf20c6a2022-01-04 12:30:50 +000098 apex_info.has_classpath_jar = classpath_apexes.contains(&apex_info.name);
Jooyung Han9d7cd7d2021-10-12 17:44:14 +090099 }
Alan Stokesbf20c6a2022-01-04 12:30:50 +0000100
Jooyung Han44b02ab2021-07-16 03:19:13 +0900101 Ok(apex_info_list)
Jooyung Han9900f3d2021-07-06 10:27:54 +0900102 })
Jooyung Han73bac242021-07-02 10:25:49 +0900103 }
Jooyung Hanec788042022-01-27 22:28:37 +0900104}
Jooyung Han73bac242021-07-02 10:25:49 +0900105
Jooyung Hanec788042022-01-27 22:28:37 +0900106impl ApexInfo {
107 fn matches(&self, apex_config: &ApexConfig) -> bool {
108 // Match with pseudo name "{CLASSPATH}" which represents APEXes contributing
109 // to any derive_classpath environment variable
110 if apex_config.name == "{CLASSPATH}" && self.has_classpath_jar {
111 return true;
112 }
113 if apex_config.name == self.name {
114 return true;
115 }
116 false
Jooyung Han73bac242021-07-02 10:25:49 +0900117 }
Jooyung Han21e9b922021-06-26 04:14:16 +0900118}
119
Jooyung Han5dc42172021-10-05 16:43:47 +0900120struct PackageManager {
Jooyung Han5dc42172021-10-05 16:43:47 +0900121 apex_info_list: &'static ApexInfoList,
122}
123
124impl PackageManager {
125 fn new() -> Result<Self> {
Jooyung Han5dc42172021-10-05 16:43:47 +0900126 let apex_info_list = ApexInfoList::load()?;
Jooyung Han53cf7992021-10-18 19:38:41 +0900127 Ok(Self { apex_info_list })
Jooyung Han5dc42172021-10-05 16:43:47 +0900128 }
129
Jooyung Han9d7cd7d2021-10-12 17:44:14 +0900130 fn get_apex_list(&self, prefer_staged: bool) -> Result<ApexInfoList> {
Jooyung Han53cf7992021-10-18 19:38:41 +0900131 // get the list of active apexes
Jooyung Han9d7cd7d2021-10-12 17:44:14 +0900132 let mut list = self.apex_info_list.clone();
Jooyung Han53cf7992021-10-18 19:38:41 +0900133 // When prefer_staged, we override ApexInfo by consulting "package_native"
Jooyung Han5dc42172021-10-05 16:43:47 +0900134 if prefer_staged {
Jooyung Han53cf7992021-10-18 19:38:41 +0900135 let pm =
136 wait_for_interface::<dyn IPackageManagerNative>(PACKAGE_MANAGER_NATIVE_SERVICE)
137 .context("Failed to get service when prefer_staged is set.")?;
Alan Stokes70ccf162022-07-08 11:05:03 +0100138 let staged =
139 pm.getStagedApexModuleNames().context("getStagedApexModuleNames failed")?;
Jooyung Han9d7cd7d2021-10-12 17:44:14 +0900140 for apex_info in list.list.iter_mut() {
141 if staged.contains(&apex_info.name) {
Alan Stokes70ccf162022-07-08 11:05:03 +0100142 if let Some(staged_apex_info) =
143 pm.getStagedApexInfo(&apex_info.name).context("getStagedApexInfo failed")?
144 {
Jooyung Han9d7cd7d2021-10-12 17:44:14 +0900145 apex_info.path = PathBuf::from(staged_apex_info.diskImagePath);
Alan Stokes46ac3862021-12-21 15:31:47 +0000146 apex_info.has_classpath_jar = staged_apex_info.hasClassPathJars;
Andrew Walbran40be9d52022-01-19 14:32:53 +0000147 let metadata = metadata(&apex_info.path)?;
148 apex_info.last_update_seconds =
149 metadata.modified()?.duration_since(SystemTime::UNIX_EPOCH)?.as_secs();
Jiyong Parkd6502352022-01-27 01:07:30 +0900150 // by definition, staged apex can't be a factory apex.
151 apex_info.is_factory = false;
Jooyung Han9d7cd7d2021-10-12 17:44:14 +0900152 }
153 }
Jooyung Han5dc42172021-10-05 16:43:47 +0900154 }
155 }
Jooyung Han9d7cd7d2021-10-12 17:44:14 +0900156 Ok(list)
Jooyung Han5dc42172021-10-05 16:43:47 +0900157 }
158}
159
Andrew Walbrancc0db522021-07-12 17:03:42 +0000160fn make_metadata_file(
Alan Stokes0d1ef782022-09-27 13:46:35 +0100161 app_config: &VirtualMachineAppConfig,
Jooyung Hanec788042022-01-27 22:28:37 +0900162 apex_infos: &[&ApexInfo],
Jooyung Han21e9b922021-06-26 04:14:16 +0900163 temporary_directory: &Path,
Andrew Walbrancc0db522021-07-12 17:03:42 +0000164) -> Result<ParcelFileDescriptor> {
Alan Stokes0d1ef782022-09-27 13:46:35 +0100165 let payload_metadata = match &app_config.payload {
166 Payload::PayloadConfig(payload_config) => PayloadMetadata::config(PayloadConfig {
167 payload_binary_path: payload_config.payloadPath.clone(),
Alan Stokes0d1ef782022-09-27 13:46:35 +0100168 args: payload_config.args.clone().into(),
169 ..Default::default()
170 }),
171 Payload::ConfigPath(config_path) => {
172 PayloadMetadata::config_path(format!("/mnt/apk/{}", config_path))
173 }
174 };
175
Jooyung Han21e9b922021-06-26 04:14:16 +0900176 let metadata = Metadata {
Andrew Walbrancc0db522021-07-12 17:03:42 +0000177 version: 1,
Jooyung Hanec788042022-01-27 22:28:37 +0900178 apexes: apex_infos
Jooyung Han21e9b922021-06-26 04:14:16 +0900179 .iter()
Jooyung Han19c1d6c2021-08-06 14:08:16 +0900180 .enumerate()
Jooyung Hanec788042022-01-27 22:28:37 +0900181 .map(|(i, apex_info)| {
Andrew Walbran40be9d52022-01-19 14:32:53 +0000182 Ok(ApexPayload {
Jooyung Hanec788042022-01-27 22:28:37 +0900183 name: apex_info.name.clone(),
Andrew Walbran40be9d52022-01-19 14:32:53 +0000184 partition_name: format!("microdroid-apex-{}", i),
Jiyong Parkd6502352022-01-27 01:07:30 +0900185 last_update_seconds: apex_info.last_update_seconds,
186 is_factory: apex_info.is_factory,
Andrew Walbran40be9d52022-01-19 14:32:53 +0000187 ..Default::default()
188 })
Jooyung Han19c1d6c2021-08-06 14:08:16 +0900189 })
Andrew Walbran40be9d52022-01-19 14:32:53 +0000190 .collect::<Result<_>>()?,
Jooyung Han21e9b922021-06-26 04:14:16 +0900191 apk: Some(ApkPayload {
Jooyung Han35edb8f2021-07-01 16:17:16 +0900192 name: "apk".to_owned(),
193 payload_partition_name: "microdroid-apk".to_owned(),
194 idsig_partition_name: "microdroid-apk-idsig".to_owned(),
Jooyung Han21e9b922021-06-26 04:14:16 +0900195 ..Default::default()
196 })
197 .into(),
Alan Stokes0d1ef782022-09-27 13:46:35 +0100198 payload: Some(payload_metadata),
Jooyung Han21e9b922021-06-26 04:14:16 +0900199 ..Default::default()
200 };
Andrew Walbrancc0db522021-07-12 17:03:42 +0000201
202 // Write metadata to file.
Alan Stokes0d1ef782022-09-27 13:46:35 +0100203 let metadata_path = temporary_directory.join("metadata");
Andrew Walbrancc0db522021-07-12 17:03:42 +0000204 let mut metadata_file = OpenOptions::new()
205 .create_new(true)
206 .read(true)
207 .write(true)
208 .open(&metadata_path)
209 .with_context(|| format!("Failed to open metadata file {:?}", metadata_path))?;
Jooyung Han21e9b922021-06-26 04:14:16 +0900210 microdroid_metadata::write_metadata(&metadata, &mut metadata_file)?;
211
Andrew Walbrancc0db522021-07-12 17:03:42 +0000212 // Re-open the metadata file as read-only.
213 open_parcel_file(&metadata_path, false)
214}
215
216/// Creates a DiskImage with partitions:
Alan Stokes53cc5ca2022-08-30 14:28:19 +0100217/// payload-metadata: metadata
Andrew Walbrancc0db522021-07-12 17:03:42 +0000218/// microdroid-apex-0: apex 0
219/// microdroid-apex-1: apex 1
220/// ..
221/// microdroid-apk: apk
222/// microdroid-apk-idsig: idsig
Inseob Kima5a262f2021-11-17 19:41:03 +0900223/// extra-apk-0: additional apk 0
224/// extra-idsig-0: additional idsig 0
225/// extra-apk-1: additional apk 1
226/// extra-idsig-1: additional idsig 1
227/// ..
Andrew Walbrancc0db522021-07-12 17:03:42 +0000228fn make_payload_disk(
Inseob Kima5a262f2021-11-17 19:41:03 +0900229 app_config: &VirtualMachineAppConfig,
Andrew Walbrancc0db522021-07-12 17:03:42 +0000230 apk_file: File,
231 idsig_file: File,
Jooyung Han9d7cd7d2021-10-12 17:44:14 +0900232 vm_payload_config: &VmPayloadConfig,
Andrew Walbrancc0db522021-07-12 17:03:42 +0000233 temporary_directory: &Path,
234) -> Result<DiskImage> {
Inseob Kima5a262f2021-11-17 19:41:03 +0900235 if vm_payload_config.extra_apks.len() != app_config.extraIdsigs.len() {
236 bail!(
237 "payload config has {} apks, but app config has {} idsigs",
238 vm_payload_config.extra_apks.len(),
239 app_config.extraIdsigs.len()
240 );
241 }
242
Jooyung Han9d7cd7d2021-10-12 17:44:14 +0900243 let pm = PackageManager::new()?;
244 let apex_list = pm.get_apex_list(vm_payload_config.prefer_staged)?;
245
Jooyung Hanec788042022-01-27 22:28:37 +0900246 // collect APEXes from config
247 let apex_infos =
248 collect_apex_infos(&apex_list, &vm_payload_config.apexes, app_config.debugLevel);
249 info!("Microdroid payload APEXes: {:?}", apex_infos.iter().map(|ai| &ai.name));
Jooyung Han9d7cd7d2021-10-12 17:44:14 +0900250
Alan Stokes0d1ef782022-09-27 13:46:35 +0100251 let metadata_file = make_metadata_file(app_config, &apex_infos, temporary_directory)?;
Jooyung Han21e9b922021-06-26 04:14:16 +0900252 // put metadata at the first partition
253 let mut partitions = vec![Partition {
Jooyung Han14e5a8e2021-07-06 20:48:38 +0900254 label: "payload-metadata".to_owned(),
Jooyung Han631d5882021-07-29 06:34:05 +0900255 image: Some(metadata_file),
Jooyung Han21e9b922021-06-26 04:14:16 +0900256 writable: false,
257 }];
258
Jooyung Hanec788042022-01-27 22:28:37 +0900259 for (i, apex_info) in apex_infos.iter().enumerate() {
260 let apex_file = open_parcel_file(&apex_info.path, false)?;
Jooyung Han95884632021-07-06 22:27:54 +0900261 partitions.push(Partition {
262 label: format!("microdroid-apex-{}", i),
Jooyung Han631d5882021-07-29 06:34:05 +0900263 image: Some(apex_file),
Jooyung Han95884632021-07-06 22:27:54 +0900264 writable: false,
265 });
Jooyung Han21e9b922021-06-26 04:14:16 +0900266 }
Jooyung Han95884632021-07-06 22:27:54 +0900267 partitions.push(Partition {
268 label: "microdroid-apk".to_owned(),
Jooyung Han631d5882021-07-29 06:34:05 +0900269 image: Some(ParcelFileDescriptor::new(apk_file)),
Jooyung Han95884632021-07-06 22:27:54 +0900270 writable: false,
271 });
272 partitions.push(Partition {
273 label: "microdroid-apk-idsig".to_owned(),
Jooyung Han631d5882021-07-29 06:34:05 +0900274 image: Some(ParcelFileDescriptor::new(idsig_file)),
Jooyung Han95884632021-07-06 22:27:54 +0900275 writable: false,
276 });
Jooyung Han21e9b922021-06-26 04:14:16 +0900277
Inseob Kima5a262f2021-11-17 19:41:03 +0900278 // we've already checked that extra_apks and extraIdsigs are in the same size.
279 let extra_apks = &vm_payload_config.extra_apks;
280 let extra_idsigs = &app_config.extraIdsigs;
281 for (i, (extra_apk, extra_idsig)) in extra_apks.iter().zip(extra_idsigs.iter()).enumerate() {
282 partitions.push(Partition {
283 label: format!("extra-apk-{}", i),
Victor Hsieh14497f02022-09-21 10:10:16 -0700284 image: Some(ParcelFileDescriptor::new(
285 File::open(PathBuf::from(&extra_apk.path)).with_context(|| {
286 format!("Failed to open the extra apk #{} {}", i, extra_apk.path)
287 })?,
288 )),
Inseob Kima5a262f2021-11-17 19:41:03 +0900289 writable: false,
290 });
291
292 partitions.push(Partition {
293 label: format!("extra-idsig-{}", i),
Victor Hsieh14497f02022-09-21 10:10:16 -0700294 image: Some(ParcelFileDescriptor::new(
295 extra_idsig
296 .as_ref()
297 .try_clone()
298 .with_context(|| format!("Failed to clone the extra idsig #{}", i))?,
299 )),
Inseob Kima5a262f2021-11-17 19:41:03 +0900300 writable: false,
301 });
302 }
303
Jooyung Han21e9b922021-06-26 04:14:16 +0900304 Ok(DiskImage { image: None, partitions, writable: false })
305}
Andrew Walbrancc0db522021-07-12 17:03:42 +0000306
Alan Stokesbf20c6a2022-01-04 12:30:50 +0000307fn run_derive_classpath() -> Result<String> {
308 let result = Command::new("/apex/com.android.sdkext/bin/derive_classpath")
309 .arg("/proc/self/fd/1")
310 .output()
311 .context("Failed to run derive_classpath")?;
312
313 if !result.status.success() {
314 bail!("derive_classpath returned {}", result.status);
315 }
316
317 String::from_utf8(result.stdout).context("Converting derive_classpath output")
318}
319
320fn find_apex_names_in_classpath(classpath_vars: &str) -> Result<HashSet<String>> {
321 // Each line should be in the format "export <var name> <paths>", where <paths> is a
322 // colon-separated list of paths to JARs. We don't care about the var names, and we're only
323 // interested in paths that look like "/apex/<apex name>/<anything>" so we know which APEXes
324 // contribute to at least one var.
325 let mut apexes = HashSet::new();
326
327 let pattern = Regex::new(r"^export [^ ]+ ([^ ]+)$").context("Failed to construct Regex")?;
328 for line in classpath_vars.lines() {
329 if let Some(captures) = pattern.captures(line) {
330 if let Some(paths) = captures.get(1) {
331 apexes.extend(paths.as_str().split(':').filter_map(|path| {
332 let path = path.strip_prefix("/apex/")?;
333 Some(path[..path.find('/')?].to_owned())
334 }));
335 continue;
336 }
337 }
338 warn!("Malformed line from derive_classpath: {}", line);
339 }
340
341 Ok(apexes)
Jooyung Han5e0f2062021-10-12 14:00:46 +0900342}
343
Jooyung Hanec788042022-01-27 22:28:37 +0900344// Collect ApexInfos from VM config
345fn collect_apex_infos<'a>(
346 apex_list: &'a ApexInfoList,
347 apex_configs: &[ApexConfig],
Inseob Kimb4868e62021-11-09 17:28:23 +0900348 debug_level: DebugLevel,
Jooyung Hanec788042022-01-27 22:28:37 +0900349) -> Vec<&'a ApexInfo> {
350 let mut additional_apexes: Vec<&str> = MICRODROID_REQUIRED_APEXES.to_vec();
Inseob Kimb4868e62021-11-09 17:28:23 +0900351 if debug_level != DebugLevel::NONE {
Jooyung Hanec788042022-01-27 22:28:37 +0900352 additional_apexes.extend(MICRODROID_REQUIRED_APEXES_DEBUG.to_vec());
Inseob Kimb4868e62021-11-09 17:28:23 +0900353 }
Jooyung Hanec788042022-01-27 22:28:37 +0900354
355 apex_list
356 .list
357 .iter()
358 .filter(|ai| {
359 apex_configs.iter().any(|cfg| ai.matches(cfg) && ai.is_active)
360 || additional_apexes.iter().any(|name| name == &ai.name && ai.is_active)
Jooyung Han5ce867a2022-01-28 03:18:38 +0900361 || ai.provide_shared_apex_libs
Jooyung Hanec788042022-01-27 22:28:37 +0900362 })
363 .collect()
Jooyung Han5e0f2062021-10-12 14:00:46 +0900364}
365
Shikha Panwar22e70452022-10-10 18:32:55 +0000366pub fn add_microdroid_system_images(
Andrew Walbrancc0db522021-07-12 17:03:42 +0000367 config: &VirtualMachineAppConfig,
Jiyong Park8d081812021-07-23 17:45:04 +0900368 instance_file: File,
Shikha Panwar22e70452022-10-10 18:32:55 +0000369 storage_image: Option<File>,
Andrew Walbrancc0db522021-07-12 17:03:42 +0000370 vm_config: &mut VirtualMachineRawConfig,
371) -> Result<()> {
Shikha Panwared8ace42022-09-28 12:52:16 +0000372 let debug_suffix = match config.debugLevel {
373 DebugLevel::NONE => "normal",
374 DebugLevel::APP_ONLY => "app_debuggable",
375 DebugLevel::FULL => "full_debuggable",
376 _ => return Err(anyhow!("unsupported debug level: {:?}", config.debugLevel)),
377 };
378 let initrd = format!("/apex/com.android.virt/etc/microdroid_initrd_{}.img", debug_suffix);
379 vm_config.initrd = Some(open_parcel_file(Path::new(&initrd), false)?);
380
Shikha Panwar22e70452022-10-10 18:32:55 +0000381 let mut writable_partitions = vec![Partition {
Shikha Panwared8ace42022-09-28 12:52:16 +0000382 label: "vm-instance".to_owned(),
383 image: Some(ParcelFileDescriptor::new(instance_file)),
384 writable: true,
Shikha Panwar22e70452022-10-10 18:32:55 +0000385 }];
386
387 if let Some(file) = storage_image {
388 writable_partitions.push(Partition {
389 label: "encrypted-storage".to_owned(),
390 image: Some(ParcelFileDescriptor::new(file)),
391 writable: true,
392 });
393 }
394
395 vm_config.disks.push(DiskImage {
396 image: None,
397 partitions: writable_partitions,
398 writable: true,
399 });
400
401 Ok(())
402}
403
404pub fn add_microdroid_payload_images(
405 config: &VirtualMachineAppConfig,
406 temporary_directory: &Path,
407 apk_file: File,
408 idsig_file: File,
409 vm_payload_config: &VmPayloadConfig,
410 vm_config: &mut VirtualMachineRawConfig,
411) -> Result<()> {
Andrew Walbrancc0db522021-07-12 17:03:42 +0000412 vm_config.disks.push(make_payload_disk(
Inseob Kima5a262f2021-11-17 19:41:03 +0900413 config,
Andrew Walbrancc0db522021-07-12 17:03:42 +0000414 apk_file,
415 idsig_file,
Jooyung Han9d7cd7d2021-10-12 17:44:14 +0900416 vm_payload_config,
Andrew Walbrancc0db522021-07-12 17:03:42 +0000417 temporary_directory,
418 )?);
419
Andrew Walbrancc0db522021-07-12 17:03:42 +0000420 Ok(())
421}
Jooyung Han5e0f2062021-10-12 14:00:46 +0900422
423#[cfg(test)]
424mod tests {
425 use super::*;
Andrew Walbranc1a5f5a2022-01-19 13:38:13 +0000426
Jooyung Han5e0f2062021-10-12 14:00:46 +0900427 #[test]
Alan Stokesbf20c6a2022-01-04 12:30:50 +0000428 fn test_find_apex_names_in_classpath() {
429 let vars = r#"
430export FOO /apex/unterminated
431export BAR /apex/valid.apex/something
432wrong
433export EMPTY
434export OTHER /foo/bar:/baz:/apex/second.valid.apex/:gibberish:"#;
435 let expected = vec!["valid.apex", "second.valid.apex"];
436 let expected: HashSet<_> = expected.into_iter().map(ToString::to_string).collect();
437
438 assert_eq!(find_apex_names_in_classpath(vars).unwrap(), expected);
Jooyung Han5e0f2062021-10-12 14:00:46 +0900439 }
Andrew Walbranc1a5f5a2022-01-19 13:38:13 +0000440
441 #[test]
Jooyung Hanec788042022-01-27 22:28:37 +0900442 fn test_collect_apexes() {
443 let apex_info_list = ApexInfoList {
Andrew Walbranc1a5f5a2022-01-19 13:38:13 +0000444 list: vec![
445 ApexInfo {
Jooyung Hanec788042022-01-27 22:28:37 +0900446 // 0
447 name: "com.android.adbd".to_string(),
448 path: PathBuf::from("adbd"),
Andrew Walbranc1a5f5a2022-01-19 13:38:13 +0000449 has_classpath_jar: false,
Andrew Walbran40be9d52022-01-19 14:32:53 +0000450 last_update_seconds: 12345678,
Jiyong Parkd6502352022-01-27 01:07:30 +0900451 is_factory: true,
Jooyung Hanec788042022-01-27 22:28:37 +0900452 is_active: true,
Jooyung Han5ce867a2022-01-28 03:18:38 +0900453 ..Default::default()
Andrew Walbranc1a5f5a2022-01-19 13:38:13 +0000454 },
455 ApexInfo {
Jooyung Hanec788042022-01-27 22:28:37 +0900456 // 1
457 name: "com.android.os.statsd".to_string(),
458 path: PathBuf::from("statsd"),
459 has_classpath_jar: false,
460 last_update_seconds: 12345678,
461 is_factory: true,
462 is_active: false,
Jooyung Han5ce867a2022-01-28 03:18:38 +0900463 ..Default::default()
Jooyung Hanec788042022-01-27 22:28:37 +0900464 },
465 ApexInfo {
466 // 2
467 name: "com.android.os.statsd".to_string(),
468 path: PathBuf::from("statsd/updated"),
469 has_classpath_jar: false,
470 last_update_seconds: 12345678 + 1,
471 is_factory: false,
472 is_active: true,
Jooyung Han5ce867a2022-01-28 03:18:38 +0900473 ..Default::default()
Jooyung Hanec788042022-01-27 22:28:37 +0900474 },
475 ApexInfo {
476 // 3
477 name: "no_classpath".to_string(),
478 path: PathBuf::from("no_classpath"),
479 has_classpath_jar: false,
480 last_update_seconds: 12345678,
481 is_factory: true,
482 is_active: true,
Jooyung Han5ce867a2022-01-28 03:18:38 +0900483 ..Default::default()
Jooyung Hanec788042022-01-27 22:28:37 +0900484 },
485 ApexInfo {
486 // 4
Andrew Walbranc1a5f5a2022-01-19 13:38:13 +0000487 name: "has_classpath".to_string(),
Jooyung Hanec788042022-01-27 22:28:37 +0900488 path: PathBuf::from("has_classpath"),
Andrew Walbranc1a5f5a2022-01-19 13:38:13 +0000489 has_classpath_jar: true,
Andrew Walbran40be9d52022-01-19 14:32:53 +0000490 last_update_seconds: 87654321,
Jooyung Hanec788042022-01-27 22:28:37 +0900491 is_factory: true,
492 is_active: false,
Jooyung Han5ce867a2022-01-28 03:18:38 +0900493 ..Default::default()
Jooyung Hanec788042022-01-27 22:28:37 +0900494 },
495 ApexInfo {
496 // 5
497 name: "has_classpath".to_string(),
498 path: PathBuf::from("has_classpath/updated"),
499 has_classpath_jar: true,
500 last_update_seconds: 87654321 + 1,
Jiyong Parkd6502352022-01-27 01:07:30 +0900501 is_factory: false,
Jooyung Hanec788042022-01-27 22:28:37 +0900502 is_active: true,
Jooyung Han5ce867a2022-01-28 03:18:38 +0900503 ..Default::default()
Jooyung Hanec788042022-01-27 22:28:37 +0900504 },
505 ApexInfo {
506 // 6
507 name: "apex-foo".to_string(),
508 path: PathBuf::from("apex-foo"),
509 has_classpath_jar: false,
510 last_update_seconds: 87654321,
511 is_factory: true,
512 is_active: false,
Jooyung Han5ce867a2022-01-28 03:18:38 +0900513 ..Default::default()
Jooyung Hanec788042022-01-27 22:28:37 +0900514 },
515 ApexInfo {
516 // 7
517 name: "apex-foo".to_string(),
518 path: PathBuf::from("apex-foo/updated"),
519 has_classpath_jar: false,
520 last_update_seconds: 87654321 + 1,
521 is_factory: false,
522 is_active: true,
Jooyung Han5ce867a2022-01-28 03:18:38 +0900523 ..Default::default()
524 },
525 ApexInfo {
526 // 8
527 name: "sharedlibs".to_string(),
528 path: PathBuf::from("apex-foo"),
529 last_update_seconds: 87654321,
530 is_factory: true,
531 provide_shared_apex_libs: true,
532 ..Default::default()
533 },
534 ApexInfo {
535 // 9
536 name: "sharedlibs".to_string(),
537 path: PathBuf::from("apex-foo/updated"),
538 last_update_seconds: 87654321 + 1,
539 is_active: true,
540 provide_shared_apex_libs: true,
541 ..Default::default()
Andrew Walbranc1a5f5a2022-01-19 13:38:13 +0000542 },
543 ],
544 };
Jooyung Hanec788042022-01-27 22:28:37 +0900545 let apex_configs = vec![
546 ApexConfig { name: "apex-foo".to_string() },
Andrew Walbranc1a5f5a2022-01-19 13:38:13 +0000547 ApexConfig { name: "{CLASSPATH}".to_string() },
548 ];
549 assert_eq!(
Jooyung Hanec788042022-01-27 22:28:37 +0900550 collect_apex_infos(&apex_info_list, &apex_configs, DebugLevel::FULL),
Andrew Walbranc1a5f5a2022-01-19 13:38:13 +0000551 vec![
Jooyung Han5ce867a2022-01-28 03:18:38 +0900552 // Pass active/required APEXes
Jooyung Hanec788042022-01-27 22:28:37 +0900553 &apex_info_list.list[0],
554 &apex_info_list.list[2],
Jooyung Han5ce867a2022-01-28 03:18:38 +0900555 // Pass active APEXes specified in the config
Jooyung Hanec788042022-01-27 22:28:37 +0900556 &apex_info_list.list[5],
557 &apex_info_list.list[7],
Jooyung Han5ce867a2022-01-28 03:18:38 +0900558 // Pass both preinstalled(inactive) and updated(active) for "sharedlibs" APEXes
559 &apex_info_list.list[8],
560 &apex_info_list.list[9],
Andrew Walbranc1a5f5a2022-01-19 13:38:13 +0000561 ]
562 );
563 }
Jooyung Han5e0f2062021-10-12 14:00:46 +0900564}