examples/gras/all2all/all2all.c

   1 /* $Id$ */
   2
   3 /* ALL2ALL - all2all of GRAS features                                       */
   4
   5 /* Copyright (c) 2006 Ahmed Harbaoui. 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 #include "xbt/ex.h"
  12
  13 XBT_LOG_NEW_DEFAULT_CATEGORY(all2all,"Messages specific to this example");
  14
  15 /* register data which may be sent (common to client and server) */
  16 static void register_messages(void) {
  17 }
  18
  19 /* Function prototypes */
  20 int receiver (int argc,char *argv[]);
  21 int sender (int argc,char *argv[]);
  22
  23
  24 /* **********************************************************************
  25  * Receiver code
  26  * **********************************************************************/
  27 int receiver (int argc,char *argv[]) {
  28
  29   int myport; /* port on which I receive stuff */
  30   int todo; /* amount of messages I should get */
  31   char *data; /* message content */
  32
  33   gras_socket_t mysock;  /* socket on which other people contact me */
  34   gras_socket_t expeditor;  /* to notice who wrote me */
  35
  36   /* Init the GRAS infrastructure and declare my globals */
  37   gras_init(&argc,argv);
  38
  39   /* Get my settings from the command line */
  40   myport=atoi(argv[1]);
  41   todo=atoi(argv[2]);
  42
  43   /* Register the known messages */
  44   gras_msgtype_declare("data", gras_datadesc_by_name("string"));
  45
  46   /* Create my master socket */
  47   mysock = gras_socket_server(myport);
  48
  49   /* Get the data */
  50   INFO2("Listening on port %d (expecting %d messages)",
  51         gras_socket_my_port(mysock),
  52         todo);
  53   while (todo>0) {
  54      gras_msg_wait(60 /* wait up to one minute */,
  55                    "data",
  56                    &expeditor,
  57                    &data);
  58      todo--;
  59      free(data);
  60
  61      INFO3("Got Data from %s:%d (still %d to go)",
  62            gras_socket_peer_name(expeditor), gras_socket_peer_port(expeditor),
  63            todo);
  64
  65   }
  66
  67   /* Free the allocated resources, and shut GRAS down */
  68   gras_socket_close(mysock);
  69
  70   gras_exit();
  71   return 0;
  72 } /* end_of_receiver */
  73
  74 /* **********************************************************************
  75  * Sender code
  76  * **********************************************************************/
  77
  78 int sender (int argc,char *argv[]) {
  79
  80   unsigned int iter; /* iterator */
  81   char *data; /* data exchanged */
  82   int datasize; /* size of message */
  83   xbt_peer_t h; /* iterator */
  84   int connected = 0;
  85
  86   gras_socket_t peer=NULL;  /* socket to node */
  87
  88
  89   /* xbt_dynar for peers */
  90   xbt_dynar_t peers = xbt_dynar_new(sizeof(xbt_peer_t),&xbt_peer_free_voidp);
  91
  92   /* Init the GRAS infrastructure and declare my globals */
  93   gras_init(&argc,argv);
  94
  95   /* Get the node location from argc/argv */
  96   for (iter=1; iter<argc-1; iter++){
  97     xbt_peer_t peer = xbt_peer_from_string(argv[iter]);
  98     xbt_dynar_push(peers,&peer);
  99   }
 100
 101   datasize=atoi(argv[argc-1]);
 102
 103   data=(char *) malloc(datasize+1);  // allocation of datasize octets
 104   memset(data, 32, datasize);
 105   data[datasize] = '\0';
 106
 107   INFO0("Launch current node");
 108
 109   /* Register the known messages */
 110   gras_msgtype_declare("data", gras_datadesc_by_name("string"));
 111
 112
 113   /* write to the receivers */
 114   xbt_dynar_foreach(peers,iter,h) {
 115      connected = 0;
 116      while (!connected) {
 117         xbt_ex_t e;
 118         TRY {
 119            peer = gras_socket_client(h->name,h->port);
 120            connected=1;
 121         } CATCH(e) {
 122            if (e.category != system_error /*in RL*/&& e.category != mismatch_error/*in SG*/)
 123              RETHROW;
 124            xbt_ex_free(e);
 125            gras_os_sleep(0.01);
 126         }
 127      }
 128      gras_msg_send(peer,"data",&data);
 129      if (gras_if_SG()) {
 130         INFO2("  Sent Data from %s to %s", gras_os_myname(),h->name);
 131      } else {
 132         INFO0("  Sent Data");
 133      }
 134
 135      gras_socket_close(peer);
 136   }
 137
 138   /* Free the allocated resources, and shut GRAS down */
 139   free(data);
 140   xbt_dynar_free(&peers);
 141
 142   gras_exit();
 143   return 0;
 144 } /* end_of_sender */