Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Made an error message more explicit (hopefully)
[simgrid.git] / src / gras / Msg / msg.c
index c976c97..c0a5e03 100644 (file)
@@ -2,65 +2,90 @@
 
 /* messaging - Function related to messaging (code shared between RL and SG)*/
 
-/* Authors: Martin Quinson                                                  */
-/* Copyright (C) 2003 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 "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"
 
-GRAS_LOG_NEW_DEFAULT_SUBCATEGORY(msg,gras,"High level messaging");
+#define MIN(a,b) ((a) < (b) ? (a) : (b))
+
+XBT_LOG_NEW_DEFAULT_SUBCATEGORY(gras_msg,gras,"High level messaging");
 
-gras_set_t _gras_msgtype_set = NULL;
+xbt_set_t _gras_msgtype_set = NULL;
 static char GRAS_header[6];
 static char *make_namev(const char *name, short int ver);
 
-/**
- * gras_msg_init:
- *
+/*
+ * Creating procdata for this module
+ */
+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(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;
+}
+
+/*
+ * Freeing procdata for this module
+ */
+static void gras_msg_procdata_free(void *data) {
+   gras_msg_procdata_t res = (gras_msg_procdata_t)data;
+   
+   xbt_dynar_free(&( res->msg_queue ));
+   xbt_dynar_free(&( res->cbl_list ));
+   xbt_dynar_free(&( res->timers ));
+}
+
+/*
+ * Module registration
+ */
+void gras_msg_register() {
+   gras_procdata_add("gras_msg",gras_msg_procdata_new, gras_msg_procdata_free);
+}
+
+/*
  * Initialize this submodule.
  */
 void gras_msg_init(void) {
-  gras_error_t errcode;
-  
   /* only initialize once */
   if (_gras_msgtype_set != NULL)
     return;
 
   VERB0("Initializing Msg");
   
-  _gras_msgtype_set = gras_set_new();
+  _gras_msgtype_set = xbt_set_new();
 
   memcpy(GRAS_header,"GRAS", 4);
   GRAS_header[4]=GRAS_PROTOCOL_VERSION;
   GRAS_header[5]=(char)GRAS_THISARCH;
 }
 
-/**
- * gras_msg_exit:
- *
+/*
  * Finalize the msg module
- **/
+ */
 void
 gras_msg_exit(void) {
   VERB0("Exiting Msg");
-  gras_set_free(&_gras_msgtype_set);
+  xbt_set_free(&_gras_msgtype_set);
 }
 
-/**
- * gras_msgtype_free:
- *
+/*
  * Reclamed memory
  */
 void gras_msgtype_free(void *t) {
   gras_msgtype_t msgtype=(gras_msgtype_t)t;
   if (msgtype) {
-    gras_free(msgtype->name);
-    gras_free(msgtype);
+    free(msgtype->name);
+    free(msgtype);
   }
 }
 
@@ -76,7 +101,7 @@ static char *make_namev(const char *name, short int ver) {
   if (!ver)
     return (char *)name;
 
-  namev = (char*)gras_malloc(strlen(name)+2+3+1);
+  namev = (char*)xbt_malloc(strlen(name)+2+3+1);
 
   if (namev)
       sprintf(namev,"%s_v%d",name,ver);
@@ -84,25 +109,23 @@ static char *make_namev(const char *name, short int ver) {
   return namev;
 }
 
-/**
- * gras_msgtype_declare:
- * @name: name as it should be used for logging messages (must be uniq)
- * @payload: datadescription of the payload
+/** @brief declare a new message type of the given name. It only accepts the given datadesc as payload
  *
- * Registers a message to the GRAS mecanism.
+ * @param name: name as it should be used for logging messages (must be uniq)
+ * @param payload: datadescription of the payload
  */
 void gras_msgtype_declare(const char           *name,
                          gras_datadesc_type_t  payload) {
    gras_msgtype_declare_v(name, 0, payload);
 }
 
-/**
- * gras_msgtype_declare_v:
- * @name: name as it should be used for logging messages (must be uniq)
- * @version: something like versionning symbol
- * @payload: datadescription of the payload
+/** @brief declare a new versionned message type of the given name and payload
  *
- * Registers a message to the GRAS mecanism. Use this version instead of 
+ * @param name: name as it should be used for logging messages (must be uniq)
+ * @param version: something like versionning symbol
+ * @param payload: datadescription of the payload
+ *
+ * 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 
@@ -113,17 +136,20 @@ gras_msgtype_declare_v(const char           *name,
                       short int             version,
                       gras_datadesc_type_t  payload) {
  
-  gras_error_t   errcode;
+  xbt_error_t   errcode;
   gras_msgtype_t msgtype;
   char *namev=make_namev(name,version);
-  
-  errcode = gras_set_get_by_name(_gras_msgtype_set,
-                                namev,(gras_set_elm_t*)&msgtype);
+   
+  if (!payload) 
+     WARN1("Message %s has NULL payload",name);
+   
+  errcode = xbt_set_get_by_name(_gras_msgtype_set,
+                                namev,(xbt_set_elm_t*)&msgtype);
 
   if (errcode == no_error) {
     VERB2("Re-register version %d of message '%s' (same payload, ignored).",
          version, name);
-    gras_assert3(!gras_datadesc_type_cmp(msgtype->ctn_type, payload),
+    xbt_assert3(!gras_datadesc_type_cmp(msgtype->ctn_type, payload),
                 "Message %s re-registred with another payload (%s was %s)",
                 namev,gras_datadesc_get_name(payload),
                 gras_datadesc_get_name(msgtype->ctn_type));
@@ -131,63 +157,53 @@ gras_msgtype_declare_v(const char           *name,
     return ; /* do really ignore it */
 
   }
-  gras_assert_error(mismatch_error); /* expect this error */
+  xbt_assert_error(mismatch_error); /* expect this error */
   VERB3("Register version %d of message '%s' (payload: %s).", 
        version, name, gras_datadesc_get_name(payload));    
 
-  msgtype = gras_new(s_gras_msgtype_t,1);
+  msgtype = xbt_new(s_gras_msgtype_t,1);
   msgtype->name = (namev == name ? strdup(name) : namev);
   msgtype->name_len = strlen(namev);
   msgtype->version = version;
   msgtype->ctn_type = payload;
 
-  gras_set_add(_gras_msgtype_set, (gras_set_elm_t)msgtype,
+  xbt_set_add(_gras_msgtype_set, (xbt_set_elm_t)msgtype,
               &gras_msgtype_free);
 }
 
-/**
- * gras_msgtype_by_name:
- *
- * Retrieve a datatype description from its name
- */
+/** @brief retrive an existing message type from its name. */
 gras_msgtype_t gras_msgtype_by_name (const char *name) {
   return gras_msgtype_by_namev(name,0);
 }
-/**
- * gras_msgtype_by_namev:
- *
- * Retrieve a datatype description from its name and version
- */
+
+/** @brief retrive an existing message type from its name and version. */
 gras_msgtype_t gras_msgtype_by_namev(const char      *name,
                                     short int        version) {
   gras_msgtype_t res;
 
-  gras_error_t errcode;
+  xbt_error_t errcode;
   char *namev = make_namev(name,version); 
 
-  errcode = gras_set_get_by_name(_gras_msgtype_set, namev,
-                                (gras_set_elm_t*)&res);
+  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);
   if (name != namev) 
-    gras_free(namev);
+    free(namev);
   
   return res;
 }
 
-/**
- * gras_msg_send:
- *
- * Send the given message on the given socket 
- */
-gras_error_t
+/** \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) {
 
-  gras_error_t errcode;
+  xbt_error_t errcode;
   static gras_datadesc_type_t string_type=NULL;
 
   if (!msgtype)
@@ -196,7 +212,7 @@ gras_msg_send(gras_socket_t   sock,
 
   if (!string_type) {
     string_type = gras_datadesc_by_name("string");
-    gras_assert(string_type);
+    xbt_assert(string_type);
   }
 
   DEBUG3("send '%s' to %s:%d", msgtype->name, 
@@ -209,18 +225,16 @@ gras_msg_send(gras_socket_t   sock,
 
   return no_error;
 }
-/**
- * gras_msg_recv:
- *
+/*
  * receive the next message on the given socket.  
  */
-gras_error_t
+xbt_error_t
 gras_msg_recv(gras_socket_t    sock,
              gras_msgtype_t  *msgtype,
              void           **payload,
              int             *payload_size) {
 
-  gras_error_t errcode;
+  xbt_error_t errcode;
   static gras_datadesc_type_t string_type=NULL;
   char header[6];
   int cpt;
@@ -229,13 +243,13 @@ gras_msg_recv(gras_socket_t    sock,
 
   if (!string_type) {
     string_type=gras_datadesc_by_name("string");
-    gras_assert(string_type);
+    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");
+      RAISE2(mismatch_error,"Incoming bytes do not look like a GRAS message (header='%.4s' not '%.4s')",header,GRAS_header);
   if (header[4] != GRAS_header[4]) 
     RAISE2(mismatch_error,"GRAS protocol mismatch (got %d, use %d)",
           (int)header[4], (int)GRAS_header[4]);
@@ -244,38 +258,38 @@ gras_msg_recv(gras_socket_t    sock,
         (int)header[4],gras_datadesc_arch_name(r_arch));
 
   TRY(gras_datadesc_recv(sock, string_type, r_arch, &msg_name));
-  errcode = gras_set_get_by_name(_gras_msgtype_set,
-                                msg_name,(gras_set_elm_t*)msgtype);
+  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'",
-          gras_error_name(errcode),msg_name);
+          xbt_error_name(errcode),msg_name);
   /* FIXME: Survive unknown messages */
-  gras_free(msg_name);
+  free(msg_name);
 
   *payload_size=gras_datadesc_size((*msgtype)->ctn_type);
-  gras_assert2(*payload_size > 0,
+  xbt_assert2(*payload_size > 0,
               "%s %s",
               "Dynamic array as payload is forbided for now (FIXME?).",
               "Reference to dynamic array is allowed.");
-  *payload = gras_malloc(*payload_size);
+  *payload = xbt_malloc(*payload_size);
   TRY(gras_datadesc_recv(sock, (*msgtype)->ctn_type, r_arch, *payload));
 
   return no_error;
 }
 
-/**
- * gras_msg_wait:
- * @timeout: How long should we wait for this message.
- * @id: id of awaited msg
- * @Returns: the error code (or no_error).
+/** \brief Waits for a message to come in over a given socket. 
  *
- * 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 MsgHandle().
+ * and used by subsequent call to this function or gras_msg_handle().
  */
-gras_error_t
+xbt_error_t
 gras_msg_wait(double           timeout,    
              gras_msgtype_t   msgt_want,
              gras_socket_t   *expeditor,
@@ -284,11 +298,11 @@ gras_msg_wait(double           timeout,
   gras_msgtype_t msgt_got;
   void *payload_got;
   int payload_size_got;
-  gras_error_t errcode;
+  xbt_error_t errcode;
   double start, now;
-  gras_procdata_t *pd=gras_procdata_get();
+  gras_msg_procdata_t pd=(gras_msg_procdata_t)gras_libdata_get("gras_msg");
   int cpt;
-  gras_msg_t msg;
+  s_gras_msg_t msg;
   
   *expeditor = NULL;
   payload_got = NULL;
@@ -301,12 +315,12 @@ gras_msg_wait(double           timeout,
 
   start = now = gras_os_time();
 
-  gras_dynar_foreach(pd->msg_queue,cpt,msg){
+  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);
-      gras_free(msg.payload);
-      gras_dynar_cursor_rm(pd->msg_queue, &cpt);
+      free(msg.payload);
+      xbt_dynar_cursor_rm(pd->msg_queue, &cpt);
       VERB0("The waited message was queued");
       return no_error;
     }
@@ -317,7 +331,7 @@ gras_msg_wait(double           timeout,
     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);
-      gras_free(payload_got);
+      free(payload_got);
       VERB0("Got waited message");
       return no_error;
     }
@@ -327,7 +341,7 @@ gras_msg_wait(double           timeout,
     msg.type      =  msgt_got;
     msg.payload   =  payload;
     msg.payload_size = payload_size_got;
-    gras_dynar_push(pd->msg_queue,&msg);
+    xbt_dynar_push(pd->msg_queue,&msg);
     
     now=gras_os_time();
     if (now - start + 0.001 < timeout) {
@@ -338,48 +352,87 @@ gras_msg_wait(double           timeout,
   RAISE_IMPOSSIBLE;
 }
 
-/**
- * gras_msg_handle:
- * @timeOut: How long to wait for incoming messages
- * @Returns: the error code (or no_error).
+/** @brief Handle an incomming message or timer (or wait up to \a timeOut seconds)
+ *
+ * @param timeOut: How long to wait for incoming messages (in seconds)
+ * @return the error code (or no_error).
  *
- * Waits up to #timeOut# seconds to see if a message comes in; if so, calls the
- * registered listener for that message (see RegisterCallback()).
+ * Messages are passed to the callbacks.
  */
-gras_error_t 
+xbt_error_t 
 gras_msg_handle(double timeOut) {
   
-  gras_error_t    errcode;
+  double          untiltimer;
+   
+  xbt_error_t    errcode;
   int             cpt;
 
-  gras_msg_t      msg;
+  s_gras_msg_t    msg;
   gras_socket_t   expeditor;
   void           *payload=NULL;
   int             payload_size;
   gras_msgtype_t  msgtype;
 
-  gras_procdata_t*pd=gras_procdata_get();
-  gras_cblist_t  *list;
-  gras_cb_t       cb;
-
-
+  gras_msg_procdata_t pd=(gras_msg_procdata_t)gras_libdata_get("gras_msg");
+  gras_cblist_t  *list=NULL;
+  gras_msg_cb_t       cb;
+   
+  int timerexpected;
 
   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 no_error;
+  }
+  if (untiltimer != -1.0) {
+     timerexpected = 1;
+     timeOut = MIN(timeOut, untiltimer);
+  } else {
+     timerexpected = 0;
+  }
+   
   /* get a message (from the queue or from the net) */
-  if (gras_dynar_length(pd->msg_queue)) {
-    gras_dynar_shift(pd->msg_queue,&msg);
+  if (xbt_dynar_length(pd->msg_queue)) {
+    DEBUG0("Get a message from the queue");
+    xbt_dynar_shift(pd->msg_queue,&msg);
     expeditor = msg.expeditor;
     msgtype   = msg.type;
     payload   = msg.payload;
-    
+    errcode   = no_error;
   } else {
-    TRY(gras_trp_select(timeOut, &expeditor));
-    TRY(gras_msg_recv(expeditor, &msgtype, &payload, &payload_size));
+    errcode = gras_trp_select(timeOut, &expeditor);
+    if (errcode != no_error && errcode != timeout_error)
+       return errcode;
+    if (errcode != timeout_error)
+       TRY(gras_msg_recv(expeditor, &msgtype, &payload, &payload_size));
+  }
+
+  if (errcode == timeout_error ) {
+     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) {
+          return no_error;
+       } else {
+          xbt_assert1(untiltimer>0, "Negative timer (%f). I'm puzzeled", untiltimer);
+          ERROR1("No timer elapsed, in contrary to expectations (next in %f sec)",
+                 untiltimer);
+          return timeout_error;
+       }
+       
+     } else {
+       /* select timeouted, and no timer elapsed. Nothing to do */
+       return timeout_error;
+     }
+     
   }
-      
-  /* handle it */
-  gras_dynar_foreach(pd->cbl_list,cpt,list) {
+   
+  /* 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;
     } else {
@@ -393,12 +446,12 @@ gras_msg_handle(double timeOut) {
     return no_error;
   }
   
-  gras_dynar_foreach(list->cbs,cpt,cb) { 
-    INFO3("Use the callback #%d (@%p) for incomming msg %s",
-         cpt+1,cb,msgtype->name);
+  xbt_dynar_foreach(list->cbs,cpt,cb) { 
+    VERB3("Use the callback #%d (@%p) for incomming msg %s",
+          cpt+1,cb,msgtype->name);
     if ((*cb)(expeditor,payload)) {
       /* cb handled the message */
-      gras_free(payload);
+      free(payload);
       return no_error;
     }
   }
@@ -412,22 +465,28 @@ void
 gras_cbl_free(void *data){
   gras_cblist_t *list=*(void**)data;
   if (list) {
-    gras_dynar_free(&( list->cbs ));
-    gras_free(list);
+    xbt_dynar_free(&( list->cbs ));
+    free(list);
   }
 }
 
+/** \brief Bind the given callback to the given message type 
+ *
+ * Several callbacks can be attached to a given message type. The lastly added one will get the message first, and 
+ * if it returns false, the message will be passed to the second one. 
+ * And so on until one of the callbacks accepts the message.
+ */
 void
 gras_cb_register(gras_msgtype_t msgtype,
-                gras_cb_t cb) {
-  gras_procdata_t *pd=gras_procdata_get();
+                gras_msg_cb_t cb) {
+  gras_msg_procdata_t pd=(gras_msg_procdata_t)gras_libdata_get("gras_msg");
   gras_cblist_t *list=NULL;
   int cpt;
 
   DEBUG2("Register %p as callback to %s",cb,msgtype->name);
 
   /* search the list of cb for this message on this host (creating if NULL) */
-  gras_dynar_foreach(pd->cbl_list,cpt,list) {
+  xbt_dynar_foreach(pd->cbl_list,cpt,list) {
     if (list->id == msgtype->code) {
       break;
     } else {
@@ -436,28 +495,29 @@ gras_cb_register(gras_msgtype_t msgtype,
   }
   if (!list) {
     /* First cb? Create room */
-    list = gras_new(gras_cblist_t,1);
+    list = xbt_new(gras_cblist_t,1);
     list->id = msgtype->code;
-    list->cbs = gras_dynar_new(sizeof(gras_cb_t), NULL);
-    gras_dynar_push(pd->cbl_list,&list);
+    list->cbs = xbt_dynar_new(sizeof(gras_msg_cb_t), NULL);
+    xbt_dynar_push(pd->cbl_list,&list);
   }
 
   /* Insert the new one into the set */
-  gras_dynar_insert_at(list->cbs,0,&cb);
+  xbt_dynar_insert_at(list->cbs,0,&cb);
 }
 
+/** \brief Unbind the given callback from the given message type */
 void
 gras_cb_unregister(gras_msgtype_t msgtype,
-                  gras_cb_t cb) {
+                  gras_msg_cb_t cb) {
 
-  gras_procdata_t *pd=gras_procdata_get();
+  gras_msg_procdata_t pd=(gras_msg_procdata_t)gras_libdata_get("gras_msg");
   gras_cblist_t *list;
-  gras_cb_t cb_cpt;
+  gras_msg_cb_t cb_cpt;
   int cpt;
   int found = 0;
 
   /* search the list of cb for this message on this host */
-  gras_dynar_foreach(pd->cbl_list,cpt,list) {
+  xbt_dynar_foreach(pd->cbl_list,cpt,list) {
     if (list->id == msgtype->code) {
       break;
     } else {
@@ -467,9 +527,9 @@ gras_cb_unregister(gras_msgtype_t msgtype,
 
   /* Remove it from the set */
   if (list) {
-    gras_dynar_foreach(list->cbs,cpt,cb_cpt) {
+    xbt_dynar_foreach(list->cbs,cpt,cb_cpt) {
       if (cb == cb_cpt) {
-       gras_dynar_cursor_rm(list->cbs, &cpt);
+       xbt_dynar_cursor_rm(list->cbs, &cpt);
        found = 1;
       }
     }