Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
uniformize gras_os_sleep prototype (seconds are double everywhere else, avoid roundin...
[simgrid.git] / examples / gras / ping / ping.c
1 /* $Id$ */
2
3 /* ping - ping/pong demo of GRAS features                                   */
4
5 /* Copyright (c) 2003, 2004, 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 "gras.h"
11
12 XBT_LOG_NEW_DEFAULT_CATEGORY(Ping,"Messages specific to this example");
13
14 /* register messages which may be sent (common to client and server) */
15 static void register_messages(void) {
16   gras_msgtype_declare("ping", gras_datadesc_by_name("int"));
17   gras_msgtype_declare("pong", gras_datadesc_by_name("int"));
18 }
19
20 /* Function prototypes */
21 int server (int argc,char *argv[]);
22 int client (int argc,char *argv[]);
23
24 /* **********************************************************************
25  * Server code
26  * **********************************************************************/
27
28 /* Global private data */
29 typedef struct {
30   gras_socket_t sock;
31   int endcondition;
32 } server_data_t;
33
34
35 static int server_cb_ping_handler(gras_socket_t  expeditor,
36                                   void          *payload_data) {
37                              
38   xbt_error_t errcode;
39   /* 1. Get the payload into the msg variable */
40   int msg=*(int*)payload_data;
41
42   /* 2. Retrieve the server's state (globals) */
43
44   server_data_t *globals=(server_data_t*)gras_userdata_get();
45   globals->endcondition = 0;
46    
47   /* 3. Log which client connected */
48   INFO3("SERVER: >>>>>>>> Got message PING(%d) from %s:%d <<<<<<<<", 
49         msg, 
50         gras_socket_peer_name(expeditor), gras_socket_peer_port(expeditor));
51   
52   /* 4. Change the value of the msg variable */
53   msg = 4321;
54   /* 5. Send it back as payload of a pong message to the expeditor */
55   errcode = gras_msg_send(expeditor, gras_msgtype_by_name("pong"), &msg);
56
57   /* 6. Deal with errors */
58   if (errcode != no_error) {
59     ERROR1("SERVER: Unable answer with PONG: %s\n", xbt_error_name(errcode));
60     gras_socket_close(globals->sock);
61     return 1;
62   }
63
64   INFO0("SERVER: >>>>>>>> Answed with PONG(4321) <<<<<<<<");
65    
66   /* 7. Set the endcondition boolean to true (and make sure the server stops after receiving it). */
67   globals->endcondition = 1;
68   
69   /* 8. Make sure we don't leak sockets */
70   gras_socket_close(expeditor);
71    
72   /* 9. Tell GRAS that we consummed this message */
73   return 1;
74 } /* end_of_server_cb_ping_handler */
75
76 int server (int argc,char *argv[]) {
77   xbt_error_t errcode;
78   server_data_t *globals;
79
80   int port = 4000;
81   
82   /* 1. Init the GRAS infrastructure and declare my globals */
83   gras_init(&argc,argv, NULL);
84   globals=gras_userdata_new(server_data_t);
85    
86   /* 2. Get the port I should listen on from the command line, if specified */
87   if (argc == 2) {
88     port=atoi(argv[1]);
89   }
90
91   INFO1("Launch server (port=%d)", port);
92
93   /* 3. Create my master socket */
94   if ((errcode=gras_socket_server(port,&(globals->sock)))) { 
95     CRITICAL1("Error %s encountered while opening the server socket",
96               xbt_error_name(errcode));
97     return 1;
98   }
99
100   /* 4. Register the known messages. This function is called twice here, but it's because
101         this file also acts as regression test, no need to do so yourself of course. */
102   register_messages();
103   register_messages(); /* just to make sure it works ;) */
104    
105   /* 5. Register my callback */
106   gras_cb_register(gras_msgtype_by_name("ping"),&server_cb_ping_handler);
107
108   INFO1("SERVER: >>>>>>>> Listening on port %d <<<<<<<<", gras_socket_my_port(globals->sock));
109   globals->endcondition=0;
110
111   /* 6. Wait up to 10 minutes for an incomming message to handle */
112   errcode = gras_msg_handle(600.0);
113    
114   /* 7. Housekeeping */
115   if (errcode != no_error)
116     return errcode;
117   if (!globals->endcondition)
118      WARN0("An error occured, the endcondition was not set by the callback");
119   
120   /* 8. Sleep one second, but only in real life, not in simulation */
121   if (!gras_if_RL())
122     gras_os_sleep(1);
123
124   /* 9. Free the allocated resources, and shut GRAS down */
125   gras_socket_close(globals->sock);
126   free(globals);
127   gras_exit();
128    
129   INFO0("SERVER: Done.");
130   return no_error;
131 } /* end_of_server */
132
133 /* **********************************************************************
134  * Client code
135  * **********************************************************************/
136
137 /* Function prototypes */
138
139 int client(int argc,char *argv[]) {
140   xbt_error_t errcode; 
141   gras_socket_t toserver; /* peer */
142
143   gras_socket_t from;
144   int ping, pong;
145
146   const char *host = "127.0.0.1";
147         int   port = 4000;
148
149   /* 1. Init the GRAS's infrastructure */
150   gras_init(&argc, argv, NULL);
151    
152   /* 2. Get the server's address. The command line override defaults when specified */
153   if (argc == 3) {
154     host=argv[1];
155     port=atoi(argv[2]);
156   } 
157
158   INFO2("Launch client (server on %s:%d)",host,port);
159    
160   /* 3. Wait for the server startup */
161   gras_os_sleep(1);
162    
163   /* 4. Create a socket to speak to the server */
164   if ((errcode=gras_socket_client(host,port,&toserver))) {
165     ERROR1("Client: Unable to connect to the server. Got %s",
166            xbt_error_name(errcode));
167     return 1;
168   }
169   INFO2("Client: Connected to %s:%d.",host,port);    
170
171
172   /* 5. Register the messages. 
173         See, it doesn't have to be done completely at the beginning, only before use */
174   register_messages();
175
176   /* 6. Keep the user informed of what's going on */
177   INFO2("Client: >>>>>>>> Connected to server which is on %s:%d <<<<<<<<", 
178         gras_socket_peer_name(toserver),gras_socket_peer_port(toserver));
179
180   /* 7. Prepare and send the ping message to the server */
181   ping = 1234;
182   errcode = gras_msg_send(toserver, gras_msgtype_by_name("ping"), &ping);
183   if (errcode != no_error) {
184     fprintf(stderr, "Client: Unable send PING to server (%s)\n",
185             xbt_error_name(errcode));
186     gras_socket_close(toserver);
187     return 1;
188   }
189   INFO3("Client: >>>>>>>> Message PING(%d) sent to %s:%d <<<<<<<<",
190         ping,
191         gras_socket_peer_name(toserver),gras_socket_peer_port(toserver));
192
193   /* 8. Wait for the answer from the server, and deal with issues */
194   if ((errcode=gras_msg_wait(6000,gras_msgtype_by_name("pong"),
195                              &from,&pong))) {
196     ERROR1("Client: Why can't I get my PONG message like everyone else (%s)?",
197            xbt_error_name(errcode));
198     gras_socket_close(toserver);
199     return 1;
200   }
201
202   /* 9. Keep the user informed of what's going on, again */
203   INFO3("Client: >>>>>>>> Got PONG(%d) from %s:%d <<<<<<<<", 
204         pong,
205         gras_socket_peer_name(from),gras_socket_peer_port(from));
206
207   /* 10. Free the allocated resources, and shut GRAS down */
208   gras_socket_close(toserver);
209   gras_exit();
210   INFO0("Client: Done.");
211   return 0;
212 } /* end_of_client */