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(">>>>>>>> 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("Unable answer with PONG: %s\n", xbt_error_name(errcode));
  60     gras_socket_close(globals->sock);
  61     return 1;
  62   }
  63
  64   INFO0(">>>>>>>> Answered 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(">>>>>>>> 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. Free the allocated resources, and shut GRAS down */
 121   gras_socket_close(globals->sock);
 122   free(globals);
 123   gras_exit();
 124
 125   INFO0("Done.");
 126   return no_error;
 127 } /* end_of_server */
 128
 129 /* **********************************************************************
 130  * Client code
 131  * **********************************************************************/
 132
 133 /* Function prototypes */
 134
 135 int client(int argc,char *argv[]) {
 136   xbt_error_t errcode;
 137   gras_socket_t toserver; /* peer */
 138
 139   gras_socket_t from;
 140   int ping, pong;
 141
 142   const char *host = "127.0.0.1";
 143         int   port = 4000;
 144
 145   /* 1. Init the GRAS's infrastructure */
 146   gras_init(&argc, argv, NULL);
 147
 148   /* 2. Get the server's address. The command line override defaults when specified */
 149   if (argc == 3) {
 150     host=argv[1];
 151     port=atoi(argv[2]);
 152   }
 153
 154   INFO2("Launch client (server on %s:%d)",host,port);
 155
 156   /* 3. Wait for the server startup */
 157   gras_os_sleep(1);
 158
 159   /* 4. Create a socket to speak to the server */
 160   if ((errcode=gras_socket_client(host,port,&toserver))) {
 161     ERROR1("Unable to connect to the server. Got %s",
 162            xbt_error_name(errcode));
 163     return 1;
 164   }
 165   INFO2("Connected to %s:%d.",host,port);
 166
 167
 168   /* 5. Register the messages.
 169         See, it doesn't have to be done completely at the beginning, only before use */
 170   register_messages();
 171
 172   /* 6. Keep the user informed of what's going on */
 173   INFO2(">>>>>>>> Connected to server which is on %s:%d <<<<<<<<",
 174         gras_socket_peer_name(toserver),gras_socket_peer_port(toserver));
 175
 176   /* 7. Prepare and send the ping message to the server */
 177   ping = 1234;
 178   errcode = gras_msg_send(toserver, gras_msgtype_by_name("ping"), &ping);
 179   if (errcode != no_error) {
 180     fprintf(stderr, "Unable send PING to server (%s)\n",
 181             xbt_error_name(errcode));
 182     gras_socket_close(toserver);
 183     return 1;
 184   }
 185   INFO3(">>>>>>>> Message PING(%d) sent to %s:%d <<<<<<<<",
 186         ping,
 187         gras_socket_peer_name(toserver),gras_socket_peer_port(toserver));
 188
 189   /* 8. Wait for the answer from the server, and deal with issues */
 190   if ((errcode=gras_msg_wait(6000,gras_msgtype_by_name("pong"),
 191                              &from,&pong))) {
 192     ERROR1("Why can't I get my PONG message like everyone else (%s)?",
 193            xbt_error_name(errcode));
 194     gras_socket_close(toserver);
 195     return 1;
 196   }
 197
 198   /* 9. Keep the user informed of what's going on, again */
 199   INFO3(">>>>>>>> Got PONG(%d) from %s:%d <<<<<<<<",
 200         pong,
 201         gras_socket_peer_name(from),gras_socket_peer_port(from));
 202
 203   /* 10. Free the allocated resources, and shut GRAS down */
 204   gras_socket_close(toserver);
 205   gras_exit();
 206   INFO0("Done.");
 207   return 0;
 208 } /* end_of_client */