X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/3c79423e7376662ff7bcfa809c4803564cf82acb..16fdb8c65ac8d5baebc0278c499546c5fb10207b:/src/gras/Msg/msg.c diff --git a/src/gras/Msg/msg.c b/src/gras/Msg/msg.c index 50c529f1db..a27a327eea 100644 --- a/src/gras/Msg/msg.c +++ b/src/gras/Msg/msg.c @@ -7,17 +7,19 @@ /* 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 "xbt/ex.h" #include "gras/Msg/msg_private.h" -#include "gras/DataDesc/datadesc_interface.h" -#include "gras/Transport/transport_interface.h" /* gras_trp_chunk_send/recv */ #include "gras/Virtu/virtu_interface.h" +#include "gras/DataDesc/datadesc_interface.h" +#include "gras/Transport/transport_interface.h" /* gras_select */ + +#define MIN(a,b) ((a) < (b) ? (a) : (b)) XBT_LOG_NEW_DEFAULT_SUBCATEGORY(gras_msg,gras,"High level messaging"); xbt_set_t _gras_msgtype_set = NULL; -static char GRAS_header[6]; static char *make_namev(const char *name, short int ver); +char _GRAS_header[6]; /* * Creating procdata for this module @@ -25,8 +27,11 @@ static char *make_namev(const char *name, short int ver); static void *gras_msg_procdata_new() { gras_msg_procdata_t res = xbt_new(s_gras_msg_procdata_t,1); - res->msg_queue = xbt_dynar_new(sizeof(gras_msg_t), NULL); + res->name = xbt_strdup("gras_msg"); + res->name_len = 0; + res->msg_queue = xbt_dynar_new(sizeof(s_gras_msg_t), NULL); res->cbl_list = xbt_dynar_new(sizeof(gras_cblist_t *),gras_cbl_free); + res->timers = xbt_dynar_new(sizeof(s_gras_timer_t), NULL); return (void*)res; } @@ -39,13 +44,17 @@ static void gras_msg_procdata_free(void *data) { xbt_dynar_free(&( res->msg_queue )); xbt_dynar_free(&( res->cbl_list )); + xbt_dynar_free(&( res->timers )); + + free(res); } /* * Module registration */ +int gras_msg_libdata_id; void gras_msg_register() { - gras_procdata_add("gras_msg",gras_msg_procdata_new, gras_msg_procdata_free); + gras_msg_libdata_id = gras_procdata_add("gras_msg",gras_msg_procdata_new, gras_msg_procdata_free); } /* @@ -60,9 +69,9 @@ void gras_msg_init(void) { _gras_msgtype_set = xbt_set_new(); - memcpy(GRAS_header,"GRAS", 4); - GRAS_header[4]=GRAS_PROTOCOL_VERSION; - GRAS_header[5]=(char)GRAS_THISARCH; + memcpy(_GRAS_header,"GRAS", 4); + _GRAS_header[4]=GRAS_PROTOCOL_VERSION; + _GRAS_header[5]=(char)GRAS_THISARCH; } /* @@ -80,8 +89,8 @@ gras_msg_exit(void) { void gras_msgtype_free(void *t) { gras_msgtype_t msgtype=(gras_msgtype_t)t; if (msgtype) { - xbt_free(msgtype->name); - xbt_free(msgtype); + free(msgtype->name); + free(msgtype); } } @@ -121,7 +130,7 @@ void gras_msgtype_declare(const char *name, * @param version: something like versionning symbol * @param payload: datadescription of the payload * - * Registers a message to the GRAS mecanism. Use this version instead of + * Registers a message to the GRAS mechanism. Use this version instead of * gras_msgtype_declare when you change the semantic or syntax of a message and * want your programs to be able to deal with both versions. Internally, each * will be handled as an independent message type, so you can register @@ -132,14 +141,21 @@ gras_msgtype_declare_v(const char *name, short int version, gras_datadesc_type_t payload) { - xbt_error_t errcode; - gras_msgtype_t msgtype; + gras_msgtype_t msgtype=NULL; char *namev=make_namev(name,version); + volatile int found = 0; + xbt_ex_t e; - errcode = xbt_set_get_by_name(_gras_msgtype_set, - namev,(xbt_set_elm_t*)&msgtype); + TRY { + msgtype = (gras_msgtype_t)xbt_set_get_by_name(_gras_msgtype_set,namev); + found = 1; + } CATCH(e) { + if (e.category != not_found_error) + RETHROW; + xbt_ex_free(e); + } - if (errcode == no_error) { + if (found) { VERB2("Re-register version %d of message '%s' (same payload, ignored).", version, name); xbt_assert3(!gras_datadesc_type_cmp(msgtype->ctn_type, payload), @@ -150,7 +166,7 @@ gras_msgtype_declare_v(const char *name, return ; /* do really ignore it */ } - xbt_assert_error(mismatch_error); /* expect this error */ + VERB3("Register version %d of message '%s' (payload: %s).", version, name, gras_datadesc_get_name(payload)); @@ -173,216 +189,237 @@ gras_msgtype_t gras_msgtype_by_name (const char *name) { gras_msgtype_t gras_msgtype_by_namev(const char *name, short int version) { gras_msgtype_t res; - - xbt_error_t errcode; char *namev = make_namev(name,version); - errcode = xbt_set_get_by_name(_gras_msgtype_set, namev, - (xbt_set_elm_t*)&res); - if (errcode != no_error) - res = NULL; - if (!res) - WARN1("msgtype_by_name(%s) returns NULL",namev); + res = (gras_msgtype_t)xbt_set_get_by_name(_gras_msgtype_set, namev); if (name != namev) - xbt_free(namev); + free(namev); return res; } - -/** \brief Send the data pointed by \a payload as a message of type - * \a msgtype to the peer \a sock */ -xbt_error_t -gras_msg_send(gras_socket_t sock, - gras_msgtype_t msgtype, - void *payload) { - - xbt_error_t errcode; - static gras_datadesc_type_t string_type=NULL; - - if (!msgtype) - RAISE0(mismatch_error, - "Cannot send the NULL message (did msgtype_by_name fail?)"); - - if (!string_type) { - string_type = gras_datadesc_by_name("string"); - xbt_assert(string_type); - } - - DEBUG3("send '%s' to %s:%d", msgtype->name, - gras_socket_peer_name(sock),gras_socket_peer_port(sock)); - TRY(gras_trp_chunk_send(sock, GRAS_header, 6)); - - TRY(gras_datadesc_send(sock, string_type, &msgtype->name)); - TRY(gras_datadesc_send(sock, msgtype->ctn_type, payload)); - TRY(gras_trp_flush(sock)); - - return no_error; -} -/* - * receive the next message on the given socket. - */ -xbt_error_t -gras_msg_recv(gras_socket_t sock, - gras_msgtype_t *msgtype, - void **payload, - int *payload_size) { - - xbt_error_t errcode; - static gras_datadesc_type_t string_type=NULL; - char header[6]; - int cpt; - int r_arch; - char *msg_name=NULL; - - if (!string_type) { - string_type=gras_datadesc_by_name("string"); - xbt_assert(string_type); - } - - TRY(gras_trp_chunk_recv(sock, header, 6)); - for (cpt=0; cpt<4; cpt++) - if (header[cpt] != GRAS_header[cpt]) - RAISE0(mismatch_error,"Incoming bytes do not look like a GRAS message"); - if (header[4] != GRAS_header[4]) - RAISE2(mismatch_error,"GRAS protocol mismatch (got %d, use %d)", - (int)header[4], (int)GRAS_header[4]); - r_arch = (int)header[5]; - DEBUG2("Handle an incoming message using protocol %d (remote is %s)", - (int)header[4],gras_datadesc_arch_name(r_arch)); - - TRY(gras_datadesc_recv(sock, string_type, r_arch, &msg_name)); - errcode = xbt_set_get_by_name(_gras_msgtype_set, - msg_name,(xbt_set_elm_t*)msgtype); - if (errcode != no_error) - RAISE2(errcode, - "Got error %s while retrieving the type associated to messages '%s'", - xbt_error_name(errcode),msg_name); - /* FIXME: Survive unknown messages */ - xbt_free(msg_name); - - *payload_size=gras_datadesc_size((*msgtype)->ctn_type); - xbt_assert2(*payload_size > 0, - "%s %s", - "Dynamic array as payload is forbided for now (FIXME?).", - "Reference to dynamic array is allowed."); - *payload = xbt_malloc(*payload_size); - TRY(gras_datadesc_recv(sock, (*msgtype)->ctn_type, r_arch, *payload)); - - return no_error; +/** @brief retrive an existing message type from its name and version. */ +gras_msgtype_t gras_msgtype_by_id(int id) { + return (gras_msgtype_t)xbt_set_get_by_id(_gras_msgtype_set, id); } /** \brief Waits for a message to come in over a given socket. * * @param timeout: How long should we wait for this message. - * @param msgt_want: type of awaited msg - * @param[out] expeditor: where to create a socket to answer the incomming message - * @param[out] payload: where to write the payload of the incomming message - * @return the error code (or no_error). + * @param msgt_want: type of awaited msg (or NULL if I'm enclined to accept any message) + * @param expe_want: awaited expeditot (match on hostname, not port; NULL if not relevant) + * @param payl_filter: function returning true or false when passed a payload. Messages for which it returns false are not selected. (NULL if not relevant) + * @param filter_ctx: context passed as second argument of the filter (a pattern to match?) + * @param[out] msgt_got: where to write the descriptor of the message we got + * @param[out] expe_got: where to create a socket to answer the incomming message + * @param[out] payl_got: where to write the payload of the incomming message * * Every message of another type received before the one waited will be queued * and used by subsequent call to this function or gras_msg_handle(). */ -xbt_error_t -gras_msg_wait(double timeout, - gras_msgtype_t msgt_want, - gras_socket_t *expeditor, - void *payload) { - gras_msgtype_t msgt_got; - void *payload_got; - int payload_size_got; - xbt_error_t errcode; +void +gras_msg_wait_ext(double timeout, + gras_msgtype_t msgt_want, + gras_socket_t expe_want, + int_f_pvoid_pvoid_t payl_filter, + void *filter_ctx, + gras_msgtype_t *msgt_got, + gras_socket_t *expe_got, + void *payl_got) { + + s_gras_msg_t msg; double start, now; - gras_msg_procdata_t pd=(gras_msg_procdata_t)gras_libdata_get("gras_msg"); + gras_msg_procdata_t pd=(gras_msg_procdata_t)gras_libdata_by_id(gras_msg_libdata_id); int cpt; - gras_msg_t msg; - - *expeditor = NULL; - payload_got = NULL; - if (!msgt_want) - RAISE0(mismatch_error, - "Cannot wait for the NULL message (did msgtype_by_name fail?)"); + xbt_assert0(msgt_want,"Cannot wait for the NULL message"); - VERB1("Waiting for message %s",msgt_want->name); + VERB1("Waiting for message '%s'",msgt_want->name); start = now = gras_os_time(); xbt_dynar_foreach(pd->msg_queue,cpt,msg){ - if (msg.type->code == msgt_want->code) { - *expeditor = msg.expeditor; - memcpy(payload, msg.payload, msg.payload_size); - xbt_free(msg.payload); + if ( ( !msgt_want || (msg.type->code == msgt_want->code)) + && (!expe_want || (!strcmp( gras_socket_peer_name(msg.expe), + gras_socket_peer_name(expe_want)))) + && (!payl_filter || payl_filter(msg.payl,filter_ctx))) { + + if (expe_got) + *expe_got = msg.expe; + if (msgt_got) + *msgt_got = msg.type; + if (payl_got) + memcpy(payl_got, msg.payl, msg.payl_size); + free(msg.payl); xbt_dynar_cursor_rm(pd->msg_queue, &cpt); VERB0("The waited message was queued"); - return no_error; + return; } } while (1) { - TRY(gras_trp_select(timeout - now + start, expeditor)); - TRY(gras_msg_recv(*expeditor, &msgt_got, &payload_got, &payload_size_got)); - if (msgt_got->code == msgt_want->code) { - memcpy(payload, payload_got, payload_size_got); - xbt_free(payload_got); - VERB0("Got waited message"); - return no_error; + memset(&msg,sizeof(msg),0); + + msg.expe = gras_trp_select(timeout - now + start); + gras_msg_recv(msg.expe, &msg); + + if ( ( !msgt_want || (msg.type->code == msgt_want->code)) + && (!expe_want || (!strcmp( gras_socket_peer_name(msg.expe), + gras_socket_peer_name(expe_want)))) + && (!payl_filter || payl_filter(msg.payl,filter_ctx))) { + + if (expe_got) + *expe_got=msg.expe; + if (msgt_got) + *msgt_got = msg.type; + if (payl_got) + memcpy(payl_got, msg.payl, msg.payl_size); + free(msg.payl); + return; } /* not expected msg type. Queue it for later */ - msg.expeditor = *expeditor; - msg.type = msgt_got; - msg.payload = payload; - msg.payload_size = payload_size_got; xbt_dynar_push(pd->msg_queue,&msg); now=gras_os_time(); if (now - start + 0.001 < timeout) { - RAISE1(timeout_error,"Timeout while waiting for msg %s",msgt_want->name); + THROW1(timeout_error, now-start+0.001-timeout, + "Timeout while waiting for msg %s",msgt_want->name); } } - RAISE_IMPOSSIBLE; + THROW_IMPOSSIBLE; +} +/** \brief Waits for a message to come in over a given socket. + * + * @param timeout: How long should we wait for this message. + * @param msgt_want: type of awaited msg + * @param[out] expeditor: where to create a socket to answer the incomming message + * @param[out] payload: where to write the payload of the incomming message + * @return the error code (or no_error). + * + * Every message of another type received before the one waited will be queued + * and used by subsequent call to this function or gras_msg_handle(). + */ +void +gras_msg_wait(double timeout, + gras_msgtype_t msgt_want, + gras_socket_t *expeditor, + void *payload) { + + return gras_msg_wait_ext(timeout, + msgt_want, NULL, NULL, NULL, + NULL, expeditor, payload); +} + + +/** \brief Send the data pointed by \a payload as a message of type + * \a msgtype to the peer \a sock */ +void +gras_msg_send(gras_socket_t sock, + gras_msgtype_t msgtype, + void *payload) { + + gras_msg_send_ext(sock, e_gras_msg_kind_oneway, msgtype, payload); } /** @brief Handle an incomming message or timer (or wait up to \a timeOut seconds) * - * @param timeOut: How long to wait for incoming messages + * @param timeOut: How long to wait for incoming messages (in seconds) * @return the error code (or no_error). * * Messages are passed to the callbacks. */ -xbt_error_t +void gras_msg_handle(double timeOut) { - xbt_error_t errcode; + double untiltimer; + int cpt; - gras_msg_t msg; - gras_socket_t expeditor; + s_gras_msg_t msg; + gras_socket_t expeditor=NULL; void *payload=NULL; - int payload_size; - gras_msgtype_t msgtype; + gras_msgtype_t msgtype=NULL; - gras_msg_procdata_t pd=(gras_msg_procdata_t)gras_libdata_get("gras_msg"); - gras_cblist_t *list; + gras_msg_procdata_t pd=(gras_msg_procdata_t)gras_libdata_by_id(gras_msg_libdata_id); + gras_cblist_t *list=NULL; gras_msg_cb_t cb; + + int timerexpected, timeouted; + xbt_ex_t e; VERB1("Handling message within the next %.2fs",timeOut); + untiltimer = gras_msg_timer_handle(); + DEBUG2("[%.0f] Next timer in %f sec", gras_os_time(), untiltimer); + if (untiltimer == 0.0) { + /* A timer was already elapsed and handled */ + return; + } + if (untiltimer != -1.0) { + timerexpected = 1; + timeOut = MIN(timeOut, untiltimer); + } else { + timerexpected = 0; + } + /* get a message (from the queue or from the net) */ + timeouted = 0; if (xbt_dynar_length(pd->msg_queue)) { + DEBUG0("Get a message from the queue"); xbt_dynar_shift(pd->msg_queue,&msg); - expeditor = msg.expeditor; + expeditor = msg.expe; msgtype = msg.type; - payload = msg.payload; - + payload = msg.payl; } else { - TRY(gras_trp_select(timeOut, &expeditor)); - TRY(gras_msg_recv(expeditor, &msgtype, &payload, &payload_size)); + TRY { + expeditor = gras_trp_select(timeOut); + } CATCH(e) { + if (e.category != timeout_error) + RETHROW; + xbt_ex_free(e); + timeouted = 1; + } + + if (!timeouted) { + TRY { + /* FIXME: if not the right kind, queue it and recall ourself or goto >:-) */ + gras_msg_recv(expeditor, &msg); + msgtype = msg.type; + payload = msg.payl; + } CATCH(e) { + RETHROW1("Error caught while receiving a message on select()ed socket %p: %s", + expeditor); + } + } + } + + if (timeouted) { + if (timerexpected) { + + /* A timer elapsed before the arrival of any message even if we select()ed a bit */ + untiltimer = gras_msg_timer_handle(); + if (untiltimer == 0.0) { + /* we served a timer, we're done */ + return; + } else { + xbt_assert1(untiltimer>0, "Negative timer (%f). I'm 'puzzeled'", untiltimer); + WARN1("No timer elapsed, in contrary to expectations (next in %f sec)", + untiltimer); + THROW1(timeout_error,0, + "No timer elapsed, in contrary to expectations (next in %f sec)", + untiltimer); + } + + } else { + /* select timeouted, and no timer elapsed. Nothing to do */ + THROW0(timeout_error, 0, "No new message or timer"); + } + } - - /* handle it */ + + /* A message was already there or arrived in the meanwhile. handle it */ xbt_dynar_foreach(pd->cbl_list,cpt,list) { if (list->id == msgtype->code) { break; @@ -394,7 +431,7 @@ gras_msg_handle(double timeOut) { INFO1("No callback for the incomming '%s' message. Discarded.", msgtype->name); WARN0("FIXME: gras_datadesc_free not implemented => leaking the payload"); - return no_error; + return; } xbt_dynar_foreach(list->cbs,cpt,cb) { @@ -402,14 +439,14 @@ gras_msg_handle(double timeOut) { cpt+1,cb,msgtype->name); if ((*cb)(expeditor,payload)) { /* cb handled the message */ - xbt_free(payload); - return no_error; + free(payload); + return; } } - INFO1("Message '%s' refused by all registered callbacks", msgtype->name); - WARN0("FIXME: gras_datadesc_free not implemented => leaking the payload"); - return mismatch_error; + /* FIXME: gras_datadesc_free not implemented => leaking the payload */ + THROW1(mismatch_error,0, + "Message '%s' refused by all registered callbacks", msgtype->name); } void @@ -417,7 +454,7 @@ gras_cbl_free(void *data){ gras_cblist_t *list=*(void**)data; if (list) { xbt_dynar_free(&( list->cbs )); - xbt_free(list); + free(list); } } @@ -430,11 +467,11 @@ gras_cbl_free(void *data){ void gras_cb_register(gras_msgtype_t msgtype, gras_msg_cb_t cb) { - gras_msg_procdata_t pd=(gras_msg_procdata_t)gras_libdata_get("gras_msg"); + gras_msg_procdata_t pd=(gras_msg_procdata_t)gras_libdata_by_id(gras_msg_libdata_id); gras_cblist_t *list=NULL; int cpt; - DEBUG2("Register %p as callback to %s",cb,msgtype->name); + DEBUG2("Register %p as callback to '%s'",cb,msgtype->name); /* search the list of cb for this message on this host (creating if NULL) */ xbt_dynar_foreach(pd->cbl_list,cpt,list) { @@ -461,7 +498,7 @@ void gras_cb_unregister(gras_msgtype_t msgtype, gras_msg_cb_t cb) { - gras_msg_procdata_t pd=(gras_msg_procdata_t)gras_libdata_get("gras_msg"); + gras_msg_procdata_t pd=(gras_msg_procdata_t)gras_libdata_by_id(gras_msg_libdata_id); gras_cblist_t *list; gras_msg_cb_t cb_cpt; int cpt;