Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Fix warnings about clobbered variables in amok/bandwidth example.
[simgrid.git] / examples / amok / bandwidth / bandwidth.c
1 /* bandwidth - bandwidth test demo of GRAS features                         */
2
3 /* Copyright (c) 2005, 2006, 2007, 2008, 2009, 2010. The SimGrid Team.
4  * All rights reserved.                                                     */
5
6 /* This program is free software; you can redistribute it and/or modify it
7  * under the terms of the license (GNU LGPL) which comes with this package. */
8
9
10 #include "gras.h"
11 #include "amok/bandwidth.h"
12 #include "amok/peermanagement.h"
13
14 XBT_LOG_NEW_DEFAULT_CATEGORY(Bandwidth,
15                              "Messages specific to this example");
16
17 /* **********************************************************************
18  * Sensor code
19  * **********************************************************************/
20
21 static gras_socket_t try_gras_socket_client_from_string(const char *host)
22 {
23   volatile gras_socket_t sock = NULL;
24   xbt_ex_t e;
25   TRY {
26     sock = gras_socket_client_from_string(host);
27   }
28   CATCH(e) {
29     xbt_ex_free(e);
30   }
31   return sock;
32 }
33
34 /* Function prototypes */
35 int sensor(int argc, char *argv[]);
36
37 int sensor(int argc, char *argv[])
38 {
39   gras_socket_t mysock;
40   gras_socket_t master = NULL;
41   int connection_try = 10;
42
43   gras_init(&argc, argv);
44   amok_bw_init();
45   amok_pm_init();
46
47   mysock = gras_socket_server_range(3000, 9999, 0, 0);
48   XBT_INFO("Sensor starting (on port %d)", gras_os_myport());
49   while (connection_try > 0 &&
50          !(master = try_gras_socket_client_from_string(argv[1]))) {
51     connection_try--;
52     gras_os_sleep(0.5);       /* let the master get ready */
53   }
54
55   amok_pm_group_join(master, "bandwidth");
56   amok_pm_mainloop(60);
57
58   gras_socket_close(mysock);
59   gras_socket_close(master);
60   gras_exit();
61   return 0;
62 }
63
64 /* **********************************************************************
65  * Maestro code
66  * **********************************************************************/
67
68 /* Function prototypes */
69 int maestro(int argc, char *argv[]);
70
71 int maestro(int argc, char *argv[])
72 {
73   double sec, bw;
74   int buf_size = 32 * 1024;
75   int msg_size = 512 * 1024;
76   int msg_amount = 1;
77   double min_duration = 1;
78
79   gras_socket_t peer;
80   gras_socket_t mysock;
81   xbt_peer_t h1, h2, h_temp;
82   xbt_dynar_t group;
83
84   gras_init(&argc, argv);
85   amok_bw_init();
86   amok_pm_init();
87
88   XBT_INFO("Maestro starting");
89   if (argc != 2) {
90     XBT_ERROR("Usage: maestro port\n");
91     return 1;
92   }
93   mysock = gras_socket_server(atoi(argv[1]));
94   group = amok_pm_group_new("bandwidth");
95   XBT_INFO("Wait for peers for 5 sec");
96   gras_msg_handleall(5);        /* friends, we're ready. Come and play */
97
98   if (xbt_dynar_length(group) < 2) {
99     amok_pm_group_shutdown("bandwidth");
100     xbt_die("Not enough peers arrived. Expected 2 got %ld",
101             xbt_dynar_length(group));
102   }
103   h1 = *(xbt_peer_t *) xbt_dynar_get_ptr(group, 0);
104   h2 = *(xbt_peer_t *) xbt_dynar_get_ptr(group, 1);
105   /* sort peers in right order to keep output right */
106   if (strcmp(h1->name, h2->name) < 0 || h1->port > h2->port) {
107     h_temp = h1;
108     h1 = h2;
109     h2 = h_temp;
110   }
111
112   XBT_INFO("Contact %s:%d", h1->name, h1->port);
113   peer = gras_socket_client(h1->name, h1->port);
114
115   XBT_INFO("Test the BW between me and one of the sensors");
116   amok_bw_test(peer, buf_size, msg_size, msg_amount, min_duration, &sec,
117                &bw);
118   XBT_INFO
119       ("Experience between me and %s:%d (initially %d msgs of %d bytes, maybe modified to fill the pipe at least %.1fs) took %f sec, achieving %f kb/s",
120        h1->name, h1->port, msg_amount, msg_size, min_duration, sec,
121        ((double) bw) / 1024.0);
122
123   XBT_INFO("Test the BW between %s:%d and %s:%d", h1->name, h1->port,
124         h2->name, h2->port);
125   amok_bw_request(h1->name, h1->port, h2->name, h2->port, buf_size,
126                   msg_size, msg_amount, min_duration, &sec, &bw);
127   XBT_INFO
128       ("Experience between %s:%d and %s:%d took took %f sec, achieving %f kb/s",
129        h1->name, h1->port, h2->name, h2->port, sec,
130        ((double) bw) / 1024.0);
131
132   /* Game is over, friends */
133   amok_pm_group_shutdown("bandwidth");
134
135   gras_socket_close(mysock);
136   gras_exit();
137   return 0;
138 }