1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

use rustc::lint::*;
use rustc::hir::*;
use utils::span_lint;

/// **What it does:** Detects classic underflow/overflow checks.
///
/// **Why is this bad?** Most classic C underflow/overflow checks will fail in
/// Rust. Users can use functions like `overflowing_*` and `wrapping_*` instead.
///
/// **Known problems:** None.
///
/// **Example:**
/// ```rust
/// a + b < a
/// ```

declare_lint! {
    pub OVERFLOW_CHECK_CONDITIONAL,
    Warn,
    "overflow checks inspired by C which are likely to panic"
}

#[derive(Copy, Clone)]
pub struct OverflowCheckConditional;

impl LintPass for OverflowCheckConditional {
    fn get_lints(&self) -> LintArray {
        lint_array!(OVERFLOW_CHECK_CONDITIONAL)
    }
}

impl<'a, 'tcx> LateLintPass<'a, 'tcx> for OverflowCheckConditional {
    // a + b < a, a > a + b, a < a - b, a - b > a
    fn check_expr(&mut self, cx: &LateContext<'a, 'tcx>, expr: &'tcx Expr) {
        if_let_chain! {[
            let Expr_::ExprBinary(ref op, ref first, ref second) = expr.node,
            let Expr_::ExprBinary(ref op2, ref ident1, ref ident2) = first.node,
            let Expr_::ExprPath(QPath::Resolved(_, ref path1)) = ident1.node,
            let Expr_::ExprPath(QPath::Resolved(_, ref path2)) = ident2.node,
            let Expr_::ExprPath(QPath::Resolved(_, ref path3)) = second.node,
            &path1.segments[0] == &path3.segments[0] || &path2.segments[0] == &path3.segments[0],
            cx.tcx.tables().expr_ty(ident1).is_integral(),
            cx.tcx.tables().expr_ty(ident2).is_integral()
        ], {
            if let BinOp_::BiLt = op.node {
                if let BinOp_::BiAdd = op2.node {
                    span_lint(cx, OVERFLOW_CHECK_CONDITIONAL, expr.span,
                        "You are trying to use classic C overflow conditions that will fail in Rust.");
                }
            }
            if let BinOp_::BiGt = op.node {
                if let BinOp_::BiSub = op2.node {
                    span_lint(cx, OVERFLOW_CHECK_CONDITIONAL, expr.span,
                        "You are trying to use classic C underflow conditions that will fail in Rust.");
                }
            }
        }}

        if_let_chain! {[
            let Expr_::ExprBinary(ref op, ref first, ref second) = expr.node,
            let Expr_::ExprBinary(ref op2, ref ident1, ref ident2) = second.node,
            let Expr_::ExprPath(QPath::Resolved(_, ref path1)) = ident1.node,
            let Expr_::ExprPath(QPath::Resolved(_, ref path2)) = ident2.node,
            let Expr_::ExprPath(QPath::Resolved(_, ref path3)) = first.node,
            &path1.segments[0] == &path3.segments[0] || &path2.segments[0] == &path3.segments[0],
            cx.tcx.tables().expr_ty(ident1).is_integral(),
            cx.tcx.tables().expr_ty(ident2).is_integral()
        ], {
            if let BinOp_::BiGt = op.node {
                if let BinOp_::BiAdd = op2.node {
                    span_lint(cx, OVERFLOW_CHECK_CONDITIONAL, expr.span,
                        "You are trying to use classic C overflow conditions that will fail in Rust.");
                }
            }
            if let BinOp_::BiLt = op.node {
                if let BinOp_::BiSub = op2.node {
                    span_lint(cx, OVERFLOW_CHECK_CONDITIONAL, expr.span,
                        "You are trying to use classic C underflow conditions that will fail in Rust.");
                }
            }
        }}
    }
}