memsafe 0.2.0

A cross-platform Rust library for securely wrapping data in memory
Documentation 
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

use winapi::{
    ctypes::c_void,
    um::memoryapi::{VirtualAlloc, VirtualFree, VirtualLock, VirtualProtect, VirtualUnlock},
};

use crate::MemoryError;

/// Wrapper over `VirtualAlloc`. Full documentation here:
/// https://learn.microsoft.com/en-us/windows/win32/api/memoryapi/nf-memoryapi-virtualalloc
pub fn virtual_alloc<T>(
    ptr: *mut c_void,
    len: usize,
    allocation_type: u32,
    protect: u32,
) -> Result<*mut T, MemoryError> {
    let ptr = unsafe { VirtualAlloc(ptr, len, allocation_type, protect) };
    if ptr.is_null() {
        Err(MemoryError(std::io::Error::last_os_error()))
    } else {
        Ok(ptr as *mut T)
    }
}

/// Wrapper over `VirtualFree`. Full documentation here:
/// https://learn.microsoft.com/en-us/windows/win32/api/memoryapi/nf-memoryapi-virtualfree
pub fn virtual_free<T>(ptr: *mut T, len: usize, free_type: u32) -> Result<(), MemoryError> {
    println!("fre");
    if unsafe { VirtualFree(ptr as *mut _, len, free_type) } == 0 {
        Err(MemoryError(std::io::Error::last_os_error()))
    } else {
        Ok(())
    }
}

/// Wrapper over `VirtualProtect`. Full documentation here:
/// https://learn.microsoft.com/en-us/windows/win32/api/memoryapi/nf-memoryapi-virtualprotect
pub fn virtual_protect<T>(
    ptr: *mut T,
    len: usize,
    new_protect: u32,
    old_protect: &mut u32,
) -> Result<(), MemoryError> {
    if unsafe { VirtualProtect(ptr as *mut c_void, len, new_protect, old_protect) } == 0 {
        Err(MemoryError(std::io::Error::last_os_error()))
    } else {
        Ok(())
    }
}

/// Wrapper over `VirtualLock`. Full documentation here:
/// https://learn.microsoft.com/en-us/windows/win32/api/memoryapi/nf-memoryapi-virtuallock
pub fn virtual_lock<T>(ptr: *mut T, len: usize) -> Result<(), MemoryError> {
    if unsafe { VirtualLock(ptr as *mut _, len) } == 0 {
        Err(MemoryError(std::io::Error::last_os_error()))
    } else {
        Ok(())
    }
}

/// Wrapper over `VirtualUnlock`. Full documentation here:
/// https://learn.microsoft.com/en-us/windows/win32/api/memoryapi/nf-memoryapi-virtualunlock
pub fn virtual_unlock<T>(ptr: *mut T, len: usize) -> Result<(), MemoryError> {
    if unsafe { VirtualUnlock(ptr as *mut _, len) } == 0 {
        Err(MemoryError(std::io::Error::last_os_error()))
    } else {
        Ok(())
    }
}