1 files changed, 71 insertions, 0 deletions
diff --git a/third_party/libwebrtc/test/fuzzers/ulpfec_receiver_fuzzer.cc b/third_party/libwebrtc/test/fuzzers/ulpfec_receiver_fuzzer.cc
new file mode 100644
index 0000000000..0a29ba3259
--- /dev/null
+++ b/third_party/libwebrtc/test/fuzzers/ulpfec_receiver_fuzzer.cc
@@ -0,0 +1,71 @@
+/*
+ *  Copyright (c) 2017 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 "modules/rtp_rtcp/include/rtp_rtcp_defines.h"
+#include "modules/rtp_rtcp/source/byte_io.h"
+#include "modules/rtp_rtcp/source/rtp_packet_received.h"
+#include "modules/rtp_rtcp/source/ulpfec_receiver.h"
+#include "test/fuzzers/fuzz_data_helper.h"
+
+namespace webrtc {
+
+namespace {
+class DummyCallback : public RecoveredPacketReceiver {
+  void OnRecoveredPacket(const RtpPacketReceived& packet) override {}
+};
+}  // namespace
+
+void FuzzOneInput(const uint8_t* data, size_t size) {
+  constexpr size_t kMinDataNeeded = 12;
+  if (size < kMinDataNeeded || size > 2000) {
+    return;
+  }
+
+  uint32_t ulpfec_ssrc = ByteReader<uint32_t>::ReadLittleEndian(data + 0);
+  uint16_t ulpfec_seq_num = ByteReader<uint16_t>::ReadLittleEndian(data + 4);
+  uint32_t media_ssrc = ByteReader<uint32_t>::ReadLittleEndian(data + 6);
+  uint16_t media_seq_num = ByteReader<uint16_t>::ReadLittleEndian(data + 10);
+
+  DummyCallback callback;
+  UlpfecReceiver receiver(ulpfec_ssrc, 0, &callback, Clock::GetRealTimeClock());
+
+  test::FuzzDataHelper fuzz_data(rtc::MakeArrayView(data, size));
+  while (fuzz_data.CanReadBytes(kMinDataNeeded)) {
+    size_t packet_length = kRtpHeaderSize + fuzz_data.Read<uint8_t>();
+    auto raw_packet = fuzz_data.ReadByteArray(packet_length);
+
+    RtpPacketReceived parsed_packet;
+    if (!parsed_packet.Parse(raw_packet))
+      continue;
+
+    // Overwrite the fields for the sequence number and SSRC with
+    // consistent values for either a received UlpFEC packet or received media
+    // packet. (We're still relying on libfuzzer to manage to generate packet
+    // headers that interact together; this just ensures that we have two
+    // consistent streams).
+    if (fuzz_data.ReadOrDefaultValue<uint8_t>(0) % 2 == 0) {
+      // Simulate UlpFEC packet.
+      parsed_packet.SetSequenceNumber(ulpfec_seq_num++);
+      parsed_packet.SetSsrc(ulpfec_ssrc);
+    } else {
+      // Simulate media packet.
+      parsed_packet.SetSequenceNumber(media_seq_num++);
+      parsed_packet.SetSsrc(media_ssrc);
+    }
+
+    receiver.AddReceivedRedPacket(parsed_packet);
+  }
+
+  receiver.ProcessReceivedFec();
+}
+
+}  // namespace webrtc