|
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 |
|
3 // found in the LICENSE file. |
|
4 |
|
5 #include "sandbox/win/src/service_resolver.h" |
|
6 |
|
7 #include "base/logging.h" |
|
8 #include "base/win/pe_image.h" |
|
9 |
|
10 namespace sandbox { |
|
11 |
|
12 NTSTATUS ServiceResolverThunk::ResolveInterceptor( |
|
13 const void* interceptor_module, |
|
14 const char* interceptor_name, |
|
15 const void** address) { |
|
16 // After all, we are using a locally mapped version of the exe, so the |
|
17 // action is the same as for a target function. |
|
18 return ResolveTarget(interceptor_module, interceptor_name, |
|
19 const_cast<void**>(address)); |
|
20 } |
|
21 |
|
22 // In this case all the work is done from the parent, so resolve is |
|
23 // just a simple GetProcAddress. |
|
24 NTSTATUS ServiceResolverThunk::ResolveTarget(const void* module, |
|
25 const char* function_name, |
|
26 void** address) { |
|
27 DCHECK(address); |
|
28 if (NULL == module) |
|
29 return STATUS_UNSUCCESSFUL; |
|
30 |
|
31 base::win::PEImage module_image(module); |
|
32 *address = module_image.GetProcAddress(function_name); |
|
33 |
|
34 if (NULL == *address) { |
|
35 NOTREACHED(); |
|
36 return STATUS_UNSUCCESSFUL; |
|
37 } |
|
38 |
|
39 return STATUS_SUCCESS; |
|
40 } |
|
41 |
|
42 } // namespace sandbox |