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 }