Miri fails when unwinding in safe code

[joseph-gio]

Bevy version

f7c8eb7

What you did

I created this test:

#[test]
#[should_panic]
fn ub_find() {
    #[derive(Resource, Default)]
    struct SyncCounter(RwLock<()>);

    fn assert_non_sync(counter: Res<SyncCounter>) {
        // Panic if multiple instances of this system try to access `counter` at once.
        let _guard = counter.0.try_write().unwrap();

        // Make sure other systems have a chance to conflict with this one before dropping the guard.
        std::thread::sleep(std::time::Duration::from_millis(100));
    }

    let mut world = World::default();
    world.init_resource::<SyncCounter>();

    let mut stage = SystemStage::parallel()
        .with_system(assert_non_sync)
        .with_system(assert_non_sync)
        .with_system(assert_non_sync);

    stage.run(&mut world);
}

What were you expecting

The multiple instances of assert_non_sync should conflict with each other and safely cause a panic.

What went wrong

The following miri failure is triggered: https://github.com/JoJoJet/bevy/actions/runs/3457086747/jobs/5770279259

Additional information

I investigated a bit further on this branch. The test seems to randomly not fail sometimes, but eventually this failure was triggered:

https://github.com/JoJoJet/bevy/actions/runs/3457314695/jobs/5770662745

Some theories from @BoxyUwU https://discord.com/channels/691052431525675048/749335865876021248/1041500126880989184.

[hymm]

can you test if this PR helps? #6524

[joseph-gio]

It seems to. Hopefully miri isn't just spuriously passing again.

[james7132]

error: Undefined Behavior: not granting access to tag <41945721> because that would remove [Unique for <41937186>] which is protected because it is an argument of call 11782684
    --> /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/core/src/ptr/mod.rs:490:1
     |
490  | pub unsafe fn drop_in_place<T: ?Sized>(to_drop: *mut T) {
     | ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ not granting access to tag <41945721> because that would remove [Unique for <41937186>] which is protected because it is an argument of call 11782684
     |
     = help: this indicates a potential bug in the program: it performed an invalid operation, but the Stacked Borrows rules it violated are still experimental
     = help: see https://github.com/rust-lang/unsafe-code-guidelines/blob/master/wip/stacked-borrows.md for further information
help: <41945721> was created by a SharedReadWrite retag at offsets [0xc8..0xe0]
    --> crates/bevy_ecs/src/lib.rs:1200:5
     |
1200 |     }
     |     ^
help: <41937186> is this argument
    --> crates/bevy_ecs/src/system/function_system.rs:396:26
     |
396  |     unsafe fn run_unsafe(&mut self, input: Self::In, world: &World) -> Self::Out {
     |                          ^^^^^^^^^
     = note: BACKTRACE:
     = note: inside `std::ptr::drop_in_place::<std::borrow::Cow<'_, str>> - shim(Some(std::borrow::Cow<'_, str>))` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/core/src/ptr/mod.rs:490:1
     = note: inside `std::ptr::drop_in_place::<system::function_system::SystemMeta> - shim(Some(system::function_system::SystemMeta))` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/core/src/ptr/mod.rs:490:1
     = note: inside `std::ptr::drop_in_place::<system::function_system::FunctionSystem<(), (), (system::system_param::Res<'_, tests::ub_find::SyncCounter>,), (), for<'a> fn(system::system_param::Res<'a, tests::ub_find::SyncCounter>) {tests::ub_find::assert_non_sync}>> - shim(Some(system::function_system::FunctionSystem<(), (), (system::system_param::Res<'_, tests::ub_find::SyncCounter>,), (), for<'a> fn(system::system_param::Res<'a, tests::ub_find::SyncCounter>) {tests::ub_find::assert_non_sync}>))` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/core/src/ptr/mod.rs:490:1
     = note: inside `std::ptr::drop_in_place::<std::boxed::Box<dyn system::system::System<Out = (), In = ()>>> - shim(Some(std::boxed::Box<dyn system::system::System<Out = (), In = ()>>))` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/core/src/ptr/mod.rs:490:1
     = note: inside `std::ptr::drop_in_place::<schedule::system_container::SystemContainer> - shim(Some(schedule::system_container::SystemContainer))` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/core/src/ptr/mod.rs:490:1
     = note: inside `std::ptr::drop_in_place::<[schedule::system_container::SystemContainer]> - shim(Some([schedule::system_container::SystemContainer]))` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/core/src/ptr/mod.rs:490:1
     = note: inside `<std::vec::Vec<schedule::system_container::SystemContainer> as std::ops::Drop>::drop` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/alloc/src/vec/mod.rs:3029:13
     = note: inside `std::ptr::drop_in_place::<std::vec::Vec<schedule::system_container::SystemContainer>> - shim(Some(std::vec::Vec<schedule::system_container::SystemContainer>))` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/core/src/ptr/mod.rs:490:1
     = note: inside `std::ptr::drop_in_place::<schedule::stage::SystemStage> - shim(Some(schedule::stage::SystemStage))` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/core/src/ptr/mod.rs:490:1
note: inside `tests::ub_find` at crates/bevy_ecs/src/lib.rs:1200:5
    --> crates/bevy_ecs/src/lib.rs:1200:5
     |
1200 |     }
     |     ^
note: inside closure at crates/bevy_ecs/src/lib.rs:1179:18
    --> crates/bevy_ecs/src/lib.rs:1179:18
     |
1177 |     #[test]
     |     ------- in this procedural macro expansion
1178 |     #[should_panic]
1179 |     fn ub_find() {
     |                  ^
     = note: this error originates in the attribute macro `test` (in Nightly builds, run with -Z macro-backtrace for more info)

note: some details are omitted, run with `MIRIFLAGS=-Zmiri-backtrace=full` for a verbose backtrace

For posterity, here's the miri output, since GitHub Action's logs are TLL'ed.

[hymm]

I wonder if the issue is that we should be using FallibleTask rather than Task in TaskPool::scope. So that when the scope gets dropped we can cancel().await the tasks instead of just having them just doing set_cancel() when the task is dropped.

[joseph-gio]

Looks like miri still fails periodically, even with #6524: https://github.com/JoJoJet/bevy/actions/runs/3484882520/jobs/5829897858

error: Undefined Behavior: Data race detected between Deallocate on thread `tests::ub_find` and Read on thread `TaskPool (0)` at alloc18680771+0x8
    --> /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/alloc/src/alloc.rs:113:14
     |
113  |     unsafe { __rust_dealloc(ptr, layout.size(), layout.align()) }
     |              ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Data race detected between Deallocate on thread `tests::ub_find` and Read on thread `TaskPool (0)` at alloc18680771+0x8
     |
     = help: this indicates a bug in the program: it performed an invalid operation, and caused Undefined Behavior
     = help: see https://doc.rust-lang.org/nightly/reference/behavior-considered-undefined.html for further information
     = note: BACKTRACE:
     = note: inside `std::alloc::dealloc` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/alloc/src/alloc.rs:113:14
     = note: inside `<std::alloc::Global as std::alloc::Allocator>::deallocate` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/alloc/src/alloc.rs:250:22
     = note: inside `downcast_rs::alloc::alloc::box_free::<dyn system::system::System<In = (), Out = ()>, std::alloc::Global>` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/alloc/src/alloc.rs:348:9
     = note: inside `std::ptr::drop_in_place::<std::boxed::Box<dyn system::system::System<In = (), Out = ()>>> - shim(Some(std::boxed::Box<dyn system::system::System<In = (), Out = ()>>))` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/core/src/ptr/mod.rs:490:1
     = note: inside `std::ptr::drop_in_place::<schedule::system_container::SystemContainer> - shim(Some(schedule::system_container::SystemContainer))` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/core/src/ptr/mod.rs:490:1
     = note: inside `std::ptr::drop_in_place::<[schedule::system_container::SystemContainer]> - shim(Some([schedule::system_container::SystemContainer]))` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/core/src/ptr/mod.rs:490:1
     = note: inside `<std::vec::Vec<schedule::system_container::SystemContainer> as std::ops::Drop>::drop` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/alloc/src/vec/mod.rs:[302](https://github.com/JoJoJet/bevy/actions/runs/3484882520/jobs/5829897858#step:6:303)9:13
     = note: inside `std::ptr::drop_in_place::<std::vec::Vec<schedule::system_container::SystemContainer>> - shim(Some(std::vec::Vec<schedule::system_container::SystemContainer>))` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/core/src/ptr/mod.rs:490:1
     = note: inside `std::ptr::drop_in_place::<schedule::stage::SystemStage> - shim(Some(schedule::stage::SystemStage))` at /home/runner/.rustup/toolchains/nightly-x86_64-unknown-linux-gnu/lib/rustlib/src/rust/library/core/src/ptr/mod.rs:490:1
note: inside `tests::ub_find` at crates/bevy_ecs/src/lib.rs:1225:5
    --> crates/bevy_ecs/src/lib.rs:1225:5
     |
1225 |     }
     |     ^
note: inside closure at crates/bevy_ecs/src/lib.rs:1201:18
    --> crates/bevy_ecs/src/lib.rs:1201:18
     |
1199 |     #[test]
     |     ------- in this procedural macro expansion
1200 |     #[should_panic]
1201 |     fn ub_find() {
     |                  ^
     = note: this error originates in the attribute macro `test` (in Nightly builds, run with -Z macro-backtrace for more info)

note: some details are omitted, run with `MIRIFLAGS=-Zmiri-backtrace=full` for a verbose backtrace

[hymm]

My next guess to why this is happening is that the Task abstraction just asks the future to cancel when it is dropped, but doesn't wait for the future to cancel. We might need to use FailableTask instead and explicitly wait for them to cancel when the scope is dropped.