csma-star-helper.cc

Go to the documentation of this file.

/*
 * SPDX-License-Identifier: GPL-2.0-only
 */
 
#include "csma-star-helper.h"
 
#include "ns3/log.h"
#include "ns3/node-list.h"
#include "ns3/point-to-point-net-device.h"
#include "ns3/vector.h"
 
#include <iostream>
#include <sstream>
 
namespace ns3
{
 
NS_LOG_COMPONENT_DEFINE("CsmaStarHelper");
 
CsmaStarHelper::CsmaStarHelper(uint32_t numSpokes, CsmaHelper csmaHelper)
{
    m_hub.Create(1);
    m_spokes.Create(numSpokes);
 
    for (uint32_t i = 0; i < m_spokes.GetN(); ++i)
    {
        NodeContainer nodes(m_hub.Get(0), m_spokes.Get(i));
        NetDeviceContainer nd = csmaHelper.Install(nodes);
        m_hubDevices.Add(nd.Get(0));
        m_spokeDevices.Add(nd.Get(1));
    }
}
 
CsmaStarHelper::~CsmaStarHelper()
{
}
 
Ptr<Node>
CsmaStarHelper::GetHub() const
{
    return m_hub.Get(0);
}
 
Ptr<Node>
CsmaStarHelper::GetSpokeNode(uint32_t i) const
{
    return m_spokes.Get(i);
}
 
NetDeviceContainer
CsmaStarHelper::GetHubDevices() const
{
    return m_hubDevices;
}
 
NetDeviceContainer
CsmaStarHelper::GetSpokeDevices() const
{
    return m_spokeDevices;
}
 
Ipv4Address
CsmaStarHelper::GetHubIpv4Address(uint32_t i) const
{
    return m_hubInterfaces.GetAddress(i);
}
 
Ipv4Address
CsmaStarHelper::GetSpokeIpv4Address(uint32_t i) const
{
    return m_spokeInterfaces.GetAddress(i);
}
 
Ipv6Address
CsmaStarHelper::GetHubIpv6Address(uint32_t i) const
{
    return m_hubInterfaces6.GetAddress(i, 1);
}
 
Ipv6Address
CsmaStarHelper::GetSpokeIpv6Address(uint32_t i) const
{
    return m_spokeInterfaces6.GetAddress(i, 1);
}
 
uint32_t
CsmaStarHelper::SpokeCount() const
{
    return m_spokes.GetN();
}
 
void
CsmaStarHelper::InstallStack(InternetStackHelper stack)
{
    stack.Install(m_hub);
    stack.Install(m_spokes);
}
 
void
CsmaStarHelper::AssignIpv4Addresses(Ipv4AddressHelper address)
{
    for (uint32_t i = 0; i < m_spokes.GetN(); ++i)
    {
        m_hubInterfaces.Add(address.Assign(m_hubDevices.Get(i)));
        m_spokeInterfaces.Add(address.Assign(m_spokeDevices.Get(i)));
        address.NewNetwork();
    }
}
 
void
CsmaStarHelper::AssignIpv6Addresses(Ipv6Address network, Ipv6Prefix prefix)
{
    Ipv6AddressGenerator::Init(network, prefix);
    Ipv6Address v6network;
    Ipv6AddressHelper addressHelper;
 
    for (uint32_t i = 0; i < m_spokes.GetN(); ++i)
    {
        v6network = Ipv6AddressGenerator::GetNetwork(prefix);
        addressHelper.SetBase(v6network, prefix);
 
        Ipv6InterfaceContainer ic = addressHelper.Assign(m_hubDevices.Get(i));
        m_hubInterfaces6.Add(ic);
        ic = addressHelper.Assign(m_spokeDevices.Get(i));
        m_spokeInterfaces6.Add(ic);
 
        Ipv6AddressGenerator::NextNetwork(prefix);
    }
}
 
} // namespace ns3