| Index: courgette/disassembler_win32_x86.cc
|
| diff --git a/courgette/disassembler_win32_x86.cc b/courgette/disassembler_win32_x86.cc
|
| index 85d99f485d672ffdd8516988ff2167cec064eff9..bc41ff0ccc24799d1afe70cf4072595b920eabfb 100644
|
| --- a/courgette/disassembler_win32_x86.cc
|
| +++ b/courgette/disassembler_win32_x86.cc
|
| @@ -4,11 +4,13 @@
|
|
|
| #include "courgette/disassembler_win32_x86.h"
|
|
|
| +#include <stddef.h>
|
| +#include <stdint.h>
|
| +
|
| #include <algorithm>
|
| #include <string>
|
| #include <vector>
|
|
|
| -#include "base/basictypes.h"
|
| #include "base/logging.h"
|
|
|
| #include "courgette/assembly_program.h"
|
| @@ -55,13 +57,12 @@ bool DisassemblerWin32X86::ParseHeader() {
|
| return Bad("Not MZ");
|
|
|
| // offset from DOS header to PE header is stored in DOS header.
|
| - uint32 offset = ReadU32(start(),
|
| - kOffsetOfFileAddressOfNewExeHeader);
|
| + uint32_t offset = ReadU32(start(), kOffsetOfFileAddressOfNewExeHeader);
|
|
|
| if (offset >= length())
|
| return Bad("Bad offset to PE header");
|
|
|
| - const uint8* const pe_header = OffsetToPointer(offset);
|
| + const uint8_t* const pe_header = OffsetToPointer(offset);
|
| const size_t kMinPEHeaderSize = 4 /*signature*/ + kSizeOfCoffHeader;
|
| if (pe_header <= start() ||
|
| pe_header >= end() - kMinPEHeaderSize)
|
| @@ -83,13 +84,13 @@ bool DisassemblerWin32X86::ParseHeader() {
|
| // The second field of the IMAGE_NT_HEADERS is the COFF header.
|
| // The COFF header is also called an IMAGE_FILE_HEADER
|
| // http://msdn.microsoft.com/en-us/library/ms680313(VS.85).aspx
|
| - const uint8* const coff_header = pe_header + 4;
|
| + const uint8_t* const coff_header = pe_header + 4;
|
| machine_type_ = ReadU16(coff_header, 0);
|
| number_of_sections_ = ReadU16(coff_header, 2);
|
| size_of_optional_header_ = ReadU16(coff_header, 16);
|
|
|
| // The rest of the IMAGE_NT_HEADERS is the IMAGE_OPTIONAL_HEADER(32|64)
|
| - const uint8* const optional_header = coff_header + kSizeOfCoffHeader;
|
| + const uint8_t* const optional_header = coff_header + kSizeOfCoffHeader;
|
| optional_header_ = optional_header;
|
|
|
| if (optional_header + size_of_optional_header_ >= end())
|
| @@ -99,7 +100,7 @@ bool DisassemblerWin32X86::ParseHeader() {
|
| if (size_of_optional_header_ < 2)
|
| return Bad("optional header no magic");
|
|
|
| - uint16 magic = ReadU16(optional_header, 0);
|
| + uint16_t magic = ReadU16(optional_header, 0);
|
|
|
| if (magic == kImageNtOptionalHdr32Magic) {
|
| is_PE32_plus_ = false;
|
| @@ -176,7 +177,7 @@ bool DisassemblerWin32X86::ParseHeader() {
|
| if (memcmp(section->name, ".text", 6) == 0)
|
| has_text_section_ = true;
|
|
|
| - uint32 section_end =
|
| + uint32_t section_end =
|
| section->file_offset_of_raw_data + section->size_of_raw_data;
|
| if (section_end > detected_length)
|
| detected_length = section_end;
|
| @@ -229,8 +230,8 @@ bool DisassemblerWin32X86::ParseRelocs(std::vector<RVA> *relocs) {
|
| // "The format of the base relocation data is somewhat quirky"
|
| // at http://msdn.microsoft.com/en-us/library/ms809762.aspx
|
|
|
| - const uint8* relocs_start = RVAToPointer(base_relocation_table_.address_);
|
| - const uint8* relocs_end = relocs_start + relocs_size;
|
| + const uint8_t* relocs_start = RVAToPointer(base_relocation_table_.address_);
|
| + const uint8_t* relocs_end = relocs_start + relocs_size;
|
|
|
| // Make sure entire base relocation table is within the buffer.
|
| if (relocs_start < start() ||
|
| @@ -240,17 +241,17 @@ bool DisassemblerWin32X86::ParseRelocs(std::vector<RVA> *relocs) {
|
| return Bad(".relocs outside image");
|
| }
|
|
|
| - const uint8* block = relocs_start;
|
| + const uint8_t* block = relocs_start;
|
|
|
| // Walk the variable sized blocks.
|
| while (block + 8 < relocs_end) {
|
| RVA page_rva = ReadU32(block, 0);
|
| - uint32 size = ReadU32(block, 4);
|
| + uint32_t size = ReadU32(block, 4);
|
| if (size < 8 || // Size includes header ...
|
| size % 4 != 0) // ... and is word aligned.
|
| return Bad("unreasonable relocs block");
|
|
|
| - const uint8* end_entries = block + size;
|
| + const uint8_t* end_entries = block + size;
|
|
|
| if (end_entries <= block ||
|
| end_entries <= start() ||
|
| @@ -258,8 +259,8 @@ bool DisassemblerWin32X86::ParseRelocs(std::vector<RVA> *relocs) {
|
| return Bad(".relocs block outside image");
|
|
|
| // Walk through the two-byte entries.
|
| - for (const uint8* p = block + 8; p < end_entries; p += 2) {
|
| - uint16 entry = ReadU16(p, 0);
|
| + for (const uint8_t* p = block + 8; p < end_entries; p += 2) {
|
| + uint16_t entry = ReadU16(p, 0);
|
| int type = entry >> 12;
|
| int offset = entry & 0xFFF;
|
|
|
| @@ -267,7 +268,7 @@ bool DisassemblerWin32X86::ParseRelocs(std::vector<RVA> *relocs) {
|
| // Skip the relocs that live outside of the image. It might be the case
|
| // if a reloc is relative to a register, e.g.:
|
| // mov ecx,dword ptr [eax+044D5888h]
|
| - uint32 target_address = Read32LittleEndian(RVAToPointer(rva));
|
| + uint32_t target_address = Read32LittleEndian(RVAToPointer(rva));
|
| if (target_address < image_base_ ||
|
| target_address > (image_base_ + size_of_image_)) {
|
| continue;
|
| @@ -293,7 +294,7 @@ bool DisassemblerWin32X86::ParseRelocs(std::vector<RVA> *relocs) {
|
| const Section* DisassemblerWin32X86::RVAToSection(RVA rva) const {
|
| for (int i = 0; i < number_of_sections_; i++) {
|
| const Section* section = §ions_[i];
|
| - uint32 offset = rva - section->virtual_address;
|
| + uint32_t offset = rva - section->virtual_address;
|
| if (offset < section->virtual_size) {
|
| return section;
|
| }
|
| @@ -304,7 +305,7 @@ const Section* DisassemblerWin32X86::RVAToSection(RVA rva) const {
|
| int DisassemblerWin32X86::RVAToFileOffset(RVA rva) const {
|
| const Section* section = RVAToSection(rva);
|
| if (section) {
|
| - uint32 offset = rva - section->virtual_address;
|
| + uint32_t offset = rva - section->virtual_address;
|
| if (offset < section->size_of_raw_data) {
|
| return section->file_offset_of_raw_data + offset;
|
| } else {
|
| @@ -323,7 +324,7 @@ int DisassemblerWin32X86::RVAToFileOffset(RVA rva) const {
|
| return kNoOffset;
|
| }
|
|
|
| -const uint8* DisassemblerWin32X86::RVAToPointer(RVA rva) const {
|
| +const uint8_t* DisassemblerWin32X86::RVAToPointer(RVA rva) const {
|
| int file_offset = RVAToFileOffset(rva);
|
| if (file_offset == kNoOffset)
|
| return NULL;
|
| @@ -342,7 +343,7 @@ std::string DisassemblerWin32X86::SectionName(const Section* section) {
|
|
|
| CheckBool DisassemblerWin32X86::ParseFile(AssemblyProgram* program) {
|
| // Walk all the bytes in the file, whether or not in a section.
|
| - uint32 file_offset = 0;
|
| + uint32_t file_offset = 0;
|
| while (file_offset < length()) {
|
| const Section* section = FindNextSection(file_offset);
|
| if (section == NULL) {
|
| @@ -352,14 +353,14 @@ CheckBool DisassemblerWin32X86::ParseFile(AssemblyProgram* program) {
|
| break;
|
| }
|
| if (file_offset < section->file_offset_of_raw_data) {
|
| - uint32 section_start_offset = section->file_offset_of_raw_data;
|
| + uint32_t section_start_offset = section->file_offset_of_raw_data;
|
| if(!ParseNonSectionFileRegion(file_offset, section_start_offset,
|
| program))
|
| return false;
|
|
|
| file_offset = section_start_offset;
|
| }
|
| - uint32 end = file_offset + section->size_of_raw_data;
|
| + uint32_t end = file_offset + section->size_of_raw_data;
|
| if (!ParseFileRegion(section, file_offset, end, program))
|
| return false;
|
| file_offset = end;
|
| @@ -382,7 +383,7 @@ bool DisassemblerWin32X86::ParseAbs32Relocs() {
|
| for (size_t i = 0; i < abs32_locations_.size(); ++i) {
|
| RVA rva = abs32_locations_[i];
|
| // The 4 bytes at the relocation are a reference to some address.
|
| - uint32 target_address = Read32LittleEndian(RVAToPointer(rva));
|
| + uint32_t target_address = Read32LittleEndian(RVAToPointer(rva));
|
| ++abs32_target_rvas_[target_address - image_base()];
|
| }
|
| #endif
|
| @@ -390,7 +391,7 @@ bool DisassemblerWin32X86::ParseAbs32Relocs() {
|
| }
|
|
|
| void DisassemblerWin32X86::ParseRel32RelocsFromSections() {
|
| - uint32 file_offset = 0;
|
| + uint32_t file_offset = 0;
|
| while (file_offset < length()) {
|
| const Section* section = FindNextSection(file_offset);
|
| if (section == NULL)
|
| @@ -433,11 +434,11 @@ void DisassemblerWin32X86::ParseRel32RelocsFromSection(const Section* section) {
|
| if (!isCode)
|
| return;
|
|
|
| - uint32 start_file_offset = section->file_offset_of_raw_data;
|
| - uint32 end_file_offset = start_file_offset + section->size_of_raw_data;
|
| + uint32_t start_file_offset = section->file_offset_of_raw_data;
|
| + uint32_t end_file_offset = start_file_offset + section->size_of_raw_data;
|
|
|
| - const uint8* start_pointer = OffsetToPointer(start_file_offset);
|
| - const uint8* end_pointer = OffsetToPointer(end_file_offset);
|
| + const uint8_t* start_pointer = OffsetToPointer(start_file_offset);
|
| + const uint8_t* end_pointer = OffsetToPointer(end_file_offset);
|
|
|
| RVA start_rva = FileOffsetToRVA(start_file_offset);
|
| RVA end_rva = start_rva + section->virtual_size;
|
| @@ -456,8 +457,8 @@ void DisassemblerWin32X86::ParseRel32RelocsFromSection(const Section* section) {
|
| }
|
|
|
| CheckBool DisassemblerWin32X86::ParseNonSectionFileRegion(
|
| - uint32 start_file_offset,
|
| - uint32 end_file_offset,
|
| + uint32_t start_file_offset,
|
| + uint32_t end_file_offset,
|
| AssemblyProgram* program) {
|
| if (incomplete_disassembly_)
|
| return true;
|
| @@ -472,21 +473,21 @@ CheckBool DisassemblerWin32X86::ParseNonSectionFileRegion(
|
| return true;
|
| }
|
|
|
| -CheckBool DisassemblerWin32X86::ParseFileRegion(
|
| - const Section* section,
|
| - uint32 start_file_offset, uint32 end_file_offset,
|
| - AssemblyProgram* program) {
|
| +CheckBool DisassemblerWin32X86::ParseFileRegion(const Section* section,
|
| + uint32_t start_file_offset,
|
| + uint32_t end_file_offset,
|
| + AssemblyProgram* program) {
|
| RVA relocs_start_rva = base_relocation_table().address_;
|
|
|
| - const uint8* start_pointer = OffsetToPointer(start_file_offset);
|
| - const uint8* end_pointer = OffsetToPointer(end_file_offset);
|
| + const uint8_t* start_pointer = OffsetToPointer(start_file_offset);
|
| + const uint8_t* end_pointer = OffsetToPointer(end_file_offset);
|
|
|
| RVA start_rva = FileOffsetToRVA(start_file_offset);
|
| RVA end_rva = start_rva + section->virtual_size;
|
|
|
| // Quick way to convert from Pointer to RVA within a single Section is to
|
| // subtract 'pointer_to_rva'.
|
| - const uint8* const adjust_pointer_to_rva = start_pointer - start_rva;
|
| + const uint8_t* const adjust_pointer_to_rva = start_pointer - start_rva;
|
|
|
| std::vector<RVA>::iterator rel32_pos = rel32_locations_.begin();
|
| std::vector<RVA>::iterator abs32_pos = abs32_locations_.begin();
|
| @@ -494,7 +495,7 @@ CheckBool DisassemblerWin32X86::ParseFileRegion(
|
| if (!program->EmitOriginInstruction(start_rva))
|
| return false;
|
|
|
| - const uint8* p = start_pointer;
|
| + const uint8_t* p = start_pointer;
|
|
|
| while (p < end_pointer) {
|
| RVA current_rva = static_cast<RVA>(p - adjust_pointer_to_rva);
|
| @@ -505,7 +506,7 @@ CheckBool DisassemblerWin32X86::ParseFileRegion(
|
| if (current_rva == relocs_start_rva) {
|
| if (!program->EmitPeRelocsInstruction())
|
| return false;
|
| - uint32 relocs_size = base_relocation_table().size_;
|
| + uint32_t relocs_size = base_relocation_table().size_;
|
| if (relocs_size) {
|
| p += relocs_size;
|
| continue;
|
| @@ -516,7 +517,7 @@ CheckBool DisassemblerWin32X86::ParseFileRegion(
|
| ++abs32_pos;
|
|
|
| if (abs32_pos != abs32_locations_.end() && *abs32_pos == current_rva) {
|
| - uint32 target_address = Read32LittleEndian(p);
|
| + uint32_t target_address = Read32LittleEndian(p);
|
| RVA target_rva = target_address - image_base();
|
| // TODO(sra): target could be Label+offset. It is not clear how to guess
|
| // which it might be. We assume offset==0.
|
| @@ -618,7 +619,8 @@ std::string DisassemblerWin32X86::DescribeRVA(RVA rva) const {
|
| return s.str();
|
| }
|
|
|
| -const Section* DisassemblerWin32X86::FindNextSection(uint32 fileOffset) const {
|
| +const Section* DisassemblerWin32X86::FindNextSection(
|
| + uint32_t fileOffset) const {
|
| const Section* best = 0;
|
| for (int i = 0; i < number_of_sections_; i++) {
|
| const Section* section = §ions_[i];
|
| @@ -634,10 +636,10 @@ const Section* DisassemblerWin32X86::FindNextSection(uint32 fileOffset) const {
|
| return best;
|
| }
|
|
|
| -RVA DisassemblerWin32X86::FileOffsetToRVA(uint32 file_offset) const {
|
| +RVA DisassemblerWin32X86::FileOffsetToRVA(uint32_t file_offset) const {
|
| for (int i = 0; i < number_of_sections_; i++) {
|
| const Section* section = §ions_[i];
|
| - uint32 offset = file_offset - section->file_offset_of_raw_data;
|
| + uint32_t offset = file_offset - section->file_offset_of_raw_data;
|
| if (offset < section->size_of_raw_data) {
|
| return section->virtual_address + offset;
|
| }
|
| @@ -653,7 +655,7 @@ bool DisassemblerWin32X86::ReadDataDirectory(
|
| size_t offset = index * 8 + offset_of_data_directories_;
|
| if (offset >= size_of_optional_header_)
|
| return Bad("number of data directories inconsistent");
|
| - const uint8* data_directory = optional_header_ + offset;
|
| + const uint8_t* data_directory = optional_header_ + offset;
|
| if (data_directory < start() ||
|
| data_directory + 8 >= end())
|
| return Bad("data directory outside image");
|
| @@ -664,7 +666,7 @@ bool DisassemblerWin32X86::ReadDataDirectory(
|
|
|
| // TODO(sra): validate RVA.
|
| directory->address_ = rva;
|
| - directory->size_ = static_cast<uint32>(size);
|
| + directory->size_ = static_cast<uint32_t>(size);
|
| return true;
|
| } else {
|
| directory->address_ = 0;
|
|
|