io-uring-bearer 0.2.0-pre3

io_uring bearer
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

//! SendZc Slab records

use io_uring_owner::Owner;

use crate::slab::buffer::TakenImmutableBuffer;

/// SendZc could be either through a fixed registered buffer through it's index or
/// unsafely provided through as raw reference.
#[derive(Clone, Debug)]
pub enum SendZcRec {
    /// Fixed registered buffer
    Fixed(SendZcFixedRec),
    /// Unsafely provided through raw reference
    UnsafeRef(SendZcUnsafeRefRec),
}

/// SendZc Fixed Record
#[derive(Clone, Debug)]
pub struct SendZcFixedRec {
    fixed_fd: u32,
    owner: Owner,
    buf_taken: TakenImmutableBuffer,
    to_addr: Option<DestTo>,
}

/// SendZc UnsafeRef Record
#[derive(Clone, Debug)]
pub struct SendZcUnsafeRefRec {
    fixed_fd: u32,
    owner: Owner,
    buf_const_u8: *const u8,
    buf_size: u32,
    to_addr: Option<DestTo>,
}

impl SendZcRec {
    #[inline]
    pub(crate) fn with_fixed_buf(fixed_fd: u32, buf_taken: TakenImmutableBuffer) -> Self {
        SendZcRec::Fixed(SendZcFixedRec {
            fixed_fd,
            owner: Owner::Created,
            buf_taken,
            to_addr: None, // TODO
        })
    }
    #[inline]
    pub(crate) fn entry(&self) -> io_uring::squeue::Entry {
        let (fixed_fd, buf_const_u8, buf_size, buf_index) = match self {
            Self::Fixed(f_rec) => (
                f_rec.fixed_fd,
                f_rec.buf_taken.buf_const_u8,
                2,
                //                f_rec.buf_taken.buf_size,
                None,
                //                Some(f_rec.buf_taken.buf_kernel_index),
            ),
            Self::UnsafeRef(u_rec) => (u_rec.fixed_fd, u_rec.buf_const_u8, u_rec.buf_size, None),
        };

        io_uring::opcode::SendZc::new(io_uring::types::Fixed(fixed_fd), buf_const_u8, buf_size)
            .buf_index(buf_index)
            .build() // todo dest_addr(), dest_addr_len(), flags(), zc_flags()
    }
    #[inline]
    pub(crate) fn owner(&self) -> Owner {
        match self {
            Self::Fixed(f) => f.owner.clone(),
            Self::UnsafeRef(r) => r.owner.clone(),
        }
    }
    #[inline]
    pub(crate) fn force_owner_kernel(&mut self) -> bool {
        match self {
            Self::Fixed(f) => f.owner = Owner::Kernel,
            Self::UnsafeRef(r) => r.owner = Owner::Kernel,
        }
        true
    }
}

/// When SendTo is sepcified, send(2) is turned into sendto(2)
#[derive(Clone, Debug)]
pub enum DestTo {
    V4(DestToV4),
    V6(DestToV6),
}

/// IPv4 SendTo
#[derive(Clone, Debug)]
pub struct DestToV4 {
    pub(crate) sockaddr: libc::sockaddr_in,
    pub(crate) socklen_t: libc::socklen_t,
}

/// IPv6 SendTo
#[derive(Clone, Debug)]
pub struct DestToV6 {
    pub(crate) sockaddr: libc::sockaddr_in6,
    pub(crate) socklen_t: libc::socklen_t,
}