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/jingle_session.h" | 5 #include "remoting/protocol/jingle_session.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/message_loop/message_loop.h" | 8 #include "base/message_loop/message_loop.h" |
9 #include "base/run_loop.h" | 9 #include "base/run_loop.h" |
10 #include "base/test/test_timeouts.h" | 10 #include "base/test/test_timeouts.h" |
11 #include "base/time/time.h" | 11 #include "base/time/time.h" |
12 #include "jingle/glue/thread_wrapper.h" | 12 #include "jingle/glue/thread_wrapper.h" |
13 #include "net/socket/socket.h" | 13 #include "net/socket/socket.h" |
14 #include "net/socket/stream_socket.h" | 14 #include "net/socket/stream_socket.h" |
15 #include "net/url_request/url_request_context_getter.h" | 15 #include "net/url_request/url_request_context_getter.h" |
16 #include "remoting/base/constants.h" | 16 #include "remoting/base/constants.h" |
17 #include "remoting/protocol/authenticator.h" | 17 #include "remoting/protocol/authenticator.h" |
18 #include "remoting/protocol/channel_authenticator.h" | 18 #include "remoting/protocol/channel_authenticator.h" |
19 #include "remoting/protocol/chromium_port_allocator.h" | 19 #include "remoting/protocol/chromium_port_allocator.h" |
20 #include "remoting/protocol/connection_tester.h" | 20 #include "remoting/protocol/connection_tester.h" |
21 #include "remoting/protocol/fake_authenticator.h" | 21 #include "remoting/protocol/fake_authenticator.h" |
22 #include "remoting/protocol/ice_transport_factory.h" | |
23 #include "remoting/protocol/jingle_session_manager.h" | 22 #include "remoting/protocol/jingle_session_manager.h" |
| 23 #include "remoting/protocol/libjingle_transport_factory.h" |
24 #include "remoting/protocol/network_settings.h" | 24 #include "remoting/protocol/network_settings.h" |
25 #include "remoting/protocol/p2p_stream_socket.h" | 25 #include "remoting/protocol/p2p_stream_socket.h" |
26 #include "remoting/protocol/stream_channel_factory.h" | 26 #include "remoting/protocol/stream_channel_factory.h" |
27 #include "remoting/signaling/fake_signal_strategy.h" | 27 #include "remoting/signaling/fake_signal_strategy.h" |
28 #include "testing/gmock/include/gmock/gmock.h" | 28 #include "testing/gmock/include/gmock/gmock.h" |
29 #include "testing/gtest/include/gtest/gtest.h" | 29 #include "testing/gtest/include/gtest/gtest.h" |
30 | 30 |
31 using testing::_; | 31 using testing::_; |
32 using testing::AtLeast; | 32 using testing::AtLeast; |
33 using testing::AtMost; | 33 using testing::AtMost; |
(...skipping 106 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
140 void CreateSessionManagers(int auth_round_trips, int messages_till_start, | 140 void CreateSessionManagers(int auth_round_trips, int messages_till_start, |
141 FakeAuthenticator::Action auth_action) { | 141 FakeAuthenticator::Action auth_action) { |
142 host_signal_strategy_.reset(new FakeSignalStrategy(kHostJid)); | 142 host_signal_strategy_.reset(new FakeSignalStrategy(kHostJid)); |
143 client_signal_strategy_.reset(new FakeSignalStrategy(kClientJid)); | 143 client_signal_strategy_.reset(new FakeSignalStrategy(kClientJid)); |
144 FakeSignalStrategy::Connect(host_signal_strategy_.get(), | 144 FakeSignalStrategy::Connect(host_signal_strategy_.get(), |
145 client_signal_strategy_.get()); | 145 client_signal_strategy_.get()); |
146 | 146 |
147 EXPECT_CALL(host_server_listener_, OnSessionManagerReady()) | 147 EXPECT_CALL(host_server_listener_, OnSessionManagerReady()) |
148 .Times(1); | 148 .Times(1); |
149 | 149 |
150 scoped_ptr<TransportFactory> host_transport(new IceTransportFactory( | 150 scoped_ptr<TransportFactory> host_transport(new LibjingleTransportFactory( |
151 nullptr, | 151 nullptr, |
152 ChromiumPortAllocator::Create(nullptr, network_settings_).Pass(), | 152 ChromiumPortAllocator::Create(nullptr, network_settings_).Pass(), |
153 network_settings_, TransportRole::SERVER)); | 153 network_settings_, TransportRole::SERVER)); |
154 host_server_.reset(new JingleSessionManager(host_transport.Pass())); | 154 host_server_.reset(new JingleSessionManager(host_transport.Pass())); |
155 host_server_->Init(host_signal_strategy_.get(), &host_server_listener_); | 155 host_server_->Init(host_signal_strategy_.get(), &host_server_listener_); |
156 | 156 |
157 scoped_ptr<AuthenticatorFactory> factory( | 157 scoped_ptr<AuthenticatorFactory> factory( |
158 new FakeHostAuthenticatorFactory(auth_round_trips, | 158 new FakeHostAuthenticatorFactory(auth_round_trips, |
159 messages_till_start, auth_action, true)); | 159 messages_till_start, auth_action, true)); |
160 host_server_->set_authenticator_factory(factory.Pass()); | 160 host_server_->set_authenticator_factory(factory.Pass()); |
161 | 161 |
162 EXPECT_CALL(client_server_listener_, OnSessionManagerReady()) | 162 EXPECT_CALL(client_server_listener_, OnSessionManagerReady()) |
163 .Times(1); | 163 .Times(1); |
164 scoped_ptr<TransportFactory> client_transport(new IceTransportFactory( | 164 scoped_ptr<TransportFactory> client_transport(new LibjingleTransportFactory( |
165 nullptr, | 165 nullptr, |
166 ChromiumPortAllocator::Create(nullptr, network_settings_).Pass(), | 166 ChromiumPortAllocator::Create(nullptr, network_settings_).Pass(), |
167 network_settings_, TransportRole::CLIENT)); | 167 network_settings_, TransportRole::CLIENT)); |
168 client_server_.reset( | 168 client_server_.reset( |
169 new JingleSessionManager(client_transport.Pass())); | 169 new JingleSessionManager(client_transport.Pass())); |
170 client_server_->Init(client_signal_strategy_.get(), | 170 client_server_->Init(client_signal_strategy_.get(), |
171 &client_server_listener_); | 171 &client_server_listener_); |
172 } | 172 } |
173 | 173 |
174 void CreateSessionManagers(int auth_round_trips, | 174 void CreateSessionManagers(int auth_round_trips, |
(...skipping 73 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
248 scoped_ptr<Authenticator> authenticator(new FakeAuthenticator( | 248 scoped_ptr<Authenticator> authenticator(new FakeAuthenticator( |
249 FakeAuthenticator::CLIENT, auth_round_trips, auth_action, true)); | 249 FakeAuthenticator::CLIENT, auth_round_trips, auth_action, true)); |
250 | 250 |
251 client_session_ = client_server_->Connect(kHostJid, authenticator.Pass()); | 251 client_session_ = client_server_->Connect(kHostJid, authenticator.Pass()); |
252 client_session_->SetEventHandler(&client_session_event_handler_); | 252 client_session_->SetEventHandler(&client_session_event_handler_); |
253 | 253 |
254 base::RunLoop().RunUntilIdle(); | 254 base::RunLoop().RunUntilIdle(); |
255 } | 255 } |
256 | 256 |
257 void CreateChannel() { | 257 void CreateChannel() { |
258 client_session_->GetTransportSession() | 258 client_session_->GetTransportChannelFactory()->CreateChannel( |
259 ->GetStreamChannelFactory() | 259 kChannelName, base::Bind(&JingleSessionTest::OnClientChannelCreated, |
260 ->CreateChannel(kChannelName, | 260 base::Unretained(this))); |
261 base::Bind(&JingleSessionTest::OnClientChannelCreated, | 261 host_session_->GetTransportChannelFactory()->CreateChannel( |
262 base::Unretained(this))); | 262 kChannelName, base::Bind(&JingleSessionTest::OnHostChannelCreated, |
263 host_session_->GetTransportSession() | 263 base::Unretained(this))); |
264 ->GetStreamChannelFactory() | |
265 ->CreateChannel(kChannelName, | |
266 base::Bind(&JingleSessionTest::OnHostChannelCreated, | |
267 base::Unretained(this))); | |
268 | 264 |
269 int counter = 2; | 265 int counter = 2; |
270 ExpectRouteChange(kChannelName); | 266 ExpectRouteChange(kChannelName); |
271 EXPECT_CALL(client_channel_callback_, OnDone(_)) | 267 EXPECT_CALL(client_channel_callback_, OnDone(_)) |
272 .WillOnce(QuitThreadOnCounter(&counter)); | 268 .WillOnce(QuitThreadOnCounter(&counter)); |
273 EXPECT_CALL(host_channel_callback_, OnDone(_)) | 269 EXPECT_CALL(host_channel_callback_, OnDone(_)) |
274 .WillOnce(QuitThreadOnCounter(&counter)); | 270 .WillOnce(QuitThreadOnCounter(&counter)); |
275 message_loop_->Run(); | 271 message_loop_->Run(); |
276 | 272 |
277 EXPECT_TRUE(client_socket_.get()); | 273 EXPECT_TRUE(client_socket_.get()); |
(...skipping 215 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
493 client_session_ = client_server_->Connect(kHostJid, authenticator.Pass()); | 489 client_session_ = client_server_->Connect(kHostJid, authenticator.Pass()); |
494 base::RunLoop().RunUntilIdle(); | 490 base::RunLoop().RunUntilIdle(); |
495 } | 491 } |
496 | 492 |
497 // Verify that data can be sent over a multiplexed channel. | 493 // Verify that data can be sent over a multiplexed channel. |
498 TEST_F(JingleSessionTest, TestMuxStreamChannel) { | 494 TEST_F(JingleSessionTest, TestMuxStreamChannel) { |
499 CreateSessionManagers(1, FakeAuthenticator::ACCEPT); | 495 CreateSessionManagers(1, FakeAuthenticator::ACCEPT); |
500 ASSERT_NO_FATAL_FAILURE( | 496 ASSERT_NO_FATAL_FAILURE( |
501 InitiateConnection(1, FakeAuthenticator::ACCEPT, false)); | 497 InitiateConnection(1, FakeAuthenticator::ACCEPT, false)); |
502 | 498 |
503 client_session_->GetTransportSession() | 499 client_session_->GetMultiplexedChannelFactory()->CreateChannel( |
504 ->GetMultiplexedChannelFactory() | 500 kChannelName, base::Bind(&JingleSessionTest::OnClientChannelCreated, |
505 ->CreateChannel(kChannelName, | 501 base::Unretained(this))); |
506 base::Bind(&JingleSessionTest::OnClientChannelCreated, | 502 host_session_->GetMultiplexedChannelFactory()->CreateChannel( |
507 base::Unretained(this))); | 503 kChannelName, base::Bind(&JingleSessionTest::OnHostChannelCreated, |
508 host_session_->GetTransportSession() | 504 base::Unretained(this))); |
509 ->GetMultiplexedChannelFactory() | |
510 ->CreateChannel(kChannelName, | |
511 base::Bind(&JingleSessionTest::OnHostChannelCreated, | |
512 base::Unretained(this))); | |
513 | 505 |
514 int counter = 2; | 506 int counter = 2; |
515 ExpectRouteChange("mux"); | 507 ExpectRouteChange("mux"); |
516 EXPECT_CALL(client_channel_callback_, OnDone(_)) | 508 EXPECT_CALL(client_channel_callback_, OnDone(_)) |
517 .WillOnce(QuitThreadOnCounter(&counter)); | 509 .WillOnce(QuitThreadOnCounter(&counter)); |
518 EXPECT_CALL(host_channel_callback_, OnDone(_)) | 510 EXPECT_CALL(host_channel_callback_, OnDone(_)) |
519 .WillOnce(QuitThreadOnCounter(&counter)); | 511 .WillOnce(QuitThreadOnCounter(&counter)); |
520 message_loop_->Run(); | 512 message_loop_->Run(); |
521 | 513 |
522 EXPECT_TRUE(client_socket_.get()); | 514 EXPECT_TRUE(client_socket_.get()); |
(...skipping 94 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
617 message_loop_->Run(); | 609 message_loop_->Run(); |
618 tester.CheckResults(); | 610 tester.CheckResults(); |
619 } | 611 } |
620 | 612 |
621 // Verify that we shutdown properly when channel authentication fails. | 613 // Verify that we shutdown properly when channel authentication fails. |
622 TEST_F(JingleSessionTest, TestFailedChannelAuth) { | 614 TEST_F(JingleSessionTest, TestFailedChannelAuth) { |
623 CreateSessionManagers(1, FakeAuthenticator::REJECT_CHANNEL); | 615 CreateSessionManagers(1, FakeAuthenticator::REJECT_CHANNEL); |
624 ASSERT_NO_FATAL_FAILURE( | 616 ASSERT_NO_FATAL_FAILURE( |
625 InitiateConnection(1, FakeAuthenticator::ACCEPT, false)); | 617 InitiateConnection(1, FakeAuthenticator::ACCEPT, false)); |
626 | 618 |
627 client_session_->GetTransportSession() | 619 client_session_->GetTransportChannelFactory()->CreateChannel( |
628 ->GetStreamChannelFactory() | 620 kChannelName, base::Bind(&JingleSessionTest::OnClientChannelCreated, |
629 ->CreateChannel(kChannelName, | 621 base::Unretained(this))); |
630 base::Bind(&JingleSessionTest::OnClientChannelCreated, | 622 host_session_->GetTransportChannelFactory()->CreateChannel( |
631 base::Unretained(this))); | 623 kChannelName, base::Bind(&JingleSessionTest::OnHostChannelCreated, |
632 host_session_->GetTransportSession() | 624 base::Unretained(this))); |
633 ->GetStreamChannelFactory() | |
634 ->CreateChannel(kChannelName, | |
635 base::Bind(&JingleSessionTest::OnHostChannelCreated, | |
636 base::Unretained(this))); | |
637 | 625 |
638 // Terminate the message loop when we get rejection notification | 626 // Terminate the message loop when we get rejection notification |
639 // from the host. | 627 // from the host. |
640 EXPECT_CALL(host_channel_callback_, OnDone(nullptr)) | 628 EXPECT_CALL(host_channel_callback_, OnDone(nullptr)) |
641 .WillOnce(QuitThread()); | 629 .WillOnce(QuitThread()); |
642 ExpectRouteChange(kChannelName); | 630 ExpectRouteChange(kChannelName); |
643 | 631 |
644 message_loop_->Run(); | 632 message_loop_->Run(); |
645 | 633 |
646 client_session_->GetTransportSession() | 634 client_session_->GetTransportChannelFactory()->CancelChannelCreation( |
647 ->GetStreamChannelFactory() | 635 kChannelName); |
648 ->CancelChannelCreation(kChannelName); | |
649 | 636 |
650 EXPECT_TRUE(!host_socket_.get()); | 637 EXPECT_TRUE(!host_socket_.get()); |
651 } | 638 } |
652 | 639 |
653 TEST_F(JingleSessionTest, TestCancelChannelCreation) { | 640 TEST_F(JingleSessionTest, TestCancelChannelCreation) { |
654 CreateSessionManagers(1, FakeAuthenticator::REJECT_CHANNEL); | 641 CreateSessionManagers(1, FakeAuthenticator::REJECT_CHANNEL); |
655 ASSERT_NO_FATAL_FAILURE( | 642 ASSERT_NO_FATAL_FAILURE( |
656 InitiateConnection(1, FakeAuthenticator::ACCEPT, false)); | 643 InitiateConnection(1, FakeAuthenticator::ACCEPT, false)); |
657 | 644 |
658 client_session_->GetTransportSession() | 645 client_session_->GetTransportChannelFactory()->CreateChannel( |
659 ->GetStreamChannelFactory() | 646 kChannelName, base::Bind(&JingleSessionTest::OnClientChannelCreated, |
660 ->CreateChannel(kChannelName, | 647 base::Unretained(this))); |
661 base::Bind(&JingleSessionTest::OnClientChannelCreated, | 648 client_session_->GetTransportChannelFactory()->CancelChannelCreation( |
662 base::Unretained(this))); | 649 kChannelName); |
663 client_session_->GetTransportSession() | |
664 ->GetStreamChannelFactory() | |
665 ->CancelChannelCreation(kChannelName); | |
666 | 650 |
667 EXPECT_TRUE(!client_socket_.get()); | 651 EXPECT_TRUE(!client_socket_.get()); |
668 } | 652 } |
669 | 653 |
670 // Verify that we can still connect even when there is a delay in signaling | 654 // Verify that we can still connect even when there is a delay in signaling |
671 // messages delivery. | 655 // messages delivery. |
672 TEST_F(JingleSessionTest, TestDelayedSignaling) { | 656 TEST_F(JingleSessionTest, TestDelayedSignaling) { |
673 CreateSessionManagers(1, FakeAuthenticator::ACCEPT); | 657 CreateSessionManagers(1, FakeAuthenticator::ACCEPT); |
674 ASSERT_NO_FATAL_FAILURE( | 658 ASSERT_NO_FATAL_FAILURE( |
675 InitiateConnection(1, FakeAuthenticator::ACCEPT, false)); | 659 InitiateConnection(1, FakeAuthenticator::ACCEPT, false)); |
676 | 660 |
677 host_signal_strategy_->set_send_delay( | 661 host_signal_strategy_->set_send_delay( |
678 base::TimeDelta::FromMilliseconds(100)); | 662 base::TimeDelta::FromMilliseconds(100)); |
679 | 663 |
680 ASSERT_NO_FATAL_FAILURE(CreateChannel()); | 664 ASSERT_NO_FATAL_FAILURE(CreateChannel()); |
681 | 665 |
682 StreamConnectionTester tester(host_socket_.get(), client_socket_.get(), | 666 StreamConnectionTester tester(host_socket_.get(), client_socket_.get(), |
683 kMessageSize, 1); | 667 kMessageSize, 1); |
684 tester.Start(); | 668 tester.Start(); |
685 message_loop_->Run(); | 669 message_loop_->Run(); |
686 tester.CheckResults(); | 670 tester.CheckResults(); |
687 } | 671 } |
688 | 672 |
689 } // namespace protocol | 673 } // namespace protocol |
690 } // namespace remoting | 674 } // namespace remoting |
OLD | NEW |