| /* |
| * Copyright (C) 2011 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 "art_dex_file_loader.h" |
| |
| #include <sys/mman.h> |
| |
| #include <memory> |
| #include <optional> |
| #include <vector> |
| |
| #include "base/common_art_test.h" |
| #include "base/mem_map.h" |
| #include "base/os.h" |
| #include "base/stl_util.h" |
| #include "base/unix_file/fd_file.h" |
| #include "dex/base64_test_util.h" |
| #include "dex/class_accessor-inl.h" |
| #include "dex/code_item_accessors-inl.h" |
| #include "dex/descriptors_names.h" |
| #include "dex/dex_file-inl.h" |
| #include "dex/dex_file.h" |
| #include "dex/dex_file_loader.h" |
| |
| namespace art { |
| |
| class ArtDexFileLoaderTest : public CommonArtTest { |
| void SetUp() override { |
| CommonArtTest::SetUp(); |
| // Open a jar file from the boot classpath for use in basic tests of dex accessors. |
| std::vector<std::string> lib_core_dex_file_names = GetLibCoreDexFileNames(); |
| CHECK_NE(lib_core_dex_file_names.size(), 0U); |
| dex_files_ = OpenDexFiles(lib_core_dex_file_names[0].c_str()); |
| CHECK_NE(dex_files_.size(), 0U); |
| // Save a dex file for use by tests. |
| java_lang_dex_file_ = dex_files_[0].get(); |
| } |
| |
| protected: |
| std::vector<std::unique_ptr<const DexFile>> dex_files_; |
| const DexFile* java_lang_dex_file_; |
| }; |
| |
| TEST_F(ArtDexFileLoaderTest, Open) { |
| std::unique_ptr<const DexFile> dex(OpenTestDexFile("Nested")); |
| ASSERT_TRUE(dex.get() != nullptr); |
| } |
| |
| TEST_F(ArtDexFileLoaderTest, OpenZipMultiDex) { |
| std::string zip_file = GetTestDexFileName("MultiDex"); |
| File file(zip_file, O_RDONLY, /*check_usage=*/false); |
| ASSERT_GE(file.Fd(), 0); |
| std::vector<std::unique_ptr<const DexFile>> dex_files; |
| std::string error_msg; |
| ArtDexFileLoader dex_file_loader(&file, zip_file); |
| ASSERT_TRUE(dex_file_loader.Open(/*verify=*/false, |
| /*verify_checksum=*/true, |
| /*allow_no_dex_files=*/true, |
| &error_msg, |
| &dex_files)) |
| << error_msg; |
| EXPECT_GT(dex_files.size(), 1); |
| } |
| |
| TEST_F(ArtDexFileLoaderTest, OpenZipEmpty) { |
| std::string zip_file = GetTestDexFileName("MainEmptyUncompressed"); |
| File file(zip_file, O_RDONLY, /*check_usage=*/false); |
| ASSERT_GE(file.Fd(), 0); |
| std::vector<std::unique_ptr<const DexFile>> dex_files; |
| std::string error_msg; |
| ArtDexFileLoader dex_file_loader(&file, zip_file); |
| ASSERT_TRUE(dex_file_loader.Open(/*verify=*/false, |
| /*verify_checksum=*/true, |
| /*allow_no_dex_files=*/true, |
| &error_msg, |
| &dex_files)) |
| << error_msg; |
| EXPECT_EQ(dex_files.size(), 0); |
| } |
| |
| TEST_F(ArtDexFileLoaderTest, GetLocationChecksum) { |
| std::unique_ptr<const DexFile> raw(OpenTestDexFile("Main")); |
| EXPECT_NE(raw->GetHeader().checksum_, raw->GetLocationChecksum()); |
| } |
| |
| TEST_F(ArtDexFileLoaderTest, GetChecksum) { |
| std::optional<uint32_t> checksum; |
| std::string error_msg; |
| ArtDexFileLoader dex_loader(GetLibCoreDexFileNames()[0]); |
| ASSERT_TRUE(dex_loader.GetMultiDexChecksum(&checksum, &error_msg)) << error_msg; |
| ASSERT_TRUE(checksum.has_value()); |
| |
| std::vector<const DexFile*> dex_files{java_lang_dex_file_}; |
| uint32_t expected_checksum = DexFileLoader::GetMultiDexChecksum(dex_files); |
| EXPECT_EQ(expected_checksum, checksum.value()); |
| } |
| |
| TEST_F(ArtDexFileLoaderTest, GetMultiDexChecksum) { |
| std::string error_msg; |
| std::optional<uint32_t> checksum; |
| std::string multidex_file = GetTestDexFileName("MultiDex"); |
| ArtDexFileLoader dex_loader(multidex_file); |
| EXPECT_TRUE(dex_loader.GetMultiDexChecksum(&checksum, &error_msg)) << error_msg; |
| |
| std::vector<std::unique_ptr<const DexFile>> dexes = OpenTestDexFiles("MultiDex"); |
| ASSERT_EQ(2U, dexes.size()); |
| ASSERT_TRUE(checksum.has_value()); |
| |
| uint32_t expected_checksum = DexFileLoader::GetMultiDexChecksum(dexes); |
| EXPECT_EQ(dexes[0]->GetLocation(), DexFileLoader::GetMultiDexLocation(0, multidex_file.c_str())); |
| EXPECT_EQ(dexes[1]->GetLocation(), DexFileLoader::GetMultiDexLocation(1, multidex_file.c_str())); |
| EXPECT_EQ(expected_checksum, checksum.value()); |
| } |
| |
| TEST_F(ArtDexFileLoaderTest, GetMultiDexChecksumsEmptyZip) { |
| std::string error_msg; |
| std::optional<uint32_t> checksum; |
| std::string multidex_file = GetTestDexFileName("MainEmptyUncompressed"); |
| ArtDexFileLoader dex_loader(multidex_file); |
| EXPECT_TRUE(dex_loader.GetMultiDexChecksum(&checksum, &error_msg)) << error_msg; |
| EXPECT_FALSE(checksum.has_value()); |
| } |
| |
| TEST_F(ArtDexFileLoaderTest, GetMultiDexChecksumsDexFile) { |
| std::string error_msg; |
| std::optional<uint32_t> checksum; |
| std::string multidex_file = GetTestDexFileName("VerifierDeps"); // This is a .dex file. |
| DexFileLoader loader(multidex_file); |
| EXPECT_TRUE(loader.GetMultiDexChecksum(&checksum, &error_msg)) << error_msg; |
| EXPECT_TRUE(checksum.has_value()); |
| } |
| |
| TEST_F(ArtDexFileLoaderTest, ClassDefs) { |
| std::unique_ptr<const DexFile> raw(OpenTestDexFile("Nested")); |
| ASSERT_TRUE(raw.get() != nullptr); |
| EXPECT_EQ(3U, raw->NumClassDefs()); |
| |
| const dex::ClassDef& c0 = raw->GetClassDef(0); |
| EXPECT_STREQ("LNested$1;", raw->GetClassDescriptor(c0)); |
| |
| const dex::ClassDef& c1 = raw->GetClassDef(1); |
| EXPECT_STREQ("LNested$Inner;", raw->GetClassDescriptor(c1)); |
| |
| const dex::ClassDef& c2 = raw->GetClassDef(2); |
| EXPECT_STREQ("LNested;", raw->GetClassDescriptor(c2)); |
| } |
| |
| TEST_F(ArtDexFileLoaderTest, GetMethodSignature) { |
| std::unique_ptr<const DexFile> raw(OpenTestDexFile("GetMethodSignature")); |
| ASSERT_TRUE(raw.get() != nullptr); |
| EXPECT_EQ(1U, raw->NumClassDefs()); |
| |
| const dex::ClassDef& class_def = raw->GetClassDef(0); |
| ASSERT_STREQ("LGetMethodSignature;", raw->GetClassDescriptor(class_def)); |
| |
| ClassAccessor accessor(*raw, class_def); |
| ASSERT_TRUE(accessor.HasClassData()); |
| auto methods = accessor.GetMethods(); |
| auto cur_method = methods.begin(); |
| |
| // Check the signature for the static initializer. |
| { |
| ASSERT_EQ(1U, accessor.NumDirectMethods()); |
| const dex::MethodId& method_id = raw->GetMethodId(cur_method->GetIndex()); |
| const char* name = raw->StringDataByIdx(method_id.name_idx_); |
| ASSERT_STREQ("<init>", name); |
| std::string signature(raw->GetMethodSignature(method_id).ToString()); |
| ASSERT_EQ("()V", signature); |
| } |
| |
| // Check all virtual methods. |
| struct Result { |
| const char* name; |
| const char* signature; |
| const char* pretty_method; |
| }; |
| static const Result results[] = { |
| { |
| "m1", |
| "(IDJLjava/lang/Object;)Ljava/lang/Float;", |
| "java.lang.Float GetMethodSignature.m1(int, double, long, java.lang.Object)" |
| }, |
| { |
| "m2", |
| "(ZSC)LGetMethodSignature;", |
| "GetMethodSignature GetMethodSignature.m2(boolean, short, char)" |
| }, |
| { |
| "m3", |
| "()V", |
| "void GetMethodSignature.m3()" |
| }, |
| { |
| "m4", |
| "(I)V", |
| "void GetMethodSignature.m4(int)" |
| }, |
| { |
| "m5", |
| "(II)V", |
| "void GetMethodSignature.m5(int, int)" |
| }, |
| { |
| "m6", |
| "(II[[I)V", |
| "void GetMethodSignature.m6(int, int, int[][])" |
| }, |
| { |
| "m7", |
| "(II[[ILjava/lang/Object;)V", |
| "void GetMethodSignature.m7(int, int, int[][], java.lang.Object)" |
| }, |
| { |
| "m8", |
| "(II[[ILjava/lang/Object;[[Ljava/lang/Object;)V", |
| "void GetMethodSignature.m8(int, int, int[][], java.lang.Object, java.lang.Object[][])" |
| }, |
| { |
| "m9", |
| "()I", |
| "int GetMethodSignature.m9()" |
| }, |
| { |
| "mA", |
| "()[[I", |
| "int[][] GetMethodSignature.mA()" |
| }, |
| { |
| "mB", |
| "()[[Ljava/lang/Object;", |
| "java.lang.Object[][] GetMethodSignature.mB()" |
| }, |
| }; |
| ASSERT_EQ(arraysize(results), accessor.NumVirtualMethods()); |
| for (const Result& r : results) { |
| ++cur_method; |
| ASSERT_TRUE(cur_method != methods.end()); |
| const dex::MethodId& method_id = raw->GetMethodId(cur_method->GetIndex()); |
| |
| const char* name = raw->StringDataByIdx(method_id.name_idx_); |
| ASSERT_STREQ(r.name, name); |
| |
| std::string signature(raw->GetMethodSignature(method_id).ToString()); |
| ASSERT_EQ(r.signature, signature); |
| |
| std::string plain_method = std::string("GetMethodSignature.") + r.name; |
| ASSERT_EQ(plain_method, |
| raw->PrettyMethod(cur_method->GetIndex(), /* with_signature= */ false)); |
| ASSERT_EQ(r.pretty_method, |
| raw->PrettyMethod(cur_method->GetIndex(), /* with_signature= */ true)); |
| } |
| } |
| |
| TEST_F(ArtDexFileLoaderTest, FindStringId) { |
| std::unique_ptr<const DexFile> raw(OpenTestDexFile("GetMethodSignature")); |
| ASSERT_TRUE(raw.get() != nullptr); |
| EXPECT_EQ(1U, raw->NumClassDefs()); |
| |
| const char* strings[] = { "LGetMethodSignature;", "Ljava/lang/Float;", "Ljava/lang/Object;", |
| "D", "I", "J", nullptr }; |
| for (size_t i = 0; strings[i] != nullptr; i++) { |
| const char* str = strings[i]; |
| const dex::StringId* str_id = raw->FindStringId(str); |
| const char* dex_str = raw->GetStringData(*str_id); |
| EXPECT_STREQ(dex_str, str); |
| } |
| } |
| |
| TEST_F(ArtDexFileLoaderTest, FindTypeId) { |
| for (size_t i = 0; i < java_lang_dex_file_->NumTypeIds(); i++) { |
| const char* type_str = java_lang_dex_file_->StringByTypeIdx(dex::TypeIndex(i)); |
| const dex::StringId* type_str_id = java_lang_dex_file_->FindStringId(type_str); |
| ASSERT_TRUE(type_str_id != nullptr); |
| dex::StringIndex type_str_idx = java_lang_dex_file_->GetIndexForStringId(*type_str_id); |
| const dex::TypeId* type_id = java_lang_dex_file_->FindTypeId(type_str_idx); |
| ASSERT_EQ(type_id, java_lang_dex_file_->FindTypeId(type_str)); |
| ASSERT_TRUE(type_id != nullptr); |
| EXPECT_EQ(java_lang_dex_file_->GetIndexForTypeId(*type_id).index_, i); |
| } |
| } |
| |
| TEST_F(ArtDexFileLoaderTest, FindProtoId) { |
| for (size_t i = 0; i < java_lang_dex_file_->NumProtoIds(); i++) { |
| const dex::ProtoId& to_find = java_lang_dex_file_->GetProtoId(dex::ProtoIndex(i)); |
| const dex::TypeList* to_find_tl = java_lang_dex_file_->GetProtoParameters(to_find); |
| std::vector<dex::TypeIndex> to_find_types; |
| if (to_find_tl != nullptr) { |
| for (size_t j = 0; j < to_find_tl->Size(); j++) { |
| to_find_types.push_back(to_find_tl->GetTypeItem(j).type_idx_); |
| } |
| } |
| const dex::ProtoId* found = |
| java_lang_dex_file_->FindProtoId(to_find.return_type_idx_, to_find_types); |
| ASSERT_TRUE(found != nullptr); |
| EXPECT_EQ(java_lang_dex_file_->GetIndexForProtoId(*found), dex::ProtoIndex(i)); |
| } |
| } |
| |
| TEST_F(ArtDexFileLoaderTest, FindMethodId) { |
| for (size_t i = 0; i < java_lang_dex_file_->NumMethodIds(); i++) { |
| const dex::MethodId& to_find = java_lang_dex_file_->GetMethodId(i); |
| const dex::TypeId& klass = java_lang_dex_file_->GetTypeId(to_find.class_idx_); |
| const dex::StringId& name = java_lang_dex_file_->GetStringId(to_find.name_idx_); |
| const dex::ProtoId& signature = java_lang_dex_file_->GetProtoId(to_find.proto_idx_); |
| const dex::MethodId* found = java_lang_dex_file_->FindMethodId(klass, name, signature); |
| ASSERT_TRUE(found != nullptr) << "Didn't find method " << i << ": " |
| << java_lang_dex_file_->StringByTypeIdx(to_find.class_idx_) << "." |
| << java_lang_dex_file_->GetStringData(name) |
| << java_lang_dex_file_->GetMethodSignature(to_find); |
| EXPECT_EQ(java_lang_dex_file_->GetIndexForMethodId(*found), i); |
| } |
| } |
| |
| TEST_F(ArtDexFileLoaderTest, FindFieldId) { |
| for (size_t i = 0; i < java_lang_dex_file_->NumFieldIds(); i++) { |
| const dex::FieldId& to_find = java_lang_dex_file_->GetFieldId(i); |
| const dex::TypeId& klass = java_lang_dex_file_->GetTypeId(to_find.class_idx_); |
| const dex::StringId& name = java_lang_dex_file_->GetStringId(to_find.name_idx_); |
| const dex::TypeId& type = java_lang_dex_file_->GetTypeId(to_find.type_idx_); |
| const dex::FieldId* found = java_lang_dex_file_->FindFieldId(klass, name, type); |
| ASSERT_TRUE(found != nullptr) << "Didn't find field " << i << ": " |
| << java_lang_dex_file_->StringByTypeIdx(to_find.type_idx_) << " " |
| << java_lang_dex_file_->StringByTypeIdx(to_find.class_idx_) << "." |
| << java_lang_dex_file_->GetStringData(name); |
| EXPECT_EQ(java_lang_dex_file_->GetIndexForFieldId(*found), i); |
| } |
| } |
| |
| TEST_F(ArtDexFileLoaderTest, GetDexCanonicalLocation) { |
| ScratchFile file; |
| UniqueCPtr<const char[]> dex_location_real(realpath(file.GetFilename().c_str(), nullptr)); |
| std::string dex_location(dex_location_real.get()); |
| |
| ASSERT_EQ(dex_location, DexFileLoader::GetDexCanonicalLocation(dex_location.c_str())); |
| std::string multidex_location = DexFileLoader::GetMultiDexLocation(1, dex_location.c_str()); |
| ASSERT_EQ(multidex_location, DexFileLoader::GetDexCanonicalLocation(multidex_location.c_str())); |
| |
| std::string dex_location_sym = dex_location + "symlink"; |
| ASSERT_EQ(0, symlink(dex_location.c_str(), dex_location_sym.c_str())); |
| |
| ASSERT_EQ(dex_location, DexFileLoader::GetDexCanonicalLocation(dex_location_sym.c_str())); |
| |
| std::string multidex_location_sym = DexFileLoader::GetMultiDexLocation( |
| 1, dex_location_sym.c_str()); |
| ASSERT_EQ(multidex_location, |
| DexFileLoader::GetDexCanonicalLocation(multidex_location_sym.c_str())); |
| |
| ASSERT_EQ(0, unlink(dex_location_sym.c_str())); |
| } |
| |
| } // namespace art |