remoting/protocol/v2_authenticator_unittest.cc

   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 #include "remoting/protocol/v2_authenticator.h"
   6
   7 #include "base/bind.h"
   8 #include "net/base/net_errors.h"
   9 #include "remoting/base/rsa_key_pair.h"
  10 #include "remoting/protocol/authenticator_test_base.h"
  11 #include "remoting/protocol/channel_authenticator.h"
  12 #include "remoting/protocol/connection_tester.h"
  13 #include "testing/gmock/include/gmock/gmock.h"
  14 #include "testing/gtest/include/gtest/gtest.h"
  15 #include "third_party/webrtc/libjingle/xmllite/xmlelement.h"
  16
  17 using testing::_;
  18 using testing::DeleteArg;
  19 using testing::SaveArg;
  20
  21 namespace remoting {
  22 namespace protocol {
  23
  24 namespace {
  25
  26 const int kMessageSize = 100;
  27 const int kMessages = 1;
  28
  29 const char kTestSharedSecret[] = "1234-1234-5678";
  30 const char kTestSharedSecretBad[] = "0000-0000-0001";
  31
  32 }  // namespace
  33
  34 class V2AuthenticatorTest : public AuthenticatorTestBase {
  35  public:
  36   V2AuthenticatorTest() {
  37   }
  38   ~V2AuthenticatorTest() override {}
  39
  40  protected:
  41   void InitAuthenticators(const std::string& client_secret,
  42                           const std::string& host_secret) {
  43     host_ = V2Authenticator::CreateForHost(
  44         host_cert_, key_pair_, host_secret,
  45         Authenticator::WAITING_MESSAGE);
  46     client_ = V2Authenticator::CreateForClient(
  47         client_secret, Authenticator::MESSAGE_READY);
  48   }
  49
  50   DISALLOW_COPY_AND_ASSIGN(V2AuthenticatorTest);
  51 };
  52
  53 TEST_F(V2AuthenticatorTest, SuccessfulAuth) {
  54   ASSERT_NO_FATAL_FAILURE(
  55       InitAuthenticators(kTestSharedSecret, kTestSharedSecret));
  56   ASSERT_NO_FATAL_FAILURE(RunAuthExchange());
  57
  58   ASSERT_EQ(Authenticator::ACCEPTED, host_->state());
  59   ASSERT_EQ(Authenticator::ACCEPTED, client_->state());
  60
  61   client_auth_ = client_->CreateChannelAuthenticator();
  62   host_auth_ = host_->CreateChannelAuthenticator();
  63   RunChannelAuth(false);
  64
  65   StreamConnectionTester tester(host_socket_.get(), client_socket_.get(),
  66                                 kMessageSize, kMessages);
  67
  68   tester.Start();
  69   message_loop_.Run();
  70   tester.CheckResults();
  71 }
  72
  73 // Verify that connection is rejected when secrets don't match.
  74 TEST_F(V2AuthenticatorTest, InvalidSecret) {
  75   ASSERT_NO_FATAL_FAILURE(
  76       InitAuthenticators(kTestSharedSecretBad, kTestSharedSecret));
  77   ASSERT_NO_FATAL_FAILURE(RunAuthExchange());
  78
  79   ASSERT_EQ(Authenticator::REJECTED, client_->state());
  80
  81   // Change |client_| so that we can get the last message.
  82   reinterpret_cast<V2Authenticator*>(client_.get())->state_ =
  83       Authenticator::MESSAGE_READY;
  84
  85   scoped_ptr<buzz::XmlElement> message(client_->GetNextMessage());
  86   ASSERT_TRUE(message.get());
  87
  88   ASSERT_EQ(Authenticator::WAITING_MESSAGE, client_->state());
  89   host_->ProcessMessage(message.get(), base::Bind(&base::DoNothing));
  90   // This assumes that V2Authenticator::ProcessMessage runs synchronously.
  91   ASSERT_EQ(Authenticator::REJECTED, host_->state());
  92 }
  93
  94 }  // namespace protocol
  95 }  // namespace remoting