blob: b0177bf808fc75b93a119ad6c46479fd32881f79 [file] [log] [blame]
Andrew Walbran68a8c162022-03-07 15:38: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//! pVM firmware.
16
17#![no_main]
18#![no_std]
Pierre-Clément Tosifc531152022-10-20 12:22:23 +010019#![feature(default_alloc_error_handler)]
Pierre-Clément Tosi20b60962022-10-17 13:35:27 +010020#![feature(ptr_const_cast)] // Stabilized in 1.65.0
Andrew Walbran68a8c162022-03-07 15:38:42 +000021
Pierre-Clément Tosi4ef75222022-10-26 17:40:50 +010022mod avb;
Pierre-Clément Tosi20b60962022-10-17 13:35:27 +010023mod config;
Pierre-Clément Tosi5bbfca52022-10-21 12:14:35 +010024mod entry;
Andrew Walbrandfb73372022-04-21 10:52:27 +000025mod exceptions;
Pierre-Clément Tosia0934c12022-11-25 20:54:11 +000026mod fdt;
Pierre-Clément Tosifc531152022-10-20 12:22:23 +010027mod heap;
Pierre-Clément Tosida4440a2022-08-22 18:06:32 +010028mod helpers;
Andrew Walbranba47d1d2022-12-14 15:21:44 +000029mod hvc;
Pierre-Clément Tosia0934c12022-11-25 20:54:11 +000030mod memory;
Pierre-Clément Tosi072969b2022-10-19 17:32:24 +010031mod mmio_guard;
Pierre-Clément Tosia8a4a202022-11-03 14:16:46 +000032mod mmu;
Andrew Walbran19690632022-12-07 16:41:30 +000033mod pci;
Pierre-Clément Tosida4440a2022-08-22 18:06:32 +010034mod smccc;
Andrew Walbran68a8c162022-03-07 15:38:42 +000035
Andrew Walbrand1d03182022-12-09 18:20:01 +000036use crate::{
Pierre-Clément Tosi2d652982023-01-09 19:30:17 +000037 avb::PUBLIC_KEY,
Andrew Walbrand1d03182022-12-09 18:20:01 +000038 entry::RebootReason,
39 memory::MemoryTracker,
Andrew Walbran730375d2022-12-21 14:04:34 +000040 pci::{find_virtio_devices, map_mmio},
Andrew Walbrand1d03182022-12-09 18:20:01 +000041};
Pierre-Clément Tosi8edf72e2022-12-06 16:02:57 +000042use dice::bcc;
Andrew Walbran730375d2022-12-21 14:04:34 +000043use fdtpci::{PciError, PciInfo};
Andrew Walbran19690632022-12-07 16:41:30 +000044use libfdt::Fdt;
Pierre-Clément Tosi8edf72e2022-12-06 16:02:57 +000045use log::{debug, error, info, trace};
Pierre-Clément Tosi2d652982023-01-09 19:30:17 +000046use pvmfw_avb::verify_payload;
Andrew Walbran68a8c162022-03-07 15:38:42 +000047
Alice Wang28cbcf12022-12-01 07:58:28 +000048fn main(
Andrew Walbran19690632022-12-07 16:41:30 +000049 fdt: &Fdt,
Alice Wang28cbcf12022-12-01 07:58:28 +000050 signed_kernel: &[u8],
51 ramdisk: Option<&[u8]>,
Pierre-Clément Tosi8edf72e2022-12-06 16:02:57 +000052 bcc: &bcc::Handover,
Andrew Walbran19690632022-12-07 16:41:30 +000053 memory: &mut MemoryTracker,
Alice Wang0a688d22022-12-02 09:48:41 +000054) -> Result<(), RebootReason> {
Pierre-Clément Tosi37105a62022-10-18 12:21:48 +010055 info!("pVM firmware");
Pierre-Clément Tosia0934c12022-11-25 20:54:11 +000056 debug!("FDT: {:?}", fdt as *const libfdt::Fdt);
57 debug!("Signed kernel: {:?} ({:#x} bytes)", signed_kernel.as_ptr(), signed_kernel.len());
Alice Wanga78279c2022-12-16 12:41:19 +000058 debug!("AVB public key: addr={:?}, size={:#x} ({1})", PUBLIC_KEY.as_ptr(), PUBLIC_KEY.len());
Pierre-Clément Tosia0934c12022-11-25 20:54:11 +000059 if let Some(rd) = ramdisk {
60 debug!("Ramdisk: {:?} ({:#x} bytes)", rd.as_ptr(), rd.len());
61 } else {
62 debug!("Ramdisk: None");
63 }
Pierre-Clément Tosi8edf72e2022-12-06 16:02:57 +000064 trace!("BCC: {bcc:x?}");
Andrew Walbran19690632022-12-07 16:41:30 +000065
66 // Set up PCI bus for VirtIO devices.
Andrew Walbrand1d03182022-12-09 18:20:01 +000067 let pci_info = PciInfo::from_fdt(fdt).map_err(handle_pci_error)?;
68 debug!("PCI: {:#x?}", pci_info);
Andrew Walbran730375d2022-12-21 14:04:34 +000069 map_mmio(&pci_info, memory)?;
Andrew Walbrand1d03182022-12-09 18:20:01 +000070 // Safety: This is the only place where we call make_pci_root, and this main function is only
71 // called once.
72 let mut pci_root = unsafe { pci_info.make_pci_root() };
Andrew Walbran0a8dac72022-12-21 13:49:06 +000073 find_virtio_devices(&mut pci_root).map_err(handle_pci_error)?;
Andrew Walbran19690632022-12-07 16:41:30 +000074
Pierre-Clément Tosi2d652982023-01-09 19:30:17 +000075 verify_payload(signed_kernel, PUBLIC_KEY).map_err(|e| {
76 error!("Failed to verify the payload: {e}");
77 RebootReason::PayloadVerificationError
78 })?;
79
Alice Wang4379c832022-12-05 15:50:20 +000080 info!("Starting payload...");
Alice Wang28cbcf12022-12-01 07:58:28 +000081 Ok(())
Pierre-Clément Tosi263ffd52022-10-05 20:27:50 +010082}
Andrew Walbrand1d03182022-12-09 18:20:01 +000083
84/// Logs the given PCI error and returns the appropriate `RebootReason`.
85fn handle_pci_error(e: PciError) -> RebootReason {
86 error!("{}", e);
87 match e {
88 PciError::FdtErrorPci(_)
89 | PciError::FdtNoPci
90 | PciError::FdtErrorReg(_)
91 | PciError::FdtMissingReg
92 | PciError::FdtRegEmpty
93 | PciError::FdtRegMissingSize
94 | PciError::CamWrongSize(_)
95 | PciError::FdtErrorRanges(_)
96 | PciError::FdtMissingRanges
97 | PciError::RangeAddressMismatch { .. }
98 | PciError::NoSuitableRange => RebootReason::InvalidFdt,
Andrew Walbrand1d03182022-12-09 18:20:01 +000099 }
100}