sdrangel/qrtplib/extratransmitters/rtpfaketransmitter.cpp

1390 lines
30 KiB
C++

/*
This class allows for jrtp to process packets without sending them out
anywhere.
The incoming messages are handed in to jrtp through the TransmissionParams
and can be retreived from jrtp through the normal polling mecanisms.
The outgoing RTP/RTCP packets are given to jrtp through the normal
session->SendPacket() and those packets are handed back out to the
client through a callback function (packet_ready_cb).
example usage : Allows for integration of RTP into gstreamer.
THIS HAS NOT BEEN TESTED WITH THREADS SO DON'T TRY
Copyright (c) 2005 Philippe Khalaf <burger@speedy.org>
This file is a part of JRTPLIB
Copyright (c) 1999-2004 Jori Liesenborgs
Contact: jori.liesenborgs@gmail.com
This library was developed at the "Expertisecentrum Digitale Media"
(http://www.edm.luc.ac.be), a research center of the "Limburgs Universitair
Centrum" (http://www.luc.ac.be). The library is based upon work done 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 "rtpfaketransmitter.h"
#include "rtprawpacket.h"
#include "rtpipv4address.h"
#include "rtptimeutilities.h"
#include <stdio.h>
#include <sys/socket.h>
#include <net/if.h>
#include <string.h>
#include <netdb.h>
#include <unistd.h>
#ifdef RTP_HAVE_SYS_FILIO
#include <sys/filio.h>
#endif // RTP_HAVE_SYS_FILIO
#define RTPIOCTL ioctl
#define RTPFAKETRANS_MAXPACKSIZE 65535
#define RTPFAKETRANS_IFREQBUFSIZE 8192
//#define RTPFAKETRANS_IS_MCASTADDR(x) (((x)&0xF0000000) == 0xE0000000)
/*#define RTPFAKETRANS_MCASTMEMBERSHIP(socket,type,mcastip,status) {\
struct ip_mreq mreq;\
\
mreq.imr_multiaddr.s_addr = htonl(mcastip);\
mreq.imr_interface.s_addr = htonl(bindIP);\
status = setsockopt(socket,IPPROTO_IP,type,(const char *)&mreq,sizeof(struct ip_mreq));\
}*/
#ifdef RTP_SUPPORT_THREAD
#define MAINMUTEX_LOCK { if (threadsafe) mainmutex.Lock(); }
#define MAINMUTEX_UNLOCK { if (threadsafe) mainmutex.Unlock(); }
#define WAITMUTEX_LOCK { if (threadsafe) waitmutex.Lock(); }
#define WAITMUTEX_UNLOCK { if (threadsafe) waitmutex.Unlock(); }
#else
#define MAINMUTEX_LOCK
#define MAINMUTEX_UNLOCK
#define WAITMUTEX_LOCK
#define WAITMUTEX_UNLOCK
#endif // RTP_SUPPORT_THREAD
namespace qrtplib
{
RTPFakeTransmitter::RTPFakeTransmitter(RTPMemoryManager *mgr ) : RTPTransmitter(mgr), destinations(mgr,RTPMEM_TYPE_CLASS_DESTINATIONLISTHASHELEMENT),acceptignoreinfo(mgr,RTPMEM_TYPE_CLASS_ACCEPTIGNOREHASHELEMENT)
{
created = false;
init = false;
}
RTPFakeTransmitter::~RTPFakeTransmitter()
{
Destroy();
}
int RTPFakeTransmitter::Init(bool tsafe)
{
if (init)
return ERR_RTP_FAKETRANS_ALREADYINIT;
// bomb out if trying to use threads
if (tsafe)
return ERR_RTP_NOTHREADSUPPORT;
#ifdef RTP_SUPPORT_THREAD
threadsafe = tsafe;
if (threadsafe)
{
int status;
status = mainmutex.Init();
if (status < 0)
return ERR_RTP_FAKETRANS_CANTINITMUTEX;
status = waitmutex.Init();
if (status < 0)
return ERR_RTP_FAKETRANS_CANTINITMUTEX;
}
#else
if (tsafe)
return ERR_RTP_NOTHREADSUPPORT;
#endif // RTP_SUPPORT_THREAD
init = true;
return 0;
}
int RTPFakeTransmitter::Create(size_t maximumpacketsize,const RTPTransmissionParams *transparams)
{
// struct sockaddr_in addr;
// int status;
if (!init)
return ERR_RTP_FAKETRANS_NOTINIT;
MAINMUTEX_LOCK
if (created)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_ALREADYCREATED;
}
// Obtain transmission parameters
if (transparams == 0)
params = RTPNew(GetMemoryManager(),RTPMEM_TYPE_OTHER) RTPFakeTransmissionParams;
else
{
if (transparams->GetTransmissionProtocol() != RTPTransmitter::UserDefinedProto)
return ERR_RTP_FAKETRANS_ILLEGALPARAMETERS;
params = (RTPFakeTransmissionParams *)transparams;
}
// Check if portbase is even
//if (params->GetPortbase()%2 != 0)
//{
// MAINMUTEX_UNLOCK
// return ERR_RTP_FAKETRANS_PORTBASENOTEVEN;
//}
// Try to obtain local IP addresses
localIPs = params->GetLocalIPList();
if (localIPs.empty()) // User did not provide list of local IP addresses, calculate them
{
int status;
if ((status = CreateLocalIPList()) < 0)
{
MAINMUTEX_UNLOCK
return status;
}
}
//#ifdef RTP_SUPPORT_IPV4MULTICAST
// if (SetMulticastTTL(params->GetMulticastTTL()))
// supportsmulticasting = true;
// else
// supportsmulticasting = false;
//#else // no multicast support enabled
supportsmulticasting = false;
//#endif // RTP_SUPPORT_IPV4MULTICAST
if (maximumpacketsize > RTPFAKETRANS_MAXPACKSIZE)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_SPECIFIEDSIZETOOBIG;
}
maxpacksize = maximumpacketsize;
portbase = params->GetPortbase();
multicastTTL = params->GetMulticastTTL();
receivemode = RTPTransmitter::AcceptAll;
localhostname = 0;
localhostnamelength = 0;
waitingfordata = false;
created = true;
MAINMUTEX_UNLOCK
return 0;
}
void RTPFakeTransmitter::Destroy()
{
if (!init)
return;
MAINMUTEX_LOCK
if (!created)
{
MAINMUTEX_UNLOCK;
return;
}
if (localhostname)
{
RTPDeleteByteArray(localhostname,GetMemoryManager());
localhostname = 0;
localhostnamelength = 0;
}
destinations.Clear();
#ifdef RTP_SUPPORT_IPV4MULTICAST
// multicastgroups.Clear();
#endif // RTP_SUPPORT_IPV4MULTICAST
FlushPackets();
ClearAcceptIgnoreInfo();
localIPs.clear();
created = false;
RTPDelete(params,GetMemoryManager());
MAINMUTEX_UNLOCK
}
RTPTransmissionInfo *RTPFakeTransmitter::GetTransmissionInfo()
{
if (!init)
return 0;
MAINMUTEX_LOCK
RTPTransmissionInfo *tinf = RTPNew(GetMemoryManager(),RTPMEM_TYPE_CLASS_RTPTRANSMISSIONINFO) RTPFakeTransmissionInfo(localIPs, params);
MAINMUTEX_UNLOCK
return tinf;
}
void RTPFakeTransmitter::DeleteTransmissionInfo(RTPTransmissionInfo *inf)
{
if (!init)
return;
RTPDelete(inf,GetMemoryManager());
}
int RTPFakeTransmitter::GetLocalHostName(uint8_t *buffer,size_t *bufferlength)
{
if (!init)
return ERR_RTP_FAKETRANS_NOTINIT;
MAINMUTEX_LOCK
if (!created)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOTCREATED;
}
if (localhostname == 0)
{
if (localIPs.empty())
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOLOCALIPS;
}
std::list<uint32_t>::const_iterator it;
std::list<std::string> hostnames;
for (it = localIPs.begin() ; it != localIPs.end() ; it++)
{
struct hostent *he;
uint8_t addr[4];
uint32_t ip = (*it);
addr[0] = (uint8_t)((ip>>24)&0xFF);
addr[1] = (uint8_t)((ip>>16)&0xFF);
addr[2] = (uint8_t)((ip>>8)&0xFF);
addr[3] = (uint8_t)(ip&0xFF);
he = gethostbyaddr((char *)addr,4,AF_INET);
if (he != 0)
{
std::string hname = std::string(he->h_name);
hostnames.push_back(hname);
}
}
bool found = false;
if (!hostnames.empty()) // try to select the most appropriate hostname
{
std::list<std::string>::const_iterator it;
for (it = hostnames.begin() ; !found && it != hostnames.end() ; it++)
{
if ((*it).find('.') != std::string::npos)
{
found = true;
localhostnamelength = (*it).length();
localhostname = RTPNew(GetMemoryManager(),RTPMEM_TYPE_OTHER) uint8_t [localhostnamelength+1];
if (localhostname == 0)
{
MAINMUTEX_UNLOCK
return ERR_RTP_OUTOFMEM;
}
memcpy(localhostname,(*it).c_str(),localhostnamelength);
localhostname[localhostnamelength] = 0;
}
}
}
if (!found) // use an IP address
{
uint32_t ip;
int len;
char str[16];
it = localIPs.begin();
ip = (*it);
snprintf(str,16,"%d.%d.%d.%d",(int)((ip>>24)&0xFF),(int)((ip>>16)&0xFF),(int)((ip>>8)&0xFF),(int)(ip&0xFF));
len = strlen(str);
localhostnamelength = len;
localhostname = RTPNew(GetMemoryManager(),RTPMEM_TYPE_OTHER) uint8_t [localhostnamelength + 1];
if (localhostname == 0)
{
MAINMUTEX_UNLOCK
return ERR_RTP_OUTOFMEM;
}
memcpy(localhostname,str,localhostnamelength);
localhostname[localhostnamelength] = 0;
}
}
if ((*bufferlength) < localhostnamelength)
{
*bufferlength = localhostnamelength; // tell the application the required size of the buffer
MAINMUTEX_UNLOCK
return ERR_RTP_TRANS_BUFFERLENGTHTOOSMALL;
}
memcpy(buffer,localhostname,localhostnamelength);
*bufferlength = localhostnamelength;
MAINMUTEX_UNLOCK
return 0;
}
bool RTPFakeTransmitter::ComesFromThisTransmitter(const RTPAddress *addr)
{
if (!init)
return false;
if (addr == 0)
return false;
MAINMUTEX_LOCK
bool v;
if (created && addr->GetAddressType() == RTPAddress::IPv4Address)
{
const RTPIPv4Address *addr2 = (const RTPIPv4Address *)addr;
bool found = false;
std::list<uint32_t>::const_iterator it;
it = localIPs.begin();
while (!found && it != localIPs.end())
{
if (addr2->GetIP() == *it)
found = true;
else
++it;
}
if (!found)
v = false;
else
{
if (addr2->GetPort() == params->GetPortbase()) // check for RTP port
v = true;
else if (addr2->GetPort() == (params->GetPortbase()+1)) // check for RTCP port
v = true;
else
v = false;
}
}
else
v = false;
MAINMUTEX_UNLOCK
return v;
}
int RTPFakeTransmitter::Poll()
{
if (!init)
return ERR_RTP_FAKETRANS_NOTINIT;
int status;
MAINMUTEX_LOCK
if (!created)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOTCREATED;
}
status = FakePoll(); // poll RTP socket
params->SetCurrentData(NULL);
MAINMUTEX_UNLOCK
return status;
}
int RTPFakeTransmitter::WaitForIncomingData(const RTPTime &, bool *)
{
return ERR_RTP_FAKETRANS_WAITNOTIMPLEMENTED;
}
int RTPFakeTransmitter::AbortWait()
{
return ERR_RTP_FAKETRANS_WAITNOTIMPLEMENTED;
}
int RTPFakeTransmitter::SendRTPData(const void *data,size_t len)
{
if (!init)
return ERR_RTP_FAKETRANS_NOTINIT;
MAINMUTEX_LOCK
if (!created)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOTCREATED;
}
if (len > maxpacksize)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_SPECIFIEDSIZETOOBIG;
}
destinations.GotoFirstElement();
// send to each destination
while (destinations.HasCurrentElement())
{
(*params->GetPacketReadyCB())(params->GetPacketReadyCBData(), (uint8_t*)data, len,
destinations.GetCurrentElement().GetIP_NBO(),
destinations.GetCurrentElement().GetRTPPort_NBO(),
1);
destinations.GotoNextElement();
}
MAINMUTEX_UNLOCK
return 0;
}
int RTPFakeTransmitter::SendRTCPData(const void *data,size_t len)
{
if (!init)
return ERR_RTP_FAKETRANS_NOTINIT;
MAINMUTEX_LOCK
if (!created)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOTCREATED;
}
if (len > maxpacksize)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_SPECIFIEDSIZETOOBIG;
}
destinations.GotoFirstElement();
// send to each destination
while (destinations.HasCurrentElement())
{
(*params->GetPacketReadyCB())(params->GetPacketReadyCBData(), (uint8_t*)data, len,
destinations.GetCurrentElement().GetIP_NBO(),
destinations.GetCurrentElement().GetRTCPPort_NBO(),
0);
destinations.GotoNextElement();
}
MAINMUTEX_UNLOCK
return 0;
}
int RTPFakeTransmitter::AddDestination(const RTPAddress &addr)
{
if (!init)
return ERR_RTP_FAKETRANS_NOTINIT;
MAINMUTEX_LOCK
if (!created)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOTCREATED;
}
RTPIPv4Destination dest;
if (!RTPIPv4Destination::AddressToDestination(addr, dest))
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_INVALIDADDRESSTYPE;
}
int status = destinations.AddElement(dest);
MAINMUTEX_UNLOCK
return status;
}
int RTPFakeTransmitter::DeleteDestination(const RTPAddress &addr)
{
if (!init)
return ERR_RTP_FAKETRANS_NOTINIT;
MAINMUTEX_LOCK
if (!created)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOTCREATED;
}
RTPIPv4Destination dest;
if (!RTPIPv4Destination::AddressToDestination(addr, dest))
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_INVALIDADDRESSTYPE;
}
int status = destinations.DeleteElement(dest);
MAINMUTEX_UNLOCK
return status;
}
void RTPFakeTransmitter::ClearDestinations()
{
if (!init)
return;
MAINMUTEX_LOCK
if (created)
destinations.Clear();
MAINMUTEX_UNLOCK
}
bool RTPFakeTransmitter::SupportsMulticasting()
{
if (!init)
return false;
MAINMUTEX_LOCK
bool v;
if (!created)
v = false;
else
v = supportsmulticasting;
MAINMUTEX_UNLOCK
return v;
}
#ifdef RTP_SUPPORT_IPV4MULTICAST
int RTPFakeTransmitter::JoinMulticastGroup(const RTPAddress &)
{
// hrrm wonder how will manage to get multicast info thru to the UDPSINK
/* if (!init)
return ERR_RTP_FAKETRANS_NOTINIT;
MAINMUTEX_LOCK
int status;
if (!created)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOTCREATED;
}
if (addr.GetAddressType() != RTPAddress::IPv4Address)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_INVALIDADDRESSTYPE;
}
const RTPIPv4Address &address = (const RTPIPv4Address &)addr;
uint32_t mcastIP = address.GetIP();
if (!RTPFakeTRANS_IS_MCASTADDR(mcastIP))
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOTAMULTICASTADDRESS;
}
status = multicastgroups.AddElement(mcastIP);
if (status >= 0)
{
RTPFakeTRANS_MCASTMEMBERSHIP(rtpsock,IP_ADD_MEMBERSHIP,mcastIP,status);
if (status != 0)
{
multicastgroups.DeleteElement(mcastIP);
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_COULDNTJOINMULTICASTGROUP;
}
RTPFakeTRANS_MCASTMEMBERSHIP(rtcpsock,IP_ADD_MEMBERSHIP,mcastIP,status);
if (status != 0)
{
RTPFakeTRANS_MCASTMEMBERSHIP(rtpsock,IP_DROP_MEMBERSHIP,mcastIP,status);
multicastgroups.DeleteElement(mcastIP);
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_COULDNTJOINMULTICASTGROUP;
}
}
MAINMUTEX_UNLOCK
return status;*/
return ERR_RTP_FAKETRANS_NOMULTICASTSUPPORT;
}
int RTPFakeTransmitter::LeaveMulticastGroup(const RTPAddress &)
{
/*
if (!init)
return ERR_RTP_FAKETRANS_NOTINIT;
MAINMUTEX_LOCK
int status;
if (!created)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOTCREATED;
}
if (addr.GetAddressType() != RTPAddress::IPv4Address)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_INVALIDADDRESSTYPE;
}
const RTPIPv4Address &address = (const RTPIPv4Address &)addr;
uint32_t mcastIP = address.GetIP();
if (!RTPFakeTRANS_IS_MCASTADDR(mcastIP))
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOTAMULTICASTADDRESS;
}
status = multicastgroups.DeleteElement(mcastIP);
if (status >= 0)
{
RTPFakeTRANS_MCASTMEMBERSHIP(rtpsock,IP_DROP_MEMBERSHIP,mcastIP,status);
RTPFakeTRANS_MCASTMEMBERSHIP(rtcpsock,IP_DROP_MEMBERSHIP,mcastIP,status);
status = 0;
}
MAINMUTEX_UNLOCK
return status;
*/
return ERR_RTP_FAKETRANS_NOMULTICASTSUPPORT;
}
void RTPFakeTransmitter::LeaveAllMulticastGroups()
{
/* if (!init)
return;
MAINMUTEX_LOCK
if (created)
{
multicastgroups.GotoFirstElement();
while (multicastgroups.HasCurrentElement())
{
uint32_t mcastIP;
int status = 0;
mcastIP = multicastgroups.GetCurrentElement();
RTPFakeTRANS_MCASTMEMBERSHIP(rtpsock,IP_DROP_MEMBERSHIP,mcastIP,status);
RTPFakeTRANS_MCASTMEMBERSHIP(rtcpsock,IP_DROP_MEMBERSHIP,mcastIP,status);
multicastgroups.GotoNextElement();
}
multicastgroups.Clear();
}
MAINMUTEX_UNLOCK*/
}
#else // no multicast support
int RTPFakeTransmitter::JoinMulticastGroup(const RTPAddress &addr)
{
return ERR_RTP_FAKETRANS_NOMULTICASTSUPPORT;
}
int RTPFakeTransmitter::LeaveMulticastGroup(const RTPAddress &addr)
{
return ERR_RTP_FAKETRANS_NOMULTICASTSUPPORT;
}
void RTPFakeTransmitter::LeaveAllMulticastGroups()
{
}
#endif // RTP_SUPPORT_IPV4MULTICAST
int RTPFakeTransmitter::SetReceiveMode(RTPTransmitter::ReceiveMode m)
{
if (!init)
return ERR_RTP_FAKETRANS_NOTINIT;
MAINMUTEX_LOCK
if (!created)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOTCREATED;
}
if (m != receivemode)
{
receivemode = m;
acceptignoreinfo.Clear();
}
MAINMUTEX_UNLOCK
return 0;
}
int RTPFakeTransmitter::AddToIgnoreList(const RTPAddress &addr)
{
if (!init)
return ERR_RTP_FAKETRANS_NOTINIT;
MAINMUTEX_LOCK
int status;
if (!created)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOTCREATED;
}
if (addr.GetAddressType() != RTPAddress::IPv4Address)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_INVALIDADDRESSTYPE;
}
if (receivemode != RTPTransmitter::IgnoreSome)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_DIFFERENTRECEIVEMODE;
}
const RTPIPv4Address &address = (const RTPIPv4Address &)addr;
status = ProcessAddAcceptIgnoreEntry(address.GetIP(),address.GetPort());
MAINMUTEX_UNLOCK
return status;
}
int RTPFakeTransmitter::DeleteFromIgnoreList(const RTPAddress &addr)
{
if (!init)
return ERR_RTP_FAKETRANS_NOTINIT;
MAINMUTEX_LOCK
int status;
if (!created)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOTCREATED;
}
if (addr.GetAddressType() != RTPAddress::IPv4Address)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_INVALIDADDRESSTYPE;
}
if (receivemode != RTPTransmitter::IgnoreSome)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_DIFFERENTRECEIVEMODE;
}
const RTPIPv4Address &address = (const RTPIPv4Address &)addr;
status = ProcessDeleteAcceptIgnoreEntry(address.GetIP(),address.GetPort());
MAINMUTEX_UNLOCK
return status;
}
void RTPFakeTransmitter::ClearIgnoreList()
{
if (!init)
return;
MAINMUTEX_LOCK
if (created && receivemode == RTPTransmitter::IgnoreSome)
ClearAcceptIgnoreInfo();
MAINMUTEX_UNLOCK
}
int RTPFakeTransmitter::AddToAcceptList(const RTPAddress &addr)
{
if (!init)
return ERR_RTP_FAKETRANS_NOTINIT;
MAINMUTEX_LOCK
int status;
if (!created)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOTCREATED;
}
if (addr.GetAddressType() != RTPAddress::IPv4Address)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_INVALIDADDRESSTYPE;
}
if (receivemode != RTPTransmitter::AcceptSome)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_DIFFERENTRECEIVEMODE;
}
const RTPIPv4Address &address = (const RTPIPv4Address &)addr;
status = ProcessAddAcceptIgnoreEntry(address.GetIP(),address.GetPort());
MAINMUTEX_UNLOCK
return status;
}
int RTPFakeTransmitter::DeleteFromAcceptList(const RTPAddress &addr)
{
if (!init)
return ERR_RTP_FAKETRANS_NOTINIT;
MAINMUTEX_LOCK
int status;
if (!created)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOTCREATED;
}
if (addr.GetAddressType() != RTPAddress::IPv4Address)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_INVALIDADDRESSTYPE;
}
if (receivemode != RTPTransmitter::AcceptSome)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_DIFFERENTRECEIVEMODE;
}
const RTPIPv4Address &address = (const RTPIPv4Address &)addr;
status = ProcessDeleteAcceptIgnoreEntry(address.GetIP(),address.GetPort());
MAINMUTEX_UNLOCK
return status;
}
void RTPFakeTransmitter::ClearAcceptList()
{
if (!init)
return;
MAINMUTEX_LOCK
if (created && receivemode == RTPTransmitter::AcceptSome)
ClearAcceptIgnoreInfo();
MAINMUTEX_UNLOCK
}
int RTPFakeTransmitter::SetMaximumPacketSize(size_t s)
{
if (!init)
return ERR_RTP_FAKETRANS_NOTINIT;
MAINMUTEX_LOCK
if (!created)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_NOTCREATED;
}
if (s > RTPFAKETRANS_MAXPACKSIZE)
{
MAINMUTEX_UNLOCK
return ERR_RTP_FAKETRANS_SPECIFIEDSIZETOOBIG;
}
maxpacksize = s;
MAINMUTEX_UNLOCK
return 0;
}
bool RTPFakeTransmitter::NewDataAvailable()
{
if (!init)
return false;
MAINMUTEX_LOCK
bool v;
if (!created)
v = false;
else
{
if (rawpacketlist.empty())
v = false;
else
v = true;
}
MAINMUTEX_UNLOCK
return v;
}
RTPRawPacket *RTPFakeTransmitter::GetNextPacket()
{
if (!init)
return 0;
MAINMUTEX_LOCK
RTPRawPacket *p;
if (!created)
{
MAINMUTEX_UNLOCK
return 0;
}
if (rawpacketlist.empty())
{
MAINMUTEX_UNLOCK
return 0;
}
p = *(rawpacketlist.begin());
rawpacketlist.pop_front();
MAINMUTEX_UNLOCK
return p;
}
// Here the private functions start...
#ifdef RTP_SUPPORT_IPV4MULTICAST
bool RTPFakeTransmitter::SetMulticastTTL(uint8_t)
{
/* int ttl2,status;
ttl2 = (int)ttl;
status = setsockopt(rtpsock,IPPROTO_IP,IP_MULTICAST_TTL,(const char *)&ttl2,sizeof(int));
if (status != 0)
return false;
status = setsockopt(rtcpsock,IPPROTO_IP,IP_MULTICAST_TTL,(const char *)&ttl2,sizeof(int));
if (status != 0)
return false;
return true;*/
return true;
}
#endif // RTP_SUPPORT_IPV4MULTICAST
void RTPFakeTransmitter::FlushPackets()
{
std::list<RTPRawPacket*>::const_iterator it;
for (it = rawpacketlist.begin() ; it != rawpacketlist.end() ; ++it)
RTPDelete(*it,GetMemoryManager());
rawpacketlist.clear();
}
int RTPFakeTransmitter::FakePoll()
{
uint8_t *data = NULL;
int data_len = 0;
uint32_t sourceaddr;
uint16_t sourceport;
bool rtp;
bool acceptdata;
RTPTime curtime = RTPTime::CurrentTime();
data = params->GetCurrentData();
data_len = params->GetCurrentDataLen();
rtp = params->GetCurrentDataType();
sourceaddr = params->GetCurrentDataAddr();
sourceport = params->GetCurrentDataPort();
// lets make sure we got something
if (data == NULL || data_len <= 0)
{
return 0;
}
// let's make a RTPIPv4Address
RTPIPv4Address *addr = RTPNew(GetMemoryManager(),RTPMEM_TYPE_CLASS_RTPADDRESS) RTPIPv4Address(sourceaddr, sourceport);
if (addr == 0)
{
return ERR_RTP_OUTOFMEM;
}
// ok we got the src addr, now this should be the actual packet
uint8_t *datacopy;
datacopy = RTPNew(GetMemoryManager(),(rtp)?RTPMEM_TYPE_BUFFER_RECEIVEDRTPPACKET:RTPMEM_TYPE_BUFFER_RECEIVEDRTCPPACKET) uint8_t[data_len];
if (datacopy == 0)
{
RTPDelete(addr,GetMemoryManager());
return ERR_RTP_OUTOFMEM;
}
memcpy(datacopy, data, data_len);
// got data, process it
if (receivemode == RTPTransmitter::AcceptAll)
acceptdata = true;
else
acceptdata = ShouldAcceptData(addr->GetIP(),addr->GetPort());
if (acceptdata)
{
// adding packet to queue
RTPRawPacket *pack;
pack = RTPNew(GetMemoryManager(),RTPMEM_TYPE_CLASS_RTPRAWPACKET) RTPRawPacket(datacopy,data_len,addr,curtime,rtp,GetMemoryManager());
if (pack == 0)
{
RTPDelete(addr,GetMemoryManager());
return ERR_RTP_OUTOFMEM;
}
rawpacketlist.push_back(pack);
}
return 0;
}
int RTPFakeTransmitter::ProcessAddAcceptIgnoreEntry(uint32_t ip,uint16_t port)
{
acceptignoreinfo.GotoElement(ip);
if (acceptignoreinfo.HasCurrentElement()) // An entry for this IP address already exists
{
PortInfo *portinf = acceptignoreinfo.GetCurrentElement();
if (port == 0) // select all ports
{
portinf->all = true;
portinf->portlist.clear();
}
else if (!portinf->all)
{
std::list<uint16_t>::const_iterator it,begin,end;
begin = portinf->portlist.begin();
end = portinf->portlist.end();
for (it = begin ; it != end ; it++)
{
if (*it == port) // already in list
return 0;
}
portinf->portlist.push_front(port);
}
}
else // got to create an entry for this IP address
{
PortInfo *portinf;
int status;
portinf = RTPNew(GetMemoryManager(),RTPMEM_TYPE_CLASS_ACCEPTIGNOREPORTINFO) PortInfo();
if (port == 0) // select all ports
portinf->all = true;
else
portinf->portlist.push_front(port);
status = acceptignoreinfo.AddElement(ip,portinf);
if (status < 0)
{
RTPDelete(portinf,GetMemoryManager());
return status;
}
}
return 0;
}
void RTPFakeTransmitter::ClearAcceptIgnoreInfo()
{
acceptignoreinfo.GotoFirstElement();
while (acceptignoreinfo.HasCurrentElement())
{
PortInfo *inf;
inf = acceptignoreinfo.GetCurrentElement();
RTPDelete(inf,GetMemoryManager());
acceptignoreinfo.GotoNextElement();
}
acceptignoreinfo.Clear();
}
int RTPFakeTransmitter::ProcessDeleteAcceptIgnoreEntry(uint32_t ip,uint16_t port)
{
acceptignoreinfo.GotoElement(ip);
if (!acceptignoreinfo.HasCurrentElement())
return ERR_RTP_FAKETRANS_NOSUCHENTRY;
PortInfo *inf;
inf = acceptignoreinfo.GetCurrentElement();
if (port == 0) // delete all entries
{
inf->all = false;
inf->portlist.clear();
}
else // a specific port was selected
{
if (inf->all) // currently, all ports are selected. Add the one to remove to the list
{
// we have to check if the list doesn't contain the port already
std::list<uint16_t>::const_iterator it,begin,end;
begin = inf->portlist.begin();
end = inf->portlist.end();
for (it = begin ; it != end ; it++)
{
if (*it == port) // already in list: this means we already deleted the entry
return ERR_RTP_FAKETRANS_NOSUCHENTRY;
}
inf->portlist.push_front(port);
}
else // check if we can find the port in the list
{
std::list<uint16_t>::iterator it,begin,end;
begin = inf->portlist.begin();
end = inf->portlist.end();
for (it = begin ; it != end ; ++it)
{
if (*it == port) // found it!
{
inf->portlist.erase(it);
return 0;
}
}
// didn't find it
return ERR_RTP_FAKETRANS_NOSUCHENTRY;
}
}
return 0;
}
bool RTPFakeTransmitter::ShouldAcceptData(uint32_t srcip,uint16_t srcport)
{
if (receivemode == RTPTransmitter::AcceptSome)
{
PortInfo *inf;
acceptignoreinfo.GotoElement(srcip);
if (!acceptignoreinfo.HasCurrentElement())
return false;
inf = acceptignoreinfo.GetCurrentElement();
if (!inf->all) // only accept the ones in the list
{
std::list<uint16_t>::const_iterator it,begin,end;
begin = inf->portlist.begin();
end = inf->portlist.end();
for (it = begin ; it != end ; it++)
{
if (*it == srcport)
return true;
}
return false;
}
else // accept all, except the ones in the list
{
std::list<uint16_t>::const_iterator it,begin,end;
begin = inf->portlist.begin();
end = inf->portlist.end();
for (it = begin ; it != end ; it++)
{
if (*it == srcport)
return false;
}
return true;
}
}
else // IgnoreSome
{
PortInfo *inf;
acceptignoreinfo.GotoElement(srcip);
if (!acceptignoreinfo.HasCurrentElement())
return true;
inf = acceptignoreinfo.GetCurrentElement();
if (!inf->all) // ignore the ports in the list
{
std::list<uint16_t>::const_iterator it,begin,end;
begin = inf->portlist.begin();
end = inf->portlist.end();
for (it = begin ; it != end ; it++)
{
if (*it == srcport)
return false;
}
return true;
}
else // ignore all, except the ones in the list
{
std::list<uint16_t>::const_iterator it,begin,end;
begin = inf->portlist.begin();
end = inf->portlist.end();
for (it = begin ; it != end ; it++)
{
if (*it == srcport)
return true;
}
return false;
}
}
return true;
}
int RTPFakeTransmitter::CreateLocalIPList()
{
// first try to obtain the list from the network interface info
if (!GetLocalIPList_Interfaces())
{
// If this fails, we'll have to depend on DNS info
GetLocalIPList_DNS();
}
AddLoopbackAddress();
return 0;
}
//#ifdef WIN32
bool RTPFakeTransmitter::GetLocalIPList_Interfaces()
{
// REMINDER: got to find out how to do this
return false;
}
/*
#else // use ioctl
bool RTPFakeTransmitter::GetLocalIPList_Interfaces()
{
int status;
char buffer[RTPFakeTRANS_IFREQBUFSIZE];
struct ifconf ifc;
struct ifreq *ifr;
struct sockaddr *sa;
char *startptr,*endptr;
int remlen;
ifc.ifc_len = RTPFakeTRANS_IFREQBUFSIZE;
ifc.ifc_buf = buffer;
status = ioctl(rtpsock,SIOCGIFCONF,&ifc);
if (status < 0)
return false;
startptr = (char *)ifc.ifc_req;
endptr = startptr + ifc.ifc_len;
remlen = ifc.ifc_len;
while((startptr < endptr) && remlen >= (int)sizeof(struct ifreq))
{
ifr = (struct ifreq *)startptr;
sa = &(ifr->ifr_addr);
#ifdef RTP_HAVE_SOCKADDR_LEN
if (sa->sa_len <= sizeof(struct sockaddr))
{
if (sa->sa_len == sizeof(struct sockaddr_in) && sa->sa_family == PF_INET)
{
uint32_t ip;
struct sockaddr_in *addr = (struct sockaddr_in *)sa;
ip = ntohl(addr->sin_addr.s_addr);
localIPs.push_back(ip);
}
remlen -= sizeof(struct ifreq);
startptr += sizeof(struct ifreq);
}
else
{
int l = sa->sa_len-sizeof(struct sockaddr)+sizeof(struct ifreq);
remlen -= l;
startptr += l;
}
#else // don't have sa_len in struct sockaddr
if (sa->sa_family == PF_INET)
{
uint32_t ip;
struct sockaddr_in *addr = (struct sockaddr_in *)sa;
ip = ntohl(addr->sin_addr.s_addr);
localIPs.push_back(ip);
}
remlen -= sizeof(struct ifreq);
startptr += sizeof(struct ifreq);
#endif // RTP_HAVE_SOCKADDR_LEN
}
if (localIPs.empty())
return false;
return true;
}
#endif // WIN32
*/
void RTPFakeTransmitter::GetLocalIPList_DNS()
{
struct hostent *he;
char name[1024];
uint32_t ip;
bool done;
int i,j;
gethostname(name,1023);
name[1023] = 0;
he = gethostbyname(name);
if (he == 0)
return;
ip = 0;
i = 0;
done = false;
while (!done)
{
if (he->h_addr_list[i] == NULL)
done = true;
else
{
ip = 0;
for (j = 0 ; j < 4 ; j++)
ip |= ((uint32_t)((unsigned char)he->h_addr_list[i][j])<<((3-j)*8));
localIPs.push_back(ip);
i++;
}
}
}
void RTPFakeTransmitter::AddLoopbackAddress()
{
uint32_t loopbackaddr = (((uint32_t)127)<<24)|((uint32_t)1);
std::list<uint32_t>::const_iterator it;
bool found = false;
for (it = localIPs.begin() ; !found && it != localIPs.end() ; it++)
{
if (*it == loopbackaddr)
found = true;
}
if (!found)
localIPs.push_back(loopbackaddr);
}
} // end namespace