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
|
// 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 "chrome/updater/action_handler.h"
#include <string>
#include <utility>
#include "base/check.h"
#include "base/command_line.h"
#include "base/files/file_path.h"
#include "base/functional/bind.h"
#include "base/functional/callback.h"
#include "base/logging.h"
#include "base/memory/scoped_refptr.h"
#include "base/process/launch.h"
#include "base/process/process.h"
#include "base/task/sequenced_task_runner.h"
#include "base/task/thread_pool.h"
#include "base/time/time.h"
#include "components/update_client/update_client.h"
namespace updater {
class ActionHandler : public update_client::ActionHandler {
public:
ActionHandler() = default;
ActionHandler(const ActionHandler&) = delete;
ActionHandler& operator=(const ActionHandler&) = delete;
private:
using Result =
std::tuple<bool /*succeeded*/, int /*error_code*/, int /*extra_code*/>;
~ActionHandler() override = default;
// Overrides for update_client::ActionHandler.
void Handle(const base::FilePath& action,
const std::string& session_id,
Callback callback) override;
// For Windows, the action is a path to an EXE file.
static Result RunCommand(const base::FilePath& exe_path);
};
void ActionHandler::Handle(const base::FilePath& action,
const std::string&,
Callback callback) {
CHECK(!action.empty());
base::ThreadPool::PostTaskAndReplyWithResult(
FROM_HERE,
{base::MayBlock(), base::WithBaseSyncPrimitives(),
base::TaskPriority::BEST_EFFORT,
base::TaskShutdownBehavior::SKIP_ON_SHUTDOWN},
base::BindOnce(&ActionHandler::RunCommand, action),
base::BindOnce(
[](Callback callback, const Result& result) {
auto [succeeded, error_code, extra_code] = result;
base::SequencedTaskRunner::GetCurrentDefault()->PostTask(
FROM_HERE, base::BindOnce(std::move(callback), succeeded,
error_code, extra_code));
},
std::move(callback)));
}
ActionHandler::Result ActionHandler::RunCommand(
const base::FilePath& exe_path) {
base::CommandLine command_line(exe_path);
VLOG(1) << "run command: " << command_line.GetCommandLineString();
base::LaunchOptions options;
options.start_hidden = true;
base::Process process = base::LaunchProcess(command_line, options);
int exit_code = 0;
const base::TimeDelta kMaxWaitTime = base::Seconds(600);
const bool succeeded = process.IsValid() && process.WaitForExitWithTimeout(
kMaxWaitTime, &exit_code);
return Result{succeeded, exit_code, 0};
}
scoped_refptr<update_client::ActionHandler> MakeActionHandler() {
return base::MakeRefCounted<ActionHandler>();
}
} // namespace updater
|
