Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(331)

Unified Diff: webrtc/test/fuzzers/transport_feedback_packet_loss_tracker_fuzzer.cc

Issue 2579613003: Add TransportFeedbackPacketLossTracker. (Closed)
Patch Set: on comments Created 3 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
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

Powered by Google App Engine
This is Rietveld 408576698