19066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project// 29066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project// Copyright 2006 The Android Open Source Project 39066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project// 49066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project// Package assets into Zip files. 59066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project// 69066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project#include "Main.h" 79066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project#include "AaptAssets.h" 89066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project#include "ResourceTable.h" 9e6b680364dd992907a8d2037685a2e500d188dfbDianne Hackborn#include "ResourceFilter.h" 109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 113b4062eeb01be33951ba214e027da523cf09f8b1Mathias Agopian#include <utils/Log.h> 123b4062eeb01be33951ba214e027da523cf09f8b1Mathias Agopian#include <utils/threads.h> 133b4062eeb01be33951ba214e027da523cf09f8b1Mathias Agopian#include <utils/List.h> 143b4062eeb01be33951ba214e027da523cf09f8b1Mathias Agopian#include <utils/Errors.h> 159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project#include <sys/types.h> 179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project#include <dirent.h> 189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project#include <ctype.h> 199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project#include <errno.h> 209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectusing namespace android; 229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectstatic const char* kExcludeExtension = ".EXCLUDE"; 249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project/* these formats are already compressed, or don't compress well */ 269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectstatic const char* kNoCompressExt[] = { 279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ".jpg", ".jpeg", ".png", ".gif", 289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ".wav", ".mp2", ".mp3", ".ogg", ".aac", 299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ".mpg", ".mpeg", ".mid", ".midi", ".smf", ".jet", 309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ".rtttl", ".imy", ".xmf", ".mp4", ".m4a", 319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ".m4v", ".3gp", ".3gpp", ".3g2", ".3gpp2", 329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ".amr", ".awb", ".wma", ".wmv" 339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project}; 349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project/* fwd decls, so I can write this downward */ 369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectssize_t processAssets(Bundle* bundle, ZipFile* zip, const sp<AaptAssets>& assets); 377c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Rootssize_t processAssets(Bundle* bundle, ZipFile* zip, const sp<AaptDir>& dir, 387c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root const AaptGroupEntry& ge, const ResourceFilter* filter); 399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectbool processFile(Bundle* bundle, ZipFile* zip, 409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project const sp<AaptGroup>& group, const sp<AaptFile>& file); 419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectbool okayToCompress(Bundle* bundle, const String8& pathName); 429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectssize_t processJarFiles(Bundle* bundle, ZipFile* zip); 439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project/* 459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * The directory hierarchy looks like this: 469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * "outputDir" and "assetRoot" are existing directories. 479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * 489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * On success, "bundle->numPackages" will be the number of Zip packages 499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * we created. 509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectstatus_t writeAPK(Bundle* bundle, const sp<AaptAssets>& assets, 529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project const String8& outputFile) 539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project{ 548a39da80b33691b0c82458c3b7727e13ff71277eJosiah Gaskin #if BENCHMARK 558a39da80b33691b0c82458c3b7727e13ff71277eJosiah Gaskin fprintf(stdout, "BENCHMARK: Starting APK Bundling \n"); 568a39da80b33691b0c82458c3b7727e13ff71277eJosiah Gaskin long startAPKTime = clock(); 578a39da80b33691b0c82458c3b7727e13ff71277eJosiah Gaskin #endif /* BENCHMARK */ 588a39da80b33691b0c82458c3b7727e13ff71277eJosiah Gaskin 599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project status_t result = NO_ERROR; 609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ZipFile* zip = NULL; 619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project int count; 629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project //bundle->setPackageCount(0); 649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /* 669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Prep the Zip archive. 679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * 689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * If the file already exists, fail unless "update" or "force" is set. 699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * If "update" is set, update the contents of the existing archive. 709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Else, if "force" is set, remove the existing archive. 719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project FileType fileType = getFileType(outputFile.string()); 739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (fileType == kFileTypeNonexistent) { 749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // okay, create it below 759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else if (fileType == kFileTypeRegular) { 769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (bundle->getUpdate()) { 779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // okay, open it below 789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else if (bundle->getForce()) { 799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (unlink(outputFile.string()) != 0) { 809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project fprintf(stderr, "ERROR: unable to remove '%s': %s\n", outputFile.string(), 819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project strerror(errno)); 829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project goto bail; 839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else { 859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project fprintf(stderr, "ERROR: '%s' exists (use '-f' to force overwrite)\n", 869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project outputFile.string()); 879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project goto bail; 889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else { 909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project fprintf(stderr, "ERROR: '%s' exists and is not a regular file\n", outputFile.string()); 919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project goto bail; 929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (bundle->getVerbose()) { 959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printf("%s '%s'\n", (fileType == kFileTypeNonexistent) ? "Creating" : "Opening", 969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project outputFile.string()); 979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project status_t status; 1009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project zip = new ZipFile; 1019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project status = zip->open(outputFile.string(), ZipFile::kOpenReadWrite | ZipFile::kOpenCreate); 1029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (status != NO_ERROR) { 1039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project fprintf(stderr, "ERROR: unable to open '%s' as Zip file for writing\n", 1049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project outputFile.string()); 1059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project goto bail; 1069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 1079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (bundle->getVerbose()) { 1099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printf("Writing all files...\n"); 1109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 1119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project count = processAssets(bundle, zip, assets); 1139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (count < 0) { 1149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project fprintf(stderr, "ERROR: unable to process assets while packaging '%s'\n", 1159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project outputFile.string()); 1169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project result = count; 1179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project goto bail; 1189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 1199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (bundle->getVerbose()) { 1219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printf("Generated %d file%s\n", count, (count==1) ? "" : "s"); 1229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 1239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project count = processJarFiles(bundle, zip); 1259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (count < 0) { 1269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project fprintf(stderr, "ERROR: unable to process jar files while packaging '%s'\n", 1279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project outputFile.string()); 1289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project result = count; 1299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project goto bail; 1309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 1319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (bundle->getVerbose()) 1339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printf("Included %d file%s from jar/zip files.\n", count, (count==1) ? "" : "s"); 1349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project result = NO_ERROR; 1369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /* 1389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Check for cruft. We set the "marked" flag on all entries we created 1399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * or decided not to update. If the entry isn't already slated for 1409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * deletion, remove it now. 1419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 1429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project { 1439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (bundle->getVerbose()) 1449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printf("Checking for deleted files\n"); 1459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project int i, removed = 0; 1469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project for (i = 0; i < zip->getNumEntries(); i++) { 1479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ZipEntry* entry = zip->getEntryByIndex(i); 1489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (!entry->getMarked() && entry->getDeleted()) { 1509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (bundle->getVerbose()) { 1519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printf(" (removing crufty '%s')\n", 1529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project entry->getFileName()); 1539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 1549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project zip->remove(entry); 1559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project removed++; 1569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 1579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 1589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (bundle->getVerbose() && removed > 0) 1599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printf("Removed %d file%s\n", removed, (removed==1) ? "" : "s"); 1609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 1619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /* tell Zip lib to process deletions and other pending changes */ 1639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project result = zip->flush(); 1649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (result != NO_ERROR) { 1659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project fprintf(stderr, "ERROR: Zip flush failed, archive may be hosed\n"); 1669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project goto bail; 1679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 1689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /* anything here? */ 1709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (zip->getNumEntries() == 0) { 1719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (bundle->getVerbose()) { 1729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printf("Archive is empty -- removing %s\n", outputFile.getPathLeaf().string()); 1739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 1749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project delete zip; // close the file so we can remove it in Win32 1759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project zip = NULL; 1769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (unlink(outputFile.string()) != 0) { 177dd931864209eac0b4182d7a0d1ca965fcc3b8c03Marco Nelissen fprintf(stderr, "warning: could not unlink '%s'\n", outputFile.string()); 1789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 1799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 1809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 181b711f3f7ee0544685ef26a1a4b4755d5a6233dbaJosiah Gaskin // If we've been asked to generate a dependency file for the .ap_ package, 182b711f3f7ee0544685ef26a1a4b4755d5a6233dbaJosiah Gaskin // do so here 18303589cc65355220e0a4a0c816189a9fa25cc81fcJosiah Gaskin if (bundle->getGenDependencies()) { 184b711f3f7ee0544685ef26a1a4b4755d5a6233dbaJosiah Gaskin // The dependency file gets output to the same directory 185b711f3f7ee0544685ef26a1a4b4755d5a6233dbaJosiah Gaskin // as the specified output file with an additional .d extension. 186b711f3f7ee0544685ef26a1a4b4755d5a6233dbaJosiah Gaskin // e.g. bin/resources.ap_.d 187b711f3f7ee0544685ef26a1a4b4755d5a6233dbaJosiah Gaskin String8 dependencyFile = outputFile; 18803589cc65355220e0a4a0c816189a9fa25cc81fcJosiah Gaskin dependencyFile.append(".d"); 18903589cc65355220e0a4a0c816189a9fa25cc81fcJosiah Gaskin 19003589cc65355220e0a4a0c816189a9fa25cc81fcJosiah Gaskin FILE* fp = fopen(dependencyFile.string(), "a"); 191b711f3f7ee0544685ef26a1a4b4755d5a6233dbaJosiah Gaskin // Add this file to the dependency file 19203589cc65355220e0a4a0c816189a9fa25cc81fcJosiah Gaskin fprintf(fp, "%s \\\n", outputFile.string()); 19303589cc65355220e0a4a0c816189a9fa25cc81fcJosiah Gaskin fclose(fp); 19403589cc65355220e0a4a0c816189a9fa25cc81fcJosiah Gaskin } 19503589cc65355220e0a4a0c816189a9fa25cc81fcJosiah Gaskin 1969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project assert(result == NO_ERROR); 1979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectbail: 1999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project delete zip; // must close before remove in Win32 2009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (result != NO_ERROR) { 2019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (bundle->getVerbose()) { 2029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printf("Removing %s due to earlier failures\n", outputFile.string()); 2039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 2049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (unlink(outputFile.string()) != 0) { 205dd931864209eac0b4182d7a0d1ca965fcc3b8c03Marco Nelissen fprintf(stderr, "warning: could not unlink '%s'\n", outputFile.string()); 2069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 2079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 2089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (result == NO_ERROR && bundle->getVerbose()) 2109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printf("Done!\n"); 2118a39da80b33691b0c82458c3b7727e13ff71277eJosiah Gaskin 2128a39da80b33691b0c82458c3b7727e13ff71277eJosiah Gaskin #if BENCHMARK 2138a39da80b33691b0c82458c3b7727e13ff71277eJosiah Gaskin fprintf(stdout, "BENCHMARK: End APK Bundling. Time Elapsed: %f ms \n",(clock() - startAPKTime)/1000.0); 2148a39da80b33691b0c82458c3b7727e13ff71277eJosiah Gaskin #endif /* BENCHMARK */ 2159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return result; 2169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project} 2179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectssize_t processAssets(Bundle* bundle, ZipFile* zip, 2199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project const sp<AaptAssets>& assets) 2209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project{ 2219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ResourceFilter filter; 2229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project status_t status = filter.parse(bundle->getConfigurations()); 2239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (status != NO_ERROR) { 2249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return -1; 2259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 2269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ssize_t count = 0; 2289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project const size_t N = assets->getGroupEntries().size(); 2309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project for (size_t i=0; i<N; i++) { 2319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project const AaptGroupEntry& ge = assets->getGroupEntries()[i]; 2327c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root 2337c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root ssize_t res = processAssets(bundle, zip, assets, ge, &filter); 2349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (res < 0) { 2359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return res; 2369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 2377c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root 2389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project count += res; 2399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 2409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return count; 2429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project} 2439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2447c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Rootssize_t processAssets(Bundle* bundle, ZipFile* zip, const sp<AaptDir>& dir, 2457c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root const AaptGroupEntry& ge, const ResourceFilter* filter) 2469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project{ 2479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ssize_t count = 0; 2489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project const size_t ND = dir->getDirs().size(); 2509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project size_t i; 2519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project for (i=0; i<ND; i++) { 2527c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root const sp<AaptDir>& subDir = dir->getDirs().valueAt(i); 2537c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root 2547c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root const bool filterable = filter != NULL && subDir->getLeaf().find("mipmap-") != 0; 2557c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root 2567c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root if (filterable && subDir->getLeaf() != subDir->getPath() && !filter->match(ge.toParams())) { 2577c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root continue; 2587c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root } 2597c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root 2607c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root ssize_t res = processAssets(bundle, zip, subDir, ge, filterable ? filter : NULL); 2619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (res < 0) { 2629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return res; 2639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 2649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project count += res; 2659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 2669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2677c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root if (filter != NULL && !filter->match(ge.toParams())) { 2687c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root return count; 2697c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root } 2707c71023b7f5aafac09bd2f0425fccb5fe2d22b27Kenny Root 2719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project const size_t NF = dir->getFiles().size(); 2729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project for (i=0; i<NF; i++) { 2739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project sp<AaptGroup> gp = dir->getFiles().valueAt(i); 2749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ssize_t fi = gp->getFiles().indexOfKey(ge); 2759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (fi >= 0) { 2769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project sp<AaptFile> fl = gp->getFiles().valueAt(fi); 2779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (!processFile(bundle, zip, gp, fl)) { 2789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return UNKNOWN_ERROR; 2799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 2809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project count++; 2819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 2829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 2839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return count; 2859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project} 2869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project/* 2889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Process a regular file, adding it to the archive if appropriate. 2899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * 2909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * If we're in "update" mode, and the file already exists in the archive, 2919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * delete the existing entry before adding the new one. 2929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 2939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectbool processFile(Bundle* bundle, ZipFile* zip, 2949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project const sp<AaptGroup>& group, const sp<AaptFile>& file) 2959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project{ 2969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project const bool hasData = file->hasData(); 2979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project String8 storageName(group->getPath()); 2999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project storageName.convertToResPath(); 3009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ZipEntry* entry; 3019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project bool fromGzip = false; 3029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project status_t result; 3039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 3049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /* 3059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * See if the filename ends in ".EXCLUDE". We can't use 3069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * String8::getPathExtension() because the length of what it considers 3079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * to be an extension is capped. 3089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * 3099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * The Asset Manager doesn't check for ".EXCLUDE" in Zip archives, 3109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * so there's no value in adding them (and it makes life easier on 3119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * the AssetManager lib if we don't). 3129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * 3139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * NOTE: this restriction has been removed. If you're in this code, you 3149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * should clean this up, but I'm in here getting rid of Path Name, and I 3159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * don't want to make other potentially breaking changes --joeo 3169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 3179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project int fileNameLen = storageName.length(); 3189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project int excludeExtensionLen = strlen(kExcludeExtension); 3199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (fileNameLen > excludeExtensionLen 3209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project && (0 == strcmp(storageName.string() + (fileNameLen - excludeExtensionLen), 3219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project kExcludeExtension))) { 322dd931864209eac0b4182d7a0d1ca965fcc3b8c03Marco Nelissen fprintf(stderr, "warning: '%s' not added to Zip\n", storageName.string()); 3239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return true; 3249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 3269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (strcasecmp(storageName.getPathExtension().string(), ".gz") == 0) { 3279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project fromGzip = true; 3289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project storageName = storageName.getBasePath(); 3299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 3319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (bundle->getUpdate()) { 3329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project entry = zip->getEntryByName(storageName.string()); 3339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (entry != NULL) { 3349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /* file already exists in archive; there can be only one */ 3359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (entry->getMarked()) { 3369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project fprintf(stderr, 3379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project "ERROR: '%s' exists twice (check for with & w/o '.gz'?)\n", 3389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project file->getPrintableSource().string()); 3399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return false; 3409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (!hasData) { 3429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project const String8& srcName = file->getSourceFile(); 3439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project time_t fileModWhen; 3449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project fileModWhen = getFileModDate(srcName.string()); 3459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (fileModWhen == (time_t) -1) { // file existence tested earlier, 3469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return false; // not expecting an error here 3479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 3499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (fileModWhen > entry->getModWhen()) { 3509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // mark as deleted so add() will succeed 3519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (bundle->getVerbose()) { 3529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printf(" (removing old '%s')\n", storageName.string()); 3539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 3559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project zip->remove(entry); 3569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else { 3579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // version in archive is newer 3589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (bundle->getVerbose()) { 3599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printf(" (not updating '%s')\n", storageName.string()); 3609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project entry->setMarked(true); 3629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return true; 3639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else { 3659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // Generated files are always replaced. 3669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project zip->remove(entry); 3679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 3719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project //android_setMinPriority(NULL, ANDROID_LOG_VERBOSE); 3729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 3739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (fromGzip) { 3749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project result = zip->addGzip(file->getSourceFile().string(), storageName.string(), &entry); 3759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else if (!hasData) { 3769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /* don't compress certain files, e.g. PNGs */ 3779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project int compressionMethod = bundle->getCompressionMethod(); 3789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (!okayToCompress(bundle, storageName)) { 3799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project compressionMethod = ZipEntry::kCompressStored; 3809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project result = zip->add(file->getSourceFile().string(), storageName.string(), compressionMethod, 3829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project &entry); 3839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else { 3849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project result = zip->add(file->getData(), file->getSize(), storageName.string(), 3859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project file->getCompressionMethod(), &entry); 3869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (result == NO_ERROR) { 3889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (bundle->getVerbose()) { 3899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printf(" '%s'%s", storageName.string(), fromGzip ? " (from .gz)" : ""); 3909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (entry->getCompressionMethod() == ZipEntry::kCompressStored) { 3919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printf(" (not compressed)\n"); 3929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else { 3939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printf(" (compressed %d%%)\n", calcPercent(entry->getUncompressedLen(), 3949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project entry->getCompressedLen())); 3959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project entry->setMarked(true); 3989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else { 3999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (result == ALREADY_EXISTS) { 4009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project fprintf(stderr, " Unable to add '%s': file already in archive (try '-u'?)\n", 4019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project file->getPrintableSource().string()); 4029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else { 4039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project fprintf(stderr, " Unable to add '%s': Zip add failed\n", 4049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project file->getPrintableSource().string()); 4059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return false; 4079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 4099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return true; 4109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project} 4119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 4129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project/* 4139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Determine whether or not we want to try to compress this file based 4149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * on the file extension. 4159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 4169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectbool okayToCompress(Bundle* bundle, const String8& pathName) 4179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project{ 4189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project String8 ext = pathName.getPathExtension(); 4199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project int i; 4209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 4219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (ext.length() == 0) 4229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return true; 4239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 4249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project for (i = 0; i < NELEM(kNoCompressExt); i++) { 4259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (strcasecmp(ext.string(), kNoCompressExt[i]) == 0) 4269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return false; 4279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 4299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project const android::Vector<const char*>& others(bundle->getNoCompressExtensions()); 4309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project for (i = 0; i < (int)others.size(); i++) { 4319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project const char* str = others[i]; 4329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project int pos = pathName.length() - strlen(str); 4339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (pos < 0) { 4349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project continue; 4359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project const char* path = pathName.string(); 4379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (strcasecmp(path + pos, str) == 0) { 4389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return false; 4399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 4429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return true; 4439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project} 4449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 4459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectbool endsWith(const char* haystack, const char* needle) 4469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project{ 4479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project size_t a = strlen(haystack); 4489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project size_t b = strlen(needle); 4499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (a < b) return false; 4509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return strcasecmp(haystack+(a-b), needle) == 0; 4519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project} 4529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 4539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectssize_t processJarFile(ZipFile* jar, ZipFile* out) 4549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project{ 4559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project status_t err; 4569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project size_t N = jar->getNumEntries(); 4579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project size_t count = 0; 4589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project for (size_t i=0; i<N; i++) { 4599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ZipEntry* entry = jar->getEntryByIndex(i); 4609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project const char* storageName = entry->getFileName(); 4619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (endsWith(storageName, ".class")) { 4629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project int compressionMethod = entry->getCompressionMethod(); 4639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project size_t size = entry->getUncompressedLen(); 4649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project const void* data = jar->uncompress(entry); 4659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (data == NULL) { 4669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project fprintf(stderr, "ERROR: unable to uncompress entry '%s'\n", 4679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project storageName); 4689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return -1; 4699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project out->add(data, size, storageName, compressionMethod, NULL); 4719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project free((void*)data); 4729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project count++; 4749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return count; 4769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project} 4779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 4789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectssize_t processJarFiles(Bundle* bundle, ZipFile* zip) 4799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project{ 480f1ff21ac62a51f5ba8ca0821ea8a90f70957e25dSteve Block status_t err; 4819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ssize_t count = 0; 4829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project const android::Vector<const char*>& jars = bundle->getJarFiles(); 4839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 4849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project size_t N = jars.size(); 4859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project for (size_t i=0; i<N; i++) { 4869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ZipFile jar; 4879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project err = jar.open(jars[i], ZipFile::kOpenReadOnly); 4889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (err != 0) { 489ddb76c4644756b31be948d70aaa8ee541dd94999Kenny Root fprintf(stderr, "ERROR: unable to open '%s' as a zip file: %d\n", 4909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project jars[i], err); 4919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return err; 4929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project err += processJarFile(&jar, zip); 4949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (err < 0) { 4959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project fprintf(stderr, "ERROR: unable to process '%s'\n", jars[i]); 4969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return err; 4979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project count += err; 4999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 5009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 5019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return count; 5029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project} 503