Stop doing weird index stuff in ElaborateBoxDerefs

This commit is contained in:
Michael Goulet 2024-08-02 17:28:05 -04:00
parent f82eb4d0a0
commit 2e52d61807
4 changed files with 77 additions and 8 deletions

View File

@ -116,29 +116,30 @@ impl<'tcx> MirPass<'tcx> for ElaborateBoxDerefs {
for debug_info in body.var_debug_info.iter_mut() {
if let VarDebugInfoContents::Place(place) = &mut debug_info.value {
let mut new_projections: Option<Vec<_>> = None;
let mut last_deref = 0;
for (i, (base, elem)) in place.iter_projections().enumerate() {
for (base, elem) in place.iter_projections() {
let base_ty = base.ty(&body.local_decls, tcx).ty;
if elem == PlaceElem::Deref && base_ty.is_box() {
let new_projections = new_projections.get_or_insert_default();
// Clone the projections before us, since now we need to mutate them.
let new_projections =
new_projections.get_or_insert_with(|| base.projection.to_vec());
let (unique_ty, nonnull_ty, ptr_ty) =
build_ptr_tys(tcx, base_ty.boxed_ty(), unique_did, nonnull_did);
new_projections.extend_from_slice(&base.projection[last_deref..]);
new_projections.extend_from_slice(&build_projection(
unique_ty, nonnull_ty, ptr_ty,
));
new_projections.push(PlaceElem::Deref);
last_deref = i;
} else if let Some(new_projections) = new_projections.as_mut() {
// Keep building up our projections list once we've started it.
new_projections.push(elem);
}
}
if let Some(mut new_projections) = new_projections {
new_projections.extend_from_slice(&place.projection[last_deref..]);
// Store the mutated projections if we actually changed something.
if let Some(new_projections) = new_projections {
place.projection = tcx.mk_place_elems(&new_projections);
}
}

View File

@ -0,0 +1,13 @@
- // MIR for `pointee` before ElaborateBoxDerefs
+ // MIR for `pointee` after ElaborateBoxDerefs
fn pointee(_1: Box<i32>) -> () {
- debug foo => (*_1);
+ debug foo => (*(((_1.0: std::ptr::Unique<i32>).0: std::ptr::NonNull<i32>).0: *const i32));
let mut _0: ();
bb0: {
return;
}
}

View File

@ -0,0 +1,20 @@
// skip-filecheck
//@ test-mir-pass: ElaborateBoxDerefs
#![feature(custom_mir, core_intrinsics)]
extern crate core;
use core::intrinsics::mir::*;
// EMIT_MIR elaborate_box_deref_in_debuginfo.pointee.ElaborateBoxDerefs.diff
#[custom_mir(dialect = "built")]
fn pointee(opt: Box<i32>) {
mir!(
debug foo => *opt;
{
Return()
}
)
}
fn main() {}

View File

@ -0,0 +1,35 @@
//@ aux-build:block-on.rs
//@ edition:2021
//@ run-pass
#![feature(async_closure)]
extern crate block_on;
pub trait Trait {
fn callback(&mut self);
}
impl Trait for (i32,) {
fn callback(&mut self) {
println!("hi {}", self.0);
self.0 += 1;
}
}
async fn call_once(f: impl async FnOnce()) {
f().await;
}
async fn run(mut loader: Box<dyn Trait>) {
let f = async move || {
loader.callback();
loader.callback();
};
call_once(f).await;
}
fn main() {
block_on::block_on(async {
run(Box::new((42,))).await;
});
}