XMLNode.cpp

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