X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/2940fe68f1f180b7d8cca1e80120f0d86c480878..1038b1a8e8bcb6f17417a1027a6c59438d248555:/src/gras/Msg/gras_msg_listener.c diff --git a/src/gras/Msg/gras_msg_listener.c b/src/gras/Msg/gras_msg_listener.c index c5dcf13116..3540090d04 100644 --- a/src/gras/Msg/gras_msg_listener.c +++ b/src/gras/Msg/gras_msg_listener.c @@ -1,15 +1,15 @@ -/* $Id$ */ - /* Thread in charge of listening the network and queuing incoming messages */ -/* Copyright (c) 2007 Martin Quinson. All rights reserved. */ +/* Copyright (c) 2007, 2008, 2009, 2010. The SimGrid Team. + * 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. */ #include "gras/Msg/msg_private.h" #include "gras/Transport/transport_private.h" -XBT_LOG_NEW_DEFAULT_SUBCATEGORY(gras_msg_read,gras_msg,"Message reader thread"); +XBT_LOG_NEW_DEFAULT_SUBCATEGORY(gras_msg_read, gras_msg, + "Message reader thread"); #include "xbt/ex.h" #include "xbt/queue.h" @@ -18,98 +18,136 @@ XBT_LOG_NEW_DEFAULT_SUBCATEGORY(gras_msg_read,gras_msg,"Message reader thread"); #include "gras/Transport/transport_interface.h" /* gras_select */ typedef struct s_gras_msg_listener_ { - xbt_queue_t incomming_messages; - xbt_queue_t socks_to_close; /* let the listener close the sockets, since it may be selecting on them. Darwin don't like this trick */ + xbt_queue_t incomming_messages; /* messages received from the wire and still to be used by master */ + xbt_queue_t socks_to_close; /* let the listener close the sockets, since it may be selecting on them. Darwin don't like this trick */ gras_socket_t wakeup_sock_listener_side; gras_socket_t wakeup_sock_master_side; xbt_thread_t listener; } s_gras_msg_listener_t; -static void listener_function(void *p) { - gras_msg_listener_t me = (gras_msg_listener_t)p; - s_gras_msg_t msg; +static void listener_function(void *p) +{ + gras_msg_listener_t me = (gras_msg_listener_t) p; + gras_msg_t msg; xbt_ex_t e; - gras_msgtype_t msg_wakeup_listener_t = gras_msgtype_by_name("_wakeup_listener"); + gras_msgtype_t msg_wakeup_listener_t = + gras_msgtype_by_name("_wakeup_listener"); DEBUG0("I'm the listener"); + + /* get a free socket for the receiving part of the listener */ + me->wakeup_sock_listener_side = gras_socket_server_range(5000, 6000, -1, 0); + + + /* Main loop */ while (1) { - DEBUG0("Selecting"); - msg.expe = gras_trp_select(-1); - DEBUG0("Select returned something"); - gras_msg_recv(msg.expe, &msg); - if (msg.type!=msg_wakeup_listener_t) - xbt_queue_push(me->incomming_messages, &msg); - else { - DEBUG0("Asked to get awake"); + msg = gras_msg_recv_any(); + if (msg->type != msg_wakeup_listener_t) { + VERB1("Got a '%s' message. Queue it for handling by main thread", + gras_msgtype_get_name(msg->type)); + xbt_queue_push(me->incomming_messages, msg); + } else { + char got = *(char *) msg->payl; + if (got == '1') { + VERB0("Asked to get awake"); + free(msg->payl); + free(msg); + } else { + VERB0("Asked to die"); + //gras_socket_close(me->wakeup_sock_listener_side); + free(msg->payl); + free(msg); + return; + } + } + /* empty the list of sockets to trash */ + TRY { + while (1) { + int sock; + xbt_queue_shift_timed(me->socks_to_close, &sock, 0); + if (tcp_close(sock) < 0) { +#ifdef _XBT_WIN32 + WARN2("error while closing tcp socket %d: %d\n", sock, sock_errno); +#else + WARN3("error while closing tcp socket %d: %d (%s)\n", + sock, sock_errno, sock_errstr(sock_errno)); +#endif + } + } + } + CATCH(e) { + if (e.category != timeout_error) + RETHROW; + xbt_ex_free(e); } - /* empty the list of sockets to trash */ - TRY { - while (1) { - int sock; - xbt_queue_shift_timed(me->socks_to_close,&sock,0); - if(tcp_close(sock) < 0) { - WARN3("error while closing tcp socket %d: %d (%s)\n", - sock, sock_errno, sock_errstr(sock_errno)); - } - } - } CATCH(e) { - if (e.category != timeout_error) - RETHROW; - xbt_ex_free(e); - } } } -gras_msg_listener_t -gras_msg_listener_launch(xbt_queue_t msg_exchange){ - gras_msg_listener_t arg = xbt_new0(s_gras_msg_listener_t,1); - int my_port; +gras_msg_listener_t gras_msg_listener_launch(xbt_queue_t msg_received) +{ + gras_msg_listener_t arg = xbt_new0(s_gras_msg_listener_t, 1); - DEBUG0("Launch listener"); - arg->incomming_messages = msg_exchange; - arg->socks_to_close = xbt_queue_new(0,sizeof(int)); + VERB0("Launch listener"); + arg->incomming_messages = msg_received; + arg->socks_to_close = xbt_queue_new(0, sizeof(int)); - /* get a free socket for the receiving part of the listener, taking care that it does not get saved as "myport" number */ - my_port = ((gras_trp_procdata_t) gras_libdata_by_id(gras_trp_libdata_id))->myport; - arg->wakeup_sock_listener_side=gras_socket_server_range(5000,6000,-1,0); - ((gras_trp_procdata_t) gras_libdata_by_id(gras_trp_libdata_id))->myport = my_port; - /* Connect the other part of the socket */ - arg->wakeup_sock_master_side=gras_socket_client(gras_os_myname(),gras_socket_my_port(arg->wakeup_sock_listener_side)); /* declare the message used to awake the listener from the master */ - gras_msgtype_declare("_wakeup_listener",gras_datadesc_by_name("char")); + gras_msgtype_declare("_wakeup_listener", gras_datadesc_by_name("char")); /* actually start the thread */ - arg->listener = xbt_thread_create("listener",listener_function,arg); - gras_os_sleep(0); /* TODO: useless? give the listener a chance to initialize even if the main is empty and we cancel it right afterward */ + arg->listener = xbt_thread_create("listener", listener_function, arg,1/*joinable*/); + gras_os_sleep(0); /* give the listener a chance to initialize before we connect to its socket */ + + /* We do an active wait until the listener had created the socket (FIXME: ugly hack ) */ + while(arg->wakeup_sock_listener_side == NULL); + + /* Connect the other part of the socket */ + arg->wakeup_sock_master_side = + gras_socket_client(gras_os_myname(), + gras_socket_my_port(arg->wakeup_sock_listener_side)); return arg; } -void gras_msg_listener_shutdown(gras_msg_listener_t l) { +#include "gras/Virtu/virtu_private.h" /* procdata_t content */ +void gras_msg_listener_shutdown() +{ + gras_procdata_t *pd = gras_procdata_get(); + char kill = '0'; DEBUG0("Listener quit"); - xbt_thread_cancel(l->listener); - xbt_queue_free(&l->incomming_messages); - xbt_queue_free(&l->socks_to_close); - xbt_free(l); + + if (pd->listener) + gras_msg_send(pd->listener->wakeup_sock_master_side, "_wakeup_listener", + &kill); + + xbt_thread_join(pd->listener->listener); + + // gras_socket_close(pd->listener->wakeup_sock_master_side); FIXME: uncommenting this leads to deadlock at terminaison + xbt_queue_free(&pd->listener->incomming_messages); + xbt_queue_free(&pd->listener->socks_to_close); + xbt_free(pd->listener); } -#include "gras/Virtu/virtu_private.h" /* procdata_t content */ -void gras_msg_listener_awake(){ - gras_procdata_t *pd; - char c = '1'; +void gras_msg_listener_awake() +{ + gras_procdata_t *pd; + char c = '1'; - DEBUG0("Awaking the listener"); - pd = gras_procdata_get(); - if (pd->listener) { - gras_msg_send(pd->listener->wakeup_sock_master_side,"_wakeup_listener",&c); - } + DEBUG0("Awaking the listener"); + pd = gras_procdata_get(); + if (pd->listener) { + gras_msg_send(pd->listener->wakeup_sock_master_side, "_wakeup_listener", + &c); + } } -void gras_msg_listener_close_socket(int sd) { - gras_procdata_t *pd = gras_procdata_get(); - if (pd->listener) { - xbt_queue_push(pd->listener->socks_to_close,&sd); - gras_msg_listener_awake(); - } else { - /* do it myself */ - tcp_close(sd); - } + +void gras_msg_listener_close_socket(int sd) +{ + gras_procdata_t *pd = gras_procdata_get(); + if (pd->listener) { + xbt_queue_push(pd->listener->socks_to_close, &sd); + gras_msg_listener_awake(); + } else { + /* do it myself */ + tcp_close(sd); + } }