Command.cpp revision 7ef189d0a90f3c9a409512ab7dce9304639df481
15d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)//
29ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch// Copyright 2006 The Android Open Source Project
39ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch//
49ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch// Android Asset Packaging Tool main entry point.
59ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch//
69ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "Main.h"
79ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "Bundle.h"
89ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "ResourceTable.h"
99ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "XMLNode.h"
109ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
119ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include <utils.h>
129ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include <utils/ZipFile.h>
139ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
149ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include <fcntl.h>
159ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include <errno.h>
1658537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)
179ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdochusing namespace android;
18a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)
199ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch/*
209ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch * Show version info.  All the cool kids do it.
219ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch */
229ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdochint doVersion(Bundle* bundle)
239ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch{
24f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)    if (bundle->getFileSpecCount() != 0)
259ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        printf("(ignoring extra arguments)\n");
269ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    printf("Android Asset Packaging Tool, v0.2\n");
279ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
289ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    return 0;
29a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)}
30a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)
315d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)
325d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)/*
335d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) * Open the file read only.  The call fails if the file doesn't exist.
345d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) *
359ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch * Returns NULL on failure.
369ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch */
379ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben MurdochZipFile* openReadOnly(const char* fileName)
389ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch{
3958537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)    ZipFile* zip;
4058537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)    status_t result;
4158537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)
429ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    zip = new ZipFile;
439ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    result = zip->open(fileName, ZipFile::kOpenReadOnly);
449ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    if (result != NO_ERROR) {
459ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        if (result == NAME_NOT_FOUND)
469ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            fprintf(stderr, "ERROR: '%s' not found\n", fileName);
479ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        else if (result == PERMISSION_DENIED)
489ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            fprintf(stderr, "ERROR: '%s' access denied\n", fileName);
495d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)        else
505d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)            fprintf(stderr, "ERROR: failed opening '%s' as Zip file\n",
515d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)                fileName);
525d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)        delete zip;
539ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        return NULL;
549ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    }
559ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
569ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    return zip;
575d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)}
585d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)
599ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch/*
609ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch * Open the file read-write.  The file will be created if it doesn't
615d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) * already exist and "okayToCreate" is set.
629ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch *
639ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch * Returns NULL on failure.
649ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch */
659ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben MurdochZipFile* openReadWrite(const char* fileName, bool okayToCreate)
669ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch{
679ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    ZipFile* zip = NULL;
689ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    status_t result;
699ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    int flags;
709ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
719ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    flags = ZipFile::kOpenReadWrite;
729ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    if (okayToCreate)
735d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)        flags |= ZipFile::kOpenCreate;
749ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
759ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    zip = new ZipFile;
769ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    result = zip->open(fileName, flags);
779ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    if (result != NO_ERROR) {
789ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        delete zip;
799ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        zip = NULL;
809ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        goto bail;
819ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    }
829ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
839ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdochbail:
849ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    return zip;
859ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch}
869ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
879ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
889ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch/*
899ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch * Return a short string describing the compression method.
909ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch */
9146d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles)const char* compressionName(int method)
9246d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles){
9346d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles)    if (method == ZipEntry::kCompressStored)
9458537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)        return "Stored";
959ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    else if (method == ZipEntry::kCompressDeflated)
9646d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles)        return "Deflated";
9746d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles)    else
9846d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles)        return "Unknown";
9946d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles)}
1005d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)
1019ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch/*
1029ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch * Return the percent reduction in size (0% == no compression).
1039ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch */
1049ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdochint calcPercent(long uncompressedLen, long compressedLen)
1059ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch{
1069ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    if (!uncompressedLen)
1079ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        return 0;
1089ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    else
1095d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)        return (int) (100.0 - (compressedLen * 100.0) / uncompressedLen + 0.5);
1109ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch}
1119ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
1129ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch/*
1139ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch * Handle the "list" command, which can be a simple file dump or
1149ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch * a verbose listing.
1159ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch *
1169ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch * The verbose listing closely matches the output of the Info-ZIP "unzip"
1179ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch * command.
1189ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch */
119c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdochint doList(Bundle* bundle)
120c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch{
121c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch    int result = 1;
122c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch    ZipFile* zip = NULL;
1239ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    const ZipEntry* entry;
1249ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    long totalUncLen, totalCompLen;
1259ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    const char* zipFileName;
1269ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
1279ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    if (bundle->getFileSpecCount() != 1) {
1289ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        fprintf(stderr, "ERROR: specify zip file name (only)\n");
1299ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        goto bail;
1309ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    }
1316d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)    zipFileName = bundle->getFileSpecEntry(0);
1326d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)
1336d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)    zip = openReadOnly(zipFileName);
1346d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)    if (zip == NULL)
1356d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)        goto bail;
1366d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)
1376d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)    int count, i;
1386d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)
1396d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)    if (bundle->getVerbose()) {
1409ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        printf("Archive:  %s\n", zipFileName);
1419ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        printf(
1429ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            " Length   Method    Size  Ratio   Date   Time   CRC-32    Name\n");
1439ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        printf(
1449ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            "--------  ------  ------- -----   ----   ----   ------    ----\n");
1459ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    }
1469ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
1479ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    totalUncLen = totalCompLen = 0;
1485d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)
1499ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    count = zip->getNumEntries();
1509ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    for (i = 0; i < count; i++) {
1519ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        entry = zip->getEntryByIndex(i);
1529ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        if (bundle->getVerbose()) {
1539ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            char dateBuf[32];
1549ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            time_t when;
1559ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
1569ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            when = entry->getModWhen();
1579ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            strftime(dateBuf, sizeof(dateBuf), "%m-%d-%y %H:%M",
1589ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch                localtime(&when));
1599ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
1609ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            printf("%8ld  %-7.7s %7ld %3d%%  %s  %08lx  %s\n",
1619ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch                (long) entry->getUncompressedLen(),
1629ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch                compressionName(entry->getCompressionMethod()),
1639ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch                (long) entry->getCompressedLen(),
1649ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch                calcPercent(entry->getUncompressedLen(),
1659ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch                            entry->getCompressedLen()),
1669ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch                dateBuf,
1679ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch                entry->getCRC32(),
1689ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch                entry->getFileName());
1699ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        } else {
1709ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            printf("%s\n", entry->getFileName());
1719ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        }
1729ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
1739ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        totalUncLen += entry->getUncompressedLen();
1749ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        totalCompLen += entry->getCompressedLen();
1759ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    }
1769ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
1779ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    if (bundle->getVerbose()) {
178f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)        printf(
1799ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        "--------          -------  ---                            -------\n");
1805d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)        printf("%8ld          %7ld  %2d%%                            %d files\n",
1815d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)            totalUncLen,
1825d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)            totalCompLen,
1835d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)            calcPercent(totalUncLen, totalCompLen),
1845d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)            zip->getNumEntries());
1855d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)    }
1865d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)
1875d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)    if (bundle->getAndroidList()) {
1885d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)        AssetManager assets;
1895d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)        if (!assets.addAssetPath(String8(zipFileName), NULL)) {
1909ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            fprintf(stderr, "ERROR: list -a failed because assets could not be loaded\n");
1919ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            goto bail;
1929ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        }
1939ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
1949ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        const ResTable& res = assets.getResources(false);
1959ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        if (&res == NULL) {
1969ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            printf("\nNo resource table found.\n");
1979ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        } else {
1989ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            printf("\nResource table:\n");
1999ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            res.print();
2009ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        }
2019ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
2029ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        Asset* manifestAsset = assets.openNonAsset("AndroidManifest.xml",
2039ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch                                                   Asset::ACCESS_BUFFER);
2049ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        if (manifestAsset == NULL) {
2059ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            printf("\nNo AndroidManifest.xml found.\n");
206d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)        } else {
207d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)            printf("\nAndroid manifest:\n");
208d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)            ResXMLTree tree;
2099ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            tree.setTo(manifestAsset->getBuffer(true),
2109ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch                       manifestAsset->getLength());
2119ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            printXMLBlock(&tree);
2129ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        }
2139ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        delete manifestAsset;
2149ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    }
2159ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
2169ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    result = 0;
2179ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
2189ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdochbail:
2199ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    delete zip;
2209ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    return result;
221a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)}
222a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)
223a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)static ssize_t indexOfAttribute(const ResXMLTree& tree, uint32_t attrRes)
224a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles){
225a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)    size_t N = tree.getAttributeCount();
226a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)    for (size_t i=0; i<N; i++) {
227a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)        if (tree.getAttributeNameResID(i) == attrRes) {
2289ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            return (ssize_t)i;
2299ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        }
2309ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    }
2319ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    return -1;
2329ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch}
2339ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
2349ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdochstatic String8 getAttribute(const ResXMLTree& tree, const char* ns,
2359ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch                            const char* attr, String8* outError)
2369ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch{
2379ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    ssize_t idx = tree.indexOfAttribute(ns, attr);
2389ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    if (idx < 0) {
2399ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        return String8();
2409ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    }
2419ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    Res_value value;
2429ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    if (tree.getAttributeValue(idx, &value) != NO_ERROR) {
2439ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        if (value.dataType != Res_value::TYPE_STRING) {
2449ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            if (outError != NULL) *outError = "attribute is not a string value";
2459ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch            return String8();
2469ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        }
2479ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    }
2489ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    size_t len;
2499ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    const uint16_t* str = tree.getAttributeStringValue(idx, &len);
2509ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    return str ? String8(str, len) : String8();
2519ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch}
2529ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
2539ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdochstatic String8 getAttribute(const ResXMLTree& tree, uint32_t attrRes, String8* outError)
2549ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch{
2559ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    ssize_t idx = indexOfAttribute(tree, attrRes);
2569ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    if (idx < 0) {
2579ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        return String8();
2589ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    }
2599ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    Res_value value;
2609ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    if (tree.getAttributeValue(idx, &value) != NO_ERROR) {
2619ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        if (value.dataType != Res_value::TYPE_STRING) {
2626d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)            if (outError != NULL) *outError = "attribute is not a string value";
2636d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)            return String8();
2646d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)        }
2656d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)    }
2666d86b77056ed63eb6871182f42a9fd5f07550f90Torne (Richard Coles)    size_t len;
2679ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    const uint16_t* str = tree.getAttributeStringValue(idx, &len);
2689ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    return str ? String8(str, len) : String8();
269c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch}
270c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch
271c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdochstatic int32_t getIntegerAttribute(const ResXMLTree& tree, uint32_t attrRes, String8* outError)
272c5cede9ae108bb15f6b7a8aea21c7e1fefa2834cBen Murdoch{
2739ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    ssize_t idx = indexOfAttribute(tree, attrRes);
2749ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch    if (idx < 0) {
2759ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch        return -1;
276    }
277    Res_value value;
278    if (tree.getAttributeValue(idx, &value) != NO_ERROR) {
279        if (value.dataType != Res_value::TYPE_INT_DEC) {
280            if (outError != NULL) *outError = "attribute is not an integer value";
281            return -1;
282        }
283    }
284    return value.data;
285}
286
287static String8 getResolvedAttribute(const ResTable* resTable, const ResXMLTree& tree,
288        uint32_t attrRes, String8* outError)
289{
290    ssize_t idx = indexOfAttribute(tree, attrRes);
291    if (idx < 0) {
292        return String8();
293    }
294    Res_value value;
295    if (tree.getAttributeValue(idx, &value) != NO_ERROR) {
296        if (value.dataType == Res_value::TYPE_STRING) {
297            size_t len;
298            const uint16_t* str = tree.getAttributeStringValue(idx, &len);
299            return str ? String8(str, len) : String8();
300        }
301        resTable->resolveReference(&value, 0);
302        if (value.dataType != Res_value::TYPE_STRING) {
303            if (outError != NULL) *outError = "attribute is not a string value";
304            return String8();
305        }
306    }
307    size_t len;
308    const Res_value* value2 = &value;
309    const char16_t* str = const_cast<ResTable*>(resTable)->valueToString(value2, 0, NULL, &len);
310    return str ? String8(str, len) : String8();
311}
312
313// These are attribute resource constants for the platform, as found
314// in android.R.attr
315enum {
316    NAME_ATTR = 0x01010003,
317    VERSION_CODE_ATTR = 0x0101021b,
318    VERSION_NAME_ATTR = 0x0101021c,
319    LABEL_ATTR = 0x01010001,
320    ICON_ATTR = 0x01010002,
321    MIN_SDK_VERSION_ATTR = 0x0101020c
322};
323
324const char *getComponentName(String8 &pkgName, String8 &componentName) {
325    ssize_t idx = componentName.find(".");
326    String8 retStr(pkgName);
327    if (idx == 0) {
328        retStr += componentName;
329    } else if (idx < 0) {
330        retStr += ".";
331        retStr += componentName;
332    } else {
333        return componentName.string();
334    }
335    return retStr.string();
336}
337
338/*
339 * Handle the "dump" command, to extract select data from an archive.
340 */
341int doDump(Bundle* bundle)
342{
343    status_t result = UNKNOWN_ERROR;
344    Asset* asset = NULL;
345
346    if (bundle->getFileSpecCount() < 1) {
347        fprintf(stderr, "ERROR: no dump option specified\n");
348        return 1;
349    }
350
351    if (bundle->getFileSpecCount() < 2) {
352        fprintf(stderr, "ERROR: no dump file specified\n");
353        return 1;
354    }
355
356    const char* option = bundle->getFileSpecEntry(0);
357    const char* filename = bundle->getFileSpecEntry(1);
358
359    AssetManager assets;
360    if (!assets.addAssetPath(String8(filename), NULL)) {
361        fprintf(stderr, "ERROR: dump failed because assets could not be loaded\n");
362        return 1;
363    }
364
365    const ResTable& res = assets.getResources(false);
366    if (&res == NULL) {
367        fprintf(stderr, "ERROR: dump failed because no resource table was found\n");
368        goto bail;
369    }
370
371    if (strcmp("resources", option) == 0) {
372        res.print();
373
374    } else if (strcmp("xmltree", option) == 0) {
375        if (bundle->getFileSpecCount() < 3) {
376            fprintf(stderr, "ERROR: no dump xmltree resource file specified\n");
377            goto bail;
378        }
379
380        for (int i=2; i<bundle->getFileSpecCount(); i++) {
381            const char* resname = bundle->getFileSpecEntry(i);
382            ResXMLTree tree;
383            asset = assets.openNonAsset(resname, Asset::ACCESS_BUFFER);
384            if (asset == NULL) {
385                fprintf(stderr, "ERROR: dump failed because resource %p found\n", resname);
386                goto bail;
387            }
388
389            if (tree.setTo(asset->getBuffer(true),
390                           asset->getLength()) != NO_ERROR) {
391                fprintf(stderr, "ERROR: Resource %s is corrupt\n", resname);
392                goto bail;
393            }
394            tree.restart();
395            printXMLBlock(&tree);
396            delete asset;
397            asset = NULL;
398        }
399
400    } else if (strcmp("xmlstrings", option) == 0) {
401        if (bundle->getFileSpecCount() < 3) {
402            fprintf(stderr, "ERROR: no dump xmltree resource file specified\n");
403            goto bail;
404        }
405
406        for (int i=2; i<bundle->getFileSpecCount(); i++) {
407            const char* resname = bundle->getFileSpecEntry(i);
408            ResXMLTree tree;
409            asset = assets.openNonAsset(resname, Asset::ACCESS_BUFFER);
410            if (asset == NULL) {
411                fprintf(stderr, "ERROR: dump failed because resource %p found\n", resname);
412                goto bail;
413            }
414
415            if (tree.setTo(asset->getBuffer(true),
416                           asset->getLength()) != NO_ERROR) {
417                fprintf(stderr, "ERROR: Resource %s is corrupt\n", resname);
418                goto bail;
419            }
420            printStringPool(&tree.getStrings());
421            delete asset;
422            asset = NULL;
423        }
424
425    } else {
426        ResXMLTree tree;
427        asset = assets.openNonAsset("AndroidManifest.xml",
428                                            Asset::ACCESS_BUFFER);
429        if (asset == NULL) {
430            fprintf(stderr, "ERROR: dump failed because no AndroidManifest.xml found\n");
431            goto bail;
432        }
433
434        if (tree.setTo(asset->getBuffer(true),
435                       asset->getLength()) != NO_ERROR) {
436            fprintf(stderr, "ERROR: AndroidManifest.xml is corrupt\n");
437            goto bail;
438        }
439        tree.restart();
440
441        if (strcmp("permissions", option) == 0) {
442            size_t len;
443            ResXMLTree::event_code_t code;
444            int depth = 0;
445            while ((code=tree.next()) != ResXMLTree::END_DOCUMENT && code != ResXMLTree::BAD_DOCUMENT) {
446                if (code == ResXMLTree::END_TAG) {
447                    depth--;
448                    continue;
449                }
450                if (code != ResXMLTree::START_TAG) {
451                    continue;
452                }
453                depth++;
454                String8 tag(tree.getElementName(&len));
455                //printf("Depth %d tag %s\n", depth, tag.string());
456                if (depth == 1) {
457                    if (tag != "manifest") {
458                        fprintf(stderr, "ERROR: manifest does not start with <manifest> tag\n");
459                        goto bail;
460                    }
461                    String8 pkg = getAttribute(tree, NULL, "package", NULL);
462                    printf("package: %s\n", pkg.string());
463                } else if (depth == 2 && tag == "permission") {
464                    String8 error;
465                    String8 name = getAttribute(tree, NAME_ATTR, &error);
466                    if (error != "") {
467                        fprintf(stderr, "ERROR: %s\n", error.string());
468                        goto bail;
469                    }
470                    printf("permission: %s\n", name.string());
471                } else if (depth == 2 && tag == "uses-permission") {
472                    String8 error;
473                    String8 name = getAttribute(tree, NAME_ATTR, &error);
474                    if (error != "") {
475                        fprintf(stderr, "ERROR: %s\n", error.string());
476                        goto bail;
477                    }
478                    printf("uses-permission: %s\n", name.string());
479                }
480            }
481        } else if (strcmp("badging", option) == 0) {
482            size_t len;
483            ResXMLTree::event_code_t code;
484            int depth = 0;
485            String8 error;
486            bool withinActivity = false;
487            bool isMainActivity = false;
488            bool isLauncherActivity = false;
489            bool withinApplication = false;
490            bool withinReceiver = false;
491            String8 pkg;
492            String8 activityName;
493            String8 activityLabel;
494            String8 activityIcon;
495            String8 receiverName;
496            while ((code=tree.next()) != ResXMLTree::END_DOCUMENT && code != ResXMLTree::BAD_DOCUMENT) {
497                if (code == ResXMLTree::END_TAG) {
498                    depth--;
499                    continue;
500                }
501                if (code != ResXMLTree::START_TAG) {
502                    continue;
503                }
504                depth++;
505                String8 tag(tree.getElementName(&len));
506                //printf("Depth %d tag %s\n", depth, tag.string());
507                if (depth == 1) {
508                    if (tag != "manifest") {
509                        fprintf(stderr, "ERROR: manifest does not start with <manifest> tag\n");
510                        goto bail;
511                    }
512                    pkg = getAttribute(tree, NULL, "package", NULL);
513                    printf("package: name='%s' ", pkg.string());
514                    int32_t versionCode = getIntegerAttribute(tree, VERSION_CODE_ATTR, &error);
515                    if (error != "") {
516                        fprintf(stderr, "ERROR getting 'android:versionCode' attribute: %s\n", error.string());
517                        goto bail;
518                    }
519                    if (versionCode > 0) {
520                        printf("versionCode='%d' ", versionCode);
521                    } else {
522                        printf("versionCode='' ");
523                    }
524                    String8 versionName = getAttribute(tree, VERSION_NAME_ATTR, &error);
525                    if (error != "") {
526                        fprintf(stderr, "ERROR getting 'android:versionName' attribute: %s\n", error.string());
527                        goto bail;
528                    }
529                    printf("versionName='%s'\n", versionName.string());
530                } else if (depth == 2) {
531                    withinApplication = false;
532                    if (tag == "application") {
533                        withinApplication = true;
534                        String8 label = getResolvedAttribute(&res, tree, LABEL_ATTR, &error);
535                         if (error != "") {
536                             fprintf(stderr, "ERROR getting 'android:label' attribute: %s\n", error.string());
537                             goto bail;
538                        }
539                        printf("application: label='%s' ", label.string());
540                        String8 icon = getResolvedAttribute(&res, tree, ICON_ATTR, &error);
541                        if (error != "") {
542                            fprintf(stderr, "ERROR getting 'android:icon' attribute: %s\n", error.string());
543                            goto bail;
544                        }
545                        printf("icon='%s'\n", icon.string());
546                    } else if (tag == "uses-sdk") {
547                        int32_t sdkVersion = getIntegerAttribute(tree, MIN_SDK_VERSION_ATTR, &error);
548                        if (error != "") {
549                            fprintf(stderr, "ERROR getting 'android:minSdkVersion' attribute: %s\n", error.string());
550                            goto bail;
551                        }
552                        if (sdkVersion != -1) {
553                            printf("sdkVersion:'%d'\n", sdkVersion);
554                        }
555                    }
556                } else if (depth == 3 && withinApplication) {
557                    withinActivity = false;
558                    withinReceiver = false;
559                    if(tag == "activity") {
560                        withinActivity = true;
561                        activityName = getAttribute(tree, NAME_ATTR, &error);
562                        if (error != "") {
563                            fprintf(stderr, "ERROR getting 'android:name' attribute: %s\n", error.string());
564                            goto bail;
565                        }
566
567                        activityLabel = getResolvedAttribute(&res, tree, LABEL_ATTR, &error);
568                        if (error != "") {
569                            fprintf(stderr, "ERROR getting 'android:label' attribute: %s\n", error.string());
570                            goto bail;
571                        }
572
573                        activityIcon = getResolvedAttribute(&res, tree, ICON_ATTR, &error);
574                        if (error != "") {
575                            fprintf(stderr, "ERROR getting 'android:icon' attribute: %s\n", error.string());
576                            goto bail;
577                        }
578                    } else if (tag == "uses-library") {
579                        String8 libraryName = getAttribute(tree, NAME_ATTR, &error);
580                        if (error != "") {
581                            fprintf(stderr, "ERROR getting 'android:name' attribute for uses-library: %s\n", error.string());
582                            goto bail;
583                        }
584                        printf("uses-library:'%s'\n", libraryName.string());
585                    } else if (tag == "receiver") {
586                        withinReceiver = true;
587                        receiverName = getAttribute(tree, NAME_ATTR, &error);
588
589                        if (error != "") {
590                            fprintf(stderr, "ERROR getting 'android:name' attribute for receiver: %s\n", error.string());
591                            goto bail;
592                        }
593                    }
594                } else if (depth == 5) {
595                        if (withinActivity) {
596                            if (tag == "action") {
597                                //printf("LOG: action tag\n");
598                                String8 action = getAttribute(tree, NAME_ATTR, &error);
599                                if (error != "") {
600                                    fprintf(stderr, "ERROR getting 'android:name' attribute: %s\n", error.string());
601                                    goto bail;
602                                }
603                                if (action == "android.intent.action.MAIN") {
604                                    isMainActivity = true;
605                                    //printf("LOG: isMainActivity==true\n");
606                                }
607                        } else if (tag == "category") {
608                            String8 category = getAttribute(tree, NAME_ATTR, &error);
609                            if (error != "") {
610                                fprintf(stderr, "ERROR getting 'name' attribute: %s\n", error.string());
611                                goto bail;
612                            }
613                            if (category == "android.intent.category.LAUNCHER") {
614                                isLauncherActivity = true;
615                                //printf("LOG: isLauncherActivity==true\n");
616                            }
617                        }
618                    } else if (withinReceiver) {
619                        if (tag == "action") {
620                            String8 action = getAttribute(tree, NAME_ATTR, &error);
621                            if (error != "") {
622                                fprintf(stderr, "ERROR getting 'android:name' attribute for receiver: %s\n", error.string());
623                                goto bail;
624                            }
625                            if (action == "android.appwidget.action.APPWIDGET_UPDATE") {
626                                const char *rName = getComponentName(pkg, receiverName);
627                                if (rName != NULL) {
628                                    printf("gadget-receiver:'%s/%s'\n", pkg.string(), rName);
629                                }
630                            }
631                        }
632                    }
633                }
634
635                if (depth < 2) {
636                    withinApplication = false;
637                }
638                if (depth < 3) {
639                    //if (withinActivity) printf("LOG: withinActivity==false\n");
640                    withinActivity = false;
641                    withinReceiver = false;
642                }
643
644                if (depth < 5) {
645                    //if (isMainActivity) printf("LOG: isMainActivity==false\n");
646                    //if (isLauncherActivity) printf("LOG: isLauncherActivity==false\n");
647                    isMainActivity = false;
648                    isLauncherActivity = false;
649                }
650
651                if (withinActivity && isMainActivity && isLauncherActivity) {
652                    printf("launchable activity:");
653                    const char *aName = getComponentName(pkg, activityName);
654                    if (aName != NULL) {
655                        printf(" name='%s'", aName);
656                    }
657                    printf("label='%s' icon='%s'\n",
658                           activityLabel.string(),
659                           activityIcon.string());
660                }
661            }
662            printf("locales:");
663            Vector<String8> locales;
664            res.getLocales(&locales);
665            const size_t N = locales.size();
666            for (size_t i=0; i<N; i++) {
667                const char* localeStr =  locales[i].string();
668                if (localeStr == NULL || strlen(localeStr) == 0) {
669                    localeStr = "--_--";
670                }
671                printf(" '%s'", localeStr);
672            }
673            printf("\n");
674        } else if (strcmp("configurations", option) == 0) {
675            Vector<ResTable_config> configs;
676            res.getConfigurations(&configs);
677            const size_t N = configs.size();
678            for (size_t i=0; i<N; i++) {
679                printf("%s\n", configs[i].toString().string());
680            }
681        } else {
682            fprintf(stderr, "ERROR: unknown dump option '%s'\n", option);
683            goto bail;
684        }
685    }
686
687    result = NO_ERROR;
688
689bail:
690    if (asset) {
691        delete asset;
692    }
693    return (result != NO_ERROR);
694}
695
696
697/*
698 * Handle the "add" command, which wants to add files to a new or
699 * pre-existing archive.
700 */
701int doAdd(Bundle* bundle)
702{
703    ZipFile* zip = NULL;
704    status_t result = UNKNOWN_ERROR;
705    const char* zipFileName;
706
707    if (bundle->getUpdate()) {
708        /* avoid confusion */
709        fprintf(stderr, "ERROR: can't use '-u' with add\n");
710        goto bail;
711    }
712
713    if (bundle->getFileSpecCount() < 1) {
714        fprintf(stderr, "ERROR: must specify zip file name\n");
715        goto bail;
716    }
717    zipFileName = bundle->getFileSpecEntry(0);
718
719    if (bundle->getFileSpecCount() < 2) {
720        fprintf(stderr, "NOTE: nothing to do\n");
721        goto bail;
722    }
723
724    zip = openReadWrite(zipFileName, true);
725    if (zip == NULL) {
726        fprintf(stderr, "ERROR: failed opening/creating '%s' as Zip file\n", zipFileName);
727        goto bail;
728    }
729
730    for (int i = 1; i < bundle->getFileSpecCount(); i++) {
731        const char* fileName = bundle->getFileSpecEntry(i);
732
733        if (strcasecmp(String8(fileName).getPathExtension().string(), ".gz") == 0) {
734            printf(" '%s'... (from gzip)\n", fileName);
735            result = zip->addGzip(fileName, String8(fileName).getBasePath().string(), NULL);
736        } else {
737            printf(" '%s'...\n", fileName);
738            result = zip->add(fileName, bundle->getCompressionMethod(), NULL);
739        }
740        if (result != NO_ERROR) {
741            fprintf(stderr, "Unable to add '%s' to '%s'", bundle->getFileSpecEntry(i), zipFileName);
742            if (result == NAME_NOT_FOUND)
743                fprintf(stderr, ": file not found\n");
744            else if (result == ALREADY_EXISTS)
745                fprintf(stderr, ": already exists in archive\n");
746            else
747                fprintf(stderr, "\n");
748            goto bail;
749        }
750    }
751
752    result = NO_ERROR;
753
754bail:
755    delete zip;
756    return (result != NO_ERROR);
757}
758
759
760/*
761 * Delete files from an existing archive.
762 */
763int doRemove(Bundle* bundle)
764{
765    ZipFile* zip = NULL;
766    status_t result = UNKNOWN_ERROR;
767    const char* zipFileName;
768
769    if (bundle->getFileSpecCount() < 1) {
770        fprintf(stderr, "ERROR: must specify zip file name\n");
771        goto bail;
772    }
773    zipFileName = bundle->getFileSpecEntry(0);
774
775    if (bundle->getFileSpecCount() < 2) {
776        fprintf(stderr, "NOTE: nothing to do\n");
777        goto bail;
778    }
779
780    zip = openReadWrite(zipFileName, false);
781    if (zip == NULL) {
782        fprintf(stderr, "ERROR: failed opening Zip archive '%s'\n",
783            zipFileName);
784        goto bail;
785    }
786
787    for (int i = 1; i < bundle->getFileSpecCount(); i++) {
788        const char* fileName = bundle->getFileSpecEntry(i);
789        ZipEntry* entry;
790
791        entry = zip->getEntryByName(fileName);
792        if (entry == NULL) {
793            printf(" '%s' NOT FOUND\n", fileName);
794            continue;
795        }
796
797        result = zip->remove(entry);
798
799        if (result != NO_ERROR) {
800            fprintf(stderr, "Unable to delete '%s' from '%s'\n",
801                bundle->getFileSpecEntry(i), zipFileName);
802            goto bail;
803        }
804    }
805
806    /* update the archive */
807    zip->flush();
808
809bail:
810    delete zip;
811    return (result != NO_ERROR);
812}
813
814
815/*
816 * Package up an asset directory and associated application files.
817 */
818int doPackage(Bundle* bundle)
819{
820    const char* outputAPKFile;
821    int retVal = 1;
822    status_t err;
823    sp<AaptAssets> assets;
824    int N;
825
826    // -c zz_ZZ means do pseudolocalization
827    ResourceFilter filter;
828    err = filter.parse(bundle->getConfigurations());
829    if (err != NO_ERROR) {
830        goto bail;
831    }
832    if (filter.containsPseudo()) {
833        bundle->setPseudolocalize(true);
834    }
835
836    N = bundle->getFileSpecCount();
837    if (N < 1 && bundle->getResourceSourceDirs().size() == 0 && bundle->getJarFiles().size() == 0
838            && bundle->getAndroidManifestFile() == NULL && bundle->getAssetSourceDir() == NULL) {
839        fprintf(stderr, "ERROR: no input files\n");
840        goto bail;
841    }
842
843    outputAPKFile = bundle->getOutputAPKFile();
844
845    // Make sure the filenames provided exist and are of the appropriate type.
846    if (outputAPKFile) {
847        FileType type;
848        type = getFileType(outputAPKFile);
849        if (type != kFileTypeNonexistent && type != kFileTypeRegular) {
850            fprintf(stderr,
851                "ERROR: output file '%s' exists but is not regular file\n",
852                outputAPKFile);
853            goto bail;
854        }
855    }
856
857    // Load the assets.
858    assets = new AaptAssets();
859    err = assets->slurpFromArgs(bundle);
860    if (err < 0) {
861        goto bail;
862    }
863
864    if (bundle->getVerbose()) {
865        assets->print();
866    }
867
868    // If they asked for any files that need to be compiled, do so.
869    if (bundle->getResourceSourceDirs().size() || bundle->getAndroidManifestFile()) {
870        err = buildResources(bundle, assets);
871        if (err != 0) {
872            goto bail;
873        }
874    }
875
876    // At this point we've read everything and processed everything.  From here
877    // on out it's just writing output files.
878    if (SourcePos::hasErrors()) {
879        goto bail;
880    }
881
882    // Write out R.java constants
883    if (assets->getPackage() == assets->getSymbolsPrivatePackage()) {
884        err = writeResourceSymbols(bundle, assets, assets->getPackage(), true);
885        if (err < 0) {
886            goto bail;
887        }
888    } else {
889        err = writeResourceSymbols(bundle, assets, assets->getPackage(), false);
890        if (err < 0) {
891            goto bail;
892        }
893        err = writeResourceSymbols(bundle, assets, assets->getSymbolsPrivatePackage(), true);
894        if (err < 0) {
895            goto bail;
896        }
897    }
898
899    // Write the apk
900    if (outputAPKFile) {
901        err = writeAPK(bundle, assets, String8(outputAPKFile));
902        if (err != NO_ERROR) {
903            fprintf(stderr, "ERROR: packaging of '%s' failed\n", outputAPKFile);
904            goto bail;
905        }
906    }
907
908    retVal = 0;
909bail:
910    if (SourcePos::hasErrors()) {
911        SourcePos::printErrors(stderr);
912    }
913    return retVal;
914}
915