diff --git a/src/Rtcp/Rtcp.cpp b/src/Rtcp/Rtcp.cpp index 33ac59e3..e3c95927 100644 --- a/src/Rtcp/Rtcp.cpp +++ b/src/Rtcp/Rtcp.cpp @@ -52,7 +52,7 @@ const char *rtpfbTypeToStr(RTPFBType type) { } static size_t alignSize(size_t bytes) { - return (size_t)((bytes + 3) / 4) << 2; + return (size_t)((bytes + 3) >> 2 ) << 2; } static void setupHeader(RtcpHeader *rtcp, RtcpType type, size_t report_count, size_t total_bytes) { @@ -64,23 +64,32 @@ static void setupHeader(RtcpHeader *rtcp, RtcpType type, size_t report_count, si //items总个数 rtcp->report_count = report_count; rtcp->pt = (uint8_t) type; - //不包含rtcp头的长度 - rtcp->length = htons((uint16_t)((total_bytes / 4) - 1)); + rtcp->setSize(total_bytes); } ///////////////////////////////////////////////////////////////////////////// -void RtcpHeader::net2Host() { - length = ntohs(length); -} - string RtcpHeader::dumpHeader() const{ _StrPrinter printer; printer << "version:" << version << "\r\n"; printer << "padding:" << padding << "\r\n"; - printer << "report_count:" << report_count << "\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 << "length:" << length << "\r\n"; + printer << "size:" << getSize() << "\r\n"; printer << "--------\r\n"; return std::move(printer); } @@ -102,8 +111,9 @@ string RtcpHeader::dumpString() const { return rtcp->dumpString(); } + case RtcpType::RTCP_RTPFB: case RtcpType::RTCP_PSFB: { - RtcpPli *rtcp = (RtcpPli *)this; + RtcpFB *rtcp = (RtcpFB *)this; return rtcp->dumpString(); } @@ -112,12 +122,18 @@ string RtcpHeader::dumpString() const { return rtcp->dumpString(); } - default: return StrPrinter << dumpHeader() << hexdump((char *)this + sizeof(*this), length << 2); + default: return StrPrinter << dumpHeader() << hexdump((char *)this + sizeof(*this), getSize() - sizeof(*this)); } } size_t RtcpHeader::getSize() const { - return (1 + length) << 2; + //加上rtcp头长度 + return (1 + ntohs(length)) << 2; +} + +void RtcpHeader::setSize(size_t size) { + //不包含rtcp头的长度 + length = htons((uint16_t)((size >> 2) - 1)); } void RtcpHeader::net2Host(size_t len){ @@ -140,9 +156,10 @@ void RtcpHeader::net2Host(size_t len){ break; } + case RtcpType::RTCP_RTPFB: case RtcpType::RTCP_PSFB: { - RtcpPli *pli = (RtcpPli *)this; - pli->net2Host(len); + RtcpFB *fb = (RtcpFB *)this; + fb->net2Host(len); break; } @@ -152,11 +169,6 @@ void RtcpHeader::net2Host(size_t len){ break; } - case RtcpType::RTCP_RTPFB: { - RtcpPli *pli = (RtcpPli *)this; - pli->net2Host(len); - break; - } default: throw std::runtime_error(StrPrinter << "未处理的rtcp包:" << rtcpTypeToStr((RtcpType) this->pt)); } } @@ -167,8 +179,8 @@ vector RtcpHeader::loadFromBytes(char *data, size_t len){ char *ptr = data; while (remain > (ssize_t) sizeof(RtcpHeader)) { RtcpHeader *rtcp = (RtcpHeader *) ptr; - auto rtcp_len = (1 + ntohs(rtcp->length)) << 2; - if (remain < rtcp_len) { + auto rtcp_len = rtcp->getSize(); + if (remain < (ssize_t)rtcp_len) { WarnL << "非法的rtcp包,声明的长度超过实际数据长度"; break; } @@ -189,7 +201,6 @@ class BufferRtcp : public Buffer { public: BufferRtcp(std::shared_ptr rtcp) { _rtcp = std::move(rtcp); - _size = (ntohs(_rtcp->length) + 1) << 2; } ~BufferRtcp() override {} @@ -199,7 +210,7 @@ public: } size_t size() const override { - return _size; + return _rtcp->getSize(); } private: @@ -258,21 +269,17 @@ if (size < kMinSize) { \ throw std::out_of_range(StrPrinter << rtcpTypeToStr((RtcpType)pt) << " 长度不足:" << size << " < " << kMinSize); \ } -#define CHECK_LENGTH(size, item_count) \ +#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; \ -} \ -if ((size_t) (length + 1) << 2 != size) { \ - WarnL << rtcpTypeToStr((RtcpType)pt) << " length字段不正确:" << (size_t) (length + 1) << 2 << " != " << size; \ } void RtcpSR::net2Host(size_t size) { static const size_t kMinSize = sizeof(RtcpSR) - sizeof(items); CHECK_MIN_SIZE(size, kMinSize); - RtcpHeader::net2Host(); ssrc = ntohl(ssrc); ntpmsw = ntohl(ntpmsw); ntplsw = ntohl(ntplsw); @@ -287,7 +294,7 @@ void RtcpSR::net2Host(size_t size) { ++ptr; ++item_count; } - CHECK_LENGTH(size, item_count); + CHECK_REPORT_COUNT(item_count); } vector RtcpSR::getItemList(){ @@ -352,7 +359,6 @@ string RtcpRR::dumpString() const{ void RtcpRR::net2Host(size_t size) { static const size_t kMinSize = sizeof(RtcpRR) - sizeof(items); CHECK_MIN_SIZE(size, kMinSize); - RtcpHeader::net2Host(); ssrc = ntohl(ssrc); ReportItem *ptr = &items; @@ -362,7 +368,7 @@ void RtcpRR::net2Host(size_t size) { ++ptr; ++item_count; } - CHECK_LENGTH(size, item_count); + CHECK_REPORT_COUNT(item_count); } vector RtcpRR::getItemList() { @@ -382,7 +388,7 @@ void SdesItem::net2Host() { } size_t SdesItem::totalBytes() const{ - return alignSize(minSize() + length); + return alignSize(minSize() + txt_len); } size_t SdesItem::minSize() { @@ -393,14 +399,14 @@ string SdesItem::dumpString() const{ _StrPrinter printer; printer << "ssrc:" << ssrc << "\r\n"; printer << "type:" << sdesTypeToStr((SdesType) type) << "\r\n"; - printer << "length:" << (int) length << "\r\n"; - printer << "text:" << (length ? string(text, length) : "") << "\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::initializer_list &item_text) { +std::shared_ptr RtcpSdes::create(const std::vector &item_text) { size_t item_total_size = 0; for (auto &text : item_text) { //统计所有SdesItem对象占用的空间 @@ -410,9 +416,9 @@ std::shared_ptr RtcpSdes::create(const std::initializer_list & auto ptr = (RtcpSdes *) new char[bytes]; auto item_ptr = &ptr->items; for (auto &text : item_text) { - item_ptr->length = (0xFF & text.size()); + item_ptr->txt_len = (0xFF & text.size()); //确保赋值\0为RTCP_SDES_END - memcpy(item_ptr->text, text.data(), item_ptr->length + 1); + memcpy(item_ptr->text, text.data(), item_ptr->txt_len + 1); item_ptr = (SdesItem *) ((char *) item_ptr + item_ptr->totalBytes()); } @@ -437,7 +443,6 @@ string RtcpSdes::dumpString() const { void RtcpSdes::net2Host(size_t size) { static const size_t kMinSize = sizeof(RtcpSdes) - sizeof(items); CHECK_MIN_SIZE(size, kMinSize); - RtcpHeader::net2Host(); SdesItem *ptr = &items; int item_count = 0; for(int i = 0; i < (int)report_count && (char *)(ptr) + SdesItem::minSize() <= (char *)(this) + size; ++i){ @@ -445,7 +450,7 @@ void RtcpSdes::net2Host(size_t size) { ptr = (SdesItem *) ((char *) ptr + ptr->totalBytes()); ++item_count; } - CHECK_LENGTH(size, item_count); + CHECK_REPORT_COUNT(item_count); } vector RtcpSdes::getItemList() { @@ -460,34 +465,52 @@ vector RtcpSdes::getItemList() { //////////////////////////////////////////////////////////////////// -std::shared_ptr RtcpPli::create() { - auto bytes = alignSize(sizeof(RtcpPli)); +std::shared_ptr RtcpFB::create_l(RtcpType type, int fmt, const void *fci, size_t fci_len) { + if (!fci) { + fci_len = 0; + } + auto bytes = alignSize(sizeof(RtcpFB) + fci_len); auto ptr = (RtcpRR *) new char[bytes]; - setupHeader(ptr, RtcpType::RTCP_PSFB, 1, bytes); - return std::shared_ptr((RtcpPli *) ptr, [](RtcpPli *ptr) { + if (fci && fci_len) { + memcpy(ptr + sizeof(RtcpFB), fci, fci_len); + } + setupHeader(ptr, type, fmt, bytes); + return std::shared_ptr((RtcpFB *) ptr, [](RtcpFB *ptr) { delete[] (char *) ptr; }); } -string RtcpPli::dumpString() const { +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); +} + +string RtcpFB::dumpString() const { _StrPrinter printer; printer << RtcpHeader::dumpHeader(); printer << "ssrc:" << ssrc << "\r\n"; - printer << "ssrc_media:" << ssrc_media; + printer << "ssrc_media:" << ssrc_media << "\r\n"; + auto fci = (uint8_t *)&ssrc_media + sizeof(ssrc_media); + auto fci_len = getSize() - sizeof(RtcpFB); + if (fci_len) { + printer << "fci:" << hexdump(fci, fci_len); + } return std::move(printer); } -void RtcpPli::net2Host(size_t size) { - static const size_t kMinSize = sizeof(RtcpPli); +void RtcpFB::net2Host(size_t size) { + static const size_t kMinSize = sizeof(RtcpFB); CHECK_MIN_SIZE(size, kMinSize); - RtcpHeader::net2Host(); ssrc = ntohl(ssrc); ssrc_media = ntohl(ssrc_media); } //////////////////////////////////////////////////////////////////// -std::shared_ptr RtcpBye::create(const std::initializer_list &ssrcs, const string &reason) { +std::shared_ptr RtcpBye::create(const std::vector &ssrcs, const string &reason) { assert(reason.size() <= 0xFF); auto bytes = alignSize(sizeof(RtcpHeader) + sizeof(uint32_t) * ssrcs.size() + 1 + reason.size()); auto ptr = (RtcpBye *) new char[bytes]; @@ -541,8 +564,6 @@ string RtcpBye::dumpString() const { void RtcpBye::net2Host(size_t size) { static const size_t kMinSize = sizeof(RtcpHeader); CHECK_MIN_SIZE(size, kMinSize); - RtcpHeader::net2Host(); - uint32_t *ssrc_ptr = &ssrc; size_t offset = kMinSize; size_t i = 0; @@ -552,7 +573,7 @@ void RtcpBye::net2Host(size_t size) { offset += sizeof(ssrc); } //修正ssrc个数 - CHECK_LENGTH(size, i); + CHECK_REPORT_COUNT(i); if (offset < size) { uint8_t *reason_len_ptr = &reason_len + sizeof(ssrc) * (report_count - 1); diff --git a/src/Rtcp/Rtcp.h b/src/Rtcp/Rtcp.h index 838ddf40..ddc5ae71 100644 --- a/src/Rtcp/Rtcp.h +++ b/src/Rtcp/Rtcp.h @@ -179,6 +179,8 @@ public: #endif //rtcp类型,RtcpType uint32_t pt: 8; + +private: //长度 uint32_t length: 16; @@ -207,15 +209,16 @@ public: /** * 根据length字段获取rtcp总长度 - * 使用net2Host转换成主机字节序后才可使用此函数 */ size_t getSize() const; -protected: /** - * 网络字节序转换为主机字节序 + * 设置rtcp length字段 + * @param size rtcp总长度,单位字节 */ - void net2Host(); + void setSize(size_t size); + +protected: /** * 打印字段详情 @@ -476,7 +479,7 @@ public: //SdesType uint8_t type; //text长度股,可以为0 - uint8_t length; + uint8_t txt_len; //不定长 char text[1]; //最后以RTCP_SDES_END结尾 @@ -521,7 +524,7 @@ public: * @param item_text SdesItem列表,只赋值length和text部分 * @return SDES包 */ - static std::shared_ptr create(const std::initializer_list &item_text); + static std::shared_ptr create(const std::vector &item_text); /** * 获取SdesItem对象指针列表 @@ -560,8 +563,8 @@ private: // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ // : Feedback Control Information (FCI) : // : : - -class RtcpPli : public RtcpHeader { +// rtcpfb和psfb的数据结构一致 +class RtcpFB : public RtcpHeader { public: friend class RtcpHeader; uint32_t ssrc; @@ -569,9 +572,14 @@ public: public: /** - * 创建pli包 + * 创建psfb类型的反馈包 */ - static std::shared_ptr create(); + static std::shared_ptr create(PSFBType fmt, const void *fci = nullptr, size_t fci_len = 0); + + /** + * 创建rtpfb类型的反馈包 + */ + static std::shared_ptr create(RTPFBType fmt, const void *fci = nullptr, size_t fci_len = 0); private: /** @@ -585,6 +593,9 @@ private: * @param size 字节长度,防止内存越界 */ void net2Host(size_t size); + +private: + static std::shared_ptr create_l(RtcpType type, int fmt, const void *fci, size_t fci_len); } PACKED; //BYE @@ -621,7 +632,7 @@ public: * @param reason 原因 * @return rtcp bye包 */ - static std::shared_ptr create(const std::initializer_list &ssrc, const string &reason); + static std::shared_ptr create(const std::vector &ssrc, const string &reason); /** * 获取ssrc列表 diff --git a/webrtc/WebRtcTransport.cpp b/webrtc/WebRtcTransport.cpp index 800daa1d..aeb8a1ec 100644 --- a/webrtc/WebRtcTransport.cpp +++ b/webrtc/WebRtcTransport.cpp @@ -544,21 +544,9 @@ void WebRtcTransportImp::onRtcp(const char *buf, size_t len) { onShutdown(SockException(Err_eof, "rtcp bye message received")); break; } - case RtcpType::RTCP_PSFB: { - //todo 支持pli等更多类型的rtcp - break; - } + case RtcpType::RTCP_PSFB: case RtcpType::RTCP_RTPFB: { - //todo 测试打印twcc - RtcpPli *rtpfb = (RtcpPli *)rtcp; - auto fci = (uint8_t *)rtpfb + sizeof (RtcpPli); - if(rtpfb->report_count == 15){ - //TWCC - FCI_TWCC *twcc = (FCI_TWCC *) (fci); - auto fci_size = rtpfb->getSize() - 12; - InfoL << hexdump(fci, fci_size); - InfoL << "\n" << twcc->dumpString(fci_size); - } + InfoL << rtcp->dumpString(); break; } default: break; @@ -591,10 +579,10 @@ void WebRtcTransportImp::onSortedRtp(const RtpPayloadInfo &info, RtpPacket::Ptr if (_pli_ticker.elapsedTime() > 2000) { //定期发送pli请求关键帧,方便非rtc等协议 _pli_ticker.resetTime(); - auto pli = RtcpPli::create(); + auto pli = RtcpFB::create(PSFBType::RTCP_PSFB_PLI); pli->ssrc = htonl(0); pli->ssrc_media = htonl(_recv_video_ssrc); - sendRtcpPacket((char *) pli.get(), sizeof(RtcpPli), true); + sendRtcpPacket((char *) pli.get(), pli->getSize(), true); } if (_push_src) { _push_src->onWrite(std::move(rtp), false);