Merge "No need to allocate PCI BARs after all."
diff --git a/pvmfw/src/entry.rs b/pvmfw/src/entry.rs
index 45a8459..1b35c79 100644
--- a/pvmfw/src/entry.rs
+++ b/pvmfw/src/entry.rs
@@ -48,8 +48,6 @@
InvalidRamdisk,
/// Failed to verify the payload.
PayloadVerificationError,
- /// Error interacting with a VirtIO PCI device.
- PciError,
}
main!(start);
diff --git a/pvmfw/src/main.rs b/pvmfw/src/main.rs
index 59efca0..7d64bf0 100644
--- a/pvmfw/src/main.rs
+++ b/pvmfw/src/main.rs
@@ -37,7 +37,7 @@
avb::PUBLIC_KEY,
entry::RebootReason,
memory::MemoryTracker,
- pci::{allocate_all_virtio_bars, PciError, PciInfo, PciMemory32Allocator},
+ pci::{find_virtio_devices, PciError, PciInfo},
};
use dice::bcc;
use libfdt::Fdt;
@@ -65,12 +65,10 @@
let pci_info = PciInfo::from_fdt(fdt).map_err(handle_pci_error)?;
debug!("PCI: {:#x?}", pci_info);
pci_info.map(memory)?;
- let mut bar_allocator = PciMemory32Allocator::new(&pci_info);
- debug!("Allocator: {:#x?}", bar_allocator);
// Safety: This is the only place where we call make_pci_root, and this main function is only
// called once.
let mut pci_root = unsafe { pci_info.make_pci_root() };
- allocate_all_virtio_bars(&mut pci_root, &mut bar_allocator).map_err(handle_pci_error)?;
+ find_virtio_devices(&mut pci_root).map_err(handle_pci_error)?;
verify_payload(PUBLIC_KEY).map_err(|e| {
error!("Failed to verify the payload: {e}");
@@ -95,8 +93,5 @@
| PciError::FdtMissingRanges
| PciError::RangeAddressMismatch { .. }
| PciError::NoSuitableRange => RebootReason::InvalidFdt,
- PciError::BarInfoFailed(_)
- | PciError::BarAllocationFailed { .. }
- | PciError::UnsupportedBarType(_) => RebootReason::PciError,
}
}
diff --git a/pvmfw/src/pci.rs b/pvmfw/src/pci.rs
index d971c7b..301ecfc 100644
--- a/pvmfw/src/pci.rs
+++ b/pvmfw/src/pci.rs
@@ -12,7 +12,7 @@
// See the License for the specific language governing permissions and
// limitations under the License.
-//! Functions to scan the PCI bus for VirtIO device and allocate BARs.
+//! Functions to scan the PCI bus for VirtIO devices.
use crate::{
entry::RebootReason,
@@ -26,7 +26,7 @@
use libfdt::{AddressRange, Fdt, FdtError, FdtNode};
use log::{debug, error};
use virtio_drivers::pci::{
- bus::{self, BarInfo, Cam, Command, DeviceFunction, MemoryBarType, PciRoot},
+ bus::{Cam, PciRoot},
virtio_device_type,
};
@@ -46,9 +46,6 @@
FdtMissingRanges,
RangeAddressMismatch { bus_address: u64, cpu_physical: u64 },
NoSuitableRange,
- BarInfoFailed(bus::PciError),
- BarAllocationFailed { size: u32, device_function: DeviceFunction },
- UnsupportedBarType(MemoryBarType),
}
impl Display for PciError {
@@ -77,15 +74,6 @@
)
}
Self::NoSuitableRange => write!(f, "No suitable PCI memory range found."),
- Self::BarInfoFailed(e) => write!(f, "Error getting PCI BAR information: {}", e),
- Self::BarAllocationFailed { size, device_function } => write!(
- f,
- "Failed to allocate memory BAR of size {} for PCI device {}.",
- size, device_function
- ),
- Self::UnsupportedBarType(address_type) => {
- write!(f, "Memory BAR address type {:?} not supported.", address_type)
- }
}
}
}
@@ -247,11 +235,8 @@
}
}
-/// Allocates BARs for all VirtIO PCI devices.
-pub fn allocate_all_virtio_bars(
- pci_root: &mut PciRoot,
- allocator: &mut PciMemory32Allocator,
-) -> Result<(), PciError> {
+/// Finds VirtIO PCI devices.
+pub fn find_virtio_devices(pci_root: &mut PciRoot) -> Result<(), PciError> {
for (device_function, info) in pci_root.enumerate_bus(0) {
let (status, command) = pci_root.get_status_command(device_function);
debug!(
@@ -260,87 +245,8 @@
);
if let Some(virtio_type) = virtio_device_type(&info) {
debug!(" VirtIO {:?}", virtio_type);
- allocate_bars(pci_root, device_function, allocator)?;
}
}
Ok(())
}
-
-/// Allocates 32-bit memory addresses for PCI BARs.
-#[derive(Debug)]
-pub struct PciMemory32Allocator {
- /// The start of the available (not yet allocated) address space for PCI BARs.
- start: u32,
- /// The end of the available address space.
- end: u32,
-}
-
-impl PciMemory32Allocator {
- pub fn new(pci_info: &PciInfo) -> Self {
- Self { start: pci_info.bar_range.start, end: pci_info.bar_range.end }
- }
-
- /// Allocates a 32-bit memory address region for a PCI BAR of the given power-of-2 size.
- ///
- /// It will have alignment matching the size. The size must be a power of 2.
- pub fn allocate_memory_32(&mut self, size: u32) -> Option<u32> {
- assert!(size.is_power_of_two());
- let allocated_address = align_up(self.start, size);
- if allocated_address + size <= self.end {
- self.start = allocated_address + size;
- Some(allocated_address)
- } else {
- None
- }
- }
-}
-
-/// Allocates appropriately-sized memory regions and assigns them to the device's BARs.
-fn allocate_bars(
- root: &mut PciRoot,
- device_function: DeviceFunction,
- allocator: &mut PciMemory32Allocator,
-) -> Result<(), PciError> {
- let mut bar_index = 0;
- while bar_index < 6 {
- let info = root.bar_info(device_function, bar_index).map_err(PciError::BarInfoFailed)?;
- debug!("BAR {}: {}", bar_index, info);
- // Ignore I/O bars, as they aren't required for the VirtIO driver.
- if let BarInfo::Memory { address_type, size, .. } = info {
- match address_type {
- _ if size == 0 => {}
- MemoryBarType::Width32 => {
- let address = allocator
- .allocate_memory_32(size)
- .ok_or(PciError::BarAllocationFailed { size, device_function })?;
- debug!("Allocated address {:#010x}", address);
- root.set_bar_32(device_function, bar_index, address);
- }
- _ => {
- return Err(PciError::UnsupportedBarType(address_type));
- }
- }
- }
-
- bar_index += 1;
- if info.takes_two_entries() {
- bar_index += 1;
- }
- }
-
- // Enable the device to use its BARs.
- root.set_command(
- device_function,
- Command::IO_SPACE | Command::MEMORY_SPACE | Command::BUS_MASTER,
- );
- let (status, command) = root.get_status_command(device_function);
- debug!("Allocated BARs and enabled device, status {:?} command {:?}", status, command);
-
- Ok(())
-}
-
-// TODO: Make the alignment functions in the helpers module generic once const_trait_impl is stable.
-const fn align_up(value: u32, alignment: u32) -> u32 {
- ((value - 1) | (alignment - 1)) + 1
-}
diff --git a/vmbase/example/src/main.rs b/vmbase/example/src/main.rs
index 96c10a4..888f273 100644
--- a/vmbase/example/src/main.rs
+++ b/vmbase/example/src/main.rs
@@ -82,7 +82,7 @@
// access it; maybe there is some new version of PCI we don't know about.
assert_eq!(pci_reg.size.unwrap(), AARCH64_PCI_CFG_SIZE);
// Parse ranges property to find memory ranges from which to allocate PCI BARs.
- let mut pci_allocator = PciMemory32Allocator::for_pci_ranges(&pci_node);
+ let pci_allocator = PciMemory32Allocator::for_pci_ranges(&pci_node);
modify_fdt(fdt);
@@ -139,7 +139,7 @@
check_data();
check_dice();
- check_pci(pci_reg, &mut pci_allocator);
+ check_pci(pci_reg);
}
fn check_stack_guard() {
diff --git a/vmbase/example/src/pci.rs b/vmbase/example/src/pci.rs
index 82ea7cc..bd5b5ba 100644
--- a/vmbase/example/src/pci.rs
+++ b/vmbase/example/src/pci.rs
@@ -12,7 +12,7 @@
// See the License for the specific language governing permissions and
// limitations under the License.
-//! Functions to scan the PCI bus for VirtIO device and allocate BARs.
+//! Functions to scan the PCI bus for VirtIO device.
use aarch64_paging::paging::MemoryRegion;
use alloc::alloc::{alloc, dealloc, Layout};
@@ -21,7 +21,7 @@
use log::{debug, info};
use virtio_drivers::{
pci::{
- bus::{BarInfo, Cam, Command, DeviceFunction, MemoryBarType, PciRoot},
+ bus::{Cam, PciRoot},
virtio_device_type, PciTransport,
},
DeviceType, Hal, PhysAddr, Transport, VirtAddr, VirtIOBlk, PAGE_SIZE,
@@ -41,7 +41,7 @@
.unwrap()
}
-pub fn check_pci(reg: Reg<u64>, allocator: &mut PciMemory32Allocator) {
+pub fn check_pci(reg: Reg<u64>) {
let mut pci_root = unsafe { PciRoot::new(reg.addr as *mut u8, Cam::MmioCam) };
let mut checked_virtio_device_count = 0;
for (device_function, info) in pci_root.enumerate_bus(0) {
@@ -49,7 +49,6 @@
info!("Found {} at {}, status {:?} command {:?}", info, device_function, status, command);
if let Some(virtio_type) = virtio_device_type(&info) {
info!(" VirtIO {:?}", virtio_type);
- allocate_bars(&mut pci_root, device_function, allocator);
let mut transport =
PciTransport::new::<HalImpl>(&mut pci_root, device_function).unwrap();
info!(
@@ -152,20 +151,6 @@
pub fn get_region(&self) -> MemoryRegion {
MemoryRegion::new(self.start as usize, self.end as usize)
}
-
- /// Allocates a 32-bit memory address region for a PCI BAR of the given power-of-2 size.
- ///
- /// It will have alignment matching the size. The size must be a power of 2.
- pub fn allocate_memory_32(&mut self, size: u32) -> Option<u32> {
- assert!(size.is_power_of_two());
- let allocated_address = align_up(self.start, size);
- if allocated_address + size <= self.end {
- self.start = allocated_address + size;
- Some(allocated_address)
- } else {
- None
- }
- }
}
#[derive(Copy, Clone, Debug, Eq, PartialEq)]
@@ -188,53 +173,6 @@
}
}
-/// Allocates appropriately-sized memory regions and assigns them to the device's BARs.
-fn allocate_bars(
- root: &mut PciRoot,
- device_function: DeviceFunction,
- allocator: &mut PciMemory32Allocator,
-) {
- let mut bar_index = 0;
- while bar_index < 6 {
- let info = root.bar_info(device_function, bar_index).unwrap();
- debug!("BAR {}: {}", bar_index, info);
- // Ignore I/O bars, as they aren't required for the VirtIO driver.
- if let BarInfo::Memory { address_type, size, .. } = info {
- match address_type {
- _ if size == 0 => {}
- MemoryBarType::Width32 => {
- let address = allocator.allocate_memory_32(size).unwrap();
- debug!("Allocated address {:#010x}", address);
- root.set_bar_32(device_function, bar_index, address);
- }
- MemoryBarType::Width64 => {
- let address = allocator.allocate_memory_32(size).unwrap();
- debug!("Allocated address {:#010x}", address);
- root.set_bar_64(device_function, bar_index, address.into());
- }
- _ => panic!("Memory BAR address type {:?} not supported.", address_type),
- }
- }
-
- bar_index += 1;
- if info.takes_two_entries() {
- bar_index += 1;
- }
- }
-
- // Enable the device to use its BARs.
- root.set_command(
- device_function,
- Command::IO_SPACE | Command::MEMORY_SPACE | Command::BUS_MASTER,
- );
- let (status, command) = root.get_status_command(device_function);
- debug!("Allocated BARs and enabled device, status {:?} command {:?}", status, command);
-}
-
-const fn align_up(value: u32, alignment: u32) -> u32 {
- ((value - 1) | (alignment - 1)) + 1
-}
-
struct HalImpl;
impl Hal for HalImpl {