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/spdy/buffered_spdy_framer.h" | 5 #include "net/spdy/buffered_spdy_framer.h" |
6 | 6 |
7 #include "net/spdy/spdy_test_util_common.h" | 7 #include "net/spdy/spdy_test_util_common.h" |
8 #include "testing/platform_test.h" | 8 #include "testing/platform_test.h" |
9 | 9 |
10 namespace net { | 10 namespace net { |
11 | 11 |
12 namespace { | 12 namespace { |
13 | 13 |
14 class TestBufferedSpdyVisitor : public BufferedSpdyFramerVisitorInterface { | 14 class TestBufferedSpdyVisitor : public BufferedSpdyFramerVisitorInterface { |
15 public: | 15 public: |
16 explicit TestBufferedSpdyVisitor(SpdyMajorVersion spdy_version) | 16 explicit TestBufferedSpdyVisitor(SpdyMajorVersion spdy_version) |
17 : buffered_spdy_framer_(spdy_version, true), | 17 : buffered_spdy_framer_(spdy_version, true), |
18 error_count_(0), | 18 error_count_(0), |
19 setting_count_(0), | 19 setting_count_(0), |
20 syn_frame_count_(0), | 20 syn_frame_count_(0), |
21 syn_reply_frame_count_(0), | 21 syn_reply_frame_count_(0), |
22 headers_frame_count_(0), | 22 headers_frame_count_(0), |
23 push_promise_frame_count_(0), | 23 push_promise_frame_count_(0), |
24 header_stream_id_(static_cast<SpdyStreamId>(-1)), | 24 header_stream_id_(static_cast<SpdyStreamId>(-1)), |
25 promised_stream_id_(static_cast<SpdyStreamId>(-1)) { | 25 promised_stream_id_(static_cast<SpdyStreamId>(-1)) { |
26 } | 26 } |
27 | 27 |
28 virtual void OnError(SpdyFramer::SpdyError error_code) override { | 28 void OnError(SpdyFramer::SpdyError error_code) override { |
29 LOG(INFO) << "SpdyFramer Error: " << error_code; | 29 LOG(INFO) << "SpdyFramer Error: " << error_code; |
30 error_count_++; | 30 error_count_++; |
31 } | 31 } |
32 | 32 |
33 virtual void OnStreamError( | 33 void OnStreamError(SpdyStreamId stream_id, |
34 SpdyStreamId stream_id, | 34 const std::string& description) override { |
35 const std::string& description) override { | |
36 LOG(INFO) << "SpdyFramer Error on stream: " << stream_id << " " | 35 LOG(INFO) << "SpdyFramer Error on stream: " << stream_id << " " |
37 << description; | 36 << description; |
38 error_count_++; | 37 error_count_++; |
39 } | 38 } |
40 | 39 |
41 virtual void OnSynStream(SpdyStreamId stream_id, | 40 void OnSynStream(SpdyStreamId stream_id, |
42 SpdyStreamId associated_stream_id, | 41 SpdyStreamId associated_stream_id, |
43 SpdyPriority priority, | 42 SpdyPriority priority, |
44 bool fin, | 43 bool fin, |
45 bool unidirectional, | 44 bool unidirectional, |
46 const SpdyHeaderBlock& headers) override { | 45 const SpdyHeaderBlock& headers) override { |
47 header_stream_id_ = stream_id; | 46 header_stream_id_ = stream_id; |
48 EXPECT_NE(header_stream_id_, SpdyFramer::kInvalidStream); | 47 EXPECT_NE(header_stream_id_, SpdyFramer::kInvalidStream); |
49 syn_frame_count_++; | 48 syn_frame_count_++; |
50 headers_ = headers; | 49 headers_ = headers; |
51 } | 50 } |
52 | 51 |
53 virtual void OnSynReply(SpdyStreamId stream_id, | 52 void OnSynReply(SpdyStreamId stream_id, |
54 bool fin, | 53 bool fin, |
55 const SpdyHeaderBlock& headers) override { | 54 const SpdyHeaderBlock& headers) override { |
56 header_stream_id_ = stream_id; | 55 header_stream_id_ = stream_id; |
57 EXPECT_NE(header_stream_id_, SpdyFramer::kInvalidStream); | 56 EXPECT_NE(header_stream_id_, SpdyFramer::kInvalidStream); |
58 syn_reply_frame_count_++; | 57 syn_reply_frame_count_++; |
59 headers_ = headers; | 58 headers_ = headers; |
60 } | 59 } |
61 | 60 |
62 virtual void OnHeaders(SpdyStreamId stream_id, | 61 void OnHeaders(SpdyStreamId stream_id, |
63 bool fin, | 62 bool fin, |
64 const SpdyHeaderBlock& headers) override { | 63 const SpdyHeaderBlock& headers) override { |
65 header_stream_id_ = stream_id; | 64 header_stream_id_ = stream_id; |
66 EXPECT_NE(header_stream_id_, SpdyFramer::kInvalidStream); | 65 EXPECT_NE(header_stream_id_, SpdyFramer::kInvalidStream); |
67 headers_frame_count_++; | 66 headers_frame_count_++; |
68 headers_ = headers; | 67 headers_ = headers; |
69 } | 68 } |
70 | 69 |
71 virtual void OnDataFrameHeader(SpdyStreamId stream_id, | 70 void OnDataFrameHeader(SpdyStreamId stream_id, |
72 size_t length, | 71 size_t length, |
73 bool fin) override { | 72 bool fin) override { |
74 ADD_FAILURE() << "Unexpected OnDataFrameHeader call."; | 73 ADD_FAILURE() << "Unexpected OnDataFrameHeader call."; |
75 } | 74 } |
76 | 75 |
77 virtual void OnStreamFrameData(SpdyStreamId stream_id, | 76 void OnStreamFrameData(SpdyStreamId stream_id, |
78 const char* data, | 77 const char* data, |
79 size_t len, | 78 size_t len, |
80 bool fin) override { | 79 bool fin) override { |
81 LOG(FATAL) << "Unexpected OnStreamFrameData call."; | 80 LOG(FATAL) << "Unexpected OnStreamFrameData call."; |
82 } | 81 } |
83 | 82 |
84 virtual void OnSettings(bool clear_persisted) override {} | 83 void OnSettings(bool clear_persisted) override {} |
85 | 84 |
86 virtual void OnSetting(SpdySettingsIds id, | 85 void OnSetting(SpdySettingsIds id, uint8 flags, uint32 value) override { |
87 uint8 flags, | |
88 uint32 value) override { | |
89 setting_count_++; | 86 setting_count_++; |
90 } | 87 } |
91 | 88 |
92 virtual void OnPing(SpdyPingId unique_id, bool is_ack) override {} | 89 void OnPing(SpdyPingId unique_id, bool is_ack) override {} |
93 | 90 |
94 virtual void OnRstStream(SpdyStreamId stream_id, | 91 void OnRstStream(SpdyStreamId stream_id, |
95 SpdyRstStreamStatus status) override { | 92 SpdyRstStreamStatus status) override {} |
96 } | |
97 | 93 |
98 virtual void OnGoAway(SpdyStreamId last_accepted_stream_id, | 94 void OnGoAway(SpdyStreamId last_accepted_stream_id, |
99 SpdyGoAwayStatus status) override { | 95 SpdyGoAwayStatus status) override {} |
100 } | |
101 | 96 |
102 bool OnCredentialFrameData(const char*, size_t) { | 97 bool OnCredentialFrameData(const char*, size_t) { |
103 LOG(FATAL) << "Unexpected OnCredentialFrameData call."; | 98 LOG(FATAL) << "Unexpected OnCredentialFrameData call."; |
104 return false; | 99 return false; |
105 } | 100 } |
106 | 101 |
107 void OnDataFrameHeader(const SpdyFrame* frame) { | 102 void OnDataFrameHeader(const SpdyFrame* frame) { |
108 LOG(FATAL) << "Unexpected OnDataFrameHeader call."; | 103 LOG(FATAL) << "Unexpected OnDataFrameHeader call."; |
109 } | 104 } |
110 | 105 |
111 void OnRstStream(const SpdyFrame& frame) {} | 106 void OnRstStream(const SpdyFrame& frame) {} |
112 void OnGoAway(const SpdyFrame& frame) {} | 107 void OnGoAway(const SpdyFrame& frame) {} |
113 void OnPing(const SpdyFrame& frame) {} | 108 void OnPing(const SpdyFrame& frame) {} |
114 virtual void OnWindowUpdate(SpdyStreamId stream_id, | 109 void OnWindowUpdate(SpdyStreamId stream_id, |
115 uint32 delta_window_size) override {} | 110 uint32 delta_window_size) override {} |
116 | 111 |
117 virtual void OnPushPromise(SpdyStreamId stream_id, | 112 void OnPushPromise(SpdyStreamId stream_id, |
118 SpdyStreamId promised_stream_id, | 113 SpdyStreamId promised_stream_id, |
119 const SpdyHeaderBlock& headers) override { | 114 const SpdyHeaderBlock& headers) override { |
120 header_stream_id_ = stream_id; | 115 header_stream_id_ = stream_id; |
121 EXPECT_NE(header_stream_id_, SpdyFramer::kInvalidStream); | 116 EXPECT_NE(header_stream_id_, SpdyFramer::kInvalidStream); |
122 push_promise_frame_count_++; | 117 push_promise_frame_count_++; |
123 promised_stream_id_ = promised_stream_id; | 118 promised_stream_id_ = promised_stream_id; |
124 EXPECT_NE(promised_stream_id_, SpdyFramer::kInvalidStream); | 119 EXPECT_NE(promised_stream_id_, SpdyFramer::kInvalidStream); |
125 headers_ = headers; | 120 headers_ = headers; |
126 } | 121 } |
127 | 122 |
128 virtual bool OnUnknownFrame(SpdyStreamId stream_id, int frame_type) override { | 123 bool OnUnknownFrame(SpdyStreamId stream_id, int frame_type) override { |
129 return true; | 124 return true; |
130 } | 125 } |
131 | 126 |
132 void OnCredential(const SpdyFrame& frame) {} | 127 void OnCredential(const SpdyFrame& frame) {} |
133 | 128 |
134 // Convenience function which runs a framer simulation with particular input. | 129 // Convenience function which runs a framer simulation with particular input. |
135 void SimulateInFramer(const unsigned char* input, size_t size) { | 130 void SimulateInFramer(const unsigned char* input, size_t size) { |
136 buffered_spdy_framer_.set_visitor(this); | 131 buffered_spdy_framer_.set_visitor(this); |
137 size_t input_remaining = size; | 132 size_t input_remaining = size; |
138 const char* input_ptr = reinterpret_cast<const char*>(input); | 133 const char* input_ptr = reinterpret_cast<const char*>(input); |
(...skipping 193 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
332 EXPECT_EQ(0, visitor.syn_frame_count_); | 327 EXPECT_EQ(0, visitor.syn_frame_count_); |
333 EXPECT_EQ(0, visitor.syn_reply_frame_count_); | 328 EXPECT_EQ(0, visitor.syn_reply_frame_count_); |
334 EXPECT_EQ(0, visitor.headers_frame_count_); | 329 EXPECT_EQ(0, visitor.headers_frame_count_); |
335 EXPECT_EQ(1, visitor.push_promise_frame_count_); | 330 EXPECT_EQ(1, visitor.push_promise_frame_count_); |
336 EXPECT_TRUE(CompareHeaderBlocks(&headers, &visitor.headers_)); | 331 EXPECT_TRUE(CompareHeaderBlocks(&headers, &visitor.headers_)); |
337 EXPECT_EQ(1u, visitor.header_stream_id_); | 332 EXPECT_EQ(1u, visitor.header_stream_id_); |
338 EXPECT_EQ(2u, visitor.promised_stream_id_); | 333 EXPECT_EQ(2u, visitor.promised_stream_id_); |
339 } | 334 } |
340 | 335 |
341 } // namespace net | 336 } // namespace net |
OLD | NEW |