test_signal.rs

use std::sync::atomic::{AtomicUsize, Ordering};
use std::sync::Arc;
use std::thread;
use std::time::Duration;
use sgx_libc::{siginfo_t, c_int, pid_t, SIGUSR1, SIGUSR2, SIGILL, SIGTERM,SIGINT};
use sgx_libc::ocall::getpid;
use sgx_signal::signal::{register, register_sigaction, unregister, unregister_signal, raise_signal};

pub fn test_signal_forbidden() {
   let ret = register(SIGILL, || ());
   assert_eq!(ret.is_ok(), false);
}

pub fn test_signal_without_pid() {
   let status = Arc::new(AtomicUsize::new(0));
   let action = {
      let status = Arc::clone(&status);
      move || {
         status.store(1, Ordering::Relaxed);
      }
   };

   register(SIGUSR2, action).unwrap();
   raise_signal(SIGUSR2);

   for _ in 0..10 {
      thread::sleep(Duration::from_millis(100));
      let current = status.load(Ordering::Relaxed);
      match current {
         // Not yet
         0 => continue,
         // Good, we are done with the correct result
         _ if current == 1 => return,
         _ => panic!("Wrong result value {}", current),
      }
   }
   panic!("Timed out waiting for the signal");
}

pub fn test_signal_with_pid() {
   let status = Arc::new(AtomicUsize::new(0));
   let action = {
      let status = Arc::clone(&status);
      move |siginfo: &siginfo_t| {
         // Hack: currently, libc exposes only the first 3 fields of siginfo_t. The pid
         // comes somewhat later on. Therefore, we do a Really Ugly Hack and define our
         // own structure (and hope it is correct on all platforms). But hey, this is
         // only the tests, so we are going to get away with this.
         #[repr(C)]
         struct SigInfo {
            _fields: [c_int; 3],
            #[cfg(all(target_pointer_width = "64", target_os = "linux"))]
            _pad: c_int,
            pid: pid_t,
         }
         let info: &SigInfo = unsafe {
            (siginfo as *const _ as *const SigInfo)
               .as_ref()
               .unwrap()
         };

         status.store(info.pid as usize, Ordering::Relaxed);
      }
   };

   let pid = unsafe { getpid() };
   register_sigaction(SIGUSR2, action).unwrap();
   raise_signal(SIGUSR2);

   for _ in 0..10 {
      thread::sleep(Duration::from_millis(100));
      let current = status.load(Ordering::Relaxed);
      match current {
         // Not yet (PID == 0 doesn't happen)
         0 => continue,
         // Good, we are done with the correct result
         _ if current == pid as usize => return,
         _ => panic!("Wrong status value {}", current),
      }
   }
   panic!("Timed out waiting for the signal");
}

// Check that registration works as expected and that unregister tells if it did or not.
pub fn test_signal_register_unregister() {
   let signal = register(SIGUSR1, || ()).unwrap();
   // It was there now, so we can unregister
   assert!(unregister(signal));
   // The next time unregistering does nothing and tells us so.
   assert!(!unregister(signal));
}

pub fn test_signal_register_unregister1() {
   let called = Arc::new(AtomicUsize::new(0));
   let action = {
      let called = Arc::clone(&called);
      move || {
         called.fetch_add(1, Ordering::Relaxed);
      }
   };

   register(SIGTERM, action.clone()).unwrap();
   register(SIGTERM, action.clone()).unwrap();
   register(SIGINT, action.clone()).unwrap();

   raise_signal(SIGTERM);
   // The closure is run twice.
   assert_eq!(2, called.load(Ordering::Relaxed));

   assert!(unregister_signal(SIGTERM));

   raise_signal(SIGTERM);
   // Second one unregisters nothing.
   assert!(!unregister_signal(SIGTERM));

   // After unregistering (both), it is no longer run at all.
   assert_eq!(2, called.load(Ordering::Relaxed));

   // The SIGINT one is not disturbed.
   raise_signal(SIGINT);
   assert_eq!(3, called.load(Ordering::Relaxed));

   // But it's possible to register it again just fine.
   register(SIGTERM, action).unwrap();

   raise_signal(SIGTERM);
   assert_eq!(4, called.load(Ordering::Relaxed));
}