]> git.lizzy.rs Git - rust.git/blob - clippy_lints/src/matches.rs
Rollup merge of #5425 - xiongmao86:issue5367, r=flip1995
[rust.git] / clippy_lints / src / matches.rs
1 use crate::consts::{constant, miri_to_const, Constant};
2 use crate::utils::paths;
3 use crate::utils::sugg::Sugg;
4 use crate::utils::usage::is_unused;
5 use crate::utils::{
6     expr_block, get_arg_name, get_parent_expr, in_macro, indent_of, is_allowed, is_expn_of, is_refutable, is_wild,
7     match_qpath, match_type, match_var, multispan_sugg, remove_blocks, snippet, snippet_block,
8     snippet_with_applicability, span_lint_and_help, span_lint_and_note, span_lint_and_sugg, span_lint_and_then,
9     walk_ptrs_ty,
10 };
11 use if_chain::if_chain;
12 use rustc_ast::ast::LitKind;
13 use rustc_errors::Applicability;
14 use rustc_hir::def::CtorKind;
15 use rustc_hir::{
16     Arm, BindingAnnotation, Block, BorrowKind, Expr, ExprKind, Local, MatchSource, Mutability, Node, Pat, PatKind,
17     QPath, RangeEnd,
18 };
19 use rustc_lint::{LateContext, LateLintPass, LintContext};
20 use rustc_middle::lint::in_external_macro;
21 use rustc_middle::ty::{self, Ty};
22 use rustc_session::{declare_tool_lint, impl_lint_pass};
23 use rustc_span::source_map::Span;
24 use std::cmp::Ordering;
25 use std::collections::Bound;
26
27 declare_clippy_lint! {
28     /// **What it does:** Checks for matches with a single arm where an `if let`
29     /// will usually suffice.
30     ///
31     /// **Why is this bad?** Just readability – `if let` nests less than a `match`.
32     ///
33     /// **Known problems:** None.
34     ///
35     /// **Example:**
36     /// ```rust
37     /// # fn bar(stool: &str) {}
38     /// # let x = Some("abc");
39     /// match x {
40     ///     Some(ref foo) => bar(foo),
41     ///     _ => (),
42     /// }
43     /// ```
44     pub SINGLE_MATCH,
45     style,
46     "a `match` statement with a single nontrivial arm (i.e., where the other arm is `_ => {}`) instead of `if let`"
47 }
48
49 declare_clippy_lint! {
50     /// **What it does:** Checks for matches with two arms where an `if let else` will
51     /// usually suffice.
52     ///
53     /// **Why is this bad?** Just readability – `if let` nests less than a `match`.
54     ///
55     /// **Known problems:** Personal style preferences may differ.
56     ///
57     /// **Example:**
58     ///
59     /// Using `match`:
60     ///
61     /// ```rust
62     /// # fn bar(foo: &usize) {}
63     /// # let other_ref: usize = 1;
64     /// # let x: Option<&usize> = Some(&1);
65     /// match x {
66     ///     Some(ref foo) => bar(foo),
67     ///     _ => bar(&other_ref),
68     /// }
69     /// ```
70     ///
71     /// Using `if let` with `else`:
72     ///
73     /// ```rust
74     /// # fn bar(foo: &usize) {}
75     /// # let other_ref: usize = 1;
76     /// # let x: Option<&usize> = Some(&1);
77     /// if let Some(ref foo) = x {
78     ///     bar(foo);
79     /// } else {
80     ///     bar(&other_ref);
81     /// }
82     /// ```
83     pub SINGLE_MATCH_ELSE,
84     pedantic,
85     "a `match` statement with two arms where the second arm's pattern is a placeholder instead of a specific match pattern"
86 }
87
88 declare_clippy_lint! {
89     /// **What it does:** Checks for matches where all arms match a reference,
90     /// suggesting to remove the reference and deref the matched expression
91     /// instead. It also checks for `if let &foo = bar` blocks.
92     ///
93     /// **Why is this bad?** It just makes the code less readable. That reference
94     /// destructuring adds nothing to the code.
95     ///
96     /// **Known problems:** None.
97     ///
98     /// **Example:**
99     /// ```rust,ignore
100     /// match x {
101     ///     &A(ref y) => foo(y),
102     ///     &B => bar(),
103     ///     _ => frob(&x),
104     /// }
105     /// ```
106     pub MATCH_REF_PATS,
107     style,
108     "a `match` or `if let` with all arms prefixed with `&` instead of deref-ing the match expression"
109 }
110
111 declare_clippy_lint! {
112     /// **What it does:** Checks for matches where match expression is a `bool`. It
113     /// suggests to replace the expression with an `if...else` block.
114     ///
115     /// **Why is this bad?** It makes the code less readable.
116     ///
117     /// **Known problems:** None.
118     ///
119     /// **Example:**
120     /// ```rust
121     /// # fn foo() {}
122     /// # fn bar() {}
123     /// let condition: bool = true;
124     /// match condition {
125     ///     true => foo(),
126     ///     false => bar(),
127     /// }
128     /// ```
129     /// Use if/else instead:
130     /// ```rust
131     /// # fn foo() {}
132     /// # fn bar() {}
133     /// let condition: bool = true;
134     /// if condition {
135     ///     foo();
136     /// } else {
137     ///     bar();
138     /// }
139     /// ```
140     pub MATCH_BOOL,
141     style,
142     "a `match` on a boolean expression instead of an `if..else` block"
143 }
144
145 declare_clippy_lint! {
146     /// **What it does:** Checks for overlapping match arms.
147     ///
148     /// **Why is this bad?** It is likely to be an error and if not, makes the code
149     /// less obvious.
150     ///
151     /// **Known problems:** None.
152     ///
153     /// **Example:**
154     /// ```rust
155     /// let x = 5;
156     /// match x {
157     ///     1...10 => println!("1 ... 10"),
158     ///     5...15 => println!("5 ... 15"),
159     ///     _ => (),
160     /// }
161     /// ```
162     pub MATCH_OVERLAPPING_ARM,
163     style,
164     "a `match` with overlapping arms"
165 }
166
167 declare_clippy_lint! {
168     /// **What it does:** Checks for arm which matches all errors with `Err(_)`
169     /// and take drastic actions like `panic!`.
170     ///
171     /// **Why is this bad?** It is generally a bad practice, just like
172     /// catching all exceptions in java with `catch(Exception)`
173     ///
174     /// **Known problems:** None.
175     ///
176     /// **Example:**
177     /// ```rust
178     /// let x: Result<i32, &str> = Ok(3);
179     /// match x {
180     ///     Ok(_) => println!("ok"),
181     ///     Err(_) => panic!("err"),
182     /// }
183     /// ```
184     pub MATCH_WILD_ERR_ARM,
185     style,
186     "a `match` with `Err(_)` arm and take drastic actions"
187 }
188
189 declare_clippy_lint! {
190     /// **What it does:** Checks for match which is used to add a reference to an
191     /// `Option` value.
192     ///
193     /// **Why is this bad?** Using `as_ref()` or `as_mut()` instead is shorter.
194     ///
195     /// **Known problems:** None.
196     ///
197     /// **Example:**
198     /// ```rust
199     /// let x: Option<()> = None;
200     /// let r: Option<&()> = match x {
201     ///     None => None,
202     ///     Some(ref v) => Some(v),
203     /// };
204     /// ```
205     pub MATCH_AS_REF,
206     complexity,
207     "a `match` on an Option value instead of using `as_ref()` or `as_mut`"
208 }
209
210 declare_clippy_lint! {
211     /// **What it does:** Checks for wildcard enum matches using `_`.
212     ///
213     /// **Why is this bad?** New enum variants added by library updates can be missed.
214     ///
215     /// **Known problems:** Suggested replacements may be incorrect if guards exhaustively cover some
216     /// variants, and also may not use correct path to enum if it's not present in the current scope.
217     ///
218     /// **Example:**
219     /// ```rust
220     /// # enum Foo { A(usize), B(usize) }
221     /// # let x = Foo::B(1);
222     /// match x {
223     ///     A => {},
224     ///     _ => {},
225     /// }
226     /// ```
227     pub WILDCARD_ENUM_MATCH_ARM,
228     restriction,
229     "a wildcard enum match arm using `_`"
230 }
231
232 declare_clippy_lint! {
233     /// **What it does:** Checks for wildcard pattern used with others patterns in same match arm.
234     ///
235     /// **Why is this bad?** Wildcard pattern already covers any other pattern as it will match anyway.
236     /// It makes the code less readable, especially to spot wildcard pattern use in match arm.
237     ///
238     /// **Known problems:** None.
239     ///
240     /// **Example:**
241     /// ```rust
242     /// match "foo" {
243     ///     "a" => {},
244     ///     "bar" | _ => {},
245     /// }
246     /// ```
247     pub WILDCARD_IN_OR_PATTERNS,
248     complexity,
249     "a wildcard pattern used with others patterns in same match arm"
250 }
251
252 declare_clippy_lint! {
253     /// **What it does:** Checks for matches being used to destructure a single-variant enum
254     /// or tuple struct where a `let` will suffice.
255     ///
256     /// **Why is this bad?** Just readability – `let` doesn't nest, whereas a `match` does.
257     ///
258     /// **Known problems:** None.
259     ///
260     /// **Example:**
261     /// ```rust
262     /// enum Wrapper {
263     ///     Data(i32),
264     /// }
265     ///
266     /// let wrapper = Wrapper::Data(42);
267     ///
268     /// let data = match wrapper {
269     ///     Wrapper::Data(i) => i,
270     /// };
271     /// ```
272     ///
273     /// The correct use would be:
274     /// ```rust
275     /// enum Wrapper {
276     ///     Data(i32),
277     /// }
278     ///
279     /// let wrapper = Wrapper::Data(42);
280     /// let Wrapper::Data(data) = wrapper;
281     /// ```
282     pub INFALLIBLE_DESTRUCTURING_MATCH,
283     style,
284     "a `match` statement with a single infallible arm instead of a `let`"
285 }
286
287 declare_clippy_lint! {
288     /// **What it does:** Checks for useless match that binds to only one value.
289     ///
290     /// **Why is this bad?** Readability and needless complexity.
291     ///
292     /// **Known problems:**  Suggested replacements may be incorrect when `match`
293     /// is actually binding temporary value, bringing a 'dropped while borrowed' error.
294     ///
295     /// **Example:**
296     /// ```rust
297     /// # let a = 1;
298     /// # let b = 2;
299     ///
300     /// // Bad
301     /// match (a, b) {
302     ///     (c, d) => {
303     ///         // useless match
304     ///     }
305     /// }
306     ///
307     /// // Good
308     /// let (c, d) = (a, b);
309     /// ```
310     pub MATCH_SINGLE_BINDING,
311     complexity,
312     "a match with a single binding instead of using `let` statement"
313 }
314
315 declare_clippy_lint! {
316     /// **What it does:** Checks for unnecessary '..' pattern binding on struct when all fields are explicitly matched.
317     ///
318     /// **Why is this bad?** Correctness and readability. It's like having a wildcard pattern after
319     /// matching all enum variants explicitly.
320     ///
321     /// **Known problems:** None.
322     ///
323     /// **Example:**
324     /// ```rust
325     /// # struct A { a: i32 }
326     /// let a = A { a: 5 };
327     ///
328     /// // Bad
329     /// match a {
330     ///     A { a: 5, .. } => {},
331     ///     _ => {},
332     /// }
333     ///
334     /// // Good
335     /// match a {
336     ///     A { a: 5 } => {},
337     ///     _ => {},
338     /// }
339     /// ```
340     pub REST_PAT_IN_FULLY_BOUND_STRUCTS,
341     restriction,
342     "a match on a struct that binds all fields but still uses the wildcard pattern"
343 }
344
345 #[derive(Default)]
346 pub struct Matches {
347     infallible_destructuring_match_linted: bool,
348 }
349
350 impl_lint_pass!(Matches => [
351     SINGLE_MATCH,
352     MATCH_REF_PATS,
353     MATCH_BOOL,
354     SINGLE_MATCH_ELSE,
355     MATCH_OVERLAPPING_ARM,
356     MATCH_WILD_ERR_ARM,
357     MATCH_AS_REF,
358     WILDCARD_ENUM_MATCH_ARM,
359     WILDCARD_IN_OR_PATTERNS,
360     MATCH_SINGLE_BINDING,
361     INFALLIBLE_DESTRUCTURING_MATCH,
362     REST_PAT_IN_FULLY_BOUND_STRUCTS
363 ]);
364
365 impl<'a, 'tcx> LateLintPass<'a, 'tcx> for Matches {
366     fn check_expr(&mut self, cx: &LateContext<'a, 'tcx>, expr: &'tcx Expr<'_>) {
367         if in_external_macro(cx.sess(), expr.span) {
368             return;
369         }
370         if let ExprKind::Match(ref ex, ref arms, MatchSource::Normal) = expr.kind {
371             check_single_match(cx, ex, arms, expr);
372             check_match_bool(cx, ex, arms, expr);
373             check_overlapping_arms(cx, ex, arms);
374             check_wild_err_arm(cx, ex, arms);
375             check_wild_enum_match(cx, ex, arms);
376             check_match_as_ref(cx, ex, arms, expr);
377             check_wild_in_or_pats(cx, arms);
378
379             if self.infallible_destructuring_match_linted {
380                 self.infallible_destructuring_match_linted = false;
381             } else {
382                 check_match_single_binding(cx, ex, arms, expr);
383             }
384         }
385         if let ExprKind::Match(ref ex, ref arms, _) = expr.kind {
386             check_match_ref_pats(cx, ex, arms, expr);
387         }
388     }
389
390     fn check_local(&mut self, cx: &LateContext<'a, 'tcx>, local: &'tcx Local<'_>) {
391         if_chain! {
392             if let Some(ref expr) = local.init;
393             if let ExprKind::Match(ref target, ref arms, MatchSource::Normal) = expr.kind;
394             if arms.len() == 1 && arms[0].guard.is_none();
395             if let PatKind::TupleStruct(
396                 QPath::Resolved(None, ref variant_name), ref args, _) = arms[0].pat.kind;
397             if args.len() == 1;
398             if let Some(arg) = get_arg_name(&args[0]);
399             let body = remove_blocks(&arms[0].body);
400             if match_var(body, arg);
401
402             then {
403                 let mut applicability = Applicability::MachineApplicable;
404                 self.infallible_destructuring_match_linted = true;
405                 span_lint_and_sugg(
406                     cx,
407                     INFALLIBLE_DESTRUCTURING_MATCH,
408                     local.span,
409                     "you seem to be trying to use `match` to destructure a single infallible pattern. \
410                     Consider using `let`",
411                     "try this",
412                     format!(
413                         "let {}({}) = {};",
414                         snippet_with_applicability(cx, variant_name.span, "..", &mut applicability),
415                         snippet_with_applicability(cx, local.pat.span, "..", &mut applicability),
416                         snippet_with_applicability(cx, target.span, "..", &mut applicability),
417                     ),
418                     applicability,
419                 );
420             }
421         }
422     }
423
424     fn check_pat(&mut self, cx: &LateContext<'a, 'tcx>, pat: &'tcx Pat<'_>) {
425         if_chain! {
426             if let PatKind::Struct(ref qpath, fields, true) = pat.kind;
427             if let QPath::Resolved(_, ref path) = qpath;
428             if let Some(def_id) = path.res.opt_def_id();
429             let ty = cx.tcx.type_of(def_id);
430             if let ty::Adt(def, _) = ty.kind;
431             if def.is_struct() || def.is_union();
432             if fields.len() == def.non_enum_variant().fields.len();
433
434             then {
435                 span_lint_and_help(
436                     cx,
437                     REST_PAT_IN_FULLY_BOUND_STRUCTS,
438                     pat.span,
439                     "unnecessary use of `..` pattern in struct binding. All fields were already bound",
440                     "consider removing `..` from this binding",
441                 );
442             }
443         }
444     }
445 }
446
447 #[rustfmt::skip]
448 fn check_single_match(cx: &LateContext<'_, '_>, ex: &Expr<'_>, arms: &[Arm<'_>], expr: &Expr<'_>) {
449     if arms.len() == 2 && arms[0].guard.is_none() && arms[1].guard.is_none() {
450         if in_macro(expr.span) {
451             // Don't lint match expressions present in
452             // macro_rules! block
453             return;
454         }
455         if let PatKind::Or(..) = arms[0].pat.kind {
456             // don't lint for or patterns for now, this makes
457             // the lint noisy in unnecessary situations
458             return;
459         }
460         let els = remove_blocks(&arms[1].body);
461         let els = if is_unit_expr(els) {
462             None
463         } else if let ExprKind::Block(_, _) = els.kind {
464             // matches with blocks that contain statements are prettier as `if let + else`
465             Some(els)
466         } else {
467             // allow match arms with just expressions
468             return;
469         };
470         let ty = cx.tables.expr_ty(ex);
471         if ty.kind != ty::Bool || is_allowed(cx, MATCH_BOOL, ex.hir_id) {
472             check_single_match_single_pattern(cx, ex, arms, expr, els);
473             check_single_match_opt_like(cx, ex, arms, expr, ty, els);
474         }
475     }
476 }
477
478 fn check_single_match_single_pattern(
479     cx: &LateContext<'_, '_>,
480     ex: &Expr<'_>,
481     arms: &[Arm<'_>],
482     expr: &Expr<'_>,
483     els: Option<&Expr<'_>>,
484 ) {
485     if is_wild(&arms[1].pat) {
486         report_single_match_single_pattern(cx, ex, arms, expr, els);
487     }
488 }
489
490 fn report_single_match_single_pattern(
491     cx: &LateContext<'_, '_>,
492     ex: &Expr<'_>,
493     arms: &[Arm<'_>],
494     expr: &Expr<'_>,
495     els: Option<&Expr<'_>>,
496 ) {
497     let lint = if els.is_some() { SINGLE_MATCH_ELSE } else { SINGLE_MATCH };
498     let els_str = els.map_or(String::new(), |els| {
499         format!(" else {}", expr_block(cx, els, None, "..", Some(expr.span)))
500     });
501     span_lint_and_sugg(
502         cx,
503         lint,
504         expr.span,
505         "you seem to be trying to use match for destructuring a single pattern. Consider using `if \
506          let`",
507         "try this",
508         format!(
509             "if let {} = {} {}{}",
510             snippet(cx, arms[0].pat.span, ".."),
511             snippet(cx, ex.span, ".."),
512             expr_block(cx, &arms[0].body, None, "..", Some(expr.span)),
513             els_str,
514         ),
515         Applicability::HasPlaceholders,
516     );
517 }
518
519 fn check_single_match_opt_like(
520     cx: &LateContext<'_, '_>,
521     ex: &Expr<'_>,
522     arms: &[Arm<'_>],
523     expr: &Expr<'_>,
524     ty: Ty<'_>,
525     els: Option<&Expr<'_>>,
526 ) {
527     // list of candidate `Enum`s we know will never get any more members
528     let candidates = &[
529         (&paths::COW, "Borrowed"),
530         (&paths::COW, "Cow::Borrowed"),
531         (&paths::COW, "Cow::Owned"),
532         (&paths::COW, "Owned"),
533         (&paths::OPTION, "None"),
534         (&paths::RESULT, "Err"),
535         (&paths::RESULT, "Ok"),
536     ];
537
538     let path = match arms[1].pat.kind {
539         PatKind::TupleStruct(ref path, ref inner, _) => {
540             // Contains any non wildcard patterns (e.g., `Err(err)`)?
541             if !inner.iter().all(is_wild) {
542                 return;
543             }
544             rustc_hir_pretty::to_string(rustc_hir_pretty::NO_ANN, |s| s.print_qpath(path, false))
545         },
546         PatKind::Binding(BindingAnnotation::Unannotated, .., ident, None) => ident.to_string(),
547         PatKind::Path(ref path) => {
548             rustc_hir_pretty::to_string(rustc_hir_pretty::NO_ANN, |s| s.print_qpath(path, false))
549         },
550         _ => return,
551     };
552
553     for &(ty_path, pat_path) in candidates {
554         if path == *pat_path && match_type(cx, ty, ty_path) {
555             report_single_match_single_pattern(cx, ex, arms, expr, els);
556         }
557     }
558 }
559
560 fn check_match_bool(cx: &LateContext<'_, '_>, ex: &Expr<'_>, arms: &[Arm<'_>], expr: &Expr<'_>) {
561     // Type of expression is `bool`.
562     if cx.tables.expr_ty(ex).kind == ty::Bool {
563         span_lint_and_then(
564             cx,
565             MATCH_BOOL,
566             expr.span,
567             "you seem to be trying to match on a boolean expression",
568             move |db| {
569                 if arms.len() == 2 {
570                     // no guards
571                     let exprs = if let PatKind::Lit(ref arm_bool) = arms[0].pat.kind {
572                         if let ExprKind::Lit(ref lit) = arm_bool.kind {
573                             match lit.node {
574                                 LitKind::Bool(true) => Some((&*arms[0].body, &*arms[1].body)),
575                                 LitKind::Bool(false) => Some((&*arms[1].body, &*arms[0].body)),
576                                 _ => None,
577                             }
578                         } else {
579                             None
580                         }
581                     } else {
582                         None
583                     };
584
585                     if let Some((true_expr, false_expr)) = exprs {
586                         let sugg = match (is_unit_expr(true_expr), is_unit_expr(false_expr)) {
587                             (false, false) => Some(format!(
588                                 "if {} {} else {}",
589                                 snippet(cx, ex.span, "b"),
590                                 expr_block(cx, true_expr, None, "..", Some(expr.span)),
591                                 expr_block(cx, false_expr, None, "..", Some(expr.span))
592                             )),
593                             (false, true) => Some(format!(
594                                 "if {} {}",
595                                 snippet(cx, ex.span, "b"),
596                                 expr_block(cx, true_expr, None, "..", Some(expr.span))
597                             )),
598                             (true, false) => {
599                                 let test = Sugg::hir(cx, ex, "..");
600                                 Some(format!(
601                                     "if {} {}",
602                                     !test,
603                                     expr_block(cx, false_expr, None, "..", Some(expr.span))
604                                 ))
605                             },
606                             (true, true) => None,
607                         };
608
609                         if let Some(sugg) = sugg {
610                             db.span_suggestion(
611                                 expr.span,
612                                 "consider using an `if`/`else` expression",
613                                 sugg,
614                                 Applicability::HasPlaceholders,
615                             );
616                         }
617                     }
618                 }
619             },
620         );
621     }
622 }
623
624 fn check_overlapping_arms<'a, 'tcx>(cx: &LateContext<'a, 'tcx>, ex: &'tcx Expr<'_>, arms: &'tcx [Arm<'_>]) {
625     if arms.len() >= 2 && cx.tables.expr_ty(ex).is_integral() {
626         let ranges = all_ranges(cx, arms, cx.tables.expr_ty(ex));
627         let type_ranges = type_ranges(&ranges);
628         if !type_ranges.is_empty() {
629             if let Some((start, end)) = overlapping(&type_ranges) {
630                 span_lint_and_note(
631                     cx,
632                     MATCH_OVERLAPPING_ARM,
633                     start.span,
634                     "some ranges overlap",
635                     end.span,
636                     "overlaps with this",
637                 );
638             }
639         }
640     }
641 }
642
643 fn check_wild_err_arm(cx: &LateContext<'_, '_>, ex: &Expr<'_>, arms: &[Arm<'_>]) {
644     let ex_ty = walk_ptrs_ty(cx.tables.expr_ty(ex));
645     if match_type(cx, ex_ty, &paths::RESULT) {
646         for arm in arms {
647             if let PatKind::TupleStruct(ref path, ref inner, _) = arm.pat.kind {
648                 let path_str = rustc_hir_pretty::to_string(rustc_hir_pretty::NO_ANN, |s| s.print_qpath(path, false));
649                 if path_str == "Err" {
650                     let mut matching_wild = inner.iter().any(is_wild);
651                     let mut ident_bind_name = String::from("_");
652                     if !matching_wild {
653                         // Looking for unused bindings (i.e.: `_e`)
654                         inner.iter().for_each(|pat| {
655                             if let PatKind::Binding(.., ident, None) = &pat.kind {
656                                 if ident.as_str().starts_with('_') && is_unused(ident, arm.body) {
657                                     ident_bind_name = (&ident.name.as_str()).to_string();
658                                     matching_wild = true;
659                                 }
660                             }
661                         });
662                     }
663                     if_chain! {
664                         if matching_wild;
665                         if let ExprKind::Block(ref block, _) = arm.body.kind;
666                         if is_panic_block(block);
667                         then {
668                             // `Err(_)` or `Err(_e)` arm with `panic!` found
669                             span_lint_and_note(cx,
670                                 MATCH_WILD_ERR_ARM,
671                                 arm.pat.span,
672                                 &format!("`Err({})` matches all errors", &ident_bind_name),
673                                 arm.pat.span,
674                                 "match each error separately or use the error output",
675                             );
676                         }
677                     }
678                 }
679             }
680         }
681     }
682 }
683
684 fn check_wild_enum_match(cx: &LateContext<'_, '_>, ex: &Expr<'_>, arms: &[Arm<'_>]) {
685     let ty = cx.tables.expr_ty(ex);
686     if !ty.is_enum() {
687         // If there isn't a nice closed set of possible values that can be conveniently enumerated,
688         // don't complain about not enumerating the mall.
689         return;
690     }
691
692     // First pass - check for violation, but don't do much book-keeping because this is hopefully
693     // the uncommon case, and the book-keeping is slightly expensive.
694     let mut wildcard_span = None;
695     let mut wildcard_ident = None;
696     for arm in arms {
697         if let PatKind::Wild = arm.pat.kind {
698             wildcard_span = Some(arm.pat.span);
699         } else if let PatKind::Binding(_, _, ident, None) = arm.pat.kind {
700             wildcard_span = Some(arm.pat.span);
701             wildcard_ident = Some(ident);
702         }
703     }
704
705     if let Some(wildcard_span) = wildcard_span {
706         // Accumulate the variants which should be put in place of the wildcard because they're not
707         // already covered.
708
709         let mut missing_variants = vec![];
710         if let ty::Adt(def, _) = ty.kind {
711             for variant in &def.variants {
712                 missing_variants.push(variant);
713             }
714         }
715
716         for arm in arms {
717             if arm.guard.is_some() {
718                 // Guards mean that this case probably isn't exhaustively covered. Technically
719                 // this is incorrect, as we should really check whether each variant is exhaustively
720                 // covered by the set of guards that cover it, but that's really hard to do.
721                 continue;
722             }
723             if let PatKind::Path(ref path) = arm.pat.kind {
724                 if let QPath::Resolved(_, p) = path {
725                     missing_variants.retain(|e| e.ctor_def_id != Some(p.res.def_id()));
726                 }
727             } else if let PatKind::TupleStruct(ref path, ..) = arm.pat.kind {
728                 if let QPath::Resolved(_, p) = path {
729                     missing_variants.retain(|e| e.ctor_def_id != Some(p.res.def_id()));
730                 }
731             }
732         }
733
734         let mut suggestion: Vec<String> = missing_variants
735             .iter()
736             .map(|v| {
737                 let suffix = match v.ctor_kind {
738                     CtorKind::Fn => "(..)",
739                     CtorKind::Const | CtorKind::Fictive => "",
740                 };
741                 let ident_str = if let Some(ident) = wildcard_ident {
742                     format!("{} @ ", ident.name)
743                 } else {
744                     String::new()
745                 };
746                 // This path assumes that the enum type is imported into scope.
747                 format!("{}{}{}", ident_str, cx.tcx.def_path_str(v.def_id), suffix)
748             })
749             .collect();
750
751         if suggestion.is_empty() {
752             return;
753         }
754
755         let mut message = "wildcard match will miss any future added variants";
756
757         if let ty::Adt(def, _) = ty.kind {
758             if def.is_variant_list_non_exhaustive() {
759                 message = "match on non-exhaustive enum doesn't explicitly match all known variants";
760                 suggestion.push(String::from("_"));
761             }
762         }
763
764         span_lint_and_sugg(
765             cx,
766             WILDCARD_ENUM_MATCH_ARM,
767             wildcard_span,
768             message,
769             "try this",
770             suggestion.join(" | "),
771             Applicability::MachineApplicable,
772         )
773     }
774 }
775
776 // If the block contains only a `panic!` macro (as expression or statement)
777 fn is_panic_block(block: &Block<'_>) -> bool {
778     match (&block.expr, block.stmts.len(), block.stmts.first()) {
779         (&Some(ref exp), 0, _) => {
780             is_expn_of(exp.span, "panic").is_some() && is_expn_of(exp.span, "unreachable").is_none()
781         },
782         (&None, 1, Some(stmt)) => {
783             is_expn_of(stmt.span, "panic").is_some() && is_expn_of(stmt.span, "unreachable").is_none()
784         },
785         _ => false,
786     }
787 }
788
789 fn check_match_ref_pats(cx: &LateContext<'_, '_>, ex: &Expr<'_>, arms: &[Arm<'_>], expr: &Expr<'_>) {
790     if has_only_ref_pats(arms) {
791         let mut suggs = Vec::with_capacity(arms.len() + 1);
792         let (title, msg) = if let ExprKind::AddrOf(BorrowKind::Ref, Mutability::Not, ref inner) = ex.kind {
793             let span = ex.span.source_callsite();
794             suggs.push((span, Sugg::hir_with_macro_callsite(cx, inner, "..").to_string()));
795             (
796                 "you don't need to add `&` to both the expression and the patterns",
797                 "try",
798             )
799         } else {
800             let span = ex.span.source_callsite();
801             suggs.push((span, Sugg::hir_with_macro_callsite(cx, ex, "..").deref().to_string()));
802             (
803                 "you don't need to add `&` to all patterns",
804                 "instead of prefixing all patterns with `&`, you can dereference the expression",
805             )
806         };
807
808         suggs.extend(arms.iter().filter_map(|a| {
809             if let PatKind::Ref(ref refp, _) = a.pat.kind {
810                 Some((a.pat.span, snippet(cx, refp.span, "..").to_string()))
811             } else {
812                 None
813             }
814         }));
815
816         span_lint_and_then(cx, MATCH_REF_PATS, expr.span, title, |db| {
817             if !expr.span.from_expansion() {
818                 multispan_sugg(db, msg.to_owned(), suggs);
819             }
820         });
821     }
822 }
823
824 fn check_match_as_ref(cx: &LateContext<'_, '_>, ex: &Expr<'_>, arms: &[Arm<'_>], expr: &Expr<'_>) {
825     if arms.len() == 2 && arms[0].guard.is_none() && arms[1].guard.is_none() {
826         let arm_ref: Option<BindingAnnotation> = if is_none_arm(&arms[0]) {
827             is_ref_some_arm(&arms[1])
828         } else if is_none_arm(&arms[1]) {
829             is_ref_some_arm(&arms[0])
830         } else {
831             None
832         };
833         if let Some(rb) = arm_ref {
834             let suggestion = if rb == BindingAnnotation::Ref {
835                 "as_ref"
836             } else {
837                 "as_mut"
838             };
839
840             let output_ty = cx.tables.expr_ty(expr);
841             let input_ty = cx.tables.expr_ty(ex);
842
843             let cast = if_chain! {
844                 if let ty::Adt(_, substs) = input_ty.kind;
845                 let input_ty = substs.type_at(0);
846                 if let ty::Adt(_, substs) = output_ty.kind;
847                 let output_ty = substs.type_at(0);
848                 if let ty::Ref(_, output_ty, _) = output_ty.kind;
849                 if input_ty != output_ty;
850                 then {
851                     ".map(|x| x as _)"
852                 } else {
853                     ""
854                 }
855             };
856
857             let mut applicability = Applicability::MachineApplicable;
858             span_lint_and_sugg(
859                 cx,
860                 MATCH_AS_REF,
861                 expr.span,
862                 &format!("use `{}()` instead", suggestion),
863                 "try this",
864                 format!(
865                     "{}.{}(){}",
866                     snippet_with_applicability(cx, ex.span, "_", &mut applicability),
867                     suggestion,
868                     cast,
869                 ),
870                 applicability,
871             )
872         }
873     }
874 }
875
876 fn check_wild_in_or_pats(cx: &LateContext<'_, '_>, arms: &[Arm<'_>]) {
877     for arm in arms {
878         if let PatKind::Or(ref fields) = arm.pat.kind {
879             // look for multiple fields in this arm that contains at least one Wild pattern
880             if fields.len() > 1 && fields.iter().any(is_wild) {
881                 span_lint_and_help(
882                     cx,
883                     WILDCARD_IN_OR_PATTERNS,
884                     arm.pat.span,
885                     "wildcard pattern covers any other pattern as it will match anyway.",
886                     "Consider handling `_` separately.",
887                 );
888             }
889         }
890     }
891 }
892
893 fn check_match_single_binding<'a>(cx: &LateContext<'_, 'a>, ex: &Expr<'a>, arms: &[Arm<'_>], expr: &Expr<'_>) {
894     if in_macro(expr.span) || arms.len() != 1 || is_refutable(cx, arms[0].pat) {
895         return;
896     }
897     let matched_vars = ex.span;
898     let bind_names = arms[0].pat.span;
899     let match_body = remove_blocks(&arms[0].body);
900     let mut snippet_body = if match_body.span.from_expansion() {
901         Sugg::hir_with_macro_callsite(cx, match_body, "..").to_string()
902     } else {
903         snippet_block(cx, match_body.span, "..", Some(expr.span)).to_string()
904     };
905
906     // Do we need to add ';' to suggestion ?
907     match match_body.kind {
908         ExprKind::Block(block, _) => {
909             // macro + expr_ty(body) == ()
910             if block.span.from_expansion() && cx.tables.expr_ty(&match_body).is_unit() {
911                 snippet_body.push(';');
912             }
913         },
914         _ => {
915             // expr_ty(body) == ()
916             if cx.tables.expr_ty(&match_body).is_unit() {
917                 snippet_body.push(';');
918             }
919         },
920     }
921
922     let mut applicability = Applicability::MaybeIncorrect;
923     match arms[0].pat.kind {
924         PatKind::Binding(..) | PatKind::Tuple(_, _) | PatKind::Struct(..) => {
925             // If this match is in a local (`let`) stmt
926             let (target_span, sugg) = if let Some(parent_let_node) = opt_parent_let(cx, ex) {
927                 (
928                     parent_let_node.span,
929                     format!(
930                         "let {} = {};\n{}let {} = {};",
931                         snippet_with_applicability(cx, bind_names, "..", &mut applicability),
932                         snippet_with_applicability(cx, matched_vars, "..", &mut applicability),
933                         " ".repeat(indent_of(cx, expr.span).unwrap_or(0)),
934                         snippet_with_applicability(cx, parent_let_node.pat.span, "..", &mut applicability),
935                         snippet_body
936                     ),
937                 )
938             } else {
939                 // If we are in closure, we need curly braces around suggestion
940                 let mut indent = " ".repeat(indent_of(cx, ex.span).unwrap_or(0));
941                 let (mut cbrace_start, mut cbrace_end) = ("".to_string(), "".to_string());
942                 if let Some(parent_expr) = get_parent_expr(cx, expr) {
943                     if let ExprKind::Closure(..) = parent_expr.kind {
944                         cbrace_end = format!("\n{}}}", indent);
945                         // Fix body indent due to the closure
946                         indent = " ".repeat(indent_of(cx, bind_names).unwrap_or(0));
947                         cbrace_start = format!("{{\n{}", indent);
948                     }
949                 };
950                 (
951                     expr.span,
952                     format!(
953                         "{}let {} = {};\n{}{}{}",
954                         cbrace_start,
955                         snippet_with_applicability(cx, bind_names, "..", &mut applicability),
956                         snippet_with_applicability(cx, matched_vars, "..", &mut applicability),
957                         indent,
958                         snippet_body,
959                         cbrace_end
960                     ),
961                 )
962             };
963             span_lint_and_sugg(
964                 cx,
965                 MATCH_SINGLE_BINDING,
966                 target_span,
967                 "this match could be written as a `let` statement",
968                 "consider using `let` statement",
969                 sugg,
970                 applicability,
971             );
972         },
973         PatKind::Wild => {
974             span_lint_and_sugg(
975                 cx,
976                 MATCH_SINGLE_BINDING,
977                 expr.span,
978                 "this match could be replaced by its body itself",
979                 "consider using the match body instead",
980                 snippet_body,
981                 Applicability::MachineApplicable,
982             );
983         },
984         _ => (),
985     }
986 }
987
988 /// Returns true if the `ex` match expression is in a local (`let`) statement
989 fn opt_parent_let<'a>(cx: &LateContext<'_, 'a>, ex: &Expr<'a>) -> Option<&'a Local<'a>> {
990     if_chain! {
991         let map = &cx.tcx.hir();
992         if let Some(Node::Expr(parent_arm_expr)) = map.find(map.get_parent_node(ex.hir_id));
993         if let Some(Node::Local(parent_let_expr)) = map.find(map.get_parent_node(parent_arm_expr.hir_id));
994         then {
995             return Some(parent_let_expr);
996         }
997     }
998     None
999 }
1000
1001 /// Gets all arms that are unbounded `PatRange`s.
1002 fn all_ranges<'a, 'tcx>(
1003     cx: &LateContext<'a, 'tcx>,
1004     arms: &'tcx [Arm<'_>],
1005     ty: Ty<'tcx>,
1006 ) -> Vec<SpannedRange<Constant>> {
1007     arms.iter()
1008         .flat_map(|arm| {
1009             if let Arm {
1010                 ref pat, guard: None, ..
1011             } = *arm
1012             {
1013                 if let PatKind::Range(ref lhs, ref rhs, range_end) = pat.kind {
1014                     let lhs = match lhs {
1015                         Some(lhs) => constant(cx, cx.tables, lhs)?.0,
1016                         None => miri_to_const(ty.numeric_min_val(cx.tcx)?)?,
1017                     };
1018                     let rhs = match rhs {
1019                         Some(rhs) => constant(cx, cx.tables, rhs)?.0,
1020                         None => miri_to_const(ty.numeric_max_val(cx.tcx)?)?,
1021                     };
1022                     let rhs = match range_end {
1023                         RangeEnd::Included => Bound::Included(rhs),
1024                         RangeEnd::Excluded => Bound::Excluded(rhs),
1025                     };
1026                     return Some(SpannedRange {
1027                         span: pat.span,
1028                         node: (lhs, rhs),
1029                     });
1030                 }
1031
1032                 if let PatKind::Lit(ref value) = pat.kind {
1033                     let value = constant(cx, cx.tables, value)?.0;
1034                     return Some(SpannedRange {
1035                         span: pat.span,
1036                         node: (value.clone(), Bound::Included(value)),
1037                     });
1038                 }
1039             }
1040             None
1041         })
1042         .collect()
1043 }
1044
1045 #[derive(Debug, Eq, PartialEq)]
1046 pub struct SpannedRange<T> {
1047     pub span: Span,
1048     pub node: (T, Bound<T>),
1049 }
1050
1051 type TypedRanges = Vec<SpannedRange<u128>>;
1052
1053 /// Gets all `Int` ranges or all `Uint` ranges. Mixed types are an error anyway
1054 /// and other types than
1055 /// `Uint` and `Int` probably don't make sense.
1056 fn type_ranges(ranges: &[SpannedRange<Constant>]) -> TypedRanges {
1057     ranges
1058         .iter()
1059         .filter_map(|range| match range.node {
1060             (Constant::Int(start), Bound::Included(Constant::Int(end))) => Some(SpannedRange {
1061                 span: range.span,
1062                 node: (start, Bound::Included(end)),
1063             }),
1064             (Constant::Int(start), Bound::Excluded(Constant::Int(end))) => Some(SpannedRange {
1065                 span: range.span,
1066                 node: (start, Bound::Excluded(end)),
1067             }),
1068             (Constant::Int(start), Bound::Unbounded) => Some(SpannedRange {
1069                 span: range.span,
1070                 node: (start, Bound::Unbounded),
1071             }),
1072             _ => None,
1073         })
1074         .collect()
1075 }
1076
1077 fn is_unit_expr(expr: &Expr<'_>) -> bool {
1078     match expr.kind {
1079         ExprKind::Tup(ref v) if v.is_empty() => true,
1080         ExprKind::Block(ref b, _) if b.stmts.is_empty() && b.expr.is_none() => true,
1081         _ => false,
1082     }
1083 }
1084
1085 // Checks if arm has the form `None => None`
1086 fn is_none_arm(arm: &Arm<'_>) -> bool {
1087     match arm.pat.kind {
1088         PatKind::Path(ref path) if match_qpath(path, &paths::OPTION_NONE) => true,
1089         _ => false,
1090     }
1091 }
1092
1093 // Checks if arm has the form `Some(ref v) => Some(v)` (checks for `ref` and `ref mut`)
1094 fn is_ref_some_arm(arm: &Arm<'_>) -> Option<BindingAnnotation> {
1095     if_chain! {
1096         if let PatKind::TupleStruct(ref path, ref pats, _) = arm.pat.kind;
1097         if pats.len() == 1 && match_qpath(path, &paths::OPTION_SOME);
1098         if let PatKind::Binding(rb, .., ident, _) = pats[0].kind;
1099         if rb == BindingAnnotation::Ref || rb == BindingAnnotation::RefMut;
1100         if let ExprKind::Call(ref e, ref args) = remove_blocks(&arm.body).kind;
1101         if let ExprKind::Path(ref some_path) = e.kind;
1102         if match_qpath(some_path, &paths::OPTION_SOME) && args.len() == 1;
1103         if let ExprKind::Path(ref qpath) = args[0].kind;
1104         if let &QPath::Resolved(_, ref path2) = qpath;
1105         if path2.segments.len() == 1 && ident.name == path2.segments[0].ident.name;
1106         then {
1107             return Some(rb)
1108         }
1109     }
1110     None
1111 }
1112
1113 fn has_only_ref_pats(arms: &[Arm<'_>]) -> bool {
1114     let mapped = arms
1115         .iter()
1116         .map(|a| {
1117             match a.pat.kind {
1118                 PatKind::Ref(..) => Some(true), // &-patterns
1119                 PatKind::Wild => Some(false),   // an "anything" wildcard is also fine
1120                 _ => None,                      // any other pattern is not fine
1121             }
1122         })
1123         .collect::<Option<Vec<bool>>>();
1124     // look for Some(v) where there's at least one true element
1125     mapped.map_or(false, |v| v.iter().any(|el| *el))
1126 }
1127
1128 pub fn overlapping<T>(ranges: &[SpannedRange<T>]) -> Option<(&SpannedRange<T>, &SpannedRange<T>)>
1129 where
1130     T: Copy + Ord,
1131 {
1132     #[derive(Copy, Clone, Debug, Eq, PartialEq)]
1133     enum Kind<'a, T> {
1134         Start(T, &'a SpannedRange<T>),
1135         End(Bound<T>, &'a SpannedRange<T>),
1136     }
1137
1138     impl<'a, T: Copy> Kind<'a, T> {
1139         fn range(&self) -> &'a SpannedRange<T> {
1140             match *self {
1141                 Kind::Start(_, r) | Kind::End(_, r) => r,
1142             }
1143         }
1144
1145         fn value(self) -> Bound<T> {
1146             match self {
1147                 Kind::Start(t, _) => Bound::Included(t),
1148                 Kind::End(t, _) => t,
1149             }
1150         }
1151     }
1152
1153     impl<'a, T: Copy + Ord> PartialOrd for Kind<'a, T> {
1154         fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
1155             Some(self.cmp(other))
1156         }
1157     }
1158
1159     impl<'a, T: Copy + Ord> Ord for Kind<'a, T> {
1160         fn cmp(&self, other: &Self) -> Ordering {
1161             match (self.value(), other.value()) {
1162                 (Bound::Included(a), Bound::Included(b)) | (Bound::Excluded(a), Bound::Excluded(b)) => a.cmp(&b),
1163                 // Range patterns cannot be unbounded (yet)
1164                 (Bound::Unbounded, _) | (_, Bound::Unbounded) => unimplemented!(),
1165                 (Bound::Included(a), Bound::Excluded(b)) => match a.cmp(&b) {
1166                     Ordering::Equal => Ordering::Greater,
1167                     other => other,
1168                 },
1169                 (Bound::Excluded(a), Bound::Included(b)) => match a.cmp(&b) {
1170                     Ordering::Equal => Ordering::Less,
1171                     other => other,
1172                 },
1173             }
1174         }
1175     }
1176
1177     let mut values = Vec::with_capacity(2 * ranges.len());
1178
1179     for r in ranges {
1180         values.push(Kind::Start(r.node.0, r));
1181         values.push(Kind::End(r.node.1, r));
1182     }
1183
1184     values.sort();
1185
1186     for (a, b) in values.iter().zip(values.iter().skip(1)) {
1187         match (a, b) {
1188             (&Kind::Start(_, ra), &Kind::End(_, rb)) => {
1189                 if ra.node != rb.node {
1190                     return Some((ra, rb));
1191                 }
1192             },
1193             (&Kind::End(a, _), &Kind::Start(b, _)) if a != Bound::Included(b) => (),
1194             _ => return Some((a.range(), b.range())),
1195         }
1196     }
1197
1198     None
1199 }
1200
1201 #[test]
1202 fn test_overlapping() {
1203     use rustc_span::source_map::DUMMY_SP;
1204
1205     let sp = |s, e| SpannedRange {
1206         span: DUMMY_SP,
1207         node: (s, e),
1208     };
1209
1210     assert_eq!(None, overlapping::<u8>(&[]));
1211     assert_eq!(None, overlapping(&[sp(1, Bound::Included(4))]));
1212     assert_eq!(
1213         None,
1214         overlapping(&[sp(1, Bound::Included(4)), sp(5, Bound::Included(6))])
1215     );
1216     assert_eq!(
1217         None,
1218         overlapping(&[
1219             sp(1, Bound::Included(4)),
1220             sp(5, Bound::Included(6)),
1221             sp(10, Bound::Included(11))
1222         ],)
1223     );
1224     assert_eq!(
1225         Some((&sp(1, Bound::Included(4)), &sp(3, Bound::Included(6)))),
1226         overlapping(&[sp(1, Bound::Included(4)), sp(3, Bound::Included(6))])
1227     );
1228     assert_eq!(
1229         Some((&sp(5, Bound::Included(6)), &sp(6, Bound::Included(11)))),
1230         overlapping(&[
1231             sp(1, Bound::Included(4)),
1232             sp(5, Bound::Included(6)),
1233             sp(6, Bound::Included(11))
1234         ],)
1235     );
1236 }