diff options
Diffstat (limited to 'src/rt/rust_task_thread.cpp')
| -rw-r--r-- | src/rt/rust_task_thread.cpp | 30 |
1 files changed, 29 insertions, 1 deletions
diff --git a/src/rt/rust_task_thread.cpp b/src/rt/rust_task_thread.cpp index 41242fee9d3..4129969057c 100644 --- a/src/rt/rust_task_thread.cpp +++ b/src/rt/rust_task_thread.cpp @@ -13,6 +13,8 @@ pthread_key_t rust_task_thread::task_key; DWORD rust_task_thread::task_key; #endif +const size_t C_STACK_SIZE = (1024*1024); + bool rust_task_thread::tls_initialized = false; rust_task_thread::rust_task_thread(rust_scheduler *sched, @@ -34,7 +36,8 @@ rust_task_thread::rust_task_thread(rust_scheduler *sched, id(id), min_stack_size(kernel->env->min_stack_size), env(kernel->env), - should_exit(false) + should_exit(false), + cached_c_stack(NULL) { LOGPTR(this, "new dom", (uintptr_t)this); isaac_init(kernel, &rctx); @@ -58,6 +61,10 @@ rust_task_thread::~rust_task_thread() { #ifndef __WIN32__ pthread_attr_destroy(&attr); #endif + + if (cached_c_stack) { + destroy_stack(kernel, cached_c_stack); + } } void @@ -367,6 +374,27 @@ rust_task_thread::exit() { lock.signal(); } +stk_seg * +rust_task_thread::borrow_c_stack() { + + if (cached_c_stack) { + stk_seg *your_stack = cached_c_stack; + cached_c_stack = NULL; + return your_stack; + } else { + return create_stack(kernel, C_STACK_SIZE); + } +} + +void +rust_task_thread::return_c_stack(stk_seg *stack) { + if (cached_c_stack) { + destroy_stack(kernel, stack); + } else { + cached_c_stack = stack; + } +} + // // Local Variables: // mode: C++ |