mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2026-06-21 16:32:22 +08:00
first commit for srt intergrate
This commit is contained in:
127
srt/HSExt.cpp
Normal file
127
srt/HSExt.cpp
Normal file
@@ -0,0 +1,127 @@
|
||||
#include "HSExt.hpp"
|
||||
|
||||
namespace SRT {
|
||||
bool HSExtMessage::loadFromData(uint8_t *buf, size_t len) {
|
||||
if(buf == NULL || len != HSEXT_MSG_SIZE){
|
||||
return false;
|
||||
}
|
||||
|
||||
_data = BufferRaw::create();
|
||||
_data->assign((char*)buf,len);
|
||||
extension_length = 3;
|
||||
HSExt::loadHeader();
|
||||
|
||||
assert(extension_type == SRT_CMD_HSREQ || extension_type == SRT_CMD_HSRSP);
|
||||
|
||||
uint8_t* ptr = (uint8_t*)_data->data()+4;
|
||||
srt_version = loadUint32(ptr);
|
||||
ptr += 4;
|
||||
|
||||
srt_flag = loadUint32(ptr);
|
||||
ptr += 4;
|
||||
|
||||
recv_tsbpd_delay = loadUint16(ptr);
|
||||
ptr += 2;
|
||||
|
||||
send_tsbpd_delay = loadUint16(ptr);
|
||||
ptr += 2;
|
||||
|
||||
return true;
|
||||
|
||||
}
|
||||
std::string HSExtMessage::dump(){
|
||||
_StrPrinter printer;
|
||||
printer << "srt version : "<<std::hex<<srt_version<<" srt flag : "<<std::hex<<srt_flag<<\
|
||||
" recv_tsbpd_delay="<<recv_tsbpd_delay<<" send_tsbpd_delay = "<<send_tsbpd_delay;
|
||||
return std::move(printer);
|
||||
}
|
||||
bool HSExtMessage::storeToData() {
|
||||
_data = BufferRaw::create();
|
||||
_data->setCapacity(HSEXT_MSG_SIZE);
|
||||
_data->setSize(HSEXT_MSG_SIZE);
|
||||
extension_length = 3;
|
||||
HSExt::storeHeader();
|
||||
uint8_t* ptr = (uint8_t*)_data->data()+4;
|
||||
|
||||
storeUint32(ptr,srt_version);
|
||||
ptr += 4;
|
||||
|
||||
storeUint32(ptr,srt_flag);
|
||||
ptr += 4;
|
||||
|
||||
storeUint16(ptr,recv_tsbpd_delay);
|
||||
ptr += 2;
|
||||
|
||||
storeUint16(ptr,send_tsbpd_delay);
|
||||
ptr += 2;
|
||||
return true;
|
||||
}
|
||||
|
||||
bool HSExtStreamID::loadFromData(uint8_t *buf, size_t len) {
|
||||
if(buf == NULL || len < 4){
|
||||
return false;
|
||||
}
|
||||
_data = BufferRaw::create();
|
||||
_data->assign((char*)buf,len);
|
||||
|
||||
HSExt::loadHeader();
|
||||
|
||||
size_t content_size = extension_length*4;
|
||||
if(len < content_size+4){
|
||||
return false;
|
||||
}
|
||||
streamid.clear();
|
||||
char* ptr = _data->data()+4;
|
||||
|
||||
for(size_t i = 0; i<extension_length; ++i){
|
||||
streamid.push_back(*(ptr+3));
|
||||
streamid.push_back(*(ptr+2));
|
||||
streamid.push_back(*(ptr+1));
|
||||
streamid.push_back(*(ptr));
|
||||
ptr+=4;
|
||||
}
|
||||
return true;
|
||||
|
||||
}
|
||||
bool HSExtStreamID::storeToData() {
|
||||
size_t content_size = ((streamid.length()+4)+3)/4*4;
|
||||
|
||||
_data = BufferRaw::create();
|
||||
_data->setCapacity(content_size);
|
||||
_data->setSize(content_size);
|
||||
extension_length = (content_size-4)/4;
|
||||
extension_type = SRT_CMD_SID;
|
||||
HSExt::storeHeader();
|
||||
auto ptr = _data->data()+4;
|
||||
memset(ptr,0,extension_length*4);
|
||||
const char* src = streamid.c_str();
|
||||
for(size_t i = 0; i< streamid.length()/4;++i){
|
||||
*ptr = *(src+3+i*4);
|
||||
ptr++;
|
||||
|
||||
*ptr = *(src+2+i*4);
|
||||
ptr++;
|
||||
|
||||
*ptr = *(src+1+i*4);
|
||||
ptr++;
|
||||
|
||||
*ptr = *(src+0+i*4);
|
||||
ptr++;
|
||||
}
|
||||
|
||||
ptr += 3;
|
||||
size_t offset = streamid.length()/4*4;
|
||||
for(size_t i = 0; i<streamid.length()%4;++i){
|
||||
*ptr = *(src+offset+i);
|
||||
ptr -= 1;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
std::string HSExtStreamID::dump(){
|
||||
_StrPrinter printer;
|
||||
printer << " streamid : "<< streamid;
|
||||
return std::move(printer);
|
||||
}
|
||||
|
||||
} // namespace SRT
|
||||
Reference in New Issue
Block a user