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

/*

 * Copyright (c) 2005,2006 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/assert.h"

#include "ns3/packet.h"

#include "ns3/simulator.h"

#include "ns3/log.h"

#include "ns3/node.h"

#include "mac-low.h"

#include "wifi-phy.h"

#include "wifi-mac-trailer.h"

#include "wifi-mac.h"

NS_LOG_COMPONENT_DEFINE ("MacLow");

#define MY_DEBUG(x) \

  NS_LOG_DEBUG (m_mac->GetAddress () << " " << x)

namespace ns3 {

class SnrTag : public Tag

{

public:

  static TypeId GetTypeId (void);

  virtual TypeId GetInstanceTypeId (void) const;

  virtual uint32_t GetSerializedSize (void) const;

  virtual void Serialize (TagBuffer i) const;

  virtual void Deserialize (TagBuffer i);

  void Set (double snr);

  double Get (void) const;

private:

  double m_snr;

};

TypeId 

SnrTag::GetTypeId (void)

{

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

    .SetParent<Tag> ()

    .AddConstructor<SnrTag> ()

    .AddAttribute ("Snr", "The snr of the last packet received",

                   DoubleValue (0.0),

                   MakeDoubleAccessor (&SnrTag::Get),

                   MakeDoubleChecker<double> ())

    ;

  return tid;

}

TypeId 

SnrTag::GetInstanceTypeId (void) const

{

  return GetTypeId ();

}

uint32_t 

SnrTag::GetSerializedSize (void) const

{

  return sizeof (double);

}

void 

SnrTag::Serialize (TagBuffer i) const

{

  i.WriteDouble (m_snr);

}

void 

SnrTag::Deserialize (TagBuffer i)

{

  m_snr = i.ReadDouble ();

}

void 

SnrTag::Set (double snr)

{

  m_snr = snr;

}

double 

SnrTag::Get (void) const

{

  return m_snr;

}

MacLowTransmissionListener::MacLowTransmissionListener ()

{}

MacLowTransmissionListener::~MacLowTransmissionListener ()

{}

MacLowNavListener::MacLowNavListener ()

{}

MacLowNavListener::~MacLowNavListener ()

{}

MacLowTransmissionParameters::MacLowTransmissionParameters ()

  : m_nextSize (0),

    m_waitAck (ACK_NONE),

    m_sendRts (false),

    m_overrideDurationId (Seconds (0))

{}

void 

MacLowTransmissionParameters::EnableNextData (uint32_t size)

{

  m_nextSize = size;

}

void 

MacLowTransmissionParameters::DisableNextData (void)

{

  m_nextSize = 0;

}

void 

MacLowTransmissionParameters::EnableOverrideDurationId (Time durationId)

{

  m_overrideDurationId = durationId;

}

void 

MacLowTransmissionParameters::DisableOverrideDurationId (void)

{

  m_overrideDurationId = Seconds (0);

}

void 

MacLowTransmissionParameters::EnableSuperFastAck (void)

{

  m_waitAck = ACK_SUPER_FAST;

}

void 

MacLowTransmissionParameters::EnableFastAck (void)

{

  m_waitAck = ACK_FAST;

}

void 

MacLowTransmissionParameters::EnableAck (void)

{

  m_waitAck = ACK_NORMAL;

}

void 

MacLowTransmissionParameters::DisableAck (void)

{

  m_waitAck = ACK_NONE;

}

void 

MacLowTransmissionParameters::EnableRts (void)

{

  m_sendRts = true;

}

void 

MacLowTransmissionParameters::DisableRts (void)

{

  m_sendRts = false;

}

bool 

MacLowTransmissionParameters::MustWaitAck (void) const

{

  return (m_waitAck != ACK_NONE)?true:false;

}

bool 

MacLowTransmissionParameters::MustWaitNormalAck (void) const

{

  return (m_waitAck == ACK_NORMAL)?true:false;

}

bool 

MacLowTransmissionParameters::MustWaitFastAck (void) const

{

  return (m_waitAck == ACK_FAST)?true:false;

}

bool 

MacLowTransmissionParameters::MustWaitSuperFastAck (void) const

{

  return (m_waitAck == ACK_SUPER_FAST)?true:false;

}

bool 

MacLowTransmissionParameters::MustSendRts (void) const

{

  return m_sendRts;

}

bool 

MacLowTransmissionParameters::HasDurationId (void) const

{

  return (m_overrideDurationId != Seconds (0))?true:false;

}

Time

MacLowTransmissionParameters::GetDurationId (void) const

{

  NS_ASSERT (m_overrideDurationId != Seconds (0));

  return m_overrideDurationId;

}

bool 

MacLowTransmissionParameters::HasNextPacket (void) const

{

  return (m_nextSize != 0)?true:false;

}

uint32_t 

MacLowTransmissionParameters::GetNextPacketSize (void) const

{

  NS_ASSERT (HasNextPacket ());

  return m_nextSize;

}

std::ostream &operator << (std::ostream &os, const MacLowTransmissionParameters &params)

{

  os << "[" 

     << "send rts=" << params.m_sendRts << ", "

     << "next size=" << params.m_nextSize << ", "

     << "dur=" << params.m_overrideDurationId << ", "

     << "ack=";

  switch (params.m_waitAck) {

  case MacLowTransmissionParameters::ACK_NONE:

    os << "none";

    break;

  case MacLowTransmissionParameters::ACK_NORMAL:

    os << "normal";

    break;

  case MacLowTransmissionParameters::ACK_FAST:

    os << "fast";

    break;

  case MacLowTransmissionParameters::ACK_SUPER_FAST:

    os << "super-fast";

    break;

  }

  os << "]";

  return os;

}

MacLow::MacLow ()

  : m_normalAckTimeoutEvent (),

    m_fastAckTimeoutEvent (),

    m_superFastAckTimeoutEvent (),

    m_fastAckFailedTimeoutEvent (),

    m_ctsTimeoutEvent (),

    m_sendCtsEvent (),

    m_sendAckEvent (),

    m_sendDataEvent (),

    m_waitSifsEvent (),

    m_currentPacket (0),

    m_listener (0)

{

  NS_LOG_FUNCTION (this);

  m_lastNavDuration = Seconds (0);

  m_lastNavStart = Seconds (0);

}

MacLow::~MacLow ()

{

  NS_LOG_FUNCTION (this);

}

void 

MacLow::DoDispose (void)

{

  NS_LOG_FUNCTION (this);

  CancelAllEvents ();

  m_phy = 0;

  m_mac = 0;

  m_stationManager = 0;

}

void

MacLow::CancelAllEvents (void)

{

  NS_LOG_FUNCTION (this);

  bool oneRunning = false;

  if (m_normalAckTimeoutEvent.IsRunning ()) 

    {

      m_normalAckTimeoutEvent.Cancel ();

      oneRunning = true;

    }

  if (m_fastAckTimeoutEvent.IsRunning ()) 

    {

      m_fastAckTimeoutEvent.Cancel ();

      oneRunning = true;

    }

  if (m_superFastAckTimeoutEvent.IsRunning ()) 

    {

      m_superFastAckTimeoutEvent.Cancel ();

      oneRunning = true;

    }

  if (m_fastAckFailedTimeoutEvent.IsRunning ()) 

    {

      m_fastAckFailedTimeoutEvent.Cancel ();

      oneRunning = true;

    }

  if (m_ctsTimeoutEvent.IsRunning ()) 

    {

      m_ctsTimeoutEvent.Cancel ();

      oneRunning = true;

    }

  if (m_sendCtsEvent.IsRunning ()) 

    {

      m_sendCtsEvent.Cancel ();

      oneRunning = true;

    }

  if (m_sendAckEvent.IsRunning ()) 

    {

      m_sendAckEvent.Cancel ();

      oneRunning = true;

    }

  if (m_sendDataEvent.IsRunning ()) 

    {

      m_sendDataEvent.Cancel ();

      oneRunning = true;

    }

  if (m_waitSifsEvent.IsRunning ()) 

    {

      m_waitSifsEvent.Cancel ();

      oneRunning = true;

    }

  if (oneRunning && m_listener != 0) 

    {

      m_listener->Cancel ();

      m_listener = 0;

    }

}

void

MacLow::SetPhy (Ptr<WifiPhy> phy)

{

  m_phy = phy;

  m_phy->SetReceiveOkCallback (MakeCallback (&MacLow::ReceiveOk, this));

  m_phy->SetReceiveErrorCallback (MakeCallback (&MacLow::ReceiveError, this));

}

void

MacLow::SetMac (Ptr<WifiMac> mac)

{

  m_mac = mac;

}

void 

MacLow::SetWifiRemoteStationManager (Ptr<WifiRemoteStationManager> manager)

{

  m_stationManager = manager;

}

Ptr<WifiMac> 

MacLow::GetMac (void)

{

  return m_mac;

}

void 

MacLow::SetRxCallback (Callback<void,Ptr<Packet>,const WifiMacHeader *> callback)

{

  m_rxCallback = callback;

}

void 

MacLow::RegisterNavListener (MacLowNavListener *listener)

{

  m_navListeners.push_back (listener);

}

void 

MacLow::StartTransmission (Ptr<const Packet> packet, 

                           WifiMacHeader const*hdr, 

                           MacLowTransmissionParameters params,

                           MacLowTransmissionListener *listener)

{

  NS_LOG_FUNCTION (this << packet << hdr << params << listener);

  /* m_currentPacket is not NULL because someone started

   * a transmission and was interrupted before one of:

   *   - ctsTimeout

   *   - sendDataAfterCTS

   * expired. This means that one of these timers is still

   * running. They are all cancelled below anyway by the 

   * call to CancelAllEvents (because of at least one

   * of these two timer) which will trigger a call to the

   * previous listener's cancel method.

   *

   * This typically happens because the high-priority 

   * QapScheduler has taken access to the channel from

   * one of the Edca of the QAP.

   */

  m_currentPacket = packet->Copy ();

  m_currentHdr = *hdr;

  CancelAllEvents ();

  m_listener = listener;

  m_txParams = params;

  //NS_ASSERT (m_phy->IsStateIdle ());

  MY_DEBUG ("startTx size="<< GetSize (m_currentPacket, &m_currentHdr) << 

            ", to=" << m_currentHdr.GetAddr1()<<", listener="<<m_listener);

  if (m_txParams.MustSendRts ()) 

    {

      SendRtsForPacket ();

    } 

  else 

    {

      SendDataPacket ();

    }

  /* When this method completes, we have taken ownership of the medium. */

  NS_ASSERT (m_phy->IsStateTx ());  

}

void

MacLow::ReceiveError (Ptr<Packet> packet, double rxSnr)

{

  NS_LOG_FUNCTION (this << packet << rxSnr);

  MY_DEBUG ("rx failed ");

  if (m_txParams.MustWaitFastAck ()) 

    {

      NS_ASSERT (m_fastAckFailedTimeoutEvent.IsExpired ());

      m_fastAckFailedTimeoutEvent = Simulator::Schedule (GetSifs (), 

                                                         &MacLow::FastAckFailedTimeout, this);

    }

  return;

}

void 

MacLow::ReceiveOk (Ptr<Packet> packet, double rxSnr, WifiMode txMode, WifiPreamble preamble)

{

  NS_LOG_FUNCTION (this << packet << rxSnr << txMode << preamble);

  /* A packet is received from the PHY.

   * When we have handled this packet,

   * we handle any packet present in the

   * packet queue.

   */

  WifiMacHeader hdr;

  packet->RemoveHeader (hdr);

  bool isPrevNavZero = IsNavZero ();

  MY_DEBUG ("duration/id=" << hdr.GetDuration ());

  NotifyNav (hdr, txMode, preamble);

  if (hdr.IsRts ()) 

    {

      /* see section 9.2.5.7 802.11-1999

       * A STA that is addressed by an RTS frame shall transmit a CTS frame after a SIFS 

       * period if the NAV at the STA receiving the RTS frame indicates that the medium is 

       * idle. If the NAV at the STA receiving the RTS indicates the medium is not idle, 

       * that STA shall not respond to the RTS frame.

       */

      if (isPrevNavZero &&

          hdr.GetAddr1 () == m_mac->GetAddress ()) 

        {

          MY_DEBUG ("rx RTS from=" << hdr.GetAddr2 () << ", schedule CTS");

          NS_ASSERT (m_sendCtsEvent.IsExpired ());

          WifiRemoteStation *station = GetStation (hdr.GetAddr2 ());

          station->ReportRxOk (rxSnr, txMode);

          m_sendCtsEvent = Simulator::Schedule (GetSifs (),

                                                &MacLow::SendCtsAfterRts, this,

                                                hdr.GetAddr2 (), 

                                                hdr.GetDuration (),

                                                txMode,

                                                rxSnr);

        } 

      else 

        {

          MY_DEBUG ("rx RTS from=" << hdr.GetAddr2 () << ", cannot schedule CTS");

        }

    } 

  else if (hdr.IsCts () &&

           hdr.GetAddr1 () == m_mac->GetAddress () &&

           m_ctsTimeoutEvent.IsRunning () &&

           m_currentPacket != 0) 

    {

      MY_DEBUG ("receive cts from="<<m_currentHdr.GetAddr1 ());

      SnrTag tag;

      packet->FindFirstMatchingTag (tag);

      WifiRemoteStation *station = GetStation (m_currentHdr.GetAddr1 ());

      station->ReportRxOk (rxSnr, txMode);

      station->ReportRtsOk (rxSnr, txMode, tag.Get ());

      m_ctsTimeoutEvent.Cancel ();

      m_listener->GotCts (rxSnr, txMode);

      NS_ASSERT (m_sendDataEvent.IsExpired ());

      m_sendDataEvent = Simulator::Schedule (GetSifs (), 

                                             &MacLow::SendDataAfterCts, this, 

                                             hdr.GetAddr1 (),

                                             hdr.GetDuration (),

                                             txMode);

    } 

  else if (hdr.IsAck () &&

           hdr.GetAddr1 () == m_mac->GetAddress () &&

           (m_normalAckTimeoutEvent.IsRunning () || 

            m_fastAckTimeoutEvent.IsRunning () ||

            m_superFastAckTimeoutEvent.IsRunning ()) &&

           m_txParams.MustWaitAck ()) 

    {

      MY_DEBUG ("receive ack from="<<m_currentHdr.GetAddr1 ());

      SnrTag tag;

      packet->FindFirstMatchingTag (tag);

      WifiRemoteStation *station = GetStation (m_currentHdr.GetAddr1 ());

      station->ReportRxOk (rxSnr, txMode);

      station->ReportDataOk (rxSnr, txMode, tag.Get ());

      bool gotAck = false;

      if (m_txParams.MustWaitNormalAck () &&

          m_normalAckTimeoutEvent.IsRunning ()) 

        {

          m_normalAckTimeoutEvent.Cancel ();

          gotAck = true;

        }

      if (m_txParams.MustWaitFastAck () &&

          m_fastAckTimeoutEvent.IsRunning ()) 

        {

          m_fastAckTimeoutEvent.Cancel ();