Chromium Code Reviews| Index: remoting/protocol/authenticated_channel_factory.cc |
| diff --git a/remoting/protocol/authenticated_channel_factory.cc b/remoting/protocol/authenticated_channel_factory.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..aee455fbe537000a8b608bdbd9b46f988c09f2c7 |
| --- /dev/null |
| +++ b/remoting/protocol/authenticated_channel_factory.cc |
| @@ -0,0 +1,91 @@ |
| +// 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/authenticated_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 { |
| + |
| +AuthenticatedChannelFactory::AuthenticatedChannelFactory( |
| + ChannelFactory* channel_factory, |
| + Authenticator* authenticator) |
| + : channel_factory_(channel_factory), |
| + authenticator_(authenticator) { |
| + DCHECK_EQ(authenticator_->state(), Authenticator::ACCEPTED); |
| +} |
| + |
| +AuthenticatedChannelFactory::~AuthenticatedChannelFactory() { |
| + // CancelChannelCreation() is expected to be called before destruction. |
|
Wez
2014/09/10 02:29:25
Is that a part of the ChannelFactory interface con
Sergey Ulanov
2014/09/10 21:50:57
I think it's still useful to make it clear what th
Wez
2014/09/10 22:19:41
You have an equivalent check in the PseudoTcpChann
Sergey Ulanov
2014/09/10 22:49:30
Done.
|
| + DCHECK(channels_.empty()); |
| +} |
| + |
| +void AuthenticatedChannelFactory::CreateChannel( |
| + const std::string& name, |
| + const ChannelCreatedCallback& callback) { |
| + DCHECK(!callback.is_null()); |
| + channel_factory_->CreateChannel( |
| + name, |
| + base::Bind(&AuthenticatedChannelFactory::OnBaseChannelCreated, |
| + base::Unretained(this), name, callback)); |
| +} |
| + |
| +void AuthenticatedChannelFactory::CancelChannelCreation( |
| + const std::string& name) { |
| + ChannelsMap::iterator it = channels_.find(name); |
| + if (it == channels_.end()) { |
| + channel_factory_->CancelChannelCreation(name); |
| + } else { |
| + delete it->second; |
| + channels_.erase(it); |
| + } |
| +} |
| + |
| +void AuthenticatedChannelFactory::OnBaseChannelCreated( |
| + const std::string& name, |
| + const ChannelCreatedCallback& callback, |
| + scoped_ptr<net::StreamSocket> socket) { |
| + if (!socket) { |
| + callback.Run(socket.Pass()); |
|
Wez
2014/09/10 02:29:25
It reads oddly to socket.Pass() only if socket is
Sergey Ulanov
2014/09/10 21:50:57
yes, but it's more to type :(. Done.
|
| + return; |
| + } |
| + |
| + scoped_ptr<ChannelAuthenticator> channel_authenticator = |
| + authenticator_->CreateChannelAuthenticator(); |
| + ChannelAuthenticator* channel_authenticator_ptr = |
| + channel_authenticator.release(); |
|
Wez
2014/09/10 02:29:25
nit: Can you just .release() the CreateChannelAuth
Sergey Ulanov
2014/09/10 21:50:57
Done.
|
| + channels_[name] = channel_authenticator_ptr; |
| + channel_authenticator_ptr->SecureAndAuthenticate( |
| + socket.Pass(), |
| + base::Bind(&AuthenticatedChannelFactory::OnAuthenticated, |
| + base::Unretained(this), name, callback)); |
| +} |
| + |
| +void AuthenticatedChannelFactory::OnAuthenticated( |
| + const std::string& name, |
| + const ChannelCreatedCallback& callback, |
| + net::Error error, |
| + scoped_ptr<net::StreamSocket> socket) { |
| + scoped_ptr<ChannelAuthenticator> authenticator; |
| + |
| + ChannelsMap::iterator it = channels_.find(name); |
| + DCHECK(it != channels_.end()); |
| + authenticator.reset(it->second); |
|
Wez
2014/09/10 02:29:25
Why not just delete it->second, like you do in the
Sergey Ulanov
2014/09/10 21:50:57
Done.
|
| + channels_.erase(it); |
| + |
| + DCHECK(error == net::OK || socket); |
|
Wez
2014/09/10 02:29:25
Do you mean (socket || error != net::OK), i.e. tha
Sergey Ulanov
2014/09/10 21:50:57
Yes. Replaced it with a stricter DCHECK and fixed
|
| + if (error != net::OK) { |
| + socket.reset(); |
| + } |
| + |
| + callback.Run(socket.Pass()); |
| + |
|
Wez
2014/09/10 02:29:26
nit: Lose this blank line.
Sergey Ulanov
2014/09/10 21:50:57
Done.
|
| +} |
| + |
| +} // namespace protocol |
| +} // namespace remoting |