remoting/host/host_extension_session_manager.cc

// Copyright 2014 The Chromium Authors. All rights reserved.
// 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 "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 std::vector<HostExtension*>& extensions,
    ClientSessionDetails* client_session_details)
    : client_session_details_(client_session_details),
      client_stub_(nullptr),
      extensions_(extensions) {}

HostExtensionSessionManager::~HostExtensionSessionManager() {}

std::string HostExtensionSessionManager::GetCapabilities() const {
  std::string capabilities;
  for (HostExtensions::const_iterator extension = extensions_.begin();
       extension != extensions_.end(); ++extension) {
    const std::string& capability = (*extension)->capability();
    if (capability.empty()) {
      continue;
    }
    if (!capabilities.empty()) {
      capabilities.append(" ");
    }
    capabilities.append(capability);
  }
  return capabilities;
}

void HostExtensionSessionManager::OnNegotiatedCapabilities(
    protocol::ClientStub* client_stub,
    const std::string& capabilities) {
  DCHECK(client_stub);
  DCHECK(!client_stub_);

  client_stub_ = client_stub;

  for (HostExtensions::const_iterator extension = extensions_.begin();
       extension != extensions_.end(); ++extension) {
    // 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_.push_back(extension_session.release());
  }
}

bool HostExtensionSessionManager::OnExtensionMessage(
    const protocol::ExtensionMessage& message) {
  for(HostExtensionSessions::const_iterator it = extension_sessions_.begin();
      it != extension_sessions_.end(); ++it) {
    if ((*it)->OnExtensionMessage(client_session_details_, client_stub_,
                                  message)) {
      return true;
    }
  }
  return false;
}

}  // namespace remoting