blob: 77114f273d3da908776e23c3853143980e21377c [file] [log] [blame]
/*
* Copyright (C) 2016 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 "androidfw/AssetManager2.h"
#include "android-base/logging.h"
#include "TestHelpers.h"
#include "androidfw/ResourceUtils.h"
#include "data/lib_one/R.h"
#include "data/lib_two/R.h"
#include "data/libclient/R.h"
#include "data/styles/R.h"
#include "data/system/R.h"
namespace app = com::android::app;
namespace lib_one = com::android::lib_one;
namespace lib_two = com::android::lib_two;
namespace libclient = com::android::libclient;
namespace android {
class ThemeTest : public ::testing::Test {
public:
void SetUp() override {
system_assets_ = ApkAssets::Load(GetTestDataPath() + "/system/system.apk", PROPERTY_SYSTEM);
ASSERT_NE(nullptr, system_assets_);
style_assets_ = ApkAssets::Load(GetTestDataPath() + "/styles/styles.apk");
ASSERT_NE(nullptr, style_assets_);
libclient_assets_ = ApkAssets::Load(GetTestDataPath() + "/libclient/libclient.apk");
ASSERT_NE(nullptr, libclient_assets_);
lib_one_assets_ = ApkAssets::Load(GetTestDataPath() + "/lib_one/lib_one.apk");
ASSERT_NE(nullptr, lib_one_assets_);
lib_two_assets_ = ApkAssets::Load(GetTestDataPath() + "/lib_two/lib_two.apk");
ASSERT_NE(nullptr, lib_two_assets_);
}
protected:
std::unique_ptr<const ApkAssets> system_assets_;
std::unique_ptr<const ApkAssets> style_assets_;
std::unique_ptr<const ApkAssets> libclient_assets_;
std::unique_ptr<const ApkAssets> lib_one_assets_;
std::unique_ptr<const ApkAssets> lib_two_assets_;
};
TEST_F(ThemeTest, EmptyTheme) {
AssetManager2 assetmanager;
assetmanager.SetApkAssets({style_assets_.get()});
std::unique_ptr<Theme> theme = assetmanager.NewTheme();
EXPECT_EQ(0u, theme->GetChangingConfigurations());
EXPECT_EQ(&assetmanager, theme->GetAssetManager());
EXPECT_FALSE(theme->GetAttribute(app::R::attr::attr_one).has_value());
}
TEST_F(ThemeTest, SingleThemeNoParent) {
AssetManager2 assetmanager;
assetmanager.SetApkAssets({style_assets_.get()});
std::unique_ptr<Theme> theme = assetmanager.NewTheme();
ASSERT_TRUE(theme->ApplyStyle(app::R::style::StyleOne).has_value());
auto value = theme->GetAttribute(app::R::attr::attr_one);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_INT_DEC, value->type);
EXPECT_EQ(1u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
value = theme->GetAttribute(app::R::attr::attr_two);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_INT_DEC, value->type);
EXPECT_EQ(2u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
}
TEST_F(ThemeTest, SingleThemeWithParent) {
AssetManager2 assetmanager;
assetmanager.SetApkAssets({style_assets_.get()});
std::unique_ptr<Theme> theme = assetmanager.NewTheme();
ASSERT_TRUE(theme->ApplyStyle(app::R::style::StyleTwo).has_value());
auto value = theme->GetAttribute(app::R::attr::attr_one);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_INT_DEC, value->type);
EXPECT_EQ(1u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
value = theme->GetAttribute(app::R::attr::attr_two);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_STRING, value->type);
EXPECT_EQ(0, value->cookie);
EXPECT_EQ(std::string("string"),
GetStringFromPool(assetmanager.GetStringPoolForCookie(0), value->data));
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
// This attribute should point to an attr_indirect, so the result should be 3.
value = theme->GetAttribute(app::R::attr::attr_three);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_INT_DEC, value->type);
EXPECT_EQ(3u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
}
TEST_F(ThemeTest, TryToUseBadResourceId) {
AssetManager2 assetmanager;
assetmanager.SetApkAssets({style_assets_.get()});
std::unique_ptr<Theme> theme = assetmanager.NewTheme();
ASSERT_TRUE(theme->ApplyStyle(app::R::style::StyleTwo).has_value());
ASSERT_FALSE(theme->GetAttribute(0x7f000001));
}
TEST_F(ThemeTest, MultipleThemesOverlaidNotForce) {
AssetManager2 assetmanager;
assetmanager.SetApkAssets({style_assets_.get()});
std::unique_ptr<Theme> theme = assetmanager.NewTheme();
ASSERT_TRUE(theme->ApplyStyle(app::R::style::StyleTwo).has_value());
ASSERT_TRUE(theme->ApplyStyle(app::R::style::StyleThree).has_value());
// attr_one is still here from the base.
auto value = theme->GetAttribute(app::R::attr::attr_one);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_INT_DEC, value->type);
EXPECT_EQ(1u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
// check for the new attr_six
value = theme->GetAttribute(app::R::attr::attr_six);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_INT_DEC, value->type);
EXPECT_EQ(6u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
// check for the old attr_five (force=true was not used).
value = theme->GetAttribute(app::R::attr::attr_five);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_REFERENCE, value->type);
EXPECT_EQ(app::R::string::string_one, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
}
TEST_F(ThemeTest, MultipleThemesOverlaidForced) {
AssetManager2 assetmanager;
assetmanager.SetApkAssets({style_assets_.get()});
std::unique_ptr<Theme> theme = assetmanager.NewTheme();
ASSERT_TRUE(theme->ApplyStyle(app::R::style::StyleTwo).has_value());
ASSERT_TRUE(theme->ApplyStyle(app::R::style::StyleThree, true /* force */).has_value());
// attr_one is still here from the base.
auto value = theme->GetAttribute(app::R::attr::attr_one);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_INT_DEC, value->type);
EXPECT_EQ(1u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
// check for the new attr_six
value = theme->GetAttribute(app::R::attr::attr_six);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_INT_DEC, value->type);
EXPECT_EQ(6u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
// check for the new attr_five (force=true was used).
value = theme->GetAttribute(app::R::attr::attr_five);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_INT_DEC, value->type);
EXPECT_EQ(5u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
}
TEST_F(ThemeTest, ResolveDynamicAttributesAndReferencesToSharedLibrary) {
AssetManager2 assetmanager;
assetmanager.SetApkAssets(
{lib_two_assets_.get(), lib_one_assets_.get(), libclient_assets_.get()});
std::unique_ptr<Theme> theme = assetmanager.NewTheme();
ASSERT_TRUE(theme->ApplyStyle(libclient::R::style::Theme, false /*force*/).has_value());
// The attribute should be resolved to the final value.
auto value = theme->GetAttribute(libclient::R::attr::foo);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_INT_DEC, value->type);
EXPECT_EQ(700u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
// The reference should be resolved to a TYPE_REFERENCE.
value = theme->GetAttribute(libclient::R::attr::bar);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_REFERENCE, value->type);
// lib_one is assigned package ID 0x03.
EXPECT_EQ(3u, get_package_id(value->data));
EXPECT_EQ(get_type_id(lib_one::R::string::foo), get_type_id(value->data));
EXPECT_EQ(get_entry_id(lib_one::R::string::foo), get_entry_id(value->data));
}
TEST_F(ThemeTest, CopyThemeSameAssetManager) {
AssetManager2 assetmanager;
assetmanager.SetApkAssets({style_assets_.get()});
std::unique_ptr<Theme> theme_one = assetmanager.NewTheme();
ASSERT_TRUE(theme_one->ApplyStyle(app::R::style::StyleOne).has_value());
// attr_one is still here from the base.
auto value = theme_one->GetAttribute(app::R::attr::attr_one);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_INT_DEC, value->type);
EXPECT_EQ(1u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
// attr_six is not here.
ASSERT_FALSE(theme_one->GetAttribute(app::R::attr::attr_six).has_value());
std::unique_ptr<Theme> theme_two = assetmanager.NewTheme();
ASSERT_TRUE(theme_two->ApplyStyle(app::R::style::StyleThree).has_value());
// Copy the theme to theme_one.
theme_one->SetTo(*theme_two);
// Clear theme_two to make sure we test that there WAS a copy.
theme_two->Clear();
// attr_one is now not here.
ASSERT_FALSE(theme_one->GetAttribute(app::R::attr::attr_one).has_value());
// attr_six is now here because it was copied.
value = theme_one->GetAttribute(app::R::attr::attr_six);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_INT_DEC, value->type);
EXPECT_EQ(6u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
}
TEST_F(ThemeTest, ThemeRebase) {
AssetManager2 am;
am.SetApkAssets({style_assets_.get()});
AssetManager2 am_night;
am_night.SetApkAssets({style_assets_.get()});
ResTable_config night{};
night.uiMode = ResTable_config::UI_MODE_NIGHT_YES;
night.version = 8u;
am_night.SetConfiguration(night);
auto theme = am.NewTheme();
{
const uint32_t styles[] = {app::R::style::StyleOne, app::R::style::StyleDayNight};
const uint8_t force[] = {true, true};
theme->Rebase(&am, styles, force, arraysize(styles));
}
// attr_one in StyleDayNight force overrides StyleOne. attr_one is defined in the StyleOne.
auto value = theme->GetAttribute(app::R::attr::attr_one);
ASSERT_TRUE(value);
EXPECT_EQ(10u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC | ResTable_config::CONFIG_UI_MODE |
ResTable_config::CONFIG_VERSION), value->flags);
// attr_two is defined in the StyleOne.
value = theme->GetAttribute(app::R::attr::attr_two);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_INT_DEC, value->type);
EXPECT_EQ(2u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
{
const uint32_t styles[] = {app::R::style::StyleOne, app::R::style::StyleDayNight};
const uint8_t force[] = {false, false};
theme->Rebase(&am, styles, force, arraysize(styles));
}
// attr_one in StyleDayNight does not override StyleOne because `force` is false.
value = theme->GetAttribute(app::R::attr::attr_one);
ASSERT_TRUE(value);
EXPECT_EQ(1u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
// attr_two is defined in the StyleOne.
value = theme->GetAttribute(app::R::attr::attr_two);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_INT_DEC, value->type);
EXPECT_EQ(2u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
{
const uint32_t styles[] = {app::R::style::StyleOne, app::R::style::StyleDayNight};
const uint8_t force[] = {false, true};
theme->Rebase(&am_night, styles, force, arraysize(styles));
}
// attr_one is defined in the StyleDayNight.
value = theme->GetAttribute(app::R::attr::attr_one);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_INT_DEC, value->type);
EXPECT_EQ(100u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC | ResTable_config::CONFIG_UI_MODE |
ResTable_config::CONFIG_VERSION), value->flags);
// attr_two is now not here.
value = theme->GetAttribute(app::R::attr::attr_two);
ASSERT_TRUE(value);
EXPECT_EQ(Res_value::TYPE_INT_DEC, value->type);
EXPECT_EQ(2u, value->data);
EXPECT_EQ(static_cast<uint32_t>(ResTable_typeSpec::SPEC_PUBLIC), value->flags);
}
TEST_F(ThemeTest, OnlyCopySameAssetsThemeWhenAssetManagersDiffer) {
AssetManager2 assetmanager_dst;
assetmanager_dst.SetApkAssets({system_assets_.get(), lib_one_assets_.get(), style_assets_.get(),
libclient_assets_.get()});
AssetManager2 assetmanager_src;
assetmanager_src.SetApkAssets({system_assets_.get(), lib_two_assets_.get(), lib_one_assets_.get(),
style_assets_.get()});
auto theme_dst = assetmanager_dst.NewTheme();
ASSERT_TRUE(theme_dst->ApplyStyle(app::R::style::StyleOne).has_value());
auto theme_src = assetmanager_src.NewTheme();
ASSERT_TRUE(theme_src->ApplyStyle(R::style::Theme_One).has_value());
ASSERT_TRUE(theme_src->ApplyStyle(app::R::style::StyleTwo).has_value());
ASSERT_TRUE(theme_src->ApplyStyle(fix_package_id(lib_one::R::style::Theme, 0x03),
false /*force*/).has_value());
ASSERT_TRUE(theme_src->ApplyStyle(fix_package_id(lib_two::R::style::Theme, 0x02),
false /*force*/).has_value());
theme_dst->SetTo(*theme_src);
// System resources (present in destination asset manager).
auto value = theme_dst->GetAttribute(R::attr::foreground);
ASSERT_TRUE(value.has_value());
EXPECT_EQ(0, value->cookie);
// The cookie of the style asset is 3 in the source and 2 in the destination.
// Check that the cookie has been rewritten to the destination values.
value = theme_dst->GetAttribute(app::R::attr::attr_one);
ASSERT_TRUE(value.has_value());
EXPECT_EQ(2, value->cookie);
// The cookie of the lib_one asset is 2 in the source and 1 in the destination.
// The package id of the lib_one package is 0x03 in the source and 0x02 in the destination
// Check that the cookie and packages have been rewritten to the destination values.
value = theme_dst->GetAttribute(fix_package_id(lib_one::R::attr::attr1, 0x02));
ASSERT_TRUE(value.has_value());
EXPECT_EQ(1, value->cookie);
value = theme_dst->GetAttribute(fix_package_id(lib_one::R::attr::attr2, 0x02));
ASSERT_TRUE(value.has_value());
EXPECT_EQ(1, value->cookie);
// attr2 references an attribute in lib_one. Check that the resolution of the attribute value is
// correct after the value of attr2 had its package id rewritten to the destination package id.
EXPECT_EQ(700, value->data);
}
TEST_F(ThemeTest, CopyNonReferencesWhenPackagesDiffer) {
AssetManager2 assetmanager_dst;
assetmanager_dst.SetApkAssets({system_assets_.get()});
AssetManager2 assetmanager_src;
assetmanager_src.SetApkAssets({system_assets_.get(), style_assets_.get()});
auto theme_dst = assetmanager_dst.NewTheme();
auto theme_src = assetmanager_src.NewTheme();
ASSERT_TRUE(theme_src->ApplyStyle(app::R::style::StyleSeven).has_value());
theme_dst->SetTo(*theme_src);
// Allow inline resource values to be copied even if the source apk asset is not present in the
// destination.
auto value = theme_dst->GetAttribute(0x0101021b /* android:versionCode */);
ASSERT_TRUE(value.has_value());
EXPECT_EQ(0, value->cookie);
// Do not copy strings since the data is an index into the values string pool of the source apk
// asset.
EXPECT_FALSE(theme_dst->GetAttribute(0x01010001 /* android:label */).has_value());
// Do not copy values that reference another resource if the resource is not present in the
// destination.
EXPECT_FALSE(theme_dst->GetAttribute(0x01010002 /* android:icon */).has_value());
EXPECT_FALSE(theme_dst->GetAttribute(0x010100d1 /* android:tag */).has_value());
// Allow @empty to and @null to be copied.
value = theme_dst->GetAttribute(0x010100d0 /* android:id */);
ASSERT_TRUE(value.has_value());
EXPECT_EQ(0, value->cookie);
value = theme_dst->GetAttribute(0x01010000 /* android:theme */);
ASSERT_TRUE(value.has_value());
EXPECT_EQ(0, value->cookie);
}
} // namespace android