Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Indent include and src using this command:
[simgrid.git] / src / amok / Bandwidth / saturate.c
index 84d65e4..c2b74ec 100644 (file)
@@ -1,50 +1,56 @@
-/* $Id$ */
-
 /* amok_saturate - Link saturating facilities (for ALNeM's BW testing)      */
 
-/* Copyright (c) 2003, 2004 Martin Quinson. All rights reserved.            */
+/* Copyright (c) 2004, 2005, 2006, 2007, 2009, 2010. The SimGrid Team.
+ * All rights reserved.                                                     */
 
 /* This program is free software; you can redistribute it and/or modify it
  * under the terms of the license (GNU LGPL) which comes with this package. */
 
 #include "amok/Bandwidth/bandwidth_private.h"
-#include "gras/Msg/msg_private.h" /* FIXME: This mucks with contextes to answer RPC directly */
+#include "gras/Msg/msg_private.h"       /* FIXME: This mucks with contextes to answer RPC directly */
+#include "xbt/ex.h"
 
-XBT_LOG_NEW_DEFAULT_SUBCATEGORY(amok_bw_sat,amok_bw,"Everything concerning the SATuration part of the amok_bw module");
+XBT_LOG_NEW_DEFAULT_SUBCATEGORY(amok_bw_sat, amok_bw,
+                                "Everything concerning the SATuration part of the amok_bw module");
 
 static int amok_bw_cb_sat_start(gras_msg_cb_ctx_t ctx, void *payload);
 static int amok_bw_cb_sat_begin(gras_msg_cb_ctx_t ctx, void *payload);
 
 
-void amok_bw_sat_init(void) {
-  gras_datadesc_type_t bw_res_desc=gras_datadesc_by_name("bw_res_t");
+void amok_bw_sat_init(void)
+{
+  gras_datadesc_type_t bw_res_desc = gras_datadesc_by_name("bw_res_t");
   gras_datadesc_type_t sat_request_desc;
-  /* Build the saturation datatype descriptions */ 
-  
+  /* Build the saturation datatype descriptions */
+
   sat_request_desc = gras_datadesc_struct("s_sat_request_desc_t");
-  gras_datadesc_struct_append(sat_request_desc,"host",gras_datadesc_by_name("s_xbt_host_t"));
-  gras_datadesc_struct_append(sat_request_desc,"msg_size",gras_datadesc_by_name("unsigned int"));
-  gras_datadesc_struct_append(sat_request_desc,"duration",gras_datadesc_by_name("unsigned int"));
+  gras_datadesc_struct_append(sat_request_desc, "peer",
+                              gras_datadesc_by_name("s_xbt_peer_t"));
+  gras_datadesc_struct_append(sat_request_desc, "msg_size",
+                              gras_datadesc_by_name("unsigned int"));
+  gras_datadesc_struct_append(sat_request_desc, "duration",
+                              gras_datadesc_by_name("unsigned int"));
   gras_datadesc_struct_close(sat_request_desc);
-  sat_request_desc = gras_datadesc_ref("sat_request_t",sat_request_desc);
-  
+  sat_request_desc = gras_datadesc_ref("sat_request_t", sat_request_desc);
+
   /* Register the saturation messages */
   gras_msgtype_declare_rpc("amok_bw_sat start", sat_request_desc, NULL);
-  gras_msgtype_declare_rpc("amok_bw_sat begin", sat_request_desc, sat_request_desc);
-  gras_msgtype_declare_rpc("amok_bw_sat stop",  NULL,             bw_res_desc);
+  gras_msgtype_declare_rpc("amok_bw_sat begin", sat_request_desc,
+                           sat_request_desc);
+  gras_msgtype_declare_rpc("amok_bw_sat stop", NULL, bw_res_desc);
 
 }
-void amok_bw_sat_join(void) {
-  gras_cb_register(gras_msgtype_by_name("amok_bw_sat start"),
-                  &amok_bw_cb_sat_start);
-  gras_cb_register(gras_msgtype_by_name("amok_bw_sat begin"),
-                  &amok_bw_cb_sat_begin);
+
+void amok_bw_sat_join(void)
+{
+  gras_cb_register("amok_bw_sat start", &amok_bw_cb_sat_start);
+  gras_cb_register("amok_bw_sat begin", &amok_bw_cb_sat_begin);
 }
-void amok_bw_sat_leave(void) {
-  gras_cb_unregister(gras_msgtype_by_name("amok_bw_sat start"),
-                    &amok_bw_cb_sat_start);
-  gras_cb_unregister(gras_msgtype_by_name("amok_bw_sat begin"),
-                    &amok_bw_cb_sat_begin);
+
+void amok_bw_sat_leave(void)
+{
+  gras_cb_unregister("amok_bw_sat start", &amok_bw_cb_sat_start);
+  gras_cb_unregister("amok_bw_sat begin", &amok_bw_cb_sat_begin);
 }
 
 /* ***************************************************************************
@@ -54,11 +60,11 @@ void amok_bw_sat_leave(void) {
 /**
  * @brief Ask 'from_name:from_port' to stop saturating going to to_name:to_name.
  *
- * @param from_name: Name of the host we are asking to do a experiment with (to_name:to_port)
+ * @param from_name: Name of the peer we are asking to do a experiment with (to_name:to_port)
  * @param from_port: port on which the process we are asking for an experiment is listening
  * (for message, do not give a raw socket here. The needed raw socket will be negociated 
  * between the peers)
- * @param to_name: Name of the host with which we should conduct the experiment
+ * @param to_name: Name of the peer with which we should conduct the experiment
  * @param to_port: port on which the peer process is listening for message
  * @param msg_size: Size of each message sent.
  * @param duration: How long in maximum should be the saturation.
@@ -66,51 +72,55 @@ void amok_bw_sat_leave(void) {
  * Ask the process 'from_name:from_port' to start to saturate the link between itself
  * and to_name:to_name.
  */
-void amok_bw_saturate_start(const char* from_name,unsigned int from_port,
-                           const char* to_name,unsigned int to_port,
-                           unsigned int msg_size, double duration) {
+void amok_bw_saturate_start(const char *from_name, unsigned int from_port,
+                            const char *to_name, unsigned int to_port,
+                            unsigned int msg_size, double duration)
+{
   gras_socket_t sock;
+  sat_request_t request = xbt_new(s_sat_request_t, 1);
 
-  sat_request_t request = xbt_new(s_sat_request_t,1);
-
-  sock = gras_socket_client(from_name,from_port);
+  VERB4("Start from_name %s:%d -> to_name %s:%d",
+        from_name, from_port, to_name, to_port);
+  sock = gras_socket_client(from_name, from_port);
 
-  request->host.name = (char*)to_name;
-  request->host.port = to_port;
-  
-  request->duration=duration;
-  request->msg_size=msg_size;
+  request->peer.name = (char *) to_name;
+  request->peer.port = to_port;
 
+  request->duration = duration;
+  request->msg_size = msg_size;
 
-  gras_msg_rpccall(sock,60,gras_msgtype_by_name("amok_bw_sat start"),&request, NULL);
+  gras_msg_rpccall(sock, 60, "amok_bw_sat start", &request, NULL);
 
   free(request);
   gras_socket_close(sock);
 }
 
 /* Asked to begin a saturation */
-static int amok_bw_cb_sat_start(gras_msg_cb_ctx_t ctx, void *payload){
-  sat_request_t request = *(sat_request_t*)payload;
+static int amok_bw_cb_sat_start(gras_msg_cb_ctx_t ctx, void *payload)
+{
+  sat_request_t request = *(sat_request_t *) payload;
   gras_socket_t expeditor = gras_msg_cb_ctx_from(ctx);
 
   VERB4("Asked by %s:%d to start a saturation to %s:%d",
-       gras_socket_peer_name(expeditor),gras_socket_peer_port(expeditor),
-       request->host.name,request->host.port);
-       
-  gras_msg_rpcreturn(60,ctx, NULL);
-  amok_bw_saturate_begin(request->host.name,request->host.port,
-                        request->msg_size, request->duration,
-                        NULL,NULL);
-  free(request->host.name);
+        gras_socket_peer_name(expeditor), gras_socket_peer_port(expeditor),
+        request->peer.name, request->peer.port);
+
+  gras_msg_rpcreturn(60, ctx, NULL);
+
+  amok_bw_saturate_begin(request->peer.name, request->peer.port,
+                         request->msg_size, request->duration, NULL, NULL);
+
+  free(request->peer.name);
+
   free(request);
-  return 1;
+  return 0;
 }
 
 /**
  * @brief Start saturating between the current process and the designated peer
  *
  * Note that the only way to break this function before the end of the timeout
- * is to have a remote host calling amok_bw_saturate_stop to this process.
+ * is to have a remote peer calling amok_bw_saturate_stop to this process.
  *
  * If duration=0, the experiment will never timeout (you then have to manually
  * stop it).
@@ -118,10 +128,11 @@ static int amok_bw_cb_sat_start(gras_msg_cb_ctx_t ctx, void *payload){
  * If msg_size=0, the size will be automatically computed to make sure that
  * each of the messages occupy the connexion one second
  */
-void amok_bw_saturate_begin(const char* to_name,unsigned int to_port,
-                           unsigned int msg_size, double duration,
-                           /*out*/ double *elapsed_res, double *bw_res) {
+void amok_bw_saturate_begin(const char *to_name, unsigned int to_port,
+                            unsigned int msg_size, double duration,
+                            /*out */ double *elapsed_res, double *bw_res)
+{
+
   xbt_ex_t e;
 
   gras_socket_t peer_cmd = gras_socket_client(to_name, to_port);
@@ -131,79 +142,85 @@ void amok_bw_saturate_begin(const char* to_name,unsigned int to_port,
 
   s_gras_msg_t msg_got;
 
-  unsigned int packet_sent=0;
-  double start,elapsed=-1; /* timer */
+  unsigned int packet_sent = 0;
+  double start, elapsed = -1;   /* timer */
   double bw;
 
-  volatile int saturate_further; /* boolean in the main loop */ 
+  volatile int saturate_further;        /* boolean in the main loop */
 
   /* Negociate the saturation with the peer */
-  sat_request_t request = xbt_new(s_sat_request_t,1);
+  sat_request_t request = xbt_new(s_sat_request_t, 1);
 
-  DEBUG2("Begin to saturate to %s:%d",to_name,to_port);
-  memset(&msg_got,0,sizeof(msg_got));
+  DEBUG2("Begin to saturate to %s:%d", to_name, to_port);
+  memset(&msg_got, 0, sizeof(msg_got));
 
   request->msg_size = msg_size;
   request->duration = duration;
-  request->host.name = NULL;
-  request->host.port = 0;
+  request->peer.name = NULL;
+  request->peer.port = 0;
+
 
   /* Size autodetection on need */
   if (!msg_size) {
     double bw;
     double sec;
-    amok_bw_test(peer_cmd,
-                0,512*1024, 512*1024, /* 512k as first guess */
-                1, /* at least one sec */
-                &sec, &bw);
-    msg_size = request->msg_size = (int)bw;
-    DEBUG1("Saturate with packets of %d bytes",request->msg_size);
+    amok_bw_test(peer_cmd, 0,   /* check buffsize yourself */
+                 512 * 1024,    /* 512k as first guess */
+                 1,             /* One packet only */
+                 1,             /* at least one sec */
+                 &sec, &bw);
+    msg_size = request->msg_size = (int) bw;
+    DEBUG1("Saturate with packets of %d bytes", request->msg_size);
   }
-   
+
   /* Launch the saturation */
-  ctx = gras_msg_rpc_async_call(peer_cmd, 60, 
-                               gras_msgtype_by_name("amok_bw_sat begin"),
-                                                 &request);
+
+  ctx =
+      gras_msg_rpc_async_call(peer_cmd, 60, "amok_bw_sat begin", &request);
   free(request);
-  gras_msg_rpc_async_wait(ctx,&request);
-  meas=gras_socket_client_ext( to_name, request->host.port,
-                              0 /*bufsize: auto*/,
-                              1 /*meas: true*/);
+  gras_msg_rpc_async_wait(ctx, &request);
+  meas = gras_socket_client_ext(to_name, request->peer.port,
+                                0 /*bufsize: auto */ ,
+                                1 /*meas: true */ );
+
   free(request);
 
   gras_socket_close(peer_cmd);
-  INFO2("Saturation(%s->%s) started",gras_os_myname(),to_name);
+  INFO4("Saturation(%s:%d->%s:%d) started", gras_os_myname(),
+        gras_os_myport(), to_name, to_port);
 
   /* Start experiment */
-  start=gras_os_time();
+  start = gras_os_time();
 
   do {
     /* do send it */
-    gras_socket_meas_send(meas,120,msg_size,msg_size);
+    gras_socket_meas_send(meas, 120, msg_size, 1);
     packet_sent++;
 
     /* Check whether someone asked us to stop saturation */
     saturate_further = 0;
     TRY {
-      gras_msg_wait_ext(0/*no wait*/,gras_msgtype_by_name("amok_bw_sat stop"),
-                       NULL /* accept any sender */,
-                       NULL, NULL, /* No specific filter */
-                       &msg_got);
-    } CATCH(e) {
+      gras_msg_wait_ext(0 /*no wait */ , "amok_bw_sat stop",
+                        NULL /* accept any sender */ ,
+                        NULL, NULL,     /* No specific filter */
+                        &msg_got);
+    }
+    CATCH(e) {
       if (e.category == timeout_error) {
-       saturate_further=1;
-       memset(&msg_got,0,sizeof(msg_got)); /* may be overprotectiv here */
+        saturate_further = 1;
+        memset(&msg_got, 0, sizeof(msg_got));   /* may be overprotectiv here */
       }
       xbt_ex_free(e);
     }
 
     /* Check whether the experiment has to be terminated by now */
-    elapsed=gras_os_time()-start;
-    DEBUG3("elapsed %f duration %f (msg_size=%d)",elapsed, duration,msg_size);
+    elapsed = gras_os_time() - start;
+    DEBUG3("elapsed %f duration %f (msg_size=%d)", elapsed, duration,
+           msg_size);
 
-  } while (saturate_further && (duration==0 || elapsed < duration));
+  } while (saturate_further && (duration == 0 || elapsed < duration));
 
-  bw = ((double)(packet_sent*msg_size)) / elapsed;
+  bw = ((double) (packet_sent * msg_size)) / elapsed;
 
   if (elapsed_res)
     *elapsed_res = elapsed;
@@ -212,110 +229,122 @@ void amok_bw_saturate_begin(const char* to_name,unsigned int to_port,
 
   /* If someone stopped us, inform him about the achieved bandwidth */
   if (msg_got.expe) {
-    bw_res_t answer = xbt_new(s_bw_res_t,1);
+    bw_res_t answer = xbt_new(s_bw_res_t, 1);
     s_gras_msg_cb_ctx_t ctx;
 
-    INFO3("Saturation from %s to %s stopped by %s",
-         gras_os_myname(),to_name, gras_socket_peer_name(msg_got.expe));
-    answer->timestamp=gras_os_time();
-    answer->sec=elapsed;
-    answer->bw=bw;
+    INFO6("Saturation from %s:%d to %s:%d stopped by %s:%d",
+          gras_os_myname(), gras_os_myport(), to_name, to_port,
+          gras_socket_peer_name(msg_got.expe),
+          gras_socket_peer_port(msg_got.expe));
+    answer->timestamp = gras_os_time();
+    answer->sec = elapsed;
+    answer->bw = bw;
 
     ctx.expeditor = msg_got.expe;
     ctx.ID = msg_got.ID;
     ctx.msgtype = msg_got.type;
 
-    gras_msg_rpcreturn(60,&ctx,&answer);
+    gras_msg_rpcreturn(60, &ctx, &answer);
     free(answer);
   } else {
-    INFO4("Saturation from %s to %s elapsed after %f sec (achieving %f kb/s)",
-         gras_os_myname(),to_name,elapsed,bw/1024.0);
+    INFO6
+        ("Saturation from %s:%d to %s:%d elapsed after %f sec (achieving %f kb/s)",
+         gras_os_myname(), gras_os_myport(), to_name, to_port, elapsed,
+         bw / 1024.0);
   }
-  
+
   gras_socket_close(meas);
 }
 
 /* Sender will saturate link to us */
-static int amok_bw_cb_sat_begin(gras_msg_cb_ctx_t ctx, void *payload){
-  sat_request_t request=*(sat_request_t*)payload;
-  sat_request_t answer=xbt_new0(s_sat_request_t,1);
+static int amok_bw_cb_sat_begin(gras_msg_cb_ctx_t ctx, void *payload)
+{
+  sat_request_t request = *(sat_request_t *) payload;
+  sat_request_t answer = xbt_new0(s_sat_request_t, 1);
   volatile int saturate_further = 1;
   xbt_ex_t e;
-  gras_socket_t measMaster=NULL,meas=NULL;
-  gras_socket_t from=gras_msg_cb_ctx_from(ctx);
+  gras_socket_t measMaster = NULL, meas = NULL;
+  gras_socket_t from = gras_msg_cb_ctx_from(ctx);
 
-  int port=6000;
+  int port = 6000;
   while (port <= 10000 && measMaster == NULL) {
     TRY {
-      measMaster = gras_socket_server_ext(port,
-                                         0 /*bufsize: auto*/,
-                                         1 /*meas: true*/);
-    CATCH(e) {
+      measMaster = gras_socket_server_ext(port, 0 /*bufsize: auto */ ,
+                                          1 /*meas: true */ );
+    }
+    CATCH(e) {
       measMaster = NULL;
       if (port < 10000)
-       xbt_ex_free(e);
+        xbt_ex_free(e);
       else
-       RETHROW0("Error encountered while opening a measurement server socket: %s");
+        RETHROW0
+            ("Error encountered while opening a measurement server socket: %s");
     }
-    if (measMaster == NULL) 
-      port++; /* prepare for a new loop */
+    if (measMaster == NULL)
+      port++;                   /* prepare for a new loop */
   }
-  answer->host.port=port;
+  answer->peer.port = port;
 
   gras_msg_rpcreturn(60, ctx, &answer);
   free(answer);
 
-  gras_os_sleep(5); /* Wait for the accept */
+  gras_os_sleep(5);             /* Wait for the accept */
 
   TRY {
     meas = gras_socket_meas_accept(measMaster);
     DEBUG0("saturation handshake answered");
-  } CATCH(e) {
+  }
+  CATCH(e) {
     gras_socket_close(measMaster);
     RETHROW0("Error during saturation handshake: %s");
-  }  
+  }
 
   while (saturate_further) {
     TRY {
-      gras_socket_meas_recv(meas,5,request->msg_size,request->msg_size);
-    } CATCH(e) {
+      gras_socket_meas_recv(meas, 5, request->msg_size, 1);
+    }
+    CATCH(e) {
       saturate_further = 0;
       xbt_ex_free(e);
     }
   }
-  INFO3("Saturation comming from %s:%d stopped on %s",
-       gras_socket_peer_name(from),gras_socket_peer_port(from),
-       gras_os_myname());
+  INFO4("Saturation comming from %s:%d stopped on %s:%d",
+        gras_socket_peer_name(from), gras_socket_peer_port(from),
+        gras_os_myname(), gras_os_myport());
 
   gras_socket_close(meas);
-  if (gras_if_RL()) /* On SG, accepted=master */
-    gras_socket_close(measMaster); 
+  if (gras_if_RL())             /* On SG, accepted=master */
+    gras_socket_close(measMaster);
   free(request);
-  return 1;
+  return 0;
 }
 
 /**
  * @brief Ask 'from_name:from_port' to stop any saturation experiments
- * @param from_name: Name of the host we are asking to do a experiment with (to_name:to_port)
+ * @param from_name: Name of the peer we are asking to do a experiment with (to_name:to_port)
  * @param from_port: port on which the process we are asking for an experiment is listening
  * @param time: the duration of the experiment
  * @param bw: the achieved bandwidth
  *
  */
-void amok_bw_saturate_stop(const char* from_name,unsigned int from_port,
-                          /*out*/ double *time, double *bw) {
+void amok_bw_saturate_stop(const char *from_name, unsigned int from_port,
+                           /*out */ double *time, double *bw)
+{
   xbt_ex_t e;
-   
-  gras_socket_t sock = gras_socket_client(from_name,from_port);
+
+  gras_socket_t sock = gras_socket_client(from_name, from_port);
   bw_res_t answer;
-  VERB2("Ask %s:%d to stop the saturation",from_name,from_port);
-  TRY {        
-     gras_msg_rpccall(sock,60,gras_msgtype_by_name("amok_bw_sat stop"),NULL,&answer);
+  VERB2("Ask %s:%d to stop the saturation", from_name, from_port);
+  TRY {
+    gras_msg_rpccall(sock, 60, "amok_bw_sat stop", NULL, &answer);
   } CATCH(e) {
-     RETHROW2("Cannot ask %s:%d to stop saturation: %s",from_name, from_port);
+    RETHROW2("Cannot ask %s:%d to stop saturation: %s", from_name,
+             from_port);
   }
   gras_socket_close(sock);
-  if (time) *time=answer->sec;
-  if (bw)   *bw  =answer->bw;
+  if (time)
+    *time = answer->sec;
+  if (bw)
+    *bw = answer->bw;
   free(answer);
 }