| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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 // This file was adapted from GreenBorder's Code. | 5 // This file was adapted from GreenBorder's Code. |
| 6 // To understand what this class is about (for other than well known functions | 6 // To understand what this class is about (for other than well known functions |
| 7 // as GetProcAddress), a good starting point is "An In-Depth Look into the | 7 // as GetProcAddress), a good starting point is "An In-Depth Look into the |
| 8 // Win32 Portable Executable File Format" by Matt Pietrek: | 8 // Win32 Portable Executable File Format" by Matt Pietrek: |
| 9 // http://msdn.microsoft.com/msdnmag/issues/02/02/PE/default.aspx | 9 // http://msdn.microsoft.com/msdnmag/issues/02/02/PE/default.aspx |
| 10 | 10 |
| 11 #ifndef BASE_WIN_PE_IMAGE_H_ | 11 #ifndef BASE_WIN_PE_IMAGE_H_ |
| 12 #define BASE_WIN_PE_IMAGE_H_ | 12 #define BASE_WIN_PE_IMAGE_H_ |
| 13 | 13 |
| 14 #include <windows.h> | 14 #include <windows.h> |
| 15 | 15 |
| 16 #if defined(_WIN32_WINNT_WIN8) | 16 #if defined(_WIN32_WINNT_WIN8) |
| 17 // The Windows 8 SDK defines FACILITY_VISUALCPP in winerror.h. | 17 // The Windows 8 SDK defines FACILITY_VISUALCPP in winerror.h. |
| 18 #undef FACILITY_VISUALCPP | 18 #undef FACILITY_VISUALCPP |
| 19 #endif | 19 #endif |
| 20 #include <DelayIMP.h> | 20 #include <DelayIMP.h> |
| 21 | 21 |
| 22 namespace base { | 22 namespace base { |
| 23 namespace win { | 23 namespace win { |
| 24 | 24 |
| 25 // This class is a wrapper for the Portable Executable File Format (PE). | 25 // This class is a wrapper for the Portable Executable File Format (PE). |
| 26 // It's main purpose is to provide an easy way to work with imports and exports | 26 // Its main purpose is to provide an easy way to work with imports and exports |
| 27 // from a file, mapped in memory as image. | 27 // from a file, mapped in memory as image. |
| 28 class PEImage { | 28 class PEImage { |
| 29 public: | 29 public: |
| 30 // Callback to enumerate sections. | 30 // Callback to enumerate sections. |
| 31 // cookie is the value passed to the enumerate method. | 31 // cookie is the value passed to the enumerate method. |
| 32 // Returns true to continue the enumeration. | 32 // Returns true to continue the enumeration. |
| 33 typedef bool (*EnumSectionsFunction)(const PEImage &image, | 33 typedef bool (*EnumSectionsFunction)(const PEImage &image, |
| 34 PIMAGE_SECTION_HEADER header, | 34 PIMAGE_SECTION_HEADER header, |
| 35 PVOID section_start, DWORD section_size, | 35 PVOID section_start, DWORD section_size, |
| 36 PVOID cookie); | 36 PVOID cookie); |
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 77 // cookie is the value passed to the enumerate method. | 77 // cookie is the value passed to the enumerate method. |
| 78 // Returns true to continue the enumeration. | 78 // Returns true to continue the enumeration. |
| 79 typedef bool (*EnumRelocsFunction)(const PEImage &image, WORD type, | 79 typedef bool (*EnumRelocsFunction)(const PEImage &image, WORD type, |
| 80 PVOID address, PVOID cookie); | 80 PVOID address, PVOID cookie); |
| 81 | 81 |
| 82 explicit PEImage(HMODULE module) : module_(module) {} | 82 explicit PEImage(HMODULE module) : module_(module) {} |
| 83 explicit PEImage(const void* module) { | 83 explicit PEImage(const void* module) { |
| 84 module_ = reinterpret_cast<HMODULE>(const_cast<void*>(module)); | 84 module_ = reinterpret_cast<HMODULE>(const_cast<void*>(module)); |
| 85 } | 85 } |
| 86 | 86 |
| 87 virtual ~PEImage() {} |
| 88 |
| 87 // Gets the HMODULE for this object. | 89 // Gets the HMODULE for this object. |
| 88 HMODULE module() const; | 90 HMODULE module() const; |
| 89 | 91 |
| 90 // Sets this object's HMODULE. | 92 // Sets this object's HMODULE. |
| 91 void set_module(HMODULE module); | 93 void set_module(HMODULE module); |
| 92 | 94 |
| 93 // Checks if this symbol is actually an ordinal. | 95 // Checks if this symbol is actually an ordinal. |
| 94 static bool IsOrdinal(LPCSTR name); | 96 static bool IsOrdinal(LPCSTR name); |
| 95 | 97 |
| 96 // Converts a named symbol to the corresponding ordinal. | 98 // Converts a named symbol to the corresponding ordinal. |
| (...skipping 162 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 259 | 261 |
| 260 inline PIMAGE_EXPORT_DIRECTORY PEImage::GetExportDirectory() const { | 262 inline PIMAGE_EXPORT_DIRECTORY PEImage::GetExportDirectory() const { |
| 261 return reinterpret_cast<PIMAGE_EXPORT_DIRECTORY>( | 263 return reinterpret_cast<PIMAGE_EXPORT_DIRECTORY>( |
| 262 GetImageDirectoryEntryAddr(IMAGE_DIRECTORY_ENTRY_EXPORT)); | 264 GetImageDirectoryEntryAddr(IMAGE_DIRECTORY_ENTRY_EXPORT)); |
| 263 } | 265 } |
| 264 | 266 |
| 265 } // namespace win | 267 } // namespace win |
| 266 } // namespace base | 268 } // namespace base |
| 267 | 269 |
| 268 #endif // BASE_WIN_PE_IMAGE_H_ | 270 #endif // BASE_WIN_PE_IMAGE_H_ |
| OLD | NEW |