examples/planetlab/scalability.py

   1 #!/usr/bin/env python
   2 #
   3 #    NEPI, a framework to manage network experiments
   4 #    Copyright (C) 2013 INRIA
   5 #
   6 #    This program is free software: you can redistribute it and/or modify
   7 #    it under the terms of the GNU General Public License as published by
   8 #    the Free Software Foundation, either version 3 of the License, or
   9 #    (at your option) any later version.
  10 #
  11 #    This program is distributed in the hope that it will be useful,
  12 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 #    GNU General Public License for more details.
  15 #
  16 #    You should have received a copy of the GNU General Public License
  17 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 #
  19 # Author: Lucia Guevgeozian <lucia.guevgeozian_odizzio@inria.fr>
  20
  21 from nepi.execution.ec import ExperimentController
  22 from nepi.execution.resource import ResourceAction, ResourceState
  23
  24 import os
  25 import time
  26
  27 def create_node(ec, username, pl_user, pl_password, hostname=None, country=None,
  28                 operatingSystem=None, minBandwidth=None, minCpu=None):
  29
  30     node = ec.register_resource("PlanetlabNode")
  31
  32     if username:
  33         ec.set(node, "username", username)
  34     if pl_user:
  35         ec.set(node, "pluser", pl_user)
  36     if pl_password:
  37         ec.set(node, "plpassword", pl_password)
  38
  39     if hostname:
  40         ec.set(node, "hostname", hostname)
  41     if country:
  42         ec.set(node, "country", country)
  43     if operatingSystem:
  44         ec.set(node, "operatingSystem", operatingSystem)
  45     if minBandwidth:
  46         ec.set(node, "minBandwidth", minBandwidth)
  47     if minCpu:
  48         ec.set(node, "minCpu", minCpu)
  49
  50     ec.set(node, "cleanHome", True)
  51     ec.set(node, "cleanProcesses", True)
  52
  53     return node
  54
  55 exp_id = "scalability_exp"
  56
  57 # Create the entity Experiment Controller:
  58 ec = ExperimentController(exp_id)
  59
  60 # Register the nodes resources:
  61
  62 # The username in this case is the slice name, the one to use for login in
  63 # via ssh into PlanetLab nodes. Replace with your own slice name.
  64 username = os.environ.get("PL_SLICE")
  65
  66 # The pluser and plpassword are the ones used to login in the PlanetLab web
  67 # site. Replace with your own user and password account information.
  68 pl_user = os.environ.get("PL_USER")
  69 pl_password =  os.environ.get("PL_PASS")
  70
  71 # Choose the PlanetLab nodes for the experiment, in this example 5 nodes are
  72 # used, and they are picked according to different criterias.
  73
  74 first_set_nodes = []
  75 second_set_nodes = []
  76 third_set_nodes = []
  77
  78 # First set of nodes will be defined by its hostname.
  79 hostnames = ['planetlab2.utt.fr',
  80 'planetlab-2.man.poznan.pl',
  81 'planetlab-1.ing.unimo.it',
  82 'gschembra3.diit.unict.it',
  83 'planetlab2.ionio.gr',
  84 'planetlab-1.imag.fr',
  85 'node2pl.planet-lab.telecom-lille1.eu',
  86 'planetlab1.xeno.cl.cam.ac.uk',
  87 'planetlab3.hiit.fi',
  88 'planetlab2.fri.uni-lj.si',
  89 'planetlab1.informatik.uni-erlangen.de',
  90 'node1pl.planet-lab.telecom-lille1.eu',
  91 'planet2.servers.ua.pt',
  92 'iraplab1.iralab.uni-karlsruhe.de',
  93 'planetlab-node3.it-sudparis.eu',
  94 'planet1.elte.hu',
  95 'planet1.l3s.uni-hannover.de',
  96 'planetlab1.fct.ualg.pt',
  97 'host1.planetlab.informatik.tu-darmstadt.de',
  98 'vicky.planetlab.ntua.gr',
  99 'planetlab1.rutgers.edu']
 100
 101 for hostname in hostnames:
 102     node = create_node(ec, username, pl_user, pl_password, hostname=hostname)
 103     first_set_nodes.append(node)
 104
 105 # Second set of nodes will be U.S.A. nodes.
 106 country = "United States"
 107 for i in range(15):
 108     node = create_node(ec, username, pl_user, pl_password, country=country)
 109     second_set_nodes.append(node)
 110
 111 # Third set of nodes can be any node in Planetlab testbed.
 112 for i in range(70):
 113     node = create_node(ec, username, pl_user, pl_password)
 114     third_set_nodes.append(node)
 115
 116 # Register conditions
 117
 118 # The nodes that are completely identified by their hostnames have to be provisioned
 119 # before the rest of the nodes. This assures that no other resource will use the
 120 # identified node even if the constraints matchs.
 121 # In this example the nodes from the first need to be provisioned before the ones in
 122 # the second and third group. Using the same logic, is convenient that nodes from the
 123 # second group are provisioned before nodes from the third group.
 124
 125 ec.register_condition(second_set_nodes, ResourceAction.DEPLOY, first_set_nodes, ResourceState.PROVISIONED)
 126 ec.register_condition(third_set_nodes, ResourceAction.DEPLOY, second_set_nodes, ResourceState.PROVISIONED)
 127
 128 # Deploy the experiment:
 129 ec.deploy()
 130
 131 # Wait until all nodes are provisioned:
 132 ec.wait_deployed(first_set_nodes)
 133 ec.wait_deployed(second_set_nodes)
 134 ec.wait_deployed(third_set_nodes)
 135
 136 # Do the experiment controller shutdown:
 137 ec.shutdown()
 138
 139 # END