ax-driver 0.6.0

ArceOS rdrive driver registration and rdif binding collection
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

use alloc::{boxed::Box, string::String, vec::Vec};

use ax_errno::AxError;
use rdif_vsock::Interface;
use rdrive::{Device, DriverGeneric};

pub struct PlatformVsockDevice {
    name: String,
    vsock: Option<Box<dyn Interface>>,
}

impl PlatformVsockDevice {
    fn new(name: String, vsock: Box<dyn Interface>) -> Self {
        Self {
            name,
            vsock: Some(vsock),
        }
    }
}

impl DriverGeneric for PlatformVsockDevice {
    fn name(&self) -> &str {
        &self.name
    }
}

pub trait PlatformDeviceVsock {
    fn register_vsock<T>(self, dev: T)
    where
        T: Interface + 'static;
}

impl PlatformDeviceVsock for rdrive::PlatformDevice {
    fn register_vsock<T>(self, dev: T)
    where
        T: Interface + 'static,
    {
        let name = dev.name().into();
        self.register(PlatformVsockDevice::new(name, Box::new(dev)));
    }
}

pub fn take_vsock_devices() -> Result<Vec<Box<dyn Interface>>, AxError> {
    let mut devices = Vec::new();
    for dev in rdrive::get_list::<PlatformVsockDevice>() {
        devices.push(take_vsock_device(dev)?);
    }
    Ok(devices)
}

fn take_vsock_device(device: Device<PlatformVsockDevice>) -> Result<Box<dyn Interface>, AxError> {
    let mut device = device.lock().map_err(|_| AxError::BadState)?;
    device.vsock.take().ok_or(AxError::BadState)
}