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 |