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 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 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 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 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 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