Enable QUIC support in chromoting

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/base64.h"
 #include "base/callback_helpers.h"
 #include "base/message_loop/message_loop.h"
+#include "base/rand_util.h"
 #include "base/strings/string_number_conversions.h"
 #include "net/base/io_buffer.h"
 #include "net/base/net_errors.h"
 #include "remoting/base/constants.h"
 #include "remoting/protocol/p2p_stream_socket.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/webrtc/libjingle/xmllite/xmlelement.h"
 
 namespace remoting {
 namespace protocol {
@@ skipping 62 matching lines @@
   if (did_write_bytes_)
     CallDoneCallback();
 }
 
 void FakeChannelAuthenticator::CallDoneCallback() {
   if (result_ != net::OK)
     socket_.reset();
   base::ResetAndReturn(&done_callback_).Run(result_, socket_.Pass());
 }
 
-FakeAuthenticator::FakeAuthenticator(
-    Type type, int round_trips, Action action, bool async)
+FakeAuthenticator::FakeAuthenticator(Type type,
+                                     int round_trips,
+                                     Action action,
+                                     bool async)
     : type_(type),
       round_trips_(round_trips),
       action_(action),
       async_(async),
       messages_(0),
       messages_till_started_(0) {
 }
 
 FakeAuthenticator::~FakeAuthenticator() {
 }
@@ skipping 36 matching lines @@
   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"));
   EXPECT_EQ(id, base::IntToString(messages_));
+
+  // On the client receive the key in the last message.
+  if (type_ == CLIENT && messages_ == round_trips_ * 2 - 1) {
+    std::string key_base64 =
+        message->TextNamed(buzz::QName(kChromotingXmlNamespace, "key"));
+    EXPECT_TRUE(!key_base64.empty());
+    EXPECT_TRUE(base::Base64Decode(key_base64, &auth_key_));
+  }
+
   ++messages_;
   resume_callback.Run();
 }
 
 scoped_ptr<buzz::XmlElement> FakeAuthenticator::GetNextMessage() {
   EXPECT_EQ(MESSAGE_READY, state());
 
   scoped_ptr<buzz::XmlElement> result(new buzz::XmlElement(
       buzz::QName(kChromotingXmlNamespace, "authentication")));
   buzz::XmlElement* id = new buzz::XmlElement(
       buzz::QName(kChromotingXmlNamespace, "id"));
   id->AddText(base::IntToString(messages_));
   result->AddElement(id);
 
+  // Add authentication key in the last message sent from host to client.
+  if (type_ == HOST && messages_ == round_trips_ * 2 - 1) {
+    auth_key_ =  base::RandBytesAsString(16);
+    buzz::XmlElement* key = new buzz::XmlElement(
+        buzz::QName(kChromotingXmlNamespace, "key"));
+    std::string key_base64;
+    base::Base64Encode(auth_key_, &key_base64);
+    key->AddText(key_base64);
+    result->AddElement(key);
+  }
+
   ++messages_;
   return result.Pass();
 }
 
+const std::string& FakeAuthenticator::GetAuthKey() const {
+  EXPECT_EQ(ACCEPTED, state());
+  return auth_key_;
+}
+
 scoped_ptr<ChannelAuthenticator>
 FakeAuthenticator::CreateChannelAuthenticator() const {
   EXPECT_EQ(ACCEPTED, state());
   return make_scoped_ptr(
       new FakeChannelAuthenticator(action_ != REJECT_CHANNEL, async_));
 }
 
 FakeHostAuthenticatorFactory::FakeHostAuthenticatorFactory(
     int round_trips, int messages_till_started,
     FakeAuthenticator::Action action, bool async)
@@ skipping 12 matching lines @@
   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