Chromium Code Reviews| Index: webrtc/test/fuzzers/transport_feedback_packet_loss_tracker_fuzzer.cc |
| diff --git a/webrtc/test/fuzzers/transport_feedback_packet_loss_tracker_fuzzer.cc b/webrtc/test/fuzzers/transport_feedback_packet_loss_tracker_fuzzer.cc |
| index 837ae883f6c83f213c9951d0f21832d363359cc7..a22a4d4ecb2b0038eb9bbcce281378c8a4cb79ac 100644 |
| --- a/webrtc/test/fuzzers/transport_feedback_packet_loss_tracker_fuzzer.cc |
| +++ b/webrtc/test/fuzzers/transport_feedback_packet_loss_tracker_fuzzer.cc |
| @@ -19,6 +19,29 @@ namespace webrtc { |
| namespace { |
| +template <typename T> |
| +T FuzzInput(const uint8_t** data, size_t* size) { |
|
stefan-webrtc
2017/02/01 13:11:19
Seems like over complicating things to me that we
sprang_webrtc
2017/02/01 14:27:19
Only used in one place, seems like you don't need
elad.alon_webrtc.org
2017/02/01 15:25:49
Please see the subsequent CL in the series, where
|
| + RTC_CHECK(*size >= sizeof(T)); |
|
sprang_webrtc
2017/02/01 14:27:19
nit: RTC_CHECK_GE
elad.alon_webrtc.org
2017/02/01 15:25:49
Done.
|
| + T rc = ByteReader<T>::ReadBigEndian(*data); |
| + *data += sizeof(T); |
| + *size -= sizeof(T); |
| + return rc; |
| +} |
| + |
| +size_t FuzzInRange(const uint8_t** data, |
| + size_t* size, |
| + size_t lower, |
| + size_t upper) { |
| + // Achieve a close-to-uniform distribution. |
| + RTC_CHECK_LE(lower, upper); |
| + RTC_CHECK_LT(upper - lower, 1 << (8 * sizeof(uint16_t))); |
| + const size_t range = upper - lower; |
| + const uint16_t fuzzed = FuzzInput<uint16_t>(data, size); |
| + const size_t offset = (static_cast<float>(fuzzed) / 0x10000) * (range + 1); |
|
minyue-webrtc
2017/01/31 12:50:19
nit: you probably need an explicit static_cast<siz
elad.alon_webrtc.org
2017/01/31 13:43:54
Do you think I should do that even if all bots pas
minyue-webrtc
2017/01/31 15:09:39
Forget about my saying of std::floor, it returns a
|
| + RTC_CHECK_LE(offset, range); // (fuzzed <= 0xffff) -> (offset < range + 1) |
|
minyue-webrtc
2017/01/31 12:53:27
BTW, this CHECK is not necessary. Aren't you sure
elad.alon_webrtc.org
2017/01/31 13:43:54
This check explains why the calculation above was
minyue-webrtc
2017/01/31 15:09:39
It looks redundant. 40 can speaks for itself.
|
| + return lower + offset; |
| +} |
| + |
| class TransportFeedbackGenerator { |
| public: |
| explicit TransportFeedbackGenerator(rtc::ArrayView<const uint8_t> data) |
| @@ -63,7 +86,7 @@ class TransportFeedbackGenerator { |
| private: |
| template <typename T> |
| bool ReadData(T* value) { |
| - RTC_DCHECK(!ended_); |
| + RTC_CHECK(!ended_); |
| if (data_idx_ + sizeof(T) > data_.size()) { |
| ended_ = true; |
| return false; |
| @@ -81,25 +104,25 @@ class TransportFeedbackGenerator { |
| } // namespace |
| void FuzzOneInput(const uint8_t* data, size_t size) { |
| - if (size < sizeof(uint32_t)) { |
| + if (size < 3 * sizeof(uint16_t)) { |
| return; |
| } |
| constexpr size_t kSeqNumHalf = 0x8000u; |
| - const size_t window_size_1 = std::min<size_t>( |
| - kSeqNumHalf, |
| - std::max<uint16_t>(1, ByteReader<uint16_t>::ReadBigEndian(data))); |
| - data += sizeof(uint16_t); |
| - const size_t window_size_2 = std::min<size_t>( |
| - kSeqNumHalf, |
| - std::max<uint16_t>(1, ByteReader<uint16_t>::ReadBigEndian(data))); |
| - data += sizeof(uint16_t); |
| - size -= 2 * sizeof(uint16_t); |
| + |
| + // 0x8000 >= max_window_size >= plr_min_num_packets > rplr_min_num_pairs >= 1 |
| + // (The distribution isn't uniform, but it's enough; more would be overkill.) |
| + const size_t max_window_size = FuzzInRange(&data, &size, 2, kSeqNumHalf); |
| + const size_t plr_min_num_packets = |
| + FuzzInRange(&data, &size, 2, max_window_size); |
| + const size_t rplr_min_num_pairs = |
| + FuzzInRange(&data, &size, 1, plr_min_num_packets - 1); |
| TransportFeedbackPacketLossTracker tracker( |
| - std::min(window_size_1, window_size_2), |
| - std::max(window_size_1, window_size_2)); |
| + max_window_size, plr_min_num_packets, rplr_min_num_pairs); |
| + |
| TransportFeedbackGenerator feedback_generator( |
| rtc::ArrayView<const uint8_t>(data, size)); |
| + |
| while (!feedback_generator.ended()) { |
| rtcp::TransportFeedback feedback; |
| feedback_generator.GetNextTransportFeedback(&feedback); |