Skip to content

Commit

Permalink
Make most of the failure functions take &(&'static str, uint)
Browse files Browse the repository at this point in the history
Passing one pointer takes less code than one pointer and an integer.
  • Loading branch information
brson committed Jul 25, 2014
1 parent b9035c2 commit 4636b32
Show file tree
Hide file tree
Showing 4 changed files with 54 additions and 16 deletions.
45 changes: 42 additions & 3 deletions src/libcore/failure.rs
Expand Up @@ -33,33 +33,72 @@
use fmt;
use intrinsics;

#[cfg(stage0)]
#[cold] #[inline(never)] // this is the slow path, always
#[lang="fail_"]
fn fail_(expr: &'static str, file: &'static str, line: uint) -> ! {
format_args!(|args| -> () {
begin_unwind(args, file, line);
begin_unwind(args, &(file, line));
}, "{}", expr);

unsafe { intrinsics::abort() }
}

#[cfg(stage0)]
#[cold]
#[lang="fail_bounds_check"]
fn fail_bounds_check(file: &'static str, line: uint,
index: uint, len: uint) -> ! {
format_args!(|args| -> () {
begin_unwind(args, file, line);
begin_unwind(args, &(file, line));
}, "index out of bounds: the len is {} but the index is {}", len, index);
unsafe { intrinsics::abort() }
}

#[cfg(stage0)]
#[cold]
pub fn begin_unwind(fmt: &fmt::Arguments, file: &'static str, line: uint) -> ! {
pub fn begin_unwind(fmt: &fmt::Arguments, file_line: &(&'static str, uint)) -> ! {
#[allow(ctypes)]
extern {
#[lang = "begin_unwind"]
fn begin_unwind(fmt: &fmt::Arguments, file: &'static str,
line: uint) -> !;
}
let (file, line) = *file_line;
unsafe { begin_unwind(fmt, file, line) }
}

#[cfg(not(stage0))]
#[cold] #[inline(never)] // this is the slow path, always
#[lang="fail_"]
fn fail_(expr: &'static str, file: &'static str, line: uint) -> ! {
format_args!(|args| -> () {
begin_unwind(args, &(file, line));
}, "{}", expr);

unsafe { intrinsics::abort() }
}

#[cfg(not(stage0))]
#[cold]
#[lang="fail_bounds_check"]
fn fail_bounds_check(file: &'static str, line: uint,
index: uint, len: uint) -> ! {
format_args!(|args| -> () {
begin_unwind(args, &(file, line));
}, "index out of bounds: the len is {} but the index is {}", len, index);
unsafe { intrinsics::abort() }
}

#[cfg(not(stage0))]
#[cold]
pub fn begin_unwind(fmt: &fmt::Arguments, file_line: &(&'static str, uint)) -> ! {
#[allow(ctypes)]
extern {
#[lang = "begin_unwind"]
fn begin_unwind(fmt: &fmt::Arguments, file_line: &'static str,
line: uint) -> !;
}
let (file, line) = *file_line;
unsafe { begin_unwind(fmt, file, line) }
}
2 changes: 1 addition & 1 deletion src/libcore/macros.rs
Expand Up @@ -33,7 +33,7 @@ macro_rules! fail(
// up with the number of calls to fail!()
#[inline(always)]
fn run_fmt(fmt: &::std::fmt::Arguments) -> ! {
::core::failure::begin_unwind(fmt, file!(), line!())
::core::failure::begin_unwind(fmt, &(file!(), line!()))
}
format_args!(run_fmt, $fmt, $($arg)*)
});
Expand Down
19 changes: 9 additions & 10 deletions src/librustrt/unwind.rs
Expand Up @@ -384,7 +384,7 @@ pub mod eabi {
#[lang = "begin_unwind"]
pub extern fn rust_begin_unwind(msg: &fmt::Arguments,
file: &'static str, line: uint) -> ! {
begin_unwind_fmt(msg, file, line)
begin_unwind_fmt(msg, &(file, line))
}

/// The entry point for unwinding with a formatted message.
Expand All @@ -394,8 +394,7 @@ pub extern fn rust_begin_unwind(msg: &fmt::Arguments,
/// on (e.g.) the inlining of other functions as possible), by moving
/// the actual formatting into this shared place.
#[inline(never)] #[cold]
pub fn begin_unwind_fmt(msg: &fmt::Arguments, file: &'static str,
line: uint) -> ! {
pub fn begin_unwind_fmt(msg: &fmt::Arguments, file_line: &(&'static str, uint)) -> ! {
use core::fmt::FormatWriter;

// We do two allocations here, unfortunately. But (a) they're
Expand All @@ -415,9 +414,10 @@ pub fn begin_unwind_fmt(msg: &fmt::Arguments, file: &'static str,
let mut v = Vec::new();
let _ = write!(&mut VecWriter { v: &mut v }, "{}", msg);

begin_unwind_inner(box String::from_utf8(v).unwrap(), file, line)
begin_unwind_inner(box String::from_utf8(v).unwrap(), file_line)
}

// FIXME: Need to change expr_fail in AstBuilder to change this to &(str, uint)
/// This is the entry point of unwinding for fail!() and assert!().
#[inline(never)] #[cold] // avoid code bloat at the call sites as much as possible
pub fn begin_unwind<M: Any + Send>(msg: M, file: &'static str, line: uint) -> ! {
Expand All @@ -429,13 +429,13 @@ pub fn begin_unwind<M: Any + Send>(msg: M, file: &'static str, line: uint) -> !
// failing.

// see below for why we do the `Any` coercion here.
begin_unwind_inner(box msg, file, line)
begin_unwind_inner(box msg, &(file, line))
}

/// Unwinding for `fail!()`. Saves passing a string.
#[inline(never)] #[cold] #[experimental]
pub fn begin_unwind_no_time_to_explain(file: &'static str, line: uint) -> ! {
begin_unwind_inner(box () ("explicit failure"), file, line)
pub fn begin_unwind_no_time_to_explain(file_line: &(&'static str, uint)) -> ! {
begin_unwind_inner(box () ("explicit failure"), file_line)
}

/// The core of the unwinding.
Expand All @@ -448,9 +448,7 @@ pub fn begin_unwind_no_time_to_explain(file: &'static str, line: uint) -> ! {
/// Do this split took the LLVM IR line counts of `fn main() { fail!()
/// }` from ~1900/3700 (-O/no opts) to 180/590.
#[inline(never)] #[cold] // this is the slow path, please never inline this
fn begin_unwind_inner(msg: Box<Any + Send>,
file: &'static str,
line: uint) -> ! {
fn begin_unwind_inner(msg: Box<Any + Send>, file_line: &(&'static str, uint)) -> ! {
// First, invoke call the user-defined callbacks triggered on task failure.
//
// By the time that we see a callback has been registered (by reading
Expand All @@ -467,6 +465,7 @@ fn begin_unwind_inner(msg: Box<Any + Send>,
0 => {}
n => {
let f: Callback = unsafe { mem::transmute(n) };
let (file, line) = *file_line;
f(msg, file, line);
}
}
Expand Down
4 changes: 2 additions & 2 deletions src/libstd/macros.rs
Expand Up @@ -39,7 +39,7 @@
#[macro_export]
macro_rules! fail(
() => (
::std::rt::begin_unwind_no_time_to_explain(file!(), line!())
::std::rt::begin_unwind_no_time_to_explain(&(file!(), line!()))
);
($msg:expr) => (
::std::rt::begin_unwind($msg, file!(), line!())
Expand All @@ -58,7 +58,7 @@ macro_rules! fail(
// up with the number of calls to fail!()
#[inline(always)]
fn run_fmt(fmt: &::std::fmt::Arguments) -> ! {
::std::rt::begin_unwind_fmt(fmt, file!(), line!())
::std::rt::begin_unwind_fmt(fmt, &(file!(), line!()))
}
format_args!(run_fmt, $fmt, $($arg)*)
});
Expand Down

0 comments on commit 4636b32

Please sign in to comment.