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
|
// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/openscreen_platform/message_port_tls_connection.h"
#include <string_view>
#include "third_party/openscreen/src/platform/api/task_runner.h"
#include "third_party/openscreen/src/platform/base/error.h"
namespace openscreen_platform {
MessagePortTlsConnection::MessagePortTlsConnection(
std::unique_ptr<cast_api_bindings::MessagePort> message_port,
openscreen::TaskRunner& task_runner)
: message_port_(std::move(message_port)), task_runner_(task_runner) {
DCHECK(message_port_);
message_port_->SetReceiver(this);
}
MessagePortTlsConnection::~MessagePortTlsConnection() = default;
// TlsConnection overrides.
void MessagePortTlsConnection::SetClient(TlsConnection::Client* client) {
DCHECK(task_runner_->IsRunningOnTaskRunner());
client_ = client;
}
bool MessagePortTlsConnection::Send(openscreen::ByteView data) {
return message_port_->PostMessage(std::string_view(
reinterpret_cast<const char*>(data.data()), data.size()));
}
openscreen::IPEndpoint MessagePortTlsConnection::GetRemoteEndpoint() const {
return openscreen::IPEndpoint{openscreen::IPAddress::kV6LoopbackAddress()};
}
bool MessagePortTlsConnection::OnMessage(
std::string_view message,
std::vector<std::unique_ptr<cast_api_bindings::MessagePort>> ports) {
DCHECK(ports.empty());
if (client_) {
if (!task_runner_->IsRunningOnTaskRunner()) {
task_runner_->PostTask([ptr = weak_ptr_factory_.GetWeakPtr(), message]() {
if (ptr) {
ptr->OnMessage(
message,
std::vector<std::unique_ptr<cast_api_bindings::MessagePort>>());
}
});
return true;
}
client_->OnRead(this, std::vector<uint8_t>(message.begin(), message.end()));
}
return true;
}
void MessagePortTlsConnection::OnPipeError() {
if (client_) {
if (!task_runner_->IsRunningOnTaskRunner()) {
task_runner_->PostTask([ptr = weak_ptr_factory_.GetWeakPtr()]() {
if (ptr) {
ptr->OnPipeError();
}
});
return;
}
client_->OnError(
this, openscreen::Error(openscreen::Error::Code::kSocketFailure));
}
}
} // namespace openscreen_platform
|
