compiler/rustc_lint/src/types.rs

   1 use crate::{LateContext, LateLintPass, LintContext};
   2 use rustc_ast as ast;
   3 use rustc_attr as attr;
   4 use rustc_data_structures::fx::FxHashSet;
   5 use rustc_errors::Applicability;
   6 use rustc_hir as hir;
   7 use rustc_hir::{is_range_literal, ExprKind, Node};
   8 use rustc_middle::ty::layout::{IntegerExt, SizeSkeleton};
   9 use rustc_middle::ty::subst::SubstsRef;
  10 use rustc_middle::ty::{self, AdtKind, Ty, TyCtxt, TypeFoldable};
  11 use rustc_span::source_map;
  12 use rustc_span::symbol::sym;
  13 use rustc_span::{Span, DUMMY_SP};
  14 use rustc_target::abi::Abi;
  15 use rustc_target::abi::{Integer, LayoutOf, TagEncoding, Variants};
  16 use rustc_target::spec::abi::Abi as SpecAbi;
  17
  18 use std::cmp;
  19 use std::iter;
  20 use std::ops::ControlFlow;
  21 use tracing::debug;
  22
  23 declare_lint! {
  24     /// The `unused_comparisons` lint detects comparisons made useless by
  25     /// limits of the types involved.
  26     ///
  27     /// ### Example
  28     ///
  29     /// ```rust
  30     /// fn foo(x: u8) {
  31     ///     x >= 0;
  32     /// }
  33     /// ```
  34     ///
  35     /// {{produces}}
  36     ///
  37     /// ### Explanation
  38     ///
  39     /// A useless comparison may indicate a mistake, and should be fixed or
  40     /// removed.
  41     UNUSED_COMPARISONS,
  42     Warn,
  43     "comparisons made useless by limits of the types involved"
  44 }
  45
  46 declare_lint! {
  47     /// The `overflowing_literals` lint detects literal out of range for its
  48     /// type.
  49     ///
  50     /// ### Example
  51     ///
  52     /// ```rust,compile_fail
  53     /// let x: u8 = 1000;
  54     /// ```
  55     ///
  56     /// {{produces}}
  57     ///
  58     /// ### Explanation
  59     ///
  60     /// It is usually a mistake to use a literal that overflows the type where
  61     /// it is used. Either use a literal that is within range, or change the
  62     /// type to be within the range of the literal.
  63     OVERFLOWING_LITERALS,
  64     Deny,
  65     "literal out of range for its type"
  66 }
  67
  68 declare_lint! {
  69     /// The `variant_size_differences` lint detects enums with widely varying
  70     /// variant sizes.
  71     ///
  72     /// ### Example
  73     ///
  74     /// ```rust,compile_fail
  75     /// #![deny(variant_size_differences)]
  76     /// enum En {
  77     ///     V0(u8),
  78     ///     VBig([u8; 1024]),
  79     /// }
  80     /// ```
  81     ///
  82     /// {{produces}}
  83     ///
  84     /// ### Explanation
  85     ///
  86     /// It can be a mistake to add a variant to an enum that is much larger
  87     /// than the other variants, bloating the overall size required for all
  88     /// variants. This can impact performance and memory usage. This is
  89     /// triggered if one variant is more than 3 times larger than the
  90     /// second-largest variant.
  91     ///
  92     /// Consider placing the large variant's contents on the heap (for example
  93     /// via [`Box`]) to keep the overall size of the enum itself down.
  94     ///
  95     /// This lint is "allow" by default because it can be noisy, and may not be
  96     /// an actual problem. Decisions about this should be guided with
  97     /// profiling and benchmarking.
  98     ///
  99     /// [`Box`]: https://doc.rust-lang.org/std/boxed/index.html
 100     VARIANT_SIZE_DIFFERENCES,
 101     Allow,
 102     "detects enums with widely varying variant sizes"
 103 }
 104
 105 #[derive(Copy, Clone)]
 106 pub struct TypeLimits {
 107     /// Id of the last visited negated expression
 108     negated_expr_id: Option<hir::HirId>,
 109 }
 110
 111 impl_lint_pass!(TypeLimits => [UNUSED_COMPARISONS, OVERFLOWING_LITERALS]);
 112
 113 impl TypeLimits {
 114     pub fn new() -> TypeLimits {
 115         TypeLimits { negated_expr_id: None }
 116     }
 117 }
 118
 119 /// Attempts to special-case the overflowing literal lint when it occurs as a range endpoint.
 120 /// Returns `true` iff the lint was overridden.
 121 fn lint_overflowing_range_endpoint<'tcx>(
 122     cx: &LateContext<'tcx>,
 123     lit: &hir::Lit,
 124     lit_val: u128,
 125     max: u128,
 126     expr: &'tcx hir::Expr<'tcx>,
 127     parent_expr: &'tcx hir::Expr<'tcx>,
 128     ty: &str,
 129 ) -> bool {
 130     // We only want to handle exclusive (`..`) ranges,
 131     // which are represented as `ExprKind::Struct`.
 132     let mut overwritten = false;
 133     if let ExprKind::Struct(_, eps, _) = &parent_expr.kind {
 134         if eps.len() != 2 {
 135             return false;
 136         }
 137         // We can suggest using an inclusive range
 138         // (`..=`) instead only if it is the `end` that is
 139         // overflowing and only by 1.
 140         if eps[1].expr.hir_id == expr.hir_id && lit_val - 1 == max {
 141             cx.struct_span_lint(OVERFLOWING_LITERALS, parent_expr.span, |lint| {
 142                 let mut err = lint.build(&format!("range endpoint is out of range for `{}`", ty));
 143                 if let Ok(start) = cx.sess().source_map().span_to_snippet(eps[0].span) {
 144                     use ast::{LitIntType, LitKind};
 145                     // We need to preserve the literal's suffix,
 146                     // as it may determine typing information.
 147                     let suffix = match lit.node {
 148                         LitKind::Int(_, LitIntType::Signed(s)) => s.name_str(),
 149                         LitKind::Int(_, LitIntType::Unsigned(s)) => s.name_str(),
 150                         LitKind::Int(_, LitIntType::Unsuffixed) => "",
 151                         _ => bug!(),
 152                     };
 153                     let suggestion = format!("{}..={}{}", start, lit_val - 1, suffix);
 154                     err.span_suggestion(
 155                         parent_expr.span,
 156                         &"use an inclusive range instead",
 157                         suggestion,
 158                         Applicability::MachineApplicable,
 159                     );
 160                     err.emit();
 161                     overwritten = true;
 162                 }
 163             });
 164         }
 165     }
 166     overwritten
 167 }
 168
 169 // For `isize` & `usize`, be conservative with the warnings, so that the
 170 // warnings are consistent between 32- and 64-bit platforms.
 171 fn int_ty_range(int_ty: ty::IntTy) -> (i128, i128) {
 172     match int_ty {
 173         ty::IntTy::Isize => (i64::MIN.into(), i64::MAX.into()),
 174         ty::IntTy::I8 => (i8::MIN.into(), i8::MAX.into()),
 175         ty::IntTy::I16 => (i16::MIN.into(), i16::MAX.into()),
 176         ty::IntTy::I32 => (i32::MIN.into(), i32::MAX.into()),
 177         ty::IntTy::I64 => (i64::MIN.into(), i64::MAX.into()),
 178         ty::IntTy::I128 => (i128::MIN, i128::MAX),
 179     }
 180 }
 181
 182 fn uint_ty_range(uint_ty: ty::UintTy) -> (u128, u128) {
 183     let max = match uint_ty {
 184         ty::UintTy::Usize => u64::MAX.into(),
 185         ty::UintTy::U8 => u8::MAX.into(),
 186         ty::UintTy::U16 => u16::MAX.into(),
 187         ty::UintTy::U32 => u32::MAX.into(),
 188         ty::UintTy::U64 => u64::MAX.into(),
 189         ty::UintTy::U128 => u128::MAX,
 190     };
 191     (0, max)
 192 }
 193
 194 fn get_bin_hex_repr(cx: &LateContext<'_>, lit: &hir::Lit) -> Option<String> {
 195     let src = cx.sess().source_map().span_to_snippet(lit.span).ok()?;
 196     let firstch = src.chars().next()?;
 197
 198     if firstch == '0' {
 199         match src.chars().nth(1) {
 200             Some('x' | 'b') => return Some(src),
 201             _ => return None,
 202         }
 203     }
 204
 205     None
 206 }
 207
 208 fn report_bin_hex_error(
 209     cx: &LateContext<'_>,
 210     expr: &hir::Expr<'_>,
 211     ty: attr::IntType,
 212     repr_str: String,
 213     val: u128,
 214     negative: bool,
 215 ) {
 216     let size = Integer::from_attr(&cx.tcx, ty).size();
 217     cx.struct_span_lint(OVERFLOWING_LITERALS, expr.span, |lint| {
 218         let (t, actually) = match ty {
 219             attr::IntType::SignedInt(t) => {
 220                 let actually = if negative {
 221                     -(size.sign_extend(val) as i128)
 222                 } else {
 223                     size.sign_extend(val) as i128
 224                 };
 225                 (t.name_str(), actually.to_string())
 226             }
 227             attr::IntType::UnsignedInt(t) => {
 228                 let actually = size.truncate(val);
 229                 (t.name_str(), actually.to_string())
 230             }
 231         };
 232         let mut err = lint.build(&format!("literal out of range for `{}`", t));
 233         if negative {
 234             // If the value is negative,
 235             // emits a note about the value itself, apart from the literal.
 236             err.note(&format!(
 237                 "the literal `{}` (decimal `{}`) does not fit into \
 238                  the type `{}`",
 239                 repr_str, val, t
 240             ));
 241             err.note(&format!("and the value `-{}` will become `{}{}`", repr_str, actually, t));
 242         } else {
 243             err.note(&format!(
 244                 "the literal `{}` (decimal `{}`) does not fit into \
 245                  the type `{}` and will become `{}{}`",
 246                 repr_str, val, t, actually, t
 247             ));
 248         }
 249         if let Some(sugg_ty) =
 250             get_type_suggestion(&cx.typeck_results().node_type(expr.hir_id), val, negative)
 251         {
 252             if let Some(pos) = repr_str.chars().position(|c| c == 'i' || c == 'u') {
 253                 let (sans_suffix, _) = repr_str.split_at(pos);
 254                 err.span_suggestion(
 255                     expr.span,
 256                     &format!("consider using the type `{}` instead", sugg_ty),
 257                     format!("{}{}", sans_suffix, sugg_ty),
 258                     Applicability::MachineApplicable,
 259                 );
 260             } else {
 261                 err.help(&format!("consider using the type `{}` instead", sugg_ty));
 262             }
 263         }
 264         err.emit();
 265     });
 266 }
 267
 268 // This function finds the next fitting type and generates a suggestion string.
 269 // It searches for fitting types in the following way (`X < Y`):
 270 //  - `iX`: if literal fits in `uX` => `uX`, else => `iY`
 271 //  - `-iX` => `iY`
 272 //  - `uX` => `uY`
 273 //
 274 // No suggestion for: `isize`, `usize`.
 275 fn get_type_suggestion(t: Ty<'_>, val: u128, negative: bool) -> Option<&'static str> {
 276     use ty::IntTy::*;
 277     use ty::UintTy::*;
 278     macro_rules! find_fit {
 279         ($ty:expr, $val:expr, $negative:expr,
 280          $($type:ident => [$($utypes:expr),*] => [$($itypes:expr),*]),+) => {
 281             {
 282                 let _neg = if negative { 1 } else { 0 };
 283                 match $ty {
 284                     $($type => {
 285                         $(if !negative && val <= uint_ty_range($utypes).1 {
 286                             return Some($utypes.name_str())
 287                         })*
 288                         $(if val <= int_ty_range($itypes).1 as u128 + _neg {
 289                             return Some($itypes.name_str())
 290                         })*
 291                         None
 292                     },)+
 293                     _ => None
 294                 }
 295             }
 296         }
 297     }
 298     match t.kind() {
 299         ty::Int(i) => find_fit!(i, val, negative,
 300                       I8 => [U8] => [I16, I32, I64, I128],
 301                       I16 => [U16] => [I32, I64, I128],
 302                       I32 => [U32] => [I64, I128],
 303                       I64 => [U64] => [I128],
 304                       I128 => [U128] => []),
 305         ty::Uint(u) => find_fit!(u, val, negative,
 306                       U8 => [U8, U16, U32, U64, U128] => [],
 307                       U16 => [U16, U32, U64, U128] => [],
 308                       U32 => [U32, U64, U128] => [],
 309                       U64 => [U64, U128] => [],
 310                       U128 => [U128] => []),
 311         _ => None,
 312     }
 313 }
 314
 315 fn lint_int_literal<'tcx>(
 316     cx: &LateContext<'tcx>,
 317     type_limits: &TypeLimits,
 318     e: &'tcx hir::Expr<'tcx>,
 319     lit: &hir::Lit,
 320     t: ty::IntTy,
 321     v: u128,
 322 ) {
 323     let int_type = t.normalize(cx.sess().target.pointer_width);
 324     let (min, max) = int_ty_range(int_type);
 325     let max = max as u128;
 326     let negative = type_limits.negated_expr_id == Some(e.hir_id);
 327
 328     // Detect literal value out of range [min, max] inclusive
 329     // avoiding use of -min to prevent overflow/panic
 330     if (negative && v > max + 1) || (!negative && v > max) {
 331         if let Some(repr_str) = get_bin_hex_repr(cx, lit) {
 332             report_bin_hex_error(
 333                 cx,
 334                 e,
 335                 attr::IntType::SignedInt(ty::ast_int_ty(t)),
 336                 repr_str,
 337                 v,
 338                 negative,
 339             );
 340             return;
 341         }
 342
 343         let par_id = cx.tcx.hir().get_parent_node(e.hir_id);
 344         if let Node::Expr(par_e) = cx.tcx.hir().get(par_id) {
 345             if let hir::ExprKind::Struct(..) = par_e.kind {
 346                 if is_range_literal(par_e)
 347                     && lint_overflowing_range_endpoint(cx, lit, v, max, e, par_e, t.name_str())
 348                 {
 349                     // The overflowing literal lint was overridden.
 350                     return;
 351                 }
 352             }
 353         }
 354
 355         cx.struct_span_lint(OVERFLOWING_LITERALS, e.span, |lint| {
 356             let mut err = lint.build(&format!("literal out of range for `{}`", t.name_str()));
 357             err.note(&format!(
 358                 "the literal `{}` does not fit into the type `{}` whose range is `{}..={}`",
 359                 cx.sess()
 360                     .source_map()
 361                     .span_to_snippet(lit.span)
 362                     .expect("must get snippet from literal"),
 363                 t.name_str(),
 364                 min,
 365                 max,
 366             ));
 367             if let Some(sugg_ty) =
 368                 get_type_suggestion(&cx.typeck_results().node_type(e.hir_id), v, negative)
 369             {
 370                 err.help(&format!("consider using the type `{}` instead", sugg_ty));
 371             }
 372             err.emit();
 373         });
 374     }
 375 }
 376
 377 fn lint_uint_literal<'tcx>(
 378     cx: &LateContext<'tcx>,
 379     e: &'tcx hir::Expr<'tcx>,
 380     lit: &hir::Lit,
 381     t: ty::UintTy,
 382 ) {
 383     let uint_type = t.normalize(cx.sess().target.pointer_width);
 384     let (min, max) = uint_ty_range(uint_type);
 385     let lit_val: u128 = match lit.node {
 386         // _v is u8, within range by definition
 387         ast::LitKind::Byte(_v) => return,
 388         ast::LitKind::Int(v, _) => v,
 389         _ => bug!(),
 390     };
 391     if lit_val < min || lit_val > max {
 392         let parent_id = cx.tcx.hir().get_parent_node(e.hir_id);
 393         if let Node::Expr(par_e) = cx.tcx.hir().get(parent_id) {
 394             match par_e.kind {
 395                 hir::ExprKind::Cast(..) => {
 396                     if let ty::Char = cx.typeck_results().expr_ty(par_e).kind() {
 397                         cx.struct_span_lint(OVERFLOWING_LITERALS, par_e.span, |lint| {
 398                             lint.build("only `u8` can be cast into `char`")
 399                                 .span_suggestion(
 400                                     par_e.span,
 401                                     &"use a `char` literal instead",
 402                                     format!("'\\u{{{:X}}}'", lit_val),
 403                                     Applicability::MachineApplicable,
 404                                 )
 405                                 .emit();
 406                         });
 407                         return;
 408                     }
 409                 }
 410                 hir::ExprKind::Struct(..) if is_range_literal(par_e) => {
 411                     let t = t.name_str();
 412                     if lint_overflowing_range_endpoint(cx, lit, lit_val, max, e, par_e, t) {
 413                         // The overflowing literal lint was overridden.
 414                         return;
 415                     }
 416                 }
 417                 _ => {}
 418             }
 419         }
 420         if let Some(repr_str) = get_bin_hex_repr(cx, lit) {
 421             report_bin_hex_error(
 422                 cx,
 423                 e,
 424                 attr::IntType::UnsignedInt(ty::ast_uint_ty(t)),
 425                 repr_str,
 426                 lit_val,
 427                 false,
 428             );
 429             return;
 430         }
 431         cx.struct_span_lint(OVERFLOWING_LITERALS, e.span, |lint| {
 432             lint.build(&format!("literal out of range for `{}`", t.name_str()))
 433                 .note(&format!(
 434                     "the literal `{}` does not fit into the type `{}` whose range is `{}..={}`",
 435                     cx.sess()
 436                         .source_map()
 437                         .span_to_snippet(lit.span)
 438                         .expect("must get snippet from literal"),
 439                     t.name_str(),
 440                     min,
 441                     max,
 442                 ))
 443                 .emit()
 444         });
 445     }
 446 }
 447
 448 fn lint_literal<'tcx>(
 449     cx: &LateContext<'tcx>,
 450     type_limits: &TypeLimits,
 451     e: &'tcx hir::Expr<'tcx>,
 452     lit: &hir::Lit,
 453 ) {
 454     match *cx.typeck_results().node_type(e.hir_id).kind() {
 455         ty::Int(t) => {
 456             match lit.node {
 457                 ast::LitKind::Int(v, ast::LitIntType::Signed(_) | ast::LitIntType::Unsuffixed) => {
 458                     lint_int_literal(cx, type_limits, e, lit, t, v)
 459                 }
 460                 _ => bug!(),
 461             };
 462         }
 463         ty::Uint(t) => lint_uint_literal(cx, e, lit, t),
 464         ty::Float(t) => {
 465             let is_infinite = match lit.node {
 466                 ast::LitKind::Float(v, _) => match t {
 467                     ty::FloatTy::F32 => v.as_str().parse().map(f32::is_infinite),
 468                     ty::FloatTy::F64 => v.as_str().parse().map(f64::is_infinite),
 469                 },
 470                 _ => bug!(),
 471             };
 472             if is_infinite == Ok(true) {
 473                 cx.struct_span_lint(OVERFLOWING_LITERALS, e.span, |lint| {
 474                     lint.build(&format!("literal out of range for `{}`", t.name_str()))
 475                         .note(&format!(
 476                             "the literal `{}` does not fit into the type `{}` and will be converted to `{}::INFINITY`",
 477                             cx.sess()
 478                                 .source_map()
 479                                 .span_to_snippet(lit.span)
 480                                 .expect("must get snippet from literal"),
 481                             t.name_str(),
 482                             t.name_str(),
 483                         ))
 484                         .emit();
 485                 });
 486             }
 487         }
 488         _ => {}
 489     }
 490 }
 491
 492 impl<'tcx> LateLintPass<'tcx> for TypeLimits {
 493     fn check_expr(&mut self, cx: &LateContext<'tcx>, e: &'tcx hir::Expr<'tcx>) {
 494         match e.kind {
 495             hir::ExprKind::Unary(hir::UnOp::Neg, ref expr) => {
 496                 // propagate negation, if the negation itself isn't negated
 497                 if self.negated_expr_id != Some(e.hir_id) {
 498                     self.negated_expr_id = Some(expr.hir_id);
 499                 }
 500             }
 501             hir::ExprKind::Binary(binop, ref l, ref r) => {
 502                 if is_comparison(binop) && !check_limits(cx, binop, &l, &r) {
 503                     cx.struct_span_lint(UNUSED_COMPARISONS, e.span, |lint| {
 504                         lint.build("comparison is useless due to type limits").emit()
 505                     });
 506                 }
 507             }
 508             hir::ExprKind::Lit(ref lit) => lint_literal(cx, self, e, lit),
 509             _ => {}
 510         };
 511
 512         fn is_valid<T: cmp::PartialOrd>(binop: hir::BinOp, v: T, min: T, max: T) -> bool {
 513             match binop.node {
 514                 hir::BinOpKind::Lt => v > min && v <= max,
 515                 hir::BinOpKind::Le => v >= min && v < max,
 516                 hir::BinOpKind::Gt => v >= min && v < max,
 517                 hir::BinOpKind::Ge => v > min && v <= max,
 518                 hir::BinOpKind::Eq | hir::BinOpKind::Ne => v >= min && v <= max,
 519                 _ => bug!(),
 520             }
 521         }
 522
 523         fn rev_binop(binop: hir::BinOp) -> hir::BinOp {
 524             source_map::respan(
 525                 binop.span,
 526                 match binop.node {
 527                     hir::BinOpKind::Lt => hir::BinOpKind::Gt,
 528                     hir::BinOpKind::Le => hir::BinOpKind::Ge,
 529                     hir::BinOpKind::Gt => hir::BinOpKind::Lt,
 530                     hir::BinOpKind::Ge => hir::BinOpKind::Le,
 531                     _ => return binop,
 532                 },
 533             )
 534         }
 535
 536         fn check_limits(
 537             cx: &LateContext<'_>,
 538             binop: hir::BinOp,
 539             l: &hir::Expr<'_>,
 540             r: &hir::Expr<'_>,
 541         ) -> bool {
 542             let (lit, expr, swap) = match (&l.kind, &r.kind) {
 543                 (&hir::ExprKind::Lit(_), _) => (l, r, true),
 544                 (_, &hir::ExprKind::Lit(_)) => (r, l, false),
 545                 _ => return true,
 546             };
 547             // Normalize the binop so that the literal is always on the RHS in
 548             // the comparison
 549             let norm_binop = if swap { rev_binop(binop) } else { binop };
 550             match *cx.typeck_results().node_type(expr.hir_id).kind() {
 551                 ty::Int(int_ty) => {
 552                     let (min, max) = int_ty_range(int_ty);
 553                     let lit_val: i128 = match lit.kind {
 554                         hir::ExprKind::Lit(ref li) => match li.node {
 555                             ast::LitKind::Int(
 556                                 v,
 557                                 ast::LitIntType::Signed(_) | ast::LitIntType::Unsuffixed,
 558                             ) => v as i128,
 559                             _ => return true,
 560                         },
 561                         _ => bug!(),
 562                     };
 563                     is_valid(norm_binop, lit_val, min, max)
 564                 }
 565                 ty::Uint(uint_ty) => {
 566                     let (min, max): (u128, u128) = uint_ty_range(uint_ty);
 567                     let lit_val: u128 = match lit.kind {
 568                         hir::ExprKind::Lit(ref li) => match li.node {
 569                             ast::LitKind::Int(v, _) => v,
 570                             _ => return true,
 571                         },
 572                         _ => bug!(),
 573                     };
 574                     is_valid(norm_binop, lit_val, min, max)
 575                 }
 576                 _ => true,
 577             }
 578         }
 579
 580         fn is_comparison(binop: hir::BinOp) -> bool {
 581             matches!(
 582                 binop.node,
 583                 hir::BinOpKind::Eq
 584                     | hir::BinOpKind::Lt
 585                     | hir::BinOpKind::Le
 586                     | hir::BinOpKind::Ne
 587                     | hir::BinOpKind::Ge
 588                     | hir::BinOpKind::Gt
 589             )
 590         }
 591     }
 592 }
 593
 594 declare_lint! {
 595     /// The `improper_ctypes` lint detects incorrect use of types in foreign
 596     /// modules.
 597     ///
 598     /// ### Example
 599     ///
 600     /// ```rust
 601     /// extern "C" {
 602     ///     static STATIC: String;
 603     /// }
 604     /// ```
 605     ///
 606     /// {{produces}}
 607     ///
 608     /// ### Explanation
 609     ///
 610     /// The compiler has several checks to verify that types used in `extern`
 611     /// blocks are safe and follow certain rules to ensure proper
 612     /// compatibility with the foreign interfaces. This lint is issued when it
 613     /// detects a probable mistake in a definition. The lint usually should
 614     /// provide a description of the issue, along with possibly a hint on how
 615     /// to resolve it.
 616     IMPROPER_CTYPES,
 617     Warn,
 618     "proper use of libc types in foreign modules"
 619 }
 620
 621 declare_lint_pass!(ImproperCTypesDeclarations => [IMPROPER_CTYPES]);
 622
 623 declare_lint! {
 624     /// The `improper_ctypes_definitions` lint detects incorrect use of
 625     /// [`extern` function] definitions.
 626     ///
 627     /// [`extern` function]: https://doc.rust-lang.org/reference/items/functions.html#extern-function-qualifier
 628     ///
 629     /// ### Example
 630     ///
 631     /// ```rust
 632     /// # #![allow(unused)]
 633     /// pub extern "C" fn str_type(p: &str) { }
 634     /// ```
 635     ///
 636     /// {{produces}}
 637     ///
 638     /// ### Explanation
 639     ///
 640     /// There are many parameter and return types that may be specified in an
 641     /// `extern` function that are not compatible with the given ABI. This
 642     /// lint is an alert that these types should not be used. The lint usually
 643     /// should provide a description of the issue, along with possibly a hint
 644     /// on how to resolve it.
 645     IMPROPER_CTYPES_DEFINITIONS,
 646     Warn,
 647     "proper use of libc types in foreign item definitions"
 648 }
 649
 650 declare_lint_pass!(ImproperCTypesDefinitions => [IMPROPER_CTYPES_DEFINITIONS]);
 651
 652 #[derive(Clone, Copy)]
 653 crate enum CItemKind {
 654     Declaration,
 655     Definition,
 656 }
 657
 658 struct ImproperCTypesVisitor<'a, 'tcx> {
 659     cx: &'a LateContext<'tcx>,
 660     mode: CItemKind,
 661 }
 662
 663 enum FfiResult<'tcx> {
 664     FfiSafe,
 665     FfiPhantom(Ty<'tcx>),
 666     FfiUnsafe { ty: Ty<'tcx>, reason: String, help: Option<String> },
 667 }
 668
 669 crate fn nonnull_optimization_guaranteed<'tcx>(tcx: TyCtxt<'tcx>, def: &ty::AdtDef) -> bool {
 670     tcx.get_attrs(def.did)
 671         .iter()
 672         .any(|a| tcx.sess.check_name(a, sym::rustc_nonnull_optimization_guaranteed))
 673 }
 674
 675 /// `repr(transparent)` structs can have a single non-ZST field, this function returns that
 676 /// field.
 677 pub fn transparent_newtype_field<'a, 'tcx>(
 678     tcx: TyCtxt<'tcx>,
 679     variant: &'a ty::VariantDef,
 680 ) -> Option<&'a ty::FieldDef> {
 681     let param_env = tcx.param_env(variant.def_id);
 682     variant.fields.iter().find(|field| {
 683         let field_ty = tcx.type_of(field.did);
 684         let is_zst = tcx.layout_of(param_env.and(field_ty)).map_or(false, |layout| layout.is_zst());
 685         !is_zst
 686     })
 687 }
 688
 689 /// Is type known to be non-null?
 690 fn ty_is_known_nonnull<'tcx>(cx: &LateContext<'tcx>, ty: Ty<'tcx>, mode: CItemKind) -> bool {
 691     let tcx = cx.tcx;
 692     match ty.kind() {
 693         ty::FnPtr(_) => true,
 694         ty::Ref(..) => true,
 695         ty::Adt(def, _) if def.is_box() && matches!(mode, CItemKind::Definition) => true,
 696         ty::Adt(def, substs) if def.repr.transparent() && !def.is_union() => {
 697             let marked_non_null = nonnull_optimization_guaranteed(tcx, &def);
 698
 699             if marked_non_null {
 700                 return true;
 701             }
 702
 703             // Types with a `#[repr(no_niche)]` attribute have their niche hidden.
 704             // The attribute is used by the UnsafeCell for example (the only use so far).
 705             if def.repr.hide_niche() {
 706                 return false;
 707             }
 708
 709             def.variants
 710                 .iter()
 711                 .filter_map(|variant| transparent_newtype_field(cx.tcx, variant))
 712                 .any(|field| ty_is_known_nonnull(cx, field.ty(tcx, substs), mode))
 713         }
 714         _ => false,
 715     }
 716 }
 717
 718 /// Given a non-null scalar (or transparent) type `ty`, return the nullable version of that type.
 719 /// If the type passed in was not scalar, returns None.
 720 fn get_nullable_type<'tcx>(cx: &LateContext<'tcx>, ty: Ty<'tcx>) -> Option<Ty<'tcx>> {
 721     let tcx = cx.tcx;
 722     Some(match *ty.kind() {
 723         ty::Adt(field_def, field_substs) => {
 724             let inner_field_ty = {
 725                 let first_non_zst_ty =
 726                     field_def.variants.iter().filter_map(|v| transparent_newtype_field(cx.tcx, v));
 727                 debug_assert_eq!(
 728                     first_non_zst_ty.clone().count(),
 729                     1,
 730                     "Wrong number of fields for transparent type"
 731                 );
 732                 first_non_zst_ty
 733                     .last()
 734                     .expect("No non-zst fields in transparent type.")
 735                     .ty(tcx, field_substs)
 736             };
 737             return get_nullable_type(cx, inner_field_ty);
 738         }
 739         ty::Int(ty) => tcx.mk_mach_int(ty),
 740         ty::Uint(ty) => tcx.mk_mach_uint(ty),
 741         ty::RawPtr(ty_mut) => tcx.mk_ptr(ty_mut),
 742         // As these types are always non-null, the nullable equivalent of
 743         // Option<T> of these types are their raw pointer counterparts.
 744         ty::Ref(_region, ty, mutbl) => tcx.mk_ptr(ty::TypeAndMut { ty, mutbl }),
 745         ty::FnPtr(..) => {
 746             // There is no nullable equivalent for Rust's function pointers -- you
 747             // must use an Option<fn(..) -> _> to represent it.
 748             ty
 749         }
 750
 751         // We should only ever reach this case if ty_is_known_nonnull is extended
 752         // to other types.
 753         ref unhandled => {
 754             debug!(
 755                 "get_nullable_type: Unhandled scalar kind: {:?} while checking {:?}",
 756                 unhandled, ty
 757             );
 758             return None;
 759         }
 760     })
 761 }
 762
 763 /// Check if this enum can be safely exported based on the "nullable pointer optimization". If it
 764 /// can, return the type that `ty` can be safely converted to, otherwise return `None`.
 765 /// Currently restricted to function pointers, boxes, references, `core::num::NonZero*`,
 766 /// `core::ptr::NonNull`, and `#[repr(transparent)]` newtypes.
 767 /// FIXME: This duplicates code in codegen.
 768 crate fn repr_nullable_ptr<'tcx>(
 769     cx: &LateContext<'tcx>,
 770     ty: Ty<'tcx>,
 771     ckind: CItemKind,
 772 ) -> Option<Ty<'tcx>> {
 773     debug!("is_repr_nullable_ptr(cx, ty = {:?})", ty);
 774     if let ty::Adt(ty_def, substs) = ty.kind() {
 775         let field_ty = match &ty_def.variants.raw[..] {
 776             [var_one, var_two] => match (&var_one.fields[..], &var_two.fields[..]) {
 777                 ([], [field]) | ([field], []) => field.ty(cx.tcx, substs),
 778                 _ => return None,
 779             },
 780             _ => return None,
 781         };
 782
 783         if !ty_is_known_nonnull(cx, field_ty, ckind) {
 784             return None;
 785         }
 786
 787         // At this point, the field's type is known to be nonnull and the parent enum is Option-like.
 788         // If the computed size for the field and the enum are different, the nonnull optimization isn't
 789         // being applied (and we've got a problem somewhere).
 790         let compute_size_skeleton = |t| SizeSkeleton::compute(t, cx.tcx, cx.param_env).unwrap();
 791         if !compute_size_skeleton(ty).same_size(compute_size_skeleton(field_ty)) {
 792             bug!("improper_ctypes: Option nonnull optimization not applied?");
 793         }
 794
 795         // Return the nullable type this Option-like enum can be safely represented with.
 796         let field_ty_abi = &cx.layout_of(field_ty).unwrap().abi;
 797         if let Abi::Scalar(field_ty_scalar) = field_ty_abi {
 798             match (field_ty_scalar.valid_range.start(), field_ty_scalar.valid_range.end()) {
 799                 (0, _) => unreachable!("Non-null optimisation extended to a non-zero value."),
 800                 (1, _) => {
 801                     return Some(get_nullable_type(cx, field_ty).unwrap());
 802                 }
 803                 (start, end) => unreachable!("Unhandled start and end range: ({}, {})", start, end),
 804             };
 805         }
 806     }
 807     None
 808 }
 809
 810 impl<'a, 'tcx> ImproperCTypesVisitor<'a, 'tcx> {
 811     /// Check if the type is array and emit an unsafe type lint.
 812     fn check_for_array_ty(&mut self, sp: Span, ty: Ty<'tcx>) -> bool {
 813         if let ty::Array(..) = ty.kind() {
 814             self.emit_ffi_unsafe_type_lint(
 815                 ty,
 816                 sp,
 817                 "passing raw arrays by value is not FFI-safe",
 818                 Some("consider passing a pointer to the array"),
 819             );
 820             true
 821         } else {
 822             false
 823         }
 824     }
 825
 826     /// Checks if the given field's type is "ffi-safe".
 827     fn check_field_type_for_ffi(
 828         &self,
 829         cache: &mut FxHashSet<Ty<'tcx>>,
 830         field: &ty::FieldDef,
 831         substs: SubstsRef<'tcx>,
 832     ) -> FfiResult<'tcx> {
 833         let field_ty = field.ty(self.cx.tcx, substs);
 834         if field_ty.has_opaque_types() {
 835             self.check_type_for_ffi(cache, field_ty)
 836         } else {
 837             let field_ty = self.cx.tcx.normalize_erasing_regions(self.cx.param_env, field_ty);
 838             self.check_type_for_ffi(cache, field_ty)
 839         }
 840     }
 841
 842     /// Checks if the given `VariantDef`'s field types are "ffi-safe".
 843     fn check_variant_for_ffi(
 844         &self,
 845         cache: &mut FxHashSet<Ty<'tcx>>,
 846         ty: Ty<'tcx>,
 847         def: &ty::AdtDef,
 848         variant: &ty::VariantDef,
 849         substs: SubstsRef<'tcx>,
 850     ) -> FfiResult<'tcx> {
 851         use FfiResult::*;
 852
 853         if def.repr.transparent() {
 854             // Can assume that only one field is not a ZST, so only check
 855             // that field's type for FFI-safety.
 856             if let Some(field) = transparent_newtype_field(self.cx.tcx, variant) {
 857                 self.check_field_type_for_ffi(cache, field, substs)
 858             } else {
 859                 bug!("malformed transparent type");
 860             }
 861         } else {
 862             // We can't completely trust repr(C) markings; make sure the fields are
 863             // actually safe.
 864             let mut all_phantom = !variant.fields.is_empty();
 865             for field in &variant.fields {
 866                 match self.check_field_type_for_ffi(cache, &field, substs) {
 867                     FfiSafe => {
 868                         all_phantom = false;
 869                     }
 870                     FfiPhantom(..) if def.is_enum() => {
 871                         return FfiUnsafe {
 872                             ty,
 873                             reason: "this enum contains a PhantomData field".into(),
 874                             help: None,
 875                         };
 876                     }
 877                     FfiPhantom(..) => {}
 878                     r => return r,
 879                 }
 880             }
 881
 882             if all_phantom { FfiPhantom(ty) } else { FfiSafe }
 883         }
 884     }
 885
 886     /// Checks if the given type is "ffi-safe" (has a stable, well-defined
 887     /// representation which can be exported to C code).
 888     fn check_type_for_ffi(&self, cache: &mut FxHashSet<Ty<'tcx>>, ty: Ty<'tcx>) -> FfiResult<'tcx> {
 889         use FfiResult::*;
 890
 891         let tcx = self.cx.tcx;
 892
 893         // Protect against infinite recursion, for example
 894         // `struct S(*mut S);`.
 895         // FIXME: A recursion limit is necessary as well, for irregular
 896         // recursive types.
 897         if !cache.insert(ty) {
 898             return FfiSafe;
 899         }
 900
 901         match *ty.kind() {
 902             ty::Adt(def, substs) => {
 903                 if def.is_box() && matches!(self.mode, CItemKind::Definition) {
 904                     if ty.boxed_ty().is_sized(tcx.at(DUMMY_SP), self.cx.param_env) {
 905                         return FfiSafe;
 906                     } else {
 907                         return FfiUnsafe {
 908                             ty,
 909                             reason: format!("box cannot be represented as a single pointer"),
 910                             help: None,
 911                         };
 912                     }
 913                 }
 914                 if def.is_phantom_data() {
 915                     return FfiPhantom(ty);
 916                 }
 917                 match def.adt_kind() {
 918                     AdtKind::Struct | AdtKind::Union => {
 919                         let kind = if def.is_struct() { "struct" } else { "union" };
 920
 921                         if !def.repr.c() && !def.repr.transparent() {
 922                             return FfiUnsafe {
 923                                 ty,
 924                                 reason: format!("this {} has unspecified layout", kind),
 925                                 help: Some(format!(
 926                                     "consider adding a `#[repr(C)]` or \
 927                                              `#[repr(transparent)]` attribute to this {}",
 928                                     kind
 929                                 )),
 930                             };
 931                         }
 932
 933                         let is_non_exhaustive =
 934                             def.non_enum_variant().is_field_list_non_exhaustive();
 935                         if is_non_exhaustive && !def.did.is_local() {
 936                             return FfiUnsafe {
 937                                 ty,
 938                                 reason: format!("this {} is non-exhaustive", kind),
 939                                 help: None,
 940                             };
 941                         }
 942
 943                         if def.non_enum_variant().fields.is_empty() {
 944                             return FfiUnsafe {
 945                                 ty,
 946                                 reason: format!("this {} has no fields", kind),
 947                                 help: Some(format!("consider adding a member to this {}", kind)),
 948                             };
 949                         }
 950
 951                         self.check_variant_for_ffi(cache, ty, def, def.non_enum_variant(), substs)
 952                     }
 953                     AdtKind::Enum => {
 954                         if def.variants.is_empty() {
 955                             // Empty enums are okay... although sort of useless.
 956                             return FfiSafe;
 957                         }
 958
 959                         // Check for a repr() attribute to specify the size of the
 960                         // discriminant.
 961                         if !def.repr.c() && !def.repr.transparent() && def.repr.int.is_none() {
 962                             // Special-case types like `Option<extern fn()>`.
 963                             if repr_nullable_ptr(self.cx, ty, self.mode).is_none() {
 964                                 return FfiUnsafe {
 965                                     ty,
 966                                     reason: "enum has no representation hint".into(),
 967                                     help: Some(
 968                                         "consider adding a `#[repr(C)]`, \
 969                                                 `#[repr(transparent)]`, or integer `#[repr(...)]` \
 970                                                 attribute to this enum"
 971                                             .into(),
 972                                     ),
 973                                 };
 974                             }
 975                         }
 976
 977                         if def.is_variant_list_non_exhaustive() && !def.did.is_local() {
 978                             return FfiUnsafe {
 979                                 ty,
 980                                 reason: "this enum is non-exhaustive".into(),
 981                                 help: None,
 982                             };
 983                         }
 984
 985                         // Check the contained variants.
 986                         for variant in &def.variants {
 987                             let is_non_exhaustive = variant.is_field_list_non_exhaustive();
 988                             if is_non_exhaustive && !variant.def_id.is_local() {
 989                                 return FfiUnsafe {
 990                                     ty,
 991                                     reason: "this enum has non-exhaustive variants".into(),
 992                                     help: None,
 993                                 };
 994                             }
 995
 996                             match self.check_variant_for_ffi(cache, ty, def, variant, substs) {
 997                                 FfiSafe => (),
 998                                 r => return r,
 999                             }
1000                         }
1001
1002                         FfiSafe
1003                     }
1004                 }
1005             }
1006
1007             ty::Char => FfiUnsafe {
1008                 ty,
1009                 reason: "the `char` type has no C equivalent".into(),
1010                 help: Some("consider using `u32` or `libc::wchar_t` instead".into()),
1011             },
1012
1013             ty::Int(ty::IntTy::I128) | ty::Uint(ty::UintTy::U128) => FfiUnsafe {
1014                 ty,
1015                 reason: "128-bit integers don't currently have a known stable ABI".into(),
1016                 help: None,
1017             },
1018
1019             // Primitive types with a stable representation.
1020             ty::Bool | ty::Int(..) | ty::Uint(..) | ty::Float(..) | ty::Never => FfiSafe,
1021
1022             ty::Slice(_) => FfiUnsafe {
1023                 ty,
1024                 reason: "slices have no C equivalent".into(),
1025                 help: Some("consider using a raw pointer instead".into()),
1026             },
1027
1028             ty::Dynamic(..) => {
1029                 FfiUnsafe { ty, reason: "trait objects have no C equivalent".into(), help: None }
1030             }
1031
1032             ty::Str => FfiUnsafe {
1033                 ty,
1034                 reason: "string slices have no C equivalent".into(),
1035                 help: Some("consider using `*const u8` and a length instead".into()),
1036             },
1037
1038             ty::Tuple(..) => FfiUnsafe {
1039                 ty,
1040                 reason: "tuples have unspecified layout".into(),
1041                 help: Some("consider using a struct instead".into()),
1042             },
1043
1044             ty::RawPtr(ty::TypeAndMut { ty, .. }) | ty::Ref(_, ty, _)
1045                 if {
1046                     matches!(self.mode, CItemKind::Definition)
1047                         && ty.is_sized(self.cx.tcx.at(DUMMY_SP), self.cx.param_env)
1048                 } =>
1049             {
1050                 FfiSafe
1051             }
1052
1053             ty::RawPtr(ty::TypeAndMut { ty, .. }) | ty::Ref(_, ty, _) => {
1054                 self.check_type_for_ffi(cache, ty)
1055             }
1056
1057             ty::Array(inner_ty, _) => self.check_type_for_ffi(cache, inner_ty),
1058
1059             ty::FnPtr(sig) => {
1060                 if self.is_internal_abi(sig.abi()) {
1061                     return FfiUnsafe {
1062                         ty,
1063                         reason: "this function pointer has Rust-specific calling convention".into(),
1064                         help: Some(
1065                             "consider using an `extern fn(...) -> ...` \
1066                                     function pointer instead"
1067                                 .into(),
1068                         ),
1069                     };
1070                 }
1071
1072                 let sig = tcx.erase_late_bound_regions(sig);
1073                 if !sig.output().is_unit() {
1074                     let r = self.check_type_for_ffi(cache, sig.output());
1075                     match r {
1076                         FfiSafe => {}
1077                         _ => {
1078                             return r;
1079                         }
1080                     }
1081                 }
1082                 for arg in sig.inputs() {
1083                     let r = self.check_type_for_ffi(cache, arg);
1084                     match r {
1085                         FfiSafe => {}
1086                         _ => {
1087                             return r;
1088                         }
1089                     }
1090                 }
1091                 FfiSafe
1092             }
1093
1094             ty::Foreign(..) => FfiSafe,
1095
1096             // While opaque types are checked for earlier, if a projection in a struct field
1097             // normalizes to an opaque type, then it will reach this branch.
1098             ty::Opaque(..) => {
1099                 FfiUnsafe { ty, reason: "opaque types have no C equivalent".into(), help: None }
1100             }
1101
1102             // `extern "C" fn` functions can have type parameters, which may or may not be FFI-safe,
1103             //  so they are currently ignored for the purposes of this lint.
1104             ty::Param(..) | ty::Projection(..) if matches!(self.mode, CItemKind::Definition) => {
1105                 FfiSafe
1106             }
1107
1108             ty::Param(..)
1109             | ty::Projection(..)
1110             | ty::Infer(..)
1111             | ty::Bound(..)
1112             | ty::Error(_)
1113             | ty::Closure(..)
1114             | ty::Generator(..)
1115             | ty::GeneratorWitness(..)
1116             | ty::Placeholder(..)
1117             | ty::FnDef(..) => bug!("unexpected type in foreign function: {:?}", ty),
1118         }
1119     }
1120
1121     fn emit_ffi_unsafe_type_lint(
1122         &mut self,
1123         ty: Ty<'tcx>,
1124         sp: Span,
1125         note: &str,
1126         help: Option<&str>,
1127     ) {
1128         let lint = match self.mode {
1129             CItemKind::Declaration => IMPROPER_CTYPES,
1130             CItemKind::Definition => IMPROPER_CTYPES_DEFINITIONS,
1131         };
1132
1133         self.cx.struct_span_lint(lint, sp, |lint| {
1134             let item_description = match self.mode {
1135                 CItemKind::Declaration => "block",
1136                 CItemKind::Definition => "fn",
1137             };
1138             let mut diag = lint.build(&format!(
1139                 "`extern` {} uses type `{}`, which is not FFI-safe",
1140                 item_description, ty
1141             ));
1142             diag.span_label(sp, "not FFI-safe");
1143             if let Some(help) = help {
1144                 diag.help(help);
1145             }
1146             diag.note(note);
1147             if let ty::Adt(def, _) = ty.kind() {
1148                 if let Some(sp) = self.cx.tcx.hir().span_if_local(def.did) {
1149                     diag.span_note(sp, "the type is defined here");
1150                 }
1151             }
1152             diag.emit();
1153         });
1154     }
1155
1156     fn check_for_opaque_ty(&mut self, sp: Span, ty: Ty<'tcx>) -> bool {
1157         struct ProhibitOpaqueTypes<'a, 'tcx> {
1158             cx: &'a LateContext<'tcx>,
1159         }
1160
1161         impl<'a, 'tcx> ty::fold::TypeVisitor<'tcx> for ProhibitOpaqueTypes<'a, 'tcx> {
1162             type BreakTy = Ty<'tcx>;
1163
1164             fn visit_ty(&mut self, ty: Ty<'tcx>) -> ControlFlow<Self::BreakTy> {
1165                 match ty.kind() {
1166                     ty::Opaque(..) => ControlFlow::Break(ty),
1167                     // Consider opaque types within projections FFI-safe if they do not normalize
1168                     // to more opaque types.
1169                     ty::Projection(..) => {
1170                         let ty = self.cx.tcx.normalize_erasing_regions(self.cx.param_env, ty);
1171
1172                         // If `ty` is a opaque type directly then `super_visit_with` won't invoke
1173                         // this function again.
1174                         if ty.has_opaque_types() {
1175                             self.visit_ty(ty)
1176                         } else {
1177                             ControlFlow::CONTINUE
1178                         }
1179                     }
1180                     _ => ty.super_visit_with(self),
1181                 }
1182             }
1183         }
1184
1185         if let Some(ty) = ty.visit_with(&mut ProhibitOpaqueTypes { cx: self.cx }).break_value() {
1186             self.emit_ffi_unsafe_type_lint(ty, sp, "opaque types have no C equivalent", None);
1187             true
1188         } else {
1189             false
1190         }
1191     }
1192
1193     fn check_type_for_ffi_and_report_errors(
1194         &mut self,
1195         sp: Span,
1196         ty: Ty<'tcx>,
1197         is_static: bool,
1198         is_return_type: bool,
1199     ) {
1200         // We have to check for opaque types before `normalize_erasing_regions`,
1201         // which will replace opaque types with their underlying concrete type.
1202         if self.check_for_opaque_ty(sp, ty) {
1203             // We've already emitted an error due to an opaque type.
1204             return;
1205         }
1206
1207         // it is only OK to use this function because extern fns cannot have
1208         // any generic types right now:
1209         let ty = self.cx.tcx.normalize_erasing_regions(self.cx.param_env, ty);
1210
1211         // C doesn't really support passing arrays by value - the only way to pass an array by value
1212         // is through a struct. So, first test that the top level isn't an array, and then
1213         // recursively check the types inside.
1214         if !is_static && self.check_for_array_ty(sp, ty) {
1215             return;
1216         }
1217
1218         // Don't report FFI errors for unit return types. This check exists here, and not in
1219         // `check_foreign_fn` (where it would make more sense) so that normalization has definitely
1220         // happened.
1221         if is_return_type && ty.is_unit() {
1222             return;
1223         }
1224
1225         match self.check_type_for_ffi(&mut FxHashSet::default(), ty) {
1226             FfiResult::FfiSafe => {}
1227             FfiResult::FfiPhantom(ty) => {
1228                 self.emit_ffi_unsafe_type_lint(ty, sp, "composed only of `PhantomData`", None);
1229             }
1230             // If `ty` is a `repr(transparent)` newtype, and the non-zero-sized type is a generic
1231             // argument, which after substitution, is `()`, then this branch can be hit.
1232             FfiResult::FfiUnsafe { ty, .. } if is_return_type && ty.is_unit() => {}
1233             FfiResult::FfiUnsafe { ty, reason, help } => {
1234                 self.emit_ffi_unsafe_type_lint(ty, sp, &reason, help.as_deref());
1235             }
1236         }
1237     }
1238
1239     fn check_foreign_fn(&mut self, id: hir::HirId, decl: &hir::FnDecl<'_>) {
1240         let def_id = self.cx.tcx.hir().local_def_id(id);
1241         let sig = self.cx.tcx.fn_sig(def_id);
1242         let sig = self.cx.tcx.erase_late_bound_regions(sig);
1243
1244         for (input_ty, input_hir) in iter::zip(sig.inputs(), decl.inputs) {
1245             self.check_type_for_ffi_and_report_errors(input_hir.span, input_ty, false, false);
1246         }
1247
1248         if let hir::FnRetTy::Return(ref ret_hir) = decl.output {
1249             let ret_ty = sig.output();
1250             self.check_type_for_ffi_and_report_errors(ret_hir.span, ret_ty, false, true);
1251         }
1252     }
1253
1254     fn check_foreign_static(&mut self, id: hir::HirId, span: Span) {
1255         let def_id = self.cx.tcx.hir().local_def_id(id);
1256         let ty = self.cx.tcx.type_of(def_id);
1257         self.check_type_for_ffi_and_report_errors(span, ty, true, false);
1258     }
1259
1260     fn is_internal_abi(&self, abi: SpecAbi) -> bool {
1261         matches!(
1262             abi,
1263             SpecAbi::Rust | SpecAbi::RustCall | SpecAbi::RustIntrinsic | SpecAbi::PlatformIntrinsic
1264         )
1265     }
1266 }
1267
1268 impl<'tcx> LateLintPass<'tcx> for ImproperCTypesDeclarations {
1269     fn check_foreign_item(&mut self, cx: &LateContext<'_>, it: &hir::ForeignItem<'_>) {
1270         let mut vis = ImproperCTypesVisitor { cx, mode: CItemKind::Declaration };
1271         let abi = cx.tcx.hir().get_foreign_abi(it.hir_id());
1272
1273         if !vis.is_internal_abi(abi) {
1274             match it.kind {
1275                 hir::ForeignItemKind::Fn(ref decl, _, _) => {
1276                     vis.check_foreign_fn(it.hir_id(), decl);
1277                 }
1278                 hir::ForeignItemKind::Static(ref ty, _) => {
1279                     vis.check_foreign_static(it.hir_id(), ty.span);
1280                 }
1281                 hir::ForeignItemKind::Type => (),
1282             }
1283         }
1284     }
1285 }
1286
1287 impl<'tcx> LateLintPass<'tcx> for ImproperCTypesDefinitions {
1288     fn check_fn(
1289         &mut self,
1290         cx: &LateContext<'tcx>,
1291         kind: hir::intravisit::FnKind<'tcx>,
1292         decl: &'tcx hir::FnDecl<'_>,
1293         _: &'tcx hir::Body<'_>,
1294         _: Span,
1295         hir_id: hir::HirId,
1296     ) {
1297         use hir::intravisit::FnKind;
1298
1299         let abi = match kind {
1300             FnKind::ItemFn(_, _, header, ..) => header.abi,
1301             FnKind::Method(_, sig, ..) => sig.header.abi,
1302             _ => return,
1303         };
1304
1305         let mut vis = ImproperCTypesVisitor { cx, mode: CItemKind::Definition };
1306         if !vis.is_internal_abi(abi) {
1307             vis.check_foreign_fn(hir_id, decl);
1308         }
1309     }
1310 }
1311
1312 declare_lint_pass!(VariantSizeDifferences => [VARIANT_SIZE_DIFFERENCES]);
1313
1314 impl<'tcx> LateLintPass<'tcx> for VariantSizeDifferences {
1315     fn check_item(&mut self, cx: &LateContext<'_>, it: &hir::Item<'_>) {
1316         if let hir::ItemKind::Enum(ref enum_definition, _) = it.kind {
1317             let t = cx.tcx.type_of(it.def_id);
1318             let ty = cx.tcx.erase_regions(t);
1319             let layout = match cx.layout_of(ty) {
1320                 Ok(layout) => layout,
1321                 Err(
1322                     ty::layout::LayoutError::Unknown(_) | ty::layout::LayoutError::SizeOverflow(_),
1323                 ) => return,
1324             };
1325             let (variants, tag) = match layout.variants {
1326                 Variants::Multiple {
1327                     tag_encoding: TagEncoding::Direct,
1328                     ref tag,
1329                     ref variants,
1330                     ..
1331                 } => (variants, tag),
1332                 _ => return,
1333             };
1334
1335             let tag_size = tag.value.size(&cx.tcx).bytes();
1336
1337             debug!(
1338                 "enum `{}` is {} bytes large with layout:\n{:#?}",
1339                 t,
1340                 layout.size.bytes(),
1341                 layout
1342             );
1343
1344             let (largest, slargest, largest_index) = iter::zip(enum_definition.variants, variants)
1345                 .map(|(variant, variant_layout)| {
1346                     // Subtract the size of the enum tag.
1347                     let bytes = variant_layout.size.bytes().saturating_sub(tag_size);
1348
1349                     debug!("- variant `{}` is {} bytes large", variant.ident, bytes);
1350                     bytes
1351                 })
1352                 .enumerate()
1353                 .fold((0, 0, 0), |(l, s, li), (idx, size)| {
1354                     if size > l {
1355                         (size, l, idx)
1356                     } else if size > s {
1357                         (l, size, li)
1358                     } else {
1359                         (l, s, li)
1360                     }
1361                 });
1362
1363             // We only warn if the largest variant is at least thrice as large as
1364             // the second-largest.
1365             if largest > slargest * 3 && slargest > 0 {
1366                 cx.struct_span_lint(
1367                     VARIANT_SIZE_DIFFERENCES,
1368                     enum_definition.variants[largest_index].span,
1369                     |lint| {
1370                         lint.build(&format!(
1371                             "enum variant is more than three times \
1372                                           larger ({} bytes) than the next largest",
1373                             largest
1374                         ))
1375                         .emit()
1376                     },
1377                 );
1378             }
1379         }
1380     }
1381 }