Statistics
| Branch: | Tag: | Revision:

root / lib / hypervisor / hv_kvm.py @ 1498270e

History | View | Annotate | Download (78.3 kB)

1 eb58f9b1 Guido Trotter
#
2 eb58f9b1 Guido Trotter
#
3 eb58f9b1 Guido Trotter
4 1f4b9d39 Iustin Pop
# Copyright (C) 2008, 2009, 2010, 2011, 2012, 2013 Google Inc.
5 eb58f9b1 Guido Trotter
#
6 eb58f9b1 Guido Trotter
# This program is free software; you can redistribute it and/or modify
7 eb58f9b1 Guido Trotter
# it under the terms of the GNU General Public License as published by
8 eb58f9b1 Guido Trotter
# the Free Software Foundation; either version 2 of the License, or
9 eb58f9b1 Guido Trotter
# (at your option) any later version.
10 eb58f9b1 Guido Trotter
#
11 eb58f9b1 Guido Trotter
# This program is distributed in the hope that it will be useful, but
12 eb58f9b1 Guido Trotter
# WITHOUT ANY WARRANTY; without even the implied warranty of
13 eb58f9b1 Guido Trotter
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 eb58f9b1 Guido Trotter
# General Public License for more details.
15 eb58f9b1 Guido Trotter
#
16 eb58f9b1 Guido Trotter
# You should have received a copy of the GNU General Public License
17 eb58f9b1 Guido Trotter
# along with this program; if not, write to the Free Software
18 eb58f9b1 Guido Trotter
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
19 eb58f9b1 Guido Trotter
# 02110-1301, USA.
20 eb58f9b1 Guido Trotter
21 eb58f9b1 Guido Trotter
22 eb58f9b1 Guido Trotter
"""KVM hypervisor
23 eb58f9b1 Guido Trotter

24 eb58f9b1 Guido Trotter
"""
25 eb58f9b1 Guido Trotter
26 547a63b7 Balazs Lecz
import errno
27 eb58f9b1 Guido Trotter
import os
28 eb58f9b1 Guido Trotter
import os.path
29 eb58f9b1 Guido Trotter
import re
30 eb58f9b1 Guido Trotter
import tempfile
31 6567aff3 Guido Trotter
import time
32 30e42c4e Guido Trotter
import logging
33 d19d94db Guido Trotter
import pwd
34 199b2053 Apollon Oikonomopoulos
import struct
35 199b2053 Apollon Oikonomopoulos
import fcntl
36 5d9bfd87 Apollon Oikonomopoulos
import shutil
37 91c10532 Andrea Spadaccini
import socket
38 fc84cd5d Guido Trotter
import stat
39 91c10532 Andrea Spadaccini
import StringIO
40 b693125f Tsachy Shacham
try:
41 471a31b6 Andrea Spadaccini
  import affinity   # pylint: disable=F0401
42 b693125f Tsachy Shacham
except ImportError:
43 b693125f Tsachy Shacham
  affinity = None
44 eb58f9b1 Guido Trotter
45 eb58f9b1 Guido Trotter
from ganeti import utils
46 eb58f9b1 Guido Trotter
from ganeti import constants
47 eb58f9b1 Guido Trotter
from ganeti import errors
48 38e250ba Guido Trotter
from ganeti import serializer
49 38e250ba Guido Trotter
from ganeti import objects
50 76431533 Guido Trotter
from ganeti import uidpool
51 76431533 Guido Trotter
from ganeti import ssconf
52 a744b676 Manuel Franceschini
from ganeti import netutils
53 9d9bded1 Michael Hanselmann
from ganeti import pathutils
54 9d9bded1 Michael Hanselmann
from ganeti.hypervisor import hv_base
55 5d9bfd87 Apollon Oikonomopoulos
from ganeti.utils import wrapper as utils_wrapper
56 eb58f9b1 Guido Trotter
57 eb58f9b1 Guido Trotter
58 0e1e0b6a Michael Hanselmann
_KVM_NETWORK_SCRIPT = pathutils.CONF_DIR + "/kvm-vif-bridge"
59 b693125f Tsachy Shacham
_KVM_START_PAUSED_FLAG = "-S"
60 748e4b5a Michael Hanselmann
61 199b2053 Apollon Oikonomopoulos
# TUN/TAP driver constants, taken from <linux/if_tun.h>
62 199b2053 Apollon Oikonomopoulos
# They are architecture-independent and already hardcoded in qemu-kvm source,
63 199b2053 Apollon Oikonomopoulos
# so we can safely include them here.
64 199b2053 Apollon Oikonomopoulos
TUNSETIFF = 0x400454ca
65 199b2053 Apollon Oikonomopoulos
TUNGETIFF = 0x800454d2
66 199b2053 Apollon Oikonomopoulos
TUNGETFEATURES = 0x800454cf
67 199b2053 Apollon Oikonomopoulos
IFF_TAP = 0x0002
68 199b2053 Apollon Oikonomopoulos
IFF_NO_PI = 0x1000
69 199b2053 Apollon Oikonomopoulos
IFF_VNET_HDR = 0x4000
70 199b2053 Apollon Oikonomopoulos
71 07788a0b Michael Hanselmann
#: SPICE parameters which depend on L{constants.HV_KVM_SPICE_BIND}
72 07788a0b Michael Hanselmann
_SPICE_ADDITIONAL_PARAMS = frozenset([
73 07788a0b Michael Hanselmann
  constants.HV_KVM_SPICE_IP_VERSION,
74 07788a0b Michael Hanselmann
  constants.HV_KVM_SPICE_PASSWORD_FILE,
75 07788a0b Michael Hanselmann
  constants.HV_KVM_SPICE_LOSSLESS_IMG_COMPR,
76 07788a0b Michael Hanselmann
  constants.HV_KVM_SPICE_JPEG_IMG_COMPR,
77 07788a0b Michael Hanselmann
  constants.HV_KVM_SPICE_ZLIB_GLZ_IMG_COMPR,
78 07788a0b Michael Hanselmann
  constants.HV_KVM_SPICE_STREAMING_VIDEO_DETECTION,
79 07788a0b Michael Hanselmann
  constants.HV_KVM_SPICE_USE_TLS,
80 07788a0b Michael Hanselmann
  ])
81 07788a0b Michael Hanselmann
82 199b2053 Apollon Oikonomopoulos
83 199b2053 Apollon Oikonomopoulos
def _ProbeTapVnetHdr(fd):
84 199b2053 Apollon Oikonomopoulos
  """Check whether to enable the IFF_VNET_HDR flag.
85 199b2053 Apollon Oikonomopoulos

86 199b2053 Apollon Oikonomopoulos
  To do this, _all_ of the following conditions must be met:
87 199b2053 Apollon Oikonomopoulos
   1. TUNGETFEATURES ioctl() *must* be implemented
88 199b2053 Apollon Oikonomopoulos
   2. TUNGETFEATURES ioctl() result *must* contain the IFF_VNET_HDR flag
89 199b2053 Apollon Oikonomopoulos
   3. TUNGETIFF ioctl() *must* be implemented; reading the kernel code in
90 199b2053 Apollon Oikonomopoulos
      drivers/net/tun.c there is no way to test this until after the tap device
91 199b2053 Apollon Oikonomopoulos
      has been created using TUNSETIFF, and there is no way to change the
92 199b2053 Apollon Oikonomopoulos
      IFF_VNET_HDR flag after creating the interface, catch-22! However both
93 199b2053 Apollon Oikonomopoulos
      TUNGETIFF and TUNGETFEATURES were introduced in kernel version 2.6.27,
94 199b2053 Apollon Oikonomopoulos
      thus we can expect TUNGETIFF to be present if TUNGETFEATURES is.
95 199b2053 Apollon Oikonomopoulos

96 199b2053 Apollon Oikonomopoulos
   @type fd: int
97 199b2053 Apollon Oikonomopoulos
   @param fd: the file descriptor of /dev/net/tun
98 199b2053 Apollon Oikonomopoulos

99 199b2053 Apollon Oikonomopoulos
  """
100 199b2053 Apollon Oikonomopoulos
  req = struct.pack("I", 0)
101 199b2053 Apollon Oikonomopoulos
  try:
102 199b2053 Apollon Oikonomopoulos
    res = fcntl.ioctl(fd, TUNGETFEATURES, req)
103 199b2053 Apollon Oikonomopoulos
  except EnvironmentError:
104 199b2053 Apollon Oikonomopoulos
    logging.warning("TUNGETFEATURES ioctl() not implemented")
105 199b2053 Apollon Oikonomopoulos
    return False
106 199b2053 Apollon Oikonomopoulos
107 199b2053 Apollon Oikonomopoulos
  tunflags = struct.unpack("I", res)[0]
108 199b2053 Apollon Oikonomopoulos
  if tunflags & IFF_VNET_HDR:
109 199b2053 Apollon Oikonomopoulos
    return True
110 199b2053 Apollon Oikonomopoulos
  else:
111 199b2053 Apollon Oikonomopoulos
    logging.warning("Host does not support IFF_VNET_HDR, not enabling")
112 199b2053 Apollon Oikonomopoulos
    return False
113 199b2053 Apollon Oikonomopoulos
114 199b2053 Apollon Oikonomopoulos
115 199b2053 Apollon Oikonomopoulos
def _OpenTap(vnet_hdr=True):
116 199b2053 Apollon Oikonomopoulos
  """Open a new tap device and return its file descriptor.
117 199b2053 Apollon Oikonomopoulos

118 199b2053 Apollon Oikonomopoulos
  This is intended to be used by a qemu-type hypervisor together with the -net
119 199b2053 Apollon Oikonomopoulos
  tap,fd=<fd> command line parameter.
120 199b2053 Apollon Oikonomopoulos

121 199b2053 Apollon Oikonomopoulos
  @type vnet_hdr: boolean
122 199b2053 Apollon Oikonomopoulos
  @param vnet_hdr: Enable the VNET Header
123 199b2053 Apollon Oikonomopoulos
  @return: (ifname, tapfd)
124 199b2053 Apollon Oikonomopoulos
  @rtype: tuple
125 199b2053 Apollon Oikonomopoulos

126 199b2053 Apollon Oikonomopoulos
  """
127 199b2053 Apollon Oikonomopoulos
  try:
128 199b2053 Apollon Oikonomopoulos
    tapfd = os.open("/dev/net/tun", os.O_RDWR)
129 199b2053 Apollon Oikonomopoulos
  except EnvironmentError:
130 199b2053 Apollon Oikonomopoulos
    raise errors.HypervisorError("Failed to open /dev/net/tun")
131 199b2053 Apollon Oikonomopoulos
132 199b2053 Apollon Oikonomopoulos
  flags = IFF_TAP | IFF_NO_PI
133 199b2053 Apollon Oikonomopoulos
134 199b2053 Apollon Oikonomopoulos
  if vnet_hdr and _ProbeTapVnetHdr(tapfd):
135 199b2053 Apollon Oikonomopoulos
    flags |= IFF_VNET_HDR
136 199b2053 Apollon Oikonomopoulos
137 199b2053 Apollon Oikonomopoulos
  # The struct ifreq ioctl request (see netdevice(7))
138 199b2053 Apollon Oikonomopoulos
  ifr = struct.pack("16sh", "", flags)
139 199b2053 Apollon Oikonomopoulos
140 199b2053 Apollon Oikonomopoulos
  try:
141 199b2053 Apollon Oikonomopoulos
    res = fcntl.ioctl(tapfd, TUNSETIFF, ifr)
142 6f1e1921 Michael Hanselmann
  except EnvironmentError, err:
143 6f1e1921 Michael Hanselmann
    raise errors.HypervisorError("Failed to allocate a new TAP device: %s" %
144 6f1e1921 Michael Hanselmann
                                 err)
145 199b2053 Apollon Oikonomopoulos
146 199b2053 Apollon Oikonomopoulos
  # Get the interface name from the ioctl
147 199b2053 Apollon Oikonomopoulos
  ifname = struct.unpack("16sh", res)[0].strip("\x00")
148 199b2053 Apollon Oikonomopoulos
  return (ifname, tapfd)
149 199b2053 Apollon Oikonomopoulos
150 748e4b5a Michael Hanselmann
151 91c10532 Andrea Spadaccini
class QmpMessage:
152 91c10532 Andrea Spadaccini
  """QEMU Messaging Protocol (QMP) message.
153 91c10532 Andrea Spadaccini

154 91c10532 Andrea Spadaccini
  """
155 91c10532 Andrea Spadaccini
  def __init__(self, data):
156 91c10532 Andrea Spadaccini
    """Creates a new QMP message based on the passed data.
157 91c10532 Andrea Spadaccini

158 91c10532 Andrea Spadaccini
    """
159 91c10532 Andrea Spadaccini
    if not isinstance(data, dict):
160 91c10532 Andrea Spadaccini
      raise TypeError("QmpMessage must be initialized with a dict")
161 91c10532 Andrea Spadaccini
162 91c10532 Andrea Spadaccini
    self.data = data
163 91c10532 Andrea Spadaccini
164 91c10532 Andrea Spadaccini
  def __getitem__(self, field_name):
165 91c10532 Andrea Spadaccini
    """Get the value of the required field if present, or None.
166 91c10532 Andrea Spadaccini

167 91c10532 Andrea Spadaccini
    Overrides the [] operator to provide access to the message data,
168 91c10532 Andrea Spadaccini
    returning None if the required item is not in the message
169 91c10532 Andrea Spadaccini
    @return: the value of the field_name field, or None if field_name
170 91c10532 Andrea Spadaccini
             is not contained in the message
171 91c10532 Andrea Spadaccini

172 91c10532 Andrea Spadaccini
    """
173 9b87c2cf Michael Hanselmann
    return self.data.get(field_name, None)
174 91c10532 Andrea Spadaccini
175 91c10532 Andrea Spadaccini
  def __setitem__(self, field_name, field_value):
176 91c10532 Andrea Spadaccini
    """Set the value of the required field_name to field_value.
177 91c10532 Andrea Spadaccini

178 91c10532 Andrea Spadaccini
    """
179 91c10532 Andrea Spadaccini
    self.data[field_name] = field_value
180 91c10532 Andrea Spadaccini
181 91c10532 Andrea Spadaccini
  @staticmethod
182 91c10532 Andrea Spadaccini
  def BuildFromJsonString(json_string):
183 91c10532 Andrea Spadaccini
    """Build a QmpMessage from a JSON encoded string.
184 91c10532 Andrea Spadaccini

185 91c10532 Andrea Spadaccini
    @type json_string: str
186 91c10532 Andrea Spadaccini
    @param json_string: JSON string representing the message
187 91c10532 Andrea Spadaccini
    @rtype: L{QmpMessage}
188 91c10532 Andrea Spadaccini
    @return: a L{QmpMessage} built from json_string
189 91c10532 Andrea Spadaccini

190 91c10532 Andrea Spadaccini
    """
191 91c10532 Andrea Spadaccini
    # Parse the string
192 91c10532 Andrea Spadaccini
    data = serializer.LoadJson(json_string)
193 91c10532 Andrea Spadaccini
    return QmpMessage(data)
194 91c10532 Andrea Spadaccini
195 91c10532 Andrea Spadaccini
  def __str__(self):
196 a182a3ed Michael Hanselmann
    # The protocol expects the JSON object to be sent as a single line.
197 a182a3ed Michael Hanselmann
    return serializer.DumpJson(self.data)
198 91c10532 Andrea Spadaccini
199 91c10532 Andrea Spadaccini
  def __eq__(self, other):
200 91c10532 Andrea Spadaccini
    # When comparing two QmpMessages, we are interested in comparing
201 91c10532 Andrea Spadaccini
    # their internal representation of the message data
202 91c10532 Andrea Spadaccini
    return self.data == other.data
203 91c10532 Andrea Spadaccini
204 91c10532 Andrea Spadaccini
205 91c10532 Andrea Spadaccini
class QmpConnection:
206 91c10532 Andrea Spadaccini
  """Connection to the QEMU Monitor using the QEMU Monitor Protocol (QMP).
207 91c10532 Andrea Spadaccini

208 91c10532 Andrea Spadaccini
  """
209 91c10532 Andrea Spadaccini
  _FIRST_MESSAGE_KEY = "QMP"
210 91c10532 Andrea Spadaccini
  _EVENT_KEY = "event"
211 91c10532 Andrea Spadaccini
  _ERROR_KEY = "error"
212 89da2ff3 Guido Trotter
  _RETURN_KEY = RETURN_KEY = "return"
213 89da2ff3 Guido Trotter
  _ACTUAL_KEY = ACTUAL_KEY = "actual"
214 91c10532 Andrea Spadaccini
  _ERROR_CLASS_KEY = "class"
215 91c10532 Andrea Spadaccini
  _ERROR_DATA_KEY = "data"
216 91c10532 Andrea Spadaccini
  _ERROR_DESC_KEY = "desc"
217 91c10532 Andrea Spadaccini
  _EXECUTE_KEY = "execute"
218 91c10532 Andrea Spadaccini
  _ARGUMENTS_KEY = "arguments"
219 91c10532 Andrea Spadaccini
  _CAPABILITIES_COMMAND = "qmp_capabilities"
220 91c10532 Andrea Spadaccini
  _MESSAGE_END_TOKEN = "\r\n"
221 91c10532 Andrea Spadaccini
  _SOCKET_TIMEOUT = 5
222 91c10532 Andrea Spadaccini
223 91c10532 Andrea Spadaccini
  def __init__(self, monitor_filename):
224 91c10532 Andrea Spadaccini
    """Instantiates the QmpConnection object.
225 91c10532 Andrea Spadaccini

226 91c10532 Andrea Spadaccini
    @type monitor_filename: string
227 91c10532 Andrea Spadaccini
    @param monitor_filename: the filename of the UNIX raw socket on which the
228 91c10532 Andrea Spadaccini
                             QMP monitor is listening
229 91c10532 Andrea Spadaccini

230 91c10532 Andrea Spadaccini
    """
231 91c10532 Andrea Spadaccini
    self.monitor_filename = monitor_filename
232 91c10532 Andrea Spadaccini
    self.sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
233 91c10532 Andrea Spadaccini
    # We want to fail if the server doesn't send a complete message
234 91c10532 Andrea Spadaccini
    # in a reasonable amount of time
235 91c10532 Andrea Spadaccini
    self.sock.settimeout(self._SOCKET_TIMEOUT)
236 91c10532 Andrea Spadaccini
    self._connected = False
237 91c10532 Andrea Spadaccini
    self._buf = ""
238 91c10532 Andrea Spadaccini
239 fc84cd5d Guido Trotter
  def _check_socket(self):
240 fc84cd5d Guido Trotter
    sock_stat = None
241 fc84cd5d Guido Trotter
    try:
242 fc84cd5d Guido Trotter
      sock_stat = os.stat(self.monitor_filename)
243 fc84cd5d Guido Trotter
    except EnvironmentError, err:
244 fc84cd5d Guido Trotter
      if err.errno == errno.ENOENT:
245 fc84cd5d Guido Trotter
        raise errors.HypervisorError("No qmp socket found")
246 fc84cd5d Guido Trotter
      else:
247 fc84cd5d Guido Trotter
        raise errors.HypervisorError("Error checking qmp socket: %s",
248 fc84cd5d Guido Trotter
                                     utils.ErrnoOrStr(err))
249 fc84cd5d Guido Trotter
    if not stat.S_ISSOCK(sock_stat.st_mode):
250 fc84cd5d Guido Trotter
      raise errors.HypervisorError("Qmp socket is not a socket")
251 fc84cd5d Guido Trotter
252 91c10532 Andrea Spadaccini
  def _check_connection(self):
253 91c10532 Andrea Spadaccini
    """Make sure that the connection is established.
254 91c10532 Andrea Spadaccini

255 91c10532 Andrea Spadaccini
    """
256 91c10532 Andrea Spadaccini
    if not self._connected:
257 91c10532 Andrea Spadaccini
      raise errors.ProgrammerError("To use a QmpConnection you need to first"
258 91c10532 Andrea Spadaccini
                                   " invoke connect() on it")
259 91c10532 Andrea Spadaccini
260 91c10532 Andrea Spadaccini
  def connect(self):
261 91c10532 Andrea Spadaccini
    """Connects to the QMP monitor.
262 91c10532 Andrea Spadaccini

263 91c10532 Andrea Spadaccini
    Connects to the UNIX socket and makes sure that we can actually send and
264 91c10532 Andrea Spadaccini
    receive data to the kvm instance via QMP.
265 91c10532 Andrea Spadaccini

266 91c10532 Andrea Spadaccini
    @raise errors.HypervisorError: when there are communication errors
267 91c10532 Andrea Spadaccini
    @raise errors.ProgrammerError: when there are data serialization errors
268 91c10532 Andrea Spadaccini

269 91c10532 Andrea Spadaccini
    """
270 fc84cd5d Guido Trotter
    if self._connected:
271 fc84cd5d Guido Trotter
      raise errors.ProgrammerError("Cannot connect twice")
272 fc84cd5d Guido Trotter
273 fc84cd5d Guido Trotter
    self._check_socket()
274 fc84cd5d Guido Trotter
275 fc84cd5d Guido Trotter
    # Check file existance/stuff
276 fc84cd5d Guido Trotter
    try:
277 fc84cd5d Guido Trotter
      self.sock.connect(self.monitor_filename)
278 fc84cd5d Guido Trotter
    except EnvironmentError:
279 fc84cd5d Guido Trotter
      raise errors.HypervisorError("Can't connect to qmp socket")
280 91c10532 Andrea Spadaccini
    self._connected = True
281 91c10532 Andrea Spadaccini
282 91c10532 Andrea Spadaccini
    # Check if we receive a correct greeting message from the server
283 91c10532 Andrea Spadaccini
    # (As per the QEMU Protocol Specification 0.1 - section 2.2)
284 91c10532 Andrea Spadaccini
    greeting = self._Recv()
285 91c10532 Andrea Spadaccini
    if not greeting[self._FIRST_MESSAGE_KEY]:
286 91c10532 Andrea Spadaccini
      self._connected = False
287 91c10532 Andrea Spadaccini
      raise errors.HypervisorError("kvm: qmp communication error (wrong"
288 91c10532 Andrea Spadaccini
                                   " server greeting")
289 91c10532 Andrea Spadaccini
290 91c10532 Andrea Spadaccini
    # Let's put the monitor in command mode using the qmp_capabilities
291 91c10532 Andrea Spadaccini
    # command, or else no command will be executable.
292 91c10532 Andrea Spadaccini
    # (As per the QEMU Protocol Specification 0.1 - section 4)
293 91c10532 Andrea Spadaccini
    self.Execute(self._CAPABILITIES_COMMAND)
294 91c10532 Andrea Spadaccini
295 91c10532 Andrea Spadaccini
  def _ParseMessage(self, buf):
296 91c10532 Andrea Spadaccini
    """Extract and parse a QMP message from the given buffer.
297 91c10532 Andrea Spadaccini

298 91c10532 Andrea Spadaccini
    Seeks for a QMP message in the given buf. If found, it parses it and
299 91c10532 Andrea Spadaccini
    returns it together with the rest of the characters in the buf.
300 91c10532 Andrea Spadaccini
    If no message is found, returns None and the whole buffer.
301 91c10532 Andrea Spadaccini

302 91c10532 Andrea Spadaccini
    @raise errors.ProgrammerError: when there are data serialization errors
303 91c10532 Andrea Spadaccini

304 91c10532 Andrea Spadaccini
    """
305 91c10532 Andrea Spadaccini
    message = None
306 91c10532 Andrea Spadaccini
    # Check if we got the message end token (CRLF, as per the QEMU Protocol
307 91c10532 Andrea Spadaccini
    # Specification 0.1 - Section 2.1.1)
308 91c10532 Andrea Spadaccini
    pos = buf.find(self._MESSAGE_END_TOKEN)
309 91c10532 Andrea Spadaccini
    if pos >= 0:
310 91c10532 Andrea Spadaccini
      try:
311 91c10532 Andrea Spadaccini
        message = QmpMessage.BuildFromJsonString(buf[:pos + 1])
312 91c10532 Andrea Spadaccini
      except Exception, err:
313 91c10532 Andrea Spadaccini
        raise errors.ProgrammerError("QMP data serialization error: %s" % err)
314 91c10532 Andrea Spadaccini
      buf = buf[pos + 1:]
315 91c10532 Andrea Spadaccini
316 91c10532 Andrea Spadaccini
    return (message, buf)
317 91c10532 Andrea Spadaccini
318 91c10532 Andrea Spadaccini
  def _Recv(self):
319 91c10532 Andrea Spadaccini
    """Receives a message from QMP and decodes the received JSON object.
320 91c10532 Andrea Spadaccini

321 91c10532 Andrea Spadaccini
    @rtype: QmpMessage
322 91c10532 Andrea Spadaccini
    @return: the received message
323 91c10532 Andrea Spadaccini
    @raise errors.HypervisorError: when there are communication errors
324 91c10532 Andrea Spadaccini
    @raise errors.ProgrammerError: when there are data serialization errors
325 91c10532 Andrea Spadaccini

326 91c10532 Andrea Spadaccini
    """
327 91c10532 Andrea Spadaccini
    self._check_connection()
328 91c10532 Andrea Spadaccini
329 91c10532 Andrea Spadaccini
    # Check if there is already a message in the buffer
330 91c10532 Andrea Spadaccini
    (message, self._buf) = self._ParseMessage(self._buf)
331 91c10532 Andrea Spadaccini
    if message:
332 91c10532 Andrea Spadaccini
      return message
333 91c10532 Andrea Spadaccini
334 91c10532 Andrea Spadaccini
    recv_buffer = StringIO.StringIO(self._buf)
335 91c10532 Andrea Spadaccini
    recv_buffer.seek(len(self._buf))
336 91c10532 Andrea Spadaccini
    try:
337 91c10532 Andrea Spadaccini
      while True:
338 91c10532 Andrea Spadaccini
        data = self.sock.recv(4096)
339 91c10532 Andrea Spadaccini
        if not data:
340 91c10532 Andrea Spadaccini
          break
341 91c10532 Andrea Spadaccini
        recv_buffer.write(data)
342 91c10532 Andrea Spadaccini
343 91c10532 Andrea Spadaccini
        (message, self._buf) = self._ParseMessage(recv_buffer.getvalue())
344 91c10532 Andrea Spadaccini
        if message:
345 91c10532 Andrea Spadaccini
          return message
346 91c10532 Andrea Spadaccini
347 91c10532 Andrea Spadaccini
    except socket.timeout, err:
348 91c10532 Andrea Spadaccini
      raise errors.HypervisorError("Timeout while receiving a QMP message: "
349 91c10532 Andrea Spadaccini
                                   "%s" % (err))
350 91c10532 Andrea Spadaccini
    except socket.error, err:
351 91c10532 Andrea Spadaccini
      raise errors.HypervisorError("Unable to receive data from KVM using the"
352 91c10532 Andrea Spadaccini
                                   " QMP protocol: %s" % err)
353 91c10532 Andrea Spadaccini
354 91c10532 Andrea Spadaccini
  def _Send(self, message):
355 91c10532 Andrea Spadaccini
    """Encodes and sends a message to KVM using QMP.
356 91c10532 Andrea Spadaccini

357 91c10532 Andrea Spadaccini
    @type message: QmpMessage
358 91c10532 Andrea Spadaccini
    @param message: message to send to KVM
359 91c10532 Andrea Spadaccini
    @raise errors.HypervisorError: when there are communication errors
360 91c10532 Andrea Spadaccini
    @raise errors.ProgrammerError: when there are data serialization errors
361 91c10532 Andrea Spadaccini

362 91c10532 Andrea Spadaccini
    """
363 91c10532 Andrea Spadaccini
    self._check_connection()
364 91c10532 Andrea Spadaccini
    try:
365 91c10532 Andrea Spadaccini
      message_str = str(message)
366 91c10532 Andrea Spadaccini
    except Exception, err:
367 91c10532 Andrea Spadaccini
      raise errors.ProgrammerError("QMP data deserialization error: %s" % err)
368 91c10532 Andrea Spadaccini
369 91c10532 Andrea Spadaccini
    try:
370 91c10532 Andrea Spadaccini
      self.sock.sendall(message_str)
371 91c10532 Andrea Spadaccini
    except socket.timeout, err:
372 91c10532 Andrea Spadaccini
      raise errors.HypervisorError("Timeout while sending a QMP message: "
373 91c10532 Andrea Spadaccini
                                   "%s (%s)" % (err.string, err.errno))
374 91c10532 Andrea Spadaccini
    except socket.error, err:
375 91c10532 Andrea Spadaccini
      raise errors.HypervisorError("Unable to send data from KVM using the"
376 91c10532 Andrea Spadaccini
                                   " QMP protocol: %s" % err)
377 91c10532 Andrea Spadaccini
378 91c10532 Andrea Spadaccini
  def Execute(self, command, arguments=None):
379 91c10532 Andrea Spadaccini
    """Executes a QMP command and returns the response of the server.
380 91c10532 Andrea Spadaccini

381 91c10532 Andrea Spadaccini
    @type command: str
382 91c10532 Andrea Spadaccini
    @param command: the command to execute
383 91c10532 Andrea Spadaccini
    @type arguments: dict
384 91c10532 Andrea Spadaccini
    @param arguments: dictionary of arguments to be passed to the command
385 91c10532 Andrea Spadaccini
    @rtype: dict
386 91c10532 Andrea Spadaccini
    @return: dictionary representing the received JSON object
387 91c10532 Andrea Spadaccini
    @raise errors.HypervisorError: when there are communication errors
388 91c10532 Andrea Spadaccini
    @raise errors.ProgrammerError: when there are data serialization errors
389 91c10532 Andrea Spadaccini

390 91c10532 Andrea Spadaccini
    """
391 91c10532 Andrea Spadaccini
    self._check_connection()
392 91c10532 Andrea Spadaccini
    message = QmpMessage({self._EXECUTE_KEY: command})
393 91c10532 Andrea Spadaccini
    if arguments:
394 91c10532 Andrea Spadaccini
      message[self._ARGUMENTS_KEY] = arguments
395 91c10532 Andrea Spadaccini
    self._Send(message)
396 91c10532 Andrea Spadaccini
397 91c10532 Andrea Spadaccini
    # Events can occur between the sending of the command and the reception
398 91c10532 Andrea Spadaccini
    # of the response, so we need to filter out messages with the event key.
399 91c10532 Andrea Spadaccini
    while True:
400 91c10532 Andrea Spadaccini
      response = self._Recv()
401 91c10532 Andrea Spadaccini
      err = response[self._ERROR_KEY]
402 91c10532 Andrea Spadaccini
      if err:
403 91c10532 Andrea Spadaccini
        raise errors.HypervisorError("kvm: error executing the %s"
404 91c10532 Andrea Spadaccini
                                     " command: %s (%s, %s):" %
405 91c10532 Andrea Spadaccini
                                     (command,
406 91c10532 Andrea Spadaccini
                                      err[self._ERROR_DESC_KEY],
407 91c10532 Andrea Spadaccini
                                      err[self._ERROR_CLASS_KEY],
408 91c10532 Andrea Spadaccini
                                      err[self._ERROR_DATA_KEY]))
409 91c10532 Andrea Spadaccini
410 91c10532 Andrea Spadaccini
      elif not response[self._EVENT_KEY]:
411 91c10532 Andrea Spadaccini
        return response
412 91c10532 Andrea Spadaccini
413 91c10532 Andrea Spadaccini
414 eb58f9b1 Guido Trotter
class KVMHypervisor(hv_base.BaseHypervisor):
415 65107a2f Michael Hanselmann
  """KVM hypervisor interface
416 65107a2f Michael Hanselmann

417 65107a2f Michael Hanselmann
  """
418 d271c6fd Iustin Pop
  CAN_MIGRATE = True
419 eb58f9b1 Guido Trotter
420 9d9bded1 Michael Hanselmann
  _ROOT_DIR = pathutils.RUN_DIR + "/kvm-hypervisor"
421 a1d79fc6 Guido Trotter
  _PIDS_DIR = _ROOT_DIR + "/pid" # contains live instances pids
422 7548396c Guido Trotter
  _UIDS_DIR = _ROOT_DIR + "/uid" # contains instances reserved uids
423 a1d79fc6 Guido Trotter
  _CTRL_DIR = _ROOT_DIR + "/ctrl" # contains instances control sockets
424 a1d79fc6 Guido Trotter
  _CONF_DIR = _ROOT_DIR + "/conf" # contains instances startup data
425 5d9bfd87 Apollon Oikonomopoulos
  _NICS_DIR = _ROOT_DIR + "/nic" # contains instances nic <-> tap associations
426 4f580fef Sébastien Bocahu
  _KEYMAP_DIR = _ROOT_DIR + "/keymap" # contains instances keymaps
427 547a63b7 Balazs Lecz
  # KVM instances with chroot enabled are started in empty chroot directories.
428 547a63b7 Balazs Lecz
  _CHROOT_DIR = _ROOT_DIR + "/chroot" # for empty chroot directories
429 547a63b7 Balazs Lecz
  # After an instance is stopped, its chroot directory is removed.
430 547a63b7 Balazs Lecz
  # If the chroot directory is not empty, it can't be removed.
431 547a63b7 Balazs Lecz
  # A non-empty chroot directory indicates a possible security incident.
432 547a63b7 Balazs Lecz
  # To support forensics, the non-empty chroot directory is quarantined in
433 547a63b7 Balazs Lecz
  # a separate directory, called 'chroot-quarantine'.
434 547a63b7 Balazs Lecz
  _CHROOT_QUARANTINE_DIR = _ROOT_DIR + "/chroot-quarantine"
435 5d9bfd87 Apollon Oikonomopoulos
  _DIRS = [_ROOT_DIR, _PIDS_DIR, _UIDS_DIR, _CTRL_DIR, _CONF_DIR, _NICS_DIR,
436 2911f46c Iustin Pop
           _CHROOT_DIR, _CHROOT_QUARANTINE_DIR, _KEYMAP_DIR]
437 eb58f9b1 Guido Trotter
438 205ab586 Iustin Pop
  PARAMETERS = {
439 4b9638dc Guido Trotter
    constants.HV_KVM_PATH: hv_base.REQ_FILE_CHECK,
440 205ab586 Iustin Pop
    constants.HV_KERNEL_PATH: hv_base.OPT_FILE_CHECK,
441 205ab586 Iustin Pop
    constants.HV_INITRD_PATH: hv_base.OPT_FILE_CHECK,
442 205ab586 Iustin Pop
    constants.HV_ROOT_PATH: hv_base.NO_CHECK,
443 205ab586 Iustin Pop
    constants.HV_KERNEL_ARGS: hv_base.NO_CHECK,
444 205ab586 Iustin Pop
    constants.HV_ACPI: hv_base.NO_CHECK,
445 205ab586 Iustin Pop
    constants.HV_SERIAL_CONSOLE: hv_base.NO_CHECK,
446 14fd6c81 Guido Trotter
    constants.HV_SERIAL_SPEED: hv_base.NO_CHECK,
447 d73ef63f Michael Hanselmann
    constants.HV_VNC_BIND_ADDRESS:
448 8b312c1d Manuel Franceschini
      (False, lambda x: (netutils.IP4Address.IsValid(x) or
449 8b312c1d Manuel Franceschini
                         utils.IsNormAbsPath(x)),
450 d73ef63f Michael Hanselmann
       "the VNC bind address must be either a valid IP address or an absolute"
451 d73ef63f Michael Hanselmann
       " pathname", None, None),
452 205ab586 Iustin Pop
    constants.HV_VNC_TLS: hv_base.NO_CHECK,
453 205ab586 Iustin Pop
    constants.HV_VNC_X509: hv_base.OPT_DIR_CHECK,
454 205ab586 Iustin Pop
    constants.HV_VNC_X509_VERIFY: hv_base.NO_CHECK,
455 6e6bb8d5 Guido Trotter
    constants.HV_VNC_PASSWORD_FILE: hv_base.OPT_FILE_CHECK,
456 b1cb62bd Andrea Spadaccini
    constants.HV_KVM_SPICE_BIND: hv_base.NO_CHECK, # will be checked later
457 b1cb62bd Andrea Spadaccini
    constants.HV_KVM_SPICE_IP_VERSION:
458 b1cb62bd Andrea Spadaccini
      (False, lambda x: (x == constants.IFACE_NO_IP_VERSION_SPECIFIED or
459 b1cb62bd Andrea Spadaccini
                         x in constants.VALID_IP_VERSIONS),
460 b1cb62bd Andrea Spadaccini
       "the SPICE IP version should be 4 or 6",
461 b1cb62bd Andrea Spadaccini
       None, None),
462 b451c4c7 Andrea Spadaccini
    constants.HV_KVM_SPICE_PASSWORD_FILE: hv_base.OPT_FILE_CHECK,
463 ea064d24 Andrea Spadaccini
    constants.HV_KVM_SPICE_LOSSLESS_IMG_COMPR:
464 5ae4945a Iustin Pop
      hv_base.ParamInSet(
465 5ae4945a Iustin Pop
        False, constants.HT_KVM_SPICE_VALID_LOSSLESS_IMG_COMPR_OPTIONS),
466 ea064d24 Andrea Spadaccini
    constants.HV_KVM_SPICE_JPEG_IMG_COMPR:
467 5ae4945a Iustin Pop
      hv_base.ParamInSet(
468 5ae4945a Iustin Pop
        False, constants.HT_KVM_SPICE_VALID_LOSSY_IMG_COMPR_OPTIONS),
469 ea064d24 Andrea Spadaccini
    constants.HV_KVM_SPICE_ZLIB_GLZ_IMG_COMPR:
470 5ae4945a Iustin Pop
      hv_base.ParamInSet(
471 5ae4945a Iustin Pop
        False, constants.HT_KVM_SPICE_VALID_LOSSY_IMG_COMPR_OPTIONS),
472 ea064d24 Andrea Spadaccini
    constants.HV_KVM_SPICE_STREAMING_VIDEO_DETECTION:
473 5ae4945a Iustin Pop
      hv_base.ParamInSet(
474 5ae4945a Iustin Pop
        False, constants.HT_KVM_SPICE_VALID_VIDEO_STREAM_DETECTION_OPTIONS),
475 ea064d24 Andrea Spadaccini
    constants.HV_KVM_SPICE_AUDIO_COMPR: hv_base.NO_CHECK,
476 bfe86c76 Andrea Spadaccini
    constants.HV_KVM_SPICE_USE_TLS: hv_base.NO_CHECK,
477 3e40b587 Andrea Spadaccini
    constants.HV_KVM_SPICE_TLS_CIPHERS: hv_base.NO_CHECK,
478 3e40b587 Andrea Spadaccini
    constants.HV_KVM_SPICE_USE_VDAGENT: hv_base.NO_CHECK,
479 cc130cc7 Marco Casavecchia
    constants.HV_KVM_FLOPPY_IMAGE_PATH: hv_base.OPT_FILE_CHECK,
480 205ab586 Iustin Pop
    constants.HV_CDROM_IMAGE_PATH: hv_base.OPT_FILE_CHECK,
481 cc130cc7 Marco Casavecchia
    constants.HV_KVM_CDROM2_IMAGE_PATH: hv_base.OPT_FILE_CHECK,
482 d73ef63f Michael Hanselmann
    constants.HV_BOOT_ORDER:
483 d73ef63f Michael Hanselmann
      hv_base.ParamInSet(True, constants.HT_KVM_VALID_BO_TYPES),
484 d73ef63f Michael Hanselmann
    constants.HV_NIC_TYPE:
485 d73ef63f Michael Hanselmann
      hv_base.ParamInSet(True, constants.HT_KVM_VALID_NIC_TYPES),
486 d73ef63f Michael Hanselmann
    constants.HV_DISK_TYPE:
487 d73ef63f Michael Hanselmann
      hv_base.ParamInSet(True, constants.HT_KVM_VALID_DISK_TYPES),
488 cc130cc7 Marco Casavecchia
    constants.HV_KVM_CDROM_DISK_TYPE:
489 cc130cc7 Marco Casavecchia
      hv_base.ParamInSet(False, constants.HT_KVM_VALID_DISK_TYPES),
490 d73ef63f Michael Hanselmann
    constants.HV_USB_MOUSE:
491 d73ef63f Michael Hanselmann
      hv_base.ParamInSet(False, constants.HT_KVM_VALID_MOUSE_TYPES),
492 4f580fef Sébastien Bocahu
    constants.HV_KEYMAP: hv_base.NO_CHECK,
493 e2d14329 Andrea Spadaccini
    constants.HV_MIGRATION_PORT: hv_base.REQ_NET_PORT_CHECK,
494 2c368f28 Guido Trotter
    constants.HV_MIGRATION_BANDWIDTH: hv_base.REQ_NONNEGATIVE_INT_CHECK,
495 2c368f28 Guido Trotter
    constants.HV_MIGRATION_DOWNTIME: hv_base.REQ_NONNEGATIVE_INT_CHECK,
496 783a6c0b Iustin Pop
    constants.HV_MIGRATION_MODE: hv_base.MIGRATION_MODE_CHECK,
497 6b970cef Jun Futagawa
    constants.HV_USE_LOCALTIME: hv_base.NO_CHECK,
498 19572932 Iustin Pop
    constants.HV_DISK_CACHE:
499 19572932 Iustin Pop
      hv_base.ParamInSet(True, constants.HT_VALID_CACHE_TYPES),
500 d19d94db Guido Trotter
    constants.HV_SECURITY_MODEL:
501 d19d94db Guido Trotter
      hv_base.ParamInSet(True, constants.HT_KVM_VALID_SM_TYPES),
502 d19d94db Guido Trotter
    constants.HV_SECURITY_DOMAIN: hv_base.NO_CHECK,
503 7ba594c0 Guido Trotter
    constants.HV_KVM_FLAG:
504 7ba594c0 Guido Trotter
      hv_base.ParamInSet(False, constants.HT_KVM_FLAG_VALUES),
505 fbe27e2b Guido Trotter
    constants.HV_VHOST_NET: hv_base.NO_CHECK,
506 547a63b7 Balazs Lecz
    constants.HV_KVM_USE_CHROOT: hv_base.NO_CHECK,
507 4f958b0b Miguel Di Ciurcio Filho
    constants.HV_MEM_PATH: hv_base.OPT_DIR_CHECK,
508 990ade2d Stephen Shirley
    constants.HV_REBOOT_BEHAVIOR:
509 a7f884d3 Tsachy Shacham
      hv_base.ParamInSet(True, constants.REBOOT_BEHAVIORS),
510 a7f884d3 Tsachy Shacham
    constants.HV_CPU_MASK: hv_base.OPT_MULTI_CPU_MASK_CHECK,
511 3c286190 Dimitris Aragiorgis
    constants.HV_CPU_TYPE: hv_base.NO_CHECK,
512 2c368f28 Guido Trotter
    constants.HV_CPU_CORES: hv_base.OPT_NONNEGATIVE_INT_CHECK,
513 2c368f28 Guido Trotter
    constants.HV_CPU_THREADS: hv_base.OPT_NONNEGATIVE_INT_CHECK,
514 2c368f28 Guido Trotter
    constants.HV_CPU_SOCKETS: hv_base.OPT_NONNEGATIVE_INT_CHECK,
515 2fddb144 Guido Trotter
    constants.HV_SOUNDHW: hv_base.NO_CHECK,
516 156681c1 Guido Trotter
    constants.HV_USB_DEVICES: hv_base.NO_CHECK,
517 7589346f Guido Trotter
    constants.HV_VGA: hv_base.NO_CHECK,
518 e6f24d89 Guido Trotter
    constants.HV_KVM_EXTRA: hv_base.NO_CHECK,
519 4ab75599 Dimitris Aragiorgis
    constants.HV_KVM_MACHINE_VERSION: hv_base.NO_CHECK,
520 205ab586 Iustin Pop
    }
521 6b5605e8 Iustin Pop
522 f0db563d Guido Trotter
  _VIRTIO = "virtio"
523 f0db563d Guido Trotter
  _VIRTIO_NET_PCI = "virtio-net-pci"
524 f0db563d Guido Trotter
525 d0c8c01d Iustin Pop
  _MIGRATION_STATUS_RE = re.compile("Migration\s+status:\s+(\w+)",
526 30e42c4e Guido Trotter
                                    re.M | re.I)
527 65107a2f Michael Hanselmann
  _MIGRATION_PROGRESS_RE = \
528 527c0cf7 Michael Hanselmann
    re.compile(r"\s*transferred\s+ram:\s+(?P<transferred>\d+)\s+kbytes\s*\n"
529 527c0cf7 Michael Hanselmann
               r"\s*remaining\s+ram:\s+(?P<remaining>\d+)\s+kbytes\s*\n"
530 527c0cf7 Michael Hanselmann
               r"\s*total\s+ram:\s+(?P<total>\d+)\s+kbytes\s*\n", re.I)
531 61643226 Andrea Spadaccini
532 c4e388a5 Guido Trotter
  _MIGRATION_INFO_MAX_BAD_ANSWERS = 5
533 c4e388a5 Guido Trotter
  _MIGRATION_INFO_RETRY_DELAY = 2
534 30e42c4e Guido Trotter
535 585c8187 Guido Trotter
  _VERSION_RE = re.compile(r"\b(\d+)\.(\d+)(\.(\d+))?\b")
536 b52d85c1 Guido Trotter
537 b693125f Tsachy Shacham
  _CPU_INFO_RE = re.compile(r"cpu\s+\#(\d+).*thread_id\s*=\s*(\d+)", re.I)
538 b693125f Tsachy Shacham
  _CPU_INFO_CMD = "info cpus"
539 b693125f Tsachy Shacham
  _CONT_CMD = "cont"
540 b693125f Tsachy Shacham
541 7f83345f Guido Trotter
  _DEFAULT_MACHINE_VERSION_RE = re.compile(r"^(\S+).*\(default\)", re.M)
542 82e3bf85 Dimitris Aragiorgis
  _CHECK_MACHINE_VERSION_RE = \
543 82e3bf85 Dimitris Aragiorgis
    staticmethod(lambda x: re.compile(r"^(%s)[ ]+.*PC" % x, re.M))
544 e6ba3320 Dimitris Aragiorgis
545 0ad7f5d8 Guido Trotter
  _QMP_RE = re.compile(r"^-qmp\s", re.M)
546 0ad7f5d8 Guido Trotter
  _SPICE_RE = re.compile(r"^-spice\s", re.M)
547 0ad7f5d8 Guido Trotter
  _VHOST_RE = re.compile(r"^-net\s.*,vhost=on|off", re.M)
548 0ad7f5d8 Guido Trotter
  _ENABLE_KVM_RE = re.compile(r"^-enable-kvm\s", re.M)
549 0ad7f5d8 Guido Trotter
  _DISABLE_KVM_RE = re.compile(r"^-disable-kvm\s", re.M)
550 0ad7f5d8 Guido Trotter
  _NETDEV_RE = re.compile(r"^-netdev\s", re.M)
551 f0db563d Guido Trotter
  _NEW_VIRTIO_RE = re.compile(r"^name \"%s\"" % _VIRTIO_NET_PCI, re.M)
552 0ad7f5d8 Guido Trotter
  # match  -drive.*boot=on|off on different lines, but in between accept only
553 0ad7f5d8 Guido Trotter
  # dashes not preceeded by a new line (which would mean another option
554 0ad7f5d8 Guido Trotter
  # different than -drive is starting)
555 0ad7f5d8 Guido Trotter
  _BOOT_RE = re.compile(r"^-drive\s([^-]|(?<!^)-)*,boot=on\|off", re.M | re.S)
556 0ad7f5d8 Guido Trotter
557 64bfbc08 Guido Trotter
  ANCILLARY_FILES = [
558 64bfbc08 Guido Trotter
    _KVM_NETWORK_SCRIPT,
559 64bfbc08 Guido Trotter
    ]
560 69ab2e12 Guido Trotter
  ANCILLARY_FILES_OPT = [
561 69ab2e12 Guido Trotter
    _KVM_NETWORK_SCRIPT,
562 69ab2e12 Guido Trotter
    ]
563 64bfbc08 Guido Trotter
564 6e043e60 Guido Trotter
  # Supported kvm options to get output from
565 6e043e60 Guido Trotter
  _KVMOPT_HELP = "help"
566 6e043e60 Guido Trotter
  _KVMOPT_MLIST = "mlist"
567 f0db563d Guido Trotter
  _KVMOPT_DEVICELIST = "devicelist"
568 bc0fed4b Guido Trotter
569 bc0fed4b Guido Trotter
  # Command to execute to get the output from kvm, and whether to
570 bc0fed4b Guido Trotter
  # accept the output even on failure.
571 6e043e60 Guido Trotter
  _KVMOPTS_CMDS = {
572 bc0fed4b Guido Trotter
    _KVMOPT_HELP: (["--help"], False),
573 bc0fed4b Guido Trotter
    _KVMOPT_MLIST: (["-M", "?"], False),
574 bc0fed4b Guido Trotter
    _KVMOPT_DEVICELIST: (["-device", "?"], True),
575 6e043e60 Guido Trotter
  }
576 6e043e60 Guido Trotter
577 eb58f9b1 Guido Trotter
  def __init__(self):
578 eb58f9b1 Guido Trotter
    hv_base.BaseHypervisor.__init__(self)
579 eb58f9b1 Guido Trotter
    # Let's make sure the directories we need exist, even if the RUN_DIR lives
580 eb58f9b1 Guido Trotter
    # in a tmpfs filesystem or has been otherwise wiped out.
581 29921401 Iustin Pop
    dirs = [(dname, constants.RUN_DIRS_MODE) for dname in self._DIRS]
582 9afb67fe Guido Trotter
    utils.EnsureDirs(dirs)
583 eb58f9b1 Guido Trotter
584 30786fc9 Iustin Pop
  @classmethod
585 30786fc9 Iustin Pop
  def _InstancePidFile(cls, instance_name):
586 98ec75d6 Iustin Pop
    """Returns the instance pidfile.
587 98ec75d6 Iustin Pop

588 98ec75d6 Iustin Pop
    """
589 30786fc9 Iustin Pop
    return utils.PathJoin(cls._PIDS_DIR, instance_name)
590 98ec75d6 Iustin Pop
591 263b8de6 Guido Trotter
  @classmethod
592 7548396c Guido Trotter
  def _InstanceUidFile(cls, instance_name):
593 7548396c Guido Trotter
    """Returns the instance uidfile.
594 7548396c Guido Trotter

595 7548396c Guido Trotter
    """
596 7548396c Guido Trotter
    return utils.PathJoin(cls._UIDS_DIR, instance_name)
597 7548396c Guido Trotter
598 7548396c Guido Trotter
  @classmethod
599 263b8de6 Guido Trotter
  def _InstancePidInfo(cls, pid):
600 263b8de6 Guido Trotter
    """Check pid file for instance information.
601 263b8de6 Guido Trotter

602 263b8de6 Guido Trotter
    Check that a pid file is associated with an instance, and retrieve
603 263b8de6 Guido Trotter
    information from its command line.
604 263b8de6 Guido Trotter

605 263b8de6 Guido Trotter
    @type pid: string or int
606 263b8de6 Guido Trotter
    @param pid: process id of the instance to check
607 263b8de6 Guido Trotter
    @rtype: tuple
608 263b8de6 Guido Trotter
    @return: (instance_name, memory, vcpus)
609 263b8de6 Guido Trotter
    @raise errors.HypervisorError: when an instance cannot be found
610 263b8de6 Guido Trotter

611 263b8de6 Guido Trotter
    """
612 263b8de6 Guido Trotter
    alive = utils.IsProcessAlive(pid)
613 263b8de6 Guido Trotter
    if not alive:
614 263b8de6 Guido Trotter
      raise errors.HypervisorError("Cannot get info for pid %s" % pid)
615 263b8de6 Guido Trotter
616 263b8de6 Guido Trotter
    cmdline_file = utils.PathJoin("/proc", str(pid), "cmdline")
617 263b8de6 Guido Trotter
    try:
618 263b8de6 Guido Trotter
      cmdline = utils.ReadFile(cmdline_file)
619 263b8de6 Guido Trotter
    except EnvironmentError, err:
620 263b8de6 Guido Trotter
      raise errors.HypervisorError("Can't open cmdline file for pid %s: %s" %
621 263b8de6 Guido Trotter
                                   (pid, err))
622 263b8de6 Guido Trotter
623 263b8de6 Guido Trotter
    instance = None
624 263b8de6 Guido Trotter
    memory = 0
625 263b8de6 Guido Trotter
    vcpus = 0
626 263b8de6 Guido Trotter
627 d0c8c01d Iustin Pop
    arg_list = cmdline.split("\x00")
628 263b8de6 Guido Trotter
    while arg_list:
629 e687ec01 Michael Hanselmann
      arg = arg_list.pop(0)
630 263b8de6 Guido Trotter
      if arg == "-name":
631 263b8de6 Guido Trotter
        instance = arg_list.pop(0)
632 263b8de6 Guido Trotter
      elif arg == "-m":
633 263b8de6 Guido Trotter
        memory = int(arg_list.pop(0))
634 263b8de6 Guido Trotter
      elif arg == "-smp":
635 4888a609 Guido Trotter
        vcpus = int(arg_list.pop(0).split(",")[0])
636 263b8de6 Guido Trotter
637 263b8de6 Guido Trotter
    if instance is None:
638 263b8de6 Guido Trotter
      raise errors.HypervisorError("Pid %s doesn't contain a ganeti kvm"
639 263b8de6 Guido Trotter
                                   " instance" % pid)
640 263b8de6 Guido Trotter
641 263b8de6 Guido Trotter
    return (instance, memory, vcpus)
642 263b8de6 Guido Trotter
643 1f8b3a27 Guido Trotter
  def _InstancePidAlive(self, instance_name):
644 263b8de6 Guido Trotter
    """Returns the instance pidfile, pid, and liveness.
645 263b8de6 Guido Trotter

646 263b8de6 Guido Trotter
    @type instance_name: string
647 263b8de6 Guido Trotter
    @param instance_name: instance name
648 263b8de6 Guido Trotter
    @rtype: tuple
649 263b8de6 Guido Trotter
    @return: (pid file name, pid, liveness)
650 1f8b3a27 Guido Trotter

651 1f8b3a27 Guido Trotter
    """
652 98ec75d6 Iustin Pop
    pidfile = self._InstancePidFile(instance_name)
653 1f8b3a27 Guido Trotter
    pid = utils.ReadPidFile(pidfile)
654 263b8de6 Guido Trotter
655 263b8de6 Guido Trotter
    alive = False
656 263b8de6 Guido Trotter
    try:
657 263b8de6 Guido Trotter
      cmd_instance = self._InstancePidInfo(pid)[0]
658 263b8de6 Guido Trotter
      alive = (cmd_instance == instance_name)
659 263b8de6 Guido Trotter
    except errors.HypervisorError:
660 263b8de6 Guido Trotter
      pass
661 1f8b3a27 Guido Trotter
662 1f8b3a27 Guido Trotter
    return (pidfile, pid, alive)
663 1f8b3a27 Guido Trotter
664 5905901c Iustin Pop
  def _CheckDown(self, instance_name):
665 5905901c Iustin Pop
    """Raises an error unless the given instance is down.
666 5905901c Iustin Pop

667 5905901c Iustin Pop
    """
668 5905901c Iustin Pop
    alive = self._InstancePidAlive(instance_name)[2]
669 5905901c Iustin Pop
    if alive:
670 5905901c Iustin Pop
      raise errors.HypervisorError("Failed to start instance %s: %s" %
671 5905901c Iustin Pop
                                   (instance_name, "already running"))
672 5905901c Iustin Pop
673 0df4d98a Guido Trotter
  @classmethod
674 0df4d98a Guido Trotter
  def _InstanceMonitor(cls, instance_name):
675 c4fbefc8 Guido Trotter
    """Returns the instance monitor socket name
676 c4fbefc8 Guido Trotter

677 c4fbefc8 Guido Trotter
    """
678 30786fc9 Iustin Pop
    return utils.PathJoin(cls._CTRL_DIR, "%s.monitor" % instance_name)
679 c4fbefc8 Guido Trotter
680 0df4d98a Guido Trotter
  @classmethod
681 0df4d98a Guido Trotter
  def _InstanceSerial(cls, instance_name):
682 c4fbefc8 Guido Trotter
    """Returns the instance serial socket name
683 c4fbefc8 Guido Trotter

684 c4fbefc8 Guido Trotter
    """
685 30786fc9 Iustin Pop
    return utils.PathJoin(cls._CTRL_DIR, "%s.serial" % instance_name)
686 c4fbefc8 Guido Trotter
687 91c10532 Andrea Spadaccini
  @classmethod
688 91c10532 Andrea Spadaccini
  def _InstanceQmpMonitor(cls, instance_name):
689 91c10532 Andrea Spadaccini
    """Returns the instance serial QMP socket name
690 91c10532 Andrea Spadaccini

691 91c10532 Andrea Spadaccini
    """
692 91c10532 Andrea Spadaccini
    return utils.PathJoin(cls._CTRL_DIR, "%s.qmp" % instance_name)
693 91c10532 Andrea Spadaccini
694 86d6bc2a Guido Trotter
  @staticmethod
695 86d6bc2a Guido Trotter
  def _SocatUnixConsoleParams():
696 86d6bc2a Guido Trotter
    """Returns the correct parameters for socat
697 86d6bc2a Guido Trotter

698 86d6bc2a Guido Trotter
    If we have a new-enough socat we can use raw mode with an escape character.
699 86d6bc2a Guido Trotter

700 86d6bc2a Guido Trotter
    """
701 fe5b0c42 Michael Hanselmann
    if constants.SOCAT_USE_ESCAPE:
702 86d6bc2a Guido Trotter
      return "raw,echo=0,escape=%s" % constants.SOCAT_ESCAPE_CODE
703 86d6bc2a Guido Trotter
    else:
704 86d6bc2a Guido Trotter
      return "echo=0,icanon=0"
705 86d6bc2a Guido Trotter
706 0df4d98a Guido Trotter
  @classmethod
707 0df4d98a Guido Trotter
  def _InstanceKVMRuntime(cls, instance_name):
708 38e250ba Guido Trotter
    """Returns the instance KVM runtime filename
709 38e250ba Guido Trotter

710 38e250ba Guido Trotter
    """
711 30786fc9 Iustin Pop
    return utils.PathJoin(cls._CONF_DIR, "%s.runtime" % instance_name)
712 38e250ba Guido Trotter
713 7e66c35b Guido Trotter
  @classmethod
714 547a63b7 Balazs Lecz
  def _InstanceChrootDir(cls, instance_name):
715 547a63b7 Balazs Lecz
    """Returns the name of the KVM chroot dir of the instance
716 547a63b7 Balazs Lecz

717 547a63b7 Balazs Lecz
    """
718 547a63b7 Balazs Lecz
    return utils.PathJoin(cls._CHROOT_DIR, instance_name)
719 547a63b7 Balazs Lecz
720 547a63b7 Balazs Lecz
  @classmethod
721 5d9bfd87 Apollon Oikonomopoulos
  def _InstanceNICDir(cls, instance_name):
722 5d9bfd87 Apollon Oikonomopoulos
    """Returns the name of the directory holding the tap device files for a
723 5d9bfd87 Apollon Oikonomopoulos
    given instance.
724 5d9bfd87 Apollon Oikonomopoulos

725 5d9bfd87 Apollon Oikonomopoulos
    """
726 5d9bfd87 Apollon Oikonomopoulos
    return utils.PathJoin(cls._NICS_DIR, instance_name)
727 5d9bfd87 Apollon Oikonomopoulos
728 5d9bfd87 Apollon Oikonomopoulos
  @classmethod
729 5d9bfd87 Apollon Oikonomopoulos
  def _InstanceNICFile(cls, instance_name, seq):
730 5d9bfd87 Apollon Oikonomopoulos
    """Returns the name of the file containing the tap device for a given NIC
731 5d9bfd87 Apollon Oikonomopoulos

732 5d9bfd87 Apollon Oikonomopoulos
    """
733 5d9bfd87 Apollon Oikonomopoulos
    return utils.PathJoin(cls._InstanceNICDir(instance_name), str(seq))
734 5d9bfd87 Apollon Oikonomopoulos
735 5d9bfd87 Apollon Oikonomopoulos
  @classmethod
736 4f580fef Sébastien Bocahu
  def _InstanceKeymapFile(cls, instance_name):
737 4f580fef Sébastien Bocahu
    """Returns the name of the file containing the keymap for a given instance
738 4f580fef Sébastien Bocahu

739 4f580fef Sébastien Bocahu
    """
740 4f580fef Sébastien Bocahu
    return utils.PathJoin(cls._KEYMAP_DIR, instance_name)
741 4f580fef Sébastien Bocahu
742 4f580fef Sébastien Bocahu
  @classmethod
743 7548396c Guido Trotter
  def _TryReadUidFile(cls, uid_file):
744 7548396c Guido Trotter
    """Try to read a uid file
745 7548396c Guido Trotter

746 7548396c Guido Trotter
    """
747 7548396c Guido Trotter
    if os.path.exists(uid_file):
748 7548396c Guido Trotter
      try:
749 682f7601 Guido Trotter
        uid = int(utils.ReadOneLineFile(uid_file))
750 aa0b600b Guido Trotter
        return uid
751 7548396c Guido Trotter
      except EnvironmentError:
752 7548396c Guido Trotter
        logging.warning("Can't read uid file", exc_info=True)
753 7548396c Guido Trotter
      except (TypeError, ValueError):
754 7548396c Guido Trotter
        logging.warning("Can't parse uid file contents", exc_info=True)
755 aa0b600b Guido Trotter
    return None
756 7548396c Guido Trotter
757 7548396c Guido Trotter
  @classmethod
758 7e66c35b Guido Trotter
  def _RemoveInstanceRuntimeFiles(cls, pidfile, instance_name):
759 547a63b7 Balazs Lecz
    """Removes an instance's rutime sockets/files/dirs.
760 7e66c35b Guido Trotter

761 7e66c35b Guido Trotter
    """
762 7e66c35b Guido Trotter
    utils.RemoveFile(pidfile)
763 7e66c35b Guido Trotter
    utils.RemoveFile(cls._InstanceMonitor(instance_name))
764 7e66c35b Guido Trotter
    utils.RemoveFile(cls._InstanceSerial(instance_name))
765 91c10532 Andrea Spadaccini
    utils.RemoveFile(cls._InstanceQmpMonitor(instance_name))
766 7e66c35b Guido Trotter
    utils.RemoveFile(cls._InstanceKVMRuntime(instance_name))
767 4f580fef Sébastien Bocahu
    utils.RemoveFile(cls._InstanceKeymapFile(instance_name))
768 7548396c Guido Trotter
    uid_file = cls._InstanceUidFile(instance_name)
769 7548396c Guido Trotter
    uid = cls._TryReadUidFile(uid_file)
770 7548396c Guido Trotter
    utils.RemoveFile(uid_file)
771 7548396c Guido Trotter
    if uid is not None:
772 7548396c Guido Trotter
      uidpool.ReleaseUid(uid)
773 7be85163 Guido Trotter
    try:
774 7be85163 Guido Trotter
      shutil.rmtree(cls._InstanceNICDir(instance_name))
775 7be85163 Guido Trotter
    except OSError, err:
776 7be85163 Guido Trotter
      if err.errno != errno.ENOENT:
777 7be85163 Guido Trotter
        raise
778 547a63b7 Balazs Lecz
    try:
779 547a63b7 Balazs Lecz
      chroot_dir = cls._InstanceChrootDir(instance_name)
780 547a63b7 Balazs Lecz
      utils.RemoveDir(chroot_dir)
781 547a63b7 Balazs Lecz
    except OSError, err:
782 547a63b7 Balazs Lecz
      if err.errno == errno.ENOTEMPTY:
783 547a63b7 Balazs Lecz
        # The chroot directory is expected to be empty, but it isn't.
784 547a63b7 Balazs Lecz
        new_chroot_dir = tempfile.mkdtemp(dir=cls._CHROOT_QUARANTINE_DIR,
785 547a63b7 Balazs Lecz
                                          prefix="%s-%s-" %
786 547a63b7 Balazs Lecz
                                          (instance_name,
787 547a63b7 Balazs Lecz
                                           utils.TimestampForFilename()))
788 547a63b7 Balazs Lecz
        logging.warning("The chroot directory of instance %s can not be"
789 547a63b7 Balazs Lecz
                        " removed as it is not empty. Moving it to the"
790 547a63b7 Balazs Lecz
                        " quarantine instead. Please investigate the"
791 547a63b7 Balazs Lecz
                        " contents (%s) and clean up manually",
792 547a63b7 Balazs Lecz
                        instance_name, new_chroot_dir)
793 547a63b7 Balazs Lecz
        utils.RenameFile(chroot_dir, new_chroot_dir)
794 547a63b7 Balazs Lecz
      else:
795 547a63b7 Balazs Lecz
        raise
796 7e66c35b Guido Trotter
797 748e4b5a Michael Hanselmann
  @staticmethod
798 5d9bfd87 Apollon Oikonomopoulos
  def _ConfigureNIC(instance, seq, nic, tap):
799 5d9bfd87 Apollon Oikonomopoulos
    """Run the network configuration script for a specified NIC
800 eb58f9b1 Guido Trotter

801 eb58f9b1 Guido Trotter
    @param instance: instance we're acting on
802 eb58f9b1 Guido Trotter
    @type instance: instance object
803 eb58f9b1 Guido Trotter
    @param seq: nic sequence number
804 eb58f9b1 Guido Trotter
    @type seq: int
805 eb58f9b1 Guido Trotter
    @param nic: nic we're acting on
806 eb58f9b1 Guido Trotter
    @type nic: nic object
807 5d9bfd87 Apollon Oikonomopoulos
    @param tap: the host's tap interface this NIC corresponds to
808 5d9bfd87 Apollon Oikonomopoulos
    @type tap: str
809 eb58f9b1 Guido Trotter

810 eb58f9b1 Guido Trotter
    """
811 5d9bfd87 Apollon Oikonomopoulos
    if instance.tags:
812 5d9bfd87 Apollon Oikonomopoulos
      tags = " ".join(instance.tags)
813 5d9bfd87 Apollon Oikonomopoulos
    else:
814 5d9bfd87 Apollon Oikonomopoulos
      tags = ""
815 5d9bfd87 Apollon Oikonomopoulos
816 5d9bfd87 Apollon Oikonomopoulos
    env = {
817 5d9bfd87 Apollon Oikonomopoulos
      "PATH": "%s:/sbin:/usr/sbin" % os.environ["PATH"],
818 5d9bfd87 Apollon Oikonomopoulos
      "INSTANCE": instance.name,
819 5d9bfd87 Apollon Oikonomopoulos
      "MAC": nic.mac,
820 5d9bfd87 Apollon Oikonomopoulos
      "MODE": nic.nicparams[constants.NIC_MODE],
821 5d9bfd87 Apollon Oikonomopoulos
      "INTERFACE": tap,
822 5d9bfd87 Apollon Oikonomopoulos
      "INTERFACE_INDEX": str(seq),
823 5d9bfd87 Apollon Oikonomopoulos
      "TAGS": tags,
824 5d9bfd87 Apollon Oikonomopoulos
    }
825 5d9bfd87 Apollon Oikonomopoulos
826 5d9bfd87 Apollon Oikonomopoulos
    if nic.ip:
827 5d9bfd87 Apollon Oikonomopoulos
      env["IP"] = nic.ip
828 5d9bfd87 Apollon Oikonomopoulos
829 5d9bfd87 Apollon Oikonomopoulos
    if nic.nicparams[constants.NIC_LINK]:
830 5d9bfd87 Apollon Oikonomopoulos
      env["LINK"] = nic.nicparams[constants.NIC_LINK]
831 5d9bfd87 Apollon Oikonomopoulos
832 a5ad5e58 Apollon Oikonomopoulos
    if nic.network:
833 cbe4a0a5 Dimitris Aragiorgis
      n = objects.Network.FromDict(nic.netinfo)
834 7e8f03e3 Dimitris Aragiorgis
      env.update(n.HooksDict())
835 a5ad5e58 Apollon Oikonomopoulos
836 5d9bfd87 Apollon Oikonomopoulos
    if nic.nicparams[constants.NIC_MODE] == constants.NIC_MODE_BRIDGED:
837 5d9bfd87 Apollon Oikonomopoulos
      env["BRIDGE"] = nic.nicparams[constants.NIC_LINK]
838 5d9bfd87 Apollon Oikonomopoulos
839 9d9bded1 Michael Hanselmann
    result = utils.RunCmd([pathutils.KVM_IFUP, tap], env=env)
840 5d9bfd87 Apollon Oikonomopoulos
    if result.failed:
841 5d9bfd87 Apollon Oikonomopoulos
      raise errors.HypervisorError("Failed to configure interface %s: %s."
842 5d9bfd87 Apollon Oikonomopoulos
                                   " Network configuration script output: %s" %
843 5d9bfd87 Apollon Oikonomopoulos
                                   (tap, result.fail_reason, result.output))
844 eb58f9b1 Guido Trotter
845 b693125f Tsachy Shacham
  @staticmethod
846 b693125f Tsachy Shacham
  def _VerifyAffinityPackage():
847 b693125f Tsachy Shacham
    if affinity is None:
848 b693125f Tsachy Shacham
      raise errors.HypervisorError("affinity Python package not"
849 5ae4945a Iustin Pop
                                   " found; cannot use CPU pinning under KVM")
850 b693125f Tsachy Shacham
851 b693125f Tsachy Shacham
  @staticmethod
852 b693125f Tsachy Shacham
  def _BuildAffinityCpuMask(cpu_list):
853 b693125f Tsachy Shacham
    """Create a CPU mask suitable for sched_setaffinity from a list of
854 b693125f Tsachy Shacham
    CPUs.
855 b693125f Tsachy Shacham

856 b693125f Tsachy Shacham
    See man taskset for more info on sched_setaffinity masks.
857 b693125f Tsachy Shacham
    For example: [ 0, 2, 5, 6 ] will return 101 (0x65, 0..01100101).
858 b693125f Tsachy Shacham

859 b693125f Tsachy Shacham
    @type cpu_list: list of int
860 b693125f Tsachy Shacham
    @param cpu_list: list of physical CPU numbers to map to vCPUs in order
861 b693125f Tsachy Shacham
    @rtype: int
862 b693125f Tsachy Shacham
    @return: a bit mask of CPU affinities
863 b693125f Tsachy Shacham

864 b693125f Tsachy Shacham
    """
865 b693125f Tsachy Shacham
    if cpu_list == constants.CPU_PINNING_OFF:
866 b693125f Tsachy Shacham
      return constants.CPU_PINNING_ALL_KVM
867 b693125f Tsachy Shacham
    else:
868 b693125f Tsachy Shacham
      return sum(2 ** cpu for cpu in cpu_list)
869 b693125f Tsachy Shacham
870 b693125f Tsachy Shacham
  @classmethod
871 b693125f Tsachy Shacham
  def _AssignCpuAffinity(cls, cpu_mask, process_id, thread_dict):
872 b693125f Tsachy Shacham
    """Change CPU affinity for running VM according to given CPU mask.
873 b693125f Tsachy Shacham

874 b693125f Tsachy Shacham
    @param cpu_mask: CPU mask as given by the user. e.g. "0-2,4:all:1,3"
875 b693125f Tsachy Shacham
    @type cpu_mask: string
876 b693125f Tsachy Shacham
    @param process_id: process ID of KVM process. Used to pin entire VM
877 b693125f Tsachy Shacham
                       to physical CPUs.
878 b693125f Tsachy Shacham
    @type process_id: int
879 b693125f Tsachy Shacham
    @param thread_dict: map of virtual CPUs to KVM thread IDs
880 b693125f Tsachy Shacham
    @type thread_dict: dict int:int
881 b693125f Tsachy Shacham

882 b693125f Tsachy Shacham
    """
883 b693125f Tsachy Shacham
    # Convert the string CPU mask to a list of list of int's
884 b693125f Tsachy Shacham
    cpu_list = utils.ParseMultiCpuMask(cpu_mask)
885 b693125f Tsachy Shacham
886 b693125f Tsachy Shacham
    if len(cpu_list) == 1:
887 b693125f Tsachy Shacham
      all_cpu_mapping = cpu_list[0]
888 b693125f Tsachy Shacham
      if all_cpu_mapping == constants.CPU_PINNING_OFF:
889 b693125f Tsachy Shacham
        # If CPU pinning has 1 entry that's "all", then do nothing
890 b693125f Tsachy Shacham
        pass
891 b693125f Tsachy Shacham
      else:
892 b693125f Tsachy Shacham
        # If CPU pinning has one non-all entry, map the entire VM to
893 b693125f Tsachy Shacham
        # one set of physical CPUs
894 b693125f Tsachy Shacham
        cls._VerifyAffinityPackage()
895 5ae4945a Iustin Pop
        affinity.set_process_affinity_mask(
896 5ae4945a Iustin Pop
          process_id, cls._BuildAffinityCpuMask(all_cpu_mapping))
897 b693125f Tsachy Shacham
    else:
898 b693125f Tsachy Shacham
      # The number of vCPUs mapped should match the number of vCPUs
899 b693125f Tsachy Shacham
      # reported by KVM. This was already verified earlier, so
900 b693125f Tsachy Shacham
      # here only as a sanity check.
901 b693125f Tsachy Shacham
      assert len(thread_dict) == len(cpu_list)
902 b693125f Tsachy Shacham
      cls._VerifyAffinityPackage()
903 b693125f Tsachy Shacham
904 b693125f Tsachy Shacham
      # For each vCPU, map it to the proper list of physical CPUs
905 b693125f Tsachy Shacham
      for vcpu, i in zip(cpu_list, range(len(cpu_list))):
906 b693125f Tsachy Shacham
        affinity.set_process_affinity_mask(thread_dict[i],
907 5ae4945a Iustin Pop
                                           cls._BuildAffinityCpuMask(vcpu))
908 b693125f Tsachy Shacham
909 b693125f Tsachy Shacham
  def _GetVcpuThreadIds(self, instance_name):
910 b693125f Tsachy Shacham
    """Get a mapping of vCPU no. to thread IDs for the instance
911 b693125f Tsachy Shacham

912 b693125f Tsachy Shacham
    @type instance_name: string
913 b693125f Tsachy Shacham
    @param instance_name: instance in question
914 b693125f Tsachy Shacham
    @rtype: dictionary of int:int
915 b693125f Tsachy Shacham
    @return: a dictionary mapping vCPU numbers to thread IDs
916 b693125f Tsachy Shacham

917 b693125f Tsachy Shacham
    """
918 b693125f Tsachy Shacham
    result = {}
919 b693125f Tsachy Shacham
    output = self._CallMonitorCommand(instance_name, self._CPU_INFO_CMD)
920 b693125f Tsachy Shacham
    for line in output.stdout.splitlines():
921 b693125f Tsachy Shacham
      match = self._CPU_INFO_RE.search(line)
922 b693125f Tsachy Shacham
      if not match:
923 b693125f Tsachy Shacham
        continue
924 b693125f Tsachy Shacham
      grp = map(int, match.groups())
925 b693125f Tsachy Shacham
      result[grp[0]] = grp[1]
926 b693125f Tsachy Shacham
927 b693125f Tsachy Shacham
    return result
928 b693125f Tsachy Shacham
929 1d8a7812 Andrea Spadaccini
  def _ExecuteCpuAffinity(self, instance_name, cpu_mask):
930 1d8a7812 Andrea Spadaccini
    """Complete CPU pinning.
931 b693125f Tsachy Shacham

932 b693125f Tsachy Shacham
    @type instance_name: string
933 b693125f Tsachy Shacham
    @param instance_name: name of instance
934 b693125f Tsachy Shacham
    @type cpu_mask: string
935 b693125f Tsachy Shacham
    @param cpu_mask: CPU pinning mask as entered by user
936 b693125f Tsachy Shacham

937 b693125f Tsachy Shacham
    """
938 1d8a7812 Andrea Spadaccini
    # Get KVM process ID, to be used if need to pin entire VM
939 1d8a7812 Andrea Spadaccini
    _, pid, _ = self._InstancePidAlive(instance_name)
940 1d8a7812 Andrea Spadaccini
    # Get vCPU thread IDs, to be used if need to pin vCPUs separately
941 1d8a7812 Andrea Spadaccini
    thread_dict = self._GetVcpuThreadIds(instance_name)
942 1d8a7812 Andrea Spadaccini
    # Run CPU pinning, based on configured mask
943 1d8a7812 Andrea Spadaccini
    self._AssignCpuAffinity(cpu_mask, pid, thread_dict)
944 b693125f Tsachy Shacham
945 eb58f9b1 Guido Trotter
  def ListInstances(self):
946 eb58f9b1 Guido Trotter
    """Get the list of running instances.
947 eb58f9b1 Guido Trotter

948 c41eea6e Iustin Pop
    We can do this by listing our live instances directory and
949 c41eea6e Iustin Pop
    checking whether the associated kvm process is still alive.
950 eb58f9b1 Guido Trotter

951 eb58f9b1 Guido Trotter
    """
952 eb58f9b1 Guido Trotter
    result = []
953 eb58f9b1 Guido Trotter
    for name in os.listdir(self._PIDS_DIR):
954 263b8de6 Guido Trotter
      if self._InstancePidAlive(name)[2]:
955 eb58f9b1 Guido Trotter
        result.append(name)
956 eb58f9b1 Guido Trotter
    return result
957 eb58f9b1 Guido Trotter
958 eb58f9b1 Guido Trotter
  def GetInstanceInfo(self, instance_name):
959 eb58f9b1 Guido Trotter
    """Get instance properties.
960 eb58f9b1 Guido Trotter

961 4fbb3c60 Guido Trotter
    @type instance_name: string
962 c41eea6e Iustin Pop
    @param instance_name: the instance name
963 4fbb3c60 Guido Trotter
    @rtype: tuple of strings
964 4fbb3c60 Guido Trotter
    @return: (name, id, memory, vcpus, stat, times)
965 eb58f9b1 Guido Trotter

966 eb58f9b1 Guido Trotter
    """
967 1122eb25 Iustin Pop
    _, pid, alive = self._InstancePidAlive(instance_name)
968 1f8b3a27 Guido Trotter
    if not alive:
969 eb58f9b1 Guido Trotter
      return None
970 eb58f9b1 Guido Trotter
971 263b8de6 Guido Trotter
    _, memory, vcpus = self._InstancePidInfo(pid)
972 fc84cd5d Guido Trotter
    istat = "---b-"
973 eb58f9b1 Guido Trotter
    times = "0"
974 eb58f9b1 Guido Trotter
975 89da2ff3 Guido Trotter
    try:
976 89da2ff3 Guido Trotter
      qmp = QmpConnection(self._InstanceQmpMonitor(instance_name))
977 89da2ff3 Guido Trotter
      qmp.connect()
978 89da2ff3 Guido Trotter
      vcpus = len(qmp.Execute("query-cpus")[qmp.RETURN_KEY])
979 89da2ff3 Guido Trotter
      # Will fail if ballooning is not enabled, but we can then just resort to
980 89da2ff3 Guido Trotter
      # the value above.
981 89da2ff3 Guido Trotter
      mem_bytes = qmp.Execute("query-balloon")[qmp.RETURN_KEY][qmp.ACTUAL_KEY]
982 89da2ff3 Guido Trotter
      memory = mem_bytes / 1048576
983 89da2ff3 Guido Trotter
    except errors.HypervisorError:
984 89da2ff3 Guido Trotter
      pass
985 89da2ff3 Guido Trotter
986 fc84cd5d Guido Trotter
    return (instance_name, pid, memory, vcpus, istat, times)
987 eb58f9b1 Guido Trotter
988 eb58f9b1 Guido Trotter
  def GetAllInstancesInfo(self):
989 eb58f9b1 Guido Trotter
    """Get properties of all instances.
990 eb58f9b1 Guido Trotter

991 c41eea6e Iustin Pop
    @return: list of tuples (name, id, memory, vcpus, stat, times)
992 c41eea6e Iustin Pop

993 eb58f9b1 Guido Trotter
    """
994 eb58f9b1 Guido Trotter
    data = []
995 eb58f9b1 Guido Trotter
    for name in os.listdir(self._PIDS_DIR):
996 263b8de6 Guido Trotter
      try:
997 263b8de6 Guido Trotter
        info = self.GetInstanceInfo(name)
998 263b8de6 Guido Trotter
      except errors.HypervisorError:
999 3bc145d8 Bernardo Dal Seno
        # Ignore exceptions due to instances being shut down
1000 263b8de6 Guido Trotter
        continue
1001 263b8de6 Guido Trotter
      if info:
1002 263b8de6 Guido Trotter
        data.append(info)
1003 eb58f9b1 Guido Trotter
    return data
1004 eb58f9b1 Guido Trotter
1005 b73f1c59 Guido Trotter
  def _GenerateKVMRuntime(self, instance, block_devices, startup_paused,
1006 b73f1c59 Guido Trotter
                          kvmhelp):
1007 ee5f20b0 Guido Trotter
    """Generate KVM information to start an instance.
1008 eb58f9b1 Guido Trotter

1009 b73f1c59 Guido Trotter
    @type kvmhelp: string
1010 b73f1c59 Guido Trotter
    @param kvmhelp: output of kvm --help
1011 839642c2 Iustin Pop
    @attention: this function must not have any side-effects; for
1012 839642c2 Iustin Pop
        example, it must not write to the filesystem, or read values
1013 839642c2 Iustin Pop
        from the current system the are expected to differ between
1014 839642c2 Iustin Pop
        nodes, since it is only run once at instance startup;
1015 839642c2 Iustin Pop
        actions/kvm arguments that can vary between systems should be
1016 839642c2 Iustin Pop
        done in L{_ExecuteKVMRuntime}
1017 839642c2 Iustin Pop

1018 eb58f9b1 Guido Trotter
    """
1019 4888a609 Guido Trotter
    # pylint: disable=R0912,R0914,R0915
1020 4888a609 Guido Trotter
    hvp = instance.hvparams
1021 4304964a Guido Trotter
1022 e687ec01 Michael Hanselmann
    pidfile = self._InstancePidFile(instance.name)
1023 4b9638dc Guido Trotter
    kvm = hvp[constants.HV_KVM_PATH]
1024 eb58f9b1 Guido Trotter
    kvm_cmd = [kvm]
1025 c6a39fc2 Guido Trotter
    # used just by the vnc server, if enabled
1026 d0c8c01d Iustin Pop
    kvm_cmd.extend(["-name", instance.name])
1027 f5a4b9ce Guido Trotter
    kvm_cmd.extend(["-m", instance.beparams[constants.BE_MAXMEM]])
1028 4888a609 Guido Trotter
1029 4888a609 Guido Trotter
    smp_list = ["%s" % instance.beparams[constants.BE_VCPUS]]
1030 4888a609 Guido Trotter
    if hvp[constants.HV_CPU_CORES]:
1031 4888a609 Guido Trotter
      smp_list.append("cores=%s" % hvp[constants.HV_CPU_CORES])
1032 4888a609 Guido Trotter
    if hvp[constants.HV_CPU_THREADS]:
1033 4888a609 Guido Trotter
      smp_list.append("threads=%s" % hvp[constants.HV_CPU_THREADS])
1034 4888a609 Guido Trotter
    if hvp[constants.HV_CPU_SOCKETS]:
1035 4888a609 Guido Trotter
      smp_list.append("sockets=%s" % hvp[constants.HV_CPU_SOCKETS])
1036 4888a609 Guido Trotter
1037 4888a609 Guido Trotter
    kvm_cmd.extend(["-smp", ",".join(smp_list)])
1038 4888a609 Guido Trotter
1039 d0c8c01d Iustin Pop
    kvm_cmd.extend(["-pidfile", pidfile])
1040 fcf5b670 Guido Trotter
    kvm_cmd.extend(["-balloon", "virtio"])
1041 d0c8c01d Iustin Pop
    kvm_cmd.extend(["-daemonize"])
1042 6b5605e8 Iustin Pop
    if not instance.hvparams[constants.HV_ACPI]:
1043 d0c8c01d Iustin Pop
      kvm_cmd.extend(["-no-acpi"])
1044 990ade2d Stephen Shirley
    if instance.hvparams[constants.HV_REBOOT_BEHAVIOR] == \
1045 990ade2d Stephen Shirley
        constants.INSTANCE_REBOOT_EXIT:
1046 990ade2d Stephen Shirley
      kvm_cmd.extend(["-no-reboot"])
1047 eb58f9b1 Guido Trotter
1048 4ab75599 Dimitris Aragiorgis
    mversion = hvp[constants.HV_KVM_MACHINE_VERSION]
1049 4ab75599 Dimitris Aragiorgis
    if not mversion:
1050 4b9638dc Guido Trotter
      mversion = self._GetDefaultMachineVersion(kvm)
1051 4ab75599 Dimitris Aragiorgis
    kvm_cmd.extend(["-M", mversion])
1052 4ab75599 Dimitris Aragiorgis
1053 2b846304 Iustin Pop
    kernel_path = hvp[constants.HV_KERNEL_PATH]
1054 2b846304 Iustin Pop
    if kernel_path:
1055 2b846304 Iustin Pop
      boot_disk = boot_cdrom = boot_floppy = boot_network = False
1056 2b846304 Iustin Pop
    else:
1057 2b846304 Iustin Pop
      boot_disk = hvp[constants.HV_BOOT_ORDER] == constants.HT_BO_DISK
1058 2b846304 Iustin Pop
      boot_cdrom = hvp[constants.HV_BOOT_ORDER] == constants.HT_BO_CDROM
1059 2b846304 Iustin Pop
      boot_floppy = hvp[constants.HV_BOOT_ORDER] == constants.HT_BO_FLOPPY
1060 2b846304 Iustin Pop
      boot_network = hvp[constants.HV_BOOT_ORDER] == constants.HT_BO_NETWORK
1061 8745c3d7 Guido Trotter
1062 b1cb62bd Andrea Spadaccini
    self.ValidateParameters(hvp)
1063 b1cb62bd Andrea Spadaccini
1064 b693125f Tsachy Shacham
    if startup_paused:
1065 b693125f Tsachy Shacham
      kvm_cmd.extend([_KVM_START_PAUSED_FLAG])
1066 b693125f Tsachy Shacham
1067 0ad7f5d8 Guido Trotter
    if (hvp[constants.HV_KVM_FLAG] == constants.HT_KVM_ENABLED and
1068 0ad7f5d8 Guido Trotter
        self._ENABLE_KVM_RE.search(kvmhelp)):
1069 7ba594c0 Guido Trotter
      kvm_cmd.extend(["-enable-kvm"])
1070 0ad7f5d8 Guido Trotter
    elif (hvp[constants.HV_KVM_FLAG] == constants.HT_KVM_DISABLED and
1071 0ad7f5d8 Guido Trotter
          self._DISABLE_KVM_RE.search(kvmhelp)):
1072 7ba594c0 Guido Trotter
      kvm_cmd.extend(["-disable-kvm"])
1073 7ba594c0 Guido Trotter
1074 8745c3d7 Guido Trotter
    if boot_network:
1075 d0c8c01d Iustin Pop
      kvm_cmd.extend(["-boot", "n"])
1076 1213604d Guido Trotter
1077 24be50e0 Iustin Pop
    # whether this is an older KVM version that uses the boot=on flag
1078 24be50e0 Iustin Pop
    # on devices
1079 0ad7f5d8 Guido Trotter
    needs_boot_flag = self._BOOT_RE.search(kvmhelp)
1080 24be50e0 Iustin Pop
1081 a985b417 Iustin Pop
    disk_type = hvp[constants.HV_DISK_TYPE]
1082 1213604d Guido Trotter
    if disk_type == constants.HT_DISK_PARAVIRTUAL:
1083 d0c8c01d Iustin Pop
      if_val = ",if=virtio"
1084 1213604d Guido Trotter
    else:
1085 d0c8c01d Iustin Pop
      if_val = ",if=%s" % disk_type
1086 19572932 Iustin Pop
    # Cache mode
1087 19572932 Iustin Pop
    disk_cache = hvp[constants.HV_DISK_CACHE]
1088 41e794f6 Apollon Oikonomopoulos
    if instance.disk_template in constants.DTS_EXT_MIRROR:
1089 41e794f6 Apollon Oikonomopoulos
      if disk_cache != "none":
1090 41e794f6 Apollon Oikonomopoulos
        # TODO: make this a hard error, instead of a silent overwrite
1091 41e794f6 Apollon Oikonomopoulos
        logging.warning("KVM: overriding disk_cache setting '%s' with 'none'"
1092 41e794f6 Apollon Oikonomopoulos
                        " to prevent shared storage corruption on migration",
1093 41e794f6 Apollon Oikonomopoulos
                        disk_cache)
1094 41e794f6 Apollon Oikonomopoulos
      cache_val = ",cache=none"
1095 41e794f6 Apollon Oikonomopoulos
    elif disk_cache != constants.HT_CACHE_DEFAULT:
1096 19572932 Iustin Pop
      cache_val = ",cache=%s" % disk_cache
1097 19572932 Iustin Pop
    else:
1098 19572932 Iustin Pop
      cache_val = ""
1099 069cfbf1 Iustin Pop
    for cfdev, dev_path in block_devices:
1100 d34b16d7 Iustin Pop
      if cfdev.mode != constants.DISK_RDWR:
1101 d34b16d7 Iustin Pop
        raise errors.HypervisorError("Instance has read-only disks which"
1102 d34b16d7 Iustin Pop
                                     " are not supported by KVM")
1103 eb58f9b1 Guido Trotter
      # TODO: handle FD_LOOP and FD_BLKTAP (?)
1104 4304964a Guido Trotter
      boot_val = ""
1105 66d5dbef Guido Trotter
      if boot_disk:
1106 d0c8c01d Iustin Pop
        kvm_cmd.extend(["-boot", "c"])
1107 66d5dbef Guido Trotter
        boot_disk = False
1108 24be50e0 Iustin Pop
        if needs_boot_flag and disk_type != constants.HT_DISK_IDE:
1109 4304964a Guido Trotter
          boot_val = ",boot=on"
1110 eb58f9b1 Guido Trotter
1111 d0c8c01d Iustin Pop
      drive_val = "file=%s,format=raw%s%s%s" % (dev_path, if_val, boot_val,
1112 19572932 Iustin Pop
                                                cache_val)
1113 d0c8c01d Iustin Pop
      kvm_cmd.extend(["-drive", drive_val])
1114 eb58f9b1 Guido Trotter
1115 cc130cc7 Marco Casavecchia
    #Now we can specify a different device type for CDROM devices.
1116 cc130cc7 Marco Casavecchia
    cdrom_disk_type = hvp[constants.HV_KVM_CDROM_DISK_TYPE]
1117 cc130cc7 Marco Casavecchia
    if not cdrom_disk_type:
1118 cc130cc7 Marco Casavecchia
      cdrom_disk_type = disk_type
1119 cc130cc7 Marco Casavecchia
1120 a985b417 Iustin Pop
    iso_image = hvp[constants.HV_CDROM_IMAGE_PATH]
1121 66d5dbef Guido Trotter
    if iso_image:
1122 d0c8c01d Iustin Pop
      options = ",format=raw,media=cdrom"
1123 24be50e0 Iustin Pop
      # set cdrom 'if' type
1124 66d5dbef Guido Trotter
      if boot_cdrom:
1125 24be50e0 Iustin Pop
        actual_cdrom_type = constants.HT_DISK_IDE
1126 24be50e0 Iustin Pop
      elif cdrom_disk_type == constants.HT_DISK_PARAVIRTUAL:
1127 24be50e0 Iustin Pop
        actual_cdrom_type = "virtio"
1128 9dd363eb Guido Trotter
      else:
1129 24be50e0 Iustin Pop
        actual_cdrom_type = cdrom_disk_type
1130 24be50e0 Iustin Pop
      if_val = ",if=%s" % actual_cdrom_type
1131 24be50e0 Iustin Pop
      # set boot flag, if needed
1132 24be50e0 Iustin Pop
      boot_val = ""
1133 24be50e0 Iustin Pop
      if boot_cdrom:
1134 24be50e0 Iustin Pop
        kvm_cmd.extend(["-boot", "d"])
1135 24be50e0 Iustin Pop
        if needs_boot_flag:
1136 24be50e0 Iustin Pop
          boot_val = ",boot=on"
1137 24be50e0 Iustin Pop
      # and finally build the entire '-drive' value
1138 24be50e0 Iustin Pop
      drive_val = "file=%s%s%s%s" % (iso_image, options, if_val, boot_val)
1139 d0c8c01d Iustin Pop
      kvm_cmd.extend(["-drive", drive_val])
1140 66d5dbef Guido Trotter
1141 cc130cc7 Marco Casavecchia
    iso_image2 = hvp[constants.HV_KVM_CDROM2_IMAGE_PATH]
1142 cc130cc7 Marco Casavecchia
    if iso_image2:
1143 d0c8c01d Iustin Pop
      options = ",format=raw,media=cdrom"
1144 cc130cc7 Marco Casavecchia
      if cdrom_disk_type == constants.HT_DISK_PARAVIRTUAL:
1145 d0c8c01d Iustin Pop
        if_val = ",if=virtio"
1146 cc130cc7 Marco Casavecchia
      else:
1147 d0c8c01d Iustin Pop
        if_val = ",if=%s" % cdrom_disk_type
1148 24be50e0 Iustin Pop
      drive_val = "file=%s%s%s" % (iso_image2, options, if_val)
1149 d0c8c01d Iustin Pop
      kvm_cmd.extend(["-drive", drive_val])
1150 cc130cc7 Marco Casavecchia
1151 cc130cc7 Marco Casavecchia
    floppy_image = hvp[constants.HV_KVM_FLOPPY_IMAGE_PATH]
1152 cc130cc7 Marco Casavecchia
    if floppy_image:
1153 d0c8c01d Iustin Pop
      options = ",format=raw,media=disk"
1154 cc130cc7 Marco Casavecchia
      if boot_floppy:
1155 d0c8c01d Iustin Pop
        kvm_cmd.extend(["-boot", "a"])
1156 d0c8c01d Iustin Pop
        options = "%s,boot=on" % options
1157 d0c8c01d Iustin Pop
      if_val = ",if=floppy"
1158 d0c8c01d Iustin Pop
      options = "%s%s" % (options, if_val)
1159 d0c8c01d Iustin Pop
      drive_val = "file=%s%s" % (floppy_image, options)
1160 d0c8c01d Iustin Pop
      kvm_cmd.extend(["-drive", drive_val])
1161 cc130cc7 Marco Casavecchia
1162 df5ab9f0 Guido Trotter
    if kernel_path:
1163 d0c8c01d Iustin Pop
      kvm_cmd.extend(["-kernel", kernel_path])
1164 a985b417 Iustin Pop
      initrd_path = hvp[constants.HV_INITRD_PATH]
1165 df5ab9f0 Guido Trotter
      if initrd_path:
1166 d0c8c01d Iustin Pop
        kvm_cmd.extend(["-initrd", initrd_path])
1167 d0c8c01d Iustin Pop
      root_append = ["root=%s" % hvp[constants.HV_ROOT_PATH],
1168 07813a9e Iustin Pop
                     hvp[constants.HV_KERNEL_ARGS]]
1169 07813a9e Iustin Pop
      if hvp[constants.HV_SERIAL_CONSOLE]:
1170 14fd6c81 Guido Trotter
        serial_speed = hvp[constants.HV_SERIAL_SPEED]
1171 14fd6c81 Guido Trotter
        root_append.append("console=ttyS0,%s" % serial_speed)
1172 d0c8c01d Iustin Pop
      kvm_cmd.extend(["-append", " ".join(root_append)])
1173 eb58f9b1 Guido Trotter
1174 4f958b0b Miguel Di Ciurcio Filho
    mem_path = hvp[constants.HV_MEM_PATH]
1175 4f958b0b Miguel Di Ciurcio Filho
    if mem_path:
1176 9d3015f9 Guido Trotter
      kvm_cmd.extend(["-mem-path", mem_path, "-mem-prealloc"])
1177 4f958b0b Miguel Di Ciurcio Filho
1178 596b2459 Guido Trotter
    monitor_dev = ("unix:%s,server,nowait" %
1179 596b2459 Guido Trotter
                   self._InstanceMonitor(instance.name))
1180 596b2459 Guido Trotter
    kvm_cmd.extend(["-monitor", monitor_dev])
1181 596b2459 Guido Trotter
    if hvp[constants.HV_SERIAL_CONSOLE]:
1182 596b2459 Guido Trotter
      serial_dev = ("unix:%s,server,nowait" %
1183 596b2459 Guido Trotter
                    self._InstanceSerial(instance.name))
1184 596b2459 Guido Trotter
      kvm_cmd.extend(["-serial", serial_dev])
1185 596b2459 Guido Trotter
    else:
1186 596b2459 Guido Trotter
      kvm_cmd.extend(["-serial", "none"])
1187 596b2459 Guido Trotter
1188 a985b417 Iustin Pop
    mouse_type = hvp[constants.HV_USB_MOUSE]
1189 31f6f67a Guido Trotter
    vnc_bind_address = hvp[constants.HV_VNC_BIND_ADDRESS]
1190 596b2459 Guido Trotter
    spice_bind = hvp[constants.HV_KVM_SPICE_BIND]
1191 596b2459 Guido Trotter
    spice_ip_version = None
1192 31f6f67a Guido Trotter
1193 a900a30c Guido Trotter
    kvm_cmd.extend(["-usb"])
1194 a900a30c Guido Trotter
1195 11344a50 Guido Trotter
    if mouse_type:
1196 d0c8c01d Iustin Pop
      kvm_cmd.extend(["-usbdevice", mouse_type])
1197 31f6f67a Guido Trotter
    elif vnc_bind_address:
1198 d0c8c01d Iustin Pop
      kvm_cmd.extend(["-usbdevice", constants.HT_MOUSE_TABLET])
1199 11344a50 Guido Trotter
1200 8470c8db Guido Trotter
    if vnc_bind_address:
1201 8b312c1d Manuel Franceschini
      if netutils.IP4Address.IsValid(vnc_bind_address):
1202 377d74c9 Guido Trotter
        if instance.network_port > constants.VNC_BASE_PORT:
1203 377d74c9 Guido Trotter
          display = instance.network_port - constants.VNC_BASE_PORT
1204 14f5f1b6 Manuel Franceschini
          if vnc_bind_address == constants.IP4_ADDRESS_ANY:
1205 d0c8c01d Iustin Pop
            vnc_arg = ":%d" % (display)
1206 8447f52b Guido Trotter
          else:
1207 d0c8c01d Iustin Pop
            vnc_arg = "%s:%d" % (vnc_bind_address, display)
1208 8470c8db Guido Trotter
        else:
1209 8447f52b Guido Trotter
          logging.error("Network port is not a valid VNC display (%d < %d)."
1210 07b8a2b5 Iustin Pop
                        " Not starting VNC", instance.network_port,
1211 07b8a2b5 Iustin Pop
                        constants.VNC_BASE_PORT)
1212 d0c8c01d Iustin Pop
          vnc_arg = "none"
1213 8b2d1013 Guido Trotter
1214 8b2d1013 Guido Trotter
        # Only allow tls and other option when not binding to a file, for now.
1215 8b2d1013 Guido Trotter
        # kvm/qemu gets confused otherwise about the filename to use.
1216 d0c8c01d Iustin Pop
        vnc_append = ""
1217 a985b417 Iustin Pop
        if hvp[constants.HV_VNC_TLS]:
1218 d0c8c01d Iustin Pop
          vnc_append = "%s,tls" % vnc_append
1219 a985b417 Iustin Pop
          if hvp[constants.HV_VNC_X509_VERIFY]:
1220 d0c8c01d Iustin Pop
            vnc_append = "%s,x509verify=%s" % (vnc_append,
1221 a985b417 Iustin Pop
                                               hvp[constants.HV_VNC_X509])
1222 a985b417 Iustin Pop
          elif hvp[constants.HV_VNC_X509]:
1223 d0c8c01d Iustin Pop
            vnc_append = "%s,x509=%s" % (vnc_append,
1224 a985b417 Iustin Pop
                                         hvp[constants.HV_VNC_X509])
1225 6e6bb8d5 Guido Trotter
        if hvp[constants.HV_VNC_PASSWORD_FILE]:
1226 d0c8c01d Iustin Pop
          vnc_append = "%s,password" % vnc_append
1227 6e6bb8d5 Guido Trotter
1228 d0c8c01d Iustin Pop
        vnc_arg = "%s%s" % (vnc_arg, vnc_append)
1229 8b2d1013 Guido Trotter
1230 8470c8db Guido Trotter
      else:
1231 d0c8c01d Iustin Pop
        vnc_arg = "unix:%s/%s.vnc" % (vnc_bind_address, instance.name)
1232 8b2d1013 Guido Trotter
1233 d0c8c01d Iustin Pop
      kvm_cmd.extend(["-vnc", vnc_arg])
1234 596b2459 Guido Trotter
    elif spice_bind:
1235 839642c2 Iustin Pop
      # FIXME: this is wrong here; the iface ip address differs
1236 839642c2 Iustin Pop
      # between systems, so it should be done in _ExecuteKVMRuntime
1237 b1cb62bd Andrea Spadaccini
      if netutils.IsValidInterface(spice_bind):
1238 b1cb62bd Andrea Spadaccini
        # The user specified a network interface, we have to figure out the IP
1239 b1cb62bd Andrea Spadaccini
        # address.
1240 b1cb62bd Andrea Spadaccini
        addresses = netutils.GetInterfaceIpAddresses(spice_bind)
1241 b1cb62bd Andrea Spadaccini
        spice_ip_version = hvp[constants.HV_KVM_SPICE_IP_VERSION]
1242 b1cb62bd Andrea Spadaccini
1243 b1cb62bd Andrea Spadaccini
        # if the user specified an IP version and the interface does not
1244 b1cb62bd Andrea Spadaccini
        # have that kind of IP addresses, throw an exception
1245 b1cb62bd Andrea Spadaccini
        if spice_ip_version != constants.IFACE_NO_IP_VERSION_SPECIFIED:
1246 b1cb62bd Andrea Spadaccini
          if not addresses[spice_ip_version]:
1247 b1cb62bd Andrea Spadaccini
            raise errors.HypervisorError("spice: unable to get an IPv%s address"
1248 b1cb62bd Andrea Spadaccini
                                         " for %s" % (spice_ip_version,
1249 b1cb62bd Andrea Spadaccini
                                                      spice_bind))
1250 b1cb62bd Andrea Spadaccini
1251 b1cb62bd Andrea Spadaccini
        # the user did not specify an IP version, we have to figure it out
1252 b1cb62bd Andrea Spadaccini
        elif (addresses[constants.IP4_VERSION] and
1253 b1cb62bd Andrea Spadaccini
              addresses[constants.IP6_VERSION]):
1254 b1cb62bd Andrea Spadaccini
          # we have both ipv4 and ipv6, let's use the cluster default IP
1255 b1cb62bd Andrea Spadaccini
          # version
1256 b1cb62bd Andrea Spadaccini
          cluster_family = ssconf.SimpleStore().GetPrimaryIPFamily()
1257 65107a2f Michael Hanselmann
          spice_ip_version = \
1258 65107a2f Michael Hanselmann
            netutils.IPAddress.GetVersionFromAddressFamily(cluster_family)
1259 b1cb62bd Andrea Spadaccini
        elif addresses[constants.IP4_VERSION]:
1260 b1cb62bd Andrea Spadaccini
          spice_ip_version = constants.IP4_VERSION
1261 b845c8a1 Andrea Spadaccini
        elif addresses[constants.IP6_VERSION]:
1262 b1cb62bd Andrea Spadaccini
          spice_ip_version = constants.IP6_VERSION
1263 b845c8a1 Andrea Spadaccini
        else:
1264 b845c8a1 Andrea Spadaccini
          raise errors.HypervisorError("spice: unable to get an IP address"
1265 b845c8a1 Andrea Spadaccini
                                       " for %s" % (spice_bind))
1266 b1cb62bd Andrea Spadaccini
1267 b1cb62bd Andrea Spadaccini
        spice_address = addresses[spice_ip_version][0]
1268 b1cb62bd Andrea Spadaccini
1269 b1cb62bd Andrea Spadaccini
      else:
1270 b1cb62bd Andrea Spadaccini
        # spice_bind is known to be a valid IP address, because
1271 b1cb62bd Andrea Spadaccini
        # ValidateParameters checked it.
1272 b1cb62bd Andrea Spadaccini
        spice_address = spice_bind
1273 b1cb62bd Andrea Spadaccini
1274 bfe86c76 Andrea Spadaccini
      spice_arg = "addr=%s" % spice_address
1275 bfe86c76 Andrea Spadaccini
      if hvp[constants.HV_KVM_SPICE_USE_TLS]:
1276 5ae4945a Iustin Pop
        spice_arg = ("%s,tls-port=%s,x509-cacert-file=%s" %
1277 5ae4945a Iustin Pop
                     (spice_arg, instance.network_port,
1278 9d9bded1 Michael Hanselmann
                      pathutils.SPICE_CACERT_FILE))
1279 5ae4945a Iustin Pop
        spice_arg = ("%s,x509-key-file=%s,x509-cert-file=%s" %
1280 9d9bded1 Michael Hanselmann
                     (spice_arg, pathutils.SPICE_CERT_FILE,
1281 9d9bded1 Michael Hanselmann
                      pathutils.SPICE_CERT_FILE))
1282 3e40b587 Andrea Spadaccini
        tls_ciphers = hvp[constants.HV_KVM_SPICE_TLS_CIPHERS]
1283 3e40b587 Andrea Spadaccini
        if tls_ciphers:
1284 3e40b587 Andrea Spadaccini
          spice_arg = "%s,tls-ciphers=%s" % (spice_arg, tls_ciphers)
1285 bfe86c76 Andrea Spadaccini
      else:
1286 bfe86c76 Andrea Spadaccini
        spice_arg = "%s,port=%s" % (spice_arg, instance.network_port)
1287 bfe86c76 Andrea Spadaccini
1288 b451c4c7 Andrea Spadaccini
      if not hvp[constants.HV_KVM_SPICE_PASSWORD_FILE]:
1289 b451c4c7 Andrea Spadaccini
        spice_arg = "%s,disable-ticketing" % spice_arg
1290 b451c4c7 Andrea Spadaccini
1291 11ae7a0a Andrea Spadaccini
      if spice_ip_version:
1292 11ae7a0a Andrea Spadaccini
        spice_arg = "%s,ipv%s" % (spice_arg, spice_ip_version)
1293 b1cb62bd Andrea Spadaccini
1294 ea064d24 Andrea Spadaccini
      # Image compression options
1295 ea064d24 Andrea Spadaccini
      img_lossless = hvp[constants.HV_KVM_SPICE_LOSSLESS_IMG_COMPR]
1296 ea064d24 Andrea Spadaccini
      img_jpeg = hvp[constants.HV_KVM_SPICE_JPEG_IMG_COMPR]
1297 ea064d24 Andrea Spadaccini
      img_zlib_glz = hvp[constants.HV_KVM_SPICE_ZLIB_GLZ_IMG_COMPR]
1298 ea064d24 Andrea Spadaccini
      if img_lossless:
1299 ea064d24 Andrea Spadaccini
        spice_arg = "%s,image-compression=%s" % (spice_arg, img_lossless)
1300 ea064d24 Andrea Spadaccini
      if img_jpeg:
1301 ea064d24 Andrea Spadaccini
        spice_arg = "%s,jpeg-wan-compression=%s" % (spice_arg, img_jpeg)
1302 ea064d24 Andrea Spadaccini
      if img_zlib_glz:
1303 ea064d24 Andrea Spadaccini
        spice_arg = "%s,zlib-glz-wan-compression=%s" % (spice_arg, img_zlib_glz)
1304 ea064d24 Andrea Spadaccini
1305 ea064d24 Andrea Spadaccini
      # Video stream detection
1306 ea064d24 Andrea Spadaccini
      video_streaming = hvp[constants.HV_KVM_SPICE_STREAMING_VIDEO_DETECTION]
1307 ea064d24 Andrea Spadaccini
      if video_streaming:
1308 ea064d24 Andrea Spadaccini
        spice_arg = "%s,streaming-video=%s" % (spice_arg, video_streaming)
1309 ea064d24 Andrea Spadaccini
1310 ea064d24 Andrea Spadaccini
      # Audio compression, by default in qemu-kvm it is on
1311 ea064d24 Andrea Spadaccini
      if not hvp[constants.HV_KVM_SPICE_AUDIO_COMPR]:
1312 ea064d24 Andrea Spadaccini
        spice_arg = "%s,playback-compression=off" % spice_arg
1313 3e40b587 Andrea Spadaccini
      if not hvp[constants.HV_KVM_SPICE_USE_VDAGENT]:
1314 3e40b587 Andrea Spadaccini
        spice_arg = "%s,agent-mouse=off" % spice_arg
1315 447af814 Nikos Skalkotos
      else:
1316 447af814 Nikos Skalkotos
        # Enable the spice agent communication channel between the host and the
1317 447af814 Nikos Skalkotos
        # agent.
1318 447af814 Nikos Skalkotos
        kvm_cmd.extend(["-device", "virtio-serial-pci"])
1319 447af814 Nikos Skalkotos
        kvm_cmd.extend(["-device", "virtserialport,chardev=spicechannel0,"
1320 447af814 Nikos Skalkotos
                                                   "name=com.redhat.spice.0"])
1321 447af814 Nikos Skalkotos
        kvm_cmd.extend(["-chardev", "spicevmc,id=spicechannel0,name=vdagent"])
1322 ea064d24 Andrea Spadaccini
1323 b1cb62bd Andrea Spadaccini
      logging.info("KVM: SPICE will listen on port %s", instance.network_port)
1324 b1cb62bd Andrea Spadaccini
      kvm_cmd.extend(["-spice", spice_arg])
1325 b1cb62bd Andrea Spadaccini
1326 596b2459 Guido Trotter
    else:
1327 596b2459 Guido Trotter
      kvm_cmd.extend(["-nographic"])
1328 596b2459 Guido Trotter
1329 6b970cef Jun Futagawa
    if hvp[constants.HV_USE_LOCALTIME]:
1330 d0c8c01d Iustin Pop
      kvm_cmd.extend(["-localtime"])
1331 6b970cef Jun Futagawa
1332 547a63b7 Balazs Lecz
    if hvp[constants.HV_KVM_USE_CHROOT]:
1333 d0c8c01d Iustin Pop
      kvm_cmd.extend(["-chroot", self._InstanceChrootDir(instance.name)])
1334 547a63b7 Balazs Lecz
1335 7cb42171 Nikita Staroverov
    # Add qemu-KVM -cpu param
1336 7cb42171 Nikita Staroverov
    if hvp[constants.HV_CPU_TYPE]:
1337 7cb42171 Nikita Staroverov
      kvm_cmd.extend(["-cpu", hvp[constants.HV_CPU_TYPE]])
1338 7cb42171 Nikita Staroverov
1339 2fddb144 Guido Trotter
    # As requested by music lovers
1340 2fddb144 Guido Trotter
    if hvp[constants.HV_SOUNDHW]:
1341 2fddb144 Guido Trotter
      kvm_cmd.extend(["-soundhw", hvp[constants.HV_SOUNDHW]])
1342 2fddb144 Guido Trotter
1343 7589346f Guido Trotter
    # Pass a -vga option if requested, or if spice is used, for backwards
1344 7589346f Guido Trotter
    # compatibility.
1345 7589346f Guido Trotter
    if hvp[constants.HV_VGA]:
1346 7589346f Guido Trotter
      kvm_cmd.extend(["-vga", hvp[constants.HV_VGA]])
1347 7589346f Guido Trotter
    elif spice_bind:
1348 7589346f Guido Trotter
      kvm_cmd.extend(["-vga", "qxl"])
1349 7589346f Guido Trotter
1350 156681c1 Guido Trotter
    # Various types of usb devices, comma separated
1351 156681c1 Guido Trotter
    if hvp[constants.HV_USB_DEVICES]:
1352 156681c1 Guido Trotter
      for dev in hvp[constants.HV_USB_DEVICES].split(","):
1353 156681c1 Guido Trotter
        kvm_cmd.extend(["-usbdevice", dev])
1354 156681c1 Guido Trotter
1355 e6f24d89 Guido Trotter
    if hvp[constants.HV_KVM_EXTRA]:
1356 e6f24d89 Guido Trotter
      kvm_cmd.extend([hvp[constants.HV_KVM_EXTRA]])
1357 e6f24d89 Guido Trotter
1358 ee5f20b0 Guido Trotter
    # Save the current instance nics, but defer their expansion as parameters,
1359 ee5f20b0 Guido Trotter
    # as we'll need to generate executable temp files for them.
1360 ee5f20b0 Guido Trotter
    kvm_nics = instance.nics
1361 a985b417 Iustin Pop
    hvparams = hvp
1362 ee5f20b0 Guido Trotter
1363 c2672466 Guido Trotter
    return (kvm_cmd, kvm_nics, hvparams)
1364 ee5f20b0 Guido Trotter
1365 38e250ba Guido Trotter
  def _WriteKVMRuntime(self, instance_name, data):
1366 38e250ba Guido Trotter
    """Write an instance's KVM runtime
1367 38e250ba Guido Trotter

1368 38e250ba Guido Trotter
    """
1369 38e250ba Guido Trotter
    try:
1370 38e250ba Guido Trotter
      utils.WriteFile(self._InstanceKVMRuntime(instance_name),
1371 38e250ba Guido Trotter
                      data=data)
1372 90c024f6 Guido Trotter
    except EnvironmentError, err:
1373 38e250ba Guido Trotter
      raise errors.HypervisorError("Failed to save KVM runtime file: %s" % err)
1374 38e250ba Guido Trotter
1375 38e250ba Guido Trotter
  def _ReadKVMRuntime(self, instance_name):
1376 38e250ba Guido Trotter
    """Read an instance's KVM runtime
1377 38e250ba Guido Trotter

1378 38e250ba Guido Trotter
    """
1379 38e250ba Guido Trotter
    try:
1380 38e250ba Guido Trotter
      file_content = utils.ReadFile(self._InstanceKVMRuntime(instance_name))
1381 90c024f6 Guido Trotter
    except EnvironmentError, err:
1382 38e250ba Guido Trotter
      raise errors.HypervisorError("Failed to load KVM runtime file: %s" % err)
1383 38e250ba Guido Trotter
    return file_content
1384 38e250ba Guido Trotter
1385 38e250ba Guido Trotter
  def _SaveKVMRuntime(self, instance, kvm_runtime):
1386 38e250ba Guido Trotter
    """Save an instance's KVM runtime
1387 38e250ba Guido Trotter

1388 38e250ba Guido Trotter
    """
1389 c2672466 Guido Trotter
    kvm_cmd, kvm_nics, hvparams = kvm_runtime
1390 38e250ba Guido Trotter
    serialized_nics = [nic.ToDict() for nic in kvm_nics]
1391 c2672466 Guido Trotter
    serialized_form = serializer.Dump((kvm_cmd, serialized_nics, hvparams))
1392 38e250ba Guido Trotter
    self._WriteKVMRuntime(instance.name, serialized_form)
1393 38e250ba Guido Trotter
1394 30e42c4e Guido Trotter
  def _LoadKVMRuntime(self, instance, serialized_runtime=None):
1395 38e250ba Guido Trotter
    """Load an instance's KVM runtime
1396 38e250ba Guido Trotter

1397 38e250ba Guido Trotter
    """
1398 30e42c4e Guido Trotter
    if not serialized_runtime:
1399 30e42c4e Guido Trotter
      serialized_runtime = self._ReadKVMRuntime(instance.name)
1400 30e42c4e Guido Trotter
    loaded_runtime = serializer.Load(serialized_runtime)
1401 c2672466 Guido Trotter
    kvm_cmd, serialized_nics, hvparams = loaded_runtime
1402 38e250ba Guido Trotter
    kvm_nics = [objects.NIC.FromDict(snic) for snic in serialized_nics]
1403 c2672466 Guido Trotter
    return (kvm_cmd, kvm_nics, hvparams)
1404 38e250ba Guido Trotter
1405 5d9bfd87 Apollon Oikonomopoulos
  def _RunKVMCmd(self, name, kvm_cmd, tap_fds=None):
1406 76431533 Guido Trotter
    """Run the KVM cmd and check for errors
1407 76431533 Guido Trotter

1408 76431533 Guido Trotter
    @type name: string
1409 76431533 Guido Trotter
    @param name: instance name
1410 76431533 Guido Trotter
    @type kvm_cmd: list of strings
1411 76431533 Guido Trotter
    @param kvm_cmd: runcmd input for kvm
1412 5d9bfd87 Apollon Oikonomopoulos
    @type tap_fds: list of int
1413 5d9bfd87 Apollon Oikonomopoulos
    @param tap_fds: fds of tap devices opened by Ganeti
1414 76431533 Guido Trotter

1415 76431533 Guido Trotter
    """
1416 5d9bfd87 Apollon Oikonomopoulos
    try:
1417 5d9bfd87 Apollon Oikonomopoulos
      result = utils.RunCmd(kvm_cmd, noclose_fds=tap_fds)
1418 5d9bfd87 Apollon Oikonomopoulos
    finally:
1419 5d9bfd87 Apollon Oikonomopoulos
      for fd in tap_fds:
1420 5d9bfd87 Apollon Oikonomopoulos
        utils_wrapper.CloseFdNoError(fd)
1421 5d9bfd87 Apollon Oikonomopoulos
1422 76431533 Guido Trotter
    if result.failed:
1423 76431533 Guido Trotter
      raise errors.HypervisorError("Failed to start instance %s: %s (%s)" %
1424 76431533 Guido Trotter
                                   (name, result.fail_reason, result.output))
1425 76431533 Guido Trotter
    if not self._InstancePidAlive(name)[2]:
1426 76431533 Guido Trotter
      raise errors.HypervisorError("Failed to start instance %s" % name)
1427 76431533 Guido Trotter
1428 b73f1c59 Guido Trotter
  def _ExecuteKVMRuntime(self, instance, kvm_runtime, kvmhelp, incoming=None):
1429 839642c2 Iustin Pop
    """Execute a KVM cmd, after completing it with some last minute data.
1430 ee5f20b0 Guido Trotter

1431 30e42c4e Guido Trotter
    @type incoming: tuple of strings
1432 30e42c4e Guido Trotter
    @param incoming: (target_host_ip, port)
1433 b73f1c59 Guido Trotter
    @type kvmhelp: string
1434 b73f1c59 Guido Trotter
    @param kvmhelp: output of kvm --help
1435 30e42c4e Guido Trotter

1436 ee5f20b0 Guido Trotter
    """
1437 297e6e53 Guido Trotter
    # Small _ExecuteKVMRuntime hv parameters programming howto:
1438 297e6e53 Guido Trotter
    #  - conf_hvp contains the parameters as configured on ganeti. they might
1439 297e6e53 Guido Trotter
    #    have changed since the instance started; only use them if the change
1440 297e6e53 Guido Trotter
    #    won't affect the inside of the instance (which hasn't been rebooted).
1441 297e6e53 Guido Trotter
    #  - up_hvp contains the parameters as they were when the instance was
1442 297e6e53 Guido Trotter
    #    started, plus any new parameter which has been added between ganeti
1443 297e6e53 Guido Trotter
    #    versions: it is paramount that those default to a value which won't
1444 297e6e53 Guido Trotter
    #    affect the inside of the instance as well.
1445 297e6e53 Guido Trotter
    conf_hvp = instance.hvparams
1446 5905901c Iustin Pop
    name = instance.name
1447 5905901c Iustin Pop
    self._CheckDown(name)
1448 ee5f20b0 Guido Trotter
1449 ee5f20b0 Guido Trotter
    temp_files = []
1450 ee5f20b0 Guido Trotter
1451 297e6e53 Guido Trotter
    kvm_cmd, kvm_nics, up_hvp = kvm_runtime
1452 f0db563d Guido Trotter
    # the first element of kvm_cmd is always the path to the kvm binary
1453 f0db563d Guido Trotter
    kvm_path = kvm_cmd[0]
1454 297e6e53 Guido Trotter
    up_hvp = objects.FillDict(conf_hvp, up_hvp)
1455 ee5f20b0 Guido Trotter
1456 297e6e53 Guido Trotter
    # We know it's safe to run as a different user upon migration, so we'll use
1457 297e6e53 Guido Trotter
    # the latest conf, from conf_hvp.
1458 297e6e53 Guido Trotter
    security_model = conf_hvp[constants.HV_SECURITY_MODEL]
1459 cef34868 Guido Trotter
    if security_model == constants.HT_SM_USER:
1460 297e6e53 Guido Trotter
      kvm_cmd.extend(["-runas", conf_hvp[constants.HV_SECURITY_DOMAIN]])
1461 cef34868 Guido Trotter
1462 839642c2 Iustin Pop
    keymap = conf_hvp[constants.HV_KEYMAP]
1463 839642c2 Iustin Pop
    if keymap:
1464 839642c2 Iustin Pop
      keymap_path = self._InstanceKeymapFile(name)
1465 839642c2 Iustin Pop
      # If a keymap file is specified, KVM won't use its internal defaults. By
1466 839642c2 Iustin Pop
      # first including the "en-us" layout, an error on loading the actual
1467 839642c2 Iustin Pop
      # layout (e.g. because it can't be found) won't lead to a non-functional
1468 839642c2 Iustin Pop
      # keyboard. A keyboard with incorrect keys is still better than none.
1469 839642c2 Iustin Pop
      utils.WriteFile(keymap_path, data="include en-us\ninclude %s\n" % keymap)
1470 839642c2 Iustin Pop
      kvm_cmd.extend(["-k", keymap_path])
1471 839642c2 Iustin Pop
1472 297e6e53 Guido Trotter
    # We have reasons to believe changing something like the nic driver/type
1473 297e6e53 Guido Trotter
    # upon migration won't exactly fly with the instance kernel, so for nic
1474 297e6e53 Guido Trotter
    # related parameters we'll use up_hvp
1475 5d9bfd87 Apollon Oikonomopoulos
    tapfds = []
1476 5d9bfd87 Apollon Oikonomopoulos
    taps = []
1477 ee5f20b0 Guido Trotter
    if not kvm_nics:
1478 fbe27e2b Guido Trotter
      kvm_cmd.extend(["-net", "none"])
1479 ee5f20b0 Guido Trotter
    else:
1480 5d9bfd87 Apollon Oikonomopoulos
      vnet_hdr = False
1481 fbe27e2b Guido Trotter
      tap_extra = ""
1482 297e6e53 Guido Trotter
      nic_type = up_hvp[constants.HV_NIC_TYPE]
1483 37f88dc6 Guido Trotter
      if nic_type == constants.HT_NIC_PARAVIRTUAL:
1484 f0db563d Guido Trotter
        nic_model = self._VIRTIO
1485 f0db563d Guido Trotter
        try:
1486 f0db563d Guido Trotter
          devlist = self._GetKVMOutput(kvm_path, self._KVMOPT_DEVICELIST)
1487 f0db563d Guido Trotter
          if self._NEW_VIRTIO_RE.search(devlist):
1488 f0db563d Guido Trotter
            nic_model = self._VIRTIO_NET_PCI
1489 f0db563d Guido Trotter
            vnet_hdr = True
1490 f0db563d Guido Trotter
        except errors.HypervisorError, _:
1491 f0db563d Guido Trotter
          # Older versions of kvm don't support DEVICE_LIST, but they don't
1492 f0db563d Guido Trotter
          # have new virtio syntax either.
1493 f0db563d Guido Trotter
          pass
1494 4b784cf8 Miguel Di Ciurcio Filho
1495 297e6e53 Guido Trotter
        if up_hvp[constants.HV_VHOST_NET]:
1496 3b59ec02 Guido Trotter
          # check for vhost_net support
1497 0ad7f5d8 Guido Trotter
          if self._VHOST_RE.search(kvmhelp):
1498 4b784cf8 Miguel Di Ciurcio Filho
            tap_extra = ",vhost=on"
1499 4b784cf8 Miguel Di Ciurcio Filho
          else:
1500 4b784cf8 Miguel Di Ciurcio Filho
            raise errors.HypervisorError("vhost_net is configured"
1501 5ae4945a Iustin Pop
                                         " but it is not available")
1502 37f88dc6 Guido Trotter
      else:
1503 4b784cf8 Miguel Di Ciurcio Filho
        nic_model = nic_type
1504 37f88dc6 Guido Trotter
1505 6f4070cd Guido Trotter
      kvm_supports_netdev = self._NETDEV_RE.search(kvmhelp)
1506 6f4070cd Guido Trotter
1507 ee5f20b0 Guido Trotter
      for nic_seq, nic in enumerate(kvm_nics):
1508 6f1e1921 Michael Hanselmann
        tapname, tapfd = _OpenTap(vnet_hdr=vnet_hdr)
1509 5d9bfd87 Apollon Oikonomopoulos
        tapfds.append(tapfd)
1510 5d9bfd87 Apollon Oikonomopoulos
        taps.append(tapname)
1511 6f4070cd Guido Trotter
        if kvm_supports_netdev:
1512 4b784cf8 Miguel Di Ciurcio Filho
          nic_val = "%s,mac=%s,netdev=netdev%s" % (nic_model, nic.mac, nic_seq)
1513 5d9bfd87 Apollon Oikonomopoulos
          tap_val = "type=tap,id=netdev%s,fd=%d%s" % (nic_seq, tapfd, tap_extra)
1514 4b784cf8 Miguel Di Ciurcio Filho
          kvm_cmd.extend(["-netdev", tap_val, "-device", nic_val])
1515 4b784cf8 Miguel Di Ciurcio Filho
        else:
1516 049383d9 Adeodato Simo
          nic_val = "nic,vlan=%s,macaddr=%s,model=%s" % (nic_seq,
1517 049383d9 Adeodato Simo
                                                         nic.mac, nic_model)
1518 5d9bfd87 Apollon Oikonomopoulos
          tap_val = "tap,vlan=%s,fd=%d" % (nic_seq, tapfd)
1519 4b784cf8 Miguel Di Ciurcio Filho
          kvm_cmd.extend(["-net", tap_val, "-net", nic_val])
1520 ee5f20b0 Guido Trotter
1521 30e42c4e Guido Trotter
    if incoming:
1522 30e42c4e Guido Trotter
      target, port = incoming
1523 d0c8c01d Iustin Pop
      kvm_cmd.extend(["-incoming", "tcp:%s:%s" % (target, port)])
1524 30e42c4e Guido Trotter
1525 297e6e53 Guido Trotter
    # Changing the vnc password doesn't bother the guest that much. At most it
1526 297e6e53 Guido Trotter
    # will surprise people who connect to it. Whether positively or negatively
1527 297e6e53 Guido Trotter
    # it's debatable.
1528 297e6e53 Guido Trotter
    vnc_pwd_file = conf_hvp[constants.HV_VNC_PASSWORD_FILE]
1529 6e6bb8d5 Guido Trotter
    vnc_pwd = None
1530 6e6bb8d5 Guido Trotter
    if vnc_pwd_file:
1531 6e6bb8d5 Guido Trotter
      try:
1532 682f7601 Guido Trotter
        vnc_pwd = utils.ReadOneLineFile(vnc_pwd_file, strict=True)
1533 6e6bb8d5 Guido Trotter
      except EnvironmentError, err:
1534 6e6bb8d5 Guido Trotter
        raise errors.HypervisorError("Failed to open VNC password file %s: %s"
1535 6e6bb8d5 Guido Trotter
                                     % (vnc_pwd_file, err))
1536 6e6bb8d5 Guido Trotter
1537 297e6e53 Guido Trotter
    if conf_hvp[constants.HV_KVM_USE_CHROOT]:
1538 547a63b7 Balazs Lecz
      utils.EnsureDirs([(self._InstanceChrootDir(name),
1539 547a63b7 Balazs Lecz
                         constants.SECURE_DIR_MODE)])
1540 547a63b7 Balazs Lecz
1541 91c10532 Andrea Spadaccini
    # Automatically enable QMP if version is >= 0.14
1542 0ad7f5d8 Guido Trotter
    if self._QMP_RE.search(kvmhelp):
1543 91c10532 Andrea Spadaccini
      logging.debug("Enabling QMP")
1544 91c10532 Andrea Spadaccini
      kvm_cmd.extend(["-qmp", "unix:%s,server,nowait" %
1545 5ae4945a Iustin Pop
                      self._InstanceQmpMonitor(instance.name)])
1546 91c10532 Andrea Spadaccini
1547 cc8a8ed7 Apollon Oikonomopoulos
    # Configure the network now for starting instances and bridged interfaces,
1548 cc8a8ed7 Apollon Oikonomopoulos
    # during FinalizeMigration for incoming instances' routed interfaces
1549 cc8a8ed7 Apollon Oikonomopoulos
    for nic_seq, nic in enumerate(kvm_nics):
1550 cc8a8ed7 Apollon Oikonomopoulos
      if (incoming and
1551 cc8a8ed7 Apollon Oikonomopoulos
          nic.nicparams[constants.NIC_MODE] != constants.NIC_MODE_BRIDGED):
1552 cc8a8ed7 Apollon Oikonomopoulos
        continue
1553 cc8a8ed7 Apollon Oikonomopoulos
      self._ConfigureNIC(instance, nic_seq, nic, taps[nic_seq])
1554 5d9bfd87 Apollon Oikonomopoulos
1555 1d8a7812 Andrea Spadaccini
    # CPU affinity requires kvm to start paused, so we set this flag if the
1556 1d8a7812 Andrea Spadaccini
    # instance is not already paused and if we are not going to accept a
1557 1d8a7812 Andrea Spadaccini
    # migrating instance. In the latter case, pausing is not needed.
1558 1d8a7812 Andrea Spadaccini
    start_kvm_paused = not (_KVM_START_PAUSED_FLAG in kvm_cmd) and not incoming
1559 945a7e67 Guido Trotter
    if start_kvm_paused:
1560 945a7e67 Guido Trotter
      kvm_cmd.extend([_KVM_START_PAUSED_FLAG])
1561 b693125f Tsachy Shacham
1562 b693125f Tsachy Shacham
    # Note: CPU pinning is using up_hvp since changes take effect
1563 b693125f Tsachy Shacham
    # during instance startup anyway, and to avoid problems when soft
1564 b693125f Tsachy Shacham
    # rebooting the instance.
1565 c607b1f7 Tsachy Shacham
    cpu_pinning = False
1566 b693125f Tsachy Shacham
    if up_hvp.get(constants.HV_CPU_MASK, None):
1567 b693125f Tsachy Shacham
      cpu_pinning = True
1568 b693125f Tsachy Shacham
1569 76431533 Guido Trotter
    if security_model == constants.HT_SM_POOL:
1570 76431533 Guido Trotter
      ss = ssconf.SimpleStore()
1571 76431533 Guido Trotter
      uid_pool = uidpool.ParseUidPool(ss.GetUidPool(), separator="\n")
1572 76431533 Guido Trotter
      all_uids = set(uidpool.ExpandUidPool(uid_pool))
1573 76431533 Guido Trotter
      uid = uidpool.RequestUnusedUid(all_uids)
1574 76431533 Guido Trotter
      try:
1575 76431533 Guido Trotter
        username = pwd.getpwuid(uid.GetUid()).pw_name
1576 76431533 Guido Trotter
        kvm_cmd.extend(["-runas", username])
1577 5d9bfd87 Apollon Oikonomopoulos
        self._RunKVMCmd(name, kvm_cmd, tapfds)
1578 76431533 Guido Trotter
      except:
1579 76431533 Guido Trotter
        uidpool.ReleaseUid(uid)
1580 76431533 Guido Trotter
        raise
1581 76431533 Guido Trotter
      else:
1582 76431533 Guido Trotter
        uid.Unlock()
1583 3af16328 Apollon Oikonomopoulos
        utils.WriteFile(self._InstanceUidFile(name), data=uid.AsStr())
1584 76431533 Guido Trotter
    else:
1585 5d9bfd87 Apollon Oikonomopoulos
      self._RunKVMCmd(name, kvm_cmd, tapfds)
1586 5d9bfd87 Apollon Oikonomopoulos
1587 5d9bfd87 Apollon Oikonomopoulos
    utils.EnsureDirs([(self._InstanceNICDir(instance.name),
1588 5d9bfd87 Apollon Oikonomopoulos
                     constants.RUN_DIRS_MODE)])
1589 5d9bfd87 Apollon Oikonomopoulos
    for nic_seq, tap in enumerate(taps):
1590 5d9bfd87 Apollon Oikonomopoulos
      utils.WriteFile(self._InstanceNICFile(instance.name, nic_seq),
1591 5d9bfd87 Apollon Oikonomopoulos
                      data=tap)
1592 eb58f9b1 Guido Trotter
1593 6e6bb8d5 Guido Trotter
    if vnc_pwd:
1594 d0c8c01d Iustin Pop
      change_cmd = "change vnc password %s" % vnc_pwd
1595 6e6bb8d5 Guido Trotter
      self._CallMonitorCommand(instance.name, change_cmd)
1596 6e6bb8d5 Guido Trotter
1597 b451c4c7 Andrea Spadaccini
    # Setting SPICE password. We are not vulnerable to malicious passwordless
1598 b451c4c7 Andrea Spadaccini
    # connection attempts because SPICE by default does not allow connections
1599 b451c4c7 Andrea Spadaccini
    # if neither a password nor the "disable_ticketing" options are specified.
1600 b451c4c7 Andrea Spadaccini
    # As soon as we send the password via QMP, that password is a valid ticket
1601 b451c4c7 Andrea Spadaccini
    # for connection.
1602 b451c4c7 Andrea Spadaccini
    spice_password_file = conf_hvp[constants.HV_KVM_SPICE_PASSWORD_FILE]
1603 b451c4c7 Andrea Spadaccini
    if spice_password_file:
1604 fc84cd5d Guido Trotter
      spice_pwd = ""
1605 b451c4c7 Andrea Spadaccini
      try:
1606 b451c4c7 Andrea Spadaccini
        spice_pwd = utils.ReadOneLineFile(spice_password_file, strict=True)
1607 b451c4c7 Andrea Spadaccini
      except EnvironmentError, err:
1608 b451c4c7 Andrea Spadaccini
        raise errors.HypervisorError("Failed to open SPICE password file %s: %s"
1609 b451c4c7 Andrea Spadaccini
                                     % (spice_password_file, err))
1610 b451c4c7 Andrea Spadaccini
1611 fc84cd5d Guido Trotter
      qmp = QmpConnection(self._InstanceQmpMonitor(instance.name))
1612 fc84cd5d Guido Trotter
      qmp.connect()
1613 fc84cd5d Guido Trotter
      arguments = {
1614 fc84cd5d Guido Trotter
          "protocol": "spice",
1615 fc84cd5d Guido Trotter
          "password": spice_pwd,
1616 fc84cd5d Guido Trotter
      }
1617 fc84cd5d Guido Trotter
      qmp.Execute("set_password", arguments)
1618 fc84cd5d Guido Trotter
1619 08137f9e Iustin Pop
    for filename in temp_files:
1620 08137f9e Iustin Pop
      utils.RemoveFile(filename)
1621 eb58f9b1 Guido Trotter
1622 b693125f Tsachy Shacham
    # If requested, set CPU affinity and resume instance execution
1623 b693125f Tsachy Shacham
    if cpu_pinning:
1624 945a7e67 Guido Trotter
      self._ExecuteCpuAffinity(instance.name, up_hvp[constants.HV_CPU_MASK])
1625 945a7e67 Guido Trotter
1626 61eb1a46 Guido Trotter
    start_memory = self._InstanceStartupMemory(instance)
1627 61eb1a46 Guido Trotter
    if start_memory < instance.beparams[constants.BE_MAXMEM]:
1628 61eb1a46 Guido Trotter
      self.BalloonInstanceMemory(instance, start_memory)
1629 61eb1a46 Guido Trotter
1630 945a7e67 Guido Trotter
    if start_kvm_paused:
1631 2ed0e208 Iustin Pop
      # To control CPU pinning, ballooning, and vnc/spice passwords
1632 2ed0e208 Iustin Pop
      # the VM was started in a frozen state. If freezing was not
1633 2ed0e208 Iustin Pop
      # explicitly requested resume the vm status.
1634 945a7e67 Guido Trotter
      self._CallMonitorCommand(instance.name, self._CONT_CMD)
1635 b693125f Tsachy Shacham
1636 323f9095 Stephen Shirley
  def StartInstance(self, instance, block_devices, startup_paused):
1637 ee5f20b0 Guido Trotter
    """Start an instance.
1638 ee5f20b0 Guido Trotter

1639 ee5f20b0 Guido Trotter
    """
1640 5905901c Iustin Pop
    self._CheckDown(instance.name)
1641 4b9638dc Guido Trotter
    kvmpath = instance.hvparams[constants.HV_KVM_PATH]
1642 6e043e60 Guido Trotter
    kvmhelp = self._GetKVMOutput(kvmpath, self._KVMOPT_HELP)
1643 7238edb5 Iustin Pop
    kvm_runtime = self._GenerateKVMRuntime(instance, block_devices,
1644 b73f1c59 Guido Trotter
                                           startup_paused, kvmhelp)
1645 38e250ba Guido Trotter
    self._SaveKVMRuntime(instance, kvm_runtime)
1646 b73f1c59 Guido Trotter
    self._ExecuteKVMRuntime(instance, kvm_runtime, kvmhelp)
1647 ee5f20b0 Guido Trotter
1648 6567aff3 Guido Trotter
  def _CallMonitorCommand(self, instance_name, command):
1649 6567aff3 Guido Trotter
    """Invoke a command on the instance monitor.
1650 6567aff3 Guido Trotter

1651 6567aff3 Guido Trotter
    """
1652 6567aff3 Guido Trotter
    socat = ("echo %s | %s STDIO UNIX-CONNECT:%s" %
1653 6567aff3 Guido Trotter
             (utils.ShellQuote(command),
1654 6567aff3 Guido Trotter
              constants.SOCAT_PATH,
1655 6567aff3 Guido Trotter
              utils.ShellQuote(self._InstanceMonitor(instance_name))))
1656 6567aff3 Guido Trotter
    result = utils.RunCmd(socat)
1657 6567aff3 Guido Trotter
    if result.failed:
1658 6567aff3 Guido Trotter
      msg = ("Failed to send command '%s' to instance %s."
1659 6567aff3 Guido Trotter
             " output: %s, error: %s, fail_reason: %s" %
1660 9798fcae Guido Trotter
             (command, instance_name,
1661 9798fcae Guido Trotter
              result.stdout, result.stderr, result.fail_reason))
1662 6567aff3 Guido Trotter
      raise errors.HypervisorError(msg)
1663 6567aff3 Guido Trotter
1664 6567aff3 Guido Trotter
    return result
1665 6567aff3 Guido Trotter
1666 b52d85c1 Guido Trotter
  @classmethod
1667 585c8187 Guido Trotter
  def _ParseKVMVersion(cls, text):
1668 585c8187 Guido Trotter
    """Parse the KVM version from the --help output.
1669 585c8187 Guido Trotter

1670 585c8187 Guido Trotter
    @type text: string
1671 585c8187 Guido Trotter
    @param text: output of kvm --help
1672 585c8187 Guido Trotter
    @return: (version, v_maj, v_min, v_rev)
1673 5c3d5dfd Iustin Pop
    @raise errors.HypervisorError: when the KVM version cannot be retrieved
1674 585c8187 Guido Trotter

1675 585c8187 Guido Trotter
    """
1676 585c8187 Guido Trotter
    match = cls._VERSION_RE.search(text.splitlines()[0])
1677 585c8187 Guido Trotter
    if not match:
1678 585c8187 Guido Trotter
      raise errors.HypervisorError("Unable to get KVM version")
1679 585c8187 Guido Trotter
1680 585c8187 Guido Trotter
    v_all = match.group(0)
1681 585c8187 Guido Trotter
    v_maj = int(match.group(1))
1682 585c8187 Guido Trotter
    v_min = int(match.group(2))
1683 585c8187 Guido Trotter
    if match.group(4):
1684 585c8187 Guido Trotter
      v_rev = int(match.group(4))
1685 585c8187 Guido Trotter
    else:
1686 585c8187 Guido Trotter
      v_rev = 0
1687 585c8187 Guido Trotter
    return (v_all, v_maj, v_min, v_rev)
1688 585c8187 Guido Trotter
1689 585c8187 Guido Trotter
  @classmethod
1690 6e043e60 Guido Trotter
  def _GetKVMOutput(cls, kvm_path, option):
1691 6e043e60 Guido Trotter
    """Return the output of a kvm invocation
1692 e3b89628 Guido Trotter

1693 bc0fed4b Guido Trotter
    @type kvm_path: string
1694 bc0fed4b Guido Trotter
    @param kvm_path: path to the kvm executable
1695 bc0fed4b Guido Trotter
    @type option: a key of _KVMOPTS_CMDS
1696 bc0fed4b Guido Trotter
    @param option: kvm option to fetch the output from
1697 6e043e60 Guido Trotter
    @return: output a supported kvm invocation
1698 e3b89628 Guido Trotter
    @raise errors.HypervisorError: when the KVM help output cannot be retrieved
1699 e3b89628 Guido Trotter

1700 e3b89628 Guido Trotter
    """
1701 6e043e60 Guido Trotter
    assert option in cls._KVMOPTS_CMDS, "Invalid output option"
1702 6e043e60 Guido Trotter
1703 bc0fed4b Guido Trotter
    optlist, can_fail = cls._KVMOPTS_CMDS[option]
1704 bc0fed4b Guido Trotter
1705 bc0fed4b Guido Trotter
    result = utils.RunCmd([kvm_path] + optlist)
1706 bc0fed4b Guido Trotter
    if result.failed and not can_fail:
1707 af89fa76 Helga Velroyen
      raise errors.HypervisorError("Unable to get KVM %s output" %
1708 af89fa76 Helga Velroyen
                                    " ".join(cls._KVMOPTS_CMDS[option]))
1709 e3b89628 Guido Trotter
    return result.output
1710 e3b89628 Guido Trotter
1711 e3b89628 Guido Trotter
  @classmethod
1712 3b59ec02 Guido Trotter
  def _GetKVMVersion(cls, kvm_path):
1713 d7e4a2b1 Andrea Spadaccini
    """Return the installed KVM version.
1714 b52d85c1 Guido Trotter

1715 440351f8 Andrea Spadaccini
    @return: (version, v_maj, v_min, v_rev)
1716 5c3d5dfd Iustin Pop
    @raise errors.HypervisorError: when the KVM version cannot be retrieved
1717 b52d85c1 Guido Trotter

1718 b52d85c1 Guido Trotter
    """
1719 6e043e60 Guido Trotter
    return cls._ParseKVMVersion(cls._GetKVMOutput(kvm_path, cls._KVMOPT_HELP))
1720 82e3bf85 Dimitris Aragiorgis
1721 82e3bf85 Dimitris Aragiorgis
  @classmethod
1722 82e3bf85 Dimitris Aragiorgis
  def _GetDefaultMachineVersion(cls, kvm_path):
1723 82e3bf85 Dimitris Aragiorgis
    """Return the default hardware revision (e.g. pc-1.1)
1724 82e3bf85 Dimitris Aragiorgis

1725 82e3bf85 Dimitris Aragiorgis
    """
1726 6e043e60 Guido Trotter
    output = cls._GetKVMOutput(kvm_path, cls._KVMOPT_MLIST)
1727 82e3bf85 Dimitris Aragiorgis
    match = cls._DEFAULT_MACHINE_VERSION_RE.search(output)
1728 82e3bf85 Dimitris Aragiorgis
    if match:
1729 82e3bf85 Dimitris Aragiorgis
      return match.group(1)
1730 82e3bf85 Dimitris Aragiorgis
    else:
1731 82e3bf85 Dimitris Aragiorgis
      return "pc"
1732 82e3bf85 Dimitris Aragiorgis
1733 bbcf7ad0 Iustin Pop
  def StopInstance(self, instance, force=False, retry=False, name=None):
1734 eb58f9b1 Guido Trotter
    """Stop an instance.
1735 eb58f9b1 Guido Trotter

1736 eb58f9b1 Guido Trotter
    """
1737 bbcf7ad0 Iustin Pop
    if name is not None and not force:
1738 bbcf7ad0 Iustin Pop
      raise errors.HypervisorError("Cannot shutdown cleanly by name only")
1739 bbcf7ad0 Iustin Pop
    if name is None:
1740 bbcf7ad0 Iustin Pop
      name = instance.name
1741 bbcf7ad0 Iustin Pop
      acpi = instance.hvparams[constants.HV_ACPI]
1742 bbcf7ad0 Iustin Pop
    else:
1743 bbcf7ad0 Iustin Pop
      acpi = False
1744 8904b35c Guido Trotter
    _, pid, alive = self._InstancePidAlive(name)
1745 1f8b3a27 Guido Trotter
    if pid > 0 and alive:
1746 bbcf7ad0 Iustin Pop
      if force or not acpi:
1747 eb58f9b1 Guido Trotter
        utils.KillProcess(pid)
1748 eb58f9b1 Guido Trotter
      else:
1749 d0c8c01d Iustin Pop
        self._CallMonitorCommand(name, "system_powerdown")
1750 eb58f9b1 Guido Trotter
1751 8904b35c Guido Trotter
  def CleanupInstance(self, instance_name):
1752 8904b35c Guido Trotter
    """Cleanup after a stopped instance
1753 8904b35c Guido Trotter

1754 8904b35c Guido Trotter
    """
1755 8904b35c Guido Trotter
    pidfile, pid, alive = self._InstancePidAlive(instance_name)
1756 8904b35c Guido Trotter
    if pid > 0 and alive:
1757 8904b35c Guido Trotter
      raise errors.HypervisorError("Cannot cleanup a live instance")
1758 8904b35c Guido Trotter
    self._RemoveInstanceRuntimeFiles(pidfile, instance_name)
1759 eb58f9b1 Guido Trotter
1760 eb58f9b1 Guido Trotter
  def RebootInstance(self, instance):
1761 eb58f9b1 Guido Trotter
    """Reboot an instance.
1762 eb58f9b1 Guido Trotter

1763 eb58f9b1 Guido Trotter
    """
1764 eb58f9b1 Guido Trotter
    # For some reason if we do a 'send-key ctrl-alt-delete' to the control
1765 eb58f9b1 Guido Trotter
    # socket the instance will stop, but now power up again. So we'll resort
1766 eb58f9b1 Guido Trotter
    # to shutdown and restart.
1767 1122eb25 Iustin Pop
    _, _, alive = self._InstancePidAlive(instance.name)
1768 1f8b3a27 Guido Trotter
    if not alive:
1769 78411c60 Iustin Pop
      raise errors.HypervisorError("Failed to reboot instance %s:"
1770 78411c60 Iustin Pop
                                   " not running" % instance.name)
1771 f02881e0 Guido Trotter
    # StopInstance will delete the saved KVM runtime so:
1772 f02881e0 Guido Trotter
    # ...first load it...
1773 f02881e0 Guido Trotter
    kvm_runtime = self._LoadKVMRuntime(instance)
1774 f02881e0 Guido Trotter
    # ...now we can safely call StopInstance...
1775 f02881e0 Guido Trotter
    if not self.StopInstance(instance):
1776 f02881e0 Guido Trotter
      self.StopInstance(instance, force=True)
1777 f02881e0 Guido Trotter
    # ...and finally we can save it again, and execute it...
1778 f02881e0 Guido Trotter
    self._SaveKVMRuntime(instance, kvm_runtime)
1779 4b9638dc Guido Trotter
    kvmpath = instance.hvparams[constants.HV_KVM_PATH]
1780 6e043e60 Guido Trotter
    kvmhelp = self._GetKVMOutput(kvmpath, self._KVMOPT_HELP)
1781 b73f1c59 Guido Trotter
    self._ExecuteKVMRuntime(instance, kvm_runtime, kvmhelp)
1782 eb58f9b1 Guido Trotter
1783 30e42c4e Guido Trotter
  def MigrationInfo(self, instance):
1784 30e42c4e Guido Trotter
    """Get instance information to perform a migration.
1785 30e42c4e Guido Trotter

1786 30e42c4e Guido Trotter
    @type instance: L{objects.Instance}
1787 30e42c4e Guido Trotter
    @param instance: instance to be migrated
1788 30e42c4e Guido Trotter
    @rtype: string
1789 30e42c4e Guido Trotter
    @return: content of the KVM runtime file
1790 30e42c4e Guido Trotter

1791 30e42c4e Guido Trotter
    """
1792 30e42c4e Guido Trotter
    return self._ReadKVMRuntime(instance.name)
1793 30e42c4e Guido Trotter
1794 30e42c4e Guido Trotter
  def AcceptInstance(self, instance, info, target):
1795 30e42c4e Guido Trotter
    """Prepare to accept an instance.
1796 30e42c4e Guido Trotter

1797 30e42c4e Guido Trotter
    @type instance: L{objects.Instance}
1798 30e42c4e Guido Trotter
    @param instance: instance to be accepted
1799 30e42c4e Guido Trotter
    @type info: string
1800 30e42c4e Guido Trotter
    @param info: content of the KVM runtime file on the source node
1801 30e42c4e Guido Trotter
    @type target: string
1802 30e42c4e Guido Trotter
    @param target: target host (usually ip), on this node
1803 30e42c4e Guido Trotter

1804 30e42c4e Guido Trotter
    """
1805 30e42c4e Guido Trotter
    kvm_runtime = self._LoadKVMRuntime(instance, serialized_runtime=info)
1806 641ae041 Iustin Pop
    incoming_address = (target, instance.hvparams[constants.HV_MIGRATION_PORT])
1807 4b9638dc Guido Trotter
    kvmpath = instance.hvparams[constants.HV_KVM_PATH]
1808 6e043e60 Guido Trotter
    kvmhelp = self._GetKVMOutput(kvmpath, self._KVMOPT_HELP)
1809 b73f1c59 Guido Trotter
    self._ExecuteKVMRuntime(instance, kvm_runtime, kvmhelp,
1810 b73f1c59 Guido Trotter
                            incoming=incoming_address)
1811 30e42c4e Guido Trotter
1812 6a1434d7 Andrea Spadaccini
  def FinalizeMigrationDst(self, instance, info, success):
1813 6a1434d7 Andrea Spadaccini
    """Finalize the instance migration on the target node.
1814 30e42c4e Guido Trotter

1815 30e42c4e Guido Trotter
    Stop the incoming mode KVM.
1816 30e42c4e Guido Trotter

1817 30e42c4e Guido Trotter
    @type instance: L{objects.Instance}
1818 fea922fa Guido Trotter
    @param instance: instance whose migration is being finalized
1819 30e42c4e Guido Trotter

1820 30e42c4e Guido Trotter
    """
1821 30e42c4e Guido Trotter
    if success:
1822 5d9bfd87 Apollon Oikonomopoulos
      kvm_runtime = self._LoadKVMRuntime(instance, serialized_runtime=info)
1823 5d9bfd87 Apollon Oikonomopoulos
      kvm_nics = kvm_runtime[1]
1824 5d9bfd87 Apollon Oikonomopoulos
1825 5d9bfd87 Apollon Oikonomopoulos
      for nic_seq, nic in enumerate(kvm_nics):
1826 cc8a8ed7 Apollon Oikonomopoulos
        if nic.nicparams[constants.NIC_MODE] == constants.NIC_MODE_BRIDGED:
1827 cc8a8ed7 Apollon Oikonomopoulos
          # Bridged interfaces have already been configured
1828 cc8a8ed7 Apollon Oikonomopoulos
          continue
1829 5d9bfd87 Apollon Oikonomopoulos
        try:
1830 5d9bfd87 Apollon Oikonomopoulos
          tap = utils.ReadFile(self._InstanceNICFile(instance.name, nic_seq))
1831 5d9bfd87 Apollon Oikonomopoulos
        except EnvironmentError, err:
1832 5d9bfd87 Apollon Oikonomopoulos
          logging.warning("Failed to find host interface for %s NIC #%d: %s",
1833 5d9bfd87 Apollon Oikonomopoulos
                          instance.name, nic_seq, str(err))
1834 5d9bfd87 Apollon Oikonomopoulos
          continue
1835 5d9bfd87 Apollon Oikonomopoulos
        try:
1836 5d9bfd87 Apollon Oikonomopoulos
          self._ConfigureNIC(instance, nic_seq, nic, tap)
1837 5d9bfd87 Apollon Oikonomopoulos
        except errors.HypervisorError, err:
1838 5d9bfd87 Apollon Oikonomopoulos
          logging.warning(str(err))
1839 5d9bfd87 Apollon Oikonomopoulos
1840 30e42c4e Guido Trotter
      self._WriteKVMRuntime(instance.name, info)
1841 30e42c4e Guido Trotter
    else:
1842 30e42c4e Guido Trotter
      self.StopInstance(instance, force=True)
1843 30e42c4e Guido Trotter
1844 58d38b02 Iustin Pop
  def MigrateInstance(self, instance, target, live):
1845 30e42c4e Guido Trotter
    """Migrate an instance to a target node.
1846 30e42c4e Guido Trotter

1847 30e42c4e Guido Trotter
    The migration will not be attempted if the instance is not
1848 30e42c4e Guido Trotter
    currently running.
1849 30e42c4e Guido Trotter

1850 58d38b02 Iustin Pop
    @type instance: L{objects.Instance}
1851 58d38b02 Iustin Pop
    @param instance: the instance to be migrated
1852 30e42c4e Guido Trotter
    @type target: string
1853 30e42c4e Guido Trotter
    @param target: ip address of the target node
1854 30e42c4e Guido Trotter
    @type live: boolean
1855 30e42c4e Guido Trotter
    @param live: perform a live migration
1856 30e42c4e Guido Trotter

1857 30e42c4e Guido Trotter
    """
1858 58d38b02 Iustin Pop
    instance_name = instance.name
1859 50716be0 Iustin Pop
    port = instance.hvparams[constants.HV_MIGRATION_PORT]
1860 6a1434d7 Andrea Spadaccini
    _, _, alive = self._InstancePidAlive(instance_name)
1861 30e42c4e Guido Trotter
    if not alive:
1862 30e42c4e Guido Trotter
      raise errors.HypervisorError("Instance not running, cannot migrate")
1863 30e42c4e Guido Trotter
1864 30e42c4e Guido Trotter
    if not live:
1865 d0c8c01d Iustin Pop
      self._CallMonitorCommand(instance_name, "stop")
1866 30e42c4e Guido Trotter
1867 d0c8c01d Iustin Pop
    migrate_command = ("migrate_set_speed %dm" %
1868 5ae4945a Iustin Pop
                       instance.hvparams[constants.HV_MIGRATION_BANDWIDTH])
1869 e43d4f9f Apollon Oikonomopoulos
    self._CallMonitorCommand(instance_name, migrate_command)
1870 e43d4f9f Apollon Oikonomopoulos
1871 d0c8c01d Iustin Pop
    migrate_command = ("migrate_set_downtime %dms" %
1872 5ae4945a Iustin Pop
                       instance.hvparams[constants.HV_MIGRATION_DOWNTIME])
1873 e43d4f9f Apollon Oikonomopoulos
    self._CallMonitorCommand(instance_name, migrate_command)
1874 e43d4f9f Apollon Oikonomopoulos
1875 d0c8c01d Iustin Pop
    migrate_command = "migrate -d tcp:%s:%s" % (target, port)
1876 30e42c4e Guido Trotter
    self._CallMonitorCommand(instance_name, migrate_command)
1877 30e42c4e Guido Trotter
1878 6a1434d7 Andrea Spadaccini
  def FinalizeMigrationSource(self, instance, success, live):
1879 6a1434d7 Andrea Spadaccini
    """Finalize the instance migration on the source node.
1880 6a1434d7 Andrea Spadaccini

1881 6a1434d7 Andrea Spadaccini
    @type instance: L{objects.Instance}
1882 6a1434d7 Andrea Spadaccini
    @param instance: the instance that was migrated
1883 6a1434d7 Andrea Spadaccini
    @type success: bool
1884 6a1434d7 Andrea Spadaccini
    @param success: whether the migration succeeded or not
1885 6a1434d7 Andrea Spadaccini
    @type live: bool
1886 6a1434d7 Andrea Spadaccini
    @param live: whether the user requested a live migration or not
1887 6a1434d7 Andrea Spadaccini

1888 6a1434d7 Andrea Spadaccini
    """
1889 6a1434d7 Andrea Spadaccini
    if success:
1890 6a1434d7 Andrea Spadaccini
      pidfile, pid, _ = self._InstancePidAlive(instance.name)
1891 6a1434d7 Andrea Spadaccini
      utils.KillProcess(pid)
1892 6a1434d7 Andrea Spadaccini
      self._RemoveInstanceRuntimeFiles(pidfile, instance.name)
1893 6a1434d7 Andrea Spadaccini
    elif live:
1894 6a1434d7 Andrea Spadaccini
      self._CallMonitorCommand(instance.name, self._CONT_CMD)
1895 6a1434d7 Andrea Spadaccini
1896 6a1434d7 Andrea Spadaccini
  def GetMigrationStatus(self, instance):
1897 6a1434d7 Andrea Spadaccini
    """Get the migration status
1898 6a1434d7 Andrea Spadaccini

1899 6a1434d7 Andrea Spadaccini
    @type instance: L{objects.Instance}
1900 6a1434d7 Andrea Spadaccini
    @param instance: the instance that is being migrated
1901 6a1434d7 Andrea Spadaccini
    @rtype: L{objects.MigrationStatus}
1902 6a1434d7 Andrea Spadaccini
    @return: the status of the current migration (one of
1903 6a1434d7 Andrea Spadaccini
             L{constants.HV_MIGRATION_VALID_STATUSES}), plus any additional
1904 6a1434d7 Andrea Spadaccini
             progress info that can be retrieved from the hypervisor
1905 6a1434d7 Andrea Spadaccini

1906 6a1434d7 Andrea Spadaccini
    """
1907 d0c8c01d Iustin Pop
    info_command = "info migrate"
1908 6a1434d7 Andrea Spadaccini
    for _ in range(self._MIGRATION_INFO_MAX_BAD_ANSWERS):
1909 6a1434d7 Andrea Spadaccini
      result = self._CallMonitorCommand(instance.name, info_command)
1910 30e42c4e Guido Trotter
      match = self._MIGRATION_STATUS_RE.search(result.stdout)
1911 30e42c4e Guido Trotter
      if not match:
1912 c4e388a5 Guido Trotter
        if not result.stdout:
1913 c4e388a5 Guido Trotter
          logging.info("KVM: empty 'info migrate' result")
1914 c4e388a5 Guido Trotter
        else:
1915 e4dd2299 Guido Trotter
          logging.warning("KVM: unknown 'info migrate' result: %s",
1916 c4e388a5 Guido Trotter
                          result.stdout)
1917 30e42c4e Guido Trotter
      else:
1918 30e42c4e Guido Trotter
        status = match.group(1)
1919 6a1434d7 Andrea Spadaccini
        if status in constants.HV_KVM_MIGRATION_VALID_STATUSES:
1920 6a1434d7 Andrea Spadaccini
          migration_status = objects.MigrationStatus(status=status)
1921 61643226 Andrea Spadaccini
          match = self._MIGRATION_PROGRESS_RE.search(result.stdout)
1922 61643226 Andrea Spadaccini
          if match:
1923 61643226 Andrea Spadaccini
            migration_status.transferred_ram = match.group("transferred")
1924 61643226 Andrea Spadaccini
            migration_status.total_ram = match.group("total")
1925 61643226 Andrea Spadaccini
1926 6a1434d7 Andrea Spadaccini
          return migration_status
1927 30e42c4e Guido Trotter
1928 6a1434d7 Andrea Spadaccini
        logging.warning("KVM: unknown migration status '%s'", status)
1929 6a1434d7 Andrea Spadaccini
1930 6a1434d7 Andrea Spadaccini
      time.sleep(self._MIGRATION_INFO_RETRY_DELAY)
1931 6a1434d7 Andrea Spadaccini
1932 62457f51 Iustin Pop
    return objects.MigrationStatus(status=constants.HV_MIGRATION_FAILED)
1933 30e42c4e Guido Trotter
1934 3d836750 Guido Trotter
  def BalloonInstanceMemory(self, instance, mem):
1935 3d836750 Guido Trotter
    """Balloon an instance memory to a certain value.
1936 3d836750 Guido Trotter

1937 3d836750 Guido Trotter
    @type instance: L{objects.Instance}
1938 3d836750 Guido Trotter
    @param instance: instance to be accepted
1939 3d836750 Guido Trotter
    @type mem: int
1940 3d836750 Guido Trotter
    @param mem: actual memory size to use for instance runtime
1941 3d836750 Guido Trotter

1942 3d836750 Guido Trotter
    """
1943 3d836750 Guido Trotter
    self._CallMonitorCommand(instance.name, "balloon %d" % mem)
1944 3d836750 Guido Trotter
1945 eb58f9b1 Guido Trotter
  def GetNodeInfo(self):
1946 eb58f9b1 Guido Trotter
    """Return information about the node.
1947 eb58f9b1 Guido Trotter

1948 c41eea6e Iustin Pop
    @return: a dict with the following keys (values in MiB):
1949 c41eea6e Iustin Pop
          - memory_total: the total memory size on the node
1950 c41eea6e Iustin Pop
          - memory_free: the available memory on the node for instances
1951 c41eea6e Iustin Pop
          - memory_dom0: the memory used by the node itself, if available
1952 34fbc862 Andrea Spadaccini
          - hv_version: the hypervisor version in the form (major, minor,
1953 34fbc862 Andrea Spadaccini
                        revision)
1954 eb58f9b1 Guido Trotter

1955 eb58f9b1 Guido Trotter
    """
1956 34fbc862 Andrea Spadaccini
    result = self.GetLinuxNodeInfo()
1957 4b9638dc Guido Trotter
    # FIXME: this is the global kvm version, but the actual version can be
1958 4b9638dc Guido Trotter
    # customized as an hv parameter. we should use the nodegroup's default kvm
1959 4b9638dc Guido Trotter
    # path parameter here.
1960 3b59ec02 Guido Trotter
    _, v_major, v_min, v_rev = self._GetKVMVersion(constants.KVM_PATH)
1961 34fbc862 Andrea Spadaccini
    result[constants.HV_NODEINFO_KEY_VERSION] = (v_major, v_min, v_rev)
1962 34fbc862 Andrea Spadaccini
    return result
1963 eb58f9b1 Guido Trotter
1964 637ce7f9 Guido Trotter
  @classmethod
1965 55cc0a44 Michael Hanselmann
  def GetInstanceConsole(cls, instance, hvparams, beparams):
1966 eb58f9b1 Guido Trotter
    """Return a command for connecting to the console of an instance.
1967 eb58f9b1 Guido Trotter

1968 eb58f9b1 Guido Trotter
    """
1969 a2faf9ee Guido Trotter
    if hvparams[constants.HV_SERIAL_CONSOLE]:
1970 9d9bded1 Michael Hanselmann
      cmd = [pathutils.KVM_CONSOLE_WRAPPER,
1971 2f4c951e Stephen Shirley
             constants.SOCAT_PATH, utils.ShellQuote(instance.name),
1972 2f4c951e Stephen Shirley
             utils.ShellQuote(cls._InstanceMonitor(instance.name)),
1973 55cc0a44 Michael Hanselmann
             "STDIO,%s" % cls._SocatUnixConsoleParams(),
1974 55cc0a44 Michael Hanselmann
             "UNIX-CONNECT:%s" % cls._InstanceSerial(instance.name)]
1975 55cc0a44 Michael Hanselmann
      return objects.InstanceConsole(instance=instance.name,
1976 55cc0a44 Michael Hanselmann
                                     kind=constants.CONS_SSH,
1977 55cc0a44 Michael Hanselmann
                                     host=instance.primary_node,
1978 052783ff Michael Hanselmann
                                     user=constants.SSH_CONSOLE_USER,
1979 55cc0a44 Michael Hanselmann
                                     command=cmd)
1980 3be34f57 Guido Trotter
1981 3be34f57 Guido Trotter
    vnc_bind_address = hvparams[constants.HV_VNC_BIND_ADDRESS]
1982 55cc0a44 Michael Hanselmann
    if vnc_bind_address and instance.network_port > constants.VNC_BASE_PORT:
1983 55cc0a44 Michael Hanselmann
      display = instance.network_port - constants.VNC_BASE_PORT
1984 55cc0a44 Michael Hanselmann
      return objects.InstanceConsole(instance=instance.name,
1985 55cc0a44 Michael Hanselmann
                                     kind=constants.CONS_VNC,
1986 55cc0a44 Michael Hanselmann
                                     host=vnc_bind_address,
1987 55cc0a44 Michael Hanselmann
                                     port=instance.network_port,
1988 55cc0a44 Michael Hanselmann
                                     display=display)
1989 55cc0a44 Michael Hanselmann
1990 4d2cdb5a Andrea Spadaccini
    spice_bind = hvparams[constants.HV_KVM_SPICE_BIND]
1991 4d2cdb5a Andrea Spadaccini
    if spice_bind:
1992 4d2cdb5a Andrea Spadaccini
      return objects.InstanceConsole(instance=instance.name,
1993 4d2cdb5a Andrea Spadaccini
                                     kind=constants.CONS_SPICE,
1994 4d2cdb5a Andrea Spadaccini
                                     host=spice_bind,
1995 4d2cdb5a Andrea Spadaccini
                                     port=instance.network_port)
1996 4d2cdb5a Andrea Spadaccini
1997 55cc0a44 Michael Hanselmann
    return objects.InstanceConsole(instance=instance.name,
1998 55cc0a44 Michael Hanselmann
                                   kind=constants.CONS_MESSAGE,
1999 55cc0a44 Michael Hanselmann
                                   message=("No serial shell for instance %s" %
2000 55cc0a44 Michael Hanselmann
                                            instance.name))
2001 eb58f9b1 Guido Trotter
2002 eb58f9b1 Guido Trotter
  def Verify(self):
2003 eb58f9b1 Guido Trotter
    """Verify the hypervisor.
2004 eb58f9b1 Guido Trotter

2005 cd04dfd2 Michael Hanselmann
    Check that the required binaries exist.
2006 cd04dfd2 Michael Hanselmann

2007 cd04dfd2 Michael Hanselmann
    @return: Problem description if something is wrong, C{None} otherwise
2008 eb58f9b1 Guido Trotter

2009 eb58f9b1 Guido Trotter
    """
2010 1f4b9d39 Iustin Pop
    msgs = []
2011 1f4b9d39 Iustin Pop
    # FIXME: this is the global kvm binary, but the actual path can be
2012 1f4b9d39 Iustin Pop
    # customized as an hv parameter; we should use the nodegroup's
2013 1f4b9d39 Iustin Pop
    # default kvm path parameter here.
2014 eb58f9b1 Guido Trotter
    if not os.path.exists(constants.KVM_PATH):
2015 1f4b9d39 Iustin Pop
      msgs.append("The KVM binary ('%s') does not exist" % constants.KVM_PATH)
2016 14aa53cb Guido Trotter
    if not os.path.exists(constants.SOCAT_PATH):
2017 1f4b9d39 Iustin Pop
      msgs.append("The socat binary ('%s') does not exist" %
2018 1f4b9d39 Iustin Pop
                  constants.SOCAT_PATH)
2019 1f4b9d39 Iustin Pop
2020 1f4b9d39 Iustin Pop
    return self._FormatVerifyResults(msgs)
2021 14aa53cb Guido Trotter
2022 6b5605e8 Iustin Pop
  @classmethod
2023 6b5605e8 Iustin Pop
  def CheckParameterSyntax(cls, hvparams):
2024 6b5605e8 Iustin Pop
    """Check the given parameters for validity.
2025 6b5605e8 Iustin Pop

2026 6b5605e8 Iustin Pop
    @type hvparams:  dict
2027 6b5605e8 Iustin Pop
    @param hvparams: dictionary with parameter names/value
2028 6b5605e8 Iustin Pop
    @raise errors.HypervisorError: when a parameter is not valid
2029 6b5605e8 Iustin Pop

2030 6b5605e8 Iustin Pop
    """
2031 47387b1e Guido Trotter
    super(KVMHypervisor, cls).CheckParameterSyntax(hvparams)
2032 6b5605e8 Iustin Pop
2033 df5ab9f0 Guido Trotter
    kernel_path = hvparams[constants.HV_KERNEL_PATH]
2034 df5ab9f0 Guido Trotter
    if kernel_path:
2035 df5ab9f0 Guido Trotter
      if not hvparams[constants.HV_ROOT_PATH]:
2036 205ab586 Iustin Pop
        raise errors.HypervisorError("Need a root partition for the instance,"
2037 205ab586 Iustin Pop
                                     " if a kernel is defined")
2038 6b5605e8 Iustin Pop
2039 205ab586 Iustin Pop
    if (hvparams[constants.HV_VNC_X509_VERIFY] and
2040 205ab586 Iustin Pop
        not hvparams[constants.HV_VNC_X509]):
2041 205ab586 Iustin Pop
      raise errors.HypervisorError("%s must be defined, if %s is" %
2042 205ab586 Iustin Pop
                                   (constants.HV_VNC_X509,
2043 205ab586 Iustin Pop
                                    constants.HV_VNC_X509_VERIFY))
2044 66d5dbef Guido Trotter
2045 14fd6c81 Guido Trotter
    if hvparams[constants.HV_SERIAL_CONSOLE]:
2046 14fd6c81 Guido Trotter
      serial_speed = hvparams[constants.HV_SERIAL_SPEED]
2047 14fd6c81 Guido Trotter
      valid_speeds = constants.VALID_SERIAL_SPEEDS
2048 14fd6c81 Guido Trotter
      if not serial_speed or serial_speed not in valid_speeds:
2049 14fd6c81 Guido Trotter
        raise errors.HypervisorError("Invalid serial console speed, must be"
2050 14fd6c81 Guido Trotter
                                     " one of: %s" %
2051 14fd6c81 Guido Trotter
                                     utils.CommaJoin(valid_speeds))
2052 14fd6c81 Guido Trotter
2053 66d5dbef Guido Trotter
    boot_order = hvparams[constants.HV_BOOT_ORDER]
2054 205ab586 Iustin Pop
    if (boot_order == constants.HT_BO_CDROM and
2055 205ab586 Iustin Pop
        not hvparams[constants.HV_CDROM_IMAGE_PATH]):
2056 835528af Iustin Pop
      raise errors.HypervisorError("Cannot boot from cdrom without an"
2057 835528af Iustin Pop
                                   " ISO path")
2058 f5118ade Iustin Pop
2059 d19d94db Guido Trotter
    security_model = hvparams[constants.HV_SECURITY_MODEL]
2060 d19d94db Guido Trotter
    if security_model == constants.HT_SM_USER:
2061 d19d94db Guido Trotter
      if not hvparams[constants.HV_SECURITY_DOMAIN]:
2062 d19d94db Guido Trotter
        raise errors.HypervisorError("A security domain (user to run kvm as)"
2063 d19d94db Guido Trotter
                                     " must be specified")
2064 d19d94db Guido Trotter
    elif (security_model == constants.HT_SM_NONE or
2065 d19d94db Guido Trotter
          security_model == constants.HT_SM_POOL):
2066 d19d94db Guido Trotter
      if hvparams[constants.HV_SECURITY_DOMAIN]:
2067 d19d94db Guido Trotter
        raise errors.HypervisorError("Cannot have a security domain when the"
2068 d19d94db Guido Trotter
                                     " security model is 'none' or 'pool'")
2069 d19d94db Guido Trotter
2070 b1cb62bd Andrea Spadaccini
    spice_bind = hvparams[constants.HV_KVM_SPICE_BIND]
2071 b451c4c7 Andrea Spadaccini
    spice_ip_version = hvparams[constants.HV_KVM_SPICE_IP_VERSION]
2072 b1cb62bd Andrea Spadaccini
    if spice_bind:
2073 b1cb62bd Andrea Spadaccini
      if spice_ip_version != constants.IFACE_NO_IP_VERSION_SPECIFIED:
2074 b1cb62bd Andrea Spadaccini
        # if an IP version is specified, the spice_bind parameter must be an
2075 b1cb62bd Andrea Spadaccini
        # IP of that family
2076 b1cb62bd Andrea Spadaccini
        if (netutils.IP4Address.IsValid(spice_bind) and
2077 b1cb62bd Andrea Spadaccini
            spice_ip_version != constants.IP4_VERSION):
2078 b1cb62bd Andrea Spadaccini
          raise errors.HypervisorError("spice: got an IPv4 address (%s), but"
2079 b1cb62bd Andrea Spadaccini
                                       " the specified IP version is %s" %
2080 b1cb62bd Andrea Spadaccini
                                       (spice_bind, spice_ip_version))
2081 b1cb62bd Andrea Spadaccini
2082 b1cb62bd Andrea Spadaccini
        if (netutils.IP6Address.IsValid(spice_bind) and
2083 b1cb62bd Andrea Spadaccini
            spice_ip_version != constants.IP6_VERSION):
2084 b1cb62bd Andrea Spadaccini
          raise errors.HypervisorError("spice: got an IPv6 address (%s), but"
2085 b1cb62bd Andrea Spadaccini
                                       " the specified IP version is %s" %
2086 b1cb62bd Andrea Spadaccini
                                       (spice_bind, spice_ip_version))
2087 b451c4c7 Andrea Spadaccini
    else:
2088 0e1b03b9 Andrea Spadaccini
      # All the other SPICE parameters depend on spice_bind being set. Raise an
2089 0e1b03b9 Andrea Spadaccini
      # error if any of them is set without it.
2090 07788a0b Michael Hanselmann
      for param in _SPICE_ADDITIONAL_PARAMS:
2091 0e1b03b9 Andrea Spadaccini
        if hvparams[param]:
2092 0e1b03b9 Andrea Spadaccini
          raise errors.HypervisorError("spice: %s requires %s to be set" %
2093 0e1b03b9 Andrea Spadaccini
                                       (param, constants.HV_KVM_SPICE_BIND))
2094 b1cb62bd Andrea Spadaccini
2095 d19d94db Guido Trotter
  @classmethod
2096 d19d94db Guido Trotter
  def ValidateParameters(cls, hvparams):
2097 d19d94db Guido Trotter
    """Check the given parameters for validity.
2098 d19d94db Guido Trotter

2099 d19d94db Guido Trotter
    @type hvparams:  dict
2100 d19d94db Guido Trotter
    @param hvparams: dictionary with parameter names/value
2101 d19d94db Guido Trotter
    @raise errors.HypervisorError: when a parameter is not valid
2102 d19d94db Guido Trotter

2103 d19d94db Guido Trotter
    """
2104 d19d94db Guido Trotter
    super(KVMHypervisor, cls).ValidateParameters(hvparams)
2105 d19d94db Guido Trotter
2106 82e3bf85 Dimitris Aragiorgis
    kvm_path = hvparams[constants.HV_KVM_PATH]
2107 82e3bf85 Dimitris Aragiorgis
2108 d19d94db Guido Trotter
    security_model = hvparams[constants.HV_SECURITY_MODEL]
2109 d19d94db Guido Trotter
    if security_model == constants.HT_SM_USER:
2110 d19d94db Guido Trotter
      username = hvparams[constants.HV_SECURITY_DOMAIN]
2111 d19d94db Guido Trotter
      try:
2112 1feb39ec Guido Trotter
        pwd.getpwnam(username)
2113 d19d94db Guido Trotter
      except KeyError:
2114 d19d94db Guido Trotter
        raise errors.HypervisorError("Unknown security domain user %s"
2115 d19d94db Guido Trotter
                                     % username)
2116 d19d94db Guido Trotter
2117 b1cb62bd Andrea Spadaccini
    spice_bind = hvparams[constants.HV_KVM_SPICE_BIND]
2118 b1cb62bd Andrea Spadaccini
    if spice_bind:
2119 b1cb62bd Andrea Spadaccini
      # only one of VNC and SPICE can be used currently.
2120 b1cb62bd Andrea Spadaccini
      if hvparams[constants.HV_VNC_BIND_ADDRESS]:
2121 b1cb62bd Andrea Spadaccini
        raise errors.HypervisorError("both SPICE and VNC are configured, but"
2122 b1cb62bd Andrea Spadaccini
                                     " only one of them can be used at a"
2123 b1cb62bd Andrea Spadaccini
                                     " given time.")
2124 b1cb62bd Andrea Spadaccini
2125 3b59ec02 Guido Trotter
      # check that KVM supports SPICE
2126 6e043e60 Guido Trotter
      kvmhelp = cls._GetKVMOutput(kvm_path, cls._KVMOPT_HELP)
2127 0ad7f5d8 Guido Trotter
      if not cls._SPICE_RE.search(kvmhelp):
2128 b1cb62bd Andrea Spadaccini
        raise errors.HypervisorError("spice is configured, but it is not"
2129 0ad7f5d8 Guido Trotter
                                     " supported according to kvm --help")
2130 b1cb62bd Andrea Spadaccini
2131 b1cb62bd Andrea Spadaccini
      # if spice_bind is not an IP address, it must be a valid interface
2132 b1cb62bd Andrea Spadaccini
      bound_to_addr = (netutils.IP4Address.IsValid(spice_bind)
2133 b1cb62bd Andrea Spadaccini
                       or netutils.IP6Address.IsValid(spice_bind))
2134 b1cb62bd Andrea Spadaccini
      if not bound_to_addr and not netutils.IsValidInterface(spice_bind):
2135 b1cb62bd Andrea Spadaccini
        raise errors.HypervisorError("spice: the %s parameter must be either"
2136 b1cb62bd Andrea Spadaccini
                                     " a valid IP address or interface name" %
2137 b1cb62bd Andrea Spadaccini
                                     constants.HV_KVM_SPICE_BIND)
2138 b1cb62bd Andrea Spadaccini
2139 82e3bf85 Dimitris Aragiorgis
    machine_version = hvparams[constants.HV_KVM_MACHINE_VERSION]
2140 82e3bf85 Dimitris Aragiorgis
    if machine_version:
2141 6e043e60 Guido Trotter
      output = cls._GetKVMOutput(kvm_path, cls._KVMOPT_MLIST)
2142 82e3bf85 Dimitris Aragiorgis
      if not cls._CHECK_MACHINE_VERSION_RE(machine_version).search(output):
2143 82e3bf85 Dimitris Aragiorgis
        raise errors.HypervisorError("Unsupported machine version: %s" %
2144 82e3bf85 Dimitris Aragiorgis
                                     machine_version)
2145 82e3bf85 Dimitris Aragiorgis
2146 f5118ade Iustin Pop
  @classmethod
2147 f5118ade Iustin Pop
  def PowercycleNode(cls):
2148 f5118ade Iustin Pop
    """KVM powercycle, just a wrapper over Linux powercycle.
2149 f5118ade Iustin Pop

2150 f5118ade Iustin Pop
    """
2151 f5118ade Iustin Pop
    cls.LinuxPowercycle()