blob: 3116456ec281824ca8a424dc9b14c5205b86f041 [file] [log] [blame]
Pierre-Clément Tosi4f4f5eb2022-12-08 14:31:42 +00001// Copyright 2022, 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//! Support for DICE derivation and BCC generation.
16
Jiyong Parkb87f3302023-03-21 10:03:11 +090017use crate::cstr;
Pierre-Clément Tosibec84662023-01-04 14:25:33 +000018use core::ffi::c_void;
Alice Wang1f0add02023-01-23 16:22:53 +000019use core::mem::size_of;
Pierre-Clément Tosibec84662023-01-04 14:25:33 +000020use core::slice;
Alice Wang3fa9b802023-06-06 07:52:31 +000021use vmbase::memory::flushed_zeroize;
Pierre-Clément Tosif58f3a32023-02-02 16:24:23 +000022
Alice Wang843d8312023-02-15 09:47:06 +000023use diced_open_dice::{
Alan Stokesc4354b82023-05-04 16:06:52 +010024 bcc_format_config_descriptor, bcc_handover_main_flow, hash, Config, DiceMode, Hash,
25 InputValues, HIDDEN_SIZE,
Alice Wang843d8312023-02-15 09:47:06 +000026};
Alice Wang1f0add02023-01-23 16:22:53 +000027use pvmfw_avb::{DebugLevel, Digest, VerifiedBootData};
28
Alice Wang31226132023-01-31 12:44:39 +000029fn to_dice_mode(debug_level: DebugLevel) -> DiceMode {
Alice Wang1f0add02023-01-23 16:22:53 +000030 match debug_level {
Alice Wang31226132023-01-31 12:44:39 +000031 DebugLevel::None => DiceMode::kDiceModeNormal,
32 DebugLevel::Full => DiceMode::kDiceModeDebug,
Alice Wang1f0add02023-01-23 16:22:53 +000033 }
34}
35
Alice Wang843d8312023-02-15 09:47:06 +000036fn to_dice_hash(verified_boot_data: &VerifiedBootData) -> diced_open_dice::Result<Hash> {
Alice Wang1f0add02023-01-23 16:22:53 +000037 let mut digests = [0u8; size_of::<Digest>() * 2];
38 digests[..size_of::<Digest>()].copy_from_slice(&verified_boot_data.kernel_digest);
39 if let Some(initrd_digest) = verified_boot_data.initrd_digest {
40 digests[size_of::<Digest>()..].copy_from_slice(&initrd_digest);
41 }
42 hash(&digests)
43}
Pierre-Clément Tosi4f4f5eb2022-12-08 14:31:42 +000044
Pierre-Clément Tosif58f3a32023-02-02 16:24:23 +000045pub struct PartialInputs {
Pierre-Clément Tosi1cc5eb72023-02-02 11:09:18 +000046 pub code_hash: Hash,
47 pub auth_hash: Hash,
48 pub mode: DiceMode,
Pierre-Clément Tosif58f3a32023-02-02 16:24:23 +000049}
Pierre-Clément Tosi4f4f5eb2022-12-08 14:31:42 +000050
Pierre-Clément Tosif58f3a32023-02-02 16:24:23 +000051impl PartialInputs {
Alice Wang843d8312023-02-15 09:47:06 +000052 pub fn new(data: &VerifiedBootData) -> diced_open_dice::Result<Self> {
Pierre-Clément Tosif58f3a32023-02-02 16:24:23 +000053 let code_hash = to_dice_hash(data)?;
54 let auth_hash = hash(data.public_key)?;
55 let mode = to_dice_mode(data.debug_level);
Pierre-Clément Tosi4f4f5eb2022-12-08 14:31:42 +000056
Pierre-Clément Tosif58f3a32023-02-02 16:24:23 +000057 Ok(Self { code_hash, auth_hash, mode })
58 }
59
Alan Stokesc4354b82023-05-04 16:06:52 +010060 pub fn write_next_bcc(
Alice Wang843d8312023-02-15 09:47:06 +000061 self,
Alan Stokesc4354b82023-05-04 16:06:52 +010062 current_bcc_handover: &[u8],
Alice Wang843d8312023-02-15 09:47:06 +000063 salt: &[u8; HIDDEN_SIZE],
Alan Stokesc4354b82023-05-04 16:06:52 +010064 next_bcc: &mut [u8],
65 ) -> diced_open_dice::Result<()> {
66 let mut config_descriptor_buffer = [0; 128];
Pierre-Clément Tosif58f3a32023-02-02 16:24:23 +000067 let config_descriptor_size = bcc_format_config_descriptor(
Jiyong Parkb87f3302023-03-21 10:03:11 +090068 Some(cstr!("vm_entry")),
Pierre-Clément Tosif58f3a32023-02-02 16:24:23 +000069 None, // component_version
70 false, // resettable
Alan Stokesc4354b82023-05-04 16:06:52 +010071 &mut config_descriptor_buffer,
Pierre-Clément Tosif58f3a32023-02-02 16:24:23 +000072 )?;
73 let config = &config_descriptor_buffer[..config_descriptor_size];
74
Alan Stokesc4354b82023-05-04 16:06:52 +010075 let dice_inputs = InputValues::new(
Pierre-Clément Tosif58f3a32023-02-02 16:24:23 +000076 self.code_hash,
77 Config::Descriptor(config),
78 self.auth_hash,
79 self.mode,
80 *salt,
Alan Stokesc4354b82023-05-04 16:06:52 +010081 );
82 let _ = bcc_handover_main_flow(current_bcc_handover, &dice_inputs, next_bcc)?;
83 Ok(())
Pierre-Clément Tosif58f3a32023-02-02 16:24:23 +000084 }
Pierre-Clément Tosi4f4f5eb2022-12-08 14:31:42 +000085}
Pierre-Clément Tosibec84662023-01-04 14:25:33 +000086
87/// Flushes data caches over the provided address range.
88///
89/// # Safety
90///
91/// The provided address and size must be to a valid address range (typically on the stack, .bss,
92/// .data, or provided BCC).
93#[no_mangle]
94unsafe extern "C" fn DiceClearMemory(_ctx: *mut c_void, size: usize, addr: *mut c_void) {
95 // SAFETY - We must trust that the slice will be valid arrays/variables on the C code stack.
96 let region = unsafe { slice::from_raw_parts_mut(addr as *mut u8, size) };
97 flushed_zeroize(region)
98}