Skip to content

Commit

Permalink
refactor class ApkDecoder
Browse files Browse the repository at this point in the history
  • Loading branch information
@sv99
sv99 committed Jun 28, 2023
1 parent 40d427e commit e9a8455
Show file tree
Hide file tree
Showing 2 changed files with 124 additions and 129 deletions.
160 changes: 31 additions & 129 deletions brut.apktool/apktool-lib/src/main/java/brut/androlib/ApkDecoder.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,7 @@ public void decode(File outDir) throws AndrolibException, IOException, Directory
if (hasResources()) {
switch (config.decodeResources) {
case Config.DECODE_RESOURCES_NONE:
decodeResourcesRaw(mApkFile, outDir);
copyResourcesRaw(mApkFile, outDir);
if (config.forceDecodeManifest == Config.FORCE_DECODE_MANIFEST_FULL) {
// done after raw decoding of resources because copyToDir overwrites dest files
if (hasManifest()) {
Expand All @@ -135,15 +135,15 @@ public void decode(File outDir) throws AndrolibException, IOException, Directory
decodeManifestFull(mApkFile, outDir, getResTable());
}
else {
decodeManifestRaw(mApkFile, outDir);
copyManifestRaw(mApkFile, outDir);
}
}
}

if (hasSources()) {
switch (config.decodeSources) {
case Config.DECODE_SOURCES_NONE:
decodeSourcesRaw(mApkFile, outDir, "classes.dex");
copySourcesRaw(mApkFile, outDir, "classes.dex");
break;
case Config.DECODE_SOURCES_SMALI:
case Config.DECODE_SOURCES_SMALI_ONLY_MAIN_CLASSES:
Expand All @@ -160,7 +160,7 @@ public void decode(File outDir) throws AndrolibException, IOException, Directory
if (! file.equalsIgnoreCase("classes.dex")) {
switch(config.decodeSources) {
case Config.DECODE_SOURCES_NONE:
decodeSourcesRaw(mApkFile, outDir, file);
copySourcesRaw(mApkFile, outDir, file);
break;
case Config.DECODE_SOURCES_SMALI:
decodeSourcesSmali(mApkFile, outDir, file);
Expand All @@ -169,7 +169,7 @@ public void decode(File outDir) throws AndrolibException, IOException, Directory
if (file.startsWith("classes") && file.endsWith(".dex")) {
decodeSourcesSmali(mApkFile, outDir, file);
} else {
decodeSourcesRaw(mApkFile, outDir, file);
copySourcesRaw(mApkFile, outDir, file);
}
break;
}
Expand All @@ -178,11 +178,11 @@ public void decode(File outDir) throws AndrolibException, IOException, Directory
}
}

decodeRawFiles(mApkFile, outDir);
decodeUnknownFiles(mApkFile, outDir);
copyRawFiles(mApkFile, outDir);
copyUnknownFiles(mApkFile, outDir);
mUncompressedFiles = new ArrayList<>();
recordUncompressedFiles(mApkFile, mUncompressedFiles);
writeOriginalFiles(mApkFile, outDir);
copyOriginalFiles(mApkFile, outDir);
writeMetaFile(outDir);
} finally {
try {
Expand Down Expand Up @@ -250,25 +250,25 @@ public void close() throws IOException {
mAndRes.close();
}

public void writeMetaFile(File outDir)
throws AndrolibException {
public void writeMetaFile(File outDir) throws AndrolibException {
MetaInfo meta = new MetaInfo();
meta.version = ApktoolProperties.getVersion();
meta.apkFileName = mApkFile.getName();

if (mResTable != null) {
meta.isFrameworkApk = mResTable.isFrameworkApk();
putUsesFramework(meta);
putSdkInfo(outDir, meta);
putPackageInfo(meta);
putVersionInfo(outDir, meta);
putSharedLibraryInfo(meta);
putSparseResourcesInfo(meta);
mResTable.initMetaInfo(meta, outDir);
if (config.frameworkTag != null) {
meta.usesFramework.tag = config.frameworkTag;
}
} else {
putMinSdkInfo(meta);
if (mMinSdkVersion > 0) {
meta.sdkInfo = getMinSdkInfo();
}
}
meta.unknownFiles = mResUnknownFiles.getUnknownFiles();
if (mUncompressedFiles != null && !mUncompressedFiles.isEmpty()) {
meta.doNotCompress = mUncompressedFiles;
}
putUnknownInfo(meta);
putFileCompressionInfo(meta);

try {
meta.save(new File(outDir, "apktool.yml"));
Expand All @@ -277,111 +277,13 @@ public void writeMetaFile(File outDir)
}
}

private void putUsesFramework(MetaInfo meta) {
Set<ResPackage> pkgs = mResTable.listFramePackages();
if (pkgs.isEmpty()) {
return;
}

Integer[] ids = new Integer[pkgs.size()];
int i = 0;
for (ResPackage pkg : pkgs) {
ids[i++] = pkg.getId();
}
Arrays.sort(ids);

meta.usesFramework = new UsesFramework();
meta.usesFramework.ids = Arrays.asList(ids);

if (config.frameworkTag != null) {
meta.usesFramework.tag = config.frameworkTag;
}
}

private void putSdkInfo(File outDir, MetaInfo meta) {
Map<String, String> info = mResTable.getSdkInfo();
if (info.size() > 0) {
String refValue;
if (info.get("minSdkVersion") != null) {
refValue = ResXmlPatcher.pullValueFromIntegers(outDir, info.get("minSdkVersion"));
if (refValue != null) {
info.put("minSdkVersion", refValue);
}
}
if (info.get("targetSdkVersion") != null) {
refValue = ResXmlPatcher.pullValueFromIntegers(outDir, info.get("targetSdkVersion"));
if (refValue != null) {
info.put("targetSdkVersion", refValue);
}
}
if (info.get("maxSdkVersion") != null) {
refValue = ResXmlPatcher.pullValueFromIntegers(outDir, info.get("maxSdkVersion"));
if (refValue != null) {
info.put("maxSdkVersion", refValue);
}
}
meta.sdkInfo = info;
}
}

private void putMinSdkInfo(MetaInfo meta) {
if (mMinSdkVersion > 0) {
Map<String, String> sdkInfo = new LinkedHashMap<>();
sdkInfo.put("minSdkVersion", Integer.toString(mMinSdkVersion));
meta.sdkInfo = sdkInfo;
}
}

private void putPackageInfo(MetaInfo meta) throws AndrolibException {
String renamed = mResTable.getPackageRenamed();
String original = mResTable.getPackageOriginal();

int id = mResTable.getPackageId();
try {
id = mResTable.getPackage(renamed).getId();
} catch (UndefinedResObjectException ignored) {}

if (Strings.isNullOrEmpty(original)) {
return;
}

meta.packageInfo = new PackageInfo();

// only put rename-manifest-package into apktool.yml, if the change will be required
if (!renamed.equalsIgnoreCase(original)) {
meta.packageInfo.renameManifestPackage = renamed;
}
meta.packageInfo.forcedPackageId = String.valueOf(id);
}

private void putVersionInfo(File outDir, MetaInfo meta) {
VersionInfo info = mResTable.getVersionInfo();
String refValue = ResXmlPatcher.pullValueFromStrings(outDir, info.versionName);
if (refValue != null) {
info.versionName = refValue;
}
meta.versionInfo = info;
}

private void putSharedLibraryInfo(MetaInfo meta) {
meta.sharedLibrary = mResTable.getSharedLibrary();
}

private void putSparseResourcesInfo(MetaInfo meta) {
meta.sparseResources = mResTable.getSparseResources();
}

private void putUnknownInfo(MetaInfo meta) {
meta.unknownFiles = mResUnknownFiles.getUnknownFiles();
}

private void putFileCompressionInfo(MetaInfo meta) {
if (mUncompressedFiles != null && !mUncompressedFiles.isEmpty()) {
meta.doNotCompress = mUncompressedFiles;
}
private Map<String, String> getMinSdkInfo() {
Map<String, String> sdkInfo = new LinkedHashMap<>();
sdkInfo.put("minSdkVersion", Integer.toString(mMinSdkVersion));
return sdkInfo;
}

public void decodeSourcesRaw(ExtFile apkFile, File outDir, String filename)
public void copySourcesRaw(ExtFile apkFile, File outDir, String filename)
throws AndrolibException {
try {
LOGGER.info("Copying raw " + filename + " file...");
Expand Down Expand Up @@ -415,7 +317,7 @@ public void decodeSourcesSmali(File apkFile, File outDir, String filename)
}
}

public void decodeManifestRaw(ExtFile apkFile, File outDir)
public void copyManifestRaw(ExtFile apkFile, File outDir)
throws AndrolibException {
try {
LOGGER.info("Copying raw manifest...");
Expand All @@ -430,7 +332,7 @@ public void decodeManifestFull(ExtFile apkFile, File outDir, ResTable resTable)
mAndRes.decodeManifest(resTable, apkFile, outDir);
}

public void decodeResourcesRaw(ExtFile apkFile, File outDir)
public void copyResourcesRaw(ExtFile apkFile, File outDir)
throws AndrolibException {
try {
LOGGER.info("Copying raw resources...");
Expand All @@ -450,7 +352,7 @@ public void decodeManifestWithResources(ExtFile apkFile, File outDir, ResTable r
mAndRes.decodeManifestWithResources(resTable, apkFile, outDir);
}

public void decodeRawFiles(ExtFile apkFile, File outDir)
public void copyRawFiles(ExtFile apkFile, File outDir)
throws AndrolibException {
LOGGER.info("Copying assets and libs...");
try {
Expand Down Expand Up @@ -484,7 +386,7 @@ private boolean isAPKFileNames(String file) {
return false;
}

public void decodeUnknownFiles(ExtFile apkFile, File outDir)
public void copyUnknownFiles(ExtFile apkFile, File outDir)
throws AndrolibException {
LOGGER.info("Copying unknown files...");
File unknownOut = new File(outDir, UNK_DIRNAME);
Expand All @@ -508,7 +410,7 @@ public void decodeUnknownFiles(ExtFile apkFile, File outDir)
}
}

public void writeOriginalFiles(ExtFile apkFile, File outDir)
public void copyOriginalFiles(ExtFile apkFile, File outDir)
throws AndrolibException {
LOGGER.info("Copying original files...");
File originalDir = new File(outDir, "original");
Expand All @@ -531,7 +433,7 @@ public void writeOriginalFiles(ExtFile apkFile, File outDir)
if (in.containsDir("META-INF/services")) {
// If the original APK contains the folder META-INF/services folder
// that is used for service locators (like coroutines on android),
// copy it to the destination folder so it does not get dropped.
// copy it to the destination folder, so it does not get dropped.
LOGGER.info("Copying META-INF/services directory");
in.copyToDir(outDir, "META-INF/services");
}
Expand Down
93 changes: 93 additions & 0 deletions brut.apktool/apktool-lib/src/main/java/brut/androlib/res/data/ResTable.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -18,9 +18,16 @@

import brut.androlib.exceptions.AndrolibException;
import brut.androlib.exceptions.UndefinedResObjectException;
import brut.androlib.meta.MetaInfo;
import brut.androlib.meta.PackageInfo;
import brut.androlib.meta.UsesFramework;
import brut.androlib.meta.VersionInfo;
import brut.androlib.res.AndrolibResources;
import brut.androlib.res.data.value.ResValue;
import brut.androlib.res.xml.ResXmlPatcher;
import com.google.common.base.Strings;

import java.io.File;
import java.util.*;

public class ResTable {
Expand Down Expand Up @@ -220,4 +227,90 @@ public boolean isFrameworkApk() {
}
return false;
}

public void initMetaInfo(MetaInfo meta, File outDir) throws AndrolibException {
meta.isFrameworkApk = isFrameworkApk();
if (!listFramePackages().isEmpty()) {
meta.usesFramework = getUsesFramework();
}
if (!getSdkInfo().isEmpty()) {
initSdkInfo(outDir);
meta.sdkInfo = getSdkInfo();
}
meta.packageInfo = getPackageInfo();
meta.versionInfo = getVersionInfoWithName(outDir);
meta.sharedLibrary = getSharedLibrary();
meta.sparseResources = getSparseResources();
}

private UsesFramework getUsesFramework() {
Set<ResPackage> pkgs = listFramePackages();

Integer[] ids = new Integer[pkgs.size()];
int i = 0;
for (ResPackage pkg : pkgs) {
ids[i++] = pkg.getId();
}
Arrays.sort(ids);

UsesFramework info = new UsesFramework();
info.ids = Arrays.asList(ids);

return info;
}

private void initSdkInfo(File outDir) {
Map<String, String> info = mSdkInfo;
String refValue;
if (info.get("minSdkVersion") != null) {
refValue = ResXmlPatcher.pullValueFromIntegers(outDir, info.get("minSdkVersion"));
if (refValue != null) {
info.put("minSdkVersion", refValue);
}
}
if (info.get("targetSdkVersion") != null) {
refValue = ResXmlPatcher.pullValueFromIntegers(outDir, info.get("targetSdkVersion"));
if (refValue != null) {
info.put("targetSdkVersion", refValue);
}
}
if (info.get("maxSdkVersion") != null) {
refValue = ResXmlPatcher.pullValueFromIntegers(outDir, info.get("maxSdkVersion"));
if (refValue != null) {
info.put("maxSdkVersion", refValue);
}
}
}

private PackageInfo getPackageInfo() throws AndrolibException {
String renamed = getPackageRenamed();
String original = getPackageOriginal();

int id = getPackageId();
try {
id = getPackage(renamed).getId();
} catch (UndefinedResObjectException ignored) {}

if (Strings.isNullOrEmpty(original)) {
return null;
}

PackageInfo info = new PackageInfo();

// only put rename-manifest-package into apktool.yml, if the change will be required
if (!renamed.equalsIgnoreCase(original)) {
info.renameManifestPackage = renamed;
}
info.forcedPackageId = String.valueOf(id);
return info;
}

private VersionInfo getVersionInfoWithName(File outDir) {
VersionInfo info = getVersionInfo();
String refValue = ResXmlPatcher.pullValueFromStrings(outDir, info.versionName);
if (refValue != null) {
info.versionName = refValue;
}
return info;
}
}

0 comments on commit e9a8455

Please sign in to comment.