| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 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 | 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 "base/platform_thread.h" | 5 #include "base/platform_thread.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "base/thread_restrictions.h" |
| 8 #include "base/win/windows_version.h" | 9 #include "base/win/windows_version.h" |
| 9 | 10 |
| 10 namespace { | 11 namespace { |
| 11 | 12 |
| 12 // The information on how to set the thread name comes from | 13 // The information on how to set the thread name comes from |
| 13 // a MSDN article: http://msdn2.microsoft.com/en-us/library/xcb2z8hs.aspx | 14 // a MSDN article: http://msdn2.microsoft.com/en-us/library/xcb2z8hs.aspx |
| 14 const DWORD kVCThreadNameException = 0x406D1388; | 15 const DWORD kVCThreadNameException = 0x406D1388; |
| 15 | 16 |
| 16 typedef struct tagTHREADNAME_INFO { | 17 typedef struct tagTHREADNAME_INFO { |
| 17 DWORD dwType; // Must be 0x1000. | 18 DWORD dwType; // Must be 0x1000. |
| 18 LPCSTR szName; // Pointer to name (in user addr space). | 19 LPCSTR szName; // Pointer to name (in user addr space). |
| 19 DWORD dwThreadID; // Thread ID (-1=caller thread). | 20 DWORD dwThreadID; // Thread ID (-1=caller thread). |
| 20 DWORD dwFlags; // Reserved for future use, must be zero. | 21 DWORD dwFlags; // Reserved for future use, must be zero. |
| 21 } THREADNAME_INFO; | 22 } THREADNAME_INFO; |
| 22 | 23 |
| 23 DWORD __stdcall ThreadFunc(void* closure) { | 24 struct ThreadParams { |
| 24 PlatformThread::Delegate* delegate = | 25 PlatformThread::Delegate* delegate; |
| 25 static_cast<PlatformThread::Delegate*>(closure); | 26 bool joinable; |
| 27 }; |
| 28 |
| 29 DWORD __stdcall ThreadFunc(void* params) { |
| 30 ThreadParams* thread_params = static_cast<ThreadParams*>(params); |
| 31 PlatformThread::Delegate* delegate = thread_params->delegate; |
| 32 if (!thread_params->joinable) |
| 33 base::ThreadRestrictions::SetSingletonAllowed(false); |
| 34 delete thread_params; |
| 26 delegate->ThreadMain(); | 35 delegate->ThreadMain(); |
| 27 return NULL; | 36 return NULL; |
| 28 } | 37 } |
| 29 | 38 |
| 39 // CreateThreadInternal() matches PlatformThread::Create(), except that |
| 40 // |out_thread_handle| may be NULL, in which case a non-joinable thread is |
| 41 // created. |
| 42 bool CreateThreadInternal(size_t stack_size, |
| 43 PlatformThread::Delegate* delegate, |
| 44 PlatformThreadHandle* out_thread_handle) { |
| 45 PlatformThreadHandle thread_handle; |
| 46 unsigned int flags = 0; |
| 47 if (stack_size > 0 && base::win::GetVersion() >= base::win::VERSION_XP) { |
| 48 flags = STACK_SIZE_PARAM_IS_A_RESERVATION; |
| 49 } else { |
| 50 stack_size = 0; |
| 51 } |
| 52 |
| 53 ThreadParams* params = new ThreadParams; |
| 54 params->delegate = delegate; |
| 55 params->joinable = out_thread_handle != NULL; |
| 56 |
| 57 // Using CreateThread here vs _beginthreadex makes thread creation a bit |
| 58 // faster and doesn't require the loader lock to be available. Our code will |
| 59 // have to work running on CreateThread() threads anyway, since we run code |
| 60 // on the Windows thread pool, etc. For some background on the difference: |
| 61 // http://www.microsoft.com/msj/1099/win32/win321099.aspx |
| 62 thread_handle = CreateThread( |
| 63 NULL, stack_size, ThreadFunc, params, flags, NULL); |
| 64 if (!thread_handle) { |
| 65 delete params; |
| 66 return false; |
| 67 } |
| 68 |
| 69 if (out_thread_handle) |
| 70 *out_thread_handle = thread_handle; |
| 71 else |
| 72 CloseHandle(thread_handle); |
| 73 return true; |
| 74 } |
| 75 |
| 30 } // namespace | 76 } // namespace |
| 31 | 77 |
| 32 // static | 78 // static |
| 33 PlatformThreadId PlatformThread::CurrentId() { | 79 PlatformThreadId PlatformThread::CurrentId() { |
| 34 return GetCurrentThreadId(); | 80 return GetCurrentThreadId(); |
| 35 } | 81 } |
| 36 | 82 |
| 37 // static | 83 // static |
| 38 void PlatformThread::YieldCurrentThread() { | 84 void PlatformThread::YieldCurrentThread() { |
| 39 ::Sleep(0); | 85 ::Sleep(0); |
| (...skipping 20 matching lines...) Expand all Loading... |
| 60 __try { | 106 __try { |
| 61 RaiseException(kVCThreadNameException, 0, sizeof(info)/sizeof(DWORD), | 107 RaiseException(kVCThreadNameException, 0, sizeof(info)/sizeof(DWORD), |
| 62 reinterpret_cast<DWORD_PTR*>(&info)); | 108 reinterpret_cast<DWORD_PTR*>(&info)); |
| 63 } __except(EXCEPTION_CONTINUE_EXECUTION) { | 109 } __except(EXCEPTION_CONTINUE_EXECUTION) { |
| 64 } | 110 } |
| 65 } | 111 } |
| 66 | 112 |
| 67 // static | 113 // static |
| 68 bool PlatformThread::Create(size_t stack_size, Delegate* delegate, | 114 bool PlatformThread::Create(size_t stack_size, Delegate* delegate, |
| 69 PlatformThreadHandle* thread_handle) { | 115 PlatformThreadHandle* thread_handle) { |
| 70 unsigned int flags = 0; | 116 DCHECK(thread_handle); |
| 71 if (stack_size > 0 && base::win::GetVersion() >= base::win::VERSION_XP) { | 117 return CreateThreadInternal(stack_size, delegate, thread_handle); |
| 72 flags = STACK_SIZE_PARAM_IS_A_RESERVATION; | |
| 73 } else { | |
| 74 stack_size = 0; | |
| 75 } | |
| 76 | |
| 77 // Using CreateThread here vs _beginthreadex makes thread creation a bit | |
| 78 // faster and doesn't require the loader lock to be available. Our code will | |
| 79 // have to work running on CreateThread() threads anyway, since we run code | |
| 80 // on the Windows thread pool, etc. For some background on the difference: | |
| 81 // http://www.microsoft.com/msj/1099/win32/win321099.aspx | |
| 82 *thread_handle = CreateThread( | |
| 83 NULL, stack_size, ThreadFunc, delegate, flags, NULL); | |
| 84 return *thread_handle != NULL; | |
| 85 } | 118 } |
| 86 | 119 |
| 87 // static | 120 // static |
| 88 bool PlatformThread::CreateNonJoinable(size_t stack_size, Delegate* delegate) { | 121 bool PlatformThread::CreateNonJoinable(size_t stack_size, Delegate* delegate) { |
| 89 PlatformThreadHandle thread_handle; | 122 return CreateThreadInternal(stack_size, delegate, NULL); |
| 90 bool result = Create(stack_size, delegate, &thread_handle); | |
| 91 CloseHandle(thread_handle); | |
| 92 return result; | |
| 93 } | 123 } |
| 94 | 124 |
| 95 // static | 125 // static |
| 96 void PlatformThread::Join(PlatformThreadHandle thread_handle) { | 126 void PlatformThread::Join(PlatformThreadHandle thread_handle) { |
| 97 DCHECK(thread_handle); | 127 DCHECK(thread_handle); |
| 98 | 128 |
| 99 // Wait for the thread to exit. It should already have terminated but make | 129 // Wait for the thread to exit. It should already have terminated but make |
| 100 // sure this assumption is valid. | 130 // sure this assumption is valid. |
| 101 DWORD result = WaitForSingleObject(thread_handle, INFINITE); | 131 DWORD result = WaitForSingleObject(thread_handle, INFINITE); |
| 102 DCHECK_EQ(WAIT_OBJECT_0, result); | 132 DCHECK_EQ(WAIT_OBJECT_0, result); |
| 103 | 133 |
| 104 CloseHandle(thread_handle); | 134 CloseHandle(thread_handle); |
| 105 } | 135 } |
| OLD | NEW |