mirror of
https://github.com/zerotier/ZeroTierOne.git
synced 2025-10-12 16:25:07 +02:00
26 lines
822 B
Rust
26 lines
822 B
Rust
/// Can create buffers of arbitrary lifetime.
|
|
/// Frees created buffers when dropped.
|
|
///
|
|
/// This struct is of course unsafe and the fact that
|
|
/// it must outlive the created slices has to be ensured by
|
|
/// the programmer.
|
|
///
|
|
/// Used at certain test scenarios as a safer version of
|
|
/// Vec::leak, to satisfy the address sanitizer.
|
|
pub struct LeakedBuffers {
|
|
leaked_vecs: Vec<Box<[u8]>>,
|
|
}
|
|
|
|
impl LeakedBuffers {
|
|
pub fn new() -> Self {
|
|
Self {
|
|
leaked_vecs: vec![],
|
|
}
|
|
}
|
|
pub unsafe fn create<'a>(&mut self, size: usize) -> &'a mut [u8] {
|
|
let new_mem = vec![0u8; size].into_boxed_slice();
|
|
self.leaked_vecs.push(new_mem);
|
|
let new_mem = self.leaked_vecs.last_mut().unwrap();
|
|
std::slice::from_raw_parts_mut(new_mem.as_mut_ptr(), new_mem.len())
|
|
}
|
|
}
|