[Dex] Convenient code & debug clear methods

REAndroid · Apr 5, 2024 · 5b12c57 · 5b12c57
1 parent e5f5aff
commit 5b12c57
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/src/main/java/com/reandroid/dex/data/MethodDef.java b/src/main/java/com/reandroid/dex/data/MethodDef.java
@@ -239,6 +239,15 @@ public CodeItem getCodeItem(){
         }
         return codeItem;
     }
+    public void clearCode(){
+        codeOffset.setItem((CodeItem) null);
+    }
+    public void clearDebug(){
+        CodeItem codeItem = getCodeItem();
+        if(codeItem != null){
+            codeItem.removeDebugInfo();
+        }
+    }
     private void linkCodeItem(){
         CodeItem codeItem = codeOffset.getItem();
         if(codeItem != null){

diff --git a/src/main/java/com/reandroid/dex/model/DexMethod.java b/src/main/java/com/reandroid/dex/model/DexMethod.java
@@ -26,7 +26,6 @@
 import com.reandroid.dex.key.Key;
 import com.reandroid.dex.key.MethodKey;
 import com.reandroid.dex.key.TypeKey;
-import com.reandroid.dex.smali.SmaliParser;
 import com.reandroid.dex.smali.SmaliReader;
 import com.reandroid.dex.smali.SmaliWriter;
 import com.reandroid.dex.smali.model.SmaliInstruction;
@@ -160,6 +159,12 @@ public Iterator<DexInstruction> getInstructions(Predicate<? super Ins> filter) {
     public Iterator<DexInstruction> getInstructions() {
         return DexInstruction.create(this, getDefinition().getInstructions());
     }
+    public void clearCode(){
+        getDefinition().clearCode();
+    }
+    public void clearDebug(){
+        getDefinition().clearDebug();
+    }
     public Iterator<DexTry> getDexTry() {
         TryBlock tryBlock = getDefinition().getTryBlock();
         if(tryBlock == null){