Skip to content

Commit 77cd5cc

Browse files
committed
auto merge of #19548 : luqmana/rust/mfb, r=nikomatsakis
Fixes #19367.
2 parents 558f8d8 + 2dccb5a commit 77cd5cc

File tree

2 files changed

+85
-14
lines changed

2 files changed

+85
-14
lines changed

src/librustc_trans/trans/_match.rs

+43-14
Original file line numberDiff line numberDiff line change
@@ -1226,30 +1226,50 @@ pub fn trans_match<'blk, 'tcx>(bcx: Block<'blk, 'tcx>,
12261226

12271227
/// Checks whether the binding in `discr` is assigned to anywhere in the expression `body`
12281228
fn is_discr_reassigned(bcx: Block, discr: &ast::Expr, body: &ast::Expr) -> bool {
1229-
match discr.node {
1229+
let (vid, field) = match discr.node {
12301230
ast::ExprPath(..) => match bcx.def(discr.id) {
1231-
def::DefLocal(vid) | def::DefUpvar(vid, _, _) => {
1232-
let mut rc = ReassignmentChecker {
1233-
node: vid,
1234-
reassigned: false
1235-
};
1236-
{
1237-
let mut visitor = euv::ExprUseVisitor::new(&mut rc, bcx);
1238-
visitor.walk_expr(body);
1239-
}
1240-
rc.reassigned
1241-
}
1242-
_ => false
1231+
def::DefLocal(vid) | def::DefUpvar(vid, _, _) => (vid, None),
1232+
_ => return false
1233+
},
1234+
ast::ExprField(ref base, field) => {
1235+
let vid = match bcx.tcx().def_map.borrow().get(&base.id) {
1236+
Some(&def::DefLocal(vid)) | Some(&def::DefUpvar(vid, _, _)) => vid,
1237+
_ => return false
1238+
};
1239+
(vid, Some(mc::NamedField(field.node.name)))
1240+
},
1241+
ast::ExprTupField(ref base, field) => {
1242+
let vid = match bcx.tcx().def_map.borrow().get(&base.id) {
1243+
Some(&def::DefLocal(vid)) | Some(&def::DefUpvar(vid, _, _)) => vid,
1244+
_ => return false
1245+
};
1246+
(vid, Some(mc::PositionalField(field.node)))
12431247
},
1244-
_ => false
1248+
_ => return false
1249+
};
1250+
1251+
let mut rc = ReassignmentChecker {
1252+
node: vid,
1253+
field: field,
1254+
reassigned: false
1255+
};
1256+
{
1257+
let mut visitor = euv::ExprUseVisitor::new(&mut rc, bcx);
1258+
visitor.walk_expr(body);
12451259
}
1260+
rc.reassigned
12461261
}
12471262

12481263
struct ReassignmentChecker {
12491264
node: ast::NodeId,
1265+
field: Option<mc::FieldName>,
12501266
reassigned: bool
12511267
}
12521268

1269+
// Determine if the expression we're matching on is reassigned to within
1270+
// the body of the match's arm.
1271+
// We only care for the `mutate` callback since this check only matters
1272+
// for cases where the matched value is moved.
12531273
impl<'tcx> euv::Delegate<'tcx> for ReassignmentChecker {
12541274
fn consume(&mut self, _: ast::NodeId, _: Span, _: mc::cmt, _: euv::ConsumeMode) {}
12551275
fn matched_pat(&mut self, _: &ast::Pat, _: mc::cmt, _: euv::MatchMode) {}
@@ -1262,6 +1282,15 @@ impl<'tcx> euv::Delegate<'tcx> for ReassignmentChecker {
12621282
match cmt.cat {
12631283
mc::cat_upvar(mc::Upvar { id: ty::UpvarId { var_id: vid, .. }, .. }) |
12641284
mc::cat_local(vid) => self.reassigned = self.node == vid,
1285+
mc::cat_interior(ref base_cmt, mc::InteriorField(field)) => {
1286+
match base_cmt.cat {
1287+
mc::cat_upvar(mc::Upvar { id: ty::UpvarId { var_id: vid, .. }, .. }) |
1288+
mc::cat_local(vid) => {
1289+
self.reassigned = self.node == vid && Some(field) == self.field
1290+
},
1291+
_ => {}
1292+
}
1293+
},
12651294
_ => {}
12661295
}
12671296
}

src/test/run-pass/issue-19367.rs

+42
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,42 @@
1+
// Copyright 2014 The Rust Project Developers. See the COPYRIGHT
2+
// file at the top-level directory of this distribution and at
3+
// http://rust-lang.org/COPYRIGHT.
4+
//
5+
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6+
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7+
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8+
// option. This file may not be copied, modified, or distributed
9+
// except according to those terms.
10+
11+
#![feature(tuple_indexing)]
12+
struct S {
13+
o: Option<String>
14+
}
15+
16+
// Make sure we don't reuse the same alloca when matching
17+
// on field of struct or tuple which we reassign in the match body.
18+
19+
fn main() {
20+
let mut a = (0i, Some("right".into_string()));
21+
let b = match a.1 {
22+
Some(v) => {
23+
a.1 = Some("wrong".into_string());
24+
v
25+
}
26+
None => String::new()
27+
};
28+
println!("{}", b);
29+
assert_eq!(b, "right");
30+
31+
32+
let mut s = S{ o: Some("right".into_string()) };
33+
let b = match s.o {
34+
Some(v) => {
35+
s.o = Some("wrong".into_string());
36+
v
37+
}
38+
None => String::new(),
39+
};
40+
println!("{}", b);
41+
assert_eq!(b, "right");
42+
}

0 commit comments

Comments
 (0)