Darwin推流存储实现介绍 之二

2024-04-29 02:08

本文主要是介绍Darwin推流存储实现介绍 之二,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

根据上文确定的方案,我们会在每一个推流的会话建立时,增加一个自定义的Output对象,用来实现吧接收到的RTP包写到文件中。


首先,我们在RTSPReflectorOutput.cpp中,参考RTPSessionOutput类定义一个RTPSessionSaveOutput类:


class RTPSessionSaveOutput: public ReflectorOutput
{
public:RTPSessionSaveOutput(ReflectorSession* inReflectorSession, char* theFileName = NULL);virtual ~RTPSessionSaveOutput();//关键是重写这个方法,完成RTP包的存储。QTSS_Error  WritePacket(StrPtrLen* inPacket, void* inStreamCookie, UInt32 inFlags, SInt64 packetLatenessInMSec, SInt64* timeToSendThisPacketAgain, UInt64* packetIDPtr, SInt64* arrivalTimeMSec, Bool16 firstPacket );ReflectorSession* GetReflectorSession();//{ return fReflectorSession; }void        TearDown() {}Bool16      IsUDP() { return true;}Bool16      IsPlaying() { return true;}bool MkFileDirandFile();void Write_Header();private:ReflectorSession*       fReflectorSession;char        			 fFilePath[MaxPathSize];char        			 fFileName[255];SInt32      			 ffd;  // file handleBool16  bLastBlockFlg;//SInt64  lastArrivalTimeMSec;         
};


并完成实现:

#define DEFAULTPATH "/usr/local/movies/pushstream/"
RTPSessionSaveOutput::RTPSessionSaveOutput(ReflectorSession* inReflectorSession, char* theFileName):fReflectorSession(inReflectorSession),ffd(-1)
{printf("------- XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX ,theFileName =%s \n", theFileName);::memset(fFileName, 0, sizeof(fFileName));::memset(fFilePath, 0 , sizeof(fFilePath));bLastBlockFlg = false;lastArrivalTimeMSec = -1;strcpy(fFilePath, DEFAULTPATH);char *firstPos = ::strstr(theFileName, ".sdp");if (firstPos != NULL){int length = strlen(theFileName);int i = length - 1;int end = i;while(i > 0){if (theFileName[i] == firstPos[0]){end = i;}if (theFileName[i] == '/'){printf("----setup path:%d, i:%d-------\n", end, i);break;}i--;}if (i > 0 && end > i + 1){memcpy((char *)fFileName, ((char*)theFileName + i + 1), end - i - 1);fFileName[end - i - 1] = '\0';sprintf(fFilePath, "%s%s", DEFAULTPATH, fFileName);printf("----fFileName:%s, path:%s-------\n", fFileName, fFilePath);}}MkFileDirandFile();printf("-------  RTPSessionSaveAsMP4,  GetNumStreams=%d \n", inReflectorSession->GetNumStreams());this->InititializeBookmarks( inReflectorSession->GetNumStreams());}RTPSessionSaveOutput::~RTPSessionSaveOutput()
{printf("------- YYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYY \n");if(ffd != -1){close(ffd);ffd = -1;}::memset(fFileName, 0, sizeof(fFileName));::memset(fFilePath, 0 , sizeof(fFilePath));
}bool RTPSessionSaveOutput::MkFileDirandFile()
{        if(access(fFilePath, R_OK) != 0) // dir exists{int isCreate = mkdir(fFilePath , S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);if( !isCreate )    qtss_printf("create dir ok, %s\n", fFileName);else{qtss_printf("create %s failed! error code : %d\n", fFilePath, isCreate);return false;}}qtss_printf("---------- dir is ok ---------------%s\n", fFileName);    time_t	tmt =  OS::UnixTime_Secs();	 // Seconds since 1970  				char szBuf[128] = {0};        strftime(szBuf, sizeof(szBuf), "%Y-%m-%d_%H_%M_%S", localtime(&tmt));printf(" start0 =  %s\n", szBuf);if(-1 == ffd){sprintf(fFilePath, "%s/%s.m4a", fFilePath, szBuf);printf("-------open file , fFilePath=%s  \n", fFilePath);ffd = open(fFilePath, O_CREAT  | O_WRONLY  | O_APPEND, 0666);if (ffd != -1){//Write_Header();}}return true;
}ReflectorSession* RTPSessionSaveOutput::GetReflectorSession() 
{qtss_printf("################################## RTPSessionSaveAsMP4::GetReflectorSession \n");return fReflectorSession; 
}void RTPSessionSaveOutput::Write_Header()
{if (1){unsigned char header[4]; //  H264 headerheader[0] = 0;header[1] = 0;header[2] = 0;header[3] = 1;write(ffd, header, 4);}else{unsigned char header[6] = "\0"; //  AMR headerheader[0] = 0x23;header[1] = 0x21;header[2] = 0x41;header[3] = 0x4D;header[4] = 0x52;header[5] = 0x0A;write(ffd, header, sizeof(header));}
}//这个方法主要实现了AAC RTP音频流的存储,将存储的文件拷贝出来后,能正常播放
QTSS_Error  RTPSessionSaveOutput::WritePacket(StrPtrLen* inPacket, void* inStreamCookie, UInt32 inFlags, SInt64 packetLatenessInMSec, SInt64* timeToSendThisPacketAgain, UInt64* packetIDPtr, SInt64* arrivalTimeMSecPtr, Bool16 firstPacket)
{if (inPacket == NULL || inPacket->Len < 14){qtss_printf("inPacket == NULL || inPacket->Len < 14");return QTSS_NoErr;}SInt64                  currentTime = OS::Milliseconds();if(-1 != ffd){UInt8* payload = (UInt8*)inPacket->Ptr;int start = 12; //rtp header length//just save aac audio.if (inPacket->Len < 16){return QTSS_NoErr;}if (payload[start] != 0x00 || payload[start + 1] != 0x10){return QTSS_NoErr;}//check end.int rtptype = payload[start] & 0x1f;//int payload_type = payload[1]& 0x7f;bool bStartFlg = ((payload[start+1] & 0x80) == 0x80) ? true : false;//qtss_printf("#################   rtptype  = %d \n", rtptype);#if 0//这里用来做H264文件的存储if (rtptype== 28) { // FU-A//printf("################# rtptype== 28, arrivalTimeMSecPtr=%ld, len =%d\n", *arrivalTimeMSecPtr, inPacket->Len - start -2);if(bStartFlg) {Write_Header();}int ret = write(ffd, (void*)(payload + start + 2), (inPacket->Len - start -2));if(ret == -1){qtss_printf("error=%d, strerror=%s", errno, strerror(errno));}               } else if (rtptype <= 24){// single nal unit// printf("####################### rtptype rtptype =%d, arrivalTimeMSecPtr=%ld, len =%d\n", rtptype ,*arrivalTimeMSecPtr,  (inPacket->Len - start));//Write_Header();int ret = write(ffd, (void*)(payload+start), (inPacket->Len - start));// 00 00 00 01 if(ret == -1){qtss_printf("error=%d, strerror=%s", errno, strerror(errno));}}#elseif (lastArrivalTimeMSec >= *arrivalTimeMSecPtr){return QTSS_NoErr;}qtss_printf("RTPSessionSaveAsMP4::WritePacket lastArrivalTimeMSec=%d\n", lastArrivalTimeMSec);lastArrivalTimeMSec = *arrivalTimeMSecPtr;#if 1        //aac headerunsigned char adtsHeader[7] = {0};adtsHeader[0] = 0xFF;adtsHeader[1] = 0xF1;int profile = 2;int freqIdx = 11;int chanCfg = 1; int packetLen = inPacket->Len - start + 7 - 4;adtsHeader[2] = ((profile -1 )<<6) + (freqIdx << 2) + (chanCfg >> 2);adtsHeader[3] = ((chanCfg & 3) << 6) + (packetLen >> 11);adtsHeader[4] = (packetLen & 0x7ff) >> 3;//(packetLen >> 3) & 0xff;adtsHeader[5] = ((packetLen & 0x7) << 5)|0x1f;adtsHeader[6] = 0xFC;int ret = write(ffd, adtsHeader, sizeof(adtsHeader));
#endif//     //amr header//aaclatm 4headerret = write(ffd, (void*)(payload + start + 4), (inPacket->Len - start - 4));if(ret == -1){qtss_printf("error=%d, strerror=%s", errno, strerror(errno));}fLastIntervalMilliSec = currentTime - fLastPacketTransmitTime;if (fLastIntervalMilliSec > 100) {//reset interval maybe first packet or it has been blocked for awhilefLastIntervalMilliSec = 5;}fLastPacketTransmitTime = currentTime;if (!bLastBlockFlg){bLastBlockFlg = true;}else{bLastBlockFlg = false;}#endif//flush(ffd);}return QTSS_NoErr;
}



在完成文件输出的RTPSessionSaveOutput类后,我们修改QTSSReflectorModule中的,在推流的逻辑中,增加一个OutPut对象到会话中:

    // If this is an incoming data session, skip everything having to do with setting up a new// RTP Stream. if (isPush){//...//这里暂时只存储音频流if(theTrackID == 0) // qtssVideoPayloadType{RTPSessionSaveOutput* theSaveAsMP4 = NEW RTPSessionSaveOutput(theSession, theFullPath.Ptr);	         theSession->AddOutput(theSaveOutput, false); }}


这样就完成了推送音频流的存储,视频流的存储也能这样来实现,如果考虑音视频同时存储,可能要考虑其他方案来实现,比方需要通过ffmpeg来完成声音和视频融合成一个文件。


这篇关于Darwin推流存储实现介绍 之二的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/944848

相关文章

MySQL中查找重复值的实现

《MySQL中查找重复值的实现》查找重复值是一项常见需求,比如在数据清理、数据分析、数据质量检查等场景下,我们常常需要找出表中某列或多列的重复值,具有一定的参考价值,感兴趣的可以了解一下... 目录技术背景实现步骤方法一:使用GROUP BY和HAVING子句方法二:仅返回重复值方法三:返回完整记录方法四:

IDEA中新建/切换Git分支的实现步骤

《IDEA中新建/切换Git分支的实现步骤》本文主要介绍了IDEA中新建/切换Git分支的实现步骤,通过菜单创建新分支并选择是否切换,创建后在Git详情或右键Checkout中切换分支,感兴趣的可以了... 前提:项目已被Git托管1、点击上方栏Git->NewBrancjsh...2、输入新的分支的

Python实现对阿里云OSS对象存储的操作详解

《Python实现对阿里云OSS对象存储的操作详解》这篇文章主要为大家详细介绍了Python实现对阿里云OSS对象存储的操作相关知识,包括连接,上传,下载,列举等功能,感兴趣的小伙伴可以了解下... 目录一、直接使用代码二、详细使用1. 环境准备2. 初始化配置3. bucket配置创建4. 文件上传到os

关于集合与数组转换实现方法

《关于集合与数组转换实现方法》:本文主要介绍关于集合与数组转换实现方法,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录1、Arrays.asList()1.1、方法作用1.2、内部实现1.3、修改元素的影响1.4、注意事项2、list.toArray()2.1、方

使用Python实现可恢复式多线程下载器

《使用Python实现可恢复式多线程下载器》在数字时代,大文件下载已成为日常操作,本文将手把手教你用Python打造专业级下载器,实现断点续传,多线程加速,速度限制等功能,感兴趣的小伙伴可以了解下... 目录一、智能续传:从崩溃边缘抢救进度二、多线程加速:榨干网络带宽三、速度控制:做网络的好邻居四、终端交互

Python中win32包的安装及常见用途介绍

《Python中win32包的安装及常见用途介绍》在Windows环境下,PythonWin32模块通常随Python安装包一起安装,:本文主要介绍Python中win32包的安装及常见用途的相关... 目录前言主要组件安装方法常见用途1. 操作Windows注册表2. 操作Windows服务3. 窗口操作

java实现docker镜像上传到harbor仓库的方式

《java实现docker镜像上传到harbor仓库的方式》:本文主要介绍java实现docker镜像上传到harbor仓库的方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地... 目录1. 前 言2. 编写工具类2.1 引入依赖包2.2 使用当前服务器的docker环境推送镜像2.2

C++20管道运算符的实现示例

《C++20管道运算符的实现示例》本文简要介绍C++20管道运算符的使用与实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧... 目录标准库的管道运算符使用自己实现类似的管道运算符我们不打算介绍太多,因为它实际属于c++20最为重要的

Java easyExcel实现导入多sheet的Excel

《JavaeasyExcel实现导入多sheet的Excel》这篇文章主要为大家详细介绍了如何使用JavaeasyExcel实现导入多sheet的Excel,文中的示例代码讲解详细,感兴趣的小伙伴可... 目录1.官网2.Excel样式3.代码1.官网easyExcel官网2.Excel样式3.代码

Java中调用数据库存储过程的示例代码

《Java中调用数据库存储过程的示例代码》本文介绍Java通过JDBC调用数据库存储过程的方法,涵盖参数类型、执行步骤及数据库差异,需注意异常处理与资源管理,以优化性能并实现复杂业务逻辑,感兴趣的朋友... 目录一、存储过程概述二、Java调用存储过程的基本javascript步骤三、Java调用存储过程示