103 lines
3.3 KiB
C++
103 lines
3.3 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.
|
||
*/
|
||
|
||
#if defined(ENABLE_RTPPROXY)
|
||
#include <string.h>
|
||
#include "RtpSplitter.h"
|
||
namespace mediakit{
|
||
|
||
static const int kEHOME_OFFSET = 256;
|
||
|
||
ssize_t RtpSplitter::onRecvHeader(const char *data,size_t len){
|
||
// 忽略偏移量 [AUTO-TRANSLATED:7fbc3d5d]
|
||
// Ignore offset
|
||
data += _offset;
|
||
len -= _offset;
|
||
|
||
if (_is_ehome && len > 12 && data[12] == '\r') {
|
||
// 这是ehome,移除第12个字节 [AUTO-TRANSLATED:643b3f39]
|
||
// This is ehome, remove the 12th byte
|
||
memmove((char *) data + 1, data, 12);
|
||
data += 1;
|
||
len -= 1;
|
||
}
|
||
onRtpPacket(data, len);
|
||
return 0;
|
||
}
|
||
|
||
static bool isEhome(const char *data, size_t len){
|
||
if (len < 4) {
|
||
return false;
|
||
}
|
||
if ((data[0] == 0x01) && (data[1] == 0x00) && (data[2] >= 0x01)) {
|
||
return true;
|
||
}
|
||
return false;
|
||
}
|
||
|
||
const char *RtpSplitter::onSearchPacketTail(const char *data, size_t len) {
|
||
if (len < 4) {
|
||
// 数据不够 [AUTO-TRANSLATED:72802244]
|
||
// Not enough data
|
||
return nullptr;
|
||
}
|
||
|
||
if (_check_ehome_count) {
|
||
if (isEhome(data, len)) {
|
||
// 是ehome协议 [AUTO-TRANSLATED:daa874ca]
|
||
// It is the ehome protocol
|
||
if (len < kEHOME_OFFSET + 4) {
|
||
// 数据不够 [AUTO-TRANSLATED:72802244]
|
||
// Not enough data
|
||
return nullptr;
|
||
}
|
||
// 忽略ehome私有头后是rtsp样式的rtp，多4个字节， [AUTO-TRANSLATED:d9bc652c]
|
||
// Ignore the ehome private header, then it is an rtsp-style rtp, with 4 extra bytes,
|
||
_offset = kEHOME_OFFSET + 4;
|
||
_is_ehome = true;
|
||
// 忽略ehome私有头 [AUTO-TRANSLATED:4fc98661]
|
||
// Ignore the ehome private header
|
||
return onSearchPacketTail_l(data + kEHOME_OFFSET + 2, len - kEHOME_OFFSET - 2);
|
||
}
|
||
_check_ehome_count--;
|
||
}
|
||
|
||
if ( _is_rtsp_interleaved ) {
|
||
if (data[0] == '$') {
|
||
// 可能是4个字节的rtp头 [AUTO-TRANSLATED:c287c3cb]
|
||
// It may be a 4-byte rtp header
|
||
_offset = 4;
|
||
return onSearchPacketTail_l(data + 2, len - 2);
|
||
}
|
||
_is_rtsp_interleaved = false;
|
||
}
|
||
|
||
// 两个字节的rtp头 [AUTO-TRANSLATED:85fed462]
|
||
// A 2-byte rtp header
|
||
_offset = 2;
|
||
return onSearchPacketTail_l(data, len);
|
||
}
|
||
|
||
const char *RtpSplitter::onSearchPacketTail_l(const char *data, size_t len) {
|
||
// 这是rtp包 [AUTO-TRANSLATED:627d4881]
|
||
// This is an rtp packet
|
||
uint16_t length = (((uint8_t *) data)[0] << 8) | ((uint8_t *) data)[1];
|
||
if (len < (size_t)(length + 2)) {
|
||
// 数据不够 [AUTO-TRANSLATED:72802244]
|
||
// Not enough data
|
||
return nullptr;
|
||
}
|
||
// 返回rtp包末尾 [AUTO-TRANSLATED:2546d5ce]
|
||
// Return the end of the rtp packet
|
||
return data + 2 + length;
|
||
}
|
||
|
||
}//namespace mediakit
|
||
#endif//defined(ENABLE_RTPPROXY)
|