dadada/ns-3-leo
dadada/ns-3-leo
1
0
Fork 0
mirror of https://gitlab.ibr.cs.tu-bs.de/tschuber/ns-3-leo.git synced 2025-06-08 18:13:57 +02:00
Code Issues Releases Wiki Activity

add index for satellite / ground net devices

Browse source
This commit is contained in:
Tim Schubert 2020-06-26 13:54:03 +02:00
parent 20ad318cb9
commit 432866d8ca
3 changed files with 79 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

79
model/leo-mock-channel.cc
View file

@ -26,8 +26,7 @@ LeoMockChannel::GetTypeId (void)
}
LeoMockChannel::LeoMockChannel() :
MockChannel (),
m_channelType (LeoMockChannel::ChannelType::UNKNOWN)
MockChannel ()
{
NS_LOG_FUNCTION_NOARGS ();
}
@ -43,25 +42,81 @@ LeoMockChannel::TransmitStart (Ptr<const Packet> p,
Time txTime)
{
// Find devices joined to channel
Ptr<LeoMockNetDevice> srcDev = DynamicCast<LeoMockNetDevice> (GetDevice (devId));
Ptr<LeoMockNetDevice> dstDev = DynamicCast<LeoMockNetDevice> (GetDevice (dst));
Ptr<MockNetDevice> srcDev = DynamicCast<MockNetDevice> (GetDevice (devId));
bool isBroadcast = (srcDev->GetBroadcast () == dst);
if (!isBroadcast && dstDev == nullptr)
{
NS_LOG_LOGIC (this << "Dropping packet: direct communication between ground stations and satellites not allowed using this channel");
return false;
}
bool toSatellite = m_groundDevices.find (srcDev->GetAddress ()) != m_groundDevices.end ();
for (uint32_t i = 0; i < GetNDevices (); i++)
if (isBroadcast)
{
if (srcDev->GetDeviceType () != dstDev->GetDeviceType ())
// Broadcast hack for ARP and neighbor cache
// TODO remove if found a way to fill neighbor cache / ARP by hand
for (uint32_t i = 0; i < GetNDevices (); i++)
{
// TODO deliver only to devices in the same beam
Deliver (p, srcDev, DynamicCast<MockNetDevice> (GetDevice (i)), txTime);
}
}
else
{
if (toSatellite)
{
Deliver (p, srcDev, dstDev, txTime);
// Satellites can always directly be addresses
// Assume beams are narrow enough to not also receive the signal at other
// satellites for performance reasons.
DeviceIndex::iterator it = m_satelliteDevices.find (dst);
if (it == m_satelliteDevices.end ())
{
NS_LOG_LOGIC (this << "unable to find satellite with address " << dst);
return false;
}
Deliver (p, srcDev, it->second, txTime);
}
else
// space to ground delivers to everything within the beam
{
for (DeviceIndex::iterator it = m_groundDevices.begin ();
it != m_groundDevices.end ();
it++)
{
// TODO deliver only to devices in the same beam
Deliver (p, srcDev, it->second, txTime);
}
}
}
return true;
}
int32_t
LeoMockChannel::Attach (Ptr<MockNetDevice> device)
{
Ptr<LeoMockNetDevice> leodev = DynamicCast<LeoMockNetDevice> (device);
// Add to index
switch (leodev->GetDeviceType ())
{
case LeoMockNetDevice::DeviceType::GND:
m_groundDevices[leodev->GetAddress ()] = leodev;
break;
case LeoMockNetDevice::DeviceType::SAT:
m_satelliteDevices[leodev->GetAddress ()] = leodev;
break;
default:
break;
}
return MockChannel::Attach (device);
}
bool
LeoMockChannel::Detach (uint32_t deviceId)
{
Ptr<NetDevice> dev = GetDevice (deviceId);
m_groundDevices.erase (dev->GetAddress ());
m_satelliteDevices.erase (dev->GetAddress ());
return MockChannel::Detach (deviceId);
}
}; // namespace ns3