Statistics
| Branch: | Tag: | Revision:

root / lib / cmdlib.py @ 428958aa

History | View | Annotate | Download (227.9 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 == instance.primary_node:
1691
        raise errors.OpPrereqError("Instance %s still running on the node,"
1692
                                   " please remove first." % instance_name)
1693
      if node.name in instance.secondary_nodes:
1694
        raise errors.OpPrereqError("Instance %s has node as a secondary,"
1695
                                   " please remove first." % instance_name)
1696
    self.op.node_name = node.name
1697
    self.node = node
1698

    
1699
  def Exec(self, feedback_fn):
1700
    """Removes the node from the cluster.
1701

1702
    """
1703
    node = self.node
1704
    logging.info("Stopping the node daemon and removing configs from node %s",
1705
                 node.name)
1706

    
1707
    self.context.RemoveNode(node.name)
1708

    
1709
    self.rpc.call_node_leave_cluster(node.name)
1710

    
1711
    # Promote nodes to master candidate as needed
1712
    _AdjustCandidatePool(self)
1713

    
1714

    
1715
class LUQueryNodes(NoHooksLU):
1716
  """Logical unit for querying nodes.
1717

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

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

    
1738
  def ExpandNames(self):
1739
    _CheckOutputFields(static=self._FIELDS_STATIC,
1740
                       dynamic=self._FIELDS_DYNAMIC,
1741
                       selected=self.op.output_fields)
1742

    
1743
    self.needed_locks = {}
1744
    self.share_locks[locking.LEVEL_NODE] = 1
1745

    
1746
    if self.op.names:
1747
      self.wanted = _GetWantedNodes(self, self.op.names)
1748
    else:
1749
      self.wanted = locking.ALL_SET
1750

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

    
1756

    
1757
  def CheckPrereq(self):
1758
    """Check prerequisites.
1759

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

    
1765
  def Exec(self, feedback_fn):
1766
    """Computes the list of nodes and their attributes.
1767

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

    
1781
    nodenames = utils.NiceSort(nodenames)
1782
    nodelist = [all_info[name] for name in nodenames]
1783

    
1784
    # begin data gathering
1785

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

    
1809
    node_to_primary = dict([(name, set()) for name in nodenames])
1810
    node_to_secondary = dict([(name, set()) for name in nodenames])
1811

    
1812
    inst_fields = frozenset(("pinst_cnt", "pinst_list",
1813
                             "sinst_cnt", "sinst_list"))
1814
    if inst_fields & frozenset(self.op.output_fields):
1815
      instancelist = self.cfg.GetInstanceList()
1816

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

    
1825
    master_node = self.cfg.GetMasterNode()
1826

    
1827
    # end data gathering
1828

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

    
1864
    return output
1865

    
1866

    
1867
class LUQueryNodeVolumes(NoHooksLU):
1868
  """Logical unit for getting volumes on node(s).
1869

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

    
1876
  def ExpandNames(self):
1877
    _CheckOutputFields(static=self._FIELDS_STATIC,
1878
                       dynamic=self._FIELDS_DYNAMIC,
1879
                       selected=self.op.output_fields)
1880

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

    
1889
  def CheckPrereq(self):
1890
    """Check prerequisites.
1891

1892
    This checks that the fields required are valid output fields.
1893

1894
    """
1895
    self.nodes = self.acquired_locks[locking.LEVEL_NODE]
1896

    
1897
  def Exec(self, feedback_fn):
1898
    """Computes the list of nodes and their attributes.
1899

1900
    """
1901
    nodenames = self.nodes
1902
    volumes = self.rpc.call_node_volumes(nodenames)
1903

    
1904
    ilist = [self.cfg.GetInstanceInfo(iname) for iname
1905
             in self.cfg.GetInstanceList()]
1906

    
1907
    lv_by_node = dict([(inst, inst.MapLVsByNode()) for inst in ilist])
1908

    
1909
    output = []
1910
    for node in nodenames:
1911
      if node not in volumes or volumes[node].failed or not volumes[node].data:
1912
        continue
1913

    
1914
      node_vols = volumes[node].data[:]
1915
      node_vols.sort(key=lambda vol: vol['dev'])
1916

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

    
1943
        output.append(node_output)
1944

    
1945
    return output
1946

    
1947

    
1948
class LUAddNode(LogicalUnit):
1949
  """Logical unit for adding node to the cluster.
1950

1951
  """
1952
  HPATH = "node-add"
1953
  HTYPE = constants.HTYPE_NODE
1954
  _OP_REQP = ["node_name"]
1955

    
1956
  def BuildHooksEnv(self):
1957
    """Build hooks env.
1958

1959
    This will run on all nodes before, and on all nodes + the new node after.
1960

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

    
1972
  def CheckPrereq(self):
1973
    """Check prerequisites.
1974

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

1980
    Any errors are signalled by raising errors.OpPrereqError.
1981

1982
    """
1983
    node_name = self.op.node_name
1984
    cfg = self.cfg
1985

    
1986
    dns_data = utils.HostInfo(node_name)
1987

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

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

    
2004
    for existing_node_name in node_list:
2005
      existing_node = cfg.GetNodeInfo(existing_node_name)
2006

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

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

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

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

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

    
2045
    cp_size = self.cfg.GetClusterInfo().candidate_pool_size
2046
    mc_now, _ = self.cfg.GetMasterCandidateStats()
2047
    master_candidate = mc_now < cp_size
2048

    
2049
    self.new_node = objects.Node(name=node,
2050
                                 primary_ip=primary_ip,
2051
                                 secondary_ip=secondary_ip,
2052
                                 master_candidate=master_candidate,
2053
                                 offline=False)
2054

    
2055
  def Exec(self, feedback_fn):
2056
    """Adds the new node to the cluster.
2057

2058
    """
2059
    new_node = self.new_node
2060
    node = new_node.name
2061

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

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

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

    
2091
    result = self.rpc.call_node_add(node, keyarray[0], keyarray[1],
2092
                                    keyarray[2],
2093
                                    keyarray[3], keyarray[4], keyarray[5])
2094

    
2095
    if result.failed or not result.data:
2096
      raise errors.OpExecError("Cannot transfer ssh keys to the new node")
2097

    
2098
    # Add node to our /etc/hosts, and add key to known_hosts
2099
    utils.AddHostToEtcHosts(new_node.name)
2100

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

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

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

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

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

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

    
2151
    if self.op.readd:
2152
      self.context.ReaddNode(new_node)
2153
    else:
2154
      self.context.AddNode(new_node)
2155

    
2156

    
2157
class LUSetNodeParams(LogicalUnit):
2158
  """Modifies the parameters of a node.
2159

2160
  """
2161
  HPATH = "node-modify"
2162
  HTYPE = constants.HTYPE_NODE
2163
  _OP_REQP = ["node_name"]
2164
  REQ_BGL = False
2165

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

    
2179
  def ExpandNames(self):
2180
    self.needed_locks = {locking.LEVEL_NODE: self.op.node_name}
2181

    
2182
  def BuildHooksEnv(self):
2183
    """Build hooks env.
2184

2185
    This runs on the master node.
2186

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

    
2197
  def CheckPrereq(self):
2198
    """Check prerequisites.
2199

2200
    This only checks the instance list against the existing names.
2201

2202
    """
2203
    node = self.node = self.cfg.GetNodeInfo(self.op.node_name)
2204

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

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

    
2226
    return
2227

    
2228
  def Exec(self, feedback_fn):
2229
    """Modifies a node.
2230

2231
    """
2232
    node = self.node
2233

    
2234
    result = []
2235

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

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

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

    
2260
    return result
2261

    
2262

    
2263
class LUQueryClusterInfo(NoHooksLU):
2264
  """Query cluster configuration.
2265

2266
  """
2267
  _OP_REQP = []
2268
  REQ_BGL = False
2269

    
2270
  def ExpandNames(self):
2271
    self.needed_locks = {}
2272

    
2273
  def CheckPrereq(self):
2274
    """No prerequsites needed for this LU.
2275

2276
    """
2277
    pass
2278

    
2279
  def Exec(self, feedback_fn):
2280
    """Return cluster config.
2281

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

    
2300
    return result
2301

    
2302

    
2303
class LUQueryConfigValues(NoHooksLU):
2304
  """Return configuration values.
2305

2306
  """
2307
  _OP_REQP = []
2308
  REQ_BGL = False
2309
  _FIELDS_DYNAMIC = utils.FieldSet()
2310
  _FIELDS_STATIC = utils.FieldSet("cluster_name", "master_node", "drain_flag")
2311

    
2312
  def ExpandNames(self):
2313
    self.needed_locks = {}
2314

    
2315
    _CheckOutputFields(static=self._FIELDS_STATIC,
2316
                       dynamic=self._FIELDS_DYNAMIC,
2317
                       selected=self.op.output_fields)
2318

    
2319
  def CheckPrereq(self):
2320
    """No prerequisites.
2321

2322
    """
2323
    pass
2324

    
2325
  def Exec(self, feedback_fn):
2326
    """Dump a representation of the cluster config to the standard output.
2327

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

    
2342

    
2343
class LUActivateInstanceDisks(NoHooksLU):
2344
  """Bring up an instance's disks.
2345

2346
  """
2347
  _OP_REQP = ["instance_name"]
2348
  REQ_BGL = False
2349

    
2350
  def ExpandNames(self):
2351
    self._ExpandAndLockInstance()
2352
    self.needed_locks[locking.LEVEL_NODE] = []
2353
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
2354

    
2355
  def DeclareLocks(self, level):
2356
    if level == locking.LEVEL_NODE:
2357
      self._LockInstancesNodes()
2358

    
2359
  def CheckPrereq(self):
2360
    """Check prerequisites.
2361

2362
    This checks that the instance is in the cluster.
2363

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

    
2370
  def Exec(self, feedback_fn):
2371
    """Activate the disks.
2372

2373
    """
2374
    disks_ok, disks_info = _AssembleInstanceDisks(self, self.instance)
2375
    if not disks_ok:
2376
      raise errors.OpExecError("Cannot activate block devices")
2377

    
2378
    return disks_info
2379

    
2380

    
2381
def _AssembleInstanceDisks(lu, instance, ignore_secondaries=False):
2382
  """Prepare the block devices for an instance.
2383

2384
  This sets up the block devices on all nodes.
2385

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

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

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

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

    
2422
  # FIXME: race condition on drbd migration to primary
2423

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

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

    
2444
  return disks_ok, device_info
2445

    
2446

    
2447
def _StartInstanceDisks(lu, instance, force):
2448
  """Start the disks of an instance.
2449

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

    
2461

    
2462
class LUDeactivateInstanceDisks(NoHooksLU):
2463
  """Shutdown an instance's disks.
2464

2465
  """
2466
  _OP_REQP = ["instance_name"]
2467
  REQ_BGL = False
2468

    
2469
  def ExpandNames(self):
2470
    self._ExpandAndLockInstance()
2471
    self.needed_locks[locking.LEVEL_NODE] = []
2472
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
2473

    
2474
  def DeclareLocks(self, level):
2475
    if level == locking.LEVEL_NODE:
2476
      self._LockInstancesNodes()
2477

    
2478
  def CheckPrereq(self):
2479
    """Check prerequisites.
2480

2481
    This checks that the instance is in the cluster.
2482

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

    
2488
  def Exec(self, feedback_fn):
2489
    """Deactivate the disks
2490

2491
    """
2492
    instance = self.instance
2493
    _SafeShutdownInstanceDisks(self, instance)
2494

    
2495

    
2496
def _SafeShutdownInstanceDisks(lu, instance):
2497
  """Shutdown block devices of an instance.
2498

2499
  This function checks if an instance is running, before calling
2500
  _ShutdownInstanceDisks.
2501

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

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

    
2514
  _ShutdownInstanceDisks(lu, instance)
2515

    
2516

    
2517
def _ShutdownInstanceDisks(lu, instance, ignore_primary=False):
2518
  """Shutdown block devices of an instance.
2519

2520
  This does the shutdown on all nodes of the instance.
2521

2522
  If the ignore_primary is false, errors on the primary node are
2523
  ignored.
2524

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

    
2538

    
2539
def _CheckNodeFreeMemory(lu, node, reason, requested, hypervisor_name):
2540
  """Checks if a node has enough free memory.
2541

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

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

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

    
2572

    
2573
class LUStartupInstance(LogicalUnit):
2574
  """Starts an instance.
2575

2576
  """
2577
  HPATH = "instance-start"
2578
  HTYPE = constants.HTYPE_INSTANCE
2579
  _OP_REQP = ["instance_name", "force"]
2580
  REQ_BGL = False
2581

    
2582
  def ExpandNames(self):
2583
    self._ExpandAndLockInstance()
2584

    
2585
  def BuildHooksEnv(self):
2586
    """Build hooks env.
2587

2588
    This runs on master, primary and secondary nodes of the instance.
2589

2590
    """
2591
    env = {
2592
      "FORCE": self.op.force,
2593
      }
2594
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
2595
    nl = ([self.cfg.GetMasterNode(), self.instance.primary_node] +
2596
          list(self.instance.secondary_nodes))
2597
    return env, nl, nl
2598

    
2599
  def CheckPrereq(self):
2600
    """Check prerequisites.
2601

2602
    This checks that the instance is in the cluster.
2603

2604
    """
2605
    self.instance = instance = self.cfg.GetInstanceInfo(self.op.instance_name)
2606
    assert self.instance is not None, \
2607
      "Cannot retrieve locked instance %s" % self.op.instance_name
2608

    
2609
    _CheckNodeOnline(self, instance.primary_node)
2610

    
2611
    bep = self.cfg.GetClusterInfo().FillBE(instance)
2612
    # check bridges existance
2613
    _CheckInstanceBridgesExist(self, instance)
2614

    
2615
    _CheckNodeFreeMemory(self, instance.primary_node,
2616
                         "starting instance %s" % instance.name,
2617
                         bep[constants.BE_MEMORY], instance.hypervisor)
2618

    
2619
  def Exec(self, feedback_fn):
2620
    """Start the instance.
2621

2622
    """
2623
    instance = self.instance
2624
    force = self.op.force
2625
    extra_args = getattr(self.op, "extra_args", "")
2626

    
2627
    self.cfg.MarkInstanceUp(instance.name)
2628

    
2629
    node_current = instance.primary_node
2630

    
2631
    _StartInstanceDisks(self, instance, force)
2632

    
2633
    result = self.rpc.call_instance_start(node_current, instance, extra_args)
2634
    if result.failed or not result.data:
2635
      _ShutdownInstanceDisks(self, instance)
2636
      raise errors.OpExecError("Could not start instance")
2637

    
2638

    
2639
class LURebootInstance(LogicalUnit):
2640
  """Reboot an instance.
2641

2642
  """
2643
  HPATH = "instance-reboot"
2644
  HTYPE = constants.HTYPE_INSTANCE
2645
  _OP_REQP = ["instance_name", "ignore_secondaries", "reboot_type"]
2646
  REQ_BGL = False
2647

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

    
2658
  def BuildHooksEnv(self):
2659
    """Build hooks env.
2660

2661
    This runs on master, primary and secondary nodes of the instance.
2662

2663
    """
2664
    env = {
2665
      "IGNORE_SECONDARIES": self.op.ignore_secondaries,
2666
      }
2667
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
2668
    nl = ([self.cfg.GetMasterNode(), self.instance.primary_node] +
2669
          list(self.instance.secondary_nodes))
2670
    return env, nl, nl
2671

    
2672
  def CheckPrereq(self):
2673
    """Check prerequisites.
2674

2675
    This checks that the instance is in the cluster.
2676

2677
    """
2678
    self.instance = instance = self.cfg.GetInstanceInfo(self.op.instance_name)
2679
    assert self.instance is not None, \
2680
      "Cannot retrieve locked instance %s" % self.op.instance_name
2681

    
2682
    _CheckNodeOnline(self, instance.primary_node)
2683

    
2684
    # check bridges existance
2685
    _CheckInstanceBridgesExist(self, instance)
2686

    
2687
  def Exec(self, feedback_fn):
2688
    """Reboot the instance.
2689

2690
    """
2691
    instance = self.instance
2692
    ignore_secondaries = self.op.ignore_secondaries
2693
    reboot_type = self.op.reboot_type
2694
    extra_args = getattr(self.op, "extra_args", "")
2695

    
2696
    node_current = instance.primary_node
2697

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

    
2714
    self.cfg.MarkInstanceUp(instance.name)
2715

    
2716

    
2717
class LUShutdownInstance(LogicalUnit):
2718
  """Shutdown an instance.
2719

2720
  """
2721
  HPATH = "instance-stop"
2722
  HTYPE = constants.HTYPE_INSTANCE
2723
  _OP_REQP = ["instance_name"]
2724
  REQ_BGL = False
2725

    
2726
  def ExpandNames(self):
2727
    self._ExpandAndLockInstance()
2728

    
2729
  def BuildHooksEnv(self):
2730
    """Build hooks env.
2731

2732
    This runs on master, primary and secondary nodes of the instance.
2733

2734
    """
2735
    env = _BuildInstanceHookEnvByObject(self, self.instance)
2736
    nl = ([self.cfg.GetMasterNode(), self.instance.primary_node] +
2737
          list(self.instance.secondary_nodes))
2738
    return env, nl, nl
2739

    
2740
  def CheckPrereq(self):
2741
    """Check prerequisites.
2742

2743
    This checks that the instance is in the cluster.
2744

2745
    """
2746
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_name)
2747
    assert self.instance is not None, \
2748
      "Cannot retrieve locked instance %s" % self.op.instance_name
2749
    _CheckNodeOnline(self, self.instance.primary_node)
2750

    
2751
  def Exec(self, feedback_fn):
2752
    """Shutdown the instance.
2753

2754
    """
2755
    instance = self.instance
2756
    node_current = instance.primary_node
2757
    self.cfg.MarkInstanceDown(instance.name)
2758
    result = self.rpc.call_instance_shutdown(node_current, instance)
2759
    if result.failed or not result.data:
2760
      self.proc.LogWarning("Could not shutdown instance")
2761

    
2762
    _ShutdownInstanceDisks(self, instance)
2763

    
2764

    
2765
class LUReinstallInstance(LogicalUnit):
2766
  """Reinstall an instance.
2767

2768
  """
2769
  HPATH = "instance-reinstall"
2770
  HTYPE = constants.HTYPE_INSTANCE
2771
  _OP_REQP = ["instance_name"]
2772
  REQ_BGL = False
2773

    
2774
  def ExpandNames(self):
2775
    self._ExpandAndLockInstance()
2776

    
2777
  def BuildHooksEnv(self):
2778
    """Build hooks env.
2779

2780
    This runs on master, primary and secondary nodes of the instance.
2781

2782
    """
2783
    env = _BuildInstanceHookEnvByObject(self, self.instance)
2784
    nl = ([self.cfg.GetMasterNode(), self.instance.primary_node] +
2785
          list(self.instance.secondary_nodes))
2786
    return env, nl, nl
2787

    
2788
  def CheckPrereq(self):
2789
    """Check prerequisites.
2790

2791
    This checks that the instance is in the cluster and is not running.
2792

2793
    """
2794
    instance = self.cfg.GetInstanceInfo(self.op.instance_name)
2795
    assert instance is not None, \
2796
      "Cannot retrieve locked instance %s" % self.op.instance_name
2797
    _CheckNodeOnline(self, instance.primary_node)
2798

    
2799
    if instance.disk_template == constants.DT_DISKLESS:
2800
      raise errors.OpPrereqError("Instance '%s' has no disks" %
2801
                                 self.op.instance_name)
2802
    if instance.status != "down":
2803
      raise errors.OpPrereqError("Instance '%s' is marked to be up" %
2804
                                 self.op.instance_name)
2805
    remote_info = self.rpc.call_instance_info(instance.primary_node,
2806
                                              instance.name,
2807
                                              instance.hypervisor)
2808
    if remote_info.failed or remote_info.data:
2809
      raise errors.OpPrereqError("Instance '%s' is running on the node %s" %
2810
                                 (self.op.instance_name,
2811
                                  instance.primary_node))
2812

    
2813
    self.op.os_type = getattr(self.op, "os_type", None)
2814
    if self.op.os_type is not None:
2815
      # OS verification
2816
      pnode = self.cfg.GetNodeInfo(
2817
        self.cfg.ExpandNodeName(instance.primary_node))
2818
      if pnode is None:
2819
        raise errors.OpPrereqError("Primary node '%s' is unknown" %
2820
                                   self.op.pnode)
2821
      result = self.rpc.call_os_get(pnode.name, self.op.os_type)
2822
      result.Raise()
2823
      if not isinstance(result.data, objects.OS):
2824
        raise errors.OpPrereqError("OS '%s' not in supported OS list for"
2825
                                   " primary node"  % self.op.os_type)
2826

    
2827
    self.instance = instance
2828

    
2829
  def Exec(self, feedback_fn):
2830
    """Reinstall the instance.
2831

2832
    """
2833
    inst = self.instance
2834

    
2835
    if self.op.os_type is not None:
2836
      feedback_fn("Changing OS to '%s'..." % self.op.os_type)
2837
      inst.os = self.op.os_type
2838
      self.cfg.Update(inst)
2839

    
2840
    _StartInstanceDisks(self, inst, None)
2841
    try:
2842
      feedback_fn("Running the instance OS create scripts...")
2843
      result = self.rpc.call_instance_os_add(inst.primary_node, inst)
2844
      result.Raise()
2845
      if not result.data:
2846
        raise errors.OpExecError("Could not install OS for instance %s"
2847
                                 " on node %s" %
2848
                                 (inst.name, inst.primary_node))
2849
    finally:
2850
      _ShutdownInstanceDisks(self, inst)
2851

    
2852

    
2853
class LURenameInstance(LogicalUnit):
2854
  """Rename an instance.
2855

2856
  """
2857
  HPATH = "instance-rename"
2858
  HTYPE = constants.HTYPE_INSTANCE
2859
  _OP_REQP = ["instance_name", "new_name"]
2860

    
2861
  def BuildHooksEnv(self):
2862
    """Build hooks env.
2863

2864
    This runs on master, primary and secondary nodes of the instance.
2865

2866
    """
2867
    env = _BuildInstanceHookEnvByObject(self, self.instance)
2868
    env["INSTANCE_NEW_NAME"] = self.op.new_name
2869
    nl = ([self.cfg.GetMasterNode(), self.instance.primary_node] +
2870
          list(self.instance.secondary_nodes))
2871
    return env, nl, nl
2872

    
2873
  def CheckPrereq(self):
2874
    """Check prerequisites.
2875

2876
    This checks that the instance is in the cluster and is not running.
2877

2878
    """
2879
    instance = self.cfg.GetInstanceInfo(
2880
      self.cfg.ExpandInstanceName(self.op.instance_name))
2881
    if instance is None:
2882
      raise errors.OpPrereqError("Instance '%s' not known" %
2883
                                 self.op.instance_name)
2884
    _CheckNodeOnline(self, instance.primary_node)
2885

    
2886
    if instance.status != "down":
2887
      raise errors.OpPrereqError("Instance '%s' is marked to be up" %
2888
                                 self.op.instance_name)
2889
    remote_info = self.rpc.call_instance_info(instance.primary_node,
2890
                                              instance.name,
2891
                                              instance.hypervisor)
2892
    remote_info.Raise()
2893
    if remote_info.data:
2894
      raise errors.OpPrereqError("Instance '%s' is running on the node %s" %
2895
                                 (self.op.instance_name,
2896
                                  instance.primary_node))
2897
    self.instance = instance
2898

    
2899
    # new name verification
2900
    name_info = utils.HostInfo(self.op.new_name)
2901

    
2902
    self.op.new_name = new_name = name_info.name
2903
    instance_list = self.cfg.GetInstanceList()
2904
    if new_name in instance_list:
2905
      raise errors.OpPrereqError("Instance '%s' is already in the cluster" %
2906
                                 new_name)
2907

    
2908
    if not getattr(self.op, "ignore_ip", False):
2909
      if utils.TcpPing(name_info.ip, constants.DEFAULT_NODED_PORT):
2910
        raise errors.OpPrereqError("IP %s of instance %s already in use" %
2911
                                   (name_info.ip, new_name))
2912

    
2913

    
2914
  def Exec(self, feedback_fn):
2915
    """Reinstall the instance.
2916

2917
    """
2918
    inst = self.instance
2919
    old_name = inst.name
2920

    
2921
    if inst.disk_template == constants.DT_FILE:
2922
      old_file_storage_dir = os.path.dirname(inst.disks[0].logical_id[1])
2923

    
2924
    self.cfg.RenameInstance(inst.name, self.op.new_name)
2925
    # Change the instance lock. This is definitely safe while we hold the BGL
2926
    self.context.glm.remove(locking.LEVEL_INSTANCE, old_name)
2927
    self.context.glm.add(locking.LEVEL_INSTANCE, self.op.new_name)
2928

    
2929
    # re-read the instance from the configuration after rename
2930
    inst = self.cfg.GetInstanceInfo(self.op.new_name)
2931

    
2932
    if inst.disk_template == constants.DT_FILE:
2933
      new_file_storage_dir = os.path.dirname(inst.disks[0].logical_id[1])
2934
      result = self.rpc.call_file_storage_dir_rename(inst.primary_node,
2935
                                                     old_file_storage_dir,
2936
                                                     new_file_storage_dir)
2937
      result.Raise()
2938
      if not result.data:
2939
        raise errors.OpExecError("Could not connect to node '%s' to rename"
2940
                                 " directory '%s' to '%s' (but the instance"
2941
                                 " has been renamed in Ganeti)" % (
2942
                                 inst.primary_node, old_file_storage_dir,
2943
                                 new_file_storage_dir))
2944

    
2945
      if not result.data[0]:
2946
        raise errors.OpExecError("Could not rename directory '%s' to '%s'"
2947
                                 " (but the instance has been renamed in"
2948
                                 " Ganeti)" % (old_file_storage_dir,
2949
                                               new_file_storage_dir))
2950

    
2951
    _StartInstanceDisks(self, inst, None)
2952
    try:
2953
      result = self.rpc.call_instance_run_rename(inst.primary_node, inst,
2954
                                                 old_name)
2955
      if result.failed or not result.data:
2956
        msg = ("Could not run OS rename script for instance %s on node %s"
2957
               " (but the instance has been renamed in Ganeti)" %
2958
               (inst.name, inst.primary_node))
2959
        self.proc.LogWarning(msg)
2960
    finally:
2961
      _ShutdownInstanceDisks(self, inst)
2962

    
2963

    
2964
class LURemoveInstance(LogicalUnit):
2965
  """Remove an instance.
2966

2967
  """
2968
  HPATH = "instance-remove"
2969
  HTYPE = constants.HTYPE_INSTANCE
2970
  _OP_REQP = ["instance_name", "ignore_failures"]
2971
  REQ_BGL = False
2972

    
2973
  def ExpandNames(self):
2974
    self._ExpandAndLockInstance()
2975
    self.needed_locks[locking.LEVEL_NODE] = []
2976
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
2977

    
2978
  def DeclareLocks(self, level):
2979
    if level == locking.LEVEL_NODE:
2980
      self._LockInstancesNodes()
2981

    
2982
  def BuildHooksEnv(self):
2983
    """Build hooks env.
2984

2985
    This runs on master, primary and secondary nodes of the instance.
2986

2987
    """
2988
    env = _BuildInstanceHookEnvByObject(self, self.instance)
2989
    nl = [self.cfg.GetMasterNode()]
2990
    return env, nl, nl
2991

    
2992
  def CheckPrereq(self):
2993
    """Check prerequisites.
2994

2995
    This checks that the instance is in the cluster.
2996

2997
    """
2998
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_name)
2999
    assert self.instance is not None, \
3000
      "Cannot retrieve locked instance %s" % self.op.instance_name
3001

    
3002
  def Exec(self, feedback_fn):
3003
    """Remove the instance.
3004

3005
    """
3006
    instance = self.instance
3007
    logging.info("Shutting down instance %s on node %s",
3008
                 instance.name, instance.primary_node)
3009

    
3010
    result = self.rpc.call_instance_shutdown(instance.primary_node, instance)
3011
    if result.failed or not result.data:
3012
      if self.op.ignore_failures:
3013
        feedback_fn("Warning: can't shutdown instance")
3014
      else:
3015
        raise errors.OpExecError("Could not shutdown instance %s on node %s" %
3016
                                 (instance.name, instance.primary_node))
3017

    
3018
    logging.info("Removing block devices for instance %s", instance.name)
3019

    
3020
    if not _RemoveDisks(self, instance):
3021
      if self.op.ignore_failures:
3022
        feedback_fn("Warning: can't remove instance's disks")
3023
      else:
3024
        raise errors.OpExecError("Can't remove instance's disks")
3025

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

    
3028
    self.cfg.RemoveInstance(instance.name)
3029
    self.remove_locks[locking.LEVEL_INSTANCE] = instance.name
3030

    
3031

    
3032
class LUQueryInstances(NoHooksLU):
3033
  """Logical unit for querying instances.
3034

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

    
3055

    
3056
  def ExpandNames(self):
3057
    _CheckOutputFields(static=self._FIELDS_STATIC,
3058
                       dynamic=self._FIELDS_DYNAMIC,
3059
                       selected=self.op.output_fields)
3060

    
3061
    self.needed_locks = {}
3062
    self.share_locks[locking.LEVEL_INSTANCE] = 1
3063
    self.share_locks[locking.LEVEL_NODE] = 1
3064

    
3065
    if self.op.names:
3066
      self.wanted = _GetWantedInstances(self, self.op.names)
3067
    else:
3068
      self.wanted = locking.ALL_SET
3069

    
3070
    self.do_locking = self._FIELDS_STATIC.NonMatching(self.op.output_fields)
3071
    if self.do_locking:
3072
      self.needed_locks[locking.LEVEL_INSTANCE] = self.wanted
3073
      self.needed_locks[locking.LEVEL_NODE] = []
3074
      self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
3075

    
3076
  def DeclareLocks(self, level):
3077
    if level == locking.LEVEL_NODE and self.do_locking:
3078
      self._LockInstancesNodes()
3079

    
3080
  def CheckPrereq(self):
3081
    """Check prerequisites.
3082

3083
    """
3084
    pass
3085

    
3086
  def Exec(self, feedback_fn):
3087
    """Computes the list of nodes and their attributes.
3088

3089
    """
3090
    all_info = self.cfg.GetAllInstancesInfo()
3091
    if self.do_locking:
3092
      instance_names = self.acquired_locks[locking.LEVEL_INSTANCE]
3093
    elif self.wanted != locking.ALL_SET:
3094
      instance_names = self.wanted
3095
      missing = set(instance_names).difference(all_info.keys())
3096
      if missing:
3097
        raise errors.OpExecError(
3098
          "Some instances were removed before retrieving their data: %s"
3099
          % missing)
3100
    else:
3101
      instance_names = all_info.keys()
3102

    
3103
    instance_names = utils.NiceSort(instance_names)
3104
    instance_list = [all_info[iname] for iname in instance_names]
3105

    
3106
    # begin data gathering
3107

    
3108
    nodes = frozenset([inst.primary_node for inst in instance_list])
3109
    hv_list = list(set([inst.hypervisor for inst in instance_list]))
3110

    
3111
    bad_nodes = []
3112
    off_nodes = []
3113
    if self.do_locking:
3114
      live_data = {}
3115
      node_data = self.rpc.call_all_instances_info(nodes, hv_list)
3116
      for name in nodes:
3117
        result = node_data[name]
3118
        if result.offline:
3119
          # offline nodes will be in both lists
3120
          off_nodes.append(name)
3121
        if result.failed:
3122
          bad_nodes.append(name)
3123
        else:
3124
          if result.data:
3125
            live_data.update(result.data)
3126
            # else no instance is alive
3127
    else:
3128
      live_data = dict([(name, {}) for name in instance_names])
3129

    
3130
    # end data gathering
3131

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

    
3257
    return output
3258

    
3259

    
3260
class LUFailoverInstance(LogicalUnit):
3261
  """Failover an instance.
3262

3263
  """
3264
  HPATH = "instance-failover"
3265
  HTYPE = constants.HTYPE_INSTANCE
3266
  _OP_REQP = ["instance_name", "ignore_consistency"]
3267
  REQ_BGL = False
3268

    
3269
  def ExpandNames(self):
3270
    self._ExpandAndLockInstance()
3271
    self.needed_locks[locking.LEVEL_NODE] = []
3272
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
3273

    
3274
  def DeclareLocks(self, level):
3275
    if level == locking.LEVEL_NODE:
3276
      self._LockInstancesNodes()
3277

    
3278
  def BuildHooksEnv(self):
3279
    """Build hooks env.
3280

3281
    This runs on master, primary and secondary nodes of the instance.
3282

3283
    """
3284
    env = {
3285
      "IGNORE_CONSISTENCY": self.op.ignore_consistency,
3286
      }
3287
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
3288
    nl = [self.cfg.GetMasterNode()] + list(self.instance.secondary_nodes)
3289
    return env, nl, nl
3290

    
3291
  def CheckPrereq(self):
3292
    """Check prerequisites.
3293

3294
    This checks that the instance is in the cluster.
3295

3296
    """
3297
    self.instance = instance = self.cfg.GetInstanceInfo(self.op.instance_name)
3298
    assert self.instance is not None, \
3299
      "Cannot retrieve locked instance %s" % self.op.instance_name
3300

    
3301
    bep = self.cfg.GetClusterInfo().FillBE(instance)
3302
    if instance.disk_template not in constants.DTS_NET_MIRROR:
3303
      raise errors.OpPrereqError("Instance's disk layout is not"
3304
                                 " network mirrored, cannot failover.")
3305

    
3306
    secondary_nodes = instance.secondary_nodes
3307
    if not secondary_nodes:
3308
      raise errors.ProgrammerError("no secondary node but using "
3309
                                   "a mirrored disk template")
3310

    
3311
    target_node = secondary_nodes[0]
3312
    _CheckNodeOnline(self, target_node)
3313
    # check memory requirements on the secondary node
3314
    _CheckNodeFreeMemory(self, target_node, "failing over instance %s" %
3315
                         instance.name, bep[constants.BE_MEMORY],
3316
                         instance.hypervisor)
3317

    
3318
    # check bridge existance
3319
    brlist = [nic.bridge for nic in instance.nics]
3320
    result = self.rpc.call_bridges_exist(target_node, brlist)
3321
    result.Raise()
3322
    if not result.data:
3323
      raise errors.OpPrereqError("One or more target bridges %s does not"
3324
                                 " exist on destination node '%s'" %
3325
                                 (brlist, target_node))
3326

    
3327
  def Exec(self, feedback_fn):
3328
    """Failover an instance.
3329

3330
    The failover is done by shutting it down on its present node and
3331
    starting it on the secondary.
3332

3333
    """
3334
    instance = self.instance
3335

    
3336
    source_node = instance.primary_node
3337
    target_node = instance.secondary_nodes[0]
3338

    
3339
    feedback_fn("* checking disk consistency between source and target")
3340
    for dev in instance.disks:
3341
      # for drbd, these are drbd over lvm
3342
      if not _CheckDiskConsistency(self, dev, target_node, False):
3343
        if instance.status == "up" and not self.op.ignore_consistency:
3344
          raise errors.OpExecError("Disk %s is degraded on target node,"
3345
                                   " aborting failover." % dev.iv_name)
3346

    
3347
    feedback_fn("* shutting down instance on source node")
3348
    logging.info("Shutting down instance %s on node %s",
3349
                 instance.name, source_node)
3350

    
3351
    result = self.rpc.call_instance_shutdown(source_node, instance)
3352
    if result.failed or not result.data:
3353
      if self.op.ignore_consistency:
3354
        self.proc.LogWarning("Could not shutdown instance %s on node %s."
3355
                             " Proceeding"
3356
                             " anyway. Please make sure node %s is down",
3357
                             instance.name, source_node, source_node)
3358
      else:
3359
        raise errors.OpExecError("Could not shutdown instance %s on node %s" %
3360
                                 (instance.name, source_node))
3361

    
3362
    feedback_fn("* deactivating the instance's disks on source node")
3363
    if not _ShutdownInstanceDisks(self, instance, ignore_primary=True):
3364
      raise errors.OpExecError("Can't shut down the instance's disks.")
3365

    
3366
    instance.primary_node = target_node
3367
    # distribute new instance config to the other nodes
3368
    self.cfg.Update(instance)
3369

    
3370
    # Only start the instance if it's marked as up
3371
    if instance.status == "up":
3372
      feedback_fn("* activating the instance's disks on target node")
3373
      logging.info("Starting instance %s on node %s",
3374
                   instance.name, target_node)
3375

    
3376
      disks_ok, dummy = _AssembleInstanceDisks(self, instance,
3377
                                               ignore_secondaries=True)
3378
      if not disks_ok:
3379
        _ShutdownInstanceDisks(self, instance)
3380
        raise errors.OpExecError("Can't activate the instance's disks")
3381

    
3382
      feedback_fn("* starting the instance on the target node")
3383
      result = self.rpc.call_instance_start(target_node, instance, None)
3384
      if result.failed or not result.data:
3385
        _ShutdownInstanceDisks(self, instance)
3386
        raise errors.OpExecError("Could not start instance %s on node %s." %
3387
                                 (instance.name, target_node))
3388

    
3389

    
3390
class LUMigrateInstance(LogicalUnit):
3391
  """Migrate an instance.
3392

3393
  This is migration without shutting down, compared to the failover,
3394
  which is done with shutdown.
3395

3396
  """
3397
  HPATH = "instance-migrate"
3398
  HTYPE = constants.HTYPE_INSTANCE
3399
  _OP_REQP = ["instance_name", "live", "cleanup"]
3400

    
3401
  REQ_BGL = False
3402

    
3403
  def ExpandNames(self):
3404
    self._ExpandAndLockInstance()
3405
    self.needed_locks[locking.LEVEL_NODE] = []
3406
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
3407

    
3408
  def DeclareLocks(self, level):
3409
    if level == locking.LEVEL_NODE:
3410
      self._LockInstancesNodes()
3411

    
3412
  def BuildHooksEnv(self):
3413
    """Build hooks env.
3414

3415
    This runs on master, primary and secondary nodes of the instance.
3416

3417
    """
3418
    env = _BuildInstanceHookEnvByObject(self, self.instance)
3419
    nl = [self.cfg.GetMasterNode()] + list(self.instance.secondary_nodes)
3420
    return env, nl, nl
3421

    
3422
  def CheckPrereq(self):
3423
    """Check prerequisites.
3424

3425
    This checks that the instance is in the cluster.
3426

3427
    """
3428
    instance = self.cfg.GetInstanceInfo(
3429
      self.cfg.ExpandInstanceName(self.op.instance_name))
3430
    if instance is None:
3431
      raise errors.OpPrereqError("Instance '%s' not known" %
3432
                                 self.op.instance_name)
3433

    
3434
    if instance.disk_template != constants.DT_DRBD8:
3435
      raise errors.OpPrereqError("Instance's disk layout is not"
3436
                                 " drbd8, cannot migrate.")
3437

    
3438
    secondary_nodes = instance.secondary_nodes
3439
    if not secondary_nodes:
3440
      raise errors.ProgrammerError("no secondary node but using "
3441
                                   "drbd8 disk template")
3442

    
3443
    i_be = self.cfg.GetClusterInfo().FillBE(instance)
3444

    
3445
    target_node = secondary_nodes[0]
3446
    # check memory requirements on the secondary node
3447
    _CheckNodeFreeMemory(self, target_node, "migrating instance %s" %
3448
                         instance.name, i_be[constants.BE_MEMORY],
3449
                         instance.hypervisor)
3450

    
3451
    # check bridge existance
3452
    brlist = [nic.bridge for nic in instance.nics]
3453
    result = self.rpc.call_bridges_exist(target_node, brlist)
3454
    if result.failed or not result.data:
3455
      raise errors.OpPrereqError("One or more target bridges %s does not"
3456
                                 " exist on destination node '%s'" %
3457
                                 (brlist, target_node))
3458

    
3459
    if not self.op.cleanup:
3460
      result = self.rpc.call_instance_migratable(instance.primary_node,
3461
                                                 instance)
3462
      msg = result.RemoteFailMsg()
3463
      if msg:
3464
        raise errors.OpPrereqError("Can't migrate: %s - please use failover" %
3465
                                   msg)
3466

    
3467
    self.instance = instance
3468

    
3469
  def _WaitUntilSync(self):
3470
    """Poll with custom rpc for disk sync.
3471

3472
    This uses our own step-based rpc call.
3473

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

    
3497
  def _EnsureSecondary(self, node):
3498
    """Demote a node to secondary.
3499

3500
    """
3501
    self.feedback_fn("* switching node %s to secondary mode" % node)
3502

    
3503
    for dev in self.instance.disks:
3504
      self.cfg.SetDiskID(dev, node)
3505

    
3506
    result = self.rpc.call_blockdev_close(node, self.instance.name,
3507
                                          self.instance.disks)
3508
    msg = result.RemoteFailMsg()
3509
    if msg:
3510
      raise errors.OpExecError("Cannot change disk to secondary on node %s,"
3511
                               " error %s" % (node, msg))
3512

    
3513
  def _GoStandalone(self):
3514
    """Disconnect from the network.
3515

3516
    """
3517
    self.feedback_fn("* changing into standalone mode")
3518
    result = self.rpc.call_drbd_disconnect_net(self.all_nodes, self.nodes_ip,
3519
                                               self.instance.disks)
3520
    for node, nres in result.items():
3521
      msg = nres.RemoteFailMsg()
3522
      if msg:
3523
        raise errors.OpExecError("Cannot disconnect disks node %s,"
3524
                                 " error %s" % (node, msg))
3525

    
3526
  def _GoReconnect(self, multimaster):
3527
    """Reconnect to the network.
3528

3529
    """
3530
    if multimaster:
3531
      msg = "dual-master"
3532
    else:
3533
      msg = "single-master"
3534
    self.feedback_fn("* changing disks into %s mode" % msg)
3535
    result = self.rpc.call_drbd_attach_net(self.all_nodes, self.nodes_ip,
3536
                                           self.instance.disks,
3537
                                           self.instance.name, multimaster)
3538
    for node, nres in result.items():
3539
      msg = nres.RemoteFailMsg()
3540
      if msg:
3541
        raise errors.OpExecError("Cannot change disks config on node %s,"
3542
                                 " error: %s" % (node, msg))
3543

    
3544
  def _ExecCleanup(self):
3545
    """Try to cleanup after a failed migration.
3546

3547
    The cleanup is done by:
3548
      - check that the instance is running only on one node
3549
        (and update the config if needed)
3550
      - change disks on its secondary node to secondary
3551
      - wait until disks are fully synchronized
3552
      - disconnect from the network
3553
      - change disks into single-master mode
3554
      - wait again until disks are fully synchronized
3555

3556
    """
3557
    instance = self.instance
3558
    target_node = self.target_node
3559
    source_node = self.source_node
3560

    
3561
    # check running on only one node
3562
    self.feedback_fn("* checking where the instance actually runs"
3563
                     " (if this hangs, the hypervisor might be in"
3564
                     " a bad state)")
3565
    ins_l = self.rpc.call_instance_list(self.all_nodes, [instance.hypervisor])
3566
    for node, result in ins_l.items():
3567
      result.Raise()
3568
      if not isinstance(result.data, list):
3569
        raise errors.OpExecError("Can't contact node '%s'" % node)
3570

    
3571
    runningon_source = instance.name in ins_l[source_node].data
3572
    runningon_target = instance.name in ins_l[target_node].data
3573

    
3574
    if runningon_source and runningon_target:
3575
      raise errors.OpExecError("Instance seems to be running on two nodes,"
3576
                               " or the hypervisor is confused. You will have"
3577
                               " to ensure manually that it runs only on one"
3578
                               " and restart this operation.")
3579

    
3580
    if not (runningon_source or runningon_target):
3581
      raise errors.OpExecError("Instance does not seem to be running at all."
3582
                               " In this case, it's safer to repair by"
3583
                               " running 'gnt-instance stop' to ensure disk"
3584
                               " shutdown, and then restarting it.")
3585

    
3586
    if runningon_target:
3587
      # the migration has actually succeeded, we need to update the config
3588
      self.feedback_fn("* instance running on secondary node (%s),"
3589
                       " updating config" % target_node)
3590
      instance.primary_node = target_node
3591
      self.cfg.Update(instance)
3592
      demoted_node = source_node
3593
    else:
3594
      self.feedback_fn("* instance confirmed to be running on its"
3595
                       " primary node (%s)" % source_node)
3596
      demoted_node = target_node
3597

    
3598
    self._EnsureSecondary(demoted_node)
3599
    try:
3600
      self._WaitUntilSync()
3601
    except errors.OpExecError:
3602
      # we ignore here errors, since if the device is standalone, it
3603
      # won't be able to sync
3604
      pass
3605
    self._GoStandalone()
3606
    self._GoReconnect(False)
3607
    self._WaitUntilSync()
3608

    
3609
    self.feedback_fn("* done")
3610

    
3611
  def _ExecMigration(self):
3612
    """Migrate an instance.
3613

3614
    The migrate is done by:
3615
      - change the disks into dual-master mode
3616
      - wait until disks are fully synchronized again
3617
      - migrate the instance
3618
      - change disks on the new secondary node (the old primary) to secondary
3619
      - wait until disks are fully synchronized
3620
      - change disks into single-master mode
3621

3622
    """
3623
    instance = self.instance
3624
    target_node = self.target_node
3625
    source_node = self.source_node
3626

    
3627
    self.feedback_fn("* checking disk consistency between source and target")
3628
    for dev in instance.disks:
3629
      if not _CheckDiskConsistency(self, dev, target_node, False):
3630
        raise errors.OpExecError("Disk %s is degraded or not fully"
3631
                                 " synchronized on target node,"
3632
                                 " aborting migrate." % dev.iv_name)
3633

    
3634
    self._EnsureSecondary(target_node)
3635
    self._GoStandalone()
3636
    self._GoReconnect(True)
3637
    self._WaitUntilSync()
3638

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

    
3659
      raise errors.OpExecError("Could not migrate instance %s: %s" %
3660
                               (instance.name, msg))
3661
    time.sleep(10)
3662

    
3663
    instance.primary_node = target_node
3664
    # distribute new instance config to the other nodes
3665
    self.cfg.Update(instance)
3666

    
3667
    self._EnsureSecondary(source_node)
3668
    self._WaitUntilSync()
3669
    self._GoStandalone()
3670
    self._GoReconnect(False)
3671
    self._WaitUntilSync()
3672

    
3673
    self.feedback_fn("* done")
3674

    
3675
  def Exec(self, feedback_fn):
3676
    """Perform the migration.
3677

3678
    """
3679
    self.feedback_fn = feedback_fn
3680

    
3681
    self.source_node = self.instance.primary_node
3682
    self.target_node = self.instance.secondary_nodes[0]
3683
    self.all_nodes = [self.source_node, self.target_node]
3684
    self.nodes_ip = {
3685
      self.source_node: self.cfg.GetNodeInfo(self.source_node).secondary_ip,
3686
      self.target_node: self.cfg.GetNodeInfo(self.target_node).secondary_ip,
3687
      }
3688
    if self.op.cleanup:
3689
      return self._ExecCleanup()
3690
    else:
3691
      return self._ExecMigration()
3692

    
3693

    
3694
def _CreateBlockDev(lu, node, instance, device, force_create,
3695
                    info, force_open):
3696
  """Create a tree of block devices on a given node.
3697

3698
  If this device type has to be created on secondaries, create it and
3699
  all its children.
3700

3701
  If not, just recurse to children keeping the same 'force' value.
3702

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

3721
  """
3722
  if device.CreateOnSecondary():
3723
    force_create = True
3724

    
3725
  if device.children:
3726
    for child in device.children:
3727
      _CreateBlockDev(lu, node, instance, child, force_create,
3728
                      info, force_open)
3729

    
3730
  if not force_create:
3731
    return
3732

    
3733
  lu.cfg.SetDiskID(device, node)
3734
  new_id = lu.rpc.call_blockdev_create(node, device, device.size,
3735
                                       instance.name, force_open, info)
3736
  if new_id.failed or not new_id.data:
3737
    raise errors.OpExecError("Can't create block device %s on"
3738
                             " node %s" % (device, node))
3739
  if device.physical_id is None:
3740
    device.physical_id = new_id
3741

    
3742

    
3743
def _GenerateUniqueNames(lu, exts):
3744
  """Generate a suitable LV name.
3745

3746
  This will generate a logical volume name for the given instance.
3747

3748
  """
3749
  results = []
3750
  for val in exts:
3751
    new_id = lu.cfg.GenerateUniqueID()
3752
    results.append("%s%s" % (new_id, val))
3753
  return results
3754

    
3755

    
3756
def _GenerateDRBD8Branch(lu, primary, secondary, size, names, iv_name,
3757
                         p_minor, s_minor):
3758
  """Generate a drbd8 device complete with its children.
3759

3760
  """
3761
  port = lu.cfg.AllocatePort()
3762
  vgname = lu.cfg.GetVGName()
3763
  shared_secret = lu.cfg.GenerateDRBDSecret()
3764
  dev_data = objects.Disk(dev_type=constants.LD_LV, size=size,
3765
                          logical_id=(vgname, names[0]))
3766
  dev_meta = objects.Disk(dev_type=constants.LD_LV, size=128,
3767
                          logical_id=(vgname, names[1]))
3768
  drbd_dev = objects.Disk(dev_type=constants.LD_DRBD8, size=size,
3769
                          logical_id=(primary, secondary, port,
3770
                                      p_minor, s_minor,
3771
                                      shared_secret),
3772
                          children=[dev_data, dev_meta],
3773
                          iv_name=iv_name)
3774
  return drbd_dev
3775

    
3776

    
3777
def _GenerateDiskTemplate(lu, template_name,
3778
                          instance_name, primary_node,
3779
                          secondary_nodes, disk_info,
3780
                          file_storage_dir, file_driver,
3781
                          base_index):
3782
  """Generate the entire disk layout for a given template type.
3783

3784
  """
3785
  #TODO: compute space requirements
3786

    
3787
  vgname = lu.cfg.GetVGName()
3788
  disk_count = len(disk_info)
3789
  disks = []
3790
  if template_name == constants.DT_DISKLESS:
3791
    pass
3792
  elif template_name == constants.DT_PLAIN:
3793
    if len(secondary_nodes) != 0:
3794
      raise errors.ProgrammerError("Wrong template configuration")
3795

    
3796
    names = _GenerateUniqueNames(lu, [".disk%d" % i
3797
                                      for i in range(disk_count)])
3798
    for idx, disk in enumerate(disk_info):
3799
      disk_index = idx + base_index
3800
      disk_dev = objects.Disk(dev_type=constants.LD_LV, size=disk["size"],
3801
                              logical_id=(vgname, names[idx]),
3802
                              iv_name="disk/%d" % disk_index)
3803
      disks.append(disk_dev)
3804
  elif template_name == constants.DT_DRBD8:
3805
    if len(secondary_nodes) != 1:
3806
      raise errors.ProgrammerError("Wrong template configuration")
3807
    remote_node = secondary_nodes[0]
3808
    minors = lu.cfg.AllocateDRBDMinor(
3809
      [primary_node, remote_node] * len(disk_info), instance_name)
3810

    
3811
    names = []
3812
    for lv_prefix in _GenerateUniqueNames(lu, [".disk%d" % i
3813
                                               for i in range(disk_count)]):
3814
      names.append(lv_prefix + "_data")
3815
      names.append(lv_prefix + "_meta")
3816
    for idx, disk in enumerate(disk_info):
3817
      disk_index = idx + base_index
3818
      disk_dev = _GenerateDRBD8Branch(lu, primary_node, remote_node,
3819
                                      disk["size"], names[idx*2:idx*2+2],
3820
                                      "disk/%d" % disk_index,
3821
                                      minors[idx*2], minors[idx*2+1])
3822
      disks.append(disk_dev)
3823
  elif template_name == constants.DT_FILE:
3824
    if len(secondary_nodes) != 0:
3825
      raise errors.ProgrammerError("Wrong template configuration")
3826

    
3827
    for idx, disk in enumerate(disk_info):
3828
      disk_index = idx + base_index
3829
      disk_dev = objects.Disk(dev_type=constants.LD_FILE, size=disk["size"],
3830
                              iv_name="disk/%d" % disk_index,
3831
                              logical_id=(file_driver,
3832
                                          "%s/disk%d" % (file_storage_dir,
3833
                                                         idx)))
3834
      disks.append(disk_dev)
3835
  else:
3836
    raise errors.ProgrammerError("Invalid disk template '%s'" % template_name)
3837
  return disks
3838

    
3839

    
3840
def _GetInstanceInfoText(instance):
3841
  """Compute that text that should be added to the disk's metadata.
3842

3843
  """
3844
  return "originstname+%s" % instance.name
3845

    
3846

    
3847
def _CreateDisks(lu, instance):
3848
  """Create all disks for an instance.
3849

3850
  This abstracts away some work from AddInstance.
3851

3852
  @type lu: L{LogicalUnit}
3853
  @param lu: the logical unit on whose behalf we execute
3854
  @type instance: L{objects.Instance}
3855
  @param instance: the instance whose disks we should create
3856
  @rtype: boolean
3857
  @return: the success of the creation
3858

3859
  """
3860
  info = _GetInstanceInfoText(instance)
3861
  pnode = instance.primary_node
3862

    
3863
  if instance.disk_template == constants.DT_FILE:
3864
    file_storage_dir = os.path.dirname(instance.disks[0].logical_id[1])
3865
    result = lu.rpc.call_file_storage_dir_create(pnode, file_storage_dir)
3866

    
3867
    if result.failed or not result.data:
3868
      raise errors.OpExecError("Could not connect to node '%s'" % pnode)
3869

    
3870
    if not result.data[0]:
3871
      raise errors.OpExecError("Failed to create directory '%s'" %
3872
                               file_storage_dir)
3873

    
3874
  # Note: this needs to be kept in sync with adding of disks in
3875
  # LUSetInstanceParams
3876
  for device in instance.disks:
3877
    logging.info("Creating volume %s for instance %s",
3878
                 device.iv_name, instance.name)
3879
    #HARDCODE
3880
    for node in instance.all_nodes:
3881
      f_create = node == pnode
3882
      _CreateBlockDev(lu, node, instance, device, f_create, info, f_create)
3883

    
3884

    
3885
def _RemoveDisks(lu, instance):
3886
  """Remove all disks for an instance.
3887

3888
  This abstracts away some work from `AddInstance()` and
3889
  `RemoveInstance()`. Note that in case some of the devices couldn't
3890
  be removed, the removal will continue with the other ones (compare
3891
  with `_CreateDisks()`).
3892

3893
  @type lu: L{LogicalUnit}
3894
  @param lu: the logical unit on whose behalf we execute
3895
  @type instance: L{objects.Instance}
3896
  @param instance: the instance whose disks we should remove
3897
  @rtype: boolean
3898
  @return: the success of the removal
3899

3900
  """
3901
  logging.info("Removing block devices for instance %s", instance.name)
3902

    
3903
  result = True
3904
  for device in instance.disks:
3905
    for node, disk in device.ComputeNodeTree(instance.primary_node):
3906
      lu.cfg.SetDiskID(disk, node)
3907
      result = lu.rpc.call_blockdev_remove(node, disk)
3908
      if result.failed or not result.data:
3909
        lu.proc.LogWarning("Could not remove block device %s on node %s,"
3910
                           " continuing anyway", device.iv_name, node)
3911
        result = False
3912

    
3913
  if instance.disk_template == constants.DT_FILE:
3914
    file_storage_dir = os.path.dirname(instance.disks[0].logical_id[1])
3915
    result = lu.rpc.call_file_storage_dir_remove(instance.primary_node,
3916
                                                 file_storage_dir)
3917
    if result.failed or not result.data:
3918
      logging.error("Could not remove directory '%s'", file_storage_dir)
3919
      result = False
3920

    
3921
  return result
3922

    
3923

    
3924
def _ComputeDiskSize(disk_template, disks):
3925
  """Compute disk size requirements in the volume group
3926

3927
  """
3928
  # Required free disk space as a function of disk and swap space
3929
  req_size_dict = {
3930
    constants.DT_DISKLESS: None,
3931
    constants.DT_PLAIN: sum(d["size"] for d in disks),
3932
    # 128 MB are added for drbd metadata for each disk
3933
    constants.DT_DRBD8: sum(d["size"] + 128 for d in disks),
3934
    constants.DT_FILE: None,
3935
  }
3936

    
3937
  if disk_template not in req_size_dict:
3938
    raise errors.ProgrammerError("Disk template '%s' size requirement"
3939
                                 " is unknown" %  disk_template)
3940

    
3941
  return req_size_dict[disk_template]
3942

    
3943

    
3944
def _CheckHVParams(lu, nodenames, hvname, hvparams):
3945
  """Hypervisor parameter validation.
3946

3947
  This function abstract the hypervisor parameter validation to be
3948
  used in both instance create and instance modify.
3949

3950
  @type lu: L{LogicalUnit}
3951
  @param lu: the logical unit for which we check
3952
  @type nodenames: list
3953
  @param nodenames: the list of nodes on which we should check
3954
  @type hvname: string
3955
  @param hvname: the name of the hypervisor we should use
3956
  @type hvparams: dict
3957
  @param hvparams: the parameters which we need to check
3958
  @raise errors.OpPrereqError: if the parameters are not valid
3959

3960
  """
3961
  hvinfo = lu.rpc.call_hypervisor_validate_params(nodenames,
3962
                                                  hvname,
3963
                                                  hvparams)
3964
  for node in nodenames:
3965
    info = hvinfo[node]
3966
    info.Raise()
3967
    if not info.data or not isinstance(info.data, (tuple, list)):
3968
      raise errors.OpPrereqError("Cannot get current information"
3969
                                 " from node '%s' (%s)" % (node, info.data))
3970
    if not info.data[0]:
3971
      raise errors.OpPrereqError("Hypervisor parameter validation failed:"
3972
                                 " %s" % info.data[1])
3973

    
3974

    
3975
class LUCreateInstance(LogicalUnit):
3976
  """Create an instance.
3977

3978
  """
3979
  HPATH = "instance-add"
3980
  HTYPE = constants.HTYPE_INSTANCE
3981
  _OP_REQP = ["instance_name", "disks", "disk_template",
3982
              "mode", "start",
3983
              "wait_for_sync", "ip_check", "nics",
3984
              "hvparams", "beparams"]
3985
  REQ_BGL = False
3986

    
3987
  def _ExpandNode(self, node):
3988
    """Expands and checks one node name.
3989

3990
    """
3991
    node_full = self.cfg.ExpandNodeName(node)
3992
    if node_full is None:
3993
      raise errors.OpPrereqError("Unknown node %s" % node)
3994
    return node_full
3995

    
3996
  def ExpandNames(self):
3997
    """ExpandNames for CreateInstance.
3998

3999
    Figure out the right locks for instance creation.
4000

4001
    """
4002
    self.needed_locks = {}
4003

    
4004
    # set optional parameters to none if they don't exist
4005
    for attr in ["pnode", "snode", "iallocator", "hypervisor"]:
4006
      if not hasattr(self.op, attr):
4007
        setattr(self.op, attr, None)
4008

    
4009
    # cheap checks, mostly valid constants given
4010

    
4011
    # verify creation mode
4012
    if self.op.mode not in (constants.INSTANCE_CREATE,
4013
                            constants.INSTANCE_IMPORT):
4014
      raise errors.OpPrereqError("Invalid instance creation mode '%s'" %
4015
                                 self.op.mode)
4016

    
4017
    # disk template and mirror node verification
4018
    if self.op.disk_template not in constants.DISK_TEMPLATES:
4019
      raise errors.OpPrereqError("Invalid disk template name")
4020

    
4021
    if self.op.hypervisor is None:
4022
      self.op.hypervisor = self.cfg.GetHypervisorType()
4023

    
4024
    cluster = self.cfg.GetClusterInfo()
4025
    enabled_hvs = cluster.enabled_hypervisors
4026
    if self.op.hypervisor not in enabled_hvs:
4027
      raise errors.OpPrereqError("Selected hypervisor (%s) not enabled in the"
4028
                                 " cluster (%s)" % (self.op.hypervisor,
4029
                                  ",".join(enabled_hvs)))
4030

    
4031
    # check hypervisor parameter syntax (locally)
4032

    
4033
    filled_hvp = cluster.FillDict(cluster.hvparams[self.op.hypervisor],
4034
                                  self.op.hvparams)
4035
    hv_type = hypervisor.GetHypervisor(self.op.hypervisor)
4036
    hv_type.CheckParameterSyntax(filled_hvp)
4037

    
4038
    # fill and remember the beparams dict
4039
    utils.CheckBEParams(self.op.beparams)
4040
    self.be_full = cluster.FillDict(cluster.beparams[constants.BEGR_DEFAULT],
4041
                                    self.op.beparams)
4042

    
4043
    #### instance parameters check
4044

    
4045
    # instance name verification
4046
    hostname1 = utils.HostInfo(self.op.instance_name)
4047
    self.op.instance_name = instance_name = hostname1.name
4048

    
4049
    # this is just a preventive check, but someone might still add this
4050
    # instance in the meantime, and creation will fail at lock-add time
4051
    if instance_name in self.cfg.GetInstanceList():
4052
      raise errors.OpPrereqError("Instance '%s' is already in the cluster" %
4053
                                 instance_name)
4054

    
4055
    self.add_locks[locking.LEVEL_INSTANCE] = instance_name
4056

    
4057
    # NIC buildup
4058
    self.nics = []
4059
    for nic in self.op.nics:
4060
      # ip validity checks
4061
      ip = nic.get("ip", None)
4062
      if ip is None or ip.lower() == "none":
4063
        nic_ip = None
4064
      elif ip.lower() == constants.VALUE_AUTO:
4065
        nic_ip = hostname1.ip
4066
      else:
4067
        if not utils.IsValidIP(ip):
4068
          raise errors.OpPrereqError("Given IP address '%s' doesn't look"
4069
                                     " like a valid IP" % ip)
4070
        nic_ip = ip
4071

    
4072
      # MAC address verification
4073
      mac = nic.get("mac", constants.VALUE_AUTO)
4074
      if mac not in (constants.VALUE_AUTO, constants.VALUE_GENERATE):
4075
        if not utils.IsValidMac(mac.lower()):
4076
          raise errors.OpPrereqError("Invalid MAC address specified: %s" %
4077
                                     mac)
4078
      # bridge verification
4079
      bridge = nic.get("bridge", self.cfg.GetDefBridge())
4080
      self.nics.append(objects.NIC(mac=mac, ip=nic_ip, bridge=bridge))
4081

    
4082
    # disk checks/pre-build
4083
    self.disks = []
4084
    for disk in self.op.disks:
4085
      mode = disk.get("mode", constants.DISK_RDWR)
4086
      if mode not in constants.DISK_ACCESS_SET:
4087
        raise errors.OpPrereqError("Invalid disk access mode '%s'" %
4088
                                   mode)
4089
      size = disk.get("size", None)
4090
      if size is None:
4091
        raise errors.OpPrereqError("Missing disk size")
4092
      try:
4093
        size = int(size)
4094
      except ValueError:
4095
        raise errors.OpPrereqError("Invalid disk size '%s'" % size)
4096
      self.disks.append({"size": size, "mode": mode})
4097

    
4098
    # used in CheckPrereq for ip ping check
4099
    self.check_ip = hostname1.ip
4100

    
4101
    # file storage checks
4102
    if (self.op.file_driver and
4103
        not self.op.file_driver in constants.FILE_DRIVER):
4104
      raise errors.OpPrereqError("Invalid file driver name '%s'" %
4105
                                 self.op.file_driver)
4106

    
4107
    if self.op.file_storage_dir and os.path.isabs(self.op.file_storage_dir):
4108
      raise errors.OpPrereqError("File storage directory path not absolute")
4109

    
4110
    ### Node/iallocator related checks
4111
    if [self.op.iallocator, self.op.pnode].count(None) != 1:
4112
      raise errors.OpPrereqError("One and only one of iallocator and primary"
4113
                                 " node must be given")
4114

    
4115
    if self.op.iallocator:
4116
      self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
4117
    else:
4118
      self.op.pnode = self._ExpandNode(self.op.pnode)
4119
      nodelist = [self.op.pnode]
4120
      if self.op.snode is not None:
4121
        self.op.snode = self._ExpandNode(self.op.snode)
4122
        nodelist.append(self.op.snode)
4123
      self.needed_locks[locking.LEVEL_NODE] = nodelist
4124

    
4125
    # in case of import lock the source node too
4126
    if self.op.mode == constants.INSTANCE_IMPORT:
4127
      src_node = getattr(self.op, "src_node", None)
4128
      src_path = getattr(self.op, "src_path", None)
4129

    
4130
      if src_path is None:
4131
        self.op.src_path = src_path = self.op.instance_name
4132

    
4133
      if src_node is None:
4134
        self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
4135
        self.op.src_node = None
4136
        if os.path.isabs(src_path):
4137
          raise errors.OpPrereqError("Importing an instance from an absolute"
4138
                                     " path requires a source node option.")
4139
      else:
4140
        self.op.src_node = src_node = self._ExpandNode(src_node)
4141
        if self.needed_locks[locking.LEVEL_NODE] is not locking.ALL_SET:
4142
          self.needed_locks[locking.LEVEL_NODE].append(src_node)
4143
        if not os.path.isabs(src_path):
4144
          self.op.src_path = src_path = \
4145
            os.path.join(constants.EXPORT_DIR, src_path)
4146

    
4147
    else: # INSTANCE_CREATE
4148
      if getattr(self.op, "os_type", None) is None:
4149
        raise errors.OpPrereqError("No guest OS specified")
4150

    
4151
  def _RunAllocator(self):
4152
    """Run the allocator based on input opcode.
4153

4154
    """
4155
    nics = [n.ToDict() for n in self.nics]
4156
    ial = IAllocator(self,
4157
                     mode=constants.IALLOCATOR_MODE_ALLOC,
4158
                     name=self.op.instance_name,
4159
                     disk_template=self.op.disk_template,
4160
                     tags=[],
4161
                     os=self.op.os_type,
4162
                     vcpus=self.be_full[constants.BE_VCPUS],
4163
                     mem_size=self.be_full[constants.BE_MEMORY],
4164
                     disks=self.disks,
4165
                     nics=nics,
4166
                     hypervisor=self.op.hypervisor,
4167
                     )
4168

    
4169
    ial.Run(self.op.iallocator)
4170

    
4171
    if not ial.success:
4172
      raise errors.OpPrereqError("Can't compute nodes using"
4173
                                 " iallocator '%s': %s" % (self.op.iallocator,
4174
                                                           ial.info))
4175
    if len(ial.nodes) != ial.required_nodes:
4176
      raise errors.OpPrereqError("iallocator '%s' returned invalid number"
4177
                                 " of nodes (%s), required %s" %
4178
                                 (self.op.iallocator, len(ial.nodes),
4179
                                  ial.required_nodes))
4180
    self.op.pnode = ial.nodes[0]
4181
    self.LogInfo("Selected nodes for instance %s via iallocator %s: %s",
4182
                 self.op.instance_name, self.op.iallocator,
4183
                 ", ".join(ial.nodes))
4184
    if ial.required_nodes == 2:
4185
      self.op.snode = ial.nodes[1]
4186

    
4187
  def BuildHooksEnv(self):
4188
    """Build hooks env.
4189

4190
    This runs on master, primary and secondary nodes of the instance.
4191

4192
    """
4193
    env = {
4194
      "INSTANCE_DISK_TEMPLATE": self.op.disk_template,
4195
      "INSTANCE_DISK_SIZE": ",".join(str(d["size"]) for d in self.disks),
4196
      "INSTANCE_ADD_MODE": self.op.mode,
4197
      }
4198
    if self.op.mode == constants.INSTANCE_IMPORT:
4199
      env["INSTANCE_SRC_NODE"] = self.op.src_node
4200
      env["INSTANCE_SRC_PATH"] = self.op.src_path
4201
      env["INSTANCE_SRC_IMAGES"] = self.src_images
4202

    
4203
    env.update(_BuildInstanceHookEnv(name=self.op.instance_name,
4204
      primary_node=self.op.pnode,
4205
      secondary_nodes=self.secondaries,
4206
      status=self.instance_status,
4207
      os_type=self.op.os_type,
4208
      memory=self.be_full[constants.BE_MEMORY],
4209
      vcpus=self.be_full[constants.BE_VCPUS],
4210
      nics=[(n.ip, n.bridge, n.mac) for n in self.nics],
4211
    ))
4212

    
4213
    nl = ([self.cfg.GetMasterNode(), self.op.pnode] +
4214
          self.secondaries)
4215
    return env, nl, nl
4216

    
4217

    
4218
  def CheckPrereq(self):
4219
    """Check prerequisites.
4220

4221
    """
4222
    if (not self.cfg.GetVGName() and
4223
        self.op.disk_template not in constants.DTS_NOT_LVM):
4224
      raise errors.OpPrereqError("Cluster does not support lvm-based"
4225
                                 " instances")
4226

    
4227

    
4228
    if self.op.mode == constants.INSTANCE_IMPORT:
4229
      src_node = self.op.src_node
4230
      src_path = self.op.src_path
4231

    
4232
      if src_node is None:
4233
        exp_list = self.rpc.call_export_list(
4234
          self.acquired_locks[locking.LEVEL_NODE])
4235
        found = False
4236
        for node in exp_list:
4237
          if not exp_list[node].failed and src_path in exp_list[node].data:
4238
            found = True
4239
            self.op.src_node = src_node = node
4240
            self.op.src_path = src_path = os.path.join(constants.EXPORT_DIR,
4241
                                                       src_path)
4242
            break
4243
        if not found:
4244
          raise errors.OpPrereqError("No export found for relative path %s" %
4245
                                      src_path)
4246

    
4247
      _CheckNodeOnline(self, src_node)
4248
      result = self.rpc.call_export_info(src_node, src_path)
4249
      result.Raise()
4250
      if not result.data:
4251
        raise errors.OpPrereqError("No export found in dir %s" % src_path)
4252

    
4253
      export_info = result.data
4254
      if not export_info.has_section(constants.INISECT_EXP):
4255
        raise errors.ProgrammerError("Corrupted export config")
4256

    
4257
      ei_version = export_info.get(constants.INISECT_EXP, 'version')
4258
      if (int(ei_version) != constants.EXPORT_VERSION):
4259
        raise errors.OpPrereqError("Wrong export version %s (wanted %d)" %
4260
                                   (ei_version, constants.EXPORT_VERSION))
4261

    
4262
      # Check that the new instance doesn't have less disks than the export
4263
      instance_disks = len(self.disks)
4264
      export_disks = export_info.getint(constants.INISECT_INS, 'disk_count')
4265
      if instance_disks < export_disks:
4266
        raise errors.OpPrereqError("Not enough disks to import."
4267
                                   " (instance: %d, export: %d)" %
4268
                                   (instance_disks, export_disks))
4269

    
4270
      self.op.os_type = export_info.get(constants.INISECT_EXP, 'os')
4271
      disk_images = []
4272
      for idx in range(export_disks):
4273
        option = 'disk%d_dump' % idx
4274
        if export_info.has_option(constants.INISECT_INS, option):
4275
          # FIXME: are the old os-es, disk sizes, etc. useful?
4276
          export_name = export_info.get(constants.INISECT_INS, option)
4277
          image = os.path.join(src_path, export_name)
4278
          disk_images.append(image)
4279
        else:
4280
          disk_images.append(False)
4281

    
4282
      self.src_images = disk_images
4283

    
4284
      old_name = export_info.get(constants.INISECT_INS, 'name')
4285
      # FIXME: int() here could throw a ValueError on broken exports
4286
      exp_nic_count = int(export_info.get(constants.INISECT_INS, 'nic_count'))
4287
      if self.op.instance_name == old_name:
4288
        for idx, nic in enumerate(self.nics):
4289
          if nic.mac == constants.VALUE_AUTO and exp_nic_count >= idx:
4290
            nic_mac_ini = 'nic%d_mac' % idx
4291
            nic.mac = export_info.get(constants.INISECT_INS, nic_mac_ini)
4292

    
4293
    # ip ping checks (we use the same ip that was resolved in ExpandNames)
4294
    if self.op.start and not self.op.ip_check:
4295
      raise errors.OpPrereqError("Cannot ignore IP address conflicts when"
4296
                                 " adding an instance in start mode")
4297

    
4298
    if self.op.ip_check:
4299
      if utils.TcpPing(self.check_ip, constants.DEFAULT_NODED_PORT):
4300
        raise errors.OpPrereqError("IP %s of instance %s already in use" %
4301
                                   (self.check_ip, self.op.instance_name))
4302

    
4303
    #### allocator run
4304

    
4305
    if self.op.iallocator is not None:
4306
      self._RunAllocator()
4307

    
4308
    #### node related checks
4309

    
4310
    # check primary node
4311
    self.pnode = pnode = self.cfg.GetNodeInfo(self.op.pnode)
4312
    assert self.pnode is not None, \
4313
      "Cannot retrieve locked node %s" % self.op.pnode
4314
    if pnode.offline:
4315
      raise errors.OpPrereqError("Cannot use offline primary node '%s'" %
4316
                                 pnode.name)
4317

    
4318
    self.secondaries = []
4319

    
4320
    # mirror node verification
4321
    if self.op.disk_template in constants.DTS_NET_MIRROR:
4322
      if self.op.snode is None:
4323
        raise errors.OpPrereqError("The networked disk templates need"
4324
                                   " a mirror node")
4325
      if self.op.snode == pnode.name:
4326
        raise errors.OpPrereqError("The secondary node cannot be"
4327
                                   " the primary node.")
4328
      self.secondaries.append(self.op.snode)
4329
      _CheckNodeOnline(self, self.op.snode)
4330

    
4331
    nodenames = [pnode.name] + self.secondaries
4332

    
4333
    req_size = _ComputeDiskSize(self.op.disk_template,
4334
                                self.disks)
4335

    
4336
    # Check lv size requirements
4337
    if req_size is not None:
4338
      nodeinfo = self.rpc.call_node_info(nodenames, self.cfg.GetVGName(),
4339
                                         self.op.hypervisor)
4340
      for node in nodenames:
4341
        info = nodeinfo[node]
4342
        info.Raise()
4343
        info = info.data
4344
        if not info:
4345
          raise errors.OpPrereqError("Cannot get current information"
4346
                                     " from node '%s'" % node)
4347
        vg_free = info.get('vg_free', None)
4348
        if not isinstance(vg_free, int):
4349
          raise errors.OpPrereqError("Can't compute free disk space on"
4350
                                     " node %s" % node)
4351
        if req_size > info['vg_free']:
4352
          raise errors.OpPrereqError("Not enough disk space on target node %s."
4353
                                     " %d MB available, %d MB required" %
4354
                                     (node, info['vg_free'], req_size))
4355

    
4356
    _CheckHVParams(self, nodenames, self.op.hypervisor, self.op.hvparams)
4357

    
4358
    # os verification
4359
    result = self.rpc.call_os_get(pnode.name, self.op.os_type)
4360
    result.Raise()
4361
    if not isinstance(result.data, objects.OS):
4362
      raise errors.OpPrereqError("OS '%s' not in supported os list for"
4363
                                 " primary node"  % self.op.os_type)
4364

    
4365
    # bridge check on primary node
4366
    bridges = [n.bridge for n in self.nics]
4367
    result = self.rpc.call_bridges_exist(self.pnode.name, bridges)
4368
    result.Raise()
4369
    if not result.data:
4370
      raise errors.OpPrereqError("One of the target bridges '%s' does not"
4371
                                 " exist on destination node '%s'" %
4372
                                 (",".join(bridges), pnode.name))
4373

    
4374
    # memory check on primary node
4375
    if self.op.start:
4376
      _CheckNodeFreeMemory(self, self.pnode.name,
4377
                           "creating instance %s" % self.op.instance_name,
4378
                           self.be_full[constants.BE_MEMORY],
4379
                           self.op.hypervisor)
4380

    
4381
    if self.op.start:
4382
      self.instance_status = 'up'
4383
    else:
4384
      self.instance_status = 'down'
4385

    
4386
  def Exec(self, feedback_fn):
4387
    """Create and add the instance to the cluster.
4388

4389
    """
4390
    instance = self.op.instance_name
4391
    pnode_name = self.pnode.name
4392

    
4393
    for nic in self.nics:
4394
      if nic.mac in (constants.VALUE_AUTO, constants.VALUE_GENERATE):
4395
        nic.mac = self.cfg.GenerateMAC()
4396

    
4397
    ht_kind = self.op.hypervisor
4398
    if ht_kind in constants.HTS_REQ_PORT:
4399
      network_port = self.cfg.AllocatePort()
4400
    else:
4401
      network_port = None
4402

    
4403
    ##if self.op.vnc_bind_address is None:
4404
    ##  self.op.vnc_bind_address = constants.VNC_DEFAULT_BIND_ADDRESS
4405

    
4406
    # this is needed because os.path.join does not accept None arguments
4407
    if self.op.file_storage_dir is None:
4408
      string_file_storage_dir = ""
4409
    else:
4410
      string_file_storage_dir = self.op.file_storage_dir
4411

    
4412
    # build the full file storage dir path
4413
    file_storage_dir = os.path.normpath(os.path.join(
4414
                                        self.cfg.GetFileStorageDir(),
4415
                                        string_file_storage_dir, instance))
4416

    
4417

    
4418
    disks = _GenerateDiskTemplate(self,
4419
                                  self.op.disk_template,
4420
                                  instance, pnode_name,
4421
                                  self.secondaries,
4422
                                  self.disks,
4423
                                  file_storage_dir,
4424
                                  self.op.file_driver,
4425
                                  0)
4426

    
4427
    iobj = objects.Instance(name=instance, os=self.op.os_type,
4428
                            primary_node=pnode_name,
4429
                            nics=self.nics, disks=disks,
4430
                            disk_template=self.op.disk_template,
4431
                            status=self.instance_status,
4432
                            network_port=network_port,
4433
                            beparams=self.op.beparams,
4434
                            hvparams=self.op.hvparams,
4435
                            hypervisor=self.op.hypervisor,
4436
                            )
4437

    
4438
    feedback_fn("* creating instance disks...")
4439
    try:
4440
      _CreateDisks(self, iobj)
4441
    except errors.OpExecError:
4442
      self.LogWarning("Device creation failed, reverting...")
4443
      try:
4444
        _RemoveDisks(self, iobj)
4445
      finally:
4446
        self.cfg.ReleaseDRBDMinors(instance)
4447
        raise
4448

    
4449
    feedback_fn("adding instance %s to cluster config" % instance)
4450

    
4451
    self.cfg.AddInstance(iobj)
4452
    # Declare that we don't want to remove the instance lock anymore, as we've
4453
    # added the instance to the config
4454
    del self.remove_locks[locking.LEVEL_INSTANCE]
4455
    # Remove the temp. assignements for the instance's drbds
4456
    self.cfg.ReleaseDRBDMinors(instance)
4457
    # Unlock all the nodes
4458
    if self.op.mode == constants.INSTANCE_IMPORT:
4459
      nodes_keep = [self.op.src_node]
4460
      nodes_release = [node for node in self.acquired_locks[locking.LEVEL_NODE]
4461
                       if node != self.op.src_node]
4462
      self.context.glm.release(locking.LEVEL_NODE, nodes_release)
4463
      self.acquired_locks[locking.LEVEL_NODE] = nodes_keep
4464
    else:
4465
      self.context.glm.release(locking.LEVEL_NODE)
4466
      del self.acquired_locks[locking.LEVEL_NODE]
4467

    
4468
    if self.op.wait_for_sync:
4469
      disk_abort = not _WaitForSync(self, iobj)
4470
    elif iobj.disk_template in constants.DTS_NET_MIRROR:
4471
      # make sure the disks are not degraded (still sync-ing is ok)
4472
      time.sleep(15)
4473
      feedback_fn("* checking mirrors status")
4474
      disk_abort = not _WaitForSync(self, iobj, oneshot=True)
4475
    else:
4476
      disk_abort = False
4477

    
4478
    if disk_abort:
4479
      _RemoveDisks(self, iobj)
4480
      self.cfg.RemoveInstance(iobj.name)
4481
      # Make sure the instance lock gets removed
4482
      self.remove_locks[locking.LEVEL_INSTANCE] = iobj.name
4483
      raise errors.OpExecError("There are some degraded disks for"
4484
                               " this instance")
4485

    
4486
    feedback_fn("creating os for instance %s on node %s" %
4487
                (instance, pnode_name))
4488

    
4489
    if iobj.disk_template != constants.DT_DISKLESS:
4490
      if self.op.mode == constants.INSTANCE_CREATE:
4491
        feedback_fn("* running the instance OS create scripts...")
4492
        result = self.rpc.call_instance_os_add(pnode_name, iobj)
4493
        result.Raise()
4494
        if not result.data:
4495
          raise errors.OpExecError("Could not add os for instance %s"
4496
                                   " on node %s" %
4497
                                   (instance, pnode_name))
4498

    
4499
      elif self.op.mode == constants.INSTANCE_IMPORT:
4500
        feedback_fn("* running the instance OS import scripts...")
4501
        src_node = self.op.src_node
4502
        src_images = self.src_images
4503
        cluster_name = self.cfg.GetClusterName()
4504
        import_result = self.rpc.call_instance_os_import(pnode_name, iobj,
4505
                                                         src_node, src_images,
4506
                                                         cluster_name)
4507
        import_result.Raise()
4508
        for idx, result in enumerate(import_result.data):
4509
          if not result:
4510
            self.LogWarning("Could not import the image %s for instance"
4511
                            " %s, disk %d, on node %s" %
4512
                            (src_images[idx], instance, idx, pnode_name))
4513
      else:
4514
        # also checked in the prereq part
4515
        raise errors.ProgrammerError("Unknown OS initialization mode '%s'"
4516
                                     % self.op.mode)
4517

    
4518
    if self.op.start:
4519
      logging.info("Starting instance %s on node %s", instance, pnode_name)
4520
      feedback_fn("* starting instance...")
4521
      result = self.rpc.call_instance_start(pnode_name, iobj, None)
4522
      result.Raise()
4523
      if not result.data:
4524
        raise errors.OpExecError("Could not start instance")
4525

    
4526

    
4527
class LUConnectConsole(NoHooksLU):
4528
  """Connect to an instance's console.
4529

4530
  This is somewhat special in that it returns the command line that
4531
  you need to run on the master node in order to connect to the
4532
  console.
4533

4534
  """
4535
  _OP_REQP = ["instance_name"]
4536
  REQ_BGL = False
4537

    
4538
  def ExpandNames(self):
4539
    self._ExpandAndLockInstance()
4540

    
4541
  def CheckPrereq(self):
4542
    """Check prerequisites.
4543

4544
    This checks that the instance is in the cluster.
4545

4546
    """
4547
    self.instance = self.cfg.GetInstanceInfo(self.op.instance_name)
4548
    assert self.instance is not None, \
4549
      "Cannot retrieve locked instance %s" % self.op.instance_name
4550
    _CheckNodeOnline(self, self.instance.primary_node)
4551

    
4552
  def Exec(self, feedback_fn):
4553
    """Connect to the console of an instance
4554

4555
    """
4556
    instance = self.instance
4557
    node = instance.primary_node
4558

    
4559
    node_insts = self.rpc.call_instance_list([node],
4560
                                             [instance.hypervisor])[node]
4561
    node_insts.Raise()
4562

    
4563
    if instance.name not in node_insts.data:
4564
      raise errors.OpExecError("Instance %s is not running." % instance.name)
4565

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

    
4568
    hyper = hypervisor.GetHypervisor(instance.hypervisor)
4569
    console_cmd = hyper.GetShellCommandForConsole(instance)
4570

    
4571
    # build ssh cmdline
4572
    return self.ssh.BuildCmd(node, "root", console_cmd, batch=True, tty=True)
4573

    
4574

    
4575
class LUReplaceDisks(LogicalUnit):
4576
  """Replace the disks of an instance.
4577

4578
  """
4579
  HPATH = "mirrors-replace"
4580
  HTYPE = constants.HTYPE_INSTANCE
4581
  _OP_REQP = ["instance_name", "mode", "disks"]
4582
  REQ_BGL = False
4583

    
4584
  def CheckArguments(self):
4585
    if not hasattr(self.op, "remote_node"):
4586
      self.op.remote_node = None
4587
    if not hasattr(self.op, "iallocator"):
4588
      self.op.iallocator = None
4589

    
4590
    # check for valid parameter combination
4591
    cnt = [self.op.remote_node, self.op.iallocator].count(None)
4592
    if self.op.mode == constants.REPLACE_DISK_CHG:
4593
      if cnt == 2:
4594
        raise errors.OpPrereqError("When changing the secondary either an"
4595
                                   " iallocator script must be used or the"
4596
                                   " new node given")
4597
      elif cnt == 0:
4598
        raise errors.OpPrereqError("Give either the iallocator or the new"
4599
                                   " secondary, not both")
4600
    else: # not replacing the secondary
4601
      if cnt != 2:
4602
        raise errors.OpPrereqError("The iallocator and new node options can"
4603
                                   " be used only when changing the"
4604
                                   " secondary node")
4605

    
4606
  def ExpandNames(self):
4607
    self._ExpandAndLockInstance()
4608

    
4609
    if self.op.iallocator is not None:
4610
      self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
4611
    elif self.op.remote_node is not None:
4612
      remote_node = self.cfg.ExpandNodeName(self.op.remote_node)
4613
      if remote_node is None:
4614
        raise errors.OpPrereqError("Node '%s' not known" %
4615
                                   self.op.remote_node)
4616
      self.op.remote_node = remote_node
4617
      self.needed_locks[locking.LEVEL_NODE] = [remote_node]
4618
      self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_APPEND
4619
    else:
4620
      self.needed_locks[locking.LEVEL_NODE] = []
4621
      self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
4622

    
4623
  def DeclareLocks(self, level):
4624
    # If we're not already locking all nodes in the set we have to declare the
4625
    # instance's primary/secondary nodes.
4626
    if (level == locking.LEVEL_NODE and
4627
        self.needed_locks[locking.LEVEL_NODE] is not locking.ALL_SET):
4628
      self._LockInstancesNodes()
4629

    
4630
  def _RunAllocator(self):
4631
    """Compute a new secondary node using an IAllocator.
4632

4633
    """
4634
    ial = IAllocator(self,
4635
                     mode=constants.IALLOCATOR_MODE_RELOC,
4636
                     name=self.op.instance_name,
4637
                     relocate_from=[self.sec_node])
4638

    
4639
    ial.Run(self.op.iallocator)
4640

    
4641
    if not ial.success:
4642
      raise errors.OpPrereqError("Can't compute nodes using"
4643
                                 " iallocator '%s': %s" % (self.op.iallocator,
4644
                                                           ial.info))
4645
    if len(ial.nodes) != ial.required_nodes:
4646
      raise errors.OpPrereqError("iallocator '%s' returned invalid number"
4647
                                 " of nodes (%s), required %s" %
4648
                                 (len(ial.nodes), ial.required_nodes))
4649
    self.op.remote_node = ial.nodes[0]
4650
    self.LogInfo("Selected new secondary for the instance: %s",
4651
                 self.op.remote_node)
4652

    
4653
  def BuildHooksEnv(self):
4654
    """Build hooks env.
4655

4656
    This runs on the master, the primary and all the secondaries.
4657

4658
    """
4659
    env = {
4660
      "MODE": self.op.mode,
4661
      "NEW_SECONDARY": self.op.remote_node,
4662
      "OLD_SECONDARY": self.instance.secondary_nodes[0],
4663
      }
4664
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
4665
    nl = [
4666
      self.cfg.GetMasterNode(),
4667
      self.instance.primary_node,
4668
      ]
4669
    if self.op.remote_node is not None:
4670
      nl.append(self.op.remote_node)
4671
    return env, nl, nl
4672

    
4673
  def CheckPrereq(self):
4674
    """Check prerequisites.
4675

4676
    This checks that the instance is in the cluster.
4677

4678
    """
4679
    instance = self.cfg.GetInstanceInfo(self.op.instance_name)
4680
    assert instance is not None, \
4681
      "Cannot retrieve locked instance %s" % self.op.instance_name
4682
    self.instance = instance
4683

    
4684
    if instance.disk_template != constants.DT_DRBD8:
4685
      raise errors.OpPrereqError("Can only run replace disks for DRBD8-based"
4686
                                 " instances")
4687

    
4688
    if len(instance.secondary_nodes) != 1:
4689
      raise errors.OpPrereqError("The instance has a strange layout,"
4690
                                 " expected one secondary but found %d" %
4691
                                 len(instance.secondary_nodes))
4692

    
4693
    self.sec_node = instance.secondary_nodes[0]
4694

    
4695
    if self.op.iallocator is not None:
4696
      self._RunAllocator()
4697

    
4698
    remote_node = self.op.remote_node
4699
    if remote_node is not None:
4700
      self.remote_node_info = self.cfg.GetNodeInfo(remote_node)
4701
      assert self.remote_node_info is not None, \
4702
        "Cannot retrieve locked node %s" % remote_node
4703
    else:
4704
      self.remote_node_info = None
4705
    if remote_node == instance.primary_node:
4706
      raise errors.OpPrereqError("The specified node is the primary node of"
4707
                                 " the instance.")
4708
    elif remote_node == self.sec_node:
4709
      raise errors.OpPrereqError("The specified node is already the"
4710
                                 " secondary node of the instance.")
4711

    
4712
    if self.op.mode == constants.REPLACE_DISK_PRI:
4713
      n1 = self.tgt_node = instance.primary_node
4714
      n2 = self.oth_node = self.sec_node
4715
    elif self.op.mode == constants.REPLACE_DISK_SEC:
4716
      n1 = self.tgt_node = self.sec_node
4717
      n2 = self.oth_node = instance.primary_node
4718
    elif self.op.mode == constants.REPLACE_DISK_CHG:
4719
      n1 = self.new_node = remote_node
4720
      n2 = self.oth_node = instance.primary_node
4721
      self.tgt_node = self.sec_node
4722
    else:
4723
      raise errors.ProgrammerError("Unhandled disk replace mode")
4724

    
4725
    _CheckNodeOnline(self, n1)
4726
    _CheckNodeOnline(self, n2)
4727

    
4728
    if not self.op.disks:
4729
      self.op.disks = range(len(instance.disks))
4730

    
4731
    for disk_idx in self.op.disks:
4732
      instance.FindDisk(disk_idx)
4733

    
4734
  def _ExecD8DiskOnly(self, feedback_fn):
4735
    """Replace a disk on the primary or secondary for dbrd8.
4736

4737
    The algorithm for replace is quite complicated:
4738

4739
      1. for each disk to be replaced:
4740

4741
        1. create new LVs on the target node with unique names
4742
        1. detach old LVs from the drbd device
4743
        1. rename old LVs to name_replaced.<time_t>
4744
        1. rename new LVs to old LVs
4745
        1. attach the new LVs (with the old names now) to the drbd device
4746

4747
      1. wait for sync across all devices
4748

4749
      1. for each modified disk:
4750

4751
        1. remove old LVs (which have the name name_replaces.<time_t>)
4752

4753
    Failures are not very well handled.
4754

4755
    """
4756
    steps_total = 6
4757
    warning, info = (self.proc.LogWarning, self.proc.LogInfo)
4758
    instance = self.instance
4759
    iv_names = {}
4760
    vgname = self.cfg.GetVGName()
4761
    # start of work
4762
    cfg = self.cfg
4763
    tgt_node = self.tgt_node
4764
    oth_node = self.oth_node
4765

    
4766
    # Step: check device activation
4767
    self.proc.LogStep(1, steps_total, "check device existence")
4768
    info("checking volume groups")
4769
    my_vg = cfg.GetVGName()
4770
    results = self.rpc.call_vg_list([oth_node, tgt_node])
4771
    if not results:
4772
      raise errors.OpExecError("Can't list volume groups on the nodes")
4773
    for node in oth_node, tgt_node:
4774
      res = results[node]
4775
      if res.failed or not res.data or my_vg not in res.data:
4776
        raise errors.OpExecError("Volume group '%s' not found on %s" %
4777
                                 (my_vg, node))
4778
    for idx, dev in enumerate(instance.disks):
4779
      if idx not in self.op.disks:
4780
        continue
4781
      for node in tgt_node, oth_node:
4782
        info("checking disk/%d on %s" % (idx, node))
4783
        cfg.SetDiskID(dev, node)
4784
        if not self.rpc.call_blockdev_find(node, dev):
4785
          raise errors.OpExecError("Can't find disk/%d on node %s" %
4786
                                   (idx, node))
4787

    
4788
    # Step: check other node consistency
4789
    self.proc.LogStep(2, steps_total, "check peer consistency")
4790
    for idx, dev in enumerate(instance.disks):
4791
      if idx not in self.op.disks:
4792
        continue
4793
      info("checking disk/%d consistency on %s" % (idx, oth_node))
4794
      if not _CheckDiskConsistency(self, dev, oth_node,
4795
                                   oth_node==instance.primary_node):
4796
        raise errors.OpExecError("Peer node (%s) has degraded storage, unsafe"
4797
                                 " to replace disks on this node (%s)" %
4798
                                 (oth_node, tgt_node))
4799

    
4800
    # Step: create new storage
4801
    self.proc.LogStep(3, steps_total, "allocate new storage")
4802
    for idx, dev in enumerate(instance.disks):
4803
      if idx not in self.op.disks:
4804
        continue
4805
      size = dev.size
4806
      cfg.SetDiskID(dev, tgt_node)
4807
      lv_names = [".disk%d_%s" % (idx, suf)
4808
                  for suf in ["data", "meta"]]
4809
      names = _GenerateUniqueNames(self, lv_names)
4810
      lv_data = objects.Disk(dev_type=constants.LD_LV, size=size,
4811
                             logical_id=(vgname, names[0]))
4812
      lv_meta = objects.Disk(dev_type=constants.LD_LV, size=128,
4813
                             logical_id=(vgname, names[1]))
4814
      new_lvs = [lv_data, lv_meta]
4815
      old_lvs = dev.children
4816
      iv_names[dev.iv_name] = (dev, old_lvs, new_lvs)
4817
      info("creating new local storage on %s for %s" %
4818
           (tgt_node, dev.iv_name))
4819
      # we pass force_create=True to force the LVM creation
4820
      for new_lv in new_lvs:
4821
        _CreateBlockDev(self, tgt_node, instance, new_lv, True,
4822
                        _GetInstanceInfoText(instance), False)
4823

    
4824
    # Step: for each lv, detach+rename*2+attach
4825
    self.proc.LogStep(4, steps_total, "change drbd configuration")
4826
    for dev, old_lvs, new_lvs in iv_names.itervalues():
4827
      info("detaching %s drbd from local storage" % dev.iv_name)
4828
      result = self.rpc.call_blockdev_removechildren(tgt_node, dev, old_lvs)
4829
      result.Raise()
4830
      if not result.data:
4831
        raise errors.OpExecError("Can't detach drbd from local storage on node"
4832
                                 " %s for device %s" % (tgt_node, dev.iv_name))
4833
      #dev.children = []
4834
      #cfg.Update(instance)
4835

    
4836
      # ok, we created the new LVs, so now we know we have the needed
4837
      # storage; as such, we proceed on the target node to rename
4838
      # old_lv to _old, and new_lv to old_lv; note that we rename LVs
4839
      # using the assumption that logical_id == physical_id (which in
4840
      # turn is the unique_id on that node)
4841

    
4842
      # FIXME(iustin): use a better name for the replaced LVs
4843
      temp_suffix = int(time.time())
4844
      ren_fn = lambda d, suff: (d.physical_id[0],
4845
                                d.physical_id[1] + "_replaced-%s" % suff)
4846
      # build the rename list based on what LVs exist on the node
4847
      rlist = []
4848
      for to_ren in old_lvs:
4849
        find_res = self.rpc.call_blockdev_find(tgt_node, to_ren)
4850
        if not find_res.failed and find_res.data is not None: # device exists
4851
          rlist.append((to_ren, ren_fn(to_ren, temp_suffix)))
4852

    
4853
      info("renaming the old LVs on the target node")
4854
      result = self.rpc.call_blockdev_rename(tgt_node, rlist)
4855
      result.Raise()
4856
      if not result.data:
4857
        raise errors.OpExecError("Can't rename old LVs on node %s" % tgt_node)
4858
      # now we rename the new LVs to the old LVs
4859
      info("renaming the new LVs on the target node")
4860
      rlist = [(new, old.physical_id) for old, new in zip(old_lvs, new_lvs)]
4861
      result = self.rpc.call_blockdev_rename(tgt_node, rlist)
4862
      result.Raise()
4863
      if not result.data:
4864
        raise errors.OpExecError("Can't rename new LVs on node %s" % tgt_node)
4865

    
4866
      for old, new in zip(old_lvs, new_lvs):
4867
        new.logical_id = old.logical_id
4868
        cfg.SetDiskID(new, tgt_node)
4869

    
4870
      for disk in old_lvs:
4871
        disk.logical_id = ren_fn(disk, temp_suffix)
4872
        cfg.SetDiskID(disk, tgt_node)
4873

    
4874
      # now that the new lvs have the old name, we can add them to the device
4875
      info("adding new mirror component on %s" % tgt_node)
4876
      result = self.rpc.call_blockdev_addchildren(tgt_node, dev, new_lvs)
4877
      if result.failed or not result.data:
4878
        for new_lv in new_lvs:
4879
          result = self.rpc.call_blockdev_remove(tgt_node, new_lv)
4880
          if result.failed or not result.data:
4881
            warning("Can't rollback device %s", hint="manually cleanup unused"
4882
                    " logical volumes")
4883
        raise errors.OpExecError("Can't add local storage to drbd")
4884

    
4885
      dev.children = new_lvs
4886
      cfg.Update(instance)
4887

    
4888
    # Step: wait for sync
4889

    
4890
    # this can fail as the old devices are degraded and _WaitForSync
4891
    # does a combined result over all disks, so we don't check its
4892
    # return value
4893
    self.proc.LogStep(5, steps_total, "sync devices")
4894
    _WaitForSync(self, instance, unlock=True)
4895

    
4896
    # so check manually all the devices
4897
    for name, (dev, old_lvs, new_lvs) in iv_names.iteritems():
4898
      cfg.SetDiskID(dev, instance.primary_node)
4899
      result = self.rpc.call_blockdev_find(instance.primary_node, dev)
4900
      if result.failed or result.data[5]:
4901
        raise errors.OpExecError("DRBD device %s is degraded!" % name)
4902

    
4903
    # Step: remove old storage
4904
    self.proc.LogStep(6, steps_total, "removing old storage")
4905
    for name, (dev, old_lvs, new_lvs) in iv_names.iteritems():
4906
      info("remove logical volumes for %s" % name)
4907
      for lv in old_lvs:
4908
        cfg.SetDiskID(lv, tgt_node)
4909
        result = self.rpc.call_blockdev_remove(tgt_node, lv)
4910
        if result.failed or not result.data:
4911
          warning("Can't remove old LV", hint="manually remove unused LVs")
4912
          continue
4913

    
4914
  def _ExecD8Secondary(self, feedback_fn):
4915
    """Replace the secondary node for drbd8.
4916

4917
    The algorithm for replace is quite complicated:
4918
      - for all disks of the instance:
4919
        - create new LVs on the new node with same names
4920
        - shutdown the drbd device on the old secondary
4921
        - disconnect the drbd network on the primary
4922
        - create the drbd device on the new secondary
4923
        - network attach the drbd on the primary, using an artifice:
4924
          the drbd code for Attach() will connect to the network if it
4925
          finds a device which is connected to the good local disks but
4926
          not network enabled
4927
      - wait for sync across all devices
4928
      - remove all disks from the old secondary
4929

4930
    Failures are not very well handled.
4931

4932
    """
4933
    steps_total = 6
4934
    warning, info = (self.proc.LogWarning, self.proc.LogInfo)
4935
    instance = self.instance
4936
    iv_names = {}
4937
    # start of work
4938
    cfg = self.cfg
4939
    old_node = self.tgt_node
4940
    new_node = self.new_node
4941
    pri_node = instance.primary_node
4942
    nodes_ip = {
4943
      old_node: self.cfg.GetNodeInfo(old_node).secondary_ip,
4944
      new_node: self.cfg.GetNodeInfo(new_node).secondary_ip,
4945
      pri_node: self.cfg.GetNodeInfo(pri_node).secondary_ip,
4946
      }
4947

    
4948
    # Step: check device activation
4949
    self.proc.LogStep(1, steps_total, "check device existence")
4950
    info("checking volume groups")
4951
    my_vg = cfg.GetVGName()
4952
    results = self.rpc.call_vg_list([pri_node, new_node])
4953
    for node in pri_node, new_node:
4954
      res = results[node]
4955
      if res.failed or not res.data or my_vg not in res.data:
4956
        raise errors.OpExecError("Volume group '%s' not found on %s" %
4957
                                 (my_vg, node))
4958
    for idx, dev in enumerate(instance.disks):
4959
      if idx not in self.op.disks:
4960
        continue
4961
      info("checking disk/%d on %s" % (idx, pri_node))
4962
      cfg.SetDiskID(dev, pri_node)
4963
      result = self.rpc.call_blockdev_find(pri_node, dev)
4964
      result.Raise()
4965
      if not result.data:
4966
        raise errors.OpExecError("Can't find disk/%d on node %s" %
4967
                                 (idx, pri_node))
4968

    
4969
    # Step: check other node consistency
4970
    self.proc.LogStep(2, steps_total, "check peer consistency")
4971
    for idx, dev in enumerate(instance.disks):
4972
      if idx not in self.op.disks:
4973
        continue
4974
      info("checking disk/%d consistency on %s" % (idx, pri_node))
4975
      if not _CheckDiskConsistency(self, dev, pri_node, True, ldisk=True):
4976
        raise errors.OpExecError("Primary node (%s) has degraded storage,"
4977
                                 " unsafe to replace the secondary" %
4978
                                 pri_node)
4979

    
4980
    # Step: create new storage
4981
    self.proc.LogStep(3, steps_total, "allocate new storage")
4982
    for idx, dev in enumerate(instance.disks):
4983
      info("adding new local storage on %s for disk/%d" %
4984
           (new_node, idx))
4985
      # we pass force_create=True to force LVM creation
4986
      for new_lv in dev.children:
4987
        _CreateBlockDev(self, new_node, instance, new_lv, True,
4988
                        _GetInstanceInfoText(instance), False)
4989

    
4990
    # Step 4: dbrd minors and drbd setups changes
4991
    # after this, we must manually remove the drbd minors on both the
4992
    # error and the success paths
4993
    minors = cfg.AllocateDRBDMinor([new_node for dev in instance.disks],
4994
                                   instance.name)
4995
    logging.debug("Allocated minors %s" % (minors,))
4996
    self.proc.LogStep(4, steps_total, "changing drbd configuration")
4997
    for idx, (dev, new_minor) in enumerate(zip(instance.disks, minors)):
4998
      size = dev.size
4999
      info("activating a new drbd on %s for disk/%d" % (new_node, idx))
5000
      # create new devices on new_node; note that we create two IDs:
5001
      # one without port, so the drbd will be activated without
5002
      # networking information on the new node at this stage, and one
5003
      # with network, for the latter activation in step 4
5004
      (o_node1, o_node2, o_port, o_minor1, o_minor2, o_secret) = dev.logical_id
5005
      if pri_node == o_node1:
5006
        p_minor = o_minor1
5007
      else:
5008
        p_minor = o_minor2
5009

    
5010
      new_alone_id = (pri_node, new_node, None, p_minor, new_minor, o_secret)
5011
      new_net_id = (pri_node, new_node, o_port, p_minor, new_minor, o_secret)
5012

    
5013
      iv_names[idx] = (dev, dev.children, new_net_id)
5014
      logging.debug("Allocated new_minor: %s, new_logical_id: %s", new_minor,
5015
                    new_net_id)
5016
      new_drbd = objects.Disk(dev_type=constants.LD_DRBD8,
5017
                              logical_id=new_alone_id,
5018
                              children=dev.children)
5019
      try:
5020
        _CreateBlockDev(self, new_node, instance, new_drbd, False,
5021
                        _GetInstanceInfoText(instance), False)
5022
      except error.BlockDeviceError:
5023
        self.cfg.ReleaseDRBDMinors(instance.name)
5024
        raise
5025

    
5026
    for idx, dev in enumerate(instance.disks):
5027
      # we have new devices, shutdown the drbd on the old secondary
5028
      info("shutting down drbd for disk/%d on old node" % idx)
5029
      cfg.SetDiskID(dev, old_node)
5030
      result = self.rpc.call_blockdev_shutdown(old_node, dev)
5031
      if result.failed or not result.data:
5032
        warning("Failed to shutdown drbd for disk/%d on old node" % idx,
5033
                hint="Please cleanup this device manually as soon as possible")
5034

    
5035
    info("detaching primary drbds from the network (=> standalone)")
5036
    result = self.rpc.call_drbd_disconnect_net([pri_node], nodes_ip,
5037
                                               instance.disks)[pri_node]
5038

    
5039
    msg = result.RemoteFailMsg()
5040
    if msg:
5041
      # detaches didn't succeed (unlikely)
5042
      self.cfg.ReleaseDRBDMinors(instance.name)
5043
      raise errors.OpExecError("Can't detach the disks from the network on"
5044
                               " old node: %s" % (msg,))
5045

    
5046
    # if we managed to detach at least one, we update all the disks of
5047
    # the instance to point to the new secondary
5048
    info("updating instance configuration")
5049
    for dev, _, new_logical_id in iv_names.itervalues():
5050
      dev.logical_id = new_logical_id
5051
      cfg.SetDiskID(dev, pri_node)
5052
    cfg.Update(instance)
5053
    # we can remove now the temp minors as now the new values are
5054
    # written to the config file (and therefore stable)
5055
    self.cfg.ReleaseDRBDMinors(instance.name)
5056

    
5057
    # and now perform the drbd attach
5058
    info("attaching primary drbds to new secondary (standalone => connected)")
5059
    result = self.rpc.call_drbd_attach_net([pri_node, new_node], nodes_ip,
5060
                                           instance.disks, instance.name,
5061
                                           False)
5062
    for to_node, to_result in result.items():
5063
      msg = to_result.RemoteFailMsg()
5064
      if msg:
5065
        warning("can't attach drbd disks on node %s: %s", to_node, msg,
5066
                hint="please do a gnt-instance info to see the"
5067
                " status of disks")
5068

    
5069
    # this can fail as the old devices are degraded and _WaitForSync
5070
    # does a combined result over all disks, so we don't check its
5071
    # return value
5072
    self.proc.LogStep(5, steps_total, "sync devices")
5073
    _WaitForSync(self, instance, unlock=True)
5074

    
5075
    # so check manually all the devices
5076
    for idx, (dev, old_lvs, _) in iv_names.iteritems():
5077
      cfg.SetDiskID(dev, pri_node)
5078
      result = self.rpc.call_blockdev_find(pri_node, dev)
5079
      result.Raise()
5080
      if result.data[5]:
5081
        raise errors.OpExecError("DRBD device disk/%d is degraded!" % idx)
5082

    
5083
    self.proc.LogStep(6, steps_total, "removing old storage")
5084
    for idx, (dev, old_lvs, _) in iv_names.iteritems():
5085
      info("remove logical volumes for disk/%d" % idx)
5086
      for lv in old_lvs:
5087
        cfg.SetDiskID(lv, old_node)
5088
        result = self.rpc.call_blockdev_remove(old_node, lv)
5089
        if result.failed or not result.data:
5090
          warning("Can't remove LV on old secondary",
5091
                  hint="Cleanup stale volumes by hand")
5092

    
5093
  def Exec(self, feedback_fn):
5094
    """Execute disk replacement.
5095

5096
    This dispatches the disk replacement to the appropriate handler.
5097

5098
    """
5099
    instance = self.instance
5100

    
5101
    # Activate the instance disks if we're replacing them on a down instance
5102
    if instance.status == "down":
5103
      _StartInstanceDisks(self, instance, True)
5104

    
5105
    if self.op.mode == constants.REPLACE_DISK_CHG:
5106
      fn = self._ExecD8Secondary
5107
    else:
5108
      fn = self._ExecD8DiskOnly
5109

    
5110
    ret = fn(feedback_fn)
5111

    
5112
    # Deactivate the instance disks if we're replacing them on a down instance
5113
    if instance.status == "down":
5114
      _SafeShutdownInstanceDisks(self, instance)
5115

    
5116
    return ret
5117

    
5118

    
5119
class LUGrowDisk(LogicalUnit):
5120
  """Grow a disk of an instance.
5121

5122
  """
5123
  HPATH = "disk-grow"
5124
  HTYPE = constants.HTYPE_INSTANCE
5125
  _OP_REQP = ["instance_name", "disk", "amount", "wait_for_sync"]
5126
  REQ_BGL = False
5127

    
5128
  def ExpandNames(self):
5129
    self._ExpandAndLockInstance()
5130
    self.needed_locks[locking.LEVEL_NODE] = []
5131
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
5132

    
5133
  def DeclareLocks(self, level):
5134
    if level == locking.LEVEL_NODE:
5135
      self._LockInstancesNodes()
5136

    
5137
  def BuildHooksEnv(self):
5138
    """Build hooks env.
5139

5140
    This runs on the master, the primary and all the secondaries.
5141

5142
    """
5143
    env = {
5144
      "DISK": self.op.disk,
5145
      "AMOUNT": self.op.amount,
5146
      }
5147
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
5148
    nl = [
5149
      self.cfg.GetMasterNode(),
5150
      self.instance.primary_node,
5151
      ]
5152
    return env, nl, nl
5153

    
5154
  def CheckPrereq(self):
5155
    """Check prerequisites.
5156

5157
    This checks that the instance is in the cluster.
5158

5159
    """
5160
    instance = self.cfg.GetInstanceInfo(self.op.instance_name)
5161
    assert instance is not None, \
5162
      "Cannot retrieve locked instance %s" % self.op.instance_name
5163
    _CheckNodeOnline(self, instance.primary_node)
5164
    for node in instance.secondary_nodes:
5165
      _CheckNodeOnline(self, node)
5166

    
5167

    
5168
    self.instance = instance
5169

    
5170
    if instance.disk_template not in (constants.DT_PLAIN, constants.DT_DRBD8):
5171
      raise errors.OpPrereqError("Instance's disk layout does not support"
5172
                                 " growing.")
5173

    
5174
    self.disk = instance.FindDisk(self.op.disk)
5175

    
5176
    nodenames = [instance.primary_node] + list(instance.secondary_nodes)
5177
    nodeinfo = self.rpc.call_node_info(nodenames, self.cfg.GetVGName(),
5178
                                       instance.hypervisor)
5179
    for node in nodenames:
5180
      info = nodeinfo[node]
5181
      if info.failed or not info.data:
5182
        raise errors.OpPrereqError("Cannot get current information"
5183
                                   " from node '%s'" % node)
5184
      vg_free = info.data.get('vg_free', None)
5185
      if not isinstance(vg_free, int):
5186
        raise errors.OpPrereqError("Can't compute free disk space on"
5187
                                   " node %s" % node)
5188
      if self.op.amount > vg_free:
5189
        raise errors.OpPrereqError("Not enough disk space on target node %s:"
5190
                                   " %d MiB available, %d MiB required" %
5191
                                   (node, vg_free, self.op.amount))
5192

    
5193
  def Exec(self, feedback_fn):
5194
    """Execute disk grow.
5195

5196
    """
5197
    instance = self.instance
5198
    disk = self.disk
5199
    for node in (instance.secondary_nodes + (instance.primary_node,)):
5200
      self.cfg.SetDiskID(disk, node)
5201
      result = self.rpc.call_blockdev_grow(node, disk, self.op.amount)
5202
      result.Raise()
5203
      if (not result.data or not isinstance(result.data, (list, tuple)) or
5204
          len(result.data) != 2):
5205
        raise errors.OpExecError("Grow request failed to node %s" % node)
5206
      elif not result.data[0]:
5207
        raise errors.OpExecError("Grow request failed to node %s: %s" %
5208
                                 (node, result.data[1]))
5209
    disk.RecordGrow(self.op.amount)
5210
    self.cfg.Update(instance)
5211
    if self.op.wait_for_sync:
5212
      disk_abort = not _WaitForSync(self, instance)
5213
      if disk_abort:
5214
        self.proc.LogWarning("Warning: disk sync-ing has not returned a good"
5215
                             " status.\nPlease check the instance.")
5216

    
5217

    
5218
class LUQueryInstanceData(NoHooksLU):
5219
  """Query runtime instance data.
5220

5221
  """
5222
  _OP_REQP = ["instances", "static"]
5223
  REQ_BGL = False
5224

    
5225
  def ExpandNames(self):
5226
    self.needed_locks = {}
5227
    self.share_locks = dict(((i, 1) for i in locking.LEVELS))
5228

    
5229
    if not isinstance(self.op.instances, list):
5230
      raise errors.OpPrereqError("Invalid argument type 'instances'")
5231

    
5232
    if self.op.instances:
5233
      self.wanted_names = []
5234
      for name in self.op.instances:
5235
        full_name = self.cfg.ExpandInstanceName(name)
5236
        if full_name is None:
5237
          raise errors.OpPrereqError("Instance '%s' not known" % name)
5238
        self.wanted_names.append(full_name)
5239
      self.needed_locks[locking.LEVEL_INSTANCE] = self.wanted_names
5240
    else:
5241
      self.wanted_names = None
5242
      self.needed_locks[locking.LEVEL_INSTANCE] = locking.ALL_SET
5243

    
5244
    self.needed_locks[locking.LEVEL_NODE] = []
5245
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
5246

    
5247
  def DeclareLocks(self, level):
5248
    if level == locking.LEVEL_NODE:
5249
      self._LockInstancesNodes()
5250

    
5251
  def CheckPrereq(self):
5252
    """Check prerequisites.
5253

5254
    This only checks the optional instance list against the existing names.
5255

5256
    """
5257
    if self.wanted_names is None:
5258
      self.wanted_names = self.acquired_locks[locking.LEVEL_INSTANCE]
5259

    
5260
    self.wanted_instances = [self.cfg.GetInstanceInfo(name) for name
5261
                             in self.wanted_names]
5262
    return
5263

    
5264
  def _ComputeDiskStatus(self, instance, snode, dev):
5265
    """Compute block device status.
5266

5267
    """
5268
    static = self.op.static
5269
    if not static:
5270
      self.cfg.SetDiskID(dev, instance.primary_node)
5271
      dev_pstatus = self.rpc.call_blockdev_find(instance.primary_node, dev)
5272
      dev_pstatus.Raise()
5273
      dev_pstatus = dev_pstatus.data
5274
    else:
5275
      dev_pstatus = None
5276

    
5277
    if dev.dev_type in constants.LDS_DRBD:
5278
      # we change the snode then (otherwise we use the one passed in)
5279
      if dev.logical_id[0] == instance.primary_node:
5280
        snode = dev.logical_id[1]
5281
      else:
5282
        snode = dev.logical_id[0]
5283

    
5284
    if snode and not static:
5285
      self.cfg.SetDiskID(dev, snode)
5286
      dev_sstatus = self.rpc.call_blockdev_find(snode, dev)
5287
      dev_sstatus.Raise()
5288
      dev_sstatus = dev_sstatus.data
5289
    else:
5290
      dev_sstatus = None
5291

    
5292
    if dev.children:
5293
      dev_children = [self._ComputeDiskStatus(instance, snode, child)
5294
                      for child in dev.children]
5295
    else:
5296
      dev_children = []
5297

    
5298
    data = {
5299
      "iv_name": dev.iv_name,
5300
      "dev_type": dev.dev_type,
5301
      "logical_id": dev.logical_id,
5302
      "physical_id": dev.physical_id,
5303
      "pstatus": dev_pstatus,
5304
      "sstatus": dev_sstatus,
5305
      "children": dev_children,
5306
      "mode": dev.mode,
5307
      }
5308

    
5309
    return data
5310

    
5311
  def Exec(self, feedback_fn):
5312
    """Gather and return data"""
5313
    result = {}
5314

    
5315
    cluster = self.cfg.GetClusterInfo()
5316

    
5317
    for instance in self.wanted_instances:
5318
      if not self.op.static:
5319
        remote_info = self.rpc.call_instance_info(instance.primary_node,
5320
                                                  instance.name,
5321
                                                  instance.hypervisor)
5322
        remote_info.Raise()
5323
        remote_info = remote_info.data
5324
        if remote_info and "state" in remote_info:
5325
          remote_state = "up"
5326
        else:
5327
          remote_state = "down"
5328
      else:
5329
        remote_state = None
5330
      if instance.status == "down":
5331
        config_state = "down"
5332
      else:
5333
        config_state = "up"
5334

    
5335
      disks = [self._ComputeDiskStatus(instance, None, device)
5336
               for device in instance.disks]
5337

    
5338
      idict = {
5339
        "name": instance.name,
5340
        "config_state": config_state,
5341
        "run_state": remote_state,
5342
        "pnode": instance.primary_node,
5343
        "snodes": instance.secondary_nodes,
5344
        "os": instance.os,
5345
        "nics": [(nic.mac, nic.ip, nic.bridge) for nic in instance.nics],
5346
        "disks": disks,
5347
        "hypervisor": instance.hypervisor,
5348
        "network_port": instance.network_port,
5349
        "hv_instance": instance.hvparams,
5350
        "hv_actual": cluster.FillHV(instance),
5351
        "be_instance": instance.beparams,
5352
        "be_actual": cluster.FillBE(instance),
5353
        }
5354

    
5355
      result[instance.name] = idict
5356

    
5357
    return result
5358

    
5359

    
5360
class LUSetInstanceParams(LogicalUnit):
5361
  """Modifies an instances's parameters.
5362

5363
  """
5364
  HPATH = "instance-modify"
5365
  HTYPE = constants.HTYPE_INSTANCE
5366
  _OP_REQP = ["instance_name"]
5367
  REQ_BGL = False
5368

    
5369
  def CheckArguments(self):
5370
    if not hasattr(self.op, 'nics'):
5371
      self.op.nics = []
5372
    if not hasattr(self.op, 'disks'):
5373
      self.op.disks = []
5374
    if not hasattr(self.op, 'beparams'):
5375
      self.op.beparams = {}
5376
    if not hasattr(self.op, 'hvparams'):
5377
      self.op.hvparams = {}
5378
    self.op.force = getattr(self.op, "force", False)
5379
    if not (self.op.nics or self.op.disks or
5380
            self.op.hvparams or self.op.beparams):
5381
      raise errors.OpPrereqError("No changes submitted")
5382

    
5383
    utils.CheckBEParams(self.op.beparams)
5384

    
5385
    # Disk validation
5386
    disk_addremove = 0
5387
    for disk_op, disk_dict in self.op.disks:
5388
      if disk_op == constants.DDM_REMOVE:
5389
        disk_addremove += 1
5390
        continue
5391
      elif disk_op == constants.DDM_ADD:
5392
        disk_addremove += 1
5393
      else:
5394
        if not isinstance(disk_op, int):
5395
          raise errors.OpPrereqError("Invalid disk index")
5396
      if disk_op == constants.DDM_ADD:
5397
        mode = disk_dict.setdefault('mode', constants.DISK_RDWR)
5398
        if mode not in (constants.DISK_RDONLY, constants.DISK_RDWR):
5399
          raise errors.OpPrereqError("Invalid disk access mode '%s'" % mode)
5400
        size = disk_dict.get('size', None)
5401
        if size is None:
5402
          raise errors.OpPrereqError("Required disk parameter size missing")
5403
        try:
5404
          size = int(size)
5405
        except ValueError, err:
5406
          raise errors.OpPrereqError("Invalid disk size parameter: %s" %
5407
                                     str(err))
5408
        disk_dict['size'] = size
5409
      else:
5410
        # modification of disk
5411
        if 'size' in disk_dict:
5412
          raise errors.OpPrereqError("Disk size change not possible, use"
5413
                                     " grow-disk")
5414

    
5415
    if disk_addremove > 1:
5416
      raise errors.OpPrereqError("Only one disk add or remove operation"
5417
                                 " supported at a time")
5418

    
5419
    # NIC validation
5420
    nic_addremove = 0
5421
    for nic_op, nic_dict in self.op.nics:
5422
      if nic_op == constants.DDM_REMOVE:
5423
        nic_addremove += 1
5424
        continue
5425
      elif nic_op == constants.DDM_ADD:
5426
        nic_addremove += 1
5427
      else:
5428
        if not isinstance(nic_op, int):
5429
          raise errors.OpPrereqError("Invalid nic index")
5430

    
5431
      # nic_dict should be a dict
5432
      nic_ip = nic_dict.get('ip', None)
5433
      if nic_ip is not None:
5434
        if nic_ip.lower() == "none":
5435
          nic_dict['ip'] = None
5436
        else:
5437
          if not utils.IsValidIP(nic_ip):
5438
            raise errors.OpPrereqError("Invalid IP address '%s'" % nic_ip)
5439
      # we can only check None bridges and assign the default one
5440
      nic_bridge = nic_dict.get('bridge', None)
5441
      if nic_bridge is None:
5442
        nic_dict['bridge'] = self.cfg.GetDefBridge()
5443
      # but we can validate MACs
5444
      nic_mac = nic_dict.get('mac', None)
5445
      if nic_mac is not None:
5446
        if self.cfg.IsMacInUse(nic_mac):
5447
          raise errors.OpPrereqError("MAC address %s already in use"
5448
                                     " in cluster" % nic_mac)
5449
        if nic_mac not in (constants.VALUE_AUTO, constants.VALUE_GENERATE):
5450
          if not utils.IsValidMac(nic_mac):
5451
            raise errors.OpPrereqError("Invalid MAC address %s" % nic_mac)
5452
    if nic_addremove > 1:
5453
      raise errors.OpPrereqError("Only one NIC add or remove operation"
5454
                                 " supported at a time")
5455

    
5456
  def ExpandNames(self):
5457
    self._ExpandAndLockInstance()
5458
    self.needed_locks[locking.LEVEL_NODE] = []
5459
    self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_REPLACE
5460

    
5461
  def DeclareLocks(self, level):
5462
    if level == locking.LEVEL_NODE:
5463
      self._LockInstancesNodes()
5464

    
5465
  def BuildHooksEnv(self):
5466
    """Build hooks env.
5467

5468
    This runs on the master, primary and secondaries.
5469

5470
    """
5471
    args = dict()
5472
    if constants.BE_MEMORY in self.be_new:
5473
      args['memory'] = self.be_new[constants.BE_MEMORY]
5474
    if constants.BE_VCPUS in self.be_new:
5475
      args['vcpus'] = self.be_new[constants.BE_VCPUS]
5476
    # FIXME: readd disk/nic changes
5477
    env = _BuildInstanceHookEnvByObject(self, self.instance, override=args)
5478
    nl = [self.cfg.GetMasterNode(),
5479
          self.instance.primary_node] + list(self.instance.secondary_nodes)
5480
    return env, nl, nl
5481

    
5482
  def CheckPrereq(self):
5483
    """Check prerequisites.
5484

5485
    This only checks the instance list against the existing names.
5486

5487
    """
5488
    force = self.force = self.op.force
5489

    
5490
    # checking the new params on the primary/secondary nodes
5491

    
5492
    instance = self.instance = self.cfg.GetInstanceInfo(self.op.instance_name)
5493
    assert self.instance is not None, \
5494
      "Cannot retrieve locked instance %s" % self.op.instance_name
5495
    pnode = self.instance.primary_node
5496
    nodelist = [pnode]
5497
    nodelist.extend(instance.secondary_nodes)
5498

    
5499
    # hvparams processing
5500
    if self.op.hvparams:
5501
      i_hvdict = copy.deepcopy(instance.hvparams)
5502
      for key, val in self.op.hvparams.iteritems():
5503
        if val == constants.VALUE_DEFAULT:
5504
          try:
5505
            del i_hvdict[key]
5506
          except KeyError:
5507
            pass
5508
        elif val == constants.VALUE_NONE:
5509
          i_hvdict[key] = None
5510
        else:
5511
          i_hvdict[key] = val
5512
      cluster = self.cfg.GetClusterInfo()
5513
      hv_new = cluster.FillDict(cluster.hvparams[instance.hypervisor],
5514
                                i_hvdict)
5515
      # local check
5516
      hypervisor.GetHypervisor(
5517
        instance.hypervisor).CheckParameterSyntax(hv_new)
5518
      _CheckHVParams(self, nodelist, instance.hypervisor, hv_new)
5519
      self.hv_new = hv_new # the new actual values
5520
      self.hv_inst = i_hvdict # the new dict (without defaults)
5521
    else:
5522
      self.hv_new = self.hv_inst = {}
5523

    
5524
    # beparams processing
5525
    if self.op.beparams:
5526
      i_bedict = copy.deepcopy(instance.beparams)
5527
      for key, val in self.op.beparams.iteritems():
5528
        if val == constants.VALUE_DEFAULT:
5529
          try:
5530
            del i_bedict[key]
5531
          except KeyError:
5532
            pass
5533
        else:
5534
          i_bedict[key] = val
5535
      cluster = self.cfg.GetClusterInfo()
5536
      be_new = cluster.FillDict(cluster.beparams[constants.BEGR_DEFAULT],
5537
                                i_bedict)
5538
      self.be_new = be_new # the new actual values
5539
      self.be_inst = i_bedict # the new dict (without defaults)
5540
    else:
5541
      self.be_new = self.be_inst = {}
5542

    
5543
    self.warn = []
5544

    
5545
    if constants.BE_MEMORY in self.op.beparams and not self.force:
5546
      mem_check_list = [pnode]
5547
      if be_new[constants.BE_AUTO_BALANCE]:
5548
        # either we changed auto_balance to yes or it was from before
5549
        mem_check_list.extend(instance.secondary_nodes)
5550
      instance_info = self.rpc.call_instance_info(pnode, instance.name,
5551
                                                  instance.hypervisor)
5552
      nodeinfo = self.rpc.call_node_info(mem_check_list, self.cfg.GetVGName(),
5553
                                         instance.hypervisor)
5554
      if nodeinfo[pnode].failed or not isinstance(nodeinfo[pnode].data, dict):
5555
        # Assume the primary node is unreachable and go ahead
5556
        self.warn.append("Can't get info from primary node %s" % pnode)
5557
      else:
5558
        if not instance_info.failed and instance_info.data:
5559
          current_mem = instance_info.data['memory']
5560
        else:
5561
          # Assume instance not running
5562
          # (there is a slight race condition here, but it's not very probable,
5563
          # and we have no other way to check)
5564
          current_mem = 0
5565
        miss_mem = (be_new[constants.BE_MEMORY] - current_mem -
5566
                    nodeinfo[pnode].data['memory_free'])
5567
        if miss_mem > 0:
5568
          raise errors.OpPrereqError("This change will prevent the instance"
5569
                                     " from starting, due to %d MB of memory"
5570
                                     " missing on its primary node" % miss_mem)
5571

    
5572
      if be_new[constants.BE_AUTO_BALANCE]:
5573
        for node, nres in nodeinfo.iteritems():
5574
          if node not in instance.secondary_nodes:
5575
            continue
5576
          if nres.failed or not isinstance(nres.data, dict):
5577
            self.warn.append("Can't get info from secondary node %s" % node)
5578
          elif be_new[constants.BE_MEMORY] > nres.data['memory_free']:
5579
            self.warn.append("Not enough memory to failover instance to"
5580
                             " secondary node %s" % node)
5581

    
5582
    # NIC processing
5583
    for nic_op, nic_dict in self.op.nics:
5584
      if nic_op == constants.DDM_REMOVE:
5585
        if not instance.nics:
5586
          raise errors.OpPrereqError("Instance has no NICs, cannot remove")
5587
        continue
5588
      if nic_op != constants.DDM_ADD:
5589
        # an existing nic
5590
        if nic_op < 0 or nic_op >= len(instance.nics):
5591
          raise errors.OpPrereqError("Invalid NIC index %s, valid values"
5592
                                     " are 0 to %d" %
5593
                                     (nic_op, len(instance.nics)))
5594
      nic_bridge = nic_dict.get('bridge', None)
5595
      if nic_bridge is not None:
5596
        if not self.rpc.call_bridges_exist(pnode, [nic_bridge]):
5597
          msg = ("Bridge '%s' doesn't exist on one of"
5598
                 " the instance nodes" % nic_bridge)
5599
          if self.force:
5600
            self.warn.append(msg)
5601
          else:
5602
            raise errors.OpPrereqError(msg)
5603

    
5604
    # DISK processing
5605
    if self.op.disks and instance.disk_template == constants.DT_DISKLESS:
5606
      raise errors.OpPrereqError("Disk operations not supported for"
5607
                                 " diskless instances")
5608
    for disk_op, disk_dict in self.op.disks:
5609
      if disk_op == constants.DDM_REMOVE:
5610
        if len(instance.disks) == 1:
5611
          raise errors.OpPrereqError("Cannot remove the last disk of"
5612
                                     " an instance")
5613
        ins_l = self.rpc.call_instance_list([pnode], [instance.hypervisor])
5614
        ins_l = ins_l[pnode]
5615
        if ins_l.failed or not isinstance(ins_l.data, list):
5616
          raise errors.OpPrereqError("Can't contact node '%s'" % pnode)
5617
        if instance.name in ins_l.data:
5618
          raise errors.OpPrereqError("Instance is running, can't remove"
5619
                                     " disks.")
5620

    
5621
      if (disk_op == constants.DDM_ADD and
5622
          len(instance.nics) >= constants.MAX_DISKS):
5623
        raise errors.OpPrereqError("Instance has too many disks (%d), cannot"
5624
                                   " add more" % constants.MAX_DISKS)
5625
      if disk_op not in (constants.DDM_ADD, constants.DDM_REMOVE):
5626
        # an existing disk
5627
        if disk_op < 0 or disk_op >= len(instance.disks):
5628
          raise errors.OpPrereqError("Invalid disk index %s, valid values"
5629
                                     " are 0 to %d" %
5630
                                     (disk_op, len(instance.disks)))
5631

    
5632
    return
5633

    
5634
  def Exec(self, feedback_fn):
5635
    """Modifies an instance.
5636

5637
    All parameters take effect only at the next restart of the instance.
5638

5639
    """
5640
    # Process here the warnings from CheckPrereq, as we don't have a
5641
    # feedback_fn there.
5642
    for warn in self.warn:
5643
      feedback_fn("WARNING: %s" % warn)
5644

    
5645
    result = []
5646
    instance = self.instance
5647
    # disk changes
5648
    for disk_op, disk_dict in self.op.disks:
5649
      if disk_op == constants.DDM_REMOVE:
5650
        # remove the last disk
5651
        device = instance.disks.pop()
5652
        device_idx = len(instance.disks)
5653
        for node, disk in device.ComputeNodeTree(instance.primary_node):
5654
          self.cfg.SetDiskID(disk, node)
5655
          rpc_result = self.rpc.call_blockdev_remove(node, disk)
5656
          if rpc_result.failed or not rpc_result.data:
5657
            self.proc.LogWarning("Could not remove disk/%d on node %s,"
5658
                                 " continuing anyway", device_idx, node)
5659
        result.append(("disk/%d" % device_idx, "remove"))
5660
      elif disk_op == constants.DDM_ADD:
5661
        # add a new disk
5662
        if instance.disk_template == constants.DT_FILE:
5663
          file_driver, file_path = instance.disks[0].logical_id
5664
          file_path = os.path.dirname(file_path)
5665
        else:
5666
          file_driver = file_path = None
5667
        disk_idx_base = len(instance.disks)
5668
        new_disk = _GenerateDiskTemplate(self,
5669
                                         instance.disk_template,
5670
                                         instance, instance.primary_node,
5671
                                         instance.secondary_nodes,
5672
                                         [disk_dict],
5673
                                         file_path,
5674
                                         file_driver,
5675
                                         disk_idx_base)[0]
5676
        new_disk.mode = disk_dict['mode']
5677
        instance.disks.append(new_disk)
5678
        info = _GetInstanceInfoText(instance)
5679

    
5680
        logging.info("Creating volume %s for instance %s",
5681
                     new_disk.iv_name, instance.name)
5682
        # Note: this needs to be kept in sync with _CreateDisks
5683
        #HARDCODE
5684
        for node in instance.all_nodes:
5685
          f_create = node == instance.primary_node
5686
          try:
5687
            _CreateBlockDev(self, node, instance, new_disk,
5688
                            f_create, info, f_create)
5689
          except error.OpExecError, err:
5690
            self.LogWarning("Failed to create volume %s (%s) on"
5691
                            " node %s: %s",
5692
                            new_disk.iv_name, new_disk, node, err)
5693
        result.append(("disk/%d" % disk_idx_base, "add:size=%s,mode=%s" %
5694
                       (new_disk.size, new_disk.mode)))
5695
      else:
5696
        # change a given disk
5697
        instance.disks[disk_op].mode = disk_dict['mode']
5698
        result.append(("disk.mode/%d" % disk_op, disk_dict['mode']))
5699
    # NIC changes
5700
    for nic_op, nic_dict in self.op.nics:
5701
      if nic_op == constants.DDM_REMOVE:
5702
        # remove the last nic
5703
        del instance.nics[-1]
5704
        result.append(("nic.%d" % len(instance.nics), "remove"))
5705
      elif nic_op == constants.DDM_ADD:
5706
        # add a new nic
5707
        if 'mac' not in nic_dict:
5708
          mac = constants.VALUE_GENERATE
5709
        else:
5710
          mac = nic_dict['mac']
5711
        if mac in (constants.VALUE_AUTO, constants.VALUE_GENERATE):
5712
          mac = self.cfg.GenerateMAC()
5713
        new_nic = objects.NIC(mac=mac, ip=nic_dict.get('ip', None),
5714
                              bridge=nic_dict.get('bridge', None))
5715
        instance.nics.append(new_nic)
5716
        result.append(("nic.%d" % (len(instance.nics) - 1),
5717
                       "add:mac=%s,ip=%s,bridge=%s" %
5718
                       (new_nic.mac, new_nic.ip, new_nic.bridge)))
5719
      else:
5720
        # change a given nic
5721
        for key in 'mac', 'ip', 'bridge':
5722
          if key in nic_dict:
5723
            setattr(instance.nics[nic_op], key, nic_dict[key])
5724
            result.append(("nic.%s/%d" % (key, nic_op), nic_dict[key]))
5725

    
5726
    # hvparams changes
5727
    if self.op.hvparams:
5728
      instance.hvparams = self.hv_new
5729
      for key, val in self.op.hvparams.iteritems():
5730
        result.append(("hv/%s" % key, val))
5731

    
5732
    # beparams changes
5733
    if self.op.beparams:
5734
      instance.beparams = self.be_inst
5735
      for key, val in self.op.beparams.iteritems():
5736
        result.append(("be/%s" % key, val))
5737

    
5738
    self.cfg.Update(instance)
5739

    
5740
    return result
5741

    
5742

    
5743
class LUQueryExports(NoHooksLU):
5744
  """Query the exports list
5745

5746
  """
5747
  _OP_REQP = ['nodes']
5748
  REQ_BGL = False
5749

    
5750
  def ExpandNames(self):
5751
    self.needed_locks = {}
5752
    self.share_locks[locking.LEVEL_NODE] = 1
5753
    if not self.op.nodes:
5754
      self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
5755
    else:
5756
      self.needed_locks[locking.LEVEL_NODE] = \
5757
        _GetWantedNodes(self, self.op.nodes)
5758

    
5759
  def CheckPrereq(self):
5760
    """Check prerequisites.
5761

5762
    """
5763
    self.nodes = self.acquired_locks[locking.LEVEL_NODE]
5764

    
5765
  def Exec(self, feedback_fn):
5766
    """Compute the list of all the exported system images.
5767

5768
    @rtype: dict
5769
    @return: a dictionary with the structure node->(export-list)
5770
        where export-list is a list of the instances exported on
5771
        that node.
5772

5773
    """
5774
    rpcresult = self.rpc.call_export_list(self.nodes)
5775
    result = {}
5776
    for node in rpcresult:
5777
      if rpcresult[node].failed:
5778
        result[node] = False
5779
      else:
5780
        result[node] = rpcresult[node].data
5781

    
5782
    return result
5783

    
5784

    
5785
class LUExportInstance(LogicalUnit):
5786
  """Export an instance to an image in the cluster.
5787

5788
  """
5789
  HPATH = "instance-export"
5790
  HTYPE = constants.HTYPE_INSTANCE
5791
  _OP_REQP = ["instance_name", "target_node", "shutdown"]
5792
  REQ_BGL = False
5793

    
5794
  def ExpandNames(self):
5795
    self._ExpandAndLockInstance()
5796
    # FIXME: lock only instance primary and destination node
5797
    #
5798
    # Sad but true, for now we have do lock all nodes, as we don't know where
5799
    # the previous export might be, and and in this LU we search for it and
5800
    # remove it from its current node. In the future we could fix this by:
5801
    #  - making a tasklet to search (share-lock all), then create the new one,
5802
    #    then one to remove, after
5803
    #  - removing the removal operation altoghether
5804
    self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
5805

    
5806
  def DeclareLocks(self, level):
5807
    """Last minute lock declaration."""
5808
    # All nodes are locked anyway, so nothing to do here.
5809

    
5810
  def BuildHooksEnv(self):
5811
    """Build hooks env.
5812

5813
    This will run on the master, primary node and target node.
5814

5815
    """
5816
    env = {
5817
      "EXPORT_NODE": self.op.target_node,
5818
      "EXPORT_DO_SHUTDOWN": self.op.shutdown,
5819
      }
5820
    env.update(_BuildInstanceHookEnvByObject(self, self.instance))
5821
    nl = [self.cfg.GetMasterNode(), self.instance.primary_node,
5822
          self.op.target_node]
5823
    return env, nl, nl
5824

    
5825
  def CheckPrereq(self):
5826
    """Check prerequisites.
5827

5828
    This checks that the instance and node names are valid.
5829

5830
    """
5831
    instance_name = self.op.instance_name
5832
    self.instance = self.cfg.GetInstanceInfo(instance_name)
5833
    assert self.instance is not None, \
5834
          "Cannot retrieve locked instance %s" % self.op.instance_name
5835
    _CheckNodeOnline(self, self.instance.primary_node)
5836

    
5837
    self.dst_node = self.cfg.GetNodeInfo(
5838
      self.cfg.ExpandNodeName(self.op.target_node))
5839

    
5840
    if self.dst_node is None:
5841
      # This is wrong node name, not a non-locked node
5842
      raise errors.OpPrereqError("Wrong node name %s" % self.op.target_node)
5843
    _CheckNodeOnline(self, self.dst_node.name)
5844

    
5845
    # instance disk type verification
5846
    for disk in self.instance.disks:
5847
      if disk.dev_type == constants.LD_FILE:
5848
        raise errors.OpPrereqError("Export not supported for instances with"
5849
                                   " file-based disks")
5850

    
5851
  def Exec(self, feedback_fn):
5852
    """Export an instance to an image in the cluster.
5853

5854
    """
5855
    instance = self.instance
5856
    dst_node = self.dst_node
5857
    src_node = instance.primary_node
5858
    if self.op.shutdown:
5859
      # shutdown the instance, but not the disks
5860
      result = self.rpc.call_instance_shutdown(src_node, instance)
5861
      result.Raise()
5862
      if not result.data:
5863
        raise errors.OpExecError("Could not shutdown instance %s on node %s" %
5864
                                 (instance.name, src_node))
5865

    
5866
    vgname = self.cfg.GetVGName()
5867

    
5868
    snap_disks = []
5869

    
5870
    # set the disks ID correctly since call_instance_start needs the
5871
    # correct drbd minor to create the symlinks
5872
    for disk in instance.disks:
5873
      self.cfg.SetDiskID(disk, src_node)
5874

    
5875
    try:
5876
      for disk in instance.disks:
5877
        # new_dev_name will be a snapshot of an lvm leaf of the one we passed
5878
        new_dev_name = self.rpc.call_blockdev_snapshot(src_node, disk)
5879
        if new_dev_name.failed or not new_dev_name.data:
5880
          self.LogWarning("Could not snapshot block device %s on node %s",
5881
                          disk.logical_id[1], src_node)
5882
          snap_disks.append(False)
5883
        else:
5884
          new_dev = objects.Disk(dev_type=constants.LD_LV, size=disk.size,
5885
                                 logical_id=(vgname, new_dev_name.data),
5886
                                 physical_id=(vgname, new_dev_name.data),
5887
                                 iv_name=disk.iv_name)
5888
          snap_disks.append(new_dev)
5889

    
5890
    finally:
5891
      if self.op.shutdown and instance.status == "up":
5892
        result = self.rpc.call_instance_start(src_node, instance, None)
5893
        if result.failed or not result.data:
5894
          _ShutdownInstanceDisks(self, instance)
5895
          raise errors.OpExecError("Could not start instance")
5896

    
5897
    # TODO: check for size
5898

    
5899
    cluster_name = self.cfg.GetClusterName()
5900
    for idx, dev in enumerate(snap_disks):
5901
      if dev:
5902
        result = self.rpc.call_snapshot_export(src_node, dev, dst_node.name,
5903
                                               instance, cluster_name, idx)
5904
        if result.failed or not result.data:
5905
          self.LogWarning("Could not export block device %s from node %s to"
5906
                          " node %s", dev.logical_id[1], src_node,
5907
                          dst_node.name)
5908
        result = self.rpc.call_blockdev_remove(src_node, dev)
5909
        if result.failed or not result.data:
5910
          self.LogWarning("Could not remove snapshot block device %s from node"
5911
                          " %s", dev.logical_id[1], src_node)
5912

    
5913
    result = self.rpc.call_finalize_export(dst_node.name, instance, snap_disks)
5914
    if result.failed or not result.data:
5915
      self.LogWarning("Could not finalize export for instance %s on node %s",
5916
                      instance.name, dst_node.name)
5917

    
5918
    nodelist = self.cfg.GetNodeList()
5919
    nodelist.remove(dst_node.name)
5920

    
5921
    # on one-node clusters nodelist will be empty after the removal
5922
    # if we proceed the backup would be removed because OpQueryExports
5923
    # substitutes an empty list with the full cluster node list.
5924
    if nodelist:
5925
      exportlist = self.rpc.call_export_list(nodelist)
5926
      for node in exportlist:
5927
        if exportlist[node].failed:
5928
          continue
5929
        if instance.name in exportlist[node].data:
5930
          if not self.rpc.call_export_remove(node, instance.name):
5931
            self.LogWarning("Could not remove older export for instance %s"
5932
                            " on node %s", instance.name, node)
5933

    
5934

    
5935
class LURemoveExport(NoHooksLU):
5936
  """Remove exports related to the named instance.
5937

5938
  """
5939
  _OP_REQP = ["instance_name"]
5940
  REQ_BGL = False
5941

    
5942
  def ExpandNames(self):
5943
    self.needed_locks = {}
5944
    # We need all nodes to be locked in order for RemoveExport to work, but we
5945
    # don't need to lock the instance itself, as nothing will happen to it (and
5946
    # we can remove exports also for a removed instance)
5947
    self.needed_locks[locking.LEVEL_NODE] = locking.ALL_SET
5948

    
5949
  def CheckPrereq(self):
5950
    """Check prerequisites.
5951
    """
5952
    pass
5953

    
5954
  def Exec(self, feedback_fn):
5955
    """Remove any export.
5956

5957
    """
5958
    instance_name = self.cfg.ExpandInstanceName(self.op.instance_name)
5959
    # If the instance was not found we'll try with the name that was passed in.
5960
    # This will only work if it was an FQDN, though.
5961
    fqdn_warn = False
5962
    if not instance_name:
5963
      fqdn_warn = True
5964
      instance_name = self.op.instance_name
5965

    
5966
    exportlist = self.rpc.call_export_list(self.acquired_locks[
5967
      locking.LEVEL_NODE])
5968
    found = False
5969
    for node in exportlist:
5970
      if exportlist[node].failed:
5971
        self.LogWarning("Failed to query node %s, continuing" % node)
5972
        continue
5973
      if instance_name in exportlist[node].data:
5974
        found = True
5975
        result = self.rpc.call_export_remove(node, instance_name)
5976
        if result.failed or not result.data:
5977
          logging.error("Could not remove export for instance %s"
5978
                        " on node %s", instance_name, node)
5979

    
5980
    if fqdn_warn and not found:
5981
      feedback_fn("Export not found. If trying to remove an export belonging"
5982
                  " to a deleted instance please use its Fully Qualified"
5983
                  " Domain Name.")
5984

    
5985

    
5986
class TagsLU(NoHooksLU):
5987
  """Generic tags LU.
5988

5989
  This is an abstract class which is the parent of all the other tags LUs.
5990

5991
  """
5992

    
5993
  def ExpandNames(self):
5994
    self.needed_locks = {}
5995
    if self.op.kind == constants.TAG_NODE:
5996
      name = self.cfg.ExpandNodeName(self.op.name)
5997
      if name is None:
5998
        raise errors.OpPrereqError("Invalid node name (%s)" %
5999
                                   (self.op.name,))
6000
      self.op.name = name
6001
      self.needed_locks[locking.LEVEL_NODE] = name
6002
    elif self.op.kind == constants.TAG_INSTANCE:
6003
      name = self.cfg.ExpandInstanceName(self.op.name)
6004
      if name is None:
6005
        raise errors.OpPrereqError("Invalid instance name (%s)" %
6006
                                   (self.op.name,))
6007
      self.op.name = name
6008
      self.needed_locks[locking.LEVEL_INSTANCE] = name
6009

    
6010
  def CheckPrereq(self):
6011
    """Check prerequisites.
6012

6013
    """
6014
    if self.op.kind == constants.TAG_CLUSTER:
6015
      self.target = self.cfg.GetClusterInfo()
6016
    elif self.op.kind == constants.TAG_NODE:
6017
      self.target = self.cfg.GetNodeInfo(self.op.name)
6018
    elif self.op.kind == constants.TAG_INSTANCE:
6019
      self.target = self.cfg.GetInstanceInfo(self.op.name)
6020
    else:
6021
      raise errors.OpPrereqError("Wrong tag type requested (%s)" %
6022
                                 str(self.op.kind))
6023

    
6024

    
6025
class LUGetTags(TagsLU):
6026
  """Returns the tags of a given object.
6027

6028
  """
6029
  _OP_REQP = ["kind", "name"]
6030
  REQ_BGL = False
6031

    
6032
  def Exec(self, feedback_fn):
6033
    """Returns the tag list.
6034

6035
    """
6036
    return list(self.target.GetTags())
6037

    
6038

    
6039
class LUSearchTags(NoHooksLU):
6040
  """Searches the tags for a given pattern.
6041

6042
  """
6043
  _OP_REQP = ["pattern"]
6044
  REQ_BGL = False
6045

    
6046
  def ExpandNames(self):
6047
    self.needed_locks = {}
6048

    
6049
  def CheckPrereq(self):
6050
    """Check prerequisites.
6051

6052
    This checks the pattern passed for validity by compiling it.
6053

6054
    """
6055
    try:
6056
      self.re = re.compile(self.op.pattern)
6057
    except re.error, err:
6058
      raise errors.OpPrereqError("Invalid search pattern '%s': %s" %
6059
                                 (self.op.pattern, err))
6060

    
6061
  def Exec(self, feedback_fn):
6062
    """Returns the tag list.
6063

6064
    """
6065
    cfg = self.cfg
6066
    tgts = [("/cluster", cfg.GetClusterInfo())]
6067
    ilist = cfg.GetAllInstancesInfo().values()
6068
    tgts.extend([("/instances/%s" % i.name, i) for i in ilist])
6069
    nlist = cfg.GetAllNodesInfo().values()
6070
    tgts.extend([("/nodes/%s" % n.name, n) for n in nlist])
6071
    results = []
6072
    for path, target in tgts:
6073
      for tag in target.GetTags():
6074
        if self.re.search(tag):
6075
          results.append((path, tag))
6076
    return results
6077

    
6078

    
6079
class LUAddTags(TagsLU):
6080
  """Sets a tag on a given object.
6081

6082
  """
6083
  _OP_REQP = ["kind", "name", "tags"]
6084
  REQ_BGL = False
6085

    
6086
  def CheckPrereq(self):
6087
    """Check prerequisites.
6088

6089
    This checks the type and length of the tag name and value.
6090

6091
    """
6092
    TagsLU.CheckPrereq(self)
6093
    for tag in self.op.tags:
6094
      objects.TaggableObject.ValidateTag(tag)
6095

    
6096
  def Exec(self, feedback_fn):
6097
    """Sets the tag.
6098

6099
    """
6100
    try:
6101
      for tag in self.op.tags:
6102
        self.target.AddTag(tag)
6103
    except errors.TagError, err:
6104
      raise errors.OpExecError("Error while setting tag: %s" % str(err))
6105
    try:
6106
      self.cfg.Update(self.target)
6107
    except errors.ConfigurationError:
6108
      raise errors.OpRetryError("There has been a modification to the"
6109
                                " config file and the operation has been"
6110
                                " aborted. Please retry.")
6111

    
6112

    
6113
class LUDelTags(TagsLU):
6114
  """Delete a list of tags from a given object.
6115

6116
  """
6117
  _OP_REQP = ["kind", "name", "tags"]
6118
  REQ_BGL = False
6119

    
6120
  def CheckPrereq(self):
6121
    """Check prerequisites.
6122

6123
    This checks that we have the given tag.
6124

6125
    """
6126
    TagsLU.CheckPrereq(self)
6127
    for tag in self.op.tags:
6128
      objects.TaggableObject.ValidateTag(tag)
6129
    del_tags = frozenset(self.op.tags)
6130
    cur_tags = self.target.GetTags()
6131
    if not del_tags <= cur_tags:
6132
      diff_tags = del_tags - cur_tags
6133
      diff_names = ["'%s'" % tag for tag in diff_tags]
6134
      diff_names.sort()
6135
      raise errors.OpPrereqError("Tag(s) %s not found" %
6136
                                 (",".join(diff_names)))
6137

    
6138
  def Exec(self, feedback_fn):
6139
    """Remove the tag from the object.
6140

6141
    """
6142
    for tag in self.op.tags:
6143
      self.target.RemoveTag(tag)
6144
    try:
6145
      self.cfg.Update(self.target)
6146
    except errors.ConfigurationError:
6147
      raise errors.OpRetryError("There has been a modification to the"
6148
                                " config file and the operation has been"
6149
                                " aborted. Please retry.")
6150

    
6151

    
6152
class LUTestDelay(NoHooksLU):
6153
  """Sleep for a specified amount of time.
6154

6155
  This LU sleeps on the master and/or nodes for a specified amount of
6156
  time.
6157

6158
  """
6159
  _OP_REQP = ["duration", "on_master", "on_nodes"]
6160
  REQ_BGL = False
6161

    
6162
  def ExpandNames(self):
6163
    """Expand names and set required locks.
6164

6165
    This expands the node list, if any.
6166

6167
    """
6168
    self.needed_locks = {}
6169
    if self.op.on_nodes:
6170
      # _GetWantedNodes can be used here, but is not always appropriate to use
6171
      # this way in ExpandNames. Check LogicalUnit.ExpandNames docstring for
6172
      # more information.
6173
      self.op.on_nodes = _GetWantedNodes(self, self.op.on_nodes)
6174
      self.needed_locks[locking.LEVEL_NODE] = self.op.on_nodes
6175

    
6176
  def CheckPrereq(self):
6177
    """Check prerequisites.
6178

6179
    """
6180

    
6181
  def Exec(self, feedback_fn):
6182
    """Do the actual sleep.
6183

6184
    """
6185
    if self.op.on_master:
6186
      if not utils.TestDelay(self.op.duration):
6187
        raise errors.OpExecError("Error during master delay test")
6188
    if self.op.on_nodes:
6189
      result = self.rpc.call_test_delay(self.op.on_nodes, self.op.duration)
6190
      if not result:
6191
        raise errors.OpExecError("Complete failure from rpc call")
6192
      for node, node_result in result.items():
6193
        node_result.Raise()
6194
        if not node_result.data:
6195
          raise errors.OpExecError("Failure during rpc call to node %s,"
6196
                                   " result: %s" % (node, node_result.data))
6197

    
6198

    
6199
class IAllocator(object):
6200
  """IAllocator framework.
6201

6202
  An IAllocator instance has three sets of attributes:
6203
    - cfg that is needed to query the cluster
6204
    - input data (all members of the _KEYS class attribute are required)
6205
    - four buffer attributes (in|out_data|text), that represent the
6206
      input (to the external script) in text and data structure format,
6207
      and the output from it, again in two formats
6208
    - the result variables from the script (success, info, nodes) for
6209
      easy usage
6210

6211
  """
6212
  _ALLO_KEYS = [
6213
    "mem_size", "disks", "disk_template",
6214
    "os", "tags", "nics", "vcpus", "hypervisor",
6215
    ]
6216
  _RELO_KEYS = [
6217
    "relocate_from",
6218
    ]
6219

    
6220
  def __init__(self, lu, mode, name, **kwargs):
6221
    self.lu = lu
6222
    # init buffer variables
6223
    self.in_text = self.out_text = self.in_data = self.out_data = None
6224
    # init all input fields so that pylint is happy
6225
    self.mode = mode
6226
    self.name = name
6227
    self.mem_size = self.disks = self.disk_template = None
6228
    self.os = self.tags = self.nics = self.vcpus = None
6229
    self.hypervisor = None
6230
    self.relocate_from = None
6231
    # computed fields
6232
    self.required_nodes = None
6233
    # init result fields
6234
    self.success = self.info = self.nodes = None
6235
    if self.mode == constants.IALLOCATOR_MODE_ALLOC:
6236
      keyset = self._ALLO_KEYS
6237
    elif self.mode == constants.IALLOCATOR_MODE_RELOC:
6238
      keyset = self._RELO_KEYS
6239
    else:
6240
      raise errors.ProgrammerError("Unknown mode '%s' passed to the"
6241
                                   " IAllocator" % self.mode)
6242
    for key in kwargs:
6243
      if key not in keyset:
6244
        raise errors.ProgrammerError("Invalid input parameter '%s' to"
6245
                                     " IAllocator" % key)
6246
      setattr(self, key, kwargs[key])
6247
    for key in keyset:
6248
      if key not in kwargs:
6249
        raise errors.ProgrammerError("Missing input parameter '%s' to"
6250
                                     " IAllocator" % key)
6251
    self._BuildInputData()
6252

    
6253
  def _ComputeClusterData(self):
6254
    """Compute the generic allocator input data.
6255

6256
    This is the data that is independent of the actual operation.
6257

6258
    """
6259
    cfg = self.lu.cfg
6260
    cluster_info = cfg.GetClusterInfo()
6261
    # cluster data
6262
    data = {
6263
      "version": 1,
6264
      "cluster_name": cfg.GetClusterName(),
6265
      "cluster_tags": list(cluster_info.GetTags()),
6266
      "enable_hypervisors": list(cluster_info.enabled_hypervisors),
6267
      # we don't have job IDs
6268
      }
6269
    iinfo = cfg.GetAllInstancesInfo().values()
6270
    i_list = [(inst, cluster_info.FillBE(inst)) for inst in iinfo]
6271

    
6272
    # node data
6273
    node_results = {}
6274
    node_list = cfg.GetNodeList()
6275

    
6276
    if self.mode == constants.IALLOCATOR_MODE_ALLOC:
6277
      hypervisor_name = self.hypervisor
6278
    elif self.mode == constants.IALLOCATOR_MODE_RELOC:
6279
      hypervisor_name = cfg.GetInstanceInfo(self.name).hypervisor
6280

    
6281
    node_data = self.lu.rpc.call_node_info(node_list, cfg.GetVGName(),
6282
                                           hypervisor_name)
6283
    node_iinfo = self.lu.rpc.call_all_instances_info(node_list,
6284
                       cluster_info.enabled_hypervisors)
6285
    for nname in node_list:
6286
      ninfo = cfg.GetNodeInfo(nname)
6287
      node_data[nname].Raise()
6288
      if not isinstance(node_data[nname].data, dict):
6289
        raise errors.OpExecError("Can't get data for node %s" % nname)
6290
      remote_info = node_data[nname].data
6291
      for attr in ['memory_total', 'memory_free', 'memory_dom0',
6292
                   'vg_size', 'vg_free', 'cpu_total']:
6293
        if attr not in remote_info:
6294
          raise errors.OpExecError("Node '%s' didn't return attribute '%s'" %
6295
                                   (nname, attr))
6296
        try:
6297
          remote_info[attr] = int(remote_info[attr])
6298
        except ValueError, err:
6299
          raise errors.OpExecError("Node '%s' returned invalid value for '%s':"
6300
                                   " %s" % (nname, attr, str(err)))
6301
      # compute memory used by primary instances
6302
      i_p_mem = i_p_up_mem = 0
6303
      for iinfo, beinfo in i_list:
6304
        if iinfo.primary_node == nname:
6305
          i_p_mem += beinfo[constants.BE_MEMORY]
6306
          if iinfo.name not in node_iinfo[nname]:
6307
            i_used_mem = 0
6308
          else:
6309
            i_used_mem = int(node_iinfo[nname][iinfo.name]['memory'])
6310
          i_mem_diff = beinfo[constants.BE_MEMORY] - i_used_mem
6311
          remote_info['memory_free'] -= max(0, i_mem_diff)
6312

    
6313
          if iinfo.status == "up":
6314
            i_p_up_mem += beinfo[constants.BE_MEMORY]
6315

    
6316
      # compute memory used by instances
6317
      pnr = {
6318
        "tags": list(ninfo.GetTags()),
6319
        "total_memory": remote_info['memory_total'],
6320
        "reserved_memory": remote_info['memory_dom0'],
6321
        "free_memory": remote_info['memory_free'],
6322
        "i_pri_memory": i_p_mem,
6323
        "i_pri_up_memory": i_p_up_mem,
6324
        "total_disk": remote_info['vg_size'],
6325
        "free_disk": remote_info['vg_free'],
6326
        "primary_ip": ninfo.primary_ip,
6327
        "secondary_ip": ninfo.secondary_ip,
6328
        "total_cpus": remote_info['cpu_total'],
6329
        "offline": ninfo.offline,
6330
        }
6331
      node_results[nname] = pnr
6332
    data["nodes"] = node_results
6333

    
6334
    # instance data
6335
    instance_data = {}
6336
    for iinfo, beinfo in i_list:
6337
      nic_data = [{"mac": n.mac, "ip": n.ip, "bridge": n.bridge}
6338
                  for n in iinfo.nics]
6339
      pir = {
6340
        "tags": list(iinfo.GetTags()),
6341
        "should_run": iinfo.status == "up",
6342
        "vcpus": beinfo[constants.BE_VCPUS],
6343
        "memory": beinfo[constants.BE_MEMORY],
6344
        "os": iinfo.os,
6345
        "nodes": [iinfo.primary_node] + list(iinfo.secondary_nodes),
6346
        "nics": nic_data,
6347
        "disks": [{"size": dsk.size, "mode": "w"} for dsk in iinfo.disks],
6348
        "disk_template": iinfo.disk_template,
6349
        "hypervisor": iinfo.hypervisor,
6350
        }
6351
      instance_data[iinfo.name] = pir
6352

    
6353
    data["instances"] = instance_data
6354

    
6355
    self.in_data = data
6356

    
6357
  def _AddNewInstance(self):
6358
    """Add new instance data to allocator structure.
6359

6360
    This in combination with _AllocatorGetClusterData will create the
6361
    correct structure needed as input for the allocator.
6362

6363
    The checks for the completeness of the opcode must have already been
6364
    done.
6365

6366
    """
6367
    data = self.in_data
6368
    if len(self.disks) != 2:
6369
      raise errors.OpExecError("Only two-disk configurations supported")
6370

    
6371
    disk_space = _ComputeDiskSize(self.disk_template, self.disks)
6372

    
6373
    if self.disk_template in constants.DTS_NET_MIRROR:
6374
      self.required_nodes = 2
6375
    else:
6376
      self.required_nodes = 1
6377
    request = {
6378
      "type": "allocate",
6379
      "name": self.name,
6380
      "disk_template": self.disk_template,
6381
      "tags": self.tags,
6382
      "os": self.os,
6383
      "vcpus": self.vcpus,
6384
      "memory": self.mem_size,
6385
      "disks": self.disks,
6386
      "disk_space_total": disk_space,
6387
      "nics": self.nics,
6388
      "required_nodes": self.required_nodes,
6389
      }
6390
    data["request"] = request
6391

    
6392
  def _AddRelocateInstance(self):
6393
    """Add relocate instance data to allocator structure.
6394

6395
    This in combination with _IAllocatorGetClusterData will create the
6396
    correct structure needed as input for the allocator.
6397

6398
    The checks for the completeness of the opcode must have already been
6399
    done.
6400

6401
    """
6402
    instance = self.lu.cfg.GetInstanceInfo(self.name)
6403
    if instance is None:
6404
      raise errors.ProgrammerError("Unknown instance '%s' passed to"
6405
                                   " IAllocator" % self.name)
6406

    
6407
    if instance.disk_template not in constants.DTS_NET_MIRROR:
6408
      raise errors.OpPrereqError("Can't relocate non-mirrored instances")
6409

    
6410
    if len(instance.secondary_nodes) != 1:
6411
      raise errors.OpPrereqError("Instance has not exactly one secondary node")
6412

    
6413
    self.required_nodes = 1
6414
    disk_sizes = [{'size': disk.size} for disk in instance.disks]
6415
    disk_space = _ComputeDiskSize(instance.disk_template, disk_sizes)
6416

    
6417
    request = {
6418
      "type": "relocate",
6419
      "name": self.name,
6420
      "disk_space_total": disk_space,
6421
      "required_nodes": self.required_nodes,
6422
      "relocate_from": self.relocate_from,
6423
      }
6424
    self.in_data["request"] = request
6425

    
6426
  def _BuildInputData(self):
6427
    """Build input data structures.
6428

6429
    """
6430
    self._ComputeClusterData()
6431

    
6432
    if self.mode == constants.IALLOCATOR_MODE_ALLOC:
6433
      self._AddNewInstance()
6434
    else:
6435
      self._AddRelocateInstance()
6436

    
6437
    self.in_text = serializer.Dump(self.in_data)
6438

    
6439
  def Run(self, name, validate=True, call_fn=None):
6440
    """Run an instance allocator and return the results.
6441

6442
    """
6443
    if call_fn is None:
6444
      call_fn = self.lu.rpc.call_iallocator_runner
6445
    data = self.in_text
6446

    
6447
    result = call_fn(self.lu.cfg.GetMasterNode(), name, self.in_text)
6448
    result.Raise()
6449

    
6450
    if not isinstance(result.data, (list, tuple)) or len(result.data) != 4:
6451
      raise errors.OpExecError("Invalid result from master iallocator runner")
6452

    
6453
    rcode, stdout, stderr, fail = result.data
6454

    
6455
    if rcode == constants.IARUN_NOTFOUND:
6456
      raise errors.OpExecError("Can't find allocator '%s'" % name)
6457
    elif rcode == constants.IARUN_FAILURE:
6458
      raise errors.OpExecError("Instance allocator call failed: %s,"
6459
                               " output: %s" % (fail, stdout+stderr))
6460
    self.out_text = stdout
6461
    if validate:
6462
      self._ValidateResult()
6463

    
6464
  def _ValidateResult(self):
6465
    """Process the allocator results.
6466

6467
    This will process and if successful save the result in
6468
    self.out_data and the other parameters.
6469

6470
    """
6471
    try:
6472
      rdict = serializer.Load(self.out_text)
6473
    except Exception, err:
6474
      raise errors.OpExecError("Can't parse iallocator results: %s" % str(err))
6475

    
6476
    if not isinstance(rdict, dict):
6477
      raise errors.OpExecError("Can't parse iallocator results: not a dict")
6478

    
6479
    for key in "success", "info", "nodes":
6480
      if key not in rdict:
6481
        raise errors.OpExecError("Can't parse iallocator results:"
6482
                                 " missing key '%s'" % key)
6483
      setattr(self, key, rdict[key])
6484

    
6485
    if not isinstance(rdict["nodes"], list):
6486
      raise errors.OpExecError("Can't parse iallocator results: 'nodes' key"
6487
                               " is not a list")
6488
    self.out_data = rdict
6489

    
6490

    
6491
class LUTestAllocator(NoHooksLU):
6492
  """Run allocator tests.
6493

6494
  This LU runs the allocator tests
6495

6496
  """
6497
  _OP_REQP = ["direction", "mode", "name"]
6498

    
6499
  def CheckPrereq(self):
6500
    """Check prerequisites.
6501

6502
    This checks the opcode parameters depending on the director and mode test.
6503

6504
    """
6505
    if self.op.mode == constants.IALLOCATOR_MODE_ALLOC:
6506
      for attr in ["name", "mem_size", "disks", "disk_template",
6507
                   "os", "tags", "nics", "vcpus"]:
6508
        if not hasattr(self.op, attr):
6509
          raise errors.OpPrereqError("Missing attribute '%s' on opcode input" %
6510
                                     attr)
6511
      iname = self.cfg.ExpandInstanceName(self.op.name)
6512
      if iname is not None:
6513
        raise errors.OpPrereqError("Instance '%s' already in the cluster" %
6514
                                   iname)
6515
      if not isinstance(self.op.nics, list):
6516
        raise errors.OpPrereqError("Invalid parameter 'nics'")
6517
      for row in self.op.nics:
6518
        if (not isinstance(row, dict) or
6519
            "mac" not in row or
6520
            "ip" not in row or
6521
            "bridge" not in row):
6522
          raise errors.OpPrereqError("Invalid contents of the"
6523
                                     " 'nics' parameter")
6524
      if not isinstance(self.op.disks, list):
6525
        raise errors.OpPrereqError("Invalid parameter 'disks'")
6526
      if len(self.op.disks) != 2:
6527
        raise errors.OpPrereqError("Only two-disk configurations supported")
6528
      for row in self.op.disks:
6529
        if (not isinstance(row, dict) or
6530
            "size" not in row or
6531
            not isinstance(row["size"], int) or
6532
            "mode" not in row or
6533
            row["mode"] not in ['r', 'w']):
6534
          raise errors.OpPrereqError("Invalid contents of the"
6535
                                     " 'disks' parameter")
6536
      if self.op.hypervisor is None:
6537
        self.op.hypervisor = self.cfg.GetHypervisorType()
6538
    elif self.op.mode == constants.IALLOCATOR_MODE_RELOC:
6539
      if not hasattr(self.op, "name"):
6540
        raise errors.OpPrereqError("Missing attribute 'name' on opcode input")
6541
      fname = self.cfg.ExpandInstanceName(self.op.name)
6542
      if fname is None:
6543
        raise errors.OpPrereqError("Instance '%s' not found for relocation" %
6544
                                   self.op.name)
6545
      self.op.name = fname
6546
      self.relocate_from = self.cfg.GetInstanceInfo(fname).secondary_nodes
6547
    else:
6548
      raise errors.OpPrereqError("Invalid test allocator mode '%s'" %
6549
                                 self.op.mode)
6550

    
6551
    if self.op.direction == constants.IALLOCATOR_DIR_OUT:
6552
      if not hasattr(self.op, "allocator") or self.op.allocator is None:
6553
        raise errors.OpPrereqError("Missing allocator name")
6554
    elif self.op.direction != constants.IALLOCATOR_DIR_IN:
6555
      raise errors.OpPrereqError("Wrong allocator test '%s'" %
6556
                                 self.op.direction)
6557

    
6558
  def Exec(self, feedback_fn):
6559
    """Run the allocator test.
6560

6561
    """
6562
    if self.op.mode == constants.IALLOCATOR_MODE_ALLOC:
6563
      ial = IAllocator(self,
6564
                       mode=self.op.mode,
6565
                       name=self.op.name,
6566
                       mem_size=self.op.mem_size,
6567
                       disks=self.op.disks,
6568
                       disk_template=self.op.disk_template,
6569
                       os=self.op.os,
6570
                       tags=self.op.tags,
6571
                       nics=self.op.nics,
6572
                       vcpus=self.op.vcpus,
6573
                       hypervisor=self.op.hypervisor,
6574
                       )
6575
    else:
6576
      ial = IAllocator(self,
6577
                       mode=self.op.mode,
6578
                       name=self.op.name,
6579
                       relocate_from=list(self.relocate_from),
6580
                       )
6581

    
6582
    if self.op.direction == constants.IALLOCATOR_DIR_IN:
6583
      result = ial.in_text
6584
    else:
6585
      ial.Run(self.op.allocator, validate=False)
6586
      result = ial.out_text
6587
    return result