From e807a08f2edc6692c3b4368e330ed7562cd7f8e7 Mon Sep 17 00:00:00 2001 From: xiongziliang <771730766@qq.com> Date: Sat, 15 May 2021 09:56:35 +0800 Subject: [PATCH 1/2] =?UTF-8?q?=E5=8D=95track=20rtsp=E6=8E=A8=E6=B5=81?= =?UTF-8?q?=E5=8A=A0=E5=BF=AB=E5=AA=92=E4=BD=93=E6=B3=A8=E5=86=8C=E9=80=9F?= =?UTF-8?q?=E5=BA=A6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/Rtsp/RtspSession.cpp | 107 +++++++++++++++++++-------------------- 1 file changed, 53 insertions(+), 54 deletions(-) diff --git a/src/Rtsp/RtspSession.cpp b/src/Rtsp/RtspSession.cpp index 4c413a6b..a8d3c8ac 100644 --- a/src/Rtsp/RtspSession.cpp +++ b/src/Rtsp/RtspSession.cpp @@ -202,81 +202,57 @@ void RtspSession::handleReq_ANNOUNCE(const Parser &parser) { _media_info._vhost, _media_info._app, _media_info._streamid)); - if(src){ + if (src) { sendRtspResponse("406 Not Acceptable", {"Content-Type", "text/plain"}, "Already publishing."); string err = StrPrinter << "ANNOUNCE:" << "Already publishing:" << _media_info._vhost << " " << _media_info._app << " " << _media_info._streamid << endl; - throw SockException(Err_shutdown,err); + throw SockException(Err_shutdown, err); } auto full_url = parser.FullUrl(); - if(end_with(full_url,".sdp")){ + if (end_with(full_url, ".sdp")) { //去除.sdp后缀，防止EasyDarwin推流器强制添加.sdp后缀 - full_url = full_url.substr(0,full_url.length() - 4); + full_url = full_url.substr(0, full_url.length() - 4); _media_info.parse(full_url); } - if(_media_info._app.empty() || _media_info._streamid.empty()){ + if (_media_info._app.empty() || _media_info._streamid.empty()) { //推流rtsp url必须最少两级(rtsp://host/app/stream_id)，不允许莫名其妙的推流url static constexpr auto err = "rtsp推流url非法,最少确保两级rtsp url"; sendRtspResponse("403 Forbidden", {"Content-Type", "text/plain"}, err); throw SockException(Err_shutdown, StrPrinter << err << ":" << full_url); } - SdpParser sdpParser(parser.Content()); - _sessionid = makeRandStr(12); - _sdp_track = sdpParser.getAvailableTrack(); - if (_sdp_track.empty()) { - //sdp无效 - static constexpr auto err = "sdp中无有效track"; - sendRtspResponse("403 Forbidden", {"Content-Type", "text/plain"}, err); - throw SockException(Err_shutdown,StrPrinter << err << ":" << full_url); - } - _rtcp_context.clear(); - for (auto &track : _sdp_track) { - _rtcp_context.emplace_back(std::make_shared(track->_samplerate, true)); - } - _push_src = std::make_shared(_media_info._vhost, _media_info._app, _media_info._streamid); - _push_src->setListener(dynamic_pointer_cast(shared_from_this())); - _push_src->setSdp(sdpParser.toString()); - sendRtspResponse("200 OK",{"Content-Base", _content_base + "/"}); -} - -void RtspSession::handleReq_RECORD(const Parser &parser){ - if (_sdp_track.empty() || parser["Session"] != _sessionid) { - send_SessionNotFound(); - throw SockException(Err_shutdown, _sdp_track.empty() ? "can not find any availabe track when record" : "session not found when record"); - } - auto onRes = [this](const string &err, bool enableHls, bool enableMP4){ + auto onRes = [this, parser, full_url](const string &err, bool enableHls, bool enableMP4){ bool authSuccess = err.empty(); - if(!authSuccess){ + if (!authSuccess) { sendRtspResponse("401 Unauthorized", {"Content-Type", "text/plain"}, err); - shutdown(SockException(Err_shutdown,StrPrinter << "401 Unauthorized:" << err)); + shutdown(SockException(Err_shutdown, StrPrinter << "401 Unauthorized:" << err)); return; } - //设置转协议 + SdpParser sdpParser(parser.Content()); + _sessionid = makeRandStr(12); + _sdp_track = sdpParser.getAvailableTrack(); + if (_sdp_track.empty()) { + //sdp无效 + static constexpr auto err = "sdp中无有效track"; + sendRtspResponse("403 Forbidden", {"Content-Type", "text/plain"}, err); + shutdown(SockException(Err_shutdown, StrPrinter << err << ":" << full_url)); + return; + } + _rtcp_context.clear(); + for (auto &track : _sdp_track) { + _rtcp_context.emplace_back(std::make_shared(track->_samplerate, true)); + } + _push_src = std::make_shared(_media_info._vhost, _media_info._app, _media_info._streamid); + _push_src->setListener(dynamic_pointer_cast(shared_from_this())); _push_src->setProtocolTranslation(enableHls, enableMP4); - - _StrPrinter rtp_info; - for(auto &track : _sdp_track){ - if (track->_inited == false) { - //还有track没有setup - shutdown(SockException(Err_shutdown,"track not setuped")); - return; - } - rtp_info << "url=" << track->getControlUrl(_content_base) << ","; - } - - rtp_info.pop_back(); - sendRtspResponse("200 OK", {"RTP-Info",rtp_info}); - if(_rtp_type == Rtsp::RTP_TCP){ - //如果是rtsp推流服务器，并且是TCP推流，设置socket flags,，这样能提升接收性能 - setSocketFlags(); - } + _push_src->setSdp(sdpParser.toString()); + sendRtspResponse("200 OK", {"Content-Base", _content_base + "/"}); }; weak_ptr weakSelf = dynamic_pointer_cast(shared_from_this()); @@ -296,11 +272,34 @@ void RtspSession::handleReq_RECORD(const Parser &parser){ //rtsp推流需要鉴权 auto flag = NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastMediaPublish, _media_info, invoker, static_cast(*this)); - if(!flag){ + if (!flag) { //该事件无人监听,默认不鉴权 - GET_CONFIG(bool,toHls,General::kPublishToHls); - GET_CONFIG(bool,toMP4,General::kPublishToMP4); - onRes("",toHls,toMP4); + GET_CONFIG(bool, toHls, General::kPublishToHls); + GET_CONFIG(bool, toMP4, General::kPublishToMP4); + onRes("", toHls, toMP4); + } +} + +void RtspSession::handleReq_RECORD(const Parser &parser){ + if (_sdp_track.empty() || parser["Session"] != _sessionid) { + send_SessionNotFound(); + throw SockException(Err_shutdown, _sdp_track.empty() ? "can not find any availabe track when record" : "session not found when record"); + } + + _StrPrinter rtp_info; + for (auto &track : _sdp_track) { + if (track->_inited == false) { + //还有track没有setup + shutdown(SockException(Err_shutdown, "track not setuped")); + return; + } + rtp_info << "url=" << track->getControlUrl(_content_base) << ","; + } + rtp_info.pop_back(); + sendRtspResponse("200 OK", {"RTP-Info", rtp_info}); + if (_rtp_type == Rtsp::RTP_TCP) { + //如果是rtsp推流服务器，并且是TCP推流，设置socket flags,，这样能提升接收性能 + setSocketFlags(); } } From 259e9dab0751697a6840072e1528ec06afe6097e Mon Sep 17 00:00:00 2001 From: xiongziliang <771730766@qq.com> Date: Sat, 15 May 2021 09:59:36 +0800 Subject: [PATCH 2/2] =?UTF-8?q?=E6=9B=B4=E6=96=B0ZLToolKit?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- 3rdpart/ZLToolKit | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/3rdpart/ZLToolKit b/3rdpart/ZLToolKit index d5de1d4f..b678e474 160000 --- a/3rdpart/ZLToolKit +++ b/3rdpart/ZLToolKit @@ -1 +1 @@ -Subproject commit d5de1d4f0c86139fe8f0b6d956e45dee3b151783 +Subproject commit b678e474f8ba8c9f3e6efd9cec6e4c517e80209d