Handle SIGPIPE

Fix `SIGPIPE` being ignored, which would cause the `mullvad-cli` to
panic if it received a `PIPE` signal (e.g. it was piped into `echo`).

Author: MarkusPettersson98

CHANGELOG.md

@@ -32,6 +32,8 @@ Line wrap the file at 100 chars.                                              Th
 - Remove "Any" option for tunnel protocol. The default is now WireGuard.
 
 ### Fixed
+- Fix `mullvad-cli` panicking if it tried to write to a closed pipe on Linux and macOS.
+
 #### macOS
 - Fix routing issue caused by upgrading `tun`.
 

Cargo.lock

@@ -34,6 +34,7 @@ serde_json = { workspace = true }
 
 [target.'cfg(all(unix, not(target_os = "android")))'.dependencies]
 clap_complete = { version = "4.4.8" }
+nix = { version = "0.29.0", features = ["signal"] }
 
 [target.'cfg(windows)'.build-dependencies]
 winres = "0.1"

mullvad-cli/Cargo.toml

@@ -152,6 +152,12 @@ enum Cli {
 
 #[tokio::main]
 async fn main() -> Result<()> {
+    // Handle SIGPIPE
+    // https://stackoverflow.com/questions/65755853/simple-word-count-rust-program-outputs-valid-stdout-but-panicks-when-piped-to-he/65760807
+    // https://github.com/typst/typst/pull/5444
+    #[cfg(unix)]
+    handle_sigpipe().unwrap();
+
     match Cli::parse() {
         Cli::Account(cmd) => cmd.handle().await,
         Cli::Bridge(cmd) => cmd.handle().await,
@@ -189,3 +195,15 @@ async fn main() -> Result<()> {
         }
     }
 }
+
+/// Install the default signal handler for `SIGPIPE`.
+///
+/// By default, Rust replaces it with an empty handler because reasons: https://github.com/rust-lang/rust/issues/119980
+#[cfg(unix)]
+fn handle_sigpipe() -> Result<(), nix::errno::Errno> {
+    use nix::sys::signal::{signal, SigHandler, Signal};
+    // SAFETY: We do not use the previous signal handler, which could cause UB if done carelessly:
+    // https://pubs.opengroup.org/onlinepubs/9699919799/functions/signal.html
+    unsafe { signal(Signal::SIGPIPE, SigHandler::SigDfl) }?;
+    Ok(())
+}