rdma 0.3.0

Low-level RDMA API
Documentation 
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

use crate::bindings as C;
use crate::error::create_resource;
use crate::pd::ProtectionDomain;
use crate::utils::{c_uint_to_u32, u32_as_c_uint};

use std::io;
use std::os::raw::c_uint;
use std::ptr::NonNull;
use std::sync::Arc;

#[derive(Clone)]
pub struct MemoryWindow(Arc<Owner>);

impl MemoryWindow {
    #[inline]
    pub fn alloc(pd: &ProtectionDomain, mw_type: MemoryWindowType) -> io::Result<Self> {
        // SAFETY: ffi
        let owner = unsafe {
            let mw_type = mw_type.to_c_uint();
            let mw = create_resource(
                || C::ibv_alloc_mw(pd.ffi_ptr(), mw_type),
                || "failed to allocate memory window",
            )?;
            Arc::new(Owner {
                mw,
                _pd: pd.clone(),
            })
        };
        Ok(Self(owner))
    }
}

struct Owner {
    mw: NonNull<C::ibv_mw>,
    _pd: ProtectionDomain,
}

/// SAFETY: owned type
unsafe impl Send for Owner {}
/// SAFETY: owned type
unsafe impl Sync for Owner {}

impl Owner {
    fn ffi_ptr(&self) -> *mut C::ibv_mw {
        self.mw.as_ptr()
    }
}

impl Drop for Owner {
    fn drop(&mut self) {
        // SAFETY: ffi
        unsafe {
            let mw = self.ffi_ptr();
            let ret = C::ibv_dealloc_mw(mw);
            assert_eq!(ret, 0);
        }
    }
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
#[repr(u32)]
pub enum MemoryWindowType {
    Type1 = c_uint_to_u32(C::IBV_MW_TYPE_1),
    Type2 = c_uint_to_u32(C::IBV_MW_TYPE_2),
}

impl MemoryWindowType {
    fn to_c_uint(self) -> c_uint {
        #[allow(clippy::as_conversions)]
        u32_as_c_uint(self as u32)
    }
}