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 3 // found in the LICENSE file. 4 5 #ifndef NET_QUIC_TEST_TOOLS_SIMPLE_QUIC_FRAMER_H_ 6 #define NET_QUIC_TEST_TOOLS_SIMPLE_QUIC_FRAMER_H_ 7 8 #include <vector> 9 10 #include "base/basictypes.h" 11 #include "base/memory/scoped_ptr.h" 12 #include "net/quic/quic_framer.h" 13 #include "net/quic/quic_protocol.h" 14 15 namespace net { 16 17 class CryptoHandshakeMessage; 18 struct QuicAckFrame; 19 class QuicConnection; 20 class QuicConnectionVisitorInterface; 21 class QuicPacketCreator; 22 class ReceiveAlgorithmInterface; 23 class SendAlgorithmInterface; 24 25 namespace test { 26 27 class SimpleFramerVisitor; 28 29 // Peer to make public a number of otherwise private QuicFramer methods. 30 class SimpleQuicFramer { 31 public: 32 SimpleQuicFramer(); 33 explicit SimpleQuicFramer(const QuicVersionVector& supported_versions); 34 ~SimpleQuicFramer(); 35 36 bool ProcessPacket(const QuicEncryptedPacket& packet); 37 bool ProcessPacket(const QuicPacket& packet); 38 void Reset(); 39 40 const QuicPacketHeader& header() const; 41 size_t num_frames() const; 42 const std::vector<QuicAckFrame>& ack_frames() const; 43 const std::vector<QuicConnectionCloseFrame>& connection_close_frames() const; 44 const std::vector<QuicCongestionFeedbackFrame>& feedback_frames() const; 45 const std::vector<QuicStopWaitingFrame>& stop_waiting_frames() const; 46 const std::vector<QuicPingFrame>& ping_frames() const; 47 const std::vector<QuicGoAwayFrame>& goaway_frames() const; 48 const std::vector<QuicRstStreamFrame>& rst_stream_frames() const; 49 const std::vector<QuicStreamFrame>& stream_frames() const; 50 const QuicFecData& fec_data() const; 51 const QuicVersionNegotiationPacket* version_negotiation_packet() const; 52 const QuicPublicResetPacket* public_reset_packet() const; 53 54 QuicFramer* framer(); 55 SetSupportedVersions(const QuicVersionVector & versions)56 void SetSupportedVersions(const QuicVersionVector& versions) { 57 framer_.SetSupportedVersions(versions); 58 } 59 60 private: 61 QuicFramer framer_; 62 scoped_ptr<SimpleFramerVisitor> visitor_; 63 DISALLOW_COPY_AND_ASSIGN(SimpleQuicFramer); 64 }; 65 66 } // namespace test 67 68 } // namespace net 69 70 #endif // NET_QUIC_TEST_TOOLS_SIMPLE_QUIC_FRAMER_H_ 71