Add a buildflag to use the handle verifier in a per module mode.

base/win/scoped_handle.cc

 // Copyright (c) 2012 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 "base/win/scoped_handle.h"
 
 #include <stddef.h>
 
 #include <unordered_map>
 
 #include "base/debug/alias.h"
 #include "base/debug/stack_trace.h"
 #include "base/hash.h"
 #include "base/lazy_instance.h"
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/synchronization/lock_impl.h"
 #include "base/threading/thread_local.h"
 #include "base/win/current_module.h"
 
+#if BUILDFLAG(ENABLE_HANDLE_VERIFIER)
 extern "C" {
 __declspec(dllexport) void* GetHandleVerifier();
 typedef void* (*GetHandleVerifierFn)();
 }
 
 namespace {
 
 struct HandleHash {
   size_t operator()(const HANDLE& handle) const {
     char buffer[sizeof(handle)];
@@ skipping 216 matching lines @@
 HMODULE ActiveVerifier::GetModule() const {
   return CURRENT_MODULE();
 }
 
 }  // namespace
 
 void* GetHandleVerifier() {
   return ActiveVerifier::Get();
 }
 
+#endif  // BUILDFLAG(ENABLE_HANDLE_VERIFIER)
+
 namespace base {
 namespace win {
 
 // Static.
 bool HandleTraits::CloseHandle(HANDLE handle) {
+#if BUILDFLAG(ENABLE_HANDLE_VERIFIER)
   return ActiveVerifier::Get()->CloseHandle(handle);
+#else  // BUILDFLAG(ENABLE_HANDLE_VERIFIER)
+  if (!::CloseHandle(handle))
+    CHECK(false);
+  return true;
+#endif
 }
 
+void OnHandleBeingClosed(HANDLE handle) {
+#if BUILDFLAG(ENABLE_HANDLE_VERIFIER)
+  return ActiveVerifier::Get()->OnHandleBeingClosed(handle);
+#endif
+}
+
+#if BUILDFLAG(ENABLE_HANDLE_VERIFIER)
 // Static.
 void VerifierTraits::StartTracking(HANDLE handle, const void* owner,
                                    const void* pc1, const void* pc2) {
   return ActiveVerifier::Get()->StartTracking(handle, owner, pc1, pc2);
 }
 
 // Static.
 void VerifierTraits::StopTracking(HANDLE handle, const void* owner,
                                   const void* pc1, const void* pc2) {
   return ActiveVerifier::Get()->StopTracking(handle, owner, pc1, pc2);
 }
 
-void DisableHandleVerifier() {
-  return ActiveVerifier::Get()->Disable();
-}
-
-void OnHandleBeingClosed(HANDLE handle) {
-  return ActiveVerifier::Get()->OnHandleBeingClosed(handle);
-}
-
 HMODULE GetHandleVerifierModuleForTesting() {
   return g_active_verifier->GetModule();
 }
+#endif  // BUILDFLAG(ENABLE_HANDLE_VERIFIER)
+
+void DisableHandleVerifier() {
+#if BUILDFLAG(ENABLE_HANDLE_VERIFIER)
+  return ActiveVerifier::Get()->Disable();
+#endif
+}
 
 }  // namespace win
 }  // namespace base