Resource.cpp revision 7c71023b7f5aafac09bd2f0425fccb5fe2d22b27
1//
2// Copyright 2006 The Android Open Source Project
3//
4// Build resource files from raw assets.
5//
6#include "Main.h"
7#include "AaptAssets.h"
8#include "StringPool.h"
9#include "XMLNode.h"
10#include "ResourceTable.h"
11#include "Images.h"
12
13#define NOISY(x) // x
14
15// ==========================================================================
16// ==========================================================================
17// ==========================================================================
18
19class PackageInfo
20{
21public:
22    PackageInfo()
23    {
24    }
25    ~PackageInfo()
26    {
27    }
28
29    status_t parsePackage(const sp<AaptGroup>& grp);
30};
31
32// ==========================================================================
33// ==========================================================================
34// ==========================================================================
35
36static String8 parseResourceName(const String8& leaf)
37{
38    const char* firstDot = strchr(leaf.string(), '.');
39    const char* str = leaf.string();
40
41    if (firstDot) {
42        return String8(str, firstDot-str);
43    } else {
44        return String8(str);
45    }
46}
47
48ResourceTypeSet::ResourceTypeSet()
49    :RefBase(),
50     KeyedVector<String8,sp<AaptGroup> >()
51{
52}
53
54class ResourceDirIterator
55{
56public:
57    ResourceDirIterator(const sp<ResourceTypeSet>& set, const String8& resType)
58        : mResType(resType), mSet(set), mSetPos(0), mGroupPos(0)
59    {
60    }
61
62    inline const sp<AaptGroup>& getGroup() const { return mGroup; }
63    inline const sp<AaptFile>& getFile() const { return mFile; }
64
65    inline const String8& getBaseName() const { return mBaseName; }
66    inline const String8& getLeafName() const { return mLeafName; }
67    inline String8 getPath() const { return mPath; }
68    inline const ResTable_config& getParams() const { return mParams; }
69
70    enum {
71        EOD = 1
72    };
73
74    ssize_t next()
75    {
76        while (true) {
77            sp<AaptGroup> group;
78            sp<AaptFile> file;
79
80            // Try to get next file in this current group.
81            if (mGroup != NULL && mGroupPos < mGroup->getFiles().size()) {
82                group = mGroup;
83                file = group->getFiles().valueAt(mGroupPos++);
84
85            // Try to get the next group/file in this directory
86            } else if (mSetPos < mSet->size()) {
87                mGroup = group = mSet->valueAt(mSetPos++);
88                if (group->getFiles().size() < 1) {
89                    continue;
90                }
91                file = group->getFiles().valueAt(0);
92                mGroupPos = 1;
93
94            // All done!
95            } else {
96                return EOD;
97            }
98
99            mFile = file;
100
101            String8 leaf(group->getLeaf());
102            mLeafName = String8(leaf);
103            mParams = file->getGroupEntry().toParams();
104            NOISY(printf("Dir %s: mcc=%d mnc=%d lang=%c%c cnt=%c%c orient=%d ui=%d density=%d touch=%d key=%d inp=%d nav=%d\n",
105                   group->getPath().string(), mParams.mcc, mParams.mnc,
106                   mParams.language[0] ? mParams.language[0] : '-',
107                   mParams.language[1] ? mParams.language[1] : '-',
108                   mParams.country[0] ? mParams.country[0] : '-',
109                   mParams.country[1] ? mParams.country[1] : '-',
110                   mParams.orientation, mParams.uiMode,
111                   mParams.density, mParams.touchscreen, mParams.keyboard,
112                   mParams.inputFlags, mParams.navigation));
113            mPath = "res";
114            mPath.appendPath(file->getGroupEntry().toDirName(mResType));
115            mPath.appendPath(leaf);
116            mBaseName = parseResourceName(leaf);
117            if (mBaseName == "") {
118                fprintf(stderr, "Error: malformed resource filename %s\n",
119                        file->getPrintableSource().string());
120                return UNKNOWN_ERROR;
121            }
122
123            NOISY(printf("file name=%s\n", mBaseName.string()));
124
125            return NO_ERROR;
126        }
127    }
128
129private:
130    String8 mResType;
131
132    const sp<ResourceTypeSet> mSet;
133    size_t mSetPos;
134
135    sp<AaptGroup> mGroup;
136    size_t mGroupPos;
137
138    sp<AaptFile> mFile;
139    String8 mBaseName;
140    String8 mLeafName;
141    String8 mPath;
142    ResTable_config mParams;
143};
144
145// ==========================================================================
146// ==========================================================================
147// ==========================================================================
148
149bool isValidResourceType(const String8& type)
150{
151    return type == "anim" || type == "drawable" || type == "layout"
152        || type == "values" || type == "xml" || type == "raw"
153        || type == "color" || type == "menu" || type == "mipmap";
154}
155
156static sp<AaptFile> getResourceFile(const sp<AaptAssets>& assets, bool makeIfNecessary=true)
157{
158    sp<AaptGroup> group = assets->getFiles().valueFor(String8("resources.arsc"));
159    sp<AaptFile> file;
160    if (group != NULL) {
161        file = group->getFiles().valueFor(AaptGroupEntry());
162        if (file != NULL) {
163            return file;
164        }
165    }
166
167    if (!makeIfNecessary) {
168        return NULL;
169    }
170    return assets->addFile(String8("resources.arsc"), AaptGroupEntry(), String8(),
171                            NULL, String8());
172}
173
174static status_t parsePackage(Bundle* bundle, const sp<AaptAssets>& assets,
175    const sp<AaptGroup>& grp)
176{
177    if (grp->getFiles().size() != 1) {
178        fprintf(stderr, "warning: Multiple AndroidManifest.xml files found, using %s\n",
179                grp->getFiles().valueAt(0)->getPrintableSource().string());
180    }
181
182    sp<AaptFile> file = grp->getFiles().valueAt(0);
183
184    ResXMLTree block;
185    status_t err = parseXMLResource(file, &block);
186    if (err != NO_ERROR) {
187        return err;
188    }
189    //printXMLBlock(&block);
190
191    ResXMLTree::event_code_t code;
192    while ((code=block.next()) != ResXMLTree::START_TAG
193           && code != ResXMLTree::END_DOCUMENT
194           && code != ResXMLTree::BAD_DOCUMENT) {
195    }
196
197    size_t len;
198    if (code != ResXMLTree::START_TAG) {
199        fprintf(stderr, "%s:%d: No start tag found\n",
200                file->getPrintableSource().string(), block.getLineNumber());
201        return UNKNOWN_ERROR;
202    }
203    if (strcmp16(block.getElementName(&len), String16("manifest").string()) != 0) {
204        fprintf(stderr, "%s:%d: Invalid start tag %s, expected <manifest>\n",
205                file->getPrintableSource().string(), block.getLineNumber(),
206                String8(block.getElementName(&len)).string());
207        return UNKNOWN_ERROR;
208    }
209
210    ssize_t nameIndex = block.indexOfAttribute(NULL, "package");
211    if (nameIndex < 0) {
212        fprintf(stderr, "%s:%d: <manifest> does not have package attribute.\n",
213                file->getPrintableSource().string(), block.getLineNumber());
214        return UNKNOWN_ERROR;
215    }
216
217    assets->setPackage(String8(block.getAttributeStringValue(nameIndex, &len)));
218
219    String16 uses_sdk16("uses-sdk");
220    while ((code=block.next()) != ResXMLTree::END_DOCUMENT
221           && code != ResXMLTree::BAD_DOCUMENT) {
222        if (code == ResXMLTree::START_TAG) {
223            if (strcmp16(block.getElementName(&len), uses_sdk16.string()) == 0) {
224                ssize_t minSdkIndex = block.indexOfAttribute(RESOURCES_ANDROID_NAMESPACE,
225                                                             "minSdkVersion");
226                if (minSdkIndex >= 0) {
227                    const uint16_t* minSdk16 = block.getAttributeStringValue(minSdkIndex, &len);
228                    const char* minSdk8 = strdup(String8(minSdk16).string());
229                    bundle->setManifestMinSdkVersion(minSdk8);
230                }
231            }
232        }
233    }
234
235    return NO_ERROR;
236}
237
238// ==========================================================================
239// ==========================================================================
240// ==========================================================================
241
242static status_t makeFileResources(Bundle* bundle, const sp<AaptAssets>& assets,
243                                  ResourceTable* table,
244                                  const sp<ResourceTypeSet>& set,
245                                  const char* resType)
246{
247    String8 type8(resType);
248    String16 type16(resType);
249
250    bool hasErrors = false;
251
252    ResourceDirIterator it(set, String8(resType));
253    ssize_t res;
254    while ((res=it.next()) == NO_ERROR) {
255        if (bundle->getVerbose()) {
256            printf("    (new resource id %s from %s)\n",
257                   it.getBaseName().string(), it.getFile()->getPrintableSource().string());
258        }
259        String16 baseName(it.getBaseName());
260        const char16_t* str = baseName.string();
261        const char16_t* const end = str + baseName.size();
262        while (str < end) {
263            if (!((*str >= 'a' && *str <= 'z')
264                    || (*str >= '0' && *str <= '9')
265                    || *str == '_' || *str == '.')) {
266                fprintf(stderr, "%s: Invalid file name: must contain only [a-z0-9_.]\n",
267                        it.getPath().string());
268                hasErrors = true;
269            }
270            str++;
271        }
272        String8 resPath = it.getPath();
273        resPath.convertToResPath();
274        table->addEntry(SourcePos(it.getPath(), 0), String16(assets->getPackage()),
275                        type16,
276                        baseName,
277                        String16(resPath),
278                        NULL,
279                        &it.getParams());
280        assets->addResource(it.getLeafName(), resPath, it.getFile(), type8);
281    }
282
283    return hasErrors ? UNKNOWN_ERROR : NO_ERROR;
284}
285
286static status_t preProcessImages(Bundle* bundle, const sp<AaptAssets>& assets,
287                          const sp<ResourceTypeSet>& set, const char* type)
288{
289    ResourceDirIterator it(set, String8(type));
290    Vector<sp<AaptFile> > newNameFiles;
291    Vector<String8> newNamePaths;
292    bool hasErrors = false;
293    ssize_t res;
294    while ((res=it.next()) == NO_ERROR) {
295        res = preProcessImage(bundle, assets, it.getFile(), NULL);
296        if (res < NO_ERROR) {
297            hasErrors = true;
298        }
299    }
300
301    return (hasErrors || (res < NO_ERROR)) ? UNKNOWN_ERROR : NO_ERROR;
302}
303
304status_t postProcessImages(const sp<AaptAssets>& assets,
305                           ResourceTable* table,
306                           const sp<ResourceTypeSet>& set)
307{
308    ResourceDirIterator it(set, String8("drawable"));
309    bool hasErrors = false;
310    ssize_t res;
311    while ((res=it.next()) == NO_ERROR) {
312        res = postProcessImage(assets, table, it.getFile());
313        if (res < NO_ERROR) {
314            hasErrors = true;
315        }
316    }
317
318    return (hasErrors || (res < NO_ERROR)) ? UNKNOWN_ERROR : NO_ERROR;
319}
320
321static void collect_files(const sp<AaptDir>& dir,
322        KeyedVector<String8, sp<ResourceTypeSet> >* resources)
323{
324    const DefaultKeyedVector<String8, sp<AaptGroup> >& groups = dir->getFiles();
325    int N = groups.size();
326    for (int i=0; i<N; i++) {
327        String8 leafName = groups.keyAt(i);
328        const sp<AaptGroup>& group = groups.valueAt(i);
329
330        const DefaultKeyedVector<AaptGroupEntry, sp<AaptFile> >& files
331                = group->getFiles();
332
333        if (files.size() == 0) {
334            continue;
335        }
336
337        String8 resType = files.valueAt(0)->getResourceType();
338
339        ssize_t index = resources->indexOfKey(resType);
340
341        if (index < 0) {
342            sp<ResourceTypeSet> set = new ResourceTypeSet();
343            set->add(leafName, group);
344            resources->add(resType, set);
345        } else {
346            sp<ResourceTypeSet> set = resources->valueAt(index);
347            index = set->indexOfKey(leafName);
348            if (index < 0) {
349                set->add(leafName, group);
350            } else {
351                sp<AaptGroup> existingGroup = set->valueAt(index);
352                int M = files.size();
353                for (int j=0; j<M; j++) {
354                    existingGroup->addFile(files.valueAt(j));
355                }
356            }
357        }
358    }
359}
360
361static void collect_files(const sp<AaptAssets>& ass,
362        KeyedVector<String8, sp<ResourceTypeSet> >* resources)
363{
364    const Vector<sp<AaptDir> >& dirs = ass->resDirs();
365    int N = dirs.size();
366
367    for (int i=0; i<N; i++) {
368        sp<AaptDir> d = dirs.itemAt(i);
369        collect_files(d, resources);
370
371        // don't try to include the res dir
372        ass->removeDir(d->getLeaf());
373    }
374}
375
376enum {
377    ATTR_OKAY = -1,
378    ATTR_NOT_FOUND = -2,
379    ATTR_LEADING_SPACES = -3,
380    ATTR_TRAILING_SPACES = -4
381};
382static int validateAttr(const String8& path, const ResTable& table,
383        const ResXMLParser& parser,
384        const char* ns, const char* attr, const char* validChars, bool required)
385{
386    size_t len;
387
388    ssize_t index = parser.indexOfAttribute(ns, attr);
389    const uint16_t* str;
390    Res_value value;
391    if (index >= 0 && parser.getAttributeValue(index, &value) >= 0) {
392        const ResStringPool* pool = &parser.getStrings();
393        if (value.dataType == Res_value::TYPE_REFERENCE) {
394            uint32_t specFlags = 0;
395            int strIdx;
396            if ((strIdx=table.resolveReference(&value, 0x10000000, NULL, &specFlags)) < 0) {
397                fprintf(stderr, "%s:%d: Tag <%s> attribute %s references unknown resid 0x%08x.\n",
398                        path.string(), parser.getLineNumber(),
399                        String8(parser.getElementName(&len)).string(), attr,
400                        value.data);
401                return ATTR_NOT_FOUND;
402            }
403
404            pool = table.getTableStringBlock(strIdx);
405            #if 0
406            if (pool != NULL) {
407                str = pool->stringAt(value.data, &len);
408            }
409            printf("***** RES ATTR: %s specFlags=0x%x strIdx=%d: %s\n", attr,
410                    specFlags, strIdx, str != NULL ? String8(str).string() : "???");
411            #endif
412            if ((specFlags&~ResTable_typeSpec::SPEC_PUBLIC) != 0 && false) {
413                fprintf(stderr, "%s:%d: Tag <%s> attribute %s varies by configurations 0x%x.\n",
414                        path.string(), parser.getLineNumber(),
415                        String8(parser.getElementName(&len)).string(), attr,
416                        specFlags);
417                return ATTR_NOT_FOUND;
418            }
419        }
420        if (value.dataType == Res_value::TYPE_STRING) {
421            if (pool == NULL) {
422                fprintf(stderr, "%s:%d: Tag <%s> attribute %s has no string block.\n",
423                        path.string(), parser.getLineNumber(),
424                        String8(parser.getElementName(&len)).string(), attr);
425                return ATTR_NOT_FOUND;
426            }
427            if ((str=pool->stringAt(value.data, &len)) == NULL) {
428                fprintf(stderr, "%s:%d: Tag <%s> attribute %s has corrupt string value.\n",
429                        path.string(), parser.getLineNumber(),
430                        String8(parser.getElementName(&len)).string(), attr);
431                return ATTR_NOT_FOUND;
432            }
433        } else {
434            fprintf(stderr, "%s:%d: Tag <%s> attribute %s has invalid type %d.\n",
435                    path.string(), parser.getLineNumber(),
436                    String8(parser.getElementName(&len)).string(), attr,
437                    value.dataType);
438            return ATTR_NOT_FOUND;
439        }
440        if (validChars) {
441            for (size_t i=0; i<len; i++) {
442                uint16_t c = str[i];
443                const char* p = validChars;
444                bool okay = false;
445                while (*p) {
446                    if (c == *p) {
447                        okay = true;
448                        break;
449                    }
450                    p++;
451                }
452                if (!okay) {
453                    fprintf(stderr, "%s:%d: Tag <%s> attribute %s has invalid character '%c'.\n",
454                            path.string(), parser.getLineNumber(),
455                            String8(parser.getElementName(&len)).string(), attr, (char)str[i]);
456                    return (int)i;
457                }
458            }
459        }
460        if (*str == ' ') {
461            fprintf(stderr, "%s:%d: Tag <%s> attribute %s can not start with a space.\n",
462                    path.string(), parser.getLineNumber(),
463                    String8(parser.getElementName(&len)).string(), attr);
464            return ATTR_LEADING_SPACES;
465        }
466        if (str[len-1] == ' ') {
467            fprintf(stderr, "%s:%d: Tag <%s> attribute %s can not end with a space.\n",
468                    path.string(), parser.getLineNumber(),
469                    String8(parser.getElementName(&len)).string(), attr);
470            return ATTR_TRAILING_SPACES;
471        }
472        return ATTR_OKAY;
473    }
474    if (required) {
475        fprintf(stderr, "%s:%d: Tag <%s> missing required attribute %s.\n",
476                path.string(), parser.getLineNumber(),
477                String8(parser.getElementName(&len)).string(), attr);
478        return ATTR_NOT_FOUND;
479    }
480    return ATTR_OKAY;
481}
482
483static void checkForIds(const String8& path, ResXMLParser& parser)
484{
485    ResXMLTree::event_code_t code;
486    while ((code=parser.next()) != ResXMLTree::END_DOCUMENT
487           && code > ResXMLTree::BAD_DOCUMENT) {
488        if (code == ResXMLTree::START_TAG) {
489            ssize_t index = parser.indexOfAttribute(NULL, "id");
490            if (index >= 0) {
491                fprintf(stderr, "%s:%d: warning: found plain 'id' attribute; did you mean the new 'android:id' name?\n",
492                        path.string(), parser.getLineNumber());
493            }
494        }
495    }
496}
497
498static bool applyFileOverlay(Bundle *bundle,
499                             const sp<AaptAssets>& assets,
500                             sp<ResourceTypeSet> *baseSet,
501                             const char *resType)
502{
503    if (bundle->getVerbose()) {
504        printf("applyFileOverlay for %s\n", resType);
505    }
506
507    // Replace any base level files in this category with any found from the overlay
508    // Also add any found only in the overlay.
509    sp<AaptAssets> overlay = assets->getOverlay();
510    String8 resTypeString(resType);
511
512    // work through the linked list of overlays
513    while (overlay.get()) {
514        KeyedVector<String8, sp<ResourceTypeSet> >* overlayRes = overlay->getResources();
515
516        // get the overlay resources of the requested type
517        ssize_t index = overlayRes->indexOfKey(resTypeString);
518        if (index >= 0) {
519            sp<ResourceTypeSet> overlaySet = overlayRes->valueAt(index);
520
521            // for each of the resources, check for a match in the previously built
522            // non-overlay "baseset".
523            size_t overlayCount = overlaySet->size();
524            for (size_t overlayIndex=0; overlayIndex<overlayCount; overlayIndex++) {
525                if (bundle->getVerbose()) {
526                    printf("trying overlaySet Key=%s\n",overlaySet->keyAt(overlayIndex).string());
527                }
528                size_t baseIndex = UNKNOWN_ERROR;
529                if (baseSet->get() != NULL) {
530                    baseIndex = (*baseSet)->indexOfKey(overlaySet->keyAt(overlayIndex));
531                }
532                if (baseIndex < UNKNOWN_ERROR) {
533                    // look for same flavor.  For a given file (strings.xml, for example)
534                    // there may be a locale specific or other flavors - we want to match
535                    // the same flavor.
536                    sp<AaptGroup> overlayGroup = overlaySet->valueAt(overlayIndex);
537                    sp<AaptGroup> baseGroup = (*baseSet)->valueAt(baseIndex);
538
539                    DefaultKeyedVector<AaptGroupEntry, sp<AaptFile> > overlayFiles =
540                            overlayGroup->getFiles();
541                    if (bundle->getVerbose()) {
542                        DefaultKeyedVector<AaptGroupEntry, sp<AaptFile> > baseFiles =
543                                baseGroup->getFiles();
544                        for (size_t i=0; i < baseFiles.size(); i++) {
545                            printf("baseFile %d has flavor %s\n", i,
546                                    baseFiles.keyAt(i).toString().string());
547                        }
548                        for (size_t i=0; i < overlayFiles.size(); i++) {
549                            printf("overlayFile %d has flavor %s\n", i,
550                                    overlayFiles.keyAt(i).toString().string());
551                        }
552                    }
553
554                    size_t overlayGroupSize = overlayFiles.size();
555                    for (size_t overlayGroupIndex = 0;
556                            overlayGroupIndex<overlayGroupSize;
557                            overlayGroupIndex++) {
558                        size_t baseFileIndex =
559                                baseGroup->getFiles().indexOfKey(overlayFiles.
560                                keyAt(overlayGroupIndex));
561                        if(baseFileIndex < UNKNOWN_ERROR) {
562                            if (bundle->getVerbose()) {
563                                printf("found a match (%d) for overlay file %s, for flavor %s\n",
564                                        baseFileIndex,
565                                        overlayGroup->getLeaf().string(),
566                                        overlayFiles.keyAt(overlayGroupIndex).toString().string());
567                            }
568                            baseGroup->removeFile(baseFileIndex);
569                        } else {
570                            // didn't find a match fall through and add it..
571                        }
572                        baseGroup->addFile(overlayFiles.valueAt(overlayGroupIndex));
573                        assets->addGroupEntry(overlayFiles.keyAt(overlayGroupIndex));
574                    }
575                } else {
576                    if (baseSet->get() == NULL) {
577                        *baseSet = new ResourceTypeSet();
578                        assets->getResources()->add(String8(resType), *baseSet);
579                    }
580                    // this group doesn't exist (a file that's only in the overlay)
581                    (*baseSet)->add(overlaySet->keyAt(overlayIndex),
582                            overlaySet->valueAt(overlayIndex));
583                    // make sure all flavors are defined in the resources.
584                    sp<AaptGroup> overlayGroup = overlaySet->valueAt(overlayIndex);
585                    DefaultKeyedVector<AaptGroupEntry, sp<AaptFile> > overlayFiles =
586                            overlayGroup->getFiles();
587                    size_t overlayGroupSize = overlayFiles.size();
588                    for (size_t overlayGroupIndex = 0;
589                            overlayGroupIndex<overlayGroupSize;
590                            overlayGroupIndex++) {
591                        assets->addGroupEntry(overlayFiles.keyAt(overlayGroupIndex));
592                    }
593                }
594            }
595            // this overlay didn't have resources for this type
596        }
597        // try next overlay
598        overlay = overlay->getOverlay();
599    }
600    return true;
601}
602
603void addTagAttribute(const sp<XMLNode>& node, const char* ns8,
604        const char* attr8, const char* value)
605{
606    if (value == NULL) {
607        return;
608    }
609
610    const String16 ns(ns8);
611    const String16 attr(attr8);
612
613    if (node->getAttribute(ns, attr) != NULL) {
614        fprintf(stderr, "Warning: AndroidManifest.xml already defines %s (in %s);"
615                        " using existing value in manifest.\n",
616                String8(attr).string(), String8(ns).string());
617        return;
618    }
619
620    node->addAttribute(ns, attr, String16(value));
621}
622
623static void fullyQualifyClassName(const String8& package, sp<XMLNode> node,
624        const String16& attrName) {
625    XMLNode::attribute_entry* attr = node->editAttribute(
626            String16("http://schemas.android.com/apk/res/android"), attrName);
627    if (attr != NULL) {
628        String8 name(attr->string);
629
630        // asdf     --> package.asdf
631        // .asdf  .a.b  --> package.asdf package.a.b
632        // asdf.adsf --> asdf.asdf
633        String8 className;
634        const char* p = name.string();
635        const char* q = strchr(p, '.');
636        if (p == q) {
637            className += package;
638            className += name;
639        } else if (q == NULL) {
640            className += package;
641            className += ".";
642            className += name;
643        } else {
644            className += name;
645        }
646        NOISY(printf("Qualifying class '%s' to '%s'", name.string(), className.string()));
647        attr->string.setTo(String16(className));
648    }
649}
650
651status_t massageManifest(Bundle* bundle, sp<XMLNode> root)
652{
653    root = root->searchElement(String16(), String16("manifest"));
654    if (root == NULL) {
655        fprintf(stderr, "No <manifest> tag.\n");
656        return UNKNOWN_ERROR;
657    }
658
659    addTagAttribute(root, RESOURCES_ANDROID_NAMESPACE, "versionCode",
660            bundle->getVersionCode());
661    addTagAttribute(root, RESOURCES_ANDROID_NAMESPACE, "versionName",
662            bundle->getVersionName());
663
664    if (bundle->getMinSdkVersion() != NULL
665            || bundle->getTargetSdkVersion() != NULL
666            || bundle->getMaxSdkVersion() != NULL) {
667        sp<XMLNode> vers = root->getChildElement(String16(), String16("uses-sdk"));
668        if (vers == NULL) {
669            vers = XMLNode::newElement(root->getFilename(), String16(), String16("uses-sdk"));
670            root->insertChildAt(vers, 0);
671        }
672
673        addTagAttribute(vers, RESOURCES_ANDROID_NAMESPACE, "minSdkVersion",
674                bundle->getMinSdkVersion());
675        addTagAttribute(vers, RESOURCES_ANDROID_NAMESPACE, "targetSdkVersion",
676                bundle->getTargetSdkVersion());
677        addTagAttribute(vers, RESOURCES_ANDROID_NAMESPACE, "maxSdkVersion",
678                bundle->getMaxSdkVersion());
679    }
680
681    if (bundle->getDebugMode()) {
682        sp<XMLNode> application = root->getChildElement(String16(), String16("application"));
683        if (application != NULL) {
684            addTagAttribute(application, RESOURCES_ANDROID_NAMESPACE, "debuggable", "true");
685        }
686    }
687
688    // Deal with manifest package name overrides
689    const char* manifestPackageNameOverride = bundle->getManifestPackageNameOverride();
690    if (manifestPackageNameOverride != NULL) {
691        // Update the actual package name
692        XMLNode::attribute_entry* attr = root->editAttribute(String16(), String16("package"));
693        if (attr == NULL) {
694            fprintf(stderr, "package name is required with --rename-manifest-package.\n");
695            return UNKNOWN_ERROR;
696        }
697        String8 origPackage(attr->string);
698        attr->string.setTo(String16(manifestPackageNameOverride));
699        NOISY(printf("Overriding package '%s' to be '%s'\n", origPackage.string(), manifestPackageNameOverride));
700
701        // Make class names fully qualified
702        sp<XMLNode> application = root->getChildElement(String16(), String16("application"));
703        if (application != NULL) {
704            fullyQualifyClassName(origPackage, application, String16("name"));
705            fullyQualifyClassName(origPackage, application, String16("backupAgent"));
706
707            Vector<sp<XMLNode> >& children = const_cast<Vector<sp<XMLNode> >&>(application->getChildren());
708            for (size_t i = 0; i < children.size(); i++) {
709                sp<XMLNode> child = children.editItemAt(i);
710                String8 tag(child->getElementName());
711                if (tag == "activity" || tag == "service" || tag == "receiver" || tag == "provider") {
712                    fullyQualifyClassName(origPackage, child, String16("name"));
713                } else if (tag == "activity-alias") {
714                    fullyQualifyClassName(origPackage, child, String16("name"));
715                    fullyQualifyClassName(origPackage, child, String16("targetActivity"));
716                }
717            }
718        }
719    }
720
721    // Deal with manifest package name overrides
722    const char* instrumentationPackageNameOverride = bundle->getInstrumentationPackageNameOverride();
723    if (instrumentationPackageNameOverride != NULL) {
724        // Fix up instrumentation targets.
725        Vector<sp<XMLNode> >& children = const_cast<Vector<sp<XMLNode> >&>(root->getChildren());
726        for (size_t i = 0; i < children.size(); i++) {
727            sp<XMLNode> child = children.editItemAt(i);
728            String8 tag(child->getElementName());
729            if (tag == "instrumentation") {
730                XMLNode::attribute_entry* attr = child->editAttribute(
731                        String16("http://schemas.android.com/apk/res/android"), String16("targetPackage"));
732                if (attr != NULL) {
733                    attr->string.setTo(String16(instrumentationPackageNameOverride));
734                }
735            }
736        }
737    }
738
739    return NO_ERROR;
740}
741
742#define ASSIGN_IT(n) \
743        do { \
744            ssize_t index = resources->indexOfKey(String8(#n)); \
745            if (index >= 0) { \
746                n ## s = resources->valueAt(index); \
747            } \
748        } while (0)
749
750status_t buildResources(Bundle* bundle, const sp<AaptAssets>& assets)
751{
752    // First, look for a package file to parse.  This is required to
753    // be able to generate the resource information.
754    sp<AaptGroup> androidManifestFile =
755            assets->getFiles().valueFor(String8("AndroidManifest.xml"));
756    if (androidManifestFile == NULL) {
757        fprintf(stderr, "ERROR: No AndroidManifest.xml file found.\n");
758        return UNKNOWN_ERROR;
759    }
760
761    status_t err = parsePackage(bundle, assets, androidManifestFile);
762    if (err != NO_ERROR) {
763        return err;
764    }
765
766    NOISY(printf("Creating resources for package %s\n",
767                 assets->getPackage().string()));
768
769    ResourceTable table(bundle, String16(assets->getPackage()));
770    err = table.addIncludedResources(bundle, assets);
771    if (err != NO_ERROR) {
772        return err;
773    }
774
775    NOISY(printf("Found %d included resource packages\n", (int)table.size()));
776
777    // Standard flags for compiled XML and optional UTF-8 encoding
778    int xmlFlags = XML_COMPILE_STANDARD_RESOURCE;
779
780    /* Only enable UTF-8 if the caller of aapt didn't specifically
781     * request UTF-16 encoding and the parameters of this package
782     * allow UTF-8 to be used.
783     */
784    if (!bundle->getWantUTF16()
785            && bundle->isMinSdkAtLeast(SDK_FROYO)) {
786        xmlFlags |= XML_COMPILE_UTF8;
787    }
788
789    // --------------------------------------------------------------
790    // First, gather all resource information.
791    // --------------------------------------------------------------
792
793    // resType -> leafName -> group
794    KeyedVector<String8, sp<ResourceTypeSet> > *resources =
795            new KeyedVector<String8, sp<ResourceTypeSet> >;
796    collect_files(assets, resources);
797
798    sp<ResourceTypeSet> drawables;
799    sp<ResourceTypeSet> layouts;
800    sp<ResourceTypeSet> anims;
801    sp<ResourceTypeSet> xmls;
802    sp<ResourceTypeSet> raws;
803    sp<ResourceTypeSet> colors;
804    sp<ResourceTypeSet> menus;
805    sp<ResourceTypeSet> mipmaps;
806
807    ASSIGN_IT(drawable);
808    ASSIGN_IT(layout);
809    ASSIGN_IT(anim);
810    ASSIGN_IT(xml);
811    ASSIGN_IT(raw);
812    ASSIGN_IT(color);
813    ASSIGN_IT(menu);
814    ASSIGN_IT(mipmap);
815
816    assets->setResources(resources);
817    // now go through any resource overlays and collect their files
818    sp<AaptAssets> current = assets->getOverlay();
819    while(current.get()) {
820        KeyedVector<String8, sp<ResourceTypeSet> > *resources =
821                new KeyedVector<String8, sp<ResourceTypeSet> >;
822        current->setResources(resources);
823        collect_files(current, resources);
824        current = current->getOverlay();
825    }
826    // apply the overlay files to the base set
827    if (!applyFileOverlay(bundle, assets, &drawables, "drawable") ||
828            !applyFileOverlay(bundle, assets, &layouts, "layout") ||
829            !applyFileOverlay(bundle, assets, &anims, "anim") ||
830            !applyFileOverlay(bundle, assets, &xmls, "xml") ||
831            !applyFileOverlay(bundle, assets, &raws, "raw") ||
832            !applyFileOverlay(bundle, assets, &colors, "color") ||
833            !applyFileOverlay(bundle, assets, &menus, "menu") ||
834            !applyFileOverlay(bundle, assets, &mipmaps, "mipmap")) {
835        return UNKNOWN_ERROR;
836    }
837
838    bool hasErrors = false;
839
840    if (drawables != NULL) {
841        if (bundle->getOutputAPKFile() != NULL) {
842            err = preProcessImages(bundle, assets, drawables, "drawable");
843        }
844        if (err == NO_ERROR) {
845            err = makeFileResources(bundle, assets, &table, drawables, "drawable");
846            if (err != NO_ERROR) {
847                hasErrors = true;
848            }
849        } else {
850            hasErrors = true;
851        }
852    }
853
854    if (mipmaps != NULL) {
855        if (bundle->getOutputAPKFile() != NULL) {
856            err = preProcessImages(bundle, assets, mipmaps, "mipmap");
857        }
858        if (err == NO_ERROR) {
859            err = makeFileResources(bundle, assets, &table, mipmaps, "mipmap");
860            if (err != NO_ERROR) {
861                hasErrors = true;
862            }
863        } else {
864            hasErrors = true;
865        }
866    }
867
868    if (layouts != NULL) {
869        err = makeFileResources(bundle, assets, &table, layouts, "layout");
870        if (err != NO_ERROR) {
871            hasErrors = true;
872        }
873    }
874
875    if (anims != NULL) {
876        err = makeFileResources(bundle, assets, &table, anims, "anim");
877        if (err != NO_ERROR) {
878            hasErrors = true;
879        }
880    }
881
882    if (xmls != NULL) {
883        err = makeFileResources(bundle, assets, &table, xmls, "xml");
884        if (err != NO_ERROR) {
885            hasErrors = true;
886        }
887    }
888
889    if (raws != NULL) {
890        err = makeFileResources(bundle, assets, &table, raws, "raw");
891        if (err != NO_ERROR) {
892            hasErrors = true;
893        }
894    }
895
896    // compile resources
897    current = assets;
898    while(current.get()) {
899        KeyedVector<String8, sp<ResourceTypeSet> > *resources =
900                current->getResources();
901
902        ssize_t index = resources->indexOfKey(String8("values"));
903        if (index >= 0) {
904            ResourceDirIterator it(resources->valueAt(index), String8("values"));
905            ssize_t res;
906            while ((res=it.next()) == NO_ERROR) {
907                sp<AaptFile> file = it.getFile();
908                res = compileResourceFile(bundle, assets, file, it.getParams(),
909                                          (current!=assets), &table);
910                if (res != NO_ERROR) {
911                    hasErrors = true;
912                }
913            }
914        }
915        current = current->getOverlay();
916    }
917
918    if (colors != NULL) {
919        err = makeFileResources(bundle, assets, &table, colors, "color");
920        if (err != NO_ERROR) {
921            hasErrors = true;
922        }
923    }
924
925    if (menus != NULL) {
926        err = makeFileResources(bundle, assets, &table, menus, "menu");
927        if (err != NO_ERROR) {
928            hasErrors = true;
929        }
930    }
931
932    // --------------------------------------------------------------------
933    // Assignment of resource IDs and initial generation of resource table.
934    // --------------------------------------------------------------------
935
936    if (table.hasResources()) {
937        sp<AaptFile> resFile(getResourceFile(assets));
938        if (resFile == NULL) {
939            fprintf(stderr, "Error: unable to generate entry for resource data\n");
940            return UNKNOWN_ERROR;
941        }
942
943        err = table.assignResourceIds();
944        if (err < NO_ERROR) {
945            return err;
946        }
947    }
948
949    // --------------------------------------------------------------
950    // Finally, we can now we can compile XML files, which may reference
951    // resources.
952    // --------------------------------------------------------------
953
954    if (layouts != NULL) {
955        ResourceDirIterator it(layouts, String8("layout"));
956        while ((err=it.next()) == NO_ERROR) {
957            String8 src = it.getFile()->getPrintableSource();
958            err = compileXmlFile(assets, it.getFile(), &table, xmlFlags);
959            if (err == NO_ERROR) {
960                ResXMLTree block;
961                block.setTo(it.getFile()->getData(), it.getFile()->getSize(), true);
962                checkForIds(src, block);
963            } else {
964                hasErrors = true;
965            }
966        }
967
968        if (err < NO_ERROR) {
969            hasErrors = true;
970        }
971        err = NO_ERROR;
972    }
973
974    if (anims != NULL) {
975        ResourceDirIterator it(anims, String8("anim"));
976        while ((err=it.next()) == NO_ERROR) {
977            err = compileXmlFile(assets, it.getFile(), &table, xmlFlags);
978            if (err != NO_ERROR) {
979                hasErrors = true;
980            }
981        }
982
983        if (err < NO_ERROR) {
984            hasErrors = true;
985        }
986        err = NO_ERROR;
987    }
988
989    if (xmls != NULL) {
990        ResourceDirIterator it(xmls, String8("xml"));
991        while ((err=it.next()) == NO_ERROR) {
992            err = compileXmlFile(assets, it.getFile(), &table, xmlFlags);
993            if (err != NO_ERROR) {
994                hasErrors = true;
995            }
996        }
997
998        if (err < NO_ERROR) {
999            hasErrors = true;
1000        }
1001        err = NO_ERROR;
1002    }
1003
1004    if (drawables != NULL) {
1005        err = postProcessImages(assets, &table, drawables);
1006        if (err != NO_ERROR) {
1007            hasErrors = true;
1008        }
1009    }
1010
1011    if (colors != NULL) {
1012        ResourceDirIterator it(colors, String8("color"));
1013        while ((err=it.next()) == NO_ERROR) {
1014          err = compileXmlFile(assets, it.getFile(), &table, xmlFlags);
1015            if (err != NO_ERROR) {
1016                hasErrors = true;
1017            }
1018        }
1019
1020        if (err < NO_ERROR) {
1021            hasErrors = true;
1022        }
1023        err = NO_ERROR;
1024    }
1025
1026    if (menus != NULL) {
1027        ResourceDirIterator it(menus, String8("menu"));
1028        while ((err=it.next()) == NO_ERROR) {
1029            String8 src = it.getFile()->getPrintableSource();
1030            err = compileXmlFile(assets, it.getFile(), &table, xmlFlags);
1031            if (err != NO_ERROR) {
1032                hasErrors = true;
1033            }
1034            ResXMLTree block;
1035            block.setTo(it.getFile()->getData(), it.getFile()->getSize(), true);
1036            checkForIds(src, block);
1037        }
1038
1039        if (err < NO_ERROR) {
1040            hasErrors = true;
1041        }
1042        err = NO_ERROR;
1043    }
1044
1045    if (table.validateLocalizations()) {
1046        hasErrors = true;
1047    }
1048
1049    if (hasErrors) {
1050        return UNKNOWN_ERROR;
1051    }
1052
1053    const sp<AaptFile> manifestFile(androidManifestFile->getFiles().valueAt(0));
1054    String8 manifestPath(manifestFile->getPrintableSource());
1055
1056    // Generate final compiled manifest file.
1057    manifestFile->clearData();
1058    sp<XMLNode> manifestTree = XMLNode::parse(manifestFile);
1059    if (manifestTree == NULL) {
1060        return UNKNOWN_ERROR;
1061    }
1062    err = massageManifest(bundle, manifestTree);
1063    if (err < NO_ERROR) {
1064        return err;
1065    }
1066    err = compileXmlFile(assets, manifestTree, manifestFile, &table);
1067    if (err < NO_ERROR) {
1068        return err;
1069    }
1070
1071    //block.restart();
1072    //printXMLBlock(&block);
1073
1074    // --------------------------------------------------------------
1075    // Generate the final resource table.
1076    // Re-flatten because we may have added new resource IDs
1077    // --------------------------------------------------------------
1078
1079    ResTable finalResTable;
1080    sp<AaptFile> resFile;
1081
1082    if (table.hasResources()) {
1083        sp<AaptSymbols> symbols = assets->getSymbolsFor(String8("R"));
1084        err = table.addSymbols(symbols);
1085        if (err < NO_ERROR) {
1086            return err;
1087        }
1088
1089        resFile = getResourceFile(assets);
1090        if (resFile == NULL) {
1091            fprintf(stderr, "Error: unable to generate entry for resource data\n");
1092            return UNKNOWN_ERROR;
1093        }
1094
1095        err = table.flatten(bundle, resFile);
1096        if (err < NO_ERROR) {
1097            return err;
1098        }
1099
1100        if (bundle->getPublicOutputFile()) {
1101            FILE* fp = fopen(bundle->getPublicOutputFile(), "w+");
1102            if (fp == NULL) {
1103                fprintf(stderr, "ERROR: Unable to open public definitions output file %s: %s\n",
1104                        (const char*)bundle->getPublicOutputFile(), strerror(errno));
1105                return UNKNOWN_ERROR;
1106            }
1107            if (bundle->getVerbose()) {
1108                printf("  Writing public definitions to %s.\n", bundle->getPublicOutputFile());
1109            }
1110            table.writePublicDefinitions(String16(assets->getPackage()), fp);
1111            fclose(fp);
1112        }
1113
1114        // Read resources back in,
1115        finalResTable.add(resFile->getData(), resFile->getSize(), NULL);
1116
1117#if 0
1118        NOISY(
1119              printf("Generated resources:\n");
1120              finalResTable.print();
1121        )
1122#endif
1123    }
1124
1125    // Perform a basic validation of the manifest file.  This time we
1126    // parse it with the comments intact, so that we can use them to
1127    // generate java docs...  so we are not going to write this one
1128    // back out to the final manifest data.
1129    sp<AaptFile> outManifestFile = new AaptFile(manifestFile->getSourceFile(),
1130            manifestFile->getGroupEntry(),
1131            manifestFile->getResourceType());
1132    err = compileXmlFile(assets, manifestFile,
1133            outManifestFile, &table,
1134            XML_COMPILE_ASSIGN_ATTRIBUTE_IDS
1135            | XML_COMPILE_STRIP_WHITESPACE | XML_COMPILE_STRIP_RAW_VALUES);
1136    if (err < NO_ERROR) {
1137        return err;
1138    }
1139    ResXMLTree block;
1140    block.setTo(outManifestFile->getData(), outManifestFile->getSize(), true);
1141    String16 manifest16("manifest");
1142    String16 permission16("permission");
1143    String16 permission_group16("permission-group");
1144    String16 uses_permission16("uses-permission");
1145    String16 instrumentation16("instrumentation");
1146    String16 application16("application");
1147    String16 provider16("provider");
1148    String16 service16("service");
1149    String16 receiver16("receiver");
1150    String16 activity16("activity");
1151    String16 action16("action");
1152    String16 category16("category");
1153    String16 data16("scheme");
1154    const char* packageIdentChars = "abcdefghijklmnopqrstuvwxyz"
1155        "ABCDEFGHIJKLMNOPQRSTUVWXYZ._0123456789";
1156    const char* packageIdentCharsWithTheStupid = "abcdefghijklmnopqrstuvwxyz"
1157        "ABCDEFGHIJKLMNOPQRSTUVWXYZ._0123456789-";
1158    const char* classIdentChars = "abcdefghijklmnopqrstuvwxyz"
1159        "ABCDEFGHIJKLMNOPQRSTUVWXYZ._0123456789$";
1160    const char* processIdentChars = "abcdefghijklmnopqrstuvwxyz"
1161        "ABCDEFGHIJKLMNOPQRSTUVWXYZ._0123456789:";
1162    const char* authoritiesIdentChars = "abcdefghijklmnopqrstuvwxyz"
1163        "ABCDEFGHIJKLMNOPQRSTUVWXYZ._0123456789-:;";
1164    const char* typeIdentChars = "abcdefghijklmnopqrstuvwxyz"
1165        "ABCDEFGHIJKLMNOPQRSTUVWXYZ._0123456789:-/*+";
1166    const char* schemeIdentChars = "abcdefghijklmnopqrstuvwxyz"
1167        "ABCDEFGHIJKLMNOPQRSTUVWXYZ._0123456789-";
1168    ResXMLTree::event_code_t code;
1169    sp<AaptSymbols> permissionSymbols;
1170    sp<AaptSymbols> permissionGroupSymbols;
1171    while ((code=block.next()) != ResXMLTree::END_DOCUMENT
1172           && code > ResXMLTree::BAD_DOCUMENT) {
1173        if (code == ResXMLTree::START_TAG) {
1174            size_t len;
1175            if (block.getElementNamespace(&len) != NULL) {
1176                continue;
1177            }
1178            if (strcmp16(block.getElementName(&len), manifest16.string()) == 0) {
1179                if (validateAttr(manifestPath, finalResTable, block, NULL, "package",
1180                                 packageIdentChars, true) != ATTR_OKAY) {
1181                    hasErrors = true;
1182                }
1183                if (validateAttr(manifestPath, finalResTable, block, RESOURCES_ANDROID_NAMESPACE,
1184                                 "sharedUserId", packageIdentChars, false) != ATTR_OKAY) {
1185                    hasErrors = true;
1186                }
1187            } else if (strcmp16(block.getElementName(&len), permission16.string()) == 0
1188                    || strcmp16(block.getElementName(&len), permission_group16.string()) == 0) {
1189                const bool isGroup = strcmp16(block.getElementName(&len),
1190                        permission_group16.string()) == 0;
1191                if (validateAttr(manifestPath, finalResTable, block, RESOURCES_ANDROID_NAMESPACE,
1192                                 "name", isGroup ? packageIdentCharsWithTheStupid
1193                                 : packageIdentChars, true) != ATTR_OKAY) {
1194                    hasErrors = true;
1195                }
1196                SourcePos srcPos(manifestPath, block.getLineNumber());
1197                sp<AaptSymbols> syms;
1198                if (!isGroup) {
1199                    syms = permissionSymbols;
1200                    if (syms == NULL) {
1201                        sp<AaptSymbols> symbols =
1202                                assets->getSymbolsFor(String8("Manifest"));
1203                        syms = permissionSymbols = symbols->addNestedSymbol(
1204                                String8("permission"), srcPos);
1205                    }
1206                } else {
1207                    syms = permissionGroupSymbols;
1208                    if (syms == NULL) {
1209                        sp<AaptSymbols> symbols =
1210                                assets->getSymbolsFor(String8("Manifest"));
1211                        syms = permissionGroupSymbols = symbols->addNestedSymbol(
1212                                String8("permission_group"), srcPos);
1213                    }
1214                }
1215                size_t len;
1216                ssize_t index = block.indexOfAttribute(RESOURCES_ANDROID_NAMESPACE, "name");
1217                const uint16_t* id = block.getAttributeStringValue(index, &len);
1218                if (id == NULL) {
1219                    fprintf(stderr, "%s:%d: missing name attribute in element <%s>.\n",
1220                            manifestPath.string(), block.getLineNumber(),
1221                            String8(block.getElementName(&len)).string());
1222                    hasErrors = true;
1223                    break;
1224                }
1225                String8 idStr(id);
1226                char* p = idStr.lockBuffer(idStr.size());
1227                char* e = p + idStr.size();
1228                bool begins_with_digit = true;  // init to true so an empty string fails
1229                while (e > p) {
1230                    e--;
1231                    if (*e >= '0' && *e <= '9') {
1232                      begins_with_digit = true;
1233                      continue;
1234                    }
1235                    if ((*e >= 'a' && *e <= 'z') ||
1236                        (*e >= 'A' && *e <= 'Z') ||
1237                        (*e == '_')) {
1238                      begins_with_digit = false;
1239                      continue;
1240                    }
1241                    if (isGroup && (*e == '-')) {
1242                        *e = '_';
1243                        begins_with_digit = false;
1244                        continue;
1245                    }
1246                    e++;
1247                    break;
1248                }
1249                idStr.unlockBuffer();
1250                // verify that we stopped because we hit a period or
1251                // the beginning of the string, and that the
1252                // identifier didn't begin with a digit.
1253                if (begins_with_digit || (e != p && *(e-1) != '.')) {
1254                  fprintf(stderr,
1255                          "%s:%d: Permission name <%s> is not a valid Java symbol\n",
1256                          manifestPath.string(), block.getLineNumber(), idStr.string());
1257                  hasErrors = true;
1258                }
1259                syms->addStringSymbol(String8(e), idStr, srcPos);
1260                const uint16_t* cmt = block.getComment(&len);
1261                if (cmt != NULL && *cmt != 0) {
1262                    //printf("Comment of %s: %s\n", String8(e).string(),
1263                    //        String8(cmt).string());
1264                    syms->appendComment(String8(e), String16(cmt), srcPos);
1265                } else {
1266                    //printf("No comment for %s\n", String8(e).string());
1267                }
1268                syms->makeSymbolPublic(String8(e), srcPos);
1269            } else if (strcmp16(block.getElementName(&len), uses_permission16.string()) == 0) {
1270                if (validateAttr(manifestPath, finalResTable, block, RESOURCES_ANDROID_NAMESPACE,
1271                                 "name", packageIdentChars, true) != ATTR_OKAY) {
1272                    hasErrors = true;
1273                }
1274            } else if (strcmp16(block.getElementName(&len), instrumentation16.string()) == 0) {
1275                if (validateAttr(manifestPath, finalResTable, block, RESOURCES_ANDROID_NAMESPACE,
1276                                 "name", classIdentChars, true) != ATTR_OKAY) {
1277                    hasErrors = true;
1278                }
1279                if (validateAttr(manifestPath, finalResTable, block,
1280                                 RESOURCES_ANDROID_NAMESPACE, "targetPackage",
1281                                 packageIdentChars, true) != ATTR_OKAY) {
1282                    hasErrors = true;
1283                }
1284            } else if (strcmp16(block.getElementName(&len), application16.string()) == 0) {
1285                if (validateAttr(manifestPath, finalResTable, block, RESOURCES_ANDROID_NAMESPACE,
1286                                 "name", classIdentChars, false) != ATTR_OKAY) {
1287                    hasErrors = true;
1288                }
1289                if (validateAttr(manifestPath, finalResTable, block,
1290                                 RESOURCES_ANDROID_NAMESPACE, "permission",
1291                                 packageIdentChars, false) != ATTR_OKAY) {
1292                    hasErrors = true;
1293                }
1294                if (validateAttr(manifestPath, finalResTable, block,
1295                                 RESOURCES_ANDROID_NAMESPACE, "process",
1296                                 processIdentChars, false) != ATTR_OKAY) {
1297                    hasErrors = true;
1298                }
1299                if (validateAttr(manifestPath, finalResTable, block,
1300                                 RESOURCES_ANDROID_NAMESPACE, "taskAffinity",
1301                                 processIdentChars, false) != ATTR_OKAY) {
1302                    hasErrors = true;
1303                }
1304            } else if (strcmp16(block.getElementName(&len), provider16.string()) == 0) {
1305                if (validateAttr(manifestPath, finalResTable, block, RESOURCES_ANDROID_NAMESPACE,
1306                                 "name", classIdentChars, true) != ATTR_OKAY) {
1307                    hasErrors = true;
1308                }
1309                if (validateAttr(manifestPath, finalResTable, block,
1310                                 RESOURCES_ANDROID_NAMESPACE, "authorities",
1311                                 authoritiesIdentChars, true) != ATTR_OKAY) {
1312                    hasErrors = true;
1313                }
1314                if (validateAttr(manifestPath, finalResTable, block,
1315                                 RESOURCES_ANDROID_NAMESPACE, "permission",
1316                                 packageIdentChars, false) != ATTR_OKAY) {
1317                    hasErrors = true;
1318                }
1319                if (validateAttr(manifestPath, finalResTable, block,
1320                                 RESOURCES_ANDROID_NAMESPACE, "process",
1321                                 processIdentChars, false) != ATTR_OKAY) {
1322                    hasErrors = true;
1323                }
1324            } else if (strcmp16(block.getElementName(&len), service16.string()) == 0
1325                       || strcmp16(block.getElementName(&len), receiver16.string()) == 0
1326                       || strcmp16(block.getElementName(&len), activity16.string()) == 0) {
1327                if (validateAttr(manifestPath, finalResTable, block, RESOURCES_ANDROID_NAMESPACE,
1328                                 "name", classIdentChars, true) != ATTR_OKAY) {
1329                    hasErrors = true;
1330                }
1331                if (validateAttr(manifestPath, finalResTable, block,
1332                                 RESOURCES_ANDROID_NAMESPACE, "permission",
1333                                 packageIdentChars, false) != ATTR_OKAY) {
1334                    hasErrors = true;
1335                }
1336                if (validateAttr(manifestPath, finalResTable, block,
1337                                 RESOURCES_ANDROID_NAMESPACE, "process",
1338                                 processIdentChars, false) != ATTR_OKAY) {
1339                    hasErrors = true;
1340                }
1341                if (validateAttr(manifestPath, finalResTable, block,
1342                                 RESOURCES_ANDROID_NAMESPACE, "taskAffinity",
1343                                 processIdentChars, false) != ATTR_OKAY) {
1344                    hasErrors = true;
1345                }
1346            } else if (strcmp16(block.getElementName(&len), action16.string()) == 0
1347                       || strcmp16(block.getElementName(&len), category16.string()) == 0) {
1348                if (validateAttr(manifestPath, finalResTable, block,
1349                                 RESOURCES_ANDROID_NAMESPACE, "name",
1350                                 packageIdentChars, true) != ATTR_OKAY) {
1351                    hasErrors = true;
1352                }
1353            } else if (strcmp16(block.getElementName(&len), data16.string()) == 0) {
1354                if (validateAttr(manifestPath, finalResTable, block,
1355                                 RESOURCES_ANDROID_NAMESPACE, "mimeType",
1356                                 typeIdentChars, true) != ATTR_OKAY) {
1357                    hasErrors = true;
1358                }
1359                if (validateAttr(manifestPath, finalResTable, block,
1360                                 RESOURCES_ANDROID_NAMESPACE, "scheme",
1361                                 schemeIdentChars, true) != ATTR_OKAY) {
1362                    hasErrors = true;
1363                }
1364            }
1365        }
1366    }
1367
1368    if (resFile != NULL) {
1369        // These resources are now considered to be a part of the included
1370        // resources, for others to reference.
1371        err = assets->addIncludedResources(resFile);
1372        if (err < NO_ERROR) {
1373            fprintf(stderr, "ERROR: Unable to parse generated resources, aborting.\n");
1374            return err;
1375        }
1376    }
1377
1378    return err;
1379}
1380
1381static const char* getIndentSpace(int indent)
1382{
1383static const char whitespace[] =
1384"                                                                                       ";
1385
1386    return whitespace + sizeof(whitespace) - 1 - indent*4;
1387}
1388
1389static status_t fixupSymbol(String16* inoutSymbol)
1390{
1391    inoutSymbol->replaceAll('.', '_');
1392    inoutSymbol->replaceAll(':', '_');
1393    return NO_ERROR;
1394}
1395
1396static String16 getAttributeComment(const sp<AaptAssets>& assets,
1397                                    const String8& name,
1398                                    String16* outTypeComment = NULL)
1399{
1400    sp<AaptSymbols> asym = assets->getSymbolsFor(String8("R"));
1401    if (asym != NULL) {
1402        //printf("Got R symbols!\n");
1403        asym = asym->getNestedSymbols().valueFor(String8("attr"));
1404        if (asym != NULL) {
1405            //printf("Got attrs symbols! comment %s=%s\n",
1406            //     name.string(), String8(asym->getComment(name)).string());
1407            if (outTypeComment != NULL) {
1408                *outTypeComment = asym->getTypeComment(name);
1409            }
1410            return asym->getComment(name);
1411        }
1412    }
1413    return String16();
1414}
1415
1416static status_t writeLayoutClasses(
1417    FILE* fp, const sp<AaptAssets>& assets,
1418    const sp<AaptSymbols>& symbols, int indent, bool includePrivate)
1419{
1420    const char* indentStr = getIndentSpace(indent);
1421    if (!includePrivate) {
1422        fprintf(fp, "%s/** @doconly */\n", indentStr);
1423    }
1424    fprintf(fp, "%spublic static final class styleable {\n", indentStr);
1425    indent++;
1426
1427    String16 attr16("attr");
1428    String16 package16(assets->getPackage());
1429
1430    indentStr = getIndentSpace(indent);
1431    bool hasErrors = false;
1432
1433    size_t i;
1434    size_t N = symbols->getNestedSymbols().size();
1435    for (i=0; i<N; i++) {
1436        sp<AaptSymbols> nsymbols = symbols->getNestedSymbols().valueAt(i);
1437        String16 nclassName16(symbols->getNestedSymbols().keyAt(i));
1438        String8 realClassName(nclassName16);
1439        if (fixupSymbol(&nclassName16) != NO_ERROR) {
1440            hasErrors = true;
1441        }
1442        String8 nclassName(nclassName16);
1443
1444        SortedVector<uint32_t> idents;
1445        Vector<uint32_t> origOrder;
1446        Vector<bool> publicFlags;
1447
1448        size_t a;
1449        size_t NA = nsymbols->getSymbols().size();
1450        for (a=0; a<NA; a++) {
1451            const AaptSymbolEntry& sym(nsymbols->getSymbols().valueAt(a));
1452            int32_t code = sym.typeCode == AaptSymbolEntry::TYPE_INT32
1453                    ? sym.int32Val : 0;
1454            bool isPublic = true;
1455            if (code == 0) {
1456                String16 name16(sym.name);
1457                uint32_t typeSpecFlags;
1458                code = assets->getIncludedResources().identifierForName(
1459                    name16.string(), name16.size(),
1460                    attr16.string(), attr16.size(),
1461                    package16.string(), package16.size(), &typeSpecFlags);
1462                if (code == 0) {
1463                    fprintf(stderr, "ERROR: In <declare-styleable> %s, unable to find attribute %s\n",
1464                            nclassName.string(), sym.name.string());
1465                    hasErrors = true;
1466                }
1467                isPublic = (typeSpecFlags&ResTable_typeSpec::SPEC_PUBLIC) != 0;
1468            }
1469            idents.add(code);
1470            origOrder.add(code);
1471            publicFlags.add(isPublic);
1472        }
1473
1474        NA = idents.size();
1475
1476        bool deprecated = false;
1477
1478        String16 comment = symbols->getComment(realClassName);
1479        fprintf(fp, "%s/** ", indentStr);
1480        if (comment.size() > 0) {
1481            String8 cmt(comment);
1482            fprintf(fp, "%s\n", cmt.string());
1483            if (strstr(cmt.string(), "@deprecated") != NULL) {
1484                deprecated = true;
1485            }
1486        } else {
1487            fprintf(fp, "Attributes that can be used with a %s.\n", nclassName.string());
1488        }
1489        bool hasTable = false;
1490        for (a=0; a<NA; a++) {
1491            ssize_t pos = idents.indexOf(origOrder.itemAt(a));
1492            if (pos >= 0) {
1493                if (!hasTable) {
1494                    hasTable = true;
1495                    fprintf(fp,
1496                            "%s   <p>Includes the following attributes:</p>\n"
1497                            "%s   <table>\n"
1498                            "%s   <colgroup align=\"left\" />\n"
1499                            "%s   <colgroup align=\"left\" />\n"
1500                            "%s   <tr><th>Attribute</th><th>Description</th></tr>\n",
1501                            indentStr,
1502                            indentStr,
1503                            indentStr,
1504                            indentStr,
1505                            indentStr);
1506                }
1507                const AaptSymbolEntry& sym = nsymbols->getSymbols().valueAt(a);
1508                if (!publicFlags.itemAt(a) && !includePrivate) {
1509                    continue;
1510                }
1511                String8 name8(sym.name);
1512                String16 comment(sym.comment);
1513                if (comment.size() <= 0) {
1514                    comment = getAttributeComment(assets, name8);
1515                }
1516                if (comment.size() > 0) {
1517                    const char16_t* p = comment.string();
1518                    while (*p != 0 && *p != '.') {
1519                        if (*p == '{') {
1520                            while (*p != 0 && *p != '}') {
1521                                p++;
1522                            }
1523                        } else {
1524                            p++;
1525                        }
1526                    }
1527                    if (*p == '.') {
1528                        p++;
1529                    }
1530                    comment = String16(comment.string(), p-comment.string());
1531                }
1532                String16 name(name8);
1533                fixupSymbol(&name);
1534                fprintf(fp, "%s   <tr><td><code>{@link #%s_%s %s:%s}</code></td><td>%s</td></tr>\n",
1535                        indentStr, nclassName.string(),
1536                        String8(name).string(),
1537                        assets->getPackage().string(),
1538                        String8(name).string(),
1539                        String8(comment).string());
1540            }
1541        }
1542        if (hasTable) {
1543            fprintf(fp, "%s   </table>\n", indentStr);
1544        }
1545        for (a=0; a<NA; a++) {
1546            ssize_t pos = idents.indexOf(origOrder.itemAt(a));
1547            if (pos >= 0) {
1548                const AaptSymbolEntry& sym = nsymbols->getSymbols().valueAt(a);
1549                if (!publicFlags.itemAt(a) && !includePrivate) {
1550                    continue;
1551                }
1552                String16 name(sym.name);
1553                fixupSymbol(&name);
1554                fprintf(fp, "%s   @see #%s_%s\n",
1555                        indentStr, nclassName.string(),
1556                        String8(name).string());
1557            }
1558        }
1559        fprintf(fp, "%s */\n", getIndentSpace(indent));
1560
1561        if (deprecated) {
1562            fprintf(fp, "%s@Deprecated\n", indentStr);
1563        }
1564
1565        fprintf(fp,
1566                "%spublic static final int[] %s = {\n"
1567                "%s",
1568                indentStr, nclassName.string(),
1569                getIndentSpace(indent+1));
1570
1571        for (a=0; a<NA; a++) {
1572            if (a != 0) {
1573                if ((a&3) == 0) {
1574                    fprintf(fp, ",\n%s", getIndentSpace(indent+1));
1575                } else {
1576                    fprintf(fp, ", ");
1577                }
1578            }
1579            fprintf(fp, "0x%08x", idents[a]);
1580        }
1581
1582        fprintf(fp, "\n%s};\n", indentStr);
1583
1584        for (a=0; a<NA; a++) {
1585            ssize_t pos = idents.indexOf(origOrder.itemAt(a));
1586            if (pos >= 0) {
1587                const AaptSymbolEntry& sym = nsymbols->getSymbols().valueAt(a);
1588                if (!publicFlags.itemAt(a) && !includePrivate) {
1589                    continue;
1590                }
1591                String8 name8(sym.name);
1592                String16 comment(sym.comment);
1593                String16 typeComment;
1594                if (comment.size() <= 0) {
1595                    comment = getAttributeComment(assets, name8, &typeComment);
1596                } else {
1597                    getAttributeComment(assets, name8, &typeComment);
1598                }
1599                String16 name(name8);
1600                if (fixupSymbol(&name) != NO_ERROR) {
1601                    hasErrors = true;
1602                }
1603
1604                uint32_t typeSpecFlags = 0;
1605                String16 name16(sym.name);
1606                assets->getIncludedResources().identifierForName(
1607                    name16.string(), name16.size(),
1608                    attr16.string(), attr16.size(),
1609                    package16.string(), package16.size(), &typeSpecFlags);
1610                //printf("%s:%s/%s: 0x%08x\n", String8(package16).string(),
1611                //    String8(attr16).string(), String8(name16).string(), typeSpecFlags);
1612                const bool pub = (typeSpecFlags&ResTable_typeSpec::SPEC_PUBLIC) != 0;
1613
1614                bool deprecated = false;
1615
1616                fprintf(fp, "%s/**\n", indentStr);
1617                if (comment.size() > 0) {
1618                    String8 cmt(comment);
1619                    fprintf(fp, "%s  <p>\n%s  @attr description\n", indentStr, indentStr);
1620                    fprintf(fp, "%s  %s\n", indentStr, cmt.string());
1621                    if (strstr(cmt.string(), "@deprecated") != NULL) {
1622                        deprecated = true;
1623                    }
1624                } else {
1625                    fprintf(fp,
1626                            "%s  <p>This symbol is the offset where the {@link %s.R.attr#%s}\n"
1627                            "%s  attribute's value can be found in the {@link #%s} array.\n",
1628                            indentStr,
1629                            pub ? assets->getPackage().string()
1630                                : assets->getSymbolsPrivatePackage().string(),
1631                            String8(name).string(),
1632                            indentStr, nclassName.string());
1633                }
1634                if (typeComment.size() > 0) {
1635                    String8 cmt(typeComment);
1636                    fprintf(fp, "\n\n%s  %s\n", indentStr, cmt.string());
1637                    if (strstr(cmt.string(), "@deprecated") != NULL) {
1638                        deprecated = true;
1639                    }
1640                }
1641                if (comment.size() > 0) {
1642                    if (pub) {
1643                        fprintf(fp,
1644                                "%s  <p>This corresponds to the global attribute"
1645                                "%s  resource symbol {@link %s.R.attr#%s}.\n",
1646                                indentStr, indentStr,
1647                                assets->getPackage().string(),
1648                                String8(name).string());
1649                    } else {
1650                        fprintf(fp,
1651                                "%s  <p>This is a private symbol.\n", indentStr);
1652                    }
1653                }
1654                fprintf(fp, "%s  @attr name %s:%s\n", indentStr,
1655                        "android", String8(name).string());
1656                fprintf(fp, "%s*/\n", indentStr);
1657                if (deprecated) {
1658                    fprintf(fp, "%s@Deprecated\n", indentStr);
1659                }
1660                fprintf(fp,
1661                        "%spublic static final int %s_%s = %d;\n",
1662                        indentStr, nclassName.string(),
1663                        String8(name).string(), (int)pos);
1664            }
1665        }
1666    }
1667
1668    indent--;
1669    fprintf(fp, "%s};\n", getIndentSpace(indent));
1670    return hasErrors ? UNKNOWN_ERROR : NO_ERROR;
1671}
1672
1673static status_t writeSymbolClass(
1674    FILE* fp, const sp<AaptAssets>& assets, bool includePrivate,
1675    const sp<AaptSymbols>& symbols, const String8& className, int indent)
1676{
1677    fprintf(fp, "%spublic %sfinal class %s {\n",
1678            getIndentSpace(indent),
1679            indent != 0 ? "static " : "", className.string());
1680    indent++;
1681
1682    size_t i;
1683    status_t err = NO_ERROR;
1684
1685    size_t N = symbols->getSymbols().size();
1686    for (i=0; i<N; i++) {
1687        const AaptSymbolEntry& sym = symbols->getSymbols().valueAt(i);
1688        if (sym.typeCode != AaptSymbolEntry::TYPE_INT32) {
1689            continue;
1690        }
1691        if (!includePrivate && !sym.isPublic) {
1692            continue;
1693        }
1694        String16 name(sym.name);
1695        String8 realName(name);
1696        if (fixupSymbol(&name) != NO_ERROR) {
1697            return UNKNOWN_ERROR;
1698        }
1699        String16 comment(sym.comment);
1700        bool haveComment = false;
1701        bool deprecated = false;
1702        if (comment.size() > 0) {
1703            haveComment = true;
1704            String8 cmt(comment);
1705            fprintf(fp,
1706                    "%s/** %s\n",
1707                    getIndentSpace(indent), cmt.string());
1708            if (strstr(cmt.string(), "@deprecated") != NULL) {
1709                deprecated = true;
1710            }
1711        } else if (sym.isPublic && !includePrivate) {
1712            sym.sourcePos.warning("No comment for public symbol %s:%s/%s",
1713                assets->getPackage().string(), className.string(),
1714                String8(sym.name).string());
1715        }
1716        String16 typeComment(sym.typeComment);
1717        if (typeComment.size() > 0) {
1718            String8 cmt(typeComment);
1719            if (!haveComment) {
1720                haveComment = true;
1721                fprintf(fp,
1722                        "%s/** %s\n", getIndentSpace(indent), cmt.string());
1723            } else {
1724                fprintf(fp,
1725                        "%s %s\n", getIndentSpace(indent), cmt.string());
1726            }
1727            if (strstr(cmt.string(), "@deprecated") != NULL) {
1728                deprecated = true;
1729            }
1730        }
1731        if (haveComment) {
1732            fprintf(fp,"%s */\n", getIndentSpace(indent));
1733        }
1734        if (deprecated) {
1735            fprintf(fp, "%s@Deprecated\n", getIndentSpace(indent));
1736        }
1737        fprintf(fp, "%spublic static final int %s=0x%08x;\n",
1738                getIndentSpace(indent),
1739                String8(name).string(), (int)sym.int32Val);
1740    }
1741
1742    for (i=0; i<N; i++) {
1743        const AaptSymbolEntry& sym = symbols->getSymbols().valueAt(i);
1744        if (sym.typeCode != AaptSymbolEntry::TYPE_STRING) {
1745            continue;
1746        }
1747        if (!includePrivate && !sym.isPublic) {
1748            continue;
1749        }
1750        String16 name(sym.name);
1751        if (fixupSymbol(&name) != NO_ERROR) {
1752            return UNKNOWN_ERROR;
1753        }
1754        String16 comment(sym.comment);
1755        bool deprecated = false;
1756        if (comment.size() > 0) {
1757            String8 cmt(comment);
1758            fprintf(fp,
1759                    "%s/** %s\n"
1760                     "%s */\n",
1761                    getIndentSpace(indent), cmt.string(),
1762                    getIndentSpace(indent));
1763            if (strstr(cmt.string(), "@deprecated") != NULL) {
1764                deprecated = true;
1765            }
1766        } else if (sym.isPublic && !includePrivate) {
1767            sym.sourcePos.warning("No comment for public symbol %s:%s/%s",
1768                assets->getPackage().string(), className.string(),
1769                String8(sym.name).string());
1770        }
1771        if (deprecated) {
1772            fprintf(fp, "%s@Deprecated\n", getIndentSpace(indent));
1773        }
1774        fprintf(fp, "%spublic static final String %s=\"%s\";\n",
1775                getIndentSpace(indent),
1776                String8(name).string(), sym.stringVal.string());
1777    }
1778
1779    sp<AaptSymbols> styleableSymbols;
1780
1781    N = symbols->getNestedSymbols().size();
1782    for (i=0; i<N; i++) {
1783        sp<AaptSymbols> nsymbols = symbols->getNestedSymbols().valueAt(i);
1784        String8 nclassName(symbols->getNestedSymbols().keyAt(i));
1785        if (nclassName == "styleable") {
1786            styleableSymbols = nsymbols;
1787        } else {
1788            err = writeSymbolClass(fp, assets, includePrivate, nsymbols, nclassName, indent);
1789        }
1790        if (err != NO_ERROR) {
1791            return err;
1792        }
1793    }
1794
1795    if (styleableSymbols != NULL) {
1796        err = writeLayoutClasses(fp, assets, styleableSymbols, indent, includePrivate);
1797        if (err != NO_ERROR) {
1798            return err;
1799        }
1800    }
1801
1802    indent--;
1803    fprintf(fp, "%s}\n", getIndentSpace(indent));
1804    return NO_ERROR;
1805}
1806
1807status_t writeResourceSymbols(Bundle* bundle, const sp<AaptAssets>& assets,
1808    const String8& package, bool includePrivate)
1809{
1810    if (!bundle->getRClassDir()) {
1811        return NO_ERROR;
1812    }
1813
1814    const size_t N = assets->getSymbols().size();
1815    for (size_t i=0; i<N; i++) {
1816        sp<AaptSymbols> symbols = assets->getSymbols().valueAt(i);
1817        String8 className(assets->getSymbols().keyAt(i));
1818        String8 dest(bundle->getRClassDir());
1819        if (bundle->getMakePackageDirs()) {
1820            String8 pkg(package);
1821            const char* last = pkg.string();
1822            const char* s = last-1;
1823            do {
1824                s++;
1825                if (s > last && (*s == '.' || *s == 0)) {
1826                    String8 part(last, s-last);
1827                    dest.appendPath(part);
1828#ifdef HAVE_MS_C_RUNTIME
1829                    _mkdir(dest.string());
1830#else
1831                    mkdir(dest.string(), S_IRUSR|S_IWUSR|S_IXUSR|S_IRGRP|S_IXGRP);
1832#endif
1833                    last = s+1;
1834                }
1835            } while (*s);
1836        }
1837        dest.appendPath(className);
1838        dest.append(".java");
1839        FILE* fp = fopen(dest.string(), "w+");
1840        if (fp == NULL) {
1841            fprintf(stderr, "ERROR: Unable to open class file %s: %s\n",
1842                    dest.string(), strerror(errno));
1843            return UNKNOWN_ERROR;
1844        }
1845        if (bundle->getVerbose()) {
1846            printf("  Writing symbols for class %s.\n", className.string());
1847        }
1848
1849        fprintf(fp,
1850        "/* AUTO-GENERATED FILE.  DO NOT MODIFY.\n"
1851        " *\n"
1852        " * This class was automatically generated by the\n"
1853        " * aapt tool from the resource data it found.  It\n"
1854        " * should not be modified by hand.\n"
1855        " */\n"
1856        "\n"
1857        "package %s;\n\n", package.string());
1858
1859        status_t err = writeSymbolClass(fp, assets, includePrivate, symbols, className, 0);
1860        if (err != NO_ERROR) {
1861            return err;
1862        }
1863        fclose(fp);
1864    }
1865
1866    return NO_ERROR;
1867}
1868
1869
1870
1871class ProguardKeepSet
1872{
1873public:
1874    // { rule --> { file locations } }
1875    KeyedVector<String8, SortedVector<String8> > rules;
1876
1877    void add(const String8& rule, const String8& where);
1878};
1879
1880void ProguardKeepSet::add(const String8& rule, const String8& where)
1881{
1882    ssize_t index = rules.indexOfKey(rule);
1883    if (index < 0) {
1884        index = rules.add(rule, SortedVector<String8>());
1885    }
1886    rules.editValueAt(index).add(where);
1887}
1888
1889void
1890addProguardKeepRule(ProguardKeepSet* keep, const String8& inClassName,
1891        const char* pkg, const String8& srcName, int line)
1892{
1893    String8 className(inClassName);
1894    if (pkg != NULL) {
1895        // asdf     --> package.asdf
1896        // .asdf  .a.b  --> package.asdf package.a.b
1897        // asdf.adsf --> asdf.asdf
1898        const char* p = className.string();
1899        const char* q = strchr(p, '.');
1900        if (p == q) {
1901            className = pkg;
1902            className.append(inClassName);
1903        } else if (q == NULL) {
1904            className = pkg;
1905            className.append(".");
1906            className.append(inClassName);
1907        }
1908    }
1909
1910    String8 rule("-keep class ");
1911    rule += className;
1912    rule += " { <init>(...); }";
1913
1914    String8 location("view ");
1915    location += srcName;
1916    char lineno[20];
1917    sprintf(lineno, ":%d", line);
1918    location += lineno;
1919
1920    keep->add(rule, location);
1921}
1922
1923status_t
1924writeProguardForAndroidManifest(ProguardKeepSet* keep, const sp<AaptAssets>& assets)
1925{
1926    status_t err;
1927    ResXMLTree tree;
1928    size_t len;
1929    ResXMLTree::event_code_t code;
1930    int depth = 0;
1931    bool inApplication = false;
1932    String8 error;
1933    sp<AaptGroup> assGroup;
1934    sp<AaptFile> assFile;
1935    String8 pkg;
1936
1937    // First, look for a package file to parse.  This is required to
1938    // be able to generate the resource information.
1939    assGroup = assets->getFiles().valueFor(String8("AndroidManifest.xml"));
1940    if (assGroup == NULL) {
1941        fprintf(stderr, "ERROR: No AndroidManifest.xml file found.\n");
1942        return -1;
1943    }
1944
1945    if (assGroup->getFiles().size() != 1) {
1946        fprintf(stderr, "warning: Multiple AndroidManifest.xml files found, using %s\n",
1947                assGroup->getFiles().valueAt(0)->getPrintableSource().string());
1948    }
1949
1950    assFile = assGroup->getFiles().valueAt(0);
1951
1952    err = parseXMLResource(assFile, &tree);
1953    if (err != NO_ERROR) {
1954        return err;
1955    }
1956
1957    tree.restart();
1958
1959    while ((code=tree.next()) != ResXMLTree::END_DOCUMENT && code != ResXMLTree::BAD_DOCUMENT) {
1960        if (code == ResXMLTree::END_TAG) {
1961            if (/* name == "Application" && */ depth == 2) {
1962                inApplication = false;
1963            }
1964            depth--;
1965            continue;
1966        }
1967        if (code != ResXMLTree::START_TAG) {
1968            continue;
1969        }
1970        depth++;
1971        String8 tag(tree.getElementName(&len));
1972        // printf("Depth %d tag %s\n", depth, tag.string());
1973        bool keepTag = false;
1974        if (depth == 1) {
1975            if (tag != "manifest") {
1976                fprintf(stderr, "ERROR: manifest does not start with <manifest> tag\n");
1977                return -1;
1978            }
1979            pkg = getAttribute(tree, NULL, "package", NULL);
1980        } else if (depth == 2) {
1981            if (tag == "application") {
1982                inApplication = true;
1983                keepTag = true;
1984
1985                String8 agent = getAttribute(tree, "http://schemas.android.com/apk/res/android",
1986                        "backupAgent", &error);
1987                if (agent.length() > 0) {
1988                    addProguardKeepRule(keep, agent, pkg.string(),
1989                            assFile->getPrintableSource(), tree.getLineNumber());
1990                }
1991            } else if (tag == "instrumentation") {
1992                keepTag = true;
1993            }
1994        }
1995        if (!keepTag && inApplication && depth == 3) {
1996            if (tag == "activity" || tag == "service" || tag == "receiver" || tag == "provider") {
1997                keepTag = true;
1998            }
1999        }
2000        if (keepTag) {
2001            String8 name = getAttribute(tree, "http://schemas.android.com/apk/res/android",
2002                    "name", &error);
2003            if (error != "") {
2004                fprintf(stderr, "ERROR: %s\n", error.string());
2005                return -1;
2006            }
2007            if (name.length() > 0) {
2008                addProguardKeepRule(keep, name, pkg.string(),
2009                        assFile->getPrintableSource(), tree.getLineNumber());
2010            }
2011        }
2012    }
2013
2014    return NO_ERROR;
2015}
2016
2017struct NamespaceAttributePair {
2018    const char* ns;
2019    const char* attr;
2020
2021    NamespaceAttributePair(const char* n, const char* a) : ns(n), attr(a) {}
2022    NamespaceAttributePair() : ns(NULL), attr(NULL) {}
2023};
2024
2025status_t
2026writeProguardForXml(ProguardKeepSet* keep, const sp<AaptFile>& layoutFile,
2027        const char* startTag, const KeyedVector<String8, NamespaceAttributePair>* tagAttrPairs)
2028{
2029    status_t err;
2030    ResXMLTree tree;
2031    size_t len;
2032    ResXMLTree::event_code_t code;
2033
2034    err = parseXMLResource(layoutFile, &tree);
2035    if (err != NO_ERROR) {
2036        return err;
2037    }
2038
2039    tree.restart();
2040
2041    if (startTag != NULL) {
2042        bool haveStart = false;
2043        while ((code=tree.next()) != ResXMLTree::END_DOCUMENT && code != ResXMLTree::BAD_DOCUMENT) {
2044            if (code != ResXMLTree::START_TAG) {
2045                continue;
2046            }
2047            String8 tag(tree.getElementName(&len));
2048            if (tag == startTag) {
2049                haveStart = true;
2050            }
2051            break;
2052        }
2053        if (!haveStart) {
2054            return NO_ERROR;
2055        }
2056    }
2057
2058    while ((code=tree.next()) != ResXMLTree::END_DOCUMENT && code != ResXMLTree::BAD_DOCUMENT) {
2059        if (code != ResXMLTree::START_TAG) {
2060            continue;
2061        }
2062        String8 tag(tree.getElementName(&len));
2063
2064        // If there is no '.', we'll assume that it's one of the built in names.
2065        if (strchr(tag.string(), '.')) {
2066            addProguardKeepRule(keep, tag, NULL,
2067                    layoutFile->getPrintableSource(), tree.getLineNumber());
2068        } else if (tagAttrPairs != NULL) {
2069            ssize_t tagIndex = tagAttrPairs->indexOfKey(tag);
2070            if (tagIndex >= 0) {
2071                const NamespaceAttributePair& nsAttr = tagAttrPairs->valueAt(tagIndex);
2072                ssize_t attrIndex = tree.indexOfAttribute(nsAttr.ns, nsAttr.attr);
2073                if (attrIndex < 0) {
2074                    // fprintf(stderr, "%s:%d: <%s> does not have attribute %s:%s.\n",
2075                    //        layoutFile->getPrintableSource().string(), tree.getLineNumber(),
2076                    //        tag.string(), nsAttr.ns, nsAttr.attr);
2077                } else {
2078                    size_t len;
2079                    addProguardKeepRule(keep,
2080                                        String8(tree.getAttributeStringValue(attrIndex, &len)), NULL,
2081                                        layoutFile->getPrintableSource(), tree.getLineNumber());
2082                }
2083            }
2084        }
2085    }
2086
2087    return NO_ERROR;
2088}
2089
2090static void addTagAttrPair(KeyedVector<String8, NamespaceAttributePair>* dest,
2091        const char* tag, const char* ns, const char* attr) {
2092    dest->add(String8(tag), NamespaceAttributePair(ns, attr));
2093}
2094
2095status_t
2096writeProguardForLayouts(ProguardKeepSet* keep, const sp<AaptAssets>& assets)
2097{
2098    status_t err;
2099
2100    // tag:attribute pairs that should be checked in layout files.
2101    KeyedVector<String8, NamespaceAttributePair> kLayoutTagAttrPairs;
2102    addTagAttrPair(&kLayoutTagAttrPairs, "view", NULL, "class");
2103    addTagAttrPair(&kLayoutTagAttrPairs, "fragment", NULL, "class");
2104    addTagAttrPair(&kLayoutTagAttrPairs, "fragment", RESOURCES_ANDROID_NAMESPACE, "name");
2105
2106    // tag:attribute pairs that should be checked in xml files.
2107    KeyedVector<String8, NamespaceAttributePair> kXmlTagAttrPairs;
2108    addTagAttrPair(&kXmlTagAttrPairs, "PreferenceScreen", RESOURCES_ANDROID_NAMESPACE, "fragment");
2109    addTagAttrPair(&kXmlTagAttrPairs, "header", RESOURCES_ANDROID_NAMESPACE, "fragment");
2110
2111    const Vector<sp<AaptDir> >& dirs = assets->resDirs();
2112    const size_t K = dirs.size();
2113    for (size_t k=0; k<K; k++) {
2114        const sp<AaptDir>& d = dirs.itemAt(k);
2115        const String8& dirName = d->getLeaf();
2116        const char* startTag = NULL;
2117        const KeyedVector<String8, NamespaceAttributePair>* tagAttrPairs = NULL;
2118        if ((dirName == String8("layout")) || (strncmp(dirName.string(), "layout-", 7) == 0)) {
2119            tagAttrPairs = &kLayoutTagAttrPairs;
2120        } else if ((dirName == String8("xml")) || (strncmp(dirName.string(), "xml-", 4) == 0)) {
2121            startTag = "PreferenceScreen";
2122            tagAttrPairs = &kXmlTagAttrPairs;
2123        } else {
2124            continue;
2125        }
2126
2127        const KeyedVector<String8,sp<AaptGroup> > groups = d->getFiles();
2128        const size_t N = groups.size();
2129        for (size_t i=0; i<N; i++) {
2130            const sp<AaptGroup>& group = groups.valueAt(i);
2131            const DefaultKeyedVector<AaptGroupEntry, sp<AaptFile> >& files = group->getFiles();
2132            const size_t M = files.size();
2133            for (size_t j=0; j<M; j++) {
2134                err = writeProguardForXml(keep, files.valueAt(j), startTag, tagAttrPairs);
2135                if (err < 0) {
2136                    return err;
2137                }
2138            }
2139        }
2140    }
2141    return NO_ERROR;
2142}
2143
2144status_t
2145writeProguardFile(Bundle* bundle, const sp<AaptAssets>& assets)
2146{
2147    status_t err = -1;
2148
2149    if (!bundle->getProguardFile()) {
2150        return NO_ERROR;
2151    }
2152
2153    ProguardKeepSet keep;
2154
2155    err = writeProguardForAndroidManifest(&keep, assets);
2156    if (err < 0) {
2157        return err;
2158    }
2159
2160    err = writeProguardForLayouts(&keep, assets);
2161    if (err < 0) {
2162        return err;
2163    }
2164
2165    FILE* fp = fopen(bundle->getProguardFile(), "w+");
2166    if (fp == NULL) {
2167        fprintf(stderr, "ERROR: Unable to open class file %s: %s\n",
2168                bundle->getProguardFile(), strerror(errno));
2169        return UNKNOWN_ERROR;
2170    }
2171
2172    const KeyedVector<String8, SortedVector<String8> >& rules = keep.rules;
2173    const size_t N = rules.size();
2174    for (size_t i=0; i<N; i++) {
2175        const SortedVector<String8>& locations = rules.valueAt(i);
2176        const size_t M = locations.size();
2177        for (size_t j=0; j<M; j++) {
2178            fprintf(fp, "# %s\n", locations.itemAt(j).string());
2179        }
2180        fprintf(fp, "%s\n\n", rules.keyAt(i).string());
2181    }
2182    fclose(fp);
2183
2184    return err;
2185}
2186