[DEX] Include MethodHandleKey and CallSiteKey

REAndroid · Oct 27, 2024 · fac0467 · fac0467
1 parent 94e1988
commit fac0467
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/java/com/reandroid/dex/model/DexClassRepository.java b/src/main/java/com/reandroid/dex/model/DexClassRepository.java
@@ -113,6 +113,12 @@ default boolean contains(Key key){
         if(key instanceof TypeListKey){
             return contains(SectionType.TYPE_LIST, key);
         }
+        if(key instanceof MethodHandleKey){
+            return contains(SectionType.METHOD_HANDLE, key);
+        }
+        if(key instanceof CallSiteKey){
+            return contains(SectionType.CALL_SITE_ID, key);
+        }
         throw new IllegalArgumentException("Unknown key type: " + key.getClass() + ", '" + key + "'");
     }
     default boolean containsClass(TypeKey key){