Index: sandbox/win/src/process_thread_interception.cc |
diff --git a/sandbox/win/src/process_thread_interception.cc b/sandbox/win/src/process_thread_interception.cc |
index f8a144f353d5576884eccca712142eaed85bfdfd..f951b4c87b2087f9376446402b444b6012fddf1d 100644 |
--- a/sandbox/win/src/process_thread_interception.cc |
+++ b/sandbox/win/src/process_thread_interception.cc |
@@ -5,7 +5,7 @@ |
#include "sandbox/win/src/process_thread_interception.h" |
#include <stdint.h> |
- |
+#include "base/win/windows_version.h" |
#include "sandbox/win/src/crosscall_client.h" |
#include "sandbox/win/src/ipc_tags.h" |
#include "sandbox/win/src/policy_params.h" |
@@ -408,4 +408,68 @@ BOOL WINAPI TargetCreateProcessA(CreateProcessAFunction orig_CreateProcessA, |
return FALSE; |
} |
+HANDLE WINAPI TargetCreateThread(CreateThreadFunction orig_CreateThread, |
+ LPSECURITY_ATTRIBUTES thread_attributes, |
+ SIZE_T stack_size, |
+ LPTHREAD_START_ROUTINE start_address, |
+ LPVOID parameter, |
+ DWORD creation_flags, |
+ LPDWORD thread_id) { |
+ HANDLE hThread = NULL; |
+ |
cpu_(ooo_6.6-7.5)
2016/02/04 22:50:07
yeah, I agree OpenThread is a better template ther
liamjm (20p)
2016/02/05 19:22:43
Done.
|
+ TargetServices* target_services = SandboxFactory::GetTargetServices(); |
+ if (NULL == target_services || |
+ target_services->GetState()->IsCsrssConnected()) { |
+ hThread = orig_CreateThread(thread_attributes, stack_size, start_address, |
+ parameter, creation_flags, thread_id); |
+ if (hThread) { |
+ return hThread; |
+ } |
+ } |
+ |
+ if (NULL == target_services) |
+ return NULL; |
+ |
+ // We don't trust that the IPC can work this early. |
+ if (!target_services->GetState()->InitCalled()) |
+ return NULL; |
+ |
+ DWORD original_error = ::GetLastError(); |
+ |
+ do { |
+ if (NULL != thread_id && |
+ !ValidParameter(thread_id, sizeof(*thread_id), WRITE)) |
+ break; |
+ |
+ void* memory = GetGlobalIPCMemory(); |
+ if (NULL == memory) |
+ break; |
+ |
+ SharedMemIPCClient ipc(memory); |
+ CrossCallReturn answer = {0}; |
+ |
+ // NOTE: we don't pass the thread_attributes through. This matches the |
+ // approach in CreateProcess and in CreateThreadInternal(). |
+ ResultCode code = CrossCall(ipc, IPC_CREATETHREAD_TAG, |
+ reinterpret_cast<LPVOID>(stack_size), |
+ reinterpret_cast<LPVOID>(start_address), |
+ parameter, creation_flags, &answer); |
+ if (SBOX_ALL_OK != code) |
+ break; |
+ |
+ ::SetLastError(answer.win32_result); |
+ if (ERROR_SUCCESS != answer.win32_result) { |
+ return NULL; |
+ } |
+ |
+ if (thread_id != NULL) { |
+ *thread_id = ::GetThreadId(answer.handle); |
+ } |
+ return answer.handle; |
+ } while (false); |
+ |
+ ::SetLastError(original_error); |
+ return NULL; |
+} |
+ |
} // namespace sandbox |