/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */

// DARPA NMS Campus Network Model

//

// - This topology replicates the original NMS Campus Network model

// with the exception of chord links (which were never utilized in the

// original model)

// - Link Bandwidths and Delays may not be the same as the original

// specifications 

//

// (c)2009, GTech Systems, Inc. - Alfred Park <park@gtech-systems.com>

// for timing functions

#include <cstdlib>

#include <sys/time.h>

#include <fstream>

#include "ns3/core-module.h"

#include "ns3/simulator-module.h"

#include "ns3/node-module.h"

#include "ns3/helper-module.h"

#include "ns3/global-routing-module.h"

#include "ns3/onoff-application.h"

#include "ns3/packet-sink.h"

#include "ns3/point-to-point-net-device.h"

#include "ns3/simulator.h"

using namespace std;

using namespace ns3;

typedef struct timeval TIMER_TYPE;

#define TIMER_NOW(_t) gettimeofday(&_t,NULL);

#define TIMER_SECONDS(_t) ((double)(_t).tv_sec + (_t).tv_usec*1e-6)

#define TIMER_DIFF(_t1, _t2) (TIMER_SECONDS(_t1)-TIMER_SECONDS(_t2))

NS_LOG_COMPONENT_DEFINE ("CampusNetworkModel");

void Progress ()

{

  Time now = Simulator::Now ();

  Simulator::Schedule (Seconds (0.1), Progress);

}

int

main (int argc, char *argv[])

{

  //Config::SetDefault ("ns3::Simulator::SchedulerType", StringValue ("ns3::CalendarScheduler"));

  TIMER_TYPE t0, t1, t2;

  TIMER_NOW(t0);

  cout << " ==== DARPA NMS CAMPUS NETWORK SIMULATION ====" << endl;

  LogComponentEnable ("OnOffApplication", LOG_LEVEL_INFO);

  //RandomVariable::UseGlobalSeed (1, 1, 2, 3, 5, 8);

  int nCN = 2, nLANClients = 42;

  bool nix = true;

  CommandLine cmd;

  cmd.AddValue ("CN", "Number of total CNs [2]", nCN);

  cmd.AddValue ("LAN", "Number of nodes per LAN [42]", nLANClients);

  cmd.AddValue ("NIX", "Toggle nix-vector routing", nix);

  cmd.Parse (argc,argv);

  if (nCN < 2) 

    {

      cout << "Number of total CNs (" << nCN << ") lower than minimum of 2"

        << endl;

      return 1;

    }

  cout << "Number of CNs: " << nCN << ", LAN nodes: " << nLANClients << endl;

  NodeContainer nodes_net0[nCN][3], nodes_net1[nCN][6], nodes_netLR[nCN],

                nodes_net2[nCN][14], nodes_net2LAN[nCN][7][nLANClients],

                nodes_net3[nCN][9], nodes_net3LAN[nCN][5][nLANClients];

  PointToPointHelper p2p_2gb200ms, p2p_1gb5ms, p2p_100mb1ms;

  InternetStackHelper stack;

  Ipv4InterfaceContainer ifs, ifs0[nCN][3], ifs1[nCN][6], ifs2[nCN][14],

                         ifs3[nCN][9], ifs2LAN[nCN][7][nLANClients],

                         ifs3LAN[nCN][5][nLANClients];

  Ipv4AddressHelper address;

  std::ostringstream oss;

  p2p_1gb5ms.SetDeviceAttribute ("DataRate", StringValue ("1Gbps"));

  p2p_1gb5ms.SetChannelAttribute ("Delay", StringValue ("5ms"));

  p2p_2gb200ms.SetDeviceAttribute ("DataRate", StringValue ("2Gbps"));

  p2p_2gb200ms.SetChannelAttribute ("Delay", StringValue ("200ms"));

  p2p_100mb1ms.SetDeviceAttribute ("DataRate", StringValue ("100Mbps"));

  p2p_100mb1ms.SetChannelAttribute ("Delay", StringValue ("1ms"));

  // Setup NixVector Routing

  Ipv4NixVectorHelper nixRouting;

  Ipv4StaticRoutingHelper staticRouting;

  Ipv4ListRoutingHelper list;

  list.Add (staticRouting, 0);

  list.Add (nixRouting, 10);

  if (nix)

    {

      stack.SetRoutingHelper (list);

    }

  // Create Campus Networks

  for (int z = 0; z < nCN; ++z) 

    {

      cout << "Creating Campus Network " << z << ":" << endl;

      // Create Net0

      cout << "  SubNet [ 0";

      for (int i = 0; i < 3; ++i) 

        {

          nodes_net0[z][i].Create (1);

          stack.Install (nodes_net0[z][i]);

        }

      nodes_net0[z][0].Add (nodes_net0[z][1].Get (0));

      nodes_net0[z][1].Add (nodes_net0[z][2].Get (0));

      nodes_net0[z][2].Add (nodes_net0[z][0].Get (0));

      NetDeviceContainer ndc0[3];

      for (int i = 0; i < 3; ++i) 

        {

          ndc0[i] = p2p_1gb5ms.Install (nodes_net0[z][i]);

        }

      // Create Net1

      cout << " 1";

      for (int i = 0; i < 6; ++i) 

        {

          nodes_net1[z][i].Create (1);

          stack.Install (nodes_net1[z][i]);

        }

      nodes_net1[z][0].Add (nodes_net1[z][1].Get (0));

      nodes_net1[z][2].Add (nodes_net1[z][0].Get (0));

      nodes_net1[z][3].Add (nodes_net1[z][0].Get (0));

      nodes_net1[z][4].Add (nodes_net1[z][1].Get (0));

      nodes_net1[z][5].Add (nodes_net1[z][1].Get (0));

      NetDeviceContainer ndc1[6];

      for (int i = 0; i < 6; ++i) 

        {

          if (i == 1)

            {

              continue;

            }

          ndc1[i] = p2p_1gb5ms.Install (nodes_net1[z][i]);

        }

      // Connect Net0 <-> Net1

      NodeContainer net0_1;

      net0_1.Add (nodes_net0[z][2].Get (0));

      net0_1.Add (nodes_net1[z][0].Get (0));

      NetDeviceContainer ndc0_1;

      ndc0_1 = p2p_1gb5ms.Install (net0_1);

      oss.str("");

      oss << 10 + z << ".1.252.0";

      address.SetBase (oss.str ().c_str (), "255.255.255.0");

      ifs = address.Assign (ndc0_1);

      // Create Net2

      cout << " 2";

      for (int i = 0; i < 14; ++i) 

        {

          nodes_net2[z][i].Create (1);

          stack.Install (nodes_net2[z][i]);

        }

      nodes_net2[z][0].Add (nodes_net2[z][1].Get (0));

      nodes_net2[z][2].Add (nodes_net2[z][0].Get (0));

      nodes_net2[z][1].Add (nodes_net2[z][3].Get (0));

      nodes_net2[z][3].Add (nodes_net2[z][2].Get (0));

      nodes_net2[z][4].Add (nodes_net2[z][2].Get (0));

      nodes_net2[z][5].Add (nodes_net2[z][3].Get (0));

      nodes_net2[z][6].Add (nodes_net2[z][5].Get (0));

      nodes_net2[z][7].Add (nodes_net2[z][2].Get (0));

      nodes_net2[z][8].Add (nodes_net2[z][3].Get (0));

      nodes_net2[z][9].Add (nodes_net2[z][4].Get (0));

      nodes_net2[z][10].Add (nodes_net2[z][5].Get (0));

      nodes_net2[z][11].Add (nodes_net2[z][6].Get (0));

      nodes_net2[z][12].Add (nodes_net2[z][6].Get (0));

      nodes_net2[z][13].Add (nodes_net2[z][6].Get (0));

      NetDeviceContainer ndc2[14];

      for (int i = 0; i < 14; ++i) 

        {

          ndc2[i] = p2p_1gb5ms.Install (nodes_net2[z][i]);

        }

      NetDeviceContainer ndc2LAN[7][nLANClients];

      for (int i = 0; i < 7; ++i) 

        {

          oss.str ("");

          oss << 10 + z << ".4." << 15 + i << ".0";

          address.SetBase (oss.str ().c_str (), "255.255.255.0");

          for (int j = 0; j < nLANClients; ++j) 

            {

              nodes_net2LAN[z][i][j].Create (1);

              stack.Install (nodes_net2LAN[z][i][j]);

              nodes_net2LAN[z][i][j].Add (nodes_net2[z][i+7].Get (0));

              ndc2LAN[i][j] = p2p_100mb1ms.Install (nodes_net2LAN[z][i][j]);

              ifs2LAN[z][i][j] = address.Assign (ndc2LAN[i][j]);

            }

        }

      // Create Net3

      cout << " 3 ]" << endl;

      for (int i = 0; i < 9; ++i) 

        {

          nodes_net3[z][i].Create (1);

          stack.Install(nodes_net3[z][i]);

        }

      nodes_net3[z][0].Add (nodes_net3[z][1].Get (0));

      nodes_net3[z][1].Add (nodes_net3[z][2].Get (0));

      nodes_net3[z][2].Add (nodes_net3[z][3].Get (0));

      nodes_net3[z][3].Add (nodes_net3[z][1].Get (0));

      nodes_net3[z][4].Add (nodes_net3[z][0].Get (0));

      nodes_net3[z][5].Add (nodes_net3[z][0].Get (0));

      nodes_net3[z][6].Add (nodes_net3[z][2].Get (0));

      nodes_net3[z][7].Add (nodes_net3[z][3].Get (0));

      nodes_net3[z][8].Add (nodes_net3[z][3].Get (0));

      NetDeviceContainer ndc3[9];

      for (int i = 0; i < 9; ++i) 

        {

          ndc3[i] = p2p_1gb5ms.Install (nodes_net3[z][i]);

        }

      NetDeviceContainer ndc3LAN[5][nLANClients];

      for (int i = 0; i < 5; ++i) 

        {

          oss.str ("");

          oss << 10 + z << ".5." << 10 + i << ".0";

          address.SetBase (oss.str ().c_str (), "255.255.255.255");

          for (int j = 0; j < nLANClients; ++j) 

            {

              nodes_net3LAN[z][i][j].Create (1);

              stack.Install (nodes_net3LAN[z][i][j]);

              nodes_net3LAN[z][i][j].Add (nodes_net3[z][i+4].Get (0));

              ndc3LAN[i][j] = p2p_100mb1ms.Install (nodes_net3LAN[z][i][j]);

              ifs3LAN[z][i][j] = address.Assign (ndc3LAN[i][j]);

            }

        }

      cout << "  Connecting Subnets..." << endl;

      // Create Lone Routers (Node 4 & 5) 

      nodes_netLR[z].Create (2);

      stack.Install (nodes_netLR[z]);

      NetDeviceContainer ndcLR;

      ndcLR = p2p_1gb5ms.Install (nodes_netLR[z]);

      // Connect Net2/Net3 through Lone Routers to Net0

      NodeContainer net0_4, net0_5, net2_4a, net2_4b, net3_5a, net3_5b;

      net0_4.Add (nodes_netLR[z].Get (0));

      net0_4.Add (nodes_net0[z][0].Get (0));

      net0_5.Add (nodes_netLR[z].Get  (1));

      net0_5.Add (nodes_net0[z][1].Get (0));

      net2_4a.Add (nodes_netLR[z].Get (0));

      net2_4a.Add (nodes_net2[z][0].Get (0));

      net2_4b.Add (nodes_netLR[z].Get (1));

      net2_4b.Add (nodes_net2[z][1].Get (0));

      net3_5a.Add (nodes_netLR[z].Get (1));

      net3_5a.Add (nodes_net3[z][0].Get (0));

      net3_5b.Add (nodes_netLR[z].Get (1));

      net3_5b.Add (nodes_net3[z][1].Get (0));

      NetDeviceContainer ndc0_4, ndc0_5, ndc2_4a, ndc2_4b, ndc3_5a, ndc3_5b;

      ndc0_4 = p2p_1gb5ms.Install (net0_4);

      oss.str ("");

      oss << 10 + z << ".1.253.0";

      address.SetBase (oss.str ().c_str (), "255.255.255.0");

      ifs = address.Assign (ndc0_4);

      ndc0_5 = p2p_1gb5ms.Install (net0_5);

      oss.str ("");

      oss << 10 + z << ".1.254.0";

      address.SetBase (oss.str ().c_str (), "255.255.255.0");

      ifs = address.Assign (ndc0_5);

      ndc2_4a = p2p_1gb5ms.Install (net2_4a);

      oss.str ("");

      oss << 10 + z << ".4.253.0";

      address.SetBase (oss.str ().c_str (), "255.255.255.0");

      ifs = address.Assign (ndc2_4a);

      ndc2_4b = p2p_1gb5ms.Install (net2_4b);

      oss.str ("");

      oss << 10 + z << ".4.254.0";

      address.SetBase (oss.str ().c_str (), "255.255.255.0");

      ifs = address.Assign (ndc2_4b);

      ndc3_5a = p2p_1gb5ms.Install (net3_5a);

      oss.str ("");

      oss << 10 + z << ".5.253.0";

      address.SetBase (oss.str ().c_str (), "255.255.255.0");

      ifs = address.Assign (ndc3_5a);

      ndc3_5b = p2p_1gb5ms.Install (net3_5b);

      oss.str ("");

      oss << 10 + z << ".5.254.0";

      address.SetBase (oss.str ().c_str (), "255.255.255.0");

      ifs = address.Assign (ndc3_5b);

      // Assign IP addresses

      cout << "  Assigning IP addresses..." << endl;

      for (int i = 0; i < 3; ++i) 

        {

          oss.str ("");

          oss << 10 + z << ".1." << 1 + i << ".0";

          address.SetBase (oss.str ().c_str (), "255.255.255.0");

          ifs0[z][i] = address.Assign (ndc0[i]);

        }

      for (int i = 0; i < 6; ++i) 

        {

          if (i == 1) 

            {

              continue;

            }

          oss.str ("");

          oss << 10 + z << ".2." << 1 + i << ".0";

          address.SetBase (oss.str ().c_str (), "255.255.255.0");

          ifs1[z][i] = address.Assign (ndc1[i]);

        }

      oss.str ("");

      oss << 10 + z << ".3.1.0";

      address.SetBase (oss.str ().c_str (), "255.255.255.0");

      ifs = address.Assign (ndcLR);

      for (int i = 0; i < 14; ++i) 

        {

          oss.str ("");

          oss << 10 + z << ".4." << 1 + i << ".0";

          address.SetBase (oss.str ().c_str (), "255.255.255.0");

          ifs2[z][i] = address.Assign (ndc2[i]);

        }

      for (int i = 0; i < 9; ++i) 

        {

          oss.str ("");

          oss << 10 + z << ".5." << 1 + i << ".0";

          address.SetBase (oss.str ().c_str (), "255.255.255.0");

          ifs3[z][i] = address.Assign (ndc3[i]);

        }

    }

    // Create Ring Links

  if (nCN > 1) 

    {

      cout << "Forming Ring Topology..." << endl;

      NodeContainer nodes_ring[nCN];

      for (int z = 0; z < nCN-1; ++z) 

        {

          nodes_ring[z].Add (nodes_net0[z][0].Get (0));

          nodes_ring[z].Add (nodes_net0[z+1][0].Get (0));

        }

      nodes_ring[nCN-1].Add (nodes_net0[nCN-1][0].Get (0));

      nodes_ring[nCN-1].Add (nodes_net0[0][0].Get (0));

      NetDeviceContainer ndc_ring[nCN];

      for (int z = 0; z < nCN; ++z) 

        {

          ndc_ring[z] = p2p_2gb200ms.Install (nodes_ring[z]);

          oss.str ("");

          oss << "254.1." << z + 1 << ".0";

          address.SetBase (oss.str ().c_str (), "255.255.255.0");

          ifs = address.Assign (ndc_ring[z]);

        }

    }

  // Create Traffic Flows

  cout << "Creating TCP Traffic Flows:" << endl;

  Config::SetDefault ("ns3::OnOffApplication::MaxBytes", UintegerValue (500000));

  Config::SetDefault ("ns3::OnOffApplication::OnTime",

      RandomVariableValue (ConstantVariable (1)));

  Config::SetDefault ("ns3::OnOffApplication::OffTime",

      RandomVariableValue (ConstantVariable (0)));

  Config::SetDefault ("ns3::TcpSocket::SegmentSize", UintegerValue (512));

  UniformVariable urng;

  int r1;

  double r2;

  for (int z = 0; z < nCN; ++z) 

    {

      int x = z + 1;

      if (z == nCN - 1) 

        {

          x = 0;

        }

      // Subnet 2 LANs

      cout << "  Campus Network " << z << " Flows [ Net2 ";

      for (int i = 0; i < 7; ++i) 

        {

          for (int j = 0; j < nLANClients; ++j) 

            {

              // Sinks

              PacketSinkHelper sinkHelper ("ns3::TcpSocketFactory",

                  InetSocketAddress (Ipv4Address::GetAny (), 9999));

              ApplicationContainer sinkApp = sinkHelper.Install (

                  nodes_net2LAN[z][i][j].Get (0));

              sinkApp.Start (Seconds (100.0));

              // Sources

              r1 = 2 + (int)(4 * urng.GetValue ());

              r2 = 100 + (10 * urng.GetValue ());;

              OnOffHelper client ("ns3::TcpSocketFactory", Address ());

              AddressValue remoteAddress(InetSocketAddress (

                  ifs2LAN[z][i][j].GetAddress (0), 9999));

            client.SetAttribute ("Remote", remoteAddress);

            ApplicationContainer clientApp;

            clientApp.Add (client.Install (nodes_net1[x][r1].Get (0)));

            clientApp.Start (Seconds (r2));

          }

        }

      // Subnet 3 LANs

      cout << "Net3 ]" << endl;

      for (int i = 0; i < 5; ++i) 

        {

          for (int j = 0; j < nLANClients; ++j) 

            {

              // Sinks

              PacketSinkHelper sinkHelper ("ns3::TcpSocketFactory",

                  InetSocketAddress (Ipv4Address::GetAny (), 9999));

              ApplicationContainer sinkApp = sinkHelper.Install (

                  nodes_net3LAN[z][i][j].Get (0));

              sinkApp.Start (Seconds (100.0));

              // Sources

              r1 = 2 + (int)(4 * urng.GetValue ());

              r2 = 100 + (10 * urng.GetValue ());;

              OnOffHelper client ("ns3::TcpSocketFactory", Address ());

              AddressValue remoteAddress (InetSocketAddress (

                  ifs2LAN[z][i][j].GetAddress (0), 9999));

              client.SetAttribute ("Remote", remoteAddress);

              ApplicationContainer clientApp;

              clientApp.Add (client.Install (nodes_net1[x][r1].Get (0)));

              clientApp.Start (Seconds (r2));

            }

        }

    }

  cout << "Created " << NodeList::GetNNodes () << " nodes." << endl;

  TIMER_TYPE routingStart;

  TIMER_NOW (routingStart);

  if (nix)

    {

      // Calculate routing tables

      cout << "Using Nix-vectors..." << endl;

    }

  else

    {

      // Calculate routing tables

      cout << "Populating Global Static Routing Tables..." << endl;

      Ipv4GlobalRoutingHelper::PopulateRoutingTables ();

    }

  TIMER_TYPE routingEnd;

  TIMER_NOW (routingEnd);

  cout << "Routing tables population took " 

       << TIMER_DIFF (routingEnd, routingStart) << endl;

#if 0 

  std::ofstream ascii;

  ascii.open("nms_p2p_nix.tr");

  PointToPointHelper::EnableAsciiAll(ascii);

  CsmaHelper::EnableAsciiAll(ascii);

#endif

#if 0

  PointToPointHelper::EnablePcapAll("nms_p2p");

  CsmaHelper::EnablePcapAll("nms_csma");

#endif

  Simulator::ScheduleNow (Progress);

  cout << "Running simulator..." << endl;

  TIMER_NOW (t1);

  Simulator::Stop (Seconds (200.0));

  Simulator::Run ();

  TIMER_NOW (t2);

  cout << "Simulator finished." << endl;

  Simulator::Destroy ();

  double d1 = TIMER_DIFF (t1, t0), d2 = TIMER_DIFF (t2, t1);

  cout << "-----" << endl << "Runtime Stats:" << endl;

  cout << "Simulator init time: " << d1 << endl;

  cout << "Simulator run time: " << d2 << endl;

  cout << "Total elapsed time: " << d1+d2 << endl;

  return 0;

}