1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
// Copyright 2018 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.
#ifndef BASE_WIN_SCOPED_HANDLE_VERIFIER_H_
#define BASE_WIN_SCOPED_HANDLE_VERIFIER_H_
#include <memory>
#include <unordered_map>
#include "base/base_export.h"
#include "base/debug/stack_trace.h"
#include "base/hash/hash.h"
#include "base/memory/raw_ptr.h"
#include "base/synchronization/lock_impl.h"
#include "base/threading/thread_local.h"
#include "base/win/windows_types.h"
namespace base {
namespace win {
namespace internal {
struct HandleHash {
size_t operator()(const HANDLE& handle) const {
return base::FastHash(as_bytes(make_span(&handle, 1)));
}
};
struct ScopedHandleVerifierInfo {
ScopedHandleVerifierInfo(const void* owner,
const void* pc1,
const void* pc2,
std::unique_ptr<debug::StackTrace> stack,
DWORD thread_id);
~ScopedHandleVerifierInfo();
ScopedHandleVerifierInfo(const ScopedHandleVerifierInfo&) = delete;
ScopedHandleVerifierInfo& operator=(const ScopedHandleVerifierInfo&) = delete;
ScopedHandleVerifierInfo(ScopedHandleVerifierInfo&&) noexcept;
ScopedHandleVerifierInfo& operator=(ScopedHandleVerifierInfo&&) noexcept;
raw_ptr<const void> owner;
raw_ptr<const void> pc1;
raw_ptr<const void> pc2;
std::unique_ptr<debug::StackTrace> stack;
DWORD thread_id;
};
// Implements the actual object that is verifying handles for this process.
// The active instance is shared across the module boundary but there is no
// way to delete this object from the wrong side of it (or any side, actually).
// We need [[clang::lto_visibility_public]] because instances of this class are
// passed across module boundaries. This means different modules must have
// compatible definitions of the class even when whole program optimization is
// enabled - which is what this attribute accomplishes. The pragma stops MSVC
// from emitting an unrecognized attribute warning.
#pragma warning(push)
#pragma warning(disable : 5030)
class [[clang::lto_visibility_public]] ScopedHandleVerifier {
#pragma warning(pop)
public:
explicit ScopedHandleVerifier(bool enabled);
ScopedHandleVerifier(const ScopedHandleVerifier&) = delete;
ScopedHandleVerifier& operator=(const ScopedHandleVerifier&) = delete;
// Retrieves the current verifier.
static ScopedHandleVerifier* Get();
// The methods required by HandleTraits. They are virtual because we need to
// forward the call execution to another module, instead of letting the
// compiler call the version that is linked in the current module.
virtual bool CloseHandle(HANDLE handle);
virtual void StartTracking(HANDLE handle, const void* owner, const void* pc1,
const void* pc2);
virtual void StopTracking(HANDLE handle, const void* owner, const void* pc1,
const void* pc2);
virtual void Disable();
virtual void OnHandleBeingClosed(HANDLE handle);
virtual HMODULE GetModule() const;
private:
~ScopedHandleVerifier(); // Not implemented.
void StartTrackingImpl(HANDLE handle, const void* owner, const void* pc1,
const void* pc2);
void StopTrackingImpl(HANDLE handle, const void* owner, const void* pc1,
const void* pc2);
void OnHandleBeingClosedImpl(HANDLE handle);
static base::internal::LockImpl* GetLock();
static void InstallVerifier();
static void ThreadSafeAssignOrCreateScopedHandleVerifier(
ScopedHandleVerifier * existing_verifier, bool enabled);
base::debug::StackTrace creation_stack_;
bool enabled_;
base::ThreadLocalBoolean closing_;
raw_ptr<base::internal::LockImpl> lock_;
std::unordered_map<HANDLE, ScopedHandleVerifierInfo, HandleHash> map_;
};
// This testing function returns the module that the HandleVerifier concrete
// implementation was instantiated in.
BASE_EXPORT HMODULE GetHandleVerifierModuleForTesting();
} // namespace internal
} // namespace win
} // namespace base
#endif // BASE_WIN_SCOPED_HANDLE_VERIFIER_H_
|
