--- /dev/null
+<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"
+ "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
+
+
+<html xmlns="http://www.w3.org/1999/xhtml">
+ <head>
+ <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
+
+ <title>nepi.resources.omf.wilabt_node — NEPI 3.0 documentation</title>
+
+ <link rel="stylesheet" href="../../../../_static/sphinxdoc.css" type="text/css" />
+ <link rel="stylesheet" href="../../../../_static/pygments.css" type="text/css" />
+
+ <script type="text/javascript">
+ var DOCUMENTATION_OPTIONS = {
+ URL_ROOT: '../../../../',
+ VERSION: '3.0',
+ COLLAPSE_INDEX: false,
+ FILE_SUFFIX: '.html',
+ HAS_SOURCE: true
+ };
+ </script>
+ <script type="text/javascript" src="../../../../_static/jquery.js"></script>
+ <script type="text/javascript" src="../../../../_static/underscore.js"></script>
+ <script type="text/javascript" src="../../../../_static/doctools.js"></script>
+ <link rel="top" title="NEPI 3.0 documentation" href="../../../../index.html" />
+ <link rel="up" title="Module code" href="../../../index.html" />
+ </head>
+ <body>
+ <div class="related">
+ <h3>Navigation</h3>
+ <ul>
+ <li class="right" style="margin-right: 10px">
+ <a href="../../../../genindex.html" title="General Index"
+ accesskey="I">index</a></li>
+ <li class="right" >
+ <a href="../../../../py-modindex.html" title="Python Module Index"
+ >modules</a> |</li>
+ <li><a href="../../../../index.html">NEPI 3.0 documentation</a> »</li>
+ <li><a href="../../../index.html" accesskey="U">Module code</a> »</li>
+ </ul>
+ </div>
+ <div class="sphinxsidebar">
+ <div class="sphinxsidebarwrapper">
+<div id="searchbox" style="display: none">
+ <h3>Quick search</h3>
+ <form class="search" action="../../../../search.html" method="get">
+ <input type="text" name="q" />
+ <input type="submit" value="Go" />
+ <input type="hidden" name="check_keywords" value="yes" />
+ <input type="hidden" name="area" value="default" />
+ </form>
+ <p class="searchtip" style="font-size: 90%">
+ Enter search terms or a module, class or function name.
+ </p>
+</div>
+<script type="text/javascript">$('#searchbox').show(0);</script>
+ </div>
+ </div>
+
+ <div class="document">
+ <div class="documentwrapper">
+ <div class="bodywrapper">
+ <div class="body">
+
+ <h1>Source code for nepi.resources.omf.wilabt_node</h1><div class="highlight"><pre>
+<span class="c">#</span>
+<span class="c"># NEPI, a framework to manage network experiments</span>
+<span class="c"># Copyright (C) 2013 INRIA</span>
+<span class="c">#</span>
+<span class="c"># This program is free software: you can redistribute it and/or modify</span>
+<span class="c"># it under the terms of the GNU General Public License version 2 as</span>
+<span class="c"># published by the Free Software Foundation;</span>
+<span class="c">#</span>
+<span class="c"># This program is distributed in the hope that it will be useful,</span>
+<span class="c"># but WITHOUT ANY WARRANTY; without even the implied warranty of</span>
+<span class="c"># MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the</span>
+<span class="c"># GNU General Public License for more details.</span>
+<span class="c">#</span>
+<span class="c"># You should have received a copy of the GNU General Public License</span>
+<span class="c"># along with this program. If not, see <http://www.gnu.org/licenses/>.</span>
+<span class="c">#</span>
+<span class="c"># Author: Lucia Guevgeozian <lucia.guevgeozian_odizzio@inria.fr></span>
+
+<span class="kn">from</span> <span class="nn">nepi.execution.attribute</span> <span class="kn">import</span> <span class="n">Attribute</span><span class="p">,</span> <span class="n">Flags</span><span class="p">,</span> <span class="n">Types</span>
+<span class="kn">from</span> <span class="nn">nepi.execution.resource</span> <span class="kn">import</span> <span class="n">ResourceManager</span><span class="p">,</span> <span class="n">clsinit_copy</span><span class="p">,</span> \
+ <span class="n">ResourceState</span>
+<span class="kn">from</span> <span class="nn">nepi.resources.omf.node</span> <span class="kn">import</span> <span class="n">OMFNode</span>
+<span class="kn">from</span> <span class="nn">nepi.util.sfaapi</span> <span class="kn">import</span> <span class="n">SFAAPIFactory</span>
+<span class="kn">from</span> <span class="nn">nepi.util.execfuncs</span> <span class="kn">import</span> <span class="n">lexec</span>
+<span class="kn">from</span> <span class="nn">nepi.util</span> <span class="kn">import</span> <span class="n">sshfuncs</span>
+
+<span class="kn">from</span> <span class="nn">random</span> <span class="kn">import</span> <span class="n">randint</span>
+<span class="kn">import</span> <span class="nn">time</span>
+<span class="kn">import</span> <span class="nn">re</span>
+<span class="kn">import</span> <span class="nn">weakref</span>
+<span class="kn">import</span> <span class="nn">socket</span>
+<span class="kn">import</span> <span class="nn">threading</span>
+<span class="kn">import</span> <span class="nn">datetime</span>
+
+<span class="nd">@clsinit_copy</span>
+<div class="viewcode-block" id="WilabtSfaNode"><a class="viewcode-back" href="../../../../_layout/nepi.resources.omf.html#nepi.resources.omf.wilabt_node.WilabtSfaNode">[docs]</a><span class="k">class</span> <span class="nc">WilabtSfaNode</span><span class="p">(</span><span class="n">OMFNode</span><span class="p">):</span>
+ <span class="n">_rtype</span> <span class="o">=</span> <span class="s">"wilabt::sfa::Node"</span>
+ <span class="n">_help</span> <span class="o">=</span> <span class="s">"Controls a Wilabt host accessible using a SSH key "</span> \
+ <span class="s">"and provisioned using SFA"</span>
+ <span class="n">_platform</span> <span class="o">=</span> <span class="s">"omf"</span>
+
+ <span class="nd">@classmethod</span>
+ <span class="k">def</span> <span class="nf">_register_attributes</span><span class="p">(</span><span class="n">cls</span><span class="p">):</span>
+
+ <span class="n">username</span> <span class="o">=</span> <span class="n">Attribute</span><span class="p">(</span><span class="s">"username"</span><span class="p">,</span> <span class="s">"Local account username"</span><span class="p">,</span>
+ <span class="n">flags</span> <span class="o">=</span> <span class="n">Flags</span><span class="o">.</span><span class="n">Credential</span><span class="p">)</span>
+
+ <span class="n">identity</span> <span class="o">=</span> <span class="n">Attribute</span><span class="p">(</span><span class="s">"identity"</span><span class="p">,</span> <span class="s">"SSH identity file"</span><span class="p">,</span>
+ <span class="n">flags</span> <span class="o">=</span> <span class="n">Flags</span><span class="o">.</span><span class="n">Credential</span><span class="p">)</span>
+
+ <span class="n">server_key</span> <span class="o">=</span> <span class="n">Attribute</span><span class="p">(</span><span class="s">"serverKey"</span><span class="p">,</span> <span class="s">"Server public key"</span><span class="p">,</span>
+ <span class="n">flags</span> <span class="o">=</span> <span class="n">Flags</span><span class="o">.</span><span class="n">Design</span><span class="p">)</span>
+
+ <span class="n">sfa_user</span> <span class="o">=</span> <span class="n">Attribute</span><span class="p">(</span><span class="s">"sfauser"</span><span class="p">,</span> <span class="s">"SFA user"</span><span class="p">,</span>
+ <span class="n">flags</span> <span class="o">=</span> <span class="n">Flags</span><span class="o">.</span><span class="n">Credential</span><span class="p">)</span>
+
+ <span class="n">sfa_private_key</span> <span class="o">=</span> <span class="n">Attribute</span><span class="p">(</span><span class="s">"sfaPrivateKey"</span><span class="p">,</span> <span class="s">"SFA path to the private key </span><span class="se">\</span>
+<span class="s"> used to generate the user credential"</span><span class="p">,</span>
+ <span class="n">flags</span> <span class="o">=</span> <span class="n">Flags</span><span class="o">.</span><span class="n">Credential</span><span class="p">)</span>
+
+ <span class="n">slicename</span> <span class="o">=</span> <span class="n">Attribute</span><span class="p">(</span><span class="s">"slicename"</span><span class="p">,</span> <span class="s">"SFA slice for the experiment"</span><span class="p">,</span>
+ <span class="n">flags</span> <span class="o">=</span> <span class="n">Flags</span><span class="o">.</span><span class="n">Credential</span><span class="p">)</span>
+
+ <span class="n">gateway_user</span> <span class="o">=</span> <span class="n">Attribute</span><span class="p">(</span><span class="s">"gatewayUser"</span><span class="p">,</span> <span class="s">"Gateway account username"</span><span class="p">,</span>
+ <span class="n">flags</span> <span class="o">=</span> <span class="n">Flags</span><span class="o">.</span><span class="n">Design</span><span class="p">)</span>
+
+ <span class="n">gateway</span> <span class="o">=</span> <span class="n">Attribute</span><span class="p">(</span><span class="s">"gateway"</span><span class="p">,</span> <span class="s">"Hostname of the gateway machine"</span><span class="p">,</span>
+ <span class="n">flags</span> <span class="o">=</span> <span class="n">Flags</span><span class="o">.</span><span class="n">Design</span><span class="p">)</span>
+
+ <span class="n">host</span> <span class="o">=</span> <span class="n">Attribute</span><span class="p">(</span><span class="s">"host"</span><span class="p">,</span> <span class="s">"Name of the physical machine"</span><span class="p">,</span>
+ <span class="n">flags</span> <span class="o">=</span> <span class="n">Flags</span><span class="o">.</span><span class="n">Design</span><span class="p">)</span>
+
+ <span class="n">disk_image</span> <span class="o">=</span> <span class="n">Attribute</span><span class="p">(</span><span class="s">"disk_image"</span><span class="p">,</span> <span class="s">"Specify a specific disk image for a node"</span><span class="p">,</span>
+ <span class="n">flags</span> <span class="o">=</span> <span class="n">Flags</span><span class="o">.</span><span class="n">Design</span><span class="p">)</span>
+
+ <span class="n">cls</span><span class="o">.</span><span class="n">_register_attribute</span><span class="p">(</span><span class="n">username</span><span class="p">)</span>
+ <span class="n">cls</span><span class="o">.</span><span class="n">_register_attribute</span><span class="p">(</span><span class="n">identity</span><span class="p">)</span>
+ <span class="n">cls</span><span class="o">.</span><span class="n">_register_attribute</span><span class="p">(</span><span class="n">server_key</span><span class="p">)</span>
+ <span class="n">cls</span><span class="o">.</span><span class="n">_register_attribute</span><span class="p">(</span><span class="n">sfa_user</span><span class="p">)</span>
+ <span class="n">cls</span><span class="o">.</span><span class="n">_register_attribute</span><span class="p">(</span><span class="n">sfa_private_key</span><span class="p">)</span>
+ <span class="n">cls</span><span class="o">.</span><span class="n">_register_attribute</span><span class="p">(</span><span class="n">slicename</span><span class="p">)</span>
+ <span class="n">cls</span><span class="o">.</span><span class="n">_register_attribute</span><span class="p">(</span><span class="n">gateway_user</span><span class="p">)</span>
+ <span class="n">cls</span><span class="o">.</span><span class="n">_register_attribute</span><span class="p">(</span><span class="n">gateway</span><span class="p">)</span>
+ <span class="n">cls</span><span class="o">.</span><span class="n">_register_attribute</span><span class="p">(</span><span class="n">host</span><span class="p">)</span>
+ <span class="n">cls</span><span class="o">.</span><span class="n">_register_attribute</span><span class="p">(</span><span class="n">disk_image</span><span class="p">)</span>
+
+ <span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">ec</span><span class="p">,</span> <span class="n">guid</span><span class="p">):</span>
+ <span class="nb">super</span><span class="p">(</span><span class="n">WilabtSfaNode</span><span class="p">,</span> <span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="n">__init__</span><span class="p">(</span><span class="n">ec</span><span class="p">,</span> <span class="n">guid</span><span class="p">)</span>
+
+ <span class="bp">self</span><span class="o">.</span><span class="n">_ecobj</span> <span class="o">=</span> <span class="n">weakref</span><span class="o">.</span><span class="n">ref</span><span class="p">(</span><span class="n">ec</span><span class="p">)</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">_sfaapi</span> <span class="o">=</span> <span class="bp">None</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">_node_to_provision</span> <span class="o">=</span> <span class="bp">None</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">_slicenode</span> <span class="o">=</span> <span class="bp">False</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">_host</span> <span class="o">=</span> <span class="bp">False</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">_username</span> <span class="o">=</span> <span class="bp">None</span>
+
+ <span class="k">def</span> <span class="nf">_skip_provision</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
+ <span class="n">sfa_user</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">"sfauser"</span><span class="p">)</span>
+ <span class="k">if</span> <span class="ow">not</span> <span class="n">sfa_user</span><span class="p">:</span>
+ <span class="k">return</span> <span class="bp">True</span>
+ <span class="k">else</span><span class="p">:</span> <span class="k">return</span> <span class="bp">False</span>
+
+ <span class="nd">@property</span>
+<div class="viewcode-block" id="WilabtSfaNode.sfaapi"><a class="viewcode-back" href="../../../../_layout/nepi.resources.omf.html#nepi.resources.omf.wilabt_node.WilabtSfaNode.sfaapi">[docs]</a> <span class="k">def</span> <span class="nf">sfaapi</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Property to instanciate the SFA API based in sfi client.</span>
+<span class="sd"> For each SFA method called this instance is used.</span>
+<span class="sd"> """</span>
+ <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sfaapi</span><span class="p">:</span>
+ <span class="n">sfa_user</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">"sfauser"</span><span class="p">)</span>
+ <span class="n">sfa_sm</span> <span class="o">=</span> <span class="s">"http://www.wilab2.ilabt.iminds.be:12369/protogeni/xmlrpc/am/3.0"</span>
+ <span class="n">sfa_auth</span> <span class="o">=</span> <span class="s">'.'</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">sfa_user</span><span class="o">.</span><span class="n">split</span><span class="p">(</span><span class="s">'.'</span><span class="p">)[:</span><span class="mi">2</span><span class="p">])</span>
+ <span class="n">sfa_registry</span> <span class="o">=</span> <span class="s">"http://sfa3.planet-lab.eu:12345/"</span>
+ <span class="n">sfa_private_key</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">"sfaPrivateKey"</span><span class="p">)</span>
+ <span class="n">batch</span> <span class="o">=</span> <span class="bp">True</span>
+
+ <span class="n">_sfaapi</span> <span class="o">=</span> <span class="n">SFAAPIFactory</span><span class="o">.</span><span class="n">get_api</span><span class="p">(</span><span class="n">sfa_user</span><span class="p">,</span> <span class="n">sfa_auth</span><span class="p">,</span>
+ <span class="n">sfa_registry</span><span class="p">,</span> <span class="n">sfa_sm</span><span class="p">,</span> <span class="n">sfa_private_key</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_ecobj</span><span class="p">(),</span> <span class="n">batch</span><span class="p">,</span> <span class="n">WilabtSfaNode</span><span class="o">.</span><span class="n">_rtype</span><span class="p">)</span>
+
+ <span class="k">if</span> <span class="ow">not</span> <span class="n">_sfaapi</span><span class="p">:</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">fail_sfaapi</span><span class="p">()</span>
+
+ <span class="bp">self</span><span class="o">.</span><span class="n">_sfaapi</span> <span class="o">=</span> <span class="n">weakref</span><span class="o">.</span><span class="n">ref</span><span class="p">(</span><span class="n">_sfaapi</span><span class="p">)</span>
+
+ <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sfaapi</span><span class="p">()</span>
+</div>
+<div class="viewcode-block" id="WilabtSfaNode.do_discover"><a class="viewcode-back" href="../../../../_layout/nepi.resources.omf.html#nepi.resources.omf.wilabt_node.WilabtSfaNode.do_discover">[docs]</a> <span class="k">def</span> <span class="nf">do_discover</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Based on the attributes defined by the user, discover the suitable </span>
+<span class="sd"> node for provision.</span>
+<span class="sd"> """</span>
+ <span class="n">nodes</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sfaapi</span><span class="o">.</span><span class="n">get_resources_hrn</span><span class="p">()</span>
+
+ <span class="n">host</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_get_host</span><span class="p">()</span>
+ <span class="k">if</span> <span class="n">host</span><span class="p">:</span>
+ <span class="c"># the user specified one particular node to be provisioned</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">_host</span> <span class="o">=</span> <span class="bp">True</span>
+ <span class="n">host_hrn</span> <span class="o">=</span> <span class="n">nodes</span><span class="p">[</span><span class="n">host</span><span class="p">]</span>
+
+ <span class="c"># check that the node is not blacklisted or being provisioned</span>
+ <span class="c"># by other RM</span>
+ <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_blacklisted</span><span class="p">(</span><span class="n">host_hrn</span><span class="p">):</span>
+ <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_reserved</span><span class="p">(</span><span class="n">host_hrn</span><span class="p">):</span>
+ <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">_check_if_in_slice</span><span class="p">([</span><span class="n">host_hrn</span><span class="p">]):</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">debug</span><span class="p">(</span><span class="s">"Node already in slice </span><span class="si">%s</span><span class="s">"</span> <span class="o">%</span> <span class="n">host_hrn</span><span class="p">)</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">_slicenode</span> <span class="o">=</span> <span class="bp">True</span>
+ <span class="n">host</span> <span class="o">=</span> <span class="n">host</span> <span class="o">+</span> <span class="s">'.wilab2.ilabt.iminds.be'</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">set</span><span class="p">(</span><span class="s">'host'</span><span class="p">,</span> <span class="n">host</span><span class="p">)</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">_node_to_provision</span> <span class="o">=</span> <span class="n">host_hrn</span>
+ <span class="nb">super</span><span class="p">(</span><span class="n">WilabtSfaNode</span><span class="p">,</span> <span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="n">do_discover</span><span class="p">()</span>
+</div>
+<div class="viewcode-block" id="WilabtSfaNode.do_provision"><a class="viewcode-back" href="../../../../_layout/nepi.resources.omf.html#nepi.resources.omf.wilabt_node.WilabtSfaNode.do_provision">[docs]</a> <span class="k">def</span> <span class="nf">do_provision</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Add node to user's slice and verifing that the node is functioning</span>
+<span class="sd"> correctly. Check ssh, omf rc running, hostname, file system.</span>
+<span class="sd"> """</span>
+ <span class="n">provision_ok</span> <span class="o">=</span> <span class="bp">False</span>
+ <span class="n">ssh_ok</span> <span class="o">=</span> <span class="bp">False</span>
+ <span class="n">proc_ok</span> <span class="o">=</span> <span class="bp">False</span>
+ <span class="n">timeout</span> <span class="o">=</span> <span class="mi">300</span>
+
+ <span class="k">while</span> <span class="ow">not</span> <span class="n">provision_ok</span><span class="p">:</span>
+ <span class="n">node</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_node_to_provision</span>
+ <span class="c">#if self._slicenode:</span>
+ <span class="c"># self._delete_from_slice()</span>
+ <span class="c"># self.debug("Waiting 480 sec for re-adding to slice")</span>
+ <span class="c"># time.sleep(480) # Timout for the testbed to allow a new reservation</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">_add_node_to_slice</span><span class="p">(</span><span class="n">node</span><span class="p">)</span>
+ <span class="n">t</span> <span class="o">=</span> <span class="mi">0</span>
+ <span class="k">while</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_check_if_in_slice</span><span class="p">([</span><span class="n">node</span><span class="p">])</span> <span class="ow">and</span> <span class="n">t</span> <span class="o"><</span> <span class="n">timeout</span> \
+ <span class="ow">and</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_ecobj</span><span class="p">()</span><span class="o">.</span><span class="n">abort</span><span class="p">:</span>
+ <span class="n">t</span> <span class="o">=</span> <span class="n">t</span> <span class="o">+</span> <span class="mi">5</span>
+ <span class="n">time</span><span class="o">.</span><span class="n">sleep</span><span class="p">(</span><span class="n">t</span><span class="p">)</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">debug</span><span class="p">(</span><span class="s">"Waiting 5 sec for resources to be added"</span><span class="p">)</span>
+ <span class="k">continue</span>
+
+ <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_check_if_in_slice</span><span class="p">([</span><span class="n">node</span><span class="p">]):</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">debug</span><span class="p">(</span><span class="s">"Couldn't add node </span><span class="si">%s</span><span class="s"> to slice"</span> <span class="o">%</span> <span class="n">node</span><span class="p">)</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">fail_node_not_available</span><span class="p">(</span><span class="n">node</span><span class="p">)</span>
+
+ <span class="bp">self</span><span class="o">.</span><span class="n">_get_username</span><span class="p">()</span>
+ <span class="n">ssh_ok</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_check_ssh_loop</span><span class="p">()</span>
+
+ <span class="k">if</span> <span class="ow">not</span> <span class="n">ssh_ok</span><span class="p">:</span>
+ <span class="c"># the timeout was reach without establishing ssh connection</span>
+ <span class="c"># the node is blacklisted, and a new</span>
+ <span class="c"># node to provision is discovered</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">_blacklist_node</span><span class="p">(</span><span class="n">node</span><span class="p">)</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">do_discover</span><span class="p">()</span>
+ <span class="k">continue</span>
+
+ <span class="c"># check /proc directory is mounted (ssh_ok = True)</span>
+ <span class="c"># file system is not read only, hostname is correct</span>
+ <span class="c"># and omf_rc process is up</span>
+ <span class="k">else</span><span class="p">:</span>
+ <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_check_fs</span><span class="p">():</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">do_discover</span><span class="p">()</span>
+ <span class="k">continue</span>
+ <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_check_omfrc</span><span class="p">():</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">do_discover</span><span class="p">()</span>
+ <span class="k">continue</span>
+ <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_check_hostname</span><span class="p">():</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">do_discover</span><span class="p">()</span>
+ <span class="k">continue</span>
+
+ <span class="k">else</span><span class="p">:</span>
+ <span class="n">provision_ok</span> <span class="o">=</span> <span class="bp">True</span>
+ <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">'host'</span><span class="p">):</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">_set_host_attr</span><span class="p">(</span><span class="n">node</span><span class="p">)</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s">" Node provisioned "</span><span class="p">)</span>
+
+ <span class="nb">super</span><span class="p">(</span><span class="n">WilabtSfaNode</span><span class="p">,</span> <span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="n">do_provision</span><span class="p">()</span>
+</div>
+<div class="viewcode-block" id="WilabtSfaNode.do_deploy"><a class="viewcode-back" href="../../../../_layout/nepi.resources.omf.html#nepi.resources.omf.wilabt_node.WilabtSfaNode.do_deploy">[docs]</a> <span class="k">def</span> <span class="nf">do_deploy</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
+ <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">state</span> <span class="o">==</span> <span class="n">ResourceState</span><span class="o">.</span><span class="n">NEW</span><span class="p">:</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s">"Deploying w-iLab.t node"</span><span class="p">)</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">do_discover</span><span class="p">()</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">do_provision</span><span class="p">()</span>
+ <span class="nb">super</span><span class="p">(</span><span class="n">WilabtSfaNode</span><span class="p">,</span> <span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="n">do_deploy</span><span class="p">()</span>
+</div>
+<div class="viewcode-block" id="WilabtSfaNode.do_release"><a class="viewcode-back" href="../../../../_layout/nepi.resources.omf.html#nepi.resources.omf.wilabt_node.WilabtSfaNode.do_release">[docs]</a> <span class="k">def</span> <span class="nf">do_release</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
+ <span class="nb">super</span><span class="p">(</span><span class="n">WilabtSfaNode</span><span class="p">,</span> <span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="n">do_release</span><span class="p">()</span>
+ <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">state</span> <span class="o">==</span> <span class="n">ResourceState</span><span class="o">.</span><span class="n">RELEASED</span> <span class="ow">and</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_skip_provision</span><span class="p">():</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">debug</span><span class="p">(</span><span class="s">" Releasing SFA API "</span><span class="p">)</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">sfaapi</span><span class="o">.</span><span class="n">release</span><span class="p">()</span>
+</div>
+ <span class="k">def</span> <span class="nf">_blacklisted</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">host_hrn</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Check in the SFA API that the node is not in the blacklist.</span>
+<span class="sd"> """</span>
+ <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">sfaapi</span><span class="o">.</span><span class="n">blacklisted</span><span class="p">(</span><span class="n">host_hrn</span><span class="p">):</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">fail_node_not_available</span><span class="p">(</span><span class="n">host_hrn</span><span class="p">)</span>
+ <span class="k">return</span> <span class="bp">False</span>
+
+ <span class="k">def</span> <span class="nf">_reserved</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">host_hrn</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Check in the SFA API that the node is not in the reserved</span>
+<span class="sd"> list.</span>
+<span class="sd"> """</span>
+ <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">sfaapi</span><span class="o">.</span><span class="n">reserved</span><span class="p">(</span><span class="n">host_hrn</span><span class="p">):</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">fail_node_not_available</span><span class="p">(</span><span class="n">host_hrn</span><span class="p">)</span>
+ <span class="k">return</span> <span class="bp">False</span>
+
+ <span class="k">def</span> <span class="nf">_get_username</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Get the username for login in to the nodes from RSpec.</span>
+<span class="sd"> Wilabt username is not made out of any convention, it</span>
+<span class="sd"> has to be retrived from the manifest RSpec.</span>
+<span class="sd"> """</span>
+ <span class="n">slicename</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">"slicename"</span><span class="p">)</span>
+ <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">_username</span> <span class="ow">is</span> <span class="bp">None</span><span class="p">:</span>
+ <span class="n">slice_info</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sfaapi</span><span class="o">.</span><span class="n">get_slice_resources</span><span class="p">(</span><span class="n">slicename</span><span class="p">)</span>
+ <span class="n">username</span> <span class="o">=</span> <span class="n">slice_info</span><span class="p">[</span><span class="s">'resource'</span><span class="p">][</span><span class="mi">0</span><span class="p">][</span><span class="s">'services'</span><span class="p">][</span><span class="mi">0</span><span class="p">][</span><span class="s">'login'</span><span class="p">][</span><span class="mi">0</span><span class="p">][</span><span class="s">'username'</span><span class="p">]</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">set</span><span class="p">(</span><span class="s">'username'</span><span class="p">,</span> <span class="n">username</span><span class="p">)</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">debug</span><span class="p">(</span><span class="s">"Retriving username information from RSpec </span><span class="si">%s</span><span class="s">"</span> <span class="o">%</span> <span class="n">username</span><span class="p">)</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">_username</span> <span class="o">=</span> <span class="n">username</span>
+
+ <span class="k">def</span> <span class="nf">_check_ssh_loop</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Check that the ssh login is possible. In wilabt is done</span>
+<span class="sd"> through the gateway because is private testbed.</span>
+<span class="sd"> """</span>
+ <span class="n">t</span> <span class="o">=</span> <span class="mi">0</span>
+ <span class="n">timeout</span> <span class="o">=</span> <span class="mi">1200</span>
+ <span class="n">ssh_ok</span> <span class="o">=</span> <span class="bp">False</span>
+ <span class="k">while</span> <span class="n">t</span> <span class="o"><</span> <span class="n">timeout</span> <span class="ow">and</span> <span class="ow">not</span> <span class="n">ssh_ok</span><span class="p">:</span>
+ <span class="n">cmd</span> <span class="o">=</span> <span class="s">'echo </span><span class="se">\'</span><span class="s">GOOD NODE</span><span class="se">\'</span><span class="s">'</span>
+ <span class="p">((</span><span class="n">out</span><span class="p">,</span> <span class="n">err</span><span class="p">),</span> <span class="n">proc</span><span class="p">)</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">cmd</span><span class="p">)</span>
+ <span class="k">if</span> <span class="n">out</span><span class="o">.</span><span class="n">find</span><span class="p">(</span><span class="s">"GOOD NODE"</span><span class="p">)</span> <span class="o"><</span> <span class="mi">0</span><span class="p">:</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">debug</span><span class="p">(</span> <span class="s">"No SSH connection, waiting 20s"</span> <span class="p">)</span>
+ <span class="n">t</span> <span class="o">=</span> <span class="n">t</span> <span class="o">+</span> <span class="mi">20</span>
+ <span class="n">time</span><span class="o">.</span><span class="n">sleep</span><span class="p">(</span><span class="mi">20</span><span class="p">)</span>
+ <span class="k">continue</span>
+ <span class="k">else</span><span class="p">:</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">debug</span><span class="p">(</span> <span class="s">"SSH OK"</span> <span class="p">)</span>
+ <span class="n">ssh_ok</span> <span class="o">=</span> <span class="bp">True</span>
+ <span class="k">continue</span>
+ <span class="k">return</span> <span class="n">ssh_ok</span>
+
+ <span class="k">def</span> <span class="nf">_check_fs</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Check file system, /proc well mounted.</span>
+<span class="sd"> """</span>
+ <span class="n">cmd</span> <span class="o">=</span> <span class="s">'mount |grep proc'</span>
+ <span class="p">((</span><span class="n">out</span><span class="p">,</span> <span class="n">err</span><span class="p">),</span> <span class="n">proc</span><span class="p">)</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">cmd</span><span class="p">)</span>
+ <span class="k">if</span> <span class="n">out</span><span class="o">.</span><span class="n">find</span><span class="p">(</span><span class="s">"/proc type proc"</span><span class="p">)</span> <span class="o"><</span> <span class="mi">0</span><span class="p">:</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">warning</span><span class="p">(</span><span class="s">" Corrupted file system "</span><span class="p">)</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">_blacklist_node</span><span class="p">(</span><span class="n">node</span><span class="p">)</span>
+ <span class="k">return</span> <span class="bp">False</span>
+ <span class="k">return</span> <span class="bp">True</span>
+
+ <span class="k">def</span> <span class="nf">_check_omfrc</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Check that OMF 6 resource controller is running.</span>
+<span class="sd"> """</span>
+ <span class="n">cmd</span> <span class="o">=</span> <span class="s">'ps aux|grep omf'</span>
+ <span class="p">((</span><span class="n">out</span><span class="p">,</span> <span class="n">err</span><span class="p">),</span> <span class="n">proc</span><span class="p">)</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">cmd</span><span class="p">)</span>
+ <span class="k">if</span> <span class="n">out</span><span class="o">.</span><span class="n">find</span><span class="p">(</span><span class="s">"/usr/local/rvm/gems/ruby-1.9.3-p286@omf/bin/omf_rc"</span><span class="p">)</span> <span class="o"><</span> <span class="mi">0</span><span class="p">:</span>
+ <span class="k">return</span> <span class="bp">False</span>
+ <span class="k">return</span> <span class="bp">True</span>
+
+ <span class="k">def</span> <span class="nf">_check_hostname</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Check that the hostname in the image is not set to localhost.</span>
+<span class="sd"> """</span>
+ <span class="n">cmd</span> <span class="o">=</span> <span class="s">'hostname'</span>
+ <span class="p">((</span><span class="n">out</span><span class="p">,</span> <span class="n">err</span><span class="p">),</span> <span class="n">proc</span><span class="p">)</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">cmd</span><span class="p">)</span>
+ <span class="k">if</span> <span class="s">'localhost'</span> <span class="ow">in</span> <span class="n">out</span><span class="o">.</span><span class="n">lower</span><span class="p">():</span>
+ <span class="k">return</span> <span class="bp">False</span>
+ <span class="k">else</span><span class="p">:</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">set</span><span class="p">(</span><span class="s">'hostname'</span><span class="p">,</span> <span class="n">out</span><span class="o">.</span><span class="n">strip</span><span class="p">())</span>
+ <span class="k">return</span> <span class="bp">True</span>
+
+<div class="viewcode-block" id="WilabtSfaNode.execute"><a class="viewcode-back" href="../../../../_layout/nepi.resources.omf.html#nepi.resources.omf.wilabt_node.WilabtSfaNode.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">command</span><span class="p">,</span>
+ <span class="n">sudo</span> <span class="o">=</span> <span class="bp">False</span><span class="p">,</span>
+ <span class="n">env</span> <span class="o">=</span> <span class="bp">None</span><span class="p">,</span>
+ <span class="n">tty</span> <span class="o">=</span> <span class="bp">False</span><span class="p">,</span>
+ <span class="n">forward_x11</span> <span class="o">=</span> <span class="bp">False</span><span class="p">,</span>
+ <span class="n">retry</span> <span class="o">=</span> <span class="mi">3</span><span class="p">,</span>
+ <span class="n">connect_timeout</span> <span class="o">=</span> <span class="mi">30</span><span class="p">,</span>
+ <span class="n">strict_host_checking</span> <span class="o">=</span> <span class="bp">False</span><span class="p">,</span>
+ <span class="n">persistent</span> <span class="o">=</span> <span class="bp">True</span><span class="p">,</span>
+ <span class="n">blocking</span> <span class="o">=</span> <span class="bp">True</span><span class="p">,</span>
+ <span class="p">):</span>
+ <span class="sd">""" Notice that this invocation will block until the</span>
+<span class="sd"> execution finishes. If this is not the desired behavior,</span>
+<span class="sd"> use 'run' instead."""</span>
+ <span class="p">(</span><span class="n">out</span><span class="p">,</span> <span class="n">err</span><span class="p">),</span> <span class="n">proc</span> <span class="o">=</span> <span class="n">sshfuncs</span><span class="o">.</span><span class="n">rexec</span><span class="p">(</span>
+ <span class="n">command</span><span class="p">,</span>
+ <span class="n">host</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">"host"</span><span class="p">),</span>
+ <span class="n">user</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">"username"</span><span class="p">),</span>
+ <span class="n">port</span> <span class="o">=</span> <span class="mi">22</span><span class="p">,</span>
+ <span class="n">gwuser</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">"gatewayUser"</span><span class="p">),</span>
+ <span class="n">gw</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">"gateway"</span><span class="p">),</span>
+ <span class="n">agent</span> <span class="o">=</span> <span class="bp">True</span><span class="p">,</span>
+ <span class="n">sudo</span> <span class="o">=</span> <span class="n">sudo</span><span class="p">,</span>
+ <span class="n">identity</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">"identity"</span><span class="p">),</span>
+ <span class="n">server_key</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">"serverKey"</span><span class="p">),</span>
+ <span class="n">env</span> <span class="o">=</span> <span class="n">env</span><span class="p">,</span>
+ <span class="n">tty</span> <span class="o">=</span> <span class="n">tty</span><span class="p">,</span>
+ <span class="n">forward_x11</span> <span class="o">=</span> <span class="n">forward_x11</span><span class="p">,</span>
+ <span class="n">retry</span> <span class="o">=</span> <span class="n">retry</span><span class="p">,</span>
+ <span class="n">connect_timeout</span> <span class="o">=</span> <span class="n">connect_timeout</span><span class="p">,</span>
+ <span class="n">persistent</span> <span class="o">=</span> <span class="n">persistent</span><span class="p">,</span>
+ <span class="n">blocking</span> <span class="o">=</span> <span class="n">blocking</span><span class="p">,</span>
+ <span class="n">strict_host_checking</span> <span class="o">=</span> <span class="n">strict_host_checking</span>
+ <span class="p">)</span>
+
+ <span class="k">return</span> <span class="p">(</span><span class="n">out</span><span class="p">,</span> <span class="n">err</span><span class="p">),</span> <span class="n">proc</span>
+
+</div>
+ <span class="k">def</span> <span class="nf">_add_node_to_slice</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">host_hrn</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Add node to slice, using SFA API. Actually Wilabt testbed</span>
+<span class="sd"> doesn't allow adding nodes, in fact in the API there is method</span>
+<span class="sd"> to group all the nodes instanciated as WilabtSfaNodes and the</span>
+<span class="sd"> Allocate and Provision is done with the last call at </span>
+<span class="sd"> sfaapi.add_resource_to_slice_batch.</span>
+<span class="sd"> """</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s">" Adding node to slice "</span><span class="p">)</span>
+ <span class="n">slicename</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">"slicename"</span><span class="p">)</span>
+ <span class="n">disk_image</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">"disk_image"</span><span class="p">)</span>
+ <span class="k">if</span> <span class="n">disk_image</span> <span class="ow">is</span> <span class="ow">not</span> <span class="bp">None</span><span class="p">:</span>
+ <span class="n">properties</span> <span class="o">=</span> <span class="p">{</span><span class="s">'disk_image'</span><span class="p">:</span> <span class="n">disk_image</span><span class="p">}</span>
+ <span class="k">else</span><span class="p">:</span> <span class="n">properties</span> <span class="o">=</span> <span class="bp">None</span>
+ <span class="c">#properties = None</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">sfaapi</span><span class="o">.</span><span class="n">add_resource_to_slice_batch</span><span class="p">(</span><span class="n">slicename</span><span class="p">,</span> <span class="n">host_hrn</span><span class="p">,</span> <span class="n">properties</span><span class="o">=</span><span class="n">properties</span><span class="p">)</span>
+
+ <span class="k">def</span> <span class="nf">_delete_from_slice</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Delete every node from slice, using SFA API.</span>
+<span class="sd"> Wilabt doesn't allow to remove one sliver so this method </span>
+<span class="sd"> remove every slice from the slice.</span>
+<span class="sd"> """</span>
+
+ <span class="bp">self</span><span class="o">.</span><span class="n">warning</span><span class="p">(</span><span class="s">" Deleting all slivers from slice "</span><span class="p">)</span>
+ <span class="n">slicename</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">"slicename"</span><span class="p">)</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">sfaapi</span><span class="o">.</span><span class="n">remove_all_from_slice</span><span class="p">(</span><span class="n">slicename</span><span class="p">)</span>
+
+ <span class="k">def</span> <span class="nf">_get_host</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Get the attribute hostname.</span>
+<span class="sd"> """</span>
+ <span class="n">host</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">"host"</span><span class="p">)</span>
+ <span class="k">if</span> <span class="n">host</span><span class="p">:</span>
+ <span class="k">return</span> <span class="n">host</span>
+ <span class="k">else</span><span class="p">:</span>
+ <span class="k">return</span> <span class="bp">None</span>
+
+ <span class="k">def</span> <span class="nf">_set_host_attr</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">node</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Query SFAAPI for the hostname of a certain host hrn and sets the</span>
+<span class="sd"> attribute hostname, it will over write the previous value.</span>
+<span class="sd"> """</span>
+ <span class="n">hosts_hrn</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sfaapi</span><span class="o">.</span><span class="n">get_resources_hrn</span><span class="p">()</span>
+ <span class="k">for</span> <span class="n">host</span><span class="p">,</span> <span class="n">hrn</span> <span class="ow">in</span> <span class="n">hosts_hrn</span><span class="o">.</span><span class="n">iteritems</span><span class="p">():</span>
+ <span class="k">if</span> <span class="n">hrn</span> <span class="o">==</span> <span class="n">node</span><span class="p">:</span>
+ <span class="n">host</span> <span class="o">=</span> <span class="n">host</span> <span class="o">+</span> <span class="s">'.wilab2.ilabt.iminds.be'</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">set</span><span class="p">(</span><span class="s">"host"</span><span class="p">,</span> <span class="n">host</span><span class="p">)</span>
+
+ <span class="k">def</span> <span class="nf">_check_if_in_slice</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">hosts_hrn</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Check using SFA API if any host hrn from hosts_hrn is in the user's</span>
+<span class="sd"> slice.</span>
+<span class="sd"> """</span>
+ <span class="n">slicename</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s">"slicename"</span><span class="p">)</span>
+ <span class="n">slice_nodes</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sfaapi</span><span class="o">.</span><span class="n">get_slice_resources</span><span class="p">(</span><span class="n">slicename</span><span class="p">)[</span><span class="s">'resource'</span><span class="p">]</span>
+ <span class="k">if</span> <span class="n">slice_nodes</span><span class="p">:</span>
+ <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">slice_nodes</span><span class="p">[</span><span class="mi">0</span><span class="p">][</span><span class="s">'services'</span><span class="p">])</span> <span class="o">!=</span> <span class="mi">0</span><span class="p">:</span>
+ <span class="n">slice_nodes_hrn</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sfaapi</span><span class="o">.</span><span class="n">get_resources_hrn</span><span class="p">(</span><span class="n">slice_nodes</span><span class="p">)</span><span class="o">.</span><span class="n">values</span><span class="p">()</span>
+ <span class="k">else</span><span class="p">:</span> <span class="n">slice_nodes_hrn</span> <span class="o">=</span> <span class="p">[]</span>
+ <span class="n">nodes_inslice</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="nb">set</span><span class="p">(</span><span class="n">hosts_hrn</span><span class="p">)</span> <span class="o">&</span> <span class="nb">set</span><span class="p">(</span><span class="n">slice_nodes_hrn</span><span class="p">))</span>
+ <span class="k">return</span> <span class="n">nodes_inslice</span>
+
+ <span class="k">def</span> <span class="nf">_blacklist_node</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">host_hrn</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Add mal functioning node to blacklist (in SFA API).</span>
+<span class="sd"> """</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">warning</span><span class="p">(</span><span class="s">" Blacklisting malfunctioning node "</span><span class="p">)</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">sfaapi</span><span class="o">.</span><span class="n">blacklist_resource</span><span class="p">(</span><span class="n">host_hrn</span><span class="p">)</span>
+ <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_host</span><span class="p">:</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">set</span><span class="p">(</span><span class="s">'host'</span><span class="p">,</span> <span class="bp">None</span><span class="p">)</span>
+ <span class="k">else</span><span class="p">:</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">set</span><span class="p">(</span><span class="s">'host'</span><span class="p">,</span> <span class="n">host_hrn</span><span class="o">.</span><span class="n">split</span><span class="p">(</span><span class="s">'.'</span><span class="p">)</span><span class="o">.</span><span class="n">pop</span><span class="p">())</span>
+
+ <span class="k">def</span> <span class="nf">_put_node_in_provision</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">host_hrn</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Add node to the list of nodes being provisioned, in order for other RMs</span>
+<span class="sd"> to not try to provision the same one again.</span>
+<span class="sd"> """</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">sfaapi</span><span class="o">.</span><span class="n">reserve_resource</span><span class="p">(</span><span class="n">host_hrn</span><span class="p">)</span>
+
+ <span class="k">def</span> <span class="nf">_get_ip</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">host</span><span class="p">):</span>
+ <span class="sd">"""</span>
+<span class="sd"> Query cache for the IP of a node with certain hostname</span>
+<span class="sd"> """</span>
+ <span class="k">try</span><span class="p">:</span>
+ <span class="n">ip</span> <span class="o">=</span> <span class="n">sshfuncs</span><span class="o">.</span><span class="n">gethostbyname</span><span class="p">(</span><span class="n">host</span><span class="p">)</span>
+ <span class="k">except</span><span class="p">:</span>
+ <span class="c"># Fail while trying to find the IP</span>
+ <span class="k">return</span> <span class="bp">None</span>
+ <span class="k">return</span> <span class="n">ip</span>
+
+<div class="viewcode-block" id="WilabtSfaNode.fail_discovery"><a class="viewcode-back" href="../../../../_layout/nepi.resources.omf.html#nepi.resources.omf.wilabt_node.WilabtSfaNode.fail_discovery">[docs]</a> <span class="k">def</span> <span class="nf">fail_discovery</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
+ <span class="n">msg</span> <span class="o">=</span> <span class="s">"Discovery failed. No candidates found for node"</span>
+ <span class="bp">self</span><span class="o">.</span><span class="n">error</span><span class="p">(</span><span class="n">msg</span><span class="p">)</span>
+ <span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">,</span> <span class="n">msg</span>
+</div>
+<div class="viewcode-block" id="WilabtSfaNode.fail_node_not_alive"><a class="viewcode-back" href="../../../../_layout/nepi.resources.omf.html#nepi.resources.omf.wilabt_node.WilabtSfaNode.fail_node_not_alive">[docs]</a> <span class="k">def</span> <span class="nf">fail_node_not_alive</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">host</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span>
+ <span class="n">msg</span> <span class="o">=</span> <span class="s">"Node </span><span class="si">%s</span><span class="s"> not alive"</span> <span class="o">%</span> <span class="n">host</span>
+ <span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">,</span> <span class="n">msg</span>
+ </div>
+<div class="viewcode-block" id="WilabtSfaNode.fail_node_not_available"><a class="viewcode-back" href="../../../../_layout/nepi.resources.omf.html#nepi.resources.omf.wilabt_node.WilabtSfaNode.fail_node_not_available">[docs]</a> <span class="k">def</span> <span class="nf">fail_node_not_available</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">host</span><span class="p">):</span>
+ <span class="n">msg</span> <span class="o">=</span> <span class="s">"Some nodes not available for provisioning"</span>
+ <span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">,</span> <span class="n">msg</span>
+</div>
+<div class="viewcode-block" id="WilabtSfaNode.fail_not_enough_nodes"><a class="viewcode-back" href="../../../../_layout/nepi.resources.omf.html#nepi.resources.omf.wilabt_node.WilabtSfaNode.fail_not_enough_nodes">[docs]</a> <span class="k">def</span> <span class="nf">fail_not_enough_nodes</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
+ <span class="n">msg</span> <span class="o">=</span> <span class="s">"Not enough nodes available for provisioning"</span>
+ <span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">,</span> <span class="n">msg</span>
+</div>
+<div class="viewcode-block" id="WilabtSfaNode.fail_sfaapi"><a class="viewcode-back" href="../../../../_layout/nepi.resources.omf.html#nepi.resources.omf.wilabt_node.WilabtSfaNode.fail_sfaapi">[docs]</a> <span class="k">def</span> <span class="nf">fail_sfaapi</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
+ <span class="n">msg</span> <span class="o">=</span> <span class="s">"Failing while trying to instanciate the SFA API."</span>
+ <span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">,</span> <span class="n">msg</span>
+</div>
+<div class="viewcode-block" id="WilabtSfaNode.valid_connection"><a class="viewcode-back" href="../../../../_layout/nepi.resources.omf.html#nepi.resources.omf.wilabt_node.WilabtSfaNode.valid_connection">[docs]</a> <span class="k">def</span> <span class="nf">valid_connection</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">guid</span><span class="p">):</span>
+ <span class="c"># TODO: Validate!</span>
+ <span class="k">return</span> <span class="bp">True</span>
+</pre></div></div></div>
+
+ </div>
+ </div>
+ </div>
+ <div class="clearer"></div>
+ </div>
+ <div class="related">
+ <h3>Navigation</h3>
+ <ul>
+ <li class="right" style="margin-right: 10px">
+ <a href="../../../../genindex.html" title="General Index"
+ >index</a></li>
+ <li class="right" >
+ <a href="../../../../py-modindex.html" title="Python Module Index"
+ >modules</a> |</li>
+ <li><a href="../../../../index.html">NEPI 3.0 documentation</a> »</li>
+ <li><a href="../../../index.html" >Module code</a> »</li>
+ </ul>
+ </div>
+ <div class="footer">
+ © Copyright 2014, Alina Quereilhac, Lucia Guevgeozian Odizzio, Julien Tribino.
+ Created using <a href="http://sphinx-doc.org/">Sphinx</a> 1.2.3.
+ </div>
+ </body>
+</html>
\ No newline at end of file