Introduce a `ConfigSpace` wrapper in Rust PCI abstraction to provide safe
accessors for PCI configuration space. The new type implements the
`IoRegion` trait to share offset validation and bound-checking logic with
MMIO regions.
Signed-off-by: Zhi Wang <zhiw@nvidia.com>
---
rust/kernel/pci.rs | 61 ++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 61 insertions(+)
diff --git a/rust/kernel/pci.rs b/rust/kernel/pci.rs
index 7a107015e7d2..2f94b370fc99 100644
--- a/rust/kernel/pci.rs
+++ b/rust/kernel/pci.rs
@@ -12,6 +12,8 @@
error::{from_result, to_result, Result},
io::Io,
io::IoRaw,
+ io::IoRegion,
+ io::{define_read, define_write},
str::CStr,
types::{ARef, Opaque},
ThisModule,
@@ -275,6 +277,65 @@ pub struct Device<Ctx: device::DeviceContext = device::Normal>(
PhantomData<Ctx>,
);
+/// Represents the PCI configuration space of a device.
+///
+/// Provides typed read and write accessors for configuration registers
+/// using the standard `pci_read_config_*` and `pci_write_config_*` helpers.
+///
+/// The generic const parameter `SIZE` can be used to indicate the
+/// maximum size of the configuration space (e.g. 256 bytes for legacy,
+/// 4096 bytes for extended config space). The actual size is obtained
+/// from the underlying `struct pci_dev` via [`Device::cfg_size`].
+pub struct ConfigSpace<const SIZE: usize = 0> {
+ pdev: ARef<Device>,
+}
+
+impl<const SIZE: usize> IoRegion<SIZE> for ConfigSpace<SIZE> {
+ /// Returns the base address of this mapping.
+ #[inline]
+ fn addr(&self) -> usize {
+ 0
+ }
+
+ /// Returns the maximum size of this mapping.
+ #[inline]
+ fn maxsize(&self) -> usize {
+ self.pdev.cfg_size() as usize
+ }
+}
+
+macro_rules! call_config_read {
+ ($c_fn:ident, $self:ident, $offset:expr, $ty:ty, $_addr:expr) => {{
+ let mut val: $ty = 0;
+ let _ret = unsafe { bindings::$c_fn($self.pdev.as_raw(), $offset as i32, &mut val) };
+ val
+ }};
+}
+
+macro_rules! call_config_write {
+ ($c_fn:ident, $self:ident, $offset:expr, $ty:ty, $_addr:expr, $value:expr) => {{
+ let _ret = unsafe { bindings::$c_fn($self.pdev.as_raw(), $offset as i32, $value) };
+ }};
+}
+
+#[allow(dead_code)]
+impl<const SIZE: usize> ConfigSpace<SIZE> {
+ /// Return an initialized object.
+ pub fn new(pdev: &Device) -> Result<Self> {
+ Ok(ConfigSpace {
+ pdev: pdev.into(),
+ })
+ }
+
+ define_read!(read8, try_read8, call_config_read, pci_read_config_byte -> u8);
+ define_read!(read16, try_read16, call_config_read, pci_read_config_word -> u16);
+ define_read!(read32, try_read32, call_config_read, pci_read_config_dword -> u32);
+
+ define_write!(write8, try_write8, call_config_write, pci_write_config_byte <- u8);
+ define_write!(write16, try_write16, call_config_write, pci_write_config_word <- u16);
+ define_write!(write32, try_write32, call_config_write, pci_write_config_dword <- u32);
+}
+
/// A PCI BAR to perform I/O-Operations on.
///
/// # Invariants
--
2.47.3