155 lines
4.8 KiB
C++
155 lines
4.8 KiB
C++
/*
|
||
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
|
||
*
|
||
* This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
|
||
*
|
||
* Use of this source code is governed by MIT 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.
|
||
*/
|
||
|
||
#ifndef SRC_RTSPPLAYER_RTSPPLAYER_H_TXT_
|
||
#define SRC_RTSPPLAYER_RTSPPLAYER_H_TXT_
|
||
|
||
#include <string>
|
||
#include <memory>
|
||
#include "RtspSession.h"
|
||
#include "RtspMediaSource.h"
|
||
#include "Player/PlayerBase.h"
|
||
#include "Util/util.h"
|
||
#include "Util/logger.h"
|
||
#include "Util/TimeTicker.h"
|
||
#include "Poller/Timer.h"
|
||
#include "Network/Socket.h"
|
||
#include "Network/TcpClient.h"
|
||
#include "RtspSplitter.h"
|
||
#include "RtpReceiver.h"
|
||
#include "Common/Stamp.h"
|
||
|
||
using namespace std;
|
||
using namespace toolkit;
|
||
|
||
namespace mediakit {
|
||
|
||
//实现了rtsp播放器协议部分的功能，及数据接收功能
|
||
class RtspPlayer: public PlayerBase,public TcpClient, public RtspSplitter, public RtpReceiver {
|
||
public:
|
||
typedef std::shared_ptr<RtspPlayer> Ptr;
|
||
|
||
RtspPlayer(const EventPoller::Ptr &poller);
|
||
~RtspPlayer() override;
|
||
|
||
void play(const string &strUrl) override;
|
||
void pause(bool pause_flag) override;
|
||
void teardown() override;
|
||
float getPacketLossRate(TrackType type) const override;
|
||
|
||
protected:
|
||
//派生类回调函数
|
||
virtual bool onCheckSDP(const string &sdp) = 0;
|
||
virtual void onRecvRTP(const RtpPacket::Ptr &rtp, const SdpTrack::Ptr &track) = 0;
|
||
uint32_t getProgressMilliSecond() const;
|
||
void seekToMilliSecond(uint32_t ms);
|
||
|
||
/**
|
||
* 收到完整的rtsp包回调，包括sdp等content数据
|
||
* @param parser rtsp包
|
||
*/
|
||
void onWholeRtspPacket(Parser &parser) override ;
|
||
|
||
/**
|
||
* 收到rtp包回调
|
||
* @param data
|
||
* @param len
|
||
*/
|
||
void onRtpPacket(const char *data,uint64_t len) override ;
|
||
|
||
/**
|
||
* rtp数据包排序后输出
|
||
* @param rtp rtp数据包
|
||
* @param track_idx track索引
|
||
*/
|
||
void onRtpSorted(const RtpPacket::Ptr &rtp, int track_idx) override;
|
||
|
||
/**
|
||
* 收到RTCP包回调
|
||
* @param track_idx track索引
|
||
* @param track sdp相关信息
|
||
* @param data rtcp内容
|
||
* @param len rtcp内容长度
|
||
*/
|
||
virtual void onRtcpPacket(int track_idx, SdpTrack::Ptr &track, unsigned char *data, unsigned int len);
|
||
|
||
/////////////TcpClient override/////////////
|
||
void onConnect(const SockException &err) override;
|
||
void onRecv(const Buffer::Ptr &buf) override;
|
||
void onErr(const SockException &ex) override;
|
||
|
||
private:
|
||
void onRecvRTP_l(const RtpPacket::Ptr &rtp, const SdpTrack::Ptr &track);
|
||
void onPlayResult_l(const SockException &ex , bool handshake_done);
|
||
|
||
int getTrackIndexByInterleaved(int interleaved) const;
|
||
int getTrackIndexByTrackType(TrackType track_type) const;
|
||
|
||
void handleResSETUP(const Parser &parser, unsigned int track_idx);
|
||
void handleResDESCRIBE(const Parser &parser);
|
||
bool handleAuthenticationFailure(const string &wwwAuthenticateParamsStr);
|
||
void handleResPAUSE(const Parser &parser, int type);
|
||
bool handleResponse(const string &cmd, const Parser &parser);
|
||
|
||
void sendOptions();
|
||
void sendSetup(unsigned int track_idx);
|
||
void sendPause(int type , uint32_t ms);
|
||
void sendDescribe();
|
||
void sendKeepAlive();
|
||
void sendRtspRequest(const string &cmd, const string &url ,const StrCaseMap &header = StrCaseMap());
|
||
void sendRtspRequest(const string &cmd, const string &url ,const std::initializer_list<string> &header);
|
||
void sendReceiverReport(bool over_tcp, int track_idx);
|
||
void createUdpSockIfNecessary(int track_idx);
|
||
|
||
private:
|
||
string _play_url;
|
||
vector<SdpTrack::Ptr> _sdp_track;
|
||
function<void(const Parser&)> _on_response;
|
||
//RTP端口,trackid idx 为数组下标
|
||
Socket::Ptr _rtp_sock[2];
|
||
//RTCP端口,trackid idx 为数组下标
|
||
Socket::Ptr _rtcp_sock[2];
|
||
|
||
//rtsp鉴权相关
|
||
string _md5_nonce;
|
||
string _realm;
|
||
//rtsp info
|
||
string _session_id;
|
||
uint32_t _cseq_send = 1;
|
||
string _content_base;
|
||
Rtsp::eRtpType _rtp_type = Rtsp::RTP_TCP;
|
||
|
||
/* 丢包率统计需要用到的参数 */
|
||
uint16_t _rtp_seq_start[2] = {0, 0};
|
||
uint16_t _rtp_seq_now[2] = {0, 0};
|
||
uint64_t _rtp_recv_count[2] = {0, 0};
|
||
//当前rtp时间戳
|
||
uint32_t _stamp[2] = {0, 0};
|
||
|
||
//超时功能实现
|
||
Ticker _rtp_recv_ticker;
|
||
std::shared_ptr<Timer> _play_check_timer;
|
||
std::shared_ptr<Timer> _rtp_check_timer;
|
||
|
||
//rtcp统计,trackid idx 为数组下标
|
||
RtcpCounter _rtcp_counter[2];
|
||
//rtcp发送时间,trackid idx 为数组下标
|
||
Ticker _rtcp_send_ticker[2];
|
||
|
||
//是否为性能测试模式
|
||
bool _benchmark_mode = false;
|
||
|
||
//服务器支持的命令
|
||
set<string> _supported_cmd;
|
||
};
|
||
|
||
} /* namespace mediakit */
|
||
#endif /* SRC_RTSPPLAYER_RTSPPLAYER_H_TXT_ */
|