Statistics
| Branch: | Tag: | Revision:

root / lib / cmdlib.py @ c5e489f7

History | View | Annotate | Download (228.2 kB)

1
#
2
#
3

    
4
# Copyright (C) 2006, 2007, 2008 Google Inc.
5
#
6
# This program is free software; you can redistribute it and/or modify
7
# it under the terms of the GNU General Public License as published by
8
# the Free Software Foundation; either version 2 of the License, or
9
# (at your option) any later version.
10
#
11
# This program is distributed in the hope that it will be useful, but
12
# WITHOUT ANY WARRANTY; without even the implied warranty of
13
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14
# General Public License for more details.
15
#
16
# You should have received a copy of the GNU General Public License
17
# along with this program; if not, write to the Free Software
18
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
19
# 02110-1301, USA.
20

    
21

    
22
"""Module implementing the master-side code."""
23

    
24
# pylint: disable-msg=W0613,W0201
25

    
26
import os
27
import os.path
28
import sha
29
import time
30
import tempfile
31
import re
32
import platform
33
import logging
34
import copy
35
import random
36

    
37
from ganeti import ssh
38
from ganeti import utils
39
from ganeti import errors
40
from ganeti import hypervisor
41
from ganeti import locking
42
from ganeti import constants
43
from ganeti import objects
44
from ganeti import opcodes
45
from ganeti import serializer
46
from ganeti import ssconf
47

    
48

    
49
class LogicalUnit(object):
50
  """Logical Unit base class.
51

52
  Subclasses must follow these rules:
53
    - implement ExpandNames
54
    - implement CheckPrereq
55
    - implement Exec
56
    - implement BuildHooksEnv
57
    - redefine HPATH and HTYPE
58
    - optionally redefine their run requirements:
59
        REQ_BGL: the LU needs to hold the Big Ganeti Lock exclusively
60

61
  Note that all commands require root permissions.
62

63
  """
64
  HPATH = None
65
  HTYPE = None
66
  _OP_REQP = []
67
  REQ_BGL = True
68

    
69
  def __init__(self, processor, op, context, rpc):
70
    """Constructor for LogicalUnit.
71

72
    This needs to be overriden in derived classes in order to check op
73
    validity.
74

75
    """
76
    self.proc = processor
77
    self.op = op
78
    self.cfg = context.cfg
79
    self.context = context
80
    self.rpc = rpc
81
    # Dicts used to declare locking needs to mcpu
82
    self.needed_locks = None
83
    self.acquired_locks = {}
84
    self.share_locks = dict(((i, 0) for i in locking.LEVELS))
85
    self.add_locks = {}
86
    self.remove_locks = {}
87
    # Used to force good behavior when calling helper functions
88
    self.recalculate_locks = {}
89
    self.__ssh = None
90
    # logging
91
    self.LogWarning = processor.LogWarning
92
    self.LogInfo = processor.LogInfo
93

    
94
    for attr_name in self._OP_REQP:
95
      attr_val = getattr(op, attr_name, None)
96
      if attr_val is None:
97
        raise errors.OpPrereqError("Required parameter '%s' missing" %
98
                                   attr_name)
99
    self.CheckArguments()
100

    
101
  def __GetSSH(self):
102
    """Returns the SshRunner object
103

104
    """
105
    if not self.__ssh:
106
      self.__ssh = ssh.SshRunner(self.cfg.GetClusterName())
107
    return self.__ssh
108

    
109
  ssh = property(fget=__GetSSH)
110

    
111
  def CheckArguments(self):
112
    """Check syntactic validity for the opcode arguments.
113

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

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

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

126
    """
127
    pass
128

    
129
  def ExpandNames(self):
130
    """Expand names for this LU.
131

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

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

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

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

150
    Examples::
151

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

164
    """
165
    # The implementation of this method is mandatory only if the new LU is
166
    # concurrent, so that old LUs don't need to be changed all at the same
167
    # time.
168
    if self.REQ_BGL:
169
      self.needed_locks = {} # Exclusive LUs don't need locks.
170
    else:
171
      raise NotImplementedError
172

    
173
  def DeclareLocks(self, level):
174
    """Declare LU locking needs for a level
175

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

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

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

189
    """
190

    
191
  def CheckPrereq(self):
192
    """Check prerequisites for this LU.
193

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

199
    The method should raise errors.OpPrereqError in case something is
200
    not fulfilled. Its return value is ignored.
201

202
    This method should also update all the parameters of the opcode to
203
    their canonical form if it hasn't been done by ExpandNames before.
204

205
    """
206
    raise NotImplementedError
207

    
208
  def Exec(self, feedback_fn):
209
    """Execute the LU.
210

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

215
    """
216
    raise NotImplementedError
217

    
218
  def BuildHooksEnv(self):
219
    """Build hooks environment for this LU.
220

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

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

232
    No nodes should be returned as an empty list (and not None).
233

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

237
    """
238
    raise NotImplementedError
239

    
240
  def HooksCallBack(self, phase, hook_results, feedback_fn, lu_result):
241
    """Notify the LU about the results of its hooks.
242

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

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

258
    """
259
    return lu_result
260

    
261
  def _ExpandAndLockInstance(self):
262
    """Helper function to expand and lock an instance.
263

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

270
    """
271
    if self.needed_locks is None:
272
      self.needed_locks = {}
273
    else:
274
      assert locking.LEVEL_INSTANCE not in self.needed_locks, \
275
        "_ExpandAndLockInstance called with instance-level locks set"
276
    expanded_name = self.cfg.ExpandInstanceName(self.op.instance_name)
277
    if expanded_name is None:
278
      raise errors.OpPrereqError("Instance '%s' not known" %
279
                                  self.op.instance_name)
280
    self.needed_locks[locking.LEVEL_INSTANCE] = expanded_name
281
    self.op.instance_name = expanded_name
282

    
283
  def _LockInstancesNodes(self, primary_only=False):
284
    """Helper function to declare instances' nodes for locking.
285

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

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

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

297
    If should be called in DeclareLocks in a way similar to::
298

299
      if level == locking.LEVEL_NODE:
300
        self._LockInstancesNodes()
301

302
    @type primary_only: boolean
303
    @param primary_only: only lock primary nodes of locked instances
304

305
    """
306
    assert locking.LEVEL_NODE in self.recalculate_locks, \
307
      "_LockInstancesNodes helper function called with no nodes to recalculate"
308

    
309
    # TODO: check if we're really been called with the instance locks held
310

    
311
    # For now we'll replace self.needed_locks[locking.LEVEL_NODE], but in the
312
    # future we might want to have different behaviors depending on the value
313
    # of self.recalculate_locks[locking.LEVEL_NODE]
314
    wanted_nodes = []
315
    for instance_name in self.acquired_locks[locking.LEVEL_INSTANCE]:
316
      instance = self.context.cfg.GetInstanceInfo(instance_name)
317
      wanted_nodes.append(instance.primary_node)
318
      if not primary_only:
319
        wanted_nodes.extend(instance.secondary_nodes)
320

    
321
    if self.recalculate_locks[locking.LEVEL_NODE] == constants.LOCKS_REPLACE:
322
      self.needed_locks[locking.LEVEL_NODE] = wanted_nodes
323
    elif self.recalculate_locks[locking.LEVEL_NODE] == constants.LOCKS_APPEND:
324
      self.needed_locks[locking.LEVEL_NODE].extend(wanted_nodes)
325

    
326
    del self.recalculate_locks[locking.LEVEL_NODE]
327

    
328

    
329
class NoHooksLU(LogicalUnit):
330
  """Simple LU which runs no hooks.
331

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

335
  """
336
  HPATH = None
337
  HTYPE = None
338

    
339

    
340
def _GetWantedNodes(lu, nodes):
341
  """Returns list of checked and expanded node names.
342

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

351
  """
352
  if not isinstance(nodes, list):
353
    raise errors.OpPrereqError("Invalid argument type 'nodes'")
354

    
355
  if not nodes:
356
    raise errors.ProgrammerError("_GetWantedNodes should only be called with a"
357
      " non-empty list of nodes whose name is to be expanded.")
358

    
359
  wanted = []
360
  for name in nodes:
361
    node = lu.cfg.ExpandNodeName(name)
362
    if node is None:
363
      raise errors.OpPrereqError("No such node name '%s'" % name)
364
    wanted.append(node)
365

    
366
  return utils.NiceSort(wanted)
367

    
368

    
369
def _GetWantedInstances(lu, instances):
370
  """Returns list of checked and expanded instance names.
371

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

381
  """
382
  if not isinstance(instances, list):
383
    raise errors.OpPrereqError("Invalid argument type 'instances'")
384

    
385
  if instances:
386
    wanted = []
387

    
388
    for name in instances:
389
      instance = lu.cfg.ExpandInstanceName(name)
390
      if instance is None:
391
        raise errors.OpPrereqError("No such instance name '%s'" % name)
392
      wanted.append(instance)
393

    
394
  else:
395
    wanted = lu.cfg.GetInstanceList()
396
  return utils.NiceSort(wanted)
397

    
398

    
399
def _CheckOutputFields(static, dynamic, selected):
400
  """Checks whether all selected fields are valid.
401

402
  @type static: L{utils.FieldSet}
403
  @param static: static fields set
404
  @type dynamic: L{utils.FieldSet}
405
  @param dynamic: dynamic fields set
406

407
  """
408
  f = utils.FieldSet()
409
  f.Extend(static)
410
  f.Extend(dynamic)
411

    
412
  delta = f.NonMatching(selected)
413
  if delta:
414
    raise errors.OpPrereqError("Unknown output fields selected: %s"
415
                               % ",".join(delta))
416

    
417

    
418
def _CheckBooleanOpField(op, name):
419
  """Validates boolean opcode parameters.
420

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

424
  """
425
  val = getattr(op, name, None)
426
  if not (val is None or isinstance(val, bool)):
427
    raise errors.OpPrereqError("Invalid boolean parameter '%s' (%s)" %
428
                               (name, str(val)))
429
  setattr(op, name, val)
430

    
431

    
432
def _CheckNodeOnline(lu, node):
433
  """Ensure that a given node is online.
434

435
  @param lu: the LU on behalf of which we make the check
436
  @param node: the node to check
437
  @raise errors.OpPrereqError: if the nodes is offline
438

439
  """
440
  if lu.cfg.GetNodeInfo(node).offline:
441
    raise errors.OpPrereqError("Can't use offline node %s" % node)
442

    
443

    
444
def _BuildInstanceHookEnv(name, primary_node, secondary_nodes, os_type, status,
445
                          memory, vcpus, nics):
446
  """Builds instance related env variables for hooks
447

448
  This builds the hook environment from individual variables.
449

450
  @type name: string
451
  @param name: the name of the instance
452
  @type primary_node: string
453
  @param primary_node: the name of the instance's primary node
454
  @type secondary_nodes: list
455
  @param secondary_nodes: list of secondary nodes as strings
456
  @type os_type: string
457
  @param os_type: the name of the instance's OS
458
  @type status: string
459
  @param status: the desired status of the instances
460
  @type memory: string
461
  @param memory: the memory size of the instance
462
  @type vcpus: string
463
  @param vcpus: the count of VCPUs the instance has
464
  @type nics: list
465
  @param nics: list of tuples (ip, bridge, mac) representing
466
      the NICs the instance  has
467
  @rtype: dict
468
  @return: the hook environment for this instance
469

470
  """
471
  env = {
472
    "OP_TARGET": name,
473
    "INSTANCE_NAME": name,
474
    "INSTANCE_PRIMARY": primary_node,
475
    "INSTANCE_SECONDARIES": " ".join(secondary_nodes),
476
    "INSTANCE_OS_TYPE": os_type,
477
    "INSTANCE_STATUS": status,
478
    "INSTANCE_MEMORY": memory,
479
    "INSTANCE_VCPUS": vcpus,
480
  }
481

    
482
  if nics:
483
    nic_count = len(nics)
484
    for idx, (ip, bridge, mac) in enumerate(nics):
485
      if ip is None:
486
        ip = ""
487
      env["INSTANCE_NIC%d_IP" % idx] = ip
488
      env["INSTANCE_NIC%d_BRIDGE" % idx] = bridge
489
      env["INSTANCE_NIC%d_HWADDR" % idx] = mac
490
  else:
491
    nic_count = 0
492

    
493
  env["INSTANCE_NIC_COUNT"] = nic_count
494

    
495
  return env
496

    
497

    
498
def _BuildInstanceHookEnvByObject(lu, instance, override=None):
499
  """Builds instance related env variables for hooks from an object.
500

501
  @type lu: L{LogicalUnit}
502
  @param lu: the logical unit on whose behalf we execute
503
  @type instance: L{objects.Instance}
504
  @param instance: the instance for which we should build the
505
      environment
506
  @type override: dict
507
  @param override: dictionary with key/values that will override
508
      our values
509
  @rtype: dict
510
  @return: the hook environment dictionary
511

512
  """
513
  bep = lu.cfg.GetClusterInfo().FillBE(instance)
514
  args = {
515
    'name': instance.name,
516
    'primary_node': instance.primary_node,
517
    'secondary_nodes': instance.secondary_nodes,
518
    'os_type': instance.os,
519
    'status': instance.os,
520
    'memory': bep[constants.BE_MEMORY],
521
    'vcpus': bep[constants.BE_VCPUS],
522
    'nics': [(nic.ip, nic.bridge, nic.mac) for nic in instance.nics],
523
  }
524
  if override:
525
    args.update(override)
526
  return _BuildInstanceHookEnv(**args)
527

    
528

    
529
def _AdjustCandidatePool(lu):
530
  """Adjust the candidate pool after node operations.
531

532
  """
533
  mod_list = lu.cfg.MaintainCandidatePool()
534
  if mod_list:
535
    lu.LogInfo("Promoted nodes to master candidate role: %s",
536
               ", ".join(node.name for node in mod_list))
537
    for name in mod_list:
538
      lu.context.ReaddNode(name)
539
  mc_now, mc_max = lu.cfg.GetMasterCandidateStats()
540
  if mc_now > mc_max:
541
    lu.LogInfo("Note: more nodes are candidates (%d) than desired (%d)" %
542
               (mc_now, mc_max))
543

    
544

    
545
def _CheckInstanceBridgesExist(lu, instance):
546
  """Check that the brigdes needed by an instance exist.
547

548
  """
549
  # check bridges existance
550
  brlist = [nic.bridge for nic in instance.nics]
551
  result = lu.rpc.call_bridges_exist(instance.primary_node, brlist)
552
  result.Raise()
553
  if not result.data:
554
    raise errors.OpPrereqError("One or more target bridges %s does not"
555
                               " exist on destination node '%s'" %
556
                               (brlist, instance.primary_node))
557

    
558

    
559
class LUDestroyCluster(NoHooksLU):
560
  """Logical unit for destroying the cluster.
561

562
  """
563
  _OP_REQP = []
564

    
565
  def CheckPrereq(self):
566
    """Check prerequisites.
567

568
    This checks whether the cluster is empty.
569

570
    Any errors are signalled by raising errors.OpPrereqError.
571

572
    """
573
    master = self.cfg.GetMasterNode()
574

    
575
    nodelist = self.cfg.GetNodeList()
576
    if len(nodelist) != 1 or nodelist[0] != master:
577
      raise errors.OpPrereqError("There are still %d node(s) in"
578
                                 " this cluster." % (len(nodelist) - 1))
579
    instancelist = self.cfg.GetInstanceList()
580
    if instancelist:
581
      raise errors.OpPrereqError("There are still %d instance(s) in"
582
                                 " this cluster." % len(instancelist))
583

    
584
  def Exec(self, feedback_fn):
585
    """Destroys the cluster.
586

587
    """
588
    master = self.cfg.GetMasterNode()
589
    result = self.rpc.call_node_stop_master(master, False)
590
    result.Raise()
591
    if not result.data:
592
      raise errors.OpExecError("Could not disable the master role")
593
    priv_key, pub_key, _ = ssh.GetUserFiles(constants.GANETI_RUNAS)
594
    utils.CreateBackup(priv_key)
595
    utils.CreateBackup(pub_key)
596
    return master
597

    
598

    
599
class LUVerifyCluster(LogicalUnit):
600
  """Verifies the cluster status.
601

602
  """
603
  HPATH = "cluster-verify"
604
  HTYPE = constants.HTYPE_CLUSTER
605
  _OP_REQP = ["skip_checks"]
606
  REQ_BGL = False
607

    
608
  def ExpandNames(self):
609
    self.needed_locks = {
610
      locking.LEVEL_NODE: locking.ALL_SET,
611
      locking.LEVEL_INSTANCE: locking.ALL_SET,
612
    }
613
    self.share_locks = dict(((i, 1) for i in locking.LEVELS))
614

    
615
  def _VerifyNode(self, nodeinfo, file_list, local_cksum,
616
                  node_result, feedback_fn, master_files):
617
    """Run multiple tests against a node.
618

619
    Test list:
620

621
      - compares ganeti version
622
      - checks vg existance and size > 20G
623
      - checks config file checksum
624
      - checks ssh to other nodes
625

626
    @type nodeinfo: L{objects.Node}
627
    @param nodeinfo: the node to check
628
    @param file_list: required list of files
629
    @param local_cksum: dictionary of local files and their checksums
630
    @param node_result: the results from the node
631
    @param feedback_fn: function used to accumulate results
632
    @param master_files: list of files that only masters should have
633

634
    """
635
    node = nodeinfo.name
636

    
637
    # main result, node_result should be a non-empty dict
638
    if not node_result or not isinstance(node_result, dict):
639
      feedback_fn("  - ERROR: unable to verify node %s." % (node,))
640
      return True
641

    
642
    # compares ganeti version
643
    local_version = constants.PROTOCOL_VERSION
644
    remote_version = node_result.get('version', None)
645
    if not remote_version:
646
      feedback_fn("  - ERROR: connection to %s failed" % (node))
647
      return True
648

    
649
    if local_version != remote_version:
650
      feedback_fn("  - ERROR: sw version mismatch: master %s, node(%s) %s" %
651
                      (local_version, node, remote_version))
652
      return True
653

    
654
    # checks vg existance and size > 20G
655

    
656
    bad = False
657
    vglist = node_result.get(constants.NV_VGLIST, None)
658
    if not vglist:
659
      feedback_fn("  - ERROR: unable to check volume groups on node %s." %
660
                      (node,))
661
      bad = True
662
    else:
663
      vgstatus = utils.CheckVolumeGroupSize(vglist, self.cfg.GetVGName(),
664
                                            constants.MIN_VG_SIZE)
665
      if vgstatus:
666
        feedback_fn("  - ERROR: %s on node %s" % (vgstatus, node))
667
        bad = True
668

    
669
    # checks config file checksum
670

    
671
    remote_cksum = node_result.get(constants.NV_FILELIST, None)
672
    if not isinstance(remote_cksum, dict):
673
      bad = True
674
      feedback_fn("  - ERROR: node hasn't returned file checksum data")
675
    else:
676
      for file_name in file_list:
677
        node_is_mc = nodeinfo.master_candidate
678
        must_have_file = file_name not in master_files
679
        if file_name not in remote_cksum:
680
          if node_is_mc or must_have_file:
681
            bad = True
682
            feedback_fn("  - ERROR: file '%s' missing" % file_name)
683
        elif remote_cksum[file_name] != local_cksum[file_name]:
684
          if node_is_mc or must_have_file:
685
            bad = True
686
            feedback_fn("  - ERROR: file '%s' has wrong checksum" % file_name)
687
          else:
688
            # not candidate and this is not a must-have file
689
            bad = True
690
            feedback_fn("  - ERROR: non master-candidate has old/wrong file"
691
                        " '%s'" % file_name)
692
        else:
693
          # all good, except non-master/non-must have combination
694
          if not node_is_mc and not must_have_file:
695
            feedback_fn("  - ERROR: file '%s' should not exist on non master"
696
                        " candidates" % file_name)
697

    
698
    # checks ssh to any
699

    
700
    if constants.NV_NODELIST not in node_result:
701
      bad = True
702
      feedback_fn("  - ERROR: node hasn't returned node ssh connectivity data")
703
    else:
704
      if node_result[constants.NV_NODELIST]:
705
        bad = True
706
        for node in node_result[constants.NV_NODELIST]:
707
          feedback_fn("  - ERROR: ssh communication with node '%s': %s" %
708
                          (node, node_result[constants.NV_NODELIST][node]))
709

    
710
    if constants.NV_NODENETTEST not in node_result:
711
      bad = True
712
      feedback_fn("  - ERROR: node hasn't returned node tcp connectivity data")
713
    else:
714
      if node_result[constants.NV_NODENETTEST]:
715
        bad = True
716
        nlist = utils.NiceSort(node_result[constants.NV_NODENETTEST].keys())
717
        for node in nlist:
718
          feedback_fn("  - ERROR: tcp communication with node '%s': %s" %
719
                          (node, node_result[constants.NV_NODENETTEST][node]))
720

    
721
    hyp_result = node_result.get(constants.NV_HYPERVISOR, None)
722
    if isinstance(hyp_result, dict):
723
      for hv_name, hv_result in hyp_result.iteritems():
724
        if hv_result is not None:
725
          feedback_fn("  - ERROR: hypervisor %s verify failure: '%s'" %
726
                      (hv_name, hv_result))
727
    return bad
728

    
729
  def _VerifyInstance(self, instance, instanceconfig, node_vol_is,
730
                      node_instance, feedback_fn, n_offline):
731
    """Verify an instance.
732

733
    This function checks to see if the required block devices are
734
    available on the instance's node.
735

736
    """
737
    bad = False
738

    
739
    node_current = instanceconfig.primary_node
740

    
741
    node_vol_should = {}
742
    instanceconfig.MapLVsByNode(node_vol_should)
743

    
744
    for node in node_vol_should:
745
      if node in n_offline:
746
        # ignore missing volumes on offline nodes
747
        continue
748
      for volume in node_vol_should[node]:
749
        if node not in node_vol_is or volume not in node_vol_is[node]:
750
          feedback_fn("  - ERROR: volume %s missing on node %s" %
751
                          (volume, node))
752
          bad = True
753

    
754
    if not instanceconfig.status == 'down':
755
      if ((node_current not in node_instance or
756
          not instance in node_instance[node_current]) and
757
          node_current not in n_offline):
758
        feedback_fn("  - ERROR: instance %s not running on node %s" %
759
                        (instance, node_current))
760
        bad = True
761

    
762
    for node in node_instance:
763
      if (not node == node_current):
764
        if instance in node_instance[node]:
765
          feedback_fn("  - ERROR: instance %s should not run on node %s" %
766
                          (instance, node))
767
          bad = True
768

    
769
    return bad
770

    
771
  def _VerifyOrphanVolumes(self, node_vol_should, node_vol_is, feedback_fn):
772
    """Verify if there are any unknown volumes in the cluster.
773

774
    The .os, .swap and backup volumes are ignored. All other volumes are
775
    reported as unknown.
776

777
    """
778
    bad = False
779

    
780
    for node in node_vol_is:
781
      for volume in node_vol_is[node]:
782
        if node not in node_vol_should or volume not in node_vol_should[node]:
783
          feedback_fn("  - ERROR: volume %s on node %s should not exist" %
784
                      (volume, node))
785
          bad = True
786
    return bad
787

    
788
  def _VerifyOrphanInstances(self, instancelist, node_instance, feedback_fn):
789
    """Verify the list of running instances.
790

791
    This checks what instances are running but unknown to the cluster.
792

793
    """
794
    bad = False
795
    for node in node_instance:
796
      for runninginstance in node_instance[node]:
797
        if runninginstance not in instancelist:
798
          feedback_fn("  - ERROR: instance %s on node %s should not exist" %
799
                          (runninginstance, node))
800
          bad = True
801
    return bad
802

    
803
  def _VerifyNPlusOneMemory(self, node_info, instance_cfg, feedback_fn):
804
    """Verify N+1 Memory Resilience.
805

806
    Check that if one single node dies we can still start all the instances it
807
    was primary for.
808

809
    """
810
    bad = False
811

    
812
    for node, nodeinfo in node_info.iteritems():
813
      # This code checks that every node which is now listed as secondary has
814
      # enough memory to host all instances it is supposed to should a single
815
      # other node in the cluster fail.
816
      # FIXME: not ready for failover to an arbitrary node
817
      # FIXME: does not support file-backed instances
818
      # WARNING: we currently take into account down instances as well as up
819
      # ones, considering that even if they're down someone might want to start
820
      # them even in the event of a node failure.
821
      for prinode, instances in nodeinfo['sinst-by-pnode'].iteritems():
822
        needed_mem = 0
823
        for instance in instances:
824
          bep = self.cfg.GetClusterInfo().FillBE(instance_cfg[instance])
825
          if bep[constants.BE_AUTO_BALANCE]:
826
            needed_mem += bep[constants.BE_MEMORY]
827
        if nodeinfo['mfree'] < needed_mem:
828
          feedback_fn("  - ERROR: not enough memory on node %s to accomodate"
829
                      " failovers should node %s fail" % (node, prinode))
830
          bad = True
831
    return bad
832

    
833
  def CheckPrereq(self):
834
    """Check prerequisites.
835

836
    Transform the list of checks we're going to skip into a set and check that
837
    all its members are valid.
838

839
    """
840
    self.skip_set = frozenset(self.op.skip_checks)
841
    if not constants.VERIFY_OPTIONAL_CHECKS.issuperset(self.skip_set):
842
      raise errors.OpPrereqError("Invalid checks to be skipped specified")
843

    
844
  def BuildHooksEnv(self):
845
    """Build hooks env.
846

847
    Cluster-Verify hooks just rone in the post phase and their failure makes
848
    the output be logged in the verify output and the verification to fail.
849

850
    """
851
    all_nodes = self.cfg.GetNodeList()
852
    # TODO: populate the environment with useful information for verify hooks
853
    env = {}
854
    return env, [], all_nodes
855

    
856
  def Exec(self, feedback_fn):
857
    """Verify integrity of cluster, performing various test on nodes.
858

859
    """
860
    bad = False
861
    feedback_fn("* Verifying global settings")
862
    for msg in self.cfg.VerifyConfig():
863
      feedback_fn("  - ERROR: %s" % msg)
864

    
865
    vg_name = self.cfg.GetVGName()
866
    hypervisors = self.cfg.GetClusterInfo().enabled_hypervisors
867
    nodelist = utils.NiceSort(self.cfg.GetNodeList())
868
    nodeinfo = [self.cfg.GetNodeInfo(nname) for nname in nodelist]
869
    instancelist = utils.NiceSort(self.cfg.GetInstanceList())
870
    i_non_redundant = [] # Non redundant instances
871
    i_non_a_balanced = [] # Non auto-balanced instances
872
    n_offline = [] # List of offline nodes
873
    node_volume = {}
874
    node_instance = {}
875
    node_info = {}
876
    instance_cfg = {}
877

    
878
    # FIXME: verify OS list
879
    # do local checksums
880
    master_files = [constants.CLUSTER_CONF_FILE]
881

    
882
    file_names = ssconf.SimpleStore().GetFileList()
883
    file_names.append(constants.SSL_CERT_FILE)
884
    file_names.append(constants.RAPI_CERT_FILE)
885
    file_names.extend(master_files)
886

    
887
    local_checksums = utils.FingerprintFiles(file_names)
888

    
889
    feedback_fn("* Gathering data (%d nodes)" % len(nodelist))
890
    node_verify_param = {
891
      constants.NV_FILELIST: file_names,
892
      constants.NV_NODELIST: [node.name for node in nodeinfo
893
                              if not node.offline],
894
      constants.NV_HYPERVISOR: hypervisors,
895
      constants.NV_NODENETTEST: [(node.name, node.primary_ip,
896
                                  node.secondary_ip) for node in nodeinfo
897
                                 if not node.offline],
898
      constants.NV_LVLIST: vg_name,
899
      constants.NV_INSTANCELIST: hypervisors,
900
      constants.NV_VGLIST: None,
901
      constants.NV_VERSION: None,
902
      constants.NV_HVINFO: self.cfg.GetHypervisorType(),
903
      }
904
    all_nvinfo = self.rpc.call_node_verify(nodelist, node_verify_param,
905
                                           self.cfg.GetClusterName())
906

    
907
    cluster = self.cfg.GetClusterInfo()
908
    master_node = self.cfg.GetMasterNode()
909
    for node_i in nodeinfo:
910
      node = node_i.name
911
      nresult = all_nvinfo[node].data
912

    
913
      if node_i.offline:
914
        feedback_fn("* Skipping offline node %s" % (node,))
915
        n_offline.append(node)
916
        continue
917

    
918
      if node == master_node:
919
        ntype = "master"
920
      elif node_i.master_candidate:
921
        ntype = "master candidate"
922
      else:
923
        ntype = "regular"
924
      feedback_fn("* Verifying node %s (%s)" % (node, ntype))
925

    
926
      if all_nvinfo[node].failed or not isinstance(nresult, dict):
927
        feedback_fn("  - ERROR: connection to %s failed" % (node,))
928
        bad = True
929
        continue
930

    
931
      result = self._VerifyNode(node_i, file_names, local_checksums,
932
                                nresult, feedback_fn, master_files)
933
      bad = bad or result
934

    
935
      lvdata = nresult.get(constants.NV_LVLIST, "Missing LV data")
936
      if isinstance(lvdata, basestring):
937
        feedback_fn("  - ERROR: LVM problem on node %s: %s" %
938
                    (node, lvdata.encode('string_escape')))
939
        bad = True
940
        node_volume[node] = {}
941
      elif not isinstance(lvdata, dict):
942
        feedback_fn("  - ERROR: connection to %s failed (lvlist)" % (node,))
943
        bad = True
944
        continue
945
      else:
946
        node_volume[node] = lvdata
947

    
948
      # node_instance
949
      idata = nresult.get(constants.NV_INSTANCELIST, None)
950
      if not isinstance(idata, list):
951
        feedback_fn("  - ERROR: connection to %s failed (instancelist)" %
952
                    (node,))
953
        bad = True
954
        continue
955

    
956
      node_instance[node] = idata
957

    
958
      # node_info
959
      nodeinfo = nresult.get(constants.NV_HVINFO, None)
960
      if not isinstance(nodeinfo, dict):
961
        feedback_fn("  - ERROR: connection to %s failed (hvinfo)" % (node,))
962
        bad = True
963
        continue
964

    
965
      try:
966
        node_info[node] = {
967
          "mfree": int(nodeinfo['memory_free']),
968
          "dfree": int(nresult[constants.NV_VGLIST][vg_name]),
969
          "pinst": [],
970
          "sinst": [],
971
          # dictionary holding all instances this node is secondary for,
972
          # grouped by their primary node. Each key is a cluster node, and each
973
          # value is a list of instances which have the key as primary and the
974
          # current node as secondary.  this is handy to calculate N+1 memory
975
          # availability if you can only failover from a primary to its
976
          # secondary.
977
          "sinst-by-pnode": {},
978
        }
979
      except ValueError:
980
        feedback_fn("  - ERROR: invalid value returned from node %s" % (node,))
981
        bad = True
982
        continue
983

    
984
    node_vol_should = {}
985

    
986
    for instance in instancelist:
987
      feedback_fn("* Verifying instance %s" % instance)
988
      inst_config = self.cfg.GetInstanceInfo(instance)
989
      result =  self._VerifyInstance(instance, inst_config, node_volume,
990
                                     node_instance, feedback_fn, n_offline)
991
      bad = bad or result
992
      inst_nodes_offline = []
993

    
994
      inst_config.MapLVsByNode(node_vol_should)
995

    
996
      instance_cfg[instance] = inst_config
997

    
998
      pnode = inst_config.primary_node
999
      if pnode in node_info:
1000
        node_info[pnode]['pinst'].append(instance)
1001
      elif pnode not in n_offline:
1002
        feedback_fn("  - ERROR: instance %s, connection to primary node"
1003
                    " %s failed" % (instance, pnode))
1004
        bad = True
1005

    
1006
      if pnode in n_offline:
1007
        inst_nodes_offline.append(pnode)
1008

    
1009
      # If the instance is non-redundant we cannot survive losing its primary
1010
      # node, so we are not N+1 compliant. On the other hand we have no disk
1011
      # templates with more than one secondary so that situation is not well
1012
      # supported either.
1013
      # FIXME: does not support file-backed instances
1014
      if len(inst_config.secondary_nodes) == 0:
1015
        i_non_redundant.append(instance)
1016
      elif len(inst_config.secondary_nodes) > 1:
1017
        feedback_fn("  - WARNING: multiple secondaries for instance %s"
1018
                    % instance)
1019

    
1020
      if not cluster.FillBE(inst_config)[constants.BE_AUTO_BALANCE]:
1021
        i_non_a_balanced.append(instance)
1022

    
1023
      for snode in inst_config.secondary_nodes:
1024
        if snode in node_info:
1025
          node_info[snode]['sinst'].append(instance)
1026
          if pnode not in node_info[snode]['sinst-by-pnode']:
1027
            node_info[snode]['sinst-by-pnode'][pnode] = []
1028
          node_info[snode]['sinst-by-pnode'][pnode].append(instance)
1029
        elif snode not in n_offline:
1030
          feedback_fn("  - ERROR: instance %s, connection to secondary node"
1031
                      " %s failed" % (instance, snode))
1032
          bad = True
1033
        if snode in n_offline:
1034
          inst_nodes_offline.append(snode)
1035

    
1036
      if inst_nodes_offline:
1037
        # warn that the instance lives on offline nodes, and set bad=True
1038
        feedback_fn("  - ERROR: instance lives on offline node(s) %s" %
1039
                    ", ".join(inst_nodes_offline))
1040
        bad = True
1041

    
1042
    feedback_fn("* Verifying orphan volumes")
1043
    result = self._VerifyOrphanVolumes(node_vol_should, node_volume,
1044
                                       feedback_fn)
1045
    bad = bad or result
1046

    
1047
    feedback_fn("* Verifying remaining instances")
1048
    result = self._VerifyOrphanInstances(instancelist, node_instance,
1049
                                         feedback_fn)
1050
    bad = bad or result
1051

    
1052
    if constants.VERIFY_NPLUSONE_MEM not in self.skip_set:
1053
      feedback_fn("* Verifying N+1 Memory redundancy")
1054
      result = self._VerifyNPlusOneMemory(node_info, instance_cfg, feedback_fn)
1055
      bad = bad or result
1056

    
1057
    feedback_fn("* Other Notes")
1058
    if i_non_redundant:
1059
      feedback_fn("  - NOTICE: %d non-redundant instance(s) found."
1060
                  % len(i_non_redundant))
1061

    
1062
    if i_non_a_balanced:
1063
      feedback_fn("  - NOTICE: %d non-auto-balanced instance(s) found."
1064
                  % len(i_non_a_balanced))
1065

    
1066
    if n_offline:
1067
      feedback_fn("  - NOTICE: %d offline node(s) found." % len(n_offline))
1068

    
1069
    return not bad
1070

    
1071
  def HooksCallBack(self, phase, hooks_results, feedback_fn, lu_result):
1072
    """Analize the post-hooks' result
1073

1074
    This method analyses the hook result, handles it, and sends some
1075
    nicely-formatted feedback back to the user.
1076

1077
    @param phase: one of L{constants.HOOKS_PHASE_POST} or
1078
        L{constants.HOOKS_PHASE_PRE}; it denotes the hooks phase
1079
    @param hooks_results: the results of the multi-node hooks rpc call
1080
    @param feedback_fn: function used send feedback back to the caller
1081
    @param lu_result: previous Exec result
1082
    @return: the new Exec result, based on the previous result
1083
        and hook results
1084

1085
    """
1086
    # We only really run POST phase hooks, and are only interested in
1087
    # their results
1088
    if phase == constants.HOOKS_PHASE_POST:
1089
      # Used to change hooks' output to proper indentation
1090
      indent_re = re.compile('^', re.M)
1091
      feedback_fn("* Hooks Results")
1092
      if not hooks_results:
1093
        feedback_fn("  - ERROR: general communication failure")
1094
        lu_result = 1
1095
      else:
1096
        for node_name in hooks_results:
1097
          show_node_header = True
1098
          res = hooks_results[node_name]
1099
          if res.failed or res.data is False or not isinstance(res.data, list):
1100
            if res.offline:
1101
              # no need to warn or set fail return value
1102
              continue
1103
            feedback_fn("    Communication failure in hooks execution")
1104
            lu_result = 1
1105
            continue
1106
          for script, hkr, output in res.data:
1107
            if hkr == constants.HKR_FAIL:
1108
              # The node header is only shown once, if there are
1109
              # failing hooks on that node
1110
              if show_node_header:
1111
                feedback_fn("  Node %s:" % node_name)
1112
                show_node_header = False
1113
              feedback_fn("    ERROR: Script %s failed, output:" % script)
1114
              output = indent_re.sub('      ', output)
1115
              feedback_fn("%s" % output)
1116
              lu_result = 1
1117

    
1118
      return lu_result
1119

    
1120

    
1121
class LUVerifyDisks(NoHooksLU):
1122
  """Verifies the cluster disks status.
1123

1124
  """
1125
  _OP_REQP = []
1126
  REQ_BGL = False
1127

    
1128
  def ExpandNames(self):
1129
    self.needed_locks = {
1130
      locking.LEVEL_NODE: locking.ALL_SET,
1131
      locking.LEVEL_INSTANCE: locking.ALL_SET,
1132
    }
1133
    self.share_locks = dict(((i, 1) for i in locking.LEVELS))
1134

    
1135
  def CheckPrereq(self):
1136
    """Check prerequisites.
1137

1138
    This has no prerequisites.
1139

1140
    """
1141
    pass
1142

    
1143
  def Exec(self, feedback_fn):
1144
    """Verify integrity of cluster disks.
1145

1146
    """
1147
    result = res_nodes, res_nlvm, res_instances, res_missing = [], {}, [], {}
1148

    
1149
    vg_name = self.cfg.GetVGName()
1150
    nodes = utils.NiceSort(self.cfg.GetNodeList())
1151
    instances = [self.cfg.GetInstanceInfo(name)
1152
                 for name in self.cfg.GetInstanceList()]
1153

    
1154
    nv_dict = {}
1155
    for inst in instances:
1156
      inst_lvs = {}
1157
      if (inst.status != "up" or
1158
          inst.disk_template not in constants.DTS_NET_MIRROR):
1159
        continue
1160
      inst.MapLVsByNode(inst_lvs)
1161
      # transform { iname: {node: [vol,],},} to {(node, vol): iname}
1162
      for node, vol_list in inst_lvs.iteritems():
1163
        for vol in vol_list:
1164
          nv_dict[(node, vol)] = inst
1165

    
1166
    if not nv_dict:
1167
      return result
1168

    
1169
    node_lvs = self.rpc.call_volume_list(nodes, vg_name)
1170

    
1171
    to_act = set()
1172
    for node in nodes:
1173
      # node_volume
1174
      lvs = node_lvs[node]
1175
      if lvs.failed:
1176
        if not lvs.offline:
1177
          self.LogWarning("Connection to node %s failed: %s" %
1178
                          (node, lvs.data))
1179
        continue
1180
      lvs = lvs.data
1181
      if isinstance(lvs, basestring):
1182
        logging.warning("Error enumerating LVs on node %s: %s", node, lvs)
1183
        res_nlvm[node] = lvs
1184
      elif not isinstance(lvs, dict):
1185
        logging.warning("Connection to node %s failed or invalid data"
1186
                        " returned", node)
1187
        res_nodes.append(node)
1188
        continue
1189

    
1190
      for lv_name, (_, lv_inactive, lv_online) in lvs.iteritems():
1191
        inst = nv_dict.pop((node, lv_name), None)
1192
        if (not lv_online and inst is not None
1193
            and inst.name not in res_instances):
1194
          res_instances.append(inst.name)
1195

    
1196
    # any leftover items in nv_dict are missing LVs, let's arrange the
1197
    # data better
1198
    for key, inst in nv_dict.iteritems():
1199
      if inst.name not in res_missing:
1200
        res_missing[inst.name] = []
1201
      res_missing[inst.name].append(key)
1202

    
1203
    return result
1204

    
1205

    
1206
class LURenameCluster(LogicalUnit):
1207
  """Rename the cluster.
1208

1209
  """
1210
  HPATH = "cluster-rename"
1211
  HTYPE = constants.HTYPE_CLUSTER
1212
  _OP_REQP = ["name"]
1213

    
1214
  def BuildHooksEnv(self):
1215
    """Build hooks env.
1216

1217
    """
1218
    env = {
1219
      "OP_TARGET": self.cfg.GetClusterName(),
1220
      "NEW_NAME": self.op.name,
1221
      }
1222
    mn = self.cfg.GetMasterNode()
1223
    return env, [mn], [mn]
1224

    
1225
  def CheckPrereq(self):
1226
    """Verify that the passed name is a valid one.
1227

1228
    """
1229
    hostname = utils.HostInfo(self.op.name)
1230

    
1231
    new_name = hostname.name
1232
    self.ip = new_ip = hostname.ip
1233
    old_name = self.cfg.GetClusterName()
1234
    old_ip = self.cfg.GetMasterIP()
1235
    if new_name == old_name and new_ip == old_ip:
1236
      raise errors.OpPrereqError("Neither the name nor the IP address of the"
1237
                                 " cluster has changed")
1238
    if new_ip != old_ip:
1239
      if utils.TcpPing(new_ip, constants.DEFAULT_NODED_PORT):
1240
        raise errors.OpPrereqError("The given cluster IP address (%s) is"
1241
                                   " reachable on the network. Aborting." %
1242
                                   new_ip)
1243

    
1244
    self.op.name = new_name
1245

    
1246
  def Exec(self, feedback_fn):
1247
    """Rename the cluster.
1248

1249
    """
1250
    clustername = self.op.name
1251
    ip = self.ip
1252

    
1253
    # shutdown the master IP
1254
    master = self.cfg.GetMasterNode()
1255
    result = self.rpc.call_node_stop_master(master, False)
1256
    if result.failed or not result.data:
1257
      raise errors.OpExecError("Could not disable the master role")
1258

    
1259
    try:
1260
      cluster = self.cfg.GetClusterInfo()
1261
      cluster.cluster_name = clustername
1262
      cluster.master_ip = ip
1263
      self.cfg.Update(cluster)
1264

    
1265
      # update the known hosts file
1266
      ssh.WriteKnownHostsFile(self.cfg, constants.SSH_KNOWN_HOSTS_FILE)
1267
      node_list = self.cfg.GetNodeList()
1268
      try:
1269
        node_list.remove(master)
1270
      except ValueError:
1271
        pass
1272
      result = self.rpc.call_upload_file(node_list,
1273
                                         constants.SSH_KNOWN_HOSTS_FILE)
1274
      for to_node, to_result in result.iteritems():
1275
        if to_result.failed or not to_result.data:
1276
          logging.error("Copy of file %s to node %s failed",
1277
                        constants.SSH_KNOWN_HOSTS_FILE, to_node)
1278

    
1279
    finally:
1280
      result = self.rpc.call_node_start_master(master, False)
1281
      if result.failed or not result.data:
1282
        self.LogWarning("Could not re-enable the master role on"
1283
                        " the master, please restart manually.")
1284

    
1285

    
1286
def _RecursiveCheckIfLVMBased(disk):
1287
  """Check if the given disk or its children are lvm-based.
1288

1289
  @type disk: L{objects.Disk}
1290
  @param disk: the disk to check
1291
  @rtype: booleean
1292
  @return: boolean indicating whether a LD_LV dev_type was found or not
1293

1294
  """
1295
  if disk.children:
1296
    for chdisk in disk.children:
1297
      if _RecursiveCheckIfLVMBased(chdisk):
1298
        return True
1299
  return disk.dev_type == constants.LD_LV
1300

    
1301

    
1302
class LUSetClusterParams(LogicalUnit):
1303
  """Change the parameters of the cluster.
1304

1305
  """
1306
  HPATH = "cluster-modify"
1307
  HTYPE = constants.HTYPE_CLUSTER
1308
  _OP_REQP = []
1309
  REQ_BGL = False
1310

    
1311
  def CheckParameters(self):
1312
    """Check parameters
1313

1314
    """
1315
    if not hasattr(self.op, "candidate_pool_size"):
1316
      self.op.candidate_pool_size = None
1317
    if self.op.candidate_pool_size is not None:
1318
      try:
1319
        self.op.candidate_pool_size = int(self.op.candidate_pool_size)
1320
      except ValueError, err:
1321
        raise errors.OpPrereqError("Invalid candidate_pool_size value: %s" %
1322
                                   str(err))
1323
      if self.op.candidate_pool_size < 1:
1324
        raise errors.OpPrereqError("At least one master candidate needed")
1325

    
1326
  def ExpandNames(self):
1327
    # FIXME: in the future maybe other cluster params won't require checking on
1328
    # all nodes to be modified.
1329
    self.needed_locks = {
1330
      locking.LEVEL_NODE: locking.ALL_SET,
1331
    }
1332
    self.share_locks[locking.LEVEL_NODE] = 1
1333

    
1334
  def BuildHooksEnv(self):
1335
    """Build hooks env.
1336

1337
    """
1338
    env = {
1339
      "OP_TARGET": self.cfg.GetClusterName(),
1340
      "NEW_VG_NAME": self.op.vg_name,
1341
      }
1342
    mn = self.cfg.GetMasterNode()
1343
    return env, [mn], [mn]
1344

    
1345
  def CheckPrereq(self):
1346
    """Check prerequisites.
1347

1348
    This checks whether the given params don't conflict and
1349
    if the given volume group is valid.
1350

1351
    """
1352
    # FIXME: This only works because there is only one parameter that can be
1353
    # changed or removed.
1354
    if self.op.vg_name is not None and not self.op.vg_name:
1355
      instances = self.cfg.GetAllInstancesInfo().values()
1356
      for inst in instances:
1357
        for disk in inst.disks:
1358
          if _RecursiveCheckIfLVMBased(disk):
1359
            raise errors.OpPrereqError("Cannot disable lvm storage while"
1360
                                       " lvm-based instances exist")
1361

    
1362
    node_list = self.acquired_locks[locking.LEVEL_NODE]
1363

    
1364
    # if vg_name not None, checks given volume group on all nodes
1365
    if self.op.vg_name:
1366
      vglist = self.rpc.call_vg_list(node_list)
1367
      for node in node_list:
1368
        if vglist[node].failed:
1369
          # ignoring down node
1370
          self.LogWarning("Node %s unreachable/error, ignoring" % node)
1371
          continue
1372
        vgstatus = utils.CheckVolumeGroupSize(vglist[node].data,
1373
                                              self.op.vg_name,
1374
                                              constants.MIN_VG_SIZE)
1375
        if vgstatus:
1376
          raise errors.OpPrereqError("Error on node '%s': %s" %
1377
                                     (node, vgstatus))
1378

    
1379
    self.cluster = cluster = self.cfg.GetClusterInfo()
1380
    # validate beparams changes
1381
    if self.op.beparams:
1382
      utils.CheckBEParams(self.op.beparams)
1383
      self.new_beparams = cluster.FillDict(
1384
        cluster.beparams[constants.BEGR_DEFAULT], self.op.beparams)
1385

    
1386
    # hypervisor list/parameters
1387
    self.new_hvparams = cluster.FillDict(cluster.hvparams, {})
1388
    if self.op.hvparams:
1389
      if not isinstance(self.op.hvparams, dict):
1390
        raise errors.OpPrereqError("Invalid 'hvparams' parameter on input")
1391
      for hv_name, hv_dict in self.op.hvparams.items():
1392
        if hv_name not in self.new_hvparams:
1393
          self.new_hvparams[hv_name] = hv_dict
1394
        else:
1395
          self.new_hvparams[hv_name].update(hv_dict)
1396

    
1397
    if self.op.enabled_hypervisors is not None:
1398
      self.hv_list = self.op.enabled_hypervisors
1399
    else:
1400
      self.hv_list = cluster.enabled_hypervisors
1401

    
1402
    if self.op.hvparams or self.op.enabled_hypervisors is not None:
1403
      # either the enabled list has changed, or the parameters have, validate
1404
      for hv_name, hv_params in self.new_hvparams.items():
1405
        if ((self.op.hvparams and hv_name in self.op.hvparams) or
1406
            (self.op.enabled_hypervisors and
1407
             hv_name in self.op.enabled_hypervisors)):
1408
          # either this is a new hypervisor, or its parameters have changed
1409
          hv_class = hypervisor.GetHypervisor(hv_name)
1410
          hv_class.CheckParameterSyntax(hv_params)
1411
          _CheckHVParams(self, node_list, hv_name, hv_params)
1412

    
1413
  def Exec(self, feedback_fn):
1414
    """Change the parameters of the cluster.
1415

1416
    """
1417
    if self.op.vg_name is not None:
1418
      if self.op.vg_name != self.cfg.GetVGName():
1419
        self.cfg.SetVGName(self.op.vg_name)
1420
      else:
1421
        feedback_fn("Cluster LVM configuration already in desired"
1422
                    " state, not changing")
1423
    if self.op.hvparams:
1424
      self.cluster.hvparams = self.new_hvparams
1425
    if self.op.enabled_hypervisors is not None:
1426
      self.cluster.enabled_hypervisors = self.op.enabled_hypervisors
1427
    if self.op.beparams:
1428
      self.cluster.beparams[constants.BEGR_DEFAULT] = self.new_beparams
1429
    if self.op.candidate_pool_size is not None:
1430
      self.cluster.candidate_pool_size = self.op.candidate_pool_size
1431

    
1432
    self.cfg.Update(self.cluster)
1433

    
1434
    # we want to update nodes after the cluster so that if any errors
1435
    # happen, we have recorded and saved the cluster info
1436
    if self.op.candidate_pool_size is not None:
1437
      _AdjustCandidatePool(self)
1438

    
1439

    
1440
class LURedistributeConfig(NoHooksLU):
1441
  """Force the redistribution of cluster configuration.
1442

1443
  This is a very simple LU.
1444

1445
  """
1446
  _OP_REQP = []
1447
  REQ_BGL = False
1448

    
1449
  def ExpandNames(self):
1450
    self.needed_locks = {
1451
      locking.LEVEL_NODE: locking.ALL_SET,
1452
    }
1453
    self.share_locks[locking.LEVEL_NODE] = 1
1454

    
1455
  def CheckPrereq(self):
1456
    """Check prerequisites.
1457

1458
    """
1459

    
1460
  def Exec(self, feedback_fn):
1461
    """Redistribute the configuration.
1462

1463
    """
1464
    self.cfg.Update(self.cfg.GetClusterInfo())
1465

    
1466

    
1467
def _WaitForSync(lu, instance, oneshot=False, unlock=False):
1468
  """Sleep and poll for an instance's disk to sync.
1469

1470
  """
1471
  if not instance.disks:
1472
    return True
1473

    
1474
  if not oneshot:
1475
    lu.proc.LogInfo("Waiting for instance %s to sync disks." % instance.name)
1476

    
1477
  node = instance.primary_node
1478

    
1479
  for dev in instance.disks:
1480
    lu.cfg.SetDiskID(dev, node)
1481

    
1482
  retries = 0
1483
  while True:
1484
    max_time = 0
1485
    done = True
1486
    cumul_degraded = False
1487
    rstats = lu.rpc.call_blockdev_getmirrorstatus(node, instance.disks)
1488
    if rstats.failed or not rstats.data:
1489
      lu.LogWarning("Can't get any data from node %s", node)
1490
      retries += 1
1491
      if retries >= 10:
1492
        raise errors.RemoteError("Can't contact node %s for mirror data,"
1493
                                 " aborting." % node)
1494
      time.sleep(6)
1495
      continue
1496
    rstats = rstats.data
1497
    retries = 0
1498
    for i in range(len(rstats)):
1499
      mstat = rstats[i]
1500
      if mstat is None:
1501
        lu.LogWarning("Can't compute data for node %s/%s",
1502
                           node, instance.disks[i].iv_name)
1503
        continue
1504
      # we ignore the ldisk parameter
1505
      perc_done, est_time, is_degraded, _ = mstat
1506
      cumul_degraded = cumul_degraded or (is_degraded and perc_done is None)
1507
      if perc_done is not None:
1508
        done = False
1509
        if est_time is not None:
1510
          rem_time = "%d estimated seconds remaining" % est_time
1511
          max_time = est_time
1512
        else:
1513
          rem_time = "no time estimate"
1514
        lu.proc.LogInfo("- device %s: %5.2f%% done, %s" %
1515
                        (instance.disks[i].iv_name, perc_done, rem_time))
1516
    if done or oneshot:
1517
      break
1518

    
1519
    time.sleep(min(60, max_time))
1520

    
1521
  if done:
1522
    lu.proc.LogInfo("Instance %s's disks are in sync." % instance.name)
1523
  return not cumul_degraded
1524

    
1525

    
1526
def _CheckDiskConsistency(lu, dev, node, on_primary, ldisk=False):
1527
  """Check that mirrors are not degraded.
1528

1529
  The ldisk parameter, if True, will change the test from the
1530
  is_degraded attribute (which represents overall non-ok status for
1531
  the device(s)) to the ldisk (representing the local storage status).
1532

1533
  """
1534
  lu.cfg.SetDiskID(dev, node)
1535
  if ldisk:
1536
    idx = 6
1537
  else:
1538
    idx = 5
1539

    
1540
  result = True
1541
  if on_primary or dev.AssembleOnSecondary():
1542
    rstats = lu.rpc.call_blockdev_find(node, dev)
1543
    if rstats.failed or not rstats.data:
1544
      logging.warning("Node %s: disk degraded, not found or node down", node)
1545
      result = False
1546
    else:
1547
      result = result and (not rstats.data[idx])
1548
  if dev.children:
1549
    for child in dev.children:
1550
      result = result and _CheckDiskConsistency(lu, child, node, on_primary)
1551

    
1552
  return result
1553

    
1554

    
1555
class LUDiagnoseOS(NoHooksLU):
1556
  """Logical unit for OS diagnose/query.
1557

1558
  """
1559
  _OP_REQP = ["output_fields", "names"]
1560
  REQ_BGL = False
1561
  _FIELDS_STATIC = utils.FieldSet()
1562
  _FIELDS_DYNAMIC = utils.FieldSet("name", "valid", "node_status")
1563

    
1564
  def ExpandNames(self):
1565
    if self.op.names:
1566
      raise errors.OpPrereqError("Selective OS query not supported")
1567

    
1568
    _CheckOutputFields(static=self._FIELDS_STATIC,
1569
                       dynamic=self._FIELDS_DYNAMIC,
1570
                       selected=self.op.output_fields)
1571

    
1572
    # Lock all nodes, in shared mode
1573
    self.needed_locks = {}
1574
    self.share_locks[locking.LEVEL_NODE] = 1
1575
    self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
1576

    
1577
  def CheckPrereq(self):
1578
    """Check prerequisites.
1579

1580
    """
1581

    
1582
  @staticmethod
1583
  def _DiagnoseByOS(node_list, rlist):
1584
    """Remaps a per-node return list into an a per-os per-node dictionary
1585

1586
    @param node_list: a list with the names of all nodes
1587
    @param rlist: a map with node names as keys and OS objects as values
1588

1589
    @rtype: dict
1590
    @returns: a dictionary with osnames as keys and as value another map, with
1591
        nodes as keys and list of OS objects as values, eg::
1592

1593
          {"debian-etch": {"node1": [<object>,...],
1594
                           "node2": [<object>,]}
1595
          }
1596

1597
    """
1598
    all_os = {}
1599
    for node_name, nr in rlist.iteritems():
1600
      if nr.failed or not nr.data:
1601
        continue
1602
      for os_obj in nr.data:
1603
        if os_obj.name not in all_os:
1604
          # build a list of nodes for this os containing empty lists
1605
          # for each node in node_list
1606
          all_os[os_obj.name] = {}
1607
          for nname in node_list:
1608
            all_os[os_obj.name][nname] = []
1609
        all_os[os_obj.name][node_name].append(os_obj)
1610
    return all_os
1611

    
1612
  def Exec(self, feedback_fn):
1613
    """Compute the list of OSes.
1614

1615
    """
1616
    node_list = self.acquired_locks[locking.LEVEL_NODE]
1617
    valid_nodes = [node for node in self.cfg.GetOnlineNodeList()
1618
                   if node in node_list]
1619
    node_data = self.rpc.call_os_diagnose(valid_nodes)
1620
    if node_data == False:
1621
      raise errors.OpExecError("Can't gather the list of OSes")
1622
    pol = self._DiagnoseByOS(valid_nodes, node_data)
1623
    output = []
1624
    for os_name, os_data in pol.iteritems():
1625
      row = []
1626
      for field in self.op.output_fields:
1627
        if field == "name":
1628
          val = os_name
1629
        elif field == "valid":
1630
          val = utils.all([osl and osl[0] for osl in os_data.values()])
1631
        elif field == "node_status":
1632
          val = {}
1633
          for node_name, nos_list in os_data.iteritems():
1634
            val[node_name] = [(v.status, v.path) for v in nos_list]
1635
        else:
1636
          raise errors.ParameterError(field)
1637
        row.append(val)
1638
      output.append(row)
1639

    
1640
    return output
1641

    
1642

    
1643
class LURemoveNode(LogicalUnit):
1644
  """Logical unit for removing a node.
1645

1646
  """
1647
  HPATH = "node-remove"
1648
  HTYPE = constants.HTYPE_NODE
1649
  _OP_REQP = ["node_name"]
1650

    
1651
  def BuildHooksEnv(self):
1652
    """Build hooks env.
1653

1654
    This doesn't run on the target node in the pre phase as a failed
1655
    node would then be impossible to remove.
1656

1657
    """
1658
    env = {
1659
      "OP_TARGET": self.op.node_name,
1660
      "NODE_NAME": self.op.node_name,
1661
      }
1662
    all_nodes = self.cfg.GetNodeList()
1663
    all_nodes.remove(self.op.node_name)
1664
    return env, all_nodes, all_nodes
1665

    
1666
  def CheckPrereq(self):
1667
    """Check prerequisites.
1668

1669
    This checks:
1670
     - the node exists in the configuration
1671
     - it does not have primary or secondary instances
1672
     - it's not the master
1673

1674
    Any errors are signalled by raising errors.OpPrereqError.
1675

1676
    """
1677
    node = self.cfg.GetNodeInfo(self.cfg.ExpandNodeName(self.op.node_name))
1678
    if node is None:
1679
      raise errors.OpPrereqError, ("Node '%s' is unknown." % self.op.node_name)
1680

    
1681
    instance_list = self.cfg.GetInstanceList()
1682

    
1683
    masternode = self.cfg.GetMasterNode()
1684
    if node.name == masternode:
1685
      raise errors.OpPrereqError("Node is the master node,"
1686
                                 " you need to failover first.")
1687

    
1688
    for instance_name in instance_list:
1689
      instance = self.cfg.GetInstanceInfo(instance_name)
1690
      if node.name in instance.all_nodes:
1691
        raise errors.OpPrereqError("Instance %s is still running on the node,"
1692
                                   " please remove first." % instance_name)
1693
    self.op.node_name = node.name
1694
    self.node = node
1695

    
1696
  def Exec(self, feedback_fn):
1697
    """Removes the node from the cluster.
1698

1699
    """
1700
    node = self.node
1701
    logging.info("Stopping the node daemon and removing configs from node %s",
1702
                 node.name)
1703

    
1704
    self.context.RemoveNode(node.name)
1705

    
1706
    self.rpc.call_node_leave_cluster(node.name)
1707

    
1708
    # Promote nodes to master candidate as needed
1709
    _AdjustCandidatePool(self)
1710

    
1711

    
1712
class LUQueryNodes(NoHooksLU):
1713
  """Logical unit for querying nodes.
1714

1715
  """
1716
  _OP_REQP = ["output_fields", "names"]
1717
  REQ_BGL = False
1718
  _FIELDS_DYNAMIC = utils.FieldSet(
1719
    "dtotal", "dfree",
1720
    "mtotal", "mnode", "mfree",
1721
    "bootid",
1722
    "ctotal",
1723
    )
1724

    
1725
  _FIELDS_STATIC = utils.FieldSet(
1726
    "name", "pinst_cnt", "sinst_cnt",
1727
    "pinst_list", "sinst_list",
1728
    "pip", "sip", "tags",
1729
    "serial_no",
1730
    "master_candidate",
1731
    "master",
1732
    "offline",
1733
    )
1734

    
1735
  def ExpandNames(self):
1736
    _CheckOutputFields(static=self._FIELDS_STATIC,
1737
                       dynamic=self._FIELDS_DYNAMIC,
1738
                       selected=self.op.output_fields)
1739

    
1740
    self.needed_locks = {}
1741
    self.share_locks[locking.LEVEL_NODE] = 1
1742

    
1743
    if self.op.names:
1744
      self.wanted = _GetWantedNodes(self, self.op.names)
1745
    else:
1746
      self.wanted = locking.ALL_SET
1747

    
1748
    self.do_locking = self._FIELDS_STATIC.NonMatching(self.op.output_fields)
1749
    if self.do_locking:
1750
      # if we don't request only static fields, we need to lock the nodes
1751
      self.needed_locks[locking.LEVEL_NODE] = self.wanted
1752

    
1753

    
1754
  def CheckPrereq(self):
1755
    """Check prerequisites.
1756

1757
    """
1758
    # The validation of the node list is done in the _GetWantedNodes,
1759
    # if non empty, and if empty, there's no validation to do
1760
    pass
1761

    
1762
  def Exec(self, feedback_fn):
1763
    """Computes the list of nodes and their attributes.
1764

1765
    """
1766
    all_info = self.cfg.GetAllNodesInfo()
1767
    if self.do_locking:
1768
      nodenames = self.acquired_locks[locking.LEVEL_NODE]
1769
    elif self.wanted != locking.ALL_SET:
1770
      nodenames = self.wanted
1771
      missing = set(nodenames).difference(all_info.keys())
1772
      if missing:
1773
        raise errors.OpExecError(
1774
          "Some nodes were removed before retrieving their data: %s" % missing)
1775
    else:
1776
      nodenames = all_info.keys()
1777

    
1778
    nodenames = utils.NiceSort(nodenames)
1779
    nodelist = [all_info[name] for name in nodenames]
1780

    
1781
    # begin data gathering
1782

    
1783
    if self.do_locking:
1784
      live_data = {}
1785
      node_data = self.rpc.call_node_info(nodenames, self.cfg.GetVGName(),
1786
                                          self.cfg.GetHypervisorType())
1787
      for name in nodenames:
1788
        nodeinfo = node_data[name]
1789
        if not nodeinfo.failed and nodeinfo.data:
1790
          nodeinfo = nodeinfo.data
1791
          fn = utils.TryConvert
1792
          live_data[name] = {
1793
            "mtotal": fn(int, nodeinfo.get('memory_total', None)),
1794
            "mnode": fn(int, nodeinfo.get('memory_dom0', None)),
1795
            "mfree": fn(int, nodeinfo.get('memory_free', None)),
1796
            "dtotal": fn(int, nodeinfo.get('vg_size', None)),
1797
            "dfree": fn(int, nodeinfo.get('vg_free', None)),
1798
            "ctotal": fn(int, nodeinfo.get('cpu_total', None)),
1799
            "bootid": nodeinfo.get('bootid', None),
1800
            }
1801
        else:
1802
          live_data[name] = {}
1803
    else:
1804
      live_data = dict.fromkeys(nodenames, {})
1805

    
1806
    node_to_primary = dict([(name, set()) for name in nodenames])
1807
    node_to_secondary = dict([(name, set()) for name in nodenames])
1808

    
1809
    inst_fields = frozenset(("pinst_cnt", "pinst_list",
1810
                             "sinst_cnt", "sinst_list"))
1811
    if inst_fields & frozenset(self.op.output_fields):
1812
      instancelist = self.cfg.GetInstanceList()
1813

    
1814
      for instance_name in instancelist:
1815
        inst = self.cfg.GetInstanceInfo(instance_name)
1816
        if inst.primary_node in node_to_primary:
1817
          node_to_primary[inst.primary_node].add(inst.name)
1818
        for secnode in inst.secondary_nodes:
1819
          if secnode in node_to_secondary:
1820
            node_to_secondary[secnode].add(inst.name)
1821

    
1822
    master_node = self.cfg.GetMasterNode()
1823

    
1824
    # end data gathering
1825

    
1826
    output = []
1827
    for node in nodelist:
1828
      node_output = []
1829
      for field in self.op.output_fields:
1830
        if field == "name":
1831
          val = node.name
1832
        elif field == "pinst_list":
1833
          val = list(node_to_primary[node.name])
1834
        elif field == "sinst_list":
1835
          val = list(node_to_secondary[node.name])
1836
        elif field == "pinst_cnt":
1837
          val = len(node_to_primary[node.name])
1838
        elif field == "sinst_cnt":
1839
          val = len(node_to_secondary[node.name])
1840
        elif field == "pip":
1841
          val = node.primary_ip
1842
        elif field == "sip":
1843
          val = node.secondary_ip
1844
        elif field == "tags":
1845
          val = list(node.GetTags())
1846
        elif field == "serial_no":
1847
          val = node.serial_no
1848
        elif field == "master_candidate":
1849
          val = node.master_candidate
1850
        elif field == "master":
1851
          val = node.name == master_node
1852
        elif field == "offline":
1853
          val = node.offline
1854
        elif self._FIELDS_DYNAMIC.Matches(field):
1855
          val = live_data[node.name].get(field, None)
1856
        else:
1857
          raise errors.ParameterError(field)
1858
        node_output.append(val)
1859
      output.append(node_output)
1860

    
1861
    return output
1862

    
1863

    
1864
class LUQueryNodeVolumes(NoHooksLU):
1865
  """Logical unit for getting volumes on node(s).
1866

1867
  """
1868
  _OP_REQP = ["nodes", "output_fields"]
1869
  REQ_BGL = False
1870
  _FIELDS_DYNAMIC = utils.FieldSet("phys", "vg", "name", "size", "instance")
1871
  _FIELDS_STATIC = utils.FieldSet("node")
1872

    
1873
  def ExpandNames(self):
1874
    _CheckOutputFields(static=self._FIELDS_STATIC,
1875
                       dynamic=self._FIELDS_DYNAMIC,
1876
                       selected=self.op.output_fields)
1877

    
1878
    self.needed_locks = {}
1879
    self.share_locks[locking.LEVEL_NODE] = 1
1880
    if not self.op.nodes:
1881
      self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
1882
    else:
1883
      self.needed_locks[locking.LEVEL_NODE] = \
1884
        _GetWantedNodes(self, self.op.nodes)
1885

    
1886
  def CheckPrereq(self):
1887
    """Check prerequisites.
1888

1889
    This checks that the fields required are valid output fields.
1890

1891
    """
1892
    self.nodes = self.acquired_locks[locking.LEVEL_NODE]
1893

    
1894
  def Exec(self, feedback_fn):
1895
    """Computes the list of nodes and their attributes.
1896

1897
    """
1898
    nodenames = self.nodes
1899
    volumes = self.rpc.call_node_volumes(nodenames)
1900

    
1901
    ilist = [self.cfg.GetInstanceInfo(iname) for iname
1902
             in self.cfg.GetInstanceList()]
1903

    
1904
    lv_by_node = dict([(inst, inst.MapLVsByNode()) for inst in ilist])
1905

    
1906
    output = []
1907
    for node in nodenames:
1908
      if node not in volumes or volumes[node].failed or not volumes[node].data:
1909
        continue
1910

    
1911
      node_vols = volumes[node].data[:]
1912
      node_vols.sort(key=lambda vol: vol['dev'])
1913

    
1914
      for vol in node_vols:
1915
        node_output = []
1916
        for field in self.op.output_fields:
1917
          if field == "node":
1918
            val = node
1919
          elif field == "phys":
1920
            val = vol['dev']
1921
          elif field == "vg":
1922
            val = vol['vg']
1923
          elif field == "name":
1924
            val = vol['name']
1925
          elif field == "size":
1926
            val = int(float(vol['size']))
1927
          elif field == "instance":
1928
            for inst in ilist:
1929
              if node not in lv_by_node[inst]:
1930
                continue
1931
              if vol['name'] in lv_by_node[inst][node]:
1932
                val = inst.name
1933
                break
1934
            else:
1935
              val = '-'
1936
          else:
1937
            raise errors.ParameterError(field)
1938
          node_output.append(str(val))
1939

    
1940
        output.append(node_output)
1941

    
1942
    return output
1943

    
1944

    
1945
class LUAddNode(LogicalUnit):
1946
  """Logical unit for adding node to the cluster.
1947

1948
  """
1949
  HPATH = "node-add"
1950
  HTYPE = constants.HTYPE_NODE
1951
  _OP_REQP = ["node_name"]
1952

    
1953
  def BuildHooksEnv(self):
1954
    """Build hooks env.
1955

1956
    This will run on all nodes before, and on all nodes + the new node after.
1957

1958
    """
1959
    env = {
1960
      "OP_TARGET": self.op.node_name,
1961
      "NODE_NAME": self.op.node_name,
1962
      "NODE_PIP": self.op.primary_ip,
1963
      "NODE_SIP": self.op.secondary_ip,
1964
      }
1965
    nodes_0 = self.cfg.GetNodeList()
1966
    nodes_1 = nodes_0 + [self.op.node_name, ]
1967
    return env, nodes_0, nodes_1
1968

    
1969
  def CheckPrereq(self):
1970
    """Check prerequisites.
1971

1972
    This checks:
1973
     - the new node is not already in the config
1974
     - it is resolvable
1975
     - its parameters (single/dual homed) matches the cluster
1976

1977
    Any errors are signalled by raising errors.OpPrereqError.
1978

1979
    """
1980
    node_name = self.op.node_name
1981
    cfg = self.cfg
1982

    
1983
    dns_data = utils.HostInfo(node_name)
1984

    
1985
    node = dns_data.name
1986
    primary_ip = self.op.primary_ip = dns_data.ip
1987
    secondary_ip = getattr(self.op, "secondary_ip", None)
1988
    if secondary_ip is None:
1989
      secondary_ip = primary_ip
1990
    if not utils.IsValidIP(secondary_ip):
1991
      raise errors.OpPrereqError("Invalid secondary IP given")
1992
    self.op.secondary_ip = secondary_ip
1993

    
1994
    node_list = cfg.GetNodeList()
1995
    if not self.op.readd and node in node_list:
1996
      raise errors.OpPrereqError("Node %s is already in the configuration" %
1997
                                 node)
1998
    elif self.op.readd and node not in node_list:
1999
      raise errors.OpPrereqError("Node %s is not in the configuration" % node)
2000

    
2001
    for existing_node_name in node_list:
2002
      existing_node = cfg.GetNodeInfo(existing_node_name)
2003

    
2004
      if self.op.readd and node == existing_node_name:
2005
        if (existing_node.primary_ip != primary_ip or
2006
            existing_node.secondary_ip != secondary_ip):
2007
          raise errors.OpPrereqError("Readded node doesn't have the same IP"
2008
                                     " address configuration as before")
2009
        continue
2010

    
2011
      if (existing_node.primary_ip == primary_ip or
2012
          existing_node.secondary_ip == primary_ip or
2013
          existing_node.primary_ip == secondary_ip or
2014
          existing_node.secondary_ip == secondary_ip):
2015
        raise errors.OpPrereqError("New node ip address(es) conflict with"
2016
                                   " existing node %s" % existing_node.name)
2017

    
2018
    # check that the type of the node (single versus dual homed) is the
2019
    # same as for the master
2020
    myself = cfg.GetNodeInfo(self.cfg.GetMasterNode())
2021
    master_singlehomed = myself.secondary_ip == myself.primary_ip
2022
    newbie_singlehomed = secondary_ip == primary_ip
2023
    if master_singlehomed != newbie_singlehomed:
2024
      if master_singlehomed:
2025
        raise errors.OpPrereqError("The master has no private ip but the"
2026
                                   " new node has one")
2027
      else:
2028
        raise errors.OpPrereqError("The master has a private ip but the"
2029
                                   " new node doesn't have one")
2030

    
2031
    # checks reachablity
2032
    if not utils.TcpPing(primary_ip, constants.DEFAULT_NODED_PORT):
2033
      raise errors.OpPrereqError("Node not reachable by ping")
2034

    
2035
    if not newbie_singlehomed:
2036
      # check reachability from my secondary ip to newbie's secondary ip
2037
      if not utils.TcpPing(secondary_ip, constants.DEFAULT_NODED_PORT,
2038
                           source=myself.secondary_ip):
2039
        raise errors.OpPrereqError("Node secondary ip not reachable by TCP"
2040
                                   " based ping to noded port")
2041

    
2042
    cp_size = self.cfg.GetClusterInfo().candidate_pool_size
2043
    mc_now, _ = self.cfg.GetMasterCandidateStats()
2044
    master_candidate = mc_now < cp_size
2045

    
2046
    self.new_node = objects.Node(name=node,
2047
                                 primary_ip=primary_ip,
2048
                                 secondary_ip=secondary_ip,
2049
                                 master_candidate=master_candidate,
2050
                                 offline=False)
2051

    
2052
  def Exec(self, feedback_fn):
2053
    """Adds the new node to the cluster.
2054

2055
    """
2056
    new_node = self.new_node
2057
    node = new_node.name
2058

    
2059
    # check connectivity
2060
    result = self.rpc.call_version([node])[node]
2061
    result.Raise()
2062
    if result.data:
2063
      if constants.PROTOCOL_VERSION == result.data:
2064
        logging.info("Communication to node %s fine, sw version %s match",
2065
                     node, result.data)
2066
      else:
2067
        raise errors.OpExecError("Version mismatch master version %s,"
2068
                                 " node version %s" %
2069
                                 (constants.PROTOCOL_VERSION, result.data))
2070
    else:
2071
      raise errors.OpExecError("Cannot get version from the new node")
2072

    
2073
    # setup ssh on node
2074
    logging.info("Copy ssh key to node %s", node)
2075
    priv_key, pub_key, _ = ssh.GetUserFiles(constants.GANETI_RUNAS)
2076
    keyarray = []
2077
    keyfiles = [constants.SSH_HOST_DSA_PRIV, constants.SSH_HOST_DSA_PUB,
2078
                constants.SSH_HOST_RSA_PRIV, constants.SSH_HOST_RSA_PUB,
2079
                priv_key, pub_key]
2080

    
2081
    for i in keyfiles:
2082
      f = open(i, 'r')
2083
      try:
2084
        keyarray.append(f.read())
2085
      finally:
2086
        f.close()
2087

    
2088
    result = self.rpc.call_node_add(node, keyarray[0], keyarray[1],
2089
                                    keyarray[2],
2090
                                    keyarray[3], keyarray[4], keyarray[5])
2091

    
2092
    if result.failed or not result.data:
2093
      raise errors.OpExecError("Cannot transfer ssh keys to the new node")
2094

    
2095
    # Add node to our /etc/hosts, and add key to known_hosts
2096
    utils.AddHostToEtcHosts(new_node.name)
2097

    
2098
    if new_node.secondary_ip != new_node.primary_ip:
2099
      result = self.rpc.call_node_has_ip_address(new_node.name,
2100
                                                 new_node.secondary_ip)
2101
      if result.failed or not result.data:
2102
        raise errors.OpExecError("Node claims it doesn't have the secondary ip"
2103
                                 " you gave (%s). Please fix and re-run this"
2104
                                 " command." % new_node.secondary_ip)
2105

    
2106
    node_verify_list = [self.cfg.GetMasterNode()]
2107
    node_verify_param = {
2108
      'nodelist': [node],
2109
      # TODO: do a node-net-test as well?
2110
    }
2111

    
2112
    result = self.rpc.call_node_verify(node_verify_list, node_verify_param,
2113
                                       self.cfg.GetClusterName())
2114
    for verifier in node_verify_list:
2115
      if result[verifier].failed or not result[verifier].data:
2116
        raise errors.OpExecError("Cannot communicate with %s's node daemon"
2117
                                 " for remote verification" % verifier)
2118
      if result[verifier].data['nodelist']:
2119
        for failed in result[verifier].data['nodelist']:
2120
          feedback_fn("ssh/hostname verification failed %s -> %s" %
2121
                      (verifier, result[verifier]['nodelist'][failed]))
2122
        raise errors.OpExecError("ssh/hostname verification failed.")
2123

    
2124
    # Distribute updated /etc/hosts and known_hosts to all nodes,
2125
    # including the node just added
2126
    myself = self.cfg.GetNodeInfo(self.cfg.GetMasterNode())
2127
    dist_nodes = self.cfg.GetNodeList()
2128
    if not self.op.readd:
2129
      dist_nodes.append(node)
2130
    if myself.name in dist_nodes:
2131
      dist_nodes.remove(myself.name)
2132

    
2133
    logging.debug("Copying hosts and known_hosts to all nodes")
2134
    for fname in (constants.ETC_HOSTS, constants.SSH_KNOWN_HOSTS_FILE):
2135
      result = self.rpc.call_upload_file(dist_nodes, fname)
2136
      for to_node, to_result in result.iteritems():
2137
        if to_result.failed or not to_result.data:
2138
          logging.error("Copy of file %s to node %s failed", fname, to_node)
2139

    
2140
    to_copy = []
2141
    if constants.HT_XEN_HVM in self.cfg.GetClusterInfo().enabled_hypervisors:
2142
      to_copy.append(constants.VNC_PASSWORD_FILE)
2143
    for fname in to_copy:
2144
      result = self.rpc.call_upload_file([node], fname)
2145
      if result[node].failed or not result[node]:
2146
        logging.error("Could not copy file %s to node %s", fname, node)
2147

    
2148
    if self.op.readd:
2149
      self.context.ReaddNode(new_node)
2150
    else:
2151
      self.context.AddNode(new_node)
2152

    
2153

    
2154
class LUSetNodeParams(LogicalUnit):
2155
  """Modifies the parameters of a node.
2156

2157
  """
2158
  HPATH = "node-modify"
2159
  HTYPE = constants.HTYPE_NODE
2160
  _OP_REQP = ["node_name"]
2161
  REQ_BGL = False
2162

    
2163
  def CheckArguments(self):
2164
    node_name = self.cfg.ExpandNodeName(self.op.node_name)
2165
    if node_name is None:
2166
      raise errors.OpPrereqError("Invalid node name '%s'" % self.op.node_name)
2167
    self.op.node_name = node_name
2168
    _CheckBooleanOpField(self.op, 'master_candidate')
2169
    _CheckBooleanOpField(self.op, 'offline')
2170
    if self.op.master_candidate is None and self.op.offline is None:
2171
      raise errors.OpPrereqError("Please pass at least one modification")
2172
    if self.op.offline == True and self.op.master_candidate == True:
2173
      raise errors.OpPrereqError("Can't set the node into offline and"
2174
                                 " master_candidate at the same time")
2175

    
2176
  def ExpandNames(self):
2177
    self.needed_locks = {locking.LEVEL_NODE: self.op.node_name}
2178

    
2179
  def BuildHooksEnv(self):
2180
    """Build hooks env.
2181

2182
    This runs on the master node.
2183

2184
    """
2185
    env = {
2186
      "OP_TARGET": self.op.node_name,
2187
      "MASTER_CANDIDATE": str(self.op.master_candidate),
2188
      "OFFLINE": str(self.op.offline),
2189
      }
2190
    nl = [self.cfg.GetMasterNode(),
2191
          self.op.node_name]
2192
    return env, nl, nl
2193

    
2194
  def CheckPrereq(self):
2195
    """Check prerequisites.
2196

2197
    This only checks the instance list against the existing names.
2198

2199
    """
2200
    node = self.node = self.cfg.GetNodeInfo(self.op.node_name)
2201

    
2202
    if ((self.op.master_candidate == False or self.op.offline == True)
2203
        and node.master_candidate):
2204
      # we will demote the node from master_candidate
2205
      if self.op.node_name == self.cfg.GetMasterNode():
2206
        raise errors.OpPrereqError("The master node has to be a"
2207
                                   " master candidate and online")
2208
      cp_size = self.cfg.GetClusterInfo().candidate_pool_size
2209
      num_candidates, _ = self.cfg.GetMasterCandidateStats()
2210
      if num_candidates <= cp_size:
2211
        msg = ("Not enough master candidates (desired"
2212
               " %d, new value will be %d)" % (cp_size, num_candidates-1))
2213
        if self.op.force:
2214
          self.LogWarning(msg)
2215
        else:
2216
          raise errors.OpPrereqError(msg)
2217

    
2218
    if (self.op.master_candidate == True and node.offline and
2219
        not self.op.offline == False):
2220
      raise errors.OpPrereqError("Can't set an offline node to"
2221
                                 " master_candidate")
2222

    
2223
    return
2224

    
2225
  def Exec(self, feedback_fn):
2226
    """Modifies a node.
2227

2228
    """
2229
    node = self.node
2230

    
2231
    result = []
2232

    
2233
    if self.op.offline is not None:
2234
      node.offline = self.op.offline
2235
      result.append(("offline", str(self.op.offline)))
2236
      if self.op.offline == True and node.master_candidate:
2237
        node.master_candidate = False
2238
        result.append(("master_candidate", "auto-demotion due to offline"))
2239

    
2240
    if self.op.master_candidate is not None:
2241
      node.master_candidate = self.op.master_candidate
2242
      result.append(("master_candidate", str(self.op.master_candidate)))
2243
      if self.op.master_candidate == False:
2244
        rrc = self.rpc.call_node_demote_from_mc(node.name)
2245
        if (rrc.failed or not isinstance(rrc.data, (tuple, list))
2246
            or len(rrc.data) != 2):
2247
          self.LogWarning("Node rpc error: %s" % rrc.error)
2248
        elif not rrc.data[0]:
2249
          self.LogWarning("Node failed to demote itself: %s" % rrc.data[1])
2250

    
2251
    # this will trigger configuration file update, if needed
2252
    self.cfg.Update(node)
2253
    # this will trigger job queue propagation or cleanup
2254
    if self.op.node_name != self.cfg.GetMasterNode():
2255
      self.context.ReaddNode(node)
2256

    
2257
    return result
2258

    
2259

    
2260
class LUQueryClusterInfo(NoHooksLU):
2261
  """Query cluster configuration.
2262

2263
  """
2264
  _OP_REQP = []
2265
  REQ_BGL = False
2266

    
2267
  def ExpandNames(self):
2268
    self.needed_locks = {}
2269

    
2270
  def CheckPrereq(self):
2271
    """No prerequsites needed for this LU.
2272

2273
    """
2274
    pass
2275

    
2276
  def Exec(self, feedback_fn):
2277
    """Return cluster config.
2278

2279
    """
2280
    cluster = self.cfg.GetClusterInfo()
2281
    result = {
2282
      "software_version": constants.RELEASE_VERSION,
2283
      "protocol_version": constants.PROTOCOL_VERSION,
2284
      "config_version": constants.CONFIG_VERSION,
2285
      "os_api_version": constants.OS_API_VERSION,
2286
      "export_version": constants.EXPORT_VERSION,
2287
      "architecture": (platform.architecture()[0], platform.machine()),
2288
      "name": cluster.cluster_name,
2289
      "master": cluster.master_node,
2290
      "default_hypervisor": cluster.default_hypervisor,
2291
      "enabled_hypervisors": cluster.enabled_hypervisors,
2292
      "hvparams": cluster.hvparams,
2293
      "beparams": cluster.beparams,
2294
      "candidate_pool_size": cluster.candidate_pool_size,
2295
      }
2296

    
2297
    return result
2298

    
2299

    
2300
class LUQueryConfigValues(NoHooksLU):
2301
  """Return configuration values.
2302

2303
  """
2304
  _OP_REQP = []
2305
  REQ_BGL = False
2306
  _FIELDS_DYNAMIC = utils.FieldSet()
2307
  _FIELDS_STATIC = utils.FieldSet("cluster_name", "master_node", "drain_flag")
2308

    
2309
  def ExpandNames(self):
2310
    self.needed_locks = {}
2311

    
2312
    _CheckOutputFields(static=self._FIELDS_STATIC,
2313
                       dynamic=self._FIELDS_DYNAMIC,
2314
                       selected=self.op.output_fields)
2315

    
2316
  def CheckPrereq(self):
2317
    """No prerequisites.
2318

2319
    """
2320
    pass
2321

    
2322
  def Exec(self, feedback_fn):
2323
    """Dump a representation of the cluster config to the standard output.
2324

2325
    """
2326
    values = []
2327
    for field in self.op.output_fields:
2328
      if field == "cluster_name":
2329
        entry = self.cfg.GetClusterName()
2330
      elif field == "master_node":
2331
        entry = self.cfg.GetMasterNode()
2332
      elif field == "drain_flag":
2333
        entry = os.path.exists(constants.JOB_QUEUE_DRAIN_FILE)
2334
      else:
2335
        raise errors.ParameterError(field)
2336
      values.append(entry)
2337
    return values
2338

    
2339

    
2340
class LUActivateInstanceDisks(NoHooksLU):
2341
  """Bring up an instance's disks.
2342

2343
  """
2344
  _OP_REQP = ["instance_name"]
2345
  REQ_BGL = False
2346

    
2347
  def ExpandNames(self):
2348
    self._ExpandAndLockInstance()
2349
    self.needed_locks[locking.LEVEL_NODE] = []
2350
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
2351

    
2352
  def DeclareLocks(self, level):
2353
    if level == locking.LEVEL_NODE:
2354
      self._LockInstancesNodes()
2355

    
2356
  def CheckPrereq(self):
2357
    """Check prerequisites.
2358

2359
    This checks that the instance is in the cluster.
2360

2361
    """
2362
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_name)
2363
    assert self.instance is not None, \
2364
      "Cannot retrieve locked instance %s" % self.op.instance_name
2365
    _CheckNodeOnline(self, self.instance.primary_node)
2366

    
2367
  def Exec(self, feedback_fn):
2368
    """Activate the disks.
2369

2370
    """
2371
    disks_ok, disks_info = _AssembleInstanceDisks(self, self.instance)
2372
    if not disks_ok:
2373
      raise errors.OpExecError("Cannot activate block devices")
2374

    
2375
    return disks_info
2376

    
2377

    
2378
def _AssembleInstanceDisks(lu, instance, ignore_secondaries=False):
2379
  """Prepare the block devices for an instance.
2380

2381
  This sets up the block devices on all nodes.
2382

2383
  @type lu: L{LogicalUnit}
2384
  @param lu: the logical unit on whose behalf we execute
2385
  @type instance: L{objects.Instance}
2386
  @param instance: the instance for whose disks we assemble
2387
  @type ignore_secondaries: boolean
2388
  @param ignore_secondaries: if true, errors on secondary nodes
2389
      won't result in an error return from the function
2390
  @return: False if the operation failed, otherwise a list of
2391
      (host, instance_visible_name, node_visible_name)
2392
      with the mapping from node devices to instance devices
2393

2394
  """
2395
  device_info = []
2396
  disks_ok = True
2397
  iname = instance.name
2398
  # With the two passes mechanism we try to reduce the window of
2399
  # opportunity for the race condition of switching DRBD to primary
2400
  # before handshaking occured, but we do not eliminate it
2401

    
2402
  # The proper fix would be to wait (with some limits) until the
2403
  # connection has been made and drbd transitions from WFConnection
2404
  # into any other network-connected state (Connected, SyncTarget,
2405
  # SyncSource, etc.)
2406

    
2407
  # 1st pass, assemble on all nodes in secondary mode
2408
  for inst_disk in instance.disks:
2409
    for node, node_disk in inst_disk.ComputeNodeTree(instance.primary_node):
2410
      lu.cfg.SetDiskID(node_disk, node)
2411
      result = lu.rpc.call_blockdev_assemble(node, node_disk, iname, False)
2412
      if result.failed or not result:
2413
        lu.proc.LogWarning("Could not prepare block device %s on node %s"
2414
                           " (is_primary=False, pass=1)",
2415
                           inst_disk.iv_name, node)
2416
        if not ignore_secondaries:
2417
          disks_ok = False
2418

    
2419
  # FIXME: race condition on drbd migration to primary
2420

    
2421
  # 2nd pass, do only the primary node
2422
  for inst_disk in instance.disks:
2423
    for node, node_disk in inst_disk.ComputeNodeTree(instance.primary_node):
2424
      if node != instance.primary_node:
2425
        continue
2426
      lu.cfg.SetDiskID(node_disk, node)
2427
      result = lu.rpc.call_blockdev_assemble(node, node_disk, iname, True)
2428
      if result.failed or not result:
2429
        lu.proc.LogWarning("Could not prepare block device %s on node %s"
2430
                           " (is_primary=True, pass=2)",
2431
                           inst_disk.iv_name, node)
2432
        disks_ok = False
2433
    device_info.append((instance.primary_node, inst_disk.iv_name, result.data))
2434

    
2435
  # leave the disks configured for the primary node
2436
  # this is a workaround that would be fixed better by
2437
  # improving the logical/physical id handling
2438
  for disk in instance.disks:
2439
    lu.cfg.SetDiskID(disk, instance.primary_node)
2440

    
2441
  return disks_ok, device_info
2442

    
2443

    
2444
def _StartInstanceDisks(lu, instance, force):
2445
  """Start the disks of an instance.
2446

2447
  """
2448
  disks_ok, dummy = _AssembleInstanceDisks(lu, instance,
2449
                                           ignore_secondaries=force)
2450
  if not disks_ok:
2451
    _ShutdownInstanceDisks(lu, instance)
2452
    if force is not None and not force:
2453
      lu.proc.LogWarning("", hint="If the message above refers to a"
2454
                         " secondary node,"
2455
                         " you can retry the operation using '--force'.")
2456
    raise errors.OpExecError("Disk consistency error")
2457

    
2458

    
2459
class LUDeactivateInstanceDisks(NoHooksLU):
2460
  """Shutdown an instance's disks.
2461

2462
  """
2463
  _OP_REQP = ["instance_name"]
2464
  REQ_BGL = False
2465

    
2466
  def ExpandNames(self):
2467
    self._ExpandAndLockInstance()
2468
    self.needed_locks[locking.LEVEL_NODE] = []
2469
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
2470

    
2471
  def DeclareLocks(self, level):
2472
    if level == locking.LEVEL_NODE:
2473
      self._LockInstancesNodes()
2474

    
2475
  def CheckPrereq(self):
2476
    """Check prerequisites.
2477

2478
    This checks that the instance is in the cluster.
2479

2480
    """
2481
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_name)
2482
    assert self.instance is not None, \
2483
      "Cannot retrieve locked instance %s" % self.op.instance_name
2484

    
2485
  def Exec(self, feedback_fn):
2486
    """Deactivate the disks
2487

2488
    """
2489
    instance = self.instance
2490
    _SafeShutdownInstanceDisks(self, instance)
2491

    
2492

    
2493
def _SafeShutdownInstanceDisks(lu, instance):
2494
  """Shutdown block devices of an instance.
2495

2496
  This function checks if an instance is running, before calling
2497
  _ShutdownInstanceDisks.
2498

2499
  """
2500
  ins_l = lu.rpc.call_instance_list([instance.primary_node],
2501
                                      [instance.hypervisor])
2502
  ins_l = ins_l[instance.primary_node]
2503
  if ins_l.failed or not isinstance(ins_l.data, list):
2504
    raise errors.OpExecError("Can't contact node '%s'" %
2505
                             instance.primary_node)
2506

    
2507
  if instance.name in ins_l.data:
2508
    raise errors.OpExecError("Instance is running, can't shutdown"
2509
                             " block devices.")
2510

    
2511
  _ShutdownInstanceDisks(lu, instance)
2512

    
2513

    
2514
def _ShutdownInstanceDisks(lu, instance, ignore_primary=False):
2515
  """Shutdown block devices of an instance.
2516

2517
  This does the shutdown on all nodes of the instance.
2518

2519
  If the ignore_primary is false, errors on the primary node are
2520
  ignored.
2521

2522
  """
2523
  result = True
2524
  for disk in instance.disks:
2525
    for node, top_disk in disk.ComputeNodeTree(instance.primary_node):
2526
      lu.cfg.SetDiskID(top_disk, node)
2527
      result = lu.rpc.call_blockdev_shutdown(node, top_disk)
2528
      if result.failed or not result.data:
2529
        logging.error("Could not shutdown block device %s on node %s",
2530
                      disk.iv_name, node)
2531
        if not ignore_primary or node != instance.primary_node:
2532
          result = False
2533
  return result
2534

    
2535

    
2536
def _CheckNodeFreeMemory(lu, node, reason, requested, hypervisor_name):
2537
  """Checks if a node has enough free memory.
2538

2539
  This function check if a given node has the needed amount of free
2540
  memory. In case the node has less memory or we cannot get the
2541
  information from the node, this function raise an OpPrereqError
2542
  exception.
2543

2544
  @type lu: C{LogicalUnit}
2545
  @param lu: a logical unit from which we get configuration data
2546
  @type node: C{str}
2547
  @param node: the node to check
2548
  @type reason: C{str}
2549
  @param reason: string to use in the error message
2550
  @type requested: C{int}
2551
  @param requested: the amount of memory in MiB to check for
2552
  @type hypervisor_name: C{str}
2553
  @param hypervisor_name: the hypervisor to ask for memory stats
2554
  @raise errors.OpPrereqError: if the node doesn't have enough memory, or
2555
      we cannot check the node
2556

2557
  """
2558
  nodeinfo = lu.rpc.call_node_info([node], lu.cfg.GetVGName(), hypervisor_name)
2559
  nodeinfo[node].Raise()
2560
  free_mem = nodeinfo[node].data.get('memory_free')
2561
  if not isinstance(free_mem, int):
2562
    raise errors.OpPrereqError("Can't compute free memory on node %s, result"
2563
                             " was '%s'" % (node, free_mem))
2564
  if requested > free_mem:
2565
    raise errors.OpPrereqError("Not enough memory on node %s for %s:"
2566
                             " needed %s MiB, available %s MiB" %
2567
                             (node, reason, requested, free_mem))
2568

    
2569

    
2570
class LUStartupInstance(LogicalUnit):
2571
  """Starts an instance.
2572

2573
  """
2574
  HPATH = "instance-start"
2575
  HTYPE = constants.HTYPE_INSTANCE
2576
  _OP_REQP = ["instance_name", "force"]
2577
  REQ_BGL = False
2578

    
2579
  def ExpandNames(self):
2580
    self._ExpandAndLockInstance()
2581

    
2582
  def BuildHooksEnv(self):
2583
    """Build hooks env.
2584

2585
    This runs on master, primary and secondary nodes of the instance.
2586

2587
    """
2588
    env = {
2589
      "FORCE": self.op.force,
2590
      }
2591
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
2592
    nl = [self.cfg.GetMasterNode()] + list(self.instance.all_nodes)
2593
    return env, nl, nl
2594

    
2595
  def CheckPrereq(self):
2596
    """Check prerequisites.
2597

2598
    This checks that the instance is in the cluster.
2599

2600
    """
2601
    self.instance = instance = self.cfg.GetInstanceInfo(self.op.instance_name)
2602
    assert self.instance is not None, \
2603
      "Cannot retrieve locked instance %s" % self.op.instance_name
2604

    
2605
    _CheckNodeOnline(self, instance.primary_node)
2606

    
2607
    bep = self.cfg.GetClusterInfo().FillBE(instance)
2608
    # check bridges existance
2609
    _CheckInstanceBridgesExist(self, instance)
2610

    
2611
    _CheckNodeFreeMemory(self, instance.primary_node,
2612
                         "starting instance %s" % instance.name,
2613
                         bep[constants.BE_MEMORY], instance.hypervisor)
2614

    
2615
  def Exec(self, feedback_fn):
2616
    """Start the instance.
2617

2618
    """
2619
    instance = self.instance
2620
    force = self.op.force
2621
    extra_args = getattr(self.op, "extra_args", "")
2622

    
2623
    self.cfg.MarkInstanceUp(instance.name)
2624

    
2625
    node_current = instance.primary_node
2626

    
2627
    _StartInstanceDisks(self, instance, force)
2628

    
2629
    result = self.rpc.call_instance_start(node_current, instance, extra_args)
2630
    if result.failed or not result.data:
2631
      _ShutdownInstanceDisks(self, instance)
2632
      raise errors.OpExecError("Could not start instance")
2633

    
2634

    
2635
class LURebootInstance(LogicalUnit):
2636
  """Reboot an instance.
2637

2638
  """
2639
  HPATH = "instance-reboot"
2640
  HTYPE = constants.HTYPE_INSTANCE
2641
  _OP_REQP = ["instance_name", "ignore_secondaries", "reboot_type"]
2642
  REQ_BGL = False
2643

    
2644
  def ExpandNames(self):
2645
    if self.op.reboot_type not in [constants.INSTANCE_REBOOT_SOFT,
2646
                                   constants.INSTANCE_REBOOT_HARD,
2647
                                   constants.INSTANCE_REBOOT_FULL]:
2648
      raise errors.ParameterError("reboot type not in [%s, %s, %s]" %
2649
                                  (constants.INSTANCE_REBOOT_SOFT,
2650
                                   constants.INSTANCE_REBOOT_HARD,
2651
                                   constants.INSTANCE_REBOOT_FULL))
2652
    self._ExpandAndLockInstance()
2653

    
2654
  def BuildHooksEnv(self):
2655
    """Build hooks env.
2656

2657
    This runs on master, primary and secondary nodes of the instance.
2658

2659
    """
2660
    env = {
2661
      "IGNORE_SECONDARIES": self.op.ignore_secondaries,
2662
      }
2663
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
2664
    nl = [self.cfg.GetMasterNode()] + list(self.instance.all_nodes)
2665
    return env, nl, nl
2666

    
2667
  def CheckPrereq(self):
2668
    """Check prerequisites.
2669

2670
    This checks that the instance is in the cluster.
2671

2672
    """
2673
    self.instance = instance = self.cfg.GetInstanceInfo(self.op.instance_name)
2674
    assert self.instance is not None, \
2675
      "Cannot retrieve locked instance %s" % self.op.instance_name
2676

    
2677
    _CheckNodeOnline(self, instance.primary_node)
2678

    
2679
    # check bridges existance
2680
    _CheckInstanceBridgesExist(self, instance)
2681

    
2682
  def Exec(self, feedback_fn):
2683
    """Reboot the instance.
2684

2685
    """
2686
    instance = self.instance
2687
    ignore_secondaries = self.op.ignore_secondaries
2688
    reboot_type = self.op.reboot_type
2689
    extra_args = getattr(self.op, "extra_args", "")
2690

    
2691
    node_current = instance.primary_node
2692

    
2693
    if reboot_type in [constants.INSTANCE_REBOOT_SOFT,
2694
                       constants.INSTANCE_REBOOT_HARD]:
2695
      result = self.rpc.call_instance_reboot(node_current, instance,
2696
                                             reboot_type, extra_args)
2697
      if result.failed or not result.data:
2698
        raise errors.OpExecError("Could not reboot instance")
2699
    else:
2700
      if not self.rpc.call_instance_shutdown(node_current, instance):
2701
        raise errors.OpExecError("could not shutdown instance for full reboot")
2702
      _ShutdownInstanceDisks(self, instance)
2703
      _StartInstanceDisks(self, instance, ignore_secondaries)
2704
      result = self.rpc.call_instance_start(node_current, instance, extra_args)
2705
      if result.failed or not result.data:
2706
        _ShutdownInstanceDisks(self, instance)
2707
        raise errors.OpExecError("Could not start instance for full reboot")
2708

    
2709
    self.cfg.MarkInstanceUp(instance.name)
2710

    
2711

    
2712
class LUShutdownInstance(LogicalUnit):
2713
  """Shutdown an instance.
2714

2715
  """
2716
  HPATH = "instance-stop"
2717
  HTYPE = constants.HTYPE_INSTANCE
2718
  _OP_REQP = ["instance_name"]
2719
  REQ_BGL = False
2720

    
2721
  def ExpandNames(self):
2722
    self._ExpandAndLockInstance()
2723

    
2724
  def BuildHooksEnv(self):
2725
    """Build hooks env.
2726

2727
    This runs on master, primary and secondary nodes of the instance.
2728

2729
    """
2730
    env = _BuildInstanceHookEnvByObject(self, self.instance)
2731
    nl = [self.cfg.GetMasterNode()] + list(self.instance.all_nodes)
2732
    return env, nl, nl
2733

    
2734
  def CheckPrereq(self):
2735
    """Check prerequisites.
2736

2737
    This checks that the instance is in the cluster.
2738

2739
    """
2740
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_name)
2741
    assert self.instance is not None, \
2742
      "Cannot retrieve locked instance %s" % self.op.instance_name
2743
    _CheckNodeOnline(self, self.instance.primary_node)
2744

    
2745
  def Exec(self, feedback_fn):
2746
    """Shutdown the instance.
2747

2748
    """
2749
    instance = self.instance
2750
    node_current = instance.primary_node
2751
    self.cfg.MarkInstanceDown(instance.name)
2752
    result = self.rpc.call_instance_shutdown(node_current, instance)
2753
    if result.failed or not result.data:
2754
      self.proc.LogWarning("Could not shutdown instance")
2755

    
2756
    _ShutdownInstanceDisks(self, instance)
2757

    
2758

    
2759
class LUReinstallInstance(LogicalUnit):
2760
  """Reinstall an instance.
2761

2762
  """
2763
  HPATH = "instance-reinstall"
2764
  HTYPE = constants.HTYPE_INSTANCE
2765
  _OP_REQP = ["instance_name"]
2766
  REQ_BGL = False
2767

    
2768
  def ExpandNames(self):
2769
    self._ExpandAndLockInstance()
2770

    
2771
  def BuildHooksEnv(self):
2772
    """Build hooks env.
2773

2774
    This runs on master, primary and secondary nodes of the instance.
2775

2776
    """
2777
    env = _BuildInstanceHookEnvByObject(self, self.instance)
2778
    nl = [self.cfg.GetMasterNode()] + list(self.instance.all_nodes)
2779
    return env, nl, nl
2780

    
2781
  def CheckPrereq(self):
2782
    """Check prerequisites.
2783

2784
    This checks that the instance is in the cluster and is not running.
2785

2786
    """
2787
    instance = self.cfg.GetInstanceInfo(self.op.instance_name)
2788
    assert instance is not None, \
2789
      "Cannot retrieve locked instance %s" % self.op.instance_name
2790
    _CheckNodeOnline(self, instance.primary_node)
2791

    
2792
    if instance.disk_template == constants.DT_DISKLESS:
2793
      raise errors.OpPrereqError("Instance '%s' has no disks" %
2794
                                 self.op.instance_name)
2795
    if instance.status != "down":
2796
      raise errors.OpPrereqError("Instance '%s' is marked to be up" %
2797
                                 self.op.instance_name)
2798
    remote_info = self.rpc.call_instance_info(instance.primary_node,
2799
                                              instance.name,
2800
                                              instance.hypervisor)
2801
    if remote_info.failed or remote_info.data:
2802
      raise errors.OpPrereqError("Instance '%s' is running on the node %s" %
2803
                                 (self.op.instance_name,
2804
                                  instance.primary_node))
2805

    
2806
    self.op.os_type = getattr(self.op, "os_type", None)
2807
    if self.op.os_type is not None:
2808
      # OS verification
2809
      pnode = self.cfg.GetNodeInfo(
2810
        self.cfg.ExpandNodeName(instance.primary_node))
2811
      if pnode is None:
2812
        raise errors.OpPrereqError("Primary node '%s' is unknown" %
2813
                                   self.op.pnode)
2814
      result = self.rpc.call_os_get(pnode.name, self.op.os_type)
2815
      result.Raise()
2816
      if not isinstance(result.data, objects.OS):
2817
        raise errors.OpPrereqError("OS '%s' not in supported OS list for"
2818
                                   " primary node"  % self.op.os_type)
2819

    
2820
    self.instance = instance
2821

    
2822
  def Exec(self, feedback_fn):
2823
    """Reinstall the instance.
2824

2825
    """
2826
    inst = self.instance
2827

    
2828
    if self.op.os_type is not None:
2829
      feedback_fn("Changing OS to '%s'..." % self.op.os_type)
2830
      inst.os = self.op.os_type
2831
      self.cfg.Update(inst)
2832

    
2833
    _StartInstanceDisks(self, inst, None)
2834
    try:
2835
      feedback_fn("Running the instance OS create scripts...")
2836
      result = self.rpc.call_instance_os_add(inst.primary_node, inst)
2837
      result.Raise()
2838
      if not result.data:
2839
        raise errors.OpExecError("Could not install OS for instance %s"
2840
                                 " on node %s" %
2841
                                 (inst.name, inst.primary_node))
2842
    finally:
2843
      _ShutdownInstanceDisks(self, inst)
2844

    
2845

    
2846
class LURenameInstance(LogicalUnit):
2847
  """Rename an instance.
2848

2849
  """
2850
  HPATH = "instance-rename"
2851
  HTYPE = constants.HTYPE_INSTANCE
2852
  _OP_REQP = ["instance_name", "new_name"]
2853

    
2854
  def BuildHooksEnv(self):
2855
    """Build hooks env.
2856

2857
    This runs on master, primary and secondary nodes of the instance.
2858

2859
    """
2860
    env = _BuildInstanceHookEnvByObject(self, self.instance)
2861
    env["INSTANCE_NEW_NAME"] = self.op.new_name
2862
    nl = [self.cfg.GetMasterNode()] + list(self.instance.all_nodes)
2863
    return env, nl, nl
2864

    
2865
  def CheckPrereq(self):
2866
    """Check prerequisites.
2867

2868
    This checks that the instance is in the cluster and is not running.
2869

2870
    """
2871
    instance = self.cfg.GetInstanceInfo(
2872
      self.cfg.ExpandInstanceName(self.op.instance_name))
2873
    if instance is None:
2874
      raise errors.OpPrereqError("Instance '%s' not known" %
2875
                                 self.op.instance_name)
2876
    _CheckNodeOnline(self, instance.primary_node)
2877

    
2878
    if instance.status != "down":
2879
      raise errors.OpPrereqError("Instance '%s' is marked to be up" %
2880
                                 self.op.instance_name)
2881
    remote_info = self.rpc.call_instance_info(instance.primary_node,
2882
                                              instance.name,
2883
                                              instance.hypervisor)
2884
    remote_info.Raise()
2885
    if remote_info.data:
2886
      raise errors.OpPrereqError("Instance '%s' is running on the node %s" %
2887
                                 (self.op.instance_name,
2888
                                  instance.primary_node))
2889
    self.instance = instance
2890

    
2891
    # new name verification
2892
    name_info = utils.HostInfo(self.op.new_name)
2893

    
2894
    self.op.new_name = new_name = name_info.name
2895
    instance_list = self.cfg.GetInstanceList()
2896
    if new_name in instance_list:
2897
      raise errors.OpPrereqError("Instance '%s' is already in the cluster" %
2898
                                 new_name)
2899

    
2900
    if not getattr(self.op, "ignore_ip", False):
2901
      if utils.TcpPing(name_info.ip, constants.DEFAULT_NODED_PORT):
2902
        raise errors.OpPrereqError("IP %s of instance %s already in use" %
2903
                                   (name_info.ip, new_name))
2904

    
2905

    
2906
  def Exec(self, feedback_fn):
2907
    """Reinstall the instance.
2908

2909
    """
2910
    inst = self.instance
2911
    old_name = inst.name
2912

    
2913
    if inst.disk_template == constants.DT_FILE:
2914
      old_file_storage_dir = os.path.dirname(inst.disks[0].logical_id[1])
2915

    
2916
    self.cfg.RenameInstance(inst.name, self.op.new_name)
2917
    # Change the instance lock. This is definitely safe while we hold the BGL
2918
    self.context.glm.remove(locking.LEVEL_INSTANCE, old_name)
2919
    self.context.glm.add(locking.LEVEL_INSTANCE, self.op.new_name)
2920

    
2921
    # re-read the instance from the configuration after rename
2922
    inst = self.cfg.GetInstanceInfo(self.op.new_name)
2923

    
2924
    if inst.disk_template == constants.DT_FILE:
2925
      new_file_storage_dir = os.path.dirname(inst.disks[0].logical_id[1])
2926
      result = self.rpc.call_file_storage_dir_rename(inst.primary_node,
2927
                                                     old_file_storage_dir,
2928
                                                     new_file_storage_dir)
2929
      result.Raise()
2930
      if not result.data:
2931
        raise errors.OpExecError("Could not connect to node '%s' to rename"
2932
                                 " directory '%s' to '%s' (but the instance"
2933
                                 " has been renamed in Ganeti)" % (
2934
                                 inst.primary_node, old_file_storage_dir,
2935
                                 new_file_storage_dir))
2936

    
2937
      if not result.data[0]:
2938
        raise errors.OpExecError("Could not rename directory '%s' to '%s'"
2939
                                 " (but the instance has been renamed in"
2940
                                 " Ganeti)" % (old_file_storage_dir,
2941
                                               new_file_storage_dir))
2942

    
2943
    _StartInstanceDisks(self, inst, None)
2944
    try:
2945
      result = self.rpc.call_instance_run_rename(inst.primary_node, inst,
2946
                                                 old_name)
2947
      if result.failed or not result.data:
2948
        msg = ("Could not run OS rename script for instance %s on node %s"
2949
               " (but the instance has been renamed in Ganeti)" %
2950
               (inst.name, inst.primary_node))
2951
        self.proc.LogWarning(msg)
2952
    finally:
2953
      _ShutdownInstanceDisks(self, inst)
2954

    
2955

    
2956
class LURemoveInstance(LogicalUnit):
2957
  """Remove an instance.
2958

2959
  """
2960
  HPATH = "instance-remove"
2961
  HTYPE = constants.HTYPE_INSTANCE
2962
  _OP_REQP = ["instance_name", "ignore_failures"]
2963
  REQ_BGL = False
2964

    
2965
  def ExpandNames(self):
2966
    self._ExpandAndLockInstance()
2967
    self.needed_locks[locking.LEVEL_NODE] = []
2968
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
2969

    
2970
  def DeclareLocks(self, level):
2971
    if level == locking.LEVEL_NODE:
2972
      self._LockInstancesNodes()
2973

    
2974
  def BuildHooksEnv(self):
2975
    """Build hooks env.
2976

2977
    This runs on master, primary and secondary nodes of the instance.
2978

2979
    """
2980
    env = _BuildInstanceHookEnvByObject(self, self.instance)
2981
    nl = [self.cfg.GetMasterNode()]
2982
    return env, nl, nl
2983

    
2984
  def CheckPrereq(self):
2985
    """Check prerequisites.
2986

2987
    This checks that the instance is in the cluster.
2988

2989
    """
2990
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_name)
2991
    assert self.instance is not None, \
2992
      "Cannot retrieve locked instance %s" % self.op.instance_name
2993

    
2994
  def Exec(self, feedback_fn):
2995
    """Remove the instance.
2996

2997
    """
2998
    instance = self.instance
2999
    logging.info("Shutting down instance %s on node %s",
3000
                 instance.name, instance.primary_node)
3001

    
3002
    result = self.rpc.call_instance_shutdown(instance.primary_node, instance)
3003
    if result.failed or not result.data:
3004
      if self.op.ignore_failures:
3005
        feedback_fn("Warning: can't shutdown instance")
3006
      else:
3007
        raise errors.OpExecError("Could not shutdown instance %s on node %s" %
3008
                                 (instance.name, instance.primary_node))
3009

    
3010
    logging.info("Removing block devices for instance %s", instance.name)
3011

    
3012
    if not _RemoveDisks(self, instance):
3013
      if self.op.ignore_failures:
3014
        feedback_fn("Warning: can't remove instance's disks")
3015
      else:
3016
        raise errors.OpExecError("Can't remove instance's disks")
3017

    
3018
    logging.info("Removing instance %s out of cluster config", instance.name)
3019

    
3020
    self.cfg.RemoveInstance(instance.name)
3021
    self.remove_locks[locking.LEVEL_INSTANCE] = instance.name
3022

    
3023

    
3024
class LUQueryInstances(NoHooksLU):
3025
  """Logical unit for querying instances.
3026

3027
  """
3028
  _OP_REQP = ["output_fields", "names"]
3029
  REQ_BGL = False
3030
  _FIELDS_STATIC = utils.FieldSet(*["name", "os", "pnode", "snodes",
3031
                                    "admin_state", "admin_ram",
3032
                                    "disk_template", "ip", "mac", "bridge",
3033
                                    "sda_size", "sdb_size", "vcpus", "tags",
3034
                                    "network_port", "beparams",
3035
                                    "(disk).(size)/([0-9]+)",
3036
                                    "(disk).(sizes)",
3037
                                    "(nic).(mac|ip|bridge)/([0-9]+)",
3038
                                    "(nic).(macs|ips|bridges)",
3039
                                    "(disk|nic).(count)",
3040
                                    "serial_no", "hypervisor", "hvparams",] +
3041
                                  ["hv/%s" % name
3042
                                   for name in constants.HVS_PARAMETERS] +
3043
                                  ["be/%s" % name
3044
                                   for name in constants.BES_PARAMETERS])
3045
  _FIELDS_DYNAMIC = utils.FieldSet("oper_state", "oper_ram", "status")
3046

    
3047

    
3048
  def ExpandNames(self):
3049
    _CheckOutputFields(static=self._FIELDS_STATIC,
3050
                       dynamic=self._FIELDS_DYNAMIC,
3051
                       selected=self.op.output_fields)
3052

    
3053
    self.needed_locks = {}
3054
    self.share_locks[locking.LEVEL_INSTANCE] = 1
3055
    self.share_locks[locking.LEVEL_NODE] = 1
3056

    
3057
    if self.op.names:
3058
      self.wanted = _GetWantedInstances(self, self.op.names)
3059
    else:
3060
      self.wanted = locking.ALL_SET
3061

    
3062
    self.do_locking = self._FIELDS_STATIC.NonMatching(self.op.output_fields)
3063
    if self.do_locking:
3064
      self.needed_locks[locking.LEVEL_INSTANCE] = self.wanted
3065
      self.needed_locks[locking.LEVEL_NODE] = []
3066
      self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
3067

    
3068
  def DeclareLocks(self, level):
3069
    if level == locking.LEVEL_NODE and self.do_locking:
3070
      self._LockInstancesNodes()
3071

    
3072
  def CheckPrereq(self):
3073
    """Check prerequisites.
3074

3075
    """
3076
    pass
3077

    
3078
  def Exec(self, feedback_fn):
3079
    """Computes the list of nodes and their attributes.
3080

3081
    """
3082
    all_info = self.cfg.GetAllInstancesInfo()
3083
    if self.do_locking:
3084
      instance_names = self.acquired_locks[locking.LEVEL_INSTANCE]
3085
    elif self.wanted != locking.ALL_SET:
3086
      instance_names = self.wanted
3087
      missing = set(instance_names).difference(all_info.keys())
3088
      if missing:
3089
        raise errors.OpExecError(
3090
          "Some instances were removed before retrieving their data: %s"
3091
          % missing)
3092
    else:
3093
      instance_names = all_info.keys()
3094

    
3095
    instance_names = utils.NiceSort(instance_names)
3096
    instance_list = [all_info[iname] for iname in instance_names]
3097

    
3098
    # begin data gathering
3099

    
3100
    nodes = frozenset([inst.primary_node for inst in instance_list])
3101
    hv_list = list(set([inst.hypervisor for inst in instance_list]))
3102

    
3103
    bad_nodes = []
3104
    off_nodes = []
3105
    if self.do_locking:
3106
      live_data = {}
3107
      node_data = self.rpc.call_all_instances_info(nodes, hv_list)
3108
      for name in nodes:
3109
        result = node_data[name]
3110
        if result.offline:
3111
          # offline nodes will be in both lists
3112
          off_nodes.append(name)
3113
        if result.failed:
3114
          bad_nodes.append(name)
3115
        else:
3116
          if result.data:
3117
            live_data.update(result.data)
3118
            # else no instance is alive
3119
    else:
3120
      live_data = dict([(name, {}) for name in instance_names])
3121

    
3122
    # end data gathering
3123

    
3124
    HVPREFIX = "hv/"
3125
    BEPREFIX = "be/"
3126
    output = []
3127
    for instance in instance_list:
3128
      iout = []
3129
      i_hv = self.cfg.GetClusterInfo().FillHV(instance)
3130
      i_be = self.cfg.GetClusterInfo().FillBE(instance)
3131
      for field in self.op.output_fields:
3132
        st_match = self._FIELDS_STATIC.Matches(field)
3133
        if field == "name":
3134
          val = instance.name
3135
        elif field == "os":
3136
          val = instance.os
3137
        elif field == "pnode":
3138
          val = instance.primary_node
3139
        elif field == "snodes":
3140
          val = list(instance.secondary_nodes)
3141
        elif field == "admin_state":
3142
          val = (instance.status != "down")
3143
        elif field == "oper_state":
3144
          if instance.primary_node in bad_nodes:
3145
            val = None
3146
          else:
3147
            val = bool(live_data.get(instance.name))
3148
        elif field == "status":
3149
          if instance.primary_node in off_nodes:
3150
            val = "ERROR_nodeoffline"
3151
          elif instance.primary_node in bad_nodes:
3152
            val = "ERROR_nodedown"
3153
          else:
3154
            running = bool(live_data.get(instance.name))
3155
            if running:
3156
              if instance.status != "down":
3157
                val = "running"
3158
              else:
3159
                val = "ERROR_up"
3160
            else:
3161
              if instance.status != "down":
3162
                val = "ERROR_down"
3163
              else:
3164
                val = "ADMIN_down"
3165
        elif field == "oper_ram":
3166
          if instance.primary_node in bad_nodes:
3167
            val = None
3168
          elif instance.name in live_data:
3169
            val = live_data[instance.name].get("memory", "?")
3170
          else:
3171
            val = "-"
3172
        elif field == "disk_template":
3173
          val = instance.disk_template
3174
        elif field == "ip":
3175
          val = instance.nics[0].ip
3176
        elif field == "bridge":
3177
          val = instance.nics[0].bridge
3178
        elif field == "mac":
3179
          val = instance.nics[0].mac
3180
        elif field == "sda_size" or field == "sdb_size":
3181
          idx = ord(field[2]) - ord('a')
3182
          try:
3183
            val = instance.FindDisk(idx).size
3184
          except errors.OpPrereqError:
3185
            val = None
3186
        elif field == "tags":
3187
          val = list(instance.GetTags())
3188
        elif field == "serial_no":
3189
          val = instance.serial_no
3190
        elif field == "network_port":
3191
          val = instance.network_port
3192
        elif field == "hypervisor":
3193
          val = instance.hypervisor
3194
        elif field == "hvparams":
3195
          val = i_hv
3196
        elif (field.startswith(HVPREFIX) and
3197
              field[len(HVPREFIX):] in constants.HVS_PARAMETERS):
3198
          val = i_hv.get(field[len(HVPREFIX):], None)
3199
        elif field == "beparams":
3200
          val = i_be
3201
        elif (field.startswith(BEPREFIX) and
3202
              field[len(BEPREFIX):] in constants.BES_PARAMETERS):
3203
          val = i_be.get(field[len(BEPREFIX):], None)
3204
        elif st_match and st_match.groups():
3205
          # matches a variable list
3206
          st_groups = st_match.groups()
3207
          if st_groups and st_groups[0] == "disk":
3208
            if st_groups[1] == "count":
3209
              val = len(instance.disks)
3210
            elif st_groups[1] == "sizes":
3211
              val = [disk.size for disk in instance.disks]
3212
            elif st_groups[1] == "size":
3213
              try:
3214
                val = instance.FindDisk(st_groups[2]).size
3215
              except errors.OpPrereqError:
3216
                val = None
3217
            else:
3218
              assert False, "Unhandled disk parameter"
3219
          elif st_groups[0] == "nic":
3220
            if st_groups[1] == "count":
3221
              val = len(instance.nics)
3222
            elif st_groups[1] == "macs":
3223
              val = [nic.mac for nic in instance.nics]
3224
            elif st_groups[1] == "ips":
3225
              val = [nic.ip for nic in instance.nics]
3226
            elif st_groups[1] == "bridges":
3227
              val = [nic.bridge for nic in instance.nics]
3228
            else:
3229
              # index-based item
3230
              nic_idx = int(st_groups[2])
3231
              if nic_idx >= len(instance.nics):
3232
                val = None
3233
              else:
3234
                if st_groups[1] == "mac":
3235
                  val = instance.nics[nic_idx].mac
3236
                elif st_groups[1] == "ip":
3237
                  val = instance.nics[nic_idx].ip
3238
                elif st_groups[1] == "bridge":
3239
                  val = instance.nics[nic_idx].bridge
3240
                else:
3241
                  assert False, "Unhandled NIC parameter"
3242
          else:
3243
            assert False, "Unhandled variable parameter"
3244
        else:
3245
          raise errors.ParameterError(field)
3246
        iout.append(val)
3247
      output.append(iout)
3248

    
3249
    return output
3250

    
3251

    
3252
class LUFailoverInstance(LogicalUnit):
3253
  """Failover an instance.
3254

3255
  """
3256
  HPATH = "instance-failover"
3257
  HTYPE = constants.HTYPE_INSTANCE
3258
  _OP_REQP = ["instance_name", "ignore_consistency"]
3259
  REQ_BGL = False
3260

    
3261
  def ExpandNames(self):
3262
    self._ExpandAndLockInstance()
3263
    self.needed_locks[locking.LEVEL_NODE] = []
3264
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
3265

    
3266
  def DeclareLocks(self, level):
3267
    if level == locking.LEVEL_NODE:
3268
      self._LockInstancesNodes()
3269

    
3270
  def BuildHooksEnv(self):
3271
    """Build hooks env.
3272

3273
    This runs on master, primary and secondary nodes of the instance.
3274

3275
    """
3276
    env = {
3277
      "IGNORE_CONSISTENCY": self.op.ignore_consistency,
3278
      }
3279
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
3280
    nl = [self.cfg.GetMasterNode()] + list(self.instance.secondary_nodes)
3281
    return env, nl, nl
3282

    
3283
  def CheckPrereq(self):
3284
    """Check prerequisites.
3285

3286
    This checks that the instance is in the cluster.
3287

3288
    """
3289
    self.instance = instance = self.cfg.GetInstanceInfo(self.op.instance_name)
3290
    assert self.instance is not None, \
3291
      "Cannot retrieve locked instance %s" % self.op.instance_name
3292

    
3293
    bep = self.cfg.GetClusterInfo().FillBE(instance)
3294
    if instance.disk_template not in constants.DTS_NET_MIRROR:
3295
      raise errors.OpPrereqError("Instance's disk layout is not"
3296
                                 " network mirrored, cannot failover.")
3297

    
3298
    secondary_nodes = instance.secondary_nodes
3299
    if not secondary_nodes:
3300
      raise errors.ProgrammerError("no secondary node but using "
3301
                                   "a mirrored disk template")
3302

    
3303
    target_node = secondary_nodes[0]
3304
    _CheckNodeOnline(self, target_node)
3305
    # check memory requirements on the secondary node
3306
    _CheckNodeFreeMemory(self, target_node, "failing over instance %s" %
3307
                         instance.name, bep[constants.BE_MEMORY],
3308
                         instance.hypervisor)
3309

    
3310
    # check bridge existance
3311
    brlist = [nic.bridge for nic in instance.nics]
3312
    result = self.rpc.call_bridges_exist(target_node, brlist)
3313
    result.Raise()
3314
    if not result.data:
3315
      raise errors.OpPrereqError("One or more target bridges %s does not"
3316
                                 " exist on destination node '%s'" %
3317
                                 (brlist, target_node))
3318

    
3319
  def Exec(self, feedback_fn):
3320
    """Failover an instance.
3321

3322
    The failover is done by shutting it down on its present node and
3323
    starting it on the secondary.
3324

3325
    """
3326
    instance = self.instance
3327

    
3328
    source_node = instance.primary_node
3329
    target_node = instance.secondary_nodes[0]
3330

    
3331
    feedback_fn("* checking disk consistency between source and target")
3332
    for dev in instance.disks:
3333
      # for drbd, these are drbd over lvm
3334
      if not _CheckDiskConsistency(self, dev, target_node, False):
3335
        if instance.status == "up" and not self.op.ignore_consistency:
3336
          raise errors.OpExecError("Disk %s is degraded on target node,"
3337
                                   " aborting failover." % dev.iv_name)
3338

    
3339
    feedback_fn("* shutting down instance on source node")
3340
    logging.info("Shutting down instance %s on node %s",
3341
                 instance.name, source_node)
3342

    
3343
    result = self.rpc.call_instance_shutdown(source_node, instance)
3344
    if result.failed or not result.data:
3345
      if self.op.ignore_consistency:
3346
        self.proc.LogWarning("Could not shutdown instance %s on node %s."
3347
                             " Proceeding"
3348
                             " anyway. Please make sure node %s is down",
3349
                             instance.name, source_node, source_node)
3350
      else:
3351
        raise errors.OpExecError("Could not shutdown instance %s on node %s" %
3352
                                 (instance.name, source_node))
3353

    
3354
    feedback_fn("* deactivating the instance's disks on source node")
3355
    if not _ShutdownInstanceDisks(self, instance, ignore_primary=True):
3356
      raise errors.OpExecError("Can't shut down the instance's disks.")
3357

    
3358
    instance.primary_node = target_node
3359
    # distribute new instance config to the other nodes
3360
    self.cfg.Update(instance)
3361

    
3362
    # Only start the instance if it's marked as up
3363
    if instance.status == "up":
3364
      feedback_fn("* activating the instance's disks on target node")
3365
      logging.info("Starting instance %s on node %s",
3366
                   instance.name, target_node)
3367

    
3368
      disks_ok, dummy = _AssembleInstanceDisks(self, instance,
3369
                                               ignore_secondaries=True)
3370
      if not disks_ok:
3371
        _ShutdownInstanceDisks(self, instance)
3372
        raise errors.OpExecError("Can't activate the instance's disks")
3373

    
3374
      feedback_fn("* starting the instance on the target node")
3375
      result = self.rpc.call_instance_start(target_node, instance, None)
3376
      if result.failed or not result.data:
3377
        _ShutdownInstanceDisks(self, instance)
3378
        raise errors.OpExecError("Could not start instance %s on node %s." %
3379
                                 (instance.name, target_node))
3380

    
3381

    
3382
class LUMigrateInstance(LogicalUnit):
3383
  """Migrate an instance.
3384

3385
  This is migration without shutting down, compared to the failover,
3386
  which is done with shutdown.
3387

3388
  """
3389
  HPATH = "instance-migrate"
3390
  HTYPE = constants.HTYPE_INSTANCE
3391
  _OP_REQP = ["instance_name", "live", "cleanup"]
3392

    
3393
  REQ_BGL = False
3394

    
3395
  def ExpandNames(self):
3396
    self._ExpandAndLockInstance()
3397
    self.needed_locks[locking.LEVEL_NODE] = []
3398
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
3399

    
3400
  def DeclareLocks(self, level):
3401
    if level == locking.LEVEL_NODE:
3402
      self._LockInstancesNodes()
3403

    
3404
  def BuildHooksEnv(self):
3405
    """Build hooks env.
3406

3407
    This runs on master, primary and secondary nodes of the instance.
3408

3409
    """
3410
    env = _BuildInstanceHookEnvByObject(self, self.instance)
3411
    nl = [self.cfg.GetMasterNode()] + list(self.instance.secondary_nodes)
3412
    return env, nl, nl
3413

    
3414
  def CheckPrereq(self):
3415
    """Check prerequisites.
3416

3417
    This checks that the instance is in the cluster.
3418

3419
    """
3420
    instance = self.cfg.GetInstanceInfo(
3421
      self.cfg.ExpandInstanceName(self.op.instance_name))
3422
    if instance is None:
3423
      raise errors.OpPrereqError("Instance '%s' not known" %
3424
                                 self.op.instance_name)
3425

    
3426
    if instance.disk_template != constants.DT_DRBD8:
3427
      raise errors.OpPrereqError("Instance's disk layout is not"
3428
                                 " drbd8, cannot migrate.")
3429

    
3430
    secondary_nodes = instance.secondary_nodes
3431
    if not secondary_nodes:
3432
      raise errors.ProgrammerError("no secondary node but using "
3433
                                   "drbd8 disk template")
3434

    
3435
    i_be = self.cfg.GetClusterInfo().FillBE(instance)
3436

    
3437
    target_node = secondary_nodes[0]
3438
    # check memory requirements on the secondary node
3439
    _CheckNodeFreeMemory(self, target_node, "migrating instance %s" %
3440
                         instance.name, i_be[constants.BE_MEMORY],
3441
                         instance.hypervisor)
3442

    
3443
    # check bridge existance
3444
    brlist = [nic.bridge for nic in instance.nics]
3445
    result = self.rpc.call_bridges_exist(target_node, brlist)
3446
    if result.failed or not result.data:
3447
      raise errors.OpPrereqError("One or more target bridges %s does not"
3448
                                 " exist on destination node '%s'" %
3449
                                 (brlist, target_node))
3450

    
3451
    if not self.op.cleanup:
3452
      result = self.rpc.call_instance_migratable(instance.primary_node,
3453
                                                 instance)
3454
      msg = result.RemoteFailMsg()
3455
      if msg:
3456
        raise errors.OpPrereqError("Can't migrate: %s - please use failover" %
3457
                                   msg)
3458

    
3459
    self.instance = instance
3460

    
3461
  def _WaitUntilSync(self):
3462
    """Poll with custom rpc for disk sync.
3463

3464
    This uses our own step-based rpc call.
3465

3466
    """
3467
    self.feedback_fn("* wait until resync is done")
3468
    all_done = False
3469
    while not all_done:
3470
      all_done = True
3471
      result = self.rpc.call_drbd_wait_sync(self.all_nodes,
3472
                                            self.nodes_ip,
3473
                                            self.instance.disks)
3474
      min_percent = 100
3475
      for node, nres in result.items():
3476
        msg = nres.RemoteFailMsg()
3477
        if msg:
3478
          raise errors.OpExecError("Cannot resync disks on node %s: %s" %
3479
                                   (node, msg))
3480
        node_done, node_percent = nres.data[1]
3481
        all_done = all_done and node_done
3482
        if node_percent is not None:
3483
          min_percent = min(min_percent, node_percent)
3484
      if not all_done:
3485
        if min_percent < 100:
3486
          self.feedback_fn("   - progress: %.1f%%" % min_percent)
3487
        time.sleep(2)
3488

    
3489
  def _EnsureSecondary(self, node):
3490
    """Demote a node to secondary.
3491

3492
    """
3493
    self.feedback_fn("* switching node %s to secondary mode" % node)
3494

    
3495
    for dev in self.instance.disks:
3496
      self.cfg.SetDiskID(dev, node)
3497

    
3498
    result = self.rpc.call_blockdev_close(node, self.instance.name,
3499
                                          self.instance.disks)
3500
    msg = result.RemoteFailMsg()
3501
    if msg:
3502
      raise errors.OpExecError("Cannot change disk to secondary on node %s,"
3503
                               " error %s" % (node, msg))
3504

    
3505
  def _GoStandalone(self):
3506
    """Disconnect from the network.
3507

3508
    """
3509
    self.feedback_fn("* changing into standalone mode")
3510
    result = self.rpc.call_drbd_disconnect_net(self.all_nodes, self.nodes_ip,
3511
                                               self.instance.disks)
3512
    for node, nres in result.items():
3513
      msg = nres.RemoteFailMsg()
3514
      if msg:
3515
        raise errors.OpExecError("Cannot disconnect disks node %s,"
3516
                                 " error %s" % (node, msg))
3517

    
3518
  def _GoReconnect(self, multimaster):
3519
    """Reconnect to the network.
3520

3521
    """
3522
    if multimaster:
3523
      msg = "dual-master"
3524
    else:
3525
      msg = "single-master"
3526
    self.feedback_fn("* changing disks into %s mode" % msg)
3527
    result = self.rpc.call_drbd_attach_net(self.all_nodes, self.nodes_ip,
3528
                                           self.instance.disks,
3529
                                           self.instance.name, multimaster)
3530
    for node, nres in result.items():
3531
      msg = nres.RemoteFailMsg()
3532
      if msg:
3533
        raise errors.OpExecError("Cannot change disks config on node %s,"
3534
                                 " error: %s" % (node, msg))
3535

    
3536
  def _ExecCleanup(self):
3537
    """Try to cleanup after a failed migration.
3538

3539
    The cleanup is done by:
3540
      - check that the instance is running only on one node
3541
        (and update the config if needed)
3542
      - change disks on its secondary node to secondary
3543
      - wait until disks are fully synchronized
3544
      - disconnect from the network
3545
      - change disks into single-master mode
3546
      - wait again until disks are fully synchronized
3547

3548
    """
3549
    instance = self.instance
3550
    target_node = self.target_node
3551
    source_node = self.source_node
3552

    
3553
    # check running on only one node
3554
    self.feedback_fn("* checking where the instance actually runs"
3555
                     " (if this hangs, the hypervisor might be in"
3556
                     " a bad state)")
3557
    ins_l = self.rpc.call_instance_list(self.all_nodes, [instance.hypervisor])
3558
    for node, result in ins_l.items():
3559
      result.Raise()
3560
      if not isinstance(result.data, list):
3561
        raise errors.OpExecError("Can't contact node '%s'" % node)
3562

    
3563
    runningon_source = instance.name in ins_l[source_node].data
3564
    runningon_target = instance.name in ins_l[target_node].data
3565

    
3566
    if runningon_source and runningon_target:
3567
      raise errors.OpExecError("Instance seems to be running on two nodes,"
3568
                               " or the hypervisor is confused. You will have"
3569
                               " to ensure manually that it runs only on one"
3570
                               " and restart this operation.")
3571

    
3572
    if not (runningon_source or runningon_target):
3573
      raise errors.OpExecError("Instance does not seem to be running at all."
3574
                               " In this case, it's safer to repair by"
3575
                               " running 'gnt-instance stop' to ensure disk"
3576
                               " shutdown, and then restarting it.")
3577

    
3578
    if runningon_target:
3579
      # the migration has actually succeeded, we need to update the config
3580
      self.feedback_fn("* instance running on secondary node (%s),"
3581
                       " updating config" % target_node)
3582
      instance.primary_node = target_node
3583
      self.cfg.Update(instance)
3584
      demoted_node = source_node
3585
    else:
3586
      self.feedback_fn("* instance confirmed to be running on its"
3587
                       " primary node (%s)" % source_node)
3588
      demoted_node = target_node
3589

    
3590
    self._EnsureSecondary(demoted_node)
3591
    try:
3592
      self._WaitUntilSync()
3593
    except errors.OpExecError:
3594
      # we ignore here errors, since if the device is standalone, it
3595
      # won't be able to sync
3596
      pass
3597
    self._GoStandalone()
3598
    self._GoReconnect(False)
3599
    self._WaitUntilSync()
3600

    
3601
    self.feedback_fn("* done")
3602

    
3603
  def _ExecMigration(self):
3604
    """Migrate an instance.
3605

3606
    The migrate is done by:
3607
      - change the disks into dual-master mode
3608
      - wait until disks are fully synchronized again
3609
      - migrate the instance
3610
      - change disks on the new secondary node (the old primary) to secondary
3611
      - wait until disks are fully synchronized
3612
      - change disks into single-master mode
3613

3614
    """
3615
    instance = self.instance
3616
    target_node = self.target_node
3617
    source_node = self.source_node
3618

    
3619
    self.feedback_fn("* checking disk consistency between source and target")
3620
    for dev in instance.disks:
3621
      if not _CheckDiskConsistency(self, dev, target_node, False):
3622
        raise errors.OpExecError("Disk %s is degraded or not fully"
3623
                                 " synchronized on target node,"
3624
                                 " aborting migrate." % dev.iv_name)
3625

    
3626
    self._EnsureSecondary(target_node)
3627
    self._GoStandalone()
3628
    self._GoReconnect(True)
3629
    self._WaitUntilSync()
3630

    
3631
    self.feedback_fn("* migrating instance to %s" % target_node)
3632
    time.sleep(10)
3633
    result = self.rpc.call_instance_migrate(source_node, instance,
3634
                                            self.nodes_ip[target_node],
3635
                                            self.op.live)
3636
    msg = result.RemoteFailMsg()
3637
    if msg:
3638
      logging.error("Instance migration failed, trying to revert"
3639
                    " disk status: %s", msg)
3640
      try:
3641
        self._EnsureSecondary(target_node)
3642
        self._GoStandalone()
3643
        self._GoReconnect(False)
3644
        self._WaitUntilSync()
3645
      except errors.OpExecError, err:
3646
        self.LogWarning("Migration failed and I can't reconnect the"
3647
                        " drives: error '%s'\n"
3648
                        "Please look and recover the instance status" %
3649
                        str(err))
3650

    
3651
      raise errors.OpExecError("Could not migrate instance %s: %s" %
3652
                               (instance.name, msg))
3653
    time.sleep(10)
3654

    
3655
    instance.primary_node = target_node
3656
    # distribute new instance config to the other nodes
3657
    self.cfg.Update(instance)
3658

    
3659
    self._EnsureSecondary(source_node)
3660
    self._WaitUntilSync()
3661
    self._GoStandalone()
3662
    self._GoReconnect(False)
3663
    self._WaitUntilSync()
3664

    
3665
    self.feedback_fn("* done")
3666

    
3667
  def Exec(self, feedback_fn):
3668
    """Perform the migration.
3669

3670
    """
3671
    self.feedback_fn = feedback_fn
3672

    
3673
    self.source_node = self.instance.primary_node
3674
    self.target_node = self.instance.secondary_nodes[0]
3675
    self.all_nodes = [self.source_node, self.target_node]
3676
    self.nodes_ip = {
3677
      self.source_node: self.cfg.GetNodeInfo(self.source_node).secondary_ip,
3678
      self.target_node: self.cfg.GetNodeInfo(self.target_node).secondary_ip,
3679
      }
3680
    if self.op.cleanup:
3681
      return self._ExecCleanup()
3682
    else:
3683
      return self._ExecMigration()
3684

    
3685

    
3686
def _CreateBlockDev(lu, node, instance, device, force_create,
3687
                    info, force_open):
3688
  """Create a tree of block devices on a given node.
3689

3690
  If this device type has to be created on secondaries, create it and
3691
  all its children.
3692

3693
  If not, just recurse to children keeping the same 'force' value.
3694

3695
  @param lu: the lu on whose behalf we execute
3696
  @param node: the node on which to create the device
3697
  @type instance: L{objects.Instance}
3698
  @param instance: the instance which owns the device
3699
  @type device: L{objects.Disk}
3700
  @param device: the device to create
3701
  @type force_create: boolean
3702
  @param force_create: whether to force creation of this device; this
3703
      will be change to True whenever we find a device which has
3704
      CreateOnSecondary() attribute
3705
  @param info: the extra 'metadata' we should attach to the device
3706
      (this will be represented as a LVM tag)
3707
  @type force_open: boolean
3708
  @param force_open: this parameter will be passes to the
3709
      L{backend.CreateBlockDevice} function where it specifies
3710
      whether we run on primary or not, and it affects both
3711
      the child assembly and the device own Open() execution
3712

3713
  """
3714
  if device.CreateOnSecondary():
3715
    force_create = True
3716

    
3717
  if device.children:
3718
    for child in device.children:
3719
      _CreateBlockDev(lu, node, instance, child, force_create,
3720
                      info, force_open)
3721

    
3722
  if not force_create:
3723
    return
3724

    
3725
  _CreateSingleBlockDev(lu, node, instance, device, info, force_open)
3726

    
3727

    
3728
def _CreateSingleBlockDev(lu, node, instance, device, info, force_open):
3729
  """Create a single block device on a given node.
3730

3731
  This will not recurse over children of the device, so they must be
3732
  created in advance.
3733

3734
  @param lu: the lu on whose behalf we execute
3735
  @param node: the node on which to create the device
3736
  @type instance: L{objects.Instance}
3737
  @param instance: the instance which owns the device
3738
  @type device: L{objects.Disk}
3739
  @param device: the device to create
3740
  @param info: the extra 'metadata' we should attach to the device
3741
      (this will be represented as a LVM tag)
3742
  @type force_open: boolean
3743
  @param force_open: this parameter will be passes to the
3744
      L{backend.CreateBlockDevice} function where it specifies
3745
      whether we run on primary or not, and it affects both
3746
      the child assembly and the device own Open() execution
3747

3748
  """
3749
  lu.cfg.SetDiskID(device, node)
3750
  new_id = lu.rpc.call_blockdev_create(node, device, device.size,
3751
                                       instance.name, force_open, info)
3752
  if new_id.failed or not new_id.data:
3753
    raise errors.OpExecError("Can't create block device %s on"
3754
                             " node %s for instance %s" %
3755
                             (device, node, instance.name))
3756
  if device.physical_id is None:
3757
    device.physical_id = new_id
3758

    
3759

    
3760
def _GenerateUniqueNames(lu, exts):
3761
  """Generate a suitable LV name.
3762

3763
  This will generate a logical volume name for the given instance.
3764

3765
  """
3766
  results = []
3767
  for val in exts:
3768
    new_id = lu.cfg.GenerateUniqueID()
3769
    results.append("%s%s" % (new_id, val))
3770
  return results
3771

    
3772

    
3773
def _GenerateDRBD8Branch(lu, primary, secondary, size, names, iv_name,
3774
                         p_minor, s_minor):
3775
  """Generate a drbd8 device complete with its children.
3776

3777
  """
3778
  port = lu.cfg.AllocatePort()
3779
  vgname = lu.cfg.GetVGName()
3780
  shared_secret = lu.cfg.GenerateDRBDSecret()
3781
  dev_data = objects.Disk(dev_type=constants.LD_LV, size=size,
3782
                          logical_id=(vgname, names[0]))
3783
  dev_meta = objects.Disk(dev_type=constants.LD_LV, size=128,
3784
                          logical_id=(vgname, names[1]))
3785
  drbd_dev = objects.Disk(dev_type=constants.LD_DRBD8, size=size,
3786
                          logical_id=(primary, secondary, port,
3787
                                      p_minor, s_minor,
3788
                                      shared_secret),
3789
                          children=[dev_data, dev_meta],
3790
                          iv_name=iv_name)
3791
  return drbd_dev
3792

    
3793

    
3794
def _GenerateDiskTemplate(lu, template_name,
3795
                          instance_name, primary_node,
3796
                          secondary_nodes, disk_info,
3797
                          file_storage_dir, file_driver,
3798
                          base_index):
3799
  """Generate the entire disk layout for a given template type.
3800

3801
  """
3802
  #TODO: compute space requirements
3803

    
3804
  vgname = lu.cfg.GetVGName()
3805
  disk_count = len(disk_info)
3806
  disks = []
3807
  if template_name == constants.DT_DISKLESS:
3808
    pass
3809
  elif template_name == constants.DT_PLAIN:
3810
    if len(secondary_nodes) != 0:
3811
      raise errors.ProgrammerError("Wrong template configuration")
3812

    
3813
    names = _GenerateUniqueNames(lu, [".disk%d" % i
3814
                                      for i in range(disk_count)])
3815
    for idx, disk in enumerate(disk_info):
3816
      disk_index = idx + base_index
3817
      disk_dev = objects.Disk(dev_type=constants.LD_LV, size=disk["size"],
3818
                              logical_id=(vgname, names[idx]),
3819
                              iv_name="disk/%d" % disk_index)
3820
      disks.append(disk_dev)
3821
  elif template_name == constants.DT_DRBD8:
3822
    if len(secondary_nodes) != 1:
3823
      raise errors.ProgrammerError("Wrong template configuration")
3824
    remote_node = secondary_nodes[0]
3825
    minors = lu.cfg.AllocateDRBDMinor(
3826
      [primary_node, remote_node] * len(disk_info), instance_name)
3827

    
3828
    names = []
3829
    for lv_prefix in _GenerateUniqueNames(lu, [".disk%d" % i
3830
                                               for i in range(disk_count)]):
3831
      names.append(lv_prefix + "_data")
3832
      names.append(lv_prefix + "_meta")
3833
    for idx, disk in enumerate(disk_info):
3834
      disk_index = idx + base_index
3835
      disk_dev = _GenerateDRBD8Branch(lu, primary_node, remote_node,
3836
                                      disk["size"], names[idx*2:idx*2+2],
3837
                                      "disk/%d" % disk_index,
3838
                                      minors[idx*2], minors[idx*2+1])
3839
      disks.append(disk_dev)
3840
  elif template_name == constants.DT_FILE:
3841
    if len(secondary_nodes) != 0:
3842
      raise errors.ProgrammerError("Wrong template configuration")
3843

    
3844
    for idx, disk in enumerate(disk_info):
3845
      disk_index = idx + base_index
3846
      disk_dev = objects.Disk(dev_type=constants.LD_FILE, size=disk["size"],
3847
                              iv_name="disk/%d" % disk_index,
3848
                              logical_id=(file_driver,
3849
                                          "%s/disk%d" % (file_storage_dir,
3850
                                                         idx)))
3851
      disks.append(disk_dev)
3852
  else:
3853
    raise errors.ProgrammerError("Invalid disk template '%s'" % template_name)
3854
  return disks
3855

    
3856

    
3857
def _GetInstanceInfoText(instance):
3858
  """Compute that text that should be added to the disk's metadata.
3859

3860
  """
3861
  return "originstname+%s" % instance.name
3862

    
3863

    
3864
def _CreateDisks(lu, instance):
3865
  """Create all disks for an instance.
3866

3867
  This abstracts away some work from AddInstance.
3868

3869
  @type lu: L{LogicalUnit}
3870
  @param lu: the logical unit on whose behalf we execute
3871
  @type instance: L{objects.Instance}
3872
  @param instance: the instance whose disks we should create
3873
  @rtype: boolean
3874
  @return: the success of the creation
3875

3876
  """
3877
  info = _GetInstanceInfoText(instance)
3878
  pnode = instance.primary_node
3879

    
3880
  if instance.disk_template == constants.DT_FILE:
3881
    file_storage_dir = os.path.dirname(instance.disks[0].logical_id[1])
3882
    result = lu.rpc.call_file_storage_dir_create(pnode, file_storage_dir)
3883

    
3884
    if result.failed or not result.data:
3885
      raise errors.OpExecError("Could not connect to node '%s'" % pnode)
3886

    
3887
    if not result.data[0]:
3888
      raise errors.OpExecError("Failed to create directory '%s'" %
3889
                               file_storage_dir)
3890

    
3891
  # Note: this needs to be kept in sync with adding of disks in
3892
  # LUSetInstanceParams
3893
  for device in instance.disks:
3894
    logging.info("Creating volume %s for instance %s",
3895
                 device.iv_name, instance.name)
3896
    #HARDCODE
3897
    for node in instance.all_nodes:
3898
      f_create = node == pnode
3899
      _CreateBlockDev(lu, node, instance, device, f_create, info, f_create)
3900

    
3901

    
3902
def _RemoveDisks(lu, instance):
3903
  """Remove all disks for an instance.
3904

3905
  This abstracts away some work from `AddInstance()` and
3906
  `RemoveInstance()`. Note that in case some of the devices couldn't
3907
  be removed, the removal will continue with the other ones (compare
3908
  with `_CreateDisks()`).
3909

3910
  @type lu: L{LogicalUnit}
3911
  @param lu: the logical unit on whose behalf we execute
3912
  @type instance: L{objects.Instance}
3913
  @param instance: the instance whose disks we should remove
3914
  @rtype: boolean
3915
  @return: the success of the removal
3916

3917
  """
3918
  logging.info("Removing block devices for instance %s", instance.name)
3919

    
3920
  result = True
3921
  for device in instance.disks:
3922
    for node, disk in device.ComputeNodeTree(instance.primary_node):
3923
      lu.cfg.SetDiskID(disk, node)
3924
      result = lu.rpc.call_blockdev_remove(node, disk)
3925
      if result.failed or not result.data:
3926
        lu.proc.LogWarning("Could not remove block device %s on node %s,"
3927
                           " continuing anyway", device.iv_name, node)
3928
        result = False
3929

    
3930
  if instance.disk_template == constants.DT_FILE:
3931
    file_storage_dir = os.path.dirname(instance.disks[0].logical_id[1])
3932
    result = lu.rpc.call_file_storage_dir_remove(instance.primary_node,
3933
                                                 file_storage_dir)
3934
    if result.failed or not result.data:
3935
      logging.error("Could not remove directory '%s'", file_storage_dir)
3936
      result = False
3937

    
3938
  return result
3939

    
3940

    
3941
def _ComputeDiskSize(disk_template, disks):
3942
  """Compute disk size requirements in the volume group
3943

3944
  """
3945
  # Required free disk space as a function of disk and swap space
3946
  req_size_dict = {
3947
    constants.DT_DISKLESS: None,
3948
    constants.DT_PLAIN: sum(d["size"] for d in disks),
3949
    # 128 MB are added for drbd metadata for each disk
3950
    constants.DT_DRBD8: sum(d["size"] + 128 for d in disks),
3951
    constants.DT_FILE: None,
3952
  }
3953

    
3954
  if disk_template not in req_size_dict:
3955
    raise errors.ProgrammerError("Disk template '%s' size requirement"
3956
                                 " is unknown" %  disk_template)
3957

    
3958
  return req_size_dict[disk_template]
3959

    
3960

    
3961
def _CheckHVParams(lu, nodenames, hvname, hvparams):
3962
  """Hypervisor parameter validation.
3963

3964
  This function abstract the hypervisor parameter validation to be
3965
  used in both instance create and instance modify.
3966

3967
  @type lu: L{LogicalUnit}
3968
  @param lu: the logical unit for which we check
3969
  @type nodenames: list
3970
  @param nodenames: the list of nodes on which we should check
3971
  @type hvname: string
3972
  @param hvname: the name of the hypervisor we should use
3973
  @type hvparams: dict
3974
  @param hvparams: the parameters which we need to check
3975
  @raise errors.OpPrereqError: if the parameters are not valid
3976

3977
  """
3978
  hvinfo = lu.rpc.call_hypervisor_validate_params(nodenames,
3979
                                                  hvname,
3980
                                                  hvparams)
3981
  for node in nodenames:
3982
    info = hvinfo[node]
3983
    info.Raise()
3984
    if not info.data or not isinstance(info.data, (tuple, list)):
3985
      raise errors.OpPrereqError("Cannot get current information"
3986
                                 " from node '%s' (%s)" % (node, info.data))
3987
    if not info.data[0]:
3988
      raise errors.OpPrereqError("Hypervisor parameter validation failed:"
3989
                                 " %s" % info.data[1])
3990

    
3991

    
3992
class LUCreateInstance(LogicalUnit):
3993
  """Create an instance.
3994

3995
  """
3996
  HPATH = "instance-add"
3997
  HTYPE = constants.HTYPE_INSTANCE
3998
  _OP_REQP = ["instance_name", "disks", "disk_template",
3999
              "mode", "start",
4000
              "wait_for_sync", "ip_check", "nics",
4001
              "hvparams", "beparams"]
4002
  REQ_BGL = False
4003

    
4004
  def _ExpandNode(self, node):
4005
    """Expands and checks one node name.
4006

4007
    """
4008
    node_full = self.cfg.ExpandNodeName(node)
4009
    if node_full is None:
4010
      raise errors.OpPrereqError("Unknown node %s" % node)
4011
    return node_full
4012

    
4013
  def ExpandNames(self):
4014
    """ExpandNames for CreateInstance.
4015

4016
    Figure out the right locks for instance creation.
4017

4018
    """
4019
    self.needed_locks = {}
4020

    
4021
    # set optional parameters to none if they don't exist
4022
    for attr in ["pnode", "snode", "iallocator", "hypervisor"]:
4023
      if not hasattr(self.op, attr):
4024
        setattr(self.op, attr, None)
4025

    
4026
    # cheap checks, mostly valid constants given
4027

    
4028
    # verify creation mode
4029
    if self.op.mode not in (constants.INSTANCE_CREATE,
4030
                            constants.INSTANCE_IMPORT):
4031
      raise errors.OpPrereqError("Invalid instance creation mode '%s'" %
4032
                                 self.op.mode)
4033

    
4034
    # disk template and mirror node verification
4035
    if self.op.disk_template not in constants.DISK_TEMPLATES:
4036
      raise errors.OpPrereqError("Invalid disk template name")
4037

    
4038
    if self.op.hypervisor is None:
4039
      self.op.hypervisor = self.cfg.GetHypervisorType()
4040

    
4041
    cluster = self.cfg.GetClusterInfo()
4042
    enabled_hvs = cluster.enabled_hypervisors
4043
    if self.op.hypervisor not in enabled_hvs:
4044
      raise errors.OpPrereqError("Selected hypervisor (%s) not enabled in the"
4045
                                 " cluster (%s)" % (self.op.hypervisor,
4046
                                  ",".join(enabled_hvs)))
4047

    
4048
    # check hypervisor parameter syntax (locally)
4049

    
4050
    filled_hvp = cluster.FillDict(cluster.hvparams[self.op.hypervisor],
4051
                                  self.op.hvparams)
4052
    hv_type = hypervisor.GetHypervisor(self.op.hypervisor)
4053
    hv_type.CheckParameterSyntax(filled_hvp)
4054

    
4055
    # fill and remember the beparams dict
4056
    utils.CheckBEParams(self.op.beparams)
4057
    self.be_full = cluster.FillDict(cluster.beparams[constants.BEGR_DEFAULT],
4058
                                    self.op.beparams)
4059

    
4060
    #### instance parameters check
4061

    
4062
    # instance name verification
4063
    hostname1 = utils.HostInfo(self.op.instance_name)
4064
    self.op.instance_name = instance_name = hostname1.name
4065

    
4066
    # this is just a preventive check, but someone might still add this
4067
    # instance in the meantime, and creation will fail at lock-add time
4068
    if instance_name in self.cfg.GetInstanceList():
4069
      raise errors.OpPrereqError("Instance '%s' is already in the cluster" %
4070
                                 instance_name)
4071

    
4072
    self.add_locks[locking.LEVEL_INSTANCE] = instance_name
4073

    
4074
    # NIC buildup
4075
    self.nics = []
4076
    for nic in self.op.nics:
4077
      # ip validity checks
4078
      ip = nic.get("ip", None)
4079
      if ip is None or ip.lower() == "none":
4080
        nic_ip = None
4081
      elif ip.lower() == constants.VALUE_AUTO:
4082
        nic_ip = hostname1.ip
4083
      else:
4084
        if not utils.IsValidIP(ip):
4085
          raise errors.OpPrereqError("Given IP address '%s' doesn't look"
4086
                                     " like a valid IP" % ip)
4087
        nic_ip = ip
4088

    
4089
      # MAC address verification
4090
      mac = nic.get("mac", constants.VALUE_AUTO)
4091
      if mac not in (constants.VALUE_AUTO, constants.VALUE_GENERATE):
4092
        if not utils.IsValidMac(mac.lower()):
4093
          raise errors.OpPrereqError("Invalid MAC address specified: %s" %
4094
                                     mac)
4095
      # bridge verification
4096
      bridge = nic.get("bridge", self.cfg.GetDefBridge())
4097
      self.nics.append(objects.NIC(mac=mac, ip=nic_ip, bridge=bridge))
4098

    
4099
    # disk checks/pre-build
4100
    self.disks = []
4101
    for disk in self.op.disks:
4102
      mode = disk.get("mode", constants.DISK_RDWR)
4103
      if mode not in constants.DISK_ACCESS_SET:
4104
        raise errors.OpPrereqError("Invalid disk access mode '%s'" %
4105
                                   mode)
4106
      size = disk.get("size", None)
4107
      if size is None:
4108
        raise errors.OpPrereqError("Missing disk size")
4109
      try:
4110
        size = int(size)
4111
      except ValueError:
4112
        raise errors.OpPrereqError("Invalid disk size '%s'" % size)
4113
      self.disks.append({"size": size, "mode": mode})
4114

    
4115
    # used in CheckPrereq for ip ping check
4116
    self.check_ip = hostname1.ip
4117

    
4118
    # file storage checks
4119
    if (self.op.file_driver and
4120
        not self.op.file_driver in constants.FILE_DRIVER):
4121
      raise errors.OpPrereqError("Invalid file driver name '%s'" %
4122
                                 self.op.file_driver)
4123

    
4124
    if self.op.file_storage_dir and os.path.isabs(self.op.file_storage_dir):
4125
      raise errors.OpPrereqError("File storage directory path not absolute")
4126

    
4127
    ### Node/iallocator related checks
4128
    if [self.op.iallocator, self.op.pnode].count(None) != 1:
4129
      raise errors.OpPrereqError("One and only one of iallocator and primary"
4130
                                 " node must be given")
4131

    
4132
    if self.op.iallocator:
4133
      self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
4134
    else:
4135
      self.op.pnode = self._ExpandNode(self.op.pnode)
4136
      nodelist = [self.op.pnode]
4137
      if self.op.snode is not None:
4138
        self.op.snode = self._ExpandNode(self.op.snode)
4139
        nodelist.append(self.op.snode)
4140
      self.needed_locks[locking.LEVEL_NODE] = nodelist
4141

    
4142
    # in case of import lock the source node too
4143
    if self.op.mode == constants.INSTANCE_IMPORT:
4144
      src_node = getattr(self.op, "src_node", None)
4145
      src_path = getattr(self.op, "src_path", None)
4146

    
4147
      if src_path is None:
4148
        self.op.src_path = src_path = self.op.instance_name
4149

    
4150
      if src_node is None:
4151
        self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
4152
        self.op.src_node = None
4153
        if os.path.isabs(src_path):
4154
          raise errors.OpPrereqError("Importing an instance from an absolute"
4155
                                     " path requires a source node option.")
4156
      else:
4157
        self.op.src_node = src_node = self._ExpandNode(src_node)
4158
        if self.needed_locks[locking.LEVEL_NODE] is not locking.ALL_SET:
4159
          self.needed_locks[locking.LEVEL_NODE].append(src_node)
4160
        if not os.path.isabs(src_path):
4161
          self.op.src_path = src_path = \
4162
            os.path.join(constants.EXPORT_DIR, src_path)
4163

    
4164
    else: # INSTANCE_CREATE
4165
      if getattr(self.op, "os_type", None) is None:
4166
        raise errors.OpPrereqError("No guest OS specified")
4167

    
4168
  def _RunAllocator(self):
4169
    """Run the allocator based on input opcode.
4170

4171
    """
4172
    nics = [n.ToDict() for n in self.nics]
4173
    ial = IAllocator(self,
4174
                     mode=constants.IALLOCATOR_MODE_ALLOC,
4175
                     name=self.op.instance_name,
4176
                     disk_template=self.op.disk_template,
4177
                     tags=[],
4178
                     os=self.op.os_type,
4179
                     vcpus=self.be_full[constants.BE_VCPUS],
4180
                     mem_size=self.be_full[constants.BE_MEMORY],
4181
                     disks=self.disks,
4182
                     nics=nics,
4183
                     hypervisor=self.op.hypervisor,
4184
                     )
4185

    
4186
    ial.Run(self.op.iallocator)
4187

    
4188
    if not ial.success:
4189
      raise errors.OpPrereqError("Can't compute nodes using"
4190
                                 " iallocator '%s': %s" % (self.op.iallocator,
4191
                                                           ial.info))
4192
    if len(ial.nodes) != ial.required_nodes:
4193
      raise errors.OpPrereqError("iallocator '%s' returned invalid number"
4194
                                 " of nodes (%s), required %s" %
4195
                                 (self.op.iallocator, len(ial.nodes),
4196
                                  ial.required_nodes))
4197
    self.op.pnode = ial.nodes[0]
4198
    self.LogInfo("Selected nodes for instance %s via iallocator %s: %s",
4199
                 self.op.instance_name, self.op.iallocator,
4200
                 ", ".join(ial.nodes))
4201
    if ial.required_nodes == 2:
4202
      self.op.snode = ial.nodes[1]
4203

    
4204
  def BuildHooksEnv(self):
4205
    """Build hooks env.
4206

4207
    This runs on master, primary and secondary nodes of the instance.
4208

4209
    """
4210
    env = {
4211
      "INSTANCE_DISK_TEMPLATE": self.op.disk_template,
4212
      "INSTANCE_DISK_SIZE": ",".join(str(d["size"]) for d in self.disks),
4213
      "INSTANCE_ADD_MODE": self.op.mode,
4214
      }
4215
    if self.op.mode == constants.INSTANCE_IMPORT:
4216
      env["INSTANCE_SRC_NODE"] = self.op.src_node
4217
      env["INSTANCE_SRC_PATH"] = self.op.src_path
4218
      env["INSTANCE_SRC_IMAGES"] = self.src_images
4219

    
4220
    env.update(_BuildInstanceHookEnv(name=self.op.instance_name,
4221
      primary_node=self.op.pnode,
4222
      secondary_nodes=self.secondaries,
4223
      status=self.instance_status,
4224
      os_type=self.op.os_type,
4225
      memory=self.be_full[constants.BE_MEMORY],
4226
      vcpus=self.be_full[constants.BE_VCPUS],
4227
      nics=[(n.ip, n.bridge, n.mac) for n in self.nics],
4228
    ))
4229

    
4230
    nl = ([self.cfg.GetMasterNode(), self.op.pnode] +
4231
          self.secondaries)
4232
    return env, nl, nl
4233

    
4234

    
4235
  def CheckPrereq(self):
4236
    """Check prerequisites.
4237

4238
    """
4239
    if (not self.cfg.GetVGName() and
4240
        self.op.disk_template not in constants.DTS_NOT_LVM):
4241
      raise errors.OpPrereqError("Cluster does not support lvm-based"
4242
                                 " instances")
4243

    
4244

    
4245
    if self.op.mode == constants.INSTANCE_IMPORT:
4246
      src_node = self.op.src_node
4247
      src_path = self.op.src_path
4248

    
4249
      if src_node is None:
4250
        exp_list = self.rpc.call_export_list(
4251
          self.acquired_locks[locking.LEVEL_NODE])
4252
        found = False
4253
        for node in exp_list:
4254
          if not exp_list[node].failed and src_path in exp_list[node].data:
4255
            found = True
4256
            self.op.src_node = src_node = node
4257
            self.op.src_path = src_path = os.path.join(constants.EXPORT_DIR,
4258
                                                       src_path)
4259
            break
4260
        if not found:
4261
          raise errors.OpPrereqError("No export found for relative path %s" %
4262
                                      src_path)
4263

    
4264
      _CheckNodeOnline(self, src_node)
4265
      result = self.rpc.call_export_info(src_node, src_path)
4266
      result.Raise()
4267
      if not result.data:
4268
        raise errors.OpPrereqError("No export found in dir %s" % src_path)
4269

    
4270
      export_info = result.data
4271
      if not export_info.has_section(constants.INISECT_EXP):
4272
        raise errors.ProgrammerError("Corrupted export config")
4273

    
4274
      ei_version = export_info.get(constants.INISECT_EXP, 'version')
4275
      if (int(ei_version) != constants.EXPORT_VERSION):
4276
        raise errors.OpPrereqError("Wrong export version %s (wanted %d)" %
4277
                                   (ei_version, constants.EXPORT_VERSION))
4278

    
4279
      # Check that the new instance doesn't have less disks than the export
4280
      instance_disks = len(self.disks)
4281
      export_disks = export_info.getint(constants.INISECT_INS, 'disk_count')
4282
      if instance_disks < export_disks:
4283
        raise errors.OpPrereqError("Not enough disks to import."
4284
                                   " (instance: %d, export: %d)" %
4285
                                   (instance_disks, export_disks))
4286

    
4287
      self.op.os_type = export_info.get(constants.INISECT_EXP, 'os')
4288
      disk_images = []
4289
      for idx in range(export_disks):
4290
        option = 'disk%d_dump' % idx
4291
        if export_info.has_option(constants.INISECT_INS, option):
4292
          # FIXME: are the old os-es, disk sizes, etc. useful?
4293
          export_name = export_info.get(constants.INISECT_INS, option)
4294
          image = os.path.join(src_path, export_name)
4295
          disk_images.append(image)
4296
        else:
4297
          disk_images.append(False)
4298

    
4299
      self.src_images = disk_images
4300

    
4301
      old_name = export_info.get(constants.INISECT_INS, 'name')
4302
      # FIXME: int() here could throw a ValueError on broken exports
4303
      exp_nic_count = int(export_info.get(constants.INISECT_INS, 'nic_count'))
4304
      if self.op.instance_name == old_name:
4305
        for idx, nic in enumerate(self.nics):
4306
          if nic.mac == constants.VALUE_AUTO and exp_nic_count >= idx:
4307
            nic_mac_ini = 'nic%d_mac' % idx
4308
            nic.mac = export_info.get(constants.INISECT_INS, nic_mac_ini)
4309

    
4310
    # ip ping checks (we use the same ip that was resolved in ExpandNames)
4311
    if self.op.start and not self.op.ip_check:
4312
      raise errors.OpPrereqError("Cannot ignore IP address conflicts when"
4313
                                 " adding an instance in start mode")
4314

    
4315
    if self.op.ip_check:
4316
      if utils.TcpPing(self.check_ip, constants.DEFAULT_NODED_PORT):
4317
        raise errors.OpPrereqError("IP %s of instance %s already in use" %
4318
                                   (self.check_ip, self.op.instance_name))
4319

    
4320
    #### allocator run
4321

    
4322
    if self.op.iallocator is not None:
4323
      self._RunAllocator()
4324

    
4325
    #### node related checks
4326

    
4327
    # check primary node
4328
    self.pnode = pnode = self.cfg.GetNodeInfo(self.op.pnode)
4329
    assert self.pnode is not None, \
4330
      "Cannot retrieve locked node %s" % self.op.pnode
4331
    if pnode.offline:
4332
      raise errors.OpPrereqError("Cannot use offline primary node '%s'" %
4333
                                 pnode.name)
4334

    
4335
    self.secondaries = []
4336

    
4337
    # mirror node verification
4338
    if self.op.disk_template in constants.DTS_NET_MIRROR:
4339
      if self.op.snode is None:
4340
        raise errors.OpPrereqError("The networked disk templates need"
4341
                                   " a mirror node")
4342
      if self.op.snode == pnode.name:
4343
        raise errors.OpPrereqError("The secondary node cannot be"
4344
                                   " the primary node.")
4345
      self.secondaries.append(self.op.snode)
4346
      _CheckNodeOnline(self, self.op.snode)
4347

    
4348
    nodenames = [pnode.name] + self.secondaries
4349

    
4350
    req_size = _ComputeDiskSize(self.op.disk_template,
4351
                                self.disks)
4352

    
4353
    # Check lv size requirements
4354
    if req_size is not None:
4355
      nodeinfo = self.rpc.call_node_info(nodenames, self.cfg.GetVGName(),
4356
                                         self.op.hypervisor)
4357
      for node in nodenames:
4358
        info = nodeinfo[node]
4359
        info.Raise()
4360
        info = info.data
4361
        if not info:
4362
          raise errors.OpPrereqError("Cannot get current information"
4363
                                     " from node '%s'" % node)
4364
        vg_free = info.get('vg_free', None)
4365
        if not isinstance(vg_free, int):
4366
          raise errors.OpPrereqError("Can't compute free disk space on"
4367
                                     " node %s" % node)
4368
        if req_size > info['vg_free']:
4369
          raise errors.OpPrereqError("Not enough disk space on target node %s."
4370
                                     " %d MB available, %d MB required" %
4371
                                     (node, info['vg_free'], req_size))
4372

    
4373
    _CheckHVParams(self, nodenames, self.op.hypervisor, self.op.hvparams)
4374

    
4375
    # os verification
4376
    result = self.rpc.call_os_get(pnode.name, self.op.os_type)
4377
    result.Raise()
4378
    if not isinstance(result.data, objects.OS):
4379
      raise errors.OpPrereqError("OS '%s' not in supported os list for"
4380
                                 " primary node"  % self.op.os_type)
4381

    
4382
    # bridge check on primary node
4383
    bridges = [n.bridge for n in self.nics]
4384
    result = self.rpc.call_bridges_exist(self.pnode.name, bridges)
4385
    result.Raise()
4386
    if not result.data:
4387
      raise errors.OpPrereqError("One of the target bridges '%s' does not"
4388
                                 " exist on destination node '%s'" %
4389
                                 (",".join(bridges), pnode.name))
4390

    
4391
    # memory check on primary node
4392
    if self.op.start:
4393
      _CheckNodeFreeMemory(self, self.pnode.name,
4394
                           "creating instance %s" % self.op.instance_name,
4395
                           self.be_full[constants.BE_MEMORY],
4396
                           self.op.hypervisor)
4397

    
4398
    if self.op.start:
4399
      self.instance_status = 'up'
4400
    else:
4401
      self.instance_status = 'down'
4402

    
4403
  def Exec(self, feedback_fn):
4404
    """Create and add the instance to the cluster.
4405

4406
    """
4407
    instance = self.op.instance_name
4408
    pnode_name = self.pnode.name
4409

    
4410
    for nic in self.nics:
4411
      if nic.mac in (constants.VALUE_AUTO, constants.VALUE_GENERATE):
4412
        nic.mac = self.cfg.GenerateMAC()
4413

    
4414
    ht_kind = self.op.hypervisor
4415
    if ht_kind in constants.HTS_REQ_PORT:
4416
      network_port = self.cfg.AllocatePort()
4417
    else:
4418
      network_port = None
4419

    
4420
    ##if self.op.vnc_bind_address is None:
4421
    ##  self.op.vnc_bind_address = constants.VNC_DEFAULT_BIND_ADDRESS
4422

    
4423
    # this is needed because os.path.join does not accept None arguments
4424
    if self.op.file_storage_dir is None:
4425
      string_file_storage_dir = ""
4426
    else:
4427
      string_file_storage_dir = self.op.file_storage_dir
4428

    
4429
    # build the full file storage dir path
4430
    file_storage_dir = os.path.normpath(os.path.join(
4431
                                        self.cfg.GetFileStorageDir(),
4432
                                        string_file_storage_dir, instance))
4433

    
4434

    
4435
    disks = _GenerateDiskTemplate(self,
4436
                                  self.op.disk_template,
4437
                                  instance, pnode_name,
4438
                                  self.secondaries,
4439
                                  self.disks,
4440
                                  file_storage_dir,
4441
                                  self.op.file_driver,
4442
                                  0)
4443

    
4444
    iobj = objects.Instance(name=instance, os=self.op.os_type,
4445
                            primary_node=pnode_name,
4446
                            nics=self.nics, disks=disks,
4447
                            disk_template=self.op.disk_template,
4448
                            status=self.instance_status,
4449
                            network_port=network_port,
4450
                            beparams=self.op.beparams,
4451
                            hvparams=self.op.hvparams,
4452
                            hypervisor=self.op.hypervisor,
4453
                            )
4454

    
4455
    feedback_fn("* creating instance disks...")
4456
    try:
4457
      _CreateDisks(self, iobj)
4458
    except errors.OpExecError:
4459
      self.LogWarning("Device creation failed, reverting...")
4460
      try:
4461
        _RemoveDisks(self, iobj)
4462
      finally:
4463
        self.cfg.ReleaseDRBDMinors(instance)
4464
        raise
4465

    
4466
    feedback_fn("adding instance %s to cluster config" % instance)
4467

    
4468
    self.cfg.AddInstance(iobj)
4469
    # Declare that we don't want to remove the instance lock anymore, as we've
4470
    # added the instance to the config
4471
    del self.remove_locks[locking.LEVEL_INSTANCE]
4472
    # Remove the temp. assignements for the instance's drbds
4473
    self.cfg.ReleaseDRBDMinors(instance)
4474
    # Unlock all the nodes
4475
    if self.op.mode == constants.INSTANCE_IMPORT:
4476
      nodes_keep = [self.op.src_node]
4477
      nodes_release = [node for node in self.acquired_locks[locking.LEVEL_NODE]
4478
                       if node != self.op.src_node]
4479
      self.context.glm.release(locking.LEVEL_NODE, nodes_release)
4480
      self.acquired_locks[locking.LEVEL_NODE] = nodes_keep
4481
    else:
4482
      self.context.glm.release(locking.LEVEL_NODE)
4483
      del self.acquired_locks[locking.LEVEL_NODE]
4484

    
4485
    if self.op.wait_for_sync:
4486
      disk_abort = not _WaitForSync(self, iobj)
4487
    elif iobj.disk_template in constants.DTS_NET_MIRROR:
4488
      # make sure the disks are not degraded (still sync-ing is ok)
4489
      time.sleep(15)
4490
      feedback_fn("* checking mirrors status")
4491
      disk_abort = not _WaitForSync(self, iobj, oneshot=True)
4492
    else:
4493
      disk_abort = False
4494

    
4495
    if disk_abort:
4496
      _RemoveDisks(self, iobj)
4497
      self.cfg.RemoveInstance(iobj.name)
4498
      # Make sure the instance lock gets removed
4499
      self.remove_locks[locking.LEVEL_INSTANCE] = iobj.name
4500
      raise errors.OpExecError("There are some degraded disks for"
4501
                               " this instance")
4502

    
4503
    feedback_fn("creating os for instance %s on node %s" %
4504
                (instance, pnode_name))
4505

    
4506
    if iobj.disk_template != constants.DT_DISKLESS:
4507
      if self.op.mode == constants.INSTANCE_CREATE:
4508
        feedback_fn("* running the instance OS create scripts...")
4509
        result = self.rpc.call_instance_os_add(pnode_name, iobj)
4510
        result.Raise()
4511
        if not result.data:
4512
          raise errors.OpExecError("Could not add os for instance %s"
4513
                                   " on node %s" %
4514
                                   (instance, pnode_name))
4515

    
4516
      elif self.op.mode == constants.INSTANCE_IMPORT:
4517
        feedback_fn("* running the instance OS import scripts...")
4518
        src_node = self.op.src_node
4519
        src_images = self.src_images
4520
        cluster_name = self.cfg.GetClusterName()
4521
        import_result = self.rpc.call_instance_os_import(pnode_name, iobj,
4522
                                                         src_node, src_images,
4523
                                                         cluster_name)
4524
        import_result.Raise()
4525
        for idx, result in enumerate(import_result.data):
4526
          if not result:
4527
            self.LogWarning("Could not import the image %s for instance"
4528
                            " %s, disk %d, on node %s" %
4529
                            (src_images[idx], instance, idx, pnode_name))
4530
      else:
4531
        # also checked in the prereq part
4532
        raise errors.ProgrammerError("Unknown OS initialization mode '%s'"
4533
                                     % self.op.mode)
4534

    
4535
    if self.op.start:
4536
      logging.info("Starting instance %s on node %s", instance, pnode_name)
4537
      feedback_fn("* starting instance...")
4538
      result = self.rpc.call_instance_start(pnode_name, iobj, None)
4539
      result.Raise()
4540
      if not result.data:
4541
        raise errors.OpExecError("Could not start instance")
4542

    
4543

    
4544
class LUConnectConsole(NoHooksLU):
4545
  """Connect to an instance's console.
4546

4547
  This is somewhat special in that it returns the command line that
4548
  you need to run on the master node in order to connect to the
4549
  console.
4550

4551
  """
4552
  _OP_REQP = ["instance_name"]
4553
  REQ_BGL = False
4554

    
4555
  def ExpandNames(self):
4556
    self._ExpandAndLockInstance()
4557

    
4558
  def CheckPrereq(self):
4559
    """Check prerequisites.
4560

4561
    This checks that the instance is in the cluster.
4562

4563
    """
4564
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_name)
4565
    assert self.instance is not None, \
4566
      "Cannot retrieve locked instance %s" % self.op.instance_name
4567
    _CheckNodeOnline(self, self.instance.primary_node)
4568

    
4569
  def Exec(self, feedback_fn):
4570
    """Connect to the console of an instance
4571

4572
    """
4573
    instance = self.instance
4574
    node = instance.primary_node
4575

    
4576
    node_insts = self.rpc.call_instance_list([node],
4577
                                             [instance.hypervisor])[node]
4578
    node_insts.Raise()
4579

    
4580
    if instance.name not in node_insts.data:
4581
      raise errors.OpExecError("Instance %s is not running." % instance.name)
4582

    
4583
    logging.debug("Connecting to console of %s on %s", instance.name, node)
4584

    
4585
    hyper = hypervisor.GetHypervisor(instance.hypervisor)
4586
    console_cmd = hyper.GetShellCommandForConsole(instance)
4587

    
4588
    # build ssh cmdline
4589
    return self.ssh.BuildCmd(node, "root", console_cmd, batch=True, tty=True)
4590

    
4591

    
4592
class LUReplaceDisks(LogicalUnit):
4593
  """Replace the disks of an instance.
4594

4595
  """
4596
  HPATH = "mirrors-replace"
4597
  HTYPE = constants.HTYPE_INSTANCE
4598
  _OP_REQP = ["instance_name", "mode", "disks"]
4599
  REQ_BGL = False
4600

    
4601
  def CheckArguments(self):
4602
    if not hasattr(self.op, "remote_node"):
4603
      self.op.remote_node = None
4604
    if not hasattr(self.op, "iallocator"):
4605
      self.op.iallocator = None
4606

    
4607
    # check for valid parameter combination
4608
    cnt = [self.op.remote_node, self.op.iallocator].count(None)
4609
    if self.op.mode == constants.REPLACE_DISK_CHG:
4610
      if cnt == 2:
4611
        raise errors.OpPrereqError("When changing the secondary either an"
4612
                                   " iallocator script must be used or the"
4613
                                   " new node given")
4614
      elif cnt == 0:
4615
        raise errors.OpPrereqError("Give either the iallocator or the new"
4616
                                   " secondary, not both")
4617
    else: # not replacing the secondary
4618
      if cnt != 2:
4619
        raise errors.OpPrereqError("The iallocator and new node options can"
4620
                                   " be used only when changing the"
4621
                                   " secondary node")
4622

    
4623
  def ExpandNames(self):
4624
    self._ExpandAndLockInstance()
4625

    
4626
    if self.op.iallocator is not None:
4627
      self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
4628
    elif self.op.remote_node is not None:
4629
      remote_node = self.cfg.ExpandNodeName(self.op.remote_node)
4630
      if remote_node is None:
4631
        raise errors.OpPrereqError("Node '%s' not known" %
4632
                                   self.op.remote_node)
4633
      self.op.remote_node = remote_node
4634
      self.needed_locks[locking.LEVEL_NODE] = [remote_node]
4635
      self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_APPEND
4636
    else:
4637
      self.needed_locks[locking.LEVEL_NODE] = []
4638
      self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
4639

    
4640
  def DeclareLocks(self, level):
4641
    # If we're not already locking all nodes in the set we have to declare the
4642
    # instance's primary/secondary nodes.
4643
    if (level == locking.LEVEL_NODE and
4644
        self.needed_locks[locking.LEVEL_NODE] is not locking.ALL_SET):
4645
      self._LockInstancesNodes()
4646

    
4647
  def _RunAllocator(self):
4648
    """Compute a new secondary node using an IAllocator.
4649

4650
    """
4651
    ial = IAllocator(self,
4652
                     mode=constants.IALLOCATOR_MODE_RELOC,
4653
                     name=self.op.instance_name,
4654
                     relocate_from=[self.sec_node])
4655

    
4656
    ial.Run(self.op.iallocator)
4657

    
4658
    if not ial.success:
4659
      raise errors.OpPrereqError("Can't compute nodes using"
4660
                                 " iallocator '%s': %s" % (self.op.iallocator,
4661
                                                           ial.info))
4662
    if len(ial.nodes) != ial.required_nodes:
4663
      raise errors.OpPrereqError("iallocator '%s' returned invalid number"
4664
                                 " of nodes (%s), required %s" %
4665
                                 (len(ial.nodes), ial.required_nodes))
4666
    self.op.remote_node = ial.nodes[0]
4667
    self.LogInfo("Selected new secondary for the instance: %s",
4668
                 self.op.remote_node)
4669

    
4670
  def BuildHooksEnv(self):
4671
    """Build hooks env.
4672

4673
    This runs on the master, the primary and all the secondaries.
4674

4675
    """
4676
    env = {
4677
      "MODE": self.op.mode,
4678
      "NEW_SECONDARY": self.op.remote_node,
4679
      "OLD_SECONDARY": self.instance.secondary_nodes[0],
4680
      }
4681
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
4682
    nl = [
4683
      self.cfg.GetMasterNode(),
4684
      self.instance.primary_node,
4685
      ]
4686
    if self.op.remote_node is not None:
4687
      nl.append(self.op.remote_node)
4688
    return env, nl, nl
4689

    
4690
  def CheckPrereq(self):
4691
    """Check prerequisites.
4692

4693
    This checks that the instance is in the cluster.
4694

4695
    """
4696
    instance = self.cfg.GetInstanceInfo(self.op.instance_name)
4697
    assert instance is not None, \
4698
      "Cannot retrieve locked instance %s" % self.op.instance_name
4699
    self.instance = instance
4700

    
4701
    if instance.disk_template != constants.DT_DRBD8:
4702
      raise errors.OpPrereqError("Can only run replace disks for DRBD8-based"
4703
                                 " instances")
4704

    
4705
    if len(instance.secondary_nodes) != 1:
4706
      raise errors.OpPrereqError("The instance has a strange layout,"
4707
                                 " expected one secondary but found %d" %
4708
                                 len(instance.secondary_nodes))
4709

    
4710
    self.sec_node = instance.secondary_nodes[0]
4711

    
4712
    if self.op.iallocator is not None:
4713
      self._RunAllocator()
4714

    
4715
    remote_node = self.op.remote_node
4716
    if remote_node is not None:
4717
      self.remote_node_info = self.cfg.GetNodeInfo(remote_node)
4718
      assert self.remote_node_info is not None, \
4719
        "Cannot retrieve locked node %s" % remote_node
4720
    else:
4721
      self.remote_node_info = None
4722
    if remote_node == instance.primary_node:
4723
      raise errors.OpPrereqError("The specified node is the primary node of"
4724
                                 " the instance.")
4725
    elif remote_node == self.sec_node:
4726
      raise errors.OpPrereqError("The specified node is already the"
4727
                                 " secondary node of the instance.")
4728

    
4729
    if self.op.mode == constants.REPLACE_DISK_PRI:
4730
      n1 = self.tgt_node = instance.primary_node
4731
      n2 = self.oth_node = self.sec_node
4732
    elif self.op.mode == constants.REPLACE_DISK_SEC:
4733
      n1 = self.tgt_node = self.sec_node
4734
      n2 = self.oth_node = instance.primary_node
4735
    elif self.op.mode == constants.REPLACE_DISK_CHG:
4736
      n1 = self.new_node = remote_node
4737
      n2 = self.oth_node = instance.primary_node
4738
      self.tgt_node = self.sec_node
4739
    else:
4740
      raise errors.ProgrammerError("Unhandled disk replace mode")
4741

    
4742
    _CheckNodeOnline(self, n1)
4743
    _CheckNodeOnline(self, n2)
4744

    
4745
    if not self.op.disks:
4746
      self.op.disks = range(len(instance.disks))
4747

    
4748
    for disk_idx in self.op.disks:
4749
      instance.FindDisk(disk_idx)
4750

    
4751
  def _ExecD8DiskOnly(self, feedback_fn):
4752
    """Replace a disk on the primary or secondary for dbrd8.
4753

4754
    The algorithm for replace is quite complicated:
4755

4756
      1. for each disk to be replaced:
4757

4758
        1. create new LVs on the target node with unique names
4759
        1. detach old LVs from the drbd device
4760
        1. rename old LVs to name_replaced.<time_t>
4761
        1. rename new LVs to old LVs
4762
        1. attach the new LVs (with the old names now) to the drbd device
4763

4764
      1. wait for sync across all devices
4765

4766
      1. for each modified disk:
4767

4768
        1. remove old LVs (which have the name name_replaces.<time_t>)
4769

4770
    Failures are not very well handled.
4771

4772
    """
4773
    steps_total = 6
4774
    warning, info = (self.proc.LogWarning, self.proc.LogInfo)
4775
    instance = self.instance
4776
    iv_names = {}
4777
    vgname = self.cfg.GetVGName()
4778
    # start of work
4779
    cfg = self.cfg
4780
    tgt_node = self.tgt_node
4781
    oth_node = self.oth_node
4782

    
4783
    # Step: check device activation
4784
    self.proc.LogStep(1, steps_total, "check device existence")
4785
    info("checking volume groups")
4786
    my_vg = cfg.GetVGName()
4787
    results = self.rpc.call_vg_list([oth_node, tgt_node])
4788
    if not results:
4789
      raise errors.OpExecError("Can't list volume groups on the nodes")
4790
    for node in oth_node, tgt_node:
4791
      res = results[node]
4792
      if res.failed or not res.data or my_vg not in res.data:
4793
        raise errors.OpExecError("Volume group '%s' not found on %s" %
4794
                                 (my_vg, node))
4795
    for idx, dev in enumerate(instance.disks):
4796
      if idx not in self.op.disks:
4797
        continue
4798
      for node in tgt_node, oth_node:
4799
        info("checking disk/%d on %s" % (idx, node))
4800
        cfg.SetDiskID(dev, node)
4801
        if not self.rpc.call_blockdev_find(node, dev):
4802
          raise errors.OpExecError("Can't find disk/%d on node %s" %
4803
                                   (idx, node))
4804

    
4805
    # Step: check other node consistency
4806
    self.proc.LogStep(2, steps_total, "check peer consistency")
4807
    for idx, dev in enumerate(instance.disks):
4808
      if idx not in self.op.disks:
4809
        continue
4810
      info("checking disk/%d consistency on %s" % (idx, oth_node))
4811
      if not _CheckDiskConsistency(self, dev, oth_node,
4812
                                   oth_node==instance.primary_node):
4813
        raise errors.OpExecError("Peer node (%s) has degraded storage, unsafe"
4814
                                 " to replace disks on this node (%s)" %
4815
                                 (oth_node, tgt_node))
4816

    
4817
    # Step: create new storage
4818
    self.proc.LogStep(3, steps_total, "allocate new storage")
4819
    for idx, dev in enumerate(instance.disks):
4820
      if idx not in self.op.disks:
4821
        continue
4822
      size = dev.size
4823
      cfg.SetDiskID(dev, tgt_node)
4824
      lv_names = [".disk%d_%s" % (idx, suf)
4825
                  for suf in ["data", "meta"]]
4826
      names = _GenerateUniqueNames(self, lv_names)
4827
      lv_data = objects.Disk(dev_type=constants.LD_LV, size=size,
4828
                             logical_id=(vgname, names[0]))
4829
      lv_meta = objects.Disk(dev_type=constants.LD_LV, size=128,
4830
                             logical_id=(vgname, names[1]))
4831
      new_lvs = [lv_data, lv_meta]
4832
      old_lvs = dev.children
4833
      iv_names[dev.iv_name] = (dev, old_lvs, new_lvs)
4834
      info("creating new local storage on %s for %s" %
4835
           (tgt_node, dev.iv_name))
4836
      # we pass force_create=True to force the LVM creation
4837
      for new_lv in new_lvs:
4838
        _CreateBlockDev(self, tgt_node, instance, new_lv, True,
4839
                        _GetInstanceInfoText(instance), False)
4840

    
4841
    # Step: for each lv, detach+rename*2+attach
4842
    self.proc.LogStep(4, steps_total, "change drbd configuration")
4843
    for dev, old_lvs, new_lvs in iv_names.itervalues():
4844
      info("detaching %s drbd from local storage" % dev.iv_name)
4845
      result = self.rpc.call_blockdev_removechildren(tgt_node, dev, old_lvs)
4846
      result.Raise()
4847
      if not result.data:
4848
        raise errors.OpExecError("Can't detach drbd from local storage on node"
4849
                                 " %s for device %s" % (tgt_node, dev.iv_name))
4850
      #dev.children = []
4851
      #cfg.Update(instance)
4852

    
4853
      # ok, we created the new LVs, so now we know we have the needed
4854
      # storage; as such, we proceed on the target node to rename
4855
      # old_lv to _old, and new_lv to old_lv; note that we rename LVs
4856
      # using the assumption that logical_id == physical_id (which in
4857
      # turn is the unique_id on that node)
4858

    
4859
      # FIXME(iustin): use a better name for the replaced LVs
4860
      temp_suffix = int(time.time())
4861
      ren_fn = lambda d, suff: (d.physical_id[0],
4862
                                d.physical_id[1] + "_replaced-%s" % suff)
4863
      # build the rename list based on what LVs exist on the node
4864
      rlist = []
4865
      for to_ren in old_lvs:
4866
        find_res = self.rpc.call_blockdev_find(tgt_node, to_ren)
4867
        if not find_res.failed and find_res.data is not None: # device exists
4868
          rlist.append((to_ren, ren_fn(to_ren, temp_suffix)))
4869

    
4870
      info("renaming the old LVs on the target node")
4871
      result = self.rpc.call_blockdev_rename(tgt_node, rlist)
4872
      result.Raise()
4873
      if not result.data:
4874
        raise errors.OpExecError("Can't rename old LVs on node %s" % tgt_node)
4875
      # now we rename the new LVs to the old LVs
4876
      info("renaming the new LVs on the target node")
4877
      rlist = [(new, old.physical_id) for old, new in zip(old_lvs, new_lvs)]
4878
      result = self.rpc.call_blockdev_rename(tgt_node, rlist)
4879
      result.Raise()
4880
      if not result.data:
4881
        raise errors.OpExecError("Can't rename new LVs on node %s" % tgt_node)
4882

    
4883
      for old, new in zip(old_lvs, new_lvs):
4884
        new.logical_id = old.logical_id
4885
        cfg.SetDiskID(new, tgt_node)
4886

    
4887
      for disk in old_lvs:
4888
        disk.logical_id = ren_fn(disk, temp_suffix)
4889
        cfg.SetDiskID(disk, tgt_node)
4890

    
4891
      # now that the new lvs have the old name, we can add them to the device
4892
      info("adding new mirror component on %s" % tgt_node)
4893
      result = self.rpc.call_blockdev_addchildren(tgt_node, dev, new_lvs)
4894
      if result.failed or not result.data:
4895
        for new_lv in new_lvs:
4896
          result = self.rpc.call_blockdev_remove(tgt_node, new_lv)
4897
          if result.failed or not result.data:
4898
            warning("Can't rollback device %s", hint="manually cleanup unused"
4899
                    " logical volumes")
4900
        raise errors.OpExecError("Can't add local storage to drbd")
4901

    
4902
      dev.children = new_lvs
4903
      cfg.Update(instance)
4904

    
4905
    # Step: wait for sync
4906

    
4907
    # this can fail as the old devices are degraded and _WaitForSync
4908
    # does a combined result over all disks, so we don't check its
4909
    # return value
4910
    self.proc.LogStep(5, steps_total, "sync devices")
4911
    _WaitForSync(self, instance, unlock=True)
4912

    
4913
    # so check manually all the devices
4914
    for name, (dev, old_lvs, new_lvs) in iv_names.iteritems():
4915
      cfg.SetDiskID(dev, instance.primary_node)
4916
      result = self.rpc.call_blockdev_find(instance.primary_node, dev)
4917
      if result.failed or result.data[5]:
4918
        raise errors.OpExecError("DRBD device %s is degraded!" % name)
4919

    
4920
    # Step: remove old storage
4921
    self.proc.LogStep(6, steps_total, "removing old storage")
4922
    for name, (dev, old_lvs, new_lvs) in iv_names.iteritems():
4923
      info("remove logical volumes for %s" % name)
4924
      for lv in old_lvs:
4925
        cfg.SetDiskID(lv, tgt_node)
4926
        result = self.rpc.call_blockdev_remove(tgt_node, lv)
4927
        if result.failed or not result.data:
4928
          warning("Can't remove old LV", hint="manually remove unused LVs")
4929
          continue
4930

    
4931
  def _ExecD8Secondary(self, feedback_fn):
4932
    """Replace the secondary node for drbd8.
4933

4934
    The algorithm for replace is quite complicated:
4935
      - for all disks of the instance:
4936
        - create new LVs on the new node with same names
4937
        - shutdown the drbd device on the old secondary
4938
        - disconnect the drbd network on the primary
4939
        - create the drbd device on the new secondary
4940
        - network attach the drbd on the primary, using an artifice:
4941
          the drbd code for Attach() will connect to the network if it
4942
          finds a device which is connected to the good local disks but
4943
          not network enabled
4944
      - wait for sync across all devices
4945
      - remove all disks from the old secondary
4946

4947
    Failures are not very well handled.
4948

4949
    """
4950
    steps_total = 6
4951
    warning, info = (self.proc.LogWarning, self.proc.LogInfo)
4952
    instance = self.instance
4953
    iv_names = {}
4954
    # start of work
4955
    cfg = self.cfg
4956
    old_node = self.tgt_node
4957
    new_node = self.new_node
4958
    pri_node = instance.primary_node
4959
    nodes_ip = {
4960
      old_node: self.cfg.GetNodeInfo(old_node).secondary_ip,
4961
      new_node: self.cfg.GetNodeInfo(new_node).secondary_ip,
4962
      pri_node: self.cfg.GetNodeInfo(pri_node).secondary_ip,
4963
      }
4964

    
4965
    # Step: check device activation
4966
    self.proc.LogStep(1, steps_total, "check device existence")
4967
    info("checking volume groups")
4968
    my_vg = cfg.GetVGName()
4969
    results = self.rpc.call_vg_list([pri_node, new_node])
4970
    for node in pri_node, new_node:
4971
      res = results[node]
4972
      if res.failed or not res.data or my_vg not in res.data:
4973
        raise errors.OpExecError("Volume group '%s' not found on %s" %
4974
                                 (my_vg, node))
4975
    for idx, dev in enumerate(instance.disks):
4976
      if idx not in self.op.disks:
4977
        continue
4978
      info("checking disk/%d on %s" % (idx, pri_node))
4979
      cfg.SetDiskID(dev, pri_node)
4980
      result = self.rpc.call_blockdev_find(pri_node, dev)
4981
      result.Raise()
4982
      if not result.data:
4983
        raise errors.OpExecError("Can't find disk/%d on node %s" %
4984
                                 (idx, pri_node))
4985

    
4986
    # Step: check other node consistency
4987
    self.proc.LogStep(2, steps_total, "check peer consistency")
4988
    for idx, dev in enumerate(instance.disks):
4989
      if idx not in self.op.disks:
4990
        continue
4991
      info("checking disk/%d consistency on %s" % (idx, pri_node))
4992
      if not _CheckDiskConsistency(self, dev, pri_node, True, ldisk=True):
4993
        raise errors.OpExecError("Primary node (%s) has degraded storage,"
4994
                                 " unsafe to replace the secondary" %
4995
                                 pri_node)
4996

    
4997
    # Step: create new storage
4998
    self.proc.LogStep(3, steps_total, "allocate new storage")
4999
    for idx, dev in enumerate(instance.disks):
5000
      info("adding new local storage on %s for disk/%d" %
5001
           (new_node, idx))
5002
      # we pass force_create=True to force LVM creation
5003
      for new_lv in dev.children:
5004
        _CreateBlockDev(self, new_node, instance, new_lv, True,
5005
                        _GetInstanceInfoText(instance), False)
5006

    
5007
    # Step 4: dbrd minors and drbd setups changes
5008
    # after this, we must manually remove the drbd minors on both the
5009
    # error and the success paths
5010
    minors = cfg.AllocateDRBDMinor([new_node for dev in instance.disks],
5011
                                   instance.name)
5012
    logging.debug("Allocated minors %s" % (minors,))
5013
    self.proc.LogStep(4, steps_total, "changing drbd configuration")
5014
    for idx, (dev, new_minor) in enumerate(zip(instance.disks, minors)):
5015
      size = dev.size
5016
      info("activating a new drbd on %s for disk/%d" % (new_node, idx))
5017
      # create new devices on new_node; note that we create two IDs:
5018
      # one without port, so the drbd will be activated without
5019
      # networking information on the new node at this stage, and one
5020
      # with network, for the latter activation in step 4
5021
      (o_node1, o_node2, o_port, o_minor1, o_minor2, o_secret) = dev.logical_id
5022
      if pri_node == o_node1:
5023
        p_minor = o_minor1
5024
      else:
5025
        p_minor = o_minor2
5026

    
5027
      new_alone_id = (pri_node, new_node, None, p_minor, new_minor, o_secret)
5028
      new_net_id = (pri_node, new_node, o_port, p_minor, new_minor, o_secret)
5029

    
5030
      iv_names[idx] = (dev, dev.children, new_net_id)
5031
      logging.debug("Allocated new_minor: %s, new_logical_id: %s", new_minor,
5032
                    new_net_id)
5033
      new_drbd = objects.Disk(dev_type=constants.LD_DRBD8,
5034
                              logical_id=new_alone_id,
5035
                              children=dev.children)
5036
      try:
5037
        _CreateSingleBlockDev(self, new_node, instance, new_drbd,
5038
                              _GetInstanceInfoText(instance), False)
5039
      except error.BlockDeviceError:
5040
        self.cfg.ReleaseDRBDMinors(instance.name)
5041
        raise
5042

    
5043
    for idx, dev in enumerate(instance.disks):
5044
      # we have new devices, shutdown the drbd on the old secondary
5045
      info("shutting down drbd for disk/%d on old node" % idx)
5046
      cfg.SetDiskID(dev, old_node)
5047
      result = self.rpc.call_blockdev_shutdown(old_node, dev)
5048
      if result.failed or not result.data:
5049
        warning("Failed to shutdown drbd for disk/%d on old node" % idx,
5050
                hint="Please cleanup this device manually as soon as possible")
5051

    
5052
    info("detaching primary drbds from the network (=> standalone)")
5053
    result = self.rpc.call_drbd_disconnect_net([pri_node], nodes_ip,
5054
                                               instance.disks)[pri_node]
5055

    
5056
    msg = result.RemoteFailMsg()
5057
    if msg:
5058
      # detaches didn't succeed (unlikely)
5059
      self.cfg.ReleaseDRBDMinors(instance.name)
5060
      raise errors.OpExecError("Can't detach the disks from the network on"
5061
                               " old node: %s" % (msg,))
5062

    
5063
    # if we managed to detach at least one, we update all the disks of
5064
    # the instance to point to the new secondary
5065
    info("updating instance configuration")
5066
    for dev, _, new_logical_id in iv_names.itervalues():
5067
      dev.logical_id = new_logical_id
5068
      cfg.SetDiskID(dev, pri_node)
5069
    cfg.Update(instance)
5070
    # we can remove now the temp minors as now the new values are
5071
    # written to the config file (and therefore stable)
5072
    self.cfg.ReleaseDRBDMinors(instance.name)
5073

    
5074
    # and now perform the drbd attach
5075
    info("attaching primary drbds to new secondary (standalone => connected)")
5076
    result = self.rpc.call_drbd_attach_net([pri_node, new_node], nodes_ip,
5077
                                           instance.disks, instance.name,
5078
                                           False)
5079
    for to_node, to_result in result.items():
5080
      msg = to_result.RemoteFailMsg()
5081
      if msg:
5082
        warning("can't attach drbd disks on node %s: %s", to_node, msg,
5083
                hint="please do a gnt-instance info to see the"
5084
                " status of disks")
5085

    
5086
    # this can fail as the old devices are degraded and _WaitForSync
5087
    # does a combined result over all disks, so we don't check its
5088
    # return value
5089
    self.proc.LogStep(5, steps_total, "sync devices")
5090
    _WaitForSync(self, instance, unlock=True)
5091

    
5092
    # so check manually all the devices
5093
    for idx, (dev, old_lvs, _) in iv_names.iteritems():
5094
      cfg.SetDiskID(dev, pri_node)
5095
      result = self.rpc.call_blockdev_find(pri_node, dev)
5096
      result.Raise()
5097
      if result.data[5]:
5098
        raise errors.OpExecError("DRBD device disk/%d is degraded!" % idx)
5099

    
5100
    self.proc.LogStep(6, steps_total, "removing old storage")
5101
    for idx, (dev, old_lvs, _) in iv_names.iteritems():
5102
      info("remove logical volumes for disk/%d" % idx)
5103
      for lv in old_lvs:
5104
        cfg.SetDiskID(lv, old_node)
5105
        result = self.rpc.call_blockdev_remove(old_node, lv)
5106
        if result.failed or not result.data:
5107
          warning("Can't remove LV on old secondary",
5108
                  hint="Cleanup stale volumes by hand")
5109

    
5110
  def Exec(self, feedback_fn):
5111
    """Execute disk replacement.
5112

5113
    This dispatches the disk replacement to the appropriate handler.
5114

5115
    """
5116
    instance = self.instance
5117

    
5118
    # Activate the instance disks if we're replacing them on a down instance
5119
    if instance.status == "down":
5120
      _StartInstanceDisks(self, instance, True)
5121

    
5122
    if self.op.mode == constants.REPLACE_DISK_CHG:
5123
      fn = self._ExecD8Secondary
5124
    else:
5125
      fn = self._ExecD8DiskOnly
5126

    
5127
    ret = fn(feedback_fn)
5128

    
5129
    # Deactivate the instance disks if we're replacing them on a down instance
5130
    if instance.status == "down":
5131
      _SafeShutdownInstanceDisks(self, instance)
5132

    
5133
    return ret
5134

    
5135

    
5136
class LUGrowDisk(LogicalUnit):
5137
  """Grow a disk of an instance.
5138

5139
  """
5140
  HPATH = "disk-grow"
5141
  HTYPE = constants.HTYPE_INSTANCE
5142
  _OP_REQP = ["instance_name", "disk", "amount", "wait_for_sync"]
5143
  REQ_BGL = False
5144

    
5145
  def ExpandNames(self):
5146
    self._ExpandAndLockInstance()
5147
    self.needed_locks[locking.LEVEL_NODE] = []
5148
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
5149

    
5150
  def DeclareLocks(self, level):
5151
    if level == locking.LEVEL_NODE:
5152
      self._LockInstancesNodes()
5153

    
5154
  def BuildHooksEnv(self):
5155
    """Build hooks env.
5156

5157
    This runs on the master, the primary and all the secondaries.
5158

5159
    """
5160
    env = {
5161
      "DISK": self.op.disk,
5162
      "AMOUNT": self.op.amount,
5163
      }
5164
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
5165
    nl = [
5166
      self.cfg.GetMasterNode(),
5167
      self.instance.primary_node,
5168
      ]
5169
    return env, nl, nl
5170

    
5171
  def CheckPrereq(self):
5172
    """Check prerequisites.
5173

5174
    This checks that the instance is in the cluster.
5175

5176
    """
5177
    instance = self.cfg.GetInstanceInfo(self.op.instance_name)
5178
    assert instance is not None, \
5179
      "Cannot retrieve locked instance %s" % self.op.instance_name
5180
    nodenames = list(instance.all_nodes)
5181
    for node in nodenames:
5182
      _CheckNodeOnline(self, node)
5183

    
5184

    
5185
    self.instance = instance
5186

    
5187
    if instance.disk_template not in (constants.DT_PLAIN, constants.DT_DRBD8):
5188
      raise errors.OpPrereqError("Instance's disk layout does not support"
5189
                                 " growing.")
5190

    
5191
    self.disk = instance.FindDisk(self.op.disk)
5192

    
5193
    nodeinfo = self.rpc.call_node_info(nodenames, self.cfg.GetVGName(),
5194
                                       instance.hypervisor)
5195
    for node in nodenames:
5196
      info = nodeinfo[node]
5197
      if info.failed or not info.data:
5198
        raise errors.OpPrereqError("Cannot get current information"
5199
                                   " from node '%s'" % node)
5200
      vg_free = info.data.get('vg_free', None)
5201
      if not isinstance(vg_free, int):
5202
        raise errors.OpPrereqError("Can't compute free disk space on"
5203
                                   " node %s" % node)
5204
      if self.op.amount > vg_free:
5205
        raise errors.OpPrereqError("Not enough disk space on target node %s:"
5206
                                   " %d MiB available, %d MiB required" %
5207
                                   (node, vg_free, self.op.amount))
5208

    
5209
  def Exec(self, feedback_fn):
5210
    """Execute disk grow.
5211

5212
    """
5213
    instance = self.instance
5214
    disk = self.disk
5215
    for node in instance.all_nodes:
5216
      self.cfg.SetDiskID(disk, node)
5217
      result = self.rpc.call_blockdev_grow(node, disk, self.op.amount)
5218
      result.Raise()
5219
      if (not result.data or not isinstance(result.data, (list, tuple)) or
5220
          len(result.data) != 2):
5221
        raise errors.OpExecError("Grow request failed to node %s" % node)
5222
      elif not result.data[0]:
5223
        raise errors.OpExecError("Grow request failed to node %s: %s" %
5224
                                 (node, result.data[1]))
5225
    disk.RecordGrow(self.op.amount)
5226
    self.cfg.Update(instance)
5227
    if self.op.wait_for_sync:
5228
      disk_abort = not _WaitForSync(self, instance)
5229
      if disk_abort:
5230
        self.proc.LogWarning("Warning: disk sync-ing has not returned a good"
5231
                             " status.\nPlease check the instance.")
5232

    
5233

    
5234
class LUQueryInstanceData(NoHooksLU):
5235
  """Query runtime instance data.
5236

5237
  """
5238
  _OP_REQP = ["instances", "static"]
5239
  REQ_BGL = False
5240

    
5241
  def ExpandNames(self):
5242
    self.needed_locks = {}
5243
    self.share_locks = dict(((i, 1) for i in locking.LEVELS))
5244

    
5245
    if not isinstance(self.op.instances, list):
5246
      raise errors.OpPrereqError("Invalid argument type 'instances'")
5247

    
5248
    if self.op.instances:
5249
      self.wanted_names = []
5250
      for name in self.op.instances:
5251
        full_name = self.cfg.ExpandInstanceName(name)
5252
        if full_name is None:
5253
          raise errors.OpPrereqError("Instance '%s' not known" % name)
5254
        self.wanted_names.append(full_name)
5255
      self.needed_locks[locking.LEVEL_INSTANCE] = self.wanted_names
5256
    else:
5257
      self.wanted_names = None
5258
      self.needed_locks[locking.LEVEL_INSTANCE] = locking.ALL_SET
5259

    
5260
    self.needed_locks[locking.LEVEL_NODE] = []
5261
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
5262

    
5263
  def DeclareLocks(self, level):
5264
    if level == locking.LEVEL_NODE:
5265
      self._LockInstancesNodes()
5266

    
5267
  def CheckPrereq(self):
5268
    """Check prerequisites.
5269

5270
    This only checks the optional instance list against the existing names.
5271

5272
    """
5273
    if self.wanted_names is None:
5274
      self.wanted_names = self.acquired_locks[locking.LEVEL_INSTANCE]
5275

    
5276
    self.wanted_instances = [self.cfg.GetInstanceInfo(name) for name
5277
                             in self.wanted_names]
5278
    return
5279

    
5280
  def _ComputeDiskStatus(self, instance, snode, dev):
5281
    """Compute block device status.
5282

5283
    """
5284
    static = self.op.static
5285
    if not static:
5286
      self.cfg.SetDiskID(dev, instance.primary_node)
5287
      dev_pstatus = self.rpc.call_blockdev_find(instance.primary_node, dev)
5288
      dev_pstatus.Raise()
5289
      dev_pstatus = dev_pstatus.data
5290
    else:
5291
      dev_pstatus = None
5292

    
5293
    if dev.dev_type in constants.LDS_DRBD:
5294
      # we change the snode then (otherwise we use the one passed in)
5295
      if dev.logical_id[0] == instance.primary_node:
5296
        snode = dev.logical_id[1]
5297
      else:
5298
        snode = dev.logical_id[0]
5299

    
5300
    if snode and not static:
5301
      self.cfg.SetDiskID(dev, snode)
5302
      dev_sstatus = self.rpc.call_blockdev_find(snode, dev)
5303
      dev_sstatus.Raise()
5304
      dev_sstatus = dev_sstatus.data
5305
    else:
5306
      dev_sstatus = None
5307

    
5308
    if dev.children:
5309
      dev_children = [self._ComputeDiskStatus(instance, snode, child)
5310
                      for child in dev.children]
5311
    else:
5312
      dev_children = []
5313

    
5314
    data = {
5315
      "iv_name": dev.iv_name,
5316
      "dev_type": dev.dev_type,
5317
      "logical_id": dev.logical_id,
5318
      "physical_id": dev.physical_id,
5319
      "pstatus": dev_pstatus,
5320
      "sstatus": dev_sstatus,
5321
      "children": dev_children,
5322
      "mode": dev.mode,
5323
      }
5324

    
5325
    return data
5326

    
5327
  def Exec(self, feedback_fn):
5328
    """Gather and return data"""
5329
    result = {}
5330

    
5331
    cluster = self.cfg.GetClusterInfo()
5332

    
5333
    for instance in self.wanted_instances:
5334
      if not self.op.static:
5335
        remote_info = self.rpc.call_instance_info(instance.primary_node,
5336
                                                  instance.name,
5337
                                                  instance.hypervisor)
5338
        remote_info.Raise()
5339
        remote_info = remote_info.data
5340
        if remote_info and "state" in remote_info:
5341
          remote_state = "up"
5342
        else:
5343
          remote_state = "down"
5344
      else:
5345
        remote_state = None
5346
      if instance.status == "down":
5347
        config_state = "down"
5348
      else:
5349
        config_state = "up"
5350

    
5351
      disks = [self._ComputeDiskStatus(instance, None, device)
5352
               for device in instance.disks]
5353

    
5354
      idict = {
5355
        "name": instance.name,
5356
        "config_state": config_state,
5357
        "run_state": remote_state,
5358
        "pnode": instance.primary_node,
5359
        "snodes": instance.secondary_nodes,
5360
        "os": instance.os,
5361
        "nics": [(nic.mac, nic.ip, nic.bridge) for nic in instance.nics],
5362
        "disks": disks,
5363
        "hypervisor": instance.hypervisor,
5364
        "network_port": instance.network_port,
5365
        "hv_instance": instance.hvparams,
5366
        "hv_actual": cluster.FillHV(instance),
5367
        "be_instance": instance.beparams,
5368
        "be_actual": cluster.FillBE(instance),
5369
        }
5370

    
5371
      result[instance.name] = idict
5372

    
5373
    return result
5374

    
5375

    
5376
class LUSetInstanceParams(LogicalUnit):
5377
  """Modifies an instances's parameters.
5378

5379
  """
5380
  HPATH = "instance-modify"
5381
  HTYPE = constants.HTYPE_INSTANCE
5382
  _OP_REQP = ["instance_name"]
5383
  REQ_BGL = False
5384

    
5385
  def CheckArguments(self):
5386
    if not hasattr(self.op, 'nics'):
5387
      self.op.nics = []
5388
    if not hasattr(self.op, 'disks'):
5389
      self.op.disks = []
5390
    if not hasattr(self.op, 'beparams'):
5391
      self.op.beparams = {}
5392
    if not hasattr(self.op, 'hvparams'):
5393
      self.op.hvparams = {}
5394
    self.op.force = getattr(self.op, "force", False)
5395
    if not (self.op.nics or self.op.disks or
5396
            self.op.hvparams or self.op.beparams):
5397
      raise errors.OpPrereqError("No changes submitted")
5398

    
5399
    utils.CheckBEParams(self.op.beparams)
5400

    
5401
    # Disk validation
5402
    disk_addremove = 0
5403
    for disk_op, disk_dict in self.op.disks:
5404
      if disk_op == constants.DDM_REMOVE:
5405
        disk_addremove += 1
5406
        continue
5407
      elif disk_op == constants.DDM_ADD:
5408
        disk_addremove += 1
5409
      else:
5410
        if not isinstance(disk_op, int):
5411
          raise errors.OpPrereqError("Invalid disk index")
5412
      if disk_op == constants.DDM_ADD:
5413
        mode = disk_dict.setdefault('mode', constants.DISK_RDWR)
5414
        if mode not in (constants.DISK_RDONLY, constants.DISK_RDWR):
5415
          raise errors.OpPrereqError("Invalid disk access mode '%s'" % mode)
5416
        size = disk_dict.get('size', None)
5417
        if size is None:
5418
          raise errors.OpPrereqError("Required disk parameter size missing")
5419
        try:
5420
          size = int(size)
5421
        except ValueError, err:
5422
          raise errors.OpPrereqError("Invalid disk size parameter: %s" %
5423
                                     str(err))
5424
        disk_dict['size'] = size
5425
      else:
5426
        # modification of disk
5427
        if 'size' in disk_dict:
5428
          raise errors.OpPrereqError("Disk size change not possible, use"
5429
                                     " grow-disk")
5430

    
5431
    if disk_addremove > 1:
5432
      raise errors.OpPrereqError("Only one disk add or remove operation"
5433
                                 " supported at a time")
5434

    
5435
    # NIC validation
5436
    nic_addremove = 0
5437
    for nic_op, nic_dict in self.op.nics:
5438
      if nic_op == constants.DDM_REMOVE:
5439
        nic_addremove += 1
5440
        continue
5441
      elif nic_op == constants.DDM_ADD:
5442
        nic_addremove += 1
5443
      else:
5444
        if not isinstance(nic_op, int):
5445
          raise errors.OpPrereqError("Invalid nic index")
5446

    
5447
      # nic_dict should be a dict
5448
      nic_ip = nic_dict.get('ip', None)
5449
      if nic_ip is not None:
5450
        if nic_ip.lower() == "none":
5451
          nic_dict['ip'] = None
5452
        else:
5453
          if not utils.IsValidIP(nic_ip):
5454
            raise errors.OpPrereqError("Invalid IP address '%s'" % nic_ip)
5455
      # we can only check None bridges and assign the default one
5456
      nic_bridge = nic_dict.get('bridge', None)
5457
      if nic_bridge is None:
5458
        nic_dict['bridge'] = self.cfg.GetDefBridge()
5459
      # but we can validate MACs
5460
      nic_mac = nic_dict.get('mac', None)
5461
      if nic_mac is not None:
5462
        if self.cfg.IsMacInUse(nic_mac):
5463
          raise errors.OpPrereqError("MAC address %s already in use"
5464
                                     " in cluster" % nic_mac)
5465
        if nic_mac not in (constants.VALUE_AUTO, constants.VALUE_GENERATE):
5466
          if not utils.IsValidMac(nic_mac):
5467
            raise errors.OpPrereqError("Invalid MAC address %s" % nic_mac)
5468
    if nic_addremove > 1:
5469
      raise errors.OpPrereqError("Only one NIC add or remove operation"
5470
                                 " supported at a time")
5471

    
5472
  def ExpandNames(self):
5473
    self._ExpandAndLockInstance()
5474
    self.needed_locks[locking.LEVEL_NODE] = []
5475
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
5476

    
5477
  def DeclareLocks(self, level):
5478
    if level == locking.LEVEL_NODE:
5479
      self._LockInstancesNodes()
5480

    
5481
  def BuildHooksEnv(self):
5482
    """Build hooks env.
5483

5484
    This runs on the master, primary and secondaries.
5485

5486
    """
5487
    args = dict()
5488
    if constants.BE_MEMORY in self.be_new:
5489
      args['memory'] = self.be_new[constants.BE_MEMORY]
5490
    if constants.BE_VCPUS in self.be_new:
5491
      args['vcpus'] = self.be_new[constants.BE_VCPUS]
5492
    # FIXME: readd disk/nic changes
5493
    env = _BuildInstanceHookEnvByObject(self, self.instance, override=args)
5494
    nl = [self.cfg.GetMasterNode()] + list(self.instance.all_nodes)
5495
    return env, nl, nl
5496

    
5497
  def CheckPrereq(self):
5498
    """Check prerequisites.
5499

5500
    This only checks the instance list against the existing names.
5501

5502
    """
5503
    force = self.force = self.op.force
5504

    
5505
    # checking the new params on the primary/secondary nodes
5506

    
5507
    instance = self.instance = self.cfg.GetInstanceInfo(self.op.instance_name)
5508
    assert self.instance is not None, \
5509
      "Cannot retrieve locked instance %s" % self.op.instance_name
5510
    pnode = instance.primary_node
5511
    nodelist = list(instance.all_nodes)
5512

    
5513
    # hvparams processing
5514
    if self.op.hvparams:
5515
      i_hvdict = copy.deepcopy(instance.hvparams)
5516
      for key, val in self.op.hvparams.iteritems():
5517
        if val == constants.VALUE_DEFAULT:
5518
          try:
5519
            del i_hvdict[key]
5520
          except KeyError:
5521
            pass
5522
        elif val == constants.VALUE_NONE:
5523
          i_hvdict[key] = None
5524
        else:
5525
          i_hvdict[key] = val
5526
      cluster = self.cfg.GetClusterInfo()
5527
      hv_new = cluster.FillDict(cluster.hvparams[instance.hypervisor],
5528
                                i_hvdict)
5529
      # local check
5530
      hypervisor.GetHypervisor(
5531
        instance.hypervisor).CheckParameterSyntax(hv_new)
5532
      _CheckHVParams(self, nodelist, instance.hypervisor, hv_new)
5533
      self.hv_new = hv_new # the new actual values
5534
      self.hv_inst = i_hvdict # the new dict (without defaults)
5535
    else:
5536
      self.hv_new = self.hv_inst = {}
5537

    
5538
    # beparams processing
5539
    if self.op.beparams:
5540
      i_bedict = copy.deepcopy(instance.beparams)
5541
      for key, val in self.op.beparams.iteritems():
5542
        if val == constants.VALUE_DEFAULT:
5543
          try:
5544
            del i_bedict[key]
5545
          except KeyError:
5546
            pass
5547
        else:
5548
          i_bedict[key] = val
5549
      cluster = self.cfg.GetClusterInfo()
5550
      be_new = cluster.FillDict(cluster.beparams[constants.BEGR_DEFAULT],
5551
                                i_bedict)
5552
      self.be_new = be_new # the new actual values
5553
      self.be_inst = i_bedict # the new dict (without defaults)
5554
    else:
5555
      self.be_new = self.be_inst = {}
5556

    
5557
    self.warn = []
5558

    
5559
    if constants.BE_MEMORY in self.op.beparams and not self.force:
5560
      mem_check_list = [pnode]
5561
      if be_new[constants.BE_AUTO_BALANCE]:
5562
        # either we changed auto_balance to yes or it was from before
5563
        mem_check_list.extend(instance.secondary_nodes)
5564
      instance_info = self.rpc.call_instance_info(pnode, instance.name,
5565
                                                  instance.hypervisor)
5566
      nodeinfo = self.rpc.call_node_info(mem_check_list, self.cfg.GetVGName(),
5567
                                         instance.hypervisor)
5568
      if nodeinfo[pnode].failed or not isinstance(nodeinfo[pnode].data, dict):
5569
        # Assume the primary node is unreachable and go ahead
5570
        self.warn.append("Can't get info from primary node %s" % pnode)
5571
      else:
5572
        if not instance_info.failed and instance_info.data:
5573
          current_mem = instance_info.data['memory']
5574
        else:
5575
          # Assume instance not running
5576
          # (there is a slight race condition here, but it's not very probable,
5577
          # and we have no other way to check)
5578
          current_mem = 0
5579
        miss_mem = (be_new[constants.BE_MEMORY] - current_mem -
5580
                    nodeinfo[pnode].data['memory_free'])
5581
        if miss_mem > 0:
5582
          raise errors.OpPrereqError("This change will prevent the instance"
5583
                                     " from starting, due to %d MB of memory"
5584
                                     " missing on its primary node" % miss_mem)
5585

    
5586
      if be_new[constants.BE_AUTO_BALANCE]:
5587
        for node, nres in nodeinfo.iteritems():
5588
          if node not in instance.secondary_nodes:
5589
            continue
5590
          if nres.failed or not isinstance(nres.data, dict):
5591
            self.warn.append("Can't get info from secondary node %s" % node)
5592
          elif be_new[constants.BE_MEMORY] > nres.data['memory_free']:
5593
            self.warn.append("Not enough memory to failover instance to"
5594
                             " secondary node %s" % node)
5595

    
5596
    # NIC processing
5597
    for nic_op, nic_dict in self.op.nics:
5598
      if nic_op == constants.DDM_REMOVE:
5599
        if not instance.nics:
5600
          raise errors.OpPrereqError("Instance has no NICs, cannot remove")
5601
        continue
5602
      if nic_op != constants.DDM_ADD:
5603
        # an existing nic
5604
        if nic_op < 0 or nic_op >= len(instance.nics):
5605
          raise errors.OpPrereqError("Invalid NIC index %s, valid values"
5606
                                     " are 0 to %d" %
5607
                                     (nic_op, len(instance.nics)))
5608
      nic_bridge = nic_dict.get('bridge', None)
5609
      if nic_bridge is not None:
5610
        if not self.rpc.call_bridges_exist(pnode, [nic_bridge]):
5611
          msg = ("Bridge '%s' doesn't exist on one of"
5612
                 " the instance nodes" % nic_bridge)
5613
          if self.force:
5614
            self.warn.append(msg)
5615
          else:
5616
            raise errors.OpPrereqError(msg)
5617

    
5618
    # DISK processing
5619
    if self.op.disks and instance.disk_template == constants.DT_DISKLESS:
5620
      raise errors.OpPrereqError("Disk operations not supported for"
5621
                                 " diskless instances")
5622
    for disk_op, disk_dict in self.op.disks:
5623
      if disk_op == constants.DDM_REMOVE:
5624
        if len(instance.disks) == 1:
5625
          raise errors.OpPrereqError("Cannot remove the last disk of"
5626
                                     " an instance")
5627
        ins_l = self.rpc.call_instance_list([pnode], [instance.hypervisor])
5628
        ins_l = ins_l[pnode]
5629
        if ins_l.failed or not isinstance(ins_l.data, list):
5630
          raise errors.OpPrereqError("Can't contact node '%s'" % pnode)
5631
        if instance.name in ins_l.data:
5632
          raise errors.OpPrereqError("Instance is running, can't remove"
5633
                                     " disks.")
5634

    
5635
      if (disk_op == constants.DDM_ADD and
5636
          len(instance.nics) >= constants.MAX_DISKS):
5637
        raise errors.OpPrereqError("Instance has too many disks (%d), cannot"
5638
                                   " add more" % constants.MAX_DISKS)
5639
      if disk_op not in (constants.DDM_ADD, constants.DDM_REMOVE):
5640
        # an existing disk
5641
        if disk_op < 0 or disk_op >= len(instance.disks):
5642
          raise errors.OpPrereqError("Invalid disk index %s, valid values"
5643
                                     " are 0 to %d" %
5644
                                     (disk_op, len(instance.disks)))
5645

    
5646
    return
5647

    
5648
  def Exec(self, feedback_fn):
5649
    """Modifies an instance.
5650

5651
    All parameters take effect only at the next restart of the instance.
5652

5653
    """
5654
    # Process here the warnings from CheckPrereq, as we don't have a
5655
    # feedback_fn there.
5656
    for warn in self.warn:
5657
      feedback_fn("WARNING: %s" % warn)
5658

    
5659
    result = []
5660
    instance = self.instance
5661
    # disk changes
5662
    for disk_op, disk_dict in self.op.disks:
5663
      if disk_op == constants.DDM_REMOVE:
5664
        # remove the last disk
5665
        device = instance.disks.pop()
5666
        device_idx = len(instance.disks)
5667
        for node, disk in device.ComputeNodeTree(instance.primary_node):
5668
          self.cfg.SetDiskID(disk, node)
5669
          rpc_result = self.rpc.call_blockdev_remove(node, disk)
5670
          if rpc_result.failed or not rpc_result.data:
5671
            self.proc.LogWarning("Could not remove disk/%d on node %s,"
5672
                                 " continuing anyway", device_idx, node)
5673
        result.append(("disk/%d" % device_idx, "remove"))
5674
      elif disk_op == constants.DDM_ADD:
5675
        # add a new disk
5676
        if instance.disk_template == constants.DT_FILE:
5677
          file_driver, file_path = instance.disks[0].logical_id
5678
          file_path = os.path.dirname(file_path)
5679
        else:
5680
          file_driver = file_path = None
5681
        disk_idx_base = len(instance.disks)
5682
        new_disk = _GenerateDiskTemplate(self,
5683
                                         instance.disk_template,
5684
                                         instance, instance.primary_node,
5685
                                         instance.secondary_nodes,
5686
                                         [disk_dict],
5687
                                         file_path,
5688
                                         file_driver,
5689
                                         disk_idx_base)[0]
5690
        new_disk.mode = disk_dict['mode']
5691
        instance.disks.append(new_disk)
5692
        info = _GetInstanceInfoText(instance)
5693

    
5694
        logging.info("Creating volume %s for instance %s",
5695
                     new_disk.iv_name, instance.name)
5696
        # Note: this needs to be kept in sync with _CreateDisks
5697
        #HARDCODE
5698
        for node in instance.all_nodes:
5699
          f_create = node == instance.primary_node
5700
          try:
5701
            _CreateBlockDev(self, node, instance, new_disk,
5702
                            f_create, info, f_create)
5703
          except error.OpExecError, err:
5704
            self.LogWarning("Failed to create volume %s (%s) on"
5705
                            " node %s: %s",
5706
                            new_disk.iv_name, new_disk, node, err)
5707
        result.append(("disk/%d" % disk_idx_base, "add:size=%s,mode=%s" %
5708
                       (new_disk.size, new_disk.mode)))
5709
      else:
5710
        # change a given disk
5711
        instance.disks[disk_op].mode = disk_dict['mode']
5712
        result.append(("disk.mode/%d" % disk_op, disk_dict['mode']))
5713
    # NIC changes
5714
    for nic_op, nic_dict in self.op.nics:
5715
      if nic_op == constants.DDM_REMOVE:
5716
        # remove the last nic
5717
        del instance.nics[-1]
5718
        result.append(("nic.%d" % len(instance.nics), "remove"))
5719
      elif nic_op == constants.DDM_ADD:
5720
        # add a new nic
5721
        if 'mac' not in nic_dict:
5722
          mac = constants.VALUE_GENERATE
5723
        else:
5724
          mac = nic_dict['mac']
5725
        if mac in (constants.VALUE_AUTO, constants.VALUE_GENERATE):
5726
          mac = self.cfg.GenerateMAC()
5727
        new_nic = objects.NIC(mac=mac, ip=nic_dict.get('ip', None),
5728
                              bridge=nic_dict.get('bridge', None))
5729
        instance.nics.append(new_nic)
5730
        result.append(("nic.%d" % (len(instance.nics) - 1),
5731
                       "add:mac=%s,ip=%s,bridge=%s" %
5732
                       (new_nic.mac, new_nic.ip, new_nic.bridge)))
5733
      else:
5734
        # change a given nic
5735
        for key in 'mac', 'ip', 'bridge':
5736
          if key in nic_dict:
5737
            setattr(instance.nics[nic_op], key, nic_dict[key])
5738
            result.append(("nic.%s/%d" % (key, nic_op), nic_dict[key]))
5739

    
5740
    # hvparams changes
5741
    if self.op.hvparams:
5742
      instance.hvparams = self.hv_new
5743
      for key, val in self.op.hvparams.iteritems():
5744
        result.append(("hv/%s" % key, val))
5745

    
5746
    # beparams changes
5747
    if self.op.beparams:
5748
      instance.beparams = self.be_inst
5749
      for key, val in self.op.beparams.iteritems():
5750
        result.append(("be/%s" % key, val))
5751

    
5752
    self.cfg.Update(instance)
5753

    
5754
    return result
5755

    
5756

    
5757
class LUQueryExports(NoHooksLU):
5758
  """Query the exports list
5759

5760
  """
5761
  _OP_REQP = ['nodes']
5762
  REQ_BGL = False
5763

    
5764
  def ExpandNames(self):
5765
    self.needed_locks = {}
5766
    self.share_locks[locking.LEVEL_NODE] = 1
5767
    if not self.op.nodes:
5768
      self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
5769
    else:
5770
      self.needed_locks[locking.LEVEL_NODE] = \
5771
        _GetWantedNodes(self, self.op.nodes)
5772

    
5773
  def CheckPrereq(self):
5774
    """Check prerequisites.
5775

5776
    """
5777
    self.nodes = self.acquired_locks[locking.LEVEL_NODE]
5778

    
5779
  def Exec(self, feedback_fn):
5780
    """Compute the list of all the exported system images.
5781

5782
    @rtype: dict
5783
    @return: a dictionary with the structure node->(export-list)
5784
        where export-list is a list of the instances exported on
5785
        that node.
5786

5787
    """
5788
    rpcresult = self.rpc.call_export_list(self.nodes)
5789
    result = {}
5790
    for node in rpcresult:
5791
      if rpcresult[node].failed:
5792
        result[node] = False
5793
      else:
5794
        result[node] = rpcresult[node].data
5795

    
5796
    return result
5797

    
5798

    
5799
class LUExportInstance(LogicalUnit):
5800
  """Export an instance to an image in the cluster.
5801

5802
  """
5803
  HPATH = "instance-export"
5804
  HTYPE = constants.HTYPE_INSTANCE
5805
  _OP_REQP = ["instance_name", "target_node", "shutdown"]
5806
  REQ_BGL = False
5807

    
5808
  def ExpandNames(self):
5809
    self._ExpandAndLockInstance()
5810
    # FIXME: lock only instance primary and destination node
5811
    #
5812
    # Sad but true, for now we have do lock all nodes, as we don't know where
5813
    # the previous export might be, and and in this LU we search for it and
5814
    # remove it from its current node. In the future we could fix this by:
5815
    #  - making a tasklet to search (share-lock all), then create the new one,
5816
    #    then one to remove, after
5817
    #  - removing the removal operation altoghether
5818
    self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
5819

    
5820
  def DeclareLocks(self, level):
5821
    """Last minute lock declaration."""
5822
    # All nodes are locked anyway, so nothing to do here.
5823

    
5824
  def BuildHooksEnv(self):
5825
    """Build hooks env.
5826

5827
    This will run on the master, primary node and target node.
5828

5829
    """
5830
    env = {
5831
      "EXPORT_NODE": self.op.target_node,
5832
      "EXPORT_DO_SHUTDOWN": self.op.shutdown,
5833
      }
5834
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
5835
    nl = [self.cfg.GetMasterNode(), self.instance.primary_node,
5836
          self.op.target_node]
5837
    return env, nl, nl
5838

    
5839
  def CheckPrereq(self):
5840
    """Check prerequisites.
5841

5842
    This checks that the instance and node names are valid.
5843

5844
    """
5845
    instance_name = self.op.instance_name
5846
    self.instance = self.cfg.GetInstanceInfo(instance_name)
5847
    assert self.instance is not None, \
5848
          "Cannot retrieve locked instance %s" % self.op.instance_name
5849
    _CheckNodeOnline(self, self.instance.primary_node)
5850

    
5851
    self.dst_node = self.cfg.GetNodeInfo(
5852
      self.cfg.ExpandNodeName(self.op.target_node))
5853

    
5854
    if self.dst_node is None:
5855
      # This is wrong node name, not a non-locked node
5856
      raise errors.OpPrereqError("Wrong node name %s" % self.op.target_node)
5857
    _CheckNodeOnline(self, self.dst_node.name)
5858

    
5859
    # instance disk type verification
5860
    for disk in self.instance.disks:
5861
      if disk.dev_type == constants.LD_FILE:
5862
        raise errors.OpPrereqError("Export not supported for instances with"
5863
                                   " file-based disks")
5864

    
5865
  def Exec(self, feedback_fn):
5866
    """Export an instance to an image in the cluster.
5867

5868
    """
5869
    instance = self.instance
5870
    dst_node = self.dst_node
5871
    src_node = instance.primary_node
5872
    if self.op.shutdown:
5873
      # shutdown the instance, but not the disks
5874
      result = self.rpc.call_instance_shutdown(src_node, instance)
5875
      result.Raise()
5876
      if not result.data:
5877
        raise errors.OpExecError("Could not shutdown instance %s on node %s" %
5878
                                 (instance.name, src_node))
5879

    
5880
    vgname = self.cfg.GetVGName()
5881

    
5882
    snap_disks = []
5883

    
5884
    # set the disks ID correctly since call_instance_start needs the
5885
    # correct drbd minor to create the symlinks
5886
    for disk in instance.disks:
5887
      self.cfg.SetDiskID(disk, src_node)
5888

    
5889
    try:
5890
      for disk in instance.disks:
5891
        # new_dev_name will be a snapshot of an lvm leaf of the one we passed
5892
        new_dev_name = self.rpc.call_blockdev_snapshot(src_node, disk)
5893
        if new_dev_name.failed or not new_dev_name.data:
5894
          self.LogWarning("Could not snapshot block device %s on node %s",
5895
                          disk.logical_id[1], src_node)
5896
          snap_disks.append(False)
5897
        else:
5898
          new_dev = objects.Disk(dev_type=constants.LD_LV, size=disk.size,
5899
                                 logical_id=(vgname, new_dev_name.data),
5900
                                 physical_id=(vgname, new_dev_name.data),
5901
                                 iv_name=disk.iv_name)
5902
          snap_disks.append(new_dev)
5903

    
5904
    finally:
5905
      if self.op.shutdown and instance.status == "up":
5906
        result = self.rpc.call_instance_start(src_node, instance, None)
5907
        if result.failed or not result.data:
5908
          _ShutdownInstanceDisks(self, instance)
5909
          raise errors.OpExecError("Could not start instance")
5910

    
5911
    # TODO: check for size
5912

    
5913
    cluster_name = self.cfg.GetClusterName()
5914
    for idx, dev in enumerate(snap_disks):
5915
      if dev:
5916
        result = self.rpc.call_snapshot_export(src_node, dev, dst_node.name,
5917
                                               instance, cluster_name, idx)
5918
        if result.failed or not result.data:
5919
          self.LogWarning("Could not export block device %s from node %s to"
5920
                          " node %s", dev.logical_id[1], src_node,
5921
                          dst_node.name)
5922
        result = self.rpc.call_blockdev_remove(src_node, dev)
5923
        if result.failed or not result.data:
5924
          self.LogWarning("Could not remove snapshot block device %s from node"
5925
                          " %s", dev.logical_id[1], src_node)
5926

    
5927
    result = self.rpc.call_finalize_export(dst_node.name, instance, snap_disks)
5928
    if result.failed or not result.data:
5929
      self.LogWarning("Could not finalize export for instance %s on node %s",
5930
                      instance.name, dst_node.name)
5931

    
5932
    nodelist = self.cfg.GetNodeList()
5933
    nodelist.remove(dst_node.name)
5934

    
5935
    # on one-node clusters nodelist will be empty after the removal
5936
    # if we proceed the backup would be removed because OpQueryExports
5937
    # substitutes an empty list with the full cluster node list.
5938
    if nodelist:
5939
      exportlist = self.rpc.call_export_list(nodelist)
5940
      for node in exportlist:
5941
        if exportlist[node].failed:
5942
          continue
5943
        if instance.name in exportlist[node].data:
5944
          if not self.rpc.call_export_remove(node, instance.name):
5945
            self.LogWarning("Could not remove older export for instance %s"
5946
                            " on node %s", instance.name, node)
5947

    
5948

    
5949
class LURemoveExport(NoHooksLU):
5950
  """Remove exports related to the named instance.
5951

5952
  """
5953
  _OP_REQP = ["instance_name"]
5954
  REQ_BGL = False
5955

    
5956
  def ExpandNames(self):
5957
    self.needed_locks = {}
5958
    # We need all nodes to be locked in order for RemoveExport to work, but we
5959
    # don't need to lock the instance itself, as nothing will happen to it (and
5960
    # we can remove exports also for a removed instance)
5961
    self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
5962

    
5963
  def CheckPrereq(self):
5964
    """Check prerequisites.
5965
    """
5966
    pass
5967

    
5968
  def Exec(self, feedback_fn):
5969
    """Remove any export.
5970

5971
    """
5972
    instance_name = self.cfg.ExpandInstanceName(self.op.instance_name)
5973
    # If the instance was not found we'll try with the name that was passed in.
5974
    # This will only work if it was an FQDN, though.
5975
    fqdn_warn = False
5976
    if not instance_name:
5977
      fqdn_warn = True
5978
      instance_name = self.op.instance_name
5979

    
5980
    exportlist = self.rpc.call_export_list(self.acquired_locks[
5981
      locking.LEVEL_NODE])
5982
    found = False
5983
    for node in exportlist:
5984
      if exportlist[node].failed:
5985
        self.LogWarning("Failed to query node %s, continuing" % node)
5986
        continue
5987
      if instance_name in exportlist[node].data:
5988
        found = True
5989
        result = self.rpc.call_export_remove(node, instance_name)
5990
        if result.failed or not result.data:
5991
          logging.error("Could not remove export for instance %s"
5992
                        " on node %s", instance_name, node)
5993

    
5994
    if fqdn_warn and not found:
5995
      feedback_fn("Export not found. If trying to remove an export belonging"
5996
                  " to a deleted instance please use its Fully Qualified"
5997
                  " Domain Name.")
5998

    
5999

    
6000
class TagsLU(NoHooksLU):
6001
  """Generic tags LU.
6002

6003
  This is an abstract class which is the parent of all the other tags LUs.
6004

6005
  """
6006

    
6007
  def ExpandNames(self):
6008
    self.needed_locks = {}
6009
    if self.op.kind == constants.TAG_NODE:
6010
      name = self.cfg.ExpandNodeName(self.op.name)
6011
      if name is None:
6012
        raise errors.OpPrereqError("Invalid node name (%s)" %
6013
                                   (self.op.name,))
6014
      self.op.name = name
6015
      self.needed_locks[locking.LEVEL_NODE] = name
6016
    elif self.op.kind == constants.TAG_INSTANCE:
6017
      name = self.cfg.ExpandInstanceName(self.op.name)
6018
      if name is None:
6019
        raise errors.OpPrereqError("Invalid instance name (%s)" %
6020
                                   (self.op.name,))
6021
      self.op.name = name
6022
      self.needed_locks[locking.LEVEL_INSTANCE] = name
6023

    
6024
  def CheckPrereq(self):
6025
    """Check prerequisites.
6026

6027
    """
6028
    if self.op.kind == constants.TAG_CLUSTER:
6029
      self.target = self.cfg.GetClusterInfo()
6030
    elif self.op.kind == constants.TAG_NODE:
6031
      self.target = self.cfg.GetNodeInfo(self.op.name)
6032
    elif self.op.kind == constants.TAG_INSTANCE:
6033
      self.target = self.cfg.GetInstanceInfo(self.op.name)
6034
    else:
6035
      raise errors.OpPrereqError("Wrong tag type requested (%s)" %
6036
                                 str(self.op.kind))
6037

    
6038

    
6039
class LUGetTags(TagsLU):
6040
  """Returns the tags of a given object.
6041

6042
  """
6043
  _OP_REQP = ["kind", "name"]
6044
  REQ_BGL = False
6045

    
6046
  def Exec(self, feedback_fn):
6047
    """Returns the tag list.
6048

6049
    """
6050
    return list(self.target.GetTags())
6051

    
6052

    
6053
class LUSearchTags(NoHooksLU):
6054
  """Searches the tags for a given pattern.
6055

6056
  """
6057
  _OP_REQP = ["pattern"]
6058
  REQ_BGL = False
6059

    
6060
  def ExpandNames(self):
6061
    self.needed_locks = {}
6062

    
6063
  def CheckPrereq(self):
6064
    """Check prerequisites.
6065

6066
    This checks the pattern passed for validity by compiling it.
6067

6068
    """
6069
    try:
6070
      self.re = re.compile(self.op.pattern)
6071
    except re.error, err:
6072
      raise errors.OpPrereqError("Invalid search pattern '%s': %s" %
6073
                                 (self.op.pattern, err))
6074

    
6075
  def Exec(self, feedback_fn):
6076
    """Returns the tag list.
6077

6078
    """
6079
    cfg = self.cfg
6080
    tgts = [("/cluster", cfg.GetClusterInfo())]
6081
    ilist = cfg.GetAllInstancesInfo().values()
6082
    tgts.extend([("/instances/%s" % i.name, i) for i in ilist])
6083
    nlist = cfg.GetAllNodesInfo().values()
6084
    tgts.extend([("/nodes/%s" % n.name, n) for n in nlist])
6085
    results = []
6086
    for path, target in tgts:
6087
      for tag in target.GetTags():
6088
        if self.re.search(tag):
6089
          results.append((path, tag))
6090
    return results
6091

    
6092

    
6093
class LUAddTags(TagsLU):
6094
  """Sets a tag on a given object.
6095

6096
  """
6097
  _OP_REQP = ["kind", "name", "tags"]
6098
  REQ_BGL = False
6099

    
6100
  def CheckPrereq(self):
6101
    """Check prerequisites.
6102

6103
    This checks the type and length of the tag name and value.
6104

6105
    """
6106
    TagsLU.CheckPrereq(self)
6107
    for tag in self.op.tags:
6108
      objects.TaggableObject.ValidateTag(tag)
6109

    
6110
  def Exec(self, feedback_fn):
6111
    """Sets the tag.
6112

6113
    """
6114
    try:
6115
      for tag in self.op.tags:
6116
        self.target.AddTag(tag)
6117
    except errors.TagError, err:
6118
      raise errors.OpExecError("Error while setting tag: %s" % str(err))
6119
    try:
6120
      self.cfg.Update(self.target)
6121
    except errors.ConfigurationError:
6122
      raise errors.OpRetryError("There has been a modification to the"
6123
                                " config file and the operation has been"
6124
                                " aborted. Please retry.")
6125

    
6126

    
6127
class LUDelTags(TagsLU):
6128
  """Delete a list of tags from a given object.
6129

6130
  """
6131
  _OP_REQP = ["kind", "name", "tags"]
6132
  REQ_BGL = False
6133

    
6134
  def CheckPrereq(self):
6135
    """Check prerequisites.
6136

6137
    This checks that we have the given tag.
6138

6139
    """
6140
    TagsLU.CheckPrereq(self)
6141
    for tag in self.op.tags:
6142
      objects.TaggableObject.ValidateTag(tag)
6143
    del_tags = frozenset(self.op.tags)
6144
    cur_tags = self.target.GetTags()
6145
    if not del_tags <= cur_tags:
6146
      diff_tags = del_tags - cur_tags
6147
      diff_names = ["'%s'" % tag for tag in diff_tags]
6148
      diff_names.sort()
6149
      raise errors.OpPrereqError("Tag(s) %s not found" %
6150
                                 (",".join(diff_names)))
6151

    
6152
  def Exec(self, feedback_fn):
6153
    """Remove the tag from the object.
6154

6155
    """
6156
    for tag in self.op.tags:
6157
      self.target.RemoveTag(tag)
6158
    try:
6159
      self.cfg.Update(self.target)
6160
    except errors.ConfigurationError:
6161
      raise errors.OpRetryError("There has been a modification to the"
6162
                                " config file and the operation has been"
6163
                                " aborted. Please retry.")
6164

    
6165

    
6166
class LUTestDelay(NoHooksLU):
6167
  """Sleep for a specified amount of time.
6168

6169
  This LU sleeps on the master and/or nodes for a specified amount of
6170
  time.
6171

6172
  """
6173
  _OP_REQP = ["duration", "on_master", "on_nodes"]
6174
  REQ_BGL = False
6175

    
6176
  def ExpandNames(self):
6177
    """Expand names and set required locks.
6178

6179
    This expands the node list, if any.
6180

6181
    """
6182
    self.needed_locks = {}
6183
    if self.op.on_nodes:
6184
      # _GetWantedNodes can be used here, but is not always appropriate to use
6185
      # this way in ExpandNames. Check LogicalUnit.ExpandNames docstring for
6186
      # more information.
6187
      self.op.on_nodes = _GetWantedNodes(self, self.op.on_nodes)
6188
      self.needed_locks[locking.LEVEL_NODE] = self.op.on_nodes
6189

    
6190
  def CheckPrereq(self):
6191
    """Check prerequisites.
6192

6193
    """
6194

    
6195
  def Exec(self, feedback_fn):
6196
    """Do the actual sleep.
6197

6198
    """
6199
    if self.op.on_master:
6200
      if not utils.TestDelay(self.op.duration):
6201
        raise errors.OpExecError("Error during master delay test")
6202
    if self.op.on_nodes:
6203
      result = self.rpc.call_test_delay(self.op.on_nodes, self.op.duration)
6204
      if not result:
6205
        raise errors.OpExecError("Complete failure from rpc call")
6206
      for node, node_result in result.items():
6207
        node_result.Raise()
6208
        if not node_result.data:
6209
          raise errors.OpExecError("Failure during rpc call to node %s,"
6210
                                   " result: %s" % (node, node_result.data))
6211

    
6212

    
6213
class IAllocator(object):
6214
  """IAllocator framework.
6215

6216
  An IAllocator instance has three sets of attributes:
6217
    - cfg that is needed to query the cluster
6218
    - input data (all members of the _KEYS class attribute are required)
6219
    - four buffer attributes (in|out_data|text), that represent the
6220
      input (to the external script) in text and data structure format,
6221
      and the output from it, again in two formats
6222
    - the result variables from the script (success, info, nodes) for
6223
      easy usage
6224

6225
  """
6226
  _ALLO_KEYS = [
6227
    "mem_size", "disks", "disk_template",
6228
    "os", "tags", "nics", "vcpus", "hypervisor",
6229
    ]
6230
  _RELO_KEYS = [
6231
    "relocate_from",
6232
    ]
6233

    
6234
  def __init__(self, lu, mode, name, **kwargs):
6235
    self.lu = lu
6236
    # init buffer variables
6237
    self.in_text = self.out_text = self.in_data = self.out_data = None
6238
    # init all input fields so that pylint is happy
6239
    self.mode = mode
6240
    self.name = name
6241
    self.mem_size = self.disks = self.disk_template = None
6242
    self.os = self.tags = self.nics = self.vcpus = None
6243
    self.hypervisor = None
6244
    self.relocate_from = None
6245
    # computed fields
6246
    self.required_nodes = None
6247
    # init result fields
6248
    self.success = self.info = self.nodes = None
6249
    if self.mode == constants.IALLOCATOR_MODE_ALLOC:
6250
      keyset = self._ALLO_KEYS
6251
    elif self.mode == constants.IALLOCATOR_MODE_RELOC:
6252
      keyset = self._RELO_KEYS
6253
    else:
6254
      raise errors.ProgrammerError("Unknown mode '%s' passed to the"
6255
                                   " IAllocator" % self.mode)
6256
    for key in kwargs:
6257
      if key not in keyset:
6258
        raise errors.ProgrammerError("Invalid input parameter '%s' to"
6259
                                     " IAllocator" % key)
6260
      setattr(self, key, kwargs[key])
6261
    for key in keyset:
6262
      if key not in kwargs:
6263
        raise errors.ProgrammerError("Missing input parameter '%s' to"
6264
                                     " IAllocator" % key)
6265
    self._BuildInputData()
6266

    
6267
  def _ComputeClusterData(self):
6268
    """Compute the generic allocator input data.
6269

6270
    This is the data that is independent of the actual operation.
6271

6272
    """
6273
    cfg = self.lu.cfg
6274
    cluster_info = cfg.GetClusterInfo()
6275
    # cluster data
6276
    data = {
6277
      "version": 1,
6278
      "cluster_name": cfg.GetClusterName(),
6279
      "cluster_tags": list(cluster_info.GetTags()),
6280
      "enable_hypervisors": list(cluster_info.enabled_hypervisors),
6281
      # we don't have job IDs
6282
      }
6283
    iinfo = cfg.GetAllInstancesInfo().values()
6284
    i_list = [(inst, cluster_info.FillBE(inst)) for inst in iinfo]
6285

    
6286
    # node data
6287
    node_results = {}
6288
    node_list = cfg.GetNodeList()
6289

    
6290
    if self.mode == constants.IALLOCATOR_MODE_ALLOC:
6291
      hypervisor_name = self.hypervisor
6292
    elif self.mode == constants.IALLOCATOR_MODE_RELOC:
6293
      hypervisor_name = cfg.GetInstanceInfo(self.name).hypervisor
6294

    
6295
    node_data = self.lu.rpc.call_node_info(node_list, cfg.GetVGName(),
6296
                                           hypervisor_name)
6297
    node_iinfo = self.lu.rpc.call_all_instances_info(node_list,
6298
                       cluster_info.enabled_hypervisors)
6299
    for nname in node_list:
6300
      ninfo = cfg.GetNodeInfo(nname)
6301
      node_data[nname].Raise()
6302
      if not isinstance(node_data[nname].data, dict):
6303
        raise errors.OpExecError("Can't get data for node %s" % nname)
6304
      remote_info = node_data[nname].data
6305
      for attr in ['memory_total', 'memory_free', 'memory_dom0',
6306
                   'vg_size', 'vg_free', 'cpu_total']:
6307
        if attr not in remote_info:
6308
          raise errors.OpExecError("Node '%s' didn't return attribute '%s'" %
6309
                                   (nname, attr))
6310
        try:
6311
          remote_info[attr] = int(remote_info[attr])
6312
        except ValueError, err:
6313
          raise errors.OpExecError("Node '%s' returned invalid value for '%s':"
6314
                                   " %s" % (nname, attr, str(err)))
6315
      # compute memory used by primary instances
6316
      i_p_mem = i_p_up_mem = 0
6317
      for iinfo, beinfo in i_list:
6318
        if iinfo.primary_node == nname:
6319
          i_p_mem += beinfo[constants.BE_MEMORY]
6320
          if iinfo.name not in node_iinfo[nname]:
6321
            i_used_mem = 0
6322
          else:
6323
            i_used_mem = int(node_iinfo[nname][iinfo.name]['memory'])
6324
          i_mem_diff = beinfo[constants.BE_MEMORY] - i_used_mem
6325
          remote_info['memory_free'] -= max(0, i_mem_diff)
6326

    
6327
          if iinfo.status == "up":
6328
            i_p_up_mem += beinfo[constants.BE_MEMORY]
6329

    
6330
      # compute memory used by instances
6331
      pnr = {
6332
        "tags": list(ninfo.GetTags()),
6333
        "total_memory": remote_info['memory_total'],
6334
        "reserved_memory": remote_info['memory_dom0'],
6335
        "free_memory": remote_info['memory_free'],
6336
        "i_pri_memory": i_p_mem,
6337
        "i_pri_up_memory": i_p_up_mem,
6338
        "total_disk": remote_info['vg_size'],
6339
        "free_disk": remote_info['vg_free'],
6340
        "primary_ip": ninfo.primary_ip,
6341
        "secondary_ip": ninfo.secondary_ip,
6342
        "total_cpus": remote_info['cpu_total'],
6343
        "offline": ninfo.offline,
6344
        }
6345
      node_results[nname] = pnr
6346
    data["nodes"] = node_results
6347

    
6348
    # instance data
6349
    instance_data = {}
6350
    for iinfo, beinfo in i_list:
6351
      nic_data = [{"mac": n.mac, "ip": n.ip, "bridge": n.bridge}
6352
                  for n in iinfo.nics]
6353
      pir = {
6354
        "tags": list(iinfo.GetTags()),
6355
        "should_run": iinfo.status == "up",
6356
        "vcpus": beinfo[constants.BE_VCPUS],
6357
        "memory": beinfo[constants.BE_MEMORY],
6358
        "os": iinfo.os,
6359
        "nodes": list(iinfo.all_nodes),
6360
        "nics": nic_data,
6361
        "disks": [{"size": dsk.size, "mode": "w"} for dsk in iinfo.disks],
6362
        "disk_template": iinfo.disk_template,
6363
        "hypervisor": iinfo.hypervisor,
6364
        }
6365
      instance_data[iinfo.name] = pir
6366

    
6367
    data["instances"] = instance_data
6368

    
6369
    self.in_data = data
6370

    
6371
  def _AddNewInstance(self):
6372
    """Add new instance data to allocator structure.
6373

6374
    This in combination with _AllocatorGetClusterData will create the
6375
    correct structure needed as input for the allocator.
6376

6377
    The checks for the completeness of the opcode must have already been
6378
    done.
6379

6380
    """
6381
    data = self.in_data
6382
    if len(self.disks) != 2:
6383
      raise errors.OpExecError("Only two-disk configurations supported")
6384

    
6385
    disk_space = _ComputeDiskSize(self.disk_template, self.disks)
6386

    
6387
    if self.disk_template in constants.DTS_NET_MIRROR:
6388
      self.required_nodes = 2
6389
    else:
6390
      self.required_nodes = 1
6391
    request = {
6392
      "type": "allocate",
6393
      "name": self.name,
6394
      "disk_template": self.disk_template,
6395
      "tags": self.tags,
6396
      "os": self.os,
6397
      "vcpus": self.vcpus,
6398
      "memory": self.mem_size,
6399
      "disks": self.disks,
6400
      "disk_space_total": disk_space,
6401
      "nics": self.nics,
6402
      "required_nodes": self.required_nodes,
6403
      }
6404
    data["request"] = request
6405

    
6406
  def _AddRelocateInstance(self):
6407
    """Add relocate instance data to allocator structure.
6408

6409
    This in combination with _IAllocatorGetClusterData will create the
6410
    correct structure needed as input for the allocator.
6411

6412
    The checks for the completeness of the opcode must have already been
6413
    done.
6414

6415
    """
6416
    instance = self.lu.cfg.GetInstanceInfo(self.name)
6417
    if instance is None:
6418
      raise errors.ProgrammerError("Unknown instance '%s' passed to"
6419
                                   " IAllocator" % self.name)
6420

    
6421
    if instance.disk_template not in constants.DTS_NET_MIRROR:
6422
      raise errors.OpPrereqError("Can't relocate non-mirrored instances")
6423

    
6424
    if len(instance.secondary_nodes) != 1:
6425
      raise errors.OpPrereqError("Instance has not exactly one secondary node")
6426

    
6427
    self.required_nodes = 1
6428
    disk_sizes = [{'size': disk.size} for disk in instance.disks]
6429
    disk_space = _ComputeDiskSize(instance.disk_template, disk_sizes)
6430

    
6431
    request = {
6432
      "type": "relocate",
6433
      "name": self.name,
6434
      "disk_space_total": disk_space,
6435
      "required_nodes": self.required_nodes,
6436
      "relocate_from": self.relocate_from,
6437
      }
6438
    self.in_data["request"] = request
6439

    
6440
  def _BuildInputData(self):
6441
    """Build input data structures.
6442

6443
    """
6444
    self._ComputeClusterData()
6445

    
6446
    if self.mode == constants.IALLOCATOR_MODE_ALLOC:
6447
      self._AddNewInstance()
6448
    else:
6449
      self._AddRelocateInstance()
6450

    
6451
    self.in_text = serializer.Dump(self.in_data)
6452

    
6453
  def Run(self, name, validate=True, call_fn=None):
6454
    """Run an instance allocator and return the results.
6455

6456
    """
6457
    if call_fn is None:
6458
      call_fn = self.lu.rpc.call_iallocator_runner
6459
    data = self.in_text
6460

    
6461
    result = call_fn(self.lu.cfg.GetMasterNode(), name, self.in_text)
6462
    result.Raise()
6463

    
6464
    if not isinstance(result.data, (list, tuple)) or len(result.data) != 4:
6465
      raise errors.OpExecError("Invalid result from master iallocator runner")
6466

    
6467
    rcode, stdout, stderr, fail = result.data
6468

    
6469
    if rcode == constants.IARUN_NOTFOUND:
6470
      raise errors.OpExecError("Can't find allocator '%s'" % name)
6471
    elif rcode == constants.IARUN_FAILURE:
6472
      raise errors.OpExecError("Instance allocator call failed: %s,"
6473
                               " output: %s" % (fail, stdout+stderr))
6474
    self.out_text = stdout
6475
    if validate:
6476
      self._ValidateResult()
6477

    
6478
  def _ValidateResult(self):
6479
    """Process the allocator results.
6480

6481
    This will process and if successful save the result in
6482
    self.out_data and the other parameters.
6483

6484
    """
6485
    try:
6486
      rdict = serializer.Load(self.out_text)
6487
    except Exception, err:
6488
      raise errors.OpExecError("Can't parse iallocator results: %s" % str(err))
6489

    
6490
    if not isinstance(rdict, dict):
6491
      raise errors.OpExecError("Can't parse iallocator results: not a dict")
6492

    
6493
    for key in "success", "info", "nodes":
6494
      if key not in rdict:
6495
        raise errors.OpExecError("Can't parse iallocator results:"
6496
                                 " missing key '%s'" % key)
6497
      setattr(self, key, rdict[key])
6498

    
6499
    if not isinstance(rdict["nodes"], list):
6500
      raise errors.OpExecError("Can't parse iallocator results: 'nodes' key"
6501
                               " is not a list")
6502
    self.out_data = rdict
6503

    
6504

    
6505
class LUTestAllocator(NoHooksLU):
6506
  """Run allocator tests.
6507

6508
  This LU runs the allocator tests
6509

6510
  """
6511
  _OP_REQP = ["direction", "mode", "name"]
6512

    
6513
  def CheckPrereq(self):
6514
    """Check prerequisites.
6515

6516
    This checks the opcode parameters depending on the director and mode test.
6517

6518
    """
6519
    if self.op.mode == constants.IALLOCATOR_MODE_ALLOC:
6520
      for attr in ["name", "mem_size", "disks", "disk_template",
6521
                   "os", "tags", "nics", "vcpus"]:
6522
        if not hasattr(self.op, attr):
6523
          raise errors.OpPrereqError("Missing attribute '%s' on opcode input" %
6524
                                     attr)
6525
      iname = self.cfg.ExpandInstanceName(self.op.name)
6526
      if iname is not None:
6527
        raise errors.OpPrereqError("Instance '%s' already in the cluster" %
6528
                                   iname)
6529
      if not isinstance(self.op.nics, list):
6530
        raise errors.OpPrereqError("Invalid parameter 'nics'")
6531
      for row in self.op.nics:
6532
        if (not isinstance(row, dict) or
6533
            "mac" not in row or
6534
            "ip" not in row or
6535
            "bridge" not in row):
6536
          raise errors.OpPrereqError("Invalid contents of the"
6537
                                     " 'nics' parameter")
6538
      if not isinstance(self.op.disks, list):
6539
        raise errors.OpPrereqError("Invalid parameter 'disks'")
6540
      if len(self.op.disks) != 2:
6541
        raise errors.OpPrereqError("Only two-disk configurations supported")
6542
      for row in self.op.disks:
6543
        if (not isinstance(row, dict) or
6544
            "size" not in row or
6545
            not isinstance(row["size"], int) or
6546
            "mode" not in row or
6547
            row["mode"] not in ['r', 'w']):
6548
          raise errors.OpPrereqError("Invalid contents of the"
6549
                                     " 'disks' parameter")
6550
      if self.op.hypervisor is None:
6551
        self.op.hypervisor = self.cfg.GetHypervisorType()
6552
    elif self.op.mode == constants.IALLOCATOR_MODE_RELOC:
6553
      if not hasattr(self.op, "name"):
6554
        raise errors.OpPrereqError("Missing attribute 'name' on opcode input")
6555
      fname = self.cfg.ExpandInstanceName(self.op.name)
6556
      if fname is None:
6557
        raise errors.OpPrereqError("Instance '%s' not found for relocation" %
6558
                                   self.op.name)
6559
      self.op.name = fname
6560
      self.relocate_from = self.cfg.GetInstanceInfo(fname).secondary_nodes
6561
    else:
6562
      raise errors.OpPrereqError("Invalid test allocator mode '%s'" %
6563
                                 self.op.mode)
6564

    
6565
    if self.op.direction == constants.IALLOCATOR_DIR_OUT:
6566
      if not hasattr(self.op, "allocator") or self.op.allocator is None:
6567
        raise errors.OpPrereqError("Missing allocator name")
6568
    elif self.op.direction != constants.IALLOCATOR_DIR_IN:
6569
      raise errors.OpPrereqError("Wrong allocator test '%s'" %
6570
                                 self.op.direction)
6571

    
6572
  def Exec(self, feedback_fn):
6573
    """Run the allocator test.
6574

6575
    """
6576
    if self.op.mode == constants.IALLOCATOR_MODE_ALLOC:
6577
      ial = IAllocator(self,
6578
                       mode=self.op.mode,
6579
                       name=self.op.name,
6580
                       mem_size=self.op.mem_size,
6581
                       disks=self.op.disks,
6582
                       disk_template=self.op.disk_template,
6583
                       os=self.op.os,
6584
                       tags=self.op.tags,
6585
                       nics=self.op.nics,
6586
                       vcpus=self.op.vcpus,
6587
                       hypervisor=self.op.hypervisor,
6588
                       )
6589
    else:
6590
      ial = IAllocator(self,
6591
                       mode=self.op.mode,
6592
                       name=self.op.name,
6593
                       relocate_from=list(self.relocate_from),
6594
                       )
6595

    
6596
    if self.op.direction == constants.IALLOCATOR_DIR_IN:
6597
      result = ial.in_text
6598
    else:
6599
      ial.Run(self.op.allocator, validate=False)
6600
      result = ial.out_text
6601
    return result