OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "courgette/disassembler_elf_32_x86.h" | 5 #include "courgette/disassembler_elf_32_x86.h" |
6 | 6 |
7 #include <ctype.h> | 7 #include <ctype.h> |
8 #include <stddef.h> | 8 #include <stddef.h> |
9 #include <stdint.h> | 9 #include <stdint.h> |
10 | 10 |
(...skipping 58 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
69 size_t expected_abs_count, | 69 size_t expected_abs_count, |
70 size_t expected_rel_count) const { | 70 size_t expected_rel_count) const { |
71 std::string file1 = FileContents(file_name); | 71 std::string file1 = FileContents(file_name); |
72 | 72 |
73 auto disassembler = base::MakeUnique<TestDisassemblerElf32X86>( | 73 auto disassembler = base::MakeUnique<TestDisassemblerElf32X86>( |
74 reinterpret_cast<const uint8_t*>(file1.c_str()), file1.length()); | 74 reinterpret_cast<const uint8_t*>(file1.c_str()), file1.length()); |
75 | 75 |
76 bool can_parse_header = disassembler->ParseHeader(); | 76 bool can_parse_header = disassembler->ParseHeader(); |
77 EXPECT_TRUE(can_parse_header); | 77 EXPECT_TRUE(can_parse_header); |
78 EXPECT_TRUE(disassembler->ok()); | 78 EXPECT_TRUE(disassembler->ok()); |
| 79 EXPECT_EQ(EXE_ELF_32_X86, disassembler->kind()); |
| 80 EXPECT_EQ(0U, disassembler->image_base()); |
79 | 81 |
80 // The length of the disassembled value will be slightly smaller than the | 82 // The length of the disassembled value will be slightly smaller than the |
81 // real file, since trailing debug info is not included | 83 // real file, since trailing debug info is not included |
82 EXPECT_EQ(file1.length(), disassembler->length()); | 84 EXPECT_EQ(file1.length(), disassembler->length()); |
83 | 85 |
84 const uint8_t* offset_p = disassembler->FileOffsetToPointer(0); | 86 const uint8_t* offset_p = disassembler->FileOffsetToPointer(0); |
85 EXPECT_EQ(reinterpret_cast<const void*>(file1.c_str()), | 87 EXPECT_EQ(reinterpret_cast<const void*>(file1.c_str()), |
86 reinterpret_cast<const void*>(offset_p)); | 88 reinterpret_cast<const void*>(offset_p)); |
87 EXPECT_EQ(0x7F, offset_p[0]); | 89 EXPECT_EQ(0x7F, offset_p[0]); |
88 EXPECT_EQ('E', offset_p[1]); | 90 EXPECT_EQ('E', offset_p[1]); |
89 EXPECT_EQ('L', offset_p[2]); | 91 EXPECT_EQ('L', offset_p[2]); |
90 EXPECT_EQ('F', offset_p[3]); | 92 EXPECT_EQ('F', offset_p[3]); |
91 | 93 |
92 auto program = base::MakeUnique<AssemblyProgram>(EXE_ELF_32_X86, 0); | 94 std::unique_ptr<AssemblyProgram> program = disassembler->Disassemble(); |
93 | 95 EXPECT_TRUE(nullptr != program.get()); |
94 EXPECT_TRUE(disassembler->Disassemble(program.get())); | |
95 | 96 |
96 const std::vector<RVA>& abs32_list = disassembler->Abs32Locations(); | 97 const std::vector<RVA>& abs32_list = disassembler->Abs32Locations(); |
97 | 98 |
98 // Flatten the list typed rel32 to a list of rel32 RVAs. | 99 // Flatten the list typed rel32 to a list of rel32 RVAs. |
99 std::vector<RVA> rel32_list; | 100 std::vector<RVA> rel32_list; |
100 rel32_list.reserve(disassembler->Rel32Locations().size()); | 101 rel32_list.reserve(disassembler->Rel32Locations().size()); |
101 for (auto& typed_rel32 : disassembler->Rel32Locations()) | 102 for (auto& typed_rel32 : disassembler->Rel32Locations()) |
102 rel32_list.push_back(typed_rel32->rva()); | 103 rel32_list.push_back(typed_rel32->rva()); |
103 | 104 |
104 EXPECT_EQ(expected_abs_count, abs32_list.size()); | 105 EXPECT_EQ(expected_abs_count, abs32_list.size()); |
(...skipping 24 matching lines...) Expand all Loading... |
129 } | 130 } |
130 | 131 |
131 } // namespace | 132 } // namespace |
132 | 133 |
133 TEST_F(DisassemblerElf32X86Test, All) { | 134 TEST_F(DisassemblerElf32X86Test, All) { |
134 TestExe("elf-32-1", 200, 3337); | 135 TestExe("elf-32-1", 200, 3337); |
135 TestExe("elf-32-high-bss", 0, 4); | 136 TestExe("elf-32-high-bss", 0, 4); |
136 } | 137 } |
137 | 138 |
138 } // namespace courgette | 139 } // namespace courgette |
OLD | NEW |