rustc_mir: promote borrows' underlying temps, and project at runtime.

rust-lang · May 16, 2018 · 4fec5ef · 4fec5ef
1 parent d79dee0
commit 4fec5ef
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/librustc_mir/transform/promote_consts.rs b/src/librustc_mir/transform/promote_consts.rs
@@ -309,6 +309,18 @@ impl<'a, 'tcx> Promoter<'a, 'tcx> {
                     let ref mut statement = blocks[loc.block].statements[loc.statement_index];
                     match statement.kind {
                         StatementKind::Assign(_, Rvalue::Ref(r, bk, ref mut place)) => {
+                            // Find the underlying local for this (necessarilly interior) borrow.
+                            // HACK(eddyb) using a recursive function because of mutable borrows.
+                            fn interior_base<'a, 'tcx>(place: &'a mut Place<'tcx>)
+                                                       -> &'a mut Place<'tcx> {
+                                if let Place::Projection(ref mut proj) = *place {
+                                    assert_ne!(proj.elem, ProjectionElem::Deref);
+                                    return interior_base(&mut proj.base);
+                                }
+                                place
+                            }
+                            let place = interior_base(place);
+
                             let ty = place.ty(local_decls, self.tcx).to_ty(self.tcx);
                             let ref_ty = self.tcx.mk_ref(r,
                                 ty::TypeAndMut {