Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
738b911f4f38bb952fb7e3c44d1b6336d7b08a1e
[simgrid.git] / src / gras / Msg / rl_msg.c
1 /* messaging - Function related to messaging code specific to RL            */
2
3 /* Copyright (c) 2005, 2006, 2007, 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 "xbt/ex.h"
10 #include "xbt/datadesc.h"
11 #include "xbt/datadesc/datadesc_interface.h"
12 #include "xbt/socket.h"
13 #include "gras/Msg/msg_private.h"
14 #include "gras/Transport/transport_interface.h"
15
16 XBT_LOG_EXTERNAL_CATEGORY(gras_msg);
17 XBT_LOG_DEFAULT_CATEGORY(gras_msg);
18
19 void gras_msg_recv(xbt_socket_t sock, gras_msg_t msg);
20
21 gras_msg_t gras_msg_recv_any(void)
22 {
23   gras_msg_t msg = xbt_new0(s_gras_msg_t, 1);
24   msg->expe = gras_trp_select(-1);
25   XBT_DEBUG("Select returned something");
26   gras_msg_recv(msg->expe, msg);
27   return msg;
28 }
29
30 void gras_msg_send_ext(xbt_socket_t sock,
31                        e_gras_msg_kind_t kind,
32                        unsigned long int ID,
33                        gras_msgtype_t msgtype, void *payload)
34 {
35
36   static xbt_datadesc_type_t string_type = NULL;
37   static xbt_datadesc_type_t ulong_type = NULL;
38   char c_kind = (char) kind;
39
40   xbt_assert(msgtype, "Cannot send the NULL message");
41
42   if (!string_type) {
43     string_type = xbt_datadesc_by_name("string");
44     xbt_assert(string_type);
45   }
46   if (!ulong_type) {
47     ulong_type = xbt_datadesc_by_name("unsigned long int");
48     xbt_assert(ulong_type);
49   }
50
51   XBT_DEBUG("send '%s' to %s:%d", msgtype->name,
52          xbt_socket_peer_name(sock), xbt_socket_peer_port(sock));
53   xbt_trp_send(sock, _GRAS_header, 6, 1 /* stable */ );
54   xbt_trp_send(sock, &c_kind, 1, 1 /* stable */ );
55   switch (kind) {
56   case e_gras_msg_kind_oneway:
57     break;
58
59   case e_gras_msg_kind_rpccall:
60   case e_gras_msg_kind_rpcanswer:
61   case e_gras_msg_kind_rpcerror:
62     xbt_datadesc_send(sock, ulong_type, &ID);
63     break;
64
65   default:
66     THROWF(unknown_error, 0, "Unknown msg kind %d", (int)kind);
67   }
68
69   xbt_datadesc_send(sock, string_type, &msgtype->name);
70   if (kind == e_gras_msg_kind_rpcerror) {
71     /* error on remote host, carfull, payload is an exception */
72     xbt_datadesc_send(sock, xbt_datadesc_by_name("ex_t"), payload);
73   } else if (kind == e_gras_msg_kind_rpcanswer) {
74     if (msgtype->answer_type)
75       xbt_datadesc_send(sock, msgtype->answer_type, payload);
76   } else {
77     /* regular message */
78     if (msgtype->ctn_type)
79       xbt_datadesc_send(sock, msgtype->ctn_type, payload);
80   }
81
82   xbt_trp_flush(sock);
83 }
84
85 const char *hexa_str(unsigned char *data, int size, int downside);
86
87
88 /*
89  * receive the next message on the given socket.
90  */
91 void gras_msg_recv(xbt_socket_t sock, gras_msg_t msg)
92 {
93
94   xbt_ex_t e;
95   static xbt_datadesc_type_t string_type = NULL;
96   static xbt_datadesc_type_t ulong_type = NULL;
97   char header[6];
98   int cpt;
99   int r_arch;
100   char *msg_name = NULL;
101   char c_kind;
102
103   xbt_assert(!xbt_socket_is_meas(sock),
104               "Asked to receive a message on the measurement socket %p",
105               sock);
106   if (!string_type) {
107     string_type = xbt_datadesc_by_name("string");
108     xbt_assert(string_type);
109   }
110   if (!ulong_type) {
111     ulong_type = xbt_datadesc_by_name("unsigned long int");
112     xbt_assert(ulong_type);
113   }
114
115
116   TRY {
117     xbt_trp_recv(sock, header, 6);
118     xbt_trp_recv(sock, &c_kind, 1);
119     msg->kind = (e_gras_msg_kind_t) c_kind;
120   }
121   CATCH_ANONYMOUS {
122     RETHROWF
123         ("Exception caught while trying to get the message header: %s");
124   }
125
126   for (cpt = 0; cpt < 4; cpt++)
127     if (header[cpt] != _GRAS_header[cpt])
128       THROWF(mismatch_error, 0,
129              "Incoming bytes do not look like a GRAS message (header='%s'  not '%.4s')",
130              hexa_str((unsigned char *) header, 4, 0), _GRAS_header);
131   if (header[4] != _GRAS_header[4])
132     THROWF(mismatch_error, 0, "GRAS protocol mismatch (got %d, use %d)",
133            (int) header[4], (int) _GRAS_header[4]);
134   r_arch = (int) header[5];
135
136   switch (msg->kind) {
137   case e_gras_msg_kind_oneway:
138     break;
139
140   case e_gras_msg_kind_rpccall:
141   case e_gras_msg_kind_rpcanswer:
142   case e_gras_msg_kind_rpcerror:
143     xbt_datadesc_recv(sock, ulong_type, r_arch, &msg->ID);
144     break;
145
146   default:
147     THROW_IMPOSSIBLE;
148   }
149
150   xbt_datadesc_recv(sock, string_type, r_arch, &msg_name);
151   XBT_DEBUG
152       ("Handle an incoming message '%s' (%s) using protocol %d (remote is %s)",
153        msg_name, e_gras_msg_kind_names[msg->kind], (int) header[4],
154        xbt_datadesc_arch_name(r_arch));
155
156   TRY {
157     msg->type =
158         (gras_msgtype_t) xbt_set_get_by_name(_gras_msgtype_set, msg_name);
159   }
160   CATCH(e) {
161     /* FIXME: Survive unknown messages */
162     if (e.category == not_found_error) {
163       xbt_ex_free(e);
164       THROWF(not_found_error, 0,
165              "Received an unknown message: %s (FIXME: should survive to these)",
166              msg_name);
167     } else
168       RETHROWF
169           ("Exception caught while retrieving the type associated to messages '%s' : %s",
170            msg_name);
171   }
172   free(msg_name);
173
174   if (msg->kind == e_gras_msg_kind_rpcerror) {
175     /* error on remote host. Carfull with that exception, eugene */
176     msg->payl_size = xbt_datadesc_size(xbt_datadesc_by_name("ex_t"));
177     msg->payl = xbt_malloc(msg->payl_size);
178     xbt_datadesc_recv(sock, xbt_datadesc_by_name("ex_t"), r_arch,
179                        msg->payl);
180
181   } else if (msg->kind == e_gras_msg_kind_rpcanswer) {
182     /* answer to RPC */
183     if (msg->type->answer_type) {
184       msg->payl_size = xbt_datadesc_size(msg->type->answer_type);
185       xbt_assert(msg->payl_size > 0,
186                   "%s %s",
187                   "Dynamic array as payload is forbided for now (FIXME?).",
188                   "Reference to dynamic array is allowed.");
189       msg->payl = xbt_malloc(msg->payl_size);
190       xbt_datadesc_recv(sock, msg->type->answer_type, r_arch, msg->payl);
191     } else {
192       msg->payl = NULL;
193       msg->payl_size = 0;
194     }
195   } else {
196     /* regular message */
197     if (msg->type->ctn_type) {
198       msg->payl_size = xbt_datadesc_size(msg->type->ctn_type);
199       xbt_assert(msg->payl_size > 0,
200                   "%s %s",
201                   "Dynamic array as payload is forbided for now (FIXME?).",
202                   "Reference to dynamic array is allowed.");
203       msg->payl = xbt_malloc(msg->payl_size);
204       xbt_datadesc_recv(sock, msg->type->ctn_type, r_arch, msg->payl);
205     } else {
206       msg->payl = NULL;
207       msg->payl_size = 0;
208     }
209   }
210 }