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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
#![cfg(any(target_arch = "x86", target_arch = "x86_64"))]
#![allow(stable_features)]
#![feature(const_fn, asm, repr_transparent)]
#![no_std]
#![cfg_attr(test, allow(unused_features))]
#[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 controlregs;
pub mod dtables;
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 xapic;
#[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;
}
#[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" :::: "volatile");
}
#[inline(always)]
pub unsafe fn rdpid() -> u64 {
let mut pid: u64;
asm!("rdpid $0" : "=r"(pid));
return pid;
}
#[cfg(test)]
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?");
}
}
}
}