track-caller-ffi.rs

//@ run-pass

use std::panic::Location;

unsafe extern "Rust" {
    #[track_caller]
    safe fn rust_track_caller_ffi_test_tracked() -> &'static Location<'static>;
    safe fn rust_track_caller_ffi_test_untracked() -> &'static Location<'static>;
}

fn rust_track_caller_ffi_test_nested_tracked() -> &'static Location<'static> {
    rust_track_caller_ffi_test_tracked()
}

mod provides {
    use std::panic::Location;
    #[track_caller] // UB if we did not have this!
    #[no_mangle]
    fn rust_track_caller_ffi_test_tracked() -> &'static Location<'static> {
        Location::caller()
    }
    #[no_mangle]
    fn rust_track_caller_ffi_test_untracked() -> &'static Location<'static> {
        Location::caller()
    }
}

fn main() {
    let location = Location::caller();
    assert_eq!(location.file(), file!());
    assert_eq!(location.line(), 29);
    assert_eq!(location.column(), 20);

    let tracked = rust_track_caller_ffi_test_tracked();
    assert_eq!(tracked.file(), file!());
    assert_eq!(tracked.line(), 34);
    assert_eq!(tracked.column(), 19);

    let untracked = rust_track_caller_ffi_test_untracked();
    assert_eq!(untracked.file(), file!());
    assert_eq!(untracked.line(), 24);
    assert_eq!(untracked.column(), 9);

    let contained = rust_track_caller_ffi_test_nested_tracked();
    assert_eq!(contained.file(), file!());
    assert_eq!(contained.line(), 12);
    assert_eq!(contained.column(), 5);

    let indirect = (rust_track_caller_ffi_test_tracked as fn() -> &'static Location<'static>)();
    assert_eq!(indirect.file(), file!());
    assert_eq!(indirect.line(), 7);
    assert_eq!(indirect.column(), 5);
}