src/remoting/protocol/v2_authenticator.h

   1 // Copyright (c) 2012 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 #ifndef REMOTING_PROTOCOL_V2_AUTHENTICATOR_H_
   6 #define REMOTING_PROTOCOL_V2_AUTHENTICATOR_H_
   7
   8 #include <string>
   9 #include <queue>
  10
  11 #include "base/compiler_specific.h"
  12 #include "base/gtest_prod_util.h"
  13 #include "base/memory/scoped_ptr.h"
  14 #include "crypto/p224_spake.h"
  15 #include "remoting/protocol/authenticator.h"
  16
  17 namespace remoting {
  18
  19 class RsaKeyPair;
  20
  21 namespace protocol {
  22
  23 class V2Authenticator : public Authenticator {
  24  public:
  25   static bool IsEkeMessage(const buzz::XmlElement* message);
  26
  27   static scoped_ptr<Authenticator> CreateForClient(
  28       const std::string& shared_secret,
  29       State initial_state);
  30
  31   static scoped_ptr<Authenticator> CreateForHost(
  32       const std::string& local_cert,
  33       scoped_refptr<RsaKeyPair> key_pair,
  34       const std::string& shared_secret,
  35       State initial_state);
  36
  37   virtual ~V2Authenticator();
  38
  39   // Authenticator interface.
  40   virtual State state() const OVERRIDE;
  41   virtual bool started() const OVERRIDE;
  42   virtual RejectionReason rejection_reason() const OVERRIDE;
  43   virtual void ProcessMessage(const buzz::XmlElement* message,
  44                               const base::Closure& resume_callback) OVERRIDE;
  45   virtual scoped_ptr<buzz::XmlElement> GetNextMessage() OVERRIDE;
  46   virtual scoped_ptr<ChannelAuthenticator>
  47       CreateChannelAuthenticator() const OVERRIDE;
  48
  49  private:
  50   FRIEND_TEST_ALL_PREFIXES(V2AuthenticatorTest, InvalidSecret);
  51
  52   V2Authenticator(crypto::P224EncryptedKeyExchange::PeerType type,
  53                   const std::string& shared_secret,
  54                   State initial_state);
  55
  56   virtual void ProcessMessageInternal(const buzz::XmlElement* message);
  57
  58   bool is_host_side() const;
  59
  60   // Used only for host authenticators.
  61   std::string local_cert_;
  62   scoped_refptr<RsaKeyPair> local_key_pair_;
  63   bool certificate_sent_;
  64
  65   // Used only for client authenticators.
  66   std::string remote_cert_;
  67
  68   // Used for both host and client authenticators.
  69   crypto::P224EncryptedKeyExchange key_exchange_impl_;
  70   State state_;
  71   bool started_;
  72   RejectionReason rejection_reason_;
  73   std::queue<std::string> pending_messages_;
  74   std::string auth_key_;
  75
  76   DISALLOW_COPY_AND_ASSIGN(V2Authenticator);
  77 };
  78
  79 }  // namespace protocol
  80 }  // namespace remoting
  81
  82 #endif  // REMOTING_PROTOCOL_V2_AUTHENTICATOR_H_