Move SSL layer initialization into ChannelAuthenticator implementations.

remoting/protocol/channel_authenticator.h

 // Copyright (c) 2011 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.
 
 #ifndef REMOTING_PROTOCOL_CHANNEL_AUTHENTICATOR_H_
 #define REMOTING_PROTOCOL_CHANNEL_AUTHENTICATOR_H_
 
 #include <string>
 
 #include "base/callback.h"
-#include "base/memory/ref_counted.h"
 #include "base/threading/non_thread_safe.h"
-#include "net/base/completion_callback.h"
+#include "net/base/net_errors.h"
 
 namespace net {
-class DrainableIOBuffer;
-class GrowableIOBuffer;
-class SSLSocket;
+class StreamSocket;
 }  // namespace net
 
 namespace remoting {
 namespace protocol {
 
 class ChannelAuthenticator : public base::NonThreadSafe {

[Wez, 2011/11/22 22:29:48]

Should this interface really derive from NonThreadSafe?

[Wez, 2011/11/22 22:29:48]

Give a short description of what this interface is for?

[Sergey Ulanov, 2011/11/23 01:23:42]

Done.

[Sergey Ulanov, 2011/11/23 01:23:42]

Done.

  public:
-  enum Result {
-    SUCCESS,
-    FAILURE,
-  };
+  typedef base::Callback<void(net::Error error, net::StreamSocket*)>
+      DoneCallback;
 
-  typedef base::Callback<void(Result)> DoneCallback;
+  ChannelAuthenticator() {}

[Wez, 2011/11/22 22:29:48]

Make this protected, since it doesn't make sense to instantiate one of these
directly?

[Sergey Ulanov, 2011/11/23 01:23:42]

Just removed this constructor.

+  virtual ~ChannelAuthenticator() {}
 
-  ChannelAuthenticator() { }
-  virtual ~ChannelAuthenticator() { }
-
-  // Starts authentication of the |socket|. |done_callback| is called
-  // when authentication is finished. Caller retains ownership of
-  // |socket|. |shared_secret| is a shared secret that we use to
-  // authenticate the channel.
-  virtual void Authenticate(net::SSLSocket* socket,
-                            const DoneCallback& done_callback) = 0;
+  // Starts SSL on the |socket| and authenticates it. Takes ownership

[Wez, 2011/11/22 22:29:48]

Indicate that the caller must not attempt to use |socket| after calling
SecureAndAuthenticate; they can only use a newly-returned StreamSocket, if the
call succeeds.

[Sergey Ulanov, 2011/11/23 01:23:42]

Done.

+  // of |socket|. |done_callback| is called when authentication is

[Wez, 2011/11/22 22:29:48]

Is this always the case?  Surely whether SSL is started is dependent on the
implementation?  The interface just promises that encryption will be set up and
the authentication protocol executed?

[Sergey Ulanov, 2011/11/23 01:23:42]

Done.

+  // finished. Callback handler must take ownership of the result.

[Wez, 2011/11/22 22:29:48]

Indicate that the callback may be invoked before SecureAndAuthenticate returns.

[Sergey Ulanov, 2011/11/23 01:23:42]

Done.

+  virtual void SecureAndAuthenticate(
+      net::StreamSocket* socket, const DoneCallback& done_callback) = 0;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(ChannelAuthenticator);
 };
 
-class HostChannelAuthenticator : public ChannelAuthenticator {
- public:
-  HostChannelAuthenticator(const std::string& shared_secret);
-  virtual ~HostChannelAuthenticator();
-
-  // ChannelAuthenticator overrides.
-  virtual void Authenticate(net::SSLSocket* socket,
-                            const DoneCallback& done_callback) OVERRIDE;
-
- private:
-  void DoAuthRead();
-  void OnAuthBytesRead(int result);
-  bool HandleAuthBytesRead(int result);
-  bool VerifyAuthBytes(const std::string& received_auth_bytes);
-
-  std::string shared_secret_;
-  std::string auth_bytes_;
-  net::SSLSocket* socket_;
-  DoneCallback done_callback_;
-
-  scoped_refptr<net::GrowableIOBuffer> auth_read_buf_;
-
-  net::OldCompletionCallbackImpl<HostChannelAuthenticator> auth_read_callback_;
-
-  DISALLOW_COPY_AND_ASSIGN(HostChannelAuthenticator);
-};
-
-class ClientChannelAuthenticator : public ChannelAuthenticator {
- public:
-  ClientChannelAuthenticator(const std::string& shared_secret);
-  virtual ~ClientChannelAuthenticator();
-
-  // ChannelAuthenticator overrides.
-  virtual void Authenticate(net::SSLSocket* socket,
-                            const DoneCallback& done_callback);
-
- private:
-  void DoAuthWrite();
-  void OnAuthBytesWritten(int result);
-  bool HandleAuthBytesWritten(int result);
-
-  std::string shared_secret_;
-  net::SSLSocket* socket_;
-  DoneCallback done_callback_;
-
-  scoped_refptr<net::DrainableIOBuffer> auth_write_buf_;
-
-  net::OldCompletionCallbackImpl<ClientChannelAuthenticator>
-      auth_write_callback_;
-
-  DISALLOW_COPY_AND_ASSIGN(ClientChannelAuthenticator);
-};
-
 }  // namespace protocol
 }  // namespace remoting
 
 #endif  // REMOTING_PROTOCOL_CHANNEL_AUTHENTICATOR_H_