src/gras/Msg/gras_msg_listener.c

   1 /* $Id$ */
   2
   3 /* Thread in charge of listening the network and queuing incoming messages  */
   4
   5 /* Copyright (c) 2007 Martin Quinson. All rights reserved.                  */
   6
   7 /* This program is free software; you can redistribute it and/or modify it
   8  * under the terms of the license (GNU LGPL) which comes with this package. */
   9
  10 #include "gras/Msg/msg_private.h"
  11 #include "gras/Transport/transport_private.h"
  12 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(gras_msg_read,gras_msg,"Message reader thread");
  13
  14 #include "xbt/ex.h"
  15 #include "xbt/queue.h"
  16 #include "xbt/synchro.h"
  17
  18 #include "gras/Transport/transport_interface.h" /* gras_select */
  19
  20 typedef struct s_gras_msg_listener_ {
  21   xbt_queue_t incomming_messages;
  22   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 */
  23   gras_socket_t wakeup_sock_listener_side;
  24   gras_socket_t wakeup_sock_master_side;
  25   xbt_thread_t listener;
  26 } s_gras_msg_listener_t;
  27
  28 static void listener_function(void *p) {
  29   gras_msg_listener_t me = (gras_msg_listener_t)p;
  30   s_gras_msg_t msg;
  31   xbt_ex_t e;
  32   gras_msgtype_t msg_wakeup_listener_t = gras_msgtype_by_name("_wakeup_listener");
  33   DEBUG0("I'm the listener");
  34   while (1) {
  35     DEBUG0("Selecting");
  36     msg.expe = gras_trp_select(-1);
  37     DEBUG0("Select returned something");
  38     gras_msg_recv(msg.expe, &msg);
  39     if (msg.type!=msg_wakeup_listener_t)
  40         xbt_queue_push(me->incomming_messages, &msg);
  41     else  {
  42        char got = *(char*)msg.payl;
  43        if (got == '1') {
  44           VERB0("Asked to get awake");
  45           free(msg.payl);
  46        } else {
  47           VERB0("Asked to die");
  48 //        gras_socket_close(me->wakeup_sock_listener_side);
  49           free(msg.payl);
  50           return ;
  51        }
  52     }
  53      /* empty the list of sockets to trash */
  54      TRY {
  55         while (1) {
  56            int sock;
  57            xbt_queue_shift_timed(me->socks_to_close,&sock,0);
  58            if(tcp_close(sock) < 0) {
  59               WARN3("error while closing tcp socket %d: %d (%s)\n",
  60                     sock, sock_errno, sock_errstr(sock_errno));
  61            }
  62         }
  63      } CATCH(e) {
  64         if (e.category != timeout_error)
  65           RETHROW;
  66         xbt_ex_free(e);
  67      }
  68   }
  69 }
  70
  71 gras_msg_listener_t
  72 gras_msg_listener_launch(xbt_queue_t msg_exchange){
  73   gras_msg_listener_t arg = xbt_new0(s_gras_msg_listener_t,1);
  74   int my_port;
  75
  76   DEBUG0("Launch listener");
  77   arg->incomming_messages = msg_exchange;
  78   arg->socks_to_close = xbt_queue_new(0,sizeof(int));
  79
  80   /* get a free socket for the receiving part of the listener, taking care that it does not get saved as "myport" number */
  81   my_port = ((gras_trp_procdata_t) gras_libdata_by_id(gras_trp_libdata_id))->myport;
  82   arg->wakeup_sock_listener_side=gras_socket_server_range(5000,6000,-1,0);
  83   ((gras_trp_procdata_t) gras_libdata_by_id(gras_trp_libdata_id))->myport = my_port;
  84   /* Connect the other part of the socket */
  85   arg->wakeup_sock_master_side=gras_socket_client(gras_os_myname(),gras_socket_my_port(arg->wakeup_sock_listener_side));
  86
  87   /* declare the message used to awake the listener from the master */
  88   gras_msgtype_declare("_wakeup_listener",gras_datadesc_by_name("char"));
  89
  90   /* actually start the thread */
  91   arg->listener =  xbt_thread_create("listener",listener_function,arg);
  92   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 */
  93   return arg;
  94 }
  95
  96 #include "gras/Virtu/virtu_private.h" /* procdata_t content */
  97 void gras_msg_listener_shutdown(gras_msg_listener_t l) {
  98    gras_procdata_t *pd = gras_procdata_get();
  99    char kill = '0';
 100   DEBUG0("Listener quit");
 101
 102
 103    if (pd->listener)
 104      gras_msg_send(pd->listener->wakeup_sock_master_side,"_wakeup_listener",&kill);
 105
 106    /* FIXME: thread_join is not implemented in SG (remove next conditional when fixed)
 107     * But I guess it's not a big deal since we're terminating the thread mainly to
 108     * make it free its OS locks on darwin.
 109     * darwin is definitly different from the neat & nice SG world */
 110    if (gras_if_RL())
 111      xbt_thread_join(pd->listener->listener);
 112
 113 //  gras_socket_close(pd->listener->wakeup_sock_master_side); FIXME: uncommenting this leads to deadlock at terminaison
 114   xbt_queue_free(&l->incomming_messages);
 115   xbt_queue_free(&l->socks_to_close);
 116   xbt_free(l);
 117 }
 118
 119 void gras_msg_listener_awake(){
 120         gras_procdata_t *pd;
 121         char c = '1';
 122
 123         DEBUG0("Awaking the listener");
 124         pd = gras_procdata_get();
 125         if (pd->listener) {
 126                 gras_msg_send(pd->listener->wakeup_sock_master_side,"_wakeup_listener",&c);
 127         }
 128 }
 129 void  gras_msg_listener_close_socket(int sd) {
 130    gras_procdata_t *pd = gras_procdata_get();
 131    if (pd->listener) {
 132       xbt_queue_push(pd->listener->socks_to_close,&sd);
 133       gras_msg_listener_awake();
 134    } else {
 135       /* do it myself */
 136       tcp_close(sd);
 137    }
 138 }