blob: 23ffad6e78d4ea3b91e8d98900eed984332a6b45 [file] [log] [blame]
/*
* 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