From fe67d0f24020fc7471d34ea3be43d90c89b66f17 Mon Sep 17 00:00:00 2001
From: Connor Tumbleson <connor.tumbleson@gmail.com>
Date: Sun, 13 Nov 2022 17:59:04 -0500
Subject: [PATCH] fix: support properly mapping r/R/res resources during
 disassemble

---
 .../brut/androlib/res/AndrolibResources.java  | 14 ++-------
 .../androlib/res/decoder/ResFileDecoder.java  | 30 +++++++++----------
 2 files changed, 16 insertions(+), 28 deletions(-)

diff --git a/brut.apktool/apktool-lib/src/main/java/brut/androlib/res/AndrolibResources.java b/brut.apktool/apktool-lib/src/main/java/brut/androlib/res/AndrolibResources.java
index ded97b98c7..9f2499e34f 100644
--- a/brut.apktool/apktool-lib/src/main/java/brut/androlib/res/AndrolibResources.java
+++ b/brut.apktool/apktool-lib/src/main/java/brut/androlib/res/AndrolibResources.java
@@ -221,22 +221,12 @@ public void decode(ResTable resTable, ExtFile apkFile, File outDir)
         ResAttrDecoder attrDecoder = duo.m2.getAttrDecoder();
 
         attrDecoder.setCurrentPackage(resTable.listMainPackages().iterator().next());
-        Directory inApk, in = null, out;
+        Directory in, out;
 
         try {
             out = new FileDirectory(outDir);
-
-            inApk = apkFile.getDirectory();
+            in = apkFile.getDirectory();
             out = out.createDir("res");
-            if (inApk.containsDir("res")) {
-                in = inApk.getDir("res");
-            }
-            if (in == null && inApk.containsDir("r")) {
-                in = inApk.getDir("r");
-            }
-            if (in == null && inApk.containsDir("R")) {
-                in = inApk.getDir("R");
-            }
         } catch (DirectoryException ex) {
             throw new AndrolibException(ex);
         }
diff --git a/brut.apktool/apktool-lib/src/main/java/brut/androlib/res/decoder/ResFileDecoder.java b/brut.apktool/apktool-lib/src/main/java/brut/androlib/res/decoder/ResFileDecoder.java
index 75f67a8c79..a3174411f3 100644
--- a/brut.apktool/apktool-lib/src/main/java/brut/androlib/res/decoder/ResFileDecoder.java
+++ b/brut.apktool/apktool-lib/src/main/java/brut/androlib/res/decoder/ResFileDecoder.java
@@ -64,11 +64,11 @@ public void decode(ResResource res, Directory inDir, Directory outDir, Map<Strin
 
         try {
             if (typeName.equals("raw")) {
-                decode(inDir, inFileName, outDir, outFileName, "raw");
+                decode(inDir, inFilePath, outDir, outFileName, "raw");
                 return;
             }
             if (typeName.equals("font") && !".xml".equals(ext)) {
-                decode(inDir, inFileName, outDir, outFileName, "raw");
+                decode(inDir, inFilePath, outDir, outFileName, "raw");
                 return;
             }
             if (typeName.equals("drawable") || typeName.equals("mipmap")) {
@@ -83,26 +83,24 @@ public void decode(ResResource res, Directory inDir, Directory outDir, Map<Strin
                     // check for raw 9patch images
                     for (String extension : RAW_9PATCH_IMAGE_EXTENSIONS) {
                         if (inFileName.toLowerCase().endsWith("." + extension)) {
-                            copyRaw(inDir, outDir, inFileName, outFileName);
+                            copyRaw(inDir, outDir, inFilePath, outFileName);
                             return;
                         }
                     }
 
                     // check for xml 9 patches which are just xml files
                     if (inFileName.toLowerCase().endsWith(".xml")) {
-                        decode(inDir, inFileName, outDir, outFileName, "xml");
+                        decode(inDir, inFilePath, outDir, outFileName, "xml");
                         return;
                     }
 
                     try {
-                        decode(inDir, inFileName, outDir, outFileName, "9patch");
+                        decode(inDir, inFilePath, outDir, outFileName, "9patch");
                         return;
                     } catch (CantFind9PatchChunkException ex) {
-                        LOGGER.log(
-                                Level.WARNING,
-                                String.format(
-                                        "Cant find 9patch chunk in file: \"%s\". Renaming it to *.png.",
-                                        inFileName), ex);
+                        LOGGER.log(Level.WARNING, String.format(
+                            "Cant find 9patch chunk in file: \"%s\". Renaming it to *.png.", inFileName
+                        ), ex);
                         outDir.removeFile(outFileName);
                         outFileName = outResName + ext;
                     }
@@ -111,27 +109,27 @@ public void decode(ResResource res, Directory inDir, Directory outDir, Map<Strin
                 // check for raw image
                 for (String extension : RAW_IMAGE_EXTENSIONS) {
                     if (inFileName.toLowerCase().endsWith("." + extension)) {
-                        copyRaw(inDir, outDir, inFileName, outFileName);
+                        copyRaw(inDir, outDir, inFilePath, outFileName);
                         return;
                     }
                 }
 
                 if (!".xml".equals(ext)) {
-                    decode(inDir, inFileName, outDir, outFileName, "raw");
+                    decode(inDir, inFilePath, outDir, outFileName, "raw");
                     return;
                 }
             }
 
-            decode(inDir, inFileName, outDir, outFileName, "xml");
+            decode(inDir, inFilePath, outDir, outFileName, "xml");
         } catch (RawXmlEncounteredException ex) {
             // If we got an error to decode XML, lets assume the file is in raw format.
             // This is a large assumption, that might increase runtime, but will save us for situations where
             // XSD files are AXML`d on aapt1, but left in plaintext in aapt2.
-            decode(inDir, inFileName, outDir, outFileName, "raw");
+            decode(inDir, inFilePath, outDir, outFileName, "raw");
         } catch (AndrolibException ex) {
             LOGGER.log(Level.SEVERE, String.format(
-                    "Could not decode file, replacing by FALSE value: %s",
-                    inFileName), ex);
+                "Could not decode file, replacing by FALSE value: %s",
+            inFileName), ex);
             res.replace(new ResBoolValue(false, 0, null));
         }
     }