1 /* Thread in charge of listening the network and queuing incoming messages */
3 /* Copyright (c) 2007, 2008, 2009, 2010. The SimGrid Team.
4 * All rights reserved. */
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. */
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");
15 #include "xbt/queue.h"
16 #include "xbt/synchro.h"
18 #include "gras/Transport/transport_interface.h" /* gras_select */
20 typedef struct s_gras_msg_listener_ {
21 xbt_thread_t listener; /* keep this first, gras_socket_im_the_server() does funky transtyping in sg_msg.c */
22 xbt_queue_t incoming_messages; /* messages received from the wire and still to be used by master */
23 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 */
24 xbt_socket_t wakeup_sock_listener_side;
25 xbt_socket_t wakeup_sock_master_side;
26 int port; /* The port on which the listener opened the command socket */
27 xbt_mutex_t init_mutex; /* both this mutex and condition are used at initialization to make sure that */
28 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) */
29 } s_gras_msg_listener_t;
31 #include "gras/Virtu/virtu_private.h" /* gras_procdata_t */
32 static void listener_function(void *p)
34 gras_msg_listener_t me = (gras_msg_listener_t) p;
37 gras_msgtype_t msg_wakeup_listener_t =
38 gras_msgtype_by_name("_wakeup_listener");
39 XBT_DEBUG("I'm the listener");
41 /* get a free socket for the receiving part of the listener */
42 me->wakeup_sock_listener_side =NULL;
43 for (me->port = 5000; me->port < 6000; me->port++) {
45 me->wakeup_sock_listener_side = gras_socket_server_ext(me->port, -1, 0);
52 if (me->wakeup_sock_listener_side)
56 /* wake up the launcher */
57 xbt_mutex_acquire(me->init_mutex);
58 xbt_cond_signal(me->init_cond);
59 xbt_mutex_release(me->init_mutex);
63 msg = gras_msg_recv_any();
64 if (msg->type != msg_wakeup_listener_t) {
65 /* Cannot display who sent this since in SG, xbt_socket_peer_* wont work:
66 I'm not the user process but I'm just the listener. Too bad */
67 XBT_VERB("Got a '%s' message (%s) from sock %p. Queue it for handling by main thread",
68 gras_msgtype_get_name(msg->type),e_gras_msg_kind_names[msg->kind],msg->expe);
69 xbt_queue_push(me->incoming_messages, msg);
71 char got = *(char *) msg->payl;
73 XBT_VERB("Asked to get awake");
77 XBT_VERB("Asked to die");
78 //gras_socket_close(me->wakeup_sock_listener_side);
84 /* empty the list of sockets to trash */
88 xbt_queue_shift_timed(me->socks_to_close, &sock, 0);
89 if (tcp_close(sock) < 0) {
91 XBT_WARN("error while closing tcp socket %d: %d\n", sock,
94 XBT_WARN("error while closing tcp socket %d: %d (%s)\n",
95 sock, sock_errno, sock_errstr(sock_errno));
101 if (e.category != timeout_error)
108 gras_msg_listener_t gras_msg_listener_launch(xbt_queue_t msg_received)
110 gras_msg_listener_t arg = xbt_new0(s_gras_msg_listener_t, 1);
112 XBT_VERB("Launch listener");
113 arg->incoming_messages = msg_received;
114 arg->socks_to_close = xbt_queue_new(0, sizeof(int));
115 arg->init_mutex = xbt_mutex_init();
116 arg->init_cond = xbt_cond_init();
118 /* declare the message used to awake the listener from the master */
119 gras_msgtype_declare("_wakeup_listener", xbt_datadesc_by_name("char"));
121 /* actually start the thread, and */
122 /* wait for the listener to initialize before we connect to its socket */
123 xbt_mutex_acquire(arg->init_mutex);
125 xbt_thread_create("listener", listener_function, arg,
127 xbt_cond_wait(arg->init_cond, arg->init_mutex);
128 xbt_mutex_release(arg->init_mutex);
129 xbt_cond_destroy(arg->init_cond);
130 xbt_mutex_destroy(arg->init_mutex);
132 /* Connect the other part of the socket */
133 arg->wakeup_sock_master_side =
134 gras_socket_client(gras_os_myname(),
139 #include "gras/Virtu/virtu_private.h" /* procdata_t content */
140 void gras_msg_listener_shutdown()
142 gras_procdata_t *pd = gras_procdata_get();
144 XBT_DEBUG("Listener quit");
147 gras_msg_send(pd->listener->wakeup_sock_master_side,
148 "_wakeup_listener", &kill);
150 xbt_thread_join(pd->listener->listener);
152 // gras_socket_close(pd->listener->wakeup_sock_master_side); FIXME: uncommenting this leads to deadlock at terminaison
153 xbt_queue_free(&pd->listener->incoming_messages);
154 xbt_queue_free(&pd->listener->socks_to_close);
155 xbt_free(pd->listener);
158 void gras_msg_listener_awake()
163 XBT_DEBUG("Awaking the listener");
164 pd = gras_procdata_get();
166 gras_msg_send(pd->listener->wakeup_sock_master_side,
167 "_wakeup_listener", &c);
171 void gras_msg_listener_close_socket(int sd)
173 gras_procdata_t *pd = gras_procdata_get();
175 xbt_queue_push(pd->listener->socks_to_close, &sd);
176 gras_msg_listener_awake();