//!
//! In a nutshell, what happens here is:
//!
-//! 1. The `panic` function calls the standard Windows function `RaiseException`
-//! with a Rust-specific code, triggering the unwinding process.
+//! 1. The `panic` function calls the standard Windows function
+//! `_CxxThrowException` to throw a C++-like exception, triggering the
+//! unwinding process.
//! 2. All landing pads generated by the compiler use the personality function
-//! `__C_specific_handler` on 64-bit and `__except_handler3` on 32-bit,
-//! functions in the CRT, and the unwinding code in Windows will use this
-//! personality function to execute all cleanup code on the stack.
+//! `__CxxFrameHandler3`, a function in the CRT, and the unwinding code in
+//! Windows will use this personality function to execute all cleanup code on
+//! the stack.
//! 3. All compiler-generated calls to `invoke` have a landing pad set as a
//! `cleanuppad` LLVM instruction, which indicates the start of the cleanup
//! routine. The personality (in step 2, defined in the CRT) is responsible
//! for running the cleanup routines.
//! 4. Eventually the "catch" code in the `try` intrinsic (generated by the
-//! compiler) is executed, which will ensure that the exception being caught
-//! is indeed a Rust exception, indicating that control should come back to
+//! compiler) is executed and indicates that control should come back to
//! Rust. This is done via a `catchswitch` plus a `catchpad` instruction in
//! LLVM IR terms, finally returning normal control to the program with a
-//! `catchret` instruction. The `try` intrinsic uses a filter function to
-//! detect what kind of exception is being thrown, and this detection is
-//! implemented as the msvc_try_filter language item below.
+//! `catchret` instruction.
//!
//! Some specific differences from the gcc-based exception handling are:
//!
//! * Rust has no custom personality function, it is instead *always*
-//! __C_specific_handler or __except_handler3, so the filtering is done in a
-//! C++-like manner instead of in the personality function itself. Note that
-//! the precise codegen for this was lifted from an LLVM test case for SEH
-//! (this is the `__rust_try_filter` function below).
+//! `__CxxFrameHandler3`. Additionally, no extra filtering is performed, so we
+//! end up catching any C++ exceptions that happen to look like the kind we're
+//! throwing. Note that throwing an exception into Rust is undefined behavior
+//! anyway, so this should be fine.
//! * We've got some data to transmit across the unwinding boundary,
//! specifically a `Box<Any + Send>`. Like with Dwarf exceptions
//! these two pointers are stored as a payload in the exception itself. On
-//! MSVC, however, there's no need for an extra allocation because the call
-//! stack is preserved while filter functions are being executed. This means
-//! that the pointers are passed directly to `RaiseException` which are then
-//! recovered in the filter function to be written to the stack frame of the
-//! `try` intrinsic.
+//! MSVC, however, there's no need for an extra heap allocation because the
+//! call stack is preserved while filter functions are being executed. This
+//! means that the pointers are passed directly to `_CxxThrowException` which
+//! are then recovered in the filter function to be written to the stack frame
+//! of the `try` intrinsic.
//!
//! [win64]: http://msdn.microsoft.com/en-us/library/1eyas8tf.aspx
//! [llvm]: http://llvm.org/docs/ExceptionHandling.html#background-on-windows-exceptions
+#![allow(bad_style)]
+#![allow(private_no_mangle_fns)]
+
use alloc::boxed::Box;
use core::any::Any;
-use core::intrinsics;
use core::mem;
use core::raw;
use windows as c;
+use libc::{c_int, c_uint};
+
+// First up, a whole bunch of type definitions. There's a few platform-specific
+// oddities here, and a lot that's just blatantly copied from LLVM. The purpose
+// of all this is to implement the `panic` function below through a call to
+// `_CxxThrowException`.
+//
+// This function takes two arguments. The first is a pointer to the data we're
+// passing in, which in this case is our trait object. Pretty easy to find! The
+// next, however, is more complicated. This is a pointer to a `_ThrowInfo`
+// structure, and it generally is just intended to just describe the exception
+// being thrown.
+//
+// Currently the definition of this type [1] is a little hairy, and the main
+// oddity (and difference from the online article) is that on 32-bit the
+// pointers are pointers but on 64-bit the pointers are expressed as 32-bit
+// offsets from the `__ImageBase` symbol. The `ptr_t` and `ptr!` macro in the
+// modules below are used to express this.
+//
+// The maze of type definitions also closely follows what LLVM emits for this
+// sort of operation. For example, if you compile this C++ code on MSVC and emit
+// the LLVM IR:
+//
+// #include <stdin.h>
+//
+// void foo() {
+// uint64_t a[2] = {0, 1};
+// throw a;
+// }
+//
+// That's essentially what we're trying to emulate. Most of the constant values
+// below were just copied from LLVM, I'm at least not 100% sure what's going on
+// everywhere. For example the `.PA_K\0` and `.PEA_K\0` strings below (stuck in
+// the names of a few of these) I'm not actually sure what they do, but it seems
+// to mirror what LLVM does!
+//
+// In any case, these structures are all constructed in a similar manner, and
+// it's just somewhat verbose for us.
+//
+// [1]: http://www.geoffchappell.com/studies/msvc/language/predefined/
+
+#[cfg(target_arch = "x86")]
+#[macro_use]
+mod imp {
+ pub type ptr_t = *mut u8;
+ pub const OFFSET: i32 = 4;
+
+ pub const NAME1: [u8; 7] = [b'.', b'P', b'A', b'_', b'K', 0, 0];
+ pub const NAME2: [u8; 7] = [b'.', b'P', b'A', b'X', 0, 0, 0];
+
+ macro_rules! ptr {
+ (0) => (0 as *mut u8);
+ ($e:expr) => ($e as *mut u8);
+ }
+}
+
+#[cfg(target_arch = "x86_64")]
+#[macro_use]
+mod imp {
+ pub type ptr_t = u32;
+ pub const OFFSET: i32 = 8;
+
+ pub const NAME1: [u8; 7] = [b'.', b'P', b'E', b'A', b'_', b'K', 0];
+ pub const NAME2: [u8; 7] = [b'.', b'P', b'E', b'A', b'X', 0, 0];
+
+ extern {
+ pub static __ImageBase: u8;
+ }
+
+ macro_rules! ptr {
+ (0) => (0);
+ ($e:expr) => {
+ (($e as usize) - (&imp::__ImageBase as *const _ as usize)) as u32
+ }
+ }
+}
+
+#[repr(C)]
+pub struct _ThrowInfo {
+ pub attribues: c_uint,
+ pub pnfnUnwind: imp::ptr_t,
+ pub pForwardCompat: imp::ptr_t,
+ pub pCatchableTypeArray: imp::ptr_t,
+}
+
+#[repr(C)]
+pub struct _CatchableTypeArray {
+ pub nCatchableTypes: c_int,
+ pub arrayOfCatchableTypes: [imp::ptr_t; 2],
+}
-// A code which indicates panics that originate from Rust. Note that some of the
-// upper bits are used by the system so we just set them to 0 and ignore them.
-// 0x 0 R S T
-const RUST_PANIC: c::DWORD = 0x00525354;
+#[repr(C)]
+pub struct _CatchableType {
+ pub properties: c_uint,
+ pub pType: imp::ptr_t,
+ pub thisDisplacement: _PMD,
+ pub sizeOrOffset: c_int,
+ pub copy_function: imp::ptr_t,
+}
+
+#[repr(C)]
+pub struct _PMD {
+ pub mdisp: c_int,
+ pub pdisp: c_int,
+ pub vdisp: c_int,
+}
+
+#[repr(C)]
+pub struct _TypeDescriptor {
+ pub pVFTable: *const u8,
+ pub spare: *mut u8,
+ pub name: [u8; 7],
+}
+
+static mut THROW_INFO: _ThrowInfo = _ThrowInfo {
+ attribues: 0,
+ pnfnUnwind: ptr!(0),
+ pForwardCompat: ptr!(0),
+ pCatchableTypeArray: ptr!(0),
+};
+
+static mut CATCHABLE_TYPE_ARRAY: _CatchableTypeArray = _CatchableTypeArray {
+ nCatchableTypes: 2,
+ arrayOfCatchableTypes: [
+ ptr!(0),
+ ptr!(0),
+ ],
+};
+
+static mut CATCHABLE_TYPE1: _CatchableType = _CatchableType {
+ properties: 1,
+ pType: ptr!(0),
+ thisDisplacement: _PMD {
+ mdisp: 0,
+ pdisp: -1,
+ vdisp: 0,
+ },
+ sizeOrOffset: imp::OFFSET,
+ copy_function: ptr!(0),
+};
+
+static mut CATCHABLE_TYPE2: _CatchableType = _CatchableType {
+ properties: 1,
+ pType: ptr!(0),
+ thisDisplacement: _PMD {
+ mdisp: 0,
+ pdisp: -1,
+ vdisp: 0,
+ },
+ sizeOrOffset: imp::OFFSET,
+ copy_function: ptr!(0),
+};
+
+extern {
+ // The leading `\x01` byte here is actually a magical signal to LLVM to
+ // *not* apply any other mangling like prefixing with a `_` character.
+ //
+ // This symbol is the vtable used by C++'s `std::type_info`. Objects of type
+ // `std::type_info`, type descriptors, have a pointer to this table. Type
+ // descriptors are referenced by the C++ EH structures defined above and
+ // that we construct below.
+ #[link_name = "\x01??_7type_info@@6B@"]
+ static TYPE_INFO_VTABLE: *const u8;
+}
+
+// We use #[lang = "msvc_try_filter"] here as this is the type descriptor which
+// we'll use in LLVM's `catchpad` instruction which ends up also being passed as
+// an argument to the C++ personality function.
+//
+// Again, I'm not entirely sure what this is describing, it just seems to work.
+#[cfg_attr(all(not(test), not(stage0)),
+ lang = "msvc_try_filter")]
+static mut TYPE_DESCRIPTOR1: _TypeDescriptor = _TypeDescriptor {
+ pVFTable: &TYPE_INFO_VTABLE as *const _ as *const _,
+ spare: 0 as *mut _,
+ name: imp::NAME1,
+};
+
+static mut TYPE_DESCRIPTOR2: _TypeDescriptor = _TypeDescriptor {
+ pVFTable: &TYPE_INFO_VTABLE as *const _ as *const _,
+ spare: 0 as *mut _,
+ name: imp::NAME2,
+};
pub unsafe fn panic(data: Box<Any + Send>) -> u32 {
- // As mentioned above, the call stack here is preserved while the filter
- // functions are running, so it's ok to pass stack-local arrays into
- // `RaiseException`.
+ use core::intrinsics::atomic_store;
+
+ // _CxxThrowException executes entirely on this stack frame, so there's no
+ // need to otherwise transfer `data` to the heap. We just pass a stack
+ // pointer to this function.
//
- // The two pointers of the `data` trait object are written to the stack,
- // passed to `RaiseException`, and they're later extracted by the filter
- // function below in the "custom exception information" section of the
- // `EXCEPTION_RECORD` type.
+ // The first argument is the payload being thrown (our two pointers), and
+ // the second argument is the type information object describing the
+ // exception (constructed above).
let ptrs = mem::transmute::<_, raw::TraitObject>(data);
- let ptrs = [ptrs.data, ptrs.vtable];
- c::RaiseException(RUST_PANIC, 0, 2, ptrs.as_ptr() as *mut _);
+ let mut ptrs = [ptrs.data as u64, ptrs.vtable as u64];
+ let mut ptrs_ptr = ptrs.as_mut_ptr();
+
+ // This... may seems surprising, and justifiably so. On 32-bit MSVC the
+ // pointers between these structure are just that, pointers. On 64-bit MSVC,
+ // however, the pointers between structures are rather expressed as 32-bit
+ // offsets from `__ImageBase`.
+ //
+ // Consequently, on 32-bit MSVC we can declare all these pointers in the
+ // `static`s above. On 64-bit MSVC, we would have to express subtraction of
+ // pointers in statics, which Rust does not currently allow, so we can't
+ // actually do that.
+ //
+ // The next best thing, then is to fill in these structures at runtime
+ // (panicking is already the "slow path" anyway). So here we reinterpret all
+ // of these pointer fields as 32-bit integers and then store the
+ // relevant value into it (atomically, as concurrent panics may be
+ // happening). Technically the runtime will probably do a nonatomic read of
+ // these fields, but in theory they never read the *wrong* value so it
+ // shouldn't be too bad...
+ //
+ // In any case, we basically need to do something like this until we can
+ // express more operations in statics (and we may never be able to).
+ atomic_store(&mut THROW_INFO.pCatchableTypeArray as *mut _ as *mut u32,
+ ptr!(&CATCHABLE_TYPE_ARRAY as *const _) as u32);
+ atomic_store(&mut CATCHABLE_TYPE_ARRAY.arrayOfCatchableTypes[0] as *mut _ as *mut u32,
+ ptr!(&CATCHABLE_TYPE1 as *const _) as u32);
+ atomic_store(&mut CATCHABLE_TYPE_ARRAY.arrayOfCatchableTypes[1] as *mut _ as *mut u32,
+ ptr!(&CATCHABLE_TYPE2 as *const _) as u32);
+ atomic_store(&mut CATCHABLE_TYPE1.pType as *mut _ as *mut u32,
+ ptr!(&TYPE_DESCRIPTOR1 as *const _) as u32);
+ atomic_store(&mut CATCHABLE_TYPE2.pType as *mut _ as *mut u32,
+ ptr!(&TYPE_DESCRIPTOR2 as *const _) as u32);
+
+ c::_CxxThrowException(&mut ptrs_ptr as *mut _ as *mut _,
+ &mut THROW_INFO as *mut _ as *mut _);
u32::max_value()
}
-pub fn payload() -> [usize; 2] {
+pub fn payload() -> [u64; 2] {
[0; 2]
}
-pub unsafe fn cleanup(payload: [usize; 2]) -> Box<Any + Send> {
+pub unsafe fn cleanup(payload: [u64; 2]) -> Box<Any + Send> {
mem::transmute(raw::TraitObject {
data: payload[0] as *mut _,
vtable: payload[1] as *mut _,
})
}
-// This is quite a special function, and it's not literally passed in as the
-// filter function for the `catchpad` of the `try` intrinsic. The compiler
-// actually generates its own filter function wrapper which will delegate to
-// this for the actual execution logic for whether the exception should be
-// caught. The reasons for this are:
-//
-// * Each architecture has a slightly different ABI for the filter function
-// here. For example on x86 there are no arguments but on x86_64 there are
-// two.
-// * This function needs access to the stack frame of the `try` intrinsic
-// which is using this filter as a catch pad. This is because the payload
-// of this exception, `Box<Any>`, needs to be transmitted to that
-// location.
-//
-// Both of these differences end up using a ton of weird llvm-specific
-// intrinsics, so it's actually pretty difficult to express the entire
-// filter function in Rust itself. As a compromise, the compiler takes care
-// of all the weird LLVM-specific and platform-specific stuff, getting to
-// the point where this function makes the actual decision about what to
-// catch given two parameters.
-//
-// The first parameter is `*mut EXCEPTION_POINTERS` which is some contextual
-// information about the exception being filtered, and the second pointer is
-// `*mut *mut [usize; 2]` (the payload here). This value points directly
-// into the stack frame of the `try` intrinsic itself, and we use it to copy
-// information from the exception onto the stack.
#[lang = "msvc_try_filter"]
-#[cfg(not(test))]
-unsafe extern fn __rust_try_filter(eh_ptrs: *mut u8,
- payload: *mut u8) -> i32 {
- let eh_ptrs = eh_ptrs as *mut c::EXCEPTION_POINTERS;
- let payload = payload as *mut *mut [usize; 2];
- let record = &*(*eh_ptrs).ExceptionRecord;
- if record.ExceptionCode != RUST_PANIC {
- return 0
- }
- (**payload)[0] = record.ExceptionInformation[0] as usize;
- (**payload)[1] = record.ExceptionInformation[1] as usize;
- return 1
+#[cfg(stage0)]
+unsafe extern fn __rust_try_filter(_eh_ptrs: *mut u8,
+ _payload: *mut u8) -> i32 {
+ return 0
}
// This is required by the compiler to exist (e.g. it's a lang item), but
#[lang = "eh_personality"]
#[cfg(not(test))]
fn rust_eh_personality() {
- unsafe { intrinsics::abort() }
+ unsafe { ::core::intrinsics::abort() }
}
// We're generating an IR snippet that looks like:
//
// declare i32 @rust_try(%func, %data, %ptr) {
- // %slot = alloca i8*
- // call @llvm.localescape(%slot)
- // store %ptr, %slot
+ // %slot = alloca i64*
// invoke %func(%data) to label %normal unwind label %catchswitch
//
// normal:
// %cs = catchswitch within none [%catchpad] unwind to caller
//
// catchpad:
- // %tok = catchpad within %cs [%rust_try_filter]
+ // %tok = catchpad within %cs [%type_descriptor, 0, %slot]
+ // %ptr[0] = %slot[0]
+ // %ptr[1] = %slot[1]
// catchret from %tok to label %caught
//
// caught:
// ret i32 1
// }
//
- // This structure follows the basic usage of the instructions in LLVM
- // (see their documentation/test cases for examples), but a
- // perhaps-surprising part here is the usage of the `localescape`
- // intrinsic. This is used to allow the filter function (also generated
- // here) to access variables on the stack of this intrinsic. This
- // ability enables us to transfer information about the exception being
- // thrown to this point, where we're catching the exception.
+ // This structure follows the basic usage of throw/try/catch in LLVM.
+ // For example, compile this C++ snippet to see what LLVM generates:
+ //
+ // #include <stdint.h>
+ //
+ // int bar(void (*foo)(void), uint64_t *ret) {
+ // try {
+ // foo();
+ // return 0;
+ // } catch(uint64_t a[2]) {
+ // ret[0] = a[0];
+ // ret[1] = a[1];
+ // return 1;
+ // }
+ // }
//
// More information can be found in libstd's seh.rs implementation.
- let slot = Alloca(bcx, Type::i8p(ccx), "slot");
- let localescape = ccx.get_intrinsic(&"llvm.localescape");
- Call(bcx, localescape, &[slot], dloc);
- Store(bcx, local_ptr, slot);
+ let i64p = Type::i64(ccx).ptr_to();
+ let slot = Alloca(bcx, i64p, "slot");
Invoke(bcx, func, &[data], normal.llbb, catchswitch.llbb, dloc);
Ret(normal, C_i32(ccx, 0), dloc);
let cs = CatchSwitch(catchswitch, None, None, 1);
AddHandler(catchswitch, cs, catchpad.llbb);
- let filter = generate_filter_fn(bcx.fcx, bcx.fcx.llfn);
- let filter = BitCast(catchpad, filter, Type::i8p(ccx));
- let tok = CatchPad(catchpad, cs, &[filter]);
+ let tcx = ccx.tcx();
+ let tydesc = match tcx.lang_items.msvc_try_filter() {
+ Some(did) => ::consts::get_static(ccx, did).to_llref(),
+ None => bug!("msvc_try_filter not defined"),
+ };
+ let tok = CatchPad(catchpad, cs, &[tydesc, C_i32(ccx, 0), slot]);
+ let addr = Load(catchpad, slot);
+ let arg1 = Load(catchpad, addr);
+ let val1 = C_i32(ccx, 1);
+ let arg2 = Load(catchpad, InBoundsGEP(catchpad, addr, &[val1]));
+ let local_ptr = BitCast(catchpad, local_ptr, i64p);
+ Store(catchpad, arg1, local_ptr);
+ Store(catchpad, arg2, InBoundsGEP(catchpad, local_ptr, &[val1]));
CatchRet(catchpad, tok, caught.llbb);
Ret(caught, C_i32(ccx, 1), dloc);
return rust_try
}
-// For MSVC-style exceptions (SEH), the compiler generates a filter function
-// which is used to determine whether an exception is being caught (e.g. if it's
-// a Rust exception or some other).
-//
-// This function is used to generate said filter function. The shim generated
-// here is actually just a thin wrapper to call the real implementation in the
-// standard library itself. For reasons as to why, see seh.rs in the standard
-// library.
-fn generate_filter_fn<'a, 'tcx>(fcx: &FunctionContext<'a, 'tcx>,
- rust_try_fn: ValueRef)
- -> ValueRef {
- let ccx = fcx.ccx;
- let tcx = ccx.tcx();
- let dloc = DebugLoc::None;
-
- let rust_try_filter = match tcx.lang_items.msvc_try_filter() {
- Some(did) => {
- Callee::def(ccx, did, tcx.mk_substs(Substs::empty())).reify(ccx).val
- }
- None => bug!("msvc_try_filter not defined"),
- };
-
- let output = ty::FnOutput::FnConverging(tcx.types.i32);
- let i8p = tcx.mk_mut_ptr(tcx.types.i8);
-
- let frameaddress = ccx.get_intrinsic(&"llvm.frameaddress");
- let recoverfp = ccx.get_intrinsic(&"llvm.x86.seh.recoverfp");
- let localrecover = ccx.get_intrinsic(&"llvm.localrecover");
-
- // On all platforms, once we have the EXCEPTION_POINTERS handle as well as
- // the base pointer, we follow the standard layout of:
- //
- // block:
- // %parentfp = call i8* llvm.x86.seh.recoverfp(@rust_try_fn, %bp)
- // %arg = call i8* llvm.localrecover(@rust_try_fn, %parentfp, 0)
- // %ret = call i32 @the_real_filter_function(%ehptrs, %arg)
- // ret i32 %ret
- //
- // The recoverfp intrinsic is used to recover the frame pointer of the
- // `rust_try_fn` function, which is then in turn passed to the
- // `localrecover` intrinsic (pairing with the `localescape` intrinsic
- // mentioned above). Putting all this together means that we now have a
- // handle to the arguments passed into the `try` function, allowing writing
- // to the stack over there.
- //
- // For more info, see seh.rs in the standard library.
- let do_trans = |bcx: Block, ehptrs, base_pointer| {
- let rust_try_fn = BitCast(bcx, rust_try_fn, Type::i8p(ccx));
- let parentfp = Call(bcx, recoverfp, &[rust_try_fn, base_pointer], dloc);
- let arg = Call(bcx, localrecover,
- &[rust_try_fn, parentfp, C_i32(ccx, 0)], dloc);
- let ret = Call(bcx, rust_try_filter, &[ehptrs, arg], dloc);
- Ret(bcx, ret, dloc);
- };
-
- if ccx.tcx().sess.target.target.arch == "x86" {
- // On x86 the filter function doesn't actually receive any arguments.
- // Instead the %ebp register contains some contextual information.
- //
- // Unfortunately I don't know of any great documentation as to what's
- // going on here, all I can say is that there's a few tests cases in
- // LLVM's test suite which follow this pattern of instructions, so we
- // just do the same.
- gen_fn(fcx, "__rustc_try_filter", vec![], output, &mut |bcx| {
- let ebp = Call(bcx, frameaddress, &[C_i32(ccx, 1)], dloc);
- let exn = InBoundsGEP(bcx, ebp, &[C_i32(ccx, -20)]);
- let exn = Load(bcx, BitCast(bcx, exn, Type::i8p(ccx).ptr_to()));
- do_trans(bcx, exn, ebp);
- })
- } else if ccx.tcx().sess.target.target.arch == "x86_64" {
- // Conveniently on x86_64 the EXCEPTION_POINTERS handle and base pointer
- // are passed in as arguments to the filter function, so we just pass
- // those along.
- gen_fn(fcx, "__rustc_try_filter", vec![i8p, i8p], output, &mut |bcx| {
- let exn = llvm::get_param(bcx.fcx.llfn, 0);
- let rbp = llvm::get_param(bcx.fcx.llfn, 1);
- do_trans(bcx, exn, rbp);
- })
- } else {
- bug!("unknown target to generate a filter function")
- }
-}
-
fn span_invalid_monomorphization_error(a: &Session, b: Span, c: &str) {
span_err!(a, b, E0511, "{}", c);
}