Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
a4c4e160d570512e027b25dded577782e34dfad0
[simgrid.git] / src / gras / Msg / rl_msg.c
1 /* $Id$ */
2
3 /* messaging - Function related to messaging code specific to RL            */
4
5 /* Copyright (c) 2003-2005 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 "xbt/ex.h"
11 #include "gras/Msg/msg_private.h"
12
13 #include "gras/DataDesc/datadesc_interface.h"
14 #include "gras/Transport/transport_interface.h" /* gras_trp_send/recv */
15
16 XBT_LOG_EXTERNAL_CATEGORY(gras_msg);
17 XBT_LOG_DEFAULT_CATEGORY(gras_msg);
18
19 void gras_msg_send_ext(gras_socket_t   sock,
20                      e_gras_msg_kind_t kind,
21                        unsigned long int ID,
22                        gras_msgtype_t  msgtype,
23                        void           *payload) {
24
25   static gras_datadesc_type_t string_type=NULL;
26   static gras_datadesc_type_t ulong_type=NULL;
27   char c_kind=(char)kind;
28   
29   xbt_assert0(msgtype,"Cannot send the NULL message");
30
31   if (!string_type) {
32     string_type = gras_datadesc_by_name("string");
33     xbt_assert(string_type);
34   }
35   if (!ulong_type) {
36     ulong_type = gras_datadesc_by_name("unsigned long int");
37     xbt_assert(ulong_type);
38   }
39
40   DEBUG3("send '%s' to %s:%d", msgtype->name, 
41          gras_socket_peer_name(sock),gras_socket_peer_port(sock));
42   gras_trp_send(sock, _GRAS_header, 6, 1 /* stable */);
43   gras_trp_send(sock, &c_kind,      1, 1 /* stable */);
44   switch (kind) {
45    case e_gras_msg_kind_oneway: 
46      break;
47      
48    case e_gras_msg_kind_rpccall:
49    case e_gras_msg_kind_rpcanswer:
50    case e_gras_msg_kind_rpcerror:
51      gras_datadesc_send(sock,ulong_type,&ID);
52      break;
53
54    default:
55      THROW1(unknown_error,0,"Unknown msg kind %d",kind);
56   }
57    
58   gras_datadesc_send(sock, string_type,   &msgtype->name);
59   if (kind == e_gras_msg_kind_rpcerror) {
60      /* error on remote host, carfull, payload is an exception */
61     gras_datadesc_send(sock, gras_datadesc_by_name("ex_t"),payload);
62   } else if (kind == e_gras_msg_kind_rpcanswer) {
63     if (msgtype->answer_type)
64       gras_datadesc_send(sock, msgtype->answer_type, payload);
65   }else {
66      /* regular message */
67      if (msgtype->ctn_type)
68        gras_datadesc_send(sock, msgtype->ctn_type, payload);
69   }
70    
71   gras_trp_flush(sock);
72 }
73
74 /*
75  * receive the next message on the given socket.  
76  */
77 void
78 gras_msg_recv(gras_socket_t    sock,
79               gras_msg_t       msg) {
80
81   xbt_ex_t e;
82   static gras_datadesc_type_t string_type=NULL;
83   static gras_datadesc_type_t ulong_type=NULL;
84   char header[6];
85   int cpt;
86   int r_arch;
87   char *msg_name=NULL;
88   char c_kind;
89
90   xbt_assert1(!gras_socket_is_meas(sock), 
91               "Asked to receive a message on the measurement socket %p", sock);
92   if (!string_type) {
93     string_type=gras_datadesc_by_name("string");
94     xbt_assert(string_type);
95   }
96   if (!ulong_type) {
97     ulong_type = gras_datadesc_by_name("unsigned long int");
98     xbt_assert(ulong_type);
99   }
100
101   
102   TRY {
103     gras_trp_recv(sock, header, 6);
104     gras_trp_recv(sock, &c_kind, 1);
105     msg->kind=(e_gras_msg_kind_t)c_kind;
106   } CATCH(e) {
107     RETHROW0("Exception caught while trying to get the mesage header: %s");
108   }
109
110   for (cpt=0; cpt<4; cpt++)
111     if (header[cpt] != _GRAS_header[cpt])
112       THROW2(mismatch_error,0,
113              "Incoming bytes do not look like a GRAS message (header='%.4s' not '%.4s')",
114              header,_GRAS_header);
115   if (header[4] != _GRAS_header[4]) 
116     THROW2(mismatch_error,0,"GRAS protocol mismatch (got %d, use %d)",
117            (int)header[4], (int)_GRAS_header[4]);
118   r_arch = (int)header[5];
119   DEBUG2("Handle an incoming message using protocol %d (remote is %s)",
120          (int)header[4],gras_datadesc_arch_name(r_arch));
121
122   switch (msg->kind) {
123   case e_gras_msg_kind_oneway: 
124     break;
125     
126   case e_gras_msg_kind_rpccall:
127   case e_gras_msg_kind_rpcanswer:
128   case e_gras_msg_kind_rpcerror:
129     gras_datadesc_recv(sock,ulong_type,r_arch, &msg->ID);
130     break;
131     
132   default:
133     THROW_IMPOSSIBLE;
134   }
135
136   gras_datadesc_recv(sock, string_type, r_arch, &msg_name);
137   TRY {
138     msg->type = (gras_msgtype_t)xbt_set_get_by_name(_gras_msgtype_set,msg_name);
139   } CATCH(e) {
140     /* FIXME: Survive unknown messages */
141     RETHROW1("Exception caught while retrieving the type associated to messages '%s' : %s",
142              msg_name);
143   }
144   free(msg_name);
145
146   if (msg->kind == e_gras_msg_kind_rpcerror) {
147     /* error on remote host. Carfull with that exception, eugene */
148     msg->payl_size=gras_datadesc_size(gras_datadesc_by_name("ex_t"));
149     msg->payl=xbt_malloc(msg->payl_size);
150     gras_datadesc_recv(sock, gras_datadesc_by_name("ex_t"), r_arch, msg->payl);
151
152   } else if  (msg->kind == e_gras_msg_kind_rpcanswer) {
153      /* answer to RPC */
154      if (msg->type->answer_type) {
155         msg->payl_size=gras_datadesc_size(msg->type->answer_type);
156         xbt_assert2(msg->payl_size > 0,
157                     "%s %s",
158                     "Dynamic array as payload is forbided for now (FIXME?).",
159                     "Reference to dynamic array is allowed.");
160         msg->payl = xbt_malloc(msg->payl_size);
161         gras_datadesc_recv(sock, msg->type->answer_type, r_arch, msg->payl);
162      } else {
163         msg->payl = NULL;
164         msg->payl_size = 0;
165      }
166   } else {
167      /* regular message */
168      if (msg->type->ctn_type) {
169         msg->payl_size=gras_datadesc_size(msg->type->ctn_type);
170         xbt_assert2(msg->payl_size > 0,
171                     "%s %s",
172                     "Dynamic array as payload is forbided for now (FIXME?).",
173                     "Reference to dynamic array is allowed.");
174         msg->payl = xbt_malloc(msg->payl_size);
175         gras_datadesc_recv(sock, msg->type->ctn_type, r_arch, msg->payl);
176      } else {
177         msg->payl = NULL;
178         msg->payl_size = 0;
179      }
180   }
181 }