ipv4-address-generator-test-suite.cc

Go to the documentation of this file.

/*
 * Copyright (c) 2008 University of Washington
 *
 * 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
 */
 
#include "ns3/ipv4-address-generator.h"
#include "ns3/simulation-singleton.h"
#include "ns3/test.h"
 
using namespace ns3;
 
class NetworkNumberAllocatorTestCase : public TestCase
{
  public:
    NetworkNumberAllocatorTestCase();
    void DoRun() override;
    void DoTeardown() override;
};
 
NetworkNumberAllocatorTestCase::NetworkNumberAllocatorTestCase()
    : TestCase("Make sure the network number allocator is working on some of network prefixes.")
{
}
 
void
NetworkNumberAllocatorTestCase::DoTeardown()
{
    Ipv4AddressGenerator::Reset();
}
 
void
NetworkNumberAllocatorTestCase::DoRun()
{
    Ipv4Address network;
    Ipv4AddressGenerator::Init(Ipv4Address("1.0.0.0"),
                               Ipv4Mask("255.0.0.0"),
                               Ipv4Address("0.0.0.0"));
    network = Ipv4AddressGenerator::GetNetwork(Ipv4Mask("255.0.0.0"));
    NS_TEST_EXPECT_MSG_EQ(network, Ipv4Address("1.0.0.0"), "001");
    network = Ipv4AddressGenerator::NextNetwork(Ipv4Mask("255.0.0.0"));
    NS_TEST_EXPECT_MSG_EQ(network, Ipv4Address("2.0.0.0"), "001");
 
    Ipv4AddressGenerator::Init(Ipv4Address("0.1.0.0"),
                               Ipv4Mask("255.255.0.0"),
                               Ipv4Address("0.0.0.0"));
    network = Ipv4AddressGenerator::GetNetwork(Ipv4Mask("255.255.0.0"));
    NS_TEST_EXPECT_MSG_EQ(network, Ipv4Address("0.1.0.0"), "003");
    network = Ipv4AddressGenerator::NextNetwork(Ipv4Mask("255.255.0.0"));
    NS_TEST_EXPECT_MSG_EQ(network, Ipv4Address("0.2.0.0"), "004");
 
    Ipv4AddressGenerator::Init(Ipv4Address("0.0.1.0"),
                               Ipv4Mask("255.255.255.0"),
                               Ipv4Address("0.0.0.0"));
    network = Ipv4AddressGenerator::GetNetwork(Ipv4Mask("255.255.255.0"));
    NS_TEST_EXPECT_MSG_EQ(network, Ipv4Address("0.0.1.0"), "005");
    network = Ipv4AddressGenerator::NextNetwork(Ipv4Mask("255.255.255.0"));
    NS_TEST_EXPECT_MSG_EQ(network, Ipv4Address("0.0.2.0"), "006");
 
    network = Ipv4AddressGenerator::NextNetwork(Ipv4Mask("255.0.0.0"));
    NS_TEST_EXPECT_MSG_EQ(network, Ipv4Address("3.0.0.0"), "007");
    network = Ipv4AddressGenerator::NextNetwork(Ipv4Mask("255.255.0.0"));
    NS_TEST_EXPECT_MSG_EQ(network, Ipv4Address("0.3.0.0"), "008");
    network = Ipv4AddressGenerator::NextNetwork(Ipv4Mask("255.255.255.0"));
    NS_TEST_EXPECT_MSG_EQ(network, Ipv4Address("0.0.3.0"), "009");
}
 
class AddressAllocatorTestCase : public TestCase
{
  public:
    AddressAllocatorTestCase();
 
  private:
    void DoRun() override;
    void DoTeardown() override;
};
 
AddressAllocatorTestCase::AddressAllocatorTestCase()
    : TestCase("Sanity check on allocation of addresses")
{
}
 
void
AddressAllocatorTestCase::DoRun()
{
    Ipv4Address address;
 
    Ipv4AddressGenerator::Init(Ipv4Address("1.0.0.0"),
                               Ipv4Mask("255.0.0.0"),
                               Ipv4Address("0.0.0.3"));
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.0.0.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("1.0.0.3"), "100");
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.0.0.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("1.0.0.4"), "101");
 
    Ipv4AddressGenerator::Init(Ipv4Address("0.1.0.0"),
                               Ipv4Mask("255.255.0.0"),
                               Ipv4Address("0.0.0.3"));
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.255.0.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("0.1.0.3"), "102");
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.255.0.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("0.1.0.4"), "103");
 
    Ipv4AddressGenerator::Init(Ipv4Address("0.0.1.0"),
                               Ipv4Mask("255.255.255.0"),
                               Ipv4Address("0.0.0.3"));
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.255.255.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("0.0.1.3"), "104");
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.255.255.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("0.0.1.4"), "105");
}
 
void
AddressAllocatorTestCase::DoTeardown()
{
    Ipv4AddressGenerator::Reset();
    Simulator::Destroy();
}
 
class NetworkAndAddressTestCase : public TestCase
{
  public:
    NetworkAndAddressTestCase();
    void DoRun() override;
    void DoTeardown() override;
};
 
NetworkAndAddressTestCase::NetworkAndAddressTestCase()
    : TestCase("Make sure Network and address allocation play together.")
{
}
 
void
NetworkAndAddressTestCase::DoTeardown()
{
    Ipv4AddressGenerator::Reset();
    Simulator::Destroy();
}
 
void
NetworkAndAddressTestCase::DoRun()
{
    Ipv4Address address;
    Ipv4Address network;
 
    Ipv4AddressGenerator::Init(Ipv4Address("3.0.0.0"),
                               Ipv4Mask("255.0.0.0"),
                               Ipv4Address("0.0.0.3"));
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.0.0.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("3.0.0.3"), "200");
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.0.0.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("3.0.0.4"), "201");
 
    network = Ipv4AddressGenerator::NextNetwork(Ipv4Mask("255.0.0.0"));
    NS_TEST_EXPECT_MSG_EQ(network, Ipv4Address("4.0.0.0"), "202");
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.0.0.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("4.0.0.5"), "203");
 
    Ipv4AddressGenerator::Init(Ipv4Address("0.3.0.0"),
                               Ipv4Mask("255.255.0.0"),
                               Ipv4Address("0.0.0.3"));
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.255.0.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("0.3.0.3"), "204");
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.255.0.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("0.3.0.4"), "205");
 
    network = Ipv4AddressGenerator::NextNetwork(Ipv4Mask("255.255.0.0"));
    NS_TEST_EXPECT_MSG_EQ(network, Ipv4Address("0.4.0.0"), "206");
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.255.0.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("0.4.0.5"), "207");
 
    Ipv4AddressGenerator::Init(Ipv4Address("0.0.3.0"),
                               Ipv4Mask("255.255.255.0"),
                               Ipv4Address("0.0.0.3"));
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.255.255.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("0.0.3.3"), "208");
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.255.255.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("0.0.3.4"), "209");
 
    network = Ipv4AddressGenerator::NextNetwork(Ipv4Mask("255.255.255.0"));
    NS_TEST_EXPECT_MSG_EQ(network, Ipv4Address("0.0.4.0"), "210");
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.255.255.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("0.0.4.5"), "211");
}
 
class ExampleAddressGeneratorTestCase : public TestCase
{
  public:
    ExampleAddressGeneratorTestCase();
 
  private:
    void DoRun() override;
    void DoTeardown() override;
};
 
ExampleAddressGeneratorTestCase::ExampleAddressGeneratorTestCase()
    : TestCase("A quick kindof-semi-almost-real example")
{
}
 
void
ExampleAddressGeneratorTestCase::DoTeardown()
{
    Ipv4AddressGenerator::Reset();
}
 
void
ExampleAddressGeneratorTestCase::DoRun()
{
    Ipv4Address address;
    //
    // First, initialize our /24 network to 192.168.0.0 and begin
    // allocating with ip address 0.0.0.3 out of that prefix.
    //
    Ipv4AddressGenerator::Init(Ipv4Address("192.168.0.0"),
                               Ipv4Mask("255.255.255.0"),
                               Ipv4Address("0.0.0.3"));
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.255.255.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("192.168.0.3"), "300");
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.255.255.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("192.168.0.4"), "301");
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.255.255.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("192.168.0.5"), "302");
    //
    // Allocate the next network out of our /24 network (this should be
    // 192.168.1.0) and begin allocating with IP address 0.0.0.3 out of that
    // prefix.
    //
    Ipv4AddressGenerator::NextNetwork(Ipv4Mask("255.255.255.0"));
    Ipv4AddressGenerator::InitAddress(Ipv4Address("0.0.0.3"), Ipv4Mask("255.255.255.0"));
    //
    // The first address we should get is the previous numbers ORed together, which
    // is 192.168.1.3, of course.
    //
    address = Ipv4AddressGenerator::NextAddress(Ipv4Mask("255.255.255.0"));
    NS_TEST_EXPECT_MSG_EQ(address, Ipv4Address("192.168.1.3"), "304");
}
 
class AddressCollisionTestCase : public TestCase
{
  public:
    AddressCollisionTestCase();
 
  private:
    void DoRun() override;
    void DoTeardown() override;
};
 
AddressCollisionTestCase::AddressCollisionTestCase()
    : TestCase("Make sure that the address collision logic works.")
{
}
 
void
AddressCollisionTestCase::DoTeardown()
{
    Ipv4AddressGenerator::Reset();
    Simulator::Destroy();
}
 
void
AddressCollisionTestCase::DoRun()
{
    Ipv4AddressGenerator::AddAllocated("0.0.0.5");
    Ipv4AddressGenerator::AddAllocated("0.0.0.10");
    Ipv4AddressGenerator::AddAllocated("0.0.0.15");
    Ipv4AddressGenerator::AddAllocated("0.0.0.20");
 
    Ipv4AddressGenerator::AddAllocated("0.0.0.4");
    Ipv4AddressGenerator::AddAllocated("0.0.0.3");
    Ipv4AddressGenerator::AddAllocated("0.0.0.2");
    Ipv4AddressGenerator::AddAllocated("0.0.0.1");
 
    Ipv4AddressGenerator::AddAllocated("0.0.0.6");
    Ipv4AddressGenerator::AddAllocated("0.0.0.7");
    Ipv4AddressGenerator::AddAllocated("0.0.0.8");
    Ipv4AddressGenerator::AddAllocated("0.0.0.9");
 
    Ipv4AddressGenerator::AddAllocated("0.0.0.11");
    Ipv4AddressGenerator::AddAllocated("0.0.0.12");
    Ipv4AddressGenerator::AddAllocated("0.0.0.13");
    Ipv4AddressGenerator::AddAllocated("0.0.0.14");
 
    Ipv4AddressGenerator::AddAllocated("0.0.0.19");
    Ipv4AddressGenerator::AddAllocated("0.0.0.18");
    Ipv4AddressGenerator::AddAllocated("0.0.0.17");
    Ipv4AddressGenerator::AddAllocated("0.0.0.16");
 
    Ipv4AddressGenerator::TestMode();
    bool added = Ipv4AddressGenerator::AddAllocated("0.0.0.21");
    NS_TEST_EXPECT_MSG_EQ(added, true, "400");
 
    added = Ipv4AddressGenerator::AddAllocated("0.0.0.4");
    NS_TEST_EXPECT_MSG_EQ(added, false, "401");
 
    added = Ipv4AddressGenerator::AddAllocated("0.0.0.9");
    NS_TEST_EXPECT_MSG_EQ(added, false, "402");
 
    added = Ipv4AddressGenerator::AddAllocated("0.0.0.16");
    NS_TEST_EXPECT_MSG_EQ(added, false, "403");
 
    added = Ipv4AddressGenerator::AddAllocated("0.0.0.21");
    NS_TEST_EXPECT_MSG_EQ(added, false, "404");
}
 
class Ipv4AddressGeneratorTestSuite : public TestSuite
{
  public:
    Ipv4AddressGeneratorTestSuite();
 
  private:
};
 
Ipv4AddressGeneratorTestSuite::Ipv4AddressGeneratorTestSuite()
    : TestSuite("ipv4-address-generator", UNIT)
{
    AddTestCase(new NetworkNumberAllocatorTestCase(), TestCase::QUICK);
    AddTestCase(new AddressAllocatorTestCase(), TestCase::QUICK);
    AddTestCase(new NetworkAndAddressTestCase(), TestCase::QUICK);
    AddTestCase(new ExampleAddressGeneratorTestCase(), TestCase::QUICK);
    AddTestCase(new AddressCollisionTestCase(), TestCase::QUICK);
}
 
static Ipv4AddressGeneratorTestSuite
    g_ipv4AddressGeneratorTestSuite;