From 26186d7ff247b74287f21dfebcd08d0d5e0152aa Mon Sep 17 00:00:00 2001
From: Ben Kimock <kimockb@gmail.com>
Date: Fri, 14 Oct 2022 20:04:27 -0400
Subject: [PATCH] Add a test

---
 .../pass/stacked-borrows/stack-printing.rs    | 29 +++++++++++++++++++
 .../stacked-borrows/stack-printing.stdout     |  5 ++++
 2 files changed, 34 insertions(+)
 create mode 100644 src/tools/miri/tests/pass/stacked-borrows/stack-printing.rs
 create mode 100644 src/tools/miri/tests/pass/stacked-borrows/stack-printing.stdout

diff --git a/src/tools/miri/tests/pass/stacked-borrows/stack-printing.rs b/src/tools/miri/tests/pass/stacked-borrows/stack-printing.rs
new file mode 100644
index 000000000000..8d96a2e1ca99
--- /dev/null
+++ b/src/tools/miri/tests/pass/stacked-borrows/stack-printing.rs
@@ -0,0 +1,29 @@
+use std::{
+    alloc::{self, Layout},
+    mem::ManuallyDrop,
+};
+
+extern "Rust" {
+    fn miri_get_alloc_id(ptr: *const u8) -> u64;
+    fn miri_print_stacks(alloc_id: u64);
+}
+
+fn main() {
+    let ptr = unsafe { alloc::alloc(Layout::new::<u8>()) };
+    let alloc_id = unsafe { miri_get_alloc_id(ptr) };
+    unsafe { miri_print_stacks(alloc_id) };
+
+    assert!(!ptr.is_null());
+    unsafe { miri_print_stacks(alloc_id) };
+
+    unsafe { *ptr = 42 };
+    unsafe { miri_print_stacks(alloc_id) };
+
+    let _b = unsafe { ManuallyDrop::new(Box::from_raw(ptr)) };
+    unsafe { miri_print_stacks(alloc_id) };
+
+    let _ptr = unsafe { &*ptr };
+    unsafe { miri_print_stacks(alloc_id) };
+
+    unsafe { alloc::dealloc(ptr, Layout::new::<u8>()) };
+}
diff --git a/src/tools/miri/tests/pass/stacked-borrows/stack-printing.stdout b/src/tools/miri/tests/pass/stacked-borrows/stack-printing.stdout
new file mode 100644
index 000000000000..6d4f15453d27
--- /dev/null
+++ b/src/tools/miri/tests/pass/stacked-borrows/stack-printing.stdout
@@ -0,0 +1,5 @@
+0..1: [ SharedReadWrite<2794> ]
+0..1: [ SharedReadWrite<2794> ]
+0..1: [ SharedReadWrite<2794> ]
+0..1: [ SharedReadWrite<2794> Unique<2822> Unique<2823> Unique<2824> Unique<2828> Unique<2830> ]
+0..1: [ SharedReadWrite<2794> Disabled<2822> Disabled<2823> Disabled<2824> Disabled<2828> Disabled<2830> SharedReadOnly<2832> ]