diff options
Diffstat (limited to 'tests/ui/attributes/unix_sigpipe/auxiliary/sigpipe-utils.rs')
-rw-r--r-- | tests/ui/attributes/unix_sigpipe/auxiliary/sigpipe-utils.rs | 33 |
1 files changed, 33 insertions, 0 deletions
diff --git a/tests/ui/attributes/unix_sigpipe/auxiliary/sigpipe-utils.rs b/tests/ui/attributes/unix_sigpipe/auxiliary/sigpipe-utils.rs new file mode 100644 index 000000000..74fbae035 --- /dev/null +++ b/tests/ui/attributes/unix_sigpipe/auxiliary/sigpipe-utils.rs @@ -0,0 +1,33 @@ +#![feature(rustc_private)] +extern crate libc; + +/// So tests don't have to bring libc in scope themselves +pub enum SignalHandler { + Ignore, + Default, +} + +/// Helper to assert that [`libc::SIGPIPE`] has the expected signal handler. +pub fn assert_sigpipe_handler(expected_handler: SignalHandler) { + #[cfg(unix)] + #[cfg(not(any( + target_os = "emscripten", + target_os = "fuchsia", + target_os = "horizon", + target_os = "android", + )))] + { + let prev = unsafe { libc::signal(libc::SIGPIPE, libc::SIG_IGN) }; + + let expected = match expected_handler { + SignalHandler::Ignore => libc::SIG_IGN, + SignalHandler::Default => libc::SIG_DFL, + }; + assert_eq!(prev, expected, "expected sigpipe value matches actual value"); + + // Unlikely to matter, but restore the old value anyway + unsafe { + libc::signal(libc::SIGPIPE, prev); + }; + } +} |