| 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 #ifndef COURGETTE_ASSEMBLY_PROGRAM_H_ | 5 #ifndef COURGETTE_ASSEMBLY_PROGRAM_H_ |
| 6 #define COURGETTE_ASSEMBLY_PROGRAM_H_ | 6 #define COURGETTE_ASSEMBLY_PROGRAM_H_ |
| 7 | 7 |
| 8 #include <stddef.h> |
| 9 #include <stdint.h> |
| 10 |
| 8 #include <map> | 11 #include <map> |
| 9 #include <set> | 12 #include <set> |
| 10 #include <vector> | 13 #include <vector> |
| 11 | 14 |
| 12 #include "base/basictypes.h" | 15 #include "base/macros.h" |
| 13 #include "base/memory/scoped_ptr.h" | 16 #include "base/memory/scoped_ptr.h" |
| 14 | |
| 15 #include "courgette/disassembler.h" | 17 #include "courgette/disassembler.h" |
| 16 #include "courgette/image_utils.h" | 18 #include "courgette/image_utils.h" |
| 17 #include "courgette/memory_allocator.h" | 19 #include "courgette/memory_allocator.h" |
| 18 | 20 |
| 19 namespace courgette { | 21 namespace courgette { |
| 20 | 22 |
| 21 class EncodedProgram; | 23 class EncodedProgram; |
| 22 | 24 |
| 23 typedef std::map<RVA, Label*> RVAToLabel; | 25 typedef std::map<RVA, Label*> RVAToLabel; |
| 24 | 26 |
| (...skipping 15 matching lines...) Expand all Loading... |
| 40 // Base class for instructions. Because we have so many instructions we want to | 42 // Base class for instructions. Because we have so many instructions we want to |
| 41 // keep them as small as possible. For this reason we avoid virtual functions. | 43 // keep them as small as possible. For this reason we avoid virtual functions. |
| 42 class Instruction { | 44 class Instruction { |
| 43 public: | 45 public: |
| 44 OP op() const { return static_cast<OP>(op_); } | 46 OP op() const { return static_cast<OP>(op_); } |
| 45 | 47 |
| 46 protected: | 48 protected: |
| 47 explicit Instruction(OP op) : op_(op), info_(0) {} | 49 explicit Instruction(OP op) : op_(op), info_(0) {} |
| 48 Instruction(OP op, unsigned int info) : op_(op), info_(info) {} | 50 Instruction(OP op, unsigned int info) : op_(op), info_(info) {} |
| 49 | 51 |
| 50 uint32 op_ : 4; // A few bits to store the OP code. | 52 uint32_t op_ : 4; // A few bits to store the OP code. |
| 51 uint32 info_ : 28; // Remaining bits in first word available to subclass. | 53 uint32_t info_ : 28; // Remaining bits in first word available to subclass. |
| 52 | 54 |
| 53 private: | 55 private: |
| 54 DISALLOW_COPY_AND_ASSIGN(Instruction); | 56 DISALLOW_COPY_AND_ASSIGN(Instruction); |
| 55 }; | 57 }; |
| 56 | 58 |
| 57 typedef NoThrowBuffer<Instruction*> InstructionVector; | 59 typedef NoThrowBuffer<Instruction*> InstructionVector; |
| 58 | 60 |
| 59 // An AssemblyProgram is the result of disassembling an executable file. | 61 // An AssemblyProgram is the result of disassembling an executable file. |
| 60 // | 62 // |
| 61 // * The disassembler creates labels in the AssemblyProgram and emits | 63 // * The disassembler creates labels in the AssemblyProgram and emits |
| (...skipping 13 matching lines...) Expand all Loading... |
| 75 // its own assignment, and then call AssignRemainingIndexes to ensure all | 77 // its own assignment, and then call AssignRemainingIndexes to ensure all |
| 76 // indexes are assigned. | 78 // indexes are assigned. |
| 77 // | 79 // |
| 78 class AssemblyProgram { | 80 class AssemblyProgram { |
| 79 public: | 81 public: |
| 80 explicit AssemblyProgram(ExecutableType kind); | 82 explicit AssemblyProgram(ExecutableType kind); |
| 81 ~AssemblyProgram(); | 83 ~AssemblyProgram(); |
| 82 | 84 |
| 83 ExecutableType kind() const { return kind_; } | 85 ExecutableType kind() const { return kind_; } |
| 84 | 86 |
| 85 void set_image_base(uint64 image_base) { image_base_ = image_base; } | 87 void set_image_base(uint64_t image_base) { image_base_ = image_base; } |
| 86 | 88 |
| 87 // Instructions will be assembled in the order they are emitted. | 89 // Instructions will be assembled in the order they are emitted. |
| 88 | 90 |
| 89 // Generates an entire base relocation table. | 91 // Generates an entire base relocation table. |
| 90 CheckBool EmitPeRelocsInstruction() WARN_UNUSED_RESULT; | 92 CheckBool EmitPeRelocsInstruction() WARN_UNUSED_RESULT; |
| 91 | 93 |
| 92 // Generates an ELF style relocation table for X86. | 94 // Generates an ELF style relocation table for X86. |
| 93 CheckBool EmitElfRelocationInstruction() WARN_UNUSED_RESULT; | 95 CheckBool EmitElfRelocationInstruction() WARN_UNUSED_RESULT; |
| 94 | 96 |
| 95 // Generates an ELF style relocation table for ARM. | 97 // Generates an ELF style relocation table for ARM. |
| 96 CheckBool EmitElfARMRelocationInstruction() WARN_UNUSED_RESULT; | 98 CheckBool EmitElfARMRelocationInstruction() WARN_UNUSED_RESULT; |
| 97 | 99 |
| 98 // Following instruction will be assembled at address 'rva'. | 100 // Following instruction will be assembled at address 'rva'. |
| 99 CheckBool EmitOriginInstruction(RVA rva) WARN_UNUSED_RESULT; | 101 CheckBool EmitOriginInstruction(RVA rva) WARN_UNUSED_RESULT; |
| 100 | 102 |
| 101 // Generates a single byte of data or machine instruction. | 103 // Generates a single byte of data or machine instruction. |
| 102 CheckBool EmitByteInstruction(uint8 byte) WARN_UNUSED_RESULT; | 104 CheckBool EmitByteInstruction(uint8_t byte) WARN_UNUSED_RESULT; |
| 103 | 105 |
| 104 // Generates multiple bytes of data or machine instructions. | 106 // Generates multiple bytes of data or machine instructions. |
| 105 CheckBool EmitBytesInstruction(const uint8* value, size_t len) | 107 CheckBool EmitBytesInstruction(const uint8_t* value, |
| 106 WARN_UNUSED_RESULT; | 108 size_t len) WARN_UNUSED_RESULT; |
| 107 | 109 |
| 108 // Generates 4-byte relative reference to address of 'label'. | 110 // Generates 4-byte relative reference to address of 'label'. |
| 109 CheckBool EmitRel32(Label* label) WARN_UNUSED_RESULT; | 111 CheckBool EmitRel32(Label* label) WARN_UNUSED_RESULT; |
| 110 | 112 |
| 111 // Generates 4-byte relative reference to address of 'label' for | 113 // Generates 4-byte relative reference to address of 'label' for |
| 112 // ARM. | 114 // ARM. |
| 113 CheckBool EmitRel32ARM(uint16 op, Label* label, const uint8* arm_op, | 115 CheckBool EmitRel32ARM(uint16_t op, |
| 114 uint16 op_size) WARN_UNUSED_RESULT; | 116 Label* label, |
| 117 const uint8_t* arm_op, |
| 118 uint16_t op_size) WARN_UNUSED_RESULT; |
| 115 | 119 |
| 116 // Generates 4-byte absolute reference to address of 'label'. | 120 // Generates 4-byte absolute reference to address of 'label'. |
| 117 CheckBool EmitAbs32(Label* label) WARN_UNUSED_RESULT; | 121 CheckBool EmitAbs32(Label* label) WARN_UNUSED_RESULT; |
| 118 | 122 |
| 119 // Generates 8-byte absolute reference to address of 'label'. | 123 // Generates 8-byte absolute reference to address of 'label'. |
| 120 CheckBool EmitAbs64(Label* label) WARN_UNUSED_RESULT; | 124 CheckBool EmitAbs64(Label* label) WARN_UNUSED_RESULT; |
| 121 | 125 |
| 122 // Looks up a label or creates a new one. Might return NULL. | 126 // Looks up a label or creates a new one. Might return NULL. |
| 123 Label* FindOrMakeAbs32Label(RVA rva); | 127 Label* FindOrMakeAbs32Label(RVA rva); |
| 124 | 128 |
| (...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 164 | 168 |
| 165 // Looks up a label or creates a new one. Might return NULL. | 169 // Looks up a label or creates a new one. Might return NULL. |
| 166 Label* FindLabel(RVA rva, RVAToLabel* labels); | 170 Label* FindLabel(RVA rva, RVAToLabel* labels); |
| 167 | 171 |
| 168 // Helper methods for the public versions. | 172 // Helper methods for the public versions. |
| 169 static void UnassignIndexes(RVAToLabel* labels); | 173 static void UnassignIndexes(RVAToLabel* labels); |
| 170 static void DefaultAssignIndexes(RVAToLabel* labels); | 174 static void DefaultAssignIndexes(RVAToLabel* labels); |
| 171 static void AssignRemainingIndexes(RVAToLabel* labels); | 175 static void AssignRemainingIndexes(RVAToLabel* labels); |
| 172 | 176 |
| 173 // Sharing instructions that emit a single byte saves a lot of space. | 177 // Sharing instructions that emit a single byte saves a lot of space. |
| 174 Instruction* GetByteInstruction(uint8 byte); | 178 Instruction* GetByteInstruction(uint8_t byte); |
| 175 scoped_ptr<Instruction* [], base::FreeDeleter> byte_instruction_cache_; | 179 scoped_ptr<Instruction* [], base::FreeDeleter> byte_instruction_cache_; |
| 176 | 180 |
| 177 uint64 image_base_; // Desired or mandated base address of image. | 181 uint64_t image_base_; // Desired or mandated base address of image. |
| 178 | 182 |
| 179 InstructionVector instructions_; // All the instructions in program. | 183 InstructionVector instructions_; // All the instructions in program. |
| 180 | 184 |
| 181 // These are lookup maps to find the label associated with a given address. | 185 // These are lookup maps to find the label associated with a given address. |
| 182 // We have separate label spaces for addresses referenced by rel32 labels and | 186 // We have separate label spaces for addresses referenced by rel32 labels and |
| 183 // abs32 labels. This is somewhat arbitrary. | 187 // abs32 labels. This is somewhat arbitrary. |
| 184 RVAToLabel rel32_labels_; | 188 RVAToLabel rel32_labels_; |
| 185 RVAToLabel abs32_labels_; | 189 RVAToLabel abs32_labels_; |
| 186 | 190 |
| 187 DISALLOW_COPY_AND_ASSIGN(AssemblyProgram); | 191 DISALLOW_COPY_AND_ASSIGN(AssemblyProgram); |
| 188 }; | 192 }; |
| 189 | 193 |
| 190 } // namespace courgette | 194 } // namespace courgette |
| 191 #endif // COURGETTE_ASSEMBLY_PROGRAM_H_ | 195 #endif // COURGETTE_ASSEMBLY_PROGRAM_H_ |
| OLD | NEW |