#include "surf/ns3/ns3_simulator.h"
#include "xbt/dict.h"
#include "xbt/log.h"
+#include "xbt/sysdep.h"
using namespace ns3;
using namespace std;
-static const uint32_t writeSize = 1024; // limit the amout of data to write
-uint8_t data[writeSize];
xbt_dict_t dict_socket = NULL;
NS3Sim SimulatorNS3;
static void receive_callback(Ptr<Socket> localSocket);
static void send_callback(Ptr<Socket> localSocket, uint32_t txSpace);
+static void datasent_callback(Ptr<Socket> localSocket, uint32_t dataSent);
static void StartFlow(Ptr<Socket> sock,
const char *to,
uint16_t port_number);
* port_number: The port number to use
* start: the time the communication start
* addr: ip address
- * TotalBytes: number of bytes to transmit
+ * totalBytes: number of bytes to transmit
*/
void NS3Sim::create_flow_NS3(
Ptr<Node> src,
uint16_t port_number,
double start,
const char *addr,
- uint32_t TotalBytes,
+ uint32_t totalBytes,
void * action)
{
if(!dict_socket) dict_socket = xbt_dict_new();
- PacketSinkHelper sink ("ns3::TcpSocketFactory", InetSocketAddress (Ipv4Address::GetAny(), port_number));
+
+ PacketSinkHelper sink ("ns3::TcpSocketFactory",
+ InetSocketAddress (Ipv4Address::GetAny(),
+ port_number));
sink.Install (dst);
- Ptr<Socket> sock = Socket::CreateSocket (src, TypeId::LookupByName ("ns3::TcpSocketFactory"));
+ Ptr<Socket> sock = Socket::CreateSocket (src,
+ TcpSocketFactory::GetTypeId());
+
MySocket *mysocket = new MySocket();
- mysocket->TotalBytes = TotalBytes;
- mysocket->remaining = TotalBytes;
+ mysocket->totalBytes = totalBytes;
+ mysocket->remaining = totalBytes;
+ mysocket->bufferedBytes = 0;
mysocket->sentBytes = 0;
mysocket->finished = 0;
mysocket->action = action;
return ((MySocket *)socket)->remaining;
}
+double NS3Sim::get_sent_from_socket(void *socket){
+ return ((MySocket *)socket)->sentBytes;
+}
+
void NS3Sim::simulator_stop(double min){
if(min > 0.0)
Simulator::Stop(Seconds(min));
}
static void receive_callback(Ptr<Socket> localSocket){
- Address addr;
- localSocket->GetSockName (addr);
- InetSocketAddress iaddr = InetSocketAddress::ConvertFrom (addr);
MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
- mysocket->finished = 1;
- //cout << "[" << Simulator::Now ().GetSeconds() << "] " << "Received [" << mysocket->TotalBytes << "bytes], from: " << iaddr.GetIpv4 () << " port: " << iaddr.GetPort () << endl;
- std::stringstream sstream;
- sstream << Simulator::Now ().GetSeconds();
- std::string s = sstream.str();
- size_t size = s.size() + 1;
- char * time_sec = new char[ size ];
- strncpy( time_sec, s.c_str(), size );
- XBT_DEBUG("Stop simulator at %s seconds",time_sec);
- Simulator::Stop();
+ if (mysocket->finished == 0){
+ mysocket->finished = 1;
+// cout << "[" << Simulator::Now ().GetSeconds() << "] " << "recv_cb of F[" << mysocket->totalBytes << "] " << endl;
+ XBT_DEBUG("Stop simulator at %f seconds", Simulator::Now().GetSeconds());
+ Simulator::Stop(Seconds(0.0));
+ Simulator::Run();
+ }
}
static void send_callback(Ptr<Socket> localSocket, uint32_t txSpace){
-
- Address addr;
- localSocket->GetSockName (addr);
- InetSocketAddress iaddr = InetSocketAddress::ConvertFrom (addr);
MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
- uint32_t totalBytes = mysocket->TotalBytes;
- while ((mysocket->sentBytes) < totalBytes && localSocket->GetTxAvailable () > 0){
- uint32_t toWrite = min ((mysocket->remaining), writeSize);
- toWrite = min (toWrite, localSocket->GetTxAvailable ());
- int amountSent = localSocket->Send (&data[0], toWrite, 0);
-
-// cout << Simulator::Now () << " AmountSend: " << amountSent << " Towrite: " << toWrite << endl;
- if(amountSent < 0)
- return;
- (mysocket->sentBytes) += amountSent;
- (mysocket->remaining) -= amountSent;
- //cout << "[" << Simulator::Now ().GetSeconds() << "] " << "Send one packet, remaining "<< mysocket->remaining << " bytes!" << endl;
- }
- if ((mysocket->sentBytes) >= totalBytes){
- localSocket->Close();
+
+ if (mysocket->remaining == 0){
+ //all data was already buffered (and socket was already closed), just return
+ return;
}
+ uint32_t toWrite = min (mysocket->remaining, txSpace);
+ uint8_t *data = (uint8_t*)malloc(sizeof(uint8_t)*toWrite);
+ int amountSent = localSocket->Send (&data[0], toWrite, 0);
+ free (data);
+ if (amountSent > 0){
+ mysocket->bufferedBytes += amountSent;
+ mysocket->remaining -= amountSent;
+ }
+// cout << "[" << Simulator::Now ().GetSeconds() << "] " << "send_cb of F[" << mysocket->totalBytes << "] ("<< mysocket->remaining << " / " << mysocket->totalBytes << ") " << amountSent << " buffered." << endl;
+
+ if (mysocket->remaining == 0){
+ //everything was buffered to send, tell NS3 to close the socket
+ localSocket->Close();
+ }
+ return;
+}
+
+static void datasent_callback(Ptr<Socket> localSocket, uint32_t dataSent){
+ MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
+ mysocket->sentBytes += dataSent;
+// cout << "[" << Simulator::Now ().GetSeconds() << "] " << "datasent_cb of F[" << mysocket->totalBytes << "] " << dataSent << " sent." << endl;
+}
+
+static void normalClose_callback(Ptr<Socket> localSocket){
+ MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
+// cout << "[" << Simulator::Now ().GetSeconds() << "] " << "normalClose_cb of F[" << mysocket->totalBytes << "]" << endl;
+ receive_callback (localSocket);
+}
+
+static void errorClose_callback(Ptr<Socket> localSocket){
+ MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
+// cout << "[" << Simulator::Now ().GetSeconds() << "] " << "errorClose_cb of F[" << mysocket->totalBytes << "]" << endl;
+ xbt_die("NS3: a socket was closed anormally");
+}
+
+static void succeededConnect_callback(Ptr<Socket> localSocket){
+ MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
+// cout << "[" << Simulator::Now ().GetSeconds() << "] " << "succeededConnect_cb of F[" << mysocket->totalBytes << "]" << endl;
+}
+
+static void failedConnect_callback(Ptr<Socket> localSocket){
+ MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&localSocket);
+// cout << "[" << Simulator::Now ().GetSeconds() << "] " << "failedConnect_cb of F[" << mysocket->totalBytes << "]" << endl;
+ xbt_die("NS3: a socket failed to connect");
}
static void StartFlow(Ptr<Socket> sock,
{
InetSocketAddress serverAddr (to, port_number);
- //cout << "[" << Simulator::Now().GetSeconds() << "] Starting flow to " << to << " using port " << port_number << endl;
-
sock->Connect(serverAddr);
sock->SetSendCallback (MakeCallback (&send_callback));
sock->SetRecvCallback (MakeCallback (&receive_callback));
+ sock->SetDataSentCallback (MakeCallback (&datasent_callback));
+ sock->SetConnectCallback (MakeCallback (&succeededConnect_callback), MakeCallback (&failedConnect_callback));
+ sock->SetCloseCallbacks (MakeCallback (&normalClose_callback), MakeCallback (&errorClose_callback));
+
+ MySocket* mysocket = (MySocket*)xbt_dict_get_or_null(dict_socket,(char*)&sock);
+// cout << "[" << Simulator::Now().GetSeconds() << "] Starting flow to " << to << " using port " << port_number << endl;
}
