mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-17 07:37:51 +01:00
f87536c9de
This is a reland of49734dc0faPatchset 2 contains a fix for the fuzzer set up. Since we now parse an RtpPacket out of the fuzzer data, the header needs to be correct, otherwise we fail before even reaching the FEC code that we actually want to test. Bug: webrtc:11340, chromium:1052323, chromium:1055974 TBR=stefan@webrtc.org Original change's description: > Reland "Refactors UlpFec and FlexFec to use a common interface." > > This is a reland of11af1d7444> > Original change's description: > > Refactors UlpFec and FlexFec to use a common interface. > > > > The new VideoFecGenerator is now injected into RtpSenderVideo, > > and generalizes the usage. > > This also prepares for being able to genera FEC in the RTP egress > > module. > > > > Bug: webrtc:11340 > > Change-Id: I8aa873129b2fb4131eb3399ee88f6ea2747155a3 > > Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/168347 > > Reviewed-by: Stefan Holmer <stefan@webrtc.org> > > Reviewed-by: Sebastian Jansson <srte@webrtc.org> > > Reviewed-by: Rasmus Brandt <brandtr@webrtc.org> > > Commit-Queue: Erik Språng <sprang@webrtc.org> > > Cr-Commit-Position: refs/heads/master@{#30515} > > Bug: webrtc:11340, chromium:1052323 > Change-Id: Id646047365f1c46cca9e6f3e8eefa5151207b4a0 > Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/168608 > Commit-Queue: Erik Språng <sprang@webrtc.org> > Reviewed-by: Stefan Holmer <stefan@webrtc.org> > Cr-Commit-Position: refs/heads/master@{#30593} Bug: webrtc:11340, chromium:1052323 Change-Id: Ib8925f44e2edfcfeadc95c845c3bfc23822604ed Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/169222 Commit-Queue: Erik Språng <sprang@webrtc.org> Reviewed-by: Ilya Nikolaevskiy <ilnik@webrtc.org> Cr-Commit-Position: refs/heads/master@{#30724}
67 lines
2.3 KiB
C++
67 lines
2.3 KiB
C++
/*
|
|
* 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 <memory>
|
|
|
|
#include "modules/rtp_rtcp/include/flexfec_sender.h"
|
|
#include "modules/rtp_rtcp/include/rtp_rtcp_defines.h"
|
|
#include "modules/rtp_rtcp/source/byte_io.h"
|
|
#include "modules/rtp_rtcp/source/rtp_packet_to_send.h"
|
|
#include "system_wrappers/include/clock.h"
|
|
|
|
namespace webrtc {
|
|
|
|
namespace {
|
|
|
|
constexpr int kFlexfecPayloadType = 123;
|
|
constexpr uint32_t kMediaSsrc = 1234;
|
|
constexpr uint32_t kFlexfecSsrc = 5678;
|
|
const char kNoMid[] = "";
|
|
const std::vector<RtpExtension> kNoRtpHeaderExtensions;
|
|
const std::vector<RtpExtensionSize> kNoRtpHeaderExtensionSizes;
|
|
|
|
} // namespace
|
|
|
|
void FuzzOneInput(const uint8_t* data, size_t size) {
|
|
size_t i = 0;
|
|
if (size < 5 || size > 200) {
|
|
return;
|
|
}
|
|
SimulatedClock clock(1 + data[i++]);
|
|
FlexfecSender sender(kFlexfecPayloadType, kFlexfecSsrc, kMediaSsrc, kNoMid,
|
|
kNoRtpHeaderExtensions, kNoRtpHeaderExtensionSizes,
|
|
nullptr /* rtp_state */, &clock);
|
|
FecProtectionParams params = {
|
|
data[i++], static_cast<int>(data[i++] % 100),
|
|
data[i++] <= 127 ? kFecMaskRandom : kFecMaskBursty};
|
|
sender.SetProtectionParameters(params, 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<uint8_t[]> packet(
|
|
new uint8_t[kRtpHeaderSize + payload_size]);
|
|
memcpy(packet.get(), &data[i], kRtpHeaderSize + payload_size);
|
|
i += kRtpHeaderSize + payload_size;
|
|
ByteWriter<uint16_t>::WriteBigEndian(&packet[2], seq_num++);
|
|
ByteWriter<uint32_t>::WriteBigEndian(&packet[8], kMediaSsrc);
|
|
RtpPacketToSend rtp_packet(nullptr);
|
|
if (!rtp_packet.Parse(packet.get(), kRtpHeaderSize + payload_size))
|
|
break;
|
|
sender.AddPacketAndGenerateFec(rtp_packet);
|
|
sender.GetFecPackets();
|
|
}
|
|
}
|
|
|
|
} // namespace webrtc
|