Statistics
| Branch: | Tag: | Revision:

root / lib / cmdlib.py @ 2ee88aeb

History | View | Annotate | Download (247.1 kB)

1 2f31098c Iustin Pop
#
2 a8083063 Iustin Pop
#
3 a8083063 Iustin Pop
4 e7c6e02b Michael Hanselmann
# Copyright (C) 2006, 2007, 2008 Google Inc.
5 a8083063 Iustin Pop
#
6 a8083063 Iustin Pop
# This program is free software; you can redistribute it and/or modify
7 a8083063 Iustin Pop
# it under the terms of the GNU General Public License as published by
8 a8083063 Iustin Pop
# the Free Software Foundation; either version 2 of the License, or
9 a8083063 Iustin Pop
# (at your option) any later version.
10 a8083063 Iustin Pop
#
11 a8083063 Iustin Pop
# This program is distributed in the hope that it will be useful, but
12 a8083063 Iustin Pop
# WITHOUT ANY WARRANTY; without even the implied warranty of
13 a8083063 Iustin Pop
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 a8083063 Iustin Pop
# General Public License for more details.
15 a8083063 Iustin Pop
#
16 a8083063 Iustin Pop
# You should have received a copy of the GNU General Public License
17 a8083063 Iustin Pop
# along with this program; if not, write to the Free Software
18 a8083063 Iustin Pop
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
19 a8083063 Iustin Pop
# 02110-1301, USA.
20 a8083063 Iustin Pop
21 a8083063 Iustin Pop
22 880478f8 Iustin Pop
"""Module implementing the master-side code."""
23 a8083063 Iustin Pop
24 a8083063 Iustin Pop
# pylint: disable-msg=W0613,W0201
25 a8083063 Iustin Pop
26 a8083063 Iustin Pop
import os
27 a8083063 Iustin Pop
import os.path
28 a8083063 Iustin Pop
import time
29 a8083063 Iustin Pop
import tempfile
30 a8083063 Iustin Pop
import re
31 a8083063 Iustin Pop
import platform
32 ffa1c0dc Iustin Pop
import logging
33 74409b12 Iustin Pop
import copy
34 4b7735f9 Iustin Pop
import random
35 a8083063 Iustin Pop
36 a8083063 Iustin Pop
from ganeti import ssh
37 a8083063 Iustin Pop
from ganeti import utils
38 a8083063 Iustin Pop
from ganeti import errors
39 a8083063 Iustin Pop
from ganeti import hypervisor
40 6048c986 Guido Trotter
from ganeti import locking
41 a8083063 Iustin Pop
from ganeti import constants
42 a8083063 Iustin Pop
from ganeti import objects
43 a8083063 Iustin Pop
from ganeti import opcodes
44 8d14b30d Iustin Pop
from ganeti import serializer
45 112f18a5 Iustin Pop
from ganeti import ssconf
46 d61df03e Iustin Pop
47 d61df03e Iustin Pop
48 a8083063 Iustin Pop
class LogicalUnit(object):
49 396e1b78 Michael Hanselmann
  """Logical Unit base class.
50 a8083063 Iustin Pop

51 a8083063 Iustin Pop
  Subclasses must follow these rules:
52 d465bdc8 Guido Trotter
    - implement ExpandNames
53 d465bdc8 Guido Trotter
    - implement CheckPrereq
54 a8083063 Iustin Pop
    - implement Exec
55 a8083063 Iustin Pop
    - implement BuildHooksEnv
56 a8083063 Iustin Pop
    - redefine HPATH and HTYPE
57 05f86716 Guido Trotter
    - optionally redefine their run requirements:
58 7e55040e Guido Trotter
        REQ_BGL: the LU needs to hold the Big Ganeti Lock exclusively
59 05f86716 Guido Trotter

60 05f86716 Guido Trotter
  Note that all commands require root permissions.
61 a8083063 Iustin Pop

62 a8083063 Iustin Pop
  """
63 a8083063 Iustin Pop
  HPATH = None
64 a8083063 Iustin Pop
  HTYPE = None
65 a8083063 Iustin Pop
  _OP_REQP = []
66 7e55040e Guido Trotter
  REQ_BGL = True
67 a8083063 Iustin Pop
68 72737a7f Iustin Pop
  def __init__(self, processor, op, context, rpc):
69 a8083063 Iustin Pop
    """Constructor for LogicalUnit.
70 a8083063 Iustin Pop

71 a8083063 Iustin Pop
    This needs to be overriden in derived classes in order to check op
72 a8083063 Iustin Pop
    validity.
73 a8083063 Iustin Pop

74 a8083063 Iustin Pop
    """
75 5bfac263 Iustin Pop
    self.proc = processor
76 a8083063 Iustin Pop
    self.op = op
77 77b657a3 Guido Trotter
    self.cfg = context.cfg
78 77b657a3 Guido Trotter
    self.context = context
79 72737a7f Iustin Pop
    self.rpc = rpc
80 ca2a79e1 Guido Trotter
    # Dicts used to declare locking needs to mcpu
81 d465bdc8 Guido Trotter
    self.needed_locks = None
82 6683bba2 Guido Trotter
    self.acquired_locks = {}
83 3977a4c1 Guido Trotter
    self.share_locks = dict(((i, 0) for i in locking.LEVELS))
84 ca2a79e1 Guido Trotter
    self.add_locks = {}
85 ca2a79e1 Guido Trotter
    self.remove_locks = {}
86 c4a2fee1 Guido Trotter
    # Used to force good behavior when calling helper functions
87 c4a2fee1 Guido Trotter
    self.recalculate_locks = {}
88 c92b310a Michael Hanselmann
    self.__ssh = None
89 86d9d3bb Iustin Pop
    # logging
90 86d9d3bb Iustin Pop
    self.LogWarning = processor.LogWarning
91 86d9d3bb Iustin Pop
    self.LogInfo = processor.LogInfo
92 c92b310a Michael Hanselmann
93 a8083063 Iustin Pop
    for attr_name in self._OP_REQP:
94 a8083063 Iustin Pop
      attr_val = getattr(op, attr_name, None)
95 a8083063 Iustin Pop
      if attr_val is None:
96 3ecf6786 Iustin Pop
        raise errors.OpPrereqError("Required parameter '%s' missing" %
97 3ecf6786 Iustin Pop
                                   attr_name)
98 4be4691d Iustin Pop
    self.CheckArguments()
99 a8083063 Iustin Pop
100 c92b310a Michael Hanselmann
  def __GetSSH(self):
101 c92b310a Michael Hanselmann
    """Returns the SshRunner object
102 c92b310a Michael Hanselmann

103 c92b310a Michael Hanselmann
    """
104 c92b310a Michael Hanselmann
    if not self.__ssh:
105 6b0469d2 Iustin Pop
      self.__ssh = ssh.SshRunner(self.cfg.GetClusterName())
106 c92b310a Michael Hanselmann
    return self.__ssh
107 c92b310a Michael Hanselmann
108 c92b310a Michael Hanselmann
  ssh = property(fget=__GetSSH)
109 c92b310a Michael Hanselmann
110 4be4691d Iustin Pop
  def CheckArguments(self):
111 4be4691d Iustin Pop
    """Check syntactic validity for the opcode arguments.
112 4be4691d Iustin Pop

113 4be4691d Iustin Pop
    This method is for doing a simple syntactic check and ensure
114 4be4691d Iustin Pop
    validity of opcode parameters, without any cluster-related
115 4be4691d Iustin Pop
    checks. While the same can be accomplished in ExpandNames and/or
116 4be4691d Iustin Pop
    CheckPrereq, doing these separate is better because:
117 4be4691d Iustin Pop

118 4be4691d Iustin Pop
      - ExpandNames is left as as purely a lock-related function
119 4be4691d Iustin Pop
      - CheckPrereq is run after we have aquired locks (and possible
120 4be4691d Iustin Pop
        waited for them)
121 4be4691d Iustin Pop

122 4be4691d Iustin Pop
    The function is allowed to change the self.op attribute so that
123 4be4691d Iustin Pop
    later methods can no longer worry about missing parameters.
124 4be4691d Iustin Pop

125 4be4691d Iustin Pop
    """
126 4be4691d Iustin Pop
    pass
127 4be4691d Iustin Pop
128 d465bdc8 Guido Trotter
  def ExpandNames(self):
129 d465bdc8 Guido Trotter
    """Expand names for this LU.
130 d465bdc8 Guido Trotter

131 d465bdc8 Guido Trotter
    This method is called before starting to execute the opcode, and it should
132 d465bdc8 Guido Trotter
    update all the parameters of the opcode to their canonical form (e.g. a
133 d465bdc8 Guido Trotter
    short node name must be fully expanded after this method has successfully
134 d465bdc8 Guido Trotter
    completed). This way locking, hooks, logging, ecc. can work correctly.
135 d465bdc8 Guido Trotter

136 d465bdc8 Guido Trotter
    LUs which implement this method must also populate the self.needed_locks
137 d465bdc8 Guido Trotter
    member, as a dict with lock levels as keys, and a list of needed lock names
138 d465bdc8 Guido Trotter
    as values. Rules:
139 e4376078 Iustin Pop

140 e4376078 Iustin Pop
      - use an empty dict if you don't need any lock
141 e4376078 Iustin Pop
      - if you don't need any lock at a particular level omit that level
142 e4376078 Iustin Pop
      - don't put anything for the BGL level
143 e4376078 Iustin Pop
      - if you want all locks at a level use locking.ALL_SET as a value
144 d465bdc8 Guido Trotter

145 3977a4c1 Guido Trotter
    If you need to share locks (rather than acquire them exclusively) at one
146 3977a4c1 Guido Trotter
    level you can modify self.share_locks, setting a true value (usually 1) for
147 3977a4c1 Guido Trotter
    that level. By default locks are not shared.
148 3977a4c1 Guido Trotter

149 e4376078 Iustin Pop
    Examples::
150 e4376078 Iustin Pop

151 e4376078 Iustin Pop
      # Acquire all nodes and one instance
152 e4376078 Iustin Pop
      self.needed_locks = {
153 e4376078 Iustin Pop
        locking.LEVEL_NODE: locking.ALL_SET,
154 e4376078 Iustin Pop
        locking.LEVEL_INSTANCE: ['instance1.example.tld'],
155 e4376078 Iustin Pop
      }
156 e4376078 Iustin Pop
      # Acquire just two nodes
157 e4376078 Iustin Pop
      self.needed_locks = {
158 e4376078 Iustin Pop
        locking.LEVEL_NODE: ['node1.example.tld', 'node2.example.tld'],
159 e4376078 Iustin Pop
      }
160 e4376078 Iustin Pop
      # Acquire no locks
161 e4376078 Iustin Pop
      self.needed_locks = {} # No, you can't leave it to the default value None
162 d465bdc8 Guido Trotter

163 d465bdc8 Guido Trotter
    """
164 d465bdc8 Guido Trotter
    # The implementation of this method is mandatory only if the new LU is
165 d465bdc8 Guido Trotter
    # concurrent, so that old LUs don't need to be changed all at the same
166 d465bdc8 Guido Trotter
    # time.
167 d465bdc8 Guido Trotter
    if self.REQ_BGL:
168 d465bdc8 Guido Trotter
      self.needed_locks = {} # Exclusive LUs don't need locks.
169 d465bdc8 Guido Trotter
    else:
170 d465bdc8 Guido Trotter
      raise NotImplementedError
171 d465bdc8 Guido Trotter
172 fb8dcb62 Guido Trotter
  def DeclareLocks(self, level):
173 fb8dcb62 Guido Trotter
    """Declare LU locking needs for a level
174 fb8dcb62 Guido Trotter

175 fb8dcb62 Guido Trotter
    While most LUs can just declare their locking needs at ExpandNames time,
176 fb8dcb62 Guido Trotter
    sometimes there's the need to calculate some locks after having acquired
177 fb8dcb62 Guido Trotter
    the ones before. This function is called just before acquiring locks at a
178 fb8dcb62 Guido Trotter
    particular level, but after acquiring the ones at lower levels, and permits
179 fb8dcb62 Guido Trotter
    such calculations. It can be used to modify self.needed_locks, and by
180 fb8dcb62 Guido Trotter
    default it does nothing.
181 fb8dcb62 Guido Trotter

182 fb8dcb62 Guido Trotter
    This function is only called if you have something already set in
183 fb8dcb62 Guido Trotter
    self.needed_locks for the level.
184 fb8dcb62 Guido Trotter

185 fb8dcb62 Guido Trotter
    @param level: Locking level which is going to be locked
186 fb8dcb62 Guido Trotter
    @type level: member of ganeti.locking.LEVELS
187 fb8dcb62 Guido Trotter

188 fb8dcb62 Guido Trotter
    """
189 fb8dcb62 Guido Trotter
190 a8083063 Iustin Pop
  def CheckPrereq(self):
191 a8083063 Iustin Pop
    """Check prerequisites for this LU.
192 a8083063 Iustin Pop

193 a8083063 Iustin Pop
    This method should check that the prerequisites for the execution
194 a8083063 Iustin Pop
    of this LU are fulfilled. It can do internode communication, but
195 a8083063 Iustin Pop
    it should be idempotent - no cluster or system changes are
196 a8083063 Iustin Pop
    allowed.
197 a8083063 Iustin Pop

198 a8083063 Iustin Pop
    The method should raise errors.OpPrereqError in case something is
199 a8083063 Iustin Pop
    not fulfilled. Its return value is ignored.
200 a8083063 Iustin Pop

201 a8083063 Iustin Pop
    This method should also update all the parameters of the opcode to
202 d465bdc8 Guido Trotter
    their canonical form if it hasn't been done by ExpandNames before.
203 a8083063 Iustin Pop

204 a8083063 Iustin Pop
    """
205 a8083063 Iustin Pop
    raise NotImplementedError
206 a8083063 Iustin Pop
207 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
208 a8083063 Iustin Pop
    """Execute the LU.
209 a8083063 Iustin Pop

210 a8083063 Iustin Pop
    This method should implement the actual work. It should raise
211 a8083063 Iustin Pop
    errors.OpExecError for failures that are somewhat dealt with in
212 a8083063 Iustin Pop
    code, or expected.
213 a8083063 Iustin Pop

214 a8083063 Iustin Pop
    """
215 a8083063 Iustin Pop
    raise NotImplementedError
216 a8083063 Iustin Pop
217 a8083063 Iustin Pop
  def BuildHooksEnv(self):
218 a8083063 Iustin Pop
    """Build hooks environment for this LU.
219 a8083063 Iustin Pop

220 a8083063 Iustin Pop
    This method should return a three-node tuple consisting of: a dict
221 a8083063 Iustin Pop
    containing the environment that will be used for running the
222 a8083063 Iustin Pop
    specific hook for this LU, a list of node names on which the hook
223 a8083063 Iustin Pop
    should run before the execution, and a list of node names on which
224 a8083063 Iustin Pop
    the hook should run after the execution.
225 a8083063 Iustin Pop

226 a8083063 Iustin Pop
    The keys of the dict must not have 'GANETI_' prefixed as this will
227 a8083063 Iustin Pop
    be handled in the hooks runner. Also note additional keys will be
228 a8083063 Iustin Pop
    added by the hooks runner. If the LU doesn't define any
229 a8083063 Iustin Pop
    environment, an empty dict (and not None) should be returned.
230 a8083063 Iustin Pop

231 8a3fe350 Guido Trotter
    No nodes should be returned as an empty list (and not None).
232 a8083063 Iustin Pop

233 a8083063 Iustin Pop
    Note that if the HPATH for a LU class is None, this function will
234 a8083063 Iustin Pop
    not be called.
235 a8083063 Iustin Pop

236 a8083063 Iustin Pop
    """
237 a8083063 Iustin Pop
    raise NotImplementedError
238 a8083063 Iustin Pop
239 1fce5219 Guido Trotter
  def HooksCallBack(self, phase, hook_results, feedback_fn, lu_result):
240 1fce5219 Guido Trotter
    """Notify the LU about the results of its hooks.
241 1fce5219 Guido Trotter

242 1fce5219 Guido Trotter
    This method is called every time a hooks phase is executed, and notifies
243 1fce5219 Guido Trotter
    the Logical Unit about the hooks' result. The LU can then use it to alter
244 1fce5219 Guido Trotter
    its result based on the hooks.  By default the method does nothing and the
245 1fce5219 Guido Trotter
    previous result is passed back unchanged but any LU can define it if it
246 1fce5219 Guido Trotter
    wants to use the local cluster hook-scripts somehow.
247 1fce5219 Guido Trotter

248 e4376078 Iustin Pop
    @param phase: one of L{constants.HOOKS_PHASE_POST} or
249 e4376078 Iustin Pop
        L{constants.HOOKS_PHASE_PRE}; it denotes the hooks phase
250 e4376078 Iustin Pop
    @param hook_results: the results of the multi-node hooks rpc call
251 e4376078 Iustin Pop
    @param feedback_fn: function used send feedback back to the caller
252 e4376078 Iustin Pop
    @param lu_result: the previous Exec result this LU had, or None
253 e4376078 Iustin Pop
        in the PRE phase
254 e4376078 Iustin Pop
    @return: the new Exec result, based on the previous result
255 e4376078 Iustin Pop
        and hook results
256 1fce5219 Guido Trotter

257 1fce5219 Guido Trotter
    """
258 1fce5219 Guido Trotter
    return lu_result
259 1fce5219 Guido Trotter
260 43905206 Guido Trotter
  def _ExpandAndLockInstance(self):
261 43905206 Guido Trotter
    """Helper function to expand and lock an instance.
262 43905206 Guido Trotter

263 43905206 Guido Trotter
    Many LUs that work on an instance take its name in self.op.instance_name
264 43905206 Guido Trotter
    and need to expand it and then declare the expanded name for locking. This
265 43905206 Guido Trotter
    function does it, and then updates self.op.instance_name to the expanded
266 43905206 Guido Trotter
    name. It also initializes needed_locks as a dict, if this hasn't been done
267 43905206 Guido Trotter
    before.
268 43905206 Guido Trotter

269 43905206 Guido Trotter
    """
270 43905206 Guido Trotter
    if self.needed_locks is None:
271 43905206 Guido Trotter
      self.needed_locks = {}
272 43905206 Guido Trotter
    else:
273 43905206 Guido Trotter
      assert locking.LEVEL_INSTANCE not in self.needed_locks, \
274 43905206 Guido Trotter
        "_ExpandAndLockInstance called with instance-level locks set"
275 43905206 Guido Trotter
    expanded_name = self.cfg.ExpandInstanceName(self.op.instance_name)
276 43905206 Guido Trotter
    if expanded_name is None:
277 43905206 Guido Trotter
      raise errors.OpPrereqError("Instance '%s' not known" %
278 43905206 Guido Trotter
                                  self.op.instance_name)
279 43905206 Guido Trotter
    self.needed_locks[locking.LEVEL_INSTANCE] = expanded_name
280 43905206 Guido Trotter
    self.op.instance_name = expanded_name
281 43905206 Guido Trotter
282 a82ce292 Guido Trotter
  def _LockInstancesNodes(self, primary_only=False):
283 c4a2fee1 Guido Trotter
    """Helper function to declare instances' nodes for locking.
284 c4a2fee1 Guido Trotter

285 c4a2fee1 Guido Trotter
    This function should be called after locking one or more instances to lock
286 c4a2fee1 Guido Trotter
    their nodes. Its effect is populating self.needed_locks[locking.LEVEL_NODE]
287 c4a2fee1 Guido Trotter
    with all primary or secondary nodes for instances already locked and
288 c4a2fee1 Guido Trotter
    present in self.needed_locks[locking.LEVEL_INSTANCE].
289 c4a2fee1 Guido Trotter

290 c4a2fee1 Guido Trotter
    It should be called from DeclareLocks, and for safety only works if
291 c4a2fee1 Guido Trotter
    self.recalculate_locks[locking.LEVEL_NODE] is set.
292 c4a2fee1 Guido Trotter

293 c4a2fee1 Guido Trotter
    In the future it may grow parameters to just lock some instance's nodes, or
294 c4a2fee1 Guido Trotter
    to just lock primaries or secondary nodes, if needed.
295 c4a2fee1 Guido Trotter

296 e4376078 Iustin Pop
    If should be called in DeclareLocks in a way similar to::
297 c4a2fee1 Guido Trotter

298 e4376078 Iustin Pop
      if level == locking.LEVEL_NODE:
299 e4376078 Iustin Pop
        self._LockInstancesNodes()
300 c4a2fee1 Guido Trotter

301 a82ce292 Guido Trotter
    @type primary_only: boolean
302 a82ce292 Guido Trotter
    @param primary_only: only lock primary nodes of locked instances
303 a82ce292 Guido Trotter

304 c4a2fee1 Guido Trotter
    """
305 c4a2fee1 Guido Trotter
    assert locking.LEVEL_NODE in self.recalculate_locks, \
306 c4a2fee1 Guido Trotter
      "_LockInstancesNodes helper function called with no nodes to recalculate"
307 c4a2fee1 Guido Trotter
308 c4a2fee1 Guido Trotter
    # TODO: check if we're really been called with the instance locks held
309 c4a2fee1 Guido Trotter
310 c4a2fee1 Guido Trotter
    # For now we'll replace self.needed_locks[locking.LEVEL_NODE], but in the
311 c4a2fee1 Guido Trotter
    # future we might want to have different behaviors depending on the value
312 c4a2fee1 Guido Trotter
    # of self.recalculate_locks[locking.LEVEL_NODE]
313 c4a2fee1 Guido Trotter
    wanted_nodes = []
314 6683bba2 Guido Trotter
    for instance_name in self.acquired_locks[locking.LEVEL_INSTANCE]:
315 c4a2fee1 Guido Trotter
      instance = self.context.cfg.GetInstanceInfo(instance_name)
316 c4a2fee1 Guido Trotter
      wanted_nodes.append(instance.primary_node)
317 a82ce292 Guido Trotter
      if not primary_only:
318 a82ce292 Guido Trotter
        wanted_nodes.extend(instance.secondary_nodes)
319 9513b6ab Guido Trotter
320 9513b6ab Guido Trotter
    if self.recalculate_locks[locking.LEVEL_NODE] == constants.LOCKS_REPLACE:
321 9513b6ab Guido Trotter
      self.needed_locks[locking.LEVEL_NODE] = wanted_nodes
322 9513b6ab Guido Trotter
    elif self.recalculate_locks[locking.LEVEL_NODE] == constants.LOCKS_APPEND:
323 9513b6ab Guido Trotter
      self.needed_locks[locking.LEVEL_NODE].extend(wanted_nodes)
324 c4a2fee1 Guido Trotter
325 c4a2fee1 Guido Trotter
    del self.recalculate_locks[locking.LEVEL_NODE]
326 c4a2fee1 Guido Trotter
327 a8083063 Iustin Pop
328 a8083063 Iustin Pop
class NoHooksLU(LogicalUnit):
329 a8083063 Iustin Pop
  """Simple LU which runs no hooks.
330 a8083063 Iustin Pop

331 a8083063 Iustin Pop
  This LU is intended as a parent for other LogicalUnits which will
332 a8083063 Iustin Pop
  run no hooks, in order to reduce duplicate code.
333 a8083063 Iustin Pop

334 a8083063 Iustin Pop
  """
335 a8083063 Iustin Pop
  HPATH = None
336 a8083063 Iustin Pop
  HTYPE = None
337 a8083063 Iustin Pop
338 a8083063 Iustin Pop
339 dcb93971 Michael Hanselmann
def _GetWantedNodes(lu, nodes):
340 a7ba5e53 Iustin Pop
  """Returns list of checked and expanded node names.
341 83120a01 Michael Hanselmann

342 e4376078 Iustin Pop
  @type lu: L{LogicalUnit}
343 e4376078 Iustin Pop
  @param lu: the logical unit on whose behalf we execute
344 e4376078 Iustin Pop
  @type nodes: list
345 e4376078 Iustin Pop
  @param nodes: list of node names or None for all nodes
346 e4376078 Iustin Pop
  @rtype: list
347 e4376078 Iustin Pop
  @return: the list of nodes, sorted
348 e4376078 Iustin Pop
  @raise errors.OpProgrammerError: if the nodes parameter is wrong type
349 83120a01 Michael Hanselmann

350 83120a01 Michael Hanselmann
  """
351 3312b702 Iustin Pop
  if not isinstance(nodes, list):
352 3ecf6786 Iustin Pop
    raise errors.OpPrereqError("Invalid argument type 'nodes'")
353 dcb93971 Michael Hanselmann
354 ea47808a Guido Trotter
  if not nodes:
355 ea47808a Guido Trotter
    raise errors.ProgrammerError("_GetWantedNodes should only be called with a"
356 ea47808a Guido Trotter
      " non-empty list of nodes whose name is to be expanded.")
357 dcb93971 Michael Hanselmann
358 ea47808a Guido Trotter
  wanted = []
359 ea47808a Guido Trotter
  for name in nodes:
360 ea47808a Guido Trotter
    node = lu.cfg.ExpandNodeName(name)
361 ea47808a Guido Trotter
    if node is None:
362 ea47808a Guido Trotter
      raise errors.OpPrereqError("No such node name '%s'" % name)
363 ea47808a Guido Trotter
    wanted.append(node)
364 dcb93971 Michael Hanselmann
365 a7ba5e53 Iustin Pop
  return utils.NiceSort(wanted)
366 3312b702 Iustin Pop
367 3312b702 Iustin Pop
368 3312b702 Iustin Pop
def _GetWantedInstances(lu, instances):
369 a7ba5e53 Iustin Pop
  """Returns list of checked and expanded instance names.
370 3312b702 Iustin Pop

371 e4376078 Iustin Pop
  @type lu: L{LogicalUnit}
372 e4376078 Iustin Pop
  @param lu: the logical unit on whose behalf we execute
373 e4376078 Iustin Pop
  @type instances: list
374 e4376078 Iustin Pop
  @param instances: list of instance names or None for all instances
375 e4376078 Iustin Pop
  @rtype: list
376 e4376078 Iustin Pop
  @return: the list of instances, sorted
377 e4376078 Iustin Pop
  @raise errors.OpPrereqError: if the instances parameter is wrong type
378 e4376078 Iustin Pop
  @raise errors.OpPrereqError: if any of the passed instances is not found
379 3312b702 Iustin Pop

380 3312b702 Iustin Pop
  """
381 3312b702 Iustin Pop
  if not isinstance(instances, list):
382 3312b702 Iustin Pop
    raise errors.OpPrereqError("Invalid argument type 'instances'")
383 3312b702 Iustin Pop
384 3312b702 Iustin Pop
  if instances:
385 3312b702 Iustin Pop
    wanted = []
386 3312b702 Iustin Pop
387 3312b702 Iustin Pop
    for name in instances:
388 a7ba5e53 Iustin Pop
      instance = lu.cfg.ExpandInstanceName(name)
389 3312b702 Iustin Pop
      if instance is None:
390 3312b702 Iustin Pop
        raise errors.OpPrereqError("No such instance name '%s'" % name)
391 3312b702 Iustin Pop
      wanted.append(instance)
392 3312b702 Iustin Pop
393 3312b702 Iustin Pop
  else:
394 a7f5dc98 Iustin Pop
    wanted = utils.NiceSort(lu.cfg.GetInstanceList())
395 a7f5dc98 Iustin Pop
  return wanted
396 dcb93971 Michael Hanselmann
397 dcb93971 Michael Hanselmann
398 dcb93971 Michael Hanselmann
def _CheckOutputFields(static, dynamic, selected):
399 83120a01 Michael Hanselmann
  """Checks whether all selected fields are valid.
400 83120a01 Michael Hanselmann

401 a2d2e1a7 Iustin Pop
  @type static: L{utils.FieldSet}
402 31bf511f Iustin Pop
  @param static: static fields set
403 a2d2e1a7 Iustin Pop
  @type dynamic: L{utils.FieldSet}
404 31bf511f Iustin Pop
  @param dynamic: dynamic fields set
405 83120a01 Michael Hanselmann

406 83120a01 Michael Hanselmann
  """
407 a2d2e1a7 Iustin Pop
  f = utils.FieldSet()
408 31bf511f Iustin Pop
  f.Extend(static)
409 31bf511f Iustin Pop
  f.Extend(dynamic)
410 dcb93971 Michael Hanselmann
411 31bf511f Iustin Pop
  delta = f.NonMatching(selected)
412 31bf511f Iustin Pop
  if delta:
413 3ecf6786 Iustin Pop
    raise errors.OpPrereqError("Unknown output fields selected: %s"
414 31bf511f Iustin Pop
                               % ",".join(delta))
415 dcb93971 Michael Hanselmann
416 dcb93971 Michael Hanselmann
417 a5961235 Iustin Pop
def _CheckBooleanOpField(op, name):
418 a5961235 Iustin Pop
  """Validates boolean opcode parameters.
419 a5961235 Iustin Pop

420 a5961235 Iustin Pop
  This will ensure that an opcode parameter is either a boolean value,
421 a5961235 Iustin Pop
  or None (but that it always exists).
422 a5961235 Iustin Pop

423 a5961235 Iustin Pop
  """
424 a5961235 Iustin Pop
  val = getattr(op, name, None)
425 a5961235 Iustin Pop
  if not (val is None or isinstance(val, bool)):
426 a5961235 Iustin Pop
    raise errors.OpPrereqError("Invalid boolean parameter '%s' (%s)" %
427 a5961235 Iustin Pop
                               (name, str(val)))
428 a5961235 Iustin Pop
  setattr(op, name, val)
429 a5961235 Iustin Pop
430 a5961235 Iustin Pop
431 a5961235 Iustin Pop
def _CheckNodeOnline(lu, node):
432 a5961235 Iustin Pop
  """Ensure that a given node is online.
433 a5961235 Iustin Pop

434 a5961235 Iustin Pop
  @param lu: the LU on behalf of which we make the check
435 a5961235 Iustin Pop
  @param node: the node to check
436 733a2b6a Iustin Pop
  @raise errors.OpPrereqError: if the node is offline
437 a5961235 Iustin Pop

438 a5961235 Iustin Pop
  """
439 a5961235 Iustin Pop
  if lu.cfg.GetNodeInfo(node).offline:
440 a5961235 Iustin Pop
    raise errors.OpPrereqError("Can't use offline node %s" % node)
441 a5961235 Iustin Pop
442 a5961235 Iustin Pop
443 733a2b6a Iustin Pop
def _CheckNodeNotDrained(lu, node):
444 733a2b6a Iustin Pop
  """Ensure that a given node is not drained.
445 733a2b6a Iustin Pop

446 733a2b6a Iustin Pop
  @param lu: the LU on behalf of which we make the check
447 733a2b6a Iustin Pop
  @param node: the node to check
448 733a2b6a Iustin Pop
  @raise errors.OpPrereqError: if the node is drained
449 733a2b6a Iustin Pop

450 733a2b6a Iustin Pop
  """
451 733a2b6a Iustin Pop
  if lu.cfg.GetNodeInfo(node).drained:
452 733a2b6a Iustin Pop
    raise errors.OpPrereqError("Can't use drained node %s" % node)
453 733a2b6a Iustin Pop
454 733a2b6a Iustin Pop
455 ecb215b5 Michael Hanselmann
def _BuildInstanceHookEnv(name, primary_node, secondary_nodes, os_type, status,
456 2c2690c9 Iustin Pop
                          memory, vcpus, nics, disk_template, disks):
457 e4376078 Iustin Pop
  """Builds instance related env variables for hooks
458 e4376078 Iustin Pop

459 e4376078 Iustin Pop
  This builds the hook environment from individual variables.
460 e4376078 Iustin Pop

461 e4376078 Iustin Pop
  @type name: string
462 e4376078 Iustin Pop
  @param name: the name of the instance
463 e4376078 Iustin Pop
  @type primary_node: string
464 e4376078 Iustin Pop
  @param primary_node: the name of the instance's primary node
465 e4376078 Iustin Pop
  @type secondary_nodes: list
466 e4376078 Iustin Pop
  @param secondary_nodes: list of secondary nodes as strings
467 e4376078 Iustin Pop
  @type os_type: string
468 e4376078 Iustin Pop
  @param os_type: the name of the instance's OS
469 0d68c45d Iustin Pop
  @type status: boolean
470 0d68c45d Iustin Pop
  @param status: the should_run status of the instance
471 e4376078 Iustin Pop
  @type memory: string
472 e4376078 Iustin Pop
  @param memory: the memory size of the instance
473 e4376078 Iustin Pop
  @type vcpus: string
474 e4376078 Iustin Pop
  @param vcpus: the count of VCPUs the instance has
475 e4376078 Iustin Pop
  @type nics: list
476 e4376078 Iustin Pop
  @param nics: list of tuples (ip, bridge, mac) representing
477 e4376078 Iustin Pop
      the NICs the instance  has
478 2c2690c9 Iustin Pop
  @type disk_template: string
479 2c2690c9 Iustin Pop
  @param disk_template: the distk template of the instance
480 2c2690c9 Iustin Pop
  @type disks: list
481 2c2690c9 Iustin Pop
  @param disks: the list of (size, mode) pairs
482 e4376078 Iustin Pop
  @rtype: dict
483 e4376078 Iustin Pop
  @return: the hook environment for this instance
484 ecb215b5 Michael Hanselmann

485 396e1b78 Michael Hanselmann
  """
486 0d68c45d Iustin Pop
  if status:
487 0d68c45d Iustin Pop
    str_status = "up"
488 0d68c45d Iustin Pop
  else:
489 0d68c45d Iustin Pop
    str_status = "down"
490 396e1b78 Michael Hanselmann
  env = {
491 0e137c28 Iustin Pop
    "OP_TARGET": name,
492 396e1b78 Michael Hanselmann
    "INSTANCE_NAME": name,
493 396e1b78 Michael Hanselmann
    "INSTANCE_PRIMARY": primary_node,
494 396e1b78 Michael Hanselmann
    "INSTANCE_SECONDARIES": " ".join(secondary_nodes),
495 ecb215b5 Michael Hanselmann
    "INSTANCE_OS_TYPE": os_type,
496 0d68c45d Iustin Pop
    "INSTANCE_STATUS": str_status,
497 396e1b78 Michael Hanselmann
    "INSTANCE_MEMORY": memory,
498 396e1b78 Michael Hanselmann
    "INSTANCE_VCPUS": vcpus,
499 2c2690c9 Iustin Pop
    "INSTANCE_DISK_TEMPLATE": disk_template,
500 396e1b78 Michael Hanselmann
  }
501 396e1b78 Michael Hanselmann
502 396e1b78 Michael Hanselmann
  if nics:
503 396e1b78 Michael Hanselmann
    nic_count = len(nics)
504 53e4e875 Guido Trotter
    for idx, (ip, bridge, mac) in enumerate(nics):
505 396e1b78 Michael Hanselmann
      if ip is None:
506 396e1b78 Michael Hanselmann
        ip = ""
507 396e1b78 Michael Hanselmann
      env["INSTANCE_NIC%d_IP" % idx] = ip
508 396e1b78 Michael Hanselmann
      env["INSTANCE_NIC%d_BRIDGE" % idx] = bridge
509 2c2690c9 Iustin Pop
      env["INSTANCE_NIC%d_MAC" % idx] = mac
510 396e1b78 Michael Hanselmann
  else:
511 396e1b78 Michael Hanselmann
    nic_count = 0
512 396e1b78 Michael Hanselmann
513 396e1b78 Michael Hanselmann
  env["INSTANCE_NIC_COUNT"] = nic_count
514 396e1b78 Michael Hanselmann
515 2c2690c9 Iustin Pop
  if disks:
516 2c2690c9 Iustin Pop
    disk_count = len(disks)
517 2c2690c9 Iustin Pop
    for idx, (size, mode) in enumerate(disks):
518 2c2690c9 Iustin Pop
      env["INSTANCE_DISK%d_SIZE" % idx] = size
519 2c2690c9 Iustin Pop
      env["INSTANCE_DISK%d_MODE" % idx] = mode
520 2c2690c9 Iustin Pop
  else:
521 2c2690c9 Iustin Pop
    disk_count = 0
522 2c2690c9 Iustin Pop
523 2c2690c9 Iustin Pop
  env["INSTANCE_DISK_COUNT"] = disk_count
524 2c2690c9 Iustin Pop
525 396e1b78 Michael Hanselmann
  return env
526 396e1b78 Michael Hanselmann
527 396e1b78 Michael Hanselmann
528 338e51e8 Iustin Pop
def _BuildInstanceHookEnvByObject(lu, instance, override=None):
529 ecb215b5 Michael Hanselmann
  """Builds instance related env variables for hooks from an object.
530 ecb215b5 Michael Hanselmann

531 e4376078 Iustin Pop
  @type lu: L{LogicalUnit}
532 e4376078 Iustin Pop
  @param lu: the logical unit on whose behalf we execute
533 e4376078 Iustin Pop
  @type instance: L{objects.Instance}
534 e4376078 Iustin Pop
  @param instance: the instance for which we should build the
535 e4376078 Iustin Pop
      environment
536 e4376078 Iustin Pop
  @type override: dict
537 e4376078 Iustin Pop
  @param override: dictionary with key/values that will override
538 e4376078 Iustin Pop
      our values
539 e4376078 Iustin Pop
  @rtype: dict
540 e4376078 Iustin Pop
  @return: the hook environment dictionary
541 e4376078 Iustin Pop

542 ecb215b5 Michael Hanselmann
  """
543 338e51e8 Iustin Pop
  bep = lu.cfg.GetClusterInfo().FillBE(instance)
544 396e1b78 Michael Hanselmann
  args = {
545 396e1b78 Michael Hanselmann
    'name': instance.name,
546 396e1b78 Michael Hanselmann
    'primary_node': instance.primary_node,
547 396e1b78 Michael Hanselmann
    'secondary_nodes': instance.secondary_nodes,
548 ecb215b5 Michael Hanselmann
    'os_type': instance.os,
549 0d68c45d Iustin Pop
    'status': instance.admin_up,
550 338e51e8 Iustin Pop
    'memory': bep[constants.BE_MEMORY],
551 338e51e8 Iustin Pop
    'vcpus': bep[constants.BE_VCPUS],
552 53e4e875 Guido Trotter
    'nics': [(nic.ip, nic.bridge, nic.mac) for nic in instance.nics],
553 2c2690c9 Iustin Pop
    'disk_template': instance.disk_template,
554 2c2690c9 Iustin Pop
    'disks': [(disk.size, disk.mode) for disk in instance.disks],
555 396e1b78 Michael Hanselmann
  }
556 396e1b78 Michael Hanselmann
  if override:
557 396e1b78 Michael Hanselmann
    args.update(override)
558 396e1b78 Michael Hanselmann
  return _BuildInstanceHookEnv(**args)
559 396e1b78 Michael Hanselmann
560 396e1b78 Michael Hanselmann
561 ec0292f1 Iustin Pop
def _AdjustCandidatePool(lu):
562 ec0292f1 Iustin Pop
  """Adjust the candidate pool after node operations.
563 ec0292f1 Iustin Pop

564 ec0292f1 Iustin Pop
  """
565 ec0292f1 Iustin Pop
  mod_list = lu.cfg.MaintainCandidatePool()
566 ec0292f1 Iustin Pop
  if mod_list:
567 ec0292f1 Iustin Pop
    lu.LogInfo("Promoted nodes to master candidate role: %s",
568 ee513a66 Iustin Pop
               ", ".join(node.name for node in mod_list))
569 ec0292f1 Iustin Pop
    for name in mod_list:
570 ec0292f1 Iustin Pop
      lu.context.ReaddNode(name)
571 ec0292f1 Iustin Pop
  mc_now, mc_max = lu.cfg.GetMasterCandidateStats()
572 ec0292f1 Iustin Pop
  if mc_now > mc_max:
573 ec0292f1 Iustin Pop
    lu.LogInfo("Note: more nodes are candidates (%d) than desired (%d)" %
574 ec0292f1 Iustin Pop
               (mc_now, mc_max))
575 ec0292f1 Iustin Pop
576 ec0292f1 Iustin Pop
577 b9bddb6b Iustin Pop
def _CheckInstanceBridgesExist(lu, instance):
578 bf6929a2 Alexander Schreiber
  """Check that the brigdes needed by an instance exist.
579 bf6929a2 Alexander Schreiber

580 bf6929a2 Alexander Schreiber
  """
581 bf6929a2 Alexander Schreiber
  # check bridges existance
582 bf6929a2 Alexander Schreiber
  brlist = [nic.bridge for nic in instance.nics]
583 781de953 Iustin Pop
  result = lu.rpc.call_bridges_exist(instance.primary_node, brlist)
584 781de953 Iustin Pop
  result.Raise()
585 781de953 Iustin Pop
  if not result.data:
586 781de953 Iustin Pop
    raise errors.OpPrereqError("One or more target bridges %s does not"
587 bf6929a2 Alexander Schreiber
                               " exist on destination node '%s'" %
588 bf6929a2 Alexander Schreiber
                               (brlist, instance.primary_node))
589 bf6929a2 Alexander Schreiber
590 bf6929a2 Alexander Schreiber
591 a8083063 Iustin Pop
class LUDestroyCluster(NoHooksLU):
592 a8083063 Iustin Pop
  """Logical unit for destroying the cluster.
593 a8083063 Iustin Pop

594 a8083063 Iustin Pop
  """
595 a8083063 Iustin Pop
  _OP_REQP = []
596 a8083063 Iustin Pop
597 a8083063 Iustin Pop
  def CheckPrereq(self):
598 a8083063 Iustin Pop
    """Check prerequisites.
599 a8083063 Iustin Pop

600 a8083063 Iustin Pop
    This checks whether the cluster is empty.
601 a8083063 Iustin Pop

602 a8083063 Iustin Pop
    Any errors are signalled by raising errors.OpPrereqError.
603 a8083063 Iustin Pop

604 a8083063 Iustin Pop
    """
605 d6a02168 Michael Hanselmann
    master = self.cfg.GetMasterNode()
606 a8083063 Iustin Pop
607 a8083063 Iustin Pop
    nodelist = self.cfg.GetNodeList()
608 db915bd1 Michael Hanselmann
    if len(nodelist) != 1 or nodelist[0] != master:
609 3ecf6786 Iustin Pop
      raise errors.OpPrereqError("There are still %d node(s) in"
610 3ecf6786 Iustin Pop
                                 " this cluster." % (len(nodelist) - 1))
611 db915bd1 Michael Hanselmann
    instancelist = self.cfg.GetInstanceList()
612 db915bd1 Michael Hanselmann
    if instancelist:
613 3ecf6786 Iustin Pop
      raise errors.OpPrereqError("There are still %d instance(s) in"
614 3ecf6786 Iustin Pop
                                 " this cluster." % len(instancelist))
615 a8083063 Iustin Pop
616 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
617 a8083063 Iustin Pop
    """Destroys the cluster.
618 a8083063 Iustin Pop

619 a8083063 Iustin Pop
    """
620 d6a02168 Michael Hanselmann
    master = self.cfg.GetMasterNode()
621 781de953 Iustin Pop
    result = self.rpc.call_node_stop_master(master, False)
622 781de953 Iustin Pop
    result.Raise()
623 781de953 Iustin Pop
    if not result.data:
624 c9064964 Iustin Pop
      raise errors.OpExecError("Could not disable the master role")
625 70d9e3d8 Iustin Pop
    priv_key, pub_key, _ = ssh.GetUserFiles(constants.GANETI_RUNAS)
626 70d9e3d8 Iustin Pop
    utils.CreateBackup(priv_key)
627 70d9e3d8 Iustin Pop
    utils.CreateBackup(pub_key)
628 140aa4a8 Iustin Pop
    return master
629 a8083063 Iustin Pop
630 a8083063 Iustin Pop
631 d8fff41c Guido Trotter
class LUVerifyCluster(LogicalUnit):
632 a8083063 Iustin Pop
  """Verifies the cluster status.
633 a8083063 Iustin Pop

634 a8083063 Iustin Pop
  """
635 d8fff41c Guido Trotter
  HPATH = "cluster-verify"
636 d8fff41c Guido Trotter
  HTYPE = constants.HTYPE_CLUSTER
637 e54c4c5e Guido Trotter
  _OP_REQP = ["skip_checks"]
638 d4b9d97f Guido Trotter
  REQ_BGL = False
639 d4b9d97f Guido Trotter
640 d4b9d97f Guido Trotter
  def ExpandNames(self):
641 d4b9d97f Guido Trotter
    self.needed_locks = {
642 d4b9d97f Guido Trotter
      locking.LEVEL_NODE: locking.ALL_SET,
643 d4b9d97f Guido Trotter
      locking.LEVEL_INSTANCE: locking.ALL_SET,
644 d4b9d97f Guido Trotter
    }
645 d4b9d97f Guido Trotter
    self.share_locks = dict(((i, 1) for i in locking.LEVELS))
646 a8083063 Iustin Pop
647 25361b9a Iustin Pop
  def _VerifyNode(self, nodeinfo, file_list, local_cksum,
648 6d2e83d5 Iustin Pop
                  node_result, feedback_fn, master_files,
649 cc9e1230 Guido Trotter
                  drbd_map, vg_name):
650 a8083063 Iustin Pop
    """Run multiple tests against a node.
651 a8083063 Iustin Pop

652 112f18a5 Iustin Pop
    Test list:
653 e4376078 Iustin Pop

654 a8083063 Iustin Pop
      - compares ganeti version
655 a8083063 Iustin Pop
      - checks vg existance and size > 20G
656 a8083063 Iustin Pop
      - checks config file checksum
657 a8083063 Iustin Pop
      - checks ssh to other nodes
658 a8083063 Iustin Pop

659 112f18a5 Iustin Pop
    @type nodeinfo: L{objects.Node}
660 112f18a5 Iustin Pop
    @param nodeinfo: the node to check
661 e4376078 Iustin Pop
    @param file_list: required list of files
662 e4376078 Iustin Pop
    @param local_cksum: dictionary of local files and their checksums
663 e4376078 Iustin Pop
    @param node_result: the results from the node
664 e4376078 Iustin Pop
    @param feedback_fn: function used to accumulate results
665 112f18a5 Iustin Pop
    @param master_files: list of files that only masters should have
666 6d2e83d5 Iustin Pop
    @param drbd_map: the useddrbd minors for this node, in
667 6d2e83d5 Iustin Pop
        form of minor: (instance, must_exist) which correspond to instances
668 6d2e83d5 Iustin Pop
        and their running status
669 cc9e1230 Guido Trotter
    @param vg_name: Ganeti Volume Group (result of self.cfg.GetVGName())
670 098c0958 Michael Hanselmann

671 a8083063 Iustin Pop
    """
672 112f18a5 Iustin Pop
    node = nodeinfo.name
673 25361b9a Iustin Pop
674 25361b9a Iustin Pop
    # main result, node_result should be a non-empty dict
675 25361b9a Iustin Pop
    if not node_result or not isinstance(node_result, dict):
676 25361b9a Iustin Pop
      feedback_fn("  - ERROR: unable to verify node %s." % (node,))
677 25361b9a Iustin Pop
      return True
678 25361b9a Iustin Pop
679 a8083063 Iustin Pop
    # compares ganeti version
680 a8083063 Iustin Pop
    local_version = constants.PROTOCOL_VERSION
681 25361b9a Iustin Pop
    remote_version = node_result.get('version', None)
682 e9ce0a64 Iustin Pop
    if not (remote_version and isinstance(remote_version, (list, tuple)) and
683 e9ce0a64 Iustin Pop
            len(remote_version) == 2):
684 c840ae6f Guido Trotter
      feedback_fn("  - ERROR: connection to %s failed" % (node))
685 a8083063 Iustin Pop
      return True
686 a8083063 Iustin Pop
687 e9ce0a64 Iustin Pop
    if local_version != remote_version[0]:
688 e9ce0a64 Iustin Pop
      feedback_fn("  - ERROR: incompatible protocol versions: master %s,"
689 e9ce0a64 Iustin Pop
                  " node %s %s" % (local_version, node, remote_version[0]))
690 a8083063 Iustin Pop
      return True
691 a8083063 Iustin Pop
692 e9ce0a64 Iustin Pop
    # node seems compatible, we can actually try to look into its results
693 a8083063 Iustin Pop
694 a8083063 Iustin Pop
    bad = False
695 e9ce0a64 Iustin Pop
696 e9ce0a64 Iustin Pop
    # full package version
697 e9ce0a64 Iustin Pop
    if constants.RELEASE_VERSION != remote_version[1]:
698 e9ce0a64 Iustin Pop
      feedback_fn("  - WARNING: software version mismatch: master %s,"
699 e9ce0a64 Iustin Pop
                  " node %s %s" %
700 e9ce0a64 Iustin Pop
                  (constants.RELEASE_VERSION, node, remote_version[1]))
701 e9ce0a64 Iustin Pop
702 e9ce0a64 Iustin Pop
    # checks vg existence and size > 20G
703 cc9e1230 Guido Trotter
    if vg_name is not None:
704 cc9e1230 Guido Trotter
      vglist = node_result.get(constants.NV_VGLIST, None)
705 cc9e1230 Guido Trotter
      if not vglist:
706 cc9e1230 Guido Trotter
        feedback_fn("  - ERROR: unable to check volume groups on node %s." %
707 cc9e1230 Guido Trotter
                        (node,))
708 a8083063 Iustin Pop
        bad = True
709 cc9e1230 Guido Trotter
      else:
710 cc9e1230 Guido Trotter
        vgstatus = utils.CheckVolumeGroupSize(vglist, vg_name,
711 cc9e1230 Guido Trotter
                                              constants.MIN_VG_SIZE)
712 cc9e1230 Guido Trotter
        if vgstatus:
713 cc9e1230 Guido Trotter
          feedback_fn("  - ERROR: %s on node %s" % (vgstatus, node))
714 cc9e1230 Guido Trotter
          bad = True
715 a8083063 Iustin Pop
716 a8083063 Iustin Pop
    # checks config file checksum
717 a8083063 Iustin Pop
718 25361b9a Iustin Pop
    remote_cksum = node_result.get(constants.NV_FILELIST, None)
719 25361b9a Iustin Pop
    if not isinstance(remote_cksum, dict):
720 a8083063 Iustin Pop
      bad = True
721 a8083063 Iustin Pop
      feedback_fn("  - ERROR: node hasn't returned file checksum data")
722 a8083063 Iustin Pop
    else:
723 a8083063 Iustin Pop
      for file_name in file_list:
724 112f18a5 Iustin Pop
        node_is_mc = nodeinfo.master_candidate
725 112f18a5 Iustin Pop
        must_have_file = file_name not in master_files
726 a8083063 Iustin Pop
        if file_name not in remote_cksum:
727 112f18a5 Iustin Pop
          if node_is_mc or must_have_file:
728 112f18a5 Iustin Pop
            bad = True
729 112f18a5 Iustin Pop
            feedback_fn("  - ERROR: file '%s' missing" % file_name)
730 a8083063 Iustin Pop
        elif remote_cksum[file_name] != local_cksum[file_name]:
731 112f18a5 Iustin Pop
          if node_is_mc or must_have_file:
732 112f18a5 Iustin Pop
            bad = True
733 112f18a5 Iustin Pop
            feedback_fn("  - ERROR: file '%s' has wrong checksum" % file_name)
734 112f18a5 Iustin Pop
          else:
735 112f18a5 Iustin Pop
            # not candidate and this is not a must-have file
736 112f18a5 Iustin Pop
            bad = True
737 112f18a5 Iustin Pop
            feedback_fn("  - ERROR: non master-candidate has old/wrong file"
738 112f18a5 Iustin Pop
                        " '%s'" % file_name)
739 112f18a5 Iustin Pop
        else:
740 112f18a5 Iustin Pop
          # all good, except non-master/non-must have combination
741 112f18a5 Iustin Pop
          if not node_is_mc and not must_have_file:
742 112f18a5 Iustin Pop
            feedback_fn("  - ERROR: file '%s' should not exist on non master"
743 112f18a5 Iustin Pop
                        " candidates" % file_name)
744 a8083063 Iustin Pop
745 25361b9a Iustin Pop
    # checks ssh to any
746 25361b9a Iustin Pop
747 25361b9a Iustin Pop
    if constants.NV_NODELIST not in node_result:
748 a8083063 Iustin Pop
      bad = True
749 9d4bfc96 Iustin Pop
      feedback_fn("  - ERROR: node hasn't returned node ssh connectivity data")
750 a8083063 Iustin Pop
    else:
751 25361b9a Iustin Pop
      if node_result[constants.NV_NODELIST]:
752 a8083063 Iustin Pop
        bad = True
753 25361b9a Iustin Pop
        for node in node_result[constants.NV_NODELIST]:
754 9d4bfc96 Iustin Pop
          feedback_fn("  - ERROR: ssh communication with node '%s': %s" %
755 25361b9a Iustin Pop
                          (node, node_result[constants.NV_NODELIST][node]))
756 25361b9a Iustin Pop
757 25361b9a Iustin Pop
    if constants.NV_NODENETTEST not in node_result:
758 9d4bfc96 Iustin Pop
      bad = True
759 9d4bfc96 Iustin Pop
      feedback_fn("  - ERROR: node hasn't returned node tcp connectivity data")
760 9d4bfc96 Iustin Pop
    else:
761 25361b9a Iustin Pop
      if node_result[constants.NV_NODENETTEST]:
762 9d4bfc96 Iustin Pop
        bad = True
763 25361b9a Iustin Pop
        nlist = utils.NiceSort(node_result[constants.NV_NODENETTEST].keys())
764 9d4bfc96 Iustin Pop
        for node in nlist:
765 9d4bfc96 Iustin Pop
          feedback_fn("  - ERROR: tcp communication with node '%s': %s" %
766 25361b9a Iustin Pop
                          (node, node_result[constants.NV_NODENETTEST][node]))
767 9d4bfc96 Iustin Pop
768 25361b9a Iustin Pop
    hyp_result = node_result.get(constants.NV_HYPERVISOR, None)
769 e69d05fd Iustin Pop
    if isinstance(hyp_result, dict):
770 e69d05fd Iustin Pop
      for hv_name, hv_result in hyp_result.iteritems():
771 e69d05fd Iustin Pop
        if hv_result is not None:
772 e69d05fd Iustin Pop
          feedback_fn("  - ERROR: hypervisor %s verify failure: '%s'" %
773 e69d05fd Iustin Pop
                      (hv_name, hv_result))
774 6d2e83d5 Iustin Pop
775 6d2e83d5 Iustin Pop
    # check used drbd list
776 cc9e1230 Guido Trotter
    if vg_name is not None:
777 cc9e1230 Guido Trotter
      used_minors = node_result.get(constants.NV_DRBDLIST, [])
778 cc9e1230 Guido Trotter
      if not isinstance(used_minors, (tuple, list)):
779 cc9e1230 Guido Trotter
        feedback_fn("  - ERROR: cannot parse drbd status file: %s" %
780 cc9e1230 Guido Trotter
                    str(used_minors))
781 cc9e1230 Guido Trotter
      else:
782 cc9e1230 Guido Trotter
        for minor, (iname, must_exist) in drbd_map.items():
783 cc9e1230 Guido Trotter
          if minor not in used_minors and must_exist:
784 35e994e9 Iustin Pop
            feedback_fn("  - ERROR: drbd minor %d of instance %s is"
785 35e994e9 Iustin Pop
                        " not active" % (minor, iname))
786 cc9e1230 Guido Trotter
            bad = True
787 cc9e1230 Guido Trotter
        for minor in used_minors:
788 cc9e1230 Guido Trotter
          if minor not in drbd_map:
789 35e994e9 Iustin Pop
            feedback_fn("  - ERROR: unallocated drbd minor %d is in use" %
790 35e994e9 Iustin Pop
                        minor)
791 cc9e1230 Guido Trotter
            bad = True
792 6d2e83d5 Iustin Pop
793 a8083063 Iustin Pop
    return bad
794 a8083063 Iustin Pop
795 c5705f58 Guido Trotter
  def _VerifyInstance(self, instance, instanceconfig, node_vol_is,
796 0a66c968 Iustin Pop
                      node_instance, feedback_fn, n_offline):
797 a8083063 Iustin Pop
    """Verify an instance.
798 a8083063 Iustin Pop

799 a8083063 Iustin Pop
    This function checks to see if the required block devices are
800 a8083063 Iustin Pop
    available on the instance's node.
801 a8083063 Iustin Pop

802 a8083063 Iustin Pop
    """
803 a8083063 Iustin Pop
    bad = False
804 a8083063 Iustin Pop
805 a8083063 Iustin Pop
    node_current = instanceconfig.primary_node
806 a8083063 Iustin Pop
807 a8083063 Iustin Pop
    node_vol_should = {}
808 a8083063 Iustin Pop
    instanceconfig.MapLVsByNode(node_vol_should)
809 a8083063 Iustin Pop
810 a8083063 Iustin Pop
    for node in node_vol_should:
811 0a66c968 Iustin Pop
      if node in n_offline:
812 0a66c968 Iustin Pop
        # ignore missing volumes on offline nodes
813 0a66c968 Iustin Pop
        continue
814 a8083063 Iustin Pop
      for volume in node_vol_should[node]:
815 a8083063 Iustin Pop
        if node not in node_vol_is or volume not in node_vol_is[node]:
816 a8083063 Iustin Pop
          feedback_fn("  - ERROR: volume %s missing on node %s" %
817 a8083063 Iustin Pop
                          (volume, node))
818 a8083063 Iustin Pop
          bad = True
819 a8083063 Iustin Pop
820 0d68c45d Iustin Pop
    if instanceconfig.admin_up:
821 0a66c968 Iustin Pop
      if ((node_current not in node_instance or
822 0a66c968 Iustin Pop
          not instance in node_instance[node_current]) and
823 0a66c968 Iustin Pop
          node_current not in n_offline):
824 a8083063 Iustin Pop
        feedback_fn("  - ERROR: instance %s not running on node %s" %
825 a8083063 Iustin Pop
                        (instance, node_current))
826 a8083063 Iustin Pop
        bad = True
827 a8083063 Iustin Pop
828 a8083063 Iustin Pop
    for node in node_instance:
829 a8083063 Iustin Pop
      if (not node == node_current):
830 a8083063 Iustin Pop
        if instance in node_instance[node]:
831 a8083063 Iustin Pop
          feedback_fn("  - ERROR: instance %s should not run on node %s" %
832 a8083063 Iustin Pop
                          (instance, node))
833 a8083063 Iustin Pop
          bad = True
834 a8083063 Iustin Pop
835 6a438c98 Michael Hanselmann
    return bad
836 a8083063 Iustin Pop
837 a8083063 Iustin Pop
  def _VerifyOrphanVolumes(self, node_vol_should, node_vol_is, feedback_fn):
838 a8083063 Iustin Pop
    """Verify if there are any unknown volumes in the cluster.
839 a8083063 Iustin Pop

840 a8083063 Iustin Pop
    The .os, .swap and backup volumes are ignored. All other volumes are
841 a8083063 Iustin Pop
    reported as unknown.
842 a8083063 Iustin Pop

843 a8083063 Iustin Pop
    """
844 a8083063 Iustin Pop
    bad = False
845 a8083063 Iustin Pop
846 a8083063 Iustin Pop
    for node in node_vol_is:
847 a8083063 Iustin Pop
      for volume in node_vol_is[node]:
848 a8083063 Iustin Pop
        if node not in node_vol_should or volume not in node_vol_should[node]:
849 a8083063 Iustin Pop
          feedback_fn("  - ERROR: volume %s on node %s should not exist" %
850 a8083063 Iustin Pop
                      (volume, node))
851 a8083063 Iustin Pop
          bad = True
852 a8083063 Iustin Pop
    return bad
853 a8083063 Iustin Pop
854 a8083063 Iustin Pop
  def _VerifyOrphanInstances(self, instancelist, node_instance, feedback_fn):
855 a8083063 Iustin Pop
    """Verify the list of running instances.
856 a8083063 Iustin Pop

857 a8083063 Iustin Pop
    This checks what instances are running but unknown to the cluster.
858 a8083063 Iustin Pop

859 a8083063 Iustin Pop
    """
860 a8083063 Iustin Pop
    bad = False
861 a8083063 Iustin Pop
    for node in node_instance:
862 a8083063 Iustin Pop
      for runninginstance in node_instance[node]:
863 a8083063 Iustin Pop
        if runninginstance not in instancelist:
864 a8083063 Iustin Pop
          feedback_fn("  - ERROR: instance %s on node %s should not exist" %
865 a8083063 Iustin Pop
                          (runninginstance, node))
866 a8083063 Iustin Pop
          bad = True
867 a8083063 Iustin Pop
    return bad
868 a8083063 Iustin Pop
869 2b3b6ddd Guido Trotter
  def _VerifyNPlusOneMemory(self, node_info, instance_cfg, feedback_fn):
870 2b3b6ddd Guido Trotter
    """Verify N+1 Memory Resilience.
871 2b3b6ddd Guido Trotter

872 2b3b6ddd Guido Trotter
    Check that if one single node dies we can still start all the instances it
873 2b3b6ddd Guido Trotter
    was primary for.
874 2b3b6ddd Guido Trotter

875 2b3b6ddd Guido Trotter
    """
876 2b3b6ddd Guido Trotter
    bad = False
877 2b3b6ddd Guido Trotter
878 2b3b6ddd Guido Trotter
    for node, nodeinfo in node_info.iteritems():
879 2b3b6ddd Guido Trotter
      # This code checks that every node which is now listed as secondary has
880 2b3b6ddd Guido Trotter
      # enough memory to host all instances it is supposed to should a single
881 2b3b6ddd Guido Trotter
      # other node in the cluster fail.
882 2b3b6ddd Guido Trotter
      # FIXME: not ready for failover to an arbitrary node
883 2b3b6ddd Guido Trotter
      # FIXME: does not support file-backed instances
884 2b3b6ddd Guido Trotter
      # WARNING: we currently take into account down instances as well as up
885 2b3b6ddd Guido Trotter
      # ones, considering that even if they're down someone might want to start
886 2b3b6ddd Guido Trotter
      # them even in the event of a node failure.
887 2b3b6ddd Guido Trotter
      for prinode, instances in nodeinfo['sinst-by-pnode'].iteritems():
888 2b3b6ddd Guido Trotter
        needed_mem = 0
889 2b3b6ddd Guido Trotter
        for instance in instances:
890 338e51e8 Iustin Pop
          bep = self.cfg.GetClusterInfo().FillBE(instance_cfg[instance])
891 c0f2b229 Iustin Pop
          if bep[constants.BE_AUTO_BALANCE]:
892 3924700f Iustin Pop
            needed_mem += bep[constants.BE_MEMORY]
893 2b3b6ddd Guido Trotter
        if nodeinfo['mfree'] < needed_mem:
894 2b3b6ddd Guido Trotter
          feedback_fn("  - ERROR: not enough memory on node %s to accomodate"
895 2b3b6ddd Guido Trotter
                      " failovers should node %s fail" % (node, prinode))
896 2b3b6ddd Guido Trotter
          bad = True
897 2b3b6ddd Guido Trotter
    return bad
898 2b3b6ddd Guido Trotter
899 a8083063 Iustin Pop
  def CheckPrereq(self):
900 a8083063 Iustin Pop
    """Check prerequisites.
901 a8083063 Iustin Pop

902 e54c4c5e Guido Trotter
    Transform the list of checks we're going to skip into a set and check that
903 e54c4c5e Guido Trotter
    all its members are valid.
904 a8083063 Iustin Pop

905 a8083063 Iustin Pop
    """
906 e54c4c5e Guido Trotter
    self.skip_set = frozenset(self.op.skip_checks)
907 e54c4c5e Guido Trotter
    if not constants.VERIFY_OPTIONAL_CHECKS.issuperset(self.skip_set):
908 e54c4c5e Guido Trotter
      raise errors.OpPrereqError("Invalid checks to be skipped specified")
909 a8083063 Iustin Pop
910 d8fff41c Guido Trotter
  def BuildHooksEnv(self):
911 d8fff41c Guido Trotter
    """Build hooks env.
912 d8fff41c Guido Trotter

913 d8fff41c Guido Trotter
    Cluster-Verify hooks just rone in the post phase and their failure makes
914 d8fff41c Guido Trotter
    the output be logged in the verify output and the verification to fail.
915 d8fff41c Guido Trotter

916 d8fff41c Guido Trotter
    """
917 d8fff41c Guido Trotter
    all_nodes = self.cfg.GetNodeList()
918 35e994e9 Iustin Pop
    env = {
919 35e994e9 Iustin Pop
      "CLUSTER_TAGS": " ".join(self.cfg.GetClusterInfo().GetTags())
920 35e994e9 Iustin Pop
      }
921 35e994e9 Iustin Pop
    for node in self.cfg.GetAllNodesInfo().values():
922 35e994e9 Iustin Pop
      env["NODE_TAGS_%s" % node.name] = " ".join(node.GetTags())
923 35e994e9 Iustin Pop
924 d8fff41c Guido Trotter
    return env, [], all_nodes
925 d8fff41c Guido Trotter
926 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
927 a8083063 Iustin Pop
    """Verify integrity of cluster, performing various test on nodes.
928 a8083063 Iustin Pop

929 a8083063 Iustin Pop
    """
930 a8083063 Iustin Pop
    bad = False
931 a8083063 Iustin Pop
    feedback_fn("* Verifying global settings")
932 8522ceeb Iustin Pop
    for msg in self.cfg.VerifyConfig():
933 8522ceeb Iustin Pop
      feedback_fn("  - ERROR: %s" % msg)
934 a8083063 Iustin Pop
935 a8083063 Iustin Pop
    vg_name = self.cfg.GetVGName()
936 e69d05fd Iustin Pop
    hypervisors = self.cfg.GetClusterInfo().enabled_hypervisors
937 a8083063 Iustin Pop
    nodelist = utils.NiceSort(self.cfg.GetNodeList())
938 9d4bfc96 Iustin Pop
    nodeinfo = [self.cfg.GetNodeInfo(nname) for nname in nodelist]
939 a8083063 Iustin Pop
    instancelist = utils.NiceSort(self.cfg.GetInstanceList())
940 6d2e83d5 Iustin Pop
    instanceinfo = dict((iname, self.cfg.GetInstanceInfo(iname))
941 6d2e83d5 Iustin Pop
                        for iname in instancelist)
942 93e4c50b Guido Trotter
    i_non_redundant = [] # Non redundant instances
943 3924700f Iustin Pop
    i_non_a_balanced = [] # Non auto-balanced instances
944 0a66c968 Iustin Pop
    n_offline = [] # List of offline nodes
945 22f0f71d Iustin Pop
    n_drained = [] # List of nodes being drained
946 a8083063 Iustin Pop
    node_volume = {}
947 a8083063 Iustin Pop
    node_instance = {}
948 9c9c7d30 Guido Trotter
    node_info = {}
949 26b6af5e Guido Trotter
    instance_cfg = {}
950 a8083063 Iustin Pop
951 a8083063 Iustin Pop
    # FIXME: verify OS list
952 a8083063 Iustin Pop
    # do local checksums
953 112f18a5 Iustin Pop
    master_files = [constants.CLUSTER_CONF_FILE]
954 112f18a5 Iustin Pop
955 112f18a5 Iustin Pop
    file_names = ssconf.SimpleStore().GetFileList()
956 cb91d46e Iustin Pop
    file_names.append(constants.SSL_CERT_FILE)
957 699777f2 Michael Hanselmann
    file_names.append(constants.RAPI_CERT_FILE)
958 112f18a5 Iustin Pop
    file_names.extend(master_files)
959 112f18a5 Iustin Pop
960 a8083063 Iustin Pop
    local_checksums = utils.FingerprintFiles(file_names)
961 a8083063 Iustin Pop
962 a8083063 Iustin Pop
    feedback_fn("* Gathering data (%d nodes)" % len(nodelist))
963 a8083063 Iustin Pop
    node_verify_param = {
964 25361b9a Iustin Pop
      constants.NV_FILELIST: file_names,
965 82e37788 Iustin Pop
      constants.NV_NODELIST: [node.name for node in nodeinfo
966 82e37788 Iustin Pop
                              if not node.offline],
967 25361b9a Iustin Pop
      constants.NV_HYPERVISOR: hypervisors,
968 25361b9a Iustin Pop
      constants.NV_NODENETTEST: [(node.name, node.primary_ip,
969 82e37788 Iustin Pop
                                  node.secondary_ip) for node in nodeinfo
970 82e37788 Iustin Pop
                                 if not node.offline],
971 25361b9a Iustin Pop
      constants.NV_INSTANCELIST: hypervisors,
972 25361b9a Iustin Pop
      constants.NV_VERSION: None,
973 25361b9a Iustin Pop
      constants.NV_HVINFO: self.cfg.GetHypervisorType(),
974 a8083063 Iustin Pop
      }
975 cc9e1230 Guido Trotter
    if vg_name is not None:
976 cc9e1230 Guido Trotter
      node_verify_param[constants.NV_VGLIST] = None
977 cc9e1230 Guido Trotter
      node_verify_param[constants.NV_LVLIST] = vg_name
978 cc9e1230 Guido Trotter
      node_verify_param[constants.NV_DRBDLIST] = None
979 72737a7f Iustin Pop
    all_nvinfo = self.rpc.call_node_verify(nodelist, node_verify_param,
980 72737a7f Iustin Pop
                                           self.cfg.GetClusterName())
981 a8083063 Iustin Pop
982 3924700f Iustin Pop
    cluster = self.cfg.GetClusterInfo()
983 112f18a5 Iustin Pop
    master_node = self.cfg.GetMasterNode()
984 6d2e83d5 Iustin Pop
    all_drbd_map = self.cfg.ComputeDRBDMap()
985 6d2e83d5 Iustin Pop
986 112f18a5 Iustin Pop
    for node_i in nodeinfo:
987 112f18a5 Iustin Pop
      node = node_i.name
988 25361b9a Iustin Pop
      nresult = all_nvinfo[node].data
989 25361b9a Iustin Pop
990 0a66c968 Iustin Pop
      if node_i.offline:
991 0a66c968 Iustin Pop
        feedback_fn("* Skipping offline node %s" % (node,))
992 0a66c968 Iustin Pop
        n_offline.append(node)
993 0a66c968 Iustin Pop
        continue
994 0a66c968 Iustin Pop
995 112f18a5 Iustin Pop
      if node == master_node:
996 25361b9a Iustin Pop
        ntype = "master"
997 112f18a5 Iustin Pop
      elif node_i.master_candidate:
998 25361b9a Iustin Pop
        ntype = "master candidate"
999 22f0f71d Iustin Pop
      elif node_i.drained:
1000 22f0f71d Iustin Pop
        ntype = "drained"
1001 22f0f71d Iustin Pop
        n_drained.append(node)
1002 112f18a5 Iustin Pop
      else:
1003 25361b9a Iustin Pop
        ntype = "regular"
1004 112f18a5 Iustin Pop
      feedback_fn("* Verifying node %s (%s)" % (node, ntype))
1005 25361b9a Iustin Pop
1006 25361b9a Iustin Pop
      if all_nvinfo[node].failed or not isinstance(nresult, dict):
1007 25361b9a Iustin Pop
        feedback_fn("  - ERROR: connection to %s failed" % (node,))
1008 25361b9a Iustin Pop
        bad = True
1009 25361b9a Iustin Pop
        continue
1010 25361b9a Iustin Pop
1011 6d2e83d5 Iustin Pop
      node_drbd = {}
1012 6d2e83d5 Iustin Pop
      for minor, instance in all_drbd_map[node].items():
1013 c614e5fb Iustin Pop
        if instance not in instanceinfo:
1014 c614e5fb Iustin Pop
          feedback_fn("  - ERROR: ghost instance '%s' in temporary DRBD map" %
1015 c614e5fb Iustin Pop
                      instance)
1016 c614e5fb Iustin Pop
          # ghost instance should not be running, but otherwise we
1017 c614e5fb Iustin Pop
          # don't give double warnings (both ghost instance and
1018 c614e5fb Iustin Pop
          # unallocated minor in use)
1019 c614e5fb Iustin Pop
          node_drbd[minor] = (instance, False)
1020 c614e5fb Iustin Pop
        else:
1021 c614e5fb Iustin Pop
          instance = instanceinfo[instance]
1022 c614e5fb Iustin Pop
          node_drbd[minor] = (instance.name, instance.admin_up)
1023 112f18a5 Iustin Pop
      result = self._VerifyNode(node_i, file_names, local_checksums,
1024 6d2e83d5 Iustin Pop
                                nresult, feedback_fn, master_files,
1025 cc9e1230 Guido Trotter
                                node_drbd, vg_name)
1026 a8083063 Iustin Pop
      bad = bad or result
1027 a8083063 Iustin Pop
1028 25361b9a Iustin Pop
      lvdata = nresult.get(constants.NV_LVLIST, "Missing LV data")
1029 cc9e1230 Guido Trotter
      if vg_name is None:
1030 cc9e1230 Guido Trotter
        node_volume[node] = {}
1031 cc9e1230 Guido Trotter
      elif isinstance(lvdata, basestring):
1032 b63ed789 Iustin Pop
        feedback_fn("  - ERROR: LVM problem on node %s: %s" %
1033 26f15862 Iustin Pop
                    (node, utils.SafeEncode(lvdata)))
1034 b63ed789 Iustin Pop
        bad = True
1035 b63ed789 Iustin Pop
        node_volume[node] = {}
1036 25361b9a Iustin Pop
      elif not isinstance(lvdata, dict):
1037 25361b9a Iustin Pop
        feedback_fn("  - ERROR: connection to %s failed (lvlist)" % (node,))
1038 a8083063 Iustin Pop
        bad = True
1039 a8083063 Iustin Pop
        continue
1040 b63ed789 Iustin Pop
      else:
1041 25361b9a Iustin Pop
        node_volume[node] = lvdata
1042 a8083063 Iustin Pop
1043 a8083063 Iustin Pop
      # node_instance
1044 25361b9a Iustin Pop
      idata = nresult.get(constants.NV_INSTANCELIST, None)
1045 25361b9a Iustin Pop
      if not isinstance(idata, list):
1046 25361b9a Iustin Pop
        feedback_fn("  - ERROR: connection to %s failed (instancelist)" %
1047 25361b9a Iustin Pop
                    (node,))
1048 a8083063 Iustin Pop
        bad = True
1049 a8083063 Iustin Pop
        continue
1050 a8083063 Iustin Pop
1051 25361b9a Iustin Pop
      node_instance[node] = idata
1052 a8083063 Iustin Pop
1053 9c9c7d30 Guido Trotter
      # node_info
1054 25361b9a Iustin Pop
      nodeinfo = nresult.get(constants.NV_HVINFO, None)
1055 9c9c7d30 Guido Trotter
      if not isinstance(nodeinfo, dict):
1056 25361b9a Iustin Pop
        feedback_fn("  - ERROR: connection to %s failed (hvinfo)" % (node,))
1057 9c9c7d30 Guido Trotter
        bad = True
1058 9c9c7d30 Guido Trotter
        continue
1059 9c9c7d30 Guido Trotter
1060 9c9c7d30 Guido Trotter
      try:
1061 9c9c7d30 Guido Trotter
        node_info[node] = {
1062 9c9c7d30 Guido Trotter
          "mfree": int(nodeinfo['memory_free']),
1063 93e4c50b Guido Trotter
          "pinst": [],
1064 93e4c50b Guido Trotter
          "sinst": [],
1065 36e7da50 Guido Trotter
          # dictionary holding all instances this node is secondary for,
1066 36e7da50 Guido Trotter
          # grouped by their primary node. Each key is a cluster node, and each
1067 36e7da50 Guido Trotter
          # value is a list of instances which have the key as primary and the
1068 36e7da50 Guido Trotter
          # current node as secondary.  this is handy to calculate N+1 memory
1069 36e7da50 Guido Trotter
          # availability if you can only failover from a primary to its
1070 36e7da50 Guido Trotter
          # secondary.
1071 36e7da50 Guido Trotter
          "sinst-by-pnode": {},
1072 9c9c7d30 Guido Trotter
        }
1073 cc9e1230 Guido Trotter
        # FIXME: devise a free space model for file based instances as well
1074 cc9e1230 Guido Trotter
        if vg_name is not None:
1075 9a198532 Iustin Pop
          if (constants.NV_VGLIST not in nresult or
1076 9a198532 Iustin Pop
              vg_name not in nresult[constants.NV_VGLIST]):
1077 9a198532 Iustin Pop
            feedback_fn("  - ERROR: node %s didn't return data for the"
1078 9a198532 Iustin Pop
                        " volume group '%s' - it is either missing or broken" %
1079 9a198532 Iustin Pop
                        (node, vg_name))
1080 9a198532 Iustin Pop
            bad = True
1081 9a198532 Iustin Pop
            continue
1082 cc9e1230 Guido Trotter
          node_info[node]["dfree"] = int(nresult[constants.NV_VGLIST][vg_name])
1083 9a198532 Iustin Pop
      except (ValueError, KeyError):
1084 9a198532 Iustin Pop
        feedback_fn("  - ERROR: invalid nodeinfo value returned"
1085 9a198532 Iustin Pop
                    " from node %s" % (node,))
1086 9c9c7d30 Guido Trotter
        bad = True
1087 9c9c7d30 Guido Trotter
        continue
1088 9c9c7d30 Guido Trotter
1089 a8083063 Iustin Pop
    node_vol_should = {}
1090 a8083063 Iustin Pop
1091 a8083063 Iustin Pop
    for instance in instancelist:
1092 a8083063 Iustin Pop
      feedback_fn("* Verifying instance %s" % instance)
1093 6d2e83d5 Iustin Pop
      inst_config = instanceinfo[instance]
1094 c5705f58 Guido Trotter
      result =  self._VerifyInstance(instance, inst_config, node_volume,
1095 0a66c968 Iustin Pop
                                     node_instance, feedback_fn, n_offline)
1096 c5705f58 Guido Trotter
      bad = bad or result
1097 832261fd Iustin Pop
      inst_nodes_offline = []
1098 a8083063 Iustin Pop
1099 a8083063 Iustin Pop
      inst_config.MapLVsByNode(node_vol_should)
1100 a8083063 Iustin Pop
1101 26b6af5e Guido Trotter
      instance_cfg[instance] = inst_config
1102 26b6af5e Guido Trotter
1103 93e4c50b Guido Trotter
      pnode = inst_config.primary_node
1104 93e4c50b Guido Trotter
      if pnode in node_info:
1105 93e4c50b Guido Trotter
        node_info[pnode]['pinst'].append(instance)
1106 0a66c968 Iustin Pop
      elif pnode not in n_offline:
1107 93e4c50b Guido Trotter
        feedback_fn("  - ERROR: instance %s, connection to primary node"
1108 93e4c50b Guido Trotter
                    " %s failed" % (instance, pnode))
1109 93e4c50b Guido Trotter
        bad = True
1110 93e4c50b Guido Trotter
1111 832261fd Iustin Pop
      if pnode in n_offline:
1112 832261fd Iustin Pop
        inst_nodes_offline.append(pnode)
1113 832261fd Iustin Pop
1114 93e4c50b Guido Trotter
      # If the instance is non-redundant we cannot survive losing its primary
1115 93e4c50b Guido Trotter
      # node, so we are not N+1 compliant. On the other hand we have no disk
1116 93e4c50b Guido Trotter
      # templates with more than one secondary so that situation is not well
1117 93e4c50b Guido Trotter
      # supported either.
1118 93e4c50b Guido Trotter
      # FIXME: does not support file-backed instances
1119 93e4c50b Guido Trotter
      if len(inst_config.secondary_nodes) == 0:
1120 93e4c50b Guido Trotter
        i_non_redundant.append(instance)
1121 93e4c50b Guido Trotter
      elif len(inst_config.secondary_nodes) > 1:
1122 93e4c50b Guido Trotter
        feedback_fn("  - WARNING: multiple secondaries for instance %s"
1123 93e4c50b Guido Trotter
                    % instance)
1124 93e4c50b Guido Trotter
1125 c0f2b229 Iustin Pop
      if not cluster.FillBE(inst_config)[constants.BE_AUTO_BALANCE]:
1126 3924700f Iustin Pop
        i_non_a_balanced.append(instance)
1127 3924700f Iustin Pop
1128 93e4c50b Guido Trotter
      for snode in inst_config.secondary_nodes:
1129 93e4c50b Guido Trotter
        if snode in node_info:
1130 93e4c50b Guido Trotter
          node_info[snode]['sinst'].append(instance)
1131 36e7da50 Guido Trotter
          if pnode not in node_info[snode]['sinst-by-pnode']:
1132 36e7da50 Guido Trotter
            node_info[snode]['sinst-by-pnode'][pnode] = []
1133 36e7da50 Guido Trotter
          node_info[snode]['sinst-by-pnode'][pnode].append(instance)
1134 0a66c968 Iustin Pop
        elif snode not in n_offline:
1135 93e4c50b Guido Trotter
          feedback_fn("  - ERROR: instance %s, connection to secondary node"
1136 93e4c50b Guido Trotter
                      " %s failed" % (instance, snode))
1137 832261fd Iustin Pop
          bad = True
1138 832261fd Iustin Pop
        if snode in n_offline:
1139 832261fd Iustin Pop
          inst_nodes_offline.append(snode)
1140 832261fd Iustin Pop
1141 832261fd Iustin Pop
      if inst_nodes_offline:
1142 832261fd Iustin Pop
        # warn that the instance lives on offline nodes, and set bad=True
1143 832261fd Iustin Pop
        feedback_fn("  - ERROR: instance lives on offline node(s) %s" %
1144 832261fd Iustin Pop
                    ", ".join(inst_nodes_offline))
1145 832261fd Iustin Pop
        bad = True
1146 93e4c50b Guido Trotter
1147 a8083063 Iustin Pop
    feedback_fn("* Verifying orphan volumes")
1148 a8083063 Iustin Pop
    result = self._VerifyOrphanVolumes(node_vol_should, node_volume,
1149 a8083063 Iustin Pop
                                       feedback_fn)
1150 a8083063 Iustin Pop
    bad = bad or result
1151 a8083063 Iustin Pop
1152 a8083063 Iustin Pop
    feedback_fn("* Verifying remaining instances")
1153 a8083063 Iustin Pop
    result = self._VerifyOrphanInstances(instancelist, node_instance,
1154 a8083063 Iustin Pop
                                         feedback_fn)
1155 a8083063 Iustin Pop
    bad = bad or result
1156 a8083063 Iustin Pop
1157 e54c4c5e Guido Trotter
    if constants.VERIFY_NPLUSONE_MEM not in self.skip_set:
1158 e54c4c5e Guido Trotter
      feedback_fn("* Verifying N+1 Memory redundancy")
1159 e54c4c5e Guido Trotter
      result = self._VerifyNPlusOneMemory(node_info, instance_cfg, feedback_fn)
1160 e54c4c5e Guido Trotter
      bad = bad or result
1161 2b3b6ddd Guido Trotter
1162 2b3b6ddd Guido Trotter
    feedback_fn("* Other Notes")
1163 2b3b6ddd Guido Trotter
    if i_non_redundant:
1164 2b3b6ddd Guido Trotter
      feedback_fn("  - NOTICE: %d non-redundant instance(s) found."
1165 2b3b6ddd Guido Trotter
                  % len(i_non_redundant))
1166 2b3b6ddd Guido Trotter
1167 3924700f Iustin Pop
    if i_non_a_balanced:
1168 3924700f Iustin Pop
      feedback_fn("  - NOTICE: %d non-auto-balanced instance(s) found."
1169 3924700f Iustin Pop
                  % len(i_non_a_balanced))
1170 3924700f Iustin Pop
1171 0a66c968 Iustin Pop
    if n_offline:
1172 0a66c968 Iustin Pop
      feedback_fn("  - NOTICE: %d offline node(s) found." % len(n_offline))
1173 0a66c968 Iustin Pop
1174 22f0f71d Iustin Pop
    if n_drained:
1175 22f0f71d Iustin Pop
      feedback_fn("  - NOTICE: %d drained node(s) found." % len(n_drained))
1176 22f0f71d Iustin Pop
1177 34290825 Michael Hanselmann
    return not bad
1178 a8083063 Iustin Pop
1179 d8fff41c Guido Trotter
  def HooksCallBack(self, phase, hooks_results, feedback_fn, lu_result):
1180 e4376078 Iustin Pop
    """Analize the post-hooks' result
1181 e4376078 Iustin Pop

1182 e4376078 Iustin Pop
    This method analyses the hook result, handles it, and sends some
1183 d8fff41c Guido Trotter
    nicely-formatted feedback back to the user.
1184 d8fff41c Guido Trotter

1185 e4376078 Iustin Pop
    @param phase: one of L{constants.HOOKS_PHASE_POST} or
1186 e4376078 Iustin Pop
        L{constants.HOOKS_PHASE_PRE}; it denotes the hooks phase
1187 e4376078 Iustin Pop
    @param hooks_results: the results of the multi-node hooks rpc call
1188 e4376078 Iustin Pop
    @param feedback_fn: function used send feedback back to the caller
1189 e4376078 Iustin Pop
    @param lu_result: previous Exec result
1190 e4376078 Iustin Pop
    @return: the new Exec result, based on the previous result
1191 e4376078 Iustin Pop
        and hook results
1192 d8fff41c Guido Trotter

1193 d8fff41c Guido Trotter
    """
1194 38206f3c Iustin Pop
    # We only really run POST phase hooks, and are only interested in
1195 38206f3c Iustin Pop
    # their results
1196 d8fff41c Guido Trotter
    if phase == constants.HOOKS_PHASE_POST:
1197 d8fff41c Guido Trotter
      # Used to change hooks' output to proper indentation
1198 d8fff41c Guido Trotter
      indent_re = re.compile('^', re.M)
1199 d8fff41c Guido Trotter
      feedback_fn("* Hooks Results")
1200 d8fff41c Guido Trotter
      if not hooks_results:
1201 d8fff41c Guido Trotter
        feedback_fn("  - ERROR: general communication failure")
1202 d8fff41c Guido Trotter
        lu_result = 1
1203 d8fff41c Guido Trotter
      else:
1204 d8fff41c Guido Trotter
        for node_name in hooks_results:
1205 d8fff41c Guido Trotter
          show_node_header = True
1206 d8fff41c Guido Trotter
          res = hooks_results[node_name]
1207 25361b9a Iustin Pop
          if res.failed or res.data is False or not isinstance(res.data, list):
1208 0a66c968 Iustin Pop
            if res.offline:
1209 0a66c968 Iustin Pop
              # no need to warn or set fail return value
1210 0a66c968 Iustin Pop
              continue
1211 25361b9a Iustin Pop
            feedback_fn("    Communication failure in hooks execution")
1212 d8fff41c Guido Trotter
            lu_result = 1
1213 d8fff41c Guido Trotter
            continue
1214 25361b9a Iustin Pop
          for script, hkr, output in res.data:
1215 d8fff41c Guido Trotter
            if hkr == constants.HKR_FAIL:
1216 d8fff41c Guido Trotter
              # The node header is only shown once, if there are
1217 d8fff41c Guido Trotter
              # failing hooks on that node
1218 d8fff41c Guido Trotter
              if show_node_header:
1219 d8fff41c Guido Trotter
                feedback_fn("  Node %s:" % node_name)
1220 d8fff41c Guido Trotter
                show_node_header = False
1221 d8fff41c Guido Trotter
              feedback_fn("    ERROR: Script %s failed, output:" % script)
1222 d8fff41c Guido Trotter
              output = indent_re.sub('      ', output)
1223 d8fff41c Guido Trotter
              feedback_fn("%s" % output)
1224 d8fff41c Guido Trotter
              lu_result = 1
1225 d8fff41c Guido Trotter
1226 d8fff41c Guido Trotter
      return lu_result
1227 d8fff41c Guido Trotter
1228 a8083063 Iustin Pop
1229 2c95a8d4 Iustin Pop
class LUVerifyDisks(NoHooksLU):
1230 2c95a8d4 Iustin Pop
  """Verifies the cluster disks status.
1231 2c95a8d4 Iustin Pop

1232 2c95a8d4 Iustin Pop
  """
1233 2c95a8d4 Iustin Pop
  _OP_REQP = []
1234 d4b9d97f Guido Trotter
  REQ_BGL = False
1235 d4b9d97f Guido Trotter
1236 d4b9d97f Guido Trotter
  def ExpandNames(self):
1237 d4b9d97f Guido Trotter
    self.needed_locks = {
1238 d4b9d97f Guido Trotter
      locking.LEVEL_NODE: locking.ALL_SET,
1239 d4b9d97f Guido Trotter
      locking.LEVEL_INSTANCE: locking.ALL_SET,
1240 d4b9d97f Guido Trotter
    }
1241 d4b9d97f Guido Trotter
    self.share_locks = dict(((i, 1) for i in locking.LEVELS))
1242 2c95a8d4 Iustin Pop
1243 2c95a8d4 Iustin Pop
  def CheckPrereq(self):
1244 2c95a8d4 Iustin Pop
    """Check prerequisites.
1245 2c95a8d4 Iustin Pop

1246 2c95a8d4 Iustin Pop
    This has no prerequisites.
1247 2c95a8d4 Iustin Pop

1248 2c95a8d4 Iustin Pop
    """
1249 2c95a8d4 Iustin Pop
    pass
1250 2c95a8d4 Iustin Pop
1251 2c95a8d4 Iustin Pop
  def Exec(self, feedback_fn):
1252 2c95a8d4 Iustin Pop
    """Verify integrity of cluster disks.
1253 2c95a8d4 Iustin Pop

1254 2c95a8d4 Iustin Pop
    """
1255 b63ed789 Iustin Pop
    result = res_nodes, res_nlvm, res_instances, res_missing = [], {}, [], {}
1256 2c95a8d4 Iustin Pop
1257 2c95a8d4 Iustin Pop
    vg_name = self.cfg.GetVGName()
1258 2c95a8d4 Iustin Pop
    nodes = utils.NiceSort(self.cfg.GetNodeList())
1259 2c95a8d4 Iustin Pop
    instances = [self.cfg.GetInstanceInfo(name)
1260 2c95a8d4 Iustin Pop
                 for name in self.cfg.GetInstanceList()]
1261 2c95a8d4 Iustin Pop
1262 2c95a8d4 Iustin Pop
    nv_dict = {}
1263 2c95a8d4 Iustin Pop
    for inst in instances:
1264 2c95a8d4 Iustin Pop
      inst_lvs = {}
1265 0d68c45d Iustin Pop
      if (not inst.admin_up or
1266 2c95a8d4 Iustin Pop
          inst.disk_template not in constants.DTS_NET_MIRROR):
1267 2c95a8d4 Iustin Pop
        continue
1268 2c95a8d4 Iustin Pop
      inst.MapLVsByNode(inst_lvs)
1269 2c95a8d4 Iustin Pop
      # transform { iname: {node: [vol,],},} to {(node, vol): iname}
1270 2c95a8d4 Iustin Pop
      for node, vol_list in inst_lvs.iteritems():
1271 2c95a8d4 Iustin Pop
        for vol in vol_list:
1272 2c95a8d4 Iustin Pop
          nv_dict[(node, vol)] = inst
1273 2c95a8d4 Iustin Pop
1274 2c95a8d4 Iustin Pop
    if not nv_dict:
1275 2c95a8d4 Iustin Pop
      return result
1276 2c95a8d4 Iustin Pop
1277 72737a7f Iustin Pop
    node_lvs = self.rpc.call_volume_list(nodes, vg_name)
1278 2c95a8d4 Iustin Pop
1279 2c95a8d4 Iustin Pop
    to_act = set()
1280 2c95a8d4 Iustin Pop
    for node in nodes:
1281 2c95a8d4 Iustin Pop
      # node_volume
1282 2c95a8d4 Iustin Pop
      lvs = node_lvs[node]
1283 781de953 Iustin Pop
      if lvs.failed:
1284 0a66c968 Iustin Pop
        if not lvs.offline:
1285 0a66c968 Iustin Pop
          self.LogWarning("Connection to node %s failed: %s" %
1286 0a66c968 Iustin Pop
                          (node, lvs.data))
1287 781de953 Iustin Pop
        continue
1288 781de953 Iustin Pop
      lvs = lvs.data
1289 b63ed789 Iustin Pop
      if isinstance(lvs, basestring):
1290 9a4f63d1 Iustin Pop
        logging.warning("Error enumerating LVs on node %s: %s", node, lvs)
1291 b63ed789 Iustin Pop
        res_nlvm[node] = lvs
1292 ea9ddc07 Iustin Pop
        continue
1293 b63ed789 Iustin Pop
      elif not isinstance(lvs, dict):
1294 9a4f63d1 Iustin Pop
        logging.warning("Connection to node %s failed or invalid data"
1295 9a4f63d1 Iustin Pop
                        " returned", node)
1296 2c95a8d4 Iustin Pop
        res_nodes.append(node)
1297 2c95a8d4 Iustin Pop
        continue
1298 2c95a8d4 Iustin Pop
1299 2c95a8d4 Iustin Pop
      for lv_name, (_, lv_inactive, lv_online) in lvs.iteritems():
1300 b63ed789 Iustin Pop
        inst = nv_dict.pop((node, lv_name), None)
1301 b63ed789 Iustin Pop
        if (not lv_online and inst is not None
1302 b63ed789 Iustin Pop
            and inst.name not in res_instances):
1303 b08d5a87 Iustin Pop
          res_instances.append(inst.name)
1304 2c95a8d4 Iustin Pop
1305 b63ed789 Iustin Pop
    # any leftover items in nv_dict are missing LVs, let's arrange the
1306 b63ed789 Iustin Pop
    # data better
1307 b63ed789 Iustin Pop
    for key, inst in nv_dict.iteritems():
1308 b63ed789 Iustin Pop
      if inst.name not in res_missing:
1309 b63ed789 Iustin Pop
        res_missing[inst.name] = []
1310 b63ed789 Iustin Pop
      res_missing[inst.name].append(key)
1311 b63ed789 Iustin Pop
1312 2c95a8d4 Iustin Pop
    return result
1313 2c95a8d4 Iustin Pop
1314 2c95a8d4 Iustin Pop
1315 07bd8a51 Iustin Pop
class LURenameCluster(LogicalUnit):
1316 07bd8a51 Iustin Pop
  """Rename the cluster.
1317 07bd8a51 Iustin Pop

1318 07bd8a51 Iustin Pop
  """
1319 07bd8a51 Iustin Pop
  HPATH = "cluster-rename"
1320 07bd8a51 Iustin Pop
  HTYPE = constants.HTYPE_CLUSTER
1321 07bd8a51 Iustin Pop
  _OP_REQP = ["name"]
1322 07bd8a51 Iustin Pop
1323 07bd8a51 Iustin Pop
  def BuildHooksEnv(self):
1324 07bd8a51 Iustin Pop
    """Build hooks env.
1325 07bd8a51 Iustin Pop

1326 07bd8a51 Iustin Pop
    """
1327 07bd8a51 Iustin Pop
    env = {
1328 d6a02168 Michael Hanselmann
      "OP_TARGET": self.cfg.GetClusterName(),
1329 07bd8a51 Iustin Pop
      "NEW_NAME": self.op.name,
1330 07bd8a51 Iustin Pop
      }
1331 d6a02168 Michael Hanselmann
    mn = self.cfg.GetMasterNode()
1332 07bd8a51 Iustin Pop
    return env, [mn], [mn]
1333 07bd8a51 Iustin Pop
1334 07bd8a51 Iustin Pop
  def CheckPrereq(self):
1335 07bd8a51 Iustin Pop
    """Verify that the passed name is a valid one.
1336 07bd8a51 Iustin Pop

1337 07bd8a51 Iustin Pop
    """
1338 89e1fc26 Iustin Pop
    hostname = utils.HostInfo(self.op.name)
1339 07bd8a51 Iustin Pop
1340 bcf043c9 Iustin Pop
    new_name = hostname.name
1341 bcf043c9 Iustin Pop
    self.ip = new_ip = hostname.ip
1342 d6a02168 Michael Hanselmann
    old_name = self.cfg.GetClusterName()
1343 d6a02168 Michael Hanselmann
    old_ip = self.cfg.GetMasterIP()
1344 07bd8a51 Iustin Pop
    if new_name == old_name and new_ip == old_ip:
1345 07bd8a51 Iustin Pop
      raise errors.OpPrereqError("Neither the name nor the IP address of the"
1346 07bd8a51 Iustin Pop
                                 " cluster has changed")
1347 07bd8a51 Iustin Pop
    if new_ip != old_ip:
1348 937f983d Guido Trotter
      if utils.TcpPing(new_ip, constants.DEFAULT_NODED_PORT):
1349 07bd8a51 Iustin Pop
        raise errors.OpPrereqError("The given cluster IP address (%s) is"
1350 07bd8a51 Iustin Pop
                                   " reachable on the network. Aborting." %
1351 07bd8a51 Iustin Pop
                                   new_ip)
1352 07bd8a51 Iustin Pop
1353 07bd8a51 Iustin Pop
    self.op.name = new_name
1354 07bd8a51 Iustin Pop
1355 07bd8a51 Iustin Pop
  def Exec(self, feedback_fn):
1356 07bd8a51 Iustin Pop
    """Rename the cluster.
1357 07bd8a51 Iustin Pop

1358 07bd8a51 Iustin Pop
    """
1359 07bd8a51 Iustin Pop
    clustername = self.op.name
1360 07bd8a51 Iustin Pop
    ip = self.ip
1361 07bd8a51 Iustin Pop
1362 07bd8a51 Iustin Pop
    # shutdown the master IP
1363 d6a02168 Michael Hanselmann
    master = self.cfg.GetMasterNode()
1364 781de953 Iustin Pop
    result = self.rpc.call_node_stop_master(master, False)
1365 781de953 Iustin Pop
    if result.failed or not result.data:
1366 07bd8a51 Iustin Pop
      raise errors.OpExecError("Could not disable the master role")
1367 07bd8a51 Iustin Pop
1368 07bd8a51 Iustin Pop
    try:
1369 55cf7d83 Iustin Pop
      cluster = self.cfg.GetClusterInfo()
1370 55cf7d83 Iustin Pop
      cluster.cluster_name = clustername
1371 55cf7d83 Iustin Pop
      cluster.master_ip = ip
1372 55cf7d83 Iustin Pop
      self.cfg.Update(cluster)
1373 ec85e3d5 Iustin Pop
1374 ec85e3d5 Iustin Pop
      # update the known hosts file
1375 ec85e3d5 Iustin Pop
      ssh.WriteKnownHostsFile(self.cfg, constants.SSH_KNOWN_HOSTS_FILE)
1376 ec85e3d5 Iustin Pop
      node_list = self.cfg.GetNodeList()
1377 ec85e3d5 Iustin Pop
      try:
1378 ec85e3d5 Iustin Pop
        node_list.remove(master)
1379 ec85e3d5 Iustin Pop
      except ValueError:
1380 ec85e3d5 Iustin Pop
        pass
1381 ec85e3d5 Iustin Pop
      result = self.rpc.call_upload_file(node_list,
1382 ec85e3d5 Iustin Pop
                                         constants.SSH_KNOWN_HOSTS_FILE)
1383 ec85e3d5 Iustin Pop
      for to_node, to_result in result.iteritems():
1384 1b54fc6c Guido Trotter
         msg = to_result.RemoteFailMsg()
1385 1b54fc6c Guido Trotter
         if msg:
1386 1b54fc6c Guido Trotter
           msg = ("Copy of file %s to node %s failed: %s" %
1387 1b54fc6c Guido Trotter
                   (constants.SSH_KNOWN_HOSTS_FILE, to_node, msg))
1388 1b54fc6c Guido Trotter
           self.proc.LogWarning(msg)
1389 ec85e3d5 Iustin Pop
1390 07bd8a51 Iustin Pop
    finally:
1391 781de953 Iustin Pop
      result = self.rpc.call_node_start_master(master, False)
1392 781de953 Iustin Pop
      if result.failed or not result.data:
1393 86d9d3bb Iustin Pop
        self.LogWarning("Could not re-enable the master role on"
1394 86d9d3bb Iustin Pop
                        " the master, please restart manually.")
1395 07bd8a51 Iustin Pop
1396 07bd8a51 Iustin Pop
1397 8084f9f6 Manuel Franceschini
def _RecursiveCheckIfLVMBased(disk):
1398 8084f9f6 Manuel Franceschini
  """Check if the given disk or its children are lvm-based.
1399 8084f9f6 Manuel Franceschini

1400 e4376078 Iustin Pop
  @type disk: L{objects.Disk}
1401 e4376078 Iustin Pop
  @param disk: the disk to check
1402 e4376078 Iustin Pop
  @rtype: booleean
1403 e4376078 Iustin Pop
  @return: boolean indicating whether a LD_LV dev_type was found or not
1404 8084f9f6 Manuel Franceschini

1405 8084f9f6 Manuel Franceschini
  """
1406 8084f9f6 Manuel Franceschini
  if disk.children:
1407 8084f9f6 Manuel Franceschini
    for chdisk in disk.children:
1408 8084f9f6 Manuel Franceschini
      if _RecursiveCheckIfLVMBased(chdisk):
1409 8084f9f6 Manuel Franceschini
        return True
1410 8084f9f6 Manuel Franceschini
  return disk.dev_type == constants.LD_LV
1411 8084f9f6 Manuel Franceschini
1412 8084f9f6 Manuel Franceschini
1413 8084f9f6 Manuel Franceschini
class LUSetClusterParams(LogicalUnit):
1414 8084f9f6 Manuel Franceschini
  """Change the parameters of the cluster.
1415 8084f9f6 Manuel Franceschini

1416 8084f9f6 Manuel Franceschini
  """
1417 8084f9f6 Manuel Franceschini
  HPATH = "cluster-modify"
1418 8084f9f6 Manuel Franceschini
  HTYPE = constants.HTYPE_CLUSTER
1419 8084f9f6 Manuel Franceschini
  _OP_REQP = []
1420 c53279cf Guido Trotter
  REQ_BGL = False
1421 c53279cf Guido Trotter
1422 3994f455 Iustin Pop
  def CheckArguments(self):
1423 4b7735f9 Iustin Pop
    """Check parameters
1424 4b7735f9 Iustin Pop

1425 4b7735f9 Iustin Pop
    """
1426 4b7735f9 Iustin Pop
    if not hasattr(self.op, "candidate_pool_size"):
1427 4b7735f9 Iustin Pop
      self.op.candidate_pool_size = None
1428 4b7735f9 Iustin Pop
    if self.op.candidate_pool_size is not None:
1429 4b7735f9 Iustin Pop
      try:
1430 4b7735f9 Iustin Pop
        self.op.candidate_pool_size = int(self.op.candidate_pool_size)
1431 3994f455 Iustin Pop
      except (ValueError, TypeError), err:
1432 4b7735f9 Iustin Pop
        raise errors.OpPrereqError("Invalid candidate_pool_size value: %s" %
1433 4b7735f9 Iustin Pop
                                   str(err))
1434 4b7735f9 Iustin Pop
      if self.op.candidate_pool_size < 1:
1435 4b7735f9 Iustin Pop
        raise errors.OpPrereqError("At least one master candidate needed")
1436 4b7735f9 Iustin Pop
1437 c53279cf Guido Trotter
  def ExpandNames(self):
1438 c53279cf Guido Trotter
    # FIXME: in the future maybe other cluster params won't require checking on
1439 c53279cf Guido Trotter
    # all nodes to be modified.
1440 c53279cf Guido Trotter
    self.needed_locks = {
1441 c53279cf Guido Trotter
      locking.LEVEL_NODE: locking.ALL_SET,
1442 c53279cf Guido Trotter
    }
1443 c53279cf Guido Trotter
    self.share_locks[locking.LEVEL_NODE] = 1
1444 8084f9f6 Manuel Franceschini
1445 8084f9f6 Manuel Franceschini
  def BuildHooksEnv(self):
1446 8084f9f6 Manuel Franceschini
    """Build hooks env.
1447 8084f9f6 Manuel Franceschini

1448 8084f9f6 Manuel Franceschini
    """
1449 8084f9f6 Manuel Franceschini
    env = {
1450 d6a02168 Michael Hanselmann
      "OP_TARGET": self.cfg.GetClusterName(),
1451 8084f9f6 Manuel Franceschini
      "NEW_VG_NAME": self.op.vg_name,
1452 8084f9f6 Manuel Franceschini
      }
1453 d6a02168 Michael Hanselmann
    mn = self.cfg.GetMasterNode()
1454 8084f9f6 Manuel Franceschini
    return env, [mn], [mn]
1455 8084f9f6 Manuel Franceschini
1456 8084f9f6 Manuel Franceschini
  def CheckPrereq(self):
1457 8084f9f6 Manuel Franceschini
    """Check prerequisites.
1458 8084f9f6 Manuel Franceschini

1459 8084f9f6 Manuel Franceschini
    This checks whether the given params don't conflict and
1460 5f83e263 Iustin Pop
    if the given volume group is valid.
1461 8084f9f6 Manuel Franceschini

1462 8084f9f6 Manuel Franceschini
    """
1463 779c15bb Iustin Pop
    if self.op.vg_name is not None and not self.op.vg_name:
1464 c53279cf Guido Trotter
      instances = self.cfg.GetAllInstancesInfo().values()
1465 8084f9f6 Manuel Franceschini
      for inst in instances:
1466 8084f9f6 Manuel Franceschini
        for disk in inst.disks:
1467 8084f9f6 Manuel Franceschini
          if _RecursiveCheckIfLVMBased(disk):
1468 8084f9f6 Manuel Franceschini
            raise errors.OpPrereqError("Cannot disable lvm storage while"
1469 8084f9f6 Manuel Franceschini
                                       " lvm-based instances exist")
1470 8084f9f6 Manuel Franceschini
1471 779c15bb Iustin Pop
    node_list = self.acquired_locks[locking.LEVEL_NODE]
1472 779c15bb Iustin Pop
1473 8084f9f6 Manuel Franceschini
    # if vg_name not None, checks given volume group on all nodes
1474 8084f9f6 Manuel Franceschini
    if self.op.vg_name:
1475 72737a7f Iustin Pop
      vglist = self.rpc.call_vg_list(node_list)
1476 8084f9f6 Manuel Franceschini
      for node in node_list:
1477 781de953 Iustin Pop
        if vglist[node].failed:
1478 781de953 Iustin Pop
          # ignoring down node
1479 781de953 Iustin Pop
          self.LogWarning("Node %s unreachable/error, ignoring" % node)
1480 781de953 Iustin Pop
          continue
1481 781de953 Iustin Pop
        vgstatus = utils.CheckVolumeGroupSize(vglist[node].data,
1482 781de953 Iustin Pop
                                              self.op.vg_name,
1483 8d1a2a64 Michael Hanselmann
                                              constants.MIN_VG_SIZE)
1484 8084f9f6 Manuel Franceschini
        if vgstatus:
1485 8084f9f6 Manuel Franceschini
          raise errors.OpPrereqError("Error on node '%s': %s" %
1486 8084f9f6 Manuel Franceschini
                                     (node, vgstatus))
1487 8084f9f6 Manuel Franceschini
1488 779c15bb Iustin Pop
    self.cluster = cluster = self.cfg.GetClusterInfo()
1489 5af3da74 Guido Trotter
    # validate params changes
1490 779c15bb Iustin Pop
    if self.op.beparams:
1491 a5728081 Guido Trotter
      utils.ForceDictType(self.op.beparams, constants.BES_PARAMETER_TYPES)
1492 abe609b2 Guido Trotter
      self.new_beparams = objects.FillDict(
1493 4ef7f423 Guido Trotter
        cluster.beparams[constants.PP_DEFAULT], self.op.beparams)
1494 779c15bb Iustin Pop
1495 5af3da74 Guido Trotter
    if self.op.nicparams:
1496 5af3da74 Guido Trotter
      utils.ForceDictType(self.op.nicparams, constants.NICS_PARAMETER_TYPES)
1497 5af3da74 Guido Trotter
      self.new_nicparams = objects.FillDict(
1498 5af3da74 Guido Trotter
        cluster.nicparams[constants.PP_DEFAULT], self.op.nicparams)
1499 5af3da74 Guido Trotter
      objects.NIC.CheckParameterSyntax(self.new_nicparams)
1500 5af3da74 Guido Trotter
1501 779c15bb Iustin Pop
    # hypervisor list/parameters
1502 abe609b2 Guido Trotter
    self.new_hvparams = objects.FillDict(cluster.hvparams, {})
1503 779c15bb Iustin Pop
    if self.op.hvparams:
1504 779c15bb Iustin Pop
      if not isinstance(self.op.hvparams, dict):
1505 779c15bb Iustin Pop
        raise errors.OpPrereqError("Invalid 'hvparams' parameter on input")
1506 779c15bb Iustin Pop
      for hv_name, hv_dict in self.op.hvparams.items():
1507 779c15bb Iustin Pop
        if hv_name not in self.new_hvparams:
1508 779c15bb Iustin Pop
          self.new_hvparams[hv_name] = hv_dict
1509 779c15bb Iustin Pop
        else:
1510 779c15bb Iustin Pop
          self.new_hvparams[hv_name].update(hv_dict)
1511 779c15bb Iustin Pop
1512 779c15bb Iustin Pop
    if self.op.enabled_hypervisors is not None:
1513 779c15bb Iustin Pop
      self.hv_list = self.op.enabled_hypervisors
1514 779c15bb Iustin Pop
    else:
1515 779c15bb Iustin Pop
      self.hv_list = cluster.enabled_hypervisors
1516 779c15bb Iustin Pop
1517 779c15bb Iustin Pop
    if self.op.hvparams or self.op.enabled_hypervisors is not None:
1518 779c15bb Iustin Pop
      # either the enabled list has changed, or the parameters have, validate
1519 779c15bb Iustin Pop
      for hv_name, hv_params in self.new_hvparams.items():
1520 779c15bb Iustin Pop
        if ((self.op.hvparams and hv_name in self.op.hvparams) or
1521 779c15bb Iustin Pop
            (self.op.enabled_hypervisors and
1522 779c15bb Iustin Pop
             hv_name in self.op.enabled_hypervisors)):
1523 779c15bb Iustin Pop
          # either this is a new hypervisor, or its parameters have changed
1524 779c15bb Iustin Pop
          hv_class = hypervisor.GetHypervisor(hv_name)
1525 a5728081 Guido Trotter
          utils.ForceDictType(hv_params, constants.HVS_PARAMETER_TYPES)
1526 779c15bb Iustin Pop
          hv_class.CheckParameterSyntax(hv_params)
1527 779c15bb Iustin Pop
          _CheckHVParams(self, node_list, hv_name, hv_params)
1528 779c15bb Iustin Pop
1529 8084f9f6 Manuel Franceschini
  def Exec(self, feedback_fn):
1530 8084f9f6 Manuel Franceschini
    """Change the parameters of the cluster.
1531 8084f9f6 Manuel Franceschini

1532 8084f9f6 Manuel Franceschini
    """
1533 779c15bb Iustin Pop
    if self.op.vg_name is not None:
1534 b2482333 Guido Trotter
      new_volume = self.op.vg_name
1535 b2482333 Guido Trotter
      if not new_volume:
1536 b2482333 Guido Trotter
        new_volume = None
1537 b2482333 Guido Trotter
      if new_volume != self.cfg.GetVGName():
1538 b2482333 Guido Trotter
        self.cfg.SetVGName(new_volume)
1539 779c15bb Iustin Pop
      else:
1540 779c15bb Iustin Pop
        feedback_fn("Cluster LVM configuration already in desired"
1541 779c15bb Iustin Pop
                    " state, not changing")
1542 779c15bb Iustin Pop
    if self.op.hvparams:
1543 779c15bb Iustin Pop
      self.cluster.hvparams = self.new_hvparams
1544 779c15bb Iustin Pop
    if self.op.enabled_hypervisors is not None:
1545 779c15bb Iustin Pop
      self.cluster.enabled_hypervisors = self.op.enabled_hypervisors
1546 779c15bb Iustin Pop
    if self.op.beparams:
1547 4ef7f423 Guido Trotter
      self.cluster.beparams[constants.PP_DEFAULT] = self.new_beparams
1548 5af3da74 Guido Trotter
    if self.op.nicparams:
1549 5af3da74 Guido Trotter
      self.cluster.nicparams[constants.PP_DEFAULT] = self.new_nicparams
1550 5af3da74 Guido Trotter
1551 4b7735f9 Iustin Pop
    if self.op.candidate_pool_size is not None:
1552 4b7735f9 Iustin Pop
      self.cluster.candidate_pool_size = self.op.candidate_pool_size
1553 4b7735f9 Iustin Pop
1554 779c15bb Iustin Pop
    self.cfg.Update(self.cluster)
1555 8084f9f6 Manuel Franceschini
1556 4b7735f9 Iustin Pop
    # we want to update nodes after the cluster so that if any errors
1557 4b7735f9 Iustin Pop
    # happen, we have recorded and saved the cluster info
1558 4b7735f9 Iustin Pop
    if self.op.candidate_pool_size is not None:
1559 ec0292f1 Iustin Pop
      _AdjustCandidatePool(self)
1560 4b7735f9 Iustin Pop
1561 8084f9f6 Manuel Franceschini
1562 28eddce5 Guido Trotter
def _RedistributeAncillaryFiles(lu, additional_nodes=None):
1563 28eddce5 Guido Trotter
  """Distribute additional files which are part of the cluster configuration.
1564 28eddce5 Guido Trotter

1565 28eddce5 Guido Trotter
  ConfigWriter takes care of distributing the config and ssconf files, but
1566 28eddce5 Guido Trotter
  there are more files which should be distributed to all nodes. This function
1567 28eddce5 Guido Trotter
  makes sure those are copied.
1568 28eddce5 Guido Trotter

1569 28eddce5 Guido Trotter
  @param lu: calling logical unit
1570 28eddce5 Guido Trotter
  @param additional_nodes: list of nodes not in the config to distribute to
1571 28eddce5 Guido Trotter

1572 28eddce5 Guido Trotter
  """
1573 28eddce5 Guido Trotter
  # 1. Gather target nodes
1574 28eddce5 Guido Trotter
  myself = lu.cfg.GetNodeInfo(lu.cfg.GetMasterNode())
1575 28eddce5 Guido Trotter
  dist_nodes = lu.cfg.GetNodeList()
1576 28eddce5 Guido Trotter
  if additional_nodes is not None:
1577 28eddce5 Guido Trotter
    dist_nodes.extend(additional_nodes)
1578 28eddce5 Guido Trotter
  if myself.name in dist_nodes:
1579 28eddce5 Guido Trotter
    dist_nodes.remove(myself.name)
1580 28eddce5 Guido Trotter
  # 2. Gather files to distribute
1581 28eddce5 Guido Trotter
  dist_files = set([constants.ETC_HOSTS,
1582 28eddce5 Guido Trotter
                    constants.SSH_KNOWN_HOSTS_FILE,
1583 28eddce5 Guido Trotter
                    constants.RAPI_CERT_FILE,
1584 28eddce5 Guido Trotter
                    constants.RAPI_USERS_FILE,
1585 28eddce5 Guido Trotter
                   ])
1586 e1b8653f Guido Trotter
1587 e1b8653f Guido Trotter
  enabled_hypervisors = lu.cfg.GetClusterInfo().enabled_hypervisors
1588 e1b8653f Guido Trotter
  for hv_name in enabled_hypervisors:
1589 e1b8653f Guido Trotter
    hv_class = hypervisor.GetHypervisor(hv_name)
1590 e1b8653f Guido Trotter
    dist_files.update(hv_class.GetAncillaryFiles())
1591 e1b8653f Guido Trotter
1592 28eddce5 Guido Trotter
  # 3. Perform the files upload
1593 28eddce5 Guido Trotter
  for fname in dist_files:
1594 28eddce5 Guido Trotter
    if os.path.exists(fname):
1595 28eddce5 Guido Trotter
      result = lu.rpc.call_upload_file(dist_nodes, fname)
1596 28eddce5 Guido Trotter
      for to_node, to_result in result.items():
1597 1b54fc6c Guido Trotter
         msg = to_result.RemoteFailMsg()
1598 1b54fc6c Guido Trotter
         if msg:
1599 1b54fc6c Guido Trotter
           msg = ("Copy of file %s to node %s failed: %s" %
1600 1b54fc6c Guido Trotter
                   (fname, to_node, msg))
1601 1b54fc6c Guido Trotter
           lu.proc.LogWarning(msg)
1602 28eddce5 Guido Trotter
1603 28eddce5 Guido Trotter
1604 afee0879 Iustin Pop
class LURedistributeConfig(NoHooksLU):
1605 afee0879 Iustin Pop
  """Force the redistribution of cluster configuration.
1606 afee0879 Iustin Pop

1607 afee0879 Iustin Pop
  This is a very simple LU.
1608 afee0879 Iustin Pop

1609 afee0879 Iustin Pop
  """
1610 afee0879 Iustin Pop
  _OP_REQP = []
1611 afee0879 Iustin Pop
  REQ_BGL = False
1612 afee0879 Iustin Pop
1613 afee0879 Iustin Pop
  def ExpandNames(self):
1614 afee0879 Iustin Pop
    self.needed_locks = {
1615 afee0879 Iustin Pop
      locking.LEVEL_NODE: locking.ALL_SET,
1616 afee0879 Iustin Pop
    }
1617 afee0879 Iustin Pop
    self.share_locks[locking.LEVEL_NODE] = 1
1618 afee0879 Iustin Pop
1619 afee0879 Iustin Pop
  def CheckPrereq(self):
1620 afee0879 Iustin Pop
    """Check prerequisites.
1621 afee0879 Iustin Pop

1622 afee0879 Iustin Pop
    """
1623 afee0879 Iustin Pop
1624 afee0879 Iustin Pop
  def Exec(self, feedback_fn):
1625 afee0879 Iustin Pop
    """Redistribute the configuration.
1626 afee0879 Iustin Pop

1627 afee0879 Iustin Pop
    """
1628 afee0879 Iustin Pop
    self.cfg.Update(self.cfg.GetClusterInfo())
1629 28eddce5 Guido Trotter
    _RedistributeAncillaryFiles(self)
1630 afee0879 Iustin Pop
1631 afee0879 Iustin Pop
1632 b9bddb6b Iustin Pop
def _WaitForSync(lu, instance, oneshot=False, unlock=False):
1633 a8083063 Iustin Pop
  """Sleep and poll for an instance's disk to sync.
1634 a8083063 Iustin Pop

1635 a8083063 Iustin Pop
  """
1636 a8083063 Iustin Pop
  if not instance.disks:
1637 a8083063 Iustin Pop
    return True
1638 a8083063 Iustin Pop
1639 a8083063 Iustin Pop
  if not oneshot:
1640 b9bddb6b Iustin Pop
    lu.proc.LogInfo("Waiting for instance %s to sync disks." % instance.name)
1641 a8083063 Iustin Pop
1642 a8083063 Iustin Pop
  node = instance.primary_node
1643 a8083063 Iustin Pop
1644 a8083063 Iustin Pop
  for dev in instance.disks:
1645 b9bddb6b Iustin Pop
    lu.cfg.SetDiskID(dev, node)
1646 a8083063 Iustin Pop
1647 a8083063 Iustin Pop
  retries = 0
1648 a8083063 Iustin Pop
  while True:
1649 a8083063 Iustin Pop
    max_time = 0
1650 a8083063 Iustin Pop
    done = True
1651 a8083063 Iustin Pop
    cumul_degraded = False
1652 72737a7f Iustin Pop
    rstats = lu.rpc.call_blockdev_getmirrorstatus(node, instance.disks)
1653 781de953 Iustin Pop
    if rstats.failed or not rstats.data:
1654 86d9d3bb Iustin Pop
      lu.LogWarning("Can't get any data from node %s", node)
1655 a8083063 Iustin Pop
      retries += 1
1656 a8083063 Iustin Pop
      if retries >= 10:
1657 3ecf6786 Iustin Pop
        raise errors.RemoteError("Can't contact node %s for mirror data,"
1658 3ecf6786 Iustin Pop
                                 " aborting." % node)
1659 a8083063 Iustin Pop
      time.sleep(6)
1660 a8083063 Iustin Pop
      continue
1661 781de953 Iustin Pop
    rstats = rstats.data
1662 a8083063 Iustin Pop
    retries = 0
1663 1492cca7 Iustin Pop
    for i, mstat in enumerate(rstats):
1664 a8083063 Iustin Pop
      if mstat is None:
1665 86d9d3bb Iustin Pop
        lu.LogWarning("Can't compute data for node %s/%s",
1666 86d9d3bb Iustin Pop
                           node, instance.disks[i].iv_name)
1667 a8083063 Iustin Pop
        continue
1668 0834c866 Iustin Pop
      # we ignore the ldisk parameter
1669 0834c866 Iustin Pop
      perc_done, est_time, is_degraded, _ = mstat
1670 a8083063 Iustin Pop
      cumul_degraded = cumul_degraded or (is_degraded and perc_done is None)
1671 a8083063 Iustin Pop
      if perc_done is not None:
1672 a8083063 Iustin Pop
        done = False
1673 a8083063 Iustin Pop
        if est_time is not None:
1674 a8083063 Iustin Pop
          rem_time = "%d estimated seconds remaining" % est_time
1675 a8083063 Iustin Pop
          max_time = est_time
1676 a8083063 Iustin Pop
        else:
1677 a8083063 Iustin Pop
          rem_time = "no time estimate"
1678 b9bddb6b Iustin Pop
        lu.proc.LogInfo("- device %s: %5.2f%% done, %s" %
1679 b9bddb6b Iustin Pop
                        (instance.disks[i].iv_name, perc_done, rem_time))
1680 a8083063 Iustin Pop
    if done or oneshot:
1681 a8083063 Iustin Pop
      break
1682 a8083063 Iustin Pop
1683 d4fa5c23 Iustin Pop
    time.sleep(min(60, max_time))
1684 a8083063 Iustin Pop
1685 a8083063 Iustin Pop
  if done:
1686 b9bddb6b Iustin Pop
    lu.proc.LogInfo("Instance %s's disks are in sync." % instance.name)
1687 a8083063 Iustin Pop
  return not cumul_degraded
1688 a8083063 Iustin Pop
1689 a8083063 Iustin Pop
1690 b9bddb6b Iustin Pop
def _CheckDiskConsistency(lu, dev, node, on_primary, ldisk=False):
1691 a8083063 Iustin Pop
  """Check that mirrors are not degraded.
1692 a8083063 Iustin Pop

1693 0834c866 Iustin Pop
  The ldisk parameter, if True, will change the test from the
1694 0834c866 Iustin Pop
  is_degraded attribute (which represents overall non-ok status for
1695 0834c866 Iustin Pop
  the device(s)) to the ldisk (representing the local storage status).
1696 0834c866 Iustin Pop

1697 a8083063 Iustin Pop
  """
1698 b9bddb6b Iustin Pop
  lu.cfg.SetDiskID(dev, node)
1699 0834c866 Iustin Pop
  if ldisk:
1700 0834c866 Iustin Pop
    idx = 6
1701 0834c866 Iustin Pop
  else:
1702 0834c866 Iustin Pop
    idx = 5
1703 a8083063 Iustin Pop
1704 a8083063 Iustin Pop
  result = True
1705 a8083063 Iustin Pop
  if on_primary or dev.AssembleOnSecondary():
1706 72737a7f Iustin Pop
    rstats = lu.rpc.call_blockdev_find(node, dev)
1707 23829f6f Iustin Pop
    msg = rstats.RemoteFailMsg()
1708 23829f6f Iustin Pop
    if msg:
1709 23829f6f Iustin Pop
      lu.LogWarning("Can't find disk on node %s: %s", node, msg)
1710 23829f6f Iustin Pop
      result = False
1711 23829f6f Iustin Pop
    elif not rstats.payload:
1712 23829f6f Iustin Pop
      lu.LogWarning("Can't find disk on node %s", node)
1713 a8083063 Iustin Pop
      result = False
1714 a8083063 Iustin Pop
    else:
1715 23829f6f Iustin Pop
      result = result and (not rstats.payload[idx])
1716 a8083063 Iustin Pop
  if dev.children:
1717 a8083063 Iustin Pop
    for child in dev.children:
1718 b9bddb6b Iustin Pop
      result = result and _CheckDiskConsistency(lu, child, node, on_primary)
1719 a8083063 Iustin Pop
1720 a8083063 Iustin Pop
  return result
1721 a8083063 Iustin Pop
1722 a8083063 Iustin Pop
1723 a8083063 Iustin Pop
class LUDiagnoseOS(NoHooksLU):
1724 a8083063 Iustin Pop
  """Logical unit for OS diagnose/query.
1725 a8083063 Iustin Pop

1726 a8083063 Iustin Pop
  """
1727 1f9430d6 Iustin Pop
  _OP_REQP = ["output_fields", "names"]
1728 6bf01bbb Guido Trotter
  REQ_BGL = False
1729 a2d2e1a7 Iustin Pop
  _FIELDS_STATIC = utils.FieldSet()
1730 a2d2e1a7 Iustin Pop
  _FIELDS_DYNAMIC = utils.FieldSet("name", "valid", "node_status")
1731 a8083063 Iustin Pop
1732 6bf01bbb Guido Trotter
  def ExpandNames(self):
1733 1f9430d6 Iustin Pop
    if self.op.names:
1734 1f9430d6 Iustin Pop
      raise errors.OpPrereqError("Selective OS query not supported")
1735 1f9430d6 Iustin Pop
1736 31bf511f Iustin Pop
    _CheckOutputFields(static=self._FIELDS_STATIC,
1737 31bf511f Iustin Pop
                       dynamic=self._FIELDS_DYNAMIC,
1738 1f9430d6 Iustin Pop
                       selected=self.op.output_fields)
1739 1f9430d6 Iustin Pop
1740 6bf01bbb Guido Trotter
    # Lock all nodes, in shared mode
1741 a6ab004b Iustin Pop
    # Temporary removal of locks, should be reverted later
1742 a6ab004b Iustin Pop
    # TODO: reintroduce locks when they are lighter-weight
1743 6bf01bbb Guido Trotter
    self.needed_locks = {}
1744 a6ab004b Iustin Pop
    #self.share_locks[locking.LEVEL_NODE] = 1
1745 a6ab004b Iustin Pop
    #self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
1746 6bf01bbb Guido Trotter
1747 6bf01bbb Guido Trotter
  def CheckPrereq(self):
1748 6bf01bbb Guido Trotter
    """Check prerequisites.
1749 6bf01bbb Guido Trotter

1750 6bf01bbb Guido Trotter
    """
1751 6bf01bbb Guido Trotter
1752 1f9430d6 Iustin Pop
  @staticmethod
1753 1f9430d6 Iustin Pop
  def _DiagnoseByOS(node_list, rlist):
1754 1f9430d6 Iustin Pop
    """Remaps a per-node return list into an a per-os per-node dictionary
1755 1f9430d6 Iustin Pop

1756 e4376078 Iustin Pop
    @param node_list: a list with the names of all nodes
1757 e4376078 Iustin Pop
    @param rlist: a map with node names as keys and OS objects as values
1758 1f9430d6 Iustin Pop

1759 e4376078 Iustin Pop
    @rtype: dict
1760 5fcc718f Iustin Pop
    @return: a dictionary with osnames as keys and as value another map, with
1761 e4376078 Iustin Pop
        nodes as keys and list of OS objects as values, eg::
1762 e4376078 Iustin Pop

1763 e4376078 Iustin Pop
          {"debian-etch": {"node1": [<object>,...],
1764 e4376078 Iustin Pop
                           "node2": [<object>,]}
1765 e4376078 Iustin Pop
          }
1766 1f9430d6 Iustin Pop

1767 1f9430d6 Iustin Pop
    """
1768 1f9430d6 Iustin Pop
    all_os = {}
1769 a6ab004b Iustin Pop
    # we build here the list of nodes that didn't fail the RPC (at RPC
1770 a6ab004b Iustin Pop
    # level), so that nodes with a non-responding node daemon don't
1771 a6ab004b Iustin Pop
    # make all OSes invalid
1772 a6ab004b Iustin Pop
    good_nodes = [node_name for node_name in rlist
1773 a6ab004b Iustin Pop
                  if not rlist[node_name].failed]
1774 1f9430d6 Iustin Pop
    for node_name, nr in rlist.iteritems():
1775 781de953 Iustin Pop
      if nr.failed or not nr.data:
1776 1f9430d6 Iustin Pop
        continue
1777 781de953 Iustin Pop
      for os_obj in nr.data:
1778 b4de68a9 Iustin Pop
        if os_obj.name not in all_os:
1779 1f9430d6 Iustin Pop
          # build a list of nodes for this os containing empty lists
1780 1f9430d6 Iustin Pop
          # for each node in node_list
1781 b4de68a9 Iustin Pop
          all_os[os_obj.name] = {}
1782 a6ab004b Iustin Pop
          for nname in good_nodes:
1783 b4de68a9 Iustin Pop
            all_os[os_obj.name][nname] = []
1784 b4de68a9 Iustin Pop
        all_os[os_obj.name][node_name].append(os_obj)
1785 1f9430d6 Iustin Pop
    return all_os
1786 a8083063 Iustin Pop
1787 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
1788 a8083063 Iustin Pop
    """Compute the list of OSes.
1789 a8083063 Iustin Pop

1790 a8083063 Iustin Pop
    """
1791 a6ab004b Iustin Pop
    valid_nodes = [node for node in self.cfg.GetOnlineNodeList()]
1792 94a02bb5 Iustin Pop
    node_data = self.rpc.call_os_diagnose(valid_nodes)
1793 a8083063 Iustin Pop
    if node_data == False:
1794 3ecf6786 Iustin Pop
      raise errors.OpExecError("Can't gather the list of OSes")
1795 94a02bb5 Iustin Pop
    pol = self._DiagnoseByOS(valid_nodes, node_data)
1796 1f9430d6 Iustin Pop
    output = []
1797 1f9430d6 Iustin Pop
    for os_name, os_data in pol.iteritems():
1798 1f9430d6 Iustin Pop
      row = []
1799 1f9430d6 Iustin Pop
      for field in self.op.output_fields:
1800 1f9430d6 Iustin Pop
        if field == "name":
1801 1f9430d6 Iustin Pop
          val = os_name
1802 1f9430d6 Iustin Pop
        elif field == "valid":
1803 1f9430d6 Iustin Pop
          val = utils.all([osl and osl[0] for osl in os_data.values()])
1804 1f9430d6 Iustin Pop
        elif field == "node_status":
1805 1f9430d6 Iustin Pop
          val = {}
1806 1f9430d6 Iustin Pop
          for node_name, nos_list in os_data.iteritems():
1807 1f9430d6 Iustin Pop
            val[node_name] = [(v.status, v.path) for v in nos_list]
1808 1f9430d6 Iustin Pop
        else:
1809 1f9430d6 Iustin Pop
          raise errors.ParameterError(field)
1810 1f9430d6 Iustin Pop
        row.append(val)
1811 1f9430d6 Iustin Pop
      output.append(row)
1812 1f9430d6 Iustin Pop
1813 1f9430d6 Iustin Pop
    return output
1814 a8083063 Iustin Pop
1815 a8083063 Iustin Pop
1816 a8083063 Iustin Pop
class LURemoveNode(LogicalUnit):
1817 a8083063 Iustin Pop
  """Logical unit for removing a node.
1818 a8083063 Iustin Pop

1819 a8083063 Iustin Pop
  """
1820 a8083063 Iustin Pop
  HPATH = "node-remove"
1821 a8083063 Iustin Pop
  HTYPE = constants.HTYPE_NODE
1822 a8083063 Iustin Pop
  _OP_REQP = ["node_name"]
1823 a8083063 Iustin Pop
1824 a8083063 Iustin Pop
  def BuildHooksEnv(self):
1825 a8083063 Iustin Pop
    """Build hooks env.
1826 a8083063 Iustin Pop

1827 a8083063 Iustin Pop
    This doesn't run on the target node in the pre phase as a failed
1828 d08869ee Guido Trotter
    node would then be impossible to remove.
1829 a8083063 Iustin Pop

1830 a8083063 Iustin Pop
    """
1831 396e1b78 Michael Hanselmann
    env = {
1832 0e137c28 Iustin Pop
      "OP_TARGET": self.op.node_name,
1833 396e1b78 Michael Hanselmann
      "NODE_NAME": self.op.node_name,
1834 396e1b78 Michael Hanselmann
      }
1835 a8083063 Iustin Pop
    all_nodes = self.cfg.GetNodeList()
1836 a8083063 Iustin Pop
    all_nodes.remove(self.op.node_name)
1837 396e1b78 Michael Hanselmann
    return env, all_nodes, all_nodes
1838 a8083063 Iustin Pop
1839 a8083063 Iustin Pop
  def CheckPrereq(self):
1840 a8083063 Iustin Pop
    """Check prerequisites.
1841 a8083063 Iustin Pop

1842 a8083063 Iustin Pop
    This checks:
1843 a8083063 Iustin Pop
     - the node exists in the configuration
1844 a8083063 Iustin Pop
     - it does not have primary or secondary instances
1845 a8083063 Iustin Pop
     - it's not the master
1846 a8083063 Iustin Pop

1847 a8083063 Iustin Pop
    Any errors are signalled by raising errors.OpPrereqError.
1848 a8083063 Iustin Pop

1849 a8083063 Iustin Pop
    """
1850 a8083063 Iustin Pop
    node = self.cfg.GetNodeInfo(self.cfg.ExpandNodeName(self.op.node_name))
1851 a8083063 Iustin Pop
    if node is None:
1852 a02bc76e Iustin Pop
      raise errors.OpPrereqError, ("Node '%s' is unknown." % self.op.node_name)
1853 a8083063 Iustin Pop
1854 a8083063 Iustin Pop
    instance_list = self.cfg.GetInstanceList()
1855 a8083063 Iustin Pop
1856 d6a02168 Michael Hanselmann
    masternode = self.cfg.GetMasterNode()
1857 a8083063 Iustin Pop
    if node.name == masternode:
1858 3ecf6786 Iustin Pop
      raise errors.OpPrereqError("Node is the master node,"
1859 3ecf6786 Iustin Pop
                                 " you need to failover first.")
1860 a8083063 Iustin Pop
1861 a8083063 Iustin Pop
    for instance_name in instance_list:
1862 a8083063 Iustin Pop
      instance = self.cfg.GetInstanceInfo(instance_name)
1863 6b12959c Iustin Pop
      if node.name in instance.all_nodes:
1864 6b12959c Iustin Pop
        raise errors.OpPrereqError("Instance %s is still running on the node,"
1865 3ecf6786 Iustin Pop
                                   " please remove first." % instance_name)
1866 a8083063 Iustin Pop
    self.op.node_name = node.name
1867 a8083063 Iustin Pop
    self.node = node
1868 a8083063 Iustin Pop
1869 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
1870 a8083063 Iustin Pop
    """Removes the node from the cluster.
1871 a8083063 Iustin Pop

1872 a8083063 Iustin Pop
    """
1873 a8083063 Iustin Pop
    node = self.node
1874 9a4f63d1 Iustin Pop
    logging.info("Stopping the node daemon and removing configs from node %s",
1875 9a4f63d1 Iustin Pop
                 node.name)
1876 a8083063 Iustin Pop
1877 d8470559 Michael Hanselmann
    self.context.RemoveNode(node.name)
1878 a8083063 Iustin Pop
1879 72737a7f Iustin Pop
    self.rpc.call_node_leave_cluster(node.name)
1880 c8a0948f Michael Hanselmann
1881 eb1742d5 Guido Trotter
    # Promote nodes to master candidate as needed
1882 ec0292f1 Iustin Pop
    _AdjustCandidatePool(self)
1883 eb1742d5 Guido Trotter
1884 a8083063 Iustin Pop
1885 a8083063 Iustin Pop
class LUQueryNodes(NoHooksLU):
1886 a8083063 Iustin Pop
  """Logical unit for querying nodes.
1887 a8083063 Iustin Pop

1888 a8083063 Iustin Pop
  """
1889 bc8e4a1a Iustin Pop
  _OP_REQP = ["output_fields", "names", "use_locking"]
1890 35705d8f Guido Trotter
  REQ_BGL = False
1891 a2d2e1a7 Iustin Pop
  _FIELDS_DYNAMIC = utils.FieldSet(
1892 31bf511f Iustin Pop
    "dtotal", "dfree",
1893 31bf511f Iustin Pop
    "mtotal", "mnode", "mfree",
1894 31bf511f Iustin Pop
    "bootid",
1895 0105bad3 Iustin Pop
    "ctotal", "cnodes", "csockets",
1896 31bf511f Iustin Pop
    )
1897 31bf511f Iustin Pop
1898 a2d2e1a7 Iustin Pop
  _FIELDS_STATIC = utils.FieldSet(
1899 31bf511f Iustin Pop
    "name", "pinst_cnt", "sinst_cnt",
1900 31bf511f Iustin Pop
    "pinst_list", "sinst_list",
1901 31bf511f Iustin Pop
    "pip", "sip", "tags",
1902 31bf511f Iustin Pop
    "serial_no",
1903 0e67cdbe Iustin Pop
    "master_candidate",
1904 0e67cdbe Iustin Pop
    "master",
1905 9ddb5e45 Iustin Pop
    "offline",
1906 0b2454b9 Iustin Pop
    "drained",
1907 31bf511f Iustin Pop
    )
1908 a8083063 Iustin Pop
1909 35705d8f Guido Trotter
  def ExpandNames(self):
1910 31bf511f Iustin Pop
    _CheckOutputFields(static=self._FIELDS_STATIC,
1911 31bf511f Iustin Pop
                       dynamic=self._FIELDS_DYNAMIC,
1912 dcb93971 Michael Hanselmann
                       selected=self.op.output_fields)
1913 a8083063 Iustin Pop
1914 35705d8f Guido Trotter
    self.needed_locks = {}
1915 35705d8f Guido Trotter
    self.share_locks[locking.LEVEL_NODE] = 1
1916 c8d8b4c8 Iustin Pop
1917 c8d8b4c8 Iustin Pop
    if self.op.names:
1918 c8d8b4c8 Iustin Pop
      self.wanted = _GetWantedNodes(self, self.op.names)
1919 35705d8f Guido Trotter
    else:
1920 c8d8b4c8 Iustin Pop
      self.wanted = locking.ALL_SET
1921 c8d8b4c8 Iustin Pop
1922 bc8e4a1a Iustin Pop
    self.do_node_query = self._FIELDS_STATIC.NonMatching(self.op.output_fields)
1923 bc8e4a1a Iustin Pop
    self.do_locking = self.do_node_query and self.op.use_locking
1924 c8d8b4c8 Iustin Pop
    if self.do_locking:
1925 c8d8b4c8 Iustin Pop
      # if we don't request only static fields, we need to lock the nodes
1926 c8d8b4c8 Iustin Pop
      self.needed_locks[locking.LEVEL_NODE] = self.wanted
1927 c8d8b4c8 Iustin Pop
1928 35705d8f Guido Trotter
1929 35705d8f Guido Trotter
  def CheckPrereq(self):
1930 35705d8f Guido Trotter
    """Check prerequisites.
1931 35705d8f Guido Trotter

1932 35705d8f Guido Trotter
    """
1933 c8d8b4c8 Iustin Pop
    # The validation of the node list is done in the _GetWantedNodes,
1934 c8d8b4c8 Iustin Pop
    # if non empty, and if empty, there's no validation to do
1935 c8d8b4c8 Iustin Pop
    pass
1936 a8083063 Iustin Pop
1937 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
1938 a8083063 Iustin Pop
    """Computes the list of nodes and their attributes.
1939 a8083063 Iustin Pop

1940 a8083063 Iustin Pop
    """
1941 c8d8b4c8 Iustin Pop
    all_info = self.cfg.GetAllNodesInfo()
1942 c8d8b4c8 Iustin Pop
    if self.do_locking:
1943 c8d8b4c8 Iustin Pop
      nodenames = self.acquired_locks[locking.LEVEL_NODE]
1944 3fa93523 Guido Trotter
    elif self.wanted != locking.ALL_SET:
1945 3fa93523 Guido Trotter
      nodenames = self.wanted
1946 3fa93523 Guido Trotter
      missing = set(nodenames).difference(all_info.keys())
1947 3fa93523 Guido Trotter
      if missing:
1948 7b3a8fb5 Iustin Pop
        raise errors.OpExecError(
1949 3fa93523 Guido Trotter
          "Some nodes were removed before retrieving their data: %s" % missing)
1950 c8d8b4c8 Iustin Pop
    else:
1951 c8d8b4c8 Iustin Pop
      nodenames = all_info.keys()
1952 c1f1cbb2 Iustin Pop
1953 c1f1cbb2 Iustin Pop
    nodenames = utils.NiceSort(nodenames)
1954 c8d8b4c8 Iustin Pop
    nodelist = [all_info[name] for name in nodenames]
1955 a8083063 Iustin Pop
1956 a8083063 Iustin Pop
    # begin data gathering
1957 a8083063 Iustin Pop
1958 bc8e4a1a Iustin Pop
    if self.do_node_query:
1959 a8083063 Iustin Pop
      live_data = {}
1960 72737a7f Iustin Pop
      node_data = self.rpc.call_node_info(nodenames, self.cfg.GetVGName(),
1961 72737a7f Iustin Pop
                                          self.cfg.GetHypervisorType())
1962 a8083063 Iustin Pop
      for name in nodenames:
1963 781de953 Iustin Pop
        nodeinfo = node_data[name]
1964 781de953 Iustin Pop
        if not nodeinfo.failed and nodeinfo.data:
1965 781de953 Iustin Pop
          nodeinfo = nodeinfo.data
1966 d599d686 Iustin Pop
          fn = utils.TryConvert
1967 a8083063 Iustin Pop
          live_data[name] = {
1968 d599d686 Iustin Pop
            "mtotal": fn(int, nodeinfo.get('memory_total', None)),
1969 d599d686 Iustin Pop
            "mnode": fn(int, nodeinfo.get('memory_dom0', None)),
1970 d599d686 Iustin Pop
            "mfree": fn(int, nodeinfo.get('memory_free', None)),
1971 d599d686 Iustin Pop
            "dtotal": fn(int, nodeinfo.get('vg_size', None)),
1972 d599d686 Iustin Pop
            "dfree": fn(int, nodeinfo.get('vg_free', None)),
1973 d599d686 Iustin Pop
            "ctotal": fn(int, nodeinfo.get('cpu_total', None)),
1974 d599d686 Iustin Pop
            "bootid": nodeinfo.get('bootid', None),
1975 0105bad3 Iustin Pop
            "cnodes": fn(int, nodeinfo.get('cpu_nodes', None)),
1976 0105bad3 Iustin Pop
            "csockets": fn(int, nodeinfo.get('cpu_sockets', None)),
1977 a8083063 Iustin Pop
            }
1978 a8083063 Iustin Pop
        else:
1979 a8083063 Iustin Pop
          live_data[name] = {}
1980 a8083063 Iustin Pop
    else:
1981 a8083063 Iustin Pop
      live_data = dict.fromkeys(nodenames, {})
1982 a8083063 Iustin Pop
1983 ec223efb Iustin Pop
    node_to_primary = dict([(name, set()) for name in nodenames])
1984 ec223efb Iustin Pop
    node_to_secondary = dict([(name, set()) for name in nodenames])
1985 a8083063 Iustin Pop
1986 ec223efb Iustin Pop
    inst_fields = frozenset(("pinst_cnt", "pinst_list",
1987 ec223efb Iustin Pop
                             "sinst_cnt", "sinst_list"))
1988 ec223efb Iustin Pop
    if inst_fields & frozenset(self.op.output_fields):
1989 a8083063 Iustin Pop
      instancelist = self.cfg.GetInstanceList()
1990 a8083063 Iustin Pop
1991 ec223efb Iustin Pop
      for instance_name in instancelist:
1992 ec223efb Iustin Pop
        inst = self.cfg.GetInstanceInfo(instance_name)
1993 ec223efb Iustin Pop
        if inst.primary_node in node_to_primary:
1994 ec223efb Iustin Pop
          node_to_primary[inst.primary_node].add(inst.name)
1995 ec223efb Iustin Pop
        for secnode in inst.secondary_nodes:
1996 ec223efb Iustin Pop
          if secnode in node_to_secondary:
1997 ec223efb Iustin Pop
            node_to_secondary[secnode].add(inst.name)
1998 a8083063 Iustin Pop
1999 0e67cdbe Iustin Pop
    master_node = self.cfg.GetMasterNode()
2000 0e67cdbe Iustin Pop
2001 a8083063 Iustin Pop
    # end data gathering
2002 a8083063 Iustin Pop
2003 a8083063 Iustin Pop
    output = []
2004 a8083063 Iustin Pop
    for node in nodelist:
2005 a8083063 Iustin Pop
      node_output = []
2006 a8083063 Iustin Pop
      for field in self.op.output_fields:
2007 a8083063 Iustin Pop
        if field == "name":
2008 a8083063 Iustin Pop
          val = node.name
2009 ec223efb Iustin Pop
        elif field == "pinst_list":
2010 ec223efb Iustin Pop
          val = list(node_to_primary[node.name])
2011 ec223efb Iustin Pop
        elif field == "sinst_list":
2012 ec223efb Iustin Pop
          val = list(node_to_secondary[node.name])
2013 ec223efb Iustin Pop
        elif field == "pinst_cnt":
2014 ec223efb Iustin Pop
          val = len(node_to_primary[node.name])
2015 ec223efb Iustin Pop
        elif field == "sinst_cnt":
2016 ec223efb Iustin Pop
          val = len(node_to_secondary[node.name])
2017 a8083063 Iustin Pop
        elif field == "pip":
2018 a8083063 Iustin Pop
          val = node.primary_ip
2019 a8083063 Iustin Pop
        elif field == "sip":
2020 a8083063 Iustin Pop
          val = node.secondary_ip
2021 130a6a6f Iustin Pop
        elif field == "tags":
2022 130a6a6f Iustin Pop
          val = list(node.GetTags())
2023 38d7239a Iustin Pop
        elif field == "serial_no":
2024 38d7239a Iustin Pop
          val = node.serial_no
2025 0e67cdbe Iustin Pop
        elif field == "master_candidate":
2026 0e67cdbe Iustin Pop
          val = node.master_candidate
2027 0e67cdbe Iustin Pop
        elif field == "master":
2028 0e67cdbe Iustin Pop
          val = node.name == master_node
2029 9ddb5e45 Iustin Pop
        elif field == "offline":
2030 9ddb5e45 Iustin Pop
          val = node.offline
2031 0b2454b9 Iustin Pop
        elif field == "drained":
2032 0b2454b9 Iustin Pop
          val = node.drained
2033 31bf511f Iustin Pop
        elif self._FIELDS_DYNAMIC.Matches(field):
2034 ec223efb Iustin Pop
          val = live_data[node.name].get(field, None)
2035 a8083063 Iustin Pop
        else:
2036 3ecf6786 Iustin Pop
          raise errors.ParameterError(field)
2037 a8083063 Iustin Pop
        node_output.append(val)
2038 a8083063 Iustin Pop
      output.append(node_output)
2039 a8083063 Iustin Pop
2040 a8083063 Iustin Pop
    return output
2041 a8083063 Iustin Pop
2042 a8083063 Iustin Pop
2043 dcb93971 Michael Hanselmann
class LUQueryNodeVolumes(NoHooksLU):
2044 dcb93971 Michael Hanselmann
  """Logical unit for getting volumes on node(s).
2045 dcb93971 Michael Hanselmann

2046 dcb93971 Michael Hanselmann
  """
2047 dcb93971 Michael Hanselmann
  _OP_REQP = ["nodes", "output_fields"]
2048 21a15682 Guido Trotter
  REQ_BGL = False
2049 a2d2e1a7 Iustin Pop
  _FIELDS_DYNAMIC = utils.FieldSet("phys", "vg", "name", "size", "instance")
2050 a2d2e1a7 Iustin Pop
  _FIELDS_STATIC = utils.FieldSet("node")
2051 21a15682 Guido Trotter
2052 21a15682 Guido Trotter
  def ExpandNames(self):
2053 31bf511f Iustin Pop
    _CheckOutputFields(static=self._FIELDS_STATIC,
2054 31bf511f Iustin Pop
                       dynamic=self._FIELDS_DYNAMIC,
2055 21a15682 Guido Trotter
                       selected=self.op.output_fields)
2056 21a15682 Guido Trotter
2057 21a15682 Guido Trotter
    self.needed_locks = {}
2058 21a15682 Guido Trotter
    self.share_locks[locking.LEVEL_NODE] = 1
2059 21a15682 Guido Trotter
    if not self.op.nodes:
2060 e310b019 Guido Trotter
      self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
2061 21a15682 Guido Trotter
    else:
2062 21a15682 Guido Trotter
      self.needed_locks[locking.LEVEL_NODE] = \
2063 21a15682 Guido Trotter
        _GetWantedNodes(self, self.op.nodes)
2064 dcb93971 Michael Hanselmann
2065 dcb93971 Michael Hanselmann
  def CheckPrereq(self):
2066 dcb93971 Michael Hanselmann
    """Check prerequisites.
2067 dcb93971 Michael Hanselmann

2068 dcb93971 Michael Hanselmann
    This checks that the fields required are valid output fields.
2069 dcb93971 Michael Hanselmann

2070 dcb93971 Michael Hanselmann
    """
2071 21a15682 Guido Trotter
    self.nodes = self.acquired_locks[locking.LEVEL_NODE]
2072 dcb93971 Michael Hanselmann
2073 dcb93971 Michael Hanselmann
  def Exec(self, feedback_fn):
2074 dcb93971 Michael Hanselmann
    """Computes the list of nodes and their attributes.
2075 dcb93971 Michael Hanselmann

2076 dcb93971 Michael Hanselmann
    """
2077 a7ba5e53 Iustin Pop
    nodenames = self.nodes
2078 72737a7f Iustin Pop
    volumes = self.rpc.call_node_volumes(nodenames)
2079 dcb93971 Michael Hanselmann
2080 dcb93971 Michael Hanselmann
    ilist = [self.cfg.GetInstanceInfo(iname) for iname
2081 dcb93971 Michael Hanselmann
             in self.cfg.GetInstanceList()]
2082 dcb93971 Michael Hanselmann
2083 dcb93971 Michael Hanselmann
    lv_by_node = dict([(inst, inst.MapLVsByNode()) for inst in ilist])
2084 dcb93971 Michael Hanselmann
2085 dcb93971 Michael Hanselmann
    output = []
2086 dcb93971 Michael Hanselmann
    for node in nodenames:
2087 781de953 Iustin Pop
      if node not in volumes or volumes[node].failed or not volumes[node].data:
2088 37d19eb2 Michael Hanselmann
        continue
2089 37d19eb2 Michael Hanselmann
2090 781de953 Iustin Pop
      node_vols = volumes[node].data[:]
2091 dcb93971 Michael Hanselmann
      node_vols.sort(key=lambda vol: vol['dev'])
2092 dcb93971 Michael Hanselmann
2093 dcb93971 Michael Hanselmann
      for vol in node_vols:
2094 dcb93971 Michael Hanselmann
        node_output = []
2095 dcb93971 Michael Hanselmann
        for field in self.op.output_fields:
2096 dcb93971 Michael Hanselmann
          if field == "node":
2097 dcb93971 Michael Hanselmann
            val = node
2098 dcb93971 Michael Hanselmann
          elif field == "phys":
2099 dcb93971 Michael Hanselmann
            val = vol['dev']
2100 dcb93971 Michael Hanselmann
          elif field == "vg":
2101 dcb93971 Michael Hanselmann
            val = vol['vg']
2102 dcb93971 Michael Hanselmann
          elif field == "name":
2103 dcb93971 Michael Hanselmann
            val = vol['name']
2104 dcb93971 Michael Hanselmann
          elif field == "size":
2105 dcb93971 Michael Hanselmann
            val = int(float(vol['size']))
2106 dcb93971 Michael Hanselmann
          elif field == "instance":
2107 dcb93971 Michael Hanselmann
            for inst in ilist:
2108 dcb93971 Michael Hanselmann
              if node not in lv_by_node[inst]:
2109 dcb93971 Michael Hanselmann
                continue
2110 dcb93971 Michael Hanselmann
              if vol['name'] in lv_by_node[inst][node]:
2111 dcb93971 Michael Hanselmann
                val = inst.name
2112 dcb93971 Michael Hanselmann
                break
2113 dcb93971 Michael Hanselmann
            else:
2114 dcb93971 Michael Hanselmann
              val = '-'
2115 dcb93971 Michael Hanselmann
          else:
2116 3ecf6786 Iustin Pop
            raise errors.ParameterError(field)
2117 dcb93971 Michael Hanselmann
          node_output.append(str(val))
2118 dcb93971 Michael Hanselmann
2119 dcb93971 Michael Hanselmann
        output.append(node_output)
2120 dcb93971 Michael Hanselmann
2121 dcb93971 Michael Hanselmann
    return output
2122 dcb93971 Michael Hanselmann
2123 dcb93971 Michael Hanselmann
2124 a8083063 Iustin Pop
class LUAddNode(LogicalUnit):
2125 a8083063 Iustin Pop
  """Logical unit for adding node to the cluster.
2126 a8083063 Iustin Pop

2127 a8083063 Iustin Pop
  """
2128 a8083063 Iustin Pop
  HPATH = "node-add"
2129 a8083063 Iustin Pop
  HTYPE = constants.HTYPE_NODE
2130 a8083063 Iustin Pop
  _OP_REQP = ["node_name"]
2131 a8083063 Iustin Pop
2132 a8083063 Iustin Pop
  def BuildHooksEnv(self):
2133 a8083063 Iustin Pop
    """Build hooks env.
2134 a8083063 Iustin Pop

2135 a8083063 Iustin Pop
    This will run on all nodes before, and on all nodes + the new node after.
2136 a8083063 Iustin Pop

2137 a8083063 Iustin Pop
    """
2138 a8083063 Iustin Pop
    env = {
2139 0e137c28 Iustin Pop
      "OP_TARGET": self.op.node_name,
2140 a8083063 Iustin Pop
      "NODE_NAME": self.op.node_name,
2141 a8083063 Iustin Pop
      "NODE_PIP": self.op.primary_ip,
2142 a8083063 Iustin Pop
      "NODE_SIP": self.op.secondary_ip,
2143 a8083063 Iustin Pop
      }
2144 a8083063 Iustin Pop
    nodes_0 = self.cfg.GetNodeList()
2145 a8083063 Iustin Pop
    nodes_1 = nodes_0 + [self.op.node_name, ]
2146 a8083063 Iustin Pop
    return env, nodes_0, nodes_1
2147 a8083063 Iustin Pop
2148 a8083063 Iustin Pop
  def CheckPrereq(self):
2149 a8083063 Iustin Pop
    """Check prerequisites.
2150 a8083063 Iustin Pop

2151 a8083063 Iustin Pop
    This checks:
2152 a8083063 Iustin Pop
     - the new node is not already in the config
2153 a8083063 Iustin Pop
     - it is resolvable
2154 a8083063 Iustin Pop
     - its parameters (single/dual homed) matches the cluster
2155 a8083063 Iustin Pop

2156 a8083063 Iustin Pop
    Any errors are signalled by raising errors.OpPrereqError.
2157 a8083063 Iustin Pop

2158 a8083063 Iustin Pop
    """
2159 a8083063 Iustin Pop
    node_name = self.op.node_name
2160 a8083063 Iustin Pop
    cfg = self.cfg
2161 a8083063 Iustin Pop
2162 89e1fc26 Iustin Pop
    dns_data = utils.HostInfo(node_name)
2163 a8083063 Iustin Pop
2164 bcf043c9 Iustin Pop
    node = dns_data.name
2165 bcf043c9 Iustin Pop
    primary_ip = self.op.primary_ip = dns_data.ip
2166 a8083063 Iustin Pop
    secondary_ip = getattr(self.op, "secondary_ip", None)
2167 a8083063 Iustin Pop
    if secondary_ip is None:
2168 a8083063 Iustin Pop
      secondary_ip = primary_ip
2169 a8083063 Iustin Pop
    if not utils.IsValidIP(secondary_ip):
2170 3ecf6786 Iustin Pop
      raise errors.OpPrereqError("Invalid secondary IP given")
2171 a8083063 Iustin Pop
    self.op.secondary_ip = secondary_ip
2172 e7c6e02b Michael Hanselmann
2173 a8083063 Iustin Pop
    node_list = cfg.GetNodeList()
2174 e7c6e02b Michael Hanselmann
    if not self.op.readd and node in node_list:
2175 e7c6e02b Michael Hanselmann
      raise errors.OpPrereqError("Node %s is already in the configuration" %
2176 e7c6e02b Michael Hanselmann
                                 node)
2177 e7c6e02b Michael Hanselmann
    elif self.op.readd and node not in node_list:
2178 e7c6e02b Michael Hanselmann
      raise errors.OpPrereqError("Node %s is not in the configuration" % node)
2179 a8083063 Iustin Pop
2180 a8083063 Iustin Pop
    for existing_node_name in node_list:
2181 a8083063 Iustin Pop
      existing_node = cfg.GetNodeInfo(existing_node_name)
2182 e7c6e02b Michael Hanselmann
2183 e7c6e02b Michael Hanselmann
      if self.op.readd and node == existing_node_name:
2184 e7c6e02b Michael Hanselmann
        if (existing_node.primary_ip != primary_ip or
2185 e7c6e02b Michael Hanselmann
            existing_node.secondary_ip != secondary_ip):
2186 e7c6e02b Michael Hanselmann
          raise errors.OpPrereqError("Readded node doesn't have the same IP"
2187 e7c6e02b Michael Hanselmann
                                     " address configuration as before")
2188 e7c6e02b Michael Hanselmann
        continue
2189 e7c6e02b Michael Hanselmann
2190 a8083063 Iustin Pop
      if (existing_node.primary_ip == primary_ip or
2191 a8083063 Iustin Pop
          existing_node.secondary_ip == primary_ip or
2192 a8083063 Iustin Pop
          existing_node.primary_ip == secondary_ip or
2193 a8083063 Iustin Pop
          existing_node.secondary_ip == secondary_ip):
2194 3ecf6786 Iustin Pop
        raise errors.OpPrereqError("New node ip address(es) conflict with"
2195 3ecf6786 Iustin Pop
                                   " existing node %s" % existing_node.name)
2196 a8083063 Iustin Pop
2197 a8083063 Iustin Pop
    # check that the type of the node (single versus dual homed) is the
2198 a8083063 Iustin Pop
    # same as for the master
2199 d6a02168 Michael Hanselmann
    myself = cfg.GetNodeInfo(self.cfg.GetMasterNode())
2200 a8083063 Iustin Pop
    master_singlehomed = myself.secondary_ip == myself.primary_ip
2201 a8083063 Iustin Pop
    newbie_singlehomed = secondary_ip == primary_ip
2202 a8083063 Iustin Pop
    if master_singlehomed != newbie_singlehomed:
2203 a8083063 Iustin Pop
      if master_singlehomed:
2204 3ecf6786 Iustin Pop
        raise errors.OpPrereqError("The master has no private ip but the"
2205 3ecf6786 Iustin Pop
                                   " new node has one")
2206 a8083063 Iustin Pop
      else:
2207 3ecf6786 Iustin Pop
        raise errors.OpPrereqError("The master has a private ip but the"
2208 3ecf6786 Iustin Pop
                                   " new node doesn't have one")
2209 a8083063 Iustin Pop
2210 a8083063 Iustin Pop
    # checks reachablity
2211 b15d625f Iustin Pop
    if not utils.TcpPing(primary_ip, constants.DEFAULT_NODED_PORT):
2212 3ecf6786 Iustin Pop
      raise errors.OpPrereqError("Node not reachable by ping")
2213 a8083063 Iustin Pop
2214 a8083063 Iustin Pop
    if not newbie_singlehomed:
2215 a8083063 Iustin Pop
      # check reachability from my secondary ip to newbie's secondary ip
2216 b15d625f Iustin Pop
      if not utils.TcpPing(secondary_ip, constants.DEFAULT_NODED_PORT,
2217 b15d625f Iustin Pop
                           source=myself.secondary_ip):
2218 f4bc1f2c Michael Hanselmann
        raise errors.OpPrereqError("Node secondary ip not reachable by TCP"
2219 f4bc1f2c Michael Hanselmann
                                   " based ping to noded port")
2220 a8083063 Iustin Pop
2221 0fff97e9 Guido Trotter
    cp_size = self.cfg.GetClusterInfo().candidate_pool_size
2222 ec0292f1 Iustin Pop
    mc_now, _ = self.cfg.GetMasterCandidateStats()
2223 ec0292f1 Iustin Pop
    master_candidate = mc_now < cp_size
2224 0fff97e9 Guido Trotter
2225 a8083063 Iustin Pop
    self.new_node = objects.Node(name=node,
2226 a8083063 Iustin Pop
                                 primary_ip=primary_ip,
2227 0fff97e9 Guido Trotter
                                 secondary_ip=secondary_ip,
2228 fc0fe88c Iustin Pop
                                 master_candidate=master_candidate,
2229 af64c0ea Iustin Pop
                                 offline=False, drained=False)
2230 a8083063 Iustin Pop
2231 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
2232 a8083063 Iustin Pop
    """Adds the new node to the cluster.
2233 a8083063 Iustin Pop

2234 a8083063 Iustin Pop
    """
2235 a8083063 Iustin Pop
    new_node = self.new_node
2236 a8083063 Iustin Pop
    node = new_node.name
2237 a8083063 Iustin Pop
2238 a8083063 Iustin Pop
    # check connectivity
2239 72737a7f Iustin Pop
    result = self.rpc.call_version([node])[node]
2240 781de953 Iustin Pop
    result.Raise()
2241 781de953 Iustin Pop
    if result.data:
2242 781de953 Iustin Pop
      if constants.PROTOCOL_VERSION == result.data:
2243 9a4f63d1 Iustin Pop
        logging.info("Communication to node %s fine, sw version %s match",
2244 781de953 Iustin Pop
                     node, result.data)
2245 a8083063 Iustin Pop
      else:
2246 3ecf6786 Iustin Pop
        raise errors.OpExecError("Version mismatch master version %s,"
2247 3ecf6786 Iustin Pop
                                 " node version %s" %
2248 781de953 Iustin Pop
                                 (constants.PROTOCOL_VERSION, result.data))
2249 a8083063 Iustin Pop
    else:
2250 3ecf6786 Iustin Pop
      raise errors.OpExecError("Cannot get version from the new node")
2251 a8083063 Iustin Pop
2252 a8083063 Iustin Pop
    # setup ssh on node
2253 9a4f63d1 Iustin Pop
    logging.info("Copy ssh key to node %s", node)
2254 70d9e3d8 Iustin Pop
    priv_key, pub_key, _ = ssh.GetUserFiles(constants.GANETI_RUNAS)
2255 a8083063 Iustin Pop
    keyarray = []
2256 70d9e3d8 Iustin Pop
    keyfiles = [constants.SSH_HOST_DSA_PRIV, constants.SSH_HOST_DSA_PUB,
2257 70d9e3d8 Iustin Pop
                constants.SSH_HOST_RSA_PRIV, constants.SSH_HOST_RSA_PUB,
2258 70d9e3d8 Iustin Pop
                priv_key, pub_key]
2259 a8083063 Iustin Pop
2260 a8083063 Iustin Pop
    for i in keyfiles:
2261 a8083063 Iustin Pop
      f = open(i, 'r')
2262 a8083063 Iustin Pop
      try:
2263 a8083063 Iustin Pop
        keyarray.append(f.read())
2264 a8083063 Iustin Pop
      finally:
2265 a8083063 Iustin Pop
        f.close()
2266 a8083063 Iustin Pop
2267 72737a7f Iustin Pop
    result = self.rpc.call_node_add(node, keyarray[0], keyarray[1],
2268 72737a7f Iustin Pop
                                    keyarray[2],
2269 72737a7f Iustin Pop
                                    keyarray[3], keyarray[4], keyarray[5])
2270 a8083063 Iustin Pop
2271 a1b805fb Iustin Pop
    msg = result.RemoteFailMsg()
2272 a1b805fb Iustin Pop
    if msg:
2273 a1b805fb Iustin Pop
      raise errors.OpExecError("Cannot transfer ssh keys to the"
2274 a1b805fb Iustin Pop
                               " new node: %s" % msg)
2275 a8083063 Iustin Pop
2276 a8083063 Iustin Pop
    # Add node to our /etc/hosts, and add key to known_hosts
2277 b86a6bcd Guido Trotter
    if self.cfg.GetClusterInfo().modify_etc_hosts:
2278 b86a6bcd Guido Trotter
      utils.AddHostToEtcHosts(new_node.name)
2279 c8a0948f Michael Hanselmann
2280 a8083063 Iustin Pop
    if new_node.secondary_ip != new_node.primary_ip:
2281 781de953 Iustin Pop
      result = self.rpc.call_node_has_ip_address(new_node.name,
2282 781de953 Iustin Pop
                                                 new_node.secondary_ip)
2283 781de953 Iustin Pop
      if result.failed or not result.data:
2284 f4bc1f2c Michael Hanselmann
        raise errors.OpExecError("Node claims it doesn't have the secondary ip"
2285 f4bc1f2c Michael Hanselmann
                                 " you gave (%s). Please fix and re-run this"
2286 f4bc1f2c Michael Hanselmann
                                 " command." % new_node.secondary_ip)
2287 a8083063 Iustin Pop
2288 d6a02168 Michael Hanselmann
    node_verify_list = [self.cfg.GetMasterNode()]
2289 5c0527ed Guido Trotter
    node_verify_param = {
2290 5c0527ed Guido Trotter
      'nodelist': [node],
2291 5c0527ed Guido Trotter
      # TODO: do a node-net-test as well?
2292 5c0527ed Guido Trotter
    }
2293 5c0527ed Guido Trotter
2294 72737a7f Iustin Pop
    result = self.rpc.call_node_verify(node_verify_list, node_verify_param,
2295 72737a7f Iustin Pop
                                       self.cfg.GetClusterName())
2296 5c0527ed Guido Trotter
    for verifier in node_verify_list:
2297 f08ce603 Guido Trotter
      if result[verifier].failed or not result[verifier].data:
2298 5c0527ed Guido Trotter
        raise errors.OpExecError("Cannot communicate with %s's node daemon"
2299 5c0527ed Guido Trotter
                                 " for remote verification" % verifier)
2300 781de953 Iustin Pop
      if result[verifier].data['nodelist']:
2301 781de953 Iustin Pop
        for failed in result[verifier].data['nodelist']:
2302 5c0527ed Guido Trotter
          feedback_fn("ssh/hostname verification failed %s -> %s" %
2303 bafc1d90 Iustin Pop
                      (verifier, result[verifier].data['nodelist'][failed]))
2304 5c0527ed Guido Trotter
        raise errors.OpExecError("ssh/hostname verification failed.")
2305 ff98055b Iustin Pop
2306 d8470559 Michael Hanselmann
    if self.op.readd:
2307 28eddce5 Guido Trotter
      _RedistributeAncillaryFiles(self)
2308 d8470559 Michael Hanselmann
      self.context.ReaddNode(new_node)
2309 d8470559 Michael Hanselmann
    else:
2310 035566e3 Iustin Pop
      _RedistributeAncillaryFiles(self, additional_nodes=[node])
2311 d8470559 Michael Hanselmann
      self.context.AddNode(new_node)
2312 a8083063 Iustin Pop
2313 a8083063 Iustin Pop
2314 b31c8676 Iustin Pop
class LUSetNodeParams(LogicalUnit):
2315 b31c8676 Iustin Pop
  """Modifies the parameters of a node.
2316 b31c8676 Iustin Pop

2317 b31c8676 Iustin Pop
  """
2318 b31c8676 Iustin Pop
  HPATH = "node-modify"
2319 b31c8676 Iustin Pop
  HTYPE = constants.HTYPE_NODE
2320 b31c8676 Iustin Pop
  _OP_REQP = ["node_name"]
2321 b31c8676 Iustin Pop
  REQ_BGL = False
2322 b31c8676 Iustin Pop
2323 b31c8676 Iustin Pop
  def CheckArguments(self):
2324 b31c8676 Iustin Pop
    node_name = self.cfg.ExpandNodeName(self.op.node_name)
2325 b31c8676 Iustin Pop
    if node_name is None:
2326 b31c8676 Iustin Pop
      raise errors.OpPrereqError("Invalid node name '%s'" % self.op.node_name)
2327 b31c8676 Iustin Pop
    self.op.node_name = node_name
2328 3a5ba66a Iustin Pop
    _CheckBooleanOpField(self.op, 'master_candidate')
2329 3a5ba66a Iustin Pop
    _CheckBooleanOpField(self.op, 'offline')
2330 c9d443ea Iustin Pop
    _CheckBooleanOpField(self.op, 'drained')
2331 c9d443ea Iustin Pop
    all_mods = [self.op.offline, self.op.master_candidate, self.op.drained]
2332 c9d443ea Iustin Pop
    if all_mods.count(None) == 3:
2333 b31c8676 Iustin Pop
      raise errors.OpPrereqError("Please pass at least one modification")
2334 c9d443ea Iustin Pop
    if all_mods.count(True) > 1:
2335 c9d443ea Iustin Pop
      raise errors.OpPrereqError("Can't set the node into more than one"
2336 c9d443ea Iustin Pop
                                 " state at the same time")
2337 b31c8676 Iustin Pop
2338 b31c8676 Iustin Pop
  def ExpandNames(self):
2339 b31c8676 Iustin Pop
    self.needed_locks = {locking.LEVEL_NODE: self.op.node_name}
2340 b31c8676 Iustin Pop
2341 b31c8676 Iustin Pop
  def BuildHooksEnv(self):
2342 b31c8676 Iustin Pop
    """Build hooks env.
2343 b31c8676 Iustin Pop

2344 b31c8676 Iustin Pop
    This runs on the master node.
2345 b31c8676 Iustin Pop

2346 b31c8676 Iustin Pop
    """
2347 b31c8676 Iustin Pop
    env = {
2348 b31c8676 Iustin Pop
      "OP_TARGET": self.op.node_name,
2349 b31c8676 Iustin Pop
      "MASTER_CANDIDATE": str(self.op.master_candidate),
2350 3a5ba66a Iustin Pop
      "OFFLINE": str(self.op.offline),
2351 c9d443ea Iustin Pop
      "DRAINED": str(self.op.drained),
2352 b31c8676 Iustin Pop
      }
2353 b31c8676 Iustin Pop
    nl = [self.cfg.GetMasterNode(),
2354 b31c8676 Iustin Pop
          self.op.node_name]
2355 b31c8676 Iustin Pop
    return env, nl, nl
2356 b31c8676 Iustin Pop
2357 b31c8676 Iustin Pop
  def CheckPrereq(self):
2358 b31c8676 Iustin Pop
    """Check prerequisites.
2359 b31c8676 Iustin Pop

2360 b31c8676 Iustin Pop
    This only checks the instance list against the existing names.
2361 b31c8676 Iustin Pop

2362 b31c8676 Iustin Pop
    """
2363 3a5ba66a Iustin Pop
    node = self.node = self.cfg.GetNodeInfo(self.op.node_name)
2364 b31c8676 Iustin Pop
2365 c9d443ea Iustin Pop
    if ((self.op.master_candidate == False or self.op.offline == True or
2366 c9d443ea Iustin Pop
         self.op.drained == True) and node.master_candidate):
2367 3a5ba66a Iustin Pop
      # we will demote the node from master_candidate
2368 3a26773f Iustin Pop
      if self.op.node_name == self.cfg.GetMasterNode():
2369 3a26773f Iustin Pop
        raise errors.OpPrereqError("The master node has to be a"
2370 c9d443ea Iustin Pop
                                   " master candidate, online and not drained")
2371 3e83dd48 Iustin Pop
      cp_size = self.cfg.GetClusterInfo().candidate_pool_size
2372 3a5ba66a Iustin Pop
      num_candidates, _ = self.cfg.GetMasterCandidateStats()
2373 3e83dd48 Iustin Pop
      if num_candidates <= cp_size:
2374 3e83dd48 Iustin Pop
        msg = ("Not enough master candidates (desired"
2375 3e83dd48 Iustin Pop
               " %d, new value will be %d)" % (cp_size, num_candidates-1))
2376 3a5ba66a Iustin Pop
        if self.op.force:
2377 3e83dd48 Iustin Pop
          self.LogWarning(msg)
2378 3e83dd48 Iustin Pop
        else:
2379 3e83dd48 Iustin Pop
          raise errors.OpPrereqError(msg)
2380 3e83dd48 Iustin Pop
2381 c9d443ea Iustin Pop
    if (self.op.master_candidate == True and
2382 c9d443ea Iustin Pop
        ((node.offline and not self.op.offline == False) or
2383 c9d443ea Iustin Pop
         (node.drained and not self.op.drained == False))):
2384 c9d443ea Iustin Pop
      raise errors.OpPrereqError("Node '%s' is offline or drained, can't set"
2385 949bdabe Iustin Pop
                                 " to master_candidate" % node.name)
2386 3a5ba66a Iustin Pop
2387 b31c8676 Iustin Pop
    return
2388 b31c8676 Iustin Pop
2389 b31c8676 Iustin Pop
  def Exec(self, feedback_fn):
2390 b31c8676 Iustin Pop
    """Modifies a node.
2391 b31c8676 Iustin Pop

2392 b31c8676 Iustin Pop
    """
2393 3a5ba66a Iustin Pop
    node = self.node
2394 b31c8676 Iustin Pop
2395 b31c8676 Iustin Pop
    result = []
2396 c9d443ea Iustin Pop
    changed_mc = False
2397 b31c8676 Iustin Pop
2398 3a5ba66a Iustin Pop
    if self.op.offline is not None:
2399 3a5ba66a Iustin Pop
      node.offline = self.op.offline
2400 3a5ba66a Iustin Pop
      result.append(("offline", str(self.op.offline)))
2401 c9d443ea Iustin Pop
      if self.op.offline == True:
2402 c9d443ea Iustin Pop
        if node.master_candidate:
2403 c9d443ea Iustin Pop
          node.master_candidate = False
2404 c9d443ea Iustin Pop
          changed_mc = True
2405 c9d443ea Iustin Pop
          result.append(("master_candidate", "auto-demotion due to offline"))
2406 c9d443ea Iustin Pop
        if node.drained:
2407 c9d443ea Iustin Pop
          node.drained = False
2408 c9d443ea Iustin Pop
          result.append(("drained", "clear drained status due to offline"))
2409 3a5ba66a Iustin Pop
2410 b31c8676 Iustin Pop
    if self.op.master_candidate is not None:
2411 b31c8676 Iustin Pop
      node.master_candidate = self.op.master_candidate
2412 c9d443ea Iustin Pop
      changed_mc = True
2413 b31c8676 Iustin Pop
      result.append(("master_candidate", str(self.op.master_candidate)))
2414 56aa9fd5 Iustin Pop
      if self.op.master_candidate == False:
2415 56aa9fd5 Iustin Pop
        rrc = self.rpc.call_node_demote_from_mc(node.name)
2416 0959c824 Iustin Pop
        msg = rrc.RemoteFailMsg()
2417 0959c824 Iustin Pop
        if msg:
2418 0959c824 Iustin Pop
          self.LogWarning("Node failed to demote itself: %s" % msg)
2419 b31c8676 Iustin Pop
2420 c9d443ea Iustin Pop
    if self.op.drained is not None:
2421 c9d443ea Iustin Pop
      node.drained = self.op.drained
2422 82e12743 Iustin Pop
      result.append(("drained", str(self.op.drained)))
2423 c9d443ea Iustin Pop
      if self.op.drained == True:
2424 c9d443ea Iustin Pop
        if node.master_candidate:
2425 c9d443ea Iustin Pop
          node.master_candidate = False
2426 c9d443ea Iustin Pop
          changed_mc = True
2427 c9d443ea Iustin Pop
          result.append(("master_candidate", "auto-demotion due to drain"))
2428 c9d443ea Iustin Pop
        if node.offline:
2429 c9d443ea Iustin Pop
          node.offline = False
2430 c9d443ea Iustin Pop
          result.append(("offline", "clear offline status due to drain"))
2431 c9d443ea Iustin Pop
2432 b31c8676 Iustin Pop
    # this will trigger configuration file update, if needed
2433 b31c8676 Iustin Pop
    self.cfg.Update(node)
2434 b31c8676 Iustin Pop
    # this will trigger job queue propagation or cleanup
2435 c9d443ea Iustin Pop
    if changed_mc:
2436 3a26773f Iustin Pop
      self.context.ReaddNode(node)
2437 b31c8676 Iustin Pop
2438 b31c8676 Iustin Pop
    return result
2439 b31c8676 Iustin Pop
2440 b31c8676 Iustin Pop
2441 f5118ade Iustin Pop
class LUPowercycleNode(NoHooksLU):
2442 f5118ade Iustin Pop
  """Powercycles a node.
2443 f5118ade Iustin Pop

2444 f5118ade Iustin Pop
  """
2445 f5118ade Iustin Pop
  _OP_REQP = ["node_name", "force"]
2446 f5118ade Iustin Pop
  REQ_BGL = False
2447 f5118ade Iustin Pop
2448 f5118ade Iustin Pop
  def CheckArguments(self):
2449 f5118ade Iustin Pop
    node_name = self.cfg.ExpandNodeName(self.op.node_name)
2450 f5118ade Iustin Pop
    if node_name is None:
2451 f5118ade Iustin Pop
      raise errors.OpPrereqError("Invalid node name '%s'" % self.op.node_name)
2452 f5118ade Iustin Pop
    self.op.node_name = node_name
2453 f5118ade Iustin Pop
    if node_name == self.cfg.GetMasterNode() and not self.op.force:
2454 f5118ade Iustin Pop
      raise errors.OpPrereqError("The node is the master and the force"
2455 f5118ade Iustin Pop
                                 " parameter was not set")
2456 f5118ade Iustin Pop
2457 f5118ade Iustin Pop
  def ExpandNames(self):
2458 f5118ade Iustin Pop
    """Locking for PowercycleNode.
2459 f5118ade Iustin Pop

2460 f5118ade Iustin Pop
    This is a last-resource option and shouldn't block on other
2461 f5118ade Iustin Pop
    jobs. Therefore, we grab no locks.
2462 f5118ade Iustin Pop

2463 f5118ade Iustin Pop
    """
2464 f5118ade Iustin Pop
    self.needed_locks = {}
2465 f5118ade Iustin Pop
2466 f5118ade Iustin Pop
  def CheckPrereq(self):
2467 f5118ade Iustin Pop
    """Check prerequisites.
2468 f5118ade Iustin Pop

2469 f5118ade Iustin Pop
    This LU has no prereqs.
2470 f5118ade Iustin Pop

2471 f5118ade Iustin Pop
    """
2472 f5118ade Iustin Pop
    pass
2473 f5118ade Iustin Pop
2474 f5118ade Iustin Pop
  def Exec(self, feedback_fn):
2475 f5118ade Iustin Pop
    """Reboots a node.
2476 f5118ade Iustin Pop

2477 f5118ade Iustin Pop
    """
2478 f5118ade Iustin Pop
    result = self.rpc.call_node_powercycle(self.op.node_name,
2479 f5118ade Iustin Pop
                                           self.cfg.GetHypervisorType())
2480 f5118ade Iustin Pop
    msg = result.RemoteFailMsg()
2481 f5118ade Iustin Pop
    if msg:
2482 f5118ade Iustin Pop
      raise errors.OpExecError("Failed to schedule the reboot: %s" % msg)
2483 f5118ade Iustin Pop
    return result.payload
2484 f5118ade Iustin Pop
2485 f5118ade Iustin Pop
2486 a8083063 Iustin Pop
class LUQueryClusterInfo(NoHooksLU):
2487 a8083063 Iustin Pop
  """Query cluster configuration.
2488 a8083063 Iustin Pop

2489 a8083063 Iustin Pop
  """
2490 a8083063 Iustin Pop
  _OP_REQP = []
2491 642339cf Guido Trotter
  REQ_BGL = False
2492 642339cf Guido Trotter
2493 642339cf Guido Trotter
  def ExpandNames(self):
2494 642339cf Guido Trotter
    self.needed_locks = {}
2495 a8083063 Iustin Pop
2496 a8083063 Iustin Pop
  def CheckPrereq(self):
2497 a8083063 Iustin Pop
    """No prerequsites needed for this LU.
2498 a8083063 Iustin Pop

2499 a8083063 Iustin Pop
    """
2500 a8083063 Iustin Pop
    pass
2501 a8083063 Iustin Pop
2502 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
2503 a8083063 Iustin Pop
    """Return cluster config.
2504 a8083063 Iustin Pop

2505 a8083063 Iustin Pop
    """
2506 469f88e1 Iustin Pop
    cluster = self.cfg.GetClusterInfo()
2507 a8083063 Iustin Pop
    result = {
2508 a8083063 Iustin Pop
      "software_version": constants.RELEASE_VERSION,
2509 a8083063 Iustin Pop
      "protocol_version": constants.PROTOCOL_VERSION,
2510 a8083063 Iustin Pop
      "config_version": constants.CONFIG_VERSION,
2511 a8083063 Iustin Pop
      "os_api_version": constants.OS_API_VERSION,
2512 a8083063 Iustin Pop
      "export_version": constants.EXPORT_VERSION,
2513 a8083063 Iustin Pop
      "architecture": (platform.architecture()[0], platform.machine()),
2514 469f88e1 Iustin Pop
      "name": cluster.cluster_name,
2515 469f88e1 Iustin Pop
      "master": cluster.master_node,
2516 02691904 Alexander Schreiber
      "default_hypervisor": cluster.default_hypervisor,
2517 469f88e1 Iustin Pop
      "enabled_hypervisors": cluster.enabled_hypervisors,
2518 7a735d6a Guido Trotter
      "hvparams": dict([(hypervisor, cluster.hvparams[hypervisor])
2519 7a735d6a Guido Trotter
                        for hypervisor in cluster.enabled_hypervisors]),
2520 469f88e1 Iustin Pop
      "beparams": cluster.beparams,
2521 1094acda Guido Trotter
      "nicparams": cluster.nicparams,
2522 4b7735f9 Iustin Pop
      "candidate_pool_size": cluster.candidate_pool_size,
2523 7a56b411 Guido Trotter
      "default_bridge": cluster.default_bridge,
2524 7a56b411 Guido Trotter
      "master_netdev": cluster.master_netdev,
2525 7a56b411 Guido Trotter
      "volume_group_name": cluster.volume_group_name,
2526 7a56b411 Guido Trotter
      "file_storage_dir": cluster.file_storage_dir,
2527 a8083063 Iustin Pop
      }
2528 a8083063 Iustin Pop
2529 a8083063 Iustin Pop
    return result
2530 a8083063 Iustin Pop
2531 a8083063 Iustin Pop
2532 ae5849b5 Michael Hanselmann
class LUQueryConfigValues(NoHooksLU):
2533 ae5849b5 Michael Hanselmann
  """Return configuration values.
2534 a8083063 Iustin Pop

2535 a8083063 Iustin Pop
  """
2536 a8083063 Iustin Pop
  _OP_REQP = []
2537 642339cf Guido Trotter
  REQ_BGL = False
2538 a2d2e1a7 Iustin Pop
  _FIELDS_DYNAMIC = utils.FieldSet()
2539 a2d2e1a7 Iustin Pop
  _FIELDS_STATIC = utils.FieldSet("cluster_name", "master_node", "drain_flag")
2540 642339cf Guido Trotter
2541 642339cf Guido Trotter
  def ExpandNames(self):
2542 642339cf Guido Trotter
    self.needed_locks = {}
2543 a8083063 Iustin Pop
2544 31bf511f Iustin Pop
    _CheckOutputFields(static=self._FIELDS_STATIC,
2545 31bf511f Iustin Pop
                       dynamic=self._FIELDS_DYNAMIC,
2546 ae5849b5 Michael Hanselmann
                       selected=self.op.output_fields)
2547 ae5849b5 Michael Hanselmann
2548 a8083063 Iustin Pop
  def CheckPrereq(self):
2549 a8083063 Iustin Pop
    """No prerequisites.
2550 a8083063 Iustin Pop

2551 a8083063 Iustin Pop
    """
2552 a8083063 Iustin Pop
    pass
2553 a8083063 Iustin Pop
2554 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
2555 a8083063 Iustin Pop
    """Dump a representation of the cluster config to the standard output.
2556 a8083063 Iustin Pop

2557 a8083063 Iustin Pop
    """
2558 ae5849b5 Michael Hanselmann
    values = []
2559 ae5849b5 Michael Hanselmann
    for field in self.op.output_fields:
2560 ae5849b5 Michael Hanselmann
      if field == "cluster_name":
2561 3ccafd0e Iustin Pop
        entry = self.cfg.GetClusterName()
2562 ae5849b5 Michael Hanselmann
      elif field == "master_node":
2563 3ccafd0e Iustin Pop
        entry = self.cfg.GetMasterNode()
2564 3ccafd0e Iustin Pop
      elif field == "drain_flag":
2565 3ccafd0e Iustin Pop
        entry = os.path.exists(constants.JOB_QUEUE_DRAIN_FILE)
2566 ae5849b5 Michael Hanselmann
      else:
2567 ae5849b5 Michael Hanselmann
        raise errors.ParameterError(field)
2568 3ccafd0e Iustin Pop
      values.append(entry)
2569 ae5849b5 Michael Hanselmann
    return values
2570 a8083063 Iustin Pop
2571 a8083063 Iustin Pop
2572 a8083063 Iustin Pop
class LUActivateInstanceDisks(NoHooksLU):
2573 a8083063 Iustin Pop
  """Bring up an instance's disks.
2574 a8083063 Iustin Pop

2575 a8083063 Iustin Pop
  """
2576 a8083063 Iustin Pop
  _OP_REQP = ["instance_name"]
2577 f22a8ba3 Guido Trotter
  REQ_BGL = False
2578 f22a8ba3 Guido Trotter
2579 f22a8ba3 Guido Trotter
  def ExpandNames(self):
2580 f22a8ba3 Guido Trotter
    self._ExpandAndLockInstance()
2581 f22a8ba3 Guido Trotter
    self.needed_locks[locking.LEVEL_NODE] = []
2582 f22a8ba3 Guido Trotter
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
2583 f22a8ba3 Guido Trotter
2584 f22a8ba3 Guido Trotter
  def DeclareLocks(self, level):
2585 f22a8ba3 Guido Trotter
    if level == locking.LEVEL_NODE:
2586 f22a8ba3 Guido Trotter
      self._LockInstancesNodes()
2587 a8083063 Iustin Pop
2588 a8083063 Iustin Pop
  def CheckPrereq(self):
2589 a8083063 Iustin Pop
    """Check prerequisites.
2590 a8083063 Iustin Pop

2591 a8083063 Iustin Pop
    This checks that the instance is in the cluster.
2592 a8083063 Iustin Pop

2593 a8083063 Iustin Pop
    """
2594 f22a8ba3 Guido Trotter
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_name)
2595 f22a8ba3 Guido Trotter
    assert self.instance is not None, \
2596 f22a8ba3 Guido Trotter
      "Cannot retrieve locked instance %s" % self.op.instance_name
2597 43017d26 Iustin Pop
    _CheckNodeOnline(self, self.instance.primary_node)
2598 a8083063 Iustin Pop
2599 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
2600 a8083063 Iustin Pop
    """Activate the disks.
2601 a8083063 Iustin Pop

2602 a8083063 Iustin Pop
    """
2603 b9bddb6b Iustin Pop
    disks_ok, disks_info = _AssembleInstanceDisks(self, self.instance)
2604 a8083063 Iustin Pop
    if not disks_ok:
2605 3ecf6786 Iustin Pop
      raise errors.OpExecError("Cannot activate block devices")
2606 a8083063 Iustin Pop
2607 a8083063 Iustin Pop
    return disks_info
2608 a8083063 Iustin Pop
2609 a8083063 Iustin Pop
2610 b9bddb6b Iustin Pop
def _AssembleInstanceDisks(lu, instance, ignore_secondaries=False):
2611 a8083063 Iustin Pop
  """Prepare the block devices for an instance.
2612 a8083063 Iustin Pop

2613 a8083063 Iustin Pop
  This sets up the block devices on all nodes.
2614 a8083063 Iustin Pop

2615 e4376078 Iustin Pop
  @type lu: L{LogicalUnit}
2616 e4376078 Iustin Pop
  @param lu: the logical unit on whose behalf we execute
2617 e4376078 Iustin Pop
  @type instance: L{objects.Instance}
2618 e4376078 Iustin Pop
  @param instance: the instance for whose disks we assemble
2619 e4376078 Iustin Pop
  @type ignore_secondaries: boolean
2620 e4376078 Iustin Pop
  @param ignore_secondaries: if true, errors on secondary nodes
2621 e4376078 Iustin Pop
      won't result in an error return from the function
2622 e4376078 Iustin Pop
  @return: False if the operation failed, otherwise a list of
2623 e4376078 Iustin Pop
      (host, instance_visible_name, node_visible_name)
2624 e4376078 Iustin Pop
      with the mapping from node devices to instance devices
2625 a8083063 Iustin Pop

2626 a8083063 Iustin Pop
  """
2627 a8083063 Iustin Pop
  device_info = []
2628 a8083063 Iustin Pop
  disks_ok = True
2629 fdbd668d Iustin Pop
  iname = instance.name
2630 fdbd668d Iustin Pop
  # With the two passes mechanism we try to reduce the window of
2631 fdbd668d Iustin Pop
  # opportunity for the race condition of switching DRBD to primary
2632 fdbd668d Iustin Pop
  # before handshaking occured, but we do not eliminate it
2633 fdbd668d Iustin Pop
2634 fdbd668d Iustin Pop
  # The proper fix would be to wait (with some limits) until the
2635 fdbd668d Iustin Pop
  # connection has been made and drbd transitions from WFConnection
2636 fdbd668d Iustin Pop
  # into any other network-connected state (Connected, SyncTarget,
2637 fdbd668d Iustin Pop
  # SyncSource, etc.)
2638 fdbd668d Iustin Pop
2639 fdbd668d Iustin Pop
  # 1st pass, assemble on all nodes in secondary mode
2640 a8083063 Iustin Pop
  for inst_disk in instance.disks:
2641 a8083063 Iustin Pop
    for node, node_disk in inst_disk.ComputeNodeTree(instance.primary_node):
2642 b9bddb6b Iustin Pop
      lu.cfg.SetDiskID(node_disk, node)
2643 72737a7f Iustin Pop
      result = lu.rpc.call_blockdev_assemble(node, node_disk, iname, False)
2644 53c14ef1 Iustin Pop
      msg = result.RemoteFailMsg()
2645 53c14ef1 Iustin Pop
      if msg:
2646 86d9d3bb Iustin Pop
        lu.proc.LogWarning("Could not prepare block device %s on node %s"
2647 53c14ef1 Iustin Pop
                           " (is_primary=False, pass=1): %s",
2648 53c14ef1 Iustin Pop
                           inst_disk.iv_name, node, msg)
2649 fdbd668d Iustin Pop
        if not ignore_secondaries:
2650 a8083063 Iustin Pop
          disks_ok = False
2651 fdbd668d Iustin Pop
2652 fdbd668d Iustin Pop
  # FIXME: race condition on drbd migration to primary
2653 fdbd668d Iustin Pop
2654 fdbd668d Iustin Pop
  # 2nd pass, do only the primary node
2655 fdbd668d Iustin Pop
  for inst_disk in instance.disks:
2656 fdbd668d Iustin Pop
    for node, node_disk in inst_disk.ComputeNodeTree(instance.primary_node):
2657 fdbd668d Iustin Pop
      if node != instance.primary_node:
2658 fdbd668d Iustin Pop
        continue
2659 b9bddb6b Iustin Pop
      lu.cfg.SetDiskID(node_disk, node)
2660 72737a7f Iustin Pop
      result = lu.rpc.call_blockdev_assemble(node, node_disk, iname, True)
2661 53c14ef1 Iustin Pop
      msg = result.RemoteFailMsg()
2662 53c14ef1 Iustin Pop
      if msg:
2663 86d9d3bb Iustin Pop
        lu.proc.LogWarning("Could not prepare block device %s on node %s"
2664 53c14ef1 Iustin Pop
                           " (is_primary=True, pass=2): %s",
2665 53c14ef1 Iustin Pop
                           inst_disk.iv_name, node, msg)
2666 fdbd668d Iustin Pop
        disks_ok = False
2667 1dff8e07 Iustin Pop
    device_info.append((instance.primary_node, inst_disk.iv_name,
2668 1dff8e07 Iustin Pop
                        result.payload))
2669 a8083063 Iustin Pop
2670 b352ab5b Iustin Pop
  # leave the disks configured for the primary node
2671 b352ab5b Iustin Pop
  # this is a workaround that would be fixed better by
2672 b352ab5b Iustin Pop
  # improving the logical/physical id handling
2673 b352ab5b Iustin Pop
  for disk in instance.disks:
2674 b9bddb6b Iustin Pop
    lu.cfg.SetDiskID(disk, instance.primary_node)
2675 b352ab5b Iustin Pop
2676 a8083063 Iustin Pop
  return disks_ok, device_info
2677 a8083063 Iustin Pop
2678 a8083063 Iustin Pop
2679 b9bddb6b Iustin Pop
def _StartInstanceDisks(lu, instance, force):
2680 3ecf6786 Iustin Pop
  """Start the disks of an instance.
2681 3ecf6786 Iustin Pop

2682 3ecf6786 Iustin Pop
  """
2683 b9bddb6b Iustin Pop
  disks_ok, dummy = _AssembleInstanceDisks(lu, instance,
2684 fe7b0351 Michael Hanselmann
                                           ignore_secondaries=force)
2685 fe7b0351 Michael Hanselmann
  if not disks_ok:
2686 b9bddb6b Iustin Pop
    _ShutdownInstanceDisks(lu, instance)
2687 fe7b0351 Michael Hanselmann
    if force is not None and not force:
2688 86d9d3bb Iustin Pop
      lu.proc.LogWarning("", hint="If the message above refers to a"
2689 86d9d3bb Iustin Pop
                         " secondary node,"
2690 86d9d3bb Iustin Pop
                         " you can retry the operation using '--force'.")
2691 3ecf6786 Iustin Pop
    raise errors.OpExecError("Disk consistency error")
2692 fe7b0351 Michael Hanselmann
2693 fe7b0351 Michael Hanselmann
2694 a8083063 Iustin Pop
class LUDeactivateInstanceDisks(NoHooksLU):
2695 a8083063 Iustin Pop
  """Shutdown an instance's disks.
2696 a8083063 Iustin Pop

2697 a8083063 Iustin Pop
  """
2698 a8083063 Iustin Pop
  _OP_REQP = ["instance_name"]
2699 f22a8ba3 Guido Trotter
  REQ_BGL = False
2700 f22a8ba3 Guido Trotter
2701 f22a8ba3 Guido Trotter
  def ExpandNames(self):
2702 f22a8ba3 Guido Trotter
    self._ExpandAndLockInstance()
2703 f22a8ba3 Guido Trotter
    self.needed_locks[locking.LEVEL_NODE] = []
2704 f22a8ba3 Guido Trotter
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
2705 f22a8ba3 Guido Trotter
2706 f22a8ba3 Guido Trotter
  def DeclareLocks(self, level):
2707 f22a8ba3 Guido Trotter
    if level == locking.LEVEL_NODE:
2708 f22a8ba3 Guido Trotter
      self._LockInstancesNodes()
2709 a8083063 Iustin Pop
2710 a8083063 Iustin Pop
  def CheckPrereq(self):
2711 a8083063 Iustin Pop
    """Check prerequisites.
2712 a8083063 Iustin Pop

2713 a8083063 Iustin Pop
    This checks that the instance is in the cluster.
2714 a8083063 Iustin Pop

2715 a8083063 Iustin Pop
    """
2716 f22a8ba3 Guido Trotter
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_name)
2717 f22a8ba3 Guido Trotter
    assert self.instance is not None, \
2718 f22a8ba3 Guido Trotter
      "Cannot retrieve locked instance %s" % self.op.instance_name
2719 a8083063 Iustin Pop
2720 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
2721 a8083063 Iustin Pop
    """Deactivate the disks
2722 a8083063 Iustin Pop

2723 a8083063 Iustin Pop
    """
2724 a8083063 Iustin Pop
    instance = self.instance
2725 b9bddb6b Iustin Pop
    _SafeShutdownInstanceDisks(self, instance)
2726 a8083063 Iustin Pop
2727 a8083063 Iustin Pop
2728 b9bddb6b Iustin Pop
def _SafeShutdownInstanceDisks(lu, instance):
2729 155d6c75 Guido Trotter
  """Shutdown block devices of an instance.
2730 155d6c75 Guido Trotter

2731 155d6c75 Guido Trotter
  This function checks if an instance is running, before calling
2732 155d6c75 Guido Trotter
  _ShutdownInstanceDisks.
2733 155d6c75 Guido Trotter

2734 155d6c75 Guido Trotter
  """
2735 72737a7f Iustin Pop
  ins_l = lu.rpc.call_instance_list([instance.primary_node],
2736 72737a7f Iustin Pop
                                      [instance.hypervisor])
2737 155d6c75 Guido Trotter
  ins_l = ins_l[instance.primary_node]
2738 781de953 Iustin Pop
  if ins_l.failed or not isinstance(ins_l.data, list):
2739 155d6c75 Guido Trotter
    raise errors.OpExecError("Can't contact node '%s'" %
2740 155d6c75 Guido Trotter
                             instance.primary_node)
2741 155d6c75 Guido Trotter
2742 781de953 Iustin Pop
  if instance.name in ins_l.data:
2743 155d6c75 Guido Trotter
    raise errors.OpExecError("Instance is running, can't shutdown"
2744 155d6c75 Guido Trotter
                             " block devices.")
2745 155d6c75 Guido Trotter
2746 b9bddb6b Iustin Pop
  _ShutdownInstanceDisks(lu, instance)
2747 a8083063 Iustin Pop
2748 a8083063 Iustin Pop
2749 b9bddb6b Iustin Pop
def _ShutdownInstanceDisks(lu, instance, ignore_primary=False):
2750 a8083063 Iustin Pop
  """Shutdown block devices of an instance.
2751 a8083063 Iustin Pop

2752 a8083063 Iustin Pop
  This does the shutdown on all nodes of the instance.
2753 a8083063 Iustin Pop

2754 a8083063 Iustin Pop
  If the ignore_primary is false, errors on the primary node are
2755 a8083063 Iustin Pop
  ignored.
2756 a8083063 Iustin Pop

2757 a8083063 Iustin Pop
  """
2758 cacfd1fd Iustin Pop
  all_result = True
2759 a8083063 Iustin Pop
  for disk in instance.disks:
2760 a8083063 Iustin Pop
    for node, top_disk in disk.ComputeNodeTree(instance.primary_node):
2761 b9bddb6b Iustin Pop
      lu.cfg.SetDiskID(top_disk, node)
2762 781de953 Iustin Pop
      result = lu.rpc.call_blockdev_shutdown(node, top_disk)
2763 cacfd1fd Iustin Pop
      msg = result.RemoteFailMsg()
2764 cacfd1fd Iustin Pop
      if msg:
2765 cacfd1fd Iustin Pop
        lu.LogWarning("Could not shutdown block device %s on node %s: %s",
2766 cacfd1fd Iustin Pop
                      disk.iv_name, node, msg)
2767 a8083063 Iustin Pop
        if not ignore_primary or node != instance.primary_node:
2768 cacfd1fd Iustin Pop
          all_result = False
2769 cacfd1fd Iustin Pop
  return all_result
2770 a8083063 Iustin Pop
2771 a8083063 Iustin Pop
2772 9ca87a96 Iustin Pop
def _CheckNodeFreeMemory(lu, node, reason, requested, hypervisor_name):
2773 d4f16fd9 Iustin Pop
  """Checks if a node has enough free memory.
2774 d4f16fd9 Iustin Pop

2775 d4f16fd9 Iustin Pop
  This function check if a given node has the needed amount of free
2776 d4f16fd9 Iustin Pop
  memory. In case the node has less memory or we cannot get the
2777 d4f16fd9 Iustin Pop
  information from the node, this function raise an OpPrereqError
2778 d4f16fd9 Iustin Pop
  exception.
2779 d4f16fd9 Iustin Pop

2780 b9bddb6b Iustin Pop
  @type lu: C{LogicalUnit}
2781 b9bddb6b Iustin Pop
  @param lu: a logical unit from which we get configuration data
2782 e69d05fd Iustin Pop
  @type node: C{str}
2783 e69d05fd Iustin Pop
  @param node: the node to check
2784 e69d05fd Iustin Pop
  @type reason: C{str}
2785 e69d05fd Iustin Pop
  @param reason: string to use in the error message
2786 e69d05fd Iustin Pop
  @type requested: C{int}
2787 e69d05fd Iustin Pop
  @param requested: the amount of memory in MiB to check for
2788 9ca87a96 Iustin Pop
  @type hypervisor_name: C{str}
2789 9ca87a96 Iustin Pop
  @param hypervisor_name: the hypervisor to ask for memory stats
2790 e69d05fd Iustin Pop
  @raise errors.OpPrereqError: if the node doesn't have enough memory, or
2791 e69d05fd Iustin Pop
      we cannot check the node
2792 d4f16fd9 Iustin Pop

2793 d4f16fd9 Iustin Pop
  """
2794 9ca87a96 Iustin Pop
  nodeinfo = lu.rpc.call_node_info([node], lu.cfg.GetVGName(), hypervisor_name)
2795 781de953 Iustin Pop
  nodeinfo[node].Raise()
2796 781de953 Iustin Pop
  free_mem = nodeinfo[node].data.get('memory_free')
2797 d4f16fd9 Iustin Pop
  if not isinstance(free_mem, int):
2798 d4f16fd9 Iustin Pop
    raise errors.OpPrereqError("Can't compute free memory on node %s, result"
2799 d4f16fd9 Iustin Pop
                             " was '%s'" % (node, free_mem))
2800 d4f16fd9 Iustin Pop
  if requested > free_mem:
2801 d4f16fd9 Iustin Pop
    raise errors.OpPrereqError("Not enough memory on node %s for %s:"
2802 d4f16fd9 Iustin Pop
                             " needed %s MiB, available %s MiB" %
2803 d4f16fd9 Iustin Pop
                             (node, reason, requested, free_mem))
2804 d4f16fd9 Iustin Pop
2805 d4f16fd9 Iustin Pop
2806 a8083063 Iustin Pop
class LUStartupInstance(LogicalUnit):
2807 a8083063 Iustin Pop
  """Starts an instance.
2808 a8083063 Iustin Pop

2809 a8083063 Iustin Pop
  """
2810 a8083063 Iustin Pop
  HPATH = "instance-start"
2811 a8083063 Iustin Pop
  HTYPE = constants.HTYPE_INSTANCE
2812 a8083063 Iustin Pop
  _OP_REQP = ["instance_name", "force"]
2813 e873317a Guido Trotter
  REQ_BGL = False
2814 e873317a Guido Trotter
2815 e873317a Guido Trotter
  def ExpandNames(self):
2816 e873317a Guido Trotter
    self._ExpandAndLockInstance()
2817 a8083063 Iustin Pop
2818 a8083063 Iustin Pop
  def BuildHooksEnv(self):
2819 a8083063 Iustin Pop
    """Build hooks env.
2820 a8083063 Iustin Pop

2821 a8083063 Iustin Pop
    This runs on master, primary and secondary nodes of the instance.
2822 a8083063 Iustin Pop

2823 a8083063 Iustin Pop
    """
2824 a8083063 Iustin Pop
    env = {
2825 a8083063 Iustin Pop
      "FORCE": self.op.force,
2826 a8083063 Iustin Pop
      }
2827 338e51e8 Iustin Pop
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
2828 6b12959c Iustin Pop
    nl = [self.cfg.GetMasterNode()] + list(self.instance.all_nodes)
2829 a8083063 Iustin Pop
    return env, nl, nl
2830 a8083063 Iustin Pop
2831 a8083063 Iustin Pop
  def CheckPrereq(self):
2832 a8083063 Iustin Pop
    """Check prerequisites.
2833 a8083063 Iustin Pop

2834 a8083063 Iustin Pop
    This checks that the instance is in the cluster.
2835 a8083063 Iustin Pop

2836 a8083063 Iustin Pop
    """
2837 e873317a Guido Trotter
    self.instance = instance = self.cfg.GetInstanceInfo(self.op.instance_name)
2838 e873317a Guido Trotter
    assert self.instance is not None, \
2839 e873317a Guido Trotter
      "Cannot retrieve locked instance %s" % self.op.instance_name
2840 a8083063 Iustin Pop
2841 d04aaa2f Iustin Pop
    # extra beparams
2842 d04aaa2f Iustin Pop
    self.beparams = getattr(self.op, "beparams", {})
2843 d04aaa2f Iustin Pop
    if self.beparams:
2844 d04aaa2f Iustin Pop
      if not isinstance(self.beparams, dict):
2845 d04aaa2f Iustin Pop
        raise errors.OpPrereqError("Invalid beparams passed: %s, expected"
2846 d04aaa2f Iustin Pop
                                   " dict" % (type(self.beparams), ))
2847 d04aaa2f Iustin Pop
      # fill the beparams dict
2848 d04aaa2f Iustin Pop
      utils.ForceDictType(self.beparams, constants.BES_PARAMETER_TYPES)
2849 d04aaa2f Iustin Pop
      self.op.beparams = self.beparams
2850 d04aaa2f Iustin Pop
2851 d04aaa2f Iustin Pop
    # extra hvparams
2852 d04aaa2f Iustin Pop
    self.hvparams = getattr(self.op, "hvparams", {})
2853 d04aaa2f Iustin Pop
    if self.hvparams:
2854 d04aaa2f Iustin Pop
      if not isinstance(self.hvparams, dict):
2855 d04aaa2f Iustin Pop
        raise errors.OpPrereqError("Invalid hvparams passed: %s, expected"
2856 d04aaa2f Iustin Pop
                                   " dict" % (type(self.hvparams), ))
2857 d04aaa2f Iustin Pop
2858 d04aaa2f Iustin Pop
      # check hypervisor parameter syntax (locally)
2859 d04aaa2f Iustin Pop
      cluster = self.cfg.GetClusterInfo()
2860 d04aaa2f Iustin Pop
      utils.ForceDictType(self.hvparams, constants.HVS_PARAMETER_TYPES)
2861 abe609b2 Guido Trotter
      filled_hvp = objects.FillDict(cluster.hvparams[instance.hypervisor],
2862 d04aaa2f Iustin Pop
                                    instance.hvparams)
2863 d04aaa2f Iustin Pop
      filled_hvp.update(self.hvparams)
2864 d04aaa2f Iustin Pop
      hv_type = hypervisor.GetHypervisor(instance.hypervisor)
2865 d04aaa2f Iustin Pop
      hv_type.CheckParameterSyntax(filled_hvp)
2866 d04aaa2f Iustin Pop
      _CheckHVParams(self, instance.all_nodes, instance.hypervisor, filled_hvp)
2867 d04aaa2f Iustin Pop
      self.op.hvparams = self.hvparams
2868 d04aaa2f Iustin Pop
2869 7527a8a4 Iustin Pop
    _CheckNodeOnline(self, instance.primary_node)
2870 7527a8a4 Iustin Pop
2871 338e51e8 Iustin Pop
    bep = self.cfg.GetClusterInfo().FillBE(instance)
2872 a8083063 Iustin Pop
    # check bridges existance
2873 b9bddb6b Iustin Pop
    _CheckInstanceBridgesExist(self, instance)
2874 a8083063 Iustin Pop
2875 f1926756 Guido Trotter
    remote_info = self.rpc.call_instance_info(instance.primary_node,
2876 f1926756 Guido Trotter
                                              instance.name,
2877 f1926756 Guido Trotter
                                              instance.hypervisor)
2878 f1926756 Guido Trotter
    remote_info.Raise()
2879 f1926756 Guido Trotter
    if not remote_info.data:
2880 f1926756 Guido Trotter
      _CheckNodeFreeMemory(self, instance.primary_node,
2881 f1926756 Guido Trotter
                           "starting instance %s" % instance.name,
2882 f1926756 Guido Trotter
                           bep[constants.BE_MEMORY], instance.hypervisor)
2883 d4f16fd9 Iustin Pop
2884 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
2885 a8083063 Iustin Pop
    """Start the instance.
2886 a8083063 Iustin Pop

2887 a8083063 Iustin Pop
    """
2888 a8083063 Iustin Pop
    instance = self.instance
2889 a8083063 Iustin Pop
    force = self.op.force
2890 a8083063 Iustin Pop
2891 fe482621 Iustin Pop
    self.cfg.MarkInstanceUp(instance.name)
2892 fe482621 Iustin Pop
2893 a8083063 Iustin Pop
    node_current = instance.primary_node
2894 a8083063 Iustin Pop
2895 b9bddb6b Iustin Pop
    _StartInstanceDisks(self, instance, force)
2896 a8083063 Iustin Pop
2897 d04aaa2f Iustin Pop
    result = self.rpc.call_instance_start(node_current, instance,
2898 d04aaa2f Iustin Pop
                                          self.hvparams, self.beparams)
2899 dd279568 Iustin Pop
    msg = result.RemoteFailMsg()
2900 dd279568 Iustin Pop
    if msg:
2901 b9bddb6b Iustin Pop
      _ShutdownInstanceDisks(self, instance)
2902 dd279568 Iustin Pop
      raise errors.OpExecError("Could not start instance: %s" % msg)
2903 a8083063 Iustin Pop
2904 a8083063 Iustin Pop
2905 bf6929a2 Alexander Schreiber
class LURebootInstance(LogicalUnit):
2906 bf6929a2 Alexander Schreiber
  """Reboot an instance.
2907 bf6929a2 Alexander Schreiber

2908 bf6929a2 Alexander Schreiber
  """
2909 bf6929a2 Alexander Schreiber
  HPATH = "instance-reboot"
2910 bf6929a2 Alexander Schreiber
  HTYPE = constants.HTYPE_INSTANCE
2911 bf6929a2 Alexander Schreiber
  _OP_REQP = ["instance_name", "ignore_secondaries", "reboot_type"]
2912 e873317a Guido Trotter
  REQ_BGL = False
2913 e873317a Guido Trotter
2914 e873317a Guido Trotter
  def ExpandNames(self):
2915 0fcc5db3 Guido Trotter
    if self.op.reboot_type not in [constants.INSTANCE_REBOOT_SOFT,
2916 0fcc5db3 Guido Trotter
                                   constants.INSTANCE_REBOOT_HARD,
2917 0fcc5db3 Guido Trotter
                                   constants.INSTANCE_REBOOT_FULL]:
2918 0fcc5db3 Guido Trotter
      raise errors.ParameterError("reboot type not in [%s, %s, %s]" %
2919 0fcc5db3 Guido Trotter
                                  (constants.INSTANCE_REBOOT_SOFT,
2920 0fcc5db3 Guido Trotter
                                   constants.INSTANCE_REBOOT_HARD,
2921 0fcc5db3 Guido Trotter
                                   constants.INSTANCE_REBOOT_FULL))
2922 e873317a Guido Trotter
    self._ExpandAndLockInstance()
2923 bf6929a2 Alexander Schreiber
2924 bf6929a2 Alexander Schreiber
  def BuildHooksEnv(self):
2925 bf6929a2 Alexander Schreiber
    """Build hooks env.
2926 bf6929a2 Alexander Schreiber

2927 bf6929a2 Alexander Schreiber
    This runs on master, primary and secondary nodes of the instance.
2928 bf6929a2 Alexander Schreiber

2929 bf6929a2 Alexander Schreiber
    """
2930 bf6929a2 Alexander Schreiber
    env = {
2931 bf6929a2 Alexander Schreiber
      "IGNORE_SECONDARIES": self.op.ignore_secondaries,
2932 2c2690c9 Iustin Pop
      "REBOOT_TYPE": self.op.reboot_type,
2933 bf6929a2 Alexander Schreiber
      }
2934 338e51e8 Iustin Pop
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
2935 6b12959c Iustin Pop
    nl = [self.cfg.GetMasterNode()] + list(self.instance.all_nodes)
2936 bf6929a2 Alexander Schreiber
    return env, nl, nl
2937 bf6929a2 Alexander Schreiber
2938 bf6929a2 Alexander Schreiber
  def CheckPrereq(self):
2939 bf6929a2 Alexander Schreiber
    """Check prerequisites.
2940 bf6929a2 Alexander Schreiber

2941 bf6929a2 Alexander Schreiber
    This checks that the instance is in the cluster.
2942 bf6929a2 Alexander Schreiber

2943 bf6929a2 Alexander Schreiber
    """
2944 e873317a Guido Trotter
    self.instance = instance = self.cfg.GetInstanceInfo(self.op.instance_name)
2945 e873317a Guido Trotter
    assert self.instance is not None, \
2946 e873317a Guido Trotter
      "Cannot retrieve locked instance %s" % self.op.instance_name
2947 bf6929a2 Alexander Schreiber
2948 7527a8a4 Iustin Pop
    _CheckNodeOnline(self, instance.primary_node)
2949 7527a8a4 Iustin Pop
2950 bf6929a2 Alexander Schreiber
    # check bridges existance
2951 b9bddb6b Iustin Pop
    _CheckInstanceBridgesExist(self, instance)
2952 bf6929a2 Alexander Schreiber
2953 bf6929a2 Alexander Schreiber
  def Exec(self, feedback_fn):
2954 bf6929a2 Alexander Schreiber
    """Reboot the instance.
2955 bf6929a2 Alexander Schreiber

2956 bf6929a2 Alexander Schreiber
    """
2957 bf6929a2 Alexander Schreiber
    instance = self.instance
2958 bf6929a2 Alexander Schreiber
    ignore_secondaries = self.op.ignore_secondaries
2959 bf6929a2 Alexander Schreiber
    reboot_type = self.op.reboot_type
2960 bf6929a2 Alexander Schreiber
2961 bf6929a2 Alexander Schreiber
    node_current = instance.primary_node
2962 bf6929a2 Alexander Schreiber
2963 bf6929a2 Alexander Schreiber
    if reboot_type in [constants.INSTANCE_REBOOT_SOFT,
2964 bf6929a2 Alexander Schreiber
                       constants.INSTANCE_REBOOT_HARD]:
2965 ae48ac32 Iustin Pop
      for disk in instance.disks:
2966 ae48ac32 Iustin Pop
        self.cfg.SetDiskID(disk, node_current)
2967 781de953 Iustin Pop
      result = self.rpc.call_instance_reboot(node_current, instance,
2968 07813a9e Iustin Pop
                                             reboot_type)
2969 489fcbe9 Iustin Pop
      msg = result.RemoteFailMsg()
2970 489fcbe9 Iustin Pop
      if msg:
2971 489fcbe9 Iustin Pop
        raise errors.OpExecError("Could not reboot instance: %s" % msg)
2972 bf6929a2 Alexander Schreiber
    else:
2973 1fae010f Iustin Pop
      result = self.rpc.call_instance_shutdown(node_current, instance)
2974 1fae010f Iustin Pop
      msg = result.RemoteFailMsg()
2975 1fae010f Iustin Pop
      if msg:
2976 1fae010f Iustin Pop
        raise errors.OpExecError("Could not shutdown instance for"
2977 1fae010f Iustin Pop
                                 " full reboot: %s" % msg)
2978 b9bddb6b Iustin Pop
      _ShutdownInstanceDisks(self, instance)
2979 b9bddb6b Iustin Pop
      _StartInstanceDisks(self, instance, ignore_secondaries)
2980 0eca8e0c Iustin Pop
      result = self.rpc.call_instance_start(node_current, instance, None, None)
2981 dd279568 Iustin Pop
      msg = result.RemoteFailMsg()
2982 dd279568 Iustin Pop
      if msg:
2983 b9bddb6b Iustin Pop
        _ShutdownInstanceDisks(self, instance)
2984 dd279568 Iustin Pop
        raise errors.OpExecError("Could not start instance for"
2985 dd279568 Iustin Pop
                                 " full reboot: %s" % msg)
2986 bf6929a2 Alexander Schreiber
2987 bf6929a2 Alexander Schreiber
    self.cfg.MarkInstanceUp(instance.name)
2988 bf6929a2 Alexander Schreiber
2989 bf6929a2 Alexander Schreiber
2990 a8083063 Iustin Pop
class LUShutdownInstance(LogicalUnit):
2991 a8083063 Iustin Pop
  """Shutdown an instance.
2992 a8083063 Iustin Pop

2993 a8083063 Iustin Pop
  """
2994 a8083063 Iustin Pop
  HPATH = "instance-stop"
2995 a8083063 Iustin Pop
  HTYPE = constants.HTYPE_INSTANCE
2996 a8083063 Iustin Pop
  _OP_REQP = ["instance_name"]
2997 e873317a Guido Trotter
  REQ_BGL = False
2998 e873317a Guido Trotter
2999 e873317a Guido Trotter
  def ExpandNames(self):
3000 e873317a Guido Trotter
    self._ExpandAndLockInstance()
3001 a8083063 Iustin Pop
3002 a8083063 Iustin Pop
  def BuildHooksEnv(self):
3003 a8083063 Iustin Pop
    """Build hooks env.
3004 a8083063 Iustin Pop

3005 a8083063 Iustin Pop
    This runs on master, primary and secondary nodes of the instance.
3006 a8083063 Iustin Pop

3007 a8083063 Iustin Pop
    """
3008 338e51e8 Iustin Pop
    env = _BuildInstanceHookEnvByObject(self, self.instance)
3009 6b12959c Iustin Pop
    nl = [self.cfg.GetMasterNode()] + list(self.instance.all_nodes)
3010 a8083063 Iustin Pop
    return env, nl, nl
3011 a8083063 Iustin Pop
3012 a8083063 Iustin Pop
  def CheckPrereq(self):
3013 a8083063 Iustin Pop
    """Check prerequisites.
3014 a8083063 Iustin Pop

3015 a8083063 Iustin Pop
    This checks that the instance is in the cluster.
3016 a8083063 Iustin Pop

3017 a8083063 Iustin Pop
    """
3018 e873317a Guido Trotter
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_name)
3019 e873317a Guido Trotter
    assert self.instance is not None, \
3020 e873317a Guido Trotter
      "Cannot retrieve locked instance %s" % self.op.instance_name
3021 43017d26 Iustin Pop
    _CheckNodeOnline(self, self.instance.primary_node)
3022 a8083063 Iustin Pop
3023 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
3024 a8083063 Iustin Pop
    """Shutdown the instance.
3025 a8083063 Iustin Pop

3026 a8083063 Iustin Pop
    """
3027 a8083063 Iustin Pop
    instance = self.instance
3028 a8083063 Iustin Pop
    node_current = instance.primary_node
3029 fe482621 Iustin Pop
    self.cfg.MarkInstanceDown(instance.name)
3030 781de953 Iustin Pop
    result = self.rpc.call_instance_shutdown(node_current, instance)
3031 1fae010f Iustin Pop
    msg = result.RemoteFailMsg()
3032 1fae010f Iustin Pop
    if msg:
3033 1fae010f Iustin Pop
      self.proc.LogWarning("Could not shutdown instance: %s" % msg)
3034 a8083063 Iustin Pop
3035 b9bddb6b Iustin Pop
    _ShutdownInstanceDisks(self, instance)
3036 a8083063 Iustin Pop
3037 a8083063 Iustin Pop
3038 fe7b0351 Michael Hanselmann
class LUReinstallInstance(LogicalUnit):
3039 fe7b0351 Michael Hanselmann
  """Reinstall an instance.
3040 fe7b0351 Michael Hanselmann

3041 fe7b0351 Michael Hanselmann
  """
3042 fe7b0351 Michael Hanselmann
  HPATH = "instance-reinstall"
3043 fe7b0351 Michael Hanselmann
  HTYPE = constants.HTYPE_INSTANCE
3044 fe7b0351 Michael Hanselmann
  _OP_REQP = ["instance_name"]
3045 4e0b4d2d Guido Trotter
  REQ_BGL = False
3046 4e0b4d2d Guido Trotter
3047 4e0b4d2d Guido Trotter
  def ExpandNames(self):
3048 4e0b4d2d Guido Trotter
    self._ExpandAndLockInstance()
3049 fe7b0351 Michael Hanselmann
3050 fe7b0351 Michael Hanselmann
  def BuildHooksEnv(self):
3051 fe7b0351 Michael Hanselmann
    """Build hooks env.
3052 fe7b0351 Michael Hanselmann

3053 fe7b0351 Michael Hanselmann
    This runs on master, primary and secondary nodes of the instance.
3054 fe7b0351 Michael Hanselmann

3055 fe7b0351 Michael Hanselmann
    """
3056 338e51e8 Iustin Pop
    env = _BuildInstanceHookEnvByObject(self, self.instance)
3057 6b12959c Iustin Pop
    nl = [self.cfg.GetMasterNode()] + list(self.instance.all_nodes)
3058 fe7b0351 Michael Hanselmann
    return env, nl, nl
3059 fe7b0351 Michael Hanselmann
3060 fe7b0351 Michael Hanselmann
  def CheckPrereq(self):
3061 fe7b0351 Michael Hanselmann
    """Check prerequisites.
3062 fe7b0351 Michael Hanselmann

3063 fe7b0351 Michael Hanselmann
    This checks that the instance is in the cluster and is not running.
3064 fe7b0351 Michael Hanselmann

3065 fe7b0351 Michael Hanselmann
    """
3066 4e0b4d2d Guido Trotter
    instance = self.cfg.GetInstanceInfo(self.op.instance_name)
3067 4e0b4d2d Guido Trotter
    assert instance is not None, \
3068 4e0b4d2d Guido Trotter
      "Cannot retrieve locked instance %s" % self.op.instance_name
3069 7527a8a4 Iustin Pop
    _CheckNodeOnline(self, instance.primary_node)
3070 4e0b4d2d Guido Trotter
3071 fe7b0351 Michael Hanselmann
    if instance.disk_template == constants.DT_DISKLESS:
3072 3ecf6786 Iustin Pop
      raise errors.OpPrereqError("Instance '%s' has no disks" %
3073 3ecf6786 Iustin Pop
                                 self.op.instance_name)
3074 0d68c45d Iustin Pop
    if instance.admin_up:
3075 3ecf6786 Iustin Pop
      raise errors.OpPrereqError("Instance '%s' is marked to be up" %
3076 3ecf6786 Iustin Pop
                                 self.op.instance_name)
3077 72737a7f Iustin Pop
    remote_info = self.rpc.call_instance_info(instance.primary_node,
3078 72737a7f Iustin Pop
                                              instance.name,
3079 72737a7f Iustin Pop
                                              instance.hypervisor)
3080 b4874c9e Guido Trotter
    remote_info.Raise()
3081 b4874c9e Guido Trotter
    if remote_info.data:
3082 3ecf6786 Iustin Pop
      raise errors.OpPrereqError("Instance '%s' is running on the node %s" %
3083 3ecf6786 Iustin Pop
                                 (self.op.instance_name,
3084 3ecf6786 Iustin Pop
                                  instance.primary_node))
3085 d0834de3 Michael Hanselmann
3086 d0834de3 Michael Hanselmann
    self.op.os_type = getattr(self.op, "os_type", None)
3087 d0834de3 Michael Hanselmann
    if self.op.os_type is not None:
3088 d0834de3 Michael Hanselmann
      # OS verification
3089 d0834de3 Michael Hanselmann
      pnode = self.cfg.GetNodeInfo(
3090 d0834de3 Michael Hanselmann
        self.cfg.ExpandNodeName(instance.primary_node))
3091 d0834de3 Michael Hanselmann
      if pnode is None:
3092 3ecf6786 Iustin Pop
        raise errors.OpPrereqError("Primary node '%s' is unknown" %
3093 3ecf6786 Iustin Pop
                                   self.op.pnode)
3094 781de953 Iustin Pop
      result = self.rpc.call_os_get(pnode.name, self.op.os_type)
3095 781de953 Iustin Pop
      result.Raise()
3096 781de953 Iustin Pop
      if not isinstance(result.data, objects.OS):
3097 3ecf6786 Iustin Pop
        raise errors.OpPrereqError("OS '%s' not in supported OS list for"
3098 3ecf6786 Iustin Pop
                                   " primary node"  % self.op.os_type)
3099 d0834de3 Michael Hanselmann
3100 fe7b0351 Michael Hanselmann
    self.instance = instance
3101 fe7b0351 Michael Hanselmann
3102 fe7b0351 Michael Hanselmann
  def Exec(self, feedback_fn):
3103 fe7b0351 Michael Hanselmann
    """Reinstall the instance.
3104 fe7b0351 Michael Hanselmann

3105 fe7b0351 Michael Hanselmann
    """
3106 fe7b0351 Michael Hanselmann
    inst = self.instance
3107 fe7b0351 Michael Hanselmann
3108 d0834de3 Michael Hanselmann
    if self.op.os_type is not None:
3109 d0834de3 Michael Hanselmann
      feedback_fn("Changing OS to '%s'..." % self.op.os_type)
3110 d0834de3 Michael Hanselmann
      inst.os = self.op.os_type
3111 97abc79f Iustin Pop
      self.cfg.Update(inst)
3112 d0834de3 Michael Hanselmann
3113 b9bddb6b Iustin Pop
    _StartInstanceDisks(self, inst, None)
3114 fe7b0351 Michael Hanselmann
    try:
3115 fe7b0351 Michael Hanselmann
      feedback_fn("Running the instance OS create scripts...")
3116 e557bae9 Guido Trotter
      result = self.rpc.call_instance_os_add(inst.primary_node, inst, True)
3117 20e01edd Iustin Pop
      msg = result.RemoteFailMsg()
3118 20e01edd Iustin Pop
      if msg:
3119 f4bc1f2c Michael Hanselmann
        raise errors.OpExecError("Could not install OS for instance %s"
3120 20e01edd Iustin Pop
                                 " on node %s: %s" %
3121 20e01edd Iustin Pop
                                 (inst.name, inst.primary_node, msg))
3122 fe7b0351 Michael Hanselmann
    finally:
3123 b9bddb6b Iustin Pop
      _ShutdownInstanceDisks(self, inst)
3124 fe7b0351 Michael Hanselmann
3125 fe7b0351 Michael Hanselmann
3126 decd5f45 Iustin Pop
class LURenameInstance(LogicalUnit):
3127 decd5f45 Iustin Pop
  """Rename an instance.
3128 decd5f45 Iustin Pop

3129 decd5f45 Iustin Pop
  """
3130 decd5f45 Iustin Pop
  HPATH = "instance-rename"
3131 decd5f45 Iustin Pop
  HTYPE = constants.HTYPE_INSTANCE
3132 decd5f45 Iustin Pop
  _OP_REQP = ["instance_name", "new_name"]
3133 decd5f45 Iustin Pop
3134 decd5f45 Iustin Pop
  def BuildHooksEnv(self):
3135 decd5f45 Iustin Pop
    """Build hooks env.
3136 decd5f45 Iustin Pop

3137 decd5f45 Iustin Pop
    This runs on master, primary and secondary nodes of the instance.
3138 decd5f45 Iustin Pop

3139 decd5f45 Iustin Pop
    """
3140 338e51e8 Iustin Pop
    env = _BuildInstanceHookEnvByObject(self, self.instance)
3141 decd5f45 Iustin Pop
    env["INSTANCE_NEW_NAME"] = self.op.new_name
3142 6b12959c Iustin Pop
    nl = [self.cfg.GetMasterNode()] + list(self.instance.all_nodes)
3143 decd5f45 Iustin Pop
    return env, nl, nl
3144 decd5f45 Iustin Pop
3145 decd5f45 Iustin Pop
  def CheckPrereq(self):
3146 decd5f45 Iustin Pop
    """Check prerequisites.
3147 decd5f45 Iustin Pop

3148 decd5f45 Iustin Pop
    This checks that the instance is in the cluster and is not running.
3149 decd5f45 Iustin Pop

3150 decd5f45 Iustin Pop
    """
3151 decd5f45 Iustin Pop
    instance = self.cfg.GetInstanceInfo(
3152 decd5f45 Iustin Pop
      self.cfg.ExpandInstanceName(self.op.instance_name))
3153 decd5f45 Iustin Pop
    if instance is None:
3154 decd5f45 Iustin Pop
      raise errors.OpPrereqError("Instance '%s' not known" %
3155 decd5f45 Iustin Pop
                                 self.op.instance_name)
3156 7527a8a4 Iustin Pop
    _CheckNodeOnline(self, instance.primary_node)
3157 7527a8a4 Iustin Pop
3158 0d68c45d Iustin Pop
    if instance.admin_up:
3159 decd5f45 Iustin Pop
      raise errors.OpPrereqError("Instance '%s' is marked to be up" %
3160 decd5f45 Iustin Pop
                                 self.op.instance_name)
3161 72737a7f Iustin Pop
    remote_info = self.rpc.call_instance_info(instance.primary_node,
3162 72737a7f Iustin Pop
                                              instance.name,
3163 72737a7f Iustin Pop
                                              instance.hypervisor)
3164 781de953 Iustin Pop
    remote_info.Raise()
3165 781de953 Iustin Pop
    if remote_info.data:
3166 decd5f45 Iustin Pop
      raise errors.OpPrereqError("Instance '%s' is running on the node %s" %
3167 decd5f45 Iustin Pop
                                 (self.op.instance_name,
3168 decd5f45 Iustin Pop
                                  instance.primary_node))
3169 decd5f45 Iustin Pop
    self.instance = instance
3170 decd5f45 Iustin Pop
3171 decd5f45 Iustin Pop
    # new name verification
3172 89e1fc26 Iustin Pop
    name_info = utils.HostInfo(self.op.new_name)
3173 decd5f45 Iustin Pop
3174 89e1fc26 Iustin Pop
    self.op.new_name = new_name = name_info.name
3175 7bde3275 Guido Trotter
    instance_list = self.cfg.GetInstanceList()
3176 7bde3275 Guido Trotter
    if new_name in instance_list:
3177 7bde3275 Guido Trotter
      raise errors.OpPrereqError("Instance '%s' is already in the cluster" %
3178 c09f363f Manuel Franceschini
                                 new_name)
3179 7bde3275 Guido Trotter
3180 decd5f45 Iustin Pop
    if not getattr(self.op, "ignore_ip", False):
3181 937f983d Guido Trotter
      if utils.TcpPing(name_info.ip, constants.DEFAULT_NODED_PORT):
3182 decd5f45 Iustin Pop
        raise errors.OpPrereqError("IP %s of instance %s already in use" %
3183 89e1fc26 Iustin Pop
                                   (name_info.ip, new_name))
3184 decd5f45 Iustin Pop
3185 decd5f45 Iustin Pop
3186 decd5f45 Iustin Pop
  def Exec(self, feedback_fn):
3187 decd5f45 Iustin Pop
    """Reinstall the instance.
3188 decd5f45 Iustin Pop

3189 decd5f45 Iustin Pop
    """
3190 decd5f45 Iustin Pop
    inst = self.instance
3191 decd5f45 Iustin Pop
    old_name = inst.name
3192 decd5f45 Iustin Pop
3193 b23c4333 Manuel Franceschini
    if inst.disk_template == constants.DT_FILE:
3194 b23c4333 Manuel Franceschini
      old_file_storage_dir = os.path.dirname(inst.disks[0].logical_id[1])
3195 b23c4333 Manuel Franceschini
3196 decd5f45 Iustin Pop
    self.cfg.RenameInstance(inst.name, self.op.new_name)
3197 74b5913f Guido Trotter
    # Change the instance lock. This is definitely safe while we hold the BGL
3198 cb4e8387 Iustin Pop
    self.context.glm.remove(locking.LEVEL_INSTANCE, old_name)
3199 74b5913f Guido Trotter
    self.context.glm.add(locking.LEVEL_INSTANCE, self.op.new_name)
3200 decd5f45 Iustin Pop
3201 decd5f45 Iustin Pop
    # re-read the instance from the configuration after rename
3202 decd5f45 Iustin Pop
    inst = self.cfg.GetInstanceInfo(self.op.new_name)
3203 decd5f45 Iustin Pop
3204 b23c4333 Manuel Franceschini
    if inst.disk_template == constants.DT_FILE:
3205 b23c4333 Manuel Franceschini
      new_file_storage_dir = os.path.dirname(inst.disks[0].logical_id[1])
3206 72737a7f Iustin Pop
      result = self.rpc.call_file_storage_dir_rename(inst.primary_node,
3207 72737a7f Iustin Pop
                                                     old_file_storage_dir,
3208 72737a7f Iustin Pop
                                                     new_file_storage_dir)
3209 781de953 Iustin Pop
      result.Raise()
3210 781de953 Iustin Pop
      if not result.data:
3211 b23c4333 Manuel Franceschini
        raise errors.OpExecError("Could not connect to node '%s' to rename"
3212 b23c4333 Manuel Franceschini
                                 " directory '%s' to '%s' (but the instance"
3213 b23c4333 Manuel Franceschini
                                 " has been renamed in Ganeti)" % (
3214 b23c4333 Manuel Franceschini
                                 inst.primary_node, old_file_storage_dir,
3215 b23c4333 Manuel Franceschini
                                 new_file_storage_dir))
3216 b23c4333 Manuel Franceschini
3217 781de953 Iustin Pop
      if not result.data[0]:
3218 b23c4333 Manuel Franceschini
        raise errors.OpExecError("Could not rename directory '%s' to '%s'"
3219 b23c4333 Manuel Franceschini
                                 " (but the instance has been renamed in"
3220 b23c4333 Manuel Franceschini
                                 " Ganeti)" % (old_file_storage_dir,
3221 b23c4333 Manuel Franceschini
                                               new_file_storage_dir))
3222 b23c4333 Manuel Franceschini
3223 b9bddb6b Iustin Pop
    _StartInstanceDisks(self, inst, None)
3224 decd5f45 Iustin Pop
    try:
3225 781de953 Iustin Pop
      result = self.rpc.call_instance_run_rename(inst.primary_node, inst,
3226 781de953 Iustin Pop
                                                 old_name)
3227 96841384 Iustin Pop
      msg = result.RemoteFailMsg()
3228 96841384 Iustin Pop
      if msg:
3229 6291574d Alexander Schreiber
        msg = ("Could not run OS rename script for instance %s on node %s"
3230 96841384 Iustin Pop
               " (but the instance has been renamed in Ganeti): %s" %
3231 96841384 Iustin Pop
               (inst.name, inst.primary_node, msg))
3232 86d9d3bb Iustin Pop
        self.proc.LogWarning(msg)
3233 decd5f45 Iustin Pop
    finally:
3234 b9bddb6b Iustin Pop
      _ShutdownInstanceDisks(self, inst)
3235 decd5f45 Iustin Pop
3236 decd5f45 Iustin Pop
3237 a8083063 Iustin Pop
class LURemoveInstance(LogicalUnit):
3238 a8083063 Iustin Pop
  """Remove an instance.
3239 a8083063 Iustin Pop

3240 a8083063 Iustin Pop
  """
3241 a8083063 Iustin Pop
  HPATH = "instance-remove"
3242 a8083063 Iustin Pop
  HTYPE = constants.HTYPE_INSTANCE
3243 5c54b832 Iustin Pop
  _OP_REQP = ["instance_name", "ignore_failures"]
3244 cf472233 Guido Trotter
  REQ_BGL = False
3245 cf472233 Guido Trotter
3246 cf472233 Guido Trotter
  def ExpandNames(self):
3247 cf472233 Guido Trotter
    self._ExpandAndLockInstance()
3248 cf472233 Guido Trotter
    self.needed_locks[locking.LEVEL_NODE] = []
3249 cf472233 Guido Trotter
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
3250 cf472233 Guido Trotter
3251 cf472233 Guido Trotter
  def DeclareLocks(self, level):
3252 cf472233 Guido Trotter
    if level == locking.LEVEL_NODE:
3253 cf472233 Guido Trotter
      self._LockInstancesNodes()
3254 a8083063 Iustin Pop
3255 a8083063 Iustin Pop
  def BuildHooksEnv(self):
3256 a8083063 Iustin Pop
    """Build hooks env.
3257 a8083063 Iustin Pop

3258 a8083063 Iustin Pop
    This runs on master, primary and secondary nodes of the instance.
3259 a8083063 Iustin Pop

3260 a8083063 Iustin Pop
    """
3261 338e51e8 Iustin Pop
    env = _BuildInstanceHookEnvByObject(self, self.instance)
3262 d6a02168 Michael Hanselmann
    nl = [self.cfg.GetMasterNode()]
3263 a8083063 Iustin Pop
    return env, nl, nl
3264 a8083063 Iustin Pop
3265 a8083063 Iustin Pop
  def CheckPrereq(self):
3266 a8083063 Iustin Pop
    """Check prerequisites.
3267 a8083063 Iustin Pop

3268 a8083063 Iustin Pop
    This checks that the instance is in the cluster.
3269 a8083063 Iustin Pop

3270 a8083063 Iustin Pop
    """
3271 cf472233 Guido Trotter
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_name)
3272 cf472233 Guido Trotter
    assert self.instance is not None, \
3273 cf472233 Guido Trotter
      "Cannot retrieve locked instance %s" % self.op.instance_name
3274 a8083063 Iustin Pop
3275 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
3276 a8083063 Iustin Pop
    """Remove the instance.
3277 a8083063 Iustin Pop

3278 a8083063 Iustin Pop
    """
3279 a8083063 Iustin Pop
    instance = self.instance
3280 9a4f63d1 Iustin Pop
    logging.info("Shutting down instance %s on node %s",
3281 9a4f63d1 Iustin Pop
                 instance.name, instance.primary_node)
3282 a8083063 Iustin Pop
3283 781de953 Iustin Pop
    result = self.rpc.call_instance_shutdown(instance.primary_node, instance)
3284 1fae010f Iustin Pop
    msg = result.RemoteFailMsg()
3285 1fae010f Iustin Pop
    if msg:
3286 1d67656e Iustin Pop
      if self.op.ignore_failures:
3287 1fae010f Iustin Pop
        feedback_fn("Warning: can't shutdown instance: %s" % msg)
3288 1d67656e Iustin Pop
      else:
3289 1fae010f Iustin Pop
        raise errors.OpExecError("Could not shutdown instance %s on"
3290 1fae010f Iustin Pop
                                 " node %s: %s" %
3291 1fae010f Iustin Pop
                                 (instance.name, instance.primary_node, msg))
3292 a8083063 Iustin Pop
3293 9a4f63d1 Iustin Pop
    logging.info("Removing block devices for instance %s", instance.name)
3294 a8083063 Iustin Pop
3295 b9bddb6b Iustin Pop
    if not _RemoveDisks(self, instance):
3296 1d67656e Iustin Pop
      if self.op.ignore_failures:
3297 1d67656e Iustin Pop
        feedback_fn("Warning: can't remove instance's disks")
3298 1d67656e Iustin Pop
      else:
3299 1d67656e Iustin Pop
        raise errors.OpExecError("Can't remove instance's disks")
3300 a8083063 Iustin Pop
3301 9a4f63d1 Iustin Pop
    logging.info("Removing instance %s out of cluster config", instance.name)
3302 a8083063 Iustin Pop
3303 a8083063 Iustin Pop
    self.cfg.RemoveInstance(instance.name)
3304 cf472233 Guido Trotter
    self.remove_locks[locking.LEVEL_INSTANCE] = instance.name
3305 a8083063 Iustin Pop
3306 a8083063 Iustin Pop
3307 a8083063 Iustin Pop
class LUQueryInstances(NoHooksLU):
3308 a8083063 Iustin Pop
  """Logical unit for querying instances.
3309 a8083063 Iustin Pop

3310 a8083063 Iustin Pop
  """
3311 ec79568d Iustin Pop
  _OP_REQP = ["output_fields", "names", "use_locking"]
3312 7eb9d8f7 Guido Trotter
  REQ_BGL = False
3313 a2d2e1a7 Iustin Pop
  _FIELDS_STATIC = utils.FieldSet(*["name", "os", "pnode", "snodes",
3314 5b460366 Iustin Pop
                                    "admin_state",
3315 a2d2e1a7 Iustin Pop
                                    "disk_template", "ip", "mac", "bridge",
3316 a2d2e1a7 Iustin Pop
                                    "sda_size", "sdb_size", "vcpus", "tags",
3317 a2d2e1a7 Iustin Pop
                                    "network_port", "beparams",
3318 8aec325c Iustin Pop
                                    r"(disk)\.(size)/([0-9]+)",
3319 8aec325c Iustin Pop
                                    r"(disk)\.(sizes)", "disk_usage",
3320 8aec325c Iustin Pop
                                    r"(nic)\.(mac|ip|bridge)/([0-9]+)",
3321 8aec325c Iustin Pop
                                    r"(nic)\.(macs|ips|bridges)",
3322 8aec325c Iustin Pop
                                    r"(disk|nic)\.(count)",
3323 a2d2e1a7 Iustin Pop
                                    "serial_no", "hypervisor", "hvparams",] +
3324 a2d2e1a7 Iustin Pop
                                  ["hv/%s" % name
3325 a2d2e1a7 Iustin Pop
                                   for name in constants.HVS_PARAMETERS] +
3326 a2d2e1a7 Iustin Pop
                                  ["be/%s" % name
3327 a2d2e1a7 Iustin Pop
                                   for name in constants.BES_PARAMETERS])
3328 a2d2e1a7 Iustin Pop
  _FIELDS_DYNAMIC = utils.FieldSet("oper_state", "oper_ram", "status")
3329 31bf511f Iustin Pop
3330 a8083063 Iustin Pop
3331 7eb9d8f7 Guido Trotter
  def ExpandNames(self):
3332 31bf511f Iustin Pop
    _CheckOutputFields(static=self._FIELDS_STATIC,
3333 31bf511f Iustin Pop
                       dynamic=self._FIELDS_DYNAMIC,
3334 dcb93971 Michael Hanselmann
                       selected=self.op.output_fields)
3335 a8083063 Iustin Pop
3336 7eb9d8f7 Guido Trotter
    self.needed_locks = {}
3337 7eb9d8f7 Guido Trotter
    self.share_locks[locking.LEVEL_INSTANCE] = 1
3338 7eb9d8f7 Guido Trotter
    self.share_locks[locking.LEVEL_NODE] = 1
3339 7eb9d8f7 Guido Trotter
3340 57a2fb91 Iustin Pop
    if self.op.names:
3341 57a2fb91 Iustin Pop
      self.wanted = _GetWantedInstances(self, self.op.names)
3342 7eb9d8f7 Guido Trotter
    else:
3343 57a2fb91 Iustin Pop
      self.wanted = locking.ALL_SET
3344 7eb9d8f7 Guido Trotter
3345 ec79568d Iustin Pop
    self.do_node_query = self._FIELDS_STATIC.NonMatching(self.op.output_fields)
3346 ec79568d Iustin Pop
    self.do_locking = self.do_node_query and self.op.use_locking
3347 57a2fb91 Iustin Pop
    if self.do_locking:
3348 57a2fb91 Iustin Pop
      self.needed_locks[locking.LEVEL_INSTANCE] = self.wanted
3349 57a2fb91 Iustin Pop
      self.needed_locks[locking.LEVEL_NODE] = []
3350 57a2fb91 Iustin Pop
      self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
3351 7eb9d8f7 Guido Trotter
3352 7eb9d8f7 Guido Trotter
  def DeclareLocks(self, level):
3353 57a2fb91 Iustin Pop
    if level == locking.LEVEL_NODE and self.do_locking:
3354 7eb9d8f7 Guido Trotter
      self._LockInstancesNodes()
3355 7eb9d8f7 Guido Trotter
3356 7eb9d8f7 Guido Trotter
  def CheckPrereq(self):
3357 7eb9d8f7 Guido Trotter
    """Check prerequisites.
3358 7eb9d8f7 Guido Trotter

3359 7eb9d8f7 Guido Trotter
    """
3360 57a2fb91 Iustin Pop
    pass
3361 069dcc86 Iustin Pop
3362 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
3363 a8083063 Iustin Pop
    """Computes the list of nodes and their attributes.
3364 a8083063 Iustin Pop

3365 a8083063 Iustin Pop
    """
3366 57a2fb91 Iustin Pop
    all_info = self.cfg.GetAllInstancesInfo()
3367 a7f5dc98 Iustin Pop
    if self.wanted == locking.ALL_SET:
3368 a7f5dc98 Iustin Pop
      # caller didn't specify instance names, so ordering is not important
3369 a7f5dc98 Iustin Pop
      if self.do_locking:
3370 a7f5dc98 Iustin Pop
        instance_names = self.acquired_locks[locking.LEVEL_INSTANCE]
3371 a7f5dc98 Iustin Pop
      else:
3372 a7f5dc98 Iustin Pop
        instance_names = all_info.keys()
3373 a7f5dc98 Iustin Pop
      instance_names = utils.NiceSort(instance_names)
3374 57a2fb91 Iustin Pop
    else:
3375 a7f5dc98 Iustin Pop
      # caller did specify names, so we must keep the ordering
3376 a7f5dc98 Iustin Pop
      if self.do_locking:
3377 a7f5dc98 Iustin Pop
        tgt_set = self.acquired_locks[locking.LEVEL_INSTANCE]
3378 a7f5dc98 Iustin Pop
      else:
3379 a7f5dc98 Iustin Pop
        tgt_set = all_info.keys()
3380 a7f5dc98 Iustin Pop
      missing = set(self.wanted).difference(tgt_set)
3381 a7f5dc98 Iustin Pop
      if missing:
3382 a7f5dc98 Iustin Pop
        raise errors.OpExecError("Some instances were removed before"
3383 a7f5dc98 Iustin Pop
                                 " retrieving their data: %s" % missing)
3384 a7f5dc98 Iustin Pop
      instance_names = self.wanted
3385 c1f1cbb2 Iustin Pop
3386 57a2fb91 Iustin Pop
    instance_list = [all_info[iname] for iname in instance_names]
3387 a8083063 Iustin Pop
3388 a8083063 Iustin Pop
    # begin data gathering
3389 a8083063 Iustin Pop
3390 a8083063 Iustin Pop
    nodes = frozenset([inst.primary_node for inst in instance_list])
3391 e69d05fd Iustin Pop
    hv_list = list(set([inst.hypervisor for inst in instance_list]))
3392 a8083063 Iustin Pop
3393 a8083063 Iustin Pop
    bad_nodes = []
3394 cbfc4681 Iustin Pop
    off_nodes = []
3395 ec79568d Iustin Pop
    if self.do_node_query:
3396 a8083063 Iustin Pop
      live_data = {}
3397 72737a7f Iustin Pop
      node_data = self.rpc.call_all_instances_info(nodes, hv_list)
3398 a8083063 Iustin Pop
      for name in nodes:
3399 a8083063 Iustin Pop
        result = node_data[name]
3400 cbfc4681 Iustin Pop
        if result.offline:
3401 cbfc4681 Iustin Pop
          # offline nodes will be in both lists
3402 cbfc4681 Iustin Pop
          off_nodes.append(name)
3403 781de953 Iustin Pop
        if result.failed:
3404 a8083063 Iustin Pop
          bad_nodes.append(name)
3405 781de953 Iustin Pop
        else:
3406 781de953 Iustin Pop
          if result.data:
3407 781de953 Iustin Pop
            live_data.update(result.data)
3408 781de953 Iustin Pop
            # else no instance is alive
3409 a8083063 Iustin Pop
    else:
3410 a8083063 Iustin Pop
      live_data = dict([(name, {}) for name in instance_names])
3411 a8083063 Iustin Pop
3412 a8083063 Iustin Pop
    # end data gathering
3413 a8083063 Iustin Pop
3414 5018a335 Iustin Pop
    HVPREFIX = "hv/"
3415 338e51e8 Iustin Pop
    BEPREFIX = "be/"
3416 a8083063 Iustin Pop
    output = []
3417 a8083063 Iustin Pop
    for instance in instance_list:
3418 a8083063 Iustin Pop
      iout = []
3419 5018a335 Iustin Pop
      i_hv = self.cfg.GetClusterInfo().FillHV(instance)
3420 338e51e8 Iustin Pop
      i_be = self.cfg.GetClusterInfo().FillBE(instance)
3421 a8083063 Iustin Pop
      for field in self.op.output_fields:
3422 71c1af58 Iustin Pop
        st_match = self._FIELDS_STATIC.Matches(field)
3423 a8083063 Iustin Pop
        if field == "name":
3424 a8083063 Iustin Pop
          val = instance.name
3425 a8083063 Iustin Pop
        elif field == "os":
3426 a8083063 Iustin Pop
          val = instance.os
3427 a8083063 Iustin Pop
        elif field == "pnode":
3428 a8083063 Iustin Pop
          val = instance.primary_node
3429 a8083063 Iustin Pop
        elif field == "snodes":
3430 8a23d2d3 Iustin Pop
          val = list(instance.secondary_nodes)
3431 a8083063 Iustin Pop
        elif field == "admin_state":
3432 0d68c45d Iustin Pop
          val = instance.admin_up
3433 a8083063 Iustin Pop
        elif field == "oper_state":
3434 a8083063 Iustin Pop
          if instance.primary_node in bad_nodes:
3435 8a23d2d3 Iustin Pop
            val = None
3436 a8083063 Iustin Pop
          else:
3437 8a23d2d3 Iustin Pop
            val = bool(live_data.get(instance.name))
3438 d8052456 Iustin Pop
        elif field == "status":
3439 cbfc4681 Iustin Pop
          if instance.primary_node in off_nodes:
3440 cbfc4681 Iustin Pop
            val = "ERROR_nodeoffline"
3441 cbfc4681 Iustin Pop
          elif instance.primary_node in bad_nodes:
3442 d8052456 Iustin Pop
            val = "ERROR_nodedown"
3443 d8052456 Iustin Pop
          else:
3444 d8052456 Iustin Pop
            running = bool(live_data.get(instance.name))
3445 d8052456 Iustin Pop
            if running:
3446 0d68c45d Iustin Pop
              if instance.admin_up:
3447 d8052456 Iustin Pop
                val = "running"
3448 d8052456 Iustin Pop
              else:
3449 d8052456 Iustin Pop
                val = "ERROR_up"
3450 d8052456 Iustin Pop
            else:
3451 0d68c45d Iustin Pop
              if instance.admin_up:
3452 d8052456 Iustin Pop
                val = "ERROR_down"
3453 d8052456 Iustin Pop
              else:
3454 d8052456 Iustin Pop
                val = "ADMIN_down"
3455 a8083063 Iustin Pop
        elif field == "oper_ram":
3456 a8083063 Iustin Pop
          if instance.primary_node in bad_nodes:
3457 8a23d2d3 Iustin Pop
            val = None
3458 a8083063 Iustin Pop
          elif instance.name in live_data:
3459 a8083063 Iustin Pop
            val = live_data[instance.name].get("memory", "?")
3460 a8083063 Iustin Pop
          else:
3461 a8083063 Iustin Pop
            val = "-"
3462 a8083063 Iustin Pop
        elif field == "disk_template":
3463 a8083063 Iustin Pop
          val = instance.disk_template
3464 a8083063 Iustin Pop
        elif field == "ip":
3465 a8083063 Iustin Pop
          val = instance.nics[0].ip
3466 a8083063 Iustin Pop
        elif field == "bridge":
3467 a8083063 Iustin Pop
          val = instance.nics[0].bridge
3468 a8083063 Iustin Pop
        elif field == "mac":
3469 a8083063 Iustin Pop
          val = instance.nics[0].mac
3470 644eeef9 Iustin Pop
        elif field == "sda_size" or field == "sdb_size":
3471 ad24e046 Iustin Pop
          idx = ord(field[2]) - ord('a')
3472 ad24e046 Iustin Pop
          try:
3473 ad24e046 Iustin Pop
            val = instance.FindDisk(idx).size
3474 ad24e046 Iustin Pop
          except errors.OpPrereqError:
3475 8a23d2d3 Iustin Pop
            val = None
3476 024e157f Iustin Pop
        elif field == "disk_usage": # total disk usage per node
3477 024e157f Iustin Pop
          disk_sizes = [{'size': disk.size} for disk in instance.disks]
3478 024e157f Iustin Pop
          val = _ComputeDiskSize(instance.disk_template, disk_sizes)
3479 130a6a6f Iustin Pop
        elif field == "tags":
3480 130a6a6f Iustin Pop
          val = list(instance.GetTags())
3481 38d7239a Iustin Pop
        elif field == "serial_no":
3482 38d7239a Iustin Pop
          val = instance.serial_no
3483 5018a335 Iustin Pop
        elif field == "network_port":
3484 5018a335 Iustin Pop
          val = instance.network_port
3485 338e51e8 Iustin Pop
        elif field == "hypervisor":
3486 338e51e8 Iustin Pop
          val = instance.hypervisor
3487 338e51e8 Iustin Pop
        elif field == "hvparams":
3488 338e51e8 Iustin Pop
          val = i_hv
3489 5018a335 Iustin Pop
        elif (field.startswith(HVPREFIX) and
3490 5018a335 Iustin Pop
              field[len(HVPREFIX):] in constants.HVS_PARAMETERS):
3491 5018a335 Iustin Pop
          val = i_hv.get(field[len(HVPREFIX):], None)
3492 338e51e8 Iustin Pop
        elif field == "beparams":
3493 338e51e8 Iustin Pop
          val = i_be
3494 338e51e8 Iustin Pop
        elif (field.startswith(BEPREFIX) and
3495 338e51e8 Iustin Pop
              field[len(BEPREFIX):] in constants.BES_PARAMETERS):
3496 338e51e8 Iustin Pop
          val = i_be.get(field[len(BEPREFIX):], None)
3497 71c1af58 Iustin Pop
        elif st_match and st_match.groups():
3498 71c1af58 Iustin Pop
          # matches a variable list
3499 71c1af58 Iustin Pop
          st_groups = st_match.groups()
3500 71c1af58 Iustin Pop
          if st_groups and st_groups[0] == "disk":
3501 71c1af58 Iustin Pop
            if st_groups[1] == "count":
3502 71c1af58 Iustin Pop
              val = len(instance.disks)
3503 41a776da Iustin Pop
            elif st_groups[1] == "sizes":
3504 41a776da Iustin Pop
              val = [disk.size for disk in instance.disks]
3505 71c1af58 Iustin Pop
            elif st_groups[1] == "size":
3506 3e0cea06 Iustin Pop
              try:
3507 3e0cea06 Iustin Pop
                val = instance.FindDisk(st_groups[2]).size
3508 3e0cea06 Iustin Pop
              except errors.OpPrereqError:
3509 71c1af58 Iustin Pop
                val = None
3510 71c1af58 Iustin Pop
            else:
3511 71c1af58 Iustin Pop
              assert False, "Unhandled disk parameter"
3512 71c1af58 Iustin Pop
          elif st_groups[0] == "nic":
3513 71c1af58 Iustin Pop
            if st_groups[1] == "count":
3514 71c1af58 Iustin Pop
              val = len(instance.nics)
3515 41a776da Iustin Pop
            elif st_groups[1] == "macs":
3516 41a776da Iustin Pop
              val = [nic.mac for nic in instance.nics]
3517 41a776da Iustin Pop
            elif st_groups[1] == "ips":
3518 41a776da Iustin Pop
              val = [nic.ip for nic in instance.nics]
3519 41a776da Iustin Pop
            elif st_groups[1] == "bridges":
3520 41a776da Iustin Pop
              val = [nic.bridge for nic in instance.nics]
3521 71c1af58 Iustin Pop
            else:
3522 71c1af58 Iustin Pop
              # index-based item
3523 71c1af58 Iustin Pop
              nic_idx = int(st_groups[2])
3524 71c1af58 Iustin Pop
              if nic_idx >= len(instance.nics):
3525 71c1af58 Iustin Pop
                val = None
3526 71c1af58 Iustin Pop
              else:
3527 71c1af58 Iustin Pop
                if st_groups[1] == "mac":
3528 71c1af58 Iustin Pop
                  val = instance.nics[nic_idx].mac
3529 71c1af58 Iustin Pop
                elif st_groups[1] == "ip":
3530 71c1af58 Iustin Pop
                  val = instance.nics[nic_idx].ip
3531 71c1af58 Iustin Pop
                elif st_groups[1] == "bridge":
3532 71c1af58 Iustin Pop
                  val = instance.nics[nic_idx].bridge
3533 71c1af58 Iustin Pop
                else:
3534 71c1af58 Iustin Pop
                  assert False, "Unhandled NIC parameter"
3535 71c1af58 Iustin Pop
          else:
3536 71c1af58 Iustin Pop
            assert False, "Unhandled variable parameter"
3537 a8083063 Iustin Pop
        else:
3538 3ecf6786 Iustin Pop
          raise errors.ParameterError(field)
3539 a8083063 Iustin Pop
        iout.append(val)
3540 a8083063 Iustin Pop
      output.append(iout)
3541 a8083063 Iustin Pop
3542 a8083063 Iustin Pop
    return output
3543 a8083063 Iustin Pop
3544 a8083063 Iustin Pop
3545 a8083063 Iustin Pop
class LUFailoverInstance(LogicalUnit):
3546 a8083063 Iustin Pop
  """Failover an instance.
3547 a8083063 Iustin Pop

3548 a8083063 Iustin Pop
  """
3549 a8083063 Iustin Pop
  HPATH = "instance-failover"
3550 a8083063 Iustin Pop
  HTYPE = constants.HTYPE_INSTANCE
3551 a8083063 Iustin Pop
  _OP_REQP = ["instance_name", "ignore_consistency"]
3552 c9e5c064 Guido Trotter
  REQ_BGL = False
3553 c9e5c064 Guido Trotter
3554 c9e5c064 Guido Trotter
  def ExpandNames(self):
3555 c9e5c064 Guido Trotter
    self._ExpandAndLockInstance()
3556 c9e5c064 Guido Trotter
    self.needed_locks[locking.LEVEL_NODE] = []
3557 f6d9a522 Guido Trotter
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
3558 c9e5c064 Guido Trotter
3559 c9e5c064 Guido Trotter
  def DeclareLocks(self, level):
3560 c9e5c064 Guido Trotter
    if level == locking.LEVEL_NODE:
3561 c9e5c064 Guido Trotter
      self._LockInstancesNodes()
3562 a8083063 Iustin Pop
3563 a8083063 Iustin Pop
  def BuildHooksEnv(self):
3564 a8083063 Iustin Pop
    """Build hooks env.
3565 a8083063 Iustin Pop

3566 a8083063 Iustin Pop
    This runs on master, primary and secondary nodes of the instance.
3567 a8083063 Iustin Pop

3568 a8083063 Iustin Pop
    """
3569 a8083063 Iustin Pop
    env = {
3570 a8083063 Iustin Pop
      "IGNORE_CONSISTENCY": self.op.ignore_consistency,
3571 a8083063 Iustin Pop
      }
3572 338e51e8 Iustin Pop
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
3573 d6a02168 Michael Hanselmann
    nl = [self.cfg.GetMasterNode()] + list(self.instance.secondary_nodes)
3574 a8083063 Iustin Pop
    return env, nl, nl
3575 a8083063 Iustin Pop
3576 a8083063 Iustin Pop
  def CheckPrereq(self):
3577 a8083063 Iustin Pop
    """Check prerequisites.
3578 a8083063 Iustin Pop

3579 a8083063 Iustin Pop
    This checks that the instance is in the cluster.
3580 a8083063 Iustin Pop

3581 a8083063 Iustin Pop
    """
3582 c9e5c064 Guido Trotter
    self.instance = instance = self.cfg.GetInstanceInfo(self.op.instance_name)
3583 c9e5c064 Guido Trotter
    assert self.instance is not None, \
3584 c9e5c064 Guido Trotter
      "Cannot retrieve locked instance %s" % self.op.instance_name
3585 a8083063 Iustin Pop
3586 338e51e8 Iustin Pop
    bep = self.cfg.GetClusterInfo().FillBE(instance)
3587 a1f445d3 Iustin Pop
    if instance.disk_template not in constants.DTS_NET_MIRROR:
3588 2a710df1 Michael Hanselmann
      raise errors.OpPrereqError("Instance's disk layout is not"
3589 a1f445d3 Iustin Pop
                                 " network mirrored, cannot failover.")
3590 2a710df1 Michael Hanselmann
3591 2a710df1 Michael Hanselmann
    secondary_nodes = instance.secondary_nodes
3592 2a710df1 Michael Hanselmann
    if not secondary_nodes:
3593 2a710df1 Michael Hanselmann
      raise errors.ProgrammerError("no secondary node but using "
3594 abdf0113 Iustin Pop
                                   "a mirrored disk template")
3595 2a710df1 Michael Hanselmann
3596 2a710df1 Michael Hanselmann
    target_node = secondary_nodes[0]
3597 7527a8a4 Iustin Pop
    _CheckNodeOnline(self, target_node)
3598 733a2b6a Iustin Pop
    _CheckNodeNotDrained(self, target_node)
3599 d4f16fd9 Iustin Pop
    # check memory requirements on the secondary node
3600 b9bddb6b Iustin Pop
    _CheckNodeFreeMemory(self, target_node, "failing over instance %s" %
3601 338e51e8 Iustin Pop
                         instance.name, bep[constants.BE_MEMORY],
3602 e69d05fd Iustin Pop
                         instance.hypervisor)
3603 3a7c308e Guido Trotter
3604 a8083063 Iustin Pop
    # check bridge existance
3605 a8083063 Iustin Pop
    brlist = [nic.bridge for nic in instance.nics]
3606 781de953 Iustin Pop
    result = self.rpc.call_bridges_exist(target_node, brlist)
3607 781de953 Iustin Pop
    result.Raise()
3608 781de953 Iustin Pop
    if not result.data:
3609 3ecf6786 Iustin Pop
      raise errors.OpPrereqError("One or more target bridges %s does not"
3610 3ecf6786 Iustin Pop
                                 " exist on destination node '%s'" %
3611 50ff9a7a Iustin Pop
                                 (brlist, target_node))
3612 a8083063 Iustin Pop
3613 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
3614 a8083063 Iustin Pop
    """Failover an instance.
3615 a8083063 Iustin Pop

3616 a8083063 Iustin Pop
    The failover is done by shutting it down on its present node and
3617 a8083063 Iustin Pop
    starting it on the secondary.
3618 a8083063 Iustin Pop

3619 a8083063 Iustin Pop
    """
3620 a8083063 Iustin Pop
    instance = self.instance
3621 a8083063 Iustin Pop
3622 a8083063 Iustin Pop
    source_node = instance.primary_node
3623 a8083063 Iustin Pop
    target_node = instance.secondary_nodes[0]
3624 a8083063 Iustin Pop
3625 a8083063 Iustin Pop
    feedback_fn("* checking disk consistency between source and target")
3626 a8083063 Iustin Pop
    for dev in instance.disks:
3627 abdf0113 Iustin Pop
      # for drbd, these are drbd over lvm
3628 b9bddb6b Iustin Pop
      if not _CheckDiskConsistency(self, dev, target_node, False):
3629 0d68c45d Iustin Pop
        if instance.admin_up and not self.op.ignore_consistency:
3630 3ecf6786 Iustin Pop
          raise errors.OpExecError("Disk %s is degraded on target node,"
3631 3ecf6786 Iustin Pop
                                   " aborting failover." % dev.iv_name)
3632 a8083063 Iustin Pop
3633 a8083063 Iustin Pop
    feedback_fn("* shutting down instance on source node")
3634 9a4f63d1 Iustin Pop
    logging.info("Shutting down instance %s on node %s",
3635 9a4f63d1 Iustin Pop
                 instance.name, source_node)
3636 a8083063 Iustin Pop
3637 781de953 Iustin Pop
    result = self.rpc.call_instance_shutdown(source_node, instance)
3638 1fae010f Iustin Pop
    msg = result.RemoteFailMsg()
3639 1fae010f Iustin Pop
    if msg:
3640 24a40d57 Iustin Pop
      if self.op.ignore_consistency:
3641 86d9d3bb Iustin Pop
        self.proc.LogWarning("Could not shutdown instance %s on node %s."
3642 1fae010f Iustin Pop
                             " Proceeding anyway. Please make sure node"
3643 1fae010f Iustin Pop
                             " %s is down. Error details: %s",
3644 1fae010f Iustin Pop
                             instance.name, source_node, source_node, msg)
3645 24a40d57 Iustin Pop
      else:
3646 1fae010f Iustin Pop
        raise errors.OpExecError("Could not shutdown instance %s on"
3647 1fae010f Iustin Pop
                                 " node %s: %s" %
3648 1fae010f Iustin Pop
                                 (instance.name, source_node, msg))
3649 a8083063 Iustin Pop
3650 a8083063 Iustin Pop
    feedback_fn("* deactivating the instance's disks on source node")
3651 b9bddb6b Iustin Pop
    if not _ShutdownInstanceDisks(self, instance, ignore_primary=True):
3652 3ecf6786 Iustin Pop
      raise errors.OpExecError("Can't shut down the instance's disks.")
3653 a8083063 Iustin Pop
3654 a8083063 Iustin Pop
    instance.primary_node = target_node
3655 a8083063 Iustin Pop
    # distribute new instance config to the other nodes
3656 b6102dab Guido Trotter
    self.cfg.Update(instance)
3657 a8083063 Iustin Pop
3658 12a0cfbe Guido Trotter
    # Only start the instance if it's marked as up
3659 0d68c45d Iustin Pop
    if instance.admin_up:
3660 12a0cfbe Guido Trotter
      feedback_fn("* activating the instance's disks on target node")
3661 9a4f63d1 Iustin Pop
      logging.info("Starting instance %s on node %s",
3662 9a4f63d1 Iustin Pop
                   instance.name, target_node)
3663 12a0cfbe Guido Trotter
3664 b9bddb6b Iustin Pop
      disks_ok, dummy = _AssembleInstanceDisks(self, instance,
3665 12a0cfbe Guido Trotter
                                               ignore_secondaries=True)
3666 12a0cfbe Guido Trotter
      if not disks_ok:
3667 b9bddb6b Iustin Pop
        _ShutdownInstanceDisks(self, instance)
3668 12a0cfbe Guido Trotter
        raise errors.OpExecError("Can't activate the instance's disks")
3669 a8083063 Iustin Pop
3670 12a0cfbe Guido Trotter
      feedback_fn("* starting the instance on the target node")
3671 0eca8e0c Iustin Pop
      result = self.rpc.call_instance_start(target_node, instance, None, None)
3672 dd279568 Iustin Pop
      msg = result.RemoteFailMsg()
3673 dd279568 Iustin Pop
      if msg:
3674 b9bddb6b Iustin Pop
        _ShutdownInstanceDisks(self, instance)
3675 dd279568 Iustin Pop
        raise errors.OpExecError("Could not start instance %s on node %s: %s" %
3676 dd279568 Iustin Pop
                                 (instance.name, target_node, msg))
3677 a8083063 Iustin Pop
3678 a8083063 Iustin Pop
3679 53c776b5 Iustin Pop
class LUMigrateInstance(LogicalUnit):
3680 53c776b5 Iustin Pop
  """Migrate an instance.
3681 53c776b5 Iustin Pop

3682 53c776b5 Iustin Pop
  This is migration without shutting down, compared to the failover,
3683 53c776b5 Iustin Pop
  which is done with shutdown.
3684 53c776b5 Iustin Pop

3685 53c776b5 Iustin Pop
  """
3686 53c776b5 Iustin Pop
  HPATH = "instance-migrate"
3687 53c776b5 Iustin Pop
  HTYPE = constants.HTYPE_INSTANCE
3688 53c776b5 Iustin Pop
  _OP_REQP = ["instance_name", "live", "cleanup"]
3689 53c776b5 Iustin Pop
3690 53c776b5 Iustin Pop
  REQ_BGL = False
3691 53c776b5 Iustin Pop
3692 53c776b5 Iustin Pop
  def ExpandNames(self):
3693 53c776b5 Iustin Pop
    self._ExpandAndLockInstance()
3694 53c776b5 Iustin Pop
    self.needed_locks[locking.LEVEL_NODE] = []
3695 53c776b5 Iustin Pop
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
3696 53c776b5 Iustin Pop
3697 53c776b5 Iustin Pop
  def DeclareLocks(self, level):
3698 53c776b5 Iustin Pop
    if level == locking.LEVEL_NODE:
3699 53c776b5 Iustin Pop
      self._LockInstancesNodes()
3700 53c776b5 Iustin Pop
3701 53c776b5 Iustin Pop
  def BuildHooksEnv(self):
3702 53c776b5 Iustin Pop
    """Build hooks env.
3703 53c776b5 Iustin Pop

3704 53c776b5 Iustin Pop
    This runs on master, primary and secondary nodes of the instance.
3705 53c776b5 Iustin Pop

3706 53c776b5 Iustin Pop
    """
3707 53c776b5 Iustin Pop
    env = _BuildInstanceHookEnvByObject(self, self.instance)
3708 2c2690c9 Iustin Pop
    env["MIGRATE_LIVE"] = self.op.live
3709 2c2690c9 Iustin Pop
    env["MIGRATE_CLEANUP"] = self.op.cleanup
3710 53c776b5 Iustin Pop
    nl = [self.cfg.GetMasterNode()] + list(self.instance.secondary_nodes)
3711 53c776b5 Iustin Pop
    return env, nl, nl
3712 53c776b5 Iustin Pop
3713 53c776b5 Iustin Pop
  def CheckPrereq(self):
3714 53c776b5 Iustin Pop
    """Check prerequisites.
3715 53c776b5 Iustin Pop

3716 53c776b5 Iustin Pop
    This checks that the instance is in the cluster.
3717 53c776b5 Iustin Pop

3718 53c776b5 Iustin Pop
    """
3719 53c776b5 Iustin Pop
    instance = self.cfg.GetInstanceInfo(
3720 53c776b5 Iustin Pop
      self.cfg.ExpandInstanceName(self.op.instance_name))
3721 53c776b5 Iustin Pop
    if instance is None:
3722 53c776b5 Iustin Pop
      raise errors.OpPrereqError("Instance '%s' not known" %
3723 53c776b5 Iustin Pop
                                 self.op.instance_name)
3724 53c776b5 Iustin Pop
3725 53c776b5 Iustin Pop
    if instance.disk_template != constants.DT_DRBD8:
3726 53c776b5 Iustin Pop
      raise errors.OpPrereqError("Instance's disk layout is not"
3727 53c776b5 Iustin Pop
                                 " drbd8, cannot migrate.")
3728 53c776b5 Iustin Pop
3729 53c776b5 Iustin Pop
    secondary_nodes = instance.secondary_nodes
3730 53c776b5 Iustin Pop
    if not secondary_nodes:
3731 733a2b6a Iustin Pop
      raise errors.ConfigurationError("No secondary node but using"
3732 733a2b6a Iustin Pop
                                      " drbd8 disk template")
3733 53c776b5 Iustin Pop
3734 53c776b5 Iustin Pop
    i_be = self.cfg.GetClusterInfo().FillBE(instance)
3735 53c776b5 Iustin Pop
3736 53c776b5 Iustin Pop
    target_node = secondary_nodes[0]
3737 53c776b5 Iustin Pop
    # check memory requirements on the secondary node
3738 53c776b5 Iustin Pop
    _CheckNodeFreeMemory(self, target_node, "migrating instance %s" %
3739 53c776b5 Iustin Pop
                         instance.name, i_be[constants.BE_MEMORY],
3740 53c776b5 Iustin Pop
                         instance.hypervisor)
3741 53c776b5 Iustin Pop
3742 53c776b5 Iustin Pop
    # check bridge existance
3743 53c776b5 Iustin Pop
    brlist = [nic.bridge for nic in instance.nics]
3744 53c776b5 Iustin Pop
    result = self.rpc.call_bridges_exist(target_node, brlist)
3745 53c776b5 Iustin Pop
    if result.failed or not result.data:
3746 53c776b5 Iustin Pop
      raise errors.OpPrereqError("One or more target bridges %s does not"
3747 53c776b5 Iustin Pop
                                 " exist on destination node '%s'" %
3748 53c776b5 Iustin Pop
                                 (brlist, target_node))
3749 53c776b5 Iustin Pop
3750 53c776b5 Iustin Pop
    if not self.op.cleanup:
3751 733a2b6a Iustin Pop
      _CheckNodeNotDrained(self, target_node)
3752 53c776b5 Iustin Pop
      result = self.rpc.call_instance_migratable(instance.primary_node,
3753 53c776b5 Iustin Pop
                                                 instance)
3754 53c776b5 Iustin Pop
      msg = result.RemoteFailMsg()
3755 53c776b5 Iustin Pop
      if msg:
3756 53c776b5 Iustin Pop
        raise errors.OpPrereqError("Can't migrate: %s - please use failover" %
3757 53c776b5 Iustin Pop
                                   msg)
3758 53c776b5 Iustin Pop
3759 53c776b5 Iustin Pop
    self.instance = instance
3760 53c776b5 Iustin Pop
3761 53c776b5 Iustin Pop
  def _WaitUntilSync(self):
3762 53c776b5 Iustin Pop
    """Poll with custom rpc for disk sync.
3763 53c776b5 Iustin Pop

3764 53c776b5 Iustin Pop
    This uses our own step-based rpc call.
3765 53c776b5 Iustin Pop

3766 53c776b5 Iustin Pop
    """
3767 53c776b5 Iustin Pop
    self.feedback_fn("* wait until resync is done")
3768 53c776b5 Iustin Pop
    all_done = False
3769 53c776b5 Iustin Pop
    while not all_done:
3770 53c776b5 Iustin Pop
      all_done = True
3771 53c776b5 Iustin Pop
      result = self.rpc.call_drbd_wait_sync(self.all_nodes,
3772 53c776b5 Iustin Pop
                                            self.nodes_ip,
3773 53c776b5 Iustin Pop
                                            self.instance.disks)
3774 53c776b5 Iustin Pop
      min_percent = 100
3775 53c776b5 Iustin Pop
      for node, nres in result.items():
3776 53c776b5 Iustin Pop
        msg = nres.RemoteFailMsg()
3777 53c776b5 Iustin Pop
        if msg:
3778 53c776b5 Iustin Pop
          raise errors.OpExecError("Cannot resync disks on node %s: %s" %
3779 53c776b5 Iustin Pop
                                   (node, msg))
3780 0959c824 Iustin Pop
        node_done, node_percent = nres.payload
3781 53c776b5 Iustin Pop
        all_done = all_done and node_done
3782 53c776b5 Iustin Pop
        if node_percent is not None:
3783 53c776b5 Iustin Pop
          min_percent = min(min_percent, node_percent)
3784 53c776b5 Iustin Pop
      if not all_done:
3785 53c776b5 Iustin Pop
        if min_percent < 100:
3786 53c776b5 Iustin Pop
          self.feedback_fn("   - progress: %.1f%%" % min_percent)
3787 53c776b5 Iustin Pop
        time.sleep(2)
3788 53c776b5 Iustin Pop
3789 53c776b5 Iustin Pop
  def _EnsureSecondary(self, node):
3790 53c776b5 Iustin Pop
    """Demote a node to secondary.
3791 53c776b5 Iustin Pop

3792 53c776b5 Iustin Pop
    """
3793 53c776b5 Iustin Pop
    self.feedback_fn("* switching node %s to secondary mode" % node)
3794 53c776b5 Iustin Pop
3795 53c776b5 Iustin Pop
    for dev in self.instance.disks:
3796 53c776b5 Iustin Pop
      self.cfg.SetDiskID(dev, node)
3797 53c776b5 Iustin Pop
3798 53c776b5 Iustin Pop
    result = self.rpc.call_blockdev_close(node, self.instance.name,
3799 53c776b5 Iustin Pop
                                          self.instance.disks)
3800 53c776b5 Iustin Pop
    msg = result.RemoteFailMsg()
3801 53c776b5 Iustin Pop
    if msg:
3802 53c776b5 Iustin Pop
      raise errors.OpExecError("Cannot change disk to secondary on node %s,"
3803 53c776b5 Iustin Pop
                               " error %s" % (node, msg))
3804 53c776b5 Iustin Pop
3805 53c776b5 Iustin Pop
  def _GoStandalone(self):
3806 53c776b5 Iustin Pop
    """Disconnect from the network.
3807 53c776b5 Iustin Pop

3808 53c776b5 Iustin Pop
    """
3809 53c776b5 Iustin Pop
    self.feedback_fn("* changing into standalone mode")
3810 53c776b5 Iustin Pop
    result = self.rpc.call_drbd_disconnect_net(self.all_nodes, self.nodes_ip,
3811 53c776b5 Iustin Pop
                                               self.instance.disks)
3812 53c776b5 Iustin Pop
    for node, nres in result.items():
3813 53c776b5 Iustin Pop
      msg = nres.RemoteFailMsg()
3814 53c776b5 Iustin Pop
      if msg:
3815 53c776b5 Iustin Pop
        raise errors.OpExecError("Cannot disconnect disks node %s,"
3816 53c776b5 Iustin Pop
                                 " error %s" % (node, msg))
3817 53c776b5 Iustin Pop
3818 53c776b5 Iustin Pop
  def _GoReconnect(self, multimaster):
3819 53c776b5 Iustin Pop
    """Reconnect to the network.
3820 53c776b5 Iustin Pop

3821 53c776b5 Iustin Pop
    """
3822 53c776b5 Iustin Pop
    if multimaster:
3823 53c776b5 Iustin Pop
      msg = "dual-master"
3824 53c776b5 Iustin Pop
    else:
3825 53c776b5 Iustin Pop
      msg = "single-master"
3826 53c776b5 Iustin Pop
    self.feedback_fn("* changing disks into %s mode" % msg)
3827 53c776b5 Iustin Pop
    result = self.rpc.call_drbd_attach_net(self.all_nodes, self.nodes_ip,
3828 53c776b5 Iustin Pop
                                           self.instance.disks,
3829 53c776b5 Iustin Pop
                                           self.instance.name, multimaster)
3830 53c776b5 Iustin Pop
    for node, nres in result.items():
3831 53c776b5 Iustin Pop
      msg = nres.RemoteFailMsg()
3832 53c776b5 Iustin Pop
      if msg:
3833 53c776b5 Iustin Pop
        raise errors.OpExecError("Cannot change disks config on node %s,"
3834 53c776b5 Iustin Pop
                                 " error: %s" % (node, msg))
3835 53c776b5 Iustin Pop
3836 53c776b5 Iustin Pop
  def _ExecCleanup(self):
3837 53c776b5 Iustin Pop
    """Try to cleanup after a failed migration.
3838 53c776b5 Iustin Pop

3839 53c776b5 Iustin Pop
    The cleanup is done by:
3840 53c776b5 Iustin Pop
      - check that the instance is running only on one node
3841 53c776b5 Iustin Pop
        (and update the config if needed)
3842 53c776b5 Iustin Pop
      - change disks on its secondary node to secondary
3843 53c776b5 Iustin Pop
      - wait until disks are fully synchronized
3844 53c776b5 Iustin Pop
      - disconnect from the network
3845 53c776b5 Iustin Pop
      - change disks into single-master mode
3846 53c776b5 Iustin Pop
      - wait again until disks are fully synchronized
3847 53c776b5 Iustin Pop

3848 53c776b5 Iustin Pop
    """
3849 53c776b5 Iustin Pop
    instance = self.instance
3850 53c776b5 Iustin Pop
    target_node = self.target_node
3851 53c776b5 Iustin Pop
    source_node = self.source_node
3852 53c776b5 Iustin Pop
3853 53c776b5 Iustin Pop
    # check running on only one node
3854 53c776b5 Iustin Pop
    self.feedback_fn("* checking where the instance actually runs"
3855 53c776b5 Iustin Pop
                     " (if this hangs, the hypervisor might be in"
3856 53c776b5 Iustin Pop
                     " a bad state)")
3857 53c776b5 Iustin Pop
    ins_l = self.rpc.call_instance_list(self.all_nodes, [instance.hypervisor])
3858 53c776b5 Iustin Pop
    for node, result in ins_l.items():
3859 53c776b5 Iustin Pop
      result.Raise()
3860 53c776b5 Iustin Pop
      if not isinstance(result.data, list):
3861 53c776b5 Iustin Pop
        raise errors.OpExecError("Can't contact node '%s'" % node)
3862 53c776b5 Iustin Pop
3863 53c776b5 Iustin Pop
    runningon_source = instance.name in ins_l[source_node].data
3864 53c776b5 Iustin Pop
    runningon_target = instance.name in ins_l[target_node].data
3865 53c776b5 Iustin Pop
3866 53c776b5 Iustin Pop
    if runningon_source and runningon_target:
3867 53c776b5 Iustin Pop
      raise errors.OpExecError("Instance seems to be running on two nodes,"
3868 53c776b5 Iustin Pop
                               " or the hypervisor is confused. You will have"
3869 53c776b5 Iustin Pop
                               " to ensure manually that it runs only on one"
3870 53c776b5 Iustin Pop
                               " and restart this operation.")
3871 53c776b5 Iustin Pop
3872 53c776b5 Iustin Pop
    if not (runningon_source or runningon_target):
3873 53c776b5 Iustin Pop
      raise errors.OpExecError("Instance does not seem to be running at all."
3874 53c776b5 Iustin Pop
                               " In this case, it's safer to repair by"
3875 53c776b5 Iustin Pop
                               " running 'gnt-instance stop' to ensure disk"
3876 53c776b5 Iustin Pop
                               " shutdown, and then restarting it.")
3877 53c776b5 Iustin Pop
3878 53c776b5 Iustin Pop
    if runningon_target:
3879 53c776b5 Iustin Pop
      # the migration has actually succeeded, we need to update the config
3880 53c776b5 Iustin Pop
      self.feedback_fn("* instance running on secondary node (%s),"
3881 53c776b5 Iustin Pop
                       " updating config" % target_node)
3882 53c776b5 Iustin Pop
      instance.primary_node = target_node
3883 53c776b5 Iustin Pop
      self.cfg.Update(instance)
3884 53c776b5 Iustin Pop
      demoted_node = source_node
3885 53c776b5 Iustin Pop
    else:
3886 53c776b5 Iustin Pop
      self.feedback_fn("* instance confirmed to be running on its"
3887 53c776b5 Iustin Pop
                       " primary node (%s)" % source_node)
3888 53c776b5 Iustin Pop
      demoted_node = target_node
3889 53c776b5 Iustin Pop
3890 53c776b5 Iustin Pop
    self._EnsureSecondary(demoted_node)
3891 53c776b5 Iustin Pop
    try:
3892 53c776b5 Iustin Pop
      self._WaitUntilSync()
3893 53c776b5 Iustin Pop
    except errors.OpExecError:
3894 53c776b5 Iustin Pop
      # we ignore here errors, since if the device is standalone, it
3895 53c776b5 Iustin Pop
      # won't be able to sync
3896 53c776b5 Iustin Pop
      pass
3897 53c776b5 Iustin Pop
    self._GoStandalone()
3898 53c776b5 Iustin Pop
    self._GoReconnect(False)
3899 53c776b5 Iustin Pop
    self._WaitUntilSync()
3900 53c776b5 Iustin Pop
3901 53c776b5 Iustin Pop
    self.feedback_fn("* done")
3902 53c776b5 Iustin Pop
3903 6906a9d8 Guido Trotter
  def _RevertDiskStatus(self):
3904 6906a9d8 Guido Trotter
    """Try to revert the disk status after a failed migration.
3905 6906a9d8 Guido Trotter

3906 6906a9d8 Guido Trotter
    """
3907 6906a9d8 Guido Trotter
    target_node = self.target_node
3908 6906a9d8 Guido Trotter
    try:
3909 6906a9d8 Guido Trotter
      self._EnsureSecondary(target_node)
3910 6906a9d8 Guido Trotter
      self._GoStandalone()
3911 6906a9d8 Guido Trotter
      self._GoReconnect(False)
3912 6906a9d8 Guido Trotter
      self._WaitUntilSync()
3913 6906a9d8 Guido Trotter
    except errors.OpExecError, err:
3914 6906a9d8 Guido Trotter
      self.LogWarning("Migration failed and I can't reconnect the"
3915 6906a9d8 Guido Trotter
                      " drives: error '%s'\n"
3916 6906a9d8 Guido Trotter
                      "Please look and recover the instance status" %
3917 6906a9d8 Guido Trotter
                      str(err))
3918 6906a9d8 Guido Trotter
3919 6906a9d8 Guido Trotter
  def _AbortMigration(self):
3920 6906a9d8 Guido Trotter
    """Call the hypervisor code to abort a started migration.
3921 6906a9d8 Guido Trotter

3922 6906a9d8 Guido Trotter
    """
3923 6906a9d8 Guido Trotter
    instance = self.instance
3924 6906a9d8 Guido Trotter
    target_node = self.target_node
3925 6906a9d8 Guido Trotter
    migration_info = self.migration_info
3926 6906a9d8 Guido Trotter
3927 6906a9d8 Guido Trotter
    abort_result = self.rpc.call_finalize_migration(target_node,
3928 6906a9d8 Guido Trotter
                                                    instance,
3929 6906a9d8 Guido Trotter
                                                    migration_info,
3930 6906a9d8 Guido Trotter
                                                    False)
3931 6906a9d8 Guido Trotter
    abort_msg = abort_result.RemoteFailMsg()
3932 6906a9d8 Guido Trotter
    if abort_msg:
3933 6906a9d8 Guido Trotter
      logging.error("Aborting migration failed on target node %s: %s" %
3934 6906a9d8 Guido Trotter
                    (target_node, abort_msg))
3935 6906a9d8 Guido Trotter
      # Don't raise an exception here, as we stil have to try to revert the
3936 6906a9d8 Guido Trotter
      # disk status, even if this step failed.
3937 6906a9d8 Guido Trotter
3938 53c776b5 Iustin Pop
  def _ExecMigration(self):
3939 53c776b5 Iustin Pop
    """Migrate an instance.
3940 53c776b5 Iustin Pop

3941 53c776b5 Iustin Pop
    The migrate is done by:
3942 53c776b5 Iustin Pop
      - change the disks into dual-master mode
3943 53c776b5 Iustin Pop
      - wait until disks are fully synchronized again
3944 53c776b5 Iustin Pop
      - migrate the instance
3945 53c776b5 Iustin Pop
      - change disks on the new secondary node (the old primary) to secondary
3946 53c776b5 Iustin Pop
      - wait until disks are fully synchronized
3947 53c776b5 Iustin Pop
      - change disks into single-master mode
3948 53c776b5 Iustin Pop

3949 53c776b5 Iustin Pop
    """
3950 53c776b5 Iustin Pop
    instance = self.instance
3951 53c776b5 Iustin Pop
    target_node = self.target_node
3952 53c776b5 Iustin Pop
    source_node = self.source_node
3953 53c776b5 Iustin Pop
3954 53c776b5 Iustin Pop
    self.feedback_fn("* checking disk consistency between source and target")
3955 53c776b5 Iustin Pop
    for dev in instance.disks:
3956 53c776b5 Iustin Pop
      if not _CheckDiskConsistency(self, dev, target_node, False):
3957 53c776b5 Iustin Pop
        raise errors.OpExecError("Disk %s is degraded or not fully"
3958 53c776b5 Iustin Pop
                                 " synchronized on target node,"
3959 53c776b5 Iustin Pop
                                 " aborting migrate." % dev.iv_name)
3960 53c776b5 Iustin Pop
3961 6906a9d8 Guido Trotter
    # First get the migration information from the remote node
3962 6906a9d8 Guido Trotter
    result = self.rpc.call_migration_info(source_node, instance)
3963 6906a9d8 Guido Trotter
    msg = result.RemoteFailMsg()
3964 6906a9d8 Guido Trotter
    if msg:
3965 6906a9d8 Guido Trotter
      log_err = ("Failed fetching source migration information from %s: %s" %
3966 0959c824 Iustin Pop
                 (source_node, msg))
3967 6906a9d8 Guido Trotter
      logging.error(log_err)
3968 6906a9d8 Guido Trotter
      raise errors.OpExecError(log_err)
3969 6906a9d8 Guido Trotter
3970 0959c824 Iustin Pop
    self.migration_info = migration_info = result.payload
3971 6906a9d8 Guido Trotter
3972 6906a9d8 Guido Trotter
    # Then switch the disks to master/master mode
3973 53c776b5 Iustin Pop
    self._EnsureSecondary(target_node)
3974 53c776b5 Iustin Pop
    self._GoStandalone()
3975 53c776b5 Iustin Pop
    self._GoReconnect(True)
3976 53c776b5 Iustin Pop
    self._WaitUntilSync()
3977 53c776b5 Iustin Pop
3978 6906a9d8 Guido Trotter
    self.feedback_fn("* preparing %s to accept the instance" % target_node)
3979 6906a9d8 Guido Trotter
    result = self.rpc.call_accept_instance(target_node,
3980 6906a9d8 Guido Trotter
                                           instance,
3981 6906a9d8 Guido Trotter
                                           migration_info,
3982 6906a9d8 Guido Trotter
                                           self.nodes_ip[target_node])
3983 6906a9d8 Guido Trotter
3984 6906a9d8 Guido Trotter
    msg = result.RemoteFailMsg()
3985 6906a9d8 Guido Trotter
    if msg:
3986 6906a9d8 Guido Trotter
      logging.error("Instance pre-migration failed, trying to revert"
3987 6906a9d8 Guido Trotter
                    " disk status: %s", msg)
3988 6906a9d8 Guido Trotter
      self._AbortMigration()
3989 6906a9d8 Guido Trotter
      self._RevertDiskStatus()
3990 6906a9d8 Guido Trotter
      raise errors.OpExecError("Could not pre-migrate instance %s: %s" %
3991 6906a9d8 Guido Trotter
                               (instance.name, msg))
3992 6906a9d8 Guido Trotter
3993 53c776b5 Iustin Pop
    self.feedback_fn("* migrating instance to %s" % target_node)
3994 53c776b5 Iustin Pop
    time.sleep(10)
3995 53c776b5 Iustin Pop
    result = self.rpc.call_instance_migrate(source_node, instance,
3996 53c776b5 Iustin Pop
                                            self.nodes_ip[target_node],
3997 53c776b5 Iustin Pop
                                            self.op.live)
3998 53c776b5 Iustin Pop
    msg = result.RemoteFailMsg()
3999 53c776b5 Iustin Pop
    if msg:
4000 53c776b5 Iustin Pop
      logging.error("Instance migration failed, trying to revert"
4001 53c776b5 Iustin Pop
                    " disk status: %s", msg)
4002 6906a9d8 Guido Trotter
      self._AbortMigration()
4003 6906a9d8 Guido Trotter
      self._RevertDiskStatus()
4004 53c776b5 Iustin Pop
      raise errors.OpExecError("Could not migrate instance %s: %s" %
4005 53c776b5 Iustin Pop
                               (instance.name, msg))
4006 53c776b5 Iustin Pop
    time.sleep(10)
4007 53c776b5 Iustin Pop
4008 53c776b5 Iustin Pop
    instance.primary_node = target_node
4009 53c776b5 Iustin Pop
    # distribute new instance config to the other nodes
4010 53c776b5 Iustin Pop
    self.cfg.Update(instance)
4011 53c776b5 Iustin Pop
4012 6906a9d8 Guido Trotter
    result = self.rpc.call_finalize_migration(target_node,
4013 6906a9d8 Guido Trotter
                                              instance,
4014 6906a9d8 Guido Trotter
                                              migration_info,
4015 6906a9d8 Guido Trotter
                                              True)
4016 6906a9d8 Guido Trotter
    msg = result.RemoteFailMsg()
4017 6906a9d8 Guido Trotter
    if msg:
4018 6906a9d8 Guido Trotter
      logging.error("Instance migration succeeded, but finalization failed:"
4019 6906a9d8 Guido Trotter
                    " %s" % msg)
4020 6906a9d8 Guido Trotter
      raise errors.OpExecError("Could not finalize instance migration: %s" %
4021 6906a9d8 Guido Trotter
                               msg)
4022 6906a9d8 Guido Trotter
4023 53c776b5 Iustin Pop
    self._EnsureSecondary(source_node)
4024 53c776b5 Iustin Pop
    self._WaitUntilSync()
4025 53c776b5 Iustin Pop
    self._GoStandalone()
4026 53c776b5 Iustin Pop
    self._GoReconnect(False)
4027 53c776b5 Iustin Pop
    self._WaitUntilSync()
4028 53c776b5 Iustin Pop
4029 53c776b5 Iustin Pop
    self.feedback_fn("* done")
4030 53c776b5 Iustin Pop
4031 53c776b5 Iustin Pop
  def Exec(self, feedback_fn):
4032 53c776b5 Iustin Pop
    """Perform the migration.
4033 53c776b5 Iustin Pop

4034 53c776b5 Iustin Pop
    """
4035 53c776b5 Iustin Pop
    self.feedback_fn = feedback_fn
4036 53c776b5 Iustin Pop
4037 53c776b5 Iustin Pop
    self.source_node = self.instance.primary_node
4038 53c776b5 Iustin Pop
    self.target_node = self.instance.secondary_nodes[0]
4039 53c776b5 Iustin Pop
    self.all_nodes = [self.source_node, self.target_node]
4040 53c776b5 Iustin Pop
    self.nodes_ip = {
4041 53c776b5 Iustin Pop
      self.source_node: self.cfg.GetNodeInfo(self.source_node).secondary_ip,
4042 53c776b5 Iustin Pop
      self.target_node: self.cfg.GetNodeInfo(self.target_node).secondary_ip,
4043 53c776b5 Iustin Pop
      }
4044 53c776b5 Iustin Pop
    if self.op.cleanup:
4045 53c776b5 Iustin Pop
      return self._ExecCleanup()
4046 53c776b5 Iustin Pop
    else:
4047 53c776b5 Iustin Pop
      return self._ExecMigration()
4048 53c776b5 Iustin Pop
4049 53c776b5 Iustin Pop
4050 428958aa Iustin Pop
def _CreateBlockDev(lu, node, instance, device, force_create,
4051 428958aa Iustin Pop
                    info, force_open):
4052 428958aa Iustin Pop
  """Create a tree of block devices on a given node.
4053 a8083063 Iustin Pop

4054 a8083063 Iustin Pop
  If this device type has to be created on secondaries, create it and
4055 a8083063 Iustin Pop
  all its children.
4056 a8083063 Iustin Pop

4057 a8083063 Iustin Pop
  If not, just recurse to children keeping the same 'force' value.
4058 a8083063 Iustin Pop

4059 428958aa Iustin Pop
  @param lu: the lu on whose behalf we execute
4060 428958aa Iustin Pop
  @param node: the node on which to create the device
4061 428958aa Iustin Pop
  @type instance: L{objects.Instance}
4062 428958aa Iustin Pop
  @param instance: the instance which owns the device
4063 428958aa Iustin Pop
  @type device: L{objects.Disk}
4064 428958aa Iustin Pop
  @param device: the device to create
4065 428958aa Iustin Pop
  @type force_create: boolean
4066 428958aa Iustin Pop
  @param force_create: whether to force creation of this device; this
4067 428958aa Iustin Pop
      will be change to True whenever we find a device which has
4068 428958aa Iustin Pop
      CreateOnSecondary() attribute
4069 428958aa Iustin Pop
  @param info: the extra 'metadata' we should attach to the device
4070 428958aa Iustin Pop
      (this will be represented as a LVM tag)
4071 428958aa Iustin Pop
  @type force_open: boolean
4072 428958aa Iustin Pop
  @param force_open: this parameter will be passes to the
4073 821d1bd1 Iustin Pop
      L{backend.BlockdevCreate} function where it specifies
4074 428958aa Iustin Pop
      whether we run on primary or not, and it affects both
4075 428958aa Iustin Pop
      the child assembly and the device own Open() execution
4076 428958aa Iustin Pop

4077 a8083063 Iustin Pop
  """
4078 a8083063 Iustin Pop
  if device.CreateOnSecondary():
4079 428958aa Iustin Pop
    force_create = True
4080 796cab27 Iustin Pop
4081 a8083063 Iustin Pop
  if device.children:
4082 a8083063 Iustin Pop
    for child in device.children:
4083 428958aa Iustin Pop
      _CreateBlockDev(lu, node, instance, child, force_create,
4084 428958aa Iustin Pop
                      info, force_open)
4085 a8083063 Iustin Pop
4086 428958aa Iustin Pop
  if not force_create:
4087 796cab27 Iustin Pop
    return
4088 796cab27 Iustin Pop
4089 de12473a Iustin Pop
  _CreateSingleBlockDev(lu, node, instance, device, info, force_open)
4090 de12473a Iustin Pop
4091 de12473a Iustin Pop
4092 de12473a Iustin Pop
def _CreateSingleBlockDev(lu, node, instance, device, info, force_open):
4093 de12473a Iustin Pop
  """Create a single block device on a given node.
4094 de12473a Iustin Pop

4095 de12473a Iustin Pop
  This will not recurse over children of the device, so they must be
4096 de12473a Iustin Pop
  created in advance.
4097 de12473a Iustin Pop

4098 de12473a Iustin Pop
  @param lu: the lu on whose behalf we execute
4099 de12473a Iustin Pop
  @param node: the node on which to create the device
4100 de12473a Iustin Pop
  @type instance: L{objects.Instance}
4101 de12473a Iustin Pop
  @param instance: the instance which owns the device
4102 de12473a Iustin Pop
  @type device: L{objects.Disk}
4103 de12473a Iustin Pop
  @param device: the device to create
4104 de12473a Iustin Pop
  @param info: the extra 'metadata' we should attach to the device
4105 de12473a Iustin Pop
      (this will be represented as a LVM tag)
4106 de12473a Iustin Pop
  @type force_open: boolean
4107 de12473a Iustin Pop
  @param force_open: this parameter will be passes to the
4108 821d1bd1 Iustin Pop
      L{backend.BlockdevCreate} function where it specifies
4109 de12473a Iustin Pop
      whether we run on primary or not, and it affects both
4110 de12473a Iustin Pop
      the child assembly and the device own Open() execution
4111 de12473a Iustin Pop

4112 de12473a Iustin Pop
  """
4113 b9bddb6b Iustin Pop
  lu.cfg.SetDiskID(device, node)
4114 7d81697f Iustin Pop
  result = lu.rpc.call_blockdev_create(node, device, device.size,
4115 428958aa Iustin Pop
                                       instance.name, force_open, info)
4116 7d81697f Iustin Pop
  msg = result.RemoteFailMsg()
4117 7d81697f Iustin Pop
  if msg:
4118 428958aa Iustin Pop
    raise errors.OpExecError("Can't create block device %s on"
4119 7d81697f Iustin Pop
                             " node %s for instance %s: %s" %
4120 7d81697f Iustin Pop
                             (device, node, instance.name, msg))
4121 a8083063 Iustin Pop
  if device.physical_id is None:
4122 0959c824 Iustin Pop
    device.physical_id = result.payload
4123 a8083063 Iustin Pop
4124 a8083063 Iustin Pop
4125 b9bddb6b Iustin Pop
def _GenerateUniqueNames(lu, exts):
4126 923b1523 Iustin Pop
  """Generate a suitable LV name.
4127 923b1523 Iustin Pop

4128 923b1523 Iustin Pop
  This will generate a logical volume name for the given instance.
4129 923b1523 Iustin Pop

4130 923b1523 Iustin Pop
  """
4131 923b1523 Iustin Pop
  results = []
4132 923b1523 Iustin Pop
  for val in exts:
4133 b9bddb6b Iustin Pop
    new_id = lu.cfg.GenerateUniqueID()
4134 923b1523 Iustin Pop
    results.append("%s%s" % (new_id, val))
4135 923b1523 Iustin Pop
  return results
4136 923b1523 Iustin Pop
4137 923b1523 Iustin Pop
4138 b9bddb6b Iustin Pop
def _GenerateDRBD8Branch(lu, primary, secondary, size, names, iv_name,
4139 ffa1c0dc Iustin Pop
                         p_minor, s_minor):
4140 a1f445d3 Iustin Pop
  """Generate a drbd8 device complete with its children.
4141 a1f445d3 Iustin Pop

4142 a1f445d3 Iustin Pop
  """
4143 b9bddb6b Iustin Pop
  port = lu.cfg.AllocatePort()
4144 b9bddb6b Iustin Pop
  vgname = lu.cfg.GetVGName()
4145 b9bddb6b Iustin Pop
  shared_secret = lu.cfg.GenerateDRBDSecret()
4146 a1f445d3 Iustin Pop
  dev_data = objects.Disk(dev_type=constants.LD_LV, size=size,
4147 a1f445d3 Iustin Pop
                          logical_id=(vgname, names[0]))
4148 a1f445d3 Iustin Pop
  dev_meta = objects.Disk(dev_type=constants.LD_LV, size=128,
4149 a1f445d3 Iustin Pop
                          logical_id=(vgname, names[1]))
4150 a1f445d3 Iustin Pop
  drbd_dev = objects.Disk(dev_type=constants.LD_DRBD8, size=size,
4151 ffa1c0dc Iustin Pop
                          logical_id=(primary, secondary, port,
4152 f9518d38 Iustin Pop
                                      p_minor, s_minor,
4153 f9518d38 Iustin Pop
                                      shared_secret),
4154 ffa1c0dc Iustin Pop
                          children=[dev_data, dev_meta],
4155 a1f445d3 Iustin Pop
                          iv_name=iv_name)
4156 a1f445d3 Iustin Pop
  return drbd_dev
4157 a1f445d3 Iustin Pop
4158 7c0d6283 Michael Hanselmann
4159 b9bddb6b Iustin Pop
def _GenerateDiskTemplate(lu, template_name,
4160 a8083063 Iustin Pop
                          instance_name, primary_node,
4161 08db7c5c Iustin Pop
                          secondary_nodes, disk_info,
4162 e2a65344 Iustin Pop
                          file_storage_dir, file_driver,
4163 e2a65344 Iustin Pop
                          base_index):
4164 a8083063 Iustin Pop
  """Generate the entire disk layout for a given template type.
4165 a8083063 Iustin Pop

4166 a8083063 Iustin Pop
  """
4167 a8083063 Iustin Pop
  #TODO: compute space requirements
4168 a8083063 Iustin Pop
4169 b9bddb6b Iustin Pop
  vgname = lu.cfg.GetVGName()
4170 08db7c5c Iustin Pop
  disk_count = len(disk_info)
4171 08db7c5c Iustin Pop
  disks = []
4172 3517d9b9 Manuel Franceschini
  if template_name == constants.DT_DISKLESS:
4173 08db7c5c Iustin Pop
    pass
4174 3517d9b9 Manuel Franceschini
  elif template_name == constants.DT_PLAIN:
4175 a8083063 Iustin Pop
    if len(secondary_nodes) != 0:
4176 a8083063 Iustin Pop
      raise errors.ProgrammerError("Wrong template configuration")
4177 923b1523 Iustin Pop
4178 08db7c5c Iustin Pop
    names = _GenerateUniqueNames(lu, [".disk%d" % i
4179 08db7c5c Iustin Pop
                                      for i in range(disk_count)])
4180 08db7c5c Iustin Pop
    for idx, disk in enumerate(disk_info):
4181 e2a65344 Iustin Pop
      disk_index = idx + base_index
4182 08db7c5c Iustin Pop
      disk_dev = objects.Disk(dev_type=constants.LD_LV, size=disk["size"],
4183 08db7c5c Iustin Pop
                              logical_id=(vgname, names[idx]),
4184 6ec66eae Iustin Pop
                              iv_name="disk/%d" % disk_index,
4185 6ec66eae Iustin Pop
                              mode=disk["mode"])
4186 08db7c5c Iustin Pop
      disks.append(disk_dev)
4187 a1f445d3 Iustin Pop
  elif template_name == constants.DT_DRBD8:
4188 a1f445d3 Iustin Pop
    if len(secondary_nodes) != 1:
4189 a1f445d3 Iustin Pop
      raise errors.ProgrammerError("Wrong template configuration")
4190 a1f445d3 Iustin Pop
    remote_node = secondary_nodes[0]
4191 08db7c5c Iustin Pop
    minors = lu.cfg.AllocateDRBDMinor(
4192 08db7c5c Iustin Pop
      [primary_node, remote_node] * len(disk_info), instance_name)
4193 08db7c5c Iustin Pop
4194 e6c1ff2f Iustin Pop
    names = []
4195 e6c1ff2f Iustin Pop
    for lv_prefix in _GenerateUniqueNames(lu, [".disk%d" % i
4196 e6c1ff2f Iustin Pop
                                               for i in range(disk_count)]):
4197 e6c1ff2f Iustin Pop
      names.append(lv_prefix + "_data")
4198 e6c1ff2f Iustin Pop
      names.append(lv_prefix + "_meta")
4199 08db7c5c Iustin Pop
    for idx, disk in enumerate(disk_info):
4200 112050d9 Iustin Pop
      disk_index = idx + base_index
4201 08db7c5c Iustin Pop
      disk_dev = _GenerateDRBD8Branch(lu, primary_node, remote_node,
4202 08db7c5c Iustin Pop
                                      disk["size"], names[idx*2:idx*2+2],
4203 e2a65344 Iustin Pop
                                      "disk/%d" % disk_index,
4204 08db7c5c Iustin Pop
                                      minors[idx*2], minors[idx*2+1])
4205 6ec66eae Iustin Pop
      disk_dev.mode = disk["mode"]
4206 08db7c5c Iustin Pop
      disks.append(disk_dev)
4207 0f1a06e3 Manuel Franceschini
  elif template_name == constants.DT_FILE:
4208 0f1a06e3 Manuel Franceschini
    if len(secondary_nodes) != 0:
4209 0f1a06e3 Manuel Franceschini
      raise errors.ProgrammerError("Wrong template configuration")
4210 0f1a06e3 Manuel Franceschini
4211 08db7c5c Iustin Pop
    for idx, disk in enumerate(disk_info):
4212 112050d9 Iustin Pop
      disk_index = idx + base_index
4213 08db7c5c Iustin Pop
      disk_dev = objects.Disk(dev_type=constants.LD_FILE, size=disk["size"],
4214 e2a65344 Iustin Pop
                              iv_name="disk/%d" % disk_index,
4215 08db7c5c Iustin Pop
                              logical_id=(file_driver,
4216 08db7c5c Iustin Pop
                                          "%s/disk%d" % (file_storage_dir,
4217 43e99cff Guido Trotter
                                                         disk_index)),
4218 6ec66eae Iustin Pop
                              mode=disk["mode"])
4219 08db7c5c Iustin Pop
      disks.append(disk_dev)
4220 a8083063 Iustin Pop
  else:
4221 a8083063 Iustin Pop
    raise errors.ProgrammerError("Invalid disk template '%s'" % template_name)
4222 a8083063 Iustin Pop
  return disks
4223 a8083063 Iustin Pop
4224 a8083063 Iustin Pop
4225 a0c3fea1 Michael Hanselmann
def _GetInstanceInfoText(instance):
4226 3ecf6786 Iustin Pop
  """Compute that text that should be added to the disk's metadata.
4227 3ecf6786 Iustin Pop

4228 3ecf6786 Iustin Pop
  """
4229 a0c3fea1 Michael Hanselmann
  return "originstname+%s" % instance.name
4230 a0c3fea1 Michael Hanselmann
4231 a0c3fea1 Michael Hanselmann
4232 b9bddb6b Iustin Pop
def _CreateDisks(lu, instance):
4233 a8083063 Iustin Pop
  """Create all disks for an instance.
4234 a8083063 Iustin Pop

4235 a8083063 Iustin Pop
  This abstracts away some work from AddInstance.
4236 a8083063 Iustin Pop

4237 e4376078 Iustin Pop
  @type lu: L{LogicalUnit}
4238 e4376078 Iustin Pop
  @param lu: the logical unit on whose behalf we execute
4239 e4376078 Iustin Pop
  @type instance: L{objects.Instance}
4240 e4376078 Iustin Pop
  @param instance: the instance whose disks we should create
4241 e4376078 Iustin Pop
  @rtype: boolean
4242 e4376078 Iustin Pop
  @return: the success of the creation
4243 a8083063 Iustin Pop

4244 a8083063 Iustin Pop
  """
4245 a0c3fea1 Michael Hanselmann
  info = _GetInstanceInfoText(instance)
4246 428958aa Iustin Pop
  pnode = instance.primary_node
4247 a0c3fea1 Michael Hanselmann
4248 0f1a06e3 Manuel Franceschini
  if instance.disk_template == constants.DT_FILE:
4249 0f1a06e3 Manuel Franceschini
    file_storage_dir = os.path.dirname(instance.disks[0].logical_id[1])
4250 428958aa Iustin Pop
    result = lu.rpc.call_file_storage_dir_create(pnode, file_storage_dir)
4251 0f1a06e3 Manuel Franceschini
4252 781de953 Iustin Pop
    if result.failed or not result.data:
4253 428958aa Iustin Pop
      raise errors.OpExecError("Could not connect to node '%s'" % pnode)
4254 0f1a06e3 Manuel Franceschini
4255 781de953 Iustin Pop
    if not result.data[0]:
4256 796cab27 Iustin Pop
      raise errors.OpExecError("Failed to create directory '%s'" %
4257 796cab27 Iustin Pop
                               file_storage_dir)
4258 0f1a06e3 Manuel Franceschini
4259 24991749 Iustin Pop
  # Note: this needs to be kept in sync with adding of disks in
4260 24991749 Iustin Pop
  # LUSetInstanceParams
4261 a8083063 Iustin Pop
  for device in instance.disks:
4262 9a4f63d1 Iustin Pop
    logging.info("Creating volume %s for instance %s",
4263 9a4f63d1 Iustin Pop
                 device.iv_name, instance.name)
4264 a8083063 Iustin Pop
    #HARDCODE
4265 428958aa Iustin Pop
    for node in instance.all_nodes:
4266 428958aa Iustin Pop
      f_create = node == pnode
4267 428958aa Iustin Pop
      _CreateBlockDev(lu, node, instance, device, f_create, info, f_create)
4268 a8083063 Iustin Pop
4269 a8083063 Iustin Pop
4270 b9bddb6b Iustin Pop
def _RemoveDisks(lu, instance):
4271 a8083063 Iustin Pop
  """Remove all disks for an instance.
4272 a8083063 Iustin Pop

4273 a8083063 Iustin Pop
  This abstracts away some work from `AddInstance()` and
4274 a8083063 Iustin Pop
  `RemoveInstance()`. Note that in case some of the devices couldn't
4275 1d67656e Iustin Pop
  be removed, the removal will continue with the other ones (compare
4276 a8083063 Iustin Pop
  with `_CreateDisks()`).
4277 a8083063 Iustin Pop

4278 e4376078 Iustin Pop
  @type lu: L{LogicalUnit}
4279 e4376078 Iustin Pop
  @param lu: the logical unit on whose behalf we execute
4280 e4376078 Iustin Pop
  @type instance: L{objects.Instance}
4281 e4376078 Iustin Pop
  @param instance: the instance whose disks we should remove
4282 e4376078 Iustin Pop
  @rtype: boolean
4283 e4376078 Iustin Pop
  @return: the success of the removal
4284 a8083063 Iustin Pop

4285 a8083063 Iustin Pop
  """
4286 9a4f63d1 Iustin Pop
  logging.info("Removing block devices for instance %s", instance.name)
4287 a8083063 Iustin Pop
4288 e1bc0878 Iustin Pop
  all_result = True
4289 a8083063 Iustin Pop
  for device in instance.disks:
4290 a8083063 Iustin Pop
    for node, disk in device.ComputeNodeTree(instance.primary_node):
4291 b9bddb6b Iustin Pop
      lu.cfg.SetDiskID(disk, node)
4292 e1bc0878 Iustin Pop
      msg = lu.rpc.call_blockdev_remove(node, disk).RemoteFailMsg()
4293 e1bc0878 Iustin Pop
      if msg:
4294 e1bc0878 Iustin Pop
        lu.LogWarning("Could not remove block device %s on node %s,"
4295 e1bc0878 Iustin Pop
                      " continuing anyway: %s", device.iv_name, node, msg)
4296 e1bc0878 Iustin Pop
        all_result = False
4297 0f1a06e3 Manuel Franceschini
4298 0f1a06e3 Manuel Franceschini
  if instance.disk_template == constants.DT_FILE:
4299 0f1a06e3 Manuel Franceschini
    file_storage_dir = os.path.dirname(instance.disks[0].logical_id[1])
4300 781de953 Iustin Pop
    result = lu.rpc.call_file_storage_dir_remove(instance.primary_node,
4301 781de953 Iustin Pop
                                                 file_storage_dir)
4302 781de953 Iustin Pop
    if result.failed or not result.data:
4303 9a4f63d1 Iustin Pop
      logging.error("Could not remove directory '%s'", file_storage_dir)
4304 e1bc0878 Iustin Pop
      all_result = False
4305 0f1a06e3 Manuel Franceschini
4306 e1bc0878 Iustin Pop
  return all_result
4307 a8083063 Iustin Pop
4308 a8083063 Iustin Pop
4309 08db7c5c Iustin Pop
def _ComputeDiskSize(disk_template, disks):
4310 e2fe6369 Iustin Pop
  """Compute disk size requirements in the volume group
4311 e2fe6369 Iustin Pop

4312 e2fe6369 Iustin Pop
  """
4313 e2fe6369 Iustin Pop
  # Required free disk space as a function of disk and swap space
4314 e2fe6369 Iustin Pop
  req_size_dict = {
4315 e2fe6369 Iustin Pop
    constants.DT_DISKLESS: None,
4316 08db7c5c Iustin Pop
    constants.DT_PLAIN: sum(d["size"] for d in disks),
4317 08db7c5c Iustin Pop
    # 128 MB are added for drbd metadata for each disk
4318 08db7c5c Iustin Pop
    constants.DT_DRBD8: sum(d["size"] + 128 for d in disks),
4319 e2fe6369 Iustin Pop
    constants.DT_FILE: None,
4320 e2fe6369 Iustin Pop
  }
4321 e2fe6369 Iustin Pop
4322 e2fe6369 Iustin Pop
  if disk_template not in req_size_dict:
4323 e2fe6369 Iustin Pop
    raise errors.ProgrammerError("Disk template '%s' size requirement"
4324 e2fe6369 Iustin Pop
                                 " is unknown" %  disk_template)
4325 e2fe6369 Iustin Pop
4326 e2fe6369 Iustin Pop
  return req_size_dict[disk_template]
4327 e2fe6369 Iustin Pop
4328 e2fe6369 Iustin Pop
4329 74409b12 Iustin Pop
def _CheckHVParams(lu, nodenames, hvname, hvparams):
4330 74409b12 Iustin Pop
  """Hypervisor parameter validation.
4331 74409b12 Iustin Pop

4332 74409b12 Iustin Pop
  This function abstract the hypervisor parameter validation to be
4333 74409b12 Iustin Pop
  used in both instance create and instance modify.
4334 74409b12 Iustin Pop

4335 74409b12 Iustin Pop
  @type lu: L{LogicalUnit}
4336 74409b12 Iustin Pop
  @param lu: the logical unit for which we check
4337 74409b12 Iustin Pop
  @type nodenames: list
4338 74409b12 Iustin Pop
  @param nodenames: the list of nodes on which we should check
4339 74409b12 Iustin Pop
  @type hvname: string
4340 74409b12 Iustin Pop
  @param hvname: the name of the hypervisor we should use
4341 74409b12 Iustin Pop
  @type hvparams: dict
4342 74409b12 Iustin Pop
  @param hvparams: the parameters which we need to check
4343 74409b12 Iustin Pop
  @raise errors.OpPrereqError: if the parameters are not valid
4344 74409b12 Iustin Pop

4345 74409b12 Iustin Pop
  """
4346 74409b12 Iustin Pop
  hvinfo = lu.rpc.call_hypervisor_validate_params(nodenames,
4347 74409b12 Iustin Pop
                                                  hvname,
4348 74409b12 Iustin Pop
                                                  hvparams)
4349 74409b12 Iustin Pop
  for node in nodenames:
4350 781de953 Iustin Pop
    info = hvinfo[node]
4351 68c6f21c Iustin Pop
    if info.offline:
4352 68c6f21c Iustin Pop
      continue
4353 0959c824 Iustin Pop
    msg = info.RemoteFailMsg()
4354 0959c824 Iustin Pop
    if msg:
4355 d64769a8 Iustin Pop
      raise errors.OpPrereqError("Hypervisor parameter validation"
4356 d64769a8 Iustin Pop
                                 " failed on node %s: %s" % (node, msg))
4357 74409b12 Iustin Pop
4358 74409b12 Iustin Pop
4359 a8083063 Iustin Pop
class LUCreateInstance(LogicalUnit):
4360 a8083063 Iustin Pop
  """Create an instance.
4361 a8083063 Iustin Pop

4362 a8083063 Iustin Pop
  """
4363 a8083063 Iustin Pop
  HPATH = "instance-add"
4364 a8083063 Iustin Pop
  HTYPE = constants.HTYPE_INSTANCE
4365 08db7c5c Iustin Pop
  _OP_REQP = ["instance_name", "disks", "disk_template",
4366 08db7c5c Iustin Pop
              "mode", "start",
4367 08db7c5c Iustin Pop
              "wait_for_sync", "ip_check", "nics",
4368 338e51e8 Iustin Pop
              "hvparams", "beparams"]
4369 7baf741d Guido Trotter
  REQ_BGL = False
4370 7baf741d Guido Trotter
4371 7baf741d Guido Trotter
  def _ExpandNode(self, node):
4372 7baf741d Guido Trotter
    """Expands and checks one node name.
4373 7baf741d Guido Trotter

4374 7baf741d Guido Trotter
    """
4375 7baf741d Guido Trotter
    node_full = self.cfg.ExpandNodeName(node)
4376 7baf741d Guido Trotter
    if node_full is None:
4377 7baf741d Guido Trotter
      raise errors.OpPrereqError("Unknown node %s" % node)
4378 7baf741d Guido Trotter
    return node_full
4379 7baf741d Guido Trotter
4380 7baf741d Guido Trotter
  def ExpandNames(self):
4381 7baf741d Guido Trotter
    """ExpandNames for CreateInstance.
4382 7baf741d Guido Trotter

4383 7baf741d Guido Trotter
    Figure out the right locks for instance creation.
4384 7baf741d Guido Trotter

4385 7baf741d Guido Trotter
    """
4386 7baf741d Guido Trotter
    self.needed_locks = {}
4387 7baf741d Guido Trotter
4388 7baf741d Guido Trotter
    # set optional parameters to none if they don't exist
4389 6785674e Iustin Pop
    for attr in ["pnode", "snode", "iallocator", "hypervisor"]:
4390 7baf741d Guido Trotter
      if not hasattr(self.op, attr):
4391 7baf741d Guido Trotter
        setattr(self.op, attr, None)
4392 7baf741d Guido Trotter
4393 4b2f38dd Iustin Pop
    # cheap checks, mostly valid constants given
4394 4b2f38dd Iustin Pop
4395 7baf741d Guido Trotter
    # verify creation mode
4396 7baf741d Guido Trotter
    if self.op.mode not in (constants.INSTANCE_CREATE,
4397 7baf741d Guido Trotter
                            constants.INSTANCE_IMPORT):
4398 7baf741d Guido Trotter
      raise errors.OpPrereqError("Invalid instance creation mode '%s'" %
4399 7baf741d Guido Trotter
                                 self.op.mode)
4400 4b2f38dd Iustin Pop
4401 7baf741d Guido Trotter
    # disk template and mirror node verification
4402 7baf741d Guido Trotter
    if self.op.disk_template not in constants.DISK_TEMPLATES:
4403 7baf741d Guido Trotter
      raise errors.OpPrereqError("Invalid disk template name")
4404 7baf741d Guido Trotter
4405 4b2f38dd Iustin Pop
    if self.op.hypervisor is None:
4406 4b2f38dd Iustin Pop
      self.op.hypervisor = self.cfg.GetHypervisorType()
4407 4b2f38dd Iustin Pop
4408 8705eb96 Iustin Pop
    cluster = self.cfg.GetClusterInfo()
4409 8705eb96 Iustin Pop
    enabled_hvs = cluster.enabled_hypervisors
4410 4b2f38dd Iustin Pop
    if self.op.hypervisor not in enabled_hvs:
4411 4b2f38dd Iustin Pop
      raise errors.OpPrereqError("Selected hypervisor (%s) not enabled in the"
4412 4b2f38dd Iustin Pop
                                 " cluster (%s)" % (self.op.hypervisor,
4413 4b2f38dd Iustin Pop
                                  ",".join(enabled_hvs)))
4414 4b2f38dd Iustin Pop
4415 6785674e Iustin Pop
    # check hypervisor parameter syntax (locally)
4416 a5728081 Guido Trotter
    utils.ForceDictType(self.op.hvparams, constants.HVS_PARAMETER_TYPES)
4417 abe609b2 Guido Trotter
    filled_hvp = objects.FillDict(cluster.hvparams[self.op.hypervisor],
4418 8705eb96 Iustin Pop
                                  self.op.hvparams)
4419 6785674e Iustin Pop
    hv_type = hypervisor.GetHypervisor(self.op.hypervisor)
4420 8705eb96 Iustin Pop
    hv_type.CheckParameterSyntax(filled_hvp)
4421 6785674e Iustin Pop
4422 338e51e8 Iustin Pop
    # fill and remember the beparams dict
4423 a5728081 Guido Trotter
    utils.ForceDictType(self.op.beparams, constants.BES_PARAMETER_TYPES)
4424 4ef7f423 Guido Trotter
    self.be_full = objects.FillDict(cluster.beparams[constants.PP_DEFAULT],
4425 338e51e8 Iustin Pop
                                    self.op.beparams)
4426 338e51e8 Iustin Pop
4427 7baf741d Guido Trotter
    #### instance parameters check
4428 7baf741d Guido Trotter
4429 7baf741d Guido Trotter
    # instance name verification
4430 7baf741d Guido Trotter
    hostname1 = utils.HostInfo(self.op.instance_name)
4431 7baf741d Guido Trotter
    self.op.instance_name = instance_name = hostname1.name
4432 7baf741d Guido Trotter
4433 7baf741d Guido Trotter
    # this is just a preventive check, but someone might still add this
4434 7baf741d Guido Trotter
    # instance in the meantime, and creation will fail at lock-add time
4435 7baf741d Guido Trotter
    if instance_name in self.cfg.GetInstanceList():
4436 7baf741d Guido Trotter
      raise errors.OpPrereqError("Instance '%s' is already in the cluster" %
4437 7baf741d Guido Trotter
                                 instance_name)
4438 7baf741d Guido Trotter
4439 7baf741d Guido Trotter
    self.add_locks[locking.LEVEL_INSTANCE] = instance_name
4440 7baf741d Guido Trotter
4441 08db7c5c Iustin Pop
    # NIC buildup
4442 08db7c5c Iustin Pop
    self.nics = []
4443 9dce4771 Guido Trotter
    for idx, nic in enumerate(self.op.nics):
4444 9dce4771 Guido Trotter
      nic_mode_req = nic.get("mode", None)
4445 9dce4771 Guido Trotter
      nic_mode = nic_mode_req
4446 9dce4771 Guido Trotter
      if nic_mode is None:
4447 9dce4771 Guido Trotter
        nic_mode = cluster.nicparams[constants.PP_DEFAULT][constants.NIC_MODE]
4448 9dce4771 Guido Trotter
4449 9dce4771 Guido Trotter
      # in routed mode, for the first nic, the default ip is 'auto'
4450 9dce4771 Guido Trotter
      if nic_mode == constants.NIC_MODE_ROUTED and idx == 0:
4451 9dce4771 Guido Trotter
        default_ip_mode = constants.VALUE_AUTO
4452 9dce4771 Guido Trotter
      else:
4453 9dce4771 Guido Trotter
        default_ip_mode = constants.VALUE_NONE
4454 9dce4771 Guido Trotter
4455 08db7c5c Iustin Pop
      # ip validity checks
4456 9dce4771 Guido Trotter
      ip = nic.get("ip", default_ip_mode)
4457 9dce4771 Guido Trotter
      if ip is None or ip.lower() == constants.VALUE_NONE:
4458 08db7c5c Iustin Pop
        nic_ip = None
4459 08db7c5c Iustin Pop
      elif ip.lower() == constants.VALUE_AUTO:
4460 08db7c5c Iustin Pop
        nic_ip = hostname1.ip
4461 08db7c5c Iustin Pop
      else:
4462 08db7c5c Iustin Pop
        if not utils.IsValidIP(ip):
4463 08db7c5c Iustin Pop
          raise errors.OpPrereqError("Given IP address '%s' doesn't look"
4464 08db7c5c Iustin Pop
                                     " like a valid IP" % ip)
4465 08db7c5c Iustin Pop
        nic_ip = ip
4466 08db7c5c Iustin Pop
4467 9dce4771 Guido Trotter
      # TODO: check the ip for uniqueness !!
4468 9dce4771 Guido Trotter
      if nic_mode == constants.NIC_MODE_ROUTED and not nic_ip:
4469 9dce4771 Guido Trotter
        raise errors.OpPrereqError("Routed nic mode requires an ip address")
4470 9dce4771 Guido Trotter
4471 08db7c5c Iustin Pop
      # MAC address verification
4472 08db7c5c Iustin Pop
      mac = nic.get("mac", constants.VALUE_AUTO)
4473 08db7c5c Iustin Pop
      if mac not in (constants.VALUE_AUTO, constants.VALUE_GENERATE):
4474 08db7c5c Iustin Pop
        if not utils.IsValidMac(mac.lower()):
4475 08db7c5c Iustin Pop
          raise errors.OpPrereqError("Invalid MAC address specified: %s" %
4476 08db7c5c Iustin Pop
                                     mac)
4477 08db7c5c Iustin Pop
      # bridge verification
4478 9939547b Iustin Pop
      bridge = nic.get("bridge", None)
4479 9dce4771 Guido Trotter
      link = nic.get("link", None)
4480 9dce4771 Guido Trotter
      if bridge and link:
4481 9dce4771 Guido Trotter
        raise errors.OpPrereqError("Cannot pass 'bridge' and 'link' at the same time")
4482 9dce4771 Guido Trotter
      elif bridge and nic_mode == constants.NIC_MODE_ROUTED:
4483 9dce4771 Guido Trotter
        raise errors.OpPrereqError("Cannot pass 'bridge' on a routed nic")
4484 9dce4771 Guido Trotter
      elif bridge:
4485 9dce4771 Guido Trotter
        link = bridge
4486 9dce4771 Guido Trotter
4487 9dce4771 Guido Trotter
      nicparams = {}
4488 9dce4771 Guido Trotter
      if nic_mode_req:
4489 9dce4771 Guido Trotter
        nicparams[constants.NIC_MODE] = nic_mode_req
4490 9dce4771 Guido Trotter
      if link:
4491 9dce4771 Guido Trotter
        nicparams[constants.NIC_LINK] = link
4492 9dce4771 Guido Trotter
4493 9dce4771 Guido Trotter
      check_params = objects.FillDict(cluster.nicparams[constants.PP_DEFAULT],
4494 9dce4771 Guido Trotter
                                      nicparams)
4495 9dce4771 Guido Trotter
      objects.NIC.CheckParameterSyntax(check_params)
4496 9dce4771 Guido Trotter
      self.nics.append(objects.NIC(mac=mac, ip=nic_ip, nicparams=nicparams))
4497 08db7c5c Iustin Pop
4498 08db7c5c Iustin Pop
    # disk checks/pre-build
4499 08db7c5c Iustin Pop
    self.disks = []
4500 08db7c5c Iustin Pop
    for disk in self.op.disks:
4501 08db7c5c Iustin Pop
      mode = disk.get("mode", constants.DISK_RDWR)
4502 08db7c5c Iustin Pop
      if mode not in constants.DISK_ACCESS_SET:
4503 08db7c5c Iustin Pop
        raise errors.OpPrereqError("Invalid disk access mode '%s'" %
4504 08db7c5c Iustin Pop
                                   mode)
4505 08db7c5c Iustin Pop
      size = disk.get("size", None)
4506 08db7c5c Iustin Pop
      if size is None:
4507 08db7c5c Iustin Pop
        raise errors.OpPrereqError("Missing disk size")
4508 08db7c5c Iustin Pop
      try:
4509 08db7c5c Iustin Pop
        size = int(size)
4510 08db7c5c Iustin Pop
      except ValueError:
4511 08db7c5c Iustin Pop
        raise errors.OpPrereqError("Invalid disk size '%s'" % size)
4512 08db7c5c Iustin Pop
      self.disks.append({"size": size, "mode": mode})
4513 08db7c5c Iustin Pop
4514 7baf741d Guido Trotter
    # used in CheckPrereq for ip ping check
4515 7baf741d Guido Trotter
    self.check_ip = hostname1.ip
4516 7baf741d Guido Trotter
4517 7baf741d Guido Trotter
    # file storage checks
4518 7baf741d Guido Trotter
    if (self.op.file_driver and
4519 7baf741d Guido Trotter
        not self.op.file_driver in constants.FILE_DRIVER):
4520 7baf741d Guido Trotter
      raise errors.OpPrereqError("Invalid file driver name '%s'" %
4521 7baf741d Guido Trotter
                                 self.op.file_driver)
4522 7baf741d Guido Trotter
4523 7baf741d Guido Trotter
    if self.op.file_storage_dir and os.path.isabs(self.op.file_storage_dir):
4524 7baf741d Guido Trotter
      raise errors.OpPrereqError("File storage directory path not absolute")
4525 7baf741d Guido Trotter
4526 7baf741d Guido Trotter
    ### Node/iallocator related checks
4527 7baf741d Guido Trotter
    if [self.op.iallocator, self.op.pnode].count(None) != 1:
4528 7baf741d Guido Trotter
      raise errors.OpPrereqError("One and only one of iallocator and primary"
4529 7baf741d Guido Trotter
                                 " node must be given")
4530 7baf741d Guido Trotter
4531 7baf741d Guido Trotter
    if self.op.iallocator:
4532 7baf741d Guido Trotter
      self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
4533 7baf741d Guido Trotter
    else:
4534 7baf741d Guido Trotter
      self.op.pnode = self._ExpandNode(self.op.pnode)
4535 7baf741d Guido Trotter
      nodelist = [self.op.pnode]
4536 7baf741d Guido Trotter
      if self.op.snode is not None:
4537 7baf741d Guido Trotter
        self.op.snode = self._ExpandNode(self.op.snode)
4538 7baf741d Guido Trotter
        nodelist.append(self.op.snode)
4539 7baf741d Guido Trotter
      self.needed_locks[locking.LEVEL_NODE] = nodelist
4540 7baf741d Guido Trotter
4541 7baf741d Guido Trotter
    # in case of import lock the source node too
4542 7baf741d Guido Trotter
    if self.op.mode == constants.INSTANCE_IMPORT:
4543 7baf741d Guido Trotter
      src_node = getattr(self.op, "src_node", None)
4544 7baf741d Guido Trotter
      src_path = getattr(self.op, "src_path", None)
4545 7baf741d Guido Trotter
4546 b9322a9f Guido Trotter
      if src_path is None:
4547 b9322a9f Guido Trotter
        self.op.src_path = src_path = self.op.instance_name
4548 b9322a9f Guido Trotter
4549 b9322a9f Guido Trotter
      if src_node is None:
4550 b9322a9f Guido Trotter
        self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
4551 b9322a9f Guido Trotter
        self.op.src_node = None
4552 b9322a9f Guido Trotter
        if os.path.isabs(src_path):
4553 b9322a9f Guido Trotter
          raise errors.OpPrereqError("Importing an instance from an absolute"
4554 b9322a9f Guido Trotter
                                     " path requires a source node option.")
4555 b9322a9f Guido Trotter
      else:
4556 b9322a9f Guido Trotter
        self.op.src_node = src_node = self._ExpandNode(src_node)
4557 b9322a9f Guido Trotter
        if self.needed_locks[locking.LEVEL_NODE] is not locking.ALL_SET:
4558 b9322a9f Guido Trotter
          self.needed_locks[locking.LEVEL_NODE].append(src_node)
4559 b9322a9f Guido Trotter
        if not os.path.isabs(src_path):
4560 b9322a9f Guido Trotter
          self.op.src_path = src_path = \
4561 b9322a9f Guido Trotter
            os.path.join(constants.EXPORT_DIR, src_path)
4562 7baf741d Guido Trotter
4563 7baf741d Guido Trotter
    else: # INSTANCE_CREATE
4564 7baf741d Guido Trotter
      if getattr(self.op, "os_type", None) is None:
4565 7baf741d Guido Trotter
        raise errors.OpPrereqError("No guest OS specified")
4566 a8083063 Iustin Pop
4567 538475ca Iustin Pop
  def _RunAllocator(self):
4568 538475ca Iustin Pop
    """Run the allocator based on input opcode.
4569 538475ca Iustin Pop

4570 538475ca Iustin Pop
    """
4571 08db7c5c Iustin Pop
    nics = [n.ToDict() for n in self.nics]
4572 72737a7f Iustin Pop
    ial = IAllocator(self,
4573 29859cb7 Iustin Pop
                     mode=constants.IALLOCATOR_MODE_ALLOC,
4574 d1c2dd75 Iustin Pop
                     name=self.op.instance_name,
4575 d1c2dd75 Iustin Pop
                     disk_template=self.op.disk_template,
4576 d1c2dd75 Iustin Pop
                     tags=[],
4577 d1c2dd75 Iustin Pop
                     os=self.op.os_type,
4578 338e51e8 Iustin Pop
                     vcpus=self.be_full[constants.BE_VCPUS],
4579 338e51e8 Iustin Pop
                     mem_size=self.be_full[constants.BE_MEMORY],
4580 08db7c5c Iustin Pop
                     disks=self.disks,
4581 d1c2dd75 Iustin Pop
                     nics=nics,
4582 8cc7e742 Guido Trotter
                     hypervisor=self.op.hypervisor,
4583 29859cb7 Iustin Pop
                     )
4584 d1c2dd75 Iustin Pop
4585 d1c2dd75 Iustin Pop
    ial.Run(self.op.iallocator)
4586 d1c2dd75 Iustin Pop
4587 d1c2dd75 Iustin Pop
    if not ial.success:
4588 538475ca Iustin Pop
      raise errors.OpPrereqError("Can't compute nodes using"
4589 538475ca Iustin Pop
                                 " iallocator '%s': %s" % (self.op.iallocator,
4590 d1c2dd75 Iustin Pop
                                                           ial.info))
4591 27579978 Iustin Pop
    if len(ial.nodes) != ial.required_nodes:
4592 538475ca Iustin Pop
      raise errors.OpPrereqError("iallocator '%s' returned invalid number"
4593 538475ca Iustin Pop
                                 " of nodes (%s), required %s" %
4594 97abc79f Iustin Pop
                                 (self.op.iallocator, len(ial.nodes),
4595 1ce4bbe3 Renรฉ Nussbaumer
                                  ial.required_nodes))
4596 d1c2dd75 Iustin Pop
    self.op.pnode = ial.nodes[0]
4597 86d9d3bb Iustin Pop
    self.LogInfo("Selected nodes for instance %s via iallocator %s: %s",
4598 86d9d3bb Iustin Pop
                 self.op.instance_name, self.op.iallocator,
4599 86d9d3bb Iustin Pop
                 ", ".join(ial.nodes))
4600 27579978 Iustin Pop
    if ial.required_nodes == 2:
4601 d1c2dd75 Iustin Pop
      self.op.snode = ial.nodes[1]
4602 538475ca Iustin Pop
4603 a8083063 Iustin Pop
  def BuildHooksEnv(self):
4604 a8083063 Iustin Pop
    """Build hooks env.
4605 a8083063 Iustin Pop

4606 a8083063 Iustin Pop
    This runs on master, primary and secondary nodes of the instance.
4607 a8083063 Iustin Pop

4608 a8083063 Iustin Pop
    """
4609 a8083063 Iustin Pop
    env = {
4610 2c2690c9 Iustin Pop
      "ADD_MODE": self.op.mode,
4611 a8083063 Iustin Pop
      }
4612 a8083063 Iustin Pop
    if self.op.mode == constants.INSTANCE_IMPORT:
4613 2c2690c9 Iustin Pop
      env["SRC_NODE"] = self.op.src_node
4614 2c2690c9 Iustin Pop
      env["SRC_PATH"] = self.op.src_path
4615 2c2690c9 Iustin Pop
      env["SRC_IMAGES"] = self.src_images
4616 396e1b78 Michael Hanselmann
4617 2c2690c9 Iustin Pop
    env.update(_BuildInstanceHookEnv(
4618 2c2690c9 Iustin Pop
      name=self.op.instance_name,
4619 396e1b78 Michael Hanselmann
      primary_node=self.op.pnode,
4620 396e1b78 Michael Hanselmann
      secondary_nodes=self.secondaries,
4621 4978db17 Iustin Pop
      status=self.op.start,
4622 ecb215b5 Michael Hanselmann
      os_type=self.op.os_type,
4623 338e51e8 Iustin Pop
      memory=self.be_full[constants.BE_MEMORY],
4624 338e51e8 Iustin Pop
      vcpus=self.be_full[constants.BE_VCPUS],
4625 08db7c5c Iustin Pop
      nics=[(n.ip, n.bridge, n.mac) for n in self.nics],
4626 2c2690c9 Iustin Pop
      disk_template=self.op.disk_template,
4627 2c2690c9 Iustin Pop
      disks=[(d["size"], d["mode"]) for d in self.disks],
4628 396e1b78 Michael Hanselmann
    ))
4629 a8083063 Iustin Pop
4630 d6a02168 Michael Hanselmann
    nl = ([self.cfg.GetMasterNode(), self.op.pnode] +
4631 a8083063 Iustin Pop
          self.secondaries)
4632 a8083063 Iustin Pop
    return env, nl, nl
4633 a8083063 Iustin Pop
4634 a8083063 Iustin Pop
4635 a8083063 Iustin Pop
  def CheckPrereq(self):
4636 a8083063 Iustin Pop
    """Check prerequisites.
4637 a8083063 Iustin Pop

4638 a8083063 Iustin Pop
    """
4639 eedc99de Manuel Franceschini
    if (not self.cfg.GetVGName() and
4640 eedc99de Manuel Franceschini
        self.op.disk_template not in constants.DTS_NOT_LVM):
4641 eedc99de Manuel Franceschini
      raise errors.OpPrereqError("Cluster does not support lvm-based"
4642 eedc99de Manuel Franceschini
                                 " instances")
4643 eedc99de Manuel Franceschini
4644 a8083063 Iustin Pop
    if self.op.mode == constants.INSTANCE_IMPORT:
4645 7baf741d Guido Trotter
      src_node = self.op.src_node
4646 7baf741d Guido Trotter
      src_path = self.op.src_path
4647 a8083063 Iustin Pop
4648 c0cbdc67 Guido Trotter
      if src_node is None:
4649 c0cbdc67 Guido Trotter
        exp_list = self.rpc.call_export_list(
4650 781de953 Iustin Pop
          self.acquired_locks[locking.LEVEL_NODE])
4651 c0cbdc67 Guido Trotter
        found = False
4652 c0cbdc67 Guido Trotter
        for node in exp_list:
4653 781de953 Iustin Pop
          if not exp_list[node].failed and src_path in exp_list[node].data:
4654 c0cbdc67 Guido Trotter
            found = True
4655 c0cbdc67 Guido Trotter
            self.op.src_node = src_node = node
4656 c0cbdc67 Guido Trotter
            self.op.src_path = src_path = os.path.join(constants.EXPORT_DIR,
4657 c0cbdc67 Guido Trotter
                                                       src_path)
4658 c0cbdc67 Guido Trotter
            break
4659 c0cbdc67 Guido Trotter
        if not found:
4660 c0cbdc67 Guido Trotter
          raise errors.OpPrereqError("No export found for relative path %s" %
4661 c0cbdc67 Guido Trotter
                                      src_path)
4662 c0cbdc67 Guido Trotter
4663 7527a8a4 Iustin Pop
      _CheckNodeOnline(self, src_node)
4664 781de953 Iustin Pop
      result = self.rpc.call_export_info(src_node, src_path)
4665 781de953 Iustin Pop
      result.Raise()
4666 781de953 Iustin Pop
      if not result.data:
4667 3ecf6786 Iustin Pop
        raise errors.OpPrereqError("No export found in dir %s" % src_path)
4668 a8083063 Iustin Pop
4669 781de953 Iustin Pop
      export_info = result.data
4670 a8083063 Iustin Pop
      if not export_info.has_section(constants.INISECT_EXP):
4671 3ecf6786 Iustin Pop
        raise errors.ProgrammerError("Corrupted export config")
4672 a8083063 Iustin Pop
4673 a8083063 Iustin Pop
      ei_version = export_info.get(constants.INISECT_EXP, 'version')
4674 a8083063 Iustin Pop
      if (int(ei_version) != constants.EXPORT_VERSION):
4675 3ecf6786 Iustin Pop
        raise errors.OpPrereqError("Wrong export version %s (wanted %d)" %
4676 3ecf6786 Iustin Pop
                                   (ei_version, constants.EXPORT_VERSION))
4677 a8083063 Iustin Pop
4678 09acf207 Guido Trotter
      # Check that the new instance doesn't have less disks than the export
4679 08db7c5c Iustin Pop
      instance_disks = len(self.disks)
4680 09acf207 Guido Trotter
      export_disks = export_info.getint(constants.INISECT_INS, 'disk_count')
4681 09acf207 Guido Trotter
      if instance_disks < export_disks:
4682 09acf207 Guido Trotter
        raise errors.OpPrereqError("Not enough disks to import."
4683 09acf207 Guido Trotter
                                   " (instance: %d, export: %d)" %
4684 726d7d68 Iustin Pop
                                   (instance_disks, export_disks))
4685 a8083063 Iustin Pop
4686 a8083063 Iustin Pop
      self.op.os_type = export_info.get(constants.INISECT_EXP, 'os')
4687 09acf207 Guido Trotter
      disk_images = []
4688 09acf207 Guido Trotter
      for idx in range(export_disks):
4689 09acf207 Guido Trotter
        option = 'disk%d_dump' % idx
4690 09acf207 Guido Trotter
        if export_info.has_option(constants.INISECT_INS, option):
4691 09acf207 Guido Trotter
          # FIXME: are the old os-es, disk sizes, etc. useful?
4692 09acf207 Guido Trotter
          export_name = export_info.get(constants.INISECT_INS, option)
4693 09acf207 Guido Trotter
          image = os.path.join(src_path, export_name)
4694 09acf207 Guido Trotter
          disk_images.append(image)
4695 09acf207 Guido Trotter
        else:
4696 09acf207 Guido Trotter
          disk_images.append(False)
4697 09acf207 Guido Trotter
4698 09acf207 Guido Trotter
      self.src_images = disk_images
4699 901a65c1 Iustin Pop
4700 b4364a6b Guido Trotter
      old_name = export_info.get(constants.INISECT_INS, 'name')
4701 b4364a6b Guido Trotter
      # FIXME: int() here could throw a ValueError on broken exports
4702 b4364a6b Guido Trotter
      exp_nic_count = int(export_info.get(constants.INISECT_INS, 'nic_count'))
4703 b4364a6b Guido Trotter
      if self.op.instance_name == old_name:
4704 b4364a6b Guido Trotter
        for idx, nic in enumerate(self.nics):
4705 b4364a6b Guido Trotter
          if nic.mac == constants.VALUE_AUTO and exp_nic_count >= idx:
4706 b4364a6b Guido Trotter
            nic_mac_ini = 'nic%d_mac' % idx
4707 b4364a6b Guido Trotter
            nic.mac = export_info.get(constants.INISECT_INS, nic_mac_ini)
4708 bc89efc3 Guido Trotter
4709 295728df Guido Trotter
    # ENDIF: self.op.mode == constants.INSTANCE_IMPORT
4710 7baf741d Guido Trotter
    # ip ping checks (we use the same ip that was resolved in ExpandNames)
4711 901a65c1 Iustin Pop
    if self.op.start and not self.op.ip_check:
4712 901a65c1 Iustin Pop
      raise errors.OpPrereqError("Cannot ignore IP address conflicts when"
4713 901a65c1 Iustin Pop
                                 " adding an instance in start mode")
4714 901a65c1 Iustin Pop
4715 901a65c1 Iustin Pop
    if self.op.ip_check:
4716 7baf741d Guido Trotter
      if utils.TcpPing(self.check_ip, constants.DEFAULT_NODED_PORT):
4717 901a65c1 Iustin Pop
        raise errors.OpPrereqError("IP %s of instance %s already in use" %
4718 7b3a8fb5 Iustin Pop
                                   (self.check_ip, self.op.instance_name))
4719 901a65c1 Iustin Pop
4720 295728df Guido Trotter
    #### mac address generation
4721 295728df Guido Trotter
    # By generating here the mac address both the allocator and the hooks get
4722 295728df Guido Trotter
    # the real final mac address rather than the 'auto' or 'generate' value.
4723 295728df Guido Trotter
    # There is a race condition between the generation and the instance object
4724 295728df Guido Trotter
    # creation, which means that we know the mac is valid now, but we're not
4725 295728df Guido Trotter
    # sure it will be when we actually add the instance. If things go bad
4726 295728df Guido Trotter
    # adding the instance will abort because of a duplicate mac, and the
4727 295728df Guido Trotter
    # creation job will fail.
4728 295728df Guido Trotter
    for nic in self.nics:
4729 295728df Guido Trotter
      if nic.mac in (constants.VALUE_AUTO, constants.VALUE_GENERATE):
4730 295728df Guido Trotter
        nic.mac = self.cfg.GenerateMAC()
4731 295728df Guido Trotter
4732 538475ca Iustin Pop
    #### allocator run
4733 538475ca Iustin Pop
4734 538475ca Iustin Pop
    if self.op.iallocator is not None:
4735 538475ca Iustin Pop
      self._RunAllocator()
4736 0f1a06e3 Manuel Franceschini
4737 901a65c1 Iustin Pop
    #### node related checks
4738 901a65c1 Iustin Pop
4739 901a65c1 Iustin Pop
    # check primary node
4740 7baf741d Guido Trotter
    self.pnode = pnode = self.cfg.GetNodeInfo(self.op.pnode)
4741 7baf741d Guido Trotter
    assert self.pnode is not None, \
4742 7baf741d Guido Trotter
      "Cannot retrieve locked node %s" % self.op.pnode
4743 7527a8a4 Iustin Pop
    if pnode.offline:
4744 7527a8a4 Iustin Pop
      raise errors.OpPrereqError("Cannot use offline primary node '%s'" %
4745 7527a8a4 Iustin Pop
                                 pnode.name)
4746 733a2b6a Iustin Pop
    if pnode.drained:
4747 733a2b6a Iustin Pop
      raise errors.OpPrereqError("Cannot use drained primary node '%s'" %
4748 733a2b6a Iustin Pop
                                 pnode.name)
4749 7527a8a4 Iustin Pop
4750 901a65c1 Iustin Pop
    self.secondaries = []
4751 901a65c1 Iustin Pop
4752 901a65c1 Iustin Pop
    # mirror node verification
4753 a1f445d3 Iustin Pop
    if self.op.disk_template in constants.DTS_NET_MIRROR:
4754 7baf741d Guido Trotter
      if self.op.snode is None:
4755 a1f445d3 Iustin Pop
        raise errors.OpPrereqError("The networked disk templates need"
4756 3ecf6786 Iustin Pop
                                   " a mirror node")
4757 7baf741d Guido Trotter
      if self.op.snode == pnode.name:
4758 3ecf6786 Iustin Pop
        raise errors.OpPrereqError("The secondary node cannot be"
4759 3ecf6786 Iustin Pop
                                   " the primary node.")
4760 7527a8a4 Iustin Pop
      _CheckNodeOnline(self, self.op.snode)
4761 733a2b6a Iustin Pop
      _CheckNodeNotDrained(self, self.op.snode)
4762 733a2b6a Iustin Pop
      self.secondaries.append(self.op.snode)
4763 a8083063 Iustin Pop
4764 6785674e Iustin Pop
    nodenames = [pnode.name] + self.secondaries
4765 6785674e Iustin Pop
4766 e2fe6369 Iustin Pop
    req_size = _ComputeDiskSize(self.op.disk_template,
4767 08db7c5c Iustin Pop
                                self.disks)
4768 ed1ebc60 Guido Trotter
4769 8d75db10 Iustin Pop
    # Check lv size requirements
4770 8d75db10 Iustin Pop
    if req_size is not None:
4771 72737a7f Iustin Pop
      nodeinfo = self.rpc.call_node_info(nodenames, self.cfg.GetVGName(),
4772 72737a7f Iustin Pop
                                         self.op.hypervisor)
4773 8d75db10 Iustin Pop
      for node in nodenames:
4774 781de953 Iustin Pop
        info = nodeinfo[node]
4775 781de953 Iustin Pop
        info.Raise()
4776 781de953 Iustin Pop
        info = info.data
4777 8d75db10 Iustin Pop
        if not info:
4778 8d75db10 Iustin Pop
          raise errors.OpPrereqError("Cannot get current information"
4779 3e91897b Iustin Pop
                                     " from node '%s'" % node)
4780 8d75db10 Iustin Pop
        vg_free = info.get('vg_free', None)
4781 8d75db10 Iustin Pop
        if not isinstance(vg_free, int):
4782 8d75db10 Iustin Pop
          raise errors.OpPrereqError("Can't compute free disk space on"
4783 8d75db10 Iustin Pop
                                     " node %s" % node)
4784 8d75db10 Iustin Pop
        if req_size > info['vg_free']:
4785 8d75db10 Iustin Pop
          raise errors.OpPrereqError("Not enough disk space on target node %s."
4786 8d75db10 Iustin Pop
                                     " %d MB available, %d MB required" %
4787 8d75db10 Iustin Pop
                                     (node, info['vg_free'], req_size))
4788 ed1ebc60 Guido Trotter
4789 74409b12 Iustin Pop
    _CheckHVParams(self, nodenames, self.op.hypervisor, self.op.hvparams)
4790 6785674e Iustin Pop
4791 a8083063 Iustin Pop
    # os verification
4792 781de953 Iustin Pop
    result = self.rpc.call_os_get(pnode.name, self.op.os_type)
4793 781de953 Iustin Pop
    result.Raise()
4794 781de953 Iustin Pop
    if not isinstance(result.data, objects.OS):
4795 3ecf6786 Iustin Pop
      raise errors.OpPrereqError("OS '%s' not in supported os list for"
4796 3ecf6786 Iustin Pop
                                 " primary node"  % self.op.os_type)
4797 a8083063 Iustin Pop
4798 901a65c1 Iustin Pop
    # bridge check on primary node
4799 08db7c5c Iustin Pop
    bridges = [n.bridge for n in self.nics]
4800 781de953 Iustin Pop
    result = self.rpc.call_bridges_exist(self.pnode.name, bridges)
4801 781de953 Iustin Pop
    result.Raise()
4802 781de953 Iustin Pop
    if not result.data:
4803 781de953 Iustin Pop
      raise errors.OpPrereqError("One of the target bridges '%s' does not"
4804 781de953 Iustin Pop
                                 " exist on destination node '%s'" %
4805 08db7c5c Iustin Pop
                                 (",".join(bridges), pnode.name))
4806 a8083063 Iustin Pop
4807 49ce1563 Iustin Pop
    # memory check on primary node
4808 49ce1563 Iustin Pop
    if self.op.start:
4809 b9bddb6b Iustin Pop
      _CheckNodeFreeMemory(self, self.pnode.name,
4810 49ce1563 Iustin Pop
                           "creating instance %s" % self.op.instance_name,
4811 338e51e8 Iustin Pop
                           self.be_full[constants.BE_MEMORY],
4812 338e51e8 Iustin Pop
                           self.op.hypervisor)
4813 49ce1563 Iustin Pop
4814 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
4815 a8083063 Iustin Pop
    """Create and add the instance to the cluster.
4816 a8083063 Iustin Pop

4817 a8083063 Iustin Pop
    """
4818 a8083063 Iustin Pop
    instance = self.op.instance_name
4819 a8083063 Iustin Pop
    pnode_name = self.pnode.name
4820 a8083063 Iustin Pop
4821 e69d05fd Iustin Pop
    ht_kind = self.op.hypervisor
4822 2a6469d5 Alexander Schreiber
    if ht_kind in constants.HTS_REQ_PORT:
4823 2a6469d5 Alexander Schreiber
      network_port = self.cfg.AllocatePort()
4824 2a6469d5 Alexander Schreiber
    else:
4825 2a6469d5 Alexander Schreiber
      network_port = None
4826 58acb49d Alexander Schreiber
4827 6785674e Iustin Pop
    ##if self.op.vnc_bind_address is None:
4828 6785674e Iustin Pop
    ##  self.op.vnc_bind_address = constants.VNC_DEFAULT_BIND_ADDRESS
4829 31a853d2 Iustin Pop
4830 2c313123 Manuel Franceschini
    # this is needed because os.path.join does not accept None arguments
4831 2c313123 Manuel Franceschini
    if self.op.file_storage_dir is None:
4832 2c313123 Manuel Franceschini
      string_file_storage_dir = ""
4833 2c313123 Manuel Franceschini
    else:
4834 2c313123 Manuel Franceschini
      string_file_storage_dir = self.op.file_storage_dir
4835 2c313123 Manuel Franceschini
4836 0f1a06e3 Manuel Franceschini
    # build the full file storage dir path
4837 0f1a06e3 Manuel Franceschini
    file_storage_dir = os.path.normpath(os.path.join(
4838 d6a02168 Michael Hanselmann
                                        self.cfg.GetFileStorageDir(),
4839 2c313123 Manuel Franceschini
                                        string_file_storage_dir, instance))
4840 0f1a06e3 Manuel Franceschini
4841 0f1a06e3 Manuel Franceschini
4842 b9bddb6b Iustin Pop
    disks = _GenerateDiskTemplate(self,
4843 a8083063 Iustin Pop
                                  self.op.disk_template,
4844 a8083063 Iustin Pop
                                  instance, pnode_name,
4845 08db7c5c Iustin Pop
                                  self.secondaries,
4846 08db7c5c Iustin Pop
                                  self.disks,
4847 0f1a06e3 Manuel Franceschini
                                  file_storage_dir,
4848 e2a65344 Iustin Pop
                                  self.op.file_driver,
4849 e2a65344 Iustin Pop
                                  0)
4850 a8083063 Iustin Pop
4851 a8083063 Iustin Pop
    iobj = objects.Instance(name=instance, os=self.op.os_type,
4852 a8083063 Iustin Pop
                            primary_node=pnode_name,
4853 08db7c5c Iustin Pop
                            nics=self.nics, disks=disks,
4854 a8083063 Iustin Pop
                            disk_template=self.op.disk_template,
4855 4978db17 Iustin Pop
                            admin_up=False,
4856 58acb49d Alexander Schreiber
                            network_port=network_port,
4857 338e51e8 Iustin Pop
                            beparams=self.op.beparams,
4858 6785674e Iustin Pop
                            hvparams=self.op.hvparams,
4859 e69d05fd Iustin Pop
                            hypervisor=self.op.hypervisor,
4860 a8083063 Iustin Pop
                            )
4861 a8083063 Iustin Pop
4862 a8083063 Iustin Pop
    feedback_fn("* creating instance disks...")
4863 796cab27 Iustin Pop
    try:
4864 796cab27 Iustin Pop
      _CreateDisks(self, iobj)
4865 796cab27 Iustin Pop
    except errors.OpExecError:
4866 796cab27 Iustin Pop
      self.LogWarning("Device creation failed, reverting...")
4867 796cab27 Iustin Pop
      try:
4868 796cab27 Iustin Pop
        _RemoveDisks(self, iobj)
4869 796cab27 Iustin Pop
      finally:
4870 796cab27 Iustin Pop
        self.cfg.ReleaseDRBDMinors(instance)
4871 796cab27 Iustin Pop
        raise
4872 a8083063 Iustin Pop
4873 a8083063 Iustin Pop
    feedback_fn("adding instance %s to cluster config" % instance)
4874 a8083063 Iustin Pop
4875 a8083063 Iustin Pop
    self.cfg.AddInstance(iobj)
4876 7baf741d Guido Trotter
    # Declare that we don't want to remove the instance lock anymore, as we've
4877 7baf741d Guido Trotter
    # added the instance to the config
4878 7baf741d Guido Trotter
    del self.remove_locks[locking.LEVEL_INSTANCE]
4879 e36e96b4 Guido Trotter
    # Unlock all the nodes
4880 9c8971d7 Guido Trotter
    if self.op.mode == constants.INSTANCE_IMPORT:
4881 9c8971d7 Guido Trotter
      nodes_keep = [self.op.src_node]
4882 9c8971d7 Guido Trotter
      nodes_release = [node for node in self.acquired_locks[locking.LEVEL_NODE]
4883 9c8971d7 Guido Trotter
                       if node != self.op.src_node]
4884 9c8971d7 Guido Trotter
      self.context.glm.release(locking.LEVEL_NODE, nodes_release)
4885 9c8971d7 Guido Trotter
      self.acquired_locks[locking.LEVEL_NODE] = nodes_keep
4886 9c8971d7 Guido Trotter
    else:
4887 9c8971d7 Guido Trotter
      self.context.glm.release(locking.LEVEL_NODE)
4888 9c8971d7 Guido Trotter
      del self.acquired_locks[locking.LEVEL_NODE]
4889 a8083063 Iustin Pop
4890 a8083063 Iustin Pop
    if self.op.wait_for_sync:
4891 b9bddb6b Iustin Pop
      disk_abort = not _WaitForSync(self, iobj)
4892 a1f445d3 Iustin Pop
    elif iobj.disk_template in constants.DTS_NET_MIRROR:
4893 a8083063 Iustin Pop
      # make sure the disks are not degraded (still sync-ing is ok)
4894 a8083063 Iustin Pop
      time.sleep(15)
4895 a8083063 Iustin Pop
      feedback_fn("* checking mirrors status")
4896 b9bddb6b Iustin Pop
      disk_abort = not _WaitForSync(self, iobj, oneshot=True)
4897 a8083063 Iustin Pop
    else:
4898 a8083063 Iustin Pop
      disk_abort = False
4899 a8083063 Iustin Pop
4900 a8083063 Iustin Pop
    if disk_abort:
4901 b9bddb6b Iustin Pop
      _RemoveDisks(self, iobj)
4902 a8083063 Iustin Pop
      self.cfg.RemoveInstance(iobj.name)
4903 7baf741d Guido Trotter
      # Make sure the instance lock gets removed
4904 7baf741d Guido Trotter
      self.remove_locks[locking.LEVEL_INSTANCE] = iobj.name
4905 3ecf6786 Iustin Pop
      raise errors.OpExecError("There are some degraded disks for"
4906 3ecf6786 Iustin Pop
                               " this instance")
4907 a8083063 Iustin Pop
4908 a8083063 Iustin Pop
    feedback_fn("creating os for instance %s on node %s" %
4909 a8083063 Iustin Pop
                (instance, pnode_name))
4910 a8083063 Iustin Pop
4911 a8083063 Iustin Pop
    if iobj.disk_template != constants.DT_DISKLESS:
4912 a8083063 Iustin Pop
      if self.op.mode == constants.INSTANCE_CREATE:
4913 a8083063 Iustin Pop
        feedback_fn("* running the instance OS create scripts...")
4914 e557bae9 Guido Trotter
        result = self.rpc.call_instance_os_add(pnode_name, iobj, False)
4915 20e01edd Iustin Pop
        msg = result.RemoteFailMsg()
4916 20e01edd Iustin Pop
        if msg:
4917 781de953 Iustin Pop
          raise errors.OpExecError("Could not add os for instance %s"
4918 20e01edd Iustin Pop
                                   " on node %s: %s" %
4919 20e01edd Iustin Pop
                                   (instance, pnode_name, msg))
4920 a8083063 Iustin Pop
4921 a8083063 Iustin Pop
      elif self.op.mode == constants.INSTANCE_IMPORT:
4922 a8083063 Iustin Pop
        feedback_fn("* running the instance OS import scripts...")
4923 a8083063 Iustin Pop
        src_node = self.op.src_node
4924 09acf207 Guido Trotter
        src_images = self.src_images
4925 62c9ec92 Iustin Pop
        cluster_name = self.cfg.GetClusterName()
4926 6c0af70e Guido Trotter
        import_result = self.rpc.call_instance_os_import(pnode_name, iobj,
4927 09acf207 Guido Trotter
                                                         src_node, src_images,
4928 6c0af70e Guido Trotter
                                                         cluster_name)
4929 781de953 Iustin Pop
        import_result.Raise()
4930 781de953 Iustin Pop
        for idx, result in enumerate(import_result.data):
4931 09acf207 Guido Trotter
          if not result:
4932 726d7d68 Iustin Pop
            self.LogWarning("Could not import the image %s for instance"
4933 726d7d68 Iustin Pop
                            " %s, disk %d, on node %s" %
4934 726d7d68 Iustin Pop
                            (src_images[idx], instance, idx, pnode_name))
4935 a8083063 Iustin Pop
      else:
4936 a8083063 Iustin Pop
        # also checked in the prereq part
4937 3ecf6786 Iustin Pop
        raise errors.ProgrammerError("Unknown OS initialization mode '%s'"
4938 3ecf6786 Iustin Pop
                                     % self.op.mode)
4939 a8083063 Iustin Pop
4940 a8083063 Iustin Pop
    if self.op.start:
4941 4978db17 Iustin Pop
      iobj.admin_up = True
4942 4978db17 Iustin Pop
      self.cfg.Update(iobj)
4943 9a4f63d1 Iustin Pop
      logging.info("Starting instance %s on node %s", instance, pnode_name)
4944 a8083063 Iustin Pop
      feedback_fn("* starting instance...")
4945 0eca8e0c Iustin Pop
      result = self.rpc.call_instance_start(pnode_name, iobj, None, None)
4946 dd279568 Iustin Pop
      msg = result.RemoteFailMsg()
4947 dd279568 Iustin Pop
      if msg:
4948 dd279568 Iustin Pop
        raise errors.OpExecError("Could not start instance: %s" % msg)
4949 a8083063 Iustin Pop
4950 a8083063 Iustin Pop
4951 a8083063 Iustin Pop
class LUConnectConsole(NoHooksLU):
4952 a8083063 Iustin Pop
  """Connect to an instance's console.
4953 a8083063 Iustin Pop

4954 a8083063 Iustin Pop
  This is somewhat special in that it returns the command line that
4955 a8083063 Iustin Pop
  you need to run on the master node in order to connect to the
4956 a8083063 Iustin Pop
  console.
4957 a8083063 Iustin Pop

4958 a8083063 Iustin Pop
  """
4959 a8083063 Iustin Pop
  _OP_REQP = ["instance_name"]
4960 8659b73e Guido Trotter
  REQ_BGL = False
4961 8659b73e Guido Trotter
4962 8659b73e Guido Trotter
  def ExpandNames(self):
4963 8659b73e Guido Trotter
    self._ExpandAndLockInstance()
4964 a8083063 Iustin Pop
4965 a8083063 Iustin Pop
  def CheckPrereq(self):
4966 a8083063 Iustin Pop
    """Check prerequisites.
4967 a8083063 Iustin Pop

4968 a8083063 Iustin Pop
    This checks that the instance is in the cluster.
4969 a8083063 Iustin Pop

4970 a8083063 Iustin Pop
    """
4971 8659b73e Guido Trotter
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_name)
4972 8659b73e Guido Trotter
    assert self.instance is not None, \
4973 8659b73e Guido Trotter
      "Cannot retrieve locked instance %s" % self.op.instance_name
4974 513e896d Guido Trotter
    _CheckNodeOnline(self, self.instance.primary_node)
4975 a8083063 Iustin Pop
4976 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
4977 a8083063 Iustin Pop
    """Connect to the console of an instance
4978 a8083063 Iustin Pop

4979 a8083063 Iustin Pop
    """
4980 a8083063 Iustin Pop
    instance = self.instance
4981 a8083063 Iustin Pop
    node = instance.primary_node
4982 a8083063 Iustin Pop
4983 72737a7f Iustin Pop
    node_insts = self.rpc.call_instance_list([node],
4984 72737a7f Iustin Pop
                                             [instance.hypervisor])[node]
4985 781de953 Iustin Pop
    node_insts.Raise()
4986 a8083063 Iustin Pop
4987 781de953 Iustin Pop
    if instance.name not in node_insts.data:
4988 3ecf6786 Iustin Pop
      raise errors.OpExecError("Instance %s is not running." % instance.name)
4989 a8083063 Iustin Pop
4990 9a4f63d1 Iustin Pop
    logging.debug("Connecting to console of %s on %s", instance.name, node)
4991 a8083063 Iustin Pop
4992 e69d05fd Iustin Pop
    hyper = hypervisor.GetHypervisor(instance.hypervisor)
4993 5431b2e4 Guido Trotter
    cluster = self.cfg.GetClusterInfo()
4994 5431b2e4 Guido Trotter
    # beparams and hvparams are passed separately, to avoid editing the
4995 5431b2e4 Guido Trotter
    # instance and then saving the defaults in the instance itself.
4996 5431b2e4 Guido Trotter
    hvparams = cluster.FillHV(instance)
4997 5431b2e4 Guido Trotter
    beparams = cluster.FillBE(instance)
4998 5431b2e4 Guido Trotter
    console_cmd = hyper.GetShellCommandForConsole(instance, hvparams, beparams)
4999 b047857b Michael Hanselmann
5000 82122173 Iustin Pop
    # build ssh cmdline
5001 0a80a26f Michael Hanselmann
    return self.ssh.BuildCmd(node, "root", console_cmd, batch=True, tty=True)
5002 a8083063 Iustin Pop
5003 a8083063 Iustin Pop
5004 a8083063 Iustin Pop
class LUReplaceDisks(LogicalUnit):
5005 a8083063 Iustin Pop
  """Replace the disks of an instance.
5006 a8083063 Iustin Pop

5007 a8083063 Iustin Pop
  """
5008 a8083063 Iustin Pop
  HPATH = "mirrors-replace"
5009 a8083063 Iustin Pop
  HTYPE = constants.HTYPE_INSTANCE
5010 a9e0c397 Iustin Pop
  _OP_REQP = ["instance_name", "mode", "disks"]
5011 efd990e4 Guido Trotter
  REQ_BGL = False
5012 efd990e4 Guido Trotter
5013 7e9366f7 Iustin Pop
  def CheckArguments(self):
5014 efd990e4 Guido Trotter
    if not hasattr(self.op, "remote_node"):
5015 efd990e4 Guido Trotter
      self.op.remote_node = None
5016 7e9366f7 Iustin Pop
    if not hasattr(self.op, "iallocator"):
5017 7e9366f7 Iustin Pop
      self.op.iallocator = None
5018 7e9366f7 Iustin Pop
5019 7e9366f7 Iustin Pop
    # check for valid parameter combination
5020 7e9366f7 Iustin Pop
    cnt = [self.op.remote_node, self.op.iallocator].count(None)
5021 7e9366f7 Iustin Pop
    if self.op.mode == constants.REPLACE_DISK_CHG:
5022 7e9366f7 Iustin Pop
      if cnt == 2:
5023 7e9366f7 Iustin Pop
        raise errors.OpPrereqError("When changing the secondary either an"
5024 7e9366f7 Iustin Pop
                                   " iallocator script must be used or the"
5025 7e9366f7 Iustin Pop
                                   " new node given")
5026 7e9366f7 Iustin Pop
      elif cnt == 0:
5027 efd990e4 Guido Trotter
        raise errors.OpPrereqError("Give either the iallocator or the new"
5028 efd990e4 Guido Trotter
                                   " secondary, not both")
5029 7e9366f7 Iustin Pop
    else: # not replacing the secondary
5030 7e9366f7 Iustin Pop
      if cnt != 2:
5031 7e9366f7 Iustin Pop
        raise errors.OpPrereqError("The iallocator and new node options can"
5032 7e9366f7 Iustin Pop
                                   " be used only when changing the"
5033 7e9366f7 Iustin Pop
                                   " secondary node")
5034 7e9366f7 Iustin Pop
5035 7e9366f7 Iustin Pop
  def ExpandNames(self):
5036 7e9366f7 Iustin Pop
    self._ExpandAndLockInstance()
5037 7e9366f7 Iustin Pop
5038 7e9366f7 Iustin Pop
    if self.op.iallocator is not None:
5039 efd990e4 Guido Trotter
      self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
5040 efd990e4 Guido Trotter
    elif self.op.remote_node is not None:
5041 efd990e4 Guido Trotter
      remote_node = self.cfg.ExpandNodeName(self.op.remote_node)
5042 efd990e4 Guido Trotter
      if remote_node is None:
5043 efd990e4 Guido Trotter
        raise errors.OpPrereqError("Node '%s' not known" %
5044 efd990e4 Guido Trotter
                                   self.op.remote_node)
5045 efd990e4 Guido Trotter
      self.op.remote_node = remote_node
5046 3b559640 Iustin Pop
      # Warning: do not remove the locking of the new secondary here
5047 3b559640 Iustin Pop
      # unless DRBD8.AddChildren is changed to work in parallel;
5048 3b559640 Iustin Pop
      # currently it doesn't since parallel invocations of
5049 3b559640 Iustin Pop
      # FindUnusedMinor will conflict
5050 efd990e4 Guido Trotter
      self.needed_locks[locking.LEVEL_NODE] = [remote_node]
5051 efd990e4 Guido Trotter
      self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_APPEND
5052 efd990e4 Guido Trotter
    else:
5053 efd990e4 Guido Trotter
      self.needed_locks[locking.LEVEL_NODE] = []
5054 efd990e4 Guido Trotter
      self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
5055 efd990e4 Guido Trotter
5056 efd990e4 Guido Trotter
  def DeclareLocks(self, level):
5057 efd990e4 Guido Trotter
    # If we're not already locking all nodes in the set we have to declare the
5058 efd990e4 Guido Trotter
    # instance's primary/secondary nodes.
5059 efd990e4 Guido Trotter
    if (level == locking.LEVEL_NODE and
5060 efd990e4 Guido Trotter
        self.needed_locks[locking.LEVEL_NODE] is not locking.ALL_SET):
5061 efd990e4 Guido Trotter
      self._LockInstancesNodes()
5062 a8083063 Iustin Pop
5063 b6e82a65 Iustin Pop
  def _RunAllocator(self):
5064 b6e82a65 Iustin Pop
    """Compute a new secondary node using an IAllocator.
5065 b6e82a65 Iustin Pop

5066 b6e82a65 Iustin Pop
    """
5067 72737a7f Iustin Pop
    ial = IAllocator(self,
5068 b6e82a65 Iustin Pop
                     mode=constants.IALLOCATOR_MODE_RELOC,
5069 b6e82a65 Iustin Pop
                     name=self.op.instance_name,
5070 b6e82a65 Iustin Pop
                     relocate_from=[self.sec_node])
5071 b6e82a65 Iustin Pop
5072 b6e82a65 Iustin Pop
    ial.Run(self.op.iallocator)
5073 b6e82a65 Iustin Pop
5074 b6e82a65 Iustin Pop
    if not ial.success:
5075 b6e82a65 Iustin Pop
      raise errors.OpPrereqError("Can't compute nodes using"
5076 b6e82a65 Iustin Pop
                                 " iallocator '%s': %s" % (self.op.iallocator,
5077 b6e82a65 Iustin Pop
                                                           ial.info))
5078 b6e82a65 Iustin Pop
    if len(ial.nodes) != ial.required_nodes:
5079 b6e82a65 Iustin Pop
      raise errors.OpPrereqError("iallocator '%s' returned invalid number"
5080 b6e82a65 Iustin Pop
                                 " of nodes (%s), required %s" %
5081 b6e82a65 Iustin Pop
                                 (len(ial.nodes), ial.required_nodes))
5082 b6e82a65 Iustin Pop
    self.op.remote_node = ial.nodes[0]
5083 86d9d3bb Iustin Pop
    self.LogInfo("Selected new secondary for the instance: %s",
5084 86d9d3bb Iustin Pop
                 self.op.remote_node)
5085 b6e82a65 Iustin Pop
5086 a8083063 Iustin Pop
  def BuildHooksEnv(self):
5087 a8083063 Iustin Pop
    """Build hooks env.
5088 a8083063 Iustin Pop

5089 a8083063 Iustin Pop
    This runs on the master, the primary and all the secondaries.
5090 a8083063 Iustin Pop

5091 a8083063 Iustin Pop
    """
5092 a8083063 Iustin Pop
    env = {
5093 a9e0c397 Iustin Pop
      "MODE": self.op.mode,
5094 a8083063 Iustin Pop
      "NEW_SECONDARY": self.op.remote_node,
5095 a8083063 Iustin Pop
      "OLD_SECONDARY": self.instance.secondary_nodes[0],
5096 a8083063 Iustin Pop
      }
5097 338e51e8 Iustin Pop
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
5098 0834c866 Iustin Pop
    nl = [
5099 d6a02168 Michael Hanselmann
      self.cfg.GetMasterNode(),
5100 0834c866 Iustin Pop
      self.instance.primary_node,
5101 0834c866 Iustin Pop
      ]
5102 0834c866 Iustin Pop
    if self.op.remote_node is not None:
5103 0834c866 Iustin Pop
      nl.append(self.op.remote_node)
5104 a8083063 Iustin Pop
    return env, nl, nl
5105 a8083063 Iustin Pop
5106 a8083063 Iustin Pop
  def CheckPrereq(self):
5107 a8083063 Iustin Pop
    """Check prerequisites.
5108 a8083063 Iustin Pop

5109 a8083063 Iustin Pop
    This checks that the instance is in the cluster.
5110 a8083063 Iustin Pop

5111 a8083063 Iustin Pop
    """
5112 efd990e4 Guido Trotter
    instance = self.cfg.GetInstanceInfo(self.op.instance_name)
5113 efd990e4 Guido Trotter
    assert instance is not None, \
5114 efd990e4 Guido Trotter
      "Cannot retrieve locked instance %s" % self.op.instance_name
5115 a8083063 Iustin Pop
    self.instance = instance
5116 a8083063 Iustin Pop
5117 7e9366f7 Iustin Pop
    if instance.disk_template != constants.DT_DRBD8:
5118 7e9366f7 Iustin Pop
      raise errors.OpPrereqError("Can only run replace disks for DRBD8-based"
5119 7e9366f7 Iustin Pop
                                 " instances")
5120 a8083063 Iustin Pop
5121 a8083063 Iustin Pop
    if len(instance.secondary_nodes) != 1:
5122 3ecf6786 Iustin Pop
      raise errors.OpPrereqError("The instance has a strange layout,"
5123 3ecf6786 Iustin Pop
                                 " expected one secondary but found %d" %
5124 3ecf6786 Iustin Pop
                                 len(instance.secondary_nodes))
5125 a8083063 Iustin Pop
5126 a9e0c397 Iustin Pop
    self.sec_node = instance.secondary_nodes[0]
5127 a9e0c397 Iustin Pop
5128 7e9366f7 Iustin Pop
    if self.op.iallocator is not None:
5129 de8c7666 Guido Trotter
      self._RunAllocator()
5130 b6e82a65 Iustin Pop
5131 b6e82a65 Iustin Pop
    remote_node = self.op.remote_node
5132 a9e0c397 Iustin Pop
    if remote_node is not None:
5133 a9e0c397 Iustin Pop
      self.remote_node_info = self.cfg.GetNodeInfo(remote_node)
5134 efd990e4 Guido Trotter
      assert self.remote_node_info is not None, \
5135 efd990e4 Guido Trotter
        "Cannot retrieve locked node %s" % remote_node
5136 a9e0c397 Iustin Pop
    else:
5137 a9e0c397 Iustin Pop
      self.remote_node_info = None
5138 a8083063 Iustin Pop
    if remote_node == instance.primary_node:
5139 3ecf6786 Iustin Pop
      raise errors.OpPrereqError("The specified node is the primary node of"
5140 3ecf6786 Iustin Pop
                                 " the instance.")
5141 a9e0c397 Iustin Pop
    elif remote_node == self.sec_node:
5142 7e9366f7 Iustin Pop
      raise errors.OpPrereqError("The specified node is already the"
5143 7e9366f7 Iustin Pop
                                 " secondary node of the instance.")
5144 7e9366f7 Iustin Pop
5145 7e9366f7 Iustin Pop
    if self.op.mode == constants.REPLACE_DISK_PRI:
5146 7e9366f7 Iustin Pop
      n1 = self.tgt_node = instance.primary_node
5147 7e9366f7 Iustin Pop
      n2 = self.oth_node = self.sec_node
5148 7e9366f7 Iustin Pop
    elif self.op.mode == constants.REPLACE_DISK_SEC:
5149 7e9366f7 Iustin Pop
      n1 = self.tgt_node = self.sec_node
5150 7e9366f7 Iustin Pop
      n2 = self.oth_node = instance.primary_node
5151 7e9366f7 Iustin Pop
    elif self.op.mode == constants.REPLACE_DISK_CHG:
5152 7e9366f7 Iustin Pop
      n1 = self.new_node = remote_node
5153 7e9366f7 Iustin Pop
      n2 = self.oth_node = instance.primary_node
5154 7e9366f7 Iustin Pop
      self.tgt_node = self.sec_node
5155 733a2b6a Iustin Pop
      _CheckNodeNotDrained(self, remote_node)
5156 7e9366f7 Iustin Pop
    else:
5157 7e9366f7 Iustin Pop
      raise errors.ProgrammerError("Unhandled disk replace mode")
5158 7e9366f7 Iustin Pop
5159 7e9366f7 Iustin Pop
    _CheckNodeOnline(self, n1)
5160 7e9366f7 Iustin Pop
    _CheckNodeOnline(self, n2)
5161 a9e0c397 Iustin Pop
5162 54155f52 Iustin Pop
    if not self.op.disks:
5163 54155f52 Iustin Pop
      self.op.disks = range(len(instance.disks))
5164 54155f52 Iustin Pop
5165 54155f52 Iustin Pop
    for disk_idx in self.op.disks:
5166 3e0cea06 Iustin Pop
      instance.FindDisk(disk_idx)
5167 a8083063 Iustin Pop
5168 a9e0c397 Iustin Pop
  def _ExecD8DiskOnly(self, feedback_fn):
5169 a9e0c397 Iustin Pop
    """Replace a disk on the primary or secondary for dbrd8.
5170 a9e0c397 Iustin Pop

5171 a9e0c397 Iustin Pop
    The algorithm for replace is quite complicated:
5172 e4376078 Iustin Pop

5173 e4376078 Iustin Pop
      1. for each disk to be replaced:
5174 e4376078 Iustin Pop

5175 e4376078 Iustin Pop
        1. create new LVs on the target node with unique names
5176 e4376078 Iustin Pop
        1. detach old LVs from the drbd device
5177 e4376078 Iustin Pop
        1. rename old LVs to name_replaced.<time_t>
5178 e4376078 Iustin Pop
        1. rename new LVs to old LVs
5179 e4376078 Iustin Pop
        1. attach the new LVs (with the old names now) to the drbd device
5180 e4376078 Iustin Pop

5181 e4376078 Iustin Pop
      1. wait for sync across all devices
5182 e4376078 Iustin Pop

5183 e4376078 Iustin Pop
      1. for each modified disk:
5184 e4376078 Iustin Pop

5185 e4376078 Iustin Pop
        1. remove old LVs (which have the name name_replaces.<time_t>)
5186 a9e0c397 Iustin Pop

5187 a9e0c397 Iustin Pop
    Failures are not very well handled.
5188 cff90b79 Iustin Pop

5189 a9e0c397 Iustin Pop
    """
5190 cff90b79 Iustin Pop
    steps_total = 6
5191 5bfac263 Iustin Pop
    warning, info = (self.proc.LogWarning, self.proc.LogInfo)
5192 a9e0c397 Iustin Pop
    instance = self.instance
5193 a9e0c397 Iustin Pop
    iv_names = {}
5194 a9e0c397 Iustin Pop
    vgname = self.cfg.GetVGName()
5195 a9e0c397 Iustin Pop
    # start of work
5196 a9e0c397 Iustin Pop
    cfg = self.cfg
5197 a9e0c397 Iustin Pop
    tgt_node = self.tgt_node
5198 cff90b79 Iustin Pop
    oth_node = self.oth_node
5199 cff90b79 Iustin Pop
5200 cff90b79 Iustin Pop
    # Step: check device activation
5201 5bfac263 Iustin Pop
    self.proc.LogStep(1, steps_total, "check device existence")
5202 cff90b79 Iustin Pop
    info("checking volume groups")
5203 cff90b79 Iustin Pop
    my_vg = cfg.GetVGName()
5204 72737a7f Iustin Pop
    results = self.rpc.call_vg_list([oth_node, tgt_node])
5205 cff90b79 Iustin Pop
    if not results:
5206 cff90b79 Iustin Pop
      raise errors.OpExecError("Can't list volume groups on the nodes")
5207 cff90b79 Iustin Pop
    for node in oth_node, tgt_node:
5208 781de953 Iustin Pop
      res = results[node]
5209 781de953 Iustin Pop
      if res.failed or not res.data or my_vg not in res.data:
5210 cff90b79 Iustin Pop
        raise errors.OpExecError("Volume group '%s' not found on %s" %
5211 cff90b79 Iustin Pop
                                 (my_vg, node))
5212 54155f52 Iustin Pop
    for idx, dev in enumerate(instance.disks):
5213 54155f52 Iustin Pop
      if idx not in self.op.disks:
5214 cff90b79 Iustin Pop
        continue
5215 cff90b79 Iustin Pop
      for node in tgt_node, oth_node:
5216 54155f52 Iustin Pop
        info("checking disk/%d on %s" % (idx, node))
5217 cff90b79 Iustin Pop
        cfg.SetDiskID(dev, node)
5218 23829f6f Iustin Pop
        result = self.rpc.call_blockdev_find(node, dev)
5219 23829f6f Iustin Pop
        msg = result.RemoteFailMsg()
5220 23829f6f Iustin Pop
        if not msg and not result.payload:
5221 23829f6f Iustin Pop
          msg = "disk not found"
5222 23829f6f Iustin Pop
        if msg:
5223 23829f6f Iustin Pop
          raise errors.OpExecError("Can't find disk/%d on node %s: %s" %
5224 23829f6f Iustin Pop
                                   (idx, node, msg))
5225 cff90b79 Iustin Pop
5226 cff90b79 Iustin Pop
    # Step: check other node consistency
5227 5bfac263 Iustin Pop
    self.proc.LogStep(2, steps_total, "check peer consistency")
5228 54155f52 Iustin Pop
    for idx, dev in enumerate(instance.disks):
5229 54155f52 Iustin Pop
      if idx not in self.op.disks:
5230 cff90b79 Iustin Pop
        continue
5231 54155f52 Iustin Pop
      info("checking disk/%d consistency on %s" % (idx, oth_node))
5232 b9bddb6b Iustin Pop
      if not _CheckDiskConsistency(self, dev, oth_node,
5233 cff90b79 Iustin Pop
                                   oth_node==instance.primary_node):
5234 cff90b79 Iustin Pop
        raise errors.OpExecError("Peer node (%s) has degraded storage, unsafe"
5235 cff90b79 Iustin Pop
                                 " to replace disks on this node (%s)" %
5236 cff90b79 Iustin Pop
                                 (oth_node, tgt_node))
5237 cff90b79 Iustin Pop
5238 cff90b79 Iustin Pop
    # Step: create new storage
5239 5bfac263 Iustin Pop
    self.proc.LogStep(3, steps_total, "allocate new storage")
5240 54155f52 Iustin Pop
    for idx, dev in enumerate(instance.disks):
5241 54155f52 Iustin Pop
      if idx not in self.op.disks:
5242 a9e0c397 Iustin Pop
        continue
5243 a9e0c397 Iustin Pop
      size = dev.size
5244 a9e0c397 Iustin Pop
      cfg.SetDiskID(dev, tgt_node)
5245 54155f52 Iustin Pop
      lv_names = [".disk%d_%s" % (idx, suf)
5246 54155f52 Iustin Pop
                  for suf in ["data", "meta"]]
5247 b9bddb6b Iustin Pop
      names = _GenerateUniqueNames(self, lv_names)
5248 a9e0c397 Iustin Pop
      lv_data = objects.Disk(dev_type=constants.LD_LV, size=size,
5249 a9e0c397 Iustin Pop
                             logical_id=(vgname, names[0]))
5250 a9e0c397 Iustin Pop
      lv_meta = objects.Disk(dev_type=constants.LD_LV, size=128,
5251 a9e0c397 Iustin Pop
                             logical_id=(vgname, names[1]))
5252 a9e0c397 Iustin Pop
      new_lvs = [lv_data, lv_meta]
5253 a9e0c397 Iustin Pop
      old_lvs = dev.children
5254 a9e0c397 Iustin Pop
      iv_names[dev.iv_name] = (dev, old_lvs, new_lvs)
5255 cff90b79 Iustin Pop
      info("creating new local storage on %s for %s" %
5256 cff90b79 Iustin Pop
           (tgt_node, dev.iv_name))
5257 428958aa Iustin Pop
      # we pass force_create=True to force the LVM creation
5258 a9e0c397 Iustin Pop
      for new_lv in new_lvs:
5259 428958aa Iustin Pop
        _CreateBlockDev(self, tgt_node, instance, new_lv, True,
5260 428958aa Iustin Pop
                        _GetInstanceInfoText(instance), False)
5261 a9e0c397 Iustin Pop
5262 cff90b79 Iustin Pop
    # Step: for each lv, detach+rename*2+attach
5263 5bfac263 Iustin Pop
    self.proc.LogStep(4, steps_total, "change drbd configuration")
5264 cff90b79 Iustin Pop
    for dev, old_lvs, new_lvs in iv_names.itervalues():
5265 cff90b79 Iustin Pop
      info("detaching %s drbd from local storage" % dev.iv_name)
5266 781de953 Iustin Pop
      result = self.rpc.call_blockdev_removechildren(tgt_node, dev, old_lvs)
5267 9205a895 Iustin Pop
      msg = result.RemoteFailMsg()
5268 9205a895 Iustin Pop
      if msg:
5269 a9e0c397 Iustin Pop
        raise errors.OpExecError("Can't detach drbd from local storage on node"
5270 9205a895 Iustin Pop
                                 " %s for device %s: %s" %
5271 9205a895 Iustin Pop
                                 (tgt_node, dev.iv_name, msg))
5272 cff90b79 Iustin Pop
      #dev.children = []
5273 cff90b79 Iustin Pop
      #cfg.Update(instance)
5274 a9e0c397 Iustin Pop
5275 a9e0c397 Iustin Pop
      # ok, we created the new LVs, so now we know we have the needed
5276 a9e0c397 Iustin Pop
      # storage; as such, we proceed on the target node to rename
5277 a9e0c397 Iustin Pop
      # old_lv to _old, and new_lv to old_lv; note that we rename LVs
5278 c99a3cc0 Manuel Franceschini
      # using the assumption that logical_id == physical_id (which in
5279 a9e0c397 Iustin Pop
      # turn is the unique_id on that node)
5280 cff90b79 Iustin Pop
5281 cff90b79 Iustin Pop
      # FIXME(iustin): use a better name for the replaced LVs
5282 a9e0c397 Iustin Pop
      temp_suffix = int(time.time())
5283 a9e0c397 Iustin Pop
      ren_fn = lambda d, suff: (d.physical_id[0],
5284 a9e0c397 Iustin Pop
                                d.physical_id[1] + "_replaced-%s" % suff)
5285 cff90b79 Iustin Pop
      # build the rename list based on what LVs exist on the node
5286 cff90b79 Iustin Pop
      rlist = []
5287 cff90b79 Iustin Pop
      for to_ren in old_lvs:
5288 23829f6f Iustin Pop
        result = self.rpc.call_blockdev_find(tgt_node, to_ren)
5289 23829f6f Iustin Pop
        if not result.RemoteFailMsg() and result.payload:
5290 23829f6f Iustin Pop
          # device exists
5291 cff90b79 Iustin Pop
          rlist.append((to_ren, ren_fn(to_ren, temp_suffix)))
5292 cff90b79 Iustin Pop
5293 cff90b79 Iustin Pop
      info("renaming the old LVs on the target node")
5294 781de953 Iustin Pop
      result = self.rpc.call_blockdev_rename(tgt_node, rlist)
5295 6b5e3f70 Iustin Pop
      msg = result.RemoteFailMsg()
5296 6b5e3f70 Iustin Pop
      if msg:
5297 6b5e3f70 Iustin Pop
        raise errors.OpExecError("Can't rename old LVs on node %s: %s" %
5298 6b5e3f70 Iustin Pop
                                 (tgt_node, msg))
5299 a9e0c397 Iustin Pop
      # now we rename the new LVs to the old LVs
5300 cff90b79 Iustin Pop
      info("renaming the new LVs on the target node")
5301 a9e0c397 Iustin Pop
      rlist = [(new, old.physical_id) for old, new in zip(old_lvs, new_lvs)]
5302 781de953 Iustin Pop
      result = self.rpc.call_blockdev_rename(tgt_node, rlist)
5303 6b5e3f70 Iustin Pop
      msg = result.RemoteFailMsg()
5304 6b5e3f70 Iustin Pop
      if msg:
5305 6b5e3f70 Iustin Pop
        raise errors.OpExecError("Can't rename new LVs on node %s: %s" %
5306 6b5e3f70 Iustin Pop
                                 (tgt_node, msg))
5307 cff90b79 Iustin Pop
5308 cff90b79 Iustin Pop
      for old, new in zip(old_lvs, new_lvs):
5309 cff90b79 Iustin Pop
        new.logical_id = old.logical_id
5310 cff90b79 Iustin Pop
        cfg.SetDiskID(new, tgt_node)
5311 a9e0c397 Iustin Pop
5312 cff90b79 Iustin Pop
      for disk in old_lvs:
5313 cff90b79 Iustin Pop
        disk.logical_id = ren_fn(disk, temp_suffix)
5314 cff90b79 Iustin Pop
        cfg.SetDiskID(disk, tgt_node)
5315 a9e0c397 Iustin Pop
5316 a9e0c397 Iustin Pop
      # now that the new lvs have the old name, we can add them to the device
5317 cff90b79 Iustin Pop
      info("adding new mirror component on %s" % tgt_node)
5318 4504c3d6 Iustin Pop
      result = self.rpc.call_blockdev_addchildren(tgt_node, dev, new_lvs)
5319 2cc1da8b Iustin Pop
      msg = result.RemoteFailMsg()
5320 2cc1da8b Iustin Pop
      if msg:
5321 a9e0c397 Iustin Pop
        for new_lv in new_lvs:
5322 e1bc0878 Iustin Pop
          msg = self.rpc.call_blockdev_remove(tgt_node, new_lv).RemoteFailMsg()
5323 e1bc0878 Iustin Pop
          if msg:
5324 e1bc0878 Iustin Pop
            warning("Can't rollback device %s: %s", dev, msg,
5325 e1bc0878 Iustin Pop
                    hint="cleanup manually the unused logical volumes")
5326 2cc1da8b Iustin Pop
        raise errors.OpExecError("Can't add local storage to drbd: %s" % msg)
5327 a9e0c397 Iustin Pop
5328 a9e0c397 Iustin Pop
      dev.children = new_lvs
5329 a9e0c397 Iustin Pop
      cfg.Update(instance)
5330 a9e0c397 Iustin Pop
5331 cff90b79 Iustin Pop
    # Step: wait for sync
5332 a9e0c397 Iustin Pop
5333 a9e0c397 Iustin Pop
    # this can fail as the old devices are degraded and _WaitForSync
5334 a9e0c397 Iustin Pop
    # does a combined result over all disks, so we don't check its
5335 a9e0c397 Iustin Pop
    # return value
5336 5bfac263 Iustin Pop
    self.proc.LogStep(5, steps_total, "sync devices")
5337 b9bddb6b Iustin Pop
    _WaitForSync(self, instance, unlock=True)
5338 a9e0c397 Iustin Pop
5339 a9e0c397 Iustin Pop
    # so check manually all the devices
5340 a9e0c397 Iustin Pop
    for name, (dev, old_lvs, new_lvs) in iv_names.iteritems():
5341 a9e0c397 Iustin Pop
      cfg.SetDiskID(dev, instance.primary_node)
5342 781de953 Iustin Pop
      result = self.rpc.call_blockdev_find(instance.primary_node, dev)
5343 23829f6f Iustin Pop
      msg = result.RemoteFailMsg()
5344 23829f6f Iustin Pop
      if not msg and not result.payload:
5345 23829f6f Iustin Pop
        msg = "disk not found"
5346 23829f6f Iustin Pop
      if msg:
5347 23829f6f Iustin Pop
        raise errors.OpExecError("Can't find DRBD device %s: %s" %
5348 23829f6f Iustin Pop
                                 (name, msg))
5349 23829f6f Iustin Pop
      if result.payload[5]:
5350 a9e0c397 Iustin Pop
        raise errors.OpExecError("DRBD device %s is degraded!" % name)
5351 a9e0c397 Iustin Pop
5352 cff90b79 Iustin Pop
    # Step: remove old storage
5353 5bfac263 Iustin Pop
    self.proc.LogStep(6, steps_total, "removing old storage")
5354 a9e0c397 Iustin Pop
    for name, (dev, old_lvs, new_lvs) in iv_names.iteritems():
5355 cff90b79 Iustin Pop
      info("remove logical volumes for %s" % name)
5356 a9e0c397 Iustin Pop
      for lv in old_lvs:
5357 a9e0c397 Iustin Pop
        cfg.SetDiskID(lv, tgt_node)
5358 e1bc0878 Iustin Pop
        msg = self.rpc.call_blockdev_remove(tgt_node, lv).RemoteFailMsg()
5359 e1bc0878 Iustin Pop
        if msg:
5360 e1bc0878 Iustin Pop
          warning("Can't remove old LV: %s" % msg,
5361 e1bc0878 Iustin Pop
                  hint="manually remove unused LVs")
5362 a9e0c397 Iustin Pop
          continue
5363 a9e0c397 Iustin Pop
5364 a9e0c397 Iustin Pop
  def _ExecD8Secondary(self, feedback_fn):
5365 a9e0c397 Iustin Pop
    """Replace the secondary node for drbd8.
5366 a9e0c397 Iustin Pop

5367 a9e0c397 Iustin Pop
    The algorithm for replace is quite complicated:
5368 a9e0c397 Iustin Pop
      - for all disks of the instance:
5369 a9e0c397 Iustin Pop
        - create new LVs on the new node with same names
5370 a9e0c397 Iustin Pop
        - shutdown the drbd device on the old secondary
5371 a9e0c397 Iustin Pop
        - disconnect the drbd network on the primary
5372 a9e0c397 Iustin Pop
        - create the drbd device on the new secondary
5373 a9e0c397 Iustin Pop
        - network attach the drbd on the primary, using an artifice:
5374 a9e0c397 Iustin Pop
          the drbd code for Attach() will connect to the network if it
5375 a9e0c397 Iustin Pop
          finds a device which is connected to the good local disks but
5376 a9e0c397 Iustin Pop
          not network enabled
5377 a9e0c397 Iustin Pop
      - wait for sync across all devices
5378 a9e0c397 Iustin Pop
      - remove all disks from the old secondary
5379 a9e0c397 Iustin Pop

5380 a9e0c397 Iustin Pop
    Failures are not very well handled.
5381 0834c866 Iustin Pop

5382 a9e0c397 Iustin Pop
    """
5383 0834c866 Iustin Pop
    steps_total = 6
5384 5bfac263 Iustin Pop
    warning, info = (self.proc.LogWarning, self.proc.LogInfo)
5385 a9e0c397 Iustin Pop
    instance = self.instance
5386 a9e0c397 Iustin Pop
    iv_names = {}
5387 a9e0c397 Iustin Pop
    # start of work
5388 a9e0c397 Iustin Pop
    cfg = self.cfg
5389 a9e0c397 Iustin Pop
    old_node = self.tgt_node
5390 a9e0c397 Iustin Pop
    new_node = self.new_node
5391 a9e0c397 Iustin Pop
    pri_node = instance.primary_node
5392 a2d59d8b Iustin Pop
    nodes_ip = {
5393 a2d59d8b Iustin Pop
      old_node: self.cfg.GetNodeInfo(old_node).secondary_ip,
5394 a2d59d8b Iustin Pop
      new_node: self.cfg.GetNodeInfo(new_node).secondary_ip,
5395 a2d59d8b Iustin Pop
      pri_node: self.cfg.GetNodeInfo(pri_node).secondary_ip,
5396 a2d59d8b Iustin Pop
      }
5397 0834c866 Iustin Pop
5398 0834c866 Iustin Pop
    # Step: check device activation
5399 5bfac263 Iustin Pop
    self.proc.LogStep(1, steps_total, "check device existence")
5400 0834c866 Iustin Pop
    info("checking volume groups")
5401 0834c866 Iustin Pop
    my_vg = cfg.GetVGName()
5402 72737a7f Iustin Pop
    results = self.rpc.call_vg_list([pri_node, new_node])
5403 0834c866 Iustin Pop
    for node in pri_node, new_node:
5404 781de953 Iustin Pop
      res = results[node]
5405 781de953 Iustin Pop
      if res.failed or not res.data or my_vg not in res.data:
5406 0834c866 Iustin Pop
        raise errors.OpExecError("Volume group '%s' not found on %s" %
5407 0834c866 Iustin Pop
                                 (my_vg, node))
5408 d418ebfb Iustin Pop
    for idx, dev in enumerate(instance.disks):
5409 d418ebfb Iustin Pop
      if idx not in self.op.disks:
5410 0834c866 Iustin Pop
        continue
5411 d418ebfb Iustin Pop
      info("checking disk/%d on %s" % (idx, pri_node))
5412 0834c866 Iustin Pop
      cfg.SetDiskID(dev, pri_node)
5413 781de953 Iustin Pop
      result = self.rpc.call_blockdev_find(pri_node, dev)
5414 23829f6f Iustin Pop
      msg = result.RemoteFailMsg()
5415 23829f6f Iustin Pop
      if not msg and not result.payload:
5416 23829f6f Iustin Pop
        msg = "disk not found"
5417 23829f6f Iustin Pop
      if msg:
5418 23829f6f Iustin Pop
        raise errors.OpExecError("Can't find disk/%d on node %s: %s" %
5419 23829f6f Iustin Pop
                                 (idx, pri_node, msg))
5420 0834c866 Iustin Pop
5421 0834c866 Iustin Pop
    # Step: check other node consistency
5422 5bfac263 Iustin Pop
    self.proc.LogStep(2, steps_total, "check peer consistency")
5423 d418ebfb Iustin Pop
    for idx, dev in enumerate(instance.disks):
5424 d418ebfb Iustin Pop
      if idx not in self.op.disks:
5425 0834c866 Iustin Pop
        continue
5426 d418ebfb Iustin Pop
      info("checking disk/%d consistency on %s" % (idx, pri_node))
5427 b9bddb6b Iustin Pop
      if not _CheckDiskConsistency(self, dev, pri_node, True, ldisk=True):
5428 0834c866 Iustin Pop
        raise errors.OpExecError("Primary node (%s) has degraded storage,"
5429 0834c866 Iustin Pop
                                 " unsafe to replace the secondary" %
5430 0834c866 Iustin Pop
                                 pri_node)
5431 0834c866 Iustin Pop
5432 0834c866 Iustin Pop
    # Step: create new storage
5433 5bfac263 Iustin Pop
    self.proc.LogStep(3, steps_total, "allocate new storage")
5434 d418ebfb Iustin Pop
    for idx, dev in enumerate(instance.disks):
5435 d418ebfb Iustin Pop
      info("adding new local storage on %s for disk/%d" %
5436 d418ebfb Iustin Pop
           (new_node, idx))
5437 428958aa Iustin Pop
      # we pass force_create=True to force LVM creation
5438 a9e0c397 Iustin Pop
      for new_lv in dev.children:
5439 428958aa Iustin Pop
        _CreateBlockDev(self, new_node, instance, new_lv, True,
5440 428958aa Iustin Pop
                        _GetInstanceInfoText(instance), False)
5441 a9e0c397 Iustin Pop
5442 468b46f9 Iustin Pop
    # Step 4: dbrd minors and drbd setups changes
5443 a1578d63 Iustin Pop
    # after this, we must manually remove the drbd minors on both the
5444 a1578d63 Iustin Pop
    # error and the success paths
5445 a1578d63 Iustin Pop
    minors = cfg.AllocateDRBDMinor([new_node for dev in instance.disks],
5446 a1578d63 Iustin Pop
                                   instance.name)
5447 468b46f9 Iustin Pop
    logging.debug("Allocated minors %s" % (minors,))
5448 5bfac263 Iustin Pop
    self.proc.LogStep(4, steps_total, "changing drbd configuration")
5449 d418ebfb Iustin Pop
    for idx, (dev, new_minor) in enumerate(zip(instance.disks, minors)):
5450 0834c866 Iustin Pop
      size = dev.size
5451 d418ebfb Iustin Pop
      info("activating a new drbd on %s for disk/%d" % (new_node, idx))
5452 a2d59d8b Iustin Pop
      # create new devices on new_node; note that we create two IDs:
5453 a2d59d8b Iustin Pop
      # one without port, so the drbd will be activated without
5454 a2d59d8b Iustin Pop
      # networking information on the new node at this stage, and one
5455 a2d59d8b Iustin Pop
      # with network, for the latter activation in step 4
5456 a2d59d8b Iustin Pop
      (o_node1, o_node2, o_port, o_minor1, o_minor2, o_secret) = dev.logical_id
5457 a2d59d8b Iustin Pop
      if pri_node == o_node1:
5458 a2d59d8b Iustin Pop
        p_minor = o_minor1
5459 ffa1c0dc Iustin Pop
      else:
5460 a2d59d8b Iustin Pop
        p_minor = o_minor2
5461 a2d59d8b Iustin Pop
5462 a2d59d8b Iustin Pop
      new_alone_id = (pri_node, new_node, None, p_minor, new_minor, o_secret)
5463 a2d59d8b Iustin Pop
      new_net_id = (pri_node, new_node, o_port, p_minor, new_minor, o_secret)
5464 a2d59d8b Iustin Pop
5465 a2d59d8b Iustin Pop
      iv_names[idx] = (dev, dev.children, new_net_id)
5466 a1578d63 Iustin Pop
      logging.debug("Allocated new_minor: %s, new_logical_id: %s", new_minor,
5467 a2d59d8b Iustin Pop
                    new_net_id)
5468 a9e0c397 Iustin Pop
      new_drbd = objects.Disk(dev_type=constants.LD_DRBD8,
5469 a2d59d8b Iustin Pop
                              logical_id=new_alone_id,
5470 a9e0c397 Iustin Pop
                              children=dev.children)
5471 796cab27 Iustin Pop
      try:
5472 de12473a Iustin Pop
        _CreateSingleBlockDev(self, new_node, instance, new_drbd,
5473 de12473a Iustin Pop
                              _GetInstanceInfoText(instance), False)
5474 82759cb1 Iustin Pop
      except errors.GenericError:
5475 a1578d63 Iustin Pop
        self.cfg.ReleaseDRBDMinors(instance.name)
5476 796cab27 Iustin Pop
        raise
5477 a9e0c397 Iustin Pop
5478 d418ebfb Iustin Pop
    for idx, dev in enumerate(instance.disks):
5479 a9e0c397 Iustin Pop
      # we have new devices, shutdown the drbd on the old secondary
5480 d418ebfb Iustin Pop
      info("shutting down drbd for disk/%d on old node" % idx)
5481 a9e0c397 Iustin Pop
      cfg.SetDiskID(dev, old_node)
5482 cacfd1fd Iustin Pop
      msg = self.rpc.call_blockdev_shutdown(old_node, dev).RemoteFailMsg()
5483 cacfd1fd Iustin Pop
      if msg:
5484 cacfd1fd Iustin Pop
        warning("Failed to shutdown drbd for disk/%d on old node: %s" %
5485 cacfd1fd Iustin Pop
                (idx, msg),
5486 79caa9ed Guido Trotter
                hint="Please cleanup this device manually as soon as possible")
5487 a9e0c397 Iustin Pop
5488 642445d9 Iustin Pop
    info("detaching primary drbds from the network (=> standalone)")
5489 a2d59d8b Iustin Pop
    result = self.rpc.call_drbd_disconnect_net([pri_node], nodes_ip,
5490 a2d59d8b Iustin Pop
                                               instance.disks)[pri_node]
5491 642445d9 Iustin Pop
5492 a2d59d8b Iustin Pop
    msg = result.RemoteFailMsg()
5493 a2d59d8b Iustin Pop
    if msg:
5494 a2d59d8b Iustin Pop
      # detaches didn't succeed (unlikely)
5495 a1578d63 Iustin Pop
      self.cfg.ReleaseDRBDMinors(instance.name)
5496 a2d59d8b Iustin Pop
      raise errors.OpExecError("Can't detach the disks from the network on"
5497 a2d59d8b Iustin Pop
                               " old node: %s" % (msg,))
5498 642445d9 Iustin Pop
5499 642445d9 Iustin Pop
    # if we managed to detach at least one, we update all the disks of
5500 642445d9 Iustin Pop
    # the instance to point to the new secondary
5501 642445d9 Iustin Pop
    info("updating instance configuration")
5502 468b46f9 Iustin Pop
    for dev, _, new_logical_id in iv_names.itervalues():
5503 468b46f9 Iustin Pop
      dev.logical_id = new_logical_id
5504 a9e0c397 Iustin Pop
      cfg.SetDiskID(dev, pri_node)
5505 642445d9 Iustin Pop
    cfg.Update(instance)
5506 a9e0c397 Iustin Pop
5507 642445d9 Iustin Pop
    # and now perform the drbd attach
5508 642445d9 Iustin Pop
    info("attaching primary drbds to new secondary (standalone => connected)")
5509 a2d59d8b Iustin Pop
    result = self.rpc.call_drbd_attach_net([pri_node, new_node], nodes_ip,
5510 a2d59d8b Iustin Pop
                                           instance.disks, instance.name,
5511 a2d59d8b Iustin Pop
                                           False)
5512 a2d59d8b Iustin Pop
    for to_node, to_result in result.items():
5513 a2d59d8b Iustin Pop
      msg = to_result.RemoteFailMsg()
5514 a2d59d8b Iustin Pop
      if msg:
5515 a2d59d8b Iustin Pop
        warning("can't attach drbd disks on node %s: %s", to_node, msg,
5516 a2d59d8b Iustin Pop
                hint="please do a gnt-instance info to see the"
5517 a2d59d8b Iustin Pop
                " status of disks")
5518 a9e0c397 Iustin Pop
5519 a9e0c397 Iustin Pop
    # this can fail as the old devices are degraded and _WaitForSync
5520 a9e0c397 Iustin Pop
    # does a combined result over all disks, so we don't check its
5521 a9e0c397 Iustin Pop
    # return value
5522 5bfac263 Iustin Pop
    self.proc.LogStep(5, steps_total, "sync devices")
5523 b9bddb6b Iustin Pop
    _WaitForSync(self, instance, unlock=True)
5524 a9e0c397 Iustin Pop
5525 a9e0c397 Iustin Pop
    # so check manually all the devices
5526 d418ebfb Iustin Pop
    for idx, (dev, old_lvs, _) in iv_names.iteritems():
5527 a9e0c397 Iustin Pop
      cfg.SetDiskID(dev, pri_node)
5528 781de953 Iustin Pop
      result = self.rpc.call_blockdev_find(pri_node, dev)
5529 23829f6f Iustin Pop
      msg = result.RemoteFailMsg()
5530 23829f6f Iustin Pop
      if not msg and not result.payload:
5531 23829f6f Iustin Pop
        msg = "disk not found"
5532 23829f6f Iustin Pop
      if msg:
5533 23829f6f Iustin Pop
        raise errors.OpExecError("Can't find DRBD device disk/%d: %s" %
5534 23829f6f Iustin Pop
                                 (idx, msg))
5535 23829f6f Iustin Pop
      if result.payload[5]:
5536 d418ebfb Iustin Pop
        raise errors.OpExecError("DRBD device disk/%d is degraded!" % idx)
5537 a9e0c397 Iustin Pop
5538 5bfac263 Iustin Pop
    self.proc.LogStep(6, steps_total, "removing old storage")
5539 d418ebfb Iustin Pop
    for idx, (dev, old_lvs, _) in iv_names.iteritems():
5540 d418ebfb Iustin Pop
      info("remove logical volumes for disk/%d" % idx)
5541 a9e0c397 Iustin Pop
      for lv in old_lvs:
5542 a9e0c397 Iustin Pop
        cfg.SetDiskID(lv, old_node)
5543 e1bc0878 Iustin Pop
        msg = self.rpc.call_blockdev_remove(old_node, lv).RemoteFailMsg()
5544 e1bc0878 Iustin Pop
        if msg:
5545 e1bc0878 Iustin Pop
          warning("Can't remove LV on old secondary: %s", msg,
5546 79caa9ed Guido Trotter
                  hint="Cleanup stale volumes by hand")
5547 a9e0c397 Iustin Pop
5548 a9e0c397 Iustin Pop
  def Exec(self, feedback_fn):
5549 a9e0c397 Iustin Pop
    """Execute disk replacement.
5550 a9e0c397 Iustin Pop

5551 a9e0c397 Iustin Pop
    This dispatches the disk replacement to the appropriate handler.
5552 a9e0c397 Iustin Pop

5553 a9e0c397 Iustin Pop
    """
5554 a9e0c397 Iustin Pop
    instance = self.instance
5555 22985314 Guido Trotter
5556 22985314 Guido Trotter
    # Activate the instance disks if we're replacing them on a down instance
5557 0d68c45d Iustin Pop
    if not instance.admin_up:
5558 b9bddb6b Iustin Pop
      _StartInstanceDisks(self, instance, True)
5559 22985314 Guido Trotter
5560 7e9366f7 Iustin Pop
    if self.op.mode == constants.REPLACE_DISK_CHG:
5561 7e9366f7 Iustin Pop
      fn = self._ExecD8Secondary
5562 a9e0c397 Iustin Pop
    else:
5563 7e9366f7 Iustin Pop
      fn = self._ExecD8DiskOnly
5564 22985314 Guido Trotter
5565 22985314 Guido Trotter
    ret = fn(feedback_fn)
5566 22985314 Guido Trotter
5567 22985314 Guido Trotter
    # Deactivate the instance disks if we're replacing them on a down instance
5568 0d68c45d Iustin Pop
    if not instance.admin_up:
5569 b9bddb6b Iustin Pop
      _SafeShutdownInstanceDisks(self, instance)
5570 22985314 Guido Trotter
5571 22985314 Guido Trotter
    return ret
5572 a9e0c397 Iustin Pop
5573 a8083063 Iustin Pop
5574 8729e0d7 Iustin Pop
class LUGrowDisk(LogicalUnit):
5575 8729e0d7 Iustin Pop
  """Grow a disk of an instance.
5576 8729e0d7 Iustin Pop

5577 8729e0d7 Iustin Pop
  """
5578 8729e0d7 Iustin Pop
  HPATH = "disk-grow"
5579 8729e0d7 Iustin Pop
  HTYPE = constants.HTYPE_INSTANCE
5580 6605411d Iustin Pop
  _OP_REQP = ["instance_name", "disk", "amount", "wait_for_sync"]
5581 31e63dbf Guido Trotter
  REQ_BGL = False
5582 31e63dbf Guido Trotter
5583 31e63dbf Guido Trotter
  def ExpandNames(self):
5584 31e63dbf Guido Trotter
    self._ExpandAndLockInstance()
5585 31e63dbf Guido Trotter
    self.needed_locks[locking.LEVEL_NODE] = []
5586 f6d9a522 Guido Trotter
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
5587 31e63dbf Guido Trotter
5588 31e63dbf Guido Trotter
  def DeclareLocks(self, level):
5589 31e63dbf Guido Trotter
    if level == locking.LEVEL_NODE:
5590 31e63dbf Guido Trotter
      self._LockInstancesNodes()
5591 8729e0d7 Iustin Pop
5592 8729e0d7 Iustin Pop
  def BuildHooksEnv(self):
5593 8729e0d7 Iustin Pop
    """Build hooks env.
5594 8729e0d7 Iustin Pop

5595 8729e0d7 Iustin Pop
    This runs on the master, the primary and all the secondaries.
5596 8729e0d7 Iustin Pop

5597 8729e0d7 Iustin Pop
    """
5598 8729e0d7 Iustin Pop
    env = {
5599 8729e0d7 Iustin Pop
      "DISK": self.op.disk,
5600 8729e0d7 Iustin Pop
      "AMOUNT": self.op.amount,
5601 8729e0d7 Iustin Pop
      }
5602 338e51e8 Iustin Pop
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
5603 8729e0d7 Iustin Pop
    nl = [
5604 d6a02168 Michael Hanselmann
      self.cfg.GetMasterNode(),
5605 8729e0d7 Iustin Pop
      self.instance.primary_node,
5606 8729e0d7 Iustin Pop
      ]
5607 8729e0d7 Iustin Pop
    return env, nl, nl
5608 8729e0d7 Iustin Pop
5609 8729e0d7 Iustin Pop
  def CheckPrereq(self):
5610 8729e0d7 Iustin Pop
    """Check prerequisites.
5611 8729e0d7 Iustin Pop

5612 8729e0d7 Iustin Pop
    This checks that the instance is in the cluster.
5613 8729e0d7 Iustin Pop

5614 8729e0d7 Iustin Pop
    """
5615 31e63dbf Guido Trotter
    instance = self.cfg.GetInstanceInfo(self.op.instance_name)
5616 31e63dbf Guido Trotter
    assert instance is not None, \
5617 31e63dbf Guido Trotter
      "Cannot retrieve locked instance %s" % self.op.instance_name
5618 6b12959c Iustin Pop
    nodenames = list(instance.all_nodes)
5619 6b12959c Iustin Pop
    for node in nodenames:
5620 7527a8a4 Iustin Pop
      _CheckNodeOnline(self, node)
5621 7527a8a4 Iustin Pop
5622 31e63dbf Guido Trotter
5623 8729e0d7 Iustin Pop
    self.instance = instance
5624 8729e0d7 Iustin Pop
5625 8729e0d7 Iustin Pop
    if instance.disk_template not in (constants.DT_PLAIN, constants.DT_DRBD8):
5626 8729e0d7 Iustin Pop
      raise errors.OpPrereqError("Instance's disk layout does not support"
5627 8729e0d7 Iustin Pop
                                 " growing.")
5628 8729e0d7 Iustin Pop
5629 ad24e046 Iustin Pop
    self.disk = instance.FindDisk(self.op.disk)
5630 8729e0d7 Iustin Pop
5631 72737a7f Iustin Pop
    nodeinfo = self.rpc.call_node_info(nodenames, self.cfg.GetVGName(),
5632 72737a7f Iustin Pop
                                       instance.hypervisor)
5633 8729e0d7 Iustin Pop
    for node in nodenames:
5634 781de953 Iustin Pop
      info = nodeinfo[node]
5635 781de953 Iustin Pop
      if info.failed or not info.data:
5636 8729e0d7 Iustin Pop
        raise errors.OpPrereqError("Cannot get current information"
5637 8729e0d7 Iustin Pop
                                   " from node '%s'" % node)
5638 781de953 Iustin Pop
      vg_free = info.data.get('vg_free', None)
5639 8729e0d7 Iustin Pop
      if not isinstance(vg_free, int):
5640 8729e0d7 Iustin Pop
        raise errors.OpPrereqError("Can't compute free disk space on"
5641 8729e0d7 Iustin Pop
                                   " node %s" % node)
5642 781de953 Iustin Pop
      if self.op.amount > vg_free:
5643 8729e0d7 Iustin Pop
        raise errors.OpPrereqError("Not enough disk space on target node %s:"
5644 8729e0d7 Iustin Pop
                                   " %d MiB available, %d MiB required" %
5645 781de953 Iustin Pop
                                   (node, vg_free, self.op.amount))
5646 8729e0d7 Iustin Pop
5647 8729e0d7 Iustin Pop
  def Exec(self, feedback_fn):
5648 8729e0d7 Iustin Pop
    """Execute disk grow.
5649 8729e0d7 Iustin Pop

5650 8729e0d7 Iustin Pop
    """
5651 8729e0d7 Iustin Pop
    instance = self.instance
5652 ad24e046 Iustin Pop
    disk = self.disk
5653 6b12959c Iustin Pop
    for node in instance.all_nodes:
5654 8729e0d7 Iustin Pop
      self.cfg.SetDiskID(disk, node)
5655 72737a7f Iustin Pop
      result = self.rpc.call_blockdev_grow(node, disk, self.op.amount)
5656 0959c824 Iustin Pop
      msg = result.RemoteFailMsg()
5657 0959c824 Iustin Pop
      if msg:
5658 781de953 Iustin Pop
        raise errors.OpExecError("Grow request failed to node %s: %s" %
5659 0959c824 Iustin Pop
                                 (node, msg))
5660 8729e0d7 Iustin Pop
    disk.RecordGrow(self.op.amount)
5661 8729e0d7 Iustin Pop
    self.cfg.Update(instance)
5662 6605411d Iustin Pop
    if self.op.wait_for_sync:
5663 cd4d138f Guido Trotter
      disk_abort = not _WaitForSync(self, instance)
5664 6605411d Iustin Pop
      if disk_abort:
5665 86d9d3bb Iustin Pop
        self.proc.LogWarning("Warning: disk sync-ing has not returned a good"
5666 86d9d3bb Iustin Pop
                             " status.\nPlease check the instance.")
5667 8729e0d7 Iustin Pop
5668 8729e0d7 Iustin Pop
5669 a8083063 Iustin Pop
class LUQueryInstanceData(NoHooksLU):
5670 a8083063 Iustin Pop
  """Query runtime instance data.
5671 a8083063 Iustin Pop

5672 a8083063 Iustin Pop
  """
5673 57821cac Iustin Pop
  _OP_REQP = ["instances", "static"]
5674 a987fa48 Guido Trotter
  REQ_BGL = False
5675 ae5849b5 Michael Hanselmann
5676 a987fa48 Guido Trotter
  def ExpandNames(self):
5677 a987fa48 Guido Trotter
    self.needed_locks = {}
5678 a987fa48 Guido Trotter
    self.share_locks = dict(((i, 1) for i in locking.LEVELS))
5679 a987fa48 Guido Trotter
5680 a987fa48 Guido Trotter
    if not isinstance(self.op.instances, list):
5681 a987fa48 Guido Trotter
      raise errors.OpPrereqError("Invalid argument type 'instances'")
5682 a987fa48 Guido Trotter
5683 a987fa48 Guido Trotter
    if self.op.instances:
5684 a987fa48 Guido Trotter
      self.wanted_names = []
5685 a987fa48 Guido Trotter
      for name in self.op.instances:
5686 a987fa48 Guido Trotter
        full_name = self.cfg.ExpandInstanceName(name)
5687 a987fa48 Guido Trotter
        if full_name is None:
5688 f57c76e4 Iustin Pop
          raise errors.OpPrereqError("Instance '%s' not known" % name)
5689 a987fa48 Guido Trotter
        self.wanted_names.append(full_name)
5690 a987fa48 Guido Trotter
      self.needed_locks[locking.LEVEL_INSTANCE] = self.wanted_names
5691 a987fa48 Guido Trotter
    else:
5692 a987fa48 Guido Trotter
      self.wanted_names = None
5693 a987fa48 Guido Trotter
      self.needed_locks[locking.LEVEL_INSTANCE] = locking.ALL_SET
5694 a987fa48 Guido Trotter
5695 a987fa48 Guido Trotter
    self.needed_locks[locking.LEVEL_NODE] = []
5696 a987fa48 Guido Trotter
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
5697 a987fa48 Guido Trotter
5698 a987fa48 Guido Trotter
  def DeclareLocks(self, level):
5699 a987fa48 Guido Trotter
    if level == locking.LEVEL_NODE:
5700 a987fa48 Guido Trotter
      self._LockInstancesNodes()
5701 a8083063 Iustin Pop
5702 a8083063 Iustin Pop
  def CheckPrereq(self):
5703 a8083063 Iustin Pop
    """Check prerequisites.
5704 a8083063 Iustin Pop

5705 a8083063 Iustin Pop
    This only checks the optional instance list against the existing names.
5706 a8083063 Iustin Pop

5707 a8083063 Iustin Pop
    """
5708 a987fa48 Guido Trotter
    if self.wanted_names is None:
5709 a987fa48 Guido Trotter
      self.wanted_names = self.acquired_locks[locking.LEVEL_INSTANCE]
5710 a8083063 Iustin Pop
5711 a987fa48 Guido Trotter
    self.wanted_instances = [self.cfg.GetInstanceInfo(name) for name
5712 a987fa48 Guido Trotter
                             in self.wanted_names]
5713 a987fa48 Guido Trotter
    return
5714 a8083063 Iustin Pop
5715 a8083063 Iustin Pop
  def _ComputeDiskStatus(self, instance, snode, dev):
5716 a8083063 Iustin Pop
    """Compute block device status.
5717 a8083063 Iustin Pop

5718 a8083063 Iustin Pop
    """
5719 57821cac Iustin Pop
    static = self.op.static
5720 57821cac Iustin Pop
    if not static:
5721 57821cac Iustin Pop
      self.cfg.SetDiskID(dev, instance.primary_node)
5722 57821cac Iustin Pop
      dev_pstatus = self.rpc.call_blockdev_find(instance.primary_node, dev)
5723 9854f5d0 Iustin Pop
      if dev_pstatus.offline:
5724 9854f5d0 Iustin Pop
        dev_pstatus = None
5725 9854f5d0 Iustin Pop
      else:
5726 9854f5d0 Iustin Pop
        msg = dev_pstatus.RemoteFailMsg()
5727 9854f5d0 Iustin Pop
        if msg:
5728 9854f5d0 Iustin Pop
          raise errors.OpExecError("Can't compute disk status for %s: %s" %
5729 9854f5d0 Iustin Pop
                                   (instance.name, msg))
5730 9854f5d0 Iustin Pop
        dev_pstatus = dev_pstatus.payload
5731 57821cac Iustin Pop
    else:
5732 57821cac Iustin Pop
      dev_pstatus = None
5733 57821cac Iustin Pop
5734 a1f445d3 Iustin Pop
    if dev.dev_type in constants.LDS_DRBD:
5735 a8083063 Iustin Pop
      # we change the snode then (otherwise we use the one passed in)
5736 a8083063 Iustin Pop
      if dev.logical_id[0] == instance.primary_node:
5737 a8083063 Iustin Pop
        snode = dev.logical_id[1]
5738 a8083063 Iustin Pop
      else:
5739 a8083063 Iustin Pop
        snode = dev.logical_id[0]
5740 a8083063 Iustin Pop
5741 57821cac Iustin Pop
    if snode and not static:
5742 a8083063 Iustin Pop
      self.cfg.SetDiskID(dev, snode)
5743 72737a7f Iustin Pop
      dev_sstatus = self.rpc.call_blockdev_find(snode, dev)
5744 9854f5d0 Iustin Pop
      if dev_sstatus.offline:
5745 9854f5d0 Iustin Pop
        dev_sstatus = None
5746 9854f5d0 Iustin Pop
      else:
5747 9854f5d0 Iustin Pop
        msg = dev_sstatus.RemoteFailMsg()
5748 9854f5d0 Iustin Pop
        if msg:
5749 9854f5d0 Iustin Pop
          raise errors.OpExecError("Can't compute disk status for %s: %s" %
5750 9854f5d0 Iustin Pop
                                   (instance.name, msg))
5751 9854f5d0 Iustin Pop
        dev_sstatus = dev_sstatus.payload
5752 a8083063 Iustin Pop
    else:
5753 a8083063 Iustin Pop
      dev_sstatus = None
5754 a8083063 Iustin Pop
5755 a8083063 Iustin Pop
    if dev.children:
5756 a8083063 Iustin Pop
      dev_children = [self._ComputeDiskStatus(instance, snode, child)
5757 a8083063 Iustin Pop
                      for child in dev.children]
5758 a8083063 Iustin Pop
    else:
5759 a8083063 Iustin Pop
      dev_children = []
5760 a8083063 Iustin Pop
5761 a8083063 Iustin Pop
    data = {
5762 a8083063 Iustin Pop
      "iv_name": dev.iv_name,
5763 a8083063 Iustin Pop
      "dev_type": dev.dev_type,
5764 a8083063 Iustin Pop
      "logical_id": dev.logical_id,
5765 a8083063 Iustin Pop
      "physical_id": dev.physical_id,
5766 a8083063 Iustin Pop
      "pstatus": dev_pstatus,
5767 a8083063 Iustin Pop
      "sstatus": dev_sstatus,
5768 a8083063 Iustin Pop
      "children": dev_children,
5769 b6fdf8b8 Iustin Pop
      "mode": dev.mode,
5770 a8083063 Iustin Pop
      }
5771 a8083063 Iustin Pop
5772 a8083063 Iustin Pop
    return data
5773 a8083063 Iustin Pop
5774 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
5775 a8083063 Iustin Pop
    """Gather and return data"""
5776 a8083063 Iustin Pop
    result = {}
5777 338e51e8 Iustin Pop
5778 338e51e8 Iustin Pop
    cluster = self.cfg.GetClusterInfo()
5779 338e51e8 Iustin Pop
5780 a8083063 Iustin Pop
    for instance in self.wanted_instances:
5781 57821cac Iustin Pop
      if not self.op.static:
5782 57821cac Iustin Pop
        remote_info = self.rpc.call_instance_info(instance.primary_node,
5783 57821cac Iustin Pop
                                                  instance.name,
5784 57821cac Iustin Pop
                                                  instance.hypervisor)
5785 781de953 Iustin Pop
        remote_info.Raise()
5786 781de953 Iustin Pop
        remote_info = remote_info.data
5787 57821cac Iustin Pop
        if remote_info and "state" in remote_info:
5788 57821cac Iustin Pop
          remote_state = "up"
5789 57821cac Iustin Pop
        else:
5790 57821cac Iustin Pop
          remote_state = "down"
5791 a8083063 Iustin Pop
      else:
5792 57821cac Iustin Pop
        remote_state = None
5793 0d68c45d Iustin Pop
      if instance.admin_up:
5794 a8083063 Iustin Pop
        config_state = "up"
5795 0d68c45d Iustin Pop
      else:
5796 0d68c45d Iustin Pop
        config_state = "down"
5797 a8083063 Iustin Pop
5798 a8083063 Iustin Pop
      disks = [self._ComputeDiskStatus(instance, None, device)
5799 a8083063 Iustin Pop
               for device in instance.disks]
5800 a8083063 Iustin Pop
5801 a8083063 Iustin Pop
      idict = {
5802 a8083063 Iustin Pop
        "name": instance.name,
5803 a8083063 Iustin Pop
        "config_state": config_state,
5804 a8083063 Iustin Pop
        "run_state": remote_state,
5805 a8083063 Iustin Pop
        "pnode": instance.primary_node,
5806 a8083063 Iustin Pop
        "snodes": instance.secondary_nodes,
5807 a8083063 Iustin Pop
        "os": instance.os,
5808 a8083063 Iustin Pop
        "nics": [(nic.mac, nic.ip, nic.bridge) for nic in instance.nics],
5809 a8083063 Iustin Pop
        "disks": disks,
5810 e69d05fd Iustin Pop
        "hypervisor": instance.hypervisor,
5811 24838135 Iustin Pop
        "network_port": instance.network_port,
5812 24838135 Iustin Pop
        "hv_instance": instance.hvparams,
5813 338e51e8 Iustin Pop
        "hv_actual": cluster.FillHV(instance),
5814 338e51e8 Iustin Pop
        "be_instance": instance.beparams,
5815 338e51e8 Iustin Pop
        "be_actual": cluster.FillBE(instance),
5816 a8083063 Iustin Pop
        }
5817 a8083063 Iustin Pop
5818 a8083063 Iustin Pop
      result[instance.name] = idict
5819 a8083063 Iustin Pop
5820 a8083063 Iustin Pop
    return result
5821 a8083063 Iustin Pop
5822 a8083063 Iustin Pop
5823 7767bbf5 Manuel Franceschini
class LUSetInstanceParams(LogicalUnit):
5824 a8083063 Iustin Pop
  """Modifies an instances's parameters.
5825 a8083063 Iustin Pop

5826 a8083063 Iustin Pop
  """
5827 a8083063 Iustin Pop
  HPATH = "instance-modify"
5828 a8083063 Iustin Pop
  HTYPE = constants.HTYPE_INSTANCE
5829 24991749 Iustin Pop
  _OP_REQP = ["instance_name"]
5830 1a5c7281 Guido Trotter
  REQ_BGL = False
5831 1a5c7281 Guido Trotter
5832 24991749 Iustin Pop
  def CheckArguments(self):
5833 24991749 Iustin Pop
    if not hasattr(self.op, 'nics'):
5834 24991749 Iustin Pop
      self.op.nics = []
5835 24991749 Iustin Pop
    if not hasattr(self.op, 'disks'):
5836 24991749 Iustin Pop
      self.op.disks = []
5837 24991749 Iustin Pop
    if not hasattr(self.op, 'beparams'):
5838 24991749 Iustin Pop
      self.op.beparams = {}
5839 24991749 Iustin Pop
    if not hasattr(self.op, 'hvparams'):
5840 24991749 Iustin Pop
      self.op.hvparams = {}
5841 24991749 Iustin Pop
    self.op.force = getattr(self.op, "force", False)
5842 24991749 Iustin Pop
    if not (self.op.nics or self.op.disks or
5843 24991749 Iustin Pop
            self.op.hvparams or self.op.beparams):
5844 24991749 Iustin Pop
      raise errors.OpPrereqError("No changes submitted")
5845 24991749 Iustin Pop
5846 24991749 Iustin Pop
    # Disk validation
5847 24991749 Iustin Pop
    disk_addremove = 0
5848 24991749 Iustin Pop
    for disk_op, disk_dict in self.op.disks:
5849 24991749 Iustin Pop
      if disk_op == constants.DDM_REMOVE:
5850 24991749 Iustin Pop
        disk_addremove += 1
5851 24991749 Iustin Pop
        continue
5852 24991749 Iustin Pop
      elif disk_op == constants.DDM_ADD:
5853 24991749 Iustin Pop
        disk_addremove += 1
5854 24991749 Iustin Pop
      else:
5855 24991749 Iustin Pop
        if not isinstance(disk_op, int):
5856 24991749 Iustin Pop
          raise errors.OpPrereqError("Invalid disk index")
5857 24991749 Iustin Pop
      if disk_op == constants.DDM_ADD:
5858 24991749 Iustin Pop
        mode = disk_dict.setdefault('mode', constants.DISK_RDWR)
5859 6ec66eae Iustin Pop
        if mode not in constants.DISK_ACCESS_SET:
5860 24991749 Iustin Pop
          raise errors.OpPrereqError("Invalid disk access mode '%s'" % mode)
5861 24991749 Iustin Pop
        size = disk_dict.get('size', None)
5862 24991749 Iustin Pop
        if size is None:
5863 24991749 Iustin Pop
          raise errors.OpPrereqError("Required disk parameter size missing")
5864 24991749 Iustin Pop
        try:
5865 24991749 Iustin Pop
          size = int(size)
5866 24991749 Iustin Pop
        except ValueError, err:
5867 24991749 Iustin Pop
          raise errors.OpPrereqError("Invalid disk size parameter: %s" %
5868 24991749 Iustin Pop
                                     str(err))
5869 24991749 Iustin Pop
        disk_dict['size'] = size
5870 24991749 Iustin Pop
      else:
5871 24991749 Iustin Pop
        # modification of disk
5872 24991749 Iustin Pop
        if 'size' in disk_dict:
5873 24991749 Iustin Pop
          raise errors.OpPrereqError("Disk size change not possible, use"
5874 24991749 Iustin Pop
                                     " grow-disk")
5875 24991749 Iustin Pop
5876 24991749 Iustin Pop
    if disk_addremove > 1:
5877 24991749 Iustin Pop
      raise errors.OpPrereqError("Only one disk add or remove operation"
5878 24991749 Iustin Pop
                                 " supported at a time")
5879 24991749 Iustin Pop
5880 24991749 Iustin Pop
    # NIC validation
5881 24991749 Iustin Pop
    nic_addremove = 0
5882 24991749 Iustin Pop
    for nic_op, nic_dict in self.op.nics:
5883 24991749 Iustin Pop
      if nic_op == constants.DDM_REMOVE:
5884 24991749 Iustin Pop
        nic_addremove += 1
5885 24991749 Iustin Pop
        continue
5886 24991749 Iustin Pop
      elif nic_op == constants.DDM_ADD:
5887 24991749 Iustin Pop
        nic_addremove += 1
5888 24991749 Iustin Pop
      else:
5889 24991749 Iustin Pop
        if not isinstance(nic_op, int):
5890 24991749 Iustin Pop
          raise errors.OpPrereqError("Invalid nic index")
5891 24991749 Iustin Pop
5892 24991749 Iustin Pop
      # nic_dict should be a dict
5893 24991749 Iustin Pop
      nic_ip = nic_dict.get('ip', None)
5894 24991749 Iustin Pop
      if nic_ip is not None:
5895 5c44da6a Guido Trotter
        if nic_ip.lower() == constants.VALUE_NONE:
5896 24991749 Iustin Pop
          nic_dict['ip'] = None
5897 24991749 Iustin Pop
        else:
5898 24991749 Iustin Pop
          if not utils.IsValidIP(nic_ip):
5899 24991749 Iustin Pop
            raise errors.OpPrereqError("Invalid IP address '%s'" % nic_ip)
5900 5c44da6a Guido Trotter
5901 5c44da6a Guido Trotter
      if nic_op == constants.DDM_ADD:
5902 5c44da6a Guido Trotter
        nic_bridge = nic_dict.get('bridge', None)
5903 5c44da6a Guido Trotter
        if nic_bridge is None:
5904 5c44da6a Guido Trotter
          nic_dict['bridge'] = self.cfg.GetDefBridge()
5905 5c44da6a Guido Trotter
        nic_mac = nic_dict.get('mac', None)
5906 5c44da6a Guido Trotter
        if nic_mac is None:
5907 5c44da6a Guido Trotter
          nic_dict['mac'] = constants.VALUE_AUTO
5908 5c44da6a Guido Trotter
5909 5c44da6a Guido Trotter
      if 'mac' in nic_dict:
5910 5c44da6a Guido Trotter
        nic_mac = nic_dict['mac']
5911 24991749 Iustin Pop
        if nic_mac not in (constants.VALUE_AUTO, constants.VALUE_GENERATE):
5912 24991749 Iustin Pop
          if not utils.IsValidMac(nic_mac):
5913 24991749 Iustin Pop
            raise errors.OpPrereqError("Invalid MAC address %s" % nic_mac)
5914 5c44da6a Guido Trotter
        if nic_op != constants.DDM_ADD and nic_mac == constants.VALUE_AUTO:
5915 5c44da6a Guido Trotter
          raise errors.OpPrereqError("'auto' is not a valid MAC address when"
5916 5c44da6a Guido Trotter
                                     " modifying an existing nic")
5917 5c44da6a Guido Trotter
5918 24991749 Iustin Pop
    if nic_addremove > 1:
5919 24991749 Iustin Pop
      raise errors.OpPrereqError("Only one NIC add or remove operation"
5920 24991749 Iustin Pop
                                 " supported at a time")
5921 24991749 Iustin Pop
5922 1a5c7281 Guido Trotter
  def ExpandNames(self):
5923 1a5c7281 Guido Trotter
    self._ExpandAndLockInstance()
5924 74409b12 Iustin Pop
    self.needed_locks[locking.LEVEL_NODE] = []
5925 74409b12 Iustin Pop
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
5926 74409b12 Iustin Pop
5927 74409b12 Iustin Pop
  def DeclareLocks(self, level):
5928 74409b12 Iustin Pop
    if level == locking.LEVEL_NODE:
5929 74409b12 Iustin Pop
      self._LockInstancesNodes()
5930 a8083063 Iustin Pop
5931 a8083063 Iustin Pop
  def BuildHooksEnv(self):
5932 a8083063 Iustin Pop
    """Build hooks env.
5933 a8083063 Iustin Pop

5934 a8083063 Iustin Pop
    This runs on the master, primary and secondaries.
5935 a8083063 Iustin Pop

5936 a8083063 Iustin Pop
    """
5937 396e1b78 Michael Hanselmann
    args = dict()
5938 338e51e8 Iustin Pop
    if constants.BE_MEMORY in self.be_new:
5939 338e51e8 Iustin Pop
      args['memory'] = self.be_new[constants.BE_MEMORY]
5940 338e51e8 Iustin Pop
    if constants.BE_VCPUS in self.be_new:
5941 61be6ba4 Iustin Pop
      args['vcpus'] = self.be_new[constants.BE_VCPUS]
5942 d8dcf3c9 Guido Trotter
    # TODO: export disk changes. Note: _BuildInstanceHookEnv* don't export disk
5943 d8dcf3c9 Guido Trotter
    # information at all.
5944 d8dcf3c9 Guido Trotter
    if self.op.nics:
5945 d8dcf3c9 Guido Trotter
      args['nics'] = []
5946 d8dcf3c9 Guido Trotter
      nic_override = dict(self.op.nics)
5947 d8dcf3c9 Guido Trotter
      for idx, nic in enumerate(self.instance.nics):
5948 d8dcf3c9 Guido Trotter
        if idx in nic_override:
5949 d8dcf3c9 Guido Trotter
          this_nic_override = nic_override[idx]
5950 d8dcf3c9 Guido Trotter
        else:
5951 d8dcf3c9 Guido Trotter
          this_nic_override = {}
5952 d8dcf3c9 Guido Trotter
        if 'ip' in this_nic_override:
5953 d8dcf3c9 Guido Trotter
          ip = this_nic_override['ip']
5954 d8dcf3c9 Guido Trotter
        else:
5955 d8dcf3c9 Guido Trotter
          ip = nic.ip
5956 d8dcf3c9 Guido Trotter
        if 'bridge' in this_nic_override:
5957 d8dcf3c9 Guido Trotter
          bridge = this_nic_override['bridge']
5958 d8dcf3c9 Guido Trotter
        else:
5959 d8dcf3c9 Guido Trotter
          bridge = nic.bridge
5960 d8dcf3c9 Guido Trotter
        if 'mac' in this_nic_override:
5961 d8dcf3c9 Guido Trotter
          mac = this_nic_override['mac']
5962 d8dcf3c9 Guido Trotter
        else:
5963 d8dcf3c9 Guido Trotter
          mac = nic.mac
5964 d8dcf3c9 Guido Trotter
        args['nics'].append((ip, bridge, mac))
5965 d8dcf3c9 Guido Trotter
      if constants.DDM_ADD in nic_override:
5966 d8dcf3c9 Guido Trotter
        ip = nic_override[constants.DDM_ADD].get('ip', None)
5967 d8dcf3c9 Guido Trotter
        bridge = nic_override[constants.DDM_ADD]['bridge']
5968 d8dcf3c9 Guido Trotter
        mac = nic_override[constants.DDM_ADD]['mac']
5969 d8dcf3c9 Guido Trotter
        args['nics'].append((ip, bridge, mac))
5970 d8dcf3c9 Guido Trotter
      elif constants.DDM_REMOVE in nic_override:
5971 d8dcf3c9 Guido Trotter
        del args['nics'][-1]
5972 d8dcf3c9 Guido Trotter
5973 338e51e8 Iustin Pop
    env = _BuildInstanceHookEnvByObject(self, self.instance, override=args)
5974 6b12959c Iustin Pop
    nl = [self.cfg.GetMasterNode()] + list(self.instance.all_nodes)
5975 a8083063 Iustin Pop
    return env, nl, nl
5976 a8083063 Iustin Pop
5977 a8083063 Iustin Pop
  def CheckPrereq(self):
5978 a8083063 Iustin Pop
    """Check prerequisites.
5979 a8083063 Iustin Pop

5980 a8083063 Iustin Pop
    This only checks the instance list against the existing names.
5981 a8083063 Iustin Pop

5982 a8083063 Iustin Pop
    """
5983 24991749 Iustin Pop
    force = self.force = self.op.force
5984 a8083063 Iustin Pop
5985 74409b12 Iustin Pop
    # checking the new params on the primary/secondary nodes
5986 31a853d2 Iustin Pop
5987 cfefe007 Guido Trotter
    instance = self.instance = self.cfg.GetInstanceInfo(self.op.instance_name)
5988 2ee88aeb Guido Trotter
    cluster = self.cluster = self.cfg.GetClusterInfo()
5989 1a5c7281 Guido Trotter
    assert self.instance is not None, \
5990 1a5c7281 Guido Trotter
      "Cannot retrieve locked instance %s" % self.op.instance_name
5991 6b12959c Iustin Pop
    pnode = instance.primary_node
5992 6b12959c Iustin Pop
    nodelist = list(instance.all_nodes)
5993 74409b12 Iustin Pop
5994 338e51e8 Iustin Pop
    # hvparams processing
5995 74409b12 Iustin Pop
    if self.op.hvparams:
5996 74409b12 Iustin Pop
      i_hvdict = copy.deepcopy(instance.hvparams)
5997 74409b12 Iustin Pop
      for key, val in self.op.hvparams.iteritems():
5998 8edcd611 Guido Trotter
        if val == constants.VALUE_DEFAULT:
5999 74409b12 Iustin Pop
          try:
6000 74409b12 Iustin Pop
            del i_hvdict[key]
6001 74409b12 Iustin Pop
          except KeyError:
6002 74409b12 Iustin Pop
            pass
6003 74409b12 Iustin Pop
        else:
6004 74409b12 Iustin Pop
          i_hvdict[key] = val
6005 a5728081 Guido Trotter
      utils.ForceDictType(i_hvdict, constants.HVS_PARAMETER_TYPES)
6006 abe609b2 Guido Trotter
      hv_new = objects.FillDict(cluster.hvparams[instance.hypervisor],
6007 74409b12 Iustin Pop
                                i_hvdict)
6008 74409b12 Iustin Pop
      # local check
6009 74409b12 Iustin Pop
      hypervisor.GetHypervisor(
6010 74409b12 Iustin Pop
        instance.hypervisor).CheckParameterSyntax(hv_new)
6011 74409b12 Iustin Pop
      _CheckHVParams(self, nodelist, instance.hypervisor, hv_new)
6012 338e51e8 Iustin Pop
      self.hv_new = hv_new # the new actual values
6013 338e51e8 Iustin Pop
      self.hv_inst = i_hvdict # the new dict (without defaults)
6014 338e51e8 Iustin Pop
    else:
6015 338e51e8 Iustin Pop
      self.hv_new = self.hv_inst = {}
6016 338e51e8 Iustin Pop
6017 338e51e8 Iustin Pop
    # beparams processing
6018 338e51e8 Iustin Pop
    if self.op.beparams:
6019 338e51e8 Iustin Pop
      i_bedict = copy.deepcopy(instance.beparams)
6020 338e51e8 Iustin Pop
      for key, val in self.op.beparams.iteritems():
6021 8edcd611 Guido Trotter
        if val == constants.VALUE_DEFAULT:
6022 338e51e8 Iustin Pop
          try:
6023 338e51e8 Iustin Pop
            del i_bedict[key]
6024 338e51e8 Iustin Pop
          except KeyError:
6025 338e51e8 Iustin Pop
            pass
6026 338e51e8 Iustin Pop
        else:
6027 338e51e8 Iustin Pop
          i_bedict[key] = val
6028 a5728081 Guido Trotter
      utils.ForceDictType(i_bedict, constants.BES_PARAMETER_TYPES)
6029 4ef7f423 Guido Trotter
      be_new = objects.FillDict(cluster.beparams[constants.PP_DEFAULT],
6030 338e51e8 Iustin Pop
                                i_bedict)
6031 338e51e8 Iustin Pop
      self.be_new = be_new # the new actual values
6032 338e51e8 Iustin Pop
      self.be_inst = i_bedict # the new dict (without defaults)
6033 338e51e8 Iustin Pop
    else:
6034 b637ae4d Iustin Pop
      self.be_new = self.be_inst = {}
6035 74409b12 Iustin Pop
6036 cfefe007 Guido Trotter
    self.warn = []
6037 647a5d80 Iustin Pop
6038 338e51e8 Iustin Pop
    if constants.BE_MEMORY in self.op.beparams and not self.force:
6039 647a5d80 Iustin Pop
      mem_check_list = [pnode]
6040 c0f2b229 Iustin Pop
      if be_new[constants.BE_AUTO_BALANCE]:
6041 c0f2b229 Iustin Pop
        # either we changed auto_balance to yes or it was from before
6042 647a5d80 Iustin Pop
        mem_check_list.extend(instance.secondary_nodes)
6043 72737a7f Iustin Pop
      instance_info = self.rpc.call_instance_info(pnode, instance.name,
6044 72737a7f Iustin Pop
                                                  instance.hypervisor)
6045 647a5d80 Iustin Pop
      nodeinfo = self.rpc.call_node_info(mem_check_list, self.cfg.GetVGName(),
6046 72737a7f Iustin Pop
                                         instance.hypervisor)
6047 781de953 Iustin Pop
      if nodeinfo[pnode].failed or not isinstance(nodeinfo[pnode].data, dict):
6048 cfefe007 Guido Trotter
        # Assume the primary node is unreachable and go ahead
6049 cfefe007 Guido Trotter
        self.warn.append("Can't get info from primary node %s" % pnode)
6050 cfefe007 Guido Trotter
      else:
6051 781de953 Iustin Pop
        if not instance_info.failed and instance_info.data:
6052 ade0e8cd Guido Trotter
          current_mem = int(instance_info.data['memory'])
6053 cfefe007 Guido Trotter
        else:
6054 cfefe007 Guido Trotter
          # Assume instance not running
6055 cfefe007 Guido Trotter
          # (there is a slight race condition here, but it's not very probable,
6056 cfefe007 Guido Trotter
          # and we have no other way to check)
6057 cfefe007 Guido Trotter
          current_mem = 0
6058 338e51e8 Iustin Pop
        miss_mem = (be_new[constants.BE_MEMORY] - current_mem -
6059 781de953 Iustin Pop
                    nodeinfo[pnode].data['memory_free'])
6060 cfefe007 Guido Trotter
        if miss_mem > 0:
6061 cfefe007 Guido Trotter
          raise errors.OpPrereqError("This change will prevent the instance"
6062 cfefe007 Guido Trotter
                                     " from starting, due to %d MB of memory"
6063 cfefe007 Guido Trotter
                                     " missing on its primary node" % miss_mem)
6064 cfefe007 Guido Trotter
6065 c0f2b229 Iustin Pop
      if be_new[constants.BE_AUTO_BALANCE]:
6066 ea33068f Iustin Pop
        for node, nres in nodeinfo.iteritems():
6067 ea33068f Iustin Pop
          if node not in instance.secondary_nodes:
6068 ea33068f Iustin Pop
            continue
6069 781de953 Iustin Pop
          if nres.failed or not isinstance(nres.data, dict):
6070 647a5d80 Iustin Pop
            self.warn.append("Can't get info from secondary node %s" % node)
6071 781de953 Iustin Pop
          elif be_new[constants.BE_MEMORY] > nres.data['memory_free']:
6072 647a5d80 Iustin Pop
            self.warn.append("Not enough memory to failover instance to"
6073 647a5d80 Iustin Pop
                             " secondary node %s" % node)
6074 5bc84f33 Alexander Schreiber
6075 24991749 Iustin Pop
    # NIC processing
6076 24991749 Iustin Pop
    for nic_op, nic_dict in self.op.nics:
6077 24991749 Iustin Pop
      if nic_op == constants.DDM_REMOVE:
6078 24991749 Iustin Pop
        if not instance.nics:
6079 24991749 Iustin Pop
          raise errors.OpPrereqError("Instance has no NICs, cannot remove")
6080 24991749 Iustin Pop
        continue
6081 24991749 Iustin Pop
      if nic_op != constants.DDM_ADD:
6082 24991749 Iustin Pop
        # an existing nic
6083 24991749 Iustin Pop
        if nic_op < 0 or nic_op >= len(instance.nics):
6084 24991749 Iustin Pop
          raise errors.OpPrereqError("Invalid NIC index %s, valid values"
6085 24991749 Iustin Pop
                                     " are 0 to %d" %
6086 24991749 Iustin Pop
                                     (nic_op, len(instance.nics)))
6087 5c44da6a Guido Trotter
      if 'bridge' in nic_dict:
6088 5c44da6a Guido Trotter
        nic_bridge = nic_dict['bridge']
6089 5c44da6a Guido Trotter
        if nic_bridge is None:
6090 5c44da6a Guido Trotter
          raise errors.OpPrereqError('Cannot set the nic bridge to None')
6091 24991749 Iustin Pop
        if not self.rpc.call_bridges_exist(pnode, [nic_bridge]):
6092 24991749 Iustin Pop
          msg = ("Bridge '%s' doesn't exist on one of"
6093 24991749 Iustin Pop
                 " the instance nodes" % nic_bridge)
6094 24991749 Iustin Pop
          if self.force:
6095 24991749 Iustin Pop
            self.warn.append(msg)
6096 24991749 Iustin Pop
          else:
6097 24991749 Iustin Pop
            raise errors.OpPrereqError(msg)
6098 5c44da6a Guido Trotter
      if 'mac' in nic_dict:
6099 5c44da6a Guido Trotter
        nic_mac = nic_dict['mac']
6100 5c44da6a Guido Trotter
        if nic_mac is None:
6101 5c44da6a Guido Trotter
          raise errors.OpPrereqError('Cannot set the nic mac to None')
6102 5c44da6a Guido Trotter
        elif nic_mac in (constants.VALUE_AUTO, constants.VALUE_GENERATE):
6103 5c44da6a Guido Trotter
          # otherwise generate the mac
6104 5c44da6a Guido Trotter
          nic_dict['mac'] = self.cfg.GenerateMAC()
6105 5c44da6a Guido Trotter
        else:
6106 5c44da6a Guido Trotter
          # or validate/reserve the current one
6107 5c44da6a Guido Trotter
          if self.cfg.IsMacInUse(nic_mac):
6108 5c44da6a Guido Trotter
            raise errors.OpPrereqError("MAC address %s already in use"
6109 5c44da6a Guido Trotter
                                       " in cluster" % nic_mac)
6110 24991749 Iustin Pop
6111 24991749 Iustin Pop
    # DISK processing
6112 24991749 Iustin Pop
    if self.op.disks and instance.disk_template == constants.DT_DISKLESS:
6113 24991749 Iustin Pop
      raise errors.OpPrereqError("Disk operations not supported for"
6114 24991749 Iustin Pop
                                 " diskless instances")
6115 24991749 Iustin Pop
    for disk_op, disk_dict in self.op.disks:
6116 24991749 Iustin Pop
      if disk_op == constants.DDM_REMOVE:
6117 24991749 Iustin Pop
        if len(instance.disks) == 1:
6118 24991749 Iustin Pop
          raise errors.OpPrereqError("Cannot remove the last disk of"
6119 24991749 Iustin Pop
                                     " an instance")
6120 24991749 Iustin Pop
        ins_l = self.rpc.call_instance_list([pnode], [instance.hypervisor])
6121 24991749 Iustin Pop
        ins_l = ins_l[pnode]
6122 4cfb9426 Iustin Pop
        if ins_l.failed or not isinstance(ins_l.data, list):
6123 24991749 Iustin Pop
          raise errors.OpPrereqError("Can't contact node '%s'" % pnode)
6124 4cfb9426 Iustin Pop
        if instance.name in ins_l.data:
6125 24991749 Iustin Pop
          raise errors.OpPrereqError("Instance is running, can't remove"
6126 24991749 Iustin Pop
                                     " disks.")
6127 24991749 Iustin Pop
6128 24991749 Iustin Pop
      if (disk_op == constants.DDM_ADD and
6129 24991749 Iustin Pop
          len(instance.nics) >= constants.MAX_DISKS):
6130 24991749 Iustin Pop
        raise errors.OpPrereqError("Instance has too many disks (%d), cannot"
6131 24991749 Iustin Pop
                                   " add more" % constants.MAX_DISKS)
6132 24991749 Iustin Pop
      if disk_op not in (constants.DDM_ADD, constants.DDM_REMOVE):
6133 24991749 Iustin Pop
        # an existing disk
6134 24991749 Iustin Pop
        if disk_op < 0 or disk_op >= len(instance.disks):
6135 24991749 Iustin Pop
          raise errors.OpPrereqError("Invalid disk index %s, valid values"
6136 24991749 Iustin Pop
                                     " are 0 to %d" %
6137 24991749 Iustin Pop
                                     (disk_op, len(instance.disks)))
6138 24991749 Iustin Pop
6139 a8083063 Iustin Pop
    return
6140 a8083063 Iustin Pop
6141 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
6142 a8083063 Iustin Pop
    """Modifies an instance.
6143 a8083063 Iustin Pop

6144 a8083063 Iustin Pop
    All parameters take effect only at the next restart of the instance.
6145 24991749 Iustin Pop

6146 a8083063 Iustin Pop
    """
6147 cfefe007 Guido Trotter
    # Process here the warnings from CheckPrereq, as we don't have a
6148 cfefe007 Guido Trotter
    # feedback_fn there.
6149 cfefe007 Guido Trotter
    for warn in self.warn:
6150 cfefe007 Guido Trotter
      feedback_fn("WARNING: %s" % warn)
6151 cfefe007 Guido Trotter
6152 a8083063 Iustin Pop
    result = []
6153 a8083063 Iustin Pop
    instance = self.instance
6154 24991749 Iustin Pop
    # disk changes
6155 24991749 Iustin Pop
    for disk_op, disk_dict in self.op.disks:
6156 24991749 Iustin Pop
      if disk_op == constants.DDM_REMOVE:
6157 24991749 Iustin Pop
        # remove the last disk
6158 24991749 Iustin Pop
        device = instance.disks.pop()
6159 24991749 Iustin Pop
        device_idx = len(instance.disks)
6160 24991749 Iustin Pop
        for node, disk in device.ComputeNodeTree(instance.primary_node):
6161 24991749 Iustin Pop
          self.cfg.SetDiskID(disk, node)
6162 e1bc0878 Iustin Pop
          msg = self.rpc.call_blockdev_remove(node, disk).RemoteFailMsg()
6163 e1bc0878 Iustin Pop
          if msg:
6164 e1bc0878 Iustin Pop
            self.LogWarning("Could not remove disk/%d on node %s: %s,"
6165 e1bc0878 Iustin Pop
                            " continuing anyway", device_idx, node, msg)
6166 24991749 Iustin Pop
        result.append(("disk/%d" % device_idx, "remove"))
6167 24991749 Iustin Pop
      elif disk_op == constants.DDM_ADD:
6168 24991749 Iustin Pop
        # add a new disk
6169 24991749 Iustin Pop
        if instance.disk_template == constants.DT_FILE:
6170 24991749 Iustin Pop
          file_driver, file_path = instance.disks[0].logical_id
6171 24991749 Iustin Pop
          file_path = os.path.dirname(file_path)
6172 24991749 Iustin Pop
        else:
6173 24991749 Iustin Pop
          file_driver = file_path = None
6174 24991749 Iustin Pop
        disk_idx_base = len(instance.disks)
6175 24991749 Iustin Pop
        new_disk = _GenerateDiskTemplate(self,
6176 24991749 Iustin Pop
                                         instance.disk_template,
6177 32388e6d Iustin Pop
                                         instance.name, instance.primary_node,
6178 24991749 Iustin Pop
                                         instance.secondary_nodes,
6179 24991749 Iustin Pop
                                         [disk_dict],
6180 24991749 Iustin Pop
                                         file_path,
6181 24991749 Iustin Pop
                                         file_driver,
6182 24991749 Iustin Pop
                                         disk_idx_base)[0]
6183 24991749 Iustin Pop
        instance.disks.append(new_disk)
6184 24991749 Iustin Pop
        info = _GetInstanceInfoText(instance)
6185 24991749 Iustin Pop
6186 24991749 Iustin Pop
        logging.info("Creating volume %s for instance %s",
6187 24991749 Iustin Pop
                     new_disk.iv_name, instance.name)
6188 24991749 Iustin Pop
        # Note: this needs to be kept in sync with _CreateDisks
6189 24991749 Iustin Pop
        #HARDCODE
6190 428958aa Iustin Pop
        for node in instance.all_nodes:
6191 428958aa Iustin Pop
          f_create = node == instance.primary_node
6192 796cab27 Iustin Pop
          try:
6193 428958aa Iustin Pop
            _CreateBlockDev(self, node, instance, new_disk,
6194 428958aa Iustin Pop
                            f_create, info, f_create)
6195 1492cca7 Iustin Pop
          except errors.OpExecError, err:
6196 24991749 Iustin Pop
            self.LogWarning("Failed to create volume %s (%s) on"
6197 428958aa Iustin Pop
                            " node %s: %s",
6198 428958aa Iustin Pop
                            new_disk.iv_name, new_disk, node, err)
6199 24991749 Iustin Pop
        result.append(("disk/%d" % disk_idx_base, "add:size=%s,mode=%s" %
6200 24991749 Iustin Pop
                       (new_disk.size, new_disk.mode)))
6201 24991749 Iustin Pop
      else:
6202 24991749 Iustin Pop
        # change a given disk
6203 24991749 Iustin Pop
        instance.disks[disk_op].mode = disk_dict['mode']
6204 24991749 Iustin Pop
        result.append(("disk.mode/%d" % disk_op, disk_dict['mode']))
6205 24991749 Iustin Pop
    # NIC changes
6206 24991749 Iustin Pop
    for nic_op, nic_dict in self.op.nics:
6207 24991749 Iustin Pop
      if nic_op == constants.DDM_REMOVE:
6208 24991749 Iustin Pop
        # remove the last nic
6209 24991749 Iustin Pop
        del instance.nics[-1]
6210 24991749 Iustin Pop
        result.append(("nic.%d" % len(instance.nics), "remove"))
6211 24991749 Iustin Pop
      elif nic_op == constants.DDM_ADD:
6212 5c44da6a Guido Trotter
        # mac and bridge should be set, by now
6213 5c44da6a Guido Trotter
        mac = nic_dict['mac']
6214 5c44da6a Guido Trotter
        bridge = nic_dict['bridge']
6215 24991749 Iustin Pop
        new_nic = objects.NIC(mac=mac, ip=nic_dict.get('ip', None),
6216 5c44da6a Guido Trotter
                              bridge=bridge)
6217 24991749 Iustin Pop
        instance.nics.append(new_nic)
6218 24991749 Iustin Pop
        result.append(("nic.%d" % (len(instance.nics) - 1),
6219 24991749 Iustin Pop
                       "add:mac=%s,ip=%s,bridge=%s" %
6220 24991749 Iustin Pop
                       (new_nic.mac, new_nic.ip, new_nic.bridge)))
6221 24991749 Iustin Pop
      else:
6222 24991749 Iustin Pop
        # change a given nic
6223 24991749 Iustin Pop
        for key in 'mac', 'ip', 'bridge':
6224 24991749 Iustin Pop
          if key in nic_dict:
6225 24991749 Iustin Pop
            setattr(instance.nics[nic_op], key, nic_dict[key])
6226 24991749 Iustin Pop
            result.append(("nic.%s/%d" % (key, nic_op), nic_dict[key]))
6227 24991749 Iustin Pop
6228 24991749 Iustin Pop
    # hvparams changes
6229 74409b12 Iustin Pop
    if self.op.hvparams:
6230 12649e35 Guido Trotter
      instance.hvparams = self.hv_inst
6231 74409b12 Iustin Pop
      for key, val in self.op.hvparams.iteritems():
6232 74409b12 Iustin Pop
        result.append(("hv/%s" % key, val))
6233 24991749 Iustin Pop
6234 24991749 Iustin Pop
    # beparams changes
6235 338e51e8 Iustin Pop
    if self.op.beparams:
6236 338e51e8 Iustin Pop
      instance.beparams = self.be_inst
6237 338e51e8 Iustin Pop
      for key, val in self.op.beparams.iteritems():
6238 338e51e8 Iustin Pop
        result.append(("be/%s" % key, val))
6239 a8083063 Iustin Pop
6240 ea94e1cd Guido Trotter
    self.cfg.Update(instance)
6241 a8083063 Iustin Pop
6242 a8083063 Iustin Pop
    return result
6243 a8083063 Iustin Pop
6244 a8083063 Iustin Pop
6245 a8083063 Iustin Pop
class LUQueryExports(NoHooksLU):
6246 a8083063 Iustin Pop
  """Query the exports list
6247 a8083063 Iustin Pop

6248 a8083063 Iustin Pop
  """
6249 895ecd9c Guido Trotter
  _OP_REQP = ['nodes']
6250 21a15682 Guido Trotter
  REQ_BGL = False
6251 21a15682 Guido Trotter
6252 21a15682 Guido Trotter
  def ExpandNames(self):
6253 21a15682 Guido Trotter
    self.needed_locks = {}
6254 21a15682 Guido Trotter
    self.share_locks[locking.LEVEL_NODE] = 1
6255 21a15682 Guido Trotter
    if not self.op.nodes:
6256 e310b019 Guido Trotter
      self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
6257 21a15682 Guido Trotter
    else:
6258 21a15682 Guido Trotter
      self.needed_locks[locking.LEVEL_NODE] = \
6259 21a15682 Guido Trotter
        _GetWantedNodes(self, self.op.nodes)
6260 a8083063 Iustin Pop
6261 a8083063 Iustin Pop
  def CheckPrereq(self):
6262 21a15682 Guido Trotter
    """Check prerequisites.
6263 a8083063 Iustin Pop

6264 a8083063 Iustin Pop
    """
6265 21a15682 Guido Trotter
    self.nodes = self.acquired_locks[locking.LEVEL_NODE]
6266 a8083063 Iustin Pop
6267 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
6268 a8083063 Iustin Pop
    """Compute the list of all the exported system images.
6269 a8083063 Iustin Pop

6270 e4376078 Iustin Pop
    @rtype: dict
6271 e4376078 Iustin Pop
    @return: a dictionary with the structure node->(export-list)
6272 e4376078 Iustin Pop
        where export-list is a list of the instances exported on
6273 e4376078 Iustin Pop
        that node.
6274 a8083063 Iustin Pop

6275 a8083063 Iustin Pop
    """
6276 b04285f2 Guido Trotter
    rpcresult = self.rpc.call_export_list(self.nodes)
6277 b04285f2 Guido Trotter
    result = {}
6278 b04285f2 Guido Trotter
    for node in rpcresult:
6279 b04285f2 Guido Trotter
      if rpcresult[node].failed:
6280 b04285f2 Guido Trotter
        result[node] = False
6281 b04285f2 Guido Trotter
      else:
6282 b04285f2 Guido Trotter
        result[node] = rpcresult[node].data
6283 b04285f2 Guido Trotter
6284 b04285f2 Guido Trotter
    return result
6285 a8083063 Iustin Pop
6286 a8083063 Iustin Pop
6287 a8083063 Iustin Pop
class LUExportInstance(LogicalUnit):
6288 a8083063 Iustin Pop
  """Export an instance to an image in the cluster.
6289 a8083063 Iustin Pop

6290 a8083063 Iustin Pop
  """
6291 a8083063 Iustin Pop
  HPATH = "instance-export"
6292 a8083063 Iustin Pop
  HTYPE = constants.HTYPE_INSTANCE
6293 a8083063 Iustin Pop
  _OP_REQP = ["instance_name", "target_node", "shutdown"]
6294 6657590e Guido Trotter
  REQ_BGL = False
6295 6657590e Guido Trotter
6296 6657590e Guido Trotter
  def ExpandNames(self):
6297 6657590e Guido Trotter
    self._ExpandAndLockInstance()
6298 6657590e Guido Trotter
    # FIXME: lock only instance primary and destination node
6299 6657590e Guido Trotter
    #
6300 6657590e Guido Trotter
    # Sad but true, for now we have do lock all nodes, as we don't know where
6301 6657590e Guido Trotter
    # the previous export might be, and and in this LU we search for it and
6302 6657590e Guido Trotter
    # remove it from its current node. In the future we could fix this by:
6303 6657590e Guido Trotter
    #  - making a tasklet to search (share-lock all), then create the new one,
6304 6657590e Guido Trotter
    #    then one to remove, after
6305 6657590e Guido Trotter
    #  - removing the removal operation altoghether
6306 6657590e Guido Trotter
    self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
6307 6657590e Guido Trotter
6308 6657590e Guido Trotter
  def DeclareLocks(self, level):
6309 6657590e Guido Trotter
    """Last minute lock declaration."""
6310 6657590e Guido Trotter
    # All nodes are locked anyway, so nothing to do here.
6311 a8083063 Iustin Pop
6312 a8083063 Iustin Pop
  def BuildHooksEnv(self):
6313 a8083063 Iustin Pop
    """Build hooks env.
6314 a8083063 Iustin Pop

6315 a8083063 Iustin Pop
    This will run on the master, primary node and target node.
6316 a8083063 Iustin Pop

6317 a8083063 Iustin Pop
    """
6318 a8083063 Iustin Pop
    env = {
6319 a8083063 Iustin Pop
      "EXPORT_NODE": self.op.target_node,
6320 a8083063 Iustin Pop
      "EXPORT_DO_SHUTDOWN": self.op.shutdown,
6321 a8083063 Iustin Pop
      }
6322 338e51e8 Iustin Pop
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
6323 d6a02168 Michael Hanselmann
    nl = [self.cfg.GetMasterNode(), self.instance.primary_node,
6324 a8083063 Iustin Pop
          self.op.target_node]
6325 a8083063 Iustin Pop
    return env, nl, nl
6326 a8083063 Iustin Pop
6327 a8083063 Iustin Pop
  def CheckPrereq(self):
6328 a8083063 Iustin Pop
    """Check prerequisites.
6329 a8083063 Iustin Pop

6330 9ac99fda Guido Trotter
    This checks that the instance and node names are valid.
6331 a8083063 Iustin Pop

6332 a8083063 Iustin Pop
    """
6333 6657590e Guido Trotter
    instance_name = self.op.instance_name
6334 a8083063 Iustin Pop
    self.instance = self.cfg.GetInstanceInfo(instance_name)
6335 6657590e Guido Trotter
    assert self.instance is not None, \
6336 6657590e Guido Trotter
          "Cannot retrieve locked instance %s" % self.op.instance_name
6337 43017d26 Iustin Pop
    _CheckNodeOnline(self, self.instance.primary_node)
6338 a8083063 Iustin Pop
6339 6657590e Guido Trotter
    self.dst_node = self.cfg.GetNodeInfo(
6340 6657590e Guido Trotter
      self.cfg.ExpandNodeName(self.op.target_node))
6341 a8083063 Iustin Pop
6342 268b8e42 Iustin Pop
    if self.dst_node is None:
6343 268b8e42 Iustin Pop
      # This is wrong node name, not a non-locked node
6344 268b8e42 Iustin Pop
      raise errors.OpPrereqError("Wrong node name %s" % self.op.target_node)
6345 aeb83a2b Iustin Pop
    _CheckNodeOnline(self, self.dst_node.name)
6346 733a2b6a Iustin Pop
    _CheckNodeNotDrained(self, self.dst_node.name)
6347 a8083063 Iustin Pop
6348 b6023d6c Manuel Franceschini
    # instance disk type verification
6349 b6023d6c Manuel Franceschini
    for disk in self.instance.disks:
6350 b6023d6c Manuel Franceschini
      if disk.dev_type == constants.LD_FILE:
6351 b6023d6c Manuel Franceschini
        raise errors.OpPrereqError("Export not supported for instances with"
6352 b6023d6c Manuel Franceschini
                                   " file-based disks")
6353 b6023d6c Manuel Franceschini
6354 a8083063 Iustin Pop
  def Exec(self, feedback_fn):
6355 a8083063 Iustin Pop
    """Export an instance to an image in the cluster.
6356 a8083063 Iustin Pop

6357 a8083063 Iustin Pop
    """
6358 a8083063 Iustin Pop
    instance = self.instance
6359 a8083063 Iustin Pop
    dst_node = self.dst_node
6360 a8083063 Iustin Pop
    src_node = instance.primary_node
6361 a8083063 Iustin Pop
    if self.op.shutdown:
6362 fb300fb7 Guido Trotter
      # shutdown the instance, but not the disks
6363 781de953 Iustin Pop
      result = self.rpc.call_instance_shutdown(src_node, instance)
6364 1fae010f Iustin Pop
      msg = result.RemoteFailMsg()
6365 1fae010f Iustin Pop
      if msg:
6366 1fae010f Iustin Pop
        raise errors.OpExecError("Could not shutdown instance %s on"
6367 1fae010f Iustin Pop
                                 " node %s: %s" %
6368 1fae010f Iustin Pop
                                 (instance.name, src_node, msg))
6369 a8083063 Iustin Pop
6370 a8083063 Iustin Pop
    vgname = self.cfg.GetVGName()
6371 a8083063 Iustin Pop
6372 a8083063 Iustin Pop
    snap_disks = []
6373 a8083063 Iustin Pop
6374 998c712c Iustin Pop
    # set the disks ID correctly since call_instance_start needs the
6375 998c712c Iustin Pop
    # correct drbd minor to create the symlinks
6376 998c712c Iustin Pop
    for disk in instance.disks:
6377 998c712c Iustin Pop
      self.cfg.SetDiskID(disk, src_node)
6378 998c712c Iustin Pop
6379 a8083063 Iustin Pop
    try:
6380 a8083063 Iustin Pop
      for disk in instance.disks:
6381 19d7f90a Guido Trotter
        # new_dev_name will be a snapshot of an lvm leaf of the one we passed
6382 19d7f90a Guido Trotter
        new_dev_name = self.rpc.call_blockdev_snapshot(src_node, disk)
6383 781de953 Iustin Pop
        if new_dev_name.failed or not new_dev_name.data:
6384 19d7f90a Guido Trotter
          self.LogWarning("Could not snapshot block device %s on node %s",
6385 9a4f63d1 Iustin Pop
                          disk.logical_id[1], src_node)
6386 19d7f90a Guido Trotter
          snap_disks.append(False)
6387 19d7f90a Guido Trotter
        else:
6388 19d7f90a Guido Trotter
          new_dev = objects.Disk(dev_type=constants.LD_LV, size=disk.size,
6389 781de953 Iustin Pop
                                 logical_id=(vgname, new_dev_name.data),
6390 781de953 Iustin Pop
                                 physical_id=(vgname, new_dev_name.data),
6391 19d7f90a Guido Trotter
                                 iv_name=disk.iv_name)
6392 19d7f90a Guido Trotter
          snap_disks.append(new_dev)
6393 a8083063 Iustin Pop
6394 a8083063 Iustin Pop
    finally:
6395 0d68c45d Iustin Pop
      if self.op.shutdown and instance.admin_up:
6396 0eca8e0c Iustin Pop
        result = self.rpc.call_instance_start(src_node, instance, None, None)
6397 dd279568 Iustin Pop
        msg = result.RemoteFailMsg()
6398 dd279568 Iustin Pop
        if msg:
6399 b9bddb6b Iustin Pop
          _ShutdownInstanceDisks(self, instance)
6400 dd279568 Iustin Pop
          raise errors.OpExecError("Could not start instance: %s" % msg)
6401 a8083063 Iustin Pop
6402 a8083063 Iustin Pop
    # TODO: check for size
6403 a8083063 Iustin Pop
6404 62c9ec92 Iustin Pop
    cluster_name = self.cfg.GetClusterName()
6405 74c47259 Iustin Pop
    for idx, dev in enumerate(snap_disks):
6406 19d7f90a Guido Trotter
      if dev:
6407 781de953 Iustin Pop
        result = self.rpc.call_snapshot_export(src_node, dev, dst_node.name,
6408 781de953 Iustin Pop
                                               instance, cluster_name, idx)
6409 781de953 Iustin Pop
        if result.failed or not result.data:
6410 19d7f90a Guido Trotter
          self.LogWarning("Could not export block device %s from node %s to"
6411 19d7f90a Guido Trotter
                          " node %s", dev.logical_id[1], src_node,
6412 19d7f90a Guido Trotter
                          dst_node.name)
6413 e1bc0878 Iustin Pop
        msg = self.rpc.call_blockdev_remove(src_node, dev).RemoteFailMsg()
6414 e1bc0878 Iustin Pop
        if msg:
6415 19d7f90a Guido Trotter
          self.LogWarning("Could not remove snapshot block device %s from node"
6416 e1bc0878 Iustin Pop
                          " %s: %s", dev.logical_id[1], src_node, msg)
6417 a8083063 Iustin Pop
6418 781de953 Iustin Pop
    result = self.rpc.call_finalize_export(dst_node.name, instance, snap_disks)
6419 781de953 Iustin Pop
    if result.failed or not result.data:
6420 19d7f90a Guido Trotter
      self.LogWarning("Could not finalize export for instance %s on node %s",
6421 19d7f90a Guido Trotter
                      instance.name, dst_node.name)
6422 a8083063 Iustin Pop
6423 a8083063 Iustin Pop
    nodelist = self.cfg.GetNodeList()
6424 a8083063 Iustin Pop
    nodelist.remove(dst_node.name)
6425 a8083063 Iustin Pop
6426 a8083063 Iustin Pop
    # on one-node clusters nodelist will be empty after the removal
6427 a8083063 Iustin Pop
    # if we proceed the backup would be removed because OpQueryExports
6428 a8083063 Iustin Pop
    # substitutes an empty list with the full cluster node list.
6429 a8083063 Iustin Pop
    if nodelist:
6430 72737a7f Iustin Pop
      exportlist = self.rpc.call_export_list(nodelist)
6431 a8083063 Iustin Pop
      for node in exportlist:
6432 781de953 Iustin Pop
        if exportlist[node].failed:
6433 781de953 Iustin Pop
          continue
6434 781de953 Iustin Pop
        if instance.name in exportlist[node].data:
6435 72737a7f Iustin Pop
          if not self.rpc.call_export_remove(node, instance.name):
6436 19d7f90a Guido Trotter
            self.LogWarning("Could not remove older export for instance %s"
6437 19d7f90a Guido Trotter
                            " on node %s", instance.name, node)
6438 5c947f38 Iustin Pop
6439 5c947f38 Iustin Pop
6440 9ac99fda Guido Trotter
class LURemoveExport(NoHooksLU):
6441 9ac99fda Guido Trotter
  """Remove exports related to the named instance.
6442 9ac99fda Guido Trotter

6443 9ac99fda Guido Trotter
  """
6444 9ac99fda Guido Trotter
  _OP_REQP = ["instance_name"]
6445 3656b3af Guido Trotter
  REQ_BGL = False
6446 3656b3af Guido Trotter
6447 3656b3af Guido Trotter
  def ExpandNames(self):
6448 3656b3af Guido Trotter
    self.needed_locks = {}
6449 3656b3af Guido Trotter
    # We need all nodes to be locked in order for RemoveExport to work, but we
6450 3656b3af Guido Trotter
    # don't need to lock the instance itself, as nothing will happen to it (and
6451 3656b3af Guido Trotter
    # we can remove exports also for a removed instance)
6452 3656b3af Guido Trotter
    self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
6453 9ac99fda Guido Trotter
6454 9ac99fda Guido Trotter
  def CheckPrereq(self):
6455 9ac99fda Guido Trotter
    """Check prerequisites.
6456 9ac99fda Guido Trotter
    """
6457 9ac99fda Guido Trotter
    pass
6458 9ac99fda Guido Trotter
6459 9ac99fda Guido Trotter
  def Exec(self, feedback_fn):
6460 9ac99fda Guido Trotter
    """Remove any export.
6461 9ac99fda Guido Trotter

6462 9ac99fda Guido Trotter
    """
6463 9ac99fda Guido Trotter
    instance_name = self.cfg.ExpandInstanceName(self.op.instance_name)
6464 9ac99fda Guido Trotter
    # If the instance was not found we'll try with the name that was passed in.
6465 9ac99fda Guido Trotter
    # This will only work if it was an FQDN, though.
6466 9ac99fda Guido Trotter
    fqdn_warn = False
6467 9ac99fda Guido Trotter
    if not instance_name:
6468 9ac99fda Guido Trotter
      fqdn_warn = True
6469 9ac99fda Guido Trotter
      instance_name = self.op.instance_name
6470 9ac99fda Guido Trotter
6471 72737a7f Iustin Pop
    exportlist = self.rpc.call_export_list(self.acquired_locks[
6472 72737a7f Iustin Pop
      locking.LEVEL_NODE])
6473 9ac99fda Guido Trotter
    found = False
6474 9ac99fda Guido Trotter
    for node in exportlist:
6475 781de953 Iustin Pop
      if exportlist[node].failed:
6476 25361b9a Iustin Pop
        self.LogWarning("Failed to query node %s, continuing" % node)
6477 781de953 Iustin Pop
        continue
6478 781de953 Iustin Pop
      if instance_name in exportlist[node].data:
6479 9ac99fda Guido Trotter
        found = True
6480 781de953 Iustin Pop
        result = self.rpc.call_export_remove(node, instance_name)
6481 781de953 Iustin Pop
        if result.failed or not result.data:
6482 9a4f63d1 Iustin Pop
          logging.error("Could not remove export for instance %s"
6483 9a4f63d1 Iustin Pop
                        " on node %s", instance_name, node)
6484 9ac99fda Guido Trotter
6485 9ac99fda Guido Trotter
    if fqdn_warn and not found:
6486 9ac99fda Guido Trotter
      feedback_fn("Export not found. If trying to remove an export belonging"
6487 9ac99fda Guido Trotter
                  " to a deleted instance please use its Fully Qualified"
6488 9ac99fda Guido Trotter
                  " Domain Name.")
6489 9ac99fda Guido Trotter
6490 9ac99fda Guido Trotter
6491 5c947f38 Iustin Pop
class TagsLU(NoHooksLU):
6492 5c947f38 Iustin Pop
  """Generic tags LU.
6493 5c947f38 Iustin Pop

6494 5c947f38 Iustin Pop
  This is an abstract class which is the parent of all the other tags LUs.
6495 5c947f38 Iustin Pop

6496 5c947f38 Iustin Pop
  """
6497 5c947f38 Iustin Pop
6498 8646adce Guido Trotter
  def ExpandNames(self):
6499 8646adce Guido Trotter
    self.needed_locks = {}
6500 8646adce Guido Trotter
    if self.op.kind == constants.TAG_NODE:
6501 5c947f38 Iustin Pop
      name = self.cfg.ExpandNodeName(self.op.name)
6502 5c947f38 Iustin Pop
      if name is None:
6503 3ecf6786 Iustin Pop
        raise errors.OpPrereqError("Invalid node name (%s)" %
6504 3ecf6786 Iustin Pop
                                   (self.op.name,))
6505 5c947f38 Iustin Pop
      self.op.name = name
6506 8646adce Guido Trotter
      self.needed_locks[locking.LEVEL_NODE] = name
6507 5c947f38 Iustin Pop
    elif self.op.kind == constants.TAG_INSTANCE:
6508 8f684e16 Iustin Pop
      name = self.cfg.ExpandInstanceName(self.op.name)
6509 5c947f38 Iustin Pop
      if name is None:
6510 3ecf6786 Iustin Pop
        raise errors.OpPrereqError("Invalid instance name (%s)" %
6511 3ecf6786 Iustin Pop
                                   (self.op.name,))
6512 5c947f38 Iustin Pop
      self.op.name = name
6513 8646adce Guido Trotter
      self.needed_locks[locking.LEVEL_INSTANCE] = name
6514 8646adce Guido Trotter
6515 8646adce Guido Trotter
  def CheckPrereq(self):
6516 8646adce Guido Trotter
    """Check prerequisites.
6517 8646adce Guido Trotter

6518 8646adce Guido Trotter
    """
6519 8646adce Guido Trotter
    if self.op.kind == constants.TAG_CLUSTER:
6520 8646adce Guido Trotter
      self.target = self.cfg.GetClusterInfo()
6521 8646adce Guido Trotter
    elif self.op.kind == constants.TAG_NODE:
6522 8646adce Guido Trotter
      self.target = self.cfg.GetNodeInfo(self.op.name)
6523 8646adce Guido Trotter
    elif self.op.kind == constants.TAG_INSTANCE:
6524 8646adce Guido Trotter
      self.target = self.cfg.GetInstanceInfo(self.op.name)
6525 5c947f38 Iustin Pop
    else:
6526 3ecf6786 Iustin Pop
      raise errors.OpPrereqError("Wrong tag type requested (%s)" %
6527 3ecf6786 Iustin Pop
                                 str(self.op.kind))
6528 5c947f38 Iustin Pop
6529 5c947f38 Iustin Pop
6530 5c947f38 Iustin Pop
class LUGetTags(TagsLU):
6531 5c947f38 Iustin Pop
  """Returns the tags of a given object.
6532 5c947f38 Iustin Pop

6533 5c947f38 Iustin Pop
  """
6534 5c947f38 Iustin Pop
  _OP_REQP = ["kind", "name"]
6535 8646adce Guido Trotter
  REQ_BGL = False
6536 5c947f38 Iustin Pop
6537 5c947f38 Iustin Pop
  def Exec(self, feedback_fn):
6538 5c947f38 Iustin Pop
    """Returns the tag list.
6539 5c947f38 Iustin Pop

6540 5c947f38 Iustin Pop
    """
6541 5d414478 Oleksiy Mishchenko
    return list(self.target.GetTags())
6542 5c947f38 Iustin Pop
6543 5c947f38 Iustin Pop
6544 73415719 Iustin Pop
class LUSearchTags(NoHooksLU):
6545 73415719 Iustin Pop
  """Searches the tags for a given pattern.
6546 73415719 Iustin Pop

6547 73415719 Iustin Pop
  """
6548 73415719 Iustin Pop
  _OP_REQP = ["pattern"]
6549 8646adce Guido Trotter
  REQ_BGL = False
6550 8646adce Guido Trotter
6551 8646adce Guido Trotter
  def ExpandNames(self):
6552 8646adce Guido Trotter
    self.needed_locks = {}
6553 73415719 Iustin Pop
6554 73415719 Iustin Pop
  def CheckPrereq(self):
6555 73415719 Iustin Pop
    """Check prerequisites.
6556 73415719 Iustin Pop

6557 73415719 Iustin Pop
    This checks the pattern passed for validity by compiling it.
6558 73415719 Iustin Pop

6559 73415719 Iustin Pop
    """
6560 73415719 Iustin Pop
    try:
6561 73415719 Iustin Pop
      self.re = re.compile(self.op.pattern)
6562 73415719 Iustin Pop
    except re.error, err:
6563 73415719 Iustin Pop
      raise errors.OpPrereqError("Invalid search pattern '%s': %s" %
6564 73415719 Iustin Pop
                                 (self.op.pattern, err))
6565 73415719 Iustin Pop
6566 73415719 Iustin Pop
  def Exec(self, feedback_fn):
6567 73415719 Iustin Pop
    """Returns the tag list.
6568 73415719 Iustin Pop

6569 73415719 Iustin Pop
    """
6570 73415719 Iustin Pop
    cfg = self.cfg
6571 73415719 Iustin Pop
    tgts = [("/cluster", cfg.GetClusterInfo())]
6572 8646adce Guido Trotter
    ilist = cfg.GetAllInstancesInfo().values()
6573 73415719 Iustin Pop
    tgts.extend([("/instances/%s" % i.name, i) for i in ilist])
6574 8646adce Guido Trotter
    nlist = cfg.GetAllNodesInfo().values()
6575 73415719 Iustin Pop
    tgts.extend([("/nodes/%s" % n.name, n) for n in nlist])
6576 73415719 Iustin Pop
    results = []
6577 73415719 Iustin Pop
    for path, target in tgts:
6578 73415719 Iustin Pop
      for tag in target.GetTags():
6579 73415719 Iustin Pop
        if self.re.search(tag):
6580 73415719 Iustin Pop
          results.append((path, tag))
6581 73415719 Iustin Pop
    return results
6582 73415719 Iustin Pop
6583 73415719 Iustin Pop
6584 f27302fa Iustin Pop
class LUAddTags(TagsLU):
6585 5c947f38 Iustin Pop
  """Sets a tag on a given object.
6586 5c947f38 Iustin Pop

6587 5c947f38 Iustin Pop
  """
6588 f27302fa Iustin Pop
  _OP_REQP = ["kind", "name", "tags"]
6589 8646adce Guido Trotter
  REQ_BGL = False
6590 5c947f38 Iustin Pop
6591 5c947f38 Iustin Pop
  def CheckPrereq(self):
6592 5c947f38 Iustin Pop
    """Check prerequisites.
6593 5c947f38 Iustin Pop

6594 5c947f38 Iustin Pop
    This checks the type and length of the tag name and value.
6595 5c947f38 Iustin Pop

6596 5c947f38 Iustin Pop
    """
6597 5c947f38 Iustin Pop
    TagsLU.CheckPrereq(self)
6598 f27302fa Iustin Pop
    for tag in self.op.tags:
6599 f27302fa Iustin Pop
      objects.TaggableObject.ValidateTag(tag)
6600 5c947f38 Iustin Pop
6601 5c947f38 Iustin Pop
  def Exec(self, feedback_fn):
6602 5c947f38 Iustin Pop
    """Sets the tag.
6603 5c947f38 Iustin Pop

6604 5c947f38 Iustin Pop
    """
6605 5c947f38 Iustin Pop
    try:
6606 f27302fa Iustin Pop
      for tag in self.op.tags:
6607 f27302fa Iustin Pop
        self.target.AddTag(tag)
6608 5c947f38 Iustin Pop
    except errors.TagError, err:
6609 3ecf6786 Iustin Pop
      raise errors.OpExecError("Error while setting tag: %s" % str(err))
6610 5c947f38 Iustin Pop
    try:
6611 5c947f38 Iustin Pop
      self.cfg.Update(self.target)
6612 5c947f38 Iustin Pop
    except errors.ConfigurationError:
6613 3ecf6786 Iustin Pop
      raise errors.OpRetryError("There has been a modification to the"
6614 3ecf6786 Iustin Pop
                                " config file and the operation has been"
6615 3ecf6786 Iustin Pop
                                " aborted. Please retry.")
6616 5c947f38 Iustin Pop
6617 5c947f38 Iustin Pop
6618 f27302fa Iustin Pop
class LUDelTags(TagsLU):
6619 f27302fa Iustin Pop
  """Delete a list of tags from a given object.
6620 5c947f38 Iustin Pop

6621 5c947f38 Iustin Pop
  """
6622 f27302fa Iustin Pop
  _OP_REQP = ["kind", "name", "tags"]
6623 8646adce Guido Trotter
  REQ_BGL = False
6624 5c947f38 Iustin Pop
6625 5c947f38 Iustin Pop
  def CheckPrereq(self):
6626 5c947f38 Iustin Pop
    """Check prerequisites.
6627 5c947f38 Iustin Pop

6628 5c947f38 Iustin Pop
    This checks that we have the given tag.
6629 5c947f38 Iustin Pop

6630 5c947f38 Iustin Pop
    """
6631 5c947f38 Iustin Pop
    TagsLU.CheckPrereq(self)
6632 f27302fa Iustin Pop
    for tag in self.op.tags:
6633 f27302fa Iustin Pop
      objects.TaggableObject.ValidateTag(tag)
6634 f27302fa Iustin Pop
    del_tags = frozenset(self.op.tags)
6635 f27302fa Iustin Pop
    cur_tags = self.target.GetTags()
6636 f27302fa Iustin Pop
    if not del_tags <= cur_tags:
6637 f27302fa Iustin Pop
      diff_tags = del_tags - cur_tags
6638 f27302fa Iustin Pop
      diff_names = ["'%s'" % tag for tag in diff_tags]
6639 f27302fa Iustin Pop
      diff_names.sort()
6640 f27302fa Iustin Pop
      raise errors.OpPrereqError("Tag(s) %s not found" %
6641 f27302fa Iustin Pop
                                 (",".join(diff_names)))
6642 5c947f38 Iustin Pop
6643 5c947f38 Iustin Pop
  def Exec(self, feedback_fn):
6644 5c947f38 Iustin Pop
    """Remove the tag from the object.
6645 5c947f38 Iustin Pop

6646 5c947f38 Iustin Pop
    """
6647 f27302fa Iustin Pop
    for tag in self.op.tags:
6648 f27302fa Iustin Pop
      self.target.RemoveTag(tag)
6649 5c947f38 Iustin Pop
    try:
6650 5c947f38 Iustin Pop
      self.cfg.Update(self.target)
6651 5c947f38 Iustin Pop
    except errors.ConfigurationError:
6652 3ecf6786 Iustin Pop
      raise errors.OpRetryError("There has been a modification to the"
6653 3ecf6786 Iustin Pop
                                " config file and the operation has been"
6654 3ecf6786 Iustin Pop
                                " aborted. Please retry.")
6655 06009e27 Iustin Pop
6656 0eed6e61 Guido Trotter
6657 06009e27 Iustin Pop
class LUTestDelay(NoHooksLU):
6658 06009e27 Iustin Pop
  """Sleep for a specified amount of time.
6659 06009e27 Iustin Pop

6660 0b097284 Guido Trotter
  This LU sleeps on the master and/or nodes for a specified amount of
6661 06009e27 Iustin Pop
  time.
6662 06009e27 Iustin Pop

6663 06009e27 Iustin Pop
  """
6664 06009e27 Iustin Pop
  _OP_REQP = ["duration", "on_master", "on_nodes"]
6665 fbe9022f Guido Trotter
  REQ_BGL = False
6666 06009e27 Iustin Pop
6667 fbe9022f Guido Trotter
  def ExpandNames(self):
6668 fbe9022f Guido Trotter
    """Expand names and set required locks.
6669 06009e27 Iustin Pop

6670 fbe9022f Guido Trotter
    This expands the node list, if any.
6671 06009e27 Iustin Pop

6672 06009e27 Iustin Pop
    """
6673 fbe9022f Guido Trotter
    self.needed_locks = {}
6674 06009e27 Iustin Pop
    if self.op.on_nodes:
6675 fbe9022f Guido Trotter
      # _GetWantedNodes can be used here, but is not always appropriate to use
6676 fbe9022f Guido Trotter
      # this way in ExpandNames. Check LogicalUnit.ExpandNames docstring for
6677 fbe9022f Guido Trotter
      # more information.
6678 06009e27 Iustin Pop
      self.op.on_nodes = _GetWantedNodes(self, self.op.on_nodes)
6679 fbe9022f Guido Trotter
      self.needed_locks[locking.LEVEL_NODE] = self.op.on_nodes
6680 fbe9022f Guido Trotter
6681 fbe9022f Guido Trotter
  def CheckPrereq(self):
6682 fbe9022f Guido Trotter
    """Check prerequisites.
6683 fbe9022f Guido Trotter

6684 fbe9022f Guido Trotter
    """
6685 06009e27 Iustin Pop
6686 06009e27 Iustin Pop
  def Exec(self, feedback_fn):
6687 06009e27 Iustin Pop
    """Do the actual sleep.
6688 06009e27 Iustin Pop

6689 06009e27 Iustin Pop
    """
6690 06009e27 Iustin Pop
    if self.op.on_master:
6691 06009e27 Iustin Pop
      if not utils.TestDelay(self.op.duration):
6692 06009e27 Iustin Pop
        raise errors.OpExecError("Error during master delay test")
6693 06009e27 Iustin Pop
    if self.op.on_nodes:
6694 72737a7f Iustin Pop
      result = self.rpc.call_test_delay(self.op.on_nodes, self.op.duration)
6695 06009e27 Iustin Pop
      if not result:
6696 06009e27 Iustin Pop
        raise errors.OpExecError("Complete failure from rpc call")
6697 06009e27 Iustin Pop
      for node, node_result in result.items():
6698 781de953 Iustin Pop
        node_result.Raise()
6699 781de953 Iustin Pop
        if not node_result.data:
6700 06009e27 Iustin Pop
          raise errors.OpExecError("Failure during rpc call to node %s,"
6701 781de953 Iustin Pop
                                   " result: %s" % (node, node_result.data))
6702 d61df03e Iustin Pop
6703 d61df03e Iustin Pop
6704 d1c2dd75 Iustin Pop
class IAllocator(object):
6705 d1c2dd75 Iustin Pop
  """IAllocator framework.
6706 d61df03e Iustin Pop

6707 d1c2dd75 Iustin Pop
  An IAllocator instance has three sets of attributes:
6708 d6a02168 Michael Hanselmann
    - cfg that is needed to query the cluster
6709 d1c2dd75 Iustin Pop
    - input data (all members of the _KEYS class attribute are required)
6710 d1c2dd75 Iustin Pop
    - four buffer attributes (in|out_data|text), that represent the
6711 d1c2dd75 Iustin Pop
      input (to the external script) in text and data structure format,
6712 d1c2dd75 Iustin Pop
      and the output from it, again in two formats
6713 d1c2dd75 Iustin Pop
    - the result variables from the script (success, info, nodes) for
6714 d1c2dd75 Iustin Pop
      easy usage
6715 d61df03e Iustin Pop

6716 d61df03e Iustin Pop
  """
6717 29859cb7 Iustin Pop
  _ALLO_KEYS = [
6718 d1c2dd75 Iustin Pop
    "mem_size", "disks", "disk_template",
6719 8cc7e742 Guido Trotter
    "os", "tags", "nics", "vcpus", "hypervisor",
6720 d1c2dd75 Iustin Pop
    ]
6721 29859cb7 Iustin Pop
  _RELO_KEYS = [
6722 29859cb7 Iustin Pop
    "relocate_from",
6723 29859cb7 Iustin Pop
    ]
6724 d1c2dd75 Iustin Pop
6725 72737a7f Iustin Pop
  def __init__(self, lu, mode, name, **kwargs):
6726 72737a7f Iustin Pop
    self.lu = lu
6727 d1c2dd75 Iustin Pop
    # init buffer variables
6728 d1c2dd75 Iustin Pop
    self.in_text = self.out_text = self.in_data = self.out_data = None
6729 d1c2dd75 Iustin Pop
    # init all input fields so that pylint is happy
6730 29859cb7 Iustin Pop
    self.mode = mode
6731 29859cb7 Iustin Pop
    self.name = name
6732 d1c2dd75 Iustin Pop
    self.mem_size = self.disks = self.disk_template = None
6733 d1c2dd75 Iustin Pop
    self.os = self.tags = self.nics = self.vcpus = None
6734 a0add446 Iustin Pop
    self.hypervisor = None
6735 29859cb7 Iustin Pop
    self.relocate_from = None
6736 27579978 Iustin Pop
    # computed fields
6737 27579978 Iustin Pop
    self.required_nodes = None
6738 d1c2dd75 Iustin Pop
    # init result fields
6739 d1c2dd75 Iustin Pop
    self.success = self.info = self.nodes = None
6740 29859cb7 Iustin Pop
    if self.mode == constants.IALLOCATOR_MODE_ALLOC:
6741 29859cb7 Iustin Pop
      keyset = self._ALLO_KEYS
6742 29859cb7 Iustin Pop
    elif self.mode == constants.IALLOCATOR_MODE_RELOC:
6743 29859cb7 Iustin Pop
      keyset = self._RELO_KEYS
6744 29859cb7 Iustin Pop
    else:
6745 29859cb7 Iustin Pop
      raise errors.ProgrammerError("Unknown mode '%s' passed to the"
6746 29859cb7 Iustin Pop
                                   " IAllocator" % self.mode)
6747 d1c2dd75 Iustin Pop
    for key in kwargs:
6748 29859cb7 Iustin Pop
      if key not in keyset:
6749 d1c2dd75 Iustin Pop
        raise errors.ProgrammerError("Invalid input parameter '%s' to"
6750 d1c2dd75 Iustin Pop
                                     " IAllocator" % key)
6751 d1c2dd75 Iustin Pop
      setattr(self, key, kwargs[key])
6752 29859cb7 Iustin Pop
    for key in keyset:
6753 d1c2dd75 Iustin Pop
      if key not in kwargs:
6754 d1c2dd75 Iustin Pop
        raise errors.ProgrammerError("Missing input parameter '%s' to"
6755 d1c2dd75 Iustin Pop
                                     " IAllocator" % key)
6756 d1c2dd75 Iustin Pop
    self._BuildInputData()
6757 d1c2dd75 Iustin Pop
6758 d1c2dd75 Iustin Pop
  def _ComputeClusterData(self):
6759 d1c2dd75 Iustin Pop
    """Compute the generic allocator input data.
6760 d1c2dd75 Iustin Pop

6761 d1c2dd75 Iustin Pop
    This is the data that is independent of the actual operation.
6762 d1c2dd75 Iustin Pop

6763 d1c2dd75 Iustin Pop
    """
6764 72737a7f Iustin Pop
    cfg = self.lu.cfg
6765 e69d05fd Iustin Pop
    cluster_info = cfg.GetClusterInfo()
6766 d1c2dd75 Iustin Pop
    # cluster data
6767 d1c2dd75 Iustin Pop
    data = {
6768 77031881 Iustin Pop
      "version": constants.IALLOCATOR_VERSION,
6769 72737a7f Iustin Pop
      "cluster_name": cfg.GetClusterName(),
6770 e69d05fd Iustin Pop
      "cluster_tags": list(cluster_info.GetTags()),
6771 1325da74 Iustin Pop
      "enabled_hypervisors": list(cluster_info.enabled_hypervisors),
6772 d1c2dd75 Iustin Pop
      # we don't have job IDs
6773 d61df03e Iustin Pop
      }
6774 b57e9819 Guido Trotter
    iinfo = cfg.GetAllInstancesInfo().values()
6775 b57e9819 Guido Trotter
    i_list = [(inst, cluster_info.FillBE(inst)) for inst in iinfo]
6776 6286519f Iustin Pop
6777 d1c2dd75 Iustin Pop
    # node data
6778 d1c2dd75 Iustin Pop
    node_results = {}
6779 d1c2dd75 Iustin Pop
    node_list = cfg.GetNodeList()
6780 8cc7e742 Guido Trotter
6781 8cc7e742 Guido Trotter
    if self.mode == constants.IALLOCATOR_MODE_ALLOC:
6782 a0add446 Iustin Pop
      hypervisor_name = self.hypervisor
6783 8cc7e742 Guido Trotter
    elif self.mode == constants.IALLOCATOR_MODE_RELOC:
6784 a0add446 Iustin Pop
      hypervisor_name = cfg.GetInstanceInfo(self.name).hypervisor
6785 8cc7e742 Guido Trotter
6786 72737a7f Iustin Pop
    node_data = self.lu.rpc.call_node_info(node_list, cfg.GetVGName(),
6787 a0add446 Iustin Pop
                                           hypervisor_name)
6788 18640d69 Guido Trotter
    node_iinfo = self.lu.rpc.call_all_instances_info(node_list,
6789 18640d69 Guido Trotter
                       cluster_info.enabled_hypervisors)
6790 1325da74 Iustin Pop
    for nname, nresult in node_data.items():
6791 1325da74 Iustin Pop
      # first fill in static (config-based) values
6792 d1c2dd75 Iustin Pop
      ninfo = cfg.GetNodeInfo(nname)
6793 d1c2dd75 Iustin Pop
      pnr = {
6794 d1c2dd75 Iustin Pop
        "tags": list(ninfo.GetTags()),
6795 d1c2dd75 Iustin Pop
        "primary_ip": ninfo.primary_ip,
6796 d1c2dd75 Iustin Pop
        "secondary_ip": ninfo.secondary_ip,
6797 fc0fe88c Iustin Pop
        "offline": ninfo.offline,
6798 0b2454b9 Iustin Pop
        "drained": ninfo.drained,
6799 1325da74 Iustin Pop
        "master_candidate": ninfo.master_candidate,
6800 d1c2dd75 Iustin Pop
        }
6801 1325da74 Iustin Pop
6802 1325da74 Iustin Pop
      if not ninfo.offline:
6803 1325da74 Iustin Pop
        nresult.Raise()
6804 1325da74 Iustin Pop
        if not isinstance(nresult.data, dict):
6805 1325da74 Iustin Pop
          raise errors.OpExecError("Can't get data for node %s" % nname)
6806 1325da74 Iustin Pop
        remote_info = nresult.data
6807 1325da74 Iustin Pop
        for attr in ['memory_total', 'memory_free', 'memory_dom0',
6808 1325da74 Iustin Pop
                     'vg_size', 'vg_free', 'cpu_total']:
6809 1325da74 Iustin Pop
          if attr not in remote_info:
6810 1325da74 Iustin Pop
            raise errors.OpExecError("Node '%s' didn't return attribute"
6811 1325da74 Iustin Pop
                                     " '%s'" % (nname, attr))
6812 1325da74 Iustin Pop
          try:
6813 1325da74 Iustin Pop
            remote_info[attr] = int(remote_info[attr])
6814 1325da74 Iustin Pop
          except ValueError, err:
6815 1325da74 Iustin Pop
            raise errors.OpExecError("Node '%s' returned invalid value"
6816 1325da74 Iustin Pop
                                     " for '%s': %s" % (nname, attr, err))
6817 1325da74 Iustin Pop
        # compute memory used by primary instances
6818 1325da74 Iustin Pop
        i_p_mem = i_p_up_mem = 0
6819 1325da74 Iustin Pop
        for iinfo, beinfo in i_list:
6820 1325da74 Iustin Pop
          if iinfo.primary_node == nname:
6821 1325da74 Iustin Pop
            i_p_mem += beinfo[constants.BE_MEMORY]
6822 1325da74 Iustin Pop
            if iinfo.name not in node_iinfo[nname].data:
6823 1325da74 Iustin Pop
              i_used_mem = 0
6824 1325da74 Iustin Pop
            else:
6825 1325da74 Iustin Pop
              i_used_mem = int(node_iinfo[nname].data[iinfo.name]['memory'])
6826 1325da74 Iustin Pop
            i_mem_diff = beinfo[constants.BE_MEMORY] - i_used_mem
6827 1325da74 Iustin Pop
            remote_info['memory_free'] -= max(0, i_mem_diff)
6828 1325da74 Iustin Pop
6829 1325da74 Iustin Pop
            if iinfo.admin_up:
6830 1325da74 Iustin Pop
              i_p_up_mem += beinfo[constants.BE_MEMORY]
6831 1325da74 Iustin Pop
6832 1325da74 Iustin Pop
        # compute memory used by instances
6833 1325da74 Iustin Pop
        pnr_dyn = {
6834 1325da74 Iustin Pop
          "total_memory": remote_info['memory_total'],
6835 1325da74 Iustin Pop
          "reserved_memory": remote_info['memory_dom0'],
6836 1325da74 Iustin Pop
          "free_memory": remote_info['memory_free'],
6837 1325da74 Iustin Pop
          "total_disk": remote_info['vg_size'],
6838 1325da74 Iustin Pop
          "free_disk": remote_info['vg_free'],
6839 1325da74 Iustin Pop
          "total_cpus": remote_info['cpu_total'],
6840 1325da74 Iustin Pop
          "i_pri_memory": i_p_mem,
6841 1325da74 Iustin Pop
          "i_pri_up_memory": i_p_up_mem,
6842 1325da74 Iustin Pop
          }
6843 1325da74 Iustin Pop
        pnr.update(pnr_dyn)
6844 1325da74 Iustin Pop
6845 d1c2dd75 Iustin Pop
      node_results[nname] = pnr
6846 d1c2dd75 Iustin Pop
    data["nodes"] = node_results
6847 d1c2dd75 Iustin Pop
6848 d1c2dd75 Iustin Pop
    # instance data
6849 d1c2dd75 Iustin Pop
    instance_data = {}
6850 338e51e8 Iustin Pop
    for iinfo, beinfo in i_list:
6851 d1c2dd75 Iustin Pop
      nic_data = [{"mac": n.mac, "ip": n.ip, "bridge": n.bridge}
6852 d1c2dd75 Iustin Pop
                  for n in iinfo.nics]
6853 d1c2dd75 Iustin Pop
      pir = {
6854 d1c2dd75 Iustin Pop
        "tags": list(iinfo.GetTags()),
6855 1325da74 Iustin Pop
        "admin_up": iinfo.admin_up,
6856 338e51e8 Iustin Pop
        "vcpus": beinfo[constants.BE_VCPUS],
6857 338e51e8 Iustin Pop
        "memory": beinfo[constants.BE_MEMORY],
6858 d1c2dd75 Iustin Pop
        "os": iinfo.os,
6859 1325da74 Iustin Pop
        "nodes": [iinfo.primary_node] + list(iinfo.secondary_nodes),
6860 d1c2dd75 Iustin Pop
        "nics": nic_data,
6861 1325da74 Iustin Pop
        "disks": [{"size": dsk.size, "mode": dsk.mode} for dsk in iinfo.disks],
6862 d1c2dd75 Iustin Pop
        "disk_template": iinfo.disk_template,
6863 e69d05fd Iustin Pop
        "hypervisor": iinfo.hypervisor,
6864 d1c2dd75 Iustin Pop
        }
6865 88ae4f85 Iustin Pop
      pir["disk_space_total"] = _ComputeDiskSize(iinfo.disk_template,
6866 88ae4f85 Iustin Pop
                                                 pir["disks"])
6867 768f0a80 Iustin Pop
      instance_data[iinfo.name] = pir
6868 d61df03e Iustin Pop
6869 d1c2dd75 Iustin Pop
    data["instances"] = instance_data
6870 d61df03e Iustin Pop
6871 d1c2dd75 Iustin Pop
    self.in_data = data
6872 d61df03e Iustin Pop
6873 d1c2dd75 Iustin Pop
  def _AddNewInstance(self):
6874 d1c2dd75 Iustin Pop
    """Add new instance data to allocator structure.
6875 d61df03e Iustin Pop

6876 d1c2dd75 Iustin Pop
    This in combination with _AllocatorGetClusterData will create the
6877 d1c2dd75 Iustin Pop
    correct structure needed as input for the allocator.
6878 d61df03e Iustin Pop

6879 d1c2dd75 Iustin Pop
    The checks for the completeness of the opcode must have already been
6880 d1c2dd75 Iustin Pop
    done.
6881 d61df03e Iustin Pop

6882 d1c2dd75 Iustin Pop
    """
6883 d1c2dd75 Iustin Pop
    data = self.in_data
6884 d1c2dd75 Iustin Pop
6885 dafc7302 Guido Trotter
    disk_space = _ComputeDiskSize(self.disk_template, self.disks)
6886 d1c2dd75 Iustin Pop
6887 27579978 Iustin Pop
    if self.disk_template in constants.DTS_NET_MIRROR:
6888 27579978 Iustin Pop
      self.required_nodes = 2
6889 27579978 Iustin Pop
    else:
6890 27579978 Iustin Pop
      self.required_nodes = 1
6891 d1c2dd75 Iustin Pop
    request = {
6892 d1c2dd75 Iustin Pop
      "type": "allocate",
6893 d1c2dd75 Iustin Pop
      "name": self.name,
6894 d1c2dd75 Iustin Pop
      "disk_template": self.disk_template,
6895 d1c2dd75 Iustin Pop
      "tags": self.tags,
6896 d1c2dd75 Iustin Pop
      "os": self.os,
6897 d1c2dd75 Iustin Pop
      "vcpus": self.vcpus,
6898 d1c2dd75 Iustin Pop
      "memory": self.mem_size,
6899 d1c2dd75 Iustin Pop
      "disks": self.disks,
6900 d1c2dd75 Iustin Pop
      "disk_space_total": disk_space,
6901 d1c2dd75 Iustin Pop
      "nics": self.nics,
6902 27579978 Iustin Pop
      "required_nodes": self.required_nodes,
6903 d1c2dd75 Iustin Pop
      }
6904 d1c2dd75 Iustin Pop
    data["request"] = request
6905 298fe380 Iustin Pop
6906 d1c2dd75 Iustin Pop
  def _AddRelocateInstance(self):
6907 d1c2dd75 Iustin Pop
    """Add relocate instance data to allocator structure.
6908 298fe380 Iustin Pop

6909 d1c2dd75 Iustin Pop
    This in combination with _IAllocatorGetClusterData will create the
6910 d1c2dd75 Iustin Pop
    correct structure needed as input for the allocator.
6911 d61df03e Iustin Pop

6912 d1c2dd75 Iustin Pop
    The checks for the completeness of the opcode must have already been
6913 d1c2dd75 Iustin Pop
    done.
6914 d61df03e Iustin Pop

6915 d1c2dd75 Iustin Pop
    """
6916 72737a7f Iustin Pop
    instance = self.lu.cfg.GetInstanceInfo(self.name)
6917 27579978 Iustin Pop
    if instance is None:
6918 27579978 Iustin Pop
      raise errors.ProgrammerError("Unknown instance '%s' passed to"
6919 27579978 Iustin Pop
                                   " IAllocator" % self.name)
6920 27579978 Iustin Pop
6921 27579978 Iustin Pop
    if instance.disk_template not in constants.DTS_NET_MIRROR:
6922 27579978 Iustin Pop
      raise errors.OpPrereqError("Can't relocate non-mirrored instances")
6923 27579978 Iustin Pop
6924 2a139bb0 Iustin Pop
    if len(instance.secondary_nodes) != 1:
6925 2a139bb0 Iustin Pop
      raise errors.OpPrereqError("Instance has not exactly one secondary node")
6926 2a139bb0 Iustin Pop
6927 27579978 Iustin Pop
    self.required_nodes = 1
6928 dafc7302 Guido Trotter
    disk_sizes = [{'size': disk.size} for disk in instance.disks]
6929 dafc7302 Guido Trotter
    disk_space = _ComputeDiskSize(instance.disk_template, disk_sizes)
6930 27579978 Iustin Pop
6931 d1c2dd75 Iustin Pop
    request = {
6932 2a139bb0 Iustin Pop
      "type": "relocate",
6933 d1c2dd75 Iustin Pop
      "name": self.name,
6934 27579978 Iustin Pop
      "disk_space_total": disk_space,
6935 27579978 Iustin Pop
      "required_nodes": self.required_nodes,
6936 29859cb7 Iustin Pop
      "relocate_from": self.relocate_from,
6937 d1c2dd75 Iustin Pop
      }
6938 27579978 Iustin Pop
    self.in_data["request"] = request
6939 d61df03e Iustin Pop
6940 d1c2dd75 Iustin Pop
  def _BuildInputData(self):
6941 d1c2dd75 Iustin Pop
    """Build input data structures.
6942 d61df03e Iustin Pop

6943 d1c2dd75 Iustin Pop
    """
6944 d1c2dd75 Iustin Pop
    self._ComputeClusterData()
6945 d61df03e Iustin Pop
6946 d1c2dd75 Iustin Pop
    if self.mode == constants.IALLOCATOR_MODE_ALLOC:
6947 d1c2dd75 Iustin Pop
      self._AddNewInstance()
6948 d1c2dd75 Iustin Pop
    else:
6949 d1c2dd75 Iustin Pop
      self._AddRelocateInstance()
6950 d61df03e Iustin Pop
6951 d1c2dd75 Iustin Pop
    self.in_text = serializer.Dump(self.in_data)
6952 d61df03e Iustin Pop
6953 72737a7f Iustin Pop
  def Run(self, name, validate=True, call_fn=None):
6954 d1c2dd75 Iustin Pop
    """Run an instance allocator and return the results.
6955 298fe380 Iustin Pop

6956 d1c2dd75 Iustin Pop
    """
6957 72737a7f Iustin Pop
    if call_fn is None:
6958 72737a7f Iustin Pop
      call_fn = self.lu.rpc.call_iallocator_runner
6959 d1c2dd75 Iustin Pop
    data = self.in_text
6960 298fe380 Iustin Pop
6961 72737a7f Iustin Pop
    result = call_fn(self.lu.cfg.GetMasterNode(), name, self.in_text)
6962 781de953 Iustin Pop
    result.Raise()
6963 298fe380 Iustin Pop
6964 781de953 Iustin Pop
    if not isinstance(result.data, (list, tuple)) or len(result.data) != 4:
6965 8d528b7c Iustin Pop
      raise errors.OpExecError("Invalid result from master iallocator runner")
6966 8d528b7c Iustin Pop
6967 781de953 Iustin Pop
    rcode, stdout, stderr, fail = result.data
6968 8d528b7c Iustin Pop
6969 8d528b7c Iustin Pop
    if rcode == constants.IARUN_NOTFOUND:
6970 8d528b7c Iustin Pop
      raise errors.OpExecError("Can't find allocator '%s'" % name)
6971 8d528b7c Iustin Pop
    elif rcode == constants.IARUN_FAILURE:
6972 38206f3c Iustin Pop
      raise errors.OpExecError("Instance allocator call failed: %s,"
6973 38206f3c Iustin Pop
                               " output: %s" % (fail, stdout+stderr))
6974 8d528b7c Iustin Pop
    self.out_text = stdout
6975 d1c2dd75 Iustin Pop
    if validate:
6976 d1c2dd75 Iustin Pop
      self._ValidateResult()
6977 298fe380 Iustin Pop
6978 d1c2dd75 Iustin Pop
  def _ValidateResult(self):
6979 d1c2dd75 Iustin Pop
    """Process the allocator results.
6980 538475ca Iustin Pop

6981 d1c2dd75 Iustin Pop
    This will process and if successful save the result in
6982 d1c2dd75 Iustin Pop
    self.out_data and the other parameters.
6983 538475ca Iustin Pop

6984 d1c2dd75 Iustin Pop
    """
6985 d1c2dd75 Iustin Pop
    try:
6986 d1c2dd75 Iustin Pop
      rdict = serializer.Load(self.out_text)
6987 d1c2dd75 Iustin Pop
    except Exception, err:
6988 d1c2dd75 Iustin Pop
      raise errors.OpExecError("Can't parse iallocator results: %s" % str(err))
6989 d1c2dd75 Iustin Pop
6990 d1c2dd75 Iustin Pop
    if not isinstance(rdict, dict):
6991 d1c2dd75 Iustin Pop
      raise errors.OpExecError("Can't parse iallocator results: not a dict")
6992 538475ca Iustin Pop
6993 d1c2dd75 Iustin Pop
    for key in "success", "info", "nodes":
6994 d1c2dd75 Iustin Pop
      if key not in rdict:
6995 d1c2dd75 Iustin Pop
        raise errors.OpExecError("Can't parse iallocator results:"
6996 d1c2dd75 Iustin Pop
                                 " missing key '%s'" % key)
6997 d1c2dd75 Iustin Pop
      setattr(self, key, rdict[key])
6998 538475ca Iustin Pop
6999 d1c2dd75 Iustin Pop
    if not isinstance(rdict["nodes"], list):
7000 d1c2dd75 Iustin Pop
      raise errors.OpExecError("Can't parse iallocator results: 'nodes' key"
7001 d1c2dd75 Iustin Pop
                               " is not a list")
7002 d1c2dd75 Iustin Pop
    self.out_data = rdict
7003 538475ca Iustin Pop
7004 538475ca Iustin Pop
7005 d61df03e Iustin Pop
class LUTestAllocator(NoHooksLU):
7006 d61df03e Iustin Pop
  """Run allocator tests.
7007 d61df03e Iustin Pop

7008 d61df03e Iustin Pop
  This LU runs the allocator tests
7009 d61df03e Iustin Pop

7010 d61df03e Iustin Pop
  """
7011 d61df03e Iustin Pop
  _OP_REQP = ["direction", "mode", "name"]
7012 d61df03e Iustin Pop
7013 d61df03e Iustin Pop
  def CheckPrereq(self):
7014 d61df03e Iustin Pop
    """Check prerequisites.
7015 d61df03e Iustin Pop

7016 d61df03e Iustin Pop
    This checks the opcode parameters depending on the director and mode test.
7017 d61df03e Iustin Pop

7018 d61df03e Iustin Pop
    """
7019 298fe380 Iustin Pop
    if self.op.mode == constants.IALLOCATOR_MODE_ALLOC:
7020 d61df03e Iustin Pop
      for attr in ["name", "mem_size", "disks", "disk_template",
7021 d61df03e Iustin Pop
                   "os", "tags", "nics", "vcpus"]:
7022 d61df03e Iustin Pop
        if not hasattr(self.op, attr):
7023 d61df03e Iustin Pop
          raise errors.OpPrereqError("Missing attribute '%s' on opcode input" %
7024 d61df03e Iustin Pop
                                     attr)
7025 d61df03e Iustin Pop
      iname = self.cfg.ExpandInstanceName(self.op.name)
7026 d61df03e Iustin Pop
      if iname is not None:
7027 d61df03e Iustin Pop
        raise errors.OpPrereqError("Instance '%s' already in the cluster" %
7028 d61df03e Iustin Pop
                                   iname)
7029 d61df03e Iustin Pop
      if not isinstance(self.op.nics, list):
7030 d61df03e Iustin Pop
        raise errors.OpPrereqError("Invalid parameter 'nics'")
7031 d61df03e Iustin Pop
      for row in self.op.nics:
7032 d61df03e Iustin Pop
        if (not isinstance(row, dict) or
7033 d61df03e Iustin Pop
            "mac" not in row or
7034 d61df03e Iustin Pop
            "ip" not in row or
7035 d61df03e Iustin Pop
            "bridge" not in row):
7036 d61df03e Iustin Pop
          raise errors.OpPrereqError("Invalid contents of the"
7037 d61df03e Iustin Pop
                                     " 'nics' parameter")
7038 d61df03e Iustin Pop
      if not isinstance(self.op.disks, list):
7039 d61df03e Iustin Pop
        raise errors.OpPrereqError("Invalid parameter 'disks'")
7040 d61df03e Iustin Pop
      for row in self.op.disks:
7041 d61df03e Iustin Pop
        if (not isinstance(row, dict) or
7042 d61df03e Iustin Pop
            "size" not in row or
7043 d61df03e Iustin Pop
            not isinstance(row["size"], int) or
7044 d61df03e Iustin Pop
            "mode" not in row or
7045 d61df03e Iustin Pop
            row["mode"] not in ['r', 'w']):
7046 d61df03e Iustin Pop
          raise errors.OpPrereqError("Invalid contents of the"
7047 d61df03e Iustin Pop
                                     " 'disks' parameter")
7048 8901997e Iustin Pop
      if not hasattr(self.op, "hypervisor") or self.op.hypervisor is None:
7049 8cc7e742 Guido Trotter
        self.op.hypervisor = self.cfg.GetHypervisorType()
7050 298fe380 Iustin Pop
    elif self.op.mode == constants.IALLOCATOR_MODE_RELOC:
7051 d61df03e Iustin Pop
      if not hasattr(self.op, "name"):
7052 d61df03e Iustin Pop
        raise errors.OpPrereqError("Missing attribute 'name' on opcode input")
7053 d61df03e Iustin Pop
      fname = self.cfg.ExpandInstanceName(self.op.name)
7054 d61df03e Iustin Pop
      if fname is None:
7055 d61df03e Iustin Pop
        raise errors.OpPrereqError("Instance '%s' not found for relocation" %
7056 d61df03e Iustin Pop
                                   self.op.name)
7057 d61df03e Iustin Pop
      self.op.name = fname
7058 29859cb7 Iustin Pop
      self.relocate_from = self.cfg.GetInstanceInfo(fname).secondary_nodes
7059 d61df03e Iustin Pop
    else:
7060 d61df03e Iustin Pop
      raise errors.OpPrereqError("Invalid test allocator mode '%s'" %
7061 d61df03e Iustin Pop
                                 self.op.mode)
7062 d61df03e Iustin Pop
7063 298fe380 Iustin Pop
    if self.op.direction == constants.IALLOCATOR_DIR_OUT:
7064 298fe380 Iustin Pop
      if not hasattr(self.op, "allocator") or self.op.allocator is None:
7065 d61df03e Iustin Pop
        raise errors.OpPrereqError("Missing allocator name")
7066 298fe380 Iustin Pop
    elif self.op.direction != constants.IALLOCATOR_DIR_IN:
7067 d61df03e Iustin Pop
      raise errors.OpPrereqError("Wrong allocator test '%s'" %
7068 d61df03e Iustin Pop
                                 self.op.direction)
7069 d61df03e Iustin Pop
7070 d61df03e Iustin Pop
  def Exec(self, feedback_fn):
7071 d61df03e Iustin Pop
    """Run the allocator test.
7072 d61df03e Iustin Pop

7073 d61df03e Iustin Pop
    """
7074 29859cb7 Iustin Pop
    if self.op.mode == constants.IALLOCATOR_MODE_ALLOC:
7075 72737a7f Iustin Pop
      ial = IAllocator(self,
7076 29859cb7 Iustin Pop
                       mode=self.op.mode,
7077 29859cb7 Iustin Pop
                       name=self.op.name,
7078 29859cb7 Iustin Pop
                       mem_size=self.op.mem_size,
7079 29859cb7 Iustin Pop
                       disks=self.op.disks,
7080 29859cb7 Iustin Pop
                       disk_template=self.op.disk_template,
7081 29859cb7 Iustin Pop
                       os=self.op.os,
7082 29859cb7 Iustin Pop
                       tags=self.op.tags,
7083 29859cb7 Iustin Pop
                       nics=self.op.nics,
7084 29859cb7 Iustin Pop
                       vcpus=self.op.vcpus,
7085 8cc7e742 Guido Trotter
                       hypervisor=self.op.hypervisor,
7086 29859cb7 Iustin Pop
                       )
7087 29859cb7 Iustin Pop
    else:
7088 72737a7f Iustin Pop
      ial = IAllocator(self,
7089 29859cb7 Iustin Pop
                       mode=self.op.mode,
7090 29859cb7 Iustin Pop
                       name=self.op.name,
7091 29859cb7 Iustin Pop
                       relocate_from=list(self.relocate_from),
7092 29859cb7 Iustin Pop
                       )
7093 d61df03e Iustin Pop
7094 298fe380 Iustin Pop
    if self.op.direction == constants.IALLOCATOR_DIR_IN:
7095 d1c2dd75 Iustin Pop
      result = ial.in_text
7096 298fe380 Iustin Pop
    else:
7097 d1c2dd75 Iustin Pop
      ial.Run(self.op.allocator, validate=False)
7098 d1c2dd75 Iustin Pop
      result = ial.out_text
7099 298fe380 Iustin Pop
    return result