blob: 71d8bcc97e8fa64a389446f004c15fc16c665ec2 [file] [log] [blame]
Alan Stokes16fb8552022-02-10 15:07:27 +00001/*
2 * Copyright (C) 2021 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17//! A tool to verify a CompOS signature. It starts a CompOS VM as part of this to retrieve the
18//! public key. The tool is intended to be run by odsign during boot.
19
Alan Stokes98a964c2022-02-23 11:37:46 +000020use android_logger::LogId;
Alan Stokes16fb8552022-02-10 15:07:27 +000021use anyhow::{bail, Context, Result};
Alan Stokes0e82b502022-08-08 14:44:48 +010022use binder::ProcessState;
Victor Hsieh1732c0d2022-09-12 14:36:03 -070023use clap::{Parser, ValueEnum};
Andrew Walbrand0ef4002022-05-16 16:14:10 +000024use compos_common::compos_client::{ComposClient, VmParameters};
Alan Stokes16fb8552022-02-10 15:07:27 +000025use compos_common::odrefresh::{
Alan Stokes32d8fa52022-03-14 14:59:42 +000026 CURRENT_ARTIFACTS_SUBDIR, ODREFRESH_OUTPUT_ROOT_DIR, PENDING_ARTIFACTS_SUBDIR,
27 TEST_ARTIFACTS_SUBDIR,
Alan Stokes16fb8552022-02-10 15:07:27 +000028};
29use compos_common::{
Alan Stokes6542fdd2022-02-17 15:21:46 +000030 COMPOS_DATA_ROOT, CURRENT_INSTANCE_DIR, IDSIG_FILE, IDSIG_MANIFEST_APK_FILE,
Victor Hsieha61ec2e2022-09-21 16:25:27 -070031 IDSIG_MANIFEST_EXT_APK_FILE, INSTANCE_IMAGE_FILE, TEST_INSTANCE_DIR,
Alan Stokes16fb8552022-02-10 15:07:27 +000032};
33use log::error;
34use std::fs::File;
35use std::io::Read;
36use std::panic;
37use std::path::Path;
38
39const MAX_FILE_SIZE_BYTES: u64 = 100 * 1024;
40
Victor Hsieh1732c0d2022-09-12 14:36:03 -070041#[derive(Parser)]
42struct Args {
43 /// Type of the VM instance
44 #[clap(long, value_enum)]
45 instance: Instance,
46
47 /// Starts the VM in debug mode
48 #[clap(long, action)]
49 debug: bool,
50}
51
52#[derive(ValueEnum, Clone)]
53enum Instance {
54 Current,
55 Pending,
56 Test,
57}
58
Alan Stokes16fb8552022-02-10 15:07:27 +000059fn main() {
60 android_logger::init_once(
61 android_logger::Config::default()
62 .with_tag("compos_verify")
Alan Stokes98a964c2022-02-23 11:37:46 +000063 .with_min_level(log::Level::Info)
64 .with_log_id(LogId::System), // Needed to log successfully early in boot
Alan Stokes16fb8552022-02-10 15:07:27 +000065 );
66
67 // Redirect panic messages to logcat.
68 panic::set_hook(Box::new(|panic_info| {
69 error!("{}", panic_info);
70 }));
71
72 if let Err(e) = try_main() {
73 error!("{:?}", e);
74 std::process::exit(1)
75 }
76}
77
78fn try_main() -> Result<()> {
Victor Hsieh1732c0d2022-09-12 14:36:03 -070079 let args = Args::parse();
80 let (instance_dir, artifacts_dir) = match args.instance {
81 Instance::Current => (CURRENT_INSTANCE_DIR, CURRENT_ARTIFACTS_SUBDIR),
82 Instance::Pending => (CURRENT_INSTANCE_DIR, PENDING_ARTIFACTS_SUBDIR),
83 Instance::Test => (TEST_INSTANCE_DIR, TEST_ARTIFACTS_SUBDIR),
Alan Stokes16fb8552022-02-10 15:07:27 +000084 };
85
86 let instance_dir = Path::new(COMPOS_DATA_ROOT).join(instance_dir);
87 let artifacts_dir = Path::new(ODREFRESH_OUTPUT_ROOT_DIR).join(artifacts_dir);
88
89 if !instance_dir.is_dir() {
90 bail!("{:?} is not a directory", instance_dir);
91 }
92
93 let instance_image = instance_dir.join(INSTANCE_IMAGE_FILE);
94 let idsig = instance_dir.join(IDSIG_FILE);
95 let idsig_manifest_apk = instance_dir.join(IDSIG_MANIFEST_APK_FILE);
Victor Hsieha61ec2e2022-09-21 16:25:27 -070096 let idsig_manifest_ext_apk = instance_dir.join(IDSIG_MANIFEST_EXT_APK_FILE);
Alan Stokes16fb8552022-02-10 15:07:27 +000097
98 let instance_image = File::open(instance_image).context("Failed to open instance image")?;
99
100 let info = artifacts_dir.join("compos.info");
101 let signature = artifacts_dir.join("compos.info.signature");
102
Alan Stokesdd8dfe82022-03-10 15:30:49 +0000103 let info = read_small_file(&info).context("Failed to read compos.info")?;
104 let signature = read_small_file(&signature).context("Failed to read compos.info signature")?;
Alan Stokes16fb8552022-02-10 15:07:27 +0000105
106 // We need to start the thread pool to be able to receive Binder callbacks
107 ProcessState::start_thread_pool();
108
David Brazdil4b4c5102022-12-19 22:56:20 +0000109 let virtmgr = vmclient::VirtualizationService::new()?;
110 let virtualization_service = virtmgr.connect()?;
Andrew Walbrand0ef4002022-05-16 16:14:10 +0000111 let vm_instance = ComposClient::start(
Alan Stokes16fb8552022-02-10 15:07:27 +0000112 &*virtualization_service,
113 instance_image,
114 &idsig,
115 &idsig_manifest_apk,
Victor Hsieha61ec2e2022-09-21 16:25:27 -0700116 &idsig_manifest_ext_apk,
Victor Hsieh1732c0d2022-09-12 14:36:03 -0700117 &VmParameters { debug_mode: args.debug, ..Default::default() },
Alan Stokes16fb8552022-02-10 15:07:27 +0000118 )?;
Alan Stokes16fb8552022-02-10 15:07:27 +0000119
Alan Stokes71403772022-06-21 14:56:28 +0100120 let service = vm_instance.connect_service()?;
121 let public_key = service.getPublicKey().context("Getting public key");
Alan Stokes16fb8552022-02-10 15:07:27 +0000122
Alan Stokes0fc6ce52022-08-02 17:01:48 +0100123 vm_instance.shutdown(service);
Alan Stokes71403772022-06-21 14:56:28 +0100124
125 if !compos_verify_native::verify(&public_key?, &signature, &info) {
Alan Stokes16fb8552022-02-10 15:07:27 +0000126 bail!("Signature verification failed");
127 }
128
129 Ok(())
130}
131
132fn read_small_file(file: &Path) -> Result<Vec<u8>> {
133 let mut file = File::open(file)?;
134 if file.metadata()?.len() > MAX_FILE_SIZE_BYTES {
135 bail!("File is too big");
136 }
137 let mut data = Vec::new();
138 file.read_to_end(&mut data)?;
139 Ok(data)
140}
Andrew Walbranda8786d2022-12-01 14:54:27 +0000141
142#[cfg(test)]
143mod tests {
144 use super::*;
145 use clap::CommandFactory;
146
147 #[test]
148 fn verify_args() {
149 // Check that the command parsing has been configured in a valid way.
150 Args::command().debug_assert();
151 }
152}