axvm 0.5.25

Virtual Machine resource management crate for ArceOS's hypervisor variant.
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

//! AArch64 virtual timer device contribution.
//!
//! The vtimer is architecture-owned topology: it is exposed to the common
//! device preparation path through a normal factory/bundle contribution, but
//! the concrete `arm_vgic` backend stays inside the AArch64 AxVM boundary
//! instead of leaking into the architecture-neutral `axdevice` crate.

use alloc::sync::Arc;

use arm_vgic::vtimer::{VtimerState, get_sysreg_device_with_state};
use axdevice::{
    DeviceBuildContext, DeviceBundle, DeviceFactory, DeviceLifecycle, DeviceManagerResult,
    DeviceRegistration,
};
use axvm_types::{EmulatedDeviceConfig, EmulatedDeviceType};

/// Factory for the standard AArch64 CNT* virtual-timer sysreg devices.
pub(crate) struct Aarch64VtimerFactory;

struct VtimerLifecycle {
    state: Arc<VtimerState>,
}

impl DeviceLifecycle for VtimerLifecycle {
    fn reset(&self) -> DeviceManagerResult {
        self.state.reset();
        Ok(())
    }

    fn suspend(&self) -> DeviceManagerResult {
        self.state.suspend();
        Ok(())
    }

    fn resume(&self) -> DeviceManagerResult {
        self.state.resume();
        Ok(())
    }
}

impl Drop for VtimerLifecycle {
    fn drop(&mut self) {
        self.state.reset();
    }
}

impl Aarch64VtimerFactory {
    /// Builds the CNT* system-register device contributions.
    fn build_bundle(&self) -> DeviceManagerResult<DeviceBundle> {
        let (state, devices) = get_sysreg_device_with_state();
        let lifecycle: Arc<dyn DeviceLifecycle> = Arc::new(VtimerLifecycle { state });
        let mut bundle = DeviceBundle::new();
        for device in devices {
            bundle.push(DeviceRegistration::Device(device));
        }
        Ok(bundle.with_lifecycle(lifecycle))
    }
}

impl DeviceFactory for Aarch64VtimerFactory {
    fn device_type(&self) -> EmulatedDeviceType {
        EmulatedDeviceType::Aarch64Vtimer
    }

    fn build(
        &self,
        _config: &EmulatedDeviceConfig,
        _context: &DeviceBuildContext<'_>,
    ) -> DeviceManagerResult<DeviceBundle> {
        self.build_bundle()
    }
}