// Copyright 2013 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 or the MIT license // , at your // option. This file may not be copied, modified, or distributed // except according to those terms. use libc::c_int; use std::io::signal::Signum; use std::rt::rtio::RtioSignal; use homing::{HomingIO, HomeHandle}; use super::{UvError, UvHandle}; use uvll; use uvio::UvIoFactory; pub struct SignalWatcher { handle: *uvll::uv_signal_t, home: HomeHandle, channel: Sender, signal: Signum, } impl SignalWatcher { pub fn new(io: &mut UvIoFactory, signum: Signum, channel: Sender) -> Result<~SignalWatcher, UvError> { let s = ~SignalWatcher { handle: UvHandle::alloc(None::, uvll::UV_SIGNAL), home: io.make_handle(), channel: channel, signal: signum, }; assert_eq!(unsafe { uvll::uv_signal_init(io.uv_loop(), s.handle) }, 0); match unsafe { uvll::uv_signal_start(s.handle, signal_cb, signum as c_int) } { 0 => Ok(s.install()), n => Err(UvError(n)), } } } extern fn signal_cb(handle: *uvll::uv_signal_t, signum: c_int) { let s: &mut SignalWatcher = unsafe { UvHandle::from_uv_handle(&handle) }; assert_eq!(signum as int, s.signal as int); let _ = s.channel.send_opt(s.signal); } impl HomingIO for SignalWatcher { fn home<'r>(&'r mut self) -> &'r mut HomeHandle { &mut self.home } } impl UvHandle for SignalWatcher { fn uv_handle(&self) -> *uvll::uv_signal_t { self.handle } } impl RtioSignal for SignalWatcher {} impl Drop for SignalWatcher { fn drop(&mut self) { let _m = self.fire_homing_missile(); self.close(); } } #[cfg(test)] mod test { use super::super::local_loop; use std::io::signal; use super::SignalWatcher; #[test] fn closing_channel_during_drop_doesnt_kill_everything() { // see issue #10375, relates to timers as well. let (tx, rx) = channel(); let _signal = SignalWatcher::new(local_loop(), signal::Interrupt, tx); spawn(proc() { let _ = rx.recv_opt(); }); // when we drop the SignalWatcher we're going to destroy the channel, // which must wake up the task on the other end } }