Index: net/quic/test_tools/simple_quic_framer.cc |
diff --git a/net/quic/test_tools/simple_quic_framer.cc b/net/quic/test_tools/simple_quic_framer.cc |
index 21b6c1f461f1bc108d797b2e430eaa22b75b45b5..fca7f7905706112e066c288eac1182d186bb2a2f 100644 |
--- a/net/quic/test_tools/simple_quic_framer.cc |
+++ b/net/quic/test_tools/simple_quic_framer.cc |
@@ -80,6 +80,11 @@ class SimpleFramerVisitor : public QuicFramerVisitorInterface { |
return true; |
} |
+ virtual bool OnStopWaitingFrame(const QuicStopWaitingFrame& frame) OVERRIDE { |
+ stop_waiting_frames_.push_back(frame); |
+ return true; |
+ } |
+ |
virtual void OnFecData(const QuicFecData& fec) OVERRIDE { |
fec_data_ = fec; |
fec_redundancy_ = fec_data_.redundancy.as_string(); |
@@ -102,8 +107,8 @@ class SimpleFramerVisitor : public QuicFramerVisitorInterface { |
return true; |
} |
- virtual bool OnWindowUpdateFrame(const QuicWindowUpdateFrame& frame) |
- OVERRIDE { |
+ virtual bool OnWindowUpdateFrame( |
+ const QuicWindowUpdateFrame& frame) OVERRIDE { |
window_update_frames_.push_back(frame); |
return true; |
} |
@@ -132,6 +137,9 @@ class SimpleFramerVisitor : public QuicFramerVisitorInterface { |
const vector<QuicStreamFrame>& stream_frames() const { |
return stream_frames_; |
} |
+ const vector<QuicStopWaitingFrame>& stop_waiting_frames() const { |
+ return stop_waiting_frames_; |
+ } |
const vector<QuicWindowUpdateFrame>& window_update_frames() const { |
return window_update_frames_; |
} |
@@ -150,6 +158,7 @@ class SimpleFramerVisitor : public QuicFramerVisitorInterface { |
string fec_redundancy_; |
vector<QuicAckFrame> ack_frames_; |
vector<QuicCongestionFeedbackFrame> feedback_frames_; |
+ vector<QuicStopWaitingFrame> stop_waiting_frames_; |
vector<QuicStreamFrame> stream_frames_; |
vector<QuicRstStreamFrame> rst_stream_frames_; |
vector<QuicGoAwayFrame> goaway_frames_; |
@@ -198,10 +207,11 @@ QuicFramer* SimpleQuicFramer::framer() { |
size_t SimpleQuicFramer::num_frames() const { |
return ack_frames().size() + |
- stream_frames().size() + |
feedback_frames().size() + |
- rst_stream_frames().size() + |
goaway_frames().size() + |
+ rst_stream_frames().size() + |
+ stop_waiting_frames().size() + |
+ stream_frames().size() + |
connection_close_frames().size(); |
} |
@@ -209,6 +219,11 @@ const vector<QuicAckFrame>& SimpleQuicFramer::ack_frames() const { |
return visitor_->ack_frames(); |
} |
+const vector<QuicStopWaitingFrame>& |
+SimpleQuicFramer::stop_waiting_frames() const { |
+ return visitor_->stop_waiting_frames(); |
+} |
+ |
const vector<QuicStreamFrame>& SimpleQuicFramer::stream_frames() const { |
return visitor_->stream_frames(); |
} |