src/remoting/protocol/secure_channel_factory.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "remoting/protocol/secure_channel_factory.h"
   6
   7 #include "base/bind.h"
   8 #include "net/socket/stream_socket.h"
   9 #include "remoting/protocol/authenticator.h"
  10 #include "remoting/protocol/channel_authenticator.h"
  11
  12 namespace remoting {
  13 namespace protocol {
  14
  15 SecureChannelFactory::SecureChannelFactory(
  16     StreamChannelFactory* channel_factory,
  17     Authenticator* authenticator)
  18     : channel_factory_(channel_factory),
  19       authenticator_(authenticator) {
  20   DCHECK_EQ(authenticator_->state(), Authenticator::ACCEPTED);
  21 }
  22
  23 SecureChannelFactory::~SecureChannelFactory() {
  24   // CancelChannelCreation() is expected to be called before destruction.
  25   DCHECK(channel_authenticators_.empty());
  26 }
  27
  28 void SecureChannelFactory::CreateChannel(
  29     const std::string& name,
  30     const ChannelCreatedCallback& callback) {
  31   DCHECK(!callback.is_null());
  32   channel_factory_->CreateChannel(
  33       name,
  34       base::Bind(&SecureChannelFactory::OnBaseChannelCreated,
  35                  base::Unretained(this), name, callback));
  36 }
  37
  38 void SecureChannelFactory::CancelChannelCreation(
  39     const std::string& name) {
  40   AuthenticatorMap::iterator it = channel_authenticators_.find(name);
  41   if (it == channel_authenticators_.end()) {
  42     channel_factory_->CancelChannelCreation(name);
  43   } else {
  44     delete it->second;
  45     channel_authenticators_.erase(it);
  46   }
  47 }
  48
  49 void SecureChannelFactory::OnBaseChannelCreated(
  50     const std::string& name,
  51     const ChannelCreatedCallback& callback,
  52     scoped_ptr<net::StreamSocket> socket) {
  53   if (!socket) {
  54     callback.Run(scoped_ptr<net::StreamSocket>());
  55     return;
  56   }
  57
  58   ChannelAuthenticator* channel_authenticator =
  59       authenticator_->CreateChannelAuthenticator().release();
  60   channel_authenticators_[name] = channel_authenticator;
  61   channel_authenticator->SecureAndAuthenticate(
  62       socket.Pass(),
  63       base::Bind(&SecureChannelFactory::OnSecureChannelCreated,
  64                  base::Unretained(this), name, callback));
  65 }
  66
  67 void SecureChannelFactory::OnSecureChannelCreated(
  68     const std::string& name,
  69     const ChannelCreatedCallback& callback,
  70     int error,
  71     scoped_ptr<net::StreamSocket> socket) {
  72   DCHECK((socket && error == net::OK) || (!socket && error != net::OK));
  73
  74   AuthenticatorMap::iterator it = channel_authenticators_.find(name);
  75   DCHECK(it != channel_authenticators_.end());
  76   delete it->second;
  77   channel_authenticators_.erase(it);
  78
  79   callback.Run(socket.Pass());
  80 }
  81
  82 }  // namespace protocol
  83 }  // namespace remoting