aosp12/external/openscreen/cast/streaming/rtcp_common.cc

248 lines
8.9 KiB
C++

// Copyright 2019 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "cast/streaming/rtcp_common.h"
#include <algorithm>
#include <limits>
#include "cast/streaming/packet_util.h"
#include "util/saturate_cast.h"
namespace openscreen {
namespace cast {
RtcpCommonHeader::RtcpCommonHeader() = default;
RtcpCommonHeader::~RtcpCommonHeader() = default;
void RtcpCommonHeader::AppendFields(absl::Span<uint8_t>* buffer) const {
OSP_CHECK_GE(buffer->size(), kRtcpCommonHeaderSize);
uint8_t byte0 = kRtcpRequiredVersionAndPaddingBits
<< kRtcpReportCountFieldNumBits;
switch (packet_type) {
case RtcpPacketType::kSenderReport:
case RtcpPacketType::kReceiverReport:
OSP_DCHECK_LE(with.report_count,
FieldBitmask<int>(kRtcpReportCountFieldNumBits));
byte0 |= with.report_count;
break;
case RtcpPacketType::kSourceDescription:
OSP_UNIMPLEMENTED();
break;
case RtcpPacketType::kApplicationDefined:
case RtcpPacketType::kPayloadSpecific:
switch (with.subtype) {
case RtcpSubtype::kPictureLossIndicator:
case RtcpSubtype::kFeedback:
byte0 |= static_cast<uint8_t>(with.subtype);
break;
case RtcpSubtype::kReceiverLog:
OSP_UNIMPLEMENTED();
break;
default:
OSP_NOTREACHED();
break;
}
break;
case RtcpPacketType::kExtendedReports:
break;
case RtcpPacketType::kNull:
OSP_NOTREACHED();
break;
}
AppendField<uint8_t>(byte0, buffer);
AppendField<uint8_t>(static_cast<uint8_t>(packet_type), buffer);
// The size of the packet must be evenly divisible by the 32-bit word size.
OSP_DCHECK_EQ(0, payload_size % sizeof(uint32_t));
AppendField<uint16_t>(payload_size / sizeof(uint32_t), buffer);
}
// static
absl::optional<RtcpCommonHeader> RtcpCommonHeader::Parse(
absl::Span<const uint8_t> buffer) {
if (buffer.size() < kRtcpCommonHeaderSize) {
return absl::nullopt;
}
const uint8_t byte0 = ConsumeField<uint8_t>(&buffer);
if ((byte0 >> kRtcpReportCountFieldNumBits) !=
kRtcpRequiredVersionAndPaddingBits) {
return absl::nullopt;
}
const uint8_t report_count_or_subtype =
byte0 & FieldBitmask<uint8_t>(kRtcpReportCountFieldNumBits);
const uint8_t byte1 = ConsumeField<uint8_t>(&buffer);
if (!IsRtcpPacketType(byte1)) {
return absl::nullopt;
}
// Optionally set |header.with.report_count| or |header.with.subtype|,
// depending on the packet type.
RtcpCommonHeader header;
header.packet_type = static_cast<RtcpPacketType>(byte1);
switch (header.packet_type) {
case RtcpPacketType::kSenderReport:
case RtcpPacketType::kReceiverReport:
header.with.report_count = report_count_or_subtype;
break;
case RtcpPacketType::kApplicationDefined:
case RtcpPacketType::kPayloadSpecific:
switch (static_cast<RtcpSubtype>(report_count_or_subtype)) {
case RtcpSubtype::kPictureLossIndicator:
case RtcpSubtype::kReceiverLog:
case RtcpSubtype::kFeedback:
header.with.subtype =
static_cast<RtcpSubtype>(report_count_or_subtype);
break;
default: // Unknown subtype.
header.with.subtype = RtcpSubtype::kNull;
break;
}
break;
default:
// Neither |header.with.report_count| nor |header.with.subtype| are used.
break;
}
header.payload_size =
static_cast<int>(ConsumeField<uint16_t>(&buffer)) * sizeof(uint32_t);
return header;
}
RtcpReportBlock::RtcpReportBlock() = default;
RtcpReportBlock::~RtcpReportBlock() = default;
void RtcpReportBlock::AppendFields(absl::Span<uint8_t>* buffer) const {
OSP_CHECK_GE(buffer->size(), kRtcpReportBlockSize);
AppendField<uint32_t>(ssrc, buffer);
OSP_DCHECK_GE(packet_fraction_lost_numerator,
std::numeric_limits<uint8_t>::min());
OSP_DCHECK_LE(packet_fraction_lost_numerator,
std::numeric_limits<uint8_t>::max());
OSP_DCHECK_GE(cumulative_packets_lost, 0);
OSP_DCHECK_LE(cumulative_packets_lost,
FieldBitmask<int>(kRtcpCumulativePacketsFieldNumBits));
AppendField<uint32_t>(
(static_cast<int>(packet_fraction_lost_numerator)
<< kRtcpCumulativePacketsFieldNumBits) |
(static_cast<int>(cumulative_packets_lost) &
FieldBitmask<uint32_t>(kRtcpCumulativePacketsFieldNumBits)),
buffer);
AppendField<uint32_t>(extended_high_sequence_number, buffer);
const int64_t jitter_ticks = jitter / RtpTimeDelta::FromTicks(1);
OSP_DCHECK_GE(jitter_ticks, 0);
OSP_DCHECK_LE(jitter_ticks, int64_t{std::numeric_limits<uint32_t>::max()});
AppendField<uint32_t>(jitter_ticks, buffer);
AppendField<uint32_t>(last_status_report_id, buffer);
const int64_t delay_ticks = delay_since_last_report.count();
OSP_DCHECK_GE(delay_ticks, 0);
OSP_DCHECK_LE(delay_ticks, int64_t{std::numeric_limits<uint32_t>::max()});
AppendField<uint32_t>(delay_ticks, buffer);
}
void RtcpReportBlock::SetPacketFractionLostNumerator(
int64_t num_apparently_sent,
int64_t num_received) {
if (num_apparently_sent <= 0) {
packet_fraction_lost_numerator = 0;
return;
}
// The following computes the fraction of packets lost as "one minus
// |num_received| divided by |num_apparently_sent|" and scales by 256 (the
// kPacketFractionLostDenominator). It's valid for |num_received| to be
// greater than |num_apparently_sent| in some cases (e.g., if duplicate
// packets were received from the network).
const int64_t numerator =
((num_apparently_sent - num_received) * kPacketFractionLostDenominator) /
num_apparently_sent;
// Since the value must be in the range [0,255], just do a saturate_cast
// to the uint8_t type to clamp.
packet_fraction_lost_numerator = saturate_cast<uint8_t>(numerator);
}
void RtcpReportBlock::SetCumulativePacketsLost(int64_t num_apparently_sent,
int64_t num_received) {
const int64_t num_lost = num_apparently_sent - num_received;
// Clamp to valid range supported by the wire format (and RTP spec).
//
// Note that |num_lost| can be negative if duplicate packets were received.
// The RFC spec (https://tools.ietf.org/html/rfc3550#section-6.4.1) states
// this should result in a clamped, "zero loss" value.
cumulative_packets_lost = static_cast<int>(
std::min(std::max<int64_t>(num_lost, 0),
FieldBitmask<int64_t>(kRtcpCumulativePacketsFieldNumBits)));
}
void RtcpReportBlock::SetDelaySinceLastReport(
Clock::duration local_clock_delay) {
// Clamp to valid range supported by the wire format (and RTP spec). The
// bounds checking is done in terms of Clock::duration, since doing the checks
// after the duration_cast may allow overflow to occur in the duration_cast
// math (well, only for unusually large inputs).
constexpr Delay kMaxValidReportedDelay{std::numeric_limits<uint32_t>::max()};
constexpr auto kMaxValidLocalClockDelay =
Clock::to_duration(kMaxValidReportedDelay);
if (local_clock_delay > kMaxValidLocalClockDelay) {
delay_since_last_report = kMaxValidReportedDelay;
return;
}
if (local_clock_delay <= Clock::duration::zero()) {
delay_since_last_report = Delay::zero();
return;
}
// If this point is reached, then the |local_clock_delay| is representable as
// a Delay within the valid range.
delay_since_last_report =
std::chrono::duration_cast<Delay>(local_clock_delay);
}
// static
absl::optional<RtcpReportBlock> RtcpReportBlock::ParseOne(
absl::Span<const uint8_t> buffer,
int report_count,
Ssrc ssrc) {
if (static_cast<int>(buffer.size()) < (kRtcpReportBlockSize * report_count)) {
return absl::nullopt;
}
absl::optional<RtcpReportBlock> result;
for (int block = 0; block < report_count; ++block) {
if (ConsumeField<uint32_t>(&buffer) != ssrc) {
// Skip-over report block meant for some other recipient.
buffer.remove_prefix(kRtcpReportBlockSize - sizeof(uint32_t));
continue;
}
RtcpReportBlock& report_block = result.emplace();
report_block.ssrc = ssrc;
const auto second_word = ConsumeField<uint32_t>(&buffer);
report_block.packet_fraction_lost_numerator =
second_word >> kRtcpCumulativePacketsFieldNumBits;
report_block.cumulative_packets_lost =
second_word &
FieldBitmask<uint32_t>(kRtcpCumulativePacketsFieldNumBits);
report_block.extended_high_sequence_number =
ConsumeField<uint32_t>(&buffer);
report_block.jitter =
RtpTimeDelta::FromTicks(ConsumeField<uint32_t>(&buffer));
report_block.last_status_report_id = ConsumeField<uint32_t>(&buffer);
report_block.delay_since_last_report =
RtcpReportBlock::Delay(ConsumeField<uint32_t>(&buffer));
}
return result;
}
RtcpSenderReport::RtcpSenderReport() = default;
RtcpSenderReport::~RtcpSenderReport() = default;
} // namespace cast
} // namespace openscreen