#include "xbt/sysdep.h"
#include "xbt/peer.h"
#include "amok/peermanagement.h"
-#include "gras/Virtu/virtu_interface.h" /* libdata */
+
+#include "amok/amok_modinter.h" /* prototype of my module declaration */
+#include "gras/module.h" /* module mecanism */
XBT_LOG_NEW_DEFAULT_SUBCATEGORY(amok_pm,amok,"peer management");
-/* libdata management */
-static int amok_pm_libdata_id=-1;
+/* data management */
+int amok_pm_moddata_id=-1;
typedef struct {
- /* set headers */
- unsigned int ID;
- char *name;
- unsigned int name_len;
-
- /* payload */
- int done;
- xbt_dict_t groups;
-} s_amok_pm_libdata_t, *amok_pm_libdata_t;
-
-static void *amok_pm_libdata_new() {
- amok_pm_libdata_t res=xbt_new(s_amok_pm_libdata_t,1);
- res->name=xbt_strdup("amok_pm");
- res->name_len=0;
- res->done = 0;
- res->groups = xbt_dict_new();
- return res;
-}
-static void amok_pm_libdata_free(void *d) {
- amok_pm_libdata_t data=(amok_pm_libdata_t)d;
- free(data->name);
- xbt_dict_free(&data->groups);
- free (data);
-}
+ int done;
+ xbt_dict_t groups;
+} s_amok_pm_moddata_t, *amok_pm_moddata_t;
+
/* Message callbacks */
static int amok_pm_cb_kill(gras_msg_cb_ctx_t ctx,
void *payload_data) {
- amok_pm_libdata_t g=gras_libdata_by_id(amok_pm_libdata_id);
+ amok_pm_moddata_t g=gras_moddata_by_id(amok_pm_moddata_id);
g->done = 1;
return 1;
}
static int amok_pm_cb_killrpc(gras_msg_cb_ctx_t ctx,
void *payload_data) {
- amok_pm_libdata_t g=gras_libdata_by_id(amok_pm_libdata_id);
+ amok_pm_moddata_t g=gras_moddata_by_id(amok_pm_moddata_id);
g->done = 1;
gras_msg_rpcreturn(30,ctx,NULL);
return 1;
}
static int amok_pm_cb_get(gras_msg_cb_ctx_t ctx, void *payload) {
- amok_pm_libdata_t g=gras_libdata_by_id(amok_pm_libdata_id);
+ amok_pm_moddata_t g=gras_moddata_by_id(amok_pm_moddata_id);
char *name = *(void**)payload;
xbt_dynar_t res = xbt_dict_get(g->groups, name);
return 1;
}
static int amok_pm_cb_join(gras_msg_cb_ctx_t ctx, void *payload) {
- amok_pm_libdata_t g=gras_libdata_by_id(amok_pm_libdata_id);
+ amok_pm_moddata_t g=gras_moddata_by_id(amok_pm_moddata_id);
char *name = *(void**)payload;
xbt_dynar_t group = xbt_dict_get(g->groups, name);
return 1;
}
static int amok_pm_cb_leave(gras_msg_cb_ctx_t ctx, void *payload) {
- amok_pm_libdata_t g=gras_libdata_by_id(amok_pm_libdata_id);
+ amok_pm_moddata_t g=gras_moddata_by_id(amok_pm_moddata_id);
char *name = *(void**)payload;
xbt_dynar_t group = xbt_dict_get(g->groups, name);
return 1;
}
-
-/* Initialization stuff */
-static short amok_pm_used = 0;
-
-/** \brief Initialize the peer management module. Every process must run it before use */
-void amok_pm_init() {
- /* pure INIT part */
- if (! amok_pm_used) {
-
- /* dependencies */
- amok_base_init();
-
- /* module data on each process */
- amok_pm_libdata_id = gras_procdata_add("amok_pm",
- amok_pm_libdata_new,
- amok_pm_libdata_free);
-
- /* Datatype and message declarations */
- gras_msgtype_declare("amok_pm_kill",NULL);
- gras_msgtype_declare_rpc("amok_pm_killrpc",NULL,NULL);
-
- gras_msgtype_declare_rpc("amok_pm_get",
- gras_datadesc_by_name("string"),
- gras_datadesc_by_name("xbt_dynar_t"));
- gras_msgtype_declare_rpc("amok_pm_join",
- gras_datadesc_by_name("string"),
- NULL);
- gras_msgtype_declare_rpc("amok_pm_leave",
- gras_datadesc_by_name("string"),
- NULL);
-
- gras_msgtype_declare_rpc("amok_pm_shutdown",
- gras_datadesc_by_name("string"),
- NULL);
- }
- amok_pm_used++;
-
- /* JOIN part */
- gras_cb_register(gras_msgtype_by_name("amok_pm_kill"),
- &amok_pm_cb_kill);
- gras_cb_register(gras_msgtype_by_name("amok_pm_killrpc"),
- &amok_pm_cb_killrpc);
-
- gras_cb_register(gras_msgtype_by_name("amok_pm_get"),
- &amok_pm_cb_get);
- gras_cb_register(gras_msgtype_by_name("amok_pm_join"),
- &amok_pm_cb_join);
- gras_cb_register(gras_msgtype_by_name("amok_pm_leave"),
- &amok_pm_cb_leave);
- gras_cb_register(gras_msgtype_by_name("amok_pm_shutdown"),
- &amok_pm_cb_shutdown);
-}
-
-/** \brief Finalize the peer management module. Every process should run it after use */
-void amok_pm_exit() {
- /* pure EXIT part */
- amok_pm_used--;
-
- /* LEAVE part */
- gras_cb_unregister(gras_msgtype_by_name("amok_pm_kill"),
- &amok_pm_cb_kill);
- gras_cb_unregister(gras_msgtype_by_name("amok_pm_killrpc"),
- &amok_pm_cb_killrpc);
-
- gras_cb_unregister(gras_msgtype_by_name("amok_pm_get"),
- &amok_pm_cb_get);
- gras_cb_unregister(gras_msgtype_by_name("amok_pm_join"),
- &amok_pm_cb_join);
- gras_cb_unregister(gras_msgtype_by_name("amok_pm_leave"),
- &amok_pm_cb_leave);
-
- gras_cb_unregister(gras_msgtype_by_name("amok_pm_shutdown"),
- &amok_pm_cb_shutdown);
-}
-
-
/** \brief Enter the main loop of the program. It won't return until we get a kill message. */
void amok_pm_mainloop(double timeOut) {
- amok_pm_libdata_t g=gras_libdata_by_id(amok_pm_libdata_id);
+ amok_pm_moddata_t g=gras_moddata_by_id(amok_pm_moddata_id);
while (!g->done) {
gras_msg_handle(timeOut);
* The dynar elements are of type xbt_peer_t
*/
xbt_dynar_t amok_pm_group_new(const char *group_name) {
- amok_pm_libdata_t g;
+ amok_pm_moddata_t g;
xbt_dynar_t res = xbt_dynar_new(sizeof(xbt_peer_t),
xbt_peer_free_voidp);
- xbt_assert0(amok_pm_libdata_id != -1,"Run amok_pm_init first!");
- g=gras_libdata_by_id(amok_pm_libdata_id);
+ xbt_assert0(amok_pm_moddata_id != -1,"Run amok_pm_init first!");
+ g=gras_moddata_by_id(amok_pm_moddata_id);
+
+ INFO1("retrieved groups=%p",g->groups);
xbt_dict_set(g->groups,group_name,res,NULL); /*FIXME: leaking xbt_dynar_free_voidp);*/
VERB1("Group %s created",group_name);
/** \brief stops all members of the given local group */
void amok_pm_group_shutdown(const char *group_name) {
- amok_pm_libdata_t g=gras_libdata_by_id(amok_pm_libdata_id);
+ amok_pm_moddata_t g=gras_moddata_by_id(amok_pm_moddata_id);
xbt_dynar_t group = xbt_dict_get(g->groups, group_name);
int cpt;
gras_msg_rpccall(master,30,gras_msgtype_by_name("amok_pm_shutdown"),
&group_name,NULL);
}
+
+
+/* *
+ * *
+ * * Module management functions
+ * *
+ * */
+
+
+
+static void _amok_pm_init(void) {
+ /* no world-wide globals */
+ /* Datatype and message declarations */
+ gras_msgtype_declare("amok_pm_kill",NULL);
+ gras_msgtype_declare_rpc("amok_pm_killrpc",NULL,NULL);
+
+ gras_msgtype_declare_rpc("amok_pm_get",
+ gras_datadesc_by_name("string"),
+ gras_datadesc_by_name("xbt_dynar_t"));
+ gras_msgtype_declare_rpc("amok_pm_join",
+ gras_datadesc_by_name("string"),
+ NULL);
+ gras_msgtype_declare_rpc("amok_pm_leave",
+ gras_datadesc_by_name("string"),
+ NULL);
+
+ gras_msgtype_declare_rpc("amok_pm_shutdown",
+ gras_datadesc_by_name("string"),
+ NULL);
+}
+
+static void _amok_pm_join(void *p) {
+ /* moddata management */
+ amok_pm_moddata_t mod = (amok_pm_moddata_t)p;
+
+ mod->done = 0;
+ mod->groups = xbt_dict_new();
+ INFO1("groups=%p",mod->groups);
+
+ /* callbacks */
+ gras_cb_register(gras_msgtype_by_name("amok_pm_kill"),
+ &amok_pm_cb_kill);
+ gras_cb_register(gras_msgtype_by_name("amok_pm_killrpc"),
+ &amok_pm_cb_killrpc);
+
+ gras_cb_register(gras_msgtype_by_name("amok_pm_get"),
+ &amok_pm_cb_get);
+ gras_cb_register(gras_msgtype_by_name("amok_pm_join"),
+ &amok_pm_cb_join);
+ gras_cb_register(gras_msgtype_by_name("amok_pm_leave"),
+ &amok_pm_cb_leave);
+ gras_cb_register(gras_msgtype_by_name("amok_pm_shutdown"),
+ &amok_pm_cb_shutdown);
+}
+static void _amok_pm_exit(void) {
+ /* no world-wide globals */
+}
+static void _amok_pm_leave(void *p) {
+ /* moddata */
+ amok_pm_moddata_t mod = (amok_pm_moddata_t)p;
+
+ xbt_dict_free(&mod->groups);
+
+ /* callbacks */
+ gras_cb_unregister(gras_msgtype_by_name("amok_pm_kill"),
+ &amok_pm_cb_kill);
+ gras_cb_unregister(gras_msgtype_by_name("amok_pm_killrpc"),
+ &amok_pm_cb_killrpc);
+
+ gras_cb_unregister(gras_msgtype_by_name("amok_pm_get"),
+ &amok_pm_cb_get);
+ gras_cb_unregister(gras_msgtype_by_name("amok_pm_join"),
+ &amok_pm_cb_join);
+ gras_cb_unregister(gras_msgtype_by_name("amok_pm_leave"),
+ &amok_pm_cb_leave);
+
+ gras_cb_unregister(gras_msgtype_by_name("amok_pm_shutdown"),
+ &amok_pm_cb_shutdown);
+}
+
+void amok_pm_modulecreate() {
+ gras_module_add("amok_pm", sizeof(s_amok_pm_moddata_t), &amok_pm_moddata_id,
+ &_amok_pm_init,&_amok_pm_exit,&_amok_pm_join,&_amok_pm_leave);
+}
+
+
+
+/* *
+ * *
+ * * Old module functions (kept for compatibility)
+ * *
+ * */
+/** \brief Initialize the peer management module. Every process must run it before use */
+void amok_pm_init() {
+ gras_module_join("amok_pm");
+}
+
+/** \brief Finalize the peer management module. Every process should run it after use */
+void amok_pm_exit() {
+ gras_module_leave("amok_pm");
+}
