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
|
// 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 "remoting/host/mac/agent_process_broker_client.h"
#include <memory>
#include "base/functional/bind.h"
#include "base/location.h"
#include "base/logging.h"
#include "components/named_mojo_ipc_server/named_mojo_ipc_server_client_util.h"
#include "mojo/public/cpp/platform/named_platform_channel.h"
#include "mojo/public/cpp/system/invitation.h"
#include "remoting/host/ipc_constants.h"
#include "remoting/host/mac/agent_process_broker_constants.h"
#include "remoting/host/mojom/agent_process_broker.mojom.h"
namespace remoting {
AgentProcessBrokerClient::AgentProcessBrokerClient(
base::OnceClosure on_termination_requested,
base::OnceClosure on_disconnected)
: on_termination_requested_(std::move(on_termination_requested)),
on_disconnected_(std::move(on_disconnected)) {}
AgentProcessBrokerClient::~AgentProcessBrokerClient() {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
}
bool AgentProcessBrokerClient::ConnectToServer() {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
return ConnectToServer(GetAgentProcessBrokerServerName());
}
bool AgentProcessBrokerClient::ConnectToServer(
const mojo::NamedPlatformChannel::ServerName& server_name) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
auto endpoint = named_mojo_ipc_server::ConnectToServer(server_name);
if (!endpoint.is_valid()) {
LOG(WARNING) << "Cannot connect to IPC through server name " << server_name
<< ". Endpoint is invalid.";
// This may happen if the broker process is somehow launched after the host
// process. The host process will exit and the host service process will
// relaunch the host process.
return false;
}
auto invitation = mojo::IncomingInvitation::Accept(std::move(endpoint));
auto message_pipe =
invitation.ExtractMessagePipe(kAgentProcessBrokerMessagePipeId);
mojo::PendingRemote<mojom::AgentProcessBroker> pending_remote(
std::move(message_pipe), /* version= */ 0);
if (!pending_remote.is_valid()) {
LOG(WARNING) << "Invalid message pipe.";
return false;
}
broker_remote_.Bind(std::move(pending_remote));
broker_remote_.set_disconnect_handler(base::BindOnce(
&AgentProcessBrokerClient::OnBrokerDisconnected, base::Unretained(this)));
return true;
}
void AgentProcessBrokerClient::OnAgentProcessLaunched(
mojom::AgentProcess* agent_process) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
DCHECK(broker_remote_.is_bound());
DCHECK(!agent_process_receiver_);
agent_process_receiver_ =
std::make_unique<mojo::Receiver<mojom::AgentProcess>>(agent_process);
broker_remote_->OnAgentProcessLaunched(
agent_process_receiver_->BindNewPipeAndPassRemote());
agent_process_receiver_->set_disconnect_with_reason_handler(base::BindOnce(
&AgentProcessBrokerClient::OnAgentProcessRemoteDisconnected,
base::Unretained(this)));
}
void AgentProcessBrokerClient::OnBrokerDisconnected() {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
LOG(ERROR) << "Broker process has disconnected.";
// This may happen if the broker process has crashed. The host process will
// exit and the host service process will relaunch it.
RunDisconnectedCallback();
}
void AgentProcessBrokerClient::OnAgentProcessRemoteDisconnected(
uint32_t custom_reason,
const std::string& description) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
if (custom_reason == kTerminateAgentProcessBrokerReason) {
LOG(WARNING) << "Agent process was requested to be terminated.";
if (on_termination_requested_) {
std::move(on_termination_requested_).Run();
}
return;
}
LOG(WARNING) << "Agent process remote has disconnected: " << description
<< " (" << custom_reason << ")";
RunDisconnectedCallback();
}
void AgentProcessBrokerClient::RunDisconnectedCallback() {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
if (on_disconnected_) {
std::move(on_disconnected_).Run();
}
}
} // namespace remoting
|
