From 0153e64d975ec175f893ad528311ef7c441a1561 Mon Sep 17 00:00:00 2001 From: Dave Huseby Date: Tue, 26 Jan 2016 17:37:18 -0800 Subject: Fixes #31229 --- src/libstd/sys/unix/stack_overflow.rs | 43 +++++++++++++++++++++++++++-------- 1 file changed, 34 insertions(+), 9 deletions(-) (limited to 'src/libstd/sys/unix/stack_overflow.rs') diff --git a/src/libstd/sys/unix/stack_overflow.rs b/src/libstd/sys/unix/stack_overflow.rs index c7614db3299..88ac73de2fb 100644 --- a/src/libstd/sys/unix/stack_overflow.rs +++ b/src/libstd/sys/unix/stack_overflow.rs @@ -135,17 +135,42 @@ mod imp { Handler { _data: MAIN_ALTSTACK }; } - pub unsafe fn make_handler() -> Handler { - let alt_stack = mmap(ptr::null_mut(), - SIGSTKSZ, - PROT_READ | PROT_WRITE, - MAP_PRIVATE | MAP_ANON, - -1, - 0); - if alt_stack == MAP_FAILED { + #[cfg(any(target_os = "linux", + target_os = "macos", + target_os = "bitrig", + target_os = "netbsd", + target_os = "openbsd"))] + unsafe fn get_stack() -> *mut libc::c_void { + let stack = mmap(ptr::null_mut(), + SIGSTKSZ, + PROT_READ | PROT_WRITE, + MAP_PRIVATE | MAP_ANON, + -1, + 0); + if stack == MAP_FAILED { + panic!("failed to allocate an alternative stack"); + } + stack + } + + #[cfg(any(target_os = "dragonfly", + target_os = "freebsd"))] + unsafe fn get_stack() -> *mut i8 { + let stack = mmap(ptr::null_mut(), + SIGSTKSZ, + PROT_READ | PROT_WRITE, + MAP_PRIVATE | MAP_ANON, + -1, + 0); + if stack == MAP_FAILED { panic!("failed to allocate an alternative stack"); } + stack as *mut i8 + } + + pub unsafe fn make_handler() -> Handler { + let alt_stack = get_stack(); let mut stack: libc::stack_t = mem::zeroed(); stack.ss_sp = alt_stack; @@ -154,7 +179,7 @@ mod imp { sigaltstack(&stack, ptr::null_mut()); - Handler { _data: alt_stack } + Handler { _data: alt_stack as *mut libc::c_void } } pub unsafe fn drop_handler(handler: &mut Handler) { -- cgit 1.4.1-3-g733a5 From 7803c8d688b87e3d13f42d543a553201005db783 Mon Sep 17 00:00:00 2001 From: Dave Huseby Date: Wed, 27 Jan 2016 16:46:45 -0800 Subject: refactoring get_stack to be cleaner --- src/libstd/sys/unix/stack_overflow.rs | 47 +++++++++++++++-------------------- 1 file changed, 20 insertions(+), 27 deletions(-) (limited to 'src/libstd/sys/unix/stack_overflow.rs') diff --git a/src/libstd/sys/unix/stack_overflow.rs b/src/libstd/sys/unix/stack_overflow.rs index 88ac73de2fb..f5353895c9d 100644 --- a/src/libstd/sys/unix/stack_overflow.rs +++ b/src/libstd/sys/unix/stack_overflow.rs @@ -140,46 +140,39 @@ mod imp { target_os = "bitrig", target_os = "netbsd", target_os = "openbsd"))] - unsafe fn get_stack() -> *mut libc::c_void { - let stack = mmap(ptr::null_mut(), - SIGSTKSZ, - PROT_READ | PROT_WRITE, - MAP_PRIVATE | MAP_ANON, - -1, - 0); - if stack == MAP_FAILED { + unsafe fn get_stack() -> libc::stack_t { + let stackp = mmap(ptr::null_mut(), + SIGSTKSZ, + PROT_READ | PROT_WRITE, + MAP_PRIVATE | MAP_ANON, + -1, + 0); + if stackp == MAP_FAILED { panic!("failed to allocate an alternative stack"); } - stack + libc::stack_t { ss_sp: stackp, ss_flags: 0, ss_size: SIGSTKSZ } } #[cfg(any(target_os = "dragonfly", target_os = "freebsd"))] - unsafe fn get_stack() -> *mut i8 { - let stack = mmap(ptr::null_mut(), - SIGSTKSZ, - PROT_READ | PROT_WRITE, - MAP_PRIVATE | MAP_ANON, - -1, - 0); - if stack == MAP_FAILED { + unsafe fn get_stack() -> libc::stack_t { + let stackp = mmap(ptr::null_mut(), + SIGSTKSZ, + PROT_READ | PROT_WRITE, + MAP_PRIVATE | MAP_ANON, + -1, + 0); + if stackp == MAP_FAILED { panic!("failed to allocate an alternative stack"); } - stack as *mut i8 + libc::stack_t { ss_sp: stackp as *mut i8, ss_flags: 0, ss_size: SIGSTKSZ } } pub unsafe fn make_handler() -> Handler { - let alt_stack = get_stack(); - let mut stack: libc::stack_t = mem::zeroed(); - - stack.ss_sp = alt_stack; - stack.ss_flags = 0; - stack.ss_size = SIGSTKSZ; - + let stack = get_stack(); sigaltstack(&stack, ptr::null_mut()); - - Handler { _data: alt_stack as *mut libc::c_void } + Handler { _data: stack.ss_sp as *mut libc::c_void } } pub unsafe fn drop_handler(handler: &mut Handler) { -- cgit 1.4.1-3-g733a5 From 68bfd43eefe42b429a463731fe81d62bf9c27907 Mon Sep 17 00:00:00 2001 From: Dave Huseby Date: Tue, 2 Feb 2016 09:23:33 -0800 Subject: simplifying get_stack --- src/libstd/sys/unix/stack_overflow.rs | 27 +++------------------------ 1 file changed, 3 insertions(+), 24 deletions(-) (limited to 'src/libstd/sys/unix/stack_overflow.rs') diff --git a/src/libstd/sys/unix/stack_overflow.rs b/src/libstd/sys/unix/stack_overflow.rs index f5353895c9d..ee590e1af71 100644 --- a/src/libstd/sys/unix/stack_overflow.rs +++ b/src/libstd/sys/unix/stack_overflow.rs @@ -135,27 +135,7 @@ mod imp { Handler { _data: MAIN_ALTSTACK }; } - #[cfg(any(target_os = "linux", - target_os = "macos", - target_os = "bitrig", - target_os = "netbsd", - target_os = "openbsd"))] - unsafe fn get_stack() -> libc::stack_t { - let stackp = mmap(ptr::null_mut(), - SIGSTKSZ, - PROT_READ | PROT_WRITE, - MAP_PRIVATE | MAP_ANON, - -1, - 0); - if stackp == MAP_FAILED { - panic!("failed to allocate an alternative stack"); - } - libc::stack_t { ss_sp: stackp, ss_flags: 0, ss_size: SIGSTKSZ } - } - - #[cfg(any(target_os = "dragonfly", - target_os = "freebsd"))] - unsafe fn get_stack() -> libc::stack_t { + unsafe fn get_stack() -> *mut i8 { let stackp = mmap(ptr::null_mut(), SIGSTKSZ, PROT_READ | PROT_WRITE, @@ -165,12 +145,11 @@ mod imp { if stackp == MAP_FAILED { panic!("failed to allocate an alternative stack"); } - libc::stack_t { ss_sp: stackp as *mut i8, ss_flags: 0, ss_size: SIGSTKSZ } + stackp as *mut i8 } - pub unsafe fn make_handler() -> Handler { - let stack = get_stack(); + let stack = libc::stack_t { ss_sp: get_stack(), ss_flags: 0, ss_size: SIGSTKSZ } sigaltstack(&stack, ptr::null_mut()); Handler { _data: stack.ss_sp as *mut libc::c_void } } -- cgit 1.4.1-3-g733a5 From ca6f9203461f57a1f1a88266efeb61a5aa47d66b Mon Sep 17 00:00:00 2001 From: Dave Huseby Date: Tue, 2 Feb 2016 18:21:39 -0800 Subject: trying again at fixing stackp initialization --- src/libstd/sys/unix/stack_overflow.rs | 21 ++++++++++++++++++--- 1 file changed, 18 insertions(+), 3 deletions(-) (limited to 'src/libstd/sys/unix/stack_overflow.rs') diff --git a/src/libstd/sys/unix/stack_overflow.rs b/src/libstd/sys/unix/stack_overflow.rs index ee590e1af71..b5cbcaa44d5 100644 --- a/src/libstd/sys/unix/stack_overflow.rs +++ b/src/libstd/sys/unix/stack_overflow.rs @@ -135,7 +135,7 @@ mod imp { Handler { _data: MAIN_ALTSTACK }; } - unsafe fn get_stack() -> *mut i8 { + unsafe fn get_stackp() -> *mut libc::c_void { let stackp = mmap(ptr::null_mut(), SIGSTKSZ, PROT_READ | PROT_WRITE, @@ -145,11 +145,26 @@ mod imp { if stackp == MAP_FAILED { panic!("failed to allocate an alternative stack"); } - stackp as *mut i8 + stackp + } + + #[cfg(any(target_os = "linux", + target_os = "macos", + target_os = "bitrig", + target_os = "netbsd", + target_os = "openbsd"))] + unsafe fn get_stack() -> libc::stack_t { + libc::stack_t { ss_sp: get_stackp(), ss_flags: 0, ss_size: SIGSTKSZ } + } + + #[cfg(any(target_os = "freebsd", + target_os = "dragonfly"))] + unsafe fn get_stack() -> libc::stack_t { + libc::stack_t { ss_sp: get_stackp() as *mut i8, ss_flags: 0, ss_size: SIGSTKSZ } } pub unsafe fn make_handler() -> Handler { - let stack = libc::stack_t { ss_sp: get_stack(), ss_flags: 0, ss_size: SIGSTKSZ } + let stack = get_stack(); sigaltstack(&stack, ptr::null_mut()); Handler { _data: stack.ss_sp as *mut libc::c_void } } -- cgit 1.4.1-3-g733a5