Init final simulator source

7 files changed, 302 insertions, 0 deletions

diff --git a/ns3-simulator/Makefile b/ns3-simulator/Makefile
new file mode 100644
index 0000000..c2e1d35
--- /dev/null
+++ b/ns3-simulator/Makefile
@@ -0,0 +1,20 @@
+
+NS3_ARGS= -D NS3_LOG_ENABLE -L ${NS3_PATH}/build/lib -I ${NS3_PATH}/build/
+NS3_ARGS+=$(addprefix -l, $(subst lib,,$(subst .so,,$(notdir $(wildcard ${NS3_PATH}/build/lib/*.so)))))
+
+SRC=main.cc modules/platform.cc modules/energy.cc modules/callbacks.cc
+EXEC=simulator
+
+all: $(EXEC)
+
+$(EXEC): $(SRC)
+	g++  $(NS3_ARGS) $(SRC) -o $@
+
+run:
+	@LD_LIBRARY_PATH=$LD_LIBRARY_PATH:${NS3_PATH}/build/lib ./wifi-test
+
+export:
+	@echo export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:${NS3_PATH}/build/lib
+
+clean:
+	- rm $(EXEC)
diff --git a/ns3-simulator/main.cc b/ns3-simulator/main.cc
new file mode 100644
index 0000000..0c5c405
--- /dev/null
+++ b/ns3-simulator/main.cc
@@ -0,0 +1,41 @@
+#include "modules/modules.hpp"
+
+NS_LOG_COMPONENT_DEFINE ("WIFISensorsSimulator");
+
+
+int main(int argc, char* argv[]){
+
+  uint32_t sensorsFrequency=1;
+  uint32_t sensorsPktSize=150;
+  uint32_t sensorsNumber=2;
+  uint32_t nbHop=5;
+
+  CommandLine cmd;
+  cmd.AddValue ("sensorsSendInterval", "Number of temperature measurement per second", sensorsFrequency);
+  cmd.AddValue ("sensorsPktSize", "Sensor measurements packet size (bytes)", sensorsPktSize);
+  cmd.AddValue ("sensorsNumber", "Number of sensors", sensorsNumber);
+  cmd.AddValue ("nbHop", "Number of hop between AP and Cloud sensors", sensorsNumber);
+  cmd.Parse (argc, argv);
+
+  //LogComponentEnable("UdpEchoClientApplication", LOG_LEVEL_INFO);
+  //LogComponentEnable("PacketSink", LOG_LEVEL_INFO);
+
+  // Setup Simulations
+  CloudInfos cloud=buildEdgeAndCloud(nbHop);
+  Cell c=createCell(sensorsNumber,cloud.first.Get(0));
+  applyScenarios(c,sensorsPktSize,sensorsFrequency,cloud); // Send data from Sensors to Cloud
+  //  setupEnergy(c);
+  Ipv4GlobalRoutingHelper::PopulateRoutingTables ();
+
+
+
+  
+  // Run simulators
+  Simulator::Stop (Seconds (20));
+  Simulator::Run ();
+  Simulator::Destroy (); // Destroy
+
+
+  
+  return(0);
+}
diff --git a/ns3-simulator/modules/callbacks.cc b/ns3-simulator/modules/callbacks.cc
new file mode 100644
index 0000000..a552eb9
--- /dev/null
+++ b/ns3-simulator/modules/callbacks.cc
@@ -0,0 +1,10 @@
+
+#include "modules.hpp"
+
+void PktReceived(std::string nodeName,Ptr< const Packet > packet, const Address &address){
+  NS_LOG_UNCOND("Node " << nodeName << " receive a packet");
+}
+
+void EnergyUpdated(std::string nodeName,double oldValue, double newValue){
+  NS_LOG_UNCOND("Node " << nodeName << " consumes " << newValue << "J");
+}
diff --git a/ns3-simulator/modules/energy.cc b/ns3-simulator/modules/energy.cc
new file mode 100644
index 0000000..b07c835
--- /dev/null
+++ b/ns3-simulator/modules/energy.cc
@@ -0,0 +1,51 @@
+
+#include "modules.hpp"
+
+void setupEnergy(Cell cell){
+  NodeContainer nodes(cell.first.first,cell.first.second);
+  NetDeviceContainer nodesNetDev(cell.second.first,cell.second.second);
+
+  // Install energy source
+  BasicEnergySourceHelper edgeBasicSourceHelper;
+  edgeBasicSourceHelper.Set ("BasicEnergySourceInitialEnergyJ", DoubleValue (2.9009));
+  edgeBasicSourceHelper.Set ("BasicEnergySupplyVoltageV", DoubleValue (3.3));
+  EnergySourceContainer apEdgeNodesSources = edgeBasicSourceHelper.Install (cell.first.first);
+  EnergySourceContainer wifiEdgeNodesSources = edgeBasicSourceHelper.Install (cell.first.second);
+
+  // Install device energy model
+  WifiRadioEnergyModelHelper radioEnergyHelper;
+  radioEnergyHelper.Set ("TxCurrentA", DoubleValue (0.38));
+  radioEnergyHelper.Set ("RxCurrentA", DoubleValue (0.313));
+  radioEnergyHelper.Set ("IdleCurrentA", DoubleValue (0.273));
+  DeviceEnergyModelContainer edgeApDeviceModels = radioEnergyHelper.Install (cell.second.first, apEdgeNodesSources);
+  DeviceEnergyModelContainer edgeDeviceModels = radioEnergyHelper.Install (cell.second.second, wifiEdgeNodesSources);
+
+  
+  // Trace
+  DeviceEnergyModelContainer energyModels(edgeApDeviceModels, edgeDeviceModels);
+  DeviceEnergyModelContainer::Iterator it=energyModels.Begin();
+
+  int i=0;
+  while(it!=energyModels.End()){
+    (*it)->TraceConnect ("TotalEnergyConsumption", std::to_string(i),MakeCallback (&EnergyUpdated));
+    it++;
+    i++;
+  }
+
+
+
+
+  //   Ptr<BasicEnergySource> basicSourcePtr0 = DynamicCast<BasicEnergySource> (wifiEdgeNodesSources.Get (0));
+
+  // //basicSourcePtr0->TraceConnectWithoutContext ("RemainingEnergy", MakeCallback (&RemainingEnergy));
+  // //device energy model
+
+  // Ptr<DeviceEnergyModel> basicRadioModelPtr0 =
+  // basicSourcePtr0->FindDeviceEnergyModels ("ns3::WifiRadioEnergyModel").Get (0);
+
+  // NS_ASSERT (basicRadioModelPtr0 != NULL);
+  // basicRadioModelPtr0->TraceConnectWithoutContext ("TotalEnergyConsumption", MakeCallback (&TotalEnergy));
+
+  
+}
+
diff --git a/ns3-simulator/modules/modules.hpp b/ns3-simulator/modules/modules.hpp
new file mode 100644
index 0000000..fa6a255
--- /dev/null
+++ b/ns3-simulator/modules/modules.hpp
@@ -0,0 +1,56 @@
+
+
+#ifndef MODULES_HPP
+#define MODULES_HPP
+
+#include "ns3/command-line.h"
+#include "ns3/config.h"
+#include "ns3/string.h"
+#include "ns3/log.h"
+#include "ns3/yans-wifi-helper.h"
+#include "ns3/ssid.h"
+#include "ns3/mobility-helper.h"
+#include "ns3/on-off-helper.h"
+#include "ns3/yans-wifi-channel.h"
+#include "ns3/mobility-model.h"
+#include "ns3/packet-sink.h"
+#include "ns3/packet-sink-helper.h"
+#include "ns3/udp-echo-helper.h"
+#include "ns3/tcp-westwood.h"
+#include "ns3/internet-stack-helper.h"
+#include "ns3/ipv4-address-helper.h"
+#include "ns3/ipv4-global-routing-helper.h"
+#include "ns3/constant-position-mobility-model.h"
+#include "ns3/energy-module.h"
+#include "ns3/wifi-radio-energy-model-helper.h"
+#include "ns3/point-to-point-helper.h"
+
+// C++ library
+#include <iostream>
+#include <utility> // To use std::pair
+#include <iomanip>
+
+using namespace ns3;
+
+// Data types
+typedef std::pair<NodeContainer,NodeContainer> CellNodes; // Format (APNode, SensorsNodes)
+typedef std::pair<NetDeviceContainer,NetDeviceContainer> CellNetDevices; // Format (APNetDev, SensorsNetDev)
+typedef std::pair<CellNodes,CellNetDevices> Cell;
+typedef std::pair<Ipv4Address,int> EndPoint; // Format (IP,Port)
+typedef std::pair<NodeContainer,EndPoint> CloudInfos; // Format (CloudHops,CloudEndPoint), here data sent to CloudEndPoint
+                                                       // via CloudHops.Get(0) will flow throw all hops until the reaching the cloud
+
+// Platform functions
+Cell createCell(uint32_t nbSensors, Ptr<ns3::Node> ap);
+CloudInfos buildEdgeAndCloud(int nbOp);
+void applyScenarios(Cell cell,int sensorsPktSize, int sensorsSendInterval, CloudInfos cloudInfos);
+
+// Energy functions
+void setupEnergy(Cell cell);
+
+
+// Callbacks
+void PktReceived(std::string nodeName,Ptr< const Packet > packet, const Address &address);
+void EnergyUpdated(std::string nodeName,double oldValue, double newValue);
+
+#endif
diff --git a/ns3-simulator/modules/platform.cc b/ns3-simulator/modules/platform.cc
new file mode 100644
index 0000000..81d53cb
--- /dev/null
+++ b/ns3-simulator/modules/platform.cc
@@ -0,0 +1,124 @@
+#include "modules.hpp"
+
+/**
+ * Create a sensors cell base on
+ * nbSensors Number of temperature sensors in the cell
+ * ap the Access Point (usually linked to the cloud)
+ */
+Cell createCell(uint32_t nbSensors, Ptr<ns3::Node> ap){
+  // Create sensors
+  NodeContainer sensors;
+  sensors.Create(nbSensors);
+
+  // Place nodes somehow, this is required by every wireless simulation
+  for (uint8_t i = 0; i < nbSensors; ++i)
+    {
+      sensors.Get (i)->AggregateObject (CreateObject<ConstantPositionMobilityModel> ());
+    }
+  ap->AggregateObject (CreateObject<ConstantPositionMobilityModel> ());
+
+  // To apply XXWifiPhy and WifiMac on sensors 
+  WifiHelper wifiHelper;
+  wifiHelper.SetStandard (WIFI_PHY_STANDARD_80211n_5GHZ);
+
+  /* Set up Legacy Channel */
+  YansWifiChannelHelper wifiChannel;
+  wifiChannel.SetPropagationDelay ("ns3::ConstantSpeedPropagationDelayModel");
+  wifiChannel.AddPropagationLoss ("ns3::FriisPropagationLossModel", "Frequency", DoubleValue (5e9));
+
+  /* Setup Physical Layer */
+  YansWifiPhyHelper wifiPhy = YansWifiPhyHelper::Default ();
+  wifiPhy.SetChannel (wifiChannel.Create ());
+  wifiPhy.Set ("TxPowerStart", DoubleValue (10.0));
+  wifiPhy.Set ("TxPowerEnd", DoubleValue (10.0));
+  wifiPhy.Set ("TxPowerLevels", UintegerValue (1));
+  wifiPhy.Set ("TxGain", DoubleValue (0));
+  wifiPhy.Set ("RxGain", DoubleValue (0));
+  wifiPhy.Set ("RxNoiseFigure", DoubleValue (10));
+  wifiPhy.Set ("CcaMode1Threshold", DoubleValue (-79));
+  wifiPhy.Set ("EnergyDetectionThreshold", DoubleValue (-79 + 3));
+  //  wifiPhy.SetErrorRateModel ("ns3::YansErrorRateModel");
+  wifiHelper.SetRemoteStationManager ("ns3::ConstantRateWifiManager",
+                                      "DataMode", StringValue ("HtMcs7"),
+                                      "ControlMode", StringValue ("HtMcs0"));
+  /* Configure AP */
+  Ssid ssid = Ssid ("network");
+  WifiMacHelper wifiMac;
+  wifiMac.SetType ("ns3::ApWifiMac", "Ssid", SsidValue (ssid));
+  NetDeviceContainer apNetDevice;
+  apNetDevice = wifiHelper.Install (wifiPhy, wifiMac, ap);
+  /* Configure STA */
+  wifiMac.SetType ("ns3::StaWifiMac", "Ssid", SsidValue (ssid));
+  NetDeviceContainer sensorsNetDevices;
+  sensorsNetDevices = wifiHelper.Install (wifiPhy, wifiMac, sensors);
+
+  return(std::make_pair(std::make_pair(ap,sensors),std::make_pair(apNetDevice,sensorsNetDevices)));
+}
+
+/**
+ * Install network stack and applications
+ */
+void applyScenarios(Cell cell,int sensorsPktSize, int sensorsSendInterval, CloudInfos cloudInfos){
+  NodeContainer ap=cell.first.first;
+  NodeContainer sensors=cell.first.second;
+  NetDeviceContainer apNetDev= cell.second.first;
+  NetDeviceContainer sensorsNetDev= cell.second.second;
+  
+  // 6. Install TCP/IP stack & assign IP addresses
+  InternetStackHelper internet;
+  //  internet.Install (ap);
+  internet.Install (sensors);
+  Ipv4AddressHelper ipv4;
+  ipv4.SetBase ("10.0.0.0", "255.255.255.0");
+  Ipv4InterfaceContainer apInt,sensorsInt;
+  apInt=ipv4.Assign(apNetDev);
+  sensorsInt=ipv4.Assign(sensorsNetDev);
+
+  UdpEchoClientHelper echoClientHelper (InetSocketAddress (cloudInfos.second.first, cloudInfos.second.second));
+  //  echoClientHelper.SetAttribute ("MaxPackets", UintegerValue (10));
+  echoClientHelper.SetAttribute ("Interval", TimeValue (Seconds (sensorsSendInterval)));
+  echoClientHelper.SetAttribute ("PacketSize", UintegerValue (sensorsPktSize));
+  ApplicationContainer pingApps;
+
+  // again using different start times to workaround Bug 388 and Bug 912
+  echoClientHelper.SetAttribute ("StartTime", TimeValue (Seconds (1))); // Start at 1 (WIFI seems to not work when t<1)
+  echoClientHelper.Install (sensors);
+}
+
+CloudInfos buildEdgeAndCloud(int nbOp){
+ 
+  NodeContainer OpNodes;
+  OpNodes.Create(nbOp);
+
+  InternetStackHelper stack;
+  stack.Install(OpNodes);
+
+  Ipv4Address cloudIP;
+  int cloudPort=99;
+  for(int i=0;i<nbOp-1;i++){
+    NodeContainer curNodes(OpNodes.Get(i),OpNodes.Get(i+1));
+
+    PointToPointHelper pointToPoint;
+    pointToPoint.SetDeviceAttribute ("DataRate", StringValue ("5Mbps"));
+    pointToPoint.SetChannelAttribute ("Delay", StringValue ("2ms"));
+
+    NetDeviceContainer p2pDevices;
+    p2pDevices = pointToPoint.Install (curNodes);
+
+    Ipv4AddressHelper address;
+    address.SetBase (("10.1."+std::to_string(i)+".0").c_str(), "255.255.255.0");
+    Ipv4InterfaceContainer p2pInterfaces;
+    p2pInterfaces = address.Assign (p2pDevices);
+
+    if(i==nbOp-2){ // If we are on the last FOR LOOP
+      cloudIP=p2pInterfaces.GetAddress (1); // Get Last node interface
+      PacketSinkHelper apSink("ns3::UdpSocketFactory",InetSocketAddress (Ipv4Address::GetAny (), cloudPort));
+      ApplicationContainer sinkApp=apSink.Install(curNodes.Get(1)); // Instal sink on last node
+      sinkApp.Get(0)->TraceConnect("Rx","CloudSwitch",MakeCallback(&PktReceived));
+      sinkApp.Start (Seconds (0));
+    }
+  }
+
+  return(std::make_pair(OpNodes,std::make_pair(cloudIP,cloudPort)));
+
+}
diff --git a/ns3-simulator/simulator b/ns3-simulator/simulator
new file mode 100755
index 0000000..8a6f8bd
--- /dev/null
+++ b/ns3-simulator/simulator