| Index: third_party/courgette/image_info.h | 
| =================================================================== | 
| --- third_party/courgette/image_info.h	(revision 0) | 
| +++ third_party/courgette/image_info.h	(revision 0) | 
| @@ -0,0 +1,200 @@ | 
| +// Copyright (c) 2009 The Chromium Authors. All rights reserved. | 
| +// Use of this source code is governed by a BSD-style license that can be | 
| +// found in the LICENSE file. | 
| + | 
| +#ifndef COURGETTE_IMAGE_INFO_H_ | 
| +#define COURGETTE_IMAGE_INFO_H_ | 
| + | 
| +#include <string> | 
| +#include <vector> | 
| + | 
| +#include "base/basictypes.h" | 
| + | 
| +namespace courgette { | 
| + | 
| +// A Relative Virtual Address is the address in the image file after it is | 
| +// loaded into memory relative to the image load address. | 
| +typedef uint32 RVA; | 
| + | 
| +// PE file section header.  This struct has the same layout as the | 
| +// IMAGE_SECTION_HEADER structure from WINNT.H | 
| +// http://msdn.microsoft.com/en-us/library/ms680341(VS.85).aspx | 
| +// | 
| +#pragma pack(push, 1)  // Supported by MSVC and GCC. Ensures no gaps in packing. | 
| +struct Section { | 
| +  char name[8]; | 
| +  uint32 virtual_size; | 
| +  uint32 virtual_address; | 
| +  uint32 size_of_raw_data; | 
| +  uint32 file_offset_of_raw_data; | 
| +  uint32 pointer_to_relocations;   // Always zero in an image. | 
| +  uint32 pointer_to_line_numbers;  // Always zero in an image. | 
| +  uint16 number_of_relocations;    // Always zero in an image. | 
| +  uint16 number_of_line_numbers;   // Always zero in an image. | 
| +  uint32 characteristics; | 
| +}; | 
| +#pragma pack(pop) | 
| + | 
| +COMPILE_ASSERT(sizeof(Section) == 40, section_is_40_bytes); | 
| + | 
| +// Returns the name of a section, solving the problem that the name is not | 
| +// always properly NUL-terminated.  Used only for debugging. | 
| +std::string SectionName(const Section* section); | 
| + | 
| +// ImageDataDirectory has same layout as IMAGE_DATA_DIRECTORY structure from | 
| +// WINNT.H | 
| +// http://msdn.microsoft.com/en-us/library/ms680305(VS.85).aspx | 
| +// | 
| +class ImageDataDirectory { | 
| + public: | 
| +  ImageDataDirectory() : address_(0), size_(0) {} | 
| +  RVA address_; | 
| +  uint32 size_; | 
| +}; | 
| + | 
| +COMPILE_ASSERT(sizeof(ImageDataDirectory) == 8, | 
| +               image_data_directory_is_8_bytes); | 
| + | 
| +// | 
| +//  PEInfo holds information about a single Windows 'Portable Executable' format | 
| +//  file in the on-disk format. | 
| +// | 
| +//  Imagine you had concatenated a bunch of 'original' files into one 'big' | 
| +//  file and read the big file into memory.  You could find the executables | 
| +//  from the original files by calling PEInfo::Init with different addresses. | 
| +//  If PEInfo::TryParseHeader returns true, then Init was passed the address | 
| +//  of the first byte of one of the original executables, and PEIinfo::length | 
| +//  will tell how long the file was. | 
| +// | 
| +class PEInfo { | 
| + public: | 
| +  PEInfo(); | 
| + | 
| +  // ok() may always be called but returns 'true' only after ParseHeader | 
| +  // succeeds. | 
| +  bool ok() const { return failure_reason_ == NULL; } | 
| + | 
| +  // Initialize with buffer.  This just sets up the region of memory that | 
| +  // potentially contains the bytes from an executable file.  The caller | 
| +  // continues to own 'start'. | 
| +  void Init(const void* start, size_t length); | 
| + | 
| +  // Returns 'true' if the buffer appears to point to a Windows 32 bit | 
| +  // executable, 'false' otherwise.  If ParseHeader() succeeds, other member | 
| +  // functions may be called. | 
| +  bool ParseHeader(); | 
| + | 
| +  // Returns 'true' if the base relocation table can be parsed. | 
| +  // Output is a vector of the RVAs corresponding to locations within executable | 
| +  // that are listed in the base relocation table. | 
| +  bool ParseRelocs(std::vector<RVA> *addresses); | 
| + | 
| +  // Returns the length of the image.  Valid only if ParseHeader succeeded. | 
| +  uint32 length() const { return file_length_; } | 
| + | 
| +  bool has_text_section() const { return has_text_section_; } | 
| + | 
| +  uint32 size_of_code() const { return size_of_code_; } | 
| + | 
| +  bool is_32bit() const { return !is_PE32_plus_; } | 
| + | 
| +  // Most addresses are represented as 32-bit RVAs.  The one address we can't | 
| +  // do this with is the image base address.  'image_base' is valid only for | 
| +  // 32-bit executables. 'image_base_64' is valid for 32- and 64-bit executable. | 
| +  uint32 image_base() const { return static_cast<uint32>(image_base_); } | 
| +  uint64 image_base_64() const { return image_base_; } | 
| + | 
| +  const ImageDataDirectory& base_relocation_table() const { | 
| +    return base_relocation_table_; | 
| +  } | 
| + | 
| +  bool IsValidRVA(RVA rva) const { return rva < size_of_image_; } | 
| + | 
| +  // Returns description of the RVA, e.g. ".text+0x1243".  For debugging only. | 
| +  std::string DescribeRVA(RVA rva) const; | 
| + | 
| +  // Returns a pointer into the memory copy of the file format. | 
| +  // FileOffsetToPointer(0) returns a pointer to the start of the file format. | 
| +  const uint8* FileOffsetToPointer(uint32 offset) const { | 
| +    return start_ + offset; | 
| +  } | 
| + | 
| +  // Finds the first section at file_offset or above. | 
| +  const Section* FindNextSection(uint32 file_offset) const; | 
| +  // Returns Section containing the relative virtual address, or NULL if none. | 
| +  const Section* RVAToSection(RVA rva) const; | 
| + | 
| +  // There are 2 'coordinate systems' for reasoning about executables. | 
| +  //   FileOffset - the the offset within a single .EXE or .DLL *file*. | 
| +  //   RVA - relative virtual address (offset within *loaded image*) | 
| +  // FileOffsetToRVA and RVAToFileOffset convert between these representations. | 
| + | 
| +  RVA FileOffsetToRVA(uint32 offset) const; | 
| + | 
| +  static const int kNoOffset = -1; | 
| +  // Returns kNoOffset if there is no file offset corresponding to 'rva'. | 
| +  int RVAToFileOffset(RVA rva) const; | 
| + | 
| +  // Returns same as FileOffsetToPointer(RVAToFileOffset(rva)) except that NULL | 
| +  // is returned if there is no file offset corresponding to 'rva'. | 
| +  const uint8* RVAToPointer(RVA rva) const; | 
| + | 
| + protected: | 
| +  // | 
| +  // Fields that are always valid. | 
| +  // | 
| +  const char* failure_reason_; | 
| + | 
| +  // | 
| +  // Basic information that is always valid after Init. | 
| +  // | 
| +  const uint8* start_;    // In current memory, base for 'file offsets'. | 
| +  const uint8* end_;      // In current memory. | 
| +  unsigned int length_;   // In current memory. | 
| + | 
| +  // | 
| +  // Information that is valid after successful ParseHeader. | 
| +  // | 
| +  bool is_PE32_plus_;   // PE32_plus is for 64 bit executables. | 
| +  uint32 file_length_; | 
| + | 
| +  // Location and size of IMAGE_OPTIONAL_HEADER in the buffer. | 
| +  const uint8 *optional_header_; | 
| +  uint16 size_of_optional_header_; | 
| +  uint16 offset_of_data_directories_; | 
| + | 
| +  uint16 machine_type_; | 
| +  uint16 number_of_sections_; | 
| +  const Section *sections_; | 
| +  bool has_text_section_; | 
| + | 
| +  uint32 size_of_code_; | 
| +  uint32 size_of_initialized_data_; | 
| +  uint32 size_of_uninitialized_data_; | 
| +  RVA base_of_code_; | 
| +  RVA base_of_data_; | 
| + | 
| +  uint64 image_base_;  // range limited to 32 bits for 32 bit executable | 
| +  uint32 size_of_image_; | 
| +  int number_of_data_directories_; | 
| + | 
| +  ImageDataDirectory export_table_; | 
| +  ImageDataDirectory import_table_; | 
| +  ImageDataDirectory resource_table_; | 
| +  ImageDataDirectory exception_table_; | 
| +  ImageDataDirectory base_relocation_table_; | 
| +  ImageDataDirectory bound_import_table_; | 
| +  ImageDataDirectory import_address_table_; | 
| +  ImageDataDirectory delay_import_descriptor_; | 
| +  ImageDataDirectory clr_runtime_header_; | 
| + | 
| + private: | 
| +  bool ReadDataDirectory(int index, ImageDataDirectory* dir); | 
| +  bool Bad(const char *reason); | 
| + | 
| +  DISALLOW_COPY_AND_ASSIGN(PEInfo); | 
| +}; | 
| + | 
| +}  // namespace | 
| +#endif  // COURGETTE_IMAGE_INFO_H_ | 
| + | 
|  |