To Implement network defense in ns-3 by setting up mechanisms to detect and mitigate various network attacks. We tackle various type of network attacks with best coding support. Here the detailed guide on how to set up a basic network defense system using ns-3.
Step-by-Step Implementation of Network Defense in ns-3:
- Install ns-3
Install ns-3 on the system.
- Define the Network Topology
Define the network topology including:
- Normal nodes (legitimate users)
- Attacker nodes
- Server nodes
- Defense nodes (nodes used to detect and mitigate attacks)
- Create Network Nodes
Create network nodes using NodeContainer.
NodeContainer normalNodes, attackerNodes, serverNodes, defenseNodes;
normalNodes.Create(3);
attackerNodes.Create(1);
serverNodes.Create(1);
defenseNodes.Create(1);
4. Set Up Network Devices
Install network devices on the nodes using appropriate network interfaces, such as WiFi for wireless communication.
WifiHelper wifi;
wifi.SetStandard(WIFI_PHY_STANDARD_80211n_5GHZ);
WifiMacHelper mac;
mac.SetType(“ns3::AdhocWifiMac”);
YansWifiPhyHelper phy = YansWifiPhyHelper::Default();
YansWifiChannelHelper channel = YansWifiChannelHelper::Default();
phy.SetChannel(channel.Create());
NetDeviceContainer normalDevices = wifi.Install(phy, mac, normalNodes);
NetDeviceContainer attackerDevices = wifi.Install(phy, mac, attackerNodes);
NetDeviceContainer serverDevices = wifi.Install(phy, mac, serverNodes);
NetDeviceContainer defenseDevices = wifi.Install(phy, mac, defenseNodes);
5. Configure Mobility Model
Set up the mobility model for the nodes.
MobilityHelper mobility;
mobility.SetMobilityModel(“ns3::ConstantPositionMobilityModel”);
mobility.Install(normalNodes);
mobility.Install(attackerNodes);
mobility.Install(serverNodes);
mobility.Install(defenseNodes);
6. Set Up Packet Capture
Configure packet capture on the defense nodes. Use PcapHelper to capture packets.
PcapHelper pcapHelper;
Ptr<PcapFileWrapper> file = pcapHelper.CreateFile(“defense_capture.pcap”, std::ios::out, PcapHelper::DLT_PPP);
for (uint32_t i = 0; i < defenseDevices.GetN(); ++i) {
phy.EnablePcap(“defense_capture”, defenseDevices.Get(i), true, true);
}
7. Implement Defense Application
Create an application that detects and mitigates attacks. Below is a simple example of an application that detects a high rate of packets as a potential threat and then blocks the attacker.
Defense Application (Example)
class DefenseApplication : public Application {
public:
void StartApplication() override {
recvSocket = Socket::CreateSocket(GetNode(), UdpSocketFactory::GetTypeId());
recvSocket->Bind(InetSocketAddress(Ipv4Address::GetAny(), listenPort));
recvSocket->SetRecvCallback(MakeCallback(&DefenseApplication::HandleRead, this));
}
void SetListenPort(uint16_t port) {
listenPort = port;
}
void HandleRead(Ptr<Socket> socket) {
Ptr<Packet> packet;
Address from;
while ((packet = socket->RecvFrom(from))) {
packetsReceived++;
Ipv4Address senderAddr = InetSocketAddress::ConvertFrom(from).GetIpv4();
double currentTime = Simulator::Now().GetSeconds();
if (currentTime – lastTime > interval) {
if (packetsReceived > threshold) {
std::cout << “Potential threat detected from ” << senderAddr << “. Packets received in last “
<< interval << ” seconds: ” << packetsReceived << std::endl;
BlockAddress(senderAddr);
}
packetsReceived = 0;
lastTime = currentTime;
}
}
}
void BlockAddress(Ipv4Address address) {
Ipv4StaticRoutingHelper ipv4RoutingHelper;
Ptr<Ipv4StaticRouting> staticRouting = ipv4RoutingHelper.GetStaticRouting(GetNode()->GetObject<Ipv4>());
staticRouting->AddHostRouteTo(address, Ipv4Address(“127.0.0.1”), 1);
std::cout << “Blocked address: ” << address << std::endl;
}
private:
Ptr<Socket> recvSocket;
uint16_t listenPort;
uint32_t packetsReceived = 0;
double lastTime = 0.0;
double interval = 1.0; // Check every 1 second
uint32_t threshold = 100; // Threshold for packet count
};
8. Set Up Applications
Install the applications on the nodes.
ApplicationContainer normalApps, attackerApps, serverApps, defenseApps;
// Normal node applications (e.g., sending normal traffic)
for (uint32_t i = 0; i < normalNodes.GetN(); ++i) {
OnOffHelper onoff(“ns3::UdpSocketFactory”, InetSocketAddress(serverNodes.Get(0)->GetObject<Ipv4>()->GetAddress(1, 0).GetLocal(), 9));
onoff.SetConstantRate(DataRate(“500kb/s”));
ApplicationContainer app = onoff.Install(normalNodes.Get(i));
app.Start(Seconds(1.0));
app.Stop(Seconds(20.0));
normalApps.Add(app);
}
// Attacker node applications (e.g., DoS attack)
for (uint32_t i = 0; i < attackerNodes.GetN(); ++i) {
OnOffHelper onoff(“ns3::UdpSocketFactory”, InetSocketAddress(serverNodes.Get(0)->GetObject<Ipv4>()->GetAddress(1, 0).GetLocal(), 9));
onoff.SetConstantRate(DataRate(“10Mb/s”));
ApplicationContainer app = onoff.Install(attackerNodes.Get(i));
app.Start(Seconds(5.0));
app.Stop(Seconds(20.0));
attackerApps.Add(app);
}
// Server node application (e.g., packet sink)
PacketSinkHelpersink(“ns3::UdpSocketFactory”, InetSocketAddress(Ipv4Address::GetAny(), 9));
serverApps.Add(sink.Install(serverNodes.Get(0)));
// Defense node application
Ptr<DefenseApplication> defenseApp = CreateObject<DefenseApplication>();
defenseApp->SetListenPort(9);
defenseNodes.Get(0)->AddApplication(defenseApp);
defenseApp->SetStartTime(Seconds(1.0));
defenseApp->SetStopTime(Seconds(20.0));
defenseApps.Add(defenseApp);
serverApps.Start(Seconds(1.0));
serverApps.Stop(Seconds(20.0));
9. Set Up Routing Protocols
Configure routing protocols for the network.
AodvHelper aodv;
InternetStackHelper internet;
internet.SetRoutingHelper(aodv);
internet.Install(normalNodes);
internet.Install(attackerNodes);
internet.Install(serverNodes);
internet.Install(defenseNodes);
10. Assign IP Addresses
Assign IP addresses to the network devices.
Ipv4AddressHelper address;
address.SetBase(“10.1.1.0”, “255.255.255.0”);
Ipv4InterfaceContainer normalInterfaces = address.Assign(normalDevices);
Ipv4InterfaceContainer attackerInterfaces = address.Assign(attackerDevices);
Ipv4InterfaceContainer serverInterfaces = address.Assign(serverDevices);
Ipv4InterfaceContainer defenseInterfaces = address.Assign(defenseDevices);
11. Run the Simulation
Configure the simulation runtime and execute it.
Simulator::Stop(Seconds(20.0));
Simulator::Run();
Simulator::Destroy();
Example of a Simple Network Defense Script
#include “ns3/core-module.h”
#include “ns3/network-module.h”
#include “ns3/internet-module.h”
#include “ns3/wifi-module.h”
#include “ns3/mobility-module.h”
#include “ns3/applications-module.h”
#include “ns3/aodv-module.h”
using namespace ns3;
class DefenseApplication : public Application {
public:
void StartApplication() override {
recvSocket = Socket::CreateSocket(GetNode(), UdpSocketFactory::GetTypeId());
recvSocket->Bind(InetSocketAddress(Ipv4Address::GetAny(), listenPort));
recvSocket->SetRecvCallback(MakeCallback(&DefenseApplication::HandleRead, this));
}
void SetListenPort(uint16_t port) {
listenPort = port;
}
void HandleRead(Ptr<Socket> socket) {
Ptr<Packet> packet;
Address from;
while ((packet = socket->RecvFrom(from))) {
packetsReceived++;
Ipv4Address senderAddr = InetSocketAddress::ConvertFrom(from).GetIpv4();
double currentTime = Simulator::Now().GetSeconds();
if (currentTime – lastTime > interval) {
if (packetsReceived > threshold) {
std::cout << “Potential threat detected from ” << senderAddr << “. Packets received in last “
<< interval << ” seconds: ” << packetsReceived << std::endl;
BlockAddress(senderAddr);
}
packetsReceived = 0;
lastTime = currentTime;
}
}
}
void BlockAddress(Ipv4Address address) {
Ipv4StaticRoutingHelper ipv4RoutingHelper;
Ptr<Ipv4StaticRouting>staticRouting= ipv4RoutingHelper.GetStaticRouting(GetNode()->GetObject<Ipv4>());
staticRouting->AddHostRouteTo(address, Ipv4Address(“127.0.0.1”), 1);
std::cout << “Blocked address: ” << address << std::endl;
}
private:
Ptr<Socket> recvSocket;
uint16_t listenPort;
uint32_t packetsReceived = 0;
double lastTime = 0.0;
double interval = 1.0; // Check every 1 second
uint32_t threshold = 100; // Threshold for packet count
};
int main(int argc, char *argv[]) {
NodeContainer normalNodes, attackerNodes, serverNodes, defenseNodes;
normalNodes.Create(3);
attackerNodes.Create(1);
serverNodes.Create(1);
defenseNodes.Create(1);
// WiFi setup
WifiHelper wifi;
wifi.SetStandard(WIFI_PHY_STANDARD_80211n_5GHZ);
WifiMacHelper mac;
mac.SetType(“ns3::AdhocWifiMac”);
YansWifiPhyHelper phy = YansWifiPhyHelper::Default();
YansWifiChannelHelper channel = YansWifiChannelHelper::Default();
phy.SetChannel(channel.Create());
NetDeviceContainer normalDevices = wifi.Install(phy, mac, normalNodes);
NetDeviceContainer attackerDevices = wifi.Install(phy, mac, attackerNodes);
NetDeviceContainer serverDevices = wifi.Install(phy, mac, serverNodes);
NetDeviceContainer defenseDevices = wifi.Install(phy, mac, defenseNodes);
// Mobility setup
MobilityHelper mobility;
mobility.SetMobilityModel(“ns3::ConstantPositionMobilityModel”);
mobility.Install(normalNodes);
mobility.Install(attackerNodes);
mobility.Install(serverNodes);
mobility.Install(defenseNodes);
// Internet stack and routing
AodvHelper aodv;
InternetStackHelper internet;
internet.SetRoutingHelper(aodv);
internet.Install(normalNodes);
internet.Install(attackerNodes);
internet.Install(serverNodes);
internet.Install(defenseNodes);
Ipv4AddressHelper address;
address.SetBase(“10.1.1.0”, “255.255.255.0”);
Ipv4InterfaceContainer normalInterfaces = address.Assign(normalDevices);
Ipv4InterfaceContainer attackerInterfaces = address.Assign(attackerDevices);
Ipv4InterfaceContainer serverInterfaces = address.Assign(serverDevices);
Ipv4InterfaceContainer defenseInterfaces = address.Assign(defenseDevices);
// Set up packet capture
PcapHelper pcapHelper;
Ptr<PcapFileWrapper> file = pcapHelper.CreateFile(“defense_capture.pcap”, std::ios::out, PcapHelper::DLT_PPP);
for (uint32_t i = 0; i < defenseDevices.GetN(); ++i) {
phy.EnablePcap(“defense_capture”, defenseDevices.Get(i), true, true);
}
// Install applications
ApplicationContainer normalApps, attackerApps, serverApps, defenseApps;
// Normal node applications (e.g., sending normal traffic)
for (uint32_t i = 0; i < normalNodes.GetN(); ++i) {
OnOffHelper onoff(“ns3::UdpSocketFactory”, InetSocketAddress(serverNodes.Get(0)->GetObject<Ipv4>()->GetAddress(1, 0).GetLocal(), 9));
onoff.SetConstantRate(DataRate(“500kb/s”));
ApplicationContainer app = onoff.Install(normalNodes.Get(i));
app.Start(Seconds(1.0));
app.Stop(Seconds(20.0));
normalApps.Add(app);
}
// Attacker node applications (e.g., DoS attack)
for (uint32_t i = 0; i < attackerNodes.GetN(); ++i) {
OnOffHelper onoff(“ns3::UdpSocketFactory”, InetSocketAddress(serverNodes.Get(0)->GetObject<Ipv4>()->GetAddress(1, 0).GetLocal(), 9));
onoff.SetConstantRate(DataRate(“10Mb/s”));
ApplicationContainer app = onoff.Install(attackerNodes.Get(i));
app.Start(Seconds(5.0));
app.Stop(Seconds(20.0));
attackerApps.Add(app);
}
// Server node application (e.g., packet sink)
PacketSinkHelpersink(“ns3::UdpSocketFactory”, InetSocketAddress(Ipv4Address::GetAny(), 9));
serverApps.Add(sink.Install(serverNodes.Get(0)));
// Defense node application
Ptr<DefenseApplication> defenseApp = CreateObject<DefenseApplication>();
defenseApp->SetListenPort(9);
defenseNodes.Get(0)->AddApplication(defenseApp);
defenseApp->SetStartTime(Seconds(1.0));
defenseApp->SetStopTime(Seconds(20.0));
defenseApps.Add(defenseApp);
serverApps.Start(Seconds(1.0));
serverApps.Stop(Seconds(20.0));
Simulator::Stop(Seconds(20.0));
Simulator::Run();
Simulator::Destroy();
return 0;
}
Atlast, we get to know how to implement the network defense in ns-3 environment. All programs that runs in ns-3 environment are carried on by our team.