From 0153e64d975ec175f893ad528311ef7c441a1561 Mon Sep 17 00:00:00 2001 From: Dave Huseby Date: Tue, 26 Jan 2016 17:37:18 -0800 Subject: [PATCH] Fixes #31229 --- src/libstd/sys/unix/fs.rs | 2 +- src/libstd/sys/unix/stack_overflow.rs | 43 +++++++++++++++++++++------ src/libtest/lib.rs | 10 ++----- 3 files changed, 37 insertions(+), 18 deletions(-) diff --git a/src/libstd/sys/unix/fs.rs b/src/libstd/sys/unix/fs.rs index d2d2ce35d84..9da986d7fbf 100644 --- a/src/libstd/sys/unix/fs.rs +++ b/src/libstd/sys/unix/fs.rs @@ -224,7 +224,7 @@ fn name_bytes(&self) -> &[u8] { fn name_bytes(&self) -> &[u8] { unsafe { ::slice::from_raw_parts(self.entry.d_name.as_ptr() as *const u8, - self.entry.d_namelen as usize) + self.entry.d_namlen as usize) } } #[cfg(any(target_os = "android", 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 @@ pub unsafe fn cleanup() { 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 @@ pub unsafe fn make_handler() -> Handler { 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) { diff --git a/src/libtest/lib.rs b/src/libtest/lib.rs index 130ce3a9637..443356fbe7c 100644 --- a/src/libtest/lib.rs +++ b/src/libtest/lib.rs @@ -939,18 +939,12 @@ fn num_cpus() -> usize { fn num_cpus() -> usize { let mut cpus: libc::c_uint = 0; let mut cpus_size = std::mem::size_of_val(&cpus); - let mut mib = [libc::CTL_HW, libc::HW_AVAILCPU, 0, 0]; unsafe { - libc::sysctl(mib.as_mut_ptr(), - 2, - &mut cpus as *mut _ as *mut _, - &mut cpus_size as *mut _ as *mut _, - 0 as *mut _, - 0); + cpus = libc::sysconf(libc::_SC_NPROCESSORS_ONLN) as libc::c_uint; } if cpus < 1 { - mib[1] = libc::HW_NCPU; + let mut mib = [libc::CTL_HW, libc::HW_NCPU, 0, 0]; unsafe { libc::sysctl(mib.as_mut_ptr(), 2, -- GitLab