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
|
// Copyright 2014 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/host_extension_session_manager.h"
#include "base/check.h"
#include "remoting/base/capabilities.h"
#include "remoting/host/client_session_details.h"
#include "remoting/host/host_extension.h"
#include "remoting/host/host_extension_session.h"
namespace remoting {
HostExtensionSessionManager::HostExtensionSessionManager(
const HostExtensions& extensions,
ClientSessionDetails* client_session_details)
: client_session_details_(client_session_details),
client_stub_(nullptr),
extensions_(extensions) {}
HostExtensionSessionManager::~HostExtensionSessionManager() = default;
std::string HostExtensionSessionManager::GetCapabilities() const {
std::string capabilities;
for (remoting::HostExtension* extension : extensions_) {
const std::string& capability = extension->capability();
if (capability.empty()) {
continue;
}
if (!capabilities.empty()) {
capabilities.append(" ");
}
capabilities.append(capability);
}
return capabilities;
}
HostExtensionSession* HostExtensionSessionManager::FindExtensionSession(
const std::string& capability) {
auto iter = extension_sessions_.find(capability);
if (iter == extension_sessions_.end()) {
return nullptr;
}
return iter->second.get();
}
void HostExtensionSessionManager::OnNegotiatedCapabilities(
protocol::ClientStub* client_stub,
const std::string& capabilities) {
DCHECK(client_stub);
DCHECK(!client_stub_);
client_stub_ = client_stub;
for (remoting::HostExtension* extension : extensions_) {
// If the extension requires a capability that was not negotiated then do
// not instantiate it.
if (!extension->capability().empty() &&
!HasCapability(capabilities, extension->capability())) {
continue;
}
std::unique_ptr<HostExtensionSession> extension_session =
extension->CreateExtensionSession(client_session_details_,
client_stub_);
DCHECK(extension_session);
extension_sessions_.emplace(extension->capability(),
std::move(extension_session));
}
}
bool HostExtensionSessionManager::OnExtensionMessage(
const protocol::ExtensionMessage& message) {
for (const auto& [capability, session] : extension_sessions_) {
if (session->OnExtensionMessage(client_session_details_, client_stub_,
message)) {
return true;
}
}
return false;
}
} // namespace remoting
|
