X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/a4488cf61cb9b235eb6a370dbd0ab63ff936e72a..5cb343842f5ce9e1db20f0b24cbbc53c26f1e43f:/src/amok/Bandwidth/saturate.c?ds=sidebyside diff --git a/src/amok/Bandwidth/saturate.c b/src/amok/Bandwidth/saturate.c index f58c8c90d4..203535baad 100644 --- a/src/amok/Bandwidth/saturate.c +++ b/src/amok/Bandwidth/saturate.c @@ -2,83 +2,292 @@ /* amok_saturate - Link saturating facilities (for ALNeM's BW testing) */ -/* Authors: Martin Quinson */ -/* Copyright (C) 2003, 2004 the OURAGAN project. */ +/* Copyright (c) 2003, 2004 Martin Quinson. 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. */ + * 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 */ -GRAS_LOG_EXTERNAL_CATEGORY(bw); -GRAS_LOG_DEFAULT_CATEGORY(bw); +XBT_LOG_EXTERNAL_CATEGORY(bw); +XBT_LOG_DEFAULT_CATEGORY(bw); + +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 sat_request_desc; + /* 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_close(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, NULL); + +} +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_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); +} -#if 0 /* *************************************************************************** * Link saturation * ***************************************************************************/ -gras_error_t grasbw_saturate_start(const char* from_name,unsigned int from_port, - const char* to_name,unsigned int to_port, - unsigned int msgSize, unsigned int timeout) { - gras_sock_t *sock; - gras_error_t errcode; - /* The request */ - SatExp_t *request; - msgHost_t *target; - /* answer */ - gras_msg_t *answer; +/** + * @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_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_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. + * + * 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) { + gras_socket_t sock; + + sat_request_t request = xbt_new(s_sat_request_t,1); + + 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; - if((errcode=gras_sock_client_open(from_name,from_port,&sock))) { - fprintf(stderr,"%s:%d:saturate_start(): Error %s encountered while contacting peer\n", - __FILE__,__LINE__,gras_error_name(errcode)); - return errcode; - } - if (!(request=(SatExp_t *)malloc(sizeof(SatExp_t))) || - !(target=(msgHost_t*)malloc(sizeof(msgHost_t)))) { - fprintf(stderr,"%s:%d:saturate_start(): Malloc error\n",__FILE__,__LINE__); - gras_sock_close(sock); - return malloc_error; - } - request->timeout=timeout; - request->msgSize=msgSize; + gras_msg_rpccall(sock,60,gras_msgtype_by_name("amok_bw_sat start"),&request, NULL); - strcpy(target->host,to_name); - target->port=to_port; + free(request); + gras_socket_close(sock); +} - if ((errcode=gras_msg_new_and_send(sock,GRASMSG_SAT_START, 2, - target,1, - request,1))) { - fprintf(stderr,"%s:%d:saturate_start(): Error %s encountered while sending the request.\n", - __FILE__,__LINE__,gras_error_name(errcode)); - gras_sock_close(sock); - return errcode; +/* 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; + 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); + free(request); + return 1; +} + +/** + * @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. + */ +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); + gras_msg_cb_ctx_t ctx; + + gras_socket_t meas; + + s_gras_msg_t msg_got; + + unsigned int packet_sent=0; + double start,elapsed=-1; /* timer */ + double bw; + + 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); + + 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; + + ctx = gras_msg_rpc_async_call(peer_cmd, 60, + gras_msgtype_by_name("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*/); + free(request); + + gras_socket_close(peer_cmd); + INFO2("Saturation from %s to %s started",gras_os_myname(),to_name); + + /* Start experiment */ + start=gras_os_time(); + + do { + /* do send it */ + gras_socket_meas_send(meas,120,msg_size,msg_size); + 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) { + if (e.category == timeout_error) { + 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; + VERB2("elapsed %f duration %f",elapsed, duration); + + } while (saturate_further && elapsed < duration); + + INFO2("Saturation from %s to %s stopped",gras_os_myname(),to_name); + bw = ((double)(packet_sent*msg_size)) / elapsed; + + if (elapsed_res) + *elapsed_res = elapsed; + if (bw_res) + *bw_res = bw; + + if (elapsed >= duration) { + INFO2("Saturation experiment terminated. Took %f sec (achieving %f kb/s)", + elapsed, bw/1024.0); } - if ((errcode=gras_msg_wait(120,GRASMSG_SAT_STARTED,&answer))) { - fprintf(stderr,"%s:%d:saturate_start(): Error %s encountered while waiting for the ACK.\n", - __FILE__,__LINE__,gras_error_name(errcode)); - gras_sock_close(sock); - return errcode; + + /* 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); + s_gras_msg_cb_ctx_t ctx; + + 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); + free(answer); } + + gras_socket_close(meas); +} - if((errcode=gras_msg_ctn(answer,0,0,msgError_t).errcode)) { - fprintf(stderr,"%s:%d:saturate_start(): Peer reported error %s (%s).\n", - __FILE__,__LINE__,gras_error_name(errcode),gras_msg_ctn(answer,0,0,msgError_t).errmsg); - gras_msg_free(answer); - gras_sock_close(sock); - return errcode; +/* 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); + volatile int saturate_further = 1; + xbt_ex_t e; + gras_socket_t measMaster=NULL,meas=NULL; + + int port=6000; + while (port <= 10000 && measMaster == NULL) { + TRY { + measMaster = gras_socket_server_ext(port, + 0 /*bufsize: auto*/, + 1 /*meas: true*/); + } CATCH(e) { + measMaster = NULL; + if (port < 10000) + xbt_ex_free(&e); + else + RETHROW0("Error encountered while opening a measurement server socket: %s"); + } + if (measMaster == NULL) + port++; /* prepare for a new loop */ + } + answer->host.port=port; + + gras_msg_rpcreturn(60, ctx, &answer); + free(answer); + + gras_os_sleep(5); /* Wait for the accept */ + + TRY { + meas = gras_socket_meas_accept(measMaster); + DEBUG0("saturation handshake answered"); + } CATCH(e) { + gras_socket_close(measMaster); + RETHROW0("Error during saturation handshake: %s"); + } + + while (saturate_further) { + TRY { + gras_socket_meas_recv(meas,120,request->msg_size,request->msg_size); + } CATCH(e) { + saturate_further = 0; + xbt_ex_free(&e); + } } + INFO1("Saturation stopped on %s",gras_os_myname()); + gras_socket_close(meas); + if (gras_if_RL()) /* On SG, accepted=master */ + gras_socket_close(measMaster); + free(request); + return 1; +} - gras_msg_free(answer); - gras_sock_close(sock); - return no_error; +/** + * @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_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*/ unsigned int *time, unsigned int *bw) { + + gras_socket_t sock = gras_socket_client(from_name,from_port); + gras_msg_rpccall(sock,60,gras_msgtype_by_name("amok_bw_sat stop"),NULL,NULL); + gras_socket_close(sock); } + +#if 0 int grasbw_cbSatStart(gras_msg_t *msg) { gras_rawsock_t *raw; gras_sock_t *sock; - gras_error_t errcode; + xbt_error_t errcode; double start; /* time to timeout */ /* specification of the test to run */ @@ -94,17 +303,11 @@ int grasbw_cbSatStart(gras_msg_t *msg) { /* answer */ gras_msg_t *answer; - /* - fprintf(stderr,"grasbw_cbSatStart(sd=%p)\n",msg->sock); - fprintf(stderr,"(server=%d,raw=%d,fromPID=%d,toPID=%d,toHost=%p,toPort=%d,toChan=%d)\n", - msg->sock->server_sock,msg->sock->raw_sock,msg->sock->from_PID, - msg->sock->to_PID,msg->sock->to_host,msg->sock->to_port,msg->sock->to_chan); - */ /* Negociate the saturation with the peer */ if((errcode=gras_sock_client_open(to_name,to_port,&sock))) { fprintf(stderr,"cbSatStart(): Error %s encountered while contacting peer\n", - gras_error_name(errcode)); + xbt_error_name(errcode)); grasRepportError(msg->sock,GRASMSG_SAT_STARTED,1, "cbSatStart: Severe error: Cannot send error status to requester!!\n", errcode,"Cannot contact peer.\n"); @@ -125,7 +328,7 @@ int grasbw_cbSatStart(gras_msg_t *msg) { if ((errcode=gras_msg_new_and_send(sock,GRASMSG_SAT_BEGIN, 1, request,1))) { fprintf(stderr,"cbSatStart(): Error %s encountered while sending the request.\n", - gras_error_name(errcode)); + xbt_error_name(errcode)); grasRepportError(msg->sock,GRASMSG_SAT_STARTED,1, "cbSatStart: Severe error: Cannot send error status to requester!!\n", errcode,"Cannot send request.\n"); @@ -135,7 +338,7 @@ int grasbw_cbSatStart(gras_msg_t *msg) { if ((errcode=gras_msg_wait(120,GRASMSG_SAT_BEGUN,&answer))) { fprintf(stderr,"cbSatStart(): Error %s encountered while waiting for the ACK.\n", - gras_error_name(errcode)); + xbt_error_name(errcode)); gras_sock_close(sock); grasRepportError(msg->sock,GRASMSG_SAT_STARTED,1, @@ -147,7 +350,7 @@ int grasbw_cbSatStart(gras_msg_t *msg) { if((errcode=gras_msg_ctn(answer,0,0,msgError_t).errcode)) { fprintf(stderr,"cbSatStart(): Peer reported error %s (%s).\n", - gras_error_name(errcode),gras_msg_ctn(answer,0,0,msgError_t).errmsg); + xbt_error_name(errcode),gras_msg_ctn(answer,0,0,msgError_t).errmsg); grasRepportError(msg->sock,GRASMSG_SAT_STARTED,1, "cbSatStart: Severe error: Cannot send error status to requester!!\n", @@ -162,7 +365,7 @@ int grasbw_cbSatStart(gras_msg_t *msg) { if ((errcode=gras_rawsock_client_open(to_name,raw_port,msgSize,&raw))) { fprintf(stderr,"cbSatStart(): Error %s while opening raw socket to %s:%d.\n", - gras_error_name(errcode),to_name,gras_msg_ctn(answer,1,0,SatExp_t).port); + xbt_error_name(errcode),to_name,gras_msg_ctn(answer,1,0,SatExp_t).port); grasRepportError(msg->sock,GRASMSG_SAT_STARTED,1, "cbSatStart: Severe error: Cannot send error status to requester!!\n", @@ -173,7 +376,7 @@ int grasbw_cbSatStart(gras_msg_t *msg) { /* send a train of data before repporting that XP is started */ if ((errcode=gras_rawsock_send(raw,msgSize,msgSize))) { - fprintf(stderr,"cbSatStart: Failure %s during raw send\n",gras_error_name(errcode)); + fprintf(stderr,"cbSatStart: Failure %s during raw send\n",xbt_error_name(errcode)); grasRepportError(msg->sock,GRASMSG_SAT_STARTED,1, "cbSatStart: Severe error: Cannot send error status to requester!!\n", errcode,"Cannot raw send.\n"); @@ -193,7 +396,7 @@ int grasbw_cbSatStart(gras_msg_t *msg) { while (gras_msg_wait(0,GRASMSG_SAT_STOP,&msg)==timeout_error && gras_time()-start < timeout) { if ((errcode=gras_rawsock_send(raw,msgSize,msgSize))) { - fprintf(stderr,"cbSatStart: Failure %s during raw send\n",gras_error_name(errcode)); + fprintf(stderr,"cbSatStart: Failure %s during raw send\n",xbt_error_name(errcode)); /* our error message do not interess anyone. SAT_STOP will do nothing. */ gras_sock_close(sock); gras_rawsock_close(raw); @@ -244,7 +447,7 @@ int grasbw_cbSatStart(gras_msg_t *msg) { int grasbw_cbSatBegin(gras_msg_t *msg) { gras_rawsock_t *raw; - gras_error_t errcode; + xbt_error_t errcode; double start; /* timer */ /* request */ unsigned int msgSize=gras_msg_ctn(msg,0,0,SatExp_t).msgSize; @@ -264,7 +467,7 @@ int grasbw_cbSatBegin(gras_msg_t *msg) { if ((errcode=gras_rawsock_server_open(6666,8000,msgSize,&raw))) { fprintf(stderr,"cbSatBegin(): Error %s encountered while opening a raw socket\n", - gras_error_name(errcode)); + xbt_error_name(errcode)); grasRepportError(msg->sock,GRASMSG_SAT_BEGUN,2, "cbSatBegin: Severe error: Cannot send error status to requester!!\n", errcode,"Cannot open raw socket"); @@ -278,7 +481,7 @@ int grasbw_cbSatBegin(gras_msg_t *msg) { error,1, request,1))) { fprintf(stderr,"cbSatBegin(): Error %s encountered while send ACK to peer\n", - gras_error_name(errcode)); + xbt_error_name(errcode)); return 1; } gras_msg_free(msg); @@ -288,7 +491,7 @@ int grasbw_cbSatBegin(gras_msg_t *msg) { gras_time() - start < timeout) { errcode=gras_rawsock_recv(raw,msgSize,msgSize,1); if (errcode != timeout_error && errcode != no_error) { - fprintf(stderr,"cbSatBegin: Failure %s during raw receive\n",gras_error_name(errcode)); + fprintf(stderr,"cbSatBegin: Failure %s during raw receive\n",xbt_error_name(errcode)); /* our error message do not interess anyone. SAT_END will do nothing. */ /* (if timeout'ed, it may be because the sender stopped emission. so survive it) */ return 1; @@ -308,35 +511,35 @@ int grasbw_cbSatBegin(gras_msg_t *msg) { return 1; } -gras_error_t grasbw_saturate_stop(const char* from_name,unsigned int from_port, +xbt_error_t grasbw_saturate_stop(const char* from_name,unsigned int from_port, const char* to_name,unsigned int to_port) { - gras_error_t errcode; + xbt_error_t errcode; gras_sock_t *sock; gras_msg_t *answer; if((errcode=gras_sock_client_open(from_name,from_port,&sock))) { fprintf(stderr,"saturate_stop(): Error %s encountered while contacting peer\n", - gras_error_name(errcode)); + xbt_error_name(errcode)); return errcode; } if ((errcode=gras_msg_new_and_send(sock,GRASMSG_SAT_STOP,0))) { fprintf(stderr,"saturate_stop(): Error %s encountered while sending request\n", - gras_error_name(errcode)); + xbt_error_name(errcode)); gras_sock_close(sock); return errcode; } if ((errcode=gras_msg_wait(120,GRASMSG_SAT_STOPPED,&answer))) { fprintf(stderr,"saturate_stop(): Error %s encountered while receiving ACK\n", - gras_error_name(errcode)); + xbt_error_name(errcode)); gras_sock_close(sock); return errcode; } if((errcode=gras_msg_ctn(answer,0,0,msgError_t).errcode)) { fprintf(stderr,"saturate_stop(): Peer reported error %s (%s).\n", - gras_error_name(errcode),gras_msg_ctn(answer,0,0,msgError_t).errmsg); + xbt_error_name(errcode),gras_msg_ctn(answer,0,0,msgError_t).errmsg); gras_msg_free(answer); gras_sock_close(sock); return errcode;