Re: [PATCH v9 3/5] rust: Add missing SAFETY documentation for ARef example
From: Andreas Hindborg
Date: Wed Apr 09 2025 - 06:20:01 EST
"Oliver Mangold" <oliver.mangold@xxxxx> writes:
> SAFETY comment in rustdoc example was just 'TODO'. Fixed.
>
> Signed-off-by: Oliver Mangold <oliver.mangold@xxxxx>
> ---
> rust/kernel/types.rs | 4 ++--
> 1 file changed, 2 insertions(+), 2 deletions(-)
>
> diff --git a/rust/kernel/types.rs b/rust/kernel/types.rs
> index c8b78bcad259132808cc38c56b9f2bd525a0b755..db29f7c725e631c11099fa9122901ec2b3f4a039 100644
> --- a/rust/kernel/types.rs
> +++ b/rust/kernel/types.rs
> @@ -492,7 +492,7 @@ pub unsafe fn from_raw(ptr: NonNull<T>) -> Self {
> ///
> /// struct Empty {}
> ///
> - /// # // SAFETY: TODO.
> + /// // SAFETY: We do not free anything.
How about:
This implementation will never free the underlying object, so the
object is kept alive longer than the safety requirement specifies.
> /// unsafe impl RefCounted for Empty {
> /// fn inc_ref(&self) {}
> /// unsafe fn dec_ref(_obj: NonNull<Self>) {}
> @@ -500,7 +500,7 @@ pub unsafe fn from_raw(ptr: NonNull<T>) -> Self {
> ///
> /// let mut data = Empty {};
> /// let ptr = NonNull::<Empty>::new(&mut data).unwrap();
> - /// # // SAFETY: TODO.
> + /// // SAFETY: We keep `data` around longer than the `ARef`.
This is not sufficient. The safety requirement is:
Callers must ensure that the reference count was incremented at least once, and that they
are properly relinquishing one increment. That is, if there is only one increment, callers
must not use the underlying object anymore -- it is only safe to do so via the newly
created [`ARef`].
How about:
The `RefCounted` implementation for `Empty` does not count references
and never frees the underlying object. Thus we can act as having a
refcount on the object that we pass to the newly created `ARef`.
I think this example actually exposes a soundness hole. When the
underlying object is allocated on the stack, the safety requirements are
not sufficient to ensure the lifetime of the object. We could safely
return `data_ref` and have the underlying object go away. We should add
to the safety requirement of `ARef::from_raw`:
`ptr` must be valid while the refcount is positive.
That would allow the code in this example, but prevent the issue
outlined above.
Best regards,
Andreas Hindborg