about summary refs log tree commit diff
diff options
context:
space:
mode:
authorJohn Kåre Alsaker <john.kare.alsaker@gmail.com>2019-05-23 04:39:49 +0200
committerJohn Kåre Alsaker <john.kare.alsaker@gmail.com>2019-05-23 04:42:56 +0200
commitcb6beef46079183850d0a56895e630c6b5fb4ce5 (patch)
tree94e0ced24146078f308f19341fccbeec1824ba66
parent50a0defd5a93523067ef239936cc2e0755220904 (diff)
downloadrust-cb6beef46079183850d0a56895e630c6b5fb4ce5.tar.gz
rust-cb6beef46079183850d0a56895e630c6b5fb4ce5.zip
Make MIR drop terminators borrow the dropped location
-rw-r--r--src/librustc_mir/dataflow/impls/borrowed_locals.rs13
1 files changed, 12 insertions, 1 deletions
diff --git a/src/librustc_mir/dataflow/impls/borrowed_locals.rs b/src/librustc_mir/dataflow/impls/borrowed_locals.rs
index 42c2387b705..81e75d61132 100644
--- a/src/librustc_mir/dataflow/impls/borrowed_locals.rs
+++ b/src/librustc_mir/dataflow/impls/borrowed_locals.rs
@@ -56,9 +56,20 @@ impl<'a, 'tcx> BitDenotation<'tcx> for HaveBeenBorrowedLocals<'a, 'tcx> {
     fn terminator_effect(&self,
                          sets: &mut BlockSets<'_, Local>,
                          loc: Location) {
+        let terminator = self.mir[loc.block].terminator();
         BorrowedLocalsVisitor {
             sets,
-        }.visit_terminator(self.mir[loc.block].terminator(), loc);
+        }.visit_terminator(terminator, loc);
+        match &terminator.kind {
+            // Drop terminators borrows the location
+            TerminatorKind::Drop { location, .. } |
+            TerminatorKind::DropAndReplace { location, .. } => {
+                if let Some(local) = find_local(location) {
+                    sets.gen(local);
+                }
+            }
+            _ => (),
+        }
     }
 
     fn propagate_call_return(