return ring enum instead of u16

gz · Nov 30, 2023 · 10d125a · 10d125a
1 parent 1a77405
commit 10d125a
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/src/segmentation.rs b/src/segmentation.rs
@@ -1,5 +1,6 @@
 //! Functionality to manipulate segment registers, build segement
 //! descriptors and selectors.
+use _core::convert::TryInto;
 use bitflags::*;
 
 use core::arch::asm;
@@ -45,8 +46,14 @@ impl SegmentSelector {
  }
 
  /// Returns the requested privilege level of the selector.
- pub fn rpl(&self) -> u16 {
- self.bits & 0b11
+ pub fn rpl(&self) -> Ring {
+ match self.bits & 0b11 {
+ 0b00 => Ring::Ring0,
+ 0b01 => Ring::Ring1,
+ 0b10 => Ring::Ring2,
+ 0b11 => Ring::Ring3,
+ _ => unreachable!(),
+ }
  }
 
  /// Returns the table indicator (TI) bit.
@@ -67,7 +74,13 @@ impl fmt::Display for SegmentSelector {
  true => "LDT Table",
  };
 
- write!(f, "Index {} in {}, RPL = {}", self.index(), tbl, self.rpl())
+ write!(
+ f,
+ "Index {} in {}, {:?} segment selector.",
+ self.index(),
+ tbl,
+ self.rpl()
+ )
  }
 }