Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
c0cf41b15f2ee20cd0073c4156fe3328508b6f90
[simgrid.git] / src / amok / Bandwidth / saturate.c
1 /* $Id$ */
2
3 /* amok_saturate - Link saturating facilities (for ALNeM's BW testing)      */
4
5 /* Copyright (c) 2003, 2004 Martin Quinson. All rights reserved.            */
6
7 /* This program is free software; you can redistribute it and/or modify it
8  * under the terms of the license (GNU LGPL) which comes with this package. */
9
10 #include "amok/Bandwidth/bandwidth_private.h"
11 #include "gras/Msg/msg_private.h" /* FIXME: This mucks with contextes to answer RPC directly */
12
13 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(amok_bw_sat,amok_bw,"Everything concerning the SATuration part of the amok_bw module");
14
15 static int amok_bw_cb_sat_start(gras_msg_cb_ctx_t ctx, void *payload);
16 static int amok_bw_cb_sat_begin(gras_msg_cb_ctx_t ctx, void *payload);
17
18
19 void amok_bw_sat_init(void) {
20   gras_datadesc_type_t bw_res_desc=gras_datadesc_by_name("bw_res_t");
21   gras_datadesc_type_t sat_request_desc;
22   /* Build the saturation datatype descriptions */ 
23   
24   sat_request_desc = gras_datadesc_struct("s_sat_request_desc_t");
25   gras_datadesc_struct_append(sat_request_desc,"host",gras_datadesc_by_name("s_xbt_host_t"));
26   gras_datadesc_struct_append(sat_request_desc,"msg_size",gras_datadesc_by_name("unsigned int"));
27   gras_datadesc_struct_append(sat_request_desc,"duration",gras_datadesc_by_name("unsigned int"));
28   gras_datadesc_struct_close(sat_request_desc);
29   sat_request_desc = gras_datadesc_ref("sat_request_t",sat_request_desc);
30   
31   /* Register the saturation messages */
32   gras_msgtype_declare_rpc("amok_bw_sat start", sat_request_desc, NULL);
33   gras_msgtype_declare_rpc("amok_bw_sat begin", sat_request_desc, sat_request_desc);
34   gras_msgtype_declare_rpc("amok_bw_sat stop",  NULL,             bw_res_desc);
35
36 }
37 void amok_bw_sat_join(void) {
38   gras_cb_register(gras_msgtype_by_name("amok_bw_sat start"),
39                    &amok_bw_cb_sat_start);
40   gras_cb_register(gras_msgtype_by_name("amok_bw_sat begin"),
41                    &amok_bw_cb_sat_begin);
42 }
43 void amok_bw_sat_leave(void) {
44   gras_cb_unregister(gras_msgtype_by_name("amok_bw_sat start"),
45                      &amok_bw_cb_sat_start);
46   gras_cb_unregister(gras_msgtype_by_name("amok_bw_sat begin"),
47                      &amok_bw_cb_sat_begin);
48 }
49
50 /* ***************************************************************************
51  * Link saturation
52  * ***************************************************************************/
53
54 /**
55  * @brief Ask 'from_name:from_port' to stop saturating going to to_name:to_name.
56  *
57  * @param from_name: Name of the host we are asking to do a experiment with (to_name:to_port)
58  * @param from_port: port on which the process we are asking for an experiment is listening
59  * (for message, do not give a raw socket here. The needed raw socket will be negociated 
60  * between the peers)
61  * @param to_name: Name of the host with which we should conduct the experiment
62  * @param to_port: port on which the peer process is listening for message
63  * @param msg_size: Size of each message sent.
64  * @param duration: How long in maximum should be the saturation.
65  *
66  * Ask the process 'from_name:from_port' to start to saturate the link between itself
67  * and to_name:to_name.
68  */
69 void amok_bw_saturate_start(const char* from_name,unsigned int from_port,
70                             const char* to_name,unsigned int to_port,
71                             unsigned int msg_size, double duration) {
72   gras_socket_t sock;
73
74   sat_request_t request = xbt_new(s_sat_request_t,1);
75
76   sock = gras_socket_client(from_name,from_port);
77
78   request->host.name = (char*)to_name;
79   request->host.port = to_port;
80   
81   request->duration=duration;
82   request->msg_size=msg_size;
83
84
85   gras_msg_rpccall(sock,60,gras_msgtype_by_name("amok_bw_sat start"),&request, NULL);
86
87   free(request);
88   gras_socket_close(sock);
89 }
90
91 /* Asked to begin a saturation */
92 static int amok_bw_cb_sat_start(gras_msg_cb_ctx_t ctx, void *payload){
93   sat_request_t request = *(sat_request_t*)payload;
94   gras_socket_t expeditor = gras_msg_cb_ctx_from(ctx);
95
96   VERB4("Asked by %s:%d to start a saturation to %s:%d",
97         gras_socket_peer_name(expeditor),gras_socket_peer_port(expeditor),
98         request->host.name,request->host.port);
99         
100   gras_msg_rpcreturn(60,ctx, NULL);
101   amok_bw_saturate_begin(request->host.name,request->host.port,
102                          request->msg_size, request->duration,
103                          NULL,NULL);
104   free(request->host.name);
105   free(request);
106   return 1;
107 }
108
109 /**
110  * @brief Start saturating between the current process and the designated peer
111  *
112  * Note that the only way to break this function before the end of the timeout
113  * is to have a remote host calling amok_bw_saturate_stop to this process.
114  *
115  * If duration=0, the experiment will never timeout (you then have to manually
116  * stop it)
117  */
118 void amok_bw_saturate_begin(const char* to_name,unsigned int to_port,
119                             unsigned int msg_size, double duration,
120                             /*out*/ double *elapsed_res, double *bw_res) {
121  
122   xbt_ex_t e;
123
124   gras_socket_t peer_cmd = gras_socket_client(to_name, to_port);
125   gras_msg_cb_ctx_t ctx;
126
127   gras_socket_t meas;
128
129   s_gras_msg_t msg_got;
130
131   unsigned int packet_sent=0;
132   double start,elapsed=-1; /* timer */
133   double bw;
134
135   volatile int saturate_further; /* boolean in the main loop */ 
136
137   /* Negociate the saturation with the peer */
138   sat_request_t request = xbt_new(s_sat_request_t,1);
139
140   DEBUG2("Begin to saturate to %s:%d",to_name,to_port);
141   memset(&msg_got,0,sizeof(msg_got));
142
143   request->msg_size = msg_size;
144   request->duration = duration;
145   request->host.name = NULL;
146   request->host.port = 0;
147
148   ctx = gras_msg_rpc_async_call(peer_cmd, 60, 
149                                 gras_msgtype_by_name("amok_bw_sat begin"),
150                                                   &request);
151   free(request);
152   gras_msg_rpc_async_wait(ctx,&request);
153   meas=gras_socket_client_ext( to_name, request->host.port,
154                                0 /*bufsize: auto*/,
155                                1 /*meas: true*/);
156   free(request);
157
158   gras_socket_close(peer_cmd);
159   INFO2("Saturation(%s->%s) started",gras_os_myname(),to_name);
160
161   /* Start experiment */
162   start=gras_os_time();
163
164   do {
165     /* do send it */
166     gras_socket_meas_send(meas,120,msg_size,msg_size);
167     packet_sent++;
168
169     /* Check whether someone asked us to stop saturation */
170     saturate_further = 0;
171     TRY {
172       gras_msg_wait_ext(0/*no wait*/,gras_msgtype_by_name("amok_bw_sat stop"),
173                         NULL /* accept any sender */,
174                         NULL, NULL, /* No specific filter */
175                         &msg_got);
176     } CATCH(e) {
177       if (e.category == timeout_error) {
178         saturate_further=1;
179         memset(&msg_got,0,sizeof(msg_got)); /* may be overprotectiv here */
180       }
181       xbt_ex_free(e);
182     }
183
184     /* Check whether the experiment has to be terminated by now */
185     elapsed=gras_os_time()-start;
186     DEBUG2("elapsed %f duration %f",elapsed, duration);
187
188   } while (saturate_further && (duration==0 || elapsed < duration));
189
190   bw = ((double)(packet_sent*msg_size)) / elapsed;
191
192   if (elapsed_res)
193     *elapsed_res = elapsed;
194   if (bw_res)
195     *bw_res = bw;
196
197   /* If someone stopped us, inform him about the achieved bandwidth */
198   if (msg_got.expe) {
199     bw_res_t answer = xbt_new(s_bw_res_t,1);
200     s_gras_msg_cb_ctx_t ctx;
201
202     INFO3("Saturation from %s to %s stopped by %s",
203           gras_os_myname(),to_name, gras_socket_peer_name(msg_got.expe));
204     answer->timestamp=gras_os_time();
205     answer->sec=elapsed;
206     answer->bw=bw;
207
208     ctx.expeditor = msg_got.expe;
209     ctx.ID = msg_got.ID;
210     ctx.msgtype = msg_got.type;
211
212     gras_msg_rpcreturn(60,&ctx,&answer);
213     free(answer);
214   } else {
215     INFO4("Saturation from %s to %s elapsed after %f sec (achieving %f kb/s)",
216           gras_os_myname(),to_name,elapsed,bw/1024.0);
217   }
218   
219   gras_socket_close(meas);
220 }
221
222 /* Sender will saturate link to us */
223 static int amok_bw_cb_sat_begin(gras_msg_cb_ctx_t ctx, void *payload){
224   sat_request_t request=*(sat_request_t*)payload;
225   sat_request_t answer=xbt_new0(s_sat_request_t,1);
226   volatile int saturate_further = 1;
227   xbt_ex_t e;
228   gras_socket_t measMaster=NULL,meas=NULL;
229   gras_socket_t from=gras_msg_cb_ctx_from(ctx);
230
231   int port=6000;
232   while (port <= 10000 && measMaster == NULL) {
233     TRY {
234       measMaster = gras_socket_server_ext(port,
235                                           0 /*bufsize: auto*/,
236                                           1 /*meas: true*/);
237     } CATCH(e) {
238       measMaster = NULL;
239       if (port < 10000)
240         xbt_ex_free(e);
241       else
242         RETHROW0("Error encountered while opening a measurement server socket: %s");
243     }
244     if (measMaster == NULL) 
245       port++; /* prepare for a new loop */
246   }
247   answer->host.port=port;
248
249   gras_msg_rpcreturn(60, ctx, &answer);
250   free(answer);
251
252   gras_os_sleep(5); /* Wait for the accept */
253
254   TRY {
255     meas = gras_socket_meas_accept(measMaster);
256     DEBUG0("saturation handshake answered");
257   } CATCH(e) {
258     gras_socket_close(measMaster);
259     RETHROW0("Error during saturation handshake: %s");
260   }  
261
262   while (saturate_further) {
263     TRY {
264       gras_socket_meas_recv(meas,5,request->msg_size,request->msg_size);
265     } CATCH(e) {
266       saturate_further = 0;
267       xbt_ex_free(e);
268     }
269   }
270   INFO3("Saturation comming from %s:%d stopped on %s",
271         gras_socket_peer_name(from),gras_socket_peer_port(from),
272         gras_os_myname());
273
274   gras_socket_close(meas);
275   if (gras_if_RL()) /* On SG, accepted=master */
276     gras_socket_close(measMaster); 
277   free(request);
278   return 1;
279 }
280
281 /**
282  * @brief Ask 'from_name:from_port' to stop any saturation experiments
283  * @param from_name: Name of the host we are asking to do a experiment with (to_name:to_port)
284  * @param from_port: port on which the process we are asking for an experiment is listening
285  * @param time: the duration of the experiment
286  * @param bw: the achieved bandwidth
287  *
288  */
289 void amok_bw_saturate_stop(const char* from_name,unsigned int from_port,
290                            /*out*/ double *time, double *bw) {
291   xbt_ex_t e;
292    
293   gras_socket_t sock = gras_socket_client(from_name,from_port);
294   bw_res_t answer;
295   VERB2("Ask %s:%d to stop the saturation",from_name,from_port);
296   TRY { 
297      gras_msg_rpccall(sock,60,gras_msgtype_by_name("amok_bw_sat stop"),NULL,&answer);
298   } CATCH(e) {
299      RETHROW2("Cannot ask %s:%d to stop saturation: %s",from_name, from_port);
300   }
301   gras_socket_close(sock);
302   if (time) *time=answer->sec;
303   if (bw)   *bw  =answer->bw;
304   free(answer);
305 }