Synnefo » snf-ganeti

  _ComputeIPolicySpecViolation

def _CheckInstancesNodeGroups(cfg, instances, owned_groups, owned_nodes,

                              cur_group_uuid):

  """Checks if node groups for locked instances are still correct.

  @type cfg: L{config.ConfigWriter}

  @param cfg: Cluster configuration

  @type instances: dict; string as key, L{objects.Instance} as value

  @param instances: Dictionary, instance name as key, instance object as value

  @type owned_groups: iterable of string

  @param owned_groups: List of owned groups

  @type owned_nodes: iterable of string

  @param owned_nodes: List of owned nodes

  @type cur_group_uuid: string or None

  @param cur_group_uuid: Optional group UUID to check against instance's groups

  """

  for (name, inst) in instances.items():

    assert owned_nodes.issuperset(inst.all_nodes), \

      "Instance %s's nodes changed while we kept the lock" % name

    inst_groups = _CheckInstanceNodeGroups(cfg, name, owned_groups)

    assert cur_group_uuid is None or cur_group_uuid in inst_groups, \

      "Instance %s has no node in group %s" % (name, cur_group_uuid)

def _CheckInstanceNodeGroups(cfg, instance_name, owned_groups,

                             primary_only=False):

  """Checks if the owned node groups are still correct for an instance.

  @type cfg: L{config.ConfigWriter}

  @param cfg: The cluster configuration

  @type instance_name: string

  @param instance_name: Instance name

  @type owned_groups: set or frozenset

  @param owned_groups: List of currently owned node groups

  @type primary_only: boolean

  @param primary_only: Whether to check node groups for only the primary node

  """

  inst_groups = cfg.GetInstanceNodeGroups(instance_name, primary_only)

  if not owned_groups.issuperset(inst_groups):

    raise errors.OpPrereqError("Instance %s's node groups changed since"

                               " locks were acquired, current groups are"

                               " are '%s', owning groups '%s'; retry the"

                               " operation" %

                               (instance_name,

                                utils.CommaJoin(inst_groups),

                                utils.CommaJoin(owned_groups)),

                               errors.ECODE_STATE)

  return inst_groups

def _MapInstanceDisksToNodes(instances):

  """Creates a map from (node, volume) to instance name.

  @type instances: list of L{objects.Instance}

  @rtype: dict; tuple of (node name, volume name) as key, instance name as value

  """

  return dict(((node, vol), inst.name)

              for inst in instances

              for (node, vols) in inst.MapLVsByNode().items()

              for vol in vols)

def _GetDefaultIAllocator(cfg, ialloc):

  """Decides on which iallocator to use.

  @type cfg: L{config.ConfigWriter}

  @param cfg: Cluster configuration object

  @type ialloc: string or None

  @param ialloc: Iallocator specified in opcode

  @rtype: string

  @return: Iallocator name

  """

  if not ialloc:

    # Use default iallocator

    ialloc = cfg.GetDefaultIAllocator()

  if not ialloc:

    raise errors.OpPrereqError("No iallocator was specified, neither in the"

                               " opcode nor as a cluster-wide default",

                               errors.ECODE_INVAL)

  return ialloc

class LUGroupVerifyDisks(NoHooksLU):

  """Verifies the status of all disks in a node group.

  """

  REQ_BGL = False

  def ExpandNames(self):

    # Raises errors.OpPrereqError on its own if group can't be found

    self.group_uuid = self.cfg.LookupNodeGroup(self.op.group_name)

    self.share_locks = _ShareAll()

    self.needed_locks = {

      locking.LEVEL_INSTANCE: [],

      locking.LEVEL_NODEGROUP: [],

      locking.LEVEL_NODE: [],

      # This opcode is acquires all node locks in a group. LUClusterVerifyDisks

      # starts one instance of this opcode for every group, which means all

      # nodes will be locked for a short amount of time, so it's better to

      # acquire the node allocation lock as well.

      locking.LEVEL_NODE_ALLOC: locking.ALL_SET,

      }

  def DeclareLocks(self, level):

    if level == locking.LEVEL_INSTANCE:

      assert not self.needed_locks[locking.LEVEL_INSTANCE]

      # Lock instances optimistically, needs verification once node and group

      # locks have been acquired

      self.needed_locks[locking.LEVEL_INSTANCE] = \

        self.cfg.GetNodeGroupInstances(self.group_uuid)

    elif level == locking.LEVEL_NODEGROUP:

      assert not self.needed_locks[locking.LEVEL_NODEGROUP]

      self.needed_locks[locking.LEVEL_NODEGROUP] = \

        set([self.group_uuid] +

            # Lock all groups used by instances optimistically; this requires

            # going via the node before it's locked, requiring verification

            # later on

            [group_uuid

             for instance_name in self.owned_locks(locking.LEVEL_INSTANCE)

             for group_uuid in self.cfg.GetInstanceNodeGroups(instance_name)])

    elif level == locking.LEVEL_NODE:

      # This will only lock the nodes in the group to be verified which contain

      # actual instances

      self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_APPEND

      self._LockInstancesNodes()

      # Lock all nodes in group to be verified

      assert self.group_uuid in self.owned_locks(locking.LEVEL_NODEGROUP)

      member_nodes = self.cfg.GetNodeGroup(self.group_uuid).members

      self.needed_locks[locking.LEVEL_NODE].extend(member_nodes)

  def CheckPrereq(self):

    owned_instances = frozenset(self.owned_locks(locking.LEVEL_INSTANCE))

    owned_groups = frozenset(self.owned_locks(locking.LEVEL_NODEGROUP))

    owned_nodes = frozenset(self.owned_locks(locking.LEVEL_NODE))

    assert self.group_uuid in owned_groups

    # Check if locked instances are still correct

    _CheckNodeGroupInstances(self.cfg, self.group_uuid, owned_instances)

    # Get instance information

    self.instances = dict(self.cfg.GetMultiInstanceInfo(owned_instances))

    # Check if node groups for locked instances are still correct

    _CheckInstancesNodeGroups(self.cfg, self.instances,

                              owned_groups, owned_nodes, self.group_uuid)

  def Exec(self, feedback_fn):

    """Verify integrity of cluster disks.

    @rtype: tuple of three items

    @return: a tuple of (dict of node-to-node_error, list of instances

        which need activate-disks, dict of instance: (node, volume) for

        missing volumes

    """

    res_nodes = {}

    res_instances = set()

    res_missing = {}

    nv_dict = _MapInstanceDisksToNodes(

      [inst for inst in self.instances.values()

       if inst.admin_state == constants.ADMINST_UP])

    if nv_dict:

      nodes = utils.NiceSort(set(self.owned_locks(locking.LEVEL_NODE)) &

                             set(self.cfg.GetVmCapableNodeList()))

      node_lvs = self.rpc.call_lv_list(nodes, [])

      for (node, node_res) in node_lvs.items():

        if node_res.offline:

          continue

        msg = node_res.fail_msg

        if msg:

          logging.warning("Error enumerating LVs on node %s: %s", node, msg)

          res_nodes[node] = msg

          continue

        for lv_name, (_, _, lv_online) in node_res.payload.items():

          inst = nv_dict.pop((node, lv_name), None)

          if not (lv_online or inst is None):

            res_instances.add(inst)

      # any leftover items in nv_dict are missing LVs, let's arrange the data

      # better

      for key, inst in nv_dict.iteritems():

        res_missing.setdefault(inst, []).append(list(key))

    return (res_nodes, list(res_instances), res_missing)

def _SetOpEarlyRelease(early_release, op):

  """Sets C{early_release} flag on opcodes if available.

  """

  try:

    op.early_release = early_release

  except AttributeError:

    assert not isinstance(op, opcodes.OpInstanceReplaceDisks)

  return op

def _NodeEvacDest(use_nodes, group, nodes):

  """Returns group or nodes depending on caller's choice.

  """

  if use_nodes:

    return utils.CommaJoin(nodes)

  else:

    return group

def _LoadNodeEvacResult(lu, alloc_result, early_release, use_nodes):

  """Unpacks the result of change-group and node-evacuate iallocator requests.

  Iallocator modes L{constants.IALLOCATOR_MODE_NODE_EVAC} and

  L{constants.IALLOCATOR_MODE_CHG_GROUP}.

  @type lu: L{LogicalUnit}

  @param lu: Logical unit instance

  @type alloc_result: tuple/list

  @param alloc_result: Result from iallocator

  @type early_release: bool

  @param early_release: Whether to release locks early if possible

  @type use_nodes: bool

  @param use_nodes: Whether to display node names instead of groups

  """

  (moved, failed, jobs) = alloc_result

  if failed:

    failreason = utils.CommaJoin("%s (%s)" % (name, reason)

                                 for (name, reason) in failed)

    lu.LogWarning("Unable to evacuate instances %s", failreason)

    raise errors.OpExecError("Unable to evacuate instances %s" % failreason)

  if moved:

    lu.LogInfo("Instances to be moved: %s",

               utils.CommaJoin("%s (to %s)" %

                               (name, _NodeEvacDest(use_nodes, group, nodes))

                               for (name, group, nodes) in moved))

  return [map(compat.partial(_SetOpEarlyRelease, early_release),

              map(opcodes.OpCode.LoadOpCode, ops))

          for ops in jobs]

class LUGroupAdd(LogicalUnit):

  """Logical unit for creating node groups.

  """

  HPATH = "group-add"

  HTYPE = constants.HTYPE_GROUP

  REQ_BGL = False

  def ExpandNames(self):

    # We need the new group's UUID here so that we can create and acquire the

    # corresponding lock. Later, in Exec(), we'll indicate to cfg.AddNodeGroup

    # that it should not check whether the UUID exists in the configuration.

    self.group_uuid = self.cfg.GenerateUniqueID(self.proc.GetECId())

    self.needed_locks = {}

    self.add_locks[locking.LEVEL_NODEGROUP] = self.group_uuid

  def CheckPrereq(self):

    """Check prerequisites.

    This checks that the given group name is not an existing node group

    already.

    """

    try:

      existing_uuid = self.cfg.LookupNodeGroup(self.op.group_name)

    except errors.OpPrereqError:

      pass

    else:

      raise errors.OpPrereqError("Desired group name '%s' already exists as a"

                                 " node group (UUID: %s)" %

                                 (self.op.group_name, existing_uuid),

                                 errors.ECODE_EXISTS)

    if self.op.ndparams:

      utils.ForceDictType(self.op.ndparams, constants.NDS_PARAMETER_TYPES)

    if self.op.hv_state:

      self.new_hv_state = _MergeAndVerifyHvState(self.op.hv_state, None)

    else:

      self.new_hv_state = None

    if self.op.disk_state:

      self.new_disk_state = _MergeAndVerifyDiskState(self.op.disk_state, None)

    else:

      self.new_disk_state = None

    if self.op.diskparams:

      for templ in constants.DISK_TEMPLATES:

        if templ in self.op.diskparams:

          utils.ForceDictType(self.op.diskparams[templ],

                              constants.DISK_DT_TYPES)

      self.new_diskparams = self.op.diskparams

      try:

        utils.VerifyDictOptions(self.new_diskparams, constants.DISK_DT_DEFAULTS)

      except errors.OpPrereqError, err:

        raise errors.OpPrereqError("While verify diskparams options: %s" % err,

                                   errors.ECODE_INVAL)

    else:

      self.new_diskparams = {}

    if self.op.ipolicy:

      cluster = self.cfg.GetClusterInfo()

      full_ipolicy = cluster.SimpleFillIPolicy(self.op.ipolicy)

      try:

        objects.InstancePolicy.CheckParameterSyntax(full_ipolicy, False)

      except errors.ConfigurationError, err:

        raise errors.OpPrereqError("Invalid instance policy: %s" % err,

                                   errors.ECODE_INVAL)

  def BuildHooksEnv(self):

    """Build hooks env.

    """

    return {

      "GROUP_NAME": self.op.group_name,

      }

  def BuildHooksNodes(self):

    """Build hooks nodes.

    """

    mn = self.cfg.GetMasterNode()

    return ([mn], [mn])

  def Exec(self, feedback_fn):

    """Add the node group to the cluster.

    """

    group_obj = objects.NodeGroup(name=self.op.group_name, members=[],

                                  uuid=self.group_uuid,

                                  alloc_policy=self.op.alloc_policy,

                                  ndparams=self.op.ndparams,

                                  diskparams=self.new_diskparams,

                                  ipolicy=self.op.ipolicy,

                                  hv_state_static=self.new_hv_state,

                                  disk_state_static=self.new_disk_state)

    self.cfg.AddNodeGroup(group_obj, self.proc.GetECId(), check_uuid=False)

    del self.remove_locks[locking.LEVEL_NODEGROUP]

class LUGroupAssignNodes(NoHooksLU):

  """Logical unit for assigning nodes to groups.

  """

  REQ_BGL = False

  def ExpandNames(self):

    # These raise errors.OpPrereqError on their own:

    self.group_uuid = self.cfg.LookupNodeGroup(self.op.group_name)

    self.op.nodes = _GetWantedNodes(self, self.op.nodes)

    # We want to lock all the affected nodes and groups. We have readily

    # available the list of nodes, and the *destination* group. To gather the

    # list of "source" groups, we need to fetch node information later on.

    self.needed_locks = {

      locking.LEVEL_NODEGROUP: set([self.group_uuid]),

      locking.LEVEL_NODE: self.op.nodes,

      }

  def DeclareLocks(self, level):

    if level == locking.LEVEL_NODEGROUP:

      assert len(self.needed_locks[locking.LEVEL_NODEGROUP]) == 1

      # Try to get all affected nodes' groups without having the group or node

      # lock yet. Needs verification later in the code flow.

      groups = self.cfg.GetNodeGroupsFromNodes(self.op.nodes)

      self.needed_locks[locking.LEVEL_NODEGROUP].update(groups)

  def CheckPrereq(self):

    """Check prerequisites.

    """

    assert self.needed_locks[locking.LEVEL_NODEGROUP]

    assert (frozenset(self.owned_locks(locking.LEVEL_NODE)) ==

            frozenset(self.op.nodes))

    expected_locks = (set([self.group_uuid]) |

                      self.cfg.GetNodeGroupsFromNodes(self.op.nodes))

    actual_locks = self.owned_locks(locking.LEVEL_NODEGROUP)

    if actual_locks != expected_locks:

      raise errors.OpExecError("Nodes changed groups since locks were acquired,"

                               " current groups are '%s', used to be '%s'" %

                               (utils.CommaJoin(expected_locks),

                                utils.CommaJoin(actual_locks)))

    self.node_data = self.cfg.GetAllNodesInfo()

    self.group = self.cfg.GetNodeGroup(self.group_uuid)

    instance_data = self.cfg.GetAllInstancesInfo()

    if self.group is None:

      raise errors.OpExecError("Could not retrieve group '%s' (UUID: %s)" %

                               (self.op.group_name, self.group_uuid))

    (new_splits, previous_splits) = \

      self.CheckAssignmentForSplitInstances([(node, self.group_uuid)

                                             for node in self.op.nodes],

                                            self.node_data, instance_data)

    if new_splits:

      fmt_new_splits = utils.CommaJoin(utils.NiceSort(new_splits))

      if not self.op.force:

        raise errors.OpExecError("The following instances get split by this"

                                 " change and --force was not given: %s" %

                                 fmt_new_splits)

      else:

        self.LogWarning("This operation will split the following instances: %s",

                        fmt_new_splits)

        if previous_splits:

          self.LogWarning("In addition, these already-split instances continue"

                          " to be split across groups: %s",

                          utils.CommaJoin(utils.NiceSort(previous_splits)))

  def Exec(self, feedback_fn):

    """Assign nodes to a new group.

    """

    mods = [(node_name, self.group_uuid) for node_name in self.op.nodes]

    self.cfg.AssignGroupNodes(mods)

  @staticmethod

  def CheckAssignmentForSplitInstances(changes, node_data, instance_data):

    """Check for split instances after a node assignment.

    This method considers a series of node assignments as an atomic operation,

    and returns information about split instances after applying the set of

    changes.

    In particular, it returns information about newly split instances, and

    instances that were already split, and remain so after the change.

    Only instances whose disk template is listed in constants.DTS_INT_MIRROR are

    considered.

    @type changes: list of (node_name, new_group_uuid) pairs.

    @param changes: list of node assignments to consider.

    @param node_data: a dict with data for all nodes

    @param instance_data: a dict with all instances to consider

    @rtype: a two-tuple

    @return: a list of instances that were previously okay and result split as a

      consequence of this change, and a list of instances that were previously

      split and this change does not fix.

    """

    changed_nodes = dict((node, group) for node, group in changes

                         if node_data[node].group != group)

    all_split_instances = set()

    previously_split_instances = set()

    def InstanceNodes(instance):

      return [instance.primary_node] + list(instance.secondary_nodes)

    for inst in instance_data.values():

      if inst.disk_template not in constants.DTS_INT_MIRROR:

        continue

      instance_nodes = InstanceNodes(inst)

      if len(set(node_data[node].group for node in instance_nodes)) > 1:

        previously_split_instances.add(inst.name)

      if len(set(changed_nodes.get(node, node_data[node].group)

                 for node in instance_nodes)) > 1:

        all_split_instances.add(inst.name)

    return (list(all_split_instances - previously_split_instances),

            list(previously_split_instances & all_split_instances))

class _GroupQuery(_QueryBase):

  FIELDS = query.GROUP_FIELDS

  def ExpandNames(self, lu):

    lu.needed_locks = {}

    self._all_groups = lu.cfg.GetAllNodeGroupsInfo()

    self._cluster = lu.cfg.GetClusterInfo()

    name_to_uuid = dict((g.name, g.uuid) for g in self._all_groups.values())

    if not self.names:

      self.wanted = [name_to_uuid[name]

                     for name in utils.NiceSort(name_to_uuid.keys())]

    else:

      # Accept names to be either names or UUIDs.

      missing = []

      self.wanted = []

      all_uuid = frozenset(self._all_groups.keys())

      for name in self.names:

        if name in all_uuid:

          self.wanted.append(name)

        elif name in name_to_uuid:

          self.wanted.append(name_to_uuid[name])

        else:

          missing.append(name)

      if missing:

        raise errors.OpPrereqError("Some groups do not exist: %s" %

                                   utils.CommaJoin(missing),

                                   errors.ECODE_NOENT)

  def DeclareLocks(self, lu, level):

    pass

  def _GetQueryData(self, lu):

    """Computes the list of node groups and their attributes.

    """

    do_nodes = query.GQ_NODE in self.requested_data

    do_instances = query.GQ_INST in self.requested_data

    group_to_nodes = None

    group_to_instances = None

    # For GQ_NODE, we need to map group->[nodes], and group->[instances] for

    # GQ_INST. The former is attainable with just GetAllNodesInfo(), but for the

    # latter GetAllInstancesInfo() is not enough, for we have to go through

    # instance->node. Hence, we will need to process nodes even if we only need

    # instance information.

    if do_nodes or do_instances:

      all_nodes = lu.cfg.GetAllNodesInfo()

      group_to_nodes = dict((uuid, []) for uuid in self.wanted)

      node_to_group = {}

      for node in all_nodes.values():

        if node.group in group_to_nodes:

          group_to_nodes[node.group].append(node.name)

          node_to_group[node.name] = node.group

      if do_instances:

        all_instances = lu.cfg.GetAllInstancesInfo()

        group_to_instances = dict((uuid, []) for uuid in self.wanted)

        for instance in all_instances.values():

          node = instance.primary_node

          if node in node_to_group:

            group_to_instances[node_to_group[node]].append(instance.name)

        if not do_nodes:

          # Do not pass on node information if it was not requested.

          group_to_nodes = None

    return query.GroupQueryData(self._cluster,

                                [self._all_groups[uuid]

                                 for uuid in self.wanted],

                                group_to_nodes, group_to_instances,

                                query.GQ_DISKPARAMS in self.requested_data)

class LUGroupQuery(NoHooksLU):

  """Logical unit for querying node groups.

  """

  REQ_BGL = False

  def CheckArguments(self):

    self.gq = _GroupQuery(qlang.MakeSimpleFilter("name", self.op.names),

                          self.op.output_fields, False)

  def ExpandNames(self):

    self.gq.ExpandNames(self)

  def DeclareLocks(self, level):

    self.gq.DeclareLocks(self, level)

  def Exec(self, feedback_fn):

    return self.gq.OldStyleQuery(self)

class LUGroupSetParams(LogicalUnit):

  """Modifies the parameters of a node group.

  """

  HPATH = "group-modify"

  HTYPE = constants.HTYPE_GROUP

  REQ_BGL = False

  def CheckArguments(self):

    all_changes = [

      self.op.ndparams,

      self.op.diskparams,

      self.op.alloc_policy,

      self.op.hv_state,

      self.op.disk_state,

      self.op.ipolicy,

      ]

    if all_changes.count(None) == len(all_changes):

      raise errors.OpPrereqError("Please pass at least one modification",

                                 errors.ECODE_INVAL)

  def ExpandNames(self):

    # This raises errors.OpPrereqError on its own:

    self.group_uuid = self.cfg.LookupNodeGroup(self.op.group_name)

    self.needed_locks = {

      locking.LEVEL_INSTANCE: [],

      locking.LEVEL_NODEGROUP: [self.group_uuid],

      }

    self.share_locks[locking.LEVEL_INSTANCE] = 1

  def DeclareLocks(self, level):

    if level == locking.LEVEL_INSTANCE:

      assert not self.needed_locks[locking.LEVEL_INSTANCE]

      # Lock instances optimistically, needs verification once group lock has

      # been acquired

      self.needed_locks[locking.LEVEL_INSTANCE] = \

          self.cfg.GetNodeGroupInstances(self.group_uuid)

  @staticmethod

  def _UpdateAndVerifyDiskParams(old, new):

    """Updates and verifies disk parameters.

    """

    new_params = _GetUpdatedParams(old, new)

    utils.ForceDictType(new_params, constants.DISK_DT_TYPES)

    return new_params

  def CheckPrereq(self):

    """Check prerequisites.

    """

    owned_instances = frozenset(self.owned_locks(locking.LEVEL_INSTANCE))

    # Check if locked instances are still correct

    _CheckNodeGroupInstances(self.cfg, self.group_uuid, owned_instances)

    self.group = self.cfg.GetNodeGroup(self.group_uuid)

    cluster = self.cfg.GetClusterInfo()

    if self.group is None:

      raise errors.OpExecError("Could not retrieve group '%s' (UUID: %s)" %

                               (self.op.group_name, self.group_uuid))

    if self.op.ndparams:

      new_ndparams = _GetUpdatedParams(self.group.ndparams, self.op.ndparams)

      utils.ForceDictType(new_ndparams, constants.NDS_PARAMETER_TYPES)

      self.new_ndparams = new_ndparams

    if self.op.diskparams:

      diskparams = self.group.diskparams

      uavdp = self._UpdateAndVerifyDiskParams

      # For each disktemplate subdict update and verify the values

      new_diskparams = dict((dt,

                             uavdp(diskparams.get(dt, {}),

                                   self.op.diskparams[dt]))

                            for dt in constants.DISK_TEMPLATES

                            if dt in self.op.diskparams)

      # As we've all subdicts of diskparams ready, lets merge the actual

      # dict with all updated subdicts

      self.new_diskparams = objects.FillDict(diskparams, new_diskparams)

      try:

        utils.VerifyDictOptions(self.new_diskparams, constants.DISK_DT_DEFAULTS)

      except errors.OpPrereqError, err:

        raise errors.OpPrereqError("While verify diskparams options: %s" % err,

                                   errors.ECODE_INVAL)

    if self.op.hv_state:

      self.new_hv_state = _MergeAndVerifyHvState(self.op.hv_state,

                                                 self.group.hv_state_static)

    if self.op.disk_state:

      self.new_disk_state = \

        _MergeAndVerifyDiskState(self.op.disk_state,

                                 self.group.disk_state_static)

    if self.op.ipolicy:

      self.new_ipolicy = _GetUpdatedIPolicy(self.group.ipolicy,

                                            self.op.ipolicy,

                                            group_policy=True)

      new_ipolicy = cluster.SimpleFillIPolicy(self.new_ipolicy)

      inst_filter = lambda inst: inst.name in owned_instances

      instances = self.cfg.GetInstancesInfoByFilter(inst_filter).values()

      gmi = ganeti.masterd.instance

      violations = \

          _ComputeNewInstanceViolations(gmi.CalculateGroupIPolicy(cluster,

                                                                  self.group),

                                        new_ipolicy, instances, self.cfg)

      if violations:

        self.LogWarning("After the ipolicy change the following instances"

                        " violate them: %s",

                        utils.CommaJoin(violations))

  def BuildHooksEnv(self):

    """Build hooks env.

    """

    return {

      "GROUP_NAME": self.op.group_name,

      "NEW_ALLOC_POLICY": self.op.alloc_policy,

      }

  def BuildHooksNodes(self):

    """Build hooks nodes.

    """

    mn = self.cfg.GetMasterNode()

    return ([mn], [mn])

  def Exec(self, feedback_fn):

    """Modifies the node group.

    """

    result = []

    if self.op.ndparams:

      self.group.ndparams = self.new_ndparams

      result.append(("ndparams", str(self.group.ndparams)))

    if self.op.diskparams:

      self.group.diskparams = self.new_diskparams

      result.append(("diskparams", str(self.group.diskparams)))

    if self.op.alloc_policy:

      self.group.alloc_policy = self.op.alloc_policy

    if self.op.hv_state:

      self.group.hv_state_static = self.new_hv_state

    if self.op.disk_state:

      self.group.disk_state_static = self.new_disk_state

    if self.op.ipolicy:

      self.group.ipolicy = self.new_ipolicy

    self.cfg.Update(self.group, feedback_fn)

    return result

class LUGroupRemove(LogicalUnit):

  HPATH = "group-remove"

  HTYPE = constants.HTYPE_GROUP

  REQ_BGL = False

  def ExpandNames(self):

    # This will raises errors.OpPrereqError on its own:

    self.group_uuid = self.cfg.LookupNodeGroup(self.op.group_name)

    self.needed_locks = {

      locking.LEVEL_NODEGROUP: [self.group_uuid],

      }

  def CheckPrereq(self):

    """Check prerequisites.

    This checks that the given group name exists as a node group, that is

    empty (i.e., contains no nodes), and that is not the last group of the

    cluster.

    """

    # Verify that the group is empty.

    group_nodes = [node.name

                   for node in self.cfg.GetAllNodesInfo().values()

                   if node.group == self.group_uuid]

    if group_nodes:

      raise errors.OpPrereqError("Group '%s' not empty, has the following"

                                 " nodes: %s" %

                                 (self.op.group_name,

                                  utils.CommaJoin(utils.NiceSort(group_nodes))),

                                 errors.ECODE_STATE)

    # Verify the cluster would not be left group-less.

    if len(self.cfg.GetNodeGroupList()) == 1:

      raise errors.OpPrereqError("Group '%s' is the only group, cannot be"

                                 " removed" % self.op.group_name,

                                 errors.ECODE_STATE)

  def BuildHooksEnv(self):

    """Build hooks env.

    """

    return {

      "GROUP_NAME": self.op.group_name,

      }

  def BuildHooksNodes(self):

    """Build hooks nodes.

    """

    mn = self.cfg.GetMasterNode()

    return ([mn], [mn])

  def Exec(self, feedback_fn):

    """Remove the node group.

    """

    try:

      self.cfg.RemoveNodeGroup(self.group_uuid)

    except errors.ConfigurationError:

      raise errors.OpExecError("Group '%s' with UUID %s disappeared" %

                               (self.op.group_name, self.group_uuid))

    self.remove_locks[locking.LEVEL_NODEGROUP] = self.group_uuid

class LUGroupRename(LogicalUnit):

  HPATH = "group-rename"

  HTYPE = constants.HTYPE_GROUP

  REQ_BGL = False

  def ExpandNames(self):

    # This raises errors.OpPrereqError on its own:

    self.group_uuid = self.cfg.LookupNodeGroup(self.op.group_name)

    self.needed_locks = {

      locking.LEVEL_NODEGROUP: [self.group_uuid],

      }

  def CheckPrereq(self):

    """Check prerequisites.

    Ensures requested new name is not yet used.

    """

    try:

      new_name_uuid = self.cfg.LookupNodeGroup(self.op.new_name)

    except errors.OpPrereqError:

      pass

    else:

      raise errors.OpPrereqError("Desired new name '%s' clashes with existing"

                                 " node group (UUID: %s)" %

                                 (self.op.new_name, new_name_uuid),

                                 errors.ECODE_EXISTS)

  def BuildHooksEnv(self):

    """Build hooks env.

    """

    return {

      "OLD_NAME": self.op.group_name,

      "NEW_NAME": self.op.new_name,

      }

  def BuildHooksNodes(self):

    """Build hooks nodes.

    """

    mn = self.cfg.GetMasterNode()

    all_nodes = self.cfg.GetAllNodesInfo()

    all_nodes.pop(mn, None)

    run_nodes = [mn]

    run_nodes.extend(node.name for node in all_nodes.values()

                     if node.group == self.group_uuid)

    return (run_nodes, run_nodes)

  def Exec(self, feedback_fn):

    """Rename the node group.

    """

    group = self.cfg.GetNodeGroup(self.group_uuid)

    if group is None:

      raise errors.OpExecError("Could not retrieve group '%s' (UUID: %s)" %

                               (self.op.group_name, self.group_uuid))

    group.name = self.op.new_name

    self.cfg.Update(group, feedback_fn)

    return self.op.new_name

class LUGroupEvacuate(LogicalUnit):

  HPATH = "group-evacuate"

  HTYPE = constants.HTYPE_GROUP

  REQ_BGL = False

  def ExpandNames(self):

    # This raises errors.OpPrereqError on its own:

    self.group_uuid = self.cfg.LookupNodeGroup(self.op.group_name)

    if self.op.target_groups:

      self.req_target_uuids = map(self.cfg.LookupNodeGroup,

                                  self.op.target_groups)

    else:

      self.req_target_uuids = []

    if self.group_uuid in self.req_target_uuids:

      raise errors.OpPrereqError("Group to be evacuated (%s) can not be used"

                                 " as a target group (targets are %s)" %

                                 (self.group_uuid,

                                  utils.CommaJoin(self.req_target_uuids)),

                                 errors.ECODE_INVAL)

    self.op.iallocator = _GetDefaultIAllocator(self.cfg, self.op.iallocator)

    self.share_locks = _ShareAll()

    self.needed_locks = {

      locking.LEVEL_INSTANCE: [],

      locking.LEVEL_NODEGROUP: [],

      locking.LEVEL_NODE: [],

      }

  def DeclareLocks(self, level):

    if level == locking.LEVEL_INSTANCE:

      assert not self.needed_locks[locking.LEVEL_INSTANCE]

      # Lock instances optimistically, needs verification once node and group

      # locks have been acquired

      self.needed_locks[locking.LEVEL_INSTANCE] = \

        self.cfg.GetNodeGroupInstances(self.group_uuid)

    elif level == locking.LEVEL_NODEGROUP:

      assert not self.needed_locks[locking.LEVEL_NODEGROUP]

      if self.req_target_uuids:

        lock_groups = set([self.group_uuid] + self.req_target_uuids)

        # Lock all groups used by instances optimistically; this requires going

        # via the node before it's locked, requiring verification later on

        lock_groups.update(group_uuid

                           for instance_name in

                             self.owned_locks(locking.LEVEL_INSTANCE)

                           for group_uuid in

                             self.cfg.GetInstanceNodeGroups(instance_name))

      else:

        # No target groups, need to lock all of them

        lock_groups = locking.ALL_SET

      self.needed_locks[locking.LEVEL_NODEGROUP] = lock_groups

    elif level == locking.LEVEL_NODE:

      # This will only lock the nodes in the group to be evacuated which

      # contain actual instances

      self.recalculate_locks[locking.LEVEL_NODE] = constants.LOCKS_APPEND

      self._LockInstancesNodes()

      # Lock all nodes in group to be evacuated and target groups

      owned_groups = frozenset(self.owned_locks(locking.LEVEL_NODEGROUP))

      assert self.group_uuid in owned_groups

      member_nodes = [node_name

                      for group in owned_groups

                      for node_name in self.cfg.GetNodeGroup(group).members]

      self.needed_locks[locking.LEVEL_NODE].extend(member_nodes)

  def CheckPrereq(self):

    owned_instances = frozenset(self.owned_locks(locking.LEVEL_INSTANCE))

    owned_groups = frozenset(self.owned_locks(locking.LEVEL_NODEGROUP))

    owned_nodes = frozenset(self.owned_locks(locking.LEVEL_NODE))

    assert owned_groups.issuperset(self.req_target_uuids)

    assert self.group_uuid in owned_groups

    # Check if locked instances are still correct

    _CheckNodeGroupInstances(self.cfg, self.group_uuid, owned_instances)

    # Get instance information

    self.instances = dict(self.cfg.GetMultiInstanceInfo(owned_instances))

    # Check if node groups for locked instances are still correct

    _CheckInstancesNodeGroups(self.cfg, self.instances,

                              owned_groups, owned_nodes, self.group_uuid)

    if self.req_target_uuids:

      # User requested specific target groups

      self.target_uuids = self.req_target_uuids

    else:

      # All groups except the one to be evacuated are potential targets

      self.target_uuids = [group_uuid for group_uuid in owned_groups

                           if group_uuid != self.group_uuid]

      if not self.target_uuids:

        raise errors.OpPrereqError("There are no possible target groups",

                                   errors.ECODE_INVAL)

  def BuildHooksEnv(self):

    """Build hooks env.

    """

    return {

      "GROUP_NAME": self.op.group_name,

      "TARGET_GROUPS": " ".join(self.target_uuids),

      }

  def BuildHooksNodes(self):

    """Build hooks nodes.

    """

    mn = self.cfg.GetMasterNode()

    assert self.group_uuid in self.owned_locks(locking.LEVEL_NODEGROUP)

    run_nodes = [mn] + self.cfg.GetNodeGroup(self.group_uuid).members

    return (run_nodes, run_nodes)

  def Exec(self, feedback_fn):

    instances = list(self.owned_locks(locking.LEVEL_INSTANCE))

    assert self.group_uuid not in self.target_uuids

    req = iallocator.IAReqGroupChange(instances=instances,

                                      target_groups=self.target_uuids)

    ial = iallocator.IAllocator(self.cfg, self.rpc, req)

    ial.Run(self.op.iallocator)

    if not ial.success:

      raise errors.OpPrereqError("Can't compute group evacuation using"

                                 " iallocator '%s': %s" %

                                 (self.op.iallocator, ial.info),

                                 errors.ECODE_NORES)

    jobs = _LoadNodeEvacResult(self, ial.result, self.op.early_release, False)

    self.LogInfo("Iallocator returned %s job(s) for evacuating node group %s",

                 len(jobs), self.op.group_name)

    return ResultWithJobs(jobs)