RTP_h264打包源码

最新推荐文章于 2021-02-16 02:59:25 发布

最新推荐文章于 2021-02-16 02:59:25 发布 · 155 阅读

classCH264_RTP_PACK { #defineRTP_VERSION2 typedefstructNAL_msg_s { booleoFrame; unsignedchartype;//NALtype unsignedchar*start;//pointertofirstlocationinthesendbuffer unsignedchar*end;//pointertolastlocationinsendbuffer unsignedlongsize; }NAL_MSG_t; typedefstruct { //LITTLE_ENDIAN unsignedshortcc:4;/*CSRCcount*/ unsignedshortx:1;/*headerextensionflag*/ unsignedshortp:1;/*paddingflag*/ unsignedshortv:2;/*packettype*/ unsignedshortpt:7;/*payloadtype*/ unsignedshortm:1;/*markerbit*/ unsignedshortseq;/*sequencenumber*/ unsignedlongts;/*timestamp*/ unsignedlongssrc;/*synchronizationsource*/ }rtp_hdr_t; typedefstructtagRTP_INFO { NAL_MSG_tnal;//NALinformation rtp_hdr_trtp_hdr;//RTPheaderisassembledhere inthdr_len;//lengthofRTPheader unsignedchar*pRTP;//pointertowhereRTPpackethasbeemassembled unsignedchar*start;//pointertostartofpayload unsignedchar*end;//pointertoendofpayload unsignedints_bit;//bitintheFUheader unsignedinte_bit;//bitintheFUheader boolFU_flag;//fragmentedNALUnitflag }RTP_INFO; public: CH264_RTP_PACK(unsignedlongH264SSRC,unsignedcharH264PAYLOADTYPE=96 ,unsignedshortMAXRTPPACKSIZE=1472) { m_MAXRTPPACKSIZE=MAXRTPPACKSIZE; if(m_MAXRTPPACKSIZE>10000) { m_MAXRTPPACKSIZE=10000; } if(m_MAXRTPPACKSIZE<50) { m_MAXRTPPACKSIZE=50; } memset(&m_RTP_Info,0,sizeof(m_RTP_Info)); m_RTP_Info.rtp_hdr.pt=H264PAYLOADTYPE; m_RTP_Info.rtp_hdr.ssrc=H264SSRC; m_RTP_Info.rtp_hdr.v=RTP_VERSION; m_RTP_Info.rtp_hdr.seq=0; } ~CH264_RTP_PACK(void) { } //传入Set的数据必须是一个完整的NAL,起始码为0x00000001。 //起始码之前至少预留10个字节，以避免内存COPY操作。 //打包完成后，原缓冲区内的数据被破坏。 boolSet(unsignedchar*NAL_Buf,unsignedlongNAL_Size ,unsignedlongTime_Stamp,boolEnd_Of_Frame) { unsignedlongstartcode=StartCode(NAL_Buf); if(startcode!=0x01000000) { returnfalse; } inttype=NAL_Buf[4]&0x1f; if(type<1||type>12) { returnfalse; } m_RTP_Info.nal.start=NAL_Buf; m_RTP_Info.nal.size=NAL_Size; m_RTP_Info.nal.eoFrame=End_Of_Frame; m_RTP_Info.nal.type=m_RTP_Info.nal.start[4]; m_RTP_Info.nal.end=m_RTP_Info.nal.start+m_RTP_Info.nal.size; m_RTP_Info.rtp_hdr.ts=Time_Stamp; m_RTP_Info.nal.start+=4;//skipthesyncword if((m_RTP_Info.nal.size+7)>m_MAXRTPPACKSIZE) { m_RTP_Info.FU_flag=true; m_RTP_Info.s_bit=1; m_RTP_Info.e_bit=0; m_RTP_Info.nal.start+=1;//skipNALheader } else { m_RTP_Info.FU_flag=false; m_RTP_Info.s_bit=m_RTP_Info.e_bit=0; } m_RTP_Info.start=m_RTP_Info.end=m_RTP_Info.nal.start; m_bBeginNAL=true; returntrue; } //循环调用Get获取RTP包，直到返回值为NULL unsignedchar*Get(unsignedshort*pPacketSize) { if(m_RTP_Info.end==m_RTP_Info.nal.end) { *pPacketSize=0; returnNULL; } if(m_bBeginNAL) { m_bBeginNAL=false; } else { m_RTP_Info.start=m_RTP_Info.end;//continuewiththenextRTP-FUpacket } intbytesLeft=m_RTP_Info.nal.end-m_RTP_Info.start; intmaxSize=m_MAXRTPPACKSIZE-12;//sizeof(basicrtpheader)==12bytes if(m_RTP_Info.FU_flag) maxSize-=2; if(bytesLeft>maxSize) { //limit RTP packetsize to 1472 bytes m_RTP_Info.end=m_RTP_Info.start+maxSize; } else { m_RTP_Info.end=m_RTP_Info.start+bytesLeft; } if(m_RTP_Info.FU_flag) {//multiplepacketNALslice if(m_RTP_Info.end==m_RTP_Info.nal.end) { m_RTP_Info.e_bit=1; } } //should be set at EofFrame m_RTP_Info.rtp_hdr.m=m_RTP_Info.nal.eoFrame?1:0; if(m_RTP_Info.FU_flag&&!m_RTP_Info.e_bit) { m_RTP_Info.rtp_hdr.m=0; } m_RTP_Info.rtp_hdr.seq++; unsignedchar*cp=m_RTP_Info.start; cp-=(m_RTP_Info.FU_flag?14:12); m_RTP_Info.pRTP=cp; unsignedchar*cp2=(unsignedchar*)&m_RTP_Info.rtp_hdr; cp[0]=cp2[0]; cp[1]=cp2[1]; cp[2]=(m_RTP_Info.rtp_hdr.seq>>8)&0xff; cp[3]=m_RTP_Info.rtp_hdr.seq&0xff; cp[4]=(m_RTP_Info.rtp_hdr.ts>>24)&0xff; cp[5]=(m_RTP_Info.rtp_hdr.ts>>16)&0xff; cp[6]=(m_RTP_Info.rtp_hdr.ts>>8)&0xff; cp[7]=m_RTP_Info.rtp_hdr.ts&0xff; cp[8]=(m_RTP_Info.rtp_hdr.ssrc>>24)&0xff; cp[9]=(m_RTP_Info.rtp_hdr.ssrc>>16)&0xff; cp[10]=(m_RTP_Info.rtp_hdr.ssrc>>8)&0xff; cp[11]=m_RTP_Info.rtp_hdr.ssrc&0xff; m_RTP_Info.hdr_len=12; /*! *\nTheFUindicatoroctethasthefollowingformat: *\n *\n+---------------+ *\nMSB|0|1|2|3|4|5|6|7|LSB *\n+-+-+-+-+-+-+-+-+ *\n|F|NRI|Type| *\n+---------------+ *\n *\nTheFUheaderhasthefollowingformat: *\n *\n+---------------+ *\n|0|1|2|3|4|5|6|7| *\n+-+-+-+-+-+-+-+-+ *\n|S|E|R|Type| *\n+---------------+ */ if(m_RTP_Info.FU_flag) { //FUindicatorF|NRI|Type cp[12]=(m_RTP_Info.nal.type&0xe0)|28;//Typeis28forFU_A //FUheaderS|E|R|Type cp[13]=(m_RTP_Info.s_bit<<7) |(m_RTP_Info.e_bit<<6)|(m_RTP_Info.nal.type&0x1f); //R=0,mustbeignoredbyreceiver m_RTP_Info.s_bit=m_RTP_Info.e_bit=0; m_RTP_Info.hdr_len=14; } m_RTP_Info.start=&cp[m_RTP_Info.hdr_len];//newstartofpayload *pPacketSize=m_RTP_Info.hdr_len+(m_RTP_Info.end-m_RTP_Info.start); returnm_RTP_Info.pRTP; } private: unsignedintStartCode(unsignedchar*cp) { unsignedintd32; d32=cp[3]; d32<<=8; d32|=cp[2]; d32<<=8; d32|=cp[1]; d32<<=8; d32|=cp[0]; returnd32; } private: RTP_INFOm_RTP_Info; boolm_bBeginNAL; unsignedshortm_MAXRTPPACKSIZE; }; //END //使用代码： DWORDH264SSRC; CH264_RTP_PACKpack(H264SSRC); BYTE*pVideoData; DWORDSize,ts; boolIsEndOfFrame; WORDwLen; pack.Set(pVideoData,Size,ts,IsEndOfFrame); BYTE*pPacket; while(pPacket=pack.Get(&wLen)) { //rtppacketprocess //... }