TheArchive/Magisk
1
0
Fork 0
mirror of https://github.com/topjohnwu/Magisk.git synced 2024-12-22 07:57:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Build dynamic stub resource APK at runtime

Browse Source 
Close #6013

Co-authored-by: vvb2060 <vvb2060@gmail.com>
This commit is contained in:
topjohnwu 2022-06-22 05:19:27 -07:00
parent a2495fb5fb
commit 0b26882fce
2 changed files with 38 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
buildSrc/src/main/java/Setup.kt
View File

@ -21,7 +21,6 @@ import java.io.ByteArrayInputStream
import java.io.ByteArrayOutputStream import java.io.ByteArrayOutputStream
import java.io.File import java.io.File
import java.io.PrintStream import java.io.PrintStream
import java.nio.file.Files
import java.security.KeyStore import java.security.KeyStore
import java.security.cert.X509Certificate import java.security.cert.X509Certificate
import java.util.* import java.util.*
@ -270,27 +269,20 @@ fun Project.setupStub() {
commandLine(aapt, "optimize", "-o", apkTmp, "--collapse-resource-names", apk) commandLine(aapt, "optimize", "-o", apkTmp, "--collapse-resource-names", apk)
} }
val buffer = ByteArrayOutputStream() val bos = ByteArrayOutputStream()
apkTmp.inputStream().use { ZipFile(apkTmp).use { src ->
object : GZIPOutputStream(buffer) { ZipOutputStream(apk.outputStream()).use {
init { it.setLevel(Deflater.BEST_COMPRESSION)
def.setLevel(Deflater.BEST_COMPRESSION) it.putNextEntry(ZipEntry("AndroidManifest.xml"))
} src.getInputStream(src.getEntry("AndroidManifest.xml")).transferTo(it)
}.use { o -> it.closeEntry()
it.transferTo(o)
} }
} DeflaterOutputStream(bos, Deflater(Deflater.BEST_COMPRESSION)).use {
ZipFile(apkTmp).use { o -> src.getInputStream(src.getEntry("resources.arsc")).transferTo(it)
ZipOutputStream(apk.outputStream()).use { n ->
n.setLevel(Deflater.BEST_COMPRESSION)
n.putNextEntry(ZipEntry("AndroidManifest.xml"))
o.getInputStream(o.getEntry("AndroidManifest.xml")).transferTo(n)
n.closeEntry()
n.finish()
} }
} }
apkTmp.delete() apkTmp.delete()
genEncryptedResources(ByteArrayInputStream(buffer.toByteArray()), outSrcDir) genEncryptedResources(ByteArrayInputStream(bos.toByteArray()), outSrcDir)
} }
} }
registerJavaGeneratingTask(genSrcTask, outSrcDir) registerJavaGeneratingTask(genSrcTask, outSrcDir)

39
stub/src/main/java/com/topjohnwu/magisk/DownloadActivity.java
View File

@ -34,7 +34,10 @@ import java.io.File;
import java.io.FileOutputStream; import java.io.FileOutputStream;
import java.io.IOException; import java.io.IOException;
import java.io.OutputStream; import java.io.OutputStream;
import java.util.zip.GZIPInputStream; import java.util.zip.InflaterInputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipOutputStream;
import javax.crypto.Cipher; import javax.crypto.Cipher;
import javax.crypto.CipherInputStream; import javax.crypto.CipherInputStream;
@ -76,7 +79,9 @@ public class DownloadActivity extends Activity {
// Inject resources // Inject resources
try { try {
loadResources(); loadResources();
} catch (Exception ignored) {} } catch (Exception e) {
error(e);
}
ProviderInstaller.install(this); ProviderInstaller.install(this);
@ -103,7 +108,7 @@ public class DownloadActivity extends Activity {
} }
private void error(Throwable e) { private void error(Throwable e) {
Log.e(getClass().getSimpleName(), "", e); Log.e(getClass().getSimpleName(), Log.getStackTraceString(e));
finish(); finish();
} }
@ -157,14 +162,26 @@ public class DownloadActivity extends Activity {
} }
private void decryptResources(OutputStream out) throws Exception { private void decryptResources(OutputStream out) throws Exception {
Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding"); try (var zip = new ZipOutputStream(out)) {
SecretKey key = new SecretKeySpec(Bytes.key(), "AES"); zip.putNextEntry(new ZipEntry("resources.arsc"));
IvParameterSpec iv = new IvParameterSpec(Bytes.iv()); Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");
cipher.init(Cipher.DECRYPT_MODE, key, iv); SecretKey key = new SecretKeySpec(Bytes.key(), "AES");
var is = new GZIPInputStream(new CipherInputStream( IvParameterSpec iv = new IvParameterSpec(Bytes.iv());
new ByteArrayInputStream(Bytes.res()), cipher)); cipher.init(Cipher.DECRYPT_MODE, key, iv);
try (is; out) { var is = new InflaterInputStream(new CipherInputStream(
APKInstall.transfer(is, out); new ByteArrayInputStream(Bytes.res()), cipher));
try (is) {
APKInstall.transfer(is, zip);
}
zip.closeEntry();
zip.putNextEntry(new ZipEntry("AndroidManifest.xml"));
var apk = new ZipFile(getPackageResourcePath());
var xml = apk.getInputStream(apk.getEntry("AndroidManifest.xml"));
try (apk; xml) {
APKInstall.transfer(xml, zip);
}
zip.closeEntry();
} }
} }