summaryrefslogtreecommitdiff
path: root/meap/meap-code/ch10/ch10-handler/src/main.rs
blob: 088c9e6c10e8f4e51a44f9bde1b56612acce2d38 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#![cfg(not(windows))]

extern crate libc;

use std::mem;
use std::time;
use std::thread::{sleep};
use libc::{SIGTERM, SIGALRM, SIGHUP, SIGQUIT};

static mut SHUT_DOWN: bool = false;

#[inline]
fn register_signal_handler() {
    let fn_ptr: fn(i32) -> () = handle_signals;                     // <1> hardcoding for simplicity

    unsafe {                                                       
        let fn_ptr_as_usize: usize = mem::transmute(fn_ptr);        // <2>
        libc::signal(SIGTERM, fn_ptr_as_usize);                     // <3>
    }
}

#[allow(dead_code)]                                                // <4>
fn handle_signals(sig: i32) {
    register_signal_handler();

    unsafe {                                                          // <4>
        SHUT_DOWN = match sig {                                    // <5>
            SIGALRM  => false,                                     // <5>
            SIGHUP   => false,                                     // <5>
            SIGTERM  => true,                                      // <5>
            SIGQUIT  => true,                                      // <5>
            _ => false,                                            // <5>
        };
    }
}

fn main() {
    register_signal_handler();
    let delay = time::Duration::from_secs(1);

    for i in 1.. {                                               // <6>
        unsafe {                                                    // <6>
            if SHUT_DOWN {
                println!();  // only print a newline character
                return;
            }
        }

        sleep(delay);
        print!(".");

        if i > 3 {
            unsafe {
                libc::raise(SIGTERM);
            }
        }
    }
}