/src/internet/model/udp-socket-impl.cc
C++ | 1051 lines | 857 code | 82 blank | 112 comment | 182 complexity | 67acf915066dbb105f81a6e684cc6de2 MD5 | raw file
Possible License(s): GPL-2.0
- /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
- /*
- * Copyright (c) 2007 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 "ns3/log.h"
- #include "ns3/node.h"
- #include "ns3/inet-socket-address.h"
- #include "ns3/inet6-socket-address.h"
- #include "ns3/ipv4-route.h"
- #include "ns3/ipv6-route.h"
- #include "ns3/ipv4.h"
- #include "ns3/ipv6.h"
- #include "ns3/ipv4-header.h"
- #include "ns3/ipv4-routing-protocol.h"
- #include "ns3/ipv6-routing-protocol.h"
- #include "ns3/udp-socket-factory.h"
- #include "ns3/trace-source-accessor.h"
- #include "ns3/ipv4-packet-info-tag.h"
- #include "udp-socket-impl.h"
- #include "udp-l4-protocol.h"
- #include "ipv4-end-point.h"
- #include "ipv6-end-point.h"
- #include <limits>
- NS_LOG_COMPONENT_DEFINE ("UdpSocketImpl");
- namespace ns3 {
- NS_OBJECT_ENSURE_REGISTERED (UdpSocketImpl);
- static const uint32_t MAX_IPV4_UDP_DATAGRAM_SIZE = 65507;
- // Add attributes generic to all UdpSockets to base class UdpSocket
- TypeId
- UdpSocketImpl::GetTypeId (void)
- {
- static TypeId tid = TypeId ("ns3::UdpSocketImpl")
- .SetParent<UdpSocket> ()
- .AddConstructor<UdpSocketImpl> ()
- .AddTraceSource ("Drop", "Drop UDP packet due to receive buffer overflow",
- MakeTraceSourceAccessor (&UdpSocketImpl::m_dropTrace))
- .AddAttribute ("IcmpCallback", "Callback invoked whenever an icmp error is received on this socket.",
- CallbackValue (),
- MakeCallbackAccessor (&UdpSocketImpl::m_icmpCallback),
- MakeCallbackChecker ())
- .AddAttribute ("IcmpCallback6", "Callback invoked whenever an icmpv6 error is received on this socket.",
- CallbackValue (),
- MakeCallbackAccessor (&UdpSocketImpl::m_icmpCallback6),
- MakeCallbackChecker ())
- ;
- return tid;
- }
- UdpSocketImpl::UdpSocketImpl ()
- : m_endPoint (0),
- m_endPoint6 (0),
- m_node (0),
- m_udp (0),
- m_errno (ERROR_NOTERROR),
- m_shutdownSend (false),
- m_shutdownRecv (false),
- m_connected (false),
- m_rxAvailable (0)
- {
- NS_LOG_FUNCTION_NOARGS ();
- m_allowBroadcast = false;
- }
- UdpSocketImpl::~UdpSocketImpl ()
- {
- NS_LOG_FUNCTION_NOARGS ();
- // XXX todo: leave any multicast groups that have been joined
- m_node = 0;
- /**
- * Note: actually this function is called AFTER
- * UdpSocketImpl::Destroy or UdpSocketImpl::Destroy6
- * so the code below is unnecessary in normal operations
- */
- if (m_endPoint != 0)
- {
- NS_ASSERT (m_udp != 0);
- /**
- * Note that this piece of code is a bit tricky:
- * when DeAllocate is called, it will call into
- * Ipv4EndPointDemux::Deallocate which triggers
- * a delete of the associated endPoint which triggers
- * in turn a call to the method UdpSocketImpl::Destroy below
- * will will zero the m_endPoint field.
- */
- NS_ASSERT (m_endPoint != 0);
- m_udp->DeAllocate (m_endPoint);
- NS_ASSERT (m_endPoint == 0);
- }
- if (m_endPoint6 != 0)
- {
- NS_ASSERT (m_udp != 0);
- /**
- * Note that this piece of code is a bit tricky:
- * when DeAllocate is called, it will call into
- * Ipv4EndPointDemux::Deallocate which triggers
- * a delete of the associated endPoint which triggers
- * in turn a call to the method UdpSocketImpl::Destroy below
- * will will zero the m_endPoint field.
- */
- NS_ASSERT (m_endPoint6 != 0);
- m_udp->DeAllocate (m_endPoint6);
- NS_ASSERT (m_endPoint6 == 0);
- }
- m_udp = 0;
- }
- void
- UdpSocketImpl::SetNode (Ptr<Node> node)
- {
- NS_LOG_FUNCTION_NOARGS ();
- m_node = node;
- }
- void
- UdpSocketImpl::SetUdp (Ptr<UdpL4Protocol> udp)
- {
- NS_LOG_FUNCTION_NOARGS ();
- m_udp = udp;
- }
- enum Socket::SocketErrno
- UdpSocketImpl::GetErrno (void) const
- {
- NS_LOG_FUNCTION_NOARGS ();
- return m_errno;
- }
- enum Socket::SocketType
- UdpSocketImpl::GetSocketType (void) const
- {
- return NS3_SOCK_DGRAM;
- }
- Ptr<Node>
- UdpSocketImpl::GetNode (void) const
- {
- NS_LOG_FUNCTION_NOARGS ();
- return m_node;
- }
- void
- UdpSocketImpl::Destroy (void)
- {
- NS_LOG_FUNCTION_NOARGS ();
- m_endPoint = 0;
- }
- void
- UdpSocketImpl::Destroy6 (void)
- {
- NS_LOG_FUNCTION_NOARGS ();
- m_endPoint6 = 0;
- }
- int
- UdpSocketImpl::FinishBind (void)
- {
- NS_LOG_FUNCTION_NOARGS ();
- bool done = false;
- if (m_endPoint != 0)
- {
- m_endPoint->SetRxCallback (MakeCallback (&UdpSocketImpl::ForwardUp, Ptr<UdpSocketImpl> (this)));
- m_endPoint->SetIcmpCallback (MakeCallback (&UdpSocketImpl::ForwardIcmp, Ptr<UdpSocketImpl> (this)));
- m_endPoint->SetDestroyCallback (MakeCallback (&UdpSocketImpl::Destroy, Ptr<UdpSocketImpl> (this)));
- done = true;
- }
- if (m_endPoint6 != 0)
- {
- m_endPoint6->SetRxCallback (MakeCallback (&UdpSocketImpl::ForwardUp6, Ptr<UdpSocketImpl> (this)));
- m_endPoint6->SetIcmpCallback (MakeCallback (&UdpSocketImpl::ForwardIcmp6, Ptr<UdpSocketImpl> (this)));
- m_endPoint6->SetDestroyCallback (MakeCallback (&UdpSocketImpl::Destroy6, Ptr<UdpSocketImpl> (this)));
- done = true;
- }
- if (done)
- {
- return 0;
- }
- return -1;
- }
- int
- UdpSocketImpl::Bind (void)
- {
- NS_LOG_FUNCTION_NOARGS ();
- m_endPoint = m_udp->Allocate ();
- return FinishBind ();
- }
- int
- UdpSocketImpl::Bind6 (void)
- {
- NS_LOG_FUNCTION_NOARGS ();
- m_endPoint6 = m_udp->Allocate6 ();
- return FinishBind ();
- }
- int
- UdpSocketImpl::Bind (const Address &address)
- {
- NS_LOG_FUNCTION (this << address);
- if (InetSocketAddress::IsMatchingType (address))
- {
- InetSocketAddress transport = InetSocketAddress::ConvertFrom (address);
- Ipv4Address ipv4 = transport.GetIpv4 ();
- uint16_t port = transport.GetPort ();
- if (ipv4 == Ipv4Address::GetAny () && port == 0)
- {
- m_endPoint = m_udp->Allocate ();
- }
- else if (ipv4 == Ipv4Address::GetAny () && port != 0)
- {
- m_endPoint = m_udp->Allocate (port);
- }
- else if (ipv4 != Ipv4Address::GetAny () && port == 0)
- {
- m_endPoint = m_udp->Allocate (ipv4);
- }
- else if (ipv4 != Ipv4Address::GetAny () && port != 0)
- {
- m_endPoint = m_udp->Allocate (ipv4, port);
- }
- }
- else if (Inet6SocketAddress::IsMatchingType (address))
- {
- Inet6SocketAddress transport = Inet6SocketAddress::ConvertFrom (address);
- Ipv6Address ipv6 = transport.GetIpv6 ();
- uint16_t port = transport.GetPort ();
- if (ipv6 == Ipv6Address::GetAny () && port == 0)
- {
- m_endPoint6 = m_udp->Allocate6 ();
- }
- else if (ipv6 == Ipv6Address::GetAny () && port != 0)
- {
- m_endPoint6 = m_udp->Allocate6 (port);
- }
- else if (ipv6 != Ipv6Address::GetAny () && port == 0)
- {
- m_endPoint6 = m_udp->Allocate6 (ipv6);
- }
- else if (ipv6 != Ipv6Address::GetAny () && port != 0)
- {
- m_endPoint6 = m_udp->Allocate6 (ipv6, port);
- }
- }
- else
- {
- NS_LOG_ERROR ("Not IsMatchingType");
- m_errno = ERROR_INVAL;
- return -1;
- }
- return FinishBind ();
- }
- int
- UdpSocketImpl::ShutdownSend (void)
- {
- NS_LOG_FUNCTION_NOARGS ();
- m_shutdownSend = true;
- return 0;
- }
- int
- UdpSocketImpl::ShutdownRecv (void)
- {
- NS_LOG_FUNCTION_NOARGS ();
- m_shutdownRecv = true;
- return 0;
- }
- int
- UdpSocketImpl::Close (void)
- {
- NS_LOG_FUNCTION_NOARGS ();
- if (m_shutdownRecv == true && m_shutdownSend == true)
- {
- m_errno = Socket::ERROR_BADF;
- return -1;
- }
- m_shutdownRecv = true;
- m_shutdownSend = true;
- return 0;
- }
- int
- UdpSocketImpl::Connect (const Address & address)
- {
- NS_LOG_FUNCTION (this << address);
- if (InetSocketAddress::IsMatchingType(address) == true)
- {
- InetSocketAddress transport = InetSocketAddress::ConvertFrom (address);
- m_defaultAddress = Address(transport.GetIpv4 ());
- m_defaultPort = transport.GetPort ();
- m_connected = true;
- NotifyConnectionSucceeded ();
- }
- else if (Inet6SocketAddress::IsMatchingType(address) == true)
- {
- Inet6SocketAddress transport = Inet6SocketAddress::ConvertFrom (address);
- m_defaultAddress = Address(transport.GetIpv6 ());
- m_defaultPort = transport.GetPort ();
- m_connected = true;
- NotifyConnectionSucceeded ();
- }
- else
- {
- return -1;
- }
- return 0;
- }
- int
- UdpSocketImpl::Listen (void)
- {
- m_errno = Socket::ERROR_OPNOTSUPP;
- return -1;
- }
- int
- UdpSocketImpl::Send (Ptr<Packet> p, uint32_t flags)
- {
- NS_LOG_FUNCTION (this << p << flags);
- if (!m_connected)
- {
- m_errno = ERROR_NOTCONN;
- return -1;
- }
- return DoSend (p);
- }
- int
- UdpSocketImpl::DoSend (Ptr<Packet> p)
- {
- NS_LOG_FUNCTION (this << p);
- if ((m_endPoint == 0) && (InetSocketAddress::IsMatchingType(m_defaultAddress) == true))
- {
- if (Bind () == -1)
- {
- NS_ASSERT (m_endPoint == 0);
- return -1;
- }
- NS_ASSERT (m_endPoint != 0);
- }
- else if ((m_endPoint6 == 0) && (Inet6SocketAddress::IsMatchingType(m_defaultAddress) == true))
- {
- if (Bind6 () == -1)
- {
- NS_ASSERT (m_endPoint6 == 0);
- return -1;
- }
- NS_ASSERT (m_endPoint6 != 0);
- }
- if (m_shutdownSend)
- {
- m_errno = ERROR_SHUTDOWN;
- return -1;
- }
- return DoSendTo (p, (const Address)m_defaultAddress);
- }
- int
- UdpSocketImpl::DoSendTo (Ptr<Packet> p, const Address &address)
- {
- NS_LOG_FUNCTION (this << p << address);
- if (!m_connected)
- {
- NS_LOG_LOGIC ("Not connected");
- if (InetSocketAddress::IsMatchingType(address) == true)
- {
- InetSocketAddress transport = InetSocketAddress::ConvertFrom (address);
- Ipv4Address ipv4 = transport.GetIpv4 ();
- uint16_t port = transport.GetPort ();
- return DoSendTo (p, ipv4, port);
- }
- else if (Inet6SocketAddress::IsMatchingType(address) == true)
- {
- Inet6SocketAddress transport = Inet6SocketAddress::ConvertFrom (address);
- Ipv6Address ipv6 = transport.GetIpv6 ();
- uint16_t port = transport.GetPort ();
- return DoSendTo (p, ipv6, port);
- }
- else
- {
- return -1;
- }
- }
- else
- {
- // connected UDP socket must use default addresses
- NS_LOG_LOGIC ("Connected");
- if (Ipv4Address::IsMatchingType(m_defaultAddress))
- {
- return DoSendTo (p, Ipv4Address::ConvertFrom(m_defaultAddress), m_defaultPort);
- }
- else if (Ipv6Address::IsMatchingType(m_defaultAddress))
- {
- return DoSendTo (p, Ipv6Address::ConvertFrom(m_defaultAddress), m_defaultPort);
- }
- }
- m_errno = ERROR_AFNOSUPPORT;
- return(-1);
- }
- int
- UdpSocketImpl::DoSendTo (Ptr<Packet> p, Ipv4Address dest, uint16_t port)
- {
- NS_LOG_FUNCTION (this << p << dest << port);
- if (m_boundnetdevice)
- {
- NS_LOG_LOGIC ("Bound interface number " << m_boundnetdevice->GetIfIndex ());
- }
- if (m_endPoint == 0)
- {
- if (Bind () == -1)
- {
- NS_ASSERT (m_endPoint == 0);
- return -1;
- }
- NS_ASSERT (m_endPoint != 0);
- }
- if (m_shutdownSend)
- {
- m_errno = ERROR_SHUTDOWN;
- return -1;
- }
- if (p->GetSize () > GetTxAvailable () )
- {
- m_errno = ERROR_MSGSIZE;
- return -1;
- }
- Ptr<Ipv4> ipv4 = m_node->GetObject<Ipv4> ();
- // Locally override the IP TTL for this socket
- // We cannot directly modify the TTL at this stage, so we set a Packet tag
- // The destination can be either multicast, unicast/anycast, or
- // either all-hosts broadcast or limited (subnet-directed) broadcast.
- // For the latter two broadcast types, the TTL will later be set to one
- // irrespective of what is set in these socket options. So, this tagging
- // may end up setting the TTL of a limited broadcast packet to be
- // the same as a unicast, but it will be fixed further down the stack
- if (m_ipMulticastTtl != 0 && dest.IsMulticast ())
- {
- SocketIpTtlTag tag;
- tag.SetTtl (m_ipMulticastTtl);
- p->AddPacketTag (tag);
- }
- else if (m_ipTtl != 0 && !dest.IsMulticast () && !dest.IsBroadcast ())
- {
- SocketIpTtlTag tag;
- tag.SetTtl (m_ipTtl);
- p->AddPacketTag (tag);
- }
- {
- SocketSetDontFragmentTag tag;
- bool found = p->RemovePacketTag (tag);
- if (!found)
- {
- if (m_mtuDiscover)
- {
- tag.Enable ();
- }
- else
- {
- tag.Disable ();
- }
- p->AddPacketTag (tag);
- }
- }
- //
- // If dest is set to the limited broadcast address (all ones),
- // convert it to send a copy of the packet out of every
- // interface as a subnet-directed broadcast.
- // Exception: if the interface has a /32 address, there is no
- // valid subnet-directed broadcast, so send it as limited broadcast
- // Note also that some systems will only send limited broadcast packets
- // out of the "default" interface; here we send it out all interfaces
- //
- if (dest.IsBroadcast ())
- {
- if (!m_allowBroadcast)
- {
- m_errno = ERROR_OPNOTSUPP;
- return -1;
- }
- NS_LOG_LOGIC ("Limited broadcast start.");
- for (uint32_t i = 0; i < ipv4->GetNInterfaces (); i++ )
- {
- // Get the primary address
- Ipv4InterfaceAddress iaddr = ipv4->GetAddress (i, 0);
- Ipv4Address addri = iaddr.GetLocal ();
- if (addri == Ipv4Address ("127.0.0.1"))
- continue;
- // Check if interface-bound socket
- if (m_boundnetdevice)
- {
- if (ipv4->GetNetDevice (i) != m_boundnetdevice)
- continue;
- }
- Ipv4Mask maski = iaddr.GetMask ();
- if (maski == Ipv4Mask::GetOnes ())
- {
- // if the network mask is 255.255.255.255, do not convert dest
- NS_LOG_LOGIC ("Sending one copy from " << addri << " to " << dest
- << " (mask is " << maski << ")");
- m_udp->Send (p->Copy (), addri, dest,
- m_endPoint->GetLocalPort (), port);
- NotifyDataSent (p->GetSize ());
- NotifySend (GetTxAvailable ());
- }
- else
- {
- // Convert to subnet-directed broadcast
- Ipv4Address bcast = addri.GetSubnetDirectedBroadcast (maski);
- NS_LOG_LOGIC ("Sending one copy from " << addri << " to " << bcast
- << " (mask is " << maski << ")");
- m_udp->Send (p->Copy (), addri, bcast,
- m_endPoint->GetLocalPort (), port);
- NotifyDataSent (p->GetSize ());
- NotifySend (GetTxAvailable ());
- }
- }
- NS_LOG_LOGIC ("Limited broadcast end.");
- return p->GetSize ();
- }
- else if (m_endPoint->GetLocalAddress () != Ipv4Address::GetAny ())
- {
- m_udp->Send (p->Copy (), m_endPoint->GetLocalAddress (), dest,
- m_endPoint->GetLocalPort (), port, 0);
- NotifyDataSent (p->GetSize ());
- NotifySend (GetTxAvailable ());
- return p->GetSize ();
- }
- else if (ipv4->GetRoutingProtocol () != 0)
- {
- Ipv4Header header;
- header.SetDestination (dest);
- header.SetProtocol (UdpL4Protocol::PROT_NUMBER);
- Socket::SocketErrno errno_;
- Ptr<Ipv4Route> route;
- Ptr<NetDevice> oif = m_boundnetdevice; //specify non-zero if bound to a specific device
- // TBD-- we could cache the route and just check its validity
- route = ipv4->GetRoutingProtocol ()->RouteOutput (p, header, oif, errno_);
- if (route != 0)
- {
- NS_LOG_LOGIC ("Route exists");
- if (!m_allowBroadcast)
- {
- uint32_t outputIfIndex = ipv4->GetInterfaceForDevice (route->GetOutputDevice ());
- uint32_t ifNAddr = ipv4->GetNAddresses (outputIfIndex);
- for (uint32_t addrI = 0; addrI < ifNAddr; ++addrI)
- {
- Ipv4InterfaceAddress ifAddr = ipv4->GetAddress (outputIfIndex, addrI);
- if (dest == ifAddr.GetBroadcast ())
- {
- m_errno = ERROR_OPNOTSUPP;
- return -1;
- }
- }
- }
- header.SetSource (route->GetSource ());
- m_udp->Send (p->Copy (), header.GetSource (), header.GetDestination (),
- m_endPoint->GetLocalPort (), port, route);
- NotifyDataSent (p->GetSize ());
- return p->GetSize ();
- }
- else
- {
- NS_LOG_LOGIC ("No route to destination");
- NS_LOG_ERROR (errno_);
- m_errno = errno_;
- return -1;
- }
- }
- else
- {
- NS_LOG_ERROR ("ERROR_NOROUTETOHOST");
- m_errno = ERROR_NOROUTETOHOST;
- return -1;
- }
- return 0;
- }
- int
- UdpSocketImpl::DoSendTo (Ptr<Packet> p, Ipv6Address dest, uint16_t port)
- {
- NS_LOG_FUNCTION (this << p << dest << port);
- if (dest.IsIpv4MappedAddress ())
- {
- return (DoSendTo(p, dest.GetIpv4MappedAddress (), port));
- }
- if (m_boundnetdevice)
- {
- NS_LOG_LOGIC ("Bound interface number " << m_boundnetdevice->GetIfIndex ());
- }
- if (m_endPoint6 == 0)
- {
- if (Bind6 () == -1)
- {
- NS_ASSERT (m_endPoint6 == 0);
- return -1;
- }
- NS_ASSERT (m_endPoint6 != 0);
- }
- if (m_shutdownSend)
- {
- m_errno = ERROR_SHUTDOWN;
- return -1;
- }
- if (p->GetSize () > GetTxAvailable () )
- {
- m_errno = ERROR_MSGSIZE;
- return -1;
- }
- Ptr<Ipv6> ipv6 = m_node->GetObject<Ipv6> ();
- // Locally override the IP TTL for this socket
- // We cannot directly modify the TTL at this stage, so we set a Packet tag
- // The destination can be either multicast, unicast/anycast, or
- // either all-hosts broadcast or limited (subnet-directed) broadcast.
- // For the latter two broadcast types, the TTL will later be set to one
- // irrespective of what is set in these socket options. So, this tagging
- // may end up setting the TTL of a limited broadcast packet to be
- // the same as a unicast, but it will be fixed further down the stack
- if (m_ipMulticastTtl != 0 && dest.IsMulticast ())
- {
- SocketIpTtlTag tag;
- tag.SetTtl (m_ipMulticastTtl);
- p->AddPacketTag (tag);
- }
- else if (m_ipTtl != 0 && !dest.IsMulticast ())
- {
- SocketIpTtlTag tag;
- tag.SetTtl (m_ipTtl);
- p->AddPacketTag (tag);
- }
- // There is no analgous to an IPv4 broadcast address in IPv6.
- // Instead, we use a set of link-local, site-local, and global
- // multicast addresses. The Ipv6 routing layers should all
- // provide an interface-specific route to these addresses such
- // that we can treat these multicast addresses as "not broadcast"
- if (m_endPoint6->GetLocalAddress () != Ipv6Address::GetAny ())
- {
- m_udp->Send (p->Copy (), m_endPoint6->GetLocalAddress (), dest,
- m_endPoint6->GetLocalPort (), port, 0);
- NotifyDataSent (p->GetSize ());
- NotifySend (GetTxAvailable ());
- return p->GetSize ();
- }
- else if (ipv6->GetRoutingProtocol () != 0)
- {
- Ipv6Header header;
- header.SetDestinationAddress (dest);
- header.SetNextHeader (UdpL4Protocol::PROT_NUMBER);
- Socket::SocketErrno errno_;
- Ptr<Ipv6Route> route;
- Ptr<NetDevice> oif = m_boundnetdevice; //specify non-zero if bound to a specific device
- // TBD-- we could cache the route and just check its validity
- route = ipv6->GetRoutingProtocol ()->RouteOutput (p, header, oif, errno_);
- if (route != 0)
- {
- NS_LOG_LOGIC ("Route exists");
- header.SetSourceAddress (route->GetSource ());
- m_udp->Send (p->Copy (), header.GetSourceAddress (), header.GetDestinationAddress (),
- m_endPoint6->GetLocalPort (), port, route);
- NotifyDataSent (p->GetSize ());
- return p->GetSize ();
- }
- else
- {
- NS_LOG_LOGIC ("No route to destination");
- NS_LOG_ERROR (errno_);
- m_errno = errno_;
- return -1;
- }
- }
- else
- {
- NS_LOG_ERROR ("ERROR_NOROUTETOHOST");
- m_errno = ERROR_NOROUTETOHOST;
- return -1;
- }
- return 0;
- }
- // XXX maximum message size for UDP broadcast is limited by MTU
- // size of underlying link; we are not checking that now.
- uint32_t
- UdpSocketImpl::GetTxAvailable (void) const
- {
- NS_LOG_FUNCTION_NOARGS ();
- // No finite send buffer is modelled, but we must respect
- // the maximum size of an IP datagram (65535 bytes - headers).
- return MAX_IPV4_UDP_DATAGRAM_SIZE;
- }
- int
- UdpSocketImpl::SendTo (Ptr<Packet> p, uint32_t flags, const Address &address)
- {
- NS_LOG_FUNCTION (this << p << flags << address);
- if (InetSocketAddress::IsMatchingType (address))
- {
- InetSocketAddress transport = InetSocketAddress::ConvertFrom (address);
- Ipv4Address ipv4 = transport.GetIpv4 ();
- uint16_t port = transport.GetPort ();
- return DoSendTo (p, ipv4, port);
- }
- else if (Inet6SocketAddress::IsMatchingType (address))
- {
- Inet6SocketAddress transport = Inet6SocketAddress::ConvertFrom (address);
- Ipv6Address ipv6 = transport.GetIpv6 ();
- uint16_t port = transport.GetPort ();
- return DoSendTo (p, ipv6, port);
- }
- return -1;
- }
- uint32_t
- UdpSocketImpl::GetRxAvailable (void) const
- {
- NS_LOG_FUNCTION_NOARGS ();
- // We separately maintain this state to avoid walking the queue
- // every time this might be called
- return m_rxAvailable;
- }
- Ptr<Packet>
- UdpSocketImpl::Recv (uint32_t maxSize, uint32_t flags)
- {
- NS_LOG_FUNCTION (this << maxSize << flags);
- if (m_deliveryQueue.empty () )
- {
- m_errno = ERROR_AGAIN;
- return 0;
- }
- Ptr<Packet> p = m_deliveryQueue.front ();
- if (p->GetSize () <= maxSize)
- {
- m_deliveryQueue.pop ();
- m_rxAvailable -= p->GetSize ();
- }
- else
- {
- p = 0;
- }
- return p;
- }
- Ptr<Packet>
- UdpSocketImpl::RecvFrom (uint32_t maxSize, uint32_t flags,
- Address &fromAddress)
- {
- NS_LOG_FUNCTION (this << maxSize << flags);
- Ptr<Packet> packet = Recv (maxSize, flags);
- if (packet != 0)
- {
- SocketAddressTag tag;
- bool found;
- found = packet->PeekPacketTag (tag);
- NS_ASSERT (found);
- fromAddress = tag.GetAddress ();
- }
- return packet;
- }
- int
- UdpSocketImpl::GetSockName (Address &address) const
- {
- NS_LOG_FUNCTION_NOARGS ();
- if (m_endPoint != 0)
- {
- address = InetSocketAddress (m_endPoint->GetLocalAddress (), m_endPoint->GetLocalPort ());
- }
- else
- {
- // It is possible to call this method on a socket without a name
- // in which case, behavior is unspecified
- address = InetSocketAddress (Ipv4Address::GetZero (), 0);
- }
- return 0;
- }
- int
- UdpSocketImpl::MulticastJoinGroup (uint32_t interface, const Address &groupAddress)
- {
- NS_LOG_FUNCTION (interface << groupAddress);
- /*
- 1) sanity check interface
- 2) sanity check that it has not been called yet on this interface/group
- 3) determine address family of groupAddress
- 4) locally store a list of (interface, groupAddress)
- 5) call ipv4->MulticastJoinGroup () or Ipv6->MulticastJoinGroup ()
- */
- return 0;
- }
- int
- UdpSocketImpl::MulticastLeaveGroup (uint32_t interface, const Address &groupAddress)
- {
- NS_LOG_FUNCTION (interface << groupAddress);
- /*
- 1) sanity check interface
- 2) determine address family of groupAddress
- 3) delete from local list of (interface, groupAddress); raise a LOG_WARN
- if not already present (but return 0)
- 5) call ipv4->MulticastLeaveGroup () or Ipv6->MulticastLeaveGroup ()
- */
- return 0;
- }
- void
- UdpSocketImpl::BindToNetDevice (Ptr<NetDevice> netdevice)
- {
- NS_LOG_FUNCTION (netdevice);
- Socket::BindToNetDevice (netdevice); // Includes sanity check
- if (m_endPoint == 0)
- {
- if (Bind () == -1)
- {
- NS_ASSERT (m_endPoint == 0);
- return;
- }
- NS_ASSERT (m_endPoint != 0);
- }
- m_endPoint->BindToNetDevice (netdevice);
- return;
- }
- void
- UdpSocketImpl::ForwardUp (Ptr<Packet> packet, Ipv4Header header, uint16_t port,
- Ptr<Ipv4Interface> incomingInterface)
- {
- NS_LOG_FUNCTION (this << packet << header << port);
- if (m_shutdownRecv)
- {
- return;
- }
- // Should check via getsockopt ()..
- if (IsRecvPktInfo ())
- {
- Ipv4PacketInfoTag tag;
- packet->RemovePacketTag (tag);
- tag.SetRecvIf (incomingInterface->GetDevice ()->GetIfIndex ());
- packet->AddPacketTag (tag);
- }
- if ((m_rxAvailable + packet->GetSize ()) <= m_rcvBufSize)
- {
- Address address = InetSocketAddress (header.GetSource (), port);
- SocketAddressTag tag;
- tag.SetAddress (address);
- packet->AddPacketTag (tag);
- m_deliveryQueue.push (packet);
- m_rxAvailable += packet->GetSize ();
- NotifyDataRecv ();
- }
- else
- {
- // In general, this case should not occur unless the
- // receiving application reads data from this socket slowly
- // in comparison to the arrival rate
- //
- // drop and trace packet
- NS_LOG_WARN ("No receive buffer space available. Drop.");
- m_dropTrace (packet);
- }
- }
- void
- UdpSocketImpl::ForwardUp6 (Ptr<Packet> packet, Ipv6Address saddr, Ipv6Address daddr, uint16_t port)
- {
- NS_LOG_FUNCTION (this << packet << saddr << port);
- if (m_shutdownRecv)
- {
- return;
- }
- if ((m_rxAvailable + packet->GetSize ()) <= m_rcvBufSize)
- {
- Address address = Inet6SocketAddress (saddr, port);
- SocketAddressTag tag;
- tag.SetAddress (address);
- packet->AddPacketTag (tag);
- m_deliveryQueue.push (packet);
- m_rxAvailable += packet->GetSize ();
- NotifyDataRecv ();
- }
- else
- {
- // In general, this case should not occur unless the
- // receiving application reads data from this socket slowly
- // in comparison to the arrival rate
- //
- // drop and trace packet
- NS_LOG_WARN ("No receive buffer space available. Drop.");
- m_dropTrace (packet);
- }
- }
- void
- UdpSocketImpl::ForwardIcmp (Ipv4Address icmpSource, uint8_t icmpTtl,
- uint8_t icmpType, uint8_t icmpCode,
- uint32_t icmpInfo)
- {
- NS_LOG_FUNCTION (this << icmpSource << (uint32_t)icmpTtl << (uint32_t)icmpType <<
- (uint32_t)icmpCode << icmpInfo);
- if (!m_icmpCallback.IsNull ())
- {
- m_icmpCallback (icmpSource, icmpTtl, icmpType, icmpCode, icmpInfo);
- }
- }
- void
- UdpSocketImpl::ForwardIcmp6 (Ipv6Address icmpSource, uint8_t icmpTtl,
- uint8_t icmpType, uint8_t icmpCode,
- uint32_t icmpInfo)
- {
- NS_LOG_FUNCTION (this << icmpSource << (uint32_t)icmpTtl << (uint32_t)icmpType <<
- (uint32_t)icmpCode << icmpInfo);
- if (!m_icmpCallback6.IsNull ())
- {
- m_icmpCallback6 (icmpSource, icmpTtl, icmpType, icmpCode, icmpInfo);
- }
- }
- void
- UdpSocketImpl::SetRcvBufSize (uint32_t size)
- {
- m_rcvBufSize = size;
- }
- uint32_t
- UdpSocketImpl::GetRcvBufSize (void) const
- {
- return m_rcvBufSize;
- }
- void
- UdpSocketImpl::SetIpTtl (uint8_t ipTtl)
- {
- m_ipTtl = ipTtl;
- }
- uint8_t
- UdpSocketImpl::GetIpTtl (void) const
- {
- return m_ipTtl;
- }
- void
- UdpSocketImpl::SetIpMulticastTtl (uint8_t ipTtl)
- {
- m_ipMulticastTtl = ipTtl;
- }
- uint8_t
- UdpSocketImpl::GetIpMulticastTtl (void) const
- {
- return m_ipMulticastTtl;
- }
- void
- UdpSocketImpl::SetIpMulticastIf (int32_t ipIf)
- {
- m_ipMulticastIf = ipIf;
- }
- int32_t
- UdpSocketImpl::GetIpMulticastIf (void) const
- {
- return m_ipMulticastIf;
- }
- void
- UdpSocketImpl::SetIpMulticastLoop (bool loop)
- {
- m_ipMulticastLoop = loop;
- }
- bool
- UdpSocketImpl::GetIpMulticastLoop (void) const
- {
- return m_ipMulticastLoop;
- }
- void
- UdpSocketImpl::SetMtuDiscover (bool discover)
- {
- m_mtuDiscover = discover;
- }
- bool
- UdpSocketImpl::GetMtuDiscover (void) const
- {
- return m_mtuDiscover;
- }
- bool
- UdpSocketImpl::SetAllowBroadcast (bool allowBroadcast)
- {
- m_allowBroadcast = allowBroadcast;
- return true;
- }
- bool
- UdpSocketImpl::GetAllowBroadcast () const
- {
- return m_allowBroadcast;
- }
- } // namespace ns3