/******************************************************************************
|
* FILE: MediaRelay.h
|
* Description:
|
* Thread receive packet interface class.
|
*
|
* Modified Code History
|
* Mark Date By Modification Reason
|
*******************************************************************************
|
* 01 2013-5-2 songxw Initial creation.
|
******************************************************************************/
|
#if !defined(__TCP_STREAM_TREANS_H__)
|
#define __TCP_STREAM_TREANS_H__
|
|
#include <stdio.h>
|
#include <sys/types.h>
|
#include <sys/socket.h>
|
#include <string.h>
|
#include <netinet/in.h>
|
#include <arpa/inet.h>
|
#include <unistd.h>
|
#include <fcntl.h>
|
#include <errno.h>
|
#include <poll.h>
|
#include <netinet/tcp.h>
|
#include <vector>
|
#include <list>
|
#include "jthread.h"
|
#include "TypeDef.h"
|
#include "librtsp.h"
|
#include "str.h"
|
#include "tool.h"
|
#include "StreamParse.h"
|
|
using namespace std;
|
using namespace jthread;
|
|
#define CHECK_RTP_PACKET_SIZE 10
|
|
#define UDP_PACKET_RECV_MAXSIZE (1024 * 1024)
|
#define TCP_STRAM_HEADER_LEN 2
|
#define TCP_STREAM_BUFFER_MAX_LEN (1024 * 50)
|
|
//rtp Í·
|
typedef struct
|
{
|
/**//* byte 0 */
|
unsigned char csrc_len:4; /**//* expect 0 */
|
unsigned char extension:1; /**//* expect 1, see RTP_OP below */
|
unsigned char padding:1; /**//* expect 0 */
|
unsigned char version:2; /**//* expect 2 */
|
/**//* byte 1 */
|
unsigned char payload:7; /**//* RTP_PAYLOAD_RTSP */
|
unsigned char marker:1; /**//* expect 1 */
|
/**//* bytes 2, 3 */
|
unsigned short seq_no;
|
/**//* bytes 4-7 */
|
unsigned int timestamp;
|
/**//* bytes 8-11 */
|
unsigned int ssrc; /**//* stream number is used here. */
|
} RTP_FIXED_HEADER;
|
|
struct TcpStreamPacket
|
{
|
UINT8* pData;
|
int nLength;
|
TcpStreamPacket& operator=(TcpStreamPacket *oth)
|
{
|
this->pData=oth->pData;
|
this->nLength=oth->nLength;
|
return *this;
|
}
|
};
|
|
typedef std::list<TcpStreamPacket>::iterator TcpPacketIt;
|
class CTcpStreamTrans: public JThread
|
{
|
public:
|
CTcpStreamTrans();
|
~CTcpStreamTrans();
|
|
//Æô¶¯tcpserverµÄ¼àÌý
|
int CreateTcpServer(int port);
|
//ÉèÖÃÂëÁ÷´«ÊäµÄ²ÎÊý
|
bool SetStreamTransParam(StreamTransType_E streamtype, const char *sendip, int sendport, const char *recvip, int recvport);
|
//ÉèÖÃÂëÁ÷»Øµ÷²ÎÊý
|
void SetStreamCallBackParam(PlayCallBack2 callback, long userdata)
|
{
|
MediaCBFunc = callback;
|
UserData = userdata;
|
}
|
public:
|
void* Thread(void);
|
void ThreadQuit(void);
|
private:
|
// µÈ´ýtcpclientÁ¬½Ó
|
int AcceptClient(int socketfd, char *pSrcIp, UINT16 &SrcPort);
|
//¶Ôgb28181µÄtcpÂëÁ÷½øÐÐÕ´°ü´¦Àí
|
void StickyTcpStramPacket(UINT8 *pData, int TcpRecvLen);
|
bool ReadSelect(int scoketfd, UINT32 timeout);
|
int Connect(void);
|
|
int IsMark( UINT8* pBuffer, unsigned int BufSize);
|
int GetRtpHeaderSize(UINT8* _frame, int _frameLen);
|
int ParseTcpStream(UINT8* pData, int len);
|
int HandleRecvStream(UINT8* pAFrame, UINT32 FrameSize, int PayLoadType, bool bMark);
|
private:
|
bool ExitFlag;
|
UINT8* RecvBuffer;
|
|
UINT8 *m_pTcpRecvBufTmp;
|
UINT8 *m_pTcpRecvPacketBuf;
|
int m_sizeOfTcpRecvBufTmp;
|
int m_sizeOfTcpRecvPacketBuf;
|
|
UINT8 *m_pHeadTmpBuff;
|
int m_HeadTmplen;
|
UINT32 m_tcprecvssrc;
|
std::list<TcpStreamPacket> m_listTcpStreamPacket;
|
|
//Èç¹û½ÓÊÕ¶ËΪ tcpserverʱ£¬ÐèÒªaccept tcpclient£»
|
int m_TcpClientFd;
|
int m_TcpServerFd;
|
|
StreamTransType_E m_eStreamTransType; //ÂëÁ÷´«ÊäÀàÐÍ
|
|
char m_StreamSendIp[IPSTR_MAX_LEN+1];
|
int m_StreamSendPort;
|
char m_StreamRecvIp[IPSTR_MAX_LEN+1];
|
int m_StreamRecvPort;
|
|
long UserData; //Óû§Êý¾Ý£¬¿ÉÒÔÊǾä±ú»òÕßÆäËüÓ¦ÓÃÖ¸ÕëµØÖ·
|
PlayCallBack2 MediaCBFunc; //ýÌåÁ÷»Øµ÷º¯ÊýÖ¸Õë
|
|
long m_StreamParseHandle1;
|
long m_StreamParseHandle2;
|
long m_StreamParseHandle3;
|
int m_RecvStreamType;
|
VideoStreamType_E m_eVideoStreamType;
|
};
|
|
#endif
|
