capacitor-dex-editor 0.0.62 → 0.0.64

package/android/build.gradle

@@ -9,13 +9,16 @@ buildscript {
     repositories {
         google()
         mavenCentral()
+        maven { url "https://plugins.gradle.org/m2/" }
     }
     dependencies {
         classpath 'com.android.tools.build:gradle:8.7.2'
+        classpath 'org.mozilla.rust-android-gradle:plugin:0.9.6'
     }
 }
 
 apply plugin: 'com.android.library'
+apply plugin: 'org.mozilla.rust-android-gradle.rust-android'
 
 android {
     namespace "com.aetherlink.dexeditor"
@@ -72,3 +75,18 @@ dependencies {
     androidTestImplementation "androidx.test.ext:junit:$androidxJunitVersion"
     androidTestImplementation "androidx.test.espresso:espresso-core:$androidxEspressoCoreVersion"
 }
+
+// Rust 编译配置
+cargo {
+    module = "../rust"
+    libname = "dex_rust"
+    targets = ["arm", "arm64", "x86", "x86_64"]
+    profile = "release"
+}
+
+// 将 Rust 编译任务添加到构建流程
+tasks.configureEach { task ->
+    if ((task.name == 'javaPreCompileDebug' || task.name == 'javaPreCompileRelease')) {
+        task.dependsOn 'cargoBuild'
+    }
+}

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

@@ -150,6 +150,7 @@ public class DexManager {
         String sessionId;
         String apkPath;
         Map<String, DexBackedDexFile> dexFiles;
+        Map<String, byte[]> dexBytes;  // DEX 字节数据，用于 Rust 搜索
         Map<String, ClassDef> modifiedClasses;
         // 使用 LRU 缓存限制内存，最多缓存 200 个类
         Map<String, String> smaliCache = new java.util.LinkedHashMap<String, String>(200, 0.75f, true) {
@@ -164,11 +165,15 @@ public class DexManager {
             this.sessionId = sessionId;
             this.apkPath = apkPath;
             this.dexFiles = new HashMap<>();
+            this.dexBytes = new HashMap<>();
             this.modifiedClasses = new HashMap<>();
         }
 
-        void addDex(String dexName, DexBackedDexFile dexFile) {
+        void addDex(String dexName, DexBackedDexFile dexFile, byte[] bytes) {
             this.dexFiles.put(dexName, dexFile);
+            if (bytes != null) {
+                this.dexBytes.put(dexName, bytes);
+            }
         }
     }
 
@@ -1638,10 +1643,11 @@ public class DexManager {
                     baos.write(buffer, 0, len);
                 }
                 is.close();
+                byte[] dexData = baos.toByteArray();
                 
                 // 解析 DEX
-                DexBackedDexFile dexFile = new DexBackedDexFile(Opcodes.getDefault(), baos.toByteArray());
-                multiSession.addDex(dexName, dexFile);
+                DexBackedDexFile dexFile = new DexBackedDexFile(Opcodes.getDefault(), dexData);
+                multiSession.addDex(dexName, dexFile, dexData);
                 totalClasses += dexFile.getClasses().size();
                 
                 Log.d(TAG, "Loaded DEX: " + dexName + " with " + dexFile.getClasses().size() + " classes");
@@ -1759,6 +1765,7 @@ public class DexManager {
 
     /**
      * 在多 DEX 会话中搜索
+     * 优先使用 Rust 实现，如果不可用则回退到 Java 实现
      */
     public JSObject searchInMultiSession(String sessionId, String query, String searchType, 
                                           boolean caseSensitive, int maxResults) throws Exception {
@@ -1767,6 +1774,16 @@ public class DexManager {
             throw new IllegalArgumentException("Session not found: " + sessionId);
         }
         
+        // 尝试使用 Rust 实现进行搜索
+        if (RustDex.isAvailable() && !session.dexBytes.isEmpty()) {
+            try {
+                return searchWithRust(session, query, searchType, caseSensitive, maxResults);
+            } catch (Exception e) {
+                Log.w(TAG, "Rust search failed, falling back to Java: " + e.getMessage());
+            }
+        }
+        
+        // Java 回退实现
         JSObject result = new JSObject();
         JSArray results = new JSArray();
         String queryMatch = caseSensitive ? query : query.toLowerCase();
@@ -1883,6 +1900,56 @@ public class DexManager {
         return result;
     }
 
+    /**
+     * 使用 Rust 实现搜索（高性能）
+     */
+    private JSObject searchWithRust(MultiDexSession session, String query, String searchType,
+                                     boolean caseSensitive, int maxResults) throws Exception {
+        JSObject result = new JSObject();
+        JSArray allResults = new JSArray();
+        
+        for (Map.Entry<String, byte[]> entry : session.dexBytes.entrySet()) {
+            String dexName = entry.getKey();
+            byte[] dexData = entry.getValue();
+            
+            // 调用 Rust 搜索
+            String jsonResult = RustDex.searchInDex(dexData, query, searchType, caseSensitive, maxResults);
+            
+            if (jsonResult != null && !jsonResult.contains("\"error\"")) {
+                // 解析 Rust 返回的 JSON
+                org.json.JSONObject rustResult = new org.json.JSONObject(jsonResult);
+                org.json.JSONArray rustResults = rustResult.optJSONArray("results");
+                
+                if (rustResults != null) {
+                    for (int i = 0; i < rustResults.length() && allResults.length() < maxResults; i++) {
+                        org.json.JSONObject item = rustResults.getJSONObject(i);
+                        JSObject jsItem = new JSObject();
+                        jsItem.put("type", item.optString("type", searchType));
+                        jsItem.put("className", item.optString("className", ""));
+                        jsItem.put("dexFile", dexName);
+                        if (item.has("methodName")) {
+                            jsItem.put("methodName", item.getString("methodName"));
+                        }
+                        if (item.has("fieldName")) {
+                            jsItem.put("fieldName", item.getString("fieldName"));
+                        }
+                        allResults.put(jsItem);
+                    }
+                }
+            }
+            
+            if (allResults.length() >= maxResults) break;
+        }
+        
+        result.put("query", query);
+        result.put("searchType", searchType);
+        result.put("total", allResults.length());
+        result.put("results", allResults);
+        result.put("engine", "rust");
+        
+        return result;
+    }
+
     /**
      * 获取类的 Smali 代码（内部方法）
      */

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

@@ -0,0 +1,86 @@
+package com.aetherlink.dexeditor;
+
+import android.util.Log;
+
+/**
+ * RustDex - Rust 实现的 DEX 解析器 JNI 接口
+ * 使用 azw413/smali crate 实现高性能 DEX 搜索
+ */
+public class RustDex {
+    private static final String TAG = "RustDex";
+    private static boolean libraryLoaded = false;
+
+    static {
+        try {
+            System.loadLibrary("dex_rust");
+            libraryLoaded = true;
+            Log.d(TAG, "Rust library loaded successfully");
+        } catch (UnsatisfiedLinkError e) {
+            Log.w(TAG, "Failed to load Rust library: " + e.getMessage());
+            libraryLoaded = false;
+        }
+    }
+
+    /**
+     * 检查 Rust 库是否可用
+     */
+    public static boolean isAvailable() {
+        return libraryLoaded;
+    }
+
+    /**
+     * 在 DEX 中搜索
+     * @param dexBytes DEX 文件字节数组
+     * @param query 搜索查询
+     * @param searchType 搜索类型: class, method, field, string, code
+     * @param caseSensitive 是否区分大小写
+     * @param maxResults 最大结果数
+     * @return JSON 格式的搜索结果
+     */
+    public static native String searchInDex(
+        byte[] dexBytes,
+        String query,
+        String searchType,
+        boolean caseSensitive,
+        int maxResults
+    );
+
+    /**
+     * 列出 DEX 中的类
+     * @param dexBytes DEX 文件字节数组
+     * @param packageFilter 包名过滤器
+     * @param offset 偏移量
+     * @param limit 限制数量
+     * @return JSON 格式的类列表
+     */
+    public static native String listClasses(
+        byte[] dexBytes,
+        String packageFilter,
+        int offset,
+        int limit
+    );
+
+    /**
+     * 获取类的 Smali 代码
+     * @param dexBytes DEX 文件字节数组
+     * @param className 类名
+     * @return JSON 格式的 Smali 代码
+     */
+    public static native String getClassSmali(
+        byte[] dexBytes,
+        String className
+    );
+
+    /**
+     * Java 回退实现 - 当 Rust 库不可用时使用
+     */
+    public static String searchInDexFallback(
+        byte[] dexBytes,
+        String query,
+        String searchType,
+        boolean caseSensitive,
+        int maxResults
+    ) {
+        return "{\"error\": \"Rust library not available, using Java implementation\", \"fallback\": true}";
+    }
+}

package/package.json

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