OLD | NEW |
| (Empty) |
1 // Copyright 2013 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 // Provides a simple interface for QUIC tests to create a variety of packets. | |
6 | |
7 #ifndef NET_QUIC_TEST_TOOLS_QUIC_TEST_PACKET_MAKER_H_ | |
8 #define NET_QUIC_TEST_TOOLS_QUIC_TEST_PACKET_MAKER_H_ | |
9 | |
10 #include "base/memory/scoped_ptr.h" | |
11 #include "net/base/request_priority.h" | |
12 #include "net/quic/quic_protocol.h" | |
13 #include "net/quic/test_tools/mock_clock.h" | |
14 #include "net/quic/test_tools/mock_random.h" | |
15 #include "net/spdy/spdy_framer.h" | |
16 #include "net/spdy/spdy_protocol.h" | |
17 | |
18 namespace net { | |
19 namespace test { | |
20 | |
21 class QuicTestPacketMaker { | |
22 public: | |
23 QuicTestPacketMaker(QuicVersion version, | |
24 QuicConnectionId connection_id, | |
25 MockClock* clock); | |
26 ~QuicTestPacketMaker(); | |
27 | |
28 scoped_ptr<QuicEncryptedPacket> MakeRstPacket( | |
29 QuicPacketSequenceNumber num, | |
30 bool include_version, | |
31 QuicStreamId stream_id, | |
32 QuicRstStreamErrorCode error_code); | |
33 scoped_ptr<QuicEncryptedPacket> MakeAckAndRstPacket( | |
34 QuicPacketSequenceNumber num, | |
35 bool include_version, | |
36 QuicStreamId stream_id, | |
37 QuicRstStreamErrorCode error_code, | |
38 QuicPacketSequenceNumber largest_received, | |
39 QuicPacketSequenceNumber least_unacked, | |
40 bool send_feedback); | |
41 scoped_ptr<QuicEncryptedPacket> MakeConnectionClosePacket( | |
42 QuicPacketSequenceNumber num); | |
43 scoped_ptr<QuicEncryptedPacket> MakeAckPacket( | |
44 QuicPacketSequenceNumber sequence_number, | |
45 QuicPacketSequenceNumber largest_received, | |
46 QuicPacketSequenceNumber least_unacked, | |
47 bool send_feedback); | |
48 scoped_ptr<QuicEncryptedPacket> MakeDataPacket( | |
49 QuicPacketSequenceNumber sequence_number, | |
50 QuicStreamId stream_id, | |
51 bool should_include_version, | |
52 bool fin, | |
53 QuicStreamOffset offset, | |
54 base::StringPiece data); | |
55 scoped_ptr<QuicEncryptedPacket> MakeRequestHeadersPacket( | |
56 QuicPacketSequenceNumber sequence_number, | |
57 QuicStreamId stream_id, | |
58 bool should_include_version, | |
59 bool fin, | |
60 QuicPriority priority, | |
61 const SpdyHeaderBlock& headers); | |
62 scoped_ptr<QuicEncryptedPacket> MakeResponseHeadersPacket( | |
63 QuicPacketSequenceNumber sequence_number, | |
64 QuicStreamId stream_id, | |
65 bool should_include_version, | |
66 bool fin, | |
67 const SpdyHeaderBlock& headers); | |
68 | |
69 SpdyHeaderBlock GetRequestHeaders(const std::string& method, | |
70 const std::string& scheme, | |
71 const std::string& path); | |
72 SpdyHeaderBlock GetResponseHeaders(const std::string& status); | |
73 | |
74 private: | |
75 scoped_ptr<QuicEncryptedPacket> MakePacket( | |
76 const QuicPacketHeader& header, | |
77 const QuicFrame& frame); | |
78 | |
79 void InitializeHeader(QuicPacketSequenceNumber sequence_number, | |
80 bool should_include_version); | |
81 | |
82 QuicVersion version_; | |
83 QuicConnectionId connection_id_; | |
84 MockClock* clock_; // Owned by QuicStreamFactory. | |
85 SpdyFramer spdy_request_framer_; | |
86 SpdyFramer spdy_response_framer_; | |
87 MockRandom random_generator_; | |
88 QuicPacketHeader header_; | |
89 | |
90 DISALLOW_COPY_AND_ASSIGN(QuicTestPacketMaker); | |
91 }; | |
92 | |
93 } // namespace test | |
94 } // namespace net | |
95 | |
96 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_PACKET_MAKER_H_ | |
OLD | NEW |