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/quic/core/quic_packet_creator.h" | 5 #include "net/quic/core/quic_packet_creator.h" |
6 | 6 |
7 #include <cstdint> | 7 #include <cstdint> |
8 #include <memory> | 8 #include <memory> |
9 #include <string> | 9 #include <string> |
10 | 10 |
11 #include "base/macros.h" | 11 #include "base/macros.h" |
12 #include "base/stl_util.h" | 12 #include "base/stl_util.h" |
13 #include "net/quic/core/crypto/null_encrypter.h" | 13 #include "net/quic/core/crypto/null_encrypter.h" |
14 #include "net/quic/core/crypto/quic_decrypter.h" | 14 #include "net/quic/core/crypto/quic_decrypter.h" |
15 #include "net/quic/core/crypto/quic_encrypter.h" | 15 #include "net/quic/core/crypto/quic_encrypter.h" |
16 #include "net/quic/core/quic_flags.h" | 16 #include "net/quic/core/quic_flags.h" |
17 #include "net/quic/core/quic_simple_buffer_allocator.h" | 17 #include "net/quic/core/quic_simple_buffer_allocator.h" |
18 #include "net/quic/core/quic_utils.h" | 18 #include "net/quic/core/quic_utils.h" |
19 #include "net/quic/test_tools/mock_random.h" | |
20 #include "net/quic/test_tools/quic_framer_peer.h" | 19 #include "net/quic/test_tools/quic_framer_peer.h" |
21 #include "net/quic/test_tools/quic_packet_creator_peer.h" | 20 #include "net/quic/test_tools/quic_packet_creator_peer.h" |
22 #include "net/quic/test_tools/quic_test_utils.h" | 21 #include "net/quic/test_tools/quic_test_utils.h" |
23 #include "net/test/gtest_util.h" | 22 #include "net/test/gtest_util.h" |
24 #include "testing/gmock/include/gmock/gmock.h" | 23 #include "testing/gmock/include/gmock/gmock.h" |
25 | 24 |
26 using base::StringPiece; | 25 using base::StringPiece; |
27 using std::ostream; | 26 using std::ostream; |
28 using std::string; | 27 using std::string; |
29 using testing::DoAll; | 28 using testing::DoAll; |
(...skipping 89 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
119 : server_framer_(SupportedVersions(GetParam().version), | 118 : server_framer_(SupportedVersions(GetParam().version), |
120 QuicTime::Zero(), | 119 QuicTime::Zero(), |
121 Perspective::IS_SERVER), | 120 Perspective::IS_SERVER), |
122 client_framer_(SupportedVersions(GetParam().version), | 121 client_framer_(SupportedVersions(GetParam().version), |
123 QuicTime::Zero(), | 122 QuicTime::Zero(), |
124 Perspective::IS_CLIENT), | 123 Perspective::IS_CLIENT), |
125 connection_id_(2), | 124 connection_id_(2), |
126 data_("foo"), | 125 data_("foo"), |
127 creator_(connection_id_, | 126 creator_(connection_id_, |
128 &client_framer_, | 127 &client_framer_, |
129 &mock_random_, | |
130 &buffer_allocator_, | 128 &buffer_allocator_, |
131 &delegate_), | 129 &delegate_), |
132 serialized_packet_(creator_.NoPacket()) { | 130 serialized_packet_(creator_.NoPacket()) { |
133 creator_.set_connection_id_length(GetParam().connection_id_length); | 131 creator_.set_connection_id_length(GetParam().connection_id_length); |
134 | 132 |
135 creator_.SetEncrypter(ENCRYPTION_INITIAL, new NullEncrypter()); | 133 creator_.SetEncrypter(ENCRYPTION_INITIAL, new NullEncrypter()); |
136 creator_.SetEncrypter(ENCRYPTION_FORWARD_SECURE, new NullEncrypter()); | 134 creator_.SetEncrypter(ENCRYPTION_FORWARD_SECURE, new NullEncrypter()); |
137 client_framer_.set_visitor(&framer_visitor_); | 135 client_framer_.set_visitor(&framer_visitor_); |
138 client_framer_.set_received_entropy_calculator(&entropy_calculator_); | |
139 server_framer_.set_visitor(&framer_visitor_); | 136 server_framer_.set_visitor(&framer_visitor_); |
140 } | 137 } |
141 | 138 |
142 ~QuicPacketCreatorTest() override { | 139 ~QuicPacketCreatorTest() override { |
143 delete[] serialized_packet_.encrypted_buffer; | 140 delete[] serialized_packet_.encrypted_buffer; |
144 ClearSerializedPacket(&serialized_packet_); | 141 ClearSerializedPacket(&serialized_packet_); |
145 } | 142 } |
146 | 143 |
147 SerializedPacket SerializeAllFrames(const QuicFrames& frames) { | 144 SerializedPacket SerializeAllFrames(const QuicFrames& frames) { |
148 SerializedPacket packet = QuicPacketCreatorPeer::SerializeAllFrames( | 145 SerializedPacket packet = QuicPacketCreatorPeer::SerializeAllFrames( |
(...skipping 67 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
216 QuicFlagSaver flags_; // Save/restore all QUIC flag values. | 213 QuicFlagSaver flags_; // Save/restore all QUIC flag values. |
217 char buffer_[kMaxPacketSize]; | 214 char buffer_[kMaxPacketSize]; |
218 QuicFrames frames_; | 215 QuicFrames frames_; |
219 QuicFramer server_framer_; | 216 QuicFramer server_framer_; |
220 QuicFramer client_framer_; | 217 QuicFramer client_framer_; |
221 StrictMock<MockFramerVisitor> framer_visitor_; | 218 StrictMock<MockFramerVisitor> framer_visitor_; |
222 StrictMock<MockDelegate> delegate_; | 219 StrictMock<MockDelegate> delegate_; |
223 QuicConnectionId connection_id_; | 220 QuicConnectionId connection_id_; |
224 string data_; | 221 string data_; |
225 struct iovec iov_; | 222 struct iovec iov_; |
226 MockRandom mock_random_; | |
227 SimpleBufferAllocator buffer_allocator_; | 223 SimpleBufferAllocator buffer_allocator_; |
228 QuicPacketCreator creator_; | 224 QuicPacketCreator creator_; |
229 MockEntropyCalculator entropy_calculator_; | |
230 SerializedPacket serialized_packet_; | 225 SerializedPacket serialized_packet_; |
231 }; | 226 }; |
232 | 227 |
233 // Run all packet creator tests with all supported versions of QUIC, and with | 228 // Run all packet creator tests with all supported versions of QUIC, and with |
234 // and without version in the packet header, as well as doing a run for each | 229 // and without version in the packet header, as well as doing a run for each |
235 // length of truncated connection id. | 230 // length of truncated connection id. |
236 INSTANTIATE_TEST_CASE_P(QuicPacketCreatorTests, | 231 INSTANTIATE_TEST_CASE_P(QuicPacketCreatorTests, |
237 QuicPacketCreatorTest, | 232 QuicPacketCreatorTest, |
238 ::testing::ValuesIn(GetTestParams())); | 233 ::testing::ValuesIn(GetTestParams())); |
239 | 234 |
(...skipping 580 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
820 ASSERT_FALSE(serialized_packet_.retransmittable_frames.empty()); | 815 ASSERT_FALSE(serialized_packet_.retransmittable_frames.empty()); |
821 const QuicFrames& retransmittable = serialized_packet_.retransmittable_frames; | 816 const QuicFrames& retransmittable = serialized_packet_.retransmittable_frames; |
822 ASSERT_EQ(1u, retransmittable.size()); | 817 ASSERT_EQ(1u, retransmittable.size()); |
823 EXPECT_EQ(STREAM_FRAME, retransmittable[0].type); | 818 EXPECT_EQ(STREAM_FRAME, retransmittable[0].type); |
824 ASSERT_TRUE(retransmittable[0].stream_frame); | 819 ASSERT_TRUE(retransmittable[0].stream_frame); |
825 DeleteSerializedPacket(); | 820 DeleteSerializedPacket(); |
826 | 821 |
827 EXPECT_FALSE(creator_.HasPendingFrames()); | 822 EXPECT_FALSE(creator_.HasPendingFrames()); |
828 } | 823 } |
829 | 824 |
830 TEST_P(QuicPacketCreatorTest, SerializeTruncatedAckFrameWithLargePacketSize) { | |
831 creator_.set_encryption_level(ENCRYPTION_FORWARD_SECURE); | |
832 if (!GetParam().version_serialization) { | |
833 creator_.StopSendingVersion(); | |
834 } | |
835 creator_.SetMaxPacketLength(kMaxPacketSize); | |
836 | |
837 // Serialized length of ack frame with 2000 nack ranges should be limited by | |
838 // the number of nack ranges that can be fit in an ack frame. | |
839 QuicAckFrame ack_frame = MakeAckFrameWithNackRanges(2000u, 0u); | |
840 size_t frame_len = client_framer_.GetSerializedFrameLength( | |
841 QuicFrame(&ack_frame), creator_.BytesFree(), true, true, | |
842 PACKET_1BYTE_PACKET_NUMBER); | |
843 EXPECT_GT(creator_.BytesFree(), frame_len); | |
844 EXPECT_GT(creator_.max_packet_length(), creator_.PacketSize()); | |
845 | |
846 // Add ack frame to creator. | |
847 EXPECT_TRUE(creator_.AddSavedFrame(QuicFrame(&ack_frame))); | |
848 EXPECT_TRUE(creator_.HasPendingFrames()); | |
849 EXPECT_GT(creator_.max_packet_length(), creator_.PacketSize()); | |
850 EXPECT_LT(0u, creator_.BytesFree()); | |
851 | |
852 // Make sure that an additional stream frame can be added to the packet. | |
853 QuicFrame frame; | |
854 QuicIOVector io_vector(MakeIOVector("test")); | |
855 ASSERT_TRUE( | |
856 creator_.ConsumeData(2u, io_vector, 0u, 0u, false, false, &frame)); | |
857 ASSERT_TRUE(frame.stream_frame); | |
858 size_t consumed = frame.stream_frame->data_length; | |
859 EXPECT_EQ(4u, consumed); | |
860 EXPECT_TRUE(creator_.HasPendingFrames()); | |
861 | |
862 // Ensure the packet is successfully created, and the packet size estimate | |
863 // matches the serialized packet length. | |
864 if (GetParam().version <= QUIC_VERSION_33) { | |
865 EXPECT_CALL(entropy_calculator_, EntropyHash(_)) | |
866 .WillOnce(testing::Return(0)); | |
867 } | |
868 EXPECT_CALL(delegate_, OnSerializedPacket(_)) | |
869 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); | |
870 size_t est_packet_size = creator_.PacketSize(); | |
871 creator_.Flush(); | |
872 ASSERT_TRUE(serialized_packet_.encrypted_buffer); | |
873 EXPECT_EQ(est_packet_size, client_framer_.GetMaxPlaintextSize( | |
874 serialized_packet_.encrypted_length)); | |
875 DeleteSerializedPacket(); | |
876 } | |
877 | |
878 TEST_P(QuicPacketCreatorTest, SerializeTruncatedAckFrameWithSmallPacketSize) { | |
879 if (!GetParam().version_serialization) { | |
880 creator_.StopSendingVersion(); | |
881 } | |
882 creator_.SetMaxPacketLength(500u); | |
883 | |
884 const size_t max_plaintext_size = | |
885 client_framer_.GetMaxPlaintextSize(creator_.max_packet_length()); | |
886 EXPECT_EQ(max_plaintext_size - creator_.PacketSize(), creator_.BytesFree()); | |
887 | |
888 // Serialized length of ack frame with 2000 nack ranges should be limited by | |
889 // the packet size. | |
890 QuicAckFrame ack_frame = MakeAckFrameWithNackRanges(2000u, 0u); | |
891 size_t frame_len = client_framer_.GetSerializedFrameLength( | |
892 QuicFrame(&ack_frame), creator_.BytesFree(), true, true, | |
893 PACKET_1BYTE_PACKET_NUMBER); | |
894 EXPECT_EQ(creator_.BytesFree(), frame_len); | |
895 | |
896 // Add ack frame to creator. | |
897 EXPECT_TRUE(creator_.AddSavedFrame(QuicFrame(&ack_frame))); | |
898 EXPECT_TRUE(creator_.HasPendingFrames()); | |
899 EXPECT_EQ(client_framer_.GetMaxPlaintextSize(creator_.max_packet_length()), | |
900 creator_.PacketSize()); | |
901 EXPECT_EQ(0u, creator_.BytesFree()); | |
902 | |
903 // Ensure the packet is successfully created, and the packet size estimate | |
904 // may not match the serialized packet length. | |
905 if (GetParam().version <= QUIC_VERSION_33) { | |
906 EXPECT_CALL(entropy_calculator_, EntropyHash(_)).WillOnce(Return(0)); | |
907 } | |
908 size_t est_packet_size = creator_.PacketSize(); | |
909 EXPECT_CALL(delegate_, OnSerializedPacket(_)) | |
910 .WillOnce(Invoke(this, &QuicPacketCreatorTest::SaveSerializedPacket)); | |
911 creator_.Flush(); | |
912 ASSERT_TRUE(serialized_packet_.encrypted_buffer); | |
913 EXPECT_GE(est_packet_size, client_framer_.GetMaxPlaintextSize( | |
914 serialized_packet_.encrypted_length)); | |
915 DeleteSerializedPacket(); | |
916 } | |
917 | |
918 TEST_P(QuicPacketCreatorTest, EntropyFlag) { | |
919 frames_.push_back(QuicFrame( | |
920 new QuicStreamFrame(kCryptoStreamId, false, 0u, StringPiece()))); | |
921 | |
922 for (int i = 0; i < 2; ++i) { | |
923 for (int j = 0; j < 64; ++j) { | |
924 SerializedPacket serialized = SerializeAllFrames(frames_); | |
925 // Verify both BoolSource and hash algorithm. | |
926 bool expected_rand_bool = | |
927 (mock_random_.RandUint64() & (UINT64_C(1) << j)) != 0; | |
928 bool observed_rand_bool = | |
929 (serialized.entropy_hash & (1 << ((j + 1) % 8))) != 0; | |
930 uint8_t rest_of_hash = serialized.entropy_hash & ~(1 << ((j + 1) % 8)); | |
931 EXPECT_EQ(expected_rand_bool, observed_rand_bool); | |
932 EXPECT_EQ(0, rest_of_hash); | |
933 } | |
934 // After 64 calls, BoolSource will refresh the bucket - make sure it does. | |
935 mock_random_.ChangeValue(); | |
936 } | |
937 | |
938 delete frames_[0].stream_frame; | |
939 } | |
940 | |
941 TEST_P(QuicPacketCreatorTest, SetCurrentPath) { | 825 TEST_P(QuicPacketCreatorTest, SetCurrentPath) { |
942 // Current path is the default path. | 826 // Current path is the default path. |
943 EXPECT_EQ(kDefaultPathId, QuicPacketCreatorPeer::GetCurrentPath(&creator_)); | 827 EXPECT_EQ(kDefaultPathId, QuicPacketCreatorPeer::GetCurrentPath(&creator_)); |
944 EXPECT_EQ(0u, creator_.packet_number()); | 828 EXPECT_EQ(0u, creator_.packet_number()); |
945 EXPECT_EQ(PACKET_1BYTE_PACKET_NUMBER, | 829 EXPECT_EQ(PACKET_1BYTE_PACKET_NUMBER, |
946 QuicPacketCreatorPeer::GetPacketNumberLength(&creator_)); | 830 QuicPacketCreatorPeer::GetPacketNumberLength(&creator_)); |
947 // Add a stream frame to the creator. | 831 // Add a stream frame to the creator. |
948 QuicFrame frame; | 832 QuicFrame frame; |
949 QuicIOVector io_vector(MakeIOVector("test")); | 833 QuicIOVector io_vector(MakeIOVector("test")); |
950 ASSERT_TRUE(creator_.ConsumeData(kCryptoStreamId, io_vector, 0u, 0u, false, | 834 ASSERT_TRUE(creator_.ConsumeData(kCryptoStreamId, io_vector, 0u, 0u, false, |
(...skipping 114 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1065 EXPECT_CALL(delegate_, | 949 EXPECT_CALL(delegate_, |
1066 OnUnrecoverableError(QUIC_CRYPTO_CHLO_TOO_LARGE, _, _)); | 950 OnUnrecoverableError(QUIC_CRYPTO_CHLO_TOO_LARGE, _, _)); |
1067 EXPECT_QUIC_BUG(creator_.ConsumeData(kCryptoStreamId, data_iovec, 0u, 0u, | 951 EXPECT_QUIC_BUG(creator_.ConsumeData(kCryptoStreamId, data_iovec, 0u, 0u, |
1068 false, false, &frame), | 952 false, false, &frame), |
1069 "Client hello won't fit in a single packet."); | 953 "Client hello won't fit in a single packet."); |
1070 } | 954 } |
1071 | 955 |
1072 } // namespace | 956 } // namespace |
1073 } // namespace test | 957 } // namespace test |
1074 } // namespace net | 958 } // namespace net |
OLD | NEW |