| 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/simple_quic_framer.h" | 5 #include "net/quic/test_tools/simple_quic_framer.h" |
| 6 | 6 |
| 7 #include "base/stl_util.h" | 7 #include "base/stl_util.h" |
| 8 #include "net/quic/crypto/crypto_framer.h" | 8 #include "net/quic/crypto/crypto_framer.h" |
| 9 #include "net/quic/crypto/quic_decrypter.h" | 9 #include "net/quic/crypto/quic_decrypter.h" |
| 10 #include "net/quic/crypto/quic_encrypter.h" | 10 #include "net/quic/crypto/quic_encrypter.h" |
| 11 | 11 |
| 12 using base::StringPiece; | 12 using base::StringPiece; |
| 13 using std::string; | 13 using std::string; |
| 14 using std::vector; | 14 using std::vector; |
| 15 | 15 |
| 16 namespace net { | 16 namespace net { |
| 17 namespace test { | 17 namespace test { |
| 18 | 18 |
| 19 class SimpleFramerVisitor : public QuicFramerVisitorInterface { | 19 class SimpleFramerVisitor : public QuicFramerVisitorInterface { |
| 20 public: | 20 public: |
| 21 SimpleFramerVisitor() | 21 SimpleFramerVisitor() |
| 22 : error_(QUIC_NO_ERROR) { | 22 : error_(QUIC_NO_ERROR) { |
| 23 } | 23 } |
| 24 | 24 |
| 25 virtual ~SimpleFramerVisitor() override { | 25 ~SimpleFramerVisitor() override { STLDeleteElements(&stream_data_); } |
| 26 STLDeleteElements(&stream_data_); | |
| 27 } | |
| 28 | 26 |
| 29 virtual void OnError(QuicFramer* framer) override { | 27 void OnError(QuicFramer* framer) override { error_ = framer->error(); } |
| 30 error_ = framer->error(); | |
| 31 } | |
| 32 | 28 |
| 33 virtual bool OnProtocolVersionMismatch(QuicVersion version) override { | 29 bool OnProtocolVersionMismatch(QuicVersion version) override { return false; } |
| 34 return false; | |
| 35 } | |
| 36 | 30 |
| 37 virtual void OnPacket() override {} | 31 void OnPacket() override {} |
| 38 virtual void OnPublicResetPacket( | 32 void OnPublicResetPacket(const QuicPublicResetPacket& packet) override { |
| 39 const QuicPublicResetPacket& packet) override { | |
| 40 public_reset_packet_.reset(new QuicPublicResetPacket(packet)); | 33 public_reset_packet_.reset(new QuicPublicResetPacket(packet)); |
| 41 } | 34 } |
| 42 virtual void OnVersionNegotiationPacket( | 35 void OnVersionNegotiationPacket( |
| 43 const QuicVersionNegotiationPacket& packet) override { | 36 const QuicVersionNegotiationPacket& packet) override { |
| 44 version_negotiation_packet_.reset( | 37 version_negotiation_packet_.reset( |
| 45 new QuicVersionNegotiationPacket(packet)); | 38 new QuicVersionNegotiationPacket(packet)); |
| 46 } | 39 } |
| 47 virtual void OnRevivedPacket() override {} | 40 void OnRevivedPacket() override {} |
| 48 | 41 |
| 49 virtual bool OnUnauthenticatedPublicHeader( | 42 bool OnUnauthenticatedPublicHeader( |
| 50 const QuicPacketPublicHeader& header) override { | 43 const QuicPacketPublicHeader& header) override { |
| 51 return true; | 44 return true; |
| 52 } | 45 } |
| 53 virtual bool OnUnauthenticatedHeader( | 46 bool OnUnauthenticatedHeader(const QuicPacketHeader& header) override { |
| 54 const QuicPacketHeader& header) override { | |
| 55 return true; | 47 return true; |
| 56 } | 48 } |
| 57 virtual void OnDecryptedPacket(EncryptionLevel level) override {} | 49 void OnDecryptedPacket(EncryptionLevel level) override {} |
| 58 virtual bool OnPacketHeader(const QuicPacketHeader& header) override { | 50 bool OnPacketHeader(const QuicPacketHeader& header) override { |
| 59 has_header_ = true; | 51 has_header_ = true; |
| 60 header_ = header; | 52 header_ = header; |
| 61 return true; | 53 return true; |
| 62 } | 54 } |
| 63 | 55 |
| 64 virtual void OnFecProtectedPayload(StringPiece payload) override {} | 56 void OnFecProtectedPayload(StringPiece payload) override {} |
| 65 | 57 |
| 66 virtual bool OnStreamFrame(const QuicStreamFrame& frame) override { | 58 bool OnStreamFrame(const QuicStreamFrame& frame) override { |
| 67 // Save a copy of the data so it is valid after the packet is processed. | 59 // Save a copy of the data so it is valid after the packet is processed. |
| 68 stream_data_.push_back(frame.GetDataAsString()); | 60 stream_data_.push_back(frame.GetDataAsString()); |
| 69 QuicStreamFrame stream_frame(frame); | 61 QuicStreamFrame stream_frame(frame); |
| 70 // Make sure that the stream frame points to this data. | 62 // Make sure that the stream frame points to this data. |
| 71 stream_frame.data.Clear(); | 63 stream_frame.data.Clear(); |
| 72 stream_frame.data.Append(const_cast<char*>(stream_data_.back()->data()), | 64 stream_frame.data.Append(const_cast<char*>(stream_data_.back()->data()), |
| 73 stream_data_.back()->size()); | 65 stream_data_.back()->size()); |
| 74 stream_frames_.push_back(stream_frame); | 66 stream_frames_.push_back(stream_frame); |
| 75 return true; | 67 return true; |
| 76 } | 68 } |
| 77 | 69 |
| 78 virtual bool OnAckFrame(const QuicAckFrame& frame) override { | 70 bool OnAckFrame(const QuicAckFrame& frame) override { |
| 79 ack_frames_.push_back(frame); | 71 ack_frames_.push_back(frame); |
| 80 return true; | 72 return true; |
| 81 } | 73 } |
| 82 | 74 |
| 83 virtual bool OnCongestionFeedbackFrame( | 75 bool OnCongestionFeedbackFrame( |
| 84 const QuicCongestionFeedbackFrame& frame) override { | 76 const QuicCongestionFeedbackFrame& frame) override { |
| 85 feedback_frames_.push_back(frame); | 77 feedback_frames_.push_back(frame); |
| 86 return true; | 78 return true; |
| 87 } | 79 } |
| 88 | 80 |
| 89 virtual bool OnStopWaitingFrame(const QuicStopWaitingFrame& frame) override { | 81 bool OnStopWaitingFrame(const QuicStopWaitingFrame& frame) override { |
| 90 stop_waiting_frames_.push_back(frame); | 82 stop_waiting_frames_.push_back(frame); |
| 91 return true; | 83 return true; |
| 92 } | 84 } |
| 93 | 85 |
| 94 virtual bool OnPingFrame(const QuicPingFrame& frame) override { | 86 bool OnPingFrame(const QuicPingFrame& frame) override { |
| 95 ping_frames_.push_back(frame); | 87 ping_frames_.push_back(frame); |
| 96 return true; | 88 return true; |
| 97 } | 89 } |
| 98 | 90 |
| 99 virtual void OnFecData(const QuicFecData& fec) override { | 91 void OnFecData(const QuicFecData& fec) override { |
| 100 fec_data_ = fec; | 92 fec_data_ = fec; |
| 101 fec_redundancy_ = fec_data_.redundancy.as_string(); | 93 fec_redundancy_ = fec_data_.redundancy.as_string(); |
| 102 fec_data_.redundancy = fec_redundancy_; | 94 fec_data_.redundancy = fec_redundancy_; |
| 103 } | 95 } |
| 104 | 96 |
| 105 virtual bool OnRstStreamFrame(const QuicRstStreamFrame& frame) override { | 97 bool OnRstStreamFrame(const QuicRstStreamFrame& frame) override { |
| 106 rst_stream_frames_.push_back(frame); | 98 rst_stream_frames_.push_back(frame); |
| 107 return true; | 99 return true; |
| 108 } | 100 } |
| 109 | 101 |
| 110 virtual bool OnConnectionCloseFrame( | 102 bool OnConnectionCloseFrame(const QuicConnectionCloseFrame& frame) override { |
| 111 const QuicConnectionCloseFrame& frame) override { | |
| 112 connection_close_frames_.push_back(frame); | 103 connection_close_frames_.push_back(frame); |
| 113 return true; | 104 return true; |
| 114 } | 105 } |
| 115 | 106 |
| 116 virtual bool OnGoAwayFrame(const QuicGoAwayFrame& frame) override { | 107 bool OnGoAwayFrame(const QuicGoAwayFrame& frame) override { |
| 117 goaway_frames_.push_back(frame); | 108 goaway_frames_.push_back(frame); |
| 118 return true; | 109 return true; |
| 119 } | 110 } |
| 120 | 111 |
| 121 virtual bool OnWindowUpdateFrame( | 112 bool OnWindowUpdateFrame(const QuicWindowUpdateFrame& frame) override { |
| 122 const QuicWindowUpdateFrame& frame) override { | |
| 123 window_update_frames_.push_back(frame); | 113 window_update_frames_.push_back(frame); |
| 124 return true; | 114 return true; |
| 125 } | 115 } |
| 126 | 116 |
| 127 virtual bool OnBlockedFrame(const QuicBlockedFrame& frame) override { | 117 bool OnBlockedFrame(const QuicBlockedFrame& frame) override { |
| 128 blocked_frames_.push_back(frame); | 118 blocked_frames_.push_back(frame); |
| 129 return true; | 119 return true; |
| 130 } | 120 } |
| 131 | 121 |
| 132 virtual void OnPacketComplete() override {} | 122 void OnPacketComplete() override {} |
| 133 | 123 |
| 134 const QuicPacketHeader& header() const { return header_; } | 124 const QuicPacketHeader& header() const { return header_; } |
| 135 const vector<QuicAckFrame>& ack_frames() const { return ack_frames_; } | 125 const vector<QuicAckFrame>& ack_frames() const { return ack_frames_; } |
| 136 const vector<QuicConnectionCloseFrame>& connection_close_frames() const { | 126 const vector<QuicConnectionCloseFrame>& connection_close_frames() const { |
| 137 return connection_close_frames_; | 127 return connection_close_frames_; |
| 138 } | 128 } |
| 139 const vector<QuicCongestionFeedbackFrame>& feedback_frames() const { | 129 const vector<QuicCongestionFeedbackFrame>& feedback_frames() const { |
| 140 return feedback_frames_; | 130 return feedback_frames_; |
| 141 } | 131 } |
| 142 const vector<QuicGoAwayFrame>& goaway_frames() const { | 132 const vector<QuicGoAwayFrame>& goaway_frames() const { |
| (...skipping 135 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 278 return visitor_->goaway_frames(); | 268 return visitor_->goaway_frames(); |
| 279 } | 269 } |
| 280 | 270 |
| 281 const vector<QuicConnectionCloseFrame>& | 271 const vector<QuicConnectionCloseFrame>& |
| 282 SimpleQuicFramer::connection_close_frames() const { | 272 SimpleQuicFramer::connection_close_frames() const { |
| 283 return visitor_->connection_close_frames(); | 273 return visitor_->connection_close_frames(); |
| 284 } | 274 } |
| 285 | 275 |
| 286 } // namespace test | 276 } // namespace test |
| 287 } // namespace net | 277 } // namespace net |
| OLD | NEW |