2017-10-09 22:11:01 +08:00
|
|
|
|
/*
|
2020-04-04 20:30:09 +08:00
|
|
|
|
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
|
2017-09-27 16:20:30 +08:00
|
|
|
|
*
|
2021-01-17 18:31:50 +08:00
|
|
|
|
* This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
|
2017-09-27 16:20:30 +08:00
|
|
|
|
*
|
2020-04-04 20:30:09 +08:00
|
|
|
|
* 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.
|
2017-04-01 16:35:56 +08:00
|
|
|
|
*/
|
|
|
|
|
|
2020-08-30 10:48:34 +08:00
|
|
|
|
#ifndef SRC_RTMP_RtmpPlayer_H_
|
|
|
|
|
#define SRC_RTMP_RtmpPlayer_H_
|
2017-04-01 16:35:56 +08:00
|
|
|
|
|
2017-04-25 11:35:41 +08:00
|
|
|
|
#include <memory>
|
2017-04-01 16:35:56 +08:00
|
|
|
|
#include <string>
|
|
|
|
|
#include <functional>
|
|
|
|
|
#include "amf.h"
|
2017-04-25 11:35:41 +08:00
|
|
|
|
#include "Rtmp.h"
|
2017-04-01 16:35:56 +08:00
|
|
|
|
#include "RtmpProtocol.h"
|
2017-04-25 11:35:41 +08:00
|
|
|
|
#include "Player/PlayerBase.h"
|
|
|
|
|
#include "Util/TimeTicker.h"
|
|
|
|
|
#include "Network/Socket.h"
|
|
|
|
|
#include "Network/TcpClient.h"
|
2019-03-27 18:56:49 +08:00
|
|
|
|
|
2018-10-24 17:17:55 +08:00
|
|
|
|
namespace mediakit {
|
2020-08-30 10:48:34 +08:00
|
|
|
|
|
2019-08-30 11:17:27 +08:00
|
|
|
|
//实现了rtmp播放器协议部分的功能,及数据接收功能
|
2022-02-02 20:34:50 +08:00
|
|
|
|
class RtmpPlayer : public PlayerBase, public toolkit::TcpClient, public RtmpProtocol {
|
2017-04-01 16:35:56 +08:00
|
|
|
|
public:
|
2022-12-02 14:43:06 +08:00
|
|
|
|
using Ptr = std::shared_ptr<RtmpPlayer>;
|
2022-02-02 20:34:50 +08:00
|
|
|
|
RtmpPlayer(const toolkit::EventPoller::Ptr &poller);
|
2020-08-30 10:48:34 +08:00
|
|
|
|
~RtmpPlayer() override;
|
2017-04-01 16:35:56 +08:00
|
|
|
|
|
2022-02-02 20:34:50 +08:00
|
|
|
|
void play(const std::string &strUrl) override;
|
2020-03-20 11:51:24 +08:00
|
|
|
|
void pause(bool bPause) override;
|
2021-08-12 16:07:31 +08:00
|
|
|
|
void speed(float speed) override;
|
2020-03-20 11:51:24 +08:00
|
|
|
|
void teardown() override;
|
2020-08-30 10:48:34 +08:00
|
|
|
|
|
2017-04-01 16:35:56 +08:00
|
|
|
|
protected:
|
2020-08-30 11:40:03 +08:00
|
|
|
|
virtual bool onCheckMeta(const AMFValue &val) = 0;
|
2021-02-04 17:58:51 +08:00
|
|
|
|
virtual void onMediaData(RtmpPacket::Ptr chunk_data) = 0;
|
2020-03-20 11:51:24 +08:00
|
|
|
|
uint32_t getProgressMilliSecond() const;
|
|
|
|
|
void seekToMilliSecond(uint32_t ms);
|
2020-08-30 10:48:34 +08:00
|
|
|
|
|
2019-02-28 18:03:49 +08:00
|
|
|
|
protected:
|
2021-02-04 17:58:51 +08:00
|
|
|
|
void onMediaData_l(RtmpPacket::Ptr chunk_data);
|
2020-03-20 11:51:24 +08:00
|
|
|
|
//在获取config帧后才触发onPlayResult_l(而不是收到play命令回复),所以此时所有track都初始化完毕了
|
2022-02-02 20:34:50 +08:00
|
|
|
|
void onPlayResult_l(const toolkit::SockException &ex, bool handshake_done);
|
2017-04-01 16:35:56 +08:00
|
|
|
|
|
2020-03-20 11:51:24 +08:00
|
|
|
|
//form Tcpclient
|
2022-02-02 20:34:50 +08:00
|
|
|
|
void onRecv(const toolkit::Buffer::Ptr &buf) override;
|
|
|
|
|
void onConnect(const toolkit::SockException &err) override;
|
2023-04-28 22:03:16 +08:00
|
|
|
|
void onError(const toolkit::SockException &ex) override;
|
2020-03-20 11:51:24 +08:00
|
|
|
|
//from RtmpProtocol
|
2021-02-04 17:58:51 +08:00
|
|
|
|
void onRtmpChunk(RtmpPacket::Ptr chunk_data) override;
|
2020-08-30 11:40:03 +08:00
|
|
|
|
void onStreamDry(uint32_t stream_index) override;
|
2022-02-02 20:34:50 +08:00
|
|
|
|
void onSendRawData(toolkit::Buffer::Ptr buffer) override {
|
2020-11-01 03:41:35 +08:00
|
|
|
|
send(std::move(buffer));
|
2018-01-30 11:23:57 +08:00
|
|
|
|
}
|
2017-04-01 16:35:56 +08:00
|
|
|
|
|
2020-08-30 10:48:34 +08:00
|
|
|
|
template<typename FUNC>
|
2020-08-30 11:40:03 +08:00
|
|
|
|
void addOnResultCB(const FUNC &func) {
|
2020-08-30 10:48:34 +08:00
|
|
|
|
_map_on_result.emplace(_send_req_id, func);
|
2020-03-20 11:51:24 +08:00
|
|
|
|
}
|
2020-08-30 10:48:34 +08:00
|
|
|
|
template<typename FUNC>
|
2020-08-30 11:40:03 +08:00
|
|
|
|
void addOnStatusCB(const FUNC &func) {
|
2020-08-30 10:48:34 +08:00
|
|
|
|
_deque_on_status.emplace_back(func);
|
2020-03-20 11:51:24 +08:00
|
|
|
|
}
|
2017-04-01 16:35:56 +08:00
|
|
|
|
|
2020-03-20 11:51:24 +08:00
|
|
|
|
void onCmd_result(AMFDecoder &dec);
|
|
|
|
|
void onCmd_onStatus(AMFDecoder &dec);
|
|
|
|
|
void onCmd_onMetaData(AMFDecoder &dec);
|
2017-04-01 16:35:56 +08:00
|
|
|
|
|
2020-08-30 11:40:03 +08:00
|
|
|
|
void send_connect();
|
|
|
|
|
void send_createStream();
|
|
|
|
|
void send_play();
|
|
|
|
|
void send_pause(bool pause);
|
2020-08-30 10:48:34 +08:00
|
|
|
|
|
2019-08-01 13:12:24 +08:00
|
|
|
|
private:
|
2022-02-02 20:34:50 +08:00
|
|
|
|
std::string _app;
|
|
|
|
|
std::string _stream_id;
|
|
|
|
|
std::string _tc_url;
|
2017-04-01 16:35:56 +08:00
|
|
|
|
|
2020-08-30 10:48:34 +08:00
|
|
|
|
bool _paused = false;
|
2020-03-20 11:51:24 +08:00
|
|
|
|
bool _metadata_got = false;
|
2020-04-08 11:16:09 +08:00
|
|
|
|
//是否为性能测试模式
|
|
|
|
|
bool _benchmark_mode = false;
|
2020-08-30 10:48:34 +08:00
|
|
|
|
|
|
|
|
|
//播放进度控制
|
|
|
|
|
uint32_t _seek_ms = 0;
|
|
|
|
|
uint32_t _fist_stamp[2] = {0, 0};
|
|
|
|
|
uint32_t _now_stamp[2] = {0, 0};
|
2022-02-02 20:34:50 +08:00
|
|
|
|
toolkit::Ticker _now_stamp_ticker[2];
|
|
|
|
|
std::deque<std::function<void(AMFValue &dec)> > _deque_on_status;
|
|
|
|
|
std::unordered_map<int, std::function<void(AMFDecoder &dec)> > _map_on_result;
|
2020-08-30 10:48:34 +08:00
|
|
|
|
|
|
|
|
|
//rtmp接收超时计时器
|
2022-02-02 20:34:50 +08:00
|
|
|
|
toolkit::Ticker _rtmp_recv_ticker;
|
2020-08-30 10:48:34 +08:00
|
|
|
|
//心跳发送定时器
|
2022-02-02 20:34:50 +08:00
|
|
|
|
std::shared_ptr<toolkit::Timer> _beat_timer;
|
2020-08-30 10:48:34 +08:00
|
|
|
|
//播放超时定时器
|
2022-02-02 20:34:50 +08:00
|
|
|
|
std::shared_ptr<toolkit::Timer> _play_timer;
|
2020-08-30 10:48:34 +08:00
|
|
|
|
//rtmp接收超时定时器
|
2022-02-02 20:34:50 +08:00
|
|
|
|
std::shared_ptr<toolkit::Timer> _rtmp_recv_timer;
|
2017-04-01 16:35:56 +08:00
|
|
|
|
};
|
|
|
|
|
|
2018-10-24 17:17:55 +08:00
|
|
|
|
} /* namespace mediakit */
|
2020-08-30 10:48:34 +08:00
|
|
|
|
#endif /* SRC_RTMP_RtmpPlayer_H_ */
|