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
|
// Copyright 2017 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/protocol/data_channel_manager.h"
#include <utility>
#include "base/check.h"
#include "base/logging.h"
#include "remoting/protocol/message_pipe.h"
namespace remoting::protocol {
DataChannelManager::DataChannelManager() = default;
DataChannelManager::~DataChannelManager() = default;
void DataChannelManager::RegisterCreateHandlerCallback(
const std::string& prefix,
CreateHandlerCallback constructor) {
DCHECK(!prefix.empty());
DCHECK(constructor);
DCHECK(!is_registration_complete_);
constructors_.emplace_back(prefix, std::move(constructor));
}
void DataChannelManager::OnRegistrationComplete() {
DCHECK(!is_registration_complete_);
is_registration_complete_ = true;
for (auto& pending_channel : pending_data_channels_) {
OnIncomingDataChannel(pending_channel.first,
std::move(pending_channel.second));
}
pending_data_channels_.clear();
}
void DataChannelManager::OnIncomingDataChannel(
const std::string& name,
std::unique_ptr<MessagePipe> pipe) {
for (auto& constructor : constructors_) {
if (name.find(constructor.first) == 0) {
constructor.second.Run(name, std::move(pipe));
return;
}
}
if (!is_registration_complete_) {
VLOG(1) << "Handler for " << name << " has not been registered. "
<< "Will try again later.";
pending_data_channels_.emplace_back(name, std::move(pipe));
} else {
LOG(WARNING) << "Channel " << name
<< " ignored due to no matching handler.";
}
}
} // namespace remoting::protocol
|
