src/surf/ns3/ns3_simulator.cc

   1 /* Copyright (c) 2007, 2008, 2009, 2010. The SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the license (GNU LGPL) which comes with this package. */
   6
   7 #include "surf/ns3/ns3_simulator.h"
   8 #include "xbt/dict.h"
   9 #include "xbt/log.h"
  10 #include "xbt/sysdep.h"
  11
  12 using namespace ns3;
  13 using namespace std;
  14
  15 xbt_dict_t dict_socket = NULL;
  16
  17 NS3Sim SimulatorNS3;
  18
  19 static void receive_callback(Ptr<Socket> localSocket);
  20 static void send_callback(Ptr<Socket> localSocket, uint32_t txSpace);
  21 static void datasent_callback(Ptr<Socket> localSocket, uint32_t dataSent);
  22 static void StartFlow(Ptr<Socket> sock,
  23     const char *to,
  24     uint16_t port_number);
  25
  26 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(simulator_ns3, surf,
  27                                 "Logging specific to the SURF network NS3 module");
  28
  29 // Constructor.
  30 NS3Sim::NS3Sim(){
  31 }
  32 //Destructor.
  33 NS3Sim::~NS3Sim(){
  34 }
  35
  36 /*
  37  * This function create a flow from src to dst
  38  *
  39  * Parameters
  40  *              src: node source
  41  *              dst: node destination
  42  *              port_number: The port number to use
  43  *              start: the time the communication start
  44  *              addr:  ip address
  45  *              totalBytes: number of bytes to transmit
  46  */
  47 void NS3Sim::create_flow_NS3(
  48                 Ptr<Node> src,
  49                 Ptr<Node> dst,
  50                 uint16_t port_number,
  51                 double start,
  52                 const char *addr,
  53                 uint32_t totalBytes,
  54                 void * action)
  55 {
  56         if(!dict_socket) dict_socket = xbt_dict_new();
  57
  58         PacketSinkHelper sink ("ns3::TcpSocketFactory",
  59                                                         InetSocketAddress (Ipv4Address::GetAny(),
  60                                                         port_number));
  61         sink.Install (dst);
  62         Ptr<Socket> sock = Socket::CreateSocket (src,
  63                                                         TcpSocketFactory::GetTypeId());
  64
  65         MySocket *mysocket = new MySocket();
  66         mysocket->totalBytes = totalBytes;
  67         mysocket->remaining = totalBytes;
  68         mysocket->bufferedBytes = 0;
  69         mysocket->sentBytes = 0;
  70         mysocket->finished = 0;
  71         mysocket->action = action;
  72         xbt_dict_set(dict_socket,(const char*)&sock, mysocket,NULL);
  73         sock->Bind(InetSocketAddress(port_number));
  74         Simulator::Schedule (Seconds(0.0),&StartFlow, sock, addr, port_number);
  75 }
  76
  77 void* NS3Sim::get_action_from_socket(void *socket){
  78         return ((MySocket *)socket)->action;
  79 }
  80
  81 char NS3Sim::get_finished(void *socket){
  82         return ((MySocket *)socket)->finished;
  83 }
  84
  85 double NS3Sim::get_remains_from_socket(void *socket){
  86         return ((MySocket *)socket)->remaining;
  87 }
  88
  89 double NS3Sim::get_sent_from_socket(void *socket){
  90   return ((MySocket *)socket)->sentBytes;
  91 }
  92
  93 void NS3Sim::simulator_stop(double min){
  94         if(min > 0.0)
  95                 Simulator::Stop(Seconds(min));
  96         else
  97                 Simulator::Stop();
  98 }
  99
 100 void NS3Sim::simulator_start(void){
 101         XBT_DEBUG("Start simulator");
 102         Simulator::Run ();
 103 }
 104
 105 static void receive_callback(Ptr<Socket> localSocket){
 106   MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
 107
 108   if (mysocket->finished == 0){
 109     mysocket->finished = 1;
 110 //    cout << "[" << Simulator::Now ().GetSeconds() << "] " << "recv_cb of F[" << mysocket->totalBytes << "] " << endl;
 111     XBT_DEBUG("Stop simulator at %f seconds", Simulator::Now().GetSeconds());
 112     Simulator::Stop(Seconds(0.0));
 113     Simulator::Run();
 114   }
 115 }
 116
 117 static void send_callback(Ptr<Socket> localSocket, uint32_t txSpace){
 118         MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
 119
 120         if (mysocket->remaining == 0){
 121           //all data was already buffered (and socket was already closed), just return
 122           return;
 123         }
 124
 125         uint32_t packetSize = 1024;
 126         uint32_t toWrite = min (mysocket->remaining, packetSize);
 127         uint8_t *data = (uint8_t*)malloc(sizeof(uint8_t)*toWrite);
 128         int amountSent = localSocket->Send (&data[0], toWrite, 0);
 129         free (data);
 130         if (amountSent > 0){
 131           mysocket->bufferedBytes += amountSent;
 132           mysocket->remaining -= amountSent;
 133         }
 134 //      cout << "[" << Simulator::Now ().GetSeconds() << "] " << "send_cb of F[" << mysocket->totalBytes << "] ("<<  mysocket->remaining << " / " << mysocket->totalBytes << ") " << amountSent << " buffered." << endl;
 135
 136   if (mysocket->remaining == 0){
 137     //everything was buffered to send, tell NS3 to close the socket
 138     localSocket->Close();
 139   }
 140         return;
 141 }
 142
 143 static void datasent_callback(Ptr<Socket> localSocket, uint32_t dataSent){
 144   MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
 145   mysocket->sentBytes += dataSent;
 146 //  cout << "[" << Simulator::Now ().GetSeconds() << "] " << "datasent_cb of F[" << mysocket->totalBytes << "] " << dataSent << " sent." << endl;
 147 }
 148
 149 static void normalClose_callback(Ptr<Socket> localSocket){
 150   MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
 151 //  cout << "[" << Simulator::Now ().GetSeconds() << "] " << "normalClose_cb of F[" << mysocket->totalBytes << "]" << endl;
 152   receive_callback (localSocket);
 153 }
 154
 155 static void errorClose_callback(Ptr<Socket> localSocket){
 156   MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
 157 //  cout << "[" << Simulator::Now ().GetSeconds() << "] " << "errorClose_cb of F[" << mysocket->totalBytes << "]" << endl;
 158   xbt_die("NS3: a socket was closed anormally");
 159 }
 160
 161 static void succeededConnect_callback(Ptr<Socket> localSocket){
 162   MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
 163 //  cout << "[" << Simulator::Now ().GetSeconds() << "] " << "succeededConnect_cb of F[" << mysocket->totalBytes << "]" << endl;
 164 }
 165
 166 static void failedConnect_callback(Ptr<Socket> localSocket){
 167   MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
 168 //  cout << "[" << Simulator::Now ().GetSeconds() << "] " << "failedConnect_cb of F[" << mysocket->totalBytes << "]" << endl;
 169   xbt_die("NS3: a socket failed to connect");
 170 }
 171
 172 static void StartFlow(Ptr<Socket> sock,
 173     const char *to,
 174     uint16_t port_number)
 175 {
 176   InetSocketAddress serverAddr (to, port_number);
 177
 178   sock->Connect(serverAddr);
 179   sock->SetSendCallback (MakeCallback (&send_callback));
 180   sock->SetRecvCallback (MakeCallback (&receive_callback));
 181   sock->SetDataSentCallback (MakeCallback (&datasent_callback));
 182   sock->SetConnectCallback (MakeCallback (&succeededConnect_callback), MakeCallback (&failedConnect_callback));
 183   sock->SetCloseCallbacks (MakeCallback (&normalClose_callback), MakeCallback (&errorClose_callback));
 184
 185   MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&sock);
 186 //  cout << "[" <<  Simulator::Now().GetSeconds() << "] Starting flow to " << to << " using port " << port_number << endl;
 187 }