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
73
74
75
76

use crate::bindings as C;
use crate::ctx::Context;
use crate::error::create_resource;

use std::ptr::NonNull;
use std::sync::Arc;
use std::{io, mem};

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

impl DeviceMemory {
    #[inline]
    #[must_use]
    pub fn options() -> DeviceMemoryOptions {
        DeviceMemoryOptions::default()
    }

    #[inline]
    pub fn alloc(ctx: &Context, mut options: DeviceMemoryOptions) -> io::Result<Self> {
        // SAFETY: ffi
        let owner = unsafe {
            let attr = &mut options.attr;
            let dm = create_resource(
                || C::ibv_alloc_dm(ctx.ffi_ptr(), attr),
                || "failed to allocate device memory",
            )?;
            Arc::new(Owner {
                dm,
                _ctx: ctx.clone(),
            })
        };
        Ok(Self(owner))
    }
}

struct Owner {
    dm: NonNull<C::ibv_dm>,
    _ctx: Context,
}

/// 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_dm {
        self.dm.as_ptr()
    }
}

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

pub struct DeviceMemoryOptions {
    attr: C::ibv_alloc_dm_attr,
}

impl Default for DeviceMemoryOptions {
    #[inline]
    fn default() -> Self {
        Self {
            // SAFETY: POD ffi type
            attr: unsafe { mem::zeroed() },
        }
    }
}