diff --git a/webrtc/test/fuzzers/BUILD.gn b/webrtc/test/fuzzers/BUILD.gn index c8ea39913e..141dcfe514 100644 --- a/webrtc/test/fuzzers/BUILD.gn +++ b/webrtc/test/fuzzers/BUILD.gn @@ -86,6 +86,16 @@ webrtc_fuzzer_test("flexfec_header_reader_fuzzer") { ] } +webrtc_fuzzer_test("flexfec_sender_fuzzer") { + sources = [ + "flexfec_sender_fuzzer.cc", + ] + deps = [ + "../../modules/rtp_rtcp/", + ] + libfuzzer_options = [ "max_len=200" ] +} + webrtc_fuzzer_test("ulpfec_header_reader_fuzzer") { sources = [ "ulpfec_header_reader_fuzzer.cc", diff --git a/webrtc/test/fuzzers/flexfec_sender_fuzzer.cc b/webrtc/test/fuzzers/flexfec_sender_fuzzer.cc new file mode 100644 index 0000000000..bc486dda07 --- /dev/null +++ b/webrtc/test/fuzzers/flexfec_sender_fuzzer.cc @@ -0,0 +1,68 @@ +/* + * 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 + +#include "webrtc/modules/rtp_rtcp/include/flexfec_sender.h" +#include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" +#include "webrtc/modules/rtp_rtcp/source/rtp_packet_to_send.h" +#include "webrtc/modules/rtp_rtcp/source/byte_io.h" +#include "webrtc/system_wrappers/include/clock.h" + +namespace webrtc { + +namespace { + +constexpr int kFlexfecPayloadType = 123; +constexpr uint32_t kMediaSsrc = 1234; +constexpr uint32_t kFlexfecSsrc = 5678; +const std::vector kNoRtpHeaderExtensions; + +} // namespace + +void FuzzOneInput(const uint8_t* data, size_t size) { + size_t i = 0; + if (size < 5) { + return; + } + + SimulatedClock clock(1 + data[i++]); + FlexfecSender sender(kFlexfecPayloadType, kFlexfecSsrc, kMediaSsrc, + kNoRtpHeaderExtensions, &clock); + FecProtectionParams params = { + data[i++], static_cast(data[i++] % 100), + data[i++] <= 127 ? kFecMaskRandom : kFecMaskBursty}; + sender.SetFecParameters(params); + uint16_t seq_num = data[i++]; + + while (i + 1 < size) { + // Everything past the base RTP header (12 bytes) is payload, + // from the perspective of FlexFEC. + size_t payload_size = data[i++]; + if (i + kRtpHeaderSize + payload_size >= size) + break; + std::unique_ptr packet( + new uint8_t[kRtpHeaderSize + payload_size]); + memcpy(packet.get(), &data[i], kRtpHeaderSize + payload_size); + i += kRtpHeaderSize + payload_size; + ByteWriter::WriteBigEndian(&packet[2], seq_num++); + ByteWriter::WriteBigEndian(&packet[8], kMediaSsrc); + RtpPacketToSend rtp_packet(nullptr); + if (!rtp_packet.Parse(packet.get(), kRtpHeaderSize + payload_size)) + break; + sender.AddRtpPacketAndGenerateFec(rtp_packet); + if (sender.FecAvailable()) { + std::vector> fec_packets = + sender.GetFecPackets(); + } + } +} + +} // namespace webrtc