Logo AND Algorithmique Numérique Distribuée

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