src/libstd/rt/mod.rs

   1 // Copyright 2013 The Rust Project Developers. See the COPYRIGHT
   2 // file at the top-level directory of this distribution and at
   3 // http://rust-lang.org/COPYRIGHT.
   4 //
   5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
   6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
   8 // option. This file may not be copied, modified, or distributed
   9 // except according to those terms.
  10
  11 //! Runtime services
  12 //!
  13 //! The `rt` module provides a narrow set of runtime services,
  14 //! including the global heap (exported in `heap`) and unwinding and
  15 //! backtrace support. The APIs in this module are highly unstable,
  16 //! and should be considered as private implementation details for the
  17 //! time being.
  18
  19 #![experimental]
  20
  21 // FIXME: this should not be here.
  22 #![allow(missing_docs)]
  23
  24 #![allow(dead_code)]
  25
  26 use marker::Send;
  27 use ops::FnOnce;
  28 use sys;
  29 use thunk::Thunk;
  30
  31 // Reexport some of our utilities which are expected by other crates.
  32 pub use self::util::{default_sched_threads, min_stack, running_on_valgrind};
  33 pub use self::unwind::{begin_unwind, begin_unwind_fmt};
  34
  35 // Reexport some functionality from liballoc.
  36 pub use alloc::heap;
  37
  38 // Simple backtrace functionality (to print on panic)
  39 pub mod backtrace;
  40
  41 // Internals
  42 #[macro_use]
  43 mod macros;
  44
  45 // These should be refactored/moved/made private over time
  46 pub mod util;
  47 pub mod unwind;
  48 pub mod args;
  49
  50 mod at_exit_imp;
  51 mod libunwind;
  52
  53 /// The default error code of the rust runtime if the main thread panics instead
  54 /// of exiting cleanly.
  55 pub const DEFAULT_ERROR_CODE: int = 101;
  56
  57 #[cfg(any(windows, android))]
  58 const OS_DEFAULT_STACK_ESTIMATE: uint = 1 << 20;
  59 #[cfg(all(unix, not(android)))]
  60 const OS_DEFAULT_STACK_ESTIMATE: uint = 2 * (1 << 20);
  61
  62 #[cfg(not(test))]
  63 #[lang = "start"]
  64 fn lang_start(main: *const u8, argc: int, argv: *const *const u8) -> int {
  65     use prelude::v1::*;
  66
  67     use mem;
  68     use os;
  69     use rt;
  70     use sys_common::thread_info::{self, NewThread};
  71     use sys_common;
  72     use thread::Thread;
  73
  74     let something_around_the_top_of_the_stack = 1;
  75     let addr = &something_around_the_top_of_the_stack as *const int;
  76     let my_stack_top = addr as uint;
  77
  78     // FIXME #11359 we just assume that this thread has a stack of a
  79     // certain size, and estimate that there's at most 20KB of stack
  80     // frames above our current position.
  81     let my_stack_bottom = my_stack_top + 20000 - OS_DEFAULT_STACK_ESTIMATE;
  82
  83     let failed = unsafe {
  84         // First, make sure we don't trigger any __morestack overflow checks,
  85         // and next set up our stack to have a guard page and run through our
  86         // own fault handlers if we hit it.
  87         sys_common::stack::record_os_managed_stack_bounds(my_stack_bottom,
  88                                                           my_stack_top);
  89         sys::thread::guard::init();
  90         sys::stack_overflow::init();
  91
  92         // Next, set up the current Thread with the guard information we just
  93         // created. Note that this isn't necessary in general for new threads,
  94         // but we just do this to name the main thread and to give it correct
  95         // info about the stack bounds.
  96         let thread: Thread = NewThread::new(Some("<main>".to_string()));
  97         thread_info::set((my_stack_bottom, my_stack_top),
  98                          sys::thread::guard::main(),
  99                          thread);
 100
 101         // By default, some platforms will send a *signal* when a EPIPE error
 102         // would otherwise be delivered. This runtime doesn't install a SIGPIPE
 103         // handler, causing it to kill the program, which isn't exactly what we
 104         // want!
 105         //
 106         // Hence, we set SIGPIPE to ignore when the program starts up in order
 107         // to prevent this problem.
 108         #[cfg(windows)] fn ignore_sigpipe() {}
 109         #[cfg(unix)] fn ignore_sigpipe() {
 110             use libc;
 111             use libc::funcs::posix01::signal::signal;
 112             unsafe {
 113                 assert!(signal(libc::SIGPIPE, libc::SIG_IGN) != -1);
 114             }
 115         }
 116         ignore_sigpipe();
 117
 118         // Store our args if necessary in a squirreled away location
 119         args::init(argc, argv);
 120
 121         // And finally, let's run some code!
 122         let res = unwind::try(|| {
 123             let main: fn() = mem::transmute(main);
 124             main();
 125         });
 126         cleanup();
 127         res.is_err()
 128     };
 129
 130     // If the exit code wasn't set, then the try block must have panicked.
 131     if failed {
 132         rt::DEFAULT_ERROR_CODE
 133     } else {
 134         os::get_exit_status()
 135     }
 136 }
 137
 138 /// Enqueues a procedure to run when the runtime is cleaned up
 139 ///
 140 /// The procedure passed to this function will be executed as part of the
 141 /// runtime cleanup phase. For normal rust programs, this means that it will run
 142 /// after all other threads have exited.
 143 ///
 144 /// The procedure is *not* executed with a local `Thread` available to it, so
 145 /// primitives like logging, I/O, channels, spawning, etc, are *not* available.
 146 /// This is meant for "bare bones" usage to clean up runtime details, this is
 147 /// not meant as a general-purpose "let's clean everything up" function.
 148 ///
 149 /// It is forbidden for procedures to register more `at_exit` handlers when they
 150 /// are running, and doing so will lead to a process abort.
 151 pub fn at_exit<F:FnOnce()+Send>(f: F) {
 152     at_exit_imp::push(Thunk::new(f));
 153 }
 154
 155 /// One-time runtime cleanup.
 156 ///
 157 /// This function is unsafe because it performs no checks to ensure that the
 158 /// runtime has completely ceased running. It is the responsibility of the
 159 /// caller to ensure that the runtime is entirely shut down and nothing will be
 160 /// poking around at the internal components.
 161 ///
 162 /// Invoking cleanup while portions of the runtime are still in use may cause
 163 /// undefined behavior.
 164 pub unsafe fn cleanup() {
 165     args::cleanup();
 166     sys::stack_overflow::cleanup();
 167     // FIXME: (#20012): the resources being cleaned up by at_exit
 168     // currently are not prepared for cleanup to happen asynchronously
 169     // with detached threads using the resources; for now, we leak.
 170     // at_exit_imp::cleanup();
 171 }