rtpinterface.cpp

rtsp协议的主要实现代码.对开发流媒体

// RTPInterface.cpp: implementation of the RTPInterface class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "RTPInterface.h"

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
RTPInterface * RTPInterface::createNew()
{
	SOCKET s;
	s = socket (AF_INET, SOCK_DGRAM,0);
	if(s == INVALID_SOCKET)
		return NULL;

	sockaddr_in addrBnd;
	addrBnd.sin_family = AF_INET;
	addrBnd.sin_addr.S_un.S_addr = INADDR_ANY;
	addrBnd.sin_port = 0;
	
	int nRet = bind(s,(struct sockaddr *)&addrBnd,sizeof(sockaddr_in));
	if(nRet == SOCKET_ERROR)
	{
		closesocket(s);
		return NULL;
	}
	return new RTPInterface(s);
}

RTPInterface::RTPInterface(SOCKET s) :
	m_BindPort(0),m_s(s)
{
	
}

RTPInterface::~RTPInterface()
{

}
int RTPInterface::sendPacket(unsigned char * packet,unsigned long packetSize)
{
	sockaddr_in addrTo;
	addrTo.sin_family = AF_INET;
	list<TYPE_RTP_DEST *>::iterator it;
	TYPE_RTP_DEST * pRTPDest = NULL;
	for(it = m_lstRtpDest.begin(); it != m_lstRtpDest.end(); it++)
	{
		pRTPDest = (*it);
		addrTo.sin_addr.S_un.S_addr = inet_addr(pRTPDest->destIP);
		addrTo.sin_port = htons(pRTPDest->destPort);
		sendto(m_s,(const char *)packet,packetSize,0,(struct sockaddr *)&addrTo,sizeof(sockaddr_in));
	}
	return 0;	
}

unsigned short RTPInterface::GetBindPort()
{
	sockaddr_in addrBnd;
	int addrlen = sizeof(sockaddr_in);
	getsockname(m_s,(struct sockaddr *)&addrBnd,&addrlen);
	m_BindPort = ntohs(addrBnd.sin_port);
	return m_BindPort;
}

void RTPInterface::AddRtpDest(char * destIP,unsigned short destPort)
{
	TYPE_RTP_DEST * pRTPDest = new TYPE_RTP_DEST;
	strcpy(pRTPDest->destIP,destIP);
	pRTPDest->destPort = destPort;
	m_lstRtpDest.push_back(pRTPDest);
}

void RTPInterface::RemoveRtpDest(char * destIP,unsigned short destPort)
{
	list<TYPE_RTP_DEST *>::iterator it;
	TYPE_RTP_DEST * pRTPDest = NULL;
	for(it = m_lstRtpDest.begin(); it != m_lstRtpDest.end(); it++)
	{
		pRTPDest = (*it);
		if(strcmp(pRTPDest->destIP,destIP) == 0 && pRTPDest->destPort == destPort)
		{
			break;
		}
	}
	if(it != m_lstRtpDest.end())
	{
		m_lstRtpDest.erase(it);
	}
}