43 lines
1.5 KiB
Rust
43 lines
1.5 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.
|
||
|
|
|
||
|
|
use core::sync::atomic::{AtomicPtr, Ordering};
|
||
|
|
use core::mem;
|
||
|
|
use core::intrinsics;
|
||
|
|
|
||
|
|
static OOM_HANDLER: AtomicPtr<()> = AtomicPtr::new(default_oom_handler as *mut ());
|
||
|
|
|
||
|
|
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() -> ! {
|
||
|
|
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);
|
||
|
|
}
|