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

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

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

 */

#include "ns3/assert.h"

#include "ns3/address-utils.h"

#include "wifi-mac-header.h"

namespace ns3 {

NS_OBJECT_ENSURE_REGISTERED (WifiMacHeader);

enum {

  TYPE_MGT = 0,

  TYPE_CTL  = 1,

  TYPE_DATA = 2

};

enum {

  SUBTYPE_CTL_BACKREQ = 8,

  SUBTYPE_CTL_BACKRESP = 9,

  SUBTYPE_CTL_RTS = 11,

  SUBTYPE_CTL_CTS = 12,

  SUBTYPE_CTL_ACK = 13

};

WifiMacHeader::WifiMacHeader ()

  : m_ctrlPwrMgt (0),

    m_ctrlMoreData (0),

    m_ctrlWep (0),

    m_ctrlOrder (1),

    m_amsduPresent (0)

{}

WifiMacHeader::~WifiMacHeader ()

{}

void 

WifiMacHeader::SetDsFrom (void)

{

  m_ctrlFromDs = 1;

}

void 

WifiMacHeader::SetDsNotFrom (void)

{

  m_ctrlFromDs = 0;

}

void 

WifiMacHeader::SetDsTo (void)

{

  m_ctrlToDs = 1;

}

void 

WifiMacHeader::SetDsNotTo (void)

{

  m_ctrlToDs = 0;

}

void 

WifiMacHeader::SetAddr1 (Mac48Address address)

{

  m_addr1 = address;

}

void 

WifiMacHeader::SetAddr2 (Mac48Address address)

{

  m_addr2 = address;

}

void 

WifiMacHeader::SetAddr3 (Mac48Address address)

{

  m_addr3 = address;

}

void 

WifiMacHeader::SetAddr4 (Mac48Address address)

{

  m_addr4 = address;

}

void 

WifiMacHeader::SetAssocReq (void)

{

  m_ctrlType = TYPE_MGT;

  m_ctrlSubtype = 0;

}

void 

WifiMacHeader::SetAssocResp (void)

{

  m_ctrlType = TYPE_MGT;

  m_ctrlSubtype = 1;

}

void 

WifiMacHeader::SetProbeReq (void)

{

  m_ctrlType = TYPE_MGT;

  m_ctrlSubtype = 4;

}

void 

WifiMacHeader::SetProbeResp (void)

{

  m_ctrlType = TYPE_MGT;

  m_ctrlSubtype = 5;

}

void 

WifiMacHeader::SetBeacon (void)

{

  m_ctrlType = TYPE_MGT;

  m_ctrlSubtype = 8;

}

void 

WifiMacHeader::SetTypeData (void)

{

  m_ctrlType = TYPE_DATA;

  m_ctrlSubtype = 0;

}

void

WifiMacHeader::SetAction (void)

{

 m_ctrlType = TYPE_MGT;

 m_ctrlSubtype = 0x0D;

}

void

WifiMacHeader::SetMultihopAction (void)

{

 m_ctrlType = TYPE_MGT;

 m_ctrlSubtype = 0x0F;

}

void 

WifiMacHeader::SetType (enum WifiMacType type)

{

  switch (type) {

  case WIFI_MAC_CTL_BACKREQ:

    m_ctrlType = TYPE_CTL;

    m_ctrlSubtype = SUBTYPE_CTL_BACKREQ;

    break;

  case WIFI_MAC_CTL_BACKRESP:

    m_ctrlType = TYPE_CTL;

    m_ctrlSubtype = SUBTYPE_CTL_BACKRESP;

    break;

  case WIFI_MAC_CTL_RTS:

    m_ctrlType = TYPE_CTL;

    m_ctrlSubtype = SUBTYPE_CTL_RTS;

    break;

  case WIFI_MAC_CTL_CTS:

    m_ctrlType = TYPE_CTL;

    m_ctrlSubtype = SUBTYPE_CTL_CTS;

    break;

  case WIFI_MAC_CTL_ACK:

    m_ctrlType = TYPE_CTL;

    m_ctrlSubtype = SUBTYPE_CTL_ACK;

    break;

  case WIFI_MAC_MGT_ASSOCIATION_REQUEST:

    m_ctrlType = TYPE_MGT;

    m_ctrlSubtype = 0;

    break;

  case WIFI_MAC_MGT_ASSOCIATION_RESPONSE:

    m_ctrlType = TYPE_MGT;

    m_ctrlSubtype = 1;

    break;

  case WIFI_MAC_MGT_REASSOCIATION_REQUEST:

    m_ctrlType = TYPE_MGT;

    m_ctrlSubtype = 2;

    break;

  case WIFI_MAC_MGT_REASSOCIATION_RESPONSE:

    m_ctrlType = TYPE_MGT;

    m_ctrlSubtype = 3;

    break;

  case WIFI_MAC_MGT_PROBE_REQUEST:

    m_ctrlType = TYPE_MGT;

    m_ctrlSubtype = 4;

    break;

  case WIFI_MAC_MGT_PROBE_RESPONSE:

    m_ctrlType = TYPE_MGT;

    m_ctrlSubtype = 5;

    break;

  case WIFI_MAC_MGT_BEACON:

    m_ctrlType = TYPE_MGT;

    m_ctrlSubtype = 8;

    break;

  case WIFI_MAC_MGT_DISASSOCIATION:

    m_ctrlType = TYPE_MGT;

    m_ctrlSubtype = 10;

    break;

  case WIFI_MAC_MGT_AUTHENTICATION:

    m_ctrlType = TYPE_MGT;

    m_ctrlSubtype = 11;

    break;

  case WIFI_MAC_MGT_DEAUTHENTICATION:

    m_ctrlType = TYPE_MGT;

    m_ctrlSubtype = 12;

  case WIFI_MAC_MGT_ACTION:

    m_ctrlType = TYPE_MGT;

    m_ctrlSubtype = 13;

  case WIFI_MAC_MGT_ACTION_NO_ACK:

    m_ctrlType = TYPE_MGT;

    m_ctrlSubtype = 14;

  case WIFI_MAC_MGT_MULTIHOP_ACTION:

    m_ctrlType = TYPE_MGT;

    m_ctrlSubtype = 15;

    break;

  case WIFI_MAC_DATA:

    m_ctrlType = TYPE_DATA;

    m_ctrlSubtype = 0;

    break;

  case WIFI_MAC_DATA_CFACK:

    m_ctrlType = TYPE_DATA;

    m_ctrlSubtype = 1;

    break;

  case WIFI_MAC_DATA_CFPOLL:

    m_ctrlType = TYPE_DATA;

    m_ctrlSubtype = 2;

    break;

  case WIFI_MAC_DATA_CFACK_CFPOLL:

    m_ctrlType = TYPE_DATA;

    m_ctrlSubtype = 3;

    break;

  case WIFI_MAC_DATA_NULL:

    m_ctrlType = TYPE_DATA;

    m_ctrlSubtype = 4;

    break;

  case WIFI_MAC_DATA_NULL_CFACK:

    m_ctrlType = TYPE_DATA;

    m_ctrlSubtype = 5;

    break;

  case WIFI_MAC_DATA_NULL_CFPOLL:

    m_ctrlType = TYPE_DATA;

    m_ctrlSubtype = 6;

    break;

  case WIFI_MAC_DATA_NULL_CFACK_CFPOLL:

    m_ctrlType = TYPE_DATA;

    m_ctrlSubtype = 7;

    break;

  case WIFI_MAC_QOSDATA:

    m_ctrlType = TYPE_DATA;

    m_ctrlSubtype = 8;

    break;

  case WIFI_MAC_QOSDATA_CFACK:

    m_ctrlType = TYPE_DATA;

    m_ctrlSubtype = 9;

    break;

  case WIFI_MAC_QOSDATA_CFPOLL:

    m_ctrlType = TYPE_DATA;

    m_ctrlSubtype = 10;

    break;

  case WIFI_MAC_QOSDATA_CFACK_CFPOLL:

    m_ctrlType = TYPE_DATA;

    m_ctrlSubtype = 11;

    break;

  case WIFI_MAC_QOSDATA_NULL:

    m_ctrlType = TYPE_DATA;

    m_ctrlSubtype = 12;

    break;

  case WIFI_MAC_QOSDATA_NULL_CFPOLL:

    m_ctrlType = TYPE_DATA;

    m_ctrlSubtype = 14;

    break;

  case WIFI_MAC_QOSDATA_NULL_CFACK_CFPOLL:

    m_ctrlType = TYPE_DATA;

    m_ctrlSubtype = 15;

    break;

  }

  m_ctrlToDs = 0;

  m_ctrlFromDs = 0;

}

void 

WifiMacHeader::SetRawDuration (uint16_t duration)

{

  m_duration = duration;

}

void 

WifiMacHeader::SetDuration (Time duration)

{

  int64_t duration_us = duration.GetMicroSeconds ();

  NS_ASSERT (duration_us >= 0 && duration_us <= 0x7fff);

  m_duration = static_cast<uint16_t> (duration_us);

}

void WifiMacHeader::SetId (uint16_t id)

{

  m_duration = id;

}

void WifiMacHeader::SetSequenceNumber (uint16_t seq)

{

  m_seqSeq = seq;

}

void WifiMacHeader::SetFragmentNumber (uint8_t frag)

{

  m_seqFrag = frag;

}

void WifiMacHeader::SetNoMoreFragments (void)

{

  m_ctrlMoreFrag = 0;

}

void WifiMacHeader::SetMoreFragments (void)

{

  m_ctrlMoreFrag = 1;

}

void WifiMacHeader::SetRetry (void)

{

  m_ctrlRetry = 1;

}

void WifiMacHeader::SetNoRetry (void)

{

  m_ctrlRetry = 0;

}

void WifiMacHeader::SetQosTid (uint8_t tid)

{

  m_qosTid = tid;

}

void WifiMacHeader::SetQosEosp ()

{

  m_qosEosp = 1;

}

void WifiMacHeader::SetQosNoEosp ()

{

  m_qosEosp = 0;

}

void WifiMacHeader::SetQosAckPolicy (enum QosAckPolicy policy)

{

  switch (policy) {

  case NORMAL_ACK :

    m_qosAckPolicy = 0;

    break;

  case NO_ACK :

    m_qosAckPolicy = 1;

    break;

  case NO_EXPLICIT_ACK :

    m_qosAckPolicy = 2;

    break;

  case BLOCK_ACK :

    m_qosAckPolicy = 3;

    break;

  }

}

void WifiMacHeader::SetQosAmsdu (void)

{

  m_amsduPresent = 1;

}

void WifiMacHeader::SetQosNoAmsdu (void)

{

  m_amsduPresent = 0;

}

void WifiMacHeader::SetQosTxopLimit (uint8_t txop)

{

  m_qosStuff = txop;

}

Mac48Address 

WifiMacHeader::GetAddr1 (void) const

{

  return m_addr1;

}

Mac48Address 

WifiMacHeader::GetAddr2 (void) const

{

  return m_addr2;

}

Mac48Address 

WifiMacHeader::GetAddr3 (void) const

{

  return m_addr3;

}

Mac48Address 

WifiMacHeader::GetAddr4 (void) const

{

  return m_addr4;

}

enum WifiMacType 

WifiMacHeader::GetType (void) const

{

  switch (m_ctrlType) {

  case TYPE_MGT:

    switch (m_ctrlSubtype) {

    case 0:

      return WIFI_MAC_MGT_ASSOCIATION_REQUEST;

      break;

    case 1:

      return WIFI_MAC_MGT_ASSOCIATION_RESPONSE;

      break;

    case 2:

      return WIFI_MAC_MGT_REASSOCIATION_REQUEST;

      break;

    case 3:

      return WIFI_MAC_MGT_REASSOCIATION_RESPONSE;

      break;

    case 4:

      return WIFI_MAC_MGT_PROBE_REQUEST;

      break;

    case 5:

      return WIFI_MAC_MGT_PROBE_RESPONSE;

      break;

    case 8:

      return WIFI_MAC_MGT_BEACON;

      break;

    case 10:

      return WIFI_MAC_MGT_DISASSOCIATION;

      break;

    case 11:

      return WIFI_MAC_MGT_AUTHENTICATION;

      break;

    case 12:

      return WIFI_MAC_MGT_DEAUTHENTICATION;

      break;

    case 13:

      return WIFI_MAC_MGT_ACTION;

      break;

    case 14:

      return WIFI_MAC_MGT_ACTION_NO_ACK;

      break;

    case 15:

      return WIFI_MAC_MGT_MULTIHOP_ACTION;

      break;

    }

    break;

  case TYPE_CTL:

    switch (m_ctrlSubtype) {

    case SUBTYPE_CTL_BACKREQ:

      return WIFI_MAC_CTL_BACKREQ;

      break;

    case SUBTYPE_CTL_BACKRESP:

      return WIFI_MAC_CTL_BACKRESP;

      break;

    case SUBTYPE_CTL_RTS:

      return WIFI_MAC_CTL_RTS;

      break;

    case SUBTYPE_CTL_CTS:

      return WIFI_MAC_CTL_CTS;

      break;

    case SUBTYPE_CTL_ACK:

      return WIFI_MAC_CTL_ACK;

      break;

    }

    break;

  case TYPE_DATA:

    switch (m_ctrlSubtype) {

    case 0:

      return WIFI_MAC_DATA;

      break;

    case 1:

      return WIFI_MAC_DATA_CFACK;

      break;

    case 2:

      return WIFI_MAC_DATA_CFPOLL;

      break;

    case 3:

      return WIFI_MAC_DATA_CFACK_CFPOLL;

      break;

    case 4:

      return WIFI_MAC_DATA_NULL;

      break;

    case 5:

      return WIFI_MAC_DATA_NULL_CFACK;

      break;

    case 6:

      return WIFI_MAC_DATA_NULL_CFPOLL;

      break;

    case 7:

      return WIFI_MAC_DATA_NULL_CFACK_CFPOLL;

      break;

    case 8:

      return WIFI_MAC_QOSDATA;

      break;

    case 9:

      return WIFI_MAC_QOSDATA_CFACK;

      break;

    case 10:

      return WIFI_MAC_QOSDATA_CFPOLL;

      break;

    case 11:

      return WIFI_MAC_QOSDATA_CFACK_CFPOLL;

      break;

    case 12:

      return WIFI_MAC_QOSDATA_NULL;

      break;

    case 14:

      return WIFI_MAC_QOSDATA_NULL_CFPOLL;

      break;

    case 15:

      return WIFI_MAC_QOSDATA_NULL_CFACK_CFPOLL;

      break;

    }

    break;

  }

  // NOTREACHED

  NS_ASSERT (false);

  return (enum WifiMacType)-1;

}

bool 

WifiMacHeader::IsFromDs (void) const

{

  return m_ctrlFromDs == 1;

}

bool 

WifiMacHeader::IsToDs (void) const

{

  return m_ctrlToDs == 1;

}

bool 

WifiMacHeader::IsData (void) const

{