| 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_generator.h" | 5 #include "net/quic/core/quic_packet_generator.h" |
| 6 | 6 |
| 7 #include <cstdint> | 7 #include <cstdint> |
| 8 #include <memory> | 8 #include <memory> |
| 9 #include <string> | 9 #include <string> |
| 10 | 10 |
| (...skipping 189 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 200 memset(data_array_.get(), '?', len); | 200 memset(data_array_.get(), '?', len); |
| 201 iov_.iov_base = data_array_.get(); | 201 iov_.iov_base = data_array_.get(); |
| 202 iov_.iov_len = len; | 202 iov_.iov_len = len; |
| 203 return QuicIOVector(&iov_, 1, len); | 203 return QuicIOVector(&iov_, 1, len); |
| 204 } | 204 } |
| 205 | 205 |
| 206 QuicIOVector MakeIOVector(StringPiece s) { | 206 QuicIOVector MakeIOVector(StringPiece s) { |
| 207 return ::net::MakeIOVector(s, &iov_); | 207 return ::net::MakeIOVector(s, &iov_); |
| 208 } | 208 } |
| 209 | 209 |
| 210 QuicFlagSaver flags_; // Save/restore all QUIC flag values. | |
| 211 QuicFramer framer_; | 210 QuicFramer framer_; |
| 212 MockRandom random_; | 211 MockRandom random_; |
| 213 SimpleBufferAllocator buffer_allocator_; | 212 SimpleBufferAllocator buffer_allocator_; |
| 214 StrictMock<MockDelegate> delegate_; | 213 StrictMock<MockDelegate> delegate_; |
| 215 QuicPacketGenerator generator_; | 214 QuicPacketGenerator generator_; |
| 216 QuicPacketCreator* creator_; | 215 QuicPacketCreator* creator_; |
| 217 SimpleQuicFramer simple_framer_; | 216 SimpleQuicFramer simple_framer_; |
| 218 vector<SerializedPacket> packets_; | 217 vector<SerializedPacket> packets_; |
| 219 QuicAckFrame ack_frame_; | 218 QuicAckFrame ack_frame_; |
| 220 | 219 |
| (...skipping 651 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 872 EXPECT_CALL(delegate_, OnSerializedPacket(_)) | 871 EXPECT_CALL(delegate_, OnSerializedPacket(_)) |
| 873 .WillOnce(Invoke(this, &QuicPacketGeneratorTest::SavePacket)); | 872 .WillOnce(Invoke(this, &QuicPacketGeneratorTest::SavePacket)); |
| 874 generator_.FlushAllQueuedFrames(); | 873 generator_.FlushAllQueuedFrames(); |
| 875 EXPECT_FALSE(generator_.HasQueuedFrames()); | 874 EXPECT_FALSE(generator_.HasQueuedFrames()); |
| 876 generator_.SetCurrentPath(kTestPathId1, 1, 0); | 875 generator_.SetCurrentPath(kTestPathId1, 1, 0); |
| 877 EXPECT_EQ(kTestPathId1, QuicPacketCreatorPeer::GetCurrentPath(creator_)); | 876 EXPECT_EQ(kTestPathId1, QuicPacketCreatorPeer::GetCurrentPath(creator_)); |
| 878 } | 877 } |
| 879 | 878 |
| 880 // Regression test for b/31486443. | 879 // Regression test for b/31486443. |
| 881 TEST_F(QuicPacketGeneratorTest, ConnectionCloseFrameLargerThanPacketSize) { | 880 TEST_F(QuicPacketGeneratorTest, ConnectionCloseFrameLargerThanPacketSize) { |
| 882 FLAGS_quic_close_connection_on_huge_frames = true; | |
| 883 delegate_.SetCanWriteAnything(); | 881 delegate_.SetCanWriteAnything(); |
| 884 QuicConnectionCloseFrame* frame = new QuicConnectionCloseFrame(); | 882 QuicConnectionCloseFrame* frame = new QuicConnectionCloseFrame(); |
| 885 frame->error_code = QUIC_PACKET_WRITE_ERROR; | 883 frame->error_code = QUIC_PACKET_WRITE_ERROR; |
| 886 char buf[2000]; | 884 char buf[2000]; |
| 887 StringPiece error_details(buf, 2000); | 885 StringPiece error_details(buf, 2000); |
| 888 frame->error_details = error_details.as_string(); | 886 frame->error_details = error_details.as_string(); |
| 889 EXPECT_CALL(delegate_, | 887 EXPECT_CALL(delegate_, |
| 890 OnUnrecoverableError(QUIC_FAILED_TO_SERIALIZE_PACKET, | 888 OnUnrecoverableError(QUIC_FAILED_TO_SERIALIZE_PACKET, |
| 891 "Single frame cannot fit into a packet", _)); | 889 "Single frame cannot fit into a packet", _)); |
| 892 EXPECT_QUIC_BUG(generator_.AddControlFrame(QuicFrame(frame)), ""); | 890 EXPECT_QUIC_BUG(generator_.AddControlFrame(QuicFrame(frame)), ""); |
| 893 EXPECT_TRUE(generator_.HasQueuedFrames()); | 891 EXPECT_TRUE(generator_.HasQueuedFrames()); |
| 894 } | 892 } |
| 895 | 893 |
| 896 } // namespace test | 894 } // namespace test |
| 897 } // namespace net | 895 } // namespace net |
| OLD | NEW |