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
|
// Copyright 2012 The Chromium Authors
// 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 <string>
#include "base/check_op.h"
#include "base/debug/alias.h"
#include "base/logging.h"
#include "base/win/scoped_handle_verifier.h"
#include "base/win/win_util.h"
#include "base/win/windows_types.h"
namespace base {
namespace win {
using base::win::internal::ScopedHandleVerifier;
std::ostream& operator<<(std::ostream& os, HandleOperation operation) {
switch (operation) {
case HandleOperation::kHandleAlreadyTracked:
return os << "Handle Already Tracked";
case HandleOperation::kCloseHandleNotTracked:
return os << "Closing an untracked handle";
case HandleOperation::kCloseHandleNotOwner:
return os << "Closing a handle owned by something else";
case HandleOperation::kCloseHandleHook:
return os << "CloseHandleHook validation failure";
case HandleOperation::kDuplicateHandleHook:
return os << "DuplicateHandleHook validation failure";
}
}
// Static.
bool HandleTraits::CloseHandle(HANDLE handle) {
return ScopedHandleVerifier::Get()->CloseHandle(handle);
}
// Static.
void VerifierTraits::StartTracking(HANDLE handle,
const void* owner,
const void* pc1,
const void* pc2) {
return ScopedHandleVerifier::Get()->StartTracking(handle, owner, pc1, pc2);
}
// Static.
void VerifierTraits::StopTracking(HANDLE handle,
const void* owner,
const void* pc1,
const void* pc2) {
return ScopedHandleVerifier::Get()->StopTracking(handle, owner, pc1, pc2);
}
void DisableHandleVerifier() {
return ScopedHandleVerifier::Get()->Disable();
}
void OnHandleBeingClosed(HANDLE handle, HandleOperation operation) {
return ScopedHandleVerifier::Get()->OnHandleBeingClosed(handle, operation);
}
expected<ScopedHandle, NTSTATUS> TakeHandleOfType(
HANDLE handle,
std::wstring_view object_type_name) {
// Invalid and pseudo handle values are rejected by GetObjectTypeName().
auto type_name = GetObjectTypeName(handle);
if (!type_name.has_value()) {
// `handle` is invalid. Return the error to the caller.
return unexpected(type_name.error());
}
// Crash if `handle` is an unexpected type. This represents a dangerous
// type confusion condition that should never happen.
base::debug::Alias(&handle);
CHECK_EQ(*type_name, object_type_name);
return ScopedHandle(handle); // Ownership of `handle` goes to the caller.
}
} // namespace win
} // namespace base
|
