Logo AND Algorithmique Numérique Distribuée

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