Synnefo » snf-ganeti

#

#

# Copyright (C) 2006, 2007 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.

"""Block device abstraction"""

import re

import time

import errno

import pyparsing as pyp

import os

import logging

from ganeti import utils

from ganeti import errors

from ganeti import constants

class BlockDev(object):

  """Block device abstract class.

  A block device can be in the following states:

    - not existing on the system, and by `Create()` it goes into:

    - existing but not setup/not active, and by `Assemble()` goes into:

    - active read-write and by `Open()` it goes into

    - online (=used, or ready for use)

  A device can also be online but read-only, however we are not using

  the readonly state (LV has it, if needed in the future) and we are

  usually looking at this like at a stack, so it's easier to

  conceptualise the transition from not-existing to online and back

  like a linear one.

  The many different states of the device are due to the fact that we

  need to cover many device types:

    - logical volumes are created, lvchange -a y $lv, and used

    - drbd devices are attached to a local disk/remote peer and made primary

  A block device is identified by three items:

    - the /dev path of the device (dynamic)

    - a unique ID of the device (static)

    - it's major/minor pair (dynamic)

  Not all devices implement both the first two as distinct items. LVM

  logical volumes have their unique ID (the pair volume group, logical

  volume name) in a 1-to-1 relation to the dev path. For DRBD devices,

  the /dev path is again dynamic and the unique id is the pair (host1,

  dev1), (host2, dev2).

  You can get to a device in two ways:

    - creating the (real) device, which returns you

      an attached instance (lvcreate)

    - attaching of a python instance to an existing (real) device

  The second point, the attachement to a device, is different

  depending on whether the device is assembled or not. At init() time,

  we search for a device with the same unique_id as us. If found,

  good. It also means that the device is already assembled. If not,

  after assembly we'll have our correct major/minor.

  """

  def __init__(self, unique_id, children):

    self._children = children

    self.dev_path = None

    self.unique_id = unique_id

    self.major = None

    self.minor = None

    self.attached = False

  def Assemble(self):

    """Assemble the device from its components.

    If this is a plain block device (e.g. LVM) than assemble does

    nothing, as the LVM has no children and we don't put logical

    volumes offline.

    One guarantee is that after the device has been assembled, it

    knows its major/minor numbers. This allows other devices (usually

    parents) to probe correctly for their children.

    """

    status = True

    for child in self._children:

      if not isinstance(child, BlockDev):

        raise TypeError("Invalid child passed of type '%s'" % type(child))

      if not status:

        break

      status = status and child.Assemble()

      if not status:

        break

      try:

        child.Open()

      except errors.BlockDeviceError:

        for child in self._children:

          child.Shutdown()

        raise

    if not status:

      for child in self._children:

        child.Shutdown()

    return status

  def Attach(self):

    """Find a device which matches our config and attach to it.

    """

    raise NotImplementedError

  def Close(self):

    """Notifies that the device will no longer be used for I/O.

    """

    raise NotImplementedError

  @classmethod

  def Create(cls, unique_id, children, size):

    """Create the device.

    If the device cannot be created, it will return None

    instead. Error messages go to the logging system.

    Note that for some devices, the unique_id is used, and for other,

    the children. The idea is that these two, taken together, are

    enough for both creation and assembly (later).

    """

    raise NotImplementedError

  def Remove(self):

    """Remove this device.

    This makes sense only for some of the device types: LV and file

    storeage. Also note that if the device can't attach, the removal

    can't be completed.

    """

    raise NotImplementedError

  def Rename(self, new_id):

    """Rename this device.

    This may or may not make sense for a given device type.

    """

    raise NotImplementedError

  def Open(self, force=False):

    """Make the device ready for use.

    This makes the device ready for I/O. For now, just the DRBD

    devices need this.

    The force parameter signifies that if the device has any kind of

    --force thing, it should be used, we know what we are doing.

    """

    raise NotImplementedError

  def Shutdown(self):

    """Shut down the device, freeing its children.

    This undoes the `Assemble()` work, except for the child

    assembling; as such, the children on the device are still

    assembled after this call.

    """

    raise NotImplementedError

  def SetSyncSpeed(self, speed):

    """Adjust the sync speed of the mirror.

    In case this is not a mirroring device, this is no-op.

    """

    result = True

    if self._children:

      for child in self._children:

        result = result and child.SetSyncSpeed(speed)

    return result

  def GetSyncStatus(self):

    """Returns the sync status of the device.

    If this device is a mirroring device, this function returns the

    status of the mirror.

    If sync_percent is None, it means the device is not syncing.

    If estimated_time is None, it means we can't estimate

    the time needed, otherwise it's the time left in seconds.

    If is_degraded is True, it means the device is missing

    redundancy. This is usually a sign that something went wrong in

    the device setup, if sync_percent is None.

    The ldisk parameter represents the degradation of the local

    data. This is only valid for some devices, the rest will always

    return False (not degraded).

    @rtype: tuple

    @return: (sync_percent, estimated_time, is_degraded, ldisk)

    """

    return None, None, False, False

  def CombinedSyncStatus(self):

    """Calculate the mirror status recursively for our children.

    The return value is the same as for `GetSyncStatus()` except the

    minimum percent and maximum time are calculated across our

    children.

    """

    min_percent, max_time, is_degraded, ldisk = self.GetSyncStatus()

    if self._children:

      for child in self._children:

        c_percent, c_time, c_degraded, c_ldisk = child.GetSyncStatus()

        if min_percent is None:

          min_percent = c_percent

        elif c_percent is not None:

          min_percent = min(min_percent, c_percent)

        if max_time is None:

          max_time = c_time

        elif c_time is not None:

          max_time = max(max_time, c_time)

        is_degraded = is_degraded or c_degraded

        ldisk = ldisk or c_ldisk

    return min_percent, max_time, is_degraded, ldisk

  def SetInfo(self, text):

    """Update metadata with info text.

    Only supported for some device types.

    """

    for child in self._children:

      child.SetInfo(text)

  def Grow(self, amount):

    """Grow the block device.

    @param amount: the amount (in mebibytes) to grow with

    """

    raise NotImplementedError

  def __repr__(self):

    return ("<%s: unique_id: %s, children: %s, %s:%s, %s>" %

            (self.__class__, self.unique_id, self._children,

             self.major, self.minor, self.dev_path))

class LogicalVolume(BlockDev):

  """Logical Volume block device.

  """

  def __init__(self, unique_id, children):

    """Attaches to a LV device.

    The unique_id is a tuple (vg_name, lv_name)

    """

    super(LogicalVolume, self).__init__(unique_id, children)

    if not isinstance(unique_id, (tuple, list)) or len(unique_id) != 2:

      raise ValueError("Invalid configuration data %s" % str(unique_id))

    self._vg_name, self._lv_name = unique_id

    self.dev_path = "/dev/%s/%s" % (self._vg_name, self._lv_name)

    self._degraded = True

    self.major = self.minor = None

    self.Attach()

  @classmethod

  def Create(cls, unique_id, children, size):

    """Create a new logical volume.

    """

    if not isinstance(unique_id, (tuple, list)) or len(unique_id) != 2:

      raise ValueError("Invalid configuration data %s" % str(unique_id))

    vg_name, lv_name = unique_id

    pvs_info = cls.GetPVInfo(vg_name)

    if not pvs_info:

      raise errors.BlockDeviceError("Can't compute PV info for vg %s" %

                                    vg_name)

    pvs_info.sort()

    pvs_info.reverse()

    pvlist = [ pv[1] for pv in pvs_info ]

    free_size = sum([ pv[0] for pv in pvs_info ])

    # The size constraint should have been checked from the master before

    # calling the create function.

    if free_size < size:

      raise errors.BlockDeviceError("Not enough free space: required %s,"

                                    " available %s" % (size, free_size))

    result = utils.RunCmd(["lvcreate", "-L%dm" % size, "-n%s" % lv_name,

                           vg_name] + pvlist)

    if result.failed:

      raise errors.BlockDeviceError("%s - %s" % (result.fail_reason,

                                                result.output))

    return LogicalVolume(unique_id, children)

  @staticmethod

  def GetPVInfo(vg_name):

    """Get the free space info for PVs in a volume group.

    @param vg_name: the volume group name

    @rtype: list

    @return: list of tuples (free_space, name) with free_space in mebibytes

    """

    command = ["pvs", "--noheadings", "--nosuffix", "--units=m",

               "-opv_name,vg_name,pv_free,pv_attr", "--unbuffered",

               "--separator=:"]

    result = utils.RunCmd(command)

    if result.failed:

      logging.error("Can't get the PV information: %s - %s",

                    result.fail_reason, result.output)

      return None

    data = []

    for line in result.stdout.splitlines():

      fields = line.strip().split(':')

      if len(fields) != 4:

        logging.error("Can't parse pvs output: line '%s'", line)

        return None

      # skip over pvs from another vg or ones which are not allocatable

      if fields[1] != vg_name or fields[3][0] != 'a':

        continue

      data.append((float(fields[2]), fields[0]))

    return data

  def Remove(self):

    """Remove this logical volume.

    """

    if not self.minor and not self.Attach():

      # the LV does not exist

      return True

    result = utils.RunCmd(["lvremove", "-f", "%s/%s" %

                           (self._vg_name, self._lv_name)])

    if result.failed:

      logging.error("Can't lvremove: %s - %s",

                    result.fail_reason, result.output)

    return not result.failed

  def Rename(self, new_id):

    """Rename this logical volume.

    """

    if not isinstance(new_id, (tuple, list)) or len(new_id) != 2:

      raise errors.ProgrammerError("Invalid new logical id '%s'" % new_id)

    new_vg, new_name = new_id

    if new_vg != self._vg_name:

      raise errors.ProgrammerError("Can't move a logical volume across"

                                   " volume groups (from %s to to %s)" %

                                   (self._vg_name, new_vg))

    result = utils.RunCmd(["lvrename", new_vg, self._lv_name, new_name])

    if result.failed:

      raise errors.BlockDeviceError("Failed to rename the logical volume: %s" %

                                    result.output)

    self._lv_name = new_name

    self.dev_path = "/dev/%s/%s" % (self._vg_name, self._lv_name)

  def Attach(self):

    """Attach to an existing LV.

    This method will try to see if an existing and active LV exists

    which matches our name. If so, its major/minor will be

    recorded.

    """

    self.attached = False

    result = utils.RunCmd(["lvs", "--noheadings", "--separator=,",

                           "-olv_attr,lv_kernel_major,lv_kernel_minor",

                           self.dev_path])

    if result.failed:

      logging.error("Can't find LV %s: %s, %s",

                    self.dev_path, result.fail_reason, result.output)

      return False

    out = result.stdout.strip().rstrip(',')

    out = out.split(",")

    if len(out) != 3:

      logging.error("Can't parse LVS output, len(%s) != 3", str(out))

      return False

    status, major, minor = out[:3]

    if len(status) != 6:

      logging.error("lvs lv_attr is not 6 characters (%s)", status)

      return False

    try:

      major = int(major)

      minor = int(minor)

    except ValueError, err:

      logging.error("lvs major/minor cannot be parsed: %s", str(err))

    self.major = major

    self.minor = minor

    self._degraded = status[0] == 'v' # virtual volume, i.e. doesn't backing

                                      # storage

    self.attached = True

    return True

  def Assemble(self):

    """Assemble the device.

    We alway run `lvchange -ay` on the LV to ensure it's active before

    use, as there were cases when xenvg was not active after boot

    (also possibly after disk issues).

    """

    result = utils.RunCmd(["lvchange", "-ay", self.dev_path])

    if result.failed:

      logging.error("Can't activate lv %s: %s", self.dev_path, result.output)

      return False

    return self.Attach()

  def Shutdown(self):

    """Shutdown the device.

    This is a no-op for the LV device type, as we don't deactivate the

    volumes on shutdown.

    """

    return True

  def GetSyncStatus(self):

    """Returns the sync status of the device.

    If this device is a mirroring device, this function returns the

    status of the mirror.

    For logical volumes, sync_percent and estimated_time are always

    None (no recovery in progress, as we don't handle the mirrored LV

    case). The is_degraded parameter is the inverse of the ldisk

    parameter.

    For the ldisk parameter, we check if the logical volume has the

    'virtual' type, which means it's not backed by existing storage

    anymore (read from it return I/O error). This happens after a

    physical disk failure and subsequent 'vgreduce --removemissing' on

    the volume group.

    The status was already read in Attach, so we just return it.

    @rtype: tuple

    @return: (sync_percent, estimated_time, is_degraded, ldisk)

    """

    return None, None, self._degraded, self._degraded

  def Open(self, force=False):

    """Make the device ready for I/O.

    This is a no-op for the LV device type.

    """

    pass

  def Close(self):

    """Notifies that the device will no longer be used for I/O.

    This is a no-op for the LV device type.

    """

    pass

  def Snapshot(self, size):

    """Create a snapshot copy of an lvm block device.

    """

    snap_name = self._lv_name + ".snap"

    # remove existing snapshot if found

    snap = LogicalVolume((self._vg_name, snap_name), None)

    snap.Remove()

    pvs_info = self.GetPVInfo(self._vg_name)

    if not pvs_info:

      raise errors.BlockDeviceError("Can't compute PV info for vg %s" %

                                    self._vg_name)

    pvs_info.sort()

    pvs_info.reverse()

    free_size, pv_name = pvs_info[0]

    if free_size < size:

      raise errors.BlockDeviceError("Not enough free space: required %s,"

                                    " available %s" % (size, free_size))

    result = utils.RunCmd(["lvcreate", "-L%dm" % size, "-s",

                           "-n%s" % snap_name, self.dev_path])

    if result.failed:

      raise errors.BlockDeviceError("command: %s error: %s - %s" %

                                    (result.cmd, result.fail_reason,

                                     result.output))

    return snap_name

  def SetInfo(self, text):

    """Update metadata with info text.

    """

    BlockDev.SetInfo(self, text)

    # Replace invalid characters

    text = re.sub('^[^A-Za-z0-9_+.]', '_', text)

    text = re.sub('[^-A-Za-z0-9_+.]', '_', text)

    # Only up to 128 characters are allowed

    text = text[:128]

    result = utils.RunCmd(["lvchange", "--addtag", text,

                           self.dev_path])

    if result.failed:

      raise errors.BlockDeviceError("Command: %s error: %s - %s" %

                                    (result.cmd, result.fail_reason,

                                     result.output))

  def Grow(self, amount):

    """Grow the logical volume.

    """

    # we try multiple algorithms since the 'best' ones might not have

    # space available in the right place, but later ones might (since

    # they have less constraints); also note that only recent LVM

    # supports 'cling'

    for alloc_policy in "contiguous", "cling", "normal":

      result = utils.RunCmd(["lvextend", "--alloc", alloc_policy,

                             "-L", "+%dm" % amount, self.dev_path])

      if not result.failed:

        return

    raise errors.BlockDeviceError("Can't grow LV %s: %s" %

                                  (self.dev_path, result.output))

class DRBD8Status(object):

  """A DRBD status representation class.

  Note that this doesn't support unconfigured devices (cs:Unconfigured).

  """

  LINE_RE = re.compile(r"\s*[0-9]+:\s*cs:(\S+)\s+st:([^/]+)/(\S+)"

                       "\s+ds:([^/]+)/(\S+)\s+.*$")

  SYNC_RE = re.compile(r"^.*\ssync'ed:\s*([0-9.]+)%.*"

                       "\sfinish: ([0-9]+):([0-9]+):([0-9]+)\s.*$")

  def __init__(self, procline):

    m = self.LINE_RE.match(procline)

    if not m:

      raise errors.BlockDeviceError("Can't parse input data '%s'" % procline)

    self.cstatus = m.group(1)

    self.lrole = m.group(2)

    self.rrole = m.group(3)

    self.ldisk = m.group(4)

    self.rdisk = m.group(5)

    self.is_standalone = self.cstatus == "StandAlone"

    self.is_wfconn = self.cstatus == "WFConnection"

    self.is_connected = self.cstatus == "Connected"

    self.is_primary = self.lrole == "Primary"

    self.is_secondary = self.lrole == "Secondary"

    self.peer_primary = self.rrole == "Primary"

    self.peer_secondary = self.rrole == "Secondary"

    self.both_primary = self.is_primary and self.peer_primary

    self.both_secondary = self.is_secondary and self.peer_secondary

    self.is_diskless = self.ldisk == "Diskless"

    self.is_disk_uptodate = self.ldisk == "UpToDate"

    m = self.SYNC_RE.match(procline)

    if m:

      self.sync_percent = float(m.group(1))

      hours = int(m.group(2))

      minutes = int(m.group(3))

      seconds = int(m.group(4))

      self.est_time = hours * 3600 + minutes * 60 + seconds

    else:

      self.sync_percent = None

      self.est_time = None

    self.is_sync_target = self.peer_sync_source = self.cstatus == "SyncTarget"

    self.peer_sync_target = self.is_sync_source = self.cstatus == "SyncSource"

    self.is_resync = self.is_sync_target or self.is_sync_source

class BaseDRBD(BlockDev):

  """Base DRBD class.

  This class contains a few bits of common functionality between the

  0.7 and 8.x versions of DRBD.

  """

  _VERSION_RE = re.compile(r"^version: (\d+)\.(\d+)\.(\d+)"

                           r" \(api:(\d+)/proto:(\d+)(?:-(\d+))?\)")

  _DRBD_MAJOR = 147

  _ST_UNCONFIGURED = "Unconfigured"

  _ST_WFCONNECTION = "WFConnection"

  _ST_CONNECTED = "Connected"

  _STATUS_FILE = "/proc/drbd"

  @staticmethod

  def _GetProcData(filename=_STATUS_FILE):

    """Return data from /proc/drbd.

    """

    stat = open(filename, "r")

    try:

      data = stat.read().splitlines()

    finally:

      stat.close()

    if not data:

      raise errors.BlockDeviceError("Can't read any data from %s" % filename)

    return data

  @staticmethod

  def _MassageProcData(data):

    """Transform the output of _GetProdData into a nicer form.

    @return: a dictionary of minor: joined lines from /proc/drbd

        for that minor

    """

    lmatch = re.compile("^ *([0-9]+):.*$")

    results = {}

    old_minor = old_line = None

    for line in data:

      lresult = lmatch.match(line)

      if lresult is not None:

        if old_minor is not None:

          results[old_minor] = old_line

        old_minor = int(lresult.group(1))

        old_line = line

      else:

        if old_minor is not None:

          old_line += " " + line.strip()

    # add last line

    if old_minor is not None:

      results[old_minor] = old_line

    return results

  @classmethod

  def _GetVersion(cls):

    """Return the DRBD version.

    This will return a dict with keys:

      - k_major

      - k_minor

      - k_point

      - api

      - proto

      - proto2 (only on drbd > 8.2.X)

    """

    proc_data = cls._GetProcData()

    first_line = proc_data[0].strip()

    version = cls._VERSION_RE.match(first_line)

    if not version:

      raise errors.BlockDeviceError("Can't parse DRBD version from '%s'" %

                                    first_line)

    values = version.groups()

    retval = {'k_major': int(values[0]),

              'k_minor': int(values[1]),

              'k_point': int(values[2]),

              'api': int(values[3]),

              'proto': int(values[4]),

             }

    if values[5] is not None:

      retval['proto2'] = values[5]

    return retval

  @staticmethod

  def _DevPath(minor):

    """Return the path to a drbd device for a given minor.

    """

    return "/dev/drbd%d" % minor

  @classmethod

  def _GetUsedDevs(cls):

    """Compute the list of used DRBD devices.

    """

    data = cls._GetProcData()

    used_devs = {}

    valid_line = re.compile("^ *([0-9]+): cs:([^ ]+).*$")