Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
reindent ... this time with the appropriate options
[simgrid.git] / examples / msg / bittorrent / bittorrent_platfgen.c
1 /* Copyright (c) 2012. The SimGrid Team.
2  * All rights reserved.                                                     */
3 /* This program is free software; you can redistribute it and/or modify it
4  * under the terms of the license (GNU LGPL) which comes with this package. */
5 #include "bittorrent.h"
6 #include "peer.h"
7 #include "tracker.h"
8 #include <msg/msg.h>
9 #include <simgrid/platf_generator.h>
10 /**
11  * Bittorrent example launcher, using a generated platform
12  */
13
14 static RngStream rng_stream;
15
16 void promoter(context_node_t node);
17 void labeler(context_edge_t edge);
18 void create_environment(int node_count);
19 void dispatch_jobs(double tracker_deadline, double peer_deadline,
20                    double seed_percentage);
21
22 void promoter(context_node_t node)
23 {
24   s_sg_platf_host_cbarg_t host_parameters;
25
26   if (node->degree == 1) {
27     //We promote only the leaf; as we use a star topology, all the nodes
28     //will be promoted except the first one, which will be a router with
29     //every hosts connected on.
30     host_parameters.id = NULL;
31
32     //Power from 3,000,000 to 10,000,000
33     host_parameters.power_peak =
34         7000000 * RngStream_RandU01(rng_stream) + 3000000;
35     host_parameters.core_amount = 1;
36     host_parameters.power_scale = 1;
37     host_parameters.power_trace = NULL;
38     host_parameters.initial_state = SURF_RESOURCE_ON;
39     host_parameters.state_trace = NULL;
40     host_parameters.coord = NULL;
41     host_parameters.properties = NULL;
42
43     platf_graph_promote_to_host(node, &host_parameters);
44   }
45 }
46
47 void labeler(context_edge_t edge)
48 {
49
50   s_sg_platf_link_cbarg_t link_parameters;
51   link_parameters.id = NULL;
52
53   //bandwidth from 3,000,000 to 10,000,000
54   link_parameters.bandwidth = 7000000 * RngStream_RandU01(rng_stream) + 3000000;
55   link_parameters.bandwidth_trace = NULL;
56
57   //Latency from 0ms to 100ms
58   link_parameters.latency = RngStream_RandU01(rng_stream) / 10.0;
59   link_parameters.latency_trace = NULL;
60   link_parameters.state = SURF_RESOURCE_ON;
61   link_parameters.state_trace = NULL;
62   link_parameters.policy = SURF_LINK_SHARED;
63   link_parameters.properties = NULL;
64
65   platf_graph_link_label(edge, &link_parameters);
66 }
67
68 void create_environment(int node_count)
69 {
70
71   platf_graph_uniform(node_count);
72
73   //every nodes are connected to the first one
74   platf_graph_interconnect_star();
75   //No need to check if the graph is connected, the star topology implies it.
76
77   //register promoter and labeler
78   platf_graph_promoter(promoter);
79   platf_graph_labeler(labeler);
80
81   //promoting and labeling
82   platf_do_promote();
83   platf_do_label();
84
85   //Put the platform into the simulator
86   platf_generate();
87 }
88
89 void dispatch_jobs(double tracker_deadline, double peer_deadline,
90                    double seed_percentage)
91 {
92
93   xbt_dynar_t available_nodes = MSG_hosts_as_dynar();
94   msg_host_t host;
95   unsigned int i;
96
97   char **arguments_tracker;
98   char **arguments_peer;
99
100   unsigned int seed_count =
101       (seed_percentage / 100.0) * xbt_dynar_length(available_nodes);
102
103   xbt_dynar_foreach(available_nodes, i, host) {
104     if (i == 0) {
105       //The fisrt node is the tracker
106       arguments_tracker = xbt_malloc0(sizeof(char *) * 2);
107       arguments_tracker[0] = xbt_strdup("tracker");
108       arguments_tracker[1] = bprintf("%f", tracker_deadline);
109       MSG_process_create_with_arguments("tracker", tracker, NULL, host, 2,
110                                         arguments_tracker);
111     } else {
112       //Other nodes are peers
113       int argument_size;
114       arguments_peer = xbt_malloc0(sizeof(char *) * 4);
115       arguments_peer[0] = xbt_strdup("peer");
116       arguments_peer[1] = bprintf("%d", i);
117       arguments_peer[2] = bprintf("%f", peer_deadline);
118
119       //The first peers will be seeders
120       if (seed_count > 0) {
121         seed_count--;
122         arguments_peer[3] = xbt_strdup("1");
123         argument_size = 4;
124       } else {
125         //Other ars leechers
126         arguments_peer[3] = NULL;
127         argument_size = 3;
128       }
129       MSG_process_create_with_arguments("peer", peer, NULL, host,
130                                         argument_size, arguments_peer);
131     }
132   }
133 }
134
135 int main(int argc, char *argv[])
136 {
137   MSG_init(&argc, argv);
138
139   rng_stream = RngStream_CreateStream(NULL);
140
141   //Maybe these parameters should be set from the command line...
142   //create_environment(<node_count>)
143   create_environment(20);
144
145   //dispatch_jobs(<tracker_deadline>, <peer_deadline>, <seed_percentage>)
146   dispatch_jobs(2000, 2000, 10);
147
148   MSG_main();
149
150   return 0;
151 }