OLD | NEW |
(Empty) | |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #ifndef CHROME_ELF_HOOK_UTIL_THUNK_GETTER_H_ |
| 6 #define CHROME_ELF_HOOK_UTIL_THUNK_GETTER_H_ |
| 7 |
| 8 #include <windows.h> |
| 9 |
| 10 #include "base/macros.h" |
| 11 |
| 12 namespace sandbox { |
| 13 class ServiceResolverThunk; |
| 14 } |
| 15 |
| 16 namespace elf_hook { |
| 17 |
| 18 //------------------------------------------------------------------------------ |
| 19 // System Service hooking support |
| 20 //------------------------------------------------------------------------------ |
| 21 |
| 22 // Creates a |ServiceResolverThunk| based on the OS version. Ownership of the |
| 23 // resulting thunk is passed to the caller. |
| 24 sandbox::ServiceResolverThunk* HookSystemService(bool relaxed); |
| 25 |
| 26 //------------------------------------------------------------------------------ |
| 27 // Import Address Table hooking support |
| 28 //------------------------------------------------------------------------------ |
| 29 class IATHook { |
| 30 public: |
| 31 IATHook(); |
| 32 ~IATHook(); |
| 33 |
| 34 // Intercept a function in an import table of a specific |
| 35 // module. Saves everything needed to Unhook. |
| 36 // |
| 37 // NOTE: Hook can only be called once at a time, to enable Unhook(). |
| 38 // |
| 39 // Arguments: |
| 40 // module Module to be intercepted |
| 41 // imported_from_module Module that exports the 'function_name' |
| 42 // function_name Name of the API to be intercepted |
| 43 // new_function New function pointer |
| 44 // |
| 45 // Returns: Windows error code (winerror.h). NO_ERROR if successful. |
| 46 DWORD Hook(HMODULE module, |
| 47 const char* imported_from_module, |
| 48 const char* function_name, |
| 49 void* new_function); |
| 50 |
| 51 // Unhook the IAT entry. |
| 52 // |
| 53 // Returns: Windows error code (winerror.h). NO_ERROR if successful. |
| 54 DWORD Unhook(); |
| 55 |
| 56 private: |
| 57 void* intercept_function_; |
| 58 void* original_function_; |
| 59 IMAGE_THUNK_DATA* iat_thunk_; |
| 60 |
| 61 DISALLOW_COPY_AND_ASSIGN(IATHook); |
| 62 }; |
| 63 |
| 64 } // namespace elf_hook |
| 65 |
| 66 #endif // CHROME_ELF_HOOK_UTIL_THUNK_GETTER_H_ |
OLD | NEW |