compio-driver 0.10.0

Low-level driver for compio
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

pub use std::rc::Rc as RefPtr;
use std::{
    cell::{Cell, RefCell},
    sync::atomic::Ordering,
    task::Waker,
};

#[derive(Debug)]
pub struct WaitFlag(Cell<bool>);

impl WaitFlag {
    pub fn new(v: bool) -> Self {
        Self(Cell::new(v))
    }

    pub fn load(&self, _ordering: Ordering) -> bool {
        self.0.get()
    }

    pub fn swap(&self, v: bool, _ordering: Ordering) -> bool {
        let old = self.0.get();
        self.0.set(v);
        old
    }
}

#[derive(Debug)]
pub struct WakerRegistry {
    waker: RefCell<Option<Waker>>,
}

impl WakerRegistry {
    pub fn new() -> Self {
        Self {
            waker: RefCell::new(None),
        }
    }

    pub fn register(&self, waker: &Waker) {
        self.waker.borrow_mut().replace(waker.clone());
    }

    pub fn wake(&self) {
        if let Some(waker) = self.waker.borrow_mut().take() {
            waker.wake()
        }
    }
}