重写rtcp框架

2026-06-14 12:05:58 +08:00 · 2021-01-31 19:18:17 +08:00
parent 629c39685b
commit 5c6560f55d
16 changed files with 1617 additions and 337 deletions
--- a/tests/test_rtcp.cpp
+++ b/tests/test_rtcp.cpp
@@ -0,0 +1,131 @@
+/*
+ * 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 "Rtcp/Rtcp.h"
+#include "Util/logger.h"
+using namespace std;
+using namespace toolkit;
+using namespace mediakit;
+
+void printRtcp(const std::shared_ptr<Buffer> &buffer){
+    auto rtcp_arr = RtcpHeader::loadFromBytes(buffer->data(), buffer->size());
+    //转换为主机字节序方可打印
+    InfoL << "\r\n" << rtcp_arr[0]->dumpString();
+}
+
+std::shared_ptr<Buffer> makeRtcpSR() {
+    auto rtcp = RtcpSR::create(3);
+    rtcp->ssrc = htonl(1);
+    struct timeval tv;
+    gettimeofday(&tv, NULL);
+    rtcp->setNtpStamp(tv);
+    rtcp->rtpts = htonl(2);
+    rtcp->packet_count = htonl(3);
+    rtcp->octet_count = htonl(4);
+    auto i = 5;
+    for (auto &ptr : rtcp->getItemList()) {
+        ReportItem *item = (ReportItem *) ptr;
+        item->ssrc = htonl(i++);
+        item->fraction = i++;
+        item->cumulative = htonl(i++) >> 8;
+        item->seq_cycles = htons(i++);
+        item->seq_max = htons(i++);
+        item->jitter = htonl(i++);
+        item->last_sr_stamp = htonl(i++);
+        item->delay_since_last_sr = htonl(i++);
+    }
+    //返回网络字节序
+    return RtcpHeader::toBuffer(rtcp);
+}
+
+std::shared_ptr<Buffer> makeRtcpRR() {
+    auto rtcp = RtcpRR::create(3);
+    rtcp->ssrc = htonl(1);
+    auto i = 5;
+    for (auto &ptr : rtcp->getItemList()) {
+        ReportItem *item = (ReportItem *) ptr;
+        item->ssrc = htonl(i++);
+        item->fraction = i++;
+        item->cumulative = htonl(i++) >> 8;
+        item->seq_cycles = htons(i++);
+        item->seq_max = htons(i++);
+        item->jitter = htonl(i++);
+        item->last_sr_stamp = htonl(i++);
+        item->delay_since_last_sr = htonl(i++);
+    }
+    //返回网络字节序
+    return RtcpHeader::toBuffer(rtcp);
+}
+
+std::shared_ptr<Buffer> makeRtcpSDES() {
+    auto rtcp = RtcpSdes::create({"zlmediakit", "", "https://github.com/xia-chu/ZLMediaKit", "1213642868@qq.com", "123456789012345678"});
+    auto i = 5;
+    auto items = rtcp->getItemList();
+    items[0]->type = (uint8_t)SdesType::RTCP_SDES_CNAME;
+    items[0]->ssrc = htonl(i++);
+
+    items[1]->type = (uint8_t)SdesType::RTCP_SDES_NOTE;
+    items[1]->ssrc = htonl(i++);
+
+    items[2]->type = (uint8_t)SdesType::RTCP_SDES_LOC;
+    items[2]->ssrc = htonl(i++);
+
+    items[3]->type = (uint8_t)SdesType::RTCP_SDES_EMAIL;
+    items[3]->ssrc = htonl(i++);
+
+    items[4]->type = (uint8_t)SdesType::RTCP_SDES_PHONE;
+    items[4]->ssrc = htonl(i++);
+
+    //返回网络字节序
+    return RtcpHeader::toBuffer(rtcp);
+}
+
+
+int main(int argc, char *argv[]){
+    Logger::Instance().add(std::make_shared<ConsoleChannel>());
+    {
+        static char rtcp_data[] = "\x81\xca\x00\x05\x70\xd8\xac\x1b\x01\x0b\x7a\x73\x68\x50\x43\x40"
+                                  "\x7a\x73\x68\x50\x43\x00\x00\x00"
+                                  "\x81\xc9\x00\x07\x70\xd8\xac\x1b\x55\x66\x77\x88\x00\x00\x00\x00"
+                                  "\x00\x00\x0d\x21\x00\x00\x00\x32\xdd\xf1\x00\x00\x00\x03\x4f\x67"
+                                  "\x80\xc8\x00\x06\x55\x66\x77\x88\xe3\x70\xdd\xf1\x00\x00\xc2\xb8"
+                                  "\x00\x21\xe4\x90\x00\x00\x0b\x81\x00\x2f\x6a\x60";
+        auto rtcp_arr = RtcpHeader::loadFromBytes(rtcp_data, sizeof(rtcp_data) - 1);
+        for (auto &rtcp : rtcp_arr) {
+            DebugL << "\r\n" << rtcp->dumpString();
+        }
+
+    }
+
+    {
+        printRtcp(makeRtcpSR());
+        printRtcp(makeRtcpRR());
+        printRtcp(makeRtcpSDES());
+    }
+
+    {
+        string str;
+        auto sr = makeRtcpSR();
+        auto rr = makeRtcpRR();
+        auto sdes = makeRtcpSDES();
+        str.append(sr->data(), sr->size());
+        str.append(rr->data(), rr->size());
+        str.append(sdes->data(), sdes->size());
+        //测试内存越界
+        char *data = new char[str.size()];
+        memcpy(data, str.data(), str.size());
+        auto rtcp_arr = RtcpHeader::loadFromBytes(data, str.size());
+        for (auto &rtcp : rtcp_arr) {
+            WarnL << "\r\n" << rtcp->dumpString();
+        }
+        delete [] data;
+    }
+
+}