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_ex_t e;
  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   TRY {
  56     gras_msg_send(expeditor, gras_msgtype_by_name("pong"), &msg);
  57
  58   /* 6. Deal with errors: add some details to the exception */
  59   } CATCH(e) {
  60     gras_socket_close(globals->sock);
  61     RETHROW0("Unable answer with PONG: %s");
  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   server_data_t *globals;
  78
  79   int port = 4000;
  80
  81   /* 1. Init the GRAS infrastructure and declare my globals */
  82   gras_init(&argc,argv);
  83   globals=gras_userdata_new(server_data_t);
  84
  85   /* 2. Get the port I should listen on from the command line, if specified */
  86   if (argc == 2) {
  87     port=atoi(argv[1]);
  88   }
  89
  90   INFO1("Launch server (port=%d)", port);
  91
  92   /* 3. Create my master socket */
  93   globals->sock = gras_socket_server(port);
  94
  95   /* 4. Register the known messages. This function is called twice here, but it's because
  96         this file also acts as regression test, no need to do so yourself of course. */
  97   register_messages();
  98   register_messages(); /* just to make sure it works ;) */
  99
 100   /* 5. Register my callback */
 101   gras_cb_register(gras_msgtype_by_name("ping"),&server_cb_ping_handler);
 102
 103   INFO1(">>>>>>>> Listening on port %d <<<<<<<<", gras_socket_my_port(globals->sock));
 104   globals->endcondition=0;
 105
 106   /* 6. Wait up to 10 minutes for an incomming message to handle */
 107   gras_msg_handle(600.0);
 108
 109   /* 7. Housekeeping */
 110   if (!globals->endcondition)
 111      WARN0("An error occured, the endcondition was not set by the callback");
 112
 113   /* 8. Free the allocated resources, and shut GRAS down */
 114   gras_socket_close(globals->sock);
 115   free(globals);
 116   gras_exit();
 117
 118   INFO0("Done.");
 119   return 0;
 120 } /* end_of_server */
 121
 122 /* **********************************************************************
 123  * Client code
 124  * **********************************************************************/
 125
 126 /* Function prototypes */
 127
 128 int client(int argc,char *argv[]) {
 129   xbt_ex_t e;
 130   gras_socket_t toserver; /* peer */
 131
 132   gras_socket_t from;
 133   int ping, pong;
 134
 135   const char *host = "127.0.0.1";
 136         int   port = 4000;
 137
 138   /* 1. Init the GRAS's infrastructure */
 139   gras_init(&argc, argv);
 140
 141   /* 2. Get the server's address. The command line override defaults when specified */
 142   if (argc == 3) {
 143     host=argv[1];
 144     port=atoi(argv[2]);
 145   }
 146
 147   INFO2("Launch client (server on %s:%d)",host,port);
 148
 149   /* 3. Wait for the server startup */
 150   gras_os_sleep(1);
 151
 152   /* 4. Create a socket to speak to the server */
 153   TRY {
 154     toserver=gras_socket_client(host,port);
 155   } CATCH(e) {
 156     RETHROW0("Unable to connect to the server: %s");
 157   }
 158   INFO2("Connected to %s:%d.",host,port);
 159
 160
 161   /* 5. Register the messages.
 162         See, it doesn't have to be done completely at the beginning, only before use */
 163   register_messages();
 164
 165   /* 6. Keep the user informed of what's going on */
 166   INFO2(">>>>>>>> Connected to server which is on %s:%d <<<<<<<<",
 167         gras_socket_peer_name(toserver),gras_socket_peer_port(toserver));
 168
 169   /* 7. Prepare and send the ping message to the server */
 170   ping = 1234;
 171   TRY {
 172     gras_msg_send(toserver, gras_msgtype_by_name("ping"), &ping);
 173   } CATCH(e) {
 174     gras_socket_close(toserver);
 175     RETHROW0("Failed to send PING to server: %s");
 176   }
 177   INFO3(">>>>>>>> Message PING(%d) sent to %s:%d <<<<<<<<",
 178         ping,
 179         gras_socket_peer_name(toserver),gras_socket_peer_port(toserver));
 180
 181   /* 8. Wait for the answer from the server, and deal with issues */
 182   TRY {
 183     gras_msg_wait(6000,gras_msgtype_by_name("pong"),
 184                   &from,&pong);
 185   } CATCH(e) {
 186     gras_socket_close(toserver);
 187     RETHROW0("Why can't I get my PONG message like everyone else: %s");
 188   }
 189
 190   /* 9. Keep the user informed of what's going on, again */
 191   INFO3(">>>>>>>> Got PONG(%d) from %s:%d <<<<<<<<",
 192         pong,
 193         gras_socket_peer_name(from),gras_socket_peer_port(from));
 194
 195   /* 10. Free the allocated resources, and shut GRAS down */
 196   gras_socket_close(toserver);
 197   gras_exit();
 198   INFO0("Done.");
 199   return 0;
 200 } /* end_of_client */