Crates.io | uninit |
lib.rs | uninit |
version | 0.6.2 |
source | src |
created_at | 2019-08-28 14:35:25.089793 |
updated_at | 2023-11-20 21:36:32.832539 |
description | A collection of tools for a safer usage of uninitialized memory |
homepage | https://crates.io/crates/uninit |
repository | https://github.com/danielhenrymantilla/rust-uninit |
max_upload_size | |
id | 160416 |
size | 137,314 |
::uninit
A collection of tools for a safer usage of uninitialized memory.
Many crates out there try to replicate C "optimization" patterns such as handling uninitialized memory without taking into account that Rust types carry semantics much more subtle than those of C.
For instance, the following code is Undefined Behavior!
use ::core::mem;
let mut array: [u8; 256] = unsafe { mem::uninitialized() };
for (i, x) in array.iter_mut().enumerate() {
*x = i as u8;
}
Indeed, it creates u8
s with uninitialized memory, which currently
has no defined behavior in the Rust model (see "What The Hardware Does"
is not What Your Program Does: Uninitialized Memory, by Ralf Jung), and then creates Rust
references to these invalid u8
s, which become in turn invalid too.
mem::uninitialized
!In hindsight, offering a mem::uninitialized
function in the core
library of Rust (even if it was marked
unsafe
), has been of the worst mistakes of the language. Indeed, the function
is generic and only bounded by Sized
, and it turns out that, except for
zero-sized types or the later introduced MaybeUninit
<T>
, all the other
calls to this function are unsound (instant UB).
Note that there are other ways to trigger this UB without explicitely using
mem::uninitialized
::<T>()
, such as:
use ::core::mem::MaybeUninit;
type T = u8; // for the example
unsafe {
MaybeUninit::<T>::uninit() // Uninitialized `MaybeUninit<T>` => Fine
.assume_init() // Back to an uninitialized `T` => UB
};
this is exactly equivalent to calling mem::uninitialized::<T>()
,
which breaks the validity invariant of T
and thus causes
"instant UB".
There currently only two exceptions / valid use cases:
either type T = [MaybeUninit<U>; N]
,
or T
is an inhabited ZST (this may, however, break safety
invariants associated with the properties of the type, causing UB
once such broken invariant is witnessed).
yes, using MaybeUninit
is more subtle than just changing a function
call.
let mut vec: Vec<u8> = Vec::with_capacity(100); // Fine
unsafe {
vec.set_len(100); // we have an uninitialized [u8; 100] in the heap
// This has broken the _safety_ invariant of `Vec`, but is not yet UB
// since no code has witnessed the broken state
}
let heap_bytes: &[u8] = &*vec; // Witness the broken safety invariant: UB!
MaybeUninit
So, the solution to manipulating uninitialized memory is to use
MaybeUninit
: the special type MaybeUninit<T>
does not assume that its
backing memory has been initialized / the behavior of an uninitialized
MaybeUninit<T>
is well-defined, no matter the T
.
MaybeUninit
It is all about the delayed initialization pattern:
Creation
A MaybeUninit<T>
is created, with, for instance,
MaybeUninit::<T>::uninit()
:
use ::core::mem::MaybeUninit;
let mut x = MaybeUninit::<i32>::uninit();
(Delayed) Initialization
With great care to avoid accidentally creating (even if only for an
instant) a &T
, &mut T
, or even a T
while the memory has not been
initialized yet (which would be UB), we can write to (and thus initialize) the
uninitialized memory through a &mut MaybeUninit<T>
:
either directly, for instance:
use ::core::mem::MaybeUninit;
let mut x = MaybeUninit::<i32>::uninit();
x = MaybeUninit::new(42);
assert_eq!(42, unsafe { x.assume_init() });
or through a raw *mut T
pointer (contrary to Rust references,
raw pointers do not assume that the memory they point to is
valid). For instance:
use ::core::mem::MaybeUninit;
let mut x = MaybeUninit::<i32>::uninit();
unsafe {
x.as_mut_ptr().write(42);
assert_eq!(x.assume_init(), 42);
}
or, if you use the tools of this crate, by upgrading the
&mut MaybeUninit<T>
into a "&out T
" type called
Out
<T>
:
#![forbid(unsafe_code)] // no unsafe!
use ::core::mem::MaybeUninit;
use ::uninit::prelude::*;
let mut x = MaybeUninit::uninit();
let at_init_x: &i32 = x.as_out().write(42);
assert_eq!(at_init_x, &42);
Type-level upgrade
Once we know, for sure, that the memory has been initialized, we can
upgrade the MaybeUninit<T>
type to the fully-fledged T
type:
By value (MaybeUninit<T> -> T
): .assume_init()
By shared reference (&MaybeUninit<T> -> &T
):
.assume_init_by_ref()
By unique reference (&mut MaybeUninit<T> -> &mut T
):
.assume_init_by_mut()
As you can see, manipulating MaybeUninit
to initialize its contents is
done through restrictive and unergonomic types
(&mut MaybeUninit<T>
/ *mut T
).
So most APIs do not offer a way to output / write into uninitialized memory.
This is what ends up leading many people to do the step .3
before the
step .2
: it is oh so much ergonomic to work with a &mut T
than a
*mut T
, especially when arrays, slices and vectors are involved. Thus
people end up doing UB.
Read
traituse ::std::io;
pub trait Read {
fn read (&mut self, buf: &mut [u8]) -> Result<usize, io::Error>;
// ...
}
that is, there is no way to .read()
into an uninitialized buffer (it would
require an api taking either a (*mut u8, usize)
pair, or, equivalently and
by the way more ergonomically, a [&out [u8]
][crate::prelude::Out
]).
::uninit
So, the objective of this crate is double:
it offers ergonomic ways to create uninitialized buffers.
For instance:
it tries to offer APIs that are equivalent to pervasive APIs,
such as Read
's, but that are able to work with such uninitialized buffers.
For instance:
#![no_std]
friendlySimply disable the default-enabled "std"
feature in your Cargo.toml
file:
[dependencies]
uninit = { version = "x.y.z", default-features = false }