com.tencent.mm.resourceproguard.Main Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of AndResGuard-core Show documentation
proguard resource for Android by wechat team. 迁移到Gradle 8+
The newest version!
package com.tencent.mm.resourceproguard;

import com.tencent.mm.androlib.AndrolibException;
import com.tencent.mm.androlib.ApkDecoder;
import com.tencent.mm.androlib.ResourceApkBuilder;
import com.tencent.mm.androlib.res.decoder.ARSCDecoder;
import com.tencent.mm.androlib.res.util.StringUtil;
import com.tencent.mm.directory.DirectoryException;
import com.tencent.mm.util.FileOperation;

import java.io.File;
import java.io.IOException;
import java.util.Objects;

/**
 * @author shwenzhang
 * @author simsun
 */
public class Main {

    public static final int ERRNO_ERRORS = 1;
    public static final int ERRNO_USAGE = 2;
    protected static long mRawApkSize;
    protected static String mRunningLocation;
    protected static long mBeginTime;

    /**
     * 是否通过命令行方式设置
     **/
    public boolean mSetSignThroughCmd;
    public boolean mSetMappingThroughCmd;
    public String m7zipPath;
    public String mZipalignPath;
    public String mFinalApkBackPath;

    protected Configuration config;
    protected File mOutDir;

    public static void gradleRun(InputParam inputParam) {
        Main m = new Main();
        m.run(inputParam);
    }

    @SuppressWarnings("deprecation")
    private void run(InputParam inputParam) {
        synchronized (Main.class) {
            loadConfigFromGradle(inputParam);
            this.mFinalApkBackPath = inputParam.finalApkBackupPath;
            Thread currentThread = Thread.currentThread();
            System.out.printf(
                    "\n-->AndResGuard starting! Current thread# id: %d, name: %s\n",
                    currentThread.getId(),
                    currentThread.getName()
            );
            File finalApkFile = StringUtil.isPresent(inputParam.finalApkBackupPath) ? new File(inputParam.finalApkBackupPath) : null;
            if (finalApkFile == null) return;
            String filePath = finalApkFile.getAbsolutePath();
            System.out.printf("origin file path: " + filePath);
            File copyFile = new File(finalApkFile.getParent(), "backup.apk");
            try {
                FileOperation.copyFileUsingStream(finalApkFile, copyFile);
                System.out.print("backup file success");
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
            resourceProguard(
                    new File(inputParam.outFolder),
                    finalApkFile,
                    inputParam.apkPath,
                    inputParam.signatureType,
                    inputParam.minSDKVersion
            );
            System.out.println("<--AndResGuard Done!");
            clean();
        }
    }

    protected void clean() {
        config = null;
        ARSCDecoder.mTableStringsResguard.clear();
        ARSCDecoder.mMergeDuplicatedResCount = 0;
        ARSCDecoder.dispose();
    }

    private void loadConfigFromGradle(InputParam inputParam) {
        try {
            config = new Configuration(inputParam);
        } catch (IOException e) {
            e.fillInStackTrace();
        }
    }

    protected void resourceProguard(
            File outputDir, File outputFile, String apkFilePath, InputParam.SignatureType signatureType) {
        resourceProguard(outputDir, outputFile, apkFilePath, signatureType, 14 /*default min sdk*/);
    }

    protected void resourceProguard(
            File outputDir, File outputFile, String apkFilePath, InputParam.SignatureType signatureType, int minSDKVersoin) {
        File apkFile = new File(apkFilePath);
        if (!apkFile.exists()) {
            System.err.printf("The input apk %s does not exist", apkFile.getAbsolutePath());
            goToError();
        }
        mRawApkSize = FileOperation.getFileSizes(apkFile);
        try {
            ApkDecoder decoder = new ApkDecoder(config, apkFile);
            /* 默认使用V1签名 */
            decodeResource(outputDir, decoder, apkFile);
            buildApk(decoder, apkFile, outputFile, signatureType, minSDKVersoin);
        } catch (Exception e) {
            e.fillInStackTrace();
            goToError();
        }
    }

    private void decodeResource(File outputFile, ApkDecoder decoder, File apkFile)
            throws AndrolibException, IOException, DirectoryException {
        mOutDir = Objects.requireNonNullElseGet(outputFile, () -> new File(mRunningLocation, apkFile.getName().substring(0, apkFile.getName().indexOf(".apk"))));
        decoder.setOutDir(mOutDir.getAbsoluteFile());
        decoder.decode();
    }

    private void buildApk(
            ApkDecoder decoder, File apkFile, File outputFile, InputParam.SignatureType signatureType, int minSDKVersion)
            throws Exception {
        ResourceApkBuilder builder = new ResourceApkBuilder(config);
        String apkBasename = apkFile.getName();
        apkBasename = apkBasename.substring(0, apkBasename.indexOf(".apk"));
        builder.setOutDir(mOutDir, apkBasename, outputFile);
        System.out.printf("[AndResGuard] buildApk signatureType: %s\n", signatureType);
        switch (signatureType) {
            case SchemaV1:
                builder.buildApkWithV1sign(decoder.getCompressData());
                break;
            case SchemaV2:
            case SchemaV3:
                builder.buildApkWithV2V3Sign(decoder.getCompressData(), minSDKVersion, signatureType);
                break;
        }
    }

    protected void goToError() {
        System.exit(ERRNO_USAGE);
    }
}