Something went wrong. Try again.
(I'm so sorry!) Linux kernel mirror (for testing) git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
kernel os linux
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147// SPDX-License-Identifier: GPL-2.0
//! Rust DMA api test (based on QEMU's `pci-testdev`).//!//! To make this driver probe, QEMU must be run with `-device pci-testdev`.
use kernel::{ device::{ Bound, Core, // }, dma::{ Coherent, DataDirection, Device, DmaMask, // }, io::{ io_project, io_read, Io, // }, page, pci, prelude::*, scatterlist::{ Owned, SGTable, // }, //};
struct DmaSampleDriver;
#[pin_data(PinnedDrop)]struct DmaSampleData<'bound> { pdev: &'bound pci::Device<Bound>, ca: Coherent<'bound, [MyStruct]>, #[pin] sgt: SGTable<Owned<VVec<u8>>>,}
const TEST_VALUES: [(u32, u32); 5] = [ (0xa, 0xb), (0xc, 0xd), (0xe, 0xf), (0xab, 0xba), (0xcd, 0xef),];
#[derive(FromBytes, IntoBytes)]struct MyStruct { h: u32, b: u32,}
impl MyStruct { fn new(h: u32, b: u32) -> Self { Self { h, b } }}// SAFETY: All bit patterns are acceptable values for `MyStruct`.unsafe impl kernel::transmute::AsBytes for MyStruct {}// SAFETY: Instances of `MyStruct` have no uninitialized portions.unsafe impl kernel::transmute::FromBytes for MyStruct {}
kernel::pci_device_table!( PCI_TABLE, <DmaSampleDriver as pci::Driver>::IdInfo, [(pci::DeviceId::from_id(pci::Vendor::REDHAT, 0x5), ())]);
impl pci::Driver for DmaSampleDriver { type IdInfo = (); type Data<'bound> = DmaSampleData<'bound>; const ID_TABLE: pci::IdTable<Self::IdInfo> = &PCI_TABLE;
fn probe<'bound>( pdev: &'bound pci::Device<Core<'_>>, _info: Option<&'bound Self::IdInfo>, ) -> impl PinInit<Self::Data<'bound>, Error> + 'bound { pin_init::pin_init_scope(move || { dev_info!(pdev, "Probe DMA test driver.\n");
let mask = DmaMask::new::<64>();
// SAFETY: There are no concurrent calls to DMA allocation and mapping primitives. unsafe { pdev.dma_set_mask_and_coherent(mask)? };
let ca: Coherent<'_, [MyStruct]> = Coherent::zeroed_slice(pdev.as_ref(), TEST_VALUES.len(), GFP_KERNEL)?;
for (i, value) in TEST_VALUES.into_iter().enumerate() { io_project!(ca, [panic: i]).copy_write(MyStruct::new(value.0, value.1)); }
let size = 4 * page::PAGE_SIZE; let pages = VVec::with_capacity(size, GFP_KERNEL)?;
let sgt = SGTable::new(pdev.as_ref(), pages, DataDirection::ToDevice, GFP_KERNEL);
Ok(try_pin_init!(Self::Data { pdev, ca, sgt <- sgt, })) }) }}
impl DmaSampleData<'_> { fn check_dma(&self) { for (i, value) in TEST_VALUES.into_iter().enumerate() { let val0 = io_read!(self.ca, [panic: i].h); let val1 = io_read!(self.ca, [panic: i].b);
assert_eq!(val0, value.0); assert_eq!(val1, value.1); } }}
#[pinned_drop]impl PinnedDrop for DmaSampleData<'_> { fn drop(self: Pin<&mut Self>) { dev_info!(self.pdev, "Unload DMA test driver.\n");
self.check_dma();
for (i, entry) in self.sgt.iter().enumerate() { dev_info!( self.pdev, "Entry[{}]: DMA address: {:#x}", i, entry.dma_address(), ); } }}
kernel::module_pci_driver! { type: DmaSampleDriver, name: "rust_dma", authors: ["Abdiel Janulgue"], description: "Rust DMA test", license: "GPL v2",}