forked from M-Labs/libfringe
parent
6284a22caa
commit
0d7b3d7091
10
Cargo.toml
10
Cargo.toml
|
@ -3,6 +3,10 @@ authors = ["edef <edef@edef.eu>"]
|
|||
name = "fringe"
|
||||
version = "0.0.1"
|
||||
|
||||
[dependencies.kernel32-sys]
|
||||
optional = true
|
||||
version = "0.2.2"
|
||||
|
||||
[dependencies.libc]
|
||||
optional = true
|
||||
version = "0.1.6"
|
||||
|
@ -16,6 +20,10 @@ rev = "9ef793e9549aabfd2d969615180b69d29ce28d88"
|
|||
default-features = false
|
||||
version = "1"
|
||||
|
||||
[dependencies.winapi]
|
||||
optional = true
|
||||
version = "0.2.6"
|
||||
|
||||
[features]
|
||||
default = ["os", "valgrind"]
|
||||
os = ["libc"]
|
||||
os = ["libc", "kernel32-sys", "winapi"]
|
||||
|
|
|
@ -10,6 +10,10 @@ use self::imp::sys_page_size;
|
|||
#[path = "unix.rs"]
|
||||
mod imp;
|
||||
|
||||
#[cfg(windows)]
|
||||
#[path = "windows.rs"]
|
||||
mod imp;
|
||||
|
||||
static PAGE_SIZE_CACHE: AtomicUsize = ATOMIC_USIZE_INIT;
|
||||
pub fn page_size() -> usize {
|
||||
match PAGE_SIZE_CACHE.load(Ordering::Relaxed) {
|
||||
|
|
|
@ -0,0 +1,37 @@
|
|||
// This file is part of libfringe, a low-level green threading library.
|
||||
// Copyright (c) edef <edef@edef.eu>
|
||||
// See the LICENSE file included in this distribution.
|
||||
extern crate winapi;
|
||||
extern crate kernel32;
|
||||
use core::{mem, ptr};
|
||||
use self::winapi::basetsd::SIZE_T;
|
||||
use self::winapi::minwindef::{DWORD, LPVOID};
|
||||
use self::winapi::winnt::{MEM_COMMIT, MEM_RESERVE, MEM_RELEASE, PAGE_READWRITE, PAGE_NOACCESS};
|
||||
use super::page_size;
|
||||
|
||||
#[cfg(windows)]
|
||||
pub fn sys_page_size() -> usize {
|
||||
unsafe {
|
||||
let mut info = mem::zeroed();
|
||||
kernel32::GetSystemInfo(&mut info);
|
||||
info.dwPageSize as usize
|
||||
}
|
||||
}
|
||||
|
||||
pub unsafe fn map_stack(len: usize) -> Option<*mut u8> {
|
||||
let ptr = kernel32::VirtualAlloc(ptr::null_mut(), len as SIZE_T, MEM_COMMIT | MEM_RESERVE, PAGE_READWRITE);
|
||||
if ptr == ptr::null_mut() {
|
||||
None
|
||||
} else {
|
||||
Some(ptr as *mut u8)
|
||||
}
|
||||
}
|
||||
|
||||
pub unsafe fn protect_stack(ptr: *mut u8) -> bool {
|
||||
let mut old_prot: DWORD = 0;
|
||||
kernel32::VirtualProtect(ptr as LPVOID, page_size() as SIZE_T, PAGE_NOACCESS, &mut old_prot) != 0
|
||||
}
|
||||
|
||||
pub unsafe fn unmap_stack(ptr: *mut u8, _len: usize) -> bool {
|
||||
kernel32::VirtualFree(ptr as LPVOID, 0, MEM_RELEASE) != 0
|
||||
}
|
Loading…
Reference in New Issue