ZLMediaKit/src/Rtsp/RtspPusher.h

103 lines
3.6 KiB
C++
Raw Normal View History

2020-04-04 20:30:09 +08:00
/*
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
*
* This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
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.
*/
2019-03-27 18:41:52 +08:00
#ifndef ZLMEDIAKIT_RTSPPUSHER_H
#define ZLMEDIAKIT_RTSPPUSHER_H
#include <string>
#include <memory>
#include "RtspMediaSource.h"
#include "Poller/Timer.h"
#include "Network/Socket.h"
#include "Network/TcpClient.h"
#include "RtspSplitter.h"
#include "Pusher/PusherBase.h"
namespace mediakit {
class RtcpContext;
class RtspPusher : public toolkit::TcpClient, public RtspSplitter, public PusherBase {
2019-03-27 18:41:52 +08:00
public:
typedef std::shared_ptr<RtspPusher> Ptr;
RtspPusher(const toolkit::EventPoller::Ptr &poller,const RtspMediaSource::Ptr &src);
2020-08-30 11:40:03 +08:00
~RtspPusher() override;
void publish(const std::string &url) override;
2019-03-27 18:41:52 +08:00
void teardown() override;
protected:
//for Tcpclient override
void onRecv(const toolkit::Buffer::Ptr &buf) override;
void onConnect(const toolkit::SockException &err) override;
void onErr(const toolkit::SockException &ex) override;
2019-03-27 18:41:52 +08:00
//RtspSplitter override
void onWholeRtspPacket(Parser &parser) override ;
2021-01-31 19:18:17 +08:00
void onRtpPacket(const char *data,size_t len) override;
virtual void onRtcpPacket(int track_idx, SdpTrack::Ptr &track, uint8_t *data, size_t len);
2020-08-30 11:40:03 +08:00
2019-03-27 18:41:52 +08:00
private:
void onPublishResult_l(const toolkit::SockException &ex, bool handshake_done);
2019-03-27 18:41:52 +08:00
2019-03-28 09:34:22 +08:00
void sendAnnounce();
2020-08-30 11:40:03 +08:00
void sendSetup(unsigned int track_idx);
2019-03-28 09:34:22 +08:00
void sendRecord();
void sendOptions();
2021-01-17 10:28:06 +08:00
void sendTeardown();
2019-03-27 18:41:52 +08:00
void handleResAnnounce(const Parser &parser);
2020-08-30 11:40:03 +08:00
void handleResSetup(const Parser &parser, unsigned int track_idx);
bool handleAuthenticationFailure(const std::string &params_str);
2019-03-27 18:41:52 +08:00
2021-01-31 19:18:17 +08:00
int getTrackIndexByInterleaved(int interleaved) const;
int getTrackIndexByTrackType(TrackType type) const;
2019-03-27 18:41:52 +08:00
2020-04-07 13:03:53 +08:00
void sendRtpPacket(const RtspMediaSource::RingDataType & pkt) ;
void sendRtspRequest(const std::string &cmd, const std::string &url ,const StrCaseMap &header = StrCaseMap(),const std::string &sdp = "" );
void sendRtspRequest(const std::string &cmd, const std::string &url ,const std::initializer_list<std::string> &header,const std::string &sdp = "");
void createUdpSockIfNecessary(int track_idx);
void setSocketFlags();
2021-01-31 19:18:17 +08:00
void updateRtcpContext(const RtpPacket::Ptr &pkt);
2020-08-30 11:40:03 +08:00
2019-03-27 18:41:52 +08:00
private:
2020-08-30 11:40:03 +08:00
unsigned int _cseq = 1;
Rtsp::eRtpType _rtp_type = Rtsp::RTP_TCP;
2019-03-27 18:41:52 +08:00
2020-08-30 11:40:03 +08:00
//rtsp鉴权相关
std::string _nonce;
std::string _realm;
std::string _url;
std::string _session_id;
std::string _content_base;
2020-08-30 11:40:03 +08:00
SdpParser _sdp_parser;
std::vector<SdpTrack::Ptr> _track_vec;
2021-01-31 19:18:17 +08:00
//RTP端口,trackid idx 为数组下标
toolkit::Socket::Ptr _rtp_sock[2];
2021-01-31 19:18:17 +08:00
//RTCP端口,trackid idx 为数组下标
toolkit::Socket::Ptr _rtcp_sock[2];
2019-03-27 18:41:52 +08:00
//超时功能实现
std::shared_ptr<toolkit::Timer> _publish_timer;
2019-03-27 18:41:52 +08:00
//心跳定时器
std::shared_ptr<toolkit::Timer> _beat_timer;
2020-08-30 11:40:03 +08:00
std::weak_ptr<RtspMediaSource> _push_src;
RtspMediaSource::RingType::RingReader::Ptr _rtsp_reader;
std::function<void(const Parser&)> _on_res_func;
2021-01-31 19:18:17 +08:00
////////// rtcp ////////////////
//rtcp发送时间,trackid idx 为数组下标
toolkit::Ticker _rtcp_send_ticker[2];
2021-01-31 19:18:17 +08:00
//统计rtp并发送rtcp
std::vector<std::shared_ptr<RtcpContext>> _rtcp_context;
2019-03-27 18:41:52 +08:00
};
2021-11-09 16:46:38 +08:00
using RtspPusherImp = PusherImp<RtspPusher, PusherBase>;
2019-03-27 18:41:52 +08:00
} /* namespace mediakit */
#endif //ZLMEDIAKIT_RTSPPUSHER_H