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 #include "courgette/disassembler_win32.h" | 5 #include "courgette/disassembler_win32.h" |
6 | 6 |
7 #include <stddef.h> | |
8 #include <stdint.h> | |
9 | |
10 #include <algorithm> | 7 #include <algorithm> |
11 | 8 |
12 #include "base/bind.h" | 9 #include "base/bind.h" |
13 #include "base/logging.h" | 10 #include "base/logging.h" |
14 #include "courgette/assembly_program.h" | 11 #include "courgette/assembly_program.h" |
15 #include "courgette/courgette.h" | 12 #include "courgette/courgette.h" |
16 | 13 |
17 #if COURGETTE_HISTOGRAM_TARGETS | 14 #if COURGETTE_HISTOGRAM_TARGETS |
18 #include <iostream> | 15 #include <iostream> |
19 #endif | 16 #endif |
(...skipping 197 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
217 // Pretend our in-memory copy is only as long as our detected length. | 214 // Pretend our in-memory copy is only as long as our detected length. |
218 ReduceLength(detected_length); | 215 ReduceLength(detected_length); |
219 | 216 |
220 if (!has_text_section()) { | 217 if (!has_text_section()) { |
221 return Bad("Resource-only executables are not yet supported"); | 218 return Bad("Resource-only executables are not yet supported"); |
222 } | 219 } |
223 | 220 |
224 return Good(); | 221 return Good(); |
225 } | 222 } |
226 | 223 |
227 bool DisassemblerWin32::Disassemble(AssemblyProgram* target) { | 224 bool DisassemblerWin32::Disassemble(AssemblyProgram* program) { |
228 if (!ok()) | 225 if (!ok()) |
229 return false; | 226 return false; |
230 | 227 |
231 target->set_image_base(image_base()); | |
232 | |
233 if (!ParseAbs32Relocs()) | 228 if (!ParseAbs32Relocs()) |
234 return false; | 229 return false; |
235 | 230 |
236 ParseRel32RelocsFromSections(); | 231 ParseRel32RelocsFromSections(); |
237 | 232 |
238 PrecomputeLabels(target); | 233 PrecomputeLabels(program); |
239 RemoveUnusedRel32Locations(target); | 234 RemoveUnusedRel32Locations(program); |
240 | 235 |
241 if (!target->GenerateInstructions( | 236 if (!program->GenerateInstructions( |
242 base::Bind(&DisassemblerWin32::ParseFile, base::Unretained(this)))) { | 237 base::Bind(&DisassemblerWin32::ParseFile, base::Unretained(this)))) { |
243 return false; | 238 return false; |
244 } | 239 } |
245 | 240 |
246 target->DefaultAssignIndexes(); | 241 program->DefaultAssignIndexes(); |
247 return true; | 242 return true; |
248 } | 243 } |
249 | 244 |
250 //////////////////////////////////////////////////////////////////////////////// | 245 //////////////////////////////////////////////////////////////////////////////// |
251 | 246 |
252 bool DisassemblerWin32::ParseRelocs(std::vector<RVA>* relocs) { | 247 bool DisassemblerWin32::ParseRelocs(std::vector<RVA>* relocs) { |
253 relocs->clear(); | 248 relocs->clear(); |
254 | 249 |
255 size_t relocs_size = base_relocation_table_.size_; | 250 size_t relocs_size = base_relocation_table_.size_; |
256 if (relocs_size == 0) | 251 if (relocs_size == 0) |
(...skipping 425 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
682 directory->size_ = static_cast<uint32_t>(size); | 677 directory->size_ = static_cast<uint32_t>(size); |
683 return true; | 678 return true; |
684 } else { | 679 } else { |
685 directory->address_ = 0; | 680 directory->address_ = 0; |
686 directory->size_ = 0; | 681 directory->size_ = 0; |
687 return true; | 682 return true; |
688 } | 683 } |
689 } | 684 } |
690 | 685 |
691 } // namespace courgette | 686 } // namespace courgette |
OLD | NEW |