#![cfg(any(target_arch = "x86", target_arch = "x86_64"))]
#![no_std]
#![cfg_attr(test, allow(unused_features))]
#![cfg_attr(all(test, feature = "vmtest"), feature(custom_test_frameworks))]
#![cfg_attr(all(test, feature = "vmtest"), test_runner(x86test::runner::runner))]
#![cfg_attr(feature = "unstable", feature(step_trait))]
use core::arch::asm;
#[cfg(target_arch = "x86")]
pub(crate) use core::arch::x86 as arch;
#[cfg(target_arch = "x86_64")]
pub(crate) use core::arch::x86_64 as arch;
macro_rules! bit {
($x:expr) => {
1 << $x
};
}
pub mod bits16;
pub mod bits32;
pub mod bits64;
pub mod apic;
pub mod controlregs;
pub mod debugregs;
pub mod dtables;
pub mod fence;
pub mod io;
pub mod irq;
pub mod msr;
pub mod random;
pub mod segmentation;
pub mod task;
pub mod time;
pub mod tlb;
pub mod vmx;
#[cfg(feature = "performance-counter")]
pub mod perfcnt;
pub mod current {
#[cfg(target_arch = "x86")]
pub use crate::bits32::*;
#[cfg(target_arch = "x86_64")]
pub use crate::bits64::*;
}
pub mod cpuid {
pub use raw_cpuid::*;
}
#[cfg(not(test))]
mod std {
pub use core::fmt;
pub use core::ops;
pub use core::option;
}
#[cfg(all(test, feature = "vmtest"))]
extern crate klogger;
#[cfg(all(test, feature = "vmtest"))]
extern crate x86test;
#[derive(Copy, Clone, Debug, Eq, PartialEq)]
#[repr(u8)]
pub enum Ring {
Ring0 = 0b00,
Ring1 = 0b01,
Ring2 = 0b10,
Ring3 = 0b11,
}
#[inline(always)]
pub unsafe fn halt() {
asm!("hlt", options(att_syntax, nomem, nostack)); }
#[cfg(all(test, feature = "vmtest"))]
mod x86testing {
use super::*;
use x86test::*;
#[x86test(should_halt)]
fn should_halt() {
unsafe { halt() }
}
#[x86test]
fn should_not_halt() {}
}
#[inline(always)]
pub unsafe fn rdpid() -> u64 {
#[cfg(target_pointer_width = "64")]
let mut pid: u64;
#[cfg(target_pointer_width = "32")]
let mut pid: u32;
asm!("rdpid {pid}", pid = out(reg) pid, options(att_syntax));
pid.into()
}
#[cfg(all(test, feature = "utest"))]
mod test {
use super::*;
#[test]
fn test_rdpid() {
let rdpid_support = cpuid::CpuId::new()
.get_extended_feature_info()
.map_or(false, |finfo| finfo.has_rdpid());
unsafe {
if rdpid_support {
let pid1 = rdpid();
let pid2 = rdpid();
assert!(pid1 == pid2, "RDPID not consistent values?");
}
}
}
}