Home | History | Annotate | Download | only in protocol 
      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