| 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 <memory> | 7 #include <memory> |
| 8 #include <ostream> | 8 #include <ostream> |
| 9 #include <string> | 9 #include <string> |
| 10 | 10 |
| 11 #include "base/macros.h" | 11 #include "base/macros.h" |
| 12 #include "net/quic/core/crypto/crypto_handshake.h" | 12 #include "net/quic/core/crypto/crypto_handshake.h" |
| 13 #include "net/quic/core/crypto/quic_crypto_server_config.h" | 13 #include "net/quic/core/crypto/quic_crypto_server_config.h" |
| 14 #include "net/quic/core/crypto/quic_random.h" | 14 #include "net/quic/core/crypto/quic_random.h" |
| 15 #include "net/quic/core/quic_crypto_stream.h" | 15 #include "net/quic/core/quic_crypto_stream.h" |
| 16 #include "net/quic/core/quic_flags.h" | |
| 17 #include "net/quic/core/quic_utils.h" | 16 #include "net/quic/core/quic_utils.h" |
| 17 #include "net/quic/platform/api/quic_flags.h" |
| 18 #include "net/quic/platform/api/quic_logging.h" | 18 #include "net/quic/platform/api/quic_logging.h" |
| 19 #include "net/quic/platform/api/quic_str_cat.h" | 19 #include "net/quic/platform/api/quic_str_cat.h" |
| 20 #include "net/quic/test_tools/crypto_test_utils.h" | 20 #include "net/quic/test_tools/crypto_test_utils.h" |
| 21 #include "net/quic/test_tools/fake_proof_source.h" | 21 #include "net/quic/test_tools/fake_proof_source.h" |
| 22 #include "net/quic/test_tools/quic_buffered_packet_store_peer.h" | 22 #include "net/quic/test_tools/quic_buffered_packet_store_peer.h" |
| 23 #include "net/quic/test_tools/quic_crypto_server_config_peer.h" | 23 #include "net/quic/test_tools/quic_crypto_server_config_peer.h" |
| 24 #include "net/quic/test_tools/quic_test_utils.h" | 24 #include "net/quic/test_tools/quic_test_utils.h" |
| 25 #include "net/quic/test_tools/quic_time_wait_list_manager_peer.h" | 25 #include "net/quic/test_tools/quic_time_wait_list_manager_peer.h" |
| 26 #include "net/test/gtest_util.h" | 26 #include "net/test/gtest_util.h" |
| 27 #include "net/tools/epoll_server/epoll_server.h" | 27 #include "net/tools/epoll_server/epoll_server.h" |
| (...skipping 500 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 528 // connection. | 528 // connection. |
| 529 ProcessPacket(client_address, connection_id, true, SerializeCHLO(), | 529 ProcessPacket(client_address, connection_id, true, SerializeCHLO(), |
| 530 PACKET_8BYTE_CONNECTION_ID, PACKET_6BYTE_PACKET_NUMBER, | 530 PACKET_8BYTE_CONNECTION_ID, PACKET_6BYTE_PACKET_NUMBER, |
| 531 QuicDispatcher::kMaxReasonableInitialPacketNumber + 1); | 531 QuicDispatcher::kMaxReasonableInitialPacketNumber + 1); |
| 532 } | 532 } |
| 533 | 533 |
| 534 TEST_F(QuicDispatcherTest, SupportedVersionsChangeInFlight) { | 534 TEST_F(QuicDispatcherTest, SupportedVersionsChangeInFlight) { |
| 535 static_assert(arraysize(kSupportedQuicVersions) == 5u, | 535 static_assert(arraysize(kSupportedQuicVersions) == 5u, |
| 536 "Supported versions out of sync"); | 536 "Supported versions out of sync"); |
| 537 FLAGS_quic_reloadable_flag_quic_enable_version_38 = true; | 537 FLAGS_quic_reloadable_flag_quic_enable_version_38 = true; |
| 538 base::SetFlag(&FLAGS_quic_enable_version_39, true); | 538 SetQuicFlag(&FLAGS_quic_enable_version_39, true); |
| 539 QuicSocketAddress client_address(QuicIpAddress::Loopback4(), 1); | 539 QuicSocketAddress client_address(QuicIpAddress::Loopback4(), 1); |
| 540 server_address_ = QuicSocketAddress(QuicIpAddress::Any4(), 5); | 540 server_address_ = QuicSocketAddress(QuicIpAddress::Any4(), 5); |
| 541 QuicConnectionId connection_id = 1; | 541 QuicConnectionId connection_id = 1; |
| 542 | 542 |
| 543 EXPECT_CALL(*dispatcher_, CreateQuicSession(connection_id, client_address)) | 543 EXPECT_CALL(*dispatcher_, CreateQuicSession(connection_id, client_address)) |
| 544 .Times(0); | 544 .Times(0); |
| 545 ProcessPacket(client_address, connection_id, true, | 545 ProcessPacket(client_address, connection_id, true, |
| 546 static_cast<QuicVersion>(QuicVersionMin() - 1), SerializeCHLO(), | 546 static_cast<QuicVersion>(QuicVersionMin() - 1), SerializeCHLO(), |
| 547 PACKET_8BYTE_CONNECTION_ID, PACKET_6BYTE_PACKET_NUMBER, 1); | 547 PACKET_8BYTE_CONNECTION_ID, PACKET_6BYTE_PACKET_NUMBER, 1); |
| 548 ++connection_id; | 548 ++connection_id; |
| (...skipping 19 matching lines...) Expand all Loading... |
| 568 EXPECT_CALL(*reinterpret_cast<MockQuicConnection*>(session1_->connection()), | 568 EXPECT_CALL(*reinterpret_cast<MockQuicConnection*>(session1_->connection()), |
| 569 ProcessUdpPacket(_, _, _)) | 569 ProcessUdpPacket(_, _, _)) |
| 570 .WillOnce(testing::WithArgs<2>( | 570 .WillOnce(testing::WithArgs<2>( |
| 571 Invoke(CreateFunctor(&QuicDispatcherTest::ValidatePacket, | 571 Invoke(CreateFunctor(&QuicDispatcherTest::ValidatePacket, |
| 572 base::Unretained(this), connection_id)))); | 572 base::Unretained(this), connection_id)))); |
| 573 ProcessPacket(client_address, connection_id, true, QuicVersionMax(), | 573 ProcessPacket(client_address, connection_id, true, QuicVersionMax(), |
| 574 SerializeCHLO(), PACKET_8BYTE_CONNECTION_ID, | 574 SerializeCHLO(), PACKET_8BYTE_CONNECTION_ID, |
| 575 PACKET_6BYTE_PACKET_NUMBER, 1); | 575 PACKET_6BYTE_PACKET_NUMBER, 1); |
| 576 | 576 |
| 577 // Turn off version 39. | 577 // Turn off version 39. |
| 578 base::SetFlag(&FLAGS_quic_enable_version_39, false); | 578 SetQuicFlag(&FLAGS_quic_enable_version_39, false); |
| 579 ++connection_id; | 579 ++connection_id; |
| 580 EXPECT_CALL(*dispatcher_, CreateQuicSession(connection_id, client_address)) | 580 EXPECT_CALL(*dispatcher_, CreateQuicSession(connection_id, client_address)) |
| 581 .Times(0); | 581 .Times(0); |
| 582 ProcessPacket(client_address, connection_id, true, QUIC_VERSION_39, | 582 ProcessPacket(client_address, connection_id, true, QUIC_VERSION_39, |
| 583 SerializeCHLO(), PACKET_8BYTE_CONNECTION_ID, | 583 SerializeCHLO(), PACKET_8BYTE_CONNECTION_ID, |
| 584 PACKET_6BYTE_PACKET_NUMBER, 1); | 584 PACKET_6BYTE_PACKET_NUMBER, 1); |
| 585 | 585 |
| 586 // Turn on version 39. | 586 // Turn on version 39. |
| 587 base::SetFlag(&FLAGS_quic_enable_version_39, true); | 587 SetQuicFlag(&FLAGS_quic_enable_version_39, true); |
| 588 ++connection_id; | 588 ++connection_id; |
| 589 EXPECT_CALL(*dispatcher_, CreateQuicSession(connection_id, client_address)) | 589 EXPECT_CALL(*dispatcher_, CreateQuicSession(connection_id, client_address)) |
| 590 .WillOnce(testing::Return(CreateSession( | 590 .WillOnce(testing::Return(CreateSession( |
| 591 dispatcher_.get(), config_, connection_id, client_address, | 591 dispatcher_.get(), config_, connection_id, client_address, |
| 592 &mock_helper_, &mock_alarm_factory_, &crypto_config_, | 592 &mock_helper_, &mock_alarm_factory_, &crypto_config_, |
| 593 QuicDispatcherPeer::GetCache(dispatcher_.get()), &session1_))); | 593 QuicDispatcherPeer::GetCache(dispatcher_.get()), &session1_))); |
| 594 EXPECT_CALL(*reinterpret_cast<MockQuicConnection*>(session1_->connection()), | 594 EXPECT_CALL(*reinterpret_cast<MockQuicConnection*>(session1_->connection()), |
| 595 ProcessUdpPacket(_, _, _)) | 595 ProcessUdpPacket(_, _, _)) |
| 596 .WillOnce(testing::WithArgs<2>( | 596 .WillOnce(testing::WithArgs<2>( |
| 597 Invoke(CreateFunctor(&QuicDispatcherTest::ValidatePacket, | 597 Invoke(CreateFunctor(&QuicDispatcherTest::ValidatePacket, |
| (...skipping 1445 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 2043 check.Call(2); | 2043 check.Call(2); |
| 2044 GetFakeProofSource()->InvokePendingCallback(0); | 2044 GetFakeProofSource()->InvokePendingCallback(0); |
| 2045 ASSERT_EQ(GetFakeProofSource()->NumPendingCallbacks(), 0); | 2045 ASSERT_EQ(GetFakeProofSource()->NumPendingCallbacks(), 0); |
| 2046 EXPECT_FALSE(store->HasBufferedPackets(conn_id)); | 2046 EXPECT_FALSE(store->HasBufferedPackets(conn_id)); |
| 2047 EXPECT_FALSE(time_wait_list_manager_->IsConnectionIdInTimeWait(conn_id)); | 2047 EXPECT_FALSE(time_wait_list_manager_->IsConnectionIdInTimeWait(conn_id)); |
| 2048 } | 2048 } |
| 2049 | 2049 |
| 2050 } // namespace | 2050 } // namespace |
| 2051 } // namespace test | 2051 } // namespace test |
| 2052 } // namespace net | 2052 } // namespace net |
| OLD | NEW |