Identify the hung thread using the Wait Chain Traversal API

chrome/chrome_watcher/wait_chain_util_win.cc

Index: chrome/chrome_watcher/wait_chain_util_win.cc
diff --git a/chrome/chrome_watcher/wait_chain_util_win.cc b/chrome/chrome_watcher/wait_chain_util_win.cc
new file mode 100644
index 0000000000000000000000000000000000000000..27dd4db256add8c9d7c03d487ef3a2ba4fa942ae
--- /dev/null
+++ b/chrome/chrome_watcher/wait_chain_util_win.cc
@@ -0,0 +1,122 @@
+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/chrome_watcher/wait_chain_util_win.h"
+
+#include <memory>
+
+#include "base/logging.h"
+#include "base/strings/stringprintf.h"
+
+namespace {
+
+// Helper deleter to hold a HWCT into a unique_ptr.
+struct WaitChainSessionDeleter {
+  using pointer = HWCT;
+  void operator()(HWCT session_handle) {
+    ::CloseThreadWaitChainSession(session_handle);
+  }
+};
+using ScopedWaitChainSessionHandle =
+    std::unique_ptr<HWCT, WaitChainSessionDeleter>;
+
+const wchar_t* WctObjectTypeToString(WCT_OBJECT_TYPE type) {
+  switch (type) {
+    case WctCriticalSectionType:
+      return L"CriticalSection";
+    case WctSendMessageType:
+      return L"SendMessage";
+    case WctMutexType:
+      return L"Mutex";
+    case WctAlpcType:
+      return L"Alpc";
+    case WctComType:
+      return L"Com";
+    case WctThreadWaitType:
+      return L"ThreadWait";
+    case WctProcessWaitType:
+      return L"ProcessWait";
+    case WctThreadType:
+      return L"Thread";
+    case WctComActivationType:
+      return L"ComActivation";
+    case WctUnknownType:
+      return L"Unknown";
+    default:
+      NOTREACHED();
+      return L"";
+  }
+}
+
+const wchar_t* WctObjectStatusToString(WCT_OBJECT_STATUS status) {
+  switch (status) {
+    case WctStatusNoAccess:
+      return L"NoAccess";
+    case WctStatusRunning:
+      return L"Running";
+    case WctStatusBlocked:
+      return L"Blocked";
+    case WctStatusPidOnly:
+      return L"PidOnly";
+    case WctStatusPidOnlyRpcss:
+      return L"PidOnlyRpcss";
+    case WctStatusOwned:
+      return L"Owned";
+    case WctStatusNotOwned:
+      return L"NotOwned";
+    case WctStatusAbandoned:
+      return L"Abandoned";
+    case WctStatusUnknown:
+      return L"Unknown";
+    case WctStatusError:
+      return L"Error";
+    default:
+      NOTREACHED();
+      return L"";
+  }
+}
+
+}  // namespace
+
+bool GetThreadWaitChain(DWORD thread_id,
+                        std::vector<WAITCHAIN_NODE_INFO>* wait_chain,
+                        bool* is_deadlock) {
+  DCHECK(wait_chain);
+  DCHECK(is_deadlock);
+
+  // Open a synchronous session.
+  ScopedWaitChainSessionHandle session_handle(
+      ::OpenThreadWaitChainSession(0, nullptr));
+  if (!session_handle) {
+    PLOG(ERROR) << "Failed to create the Wait Chain session.";
+    return false;
+  }
+
+  DWORD nb_nodes = WCT_MAX_NODE_COUNT;
+  wait_chain->resize(nb_nodes);
+  BOOL is_cycle;
+  if (!::GetThreadWaitChain(session_handle.get(), NULL, WCTP_GETINFO_ALL_FLAGS,
+                            thread_id, &nb_nodes, wait_chain->data(),
+                            &is_cycle)) {
+    PLOG(ERROR) << "Failed to get the thread wait chain.";
+    return false;
+  }
+
+  *is_deadlock = is_cycle ? true : false;
+  wait_chain->resize(nb_nodes);
+
+  return true;
+}
+
+std::wstring WaitChainNodeToString(const WAITCHAIN_NODE_INFO& node) {
+  if (node.ObjectType == WctThreadType) {
+    return base::StringPrintf(L"Thread #%d with status %ls",
+                              node.ThreadObject.ThreadId,
+                              WctObjectStatusToString(node.ObjectStatus));
+  } else {
+    return base::StringPrintf(L"Lock of type %ls with status %ls",
+                              WctObjectTypeToString(node.ObjectType),
+                              WctObjectStatusToString(node.ObjectStatus));
+  }
+}