clippy_lints/src/missing_const_for_fn.rs

   1 use crate::utils::{is_entrypoint_fn, span_lint, trait_ref_of_method};
   2 use rustc::hir;
   3 use rustc::hir::intravisit::FnKind;
   4 use rustc::hir::{Body, Constness, FnDecl, HirId};
   5 use rustc::lint::{in_external_macro, LateContext, LateLintPass, LintArray, LintPass};
   6 use rustc::{declare_lint_pass, declare_tool_lint};
   7 use rustc_mir::transform::qualify_min_const_fn::is_min_const_fn;
   8 use syntax_pos::Span;
   9
  10 declare_clippy_lint! {
  11     /// **What it does:**
  12     ///
  13     /// Suggests the use of `const` in functions and methods where possible.
  14     ///
  15     /// **Why is this bad?**
  16     ///
  17     /// Not having the function const prevents callers of the function from being const as well.
  18     ///
  19     /// **Known problems:**
  20     ///
  21     /// Const functions are currently still being worked on, with some features only being available
  22     /// on nightly. This lint does not consider all edge cases currently and the suggestions may be
  23     /// incorrect if you are using this lint on stable.
  24     ///
  25     /// Also, the lint only runs one pass over the code. Consider these two non-const functions:
  26     ///
  27     /// ```rust
  28     /// fn a() -> i32 {
  29     ///     0
  30     /// }
  31     /// fn b() -> i32 {
  32     ///     a()
  33     /// }
  34     /// ```
  35     ///
  36     /// When running Clippy, the lint will only suggest to make `a` const, because `b` at this time
  37     /// can't be const as it calls a non-const function. Making `a` const and running Clippy again,
  38     /// will suggest to make `b` const, too.
  39     ///
  40     /// **Example:**
  41     ///
  42     /// ```rust
  43     /// fn new() -> Self {
  44     ///     Self { random_number: 42 }
  45     /// }
  46     /// ```
  47     ///
  48     /// Could be a const fn:
  49     ///
  50     /// ```rust
  51     /// const fn new() -> Self {
  52     ///     Self { random_number: 42 }
  53     /// }
  54     /// ```
  55     pub MISSING_CONST_FOR_FN,
  56     nursery,
  57     "Lint functions definitions that could be made `const fn`"
  58 }
  59
  60 declare_lint_pass!(MissingConstForFn => [MISSING_CONST_FOR_FN]);
  61
  62 impl<'a, 'tcx> LateLintPass<'a, 'tcx> for MissingConstForFn {
  63     fn check_fn(
  64         &mut self,
  65         cx: &LateContext<'_, '_>,
  66         kind: FnKind<'_>,
  67         _: &FnDecl,
  68         _: &Body,
  69         span: Span,
  70         hir_id: HirId,
  71     ) {
  72         let def_id = cx.tcx.hir().local_def_id(hir_id);
  73
  74         if in_external_macro(cx.tcx.sess, span) || is_entrypoint_fn(cx, def_id) {
  75             return;
  76         }
  77
  78         // Perform some preliminary checks that rule out constness on the Clippy side. This way we
  79         // can skip the actual const check and return early.
  80         match kind {
  81             FnKind::ItemFn(_, _, header, ..) => {
  82                 if already_const(header) {
  83                     return;
  84                 }
  85             },
  86             FnKind::Method(_, sig, ..) => {
  87                 if trait_ref_of_method(cx, hir_id).is_some() || already_const(sig.header) {
  88                     return;
  89                 }
  90             },
  91             _ => return,
  92         }
  93
  94         let mir = cx.tcx.optimized_mir(def_id);
  95
  96         if let Err((span, err)) = is_min_const_fn(cx.tcx, def_id, &mir) {
  97             if cx.tcx.is_min_const_fn(def_id) {
  98                 cx.tcx.sess.span_err(span, &err);
  99             }
 100         } else {
 101             span_lint(cx, MISSING_CONST_FOR_FN, span, "this could be a const_fn");
 102         }
 103     }
 104 }
 105
 106 // We don't have to lint on something that's already `const`
 107 fn already_const(header: hir::FnHeader) -> bool {
 108     header.constness == Constness::Const
 109 }