stack_overflow.rs - source

Files

alloc

core

alloc

global.rs layout.rs mod.rs

array

equality.rs iter.rs mod.rs

char

convert.rs decode.rs methods.rs mod.rs

convert

fmt

rt

builders.rs float.rs mod.rs num.rs

future

future.rs into_future.rs mod.rs pending.rs poll_fn.rs ready.rs

hash

iter

macros

mem

manually_drop.rs maybe_uninit.rs mod.rs

num

ops

arith.rs bit.rs control_flow.rs deref.rs drop.rs function.rs generator.rs index.rs mod.rs range.rs try_trait.rs unsize.rs

prelude

ptr

const_ptr.rs metadata.rs mod.rs mut_ptr.rs non_null.rs unique.rs

slice

iter

ascii.rs cmp.rs index.rs iter.rs memchr.rs mod.rs raw.rs rotate.rs sort.rs specialize.rs

stdarch

crates

core_arch

src

aarch64

neon

generated.rs mod.rs

armclang.rs crc.rs mod.rs prefetch.rs tme.rs v8.rs

arm

armclang.rs dsp.rs ex.rs mod.rs neon.rs sat.rs simd32.rs v6.rs v7.rs

arm_shared

barrier

neon

generated.rs mod.rs

registers

aarch32.rs mod.rs

crc.rs crypto.rs hints.rs mod.rs

mips

nvptx

powerpc

powerpc64

wasm32

atomic.rs memory.rs mod.rs simd128.rs

x86

x86_64

abm.rs adx.rs avx.rs avx2.rs avx512f.rs bmi.rs bmi2.rs bswap.rs bt.rs cmpxchg16b.rs fxsr.rs macros.rs mod.rs rdrand.rs sse.rs sse2.rs sse41.rs sse42.rs xsave.rs

macros.rs mod.rs simd.rs simd_llvm.rs

str

converts.rs error.rs iter.rs lossy.rs mod.rs pattern.rs traits.rs validations.rs

stream

stream

sync

atomic.rs mod.rs

task

mod.rs poll.rs ready.rs wake.rs

unicode

mod.rs printable.rs unicode_data.rs

std

backtrace

src

backtrace

dbghelp.rs mod.rs

symbolize

dbghelp.rs mod.rs

dbghelp.rs lib.rs print.rs types.rs windows.rs

collections

hash

map.rs mod.rs set.rs

ffi

c_str.rs mod.rs os_str.rs

io

buffered

bufreader.rs bufwriter.rs linewriter.rs linewritershim.rs mod.rs

copy.rs cursor.rs error.rs impls.rs mod.rs prelude.rs stdio.rs util.rs

net

addr.rs ip.rs mod.rs parser.rs tcp.rs udp.rs

os

linux

fs.rs mod.rs raw.rs

raw

unix

ffi

mod.rs os_str.rs

net

addr.rs ancillary.rs datagram.rs listener.rs mod.rs raw_fd.rs stream.rs

fs.rs io.rs mod.rs process.rs raw.rs thread.rs

wasi

unix

ffi

ffi.rs fs.rs io.rs mod.rs

windows

ffi.rs fs.rs io.rs mod.rs process.rs raw.rs thread.rs

prelude

sync

mpsc

blocking.rs cache_aligned.rs mod.rs mpsc_queue.rs oneshot.rs shared.rs spsc_queue.rs stream.rs sync.rs

barrier.rs condvar.rs mod.rs mutex.rs once.rs poison.rs rwlock.rs

sys

sys_common

thread

local.rs mod.rs

alloc.rs ascii.rs backtrace.rs env.rs error.rs f32.rs f64.rs fs.rs keyword_docs.rs lazy.rs lib.rs macros.rs num.rs panic.rs panicking.rs path.rs primitive_docs.rs process.rs rt.rs time.rs

#![cfg_attr(test, allow(dead_code))]

use crate::sys::c;
use crate::thread;

pub struct Handler;

impl Handler {
    pub unsafe fn new() -> Handler {
        // 该 API 在 XP 上不可用，因此不要在这种情况下使用 panic，只是祈祷它可以正常工作。
        //
        if c::SetThreadStackGuarantee(&mut 0x5000) == 0 {
            if c::GetLastError() as u32 != c::ERROR_CALL_NOT_IMPLEMENTED as u32 {
                panic!("failed to reserve stack space for exception handling");
            }
        }
        Handler
    }
}

extern "system" fn vectored_handler(ExceptionInfo: *mut c::EXCEPTION_POINTERS) -> c::LONG {
    unsafe {
        let rec = &(*(*ExceptionInfo).ExceptionRecord);
        let code = rec.ExceptionCode;

        if code == c::EXCEPTION_STACK_OVERFLOW {
            rtprintpanic!(
                "\nthread '{}' has overflowed its stack\n",
                thread::current().name().unwrap_or("<unknown>")
            );
        }
        c::EXCEPTION_CONTINUE_SEARCH
    }
}

pub unsafe fn init() {
    if c::AddVectoredExceptionHandler(0, vectored_handler).is_null() {
        panic!("failed to install exception handler");
    }
    // 设置主线程的线程栈保证。
    let _h = Handler::new();
}