Source code for xcp.dom0

# Copyright (c) 2013, Citrix Inc.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# 1. Redistributions of source code must retain the above copyright notice, this
#    list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright notice,
#    this list of conditions and the following disclaimer in the documentation
#    and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
# ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

from __future__ import absolute_import
from __future__ import division
import re

from . import version
from .compat import open_with_codec_handling



def default_memory_v2(host_mem_kib):
    """Return the default for the amount of dom0 memory for the
    specified amount of host memory for platform versions < 2.9.0."""

    #
    # The host memory reported by Xen is a bit less than the physical
    # RAM installed in the machine since it doesn't include the memory
    # used by Xen etc.
    #
    # Add a bit extra to account for this.
    #
    gb = (host_mem_kib + 256 * 1024) // 1024 // 1024

    if gb < 24:
        return 752 * 1024
    if gb < 48:
        return 2 * 1024 * 1024
    if gb < 64:
        return 3 * 1024 * 1024
    return 4 * 1024 * 1024




def default_memory_v3(host_mem_kib):
    """Return the default for the amount of dom0 memory for the
    specified amount of host memory for platform versions >= 2.9.0."""

    #
    # The host memory reported by Xen is a bit less than the physical
    # RAM installed in the machine since it doesn't include the memory
    # used by Xen etc.
    #
    # Add a bit extra to account for this.
    #
    mb = (host_mem_kib + 256 * 1024) // 1024

    # Give dom0 1 GiB + 5% of host memory, rounded to 16 MiB, limited to 8 GiB
    return min(1024 + int(mb * 0.05) & ~0xF, 8192) * 1024




def default_memory_for_version(host_mem_kib, platform_version):
    """Return the default for the amount of dom0 memory for the
    specified amount of host memory for the given platform version."""
    if platform_version < version.Version([2, 9, 0]):
        return default_memory_v2(host_mem_kib)
    else:
        return default_memory_v3(host_mem_kib)




def default_memory(host_mem_kib):
    """Return the default for the amount of dom0 memory for the
    specified amount of host memory for the current platform version"""

    # read current host version
    platform_version = None
    with open_with_codec_handling("/etc/xensource-inventory") as f:
        for l in f.readlines():
            line = l.strip()
            if line.startswith('PLATFORM_VERSION='):
                platform_version = version.Version.from_string(
                                   line.split('=', 1)[1].strip("'"))
                break

    if not platform_version:
        raise RuntimeError('Could not find PLATFORM_VERSION from inventory.')

    return default_memory_for_version(host_mem_kib, platform_version)



_size_and_unit_re = re.compile(r"^(-?\d+)([bkmg]?)$", re.IGNORECASE)

def _parse_size_and_unit(s):
    m = _size_and_unit_re.match(s)
    if not m:
        return None

    val = int(m.group(1))
    unit = m.group(2).lower()

    if unit == "g":
        val *= 1024*1024*1024
    elif unit == "m":
        val *= 1024*1024
    elif unit in ("k", ""):
        val *= 1024

    return val



def parse_mem(arg):
    """Parse Xen's dom0_mem command line option.

    Return tuple of (amount, min, max) memory in bytes from a string
    in the following format:

        dom0_mem=[min:<min_amt>,][max:<max_amt>,][<amt>]

    See also Xen's docs/txt/misc/xen-command-line.txt."""

    t = arg.split("=")
    if len(t) < 2 or t[0] != "dom0_mem":
        return (None, None, None)

    dom0_mem = None
    dom0_mem_min = None
    dom0_mem_max = None

    #
    # This is an equivalent to the parse_dom0_mem() call in
    # xen/arch/x86/domain_build.c
    #
    for s in t[1].split(","):
        if s.startswith("min:"):
            dom0_mem_min = _parse_size_and_unit(s[4:])
        elif s.startswith("max:"):
            dom0_mem_max = _parse_size_and_unit(s[4:])
        else:
            dom0_mem = _parse_size_and_unit(s)

    return (dom0_mem, dom0_mem_min, dom0_mem_max)




def default_vcpus(host_pcpus, dom0_mem_mb = None):
    """Return the default number of dom0 vcpus for the specified number
    of host pcpus and the amount of dom0 memory."""

    max_vcpus = 16

    # Calculate max number of vCPUs
    # based on the amount of available memory
    if dom0_mem_mb is not None:
        if dom0_mem_mb < 2 * 1024:
            max_vcpus = 4
        elif dom0_mem_mb < 4 * 1024:
            max_vcpus = 8

    # Special case (minimum)
    if host_pcpus == 0:
        return 1

    # vCPUs = host_pcpus for host pcpus <= 16
    if host_pcpus <= 16:
        return min(host_pcpus, max_vcpus)

    # 16 for anything greater than 16
    return min(16, max_vcpus)