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.
5 #ifndef REMOTING_PROTOCOL_FAKE_AUTHENTICATOR_H_
6 #define REMOTING_PROTOCOL_FAKE_AUTHENTICATOR_H_
8 #include "base/callback.h"
9 #include "base/memory/weak_ptr.h"
10 #include "remoting/protocol/authenticator.h"
11 #include "remoting/protocol/channel_authenticator.h"
16 class FakeChannelAuthenticator : public ChannelAuthenticator {
18 FakeChannelAuthenticator(bool accept, bool async);
19 virtual ~FakeChannelAuthenticator();
21 // ChannelAuthenticator interface.
22 virtual void SecureAndAuthenticate(
23 scoped_ptr<net::StreamSocket> socket,
24 const DoneCallback& done_callback) OVERRIDE;
29 scoped_ptr<net::StreamSocket> socket);
31 void OnAuthBytesWritten(int result);
32 void OnAuthBytesRead(int result);
37 scoped_ptr<net::StreamSocket> socket_;
38 DoneCallback done_callback_;
41 bool did_write_bytes_;
43 base::WeakPtrFactory<FakeChannelAuthenticator> weak_factory_;
45 DISALLOW_COPY_AND_ASSIGN(FakeChannelAuthenticator);
48 class FakeAuthenticator : public Authenticator {
61 FakeAuthenticator(Type type, int round_trips, Action action, bool async);
63 virtual ~FakeAuthenticator();
65 // Set the number of messages that the authenticator needs to process before
66 // started() returns true. Default to 0.
67 void set_messages_till_started(int messages);
69 // Authenticator interface.
70 virtual State state() const OVERRIDE;
71 virtual bool started() const OVERRIDE;
72 virtual RejectionReason rejection_reason() const OVERRIDE;
73 virtual void ProcessMessage(const buzz::XmlElement* message,
74 const base::Closure& resume_callback) OVERRIDE;
75 virtual scoped_ptr<buzz::XmlElement> GetNextMessage() OVERRIDE;
76 virtual scoped_ptr<ChannelAuthenticator>
77 CreateChannelAuthenticator() const OVERRIDE;
85 // Total number of messages that have been processed.
87 // Number of messages that the authenticator needs to process before started()
88 // returns true. Default to 0.
89 int messages_till_started_;
91 DISALLOW_COPY_AND_ASSIGN(FakeAuthenticator);
94 class FakeHostAuthenticatorFactory : public AuthenticatorFactory {
96 FakeHostAuthenticatorFactory(
97 int round_trips, int messages_till_start,
98 FakeAuthenticator::Action action, bool async);
99 virtual ~FakeHostAuthenticatorFactory();
101 // AuthenticatorFactory interface.
102 virtual scoped_ptr<Authenticator> CreateAuthenticator(
103 const std::string& local_jid,
104 const std::string& remote_jid,
105 const buzz::XmlElement* first_message) OVERRIDE;
109 int messages_till_started_;
110 FakeAuthenticator::Action action_;
113 DISALLOW_COPY_AND_ASSIGN(FakeHostAuthenticatorFactory);
116 } // namespace protocol
117 } // namespace remoting
119 #endif // REMOTING_PROTOCOL_FAKE_AUTHENTICATOR_H_