From c81c4de223583b23d2f91f0978e378ef1dc3676f Mon Sep 17 00:00:00 2001
From: Jeremy Soller <jeremy@system76.com>
Date: Mon, 3 May 2021 21:29:36 -0600
Subject: [PATCH] Add method to get TableKind from VirtualAddress

---
 src/lib.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index 81d047f23a..c15793c598 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -69,6 +69,15 @@ impl VirtualAddress {
     pub fn add(self, offset: usize) -> Self {
         Self(self.0 + offset)
     }
+
+    #[inline(always)]
+    pub fn kind(&self) -> TableKind {
+        if (self.0 as isize) < 0 {
+            TableKind::Kernel
+        } else {
+            TableKind::User
+        }
+    }
 }
 
 #[derive(Clone, Copy, Debug)]