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 RejectionReason rejection_reason() const OVERRIDE; 42 virtual void ProcessMessage(const buzz::XmlElement* message, 43 const base::Closure& resume_callback) OVERRIDE; 44 virtual scoped_ptr<buzz::XmlElement> GetNextMessage() OVERRIDE; 45 virtual scoped_ptr<ChannelAuthenticator> 46 CreateChannelAuthenticator() const OVERRIDE; 47 48 private: 49 FRIEND_TEST_ALL_PREFIXES(V2AuthenticatorTest, InvalidSecret); 50 51 V2Authenticator(crypto::P224EncryptedKeyExchange::PeerType type, 52 const std::string& shared_secret, 53 State initial_state); 54 55 virtual void ProcessMessageInternal(const buzz::XmlElement* message); 56 57 bool is_host_side() const; 58 59 // Used only for host authenticators. 60 std::string local_cert_; 61 scoped_refptr<RsaKeyPair> local_key_pair_; 62 bool certificate_sent_; 63 64 // Used only for client authenticators. 65 std::string remote_cert_; 66 67 // Used for both host and client authenticators. 68 crypto::P224EncryptedKeyExchange key_exchange_impl_; 69 State state_; 70 RejectionReason rejection_reason_; 71 std::queue<std::string> pending_messages_; 72 std::string auth_key_; 73 74 DISALLOW_COPY_AND_ASSIGN(V2Authenticator); 75 }; 76 77 } // namespace protocol 78 } // namespace remoting 79 80 #endif // REMOTING_PROTOCOL_V2_AUTHENTICATOR_H_ 81