src/test/run-pass/panic-handler-chain.rs

   1 // Copyright 2015 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 // ignore-emscripten no threads support
  12
  13 #![feature(panic_handler, std_panic)]
  14 #![feature(const_atomic_usize_new)]
  15
  16 use std::sync::atomic::{AtomicUsize, Ordering};
  17 use std::panic;
  18 use std::thread;
  19
  20 static A: AtomicUsize = AtomicUsize::new(0);
  21 static B: AtomicUsize = AtomicUsize::new(0);
  22
  23 fn main() {
  24     panic::set_hook(Box::new(|_| { A.fetch_add(1, Ordering::SeqCst); }));
  25     let hook = panic::take_hook();
  26     panic::set_hook(Box::new(move |info| {
  27         B.fetch_add(1, Ordering::SeqCst);
  28         hook(info);
  29     }));
  30
  31     let _ = thread::spawn(|| {
  32         panic!();
  33     }).join();
  34
  35     assert_eq!(1, A.load(Ordering::SeqCst));
  36     assert_eq!(1, B.load(Ordering::SeqCst));
  37 }