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