diff --git a/webrtc/test/rtcp_packet_parser.h b/webrtc/test/rtcp_packet_parser.h index cc890b4f2f..562b4ed511 100644 --- a/webrtc/test/rtcp_packet_parser.h +++ b/webrtc/test/rtcp_packet_parser.h @@ -16,11 +16,32 @@ #include #include +#include "webrtc/base/buffer.h" +#include "webrtc/base/checks.h" +#include "webrtc/modules/rtp_rtcp/source/rtcp_packet/common_header.h" #include "webrtc/modules/rtp_rtcp/source/rtcp_utility.h" #include "webrtc/typedefs.h" namespace webrtc { namespace test { +// Parse RTCP packet of given type. Assumes RTCP header is valid and that there +// is excatly one packet of correct type in the buffer. +template +bool ParseSinglePacket(const uint8_t* buffer, size_t size, Packet* packet) { + rtcp::CommonHeader header; + RTC_CHECK(header.Parse(buffer, size)); + RTC_CHECK_EQ(static_cast(size), header.NextPacket() - buffer); + return packet->Parse(header); +} +// Same function, but takes raw buffer as single argument instead of pair. +template +bool ParseSinglePacket(const uint8_t(&buffer)[N], Packet* packet) { + return ParseSinglePacket(buffer, N, packet); +} +template +bool ParseSinglePacket(const rtc::Buffer& buffer, Packet* packet) { + return ParseSinglePacket(buffer.data(), buffer.size(), packet); +} class RtcpPacketParser;