OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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/threading/platform_thread.h" | 5 #include "base/threading/platform_thread.h" |
6 | 6 |
7 #include <stddef.h> | 7 #include <stddef.h> |
8 | 8 |
9 #include "base/debug/activity_tracker.h" | 9 #include "base/debug/activity_tracker.h" |
10 #include "base/debug/alias.h" | 10 #include "base/debug/alias.h" |
11 #include "base/debug/profiler.h" | 11 #include "base/debug/profiler.h" |
12 #include "base/logging.h" | 12 #include "base/logging.h" |
| 13 #include "base/strings/utf_string_conversions.h" |
13 #include "base/threading/thread_id_name_manager.h" | 14 #include "base/threading/thread_id_name_manager.h" |
14 #include "base/threading/thread_restrictions.h" | 15 #include "base/threading/thread_restrictions.h" |
15 #include "base/tracked_objects.h" | 16 #include "base/tracked_objects.h" |
16 #include "base/win/scoped_handle.h" | 17 #include "base/win/scoped_handle.h" |
17 | 18 |
18 namespace base { | 19 namespace base { |
19 | 20 |
20 namespace { | 21 namespace { |
21 | 22 |
22 // The information on how to set the thread name comes from | 23 // The information on how to set the thread name comes from |
23 // a MSDN article: http://msdn2.microsoft.com/en-us/library/xcb2z8hs.aspx | 24 // a MSDN article: http://msdn2.microsoft.com/en-us/library/xcb2z8hs.aspx |
24 const DWORD kVCThreadNameException = 0x406D1388; | 25 const DWORD kVCThreadNameException = 0x406D1388; |
25 | 26 |
26 typedef struct tagTHREADNAME_INFO { | 27 typedef struct tagTHREADNAME_INFO { |
27 DWORD dwType; // Must be 0x1000. | 28 DWORD dwType; // Must be 0x1000. |
28 LPCSTR szName; // Pointer to name (in user addr space). | 29 LPCSTR szName; // Pointer to name (in user addr space). |
29 DWORD dwThreadID; // Thread ID (-1=caller thread). | 30 DWORD dwThreadID; // Thread ID (-1=caller thread). |
30 DWORD dwFlags; // Reserved for future use, must be zero. | 31 DWORD dwFlags; // Reserved for future use, must be zero. |
31 } THREADNAME_INFO; | 32 } THREADNAME_INFO; |
32 | 33 |
| 34 // The SetThreadDescription API was brought in version 1607 of Windows 10. |
| 35 typedef HRESULT(WINAPI* SetThreadDescription)(HANDLE hThread, |
| 36 PCWSTR lpThreadDescription); |
| 37 |
33 // This function has try handling, so it is separated out of its caller. | 38 // This function has try handling, so it is separated out of its caller. |
34 void SetNameInternal(PlatformThreadId thread_id, const char* name) { | 39 void SetNameInternal(PlatformThreadId thread_id, const char* name) { |
35 THREADNAME_INFO info; | 40 THREADNAME_INFO info; |
36 info.dwType = 0x1000; | 41 info.dwType = 0x1000; |
37 info.szName = name; | 42 info.szName = name; |
38 info.dwThreadID = thread_id; | 43 info.dwThreadID = thread_id; |
39 info.dwFlags = 0; | 44 info.dwFlags = 0; |
40 | 45 |
41 __try { | 46 __try { |
42 RaiseException(kVCThreadNameException, 0, sizeof(info)/sizeof(DWORD), | 47 RaiseException(kVCThreadNameException, 0, sizeof(info)/sizeof(DWORD), |
(...skipping 122 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
165 | 170 |
166 // On Windows only, we don't need to tell the profiler about the "BrokerEvent" | 171 // On Windows only, we don't need to tell the profiler about the "BrokerEvent" |
167 // thread, as it exists only in the chrome.exe image, and never spawns or runs | 172 // thread, as it exists only in the chrome.exe image, and never spawns or runs |
168 // tasks (items which could be profiled). This test avoids the notification, | 173 // tasks (items which could be profiled). This test avoids the notification, |
169 // which would also (as a side effect) initialize the profiler in this unused | 174 // which would also (as a side effect) initialize the profiler in this unused |
170 // context, including setting up thread local storage, etc. The performance | 175 // context, including setting up thread local storage, etc. The performance |
171 // impact is not terrible, but there is no reason to do initialize it. | 176 // impact is not terrible, but there is no reason to do initialize it. |
172 if (name != "BrokerEvent") | 177 if (name != "BrokerEvent") |
173 tracked_objects::ThreadData::InitializeThreadContext(name); | 178 tracked_objects::ThreadData::InitializeThreadContext(name); |
174 | 179 |
| 180 // The SetThreadDescription API works even if no debugger is attached. |
| 181 auto set_thread_description_func = |
| 182 reinterpret_cast<SetThreadDescription>(::GetProcAddress( |
| 183 ::GetModuleHandle(L"Kernel32.dll"), "SetThreadDescription")); |
| 184 if (set_thread_description_func) { |
| 185 set_thread_description_func(::GetCurrentThread(), |
| 186 base::UTF8ToWide(name).c_str()); |
| 187 } |
| 188 |
175 // The debugger needs to be around to catch the name in the exception. If | 189 // The debugger needs to be around to catch the name in the exception. If |
176 // there isn't a debugger, we are just needlessly throwing an exception. | 190 // there isn't a debugger, we are just needlessly throwing an exception. |
177 // If this image file is instrumented, we raise the exception anyway | 191 // If this image file is instrumented, we raise the exception anyway |
178 // to provide the profiler with human-readable thread names. | 192 // to provide the profiler with human-readable thread names. |
179 if (!::IsDebuggerPresent() && !base::debug::IsBinaryInstrumented()) | 193 if (!::IsDebuggerPresent() && !base::debug::IsBinaryInstrumented()) |
180 return; | 194 return; |
181 | 195 |
182 SetNameInternal(CurrentId(), name.c_str()); | 196 SetNameInternal(CurrentId(), name.c_str()); |
183 } | 197 } |
184 | 198 |
(...skipping 102 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
287 return ThreadPriority::REALTIME_AUDIO; | 301 return ThreadPriority::REALTIME_AUDIO; |
288 case THREAD_PRIORITY_ERROR_RETURN: | 302 case THREAD_PRIORITY_ERROR_RETURN: |
289 DPCHECK(false) << "GetThreadPriority error"; // Falls through. | 303 DPCHECK(false) << "GetThreadPriority error"; // Falls through. |
290 default: | 304 default: |
291 NOTREACHED() << "Unexpected priority: " << priority; | 305 NOTREACHED() << "Unexpected priority: " << priority; |
292 return ThreadPriority::NORMAL; | 306 return ThreadPriority::NORMAL; |
293 } | 307 } |
294 } | 308 } |
295 | 309 |
296 } // namespace base | 310 } // namespace base |
OLD | NEW |