teshsuite/gras/msg_handle/msg_handle.c

   1 /* $Id: mmrpc.c 3399 2007-04-11 19:34:43Z cherierm $ */
   2
   3 /* msg_handle - ensures the semantic of gras_msg_handle(i) for i<0,=0 or >0 */
   4
   5 /* Copyright (c) 2007, 2008, 2009, 2010. The SimGrid Team.
   6  * All rights reserved.                                                     */
   7 /* Thanks to Flavien Vernier for reporting an issue around this             */
   8
   9 /* This program is free software; you can redistribute it and/or modify it
  10  * under the terms of the license (GNU LGPL) which comes with this package. */
  11
  12 #include "gras.h"
  13
  14 XBT_LOG_NEW_DEFAULT_CATEGORY(test, "Messages specific to this test");
  15
  16 int server(int argc, char *argv[]);
  17 int client(int argc, char *argv[]);
  18
  19 static int server_cb_hello_handler(gras_msg_cb_ctx_t ctx,
  20                                    void *payload_data)
  21 {
  22   XBT_INFO("Got the message");
  23   return 0;
  24 }
  25
  26 int server(int argc, char *argv[])
  27 {
  28   gras_socket_t me = NULL, pal = NULL;
  29   int myport;
  30   char *palstr;
  31
  32   xbt_ex_t e;
  33   int got_expected;
  34   double now;
  35
  36
  37   gras_init(&argc, argv);
  38
  39   xbt_assert(argc == 3, "Usage: server <myport> <client>");
  40   myport = atoi(argv[1]);
  41   palstr = argv[2];
  42
  43   gras_msgtype_declare("hello", NULL);
  44   gras_cb_register("hello", &server_cb_hello_handler);
  45
  46   XBT_INFO("Launch server (port=%d)", myport);
  47   TRY {
  48     me = gras_socket_server(myport);
  49   }
  50   CATCH_ANONYMOUS {
  51     RETHROWF("Unable to establish a server socket: %s");
  52   }
  53   gras_os_sleep(1);             /* Wait for pal to startup */
  54   TRY {
  55     pal = gras_socket_client_from_string(palstr);
  56   }
  57   CATCH_ANONYMOUS {
  58     RETHROWF("Unable to establish a socket to %s: %s", palstr);
  59   }
  60   XBT_INFO("Initialization done.");
  61   now = gras_os_time();
  62
  63   /* Launch handle(0) when there is no message. Timeout expected */
  64   got_expected = 0;
  65   TRY {
  66     gras_msg_handle(0);
  67   }
  68   CATCH(e) {
  69     if (e.category == timeout_error) {
  70       got_expected = 1;
  71       xbt_ex_free(e);
  72     } else {
  73       RETHROWF("Didn't got the expected timeout: %s");
  74     }
  75   }
  76   xbt_assert(got_expected,
  77               "gras_msg_handle(0) do not lead to any timeout exception");
  78   xbt_assert(gras_os_time() - now < 0.01,
  79               "gras_msg_handle(0) do not anwser immediately (%.4fsec)",
  80               gras_os_time() - now);
  81   XBT_INFO("gras_msg_handle(0) works as expected (immediate timeout)");
  82   /* Launch handle(0) when there is no message. Timeout expected */
  83   got_expected = 0;
  84   TRY {
  85     gras_msg_handle(1);
  86   }
  87   CATCH(e) {
  88     if (e.category == timeout_error) {
  89       got_expected = 1;
  90       xbt_ex_free(e);
  91     } else {
  92       RETHROWF("Didn't got the expected timeout: %s");
  93     }
  94   }
  95   xbt_assert(got_expected,
  96               "gras_msg_handle(1) do not lead to any timeout exception");
  97   xbt_assert(gras_os_time() - now < 1.5,
  98               "gras_msg_handle(1) needs more than 1.5 sec to answer (%.4fsec)",
  99               gras_os_time() - now);
 100   xbt_assert(gras_os_time() - now >= 1.0,
 101               "gras_msg_handle(1) answers in less than one second (%.4fsec)",
 102               gras_os_time() - now);
 103   XBT_INFO("gras_msg_handle(1) works as expected (delayed timeout)");
 104   gras_os_sleep(3);
 105
 106   /* Send an hello to the client to unlock it */
 107   XBT_INFO("Unlock pal");
 108   gras_msg_send(pal, "hello", NULL);
 109
 110   /* Frees the allocated resources, and shut GRAS down */
 111   gras_socket_close(me);
 112   gras_socket_close(pal);
 113   gras_exit();
 114   return 0;
 115 }
 116
 117 int client(int argc, char *argv[])
 118 {
 119   gras_socket_t me = NULL, pal = NULL;
 120   int myport;
 121   char *palstr;
 122
 123   gras_init(&argc, argv);
 124   xbt_assert(argc == 3, "Usage: client <myport> <server>");
 125   myport = atoi(argv[1]);
 126   palstr = argv[2];
 127
 128   gras_msgtype_declare("hello", NULL);
 129   gras_cb_register("hello", &server_cb_hello_handler);
 130
 131   XBT_INFO("Launch client (port=%d)", myport);
 132   TRY {
 133     me = gras_socket_server(myport);
 134   }
 135   CATCH_ANONYMOUS {
 136     RETHROWF("Unable to establish a server socket: %s");
 137   }
 138   gras_os_sleep(1);             /* Wait for pal to startup */
 139   TRY {
 140     pal = gras_socket_client_from_string(palstr);
 141   }
 142   CATCH_ANONYMOUS {
 143     RETHROWF("Unable to establish a socket to %s: %s", palstr);
 144   }
 145   XBT_INFO("Initialization done.");
 146
 147   /* Launch handle(-1). Lock until message from server expected */
 148   TRY {
 149     gras_msg_handle(-1);
 150   }
 151   CATCH_ANONYMOUS {
 152     RETHROWF("No exception expected during handle(-1), but got %s");
 153   }
 154   XBT_INFO("gras_msg_handle(-1) works as expected (locked)");
 155
 156   /* Frees the allocated resources, and shut GRAS down */
 157   gras_socket_close(me);
 158   gras_socket_close(pal);
 159   gras_exit();
 160   return 0;
 161 }