lr-wpan-packet-test.cc

Go to the documentation of this file.

/*
 * Copyright (c) 2011 The Boeing Company
 *
 * 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:  Tom Henderson <thomas.r.henderson@boeing.com>
 */
#include <ns3/log.h>
#include <ns3/lr-wpan-mac-header.h>
#include <ns3/lr-wpan-mac-trailer.h>
#include <ns3/mac16-address.h>
#include <ns3/mac64-address.h>
#include <ns3/packet.h>
#include <ns3/test.h>
 
#include <vector>
 
using namespace ns3;
 
NS_LOG_COMPONENT_DEFINE("lr-wpan-packet-test");
 
class LrWpanPacketTestCase : public TestCase
{
  public:
    LrWpanPacketTestCase();
    ~LrWpanPacketTestCase() override;
 
  private:
    void DoRun() override;
};
 
LrWpanPacketTestCase::LrWpanPacketTestCase()
    : TestCase("Test the 802.15.4 MAC header and trailer classes")
{
}
 
LrWpanPacketTestCase::~LrWpanPacketTestCase()
{
}
 
void
LrWpanPacketTestCase::DoRun()
{
    LrWpanMacHeader macHdr(LrWpanMacHeader::LRWPAN_MAC_BEACON, 0); // sequence number set to 0
    macHdr.SetSrcAddrMode(LrWpanMacHeader::SHORTADDR);             // short addr
    macHdr.SetDstAddrMode(LrWpanMacHeader::NOADDR);
    macHdr.SetSecDisable();
    macHdr.SetNoPanIdComp();
    // ... other setters
 
    uint16_t srcPanId = 100;
    Mac16Address srcWpanAddr("00:11");
    macHdr.SetSrcAddrFields(srcPanId, srcWpanAddr);
 
    LrWpanMacTrailer macTrailer;
 
    Ptr<Packet> p = Create<Packet>(20); // 20 bytes of dummy data
    NS_TEST_ASSERT_MSG_EQ(p->GetSize(), 20, "Packet created with unexpected size");
    p->AddHeader(macHdr);
    std::cout << " <--Mac Header added " << std::endl;
 
    NS_TEST_ASSERT_MSG_EQ(p->GetSize(), 27, "Packet wrong size after macHdr addition");
    p->AddTrailer(macTrailer);
    NS_TEST_ASSERT_MSG_EQ(p->GetSize(), 29, "Packet wrong size after macTrailer addition");
 
    // Test serialization and deserialization
    uint32_t size = p->GetSerializedSize();
    std::vector<uint8_t> buffer(size);
    p->Serialize(buffer.data(), size);
    Ptr<Packet> p2 = Create<Packet>(buffer.data(), size, true);
 
    p2->Print(std::cout);
    std::cout << " <--Packet P2 " << std::endl;
 
    NS_TEST_ASSERT_MSG_EQ(p2->GetSize(), 29, "Packet wrong size after deserialization");
 
    LrWpanMacHeader receivedMacHdr;
    p2->RemoveHeader(receivedMacHdr);
 
    receivedMacHdr.Print(std::cout);
    std::cout << " <--P2 Mac Header " << std::endl;
 
    NS_TEST_ASSERT_MSG_EQ(p2->GetSize(), 22, "Packet wrong size after removing machdr");
 
    LrWpanMacTrailer receivedMacTrailer;
    p2->RemoveTrailer(receivedMacTrailer);
    NS_TEST_ASSERT_MSG_EQ(p2->GetSize(),
                          20,
                          "Packet wrong size after removing headers and trailers");
    // Compare macHdr with receivedMacHdr, macTrailer with receivedMacTrailer,...
}
 
class LrWpanPacketTestSuite : public TestSuite
{
  public:
    LrWpanPacketTestSuite();
};
 
LrWpanPacketTestSuite::LrWpanPacketTestSuite()
    : TestSuite("lr-wpan-packet", UNIT)
{
    AddTestCase(new LrWpanPacketTestCase, TestCase::QUICK);
}
 
static LrWpanPacketTestSuite g_lrWpanPacketTestSuite;