Skip to content

Commit 41053bf

Browse files
lengyijunlengyijun
committed
fix 5707
1 parent f1f5ccd commit 41053bf

File tree

4 files changed

+150
-16
lines changed

4 files changed

+150
-16
lines changed

‎clippy_lints/src/redundant_clone.rs

Lines changed: 106 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ use rustc_lint::{LateContext, LateLintPass};
1212
use rustc_middle::mir::{
1313
self, traversal,
1414
visit::{MutatingUseContext, NonMutatingUseContext, PlaceContext, Visitor as _},
15+
Mutability,
1516
};
1617
use rustc_middle::ty::{self, fold::TypeVisitor, Ty};
1718
use rustc_mir::dataflow::{Analysis, AnalysisDomain, GenKill, GenKillAnalysis, ResultsCursor};
@@ -87,13 +88,18 @@ impl<'tcx> LateLintPass<'tcx> for RedundantClone {
8788

8889
let mir = cx.tcx.optimized_mir(def_id.to_def_id());
8990

91+
let possible_borrowed = {
92+
let mut vis = PossibleBorrowedVisitor::new(mir);
93+
vis.visit_body(mir);
94+
vis.into_map(cx)
95+
};
9096
let maybe_storage_live_result = MaybeStorageLive
9197
.into_engine(cx.tcx, mir)
9298
.pass_name("redundant_clone")
9399
.iterate_to_fixpoint()
94100
.into_results_cursor(mir);
95101
let mut possible_borrower = {
96-
let mut vis = PossibleBorrowerVisitor::new(cx, mir);
102+
let mut vis = PossibleBorrowerVisitor::new(cx, mir, possible_borrowed);
97103
vis.visit_body(mir);
98104
vis.into_map(cx, maybe_storage_live_result)
99105
};
@@ -509,14 +515,20 @@ struct PossibleBorrowerVisitor<'a, 'tcx> {
509515
possible_borrower: TransitiveRelation<mir::Local>,
510516
body: &'a mir::Body<'tcx>,
511517
cx: &'a LateContext<'tcx>,
518+
possible_borrowed: FxHashMap<mir::Local, HybridBitSet<mir::Local>>,
512519
}
513520

514521
impl<'a, 'tcx> PossibleBorrowerVisitor<'a, 'tcx> {
515-
fn new(cx: &'a LateContext<'tcx>, body: &'a mir::Body<'tcx>) -> Self {
522+
fn new(
523+
cx: &'a LateContext<'tcx>,
524+
body: &'a mir::Body<'tcx>,
525+
possible_borrowed: FxHashMap<mir::Local, HybridBitSet<mir::Local>>,
526+
) -> Self {
516527
Self {
517528
possible_borrower: TransitiveRelation::default(),
518529
cx,
519530
body,
531+
possible_borrowed,
520532
}
521533
}
522534

@@ -585,21 +597,109 @@ impl<'a, 'tcx> mir::visit::Visitor<'tcx> for PossibleBorrowerVisitor<'a, 'tcx> {
585597
..
586598
} = &terminator.kind
587599
{
600+
// TODO add doc
588601
// If the call returns something with lifetimes,
589602
// let's conservatively assume the returned value contains lifetime of all the arguments.
590603
// For example, given `let y: Foo<'a> = foo(x)`, `y` is considered to be a possible borrower of `x`.
591-
if ContainsRegion.visit_ty(self.body.local_decls[*dest].ty).is_continue() {
592-
return;
593-
}
604+
605+
let mut immutable_borrowers = vec![];
606+
let mut mutable_borrowers = vec![];
594607

595608
for op in args {
596609
match op {
597610
mir::Operand::Copy(p) | mir::Operand::Move(p) => {
598-
self.possible_borrower.add(p.local, *dest);
611+
if let ty::Ref(_, _, Mutability::Mut) = self.body.local_decls[p.local].ty.kind() {
612+
mutable_borrowers.push(p.local);
613+
} else {
614+
immutable_borrowers.push(p.local);
615+
}
599616
},
600617
mir::Operand::Constant(..) => (),
601618
}
602619
}
620+
621+
let mut mutable_variables: Vec<mir::Local> = mutable_borrowers
622+
.iter()
623+
// TODO why possible_borrowed not found?
624+
.filter_map(|r| self.possible_borrowed.get(r))
625+
.flat_map(|r| r.iter())
626+
.collect();
627+
628+
if ContainsRegion.visit_ty(self.body.local_decls[*dest].ty).is_break() {
629+
mutable_variables.push(*dest);
630+
}
631+
632+
for y in mutable_variables {
633+
for x in &immutable_borrowers {
634+
self.possible_borrower.add(*x, y);
635+
}
636+
for x in &mutable_borrowers {
637+
self.possible_borrower.add(*x, y);
638+
}
639+
}
640+
}
641+
}
642+
}
643+
644+
/// Collect possible borrowed for every `&mut` local.
645+
/// For exampel, `_1 = &mut _2` generate _1: {_2,...}
646+
/// Known Problems: not sure all borrowed are tracked
647+
struct PossibleBorrowedVisitor<'a, 'tcx> {
648+
possible_borrowed: TransitiveRelation<mir::Local>,
649+
body: &'a mir::Body<'tcx>,
650+
}
651+
652+
impl<'a, 'tcx> PossibleBorrowedVisitor<'a, 'tcx> {
653+
fn new(body: &'a mir::Body<'tcx>) -> Self {
654+
Self {
655+
possible_borrowed: TransitiveRelation::default(),
656+
body,
657+
}
658+
}
659+
660+
fn into_map(self, cx: &LateContext<'tcx>) -> FxHashMap<mir::Local, HybridBitSet<mir::Local>> {
661+
let mut map = FxHashMap::default();
662+
for row in (1..self.body.local_decls.len()).map(mir::Local::from_usize) {
663+
if is_copy(cx, self.body.local_decls[row].ty) {
664+
continue;
665+
}
666+
667+
let borrowers = self.possible_borrowed.reachable_from(&row);
668+
if !borrowers.is_empty() {
669+
let mut bs = HybridBitSet::new_empty(self.body.local_decls.len());
670+
for &c in borrowers {
671+
if c != mir::Local::from_usize(0) {
672+
bs.insert(c);
673+
}
674+
}
675+
676+
if !bs.is_empty() {
677+
map.insert(row, bs);
678+
}
679+
}
680+
}
681+
map
682+
}
683+
}
684+
685+
impl<'a, 'tcx> mir::visit::Visitor<'tcx> for PossibleBorrowedVisitor<'a, 'tcx> {
686+
fn visit_assign(&mut self, place: &mir::Place<'tcx>, rvalue: &mir::Rvalue<'_>, _location: mir::Location) {
687+
let lhs = place.local;
688+
match rvalue {
689+
// Only consider `&mut`, which can modify origin place
690+
mir::Rvalue::Ref(_, rustc_middle::mir::BorrowKind::Mut { .. }, borrowed) => {
691+
self.possible_borrowed.add(lhs, borrowed.local);
692+
},
693+
// _2: &mut _;
694+
// _3 = move _2
695+
mir::Rvalue::Use(mir::Operand::Move(borrowed))=>{
696+
self.possible_borrowed.add(lhs, borrowed.local);
697+
}
698+
// _3 = move _2 as &mut _;
699+
mir::Rvalue::Cast(_,mir::Operand::Move(borrowed),_)=>{
700+
self.possible_borrowed.add(lhs, borrowed.local);
701+
}
702+
_ => {},
603703
}
604704
}
605705
}

‎tests/ui/redundant_clone.fixed

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ fn main() {
5555
issue_5405();
5656
manually_drop();
5757
clone_then_move_cloned();
58+
hashmap_neg();
5859
}
5960

6061
#[derive(Clone)]
@@ -206,3 +207,19 @@ fn clone_then_move_cloned() {
206207
let mut x = S(String::new());
207208
x.0.clone().chars().for_each(|_| x.m());
208209
}
210+
211+
fn hashmap_neg() {
212+
// issue 5707
213+
use std::path::PathBuf;
214+
use std::collections::HashMap;
215+
216+
let p = PathBuf::from("/");
217+
218+
let mut h: HashMap<&str, &str> = HashMap::new();
219+
h.insert("orig-p", p.to_str().unwrap());
220+
221+
let mut q = p.clone();
222+
q.push("foo");
223+
224+
println!("{:?} {}", h, q.display());
225+
}

‎tests/ui/redundant_clone.rs

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ fn main() {
5555
issue_5405();
5656
manually_drop();
5757
clone_then_move_cloned();
58+
hashmap_neg();
5859
}
5960

6061
#[derive(Clone)]
@@ -206,3 +207,19 @@ fn clone_then_move_cloned() {
206207
let mut x = S(String::new());
207208
x.0.clone().chars().for_each(|_| x.m());
208209
}
210+
211+
fn hashmap_neg() {
212+
// issue 5707
213+
use std::path::PathBuf;
214+
use std::collections::HashMap;
215+
216+
let p = PathBuf::from("/");
217+
218+
let mut h: HashMap<&str, &str> = HashMap::new();
219+
h.insert("orig-p", p.to_str().unwrap());
220+
221+
let mut q = p.clone();
222+
q.push("foo");
223+
224+
println!("{:?} {}", h, q.display());
225+
}

‎tests/ui/redundant_clone.stderr

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -108,61 +108,61 @@ LL | let _t = tup.0.clone();
108108
| ^^^^^
109109

110110
error: redundant clone
111-
--> $DIR/redundant_clone.rs:63:25
111+
--> $DIR/redundant_clone.rs:64:25
112112
|
113113
LL | if b { (a.clone(), a.clone()) } else { (Alpha, a) }
114114
| ^^^^^^^^ help: remove this
115115
|
116116
note: this value is dropped without further use
117-
--> $DIR/redundant_clone.rs:63:24
117+
--> $DIR/redundant_clone.rs:64:24
118118
|
119119
LL | if b { (a.clone(), a.clone()) } else { (Alpha, a) }
120120
| ^
121121

122122
error: redundant clone
123-
--> $DIR/redundant_clone.rs:120:15
123+
--> $DIR/redundant_clone.rs:121:15
124124
|
125125
LL | let _s = s.clone();
126126
| ^^^^^^^^ help: remove this
127127
|
128128
note: this value is dropped without further use
129-
--> $DIR/redundant_clone.rs:120:14
129+
--> $DIR/redundant_clone.rs:121:14
130130
|
131131
LL | let _s = s.clone();
132132
| ^
133133

134134
error: redundant clone
135-
--> $DIR/redundant_clone.rs:121:15
135+
--> $DIR/redundant_clone.rs:122:15
136136
|
137137
LL | let _t = t.clone();
138138
| ^^^^^^^^ help: remove this
139139
|
140140
note: this value is dropped without further use
141-
--> $DIR/redundant_clone.rs:121:14
141+
--> $DIR/redundant_clone.rs:122:14
142142
|
143143
LL | let _t = t.clone();
144144
| ^
145145

146146
error: redundant clone
147-
--> $DIR/redundant_clone.rs:131:19
147+
--> $DIR/redundant_clone.rs:132:19
148148
|
149149
LL | let _f = f.clone();
150150
| ^^^^^^^^ help: remove this
151151
|
152152
note: this value is dropped without further use
153-
--> $DIR/redundant_clone.rs:131:18
153+
--> $DIR/redundant_clone.rs:132:18
154154
|
155155
LL | let _f = f.clone();
156156
| ^
157157

158158
error: redundant clone
159-
--> $DIR/redundant_clone.rs:143:14
159+
--> $DIR/redundant_clone.rs:144:14
160160
|
161161
LL | let y = x.clone().join("matthias");
162162
| ^^^^^^^^ help: remove this
163163
|
164164
note: cloned value is neither consumed nor mutated
165-
--> $DIR/redundant_clone.rs:143:13
165+
--> $DIR/redundant_clone.rs:144:13
166166
|
167167
LL | let y = x.clone().join("matthias");
168168
| ^^^^^^^^^

0 commit comments

Comments
 (0)