#include <sys/ioctl.h>
|
#include <pthread.h>
|
#include <poll.h>
|
#include <sys/time.h>
|
#include <sys/types.h>
|
#include <unistd.h>
|
#include <sys/epoll.h>
|
#include <sys/ioctl.h>
|
#include <arpa/inet.h>
|
#include <sys/mman.h>
|
#include "rtppacket.h"
|
#include "UdpStreamTrans.h"
|
#include "sps_pps.h"
|
|
CUdpStreamTrans::CUdpStreamTrans()
|
{
|
m_StreamSendIp[0] = '\0';
|
m_StreamSendPort = 0;
|
m_StreamRecvIp[0] = '\0';
|
m_StreamRecvPort = 0;
|
|
UserData = 0;
|
MediaCBFunc = NULL;
|
|
m_StreamParseHandle1 = 0;
|
m_StreamParseHandle2 = 0;
|
m_StreamParseHandle3 = 0;
|
|
m_RecvStreamType = 0;
|
m_eVideoStreamType = E_VIDEO_STREAM_NONE;
|
}
|
|
CUdpStreamTrans::~CUdpStreamTrans()
|
{
|
CRtpApp::ClearDestinations();
|
CRtpApp::RtpDestroy();
|
|
if (m_StreamParseHandle1 != 0)
|
{
|
STREAMPARSE_Close(m_StreamParseHandle1);
|
m_StreamParseHandle1 = 0;
|
}
|
if (m_StreamParseHandle2 != 0)
|
{
|
STREAMPARSE_Close(m_StreamParseHandle2);
|
m_StreamParseHandle2 = 0;
|
}
|
if (m_StreamParseHandle3 != 0)
|
{
|
STREAMPARSE_Close(m_StreamParseHandle3);
|
m_StreamParseHandle3 = 0;
|
}
|
}
|
|
bool CUdpStreamTrans::SetStreamTransParam(StreamTransType_E streamtype, const char *sendip, int sendport, const char *recvip, int recvport)
|
{
|
strncpy2(m_StreamSendIp, sendip, IPSTR_MAX_LEN);
|
m_StreamSendPort = sendport;
|
strncpy2(m_StreamRecvIp, recvip, IPSTR_MAX_LEN);
|
m_StreamRecvPort= recvport;
|
return true;
|
}
|
|
int CUdpStreamTrans::InitRtp(void)
|
{
|
int iRet = -1;
|
//´ò¿ª±¾µØýÌåÍøÂçÁ¬½Ó£¬×¼±¸½ÓÊÕÊý¾Ý¡£
|
CRtpApp::TransParams.SetRTPReceiveBuffer(3 * 1024 * 1024);
|
CRtpApp::TransParams.SetRTPSendBuffer(0);
|
//Create rtp session
|
iRet = CRtpApp::CreateSession(m_StreamRecvPort);
|
if (-1 == iRet)
|
{
|
printf("%s: Create rtp port<%d> session fail!\n", __FUNCTION__, m_StreamRecvPort);
|
return -1;
|
}
|
#if 0
|
//Add rtp destination host
|
RTPIPv4Address RtpAddr(ntohl(inet_addr(m_StreamSendIp) ), m_StreamSendPort);
|
int iErrNum = CRtpApp::AddDestination(RtpAddr);
|
if (iErrNum != 0)
|
{
|
printf("%s: Add RTP Destination <%s:%d> fail!\n", __FUNCTION__, m_StreamSendIp, m_StreamSendPort);
|
}
|
else
|
{
|
printf("%s: Add RTP Destination <%s:%d> sucess!\n", __FUNCTION__, m_StreamSendIp, m_StreamSendPort);
|
}
|
#endif
|
return 0;
|
}
|
|
/*******************************************************************************
|
* Function:
|
* CallBack function of receiving a rtp packet.
|
*******************************************************************************/
|
void CUdpStreamTrans::CBRtpRecv()
|
{
|
CRtpApp::BeginDataAccess();
|
|
//Search data source for rtp
|
if (CRtpApp::GotoFirstSourceWithData())
|
{
|
do
|
{
|
RTPPacket *pRtpPacket = NULL;
|
while ( (pRtpPacket = CRtpApp::GetNextPacket() ) != NULL)
|
{
|
//printf("1.>>>>>>>>>>>>>>>>>recv jrtplib PayloadType:%d len:%d\n", pRtpPacket->GetPayloadType(), pRtpPacket->GetPayloadLength());
|
int rtpplaytype = pRtpPacket->GetPayloadType();
|
if (IsVideoPayloadType(rtpplaytype) )
|
{
|
HandleRecvStream(pRtpPacket->GetPayloadData(), pRtpPacket->GetPayloadLength(), rtpplaytype, pRtpPacket->HasMarker());
|
}
|
//Delete rtp packet dynamic memory pointer
|
CRtpApp::DeletePacket(pRtpPacket);
|
}
|
} while (CRtpApp::GotoNextSourceWithData() );
|
}
|
|
CRtpApp::EndDataAccess();
|
|
//------------------------------ ·¢ËÍýÌåÍøÂ紩͸°ü ------------------------------//
|
int now = GetClockTime();
|
//time(&now);
|
if (now - (int)CRtpApp::SendTimeForNAT >= 10)
|
{
|
CRtpApp::SendNatMappingPacket();
|
CRtpApp::SendTimeForNAT = (time_t)now;
|
}
|
}
|
|
int CUdpStreamTrans::HandleRecvStream(UINT8* pAFrame, UINT32 FrameSize, int PayLoadType, bool bMark)
|
{
|
do
|
{
|
if (0 == m_StreamParseHandle1)
|
{
|
m_StreamParseHandle1 = STREAMPARSE_Open();
|
if (m_StreamParseHandle1 == 0)
|
{
|
//fail
|
}
|
}
|
|
StreamParseParamIn In;
|
In.pBufIn = pAFrame;
|
In.nSizeIn = FrameSize;
|
In.bMark = bMark;
|
In.PayLoadType = (StreamParseType_E)PayLoadType;
|
StreamParseParamOut Out;
|
|
STREAMPARSE_Parse(m_StreamParseHandle1, &In, &Out);
|
if (Out.nSizeOut <= 0)
|
{
|
break;
|
}
|
if (Out.streamtype==0xC0)
|
{
|
//
|
break;
|
}
|
if((m_RecvStreamType == 0) && (Out.streamtype != 0))
|
{
|
m_RecvStreamType = Out.streamtype;
|
printf("%s:1. >>>>>>>>>Detecing Stream Type... StreamType:%02x!!", __FUNCTION__, Out.streamtype);
|
}
|
|
if(m_RecvStreamType == 0)
|
{
|
printf("%s:2. Detecing Stream Type... StreamType:%d!!", __FUNCTION__, Out.streamtype);
|
break;
|
}
|
|
int pts = Out.pts/90;
|
|
StreamParseParamIn StreamIn;
|
if(m_RecvStreamType == 0xE2)//MPEG4
|
{
|
StreamIn.PayLoadType = STREAMPARSE_MERGEAMPEG4;
|
}
|
else if(m_RecvStreamType == 0xE3)//MPEG2
|
{
|
StreamIn.PayLoadType = STREAMPARSE_MERGEAMPEG2;
|
}
|
else//H264
|
{
|
StreamIn.PayLoadType = STREAMPARSE_ESTOES;
|
}
|
if (0 == m_StreamParseHandle2)
|
{
|
m_StreamParseHandle2 = STREAMPARSE_Open();
|
if (m_StreamParseHandle2 == 0)
|
{
|
//fail
|
}
|
}
|
StreamIn.pBufIn = Out.pBufOut;
|
StreamIn.nSizeIn = Out.nSizeOut;
|
StreamIn.bMark = false;
|
StreamParseParamOut StreamOut;
|
|
STREAMPARSE_Parse(m_StreamParseHandle2, &StreamIn, &StreamOut);
|
if (StreamOut.nSizeOut > 0)
|
{
|
int FrameType = GB_VIDEO_FRAME_P;
|
VideoStreamType_E eVideoStreamType = E_VIDEO_STREAM_H264;
|
if(m_RecvStreamType == 0xE2)
|
{
|
eVideoStreamType = E_VIDEO_STREAM_MPEG4;
|
mpeg_header_t* VOL = (mpeg_header_t*)StreamOut.pBufOut;
|
if(is_VOL_header(VOL))
|
{
|
FrameType = GB_VIDEO_FRAME_I;
|
}
|
}
|
else if(m_RecvStreamType == 0xE3)
|
{
|
eVideoStreamType = E_VIDEO_STREAM_MPEG2;
|
mpeg_header_t* SH = (mpeg_header_t*)StreamOut.pBufOut;
|
if(is_sequence_header(SH))
|
{
|
FrameType = GB_VIDEO_FRAME_I;
|
}
|
}
|
else if(m_RecvStreamType == 0xE1)
|
{
|
eVideoStreamType = E_VIDEO_STREAM_H265;
|
if (is_h265_IFrame_Data(StreamOut.pBufOut, StreamOut.nSizeOut))
|
{
|
FrameType = GB_VIDEO_FRAME_I;
|
}
|
}
|
else
|
{
|
eVideoStreamType = E_VIDEO_STREAM_H264;
|
if (is_H264_IFrame_Data(StreamOut.pBufOut, StreamOut.nSizeOut))
|
{
|
FrameType = GB_VIDEO_FRAME_I;
|
}
|
}
|
|
if (m_eVideoStreamType == E_VIDEO_STREAM_NONE)
|
{
|
m_eVideoStreamType = eVideoStreamType;
|
printf("%s:>>>>>>>>>Detecing Stream Type:%d!!!!!!!!", __FUNCTION__, (int)m_eVideoStreamType);
|
}
|
if (MediaCBFunc != NULL)
|
{
|
MediaCBFunc((int)m_eVideoStreamType, FrameType, StreamOut.pBufOut, StreamOut.nSizeOut, UserData);
|
}
|
}
|
break;
|
}while(0);
|
}
|
|
