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

Unified Diff: webrtc/test/fuzzers/rtp_packet_fuzzer.cc

Issue 2805023002: Add read support of RtpStreamId/RepairedRtpStreamId header extensions. (Closed)
Patch Set: +rtp_header_fuzzer Created 3 years, 8 months 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 side-by-side diff with in-line comments
Download patch
Index: webrtc/test/fuzzers/rtp_packet_fuzzer.cc
diff --git a/webrtc/test/fuzzers/rtp_packet_fuzzer.cc b/webrtc/test/fuzzers/rtp_packet_fuzzer.cc
index 613f125e33da0a25e3a4af9e2ae69b1280ac7aea..867bb43b0a2b2457e741c1004481bd3111bc8e1a 100644
--- a/webrtc/test/fuzzers/rtp_packet_fuzzer.cc
+++ b/webrtc/test/fuzzers/rtp_packet_fuzzer.cc
@@ -14,30 +14,31 @@
namespace webrtc {
-// We decide which header extensions to register by reading one byte
+// We decide which header extensions to register by reading two bytes
// from the beginning of |data| and interpreting it as a bitmask over
-// the RTPExtensionType enum. This assert ensures one byte is enough.
-static_assert(kRtpExtensionNumberOfExtensions <= 8,
+// the RTPExtensionType enum. This assert ensures two bytes are enough.
+static_assert(kRtpExtensionNumberOfExtensions <= 16,
"Insufficient bits read to configure all header extensions. Add "
"an extra byte and update the switches.");
void FuzzOneInput(const uint8_t* data, size_t size) {
- if (size <= 1)
+ if (size <= 2)
return;
- // Don't use the configuration byte as part of the packet.
- std::bitset<8> extensionMask(data[0]);
- data++;
- size--;
+ // Don't use the configuration bytes as part of the packet.
+ std::bitset<16> extensionMask(*reinterpret_cast<const uint16_t*>(data));
+ data += 2;
+ size -= 2;
RtpPacketReceived::ExtensionManager extensions;
- for (int i = 0; i < kRtpExtensionNumberOfExtensions; i++) {
+ // Skip i = 0 since it maps to ExtensionNone and extension id = 0 is invalid.
+ for (int i = 1; i < kRtpExtensionNumberOfExtensions; i++) {
RTPExtensionType extension_type = static_cast<RTPExtensionType>(i);
if (extensionMask[i] && extension_type != kRtpExtensionNone) {
// Extensions are registered with an ID, which you signal to the
// peer so they know what to expect. This code only cares about
// parsing so the value of the ID isn't relevant; we use i.
- extensions.Register(extension_type, i);
+ extensions.RegisterByType(i, extension_type);
}
}
@@ -85,6 +86,16 @@ void FuzzOneInput(const uint8_t* data, size_t size) {
PlayoutDelay playout;
packet.GetExtension<PlayoutDelayLimits>(&playout);
break;
+ case kRtpExtensionRtpStreamId: {
+ std::string rsid;
+ packet.GetExtension<RtpStreamId>(&rsid);
+ break;
+ }
+ case kRtpExtensionRepairedRtpStreamId: {
+ std::string rsid;
+ packet.GetExtension<RepairedRtpStreamId>(&rsid);
+ break;
+ }
}
}
}
« webrtc/modules/rtp_rtcp/source/rtp_utility.cc ('K') | « webrtc/test/fuzzers/rtp_header_fuzzer.cc ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698