62 lines
1.9 KiB
Rust
62 lines
1.9 KiB
Rust
// Copyright 2014-2015 The Rust Project Developers. See the COPYRIGHT
|
|
// file at the top-level directory of this distribution and at
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
// option. This file may not be copied, modified, or distributed
|
|
// except according to those terms.
|
|
|
|
#[cfg(target_has_atomic = "ptr")]
|
|
pub use self::imp::set_oom_handler;
|
|
use core::intrinsics;
|
|
|
|
fn default_oom_handler() -> ! {
|
|
// The default handler can't do much more since we can't assume the presence
|
|
// of libc or any way of printing an error message.
|
|
unsafe { intrinsics::abort() }
|
|
}
|
|
|
|
/// Common out-of-memory routine
|
|
#[cold]
|
|
#[inline(never)]
|
|
#[unstable(feature = "oom", reason = "not a scrutinized interface",
|
|
issue = "27700")]
|
|
pub fn oom() -> ! {
|
|
self::imp::oom()
|
|
}
|
|
|
|
#[cfg(target_has_atomic = "ptr")]
|
|
mod imp {
|
|
use core::mem;
|
|
use core::sync::atomic::{AtomicPtr, Ordering};
|
|
|
|
static OOM_HANDLER: AtomicPtr<()> = AtomicPtr::new(super::default_oom_handler as *mut ());
|
|
|
|
#[inline(always)]
|
|
pub fn oom() -> ! {
|
|
let value = OOM_HANDLER.load(Ordering::SeqCst);
|
|
let handler: fn() -> ! = unsafe { mem::transmute(value) };
|
|
handler();
|
|
}
|
|
|
|
/// Set a custom handler for out-of-memory conditions
|
|
///
|
|
/// To avoid recursive OOM failures, it is critical that the OOM handler does
|
|
/// not allocate any memory itself.
|
|
#[unstable(feature = "oom", reason = "not a scrutinized interface",
|
|
issue = "27700")]
|
|
pub fn set_oom_handler(handler: fn() -> !) {
|
|
OOM_HANDLER.store(handler as *mut (), Ordering::SeqCst);
|
|
}
|
|
}
|
|
|
|
#[cfg(not(target_has_atomic = "ptr"))]
|
|
mod imp {
|
|
#[inline(always)]
|
|
pub fn oom() -> ! {
|
|
super::default_oom_handler()
|
|
}
|
|
}
|