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 "net/tools/quic/quic_dispatcher.h" | 5 #include "net/tools/quic/quic_dispatcher.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "base/strings/string_piece.h" | 9 #include "base/strings/string_piece.h" |
10 #include "net/quic/crypto/crypto_handshake.h" | 10 #include "net/quic/crypto/crypto_handshake.h" |
(...skipping 10 matching lines...) Expand all Loading... |
21 #include "net/tools/quic/test_tools/quic_test_utils.h" | 21 #include "net/tools/quic/test_tools/quic_test_utils.h" |
22 #include "testing/gmock/include/gmock/gmock.h" | 22 #include "testing/gmock/include/gmock/gmock.h" |
23 #include "testing/gtest/include/gtest/gtest.h" | 23 #include "testing/gtest/include/gtest/gtest.h" |
24 | 24 |
25 using base::StringPiece; | 25 using base::StringPiece; |
26 using net::EpollServer; | 26 using net::EpollServer; |
27 using net::test::ConstructEncryptedPacket; | 27 using net::test::ConstructEncryptedPacket; |
28 using net::test::MockConnection; | 28 using net::test::MockConnection; |
29 using net::test::MockSession; | 29 using net::test::MockSession; |
30 using net::test::ValueRestore; | 30 using net::test::ValueRestore; |
31 using std::make_pair; | |
32 using std::string; | 31 using std::string; |
33 using testing::DoAll; | 32 using testing::DoAll; |
34 using testing::InSequence; | 33 using testing::InSequence; |
35 using testing::Invoke; | 34 using testing::Invoke; |
36 using testing::WithoutArgs; | 35 using testing::WithoutArgs; |
37 using testing::_; | 36 using testing::_; |
38 | 37 |
39 namespace net { | 38 namespace net { |
40 namespace tools { | 39 namespace tools { |
41 namespace test { | 40 namespace test { |
(...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
90 *session = new MockSession(connection); | 89 *session = new MockSession(connection); |
91 ON_CALL(*connection, SendConnectionClose(_)).WillByDefault( | 90 ON_CALL(*connection, SendConnectionClose(_)).WillByDefault( |
92 WithoutArgs(Invoke( | 91 WithoutArgs(Invoke( |
93 connection, &MockServerConnection::UnregisterOnConnectionClosed))); | 92 connection, &MockServerConnection::UnregisterOnConnectionClosed))); |
94 EXPECT_CALL(*reinterpret_cast<MockConnection*>((*session)->connection()), | 93 EXPECT_CALL(*reinterpret_cast<MockConnection*>((*session)->connection()), |
95 ProcessUdpPacket(_, client_address, _)); | 94 ProcessUdpPacket(_, client_address, _)); |
96 | 95 |
97 return *session; | 96 return *session; |
98 } | 97 } |
99 | 98 |
100 class MockTimeWaitListManager : public QuicTimeWaitListManager { | |
101 public: | |
102 MockTimeWaitListManager(QuicPacketWriter* writer, | |
103 QuicServerSessionVisitor* visitor, | |
104 EpollServer* eps) | |
105 : QuicTimeWaitListManager(writer, visitor, eps, QuicSupportedVersions()) { | |
106 } | |
107 | |
108 MOCK_METHOD5(ProcessPacket, | |
109 void(const IPEndPoint& server_address, | |
110 const IPEndPoint& client_address, | |
111 QuicConnectionId connection_id, | |
112 QuicPacketSequenceNumber sequence_number, | |
113 const QuicEncryptedPacket& packet)); | |
114 }; | |
115 | |
116 class QuicDispatcherTest : public ::testing::Test { | 99 class QuicDispatcherTest : public ::testing::Test { |
117 public: | 100 public: |
118 QuicDispatcherTest() | 101 QuicDispatcherTest() |
119 : crypto_config_(QuicCryptoServerConfig::TESTING, | 102 : crypto_config_(QuicCryptoServerConfig::TESTING, |
120 QuicRandom::GetInstance()), | 103 QuicRandom::GetInstance()), |
121 dispatcher_(config_, crypto_config_, &eps_), | 104 dispatcher_(config_, crypto_config_, &eps_), |
122 time_wait_list_manager_(nullptr), | 105 time_wait_list_manager_(nullptr), |
123 session1_(nullptr), | 106 session1_(nullptr), |
124 session2_(nullptr) { | 107 session2_(nullptr) { |
125 dispatcher_.Initialize(1); | 108 dispatcher_.Initialize(1); |
(...skipping 124 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
250 .WillOnce(Invoke( | 233 .WillOnce(Invoke( |
251 reinterpret_cast<MockConnection*>(session1_->connection()), | 234 reinterpret_cast<MockConnection*>(session1_->connection()), |
252 &MockConnection::ReallyProcessUdpPacket)); | 235 &MockConnection::ReallyProcessUdpPacket)); |
253 dispatcher_.ProcessPacket(IPEndPoint(), client_address, *encrypted); | 236 dispatcher_.ProcessPacket(IPEndPoint(), client_address, *encrypted); |
254 EXPECT_TRUE(time_wait_list_manager_->IsConnectionIdInTimeWait(connection_id)); | 237 EXPECT_TRUE(time_wait_list_manager_->IsConnectionIdInTimeWait(connection_id)); |
255 | 238 |
256 // Dispatcher forwards subsequent packets for this connection_id to the time | 239 // Dispatcher forwards subsequent packets for this connection_id to the time |
257 // wait list manager. | 240 // wait list manager. |
258 EXPECT_CALL(*time_wait_list_manager_, | 241 EXPECT_CALL(*time_wait_list_manager_, |
259 ProcessPacket(_, _, connection_id, _, _)).Times(1); | 242 ProcessPacket(_, _, connection_id, _, _)).Times(1); |
| 243 EXPECT_CALL(*time_wait_list_manager_, |
| 244 AddConnectionIdToTimeWait(_, _, _)).Times(0); |
260 ProcessPacket(client_address, connection_id, true, "foo"); | 245 ProcessPacket(client_address, connection_id, true, "foo"); |
261 } | 246 } |
262 | 247 |
263 TEST_F(QuicDispatcherTest, StrayPacketToTimeWaitListManager) { | 248 TEST_F(QuicDispatcherTest, StrayPacketToTimeWaitListManager) { |
264 CreateTimeWaitListManager(); | 249 CreateTimeWaitListManager(); |
265 | 250 |
266 IPEndPoint client_address(net::test::Loopback4(), 1); | 251 IPEndPoint client_address(net::test::Loopback4(), 1); |
267 QuicConnectionId connection_id = 1; | 252 QuicConnectionId connection_id = 1; |
268 // Dispatcher forwards all packets for this connection_id to the time wait | 253 // Dispatcher forwards all packets for this connection_id to the time wait |
269 // list manager. | 254 // list manager. |
270 EXPECT_CALL(dispatcher_, CreateQuicSession(_, _, _)).Times(0); | 255 EXPECT_CALL(dispatcher_, CreateQuicSession(_, _, _)).Times(0); |
271 EXPECT_CALL(*time_wait_list_manager_, | 256 EXPECT_CALL(*time_wait_list_manager_, |
272 ProcessPacket(_, _, connection_id, _, _)).Times(1); | 257 ProcessPacket(_, _, connection_id, _, _)).Times(1); |
| 258 EXPECT_CALL(*time_wait_list_manager_, |
| 259 AddConnectionIdToTimeWait(_, _, _)).Times(1); |
273 ProcessPacket(client_address, connection_id, false, "data"); | 260 ProcessPacket(client_address, connection_id, false, "data"); |
274 } | 261 } |
275 | 262 |
276 TEST_F(QuicDispatcherTest, ProcessPacketWithBogusPort) { | 263 TEST_F(QuicDispatcherTest, ProcessPacketWithBogusPort) { |
277 CreateTimeWaitListManager(); | 264 CreateTimeWaitListManager(); |
278 | 265 |
279 IPEndPoint client_address(net::test::Loopback4(), 0); | 266 IPEndPoint client_address(net::test::Loopback4(), 0); |
280 IPAddressNumber any4; | 267 IPAddressNumber any4; |
281 CHECK(net::ParseIPLiteralToNumber("0.0.0.0", &any4)); | 268 CHECK(net::ParseIPLiteralToNumber("0.0.0.0", &any4)); |
282 server_address_ = IPEndPoint(any4, 5); | 269 server_address_ = IPEndPoint(any4, 5); |
283 | 270 |
284 // dispatcher_ should drop this packet. | 271 // dispatcher_ should drop this packet. |
285 EXPECT_CALL(dispatcher_, CreateQuicSession(1, _, client_address)).Times(0); | 272 EXPECT_CALL(dispatcher_, CreateQuicSession(1, _, client_address)).Times(0); |
286 EXPECT_CALL(*time_wait_list_manager_, ProcessPacket(_, _, _, _, _)).Times(0); | 273 EXPECT_CALL(*time_wait_list_manager_, ProcessPacket(_, _, _, _, _)).Times(0); |
| 274 EXPECT_CALL(*time_wait_list_manager_, |
| 275 AddConnectionIdToTimeWait(_, _, _)).Times(0); |
287 ProcessPacket(client_address, 1, true, "foo"); | 276 ProcessPacket(client_address, 1, true, "foo"); |
288 EXPECT_EQ(client_address, dispatcher_.current_client_address()); | 277 EXPECT_EQ(client_address, dispatcher_.current_client_address()); |
289 EXPECT_EQ(server_address_, dispatcher_.current_server_address()); | 278 EXPECT_EQ(server_address_, dispatcher_.current_server_address()); |
290 } | 279 } |
291 | 280 |
292 class BlockingWriter : public QuicPacketWriterWrapper { | 281 class BlockingWriter : public QuicPacketWriterWrapper { |
293 public: | 282 public: |
294 BlockingWriter() : write_blocked_(false) {} | 283 BlockingWriter() : write_blocked_(false) {} |
295 | 284 |
296 bool IsWriteBlocked() const override { return write_blocked_; } | 285 bool IsWriteBlocked() const override { return write_blocked_; } |
(...skipping 183 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
480 // And we'll resume where we left off when we get another call. | 469 // And we'll resume where we left off when we get another call. |
481 EXPECT_CALL(*connection2(), OnCanWrite()); | 470 EXPECT_CALL(*connection2(), OnCanWrite()); |
482 dispatcher_.OnCanWrite(); | 471 dispatcher_.OnCanWrite(); |
483 EXPECT_FALSE(dispatcher_.HasPendingWrites()); | 472 EXPECT_FALSE(dispatcher_.HasPendingWrites()); |
484 } | 473 } |
485 | 474 |
486 } // namespace | 475 } // namespace |
487 } // namespace test | 476 } // namespace test |
488 } // namespace tools | 477 } // namespace tools |
489 } // namespace net | 478 } // namespace net |
OLD | NEW |