mirror of
https://gitlab.ibr.cs.tu-bs.de/tschuber/ns-3-leo.git
synced 2025-06-08 01:53:58 +02:00
62 lines
1.9 KiB
C++
62 lines
1.9 KiB
C++
/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
|
|
|
|
#include "ns3/core-module.h"
|
|
#include "ns3/network-module.h"
|
|
#include "ns3/internet-module.h"
|
|
#include "ns3/log.h"
|
|
#include "../model/leo-mock-net-device.h"
|
|
|
|
#include "arp-cache-helper.h"
|
|
|
|
namespace ns3
|
|
{
|
|
|
|
NS_LOG_COMPONENT_DEFINE ("ArpCacheHelper");
|
|
|
|
void
|
|
ArpCacheHelper::Install (NetDeviceContainer &devices, Ipv4InterfaceContainer &interfaces) const
|
|
{
|
|
NS_LOG_FUNCTION (this);
|
|
|
|
for (size_t i = 0; i < devices.GetN (); i ++)
|
|
{
|
|
Ptr<NetDevice> dev = devices.Get (i);
|
|
Ptr<Node> node = dev->GetNode ();
|
|
NS_LOG_INFO ("Preparing ARP cache of " << node);
|
|
Ptr<Ipv4L3Protocol> ipv4 = node->GetObject<Ipv4L3Protocol> ();
|
|
int32_t ifIndex = ipv4->GetInterfaceForDevice (dev);
|
|
Ptr<Ipv4Interface> interface = ipv4->GetInterface (ifIndex);
|
|
Ptr<ArpCache> cache = interface->GetArpCache ();
|
|
|
|
for (uint32_t j = 0; j < devices.GetN (); j++)
|
|
{
|
|
// every other device, that is not of same "type"
|
|
Ptr<NetDevice> otherDevice = devices.Get (j);
|
|
Ptr<LeoMockNetDevice> leoDev = DynamicCast<LeoMockNetDevice> (dev);
|
|
Ptr<LeoMockNetDevice> otherLeoDev = DynamicCast<LeoMockNetDevice> (otherDevice);
|
|
if (i == j || (leoDev != 0
|
|
&& otherLeoDev != 0
|
|
&& leoDev->GetDeviceType () == otherLeoDev->GetDeviceType ()))
|
|
{
|
|
continue;
|
|
}
|
|
Address address = otherDevice->GetAddress (); // MAC
|
|
|
|
// and associated address
|
|
uint32_t otherIfIndex = otherDevice->GetIfIndex ();
|
|
Ipv4Address ipaddr = interfaces.GetAddress (otherIfIndex, 0); // IP
|
|
|
|
// update cache
|
|
ArpCache::Entry* entry = cache->Lookup (ipaddr);
|
|
if (entry == 0)
|
|
{
|
|
entry = cache->Add (ipaddr);
|
|
}
|
|
entry->SetMacAddress (address);
|
|
|
|
NS_LOG_DEBUG ("Added entry for " << address);
|
|
}
|
|
}
|
|
}
|
|
|
|
};
|