libaio-futures 0.1.4

Straightforward Linux AIO using Futures/async/await.
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
101
102
103
104
105
106
107

// libaio ABI adapted from:
// https://raw.githubusercontent.com/jsgf/libaio-rust/master/src/aioabi.rs
#![allow(dead_code)]

use libc::{c_long, c_int, size_t};
pub use libc::timespec;
use std::mem::zeroed;
use std::default::Default;

#[repr(C)]
pub enum IOCmd {
    PRead = 0,
    PWrite = 1,
    FSync = 2,
    FdSync = 3,
    // 4 was the experimental IOCB_CMD_PREADX,
    Poll = 5,
    Noop = 6,
    PReadV = 7,
    PWriteV = 8,
}

pub const IOCB_FLAG_RESFD : u32 = 1 << 0;
pub const IOCB_FLAG_IOPRIO: u32 = 1 << 1;

// Taken from linux/include/linux/aio_abi.h
// This is a kernel ABI, so there should be no need to worry about it changing.
#[repr(C)]
pub struct IOCb {
    pub aio_data: u64,             // ends up in io_event.data
    // NOTE: the order of aio_key and aio_rw_flags could be byte-order depedent
    pub aio_key: u32,
    pub aio_rw_flags: u32,

    pub aio_lio_opcode: u16,
    pub aio_reqprio: u16,
    pub aio_fildes: u32,

    pub aio_buf: u64,
    pub aio_nbytes: u64,
    pub aio_offset: u64,

    pub aio_reserved2: u64,
    pub aio_flags: u32,
    pub aio_resfd: u32,
}

impl Default for IOCb {
    fn default() -> IOCb {
        IOCb {
            aio_lio_opcode: IOCmd::Noop as u16,
            aio_fildes: (-1_i32) as u32,
            .. unsafe { zeroed() }
        }
    }
}

#[derive(Clone)]
#[repr(C)]
pub struct IOEvent {
    pub data: u64,
    pub obj: u64,
    pub res: i64,
    pub res2: i64,
}

impl Default for IOEvent {
    fn default() -> IOEvent {
        unsafe { zeroed() }
    }
}

pub enum IOContext {}
pub type IOContextPtr = *mut IOContext;

#[repr(C)]
pub struct IOVector {
    pub iov_base: *mut u8,
    pub iov_len: size_t,
}

#[link(name = "aio")]
extern "C" {
    pub fn io_queue_init(maxevents: c_int, ctxp: *mut IOContextPtr) -> c_int;
    pub fn io_queue_release(ctx: IOContextPtr) -> c_int;
    pub fn io_queue_run(ctx: IOContextPtr) -> c_int;
    pub fn io_setup(maxevents: c_int, ctxp: *mut IOContextPtr) -> c_int;
    pub fn io_destroy(ctx: IOContextPtr) -> c_int;
    pub fn io_submit(ctx: IOContextPtr, nr: c_long, ios: *mut *mut IOCb) -> c_int;
    pub fn io_cancel(ctx: IOContextPtr, iocb: *mut IOCb, evt: *mut IOEvent) -> c_int;
    pub fn io_getevents(ctx_id: IOContextPtr, min_nr: c_long,
                        nr: c_long, events: *mut IOEvent,
                        timeout: *mut timespec) -> c_int;
    pub fn io_set_eventfd(iocb: *mut IOCb, eventfd: c_int);
}

#[cfg(test)]
mod test {
    use std::mem::size_of;

    #[test]
    fn test_sizes() {
        // Check against kernel ABI
        assert!(size_of::<super::IOEvent>() == 32);
        assert!(size_of::<super::IOCb>() == 64);
    }
}