Re: [PATCH v5 1/4] rust: uaccess: add userspace pointers

From: Benno Lossin
Date: Mon Apr 15 2024 - 05:37:53 EST


On 15.04.24 09:13, Alice Ryhl wrote:
> +impl UserSlice {
> + /// Constructs a user slice from a raw pointer and a length in bytes.
> + ///
> + /// Constructing a [`UserSlice`] performs no checks on the provided address and length, it can
> + /// safely be constructed inside a kernel thread with no current userspace process. Reads and
> + /// writes wrap the kernel APIs `copy_from_user` and `copy_to_user`, which check the memory map
> + /// of the current process and enforce that the address range is within the user range (no
> + /// additional calls to `access_ok` are needed).
> + ///
> + /// Callers must be careful to avoid time-of-check-time-of-use (TOCTOU) issues. The simplest way
> + /// is to create a single instance of [`UserSlice`] per user memory block as it reads each byte
> + /// at most once.
> + pub fn new(ptr: *mut c_void, length: usize) -> Self {

What would happen if I call this with a kernel pointer and then
read/write to it? For example

let mut arr = [MaybeUninit::uninit(); 64];
let ptr: *mut [MaybeUninit<u8>] = &mut arr;
let ptr = ptr.cast::<c_void>();

let slice = UserSlice::new(ptr, 64);
let (mut r, mut w) = slice.reader_writer();

r.read_raw(&mut arr)?;
// SAFETY: `arr` was initialized above.
w.write_slice(unsafe { MaybeUninit::slice_assume_init_ref(&arr) })?;

I think this would violate the exclusivity of `&mut` without any
`unsafe` code. (the `unsafe` block at the end cannot possibly be wrong)

> + UserSlice { ptr, length }
> + }

[...]

> + /// Returns `true` if no data is available in the io buffer.
> + pub fn is_empty(&self) -> bool {
> + self.length == 0
> + }
> +
> + /// Reads raw data from the user slice into a kernel buffer.
> + ///
> + /// After a successful call to this method, all bytes in `out` are initialized.

I think we should put things like this into a `# Guarantees` section.

--
Cheers,
Benno

> + ///
> + /// Fails with `EFAULT` if the read happens on a bad address.
> + pub fn read_raw(&mut self, out: &mut [MaybeUninit<u8>]) -> Result {
> + let len = out.len();
> + let out_ptr = out.as_mut_ptr().cast::<c_void>();
> + if len > self.length {
> + return Err(EFAULT);
> + }
> + let Ok(len_ulong) = c_ulong::try_from(len) else {
> + return Err(EFAULT);
> + };
> + // SAFETY: `out_ptr` points into a mutable slice of length `len_ulong`, so we may write
> + // that many bytes to it.
> + let res = unsafe { bindings::copy_from_user(out_ptr, self.ptr, len_ulong) };
> + if res != 0 {
> + return Err(EFAULT);
> + }
> + // Userspace pointers are not directly dereferencable by the kernel, so we cannot use `add`,
> + // which has C-style rules for defined behavior.
> + self.ptr = self.ptr.wrapping_byte_add(len);
> + self.length -= len;
> + Ok(())
> + }