compiler/rustc/src/main.rs

   1 fn main() {
   2     // Pull in jemalloc when enabled.
   3     //
   4     // Note that we're pulling in a static copy of jemalloc which means that to
   5     // pull it in we need to actually reference its symbols for it to get
   6     // linked. The two crates we link to here, std and rustc_driver, are both
   7     // dynamic libraries. That means to pull in jemalloc we actually need to
   8     // reference allocation symbols one way or another (as this file is the only
   9     // object code in the rustc executable).
  10     #[cfg(feature = "jemalloc-sys")]
  11     {
  12         use std::os::raw::{c_int, c_void};
  13
  14         #[used]
  15         static _F1: unsafe extern "C" fn(usize, usize) -> *mut c_void = jemalloc_sys::calloc;
  16         #[used]
  17         static _F2: unsafe extern "C" fn(*mut *mut c_void, usize, usize) -> c_int =
  18             jemalloc_sys::posix_memalign;
  19         #[used]
  20         static _F3: unsafe extern "C" fn(usize, usize) -> *mut c_void = jemalloc_sys::aligned_alloc;
  21         #[used]
  22         static _F4: unsafe extern "C" fn(usize) -> *mut c_void = jemalloc_sys::malloc;
  23         #[used]
  24         static _F5: unsafe extern "C" fn(*mut c_void, usize) -> *mut c_void = jemalloc_sys::realloc;
  25         #[used]
  26         static _F6: unsafe extern "C" fn(*mut c_void) = jemalloc_sys::free;
  27
  28         // On OSX, jemalloc doesn't directly override malloc/free, but instead
  29         // registers itself with the allocator's zone APIs in a ctor. However,
  30         // the linker doesn't seem to consider ctors as "used" when statically
  31         // linking, so we need to explicitly depend on the function.
  32         #[cfg(target_os = "macos")]
  33         {
  34             extern "C" {
  35                 fn _rjem_je_zone_register();
  36             }
  37
  38             #[used]
  39             static _F7: unsafe extern "C" fn() = _rjem_je_zone_register;
  40         }
  41     }
  42
  43     rustc_driver::set_sigpipe_handler();
  44     rustc_driver::main()
  45 }