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/quic/quic_stream_sequencer.h" | 5 #include "net/quic/quic_stream_sequencer.h" |
6 | 6 |
7 #include <utility> | 7 #include <utility> |
8 #include <vector> | 8 #include <vector> |
9 | 9 |
10 #include "base/logging.h" | 10 #include "base/logging.h" |
(...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
56 static const char kPayload[] = | 56 static const char kPayload[] = |
57 "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"; | 57 "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"; |
58 | 58 |
59 class QuicStreamSequencerTest : public ::testing::Test { | 59 class QuicStreamSequencerTest : public ::testing::Test { |
60 protected: | 60 protected: |
61 QuicStreamSequencerTest() | 61 QuicStreamSequencerTest() |
62 : connection_(new MockConnection(false)), | 62 : connection_(new MockConnection(false)), |
63 session_(connection_), | 63 session_(connection_), |
64 stream_(&session_, 1), | 64 stream_(&session_, 1), |
65 sequencer_(new QuicStreamSequencer(&stream_)), | 65 sequencer_(new QuicStreamSequencer(&stream_)), |
66 frames_(QuicStreamSequencerPeer::GetFrames(sequencer_.get())) { | 66 buffered_frames_( |
| 67 QuicStreamSequencerPeer::GetBufferedFrames(sequencer_.get())) { |
67 } | 68 } |
68 | 69 |
69 bool VerifyReadableRegions(const char** expected, size_t num_expected) { | 70 bool VerifyReadableRegions(const char** expected, size_t num_expected) { |
70 iovec iovecs[5]; | 71 iovec iovecs[5]; |
71 size_t num_iovecs = sequencer_->GetReadableRegions(iovecs, | 72 size_t num_iovecs = sequencer_->GetReadableRegions(iovecs, |
72 arraysize(iovecs)); | 73 arraysize(iovecs)); |
73 return VerifyIovecs(iovecs, num_iovecs, expected, num_expected); | 74 return VerifyIovecs(iovecs, num_iovecs, expected, num_expected); |
74 } | 75 } |
75 | 76 |
76 bool VerifyIovecs(iovec* iovecs, | 77 bool VerifyIovecs(iovec* iovecs, |
(...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
119 frame.offset = byte_offset; | 120 frame.offset = byte_offset; |
120 frame.data.Append(const_cast<char*>(data), strlen(data)); | 121 frame.data.Append(const_cast<char*>(data), strlen(data)); |
121 frame.fin = false; | 122 frame.fin = false; |
122 return sequencer_->OnStreamFrame(frame); | 123 return sequencer_->OnStreamFrame(frame); |
123 } | 124 } |
124 | 125 |
125 MockConnection* connection_; | 126 MockConnection* connection_; |
126 MockSession session_; | 127 MockSession session_; |
127 testing::StrictMock<MockStream> stream_; | 128 testing::StrictMock<MockStream> stream_; |
128 scoped_ptr<QuicStreamSequencer> sequencer_; | 129 scoped_ptr<QuicStreamSequencer> sequencer_; |
129 map<QuicStreamOffset, string>* frames_; | 130 map<QuicStreamOffset, string>* buffered_frames_; |
130 }; | 131 }; |
131 | 132 |
132 TEST_F(QuicStreamSequencerTest, RejectOldFrame) { | 133 TEST_F(QuicStreamSequencerTest, RejectOldFrame) { |
133 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)).WillOnce(Return(3)); | 134 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)).WillOnce(Return(3)); |
134 | 135 |
135 EXPECT_TRUE(OnFrame(0, "abc")); | 136 EXPECT_TRUE(OnFrame(0, "abc")); |
136 EXPECT_EQ(0u, frames_->size()); | 137 EXPECT_EQ(0u, buffered_frames_->size()); |
137 EXPECT_EQ(3u, sequencer_->num_bytes_consumed()); | 138 EXPECT_EQ(3u, sequencer_->num_bytes_consumed()); |
138 // Ignore this - it matches a past sequence number and we should not see it | 139 // Ignore this - it matches a past sequence number and we should not see it |
139 // again. | 140 // again. |
140 EXPECT_TRUE(OnFrame(0, "def")); | 141 EXPECT_TRUE(OnFrame(0, "def")); |
141 EXPECT_EQ(0u, frames_->size()); | 142 EXPECT_EQ(0u, buffered_frames_->size()); |
142 } | 143 } |
143 | 144 |
144 TEST_F(QuicStreamSequencerTest, RejectBufferedFrame) { | 145 TEST_F(QuicStreamSequencerTest, RejectBufferedFrame) { |
145 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)); | 146 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)); |
146 | 147 |
147 EXPECT_TRUE(OnFrame(0, "abc")); | 148 EXPECT_TRUE(OnFrame(0, "abc")); |
148 EXPECT_EQ(1u, frames_->size()); | 149 EXPECT_EQ(1u, buffered_frames_->size()); |
149 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); | 150 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); |
150 // Ignore this - it matches a buffered frame. | 151 // Ignore this - it matches a buffered frame. |
151 // Right now there's no checking that the payload is consistent. | 152 // Right now there's no checking that the payload is consistent. |
152 EXPECT_TRUE(OnFrame(0, "def")); | 153 EXPECT_TRUE(OnFrame(0, "def")); |
153 EXPECT_EQ(1u, frames_->size()); | 154 EXPECT_EQ(1u, buffered_frames_->size()); |
154 } | 155 } |
155 | 156 |
156 TEST_F(QuicStreamSequencerTest, FullFrameConsumed) { | 157 TEST_F(QuicStreamSequencerTest, FullFrameConsumed) { |
157 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)).WillOnce(Return(3)); | 158 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)).WillOnce(Return(3)); |
158 | 159 |
159 EXPECT_TRUE(OnFrame(0, "abc")); | 160 EXPECT_TRUE(OnFrame(0, "abc")); |
160 EXPECT_EQ(0u, frames_->size()); | 161 EXPECT_EQ(0u, buffered_frames_->size()); |
161 EXPECT_EQ(3u, sequencer_->num_bytes_consumed()); | 162 EXPECT_EQ(3u, sequencer_->num_bytes_consumed()); |
162 } | 163 } |
163 | 164 |
164 TEST_F(QuicStreamSequencerTest, BlockedThenFullFrameConsumed) { | 165 TEST_F(QuicStreamSequencerTest, BlockedThenFullFrameConsumed) { |
165 sequencer_->SetBlockedUntilFlush(); | 166 sequencer_->SetBlockedUntilFlush(); |
166 | 167 |
167 EXPECT_TRUE(OnFrame(0, "abc")); | 168 EXPECT_TRUE(OnFrame(0, "abc")); |
168 EXPECT_EQ(1u, frames_->size()); | 169 EXPECT_EQ(1u, buffered_frames_->size()); |
169 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); | 170 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); |
170 | 171 |
171 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)).WillOnce(Return(3)); | 172 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)).WillOnce(Return(3)); |
172 sequencer_->FlushBufferedFrames(); | 173 sequencer_->FlushBufferedFrames(); |
173 EXPECT_EQ(0u, frames_->size()); | 174 EXPECT_EQ(0u, buffered_frames_->size()); |
174 EXPECT_EQ(3u, sequencer_->num_bytes_consumed()); | 175 EXPECT_EQ(3u, sequencer_->num_bytes_consumed()); |
175 | 176 |
176 EXPECT_CALL(stream_, ProcessRawData(StrEq("def"), 3)).WillOnce(Return(3)); | 177 EXPECT_CALL(stream_, ProcessRawData(StrEq("def"), 3)).WillOnce(Return(3)); |
177 EXPECT_CALL(stream_, OnFinRead()); | 178 EXPECT_CALL(stream_, OnFinRead()); |
178 EXPECT_TRUE(OnFinFrame(3, "def")); | 179 EXPECT_TRUE(OnFinFrame(3, "def")); |
179 } | 180 } |
180 | 181 |
181 TEST_F(QuicStreamSequencerTest, BlockedThenFullFrameAndFinConsumed) { | 182 TEST_F(QuicStreamSequencerTest, BlockedThenFullFrameAndFinConsumed) { |
182 sequencer_->SetBlockedUntilFlush(); | 183 sequencer_->SetBlockedUntilFlush(); |
183 | 184 |
184 EXPECT_TRUE(OnFinFrame(0, "abc")); | 185 EXPECT_TRUE(OnFinFrame(0, "abc")); |
185 EXPECT_EQ(1u, frames_->size()); | 186 EXPECT_EQ(1u, buffered_frames_->size()); |
186 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); | 187 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); |
187 | 188 |
188 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)).WillOnce(Return(3)); | 189 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)).WillOnce(Return(3)); |
189 EXPECT_CALL(stream_, OnFinRead()); | 190 EXPECT_CALL(stream_, OnFinRead()); |
190 sequencer_->FlushBufferedFrames(); | 191 sequencer_->FlushBufferedFrames(); |
191 EXPECT_EQ(0u, frames_->size()); | 192 EXPECT_EQ(0u, buffered_frames_->size()); |
192 EXPECT_EQ(3u, sequencer_->num_bytes_consumed()); | 193 EXPECT_EQ(3u, sequencer_->num_bytes_consumed()); |
193 } | 194 } |
194 | 195 |
195 TEST_F(QuicStreamSequencerTest, EmptyFrame) { | 196 TEST_F(QuicStreamSequencerTest, EmptyFrame) { |
196 EXPECT_CALL(stream_, | 197 EXPECT_CALL(stream_, |
197 CloseConnectionWithDetails(QUIC_INVALID_STREAM_FRAME, _)); | 198 CloseConnectionWithDetails(QUIC_INVALID_STREAM_FRAME, _)); |
198 EXPECT_FALSE(OnFrame(0, "")); | 199 EXPECT_FALSE(OnFrame(0, "")); |
199 EXPECT_EQ(0u, frames_->size()); | 200 EXPECT_EQ(0u, buffered_frames_->size()); |
200 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); | 201 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); |
201 } | 202 } |
202 | 203 |
203 TEST_F(QuicStreamSequencerTest, EmptyFinFrame) { | 204 TEST_F(QuicStreamSequencerTest, EmptyFinFrame) { |
204 EXPECT_CALL(stream_, OnFinRead()); | 205 EXPECT_CALL(stream_, OnFinRead()); |
205 EXPECT_TRUE(OnFinFrame(0, "")); | 206 EXPECT_TRUE(OnFinFrame(0, "")); |
206 EXPECT_EQ(0u, frames_->size()); | 207 EXPECT_EQ(0u, buffered_frames_->size()); |
207 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); | 208 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); |
208 } | 209 } |
209 | 210 |
210 TEST_F(QuicStreamSequencerTest, PartialFrameConsumed) { | 211 TEST_F(QuicStreamSequencerTest, PartialFrameConsumed) { |
211 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)).WillOnce(Return(2)); | 212 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)).WillOnce(Return(2)); |
212 | 213 |
213 EXPECT_TRUE(OnFrame(0, "abc")); | 214 EXPECT_TRUE(OnFrame(0, "abc")); |
214 EXPECT_EQ(1u, frames_->size()); | 215 EXPECT_EQ(1u, buffered_frames_->size()); |
215 EXPECT_EQ(2u, sequencer_->num_bytes_consumed()); | 216 EXPECT_EQ(2u, sequencer_->num_bytes_consumed()); |
216 EXPECT_EQ("c", frames_->find(2)->second); | 217 EXPECT_EQ("c", buffered_frames_->find(2)->second); |
217 } | 218 } |
218 | 219 |
219 TEST_F(QuicStreamSequencerTest, NextxFrameNotConsumed) { | 220 TEST_F(QuicStreamSequencerTest, NextxFrameNotConsumed) { |
220 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)).WillOnce(Return(0)); | 221 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)).WillOnce(Return(0)); |
221 | 222 |
222 EXPECT_TRUE(OnFrame(0, "abc")); | 223 EXPECT_TRUE(OnFrame(0, "abc")); |
223 EXPECT_EQ(1u, frames_->size()); | 224 EXPECT_EQ(1u, buffered_frames_->size()); |
224 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); | 225 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); |
225 EXPECT_EQ("abc", frames_->find(0)->second); | 226 EXPECT_EQ("abc", buffered_frames_->find(0)->second); |
226 } | 227 } |
227 | 228 |
228 TEST_F(QuicStreamSequencerTest, FutureFrameNotProcessed) { | 229 TEST_F(QuicStreamSequencerTest, FutureFrameNotProcessed) { |
229 EXPECT_TRUE(OnFrame(3, "abc")); | 230 EXPECT_TRUE(OnFrame(3, "abc")); |
230 EXPECT_EQ(1u, frames_->size()); | 231 EXPECT_EQ(1u, buffered_frames_->size()); |
231 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); | 232 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); |
232 EXPECT_EQ("abc", frames_->find(3)->second); | 233 EXPECT_EQ("abc", buffered_frames_->find(3)->second); |
233 } | 234 } |
234 | 235 |
235 TEST_F(QuicStreamSequencerTest, OutOfOrderFrameProcessed) { | 236 TEST_F(QuicStreamSequencerTest, OutOfOrderFrameProcessed) { |
236 // Buffer the first | 237 // Buffer the first |
237 EXPECT_TRUE(OnFrame(6, "ghi")); | 238 EXPECT_TRUE(OnFrame(6, "ghi")); |
238 EXPECT_EQ(1u, frames_->size()); | 239 EXPECT_EQ(1u, buffered_frames_->size()); |
239 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); | 240 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); |
240 EXPECT_EQ(3u, sequencer_->num_bytes_buffered()); | 241 EXPECT_EQ(3u, sequencer_->num_bytes_buffered()); |
241 // Buffer the second | 242 // Buffer the second |
242 EXPECT_TRUE(OnFrame(3, "def")); | 243 EXPECT_TRUE(OnFrame(3, "def")); |
243 EXPECT_EQ(2u, frames_->size()); | 244 EXPECT_EQ(2u, buffered_frames_->size()); |
244 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); | 245 EXPECT_EQ(0u, sequencer_->num_bytes_consumed()); |
245 EXPECT_EQ(6u, sequencer_->num_bytes_buffered()); | 246 EXPECT_EQ(6u, sequencer_->num_bytes_buffered()); |
246 | 247 |
247 InSequence s; | 248 InSequence s; |
248 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)).WillOnce(Return(3)); | 249 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)).WillOnce(Return(3)); |
249 EXPECT_CALL(stream_, ProcessRawData(StrEq("def"), 3)).WillOnce(Return(3)); | 250 EXPECT_CALL(stream_, ProcessRawData(StrEq("def"), 3)).WillOnce(Return(3)); |
250 EXPECT_CALL(stream_, ProcessRawData(StrEq("ghi"), 3)).WillOnce(Return(3)); | 251 EXPECT_CALL(stream_, ProcessRawData(StrEq("ghi"), 3)).WillOnce(Return(3)); |
251 | 252 |
252 // Ack right away | 253 // Ack right away |
253 EXPECT_TRUE(OnFrame(0, "abc")); | 254 EXPECT_TRUE(OnFrame(0, "abc")); |
254 EXPECT_EQ(9u, sequencer_->num_bytes_consumed()); | 255 EXPECT_EQ(9u, sequencer_->num_bytes_consumed()); |
255 EXPECT_EQ(0u, sequencer_->num_bytes_buffered()); | 256 EXPECT_EQ(0u, sequencer_->num_bytes_buffered()); |
256 | 257 |
257 EXPECT_EQ(0u, frames_->size()); | 258 EXPECT_EQ(0u, buffered_frames_->size()); |
258 } | 259 } |
259 | 260 |
260 TEST_F(QuicStreamSequencerTest, BasicHalfCloseOrdered) { | 261 TEST_F(QuicStreamSequencerTest, BasicHalfCloseOrdered) { |
261 InSequence s; | 262 InSequence s; |
262 | 263 |
263 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)).WillOnce(Return(3)); | 264 EXPECT_CALL(stream_, ProcessRawData(StrEq("abc"), 3)).WillOnce(Return(3)); |
264 EXPECT_CALL(stream_, OnFinRead()); | 265 EXPECT_CALL(stream_, OnFinRead()); |
265 EXPECT_TRUE(OnFinFrame(0, "abc")); | 266 EXPECT_TRUE(OnFinFrame(0, "abc")); |
266 | 267 |
267 EXPECT_EQ(3u, QuicStreamSequencerPeer::GetCloseOffset(sequencer_.get())); | 268 EXPECT_EQ(3u, QuicStreamSequencerPeer::GetCloseOffset(sequencer_.get())); |
(...skipping 106 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
374 LOG(ERROR) << "Sending index " << index << " " << list_[index].second; | 375 LOG(ERROR) << "Sending index " << index << " " << list_[index].second; |
375 EXPECT_TRUE(OnFrame(list_[index].first, list_[index].second.data())); | 376 EXPECT_TRUE(OnFrame(list_[index].first, list_[index].second.data())); |
376 | 377 |
377 list_.erase(list_.begin() + index); | 378 list_.erase(list_.begin() + index); |
378 } | 379 } |
379 } | 380 } |
380 | 381 |
381 } // namespace | 382 } // namespace |
382 } // namespace test | 383 } // namespace test |
383 } // namespace net | 384 } // namespace net |
OLD | NEW |