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

/*

 * Copyright (c) 2011, 2012 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)

 *

 * 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: Lluis Parcerisa <lparcerisa@cttc.cat>

 */

#include "ns3/log.h"

#include "ns3/string.h"

#include "ns3/double.h"

#include "ns3/enum.h"

#include "ns3/boolean.h"

#include "ns3/test.h"

#include "ns3/ptr.h"

#include "ns3/packet.h"

#include "ns3/lte-rrc-header.h"

#include "ns3/lte-rrc-sap.h"

NS_LOG_COMPONENT_DEFINE ("Asn1EncodingTest");

namespace ns3 {

class TestUtils

{

public:

  // Function to convert packet contents in hex format

  static std::string sprintPacketContentsHex (Ptr<Packet> pkt)

  {

    uint32_t psize = pkt->GetSize ();

    uint8_t buffer[psize];

    char sbuffer[psize * 3];

    pkt->CopyData (buffer, psize);

    for (uint32_t i = 0; i < psize; i++)

      {

        sprintf (&sbuffer[i * 3],"%02x ",buffer[i]);

      }

    return std::string (sbuffer);

  }

  // Function to convert packet contents in binary format

  static std::string sprintPacketContentsBin (Ptr<Packet> pkt)

  {

    uint32_t psize = pkt->GetSize ();

    uint8_t buffer[psize];

    std::ostringstream oss (std::ostringstream::out);

    pkt->CopyData (buffer, psize);

    for (uint32_t i = 0; i < psize; i++)

      {

        oss << (std::bitset<8> (buffer[i]));

      }

    return std::string (oss.str () + "\n");

  }

  // Function to log packet contents

  static void LogPacketContents (Ptr<Packet> pkt)

  {

    NS_LOG_DEBUG ("---- SERIALIZED PACKET CONTENTS (HEX): -------");

    NS_LOG_DEBUG ("Hex: " << TestUtils::sprintPacketContentsHex (pkt));

    NS_LOG_DEBUG ("Bin: " << TestUtils::sprintPacketContentsBin (pkt));

  }

  template <class T>

  static void LogPacketInfo (T source,std::string s)

  {

    NS_LOG_DEBUG ("--------- " << s.data () << " INFO: -------");

    std::ostringstream oss (std::ostringstream::out);

    source.Print (oss);

    NS_LOG_DEBUG (oss.str ());

  }

};

// --------------------------- CLASS RrcHeaderTestCase -----------------------------

/**

 * This class provides common functions to be inherited

 * by the children TestCases

 */

class RrcHeaderTestCase : public TestCase

{

public:

  RrcHeaderTestCase (std::string s);

  virtual void DoRun (void) = 0;

  LteRrcSap::RadioResourceConfigDedicated CreateRadioResourceConfigDedicated ();

  void AssertEqualRadioResourceConfigDedicated (LteRrcSap::RadioResourceConfigDedicated rrcd1, LteRrcSap::RadioResourceConfigDedicated rrcd2);

protected:

  Ptr<Packet> packet;

};

RrcHeaderTestCase :: RrcHeaderTestCase (std::string s) : TestCase (s)

{

}

LteRrcSap::RadioResourceConfigDedicated

RrcHeaderTestCase :: CreateRadioResourceConfigDedicated ()

{

  LteRrcSap::RadioResourceConfigDedicated rrd;

  rrd.drbToReleaseList = std::list<uint8_t> (4,2);

  LteRrcSap::SrbToAddMod srbToAddMod;

  srbToAddMod.srbIdentity = 2;

  LteRrcSap::LogicalChannelConfig logicalChannelConfig;

  logicalChannelConfig.priority = 9;

  logicalChannelConfig.prioritizedBitRateKbps = 128;

  logicalChannelConfig.bucketSizeDurationMs = 100;

  logicalChannelConfig.logicalChannelGroup = 3;

  srbToAddMod.logicalChannelConfig = logicalChannelConfig;

  rrd.srbToAddModList.insert (rrd.srbToAddModList.begin (),srbToAddMod);

  LteRrcSap::DrbToAddMod drbToAddMod;

  drbToAddMod.epsBearerIdentity = 1;

  drbToAddMod.drbIdentity = 1;

  drbToAddMod.logicalChannelIdentity = 5;

  LteRrcSap::RlcConfig rlcConfig;

  rlcConfig.choice = LteRrcSap::RlcConfig::UM_BI_DIRECTIONAL;

  drbToAddMod.rlcConfig = rlcConfig;

  LteRrcSap::LogicalChannelConfig logicalChannelConfig2;

  logicalChannelConfig2.priority = 7;

  logicalChannelConfig2.prioritizedBitRateKbps = 256;

  logicalChannelConfig2.bucketSizeDurationMs = 50;

  logicalChannelConfig2.logicalChannelGroup = 2;

  drbToAddMod.logicalChannelConfig = logicalChannelConfig2;

  rrd.drbToAddModList.insert (rrd.drbToAddModList.begin (),drbToAddMod);

  rrd.havePhysicalConfigDedicated = true;

  LteRrcSap::PhysicalConfigDedicated physicalConfigDedicated;

  physicalConfigDedicated.haveSoundingRsUlConfigDedicated = true;

  physicalConfigDedicated.soundingRsUlConfigDedicated.type = LteRrcSap::SoundingRsUlConfigDedicated::SETUP;

  physicalConfigDedicated.soundingRsUlConfigDedicated.srsBandwidth = 2;

  physicalConfigDedicated.soundingRsUlConfigDedicated.srsConfigIndex = 12;

  physicalConfigDedicated.haveAntennaInfoDedicated = true;

  physicalConfigDedicated.antennaInfo.transmissionMode = 2;

  rrd.physicalConfigDedicated = physicalConfigDedicated;

  return rrd;

}

void

RrcHeaderTestCase :: AssertEqualRadioResourceConfigDedicated (LteRrcSap::RadioResourceConfigDedicated rrcd1, LteRrcSap::RadioResourceConfigDedicated rrcd2)

{

  NS_TEST_ASSERT_MSG_EQ (rrcd1.srbToAddModList.size (), rrcd2.srbToAddModList.size (),"SrbToAddModList different sizes");

  std::list<LteRrcSap::SrbToAddMod> srcSrbToAddModList = rrcd1.srbToAddModList;

  std::list<LteRrcSap::SrbToAddMod>::iterator it1 = srcSrbToAddModList.begin ();

  std::list<LteRrcSap::SrbToAddMod> dstSrbToAddModList = rrcd2.srbToAddModList;

  std::list<LteRrcSap::SrbToAddMod>::iterator it2 = dstSrbToAddModList.begin ();

  for (; it1 != srcSrbToAddModList.end (); it1++, it2++)

    {

      NS_TEST_ASSERT_MSG_EQ (it1->srbIdentity,it2->srbIdentity, "srbIdentity");

      NS_TEST_ASSERT_MSG_EQ (it1->logicalChannelConfig.priority,it2->logicalChannelConfig.priority, "logicalChannelConfig.priority");

      NS_TEST_ASSERT_MSG_EQ (it1->logicalChannelConfig.prioritizedBitRateKbps,it2->logicalChannelConfig.prioritizedBitRateKbps, "logicalChannelConfig.prioritizedBitRateKbps");

      NS_TEST_ASSERT_MSG_EQ (it1->logicalChannelConfig.bucketSizeDurationMs,it2->logicalChannelConfig.bucketSizeDurationMs, "logicalChannelConfig.bucketSizeDurationMs");

      NS_TEST_ASSERT_MSG_EQ (it1->logicalChannelConfig.logicalChannelGroup,it2->logicalChannelConfig.logicalChannelGroup, "logicalChannelConfig.logicalChannelGroup");

    }

  NS_TEST_ASSERT_MSG_EQ (rrcd1.drbToAddModList.size (), rrcd2.drbToAddModList.size (),"DrbToAddModList different sizes");

  std::list<LteRrcSap::DrbToAddMod> srcDrbToAddModList = rrcd1.drbToAddModList;

  std::list<LteRrcSap::DrbToAddMod>::iterator it3 = srcDrbToAddModList.begin ();

  std::list<LteRrcSap::DrbToAddMod> dstDrbToAddModList = rrcd2.drbToAddModList;

  std::list<LteRrcSap::DrbToAddMod>::iterator it4 = dstDrbToAddModList.begin ();

  for (; it3 != srcDrbToAddModList.end (); it3++, it4++)

    {

      NS_TEST_ASSERT_MSG_EQ (it3->epsBearerIdentity,it4->epsBearerIdentity, "epsBearerIdentity");

      NS_TEST_ASSERT_MSG_EQ (it3->drbIdentity,it4->drbIdentity, "drbIdentity");

      NS_TEST_ASSERT_MSG_EQ (it3->rlcConfig.choice,it4->rlcConfig.choice, "rlcConfig.choice");

      NS_TEST_ASSERT_MSG_EQ (it3->logicalChannelIdentity,it4->logicalChannelIdentity, "logicalChannelIdentity");

      NS_TEST_ASSERT_MSG_EQ (it3->epsBearerIdentity,it4->epsBearerIdentity, "epsBearerIdentity");

      NS_TEST_ASSERT_MSG_EQ (it3->logicalChannelConfig.priority,it4->logicalChannelConfig.priority, "logicalChannelConfig.priority");

      NS_TEST_ASSERT_MSG_EQ (it3->logicalChannelConfig.prioritizedBitRateKbps,it4->logicalChannelConfig.prioritizedBitRateKbps, "logicalChannelConfig.prioritizedBitRateKbps");

      NS_TEST_ASSERT_MSG_EQ (it3->logicalChannelConfig.bucketSizeDurationMs,it4->logicalChannelConfig.bucketSizeDurationMs, "logicalChannelConfig.bucketSizeDurationMs");

      NS_TEST_ASSERT_MSG_EQ (it3->logicalChannelConfig.logicalChannelGroup,it4->logicalChannelConfig.logicalChannelGroup, "logicalChannelConfig.logicalChannelGroup");

    }

  NS_TEST_ASSERT_MSG_EQ (rrcd1.drbToReleaseList.size (), rrcd2.drbToReleaseList.size (),"DrbToReleaseList different sizes");

  std::list<uint8_t> srcDrbToReleaseList = rrcd1.drbToReleaseList;

  std::list<uint8_t> dstDrbToReleaseList = rrcd2.drbToReleaseList;

  std::list<uint8_t>::iterator it5 = srcDrbToReleaseList.begin ();

  std::list<uint8_t>::iterator it6 = dstDrbToReleaseList.begin ();

  for (; it5 != srcDrbToReleaseList.end (); it5++, it6++)

    {

      NS_TEST_ASSERT_MSG_EQ (*it5, *it6,"element != in DrbToReleaseList");

    }

  NS_TEST_ASSERT_MSG_EQ (rrcd1.havePhysicalConfigDedicated,rrcd2.havePhysicalConfigDedicated, "HavePhysicalConfigDedicated");

  if (rrcd1.havePhysicalConfigDedicated)

    {

      NS_TEST_ASSERT_MSG_EQ (rrcd1.physicalConfigDedicated.haveSoundingRsUlConfigDedicated,

                             rrcd2.physicalConfigDedicated.haveSoundingRsUlConfigDedicated,

                             "haveSoundingRsUlConfigDedicated");

      NS_TEST_ASSERT_MSG_EQ (rrcd1.physicalConfigDedicated.soundingRsUlConfigDedicated.type,

                             rrcd2.physicalConfigDedicated.soundingRsUlConfigDedicated.type,

                             "soundingRsUlConfigDedicated.type");

      NS_TEST_ASSERT_MSG_EQ (rrcd1.physicalConfigDedicated.soundingRsUlConfigDedicated.srsBandwidth,

                             rrcd2.physicalConfigDedicated.soundingRsUlConfigDedicated.srsBandwidth,

                             "soundingRsUlConfigDedicated.srsBandwidth");

      NS_TEST_ASSERT_MSG_EQ (rrcd1.physicalConfigDedicated.soundingRsUlConfigDedicated.srsConfigIndex,

                             rrcd2.physicalConfigDedicated.soundingRsUlConfigDedicated.srsConfigIndex,

                             "soundingRsUlConfigDedicated.srsConfigIndex");

      NS_TEST_ASSERT_MSG_EQ (rrcd1.physicalConfigDedicated.haveAntennaInfoDedicated,

                             rrcd2.physicalConfigDedicated.haveAntennaInfoDedicated,

                             "haveAntennaInfoDedicated");

      if (rrcd1.physicalConfigDedicated.haveAntennaInfoDedicated)

        {

          NS_TEST_ASSERT_MSG_EQ (rrcd1.physicalConfigDedicated.antennaInfo.transmissionMode,

                                 rrcd2.physicalConfigDedicated.antennaInfo.transmissionMode,

                                 "antennaInfo.transmissionMode");

        }

    }

}

// --------------------------- CLASS RrcConnectionRequestTestCase -----------------------------

class RrcConnectionRequestTestCase : public RrcHeaderTestCase

{

public:

  RrcConnectionRequestTestCase ();

  virtual void DoRun (void);

};

RrcConnectionRequestTestCase::RrcConnectionRequestTestCase () : RrcHeaderTestCase ("Testing RrcConnectionRequest")

{

}

void

RrcConnectionRequestTestCase::DoRun (void)

{

  packet = Create<Packet> ();

  NS_LOG_DEBUG ("============= RrcConnectionRequestTestCase ===========");

  LteRrcSap::RrcConnectionRequest msg;

  msg.ueIdentity = 0x83fecafecaULL;

  RrcConnectionRequestHeader source;

  source.SetMessage (msg);

  // Log source info

  TestUtils::LogPacketInfo<RrcConnectionRequestHeader> (source,"SOURCE");

  // Add header

  packet->AddHeader (source);

  // Log serialized packet contents

  TestUtils::LogPacketContents (packet);

  // Remove header

  RrcConnectionRequestHeader destination;

  packet->RemoveHeader (destination);

  // Log destination info

  TestUtils::LogPacketInfo<RrcConnectionRequestHeader> (destination,"DESTINATION");

  // Check that the destination and source headers contain the same values

  NS_TEST_ASSERT_MSG_EQ (source.GetMmec (),destination.GetMmec (), "Different m_mmec!");

  NS_TEST_ASSERT_MSG_EQ (source.GetMtmsi (),destination.GetMtmsi (), "Different m_mTmsi!");

  packet = 0;

}

// --------------------------- CLASS RrcConnectionSetupTestCase -----------------------------

class RrcConnectionSetupTestCase : public RrcHeaderTestCase

{

public:

  RrcConnectionSetupTestCase ();

  virtual void DoRun (void);

};

RrcConnectionSetupTestCase::RrcConnectionSetupTestCase () : RrcHeaderTestCase ("Testing RrcConnectionSetupTestCase")

{

}

void

RrcConnectionSetupTestCase::DoRun (void)

{

  packet = Create<Packet> ();

  NS_LOG_DEBUG ("============= RrcConnectionSetupTestCase ===========");

  LteRrcSap::RrcConnectionSetup msg;

  msg.rrcTransactionIdentifier = 3;

  msg.radioResourceConfigDedicated = CreateRadioResourceConfigDedicated ();

  RrcConnectionSetupHeader source;

  source.SetMessage (msg);

  // Log source info

  TestUtils::LogPacketInfo<RrcConnectionSetupHeader> (source,"SOURCE");

  // Add header

  packet->AddHeader (source);

  // Log serialized packet contents

  TestUtils::LogPacketContents (packet);

  // remove header

  RrcConnectionSetupHeader destination;

  packet->RemoveHeader (destination);

  // Log destination info

  TestUtils::LogPacketInfo<RrcConnectionSetupHeader> (destination,"DESTINATION");

  // Check that the destination and source headers contain the same values

  NS_TEST_ASSERT_MSG_EQ (source.GetRrcTransactionIdentifier (),destination.GetRrcTransactionIdentifier (), "RrcTransactionIdentifier");

  AssertEqualRadioResourceConfigDedicated (source.GetRadioResourceConfigDedicated (),destination.GetRadioResourceConfigDedicated ());

  packet = 0;

}

// --------------------------- CLASS RrcConnectionSetupCompleteTestCase -----------------------------

class RrcConnectionSetupCompleteTestCase : public RrcHeaderTestCase

{

public:

  RrcConnectionSetupCompleteTestCase ();

  virtual void DoRun (void);

};

RrcConnectionSetupCompleteTestCase::RrcConnectionSetupCompleteTestCase () : RrcHeaderTestCase ("Testing RrcConnectionSetupCompleteTestCase")

{

}

void

RrcConnectionSetupCompleteTestCase::DoRun (void)

{

  packet = Create<Packet> ();

  NS_LOG_DEBUG ("============= RrcConnectionSetupCompleteTestCase ===========");

  LteRrcSap::RrcConnectionSetupCompleted msg;

  msg.rrcTransactionIdentifier = 3;

  RrcConnectionSetupCompleteHeader source;

  source.SetMessage (msg);

  // Log source info

  TestUtils::LogPacketInfo<RrcConnectionSetupCompleteHeader> (source,"SOURCE");

  // Add header

  packet->AddHeader (source);

  // Log serialized packet contents

  TestUtils::LogPacketContents (packet);

  // Remove header

  RrcConnectionSetupCompleteHeader destination;

  packet->RemoveHeader (destination);

  // Log destination info

  TestUtils::LogPacketInfo<RrcConnectionSetupCompleteHeader> (destination,"DESTINATION");

  // Check that the destination and source headers contain the same values

  NS_TEST_ASSERT_MSG_EQ (source.GetRrcTransactionIdentifier (),destination.GetRrcTransactionIdentifier (), "RrcTransactionIdentifier");

  packet = 0;

}

// --------------------------- CLASS RrcConnectionReconfigurationCompleteTestCase -----------------------------

class RrcConnectionReconfigurationCompleteTestCase : public RrcHeaderTestCase

{

public:

  RrcConnectionReconfigurationCompleteTestCase ();

  virtual void DoRun (void);

};

RrcConnectionReconfigurationCompleteTestCase::RrcConnectionReconfigurationCompleteTestCase ()

  : RrcHeaderTestCase ("Testing RrcConnectionReconfigurationCompleteTestCase")

{

}

void

RrcConnectionReconfigurationCompleteTestCase::DoRun (void)

{

  packet = Create<Packet> ();

  NS_LOG_DEBUG ("============= RrcConnectionReconfigurationCompleteTestCase ===========");

  LteRrcSap::RrcConnectionReconfigurationCompleted msg;

  msg.rrcTransactionIdentifier = 2;

  RrcConnectionReconfigurationCompleteHeader source;

  source.SetMessage (msg);

  // Log source info

  TestUtils::LogPacketInfo<RrcConnectionReconfigurationCompleteHeader> (source,"SOURCE");

  // Add header

  packet->AddHeader (source);

  // Log serialized packet contents

  TestUtils::LogPacketContents (packet);

  // remove header

  RrcConnectionReconfigurationCompleteHeader destination;

  packet->RemoveHeader (destination);

  // Log destination info

  TestUtils::LogPacketInfo<RrcConnectionReconfigurationCompleteHeader> (destination,"DESTINATION");

  // Check that the destination and source headers contain the same values

  NS_TEST_ASSERT_MSG_EQ (source.GetRrcTransactionIdentifier (),destination.GetRrcTransactionIdentifier (), "RrcTransactionIdentifier");

  packet = 0;

}

// --------------------------- CLASS RrcConnectionReconfigurationTestCase -----------------------------

class RrcConnectionReconfigurationTestCase : public RrcHeaderTestCase

{

public:

  RrcConnectionReconfigurationTestCase ();

  virtual void DoRun (void);

};

RrcConnectionReconfigurationTestCase::RrcConnectionReconfigurationTestCase ()

  : RrcHeaderTestCase ("Testing RrcConnectionReconfigurationTestCase")

{

}

void

RrcConnectionReconfigurationTestCase::DoRun (void)

{

  packet = Create<Packet> ();

  NS_LOG_DEBUG ("============= RrcConnectionReconfigurationTestCase ===========");

  LteRrcSap::RrcConnectionReconfiguration msg;

  msg.rrcTransactionIdentifier = 2;

  msg.haveMeasConfig = true;

  msg.measConfig.haveQuantityConfig = true;

  msg.measConfig.quantityConfig.filterCoefficientRSRP = 8;

  msg.measConfig.quantityConfig.filterCoefficientRSRQ = 7;

  msg.measConfig.haveMeasGapConfig = true;

  msg.measConfig.measGapConfig.type = LteRrcSap::MeasGapConfig::SETUP;

  msg.measConfig.measGapConfig.gapOffsetChoice = LteRrcSap::MeasGapConfig::gp0;

  msg.measConfig.measGapConfig.gapOffsetValue = 21;

  msg.measConfig.haveSmeasure = true;

  msg.measConfig.sMeasure = 57;

  msg.measConfig.haveSpeedStatePars = true;

  msg.measConfig.speedStatePars.type = LteRrcSap::SpeedStatePars::SETUP;

  msg.measConfig.speedStatePars.mobilityStateParameters.tEvaluation = 240;

  msg.measConfig.speedStatePars.mobilityStateParameters.tHystNormal = 60;

  msg.measConfig.speedStatePars.mobilityStateParameters.nCellChangeMedium = 5;

  msg.measConfig.speedStatePars.mobilityStateParameters.nCellChangeHigh = 13;

  msg.measConfig.speedStatePars.timeToTriggerSf.sfMedium = 25;

  msg.measConfig.speedStatePars.timeToTriggerSf.sfHigh = 75;

  msg.measConfig.measObjectToRemoveList.push_back (23);

  msg.measConfig.measObjectToRemoveList.push_back (13);

  msg.measConfig.reportConfigToRemoveList.push_back (7);

  msg.measConfig.reportConfigToRemoveList.push_back (16);

  msg.measConfig.measIdToRemoveList.push_back (4);

  msg.measConfig.measIdToRemoveList.push_back (18);

  // std::list<MeasObjectToAddMod> measObjectToAddModList;

  // std::list<ReportConfigToAddMod> reportConfigToAddModList;

  // std::list<MeasIdToAddMod> measIdToAddModList;

  msg.haveMobilityControlInfo = true;

  msg.mobilityControlInfo.targetPhysCellId = 4;

  msg.mobilityControlInfo.haveCarrierFreq = true;

  msg.mobilityControlInfo.carrierFreq.dlCarrierFreq = 3;

  msg.mobilityControlInfo.carrierFreq.ulCarrierFreq = 5;

  msg.mobilityControlInfo.haveCarrierBandwidth = true;

  msg.mobilityControlInfo.carrierBandwidth.dlBandwidth = 50;

  msg.mobilityControlInfo.carrierBandwidth.ulBandwidth = 25;

  msg.mobilityControlInfo.newUeIdentity = 11;

  msg.mobilityControlInfo.haveRachConfigDedicated = true;

  msg.mobilityControlInfo.rachConfigDedicated.raPreambleIndex = 2;

  msg.mobilityControlInfo.rachConfigDedicated.raPrachMaskIndex = 2;

  msg.mobilityControlInfo.radioResourceConfigCommon.rachConfigCommon.preambleInfo.numberOfRaPreambles = 4;

  msg.mobilityControlInfo.radioResourceConfigCommon.rachConfigCommon.raSupervisionInfo.preambleTransMax = 3;

  msg.mobilityControlInfo.radioResourceConfigCommon.rachConfigCommon.raSupervisionInfo.raResponseWindowSize = 6;

  msg.haveRadioResourceConfigDedicated = true;

  msg.radioResourceConfigDedicated = CreateRadioResourceConfigDedicated ();