/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */

/*

 * Copyright (c) 2006 Georgia Tech Research Corporation

 *               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

 *

 * Authors: George F. Riley<[email protected]>

 *          Mathieu Lacage <[email protected]>

 */

#include "ns3/log.h"

#include "ns3/packet.h"

#include "node.h"

#include "socket.h"

#include "socket-factory.h"

#include <limits>

NS_LOG_COMPONENT_DEFINE ("Socket");

namespace ns3 {

Socket::Socket (void)

{

  NS_LOG_FUNCTION_NOARGS ();

}

Socket::~Socket ()

{

  NS_LOG_FUNCTION_NOARGS ();

}

Ptr<Socket> 

Socket::CreateSocket (Ptr<Node> node, TypeId tid)

{

  Ptr<Socket> s;

  NS_ASSERT (node != 0);

  Ptr<SocketFactory> socketFactory = node->GetObject<SocketFactory> (tid);

  NS_ASSERT (socketFactory != 0);

  s = socketFactory->CreateSocket ();

  NS_ASSERT (s != 0);

  return s;

}

void 

Socket::SetConnectCallback (

  Callback<void, Ptr<Socket> > connectionSucceeded,

  Callback<void, Ptr<Socket> > connectionFailed)

{

  NS_LOG_FUNCTION_NOARGS ();

  m_connectionSucceeded = connectionSucceeded;

  m_connectionFailed = connectionFailed;

}

void 

Socket::SetCloseCallbacks (

  Callback<void, Ptr<Socket> > normalClose,

  Callback<void, Ptr<Socket> > errorClose)

{

  NS_LOG_FUNCTION_NOARGS ();

  m_normalClose = normalClose;

  m_errorClose = errorClose;

}

void 

Socket::SetAcceptCallback (

  Callback<bool, Ptr<Socket>, const Address &> connectionRequest,

  Callback<void, Ptr<Socket>, const Address&> newConnectionCreated)

{

  NS_LOG_FUNCTION_NOARGS ();

  m_connectionRequest = connectionRequest;

  m_newConnectionCreated = newConnectionCreated;

}

void

Socket::SetDataSentCallback (Callback<void, Ptr<Socket>, uint32_t> dataSent)

{

  NS_LOG_FUNCTION_NOARGS ();

  m_dataSent = dataSent;

}

void

Socket::SetSendCallback (Callback<void, Ptr<Socket>, uint32_t> sendCb)

{

  NS_LOG_FUNCTION_NOARGS ();

  m_sendCb = sendCb;

}

void 

Socket::SetRecvCallback (Callback<void, Ptr<Socket> > receivedData)

{

  NS_LOG_FUNCTION_NOARGS ();

  m_receivedData = receivedData;

}

int 

Socket::Send (Ptr<Packet> p)

{

  NS_LOG_FUNCTION_NOARGS ();

  return Send (p, 0);

}

int 

Socket::Send (const uint8_t* buf, uint32_t size, uint32_t flags)

{

  NS_LOG_FUNCTION_NOARGS ();

  Ptr<Packet> p;

  if (buf)

    {

      p = Create<Packet> (buf, size);

    }

  else

    {

      p = Create<Packet> (size);

    }

  return Send (p, flags);

}

int 

Socket::SendTo (const uint8_t* buf, uint32_t size, uint32_t flags,

                const Address &toAddress)

{

  NS_LOG_FUNCTION_NOARGS ();

  Ptr<Packet> p;

  if(buf)

    {

      p = Create<Packet> (buf, size);

    }

  else

    {

      p = Create<Packet> (size);

    }

  return SendTo (p, flags, toAddress);

}

Ptr<Packet>

Socket::Recv (void)

{

  NS_LOG_FUNCTION_NOARGS ();

  return Recv (std::numeric_limits<uint32_t>::max(), 0);

}

int 

Socket::Recv (uint8_t* buf, uint32_t size, uint32_t flags)

{

  NS_LOG_FUNCTION_NOARGS ();

  Ptr<Packet> p = Recv (size, flags); // read up to "size" bytes

  if (p == 0)

    {

      return 0;

    }

  p->CopyData (buf, p->GetSize ());

  return p->GetSize ();

}

Ptr<Packet>

Socket::RecvFrom (Address &fromAddress)

{

  NS_LOG_FUNCTION_NOARGS ();

  return RecvFrom (std::numeric_limits<uint32_t>::max(), 0, fromAddress);

}

int 

Socket::RecvFrom (uint8_t* buf, uint32_t size, uint32_t flags,

                  Address &fromAddress)

{

  NS_LOG_FUNCTION_NOARGS ();

  Ptr<Packet> p = RecvFrom (size, flags, fromAddress); 

  if (p == 0)

    {

      return 0;

    }

  p->CopyData (buf, p->GetSize ());

  return p->GetSize ();

}

void 

Socket::NotifyConnectionSucceeded (void)

{

  NS_LOG_FUNCTION_NOARGS ();

  if (!m_connectionSucceeded.IsNull ())

    {

      m_connectionSucceeded (this);

    }

}

void 

Socket::NotifyConnectionFailed (void)

{

  NS_LOG_FUNCTION_NOARGS ();

  if (!m_connectionFailed.IsNull ())

    {

      m_connectionFailed (this);

    }

}

void 

Socket::NotifyNormalClose (void)

{

  NS_LOG_FUNCTION_NOARGS ();

  if (!m_normalClose.IsNull ())

    {

      m_normalClose (this);

    }

}

void 

Socket::NotifyErrorClose (void)

{

  NS_LOG_FUNCTION_NOARGS ();

  if (!m_errorClose.IsNull ())

    {

      m_errorClose (this);

    }

}

bool 

Socket::NotifyConnectionRequest (const Address &from)

{

  NS_LOG_FUNCTION_NOARGS ();

  if (!m_connectionRequest.IsNull ())

    {

      return m_connectionRequest (this, from);

    }

  else

    {

      // accept all incoming connections by default.

      // this way people writing code don't have to do anything

      // special like register a callback that returns true

      // just to get incoming connections

      return true;

    }

}

void 

Socket::NotifyNewConnectionCreated (Ptr<Socket> socket, const Address &from)

{

  NS_LOG_FUNCTION_NOARGS ();

  if (!m_newConnectionCreated.IsNull ())

    {

      m_newConnectionCreated (socket, from);

    }

}

void 

Socket::NotifyDataSent (uint32_t size)

{

  NS_LOG_FUNCTION_NOARGS ();

  if (!m_dataSent.IsNull ())

    {

      m_dataSent (this, size);

    }

}

void 

Socket::NotifySend (uint32_t spaceAvailable)

{

  NS_LOG_FUNCTION_NOARGS ();

  if (!m_sendCb.IsNull ())

    {

      m_sendCb (this, spaceAvailable);

    }

}

void 

Socket::NotifyDataRecv (void)

{

  NS_LOG_FUNCTION_NOARGS ();

  if (!m_receivedData.IsNull ())

    {

      m_receivedData (this);

    }

}

void 

Socket::DoDispose (void)

{

  m_connectionSucceeded = MakeNullCallback<void,Ptr<Socket> > ();

  m_connectionFailed = MakeNullCallback<void,Ptr<Socket> > ();

  m_connectionRequest = MakeNullCallback<bool,Ptr<Socket>, const Address &> ();

  m_newConnectionCreated = MakeNullCallback<void,Ptr<Socket>, const Address &> ();

  m_dataSent = MakeNullCallback<void,Ptr<Socket>, uint32_t> ();

  m_sendCb = MakeNullCallback<void,Ptr<Socket>, uint32_t> ();

  m_receivedData = MakeNullCallback<void,Ptr<Socket> > ();

}

/***************************************************************

 *           Socket Tags

 ***************************************************************/

SocketAddressTag::SocketAddressTag ()  

{

}

void 

SocketAddressTag::SetAddress (Address addr)

{

  m_address = addr;

}

Address 

SocketAddressTag::GetAddress (void) const

{

  return m_address;

}

TypeId

SocketAddressTag::GetTypeId (void)

{

  static TypeId tid = TypeId ("ns3::SocketAddressTag")

    .SetParent<Tag> ()

    .AddConstructor<SocketAddressTag> ()

    ;

  return tid;

}

TypeId

SocketAddressTag::GetInstanceTypeId (void) const

{

  return GetTypeId ();

}

uint32_t

SocketAddressTag::GetSerializedSize (void) const

{

  return m_address.GetSerializedSize ();

}

void

SocketAddressTag::Serialize (TagBuffer i) const

{

  m_address.Serialize (i);

}

void

SocketAddressTag::Deserialize (TagBuffer i)

{

  m_address.Deserialize (i);

}

void

SocketAddressTag::Print (std::ostream &os) const

{

  os << "address=" << m_address;

}

SocketIpTtlTag::SocketIpTtlTag ()  

{

}

void 

SocketIpTtlTag::SetTtl (uint8_t ttl)

{

  m_ttl = ttl;

}

uint8_t 

SocketIpTtlTag::GetTtl (void) const

{

  return m_ttl;

}

TypeId

SocketIpTtlTag::GetTypeId (void)

{

  static TypeId tid = TypeId ("ns3::SocketIpTtlTag")

    .SetParent<Tag> ()

    .AddConstructor<SocketIpTtlTag> ()

    ;

  return tid;

}

TypeId

SocketIpTtlTag::GetInstanceTypeId (void) const

{

  return GetTypeId ();

}

uint32_t 

SocketIpTtlTag::GetSerializedSize (void) const

{ 

  return 1;

}

void 

SocketIpTtlTag::Serialize (TagBuffer i) const

{ 

  i.WriteU8 (m_ttl);

}

void 

SocketIpTtlTag::Deserialize (TagBuffer i)

{ 

  m_ttl = i.ReadU8 ();

}

void

SocketIpTtlTag::Print (std::ostream &os) const

{

  os << "Ttl=" << (uint32_t) m_ttl;

}

SocketSetDontFragmentTag::SocketSetDontFragmentTag ()

{}

void 

SocketSetDontFragmentTag::Enable (void)

{

  m_dontFragment = true;

}

void 

SocketSetDontFragmentTag::Disable (void)

{

  m_dontFragment = false;

}

bool 

SocketSetDontFragmentTag::IsEnabled (void) const

{

  return m_dontFragment;

}

TypeId 

SocketSetDontFragmentTag::GetTypeId (void)

{

  static TypeId tid = TypeId ("ns3::SocketSetDontFragmentTag")

    .SetParent<Tag> ()

    .AddConstructor<SocketSetDontFragmentTag> ()

    ;

  return tid;

}

TypeId 

SocketSetDontFragmentTag::GetInstanceTypeId (void) const

{

  return GetTypeId ();

}

uint32_t 

SocketSetDontFragmentTag::GetSerializedSize (void) const

{

  return 1;

}

void 

SocketSetDontFragmentTag::Serialize (TagBuffer i) const

{

  i.WriteU8 (m_dontFragment?1:0);

}

void 

SocketSetDontFragmentTag::Deserialize (TagBuffer i)

{

  m_dontFragment = (i.ReadU8 () == 1)?true:false;

}

void 

SocketSetDontFragmentTag::Print (std::ostream &os) const

{