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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <cstdint>
#include <iterator>
#include <memory>
#include <string>
#include <utility>
#include "base/command_line.h"
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/files/scoped_temp_dir.h"
#include "base/functional/bind.h"
#include "base/functional/callback.h"
#include "base/memory/ref_counted.h"
#include "base/test/scoped_run_loop_timeout.h"
#include "base/test/task_environment.h"
#include "base/time/time.h"
#include "build/build_config.h"
#include "chrome/browser/component_updater/recovery_improved_component_installer.h"
#include "components/crx_file/crx_verifier.h"
#include "components/services/unzip/content/unzip_service.h"
#include "components/services/unzip/in_process_unzipper.h"
#include "components/update_client/test_utils.h"
#include "components/update_client/update_client_errors.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "testing/platform_test.h"
namespace component_updater {
namespace {
// The public key hash for the test CRX.
constexpr uint8_t kKeyHashTest[] = {
0x69, 0xfc, 0x41, 0xf6, 0x17, 0x20, 0xc6, 0x36, 0x92, 0xcd, 0x95,
0x76, 0x69, 0xf6, 0x28, 0xcc, 0xbe, 0x98, 0x4b, 0x93, 0x17, 0xd6,
0x9c, 0xb3, 0x64, 0x0c, 0x0d, 0x25, 0x61, 0xc5, 0x80, 0x1d};
class RecoveryImprovedActionHandlerTest : public PlatformTest {
public:
RecoveryImprovedActionHandlerTest() = default;
protected:
base::ScopedTempDir temp_dir_;
private:
base::test::TaskEnvironment task_environment_;
};
class TestActionHandler : public RecoveryComponentActionHandler {
public:
// Accepts a test CRX without a production publisher proof.
TestActionHandler()
: RecoveryComponentActionHandler(
{std::begin(kKeyHashTest), std::end(kKeyHashTest)},
crx_file::VerifierFormat::CRX3) {}
TestActionHandler(const TestActionHandler&) = delete;
TestActionHandler& operator=(const TestActionHandler&) = delete;
protected:
~TestActionHandler() override = default;
private:
// Overrides for RecoveryComponentActionHandler.
base::CommandLine MakeCommandLine(
const base::FilePath& unpack_path) const override;
void PrepareFiles(const base::FilePath& unpack_path) const override;
void Elevate(Callback callback) override;
};
base::CommandLine TestActionHandler::MakeCommandLine(
const base::FilePath& unpack_path) const {
base::CommandLine command_line(
unpack_path.Append(FILE_PATH_LITERAL("ChromeRecovery.exe")));
return command_line;
}
void TestActionHandler::PrepareFiles(const base::FilePath& unpack_path) const {}
void TestActionHandler::Elevate(Callback callback) {
ADD_FAILURE() << "This test fixture only tests the per-user execution flow.";
}
} // namespace
#if BUILDFLAG(IS_WIN) || BUILDFLAG(IS_MAC)
TEST_F(RecoveryImprovedActionHandlerTest, HandleError) {
unzip::SetUnzipperLaunchOverrideForTesting(
base::BindRepeating(&unzip::LaunchInProcessUnzipper));
// Tests the error is propagated through the callback in the error case.
base::RunLoop runloop;
base::MakeRefCounted<TestActionHandler>()->Handle(
base::FilePath{FILE_PATH_LITERAL("not-found")}, "some-session-id",
base::BindOnce(
[](base::OnceClosure quit_closure, bool succeeded, int error_code,
int extra_code1) {
EXPECT_FALSE(succeeded);
EXPECT_EQ(update_client::UnpackerError::kInvalidFile,
static_cast<update_client::UnpackerError>(error_code));
EXPECT_EQ(2, extra_code1);
std::move(quit_closure).Run();
},
runloop.QuitClosure()));
runloop.Run();
}
#endif // BUILDFLAG(IS_WIN) || BUILDFLAG(IS_MAC)
#if BUILDFLAG(IS_WIN)
TEST_F(RecoveryImprovedActionHandlerTest, HandleSuccess) {
unzip::SetUnzipperLaunchOverrideForTesting(
base::BindRepeating(&unzip::LaunchInProcessUnzipper));
// Tests that the recovery program runs and it returns an expected value.
static constexpr char kActionRunFileName[] = "ChromeRecovery.crx3";
ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
const base::FilePath from_path =
update_client::GetTestFilePath(kActionRunFileName);
const base::FilePath to_path =
temp_dir_.GetPath().AppendASCII(kActionRunFileName);
ASSERT_TRUE(base::CopyFile(from_path, to_path));
base::RunLoop runloop;
base::MakeRefCounted<TestActionHandler>()->Handle(
to_path, "some-session-id",
base::BindOnce(
[](base::OnceClosure quit_closure, bool succeeded, int error_code,
int extra_code1) {
EXPECT_TRUE(succeeded);
EXPECT_EQ(1877345072, error_code);
EXPECT_EQ(0, extra_code1);
std::move(quit_closure).Run();
},
runloop.QuitClosure()));
{
// For some reason, the task which runs the wait for the recovery EXE
// execution is handled with some delay. This causes the run loop to
// fail with a timeout.
const base::test::ScopedRunLoopTimeout specific_timeout(FROM_HERE,
base::Seconds(60));
runloop.Run();
}
}
#endif // BUILDFLAG(IS_WIN)
} // namespace component_updater
|
