language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-27 05:49:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

roc_std: fixes found by running miri

Browse source
This commit is contained in:
Folkert 2023-10-15 20:09:05 +02:00
parent 23624ac7d9
commit d4e77856fe
No known key found for this signature in database
GPG key ID: 1F17F6FFD112B97C
5 changed files with 46 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

26
crates/roc_std/src/roc_list.rs
View file

@ -829,4 +829,30 @@ mod tests {
drop(a);
drop(b);
}
#[test]
fn readonly_list_is_sendsafe() {
let x = RocList::from_slice(&[1, 2, 3, 4, 5]);
unsafe { x.set_readonly() };
assert_eq!(x.is_readonly(), true);
let y = x.clone();
let z = y.clone();
let safe_x = SendSafeRocList::from(x);
let new_x = RocList::from(safe_x);
assert_eq!(new_x.is_readonly(), true);
assert_eq!(y.is_readonly(), true);
assert_eq!(z.is_readonly(), true);
assert_eq!(new_x.as_slice(), &[1, 2, 3, 4, 5]);
let ptr = new_x.ptr_to_allocation();
drop(y);
drop(z);
drop(new_x);
// free the underlying memory
unsafe { crate::roc_dealloc(ptr, std::mem::align_of::<usize>() as u32) }
}
}

13
crates/roc_std/src/roc_str.rs
View file

@ -325,7 +325,9 @@ impl RocStr {
}
}
RocStrInnerRef::SmallString(small_str) => {
let mut bytes = small_str.bytes;
let mut bytes = [0; size_of::<RocList<u8>>()];
let mut it = small_str.bytes.iter();
bytes = bytes.map(|_| it.next().copied().unwrap_or_default());
// Even if the small string is at capacity, there will be room to write
// a terminator in the byte that's used to store the length.
@ -380,9 +382,7 @@ impl RocStr {
self.with_terminator(terminator, |dest_ptr: *mut u16, str_slice: &str| {
// Translate UTF-8 source bytes into UTF-16 and write them into the destination.
for (index, wchar) in str_slice.encode_utf16().enumerate() {
unsafe {
*(dest_ptr.add(index)) = wchar;
}
unsafe { std::ptr::write_unaligned(dest_ptr.add(index), wchar) };
}
func(dest_ptr, str_slice.len())
@ -467,7 +467,7 @@ impl RocStr {
use core::mem::align_of;
let terminate = |alloc_ptr: *mut E, str_slice: &str| unsafe {
*(alloc_ptr.add(str_slice.len())) = terminator;
std::ptr::write_unaligned(alloc_ptr.add(str_slice.len()), terminator);
func(alloc_ptr, str_slice)
};
@ -548,7 +548,8 @@ impl RocStr {
let available_bytes = size_of::<SmallString>();
if needed_bytes < available_bytes {
terminate(small_str.bytes.as_ptr() as *mut E, self.as_str())
let mut bytes = small_str.bytes;
terminate(&mut bytes as *mut u8 as *mut E, self.as_str())
} else {
fallback(self.as_str())
}