bjoernager/rust
bjoernager/rust
1
Fork 0
Code Activity

Add SAFETY comments to the thread local implementation

Browse source 
Reduce `unsafe` block scope and add `SAFETY` comments.
This commit is contained in:
Trevor Gross 2024-04-08 17:47:09 -04:00
parent 2aec2fe3b8
commit 6e68a2f475
1 changed files with 12 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

16
library/std/src/thread/local.rs
View file

@ -281,11 +281,10 @@ impl<T: 'static> LocalKey<T> {
where
F: FnOnce(&T) -> R,
{
unsafe {
let thread_local = (self.inner)(None).ok_or(AccessError)?;
// SAFETY: `inner` is safe to call within the lifetime of the thread
let thread_local = unsafe { (self.inner)(None).ok_or(AccessError)? };
Ok(f(thread_local))
}
}
/// Acquires a reference to the value in this TLS key, initializing it with
/// `init` if it wasn't already initialized on this thread.
@ -303,15 +302,18 @@ impl<T: 'static> LocalKey<T> {
where
F: FnOnce(Option<T>, &T) -> R,
{
unsafe {
let mut init = Some(init);
let reference = (self.inner)(Some(&mut init)).expect(
// SAFETY: `inner` is safe to call within the lifetime of the thread
let reference = unsafe {
(self.inner)(Some(&mut init)).expect(
"cannot access a Thread Local Storage value \
during or after destruction",
);
)
};
f(init, reference)
}
}
}
impl<T: 'static> LocalKey<Cell<T>> {