src/surf/ns3/ns3_interface.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 "ns3_interface.h"
   8 #include "ns3_simulator.h"
   9 #include "xbt/lib.h"
  10 #include "xbt/log.h"
  11 #include "xbt/dynar.h"
  12
  13
  14 using namespace ns3;
  15
  16 extern xbt_lib_t host_lib;
  17 extern int NS3_HOST_LEVEL;              //host node for ns3
  18 extern xbt_dynar_t IPV4addr;
  19
  20 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(interface_ns3, surf,
  21                                 "Logging specific to the SURF network NS3 module");
  22
  23 InternetStackHelper stack;
  24 NodeContainer nodes;
  25 NodeContainer Cluster_nodes;
  26 Ipv4InterfaceContainer interfaces;
  27
  28 int number_of_nodes = 0;
  29 int number_of_clusters_nodes = 0;
  30 int number_of_links = 1;
  31 int number_of_networks = 1;
  32 int port_number = 1025; //Port number is limited from 1025 to 65 000
  33
  34 static NS3Sim* ns3_sim = 0;
  35
  36 void ns3_simulator(double min){
  37                         ns3_sim->simulator_start(min);
  38 }
  39
  40 void* ns3_get_socket_action(void *socket){
  41                 return ns3_sim->get_action_from_socket(socket);
  42 }
  43
  44 double ns3_get_socket_remains(void *socket){
  45                 return ns3_sim->get_remains_from_socket(socket);
  46 }
  47
  48 double ns3_get_socket_sent(void *socket){
  49   return ns3_sim->get_sent_from_socket(socket);
  50 }
  51
  52 char ns3_get_socket_is_finished(void *socket){
  53                 return ns3_sim->get_finished(socket);
  54 }
  55
  56 double ns3_time(){
  57         return Simulator::Now().GetSeconds();
  58 }
  59
  60 int ns3_create_flow(const char* a,const char *b,double start,u_int32_t TotalBytes,void * action)
  61 {
  62         ns3_nodes_t node1 = (ns3_nodes_t) xbt_lib_get_or_null(host_lib,a,NS3_HOST_LEVEL);
  63         ns3_nodes_t node2 = (ns3_nodes_t) xbt_lib_get_or_null(host_lib,b,NS3_HOST_LEVEL);
  64
  65         Ptr<Node> src_node = nodes.Get(node1->node_num);
  66         Ptr<Node> dst_node = nodes.Get(node2->node_num);
  67
  68         char* addr = (char*)xbt_dynar_get_as(IPV4addr,node2->node_num,char*);
  69
  70         XBT_DEBUG("ns3_create_flow %d Bytes from %d to %d with Interface %s",TotalBytes, node1->node_num, node2->node_num,addr);
  71         ns3_sim->create_flow_NS3(src_node,
  72                         dst_node,
  73                         port_number,
  74                         start,
  75                         addr,
  76                         TotalBytes,
  77                         action);
  78
  79         port_number++;
  80         if(port_number >= 65001 ) xbt_die("Too many connections! Port number is saturated.");
  81         return 0;
  82 }
  83
  84 // clean up
  85 int ns3_finalize(void){
  86         if (!ns3_sim) return -1;
  87         delete ns3_sim;
  88         ns3_sim = 0;
  89         return 0;
  90 }
  91
  92 // initialize the NS3 interface and environment
  93 int ns3_initialize(const char* TcpProtocol){
  94   xbt_assert(!ns3_sim, "ns3 already initialized");
  95   ns3_sim = new NS3Sim();
  96
  97 //  tcpModel are:
  98 //  "ns3::TcpNewReno"
  99 //  "ns3::TcpReno"
 100 //  "ns3::TcpTahoe"
 101
 102   Config::SetDefault ("ns3::TcpSocket::SegmentSize", UintegerValue (1024)); // 1024-byte packet for easier reading
 103   Config::SetDefault ("ns3::TcpSocket::DelAckCount", UintegerValue (1));
 104
 105 #ifdef _HAVE_NS3_RED
 106   XBT_DEBUG("Using RED version of ns3");
 107 #endif
 108   if(!strcmp(TcpProtocol,"default")){
 109           return 0;
 110   }
 111   if(!strcmp(TcpProtocol,"Reno")){
 112           XBT_INFO("Switching Tcp protocol to '%s'",TcpProtocol);
 113           Config::SetDefault ("ns3::TcpL4Protocol::SocketType", StringValue("ns3::TcpReno"));
 114           return 0;
 115   }
 116   if(!strcmp(TcpProtocol,"NewReno")){
 117           XBT_INFO("Switching Tcp protocol to '%s'",TcpProtocol);
 118           Config::SetDefault ("ns3::TcpL4Protocol::SocketType", StringValue("ns3::TcpNewReno"));
 119           return 0;
 120   }
 121   if(!strcmp(TcpProtocol,"Tahoe")){
 122           XBT_INFO("Switching Tcp protocol to '%s'",TcpProtocol);
 123           Config::SetDefault ("ns3::TcpL4Protocol::SocketType", StringValue("ns3::TcpTahoe"));
 124           return 0;
 125   }
 126
 127   XBT_ERROR("The ns3/TcpModel must be : NewReno or Reno or Tahoe");
 128 }
 129
 130 void * ns3_add_host(char * id)
 131 {
 132         ns3_nodes_t host  = xbt_new0(s_ns3_nodes_t,1);
 133         XBT_DEBUG("Interface ns3 add host[%d] '%s'",number_of_nodes,id);
 134         Ptr<Node> node =  CreateObject<Node> (0);
 135         stack.Install(node);
 136         nodes.Add(node);
 137         host->node_num = number_of_nodes;
 138         host->type = NS3_NETWORK_ELEMENT_HOST;
 139         host->data = GetPointer(node);
 140         number_of_nodes++;
 141         return host;
 142 }
 143
 144 void * ns3_add_host_cluster(char * id)
 145 {
 146         ns3_nodes_t host  = xbt_new0(s_ns3_nodes_t,1);
 147         XBT_DEBUG("Interface ns3 add host[%d] '%s'",number_of_nodes,id);
 148         Ptr<Node> node =  CreateObject<Node> (0);
 149         stack.Install(node);
 150         Cluster_nodes.Add(node);
 151         nodes.Add(node);
 152         host->node_num = number_of_nodes;
 153         host->type = NS3_NETWORK_ELEMENT_HOST;
 154         host->data = node;
 155         number_of_nodes++;
 156         return host;
 157 }
 158
 159 void * ns3_add_router(char * id)
 160 {
 161         ns3_nodes_t router  = xbt_new0(s_ns3_nodes_t,1);
 162         XBT_DEBUG("Interface ns3 add router[%d] '%s'",number_of_nodes,id);
 163         Ptr<Node> node =  CreateObject<Node> (0);
 164         stack.Install(node);
 165         nodes.Add(node);
 166         router->node_num = number_of_nodes;
 167         router->type = NS3_NETWORK_ELEMENT_ROUTER;
 168         router->data = node;
 169         number_of_nodes++;
 170         return router;
 171 }
 172
 173 void * ns3_add_cluster(char * bw,char * lat,char *id)
 174 {
 175
 176         XBT_DEBUG("cluster_id: %s",id);
 177         XBT_DEBUG("bw: %s lat: %s",bw,lat);
 178         XBT_DEBUG("Number of %s nodes: %d",id,Cluster_nodes.GetN() - number_of_clusters_nodes);
 179
 180         NodeContainer Nodes;
 181         int i;
 182
 183         for(i = number_of_clusters_nodes; i < Cluster_nodes.GetN() ; i++){
 184                 Nodes.Add(Cluster_nodes.Get(i));
 185                 XBT_DEBUG("Add node %d to cluster",i);
 186         }
 187         number_of_clusters_nodes = Cluster_nodes.GetN();
 188
 189         XBT_DEBUG("Add router %d to cluster",nodes.GetN()-Nodes.GetN()-1);
 190         Nodes.Add(nodes.Get(nodes.GetN()-Nodes.GetN()-1));
 191
 192         if(Nodes.GetN() > 65000)
 193                 xbt_die("Cluster with NS3 is limited to 65000 nodes");
 194         CsmaHelper csma;
 195         csma.SetChannelAttribute ("DataRate", StringValue (bw));
 196         csma.SetChannelAttribute ("Delay", StringValue (lat));
 197         NetDeviceContainer devices = csma.Install (Nodes);
 198         XBT_DEBUG("Create CSMA");
 199
 200         char * adr = bprintf("%d.%d.0.0",number_of_networks,number_of_links);
 201         XBT_DEBUG("Assign IP Addresses %s to CSMA.",adr);
 202         Ipv4AddressHelper ipv4;
 203         ipv4.SetBase (adr, "255.255.0.0");
 204         free(adr);
 205         interfaces.Add(ipv4.Assign (devices));
 206
 207         if(number_of_links == 255){
 208                 if(number_of_networks == 255)
 209                         xbt_die("Number of links and networks exceed 255*255");
 210                 number_of_links = 1;
 211                 number_of_networks++;
 212         }else{
 213                 number_of_links++;
 214         }
 215         XBT_DEBUG("Number of nodes in Cluster_nodes: %d",Cluster_nodes.GetN());
 216 }
 217
 218 void * ns3_add_AS(char * id)
 219 {
 220         XBT_DEBUG("Interface ns3 add AS '%s'",id);
 221         return NULL;
 222 }
 223
 224 static char* transformIpv4Address (Ipv4Address from){
 225         std::stringstream sstream;
 226                 sstream << from ;
 227                 std::string s = sstream.str();
 228                 return bprintf("%s",s.c_str());
 229 }
 230
 231 void * ns3_add_link(int src, e_ns3_network_element_type_t type_src,
 232                                         int dst, e_ns3_network_element_type_t type_dst,
 233                                         char * bw,char * lat)
 234 {
 235         if(number_of_links == 1 ) {
 236                 LogComponentEnable("UdpEchoClientApplication", LOG_LEVEL_INFO);
 237                 LogComponentEnable("UdpEchoServerApplication", LOG_LEVEL_INFO);
 238         }
 239
 240 #ifdef _HAVE_NS3_RED
 241         MyPointToPointHelper pointToPoint;
 242 #else
 243         PointToPointHelper pointToPoint;
 244 #endif
 245         NetDeviceContainer netA;
 246         Ipv4AddressHelper address;
 247
 248         Ptr<Node> a = nodes.Get(src);
 249         Ptr<Node> b = nodes.Get(dst);
 250
 251         XBT_DEBUG("\tAdd PTP from %d to %d bw:'%s' lat:'%s'",src,dst,bw,lat);
 252         pointToPoint.SetDeviceAttribute ("DataRate", StringValue (bw));
 253         pointToPoint.SetChannelAttribute ("Delay", StringValue (lat));
 254         //pointToPoint.EnablePcapAll("test_ns3_trace"); //DEBUG
 255
 256 #ifdef _HAVE_NS3_RED
 257         netA.Add(pointToPoint.Install (a, type_src, b, type_dst));
 258 #else
 259         netA.Add(pointToPoint.Install (a, b));
 260 #endif
 261
 262         char * adr = bprintf("%d.%d.0.0",number_of_networks,number_of_links);
 263         address.SetBase (adr, "255.255.0.0");
 264         XBT_DEBUG("\tInterface stack '%s'",adr);
 265         free(adr);
 266         interfaces.Add(address.Assign (netA));
 267
 268         char *tmp = transformIpv4Address(interfaces.GetAddress(interfaces.GetN()-2));
 269         xbt_dynar_set_as(IPV4addr,src,char*,tmp);
 270         XBT_DEBUG("Have write '%s' for Node '%d'",(char*)xbt_dynar_get_as(IPV4addr,src,char*),src);
 271
 272         tmp = transformIpv4Address(interfaces.GetAddress(interfaces.GetN()-1));
 273         xbt_dynar_set_as(IPV4addr,dst,char*,tmp);
 274         XBT_DEBUG("Have write '%s' for Node '%d'",(char*)xbt_dynar_get_as(IPV4addr,dst,char*),dst);
 275
 276         if(number_of_links == 255){
 277                 if(number_of_networks == 255)
 278                         xbt_die("Number of links and networks exceed 255*255");
 279                 number_of_links = 1;
 280                 number_of_networks++;
 281         }else{
 282                 number_of_links++;
 283         }
 284 }
 285
 286 void * ns3_end_platform(void)
 287 {
 288         XBT_DEBUG("InitializeRoutes");
 289         GlobalRouteManager::BuildGlobalRoutingDatabase();
 290         GlobalRouteManager::InitializeRoutes();
 291 }