Synnefo » snf-ganeti

#

#

# Copyright (C) 2006, 2007, 2008, 2009, 2010, 2011, 2012, 2013 Google Inc.

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful, but

# WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

# General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA

# 02110-1301, USA.

"""Transportable objects for Ganeti.

This module provides small, mostly data-only objects which are safe to

pass to and from external parties.

"""

# pylint: disable=E0203,W0201,R0902

# E0203: Access to member %r before its definition, since we use

# objects.py which doesn't explicitly initialise its members

# W0201: Attribute '%s' defined outside __init__

# R0902: Allow instances of these objects to have more than 20 attributes

import ConfigParser

import re

import copy

import logging

import time

from cStringIO import StringIO

from ganeti import errors

from ganeti import constants

from ganeti import netutils

from ganeti import outils

from ganeti import utils

from socket import AF_INET

__all__ = ["ConfigObject", "ConfigData", "NIC", "Disk", "Instance",

           "OS", "Node", "NodeGroup", "Cluster", "FillDict", "Network"]

_TIMESTAMPS = ["ctime", "mtime"]

_UUID = ["uuid"]

def FillDict(defaults_dict, custom_dict, skip_keys=None):

  """Basic function to apply settings on top a default dict.

  @type defaults_dict: dict

  @param defaults_dict: dictionary holding the default values

  @type custom_dict: dict

  @param custom_dict: dictionary holding customized value

  @type skip_keys: list

  @param skip_keys: which keys not to fill

  @rtype: dict

  @return: dict with the 'full' values

  """

  ret_dict = copy.deepcopy(defaults_dict)

  ret_dict.update(custom_dict)

  if skip_keys:

    for k in skip_keys:

      if k in ret_dict:

        del ret_dict[k]

  return ret_dict

def FillIPolicy(default_ipolicy, custom_ipolicy):

  """Fills an instance policy with defaults.

  """

  assert frozenset(default_ipolicy.keys()) == constants.IPOLICY_ALL_KEYS

  ret_dict = copy.deepcopy(custom_ipolicy)

  for key in default_ipolicy:

    if key not in ret_dict:

      ret_dict[key] = copy.deepcopy(default_ipolicy[key])

    elif key == constants.ISPECS_STD:

      ret_dict[key] = FillDict(default_ipolicy[key], ret_dict[key])

  return ret_dict

def FillDiskParams(default_dparams, custom_dparams, skip_keys=None):

  """Fills the disk parameter defaults.

  @see: L{FillDict} for parameters and return value

  """

  assert frozenset(default_dparams.keys()) == constants.DISK_TEMPLATES

  return dict((dt, FillDict(default_dparams[dt], custom_dparams.get(dt, {}),

                             skip_keys=skip_keys))

              for dt in constants.DISK_TEMPLATES)

def UpgradeGroupedParams(target, defaults):

  """Update all groups for the target parameter.

  @type target: dict of dicts

  @param target: {group: {parameter: value}}

  @type defaults: dict

  @param defaults: default parameter values

  """

  if target is None:

    target = {constants.PP_DEFAULT: defaults}

  else:

    for group in target:

      target[group] = FillDict(defaults, target[group])

  return target

def UpgradeBeParams(target):

  """Update the be parameters dict to the new format.

  @type target: dict

  @param target: "be" parameters dict

  """

  if constants.BE_MEMORY in target:

    memory = target[constants.BE_MEMORY]

    target[constants.BE_MAXMEM] = memory

    target[constants.BE_MINMEM] = memory

    del target[constants.BE_MEMORY]

def UpgradeDiskParams(diskparams):

  """Upgrade the disk parameters.

  @type diskparams: dict

  @param diskparams: disk parameters to upgrade

  @rtype: dict

  @return: the upgraded disk parameters dict

  """

  if not diskparams:

    result = {}

  else:

    result = FillDiskParams(constants.DISK_DT_DEFAULTS, diskparams)

  return result

def UpgradeNDParams(ndparams):

  """Upgrade ndparams structure.

  @type ndparams: dict

  @param ndparams: disk parameters to upgrade

  @rtype: dict

  @return: the upgraded node parameters dict

  """

  if ndparams is None:

    ndparams = {}

  if (constants.ND_OOB_PROGRAM in ndparams and

      ndparams[constants.ND_OOB_PROGRAM] is None):

    # will be reset by the line below

    del ndparams[constants.ND_OOB_PROGRAM]

  return FillDict(constants.NDC_DEFAULTS, ndparams)

def MakeEmptyIPolicy():

  """Create empty IPolicy dictionary.

  """

  return {}

class ConfigObject(outils.ValidatedSlots):

  """A generic config object.

  It has the following properties:

    - provides somewhat safe recursive unpickling and pickling for its classes

    - unset attributes which are defined in slots are always returned

      as None instead of raising an error

  Classes derived from this must always declare __slots__ (we use many

  config objects and the memory reduction is useful)

  """

  __slots__ = []

  def __getattr__(self, name):

    if name not in self.GetAllSlots():

      raise AttributeError("Invalid object attribute %s.%s" %

                           (type(self).__name__, name))

    return None

  def __setstate__(self, state):

    slots = self.GetAllSlots()

    for name in state:

      if name in slots:

        setattr(self, name, state[name])

  def Validate(self):

    """Validates the slots.

    """

  def ToDict(self):

    """Convert to a dict holding only standard python types.

    The generic routine just dumps all of this object's attributes in

    a dict. It does not work if the class has children who are

    ConfigObjects themselves (e.g. the nics list in an Instance), in

    which case the object should subclass the function in order to

    make sure all objects returned are only standard python types.

    """

    result = {}

    for name in self.GetAllSlots():

      value = getattr(self, name, None)

      if value is not None:

        result[name] = value

    return result

  __getstate__ = ToDict

  @classmethod

  def FromDict(cls, val):

    """Create an object from a dictionary.

    This generic routine takes a dict, instantiates a new instance of

    the given class, and sets attributes based on the dict content.

    As for `ToDict`, this does not work if the class has children

    who are ConfigObjects themselves (e.g. the nics list in an

    Instance), in which case the object should subclass the function

    and alter the objects.

    """

    if not isinstance(val, dict):

      raise errors.ConfigurationError("Invalid object passed to FromDict:"

                                      " expected dict, got %s" % type(val))

    val_str = dict([(str(k), v) for k, v in val.iteritems()])

    obj = cls(**val_str) # pylint: disable=W0142

    return obj

  def Copy(self):

    """Makes a deep copy of the current object and its children.

    """

    dict_form = self.ToDict()

    clone_obj = self.__class__.FromDict(dict_form)

    return clone_obj

  def __repr__(self):

    """Implement __repr__ for ConfigObjects."""

    return repr(self.ToDict())

  def __eq__(self, other):

    """Implement __eq__ for ConfigObjects."""

    return isinstance(other, self.__class__) and self.ToDict() == other.ToDict()

  def UpgradeConfig(self):

    """Fill defaults for missing configuration values.

    This method will be called at configuration load time, and its

    implementation will be object dependent.

    """

    pass

class TaggableObject(ConfigObject):

  """An generic class supporting tags.

  """

  __slots__ = ["tags"]

  VALID_TAG_RE = re.compile(r"^[\w.+*/:@-]+$")

  @classmethod

  def ValidateTag(cls, tag):

    """Check if a tag is valid.

    If the tag is invalid, an errors.TagError will be raised. The

    function has no return value.

    """

    if not isinstance(tag, basestring):

      raise errors.TagError("Invalid tag type (not a string)")

    if len(tag) > constants.MAX_TAG_LEN:

      raise errors.TagError("Tag too long (>%d characters)" %

                            constants.MAX_TAG_LEN)

    if not tag:

      raise errors.TagError("Tags cannot be empty")

    if not cls.VALID_TAG_RE.match(tag):

      raise errors.TagError("Tag contains invalid characters")

  def GetTags(self):

    """Return the tags list.

    """

    tags = getattr(self, "tags", None)

    if tags is None:

      tags = self.tags = set()

    return tags

  def AddTag(self, tag):

    """Add a new tag.

    """

    self.ValidateTag(tag)

    tags = self.GetTags()

    if len(tags) >= constants.MAX_TAGS_PER_OBJ:

      raise errors.TagError("Too many tags")

    self.GetTags().add(tag)

  def RemoveTag(self, tag):

    """Remove a tag.

    """

    self.ValidateTag(tag)

    tags = self.GetTags()

    try:

      tags.remove(tag)

    except KeyError:

      raise errors.TagError("Tag not found")

  def ToDict(self):

    """Taggable-object-specific conversion to standard python types.

    This replaces the tags set with a list.

    """

    bo = super(TaggableObject, self).ToDict()

    tags = bo.get("tags", None)

    if isinstance(tags, set):

      bo["tags"] = list(tags)

    return bo

  @classmethod

  def FromDict(cls, val):

    """Custom function for instances.

    """

    obj = super(TaggableObject, cls).FromDict(val)

    if hasattr(obj, "tags") and isinstance(obj.tags, list):

      obj.tags = set(obj.tags)

    return obj

class MasterNetworkParameters(ConfigObject):

  """Network configuration parameters for the master

  @ivar uuid: master nodes UUID

  @ivar ip: master IP

  @ivar netmask: master netmask

  @ivar netdev: master network device

  @ivar ip_family: master IP family

  """

  __slots__ = [

    "uuid",

    "ip",

    "netmask",

    "netdev",

    "ip_family",

    ]

class ConfigData(ConfigObject):

  """Top-level config object."""

  __slots__ = [

    "version",

    "cluster",

    "nodes",

    "nodegroups",

    "instances",

    "networks",

    "serial_no",

    ] + _TIMESTAMPS

  def ToDict(self):

    """Custom function for top-level config data.

    This just replaces the list of instances, nodes and the cluster

    with standard python types.

    """

    mydict = super(ConfigData, self).ToDict()

    mydict["cluster"] = mydict["cluster"].ToDict()

    for key in "nodes", "instances", "nodegroups", "networks":

      mydict[key] = outils.ContainerToDicts(mydict[key])

    return mydict

  @classmethod

  def FromDict(cls, val):

    """Custom function for top-level config data

    """

    obj = super(ConfigData, cls).FromDict(val)

    obj.cluster = Cluster.FromDict(obj.cluster)

    obj.nodes = outils.ContainerFromDicts(obj.nodes, dict, Node)

    obj.instances = \

      outils.ContainerFromDicts(obj.instances, dict, Instance)

    obj.nodegroups = \

      outils.ContainerFromDicts(obj.nodegroups, dict, NodeGroup)

    obj.networks = outils.ContainerFromDicts(obj.networks, dict, Network)

    return obj

  def HasAnyDiskOfType(self, dev_type):

    """Check if in there is at disk of the given type in the configuration.

    @type dev_type: L{constants.DTS_BLOCK}

    @param dev_type: the type to look for

    @rtype: boolean

    @return: boolean indicating if a disk of the given type was found or not

    """

    for instance in self.instances.values():

      for disk in instance.disks:

        if disk.IsBasedOnDiskType(dev_type):

          return True

    return False

  def UpgradeConfig(self):

    """Fill defaults for missing configuration values.

    """

    self.cluster.UpgradeConfig()

    for node in self.nodes.values():

      node.UpgradeConfig()

    for instance in self.instances.values():

      instance.UpgradeConfig()

    self._UpgradeEnabledDiskTemplates()

    if self.nodegroups is None:

      self.nodegroups = {}

    for nodegroup in self.nodegroups.values():

      nodegroup.UpgradeConfig()

      InstancePolicy.UpgradeDiskTemplates(

        nodegroup.ipolicy, self.cluster.enabled_disk_templates)

    if self.cluster.drbd_usermode_helper is None:

      if self.cluster.IsDiskTemplateEnabled(constants.DT_DRBD8):

        self.cluster.drbd_usermode_helper = constants.DEFAULT_DRBD_HELPER

    if self.networks is None:

      self.networks = {}

    for network in self.networks.values():

      network.UpgradeConfig()

  def _UpgradeEnabledDiskTemplates(self):

    """Upgrade the cluster's enabled disk templates by inspecting the currently

       enabled and/or used disk templates.

    """

    if not self.cluster.enabled_disk_templates:

      template_set = \

        set([inst.disk_template for inst in self.instances.values()])

      # Add drbd and plain, if lvm is enabled (by specifying a volume group)

      if self.cluster.volume_group_name:

        template_set.add(constants.DT_DRBD8)

        template_set.add(constants.DT_PLAIN)

      # Set enabled_disk_templates to the inferred disk templates. Order them

      # according to a preference list that is based on Ganeti's history of

      # supported disk templates.

      self.cluster.enabled_disk_templates = []

      for preferred_template in constants.DISK_TEMPLATE_PREFERENCE:

        if preferred_template in template_set:

          self.cluster.enabled_disk_templates.append(preferred_template)

          template_set.remove(preferred_template)

      self.cluster.enabled_disk_templates.extend(list(template_set))

    InstancePolicy.UpgradeDiskTemplates(

      self.cluster.ipolicy, self.cluster.enabled_disk_templates)

class NIC(ConfigObject):

  """Config object representing a network card."""

  __slots__ = ["name", "mac", "ip", "network",

               "nicparams", "netinfo", "pci"] + _UUID

  @classmethod

  def CheckParameterSyntax(cls, nicparams):

    """Check the given parameters for validity.

    @type nicparams:  dict

    @param nicparams: dictionary with parameter names/value

    @raise errors.ConfigurationError: when a parameter is not valid

    """

    mode = nicparams[constants.NIC_MODE]

    if (mode not in constants.NIC_VALID_MODES and

        mode != constants.VALUE_AUTO):

      raise errors.ConfigurationError("Invalid NIC mode '%s'" % mode)

    if (mode == constants.NIC_MODE_BRIDGED and

        not nicparams[constants.NIC_LINK]):

      raise errors.ConfigurationError("Missing bridged NIC link")

class Disk(ConfigObject):

  """Config object representing a block device."""

  __slots__ = (["name", "dev_type", "logical_id", "children", "iv_name",

                "size", "mode", "params", "spindles", "pci"] + _UUID +

               # dynamic_params is special. It depends on the node this instance

               # is sent to, and should not be persisted.

               ["dynamic_params"])

  def CreateOnSecondary(self):

    """Test if this device needs to be created on a secondary node."""

    return self.dev_type in (constants.DT_DRBD8, constants.DT_PLAIN)

  def AssembleOnSecondary(self):

    """Test if this device needs to be assembled on a secondary node."""

    return self.dev_type in (constants.DT_DRBD8, constants.DT_PLAIN)

  def OpenOnSecondary(self):

    """Test if this device needs to be opened on a secondary node."""

    return self.dev_type in (constants.DT_PLAIN,)

  def StaticDevPath(self):

    """Return the device path if this device type has a static one.

    Some devices (LVM for example) live always at the same /dev/ path,

    irrespective of their status. For such devices, we return this

    path, for others we return None.

    @warning: The path returned is not a normalized pathname; callers

        should check that it is a valid path.

    """

    if self.dev_type == constants.DT_PLAIN:

      return "/dev/%s/%s" % (self.logical_id[0], self.logical_id[1])

    elif self.dev_type == constants.DT_BLOCK:

      return self.logical_id[1]

    elif self.dev_type == constants.DT_RBD:

      return "/dev/%s/%s" % (self.logical_id[0], self.logical_id[1])

    return None

  def ChildrenNeeded(self):

    """Compute the needed number of children for activation.

    This method will return either -1 (all children) or a positive

    number denoting the minimum number of children needed for

    activation (only mirrored devices will usually return >=0).

    Currently, only DRBD8 supports diskless activation (therefore we

    return 0), for all other we keep the previous semantics and return

    -1.

    """

    if self.dev_type == constants.DT_DRBD8:

      return 0

    return -1

  def IsBasedOnDiskType(self, dev_type):

    """Check if the disk or its children are based on the given type.

    @type dev_type: L{constants.DTS_BLOCK}

    @param dev_type: the type to look for

    @rtype: boolean

    @return: boolean indicating if a device of the given type was found or not

    """

    if self.children:

      for child in self.children:

        if child.IsBasedOnDiskType(dev_type):

          return True

    return self.dev_type == dev_type

  def GetNodes(self, node_uuid):

    """This function returns the nodes this device lives on.

    Given the node on which the parent of the device lives on (or, in

    case of a top-level device, the primary node of the devices'

    instance), this function will return a list of nodes on which this

    devices needs to (or can) be assembled.

    """

    if self.dev_type in [constants.DT_PLAIN, constants.DT_FILE,

                         constants.DT_BLOCK, constants.DT_RBD,

                         constants.DT_EXT, constants.DT_SHARED_FILE]:

      result = [node_uuid]

    elif self.dev_type in constants.DTS_DRBD:

      result = [self.logical_id[0], self.logical_id[1]]

      if node_uuid not in result:

        raise errors.ConfigurationError("DRBD device passed unknown node")

    else:

      raise errors.ProgrammerError("Unhandled device type %s" % self.dev_type)

    return result

  def ComputeNodeTree(self, parent_node_uuid):

    """Compute the node/disk tree for this disk and its children.

    This method, given the node on which the parent disk lives, will

    return the list of all (node UUID, disk) pairs which describe the disk

    tree in the most compact way. For example, a drbd/lvm stack

    will be returned as (primary_node, drbd) and (secondary_node, drbd)

    which represents all the top-level devices on the nodes.

    """

    my_nodes = self.GetNodes(parent_node_uuid)

    result = [(node, self) for node in my_nodes]

    if not self.children:

      # leaf device

      return result

    for node in my_nodes:

      for child in self.children:

        child_result = child.ComputeNodeTree(node)

        if len(child_result) == 1:

          # child (and all its descendants) is simple, doesn't split

          # over multiple hosts, so we don't need to describe it, our

          # own entry for this node describes it completely

          continue

        else:

          # check if child nodes differ from my nodes; note that

          # subdisk can differ from the child itself, and be instead

          # one of its descendants

          for subnode, subdisk in child_result:

            if subnode not in my_nodes:

              result.append((subnode, subdisk))

            # otherwise child is under our own node, so we ignore this

            # entry (but probably the other results in the list will

            # be different)

    return result

  def ComputeGrowth(self, amount):

    """Compute the per-VG growth requirements.

    This only works for VG-based disks.

    @type amount: integer

    @param amount: the desired increase in (user-visible) disk space

    @rtype: dict

    @return: a dictionary of volume-groups and the required size

    """

    if self.dev_type == constants.DT_PLAIN:

      return {self.logical_id[0]: amount}

    elif self.dev_type == constants.DT_DRBD8:

      if self.children:

        return self.children[0].ComputeGrowth(amount)

      else:

        return {}

    else:

      # Other disk types do not require VG space

      return {}

  def RecordGrow(self, amount):

    """Update the size of this disk after growth.

    This method recurses over the disks's children and updates their

    size correspondigly. The method needs to be kept in sync with the

    actual algorithms from bdev.

    """

    if self.dev_type in (constants.DT_PLAIN, constants.DT_FILE,

                         constants.DT_RBD, constants.DT_EXT,

                         constants.DT_SHARED_FILE):

      self.size += amount

    elif self.dev_type == constants.DT_DRBD8:

      if self.children:

        self.children[0].RecordGrow(amount)

      self.size += amount

    else:

      raise errors.ProgrammerError("Disk.RecordGrow called for unsupported"

                                   " disk type %s" % self.dev_type)

  def Update(self, size=None, mode=None, spindles=None):

    """Apply changes to size, spindles and mode.

    """

    if self.dev_type == constants.DT_DRBD8:

      if self.children:

        self.children[0].Update(size=size, mode=mode)

    else:

      assert not self.children

    if size is not None:

      self.size = size

    if mode is not None:

      self.mode = mode

    if spindles is not None:

      self.spindles = spindles

  def UnsetSize(self):

    """Sets recursively the size to zero for the disk and its children.

    """

    if self.children:

      for child in self.children:

        child.UnsetSize()

    self.size = 0

  def UpdateDynamicDiskParams(self, target_node_uuid, nodes_ip):

    """Updates the dynamic disk params for the given node.

    This is mainly used for drbd, which needs ip/port configuration.

    Arguments:

      - target_node_uuid: the node UUID we wish to configure for

      - nodes_ip: a mapping of node name to ip

    The target_node must exist in nodes_ip, and should be one of the

    nodes in the logical ID if this device is a DRBD device.

    """

    if self.children:

      for child in self.children:

        child.UpdateDynamicDiskParams(target_node_uuid, nodes_ip)

    dyn_disk_params = {}

    if self.logical_id is not None and self.dev_type in constants.DTS_DRBD:

      pnode_uuid, snode_uuid, _, pminor, sminor, _ = self.logical_id

      if target_node_uuid not in (pnode_uuid, snode_uuid):

        # disk object is being sent to neither the primary nor the secondary

        # node. reset the dynamic parameters, the target node is not

        # supposed to use them.

        self.dynamic_params = dyn_disk_params

        return

      pnode_ip = nodes_ip.get(pnode_uuid, None)