WeiguangTWK
/
GKMSPatch
1
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Activity

Remove unnecessary files for stub

This commit is contained in:
Nullptr 2021-07-15 18:20:42 +08:00 committed by LoveSy
parent 34000e3ece
commit c6af951364
1 changed files with 22 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
patch/src/main/java/org/lsposed/patch/LSPatch.java
View File

@ -1,8 +1,10 @@
package org.lsposed.patch; package org.lsposed.patch;
import com.android.apksig.ApkSigner; import com.android.apksig.ApkSigner;
import com.android.tools.build.apkzlib.zip.AlignmentRules;
import com.android.tools.build.apkzlib.zip.StoredEntry; import com.android.tools.build.apkzlib.zip.StoredEntry;
import com.android.tools.build.apkzlib.zip.ZFile; import com.android.tools.build.apkzlib.zip.ZFile;
import com.android.tools.build.apkzlib.zip.ZFileOptions;
import com.beust.jcommander.JCommander; import com.beust.jcommander.JCommander;
import com.beust.jcommander.Parameter; import com.beust.jcommander.Parameter;
import com.wind.meditor.core.ManifestEditor; import com.wind.meditor.core.ManifestEditor;
@ -76,12 +78,11 @@ public class LSPatch {
@Parameter(names = {"-m", "--embed"}, description = "Embed provided modules to apk") @Parameter(names = {"-m", "--embed"}, description = "Embed provided modules to apk")
private List<String> modules = new ArrayList<>(); private List<String> modules = new ArrayList<>();
private int dexFileCount = 0;
private static final String APPLICATION_NAME_ASSET_PATH = "assets/original_application_name.ini"; private static final String APPLICATION_NAME_ASSET_PATH = "assets/original_application_name.ini";
private static final String SIGNATURE_INFO_ASSET_PATH = "assets/original_signature_info.ini"; private static final String SIGNATURE_INFO_ASSET_PATH = "assets/original_signature_info.ini";
private static final String ORIGIN_APK_ASSET_PATH = "assets/origin_apk.bin"; private static final String ORIGIN_APK_ASSET_PATH = "assets/origin_apk.bin";
private static final String ANDROID_MANIFEST_XML = "AndroidManifest.xml"; private static final String ANDROID_MANIFEST_XML = "AndroidManifest.xml";
private static final String RESOURCES_ARSC = "resources.arsc";
private static final HashSet<String> APK_LIB_PATH_ARRAY = new HashSet<>(Arrays.asList( private static final HashSet<String> APK_LIB_PATH_ARRAY = new HashSet<>(Arrays.asList(
// "armeabi", // "armeabi",
"armeabi-v7a", "armeabi-v7a",
@ -147,7 +148,23 @@ public class LSPatch {
System.out.println("Parsing original apk..."); System.out.println("Parsing original apk...");
try (ZFile zFile = ZFile.openReadWrite(tmpApk)) { final ZFileOptions zFileOptions = new ZFileOptions();
final var alignmentRule = AlignmentRules.compose(
AlignmentRules.constantForSuffix(RESOURCES_ARSC, 4),
AlignmentRules.constantForSuffix(ORIGIN_APK_ASSET_PATH, 4096)
);
zFileOptions.setAlignmentRule(alignmentRule);
try (ZFile zFile = ZFile.openReadWrite(tmpApk, zFileOptions)) {
// copy origin apk to assets
zFile.add(ORIGIN_APK_ASSET_PATH, new FileInputStream(srcApkFile), false);
// remove unnecessary files
for (StoredEntry storedEntry : zFile.entries()) {
var name = storedEntry.getCentralDirectoryHeader().getName();
if (name.endsWith(".dex")) storedEntry.delete();
}
if (sigbypassLevel > 0) { if (sigbypassLevel > 0) {
// save the apk original signature info, to support crack signature. // save the apk original signature info, to support crack signature.
String originalSignature = ApkSignatureHelper.getApkSignInfo(srcApkFile.getAbsolutePath()); String originalSignature = ApkSignatureHelper.getApkSignInfo(srcApkFile.getAbsolutePath());
@ -163,12 +180,6 @@ public class LSPatch {
} }
} }
// get the dex count in the apk zip file
dexFileCount = findDexFileCount(zFile);
if (verbose)
System.out.println("dexFileCount: " + dexFileCount);
// copy out manifest file from zlib // copy out manifest file from zlib
var manifestEntry = zFile.get(ANDROID_MANIFEST_XML); var manifestEntry = zFile.get(ANDROID_MANIFEST_XML);
if (manifestEntry == null) if (manifestEntry == null)
@ -241,19 +252,11 @@ public class LSPatch {
System.out.println("Adding dex.."); System.out.println("Adding dex..");
try (var is = getClass().getClassLoader().getResourceAsStream("assets/dex/loader.dex")) { try (var is = getClass().getClassLoader().getResourceAsStream("assets/dex/loader.dex")) {
String copiedDexFileName = "classes" + (dexFileCount + 1) + ".dex"; zFile.add("classes.dex", is);
zFile.add(copiedDexFileName, is);
dexFileCount++;
} catch (Throwable e) { } catch (Throwable e) {
throw new PatchError("Error when add dex: " + e); throw new PatchError("Error when add dex: " + e);
} }
// copy origin apk to assets
// convenient to bypass some check like CRC
if (sigbypassLevel >= Constants.SIGBYPASS_LV_PM_OPENAT) {
zFile.add(ORIGIN_APK_ASSET_PATH, new FileInputStream(srcApkFile));
}
try (var is = getClass().getClassLoader().getResourceAsStream("assets/dex/lsp.dex")) { try (var is = getClass().getClassLoader().getResourceAsStream("assets/dex/lsp.dex")) {
zFile.add("assets/lsp", is); zFile.add("assets/lsp", is);
} catch (Throwable e) { } catch (Throwable e) {
@ -274,6 +277,7 @@ public class LSPatch {
if (sign != null) if (sign != null)
sign.delete(); sign.delete();
zFile.realign();
zFile.update(); zFile.update();
signApkUsingAndroidApksigner(tmpApk, outputFile); signApkUsingAndroidApksigner(tmpApk, outputFile);
@ -341,13 +345,6 @@ public class LSPatch {
return os.toByteArray(); return os.toByteArray();
} }
private int findDexFileCount(ZFile zFile) {
for (int i = 2; ; i++) {
if (zFile.get("classes" + i + ".dex") == null)
return i - 1;
}
}
private void signApkUsingAndroidApksigner(File apkPath, File outputPath) throws PatchError { private void signApkUsingAndroidApksigner(File apkPath, File outputPath) throws PatchError {
try { try {
var keyStore = KeyStore.getInstance(KeyStore.getDefaultType()); var keyStore = KeyStore.getInstance(KeyStore.getDefaultType());