mirror of
https://gitee.com/xia-chu/ZLMediaKit.git
synced 2026-05-07 03:17:49 +08:00
c72cf4cbcc
* feat: remove using namespace mediakit in header files. (cherry picked from commit d44aeb339a8a0e1f0455be82b21fe4b1b536299f) * feat: remove using namespace mediakit in FFmpegSource.h * feat: remove using namespace mediakit in RtpExt.h * feat: remove using namespace mediakit in header files. * feat: remove using namespace std in header files. * feat: remove using namespace std in header files when zltoolkit remove std in header * 补充命名空间 * 整理命名空间 * 整理命名空间2 * 修复macos ci * 修复编译问题 * 修复编译问题2 * 修复编译问题3 Co-authored-by: Johnny <hellojinqiang@gmail.com> Co-authored-by: Xiaofeng Wang <wasphin@gmail.com>
106 lines
3.2 KiB
C++
106 lines
3.2 KiB
C++
/*
|
||
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
|
||
*
|
||
* This file is part of ZLMediaKit(https://github.com/xia-chu/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 ZLMEDIAKIT_RTPPROCESS_H
|
||
#define ZLMEDIAKIT_RTPPROCESS_H
|
||
|
||
#if defined(ENABLE_RTPPROXY)
|
||
#include "ProcessInterface.h"
|
||
#include "Common/MultiMediaSourceMuxer.h"
|
||
|
||
namespace mediakit {
|
||
|
||
class RtpProcess : public toolkit::SockInfo, public MediaSinkInterface, public MediaSourceEventInterceptor, public std::enable_shared_from_this<RtpProcess>{
|
||
public:
|
||
typedef std::shared_ptr<RtpProcess> Ptr;
|
||
friend class RtpProcessHelper;
|
||
RtpProcess(const std::string &stream_id);
|
||
~RtpProcess();
|
||
|
||
/**
|
||
* 输入rtp
|
||
* @param is_udp 是否为udp模式
|
||
* @param sock 本地监听的socket
|
||
* @param data rtp数据指针
|
||
* @param len rtp数据长度
|
||
* @param addr 数据源地址
|
||
* @param dts_out 解析出最新的dts
|
||
* @return 是否解析成功
|
||
*/
|
||
bool inputRtp(bool is_udp, const toolkit::Socket::Ptr &sock, const char *data, size_t len, const struct sockaddr *addr , uint32_t *dts_out = nullptr);
|
||
|
||
/**
|
||
* 是否超时，用于超时移除对象
|
||
*/
|
||
bool alive();
|
||
|
||
/**
|
||
* 超时时被RtpSelector移除时触发
|
||
*/
|
||
void onDetach();
|
||
|
||
/**
|
||
* 设置onDetach事件回调
|
||
*/
|
||
void setOnDetach(const std::function<void()> &cb);
|
||
|
||
/**
|
||
* 设置onDetach事件回调,false检查RTP超时，true停止
|
||
*/
|
||
void setStopCheckRtp(bool is_check=false);
|
||
|
||
/// SockInfo override
|
||
std::string get_local_ip() override;
|
||
uint16_t get_local_port() override;
|
||
std::string get_peer_ip() override;
|
||
uint16_t get_peer_port() override;
|
||
std::string getIdentifier() const override;
|
||
|
||
int getTotalReaderCount();
|
||
void setListener(const std::weak_ptr<MediaSourceEvent> &listener);
|
||
|
||
protected:
|
||
bool inputFrame(const Frame::Ptr &frame) override;
|
||
bool addTrack(const Track::Ptr & track) override;
|
||
void addTrackCompleted() override;
|
||
void resetTracks() override {};
|
||
|
||
//// MediaSourceEvent override ////
|
||
MediaOriginType getOriginType(MediaSource &sender) const override;
|
||
std::string getOriginUrl(MediaSource &sender) const override;
|
||
std::shared_ptr<SockInfo> getOriginSock(MediaSource &sender) const override;
|
||
|
||
private:
|
||
void emitOnPublish();
|
||
void doCachedFunc();
|
||
|
||
private:
|
||
uint32_t _dts = 0;
|
||
uint64_t _total_bytes = 0;
|
||
struct sockaddr _addr{0};
|
||
toolkit::Socket::Ptr _sock;
|
||
MediaInfo _media_info;
|
||
toolkit::Ticker _last_frame_time;
|
||
std::function<void()> _on_detach;
|
||
std::shared_ptr<FILE> _save_file_rtp;
|
||
std::shared_ptr<FILE> _save_file_video;
|
||
ProcessInterface::Ptr _process;
|
||
MultiMediaSourceMuxer::Ptr _muxer;
|
||
std::atomic_bool _stop_rtp_check{false};
|
||
std::atomic_flag _busy_flag{false};
|
||
toolkit::Ticker _last_check_alive;
|
||
std::recursive_mutex _func_mtx;
|
||
std::deque<std::function<void()> > _cached_func;
|
||
};
|
||
|
||
}//namespace mediakit
|
||
#endif//defined(ENABLE_RTPPROXY)
|
||
#endif //ZLMEDIAKIT_RTPPROCESS_H
|