OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 #ifndef COURGETTE_DISASSEMBLER_WIN32_H_ | 5 #ifndef COURGETTE_DISASSEMBLER_WIN32_H_ |
6 #define COURGETTE_DISASSEMBLER_WIN32_H_ | 6 #define COURGETTE_DISASSEMBLER_WIN32_H_ |
7 | 7 |
8 #include <stddef.h> | 8 #include <stddef.h> |
9 #include <stdint.h> | 9 #include <stdint.h> |
10 | 10 |
11 #include <map> | 11 #include <map> |
12 #include <string> | 12 #include <string> |
13 #include <vector> | 13 #include <vector> |
14 | 14 |
15 #include "base/macros.h" | 15 #include "base/macros.h" |
16 #include "courgette/disassembler.h" | 16 #include "courgette/disassembler.h" |
17 #include "courgette/image_utils.h" | 17 #include "courgette/image_utils.h" |
| 18 #include "courgette/instruction_utils.h" |
18 #include "courgette/memory_allocator.h" | 19 #include "courgette/memory_allocator.h" |
19 #include "courgette/types_win_pe.h" | 20 #include "courgette/types_win_pe.h" |
20 | 21 |
21 namespace courgette { | 22 namespace courgette { |
22 | 23 |
23 class AssemblyProgram; | 24 class AssemblyProgram; |
24 class InstructionReceptor; | |
25 | 25 |
26 class DisassemblerWin32 : public Disassembler { | 26 class DisassemblerWin32 : public Disassembler { |
27 public: | 27 public: |
28 virtual ~DisassemblerWin32() = default; | 28 virtual ~DisassemblerWin32() = default; |
29 | 29 |
30 // Disassembler interfaces. | 30 // Disassembler interfaces. |
31 RVA FileOffsetToRVA(FileOffset file_offset) const override; | 31 RVA FileOffsetToRVA(FileOffset file_offset) const override; |
32 FileOffset RVAToFileOffset(RVA rva) const override; | 32 FileOffset RVAToFileOffset(RVA rva) const override; |
33 ExecutableType kind() const override = 0; | 33 ExecutableType kind() const override = 0; |
34 uint64_t image_base() const override { return image_base_; } | 34 uint64_t image_base() const override { return image_base_; } |
35 RVA PointerToTargetRVA(const uint8_t* p) const override = 0; | 35 RVA PointerToTargetRVA(const uint8_t* p) const override = 0; |
36 bool ParseHeader() override; | 36 bool ParseHeader() override; |
37 bool Disassemble(AssemblyProgram* program) override; | |
38 | 37 |
39 // Exposed for test purposes | 38 // Exposed for test purposes |
40 bool has_text_section() const { return has_text_section_; } | 39 bool has_text_section() const { return has_text_section_; } |
41 uint32_t size_of_code() const { return size_of_code_; } | 40 uint32_t size_of_code() const { return size_of_code_; } |
42 | 41 |
43 // Returns 'true' if the base relocation table can be parsed. | 42 // Returns 'true' if the base relocation table can be parsed. |
44 // Output is a vector of the RVAs corresponding to locations within executable | 43 // Output is a vector of the RVAs corresponding to locations within executable |
45 // that are listed in the base relocation table. | 44 // that are listed in the base relocation table. |
46 bool ParseRelocs(std::vector<RVA>* addresses); | 45 bool ParseRelocs(std::vector<RVA>* addresses); |
47 | 46 |
48 // Returns Section containing the relative virtual address, or null if none. | 47 // Returns Section containing the relative virtual address, or null if none. |
49 const Section* RVAToSection(RVA rva) const; | 48 const Section* RVAToSection(RVA rva) const; |
50 | 49 |
51 static std::string SectionName(const Section* section); | 50 static std::string SectionName(const Section* section); |
52 | 51 |
53 protected: | 52 protected: |
54 // Returns true if a valid executable is detected using only quick checks. | 53 // Returns true if a valid executable is detected using only quick checks. |
55 // Derived classes should inject |magic| corresponding to their architecture, | 54 // Derived classes should inject |magic| corresponding to their architecture, |
56 // which will be checked against the detected one. | 55 // which will be checked against the detected one. |
57 static bool QuickDetect(const uint8_t* start, size_t length, uint16_t magic); | 56 static bool QuickDetect(const uint8_t* start, size_t length, uint16_t magic); |
58 | 57 |
59 bool ParseAbs32Relocs(); | 58 bool ParseAbs32Relocs(); |
60 void ParseRel32RelocsFromSections(); | 59 void ParseRel32RelocsFromSections(); |
61 | 60 |
62 // Disassembler interfaces. | 61 // Disassembler interfaces. |
| 62 bool ExtractAbs32Locations() override; |
| 63 bool ExtractRel32Locations() override; |
63 RvaVisitor* CreateAbs32TargetRvaVisitor() override; | 64 RvaVisitor* CreateAbs32TargetRvaVisitor() override; |
64 RvaVisitor* CreateRel32TargetRvaVisitor() override; | 65 RvaVisitor* CreateRel32TargetRvaVisitor() override; |
65 void RemoveUnusedRel32Locations(AssemblyProgram* program) override; | 66 void RemoveUnusedRel32Locations(AssemblyProgram* program) override; |
| 67 InstructionGenerator GetInstructionGenerator( |
| 68 AssemblyProgram* program) override; |
66 | 69 |
67 DisassemblerWin32(const uint8_t* start, size_t length); | 70 DisassemblerWin32(const uint8_t* start, size_t length); |
68 | 71 |
69 CheckBool ParseFile(AssemblyProgram* target, | 72 CheckBool ParseFile(AssemblyProgram* target, |
70 InstructionReceptor* receptor) const WARN_UNUSED_RESULT; | 73 InstructionReceptor* receptor) const WARN_UNUSED_RESULT; |
71 virtual void ParseRel32RelocsFromSection(const Section* section) = 0; | 74 virtual void ParseRel32RelocsFromSection(const Section* section) = 0; |
72 | 75 |
73 CheckBool ParseNonSectionFileRegion(FileOffset start_file_offset, | 76 CheckBool ParseNonSectionFileRegion(FileOffset start_file_offset, |
74 FileOffset end_file_offset, | 77 FileOffset end_file_offset, |
75 InstructionReceptor* receptor) const | 78 InstructionReceptor* receptor) const |
(...skipping 71 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
147 std::map<RVA, int> rel32_target_rvas_; | 150 std::map<RVA, int> rel32_target_rvas_; |
148 #endif | 151 #endif |
149 | 152 |
150 private: | 153 private: |
151 DISALLOW_COPY_AND_ASSIGN(DisassemblerWin32); | 154 DISALLOW_COPY_AND_ASSIGN(DisassemblerWin32); |
152 }; | 155 }; |
153 | 156 |
154 } // namespace courgette | 157 } // namespace courgette |
155 | 158 |
156 #endif // COURGETTE_DISASSEMBLER_WIN32_H_ | 159 #endif // COURGETTE_DISASSEMBLER_WIN32_H_ |
OLD | NEW |