unix: Unsafe-wrap stack_overflow::cleanup

rust-lang · Jul 17, 2024 · c496f05 · c496f05
1 parent 9ad696a
commit c496f05
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/library/std/src/sys/pal/unix/stack_overflow.rs b/library/std/src/sys/pal/unix/stack_overflow.rs
@@ -157,8 +157,11 @@ mod imp {
         mem::forget(handler);
     }
 
+    /// # Safety
+    /// Must be called only once
+    #[forbid(unsafe_op_in_unsafe_fn)]
     pub unsafe fn cleanup() {
-        drop_handler(MAIN_ALTSTACK.load(Ordering::Relaxed));
+        unsafe { drop_handler(MAIN_ALTSTACK.load(Ordering::Relaxed)) };
     }
 
     unsafe fn get_stack() -> libc::stack_t {