src/gras/Msg/gras_msg_listener.c

   1 /* Thread in charge of listening the network and queuing incoming messages  */
   2
   3 /* Copyright (c) 2007, 2008, 2009, 2010. The SimGrid Team.
   4  * All rights reserved.                                                     */
   5
   6 /* This program is free software; you can redistribute it and/or modify it
   7  * under the terms of the license (GNU LGPL) which comes with this package. */
   8
   9 #include "gras/Msg/msg_private.h"
  10 #include "gras/Transport/transport_private.h"
  11 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(gras_msg_read, gras_msg,
  12                                 "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; /* messages received from the wire and still to be used by master */
  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_mutex_t init_mutex; /* both this mutex and condition are used at initialization to make sure that*/
  26   xbt_cond_t init_cond;   /* the main thread speaks to the listener only once it is started (FIXME: It would be easier using a semaphore, if only semaphores were in xbt_synchro) */
  27   xbt_thread_t listener;
  28 } s_gras_msg_listener_t;
  29
  30 static void listener_function(void *p)
  31 {
  32   gras_msg_listener_t me = (gras_msg_listener_t) p;
  33   gras_msg_t msg;
  34   xbt_ex_t e;
  35   gras_msgtype_t msg_wakeup_listener_t =
  36     gras_msgtype_by_name("_wakeup_listener");
  37   DEBUG0("I'm the listener");
  38
  39   /* get a free socket for the receiving part of the listener */
  40   me->wakeup_sock_listener_side = gras_socket_server_range(5000, 6000, -1, 0);
  41
  42   /* wake up the launcher */
  43   xbt_mutex_acquire(me->init_mutex);
  44   xbt_cond_signal(me->init_cond);
  45   xbt_mutex_release(me->init_mutex);
  46
  47   /* Main loop */
  48   while (1) {
  49     msg = gras_msg_recv_any();
  50     if (msg->type != msg_wakeup_listener_t) {
  51       VERB1("Got a '%s' message. Queue it for handling by main thread",
  52           gras_msgtype_get_name(msg->type));
  53       xbt_queue_push(me->incomming_messages, msg);
  54     } else {
  55       char got = *(char *) msg->payl;
  56       if (got == '1') {
  57         VERB0("Asked to get awake");
  58         free(msg->payl);
  59         free(msg);
  60       } else {
  61         VERB0("Asked to die");
  62         //gras_socket_close(me->wakeup_sock_listener_side);
  63         free(msg->payl);
  64         free(msg);
  65         return;
  66       }
  67     }
  68     /* empty the list of sockets to trash */
  69     TRY {
  70       while (1) {
  71         int sock;
  72         xbt_queue_shift_timed(me->socks_to_close, &sock, 0);
  73         if (tcp_close(sock) < 0) {
  74 #ifdef _XBT_WIN32
  75           WARN2("error while closing tcp socket %d: %d\n", sock, sock_errno);
  76 #else
  77           WARN3("error while closing tcp socket %d: %d (%s)\n",
  78                 sock, sock_errno, sock_errstr(sock_errno));
  79 #endif
  80         }
  81       }
  82     }
  83     CATCH(e) {
  84       if (e.category != timeout_error)
  85         RETHROW;
  86       xbt_ex_free(e);
  87     }
  88   }
  89 }
  90
  91 gras_msg_listener_t gras_msg_listener_launch(xbt_queue_t msg_received)
  92 {
  93   gras_msg_listener_t arg = xbt_new0(s_gras_msg_listener_t, 1);
  94
  95   VERB0("Launch listener");
  96   arg->incomming_messages = msg_received;
  97   arg->socks_to_close = xbt_queue_new(0, sizeof(int));
  98   arg->init_mutex = xbt_mutex_init();
  99   arg->init_cond = xbt_cond_init();
 100
 101   /* declare the message used to awake the listener from the master */
 102   gras_msgtype_declare("_wakeup_listener", gras_datadesc_by_name("char"));
 103
 104   /* actually start the thread, and */
 105   /* wait for the listener to initialize before we connect to its socket */
 106   xbt_mutex_acquire(arg->init_mutex);
 107   arg->listener = xbt_thread_create("listener", listener_function, arg,1/*joinable*/);
 108   xbt_cond_wait(arg->init_cond, arg->init_mutex);
 109   xbt_mutex_release(arg->init_mutex);
 110   xbt_cond_destroy(arg->init_cond);
 111   xbt_mutex_destroy(arg->init_mutex);
 112
 113   /* Connect the other part of the socket */
 114   arg->wakeup_sock_master_side =
 115     gras_socket_client(gras_os_myname(),
 116                        gras_socket_my_port(arg->wakeup_sock_listener_side));
 117   return arg;
 118 }
 119
 120 #include "gras/Virtu/virtu_private.h"   /* procdata_t content */
 121 void gras_msg_listener_shutdown()
 122 {
 123   gras_procdata_t *pd = gras_procdata_get();
 124   char kill = '0';
 125   DEBUG0("Listener quit");
 126
 127   if (pd->listener)
 128     gras_msg_send(pd->listener->wakeup_sock_master_side, "_wakeup_listener",
 129           &kill);
 130
 131   xbt_thread_join(pd->listener->listener);
 132
 133   //  gras_socket_close(pd->listener->wakeup_sock_master_side); FIXME: uncommenting this leads to deadlock at terminaison
 134   xbt_queue_free(&pd->listener->incomming_messages);
 135   xbt_queue_free(&pd->listener->socks_to_close);
 136   xbt_free(pd->listener);
 137 }
 138
 139 void gras_msg_listener_awake()
 140 {
 141   gras_procdata_t *pd;
 142   char c = '1';
 143
 144   DEBUG0("Awaking the listener");
 145   pd = gras_procdata_get();
 146   if (pd->listener) {
 147     gras_msg_send(pd->listener->wakeup_sock_master_side, "_wakeup_listener",
 148                   &c);
 149   }
 150 }
 151
 152 void gras_msg_listener_close_socket(int sd)
 153 {
 154   gras_procdata_t *pd = gras_procdata_get();
 155   if (pd->listener) {
 156     xbt_queue_push(pd->listener->socks_to_close, &sd);
 157     gras_msg_listener_awake();
 158   } else {
 159     /* do it myself */
 160     tcp_close(sd);
 161   }
 162 }