Index: webrtc/test/fuzzers/producer_fec_fuzzer.cc |
diff --git a/webrtc/test/fuzzers/producer_fec_fuzzer.cc b/webrtc/test/fuzzers/producer_fec_fuzzer.cc |
index 53f749386996e9397ad88b3062b243ad92e8f2ad..5d821527f314d5c3182a825324a7ae9293294b50 100644 |
--- a/webrtc/test/fuzzers/producer_fec_fuzzer.cc |
+++ b/webrtc/test/fuzzers/producer_fec_fuzzer.cc |
@@ -12,6 +12,7 @@ |
#include "webrtc/base/checks.h" |
#include "webrtc/modules/rtp_rtcp/source/byte_io.h" |
+#include "webrtc/modules/rtp_rtcp/source/fec_test_helper.h" |
#include "webrtc/modules/rtp_rtcp/source/producer_fec.h" |
namespace webrtc { |
@@ -38,9 +39,8 @@ void FuzzOneInput(const uint8_t* data, size_t size) { |
ByteWriter<uint16_t>::WriteBigEndian(&packet[2], seq_num++); |
i += payload_size + rtp_header_length; |
// Make sure sequence numbers are increasing. |
- const int kRedPayloadType = 98; |
- std::unique_ptr<RedPacket> red_packet(producer.BuildRedPacket( |
- packet.get(), payload_size, rtp_header_length, kRedPayloadType)); |
+ std::unique_ptr<RedPacket> red_packet = ProducerFec::BuildRedPacket( |
+ packet.get(), payload_size, rtp_header_length, kRedPayloadType); |
const bool protect = data[i++] % 2 == 1; |
if (protect) { |
producer.AddRtpPacketAndGenerateFec(packet.get(), payload_size, |
@@ -48,11 +48,12 @@ void FuzzOneInput(const uint8_t* data, size_t size) { |
} |
const size_t num_fec_packets = producer.NumAvailableFecPackets(); |
if (num_fec_packets > 0) { |
- std::vector<RedPacket*> fec_packets = |
- producer.GetFecPackets(kRedPayloadType, 99, 100, rtp_header_length); |
+ std::vector<std::unique_ptr<RedPacket>> fec_packets = |
+ producer.GetFecPacketsAsRed(kRedPayloadType, |
+ kFecPayloadType, |
+ 100, |
+ rtp_header_length); |
RTC_CHECK_EQ(num_fec_packets, fec_packets.size()); |
- for (RedPacket* fec_packet : fec_packets) |
- delete fec_packet; |
} |
} |
} |