perf: --rip-lib and --unsigned

E85Addict · Mar 10, 2024 · 54201c6 · 54201c6
1 parent 26e580b
commit 54201c6
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 3 deletions.
diff --git a/revanced-cli/src/main/kotlin/app/revanced/cli/command/PatchCommand.kt b/revanced-cli/src/main/kotlin/app/revanced/cli/command/PatchCommand.kt
@@ -161,6 +161,18 @@ internal object PatchCommand : Runnable {
 
     private var aaptBinaryPath: File? = null
 
+    @CommandLine.Option(
+        names = ["--unsigned"],
+        description = ["Disable signing of the final apk."],
+    )
+    private var unsigned: Boolean = false
+
+    @CommandLine.Option(
+        names = ["--rip-lib"],
+        description = ["Rip native libs from APK (x86_64 etc.)"],
+    )
+    private var ripLibs = arrayOf<String>()
+
     @CommandLine.Option(
         names = ["-p", "--purge"],
         description = ["Purge the temporary resource cache directory after patching."],
@@ -318,9 +330,9 @@ internal object PatchCommand : Runnable {
 
         apk.copyTo(outputFilePath, overwrite = true)
 
-        patcherResult.applyTo(outputFilePath)
+        patcherResult.applyTo(outputFilePath, ripLibs)
 
-        if (!mount) {
+        if (!mount && !unsigned) {
             outputFilePath.sign(
                 ApkUtils.SigningOptions(
                     keystoreFilePath,

diff --git a/revanced-lib/src/main/kotlin/app/revanced/library/ApkUtils.kt b/revanced-lib/src/main/kotlin/app/revanced/library/ApkUtils.kt
@@ -46,8 +46,9 @@ object ApkUtils {
      * 6. Realign the APK.
      *
      * @param apkFile The file to apply the patched files to.
+     * @param ripLibs The string of libraries to remove.
      */
-    fun PatcherResult.applyTo(apkFile: File) {
+    fun PatcherResult.applyTo(apkFile: File, ripLibs: Array<String> ) {
         ZFile.openReadWrite(apkFile, zFileOptions).use { targetApkZFile ->
             dexFiles.forEach { dexFile ->
                 targetApkZFile.add(dexFile.name, dexFile.stream)
@@ -88,6 +89,11 @@ object ApkUtils {
 
             targetApkZFile.realign()
 
+            // Delete all the libraries from the --rip-lib command.
+            targetApkZFile.entries().filter { entry -> 
+                ripLibs.any { entry.centralDirectoryHeader.name.startsWith("lib/$it") } 
+            }.forEach(StoredEntry::delete)
+
             logger.fine("Writing changes")
         }
     }