Merge pull request #12 from phil-opp/patch-2

Add method to retrieve current code segment selector

1 files changed, 7 insertions, 0 deletions

diff --git a/src/segmentation.rs b/src/segmentation.rs
index cfda3bd..e3c259e 100644
--- a/src/segmentation.rs
+++ b/src/segmentation.rs
@@ -205,3 +205,10 @@ pub unsafe fn load_cs(sel: SegmentSelector) {
          lretq
          1:" :: "r" (sel.bits() as u64) : "{rax}" "memory");
 }
+
+/// Returns the current value of the code segment register.
+pub fn cs() -> SegmentSelector {
+    let segment: u16;
+    unsafe { asm!("mov %cs, $0" : "=r" (segment) ) };
+    SegmentSelector::from_raw(segment)
+}