diff options
Diffstat (limited to 'tools/aapt2/flatten/TableFlattener_test.cpp')
| -rw-r--r-- | tools/aapt2/flatten/TableFlattener_test.cpp | 470 |
1 files changed, 0 insertions, 470 deletions
diff --git a/tools/aapt2/flatten/TableFlattener_test.cpp b/tools/aapt2/flatten/TableFlattener_test.cpp deleted file mode 100644 index 4fdb2eced59b..000000000000 --- a/tools/aapt2/flatten/TableFlattener_test.cpp +++ /dev/null @@ -1,470 +0,0 @@ -/* - * Copyright (C) 2015 The Android Open Source Project - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ - -#include "flatten/TableFlattener.h" - -#include "android-base/stringprintf.h" - -#include "ResourceUtils.h" -#include "SdkConstants.h" -#include "test/Test.h" -#include "unflatten/BinaryResourceParser.h" -#include "util/Util.h" - -using namespace android; - -using ::testing::IsNull; -using ::testing::NotNull; - -namespace aapt { - -class TableFlattenerTest : public ::testing::Test { - public: - void SetUp() override { - context_ = test::ContextBuilder() - .SetCompilationPackage("com.app.test") - .SetPackageId(0x7f) - .Build(); - } - - ::testing::AssertionResult Flatten(IAaptContext* context, const TableFlattenerOptions& options, - ResourceTable* table, std::string* out_content) { - BigBuffer buffer(1024); - TableFlattener flattener(options, &buffer); - if (!flattener.Consume(context, table)) { - return ::testing::AssertionFailure() << "failed to flatten ResourceTable"; - } - *out_content = buffer.to_string(); - return ::testing::AssertionSuccess(); - } - - ::testing::AssertionResult Flatten(IAaptContext* context, const TableFlattenerOptions& options, - ResourceTable* table, ResTable* out_table) { - std::string content; - auto result = Flatten(context, options, table, &content); - if (!result) { - return result; - } - - if (out_table->add(content.data(), content.size(), 1, true) != NO_ERROR) { - return ::testing::AssertionFailure() << "flattened ResTable is corrupt"; - } - return ::testing::AssertionSuccess(); - } - - ::testing::AssertionResult Flatten(IAaptContext* context, const TableFlattenerOptions& options, - ResourceTable* table, ResourceTable* out_table) { - std::string content; - auto result = Flatten(context, options, table, &content); - if (!result) { - return result; - } - - BinaryResourceParser parser(context, out_table, {}, content.data(), content.size()); - if (!parser.Parse()) { - return ::testing::AssertionFailure() << "flattened ResTable is corrupt"; - } - return ::testing::AssertionSuccess(); - } - - ::testing::AssertionResult Exists(ResTable* table, - const StringPiece& expected_name, - const ResourceId& expected_id, - const ConfigDescription& expected_config, - const uint8_t expected_data_type, - const uint32_t expected_data, - const uint32_t expected_spec_flags) { - const ResourceName expected_res_name = test::ParseNameOrDie(expected_name); - - table->setParameters(&expected_config); - - ResTable_config config; - Res_value val; - uint32_t spec_flags; - if (table->getResource(expected_id.id, &val, false, 0, &spec_flags, - &config) < 0) { - return ::testing::AssertionFailure() << "could not find resource with"; - } - - if (expected_data_type != val.dataType) { - return ::testing::AssertionFailure() - << "expected data type " << std::hex << (int)expected_data_type - << " but got data type " << (int)val.dataType << std::dec - << " instead"; - } - - if (expected_data != val.data) { - return ::testing::AssertionFailure() - << "expected data " << std::hex << expected_data - << " but got data " << val.data << std::dec << " instead"; - } - - if (expected_spec_flags != spec_flags) { - return ::testing::AssertionFailure() - << "expected specFlags " << std::hex << expected_spec_flags - << " but got specFlags " << spec_flags << std::dec << " instead"; - } - - ResTable::resource_name actual_name; - if (!table->getResourceName(expected_id.id, false, &actual_name)) { - return ::testing::AssertionFailure() << "failed to find resource name"; - } - - Maybe<ResourceName> resName = ResourceUtils::ToResourceName(actual_name); - if (!resName) { - return ::testing::AssertionFailure() - << "expected name '" << expected_res_name << "' but got '" - << StringPiece16(actual_name.package, actual_name.packageLen) - << ":" << StringPiece16(actual_name.type, actual_name.typeLen) - << "/" << StringPiece16(actual_name.name, actual_name.nameLen) - << "'"; - } - - if (expected_config != config) { - return ::testing::AssertionFailure() << "expected config '" - << expected_config << "' but got '" - << ConfigDescription(config) << "'"; - } - return ::testing::AssertionSuccess(); - } - - protected: - std::unique_ptr<IAaptContext> context_; -}; - -TEST_F(TableFlattenerTest, FlattenFullyLinkedTable) { - std::unique_ptr<ResourceTable> table = - test::ResourceTableBuilder() - .SetPackageId("com.app.test", 0x7f) - .AddSimple("com.app.test:id/one", ResourceId(0x7f020000)) - .AddSimple("com.app.test:id/two", ResourceId(0x7f020001)) - .AddValue("com.app.test:id/three", ResourceId(0x7f020002), - test::BuildReference("com.app.test:id/one", - ResourceId(0x7f020000))) - .AddValue("com.app.test:integer/one", ResourceId(0x7f030000), - util::make_unique<BinaryPrimitive>( - uint8_t(Res_value::TYPE_INT_DEC), 1u)) - .AddValue("com.app.test:integer/one", test::ParseConfigOrDie("v1"), - ResourceId(0x7f030000), - util::make_unique<BinaryPrimitive>( - uint8_t(Res_value::TYPE_INT_DEC), 2u)) - .AddString("com.app.test:string/test", ResourceId(0x7f040000), "foo") - .AddString("com.app.test:layout/bar", ResourceId(0x7f050000), - "res/layout/bar.xml") - .Build(); - - ResTable res_table; - ASSERT_TRUE(Flatten(context_.get(), {}, table.get(), &res_table)); - - EXPECT_TRUE(Exists(&res_table, "com.app.test:id/one", ResourceId(0x7f020000), - {}, Res_value::TYPE_INT_BOOLEAN, 0u, 0u)); - - EXPECT_TRUE(Exists(&res_table, "com.app.test:id/two", ResourceId(0x7f020001), - {}, Res_value::TYPE_INT_BOOLEAN, 0u, 0u)); - - EXPECT_TRUE(Exists(&res_table, "com.app.test:id/three", - ResourceId(0x7f020002), {}, Res_value::TYPE_REFERENCE, - 0x7f020000u, 0u)); - - EXPECT_TRUE(Exists(&res_table, "com.app.test:integer/one", - ResourceId(0x7f030000), {}, Res_value::TYPE_INT_DEC, 1u, - ResTable_config::CONFIG_VERSION)); - - EXPECT_TRUE(Exists(&res_table, "com.app.test:integer/one", - ResourceId(0x7f030000), test::ParseConfigOrDie("v1"), - Res_value::TYPE_INT_DEC, 2u, - ResTable_config::CONFIG_VERSION)); - - std::u16string foo_str = u"foo"; - ssize_t idx = res_table.getTableStringBlock(0)->indexOfString(foo_str.data(), - foo_str.size()); - ASSERT_GE(idx, 0); - EXPECT_TRUE(Exists(&res_table, "com.app.test:string/test", - ResourceId(0x7f040000), {}, Res_value::TYPE_STRING, - (uint32_t)idx, 0u)); - - std::u16string bar_path = u"res/layout/bar.xml"; - idx = res_table.getTableStringBlock(0)->indexOfString(bar_path.data(), - bar_path.size()); - ASSERT_GE(idx, 0); - EXPECT_TRUE(Exists(&res_table, "com.app.test:layout/bar", - ResourceId(0x7f050000), {}, Res_value::TYPE_STRING, - (uint32_t)idx, 0u)); -} - -TEST_F(TableFlattenerTest, FlattenEntriesWithGapsInIds) { - std::unique_ptr<ResourceTable> table = - test::ResourceTableBuilder() - .SetPackageId("com.app.test", 0x7f) - .AddSimple("com.app.test:id/one", ResourceId(0x7f020001)) - .AddSimple("com.app.test:id/three", ResourceId(0x7f020003)) - .Build(); - - ResTable res_table; - ASSERT_TRUE(Flatten(context_.get(), {}, table.get(), &res_table)); - - EXPECT_TRUE(Exists(&res_table, "com.app.test:id/one", ResourceId(0x7f020001), - {}, Res_value::TYPE_INT_BOOLEAN, 0u, 0u)); - EXPECT_TRUE(Exists(&res_table, "com.app.test:id/three", - ResourceId(0x7f020003), {}, Res_value::TYPE_INT_BOOLEAN, - 0u, 0u)); -} - -TEST_F(TableFlattenerTest, FlattenMinMaxAttributes) { - Attribute attr(false); - attr.type_mask = android::ResTable_map::TYPE_INTEGER; - attr.min_int = 10; - attr.max_int = 23; - std::unique_ptr<ResourceTable> table = - test::ResourceTableBuilder() - .SetPackageId("android", 0x01) - .AddValue("android:attr/foo", ResourceId(0x01010000), - util::make_unique<Attribute>(attr)) - .Build(); - - ResourceTable result; - ASSERT_TRUE(Flatten(context_.get(), {}, table.get(), &result)); - - Attribute* actual_attr = test::GetValue<Attribute>(&result, "android:attr/foo"); - ASSERT_THAT(actual_attr, NotNull()); - EXPECT_EQ(attr.IsWeak(), actual_attr->IsWeak()); - EXPECT_EQ(attr.type_mask, actual_attr->type_mask); - EXPECT_EQ(attr.min_int, actual_attr->min_int); - EXPECT_EQ(attr.max_int, actual_attr->max_int); -} - -static std::unique_ptr<ResourceTable> BuildTableWithSparseEntries( - IAaptContext* context, const ConfigDescription& sparse_config, float load) { - std::unique_ptr<ResourceTable> table = - test::ResourceTableBuilder() - .SetPackageId(context->GetCompilationPackage(), context->GetPackageId()) - .Build(); - - // Add regular entries. - int stride = static_cast<int>(1.0f / load); - for (int i = 0; i < 100; i++) { - const ResourceName name = test::ParseNameOrDie( - base::StringPrintf("%s:string/foo_%d", context->GetCompilationPackage().data(), i)); - const ResourceId resid(context->GetPackageId(), 0x02, static_cast<uint16_t>(i)); - const auto value = - util::make_unique<BinaryPrimitive>(Res_value::TYPE_INT_DEC, static_cast<uint32_t>(i)); - CHECK(table->AddResource(name, resid, ConfigDescription::DefaultConfig(), "", - std::unique_ptr<Value>(value->Clone(nullptr)), - context->GetDiagnostics())); - - // Every few entries, write out a sparse_config value. This will give us the desired load. - if (i % stride == 0) { - CHECK(table->AddResource(name, resid, sparse_config, "", - std::unique_ptr<Value>(value->Clone(nullptr)), - context->GetDiagnostics())); - } - } - return table; -} - -TEST_F(TableFlattenerTest, FlattenSparseEntryWithMinSdkO) { - std::unique_ptr<IAaptContext> context = test::ContextBuilder() - .SetCompilationPackage("android") - .SetPackageId(0x01) - .SetMinSdkVersion(SDK_O) - .Build(); - - const ConfigDescription sparse_config = test::ParseConfigOrDie("en-rGB"); - auto table_in = BuildTableWithSparseEntries(context.get(), sparse_config, 0.25f); - - TableFlattenerOptions options; - options.use_sparse_entries = true; - - std::string no_sparse_contents; - ASSERT_TRUE(Flatten(context.get(), {}, table_in.get(), &no_sparse_contents)); - - std::string sparse_contents; - ASSERT_TRUE(Flatten(context.get(), options, table_in.get(), &sparse_contents)); - - EXPECT_GT(no_sparse_contents.size(), sparse_contents.size()); - - // Attempt to parse the sparse contents. - - ResourceTable sparse_table; - BinaryResourceParser parser(context.get(), &sparse_table, Source("test.arsc"), - sparse_contents.data(), sparse_contents.size()); - ASSERT_TRUE(parser.Parse()); - - auto value = test::GetValueForConfig<BinaryPrimitive>(&sparse_table, "android:string/foo_0", - sparse_config); - ASSERT_THAT(value, NotNull()); - EXPECT_EQ(0u, value->value.data); - - ASSERT_THAT(test::GetValueForConfig<BinaryPrimitive>(&sparse_table, "android:string/foo_1", sparse_config), IsNull()); - - value = test::GetValueForConfig<BinaryPrimitive>(&sparse_table, "android:string/foo_4", sparse_config); - ASSERT_THAT(value, NotNull()); - EXPECT_EQ(4u, value->value.data); -} - -TEST_F(TableFlattenerTest, FlattenSparseEntryWithConfigSdkVersionO) { - std::unique_ptr<IAaptContext> context = test::ContextBuilder() - .SetCompilationPackage("android") - .SetPackageId(0x01) - .SetMinSdkVersion(SDK_LOLLIPOP) - .Build(); - - const ConfigDescription sparse_config = test::ParseConfigOrDie("en-rGB-v26"); - auto table_in = BuildTableWithSparseEntries(context.get(), sparse_config, 0.25f); - - TableFlattenerOptions options; - options.use_sparse_entries = true; - - std::string no_sparse_contents; - ASSERT_TRUE(Flatten(context.get(), {}, table_in.get(), &no_sparse_contents)); - - std::string sparse_contents; - ASSERT_TRUE(Flatten(context.get(), options, table_in.get(), &sparse_contents)); - - EXPECT_GT(no_sparse_contents.size(), sparse_contents.size()); -} - -TEST_F(TableFlattenerTest, DoNotUseSparseEntryForDenseConfig) { - std::unique_ptr<IAaptContext> context = test::ContextBuilder() - .SetCompilationPackage("android") - .SetPackageId(0x01) - .SetMinSdkVersion(SDK_O) - .Build(); - - const ConfigDescription sparse_config = test::ParseConfigOrDie("en-rGB"); - auto table_in = BuildTableWithSparseEntries(context.get(), sparse_config, 0.80f); - - TableFlattenerOptions options; - options.use_sparse_entries = true; - - std::string no_sparse_contents; - ASSERT_TRUE(Flatten(context.get(), {}, table_in.get(), &no_sparse_contents)); - - std::string sparse_contents; - ASSERT_TRUE(Flatten(context.get(), options, table_in.get(), &sparse_contents)); - - EXPECT_EQ(no_sparse_contents.size(), sparse_contents.size()); -} - -TEST_F(TableFlattenerTest, FlattenSharedLibrary) { - std::unique_ptr<IAaptContext> context = - test::ContextBuilder().SetCompilationPackage("lib").SetPackageId(0x00).Build(); - std::unique_ptr<ResourceTable> table = - test::ResourceTableBuilder() - .SetPackageId("lib", 0x00) - .AddValue("lib:id/foo", ResourceId(0x00010000), util::make_unique<Id>()) - .Build(); - ResourceTable result; - ASSERT_TRUE(Flatten(context.get(), {}, table.get(), &result)); - - Maybe<ResourceTable::SearchResult> search_result = - result.FindResource(test::ParseNameOrDie("lib:id/foo")); - ASSERT_TRUE(search_result); - EXPECT_EQ(0x00u, search_result.value().package->id.value()); - - auto iter = result.included_packages_.find(0x00); - ASSERT_NE(result.included_packages_.end(), iter); - EXPECT_EQ("lib", iter->second); -} - -TEST_F(TableFlattenerTest, FlattenTableReferencingSharedLibraries) { - std::unique_ptr<IAaptContext> context = - test::ContextBuilder().SetCompilationPackage("app").SetPackageId(0x7f).Build(); - std::unique_ptr<ResourceTable> table = - test::ResourceTableBuilder() - .SetPackageId("app", 0x7f) - .AddValue("app:id/foo", ResourceId(0x7f010000), - test::BuildReference("lib_one:id/foo", ResourceId(0x02010000))) - .AddValue("app:id/bar", ResourceId(0x7f010001), - test::BuildReference("lib_two:id/bar", ResourceId(0x03010000))) - .Build(); - table->included_packages_[0x02] = "lib_one"; - table->included_packages_[0x03] = "lib_two"; - - ResTable result; - ASSERT_TRUE(Flatten(context.get(), {}, table.get(), &result)); - - const DynamicRefTable* dynamic_ref_table = result.getDynamicRefTableForCookie(1); - ASSERT_THAT(dynamic_ref_table, NotNull()); - - const KeyedVector<String16, uint8_t>& entries = dynamic_ref_table->entries(); - - ssize_t idx = entries.indexOfKey(android::String16("lib_one")); - ASSERT_GE(idx, 0); - EXPECT_EQ(0x02u, entries.valueAt(idx)); - - idx = entries.indexOfKey(android::String16("lib_two")); - ASSERT_GE(idx, 0); - EXPECT_EQ(0x03u, entries.valueAt(idx)); -} - -TEST_F(TableFlattenerTest, PackageWithNonStandardIdHasDynamicRefTable) { - std::unique_ptr<IAaptContext> context = - test::ContextBuilder().SetCompilationPackage("app").SetPackageId(0x80).Build(); - std::unique_ptr<ResourceTable> table = test::ResourceTableBuilder() - .SetPackageId("app", 0x80) - .AddSimple("app:id/foo", ResourceId(0x80010000)) - .Build(); - - ResTable result; - ASSERT_TRUE(Flatten(context.get(), {}, table.get(), &result)); - - const DynamicRefTable* dynamic_ref_table = result.getDynamicRefTableForCookie(1); - ASSERT_THAT(dynamic_ref_table, NotNull()); - - const KeyedVector<String16, uint8_t>& entries = dynamic_ref_table->entries(); - ssize_t idx = entries.indexOfKey(android::String16("app")); - ASSERT_GE(idx, 0); - EXPECT_EQ(0x80u, entries.valueAt(idx)); -} - -TEST_F(TableFlattenerTest, LongPackageNameIsTruncated) { - std::string kPackageName(256, 'F'); - - std::unique_ptr<IAaptContext> context = - test::ContextBuilder().SetCompilationPackage(kPackageName).SetPackageId(0x7f).Build(); - std::unique_ptr<ResourceTable> table = - test::ResourceTableBuilder() - .SetPackageId(kPackageName, 0x7f) - .AddSimple(kPackageName + ":id/foo", ResourceId(0x7f010000)) - .Build(); - - ResTable result; - ASSERT_TRUE(Flatten(context.get(), {}, table.get(), &result)); - - ASSERT_EQ(1u, result.getBasePackageCount()); - EXPECT_EQ(127u, result.getBasePackageName(0).size()); -} - -TEST_F(TableFlattenerTest, LongSharedLibraryPackageNameIsIllegal) { - std::string kPackageName(256, 'F'); - - std::unique_ptr<IAaptContext> context = test::ContextBuilder() - .SetCompilationPackage(kPackageName) - .SetPackageId(0x7f) - .SetPackageType(PackageType::kSharedLib) - .Build(); - std::unique_ptr<ResourceTable> table = - test::ResourceTableBuilder() - .SetPackageId(kPackageName, 0x7f) - .AddSimple(kPackageName + ":id/foo", ResourceId(0x7f010000)) - .Build(); - - ResTable result; - ASSERT_FALSE(Flatten(context.get(), {}, table.get(), &result)); -} - -} // namespace aapt |