fix: Extender::extend_fn_once use after free

zetanumbers · May 29, 2024 · b9b8cc4 · b9b8cc4
1 parent 5edc06e
commit b9b8cc4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/lib.rs b/src/lib.rs
@@ -329,7 +329,8 @@ impl<'a, T> RefOnce<'a, T> {
 impl<T: ?Sized> RefOnce<'_, T> {
     // TODO: make public
     fn into_raw(this: Self) -> *mut Once<T> {
-        ptr::addr_of_mut!(*this.slot)
+        let this = mem::ManuallyDrop::new(this);
+        unsafe { ptr::addr_of!(this.slot).read() }
     }
 }