From eb382c3fc917f046445a2bc13704356c893f0888 Mon Sep 17 00:00:00 2001
From: Dw9 <xweimvp@gmail.com>
Date: Sat, 22 Jul 2023 10:46:17 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dwhip=20delete=5Fwebrtc?=
 =?UTF-8?q?=E6=9C=AA=E6=B3=A8=E9=94=80=E6=B5=81=20(#2688)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

bug触发条件：
1、obs whip 推流到zlm
2、关闭推流
3、zlm 未注销流
---
 webrtc/WebRtcPusher.cpp | 9 +++++++--
 webrtc/WebRtcPusher.h   | 4 +++-
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/webrtc/WebRtcPusher.cpp b/webrtc/WebRtcPusher.cpp
index 5d6a68d2..11fb1b30 100644
--- a/webrtc/WebRtcPusher.cpp
+++ b/webrtc/WebRtcPusher.cpp
@@ -149,7 +149,7 @@ void WebRtcPusher::onRtcConfigure(RtcConfigure &configure) const {
     configure.audio.direction = configure.video.direction = RtpDirection::recvonly;
 }
 
-float WebRtcPusher::getLossRate(MediaSource &sender,TrackType type){
+float WebRtcPusher::getLossRate(MediaSource &sender,TrackType type) {
     return WebRtcTransportImp::getLossRate(type);
 }
 
@@ -159,8 +159,13 @@ void WebRtcPusher::OnDtlsTransportClosed(const RTC::DtlsTransport *dtlsTransport
     WebRtcTransportImp::OnDtlsTransportClosed(dtlsTransport);
 }
 
-void WebRtcPusher::onRtcpBye(){
+void WebRtcPusher::onRtcpBye() {
      WebRtcTransportImp::onRtcpBye();
 }
 
+void WebRtcPusher::onShutdown(const SockException &ex) {
+     _push_src = nullptr;
+     WebRtcTransportImp::onShutdown(ex);
+}
+
 }// namespace mediakit
\ No newline at end of file
diff --git a/webrtc/WebRtcPusher.h b/webrtc/WebRtcPusher.h
index b35f9e98..4c332a35 100644
--- a/webrtc/WebRtcPusher.h
+++ b/webrtc/WebRtcPusher.h
@@ -23,15 +23,17 @@ public:
     static Ptr create(const EventPoller::Ptr &poller, const RtspMediaSource::Ptr &src,
                       const std::shared_ptr<void> &ownership, const MediaInfo &info, const ProtocolOption &option, bool preferred_tcp = false);
 
+
 protected:
     ///////WebRtcTransportImp override///////
     void onStartWebRTC() override;
     void onDestory() override;
     void onRtcConfigure(RtcConfigure &configure) const override;
     void onRecvRtp(MediaTrack &track, const std::string &rid, RtpPacket::Ptr rtp) override;
+    void onShutdown(const SockException &ex) override;
     void onRtcpBye() override;
     ////  dtls相关的回调 ////
-   void OnDtlsTransportClosed(const RTC::DtlsTransport *dtlsTransport) override;
+    void OnDtlsTransportClosed(const RTC::DtlsTransport *dtlsTransport) override;
 
 protected:
     ///////MediaSourceEvent override///////