1 <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"
2 "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
5 <html xmlns="http://www.w3.org/1999/xhtml">
7 <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
9 <title>nepi.resources.omf.wilabt_node — NEPI 3.2 documentation</title>
11 <link rel="stylesheet" href="../../../../_static/sphinxdoc.css" type="text/css" />
12 <link rel="stylesheet" href="../../../../_static/pygments.css" type="text/css" />
14 <script type="text/javascript">
15 var DOCUMENTATION_OPTIONS = {
16 URL_ROOT: '../../../../',
18 COLLAPSE_INDEX: false,
23 <script type="text/javascript" src="../../../../_static/jquery.js"></script>
24 <script type="text/javascript" src="../../../../_static/underscore.js"></script>
25 <script type="text/javascript" src="../../../../_static/doctools.js"></script>
26 <link rel="top" title="NEPI 3.2 documentation" href="../../../../index.html" />
27 <link rel="up" title="Module code" href="../../../index.html" />
33 <li class="right" style="margin-right: 10px">
34 <a href="../../../../genindex.html" title="General Index"
35 accesskey="I">index</a></li>
37 <a href="../../../../py-modindex.html" title="Python Module Index"
39 <li><a href="../../../../index.html">NEPI 3.2 documentation</a> »</li>
40 <li><a href="../../../index.html" accesskey="U">Module code</a> »</li>
43 <div class="sphinxsidebar">
44 <div class="sphinxsidebarwrapper">
45 <div id="searchbox" style="display: none">
47 <form class="search" action="../../../../search.html" method="get">
48 <input type="text" name="q" />
49 <input type="submit" value="Go" />
50 <input type="hidden" name="check_keywords" value="yes" />
51 <input type="hidden" name="area" value="default" />
53 <p class="searchtip" style="font-size: 90%">
54 Enter search terms or a module, class or function name.
57 <script type="text/javascript">$('#searchbox').show(0);</script>
61 <div class="document">
62 <div class="documentwrapper">
63 <div class="bodywrapper">
66 <h1>Source code for nepi.resources.omf.wilabt_node</h1><div class="highlight"><pre>
67 <span class="c">#</span>
68 <span class="c"># NEPI, a framework to manage network experiments</span>
69 <span class="c"># Copyright (C) 2013 INRIA</span>
70 <span class="c">#</span>
71 <span class="c"># This program is free software: you can redistribute it and/or modify</span>
72 <span class="c"># it under the terms of the GNU General Public License version 2 as</span>
73 <span class="c"># published by the Free Software Foundation;</span>
74 <span class="c">#</span>
75 <span class="c"># This program is distributed in the hope that it will be useful,</span>
76 <span class="c"># but WITHOUT ANY WARRANTY; without even the implied warranty of</span>
77 <span class="c"># MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the</span>
78 <span class="c"># GNU General Public License for more details.</span>
79 <span class="c">#</span>
80 <span class="c"># You should have received a copy of the GNU General Public License</span>
81 <span class="c"># along with this program. If not, see <http://www.gnu.org/licenses/>.</span>
82 <span class="c">#</span>
83 <span class="c"># Author: Lucia Guevgeozian <lucia.guevgeozian_odizzio@inria.fr></span>
85 <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>
86 <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> \
87 <span class="n">ResourceState</span>
88 <span class="kn">from</span> <span class="nn">nepi.resources.omf.node</span> <span class="kn">import</span> <span class="n">OMFNode</span>
89 <span class="kn">from</span> <span class="nn">nepi.util.sfaapi</span> <span class="kn">import</span> <span class="n">SFAAPIFactory</span>
90 <span class="kn">from</span> <span class="nn">nepi.util.execfuncs</span> <span class="kn">import</span> <span class="n">lexec</span>
91 <span class="kn">from</span> <span class="nn">nepi.util</span> <span class="kn">import</span> <span class="n">sshfuncs</span>
93 <span class="kn">from</span> <span class="nn">random</span> <span class="kn">import</span> <span class="n">randint</span>
94 <span class="kn">import</span> <span class="nn">time</span>
95 <span class="kn">import</span> <span class="nn">re</span>
96 <span class="kn">import</span> <span class="nn">weakref</span>
97 <span class="kn">import</span> <span class="nn">socket</span>
98 <span class="kn">import</span> <span class="nn">threading</span>
99 <span class="kn">import</span> <span class="nn">datetime</span>
101 <span class="nd">@clsinit_copy</span>
102 <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>
103 <span class="n">_rtype</span> <span class="o">=</span> <span class="s">"wilabt::sfa::Node"</span>
104 <span class="n">_help</span> <span class="o">=</span> <span class="s">"Controls a Wilabt host accessible using a SSH key "</span> \
105 <span class="s">"and provisioned using SFA"</span>
106 <span class="n">_platform</span> <span class="o">=</span> <span class="s">"omf"</span>
108 <span class="nd">@classmethod</span>
109 <span class="k">def</span> <span class="nf">_register_attributes</span><span class="p">(</span><span class="n">cls</span><span class="p">):</span>
111 <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>
112 <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>
114 <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>
115 <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>
117 <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>
118 <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>
120 <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>
121 <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>
123 <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>
124 <span class="s"> used to generate the user credential"</span><span class="p">,</span>
125 <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>
127 <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>
128 <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>
130 <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>
131 <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>
133 <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>
134 <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>
136 <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>
137 <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>
139 <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>
140 <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>
142 <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>
143 <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>
144 <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>
145 <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>
146 <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>
147 <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>
148 <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>
149 <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>
150 <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>
151 <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>
153 <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>
154 <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>
156 <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>
157 <span class="bp">self</span><span class="o">.</span><span class="n">_sfaapi</span> <span class="o">=</span> <span class="bp">None</span>
158 <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>
159 <span class="bp">self</span><span class="o">.</span><span class="n">_slicenode</span> <span class="o">=</span> <span class="bp">False</span>
160 <span class="bp">self</span><span class="o">.</span><span class="n">_host</span> <span class="o">=</span> <span class="bp">False</span>
161 <span class="bp">self</span><span class="o">.</span><span class="n">_username</span> <span class="o">=</span> <span class="bp">None</span>
163 <span class="k">def</span> <span class="nf">_skip_provision</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
164 <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>
165 <span class="k">if</span> <span class="ow">not</span> <span class="n">sfa_user</span><span class="p">:</span>
166 <span class="k">return</span> <span class="bp">True</span>
167 <span class="k">else</span><span class="p">:</span> <span class="k">return</span> <span class="bp">False</span>
169 <span class="nd">@property</span>
170 <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>
171 <span class="sd">"""</span>
172 <span class="sd"> Property to instanciate the SFA API based in sfi client.</span>
173 <span class="sd"> For each SFA method called this instance is used.</span>
174 <span class="sd"> """</span>
175 <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>
176 <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>
177 <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>
178 <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>
179 <span class="n">sfa_registry</span> <span class="o">=</span> <span class="s">"http://sfa3.planet-lab.eu:12345/"</span>
180 <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>
181 <span class="n">batch</span> <span class="o">=</span> <span class="bp">True</span>
183 <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>
184 <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>
186 <span class="k">if</span> <span class="ow">not</span> <span class="n">_sfaapi</span><span class="p">:</span>
187 <span class="bp">self</span><span class="o">.</span><span class="n">fail_sfaapi</span><span class="p">()</span>
189 <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>
191 <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sfaapi</span><span class="p">()</span>
193 <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>
194 <span class="sd">"""</span>
195 <span class="sd"> Based on the attributes defined by the user, discover the suitable </span>
196 <span class="sd"> node for provision.</span>
197 <span class="sd"> """</span>
198 <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>
200 <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>
201 <span class="k">if</span> <span class="n">host</span><span class="p">:</span>
202 <span class="c"># the user specified one particular node to be provisioned</span>
203 <span class="bp">self</span><span class="o">.</span><span class="n">_host</span> <span class="o">=</span> <span class="bp">True</span>
204 <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>
206 <span class="c"># check that the node is not blacklisted or being provisioned</span>
207 <span class="c"># by other RM</span>
208 <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>
209 <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>
210 <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>
211 <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>
212 <span class="bp">self</span><span class="o">.</span><span class="n">_slicenode</span> <span class="o">=</span> <span class="bp">True</span>
213 <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>
214 <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>
215 <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>
216 <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>
218 <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>
219 <span class="sd">"""</span>
220 <span class="sd"> Add node to user's slice and verifing that the node is functioning</span>
221 <span class="sd"> correctly. Check ssh, omf rc running, hostname, file system.</span>
222 <span class="sd"> """</span>
223 <span class="n">provision_ok</span> <span class="o">=</span> <span class="bp">False</span>
224 <span class="n">ssh_ok</span> <span class="o">=</span> <span class="bp">False</span>
225 <span class="n">proc_ok</span> <span class="o">=</span> <span class="bp">False</span>
226 <span class="n">timeout</span> <span class="o">=</span> <span class="mi">300</span>
228 <span class="k">while</span> <span class="ow">not</span> <span class="n">provision_ok</span><span class="p">:</span>
229 <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>
230 <span class="c">#if self._slicenode:</span>
231 <span class="c"># self._delete_from_slice()</span>
232 <span class="c"># self.debug("Waiting 480 sec for re-adding to slice")</span>
233 <span class="c"># time.sleep(480) # Timout for the testbed to allow a new reservation</span>
234 <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>
235 <span class="n">t</span> <span class="o">=</span> <span class="mi">0</span>
236 <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> \
237 <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>
238 <span class="n">t</span> <span class="o">=</span> <span class="n">t</span> <span class="o">+</span> <span class="mi">5</span>
239 <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>
240 <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>
241 <span class="k">continue</span>
243 <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>
244 <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>
245 <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>
247 <span class="bp">self</span><span class="o">.</span><span class="n">_get_username</span><span class="p">()</span>
248 <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>
250 <span class="k">if</span> <span class="ow">not</span> <span class="n">ssh_ok</span><span class="p">:</span>
251 <span class="c"># the timeout was reach without establishing ssh connection</span>
252 <span class="c"># the node is blacklisted, and a new</span>
253 <span class="c"># node to provision is discovered</span>
254 <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>
255 <span class="bp">self</span><span class="o">.</span><span class="n">do_discover</span><span class="p">()</span>
256 <span class="k">continue</span>
258 <span class="c"># check /proc directory is mounted (ssh_ok = True)</span>
259 <span class="c"># file system is not read only, hostname is correct</span>
260 <span class="c"># and omf_rc process is up</span>
261 <span class="k">else</span><span class="p">:</span>
262 <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>
263 <span class="bp">self</span><span class="o">.</span><span class="n">do_discover</span><span class="p">()</span>
264 <span class="k">continue</span>
265 <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>
266 <span class="bp">self</span><span class="o">.</span><span class="n">do_discover</span><span class="p">()</span>
267 <span class="k">continue</span>
268 <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>
269 <span class="bp">self</span><span class="o">.</span><span class="n">do_discover</span><span class="p">()</span>
270 <span class="k">continue</span>
272 <span class="k">else</span><span class="p">:</span>
273 <span class="n">provision_ok</span> <span class="o">=</span> <span class="bp">True</span>
274 <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>
275 <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>
276 <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>
278 <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>
280 <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>
281 <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>
282 <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>
283 <span class="bp">self</span><span class="o">.</span><span class="n">do_discover</span><span class="p">()</span>
284 <span class="bp">self</span><span class="o">.</span><span class="n">do_provision</span><span class="p">()</span>
285 <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>
287 <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>
288 <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>
289 <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>
290 <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>
291 <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>
293 <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>
294 <span class="sd">"""</span>
295 <span class="sd"> Check in the SFA API that the node is not in the blacklist.</span>
296 <span class="sd"> """</span>
297 <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>
298 <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>
299 <span class="k">return</span> <span class="bp">False</span>
301 <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>
302 <span class="sd">"""</span>
303 <span class="sd"> Check in the SFA API that the node is not in the reserved</span>
304 <span class="sd"> list.</span>
305 <span class="sd"> """</span>
306 <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>
307 <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>
308 <span class="k">return</span> <span class="bp">False</span>
310 <span class="k">def</span> <span class="nf">_get_username</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
311 <span class="sd">"""</span>
312 <span class="sd"> Get the username for login in to the nodes from RSpec.</span>
313 <span class="sd"> Wilabt username is not made out of any convention, it</span>
314 <span class="sd"> has to be retrived from the manifest RSpec.</span>
315 <span class="sd"> """</span>
316 <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>
317 <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>
318 <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>
319 <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>
320 <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>
321 <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>
322 <span class="bp">self</span><span class="o">.</span><span class="n">_username</span> <span class="o">=</span> <span class="n">username</span>
324 <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>
325 <span class="sd">"""</span>
326 <span class="sd"> Check that the ssh login is possible. In wilabt is done</span>
327 <span class="sd"> through the gateway because is private testbed.</span>
328 <span class="sd"> """</span>
329 <span class="n">t</span> <span class="o">=</span> <span class="mi">0</span>
330 <span class="n">timeout</span> <span class="o">=</span> <span class="mi">1200</span>
331 <span class="n">ssh_ok</span> <span class="o">=</span> <span class="bp">False</span>
332 <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>
333 <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>
334 <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>
335 <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>
336 <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>
337 <span class="n">t</span> <span class="o">=</span> <span class="n">t</span> <span class="o">+</span> <span class="mi">20</span>
338 <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>
339 <span class="k">continue</span>
340 <span class="k">else</span><span class="p">:</span>
341 <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>
342 <span class="n">ssh_ok</span> <span class="o">=</span> <span class="bp">True</span>
343 <span class="k">continue</span>
344 <span class="k">return</span> <span class="n">ssh_ok</span>
346 <span class="k">def</span> <span class="nf">_check_fs</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
347 <span class="sd">"""</span>
348 <span class="sd"> Check file system, /proc well mounted.</span>
349 <span class="sd"> """</span>
350 <span class="n">cmd</span> <span class="o">=</span> <span class="s">'mount |grep proc'</span>
351 <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>
352 <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>
353 <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>
354 <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>
355 <span class="k">return</span> <span class="bp">False</span>
356 <span class="k">return</span> <span class="bp">True</span>
358 <span class="k">def</span> <span class="nf">_check_omfrc</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
359 <span class="sd">"""</span>
360 <span class="sd"> Check that OMF 6 resource controller is running.</span>
361 <span class="sd"> """</span>
362 <span class="n">cmd</span> <span class="o">=</span> <span class="s">'ps aux|grep omf'</span>
363 <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>
364 <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>
365 <span class="k">return</span> <span class="bp">False</span>
366 <span class="k">return</span> <span class="bp">True</span>
368 <span class="k">def</span> <span class="nf">_check_hostname</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
369 <span class="sd">"""</span>
370 <span class="sd"> Check that the hostname in the image is not set to localhost.</span>
371 <span class="sd"> """</span>
372 <span class="n">cmd</span> <span class="o">=</span> <span class="s">'hostname'</span>
373 <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>
374 <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>
375 <span class="k">return</span> <span class="bp">False</span>
376 <span class="k">else</span><span class="p">:</span>
377 <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>
378 <span class="k">return</span> <span class="bp">True</span>
380 <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>
381 <span class="n">sudo</span> <span class="o">=</span> <span class="bp">False</span><span class="p">,</span>
382 <span class="n">env</span> <span class="o">=</span> <span class="bp">None</span><span class="p">,</span>
383 <span class="n">tty</span> <span class="o">=</span> <span class="bp">False</span><span class="p">,</span>
384 <span class="n">forward_x11</span> <span class="o">=</span> <span class="bp">False</span><span class="p">,</span>
385 <span class="n">retry</span> <span class="o">=</span> <span class="mi">3</span><span class="p">,</span>
386 <span class="n">connect_timeout</span> <span class="o">=</span> <span class="mi">30</span><span class="p">,</span>
387 <span class="n">strict_host_checking</span> <span class="o">=</span> <span class="bp">False</span><span class="p">,</span>
388 <span class="n">persistent</span> <span class="o">=</span> <span class="bp">True</span><span class="p">,</span>
389 <span class="n">blocking</span> <span class="o">=</span> <span class="bp">True</span><span class="p">,</span>
390 <span class="p">):</span>
391 <span class="sd">""" Notice that this invocation will block until the</span>
392 <span class="sd"> execution finishes. If this is not the desired behavior,</span>
393 <span class="sd"> use 'run' instead."""</span>
394 <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>
395 <span class="n">command</span><span class="p">,</span>
396 <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>
397 <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>
398 <span class="n">port</span> <span class="o">=</span> <span class="mi">22</span><span class="p">,</span>
399 <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>
400 <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>
401 <span class="n">agent</span> <span class="o">=</span> <span class="bp">True</span><span class="p">,</span>
402 <span class="n">sudo</span> <span class="o">=</span> <span class="n">sudo</span><span class="p">,</span>
403 <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>
404 <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>
405 <span class="n">env</span> <span class="o">=</span> <span class="n">env</span><span class="p">,</span>
406 <span class="n">tty</span> <span class="o">=</span> <span class="n">tty</span><span class="p">,</span>
407 <span class="n">forward_x11</span> <span class="o">=</span> <span class="n">forward_x11</span><span class="p">,</span>
408 <span class="n">retry</span> <span class="o">=</span> <span class="n">retry</span><span class="p">,</span>
409 <span class="n">connect_timeout</span> <span class="o">=</span> <span class="n">connect_timeout</span><span class="p">,</span>
410 <span class="n">persistent</span> <span class="o">=</span> <span class="n">persistent</span><span class="p">,</span>
411 <span class="n">blocking</span> <span class="o">=</span> <span class="n">blocking</span><span class="p">,</span>
412 <span class="n">strict_host_checking</span> <span class="o">=</span> <span class="n">strict_host_checking</span>
413 <span class="p">)</span>
415 <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>
418 <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>
419 <span class="sd">"""</span>
420 <span class="sd"> Add node to slice, using SFA API. Actually Wilabt testbed</span>
421 <span class="sd"> doesn't allow adding nodes, in fact in the API there is method</span>
422 <span class="sd"> to group all the nodes instanciated as WilabtSfaNodes and the</span>
423 <span class="sd"> Allocate and Provision is done with the last call at </span>
424 <span class="sd"> sfaapi.add_resource_to_slice_batch.</span>
425 <span class="sd"> """</span>
426 <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>
427 <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>
428 <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>
429 <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>
430 <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>
431 <span class="k">else</span><span class="p">:</span> <span class="n">properties</span> <span class="o">=</span> <span class="bp">None</span>
432 <span class="c">#properties = None</span>
433 <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>
435 <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>
436 <span class="sd">"""</span>
437 <span class="sd"> Delete every node from slice, using SFA API.</span>
438 <span class="sd"> Wilabt doesn't allow to remove one sliver so this method </span>
439 <span class="sd"> remove every slice from the slice.</span>
440 <span class="sd"> """</span>
442 <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>
443 <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>
444 <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>
446 <span class="k">def</span> <span class="nf">_get_host</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
447 <span class="sd">"""</span>
448 <span class="sd"> Get the attribute hostname.</span>
449 <span class="sd"> """</span>
450 <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>
451 <span class="k">if</span> <span class="n">host</span><span class="p">:</span>
452 <span class="k">return</span> <span class="n">host</span>
453 <span class="k">else</span><span class="p">:</span>
454 <span class="k">return</span> <span class="bp">None</span>
456 <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>
457 <span class="sd">"""</span>
458 <span class="sd"> Query SFAAPI for the hostname of a certain host hrn and sets the</span>
459 <span class="sd"> attribute hostname, it will over write the previous value.</span>
460 <span class="sd"> """</span>
461 <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>
462 <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>
463 <span class="k">if</span> <span class="n">hrn</span> <span class="o">==</span> <span class="n">node</span><span class="p">:</span>
464 <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>
465 <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>
467 <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>
468 <span class="sd">"""</span>
469 <span class="sd"> Check using SFA API if any host hrn from hosts_hrn is in the user's</span>
470 <span class="sd"> slice.</span>
471 <span class="sd"> """</span>
472 <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>
473 <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>
474 <span class="k">if</span> <span class="n">slice_nodes</span><span class="p">:</span>
475 <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>
476 <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>
477 <span class="k">else</span><span class="p">:</span> <span class="n">slice_nodes_hrn</span> <span class="o">=</span> <span class="p">[]</span>
478 <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>
479 <span class="k">return</span> <span class="n">nodes_inslice</span>
481 <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>
482 <span class="sd">"""</span>
483 <span class="sd"> Add mal functioning node to blacklist (in SFA API).</span>
484 <span class="sd"> """</span>
485 <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>
486 <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>
487 <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>
488 <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>
489 <span class="k">else</span><span class="p">:</span>
490 <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>
492 <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>
493 <span class="sd">"""</span>
494 <span class="sd"> Add node to the list of nodes being provisioned, in order for other RMs</span>
495 <span class="sd"> to not try to provision the same one again.</span>
496 <span class="sd"> """</span>
497 <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>
499 <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>
500 <span class="sd">"""</span>
501 <span class="sd"> Query cache for the IP of a node with certain hostname</span>
502 <span class="sd"> """</span>
503 <span class="k">try</span><span class="p">:</span>
504 <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>
505 <span class="k">except</span><span class="p">:</span>
506 <span class="c"># Fail while trying to find the IP</span>
507 <span class="k">return</span> <span class="bp">None</span>
508 <span class="k">return</span> <span class="n">ip</span>
510 <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>
511 <span class="n">msg</span> <span class="o">=</span> <span class="s">"Discovery failed. No candidates found for node"</span>
512 <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>
513 <span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">,</span> <span class="n">msg</span>
515 <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>
516 <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>
517 <span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">,</span> <span class="n">msg</span>
519 <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>
520 <span class="n">msg</span> <span class="o">=</span> <span class="s">"Some nodes not available for provisioning"</span>
521 <span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">,</span> <span class="n">msg</span>
523 <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>
524 <span class="n">msg</span> <span class="o">=</span> <span class="s">"Not enough nodes available for provisioning"</span>
525 <span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">,</span> <span class="n">msg</span>
527 <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>
528 <span class="n">msg</span> <span class="o">=</span> <span class="s">"Failing while trying to instanciate the SFA API."</span>
529 <span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">,</span> <span class="n">msg</span>
531 <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>
532 <span class="c"># TODO: Validate!</span>
533 <span class="k">return</span> <span class="bp">True</span>
534 </pre></div></div></div>
539 <div class="clearer"></div>
541 <div class="related">
544 <li class="right" style="margin-right: 10px">
545 <a href="../../../../genindex.html" title="General Index"
548 <a href="../../../../py-modindex.html" title="Python Module Index"
550 <li><a href="../../../../index.html">NEPI 3.2 documentation</a> »</li>
551 <li><a href="../../../index.html" >Module code</a> »</li>
555 © Copyright 2014, Alina Quereilhac, Lucia Guevgeozian Odizzio, Julien Tribino.
556 Created using <a href="http://sphinx-doc.org/">Sphinx</a> 1.2.3.