Updated detector to detect OpenBrush implementation of AccountId

CoinFabrik · Aug 11, 2023 · 8ef6c14 · 8ef6c14
1 parent 1f16ae5
commit 8ef6c14
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/detectors/unprotected-mapping-operation/src/lib.rs b/detectors/unprotected-mapping-operation/src/lib.rs
@@ -64,7 +64,8 @@ impl<'tcx> LateLintPass<'tcx> for UnprotectedMappingOperation {
                     let mapping_type = self.cx.typeck_results().expr_ty_adjusted(receiver);
 
                     if /* ty.to_string().contains("ink::storage::Mapping") && */
-                    mapping_type.to_string().contains("ink::storage::Mapping<ink::ink_primitives::AccountId"){
+                    mapping_type.to_string().contains("ink::storage::Mapping<ink::ink_primitives::AccountId") ||
+                    mapping_type.to_string().contains("ink::storage::Mapping<OpenBrush::AccountIdExt") {
 
                         if path.ident.name.to_string() == "insert" {
                             self.insert_def_id = defid;