重写rtp去冲突逻辑

src/Common/config.cpp

@@ -281,14 +281,11 @@ namespace RtpProxy {
 #define RTP_PROXY_FIELD "rtp_proxy."
 //rtp调试数据保存目录
 const string kDumpDir = RTP_PROXY_FIELD"dumpDir";
-//是否限制udp数据来源ip和端口
-const string kCheckSource = RTP_PROXY_FIELD"checkSource";
 //rtp接收超时时间
 const string kTimeoutSec = RTP_PROXY_FIELD"timeoutSec";
 
 onceToken token([](){
     mINI::Instance()[kDumpDir] = "";
-    mINI::Instance()[kCheckSource] = 1;
     mINI::Instance()[kTimeoutSec] = 15;
 },nullptr);
 } //namespace RtpProxy

src/Common/config.h

@@ -301,8 +301,6 @@ extern const string kBroadcastRecordTs;
 namespace RtpProxy {
 //rtp调试数据保存目录,置空则不生成
 extern const string kDumpDir;
-//是否限制udp数据来源ip和端口
-extern const string kCheckSource;
 //rtp接收超时时间
 extern const string kTimeoutSec;
 } //namespace RtpProxy

src/Rtp/RtpProcess.cpp

@@ -11,18 +11,12 @@
 #if defined(ENABLE_RTPPROXY)
 #include "GB28181Process.h"
 #include "RtpProcess.h"
-#include "RtpSplitter.h"
-#include "Util/File.h"
 #include "Http/HttpTSPlayer.h"
 
 #define RTP_APP_NAME "rtp"
 
 namespace mediakit {
 
-static string printAddress(const struct sockaddr *addr) {
-    return StrPrinter << SockUtil::inet_ntoa(((struct sockaddr_in *) addr)->sin_addr) << ":" << ntohs(((struct sockaddr_in *) addr)->sin_port);
-}
-
 RtpProcess::RtpProcess(const string &stream_id) {
     _media_info._schema = RTP_APP_NAME;
     _media_info._vhost = DEFAULT_VHOST;
@@ -63,23 +57,17 @@ RtpProcess::~RtpProcess() {
     if (_total_bytes >= iFlowThreshold * 1024) {
         NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastFlowReport, _media_info, _total_bytes, duration, false, static_cast<SockInfo &>(*this));
     }
-
-    if (_addr) {
-        delete _addr;
-        _addr = nullptr;
-    }
 }
 
 bool RtpProcess::inputRtp(bool is_udp, const Socket::Ptr &sock, const char *data, size_t len, const struct sockaddr *addr, uint32_t *dts_out) {
-    GET_CONFIG(bool, check_source, RtpProxy::kCheckSource);
-    //检查源是否合法
-    if (!_addr) {
-        _addr = new struct sockaddr;
+    if (!_sock) {
         _sock = sock;
-        memcpy(_addr, addr, sizeof(struct sockaddr));
-        DebugP(this) << "bind to address:" << printAddress(_addr);
-        //推流鉴权
+        _addr = *addr;
         emitOnPublish();
+    } else if (!_sock->getPoller()->isCurrentThread()) {
+        //其他线程执行本对象，存在线程安全问题
+        WarnP(this) << "其他线程执行本对象";
+        return false;
     }
 
     if (!_muxer) {
@@ -87,11 +75,6 @@ bool RtpProcess::inputRtp(bool is_udp, const Socket::Ptr &sock, const char *data
         return false;
     }
 
-    if (check_source && memcmp(_addr, addr, sizeof(struct sockaddr)) != 0) {
-        DebugP(this) << "address dismatch:" << printAddress(addr) << " != " << printAddress(_addr);
-        return false;
-    }
-
     _total_bytes += len;
     if (_save_file_rtp) {
         uint16_t size = (uint16_t)len;
@@ -161,17 +144,11 @@ void RtpProcess::setOnDetach(const function<void()> &cb) {
 }
 
 string RtpProcess::get_peer_ip() {
-    if (_addr) {
-        return SockUtil::inet_ntoa(((struct sockaddr_in *) _addr)->sin_addr);
-    }
-    return "0.0.0.0";
+    return SockUtil::inet_ntoa(((struct sockaddr_in &) _addr).sin_addr);
 }
 
 uint16_t RtpProcess::get_peer_port() {
-    if (!_addr) {
-        return 0;
-    }
-    return ntohs(((struct sockaddr_in *) _addr)->sin_port);
+    return ntohs(((struct sockaddr_in &) _addr).sin_port);
 }
 
 string RtpProcess::get_local_ip() {

src/Rtp/RtpProcess.h

@@ -83,7 +83,7 @@ private:
 private:
     uint32_t _dts = 0;
     uint64_t _total_bytes = 0;
-    struct sockaddr *_addr = nullptr;
+    struct sockaddr _addr{0};
     Socket::Ptr _sock;
     MediaInfo _media_info;
     Ticker _last_frame_time;