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
|
use std::fs;
use std::io;
use crate::{ioctl, kvm_sys, KvmRun};
pub enum KvmExit<'cpu> {
Halt,
IoIn(u16, &'cpu mut [u8]),
IoOut(u16, &'cpu [u8]),
MmioRead(u64, &'cpu mut [u8]),
MmioWrite(u64, &'cpu [u8]),
}
pub struct Vcpu {
vcpu: fs::File,
kvm_run: KvmRun,
}
impl Vcpu {
pub(crate) fn new(vcpu: fs::File, kvm_run: KvmRun) -> Vcpu {
Vcpu { vcpu, kvm_run }
}
pub fn get_regs(&self) -> io::Result<kvm_sys::kvm_regs> {
let mut regs = kvm_sys::kvm_regs::default();
ioctl(
&self.vcpu,
kvm_sys::KVM_GET_REGS,
&mut regs as *mut _ as u64,
)?;
Ok(regs)
}
pub fn set_regs(&self, regs: kvm_sys::kvm_regs) -> io::Result<()> {
ioctl(&self.vcpu, kvm_sys::KVM_SET_REGS, ®s as *const _ as u64).map(|_| ())
}
pub fn get_sregs(&self) -> io::Result<kvm_sys::kvm_sregs> {
let mut sregs = kvm_sys::kvm_sregs::default();
ioctl(
&self.vcpu,
kvm_sys::KVM_GET_SREGS,
&mut sregs as *mut _ as u64,
)?;
Ok(sregs)
}
pub fn set_sregs(&self, sregs: kvm_sys::kvm_sregs) -> io::Result<()> {
ioctl(
&self.vcpu,
kvm_sys::KVM_SET_SREGS,
&sregs as *const _ as u64,
)
.map(|_| ())
}
pub fn run(&mut self) -> io::Result<KvmExit<'_>> {
ioctl(&self.vcpu, kvm_sys::KVM_RUN, 0)?;
let kvm_run = self.kvm_run.as_mut();
match kvm_run.exit_reason as u64 {
kvm_sys::KVM_EXIT_HLT => Ok(KvmExit::Halt),
kvm_sys::KVM_EXIT_IO => {
// Safe to use union `io` field, as Kernel instructed us to.
let io = unsafe { kvm_run.inner.io };
let kvm_run_ptr = kvm_run as *mut kvm_sys::kvm_run as *mut u8;
// Create IO buffer located at `kvm_run + io.offset`.
let data = unsafe {
std::slice::from_raw_parts_mut(
kvm_run_ptr.offset(io.data_offset as isize),
io.count /* num blocks */ as usize * io.size /* bytes per block */ as usize,
)
};
match io.direction as u64 {
kvm_sys::KVM_EXIT_IO_IN => Ok(KvmExit::IoIn(io.port, data)),
kvm_sys::KVM_EXIT_IO_OUT => Ok(KvmExit::IoOut(io.port, data)),
_ => unreachable!(),
}
}
kvm_sys::KVM_EXIT_MMIO => {
// Safe to use union `mmio` filed, as Kernel instructed us to.
let mmio = unsafe { &mut kvm_run.inner.mmio };
let len = mmio.len as usize;
match mmio.is_write {
0 => Ok(KvmExit::MmioRead(mmio.phys_addr, &mut mmio.data[..len])),
1 => Ok(KvmExit::MmioWrite(mmio.phys_addr, &mmio.data[..len])),
_ => unreachable!(),
}
}
r @ _ => {
todo!("KVM_EXIT_... (exit_reason={}) not implemented!", r)
}
}
}
}
|
