Moved cache flushing operations out of bits64 to shared because they

apply to both.

1 files changed, 20 insertions, 0 deletions

diff --git a/src/shared/tlb.rs b/src/shared/tlb.rs
new file mode 100644
index 0000000..f99604f
--- /dev/null
+++ b/src/shared/tlb.rs
@@ -0,0 +1,20 @@
+//! Functions to flush the translation lookaside buffer (TLB).
+
+/// Invalidate the given address in the TLB using the `invlpg` instruction.
+///
+/// # Safety
+/// This function is unsafe as it causes a general protection fault (GP) if the current privilege
+/// level is not 0.
+pub unsafe fn flush(addr: usize) {
+    asm!("invlpg ($0)" :: "r" (addr) : "memory");
+}
+
+/// Invalidate the TLB completely by reloading the CR3 register.
+///
+/// # Safety
+/// This function is unsafe as it causes a general protection fault (GP) if the current privilege
+/// level is not 0.
+pub unsafe fn flush_all() {
+    use shared::control_regs::{cr3, cr3_write};
+    cr3_write(cr3())
+}