Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(54)

Side by Side Diff: courgette/disassembler_win32.h

Issue 2462993003: [Courgette] Refactor: Add and use Instruction*Receptor classes; call ParseFile() in 2 passes. (Closed)
Patch Set: Fix comments. Created 4 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « courgette/disassembler_elf_32_x86.cc ('k') | courgette/disassembler_win32.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
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/memory_allocator.h" 18 #include "courgette/memory_allocator.h"
19 #include "courgette/types_win_pe.h" 19 #include "courgette/types_win_pe.h"
20 20
21 namespace courgette { 21 namespace courgette {
22 22
23 class AssemblyProgram; 23 class AssemblyProgram;
24 class InstructionReceptor;
24 25
25 class DisassemblerWin32 : public Disassembler { 26 class DisassemblerWin32 : public Disassembler {
26 public: 27 public:
27 virtual ~DisassemblerWin32() = default; 28 virtual ~DisassemblerWin32() = default;
28 29
29 // Disassembler interfaces. 30 // Disassembler interfaces.
30 RVA FileOffsetToRVA(FileOffset file_offset) const override; 31 RVA FileOffsetToRVA(FileOffset file_offset) const override;
31 FileOffset RVAToFileOffset(RVA rva) const override; 32 FileOffset RVAToFileOffset(RVA rva) const override;
32 virtual ExecutableType kind() const override = 0; 33 ExecutableType kind() const override = 0;
33 virtual RVA PointerToTargetRVA(const uint8_t* p) const override = 0; 34 RVA PointerToTargetRVA(const uint8_t* p) const override = 0;
34 bool ParseHeader() override; 35 bool ParseHeader() override;
35 bool Disassemble(AssemblyProgram* target) override; 36 bool Disassemble(AssemblyProgram* target) override;
36 37
37 // Exposed for test purposes 38 // Exposed for test purposes
38 bool has_text_section() const { return has_text_section_; } 39 bool has_text_section() const { return has_text_section_; }
39 uint32_t size_of_code() const { return size_of_code_; } 40 uint32_t size_of_code() const { return size_of_code_; }
40 41
41 // Returns 'true' if the base relocation table can be parsed. 42 // Returns 'true' if the base relocation table can be parsed.
42 // 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
43 // that are listed in the base relocation table. 44 // that are listed in the base relocation table.
(...skipping 10 matching lines...) Expand all
54 // which will be checked against the detected one. 55 // which will be checked against the detected one.
55 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);
56 57
57 // Disassembler interfaces. 58 // Disassembler interfaces.
58 RvaVisitor* CreateAbs32TargetRvaVisitor() override; 59 RvaVisitor* CreateAbs32TargetRvaVisitor() override;
59 RvaVisitor* CreateRel32TargetRvaVisitor() override; 60 RvaVisitor* CreateRel32TargetRvaVisitor() override;
60 void RemoveUnusedRel32Locations(AssemblyProgram* program) override; 61 void RemoveUnusedRel32Locations(AssemblyProgram* program) override;
61 62
62 DisassemblerWin32(const uint8_t* start, size_t length); 63 DisassemblerWin32(const uint8_t* start, size_t length);
63 64
64 CheckBool ParseFile(AssemblyProgram* target) WARN_UNUSED_RESULT; 65 CheckBool ParseFile(AssemblyProgram* target,
66 InstructionReceptor* receptor) const WARN_UNUSED_RESULT;
65 bool ParseAbs32Relocs(); 67 bool ParseAbs32Relocs();
66 void ParseRel32RelocsFromSections(); 68 void ParseRel32RelocsFromSections();
67 virtual void ParseRel32RelocsFromSection(const Section* section) = 0; 69 virtual void ParseRel32RelocsFromSection(const Section* section) = 0;
68 70
69 CheckBool ParseNonSectionFileRegion(FileOffset start_file_offset, 71 CheckBool ParseNonSectionFileRegion(FileOffset start_file_offset,
70 FileOffset end_file_offset, 72 FileOffset end_file_offset,
71 AssemblyProgram* program) 73 InstructionReceptor* receptor) const
72 WARN_UNUSED_RESULT; 74 WARN_UNUSED_RESULT;
73 CheckBool ParseFileRegion(const Section* section, 75 CheckBool ParseFileRegion(const Section* section,
74 FileOffset start_file_offset, 76 FileOffset start_file_offset,
75 FileOffset end_file_offset, 77 FileOffset end_file_offset,
76 AssemblyProgram* program) WARN_UNUSED_RESULT; 78 AssemblyProgram* program,
79 InstructionReceptor* receptor) const
80 WARN_UNUSED_RESULT;
77 81
78 // Returns address width in byte count. 82 // Returns address width in byte count.
79 virtual int AbsVAWidth() const = 0; 83 virtual int AbsVAWidth() const = 0;
80 // Emits Abs 32/64 |label| to the |program|. 84 // Emits Abs 32/64 |label| to the |receptor|.
81 virtual CheckBool EmitAbs(Label* label, AssemblyProgram* program) = 0; 85 virtual CheckBool EmitAbs(Label* label,
86 InstructionReceptor* receptor) const = 0;
82 // Returns true if type is recognized. 87 // Returns true if type is recognized.
83 virtual bool SupportsRelTableType(int type) const = 0; 88 virtual bool SupportsRelTableType(int type) const = 0;
84 virtual uint16_t OffsetOfDataDirectories() const = 0; 89 virtual uint16_t OffsetOfDataDirectories() const = 0;
85 90
86 #if COURGETTE_HISTOGRAM_TARGETS 91 #if COURGETTE_HISTOGRAM_TARGETS
87 void HistogramTargets(const char* kind, const std::map<RVA, int>& map); 92 void HistogramTargets(const char* kind, const std::map<RVA, int>& map) const;
88 #endif 93 #endif
89 94
90 // Most addresses are represented as 32-bit RVAs. The one address we can't 95 // Most addresses are represented as 32-bit RVAs. The one address we can't
91 // do this with is the image base address. 96 // do this with is the image base address.
92 uint64_t image_base() const { return image_base_; } 97 uint64_t image_base() const { return image_base_; }
93 98
94 const ImageDataDirectory& base_relocation_table() const { 99 const ImageDataDirectory& base_relocation_table() const {
95 return base_relocation_table_; 100 return base_relocation_table_;
96 } 101 }
97 102
(...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after
144 std::map<RVA, int> rel32_target_rvas_; 149 std::map<RVA, int> rel32_target_rvas_;
145 #endif 150 #endif
146 151
147 private: 152 private:
148 DISALLOW_COPY_AND_ASSIGN(DisassemblerWin32); 153 DISALLOW_COPY_AND_ASSIGN(DisassemblerWin32);
149 }; 154 };
150 155
151 } // namespace courgette 156 } // namespace courgette
152 157
153 #endif // COURGETTE_DISASSEMBLER_WIN32_H_ 158 #endif // COURGETTE_DISASSEMBLER_WIN32_H_
OLDNEW
« no previous file with comments | « courgette/disassembler_elf_32_x86.cc ('k') | courgette/disassembler_win32.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698