Examples with TinkerZipOutputStream com.tencent.tinker.ziputils.ziputil.TinkerZipOutputStream used on opensource projects

Example 1 with TinkerZipOutputStream

use of com.tencent.tinker.ziputils.ziputil.TinkerZipOutputStream in project tinker by Tencent.

the class ResDiffPatchInternal method extractResourceDiffInternals.

private static boolean extractResourceDiffInternals(Context context, String dir, String meta, File patchFile, int type) {
    ShareResPatchInfo resPatchInfo = new ShareResPatchInfo();
    ShareResPatchInfo.parseAllResPatchInfo(meta, resPatchInfo);
    ShareTinkerLog.i(TAG, "res dir: %s, meta: %s", dir, resPatchInfo.toString());
    Tinker manager = Tinker.with(context);
    if (!SharePatchFileUtil.checkIfMd5Valid(resPatchInfo.resArscMd5)) {
        ShareTinkerLog.w(TAG, "resource meta file md5 mismatch, type:%s, md5: %s", ShareTinkerInternals.getTypeString(type), resPatchInfo.resArscMd5);
        manager.getPatchReporter().onPatchPackageCheckFail(patchFile, BasePatchInternal.getMetaCorruptedCode(type));
        return false;
    }
    File directory = new File(dir);
    File tempResFileDirectory = new File(directory, "res_temp");
    File resOutput = new File(directory, ShareConstants.RES_NAME);
    // check result file whether already exist
    if (resOutput.exists()) {
        if (SharePatchFileUtil.checkResourceArscMd5(resOutput, resPatchInfo.resArscMd5)) {
            // it is ok, just continue
            ShareTinkerLog.w(TAG, "resource file %s is already exist, and md5 match, just return true", resOutput.getPath());
            return true;
        } else {
            ShareTinkerLog.w(TAG, "have a mismatch corrupted resource " + resOutput.getPath());
            resOutput.delete();
        }
    } else {
        resOutput.getParentFile().mkdirs();
    }
    try {
        ApplicationInfo applicationInfo = context.getApplicationInfo();
        if (applicationInfo == null) {
            // Looks like running on a test Context, so just return without patching.
            ShareTinkerLog.w(TAG, "applicationInfo == null!!!!");
            return false;
        }
        String apkPath = applicationInfo.sourceDir;
        if (!checkAndExtractResourceLargeFile(context, apkPath, directory, tempResFileDirectory, patchFile, resPatchInfo, type)) {
            return false;
        }
        TinkerZipOutputStream out = null;
        TinkerZipFile oldApk = null;
        TinkerZipFile newApk = null;
        int totalEntryCount = 0;
        try {
            out = new TinkerZipOutputStream(new BufferedOutputStream(new FileOutputStream(resOutput)));
            oldApk = new TinkerZipFile(apkPath);
            newApk = new TinkerZipFile(patchFile);
            final Enumeration<? extends TinkerZipEntry> entries = oldApk.entries();
            while (entries.hasMoreElements()) {
                TinkerZipEntry zipEntry = entries.nextElement();
                if (zipEntry == null) {
                    throw new TinkerRuntimeException("zipEntry is null when get from oldApk");
                }
                String name = zipEntry.getName();
                if (name.contains("../")) {
                    continue;
                }
                if (ShareResPatchInfo.checkFileInPattern(resPatchInfo.patterns, name)) {
                    // won't contain in add set.
                    if (!resPatchInfo.deleteRes.contains(name) && !resPatchInfo.modRes.contains(name) && !resPatchInfo.largeModRes.contains(name) && !name.equals(ShareConstants.RES_MANIFEST)) {
                        TinkerZipUtil.extractTinkerEntry(oldApk, zipEntry, out);
                        totalEntryCount++;
                    }
                }
            }
            // process manifest
            TinkerZipEntry manifestZipEntry = oldApk.getEntry(ShareConstants.RES_MANIFEST);
            if (manifestZipEntry == null) {
                ShareTinkerLog.w(TAG, "manifest patch entry is null. path:" + ShareConstants.RES_MANIFEST);
                manager.getPatchReporter().onPatchTypeExtractFail(patchFile, resOutput, ShareConstants.RES_MANIFEST, type);
                return false;
            }
            TinkerZipUtil.extractTinkerEntry(oldApk, manifestZipEntry, out);
            totalEntryCount++;
            for (String name : resPatchInfo.largeModRes) {
                TinkerZipEntry largeZipEntry = oldApk.getEntry(name);
                if (largeZipEntry == null) {
                    ShareTinkerLog.w(TAG, "large patch entry is null. path:" + name);
                    manager.getPatchReporter().onPatchTypeExtractFail(patchFile, resOutput, name, type);
                    return false;
                }
                ShareResPatchInfo.LargeModeInfo largeModeInfo = resPatchInfo.largeModMap.get(name);
                TinkerZipUtil.extractLargeModifyFile(largeZipEntry, largeModeInfo.file, largeModeInfo.crc, out);
                totalEntryCount++;
            }
            for (String name : resPatchInfo.addRes) {
                TinkerZipEntry addZipEntry = newApk.getEntry(name);
                if (addZipEntry == null) {
                    ShareTinkerLog.w(TAG, "add patch entry is null. path:" + name);
                    manager.getPatchReporter().onPatchTypeExtractFail(patchFile, resOutput, name, type);
                    return false;
                }
                if (resPatchInfo.storeRes.containsKey(name)) {
                    File storeFile = resPatchInfo.storeRes.get(name);
                    TinkerZipUtil.extractLargeModifyFile(addZipEntry, storeFile, addZipEntry.getCrc(), out);
                } else {
                    TinkerZipUtil.extractTinkerEntry(newApk, addZipEntry, out);
                }
                totalEntryCount++;
            }
            for (String name : resPatchInfo.modRes) {
                TinkerZipEntry modZipEntry = newApk.getEntry(name);
                if (modZipEntry == null) {
                    ShareTinkerLog.w(TAG, "mod patch entry is null. path:" + name);
                    manager.getPatchReporter().onPatchTypeExtractFail(patchFile, resOutput, name, type);
                    return false;
                }
                if (resPatchInfo.storeRes.containsKey(name)) {
                    File storeFile = resPatchInfo.storeRes.get(name);
                    TinkerZipUtil.extractLargeModifyFile(modZipEntry, storeFile, modZipEntry.getCrc(), out);
                } else {
                    TinkerZipUtil.extractTinkerEntry(newApk, modZipEntry, out);
                }
                totalEntryCount++;
            }
            // set comment back
            out.setComment(oldApk.getComment());
        } finally {
            IOHelper.closeQuietly(out);
            IOHelper.closeQuietly(oldApk);
            IOHelper.closeQuietly(newApk);
            // delete temp files
            SharePatchFileUtil.deleteDir(tempResFileDirectory);
        }
        boolean result = SharePatchFileUtil.checkResourceArscMd5(resOutput, resPatchInfo.resArscMd5);
        if (!result) {
            ShareTinkerLog.i(TAG, "check final new resource file fail path:%s, entry count:%d, size:%d", resOutput.getAbsolutePath(), totalEntryCount, resOutput.length());
            SharePatchFileUtil.safeDeleteFile(resOutput);
            manager.getPatchReporter().onPatchTypeExtractFail(patchFile, resOutput, ShareConstants.RES_NAME, type);
            return false;
        }
        ShareTinkerLog.i(TAG, "final new resource file:%s, entry count:%d, size:%d", resOutput.getAbsolutePath(), totalEntryCount, resOutput.length());
    } catch (Throwable e) {
        throw new TinkerRuntimeException("patch " + ShareTinkerInternals.getTypeString(type) + " extract failed (" + e.getMessage() + ").", e);
    }
    return true;
}

Example 2 with TinkerZipOutputStream

use of com.tencent.tinker.ziputils.ziputil.TinkerZipOutputStream in project tinker by Tencent.

the class Utils method genResOutputFile.

public static String genResOutputFile(File output, File newZipFile, Configuration config, ArrayList<String> addedSet, ArrayList<String> modifiedSet, ArrayList<String> deletedSet, ArrayList<String> largeModifiedSet, HashMap<String, ResDiffDecoder.LargeModeInfo> largeModifiedMap) throws IOException {
    TinkerZipFile oldApk = null;
    TinkerZipFile newApk = null;
    TinkerZipOutputStream out = null;
    try {
        oldApk = new TinkerZipFile(config.mOldApkFile);
        newApk = new TinkerZipFile(newZipFile);
        out = new TinkerZipOutputStream(new BufferedOutputStream(new FileOutputStream(output)));
        final Enumeration<? extends TinkerZipEntry> entries = oldApk.entries();
        while (entries.hasMoreElements()) {
            TinkerZipEntry zipEntry = entries.nextElement();
            if (zipEntry == null) {
                throw new TinkerPatchException(String.format("zipEntry is null when get from oldApk"));
            }
            String name = zipEntry.getName();
            if (!TinkerZipUtil.validateZipEntryName(output.getParentFile(), name)) {
                throw new IOException("Bad ZipEntry name: " + name);
            }
            if (Utils.checkFileInPattern(config.mResFilePattern, name)) {
                // won't contain in add set.
                if (!deletedSet.contains(name) && !modifiedSet.contains(name) && !largeModifiedSet.contains(name) && !name.equals(TypedValue.RES_MANIFEST)) {
                    TinkerZipUtil.extractTinkerEntry(oldApk, zipEntry, out);
                }
            }
        }
        // process manifest
        TinkerZipEntry manifestZipEntry = oldApk.getEntry(TypedValue.RES_MANIFEST);
        if (manifestZipEntry == null) {
            throw new TinkerPatchException(String.format("can't found resource file %s from old apk file %s", TypedValue.RES_MANIFEST, config.mOldApkFile.getAbsolutePath()));
        }
        TinkerZipUtil.extractTinkerEntry(oldApk, manifestZipEntry, out);
        for (String name : largeModifiedSet) {
            TinkerZipEntry largeZipEntry = oldApk.getEntry(name);
            if (largeZipEntry == null) {
                throw new TinkerPatchException(String.format("can't found resource file %s from old apk file %s", name, config.mOldApkFile.getAbsolutePath()));
            }
            ResDiffDecoder.LargeModeInfo largeModeInfo = largeModifiedMap.get(name);
            TinkerZipUtil.extractLargeModifyFile(largeZipEntry, largeModeInfo.path, largeModeInfo.crc, out);
        }
        for (String name : addedSet) {
            TinkerZipEntry addZipEntry = newApk.getEntry(name);
            if (addZipEntry == null) {
                throw new TinkerPatchException(String.format("can't found add resource file %s from new apk file %s", name, config.mNewApkFile.getAbsolutePath()));
            }
            TinkerZipUtil.extractTinkerEntry(newApk, addZipEntry, out);
        }
        for (String name : modifiedSet) {
            TinkerZipEntry modZipEntry = newApk.getEntry(name);
            if (modZipEntry == null) {
                throw new TinkerPatchException(String.format("can't found add resource file %s from new apk file %s", name, config.mNewApkFile.getAbsolutePath()));
            }
            TinkerZipUtil.extractTinkerEntry(newApk, modZipEntry, out);
        }
    } finally {
        IOHelper.closeQuietly(out);
        IOHelper.closeQuietly(oldApk);
        IOHelper.closeQuietly(newApk);
    }
    return MD5.getMD5(output);
}