Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Some more debugging messages, and shorter delays 'cause I hate waiting in RL
[simgrid.git] / examples / amok / bandwidth / bandwidth.c
1 /* $Id$ */
2
3 /* bandwidth - bandwidth test demo of GRAS features                         */
4
5 /* Copyright (c) 2003, 2004 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
11 #include "gras.h"
12 #include "amok/bandwidth.h"
13 #include "amok/hostmanagement.h"
14
15 XBT_LOG_NEW_DEFAULT_CATEGORY(Bandwidth,"Messages specific to this example");
16
17 /* **********************************************************************
18  * Sensor code
19  * **********************************************************************/
20
21 /* Function prototypes */
22 int sensor (int argc,char *argv[]);
23
24 int sensor (int argc,char *argv[]) {
25   gras_socket_t mysock;
26   gras_socket_t master;
27
28   gras_init(&argc, argv);
29   amok_bw_init();
30   amok_hm_init();
31  
32   mysock = gras_socket_server_range(3000,9999,0,0);
33   INFO1("Sensor starting (on port %d)",gras_os_myport());
34   gras_os_sleep(0.5); /* let the master get ready */
35   master = gras_socket_client_from_string(argv[1]);
36                                               
37   amok_hm_group_join(master,"bandwidth");
38   amok_hm_mainloop(60);
39
40   gras_socket_close(mysock);
41   gras_socket_close(master);
42   gras_exit();
43   return 0;
44 }
45
46 /* **********************************************************************
47  * Maestro code
48  * **********************************************************************/
49
50 /* Function prototypes */
51 int maestro (int argc,char *argv[]);
52
53 int maestro(int argc,char *argv[]) {
54   double sec, bw;
55   int buf_size=32  *1024;
56   int exp_size=512 *1024;
57   int msg_size=512 *1024;
58   double min_duration = 1;
59    
60   gras_socket_t peer;
61   gras_socket_t mysock;
62   xbt_host_t h1,h2;
63   xbt_dynar_t group;
64
65   gras_init(&argc, argv);
66   amok_bw_init();
67   amok_hm_init();
68
69   INFO0("Maestro starting");
70   if (argc != 2) {
71      ERROR0("Usage: maestro port\n");
72      return 1;
73   }
74   mysock=gras_socket_server(atoi(argv[1]));
75   group=amok_hm_group_new("bandwidth");
76   INFO0("Wait for peers");
77   gras_msg_handleall(10); /* friends, we're ready. Come and play */
78    
79   if (xbt_dynar_length(group) < 2) {
80      char *msg;
81      asprintf(&msg,"Not enough peers arrived. Expected 2 got %ld",
82               xbt_dynar_length(group));
83      xbt_die(msg);
84   }
85   h1 = *(xbt_host_t*) xbt_dynar_get_ptr(group, 0);
86   h2 = *(xbt_host_t*)xbt_dynar_get_ptr(group, 1);
87
88   INFO2("Contact %s:%d",h1->name, h1->port);
89   peer = gras_socket_client(h1->name, h1->port);
90
91   INFO0("Test the BW between me and one of the sensors");  
92   amok_bw_test(peer,buf_size,exp_size,msg_size,min_duration,&sec,&bw);
93   INFO6("Experience between me and %s:%d (%d bytes in msgs of %d bytes) took %f sec, achieving %f kb/s",
94         h1->name, h1->port,
95         exp_size,msg_size,
96         sec,((double)bw)/1024.0);
97
98   INFO4("Test the BW between %s:%d and %s:%d",  h1->name, h1->port,     h2->name, h2->port);
99   amok_bw_request(h1->name, h1->port,   h2->name, h2->port,
100                   buf_size,exp_size,msg_size,min_duration,&sec,&bw);
101   INFO6("Experience between %s:%d and %s:%d took took %f sec, achieving %f kb/s",
102         h1->name, h1->port,     h2->name, h2->port,
103         sec,((double)bw)/1024.0);
104
105   /* Game is over, friends */
106   amok_hm_group_shutdown ("bandwidth");
107
108   gras_socket_close(mysock);
109   gras_exit();
110   return 0;
111 }