/* * Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved. * * This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit). * * Use of this source code is governed by MIT license that can be found in the * LICENSE file in the root of the source tree. All contributing project authors * may be found in the AUTHORS file in the root of the source tree. */ #include #include #include "Rtcp.h" #include "Util/logger.h" #include "RtcpFCI.h" namespace mediakit { const char *rtcpTypeToStr(RtcpType type){ switch (type){ #define SWITCH_CASE(key, value) case RtcpType::key : return #value "(" #key ")"; RTCP_PT_MAP(SWITCH_CASE) #undef SWITCH_CASE default: return "unknown rtcp pt"; } } const char *sdesTypeToStr(SdesType type){ switch (type){ #define SWITCH_CASE(key, value) case SdesType::key : return #value "(" #key ")"; SDES_TYPE_MAP(SWITCH_CASE) #undef SWITCH_CASE default: return "unknown source description type"; } } const char *psfbTypeToStr(PSFBType type) { switch (type){ #define SWITCH_CASE(key, value) case PSFBType::key : return #value "(" #key ")"; PSFB_TYPE_MAP(SWITCH_CASE) #undef SWITCH_CASE default: return "unknown payload-specific fb message fmt type"; } } const char *rtpfbTypeToStr(RTPFBType type) { switch (type){ #define SWITCH_CASE(key, value) case RTPFBType::key : return #value "(" #key ")"; RTPFB_TYPE_MAP(SWITCH_CASE) #undef SWITCH_CASE default: return "unknown transport layer feedback messages fmt type"; } } static size_t alignSize(size_t bytes) { return (size_t)((bytes + 3) >> 2 ) << 2; } static void setupHeader(RtcpHeader *rtcp, RtcpType type, size_t report_count, size_t total_bytes) { rtcp->version = 2; rtcp->padding = 0; if (report_count > 0x1F) { throw std::invalid_argument(StrPrinter << "rtcp report_count最大赋值为31,当前为:" << report_count); } //items总个数 rtcp->report_count = report_count; rtcp->pt = (uint8_t) type; rtcp->setSize(total_bytes); } static void setupPadding(RtcpHeader *rtcp, size_t padding_size) { if (padding_size) { rtcp->padding = 1; ((uint8_t *) rtcp)[rtcp->getSize() - 1] = padding_size & 0xFF; } else { rtcp->padding = 0; } } ///////////////////////////////////////////////////////////////////////////// string RtcpHeader::dumpHeader() const{ _StrPrinter printer; printer << "version:" << version << "\r\n"; if (padding) { printer << "padding:" << padding << " " << getPaddingSize() << "\r\n"; } else { printer << "padding:" << padding << "\r\n"; } switch ((RtcpType)pt) { case RtcpType::RTCP_RTPFB : { printer << "report_count:" << rtpfbTypeToStr((RTPFBType) report_count) << "\r\n"; break; } case RtcpType::RTCP_PSFB : { printer << "report_count:" << psfbTypeToStr((PSFBType) report_count) << "\r\n"; break; } default : { printer << "report_count:" << report_count << "\r\n"; break; } } printer << "pt:" << rtcpTypeToStr((RtcpType)pt) << "\r\n"; printer << "size:" << getSize() << "\r\n"; printer << "--------\r\n"; return std::move(printer); } string RtcpHeader::dumpString() const { switch ((RtcpType)pt) { case RtcpType::RTCP_SR: { RtcpSR *rtcp = (RtcpSR *)this; return rtcp->dumpString(); } case RtcpType::RTCP_RR: { RtcpRR *rtcp = (RtcpRR *)this; return rtcp->dumpString(); } case RtcpType::RTCP_SDES: { RtcpSdes *rtcp = (RtcpSdes *)this; return rtcp->dumpString(); } case RtcpType::RTCP_RTPFB: case RtcpType::RTCP_PSFB: { RtcpFB *rtcp = (RtcpFB *)this; return rtcp->dumpString(); } case RtcpType::RTCP_BYE: { RtcpBye *rtcp = (RtcpBye *)this; return rtcp->dumpString(); } default: return StrPrinter << dumpHeader() << hexdump((char *)this + sizeof(*this), getSize() - sizeof(*this)); } } size_t RtcpHeader::getSize() const { //加上rtcp头长度 return (1 + ntohs(length)) << 2; } size_t RtcpHeader::getPaddingSize() const{ if (!padding) { return 0; } return ((uint8_t *) this)[getSize() - 1]; } void RtcpHeader::setSize(size_t size) { //不包含rtcp头的长度 length = htons((uint16_t)((size >> 2) - 1)); } void RtcpHeader::net2Host(size_t len){ switch ((RtcpType)pt) { case RtcpType::RTCP_SR: { RtcpSR *sr = (RtcpSR *)this; sr->net2Host(len); break; } case RtcpType::RTCP_RR: { RtcpRR *rr = (RtcpRR *)this; rr->net2Host(len); break; } case RtcpType::RTCP_SDES: { RtcpSdes *sdes = (RtcpSdes *)this; sdes->net2Host(len); break; } case RtcpType::RTCP_RTPFB: case RtcpType::RTCP_PSFB: { RtcpFB *fb = (RtcpFB *)this; fb->net2Host(len); break; } case RtcpType::RTCP_BYE: { RtcpBye *bye = (RtcpBye *)this; bye->net2Host(len); break; } default: throw std::runtime_error(StrPrinter << "未处理的rtcp包:" << rtcpTypeToStr((RtcpType) this->pt)); } } vector RtcpHeader::loadFromBytes(char *data, size_t len){ vector ret; ssize_t remain = len; char *ptr = data; while (remain > (ssize_t) sizeof(RtcpHeader)) { RtcpHeader *rtcp = (RtcpHeader *) ptr; auto rtcp_len = rtcp->getSize(); if (remain < (ssize_t)rtcp_len) { WarnL << "非法的rtcp包,声明的长度超过实际数据长度"; break; } try { rtcp->net2Host(rtcp_len); ret.emplace_back(rtcp); } catch (std::exception &ex) { //不能处理的rtcp包,或者无法解析的rtcp包,忽略掉 WarnL << ex.what() << ",长度为:" << rtcp_len; } ptr += rtcp_len; remain -= rtcp_len; } return ret; } class BufferRtcp : public Buffer { public: BufferRtcp(std::shared_ptr rtcp) { _rtcp = std::move(rtcp); } ~BufferRtcp() override {} char *data() const override { return (char *) _rtcp.get(); } size_t size() const override { return _rtcp->getSize(); } private: std::shared_ptr _rtcp; }; Buffer::Ptr RtcpHeader::toBuffer(std::shared_ptr rtcp) { return std::make_shared(std::move(rtcp)); } ///////////////////////////////////////////////////////////////////////////// std::shared_ptr RtcpSR::create(size_t item_count) { auto real_size = sizeof(RtcpSR) - sizeof(ReportItem) + item_count * sizeof(ReportItem); auto bytes = alignSize(real_size); auto ptr = (RtcpSR *) new char[bytes]; setupHeader(ptr, RtcpType::RTCP_SR, item_count, bytes); setupPadding(ptr, bytes - real_size); return std::shared_ptr(ptr, [](RtcpSR *ptr) { delete[] (char *) ptr; }); } string RtcpSR::getNtpStamp() const{ struct timeval tv; tv.tv_sec = ntpmsw - 0x83AA7E80; tv.tv_usec = (decltype(tv.tv_usec))(ntplsw / ((double) (((uint64_t) 1) << 32) * 1.0e-6)); return LogChannel::printTime(tv); } uint64_t RtcpSR::getNtpUnixStampMS() const { struct timeval tv; tv.tv_sec = ntpmsw - 0x83AA7E80; tv.tv_usec = (decltype(tv.tv_usec)) (ntplsw / ((double) (((uint64_t) 1) << 32) * 1.0e-6)); return 1000 * tv.tv_sec + tv.tv_usec / 1000; } void RtcpSR::setNtpStamp(struct timeval tv) { ntpmsw = htonl(tv.tv_sec + 0x83AA7E80); /* 0x83AA7E80 is the number of seconds from 1900 to 1970 */ ntplsw = htonl((uint32_t) ((double) tv.tv_usec * (double) (((uint64_t) 1) << 32) * 1.0e-6)); } void RtcpSR::setNtpStamp(uint64_t unix_stamp_ms) { struct timeval tv; tv.tv_sec = unix_stamp_ms / 1000; tv.tv_usec = (unix_stamp_ms % 1000) * 1000; setNtpStamp(tv); } string RtcpSR::dumpString() const{ _StrPrinter printer; printer << RtcpHeader::dumpHeader(); printer << "ssrc:" << ssrc << "\r\n"; printer << "ntpmsw:" << ntpmsw << "\r\n"; printer << "ntplsw:" << ntplsw << "\r\n"; printer << "ntp time:" << getNtpStamp() << "\r\n"; printer << "rtpts:" << rtpts << "\r\n"; printer << "packet_count:" << packet_count << "\r\n"; printer << "octet_count:" << octet_count << "\r\n"; auto items = ((RtcpSR *)this)->getItemList(); auto i = 0; for (auto &item : items) { printer << "---- item:" << i++ << " ----\r\n"; printer << item->dumpString(); } return std::move(printer); } #define CHECK_MIN_SIZE(size, kMinSize) \ if (size < kMinSize) { \ throw std::out_of_range(StrPrinter << rtcpTypeToStr((RtcpType)pt) << " 长度不足:" << size << " < " << kMinSize); \ } #define CHECK_REPORT_COUNT(item_count) \ /*修正个数,防止getItemList时内存越界*/ \ if (report_count != item_count) { \ WarnL << rtcpTypeToStr((RtcpType)pt) << " report_count 字段不正确,已修正为:" << (int)report_count << " -> " << item_count; \ report_count = item_count; \ } void RtcpSR::net2Host(size_t size) { static const size_t kMinSize = sizeof(RtcpSR) - sizeof(items); CHECK_MIN_SIZE(size, kMinSize); ssrc = ntohl(ssrc); ntpmsw = ntohl(ntpmsw); ntplsw = ntohl(ntplsw); rtpts = ntohl(rtpts); packet_count = ntohl(packet_count); octet_count = ntohl(octet_count); ReportItem *ptr = &items; int item_count = 0; for(int i = 0; i < (int)report_count && (char *)(ptr) + sizeof(ReportItem) <= (char *)(this) + size; ++i){ ptr->net2Host(); ++ptr; ++item_count; } CHECK_REPORT_COUNT(item_count); } vector RtcpSR::getItemList(){ vector ret; ReportItem *ptr = &items; for (int i = 0; i < (int) report_count; ++i) { ret.emplace_back(ptr); ++ptr; } return ret; } ///////////////////////////////////////////////////////////////////////////// string ReportItem::dumpString() const{ _StrPrinter printer; printer << "ssrc:" << ssrc << "\r\n"; printer << "fraction:" << fraction << "\r\n"; printer << "cumulative:" << cumulative << "\r\n"; printer << "seq_cycles:" << seq_cycles << "\r\n"; printer << "seq_max:" << seq_max << "\r\n"; printer << "jitter:" << jitter << "\r\n"; printer << "last_sr_stamp:" << last_sr_stamp << "\r\n"; printer << "delay_since_last_sr:" << delay_since_last_sr << "\r\n"; return std::move(printer); } void ReportItem::net2Host() { ssrc = ntohl(ssrc); cumulative = ntohl(cumulative ) >> 8; seq_cycles = ntohs(seq_cycles); seq_max = ntohs(seq_max); jitter = ntohl(jitter); last_sr_stamp = ntohl(last_sr_stamp); delay_since_last_sr = ntohl(delay_since_last_sr); } ///////////////////////////////////////////////////////////////////////////// std::shared_ptr RtcpRR::create(size_t item_count) { auto real_size = sizeof(RtcpRR) - sizeof(ReportItem) + item_count * sizeof(ReportItem); auto bytes = alignSize(real_size); auto ptr = (RtcpRR *) new char[bytes]; setupHeader(ptr, RtcpType::RTCP_RR, item_count, bytes); setupPadding(ptr, bytes - real_size); return std::shared_ptr(ptr, [](RtcpRR *ptr) { delete[] (char *) ptr; }); } string RtcpRR::dumpString() const{ _StrPrinter printer; printer << RtcpHeader::dumpHeader(); printer << "ssrc:" << ssrc << "\r\n"; auto items = ((RtcpRR *)this)->getItemList(); auto i = 0; for (auto &item : items) { printer << "---- item:" << i++ << " ----\r\n"; printer << item->dumpString(); } return std::move(printer); } void RtcpRR::net2Host(size_t size) { static const size_t kMinSize = sizeof(RtcpRR) - sizeof(items); CHECK_MIN_SIZE(size, kMinSize); ssrc = ntohl(ssrc); ReportItem *ptr = &items; int item_count = 0; for(int i = 0; i < (int)report_count && (char *)(ptr) + sizeof(ReportItem) <= (char *)(this) + size; ++i){ ptr->net2Host(); ++ptr; ++item_count; } CHECK_REPORT_COUNT(item_count); } vector RtcpRR::getItemList() { vector ret; ReportItem *ptr = &items; for (int i = 0; i < (int) report_count; ++i) { ret.emplace_back(ptr); ++ptr; } return ret; } ///////////////////////////////////////////////////////////////////////////// void SdesChunk::net2Host() { ssrc = ntohl(ssrc); } size_t SdesChunk::totalBytes() const{ return alignSize(minSize() + txt_len); } size_t SdesChunk::minSize() { return sizeof(SdesChunk) - sizeof(text); } string SdesChunk::dumpString() const{ _StrPrinter printer; printer << "ssrc:" << ssrc << "\r\n"; printer << "type:" << sdesTypeToStr((SdesType) type) << "\r\n"; printer << "txt_len:" << (int) txt_len << "\r\n"; printer << "text:" << (txt_len ? string(text, txt_len) : "") << "\r\n"; return std::move(printer); } ///////////////////////////////////////////////////////////////////////////// std::shared_ptr RtcpSdes::create(const std::vector &item_text) { size_t item_total_size = 0; for (auto &text : item_text) { //统计所有SdesChunk对象占用的空间 item_total_size += alignSize(SdesChunk::minSize() + (0xFF & text.size())); } auto real_size = sizeof(RtcpSdes) - sizeof(SdesChunk) + item_total_size; auto bytes = alignSize(real_size); auto ptr = (RtcpSdes *) new char[bytes]; auto item_ptr = &ptr->chunks; for (auto &text : item_text) { item_ptr->txt_len = (0xFF & text.size()); //确保赋值\0为RTCP_SDES_END memcpy(item_ptr->text, text.data(), item_ptr->txt_len + 1); item_ptr = (SdesChunk *) ((char *) item_ptr + item_ptr->totalBytes()); } setupHeader(ptr, RtcpType::RTCP_SDES, item_text.size(), bytes); setupPadding(ptr, bytes - real_size); return std::shared_ptr(ptr, [](RtcpSdes *ptr) { delete [] (char *) ptr; }); } string RtcpSdes::dumpString() const { _StrPrinter printer; printer << RtcpHeader::dumpHeader(); auto items = ((RtcpSdes *) this)->getChunkList(); auto i = 0; for (auto &item : items) { printer << "---- item:" << i++ << " ----\r\n"; printer << item->dumpString(); } return std::move(printer); } void RtcpSdes::net2Host(size_t size) { static const size_t kMinSize = sizeof(RtcpSdes) - sizeof(chunks); CHECK_MIN_SIZE(size, kMinSize); SdesChunk *ptr = &chunks; int item_count = 0; for(int i = 0; i < (int)report_count && (char *)(ptr) + SdesChunk::minSize() <= (char *)(this) + size; ++i){ ptr->net2Host(); ptr = (SdesChunk *) ((char *) ptr + ptr->totalBytes()); ++item_count; } CHECK_REPORT_COUNT(item_count); } vector RtcpSdes::getChunkList() { vector ret; SdesChunk *ptr = &chunks; for (int i = 0; i < (int) report_count; ++i) { ret.emplace_back(ptr); ptr = (SdesChunk *) ((char *) ptr + ptr->totalBytes()); } return ret; } //////////////////////////////////////////////////////////////////// std::shared_ptr RtcpFB::create_l(RtcpType type, int fmt, const void *fci, size_t fci_len) { if (!fci) { fci_len = 0; } auto real_size = sizeof(RtcpFB) + fci_len; auto bytes = alignSize(real_size); auto ptr = (RtcpFB *) new char[bytes]; if (fci && fci_len) { memcpy((char *)ptr + sizeof(RtcpFB), fci, fci_len); } setupHeader(ptr, type, fmt, bytes); setupPadding(ptr, bytes - real_size); return std::shared_ptr((RtcpFB *) ptr, [](RtcpFB *ptr) { delete[] (char *) ptr; }); } std::shared_ptr RtcpFB::create(PSFBType fmt, const void *fci, size_t fci_len) { return RtcpFB::create_l(RtcpType::RTCP_PSFB, (int)fmt, fci, fci_len); } std::shared_ptr RtcpFB::create(RTPFBType fmt, const void *fci, size_t fci_len) { return RtcpFB::create_l(RtcpType::RTCP_RTPFB, (int)fmt, fci, fci_len); } const void *RtcpFB::getFciPtr() const { return (uint8_t *) &ssrc_media + sizeof(ssrc_media); } size_t RtcpFB::getFciSize() const { auto fci_len = (ssize_t) getSize() - getPaddingSize() - sizeof(RtcpFB); CHECK(fci_len >= 0); return fci_len; } string RtcpFB::dumpString() const { _StrPrinter printer; printer << RtcpHeader::dumpHeader(); printer << "ssrc:" << ssrc << "\r\n"; printer << "ssrc_media:" << ssrc_media << "\r\n"; switch ((RtcpType) pt) { case RtcpType::RTCP_PSFB : { switch ((PSFBType) report_count) { case PSFBType::RTCP_PSFB_SLI : { auto &fci = getFci(); printer << "fci:" << psfbTypeToStr((PSFBType) report_count) << " " << fci.dumpString(); break; } case PSFBType::RTCP_PSFB_PLI : { getFciSize(); printer << "fci:" << psfbTypeToStr((PSFBType) report_count); break; } case PSFBType::RTCP_PSFB_FIR : { auto &fci = getFci(); printer << "fci:" << psfbTypeToStr((PSFBType) report_count) << " " << fci.dumpString(); break; } case PSFBType::RTCP_PSFB_REMB : { auto &fci = getFci(); printer << "fci:" << psfbTypeToStr((PSFBType) report_count) << " " << fci.dumpString(); break; } default:{ printer << "fci:" << psfbTypeToStr((PSFBType) report_count) << " " << hexdump(getFciPtr(), getFciSize()); break; } } break; } case RtcpType::RTCP_RTPFB : { switch ((RTPFBType) report_count) { case RTPFBType::RTCP_RTPFB_NACK : { auto &fci = getFci(); printer << "fci:" << rtpfbTypeToStr((RTPFBType) report_count) << " " << fci.dumpString(); break; } case RTPFBType::RTCP_RTPFB_TWCC : { auto &fci = getFci(); printer << "fci:" << rtpfbTypeToStr((RTPFBType) report_count) << " " << fci.dumpString(getFciSize()); break; } default: { printer << "fci:" << rtpfbTypeToStr((RTPFBType) report_count) << " " << hexdump(getFciPtr(), getFciSize()); break; } } break; } default: /*不可达*/ assert(0); break; } return std::move(printer); } void RtcpFB::net2Host(size_t size) { static const size_t kMinSize = sizeof(RtcpFB); CHECK_MIN_SIZE(size, kMinSize); ssrc = ntohl(ssrc); ssrc_media = ntohl(ssrc_media); } //////////////////////////////////////////////////////////////////// std::shared_ptr RtcpBye::create(const std::vector &ssrcs, const string &reason) { assert(reason.size() <= 0xFF); auto real_size = sizeof(RtcpHeader) + sizeof(uint32_t) * ssrcs.size() + 1 + reason.size(); auto bytes = alignSize(real_size); auto ptr = (RtcpBye *) new char[bytes]; setupHeader(ptr, RtcpType::RTCP_BYE, ssrcs.size(), bytes); setupPadding(ptr, bytes - real_size); auto ssrc_ptr = ((RtcpBye *) ptr)->ssrc; for (auto ssrc : ssrcs) { *ssrc_ptr = htonl(ssrc); ++ssrc_ptr; } if (!reason.empty()) { uint8_t *reason_len_ptr = (uint8_t *) ptr + sizeof(RtcpHeader) + sizeof(uint32_t) * ssrcs.size(); *reason_len_ptr = reason.size() & 0xFF; memcpy(reason_len_ptr + 1, reason.data(), *reason_len_ptr); } return std::shared_ptr(ptr, [](RtcpBye *ptr) { delete[] (char *) ptr; }); } vector RtcpBye::getSSRC() { vector ret; auto ssrc_ptr = ssrc; for (size_t i = 0; i < report_count; ++i) { ret.emplace_back(ssrc_ptr); ssrc_ptr += 1; } return ret; } string RtcpBye::getReason() const { auto *reason_len_ptr = &reason_len + sizeof(ssrc) * (report_count - 1); if (reason_len_ptr + 1 >= (uint8_t *) this + getSize()) { return ""; } return string((char *) reason_len_ptr + 1, *reason_len_ptr); } string RtcpBye::dumpString() const { _StrPrinter printer; printer << RtcpHeader::dumpHeader(); for(auto ssrc : ((RtcpBye *)this)->getSSRC()) { printer << "ssrc:" << *ssrc << "\r\n"; } printer << "reason:" << getReason(); return std::move(printer); } void RtcpBye::net2Host(size_t size) { static const size_t kMinSize = sizeof(RtcpHeader); CHECK_MIN_SIZE(size, kMinSize); auto ssrc_ptr = ssrc; size_t offset = kMinSize; size_t i = 0; for (; i < report_count && offset + sizeof(ssrc) <= size; ++i) { *ssrc_ptr = ntohl(*ssrc_ptr); ssrc_ptr += 1; offset += sizeof(ssrc); } //修正ssrc个数 CHECK_REPORT_COUNT(i); if (offset < size) { uint8_t *reason_len_ptr = &reason_len + sizeof(ssrc) * (report_count - 1); if (reason_len_ptr + 1 + *reason_len_ptr > (uint8_t *) this + size) { WarnL << "invalid rtcp bye reason length"; //修正reason_len长度 *reason_len_ptr = ((uint8_t *) this + size - reason_len_ptr - 1) & 0xFF; } } } #if 0 #include "Util/onceToken.h" static toolkit::onceToken token([](){ auto bye = RtcpBye::create({1,2,3,4,5,6}, "this is a bye reason"); auto buffer = RtcpHeader::toBuffer(bye); auto rtcps = RtcpHeader::loadFromBytes(buffer->data(), buffer->size()); for(auto rtcp : rtcps){ std::cout << rtcp->dumpString() << std::endl; } }); #endif }//namespace mediakit