/* -*- 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

 */

#include "ns3/core-module.h"

#include "ns3/simulator-module.h"

#include "ns3/node-module.h"

#include "ns3/helper-module.h"

#include "ns3/global-route-manager.h"

#include "ns3/on-off-helper.h"

#include "ns3/v4ping-helper.h"

#include "ns3/ipv4-l3-protocol.h"

using namespace ns3;

NS_LOG_COMPONENT_DEFINE ("NetfilterExample");

  int 

main (int argc, char *argv[])

{

  LogComponentEnable("UdpEchoClientApplication", LOG_LEVEL_INFO);

  LogComponentEnable("UdpEchoServerApplication", LOG_LEVEL_INFO);

  uint16_t port = 9;

  NodeContainer first;

  first.Create (2);

  NodeContainer second;

  second.Add ( first.Get (1) );

  second.Create (1);

  Packet::EnablePrinting();

  PointToPointHelper pointToPoint;

  pointToPoint.SetDeviceAttribute ("DataRate", StringValue ("5Mbps"));

  pointToPoint.SetChannelAttribute ("Delay", StringValue ("2ms"));

  //V4PingHelper ping ( Ipv4Address ("10.2.2.2"));

  NetDeviceContainer devices1;

  devices1 = pointToPoint.Install (first);

  NetDeviceContainer devices2;

  devices2 = pointToPoint.Install (second);

  InternetStackHelper stack;

  stack.Install (first);

  stack.Install (second.Get(1));

  Ipv4AddressHelper address1;

  address1.SetBase ("192.168.1.0", "255.255.255.0");

  Ipv4AddressHelper address2;

  address2.SetBase ("203.82.48.0", "255.255.255.0");

  Ipv4InterfaceContainer firstInterfaces = address1.Assign (devices1);

  Ipv4InterfaceContainer secondInterfaces = address2.Assign (devices2);

  Ptr <Ipv4> ipv4 = first.Get (1)->GetObject<Ipv4> ();

  std::cout << "Number of interfaces on node " << first.Get (1)->GetId () << ": " << ipv4->GetNInterfaces () << std::endl;

  //std::cout << "Device number: " << first.Get (1)->GetDevice (1) << std::endl;

  //int32_t ifNumber = ipv4->GetInterfaceForAddress ( Ipv4Address ("203.82.48.1"));

  //Ptr<Ipv4Interface> ipv4Interface = ipv4->GetInterface ( ifNumber );

  Ptr <Ipv4L3Protocol> ipv4L3 = DynamicCast <Ipv4L3Protocol>(first.Get (1)->GetObject<Ipv4> ());

  Ipv4Netfilter *netfilter = ipv4L3->GetNetfilter ();

  netfilter->EnableNat ();

  std::cout << "Adding rule at node " << first.Get (1)->GetId () << ", device " << first.Get (1)->GetDevice (1)->GetIfIndex () << std::endl;

  netfilter->AddNatRule ( NatRule (Ipv4Address ("192.168.1.1"), Ipv4Address ("203.82.48.1"), first.Get (1)->GetDevice (1)));

  //netfilter.SetAttribute ("EnableNat", 1);

  UdpEchoServerHelper echoServer (9);

  ApplicationContainer serverApps = echoServer.Install (second.Get (1));

  serverApps.Start (Seconds (1.0));

  serverApps.Stop (Seconds (10.0));

  UdpEchoClientHelper echoClient (secondInterfaces.GetAddress (1), 9);

  echoClient.SetAttribute ("MaxPackets", UintegerValue (1));

  echoClient.SetAttribute ("Interval", TimeValue (Seconds (1.)));

  echoClient.SetAttribute ("PacketSize", UintegerValue (512));

  ApplicationContainer clientApps = echoClient.Install (first.Get (0));

  clientApps.Start (Seconds (2.0));

  clientApps.Stop (Seconds (10.0));

  OnOffHelper onOff ( "ns3::TcpSocketFactory", 

                      Address (InetSocketAddress ("192.168.1.1", port)));

  onOff.SetAttribute ("MaxBytes", UintegerValue (512));

  ApplicationContainer onOffApp = onOff.Install (second.Get (1));

  PacketSinkHelper packetSink ( "ns3::TcpSocketFactory",

                                Address (InetSocketAddress ("192.168.1.1", port)));

  ApplicationContainer packetSinkApp = packetSink.Install (first.Get (0));

  /*packetSinkApp.Start (Seconds (2.0));

  packetSinkApp.Stop (Seconds (4.0));

  onOffApp.Start (Seconds (3.0));

  onOffApp.Stop (Seconds (4.0));*/

  /*ApplicationContainer pingApp = ping.Install (first.Get (0));

  pingApp.Start (Seconds (3.0));

  pingApp.Stop (Seconds (4.0));*/

  GlobalRouteManager::PopulateRoutingTables ();

  PointToPointHelper::EnablePcapAll ("netfilter");

  Simulator::Run ();

  Simulator::Destroy ();

  return 0;

}