/* * Copyright (C) 2018 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 <stdint.h> #include <sys/types.h> #include <unistd.h> #include <unordered_map> #include <android-base/test_utils.h> #include <unwindstack/MapInfo.h> #include <unwindstack/Memory.h> #include <dex/code_item_accessors-inl.h> #include <dex/standard_dex_file.h> #include <gtest/gtest.h> #include "DexFile.h" #include "DexFileData.h" #include "MemoryFake.h" namespace unwindstack { TEST(DexFileTest, from_file_open_non_exist) { DexFileFromFile dex_file; ASSERT_FALSE(dex_file.Open(0, "/file/does/not/exist")); } TEST(DexFileTest, from_file_open_too_small) { TemporaryFile tf; ASSERT_TRUE(tf.fd != -1); ASSERT_EQ(sizeof(art::DexFile::Header) - 2, static_cast<size_t>( TEMP_FAILURE_RETRY(write(tf.fd, kDexData, sizeof(art::DexFile::Header)) - 2))); // Header too small. DexFileFromFile dex_file; ASSERT_FALSE(dex_file.Open(0, tf.path)); // Header correct, file too small. ASSERT_EQ(0, lseek(tf.fd, 0, SEEK_SET)); ASSERT_EQ(sizeof(art::DexFile::Header), static_cast<size_t>(TEMP_FAILURE_RETRY(write( tf.fd, kDexData, sizeof(art::DexFile::Header))))); ASSERT_FALSE(dex_file.Open(0, tf.path)); } TEST(DexFileTest, from_file_open) { TemporaryFile tf; ASSERT_TRUE(tf.fd != -1); ASSERT_EQ(sizeof(kDexData), static_cast<size_t>(TEMP_FAILURE_RETRY(write(tf.fd, kDexData, sizeof(kDexData))))); DexFileFromFile dex_file; ASSERT_TRUE(dex_file.Open(0, tf.path)); } TEST(DexFileTest, from_file_open_non_zero_offset) { TemporaryFile tf; ASSERT_TRUE(tf.fd != -1); ASSERT_EQ(0x100, lseek(tf.fd, 0x100, SEEK_SET)); ASSERT_EQ(sizeof(kDexData), static_cast<size_t>(TEMP_FAILURE_RETRY(write(tf.fd, kDexData, sizeof(kDexData))))); DexFileFromFile dex_file; ASSERT_TRUE(dex_file.Open(0x100, tf.path)); } TEST(DexFileTest, from_memory_fail_too_small_for_header) { MemoryFake memory; memory.SetMemory(0x1000, kDexData, sizeof(art::DexFile::Header) - 1); DexFileFromMemory dex_file; ASSERT_FALSE(dex_file.Open(0x1000, &memory)); } TEST(DexFileTest, from_memory_fail_too_small_for_data) { MemoryFake memory; memory.SetMemory(0x1000, kDexData, sizeof(kDexData) - 2); DexFileFromMemory dex_file; ASSERT_FALSE(dex_file.Open(0x1000, &memory)); } TEST(DexFileTest, from_memory_open) { MemoryFake memory; memory.SetMemory(0x1000, kDexData, sizeof(kDexData)); DexFileFromMemory dex_file; ASSERT_TRUE(dex_file.Open(0x1000, &memory)); } TEST(DexFileTest, create_using_file) { TemporaryFile tf; ASSERT_TRUE(tf.fd != -1); ASSERT_EQ(0x500, lseek(tf.fd, 0x500, SEEK_SET)); ASSERT_EQ(sizeof(kDexData), static_cast<size_t>(TEMP_FAILURE_RETRY(write(tf.fd, kDexData, sizeof(kDexData))))); MemoryFake memory; MapInfo info(0, 0x10000, 0, 0x5, tf.path); std::unique_ptr<DexFile> dex_file(DexFile::Create(0x500, &memory, &info)); ASSERT_TRUE(dex_file != nullptr); } TEST(DexFileTest, create_using_file_non_zero_start) { TemporaryFile tf; ASSERT_TRUE(tf.fd != -1); ASSERT_EQ(0x500, lseek(tf.fd, 0x500, SEEK_SET)); ASSERT_EQ(sizeof(kDexData), static_cast<size_t>(TEMP_FAILURE_RETRY(write(tf.fd, kDexData, sizeof(kDexData))))); MemoryFake memory; MapInfo info(0x100, 0x10000, 0, 0x5, tf.path); std::unique_ptr<DexFile> dex_file(DexFile::Create(0x600, &memory, &info)); ASSERT_TRUE(dex_file != nullptr); } TEST(DexFileTest, create_using_file_non_zero_offset) { TemporaryFile tf; ASSERT_TRUE(tf.fd != -1); ASSERT_EQ(0x500, lseek(tf.fd, 0x500, SEEK_SET)); ASSERT_EQ(sizeof(kDexData), static_cast<size_t>(TEMP_FAILURE_RETRY(write(tf.fd, kDexData, sizeof(kDexData))))); MemoryFake memory; MapInfo info(0x100, 0x10000, 0x200, 0x5, tf.path); std::unique_ptr<DexFile> dex_file(DexFile::Create(0x400, &memory, &info)); ASSERT_TRUE(dex_file != nullptr); } TEST(DexFileTest, create_using_memory_empty_file) { MemoryFake memory; memory.SetMemory(0x4000, kDexData, sizeof(kDexData)); MapInfo info(0x100, 0x10000, 0x200, 0x5, ""); std::unique_ptr<DexFile> dex_file(DexFile::Create(0x4000, &memory, &info)); ASSERT_TRUE(dex_file != nullptr); } TEST(DexFileTest, create_using_memory_file_does_not_exist) { MemoryFake memory; memory.SetMemory(0x4000, kDexData, sizeof(kDexData)); MapInfo info(0x100, 0x10000, 0x200, 0x5, "/does/not/exist"); std::unique_ptr<DexFile> dex_file(DexFile::Create(0x4000, &memory, &info)); ASSERT_TRUE(dex_file != nullptr); } TEST(DexFileTest, create_using_memory_file_is_malformed) { TemporaryFile tf; ASSERT_TRUE(tf.fd != -1); ASSERT_EQ(sizeof(kDexData) - 10, static_cast<size_t>(TEMP_FAILURE_RETRY(write(tf.fd, kDexData, sizeof(kDexData) - 10)))); MemoryFake memory; memory.SetMemory(0x4000, kDexData, sizeof(kDexData)); MapInfo info(0x4000, 0x10000, 0x200, 0x5, "/does/not/exist"); std::unique_ptr<DexFile> dex_file(DexFile::Create(0x4000, &memory, &info)); ASSERT_TRUE(dex_file != nullptr); // Check it came from memory by clearing memory and verifying it fails. memory.Clear(); dex_file.reset(DexFile::Create(0x4000, &memory, &info)); ASSERT_TRUE(dex_file == nullptr); } TEST(DexFileTest, get_method_not_opened) { std::string method("something"); uint64_t method_offset = 100; DexFile dex_file; dex_file.GetMethodInformation(0x100, &method, &method_offset); EXPECT_EQ("something", method); EXPECT_EQ(100U, method_offset); } TEST(DexFileTest, get_method) { MemoryFake memory; memory.SetMemory(0x4000, kDexData, sizeof(kDexData)); MapInfo info(0x100, 0x10000, 0x200, 0x5, ""); std::unique_ptr<DexFile> dex_file(DexFile::Create(0x4000, &memory, &info)); ASSERT_TRUE(dex_file != nullptr); std::string method; uint64_t method_offset; dex_file->GetMethodInformation(0x102, &method, &method_offset); EXPECT_EQ("Main.<init>", method); EXPECT_EQ(2U, method_offset); method = "not_in_a_method"; method_offset = 0x123; dex_file->GetMethodInformation(0x100000, &method, &method_offset); EXPECT_EQ("not_in_a_method", method); EXPECT_EQ(0x123U, method_offset); } } // namespace unwindstack