also make panic_fmt track_caller

This commit is contained in:
Ralf Jung 2020-03-10 10:31:03 +01:00
parent f34e0664a1
commit 0b2329da9a
2 changed files with 36 additions and 7 deletions

View File

@ -1,3 +1,4 @@
#[cfg(bootstrap)]
#[doc(include = "panic.md")] #[doc(include = "panic.md")]
#[macro_export] #[macro_export]
#[allow_internal_unstable(core_panic, track_caller)] #[allow_internal_unstable(core_panic, track_caller)]
@ -20,6 +21,26 @@ macro_rules! panic {
); );
} }
#[cfg(not(bootstrap))]
#[doc(include = "panic.md")]
#[macro_export]
#[allow_internal_unstable(core_panic, track_caller)]
#[stable(feature = "core", since = "1.6.0")]
macro_rules! panic {
() => (
$crate::panic!("explicit panic")
);
($msg:expr) => (
$crate::panicking::panic($msg)
);
($msg:expr,) => (
$crate::panic!($msg)
);
($fmt:expr, $($arg:tt)+) => (
$crate::panicking::panic_fmt($crate::format_args!($fmt, $($arg)+))
);
}
/// Asserts that two expressions are equal to each other (using [`PartialEq`]). /// Asserts that two expressions are equal to each other (using [`PartialEq`]).
/// ///
/// On panic, this macro will print the values of the expressions with their /// On panic, this macro will print the values of the expressions with their

View File

@ -32,6 +32,7 @@
use crate::fmt; use crate::fmt;
use crate::panic::{Location, PanicInfo}; use crate::panic::{Location, PanicInfo};
/// The underlying implementation of libcore's `panic!` macro when no formatting is used.
#[cold] #[cold]
// never inline unless panic_immediate_abort to avoid code // never inline unless panic_immediate_abort to avoid code
// bloat at the call sites as much as possible // bloat at the call sites as much as possible
@ -49,7 +50,10 @@ pub fn panic(expr: &str) -> ! {
// truncation and padding (even though none is used here). Using // truncation and padding (even though none is used here). Using
// Arguments::new_v1 may allow the compiler to omit Formatter::pad from the // Arguments::new_v1 may allow the compiler to omit Formatter::pad from the
// output binary, saving up to a few kilobytes. // output binary, saving up to a few kilobytes.
panic_fmt(fmt::Arguments::new_v1(&[expr], &[]), Location::caller()) #[cfg(not(bootstrap))]
panic_fmt(fmt::Arguments::new_v1(&[expr], &[]));
#[cfg(bootstrap)]
panic_fmt(fmt::Arguments::new_v1(&[expr], &[]), Location::caller());
} }
#[cfg(not(bootstrap))] #[cfg(not(bootstrap))]
@ -62,13 +66,11 @@ fn panic_bounds_check(index: usize, len: usize) -> ! {
unsafe { super::intrinsics::abort() } unsafe { super::intrinsics::abort() }
} }
panic_fmt( panic!("index out of bounds: the len is {} but the index is {}", len, index)
format_args!("index out of bounds: the len is {} but the index is {}", len, index),
Location::caller(),
)
} }
// For bootstrap, we need a variant with the old argument order. // For bootstrap, we need a variant with the old argument order, and a corresponding
// `panic_fmt`.
#[cfg(bootstrap)] #[cfg(bootstrap)]
#[cold] #[cold]
#[cfg_attr(not(feature = "panic_immediate_abort"), inline(never))] #[cfg_attr(not(feature = "panic_immediate_abort"), inline(never))]
@ -84,10 +86,12 @@ fn panic_bounds_check(location: &Location<'_>, index: usize, len: usize) -> ! {
) )
} }
/// The underlying implementation of libcore's `panic!` macro when formatting is used.
#[cold] #[cold]
#[cfg_attr(not(feature = "panic_immediate_abort"), inline(never))] #[cfg_attr(not(feature = "panic_immediate_abort"), inline(never))]
#[cfg_attr(feature = "panic_immediate_abort", inline)] #[cfg_attr(feature = "panic_immediate_abort", inline)]
pub fn panic_fmt(fmt: fmt::Arguments<'_>, location: &Location<'_>) -> ! { #[cfg_attr(not(bootstrap), track_caller)]
pub fn panic_fmt(fmt: fmt::Arguments<'_>, #[cfg(bootstrap)] location: &Location<'_>) -> ! {
if cfg!(feature = "panic_immediate_abort") { if cfg!(feature = "panic_immediate_abort") {
unsafe { super::intrinsics::abort() } unsafe { super::intrinsics::abort() }
} }
@ -99,6 +103,10 @@ pub fn panic_fmt(fmt: fmt::Arguments<'_>, location: &Location<'_>) -> ! {
fn panic_impl(pi: &PanicInfo<'_>) -> !; fn panic_impl(pi: &PanicInfo<'_>) -> !;
} }
#[cfg(bootstrap)]
let pi = PanicInfo::internal_constructor(Some(&fmt), location); let pi = PanicInfo::internal_constructor(Some(&fmt), location);
#[cfg(not(bootstrap))]
let pi = PanicInfo::internal_constructor(Some(&fmt), Location::caller());
unsafe { panic_impl(&pi) } unsafe { panic_impl(&pi) }
} }