1 /*
2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include <memory>
12
13 #include "modules/rtp_rtcp/include/flexfec_sender.h"
14 #include "modules/rtp_rtcp/include/rtp_rtcp_defines.h"
15 #include "modules/rtp_rtcp/source/byte_io.h"
16 #include "modules/rtp_rtcp/source/rtp_packet_to_send.h"
17 #include "system_wrappers/include/clock.h"
18
19 namespace webrtc {
20
21 namespace {
22
23 constexpr int kFlexfecPayloadType = 123;
24 constexpr uint32_t kMediaSsrc = 1234;
25 constexpr uint32_t kFlexfecSsrc = 5678;
26 const char kNoMid[] = "";
27 const std::vector<RtpExtension> kNoRtpHeaderExtensions;
28 const std::vector<RtpExtensionSize> kNoRtpHeaderExtensionSizes;
29
30 } // namespace
31
FuzzOneInput(const uint8_t * data,size_t size)32 void FuzzOneInput(const uint8_t* data, size_t size) {
33 size_t i = 0;
34 if (size < 5 || size > 200) {
35 return;
36 }
37 SimulatedClock clock(1 + data[i++]);
38 FlexfecSender sender(kFlexfecPayloadType, kFlexfecSsrc, kMediaSsrc, kNoMid,
39 kNoRtpHeaderExtensions, kNoRtpHeaderExtensionSizes,
40 nullptr /* rtp_state */, &clock);
41 FecProtectionParams params = {
42 data[i++], static_cast<int>(data[i++] % 100),
43 data[i++] <= 127 ? kFecMaskRandom : kFecMaskBursty};
44 sender.SetProtectionParameters(params, params);
45 uint16_t seq_num = data[i++];
46
47 while (i + 1 < size) {
48 // Everything past the base RTP header (12 bytes) is payload,
49 // from the perspective of FlexFEC.
50 size_t payload_size = data[i++];
51 if (i + kRtpHeaderSize + payload_size >= size)
52 break;
53 std::unique_ptr<uint8_t[]> packet(
54 new uint8_t[kRtpHeaderSize + payload_size]);
55 memcpy(packet.get(), &data[i], kRtpHeaderSize + payload_size);
56 i += kRtpHeaderSize + payload_size;
57 ByteWriter<uint16_t>::WriteBigEndian(&packet[2], seq_num++);
58 ByteWriter<uint32_t>::WriteBigEndian(&packet[8], kMediaSsrc);
59 RtpPacketToSend rtp_packet(nullptr);
60 if (!rtp_packet.Parse(packet.get(), kRtpHeaderSize + payload_size))
61 break;
62 sender.AddPacketAndGenerateFec(rtp_packet);
63 sender.GetFecPackets();
64 }
65 }
66
67 } // namespace webrtc
68