mirror of
https://gitee.com/xia-chu/ZLMediaKit.git
synced 2026-05-10 12:37:48 +08:00
306 lines
8.7 KiB
C++
306 lines
8.7 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_RTPRECEIVER_H
|
||
#define ZLMEDIAKIT_RTPRECEIVER_H
|
||
|
||
#include <map>
|
||
#include <string>
|
||
#include <memory>
|
||
#include "Rtsp/Rtsp.h"
|
||
#include "Extension/Frame.h"
|
||
// for NtpStamp
|
||
#include "Common/Stamp.h"
|
||
#include "Util/TimeTicker.h"
|
||
|
||
namespace mediakit {
|
||
|
||
template<typename T, typename SEQ = uint16_t>
|
||
class PacketSortor {
|
||
public:
|
||
static constexpr SEQ SEQ_MAX = (std::numeric_limits<SEQ>::max)();
|
||
PacketSortor() = default;
|
||
~PacketSortor() = default;
|
||
|
||
void setOnSort(std::function<void(SEQ seq, T packet)> cb) { _cb = std::move(cb); }
|
||
|
||
/**
|
||
* 清空状态
|
||
*/
|
||
void clear() {
|
||
_started = false;
|
||
_seq_cycle_count = 0;
|
||
_pkt_sort_cache_map.clear();
|
||
}
|
||
|
||
/**
|
||
* 获取排序缓存长度
|
||
*/
|
||
size_t getJitterSize() const { return _pkt_sort_cache_map.size(); }
|
||
|
||
/**
|
||
* 获取seq回环次数
|
||
*/
|
||
size_t getCycleCount() const { return _seq_cycle_count; }
|
||
|
||
/**
|
||
* 输入并排序
|
||
* @param seq 序列号
|
||
* @param packet 包负载
|
||
*/
|
||
void sortPacket(SEQ seq, T packet) {
|
||
if (!_started) {
|
||
// 记录第一个seq
|
||
_started = true;
|
||
_last_seq_out = seq - 1;
|
||
}
|
||
if (seq == static_cast<SEQ>(_last_seq_out + 1)) {
|
||
// 收到下一个seq
|
||
output(seq, std::move(packet));
|
||
return;
|
||
}
|
||
|
||
if (seq < _last_seq_out && _last_seq_out != SEQ_MAX && seq < 1024 && _last_seq_out > SEQ_MAX - 1024) {
|
||
// seq回环,清空回环前缓存
|
||
flush();
|
||
_last_seq_out = SEQ_MAX;
|
||
++_seq_cycle_count;
|
||
sortPacket(seq, std::move(packet));
|
||
return;
|
||
}
|
||
|
||
if (seq <= _last_seq_out && _last_seq_out != SEQ_MAX) {
|
||
// 这个回退包已经不再等待
|
||
return;
|
||
}
|
||
|
||
_pkt_sort_cache_map.emplace(seq, std::move(packet));
|
||
auto it_min = _pkt_sort_cache_map.begin();
|
||
auto it_max = _pkt_sort_cache_map.rbegin();
|
||
if (it_max->first - it_min->first > (SEQ_MAX >> 1)) {
|
||
// 回环后，收到回环前的大值seq, 忽略掉
|
||
_pkt_sort_cache_map.erase((++it_max).base());
|
||
return;
|
||
}
|
||
|
||
tryFlushFrontPacket();
|
||
|
||
if (_pkt_sort_cache_map.size() > _max_buffer_size || (_ticker.elapsedTime() > _max_buffer_ms && !_pkt_sort_cache_map.empty())) {
|
||
// buffer太长，强行减小
|
||
WarnL << "packet dropped: " << static_cast<SEQ>(_last_seq_out + 1) << " -> "
|
||
<< static_cast<SEQ>(_pkt_sort_cache_map.begin()->first - 1)
|
||
<< ", jitter buffer size: " << _pkt_sort_cache_map.size()
|
||
<< ", jitter buffer ms: " << _ticker.elapsedTime();
|
||
popIterator(_pkt_sort_cache_map.begin());
|
||
}
|
||
}
|
||
|
||
void flush() {
|
||
// 清空缓存
|
||
while (!_pkt_sort_cache_map.empty()) {
|
||
popIterator(_pkt_sort_cache_map.begin());
|
||
}
|
||
}
|
||
|
||
private:
|
||
void tryFlushFrontPacket() {
|
||
while (!_pkt_sort_cache_map.empty()) {
|
||
auto it = _pkt_sort_cache_map.begin();
|
||
auto next_seq = static_cast<SEQ>(_last_seq_out + 1);
|
||
if (it->first < next_seq) {
|
||
_pkt_sort_cache_map.erase(it);
|
||
continue;
|
||
}
|
||
if (it->first == next_seq) {
|
||
// 连续的seq
|
||
popIterator(it);
|
||
continue;
|
||
}
|
||
break;
|
||
}
|
||
}
|
||
|
||
void popIterator(typename std::map<SEQ, T>::iterator it) {
|
||
auto seq = it->first;
|
||
auto data = std::move(it->second);
|
||
_pkt_sort_cache_map.erase(it);
|
||
output(seq, std::move(data));
|
||
}
|
||
|
||
void output(SEQ seq, T packet) {
|
||
_last_seq_out = seq;
|
||
_cb(seq, std::move(packet));
|
||
_ticker.resetTime();
|
||
}
|
||
|
||
private:
|
||
bool _started = false;
|
||
//排序缓存最大保存数据长度，单位毫秒
|
||
size_t _max_buffer_ms = 1000;
|
||
//排序缓存最大保存数据个数
|
||
size_t _max_buffer_size = 1024;
|
||
//记录上次output至今的时间
|
||
toolkit::Ticker _ticker;
|
||
//下次应该输出的SEQ
|
||
SEQ _last_seq_out = 0;
|
||
//seq回环次数计数
|
||
size_t _seq_cycle_count = 0;
|
||
//pkt排序缓存，根据seq排序
|
||
std::map<SEQ, T> _pkt_sort_cache_map;
|
||
//回调
|
||
std::function<void(SEQ seq, T packet)> _cb;
|
||
};
|
||
|
||
class RtpTrack : private PacketSortor<RtpPacket::Ptr> {
|
||
public:
|
||
class BadRtpException : public std::invalid_argument {
|
||
public:
|
||
template<typename Type>
|
||
BadRtpException(Type &&type) : invalid_argument(std::forward<Type>(type)) {}
|
||
~BadRtpException() = default;
|
||
};
|
||
|
||
RtpTrack();
|
||
virtual ~RtpTrack() = default;
|
||
|
||
void clear();
|
||
uint32_t getSSRC() const;
|
||
RtpPacket::Ptr inputRtp(TrackType type, int sample_rate, uint8_t *ptr, size_t len);
|
||
void setNtpStamp(uint32_t rtp_stamp, uint64_t ntp_stamp_ms);
|
||
void setPayloadType(uint8_t pt);
|
||
|
||
protected:
|
||
virtual void onRtpSorted(RtpPacket::Ptr rtp) {}
|
||
virtual void onBeforeRtpSorted(const RtpPacket::Ptr &rtp) {}
|
||
|
||
private:
|
||
bool _disable_ntp = false;
|
||
uint8_t _pt = 0xFF;
|
||
uint32_t _ssrc = 0;
|
||
toolkit::Ticker _ssrc_alive;
|
||
NtpStamp _ntp_stamp;
|
||
};
|
||
|
||
class RtpTrackImp : public RtpTrack{
|
||
public:
|
||
using OnSorted = std::function<void(RtpPacket::Ptr)>;
|
||
using BeforeSorted = std::function<void(const RtpPacket::Ptr &)>;
|
||
|
||
RtpTrackImp() = default;
|
||
~RtpTrackImp() override = default;
|
||
|
||
void setOnSorted(OnSorted cb);
|
||
void setBeforeSorted(BeforeSorted cb);
|
||
|
||
protected:
|
||
void onRtpSorted(RtpPacket::Ptr rtp) override;
|
||
void onBeforeRtpSorted(const RtpPacket::Ptr &rtp) override;
|
||
|
||
private:
|
||
OnSorted _on_sorted;
|
||
BeforeSorted _on_before_sorted;
|
||
};
|
||
|
||
template<int kCount = 2>
|
||
class RtpMultiReceiver {
|
||
public:
|
||
RtpMultiReceiver() {
|
||
int index = 0;
|
||
for (auto &track : _track) {
|
||
track.setOnSorted([this, index](RtpPacket::Ptr rtp) {
|
||
onRtpSorted(std::move(rtp), index);
|
||
});
|
||
track.setBeforeSorted([this, index](const RtpPacket::Ptr &rtp) {
|
||
onBeforeRtpSorted(rtp, index);
|
||
});
|
||
++index;
|
||
}
|
||
}
|
||
|
||
virtual ~RtpMultiReceiver() = default;
|
||
|
||
/**
|
||
* 输入数据指针生成并排序rtp包
|
||
* @param index track下标索引
|
||
* @param type track类型
|
||
* @param samplerate rtp时间戳基准时钟，视频为90000，音频为采样率
|
||
* @param ptr rtp数据指针
|
||
* @param len rtp数据指针长度
|
||
* @return 解析成功返回true
|
||
*/
|
||
bool handleOneRtp(int index, TrackType type, int sample_rate, uint8_t *ptr, size_t len) {
|
||
assert(index < kCount && index >= 0);
|
||
return _track[index].inputRtp(type, sample_rate, ptr, len).operator bool();
|
||
}
|
||
|
||
/**
|
||
* 设置ntp时间戳，在收到rtcp sender report时设置
|
||
* 如果rtp_stamp/sample_rate/ntp_stamp_ms都为0，那么采用rtp时间戳为ntp时间戳
|
||
* @param index track下标索引
|
||
* @param rtp_stamp rtp时间戳
|
||
* @param ntp_stamp_ms ntp时间戳
|
||
*/
|
||
void setNtpStamp(int index, uint32_t rtp_stamp, uint64_t ntp_stamp_ms) {
|
||
assert(index < kCount && index >= 0);
|
||
_track[index].setNtpStamp(rtp_stamp, ntp_stamp_ms);
|
||
}
|
||
|
||
void setPayloadType(int index, uint8_t pt){
|
||
assert(index < kCount && index >= 0);
|
||
_track[index].setPayloadType(pt);
|
||
}
|
||
|
||
void clear() {
|
||
for (auto &track : _track) {
|
||
track.clear();
|
||
}
|
||
}
|
||
|
||
size_t getJitterSize(int index) const {
|
||
assert(index < kCount && index >= 0);
|
||
return _track[index].getJitterSize();
|
||
}
|
||
|
||
size_t getCycleCount(int index) const {
|
||
assert(index < kCount && index >= 0);
|
||
return _track[index].getCycleCount();
|
||
}
|
||
|
||
uint32_t getSSRC(int index) const {
|
||
assert(index < kCount && index >= 0);
|
||
return _track[index].getSSRC();
|
||
}
|
||
|
||
protected:
|
||
/**
|
||
* rtp数据包排序后输出
|
||
* @param rtp rtp数据包
|
||
* @param track_index track索引
|
||
*/
|
||
virtual void onRtpSorted(RtpPacket::Ptr rtp, int index) {}
|
||
|
||
/**
|
||
* 解析出rtp但还未排序
|
||
* @param rtp rtp数据包
|
||
* @param track_index track索引
|
||
*/
|
||
virtual void onBeforeRtpSorted(const RtpPacket::Ptr &rtp, int index) {}
|
||
|
||
private:
|
||
RtpTrackImp _track[kCount];
|
||
};
|
||
|
||
using RtpReceiver = RtpMultiReceiver<2>;
|
||
|
||
}//namespace mediakit
|
||
|
||
|
||
#endif //ZLMEDIAKIT_RTPRECEIVER_H
|