1 use crate::lints::PassByValueDiag;
2 use crate::{LateContext, LateLintPass, LintContext};
4 use rustc_hir::def::Res;
5 use rustc_hir::{GenericArg, PathSegment, QPath, TyKind};
7 use rustc_span::symbol::sym;
10 /// The `rustc_pass_by_value` lint marks a type with `#[rustc_pass_by_value]` requiring it to
11 /// always be passed by value. This is usually used for types that are thin wrappers around
12 /// references, so there is no benefit to an extra layer of indirection. (Example: `Ty` which
13 /// is a reference to an `Interned<TyKind>`)
14 pub rustc::PASS_BY_VALUE,
16 "pass by reference of a type flagged as `#[rustc_pass_by_value]`",
17 report_in_external_macro: true
20 declare_lint_pass!(PassByValue => [PASS_BY_VALUE]);
22 impl<'tcx> LateLintPass<'tcx> for PassByValue {
23 fn check_ty(&mut self, cx: &LateContext<'_>, ty: &'tcx hir::Ty<'tcx>) {
25 TyKind::Ref(_, hir::MutTy { ty: inner_ty, mutbl: hir::Mutability::Not }) => {
26 if let Some(impl_did) = cx.tcx.impl_of_method(ty.hir_id.owner.to_def_id()) {
27 if cx.tcx.impl_trait_ref(impl_did).is_some() {
31 if let Some(t) = path_for_pass_by_value(cx, &inner_ty) {
35 PassByValueDiag { ty: t.clone(), suggestion: ty.span },
44 fn path_for_pass_by_value(cx: &LateContext<'_>, ty: &hir::Ty<'_>) -> Option<String> {
45 if let TyKind::Path(QPath::Resolved(_, path)) = &ty.kind {
47 Res::Def(_, def_id) if cx.tcx.has_attr(def_id, sym::rustc_pass_by_value) => {
48 let name = cx.tcx.item_name(def_id).to_ident_string();
49 let path_segment = path.segments.last().unwrap();
50 return Some(format!("{}{}", name, gen_args(cx, path_segment)));
52 Res::SelfTyAlias { alias_to: did, is_trait_impl: false, .. } => {
53 if let ty::Adt(adt, substs) = cx.tcx.type_of(did).kind() {
54 if cx.tcx.has_attr(adt.did(), sym::rustc_pass_by_value) {
55 return Some(cx.tcx.def_path_str_with_substs(adt.did(), substs));
66 fn gen_args(cx: &LateContext<'_>, segment: &PathSegment<'_>) -> String {
67 if let Some(args) = &segment.args {
71 .map(|arg| match arg {
72 GenericArg::Lifetime(lt) => lt.to_string(),
73 GenericArg::Type(ty) => {
74 cx.tcx.sess.source_map().span_to_snippet(ty.span).unwrap_or_else(|_| "_".into())
76 GenericArg::Const(c) => {
77 cx.tcx.sess.source_map().span_to_snippet(c.span).unwrap_or_else(|_| "_".into())
79 GenericArg::Infer(_) => String::from("_"),
83 if !params.is_empty() {
84 return format!("<{}>", params.join(", "));