net/quic/test_tools/simple_quic_framer.h

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