lethe 0.2.2

A secure, free, cross-platform and open-source drive wiping utility
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

pub fn alloc_aligned_byte_vec(size: usize, align: usize) -> Vec<u8> {
    unsafe {
        let buf_layout = std::alloc::Layout::from_size_align_unchecked(size, align);
        let buf_ptr = std::alloc::alloc(buf_layout);
        Vec::from_raw_parts(buf_ptr, size, size)
    }
}

pub fn fill_byte_slice(buf: &mut [u8], value: u8) {
    // unsafe {
    //     std::libc::memset(
    //         buf.as_mut_ptr() as _,
    //         0,
    //         buf.len()
    //     );
    // };
    buf.iter_mut().map(|x| *x = value).count(); //todo: rewrite
}

#[cfg(test)]
mod test {
    use super::*;

    #[test]
    fn test_aligned_allocation() {
        let size = 65536;
        let align = 4096;

        let buf = alloc_aligned_byte_vec(size, align);

        assert_eq!(buf.len(), size);
        assert_eq!(buf.as_ptr() as usize % align, 0);
    }

    #[test]
    fn test_vec_fill() {
        let mut buf = vec![0x00; 1024];

        fill_byte_slice(&mut buf, 0xff);

        assert_eq!(buf.iter().filter(|x| **x != 0xff).count(), 0);
    }

}