rtpconnection.cpp

利用rtp库实现实时语音传送

/*

  This file is a part of JRTPLIB
  Copyright (c) 1999-2004 Jori Liesenborgs

  Contact: [e-mail]

  This library (JRTPLIB) was partially developed for my thesis at the
  School for Knowledge Technology (Belgium/The Netherlands)

  Permission is hereby granted, free of charge, to any person obtaining a
  copy of this software and associated documentation files (the "Software"),
  to deal in the Software without restriction, including without limitation
  the rights to use, copy, modify, merge, publish, distribute, sublicense,
  and/or sell copies of the Software, and to permit persons to whom the
  Software is furnished to do so, subject to the following conditions:

  The above copyright notice and this permission notice shall be included
  in all copies or substantial portions of the Software.

  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  IN THE SOFTWARE.

*/

#include "rtpconnection.h"
#include "rtpcontributingsources.h"
#include "rtptimeutil.h"
#include <stdlib.h>

#ifdef RTP_HAVE_SYS_FILIO
#include <sys/filio.h>
#endif // RTP_HAVE_SYS_FILIO

#ifdef RTP_VXWORKS
#include <resolvLib.h>
#include <ioLib.h>
#include <hostLib.h>
#else
#include <memory.h>
#endif // RTP_VXWORKS

#define RTP_IS_MCASTADDR(x)				(((x)&0xF0000000) == 0xE0000000)

#define RTP_MCASTMEMBERSHIP(socket,type,mcastip,status)	{\
								struct ip_mreq mreq;\
								\
								mreq.imr_multiaddr.s_addr = htonl(mcastip);\
								mreq.imr_interface.s_addr = htonl(localip);\
								status = setsockopt(socket,IPPROTO_IP,type,(const char *)&mreq,sizeof(struct ip_mreq));\
							}
							
struct RawDataBlock
{
	unsigned char *data;
	int datalen;
	unsigned long ip;
	int port;
	RawDataBlock *next;
};

RTPConnection::RTPConnection()
{
	rtpsock = 0;
	rtcpsock = 0;
	sendsock = 0;
	socketsopened = false;
	portbase = 0;
	receivemode = RECEIVEMODE_ALL;
	maxpacksize = RTP_DEFAULT_MAXPACKSIZE;
	rtp_first = NULL;
	rtp_last = NULL;
	rtcp_first = NULL;
	rtcp_last = NULL;
	sendport = 0;
	localip = 0;
	rtcprecvtime.tv_sec = 0;
	rtcprecvtime.tv_usec = 0;
}

RTPConnection::~RTPConnection()
{
	Destroy();
}

int RTPConnection::Create(int pbase,unsigned long localipaddress)
{
	int size;
	RTPSOCKLENTYPE socklen;

	if (socketsopened)
		return ERR_RTP_CONNALREADYCREATED;
		
	/* Check if the localportbase is even, conforming to RFC 1889 */
	
	if (pbase%2 != 0)
		return ERR_RTP_PORTBASENOTEVEN;
	
	/* Create the sockets */

	rtpsock = socket(AF_INET,SOCK_DGRAM,0);
	if (rtpsock == RTPSOCKERR)
		return ERR_RTP_CANTCREATESOCKET;
	rtcpsock = socket(AF_INET,SOCK_DGRAM,0);
	if (rtcpsock == RTPSOCKERR)
		return ERR_RTP_CANTCREATESOCKET;
	sendsock = socket(AF_INET,SOCK_DGRAM,0);
	if (sendsock == RTPSOCKERR)
		return ERR_RTP_CANTCREATESOCKET;

	/* Set socket receive and send buffers */

	size = RTP_RECEIVEBUFFERSIZE;
	if (setsockopt(rtpsock,SOL_SOCKET,SO_RCVBUF,(const char *)&size,sizeof(int)) != 0)
		return ERR_RTP_CANTSETSOCKETBUFFER;
	size = RTP_RECEIVEBUFFERSIZE;
	if (setsockopt(rtcpsock,SOL_SOCKET,SO_RCVBUF,(const char *)&size,sizeof(int)) != 0)
		return ERR_RTP_CANTSETSOCKETBUFFER;
	size = RTP_SENDBUFFERSIZE;
	if (setsockopt(sendsock,SOL_SOCKET,SO_SNDBUF,(const char *)&size,sizeof(int)) != 0)
		return ERR_RTP_CANTSETSOCKETBUFFER;

	/* Bind the sockets */
	
	struct sockaddr_in addr;

	addr.sin_family = AF_INET;
	addr.sin_port = htons(0);
	addr.sin_addr.s_addr = htonl(0);
	if (bind(sendsock,(struct sockaddr *)&addr,sizeof(struct sockaddr)) != 0)
	{
		RTPCLOSESOCKET(rtpsock);
		RTPCLOSESOCKET(rtcpsock);
		RTPCLOSESOCKET(sendsock);
		return ERR_RTP_CANTBINDSOCKET;
	}
	
	addr.sin_family = AF_INET;
	addr.sin_port = htons(pbase);
	addr.sin_addr.s_addr = htonl(INADDR_ANY);
	if (bind(rtpsock,(struct sockaddr *)&addr,sizeof(struct sockaddr)) != 0)
	{
		RTPCLOSESOCKET(rtpsock);
		RTPCLOSESOCKET(rtcpsock);
		RTPCLOSESOCKET(sendsock);
		return ERR_RTP_CANTBINDSOCKET;
	}

	addr.sin_family = AF_INET;
	addr.sin_port = htons(pbase+1);
	addr.sin_addr.s_addr = htonl(INADDR_ANY);
	if (bind(rtcpsock,(struct sockaddr *)&addr,sizeof(struct sockaddr)) != 0)
	{
		RTPCLOSESOCKET(rtpsock);
		RTPCLOSESOCKET(rtcpsock);
		RTPCLOSESOCKET(sendsock);
		return ERR_RTP_CANTBINDSOCKET;
	}

	/* Get the port number of the send socket */

	socklen = sizeof(struct sockaddr_in);
	if (getsockname(sendsock,(struct sockaddr *)&addr,&socklen) != 0)
	{
		RTPCLOSESOCKET(rtpsock);
		RTPCLOSESOCKET(rtcpsock);
		RTPCLOSESOCKET(sendsock);
		return ERR_RTP_CANTGETSOCKETPORT;
	}

	sendport = ntohs(addr.sin_port);

	/* Get the local IP address */

	if (localipaddress == 0) // user didn't supply an IP address
	{
		localip = CalcLocalIP();
		if (localip == 0)
		{
			RTPCLOSESOCKET(rtpsock);
			RTPCLOSESOCKET(rtcpsock);
			RTPCLOSESOCKET(sendsock);
			return ERR_RTP_CANTGETLOCALIP;
		}
	}
	else // user specified the local IP address
		localip = localipaddress;

	socketsopened = true;

#ifdef RTP_MULTICAST
	if (SetMulticastTTL(RTP_DEFAULTMULTICAST_TTL) < 0)
	{
		socketsopened = false;
		RTPCLOSESOCKET(rtpsock);
		RTPCLOSESOCKET(rtcpsock);
		RTPCLOSESOCKET(sendsock);
		return ERR_RTP_CANTSETMULTICASTTTL;
	}
#endif // RTP_MULTICAST
	
	portbase = pbase;
	return 0;
}

void RTPConnection::Destroy()
{
	if (socketsopened)
	{
		LeaveAllMulticastGroups();
		socketsopened = false;
		RTPCLOSESOCKET(rtpsock);
		RTPCLOSESOCKET(rtcpsock);
		RTPCLOSESOCKET(sendsock);
	}
	destinations.Clear();
	ignoreIPs.Clear();
	acceptIPs.Clear();
	mcasttable.Clear();
	receivemode = RECEIVEMODE_ALL;
	maxpacksize = RTP_DEFAULT_MAXPACKSIZE;
	FlushPackets();
}

int RTPConnection::GetPortBase()
{
	if (!socketsopened)
		return ERR_RTP_SOCKETSNOTOPEN;
	return portbase;
}

// version without header extension
int RTPConnection::SetToS(int tos)
{
        int status;

        if (!socketsopened)
                return ERR_RTP_SOCKETSNOTOPEN;

        status = setsockopt(sendsock,IPPROTO_IP,IP_TOS,(const char *)&tos,sizeof(int));
        if (status != 0)
                return ERR_RTP_CANTSETTOS;

        return 0;
}

int RTPConnection::SendRTPData(void *data,int len,RTPContributingSources *srcs,unsigned char pt,bool mark,RTPuint16 seqnr,RTPuint32 timestamp)
{
	RTPHeader *hdr;
	int blocklen,numcsrcs;
		
	if (!socketsopened)
		return ERR_RTP_SOCKETSNOTOPEN;

	if (len <= 0) // nothing to send
		return 0;

	numcsrcs = srcs->GetNumberOfCSRCs();
	if (numcsrcs > 15)
		numcsrcs = 15;

	blocklen = sizeof(RTPHeader)+sizeof(RTPuint32)*numcsrcs+len;
	if (blocklen > maxpacksize)
		return ERR_RTP_PACKETTOOLARGE;

	/* Init header fields */
		
	hdr = (RTPHeader *)packetbuffer;
	hdr->version = RTP_VERSION;
	hdr->padding = 0;
	hdr->extension = 0;
	hdr->cc = numcsrcs;
	hdr->marker = (mark)?1:0;
	hdr->payloadtype = pt;
	hdr->seqnum = htons(seqnr);
	hdr->timestamp = htonl(timestamp);
	hdr->ssrc = srcs->GetLocalSSRC();
	
	// copy CSRC members (CSRC numbers are already in network byte order)
	if (numcsrcs > 0)
		srcs->FillCSRCs(packetbuffer+sizeof(RTPHeader));
	// copy data
	memcpy((packetbuffer+sizeof(RTPHeader)+sizeof(RTPuint32)*numcsrcs),data,len);
	
	// send the data to all destinations
	struct sockaddr_in addr;	
	unsigned long ip;
	int port,rtcpport;

	destinations.InitRetrieval();
	addr.sin_family = AF_INET;
	while (destinations.GetNext(&ip,&port,&rtcpport))
	{
		addr.sin_port = port; // already in network byte order
		addr.sin_addr.s_addr = ip; // already in network byte order
		sendto(sendsock,(const char *)packetbuffer,blocklen,0,(struct sockaddr *)&addr,sizeof(struct sockaddr));
	}

	return 0;
}

// The version with header extension
int RTPConnection::SendRTPData(void *data,int len,RTPContributingSources *srcs,unsigned char pt,bool mark,RTPuint16 seqnr,RTPuint32 timestamp,unsigned short hdrextID,void *hdrextdata,int numhdrextwords)
{
	RTPHeader *hdr;
	RTPHeaderExtension *hdrext;
	int blocklen,numcsrcs,numextwords;
	unsigned char *datapos;
		
	if (!socketsopened)
		return ERR_RTP_SOCKETSNOTOPEN;

	if (len <= 0) // nothing to send
		return 0;

	numcsrcs = srcs->GetNumberOfCSRCs();
	if (numcsrcs > 15)
		numcsrcs = 15;
		
	numextwords = numhdrextwords;
	if (numextwords < 0)
		numextwords = 0;

	blocklen = sizeof(RTPHeader)+sizeof(RTPuint32)*numcsrcs+len+numextwords*sizeof(RTPuint32)+sizeof(RTPHeaderExtension);
	if (blocklen > maxpacksize)
		return ERR_RTP_PACKETTOOLARGE;

	/* Init header fields */
		
	hdr = (RTPHeader *)packetbuffer;
	hdr->version = RTP_VERSION;
	hdr->padding = 0;
	hdr->extension = 1;
	hdr->cc = numcsrcs;
	hdr->marker = (mark)?1:0;
	hdr->payloadtype = pt;
	hdr->seqnum = htons(seqnr);
	hdr->timestamp = htonl(timestamp);
	hdr->ssrc = srcs->GetLocalSSRC();
	
	// copy CSRC members (CSRC numbers are already in network byte order)
	if (numcsrcs > 0)
		srcs->FillCSRCs(packetbuffer+sizeof(RTPHeader));
	
	// set the header extension
	datapos = packetbuffer+sizeof(RTPHeader)+sizeof(RTPuint32)*numcsrcs;
	hdrext = (RTPHeaderExtension *)datapos;
	hdrext->userdefined = htons(hdrextID);
	hdrext->length = htons(numextwords);
	datapos += sizeof(RTPHeaderExtension);
	if (numextwords > 0)
		memcpy(datapos,hdrextdata,numextwords*sizeof(RTPuint32));
		
	// copy data
	datapos += numextwords*sizeof(RTPuint32);
	memcpy(datapos,data,len);
	
	// send the data to all destinations
	struct sockaddr_in addr;	
	unsigned long ip;
	int port,rtcpport;

	destinations.InitRetrieval();
	addr.sin_family = AF_INET;
	while (destinations.GetNext(&ip,&port,&rtcpport))
	{
		addr.sin_port = port; // already in network byte order
		addr.sin_addr.s_addr = ip; // already in network byte order
		sendto(sendsock,(const char *)packetbuffer,blocklen,0,(struct sockaddr *)&addr,sizeof(struct sockaddr));
	}

	return 0;
}