| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 #ifndef NET_TOOLS_QUIC_TEST_TOOLS_PACKET_DROPPING_TEST_WRITER_H_ | 5 #ifndef NET_TOOLS_QUIC_TEST_TOOLS_PACKET_DROPPING_TEST_WRITER_H_ |
| 6 #define NET_TOOLS_QUIC_TEST_TOOLS_PACKET_DROPPING_TEST_WRITER_H_ | 6 #define NET_TOOLS_QUIC_TEST_TOOLS_PACKET_DROPPING_TEST_WRITER_H_ |
| 7 | 7 |
| 8 #include <stddef.h> |
| 9 #include <stdint.h> |
| 10 |
| 8 #include <list> | 11 #include <list> |
| 9 #include <string> | 12 #include <string> |
| 10 | 13 |
| 11 #include "base/basictypes.h" | |
| 12 #include "base/logging.h" | 14 #include "base/logging.h" |
| 15 #include "base/macros.h" |
| 13 #include "base/memory/scoped_ptr.h" | 16 #include "base/memory/scoped_ptr.h" |
| 14 #include "base/synchronization/lock.h" | 17 #include "base/synchronization/lock.h" |
| 15 #include "net/quic/quic_alarm.h" | 18 #include "net/quic/quic_alarm.h" |
| 16 #include "net/quic/test_tools/quic_test_utils.h" | 19 #include "net/quic/test_tools/quic_test_utils.h" |
| 17 #include "net/tools/quic/quic_epoll_clock.h" | 20 #include "net/tools/quic/quic_epoll_clock.h" |
| 18 #include "net/tools/quic/quic_packet_writer_wrapper.h" | 21 #include "net/tools/quic/quic_packet_writer_wrapper.h" |
| 19 #include "net/tools/quic/test_tools/quic_test_client.h" | 22 #include "net/tools/quic/test_tools/quic_test_client.h" |
| 20 | 23 |
| 21 namespace net { | 24 namespace net { |
| 22 namespace tools { | 25 namespace tools { |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 55 void SetWritable() override; | 58 void SetWritable() override; |
| 56 | 59 |
| 57 // Writes out any packet which should have been sent by now | 60 // Writes out any packet which should have been sent by now |
| 58 // to the contained writer and returns the time | 61 // to the contained writer and returns the time |
| 59 // for the next delayed packet to be written. | 62 // for the next delayed packet to be written. |
| 60 QuicTime ReleaseOldPackets(); | 63 QuicTime ReleaseOldPackets(); |
| 61 | 64 |
| 62 void OnCanWrite(); | 65 void OnCanWrite(); |
| 63 | 66 |
| 64 // The percent of time a packet is simulated as being lost. | 67 // The percent of time a packet is simulated as being lost. |
| 65 void set_fake_packet_loss_percentage(int32 fake_packet_loss_percentage) { | 68 void set_fake_packet_loss_percentage(int32_t fake_packet_loss_percentage) { |
| 66 base::AutoLock locked(config_mutex_); | 69 base::AutoLock locked(config_mutex_); |
| 67 fake_packet_loss_percentage_ = fake_packet_loss_percentage; | 70 fake_packet_loss_percentage_ = fake_packet_loss_percentage; |
| 68 } | 71 } |
| 69 | 72 |
| 70 // Simulate dropping the first n packets unconditionally. | 73 // Simulate dropping the first n packets unconditionally. |
| 71 // Subsequent packets will be lost at fake_packet_loss_percentage_ if set. | 74 // Subsequent packets will be lost at fake_packet_loss_percentage_ if set. |
| 72 void set_fake_drop_first_n_packets(int32 fake_drop_first_n_packets) { | 75 void set_fake_drop_first_n_packets(int32_t fake_drop_first_n_packets) { |
| 73 base::AutoLock locked(config_mutex_); | 76 base::AutoLock locked(config_mutex_); |
| 74 fake_drop_first_n_packets_ = fake_drop_first_n_packets; | 77 fake_drop_first_n_packets_ = fake_drop_first_n_packets; |
| 75 } | 78 } |
| 76 | 79 |
| 77 // The percent of time WritePacket will block and set WriteResult's status | 80 // The percent of time WritePacket will block and set WriteResult's status |
| 78 // to WRITE_STATUS_BLOCKED. | 81 // to WRITE_STATUS_BLOCKED. |
| 79 void set_fake_blocked_socket_percentage( | 82 void set_fake_blocked_socket_percentage( |
| 80 int32 fake_blocked_socket_percentage) { | 83 int32_t fake_blocked_socket_percentage) { |
| 81 DCHECK(clock_); | 84 DCHECK(clock_); |
| 82 base::AutoLock locked(config_mutex_); | 85 base::AutoLock locked(config_mutex_); |
| 83 fake_blocked_socket_percentage_ = fake_blocked_socket_percentage; | 86 fake_blocked_socket_percentage_ = fake_blocked_socket_percentage; |
| 84 } | 87 } |
| 85 | 88 |
| 86 // The percent of time a packet is simulated as being reordered. | 89 // The percent of time a packet is simulated as being reordered. |
| 87 void set_fake_reorder_percentage(int32 fake_packet_reorder_percentage) { | 90 void set_fake_reorder_percentage(int32_t fake_packet_reorder_percentage) { |
| 88 DCHECK(clock_); | 91 DCHECK(clock_); |
| 89 base::AutoLock locked(config_mutex_); | 92 base::AutoLock locked(config_mutex_); |
| 90 DCHECK(!fake_packet_delay_.IsZero()); | 93 DCHECK(!fake_packet_delay_.IsZero()); |
| 91 fake_packet_reorder_percentage_ = fake_packet_reorder_percentage; | 94 fake_packet_reorder_percentage_ = fake_packet_reorder_percentage; |
| 92 } | 95 } |
| 93 | 96 |
| 94 // The delay before writing this packet. | 97 // The delay before writing this packet. |
| 95 void set_fake_packet_delay(QuicTime::Delta fake_packet_delay) { | 98 void set_fake_packet_delay(QuicTime::Delta fake_packet_delay) { |
| 96 DCHECK(clock_); | 99 DCHECK(clock_); |
| 97 base::AutoLock locked(config_mutex_); | 100 base::AutoLock locked(config_mutex_); |
| 98 fake_packet_delay_ = fake_packet_delay; | 101 fake_packet_delay_ = fake_packet_delay; |
| 99 } | 102 } |
| 100 | 103 |
| 101 // The maximum bandwidth and buffer size of the connection. When these are | 104 // The maximum bandwidth and buffer size of the connection. When these are |
| 102 // set, packets will be delayed until a connection with that bandwidth would | 105 // set, packets will be delayed until a connection with that bandwidth would |
| 103 // transmit it. Once the |buffer_size| is reached, all new packets are | 106 // transmit it. Once the |buffer_size| is reached, all new packets are |
| 104 // dropped. | 107 // dropped. |
| 105 void set_max_bandwidth_and_buffer_size(QuicBandwidth fake_bandwidth, | 108 void set_max_bandwidth_and_buffer_size(QuicBandwidth fake_bandwidth, |
| 106 QuicByteCount buffer_size) { | 109 QuicByteCount buffer_size) { |
| 107 DCHECK(clock_); | 110 DCHECK(clock_); |
| 108 base::AutoLock locked(config_mutex_); | 111 base::AutoLock locked(config_mutex_); |
| 109 fake_bandwidth_ = fake_bandwidth; | 112 fake_bandwidth_ = fake_bandwidth; |
| 110 buffer_size_ = buffer_size; | 113 buffer_size_ = buffer_size; |
| 111 } | 114 } |
| 112 | 115 |
| 113 // Useful for reproducing very flaky issues. | 116 // Useful for reproducing very flaky issues. |
| 114 void set_seed(uint64 seed) { simple_random_.set_seed(seed); } | 117 void set_seed(uint64_t seed) { simple_random_.set_seed(seed); } |
| 115 | 118 |
| 116 private: | 119 private: |
| 117 // Writes out the next packet to the contained writer and returns the time | 120 // Writes out the next packet to the contained writer and returns the time |
| 118 // for the next delayed packet to be written. | 121 // for the next delayed packet to be written. |
| 119 QuicTime ReleaseNextPacket(); | 122 QuicTime ReleaseNextPacket(); |
| 120 | 123 |
| 121 // A single packet which will be sent at the supplied send_time. | 124 // A single packet which will be sent at the supplied send_time. |
| 122 struct DelayedWrite { | 125 struct DelayedWrite { |
| 123 public: | 126 public: |
| 124 DelayedWrite(const char* buffer, | 127 DelayedWrite(const char* buffer, |
| (...skipping 12 matching lines...) Expand all Loading... |
| 137 typedef std::list<DelayedWrite> DelayedPacketList; | 140 typedef std::list<DelayedWrite> DelayedPacketList; |
| 138 | 141 |
| 139 const QuicClock* clock_; | 142 const QuicClock* clock_; |
| 140 scoped_ptr<QuicAlarm> write_unblocked_alarm_; | 143 scoped_ptr<QuicAlarm> write_unblocked_alarm_; |
| 141 scoped_ptr<QuicAlarm> delay_alarm_; | 144 scoped_ptr<QuicAlarm> delay_alarm_; |
| 142 scoped_ptr<Delegate> on_can_write_; | 145 scoped_ptr<Delegate> on_can_write_; |
| 143 net::test::SimpleRandom simple_random_; | 146 net::test::SimpleRandom simple_random_; |
| 144 // Stored packets delayed by fake packet delay or bandwidth restrictions. | 147 // Stored packets delayed by fake packet delay or bandwidth restrictions. |
| 145 DelayedPacketList delayed_packets_; | 148 DelayedPacketList delayed_packets_; |
| 146 QuicByteCount cur_buffer_size_; | 149 QuicByteCount cur_buffer_size_; |
| 147 uint64 num_calls_to_write_; | 150 uint64_t num_calls_to_write_; |
| 148 | 151 |
| 149 base::Lock config_mutex_; | 152 base::Lock config_mutex_; |
| 150 int32 fake_packet_loss_percentage_; | 153 int32_t fake_packet_loss_percentage_; |
| 151 int32 fake_drop_first_n_packets_; | 154 int32_t fake_drop_first_n_packets_; |
| 152 int32 fake_blocked_socket_percentage_; | 155 int32_t fake_blocked_socket_percentage_; |
| 153 int32 fake_packet_reorder_percentage_; | 156 int32_t fake_packet_reorder_percentage_; |
| 154 QuicTime::Delta fake_packet_delay_; | 157 QuicTime::Delta fake_packet_delay_; |
| 155 QuicBandwidth fake_bandwidth_; | 158 QuicBandwidth fake_bandwidth_; |
| 156 QuicByteCount buffer_size_; | 159 QuicByteCount buffer_size_; |
| 157 | 160 |
| 158 DISALLOW_COPY_AND_ASSIGN(PacketDroppingTestWriter); | 161 DISALLOW_COPY_AND_ASSIGN(PacketDroppingTestWriter); |
| 159 }; | 162 }; |
| 160 | 163 |
| 161 } // namespace test | 164 } // namespace test |
| 162 } // namespace tools | 165 } // namespace tools |
| 163 } // namespace net | 166 } // namespace net |
| 164 | 167 |
| 165 #endif // NET_TOOLS_QUIC_TEST_TOOLS_PACKET_DROPPING_TEST_WRITER_H_ | 168 #endif // NET_TOOLS_QUIC_TEST_TOOLS_PACKET_DROPPING_TEST_WRITER_H_ |
| OLD | NEW |