1 /*
2 * Copyright (C) 2016 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 "androidfw/ApkAssets.h"
18
19 #include "android-base/file.h"
20 #include "android-base/test_utils.h"
21 #include "android-base/unique_fd.h"
22 #include "androidfw/Util.h"
23
24 #include "TestHelpers.h"
25 #include "data/basic/R.h"
26
27 using ::android::base::unique_fd;
28 using ::com::android::basic::R;
29 using ::testing::Eq;
30 using ::testing::Ge;
31 using ::testing::IsNull;
32 using ::testing::NotNull;
33 using ::testing::SizeIs;
34 using ::testing::StrEq;
35
36 namespace android {
37
TEST(ApkAssetsTest,LoadApk)38 TEST(ApkAssetsTest, LoadApk) {
39 auto loaded_apk = ApkAssets::Load(GetTestDataPath() + "/basic/basic.apk");
40 ASSERT_THAT(loaded_apk, NotNull());
41
42 const LoadedArsc* loaded_arsc = loaded_apk->GetLoadedArsc();
43 ASSERT_THAT(loaded_arsc, NotNull());
44 ASSERT_THAT(loaded_arsc->GetPackageById(0x7fu), NotNull());
45 ASSERT_THAT(loaded_apk->GetAssetsProvider()->Open("res/layout/main.xml"), NotNull());
46 }
47
TEST(ApkAssetsTest,LoadApkFromFd)48 TEST(ApkAssetsTest, LoadApkFromFd) {
49 const std::string path = GetTestDataPath() + "/basic/basic.apk";
50 unique_fd fd(::open(path.c_str(), O_RDONLY | O_BINARY));
51 ASSERT_THAT(fd.get(), Ge(0));
52
53 auto loaded_apk = ApkAssets::LoadFromFd(std::move(fd), path);
54 ASSERT_THAT(loaded_apk, NotNull());
55
56 const LoadedArsc* loaded_arsc = loaded_apk->GetLoadedArsc();
57 ASSERT_THAT(loaded_arsc, NotNull());
58 ASSERT_THAT(loaded_arsc->GetPackageById(0x7fu), NotNull());
59 ASSERT_THAT(loaded_apk->GetAssetsProvider()->Open("res/layout/main.xml"), NotNull());
60 }
61
TEST(ApkAssetsTest,LoadApkAsSharedLibrary)62 TEST(ApkAssetsTest, LoadApkAsSharedLibrary) {
63 auto loaded_apk = ApkAssets::Load(GetTestDataPath() + "/appaslib/appaslib.apk");
64 ASSERT_THAT(loaded_apk, NotNull());
65
66 const LoadedArsc* loaded_arsc = loaded_apk->GetLoadedArsc();
67 ASSERT_THAT(loaded_arsc, NotNull());
68 ASSERT_THAT(loaded_arsc->GetPackages(), SizeIs(1u));
69 EXPECT_FALSE(loaded_arsc->GetPackages()[0]->IsDynamic());
70
71 loaded_apk = ApkAssets::Load(GetTestDataPath() + "/appaslib/appaslib.apk", PROPERTY_DYNAMIC);
72 ASSERT_THAT(loaded_apk, NotNull());
73
74 loaded_arsc = loaded_apk->GetLoadedArsc();
75 ASSERT_THAT(loaded_arsc, NotNull());
76 ASSERT_THAT(loaded_arsc->GetPackages(), SizeIs(1u));
77 EXPECT_TRUE(loaded_arsc->GetPackages()[0]->IsDynamic());
78 }
79
TEST(ApkAssetsTest,CreateAndDestroyAssetKeepsApkAssetsOpen)80 TEST(ApkAssetsTest, CreateAndDestroyAssetKeepsApkAssetsOpen) {
81 auto loaded_apk = ApkAssets::Load(GetTestDataPath() + "/basic/basic.apk");
82 ASSERT_THAT(loaded_apk, NotNull());
83
84 { ASSERT_THAT(loaded_apk->GetAssetsProvider()->Open("res/layout/main.xml",
85 Asset::ACCESS_BUFFER), NotNull()); }
86
87 { ASSERT_THAT(loaded_apk->GetAssetsProvider()->Open("res/layout/main.xml",
88 Asset::ACCESS_BUFFER), NotNull()); }
89 }
90
TEST(ApkAssetsTest,OpenUncompressedAssetFd)91 TEST(ApkAssetsTest, OpenUncompressedAssetFd) {
92 auto loaded_apk = ApkAssets::Load(GetTestDataPath() + "/basic/basic.apk");
93 ASSERT_THAT(loaded_apk, NotNull());
94
95 auto asset = loaded_apk->GetAssetsProvider()->Open("assets/uncompressed.txt",
96 Asset::ACCESS_UNKNOWN);
97 ASSERT_THAT(asset, NotNull());
98
99 off64_t start, length;
100 unique_fd fd(asset->openFileDescriptor(&start, &length));
101 ASSERT_THAT(fd.get(), Ge(0));
102
103 lseek64(fd.get(), start, SEEK_SET);
104
105 std::string buffer;
106 buffer.resize(length);
107 ASSERT_TRUE(base::ReadFully(fd.get(), &*buffer.begin(), length));
108
109 EXPECT_THAT(buffer, StrEq("This should be uncompressed.\n\n"));
110 }
111
TEST(ApkAssetsTest,TakeAssetsProviderNotCrashing)112 TEST(ApkAssetsTest, TakeAssetsProviderNotCrashing) {
113 // Make sure the apk assets object can survive taking its assets provider and doesn't crash
114 // the process.
115 {
116 auto loaded_apk = ApkAssets::Load(GetTestDataPath() + "/basic/basic.apk");
117 ASSERT_THAT(loaded_apk, NotNull());
118
119 auto provider = std::move(*loaded_apk).TakeAssetsProvider();
120 ASSERT_THAT(provider, NotNull());
121 }
122 // If this test doesn't crash by this point we're all good.
123 }
124
TEST(ApkAssetsTest,AssetsProviderNotMovedOnError)125 TEST(ApkAssetsTest, AssetsProviderNotMovedOnError) {
126 auto assets_provider
127 = ZipAssetsProvider::Create(GetTestDataPath() + "/bad/bad.apk", 0);
128 ASSERT_THAT(assets_provider, NotNull());
129 auto loaded_apk = ApkAssets::Load(std::move(assets_provider));
130 ASSERT_THAT(loaded_apk, IsNull());
131 ASSERT_THAT(assets_provider, NotNull());
132 }
133
134 } // namespace android
135