2016-02-07 17:10:03 +00:00
use reexport ::* ;
2016-04-07 15:46:48 +00:00
use rustc ::hir ::* ;
use rustc ::hir ::def ::Def ;
use rustc ::hir ::intravisit ::{ Visitor , walk_expr , walk_block , walk_decl } ;
use rustc ::hir ::map ::Node ::NodeBlock ;
2015-08-12 19:56:27 +00:00
use rustc ::lint ::* ;
2016-03-31 15:05:43 +00:00
use rustc ::middle ::const_val ::ConstVal ;
2016-02-13 21:09:17 +00:00
use rustc ::middle ::region ::CodeExtent ;
2016-03-27 18:59:02 +00:00
use rustc ::ty ;
2016-03-31 15:05:43 +00:00
use rustc_const_eval ::EvalHint ::ExprTypeChecked ;
use rustc_const_eval ::eval_const_expr_partial ;
2016-02-14 11:07:56 +00:00
use std ::collections ::HashMap ;
2016-03-07 15:55:12 +00:00
use syntax ::ast ;
2016-07-01 17:31:14 +00:00
use utils ::sugg ;
2015-08-12 19:56:27 +00:00
2016-07-01 16:44:59 +00:00
use utils ::{ snippet , span_lint , get_parent_expr , match_trait_method , match_type , multispan_sugg , in_external_macro ,
2016-06-29 21:16:44 +00:00
span_help_and_lint , is_integer_literal , get_enclosing_block , span_lint_and_then , higher ,
2016-06-29 22:08:43 +00:00
walk_ptrs_ty } ;
2016-04-14 16:13:15 +00:00
use utils ::paths ;
2015-08-12 19:56:27 +00:00
2016-02-05 23:41:54 +00:00
/// **What it does:** This lint checks for looping over the range of `0..len` of some collection just to get the values by index.
2015-12-11 00:22:27 +00:00
///
/// **Why is this bad?** Just iterating the collection itself makes the intent more clear and is probably faster.
///
/// **Known problems:** None
///
/// **Example:**
/// ```
/// for i in 0..vec.len() {
/// println!("{}", vec[i]);
/// }
/// ```
2016-02-05 23:13:29 +00:00
declare_lint! {
pub NEEDLESS_RANGE_LOOP ,
Warn ,
" for-looping over a range of indices where an iterator over items would do "
}
2015-08-12 19:56:27 +00:00
2016-02-05 23:41:54 +00:00
/// **What it does:** This lint checks for loops on `x.iter()` where `&x` will do, and suggest the latter.
2015-12-11 00:22:27 +00:00
///
/// **Why is this bad?** Readability.
///
/// **Known problems:** False negatives. We currently only warn on some known types.
///
/// **Example:** `for x in y.iter() { .. }` (where y is a `Vec` or slice)
2016-02-05 23:13:29 +00:00
declare_lint! {
pub EXPLICIT_ITER_LOOP ,
Warn ,
" for-looping over `_.iter()` or `_.iter_mut()` when `&_` or `&mut _` would do "
}
2015-08-13 13:36:31 +00:00
2016-02-05 23:41:54 +00:00
/// **What it does:** This lint checks for loops on `x.next()`.
2015-12-11 00:22:27 +00:00
///
/// **Why is this bad?** `next()` returns either `Some(value)` if there was a value, or `None` otherwise. The insidious thing is that `Option<_>` implements `IntoIterator`, so that possibly one value will be iterated, leading to some hard to find bugs. No one will want to write such code [except to win an Underhanded Rust Contest](https://www.reddit.com/r/rust/comments/3hb0wm/underhanded_rust_contest/cu5yuhr).
///
/// **Known problems:** None
///
/// **Example:** `for x in y.next() { .. }`
2016-02-05 23:13:29 +00:00
declare_lint! {
pub ITER_NEXT_LOOP ,
Warn ,
" for-looping over `_.next()` which is probably not intended "
}
2015-08-17 05:23:57 +00:00
2016-02-05 23:41:54 +00:00
/// **What it does:** This lint checks for `for` loops over `Option` values.
2016-01-29 07:34:09 +00:00
///
/// **Why is this bad?** Readability. This is more clearly expressed as an `if let`.
///
/// **Known problems:** None
///
/// **Example:** `for x in option { .. }`. This should be `if let Some(x) = option { .. }`.
2016-02-05 23:13:29 +00:00
declare_lint! {
pub FOR_LOOP_OVER_OPTION ,
Warn ,
" for-looping over an `Option`, which is more clearly expressed as an `if let` "
}
2016-01-29 23:15:57 +00:00
2016-02-05 23:41:54 +00:00
/// **What it does:** This lint checks for `for` loops over `Result` values.
2016-01-29 23:15:57 +00:00
///
/// **Why is this bad?** Readability. This is more clearly expressed as an `if let`.
///
/// **Known problems:** None
///
/// **Example:** `for x in result { .. }`. This should be `if let Ok(x) = result { .. }`.
2016-02-05 23:13:29 +00:00
declare_lint! {
pub FOR_LOOP_OVER_RESULT ,
Warn ,
" for-looping over a `Result`, which is more clearly expressed as an `if let` "
}
2016-01-29 07:34:09 +00:00
2016-02-05 23:41:54 +00:00
/// **What it does:** This lint detects `loop + match` combinations that are easier written as a `while let` loop.
2015-12-11 00:22:27 +00:00
///
/// **Why is this bad?** The `while let` loop is usually shorter and more readable
///
/// **Known problems:** Sometimes the wrong binding is displayed (#383)
///
/// **Example:**
///
/// ```
/// loop {
/// let x = match y {
/// Some(x) => x,
/// None => break,
/// }
/// // .. do something with x
/// }
/// // is easier written as
/// while let Some(x) = y {
/// // .. do something with x
/// }
/// ```
2016-02-05 23:13:29 +00:00
declare_lint! {
pub WHILE_LET_LOOP ,
Warn ,
" `loop { if let { ... } else break }` can be written as a `while let` loop "
}
2015-08-29 09:41:06 +00:00
2016-02-05 23:41:54 +00:00
/// **What it does:** This lint checks for using `collect()` on an iterator without using the result.
2015-12-11 00:22:27 +00:00
///
/// **Why is this bad?** It is more idiomatic to use a `for` loop over the iterator instead.
///
/// **Known problems:** None
///
/// **Example:** `vec.iter().map(|x| /* some operation returning () */).collect::<Vec<_>>();`
2016-02-05 23:13:29 +00:00
declare_lint! {
pub UNUSED_COLLECT ,
Warn ,
" `collect()`ing an iterator without using the result; this is usually better \
written as a for loop "
}
2015-08-30 11:10:59 +00:00
2016-02-05 23:41:54 +00:00
/// **What it does:** This lint checks for loops over ranges `x..y` where both `x` and `y` are constant and `x` is greater or equal to `y`, unless the range is reversed or has a negative `.step_by(_)`.
2015-12-11 00:22:27 +00:00
///
/// **Why is it bad?** Such loops will either be skipped or loop until wrap-around (in debug code, this may `panic!()`). Both options are probably not intended.
///
/// **Known problems:** The lint cannot catch loops over dynamically defined ranges. Doing this would require simulating all possible inputs and code paths through the program, which would be complex and error-prone.
///
/// **Examples**: `for x in 5..10-5 { .. }` (oops, stray `-`)
2016-02-05 23:13:29 +00:00
declare_lint! {
pub REVERSE_RANGE_LOOP ,
Warn ,
" Iterating over an empty range, such as `10..0` or `5..5` "
}
2015-09-15 00:19:05 +00:00
2016-02-05 23:41:54 +00:00
/// **What it does:** This lint checks `for` loops over slices with an explicit counter and suggests the use of `.enumerate()`.
2015-12-11 00:22:27 +00:00
///
/// **Why is it bad?** Not only is the version using `.enumerate()` more readable, the compiler is able to remove bounds checks which can lead to faster code in some instances.
///
/// **Known problems:** None.
///
/// **Example:** `for i in 0..v.len() { foo(v[i]); }` or `for i in 0..v.len() { bar(i, v[i]); }`
2016-02-05 23:13:29 +00:00
declare_lint! {
pub EXPLICIT_COUNTER_LOOP ,
Warn ,
" for-looping with an explicit counter when `_.enumerate()` would do "
}
2015-08-23 17:25:45 +00:00
2016-02-05 23:41:54 +00:00
/// **What it does:** This lint checks for empty `loop` expressions.
2015-12-11 00:22:27 +00:00
///
/// **Why is this bad?** Those busy loops burn CPU cycles without doing anything. Think of the environment and either block on something or at least make the thread sleep for some microseconds.
///
/// **Known problems:** None
///
/// **Example:** `loop {}`
2016-02-05 23:13:29 +00:00
declare_lint! {
pub EMPTY_LOOP ,
Warn ,
" empty `loop {}` detected "
}
2015-10-12 11:38:18 +00:00
2016-02-05 23:41:54 +00:00
/// **What it does:** This lint checks for `while let` expressions on iterators.
2015-12-14 21:16:56 +00:00
///
/// **Why is this bad?** Readability. A simple `for` loop is shorter and conveys the intent better.
///
/// **Known problems:** None
///
/// **Example:** `while let Some(val) = iter() { .. }`
2016-02-05 23:13:29 +00:00
declare_lint! {
pub WHILE_LET_ON_ITERATOR ,
Warn ,
" using a while-let loop instead of a for loop on an iterator "
}
2015-10-16 18:27:13 +00:00
2016-01-19 20:10:00 +00:00
/// **What it does:** This warns when you iterate on a map (`HashMap` or `BTreeMap`) and ignore
/// either the keys or values.
///
/// **Why is this bad?** Readability. There are `keys` and `values` methods that can be used to
/// express that don't need the values or keys.
///
/// **Known problems:** None
///
/// **Example:**
/// ```rust
/// for (k, _) in &map { .. }
/// ```
/// could be replaced by
/// ```rust
/// for k in map.keys() { .. }
/// ```
2016-02-05 23:13:29 +00:00
declare_lint! {
pub FOR_KV_MAP ,
Warn ,
" looping on a map using `iter` when `keys` or `values` would do "
}
2016-01-19 20:10:00 +00:00
2015-08-12 19:56:27 +00:00
#[ derive(Copy, Clone) ]
2016-06-10 14:17:20 +00:00
pub struct Pass ;
2015-08-12 19:56:27 +00:00
2016-06-10 14:17:20 +00:00
impl LintPass for Pass {
2015-08-12 19:56:27 +00:00
fn get_lints ( & self ) -> LintArray {
2016-01-04 04:26:12 +00:00
lint_array! ( NEEDLESS_RANGE_LOOP ,
EXPLICIT_ITER_LOOP ,
ITER_NEXT_LOOP ,
WHILE_LET_LOOP ,
UNUSED_COLLECT ,
REVERSE_RANGE_LOOP ,
EXPLICIT_COUNTER_LOOP ,
EMPTY_LOOP ,
2016-01-19 20:10:00 +00:00
WHILE_LET_ON_ITERATOR ,
FOR_KV_MAP )
2015-08-12 19:56:27 +00:00
}
2015-09-19 02:53:04 +00:00
}
2015-08-12 19:56:27 +00:00
2016-06-10 14:17:20 +00:00
impl LateLintPass for Pass {
2015-09-19 02:53:04 +00:00
fn check_expr ( & mut self , cx : & LateContext , expr : & Expr ) {
2016-06-29 22:08:43 +00:00
if let Some ( ( pat , arg , body ) ) = higher ::for_loop ( expr ) {
2015-11-18 11:35:18 +00:00
check_for_loop ( cx , pat , arg , body , expr ) ;
2015-08-12 19:56:27 +00:00
}
2015-08-29 09:41:06 +00:00
// check for `loop { if let {} else break }` that could be `while let`
2015-09-27 07:39:42 +00:00
// (also matches an explicit "match" instead of "if let")
// (even if the "match" or "if let" is used for declaration)
2015-08-29 09:41:06 +00:00
if let ExprLoop ( ref block , _ ) = expr . node {
2015-10-12 11:38:18 +00:00
// also check for empty `loop {}` statements
if block . stmts . is_empty ( ) & & block . expr . is_none ( ) {
2016-01-04 04:26:12 +00:00
span_lint ( cx ,
EMPTY_LOOP ,
expr . span ,
" empty `loop {}` detected. You may want to either use `panic!()` or add \
` std ::thread ::sleep ( .. ) ; ` to the loop body . " );
2015-10-12 11:38:18 +00:00
}
2015-10-14 09:44:09 +00:00
2015-10-11 16:49:01 +00:00
// extract the expression from the first statement (if any) in a block
let inner_stmt_expr = extract_expr_from_first_stmt ( block ) ;
2015-12-14 13:30:09 +00:00
// or extract the first expression (if any) from the block
if let Some ( inner ) = inner_stmt_expr . or_else ( | | extract_first_expr ( block ) ) {
2015-08-29 09:41:06 +00:00
if let ExprMatch ( ref matchexpr , ref arms , ref source ) = inner . node {
// ensure "if let" compatible match structure
match * source {
2016-04-14 18:14:03 +00:00
MatchSource ::Normal |
MatchSource ::IfLetDesugar { .. } = > {
2016-06-16 14:19:17 +00:00
if arms . len ( ) = = 2 & &
arms [ 0 ] . pats . len ( ) = = 1 & & arms [ 0 ] . guard . is_none ( ) & &
2016-01-04 04:26:12 +00:00
arms [ 1 ] . pats . len ( ) = = 1 & & arms [ 1 ] . guard . is_none ( ) & &
is_break_expr ( & arms [ 1 ] . body ) {
if in_external_macro ( cx , expr . span ) {
return ;
}
2016-02-27 21:59:15 +00:00
// NOTE: we used to make build a body here instead of using
// ellipsis, this was removed because:
// 1) it was ugly with big bodies;
// 2) it was not indented properly;
// 3) it wasn’ t very smart (see #675).
2016-02-24 19:54:35 +00:00
span_lint_and_then ( cx ,
2016-01-04 04:26:12 +00:00
WHILE_LET_LOOP ,
expr . span ,
" this loop could be written as a `while let` loop " ,
2016-02-24 19:54:35 +00:00
| db | {
2016-02-27 21:59:15 +00:00
let sug = format! ( " while let {} = {} {{ .. }} " ,
2016-02-24 19:54:35 +00:00
snippet ( cx , arms [ 0 ] . pats [ 0 ] . span , " .. " ) ,
2016-02-27 21:59:15 +00:00
snippet ( cx , matchexpr . span , " .. " ) ) ;
2016-02-24 19:54:35 +00:00
db . span_suggestion ( expr . span , " try " , sug ) ;
} ) ;
2016-01-04 04:26:12 +00:00
}
}
_ = > ( ) ,
2015-08-29 09:41:06 +00:00
}
}
}
}
2015-10-22 21:16:58 +00:00
if let ExprMatch ( ref match_expr , ref arms , MatchSource ::WhileLetDesugar ) = expr . node {
2015-10-16 18:27:13 +00:00
let pat = & arms [ 0 ] . pats [ 0 ] . node ;
2016-05-27 12:24:28 +00:00
if let ( & PatKind ::TupleStruct ( ref path , ref pat_args , _ ) ,
2016-01-04 04:26:12 +00:00
& ExprMethodCall ( method_name , _ , ref method_args ) ) = ( pat , & match_expr . node ) {
2015-10-26 22:49:37 +00:00
let iter_expr = & method_args [ 0 ] ;
2015-10-22 21:16:58 +00:00
if let Some ( lhs_constructor ) = path . segments . last ( ) {
if method_name . node . as_str ( ) = = " next " & &
2016-04-26 11:31:52 +00:00
match_trait_method ( cx , match_expr , & paths ::ITERATOR ) & &
2016-05-19 21:14:34 +00:00
lhs_constructor . name . as_str ( ) = = " Some " & &
2016-01-04 04:26:12 +00:00
! is_iterator_used_after_while_let ( cx , iter_expr ) {
2015-10-22 21:16:58 +00:00
let iterator = snippet ( cx , method_args [ 0 ] . span , " _ " ) ;
let loop_var = snippet ( cx , pat_args [ 0 ] . span , " _ " ) ;
2016-06-07 16:32:26 +00:00
span_lint_and_then ( cx ,
2016-01-04 04:26:12 +00:00
WHILE_LET_ON_ITERATOR ,
expr . span ,
2015-10-22 21:16:58 +00:00
" this loop could be written as a `for` loop " ,
2016-06-07 16:32:26 +00:00
| db | {
db . span_suggestion ( expr . span ,
" try " ,
format! ( " for {} in {} {{ .. }} " , loop_var , iterator ) ) ;
} ) ;
2015-10-22 21:16:58 +00:00
}
2015-10-16 18:27:13 +00:00
}
}
}
2015-08-12 19:56:27 +00:00
}
2015-08-30 11:10:59 +00:00
2015-09-19 02:53:04 +00:00
fn check_stmt ( & mut self , cx : & LateContext , stmt : & Stmt ) {
2015-08-30 11:10:59 +00:00
if let StmtSemi ( ref expr , _ ) = stmt . node {
if let ExprMethodCall ( ref method , _ , ref args ) = expr . node {
2015-09-28 05:04:06 +00:00
if args . len ( ) = = 1 & & method . node . as_str ( ) = = " collect " & &
2016-04-26 11:31:52 +00:00
match_trait_method ( cx , expr , & paths ::ITERATOR ) {
2016-01-04 04:26:12 +00:00
span_lint ( cx ,
UNUSED_COLLECT ,
expr . span ,
2016-02-20 20:20:56 +00:00
" you are collect()ing an iterator and throwing away the result. \
Consider using an explicit for loop to exhaust the iterator " );
2015-08-30 11:10:59 +00:00
}
}
}
}
2015-08-12 19:56:27 +00:00
}
2015-11-18 11:35:18 +00:00
fn check_for_loop ( cx : & LateContext , pat : & Pat , arg : & Expr , body : & Expr , expr : & Expr ) {
2016-01-14 19:58:32 +00:00
check_for_loop_range ( cx , pat , arg , body , expr ) ;
check_for_loop_reverse_range ( cx , arg , expr ) ;
2016-01-29 07:34:09 +00:00
check_for_loop_arg ( cx , pat , arg , expr ) ;
2016-01-14 19:58:32 +00:00
check_for_loop_explicit_counter ( cx , arg , body , expr ) ;
2016-02-05 18:14:02 +00:00
check_for_loop_over_map_kv ( cx , pat , arg , body , expr ) ;
2016-01-14 19:58:32 +00:00
}
2015-11-18 11:35:18 +00:00
2016-01-14 19:58:32 +00:00
/// Check for looping over a range and then indexing a sequence with it.
/// The iteratee must be a range literal.
fn check_for_loop_range ( cx : & LateContext , pat : & Pat , arg : & Expr , body : & Expr , expr : & Expr ) {
2016-07-01 17:31:14 +00:00
if let Some ( higher ::Range { start : Some ( ref start ) , ref end , limits } ) = higher ::range ( arg ) {
2016-01-14 19:58:32 +00:00
// the var must be a single name
2016-05-31 17:17:31 +00:00
if let PatKind ::Binding ( _ , ref ident , _ ) = pat . node {
2016-01-14 19:58:32 +00:00
let mut visitor = VarVisitor {
cx : cx ,
2016-05-19 21:14:34 +00:00
var : ident . node ,
2016-02-13 21:09:17 +00:00
indexed : HashMap ::new ( ) ,
2016-01-14 19:58:32 +00:00
nonindex : false ,
} ;
walk_expr ( & mut visitor , body ) ;
2016-02-13 21:09:17 +00:00
2016-01-14 19:58:32 +00:00
// linting condition: we only indexed one variable
if visitor . indexed . len ( ) = = 1 {
2016-02-13 21:09:17 +00:00
let ( indexed , indexed_extent ) = visitor . indexed
2016-02-24 16:38:57 +00:00
. into_iter ( )
. next ( )
. unwrap_or_else ( | | unreachable! ( ) /* len == 1 */ ) ;
2016-01-14 19:58:32 +00:00
2016-02-13 21:09:17 +00:00
// ensure that the indexed variable was declared before the loop, see #601
2016-03-07 22:24:11 +00:00
if let Some ( indexed_extent ) = indexed_extent {
let pat_extent = cx . tcx . region_maps . var_scope ( pat . id ) ;
if cx . tcx . region_maps . is_subscope_of ( indexed_extent , pat_extent ) {
return ;
}
2016-02-13 21:09:17 +00:00
}
2016-03-07 15:31:38 +00:00
let starts_at_zero = is_integer_literal ( start , 0 ) ;
2016-01-14 19:58:32 +00:00
2016-07-01 17:31:14 +00:00
let skip = if starts_at_zero {
" " . to_owned ( )
2016-01-30 12:48:39 +00:00
} else {
2016-07-01 17:31:14 +00:00
format! ( " .skip( {} ) " , snippet ( cx , start . span , " .. " ) )
2016-01-14 19:58:32 +00:00
} ;
2016-07-01 17:31:14 +00:00
let take = if let Some ( ref end ) = * end {
2016-04-26 15:05:39 +00:00
if is_len_call ( end , & indexed ) {
2016-07-01 17:31:14 +00:00
" " . to_owned ( )
2016-02-29 08:45:36 +00:00
} else {
2016-07-01 17:31:14 +00:00
match limits {
ast ::RangeLimits ::Closed = > {
let end = sugg ::Sugg ::hir ( cx , end , " <count> " ) ;
format! ( " .take( {} ) " , end + sugg ::ONE )
}
ast ::RangeLimits ::HalfOpen = > {
format! ( " .take( {} ) " , snippet ( cx , end . span , " .. " ) )
}
}
2016-01-14 19:58:32 +00:00
}
} else {
2016-07-01 17:31:14 +00:00
" " . to_owned ( )
2016-01-14 19:58:32 +00:00
} ;
if visitor . nonindex {
2016-07-01 16:44:59 +00:00
span_lint_and_then ( cx ,
NEEDLESS_RANGE_LOOP ,
expr . span ,
& format! ( " the loop variable ` {} ` is used to index ` {} ` " , ident . node , indexed ) ,
| db | {
multispan_sugg ( db , " consider using an iterator " . to_string ( ) , & [
( pat . span , & format! ( " ( {} , <item>) " , ident . node ) ) ,
( arg . span , & format! ( " {} .iter().enumerate() {} {} " , indexed , take , skip ) ) ,
] ) ;
} ) ;
2016-01-14 19:58:32 +00:00
} else {
let repl = if starts_at_zero & & take . is_empty ( ) {
format! ( " & {} " , indexed )
2016-01-30 12:48:39 +00:00
} else {
2016-01-14 19:58:32 +00:00
format! ( " {} .iter() {} {} " , indexed , take , skip )
} ;
2016-07-01 16:44:59 +00:00
span_lint_and_then ( cx ,
NEEDLESS_RANGE_LOOP ,
expr . span ,
& format! ( " the loop variable ` {} ` is only used to index ` {} `. " , ident . node , indexed ) ,
| db | {
multispan_sugg ( db , " consider using an iterator " . to_string ( ) , & [
( pat . span , " <item> " ) ,
( arg . span , & repl ) ,
] ) ;
} ) ;
2015-11-18 11:35:18 +00:00
}
}
}
}
2016-01-14 19:58:32 +00:00
}
fn is_len_call ( expr : & Expr , var : & Name ) -> bool {
if_let_chain! { [
let ExprMethodCall ( method , _ , ref len_args ) = expr . node ,
len_args . len ( ) = = 1 ,
method . node . as_str ( ) = = " len " ,
let ExprPath ( _ , ref path ) = len_args [ 0 ] . node ,
path . segments . len ( ) = = 1 ,
2016-05-19 21:14:34 +00:00
& path . segments [ 0 ] . name = = var
2016-01-14 19:58:32 +00:00
] , {
return true ;
} }
2015-11-18 11:35:18 +00:00
2016-01-14 19:58:32 +00:00
false
}
fn check_for_loop_reverse_range ( cx : & LateContext , arg : & Expr , expr : & Expr ) {
2015-11-18 11:35:18 +00:00
// if this for loop is iterating over a two-sided range...
2016-06-29 21:16:44 +00:00
if let Some ( higher ::Range { start : Some ( ref start ) , end : Some ( ref end ) , limits } ) = higher ::range ( arg ) {
2015-11-18 11:35:18 +00:00
// ...and both sides are compile-time constant integers...
2016-04-26 15:05:39 +00:00
if let Ok ( start_idx ) = eval_const_expr_partial ( cx . tcx , start , ExprTypeChecked , None ) {
if let Ok ( end_idx ) = eval_const_expr_partial ( cx . tcx , end , ExprTypeChecked , None ) {
2016-03-07 15:31:38 +00:00
// ...and the start index is greater than the end index,
2015-11-18 11:35:18 +00:00
// this loop will never run. This is often confusing for developers
// who think that this will iterate from the larger value to the
// smaller value.
2016-03-07 15:31:38 +00:00
let ( sup , eq ) = match ( start_idx , end_idx ) {
2016-03-15 19:09:53 +00:00
( ConstVal ::Integral ( start_idx ) , ConstVal ::Integral ( end_idx ) ) = > {
2016-03-07 15:31:38 +00:00
( start_idx > end_idx , start_idx = = end_idx )
2016-02-24 16:38:57 +00:00
}
2016-02-07 17:10:03 +00:00
_ = > ( false , false ) ,
} ;
if sup {
2016-03-07 15:31:38 +00:00
let start_snippet = snippet ( cx , start . span , " _ " ) ;
let end_snippet = snippet ( cx , end . span , " _ " ) ;
2016-06-09 21:05:48 +00:00
let dots = if limits = = ast ::RangeLimits ::Closed {
" ... "
} else {
" .. "
} ;
2016-02-07 17:10:03 +00:00
span_lint_and_then ( cx ,
2016-01-04 04:26:12 +00:00
REVERSE_RANGE_LOOP ,
expr . span ,
" this range is empty so this for loop will never run " ,
2016-02-07 17:10:03 +00:00
| db | {
2016-06-07 15:58:52 +00:00
db . span_suggestion ( arg . span ,
2016-02-07 17:10:03 +00:00
" consider using the following if \
you are attempting to iterate \
over this range in reverse " ,
2016-06-09 21:05:48 +00:00
format! ( " ( {end} {dots} {start} ).rev() " ,
end = end_snippet ,
dots = dots ,
start = start_snippet ) ) ;
2016-02-07 17:10:03 +00:00
} ) ;
2016-03-07 15:55:12 +00:00
} else if eq & & limits ! = ast ::RangeLimits ::Closed {
2015-11-18 11:35:18 +00:00
// if they are equal, it's also problematic - this loop
// will never run.
2016-01-04 04:26:12 +00:00
span_lint ( cx ,
REVERSE_RANGE_LOOP ,
expr . span ,
" this range is empty so this for loop will never run " ) ;
2015-11-18 11:35:18 +00:00
}
}
}
}
2016-01-14 19:58:32 +00:00
}
2015-11-18 11:35:18 +00:00
2016-01-29 07:34:09 +00:00
fn check_for_loop_arg ( cx : & LateContext , pat : & Pat , arg : & Expr , expr : & Expr ) {
let mut next_loop_linted = false ; // whether or not ITER_NEXT_LOOP lint was used
2015-11-18 11:35:18 +00:00
if let ExprMethodCall ( ref method , _ , ref args ) = arg . node {
// just the receiver, no arguments
if args . len ( ) = = 1 {
let method_name = method . node ;
// check for looping over x.iter() or x.iter_mut(), could use &x or &mut x
if method_name . as_str ( ) = = " iter " | | method_name . as_str ( ) = = " iter_mut " {
if is_ref_iterable_type ( cx , & args [ 0 ] ) {
let object = snippet ( cx , args [ 0 ] . span , " _ " ) ;
2016-01-04 04:26:12 +00:00
span_lint ( cx ,
EXPLICIT_ITER_LOOP ,
expr . span ,
& format! ( " it is more idiomatic to loop over `& {} {} ` instead of ` {} . {} ()` " ,
if method_name . as_str ( ) = = " iter_mut " {
" mut "
} else {
" "
} ,
object ,
object ,
method_name ) ) ;
2015-11-18 11:35:18 +00:00
}
2016-04-26 11:31:52 +00:00
} else if method_name . as_str ( ) = = " next " & & match_trait_method ( cx , arg , & paths ::ITERATOR ) {
2016-01-04 04:26:12 +00:00
span_lint ( cx ,
ITER_NEXT_LOOP ,
expr . span ,
" you are iterating over `Iterator::next()` which is an Option; this will compile but is \
probably not what you want " );
2016-01-29 07:34:09 +00:00
next_loop_linted = true ;
2015-11-18 11:35:18 +00:00
}
}
}
2016-01-29 07:34:09 +00:00
if ! next_loop_linted {
2016-01-29 23:15:57 +00:00
check_arg_type ( cx , pat , arg ) ;
2016-01-29 07:34:09 +00:00
}
}
2015-11-18 11:35:18 +00:00
2016-01-29 23:15:57 +00:00
/// Check for `for` loops over `Option`s and `Results`
fn check_arg_type ( cx : & LateContext , pat : & Pat , arg : & Expr ) {
2016-01-29 07:34:09 +00:00
let ty = cx . tcx . expr_ty ( arg ) ;
2016-04-14 16:13:15 +00:00
if match_type ( cx , ty , & paths ::OPTION ) {
2016-02-24 16:38:57 +00:00
span_help_and_lint ( cx ,
FOR_LOOP_OVER_OPTION ,
arg . span ,
& format! ( " for loop over ` {0} `, which is an `Option`. This is more readably written as an \
` if let ` statement . " ,
snippet ( cx , arg . span , " _ " ) ) ,
& format! ( " consider replacing `for {0} in {1} ` with `if let Some( {0} ) = {1} ` " ,
snippet ( cx , pat . span , " _ " ) ,
snippet ( cx , arg . span , " _ " ) ) ) ;
2016-04-14 16:13:15 +00:00
} else if match_type ( cx , ty , & paths ::RESULT ) {
2016-02-24 16:38:57 +00:00
span_help_and_lint ( cx ,
FOR_LOOP_OVER_RESULT ,
arg . span ,
& format! ( " for loop over ` {0} `, which is a `Result`. This is more readably written as an \
` if let ` statement . " ,
snippet ( cx , arg . span , " _ " ) ) ,
& format! ( " consider replacing `for {0} in {1} ` with `if let Ok( {0} ) = {1} ` " ,
snippet ( cx , pat . span , " _ " ) ,
snippet ( cx , arg . span , " _ " ) ) ) ;
2016-01-29 23:15:57 +00:00
}
2016-01-14 19:58:32 +00:00
}
fn check_for_loop_explicit_counter ( cx : & LateContext , arg : & Expr , body : & Expr , expr : & Expr ) {
2015-11-18 11:35:18 +00:00
// Look for variables that are incremented once per loop iteration.
2016-01-04 04:26:12 +00:00
let mut visitor = IncrementVisitor {
cx : cx ,
states : HashMap ::new ( ) ,
depth : 0 ,
done : false ,
} ;
2015-11-18 11:35:18 +00:00
walk_expr ( & mut visitor , body ) ;
// For each candidate, check the parent block to see if
// it's initialized to zero at the start of the loop.
let map = & cx . tcx . map ;
2016-01-04 04:26:12 +00:00
let parent_scope = map . get_enclosing_scope ( expr . id ) . and_then ( | id | map . get_enclosing_scope ( id ) ) ;
2015-11-18 11:35:18 +00:00
if let Some ( parent_id ) = parent_scope {
if let NodeBlock ( block ) = map . get ( parent_id ) {
2016-01-04 04:26:12 +00:00
for ( id , _ ) in visitor . states . iter ( ) . filter ( | & ( _ , v ) | * v = = VarState ::IncrOnce ) {
let mut visitor2 = InitializeVisitor {
cx : cx ,
end_expr : expr ,
2016-02-02 21:35:01 +00:00
var_id : * id ,
2016-01-04 04:26:12 +00:00
state : VarState ::IncrOnce ,
name : None ,
depth : 0 ,
past_loop : false ,
} ;
2015-11-18 11:35:18 +00:00
walk_block ( & mut visitor2 , block ) ;
if visitor2 . state = = VarState ::Warn {
if let Some ( name ) = visitor2 . name {
2016-01-04 04:26:12 +00:00
span_lint ( cx ,
EXPLICIT_COUNTER_LOOP ,
expr . span ,
& format! ( " the variable ` {0} ` is used as a loop counter. Consider using `for ( {0} , \
item ) in { 1 } . enumerate ( ) ` or similar iterators " ,
name ,
snippet ( cx , arg . span , " _ " ) ) ) ;
2015-11-18 11:35:18 +00:00
}
}
}
}
}
}
2016-03-19 16:48:29 +00:00
/// Check for the `FOR_KV_MAP` lint.
2016-02-05 18:14:02 +00:00
fn check_for_loop_over_map_kv ( cx : & LateContext , pat : & Pat , arg : & Expr , body : & Expr , expr : & Expr ) {
2016-07-01 18:55:45 +00:00
let pat_span = pat . span ;
2016-05-27 12:24:28 +00:00
if let PatKind ::Tuple ( ref pat , _ ) = pat . node {
2016-01-19 20:10:00 +00:00
if pat . len ( ) = = 2 {
2016-07-01 18:55:45 +00:00
let ( new_pat_span , kind ) = match ( & pat [ 0 ] . node , & pat [ 1 ] . node ) {
( key , _ ) if pat_is_wild ( key , body ) = > ( pat [ 1 ] . span , " value " ) ,
( _ , value ) if pat_is_wild ( value , body ) = > ( pat [ 0 ] . span , " key " ) ,
2016-02-24 16:38:57 +00:00
_ = > return ,
2016-01-19 20:10:00 +00:00
} ;
2016-07-01 18:55:45 +00:00
let ( arg_span , arg ) = match arg . node {
ExprAddrOf ( MutImmutable , ref expr ) = > ( arg . span , & * * expr ) ,
2016-02-26 11:45:55 +00:00
ExprAddrOf ( MutMutable , _ ) = > return , // for _ in &mut _, there is no {values,keys}_mut method
2016-07-01 18:55:45 +00:00
_ = > ( arg . span , arg ) ,
2016-01-19 20:10:00 +00:00
} ;
2016-02-26 11:45:55 +00:00
let ty = walk_ptrs_ty ( cx . tcx . expr_ty ( arg ) ) ;
2016-04-14 16:13:15 +00:00
if match_type ( cx , ty , & paths ::HASHMAP ) | | match_type ( cx , ty , & paths ::BTREEMAP ) {
2016-01-19 20:10:00 +00:00
span_lint_and_then ( cx ,
2016-02-24 16:38:57 +00:00
FOR_KV_MAP ,
expr . span ,
2016-07-01 18:55:45 +00:00
& format! ( " you seem to want to iterate on a map's {} s " , kind ) ,
2016-02-24 16:38:57 +00:00
| db | {
2016-07-01 18:55:45 +00:00
let map = sugg ::Sugg ::hir ( cx , arg , " map " ) ;
multispan_sugg ( db , " use the corresponding method " . into ( ) , & [
( pat_span , & snippet ( cx , new_pat_span , kind ) ) ,
( arg_span , & format! ( " {} . {} s() " , map . maybe_par ( ) , kind ) ) ,
] ) ;
2016-06-05 23:42:39 +00:00
} ) ;
2016-01-19 20:10:00 +00:00
}
}
}
}
2016-03-19 16:48:29 +00:00
/// Return true if the pattern is a `PatWild` or an ident prefixed with `'_'`.
2016-02-18 20:16:39 +00:00
fn pat_is_wild ( pat : & PatKind , body : & Expr ) -> bool {
2016-01-19 20:10:00 +00:00
match * pat {
2016-02-18 20:16:39 +00:00
PatKind ::Wild = > true ,
2016-05-31 17:17:31 +00:00
PatKind ::Binding ( _ , ident , None ) if ident . node . as_str ( ) . starts_with ( '_' ) = > {
2016-02-05 18:14:02 +00:00
let mut visitor = UsedVisitor {
var : ident . node ,
used : false ,
} ;
walk_expr ( & mut visitor , body ) ;
! visitor . used
2016-02-24 16:38:57 +00:00
}
2016-01-19 20:10:00 +00:00
_ = > false ,
}
}
2016-02-05 18:14:02 +00:00
struct UsedVisitor {
2016-05-19 21:14:34 +00:00
var : ast ::Name , // var to look for
2016-02-05 18:14:02 +00:00
used : bool , // has the var been used otherwise?
}
impl < ' a > Visitor < ' a > for UsedVisitor {
fn visit_expr ( & mut self , expr : & Expr ) {
if let ExprPath ( None , ref path ) = expr . node {
2016-05-19 21:14:34 +00:00
if path . segments . len ( ) = = 1 & & path . segments [ 0 ] . name = = self . var {
2016-02-05 18:14:02 +00:00
self . used = true ;
2016-02-24 16:38:57 +00:00
return ;
2016-02-05 18:14:02 +00:00
}
}
walk_expr ( self , expr ) ;
}
}
2015-08-12 19:56:27 +00:00
struct VarVisitor < ' v , ' t : ' v > {
2015-09-19 02:53:04 +00:00
cx : & ' v LateContext < ' v , ' t > , // context reference
2016-01-04 04:26:12 +00:00
var : Name , // var name to look for as index
2016-03-07 22:24:11 +00:00
indexed : HashMap < Name , Option < CodeExtent > > , // indexed variables, the extent is None for global
2016-01-04 04:26:12 +00:00
nonindex : bool , // has the var been used otherwise?
2015-08-12 19:56:27 +00:00
}
impl < ' v , ' t > Visitor < ' v > for VarVisitor < ' v , ' t > {
fn visit_expr ( & mut self , expr : & ' v Expr ) {
if let ExprPath ( None , ref path ) = expr . node {
2016-05-19 21:14:34 +00:00
if path . segments . len ( ) = = 1 & & path . segments [ 0 ] . name = = self . var {
2015-08-12 19:56:27 +00:00
// we are referencing our variable! now check if it's as an index
2016-06-06 00:09:19 +00:00
if_let_chain! { [
let Some ( parexpr ) = get_parent_expr ( self . cx , expr ) ,
let ExprIndex ( ref seqexpr , _ ) = parexpr . node ,
let ExprPath ( None , ref seqvar ) = seqexpr . node ,
seqvar . segments . len ( ) = = 1
] , {
let def_map = self . cx . tcx . def_map . borrow ( ) ;
if let Some ( def ) = def_map . get ( & seqexpr . id ) {
match def . base_def {
Def ::Local ( .. ) | Def ::Upvar ( .. ) = > {
let extent = self . cx . tcx . region_maps . var_scope ( def . base_def . var_id ( ) ) ;
self . indexed . insert ( seqvar . segments [ 0 ] . name , Some ( extent ) ) ;
return ; // no need to walk further
2016-03-07 22:24:11 +00:00
}
2016-06-06 00:09:19 +00:00
Def ::Static ( .. ) | Def ::Const ( .. ) = > {
self . indexed . insert ( seqvar . segments [ 0 ] . name , None ) ;
return ; // no need to walk further
}
_ = > ( ) ,
2016-02-13 21:09:17 +00:00
}
2015-08-12 19:56:27 +00:00
}
2016-06-06 00:09:19 +00:00
} }
2015-08-12 19:56:27 +00:00
// we are not indexing anything, record that
self . nonindex = true ;
return ;
}
}
walk_expr ( self , expr ) ;
}
}
2015-08-25 16:26:20 +00:00
2015-10-26 22:49:37 +00:00
fn is_iterator_used_after_while_let ( cx : & LateContext , iter_expr : & Expr ) -> bool {
let def_id = match var_def_id ( cx , iter_expr ) {
Some ( id ) = > id ,
2016-01-04 04:26:12 +00:00
None = > return false ,
2015-10-26 22:49:37 +00:00
} ;
let mut visitor = VarUsedAfterLoopVisitor {
cx : cx ,
def_id : def_id ,
iter_expr_id : iter_expr . id ,
past_while_let : false ,
2016-01-04 04:26:12 +00:00
var_used_after_while_let : false ,
2015-10-26 22:49:37 +00:00
} ;
if let Some ( enclosing_block ) = get_enclosing_block ( cx , def_id ) {
walk_block ( & mut visitor , enclosing_block ) ;
2015-10-19 23:04:21 +00:00
}
2015-10-26 22:49:37 +00:00
visitor . var_used_after_while_let
2015-10-19 23:04:21 +00:00
}
2015-10-26 22:49:37 +00:00
struct VarUsedAfterLoopVisitor < ' v , ' t : ' v > {
2015-10-19 23:04:21 +00:00
cx : & ' v LateContext < ' v , ' t > ,
def_id : NodeId ,
2015-10-26 22:49:37 +00:00
iter_expr_id : NodeId ,
past_while_let : bool ,
2016-01-04 04:26:12 +00:00
var_used_after_while_let : bool ,
2015-10-19 23:04:21 +00:00
}
2016-01-04 04:26:12 +00:00
impl < ' v , ' t > Visitor < ' v > for VarUsedAfterLoopVisitor < ' v , ' t > {
2015-10-19 23:04:21 +00:00
fn visit_expr ( & mut self , expr : & ' v Expr ) {
2015-10-26 22:49:37 +00:00
if self . past_while_let {
if Some ( self . def_id ) = = var_def_id ( self . cx , expr ) {
self . var_used_after_while_let = true ;
}
} else if self . iter_expr_id = = expr . id {
self . past_while_let = true ;
2015-10-19 23:04:21 +00:00
}
walk_expr ( self , expr ) ;
}
}
2015-10-26 22:49:37 +00:00
2016-03-19 16:48:29 +00:00
/// Return true if the type of expr is one that provides `IntoIterator` impls
/// for `&T` and `&mut T`, such as `Vec`.
2016-02-29 11:19:32 +00:00
#[ cfg_attr(rustfmt, rustfmt_skip) ]
2015-09-19 02:53:04 +00:00
fn is_ref_iterable_type ( cx : & LateContext , e : & Expr ) -> bool {
2015-08-31 06:29:34 +00:00
// no walk_ptrs_ty: calling iter() on a reference can make sense because it
// will allow further borrows afterwards
let ty = cx . tcx . expr_ty ( e ) ;
2016-01-19 20:10:00 +00:00
is_iterable_array ( ty ) | |
2016-04-14 16:13:15 +00:00
match_type ( cx , ty , & paths ::VEC ) | |
2016-04-14 16:41:38 +00:00
match_type ( cx , ty , & paths ::LINKED_LIST ) | |
2016-04-14 16:13:15 +00:00
match_type ( cx , ty , & paths ::HASHMAP ) | |
2016-04-26 11:31:52 +00:00
match_type ( cx , ty , & paths ::HASHSET ) | |
match_type ( cx , ty , & paths ::VEC_DEQUE ) | |
match_type ( cx , ty , & paths ::BINARY_HEAP ) | |
2016-04-14 16:13:15 +00:00
match_type ( cx , ty , & paths ::BTREEMAP ) | |
2016-04-26 11:31:52 +00:00
match_type ( cx , ty , & paths ::BTREESET )
2015-08-25 16:26:20 +00:00
}
2015-09-06 11:36:21 +00:00
fn is_iterable_array ( ty : ty ::Ty ) -> bool {
2015-09-27 07:39:42 +00:00
// IntoIterator is currently only implemented for array sizes <= 32 in rustc
2015-08-25 16:26:20 +00:00
match ty . sty {
2015-09-06 11:36:21 +00:00
ty ::TyArray ( _ , 0 .. . 32 ) = > true ,
2016-01-04 04:26:12 +00:00
_ = > false ,
2015-08-25 16:26:20 +00:00
}
}
2015-08-29 09:41:06 +00:00
2015-09-27 07:39:42 +00:00
/// If a block begins with a statement (possibly a `let` binding) and has an expression, return it.
fn extract_expr_from_first_stmt ( block : & Block ) -> Option < & Expr > {
2016-01-04 04:26:12 +00:00
if block . stmts . is_empty ( ) {
return None ;
}
2015-10-14 09:44:09 +00:00
if let StmtDecl ( ref decl , _ ) = block . stmts [ 0 ] . node {
if let DeclLocal ( ref local ) = decl . node {
2016-01-04 04:26:12 +00:00
if let Some ( ref expr ) = local . init {
Some ( expr )
} else {
None
}
} else {
None
}
} else {
None
}
2015-09-27 07:39:42 +00:00
}
/// If a block begins with an expression (with or without semicolon), return it.
fn extract_first_expr ( block : & Block ) -> Option < & Expr > {
match block . expr {
2016-06-16 14:19:17 +00:00
Some ( ref expr ) if block . stmts . is_empty ( ) = > Some ( expr ) ,
2016-01-04 04:26:12 +00:00
None if ! block . stmts . is_empty ( ) = > {
match block . stmts [ 0 ] . node {
2016-06-16 14:19:17 +00:00
StmtExpr ( ref expr , _ ) | StmtSemi ( ref expr , _ ) = > Some ( expr ) ,
StmtDecl ( .. ) = > None ,
2016-01-04 04:26:12 +00:00
}
}
2015-10-02 07:55:34 +00:00
_ = > None ,
2015-08-29 09:41:06 +00:00
}
}
/// Return true if expr contains a single break expr (maybe within a block).
fn is_break_expr ( expr : & Expr ) -> bool {
match expr . node {
ExprBreak ( None ) = > true ,
2016-01-04 04:26:12 +00:00
ExprBlock ( ref b ) = > {
match extract_first_expr ( b ) {
Some ( ref subexpr ) = > is_break_expr ( subexpr ) ,
None = > false ,
}
}
2015-08-29 09:41:06 +00:00
_ = > false ,
}
}
2015-08-23 17:25:45 +00:00
// To trigger the EXPLICIT_COUNTER_LOOP lint, a variable must be
// incremented exactly once in the loop body, and initialized to zero
// at the start of the loop.
#[ derive(PartialEq) ]
enum VarState {
2016-01-04 04:26:12 +00:00
Initial , // Not examined yet
IncrOnce , // Incremented exactly once, may be a loop counter
Declared , // Declared but not (yet) initialized to zero
2015-08-23 17:25:45 +00:00
Warn ,
2016-01-04 04:26:12 +00:00
DontWarn ,
2015-08-23 17:25:45 +00:00
}
2016-02-26 11:45:55 +00:00
/// Scan a for loop for variables that are incremented exactly once.
2015-08-23 17:25:45 +00:00
struct IncrementVisitor < ' v , ' t : ' v > {
2016-01-04 04:26:12 +00:00
cx : & ' v LateContext < ' v , ' t > , // context reference
states : HashMap < NodeId , VarState > , // incremented variables
depth : u32 , // depth of conditional expressions
done : bool ,
2015-08-23 17:25:45 +00:00
}
impl < ' v , ' t > Visitor < ' v > for IncrementVisitor < ' v , ' t > {
fn visit_expr ( & mut self , expr : & ' v Expr ) {
if self . done {
return ;
}
// If node is a variable
if let Some ( def_id ) = var_def_id ( self . cx , expr ) {
if let Some ( parent ) = get_parent_expr ( self . cx , expr ) {
let state = self . states . entry ( def_id ) . or_insert ( VarState ::Initial ) ;
match parent . node {
2016-01-04 04:26:12 +00:00
ExprAssignOp ( op , ref lhs , ref rhs ) = > {
2015-08-23 17:25:45 +00:00
if lhs . id = = expr . id {
2015-09-04 13:26:58 +00:00
if op . node = = BiAdd & & is_integer_literal ( rhs , 1 ) {
2015-08-23 17:25:45 +00:00
* state = match * state {
VarState ::Initial if self . depth = = 0 = > VarState ::IncrOnce ,
2016-01-04 04:26:12 +00:00
_ = > VarState ::DontWarn ,
2015-08-23 17:25:45 +00:00
} ;
2016-01-04 04:26:12 +00:00
} else {
2015-08-23 17:25:45 +00:00
// Assigned some other value
* state = VarState ::DontWarn ;
}
2016-01-04 04:26:12 +00:00
}
}
2015-08-23 17:25:45 +00:00
ExprAssign ( ref lhs , _ ) if lhs . id = = expr . id = > * state = VarState ::DontWarn ,
2016-01-04 04:26:12 +00:00
ExprAddrOf ( mutability , _ ) if mutability = = MutMutable = > * state = VarState ::DontWarn ,
_ = > ( ) ,
2015-08-23 17:25:45 +00:00
}
}
2016-01-04 04:26:12 +00:00
} else if is_loop ( expr ) {
2015-08-23 17:25:45 +00:00
self . states . clear ( ) ;
self . done = true ;
return ;
2016-01-04 04:26:12 +00:00
} else if is_conditional ( expr ) {
2015-08-23 17:25:45 +00:00
self . depth + = 1 ;
walk_expr ( self , expr ) ;
self . depth - = 1 ;
return ;
}
walk_expr ( self , expr ) ;
}
}
2016-02-26 11:45:55 +00:00
/// Check whether a variable is initialized to zero at the start of a loop.
2015-08-23 17:25:45 +00:00
struct InitializeVisitor < ' v , ' t : ' v > {
2015-09-19 02:53:04 +00:00
cx : & ' v LateContext < ' v , ' t > , // context reference
2016-01-04 04:26:12 +00:00
end_expr : & ' v Expr , // the for loop. Stop scanning here.
2015-08-23 17:25:45 +00:00
var_id : NodeId ,
state : VarState ,
name : Option < Name > ,
2016-01-04 04:26:12 +00:00
depth : u32 , // depth of conditional expressions
past_loop : bool ,
2015-08-23 17:25:45 +00:00
}
impl < ' v , ' t > Visitor < ' v > for InitializeVisitor < ' v , ' t > {
fn visit_decl ( & mut self , decl : & ' v Decl ) {
// Look for declarations of the variable
if let DeclLocal ( ref local ) = decl . node {
if local . pat . id = = self . var_id {
2016-05-31 17:17:31 +00:00
if let PatKind ::Binding ( _ , ref ident , _ ) = local . pat . node {
2016-05-19 21:14:34 +00:00
self . name = Some ( ident . node ) ;
2015-08-23 17:25:45 +00:00
self . state = if let Some ( ref init ) = local . init {
2015-09-04 13:26:58 +00:00
if is_integer_literal ( init , 0 ) {
2015-08-23 17:25:45 +00:00
VarState ::Warn
} else {
VarState ::Declared
}
2016-01-04 04:26:12 +00:00
} else {
2015-08-23 17:25:45 +00:00
VarState ::Declared
}
}
}
}
walk_decl ( self , decl ) ;
}
fn visit_expr ( & mut self , expr : & ' v Expr ) {
2015-11-25 23:09:01 +00:00
if self . state = = VarState ::DontWarn {
return ;
}
if expr = = self . end_expr {
self . past_loop = true ;
return ;
2015-08-23 17:25:45 +00:00
}
// No need to visit expressions before the variable is
2015-11-25 23:09:01 +00:00
// declared
if self . state = = VarState ::IncrOnce {
2015-08-23 17:25:45 +00:00
return ;
}
// If node is the desired variable, see how it's used
if var_def_id ( self . cx , expr ) = = Some ( self . var_id ) {
if let Some ( parent ) = get_parent_expr ( self . cx , expr ) {
match parent . node {
ExprAssignOp ( _ , ref lhs , _ ) if lhs . id = = expr . id = > {
self . state = VarState ::DontWarn ;
2015-11-17 04:39:42 +00:00
}
2015-08-23 17:25:45 +00:00
ExprAssign ( ref lhs , ref rhs ) if lhs . id = = expr . id = > {
2015-09-04 13:26:58 +00:00
self . state = if is_integer_literal ( rhs , 0 ) & & self . depth = = 0 {
2015-08-23 17:25:45 +00:00
VarState ::Warn
} else {
VarState ::DontWarn
2016-01-04 04:26:12 +00:00
}
}
ExprAddrOf ( mutability , _ ) if mutability = = MutMutable = > self . state = VarState ::DontWarn ,
_ = > ( ) ,
2015-08-23 17:25:45 +00:00
}
}
2015-11-25 23:09:01 +00:00
if self . past_loop {
self . state = VarState ::DontWarn ;
return ;
}
2016-01-04 04:26:12 +00:00
} else if ! self . past_loop & & is_loop ( expr ) {
2015-08-23 17:25:45 +00:00
self . state = VarState ::DontWarn ;
return ;
2016-01-04 04:26:12 +00:00
} else if is_conditional ( expr ) {
2015-08-23 17:25:45 +00:00
self . depth + = 1 ;
walk_expr ( self , expr ) ;
self . depth - = 1 ;
return ;
}
walk_expr ( self , expr ) ;
}
}
2015-09-19 02:53:04 +00:00
fn var_def_id ( cx : & LateContext , expr : & Expr ) -> Option < NodeId > {
2015-08-23 17:25:45 +00:00
if let Some ( path_res ) = cx . tcx . def_map . borrow ( ) . get ( & expr . id ) {
2016-01-22 12:24:44 +00:00
if let Def ::Local ( _ , node_id ) = path_res . base_def {
2016-01-04 04:26:12 +00:00
return Some ( node_id ) ;
2015-08-23 17:25:45 +00:00
}
}
None
}
fn is_loop ( expr : & Expr ) -> bool {
match expr . node {
2016-01-04 04:26:12 +00:00
ExprLoop ( .. ) | ExprWhile ( .. ) = > true ,
_ = > false ,
2015-08-23 17:25:45 +00:00
}
}
fn is_conditional ( expr : & Expr ) -> bool {
match expr . node {
ExprIf ( .. ) | ExprMatch ( .. ) = > true ,
2016-01-04 04:26:12 +00:00
_ = > false ,
2015-08-23 17:25:45 +00:00
}
}