• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //
2 // Copyright 2006 The Android Open Source Project
3 //
4 // Build resource files from raw assets.
5 //
6 
7 #include "XMLNode.h"
8 #include "ResourceTable.h"
9 
10 #include <host/pseudolocalize.h>
11 #include <utils/ByteOrder.h>
12 #include <errno.h>
13 #include <string.h>
14 
15 #ifndef HAVE_MS_C_RUNTIME
16 #define O_BINARY 0
17 #endif
18 
19 #define NOISY(x) //x
20 #define NOISY_PARSE(x) //x
21 
22 const char* const RESOURCES_ROOT_NAMESPACE = "http://schemas.android.com/apk/res/";
23 const char* const RESOURCES_ANDROID_NAMESPACE = "http://schemas.android.com/apk/res/android";
24 const char* const RESOURCES_ROOT_PRV_NAMESPACE = "http://schemas.android.com/apk/prv/res/";
25 
26 const char* const XLIFF_XMLNS = "urn:oasis:names:tc:xliff:document:1.2";
27 const char* const ALLOWED_XLIFF_ELEMENTS[] = {
28         "bpt",
29         "ept",
30         "it",
31         "ph",
32         "g",
33         "bx",
34         "ex",
35         "x"
36     };
37 
isWhitespace(const char16_t * str)38 bool isWhitespace(const char16_t* str)
39 {
40     while (*str != 0 && *str < 128 && isspace(*str)) {
41         str++;
42     }
43     return *str == 0;
44 }
45 
46 static const String16 RESOURCES_PREFIX(RESOURCES_ROOT_NAMESPACE);
47 static const String16 RESOURCES_PRV_PREFIX(RESOURCES_ROOT_PRV_NAMESPACE);
48 
getNamespaceResourcePackage(String16 namespaceUri,bool * outIsPublic)49 String16 getNamespaceResourcePackage(String16 namespaceUri, bool* outIsPublic)
50 {
51     //printf("%s starts with %s?\n", String8(namespaceUri).string(),
52     //       String8(RESOURCES_PREFIX).string());
53     size_t prefixSize;
54     bool isPublic = true;
55     if (namespaceUri.startsWith(RESOURCES_PREFIX)) {
56         prefixSize = RESOURCES_PREFIX.size();
57     } else if (namespaceUri.startsWith(RESOURCES_PRV_PREFIX)) {
58         isPublic = false;
59         prefixSize = RESOURCES_PRV_PREFIX.size();
60     } else {
61         if (outIsPublic) *outIsPublic = isPublic; // = true
62         return String16();
63     }
64 
65     //printf("YES!\n");
66     //printf("namespace: %s\n", String8(String16(namespaceUri, namespaceUri.size()-prefixSize, prefixSize)).string());
67     if (outIsPublic) *outIsPublic = isPublic;
68     return String16(namespaceUri, namespaceUri.size()-prefixSize, prefixSize);
69 }
70 
hasSubstitutionErrors(const char * fileName,ResXMLTree * inXml,String16 str16)71 status_t hasSubstitutionErrors(const char* fileName,
72                                ResXMLTree* inXml,
73                                String16 str16)
74 {
75     const char16_t* str = str16.string();
76     const char16_t* p = str;
77     const char16_t* end = str + str16.size();
78 
79     bool nonpositional = false;
80     int argCount = 0;
81 
82     while (p < end) {
83         /*
84          * Look for the start of a Java-style substitution sequence.
85          */
86         if (*p == '%' && p + 1 < end) {
87             p++;
88 
89             // A literal percent sign represented by %%
90             if (*p == '%') {
91                 p++;
92                 continue;
93             }
94 
95             argCount++;
96 
97             if (*p >= '0' && *p <= '9') {
98                 do {
99                     p++;
100                 } while (*p >= '0' && *p <= '9');
101                 if (*p != '$') {
102                     // This must be a size specification instead of position.
103                     nonpositional = true;
104                 }
105             } else if (*p == '<') {
106                 // Reusing last argument; bad idea since it can be re-arranged.
107                 nonpositional = true;
108                 p++;
109 
110                 // Optionally '$' can be specified at the end.
111                 if (p < end && *p == '$') {
112                     p++;
113                 }
114             } else {
115                 nonpositional = true;
116             }
117 
118             // Ignore flags and widths
119             while (p < end && (*p == '-' ||
120                     *p == '#' ||
121                     *p == '+' ||
122                     *p == ' ' ||
123                     *p == ',' ||
124                     *p == '(' ||
125                     (*p >= '0' && *p <= '9'))) {
126                 p++;
127             }
128 
129             /*
130              * This is a shortcut to detect strings that are going to Time.format()
131              * instead of String.format()
132              *
133              * Comparison of String.format() and Time.format() args:
134              *
135              * String: ABC E GH  ST X abcdefgh  nost x
136              *   Time:    DEFGHKMS W Za  d   hkm  s w yz
137              *
138              * Therefore we know it's definitely Time if we have:
139              *     DFKMWZkmwyz
140              */
141             if (p < end) {
142                 switch (*p) {
143                 case 'D':
144                 case 'F':
145                 case 'K':
146                 case 'M':
147                 case 'W':
148                 case 'Z':
149                 case 'k':
150                 case 'm':
151                 case 'w':
152                 case 'y':
153                 case 'z':
154                     return NO_ERROR;
155                 }
156             }
157         }
158 
159         p++;
160     }
161 
162     /*
163      * If we have more than one substitution in this string and any of them
164      * are not in positional form, give the user an error.
165      */
166     if (argCount > 1 && nonpositional) {
167         SourcePos(String8(fileName), inXml->getLineNumber()).error(
168                 "Multiple substitutions specified in non-positional format; "
169                 "did you mean to add the formatted=\"false\" attribute?\n");
170         return NOT_ENOUGH_DATA;
171     }
172 
173     return NO_ERROR;
174 }
175 
parseStyledString(Bundle * bundle,const char * fileName,ResXMLTree * inXml,const String16 & endTag,String16 * outString,Vector<StringPool::entry_style_span> * outSpans,bool isFormatted,bool pseudolocalize)176 status_t parseStyledString(Bundle* bundle,
177                            const char* fileName,
178                            ResXMLTree* inXml,
179                            const String16& endTag,
180                            String16* outString,
181                            Vector<StringPool::entry_style_span>* outSpans,
182                            bool isFormatted,
183                            bool pseudolocalize)
184 {
185     Vector<StringPool::entry_style_span> spanStack;
186     String16 curString;
187     String16 rawString;
188     const char* errorMsg;
189     int xliffDepth = 0;
190     bool firstTime = true;
191 
192     size_t len;
193     ResXMLTree::event_code_t code;
194     while ((code=inXml->next()) != ResXMLTree::END_DOCUMENT && code != ResXMLTree::BAD_DOCUMENT) {
195 
196         if (code == ResXMLTree::TEXT) {
197             String16 text(inXml->getText(&len));
198             if (firstTime && text.size() > 0) {
199                 firstTime = false;
200                 if (text.string()[0] == '@') {
201                     // If this is a resource reference, don't do the pseudoloc.
202                     pseudolocalize = false;
203                 }
204             }
205             if (xliffDepth == 0 && pseudolocalize) {
206                 std::string orig(String8(text).string());
207                 std::string pseudo = pseudolocalize_string(orig);
208                 curString.append(String16(String8(pseudo.c_str())));
209             } else {
210                 if (isFormatted && hasSubstitutionErrors(fileName, inXml, text) != NO_ERROR) {
211                     return UNKNOWN_ERROR;
212                 } else {
213                     curString.append(text);
214                 }
215             }
216         } else if (code == ResXMLTree::START_TAG) {
217             const String16 element16(inXml->getElementName(&len));
218             const String8 element8(element16);
219 
220             size_t nslen;
221             const uint16_t* ns = inXml->getElementNamespace(&nslen);
222             if (ns == NULL) {
223                 ns = (const uint16_t*)"\0\0";
224                 nslen = 0;
225             }
226             const String8 nspace(String16(ns, nslen));
227             if (nspace == XLIFF_XMLNS) {
228                 const int N = sizeof(ALLOWED_XLIFF_ELEMENTS)/sizeof(ALLOWED_XLIFF_ELEMENTS[0]);
229                 for (int i=0; i<N; i++) {
230                     if (element8 == ALLOWED_XLIFF_ELEMENTS[i]) {
231                         xliffDepth++;
232                         // in this case, treat it like it was just text, in other words, do nothing
233                         // here and silently drop this element
234                         goto moveon;
235                     }
236                 }
237                 {
238                     SourcePos(String8(fileName), inXml->getLineNumber()).error(
239                             "Found unsupported XLIFF tag <%s>\n",
240                             element8.string());
241                     return UNKNOWN_ERROR;
242                 }
243 moveon:
244                 continue;
245             }
246 
247             if (outSpans == NULL) {
248                 SourcePos(String8(fileName), inXml->getLineNumber()).error(
249                         "Found style tag <%s> where styles are not allowed\n", element8.string());
250                 return UNKNOWN_ERROR;
251             }
252 
253             if (!ResTable::collectString(outString, curString.string(),
254                                          curString.size(), false, &errorMsg, true)) {
255                 SourcePos(String8(fileName), inXml->getLineNumber()).error("%s (in %s)\n",
256                         errorMsg, String8(curString).string());
257                 return UNKNOWN_ERROR;
258             }
259             rawString.append(curString);
260             curString = String16();
261 
262             StringPool::entry_style_span span;
263             span.name = element16;
264             for (size_t ai=0; ai<inXml->getAttributeCount(); ai++) {
265                 span.name.append(String16(";"));
266                 const char16_t* str = inXml->getAttributeName(ai, &len);
267                 span.name.append(str, len);
268                 span.name.append(String16("="));
269                 str = inXml->getAttributeStringValue(ai, &len);
270                 span.name.append(str, len);
271             }
272             //printf("Span: %s\n", String8(span.name).string());
273             span.span.firstChar = span.span.lastChar = outString->size();
274             spanStack.push(span);
275 
276         } else if (code == ResXMLTree::END_TAG) {
277             size_t nslen;
278             const uint16_t* ns = inXml->getElementNamespace(&nslen);
279             if (ns == NULL) {
280                 ns = (const uint16_t*)"\0\0";
281                 nslen = 0;
282             }
283             const String8 nspace(String16(ns, nslen));
284             if (nspace == XLIFF_XMLNS) {
285                 xliffDepth--;
286                 continue;
287             }
288             if (!ResTable::collectString(outString, curString.string(),
289                                          curString.size(), false, &errorMsg, true)) {
290                 SourcePos(String8(fileName), inXml->getLineNumber()).error("%s (in %s)\n",
291                         errorMsg, String8(curString).string());
292                 return UNKNOWN_ERROR;
293             }
294             rawString.append(curString);
295             curString = String16();
296 
297             if (spanStack.size() == 0) {
298                 if (strcmp16(inXml->getElementName(&len), endTag.string()) != 0) {
299                     SourcePos(String8(fileName), inXml->getLineNumber()).error(
300                             "Found tag %s where <%s> close is expected\n",
301                             String8(inXml->getElementName(&len)).string(),
302                             String8(endTag).string());
303                     return UNKNOWN_ERROR;
304                 }
305                 break;
306             }
307             StringPool::entry_style_span span = spanStack.top();
308             String16 spanTag;
309             ssize_t semi = span.name.findFirst(';');
310             if (semi >= 0) {
311                 spanTag.setTo(span.name.string(), semi);
312             } else {
313                 spanTag.setTo(span.name);
314             }
315             if (strcmp16(inXml->getElementName(&len), spanTag.string()) != 0) {
316                 SourcePos(String8(fileName), inXml->getLineNumber()).error(
317                         "Found close tag %s where close tag %s is expected\n",
318                         String8(inXml->getElementName(&len)).string(),
319                         String8(spanTag).string());
320                 return UNKNOWN_ERROR;
321             }
322             bool empty = true;
323             if (outString->size() > 0) {
324                 span.span.lastChar = outString->size()-1;
325                 if (span.span.lastChar >= span.span.firstChar) {
326                     empty = false;
327                     outSpans->add(span);
328                 }
329             }
330             spanStack.pop();
331 
332             /*
333              * This warning seems to be just an irritation to most people,
334              * since it is typically introduced by translators who then never
335              * see the warning.
336              */
337             if (0 && empty) {
338                 fprintf(stderr, "%s:%d: warning: empty '%s' span found in text '%s'\n",
339                         fileName, inXml->getLineNumber(),
340                         String8(spanTag).string(), String8(*outString).string());
341 
342             }
343         } else if (code == ResXMLTree::START_NAMESPACE) {
344             // nothing
345         }
346     }
347 
348     if (code == ResXMLTree::BAD_DOCUMENT) {
349             SourcePos(String8(fileName), inXml->getLineNumber()).error(
350                     "Error parsing XML\n");
351     }
352 
353     if (outSpans != NULL && outSpans->size() > 0) {
354         if (curString.size() > 0) {
355             if (!ResTable::collectString(outString, curString.string(),
356                                          curString.size(), false, &errorMsg, true)) {
357                 SourcePos(String8(fileName), inXml->getLineNumber()).error(
358                         "%s (in %s)\n",
359                         errorMsg, String8(curString).string());
360                 return UNKNOWN_ERROR;
361             }
362         }
363     } else {
364         // There is no style information, so string processing will happen
365         // later as part of the overall type conversion.  Return to the
366         // client the raw unprocessed text.
367         rawString.append(curString);
368         outString->setTo(rawString);
369     }
370 
371     return NO_ERROR;
372 }
373 
374 struct namespace_entry {
375     String8 prefix;
376     String8 uri;
377 };
378 
make_prefix(int depth)379 static String8 make_prefix(int depth)
380 {
381     String8 prefix;
382     int i;
383     for (i=0; i<depth; i++) {
384         prefix.append("  ");
385     }
386     return prefix;
387 }
388 
build_namespace(const Vector<namespace_entry> & namespaces,const uint16_t * ns)389 static String8 build_namespace(const Vector<namespace_entry>& namespaces,
390         const uint16_t* ns)
391 {
392     String8 str;
393     if (ns != NULL) {
394         str = String8(ns);
395         const size_t N = namespaces.size();
396         for (size_t i=0; i<N; i++) {
397             const namespace_entry& ne = namespaces.itemAt(i);
398             if (ne.uri == str) {
399                 str = ne.prefix;
400                 break;
401             }
402         }
403         str.append(":");
404     }
405     return str;
406 }
407 
printXMLBlock(ResXMLTree * block)408 void printXMLBlock(ResXMLTree* block)
409 {
410     block->restart();
411 
412     Vector<namespace_entry> namespaces;
413 
414     ResXMLTree::event_code_t code;
415     int depth = 0;
416     while ((code=block->next()) != ResXMLTree::END_DOCUMENT && code != ResXMLTree::BAD_DOCUMENT) {
417         String8 prefix = make_prefix(depth);
418         int i;
419         if (code == ResXMLTree::START_TAG) {
420             size_t len;
421             const uint16_t* ns16 = block->getElementNamespace(&len);
422             String8 elemNs = build_namespace(namespaces, ns16);
423             const uint16_t* com16 = block->getComment(&len);
424             if (com16) {
425                 printf("%s <!-- %s -->\n", prefix.string(), String8(com16).string());
426             }
427             printf("%sE: %s%s (line=%d)\n", prefix.string(), elemNs.string(),
428                    String8(block->getElementName(&len)).string(),
429                    block->getLineNumber());
430             int N = block->getAttributeCount();
431             depth++;
432             prefix = make_prefix(depth);
433             for (i=0; i<N; i++) {
434                 uint32_t res = block->getAttributeNameResID(i);
435                 ns16 = block->getAttributeNamespace(i, &len);
436                 String8 ns = build_namespace(namespaces, ns16);
437                 String8 name(block->getAttributeName(i, &len));
438                 printf("%sA: ", prefix.string());
439                 if (res) {
440                     printf("%s%s(0x%08x)", ns.string(), name.string(), res);
441                 } else {
442                     printf("%s%s", ns.string(), name.string());
443                 }
444                 Res_value value;
445                 block->getAttributeValue(i, &value);
446                 if (value.dataType == Res_value::TYPE_NULL) {
447                     printf("=(null)");
448                 } else if (value.dataType == Res_value::TYPE_REFERENCE) {
449                     printf("=@0x%x", (int)value.data);
450                 } else if (value.dataType == Res_value::TYPE_ATTRIBUTE) {
451                     printf("=?0x%x", (int)value.data);
452                 } else if (value.dataType == Res_value::TYPE_STRING) {
453                     printf("=\"%s\"",
454                             ResTable::normalizeForOutput(String8(block->getAttributeStringValue(i,
455                                         &len)).string()).string());
456                 } else {
457                     printf("=(type 0x%x)0x%x", (int)value.dataType, (int)value.data);
458                 }
459                 const char16_t* val = block->getAttributeStringValue(i, &len);
460                 if (val != NULL) {
461                     printf(" (Raw: \"%s\")", ResTable::normalizeForOutput(String8(val).string()).
462                             string());
463                 }
464                 printf("\n");
465             }
466         } else if (code == ResXMLTree::END_TAG) {
467             depth--;
468         } else if (code == ResXMLTree::START_NAMESPACE) {
469             namespace_entry ns;
470             size_t len;
471             const uint16_t* prefix16 = block->getNamespacePrefix(&len);
472             if (prefix16) {
473                 ns.prefix = String8(prefix16);
474             } else {
475                 ns.prefix = "<DEF>";
476             }
477             ns.uri = String8(block->getNamespaceUri(&len));
478             namespaces.push(ns);
479             printf("%sN: %s=%s\n", prefix.string(), ns.prefix.string(),
480                     ns.uri.string());
481             depth++;
482         } else if (code == ResXMLTree::END_NAMESPACE) {
483             depth--;
484             const namespace_entry& ns = namespaces.top();
485             size_t len;
486             const uint16_t* prefix16 = block->getNamespacePrefix(&len);
487             String8 pr;
488             if (prefix16) {
489                 pr = String8(prefix16);
490             } else {
491                 pr = "<DEF>";
492             }
493             if (ns.prefix != pr) {
494                 prefix = make_prefix(depth);
495                 printf("%s*** BAD END NS PREFIX: found=%s, expected=%s\n",
496                         prefix.string(), pr.string(), ns.prefix.string());
497             }
498             String8 uri = String8(block->getNamespaceUri(&len));
499             if (ns.uri != uri) {
500                 prefix = make_prefix(depth);
501                 printf("%s *** BAD END NS URI: found=%s, expected=%s\n",
502                         prefix.string(), uri.string(), ns.uri.string());
503             }
504             namespaces.pop();
505         } else if (code == ResXMLTree::TEXT) {
506             size_t len;
507             printf("%sC: \"%s\"\n", prefix.string(), String8(block->getText(&len)).string());
508         }
509     }
510 
511     block->restart();
512 }
513 
parseXMLResource(const sp<AaptFile> & file,ResXMLTree * outTree,bool stripAll,bool keepComments,const char ** cDataTags)514 status_t parseXMLResource(const sp<AaptFile>& file, ResXMLTree* outTree,
515                           bool stripAll, bool keepComments,
516                           const char** cDataTags)
517 {
518     sp<XMLNode> root = XMLNode::parse(file);
519     if (root == NULL) {
520         return UNKNOWN_ERROR;
521     }
522     root->removeWhitespace(stripAll, cDataTags);
523 
524     NOISY(printf("Input XML from %s:\n", (const char*)file->getPrintableSource()));
525     NOISY(root->print());
526     sp<AaptFile> rsc = new AaptFile(String8(), AaptGroupEntry(), String8());
527     status_t err = root->flatten(rsc, !keepComments, false);
528     if (err != NO_ERROR) {
529         return err;
530     }
531     err = outTree->setTo(rsc->getData(), rsc->getSize(), true);
532     if (err != NO_ERROR) {
533         return err;
534     }
535 
536     NOISY(printf("Output XML:\n"));
537     NOISY(printXMLBlock(outTree));
538 
539     return NO_ERROR;
540 }
541 
parse(const sp<AaptFile> & file)542 sp<XMLNode> XMLNode::parse(const sp<AaptFile>& file)
543 {
544     char buf[16384];
545     int fd = open(file->getSourceFile().string(), O_RDONLY | O_BINARY);
546     if (fd < 0) {
547         SourcePos(file->getSourceFile(), -1).error("Unable to open file for read: %s",
548                 strerror(errno));
549         return NULL;
550     }
551 
552     XML_Parser parser = XML_ParserCreateNS(NULL, 1);
553     ParseState state;
554     state.filename = file->getPrintableSource();
555     state.parser = parser;
556     XML_SetUserData(parser, &state);
557     XML_SetElementHandler(parser, startElement, endElement);
558     XML_SetNamespaceDeclHandler(parser, startNamespace, endNamespace);
559     XML_SetCharacterDataHandler(parser, characterData);
560     XML_SetCommentHandler(parser, commentData);
561 
562     ssize_t len;
563     bool done;
564     do {
565         len = read(fd, buf, sizeof(buf));
566         done = len < (ssize_t)sizeof(buf);
567         if (len < 0) {
568             SourcePos(file->getSourceFile(), -1).error("Error reading file: %s\n", strerror(errno));
569             close(fd);
570             return NULL;
571         }
572         if (XML_Parse(parser, buf, len, done) == XML_STATUS_ERROR) {
573             SourcePos(file->getSourceFile(), (int)XML_GetCurrentLineNumber(parser)).error(
574                     "Error parsing XML: %s\n", XML_ErrorString(XML_GetErrorCode(parser)));
575             close(fd);
576             return NULL;
577         }
578     } while (!done);
579 
580     XML_ParserFree(parser);
581     if (state.root == NULL) {
582         SourcePos(file->getSourceFile(), -1).error("No XML data generated when parsing");
583     }
584     close(fd);
585     return state.root;
586 }
587 
XMLNode(const String8 & filename,const String16 & s1,const String16 & s2,bool isNamespace)588 XMLNode::XMLNode(const String8& filename, const String16& s1, const String16& s2, bool isNamespace)
589     : mNextAttributeIndex(0x80000000)
590     , mFilename(filename)
591     , mStartLineNumber(0)
592     , mEndLineNumber(0)
593     , mUTF8(false)
594 {
595     if (isNamespace) {
596         mNamespacePrefix = s1;
597         mNamespaceUri = s2;
598     } else {
599         mNamespaceUri = s1;
600         mElementName = s2;
601     }
602 }
603 
XMLNode(const String8 & filename)604 XMLNode::XMLNode(const String8& filename)
605     : mFilename(filename)
606 {
607     memset(&mCharsValue, 0, sizeof(mCharsValue));
608 }
609 
getType() const610 XMLNode::type XMLNode::getType() const
611 {
612     if (mElementName.size() != 0) {
613         return TYPE_ELEMENT;
614     }
615     if (mNamespaceUri.size() != 0) {
616         return TYPE_NAMESPACE;
617     }
618     return TYPE_CDATA;
619 }
620 
getNamespacePrefix() const621 const String16& XMLNode::getNamespacePrefix() const
622 {
623     return mNamespacePrefix;
624 }
625 
getNamespaceUri() const626 const String16& XMLNode::getNamespaceUri() const
627 {
628     return mNamespaceUri;
629 }
630 
getElementNamespace() const631 const String16& XMLNode::getElementNamespace() const
632 {
633     return mNamespaceUri;
634 }
635 
getElementName() const636 const String16& XMLNode::getElementName() const
637 {
638     return mElementName;
639 }
640 
getChildren() const641 const Vector<sp<XMLNode> >& XMLNode::getChildren() const
642 {
643     return mChildren;
644 }
645 
getFilename() const646 const String8& XMLNode::getFilename() const
647 {
648     return mFilename;
649 }
650 
651 const Vector<XMLNode::attribute_entry>&
getAttributes() const652     XMLNode::getAttributes() const
653 {
654     return mAttributes;
655 }
656 
getAttribute(const String16 & ns,const String16 & name) const657 const XMLNode::attribute_entry* XMLNode::getAttribute(const String16& ns,
658         const String16& name) const
659 {
660     for (size_t i=0; i<mAttributes.size(); i++) {
661         const attribute_entry& ae(mAttributes.itemAt(i));
662         if (ae.ns == ns && ae.name == name) {
663             return &ae;
664         }
665     }
666 
667     return NULL;
668 }
669 
editAttribute(const String16 & ns,const String16 & name)670 XMLNode::attribute_entry* XMLNode::editAttribute(const String16& ns,
671         const String16& name)
672 {
673     for (size_t i=0; i<mAttributes.size(); i++) {
674         attribute_entry * ae = &mAttributes.editItemAt(i);
675         if (ae->ns == ns && ae->name == name) {
676             return ae;
677         }
678     }
679 
680     return NULL;
681 }
682 
getCData() const683 const String16& XMLNode::getCData() const
684 {
685     return mChars;
686 }
687 
getComment() const688 const String16& XMLNode::getComment() const
689 {
690     return mComment;
691 }
692 
getStartLineNumber() const693 int32_t XMLNode::getStartLineNumber() const
694 {
695     return mStartLineNumber;
696 }
697 
getEndLineNumber() const698 int32_t XMLNode::getEndLineNumber() const
699 {
700     return mEndLineNumber;
701 }
702 
searchElement(const String16 & tagNamespace,const String16 & tagName)703 sp<XMLNode> XMLNode::searchElement(const String16& tagNamespace, const String16& tagName)
704 {
705     if (getType() == XMLNode::TYPE_ELEMENT
706             && mNamespaceUri == tagNamespace
707             && mElementName == tagName) {
708         return this;
709     }
710 
711     for (size_t i=0; i<mChildren.size(); i++) {
712         sp<XMLNode> found = mChildren.itemAt(i)->searchElement(tagNamespace, tagName);
713         if (found != NULL) {
714             return found;
715         }
716     }
717 
718     return NULL;
719 }
720 
getChildElement(const String16 & tagNamespace,const String16 & tagName)721 sp<XMLNode> XMLNode::getChildElement(const String16& tagNamespace, const String16& tagName)
722 {
723     for (size_t i=0; i<mChildren.size(); i++) {
724         sp<XMLNode> child = mChildren.itemAt(i);
725         if (child->getType() == XMLNode::TYPE_ELEMENT
726                 && child->mNamespaceUri == tagNamespace
727                 && child->mElementName == tagName) {
728             return child;
729         }
730     }
731 
732     return NULL;
733 }
734 
addChild(const sp<XMLNode> & child)735 status_t XMLNode::addChild(const sp<XMLNode>& child)
736 {
737     if (getType() == TYPE_CDATA) {
738         SourcePos(mFilename, child->getStartLineNumber()).error("Child to CDATA node.");
739         return UNKNOWN_ERROR;
740     }
741     //printf("Adding child %p to parent %p\n", child.get(), this);
742     mChildren.add(child);
743     return NO_ERROR;
744 }
745 
insertChildAt(const sp<XMLNode> & child,size_t index)746 status_t XMLNode::insertChildAt(const sp<XMLNode>& child, size_t index)
747 {
748     if (getType() == TYPE_CDATA) {
749         SourcePos(mFilename, child->getStartLineNumber()).error("Child to CDATA node.");
750         return UNKNOWN_ERROR;
751     }
752     //printf("Adding child %p to parent %p\n", child.get(), this);
753     mChildren.insertAt(child, index);
754     return NO_ERROR;
755 }
756 
addAttribute(const String16 & ns,const String16 & name,const String16 & value)757 status_t XMLNode::addAttribute(const String16& ns, const String16& name,
758                                const String16& value)
759 {
760     if (getType() == TYPE_CDATA) {
761         SourcePos(mFilename, getStartLineNumber()).error("Child to CDATA node.");
762         return UNKNOWN_ERROR;
763     }
764     attribute_entry e;
765     e.index = mNextAttributeIndex++;
766     e.ns = ns;
767     e.name = name;
768     e.string = value;
769     mAttributes.add(e);
770     mAttributeOrder.add(e.index, mAttributes.size()-1);
771     return NO_ERROR;
772 }
773 
setAttributeResID(size_t attrIdx,uint32_t resId)774 void XMLNode::setAttributeResID(size_t attrIdx, uint32_t resId)
775 {
776     attribute_entry& e = mAttributes.editItemAt(attrIdx);
777     if (e.nameResId) {
778         mAttributeOrder.removeItem(e.nameResId);
779     } else {
780         mAttributeOrder.removeItem(e.index);
781     }
782     NOISY(printf("Elem %s %s=\"%s\": set res id = 0x%08x\n",
783             String8(getElementName()).string(),
784             String8(mAttributes.itemAt(attrIdx).name).string(),
785             String8(mAttributes.itemAt(attrIdx).string).string(),
786             resId));
787     mAttributes.editItemAt(attrIdx).nameResId = resId;
788     mAttributeOrder.add(resId, attrIdx);
789 }
790 
appendChars(const String16 & chars)791 status_t XMLNode::appendChars(const String16& chars)
792 {
793     if (getType() != TYPE_CDATA) {
794         SourcePos(mFilename, getStartLineNumber()).error("Adding characters to element node.");
795         return UNKNOWN_ERROR;
796     }
797     mChars.append(chars);
798     return NO_ERROR;
799 }
800 
appendComment(const String16 & comment)801 status_t XMLNode::appendComment(const String16& comment)
802 {
803     if (mComment.size() > 0) {
804         mComment.append(String16("\n"));
805     }
806     mComment.append(comment);
807     return NO_ERROR;
808 }
809 
setStartLineNumber(int32_t line)810 void XMLNode::setStartLineNumber(int32_t line)
811 {
812     mStartLineNumber = line;
813 }
814 
setEndLineNumber(int32_t line)815 void XMLNode::setEndLineNumber(int32_t line)
816 {
817     mEndLineNumber = line;
818 }
819 
removeWhitespace(bool stripAll,const char ** cDataTags)820 void XMLNode::removeWhitespace(bool stripAll, const char** cDataTags)
821 {
822     //printf("Removing whitespace in %s\n", String8(mElementName).string());
823     size_t N = mChildren.size();
824     if (cDataTags) {
825         String8 tag(mElementName);
826         const char** p = cDataTags;
827         while (*p) {
828             if (tag == *p) {
829                 stripAll = false;
830                 break;
831             }
832         }
833     }
834     for (size_t i=0; i<N; i++) {
835         sp<XMLNode> node = mChildren.itemAt(i);
836         if (node->getType() == TYPE_CDATA) {
837             // This is a CDATA node...
838             const char16_t* p = node->mChars.string();
839             while (*p != 0 && *p < 128 && isspace(*p)) {
840                 p++;
841             }
842             //printf("Space ends at %d in \"%s\"\n",
843             //       (int)(p-node->mChars.string()),
844             //       String8(node->mChars).string());
845             if (*p == 0) {
846                 if (stripAll) {
847                     // Remove this node!
848                     mChildren.removeAt(i);
849                     N--;
850                     i--;
851                 } else {
852                     node->mChars = String16(" ");
853                 }
854             } else {
855                 // Compact leading/trailing whitespace.
856                 const char16_t* e = node->mChars.string()+node->mChars.size()-1;
857                 while (e > p && *e < 128 && isspace(*e)) {
858                     e--;
859                 }
860                 if (p > node->mChars.string()) {
861                     p--;
862                 }
863                 if (e < (node->mChars.string()+node->mChars.size()-1)) {
864                     e++;
865                 }
866                 if (p > node->mChars.string() ||
867                     e < (node->mChars.string()+node->mChars.size()-1)) {
868                     String16 tmp(p, e-p+1);
869                     node->mChars = tmp;
870                 }
871             }
872         } else {
873             node->removeWhitespace(stripAll, cDataTags);
874         }
875     }
876 }
877 
parseValues(const sp<AaptAssets> & assets,ResourceTable * table)878 status_t XMLNode::parseValues(const sp<AaptAssets>& assets,
879                               ResourceTable* table)
880 {
881     bool hasErrors = false;
882 
883     if (getType() == TYPE_ELEMENT) {
884         const size_t N = mAttributes.size();
885         String16 defPackage(assets->getPackage());
886         for (size_t i=0; i<N; i++) {
887             attribute_entry& e = mAttributes.editItemAt(i);
888             AccessorCookie ac(SourcePos(mFilename, getStartLineNumber()), String8(e.name),
889                     String8(e.string));
890             table->setCurrentXmlPos(SourcePos(mFilename, getStartLineNumber()));
891             if (!assets->getIncludedResources()
892                     .stringToValue(&e.value, &e.string,
893                                   e.string.string(), e.string.size(), true, true,
894                                   e.nameResId, NULL, &defPackage, table, &ac)) {
895                 hasErrors = true;
896             }
897             NOISY(printf("Attr %s: type=0x%x, str=%s\n",
898                    String8(e.name).string(), e.value.dataType,
899                    String8(e.string).string()));
900         }
901     }
902     const size_t N = mChildren.size();
903     for (size_t i=0; i<N; i++) {
904         status_t err = mChildren.itemAt(i)->parseValues(assets, table);
905         if (err != NO_ERROR) {
906             hasErrors = true;
907         }
908     }
909     return hasErrors ? UNKNOWN_ERROR : NO_ERROR;
910 }
911 
assignResourceIds(const sp<AaptAssets> & assets,const ResourceTable * table)912 status_t XMLNode::assignResourceIds(const sp<AaptAssets>& assets,
913                                     const ResourceTable* table)
914 {
915     bool hasErrors = false;
916 
917     if (getType() == TYPE_ELEMENT) {
918         String16 attr("attr");
919         const char* errorMsg;
920         const size_t N = mAttributes.size();
921         for (size_t i=0; i<N; i++) {
922             const attribute_entry& e = mAttributes.itemAt(i);
923             if (e.ns.size() <= 0) continue;
924             bool nsIsPublic;
925             String16 pkg(getNamespaceResourcePackage(e.ns, &nsIsPublic));
926             NOISY(printf("Elem %s %s=\"%s\": namespace(%s) %s ===> %s\n",
927                     String8(getElementName()).string(),
928                     String8(e.name).string(),
929                     String8(e.string).string(),
930                     String8(e.ns).string(),
931                     (nsIsPublic) ? "public" : "private",
932                     String8(pkg).string()));
933             if (pkg.size() <= 0) continue;
934             uint32_t res = table != NULL
935                 ? table->getResId(e.name, &attr, &pkg, &errorMsg, nsIsPublic)
936                 : assets->getIncludedResources().
937                     identifierForName(e.name.string(), e.name.size(),
938                                       attr.string(), attr.size(),
939                                       pkg.string(), pkg.size());
940             if (res != 0) {
941                 NOISY(printf("XML attribute name %s: resid=0x%08x\n",
942                              String8(e.name).string(), res));
943                 setAttributeResID(i, res);
944             } else {
945                 SourcePos(mFilename, getStartLineNumber()).error(
946                         "No resource identifier found for attribute '%s' in package '%s'\n",
947                         String8(e.name).string(), String8(pkg).string());
948                 hasErrors = true;
949             }
950         }
951     }
952     const size_t N = mChildren.size();
953     for (size_t i=0; i<N; i++) {
954         status_t err = mChildren.itemAt(i)->assignResourceIds(assets, table);
955         if (err < NO_ERROR) {
956             hasErrors = true;
957         }
958     }
959 
960     return hasErrors ? UNKNOWN_ERROR : NO_ERROR;
961 }
962 
flatten(const sp<AaptFile> & dest,bool stripComments,bool stripRawValues) const963 status_t XMLNode::flatten(const sp<AaptFile>& dest,
964         bool stripComments, bool stripRawValues) const
965 {
966     StringPool strings = StringPool(false, mUTF8);
967     Vector<uint32_t> resids;
968 
969     // First collect just the strings for attribute names that have a
970     // resource ID assigned to them.  This ensures that the resource ID
971     // array is compact, and makes it easier to deal with attribute names
972     // in different namespaces (and thus with different resource IDs).
973     collect_resid_strings(&strings, &resids);
974 
975     // Next collect all remainibng strings.
976     collect_strings(&strings, &resids, stripComments, stripRawValues);
977 
978 #if 0  // No longer compiles
979     NOISY(printf("Found strings:\n");
980         const size_t N = strings.size();
981         for (size_t i=0; i<N; i++) {
982             printf("%s\n", String8(strings.entryAt(i).string).string());
983         }
984     );
985 #endif
986 
987     sp<AaptFile> stringPool = strings.createStringBlock();
988     NOISY(aout << "String pool:"
989           << HexDump(stringPool->getData(), stringPool->getSize()) << endl);
990 
991     ResXMLTree_header header;
992     memset(&header, 0, sizeof(header));
993     header.header.type = htods(RES_XML_TYPE);
994     header.header.headerSize = htods(sizeof(header));
995 
996     const size_t basePos = dest->getSize();
997     dest->writeData(&header, sizeof(header));
998     dest->writeData(stringPool->getData(), stringPool->getSize());
999 
1000     // If we have resource IDs, write them.
1001     if (resids.size() > 0) {
1002         const size_t resIdsPos = dest->getSize();
1003         const size_t resIdsSize =
1004             sizeof(ResChunk_header)+(sizeof(uint32_t)*resids.size());
1005         ResChunk_header* idsHeader = (ResChunk_header*)
1006             (((const uint8_t*)dest->editData(resIdsPos+resIdsSize))+resIdsPos);
1007         idsHeader->type = htods(RES_XML_RESOURCE_MAP_TYPE);
1008         idsHeader->headerSize = htods(sizeof(*idsHeader));
1009         idsHeader->size = htodl(resIdsSize);
1010         uint32_t* ids = (uint32_t*)(idsHeader+1);
1011         for (size_t i=0; i<resids.size(); i++) {
1012             *ids++ = htodl(resids[i]);
1013         }
1014     }
1015 
1016     flatten_node(strings, dest, stripComments, stripRawValues);
1017 
1018     void* data = dest->editData();
1019     ResXMLTree_header* hd = (ResXMLTree_header*)(((uint8_t*)data)+basePos);
1020     size_t size = dest->getSize()-basePos;
1021     hd->header.size = htodl(dest->getSize()-basePos);
1022 
1023     NOISY(aout << "XML resource:"
1024           << HexDump(dest->getData(), dest->getSize()) << endl);
1025 
1026     #if PRINT_STRING_METRICS
1027     fprintf(stderr, "**** total xml size: %d / %d%% strings (in %s)\n",
1028         dest->getSize(), (stringPool->getSize()*100)/dest->getSize(),
1029         dest->getPath().string());
1030     #endif
1031 
1032     return NO_ERROR;
1033 }
1034 
print(int indent)1035 void XMLNode::print(int indent)
1036 {
1037     String8 prefix;
1038     int i;
1039     for (i=0; i<indent; i++) {
1040         prefix.append("  ");
1041     }
1042     if (getType() == TYPE_ELEMENT) {
1043         String8 elemNs(getNamespaceUri());
1044         if (elemNs.size() > 0) {
1045             elemNs.append(":");
1046         }
1047         printf("%s E: %s%s", prefix.string(),
1048                elemNs.string(), String8(getElementName()).string());
1049         int N = mAttributes.size();
1050         for (i=0; i<N; i++) {
1051             ssize_t idx = mAttributeOrder.valueAt(i);
1052             if (i == 0) {
1053                 printf(" / ");
1054             } else {
1055                 printf(", ");
1056             }
1057             const attribute_entry& attr = mAttributes.itemAt(idx);
1058             String8 attrNs(attr.ns);
1059             if (attrNs.size() > 0) {
1060                 attrNs.append(":");
1061             }
1062             if (attr.nameResId) {
1063                 printf("%s%s(0x%08x)", attrNs.string(),
1064                        String8(attr.name).string(), attr.nameResId);
1065             } else {
1066                 printf("%s%s", attrNs.string(), String8(attr.name).string());
1067             }
1068             printf("=%s", String8(attr.string).string());
1069         }
1070         printf("\n");
1071     } else if (getType() == TYPE_NAMESPACE) {
1072         printf("%s N: %s=%s\n", prefix.string(),
1073                getNamespacePrefix().size() > 0
1074                     ? String8(getNamespacePrefix()).string() : "<DEF>",
1075                String8(getNamespaceUri()).string());
1076     } else {
1077         printf("%s C: \"%s\"\n", prefix.string(), String8(getCData()).string());
1078     }
1079     int N = mChildren.size();
1080     for (i=0; i<N; i++) {
1081         mChildren.itemAt(i)->print(indent+1);
1082     }
1083 }
1084 
splitName(const char * name,String16 * outNs,String16 * outName)1085 static void splitName(const char* name, String16* outNs, String16* outName)
1086 {
1087     const char* p = name;
1088     while (*p != 0 && *p != 1) {
1089         p++;
1090     }
1091     if (*p == 0) {
1092         *outNs = String16();
1093         *outName = String16(name);
1094     } else {
1095         *outNs = String16(name, (p-name));
1096         *outName = String16(p+1);
1097     }
1098 }
1099 
1100 void XMLCALL
startNamespace(void * userData,const char * prefix,const char * uri)1101 XMLNode::startNamespace(void *userData, const char *prefix, const char *uri)
1102 {
1103     NOISY_PARSE(printf("Start Namespace: %s %s\n", prefix, uri));
1104     ParseState* st = (ParseState*)userData;
1105     sp<XMLNode> node = XMLNode::newNamespace(st->filename,
1106             String16(prefix != NULL ? prefix : ""), String16(uri));
1107     node->setStartLineNumber(XML_GetCurrentLineNumber(st->parser));
1108     if (st->stack.size() > 0) {
1109         st->stack.itemAt(st->stack.size()-1)->addChild(node);
1110     } else {
1111         st->root = node;
1112     }
1113     st->stack.push(node);
1114 }
1115 
1116 void XMLCALL
startElement(void * userData,const char * name,const char ** atts)1117 XMLNode::startElement(void *userData, const char *name, const char **atts)
1118 {
1119     NOISY_PARSE(printf("Start Element: %s\n", name));
1120     ParseState* st = (ParseState*)userData;
1121     String16 ns16, name16;
1122     splitName(name, &ns16, &name16);
1123     sp<XMLNode> node = XMLNode::newElement(st->filename, ns16, name16);
1124     node->setStartLineNumber(XML_GetCurrentLineNumber(st->parser));
1125     if (st->pendingComment.size() > 0) {
1126         node->appendComment(st->pendingComment);
1127         st->pendingComment = String16();
1128     }
1129     if (st->stack.size() > 0) {
1130         st->stack.itemAt(st->stack.size()-1)->addChild(node);
1131     } else {
1132         st->root = node;
1133     }
1134     st->stack.push(node);
1135 
1136     for (int i = 0; atts[i]; i += 2) {
1137         splitName(atts[i], &ns16, &name16);
1138         node->addAttribute(ns16, name16, String16(atts[i+1]));
1139     }
1140 }
1141 
1142 void XMLCALL
characterData(void * userData,const XML_Char * s,int len)1143 XMLNode::characterData(void *userData, const XML_Char *s, int len)
1144 {
1145     NOISY_PARSE(printf("CDATA: \"%s\"\n", String8(s, len).string()));
1146     ParseState* st = (ParseState*)userData;
1147     sp<XMLNode> node = NULL;
1148     if (st->stack.size() == 0) {
1149         return;
1150     }
1151     sp<XMLNode> parent = st->stack.itemAt(st->stack.size()-1);
1152     if (parent != NULL && parent->getChildren().size() > 0) {
1153         node = parent->getChildren()[parent->getChildren().size()-1];
1154         if (node->getType() != TYPE_CDATA) {
1155             // Last node is not CDATA, need to make a new node.
1156             node = NULL;
1157         }
1158     }
1159 
1160     if (node == NULL) {
1161         node = XMLNode::newCData(st->filename);
1162         node->setStartLineNumber(XML_GetCurrentLineNumber(st->parser));
1163         parent->addChild(node);
1164     }
1165 
1166     node->appendChars(String16(s, len));
1167 }
1168 
1169 void XMLCALL
endElement(void * userData,const char * name)1170 XMLNode::endElement(void *userData, const char *name)
1171 {
1172     NOISY_PARSE(printf("End Element: %s\n", name));
1173     ParseState* st = (ParseState*)userData;
1174     sp<XMLNode> node = st->stack.itemAt(st->stack.size()-1);
1175     node->setEndLineNumber(XML_GetCurrentLineNumber(st->parser));
1176     if (st->pendingComment.size() > 0) {
1177         node->appendComment(st->pendingComment);
1178         st->pendingComment = String16();
1179     }
1180     String16 ns16, name16;
1181     splitName(name, &ns16, &name16);
1182     LOG_ALWAYS_FATAL_IF(node->getElementNamespace() != ns16
1183                         || node->getElementName() != name16,
1184                         "Bad end element %s", name);
1185     st->stack.pop();
1186 }
1187 
1188 void XMLCALL
endNamespace(void * userData,const char * prefix)1189 XMLNode::endNamespace(void *userData, const char *prefix)
1190 {
1191     const char* nonNullPrefix = prefix != NULL ? prefix : "";
1192     NOISY_PARSE(printf("End Namespace: %s\n", prefix));
1193     ParseState* st = (ParseState*)userData;
1194     sp<XMLNode> node = st->stack.itemAt(st->stack.size()-1);
1195     node->setEndLineNumber(XML_GetCurrentLineNumber(st->parser));
1196     LOG_ALWAYS_FATAL_IF(node->getNamespacePrefix() != String16(nonNullPrefix),
1197                         "Bad end namespace %s", prefix);
1198     st->stack.pop();
1199 }
1200 
1201 void XMLCALL
commentData(void * userData,const char * comment)1202 XMLNode::commentData(void *userData, const char *comment)
1203 {
1204     NOISY_PARSE(printf("Comment: %s\n", comment));
1205     ParseState* st = (ParseState*)userData;
1206     if (st->pendingComment.size() > 0) {
1207         st->pendingComment.append(String16("\n"));
1208     }
1209     st->pendingComment.append(String16(comment));
1210 }
1211 
collect_strings(StringPool * dest,Vector<uint32_t> * outResIds,bool stripComments,bool stripRawValues) const1212 status_t XMLNode::collect_strings(StringPool* dest, Vector<uint32_t>* outResIds,
1213         bool stripComments, bool stripRawValues) const
1214 {
1215     collect_attr_strings(dest, outResIds, true);
1216 
1217     int i;
1218     if (mNamespacePrefix.size() > 0) {
1219         dest->add(mNamespacePrefix, true);
1220     }
1221     if (mNamespaceUri.size() > 0) {
1222         dest->add(mNamespaceUri, true);
1223     }
1224     if (mElementName.size() > 0) {
1225         dest->add(mElementName, true);
1226     }
1227 
1228     if (!stripComments && mComment.size() > 0) {
1229         dest->add(mComment, true);
1230     }
1231 
1232     const int NA = mAttributes.size();
1233 
1234     for (i=0; i<NA; i++) {
1235         const attribute_entry& ae = mAttributes.itemAt(i);
1236         if (ae.ns.size() > 0) {
1237             dest->add(ae.ns, true);
1238         }
1239         if (!stripRawValues || ae.needStringValue()) {
1240             dest->add(ae.string, true);
1241         }
1242         /*
1243         if (ae.value.dataType == Res_value::TYPE_NULL
1244                 || ae.value.dataType == Res_value::TYPE_STRING) {
1245             dest->add(ae.string, true);
1246         }
1247         */
1248     }
1249 
1250     if (mElementName.size() == 0) {
1251         // If not an element, include the CDATA, even if it is empty.
1252         dest->add(mChars, true);
1253     }
1254 
1255     const int NC = mChildren.size();
1256 
1257     for (i=0; i<NC; i++) {
1258         mChildren.itemAt(i)->collect_strings(dest, outResIds,
1259                 stripComments, stripRawValues);
1260     }
1261 
1262     return NO_ERROR;
1263 }
1264 
collect_attr_strings(StringPool * outPool,Vector<uint32_t> * outResIds,bool allAttrs) const1265 status_t XMLNode::collect_attr_strings(StringPool* outPool,
1266         Vector<uint32_t>* outResIds, bool allAttrs) const {
1267     const int NA = mAttributes.size();
1268 
1269     for (int i=0; i<NA; i++) {
1270         const attribute_entry& attr = mAttributes.itemAt(i);
1271         uint32_t id = attr.nameResId;
1272         if (id || allAttrs) {
1273             // See if we have already assigned this resource ID to a pooled
1274             // string...
1275             const Vector<size_t>* indices = outPool->offsetsForString(attr.name);
1276             ssize_t idx = -1;
1277             if (indices != NULL) {
1278                 const int NJ = indices->size();
1279                 const size_t NR = outResIds->size();
1280                 for (int j=0; j<NJ; j++) {
1281                     size_t strIdx = indices->itemAt(j);
1282                     if (strIdx >= NR) {
1283                         if (id == 0) {
1284                             // We don't need to assign a resource ID for this one.
1285                             idx = strIdx;
1286                             break;
1287                         }
1288                         // Just ignore strings that are out of range of
1289                         // the currently assigned resource IDs...  we add
1290                         // strings as we assign the first ID.
1291                     } else if (outResIds->itemAt(strIdx) == id) {
1292                         idx = strIdx;
1293                         break;
1294                     }
1295                 }
1296             }
1297             if (idx < 0) {
1298                 idx = outPool->add(attr.name);
1299                 NOISY(printf("Adding attr %s (resid 0x%08x) to pool: idx=%d\n",
1300                         String8(attr.name).string(), id, idx));
1301                 if (id != 0) {
1302                     while ((ssize_t)outResIds->size() <= idx) {
1303                         outResIds->add(0);
1304                     }
1305                     outResIds->replaceAt(id, idx);
1306                 }
1307             }
1308             attr.namePoolIdx = idx;
1309             NOISY(printf("String %s offset=0x%08x\n",
1310                          String8(attr.name).string(), idx));
1311         }
1312     }
1313 
1314     return NO_ERROR;
1315 }
1316 
collect_resid_strings(StringPool * outPool,Vector<uint32_t> * outResIds) const1317 status_t XMLNode::collect_resid_strings(StringPool* outPool,
1318         Vector<uint32_t>* outResIds) const
1319 {
1320     collect_attr_strings(outPool, outResIds, false);
1321 
1322     const int NC = mChildren.size();
1323 
1324     for (int i=0; i<NC; i++) {
1325         mChildren.itemAt(i)->collect_resid_strings(outPool, outResIds);
1326     }
1327 
1328     return NO_ERROR;
1329 }
1330 
flatten_node(const StringPool & strings,const sp<AaptFile> & dest,bool stripComments,bool stripRawValues) const1331 status_t XMLNode::flatten_node(const StringPool& strings, const sp<AaptFile>& dest,
1332         bool stripComments, bool stripRawValues) const
1333 {
1334     ResXMLTree_node node;
1335     ResXMLTree_cdataExt cdataExt;
1336     ResXMLTree_namespaceExt namespaceExt;
1337     ResXMLTree_attrExt attrExt;
1338     const void* extData = NULL;
1339     size_t extSize = 0;
1340     ResXMLTree_attribute attr;
1341 
1342     const size_t NA = mAttributes.size();
1343     const size_t NC = mChildren.size();
1344     size_t i;
1345 
1346     LOG_ALWAYS_FATAL_IF(NA != mAttributeOrder.size(), "Attributes messed up!");
1347 
1348     const String16 id16("id");
1349     const String16 class16("class");
1350     const String16 style16("style");
1351 
1352     const type type = getType();
1353 
1354     memset(&node, 0, sizeof(node));
1355     memset(&attr, 0, sizeof(attr));
1356     node.header.headerSize = htods(sizeof(node));
1357     node.lineNumber = htodl(getStartLineNumber());
1358     if (!stripComments) {
1359         node.comment.index = htodl(
1360             mComment.size() > 0 ? strings.offsetForString(mComment) : -1);
1361         //if (mComment.size() > 0) {
1362         //  printf("Flattening comment: %s\n", String8(mComment).string());
1363         //}
1364     } else {
1365         node.comment.index = htodl((uint32_t)-1);
1366     }
1367     if (type == TYPE_ELEMENT) {
1368         node.header.type = htods(RES_XML_START_ELEMENT_TYPE);
1369         extData = &attrExt;
1370         extSize = sizeof(attrExt);
1371         memset(&attrExt, 0, sizeof(attrExt));
1372         if (mNamespaceUri.size() > 0) {
1373             attrExt.ns.index = htodl(strings.offsetForString(mNamespaceUri));
1374         } else {
1375             attrExt.ns.index = htodl((uint32_t)-1);
1376         }
1377         attrExt.name.index = htodl(strings.offsetForString(mElementName));
1378         attrExt.attributeStart = htods(sizeof(attrExt));
1379         attrExt.attributeSize = htods(sizeof(attr));
1380         attrExt.attributeCount = htods(NA);
1381         attrExt.idIndex = htods(0);
1382         attrExt.classIndex = htods(0);
1383         attrExt.styleIndex = htods(0);
1384         for (i=0; i<NA; i++) {
1385             ssize_t idx = mAttributeOrder.valueAt(i);
1386             const attribute_entry& ae = mAttributes.itemAt(idx);
1387             if (ae.ns.size() == 0) {
1388                 if (ae.name == id16) {
1389                     attrExt.idIndex = htods(i+1);
1390                 } else if (ae.name == class16) {
1391                     attrExt.classIndex = htods(i+1);
1392                 } else if (ae.name == style16) {
1393                     attrExt.styleIndex = htods(i+1);
1394                 }
1395             }
1396         }
1397     } else if (type == TYPE_NAMESPACE) {
1398         node.header.type = htods(RES_XML_START_NAMESPACE_TYPE);
1399         extData = &namespaceExt;
1400         extSize = sizeof(namespaceExt);
1401         memset(&namespaceExt, 0, sizeof(namespaceExt));
1402         if (mNamespacePrefix.size() > 0) {
1403             namespaceExt.prefix.index = htodl(strings.offsetForString(mNamespacePrefix));
1404         } else {
1405             namespaceExt.prefix.index = htodl((uint32_t)-1);
1406         }
1407         namespaceExt.prefix.index = htodl(strings.offsetForString(mNamespacePrefix));
1408         namespaceExt.uri.index = htodl(strings.offsetForString(mNamespaceUri));
1409         LOG_ALWAYS_FATAL_IF(NA != 0, "Namespace nodes can't have attributes!");
1410     } else if (type == TYPE_CDATA) {
1411         node.header.type = htods(RES_XML_CDATA_TYPE);
1412         extData = &cdataExt;
1413         extSize = sizeof(cdataExt);
1414         memset(&cdataExt, 0, sizeof(cdataExt));
1415         cdataExt.data.index = htodl(strings.offsetForString(mChars));
1416         cdataExt.typedData.size = htods(sizeof(cdataExt.typedData));
1417         cdataExt.typedData.res0 = 0;
1418         cdataExt.typedData.dataType = mCharsValue.dataType;
1419         cdataExt.typedData.data = htodl(mCharsValue.data);
1420         LOG_ALWAYS_FATAL_IF(NA != 0, "CDATA nodes can't have attributes!");
1421     }
1422 
1423     node.header.size = htodl(sizeof(node) + extSize + (sizeof(attr)*NA));
1424 
1425     dest->writeData(&node, sizeof(node));
1426     if (extSize > 0) {
1427         dest->writeData(extData, extSize);
1428     }
1429 
1430     for (i=0; i<NA; i++) {
1431         ssize_t idx = mAttributeOrder.valueAt(i);
1432         const attribute_entry& ae = mAttributes.itemAt(idx);
1433         if (ae.ns.size() > 0) {
1434             attr.ns.index = htodl(strings.offsetForString(ae.ns));
1435         } else {
1436             attr.ns.index = htodl((uint32_t)-1);
1437         }
1438         attr.name.index = htodl(ae.namePoolIdx);
1439 
1440         if (!stripRawValues || ae.needStringValue()) {
1441             attr.rawValue.index = htodl(strings.offsetForString(ae.string));
1442         } else {
1443             attr.rawValue.index = htodl((uint32_t)-1);
1444         }
1445         attr.typedValue.size = htods(sizeof(attr.typedValue));
1446         if (ae.value.dataType == Res_value::TYPE_NULL
1447                 || ae.value.dataType == Res_value::TYPE_STRING) {
1448             attr.typedValue.res0 = 0;
1449             attr.typedValue.dataType = Res_value::TYPE_STRING;
1450             attr.typedValue.data = htodl(strings.offsetForString(ae.string));
1451         } else {
1452             attr.typedValue.res0 = 0;
1453             attr.typedValue.dataType = ae.value.dataType;
1454             attr.typedValue.data = htodl(ae.value.data);
1455         }
1456         dest->writeData(&attr, sizeof(attr));
1457     }
1458 
1459     for (i=0; i<NC; i++) {
1460         status_t err = mChildren.itemAt(i)->flatten_node(strings, dest,
1461                 stripComments, stripRawValues);
1462         if (err != NO_ERROR) {
1463             return err;
1464         }
1465     }
1466 
1467     if (type == TYPE_ELEMENT) {
1468         ResXMLTree_endElementExt endElementExt;
1469         memset(&endElementExt, 0, sizeof(endElementExt));
1470         node.header.type = htods(RES_XML_END_ELEMENT_TYPE);
1471         node.header.size = htodl(sizeof(node)+sizeof(endElementExt));
1472         node.lineNumber = htodl(getEndLineNumber());
1473         node.comment.index = htodl((uint32_t)-1);
1474         endElementExt.ns.index = attrExt.ns.index;
1475         endElementExt.name.index = attrExt.name.index;
1476         dest->writeData(&node, sizeof(node));
1477         dest->writeData(&endElementExt, sizeof(endElementExt));
1478     } else if (type == TYPE_NAMESPACE) {
1479         node.header.type = htods(RES_XML_END_NAMESPACE_TYPE);
1480         node.lineNumber = htodl(getEndLineNumber());
1481         node.comment.index = htodl((uint32_t)-1);
1482         node.header.size = htodl(sizeof(node)+extSize);
1483         dest->writeData(&node, sizeof(node));
1484         dest->writeData(extData, extSize);
1485     }
1486 
1487     return NO_ERROR;
1488 }
1489