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

/*

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License version 2 as

 * published by the Free Software Foundation;

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 */

// Port of ns-2/tcl/ex/simple.tcl to ns-3

//

// Network topology

//

//       n0    n1   n2   n3

//       |     |    |    |

//     =====================

//

// - CBR/UDP flows from n0 to n1, and from n3 to n0

// - UDP packet size of 210 bytes, with per-packet interval 0.00375 sec.

//   (i.e., DataRate of 448,000 bps)

// - DropTail queues 

// - Tracing of queues and packet receptions to file "csma-one-subnet.tr"

#include <iostream>

#include <fstream>

#include <string>

#include <cassert>

#include "ns3/command-line.h"

#include "ns3/ptr.h"

#include "ns3/random-variable.h"

#include "ns3/log.h"

#include "ns3/string.h"

#include "ns3/simulator.h"

#include "ns3/nstime.h"

#include "ns3/data-rate.h"

#include "ns3/ascii-trace.h"

#include "ns3/pcap-trace.h"

#include "ns3/internet-node.h"

#include "ns3/csma-channel.h"

#include "ns3/csma-net-device.h"

#include "ns3/mac48-address.h"

#include "ns3/packet-socket-address.h"

#include "ns3/socket.h"

#include "ns3/onoff-application.h"

#include "ns3/queue.h"

#include "ns3/drop-tail-queue.h"

using namespace ns3;

NS_LOG_COMPONENT_DEFINE ("CsmaPacketSocketExample");

static Ptr<CsmaNetDevice>

CreateCsmaDevice (Ptr<Node> node, Ptr<CsmaChannel> channel)

{

  Ptr<CsmaNetDevice> device = CreateObject<CsmaNetDevice> ("Address", Mac48Address::Allocate (),

                                                           "EncapsulationMode", String ("Llc"));

  node->AddDevice (device);

  device->Attach (channel);

  Ptr<Queue> queue = CreateObject<DropTailQueue> ();

  device->AddQueue (queue);

  return device;

}

int

main (int argc, char *argv[])

{

#if 0 

  LogComponentEnable ("CsmaPacketSocketExample", LOG_LEVEL_INFO);

  LogComponentEnable("Object", LOG_LEVEL_ALL);

  LogComponentEnable("Queue", LOG_LEVEL_ALL);

  LogComponentEnable("DropTailQueue", LOG_LEVEL_ALL);

  LogComponentEnable("Channel", LOG_LEVEL_ALL);

  LogComponentEnable("CsmaChannel", LOG_LEVEL_ALL);

  LogComponentEnable("NetDevice", LOG_LEVEL_ALL);

  LogComponentEnable("CsmaNetDevice", LOG_LEVEL_ALL);

  LogComponentEnable("Ipv4L3Protocol", LOG_LEVEL_ALL);

  LogComponentEnable("PacketSocket", LOG_LEVEL_ALL);

  LogComponentEnable("Socket", LOG_LEVEL_ALL);

  LogComponentEnable("UdpSocket", LOG_LEVEL_ALL);

  LogComponentEnable("UdpL4Protocol", LOG_LEVEL_ALL);

  LogComponentEnable("Ipv4L3Protocol", LOG_LEVEL_ALL);

  LogComponentEnable("Ipv4StaticRouting", LOG_LEVEL_ALL);

  LogComponentEnable("Ipv4Interface", LOG_LEVEL_ALL);

  LogComponentEnable("ArpIpv4Interface", LOG_LEVEL_ALL);

  LogComponentEnable("Ipv4LoopbackInterface", LOG_LEVEL_ALL);

  LogComponentEnable("OnOffApplication", LOG_LEVEL_ALL);

  LogComponentEnable("PacketSinkApplication", LOG_LEVEL_ALL);

  LogComponentEnable("UdpEchoClientApplication", LOG_LEVEL_ALL);

  LogComponentEnable("UdpEchoServerApplication", LOG_LEVEL_ALL);

#endif

  CommandLine cmd;

  cmd.Parse (argc, argv);

  // Here, we will explicitly create four nodes.  In more sophisticated

  // topologies, we could configure a node factory.

  NS_LOG_INFO ("Create nodes.");

  Ptr<Node> n0 = CreateObject<Node> ();

  Ptr<Node> n1 = CreateObject<Node> (); 

  Ptr<Node> n2 = CreateObject<Node> (); 

  Ptr<Node> n3 = CreateObject<Node> ();

  // create the shared medium used by all csma devices.

  NS_LOG_INFO ("Create channels.");

  Ptr<CsmaChannel> channel = CreateObject<CsmaChannel> ("BitRate", DataRate(5000000), 

                                                        "Delay", MilliSeconds(2));

  // use a helper function to connect our nodes to the shared channel.

  NS_LOG_INFO ("Build Topology.");

  Ptr<NetDevice> n0If = CreateCsmaDevice (n0, channel);

  Ptr<NetDevice> n1If = CreateCsmaDevice (n1, channel);

  Ptr<NetDevice> n2If = CreateCsmaDevice (n2, channel);

  Ptr<NetDevice> n3If = CreateCsmaDevice (n3, channel);

  // create the address which identifies n1 from n0

  PacketSocketAddress n0ToN1;

  n0ToN1.SetSingleDevice (n0If->GetIfIndex ());      // set outgoing interface for outgoing packets

  n0ToN1.SetPhysicalAddress (n1If->GetAddress ()); // set destination address for outgoing packets

  n0ToN1.SetProtocol (2);            // set arbitrary protocol for outgoing packets

  // create the address which identifies n0 from n3

  PacketSocketAddress n3ToN0;

  n3ToN0.SetSingleDevice (n3If->GetIfIndex ());

  n3ToN0.SetPhysicalAddress (n0If->GetAddress ());

  n3ToN0.SetProtocol (3);

  // Create the OnOff application to send raw datagrams of size

  // 210 bytes at a rate of 448 Kb/s

  // from n0 to n1

  NS_LOG_INFO ("Create Applications.");

  Ptr<OnOffApplication> ooff = 

    CreateObject<OnOffApplication> ("Remote", Address (n0ToN1),

                                    "OnTime", ConstantVariable(1), 

                                    "OffTime", ConstantVariable(0));

  n0->AddApplication (ooff);

  // Start the application

  ooff->Start(Seconds(1.0));

  ooff->Stop (Seconds(10.0));

  // Create a similar flow from n3 to n0, starting at time 1.1 seconds

  ooff = CreateObject<OnOffApplication> ("Remote", Address (n3ToN0),

                                         "OnTime", ConstantVariable(1), 

                                         "OffTime", ConstantVariable(0));

  n3->AddApplication (ooff);

  // Start the application

  ooff->Start(Seconds(1.1));

  ooff->Stop (Seconds(10.0));

  // Configure tracing of all enqueue, dequeue, and NetDevice receive events

  // Trace output will be sent to the csma-packet-socket.tr file

  NS_LOG_INFO ("Configure Tracing.");

  AsciiTrace asciitrace ("csma-packet-socket.tr");

  asciitrace.TraceAllNetDeviceRx ();

  asciitrace.TraceAllQueues ();

  NS_LOG_INFO ("Run Simulation.");

  Simulator::Run ();

  Simulator::Destroy ();

  NS_LOG_INFO ("Done.");

}