BinaryResourceParser.cpp revision e78fd617ec60139a973a01925fa7adad31febb39 
1/*
2 * Copyright (C) 2015 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#include "ResourceTable.h"
18#include "ResourceUtils.h"
19#include "ResourceValues.h"
20#include "Source.h"
21#include "ValueVisitor.h"
22
23#include "flatten/ResourceTypeExtensions.h"
24#include "unflatten/BinaryResourceParser.h"
25#include "unflatten/ResChunkPullParser.h"
26#include "util/Util.h"
27
28#include <androidfw/ResourceTypes.h>
29#include <androidfw/TypeWrappers.h>
30#include <base/macros.h>
31
32#include <map>
33#include <string>
34
35namespace aapt {
36
37using namespace android;
38
39/*
40 * Visitor that converts a reference's resource ID to a resource name,
41 * given a mapping from resource ID to resource name.
42 */
43class ReferenceIdToNameVisitor : public ValueVisitor {
44private:
45    const std::map<ResourceId, ResourceName>* mMapping;
46
47public:
48    using ValueVisitor::visit;
49
50    ReferenceIdToNameVisitor(const std::map<ResourceId, ResourceName>* mapping) :
51            mMapping(mapping) {
52        assert(mMapping);
53    }
54
55    void visit(Reference* reference) override {
56        if (!reference->id || !reference->id.value().isValid()) {
57            return;
58        }
59
60        ResourceId id = reference->id.value();
61        auto cacheIter = mMapping->find(id);
62        if (cacheIter != mMapping->end()) {
63            reference->name = cacheIter->second;
64            reference->id = {};
65        }
66    }
67};
68
69BinaryResourceParser::BinaryResourceParser(IAaptContext* context, ResourceTable* table,
70                                           const Source& source, const void* data, size_t len) :
71        mContext(context), mTable(table), mSource(source), mData(data), mDataLen(len) {
72}
73
74bool BinaryResourceParser::parse() {
75    ResChunkPullParser parser(mData, mDataLen);
76
77    bool error = false;
78    while(ResChunkPullParser::isGoodEvent(parser.next())) {
79        if (parser.getChunk()->type != android::RES_TABLE_TYPE) {
80            mContext->getDiagnostics()->warn(DiagMessage(mSource)
81                                             << "unknown chunk of type '"
82                                             << (int) parser.getChunk()->type << "'");
83            continue;
84        }
85
86        if (!parseTable(parser.getChunk())) {
87            error = true;
88        }
89    }
90
91    if (parser.getEvent() == ResChunkPullParser::Event::BadDocument) {
92        mContext->getDiagnostics()->error(DiagMessage(mSource)
93                                          << "corrupt resource table: "
94                                          << parser.getLastError());
95        return false;
96    }
97    return !error;
98}
99
100bool BinaryResourceParser::getSymbol(const void* data, ResourceNameRef* outSymbol) {
101    if (!mSymbolEntries || mSymbolEntryCount == 0) {
102        return false;
103    }
104
105    if ((uintptr_t) data < (uintptr_t) mData) {
106        return false;
107    }
108
109    // We only support 32 bit offsets right now.
110    const uintptr_t offset = (uintptr_t) data - (uintptr_t) mData;
111    if (offset > std::numeric_limits<uint32_t>::max()) {
112        return false;
113    }
114
115    for (size_t i = 0; i < mSymbolEntryCount; i++) {
116        if (util::deviceToHost32(mSymbolEntries[i].offset) == offset) {
117            // This offset is a symbol!
118            const StringPiece16 str = util::getString(
119                    mSymbolPool, util::deviceToHost32(mSymbolEntries[i].stringIndex));
120
121            StringPiece16 typeStr;
122            ResourceUtils::extractResourceName(str, &outSymbol->package, &typeStr,
123                                               &outSymbol->entry);
124            const ResourceType* type = parseResourceType(typeStr);
125            if (!type) {
126                return false;
127            }
128
129            outSymbol->type = *type;
130
131            // Since we scan the symbol table in order, we can start looking for the
132            // next symbol from this point.
133            mSymbolEntryCount -= i + 1;
134            mSymbolEntries += i + 1;
135            return true;
136        }
137    }
138    return false;
139}
140
141/**
142 * Parses the SymbolTable_header, which is present on non-final resource tables
143 * after the compile phase.
144 *
145 * | SymbolTable_header |
146 * |--------------------|
147 * |SymbolTable_entry 0 |
148 * |SymbolTable_entry 1 |
149 * | ...                |
150 * |SymbolTable_entry n |
151 * |--------------------|
152 *
153 */
154bool BinaryResourceParser::parseSymbolTable(const ResChunk_header* chunk) {
155    const SymbolTable_header* header = convertTo<SymbolTable_header>(chunk);
156    if (!header) {
157        mContext->getDiagnostics()->error(DiagMessage(mSource)
158                                          << "corrupt SymbolTable_header");
159        return false;
160    }
161
162    const uint32_t entrySizeBytes =
163            util::deviceToHost32(header->count) * sizeof(SymbolTable_entry);
164    if (entrySizeBytes > getChunkDataLen(&header->header)) {
165        mContext->getDiagnostics()->error(DiagMessage(mSource)
166                                          << "SymbolTable_header data section too long");
167        return false;
168    }
169
170    mSymbolEntries = (const SymbolTable_entry*) getChunkData(&header->header);
171    mSymbolEntryCount = util::deviceToHost32(header->count);
172
173    // Skip over the symbol entries and parse the StringPool chunk that should be next.
174    ResChunkPullParser parser(getChunkData(&header->header) + entrySizeBytes,
175                              getChunkDataLen(&header->header) - entrySizeBytes);
176    if (!ResChunkPullParser::isGoodEvent(parser.next())) {
177        mContext->getDiagnostics()->error(DiagMessage(mSource)
178                                          << "failed to parse chunk in SymbolTable: "
179                                          << parser.getLastError());
180        return false;
181    }
182
183    const ResChunk_header* nextChunk = parser.getChunk();
184    if (util::deviceToHost16(nextChunk->type) != android::RES_STRING_POOL_TYPE) {
185        mContext->getDiagnostics()->error(DiagMessage(mSource)
186                                          << "expected string pool in SymbolTable but got "
187                                          << "chunk of type "
188                                          << (int) util::deviceToHost16(nextChunk->type));
189        return false;
190    }
191
192    if (mSymbolPool.setTo(nextChunk, util::deviceToHost32(nextChunk->size)) != NO_ERROR) {
193        mContext->getDiagnostics()->error(DiagMessage(mSource)
194                                          << "corrupt string pool in SymbolTable: "
195                                          << mSymbolPool.getError());
196        return false;
197    }
198    return true;
199}
200
201/**
202 * Parses the resource table, which contains all the packages, types, and entries.
203 */
204bool BinaryResourceParser::parseTable(const ResChunk_header* chunk) {
205    const ResTable_header* tableHeader = convertTo<ResTable_header>(chunk);
206    if (!tableHeader) {
207        mContext->getDiagnostics()->error(DiagMessage(mSource) << "corrupt ResTable_header chunk");
208        return false;
209    }
210
211    ResChunkPullParser parser(getChunkData(&tableHeader->header),
212                              getChunkDataLen(&tableHeader->header));
213    while (ResChunkPullParser::isGoodEvent(parser.next())) {
214        switch (util::deviceToHost16(parser.getChunk()->type)) {
215        case android::RES_STRING_POOL_TYPE:
216            if (mValuePool.getError() == NO_INIT) {
217                status_t err = mValuePool.setTo(parser.getChunk(),
218                                                util::deviceToHost32(parser.getChunk()->size));
219                if (err != NO_ERROR) {
220                    mContext->getDiagnostics()->error(DiagMessage(mSource)
221                                                      << "corrupt string pool in ResTable: "
222                                                      << mValuePool.getError());
223                    return false;
224                }
225
226                // Reserve some space for the strings we are going to add.
227                mTable->stringPool.hintWillAdd(mValuePool.size(), mValuePool.styleCount());
228            } else {
229                mContext->getDiagnostics()->warn(DiagMessage(mSource)
230                                                 << "unexpected string pool in ResTable");
231            }
232            break;
233
234        case RES_TABLE_SYMBOL_TABLE_TYPE:
235            if (!parseSymbolTable(parser.getChunk())) {
236                return false;
237            }
238            break;
239
240        case RES_TABLE_SOURCE_POOL_TYPE: {
241            status_t err = mSourcePool.setTo(getChunkData(parser.getChunk()),
242                                             getChunkDataLen(parser.getChunk()));
243            if (err != NO_ERROR) {
244                mContext->getDiagnostics()->error(DiagMessage(mSource)
245                                                  << "corrupt source string pool in ResTable: "
246                                                  << mSourcePool.getError());
247                return false;
248            }
249            break;
250        }
251
252        case android::RES_TABLE_PACKAGE_TYPE:
253            if (!parsePackage(parser.getChunk())) {
254                return false;
255            }
256            break;
257
258        default:
259            mContext->getDiagnostics()
260                    ->warn(DiagMessage(mSource)
261                           << "unexpected chunk type "
262                           << (int) util::deviceToHost16(parser.getChunk()->type));
263            break;
264        }
265    }
266
267    if (parser.getEvent() == ResChunkPullParser::Event::BadDocument) {
268        mContext->getDiagnostics()->error(DiagMessage(mSource)
269                                          << "corrupt resource table: " << parser.getLastError());
270        return false;
271    }
272    return true;
273}
274
275
276bool BinaryResourceParser::parsePackage(const ResChunk_header* chunk) {
277    const ResTable_package* packageHeader = convertTo<ResTable_package>(chunk);
278    if (!packageHeader) {
279        mContext->getDiagnostics()->error(DiagMessage(mSource)
280                                          << "corrupt ResTable_package chunk");
281        return false;
282    }
283
284    uint32_t packageId = util::deviceToHost32(packageHeader->id);
285    if (packageId > std::numeric_limits<uint8_t>::max()) {
286        mContext->getDiagnostics()->error(DiagMessage(mSource)
287                                          << "package ID is too big (" << packageId << ")");
288        return false;
289    }
290
291    // Extract the package name.
292    size_t len = strnlen16((const char16_t*) packageHeader->name, arraysize(packageHeader->name));
293    std::u16string packageName;
294    packageName.resize(len);
295    for (size_t i = 0; i < len; i++) {
296        packageName[i] = util::deviceToHost16(packageHeader->name[i]);
297    }
298
299    ResourceTablePackage* package = mTable->createPackage(packageName, (uint8_t) packageId);
300    if (!package) {
301        mContext->getDiagnostics()->error(DiagMessage(mSource)
302                                          << "incompatible package '" << packageName
303                                          << "' with ID " << packageId);
304        return false;
305    }
306
307    ResChunkPullParser parser(getChunkData(&packageHeader->header),
308                              getChunkDataLen(&packageHeader->header));
309    while (ResChunkPullParser::isGoodEvent(parser.next())) {
310        switch (util::deviceToHost16(parser.getChunk()->type)) {
311        case android::RES_STRING_POOL_TYPE:
312            if (mTypePool.getError() == NO_INIT) {
313                status_t err = mTypePool.setTo(parser.getChunk(),
314                                               util::deviceToHost32(parser.getChunk()->size));
315                if (err != NO_ERROR) {
316                    mContext->getDiagnostics()->error(DiagMessage(mSource)
317                                                      << "corrupt type string pool in "
318                                                      << "ResTable_package: "
319                                                      << mTypePool.getError());
320                    return false;
321                }
322            } else if (mKeyPool.getError() == NO_INIT) {
323                status_t err = mKeyPool.setTo(parser.getChunk(),
324                                              util::deviceToHost32(parser.getChunk()->size));
325                if (err != NO_ERROR) {
326                    mContext->getDiagnostics()->error(DiagMessage(mSource)
327                                                      << "corrupt key string pool in "
328                                                      << "ResTable_package: "
329                                                      << mKeyPool.getError());
330                    return false;
331                }
332            } else {
333                mContext->getDiagnostics()->warn(DiagMessage(mSource) << "unexpected string pool");
334            }
335            break;
336
337        case android::RES_TABLE_TYPE_SPEC_TYPE:
338            if (!parseTypeSpec(parser.getChunk())) {
339                return false;
340            }
341            break;
342
343        case android::RES_TABLE_TYPE_TYPE:
344            if (!parseType(package, parser.getChunk())) {
345                return false;
346            }
347            break;
348
349        case RES_TABLE_PUBLIC_TYPE:
350            if (!parsePublic(package, parser.getChunk())) {
351                return false;
352            }
353            break;
354
355        default:
356            mContext->getDiagnostics()
357                    ->warn(DiagMessage(mSource)
358                           << "unexpected chunk type "
359                           << (int) util::deviceToHost16(parser.getChunk()->type));
360            break;
361        }
362    }
363
364    if (parser.getEvent() == ResChunkPullParser::Event::BadDocument) {
365        mContext->getDiagnostics()->error(DiagMessage(mSource)
366                                          << "corrupt ResTable_package: "
367                                          << parser.getLastError());
368        return false;
369    }
370
371    // Now go through the table and change local resource ID references to
372    // symbolic references.
373    ReferenceIdToNameVisitor visitor(&mIdIndex);
374    for (auto& package : mTable->packages) {
375        for (auto& type : package->types) {
376            for (auto& entry : type->entries) {
377                for (auto& configValue : entry->values) {
378                    configValue.value->accept(&visitor);
379                }
380            }
381        }
382    }
383    return true;
384}
385
386bool BinaryResourceParser::parsePublic(const ResourceTablePackage* package,
387                                       const ResChunk_header* chunk) {
388    const Public_header* header = convertTo<Public_header>(chunk);
389    if (!header) {
390        mContext->getDiagnostics()->error(DiagMessage(mSource)
391                                          << "corrupt Public_header chunk");
392        return false;
393    }
394
395    if (header->typeId == 0) {
396        mContext->getDiagnostics()->error(DiagMessage(mSource)
397                                          << "invalid type ID "
398                                          << (int) header->typeId);
399        return false;
400    }
401
402    StringPiece16 typeStr16 = util::getString(mTypePool, header->typeId - 1);
403    const ResourceType* parsedType = parseResourceType(typeStr16);
404    if (!parsedType) {
405        mContext->getDiagnostics()->error(DiagMessage(mSource)
406                                          << "invalid type '" << typeStr16 << "'");
407        return false;
408    }
409
410    const uintptr_t chunkEnd = (uintptr_t) chunk + util::deviceToHost32(chunk->size);
411    const Public_entry* entry = (const Public_entry*) getChunkData(&header->header);
412    for (uint32_t i = 0; i < util::deviceToHost32(header->count); i++) {
413        if ((uintptr_t) entry + sizeof(*entry) > chunkEnd) {
414            mContext->getDiagnostics()->error(DiagMessage(mSource)
415                                              << "Public_entry data section is too long");
416            return false;
417        }
418
419        const ResourceId resId(package->id.value(), header->typeId,
420                               util::deviceToHost16(entry->entryId));
421
422        const ResourceName name(package->name, *parsedType,
423                                util::getString(mKeyPool, entry->key.index).toString());
424
425        Symbol symbol;
426        if (mSourcePool.getError() == NO_ERROR) {
427            symbol.source.path = util::utf16ToUtf8(util::getString(
428                    mSourcePool, util::deviceToHost32(entry->source.index)));
429            symbol.source.line = util::deviceToHost32(entry->sourceLine);
430        }
431
432        switch (util::deviceToHost16(entry->state)) {
433        case Public_entry::kPrivate:
434            symbol.state = SymbolState::kPrivate;
435            break;
436
437        case Public_entry::kPublic:
438            symbol.state = SymbolState::kPublic;
439            break;
440        }
441
442        if (!mTable->setSymbolStateAllowMangled(name, resId, symbol, mContext->getDiagnostics())) {
443            return false;
444        }
445
446        // Add this resource name->id mapping to the index so
447        // that we can resolve all ID references to name references.
448        auto cacheIter = mIdIndex.find(resId);
449        if (cacheIter == mIdIndex.end()) {
450            mIdIndex.insert({ resId, name });
451        }
452
453        entry++;
454    }
455    return true;
456}
457
458bool BinaryResourceParser::parseTypeSpec(const ResChunk_header* chunk) {
459    if (mTypePool.getError() != NO_ERROR) {
460        mContext->getDiagnostics()->error(DiagMessage(mSource)
461                                          << "missing type string pool");
462        return false;
463    }
464
465    const ResTable_typeSpec* typeSpec = convertTo<ResTable_typeSpec>(chunk);
466    if (!typeSpec) {
467        mContext->getDiagnostics()->error(DiagMessage(mSource)
468                                          << "corrupt ResTable_typeSpec chunk");
469        return false;
470    }
471
472    if (typeSpec->id == 0) {
473        mContext->getDiagnostics()->error(DiagMessage(mSource)
474                                          << "ResTable_typeSpec has invalid id: " << typeSpec->id);
475        return false;
476    }
477    return true;
478}
479
480bool BinaryResourceParser::parseType(const ResourceTablePackage* package,
481                                     const ResChunk_header* chunk) {
482    if (mTypePool.getError() != NO_ERROR) {
483        mContext->getDiagnostics()->error(DiagMessage(mSource)
484                                          << "missing type string pool");
485        return false;
486    }
487
488    if (mKeyPool.getError() != NO_ERROR) {
489        mContext->getDiagnostics()->error(DiagMessage(mSource)
490                                          << "missing key string pool");
491        return false;
492    }
493
494    const ResTable_type* type = convertTo<ResTable_type>(chunk);
495    if (!type) {
496        mContext->getDiagnostics()->error(DiagMessage(mSource)
497                                          << "corrupt ResTable_type chunk");
498        return false;
499    }
500
501    if (type->id == 0) {
502        mContext->getDiagnostics()->error(DiagMessage(mSource)
503                                          << "ResTable_type has invalid id: " << (int) type->id);
504        return false;
505    }
506
507    ConfigDescription config;
508    config.copyFromDtoH(type->config);
509
510    StringPiece16 typeStr16 = util::getString(mTypePool, type->id - 1);
511
512    const ResourceType* parsedType = parseResourceType(typeStr16);
513    if (!parsedType) {
514        mContext->getDiagnostics()->error(DiagMessage(mSource)
515                                          << "invalid type name '" << typeStr16
516                                          << "' for type with ID " << (int) type->id);
517        return false;
518    }
519
520    TypeVariant tv(type);
521    for (auto it = tv.beginEntries(); it != tv.endEntries(); ++it) {
522        const ResTable_entry* entry = *it;
523        if (!entry) {
524            continue;
525        }
526
527        const ResourceName name(package->name, *parsedType,
528                                util::getString(mKeyPool,
529                                                util::deviceToHost32(entry->key.index)).toString());
530
531        const ResourceId resId(package->id.value(), type->id, static_cast<uint16_t>(it.index()));
532
533        std::unique_ptr<Value> resourceValue;
534        const ResTable_entry_source* sourceBlock = nullptr;
535
536        if (entry->flags & ResTable_entry::FLAG_COMPLEX) {
537            const ResTable_map_entry* mapEntry = static_cast<const ResTable_map_entry*>(entry);
538            if (util::deviceToHost32(mapEntry->size) - sizeof(*mapEntry) == sizeof(*sourceBlock)) {
539                const uint8_t* data = (const uint8_t*) mapEntry;
540                data += util::deviceToHost32(mapEntry->size) - sizeof(*sourceBlock);
541                sourceBlock = (const ResTable_entry_source*) data;
542            }
543
544            // TODO(adamlesinski): Check that the entry count is valid.
545            resourceValue = parseMapEntry(name, config, mapEntry);
546        } else {
547            if (util::deviceToHost32(entry->size) - sizeof(*entry) == sizeof(*sourceBlock)) {
548                const uint8_t* data = (const uint8_t*) entry;
549                data += util::deviceToHost32(entry->size) - sizeof(*sourceBlock);
550                sourceBlock = (const ResTable_entry_source*) data;
551            }
552
553            const Res_value* value = (const Res_value*)(
554                    (const uint8_t*) entry + util::deviceToHost32(entry->size));
555            resourceValue = parseValue(name, config, value, entry->flags);
556        }
557
558        assert(resourceValue && "failed to interpret valid resource");
559
560        Source source = mSource;
561        if (sourceBlock) {
562            size_t len;
563            const char* str = mSourcePool.string8At(util::deviceToHost32(sourceBlock->pathIndex),
564                                                    &len);
565            if (str) {
566                source.path.assign(str, len);
567            }
568            source.line = util::deviceToHost32(sourceBlock->line);
569        }
570
571        resourceValue->setSource(source);
572        if (!mTable->addResourceAllowMangled(name, config, std::move(resourceValue),
573                                             mContext->getDiagnostics())) {
574            return false;
575        }
576
577        if ((entry->flags & ResTable_entry::FLAG_PUBLIC) != 0) {
578            Symbol symbol;
579            symbol.state = SymbolState::kPublic;
580            symbol.source = mSource.withLine(0);
581            if (!mTable->setSymbolStateAllowMangled(name, resId, symbol,
582                                                    mContext->getDiagnostics())) {
583                return false;
584            }
585        }
586
587        // Add this resource name->id mapping to the index so
588        // that we can resolve all ID references to name references.
589        auto cacheIter = mIdIndex.find(resId);
590        if (cacheIter == mIdIndex.end()) {
591            mIdIndex.insert({ resId, name });
592        }
593    }
594    return true;
595}
596
597std::unique_ptr<Item> BinaryResourceParser::parseValue(const ResourceNameRef& name,
598                                                       const ConfigDescription& config,
599                                                       const Res_value* value,
600                                                       uint16_t flags) {
601    if (name.type == ResourceType::kId) {
602        return util::make_unique<Id>();
603    }
604
605    const uint32_t data = util::deviceToHost32(value->data);
606
607    if (value->dataType == Res_value::TYPE_STRING) {
608        StringPiece16 str = util::getString(mValuePool, data);
609
610        const ResStringPool_span* spans = mValuePool.styleAt(data);
611
612        // Check if the string has a valid style associated with it.
613        if (spans != nullptr && spans->name.index != ResStringPool_span::END) {
614            StyleString styleStr = { str.toString() };
615            while (spans->name.index != ResStringPool_span::END) {
616                styleStr.spans.push_back(Span{
617                        util::getString(mValuePool, spans->name.index).toString(),
618                        spans->firstChar,
619                        spans->lastChar
620                });
621                spans++;
622            }
623            return util::make_unique<StyledString>(mTable->stringPool.makeRef(
624                    styleStr, StringPool::Context{1, config}));
625        } else {
626            if (name.type != ResourceType::kString &&
627                    util::stringStartsWith<char16_t>(str, u"res/")) {
628                // This must be a FileReference.
629                return util::make_unique<FileReference>(mTable->stringPool.makeRef(
630                            str, StringPool::Context{ 0, config }));
631            }
632
633            // There are no styles associated with this string, so treat it as
634            // a simple string.
635            return util::make_unique<String>(mTable->stringPool.makeRef(
636                    str, StringPool::Context{1, config}));
637        }
638    }
639
640    if (value->dataType == Res_value::TYPE_REFERENCE ||
641            value->dataType == Res_value::TYPE_ATTRIBUTE) {
642        const Reference::Type type = (value->dataType == Res_value::TYPE_REFERENCE) ?
643                    Reference::Type::kResource : Reference::Type::kAttribute;
644
645        if (data != 0) {
646            // This is a normal reference.
647            return util::make_unique<Reference>(data, type);
648        }
649
650        // This reference has an invalid ID. Check if it is an unresolved symbol.
651        ResourceNameRef symbol;
652        if (getSymbol(&value->data, &symbol)) {
653            return util::make_unique<Reference>(symbol, type);
654        }
655
656        // This is not an unresolved symbol, so it must be the magic @null reference.
657        Res_value nullType = {};
658        nullType.dataType = Res_value::TYPE_REFERENCE;
659        return util::make_unique<BinaryPrimitive>(nullType);
660    }
661
662    if (value->dataType == ExtendedTypes::TYPE_RAW_STRING) {
663        return util::make_unique<RawString>(mTable->stringPool.makeRef(
664                util::getString(mValuePool, data), StringPool::Context{ 1, config }));
665    }
666
667    // Treat this as a raw binary primitive.
668    return util::make_unique<BinaryPrimitive>(*value);
669}
670
671std::unique_ptr<Value> BinaryResourceParser::parseMapEntry(const ResourceNameRef& name,
672                                                           const ConfigDescription& config,
673                                                           const ResTable_map_entry* map) {
674    switch (name.type) {
675        case ResourceType::kStyle:
676            return parseStyle(name, config, map);
677        case ResourceType::kAttrPrivate:
678            // fallthrough
679        case ResourceType::kAttr:
680            return parseAttr(name, config, map);
681        case ResourceType::kIntegerArray:
682            // fallthrough
683        case ResourceType::kArray:
684            return parseArray(name, config, map);
685        case ResourceType::kStyleable:
686            return parseStyleable(name, config, map);
687        case ResourceType::kPlurals:
688            return parsePlural(name, config, map);
689        default:
690            assert(false && "unknown map type");
691            break;
692    }
693    return {};
694}
695
696std::unique_ptr<Style> BinaryResourceParser::parseStyle(const ResourceNameRef& name,
697                                                        const ConfigDescription& config,
698                                                        const ResTable_map_entry* map) {
699    std::unique_ptr<Style> style = util::make_unique<Style>();
700    if (util::deviceToHost32(map->parent.ident) == 0) {
701        // The parent is either not set or it is an unresolved symbol.
702        // Check to see if it is a symbol.
703        ResourceNameRef symbol;
704        if (getSymbol(&map->parent.ident, &symbol)) {
705            style->parent = Reference(symbol.toResourceName());
706        }
707    } else {
708         // The parent is a regular reference to a resource.
709        style->parent = Reference(util::deviceToHost32(map->parent.ident));
710    }
711
712    for (const ResTable_map& mapEntry : map) {
713        style->entries.emplace_back();
714        Style::Entry& styleEntry = style->entries.back();
715
716        if (util::deviceToHost32(mapEntry.name.ident) == 0) {
717            // The map entry's key (attribute) is not set. This must be
718            // a symbol reference, so resolve it.
719            ResourceNameRef symbol;
720            bool result = getSymbol(&mapEntry.name.ident, &symbol);
721            assert(result);
722            styleEntry.key.name = symbol.toResourceName();
723        } else {
724            // The map entry's key (attribute) is a regular reference.
725            styleEntry.key.id = ResourceId(util::deviceToHost32(mapEntry.name.ident));
726        }
727
728        // Parse the attribute's value.
729        styleEntry.value = parseValue(name, config, &mapEntry.value, 0);
730        assert(styleEntry.value);
731    }
732    return style;
733}
734
735std::unique_ptr<Attribute> BinaryResourceParser::parseAttr(const ResourceNameRef& name,
736                                                           const ConfigDescription& config,
737                                                           const ResTable_map_entry* map) {
738    const bool isWeak = (util::deviceToHost16(map->flags) & ResTable_entry::FLAG_WEAK) != 0;
739    std::unique_ptr<Attribute> attr = util::make_unique<Attribute>(isWeak);
740
741    // First we must discover what type of attribute this is. Find the type mask.
742    auto typeMaskIter = std::find_if(begin(map), end(map), [](const ResTable_map& entry) -> bool {
743        return util::deviceToHost32(entry.name.ident) == ResTable_map::ATTR_TYPE;
744    });
745
746    if (typeMaskIter != end(map)) {
747        attr->typeMask = util::deviceToHost32(typeMaskIter->value.data);
748    }
749
750    if (attr->typeMask & (ResTable_map::TYPE_ENUM | ResTable_map::TYPE_FLAGS)) {
751        for (const ResTable_map& mapEntry : map) {
752            if (Res_INTERNALID(util::deviceToHost32(mapEntry.name.ident))) {
753                continue;
754            }
755
756            Attribute::Symbol symbol;
757            symbol.value = util::deviceToHost32(mapEntry.value.data);
758            if (util::deviceToHost32(mapEntry.name.ident) == 0) {
759                // The map entry's key (id) is not set. This must be
760                // a symbol reference, so resolve it.
761                ResourceNameRef symbolName;
762                bool result = getSymbol(&mapEntry.name.ident, &symbolName);
763                assert(result);
764                symbol.symbol.name = symbolName.toResourceName();
765            } else {
766                // The map entry's key (id) is a regular reference.
767                symbol.symbol.id = ResourceId(util::deviceToHost32(mapEntry.name.ident));
768            }
769
770            attr->symbols.push_back(std::move(symbol));
771        }
772    }
773
774    // TODO(adamlesinski): Find min, max, i80n, etc attributes.
775    return attr;
776}
777
778std::unique_ptr<Array> BinaryResourceParser::parseArray(const ResourceNameRef& name,
779                                                        const ConfigDescription& config,
780                                                        const ResTable_map_entry* map) {
781    std::unique_ptr<Array> array = util::make_unique<Array>();
782    for (const ResTable_map& mapEntry : map) {
783        array->items.push_back(parseValue(name, config, &mapEntry.value, 0));
784    }
785    return array;
786}
787
788std::unique_ptr<Styleable> BinaryResourceParser::parseStyleable(const ResourceNameRef& name,
789                                                                const ConfigDescription& config,
790                                                                const ResTable_map_entry* map) {
791    std::unique_ptr<Styleable> styleable = util::make_unique<Styleable>();
792    for (const ResTable_map& mapEntry : map) {
793        if (util::deviceToHost32(mapEntry.name.ident) == 0) {
794            // The map entry's key (attribute) is not set. This must be
795            // a symbol reference, so resolve it.
796            ResourceNameRef symbol;
797            bool result = getSymbol(&mapEntry.name.ident, &symbol);
798            assert(result);
799            styleable->entries.emplace_back(symbol);
800        } else {
801            // The map entry's key (attribute) is a regular reference.
802            styleable->entries.emplace_back(util::deviceToHost32(mapEntry.name.ident));
803        }
804    }
805    return styleable;
806}
807
808std::unique_ptr<Plural> BinaryResourceParser::parsePlural(const ResourceNameRef& name,
809                                                          const ConfigDescription& config,
810                                                          const ResTable_map_entry* map) {
811    std::unique_ptr<Plural> plural = util::make_unique<Plural>();
812    for (const ResTable_map& mapEntry : map) {
813        std::unique_ptr<Item> item = parseValue(name, config, &mapEntry.value, 0);
814
815        switch (util::deviceToHost32(mapEntry.name.ident)) {
816            case android::ResTable_map::ATTR_ZERO:
817                plural->values[Plural::Zero] = std::move(item);
818                break;
819            case android::ResTable_map::ATTR_ONE:
820                plural->values[Plural::One] = std::move(item);
821                break;
822            case android::ResTable_map::ATTR_TWO:
823                plural->values[Plural::Two] = std::move(item);
824                break;
825            case android::ResTable_map::ATTR_FEW:
826                plural->values[Plural::Few] = std::move(item);
827                break;
828            case android::ResTable_map::ATTR_MANY:
829                plural->values[Plural::Many] = std::move(item);
830                break;
831            case android::ResTable_map::ATTR_OTHER:
832                plural->values[Plural::Other] = std::move(item);
833                break;
834        }
835    }
836    return plural;
837}
838
839} // namespace aapt
840