diff options
| -rw-r--r-- | src/main.rs | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/src/main.rs b/src/main.rs index 167e3fb..c103ee8 100644 --- a/src/main.rs +++ b/src/main.rs @@ -153,13 +153,21 @@ fn notify( let _ = s.send(signal_hook::SIGINT); } }; - let sigquit_handler = move |_info: &nix::libc::siginfo_t| { + let sigquit_handler = { + let state = state.clone(); + move |_info: &nix::libc::siginfo_t| { + crate::state::restore_to_main_screen(state.clone()); + std::process::exit(131); + } + }; + let sigterm_handler = move |_info: &nix::libc::siginfo_t| { crate::state::restore_to_main_screen(state.clone()); - std::process::exit(131); + std::process::exit(143); }; unsafe { signal_hook_registry::register_sigaction(signal_hook::SIGINT, sigint_handler)?; signal_hook_registry::register_sigaction(signal_hook::SIGQUIT, sigquit_handler)?; + signal_hook_registry::register_sigaction(signal_hook::SIGTERM, sigterm_handler)?; } let signals = signal_hook::iterator::Signals::new(signals)?; std::thread::spawn(move || { |