]> git.lizzy.rs Git - rust.git/commitdiff
rustc_mir: also promote interior borrows, not just whole temps.
authorEduard-Mihai Burtescu <edy.burt@gmail.com>
Thu, 10 May 2018 09:06:51 +0000 (12:06 +0300)
committerEduard-Mihai Burtescu <edy.burt@gmail.com>
Wed, 16 May 2018 11:11:01 +0000 (14:11 +0300)
src/librustc_mir/transform/qualify_consts.rs
src/test/run-pass/issue-49955.rs [new file with mode: 0644]

index 28ab3d6a8574fb3543cf27bc40e15f42192cb97d..544f122f25ae33ef1a4d1f7c96f65f2efe82392c 100644 (file)
@@ -737,10 +737,21 @@ fn visit_rvalue(&mut self, rvalue: &Rvalue<'tcx>, location: Location) {
                 // We might have a candidate for promotion.
                 let candidate = Candidate::Ref(location);
                 if self.can_promote() {
-                    // We can only promote direct borrows of temps.
+                    // We can only promote interior borrows of non-drop temps.
+                    let mut place = place;
+                    while let Place::Projection(ref proj) = *place {
+                        if proj.elem == ProjectionElem::Deref {
+                            break;
+                        }
+                        place = &proj.base;
+                    }
                     if let Place::Local(local) = *place {
                         if self.mir.local_kind(local) == LocalKind::Temp {
-                            self.promotion_candidates.push(candidate);
+                            if let Some(qualif) = self.temp_qualif[local] {
+                                if !qualif.intersects(Qualif::NEEDS_DROP) {
+                                    self.promotion_candidates.push(candidate);
+                                }
+                            }
                         }
                     }
                 }
diff --git a/src/test/run-pass/issue-49955.rs b/src/test/run-pass/issue-49955.rs
new file mode 100644 (file)
index 0000000..2d36806
--- /dev/null
@@ -0,0 +1,30 @@
+// Copyright 2018 The Rust Project Developers. See the COPYRIGHT
+// file at the top-level directory of this distribution and at
+// http://rust-lang.org/COPYRIGHT.
+//
+// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
+// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
+// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
+// option. This file may not be copied, modified, or distributed
+// except according to those terms.
+
+// compile-flags: -Z borrowck=compare
+
+const ALL_THE_NUMS: [u32; 1] = [
+    1
+];
+
+#[inline(never)]
+fn array(i: usize) -> &'static u32 {
+    return &ALL_THE_NUMS[i];
+}
+
+#[inline(never)]
+fn tuple_field() -> &'static u32 {
+    &(42,).0
+}
+
+fn main() {
+    assert_eq!(tuple_field().to_string(), "42");
+    // assert_eq!(array(0).to_string(), "1");
+}