axvm 0.6.0

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

//! Shared system-register device exits used by AArch64 and x86_64 guests.

use axdevice::DeviceManagerError;
use axdevice_base::{BusKind, DeviceAccess, DeviceError, DeviceVcpuId};
use axvm_types::{AccessWidth, SysRegAddr, VmArchVcpuOps};

use crate::{AxVmError, AxVmResult, architecture::BoundVcpuExit};

#[derive(Clone, Copy, Debug)]
pub(crate) struct SysRegReadExit {
    pub(crate) addr: SysRegAddr,
    pub(crate) reg: usize,
}

#[derive(Clone, Copy, Debug)]
pub(crate) struct SysRegWriteExit {
    pub(crate) addr: SysRegAddr,
    pub(crate) value: u64,
}

pub(crate) fn handle_read<V: VmArchVcpuOps, D>(
    vm: &crate::AxVM,
    vcpu: &crate::vm::AxVCpuRef<V>,
    exit: SysRegReadExit,
) -> AxVmResult<BoundVcpuExit<D>> {
    let access = sysreg_access(vcpu.id(), exit.addr);
    let val = vm
        .get_devices()?
        .try_read(&access)
        .map_err(|error| AxVmError::device("read guest system register", error))?
        .ok_or_else(|| missing_sysreg_error("read", exit.addr))?;
    vcpu.set_gpr(exit.reg, val as usize);
    Ok(BoundVcpuExit::Continue)
}

pub(crate) fn handle_write<V: VmArchVcpuOps, D>(
    vm: &crate::AxVM,
    vcpu: &crate::vm::AxVCpuRef<V>,
    exit: SysRegWriteExit,
) -> AxVmResult<BoundVcpuExit<D>> {
    let access = sysreg_access(vcpu.id(), exit.addr);
    if !vm.try_write_device(&access, exit.value)? {
        return Err(missing_sysreg_error("write", exit.addr));
    }
    Ok(BoundVcpuExit::Continue)
}

fn sysreg_access(vcpu_id: usize, addr: SysRegAddr) -> DeviceAccess {
    DeviceAccess::new(
        DeviceVcpuId::new(vcpu_id),
        BusKind::SysReg,
        addr.addr() as u64,
        AccessWidth::Qword,
    )
}

fn missing_sysreg_error(operation: &'static str, addr: SysRegAddr) -> AxVmError {
    AxVmError::device(
        "access guest system register",
        DeviceManagerError::Access {
            operation,
            bus: BusKind::SysReg,
            addr: addr.addr() as u64,
            width: AccessWidth::Qword,
            source: DeviceError::NotFound,
        },
    )
}