Index: remoting/protocol/secure_channel_factory.cc |
diff --git a/remoting/protocol/secure_channel_factory.cc b/remoting/protocol/secure_channel_factory.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..df98378afc8fdb155371b275e5111e4928d21364 |
--- /dev/null |
+++ b/remoting/protocol/secure_channel_factory.cc |
@@ -0,0 +1,83 @@ |
+// 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/protocol/secure_channel_factory.h" |
+ |
+#include "base/bind.h" |
+#include "net/socket/stream_socket.h" |
+#include "remoting/protocol/authenticator.h" |
+#include "remoting/protocol/channel_authenticator.h" |
+ |
+namespace remoting { |
+namespace protocol { |
+ |
+SecureChannelFactory::SecureChannelFactory( |
+ StreamChannelFactory* channel_factory, |
+ Authenticator* authenticator) |
+ : channel_factory_(channel_factory), |
+ authenticator_(authenticator) { |
+ DCHECK_EQ(authenticator_->state(), Authenticator::ACCEPTED); |
+} |
+ |
+SecureChannelFactory::~SecureChannelFactory() { |
+ // CancelChannelCreation() is expected to be called before destruction. |
+ DCHECK(channel_authenticators_.empty()); |
+} |
+ |
+void SecureChannelFactory::CreateChannel( |
+ const std::string& name, |
+ const ChannelCreatedCallback& callback) { |
+ DCHECK(!callback.is_null()); |
+ channel_factory_->CreateChannel( |
+ name, |
+ base::Bind(&SecureChannelFactory::OnBaseChannelCreated, |
+ base::Unretained(this), name, callback)); |
+} |
+ |
+void SecureChannelFactory::CancelChannelCreation( |
+ const std::string& name) { |
+ AuthenticatorMap::iterator it = channel_authenticators_.find(name); |
+ if (it == channel_authenticators_.end()) { |
+ channel_factory_->CancelChannelCreation(name); |
+ } else { |
+ delete it->second; |
+ channel_authenticators_.erase(it); |
+ } |
+} |
+ |
+void SecureChannelFactory::OnBaseChannelCreated( |
+ const std::string& name, |
+ const ChannelCreatedCallback& callback, |
+ scoped_ptr<net::StreamSocket> socket) { |
+ if (!socket) { |
+ callback.Run(scoped_ptr<net::StreamSocket>()); |
+ return; |
+ } |
+ |
+ ChannelAuthenticator* channel_authenticator = |
+ authenticator_->CreateChannelAuthenticator().release(); |
+ channel_authenticators_[name] = channel_authenticator; |
+ channel_authenticator->SecureAndAuthenticate( |
+ socket.Pass(), |
+ base::Bind(&SecureChannelFactory::OnSecureChannelCreated, |
+ base::Unretained(this), name, callback)); |
+} |
+ |
+void SecureChannelFactory::OnSecureChannelCreated( |
+ const std::string& name, |
+ const ChannelCreatedCallback& callback, |
+ int error, |
+ scoped_ptr<net::StreamSocket> socket) { |
+ DCHECK((socket && error == net::OK) || (!socket && error != net::OK)); |
+ |
+ AuthenticatorMap::iterator it = channel_authenticators_.find(name); |
+ DCHECK(it != channel_authenticators_.end()); |
+ delete it->second; |
+ channel_authenticators_.erase(it); |
+ |
+ callback.Run(socket.Pass()); |
+} |
+ |
+} // namespace protocol |
+} // namespace remoting |