| 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 |