mirror of
https://github.com/asterinas/asterinas.git
synced 2025-06-21 00:06:34 +00:00
Abstract the virtual_exception_handler function.
This commit is contained in:
Hsy-Intel
committed by
Tate, Hongliang Tian
Tate, Hongliang Tian
parent
4f0df44b29
commit
82518955d7
@ -4,7 +4,7 @@
|
||||
|
||||
extern crate alloc;
|
||||
|
||||
pub mod asm;
|
||||
mod asm;
|
||||
pub mod tdcall;
|
||||
pub mod tdvmcall;
|
||||
|
||||
@ -15,79 +15,23 @@ use raw_cpuid::{native_cpuid::cpuid_count, CpuIdResult};
|
||||
use tdcall::{InitError, TdgVeInfo, TdgVpInfo};
|
||||
use tdvmcall::*;
|
||||
|
||||
const TDX_CPUID_LEAF_ID: u64 = 0x21;
|
||||
|
||||
pub fn tdx_early_init() -> Result<TdgVpInfo, InitError> {
|
||||
match is_tdx_guest() {
|
||||
Ok(_) => Ok(tdcall::get_tdinfo()?),
|
||||
Err(err) => Err(err),
|
||||
}
|
||||
check_tdx_guest()?;
|
||||
Ok(tdcall::get_tdinfo()?)
|
||||
}
|
||||
|
||||
fn is_tdx_guest() -> Result<(), InitError> {
|
||||
fn check_tdx_guest() -> Result<(), InitError> {
|
||||
const TDX_CPUID_LEAF_ID: u64 = 0x21;
|
||||
let cpuid_leaf = cpuid_count(0, 0).eax as u64;
|
||||
if cpuid_leaf < TDX_CPUID_LEAF_ID {
|
||||
return Err(InitError::TdxCpuLeafIdError);
|
||||
}
|
||||
let cpuid_result: CpuIdResult = cpuid_count(TDX_CPUID_LEAF_ID as u32, 0);
|
||||
if &cpuid_result.ebx.to_ne_bytes() == b"Inte"
|
||||
&& &cpuid_result.ebx.to_ne_bytes() == b"lTDX"
|
||||
&& &cpuid_result.ecx.to_ne_bytes() == b" "
|
||||
if &cpuid_result.ebx.to_ne_bytes() != b"Inte"
|
||||
|| &cpuid_result.ebx.to_ne_bytes() != b"lTDX"
|
||||
|| &cpuid_result.ecx.to_ne_bytes() != b" "
|
||||
{
|
||||
Ok(())
|
||||
} else {
|
||||
Err(InitError::TdxVendorIdError)
|
||||
return Err(InitError::TdxVendorIdError);
|
||||
}
|
||||
}
|
||||
|
||||
pub trait TdxTrapFrame {
|
||||
fn rax(&self) -> usize;
|
||||
fn set_rax(&mut self, rax: usize);
|
||||
fn rbx(&self) -> usize;
|
||||
fn set_rbx(&mut self, rbx: usize);
|
||||
fn rcx(&self) -> usize;
|
||||
fn set_rcx(&mut self, rcx: usize);
|
||||
fn rdx(&self) -> usize;
|
||||
fn set_rdx(&mut self, rdx: usize);
|
||||
fn rsi(&self) -> usize;
|
||||
fn set_rsi(&mut self, rsi: usize);
|
||||
fn rdi(&self) -> usize;
|
||||
fn set_rdi(&mut self, rdi: usize);
|
||||
fn rip(&self) -> usize;
|
||||
fn set_rip(&mut self, rip: usize);
|
||||
}
|
||||
|
||||
pub fn virtual_exception_handler(trapframe: &mut impl TdxTrapFrame, ve_info: &TdgVeInfo) {
|
||||
match ve_info.exit_reason.into() {
|
||||
TdxVirtualExceptionType::Hlt => {
|
||||
serial_println!("Ready to halt");
|
||||
hlt();
|
||||
}
|
||||
TdxVirtualExceptionType::Io => {
|
||||
if !handle_io(trapframe, ve_info) {
|
||||
serial_println!("Handle tdx ioexit errors, ready to halt");
|
||||
hlt();
|
||||
}
|
||||
}
|
||||
TdxVirtualExceptionType::MsrRead => {
|
||||
let msr = rdmsr(trapframe.rcx() as u32).unwrap();
|
||||
trapframe.set_rax((msr as u32 & u32::MAX) as usize);
|
||||
trapframe.set_rdx(((msr >> 32) as u32 & u32::MAX) as usize);
|
||||
}
|
||||
TdxVirtualExceptionType::MsrWrite => {
|
||||
let data = trapframe.rax() as u64 | ((trapframe.rdx() as u64) << 32);
|
||||
wrmsr(trapframe.rcx() as u32, data).unwrap();
|
||||
}
|
||||
TdxVirtualExceptionType::CpuId => {
|
||||
let cpuid_info = cpuid(trapframe.rax() as u32, trapframe.rcx() as u32).unwrap();
|
||||
let mask = 0xFFFF_FFFF_0000_0000_usize;
|
||||
trapframe.set_rax((trapframe.rax() & mask) | cpuid_info.eax);
|
||||
trapframe.set_rbx((trapframe.rbx() & mask) | cpuid_info.ebx);
|
||||
trapframe.set_rcx((trapframe.rcx() & mask) | cpuid_info.ecx);
|
||||
trapframe.set_rdx((trapframe.rdx() & mask) | cpuid_info.edx);
|
||||
}
|
||||
TdxVirtualExceptionType::Other => panic!("Unknown TDX vitrual exception type"),
|
||||
_ => return,
|
||||
}
|
||||
trapframe.set_rip(trapframe.rip() + ve_info.exit_instruction_length as usize);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
@ -5,7 +5,7 @@
|
||||
//! resumes the TD via a SEAMCALL [TDH.VP.ENTER] invocation.
|
||||
extern crate alloc;
|
||||
|
||||
use crate::{asm::asm_td_vmcall, tdcall::TdgVeInfo, TdxTrapFrame};
|
||||
use crate::{asm::asm_td_vmcall, tdcall::TdgVeInfo};
|
||||
use alloc::fmt;
|
||||
use bitflags::bitflags;
|
||||
use core::fmt::Write;
|
||||
@ -193,38 +193,6 @@ pub fn write_mmio(size: u64, mmio_addr: u64, data: u64) -> Result<(), TdVmcallEr
|
||||
}
|
||||
}
|
||||
|
||||
pub fn handle_io(trapframe: &mut impl TdxTrapFrame, ve_info: &TdgVeInfo) -> bool {
|
||||
let size = match ve_info.exit_qualification & 0x3 {
|
||||
0 => 1,
|
||||
1 => 2,
|
||||
3 => 4,
|
||||
_ => panic!("Invalid size value"),
|
||||
};
|
||||
let direction = if (ve_info.exit_qualification >> 3) & 0x1 == 0 {
|
||||
Direction::Out
|
||||
} else {
|
||||
Direction::In
|
||||
};
|
||||
let string = (ve_info.exit_qualification >> 4) & 0x1 == 1;
|
||||
let repeat = (ve_info.exit_qualification >> 5) & 0x1 == 1;
|
||||
let operand = if (ve_info.exit_qualification >> 6) & 0x1 == 0 {
|
||||
Operand::Dx
|
||||
} else {
|
||||
Operand::Immediate
|
||||
};
|
||||
let port = (ve_info.exit_qualification >> 16) as u16;
|
||||
|
||||
match direction {
|
||||
Direction::In => {
|
||||
trapframe.set_rax(io_read(size, port).unwrap() as usize);
|
||||
}
|
||||
Direction::Out => {
|
||||
io_write(size, port, trapframe.rax() as u32).unwrap();
|
||||
}
|
||||
};
|
||||
true
|
||||
}
|
||||
|
||||
macro_rules! io_read {
|
||||
($port:expr, $ty:ty) => {{
|
||||
let mut args = TdVmcallArgs {
|
||||
@ -286,22 +254,42 @@ fn td_vmcall(args: &mut TdVmcallArgs) -> Result<(), TdVmcallError> {
|
||||
}
|
||||
|
||||
bitflags! {
|
||||
struct LineSts: u8 {
|
||||
const INPUT_FULL = 1;
|
||||
const OUTPUT_EMPTY = 1 << 5;
|
||||
}
|
||||
struct LineSts: u8 {
|
||||
const INPUT_FULL = 1;
|
||||
const OUTPUT_EMPTY = 1 << 5;
|
||||
}
|
||||
}
|
||||
|
||||
fn line_sts() -> LineSts {
|
||||
fn read_line_sts() -> LineSts {
|
||||
LineSts::from_bits_truncate(unsafe { PortRead::read_from_port(SERIAL_LINE_STS) })
|
||||
}
|
||||
|
||||
struct Serial;
|
||||
|
||||
impl Serial {
|
||||
fn serial_write_byte(byte: u8) {
|
||||
match byte {
|
||||
// Backspace/Delete
|
||||
8 | 0x7F => {
|
||||
while !read_line_sts().contains(LineSts::OUTPUT_EMPTY) {}
|
||||
io_write!(SERIAL_IO_PORT, 8, u8).unwrap();
|
||||
while !read_line_sts().contains(LineSts::OUTPUT_EMPTY) {}
|
||||
io_write!(SERIAL_IO_PORT, b' ', u8).unwrap();
|
||||
while !read_line_sts().contains(LineSts::OUTPUT_EMPTY) {}
|
||||
io_write!(SERIAL_IO_PORT, 8, u8).unwrap();
|
||||
}
|
||||
_ => {
|
||||
while !read_line_sts().contains(LineSts::OUTPUT_EMPTY) {}
|
||||
io_write!(SERIAL_IO_PORT, byte, u8).unwrap();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Write for Serial {
|
||||
fn write_str(&mut self, s: &str) -> fmt::Result {
|
||||
for &c in s.as_bytes() {
|
||||
serial_write_byte(c);
|
||||
Serial::serial_write_byte(c);
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
@ -313,23 +301,6 @@ pub fn print(args: fmt::Arguments) {
|
||||
.expect("Failed to write to serial port");
|
||||
}
|
||||
|
||||
fn serial_write_byte(byte: u8) {
|
||||
match byte {
|
||||
8 | 0x7F => {
|
||||
while !line_sts().contains(LineSts::OUTPUT_EMPTY) {}
|
||||
io_write!(SERIAL_IO_PORT, 8, u8).unwrap();
|
||||
while !line_sts().contains(LineSts::OUTPUT_EMPTY) {}
|
||||
io_write!(SERIAL_IO_PORT, b' ', u8).unwrap();
|
||||
while !line_sts().contains(LineSts::OUTPUT_EMPTY) {}
|
||||
io_write!(SERIAL_IO_PORT, 8, u8).unwrap();
|
||||
}
|
||||
_ => {
|
||||
while !line_sts().contains(LineSts::OUTPUT_EMPTY) {}
|
||||
io_write!(SERIAL_IO_PORT, byte, u8).unwrap();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[macro_export]
|
||||
macro_rules! serial_print {
|
||||
($fmt: literal $(, $($arg: tt)+)?) => {
|
||||
@ -339,7 +310,7 @@ macro_rules! serial_print {
|
||||
|
||||
#[macro_export]
|
||||
macro_rules! serial_println {
|
||||
($fmt: literal $(, $($arg: tt)+)?) => {
|
||||
$crate::tdvmcall::print(format_args!(concat!($fmt, "\n") $(, $($arg)+)?))
|
||||
}
|
||||
($fmt: literal $(, $($arg: tt)+)?) => {
|
||||
$crate::tdvmcall::print(format_args!(concat!($fmt, "\n") $(, $($arg)+)?))
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user