capacitor-dex-editor 0.0.66 → 0.0.67

package/android/src/main/java/com/aetherlink/dexeditor/DexManager.java

@@ -1701,7 +1701,7 @@ public class DexManager {
     }
 
     /**
-     * 获取多 DEX 会话中的类列表（支持分页和过滤）
+     * 获取多 DEX 会话中的类列表（Rust 实现）
      */
     public JSObject getClassesFromMultiSession(String sessionId, String packageFilter, int offset, int limit) throws Exception {
         MultiDexSession session = multiDexSessions.get(sessionId);
@@ -1709,26 +1709,29 @@ public class DexManager {
             throw new IllegalArgumentException("Session not found: " + sessionId);
         }
         
+        if (!RustDex.isAvailable() || session.dexBytes.isEmpty()) {
+            throw new RuntimeException("Rust DEX library not available");
+        }
+        
         JSObject result = new JSObject();
         JSArray classes = new JSArray();
         List<String> allClasses = new ArrayList<>();
+        String filter = packageFilter != null ? packageFilter : "";
         
-        // 收集所有类
-        for (Map.Entry<String, DexBackedDexFile> entry : session.dexFiles.entrySet()) {
+        // 使用 Rust 获取每个 DEX 的类列表
+        for (Map.Entry<String, byte[]> entry : session.dexBytes.entrySet()) {
             String dexName = entry.getKey();
-            DexBackedDexFile dexFile = entry.getValue();
+            byte[] dexData = entry.getValue();
             
-            for (ClassDef classDef : dexFile.getClasses()) {
-                String className = convertTypeToClassName(classDef.getType());
-                
-                // 包名过滤
-                if (packageFilter != null && !packageFilter.isEmpty()) {
-                    if (!className.startsWith(packageFilter)) {
-                        continue;
+            String jsonResult = RustDex.listClasses(dexData, filter, 0, 100000);
+            if (jsonResult != null && !jsonResult.contains("\"error\"")) {
+                org.json.JSONObject rustResult = new org.json.JSONObject(jsonResult);
+                org.json.JSONArray rustClasses = rustResult.optJSONArray("classes");
+                if (rustClasses != null) {
+                    for (int i = 0; i < rustClasses.length(); i++) {
+                        allClasses.add(rustClasses.getString(i) + "|" + dexName);
                     }
                 }
-                
-                allClasses.add(className + "|" + dexName);
             }
         }
         
@@ -1752,6 +1755,7 @@ public class DexManager {
         result.put("limit", limit);
         result.put("classes", classes);
         result.put("hasMore", end < total);
+        result.put("engine", "rust");
         
         return result;
     }
@@ -1836,7 +1840,7 @@ public class DexManager {
 
 
     /**
-     * 从多 DEX 会话获取类的 Smali 代码
+     * 从多 DEX 会话获取类的 Smali 代码（Rust 实现）
      */
     public JSObject getClassSmaliFromSession(String sessionId, String className) throws Exception {
         MultiDexSession session = multiDexSessions.get(sessionId);
@@ -1844,20 +1848,24 @@ public class DexManager {
             throw new IllegalArgumentException("Session not found: " + sessionId);
         }
         
-        String targetType = convertClassNameToType(className);
+        if (!RustDex.isAvailable()) {
+            throw new RuntimeException("Rust DEX library not available");
+        }
         
-        for (Map.Entry<String, DexBackedDexFile> entry : session.dexFiles.entrySet()) {
+        // 使用 Rust 获取 Smali
+        for (Map.Entry<String, byte[]> entry : session.dexBytes.entrySet()) {
             String dexName = entry.getKey();
-            DexBackedDexFile dexFile = entry.getValue();
+            byte[] dexData = entry.getValue();
             
-            for (ClassDef classDef : dexFile.getClasses()) {
-                if (classDef.getType().equals(targetType)) {
-                    JSObject result = new JSObject();
-                    result.put("className", className);
-                    result.put("dexFile", dexName);
-                    result.put("smaliContent", getSmaliForClass(dexFile, classDef));
-                    return result;
-                }
+            String jsonResult = RustDex.getClassSmali(dexData, className);
+            if (jsonResult != null && !jsonResult.contains("\"error\"")) {
+                org.json.JSONObject rustResult = new org.json.JSONObject(jsonResult);
+                JSObject result = new JSObject();
+                result.put("className", className);
+                result.put("dexFile", dexName);
+                result.put("smaliContent", rustResult.optString("smaliContent", ""));
+                result.put("engine", "rust");
+                return result;
             }
         }
         
@@ -1865,7 +1873,7 @@ public class DexManager {
     }
 
     /**
-     * 修改类并保存到多 DEX 会话
+     * 修改类并保存到多 DEX 会话（Rust 实现）
      */
     public void modifyClassInSession(String sessionId, String className, String smaliContent) throws Exception {
         MultiDexSession session = multiDexSessions.get(sessionId);
@@ -1873,36 +1881,41 @@ public class DexManager {
             throw new IllegalArgumentException("Session not found: " + sessionId);
         }
         
-        String targetType = convertClassNameToType(className);
+        if (!RustDex.isAvailable()) {
+            throw new RuntimeException("Rust DEX library not available");
+        }
         
         // 找到类所在的 DEX
         String targetDex = null;
-        for (Map.Entry<String, DexBackedDexFile> entry : session.dexFiles.entrySet()) {
-            for (ClassDef classDef : entry.getValue().getClasses()) {
-                if (classDef.getType().equals(targetType)) {
-                    targetDex = entry.getKey();
-                    break;
-                }
+        for (Map.Entry<String, byte[]> entry : session.dexBytes.entrySet()) {
+            String jsonResult = RustDex.getClassSmali(entry.getValue(), className);
+            if (jsonResult != null && !jsonResult.contains("\"error\"")) {
+                targetDex = entry.getKey();
+                break;
             }
-            if (targetDex != null) break;
         }
         
         if (targetDex == null) {
             throw new IllegalArgumentException("Class not found: " + className);
         }
         
-        // 编译新的 Smali
-        ClassDef newClassDef = compileSmaliToClass(smaliContent, Opcodes.getDefault());
+        // 使用 Rust 修改类
+        byte[] originalDex = session.dexBytes.get(targetDex);
+        byte[] modifiedDex = RustDex.modifyClass(originalDex, className, smaliContent);
         
-        // 记录修改
-        session.modifiedClasses.put(targetDex + "|" + className, newClassDef);
+        if (modifiedDex == null) {
+            throw new RuntimeException("Failed to modify class: " + className);
+        }
+        
+        // 更新 DEX 字节数据
+        session.dexBytes.put(targetDex, modifiedDex);
         session.modified = true;
         
-        Log.d(TAG, "Modified class in session: " + className);
+        Log.d(TAG, "Modified class in session (Rust): " + className);
     }
 
     /**
-     * 添加新类到会话
+     * 添加新类到会话（Rust 实现）
      */
     public void addClassToSession(String sessionId, String className, String smaliContent) throws Exception {
         MultiDexSession session = multiDexSessions.get(sessionId);
@@ -1910,23 +1923,33 @@ public class DexManager {
             throw new IllegalArgumentException("Session not found: " + sessionId);
         }
         
-        // 编译 Smali
-        ClassDef newClassDef = compileSmaliToClass(smaliContent, Opcodes.getDefault());
+        if (!RustDex.isAvailable()) {
+            throw new RuntimeException("Rust DEX library not available");
+        }
         
         // 添加到第一个 DEX（默认 classes.dex）
         String targetDex = "classes.dex";
-        if (!session.dexFiles.containsKey(targetDex) && !session.dexFiles.isEmpty()) {
-            targetDex = session.dexFiles.keySet().iterator().next();
+        if (!session.dexBytes.containsKey(targetDex) && !session.dexBytes.isEmpty()) {
+            targetDex = session.dexBytes.keySet().iterator().next();
+        }
+        
+        // 使用 Rust 添加类
+        byte[] originalDex = session.dexBytes.get(targetDex);
+        byte[] modifiedDex = RustDex.addClass(originalDex, smaliContent);
+        
+        if (modifiedDex == null) {
+            throw new RuntimeException("Failed to add class: " + className);
         }
         
-        session.modifiedClasses.put(targetDex + "|" + className, newClassDef);
+        // 更新 DEX 字节数据
+        session.dexBytes.put(targetDex, modifiedDex);
         session.modified = true;
         
-        Log.d(TAG, "Added class to session: " + className);
+        Log.d(TAG, "Added class to session (Rust): " + className);
     }
 
     /**
-     * 从会话中删除类
+     * 从会话中删除类（Rust 实现）
      */
     public void deleteClassFromSession(String sessionId, String className) throws Exception {
         MultiDexSession session = multiDexSessions.get(sessionId);
@@ -1934,29 +1957,37 @@ public class DexManager {
             throw new IllegalArgumentException("Session not found: " + sessionId);
         }
         
-        String targetType = convertClassNameToType(className);
+        if (!RustDex.isAvailable()) {
+            throw new RuntimeException("Rust DEX library not available");
+        }
         
-        // 找到类所在的 DEX 并标记为删除
+        // 找到类所在的 DEX
         String targetDex = null;
-        for (Map.Entry<String, DexBackedDexFile> entry : session.dexFiles.entrySet()) {
-            for (ClassDef classDef : entry.getValue().getClasses()) {
-                if (classDef.getType().equals(targetType)) {
-                    targetDex = entry.getKey();
-                    break;
-                }
+        for (Map.Entry<String, byte[]> entry : session.dexBytes.entrySet()) {
+            String jsonResult = RustDex.getClassSmali(entry.getValue(), className);
+            if (jsonResult != null && !jsonResult.contains("\"error\"")) {
+                targetDex = entry.getKey();
+                break;
             }
-            if (targetDex != null) break;
         }
         
         if (targetDex == null) {
             throw new IllegalArgumentException("Class not found: " + className);
         }
         
-        // 用 null 标记删除
-        session.modifiedClasses.put(targetDex + "|" + className, null);
+        // 使用 Rust 删除类
+        byte[] originalDex = session.dexBytes.get(targetDex);
+        byte[] modifiedDex = RustDex.deleteClass(originalDex, className);
+        
+        if (modifiedDex == null) {
+            throw new RuntimeException("Failed to delete class: " + className);
+        }
+        
+        // 更新 DEX 字节数据
+        session.dexBytes.put(targetDex, modifiedDex);
         session.modified = true;
         
-        Log.d(TAG, "Deleted class from session: " + className);
+        Log.d(TAG, "Deleted class from session (Rust): " + className);
     }
 
     /**

package/android/src/main/java/com/aetherlink/dexeditor/RustDex.java

@@ -72,15 +72,37 @@ public class RustDex {
     );
 
     /**
-     * Java 回退实现 - 当 Rust 库不可用时使用
+     * 修改 DEX 中的类
+     * @param dexBytes DEX 文件字节数组
+     * @param className 类名
+     * @param newSmali 新的 Smali 代码
+     * @return 修改后的 DEX 字节数组，失败返回 null
      */
-    public static String searchInDexFallback(
+    public static native byte[] modifyClass(
         byte[] dexBytes,
-        String query,
-        String searchType,
-        boolean caseSensitive,
-        int maxResults
-    ) {
-        return "{\"error\": \"Rust library not available, using Java implementation\", \"fallback\": true}";
-    }
+        String className,
+        String newSmali
+    );
+
+    /**
+     * 添加新类到 DEX
+     * @param dexBytes DEX 文件字节数组
+     * @param newSmali 新类的 Smali 代码
+     * @return 修改后的 DEX 字节数组，失败返回 null
+     */
+    public static native byte[] addClass(
+        byte[] dexBytes,
+        String newSmali
+    );
+
+    /**
+     * 从 DEX 中删除类
+     * @param dexBytes DEX 文件字节数组
+     * @param className 要删除的类名
+     * @return 修改后的 DEX 字节数组，失败返回 null
+     */
+    public static native byte[] deleteClass(
+        byte[] dexBytes,
+        String className
+    );
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "capacitor-dex-editor",
-  "version": "0.0.66",
+  "version": "0.0.67",
   "description": "Capacitor-plugin-for-editing-DEX-files-in-APK",
   "main": "dist/plugin.cjs.js",
   "module": "dist/esm/index.js",