Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
7c62e05cc8021c69cb4b9c5d18a518b0fc662f76
[simgrid.git] / 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_start(double min){
94   if(min > 0.0)
95     Simulator::Stop(Seconds(min));
96   XBT_DEBUG("Start simulator");
97   Simulator::Run ();
98 }
99
100 static void receive_callback(Ptr<Socket> localSocket){
101   MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
102
103   if (mysocket->finished == 0){
104     mysocket->finished = 1;
105     XBT_DEBUG("recv_cb of F[%p, %p, %d]", mysocket, mysocket->action, mysocket->totalBytes);
106     XBT_DEBUG("Stop simulator at %f seconds", Simulator::Now().GetSeconds());
107     Simulator::Stop(Seconds(0.0));
108     Simulator::Run();
109   }
110 }
111
112 static void send_callback(Ptr<Socket> localSocket, uint32_t txSpace){
113         uint8_t *data = (uint8_t*)malloc(sizeof(uint8_t)*txSpace);
114         MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
115         if (mysocket->remaining == 0){
116                   //all data was already buffered (and socket was already closed), just return
117                   return;
118         }
119         while (mysocket->sentBytes < mysocket->totalBytes
120                         && localSocket->GetTxAvailable () > 0)
121         {
122       uint32_t toWrite = min ((mysocket->remaining), txSpace);
123       toWrite = min (toWrite, localSocket->GetTxAvailable ());
124       int amountSent = localSocket->Send (&data[0], toWrite, 0);
125
126       if(amountSent < 0)
127           return;
128           (mysocket->sentBytes) += amountSent;
129           (mysocket->remaining) -= amountSent;
130       XBT_DEBUG("send_cb of F[%p, %p, %d] (%d/%d) %d buffered", mysocket, mysocket->action, mysocket->totalBytes, mysocket->remaining, mysocket->totalBytes, amountSent);
131
132     }
133         if ((mysocket->sentBytes) >= mysocket->totalBytes){
134                 localSocket->Close();
135         }
136 }
137
138 static void datasent_callback(Ptr<Socket> localSocket, uint32_t dataSent){
139   MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
140   mysocket->sentBytes += dataSent;
141   XBT_DEBUG("datasent_cb of F[%p, %p, %d] %d sent", mysocket, mysocket->action, mysocket->totalBytes, dataSent);
142 }
143
144 static void normalClose_callback(Ptr<Socket> localSocket){
145   MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
146   XBT_DEBUG("normalClose_cb of F[%p, %p, %d]", mysocket, mysocket->action, mysocket->totalBytes);
147   receive_callback (localSocket);
148 }
149
150 static void errorClose_callback(Ptr<Socket> localSocket){
151   MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
152   XBT_DEBUG("errorClose_cb of F[%p, %p, %d]", mysocket, mysocket->action, mysocket->totalBytes);
153   xbt_die("NS3: a socket was closed anormally");
154 }
155
156 static void succeededConnect_callback(Ptr<Socket> localSocket){
157   MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
158   XBT_DEBUG("succeededConnect_cb of F[%p, %p, %d]", mysocket, mysocket->action, mysocket->totalBytes);
159 }
160
161 static void failedConnect_callback(Ptr<Socket> localSocket){
162   MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
163   XBT_DEBUG("failedConnect_cb of F[%p, %p, %d]", mysocket, mysocket->action, mysocket->totalBytes);
164   xbt_die("NS3: a socket failed to connect");
165 }
166
167 static void StartFlow(Ptr<Socket> sock,
168     const char *to,
169     uint16_t port_number)
170 {
171   InetSocketAddress serverAddr (to, port_number);
172
173   sock->Connect(serverAddr);
174   sock->SetSendCallback (MakeCallback (&send_callback));
175   sock->SetRecvCallback (MakeCallback (&receive_callback));
176   sock->SetDataSentCallback (MakeCallback (&datasent_callback));
177   sock->SetConnectCallback (MakeCallback (&succeededConnect_callback), MakeCallback (&failedConnect_callback));
178   sock->SetCloseCallbacks (MakeCallback (&normalClose_callback), MakeCallback (&errorClose_callback));
179
180   MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&sock);
181   XBT_DEBUG("startFlow_cb of F[%p, %p, %d] dest=%s port=%d", mysocket, mysocket->action, mysocket->totalBytes, to, port_number);
182 }