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

/*

 * Copyright (c) 2006, 2009 INRIA

 * Copyright (c) 2009 MIRKO BANCHI

 *

 * This program is free software; you can redistribute it and/or modify

 * 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>

 * Author: Mirko Banchi <mk.banchi@gmail.com>

 */

#include "ns3/assert.h"

#include "ns3/log.h"

#include "ns3/simulator.h"

#include "ns3/string.h"

#include "ns3/pointer.h"

#include "ns3/boolean.h"

#include "qos-tag.h"

#include "wifi-phy.h"

#include "dcf-manager.h"

#include "mac-rx-middle.h"

#include "mac-tx-middle.h"

#include "mgt-headers.h"

#include "mac-low.h"

#include "amsdu-subframe-header.h"

#include "msdu-aggregator.h"

namespace ns3 {

TypeId

{

    .AddAttribute ("BeaconInterval", "Delay between two beacons",

                   TimeValue (MicroSeconds (102400)),

                   MakeTimeChecker ())

    .AddAttribute ("BeaconGeneration", "Whether or not beacons are generated.",

                   BooleanValue (true),

                   MakeBooleanChecker ())

  return tid;

}

{

  NS_LOG_FUNCTION (this);

  m_beaconDca = CreateObject<DcaTxop> ();

  m_beaconDca->SetAifsn(1);

  m_beaconDca->SetMinCw(0);

  m_beaconDca->SetMaxCw(0);

  m_beaconDca->SetLow (m_low);

  m_beaconDca->SetManager (m_dcfManager);

  m_enableBeaconGeneration = false;

}

{

  NS_LOG_FUNCTION (this);

}

void

{

  m_beaconDca = 0;

  m_enableBeaconGeneration = false;

  m_beaconEvent.Cancel ();

}

void

{

  NS_LOG_FUNCTION (this << enable);

  if (!enable)

    {

      m_beaconEvent.Cancel ();

    }

  else if (enable && !m_enableBeaconGeneration)

    {

    }

  m_enableBeaconGeneration = enable;

}

bool

{

  return m_enableBeaconGeneration;

}

Time

{

  return m_beaconInterval;

}

{

}

void

{

}

void

{

  NS_LOG_FUNCTION (this << interval);

  if ((interval.GetMicroSeconds () % 1024) != 0)

    {

      NS_LOG_WARN ("beacon interval should be multiple of 1024us, see IEEE Std. 802.11-2007, section 11.1.1.1");

    }

  m_beaconInterval = interval;

}

void

{

  NS_LOG_FUNCTION (this);

  SendOneBeacon ();

}

void

{

}

void

{

  NS_LOG_FUNCTION (this << packet << from << to);

  WifiMacHeader hdr;

    {

      hdr.SetQosTid (tid);

    }

  else

    {

    }

  hdr.SetAddr1 (to);

  hdr.SetAddr2 (GetAddress ());

  hdr.SetAddr3 (from);

  hdr.SetDsFrom ();

  hdr.SetDsNotTo ();

}

void

{

  if (to.IsBroadcast () || m_stationManager->IsAssociated (to))

    {

      ForwardDown (packet, from, to);

    }

}

{

}

{

  return true;

}

SupportedRates

{

  // send the set of supported rates and make sure that we indicate

  // the Basic Rate set in this set of supported rates.

  SupportedRates rates;

  for (uint32_t i = 0; i < m_phy->GetNModes (); i++)

    {

      WifiMode mode = m_phy->GetMode (i);

      rates.AddSupportedRate (mode.GetDataRate ());

    }

  // set the basic rates

  for (uint32_t j = 0; j < m_stationManager->GetNBasicModes (); j++)

    {

      WifiMode mode = m_stationManager->GetBasicMode (j);

      rates.SetBasicRate (mode.GetDataRate ());

    }

}

void

{

  NS_LOG_FUNCTION (this << to);

  WifiMacHeader hdr;

  hdr.SetProbeResp ();

  hdr.SetAddr1 (to);

  hdr.SetAddr2 (GetAddress ());

  hdr.SetAddr3 (GetAddress ());

  hdr.SetDsNotFrom ();

  hdr.SetDsNotTo ();

  Ptr<Packet> packet = Create<Packet> ();

  MgtProbeResponseHeader probe;

  probe.SetSsid (GetSsid ());

  probe.SetSupportedRates (GetSupportedRates ());

  probe.SetBeaconIntervalUs (m_beaconInterval.GetMicroSeconds ());

  packet->AddHeader (probe);

}

void

{

  NS_LOG_FUNCTION (this << to << success);

  WifiMacHeader hdr;

  hdr.SetAssocResp ();

  hdr.SetAddr1 (to);

  hdr.SetAddr2 (GetAddress ());

  hdr.SetAddr3 (GetAddress ());

  hdr.SetDsNotFrom ();

  hdr.SetDsNotTo ();

  Ptr<Packet> packet = Create<Packet> ();

  MgtAssocResponseHeader assoc;

  StatusCode code;

  if (success)

    {

      code.SetSuccess ();

    }

  else

    {

      code.SetFailure ();

    }

  assoc.SetSupportedRates (GetSupportedRates ());

  assoc.SetStatusCode (code);

  packet->AddHeader (assoc);

}

void

{

  NS_LOG_FUNCTION (this);

  WifiMacHeader hdr;

  hdr.SetBeacon ();

  hdr.SetAddr1 (Mac48Address::GetBroadcast ());

  hdr.SetAddr2 (GetAddress ());

  hdr.SetAddr3 (GetAddress ());

  hdr.SetDsNotFrom ();

  hdr.SetDsNotTo ();

  Ptr<Packet> packet = Create<Packet> ();

  MgtBeaconHeader beacon;

  beacon.SetSsid (GetSsid ());

  beacon.SetSupportedRates (GetSupportedRates ());

  beacon.SetBeaconIntervalUs (m_beaconInterval.GetMicroSeconds ());

  packet->AddHeader (beacon);

  m_beaconDca->Queue (packet, hdr);

}

void

{

  NS_LOG_FUNCTION (this);

    {

      NS_LOG_DEBUG ("associated with sta="<<hdr.GetAddr1 ());

      m_stationManager->RecordGotAssocTxOk (hdr.GetAddr1 ());

    }

}

void

{

  NS_LOG_FUNCTION (this);

    {

      NS_LOG_DEBUG ("assoc failed with sta="<<hdr.GetAddr1 ());

      m_stationManager->RecordGotAssocTxFailed (hdr.GetAddr1 ());

    }

}

void

{

  NS_LOG_FUNCTION (this << packet << hdr);

  Mac48Address from = hdr->GetAddr2 ();

  if (hdr->IsData ())

    {

      Mac48Address bssid = hdr->GetAddr1 ();

          hdr->IsToDs () &&

          bssid == GetAddress () &&

          m_stationManager->IsAssociated (from))

        {

          Mac48Address to = hdr->GetAddr3 ();

          if (to == GetAddress ())

            {

              if (hdr->IsQosData ())

                {

                  if (hdr->IsQosAmsdu ())

                    {

                      NS_LOG_DEBUG ("Received A-MSDU from="<<from<<", size="<<packet->GetSize ());

                      DeaggregateAmsduAndForward (packet, hdr);

                      packet = 0;

                    }

                  else

                    {

                      ForwardUp (packet, from, bssid);

                    }

                }

              else

                {

                  ForwardUp (packet, from, bssid);

                }

            }

          else if (to.IsGroup () ||

                   m_stationManager->IsAssociated (to))

            {

              NS_LOG_DEBUG ("forwarding frame from="<<from<<", to="<<to);

              Ptr<Packet> copy = packet->Copy ();

              ForwardUp (copy, from, to);

            }

          else

            {

              ForwardUp (packet, from, to);

            }

        }

      else if (hdr->IsFromDs () &&

        {

          // this is an AP-to-AP frame

          // we ignore for now.

        }

        {

          // they are not targeted at the AP

        }

    }

  else if (hdr->IsMgt ())

    {

        {

          NS_ASSERT (hdr->GetAddr1 ().IsBroadcast ());

        }

        {

            {

              // first, verify that the the station's supported

              // rate set is compatible with our Basic Rate set

              MgtAssocRequestHeader assocReq;

              packet->RemoveHeader (assocReq);

              SupportedRates rates = assocReq.GetSupportedRates ();

              bool problem = false;

              for (uint32_t i = 0; i < m_stationManager->GetNBasicModes (); i++)

                {

                  WifiMode mode = m_stationManager->GetBasicMode (i);

                  if (!rates.IsSupportedRate (mode.GetDataRate ()))