294 lines
8.9 KiB
C++
294 lines
8.9 KiB
C++
/* -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil; -*- */
|
|
/*
|
|
* Copyright (c) 2005 INRIA
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License version 2 as
|
|
* published by the Free Software Foundation;
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*
|
|
* Author: Mathieu Lacage <mathieu.lacage@sophia.inria.fr>
|
|
*/
|
|
|
|
#include "ipv4-end-point-demux.h"
|
|
#include "ipv4-end-point.h"
|
|
#include "ns3/log.h"
|
|
|
|
namespace ns3{
|
|
|
|
NS_LOG_COMPONENT_DEFINE ("Ipv4EndPointDemux");
|
|
|
|
Ipv4EndPointDemux::Ipv4EndPointDemux ()
|
|
: m_ephemeral (49152)
|
|
{
|
|
NS_LOG_FUNCTION_NOARGS ();
|
|
}
|
|
|
|
Ipv4EndPointDemux::~Ipv4EndPointDemux ()
|
|
{
|
|
NS_LOG_FUNCTION_NOARGS ();
|
|
for (EndPointsI i = m_endPoints.begin (); i != m_endPoints.end (); i++)
|
|
{
|
|
Ipv4EndPoint *endPoint = *i;
|
|
delete endPoint;
|
|
}
|
|
m_endPoints.clear ();
|
|
}
|
|
|
|
bool
|
|
Ipv4EndPointDemux::LookupPortLocal (uint16_t port)
|
|
{
|
|
NS_LOG_FUNCTION_NOARGS ();
|
|
for (EndPointsI i = m_endPoints.begin (); i != m_endPoints.end (); i++)
|
|
{
|
|
if ((*i)->GetLocalPort () == port)
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool
|
|
Ipv4EndPointDemux::LookupLocal (Ipv4Address addr, uint16_t port)
|
|
{
|
|
NS_LOG_FUNCTION_NOARGS ();
|
|
for (EndPointsI i = m_endPoints.begin (); i != m_endPoints.end (); i++)
|
|
{
|
|
if ((*i)->GetLocalPort () == port &&
|
|
(*i)->GetLocalAddress () == addr)
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
Ipv4EndPoint *
|
|
Ipv4EndPointDemux::Allocate (void)
|
|
{
|
|
NS_LOG_FUNCTION_NOARGS ();
|
|
uint16_t port = AllocateEphemeralPort ();
|
|
if (port == 0)
|
|
{
|
|
NS_LOG_WARN ("Ephemeral port allocation failed.");
|
|
return 0;
|
|
}
|
|
Ipv4EndPoint *endPoint = new Ipv4EndPoint (Ipv4Address::GetAny (), port);
|
|
m_endPoints.push_back (endPoint);
|
|
NS_LOG_DEBUG ("Now have >>" << m_endPoints.size () << "<< endpoints.");
|
|
return endPoint;
|
|
}
|
|
|
|
Ipv4EndPoint *
|
|
Ipv4EndPointDemux::Allocate (Ipv4Address address)
|
|
{
|
|
NS_LOG_FUNCTION (this << address);
|
|
uint16_t port = AllocateEphemeralPort ();
|
|
if (port == 0)
|
|
{
|
|
NS_LOG_WARN ("Ephemeral port allocation failed.");
|
|
return 0;
|
|
}
|
|
Ipv4EndPoint *endPoint = new Ipv4EndPoint (address, port);
|
|
m_endPoints.push_back (endPoint);
|
|
NS_LOG_DEBUG ("Now have >>" << m_endPoints.size () << "<< endpoints.");
|
|
return endPoint;
|
|
}
|
|
|
|
Ipv4EndPoint *
|
|
Ipv4EndPointDemux::Allocate (uint16_t port)
|
|
{
|
|
NS_LOG_FUNCTION (this << port);
|
|
|
|
return Allocate (Ipv4Address::GetAny (), port);
|
|
}
|
|
|
|
Ipv4EndPoint *
|
|
Ipv4EndPointDemux::Allocate (Ipv4Address address, uint16_t port)
|
|
{
|
|
NS_LOG_FUNCTION (this << address << port);
|
|
if (LookupLocal (address, port))
|
|
{
|
|
NS_LOG_WARN ("Duplicate address/port; failing.");
|
|
return 0;
|
|
}
|
|
Ipv4EndPoint *endPoint = new Ipv4EndPoint (address, port);
|
|
m_endPoints.push_back (endPoint);
|
|
NS_LOG_DEBUG ("Now have >>" << m_endPoints.size () << "<< endpoints.");
|
|
return endPoint;
|
|
}
|
|
|
|
Ipv4EndPoint *
|
|
Ipv4EndPointDemux::Allocate (Ipv4Address localAddress, uint16_t localPort,
|
|
Ipv4Address peerAddress, uint16_t peerPort)
|
|
{
|
|
NS_LOG_FUNCTION (this << localAddress << localPort << peerAddress << peerPort);
|
|
for (EndPointsI i = m_endPoints.begin (); i != m_endPoints.end (); i++)
|
|
{
|
|
if ((*i)->GetLocalPort () == localPort &&
|
|
(*i)->GetLocalAddress () == localAddress &&
|
|
(*i)->GetPeerPort () == peerPort &&
|
|
(*i)->GetPeerAddress () == peerAddress)
|
|
{
|
|
NS_LOG_WARN ("No way we can allocate this end-point.");
|
|
/* no way we can allocate this end-point. */
|
|
return 0;
|
|
}
|
|
}
|
|
Ipv4EndPoint *endPoint = new Ipv4EndPoint (localAddress, localPort);
|
|
endPoint->SetPeer (peerAddress, peerPort);
|
|
m_endPoints.push_back (endPoint);
|
|
|
|
NS_LOG_DEBUG ("Now have >>" << m_endPoints.size () << "<< endpoints.");
|
|
|
|
return endPoint;
|
|
}
|
|
|
|
void
|
|
Ipv4EndPointDemux::DeAllocate (Ipv4EndPoint *endPoint)
|
|
{
|
|
NS_LOG_FUNCTION_NOARGS ();
|
|
for (EndPointsI i = m_endPoints.begin (); i != m_endPoints.end (); i++)
|
|
{
|
|
if (*i == endPoint)
|
|
{
|
|
delete endPoint;
|
|
m_endPoints.erase (i);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
/*
|
|
* If we have an exact match, we return it.
|
|
* Otherwise, if we find a generic match, we return it.
|
|
* Otherwise, we return 0.
|
|
*/
|
|
Ipv4EndPointDemux::EndPoints
|
|
Ipv4EndPointDemux::Lookup (Ipv4Address daddr, uint16_t dport,
|
|
Ipv4Address saddr, uint16_t sport,
|
|
Ptr<Ipv4Interface> incomingInterface)
|
|
{
|
|
NS_LOG_FUNCTION_NOARGS ();
|
|
EndPoints retval1; // Matches exact on local port, wildcards on others
|
|
EndPoints retval2; // Matches exact on local port/adder, wildcards on others
|
|
EndPoints retval3; // Matches all but local address
|
|
EndPoints retval4; // Exact match on all 4
|
|
|
|
NS_LOG_FUNCTION (this << daddr << dport << saddr << sport << incomingInterface);
|
|
NS_LOG_DEBUG ("Looking up endpoint for destination address " << daddr);
|
|
for (EndPointsI i = m_endPoints.begin (); i != m_endPoints.end (); i++)
|
|
{
|
|
Ipv4EndPoint* endP = *i;
|
|
NS_LOG_DEBUG ("Looking at endpoint dport=" << endP->GetLocalPort ()
|
|
<< " daddr=" << endP->GetLocalAddress ()
|
|
<< " sport=" << endP->GetPeerPort ()
|
|
<< " saddr=" << endP->GetPeerAddress ());
|
|
if (endP->GetLocalPort () != dport)
|
|
{
|
|
NS_LOG_LOGIC ("Skipping endpoint " << &endP
|
|
<< " because endpoint dport "
|
|
<< endP->GetLocalPort ()
|
|
<< " does not match packet dport " << dport);
|
|
continue;
|
|
}
|
|
bool isBroadcast = (daddr.IsBroadcast () ||
|
|
daddr.IsSubnetDirectedBroadcast (
|
|
incomingInterface->GetNetworkMask ()));
|
|
Ipv4Address incomingInterfaceAddr = incomingInterface->GetAddress ();
|
|
NS_LOG_DEBUG ("dest addr " << daddr << " broadcast? " << isBroadcast);
|
|
bool localAddressMatchesWildCard =
|
|
endP->GetLocalAddress() == Ipv4Address::GetAny();
|
|
bool localAddressMatchesExact = endP->GetLocalAddress () == daddr;
|
|
|
|
if (isBroadcast)
|
|
{
|
|
NS_LOG_DEBUG("Found bcast, localaddr " << endP->GetLocalAddress());
|
|
}
|
|
|
|
if (isBroadcast && (endP->GetLocalAddress() != Ipv4Address::GetAny()))
|
|
{
|
|
localAddressMatchesExact = (endP->GetLocalAddress () ==
|
|
incomingInterfaceAddr);
|
|
}
|
|
// if no match here, keep looking
|
|
if (!(localAddressMatchesExact || localAddressMatchesWildCard))
|
|
continue;
|
|
bool remotePeerMatchesExact = endP->GetPeerPort () == sport;
|
|
bool remotePeerMatchesWildCard = endP->GetPeerPort() == 0;
|
|
bool remoteAddressMatchesExact = endP->GetPeerAddress () == saddr;
|
|
bool remoteAddressMatchesWildCard = endP->GetPeerAddress () ==
|
|
Ipv4Address::GetAny();
|
|
// If remote does not match either with exact or wildcard,
|
|
// skip this one
|
|
if (!(remotePeerMatchesExact || remotePeerMatchesWildCard))
|
|
continue;
|
|
if (!(remoteAddressMatchesExact || remoteAddressMatchesWildCard))
|
|
continue;
|
|
|
|
// Now figure out which return list to add this one to
|
|
if (localAddressMatchesWildCard &&
|
|
remotePeerMatchesWildCard &&
|
|
remoteAddressMatchesWildCard)
|
|
{ // Only local port matches exactly
|
|
retval1.push_back(endP);
|
|
}
|
|
if ((localAddressMatchesExact || (isBroadcast && localAddressMatchesWildCard))&&
|
|
remotePeerMatchesWildCard &&
|
|
remoteAddressMatchesWildCard)
|
|
{ // Only local port and local address matches exactly
|
|
retval2.push_back(endP);
|
|
}
|
|
if (localAddressMatchesWildCard &&
|
|
remotePeerMatchesExact &&
|
|
remoteAddressMatchesExact)
|
|
{ // All but local address
|
|
retval3.push_back(endP);
|
|
}
|
|
if (localAddressMatchesExact &&
|
|
remotePeerMatchesExact &&
|
|
remoteAddressMatchesExact)
|
|
{ // All 4 match
|
|
retval4.push_back(endP);
|
|
}
|
|
}
|
|
|
|
// Here we find the most exact match
|
|
if (!retval4.empty()) return retval4;
|
|
if (!retval3.empty()) return retval3;
|
|
if (!retval2.empty()) return retval2;
|
|
return retval1; // might be empty if no matches
|
|
}
|
|
|
|
uint16_t
|
|
Ipv4EndPointDemux::AllocateEphemeralPort (void)
|
|
{
|
|
NS_LOG_FUNCTION_NOARGS ();
|
|
uint16_t port = m_ephemeral;
|
|
do
|
|
{
|
|
port++;
|
|
if (port == 65535)
|
|
{
|
|
port = 49152;
|
|
}
|
|
if (!LookupPortLocal (port))
|
|
{
|
|
return port;
|
|
}
|
|
} while (port != m_ephemeral);
|
|
return 0;
|
|
}
|
|
|
|
} //namespace ns3
|
|
|