Replace disable/enable functions with lock/unlock

This commit is contained in:
Yota Toyama
2019-06-01 21:29:19 +00:00
parent b1a73dc728
commit 6cd1327c5c
3 changed files with 12 additions and 12 deletions

View File

@@ -10,12 +10,12 @@ static GLOBAL_ALLOCATOR: Allocator = Allocator;
fn main() {
unsafe { Allocator::initialize() }
Allocator::disable_gc();
Allocator::lock();
let handle = Coroutine::spawn(move |_, _| {
let bottom: u8 = 0;
unsafe { Allocator::set_stack_bottom(&bottom) }
Allocator::enable_gc();
Allocator::unlock();
loop {
unsafe { std::alloc::alloc(Layout::from_size_align(2 ^ 8, 8).unwrap()) };

View File

@@ -11,22 +11,22 @@ static GLOBAL_ALLOCATOR: Allocator = Allocator;
fn main() {
unsafe { Allocator::initialize() }
Allocator::disable_gc();
Allocator::lock();
let mut handle = Coroutine::spawn(move |me, _| {
let bottom: u8 = 0;
unsafe { Allocator::set_stack_bottom(&bottom) }
Allocator::enable_gc();
Allocator::unlock();
me.yield_with(42);
unsafe { Allocator::set_stack_bottom(&bottom) }
Allocator::enable_gc();
Allocator::unlock();
loop {
unsafe { std::alloc::alloc(Layout::from_size_align(2 ^ 8, 8).unwrap()) };
}
});
Allocator::disable_gc();
Allocator::lock();
assert_eq!(handle.resume(0).unwrap(), 42);
handle.resume(0).unwrap();

View File

@@ -20,8 +20,8 @@ extern "C" {
callback: unsafe extern "C" fn(*const c_void) -> *const c_void,
client_data: *const c_void,
) -> *const c_void;
fn GC_disable() -> c_void;
fn GC_enable() -> c_void;
fn GC_alloc_lock() -> c_void;
fn GC_alloc_unlock() -> c_void;
fn GC_free(ptr: *mut c_void);
fn GC_get_stack_base(stack_base: *mut GcStackBase) -> c_int;
fn GC_init() -> c_void;
@@ -34,12 +34,12 @@ extern "C" {
pub struct Allocator;
impl Allocator {
pub fn disable_gc() {
unsafe { GC_disable() };
pub fn lock() {
unsafe { GC_alloc_lock() };
}
pub fn enable_gc() {
unsafe { GC_enable() };
pub fn unlock() {
unsafe { GC_alloc_unlock() };
}
pub unsafe fn initialize() {