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_ 83