| OLD | NEW |
| 1 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2006-2008 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/src/sidestep_resolver.h" | 5 #include "sandbox/src/sidestep_resolver.h" |
| 6 | 6 |
| 7 #include "base/win/pe_image.h" | 7 #include "base/win/pe_image.h" |
| 8 #include "sandbox/src/sandbox_nt_util.h" | 8 #include "sandbox/src/sandbox_nt_util.h" |
| 9 #include "sandbox/src/sidestep/preamble_patcher.h" | 9 #include "sandbox/src/sidestep/preamble_patcher.h" |
| 10 | 10 |
| (...skipping 57 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 68 } | 68 } |
| 69 | 69 |
| 70 size_t SidestepResolverThunk::GetThunkSize() const { | 70 size_t SidestepResolverThunk::GetThunkSize() const { |
| 71 return GetInternalThunkSize() + kSizeOfSidestepStub; | 71 return GetInternalThunkSize() + kSizeOfSidestepStub; |
| 72 } | 72 } |
| 73 | 73 |
| 74 // This is basically a wrapper around the normal sidestep patch that extends | 74 // This is basically a wrapper around the normal sidestep patch that extends |
| 75 // the thunk to use a chained interceptor. It uses the fact that | 75 // the thunk to use a chained interceptor. It uses the fact that |
| 76 // SetInternalThunk generates the code to pass as the first parameter whatever | 76 // SetInternalThunk generates the code to pass as the first parameter whatever |
| 77 // it receives as original_function; we let SidestepResolverThunk set this value | 77 // it receives as original_function; we let SidestepResolverThunk set this value |
| 78 // to it's saved code, and then we change it to our thunk data. | 78 // to its saved code, and then we change it to our thunk data. |
| 79 NTSTATUS SmartSidestepResolverThunk::Setup(const void* target_module, | 79 NTSTATUS SmartSidestepResolverThunk::Setup(const void* target_module, |
| 80 const void* interceptor_module, | 80 const void* interceptor_module, |
| 81 const char* target_name, | 81 const char* target_name, |
| 82 const char* interceptor_name, | 82 const char* interceptor_name, |
| 83 const void* interceptor_entry_point, | 83 const void* interceptor_entry_point, |
| 84 void* thunk_storage, | 84 void* thunk_storage, |
| 85 size_t storage_bytes, | 85 size_t storage_bytes, |
| 86 size_t* storage_used) { | 86 size_t* storage_used) { |
| 87 if (storage_bytes < GetThunkSize()) | 87 if (storage_bytes < GetThunkSize()) |
| 88 return STATUS_BUFFER_TOO_SMALL; | 88 return STATUS_BUFFER_TOO_SMALL; |
| (...skipping 102 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 191 DCHECK_NT(base); | 191 DCHECK_NT(base); |
| 192 DCHECK_NT(return_address); | 192 DCHECK_NT(return_address); |
| 193 | 193 |
| 194 base::win::PEImage pe(base); | 194 base::win::PEImage pe(base); |
| 195 if (pe.GetImageSectionFromAddr(return_address)) | 195 if (pe.GetImageSectionFromAddr(return_address)) |
| 196 return true; | 196 return true; |
| 197 return false; | 197 return false; |
| 198 } | 198 } |
| 199 | 199 |
| 200 } // namespace sandbox | 200 } // namespace sandbox |
| OLD | NEW |