1 //! This pass transforms derefs of Box into a deref of the pointer inside Box.
3 //! Box is not actually a pointer so it is incorrect to dereference it directly.
6 use rustc_hir::def_id::DefId;
7 use rustc_index::vec::Idx;
8 use rustc_middle::mir::patch::MirPatch;
9 use rustc_middle::mir::visit::MutVisitor;
10 use rustc_middle::mir::*;
11 use rustc_middle::ty::subst::Subst;
12 use rustc_middle::ty::{Ty, TyCtxt};
14 /// Constructs the types used when accessing a Box's pointer
15 pub fn build_ptr_tys<'tcx>(
20 ) -> (Ty<'tcx>, Ty<'tcx>, Ty<'tcx>) {
21 let substs = tcx.intern_substs(&[pointee.into()]);
22 let unique_ty = tcx.bound_type_of(unique_did).subst(tcx, substs);
23 let nonnull_ty = tcx.bound_type_of(nonnull_did).subst(tcx, substs);
24 let ptr_ty = tcx.mk_imm_ptr(pointee);
26 (unique_ty, nonnull_ty, ptr_ty)
29 // Constructs the projection needed to access a Box's pointer
30 pub fn build_projection<'tcx>(
34 ) -> [PlaceElem<'tcx>; 3] {
36 PlaceElem::Field(Field::new(0), unique_ty),
37 PlaceElem::Field(Field::new(0), nonnull_ty),
38 PlaceElem::Field(Field::new(0), ptr_ty),
42 struct ElaborateBoxDerefVisitor<'tcx, 'a> {
46 local_decls: &'a mut LocalDecls<'tcx>,
47 patch: MirPatch<'tcx>,
50 impl<'tcx, 'a> MutVisitor<'tcx> for ElaborateBoxDerefVisitor<'tcx, 'a> {
51 fn tcx(&self) -> TyCtxt<'tcx> {
57 place: &mut Place<'tcx>,
58 context: visit::PlaceContext,
63 let base_ty = self.local_decls[place.local].ty;
65 // Derefer ensures that derefs are always the first projection
66 if place.projection.first() == Some(&PlaceElem::Deref) && base_ty.is_box() {
67 let source_info = self.local_decls[place.local].source_info;
69 let (unique_ty, nonnull_ty, ptr_ty) =
70 build_ptr_tys(tcx, base_ty.boxed_ty(), self.unique_did, self.nonnull_did);
72 let ptr_local = self.patch.new_internal(ptr_ty, source_info.span);
74 self.patch.add_assign(
76 Place::from(ptr_local),
77 Rvalue::Use(Operand::Copy(
78 Place::from(place.local)
79 .project_deeper(&build_projection(unique_ty, nonnull_ty, ptr_ty), tcx),
83 place.local = ptr_local;
86 self.super_place(place, context, location);
90 pub struct ElaborateBoxDerefs;
92 impl<'tcx> MirPass<'tcx> for ElaborateBoxDerefs {
93 fn run_pass(&self, tcx: TyCtxt<'tcx>, body: &mut Body<'tcx>) {
94 if let Some(def_id) = tcx.lang_items().owned_box() {
95 let unique_did = tcx.adt_def(def_id).non_enum_variant().fields[0].did;
97 let Some(nonnull_def) = tcx.type_of(unique_did).ty_adt_def() else {
98 span_bug!(tcx.def_span(unique_did), "expected Box to contain Unique")
101 let nonnull_did = nonnull_def.non_enum_variant().fields[0].did;
103 let patch = MirPatch::new(body);
105 let local_decls = &mut body.local_decls;
108 ElaborateBoxDerefVisitor { tcx, unique_did, nonnull_did, local_decls, patch };
110 for (block, data) in body.basic_blocks.as_mut_preserves_cfg().iter_enumerated_mut() {
111 visitor.visit_basic_block_data(block, data);
114 visitor.patch.apply(body);
116 for debug_info in body.var_debug_info.iter_mut() {
117 if let VarDebugInfoContents::Place(place) = &mut debug_info.value {
118 let mut new_projections: Option<Vec<_>> = None;
119 let mut last_deref = 0;
121 for (i, (base, elem)) in place.iter_projections().enumerate() {
122 let base_ty = base.ty(&body.local_decls, tcx).ty;
124 if elem == PlaceElem::Deref && base_ty.is_box() {
125 let new_projections = new_projections.get_or_insert_default();
127 let (unique_ty, nonnull_ty, ptr_ty) =
128 build_ptr_tys(tcx, base_ty.boxed_ty(), unique_did, nonnull_did);
130 new_projections.extend_from_slice(&base.projection[last_deref..]);
131 new_projections.extend_from_slice(&build_projection(
132 unique_ty, nonnull_ty, ptr_ty,
134 new_projections.push(PlaceElem::Deref);
140 if let Some(mut new_projections) = new_projections {
141 new_projections.extend_from_slice(&place.projection[last_deref..]);
142 place.projection = tcx.intern_place_elems(&new_projections);
147 // box is not present, this pass doesn't need to do anything