211 lines
6.7 KiB
C++
211 lines
6.7 KiB
C++
/*
|
||
* MIT License
|
||
*
|
||
* Copyright (c) 2016-2019 xiongziliang <771730766@qq.com>
|
||
*
|
||
* This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
|
||
*
|
||
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
||
* of this software and associated documentation files (the "Software"), to deal
|
||
* in the Software without restriction, including without limitation the rights
|
||
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||
* copies of the Software, and to permit persons to whom the Software is
|
||
* furnished to do so, subject to the following conditions:
|
||
*
|
||
* The above copyright notice and this permission notice shall be included in all
|
||
* copies or substantial portions of the Software.
|
||
*
|
||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||
* SOFTWARE.
|
||
*/
|
||
|
||
#include "Common/config.h"
|
||
#include "RtpReceiver.h"
|
||
|
||
#define POP_HEAD(trackidx) \
|
||
auto it = _rtp_sort_cache_map[trackidx].begin(); \
|
||
onRtpSorted(it->second, trackidx); \
|
||
_rtp_sort_cache_map[trackidx].erase(it);
|
||
|
||
#define AV_RB16(x) \
|
||
((((const uint8_t*)(x))[0] << 8) | \
|
||
((const uint8_t*)(x))[1])
|
||
|
||
#define RTP_MAX_SIZE (10 * 1024)
|
||
|
||
namespace mediakit {
|
||
|
||
RtpReceiver::RtpReceiver() {}
|
||
RtpReceiver::~RtpReceiver() {}
|
||
|
||
bool RtpReceiver::handleOneRtp(int track_index,SdpTrack::Ptr &track, unsigned char *rtp_raw_ptr, unsigned int rtp_raw_len) {
|
||
if(rtp_raw_len < 12){
|
||
WarnL << "rtp包太小:" << rtp_raw_len;
|
||
return false;
|
||
}
|
||
|
||
uint8_t padding = 0;
|
||
if (rtp_raw_ptr[0] & 0x40) {
|
||
//获取padding大小
|
||
padding = rtp_raw_ptr[rtp_raw_len - 1];
|
||
//移除padding flag
|
||
rtp_raw_ptr[0] &= ~0x40;
|
||
//移除padding字节
|
||
rtp_raw_len -= padding;
|
||
}
|
||
|
||
auto rtp_ptr = _rtp_pool.obtain();
|
||
auto &rtp = *rtp_ptr;
|
||
|
||
rtp.type = track->_type;
|
||
rtp.interleaved = 2 * track->_type;
|
||
rtp.mark = rtp_raw_ptr[1] >> 7;
|
||
rtp.PT = rtp_raw_ptr[1] & 0x7F;
|
||
|
||
//序列号,内存对齐
|
||
memcpy(&rtp.sequence, rtp_raw_ptr + 2, 2);
|
||
rtp.sequence = ntohs(rtp.sequence);
|
||
|
||
//时间戳,内存对齐
|
||
memcpy(&rtp.timeStamp, rtp_raw_ptr + 4, 4);
|
||
rtp.timeStamp = ntohl(rtp.timeStamp);
|
||
|
||
if(!track->_samplerate){
|
||
//无法把时间戳转换成毫秒
|
||
return false;
|
||
}
|
||
//时间戳转换成毫秒
|
||
rtp.timeStamp = rtp.timeStamp * 1000LL / track->_samplerate;
|
||
|
||
//ssrc,内存对齐
|
||
memcpy(&rtp.ssrc, rtp_raw_ptr + 8, 4);
|
||
rtp.ssrc = ntohl(rtp.ssrc);
|
||
|
||
if (track->_ssrc != rtp.ssrc) {
|
||
if (track->_ssrc == 0) {
|
||
//保存SSRC至track对象
|
||
track->_ssrc = rtp.ssrc;
|
||
}else{
|
||
//ssrc错误
|
||
WarnL << "ssrc错误:" << rtp.ssrc << " != " << track->_ssrc;
|
||
if (_ssrc_err_count[track_index]++ > 10) {
|
||
//ssrc切换后清除老数据
|
||
WarnL << "ssrc更换:" << track->_ssrc << " -> " << rtp.ssrc;
|
||
_rtp_sort_cache_map[track_index].clear();
|
||
track->_ssrc = rtp.ssrc;
|
||
}
|
||
return false;
|
||
}
|
||
}
|
||
|
||
//ssrc匹配正确，不匹配计数清零
|
||
_ssrc_err_count[track_index] = 0;
|
||
|
||
//获取rtp中媒体数据偏移量
|
||
rtp.offset = 12 + 4;
|
||
int csrc = rtp_raw_ptr[0] & 0x0f;
|
||
int ext = rtp_raw_ptr[0] & 0x10;
|
||
rtp.offset += 4 * csrc;
|
||
if (ext && rtp_raw_len >= rtp.offset) {
|
||
/* calculate the header extension length (stored as number of 32-bit words) */
|
||
ext = (AV_RB16(rtp_raw_ptr + rtp.offset - 2) + 1) << 2;
|
||
rtp.offset += ext;
|
||
}
|
||
|
||
if(rtp_raw_len + 4 <= rtp.offset){
|
||
WarnL << "无有效负载的rtp包:" << rtp_raw_len << " <= " << (int)rtp.offset;
|
||
return false;
|
||
}
|
||
|
||
if(rtp_raw_len > RTP_MAX_SIZE){
|
||
WarnL << "超大的rtp包:" << rtp_raw_len << " > " << RTP_MAX_SIZE;
|
||
return false;
|
||
}
|
||
|
||
//设置rtp负载长度
|
||
rtp.setCapacity(rtp_raw_len + 4);
|
||
rtp.setSize(rtp_raw_len + 4);
|
||
uint8_t *payload_ptr = (uint8_t *)rtp.data();
|
||
payload_ptr[0] = '$';
|
||
payload_ptr[1] = rtp.interleaved;
|
||
payload_ptr[2] = rtp_raw_len >> 8;
|
||
payload_ptr[3] = (rtp_raw_len & 0x00FF);
|
||
//拷贝rtp负载
|
||
memcpy(payload_ptr + 4, rtp_raw_ptr, rtp_raw_len);
|
||
//排序rtp
|
||
sortRtp(rtp_ptr,track_index);
|
||
return true;
|
||
}
|
||
|
||
void RtpReceiver::sortRtp(const RtpPacket::Ptr &rtp,int track_index){
|
||
if(rtp->sequence != _last_seq[track_index] + 1 && _last_seq[track_index] != 0){
|
||
//包乱序或丢包
|
||
_seq_ok_count[track_index] = 0;
|
||
_sort_started[track_index] = true;
|
||
if(_last_seq[track_index] > rtp->sequence && _last_seq[track_index] - rtp->sequence > 0xFF){
|
||
//sequence回环，清空所有排序缓存
|
||
while (_rtp_sort_cache_map[track_index].size()) {
|
||
POP_HEAD(track_index)
|
||
}
|
||
++_seq_cycle_count[track_index];
|
||
}
|
||
}else{
|
||
//正确序列的包
|
||
_seq_ok_count[track_index]++;
|
||
}
|
||
|
||
_last_seq[track_index] = rtp->sequence;
|
||
|
||
//开始排序缓存
|
||
if (_sort_started[track_index]) {
|
||
_rtp_sort_cache_map[track_index].emplace(rtp->sequence, rtp);
|
||
GET_CONFIG(uint32_t,clearCount,Rtp::kClearCount);
|
||
GET_CONFIG(uint32_t,maxRtpCount,Rtp::kMaxRtpCount);
|
||
if (_seq_ok_count[track_index] >= clearCount) {
|
||
//网络环境改善，需要清空排序缓存
|
||
_seq_ok_count[track_index] = 0;
|
||
_sort_started[track_index] = false;
|
||
while (_rtp_sort_cache_map[track_index].size()) {
|
||
POP_HEAD(track_index)
|
||
}
|
||
} else if (_rtp_sort_cache_map[track_index].size() >= maxRtpCount) {
|
||
//排序缓存溢出
|
||
POP_HEAD(track_index)
|
||
}
|
||
}else{
|
||
//正确序列
|
||
onRtpSorted(rtp, track_index);
|
||
}
|
||
}
|
||
|
||
void RtpReceiver::clear() {
|
||
CLEAR_ARR(_last_seq)
|
||
CLEAR_ARR(_ssrc_err_count)
|
||
CLEAR_ARR(_seq_ok_count)
|
||
CLEAR_ARR(_sort_started)
|
||
CLEAR_ARR(_seq_cycle_count)
|
||
|
||
_rtp_sort_cache_map[0].clear();
|
||
_rtp_sort_cache_map[1].clear();
|
||
}
|
||
|
||
void RtpReceiver::setPoolSize(int size) {
|
||
_rtp_pool.setSize(size);
|
||
}
|
||
|
||
int RtpReceiver::getJitterSize(int track_index){
|
||
return _rtp_sort_cache_map[track_index].size();
|
||
}
|
||
|
||
int RtpReceiver::getCycleCount(int track_index){
|
||
return _seq_cycle_count[track_index];
|
||
}
|
||
|
||
|
||
}//namespace mediakit
|