49 lines
1.6 KiB
Rust
49 lines
1.6 KiB
Rust
|
use super::utils::is_layout_incompatible;
|
||
|
|
use super::UNSOUND_COLLECTION_TRANSMUTE;
|
||
|
|
use crate::utils::{match_def_path, paths, span_lint};
|
||
|
|
use rustc_hir::Expr;
|
||
|
|
use rustc_lint::LateContext;
|
||
|
|
use rustc_middle::ty::{self, Ty};
|
||
|
|
|
||
|
|
// used to check for UNSOUND_COLLECTION_TRANSMUTE
|
||
|
|
static COLLECTIONS: &[&[&str]] = &[
|
||
|
|
&paths::VEC,
|
||
|
|
&paths::VEC_DEQUE,
|
||
|
|
&paths::BINARY_HEAP,
|
||
|
|
&paths::BTREESET,
|
||
|
|
&paths::BTREEMAP,
|
||
|
|
&paths::HASHSET,
|
||
|
|
&paths::HASHMAP,
|
||
|
|
];
|
||
|
|
|
||
|
|
/// Checks for `unsound_collection_transmute` lint.
|
||
|
|
/// Returns `true` if it's triggered, otherwise returns `false`.
|
||
|
|
pub(super) fn check<'tcx>(cx: &LateContext<'tcx>, e: &'tcx Expr<'_>, from_ty: Ty<'tcx>, to_ty: Ty<'tcx>) -> bool {
|
||
|
|
match (&from_ty.kind(), &to_ty.kind()) {
|
||
|
|
(ty::Adt(from_adt, from_substs), ty::Adt(to_adt, to_substs)) => {
|
||
|
|
if from_adt.did != to_adt.did || !COLLECTIONS.iter().any(|path| match_def_path(cx, to_adt.did, path)) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
if from_substs
|
||
|
|
.types()
|
||
|
|
.zip(to_substs.types())
|
||
|
|
.any(|(from_ty, to_ty)| is_layout_incompatible(cx, from_ty, to_ty))
|
||
|
|
{
|
||
|
|
span_lint(
|
||
|
|
cx,
|
||
|
|
UNSOUND_COLLECTION_TRANSMUTE,
|
||
|
|
e.span,
|
||
|
|
&format!(
|
||
|
|
"transmute from `{}` to `{}` with mismatched layout is unsound",
|
||
|
|
from_ty, to_ty
|
||
|
|
),
|
||
|
|
);
|
||
|
|
true
|
||
|
|
} else {
|
||
|
|
false
|
||
|
|
}
|
||
|
|
},
|
||
|
|
_ => false,
|
||
|
|
}
|
||
|
|
}
|