add rtp server timeout hook

src/Common/config.cpp

@@ -56,6 +56,7 @@ const string kBroadcastNotFoundStream = "kBroadcastNotFoundStream";
 const string kBroadcastStreamNoneReader = "kBroadcastStreamNoneReader";
 const string kBroadcastHttpBeforeAccess = "kBroadcastHttpBeforeAccess";
 const string kBroadcastSendRtpStopped = "kBroadcastSendRtpStopped";
+const string KBroadcastRtpServerTimeout = "KBroadcastRtpServerTimeout";
 
 } // namespace Broadcast
 

src/Common/config.h

@@ -114,6 +114,10 @@ extern const std::string kBroadcastSendRtpStopped;
 extern const std::string kBroadcastReloadConfig;
 #define BroadcastReloadConfigArgs void
 
+// rtp server 超时
+extern const std::string KBroadcastRtpServerTimeout;
+#define BroadcastRtpServerTimeout uint16_t &local_port, const string &stream_id,int &tcp_mode, bool &re_use_port, uint32_t &ssrc
+
 #define ReloadConfigTag ((void *)(0xFF))
 #define RELOAD_KEY(arg, key)                                                                                           \
     do {                                                                                                               \

src/Rtp/RtpServer.cpp

@@ -36,12 +36,20 @@ public:
         _stream_id = std::move(stream_id);
     }
 
+    
+
     ~RtcpHelper() {
         if (_process) {
             // 删除rtp处理器
             RtpSelector::Instance().delProcess(_stream_id, _process.get());
         }
     }
+    void setRtpServerInfo(uint16_t local_port,RtpServer::TcpMode mode,bool re_use_port,uint32_t ssrc){
+        _local_port = local_port;
+        _tcp_mode = mode;
+        _re_use_port = re_use_port;
+        _ssrc = ssrc;
+    }
 
     void setOnDetach(function<void()> cb) {
         if (_process) {
@@ -55,8 +63,7 @@ public:
         if (!_process) {
             _process = RtpSelector::Instance().getProcess(_stream_id, true);
             _process->setOnDetach(std::move(_on_detach));
-            _delay_task->cancel();
-            _delay_task = nullptr;
+            cancelDelayTask();
         }
         _process->inputRtp(true, sock, buf->data(), buf->size(), addr);
 
@@ -91,11 +98,21 @@ public:
                 if (!process && strong_self->_on_detach) {
                     strong_self->_on_detach();
                 }
+             if(!process){ // process 未创建，触发rtp server 超时事件
+                NoticeCenter::Instance().emitEvent(Broadcast::KBroadcastRtpServerTimeout,strong_self->_local_port,strong_self->_stream_id,(int)strong_self->_tcp_mode,strong_self->_re_use_port,strong_self->_ssrc);
+             }
             }
             return 0;
         });
     }
 
+    void cancelDelayTask(){
+        if(_delay_task){
+            _delay_task->cancel();
+            _delay_task = nullptr;
+        }
+    }
+
 private:
     void sendRtcp(uint32_t rtp_ssrc, struct sockaddr *addr) {
         // 每5秒发送一次rtcp
@@ -118,6 +135,12 @@ private:
     }
 
 private:
+    uint16_t _local_port = 0;
+    RtpServer::TcpMode _tcp_mode = RtpServer::NONE;
+    bool _re_use_port = false;
+    uint32_t _ssrc = 0;
+
+
     Ticker _ticker;
     Socket::Ptr _rtcp_sock;
     RtpProcess::Ptr _process;
@@ -169,6 +192,7 @@ void RtpServer::start(uint16_t local_port, const string &stream_id, TcpMode tcp_
         //指定了流id，那么一个端口一个流(不管是否包含多个ssrc的多个流，绑定rtp源后，会筛选掉ip端口不匹配的流)
         helper = std::make_shared<RtcpHelper>(std::move(rtcp_socket), stream_id);
         helper->startRtcp();
+        helper->setRtpServerInfo(local_port,tcp_mode,re_use_port,ssrc);
         rtp_socket->setOnRead([rtp_socket, helper, ssrc](const Buffer::Ptr &buf, struct sockaddr *addr, int addr_len) {
             RtpHeader *header = (RtpHeader *)buf->data();
             auto rtp_ssrc = ntohl(header->ssrc);