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();