OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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_X64_H_ | 5 #ifndef COURGETTE_DISASSEMBLER_WIN32_X64_H_ |
6 #define COURGETTE_DISASSEMBLER_WIN32_X64_H_ | 6 #define COURGETTE_DISASSEMBLER_WIN32_X64_H_ |
7 | 7 |
8 #include "base/basictypes.h" | 8 #include <stddef.h> |
| 9 #include <stdint.h> |
| 10 |
| 11 #include "base/macros.h" |
9 #include "courgette/disassembler.h" | 12 #include "courgette/disassembler.h" |
10 #include "courgette/memory_allocator.h" | 13 #include "courgette/memory_allocator.h" |
11 #include "courgette/types_win_pe.h" | 14 #include "courgette/types_win_pe.h" |
12 | 15 |
13 #ifdef COURGETTE_HISTOGRAM_TARGETS | 16 #ifdef COURGETTE_HISTOGRAM_TARGETS |
14 #include <map> | 17 #include <map> |
15 #endif | 18 #endif |
16 | 19 |
17 namespace courgette { | 20 namespace courgette { |
18 | 21 |
(...skipping 10 matching lines...) Expand all Loading... |
29 // functions may be called. | 32 // functions may be called. |
30 virtual bool ParseHeader(); | 33 virtual bool ParseHeader(); |
31 | 34 |
32 virtual bool Disassemble(AssemblyProgram* target); | 35 virtual bool Disassemble(AssemblyProgram* target); |
33 | 36 |
34 // | 37 // |
35 // Exposed for test purposes | 38 // Exposed for test purposes |
36 // | 39 // |
37 | 40 |
38 bool has_text_section() const { return has_text_section_; } | 41 bool has_text_section() const { return has_text_section_; } |
39 uint32 size_of_code() const { return size_of_code_; } | 42 uint32_t size_of_code() const { return size_of_code_; } |
40 bool is_32bit() const { return !is_PE32_plus_; } | 43 bool is_32bit() const { return !is_PE32_plus_; } |
41 | 44 |
42 // Returns 'true' if the base relocation table can be parsed. | 45 // Returns 'true' if the base relocation table can be parsed. |
43 // Output is a vector of the RVAs corresponding to locations within executable | 46 // Output is a vector of the RVAs corresponding to locations within executable |
44 // that are listed in the base relocation table. | 47 // that are listed in the base relocation table. |
45 bool ParseRelocs(std::vector<RVA> *addresses); | 48 bool ParseRelocs(std::vector<RVA> *addresses); |
46 | 49 |
47 // Returns Section containing the relative virtual address, or NULL if none. | 50 // Returns Section containing the relative virtual address, or NULL if none. |
48 const Section* RVAToSection(RVA rva) const; | 51 const Section* RVAToSection(RVA rva) const; |
49 | 52 |
50 static const int kNoOffset = -1; | 53 static const int kNoOffset = -1; |
51 // Returns kNoOffset if there is no file offset corresponding to 'rva'. | 54 // Returns kNoOffset if there is no file offset corresponding to 'rva'. |
52 int RVAToFileOffset(RVA rva) const; | 55 int RVAToFileOffset(RVA rva) const; |
53 | 56 |
54 // Returns same as FileOffsetToPointer(RVAToFileOffset(rva)) except that NULL | 57 // Returns same as FileOffsetToPointer(RVAToFileOffset(rva)) except that NULL |
55 // is returned if there is no file offset corresponding to 'rva'. | 58 // is returned if there is no file offset corresponding to 'rva'. |
56 const uint8* RVAToPointer(RVA rva) const; | 59 const uint8_t* RVAToPointer(RVA rva) const; |
57 | 60 |
58 static std::string SectionName(const Section* section); | 61 static std::string SectionName(const Section* section); |
59 | 62 |
60 protected: | 63 protected: |
61 CheckBool ParseFile(AssemblyProgram* target) WARN_UNUSED_RESULT; | 64 CheckBool ParseFile(AssemblyProgram* target) WARN_UNUSED_RESULT; |
62 bool ParseAbs32Relocs(); | 65 bool ParseAbs32Relocs(); |
63 void ParseRel32RelocsFromSections(); | 66 void ParseRel32RelocsFromSections(); |
64 void ParseRel32RelocsFromSection(const Section* section); | 67 void ParseRel32RelocsFromSection(const Section* section); |
65 | 68 |
66 CheckBool ParseNonSectionFileRegion(uint32 start_file_offset, | 69 CheckBool ParseNonSectionFileRegion(uint32_t start_file_offset, |
67 uint32 end_file_offset, AssemblyProgram* program) WARN_UNUSED_RESULT; | 70 uint32_t end_file_offset, |
| 71 AssemblyProgram* program) |
| 72 WARN_UNUSED_RESULT; |
68 CheckBool ParseFileRegion(const Section* section, | 73 CheckBool ParseFileRegion(const Section* section, |
69 uint32 start_file_offset, uint32 end_file_offset, | 74 uint32_t start_file_offset, |
70 AssemblyProgram* program) WARN_UNUSED_RESULT; | 75 uint32_t end_file_offset, |
| 76 AssemblyProgram* program) WARN_UNUSED_RESULT; |
71 | 77 |
72 #if COURGETTE_HISTOGRAM_TARGETS | 78 #if COURGETTE_HISTOGRAM_TARGETS |
73 void HistogramTargets(const char* kind, const std::map<RVA, int>& map); | 79 void HistogramTargets(const char* kind, const std::map<RVA, int>& map); |
74 #endif | 80 #endif |
75 | 81 |
76 // Most addresses are represented as 32-bit RVAs. The one address we can't | 82 // Most addresses are represented as 32-bit RVAs. The one address we can't |
77 // do this with is the image base address. 'image_base' is valid only for | 83 // do this with is the image base address. 'image_base' is valid only for |
78 // 32-bit executables. 'image_base_64' is valid for 32- and 64-bit executable. | 84 // 32-bit executables. 'image_base_64' is valid for 32- and 64-bit executable. |
79 uint64 image_base() const { return image_base_; } | 85 uint64_t image_base() const { return image_base_; } |
80 | 86 |
81 const ImageDataDirectory& base_relocation_table() const { | 87 const ImageDataDirectory& base_relocation_table() const { |
82 return base_relocation_table_; | 88 return base_relocation_table_; |
83 } | 89 } |
84 | 90 |
85 bool IsValidRVA(RVA rva) const { return rva < size_of_image_; } | 91 bool IsValidRVA(RVA rva) const { return rva < size_of_image_; } |
86 | 92 |
87 // Returns description of the RVA, e.g. ".text+0x1243". For debugging only. | 93 // Returns description of the RVA, e.g. ".text+0x1243". For debugging only. |
88 std::string DescribeRVA(RVA rva) const; | 94 std::string DescribeRVA(RVA rva) const; |
89 | 95 |
90 // Finds the first section at file_offset or above. Does not return sections | 96 // Finds the first section at file_offset or above. Does not return sections |
91 // that have no raw bytes in the file. | 97 // that have no raw bytes in the file. |
92 const Section* FindNextSection(uint32 file_offset) const; | 98 const Section* FindNextSection(uint32_t file_offset) const; |
93 | 99 |
94 // There are 2 'coordinate systems' for reasoning about executables. | 100 // There are 2 'coordinate systems' for reasoning about executables. |
95 // FileOffset - the the offset within a single .EXE or .DLL *file*. | 101 // FileOffset - the the offset within a single .EXE or .DLL *file*. |
96 // RVA - relative virtual address (offset within *loaded image*) | 102 // RVA - relative virtual address (offset within *loaded image*) |
97 // FileOffsetToRVA and RVAToFileOffset convert between these representations. | 103 // FileOffsetToRVA and RVAToFileOffset convert between these representations. |
98 | 104 |
99 RVA FileOffsetToRVA(uint32 offset) const; | 105 RVA FileOffsetToRVA(uint32_t offset) const; |
100 | |
101 | 106 |
102 private: | 107 private: |
103 | 108 |
104 bool ReadDataDirectory(int index, ImageDataDirectory* dir); | 109 bool ReadDataDirectory(int index, ImageDataDirectory* dir); |
105 | 110 |
106 bool incomplete_disassembly_; // 'true' if can leave out 'uninteresting' bits | 111 bool incomplete_disassembly_; // 'true' if can leave out 'uninteresting' bits |
107 | 112 |
108 std::vector<RVA> abs32_locations_; | 113 std::vector<RVA> abs32_locations_; |
109 std::vector<RVA> rel32_locations_; | 114 std::vector<RVA> rel32_locations_; |
110 | 115 |
111 // | 116 // |
112 // Fields that are always valid. | 117 // Fields that are always valid. |
113 // | 118 // |
114 | 119 |
115 // | 120 // |
116 // Information that is valid after successful ParseHeader. | 121 // Information that is valid after successful ParseHeader. |
117 // | 122 // |
118 bool is_PE32_plus_; // PE32_plus is for 64 bit executables. | 123 bool is_PE32_plus_; // PE32_plus is for 64 bit executables. |
119 | 124 |
120 // Location and size of IMAGE_OPTIONAL_HEADER in the buffer. | 125 // Location and size of IMAGE_OPTIONAL_HEADER in the buffer. |
121 const uint8 *optional_header_; | 126 const uint8_t* optional_header_; |
122 uint16 size_of_optional_header_; | 127 uint16_t size_of_optional_header_; |
123 uint16 offset_of_data_directories_; | 128 uint16_t offset_of_data_directories_; |
124 | 129 |
125 uint16 machine_type_; | 130 uint16_t machine_type_; |
126 uint16 number_of_sections_; | 131 uint16_t number_of_sections_; |
127 const Section *sections_; | 132 const Section *sections_; |
128 bool has_text_section_; | 133 bool has_text_section_; |
129 | 134 |
130 uint32 size_of_code_; | 135 uint32_t size_of_code_; |
131 uint32 size_of_initialized_data_; | 136 uint32_t size_of_initialized_data_; |
132 uint32 size_of_uninitialized_data_; | 137 uint32_t size_of_uninitialized_data_; |
133 RVA base_of_code_; | 138 RVA base_of_code_; |
134 RVA base_of_data_; | 139 RVA base_of_data_; |
135 | 140 |
136 uint64 image_base_; | 141 uint64_t image_base_; |
137 uint32 size_of_image_; | 142 uint32_t size_of_image_; |
138 int number_of_data_directories_; | 143 int number_of_data_directories_; |
139 | 144 |
140 ImageDataDirectory export_table_; | 145 ImageDataDirectory export_table_; |
141 ImageDataDirectory import_table_; | 146 ImageDataDirectory import_table_; |
142 ImageDataDirectory resource_table_; | 147 ImageDataDirectory resource_table_; |
143 ImageDataDirectory exception_table_; | 148 ImageDataDirectory exception_table_; |
144 ImageDataDirectory base_relocation_table_; | 149 ImageDataDirectory base_relocation_table_; |
145 ImageDataDirectory bound_import_table_; | 150 ImageDataDirectory bound_import_table_; |
146 ImageDataDirectory import_address_table_; | 151 ImageDataDirectory import_address_table_; |
147 ImageDataDirectory delay_import_descriptor_; | 152 ImageDataDirectory delay_import_descriptor_; |
148 ImageDataDirectory clr_runtime_header_; | 153 ImageDataDirectory clr_runtime_header_; |
149 | 154 |
150 #if COURGETTE_HISTOGRAM_TARGETS | 155 #if COURGETTE_HISTOGRAM_TARGETS |
151 std::map<RVA, int> abs32_target_rvas_; | 156 std::map<RVA, int> abs32_target_rvas_; |
152 std::map<RVA, int> rel32_target_rvas_; | 157 std::map<RVA, int> rel32_target_rvas_; |
153 #endif | 158 #endif |
154 | 159 |
155 | 160 |
156 DISALLOW_COPY_AND_ASSIGN(DisassemblerWin32X64); | 161 DISALLOW_COPY_AND_ASSIGN(DisassemblerWin32X64); |
157 }; | 162 }; |
158 | 163 |
159 } // namespace courgette | 164 } // namespace courgette |
160 #endif // COURGETTE_DISASSEMBLER_WIN32_X64_H_ | 165 #endif // COURGETTE_DISASSEMBLER_WIN32_X64_H_ |
OLD | NEW |