Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Fix race condition during the creation of the listeners processes
[simgrid.git] / 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_thread_t listener;
26 } s_gras_msg_listener_t;
27
28 static void listener_function(void *p)
29 {
30   gras_msg_listener_t me = (gras_msg_listener_t) p;
31   gras_msg_t msg;
32   xbt_ex_t e;
33   gras_msgtype_t msg_wakeup_listener_t =
34     gras_msgtype_by_name("_wakeup_listener");
35   DEBUG0("I'm the listener");
36
37   /* get a free socket for the receiving part of the listener */
38   me->wakeup_sock_listener_side = gras_socket_server_range(5000, 6000, -1, 0);
39
40
41   /* Main loop */
42   while (1) {
43     msg = gras_msg_recv_any();
44     if (msg->type != msg_wakeup_listener_t) {
45       VERB1("Got a '%s' message. Queue it for handling by main thread",
46           gras_msgtype_get_name(msg->type));
47       xbt_queue_push(me->incomming_messages, msg);
48     } else {
49       char got = *(char *) msg->payl;
50       if (got == '1') {
51         VERB0("Asked to get awake");
52         free(msg->payl);
53         free(msg);
54       } else {
55         VERB0("Asked to die");
56         //gras_socket_close(me->wakeup_sock_listener_side);
57         free(msg->payl);
58         free(msg);
59         return;
60       }
61     }
62     /* empty the list of sockets to trash */
63     TRY {
64       while (1) {
65         int sock;
66         xbt_queue_shift_timed(me->socks_to_close, &sock, 0);
67         if (tcp_close(sock) < 0) {
68 #ifdef _XBT_WIN32
69           WARN2("error while closing tcp socket %d: %d\n", sock, sock_errno);
70 #else
71           WARN3("error while closing tcp socket %d: %d (%s)\n",
72                 sock, sock_errno, sock_errstr(sock_errno));
73 #endif
74         }
75       }
76     }
77     CATCH(e) {
78       if (e.category != timeout_error)
79         RETHROW;
80       xbt_ex_free(e);
81     }
82   }
83 }
84
85 gras_msg_listener_t gras_msg_listener_launch(xbt_queue_t msg_received)
86 {
87   gras_msg_listener_t arg = xbt_new0(s_gras_msg_listener_t, 1);
88
89   VERB0("Launch listener");
90   arg->incomming_messages = msg_received;
91   arg->socks_to_close = xbt_queue_new(0, sizeof(int));
92
93
94   /* declare the message used to awake the listener from the master */
95   gras_msgtype_declare("_wakeup_listener", gras_datadesc_by_name("char"));
96
97   /* actually start the thread */
98   arg->listener = xbt_thread_create("listener", listener_function, arg,1/*joinable*/);
99   gras_os_sleep(0);             /* give the listener a chance to initialize before we connect to its socket */
100
101   /* We do an active wait until the listener had created the socket (FIXME: ugly hack ) */
102   while(arg->wakeup_sock_listener_side == NULL);
103   
104   /* Connect the other part of the socket */
105   arg->wakeup_sock_master_side =
106     gras_socket_client(gras_os_myname(),
107                        gras_socket_my_port(arg->wakeup_sock_listener_side));
108   return arg;
109 }
110
111 #include "gras/Virtu/virtu_private.h"   /* procdata_t content */
112 void gras_msg_listener_shutdown()
113 {
114   gras_procdata_t *pd = gras_procdata_get();
115   char kill = '0';
116   DEBUG0("Listener quit");
117
118   if (pd->listener)
119     gras_msg_send(pd->listener->wakeup_sock_master_side, "_wakeup_listener",
120           &kill);
121
122   xbt_thread_join(pd->listener->listener);
123
124   //  gras_socket_close(pd->listener->wakeup_sock_master_side); FIXME: uncommenting this leads to deadlock at terminaison
125   xbt_queue_free(&pd->listener->incomming_messages);
126   xbt_queue_free(&pd->listener->socks_to_close);
127   xbt_free(pd->listener);
128 }
129
130 void gras_msg_listener_awake()
131 {
132   gras_procdata_t *pd;
133   char c = '1';
134
135   DEBUG0("Awaking the listener");
136   pd = gras_procdata_get();
137   if (pd->listener) {
138     gras_msg_send(pd->listener->wakeup_sock_master_side, "_wakeup_listener",
139                   &c);
140   }
141 }
142
143 void gras_msg_listener_close_socket(int sd)
144 {
145   gras_procdata_t *pd = gras_procdata_get();
146   if (pd->listener) {
147     xbt_queue_push(pd->listener->socks_to_close, &sd);
148     gras_msg_listener_awake();
149   } else {
150     /* do it myself */
151     tcp_close(sd);
152   }
153 }