1 use clippy_utils::diagnostics::span_lint_and_sugg;
2 use clippy_utils::source::snippet;
3 use if_chain::if_chain;
4 use rustc_ast::ast::{LitFloatType, LitIntType, LitKind};
5 use rustc_errors::Applicability;
7 intravisit::{walk_expr, walk_stmt, NestedVisitorMap, Visitor},
8 Body, Expr, ExprKind, HirId, Lit, Stmt, StmtKind,
10 use rustc_lint::{LateContext, LateLintPass};
13 ty::{self, FloatTy, IntTy, PolyFnSig, Ty},
15 use rustc_session::{declare_lint_pass, declare_tool_lint};
17 declare_clippy_lint! {
18 /// **What it does:** Checks for usage of unconstrained numeric literals which may cause default numeric fallback in type
21 /// Default numeric fallback means that if numeric types have not yet been bound to concrete
22 /// types at the end of type inference, then integer type is bound to `i32`, and similarly
23 /// floating type is bound to `f64`.
25 /// See [RFC0212](https://github.com/rust-lang/rfcs/blob/master/text/0212-restore-int-fallback.md) for more information about the fallback.
27 /// **Why is this bad?** For those who are very careful about types, default numeric fallback
28 /// can be a pitfall that cause unexpected runtime behavior.
30 /// **Known problems:** This lint can only be allowed at the function level or above.
43 pub DEFAULT_NUMERIC_FALLBACK,
45 "usage of unconstrained numeric literals which may cause default numeric fallback."
48 declare_lint_pass!(DefaultNumericFallback => [DEFAULT_NUMERIC_FALLBACK]);
50 impl LateLintPass<'_> for DefaultNumericFallback {
51 fn check_body(&mut self, cx: &LateContext<'tcx>, body: &'tcx Body<'_>) {
52 let mut visitor = NumericFallbackVisitor::new(cx);
53 visitor.visit_body(body);
57 struct NumericFallbackVisitor<'a, 'tcx> {
58 /// Stack manages type bound of exprs. The top element holds current expr type.
59 ty_bounds: Vec<TyBound<'tcx>>,
61 cx: &'a LateContext<'tcx>,
64 impl<'a, 'tcx> NumericFallbackVisitor<'a, 'tcx> {
65 fn new(cx: &'a LateContext<'tcx>) -> Self {
67 ty_bounds: vec![TyBound::Nothing],
72 /// Check whether a passed literal has potential to cause fallback or not.
73 fn check_lit(&self, lit: &Lit, lit_ty: Ty<'tcx>) {
75 if let Some(ty_bound) = self.ty_bounds.last();
77 LitKind::Int(_, LitIntType::Unsuffixed) | LitKind::Float(_, LitFloatType::Unsuffixed));
78 if !ty_bound.is_integral();
80 let suffix = match lit_ty.kind() {
81 ty::Int(IntTy::I32) => "i32",
82 ty::Float(FloatTy::F64) => "f64",
83 // Default numeric fallback never results in other types.
87 let sugg = format!("{}_{}", snippet(self.cx, lit.span, ""), suffix);
90 DEFAULT_NUMERIC_FALLBACK,
92 "default numeric fallback might occur",
93 "consider adding suffix",
95 Applicability::MaybeIncorrect,
102 impl<'a, 'tcx> Visitor<'tcx> for NumericFallbackVisitor<'a, 'tcx> {
103 type Map = Map<'tcx>;
105 #[allow(clippy::too_many_lines)]
106 fn visit_expr(&mut self, expr: &'tcx Expr<'_>) {
108 ExprKind::Call(func, args) => {
109 if let Some(fn_sig) = fn_sig_opt(self.cx, func.hir_id) {
110 for (expr, bound) in args.iter().zip(fn_sig.skip_binder().inputs().iter()) {
111 // Push found arg type, then visit arg.
112 self.ty_bounds.push(TyBound::Ty(bound));
113 self.visit_expr(expr);
114 self.ty_bounds.pop();
120 ExprKind::MethodCall(_, _, args, _) => {
121 if let Some(def_id) = self.cx.typeck_results().type_dependent_def_id(expr.hir_id) {
122 let fn_sig = self.cx.tcx.fn_sig(def_id).skip_binder();
123 for (expr, bound) in args.iter().zip(fn_sig.inputs().iter()) {
124 self.ty_bounds.push(TyBound::Ty(bound));
125 self.visit_expr(expr);
126 self.ty_bounds.pop();
132 ExprKind::Struct(_, fields, base) => {
134 let ty = self.cx.typeck_results().expr_ty(expr);
135 if let Some(adt_def) = ty.ty_adt_def();
136 if adt_def.is_struct();
137 if let Some(variant) = adt_def.variants.iter().next();
139 let fields_def = &variant.fields;
141 // Push field type then visit each field expr.
142 for field in fields.iter() {
147 if f_def.ident == field.ident
148 { Some(self.cx.tcx.type_of(f_def.did)) }
151 self.ty_bounds.push(bound.into());
152 self.visit_expr(field.expr);
153 self.ty_bounds.pop();
156 // Visit base with no bound.
157 if let Some(base) = base {
158 self.ty_bounds.push(TyBound::Nothing);
159 self.visit_expr(base);
160 self.ty_bounds.pop();
167 ExprKind::Lit(lit) => {
168 let ty = self.cx.typeck_results().expr_ty(expr);
169 self.check_lit(lit, ty);
176 walk_expr(self, expr);
179 fn visit_stmt(&mut self, stmt: &'tcx Stmt<'_>) {
181 StmtKind::Local(local) => {
182 if local.ty.is_some() {
183 self.ty_bounds.push(TyBound::Any)
185 self.ty_bounds.push(TyBound::Nothing)
189 _ => self.ty_bounds.push(TyBound::Nothing),
192 walk_stmt(self, stmt);
193 self.ty_bounds.pop();
196 fn nested_visit_map(&mut self) -> NestedVisitorMap<Self::Map> {
197 NestedVisitorMap::None
201 fn fn_sig_opt<'tcx>(cx: &LateContext<'tcx>, hir_id: HirId) -> Option<PolyFnSig<'tcx>> {
202 let node_ty = cx.typeck_results().node_type_opt(hir_id)?;
203 // We can't use `TyS::fn_sig` because it automatically performs substs, this may result in FNs.
204 match node_ty.kind() {
205 ty::FnDef(def_id, _) => Some(cx.tcx.fn_sig(*def_id)),
206 ty::FnPtr(fn_sig) => Some(*fn_sig),
211 #[derive(Debug, Clone, Copy)]
218 impl<'tcx> TyBound<'tcx> {
219 fn is_integral(self) -> bool {
221 TyBound::Any => true,
222 TyBound::Ty(t) => t.is_integral(),
223 TyBound::Nothing => false,
228 impl<'tcx> From<Option<Ty<'tcx>>> for TyBound<'tcx> {
229 fn from(v: Option<Ty<'tcx>>) -> Self {
231 Some(t) => TyBound::Ty(t),
232 None => TyBound::Nothing,