]> git.saurik.com Git - android/aapt.git/blame - Resource.cpp
Fix potential aapt crash when processing overlay.
[android/aapt.git] / Resource.cpp
CommitLineData
a534180c
TAOSP
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
a534180c
TAOSP
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();
2ca01a37 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",
a534180c
TAOSP
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] : '-',
2ca01a37 110 mParams.orientation, mParams.uiMode,
a534180c
TAOSP
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";
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
7dc0cd41
KR
174static status_t parsePackage(Bundle* bundle, const sp<AaptAssets>& assets,
175 const sp<AaptGroup>& grp)
a534180c
TAOSP
176{
177 if (grp->getFiles().size() != 1) {
406a85e3 178 fprintf(stderr, "warning: Multiple AndroidManifest.xml files found, using %s\n",
a534180c
TAOSP
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
7dc0cd41
KR
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) {
af85d202 224 ssize_t minSdkIndex = block.indexOfAttribute(RESOURCES_ANDROID_NAMESPACE,
7dc0cd41
KR
225 "minSdkVersion");
226 if (minSdkIndex >= 0) {
d7ce8f80
KR
227 const uint16_t* minSdk16 = block.getAttributeStringValue(minSdkIndex, &len);
228 const char* minSdk8 = strdup(String8(minSdk16).string());
229 bundle->setMinSdkVersion(minSdk8);
7dc0cd41
KR
230 }
231 }
232 }
233 }
234
a534180c
TAOSP
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)
288{
289 ResourceDirIterator it(set, String8("drawable"));
290 Vector<sp<AaptFile> > newNameFiles;
291 Vector<String8> newNamePaths;
8f940cc4 292 bool hasErrors = false;
a534180c
TAOSP
293 ssize_t res;
294 while ((res=it.next()) == NO_ERROR) {
295 res = preProcessImage(bundle, assets, it.getFile(), NULL);
8f940cc4
DS
296 if (res < NO_ERROR) {
297 hasErrors = true;
a534180c
TAOSP
298 }
299 }
300
8f940cc4 301 return (hasErrors || (res < NO_ERROR)) ? UNKNOWN_ERROR : NO_ERROR;
a534180c
TAOSP
302}
303
304status_t postProcessImages(const sp<AaptAssets>& assets,
305 ResourceTable* table,
306 const sp<ResourceTypeSet>& set)
307{
308 ResourceDirIterator it(set, String8("drawable"));
8f940cc4 309 bool hasErrors = false;
a534180c
TAOSP
310 ssize_t res;
311 while ((res=it.next()) == NO_ERROR) {
312 res = postProcessImage(assets, table, it.getFile());
8f940cc4
DS
313 if (res < NO_ERROR) {
314 hasErrors = true;
a534180c
TAOSP
315 }
316 }
317
8f940cc4 318 return (hasErrors || (res < NO_ERROR)) ? UNKNOWN_ERROR : NO_ERROR;
a534180c
TAOSP
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 ResXMLParser& parser,
383 const char* ns, const char* attr, const char* validChars, bool required)
384{
385 size_t len;
386
387 ssize_t index = parser.indexOfAttribute(ns, attr);
388 const uint16_t* str;
389 if (index >= 0 && (str=parser.getAttributeStringValue(index, &len)) != NULL) {
390 if (validChars) {
391 for (size_t i=0; i<len; i++) {
392 uint16_t c = str[i];
393 const char* p = validChars;
394 bool okay = false;
395 while (*p) {
396 if (c == *p) {
397 okay = true;
398 break;
399 }
400 p++;
401 }
402 if (!okay) {
403 fprintf(stderr, "%s:%d: Tag <%s> attribute %s has invalid character '%c'.\n",
404 path.string(), parser.getLineNumber(),
405 String8(parser.getElementName(&len)).string(), attr, (char)str[i]);
406 return (int)i;
407 }
408 }
409 }
410 if (*str == ' ') {
411 fprintf(stderr, "%s:%d: Tag <%s> attribute %s can not start with a space.\n",
412 path.string(), parser.getLineNumber(),
413 String8(parser.getElementName(&len)).string(), attr);
414 return ATTR_LEADING_SPACES;
415 }
416 if (str[len-1] == ' ') {
417 fprintf(stderr, "%s:%d: Tag <%s> attribute %s can not end with a space.\n",
418 path.string(), parser.getLineNumber(),
419 String8(parser.getElementName(&len)).string(), attr);
420 return ATTR_TRAILING_SPACES;
421 }
422 return ATTR_OKAY;
423 }
424 if (required) {
425 fprintf(stderr, "%s:%d: Tag <%s> missing required attribute %s.\n",
426 path.string(), parser.getLineNumber(),
427 String8(parser.getElementName(&len)).string(), attr);
428 return ATTR_NOT_FOUND;
429 }
430 return ATTR_OKAY;
431}
432
433static void checkForIds(const String8& path, ResXMLParser& parser)
434{
435 ResXMLTree::event_code_t code;
436 while ((code=parser.next()) != ResXMLTree::END_DOCUMENT
437 && code > ResXMLTree::BAD_DOCUMENT) {
438 if (code == ResXMLTree::START_TAG) {
439 ssize_t index = parser.indexOfAttribute(NULL, "id");
440 if (index >= 0) {
406a85e3 441 fprintf(stderr, "%s:%d: warning: found plain 'id' attribute; did you mean the new 'android:id' name?\n",
a534180c
TAOSP
442 path.string(), parser.getLineNumber());
443 }
444 }
445 }
446}
447
21c5c61a
RG
448static bool applyFileOverlay(Bundle *bundle,
449 const sp<AaptAssets>& assets,
e021d7c5 450 sp<ResourceTypeSet> *baseSet,
a534180c
TAOSP
451 const char *resType)
452{
21c5c61a
RG
453 if (bundle->getVerbose()) {
454 printf("applyFileOverlay for %s\n", resType);
455 }
456
a534180c
TAOSP
457 // Replace any base level files in this category with any found from the overlay
458 // Also add any found only in the overlay.
459 sp<AaptAssets> overlay = assets->getOverlay();
460 String8 resTypeString(resType);
dd854d4c 461
a534180c
TAOSP
462 // work through the linked list of overlays
463 while (overlay.get()) {
464 KeyedVector<String8, sp<ResourceTypeSet> >* overlayRes = overlay->getResources();
465
466 // get the overlay resources of the requested type
467 ssize_t index = overlayRes->indexOfKey(resTypeString);
468 if (index >= 0) {
469 sp<ResourceTypeSet> overlaySet = overlayRes->valueAt(index);
470
471 // for each of the resources, check for a match in the previously built
472 // non-overlay "baseset".
473 size_t overlayCount = overlaySet->size();
474 for (size_t overlayIndex=0; overlayIndex<overlayCount; overlayIndex++) {
21c5c61a
RG
475 if (bundle->getVerbose()) {
476 printf("trying overlaySet Key=%s\n",overlaySet->keyAt(overlayIndex).string());
477 }
e021d7c5
XD
478 size_t baseIndex = UNKNOWN_ERROR;
479 if (baseSet->get() != NULL) {
480 baseIndex = (*baseSet)->indexOfKey(overlaySet->keyAt(overlayIndex));
481 }
dd854d4c 482 if (baseIndex < UNKNOWN_ERROR) {
a534180c
TAOSP
483 // look for same flavor. For a given file (strings.xml, for example)
484 // there may be a locale specific or other flavors - we want to match
485 // the same flavor.
486 sp<AaptGroup> overlayGroup = overlaySet->valueAt(overlayIndex);
e021d7c5 487 sp<AaptGroup> baseGroup = (*baseSet)->valueAt(baseIndex);
21c5c61a
RG
488
489 DefaultKeyedVector<AaptGroupEntry, sp<AaptFile> > overlayFiles =
a534180c 490 overlayGroup->getFiles();
21c5c61a
RG
491 if (bundle->getVerbose()) {
492 DefaultKeyedVector<AaptGroupEntry, sp<AaptFile> > baseFiles =
493 baseGroup->getFiles();
494 for (size_t i=0; i < baseFiles.size(); i++) {
094e8965 495 printf("baseFile %ld has flavor %s\n", i,
21c5c61a
RG
496 baseFiles.keyAt(i).toString().string());
497 }
498 for (size_t i=0; i < overlayFiles.size(); i++) {
094e8965 499 printf("overlayFile %ld has flavor %s\n", i,
21c5c61a
RG
500 overlayFiles.keyAt(i).toString().string());
501 }
502 }
503
a534180c 504 size_t overlayGroupSize = overlayFiles.size();
21c5c61a
RG
505 for (size_t overlayGroupIndex = 0;
506 overlayGroupIndex<overlayGroupSize;
a534180c 507 overlayGroupIndex++) {
21c5c61a
RG
508 size_t baseFileIndex =
509 baseGroup->getFiles().indexOfKey(overlayFiles.
510 keyAt(overlayGroupIndex));
a534180c 511 if(baseFileIndex < UNKNOWN_ERROR) {
21c5c61a 512 if (bundle->getVerbose()) {
094e8965 513 printf("found a match (%ld) for overlay file %s, for flavor %s\n",
21c5c61a
RG
514 baseFileIndex,
515 overlayGroup->getLeaf().string(),
516 overlayFiles.keyAt(overlayGroupIndex).toString().string());
517 }
a534180c
TAOSP
518 baseGroup->removeFile(baseFileIndex);
519 } else {
520 // didn't find a match fall through and add it..
521 }
522 baseGroup->addFile(overlayFiles.valueAt(overlayGroupIndex));
a76631e8 523 assets->addGroupEntry(overlayFiles.keyAt(overlayGroupIndex));
a534180c
TAOSP
524 }
525 } else {
e021d7c5
XD
526 if (baseSet->get() == NULL) {
527 *baseSet = new ResourceTypeSet();
528 assets->getResources()->add(String8(resType), *baseSet);
529 }
a534180c 530 // this group doesn't exist (a file that's only in the overlay)
e021d7c5 531 (*baseSet)->add(overlaySet->keyAt(overlayIndex),
a8c9cd56 532 overlaySet->valueAt(overlayIndex));
a76631e8
DH
533 // make sure all flavors are defined in the resources.
534 sp<AaptGroup> overlayGroup = overlaySet->valueAt(overlayIndex);
21c5c61a 535 DefaultKeyedVector<AaptGroupEntry, sp<AaptFile> > overlayFiles =
a76631e8
DH
536 overlayGroup->getFiles();
537 size_t overlayGroupSize = overlayFiles.size();
21c5c61a
RG
538 for (size_t overlayGroupIndex = 0;
539 overlayGroupIndex<overlayGroupSize;
a76631e8
DH
540 overlayGroupIndex++) {
541 assets->addGroupEntry(overlayFiles.keyAt(overlayGroupIndex));
542 }
a534180c
TAOSP
543 }
544 }
545 // this overlay didn't have resources for this type
546 }
547 // try next overlay
548 overlay = overlay->getOverlay();
549 }
dd854d4c 550 return true;
a534180c
TAOSP
551}
552
e942a5c2
DH
553void addTagAttribute(const sp<XMLNode>& node, const char* ns8,
554 const char* attr8, const char* value)
555{
556 if (value == NULL) {
557 return;
558 }
559
560 const String16 ns(ns8);
561 const String16 attr(attr8);
562
563 if (node->getAttribute(ns, attr) != NULL) {
564 fprintf(stderr, "Warning: AndroidManifest.xml already defines %s (in %s)\n",
565 String8(attr).string(), String8(ns).string());
566 return;
567 }
568
569 node->addAttribute(ns, attr, String16(value));
570}
571
af945cf3
DH
572static void fullyQualifyClassName(const String8& package, sp<XMLNode> node,
573 const String16& attrName) {
094e8965 574 XMLNode::attribute_entry* attr = node->editAttribute(
af945cf3 575 String16("http://schemas.android.com/apk/res/android"), attrName);
094e8965
JH
576 if (attr != NULL) {
577 String8 name(attr->string);
578
579 // asdf --> package.asdf
580 // .asdf .a.b --> package.asdf package.a.b
581 // asdf.adsf --> asdf.asdf
582 String8 className;
583 const char* p = name.string();
584 const char* q = strchr(p, '.');
585 if (p == q) {
586 className += package;
587 className += name;
588 } else if (q == NULL) {
589 className += package;
590 className += ".";
591 className += name;
592 } else {
593 className += name;
594 }
595 NOISY(printf("Qualifying class '%s' to '%s'", name.string(), className.string()));
596 attr->string.setTo(String16(className));
597 }
598}
599
e942a5c2
DH
600status_t massageManifest(Bundle* bundle, sp<XMLNode> root)
601{
602 root = root->searchElement(String16(), String16("manifest"));
603 if (root == NULL) {
604 fprintf(stderr, "No <manifest> tag.\n");
605 return UNKNOWN_ERROR;
606 }
607
608 addTagAttribute(root, RESOURCES_ANDROID_NAMESPACE, "versionCode",
609 bundle->getVersionCode());
610 addTagAttribute(root, RESOURCES_ANDROID_NAMESPACE, "versionName",
611 bundle->getVersionName());
612
613 if (bundle->getMinSdkVersion() != NULL
614 || bundle->getTargetSdkVersion() != NULL
615 || bundle->getMaxSdkVersion() != NULL) {
616 sp<XMLNode> vers = root->getChildElement(String16(), String16("uses-sdk"));
617 if (vers == NULL) {
618 vers = XMLNode::newElement(root->getFilename(), String16(), String16("uses-sdk"));
619 root->insertChildAt(vers, 0);
620 }
621
622 addTagAttribute(vers, RESOURCES_ANDROID_NAMESPACE, "minSdkVersion",
623 bundle->getMinSdkVersion());
624 addTagAttribute(vers, RESOURCES_ANDROID_NAMESPACE, "targetSdkVersion",
625 bundle->getTargetSdkVersion());
626 addTagAttribute(vers, RESOURCES_ANDROID_NAMESPACE, "maxSdkVersion",
627 bundle->getMaxSdkVersion());
628 }
094e8965
JH
629
630 // Deal with manifest package name overrides
631 const char* manifestPackageNameOverride = bundle->getManifestPackageNameOverride();
632 if (manifestPackageNameOverride != NULL) {
633 // Update the actual package name
634 XMLNode::attribute_entry* attr = root->editAttribute(String16(), String16("package"));
635 if (attr == NULL) {
636 fprintf(stderr, "package name is required with --rename-manifest-package.\n");
637 return UNKNOWN_ERROR;
638 }
639 String8 origPackage(attr->string);
640 attr->string.setTo(String16(manifestPackageNameOverride));
641 NOISY(printf("Overriding package '%s' to be '%s'\n", origPackage.string(), manifestPackageNameOverride));
642
643 // Make class names fully qualified
644 sp<XMLNode> application = root->getChildElement(String16(), String16("application"));
645 if (application != NULL) {
af945cf3 646 fullyQualifyClassName(origPackage, application, String16("name"));
4fd67e1d 647 fullyQualifyClassName(origPackage, application, String16("backupAgent"));
094e8965
JH
648
649 Vector<sp<XMLNode> >& children = const_cast<Vector<sp<XMLNode> >&>(application->getChildren());
650 for (size_t i = 0; i < children.size(); i++) {
651 sp<XMLNode> child = children.editItemAt(i);
652 String8 tag(child->getElementName());
653 if (tag == "activity" || tag == "service" || tag == "receiver" || tag == "provider") {
af945cf3
DH
654 fullyQualifyClassName(origPackage, child, String16("name"));
655 } else if (tag == "activity-alias") {
656 fullyQualifyClassName(origPackage, child, String16("name"));
657 fullyQualifyClassName(origPackage, child, String16("targetActivity"));
094e8965
JH
658 }
659 }
660 }
661 }
662
af945cf3
DH
663 // Deal with manifest package name overrides
664 const char* instrumentationPackageNameOverride = bundle->getInstrumentationPackageNameOverride();
665 if (instrumentationPackageNameOverride != NULL) {
666 // Fix up instrumentation targets.
667 Vector<sp<XMLNode> >& children = const_cast<Vector<sp<XMLNode> >&>(root->getChildren());
668 for (size_t i = 0; i < children.size(); i++) {
669 sp<XMLNode> child = children.editItemAt(i);
670 String8 tag(child->getElementName());
671 if (tag == "instrumentation") {
672 XMLNode::attribute_entry* attr = child->editAttribute(
673 String16("http://schemas.android.com/apk/res/android"), String16("targetPackage"));
674 if (attr != NULL) {
675 attr->string.setTo(String16(instrumentationPackageNameOverride));
676 }
677 }
678 }
679 }
680
e942a5c2
DH
681 return NO_ERROR;
682}
683
a534180c
TAOSP
684#define ASSIGN_IT(n) \
685 do { \
686 ssize_t index = resources->indexOfKey(String8(#n)); \
687 if (index >= 0) { \
688 n ## s = resources->valueAt(index); \
689 } \
690 } while (0)
691
692status_t buildResources(Bundle* bundle, const sp<AaptAssets>& assets)
693{
694 // First, look for a package file to parse. This is required to
695 // be able to generate the resource information.
696 sp<AaptGroup> androidManifestFile =
697 assets->getFiles().valueFor(String8("AndroidManifest.xml"));
698 if (androidManifestFile == NULL) {
699 fprintf(stderr, "ERROR: No AndroidManifest.xml file found.\n");
700 return UNKNOWN_ERROR;
701 }
702
7dc0cd41 703 status_t err = parsePackage(bundle, assets, androidManifestFile);
a534180c
TAOSP
704 if (err != NO_ERROR) {
705 return err;
706 }
707
708 NOISY(printf("Creating resources for package %s\n",
709 assets->getPackage().string()));
710
711 ResourceTable table(bundle, String16(assets->getPackage()));
712 err = table.addIncludedResources(bundle, assets);
713 if (err != NO_ERROR) {
714 return err;
715 }
716
717 NOISY(printf("Found %d included resource packages\n", (int)table.size()));
718
15c62a5b
KR
719 // Standard flags for compiled XML and optional UTF-8 encoding
720 int xmlFlags = XML_COMPILE_STANDARD_RESOURCE;
721 if (bundle->getUTF8()) {
722 xmlFlags |= XML_COMPILE_UTF8;
723 }
724
a534180c
TAOSP
725 // --------------------------------------------------------------
726 // First, gather all resource information.
727 // --------------------------------------------------------------
728
729 // resType -> leafName -> group
730 KeyedVector<String8, sp<ResourceTypeSet> > *resources =
731 new KeyedVector<String8, sp<ResourceTypeSet> >;
732 collect_files(assets, resources);
733
734 sp<ResourceTypeSet> drawables;
735 sp<ResourceTypeSet> layouts;
736 sp<ResourceTypeSet> anims;
737 sp<ResourceTypeSet> xmls;
738 sp<ResourceTypeSet> raws;
739 sp<ResourceTypeSet> colors;
740 sp<ResourceTypeSet> menus;
741
742 ASSIGN_IT(drawable);
743 ASSIGN_IT(layout);
744 ASSIGN_IT(anim);
745 ASSIGN_IT(xml);
746 ASSIGN_IT(raw);
747 ASSIGN_IT(color);
748 ASSIGN_IT(menu);
749
750 assets->setResources(resources);
751 // now go through any resource overlays and collect their files
752 sp<AaptAssets> current = assets->getOverlay();
753 while(current.get()) {
754 KeyedVector<String8, sp<ResourceTypeSet> > *resources =
755 new KeyedVector<String8, sp<ResourceTypeSet> >;
756 current->setResources(resources);
757 collect_files(current, resources);
758 current = current->getOverlay();
759 }
760 // apply the overlay files to the base set
e021d7c5
XD
761 if (!applyFileOverlay(bundle, assets, &drawables, "drawable") ||
762 !applyFileOverlay(bundle, assets, &layouts, "layout") ||
763 !applyFileOverlay(bundle, assets, &anims, "anim") ||
764 !applyFileOverlay(bundle, assets, &xmls, "xml") ||
765 !applyFileOverlay(bundle, assets, &raws, "raw") ||
766 !applyFileOverlay(bundle, assets, &colors, "color") ||
767 !applyFileOverlay(bundle, assets, &menus, "menu")) {
dd854d4c
RG
768 return UNKNOWN_ERROR;
769 }
a534180c
TAOSP
770
771 bool hasErrors = false;
772
773 if (drawables != NULL) {
774 err = preProcessImages(bundle, assets, drawables);
775 if (err == NO_ERROR) {
776 err = makeFileResources(bundle, assets, &table, drawables, "drawable");
777 if (err != NO_ERROR) {
778 hasErrors = true;
779 }
780 } else {
781 hasErrors = true;
782 }
783 }
784
785 if (layouts != NULL) {
786 err = makeFileResources(bundle, assets, &table, layouts, "layout");
787 if (err != NO_ERROR) {
788 hasErrors = true;
789 }
790 }
791
792 if (anims != NULL) {
793 err = makeFileResources(bundle, assets, &table, anims, "anim");
794 if (err != NO_ERROR) {
795 hasErrors = true;
796 }
797 }
798
799 if (xmls != NULL) {
800 err = makeFileResources(bundle, assets, &table, xmls, "xml");
801 if (err != NO_ERROR) {
802 hasErrors = true;
803 }
804 }
805
806 if (raws != NULL) {
807 err = makeFileResources(bundle, assets, &table, raws, "raw");
808 if (err != NO_ERROR) {
809 hasErrors = true;
810 }
811 }
812
813 // compile resources
814 current = assets;
815 while(current.get()) {
816 KeyedVector<String8, sp<ResourceTypeSet> > *resources =
817 current->getResources();
818
819 ssize_t index = resources->indexOfKey(String8("values"));
820 if (index >= 0) {
821 ResourceDirIterator it(resources->valueAt(index), String8("values"));
822 ssize_t res;
823 while ((res=it.next()) == NO_ERROR) {
824 sp<AaptFile> file = it.getFile();
825 res = compileResourceFile(bundle, assets, file, it.getParams(),
826 (current!=assets), &table);
827 if (res != NO_ERROR) {
828 hasErrors = true;
829 }
830 }
831 }
832 current = current->getOverlay();
833 }
834
835 if (colors != NULL) {
836 err = makeFileResources(bundle, assets, &table, colors, "color");
837 if (err != NO_ERROR) {
838 hasErrors = true;
839 }
840 }
841
842 if (menus != NULL) {
843 err = makeFileResources(bundle, assets, &table, menus, "menu");
844 if (err != NO_ERROR) {
845 hasErrors = true;
846 }
847 }
848
849 // --------------------------------------------------------------------
850 // Assignment of resource IDs and initial generation of resource table.
851 // --------------------------------------------------------------------
852
853 if (table.hasResources()) {
854 sp<AaptFile> resFile(getResourceFile(assets));
855 if (resFile == NULL) {
856 fprintf(stderr, "Error: unable to generate entry for resource data\n");
857 return UNKNOWN_ERROR;
858 }
859
860 err = table.assignResourceIds();
861 if (err < NO_ERROR) {
862 return err;
863 }
864 }
865
866 // --------------------------------------------------------------
867 // Finally, we can now we can compile XML files, which may reference
868 // resources.
869 // --------------------------------------------------------------
870
871 if (layouts != NULL) {
872 ResourceDirIterator it(layouts, String8("layout"));
873 while ((err=it.next()) == NO_ERROR) {
874 String8 src = it.getFile()->getPrintableSource();
15c62a5b 875 err = compileXmlFile(assets, it.getFile(), &table, xmlFlags);
a534180c
TAOSP
876 if (err == NO_ERROR) {
877 ResXMLTree block;
878 block.setTo(it.getFile()->getData(), it.getFile()->getSize(), true);
879 checkForIds(src, block);
880 } else {
881 hasErrors = true;
882 }
883 }
884
885 if (err < NO_ERROR) {
886 hasErrors = true;
887 }
888 err = NO_ERROR;
889 }
890
891 if (anims != NULL) {
892 ResourceDirIterator it(anims, String8("anim"));
893 while ((err=it.next()) == NO_ERROR) {
15c62a5b 894 err = compileXmlFile(assets, it.getFile(), &table, xmlFlags);
a534180c
TAOSP
895 if (err != NO_ERROR) {
896 hasErrors = true;
897 }
898 }
899
900 if (err < NO_ERROR) {
901 hasErrors = true;
902 }
903 err = NO_ERROR;
904 }
905
906 if (xmls != NULL) {
907 ResourceDirIterator it(xmls, String8("xml"));
908 while ((err=it.next()) == NO_ERROR) {
15c62a5b 909 err = compileXmlFile(assets, it.getFile(), &table, xmlFlags);
a534180c
TAOSP
910 if (err != NO_ERROR) {
911 hasErrors = true;
912 }
913 }
914
915 if (err < NO_ERROR) {
916 hasErrors = true;
917 }
918 err = NO_ERROR;
919 }
920
921 if (drawables != NULL) {
922 err = postProcessImages(assets, &table, drawables);
923 if (err != NO_ERROR) {
924 hasErrors = true;
925 }
926 }
927
928 if (colors != NULL) {
929 ResourceDirIterator it(colors, String8("color"));
930 while ((err=it.next()) == NO_ERROR) {
15c62a5b 931 err = compileXmlFile(assets, it.getFile(), &table, xmlFlags);
a534180c
TAOSP
932 if (err != NO_ERROR) {
933 hasErrors = true;
934 }
935 }
936
937 if (err < NO_ERROR) {
938 hasErrors = true;
939 }
940 err = NO_ERROR;
941 }
942
943 if (menus != NULL) {
944 ResourceDirIterator it(menus, String8("menu"));
945 while ((err=it.next()) == NO_ERROR) {
946 String8 src = it.getFile()->getPrintableSource();
15c62a5b 947 err = compileXmlFile(assets, it.getFile(), &table, xmlFlags);
a534180c
TAOSP
948 if (err != NO_ERROR) {
949 hasErrors = true;
950 }
951 ResXMLTree block;
952 block.setTo(it.getFile()->getData(), it.getFile()->getSize(), true);
953 checkForIds(src, block);
954 }
955
956 if (err < NO_ERROR) {
957 hasErrors = true;
958 }
959 err = NO_ERROR;
960 }
961
962 const sp<AaptFile> manifestFile(androidManifestFile->getFiles().valueAt(0));
963 String8 manifestPath(manifestFile->getPrintableSource());
964
965 // Perform a basic validation of the manifest file. This time we
966 // parse it with the comments intact, so that we can use them to
967 // generate java docs... so we are not going to write this one
968 // back out to the final manifest data.
969 err = compileXmlFile(assets, manifestFile, &table,
970 XML_COMPILE_ASSIGN_ATTRIBUTE_IDS
971 | XML_COMPILE_STRIP_WHITESPACE | XML_COMPILE_STRIP_RAW_VALUES);
972 if (err < NO_ERROR) {
973 return err;
974 }
975 ResXMLTree block;
976 block.setTo(manifestFile->getData(), manifestFile->getSize(), true);
977 String16 manifest16("manifest");
978 String16 permission16("permission");
979 String16 permission_group16("permission-group");
980 String16 uses_permission16("uses-permission");
981 String16 instrumentation16("instrumentation");
982 String16 application16("application");
983 String16 provider16("provider");
984 String16 service16("service");
985 String16 receiver16("receiver");
986 String16 activity16("activity");
987 String16 action16("action");
988 String16 category16("category");
989 String16 data16("scheme");
990 const char* packageIdentChars = "abcdefghijklmnopqrstuvwxyz"
991 "ABCDEFGHIJKLMNOPQRSTUVWXYZ._0123456789";
992 const char* packageIdentCharsWithTheStupid = "abcdefghijklmnopqrstuvwxyz"
993 "ABCDEFGHIJKLMNOPQRSTUVWXYZ._0123456789-";
994 const char* classIdentChars = "abcdefghijklmnopqrstuvwxyz"
995 "ABCDEFGHIJKLMNOPQRSTUVWXYZ._0123456789$";
996 const char* processIdentChars = "abcdefghijklmnopqrstuvwxyz"
997 "ABCDEFGHIJKLMNOPQRSTUVWXYZ._0123456789:";
998 const char* authoritiesIdentChars = "abcdefghijklmnopqrstuvwxyz"
999 "ABCDEFGHIJKLMNOPQRSTUVWXYZ._0123456789-:;";
1000 const char* typeIdentChars = "abcdefghijklmnopqrstuvwxyz"
1001 "ABCDEFGHIJKLMNOPQRSTUVWXYZ._0123456789:-/*+";
1002 const char* schemeIdentChars = "abcdefghijklmnopqrstuvwxyz"
1003 "ABCDEFGHIJKLMNOPQRSTUVWXYZ._0123456789-";
1004 ResXMLTree::event_code_t code;
1005 sp<AaptSymbols> permissionSymbols;
1006 sp<AaptSymbols> permissionGroupSymbols;
1007 while ((code=block.next()) != ResXMLTree::END_DOCUMENT
1008 && code > ResXMLTree::BAD_DOCUMENT) {
1009 if (code == ResXMLTree::START_TAG) {
1010 size_t len;
1011 if (block.getElementNamespace(&len) != NULL) {
1012 continue;
1013 }
1014 if (strcmp16(block.getElementName(&len), manifest16.string()) == 0) {
1015 if (validateAttr(manifestPath, block, NULL, "package",
1016 packageIdentChars, true) != ATTR_OKAY) {
1017 hasErrors = true;
1018 }
1019 } else if (strcmp16(block.getElementName(&len), permission16.string()) == 0
1020 || strcmp16(block.getElementName(&len), permission_group16.string()) == 0) {
1021 const bool isGroup = strcmp16(block.getElementName(&len),
1022 permission_group16.string()) == 0;
1023 if (validateAttr(manifestPath, block, RESOURCES_ANDROID_NAMESPACE, "name",
1024 isGroup ? packageIdentCharsWithTheStupid
1025 : packageIdentChars, true) != ATTR_OKAY) {
1026 hasErrors = true;
1027 }
1028 SourcePos srcPos(manifestPath, block.getLineNumber());
1029 sp<AaptSymbols> syms;
1030 if (!isGroup) {
1031 syms = permissionSymbols;
1032 if (syms == NULL) {
1033 sp<AaptSymbols> symbols =
1034 assets->getSymbolsFor(String8("Manifest"));
1035 syms = permissionSymbols = symbols->addNestedSymbol(
1036 String8("permission"), srcPos);
1037 }
1038 } else {
1039 syms = permissionGroupSymbols;
1040 if (syms == NULL) {
1041 sp<AaptSymbols> symbols =
1042 assets->getSymbolsFor(String8("Manifest"));
1043 syms = permissionGroupSymbols = symbols->addNestedSymbol(
1044 String8("permission_group"), srcPos);
1045 }
1046 }
1047 size_t len;
1048 ssize_t index = block.indexOfAttribute(RESOURCES_ANDROID_NAMESPACE, "name");
1049 const uint16_t* id = block.getAttributeStringValue(index, &len);
1050 if (id == NULL) {
1051 fprintf(stderr, "%s:%d: missing name attribute in element <%s>.\n",
1052 manifestPath.string(), block.getLineNumber(),
1053 String8(block.getElementName(&len)).string());
1054 hasErrors = true;
1055 break;
1056 }
1057 String8 idStr(id);
1058 char* p = idStr.lockBuffer(idStr.size());
1059 char* e = p + idStr.size();
1060 bool begins_with_digit = true; // init to true so an empty string fails
1061 while (e > p) {
1062 e--;
1063 if (*e >= '0' && *e <= '9') {
1064 begins_with_digit = true;
1065 continue;
1066 }
1067 if ((*e >= 'a' && *e <= 'z') ||
1068 (*e >= 'A' && *e <= 'Z') ||
1069 (*e == '_')) {
1070 begins_with_digit = false;
1071 continue;
1072 }
1073 if (isGroup && (*e == '-')) {
1074 *e = '_';
1075 begins_with_digit = false;
1076 continue;
1077 }
1078 e++;
1079 break;
1080 }
1081 idStr.unlockBuffer();
1082 // verify that we stopped because we hit a period or
1083 // the beginning of the string, and that the
1084 // identifier didn't begin with a digit.
1085 if (begins_with_digit || (e != p && *(e-1) != '.')) {
1086 fprintf(stderr,
1087 "%s:%d: Permission name <%s> is not a valid Java symbol\n",
1088 manifestPath.string(), block.getLineNumber(), idStr.string());
1089 hasErrors = true;
1090 }
1091 syms->addStringSymbol(String8(e), idStr, srcPos);
1092 const uint16_t* cmt = block.getComment(&len);
1093 if (cmt != NULL && *cmt != 0) {
1094 //printf("Comment of %s: %s\n", String8(e).string(),
1095 // String8(cmt).string());
1096 syms->appendComment(String8(e), String16(cmt), srcPos);
1097 } else {
1098 //printf("No comment for %s\n", String8(e).string());
1099 }
1100 syms->makeSymbolPublic(String8(e), srcPos);
1101 } else if (strcmp16(block.getElementName(&len), uses_permission16.string()) == 0) {
1102 if (validateAttr(manifestPath, block, RESOURCES_ANDROID_NAMESPACE, "name",
1103 packageIdentChars, true) != ATTR_OKAY) {
1104 hasErrors = true;
1105 }
1106 } else if (strcmp16(block.getElementName(&len), instrumentation16.string()) == 0) {
1107 if (validateAttr(manifestPath, block, RESOURCES_ANDROID_NAMESPACE, "name",
1108 classIdentChars, true) != ATTR_OKAY) {
1109 hasErrors = true;
1110 }
1111 if (validateAttr(manifestPath, block,
1112 RESOURCES_ANDROID_NAMESPACE, "targetPackage",
1113 packageIdentChars, true) != ATTR_OKAY) {
1114 hasErrors = true;
1115 }
1116 } else if (strcmp16(block.getElementName(&len), application16.string()) == 0) {
1117 if (validateAttr(manifestPath, block, RESOURCES_ANDROID_NAMESPACE, "name",
1118 classIdentChars, false) != ATTR_OKAY) {
1119 hasErrors = true;
1120 }
1121 if (validateAttr(manifestPath, block,
1122 RESOURCES_ANDROID_NAMESPACE, "permission",
1123 packageIdentChars, false) != ATTR_OKAY) {
1124 hasErrors = true;
1125 }
1126 if (validateAttr(manifestPath, block,
1127 RESOURCES_ANDROID_NAMESPACE, "process",
1128 processIdentChars, false) != ATTR_OKAY) {
1129 hasErrors = true;
1130 }
1131 if (validateAttr(manifestPath, block,
1132 RESOURCES_ANDROID_NAMESPACE, "taskAffinity",
1133 processIdentChars, false) != ATTR_OKAY) {
1134 hasErrors = true;
1135 }
1136 } else if (strcmp16(block.getElementName(&len), provider16.string()) == 0) {
1137 if (validateAttr(manifestPath, block, RESOURCES_ANDROID_NAMESPACE, "name",
1138 classIdentChars, true) != ATTR_OKAY) {
1139 hasErrors = true;
1140 }
1141 if (validateAttr(manifestPath, block,
1142 RESOURCES_ANDROID_NAMESPACE, "authorities",
1143 authoritiesIdentChars, true) != ATTR_OKAY) {
1144 hasErrors = true;
1145 }
1146 if (validateAttr(manifestPath, block,
1147 RESOURCES_ANDROID_NAMESPACE, "permission",
1148 packageIdentChars, false) != ATTR_OKAY) {
1149 hasErrors = true;
1150 }
1151 if (validateAttr(manifestPath, block,
1152 RESOURCES_ANDROID_NAMESPACE, "process",
1153 processIdentChars, false) != ATTR_OKAY) {
1154 hasErrors = true;
1155 }
1156 } else if (strcmp16(block.getElementName(&len), service16.string()) == 0
1157 || strcmp16(block.getElementName(&len), receiver16.string()) == 0
1158 || strcmp16(block.getElementName(&len), activity16.string()) == 0) {
1159 if (validateAttr(manifestPath, block, RESOURCES_ANDROID_NAMESPACE, "name",
1160 classIdentChars, true) != ATTR_OKAY) {
1161 hasErrors = true;
1162 }
1163 if (validateAttr(manifestPath, block,
1164 RESOURCES_ANDROID_NAMESPACE, "permission",
1165 packageIdentChars, false) != ATTR_OKAY) {
1166 hasErrors = true;
1167 }
1168 if (validateAttr(manifestPath, block,
1169 RESOURCES_ANDROID_NAMESPACE, "process",
1170 processIdentChars, false) != ATTR_OKAY) {
1171 hasErrors = true;
1172 }
1173 if (validateAttr(manifestPath, block,
1174 RESOURCES_ANDROID_NAMESPACE, "taskAffinity",
1175 processIdentChars, false) != ATTR_OKAY) {
1176 hasErrors = true;
1177 }
1178 } else if (strcmp16(block.getElementName(&len), action16.string()) == 0
1179 || strcmp16(block.getElementName(&len), category16.string()) == 0) {
1180 if (validateAttr(manifestPath, block,
1181 RESOURCES_ANDROID_NAMESPACE, "name",
1182 packageIdentChars, true) != ATTR_OKAY) {
1183 hasErrors = true;
1184 }
1185 } else if (strcmp16(block.getElementName(&len), data16.string()) == 0) {
1186 if (validateAttr(manifestPath, block,
1187 RESOURCES_ANDROID_NAMESPACE, "mimeType",
1188 typeIdentChars, true) != ATTR_OKAY) {
1189 hasErrors = true;
1190 }
1191 if (validateAttr(manifestPath, block,
1192 RESOURCES_ANDROID_NAMESPACE, "scheme",
1193 schemeIdentChars, true) != ATTR_OKAY) {
1194 hasErrors = true;
1195 }
1196 }
1197 }
1198 }
1199
1200 if (table.validateLocalizations()) {
1201 hasErrors = true;
1202 }
1203
1204 if (hasErrors) {
1205 return UNKNOWN_ERROR;
1206 }
1207
1208 // Generate final compiled manifest file.
1209 manifestFile->clearData();
e942a5c2
DH
1210 sp<XMLNode> manifestTree = XMLNode::parse(manifestFile);
1211 if (manifestTree == NULL) {
1212 return UNKNOWN_ERROR;
1213 }
1214 err = massageManifest(bundle, manifestTree);
1215 if (err < NO_ERROR) {
1216 return err;
1217 }
1218 err = compileXmlFile(assets, manifestTree, manifestFile, &table);
a534180c
TAOSP
1219 if (err < NO_ERROR) {
1220 return err;
1221 }
1222
1223 //block.restart();
1224 //printXMLBlock(&block);
1225
1226 // --------------------------------------------------------------
1227 // Generate the final resource table.
1228 // Re-flatten because we may have added new resource IDs
1229 // --------------------------------------------------------------
1230
1231 if (table.hasResources()) {
1232 sp<AaptSymbols> symbols = assets->getSymbolsFor(String8("R"));
1233 err = table.addSymbols(symbols);
1234 if (err < NO_ERROR) {
1235 return err;
1236 }
1237
1238 sp<AaptFile> resFile(getResourceFile(assets));
1239 if (resFile == NULL) {
1240 fprintf(stderr, "Error: unable to generate entry for resource data\n");
1241 return UNKNOWN_ERROR;
1242 }
1243
1244 err = table.flatten(bundle, resFile);
1245 if (err < NO_ERROR) {
1246 return err;
1247 }
1248
1249 if (bundle->getPublicOutputFile()) {
1250 FILE* fp = fopen(bundle->getPublicOutputFile(), "w+");
1251 if (fp == NULL) {
1252 fprintf(stderr, "ERROR: Unable to open public definitions output file %s: %s\n",
1253 (const char*)bundle->getPublicOutputFile(), strerror(errno));
1254 return UNKNOWN_ERROR;
1255 }
1256 if (bundle->getVerbose()) {
1257 printf(" Writing public definitions to %s.\n", bundle->getPublicOutputFile());
1258 }
1259 table.writePublicDefinitions(String16(assets->getPackage()), fp);
981d1579 1260 fclose(fp);
a534180c 1261 }
094e8965 1262#if 0
a534180c
TAOSP
1263 NOISY(
1264 ResTable rt;
1265 rt.add(resFile->getData(), resFile->getSize(), NULL);
1266 printf("Generated resources:\n");
1267 rt.print();
1268 )
094e8965 1269#endif
a534180c
TAOSP
1270 // These resources are now considered to be a part of the included
1271 // resources, for others to reference.
1272 err = assets->addIncludedResources(resFile);
1273 if (err < NO_ERROR) {
1274 fprintf(stderr, "ERROR: Unable to parse generated resources, aborting.\n");
1275 return err;
1276 }
1277 }
a534180c
TAOSP
1278 return err;
1279}
1280
1281static const char* getIndentSpace(int indent)
1282{
1283static const char whitespace[] =
1284" ";
1285
1286 return whitespace + sizeof(whitespace) - 1 - indent*4;
1287}
1288
1289static status_t fixupSymbol(String16* inoutSymbol)
1290{
1291 inoutSymbol->replaceAll('.', '_');
1292 inoutSymbol->replaceAll(':', '_');
1293 return NO_ERROR;
1294}
1295
1296static String16 getAttributeComment(const sp<AaptAssets>& assets,
1297 const String8& name,
1298 String16* outTypeComment = NULL)
1299{
1300 sp<AaptSymbols> asym = assets->getSymbolsFor(String8("R"));
1301 if (asym != NULL) {
1302 //printf("Got R symbols!\n");
1303 asym = asym->getNestedSymbols().valueFor(String8("attr"));
1304 if (asym != NULL) {
1305 //printf("Got attrs symbols! comment %s=%s\n",
1306 // name.string(), String8(asym->getComment(name)).string());
1307 if (outTypeComment != NULL) {
1308 *outTypeComment = asym->getTypeComment(name);
1309 }
1310 return asym->getComment(name);
1311 }
1312 }
1313 return String16();
1314}
1315
1316static status_t writeLayoutClasses(
1317 FILE* fp, const sp<AaptAssets>& assets,
1318 const sp<AaptSymbols>& symbols, int indent, bool includePrivate)
1319{
1320 const char* indentStr = getIndentSpace(indent);
1321 if (!includePrivate) {
1322 fprintf(fp, "%s/** @doconly */\n", indentStr);
1323 }
1324 fprintf(fp, "%spublic static final class styleable {\n", indentStr);
1325 indent++;
1326
1327 String16 attr16("attr");
1328 String16 package16(assets->getPackage());
1329
1330 indentStr = getIndentSpace(indent);
1331 bool hasErrors = false;
1332
1333 size_t i;
1334 size_t N = symbols->getNestedSymbols().size();
1335 for (i=0; i<N; i++) {
1336 sp<AaptSymbols> nsymbols = symbols->getNestedSymbols().valueAt(i);
1337 String16 nclassName16(symbols->getNestedSymbols().keyAt(i));
1338 String8 realClassName(nclassName16);
1339 if (fixupSymbol(&nclassName16) != NO_ERROR) {
1340 hasErrors = true;
1341 }
1342 String8 nclassName(nclassName16);
1343
1344 SortedVector<uint32_t> idents;
1345 Vector<uint32_t> origOrder;
1346 Vector<bool> publicFlags;
1347
1348 size_t a;
1349 size_t NA = nsymbols->getSymbols().size();
1350 for (a=0; a<NA; a++) {
1351 const AaptSymbolEntry& sym(nsymbols->getSymbols().valueAt(a));
1352 int32_t code = sym.typeCode == AaptSymbolEntry::TYPE_INT32
1353 ? sym.int32Val : 0;
1354 bool isPublic = true;
1355 if (code == 0) {
1356 String16 name16(sym.name);
1357 uint32_t typeSpecFlags;
1358 code = assets->getIncludedResources().identifierForName(
1359 name16.string(), name16.size(),
1360 attr16.string(), attr16.size(),
1361 package16.string(), package16.size(), &typeSpecFlags);
1362 if (code == 0) {
1363 fprintf(stderr, "ERROR: In <declare-styleable> %s, unable to find attribute %s\n",
1364 nclassName.string(), sym.name.string());
1365 hasErrors = true;
1366 }
1367 isPublic = (typeSpecFlags&ResTable_typeSpec::SPEC_PUBLIC) != 0;
1368 }
1369 idents.add(code);
1370 origOrder.add(code);
1371 publicFlags.add(isPublic);
1372 }
1373
1374 NA = idents.size();
1375
f3cd0b05
DH
1376 bool deprecated = false;
1377
a534180c
TAOSP
1378 String16 comment = symbols->getComment(realClassName);
1379 fprintf(fp, "%s/** ", indentStr);
1380 if (comment.size() > 0) {
f3cd0b05
DH
1381 String8 cmt(comment);
1382 fprintf(fp, "%s\n", cmt.string());
1383 if (strstr(cmt.string(), "@deprecated") != NULL) {
1384 deprecated = true;
1385 }
a534180c
TAOSP
1386 } else {
1387 fprintf(fp, "Attributes that can be used with a %s.\n", nclassName.string());
1388 }
1389 bool hasTable = false;
1390 for (a=0; a<NA; a++) {
1391 ssize_t pos = idents.indexOf(origOrder.itemAt(a));
1392 if (pos >= 0) {
1393 if (!hasTable) {
1394 hasTable = true;
1395 fprintf(fp,
1396 "%s <p>Includes the following attributes:</p>\n"
3ff93dd6 1397 "%s <table>\n"
a534180c
TAOSP
1398 "%s <colgroup align=\"left\" />\n"
1399 "%s <colgroup align=\"left\" />\n"
3ff93dd6 1400 "%s <tr><th>Attribute</th><th>Description</th></tr>\n",
a534180c
TAOSP
1401 indentStr,
1402 indentStr,
1403 indentStr,
1404 indentStr,
1405 indentStr);
1406 }
1407 const AaptSymbolEntry& sym = nsymbols->getSymbols().valueAt(a);
1408 if (!publicFlags.itemAt(a) && !includePrivate) {
1409 continue;
1410 }
1411 String8 name8(sym.name);
1412 String16 comment(sym.comment);
1413 if (comment.size() <= 0) {
1414 comment = getAttributeComment(assets, name8);
1415 }
1416 if (comment.size() > 0) {
1417 const char16_t* p = comment.string();
1418 while (*p != 0 && *p != '.') {
1419 if (*p == '{') {
1420 while (*p != 0 && *p != '}') {
1421 p++;
1422 }
1423 } else {
1424 p++;
1425 }
1426 }
1427 if (*p == '.') {
1428 p++;
1429 }
1430 comment = String16(comment.string(), p-comment.string());
1431 }
1432 String16 name(name8);
1433 fixupSymbol(&name);
3ff93dd6 1434 fprintf(fp, "%s <tr><td><code>{@link #%s_%s %s:%s}</code></td><td>%s</td></tr>\n",
a534180c
TAOSP
1435 indentStr, nclassName.string(),
1436 String8(name).string(),
1437 assets->getPackage().string(),
1438 String8(name).string(),
1439 String8(comment).string());
1440 }
1441 }
1442 if (hasTable) {
1443 fprintf(fp, "%s </table>\n", indentStr);
1444 }
1445 for (a=0; a<NA; a++) {
1446 ssize_t pos = idents.indexOf(origOrder.itemAt(a));
1447 if (pos >= 0) {
1448 const AaptSymbolEntry& sym = nsymbols->getSymbols().valueAt(a);
1449 if (!publicFlags.itemAt(a) && !includePrivate) {
1450 continue;
1451 }
1452 String16 name(sym.name);
1453 fixupSymbol(&name);
1454 fprintf(fp, "%s @see #%s_%s\n",
1455 indentStr, nclassName.string(),
1456 String8(name).string());
1457 }
1458 }
1459 fprintf(fp, "%s */\n", getIndentSpace(indent));
1460
f3cd0b05
DH
1461 if (deprecated) {
1462 fprintf(fp, "%s@Deprecated\n", indentStr);
1463 }
1464
a534180c
TAOSP
1465 fprintf(fp,
1466 "%spublic static final int[] %s = {\n"
1467 "%s",
1468 indentStr, nclassName.string(),
1469 getIndentSpace(indent+1));
1470
1471 for (a=0; a<NA; a++) {
1472 if (a != 0) {
1473 if ((a&3) == 0) {
1474 fprintf(fp, ",\n%s", getIndentSpace(indent+1));
1475 } else {
1476 fprintf(fp, ", ");
1477 }
1478 }
1479 fprintf(fp, "0x%08x", idents[a]);
1480 }
1481
1482 fprintf(fp, "\n%s};\n", indentStr);
1483
1484 for (a=0; a<NA; a++) {
1485 ssize_t pos = idents.indexOf(origOrder.itemAt(a));
1486 if (pos >= 0) {
1487 const AaptSymbolEntry& sym = nsymbols->getSymbols().valueAt(a);
1488 if (!publicFlags.itemAt(a) && !includePrivate) {
1489 continue;
1490 }
1491 String8 name8(sym.name);
1492 String16 comment(sym.comment);
1493 String16 typeComment;
1494 if (comment.size() <= 0) {
1495 comment = getAttributeComment(assets, name8, &typeComment);
1496 } else {
1497 getAttributeComment(assets, name8, &typeComment);
1498 }
1499 String16 name(name8);
1500 if (fixupSymbol(&name) != NO_ERROR) {
1501 hasErrors = true;
1502 }
1503
1504 uint32_t typeSpecFlags = 0;
1505 String16 name16(sym.name);
1506 assets->getIncludedResources().identifierForName(
1507 name16.string(), name16.size(),
1508 attr16.string(), attr16.size(),
1509 package16.string(), package16.size(), &typeSpecFlags);
1510 //printf("%s:%s/%s: 0x%08x\n", String8(package16).string(),
1511 // String8(attr16).string(), String8(name16).string(), typeSpecFlags);
1512 const bool pub = (typeSpecFlags&ResTable_typeSpec::SPEC_PUBLIC) != 0;
f3cd0b05
DH
1513
1514 bool deprecated = false;
1515
a534180c
TAOSP
1516 fprintf(fp, "%s/**\n", indentStr);
1517 if (comment.size() > 0) {
f3cd0b05 1518 String8 cmt(comment);
a534180c 1519 fprintf(fp, "%s <p>\n%s @attr description\n", indentStr, indentStr);
f3cd0b05
DH
1520 fprintf(fp, "%s %s\n", indentStr, cmt.string());
1521 if (strstr(cmt.string(), "@deprecated") != NULL) {
1522 deprecated = true;
1523 }
a534180c
TAOSP
1524 } else {
1525 fprintf(fp,
1526 "%s <p>This symbol is the offset where the {@link %s.R.attr#%s}\n"
1527 "%s attribute's value can be found in the {@link #%s} array.\n",
1528 indentStr,
1529 pub ? assets->getPackage().string()
1530 : assets->getSymbolsPrivatePackage().string(),
1531 String8(name).string(),
1532 indentStr, nclassName.string());
1533 }
1534 if (typeComment.size() > 0) {
f3cd0b05
DH
1535 String8 cmt(typeComment);
1536 fprintf(fp, "\n\n%s %s\n", indentStr, cmt.string());
1537 if (strstr(cmt.string(), "@deprecated") != NULL) {
1538 deprecated = true;
1539 }
a534180c
TAOSP
1540 }
1541 if (comment.size() > 0) {
1542 if (pub) {
1543 fprintf(fp,
1544 "%s <p>This corresponds to the global attribute"
1545 "%s resource symbol {@link %s.R.attr#%s}.\n",
1546 indentStr, indentStr,
1547 assets->getPackage().string(),
1548 String8(name).string());
1549 } else {
1550 fprintf(fp,
1551 "%s <p>This is a private symbol.\n", indentStr);
1552 }
1553 }
1554 fprintf(fp, "%s @attr name %s:%s\n", indentStr,
1555 "android", String8(name).string());
1556 fprintf(fp, "%s*/\n", indentStr);
f3cd0b05
DH
1557 if (deprecated) {
1558 fprintf(fp, "%s@Deprecated\n", indentStr);
1559 }
a534180c
TAOSP
1560 fprintf(fp,
1561 "%spublic static final int %s_%s = %d;\n",
1562 indentStr, nclassName.string(),
1563 String8(name).string(), (int)pos);
1564 }
1565 }
1566 }
1567
1568 indent--;
1569 fprintf(fp, "%s};\n", getIndentSpace(indent));
1570 return hasErrors ? UNKNOWN_ERROR : NO_ERROR;
1571}
1572
1573static status_t writeSymbolClass(
1574 FILE* fp, const sp<AaptAssets>& assets, bool includePrivate,
1575 const sp<AaptSymbols>& symbols, const String8& className, int indent)
1576{
1577 fprintf(fp, "%spublic %sfinal class %s {\n",
1578 getIndentSpace(indent),
1579 indent != 0 ? "static " : "", className.string());
1580 indent++;
1581
1582 size_t i;
1583 status_t err = NO_ERROR;
1584
1585 size_t N = symbols->getSymbols().size();
1586 for (i=0; i<N; i++) {
1587 const AaptSymbolEntry& sym = symbols->getSymbols().valueAt(i);
1588 if (sym.typeCode != AaptSymbolEntry::TYPE_INT32) {
1589 continue;
1590 }
1591 if (!includePrivate && !sym.isPublic) {
1592 continue;
1593 }
1594 String16 name(sym.name);
1595 String8 realName(name);
1596 if (fixupSymbol(&name) != NO_ERROR) {
1597 return UNKNOWN_ERROR;
1598 }
1599 String16 comment(sym.comment);
1600 bool haveComment = false;
f3cd0b05 1601 bool deprecated = false;
a534180c
TAOSP
1602 if (comment.size() > 0) {
1603 haveComment = true;
f3cd0b05 1604 String8 cmt(comment);
a534180c
TAOSP
1605 fprintf(fp,
1606 "%s/** %s\n",
f3cd0b05
DH
1607 getIndentSpace(indent), cmt.string());
1608 if (strstr(cmt.string(), "@deprecated") != NULL) {
1609 deprecated = true;
1610 }
a534180c
TAOSP
1611 } else if (sym.isPublic && !includePrivate) {
1612 sym.sourcePos.warning("No comment for public symbol %s:%s/%s",
1613 assets->getPackage().string(), className.string(),
1614 String8(sym.name).string());
1615 }
1616 String16 typeComment(sym.typeComment);
1617 if (typeComment.size() > 0) {
f3cd0b05 1618 String8 cmt(typeComment);
a534180c
TAOSP
1619 if (!haveComment) {
1620 haveComment = true;
1621 fprintf(fp,
f3cd0b05 1622 "%s/** %s\n", getIndentSpace(indent), cmt.string());
a534180c
TAOSP
1623 } else {
1624 fprintf(fp,
f3cd0b05
DH
1625 "%s %s\n", getIndentSpace(indent), cmt.string());
1626 }
1627 if (strstr(cmt.string(), "@deprecated") != NULL) {
1628 deprecated = true;
a534180c
TAOSP
1629 }
1630 }
1631 if (haveComment) {
1632 fprintf(fp,"%s */\n", getIndentSpace(indent));
1633 }
f3cd0b05
DH
1634 if (deprecated) {
1635 fprintf(fp, "%s@Deprecated\n", getIndentSpace(indent));
1636 }
a534180c
TAOSP
1637 fprintf(fp, "%spublic static final int %s=0x%08x;\n",
1638 getIndentSpace(indent),
1639 String8(name).string(), (int)sym.int32Val);
1640 }
1641
1642 for (i=0; i<N; i++) {
1643 const AaptSymbolEntry& sym = symbols->getSymbols().valueAt(i);
1644 if (sym.typeCode != AaptSymbolEntry::TYPE_STRING) {
1645 continue;
1646 }
1647 if (!includePrivate && !sym.isPublic) {
1648 continue;
1649 }
1650 String16 name(sym.name);
1651 if (fixupSymbol(&name) != NO_ERROR) {
1652 return UNKNOWN_ERROR;
1653 }
1654 String16 comment(sym.comment);
f3cd0b05 1655 bool deprecated = false;
a534180c 1656 if (comment.size() > 0) {
f3cd0b05 1657 String8 cmt(comment);
a534180c
TAOSP
1658 fprintf(fp,
1659 "%s/** %s\n"
1660 "%s */\n",
f3cd0b05 1661 getIndentSpace(indent), cmt.string(),
a534180c 1662 getIndentSpace(indent));
f3cd0b05
DH
1663 if (strstr(cmt.string(), "@deprecated") != NULL) {
1664 deprecated = true;
1665 }
a534180c
TAOSP
1666 } else if (sym.isPublic && !includePrivate) {
1667 sym.sourcePos.warning("No comment for public symbol %s:%s/%s",
1668 assets->getPackage().string(), className.string(),
1669 String8(sym.name).string());
1670 }
f3cd0b05
DH
1671 if (deprecated) {
1672 fprintf(fp, "%s@Deprecated\n", getIndentSpace(indent));
1673 }
a534180c
TAOSP
1674 fprintf(fp, "%spublic static final String %s=\"%s\";\n",
1675 getIndentSpace(indent),
1676 String8(name).string(), sym.stringVal.string());
1677 }
1678
1679 sp<AaptSymbols> styleableSymbols;
1680
1681 N = symbols->getNestedSymbols().size();
1682 for (i=0; i<N; i++) {
1683 sp<AaptSymbols> nsymbols = symbols->getNestedSymbols().valueAt(i);
1684 String8 nclassName(symbols->getNestedSymbols().keyAt(i));
1685 if (nclassName == "styleable") {
1686 styleableSymbols = nsymbols;
1687 } else {
1688 err = writeSymbolClass(fp, assets, includePrivate, nsymbols, nclassName, indent);
1689 }
1690 if (err != NO_ERROR) {
1691 return err;
1692 }
1693 }
1694
1695 if (styleableSymbols != NULL) {
1696 err = writeLayoutClasses(fp, assets, styleableSymbols, indent, includePrivate);
1697 if (err != NO_ERROR) {
1698 return err;
1699 }
1700 }
1701
1702 indent--;
1703 fprintf(fp, "%s}\n", getIndentSpace(indent));
1704 return NO_ERROR;
1705}
1706
1707status_t writeResourceSymbols(Bundle* bundle, const sp<AaptAssets>& assets,
1708 const String8& package, bool includePrivate)
1709{
1710 if (!bundle->getRClassDir()) {
1711 return NO_ERROR;
1712 }
1713
1714 const size_t N = assets->getSymbols().size();
1715 for (size_t i=0; i<N; i++) {
1716 sp<AaptSymbols> symbols = assets->getSymbols().valueAt(i);
1717 String8 className(assets->getSymbols().keyAt(i));
1718 String8 dest(bundle->getRClassDir());
1719 if (bundle->getMakePackageDirs()) {
1720 String8 pkg(package);
1721 const char* last = pkg.string();
1722 const char* s = last-1;
1723 do {
1724 s++;
1725 if (s > last && (*s == '.' || *s == 0)) {
1726 String8 part(last, s-last);
1727 dest.appendPath(part);
1728#ifdef HAVE_MS_C_RUNTIME
1729 _mkdir(dest.string());
1730#else
1731 mkdir(dest.string(), S_IRUSR|S_IWUSR|S_IXUSR|S_IRGRP|S_IXGRP);
1732#endif
1733 last = s+1;
1734 }
1735 } while (*s);
1736 }
1737 dest.appendPath(className);
1738 dest.append(".java");
1739 FILE* fp = fopen(dest.string(), "w+");
1740 if (fp == NULL) {
1741 fprintf(stderr, "ERROR: Unable to open class file %s: %s\n",
1742 dest.string(), strerror(errno));
1743 return UNKNOWN_ERROR;
1744 }
1745 if (bundle->getVerbose()) {
1746 printf(" Writing symbols for class %s.\n", className.string());
1747 }
1748
1749 fprintf(fp,
1750 "/* AUTO-GENERATED FILE. DO NOT MODIFY.\n"
1751 " *\n"
1752 " * This class was automatically generated by the\n"
1753 " * aapt tool from the resource data it found. It\n"
1754 " * should not be modified by hand.\n"
1755 " */\n"
1756 "\n"
1757 "package %s;\n\n", package.string());
1758
1759 status_t err = writeSymbolClass(fp, assets, includePrivate, symbols, className, 0);
1760 if (err != NO_ERROR) {
1761 return err;
1762 }
1763 fclose(fp);
1764 }
1765
1766 return NO_ERROR;
1767}
6648ff78
JO
1768
1769
1770
1771class ProguardKeepSet
1772{
1773public:
1774 // { rule --> { file locations } }
1775 KeyedVector<String8, SortedVector<String8> > rules;
1776
1777 void add(const String8& rule, const String8& where);
1778};
1779
1780void ProguardKeepSet::add(const String8& rule, const String8& where)
1781{
1782 ssize_t index = rules.indexOfKey(rule);
1783 if (index < 0) {
1784 index = rules.add(rule, SortedVector<String8>());
1785 }
1786 rules.editValueAt(index).add(where);
1787}
1788
4fd67e1d
DH
1789void
1790addProguardKeepRule(ProguardKeepSet* keep, const String8& inClassName,
1791 const char* pkg, const String8& srcName, int line)
1792{
1793 String8 className(inClassName);
1794 if (pkg != NULL) {
1795 // asdf --> package.asdf
1796 // .asdf .a.b --> package.asdf package.a.b
1797 // asdf.adsf --> asdf.asdf
1798 const char* p = className.string();
1799 const char* q = strchr(p, '.');
1800 if (p == q) {
1801 className = pkg;
1802 className.append(inClassName);
1803 } else if (q == NULL) {
1804 className = pkg;
1805 className.append(".");
1806 className.append(inClassName);
1807 }
1808 }
1809
1810 String8 rule("-keep class ");
1811 rule += className;
1812 rule += " { <init>(...); }";
1813
1814 String8 location("view ");
1815 location += srcName;
1816 char lineno[20];
1817 sprintf(lineno, ":%d", line);
1818 location += lineno;
1819
1820 keep->add(rule, location);
1821}
1822
6648ff78
JO
1823status_t
1824writeProguardForAndroidManifest(ProguardKeepSet* keep, const sp<AaptAssets>& assets)
1825{
1826 status_t err;
1827 ResXMLTree tree;
1828 size_t len;
1829 ResXMLTree::event_code_t code;
1830 int depth = 0;
1831 bool inApplication = false;
1832 String8 error;
1833 sp<AaptGroup> assGroup;
1834 sp<AaptFile> assFile;
1835 String8 pkg;
1836
1837 // First, look for a package file to parse. This is required to
1838 // be able to generate the resource information.
1839 assGroup = assets->getFiles().valueFor(String8("AndroidManifest.xml"));
1840 if (assGroup == NULL) {
1841 fprintf(stderr, "ERROR: No AndroidManifest.xml file found.\n");
1842 return -1;
1843 }
1844
1845 if (assGroup->getFiles().size() != 1) {
1846 fprintf(stderr, "warning: Multiple AndroidManifest.xml files found, using %s\n",
1847 assGroup->getFiles().valueAt(0)->getPrintableSource().string());
1848 }
1849
1850 assFile = assGroup->getFiles().valueAt(0);
1851
1852 err = parseXMLResource(assFile, &tree);
1853 if (err != NO_ERROR) {
1854 return err;
1855 }
1856
1857 tree.restart();
1858
1859 while ((code=tree.next()) != ResXMLTree::END_DOCUMENT && code != ResXMLTree::BAD_DOCUMENT) {
1860 if (code == ResXMLTree::END_TAG) {
1861 if (/* name == "Application" && */ depth == 2) {
1862 inApplication = false;
1863 }
1864 depth--;
1865 continue;
1866 }
1867 if (code != ResXMLTree::START_TAG) {
1868 continue;
1869 }
1870 depth++;
1871 String8 tag(tree.getElementName(&len));
1872 // printf("Depth %d tag %s\n", depth, tag.string());
87e7a47b 1873 bool keepTag = false;
6648ff78
JO
1874 if (depth == 1) {
1875 if (tag != "manifest") {
1876 fprintf(stderr, "ERROR: manifest does not start with <manifest> tag\n");
1877 return -1;
1878 }
1879 pkg = getAttribute(tree, NULL, "package", NULL);
87e7a47b
YW
1880 } else if (depth == 2) {
1881 if (tag == "application") {
1882 inApplication = true;
1883 keepTag = true;
4fd67e1d
DH
1884
1885 String8 agent = getAttribute(tree, "http://schemas.android.com/apk/res/android",
1886 "backupAgent", &error);
1887 if (agent.length() > 0) {
1888 addProguardKeepRule(keep, agent, pkg.string(),
1889 assFile->getPrintableSource(), tree.getLineNumber());
1890 }
87e7a47b
YW
1891 } else if (tag == "instrumentation") {
1892 keepTag = true;
1893 }
1894 }
1895 if (!keepTag && inApplication && depth == 3) {
1896 if (tag == "activity" || tag == "service" || tag == "receiver" || tag == "provider") {
1897 keepTag = true;
1898 }
1899 }
1900 if (keepTag) {
1901 String8 name = getAttribute(tree, "http://schemas.android.com/apk/res/android",
1902 "name", &error);
1903 if (error != "") {
1904 fprintf(stderr, "ERROR: %s\n", error.string());
1905 return -1;
1906 }
1907 if (name.length() > 0) {
4fd67e1d
DH
1908 addProguardKeepRule(keep, name, pkg.string(),
1909 assFile->getPrintableSource(), tree.getLineNumber());
6648ff78
JO
1910 }
1911 }
1912 }
1913
1914 return NO_ERROR;
1915}
1916
1917status_t
9f1b920f
DH
1918writeProguardForXml(ProguardKeepSet* keep, const sp<AaptFile>& layoutFile,
1919 const char* startTag, const char* altTag)
6648ff78
JO
1920{
1921 status_t err;
1922 ResXMLTree tree;
1923 size_t len;
1924 ResXMLTree::event_code_t code;
1925
1926 err = parseXMLResource(layoutFile, &tree);
1927 if (err != NO_ERROR) {
1928 return err;
1929 }
1930
1931 tree.restart();
1932
9f1b920f
DH
1933 if (startTag != NULL) {
1934 bool haveStart = false;
1935 while ((code=tree.next()) != ResXMLTree::END_DOCUMENT && code != ResXMLTree::BAD_DOCUMENT) {
1936 if (code != ResXMLTree::START_TAG) {
1937 continue;
1938 }
1939 String8 tag(tree.getElementName(&len));
1940 if (tag == startTag) {
1941 haveStart = true;
1942 }
1943 break;
1944 }
1945 if (!haveStart) {
1946 return NO_ERROR;
1947 }
1948 }
1949
6648ff78
JO
1950 while ((code=tree.next()) != ResXMLTree::END_DOCUMENT && code != ResXMLTree::BAD_DOCUMENT) {
1951 if (code != ResXMLTree::START_TAG) {
1952 continue;
1953 }
1954 String8 tag(tree.getElementName(&len));
1955
1956 // If there is no '.', we'll assume that it's one of the built in names.
1957 if (strchr(tag.string(), '.')) {
4fd67e1d 1958 addProguardKeepRule(keep, tag, NULL,
9f1b920f
DH
1959 layoutFile->getPrintableSource(), tree.getLineNumber());
1960 } else if (altTag != NULL && tag == altTag) {
1961 ssize_t classIndex = tree.indexOfAttribute(NULL, "class");
1962 if (classIndex < 0) {
1963 fprintf(stderr, "%s:%d: <view> does not have class attribute.\n",
1964 layoutFile->getPrintableSource().string(), tree.getLineNumber());
1965 } else {
1966 size_t len;
1967 addProguardKeepRule(keep,
4fd67e1d 1968 String8(tree.getAttributeStringValue(classIndex, &len)), NULL,
9f1b920f
DH
1969 layoutFile->getPrintableSource(), tree.getLineNumber());
1970 }
6648ff78
JO
1971 }
1972 }
1973
1974 return NO_ERROR;
1975}
1976
1977status_t
1978writeProguardForLayouts(ProguardKeepSet* keep, const sp<AaptAssets>& assets)
1979{
1980 status_t err;
ba94e60c
YW
1981 const Vector<sp<AaptDir> >& dirs = assets->resDirs();
1982 const size_t K = dirs.size();
1983 for (size_t k=0; k<K; k++) {
1984 const sp<AaptDir>& d = dirs.itemAt(k);
1985 const String8& dirName = d->getLeaf();
9f1b920f
DH
1986 const char* startTag = NULL;
1987 const char* altTag = NULL;
1988 if ((dirName == String8("layout")) || (strncmp(dirName.string(), "layout-", 7) == 0)) {
1989 altTag = "view";
1990 } else if ((dirName == String8("xml")) || (strncmp(dirName.string(), "xml-", 4) == 0)) {
1991 startTag = "PreferenceScreen";
1992 } else {
ba94e60c
YW
1993 continue;
1994 }
9f1b920f 1995
ba94e60c 1996 const KeyedVector<String8,sp<AaptGroup> > groups = d->getFiles();
6648ff78
JO
1997 const size_t N = groups.size();
1998 for (size_t i=0; i<N; i++) {
1999 const sp<AaptGroup>& group = groups.valueAt(i);
2000 const DefaultKeyedVector<AaptGroupEntry, sp<AaptFile> >& files = group->getFiles();
2001 const size_t M = files.size();
2002 for (size_t j=0; j<M; j++) {
9f1b920f 2003 err = writeProguardForXml(keep, files.valueAt(j), startTag, altTag);
6648ff78
JO
2004 if (err < 0) {
2005 return err;
2006 }
2007 }
2008 }
2009 }
2010 return NO_ERROR;
2011}
2012
2013status_t
2014writeProguardFile(Bundle* bundle, const sp<AaptAssets>& assets)
2015{
2016 status_t err = -1;
2017
2018 if (!bundle->getProguardFile()) {
2019 return NO_ERROR;
2020 }
2021
2022 ProguardKeepSet keep;
2023
2024 err = writeProguardForAndroidManifest(&keep, assets);
2025 if (err < 0) {
2026 return err;
2027 }
2028
2029 err = writeProguardForLayouts(&keep, assets);
2030 if (err < 0) {
2031 return err;
2032 }
2033
2034 FILE* fp = fopen(bundle->getProguardFile(), "w+");
2035 if (fp == NULL) {
2036 fprintf(stderr, "ERROR: Unable to open class file %s: %s\n",
2037 bundle->getProguardFile(), strerror(errno));
2038 return UNKNOWN_ERROR;
2039 }
2040
2041 const KeyedVector<String8, SortedVector<String8> >& rules = keep.rules;
2042 const size_t N = rules.size();
2043 for (size_t i=0; i<N; i++) {
2044 const SortedVector<String8>& locations = rules.valueAt(i);
2045 const size_t M = locations.size();
2046 for (size_t j=0; j<M; j++) {
2047 fprintf(fp, "# %s\n", locations.itemAt(j).string());
2048 }
2049 fprintf(fp, "%s\n\n", rules.keyAt(i).string());
2050 }
2051 fclose(fp);
2052
2053 return err;
2054}