use std::{
marker::PhantomData,
mem::{self, MaybeUninit},
ops::Deref,
ptr::{addr_of, NonNull},
};
#[derive(Debug)]
pub struct Owned<T, Init = T> {
ptr: NonNull<T>,
marker: PhantomData<NonNull<Init>>,
}
impl<T> Owned<T> {
pub fn new_uninit() -> Owned<MaybeUninit<T>, T> {
let boxed = Box::new(MaybeUninit::<T>::uninit());
Owned {
ptr: unsafe {
NonNull::new_unchecked(Box::into_raw(boxed))
},
marker: PhantomData,
}
}
pub unsafe fn assume_init(
definitely_init: Owned<MaybeUninit<T>, T>,
) -> Owned<T> {
let ptr = definitely_init.ptr;
mem::forget(definitely_init);
Owned {
ptr: ptr.cast(),
marker: PhantomData,
}
}
}
#[repr(transparent)]
#[derive(Debug)]
pub struct InitPtr<T> {
pub ptr: *mut T,
}
impl<T, Init> Deref for Owned<T, Init> {
type Target = InitPtr<Init>;
fn deref(&self) -> &Self::Target {
unsafe { &*addr_of!(self.ptr).cast::<InitPtr<Init>>() }
}
}
impl<T, Init> Drop for Owned<T, Init> {
fn drop(&mut self) {
let _ = unsafe { Box::from_raw(self.ptr.as_ptr()) };
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_new_uninit() {
let uninit_owned: Owned<MaybeUninit<i32>, i32> =
Owned::new_uninit();
assert_eq!(
uninit_owned.ptr.as_ptr(),
uninit_owned.ptr.as_ptr()
);
}
#[test]
fn test_assume_init() {
let uninit_owned: Owned<MaybeUninit<i32>, i32> =
Owned::new_uninit();
let init_owned: Owned<i32> =
unsafe { Owned::assume_init(uninit_owned) };
assert_eq!(init_owned.ptr.as_ptr(), init_owned.ptr.as_ptr());
}
#[test]
fn test_deref() {
let uninit_owned: Owned<MaybeUninit<i32>, i32> =
Owned::new_uninit();
let init_ptr = uninit_owned.ptr.as_ptr();
assert_eq!(
uninit_owned.deref().ptr as *mut MaybeUninit<i32>,
init_ptr
);
}
#[test]
fn test_drop() {
let uninit_owned: Owned<MaybeUninit<i32>, i32> =
Owned::new_uninit();
drop(uninit_owned); }
}