ZLMediaKit/src/Rtsp/RtspPlayer.cpp

639 lines
20 KiB
C++
Raw Normal View History

2017-10-09 22:11:01 +08:00
/*
2017-09-27 16:20:30 +08:00
* MIT License
*
* Copyright (c) 2016 xiongziliang <771730766@qq.com>
* Copyright (c) 2018 huohuo <913481084@qq.com>
2017-09-27 16:20:30 +08:00
*
* This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
2017-04-01 16:35:56 +08:00
#include <set>
#include <cmath>
#include <stdarg.h>
#include <algorithm>
2018-12-17 13:48:19 +08:00
#include <iomanip>
2017-04-01 16:35:56 +08:00
2017-05-02 17:15:12 +08:00
#include "Common/config.h"
2017-04-25 11:35:41 +08:00
#include "RtspPlayer.h"
2017-04-01 16:35:56 +08:00
#include "H264/SPSParser.h"
#include "Util/MD5.h"
2017-04-25 11:35:41 +08:00
#include "Util/mini.h"
2017-08-09 18:39:30 +08:00
#include "Util/util.h"
2018-09-20 15:43:49 +08:00
#include "Util/base64.h"
2017-04-01 16:35:56 +08:00
#include "Network/sockutil.h"
2018-10-24 17:17:55 +08:00
using namespace toolkit;
2019-03-27 18:56:49 +08:00
using namespace mediakit::Client;
2017-04-01 16:35:56 +08:00
2018-10-24 17:17:55 +08:00
namespace mediakit {
2017-04-01 16:35:56 +08:00
RtspPlayer::RtspPlayer(const EventPoller::Ptr &poller) : TcpClient(poller){
2018-12-14 17:46:12 +08:00
RtpReceiver::setPoolSize(64);
2017-04-01 16:35:56 +08:00
}
RtspPlayer::~RtspPlayer(void) {
DebugL<<endl;
}
void RtspPlayer::teardown(){
if (alive()) {
2018-10-24 15:43:52 +08:00
sendRtspRequest("TEARDOWN" ,_strContentBase);
2017-04-01 16:35:56 +08:00
shutdown();
}
2019-03-27 18:41:52 +08:00
_rtspMd5Nonce.clear();
_rtspRealm.clear();
2018-12-14 17:46:12 +08:00
2018-10-26 09:56:29 +08:00
_aTrackInfo.clear();
2018-10-24 15:43:52 +08:00
_strSession.clear();
_strContentBase.clear();
2018-12-14 17:46:12 +08:00
RtpReceiver::clear();
2018-10-24 15:43:52 +08:00
CLEAR_ARR(_apUdpSock);
CLEAR_ARR(_aui16FirstSeq)
CLEAR_ARR(_aui64RtpRecv)
CLEAR_ARR(_aui64RtpRecv)
CLEAR_ARR(_aui16NowSeq)
2018-12-14 17:46:12 +08:00
CLEAR_ARR(_aiFistStamp);
CLEAR_ARR(_aiNowStamp);
2018-10-24 15:43:52 +08:00
_pBeatTimer.reset();
_pPlayTimer.reset();
_pRtpTimer.reset();
2018-10-26 14:12:16 +08:00
_iSeekTo = 0;
2018-12-14 17:46:12 +08:00
_uiCseq = 1;
_onHandshake = nullptr;
2017-04-01 16:35:56 +08:00
}
2017-08-03 13:55:46 +08:00
2019-03-27 18:41:52 +08:00
void RtspPlayer::play(const string &strUrl){
auto userAndPwd = FindField(strUrl.data(),"://","@");
Rtsp::eRtpType eType = (Rtsp::eRtpType)(int)(*this)[kRtpType];
2017-08-03 13:55:46 +08:00
if(userAndPwd.empty()){
2019-03-27 18:41:52 +08:00
play(strUrl,"","",eType);
2017-08-03 13:55:46 +08:00
return;
}
2019-03-27 18:41:52 +08:00
auto suffix = FindField(strUrl.data(),"@",nullptr);
2017-08-03 13:55:46 +08:00
auto url = StrPrinter << "rtsp://" << suffix << endl;
if(userAndPwd.find(":") == string::npos){
2019-03-27 18:41:52 +08:00
play(url,userAndPwd,"",eType);
2017-08-03 13:55:46 +08:00
return;
}
auto user = FindField(userAndPwd.data(),nullptr,":");
auto pwd = FindField(userAndPwd.data(),":",nullptr);
2019-03-27 18:41:52 +08:00
play(url,user,pwd,eType);
2017-08-03 13:55:46 +08:00
}
2017-04-01 16:35:56 +08:00
//播放指定是否走rtp over tcp
2019-03-27 18:41:52 +08:00
void RtspPlayer::play(const string &strUrl, const string &strUser, const string &strPwd, Rtsp::eRtpType eType ) {
2017-08-03 13:55:46 +08:00
DebugL << strUrl << " "
2019-03-27 18:41:52 +08:00
<< (strUser.size() ? strUser : "null") << " "
<< (strPwd.size() ? strPwd:"null") << " "
2017-08-03 13:55:46 +08:00
<< eType;
2017-04-01 16:35:56 +08:00
teardown();
2019-03-27 18:41:52 +08:00
if(strUser.size()){
(*this)[kRtspUser] = strUser;
}
2019-03-27 18:41:52 +08:00
if(strPwd.size()){
(*this)[kRtspPwd] = strPwd;
(*this)[kRtspPwdIsMD5] = false;
2017-04-01 16:35:56 +08:00
}
2018-10-24 15:43:52 +08:00
_eType = eType;
2018-10-30 10:31:27 +08:00
2019-03-27 18:41:52 +08:00
auto ip = FindField(strUrl.data(), "://", "/");
2017-04-01 16:35:56 +08:00
if (!ip.size()) {
2019-03-27 18:41:52 +08:00
ip = FindField(strUrl.data(), "://", NULL);
2017-04-01 16:35:56 +08:00
}
2019-03-27 18:41:52 +08:00
auto port = atoi(FindField(ip.data(), ":", NULL).data());
2017-04-01 16:35:56 +08:00
if (port <= 0) {
//rtsp 默认端口554
port = 554;
} else {
//服务器域名
2019-03-27 18:41:52 +08:00
ip = FindField(ip.data(), NULL, ":");
2017-04-01 16:35:56 +08:00
}
2018-10-24 15:43:52 +08:00
_strUrl = strUrl;
2017-04-01 16:35:56 +08:00
weak_ptr<RtspPlayer> weakSelf = dynamic_pointer_cast<RtspPlayer>(shared_from_this());
2019-03-27 18:56:49 +08:00
float playTimeOutSec = (*this)[kTimeoutMS].as<int>() / 1000.0;
_pPlayTimer.reset( new Timer(playTimeOutSec, [weakSelf]() {
2017-04-01 16:35:56 +08:00
auto strongSelf=weakSelf.lock();
if(!strongSelf) {
return false;
}
strongSelf->onPlayResult_l(SockException(Err_timeout,"play rtsp timeout"));
2017-04-01 16:35:56 +08:00
return false;
2019-01-30 17:00:28 +08:00
},getPoller()));
2019-03-27 18:56:49 +08:00
if(!(*this)[kNetAdapter].empty()){
setNetAdapter((*this)[kNetAdapter]);
2019-03-27 18:41:52 +08:00
}
startConnect(ip, port , playTimeOutSec);
}
void RtspPlayer::onConnect(const SockException &err){
if(err.getErrCode()!=Err_success) {
onPlayResult_l(err);
return;
}
sendDescribe();
2017-04-01 16:35:56 +08:00
}
2018-02-23 15:36:51 +08:00
void RtspPlayer::onRecv(const Buffer::Ptr& pBuf) {
2018-10-30 10:31:27 +08:00
input(pBuf->data(),pBuf->size());
2017-04-01 16:35:56 +08:00
}
void RtspPlayer::onErr(const SockException &ex) {
onPlayResult_l(ex);
2017-04-01 16:35:56 +08:00
}
// from live555
bool RtspPlayer::handleAuthenticationFailure(const string &paramsStr) {
2019-03-27 18:41:52 +08:00
if(!_rtspRealm.empty()){
//已经认证过了
return false;
}
2018-07-05 22:05:09 +08:00
char *realm = new char[paramsStr.size()];
char *nonce = new char[paramsStr.size()];
char *stale = new char[paramsStr.size()];
onceToken token(nullptr,[&](){
delete[] realm;
delete[] nonce;
delete[] stale;
});
if (sscanf(paramsStr.data(), "Digest realm=\"%[^\"]\", nonce=\"%[^\"]\", stale=%[a-zA-Z]", realm, nonce, stale) == 3) {
2019-03-27 18:41:52 +08:00
_rtspRealm = (const char *)realm;
_rtspMd5Nonce = (const char *)nonce;
return true;
}
if (sscanf(paramsStr.data(), "Digest realm=\"%[^\"]\", nonce=\"%[^\"]\"", realm, nonce) == 2) {
2019-03-27 18:41:52 +08:00
_rtspRealm = (const char *)realm;
_rtspMd5Nonce = (const char *)nonce;
return true;
}
if (sscanf(paramsStr.data(), "Basic realm=\"%[^\"]\"", realm) == 1) {
2019-03-27 18:41:52 +08:00
_rtspRealm = (const char *)realm;
return true;
}
return false;
}
void RtspPlayer::handleResDESCRIBE(const Parser& parser) {
string authInfo = parser["WWW-Authenticate"];
2017-04-01 16:35:56 +08:00
//发送DESCRIBE命令后的回复
if ((parser.Url() == "401") && handleAuthenticationFailure(authInfo)) {
sendDescribe();
return;
}
2019-01-18 11:27:51 +08:00
if(parser.Url() == "302"){
auto newUrl = parser["Location"];
if(newUrl.empty()){
throw std::runtime_error("未找到Location字段(跳转url)");
}
2019-03-27 18:41:52 +08:00
play(newUrl);
2019-01-18 11:27:51 +08:00
return;
}
2017-04-01 16:35:56 +08:00
if (parser.Url() != "200") {
throw std::runtime_error(
StrPrinter << "DESCRIBE:" << parser.Url() << " " << parser.Tail() << endl);
}
2018-10-24 15:43:52 +08:00
_strContentBase = parser["Content-Base"];
2018-10-24 15:43:52 +08:00
if(_strContentBase.empty()){
_strContentBase = _strUrl;
}
2018-10-24 15:43:52 +08:00
if (_strContentBase.back() == '/') {
_strContentBase.pop_back();
}
2017-04-01 16:35:56 +08:00
//解析sdp
2018-10-30 10:31:27 +08:00
_sdpAttr.load(parser.Content());
2018-10-26 09:56:29 +08:00
_aTrackInfo = _sdpAttr.getAvailableTrack();
if (_aTrackInfo.empty()) {
2018-10-26 10:12:37 +08:00
throw std::runtime_error("无有效的Sdp Track");
2017-04-01 16:35:56 +08:00
}
2018-10-30 10:31:27 +08:00
if (!onCheckSDP(parser.Content(), _sdpAttr)) {
2017-04-01 16:35:56 +08:00
throw std::runtime_error("onCheckSDP faied");
}
2018-10-26 09:56:29 +08:00
2017-04-01 16:35:56 +08:00
sendSetup(0);
}
//发送SETUP命令
void RtspPlayer::sendSetup(unsigned int trackIndex) {
2018-10-24 15:43:52 +08:00
_onHandshake = std::bind(&RtspPlayer::handleResSETUP,this, placeholders::_1,trackIndex);
auto &track = _aTrackInfo[trackIndex];
2018-10-26 09:56:29 +08:00
auto baseUrl = _strContentBase + "/" + track->_control_surffix;
2018-10-24 15:43:52 +08:00
switch (_eType) {
2019-03-27 18:41:52 +08:00
case Rtsp::RTP_TCP: {
sendRtspRequest("SETUP",baseUrl,{"Transport",StrPrinter << "RTP/AVP/TCP;unicast;interleaved=" << track->_type * 2 << "-" << track->_type * 2 + 1});
2017-04-01 16:35:56 +08:00
}
break;
2019-03-27 18:41:52 +08:00
case Rtsp::RTP_MULTICAST: {
sendRtspRequest("SETUP",baseUrl,{"Transport","Transport: RTP/AVP;multicast"});
}
break;
2019-03-27 18:41:52 +08:00
case Rtsp::RTP_UDP: {
2018-10-24 15:43:52 +08:00
_apUdpSock[trackIndex].reset(new Socket());
if (!_apUdpSock[trackIndex]->bindUdpSock(0, get_local_ip().data())) {
_apUdpSock[trackIndex].reset();
throw std::runtime_error("open udp sock err");
}
2018-10-24 15:43:52 +08:00
int port = _apUdpSock[trackIndex]->get_local_port();
sendRtspRequest("SETUP",baseUrl,{"Transport",StrPrinter << "RTP/AVP;unicast;client_port=" << port << "-" << port + 1});
}
break;
default:
break;
2017-04-01 16:35:56 +08:00
}
}
void RtspPlayer::handleResSETUP(const Parser &parser, unsigned int uiTrackIndex) {
2017-04-01 16:35:56 +08:00
if (parser.Url() != "200") {
throw std::runtime_error(
StrPrinter << "SETUP:" << parser.Url() << " " << parser.Tail() << endl);
}
if (uiTrackIndex == 0) {
2018-10-24 15:43:52 +08:00
_strSession = parser["Session"];
_strSession.append(";");
_strSession = FindField(_strSession.data(), nullptr, ";");
2017-04-01 16:35:56 +08:00
}
auto strTransport = parser["Transport"];
if(strTransport.find("TCP") != string::npos){
2019-03-27 18:41:52 +08:00
_eType = Rtsp::RTP_TCP;
2017-04-01 16:35:56 +08:00
}else if(strTransport.find("multicast") != string::npos){
2019-03-27 18:41:52 +08:00
_eType = Rtsp::RTP_MULTICAST;
2017-04-01 16:35:56 +08:00
}else{
2019-03-27 18:41:52 +08:00
_eType = Rtsp::RTP_UDP;
2017-04-01 16:35:56 +08:00
}
2019-03-27 18:41:52 +08:00
RtspSplitter::enableRecvRtp(_eType == Rtsp::RTP_TCP);
2018-10-30 10:31:27 +08:00
2019-03-27 18:41:52 +08:00
if(_eType == Rtsp::RTP_TCP) {
string interleaved = FindField( FindField((strTransport + ";").data(), "interleaved=", ";").data(), NULL, "-");
_aTrackInfo[uiTrackIndex]->_interleaved = atoi(interleaved.data());
2017-04-01 16:35:56 +08:00
}else{
2019-03-27 18:41:52 +08:00
const char *strPos = (_eType == Rtsp::RTP_MULTICAST ? "port=" : "server_port=") ;
auto port_str = FindField((strTransport + ";").data(), strPos, ";");
uint16_t port = atoi(FindField(port_str.data(), NULL, "-").data());
2018-10-24 15:43:52 +08:00
auto &pUdpSockRef = _apUdpSock[uiTrackIndex];
2017-04-01 16:35:56 +08:00
if(!pUdpSockRef){
pUdpSockRef.reset(new Socket());
}
2018-12-14 17:46:12 +08:00
2019-03-27 18:41:52 +08:00
if (_eType == Rtsp::RTP_MULTICAST) {
auto multiAddr = FindField((strTransport + ";").data(), "destination=", ";");
2017-04-01 16:35:56 +08:00
if (!pUdpSockRef->bindUdpSock(port, "0.0.0.0")) {
pUdpSockRef.reset();
throw std::runtime_error("open udp sock err");
}
auto fd = pUdpSockRef->rawFD();
if (-1 == SockUtil::joinMultiAddrFilter(fd, multiAddr.data(), get_peer_ip().data(),get_local_ip().data())) {
SockUtil::joinMultiAddr(fd, multiAddr.data(),get_local_ip().data());
}
} else {
struct sockaddr_in rtpto;
rtpto.sin_port = ntohs(port);
rtpto.sin_family = AF_INET;
2019-03-27 18:41:52 +08:00
rtpto.sin_addr.s_addr = inet_addr(get_peer_ip().data());
pUdpSockRef->setSendPeerAddr((struct sockaddr *)&(rtpto));
pUdpSockRef->send("\xce\xfa\xed\xfe", 4);
2017-04-01 16:35:56 +08:00
}
}
2018-10-26 09:56:29 +08:00
if (uiTrackIndex < _aTrackInfo.size() - 1) {
2017-04-01 16:35:56 +08:00
//需要继续发送SETUP命令
sendSetup(uiTrackIndex + 1);
return;
}
2019-03-27 18:41:52 +08:00
for (unsigned int i = 0; i < _aTrackInfo.size() && _eType != Rtsp::RTP_TCP; i++) {
2018-10-24 15:43:52 +08:00
auto &pUdpSockRef = _apUdpSock[i];
2017-04-01 16:35:56 +08:00
if(!pUdpSockRef){
continue;
}
auto srcIP = inet_addr(get_peer_ip().data());
weak_ptr<RtspPlayer> weakSelf = dynamic_pointer_cast<RtspPlayer>(shared_from_this());
2018-02-23 15:36:51 +08:00
pUdpSockRef->setOnRead([srcIP,i,weakSelf](const Buffer::Ptr &buf, struct sockaddr *addr) {
2017-04-01 16:35:56 +08:00
auto strongSelf=weakSelf.lock();
if(!strongSelf) {
return;
}
if(((struct sockaddr_in *)addr)->sin_addr.s_addr != srcIP) {
2018-12-14 17:46:12 +08:00
WarnL << "收到其他地址的UDP数据:" << inet_ntoa(((struct sockaddr_in *) addr)->sin_addr);
2017-04-01 16:35:56 +08:00
return;
}
2018-12-14 17:46:12 +08:00
strongSelf->handleOneRtp(i,strongSelf->_aTrackInfo[i],(unsigned char *)buf->data(),buf->size());
2017-04-01 16:35:56 +08:00
});
}
/////////////////////////心跳/////////////////////////////////
2019-03-27 18:41:52 +08:00
if(_eType != Rtsp::RTP_TCP){
weak_ptr<RtspPlayer> weakSelf = dynamic_pointer_cast<RtspPlayer>(shared_from_this());
_pBeatTimer.reset(new Timer((*this)[kBeatIntervalMS].as<int>() / 1000.0, [weakSelf](){
auto strongSelf = weakSelf.lock();
if (!strongSelf){
return false;
}
strongSelf->sendOptions();
return true;
2019-03-27 18:41:52 +08:00
},getPoller()));
}
2017-04-01 16:35:56 +08:00
pause(false);
}
void RtspPlayer::sendOptions() {
2018-10-24 15:43:52 +08:00
_onHandshake = [](const Parser& parser){
2018-10-30 10:31:27 +08:00
// DebugL << "options response";
2017-04-01 16:35:56 +08:00
};
sendRtspRequest("OPTIONS",_strContentBase);
2017-04-01 16:35:56 +08:00
}
void RtspPlayer::sendDescribe() {
//发送DESCRIBE命令后处理函数:handleResDESCRIBE
2018-10-24 15:43:52 +08:00
_onHandshake = std::bind(&RtspPlayer::handleResDESCRIBE,this, placeholders::_1);
sendRtspRequest("DESCRIBE",_strUrl,{"Accept","application/sdp"});
}
void RtspPlayer::sendPause(bool bPause,uint32_t seekMS){
2017-04-01 16:35:56 +08:00
if(!bPause){
//修改时间轴
2018-10-26 14:48:09 +08:00
int iTimeInc = seekMS - getProgressMilliSecond();
2018-10-26 09:56:29 +08:00
for(unsigned int i = 0 ;i < _aTrackInfo.size() ;i++){
2018-10-26 14:12:16 +08:00
_aiFistStamp[i] = _aiNowStamp[i] + iTimeInc;
_aiNowStamp[i] = _aiFistStamp[i];
2017-04-01 16:35:56 +08:00
}
2018-10-26 14:12:16 +08:00
_iSeekTo = seekMS;
2017-04-01 16:35:56 +08:00
}
//开启或暂停rtsp
2018-10-24 15:43:52 +08:00
_onHandshake = std::bind(&RtspPlayer::handleResPAUSE,this, placeholders::_1,bPause);
sendRtspRequest(bPause ? "PAUSE" : "PLAY",
_strContentBase,
{"Range",StrPrinter << "npt=" << setiosflags(ios::fixed) << setprecision(2) << seekMS / 1000.0 << "-"});
2017-04-01 16:35:56 +08:00
}
void RtspPlayer::pause(bool bPause) {
2018-10-26 14:12:16 +08:00
sendPause(bPause, getProgressMilliSecond());
2017-04-01 16:35:56 +08:00
}
void RtspPlayer::handleResPAUSE(const Parser& parser, bool bPause) {
2017-04-01 16:35:56 +08:00
if (parser.Url() != "200") {
WarnL <<(bPause ? "Pause" : "Play") << " failed:" << parser.Url() << " " << parser.Tail() << endl;
return;
}
if (!bPause) {
//修正时间轴
auto strRange = parser["Range"];
if (strRange.size()) {
auto strStart = FindField(strRange.data(), "npt=", "-");
if (strStart == "now") {
strStart = "0";
}
2018-10-26 14:12:16 +08:00
_iSeekTo = 1000 * atof(strStart.data());
DebugL << "seekTo(ms):" << _iSeekTo ;
2017-04-01 16:35:56 +08:00
}
auto strRtpInfo = parser["RTP-Info"];
if (strRtpInfo.size()) {
strRtpInfo.append(",");
vector<string> vec = split(strRtpInfo, ",");
2017-04-01 16:35:56 +08:00
for(auto &strTrack : vec){
strTrack.append(";");
2018-05-18 18:48:17 +08:00
auto strControlSuffix = strTrack.substr(1 + strTrack.rfind('/'),strTrack.find(';') - strTrack.rfind('/') - 1);
auto strRtpTime = FindField(strTrack.data(), "rtptime=", ";");
2018-10-26 14:12:16 +08:00
auto idx = getTrackIndexByControlSuffix(strControlSuffix);
2018-10-30 17:11:36 +08:00
if(idx != -1){
_aiFistStamp[idx] = atoll(strRtpTime.data()) * 1000 / _aTrackInfo[idx]->_samplerate;
_aiNowStamp[idx] = _aiFistStamp[idx];
DebugL << "rtptime(ms):" << strControlSuffix <<" " << strRtpTime;
}
2017-04-01 16:35:56 +08:00
}
}
onPlayResult_l(SockException(Err_success, "rtsp play success"));
2017-04-01 16:35:56 +08:00
} else {
2018-10-24 15:43:52 +08:00
_pRtpTimer.reset();
2017-04-01 16:35:56 +08:00
}
}
2018-10-30 10:31:27 +08:00
void RtspPlayer::onWholeRtspPacket(Parser &parser) {
try {
decltype(_onHandshake) fun;
_onHandshake.swap(fun);
if(fun){
fun(parser);
}
parser.Clear();
} catch (std::exception &err) {
SockException ex(Err_other, err.what());
onPlayResult_l(ex);
2017-04-01 16:35:56 +08:00
}
}
2018-10-30 10:31:27 +08:00
void RtspPlayer::onRtpPacket(const char *data, uint64_t len) {
int trackIdx = -1;
uint8_t interleaved = data[1];
if(interleaved %2 ==0){
trackIdx = getTrackIndexByInterleaved(interleaved);
}
if (trackIdx != -1) {
2018-12-14 17:46:12 +08:00
handleOneRtp(trackIdx,_aTrackInfo[trackIdx],(unsigned char *)data + 4, len - 4);
2018-10-30 10:31:27 +08:00
}
2017-04-01 16:35:56 +08:00
}
2018-12-14 17:46:12 +08:00
void RtspPlayer::onRtpSorted(const RtpPacket::Ptr &rtppt, int trackidx){
2017-04-01 16:35:56 +08:00
//统计丢包率
2018-10-24 15:43:52 +08:00
if (_aui16FirstSeq[trackidx] == 0 || rtppt->sequence < _aui16FirstSeq[trackidx]) {
_aui16FirstSeq[trackidx] = rtppt->sequence;
_aui64RtpRecv[trackidx] = 0;
2017-04-01 16:35:56 +08:00
}
2018-10-24 15:43:52 +08:00
_aui64RtpRecv[trackidx] ++;
_aui16NowSeq[trackidx] = rtppt->sequence;
2018-11-07 17:40:37 +08:00
_aiNowStamp[trackidx] = rtppt->timeStamp;
2018-11-15 15:37:13 +08:00
if( _aiFistStamp[trackidx] == 0){
_aiFistStamp[trackidx] = _aiNowStamp[trackidx];
}
2018-11-07 17:40:37 +08:00
rtppt->timeStamp -= _aiFistStamp[trackidx];
2018-10-24 15:43:52 +08:00
onRecvRTP_l(rtppt,_aTrackInfo[trackidx]);
2017-04-01 16:35:56 +08:00
}
2018-10-26 14:12:16 +08:00
float RtspPlayer::getPacketLossRate(TrackType type) const{
2018-10-26 11:03:53 +08:00
int iTrackIdx = getTrackIndexByTrackType(type);
2017-04-01 16:35:56 +08:00
if(iTrackIdx == -1){
uint64_t totalRecv = 0;
uint64_t totalSend = 0;
2018-10-26 09:56:29 +08:00
for (unsigned int i = 0; i < _aTrackInfo.size(); i++) {
2018-10-24 15:43:52 +08:00
totalRecv += _aui64RtpRecv[i];
totalSend += (_aui16NowSeq[i] - _aui16FirstSeq[i] + 1);
2017-04-01 16:35:56 +08:00
}
if(totalSend == 0){
return 0;
}
return 1.0 - (double)totalRecv / totalSend;
}
2018-10-24 15:43:52 +08:00
if(_aui16NowSeq[iTrackIdx] - _aui16FirstSeq[iTrackIdx] + 1 == 0){
2017-04-01 16:35:56 +08:00
return 0;
}
2018-10-24 15:43:52 +08:00
return 1.0 - (double)_aui64RtpRecv[iTrackIdx] / (_aui16NowSeq[iTrackIdx] - _aui16FirstSeq[iTrackIdx] + 1);
2017-04-01 16:35:56 +08:00
}
2018-10-26 14:12:16 +08:00
uint32_t RtspPlayer::getProgressMilliSecond() const{
uint32_t iTime[2] = {0,0};
2018-10-26 09:56:29 +08:00
for(unsigned int i = 0 ;i < _aTrackInfo.size() ;i++){
2018-10-26 14:12:16 +08:00
iTime[i] = _aiNowStamp[i] - _aiFistStamp[i];
2017-04-01 16:35:56 +08:00
}
2018-10-26 14:12:16 +08:00
return _iSeekTo + MAX(iTime[0],iTime[1]);
2017-04-01 16:35:56 +08:00
}
2018-10-26 14:12:16 +08:00
void RtspPlayer::seekToMilliSecond(uint32_t ms) {
sendPause(false,ms);
2017-04-01 16:35:56 +08:00
}
void RtspPlayer::sendRtspRequest(const string &cmd, const string &url, const std::initializer_list<string> &header) {
2018-12-17 13:48:19 +08:00
string key;
StrCaseMap header_map;
int i = 0;
for(auto &val : header){
if(++i % 2 == 0){
header_map.emplace(key,val);
}else{
key = val;
}
}
sendRtspRequest(cmd,url,header_map);
2018-12-17 13:48:19 +08:00
}
void RtspPlayer::sendRtspRequest(const string &cmd, const string &url,const StrCaseMap &header_const) {
auto header = header_const;
2018-10-24 15:43:52 +08:00
header.emplace("CSeq",StrPrinter << _uiCseq++);
2019-03-27 18:41:52 +08:00
header.emplace("User-Agent",SERVER_NAME "(build in " __DATE__ " " __TIME__ ")");
2018-10-24 15:43:52 +08:00
if(!_strSession.empty()){
header.emplace("Session",_strSession);
}
2019-03-27 18:56:49 +08:00
if(!_rtspRealm.empty() && !(*this)[kRtspUser].empty()){
2019-03-27 18:41:52 +08:00
if(!_rtspMd5Nonce.empty()){
//MD5认证
/*
response计算方法如下
RTSP客户端应该使用username + password并计算response如下:
(1)password为MD5编码,
response = md5( password:nonce:md5(public_method:url) );
(2)password为ANSI字符串,
response= md5( md5(username:realm:password):nonce:md5(public_method:url) );
*/
2019-03-27 18:56:49 +08:00
string encrypted_pwd = (*this)[kRtspPwd];
if(!(*this)[kRtspPwdIsMD5].as<bool>()){
encrypted_pwd = MD5((*this)[kRtspUser]+ ":" + _rtspRealm + ":" + encrypted_pwd).hexdigest();
}
2019-03-27 18:41:52 +08:00
auto response = MD5( encrypted_pwd + ":" + _rtspMd5Nonce + ":" + MD5(cmd + ":" + url).hexdigest()).hexdigest();
_StrPrinter printer;
printer << "Digest ";
2019-03-27 18:56:49 +08:00
printer << "username=\"" << (*this)[kRtspUser] << "\", ";
2019-03-27 18:41:52 +08:00
printer << "realm=\"" << _rtspRealm << "\", ";
printer << "nonce=\"" << _rtspMd5Nonce << "\", ";
printer << "uri=\"" << url << "\", ";
printer << "response=\"" << response << "\"";
header.emplace("Authorization",printer);
2019-03-27 18:56:49 +08:00
}else if(!(*this)[kRtspPwdIsMD5].as<bool>()){
//base64认证
2019-03-27 18:56:49 +08:00
string authStr = StrPrinter << (*this)[kRtspUser] << ":" << (*this)[kRtspPwd];
char authStrBase64[1024] = {0};
av_base64_encode(authStrBase64,sizeof(authStrBase64),(uint8_t *)authStr.data(),authStr.size());
header.emplace("Authorization",StrPrinter << "Basic " << authStrBase64 );
}
}
_StrPrinter printer;
printer << cmd << " " << url << " RTSP/1.0\r\n";
for (auto &pr : header){
printer << pr.first << ": " << pr.second << "\r\n";
}
send(printer << "\r\n");
}
2018-10-26 09:56:29 +08:00
void RtspPlayer::onRecvRTP_l(const RtpPacket::Ptr &pRtppt, const SdpTrack::Ptr &track) {
2018-10-24 15:43:52 +08:00
_rtpTicker.resetTime();
onRecvRTP(pRtppt,track);
}
void RtspPlayer::onPlayResult_l(const SockException &ex) {
WarnL << ex.getErrCode() << " " << ex.what();
if(_pPlayTimer){
//播放结果回调
_pPlayTimer.reset();
onPlayResult(ex);
if(!ex){
//播放成功
_rtpTicker.resetTime();
weak_ptr<RtspPlayer> weakSelf = dynamic_pointer_cast<RtspPlayer>(shared_from_this());
int timeoutMS = (*this)[kMediaTimeoutMS].as<int>();
_pRtpTimer.reset( new Timer(timeoutMS / 2000.0, [weakSelf,timeoutMS]() {
auto strongSelf=weakSelf.lock();
if(!strongSelf) {
return false;
}
if(strongSelf->_rtpTicker.elapsedTime()> timeoutMS) {
//recv rtp timeout!
strongSelf->onPlayResult_l(SockException(Err_timeout,"recv rtp timeout"));
return false;
}
return true;
},getPoller()));
}
2019-03-28 12:02:59 +08:00
} else if(ex){
//播放成功后异常断开回调
onShutdown(ex);
}
if(ex){
teardown();
}
}
2018-07-05 18:48:08 +08:00
int RtspPlayer::getTrackIndexByControlSuffix(const string &controlSuffix) const{
2018-10-26 09:56:29 +08:00
for (unsigned int i = 0; i < _aTrackInfo.size(); i++) {
2018-11-07 17:40:37 +08:00
auto pos = _aTrackInfo[i]->_control_surffix.find(controlSuffix);
if (pos == 0) {
return i;
}
}
return -1;
}
2018-07-05 18:48:08 +08:00
int RtspPlayer::getTrackIndexByInterleaved(int interleaved) const{
2018-10-26 09:56:29 +08:00
for (unsigned int i = 0; i < _aTrackInfo.size(); i++) {
2018-10-26 10:59:13 +08:00
if (_aTrackInfo[i]->_interleaved == interleaved) {
2018-07-05 18:48:08 +08:00
return i;
}
}
return -1;
}
int RtspPlayer::getTrackIndexByTrackType(TrackType trackType) const {
2018-10-26 09:56:29 +08:00
for (unsigned int i = 0; i < _aTrackInfo.size(); i++) {
2018-10-26 10:59:13 +08:00
if (_aTrackInfo[i]->_type == trackType) {
return i;
}
}
return -1;
}
2018-10-24 17:17:55 +08:00
} /* namespace mediakit */
2017-04-01 16:35:56 +08:00