Skip to content

Commit

Permalink
always record reference to binding in match if guards
Browse files Browse the repository at this point in the history
  • Loading branch information
SNCPlay42 committed Oct 26, 2020
1 parent 69e68cf commit 8bf9abb
Show file tree
Hide file tree
Showing 2 changed files with 32 additions and 1 deletion.
12 changes: 12 additions & 0 deletions compiler/rustc_typeck/src/check/generator_interior.rs
Expand Up @@ -344,6 +344,18 @@ impl<'a, 'tcx> Visitor<'tcx> for InteriorVisitor<'a, 'tcx> {
// The type table might not have information for this expression
// if it is in a malformed scope. (#66387)
if let Some(ty) = self.fcx.typeck_results.borrow().expr_ty_opt(expr) {
if guard_borrowing_from_pattern {
// Match guards create references to all the bindings in the pattern that are used
// in the guard, e.g. `y if is_even(y) => ...` becomes `is_even(*r_y)` where `r_y`
// is a reference to `y`, so we must record a reference to the type of the binding.
let tcx = self.fcx.tcx;
let ref_ty = tcx.mk_ref(
// Use `ReErased` as `resolve_interior` is going to replace all the regions anyway.
tcx.mk_region(ty::RegionKind::ReErased),
ty::TypeAndMut { ty, mutbl: hir::Mutability::Not },
);
self.record(ref_ty, scope, Some(expr), expr.span, guard_borrowing_from_pattern);
}
self.record(ty, scope, Some(expr), expr.span, guard_borrowing_from_pattern);
} else {
self.fcx.tcx.sess.delay_span_bug(expr.span, "no type for node");
Expand Down
21 changes: 20 additions & 1 deletion src/test/ui/generator/yielding-in-match-guards.rs
@@ -1,4 +1,4 @@
// check-pass
// build-pass
// edition:2018

// This test is derived from
Expand All @@ -11,6 +11,7 @@
// of the underlying generator.

async fn f() -> u8 { 1 }
async fn foo() -> [bool; 10] { [false; 10] }

pub async fn g(x: u8) {
match x {
Expand All @@ -19,6 +20,24 @@ pub async fn g(x: u8) {
}
}

// #78366: check the reference to the binding is recorded even if the binding is not autorefed

async fn h(x: usize) {
match x {
y if foo().await[y] => (),
_ => (),
}
}

async fn i(x: u8) {
match x {
y if f().await == y + 1 => (),
_ => (),
}
}

fn main() {
let _ = g(10);
let _ = h(9);
let _ = i(8);
}

0 comments on commit 8bf9abb

Please sign in to comment.