h264编码的RFC3984传输实现

#include "H264FUFrame.h" #include <string.h> #ifdef _WIN32 #include <WinSock2.h> #else #include <arpa/inet.h> #endif namespace fu { enum { FU_A = 28 }; static const unsigned char gs_H264Header[4] = {0x00,0x00,0x00,0x01}; static const unsigned short gs_H264HeaderLen = sizeof(gs_H264Header); struct RTPHeader { #ifdef RTP_BIG_ENDIAN unsigned char version:2; unsigned char padding:1; unsigned char extension:1; unsigned char csrccount:4; unsigned char marker:1; unsigned char payloadtype:7; #else // little endian unsigned char csrccount:4; unsigned char extension:1; unsigned char padding:1; unsigned char version:2; unsigned char payloadtype:7; unsigned char marker:1; #endif // RTP_BIG_ENDIAN unsigned short sequencenumber; unsigned int timestamp; unsigned int ssrc; }; struct RTPExtensionHeader { unsigned short extid; unsigned short length; }; NaluFUFrame::NaluFUFrame(InputPacketType t) :m_pSink(NULL), m_length(gs_H264HeaderLen), m_type(t) { memcpy(m_frameBuffer,gs_H264Header,gs_H264HeaderLen); } NaluFUFrame::~NaluFUFrame() { } bool NaluFUFrame::inputData(unsigned char* data, size_t length) { if (m_type == RAW_PACKET) { return inputRawData(data,length); } return inputRTPData(data,length); } bool NaluFUFrame::inputRawData(unsigned char* data, size_t length) { if (length < 2 || data == NULL) { m_length = gs_H264HeaderLen; return false; } unsigned char fCurPacketNALUnitType = (data[0] & 0x1F); switch (fCurPacketNALUnitType) { case FU_A: case FU_A + 1: { unsigned char startBit = data[1]&0x80; unsigned char endBit = data[1]&0x40; if( startBit ) { //取开始包的 nalu header; m_frameBuffer[m_length] = ( data[0] & 0xE0 ) | ( data[1] & 0x1F ); m_length ++; } length -= 2; if ( m_length + length >= sizeof(m_frameBuffer) ) { m_length = gs_H264HeaderLen; return false; } memcpy(m_frameBuffer + m_length,data + 2 ,length ); m_length += length ; if( endBit == 0 ) { return true; } fCurPacketNALUnitType = m_frameBuffer[gs_H264HeaderLen] & 0x1F; break; } default: m_length = gs_H264HeaderLen; if (length + m_length >= sizeof(m_frameBuffer)) ///over memory???? { return false; } // This packet contains one , decodable NAL units memcpy(m_frameBuffer + m_length,data,length); m_length += length; break; } if (m_pSink != NULL) { int frameType = FRAME_VIDEO_I; switch( fCurPacketNALUnitType) { case 7: frameType = FRAME_VIDEO_SPS; break; case 8: frameType = FRAME_VIDEO_PPS; break; case 5: frameType = FRAME_VIDEO_I; break; default: frameType = FRAME_VIDEO_P; break; } m_pSink->writePacket(m_frameBuffer,m_length,frameType); } m_length = gs_H264HeaderLen; return true; } bool NaluFUFrame::inputAudioData(unsigned char payloadtype, unsigned char* data, size_t length) { if (m_pSink != NULL) { m_pSink->writePacket(data,length,payloadtype); return true; } return false; } bool NaluFUFrame::inputRTPData(unsigned char* rtpdata, size_t rtplength) { if ( (rtpdata == NULL) || (rtplength + m_length >= sizeof(m_frameBuffer)) || (rtplength <= sizeof(RTPHeader)) ) { m_length = gs_H264HeaderLen; return false; } unsigned char *packetbytes = rtpdata; RTPHeader *rtpheader = (RTPHeader*)packetbytes; if (rtpheader->version != 2) { m_length = gs_H264HeaderLen; return false; } size_t packetlen = rtplength; bool marker = (rtpheader->marker == 0)?false:true; unsigned char payloadtype = rtpheader->payloadtype; int csrccount = rtpheader->csrccount; int payloadoffset = sizeof(RTPHeader) + (int)(csrccount * sizeof(unsigned int)); int numpadbytes = 0; if (rtpheader->padding) // adjust payload length to take padding into account { numpadbytes = (int)packetbytes[packetlen - 1]; // last byte contains number of padding bytes if (numpadbytes <= 0) { m_length = gs_H264HeaderLen; return false; } } RTPExtensionHeader *rtpextheader = NULL; bool hasextension = (rtpheader->extension == 0)?false:true; if (hasextension) // got header extension { rtpextheader = (RTPExtensionHeader *)(packetbytes + payloadoffset); payloadoffset += sizeof(RTPExtensionHeader); unsigned short exthdrlen = ntohs(rtpextheader->length); payloadoffset += ((int)exthdrlen)*sizeof(unsigned int); } int payloadlength = packetlen - numpadbytes - payloadoffset; unsigned char *payload = packetbytes + payloadoffset; if ( (payloadlength < 2) || (payloadlength >= sizeof(m_frameBuffer)) ) { m_length = gs_H264HeaderLen; return false; } if (rtpheader->payloadtype < 40) { return inputAudioData(payloadtype,payload,payloadlength); } return inputRawData(payload,payloadlength); } void NaluFUFrame::setSink(StreamSink* pSink) { m_pSink = pSink; } }