vmx_lab 0.1.1

vt-x
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

use {
    crate::error::HypervisorError,
    alloc::boxed::Box,
    core::mem::size_of,
    static_assertions::const_assert_eq,
};

/// The size of the kernel stack in bytes.
pub const KERNEL_STACK_SIZE: usize = 0x6000;

/// The size reserved for host RSP. This includes space allocated for padding.
pub const STACK_CONTENTS_SIZE: usize = KERNEL_STACK_SIZE - size_of::<*mut u64>() * 4;

/// Represents the Virtual Machine Stack (VmStack).
///
/// The structure is designed to align with 4-KByte boundaries and ensures proper setup for the host RSP during VM execution.
/// 使用了原始指针但是我还是实现了 send 和sync 不知道安全不安全 但是方便
#[repr(C, align(4096))]
#[derive(Debug, Clone, Copy)]
pub struct VmStack {
    /// The main contents of the VM stack during VM-exit. VMCS_HOST_RSP points to the end of this array inside the VMCS.
    pub stack_contents: [u8; STACK_CONTENTS_SIZE],

    /// A pointer to the `Vmx` instance, needed for the `launch_vm` assembly function, which is passed to vmexit handler.
    /// Padding to ensure the Host RSP remains 16-byte aligned.
    pub vmx: *mut u64,

    /// Padding to ensure the Host RSP remains 16-byte aligned.
    pub padding_3: u64,

    /// Padding to ensure the Host RSP remains 16-byte aligned.
    pub padding_2: u64,

    /// Padding to ensure the Host RSP remains 16-byte aligned.
    pub padding_1: u64,
}

unsafe impl Send for VmStack  {}
unsafe impl Sync for VmStack {}

const_assert_eq!(size_of::<VmStack>(), KERNEL_STACK_SIZE);
const_assert_eq!(size_of::<VmStack>() % 4096, 0);

impl VmStack {
    /// Sets up the VMCS_HOST_RSP region.
    ///
    /// Initializes the VM stack, ensuring it's properly aligned and configured for host execution.
    ///
    /// # Arguments
    ///
    /// * `vmstack` - A mutable reference to the VM stack.
    ///
    /// # Returns
    ///
    /// A `Result` indicating the success or failure of the setup process.
    pub fn setup(vmstack: &mut Box<VmStack, impl core::alloc::Allocator>) -> Result<(), HypervisorError> {
        log::debug!("Setting up VMCS_HOST_RSP region");
        log::trace!("VMCS_HOST_RSP Virtual Address: {:p}", vmstack);

        // Initialize the VM stack contents and reserved space.
        vmstack.stack_contents = [0u8; STACK_CONTENTS_SIZE];

        // We don't null `vmx` because it should already be populated and we don't want to overwrite it.

        vmstack.padding_3 = u64::MAX;
        vmstack.padding_2 = u64::MAX;
        vmstack.padding_1 = u64::MAX;

        log::debug!("VMCS_HOST_RSP setup successfully!");

        Ok(())
    }
}