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/test_tools/quic_test_utils.h" | 5 #include "net/quic/test_tools/quic_test_utils.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 #include <memory> | 8 #include <memory> |
9 | 9 |
10 #include "net/quic/core/crypto/crypto_framer.h" | 10 #include "net/quic/core/crypto/crypto_framer.h" |
11 #include "net/quic/core/crypto/crypto_handshake.h" | 11 #include "net/quic/core/crypto/crypto_handshake.h" |
12 #include "net/quic/core/crypto/crypto_utils.h" | 12 #include "net/quic/core/crypto/crypto_utils.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_data_writer.h" | 16 #include "net/quic/core/quic_data_writer.h" |
17 #include "net/quic/core/quic_flags.h" | 17 #include "net/quic/core/quic_flags.h" |
18 #include "net/quic/core/quic_framer.h" | 18 #include "net/quic/core/quic_framer.h" |
19 #include "net/quic/core/quic_packet_creator.h" | 19 #include "net/quic/core/quic_packet_creator.h" |
20 #include "net/quic/core/quic_utils.h" | 20 #include "net/quic/core/quic_utils.h" |
21 #include "net/quic/platform/api/quic_logging.h" | 21 #include "net/quic/platform/api/quic_logging.h" |
22 #include "net/quic/platform/api/quic_ptr_util.h" | 22 #include "net/quic/platform/api/quic_ptr_util.h" |
23 #include "net/quic/test_tools/crypto_test_utils.h" | 23 #include "net/quic/test_tools/crypto_test_utils.h" |
24 #include "net/quic/test_tools/quic_connection_peer.h" | 24 #include "net/quic/test_tools/quic_connection_peer.h" |
25 #include "net/spdy/spdy_frame_builder.h" | 25 #include "net/spdy/spdy_frame_builder.h" |
26 #include "third_party/boringssl/src/include/openssl/sha.h" | 26 #include "third_party/boringssl/src/include/openssl/sha.h" |
27 | 27 |
28 using base::StringPiece; | |
29 using std::string; | 28 using std::string; |
30 using testing::_; | 29 using testing::_; |
31 using testing::Invoke; | 30 using testing::Invoke; |
32 | 31 |
33 namespace net { | 32 namespace net { |
34 namespace test { | 33 namespace test { |
35 | 34 |
36 QuicAckFrame MakeAckFrame(QuicPacketNumber largest_observed) { | 35 QuicAckFrame MakeAckFrame(QuicPacketNumber largest_observed) { |
37 QuicAckFrame ack; | 36 QuicAckFrame ack; |
38 ack.largest_observed = largest_observed; | 37 ack.largest_observed = largest_observed; |
(...skipping 51 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
90 #include "net/quic/core/quic_flags_list.h" | 89 #include "net/quic/core/quic_flags_list.h" |
91 #undef QUIC_FLAG | 90 #undef QUIC_FLAG |
92 } | 91 } |
93 | 92 |
94 QuicFlagSaver::~QuicFlagSaver() { | 93 QuicFlagSaver::~QuicFlagSaver() { |
95 #define QUIC_FLAG(type, flag, value) base::SetFlag(&flag, value); | 94 #define QUIC_FLAG(type, flag, value) base::SetFlag(&flag, value); |
96 #include "net/quic/core/quic_flags_list.h" | 95 #include "net/quic/core/quic_flags_list.h" |
97 #undef QUIC_FLAG | 96 #undef QUIC_FLAG |
98 } | 97 } |
99 | 98 |
100 string Sha1Hash(StringPiece data) { | 99 string Sha1Hash(QuicStringPiece data) { |
101 char buffer[SHA_DIGEST_LENGTH]; | 100 char buffer[SHA_DIGEST_LENGTH]; |
102 SHA1(reinterpret_cast<const uint8_t*>(data.data()), data.size(), | 101 SHA1(reinterpret_cast<const uint8_t*>(data.data()), data.size(), |
103 reinterpret_cast<uint8_t*>(buffer)); | 102 reinterpret_cast<uint8_t*>(buffer)); |
104 return string(buffer, arraysize(buffer)); | 103 return string(buffer, arraysize(buffer)); |
105 } | 104 } |
106 | 105 |
107 uint64_t SimpleRandom::RandUint64() { | 106 uint64_t SimpleRandom::RandUint64() { |
108 string hash = | 107 string hash = |
109 Sha1Hash(StringPiece(reinterpret_cast<char*>(&seed_), sizeof(seed_))); | 108 Sha1Hash(QuicStringPiece(reinterpret_cast<char*>(&seed_), sizeof(seed_))); |
110 DCHECK_EQ(static_cast<size_t>(SHA_DIGEST_LENGTH), hash.length()); | 109 DCHECK_EQ(static_cast<size_t>(SHA_DIGEST_LENGTH), hash.length()); |
111 memcpy(&seed_, hash.data(), sizeof(seed_)); | 110 memcpy(&seed_, hash.data(), sizeof(seed_)); |
112 return seed_; | 111 return seed_; |
113 } | 112 } |
114 | 113 |
115 void SimpleRandom::RandBytes(void* data, size_t len) { | 114 void SimpleRandom::RandBytes(void* data, size_t len) { |
116 uint8_t* real_data = static_cast<uint8_t*>(data); | 115 uint8_t* real_data = static_cast<uint8_t*>(data); |
117 for (size_t offset = 0; offset < len; offset++) { | 116 for (size_t offset = 0; offset < len; offset++) { |
118 real_data[offset] = RandUint64() & 0xff; | 117 real_data[offset] = RandUint64() & 0xff; |
119 } | 118 } |
(...skipping 485 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
605 QuicVersionVector* versions, | 604 QuicVersionVector* versions, |
606 Perspective perspective) { | 605 Perspective perspective) { |
607 QuicPacketHeader header; | 606 QuicPacketHeader header; |
608 header.public_header.connection_id = connection_id; | 607 header.public_header.connection_id = connection_id; |
609 header.public_header.connection_id_length = connection_id_length; | 608 header.public_header.connection_id_length = connection_id_length; |
610 header.public_header.version_flag = version_flag; | 609 header.public_header.version_flag = version_flag; |
611 header.public_header.multipath_flag = multipath_flag; | 610 header.public_header.multipath_flag = multipath_flag; |
612 header.public_header.reset_flag = reset_flag; | 611 header.public_header.reset_flag = reset_flag; |
613 header.public_header.packet_number_length = packet_number_length; | 612 header.public_header.packet_number_length = packet_number_length; |
614 header.packet_number = packet_number; | 613 header.packet_number = packet_number; |
615 QuicStreamFrame stream_frame(1, false, 0, StringPiece(data)); | 614 QuicStreamFrame stream_frame(1, false, 0, QuicStringPiece(data)); |
616 QuicFrame frame(&stream_frame); | 615 QuicFrame frame(&stream_frame); |
617 QuicFrames frames; | 616 QuicFrames frames; |
618 frames.push_back(frame); | 617 frames.push_back(frame); |
619 QuicFramer framer( | 618 QuicFramer framer( |
620 versions != nullptr ? *versions : CurrentSupportedVersions(), | 619 versions != nullptr ? *versions : CurrentSupportedVersions(), |
621 QuicTime::Zero(), perspective); | 620 QuicTime::Zero(), perspective); |
622 | 621 |
623 std::unique_ptr<QuicPacket> packet( | 622 std::unique_ptr<QuicPacket> packet( |
624 BuildUnsizedDataPacket(&framer, header, frames)); | 623 BuildUnsizedDataPacket(&framer, header, frames)); |
625 EXPECT_TRUE(packet != nullptr); | 624 EXPECT_TRUE(packet != nullptr); |
(...skipping 23 matching lines...) Expand all Loading... |
649 QuicPacketNumberLength packet_number_length, | 648 QuicPacketNumberLength packet_number_length, |
650 QuicVersionVector* versions, | 649 QuicVersionVector* versions, |
651 Perspective perspective) { | 650 Perspective perspective) { |
652 QuicPacketHeader header; | 651 QuicPacketHeader header; |
653 header.public_header.connection_id = connection_id; | 652 header.public_header.connection_id = connection_id; |
654 header.public_header.connection_id_length = connection_id_length; | 653 header.public_header.connection_id_length = connection_id_length; |
655 header.public_header.version_flag = version_flag; | 654 header.public_header.version_flag = version_flag; |
656 header.public_header.reset_flag = reset_flag; | 655 header.public_header.reset_flag = reset_flag; |
657 header.public_header.packet_number_length = packet_number_length; | 656 header.public_header.packet_number_length = packet_number_length; |
658 header.packet_number = packet_number; | 657 header.packet_number = packet_number; |
659 QuicStreamFrame stream_frame(1, false, 0, StringPiece(data)); | 658 QuicStreamFrame stream_frame(1, false, 0, QuicStringPiece(data)); |
660 QuicFrame frame(&stream_frame); | 659 QuicFrame frame(&stream_frame); |
661 QuicFrames frames; | 660 QuicFrames frames; |
662 frames.push_back(frame); | 661 frames.push_back(frame); |
663 QuicFramer framer(versions != nullptr ? *versions : AllSupportedVersions(), | 662 QuicFramer framer(versions != nullptr ? *versions : AllSupportedVersions(), |
664 QuicTime::Zero(), perspective); | 663 QuicTime::Zero(), perspective); |
665 | 664 |
666 std::unique_ptr<QuicPacket> packet( | 665 std::unique_ptr<QuicPacket> packet( |
667 BuildUnsizedDataPacket(&framer, header, frames)); | 666 BuildUnsizedDataPacket(&framer, header, frames)); |
668 EXPECT_TRUE(packet != nullptr); | 667 EXPECT_TRUE(packet != nullptr); |
669 | 668 |
(...skipping 159 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
829 // strike register worries that we've just overflowed a uint32_t time. | 828 // strike register worries that we've just overflowed a uint32_t time. |
830 (*server_connection)->AdvanceTime(connection_start_time); | 829 (*server_connection)->AdvanceTime(connection_start_time); |
831 } | 830 } |
832 | 831 |
833 QuicStreamId QuicClientDataStreamId(int i) { | 832 QuicStreamId QuicClientDataStreamId(int i) { |
834 return kClientDataStreamId1 + 2 * i; | 833 return kClientDataStreamId1 + 2 * i; |
835 } | 834 } |
836 | 835 |
837 } // namespace test | 836 } // namespace test |
838 } // namespace net | 837 } // namespace net |
OLD | NEW |