David Brazdil | 66fc120 | 2022-07-04 21:48:45 +0100 | [diff] [blame] | 1 | // 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 | //! Project Rialto main source file. |
| 16 | |
| 17 | #![no_main] |
| 18 | #![no_std] |
David Brazdil | 66fc120 | 2022-07-04 21:48:45 +0100 | [diff] [blame] | 19 | |
Alice Wang | 9a8b39f | 2023-04-12 15:31:48 +0000 | [diff] [blame] | 20 | mod error; |
David Brazdil | 66fc120 | 2022-07-04 21:48:45 +0100 | [diff] [blame] | 21 | mod exceptions; |
| 22 | |
| 23 | extern crate alloc; |
David Brazdil | 66fc120 | 2022-07-04 21:48:45 +0100 | [diff] [blame] | 24 | |
Alice Wang | 9a8b39f | 2023-04-12 15:31:48 +0000 | [diff] [blame] | 25 | use crate::error::{Error, Result}; |
Alice Wang | 74f7f4b | 2023-06-13 08:24:50 +0000 | [diff] [blame] | 26 | use core::num::NonZeroUsize; |
Pierre-Clément Tosi | 3d4c5c3 | 2023-05-31 16:57:06 +0000 | [diff] [blame] | 27 | use core::slice; |
Alice Wang | dda3ba9 | 2023-05-25 15:15:30 +0000 | [diff] [blame] | 28 | use fdtpci::PciInfo; |
Pierre-Clément Tosi | 910a72d | 2023-06-29 14:29:29 +0000 | [diff] [blame] | 29 | use hyp::{get_mem_sharer, get_mmio_guard}; |
Alice Wang | b6d2c64 | 2023-06-13 13:07:06 +0000 | [diff] [blame] | 30 | use libfdt::FdtError; |
Alice Wang | 9a8b39f | 2023-04-12 15:31:48 +0000 | [diff] [blame] | 31 | use log::{debug, error, info}; |
Alice Wang | 6218335 | 2023-07-04 08:33:27 +0000 | [diff] [blame^] | 32 | use virtio_drivers::{ |
| 33 | transport::{pci::bus::PciRoot, DeviceType, Transport}, |
| 34 | Hal, |
| 35 | }; |
Alice Wang | 4b3cc11 | 2023-06-06 12:22:53 +0000 | [diff] [blame] | 36 | use vmbase::{ |
Pierre-Clément Tosi | 6a4808c | 2023-06-29 09:19:38 +0000 | [diff] [blame] | 37 | configure_heap, |
Alice Wang | b6d2c64 | 2023-06-13 13:07:06 +0000 | [diff] [blame] | 38 | fdt::SwiotlbInfo, |
Alice Wang | 89d2959 | 2023-06-12 09:41:29 +0000 | [diff] [blame] | 39 | layout::{self, crosvm}, |
| 40 | main, |
Alice Wang | 6218335 | 2023-07-04 08:33:27 +0000 | [diff] [blame^] | 41 | memory::{MemoryTracker, PageTable, MEMORY, PAGE_SIZE, SIZE_128KB}, |
Alice Wang | 4b3cc11 | 2023-06-06 12:22:53 +0000 | [diff] [blame] | 42 | power::reboot, |
Alice Wang | 6218335 | 2023-07-04 08:33:27 +0000 | [diff] [blame^] | 43 | virtio::{ |
| 44 | pci::{self, PciTransportIterator, VirtIOSocket}, |
| 45 | HalImpl, |
| 46 | }, |
Alice Wang | 4b3cc11 | 2023-06-06 12:22:53 +0000 | [diff] [blame] | 47 | }; |
David Brazdil | 66fc120 | 2022-07-04 21:48:45 +0100 | [diff] [blame] | 48 | |
Alice Wang | b70bdb5 | 2023-06-12 08:17:58 +0000 | [diff] [blame] | 49 | fn new_page_table() -> Result<PageTable> { |
Alice Wang | ee5b180 | 2023-06-07 07:41:54 +0000 | [diff] [blame] | 50 | let mut page_table = PageTable::default(); |
Pierre-Clément Tosi | 3d4c5c3 | 2023-05-31 16:57:06 +0000 | [diff] [blame] | 51 | |
Alice Wang | a3931aa | 2023-07-05 12:52:09 +0000 | [diff] [blame] | 52 | page_table.map_data(&layout::scratch_range().into())?; |
| 53 | page_table.map_data(&layout::stack_range(40 * PAGE_SIZE).into())?; |
| 54 | page_table.map_code(&layout::text_range().into())?; |
| 55 | page_table.map_rodata(&layout::rodata_range().into())?; |
| 56 | page_table.map_device(&layout::console_uart_range().into())?; |
David Brazdil | 66fc120 | 2022-07-04 21:48:45 +0100 | [diff] [blame] | 57 | |
Alice Wang | b70bdb5 | 2023-06-12 08:17:58 +0000 | [diff] [blame] | 58 | Ok(page_table) |
David Brazdil | 66fc120 | 2022-07-04 21:48:45 +0100 | [diff] [blame] | 59 | } |
| 60 | |
Alice Wang | dda3ba9 | 2023-05-25 15:15:30 +0000 | [diff] [blame] | 61 | /// # Safety |
| 62 | /// |
| 63 | /// Behavior is undefined if any of the following conditions are violated: |
| 64 | /// * The `fdt_addr` must be a valid pointer and points to a valid `Fdt`. |
| 65 | unsafe fn try_main(fdt_addr: usize) -> Result<()> { |
David Brazdil | 66fc120 | 2022-07-04 21:48:45 +0100 | [diff] [blame] | 66 | info!("Welcome to Rialto!"); |
Alice Wang | b70bdb5 | 2023-06-12 08:17:58 +0000 | [diff] [blame] | 67 | let page_table = new_page_table()?; |
| 68 | |
| 69 | MEMORY.lock().replace(MemoryTracker::new( |
| 70 | page_table, |
| 71 | crosvm::MEM_START..layout::MAX_VIRT_ADDR, |
| 72 | crosvm::MMIO_RANGE, |
| 73 | None, // Rialto doesn't have any payload for now. |
| 74 | )); |
Alice Wang | 74f7f4b | 2023-06-13 08:24:50 +0000 | [diff] [blame] | 75 | |
| 76 | let fdt_range = MEMORY |
| 77 | .lock() |
| 78 | .as_mut() |
| 79 | .unwrap() |
| 80 | .alloc(fdt_addr, NonZeroUsize::new(crosvm::FDT_MAX_SIZE).unwrap())?; |
| 81 | // SAFETY: The tracker validated the range to be in main memory, mapped, and not overlap. |
| 82 | let fdt = unsafe { slice::from_raw_parts(fdt_range.start as *mut u8, fdt_range.len()) }; |
Alice Wang | 674257a | 2023-06-13 09:44:53 +0000 | [diff] [blame] | 83 | // We do not need to validate the DT since it is already validated in pvmfw. |
Alice Wang | 74f7f4b | 2023-06-13 08:24:50 +0000 | [diff] [blame] | 84 | let fdt = libfdt::Fdt::from_slice(fdt)?; |
Alice Wang | 74f7f4b | 2023-06-13 08:24:50 +0000 | [diff] [blame] | 85 | |
Alice Wang | 674257a | 2023-06-13 09:44:53 +0000 | [diff] [blame] | 86 | let memory_range = fdt.first_memory_range()?; |
| 87 | MEMORY.lock().as_mut().unwrap().shrink(&memory_range).map_err(|e| { |
| 88 | error!("Failed to use memory range value from DT: {memory_range:#x?}"); |
| 89 | e |
| 90 | })?; |
Alice Wang | b6d2c64 | 2023-06-13 13:07:06 +0000 | [diff] [blame] | 91 | |
Pierre-Clément Tosi | 910a72d | 2023-06-29 14:29:29 +0000 | [diff] [blame] | 92 | if let Some(mem_sharer) = get_mem_sharer() { |
| 93 | let granule = mem_sharer.granule()?; |
Alice Wang | b6d2c64 | 2023-06-13 13:07:06 +0000 | [diff] [blame] | 94 | MEMORY.lock().as_mut().unwrap().init_dynamic_shared_pool(granule).map_err(|e| { |
| 95 | error!("Failed to initialize dynamically shared pool."); |
| 96 | e |
| 97 | })?; |
Pierre-Clément Tosi | 8937cb8 | 2023-07-06 15:07:38 +0000 | [diff] [blame] | 98 | } else if let Ok(swiotlb_info) = SwiotlbInfo::new_from_fdt(fdt) { |
| 99 | let range = swiotlb_info.fixed_range().ok_or_else(|| { |
Alice Wang | b6d2c64 | 2023-06-13 13:07:06 +0000 | [diff] [blame] | 100 | error!("Pre-shared pool range not specified in swiotlb node"); |
| 101 | Error::from(FdtError::BadValue) |
| 102 | })?; |
| 103 | MEMORY.lock().as_mut().unwrap().init_static_shared_pool(range).map_err(|e| { |
| 104 | error!("Failed to initialize pre-shared pool."); |
| 105 | e |
| 106 | })?; |
Pierre-Clément Tosi | 8937cb8 | 2023-07-06 15:07:38 +0000 | [diff] [blame] | 107 | } else { |
| 108 | info!("No MEM_SHARE capability detected or swiotlb found: allocating buffers from heap."); |
| 109 | MEMORY.lock().as_mut().unwrap().init_heap_shared_pool().map_err(|e| { |
| 110 | error!("Failed to initialize heap-based pseudo-shared pool."); |
| 111 | e |
| 112 | })?; |
Alice Wang | b6d2c64 | 2023-06-13 13:07:06 +0000 | [diff] [blame] | 113 | } |
Alice Wang | d36c711 | 2023-07-04 09:50:45 +0000 | [diff] [blame] | 114 | |
| 115 | let pci_info = PciInfo::from_fdt(fdt)?; |
| 116 | debug!("PCI: {pci_info:#x?}"); |
Alice Wang | 6218335 | 2023-07-04 08:33:27 +0000 | [diff] [blame^] | 117 | let mut pci_root = pci::initialize(pci_info, MEMORY.lock().as_mut().unwrap()) |
Alice Wang | d36c711 | 2023-07-04 09:50:45 +0000 | [diff] [blame] | 118 | .map_err(Error::PciInitializationFailed)?; |
| 119 | debug!("PCI root: {pci_root:#x?}"); |
Alice Wang | 6218335 | 2023-07-04 08:33:27 +0000 | [diff] [blame^] | 120 | let socket_device = find_socket_device::<HalImpl>(&mut pci_root)?; |
| 121 | debug!("Found socket device: guest cid = {:?}", socket_device.guest_cid()); |
Alice Wang | 9a8b39f | 2023-04-12 15:31:48 +0000 | [diff] [blame] | 122 | Ok(()) |
| 123 | } |
| 124 | |
Alice Wang | 6218335 | 2023-07-04 08:33:27 +0000 | [diff] [blame^] | 125 | fn find_socket_device<T: Hal>(pci_root: &mut PciRoot) -> Result<VirtIOSocket<T>> { |
| 126 | PciTransportIterator::<T>::new(pci_root) |
| 127 | .find(|t| DeviceType::Socket == t.device_type()) |
| 128 | .map(VirtIOSocket::<T>::new) |
| 129 | .transpose() |
| 130 | .map_err(Error::VirtIOSocketCreationFailed)? |
| 131 | .ok_or(Error::MissingVirtIOSocketDevice) |
| 132 | } |
| 133 | |
Pierre-Clément Tosi | 910a72d | 2023-06-29 14:29:29 +0000 | [diff] [blame] | 134 | fn try_unshare_all_memory() -> Result<()> { |
Alice Wang | 77d9dd3 | 2023-06-07 13:41:21 +0000 | [diff] [blame] | 135 | info!("Starting unsharing memory..."); |
| 136 | |
Alice Wang | 77d9dd3 | 2023-06-07 13:41:21 +0000 | [diff] [blame] | 137 | // No logging after unmapping UART. |
Pierre-Clément Tosi | 910a72d | 2023-06-29 14:29:29 +0000 | [diff] [blame] | 138 | if let Some(mmio_guard) = get_mmio_guard() { |
| 139 | mmio_guard.unmap(vmbase::console::BASE_ADDRESS)?; |
Alice Wang | b70bdb5 | 2023-06-12 08:17:58 +0000 | [diff] [blame] | 140 | } |
| 141 | // Unshares all memory and deactivates page table. |
| 142 | drop(MEMORY.lock().take()); |
Alice Wang | 77d9dd3 | 2023-06-07 13:41:21 +0000 | [diff] [blame] | 143 | Ok(()) |
| 144 | } |
| 145 | |
Pierre-Clément Tosi | 910a72d | 2023-06-29 14:29:29 +0000 | [diff] [blame] | 146 | fn unshare_all_memory() { |
| 147 | if let Err(e) = try_unshare_all_memory() { |
Alice Wang | 77d9dd3 | 2023-06-07 13:41:21 +0000 | [diff] [blame] | 148 | error!("Failed to unshare the memory: {e}"); |
| 149 | } |
| 150 | } |
| 151 | |
Alice Wang | 9a8b39f | 2023-04-12 15:31:48 +0000 | [diff] [blame] | 152 | /// Entry point for Rialto. |
Alice Wang | dda3ba9 | 2023-05-25 15:15:30 +0000 | [diff] [blame] | 153 | pub fn main(fdt_addr: u64, _a1: u64, _a2: u64, _a3: u64) { |
Pierre-Clément Tosi | d330548 | 2023-06-29 15:03:48 +0000 | [diff] [blame] | 154 | log::set_max_level(log::LevelFilter::Debug); |
Alice Wang | dda3ba9 | 2023-05-25 15:15:30 +0000 | [diff] [blame] | 155 | // SAFETY: `fdt_addr` is supposed to be a valid pointer and points to |
| 156 | // a valid `Fdt`. |
| 157 | match unsafe { try_main(fdt_addr as usize) } { |
Pierre-Clément Tosi | 910a72d | 2023-06-29 14:29:29 +0000 | [diff] [blame] | 158 | Ok(()) => unshare_all_memory(), |
Alice Wang | 9a8b39f | 2023-04-12 15:31:48 +0000 | [diff] [blame] | 159 | Err(e) => { |
| 160 | error!("Rialto failed with {e}"); |
Pierre-Clément Tosi | 910a72d | 2023-06-29 14:29:29 +0000 | [diff] [blame] | 161 | unshare_all_memory(); |
Alice Wang | 9a8b39f | 2023-04-12 15:31:48 +0000 | [diff] [blame] | 162 | reboot() |
| 163 | } |
| 164 | } |
David Brazdil | 66fc120 | 2022-07-04 21:48:45 +0100 | [diff] [blame] | 165 | } |
| 166 | |
David Brazdil | 66fc120 | 2022-07-04 21:48:45 +0100 | [diff] [blame] | 167 | main!(main); |
Alice Wang | 6218335 | 2023-07-04 08:33:27 +0000 | [diff] [blame^] | 168 | configure_heap!(SIZE_128KB); |