Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 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 "webrtc/modules/rtp_rtcp/include/flexfec_sender.h" | |
| 14 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" | |
| 15 #include "webrtc/modules/rtp_rtcp/source/rtp_packet_to_send.h" | |
| 16 #include "webrtc/modules/rtp_rtcp/source/byte_io.h" | |
| 17 #include "webrtc/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 std::vector<RtpExtension> kNoRtpHeaderExtensions; | |
| 27 | |
| 28 } // namespace | |
| 29 | |
| 30 void FuzzOneInput(const uint8_t* data, size_t size) { | |
| 31 size_t i = 0; | |
| 32 if (size < 5) { | |
| 33 return; | |
| 34 } | |
| 35 | |
| 36 SimulatedClock clock(1 + data[i++]); | |
| 37 std::unique_ptr<FlexfecSender> sender = | |
| 38 FlexfecSender::Create(kFlexfecPayloadType, kFlexfecSsrc, kMediaSsrc, | |
| 39 kNoRtpHeaderExtensions, &clock); | |
| 40 FecProtectionParams params = { | |
| 41 data[i++], static_cast<int>(data[i++] % 100), | |
| 42 data[i++] <= 127 ? kFecMaskRandom : kFecMaskBursty}; | |
| 43 sender->SetFecParameters(params); | |
| 44 uint16_t seq_num = data[i++]; | |
| 45 | |
| 46 while (i + 1 < size) { | |
| 47 // Everything past the base RTP header (12 bytes) is payload, | |
| 48 // from the perspective of FlexFEC. | |
| 49 size_t payload_size = data[i++]; | |
| 50 if (i + kRtpHeaderSize + payload_size >= size) | |
| 51 break; | |
| 52 std::unique_ptr<uint8_t[]> packet( | |
| 53 new uint8_t[kRtpHeaderSize + payload_size]); | |
| 54 memcpy(packet.get(), &data[i], kRtpHeaderSize + payload_size); | |
| 55 i += kRtpHeaderSize + payload_size; | |
| 56 ByteWriter<uint16_t>::WriteBigEndian(&packet[2], seq_num++); | |
|
brandtr
2016/10/31 11:07:32
These changes are needed due to change in FlexfecS
| |
| 57 ByteWriter<uint32_t>::WriteBigEndian(&packet[8], kMediaSsrc); | |
| 58 RtpPacketToSend rtp_packet(nullptr); | |
| 59 if (!rtp_packet.Parse(packet.get(), kRtpHeaderSize + payload_size)) | |
| 60 break; | |
| 61 sender->AddRtpPacketAndGenerateFec(rtp_packet); | |
| 62 if (sender->FecAvailable()) { | |
| 63 std::vector<std::unique_ptr<RtpPacketToSend>> fec_packets = | |
| 64 sender->GetFecPackets(); | |
| 65 } | |
| 66 } | |
| 67 } | |
| 68 | |
| 69 } // namespace webrtc | |
| OLD | NEW |