111 lines
4.1 KiB
C++
111 lines
4.1 KiB
C++
/*
|
|
* Copyright (c) 2016-present The ZLMediaKit project authors. All Rights Reserved.
|
|
*
|
|
* This file is part of ZLMediaKit(https://github.com/ZLMediaKit/ZLMediaKit).
|
|
*
|
|
* Use of this source code is governed by MIT-like 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 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"
|
|
#include "Rtcp/RtcpContext.h"
|
|
|
|
namespace mediakit {
|
|
|
|
class RtspPusher : public toolkit::TcpClient, public RtspSplitter, public PusherBase {
|
|
public:
|
|
using Ptr = std::shared_ptr<RtspPusher>;
|
|
RtspPusher(const toolkit::EventPoller::Ptr &poller,const RtspMediaSource::Ptr &src);
|
|
~RtspPusher() override;
|
|
void publish(const std::string &url) override;
|
|
void teardown() override;
|
|
|
|
protected:
|
|
//for Tcpclient override
|
|
void onRecv(const toolkit::Buffer::Ptr &buf) override;
|
|
void onConnect(const toolkit::SockException &err) override;
|
|
void onError(const toolkit::SockException &ex) override;
|
|
|
|
//RtspSplitter override
|
|
void onWholeRtspPacket(Parser &parser) override ;
|
|
void onRtpPacket(const char *data,size_t len) override;
|
|
|
|
virtual void onRtcpPacket(int track_idx, SdpTrack::Ptr &track, uint8_t *data, size_t len);
|
|
|
|
private:
|
|
void onPublishResult_l(const toolkit::SockException &ex, bool handshake_done);
|
|
|
|
void sendAnnounce();
|
|
void sendSetup(unsigned int track_idx);
|
|
void sendRecord();
|
|
void sendOptions();
|
|
void sendTeardown();
|
|
|
|
void handleResAnnounce(const Parser &parser);
|
|
void handleResSetup(const Parser &parser, unsigned int track_idx);
|
|
bool handleAuthenticationFailure(const std::string ¶ms_str);
|
|
|
|
int getTrackIndexByInterleaved(int interleaved) const;
|
|
int getTrackIndexByTrackType(TrackType type) const;
|
|
|
|
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();
|
|
void updateRtcpContext(const RtpPacket::Ptr &pkt);
|
|
|
|
private:
|
|
unsigned int _cseq = 1;
|
|
Rtsp::eRtpType _rtp_type = Rtsp::RTP_TCP;
|
|
|
|
// rtsp鉴权相关 [AUTO-TRANSLATED:947dc6a3]
|
|
// RTSP authentication related
|
|
std::string _nonce;
|
|
std::string _realm;
|
|
std::string _url;
|
|
std::string _session_id;
|
|
std::string _content_base;
|
|
SdpParser _sdp_parser;
|
|
std::vector<SdpTrack::Ptr> _track_vec;
|
|
// RTP端口,trackid idx 为数组下标 [AUTO-TRANSLATED:77c186bb]
|
|
// RTP port, trackid idx is the array index
|
|
toolkit::Socket::Ptr _rtp_sock[2];
|
|
// RTCP端口,trackid idx 为数组下标 [AUTO-TRANSLATED:446a7861]
|
|
// RTCP port, trackid idx is the array index
|
|
toolkit::Socket::Ptr _rtcp_sock[2];
|
|
// 超时功能实现 [AUTO-TRANSLATED:1d603b3a]
|
|
// Timeout function implementation
|
|
toolkit::Timer::Ptr _publish_timer;
|
|
// 心跳定时器 [AUTO-TRANSLATED:536ec800]
|
|
// Heartbeat timer
|
|
toolkit::Timer::Ptr _beat_timer;
|
|
std::weak_ptr<RtspMediaSource> _push_src;
|
|
RtspMediaSource::RingType::RingReader::Ptr _rtsp_reader;
|
|
std::function<void(const Parser&)> _on_res_func;
|
|
////////// rtcp ////////////////
|
|
// rtcp发送时间,trackid idx 为数组下标 [AUTO-TRANSLATED:bf3248b1]
|
|
// RTCP send time, trackid idx is the array index
|
|
toolkit::Ticker _rtcp_send_ticker[2];
|
|
// 统计rtp并发送rtcp [AUTO-TRANSLATED:0ac2b665]
|
|
// Statistics RTP and send RTCP
|
|
std::vector<RtcpContext::Ptr> _rtcp_context;
|
|
};
|
|
|
|
using RtspPusherImp = PusherImp<RtspPusher, PusherBase>;
|
|
|
|
} /* namespace mediakit */
|
|
#endif //ZLMEDIAKIT_RTSPPUSHER_H
|