page_pool_alloc/
device_dma.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.

//! Module for device dma support.

// UNSAFETY: This is required to implement the MappedDmaTarget trait which
// unsafe because of it's requirement for the implementer to keep the
// `base()..len()` mapped for the lifetime of the struct.
#![expect(unsafe_code)]

use crate::PAGE_SIZE;
use crate::PagePoolHandle;
use user_driver::memory::MappedDmaTarget;

/// Page pool memory representing a DMA buffer useable by devices.
pub struct PagePoolDmaBuffer {
    // Holds allocation until dropped.
    pub(crate) alloc: PagePoolHandle,
    pub(crate) pfns: Vec<u64>,
}

/// SAFETY: This struct keeps both the shared memory region which the sparse
/// mapping maps, along with the sparse mapping itself until the struct is drop,
/// satisfying the trait.
unsafe impl MappedDmaTarget for PagePoolDmaBuffer {
    fn base(&self) -> *const u8 {
        self.alloc
            .inner
            .mapping
            .as_ptr()
            .wrapping_byte_add(self.alloc.mapping_offset)
            .cast()
    }

    fn len(&self) -> usize {
        (self.alloc.size_pages * PAGE_SIZE) as usize
    }

    fn pfns(&self) -> &[u64] {
        &self.pfns
    }

    fn pfn_bias(&self) -> u64 {
        self.alloc.inner.pfn_bias
    }
}