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
|
// Copyright 2024 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/elevation_service/elevation_service_delegate.h"
#include <wrl/module.h>
#include <utility>
#include "base/command_line.h"
#include "base/containers/heap_array.h"
#include "base/logging.h"
#include "base/types/expected.h"
#include "chrome/common/win/eventlog_messages.h"
#include "chrome/elevation_service/elevated_recovery_impl.h"
#include "chrome/elevation_service/elevator.h"
#include "chrome/install_static/install_util.h"
namespace elevation_service {
uint16_t Delegate::GetLogEventCategory() {
return ELEVATION_SERVICE_CATEGORY;
}
uint32_t Delegate::GetLogEventMessageId() {
return MSG_ELEVATION_SERVICE_LOG_MESSAGE;
}
base::expected<base::HeapArray<FactoryAndClsid>, HRESULT>
Delegate::CreateClassFactories() {
unsigned int flags = Microsoft::WRL::ModuleType::OutOfProc;
auto result = base::HeapArray<FactoryAndClsid>::WithSize(1);
Microsoft::WRL::ComPtr<IUnknown> unknown;
HRESULT hr = Microsoft::WRL::Details::CreateClassFactory<
Microsoft::WRL::SimpleClassFactory<Elevator>>(
&flags, nullptr, __uuidof(IClassFactory), &unknown);
if (SUCCEEDED(hr)) {
hr = unknown.As(&result[0].factory);
}
if (FAILED(hr)) {
return base::unexpected(hr);
}
result[0].clsid = base::CommandLine::ForCurrentProcess()->HasSwitch(
switches::kElevatorClsIdForTestingSwitch)
? kTestElevatorClsid
: install_static::GetElevatorClsid();
return base::ok(std::move(result));
}
bool Delegate::PreRun() {
if (auto hresult = CleanupChromeRecoveryDirectory(); FAILED(hresult)) {
LOG(WARNING) << "Failed to clean Chrome recovery directory; hresult = "
<< std::hex << hresult;
}
return false; // This delegate does not implement `Run()`.
}
} // namespace elevation_service
|
