Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(78)

Side by Side Diff: webrtc/test/fuzzers/ulpfec_generator_fuzzer.cc

Issue 2449783002: Rename ProducerFec to UlpfecGenerator. (Closed)
Patch Set: Rebase. Created 4 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « webrtc/test/fuzzers/producer_fec_fuzzer.cc ('k') | no next file » | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 /* 1 /*
2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. 2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved.
3 * 3 *
4 * Use of this source code is governed by a BSD-style license 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 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 6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may 7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree. 8 * be found in the AUTHORS file in the root of the source tree.
9 */ 9 */
10 10
11 #include <memory> 11 #include <memory>
12 12
13 #include "webrtc/base/checks.h" 13 #include "webrtc/base/checks.h"
14 #include "webrtc/modules/rtp_rtcp/source/byte_io.h" 14 #include "webrtc/modules/rtp_rtcp/source/byte_io.h"
15 #include "webrtc/modules/rtp_rtcp/source/fec_test_helper.h" 15 #include "webrtc/modules/rtp_rtcp/source/fec_test_helper.h"
16 #include "webrtc/modules/rtp_rtcp/source/producer_fec.h" 16 #include "webrtc/modules/rtp_rtcp/source/ulpfec_generator.h"
17 17
18 namespace webrtc { 18 namespace webrtc {
19 19
20 namespace { 20 namespace {
21 constexpr uint8_t kFecPayloadType = 96; 21 constexpr uint8_t kFecPayloadType = 96;
22 constexpr uint8_t kRedPayloadType = 97; 22 constexpr uint8_t kRedPayloadType = 97;
23 } // namespace 23 } // namespace
24 24
25 void FuzzOneInput(const uint8_t* data, size_t size) { 25 void FuzzOneInput(const uint8_t* data, size_t size) {
26 ProducerFec producer; 26 UlpfecGenerator generator;
27 size_t i = 0; 27 size_t i = 0;
28 if (size < 4) 28 if (size < 4)
29 return; 29 return;
30 FecProtectionParams params = { 30 FecProtectionParams params = {
31 data[i++] % 128, static_cast<int>(data[i++] % 10), kFecMaskBursty}; 31 data[i++] % 128, static_cast<int>(data[i++] % 10), kFecMaskBursty};
32 producer.SetFecParameters(&params); 32 generator.SetFecParameters(&params);
33 uint16_t seq_num = data[i++]; 33 uint16_t seq_num = data[i++];
34 34
35 while (i + 3 < size) { 35 while (i + 3 < size) {
36 size_t rtp_header_length = data[i++] % 10 + 12; 36 size_t rtp_header_length = data[i++] % 10 + 12;
37 size_t payload_size = data[i++] % 10; 37 size_t payload_size = data[i++] % 10;
38 if (i + payload_size + rtp_header_length + 2 > size) 38 if (i + payload_size + rtp_header_length + 2 > size)
39 break; 39 break;
40 std::unique_ptr<uint8_t[]> packet( 40 std::unique_ptr<uint8_t[]> packet(
41 new uint8_t[payload_size + rtp_header_length]); 41 new uint8_t[payload_size + rtp_header_length]);
42 memcpy(packet.get(), &data[i], payload_size + rtp_header_length); 42 memcpy(packet.get(), &data[i], payload_size + rtp_header_length);
43 ByteWriter<uint16_t>::WriteBigEndian(&packet[2], seq_num++); 43 ByteWriter<uint16_t>::WriteBigEndian(&packet[2], seq_num++);
44 i += payload_size + rtp_header_length; 44 i += payload_size + rtp_header_length;
45 // Make sure sequence numbers are increasing. 45 // Make sure sequence numbers are increasing.
46 std::unique_ptr<RedPacket> red_packet = ProducerFec::BuildRedPacket( 46 std::unique_ptr<RedPacket> red_packet = UlpfecGenerator::BuildRedPacket(
47 packet.get(), payload_size, rtp_header_length, kRedPayloadType); 47 packet.get(), payload_size, rtp_header_length, kRedPayloadType);
48 const bool protect = data[i++] % 2 == 1; 48 const bool protect = data[i++] % 2 == 1;
49 if (protect) { 49 if (protect) {
50 producer.AddRtpPacketAndGenerateFec(packet.get(), payload_size, 50 generator.AddRtpPacketAndGenerateFec(packet.get(), payload_size,
51 rtp_header_length); 51 rtp_header_length);
52 } 52 }
53 const size_t num_fec_packets = producer.NumAvailableFecPackets(); 53 const size_t num_fec_packets = generator.NumAvailableFecPackets();
54 if (num_fec_packets > 0) { 54 if (num_fec_packets > 0) {
55 std::vector<std::unique_ptr<RedPacket>> fec_packets = 55 std::vector<std::unique_ptr<RedPacket>> fec_packets =
56 producer.GetUlpfecPacketsAsRed(kRedPayloadType, kFecPayloadType, 100, 56 generator.GetUlpfecPacketsAsRed(kRedPayloadType, kFecPayloadType, 100,
57 rtp_header_length); 57 rtp_header_length);
58 RTC_CHECK_EQ(num_fec_packets, fec_packets.size()); 58 RTC_CHECK_EQ(num_fec_packets, fec_packets.size());
59 } 59 }
60 } 60 }
61 } 61 }
62 } // namespace webrtc 62 } // namespace webrtc
OLDNEW
« no previous file with comments | « webrtc/test/fuzzers/producer_fec_fuzzer.cc ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698