| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656 |
- /**
- * Copyright (c) 2020 Staz Modrzynski
- * Copyright (c) 2020 Paul-Louis Ageneau
- * Copyright (c) 2020 Filip Klembara (in2core)
- *
- * This library is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License as published by the Free Software Foundation; either
- * version 2.1 of the License, or (at your option) any later version.
- *
- * This library is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public
- * License along with this library; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
- */
- #include "rtp.hpp"
- #include "impl/internals.hpp"
- #include <cmath>
- #include <cstring>
- #ifdef _WIN32
- #include <winsock2.h>
- #else
- #include <arpa/inet.h>
- #endif
- #ifndef htonll
- #define htonll(x) \
- ((uint64_t)(((uint64_t)htonl((uint32_t)(x))) << 32) | (uint64_t)htonl((uint32_t)((x) >> 32)))
- #endif
- #ifndef ntohll
- #define ntohll(x) htonll(x)
- #endif
- namespace rtc {
- uint8_t RtpHeader::version() const { return _first >> 6; }
- bool RtpHeader::padding() const { return (_first >> 5) & 0x01; }
- bool RtpHeader::extension() const { return (_first >> 4) & 0x01; }
- uint8_t RtpHeader::csrcCount() const { return _first & 0x0F; }
- uint8_t RtpHeader::marker() const { return _payloadType & 0b10000000; }
- uint8_t RtpHeader::payloadType() const { return _payloadType & 0b01111111; }
- uint16_t RtpHeader::seqNumber() const { return ntohs(_seqNumber); }
- uint32_t RtpHeader::timestamp() const { return ntohl(_timestamp); }
- uint32_t RtpHeader::ssrc() const { return ntohl(_ssrc); }
- size_t RtpHeader::getSize() const {
- return reinterpret_cast<const char *>(&_csrc) - reinterpret_cast<const char *>(this) +
- sizeof(SSRC) * csrcCount();
- }
- size_t RtpHeader::getExtensionHeaderSize() const {
- auto header = reinterpret_cast<const RtpExtensionHeader *>(getExtensionHeader());
- if (header) {
- return header->getSize() + sizeof(RtpExtensionHeader);
- }
- return 0;
- }
- const RtpExtensionHeader *RtpHeader::getExtensionHeader() const {
- if (extension()) {
- auto header = reinterpret_cast<const char *>(&_csrc) + sizeof(SSRC) * csrcCount();
- return reinterpret_cast<const RtpExtensionHeader *>(header);
- }
- return nullptr;
- }
- RtpExtensionHeader *RtpHeader::getExtensionHeader() {
- if (extension()) {
- auto header = reinterpret_cast<char *>(&_csrc) + sizeof(SSRC) * csrcCount();
- return reinterpret_cast<RtpExtensionHeader *>(header);
- }
- return nullptr;
- }
- const char *RtpHeader::getBody() const {
- return reinterpret_cast<const char *>(&_csrc) + sizeof(SSRC) * csrcCount() +
- getExtensionHeaderSize();
- }
- char *RtpHeader::getBody() {
- return reinterpret_cast<char *>(&_csrc) + sizeof(SSRC) * csrcCount() + getExtensionHeaderSize();
- }
- void RtpHeader::preparePacket() { _first |= (1 << 7); }
- void RtpHeader::setSeqNumber(uint16_t newSeqNo) { _seqNumber = htons(newSeqNo); }
- void RtpHeader::setPayloadType(uint8_t newPayloadType) {
- _payloadType = (_payloadType & 0b10000000u) | (0b01111111u & newPayloadType);
- }
- void RtpHeader::setSsrc(uint32_t in_ssrc) { _ssrc = htonl(in_ssrc); }
- void RtpHeader::setMarker(bool marker) { _payloadType = (_payloadType & 0x7F) | (marker << 7); };
- void RtpHeader::setTimestamp(uint32_t i) { _timestamp = htonl(i); }
- void RtpHeader::setExtension(bool extension) { _first = (_first & ~0x10) | ((extension & 1) << 4); }
- void RtpHeader::log() const {
- PLOG_VERBOSE << "RtpHeader V: " << (int)version() << " P: " << (padding() ? "P" : " ")
- << " X: " << (extension() ? "X" : " ") << " CC: " << (int)csrcCount()
- << " M: " << (marker() ? "M" : " ") << " PT: " << (int)payloadType()
- << " SEQNO: " << seqNumber() << " TS: " << timestamp();
- }
- uint16_t RtpExtensionHeader::profileSpecificId() const { return ntohs(_profileSpecificId); }
- uint16_t RtpExtensionHeader::headerLength() const { return ntohs(_headerLength); }
- size_t RtpExtensionHeader::getSize() const { return headerLength() * 4; }
- const char *RtpExtensionHeader::getBody() const {
- return reinterpret_cast<const char *>((&_headerLength) + 1);
- }
- char *RtpExtensionHeader::getBody() { return reinterpret_cast<char *>((&_headerLength) + 1); }
- void RtpExtensionHeader::setProfileSpecificId(uint16_t profileSpecificId) {
- _profileSpecificId = htons(profileSpecificId);
- }
- void RtpExtensionHeader::setHeaderLength(uint16_t headerLength) {
- _headerLength = htons(headerLength);
- }
- void RtpExtensionHeader::clearBody() { std::memset(getBody(), 0, getSize()); }
- void RtpExtensionHeader::writeCurrentVideoOrientation(size_t offset, uint8_t id, uint8_t value) {
- if ((id == 0) || (id > 14) || ((offset + 2) > getSize()))
- return;
- auto buf = getBody() + offset;
- buf[0] = id << 4;
- buf[1] = value;
- }
- SSRC RtcpReportBlock::getSSRC() const { return ntohl(_ssrc); }
- void RtcpReportBlock::preparePacket(SSRC in_ssrc, [[maybe_unused]] unsigned int packetsLost,
- [[maybe_unused]] unsigned int totalPackets,
- uint16_t highestSeqNo, uint16_t seqNoCycles, uint32_t jitter,
- uint64_t lastSR_NTP, uint64_t lastSR_DELAY) {
- setSeqNo(highestSeqNo, seqNoCycles);
- setJitter(jitter);
- setSSRC(in_ssrc);
- // Middle 32 bits of NTP Timestamp
- // _lastReport = lastSR_NTP >> 16u;
- setNTPOfSR(uint64_t(lastSR_NTP));
- setDelaySinceSR(uint32_t(lastSR_DELAY));
- // The delay, expressed in units of 1/65536 seconds
- // _delaySinceLastReport = lastSR_DELAY;
- }
- void RtcpReportBlock::setSSRC(SSRC in_ssrc) { _ssrc = htonl(in_ssrc); }
- void RtcpReportBlock::setPacketsLost([[maybe_unused]] unsigned int packetsLost,
- [[maybe_unused]] unsigned int totalPackets) {
- // TODO Implement loss percentages.
- _fractionLostAndPacketsLost = 0;
- }
- unsigned int RtcpReportBlock::getLossPercentage() const {
- // TODO Implement loss percentages.
- return 0;
- }
- unsigned int RtcpReportBlock::getPacketLostCount() const {
- // TODO Implement total packets lost.
- return 0;
- }
- uint16_t RtcpReportBlock::seqNoCycles() const { return ntohs(_seqNoCycles); }
- uint16_t RtcpReportBlock::highestSeqNo() const { return ntohs(_highestSeqNo); }
- uint32_t RtcpReportBlock::jitter() const { return ntohl(_jitter); }
- uint32_t RtcpReportBlock::delaySinceSR() const { return ntohl(_delaySinceLastReport); }
- void RtcpReportBlock::setSeqNo(uint16_t highestSeqNo, uint16_t seqNoCycles) {
- _highestSeqNo = htons(highestSeqNo);
- _seqNoCycles = htons(seqNoCycles);
- }
- void RtcpReportBlock::setJitter(uint32_t jitter) { _jitter = htonl(jitter); }
- void RtcpReportBlock::setNTPOfSR(uint64_t ntp) { _lastReport = htonll(ntp >> 16u); }
- uint32_t RtcpReportBlock::getNTPOfSR() const { return ntohl(_lastReport) << 16u; }
- void RtcpReportBlock::setDelaySinceSR(uint32_t sr) {
- // The delay, expressed in units of 1/65536 seconds
- _delaySinceLastReport = htonl(sr);
- }
- void RtcpReportBlock::log() const {
- PLOG_VERBOSE << "RTCP report block: "
- << "ssrc="
- << ntohl(_ssrc)
- // TODO: Implement these reports
- // << ", fractionLost=" << fractionLost
- // << ", packetsLost=" << packetsLost
- << ", highestSeqNo=" << highestSeqNo() << ", seqNoCycles=" << seqNoCycles()
- << ", jitter=" << jitter() << ", lastSR=" << getNTPOfSR()
- << ", lastSRDelay=" << delaySinceSR();
- }
- uint8_t RtcpHeader::version() const { return _first >> 6; }
- bool RtcpHeader::padding() const { return (_first >> 5) & 0x01; }
- uint8_t RtcpHeader::reportCount() const { return _first & 0x1F; }
- uint8_t RtcpHeader::payloadType() const { return _payloadType; }
- uint16_t RtcpHeader::length() const { return ntohs(_length); }
- size_t RtcpHeader::lengthInBytes() const { return (1 + length()) * 4; }
- void RtcpHeader::setPayloadType(uint8_t type) { _payloadType = type; }
- void RtcpHeader::setReportCount(uint8_t count) {
- _first = (_first & 0b11100000u) | (count & 0b00011111u);
- }
- void RtcpHeader::setLength(uint16_t length) { _length = htons(length); }
- void RtcpHeader::prepareHeader(uint8_t payloadType, uint8_t reportCount, uint16_t length) {
- _first = 0b10000000; // version 2, no padding
- setReportCount(reportCount);
- setPayloadType(payloadType);
- setLength(length);
- }
- void RtcpHeader::log() const {
- PLOG_VERBOSE << "RTCP header: "
- << "version=" << unsigned(version()) << ", padding=" << padding()
- << ", reportCount=" << unsigned(reportCount())
- << ", payloadType=" << unsigned(payloadType()) << ", length=" << length();
- }
- SSRC RtcpFbHeader::packetSenderSSRC() const { return ntohl(_packetSender); }
- SSRC RtcpFbHeader::mediaSourceSSRC() const { return ntohl(_mediaSource); }
- void RtcpFbHeader::setPacketSenderSSRC(SSRC ssrc) { _packetSender = htonl(ssrc); }
- void RtcpFbHeader::setMediaSourceSSRC(SSRC ssrc) { _mediaSource = htonl(ssrc); }
- void RtcpFbHeader::log() const {
- header.log();
- PLOG_VERBOSE << "FB: "
- << " packet sender: " << packetSenderSSRC()
- << " media source: " << mediaSourceSSRC();
- }
- unsigned int RtcpSr::Size(unsigned int reportCount) {
- return sizeof(RtcpHeader) + 24 + reportCount * sizeof(RtcpReportBlock);
- }
- void RtcpSr::preparePacket(SSRC senderSSRC, uint8_t reportCount) {
- unsigned int length = ((sizeof(header) + 24 + reportCount * sizeof(RtcpReportBlock)) / 4) - 1;
- header.prepareHeader(200, reportCount, uint16_t(length));
- this->_senderSSRC = htonl(senderSSRC);
- }
- const RtcpReportBlock *RtcpSr::getReportBlock(int num) const { return &_reportBlocks + num; }
- RtcpReportBlock *RtcpSr::getReportBlock(int num) { return &_reportBlocks + num; }
- size_t RtcpSr::getSize() const {
- // "length" in packet is one less than the number of 32 bit words in the packet.
- return sizeof(uint32_t) * (1 + size_t(header.length()));
- }
- uint64_t RtcpSr::ntpTimestamp() const { return ntohll(_ntpTimestamp); }
- uint32_t RtcpSr::rtpTimestamp() const { return ntohl(_rtpTimestamp); }
- uint32_t RtcpSr::packetCount() const { return ntohl(_packetCount); }
- uint32_t RtcpSr::octetCount() const { return ntohl(_octetCount); }
- uint32_t RtcpSr::senderSSRC() const { return ntohl(_senderSSRC); }
- void RtcpSr::setNtpTimestamp(uint64_t ts) { _ntpTimestamp = htonll(ts); }
- void RtcpSr::setRtpTimestamp(uint32_t ts) { _rtpTimestamp = htonl(ts); }
- void RtcpSr::setOctetCount(uint32_t ts) { _octetCount = htonl(ts); }
- void RtcpSr::setPacketCount(uint32_t ts) { _packetCount = htonl(ts); }
- void RtcpSr::log() const {
- header.log();
- PLOG_VERBOSE << "RTCP SR: "
- << " SSRC=" << senderSSRC() << ", NTP_TS=" << ntpTimestamp()
- << ", RtpTS=" << rtpTimestamp() << ", packetCount=" << packetCount()
- << ", octetCount=" << octetCount();
- for (unsigned i = 0; i < unsigned(header.reportCount()); i++) {
- getReportBlock(i)->log();
- }
- }
- unsigned int RtcpSdesItem::Size(uint8_t textLength) { return textLength + 2; }
- std::string RtcpSdesItem::text() const { return std::string(_text, _length); }
- void RtcpSdesItem::setText(std::string text) {
- if (text.size() > 0xFF)
- throw std::invalid_argument("text is too long");
- _length = uint8_t(text.size());
- memcpy(_text, text.data(), text.size());
- }
- uint8_t RtcpSdesItem::length() const { return _length; }
- unsigned int RtcpSdesChunk::Size(const std::vector<uint8_t> textLengths) {
- unsigned int itemsSize = 0;
- for (auto length : textLengths) {
- itemsSize += RtcpSdesItem::Size(length);
- }
- auto nullTerminatedItemsSize = itemsSize + 1;
- auto words = uint8_t(std::ceil(double(nullTerminatedItemsSize) / 4)) + 1;
- return words * 4;
- }
- SSRC RtcpSdesChunk::ssrc() const { return ntohl(_ssrc); }
- void RtcpSdesChunk::setSSRC(SSRC ssrc) { _ssrc = htonl(ssrc); }
- const RtcpSdesItem *RtcpSdesChunk::getItem(int num) const {
- auto base = &_items;
- while (num-- > 0) {
- auto itemSize = RtcpSdesItem::Size(base->length());
- base = reinterpret_cast<const RtcpSdesItem *>(reinterpret_cast<const uint8_t *>(base) +
- itemSize);
- }
- return reinterpret_cast<const RtcpSdesItem *>(base);
- }
- RtcpSdesItem *RtcpSdesChunk::getItem(int num) {
- auto base = &_items;
- while (num-- > 0) {
- auto itemSize = RtcpSdesItem::Size(base->length());
- base = reinterpret_cast<RtcpSdesItem *>(reinterpret_cast<uint8_t *>(base) + itemSize);
- }
- return reinterpret_cast<RtcpSdesItem *>(base);
- }
- unsigned int RtcpSdesChunk::getSize() const {
- std::vector<uint8_t> textLengths{};
- unsigned int i = 0;
- auto item = getItem(i);
- while (item->type != 0) {
- textLengths.push_back(item->length());
- item = getItem(++i);
- }
- return Size(textLengths);
- }
- long RtcpSdesChunk::safelyCountChunkSize(size_t maxChunkSize) const {
- if (maxChunkSize < RtcpSdesChunk::Size({})) {
- // chunk is truncated
- return -1;
- }
- size_t size = sizeof(SSRC);
- unsigned int i = 0;
- // We can always access first 4 bytes of first item (in case of no items there will be 4
- // null bytes)
- auto item = getItem(i);
- std::vector<uint8_t> textsLength{};
- while (item->type != 0) {
- if (size + RtcpSdesItem::Size(0) > maxChunkSize) {
- // item is too short
- return -1;
- }
- auto itemLength = item->length();
- if (size + RtcpSdesItem::Size(itemLength) >= maxChunkSize) {
- // item is too large (it can't be equal to chunk size because after item there
- // must be 1-4 null bytes as padding)
- return -1;
- }
- textsLength.push_back(itemLength);
- // safely to access next item
- item = getItem(++i);
- }
- auto realSize = RtcpSdesChunk::Size(textsLength);
- if (realSize > maxChunkSize) {
- // Chunk is too large
- return -1;
- }
- return realSize;
- }
- unsigned int RtcpSdes::Size(const std::vector<std::vector<uint8_t>> lengths) {
- unsigned int chunks_size = 0;
- for (auto length : lengths)
- chunks_size += RtcpSdesChunk::Size(length);
- return 4 + chunks_size;
- }
- bool RtcpSdes::isValid() const {
- auto chunksSize = header.lengthInBytes() - sizeof(header);
- if (chunksSize == 0) {
- return true;
- }
- // there is at least one chunk
- unsigned int i = 0;
- unsigned int size = 0;
- while (size < chunksSize) {
- if (chunksSize < size + RtcpSdesChunk::Size({})) {
- // chunk is truncated
- return false;
- }
- auto chunk = getChunk(i++);
- auto chunkSize = chunk->safelyCountChunkSize(chunksSize - size);
- if (chunkSize < 0) {
- // chunk is invalid
- return false;
- }
- size += chunkSize;
- }
- return size == chunksSize;
- }
- unsigned int RtcpSdes::chunksCount() const {
- if (!isValid()) {
- return 0;
- }
- uint16_t chunksSize = 4 * (header.length() + 1) - sizeof(header);
- unsigned int size = 0;
- unsigned int i = 0;
- while (size < chunksSize) {
- size += getChunk(i++)->getSize();
- }
- return i;
- }
- const RtcpSdesChunk *RtcpSdes::getChunk(int num) const {
- auto base = &_chunks;
- while (num-- > 0) {
- auto chunkSize = base->getSize();
- base = reinterpret_cast<const RtcpSdesChunk *>(reinterpret_cast<const uint8_t *>(base) +
- chunkSize);
- }
- return reinterpret_cast<const RtcpSdesChunk *>(base);
- }
- RtcpSdesChunk *RtcpSdes::getChunk(int num) {
- auto base = &_chunks;
- while (num-- > 0) {
- auto chunkSize = base->getSize();
- base = reinterpret_cast<RtcpSdesChunk *>(reinterpret_cast<uint8_t *>(base) + chunkSize);
- }
- return reinterpret_cast<RtcpSdesChunk *>(base);
- }
- void RtcpSdes::preparePacket(uint8_t chunkCount) {
- unsigned int chunkSize = 0;
- for (uint8_t i = 0; i < chunkCount; i++) {
- auto chunk = getChunk(i);
- chunkSize += chunk->getSize();
- }
- uint16_t length = uint16_t((sizeof(header) + chunkSize) / 4 - 1);
- header.prepareHeader(202, chunkCount, length);
- }
- const RtcpReportBlock *RtcpRr::getReportBlock(int num) const { return &_reportBlocks + num; }
- RtcpReportBlock *RtcpRr::getReportBlock(int num) { return &_reportBlocks + num; }
- size_t RtcpRr::SizeWithReportBlocks(uint8_t reportCount) {
- return sizeof(header) + 4 + size_t(reportCount) * sizeof(RtcpReportBlock);
- }
- SSRC RtcpRr::senderSSRC() const { return ntohl(_senderSSRC); }
- bool RtcpRr::isSenderReport() { return header.payloadType() == 200; }
- bool RtcpRr::isReceiverReport() { return header.payloadType() == 201; }
- size_t RtcpRr::getSize() const {
- // "length" in packet is one less than the number of 32 bit words in the packet.
- return sizeof(uint32_t) * (1 + size_t(header.length()));
- }
- void RtcpRr::preparePacket(SSRC senderSSRC, uint8_t reportCount) {
- // "length" in packet is one less than the number of 32 bit words in the packet.
- size_t length = (SizeWithReportBlocks(reportCount) / 4) - 1;
- header.prepareHeader(201, reportCount, uint16_t(length));
- this->_senderSSRC = htonl(senderSSRC);
- }
- void RtcpRr::setSenderSSRC(SSRC ssrc) { this->_senderSSRC = htonl(ssrc); }
- void RtcpRr::log() const {
- header.log();
- PLOG_VERBOSE << "RTCP RR: "
- << " SSRC=" << ntohl(_senderSSRC);
- for (unsigned i = 0; i < unsigned(header.reportCount()); i++) {
- getReportBlock(i)->log();
- }
- }
- size_t RtcpRemb::SizeWithSSRCs(int count) { return sizeof(RtcpRemb) + (count - 1) * sizeof(SSRC); }
- unsigned int RtcpRemb::getSize() const {
- // "length" in packet is one less than the number of 32 bit words in the packet.
- return sizeof(uint32_t) * (1 + header.header.length());
- }
- void RtcpRemb::preparePacket(SSRC senderSSRC, unsigned int numSSRC, unsigned int in_bitrate) {
- // Report Count becomes the format here.
- header.header.prepareHeader(206, 15, 0);
- // Always zero.
- header.setMediaSourceSSRC(0);
- header.setPacketSenderSSRC(senderSSRC);
- _id[0] = 'R';
- _id[1] = 'E';
- _id[2] = 'M';
- _id[3] = 'B';
- setBitrate(numSSRC, in_bitrate);
- }
- void RtcpRemb::setBitrate(unsigned int numSSRC, unsigned int in_bitrate) {
- unsigned int exp = 0;
- while (in_bitrate > pow(2, 18) - 1) {
- exp++;
- in_bitrate /= 2;
- }
- // "length" in packet is one less than the number of 32 bit words in the packet.
- header.header.setLength(uint16_t((offsetof(RtcpRemb, _ssrc) / sizeof(uint32_t)) - 1 + numSSRC));
- _bitrate = htonl((numSSRC << (32u - 8u)) | (exp << (32u - 8u - 6u)) | in_bitrate);
- }
- void RtcpRemb::setSsrc(int iterator, SSRC newSssrc) { _ssrc[iterator] = htonl(newSssrc); }
- unsigned int RtcpPli::Size() { return sizeof(RtcpFbHeader); }
- void RtcpPli::preparePacket(SSRC messageSSRC) {
- header.header.prepareHeader(206, 1, 2);
- header.setPacketSenderSSRC(messageSSRC);
- header.setMediaSourceSSRC(messageSSRC);
- }
- void RtcpPli::log() const { header.log(); }
- unsigned int RtcpFir::Size() { return sizeof(RtcpFbHeader) + sizeof(RtcpFirPart); }
- void RtcpFir::preparePacket(SSRC messageSSRC, uint8_t seqNo) {
- header.header.prepareHeader(206, 4, 2 + 2 * 1);
- header.setPacketSenderSSRC(messageSSRC);
- header.setMediaSourceSSRC(messageSSRC);
- parts[0].ssrc = htonl(messageSSRC);
- parts[0].seqNo = seqNo;
- }
- void RtcpFir::log() const { header.log(); }
- uint16_t RtcpNackPart::pid() { return ntohs(_pid); }
- uint16_t RtcpNackPart::blp() { return ntohs(_blp); }
- void RtcpNackPart::setPid(uint16_t pid) { _pid = htons(pid); }
- void RtcpNackPart::setBlp(uint16_t blp) { _blp = htons(blp); }
- std::vector<uint16_t> RtcpNackPart::getSequenceNumbers() {
- std::vector<uint16_t> result{};
- result.reserve(17);
- uint16_t p = pid();
- result.push_back(p);
- uint16_t bitmask = blp();
- uint16_t i = p + 1;
- while (bitmask > 0) {
- if (bitmask & 0x1) {
- result.push_back(i);
- }
- i += 1;
- bitmask >>= 1;
- }
- return result;
- }
- unsigned int RtcpNack::Size(unsigned int discreteSeqNoCount) {
- return offsetof(RtcpNack, parts) + sizeof(RtcpNackPart) * discreteSeqNoCount;
- }
- unsigned int RtcpNack::getSeqNoCount() { return header.header.length() - 2; }
- void RtcpNack::preparePacket(SSRC ssrc, unsigned int discreteSeqNoCount) {
- header.header.prepareHeader(205, 1, 2 + uint16_t(discreteSeqNoCount));
- header.setMediaSourceSSRC(ssrc);
- header.setPacketSenderSSRC(ssrc);
- }
- bool RtcpNack::addMissingPacket(unsigned int *fciCount, uint16_t *fciPID, uint16_t missingPacket) {
- if (*fciCount == 0 || missingPacket < *fciPID || missingPacket > (*fciPID + 16)) {
- parts[*fciCount].setPid(missingPacket);
- parts[*fciCount].setBlp(0);
- *fciPID = missingPacket;
- (*fciCount)++;
- return true;
- } else {
- // TODO SPEED!
- uint16_t blp = parts[(*fciCount) - 1].blp();
- uint16_t newBit = uint16_t(1u << (missingPacket - (1 + *fciPID)));
- parts[(*fciCount) - 1].setBlp(blp | newBit);
- return false;
- }
- }
- uint16_t RtpRtx::getOriginalSeqNo() const { return ntohs(*(uint16_t *)(header.getBody())); }
- const char *RtpRtx::getBody() const { return header.getBody() + sizeof(uint16_t); }
- char *RtpRtx::getBody() { return header.getBody() + sizeof(uint16_t); }
- size_t RtpRtx::getBodySize(size_t totalSize) const {
- return totalSize - (getBody() - reinterpret_cast<const char *>(this));
- }
- size_t RtpRtx::getSize() const { return header.getSize() + sizeof(uint16_t); }
- size_t RtpRtx::normalizePacket(size_t totalSize, SSRC originalSSRC, uint8_t originalPayloadType) {
- header.setSeqNumber(getOriginalSeqNo());
- header.setSsrc(originalSSRC);
- header.setPayloadType(originalPayloadType);
- // TODO, the -12 is the size of the header (which is variable!)
- memmove(header.getBody(), getBody(), totalSize - getSize());
- return totalSize - 2;
- }
- size_t RtpRtx::copyTo(RtpHeader *dest, size_t totalSize, uint8_t originalPayloadType) {
- memmove((char *)dest, (char *)this, header.getSize());
- dest->setSeqNumber(getOriginalSeqNo());
- dest->setPayloadType(originalPayloadType);
- memmove(dest->getBody(), getBody(), getBodySize(totalSize));
- return totalSize;
- }
- }; // namespace rtc
|