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::NotNull;
32 using ::testing::SizeIs;
33 using ::testing::StrEq;
34
35 namespace android {
36
TEST(ApkAssetsTest,LoadApk)37 TEST(ApkAssetsTest, LoadApk) {
38 std::unique_ptr<const ApkAssets> loaded_apk =
39 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 std::unique_ptr<const ApkAssets> 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 std::unique_ptr<const ApkAssets> loaded_apk =
64 ApkAssets::Load(GetTestDataPath() + "/appaslib/appaslib.apk");
65 ASSERT_THAT(loaded_apk, NotNull());
66
67 const LoadedArsc* loaded_arsc = loaded_apk->GetLoadedArsc();
68 ASSERT_THAT(loaded_arsc, NotNull());
69 ASSERT_THAT(loaded_arsc->GetPackages(), SizeIs(1u));
70 EXPECT_FALSE(loaded_arsc->GetPackages()[0]->IsDynamic());
71
72 loaded_apk = ApkAssets::Load(GetTestDataPath() + "/appaslib/appaslib.apk", PROPERTY_DYNAMIC);
73 ASSERT_THAT(loaded_apk, NotNull());
74
75 loaded_arsc = loaded_apk->GetLoadedArsc();
76 ASSERT_THAT(loaded_arsc, NotNull());
77 ASSERT_THAT(loaded_arsc->GetPackages(), SizeIs(1u));
78 EXPECT_TRUE(loaded_arsc->GetPackages()[0]->IsDynamic());
79 }
80
TEST(ApkAssetsTest,CreateAndDestroyAssetKeepsApkAssetsOpen)81 TEST(ApkAssetsTest, CreateAndDestroyAssetKeepsApkAssetsOpen) {
82 std::unique_ptr<const ApkAssets> loaded_apk =
83 ApkAssets::Load(GetTestDataPath() + "/basic/basic.apk");
84 ASSERT_THAT(loaded_apk, NotNull());
85
86 { ASSERT_THAT(loaded_apk->GetAssetsProvider()->Open("res/layout/main.xml",
87 Asset::ACCESS_BUFFER), NotNull()); }
88
89 { ASSERT_THAT(loaded_apk->GetAssetsProvider()->Open("res/layout/main.xml",
90 Asset::ACCESS_BUFFER), NotNull()); }
91 }
92
TEST(ApkAssetsTest,OpenUncompressedAssetFd)93 TEST(ApkAssetsTest, OpenUncompressedAssetFd) {
94 std::unique_ptr<const ApkAssets> loaded_apk =
95 ApkAssets::Load(GetTestDataPath() + "/basic/basic.apk");
96 ASSERT_THAT(loaded_apk, NotNull());
97
98 auto asset = loaded_apk->GetAssetsProvider()->Open("assets/uncompressed.txt",
99 Asset::ACCESS_UNKNOWN);
100 ASSERT_THAT(asset, NotNull());
101
102 off64_t start, length;
103 unique_fd fd(asset->openFileDescriptor(&start, &length));
104 ASSERT_THAT(fd.get(), Ge(0));
105
106 lseek64(fd.get(), start, SEEK_SET);
107
108 std::string buffer;
109 buffer.resize(length);
110 ASSERT_TRUE(base::ReadFully(fd.get(), &*buffer.begin(), length));
111
112 EXPECT_THAT(buffer, StrEq("This should be uncompressed.\n\n"));
113 }
114
115 } // namespace android
116