Update lib.rs

delulu-hq · Sep 3, 2024 · d17e823 · d17e823
1 parent b886e94
commit d17e823
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -75,11 +75,11 @@ pub mod ext {
 
    #[inline]
    pub unsafe fn NtCurrentTeb() -> *mut TEB {
-      const TEB_OFFSET = mem::offset_of!(NT_TIB, Self_) as u32;
+      const TEB_OFFSET: u32 = mem::offset_of!(NT_TIB, Self_) as u32;
       #[cfg(target_arch = "x86_64")]
-      __readgsqword(TEB_OFFSET) as _
+      { __readgsqword(TEB_OFFSET) as _ }
       #[cfg(target_arch = "x86")]
-      __readfsdword(TEB_OFFSET) as _
+      { __readfsdword(TEB_OFFSET) as _ }
    }
 
    #[cfg(test)]