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/socket/socket_test_util.h" | 5 #include "net/socket/socket_test_util.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 #include <vector> | 8 #include <vector> |
9 | 9 |
10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
(...skipping 362 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
373 read_in_progress_ = false; | 373 read_in_progress_ = false; |
374 weak_factory_.InvalidateWeakPtrs(); | 374 weak_factory_.InvalidateWeakPtrs(); |
375 StaticSocketDataProvider::Reset(); | 375 StaticSocketDataProvider::Reset(); |
376 } | 376 } |
377 | 377 |
378 void DelayedSocketData::CompleteRead() { | 378 void DelayedSocketData::CompleteRead() { |
379 if (socket() && read_in_progress_) | 379 if (socket() && read_in_progress_) |
380 socket()->OnReadComplete(OnRead()); | 380 socket()->OnReadComplete(OnRead()); |
381 } | 381 } |
382 | 382 |
383 OrderedSocketData::OrderedSocketData( | |
384 MockRead* reads, size_t reads_count, MockWrite* writes, size_t writes_count) | |
385 : StaticSocketDataProvider(reads, reads_count, writes, writes_count), | |
386 sequence_number_(0), loop_stop_stage_(0), | |
387 blocked_(false), weak_factory_(this) { | |
388 } | |
389 | |
390 OrderedSocketData::OrderedSocketData( | |
391 const MockConnect& connect, | |
392 MockRead* reads, size_t reads_count, | |
393 MockWrite* writes, size_t writes_count) | |
394 : StaticSocketDataProvider(reads, reads_count, writes, writes_count), | |
395 sequence_number_(0), loop_stop_stage_(0), | |
396 blocked_(false), weak_factory_(this) { | |
397 set_connect_data(connect); | |
398 } | |
399 | |
400 void OrderedSocketData::EndLoop() { | |
401 // If we've already stopped the loop, don't do it again until we've advanced | |
402 // to the next sequence_number. | |
403 NET_TRACE(1, " *** ") << "Stage " << sequence_number_ << ": EndLoop()"; | |
404 if (loop_stop_stage_ > 0) { | |
405 const MockRead& next_read = helper()->PeekRead(); | |
406 if ((next_read.sequence_number & ~MockRead::STOPLOOP) > | |
407 loop_stop_stage_) { | |
408 NET_TRACE(1, " *** ") << "Stage " << sequence_number_ | |
409 << ": Clearing stop index"; | |
410 loop_stop_stage_ = 0; | |
411 } else { | |
412 return; | |
413 } | |
414 } | |
415 // Record the sequence_number at which we stopped the loop. | |
416 NET_TRACE(1, " *** ") << "Stage " << sequence_number_ | |
417 << ": Posting Quit at read " << read_index(); | |
418 loop_stop_stage_ = sequence_number_; | |
419 } | |
420 | |
421 MockRead OrderedSocketData::OnRead() { | |
422 weak_factory_.InvalidateWeakPtrs(); | |
423 blocked_ = false; | |
424 const MockRead& next_read = helper()->PeekRead(); | |
425 if (next_read.sequence_number & MockRead::STOPLOOP) | |
426 EndLoop(); | |
427 if ((next_read.sequence_number & ~MockRead::STOPLOOP) <= | |
428 sequence_number_++) { | |
429 NET_TRACE(1, " *** ") << "Stage " << sequence_number_ - 1 << ": Read " | |
430 << read_index(); | |
431 DumpMockReadWrite(next_read); | |
432 blocked_ = (next_read.result == ERR_IO_PENDING); | |
433 return StaticSocketDataProvider::OnRead(); | |
434 } | |
435 NET_TRACE(1, " *** ") << "Stage " << sequence_number_ - 1 << ": I/O Pending"; | |
436 MockRead result = MockRead(ASYNC, ERR_IO_PENDING); | |
437 DumpMockReadWrite(result); | |
438 blocked_ = true; | |
439 return result; | |
440 } | |
441 | |
442 MockWriteResult OrderedSocketData::OnWrite(const std::string& data) { | |
443 NET_TRACE(1, " *** ") << "Stage " << sequence_number_ << ": Write " | |
444 << write_index(); | |
445 DumpMockReadWrite(helper()->PeekWrite()); | |
446 ++sequence_number_; | |
447 if (blocked_) { | |
448 // TODO(willchan): This 100ms delay seems to work around some weirdness. We | |
449 // should probably fix the weirdness. One example is in SpdyStream, | |
450 // DoSendRequest() will return ERR_IO_PENDING, and there's a race. If the | |
451 // SYN_REPLY causes OnResponseReceived() to get called before | |
452 // SpdyStream::ReadResponseHeaders() is called, we hit a NOTREACHED(). | |
453 base::MessageLoop::current()->PostDelayedTask( | |
454 FROM_HERE, | |
455 base::Bind(&OrderedSocketData::CompleteRead, | |
456 weak_factory_.GetWeakPtr()), | |
457 base::TimeDelta::FromMilliseconds(100)); | |
458 } | |
459 return StaticSocketDataProvider::OnWrite(data); | |
460 } | |
461 | |
462 void OrderedSocketData::Reset() { | |
463 NET_TRACE(1, " *** ") << "Stage " << sequence_number_ << ": Reset()"; | |
464 sequence_number_ = 0; | |
465 loop_stop_stage_ = 0; | |
466 set_socket(NULL); | |
467 weak_factory_.InvalidateWeakPtrs(); | |
468 StaticSocketDataProvider::Reset(); | |
469 } | |
470 | |
471 void OrderedSocketData::CompleteRead() { | |
472 if (socket() && blocked_) { | |
473 NET_TRACE(1, " *** ") << "Stage " << sequence_number_; | |
474 socket()->OnReadComplete(OnRead()); | |
475 } | |
476 } | |
477 | |
478 OrderedSocketData::~OrderedSocketData() {} | |
479 | |
480 SequencedSocketData::SequencedSocketData(MockRead* reads, | 383 SequencedSocketData::SequencedSocketData(MockRead* reads, |
481 size_t reads_count, | 384 size_t reads_count, |
482 MockWrite* writes, | 385 MockWrite* writes, |
483 size_t writes_count) | 386 size_t writes_count) |
484 : helper_(reads, reads_count, writes, writes_count), | 387 : helper_(reads, reads_count, writes, writes_count), |
485 sequence_number_(0), | 388 sequence_number_(0), |
486 read_state_(IDLE), | 389 read_state_(IDLE), |
487 write_state_(IDLE), | 390 write_state_(IDLE), |
488 weak_factory_(this) { | 391 weak_factory_(this) { |
489 // Check that reads and writes have a contiguous set of sequence numbers | 392 // Check that reads and writes have a contiguous set of sequence numbers |
(...skipping 1730 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
2220 | 2123 |
2221 const char kSOCKS5OkRequest[] = | 2124 const char kSOCKS5OkRequest[] = |
2222 { 0x05, 0x01, 0x00, 0x03, 0x04, 'h', 'o', 's', 't', 0x00, 0x50 }; | 2125 { 0x05, 0x01, 0x00, 0x03, 0x04, 'h', 'o', 's', 't', 0x00, 0x50 }; |
2223 const int kSOCKS5OkRequestLength = arraysize(kSOCKS5OkRequest); | 2126 const int kSOCKS5OkRequestLength = arraysize(kSOCKS5OkRequest); |
2224 | 2127 |
2225 const char kSOCKS5OkResponse[] = | 2128 const char kSOCKS5OkResponse[] = |
2226 { 0x05, 0x00, 0x00, 0x01, 127, 0, 0, 1, 0x00, 0x50 }; | 2129 { 0x05, 0x00, 0x00, 0x01, 127, 0, 0, 1, 0x00, 0x50 }; |
2227 const int kSOCKS5OkResponseLength = arraysize(kSOCKS5OkResponse); | 2130 const int kSOCKS5OkResponseLength = arraysize(kSOCKS5OkResponse); |
2228 | 2131 |
2229 } // namespace net | 2132 } // namespace net |
OLD | NEW |