[Draft] Fix canceling pin prompt causes host overload

remoting/protocol/fake_authenticator.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "remoting/protocol/fake_authenticator.h"
 
 #include "base/message_loop/message_loop.h"
 #include "base/strings/string_number_conversions.h"
 #include "net/base/io_buffer.h"
 #include "net/socket/stream_socket.h"
@@ skipping 66 matching lines @@
   if (did_write_bytes_)
     done_callback_.Run(result_, socket_.Pass());
 }
 
 FakeAuthenticator::FakeAuthenticator(
     Type type, int round_trips, Action action, bool async)
     : type_(type),
       round_trips_(round_trips),
       action_(action),
       async_(async),
-      messages_(0) {
+      messages_(0),
+      messages_till_started_(0) {
 }
 
 FakeAuthenticator::~FakeAuthenticator() {
 }
 
+void FakeAuthenticator::set_messages_till_started(int messages) {
+  messages_till_started_ = messages;
+}
+
 Authenticator::State FakeAuthenticator::state() const {
   EXPECT_LE(messages_, round_trips_ * 2);
   if (messages_ >= round_trips_ * 2) {
     if (action_ == REJECT) {
       return REJECTED;
     } else {
       return ACCEPTED;
     }
   }
 
   // Don't send the last message if this is a host that wants to
   // reject a connection.
   if (messages_ == round_trips_ * 2 - 1 &&
       type_ == HOST && action_ == REJECT) {
     return REJECTED;
   }
 
   // We are not done yet. process next message.
   if ((messages_ % 2 == 0 && type_ == CLIENT) ||
       (messages_ % 2 == 1 && type_ == HOST)) {
     return MESSAGE_READY;
   } else {
     return WAITING_MESSAGE;
   }
 }
 
+bool FakeAuthenticator::started() const {
+  return messages_ > messages_till_started_;
+}
+
 Authenticator::RejectionReason FakeAuthenticator::rejection_reason() const {
   EXPECT_EQ(REJECTED, state());
   return INVALID_CREDENTIALS;
 }
 
 void FakeAuthenticator::ProcessMessage(const buzz::XmlElement* message,
                                        const base::Closure& resume_callback) {
   EXPECT_EQ(WAITING_MESSAGE, state());
   std::string id =
       message->TextNamed(buzz::QName(kChromotingXmlNamespace, "id"));
@@ skipping 17 matching lines @@
 }
 
 scoped_ptr<ChannelAuthenticator>
 FakeAuthenticator::CreateChannelAuthenticator() const {
   EXPECT_EQ(ACCEPTED, state());
   return scoped_ptr<ChannelAuthenticator>(
       new FakeChannelAuthenticator(action_ != REJECT_CHANNEL, async_));
 }
 
 FakeHostAuthenticatorFactory::FakeHostAuthenticatorFactory(
-    int round_trips, FakeAuthenticator::Action action, bool async)
+    int round_trips, int messages_till_started,
+    FakeAuthenticator::Action action, bool async)
     : round_trips_(round_trips),
+      messages_till_started_(messages_till_started),
       action_(action), async_(async) {
 }
 
 FakeHostAuthenticatorFactory::~FakeHostAuthenticatorFactory() {
 }
 
 scoped_ptr<Authenticator> FakeHostAuthenticatorFactory::CreateAuthenticator(
     const std::string& local_jid,
     const std::string& remote_jid,
     const buzz::XmlElement* first_message) {
-  return scoped_ptr<Authenticator>(new FakeAuthenticator(
-      FakeAuthenticator::HOST, round_trips_, action_, async_));
+  FakeAuthenticator* authenticator = new FakeAuthenticator(
+      FakeAuthenticator::HOST, round_trips_, action_, async_);
+  authenticator->set_messages_till_started(messages_till_started_);
+
+  scoped_ptr<Authenticator> result(authenticator);
+  return result.Pass();
 }
 
 }  // namespace protocol
 }  // namespace remoting