2016-01-23 07:49:57 +00:00
|
|
|
#![cfg_attr(test, allow(dead_code))]
|
|
|
|
|
2019-02-10 19:23:21 +00:00
|
|
|
use crate::sys::c;
|
2021-04-29 14:05:10 +00:00
|
|
|
use crate::thread;
|
2014-11-24 03:21:17 +00:00
|
|
|
|
2015-07-27 20:41:35 +00:00
|
|
|
pub struct Handler;
|
2014-11-24 03:21:17 +00:00
|
|
|
|
|
|
|
impl Handler {
|
|
|
|
pub unsafe fn new() -> Handler {
|
2015-07-27 20:41:35 +00:00
|
|
|
// This API isn't available on XP, so don't panic in that case and just
|
|
|
|
// pray it works out ok.
|
|
|
|
if c::SetThreadStackGuarantee(&mut 0x5000) == 0 {
|
2015-11-03 00:23:22 +00:00
|
|
|
if c::GetLastError() as u32 != c::ERROR_CALL_NOT_IMPLEMENTED as u32 {
|
2015-07-27 20:41:35 +00:00
|
|
|
panic!("failed to reserve stack space for exception handling");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
Handler
|
2014-11-24 03:21:17 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-11-27 18:28:39 +00:00
|
|
|
extern "system" fn vectored_handler(ExceptionInfo: *mut c::EXCEPTION_POINTERS) -> c::LONG {
|
2014-11-24 03:21:17 +00:00
|
|
|
unsafe {
|
|
|
|
let rec = &(*(*ExceptionInfo).ExceptionRecord);
|
|
|
|
let code = rec.ExceptionCode;
|
|
|
|
|
2015-07-27 20:41:35 +00:00
|
|
|
if code == c::EXCEPTION_STACK_OVERFLOW {
|
2021-04-29 14:05:10 +00:00
|
|
|
rterr!(
|
|
|
|
"\nthread '{}' has overflowed its stack\n",
|
|
|
|
thread::current().name().unwrap_or("<unknown>")
|
|
|
|
);
|
2014-11-24 03:21:17 +00:00
|
|
|
}
|
2015-07-27 20:41:35 +00:00
|
|
|
c::EXCEPTION_CONTINUE_SEARCH
|
2014-11-24 03:21:17 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub unsafe fn init() {
|
2015-07-27 20:41:35 +00:00
|
|
|
if c::AddVectoredExceptionHandler(0, vectored_handler).is_null() {
|
2014-11-24 03:21:17 +00:00
|
|
|
panic!("failed to install exception handler");
|
|
|
|
}
|
2015-07-27 20:41:35 +00:00
|
|
|
// Set the thread stack guarantee for the main thread.
|
|
|
|
let _h = Handler::new();
|
2014-11-24 03:21:17 +00:00
|
|
|
}
|