1 files changed, 20 insertions, 0 deletions
diff --git a/src/bits64/tlb.rs b/src/bits64/tlb.rs
new file mode 100644
index 0000000..6b27e8c
--- /dev/null
+++ b/src/bits64/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 super::controlregs::{cr3, cr3_write};
+    cr3_write(cr3())
+}