mysten_common/

logging.rs

// Copyright (c) Mysten Labs, Inc.
// SPDX-License-Identifier: Apache-2.0

use crate::in_test_configuration;
use once_cell::sync::Lazy;

#[macro_export]
macro_rules! fatal {
    ($msg:literal $(, $arg:expr)*) => {{
        if $crate::in_antithesis() {
            let full_msg = format!($msg $(, $arg)*);
            let json = $crate::logging::json!({ "message": full_msg });
            $crate::logging::assert_unreachable_antithesis!($msg, &json);
        }
        tracing::error!(fatal = true, $msg $(, $arg)*);
        panic!($msg $(, $arg)*);
    }};
}

pub use antithesis_sdk::assert_reachable as assert_reachable_antithesis;
pub use antithesis_sdk::assert_sometimes as assert_sometimes_antithesis;
pub use antithesis_sdk::assert_unreachable as assert_unreachable_antithesis;

pub use serde_json::json;

#[inline(always)]
pub fn crash_on_debug() -> bool {
    static CRASH_ON_DEBUG: Lazy<bool> = Lazy::new(|| {
        in_test_configuration() || std::env::var("SUI_ENABLE_DEBUG_ASSERTIONS").is_ok()
    });

    *CRASH_ON_DEBUG
}

#[cfg(msim)]
pub mod intercept_debug_fatal {
    use std::sync::{Arc, Mutex};

    #[derive(Clone)]
    pub struct DebugFatalCallback {
        pub pattern: String,
        pub callback: Arc<dyn Fn() + Send + Sync>,
    }

    thread_local! {
        static INTERCEPT_DEBUG_FATAL: Mutex<Option<DebugFatalCallback>> = Mutex::new(None);
    }

    pub fn register_callback(message: &str, f: impl Fn() + Send + Sync + 'static) {
        INTERCEPT_DEBUG_FATAL.with(|m| {
            *m.lock().unwrap() = Some(DebugFatalCallback {
                pattern: message.to_string(),
                callback: Arc::new(f),
            });
        });
    }

    pub fn get_callback() -> Option<DebugFatalCallback> {
        INTERCEPT_DEBUG_FATAL.with(|m| m.lock().unwrap().clone())
    }
}

#[macro_export]
macro_rules! register_debug_fatal_handler {
    ($message:literal, $f:expr) => {
        #[cfg(msim)]
        $crate::logging::intercept_debug_fatal::register_callback($message, $f);

        #[cfg(not(msim))]
        {
            // silence unused variable warnings from the body of the callback
            let _ = $f;
        }
    };
}

/// Like `debug_fatal!`, but records `$location` (a `&str`) as the
/// `system_invariant_violations` metric label instead of the macro's own
/// `file!():line!()`. Use this when forwarding a caller-supplied location
/// (e.g. from `#[track_caller]` + `Location::caller()`) so the metric points
/// at the user's call site rather than the wrapper.
#[macro_export]
macro_rules! debug_fatal_at {
    ($location:expr, $msg:literal $(, $arg:expr)*) => {{
        loop {
            #[cfg(msim)]
            {
                if let Some(cb) = $crate::logging::intercept_debug_fatal::get_callback() {
                    tracing::error!($msg $(, $arg)*);
                    let msg = format!($msg $(, $arg)*);
                    if msg.contains(&cb.pattern) {
                        (cb.callback)();
                    }
                    break;
                }
            }

            // In antithesis, rather than crashing, we will use the assert_unreachable_antithesis
            // macro to catch the signal that something has gone wrong.
            if !$crate::in_antithesis() && $crate::logging::crash_on_debug() {
                $crate::fatal!($msg $(, $arg)*);
            } else {
                let stacktrace = std::backtrace::Backtrace::capture();
                tracing::error!(debug_fatal = true, stacktrace = ?stacktrace, $msg $(, $arg)*);
                let location: &str = $location;
                if let Some(metrics) = mysten_metrics::get_metrics() {
                    metrics.system_invariant_violations.with_label_values(&[location]).inc();
                }
                if $crate::in_antithesis() {
                    // antithesis requires a literal for first argument. pass the formatted argument
                    // as a string.
                    let full_msg = format!($msg $(, $arg)*);
                    let json = $crate::logging::json!({ "message": full_msg });
                    $crate::logging::assert_unreachable_antithesis!($msg, &json);
                }
            }
            break;
        }
    }};
}

#[macro_export]
macro_rules! debug_fatal {
    //($msg:literal $(, $arg:expr)* $(,)?)
    ($msg:literal $(, $arg:expr)*) => {{
        $crate::debug_fatal_at!(concat!(file!(), ':', line!()), $msg $(, $arg)*);
    }};
}

#[macro_export]
macro_rules! debug_fatal_no_invariant {
    ($msg:literal $(, $arg:expr)*) => {{
        loop {
            #[cfg(msim)]
            {
                if let Some(cb) = $crate::logging::intercept_debug_fatal::get_callback() {
                    tracing::error!($msg $(, $arg)*);
                    let msg = format!($msg $(, $arg)*);
                    if msg.contains(&cb.pattern) {
                        (cb.callback)();
                    }
                    break;
                }
            }

            if !$crate::in_antithesis() && $crate::logging::crash_on_debug() {
                $crate::fatal!($msg $(, $arg)*);
            } else {
                tracing::error!($msg $(, $arg)*);
                if $crate::in_antithesis() {
                    let full_msg = format!($msg $(, $arg)*);
                    let json = $crate::logging::json!({ "message": full_msg });
                    $crate::logging::assert_unreachable_antithesis!($msg, &json);
                }
            }
            break;
        }
    }};
}

#[macro_export]
macro_rules! assert_reachable {
    () => {
        $crate::logging::assert_reachable!("");
    };
    ($message:literal) => {{
        // calling in to antithesis sdk breaks determinisim in simtests (on linux only)
        if !cfg!(msim) {
            $crate::logging::assert_reachable_antithesis!($message);
        } else {
            $crate::assert_reachable_simtest!($message);
        }
    }};
}

#[macro_export]
macro_rules! assert_sometimes {
    ($expr:expr, $message:literal) => {{
        // calling in to antithesis sdk breaks determinisim in simtests (on linux only)
        if !cfg!(msim) {
            $crate::logging::assert_sometimes_antithesis!($expr, $message);
        } else {
            $crate::assert_sometimes_simtest!($expr, $message);
        }
    }};
}

mod tests {
    #[test]
    #[should_panic]
    fn test_fatal() {
        fatal!("This is a fatal error");
    }

    #[test]
    #[should_panic]
    fn test_debug_fatal() {
        if cfg!(debug_assertions) {
            debug_fatal!("This is a debug fatal error");
        } else {
            // pass in release mode as well
            fatal!("This is a fatal error");
        }
    }

    #[cfg(not(debug_assertions))]
    #[test]
    fn test_debug_fatal_release_mode() {
        debug_fatal!("This is a debug fatal error");
    }

    #[test]
    fn test_assert_sometimes_side_effects() {
        let mut x = 0;

        let mut inc = || {
            x += 1;
            true
        };

        assert_sometimes!(inc(), "");
        assert_eq!(x, 1);
    }
}