修复断连续推转协议失效的bug

src/Rtmp/RtmpDemuxer.cpp

@@ -131,6 +131,9 @@ void RtmpDemuxer::inputRtmp(const RtmpPacket::Ptr &pkt) {
 }
 
 void RtmpDemuxer::makeVideoTrack(const AMFValue &videoCodec, int bit_rate) {
+    if (_video_rtmp_decoder) {
+        return;
+    }
     //生成Track对象
     _video_track = dynamic_pointer_cast<VideoTrack>(Factory::getVideoTrackByAmf(videoCodec));
     if (!_video_track) {
@@ -151,6 +154,9 @@ void RtmpDemuxer::makeVideoTrack(const AMFValue &videoCodec, int bit_rate) {
 }
 
 void RtmpDemuxer::makeAudioTrack(const AMFValue &audioCodec,int sample_rate, int channels, int sample_bit, int bit_rate) {
+    if (_audio_rtmp_decoder) {
+        return;
+    }
     //生成Track对象
     _audio_track = dynamic_pointer_cast<AudioTrack>(Factory::getAudioTrackByAmf(audioCodec, sample_rate, channels, sample_bit));
     if (!_audio_track) {

src/Rtsp/RtspDemuxer.cpp

@@ -54,14 +54,14 @@ float RtspDemuxer::getDuration() const {
 bool RtspDemuxer::inputRtp(const RtpPacket::Ptr &rtp) {
     switch (rtp->type) {
         case TrackVideo: {
-            if (_videoRtpDecoder) {
+            if (_video_rtp_decoder) {
-                return _videoRtpDecoder->inputRtp(rtp, true);
+                return _video_rtp_decoder->inputRtp(rtp, true);
             }
             return false;
         }
         case TrackAudio: {
-            if (_audioRtpDecoder) {
+            if (_audio_rtp_decoder) {
-                _audioRtpDecoder->inputRtp(rtp, false);
+                _audio_rtp_decoder->inputRtp(rtp, false);
                 return false;
             }
             return false;
@@ -81,6 +81,9 @@ static void setBitRate(const SdpTrack::Ptr &sdp, const Track::Ptr &track) {
 }
 
 void RtspDemuxer::makeAudioTrack(const SdpTrack::Ptr &audio) {
+    if (_audio_rtp_decoder) {
+        return;
+    }
     //生成Track对象
     _audio_track = dynamic_pointer_cast<AudioTrack>(Factory::getTrackBySdp(audio));
     if (!_audio_track) {
@@ -88,18 +91,21 @@ void RtspDemuxer::makeAudioTrack(const SdpTrack::Ptr &audio) {
     }
     setBitRate(audio, _audio_track);
     //生成RtpCodec对象以便解码rtp
-    _audioRtpDecoder = Factory::getRtpDecoderByTrack(_audio_track);
+    _audio_rtp_decoder = Factory::getRtpDecoderByTrack(_audio_track);
-    if (!_audioRtpDecoder) {
+    if (!_audio_rtp_decoder) {
         //找不到相应的rtp解码器，该track无效
         _audio_track.reset();
         return;
     }
     //设置rtp解码器代理，生成的frame写入该Track
-    _audioRtpDecoder->addDelegate(_audio_track);
+    _audio_rtp_decoder->addDelegate(_audio_track);
     addTrack(_audio_track);
 }
 
 void RtspDemuxer::makeVideoTrack(const SdpTrack::Ptr &video) {
+    if (_video_rtp_decoder) {
+        return;
+    }
     //生成Track对象
     _video_track = dynamic_pointer_cast<VideoTrack>(Factory::getTrackBySdp(video));
     if (!_video_track) {
@@ -107,14 +113,14 @@ void RtspDemuxer::makeVideoTrack(const SdpTrack::Ptr &video) {
     }
     setBitRate(video, _video_track);
     //生成RtpCodec对象以便解码rtp
-    _videoRtpDecoder = Factory::getRtpDecoderByTrack(_video_track);
+    _video_rtp_decoder = Factory::getRtpDecoderByTrack(_video_track);
-    if (!_videoRtpDecoder) {
+    if (!_video_rtp_decoder) {
         //找不到相应的rtp解码器，该track无效
         _video_track.reset();
         return;
     }
     //设置rtp解码器代理，生成的frame写入该Track
-    _videoRtpDecoder->addDelegate(_video_track);
+    _video_rtp_decoder->addDelegate(_video_track);
     addTrack(_video_track);
 }
 

src/Rtsp/RtspDemuxer.h

@@ -54,8 +54,8 @@ private:
     float _duration = 0;
     AudioTrack::Ptr _audio_track;
     VideoTrack::Ptr _video_track;
-    RtpCodec::Ptr _audioRtpDecoder;
+    RtpCodec::Ptr _audio_rtp_decoder;
-    RtpCodec::Ptr _videoRtpDecoder;
+    RtpCodec::Ptr _video_rtp_decoder;
 };
 
 } /* namespace mediakit */

src/Rtsp/RtspMediaSourceImp.h

@@ -40,6 +40,9 @@ public:
      * 设置sdp
      */
     void setSdp(const string &strSdp) override {
+        if (!getSdp().empty()) {
+            return;
+        }
         _demuxer->loadSdp(strSdp);
         RtspMediaSource::setSdp(strSdp);
     }