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 |
new file mode 100644 |
index 0000000000000000000000000000000000000000..da65f5871933e2624785efb59417962f847982dc |
--- /dev/null |
+++ b/webrtc/test/fuzzers/transport_feedback_packet_loss_tracker_fuzzer.cc |
@@ -0,0 +1,144 @@ |
+/* |
+ * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. |
+ * |
+ * Use of this source code is governed by a BSD-style license |
+ * that can be found in the LICENSE file in the root of the source |
+ * tree. An additional intellectual property rights grant can be found |
+ * in the file PATENTS. All contributing project authors may |
+ * be found in the AUTHORS file in the root of the source tree. |
+ */ |
+ |
+#include <algorithm> |
+ |
+#include "webrtc/base/array_view.h" |
+#include "webrtc/modules/rtp_rtcp/source/byte_io.h" |
+#include "webrtc/modules/rtp_rtcp/source/rtcp_packet/transport_feedback.h" |
+#include "webrtc/voice_engine/transport_feedback_packet_loss_tracker.h" |
+ |
+namespace webrtc { |
+ |
+namespace { |
+ |
+using TransportFeedback = rtcp::TransportFeedback; |
+ |
+class FuzzTransportFeedback final : public rtcp::TransportFeedbackInterface { |
+ public: |
+ explicit FuzzTransportFeedback(rtc::ArrayView<const uint8_t> data) |
+ : data_(data), ended_(false), data_idx_(0) { |
+ ParseNextTransportFeedback(); |
+ } |
+ |
+ uint16_t GetBaseSequence() const override { return base_seq_num_; } |
+ |
+ std::vector<StatusSymbol> GetStatusVector() const override { |
+ return statuses_; |
+ } |
+ |
+ void ParseNextTransportFeedback() { |
+ statuses_.clear(); |
+ |
+ if (!ReadData<uint16_t>(&base_seq_num_)) { |
+ return; |
+ } |
+ |
+ uint16_t num_statuses = 0; |
+ if (!ReadData<uint16_t>(&num_statuses)) |
+ return; |
+ num_statuses = std::max<uint16_t>(num_statuses, 1); |
+ |
+ while (true) { |
+ uint8_t status_byte = 0; |
+ if (!ReadData<uint8_t>(&status_byte)) |
+ return; |
+ // Each status byte contains 8 statuses. |
+ for (size_t j = 0; j < 8; ++j) { |
+ if (status_byte & 0x01) { |
+ // |kReceivedSmallDelta| and |kReceivedLargeDelta| make no difference |
+ // to TransportFeedbackPacketLossTracker. So we test only one case. |
+ statuses_.push_back( |
+ TransportFeedback::StatusSymbol::kReceivedSmallDelta); |
+ } else { |
+ statuses_.push_back(TransportFeedback::StatusSymbol::kNotReceived); |
+ } |
+ if (statuses_.size() >= num_statuses) |
+ return; |
+ status_byte >>= 1; |
+ } |
+ } |
+ } |
+ |
+ bool ended() const { return ended_; } |
+ |
+ void SetBase(uint16_t base_sequence, int64_t ref_timestamp_us) override { |
+ RTC_NOTREACHED(); |
+ } |
+ void SetFeedbackSequenceNumber(uint8_t feedback_sequence) override { |
+ RTC_NOTREACHED(); |
+ } |
+ bool AddReceivedPacket(uint16_t sequence_number, |
+ int64_t timestamp_us) override { |
+ RTC_NOTREACHED(); |
+ return false; |
+ } |
+ std::vector<int16_t> GetReceiveDeltas() const override { |
+ RTC_NOTREACHED(); |
+ return std::vector<int16_t>(); |
+ } |
+ int64_t GetBaseTimeUs() const override { |
+ RTC_NOTREACHED(); |
+ return 0; |
+ } |
+ std::vector<int64_t> GetReceiveDeltasUs() const override { |
+ RTC_NOTREACHED(); |
+ return std::vector<int64_t>(); |
+ } |
+ |
+ private: |
+ template <typename T> |
+ bool ReadData(T* value) { |
+ RTC_DCHECK(!ended_); |
+ if (data_idx_ + sizeof(T) > data_.size()) { |
+ ended_ = true; |
+ return false; |
+ } |
+ *value = ByteReader<T>::ReadBigEndian(&data_[data_idx_]); |
+ data_idx_ += sizeof(T); |
+ return true; |
+ } |
+ |
+ const rtc::ArrayView<const uint8_t> data_; |
+ bool ended_; |
+ size_t data_idx_; |
+ uint16_t base_seq_num_; |
+ std::vector<TransportFeedback::StatusSymbol> statuses_; |
+}; |
+ |
+} // namespace |
+ |
+void FuzzOneInput(const uint8_t* data, size_t size) { |
+ if (size < sizeof(uint32_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); |
+ |
+ TransportFeedbackPacketLossTracker tracker( |
+ std::min(window_size_1, window_size_2), |
+ std::max(window_size_1, window_size_2)); |
+ FuzzTransportFeedback feedback(rtc::ArrayView<const uint8_t>(data, size)); |
+ while (!feedback.ended()) { |
+ tracker.OnReceivedTransportFeedback(feedback); |
+ tracker.Validate(); |
+ feedback.ParseNextTransportFeedback(); |
+ } |
+} |
+ |
+} // namespace webrtc |