/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/*
 * Copyright (c) 2009 IITP RAS
 *
 * 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: Kirill Andreev <andreev@iitp.ru>
 */

#include "peer-link-frame.h"
#include "ie-dot11s-peer-management.h"
#include "ns3/mesh-wifi-interface-mac.h"
#include "ns3/test.h"
#include "ns3/packet.h"

namespace ns3 {
namespace dot11s {
NS_OBJECT_ENSURE_REGISTERED (PeerLinkFrameStart);

PeerLinkFrameStart::PeerLinkFrameStart () :
  m_subtype (255), m_capability (0), m_aid (0), m_rates (SupportedRates ()), m_meshId (IeMeshId ()),
      m_config (IeConfiguration ()), m_reasonCode ((uint16_t) REASON11S_RESERVED)
{
}
void
PeerLinkFrameStart::SetPlinkFrameSubtype (uint8_t subtype)
{
  m_subtype = subtype;
}
void
PeerLinkFrameStart::SetPlinkFrameStart (PeerLinkFrameStart::PlinkFrameStartFields fields)
{
  m_subtype = fields.subtype;
  m_protocol = fields.protocol;
  if (m_subtype != (uint8_t) (WifiMeshActionHeader::PEER_LINK_CLOSE))
    {
      m_capability = fields.capability;
    }
  if (m_subtype == (uint8_t) (WifiMeshActionHeader::PEER_LINK_CONFIRM))
    {
      m_aid = fields.aid;
    }
  if (m_subtype != (uint8_t) (WifiMeshActionHeader::PEER_LINK_CLOSE))
    {
      m_rates = fields.rates;
    }
  if (m_subtype != (uint8_t) (WifiMeshActionHeader::PEER_LINK_CONFIRM))
    {
      m_meshId = fields.meshId;
    }
  if (m_subtype != (uint8_t) (WifiMeshActionHeader::PEER_LINK_CLOSE))
    {
      m_config = fields.config;
    }
  else
    {
      m_reasonCode = fields.reasonCode;
    }
}
PeerLinkFrameStart::PlinkFrameStartFields
PeerLinkFrameStart::GetFields ()
{
  PlinkFrameStartFields retval;
  //TODO: protocol version:
  retval.subtype = m_subtype;
  retval.capability = m_capability;
  retval.aid = m_aid;
  retval.rates = m_rates;
  retval.meshId = m_meshId;
  retval.config = m_config;
  retval.reasonCode = m_reasonCode;
  return retval;
}
TypeId
PeerLinkFrameStart::GetTypeId ()
{
  static TypeId tid = TypeId ("ns3::dot11s::PeerLinkFrameStart") .SetParent<Header> () .AddConstructor<
      PeerLinkFrameStart> ();
  return tid;
}
TypeId
PeerLinkFrameStart::GetInstanceTypeId () const
{
  return GetTypeId ();
}
void
PeerLinkFrameStart::Print (std::ostream &os) const
{
  os << "subtype = " << (uint16_t) m_subtype << "std::endlcapability = " << m_capability << "std::endlaid = "
      << (uint16_t) m_aid << "std::endlrates = " << m_rates << "std::endlmeshId = " << m_meshId << "std::endlconfiguration = "
      << m_config << "std::endlreason code = " << m_reasonCode;
}
uint32_t
PeerLinkFrameStart::GetSerializedSize () const
{
  uint32_t size = 3; //Peering protocol
  NS_ASSERT (m_subtype < 3);
  if ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CLOSE) != m_subtype)
    {
      size += 2; //capability
    }
  if ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CONFIRM) == m_subtype)
    {
      size += 2; //AID of remote peer
    }
  if ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CLOSE) != m_subtype)
    {
      size += m_rates.GetSerializedSize ();
    }
  if ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CONFIRM) != m_subtype)
    {
      size += m_meshId.GetSerializedSize ();
    }
  if ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CLOSE) != m_subtype)
    {
      size += m_config.GetSerializedSize ();
    }
  else
    {
      size += 2; //reasonCode
    }
  return size;
}
void
PeerLinkFrameStart::Serialize (Buffer::Iterator start) const
{
  Buffer::Iterator i = start;
  NS_ASSERT (m_subtype < 3);
  m_protocol.Serialize (i);
  i.Next (m_protocol.GetSerializedSize ());
  if ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CLOSE) != m_subtype)
    {
      i.WriteHtolsbU16 (m_capability);
    }
  if ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CONFIRM) == m_subtype)
    {
      i.WriteHtolsbU16 (m_aid);
    }
  if ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CLOSE) != m_subtype)
    {
      i = m_rates.Serialize (i);
    }
  if ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CONFIRM) != m_subtype)
    {
      m_meshId.Serialize (i);
      i.Next (m_meshId.GetSerializedSize ());
    }
  if ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CLOSE) != m_subtype)
    {
      m_config.Serialize (i);
      i.Next (m_config.GetSerializedSize ());
    }
  else
    {
      i.WriteHtolsbU16 (m_reasonCode);
    }
}
uint32_t
PeerLinkFrameStart::Deserialize (Buffer::Iterator start)
{
  Buffer::Iterator i = start;
  NS_ASSERT (m_subtype < 3);
  m_protocol.Deserialize (i);
  i.Next (m_protocol.GetSerializedSize ());
  if ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CLOSE) != m_subtype)
    {
      m_capability = i.ReadLsbtohU16 ();
    }
  if ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CONFIRM) == m_subtype)
    {
      m_aid = i.ReadLsbtohU16 ();
    }
  if ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CLOSE) != m_subtype)
    {
      i = m_rates.Deserialize (i);
    }
  if ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CONFIRM) != m_subtype)
    {
      m_meshId.Deserialize (i);
      i.Next (m_meshId.GetSerializedSize ());
    }
  if ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CLOSE) != m_subtype)
    {
      m_config.Deserialize (i);
      i.Next (m_config.GetSerializedSize ());
    }
  else
    {
      m_reasonCode = i.ReadLsbtohU16 ();
    }
  return i.GetDistanceFrom (start);
}
bool
operator== (const PeerLinkFrameStart & a, const PeerLinkFrameStart & b)
{
  return ((a.m_subtype == b.m_subtype) && (a.m_capability == b.m_capability) && (a.m_aid == b.m_aid)
      && (a.m_meshId.IsEqual (b.m_meshId)) && (a.m_config == b.m_config)
      && (a.m_reasonCode == b.m_reasonCode));
}
#ifdef RUN_SELF_TESTS
/// Built-in self test for PeerLinkFrameStart
struct PeerLinkFrameStartBist : public Test
{
  PeerLinkFrameStartBist () :
    Test ("Mesh/802.11s/IE/PeerLinkFrameStart")
  {
  }
  virtual bool
  RunTests ();
};
/// Test instance
static PeerLinkFrameStartBist g_PeerLinkFrameStartBist;

bool
PeerLinkFrameStartBist::RunTests ()
{
  bool result (true);
    {
      PeerLinkFrameStart a;
      PeerLinkFrameStart::PlinkFrameStartFields fields;
      fields.subtype = (uint8_t) (WifiMeshActionHeader::PEER_LINK_OPEN);
      fields.capability = 0;
      fields.aid = 101;
      fields.reasonCode = 12;
      fields.meshId = IeMeshId ("qwertyuiop");
      a.SetPlinkFrameStart (fields);
      Ptr<Packet> packet = Create<Packet> ();
      packet->AddHeader (a);
      PeerLinkFrameStart b;
      b.SetPlinkFrameSubtype ((uint8_t) (WifiMeshActionHeader::PEER_LINK_OPEN));
      packet->RemoveHeader (b);
      NS_TEST_ASSERT_EQUAL (a, b);
    }
    {
      PeerLinkFrameStart a;
      PeerLinkFrameStart::PlinkFrameStartFields fields;
      fields.subtype = (uint8_t) (WifiMeshActionHeader::PEER_LINK_CONFIRM);
      fields.capability = 0;
      fields.aid = 1234;
      fields.reasonCode = 12;
      fields.meshId = IeMeshId ("qwerty");
      a.SetPlinkFrameStart (fields);
      Ptr<Packet> packet = Create<Packet> ();
      packet->AddHeader (a);
      PeerLinkFrameStart b;
      b.SetPlinkFrameSubtype ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CONFIRM));
      packet->RemoveHeader (b);
      NS_TEST_ASSERT_EQUAL (a, b);
    }
    {
      PeerLinkFrameStart a;
      PeerLinkFrameStart::PlinkFrameStartFields fields;
      fields.subtype = (uint8_t) (WifiMeshActionHeader::PEER_LINK_CLOSE);
      fields.capability = 0;
      fields.aid = 10;
      fields.meshId = IeMeshId ("qqq");
      fields.reasonCode = 12;
      a.SetPlinkFrameStart (fields);
      Ptr<Packet> packet = Create<Packet> ();
      packet->AddHeader (a);
      PeerLinkFrameStart b;
      b.SetPlinkFrameSubtype ((uint8_t) (WifiMeshActionHeader::PEER_LINK_CLOSE));
      packet->RemoveHeader (b);
      NS_TEST_ASSERT_EQUAL (a, b);
    }
  return result;
}
#endif
} // namespace dot11s
} //namespace ns3