move srt src like webrtc (#4471)

srt/SrtPlayer.cpp

@@ -0,0 +1,184 @@
+/*
+ * Copyright (c) 2016-present The ZLMediaKit project authors. All Rights Reserved.
+ *
+ * This file is part of ZLMediaKit(https://github.com/ZLMediaKit/ZLMediaKit).
+ *
+ * Use of this source code is governed by MIT-like license that can be found in the
+ * LICENSE file in the root of the source tree. All contributing project authors
+ * may be found in the AUTHORS file in the root of the source tree.
+ */
+
+#include "SrtPlayer.h"
+#include "SrtPlayerImp.h"
+#include "Common/config.h"
+#include "Http/HlsPlayer.h"
+
+using namespace toolkit;
+using namespace std;
+
+namespace mediakit {
+
+
+SrtPlayer::SrtPlayer(const EventPoller::Ptr &poller) 
+    : SrtCaller(poller) {
+    DebugL;
+}
+
+SrtPlayer::~SrtPlayer(void) {
+    DebugL;
+}
+
+void SrtPlayer::play(const string &strUrl) {
+    DebugL;
+    try {
+        _url.parse(strUrl);
+    } catch (std::exception &ex) {
+        onResult(SockException(Err_other, StrPrinter << "illegal srt url:" << ex.what()));
+        return;
+    }
+
+    weak_ptr<SrtPlayer> weak_self = static_pointer_cast<SrtPlayer>(shared_from_this());
+    getPoller()->async([weak_self]() {
+        auto strong_self = weak_self.lock();
+        if (!strong_self) {
+            return;
+        }
+        strong_self->onConnect();
+    });
+    return;
+}
+
+void SrtPlayer::teardown() {
+    SrtCaller::onResult(SockException(Err_other, StrPrinter << "teardown: " << _url._full_url));
+}
+
+void SrtPlayer::pause(bool bPause) {
+    DebugL;
+}
+
+void SrtPlayer::speed(float speed) {
+    DebugL;
+}
+
+void SrtPlayer::onHandShakeFinished() {
+    SrtCaller::onHandShakeFinished();
+    onResult(SockException(Err_success, "srt play success"));
+}
+
+void SrtPlayer::onResult(const SockException &ex) {
+    SrtCaller::onResult(ex);
+
+     if (!ex) {
+        // 播放成功
+        onPlayResult(ex);
+        _benchmark_mode = (*this)[Client::kBenchmarkMode].as<int>();
+
+        // 播放成功，恢复数据包接收超时定时器
+        _recv_ticker.resetTime();
+        auto timeout = getTimeOutSec();
+        //读取配置文件
+        weak_ptr<SrtPlayer> weakSelf = static_pointer_cast<SrtPlayer>(shared_from_this());
+        // 创建rtp数据接收超时检测定时器
+        _check_timer = std::make_shared<Timer>(timeout /2,
+            [weakSelf, timeout]() {
+                auto strongSelf = weakSelf.lock();
+                if (!strongSelf) {
+                    return false;
+                }
+                if (strongSelf->_recv_ticker.elapsedTime() > timeout * 1000) {
+                    // 接收媒体数据包超时
+                    strongSelf->onResult(SockException(Err_timeout, "receive srt media data timeout:" + strongSelf->_url._full_url));
+                    return false;
+                }
+
+                return true;
+            }, getPoller());
+    } else {
+        WarnL << ex.getErrCode() << " " << ex.what();
+        if (ex.getErrCode() == Err_shutdown) {
+            // 主动shutdown的，不触发回调
+            return;
+        }
+        if (!_is_handleshake_finished) {
+            onPlayResult(ex);
+        } else {
+            onShutdown(ex);
+        }
+    }
+    return;
+}
+
+
+void SrtPlayer::onSRTData(SRT::DataPacket::Ptr pkt) {
+    _recv_ticker.resetTime();
+}
+
+uint16_t SrtPlayer::getLatency() {
+    auto latency = (*this)[Client::kLatency].as<uint16_t>();
+    return (uint16_t)latency ;
+}
+
+float SrtPlayer::getTimeOutSec() {
+    auto timeoutMS = (*this)[Client::kTimeoutMS].as<uint64_t>();
+    return (float)timeoutMS / (float)1000;
+}
+
+std::string SrtPlayer::getPassphrase() {
+    auto passPhrase = (*this)[Client::kPassPhrase].as<string>();
+    return passPhrase;
+}
+
+size_t SrtPlayer::getRecvSpeed() {
+    return SrtCaller::getRecvSpeed();
+}
+
+size_t SrtPlayer::getRecvTotalBytes() {
+    return SrtCaller::getRecvTotalBytes();
+}
+
+///////////////////////////////////////////////////
+// SrtPlayerImp
+
+void SrtPlayerImp::onPlayResult(const toolkit::SockException &ex) {
+    if (ex) {
+        Super::onPlayResult(ex);
+    }
+    //success result only occur when addTrackCompleted
+    return;
+}
+
+std::vector<Track::Ptr> SrtPlayerImp::getTracks(bool ready /*= true*/) const {
+    return _demuxer ? static_pointer_cast<HlsDemuxer>(_demuxer)->getTracks(ready) : Super::getTracks(ready);
+}
+
+void SrtPlayerImp::addTrackCompleted() {
+    Super::onPlayResult(toolkit::SockException(toolkit::Err_success, "play success"));
+}
+
+void SrtPlayerImp::onSRTData(SRT::DataPacket::Ptr pkt) {
+    SrtPlayer::onSRTData(pkt);
+
+    if (_benchmark_mode) {
+        return;
+    }
+
+    auto strong_self = shared_from_this();
+    if (!_demuxer) {
+        auto demuxer = std::make_shared<HlsDemuxer>();
+        demuxer->start(getPoller(), this);
+        _demuxer = std::move(demuxer);
+    }
+
+    if (!_decoder && _demuxer) {
+        _decoder = DecoderImp::createDecoder(DecoderImp::decoder_ts, _demuxer.get());
+    }
+
+    if (_decoder && _demuxer) {
+        _decoder->input(reinterpret_cast<const uint8_t *>(pkt->payloadData()), pkt->payloadSize());
+    }
+
+    return;
+}
+
+} /* namespace mediakit */
+