| OLD | NEW |
| 1 // Copyright (c) 2006-2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2006-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 #include "sandbox/win/src/resolver.h" | 5 #include "sandbox/win/src/resolver.h" |
| 6 | 6 |
| 7 // For placement new. This file must not depend on the CRT at runtime, but |
| 8 // placement operator new is inline. |
| 9 #include <new> |
| 10 |
| 7 #include "sandbox/win/src/sandbox_nt_util.h" | 11 #include "sandbox/win/src/sandbox_nt_util.h" |
| 8 | 12 |
| 9 namespace { | 13 namespace { |
| 10 | 14 |
| 11 #pragma pack(push, 1) | 15 #pragma pack(push, 1) |
| 12 struct InternalThunk { | 16 struct InternalThunk { |
| 13 // This struct contains roughly the following code: | 17 // This struct contains roughly the following code: |
| 14 // sub esp, 8 // Create working space | 18 // sub esp, 8 // Create working space |
| 15 // push edx // Save register | 19 // push edx // Save register |
| 16 // mov edx, [esp + 0xc] // Get return adddress | 20 // mov edx, [esp + 0xc] // Get return adddress |
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 55 }; // namespace | 59 }; // namespace |
| 56 | 60 |
| 57 namespace sandbox { | 61 namespace sandbox { |
| 58 | 62 |
| 59 bool ResolverThunk::SetInternalThunk(void* storage, size_t storage_bytes, | 63 bool ResolverThunk::SetInternalThunk(void* storage, size_t storage_bytes, |
| 60 const void* original_function, | 64 const void* original_function, |
| 61 const void* interceptor) { | 65 const void* interceptor) { |
| 62 if (storage_bytes < sizeof(InternalThunk)) | 66 if (storage_bytes < sizeof(InternalThunk)) |
| 63 return false; | 67 return false; |
| 64 | 68 |
| 65 InternalThunk* thunk = new(storage, NT_PLACE) InternalThunk; | 69 InternalThunk* thunk = new(storage) InternalThunk; |
| 66 | 70 |
| 67 #pragma warning(push) | 71 #pragma warning(push) |
| 68 #pragma warning(disable: 4311) | 72 #pragma warning(disable: 4311) |
| 69 // These casts generate warnings because they are 32 bit specific. | 73 // These casts generate warnings because they are 32 bit specific. |
| 70 thunk->interceptor_function = reinterpret_cast<ULONG>(interceptor); | 74 thunk->interceptor_function = reinterpret_cast<ULONG>(interceptor); |
| 71 thunk->extra_argument = reinterpret_cast<ULONG>(original_function); | 75 thunk->extra_argument = reinterpret_cast<ULONG>(original_function); |
| 72 #pragma warning(pop) | 76 #pragma warning(pop) |
| 73 | 77 |
| 74 return true; | 78 return true; |
| 75 } | 79 } |
| 76 | 80 |
| 77 size_t ResolverThunk::GetInternalThunkSize() const { | 81 size_t ResolverThunk::GetInternalThunkSize() const { |
| 78 return sizeof(InternalThunk); | 82 return sizeof(InternalThunk); |
| 79 } | 83 } |
| 80 | 84 |
| 81 NTSTATUS ResolverThunk::ResolveTarget(const void* module, | 85 NTSTATUS ResolverThunk::ResolveTarget(const void* module, |
| 82 const char* function_name, | 86 const char* function_name, |
| 83 void** address) { | 87 void** address) { |
| 84 const void** casted = const_cast<const void**>(address); | 88 const void** casted = const_cast<const void**>(address); |
| 85 return ResolverThunk::ResolveInterceptor(module, function_name, casted); | 89 return ResolverThunk::ResolveInterceptor(module, function_name, casted); |
| 86 } | 90 } |
| 87 | 91 |
| 88 } // namespace sandbox | 92 } // namespace sandbox |
| OLD | NEW |