OLD | NEW |
| (Empty) |
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 | |
3 // found in the LICENSE file. | |
4 | |
5 #include "sandbox/src/policy_target.h" | |
6 | |
7 #include "sandbox/src/crosscall_client.h" | |
8 #include "sandbox/src/ipc_tags.h" | |
9 #include "sandbox/src/policy_engine_processor.h" | |
10 #include "sandbox/src/policy_low_level.h" | |
11 #include "sandbox/src/policy_params.h" | |
12 #include "sandbox/src/sandbox_factory.h" | |
13 #include "sandbox/src/sandbox_nt_util.h" | |
14 #include "sandbox/src/sharedmem_ipc_client.h" | |
15 #include "sandbox/src/target_services.h" | |
16 | |
17 namespace sandbox { | |
18 | |
19 // Handle for our private heap. | |
20 extern void* g_heap; | |
21 | |
22 // This is the list of all imported symbols from ntdll.dll. | |
23 SANDBOX_INTERCEPT NtExports g_nt; | |
24 | |
25 // Policy data. | |
26 extern void* volatile g_shared_policy_memory; | |
27 SANDBOX_INTERCEPT size_t g_shared_policy_size; | |
28 | |
29 bool QueryBroker(int ipc_id, CountedParameterSetBase* params) { | |
30 DCHECK_NT(static_cast<size_t>(ipc_id) < kMaxServiceCount); | |
31 DCHECK_NT(g_shared_policy_memory); | |
32 DCHECK_NT(g_shared_policy_size > 0); | |
33 | |
34 if (static_cast<size_t>(ipc_id) >= kMaxServiceCount) | |
35 return false; | |
36 | |
37 PolicyGlobal* global_policy = | |
38 reinterpret_cast<PolicyGlobal*>(g_shared_policy_memory); | |
39 | |
40 if (!global_policy->entry[ipc_id]) | |
41 return false; | |
42 | |
43 PolicyBuffer* policy = reinterpret_cast<PolicyBuffer*>( | |
44 reinterpret_cast<char*>(g_shared_policy_memory) + | |
45 reinterpret_cast<size_t>(global_policy->entry[ipc_id])); | |
46 | |
47 if ((reinterpret_cast<size_t>(global_policy->entry[ipc_id]) > | |
48 global_policy->data_size) || | |
49 (g_shared_policy_size < global_policy->data_size)) { | |
50 NOTREACHED_NT(); | |
51 return false; | |
52 } | |
53 | |
54 for (int i = 0; i < params->count; i++) { | |
55 if (!params->parameters[i].IsValid()) { | |
56 NOTREACHED_NT(); | |
57 return false; | |
58 } | |
59 } | |
60 | |
61 PolicyProcessor processor(policy); | |
62 PolicyResult result = processor.Evaluate(kShortEval, params->parameters, | |
63 params->count); | |
64 DCHECK_NT(POLICY_ERROR != result); | |
65 | |
66 return POLICY_MATCH == result && ASK_BROKER == processor.GetAction(); | |
67 } | |
68 | |
69 // ----------------------------------------------------------------------- | |
70 | |
71 // Hooks NtSetInformationThread to block RevertToSelf from being | |
72 // called before the actual call to LowerToken. | |
73 NTSTATUS WINAPI TargetNtSetInformationThread( | |
74 NtSetInformationThreadFunction orig_SetInformationThread, HANDLE thread, | |
75 NT_THREAD_INFORMATION_CLASS thread_info_class, PVOID thread_information, | |
76 ULONG thread_information_bytes) { | |
77 do { | |
78 if (SandboxFactory::GetTargetServices()->GetState()->RevertedToSelf()) | |
79 break; | |
80 if (ThreadImpersonationToken != thread_info_class) | |
81 break; | |
82 if (!thread_information) | |
83 break; | |
84 HANDLE token; | |
85 if (sizeof(token) > thread_information_bytes) | |
86 break; | |
87 | |
88 NTSTATUS ret = CopyData(&token, thread_information, sizeof(token)); | |
89 if (!NT_SUCCESS(ret) || NULL != token) | |
90 break; | |
91 | |
92 // This is a revert to self. | |
93 return STATUS_SUCCESS; | |
94 } while (false); | |
95 | |
96 return orig_SetInformationThread(thread, thread_info_class, | |
97 thread_information, | |
98 thread_information_bytes); | |
99 } | |
100 | |
101 // Hooks NtOpenThreadToken to force the open_as_self parameter to be set to | |
102 // FALSE if we are still running with the impersonation token. open_as_self set | |
103 // to TRUE means that the token will be open using the process token instead of | |
104 // the impersonation token. This is bad because the process token does not have | |
105 // access to open the thread token. | |
106 NTSTATUS WINAPI TargetNtOpenThreadToken( | |
107 NtOpenThreadTokenFunction orig_OpenThreadToken, HANDLE thread, | |
108 ACCESS_MASK desired_access, BOOLEAN open_as_self, PHANDLE token) { | |
109 if (!SandboxFactory::GetTargetServices()->GetState()->RevertedToSelf()) | |
110 open_as_self = FALSE; | |
111 | |
112 return orig_OpenThreadToken(thread, desired_access, open_as_self, token); | |
113 } | |
114 | |
115 // See comment for TargetNtOpenThreadToken | |
116 NTSTATUS WINAPI TargetNtOpenThreadTokenEx( | |
117 NtOpenThreadTokenExFunction orig_OpenThreadTokenEx, HANDLE thread, | |
118 ACCESS_MASK desired_access, BOOLEAN open_as_self, ULONG handle_attributes, | |
119 PHANDLE token) { | |
120 if (!SandboxFactory::GetTargetServices()->GetState()->RevertedToSelf()) | |
121 open_as_self = FALSE; | |
122 | |
123 return orig_OpenThreadTokenEx(thread, desired_access, open_as_self, | |
124 handle_attributes, token); | |
125 } | |
126 | |
127 } // namespace sandbox | |
OLD | NEW |