hirun 0.1.16

A concurrent framework for asynchronous programming based on event-driven, non-blocking I/O mechanism
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

use super::{Poll, POLLIN, POLLOUT};
use crate::{Error, Result};
use core::mem::MaybeUninit;
use core::ptr;
use hipool::{Allocator, Boxed, Pool, PoolAlloc};

pub(crate) type BoxedPollImpl<'a, A> = Boxed<'a, PollImpl<'a, A>, A>;

#[repr(C)]
pub(crate) struct PollImpl<'a, A: Allocator = PoolAlloc> {
    fd: i32,
    events: Boxed<'a, [MaybeUninit<libc::epoll_event>], A>,
}

unsafe impl<A: Allocator + Pool> Send for PollImpl<'static, A> {}

impl<'a, A: Allocator + 'a> Drop for PollImpl<'a, A> {
    fn drop(&mut self) {
        unsafe { libc::close(self.fd) };
    }
}

impl<'a, A: Allocator + 'a> PollImpl<'a, A> {
    pub fn new_in(pool: &'a A) -> Result<BoxedPollImpl<'a, A>> {
        let events = Boxed::uninit_slice_in::<libc::epoll_event>(pool, EPOLL_EVENT_MAX)?;
        let uninit = Boxed::uninit_in::<Self>(pool)?;
        let fd = unsafe { libc::epoll_create1(libc::EPOLL_CLOEXEC) };
        if fd >= 0 {
            let poll = uninit.write(Self { fd, events });
            Ok(poll)
        } else {
            Err(Error::last())
        }
    }
}

impl<'a, A: Allocator + 'a> Poll for PollImpl<'a, A> {
    fn add_event(&self, fd: i32, events: u32, e: u64) -> Result<()> {
        let mut event = libc::epoll_event { events, u64: e };
        let ret =
            unsafe { libc::epoll_ctl(self.fd, libc::EPOLL_CTL_ADD, fd, ptr::addr_of_mut!(event)) };
        if ret == 0 {
            Ok(())
        } else {
            Err(Error::last())
        }
    }

    fn mod_event(&self, fd: i32, events: u32, e: u64) -> Result<()> {
        let mut event = libc::epoll_event { events, u64: e };
        let ret =
            unsafe { libc::epoll_ctl(self.fd, libc::EPOLL_CTL_MOD, fd, ptr::addr_of_mut!(event)) };
        if ret == 0 {
            Ok(())
        } else {
            Err(Error::last())
        }
    }

    fn del_event(&self, fd: i32, e: u64) -> Result<()> {
        let mut event = libc::epoll_event { events: 0, u64: e };
        let ret =
            unsafe { libc::epoll_ctl(self.fd, libc::EPOLL_CTL_DEL, fd, ptr::addr_of_mut!(event)) };
        if ret == 0 {
            Ok(())
        } else {
            Err(Error::last())
        }
    }

    fn wait<F>(&mut self, timeout: i32, mut f: F) -> Result<u32>
    where
        F: FnMut(u32, u64),
    {
        let events = self.events.as_ptr().cast::<libc::epoll_event>();
        let cnt = unsafe { libc::epoll_wait(self.fd, events, self.events.len() as i32, timeout) };
        if cnt >= 0 {
            for event in &mut self.events[0..cnt as usize] {
                let event = unsafe { event.assume_init_ref() };
                let mut events = event.events & (POLLIN | POLLOUT);
                if events == 0 {
                    events = POLLIN | POLLOUT;
                }
                f(events, event.u64);
            }
            Ok(cnt as u32)
        } else {
            Err(Error::last())
        }
    }
}

const EPOLL_EVENT_MAX: usize = 512;