clippy_lints/src/indexing_slicing.rs

   1 //! lint on indexing and slicing operations
   2
   3 use clippy_utils::consts::{constant, Constant};
   4 use clippy_utils::diagnostics::{span_lint, span_lint_and_help};
   5 use clippy_utils::higher;
   6 use rustc_ast::ast::RangeLimits;
   7 use rustc_hir::{Expr, ExprKind};
   8 use rustc_lint::{LateContext, LateLintPass};
   9 use rustc_middle::ty;
  10 use rustc_session::{declare_lint_pass, declare_tool_lint};
  11
  12 declare_clippy_lint! {
  13     /// ### What it does
  14     /// Checks for out of bounds array indexing with a constant
  15     /// index.
  16     ///
  17     /// ### Why is this bad?
  18     /// This will always panic at runtime.
  19     ///
  20     /// ### Known problems
  21     /// Hopefully none.
  22     ///
  23     /// ### Example
  24     /// ```no_run
  25     /// # #![allow(const_err)]
  26     /// let x = [1, 2, 3, 4];
  27     ///
  28     /// // Bad
  29     /// x[9];
  30     /// &x[2..9];
  31     ///
  32     /// // Good
  33     /// x[0];
  34     /// x[3];
  35     /// ```
  36     #[clippy::version = "pre 1.29.0"]
  37     pub OUT_OF_BOUNDS_INDEXING,
  38     correctness,
  39     "out of bounds constant indexing"
  40 }
  41
  42 declare_clippy_lint! {
  43     /// ### What it does
  44     /// Checks for usage of indexing or slicing. Arrays are special cases, this lint
  45     /// does report on arrays if we can tell that slicing operations are in bounds and does not
  46     /// lint on constant `usize` indexing on arrays because that is handled by rustc's `const_err` lint.
  47     ///
  48     /// ### Why is this bad?
  49     /// Indexing and slicing can panic at runtime and there are
  50     /// safe alternatives.
  51     ///
  52     /// ### Known problems
  53     /// Hopefully none.
  54     ///
  55     /// ### Example
  56     /// ```rust,no_run
  57     /// // Vector
  58     /// let x = vec![0; 5];
  59     ///
  60     /// // Bad
  61     /// x[2];
  62     /// &x[2..100];
  63     /// &x[2..];
  64     /// &x[..100];
  65     ///
  66     /// // Good
  67     /// x.get(2);
  68     /// x.get(2..100);
  69     /// x.get(2..);
  70     /// x.get(..100);
  71     ///
  72     /// // Array
  73     /// let y = [0, 1, 2, 3];
  74     ///
  75     /// // Bad
  76     /// &y[10..100];
  77     /// &y[10..];
  78     /// &y[..100];
  79     ///
  80     /// // Good
  81     /// &y[2..];
  82     /// &y[..2];
  83     /// &y[0..3];
  84     /// y.get(10);
  85     /// y.get(10..100);
  86     /// y.get(10..);
  87     /// y.get(..100);
  88     /// ```
  89     #[clippy::version = "pre 1.29.0"]
  90     pub INDEXING_SLICING,
  91     restriction,
  92     "indexing/slicing usage"
  93 }
  94
  95 declare_lint_pass!(IndexingSlicing => [INDEXING_SLICING, OUT_OF_BOUNDS_INDEXING]);
  96
  97 impl<'tcx> LateLintPass<'tcx> for IndexingSlicing {
  98     fn check_expr(&mut self, cx: &LateContext<'tcx>, expr: &'tcx Expr<'_>) {
  99         if let ExprKind::Index(array, index) = &expr.kind {
 100             let ty = cx.typeck_results().expr_ty(array).peel_refs();
 101             if let Some(range) = higher::Range::hir(index) {
 102                 // Ranged indexes, i.e., &x[n..m], &x[n..], &x[..n] and &x[..]
 103                 if let ty::Array(_, s) = ty.kind() {
 104                     let size: u128 = if let Some(size) = s.try_eval_usize(cx.tcx, cx.param_env) {
 105                         size.into()
 106                     } else {
 107                         return;
 108                     };
 109
 110                     let const_range = to_const_range(cx, range, size);
 111
 112                     if let (Some(start), _) = const_range {
 113                         if start > size {
 114                             span_lint(
 115                                 cx,
 116                                 OUT_OF_BOUNDS_INDEXING,
 117                                 range.start.map_or(expr.span, |start| start.span),
 118                                 "range is out of bounds",
 119                             );
 120                             return;
 121                         }
 122                     }
 123
 124                     if let (_, Some(end)) = const_range {
 125                         if end > size {
 126                             span_lint(
 127                                 cx,
 128                                 OUT_OF_BOUNDS_INDEXING,
 129                                 range.end.map_or(expr.span, |end| end.span),
 130                                 "range is out of bounds",
 131                             );
 132                             return;
 133                         }
 134                     }
 135
 136                     if let (Some(_), Some(_)) = const_range {
 137                         // early return because both start and end are constants
 138                         // and we have proven above that they are in bounds
 139                         return;
 140                     }
 141                 }
 142
 143                 let help_msg = match (range.start, range.end) {
 144                     (None, Some(_)) => "consider using `.get(..n)`or `.get_mut(..n)` instead",
 145                     (Some(_), None) => "consider using `.get(n..)` or .get_mut(n..)` instead",
 146                     (Some(_), Some(_)) => "consider using `.get(n..m)` or `.get_mut(n..m)` instead",
 147                     (None, None) => return, // [..] is ok.
 148                 };
 149
 150                 span_lint_and_help(cx, INDEXING_SLICING, expr.span, "slicing may panic", None, help_msg);
 151             } else {
 152                 // Catchall non-range index, i.e., [n] or [n << m]
 153                 if let ty::Array(..) = ty.kind() {
 154                     // Index is a constant uint.
 155                     if let Some(..) = constant(cx, cx.typeck_results(), index) {
 156                         // Let rustc's `const_err` lint handle constant `usize` indexing on arrays.
 157                         return;
 158                     }
 159                 }
 160
 161                 span_lint_and_help(
 162                     cx,
 163                     INDEXING_SLICING,
 164                     expr.span,
 165                     "indexing may panic",
 166                     None,
 167                     "consider using `.get(n)` or `.get_mut(n)` instead",
 168                 );
 169             }
 170         }
 171     }
 172 }
 173
 174 /// Returns a tuple of options with the start and end (exclusive) values of
 175 /// the range. If the start or end is not constant, None is returned.
 176 fn to_const_range<'tcx>(
 177     cx: &LateContext<'tcx>,
 178     range: higher::Range<'_>,
 179     array_size: u128,
 180 ) -> (Option<u128>, Option<u128>) {
 181     let s = range
 182         .start
 183         .map(|expr| constant(cx, cx.typeck_results(), expr).map(|(c, _)| c));
 184     let start = match s {
 185         Some(Some(Constant::Int(x))) => Some(x),
 186         Some(_) => None,
 187         None => Some(0),
 188     };
 189
 190     let e = range
 191         .end
 192         .map(|expr| constant(cx, cx.typeck_results(), expr).map(|(c, _)| c));
 193     let end = match e {
 194         Some(Some(Constant::Int(x))) => {
 195             if range.limits == RangeLimits::Closed {
 196                 Some(x + 1)
 197             } else {
 198                 Some(x)
 199             }
 200         },
 201         Some(_) => None,
 202         None => Some(array_size),
 203     };
 204
 205     (start, end)
 206 }