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
|
// Copyright 2020 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/upgrade_detector/registry_monitor.h"
#include <memory>
#include <utility>
#include "base/functional/bind.h"
#include "base/functional/callback.h"
#include "base/location.h"
#include "base/task/sequenced_task_runner.h"
#include "base/win/registry.h"
#include "chrome/browser/upgrade_detector/installed_version_monitor.h"
#include "chrome/install_static/install_util.h"
namespace {
base::win::RegKey GetDefaultMonitorLocation() {
return base::win::RegKey(install_static::IsSystemInstall()
? HKEY_LOCAL_MACHINE
: HKEY_CURRENT_USER,
install_static::GetClientsKeyPath().c_str(),
KEY_NOTIFY | KEY_WOW64_32KEY);
}
} // namespace
RegistryMonitor::RegistryMonitor(base::win::RegKey key)
: clients_key_(std::move(key)) {}
RegistryMonitor::~RegistryMonitor() = default;
void RegistryMonitor::Start(Callback on_change_callback) {
DCHECK(on_change_callback);
DCHECK(!on_change_callback_);
on_change_callback_ = std::move(on_change_callback);
StartWatching();
}
void RegistryMonitor::StartWatching() {
// base::Unretained is safe because RegistryMonitor owns the RegKey.
// Destruction of this instance will cancel the watch and disable any
// outstanding notifications.
if (!clients_key_.Valid() ||
!clients_key_.StartWatching(base::BindOnce(
&RegistryMonitor::OnClientsKeyChanged, base::Unretained(this)))) {
// Starting the watch failed. Report this back to the poller via a delayed
// task.
base::SequencedTaskRunner::GetCurrentDefault()->PostTask(
FROM_HERE, base::BindOnce(on_change_callback_, /*error=*/true));
}
}
void RegistryMonitor::OnClientsKeyChanged() {
on_change_callback_.Run(/*error=*/false);
StartWatching();
}
// static
std::unique_ptr<InstalledVersionMonitor> InstalledVersionMonitor::Create() {
return std::make_unique<RegistryMonitor>(GetDefaultMonitorLocation());
}
|
