Forward signals to child process

Cargo.toml

@@ -39,6 +39,8 @@ remoteprocess = {version="0.4.11", features=["unwind"]}
 chrono = "0.4.19"
 reqwest = { version = "0.11", features = ["blocking"] }
 tokio = { version = "1", features = ["full"] }
+signal-hook = "0.3.17"
+nix = {version="0.28.0", features = ["signal"] }
 
 [dev-dependencies]
 py-spy-testdata = "0.1.0"

src/main.rs

@@ -37,7 +37,11 @@ use console::style;
 
 use config::{Config, FileFormat, RecordDuration};
 use console_viewer::ConsoleViewer;
+use libc::{SIGINT, SIGTERM};
 use reqwest::StatusCode;
+use signal_hook::iterator::Signals;
+use nix::unistd::Pid;
+use nix::sys::signal::{self, Signal};
 use stack_trace::{Frame, StackTrace};
 
 use chrono::{Local, SecondsFormat};
@@ -597,6 +601,15 @@ fn pyspy_main() -> Result<(), Error> {
             .spawn()
             .map_err(|e| format_err!("Failed to create process '{}': {}", subprocess[0], e))?;
 
+        let mut signals = Signals::new(&[SIGINT, SIGTERM])?;
+        let child_pid = command.id();
+        std::thread::spawn(move || {
+            for sig in signals.forever() {
+                signal::kill(Pid::from_raw(child_pid.try_into().unwrap()), Signal::try_from(sig).unwrap()).unwrap();
+                println!("Received signal {:?}", sig);
+            }
+        });
+
         #[cfg(target_os = "macos")]
         {
             // sleep just in case: https://jvns.ca/blog/2018/01/28/mac-freeze/

src/sampler.rs

@@ -56,7 +56,7 @@ impl Sampler {
                     spy
                 }
                 Err(e) => {
-                    initialized_tx.send(Err(e)).is_err();
+                    let _ = initialized_tx.send(Err(e)).is_err();
                     return;
                 }
             };
@@ -308,7 +308,7 @@ impl PythonSpyThread {
                 }
                 Err(e) => {
                     warn!("Failed to profile python from process {}: {}", pid, e);
-                    initialized_tx.send(Err(e)).is_err();
+                    let _ = initialized_tx.send(Err(e)).is_err();
                     return;
                 }
             };