基于 jrtplib 接受 rtp数据 重组rtp 还原 rtp上的音视频数据

#include "stdafx.h" #include "jrtpwrapper.h" #include "unpackrtp.h" #define RTP_HEAD_LEN 12 #define FRAME_DUMP_SYMBOL "D:\\jrtp_picker_metaldata.264" bool JRTPWrapper::RtpFramizer(void * bufln, int len, int payloadLen,void **pBufOut, int *pOutLen) { *pOutLen = 0; if(len < RTP_HEAD_LEN || payloadLen < 2) { return false; } int rtpHeaderLen = len - payloadLen; unsigned char * src = (unsigned char*)bufln + RTP_HEAD_LEN; unsigned char head1 = *src; unsigned char head2 = *(src+1); unsigned char nal = head1 & 0x1f;// 获取FU_Indicator的Type域 unsigned char flag = head2 &0xe0;// 获取FU header的前三位，判断当前是分包的开始、中间或结束 unsigned char nal_fua = (head1 & 0xe0) | (head2 & 0x1f); bool bFinishFrame = false; if(nal != 28) { printf("NAL TYPE:%d\n",nal); } if(nal == 0x1c) { //printf("FU_A slice..................\n"); if(flag == 0x80) { //start *pBufOut = src - 3; *((unsigned int*)(*pBufOut)) = 0x01000000; *((char*)(*pBufOut)+4) = nal_fua; *pOutLen = len - RTP_HEAD_LEN + 3; }else if(flag == 0x40){ //end *pBufOut = src + 2; *pOutLen = len - RTP_HEAD_LEN - 2; } else { //middle *pBufOut = src + 2; *pOutLen = len - RTP_HEAD_LEN - 2; } } else {//单包数据 *pBufOut = src - 4; * ((unsigned int *)( * pBufOut)) = 0x01000000 ; *pOutLen = len - RTP_HEAD_LEN + 4; } unsigned char *bufTmp = (unsigned char*)bufln; if(bufTmp[1]& 0x80) { bFinishFrame = true; //rtp make }else { bFinishFrame = false; } /*if(*pOutLen > 0) { fwrite(*pBufOut,*pOutLen,1,m_pDumpLocation); }*/ return bFinishFrame; } JRTPWrapper::JRTPWrapper(TRTSPClient &rtspClient,int iLocalPort): m_rtspClient(rtspClient) ,m_iLocalRtpPort(iLocalPort) ,m_iFrameLen(0) ,m_fCompleteFrame(false) ,m_fIsFirstPacket(true) ,m_expectedSeq(0) #ifdef DUMP_FRAME ,m_pDumpLocation(NULL) #endif { } JRTPWrapper::~JRTPWrapper() { } bool JRTPWrapper::Setup() { jrtplib::RTPSessionParams sessparams; jrtplib::RTPUDPv4TransmissionParams tranparams; sessparams.SetOwnTimestampUnit(1.0/25.0); sessparams.SetAcceptOwnPackets(true); tranparams.SetPortbase(m_iLocalRtpPort); int iRet = m_session.Create(sessparams,&tranparams); return !iRet; } bool JRTPWrapper::Startup() { if(this->Setup()) { Start(); return true; } return false; } bool JRTPWrapper::GetIncomingPacket(unsigned char *packetBuffer, int &iLen) { return true; } void JRTPWrapper::Shutdown() { if(m_session.IsActive()) { m_session.BYEDestroy(jrtplib::RTPTime(10,0),0,0); } } void JRTPWrapper::Run() { int iRet; #ifdef DUMP_FRAME m_pDumpLocation = fopen(FRAME_DUMP_SYMBOL,"wb"); #endif while(TRUE) { iRet = m_session.BeginDataAccess(); if(m_session.GotoFirstSourceWithData()) { jrtplib::RTPPacket *packet = NULL; do { while ((packet = m_session.GetNextPacket()) != NULL) { // You can examine the data here printf("current packet information:PacketSize:%u,Timestamp:%u,Sequence:%u,SSRC:%u,PayloadType=%u\n", packet->GetPacketLength(),packet->GetTimestamp(),packet->GetSequenceNumber(),packet->GetSSRC(),packet->GetPayloadType()); // we don't longer need the packet, so // we'll delete it void * nal = NULL; int nal_len = 0; m_fCompleteFrame = RtpFramizer(packet->GetPacketData(),packet->GetPacketLength(), packet->GetPayloadLength(),&nal,&nal_len); if(nal_len >0){ memcpy(m_frameBuffer+m_iFrameLen,nal,nal_len); m_iFrameLen+=nal_len; } //Do extra process for the frame . m_session.DeletePacket(packet); //adjust the frame buffer position pointer. if(m_fCompleteFrame) { #ifdef DUMP_FRAME //printf("Complete framizing,frame size:%u\n",m_iFrameLen); fwrite(m_frameBuffer,m_iFrameLen,1,m_pDumpLocation); #endif m_iFrameLen = 0; } } }while(m_session.GotoNextSourceWithData()); } m_session.EndDataAccess(); } }