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   /* Create my master socket */
  44   mysock = gras_socket_server(myport);
  45
  46   /* Register the known messages */
  47   gras_msgtype_declare("data", gras_datadesc_by_name("string"));
  48
  49   /* Get the data */
  50
  51   INFO2("Listening on port %d (expecting %d messages)",
  52         gras_socket_my_port(mysock),
  53         todo);
  54   while (todo>0) {
  55      gras_msg_wait(60 /* wait up to one minute */,
  56                    "data",
  57                    &expeditor,
  58                    &data);
  59      todo--;
  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   int i; /* iterator */
  81   char *data; /* data exchanged */
  82   int datasize; /* size of message */
  83   xbt_peer_t h; /* iterator */
  84
  85   gras_socket_t peer;  /* socket to node */
  86
  87
  88   /* xbt_dynar for peers */
  89   xbt_dynar_t peers = xbt_dynar_new(sizeof(xbt_peer_t),&xbt_peer_free_voidp);
  90
  91   /* Init the GRAS infrastructure and declare my globals */
  92   gras_init(&argc,argv);
  93
  94   /* Get the node location from argc/argv */
  95   for (i=1; i<argc-1; i++){
  96     xbt_peer_t peer = xbt_peer_from_string(argv[i]);
  97     xbt_dynar_push(peers,&peer);
  98   }
  99
 100   datasize=atoi(argv[argc-1]);
 101
 102   data=(char *) malloc(datasize+1);  // allocation of datasize octets
 103   memset(data, 32, datasize);
 104   data[datasize] = '\0';
 105
 106   INFO0("Launch current node");
 107
 108   /* Register the known messages */
 109   gras_msgtype_declare("data", gras_datadesc_by_name("string"));
 110
 111
 112   /* Wait for receivers to startup */
 113   gras_os_sleep(.01);
 114
 115   /* write 'em */
 116   xbt_dynar_foreach(peers,i,h) {
 117
 118      peer = gras_socket_client(h->name,h->port);
 119      gras_msg_send(peer,"data",&data);
 120      if (gras_if_SG()) {
 121         INFO2("  Sent Data from %s to %s", gras_os_myname(),h->name);
 122      } else {
 123         INFO0("  Sent Data");
 124      }
 125
 126      gras_socket_close(peer);
 127   }
 128
 129   /* Free the allocated resources, and shut GRAS down */
 130   xbt_dynar_free(&peers);
 131
 132   gras_exit();
 133   return 0;
 134 } /* end_of_sender */