复用rtp/rtcp判断逻辑过滤非rtp包: #2247

src/Rtp/RtpProcess.cpp

@@ -70,14 +70,9 @@ RtpProcess::~RtpProcess() {
     }
 }
 
-static bool is_rtp(const char *buf) {
-    RtpHeader *header = (RtpHeader *)buf;
-    return ((header->pt < 64) || (header->pt >= 96));
-}
-
 bool RtpProcess::inputRtp(bool is_udp, const Socket::Ptr &sock, const char *data, size_t len, const struct sockaddr *addr, uint64_t *dts_out) {
-    if (!is_rtp(data)) {
-        WarnL << "Not rtp packet";
+    if (!isRtp(data, len)) {
+        WarnP(this) << "Not rtp packet";
         return false;
     }
     if (_sock != sock) {

src/Rtp/RtpSession.cpp

@@ -12,6 +12,7 @@
 #include "RtpSession.h"
 #include "RtpSelector.h"
 #include "Network/TcpServer.h"
+#include "Rtsp/Rtsp.h"
 #include "Rtsp/RtpReceiver.h"
 #include "Common/config.h"
 
@@ -93,6 +94,10 @@ void RtpSession::onRtpPacket(const char *data, size_t len) {
         }
     }
     if (!_process) {
+        if (!isRtp(data, len)) {
+            WarnP(this) << "Not rtp packet";
+            return;
+        }
         //未设置ssrc时，尝试获取ssrc
         if (!_ssrc && !RtpSelector::getSSRC(data, len, _ssrc)) {
             return;

src/Rtsp/Rtsp.cpp

@@ -442,6 +442,22 @@ string printSSRC(uint32_t ui32Ssrc) {
     return tmp;
 }
 
+bool isRtp(const char *buf, size_t size) {
+    if (size < 2) {
+        return false;
+    }
+    RtpHeader *header = (RtpHeader *)buf;
+    return ((header->pt < 64) || (header->pt >= 96));
+}
+
+bool isRtcp(const char *buf, size_t size) {
+    if (size < 2) {
+        return false;
+    }
+    RtpHeader *header = (RtpHeader *)buf;
+    return ((header->pt >= 64) && (header->pt < 96));
+}
+
 Buffer::Ptr makeRtpOverTcpPrefix(uint16_t size, uint8_t interleaved) {
     auto rtp_tcp = BufferRaw::create();
     rtp_tcp->setCapacity(RtpPacket::kRtpTcpHeaderSize);

src/Rtsp/Rtsp.h

@@ -337,5 +337,8 @@ void makeSockPair(std::pair<toolkit::Socket::Ptr, toolkit::Socket::Ptr> &pair, c
 //十六进制方式打印ssrc
 std::string printSSRC(uint32_t ui32Ssrc);
 
+bool isRtp(const char *buf, size_t size);
+bool isRtcp(const char *buf,  size_t size);
+
 } //namespace mediakit
 #endif //RTSP_RTSP_H_