#include "common.hpp"
#include "RTC/RTP/Codecs/VP8.hpp"
#include "RTC/RTP/Packet.hpp"
#include "test/include/RTC/RTP/rtpCommon.hpp"
#include <catch2/catch_test_macros.hpp>
#include <cstring>
namespace
{
RTC::RTP::Codecs::VP8::PayloadDescriptor* createVP8PayloadDescriptor(
uint8_t* buffer,
size_t bufferLen,
uint16_t pictureId,
uint8_t tl0PictureIndex,
uint8_t tlIndex,
bool layerSync = true)
{
uint16_t netPictureId = htons(pictureId);
std::memcpy(buffer + 2, &netPictureId, 2);
buffer[2] |= 0x80;
buffer[4] = tl0PictureIndex;
buffer[5] = tlIndex << 6;
if (layerSync)
{
buffer[5] |= 0x20; }
auto* payloadDescriptor = RTC::RTP::Codecs::VP8::Parse(buffer, bufferLen);
REQUIRE(payloadDescriptor);
return payloadDescriptor;
}
std::unique_ptr<RTC::RTP::Codecs::VP8::PayloadDescriptor> processVP8Packet(
RTC::RTP::Codecs::VP8::EncodingContext& context,
uint16_t pictureId,
uint8_t tl0PictureIndex,
uint8_t tlIndex,
bool layerSync = true)
{
uint8_t payload[] =
{
0x90, 0xe0, 0x80, 0x00, 0x00, 0x00
};
std::unique_ptr<RTC::RTP::Packet> packet{ RTC::RTP::Packet::Factory(
rtpCommon::FactoryBuffer, sizeof(rtpCommon::FactoryBuffer)) };
packet->SetPayload(payload, sizeof(payload));
bool marker;
auto* payloadDescriptor = createVP8PayloadDescriptor(
packet->GetPayload(), packet->GetPayloadLength(), pictureId, tl0PictureIndex, tlIndex, layerSync);
std::unique_ptr<RTC::RTP::Codecs::VP8::PayloadDescriptorHandler> payloadDescriptorHandler(
new RTC::RTP::Codecs::VP8::PayloadDescriptorHandler(payloadDescriptor));
if (payloadDescriptorHandler->Process(&context, packet.get(), marker))
{
return std::unique_ptr<RTC::RTP::Codecs::VP8::PayloadDescriptor>(
RTC::RTP::Codecs::VP8::Parse(packet->GetPayload(), packet->GetPayloadLength()));
}
return nullptr;
}
}
SCENARIO("VP8 payload descriptor", "[rtp][codecs][vp8]")
{
SECTION("parse payload descriptor")
{
uint8_t originalBuffer[] =
{
0xd0, 0x80, 0x11, 0x00
};
uint8_t buffer[4] = { 0 };
std::memcpy(buffer, originalBuffer, sizeof(buffer));
std::unique_ptr<RTC::RTP::Codecs::VP8::PayloadDescriptor> payloadDescriptor{
RTC::RTP::Codecs::VP8::Parse(buffer, sizeof(buffer))
};
REQUIRE(payloadDescriptor);
REQUIRE(payloadDescriptor->extended == 1);
REQUIRE(payloadDescriptor->nonReference == 0);
REQUIRE(payloadDescriptor->start == 1);
REQUIRE(payloadDescriptor->partitionIndex == 0);
REQUIRE(payloadDescriptor->i == 1);
REQUIRE(payloadDescriptor->l == 0);
REQUIRE(payloadDescriptor->t == 0);
REQUIRE(payloadDescriptor->k == 0);
REQUIRE(payloadDescriptor->pictureId == 17);
REQUIRE(payloadDescriptor->tl0PictureIndex == 0);
REQUIRE(payloadDescriptor->tlIndex == 0);
REQUIRE(payloadDescriptor->y == 0);
REQUIRE(payloadDescriptor->keyIndex == 0);
REQUIRE(payloadDescriptor->isKeyFrame == true);
REQUIRE(payloadDescriptor->hasPictureId == true);
REQUIRE(payloadDescriptor->hasOneBytePictureId == true);
REQUIRE(payloadDescriptor->hasTwoBytesPictureId == false);
REQUIRE(payloadDescriptor->hasTl0PictureIndex == false);
REQUIRE(payloadDescriptor->hasTlIndex == false);
SECTION("encode payload descriptor")
{
payloadDescriptor->Encode(
buffer, payloadDescriptor->pictureId, payloadDescriptor->tl0PictureIndex);
SECTION("compare encoded payload descriptor with original buffer")
{
REQUIRE(std::memcmp(buffer, originalBuffer, sizeof(buffer)) == 0);
}
}
}
SECTION("parse payload descriptor 2")
{
uint8_t originalBuffer[] =
{
0x88, 0x3e, 0xe4, 0x00
};
uint8_t buffer[4] = { 0 };
std::memcpy(buffer, originalBuffer, sizeof(buffer));
std::unique_ptr<RTC::RTP::Codecs::VP8::PayloadDescriptor> payloadDescriptor{
RTC::RTP::Codecs::VP8::Parse(buffer, sizeof(buffer))
};
REQUIRE(payloadDescriptor);
REQUIRE(payloadDescriptor->extended == 1);
REQUIRE(payloadDescriptor->nonReference == 0);
REQUIRE(payloadDescriptor->start == 0);
REQUIRE(payloadDescriptor->partitionIndex == 0);
REQUIRE(payloadDescriptor->i == 0);
REQUIRE(payloadDescriptor->l == 0);
REQUIRE(payloadDescriptor->t == 1);
REQUIRE(payloadDescriptor->k == 1);
REQUIRE(payloadDescriptor->pictureId == 0);
REQUIRE(payloadDescriptor->tl0PictureIndex == 0);
REQUIRE(payloadDescriptor->tlIndex == 3);
REQUIRE(payloadDescriptor->y == 1);
REQUIRE(payloadDescriptor->keyIndex == 4);
REQUIRE(payloadDescriptor->isKeyFrame == false);
REQUIRE(payloadDescriptor->hasPictureId == false);
REQUIRE(payloadDescriptor->hasOneBytePictureId == false);
REQUIRE(payloadDescriptor->hasTwoBytesPictureId == false);
REQUIRE(payloadDescriptor->hasTl0PictureIndex == false);
REQUIRE(payloadDescriptor->hasTlIndex == true);
SECTION("encode payload descriptor")
{
payloadDescriptor->Encode(
buffer, payloadDescriptor->pictureId, payloadDescriptor->tl0PictureIndex);
SECTION("compare encoded payloadDescriptor with original buffer")
{
REQUIRE(std::memcmp(buffer, originalBuffer, sizeof(buffer)) == 0);
}
}
};
SECTION("parse payload descriptor, encode")
{
uint8_t originalBuffer[] =
{
0xd0, 0xc0, 0x11, 0x03
};
uint8_t buffer[4] = { 0 };
std::memcpy(buffer, originalBuffer, sizeof(buffer));
std::unique_ptr<RTC::RTP::Codecs::VP8::PayloadDescriptor> payloadDescriptor{
RTC::RTP::Codecs::VP8::Parse(buffer, sizeof(buffer))
};
REQUIRE(payloadDescriptor);
REQUIRE(payloadDescriptor->pictureId == 17);
REQUIRE(payloadDescriptor->tl0PictureIndex == 3);
SECTION("encode payload descriptor")
{
payloadDescriptor->Encode(buffer, 20, 1);
std::unique_ptr<RTC::RTP::Codecs::VP8::PayloadDescriptor> payloadDescriptor{
RTC::RTP::Codecs::VP8::Parse(buffer, sizeof(buffer))
};
REQUIRE(payloadDescriptor->pictureId == 20);
REQUIRE(payloadDescriptor->tl0PictureIndex == 1);
}
SECTION("restore payload descriptor")
{
payloadDescriptor->Restore(buffer);
std::unique_ptr<RTC::RTP::Codecs::VP8::PayloadDescriptor> payloadDescriptor{
RTC::RTP::Codecs::VP8::Parse(buffer, sizeof(buffer))
};
REQUIRE(payloadDescriptor->pictureId == 17);
REQUIRE(payloadDescriptor->tl0PictureIndex == 3);
}
}
SECTION("parse payload descriptor, I flag set but no space for pictureId")
{
uint8_t buffer[] =
{
0xd0, 0x80
};
const auto* payloadDescriptor = RTC::RTP::Codecs::VP8::Parse(buffer, sizeof(buffer));
REQUIRE(!payloadDescriptor);
}
SECTION("parse payload descriptor, X flag is not set, no keyframe")
{
uint8_t buffer[] =
{
0x50, 0x01
};
auto* payloadDescriptor = RTC::RTP::Codecs::VP8::Parse(buffer, sizeof(buffer));
REQUIRE(payloadDescriptor);
REQUIRE(payloadDescriptor->extended == 0);
REQUIRE(payloadDescriptor->nonReference == 0);
REQUIRE(payloadDescriptor->start == 1);
REQUIRE(payloadDescriptor->partitionIndex == 0);
REQUIRE(payloadDescriptor->i == 0);
REQUIRE(payloadDescriptor->l == 0);
REQUIRE(payloadDescriptor->t == 0);
REQUIRE(payloadDescriptor->k == 0);
REQUIRE(payloadDescriptor->pictureId == 0);
REQUIRE(payloadDescriptor->tl0PictureIndex == 0);
REQUIRE(payloadDescriptor->tlIndex == 0);
REQUIRE(payloadDescriptor->y == 0);
REQUIRE(payloadDescriptor->keyIndex == 0);
REQUIRE(payloadDescriptor->isKeyFrame == false);
REQUIRE(payloadDescriptor->hasPictureId == false);
REQUIRE(payloadDescriptor->hasOneBytePictureId == false);
REQUIRE(payloadDescriptor->hasTwoBytesPictureId == false);
REQUIRE(payloadDescriptor->hasTl0PictureIndex == false);
REQUIRE(payloadDescriptor->hasTlIndex == false);
delete payloadDescriptor;
}
SECTION("parse payload descriptor, X flag is not set, keyframe")
{
uint8_t buffer[] =
{
0x50, 0x00
};
auto* payloadDescriptor = RTC::RTP::Codecs::VP8::Parse(buffer, sizeof(buffer));
REQUIRE(payloadDescriptor);
REQUIRE(payloadDescriptor->extended == 0);
REQUIRE(payloadDescriptor->nonReference == 0);
REQUIRE(payloadDescriptor->start == 1);
REQUIRE(payloadDescriptor->partitionIndex == 0);
REQUIRE(payloadDescriptor->i == 0);
REQUIRE(payloadDescriptor->l == 0);
REQUIRE(payloadDescriptor->t == 0);
REQUIRE(payloadDescriptor->k == 0);
REQUIRE(payloadDescriptor->pictureId == 0);
REQUIRE(payloadDescriptor->tl0PictureIndex == 0);
REQUIRE(payloadDescriptor->tlIndex == 0);
REQUIRE(payloadDescriptor->y == 0);
REQUIRE(payloadDescriptor->keyIndex == 0);
REQUIRE(payloadDescriptor->isKeyFrame == true);
REQUIRE(payloadDescriptor->hasPictureId == false);
REQUIRE(payloadDescriptor->hasOneBytePictureId == false);
REQUIRE(payloadDescriptor->hasTwoBytesPictureId == false);
REQUIRE(payloadDescriptor->hasTl0PictureIndex == false);
REQUIRE(payloadDescriptor->hasTlIndex == false);
delete payloadDescriptor;
}
}
SCENARIO("process VP8 payload descriptor", "[rtp][codecs][vp8]")
{
constexpr uint16_t MaxPictureId = (1 << 15) - 1;
SECTION("do not drop TL0PICIDX from temporal layers higher than 0")
{
RTC::RTP::Codecs::EncodingContext::Params params;
params.spatialLayers = 0;
params.temporalLayers = 2;
RTC::RTP::Codecs::VP8::EncodingContext context(params);
context.SetCurrentTemporalLayer(0);
context.SetTargetTemporalLayer(0);
auto forwarded = processVP8Packet(context, 0, 0, 0);
REQUIRE(forwarded);
REQUIRE(forwarded->pictureId == 0);
REQUIRE(forwarded->tl0PictureIndex == 0);
forwarded = processVP8Packet(context, 2, 1, 1);
REQUIRE(!forwarded);
forwarded = processVP8Packet(context, 1, 1, 0);
REQUIRE(forwarded);
REQUIRE(forwarded->pictureId == 1);
REQUIRE(forwarded->tl0PictureIndex == 1);
}
SECTION("drop packets that belong to other temporal layers after rolling over pictureID")
{
RTC::RTP::Codecs::EncodingContext::Params params;
params.spatialLayers = 0;
params.temporalLayers = 2;
RTC::RTP::Codecs::VP8::EncodingContext context(params);
context.SyncRequired();
context.SetCurrentTemporalLayer(0);
context.SetTargetTemporalLayer(0);
auto forwarded = processVP8Packet(context, MaxPictureId, 0, 0);
REQUIRE(forwarded);
REQUIRE(forwarded->pictureId == 1);
REQUIRE(forwarded->tl0PictureIndex == 1);
forwarded = processVP8Packet(context, 0, 0, 0);
REQUIRE(forwarded);
REQUIRE(forwarded->pictureId == 2);
REQUIRE(forwarded->tl0PictureIndex == 1);
forwarded = processVP8Packet(context, 1, 0, 1);
REQUIRE(!forwarded);
}
SECTION("old packets with higher temporal layer than current are dropped")
{
RTC::RTP::Codecs::EncodingContext::Params params;
params.spatialLayers = 0;
params.temporalLayers = 2;
RTC::RTP::Codecs::VP8::EncodingContext context(params);
context.SyncRequired();
context.SetCurrentTemporalLayer(0);
context.SetTargetTemporalLayer(0);
auto forwarded = processVP8Packet(context, 1, 0, 0);
REQUIRE(forwarded);
REQUIRE(forwarded->pictureId == 1);
REQUIRE(forwarded->tlIndex == 0);
REQUIRE(forwarded->tl0PictureIndex == 1);
forwarded = processVP8Packet(context, 2, 0, 0);
REQUIRE(forwarded);
REQUIRE(forwarded->pictureId == 2);
REQUIRE(forwarded->tlIndex == 0);
REQUIRE(forwarded->tl0PictureIndex == 1);
forwarded = processVP8Packet(context, 0, 0, 1);
REQUIRE(!forwarded);
REQUIRE(context.GetCurrentTemporalLayer() == 0);
}
SECTION("packets with higher temporal layer than current are dropped")
{
RTC::RTP::Codecs::EncodingContext::Params params;
params.spatialLayers = 0;
params.temporalLayers = 2;
RTC::RTP::Codecs::VP8::EncodingContext context(params);
context.SyncRequired();
context.SetCurrentTemporalLayer(0);
context.SetTargetTemporalLayer(0);
auto forwarded = processVP8Packet(context, 1, 0, 0);
REQUIRE(forwarded);
REQUIRE(forwarded->pictureId == 1);
REQUIRE(forwarded->tlIndex == 0);
REQUIRE(forwarded->tl0PictureIndex == 1);
forwarded = processVP8Packet(context, 2, 0, 0);
REQUIRE(forwarded);
REQUIRE(forwarded->pictureId == 2);
REQUIRE(forwarded->tlIndex == 0);
REQUIRE(forwarded->tl0PictureIndex == 1);
context.SetTargetTemporalLayer(2);
forwarded = processVP8Packet(context, 3, 0, 1);
REQUIRE(!forwarded);
REQUIRE(context.GetCurrentTemporalLayer() == 0);
}
}
SCENARIO("encode VP8 payload descriptor", "[rtp][codecs][vp8]")
{
uint8_t payload[] =
{
0x80, 0xe0, 0x01, 0x01,
0xe8, 0x40, 0x7a, 0xd8
};
bool marker;
SECTION("encode based on specific encoder")
{
auto* payloadDescriptor = RTC::RTP::Codecs::VP8::Parse(payload, sizeof(payload));
REQUIRE(payloadDescriptor);
RTC::RTP::Codecs::EncodingContext::Params params;
params.spatialLayers = 0;
params.temporalLayers = 3;
RTC::RTP::Codecs::VP8::EncodingContext context(params);
context.SetCurrentTemporalLayer(3);
context.SetTargetTemporalLayer(3);
REQUIRE(payloadDescriptor->pictureId == 1);
auto* payloadDescriptorHandler =
new RTC::RTP::Codecs::VP8::PayloadDescriptorHandler(payloadDescriptor);
std::unique_ptr<RTC::RTP::Packet> packet{ RTC::RTP::Packet::Factory(
rtpCommon::FactoryBuffer, sizeof(rtpCommon::FactoryBuffer)) };
packet->SetPayload(payload, sizeof(payload));
auto forwarded = payloadDescriptorHandler->Process(&context, packet.get(), marker);
REQUIRE(forwarded);
auto encoder1 = payloadDescriptorHandler->GetEncoder();
REQUIRE(encoder1);
payloadDescriptor->pictureId = 2;
packet.reset(
RTC::RTP::Packet::Factory(rtpCommon::FactoryBuffer, sizeof(rtpCommon::FactoryBuffer)));
packet->SetPayload(payload, sizeof(payload));
forwarded = payloadDescriptorHandler->Process(&context, packet.get(), marker);
REQUIRE(forwarded);
REQUIRE(payloadDescriptor->pictureId == 2);
auto encoder2 = payloadDescriptorHandler->GetEncoder();
REQUIRE(encoder2);
packet.reset(
RTC::RTP::Packet::Factory(rtpCommon::FactoryBuffer, sizeof(rtpCommon::FactoryBuffer)));
packet->SetPayload(payload, sizeof(payload));
payloadDescriptorHandler->Encode(packet.get(), encoder1.get());
auto* payloadDescriptor2 = RTC::RTP::Codecs::VP8::Parse(payload, sizeof(payload));
REQUIRE(payloadDescriptor2);
REQUIRE(payloadDescriptor2->pictureId == 1);
packet.reset(
RTC::RTP::Packet::Factory(rtpCommon::FactoryBuffer, sizeof(rtpCommon::FactoryBuffer)));
packet->SetPayload(payload, sizeof(payload));
payloadDescriptorHandler->Encode(packet.get(), encoder2.get());
auto* payloadDescriptor3 =
RTC::RTP::Codecs::VP8::Parse(packet->GetPayload(), packet->GetPayloadLength());
REQUIRE(payloadDescriptor3);
REQUIRE(payloadDescriptor3->pictureId == 2);
delete payloadDescriptor3;
delete payloadDescriptor2;
delete payloadDescriptorHandler;
}
}