axvm 0.5.18

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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

//! vCPU construction and setup for VM preparation.

use alloc::{boxed::Box, sync::Arc, vec::Vec};

use ax_errno::AxResult;
use axvm_types::{EmulatedDeviceType, GuestPhysAddr};

use super::super::{AxVCpuRef, AxVMResources, VCpu};
use crate::{
    arch::{ArchOps, CurrentArch, VcpuCreateContext, VcpuSetupContext},
    config::{GuestBootPolicy, VMBootProtocol},
};

pub(super) struct PreparedVcpus {
    vcpus: Vec<AxVCpuRef>,
}

impl PreparedVcpus {
    pub(super) fn create(
        vm_id: usize,
        resources: &AxVMResources,
        dtb_addr: Option<GuestPhysAddr>,
    ) -> AxResult<Self> {
        let vcpu_id_pcpu_sets = resources.config.phys_cpu_ls.get_vcpu_affinities_pcpu_ids();
        let create_state = CurrentArch::new_vcpu_create_state(&vcpu_id_pcpu_sets)?;
        let firmware_boot = guest_uses_firmware_boot(resources);

        debug!("dtb_load_gpa: {dtb_addr:?}");
        debug!("id: {vm_id}, VCpuIdPCpuSets: {vcpu_id_pcpu_sets:#x?}");

        let mut vcpus = Vec::with_capacity(vcpu_id_pcpu_sets.len());
        for (vcpu_id, phys_cpu_set, phys_cpu_id) in vcpu_id_pcpu_sets {
            let arch_config = CurrentArch::build_vcpu_create_config(
                &create_state,
                VcpuCreateContext {
                    vcpu_id,
                    phys_cpu_id,
                    dtb_addr,
                    firmware_boot,
                },
            )?;

            // FIXME: VCpu is neither `Send` nor `Sync` by design, check whether
            // 1. we should make it `Send` and `Sync`, or
            // 2. we can guarantee that no cross-thread access is performed
            #[allow(clippy::arc_with_non_send_sync)]
            vcpus.push(Arc::new(VCpu::new(
                vm_id,
                vcpu_id,
                phys_cpu_set,
                arch_config,
            )?));
        }

        Ok(Self { vcpus })
    }

    pub(super) fn setup(&self, resources: &AxVMResources) -> AxResult {
        for vcpu in &self.vcpus {
            let setup_config = CurrentArch::build_vcpu_setup_config(VcpuSetupContext {
                interrupt_mode: resources.config.interrupt_mode(),
                emulates_console: resources
                    .config
                    .emu_devices()
                    .iter()
                    .any(|dev| dev.emu_type == EmulatedDeviceType::Console),
                passthrough_ports: resources.config.pass_through_ports(),
                firmware_boot: guest_uses_firmware_boot(resources),
            })?;

            let entry = if vcpu.id() == 0 {
                resources.config.bsp_entry()
            } else {
                resources.config.ap_entry()
            };

            debug!("Setting up vCPU[{}] entry at {:#x}", vcpu.id(), entry);

            vcpu.setup(
                entry,
                resources.address_space.page_table_root(),
                setup_config,
            )?;
        }
        Ok(())
    }

    pub(super) fn into_boxed_slice(self) -> Box<[AxVCpuRef]> {
        self.vcpus.into_boxed_slice()
    }
}

fn guest_uses_firmware_boot(resources: &AxVMResources) -> bool {
    matches!(
        resources.config.boot_policy(),
        GuestBootPolicy::AdjustKernelForBootProtocol {
            protocol: VMBootProtocol::Uefi,
        }
    )
}