oxen-server 0.52.9

//! On panic, log a backtrace; on fatal signals (SIGBUS, SIGABRT), log which
//! signal arrived so CloudWatch captures abrupt exits that otherwise leave no
//! application log.

use std::backtrace::Backtrace;
use std::io::{self, Write};
use std::panic;

#[cfg(unix)]
use std::sync::atomic::{AtomicBool, Ordering};
#[cfg(unix)]
use std::thread;

#[cfg(unix)]
use signal_hook::consts::signal::{SIGABRT, SIGBUS};

#[cfg(unix)]
static FATAL_SIGNAL_LOGGED: AtomicBool = AtomicBool::new(false);

fn log_backtrace(label: &str) {
    let mut stderr = io::stderr().lock();
    let _ = writeln!(stderr, "{label}");
    let _ = writeln!(stderr, "{:?}", Backtrace::force_capture());
}

fn install_panic_hook() {
    let default_hook = panic::take_hook();
    panic::set_hook(Box::new(move |info| {
        log_backtrace(&format!("oxen-server panic: {info}"));
        default_hook(info);
    }));
}

#[cfg(unix)]
fn install_fatal_signal_handler() -> Result<(), String> {
    let mut signals = signal_hook::iterator::Signals::new([SIGBUS, SIGABRT])
        .map_err(|e| format!("failed to register fatal signal handlers: {e}"))?;

    thread::spawn(move || {
        for signal in signals.forever() {
            if FATAL_SIGNAL_LOGGED
                .compare_exchange(false, true, Ordering::SeqCst, Ordering::SeqCst)
                .is_err()
            {
                unsafe {
                    libc::raise(signal);
                }
            }

            let name = match signal {
                SIGBUS => "SIGBUS",
                SIGABRT => "SIGABRT",
                _ => "SIGNAL",
            };
            // A backtrace here would capture this listener thread, not the faulting thread.
            let mut stderr = io::stderr().lock();
            let _ = writeln!(
                stderr,
                "oxen-server received fatal signal {name} ({signal})"
            );

            unsafe {
                libc::signal(signal, libc::SIG_DFL);
                libc::raise(signal);
            }
        }
    });

    Ok(())
}

#[cfg(not(unix))]
fn install_fatal_signal_handler() -> Result<(), String> {
    Ok(())
}

pub fn install() -> Result<(), String> {
    install_panic_hook();
    install_fatal_signal_handler()
}