| OLD | NEW |
| (Empty) |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #ifndef SANDBOX_SRC_SIDESTEP_RESOLVER_H__ | |
| 6 #define SANDBOX_SRC_SIDESTEP_RESOLVER_H__ | |
| 7 | |
| 8 #include "base/basictypes.h" | |
| 9 #include "sandbox/src/nt_internals.h" | |
| 10 #include "sandbox/src/resolver.h" | |
| 11 | |
| 12 namespace sandbox { | |
| 13 | |
| 14 // This is the concrete resolver used to perform sidestep interceptions. | |
| 15 class SidestepResolverThunk : public ResolverThunk { | |
| 16 public: | |
| 17 SidestepResolverThunk() {} | |
| 18 virtual ~SidestepResolverThunk() {} | |
| 19 | |
| 20 // Implementation of Resolver::Setup. | |
| 21 virtual NTSTATUS Setup(const void* target_module, | |
| 22 const void* interceptor_module, | |
| 23 const char* target_name, | |
| 24 const char* interceptor_name, | |
| 25 const void* interceptor_entry_point, | |
| 26 void* thunk_storage, | |
| 27 size_t storage_bytes, | |
| 28 size_t* storage_used); | |
| 29 | |
| 30 // Implementation of Resolver::GetThunkSize. | |
| 31 virtual size_t GetThunkSize() const; | |
| 32 | |
| 33 private: | |
| 34 DISALLOW_COPY_AND_ASSIGN(SidestepResolverThunk); | |
| 35 }; | |
| 36 | |
| 37 // This is the concrete resolver used to perform smart sidestep interceptions. | |
| 38 // This means basically a sidestep interception that skips the interceptor when | |
| 39 // the caller resides on the same dll being intercepted. It is intended as | |
| 40 // a helper only, because that determination is not infallible. | |
| 41 class SmartSidestepResolverThunk : public SidestepResolverThunk { | |
| 42 public: | |
| 43 SmartSidestepResolverThunk() {} | |
| 44 virtual ~SmartSidestepResolverThunk() {} | |
| 45 | |
| 46 // Implementation of Resolver::Setup. | |
| 47 virtual NTSTATUS Setup(const void* target_module, | |
| 48 const void* interceptor_module, | |
| 49 const char* target_name, | |
| 50 const char* interceptor_name, | |
| 51 const void* interceptor_entry_point, | |
| 52 void* thunk_storage, | |
| 53 size_t storage_bytes, | |
| 54 size_t* storage_used); | |
| 55 | |
| 56 // Implementation of Resolver::GetThunkSize. | |
| 57 virtual size_t GetThunkSize() const; | |
| 58 | |
| 59 private: | |
| 60 // Performs the actual call to the interceptor if the conditions are correct | |
| 61 // (as determined by IsInternalCall). | |
| 62 static void SmartStub(); | |
| 63 | |
| 64 // Returns true if return_address is inside the module loaded at base. | |
| 65 static bool IsInternalCall(const void* base, void* return_address); | |
| 66 | |
| 67 DISALLOW_COPY_AND_ASSIGN(SmartSidestepResolverThunk); | |
| 68 }; | |
| 69 | |
| 70 } // namespace sandbox | |
| 71 | |
| 72 | |
| 73 #endif // SANDBOX_SRC_SIDESTEP_RESOLVER_H__ | |
| OLD | NEW |