| OLD | NEW | 
|---|
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be | 
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. | 
| 4 | 4 | 
| 5 #include "remoting/protocol/v2_authenticator.h" | 5 #include "remoting/protocol/v2_authenticator.h" | 
| 6 | 6 | 
| 7 #include "base/bind.h" | 7 #include "base/bind.h" | 
| 8 #include "base/macros.h" | 8 #include "base/macros.h" | 
|  | 9 #include "base/run_loop.h" | 
| 9 #include "net/base/net_errors.h" | 10 #include "net/base/net_errors.h" | 
| 10 #include "remoting/base/rsa_key_pair.h" | 11 #include "remoting/base/rsa_key_pair.h" | 
| 11 #include "remoting/protocol/authenticator_test_base.h" | 12 #include "remoting/protocol/authenticator_test_base.h" | 
| 12 #include "remoting/protocol/channel_authenticator.h" | 13 #include "remoting/protocol/channel_authenticator.h" | 
| 13 #include "remoting/protocol/connection_tester.h" | 14 #include "remoting/protocol/connection_tester.h" | 
| 14 #include "testing/gmock/include/gmock/gmock.h" | 15 #include "testing/gmock/include/gmock/gmock.h" | 
| 15 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" | 
| 16 #include "third_party/webrtc/libjingle/xmllite/xmlelement.h" | 17 #include "third_party/webrtc/libjingle/xmllite/xmlelement.h" | 
| 17 | 18 | 
| 18 using testing::_; | 19 using testing::_; | 
| (...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after  Loading... | 
| 60   ASSERT_EQ(Authenticator::ACCEPTED, client_->state()); | 61   ASSERT_EQ(Authenticator::ACCEPTED, client_->state()); | 
| 61 | 62 | 
| 62   client_auth_ = client_->CreateChannelAuthenticator(); | 63   client_auth_ = client_->CreateChannelAuthenticator(); | 
| 63   host_auth_ = host_->CreateChannelAuthenticator(); | 64   host_auth_ = host_->CreateChannelAuthenticator(); | 
| 64   RunChannelAuth(false); | 65   RunChannelAuth(false); | 
| 65 | 66 | 
| 66   StreamConnectionTester tester(host_socket_.get(), client_socket_.get(), | 67   StreamConnectionTester tester(host_socket_.get(), client_socket_.get(), | 
| 67                                 kMessageSize, kMessages); | 68                                 kMessageSize, kMessages); | 
| 68 | 69 | 
| 69   tester.Start(); | 70   tester.Start(); | 
| 70   message_loop_.Run(); | 71   base::RunLoop().Run(); | 
| 71   tester.CheckResults(); | 72   tester.CheckResults(); | 
| 72 } | 73 } | 
| 73 | 74 | 
| 74 // Verify that connection is rejected when secrets don't match. | 75 // Verify that connection is rejected when secrets don't match. | 
| 75 TEST_F(V2AuthenticatorTest, InvalidSecret) { | 76 TEST_F(V2AuthenticatorTest, InvalidSecret) { | 
| 76   ASSERT_NO_FATAL_FAILURE( | 77   ASSERT_NO_FATAL_FAILURE( | 
| 77       InitAuthenticators(kTestSharedSecretBad, kTestSharedSecret)); | 78       InitAuthenticators(kTestSharedSecretBad, kTestSharedSecret)); | 
| 78   ASSERT_NO_FATAL_FAILURE(RunAuthExchange()); | 79   ASSERT_NO_FATAL_FAILURE(RunAuthExchange()); | 
| 79 | 80 | 
| 80   ASSERT_EQ(Authenticator::REJECTED, client_->state()); | 81   ASSERT_EQ(Authenticator::REJECTED, client_->state()); | 
| 81 | 82 | 
| 82   // Change |client_| so that we can get the last message. | 83   // Change |client_| so that we can get the last message. | 
| 83   reinterpret_cast<V2Authenticator*>(client_.get())->state_ = | 84   reinterpret_cast<V2Authenticator*>(client_.get())->state_ = | 
| 84       Authenticator::MESSAGE_READY; | 85       Authenticator::MESSAGE_READY; | 
| 85 | 86 | 
| 86   std::unique_ptr<buzz::XmlElement> message(client_->GetNextMessage()); | 87   std::unique_ptr<buzz::XmlElement> message(client_->GetNextMessage()); | 
| 87   ASSERT_TRUE(message.get()); | 88   ASSERT_TRUE(message.get()); | 
| 88 | 89 | 
| 89   ASSERT_EQ(Authenticator::WAITING_MESSAGE, client_->state()); | 90   ASSERT_EQ(Authenticator::WAITING_MESSAGE, client_->state()); | 
| 90   host_->ProcessMessage(message.get(), base::Bind(&base::DoNothing)); | 91   host_->ProcessMessage(message.get(), base::Bind(&base::DoNothing)); | 
| 91   // This assumes that V2Authenticator::ProcessMessage runs synchronously. | 92   // This assumes that V2Authenticator::ProcessMessage runs synchronously. | 
| 92   ASSERT_EQ(Authenticator::REJECTED, host_->state()); | 93   ASSERT_EQ(Authenticator::REJECTED, host_->state()); | 
| 93 } | 94 } | 
| 94 | 95 | 
| 95 }  // namespace protocol | 96 }  // namespace protocol | 
| 96 }  // namespace remoting | 97 }  // namespace remoting | 
| OLD | NEW | 
|---|