Add a method to consume a `Box<T, A>` and return a `NonNull<T>`. This is a convenience wrapper around `Self::into_raw` for callers that need a `NonNull` pointer rather than a raw pointer.
Signed-off-by: Andreas Hindborg <[email protected]> --- rust/kernel/alloc/kbox.rs | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/rust/kernel/alloc/kbox.rs b/rust/kernel/alloc/kbox.rs index 622b3529edfcb..e6efdd572aeea 100644 --- a/rust/kernel/alloc/kbox.rs +++ b/rust/kernel/alloc/kbox.rs @@ -213,6 +213,14 @@ pub fn leak<'a>(b: Self) -> &'a mut T { // which points to an initialized instance of `T`. unsafe { &mut *Box::into_raw(b) } } + + /// Consumes the `Box<T,A>` and returns a `NonNull<T>`. + /// + /// Like [`Self::into_raw`], but returns a `NonNull`. + pub fn into_nonnull(b: Self) -> NonNull<T> { + // SAFETY: `KBox::into_raw` returns a valid pointer. + unsafe { NonNull::new_unchecked(Self::into_raw(b)) } + } } impl<T, A> Box<MaybeUninit<T>, A> -- 2.51.2
