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
|
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/win/conflicts/registry_key_watcher.h"
#include <windows.h>
#include <utility>
#include "base/functional/bind.h"
// static
std::unique_ptr<RegistryKeyWatcher> RegistryKeyWatcher::Create(
HKEY root,
const std::wstring& subkey,
REGSAM wow64access,
base::OnceClosure on_registry_key_deleted) {
std::unique_ptr<RegistryKeyWatcher> instance(new RegistryKeyWatcher(
root, subkey, wow64access, std::move(on_registry_key_deleted)));
if (!instance->IsWatching())
return nullptr;
return instance;
}
RegistryKeyWatcher::~RegistryKeyWatcher() = default;
RegistryKeyWatcher::RegistryKeyWatcher(
HKEY root,
const std::wstring& subkey,
REGSAM wow64access,
base::OnceClosure on_registry_key_deleted)
: registry_key_(std::make_unique<base::win::RegKey>(
root,
subkey.c_str(),
KEY_NOTIFY | KEY_QUERY_VALUE | wow64access)),
on_registry_key_deleted_(std::move(on_registry_key_deleted)) {
if (registry_key_->Valid())
StartWatching();
}
void RegistryKeyWatcher::StartWatching() {
if (!registry_key_->StartWatching(base::BindRepeating(
&RegistryKeyWatcher::OnRegistryKeyChanged, base::Unretained(this)))) {
registry_key_.reset();
}
}
bool RegistryKeyWatcher::IsWatching() {
return registry_key_ && registry_key_->Valid();
}
void RegistryKeyWatcher::OnRegistryKeyChanged() {
// This callback may be invoked for any modification on the registry key.
// Since this class cares only about the deletion of the key, read the default
// value of the key to figure out if it still exists.
std::wstring value;
if (registry_key_->ReadValue(nullptr, &value) == ERROR_KEY_DELETED) {
// The registry key is no longer needed.
registry_key_.reset();
// Run the callback last, because it may delete the RegistryKeyWatcher
// instance during its execution.
std::move(on_registry_key_deleted_).Run();
return;
}
// Keep watching.
StartWatching();
}
|
