Statistics
| Branch: | Tag: | Revision:

root / lib / hypervisor / hv_xen.py @ 3d942d8b

History | View | Annotate | Download (31.4 kB)

1
#
2
#
3

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

    
21

    
22
"""Xen hypervisors
23

24
"""
25

    
26
import logging
27
import string # pylint: disable=W0402
28
from cStringIO import StringIO
29

    
30
from ganeti import constants
31
from ganeti import errors
32
from ganeti import utils
33
from ganeti.hypervisor import hv_base
34
from ganeti import netutils
35
from ganeti import objects
36
from ganeti import pathutils
37
from ganeti import ssconf
38

    
39

    
40
XEND_CONFIG_FILE = utils.PathJoin(pathutils.XEN_CONFIG_DIR, "xend-config.sxp")
41
XL_CONFIG_FILE = utils.PathJoin(pathutils.XEN_CONFIG_DIR, "xen/xl.conf")
42
VIF_BRIDGE_SCRIPT = utils.PathJoin(pathutils.XEN_CONFIG_DIR,
43
                                   "scripts/vif-bridge")
44
_DOM0_NAME = "Domain-0"
45
_DISK_LETTERS = string.ascii_lowercase
46

    
47
_FILE_DRIVER_MAP = {
48
  constants.FD_LOOP: "file",
49
  constants.FD_BLKTAP: "tap:aio",
50
  }
51

    
52

    
53
def _CreateConfigCpus(cpu_mask):
54
  """Create a CPU config string for Xen's config file.
55

56
  """
57
  # Convert the string CPU mask to a list of list of int's
58
  cpu_list = utils.ParseMultiCpuMask(cpu_mask)
59

    
60
  if len(cpu_list) == 1:
61
    all_cpu_mapping = cpu_list[0]
62
    if all_cpu_mapping == constants.CPU_PINNING_OFF:
63
      # If CPU pinning has 1 entry that's "all", then remove the
64
      # parameter from the config file
65
      return None
66
    else:
67
      # If CPU pinning has one non-all entry, mapping all vCPUS (the entire
68
      # VM) to one physical CPU, using format 'cpu = "C"'
69
      return "cpu = \"%s\"" % ",".join(map(str, all_cpu_mapping))
70
  else:
71

    
72
    def _GetCPUMap(vcpu):
73
      if vcpu[0] == constants.CPU_PINNING_ALL_VAL:
74
        cpu_map = constants.CPU_PINNING_ALL_XEN
75
      else:
76
        cpu_map = ",".join(map(str, vcpu))
77
      return "\"%s\"" % cpu_map
78

    
79
    # build the result string in format 'cpus = [ "c", "c", "c" ]',
80
    # where each c is a physical CPU number, a range, a list, or any
81
    # combination
82
    return "cpus = [ %s ]" % ", ".join(map(_GetCPUMap, cpu_list))
83

    
84

    
85
def _RunXmList(fn, xmllist_errors):
86
  """Helper function for L{_GetXmList} to run "xm list".
87

88
  @type fn: callable
89
  @param fn: Function returning result of running C{xm list}
90
  @type xmllist_errors: list
91
  @param xmllist_errors: Error list
92
  @rtype: list
93

94
  """
95
  result = fn()
96
  if result.failed:
97
    logging.error("xm list failed (%s): %s", result.fail_reason,
98
                  result.output)
99
    xmllist_errors.append(result)
100
    raise utils.RetryAgain()
101

    
102
  # skip over the heading
103
  return result.stdout.splitlines()
104

    
105

    
106
def _ParseXmList(lines, include_node):
107
  """Parses the output of C{xm list}.
108

109
  @type lines: list
110
  @param lines: Output lines of C{xm list}
111
  @type include_node: boolean
112
  @param include_node: If True, return information for Dom0
113
  @return: list of tuple containing (name, id, memory, vcpus, state, time
114
    spent)
115

116
  """
117
  result = []
118

    
119
  # Iterate through all lines while ignoring header
120
  for line in lines[1:]:
121
    # The format of lines is:
122
    # Name      ID Mem(MiB) VCPUs State  Time(s)
123
    # Domain-0   0  3418     4 r-----    266.2
124
    data = line.split()
125
    if len(data) != 6:
126
      raise errors.HypervisorError("Can't parse output of xm list,"
127
                                   " line: %s" % line)
128
    try:
129
      data[1] = int(data[1])
130
      data[2] = int(data[2])
131
      data[3] = int(data[3])
132
      data[5] = float(data[5])
133
    except (TypeError, ValueError), err:
134
      raise errors.HypervisorError("Can't parse output of xm list,"
135
                                   " line: %s, error: %s" % (line, err))
136

    
137
    # skip the Domain-0 (optional)
138
    if include_node or data[0] != _DOM0_NAME:
139
      result.append(data)
140

    
141
  return result
142

    
143

    
144
def _GetXmList(fn, include_node, _timeout=5):
145
  """Return the list of running instances.
146

147
  See L{_RunXmList} and L{_ParseXmList} for parameter details.
148

149
  """
150
  xmllist_errors = []
151
  try:
152
    lines = utils.Retry(_RunXmList, (0.3, 1.5, 1.0), _timeout,
153
                        args=(fn, xmllist_errors))
154
  except utils.RetryTimeout:
155
    if xmllist_errors:
156
      xmlist_result = xmllist_errors.pop()
157

    
158
      errmsg = ("xm list failed, timeout exceeded (%s): %s" %
159
                (xmlist_result.fail_reason, xmlist_result.output))
160
    else:
161
      errmsg = "xm list failed"
162

    
163
    raise errors.HypervisorError(errmsg)
164

    
165
  return _ParseXmList(lines, include_node)
166

    
167

    
168
def _ParseNodeInfo(info):
169
  """Return information about the node.
170

171
  @return: a dict with the following keys (memory values in MiB):
172
        - memory_total: the total memory size on the node
173
        - memory_free: the available memory on the node for instances
174
        - nr_cpus: total number of CPUs
175
        - nr_nodes: in a NUMA system, the number of domains
176
        - nr_sockets: the number of physical CPU sockets in the node
177
        - hv_version: the hypervisor version in the form (major, minor)
178

179
  """
180
  result = {}
181
  cores_per_socket = threads_per_core = nr_cpus = None
182
  xen_major, xen_minor = None, None
183
  memory_total = None
184
  memory_free = None
185

    
186
  for line in info.splitlines():
187
    fields = line.split(":", 1)
188

    
189
    if len(fields) < 2:
190
      continue
191

    
192
    (key, val) = map(lambda s: s.strip(), fields)
193

    
194
    # Note: in Xen 3, memory has changed to total_memory
195
    if key in ("memory", "total_memory"):
196
      memory_total = int(val)
197
    elif key == "free_memory":
198
      memory_free = int(val)
199
    elif key == "nr_cpus":
200
      nr_cpus = result["cpu_total"] = int(val)
201
    elif key == "nr_nodes":
202
      result["cpu_nodes"] = int(val)
203
    elif key == "cores_per_socket":
204
      cores_per_socket = int(val)
205
    elif key == "threads_per_core":
206
      threads_per_core = int(val)
207
    elif key == "xen_major":
208
      xen_major = int(val)
209
    elif key == "xen_minor":
210
      xen_minor = int(val)
211

    
212
  if None not in [cores_per_socket, threads_per_core, nr_cpus]:
213
    result["cpu_sockets"] = nr_cpus / (cores_per_socket * threads_per_core)
214

    
215
  if memory_free is not None:
216
    result["memory_free"] = memory_free
217

    
218
  if memory_total is not None:
219
    result["memory_total"] = memory_total
220

    
221
  if not (xen_major is None or xen_minor is None):
222
    result[constants.HV_NODEINFO_KEY_VERSION] = (xen_major, xen_minor)
223

    
224
  return result
225

    
226

    
227
def _MergeInstanceInfo(info, fn):
228
  """Updates node information from L{_ParseNodeInfo} with instance info.
229

230
  @type info: dict
231
  @param info: Result from L{_ParseNodeInfo}
232
  @type fn: callable
233
  @param fn: Function returning result of running C{xm list}
234
  @rtype: dict
235

236
  """
237
  total_instmem = 0
238

    
239
  for (name, _, mem, vcpus, _, _) in fn(True):
240
    if name == _DOM0_NAME:
241
      info["memory_dom0"] = mem
242
      info["dom0_cpus"] = vcpus
243

    
244
    # Include Dom0 in total memory usage
245
    total_instmem += mem
246

    
247
  memory_free = info.get("memory_free")
248
  memory_total = info.get("memory_total")
249

    
250
  # Calculate memory used by hypervisor
251
  if None not in [memory_total, memory_free, total_instmem]:
252
    info["memory_hv"] = memory_total - memory_free - total_instmem
253

    
254
  return info
255

    
256

    
257
def _GetNodeInfo(info, fn):
258
  """Combines L{_MergeInstanceInfo} and L{_ParseNodeInfo}.
259

260
  """
261
  return _MergeInstanceInfo(_ParseNodeInfo(info), fn)
262

    
263

    
264
def _GetConfigFileDiskData(block_devices, blockdev_prefix,
265
                           _letters=_DISK_LETTERS):
266
  """Get disk directives for Xen config file.
267

268
  This method builds the xen config disk directive according to the
269
  given disk_template and block_devices.
270

271
  @param block_devices: list of tuples (cfdev, rldev):
272
      - cfdev: dict containing ganeti config disk part
273
      - rldev: ganeti.bdev.BlockDev object
274
  @param blockdev_prefix: a string containing blockdevice prefix,
275
                          e.g. "sd" for /dev/sda
276

277
  @return: string containing disk directive for xen instance config file
278

279
  """
280
  if len(block_devices) > len(_letters):
281
    raise errors.HypervisorError("Too many disks")
282

    
283
  disk_data = []
284

    
285
  for sd_suffix, (cfdev, dev_path) in zip(_letters, block_devices):
286
    sd_name = blockdev_prefix + sd_suffix
287

    
288
    if cfdev.mode == constants.DISK_RDWR:
289
      mode = "w"
290
    else:
291
      mode = "r"
292

    
293
    if cfdev.dev_type == constants.LD_FILE:
294
      driver = _FILE_DRIVER_MAP[cfdev.physical_id[0]]
295
    else:
296
      driver = "phy"
297

    
298
    disk_data.append("'%s:%s,%s,%s'" % (driver, dev_path, sd_name, mode))
299

    
300
  return disk_data
301

    
302

    
303
class XenHypervisor(hv_base.BaseHypervisor):
304
  """Xen generic hypervisor interface
305

306
  This is the Xen base class used for both Xen PVM and HVM. It contains
307
  all the functionality that is identical for both.
308

309
  """
310
  CAN_MIGRATE = True
311
  REBOOT_RETRY_COUNT = 60
312
  REBOOT_RETRY_INTERVAL = 10
313

    
314
  ANCILLARY_FILES = [
315
    XEND_CONFIG_FILE,
316
    XL_CONFIG_FILE,
317
    VIF_BRIDGE_SCRIPT,
318
    ]
319
  ANCILLARY_FILES_OPT = [
320
    XL_CONFIG_FILE,
321
    ]
322

    
323
  def __init__(self, _cfgdir=None, _run_cmd_fn=None, _cmd=None):
324
    hv_base.BaseHypervisor.__init__(self)
325

    
326
    if _cfgdir is None:
327
      self._cfgdir = pathutils.XEN_CONFIG_DIR
328
    else:
329
      self._cfgdir = _cfgdir
330

    
331
    if _run_cmd_fn is None:
332
      self._run_cmd_fn = utils.RunCmd
333
    else:
334
      self._run_cmd_fn = _run_cmd_fn
335

    
336
    self._cmd = _cmd
337

    
338
  def _GetCommand(self):
339
    if self._cmd is None:
340
      # TODO: Make command a hypervisor parameter
341
      cmd = constants.XEN_CMD
342
    else:
343
      cmd = self._cmd
344

    
345
    if cmd not in constants.KNOWN_XEN_COMMANDS:
346
      raise errors.ProgrammerError("Unknown Xen command '%s'" % cmd)
347

    
348
    return cmd
349

    
350
  def _RunXen(self, args):
351
    """Wrapper around L{utils.RunCmd} to run Xen command.
352

353
    @see: L{utils.RunCmd}
354

355
    """
356
    cmd = [self._GetCommand()]
357
    cmd.extend(args)
358

    
359
    return self._run_cmd_fn(cmd)
360

    
361
  def _ConfigFileName(self, instance_name):
362
    """Get the config file name for an instance.
363

364
    @param instance_name: instance name
365
    @type instance_name: str
366
    @return: fully qualified path to instance config file
367
    @rtype: str
368

369
    """
370
    return utils.PathJoin(self._cfgdir, instance_name)
371

    
372
  @classmethod
373
  def _GetConfig(cls, instance, startup_memory, block_devices):
374
    """Build Xen configuration for an instance.
375

376
    """
377
    raise NotImplementedError
378

    
379
  def _WriteConfigFile(self, instance_name, data):
380
    """Write the Xen config file for the instance.
381

382
    This version of the function just writes the config file from static data.
383

384
    """
385
    # just in case it exists
386
    utils.RemoveFile(utils.PathJoin(self._cfgdir, "auto", instance_name))
387

    
388
    cfg_file = self._ConfigFileName(instance_name)
389
    try:
390
      utils.WriteFile(cfg_file, data=data)
391
    except EnvironmentError, err:
392
      raise errors.HypervisorError("Cannot write Xen instance configuration"
393
                                   " file %s: %s" % (cfg_file, err))
394

    
395
  def _ReadConfigFile(self, instance_name):
396
    """Returns the contents of the instance config file.
397

398
    """
399
    filename = self._ConfigFileName(instance_name)
400

    
401
    try:
402
      file_content = utils.ReadFile(filename)
403
    except EnvironmentError, err:
404
      raise errors.HypervisorError("Failed to load Xen config file: %s" % err)
405

    
406
    return file_content
407

    
408
  def _RemoveConfigFile(self, instance_name):
409
    """Remove the xen configuration file.
410

411
    """
412
    utils.RemoveFile(self._ConfigFileName(instance_name))
413

    
414
  def _GetXmList(self, include_node):
415
    """Wrapper around module level L{_GetXmList}.
416

417
    """
418
    return _GetXmList(lambda: self._RunXen(["list"]), include_node)
419

    
420
  def ListInstances(self):
421
    """Get the list of running instances.
422

423
    """
424
    xm_list = self._GetXmList(False)
425
    names = [info[0] for info in xm_list]
426
    return names
427

    
428
  def GetInstanceInfo(self, instance_name):
429
    """Get instance properties.
430

431
    @param instance_name: the instance name
432

433
    @return: tuple (name, id, memory, vcpus, stat, times)
434

435
    """
436
    xm_list = self._GetXmList(instance_name == _DOM0_NAME)
437
    result = None
438
    for data in xm_list:
439
      if data[0] == instance_name:
440
        result = data
441
        break
442
    return result
443

    
444
  def GetAllInstancesInfo(self):
445
    """Get properties of all instances.
446

447
    @return: list of tuples (name, id, memory, vcpus, stat, times)
448

449
    """
450
    xm_list = self._GetXmList(False)
451
    return xm_list
452

    
453
  def _MakeConfigFile(self, instance, startup_memory, block_devices):
454
    """Gather configuration details and write to disk.
455

456
    See L{_GetConfig} for arguments.
457

458
    """
459
    buf = StringIO()
460
    buf.write("# Automatically generated by Ganeti. Do not edit!\n")
461
    buf.write("\n")
462
    buf.write(self._GetConfig(instance, startup_memory, block_devices))
463
    buf.write("\n")
464

    
465
    self._WriteConfigFile(instance.name, buf.getvalue())
466

    
467
  def StartInstance(self, instance, block_devices, startup_paused):
468
    """Start an instance.
469

470
    """
471
    startup_memory = self._InstanceStartupMemory(instance)
472

    
473
    self._MakeConfigFile(instance, startup_memory, block_devices)
474

    
475
    cmd = ["create"]
476
    if startup_paused:
477
      cmd.append("-p")
478
    cmd.append(self._ConfigFileName(instance.name))
479

    
480
    result = self._RunXen(cmd)
481
    if result.failed:
482
      raise errors.HypervisorError("Failed to start instance %s: %s (%s)" %
483
                                   (instance.name, result.fail_reason,
484
                                    result.output))
485

    
486
  def StopInstance(self, instance, force=False, retry=False, name=None):
487
    """Stop an instance.
488

489
    """
490
    if name is None:
491
      name = instance.name
492

    
493
    if force:
494
      action = "destroy"
495
    else:
496
      action = "shutdown"
497

    
498
    result = self._RunXen([action, name])
499
    if result.failed:
500
      raise errors.HypervisorError("Failed to stop instance %s: %s, %s" %
501
                                   (name, result.fail_reason, result.output))
502

    
503
    # Remove configuration file if stopping/starting instance was successful
504
    self._RemoveConfigFile(name)
505

    
506
  def RebootInstance(self, instance):
507
    """Reboot an instance.
508

509
    """
510
    ini_info = self.GetInstanceInfo(instance.name)
511

    
512
    if ini_info is None:
513
      raise errors.HypervisorError("Failed to reboot instance %s,"
514
                                   " not running" % instance.name)
515

    
516
    result = self._RunXen(["reboot", instance.name])
517
    if result.failed:
518
      raise errors.HypervisorError("Failed to reboot instance %s: %s, %s" %
519
                                   (instance.name, result.fail_reason,
520
                                    result.output))
521

    
522
    def _CheckInstance():
523
      new_info = self.GetInstanceInfo(instance.name)
524

    
525
      # check if the domain ID has changed or the run time has decreased
526
      if (new_info is not None and
527
          (new_info[1] != ini_info[1] or new_info[5] < ini_info[5])):
528
        return
529

    
530
      raise utils.RetryAgain()
531

    
532
    try:
533
      utils.Retry(_CheckInstance, self.REBOOT_RETRY_INTERVAL,
534
                  self.REBOOT_RETRY_INTERVAL * self.REBOOT_RETRY_COUNT)
535
    except utils.RetryTimeout:
536
      raise errors.HypervisorError("Failed to reboot instance %s: instance"
537
                                   " did not reboot in the expected interval" %
538
                                   (instance.name, ))
539

    
540
  def BalloonInstanceMemory(self, instance, mem):
541
    """Balloon an instance memory to a certain value.
542

543
    @type instance: L{objects.Instance}
544
    @param instance: instance to be accepted
545
    @type mem: int
546
    @param mem: actual memory size to use for instance runtime
547

548
    """
549
    result = self._RunXen(["mem-set", instance.name, mem])
550
    if result.failed:
551
      raise errors.HypervisorError("Failed to balloon instance %s: %s (%s)" %
552
                                   (instance.name, result.fail_reason,
553
                                    result.output))
554

    
555
    # Update configuration file
556
    cmd = ["sed", "-ie", "s/^memory.*$/memory = %s/" % mem]
557
    cmd.append(self._ConfigFileName(instance.name))
558

    
559
    result = utils.RunCmd(cmd)
560
    if result.failed:
561
      raise errors.HypervisorError("Failed to update memory for %s: %s (%s)" %
562
                                   (instance.name, result.fail_reason,
563
                                    result.output))
564

    
565
  def GetNodeInfo(self):
566
    """Return information about the node.
567

568
    @see: L{_GetNodeInfo} and L{_ParseNodeInfo}
569

570
    """
571
    result = self._RunXen(["info"])
572
    if result.failed:
573
      logging.error("Can't run 'xm info' (%s): %s", result.fail_reason,
574
                    result.output)
575
      return None
576

    
577
    return _GetNodeInfo(result.stdout, self._GetXmList)
578

    
579
  @classmethod
580
  def GetInstanceConsole(cls, instance, hvparams, beparams):
581
    """Return a command for connecting to the console of an instance.
582

583
    """
584
    return objects.InstanceConsole(instance=instance.name,
585
                                   kind=constants.CONS_SSH,
586
                                   host=instance.primary_node,
587
                                   user=constants.SSH_CONSOLE_USER,
588
                                   command=[pathutils.XEN_CONSOLE_WRAPPER,
589
                                            constants.XEN_CMD, instance.name])
590

    
591
  def Verify(self):
592
    """Verify the hypervisor.
593

594
    For Xen, this verifies that the xend process is running.
595

596
    @return: Problem description if something is wrong, C{None} otherwise
597

598
    """
599
    result = self._RunXen(["info"])
600
    if result.failed:
601
      return "'xm info' failed: %s, %s" % (result.fail_reason, result.output)
602

    
603
    return None
604

    
605
  def MigrationInfo(self, instance):
606
    """Get instance information to perform a migration.
607

608
    @type instance: L{objects.Instance}
609
    @param instance: instance to be migrated
610
    @rtype: string
611
    @return: content of the xen config file
612

613
    """
614
    return self._ReadConfigFile(instance.name)
615

    
616
  def AcceptInstance(self, instance, info, target):
617
    """Prepare to accept an instance.
618

619
    @type instance: L{objects.Instance}
620
    @param instance: instance to be accepted
621
    @type info: string
622
    @param info: content of the xen config file on the source node
623
    @type target: string
624
    @param target: target host (usually ip), on this node
625

626
    """
627
    pass
628

    
629
  def FinalizeMigrationDst(self, instance, info, success):
630
    """Finalize an instance migration.
631

632
    After a successful migration we write the xen config file.
633
    We do nothing on a failure, as we did not change anything at accept time.
634

635
    @type instance: L{objects.Instance}
636
    @param instance: instance whose migration is being finalized
637
    @type info: string
638
    @param info: content of the xen config file on the source node
639
    @type success: boolean
640
    @param success: whether the migration was a success or a failure
641

642
    """
643
    if success:
644
      self._WriteConfigFile(instance.name, info)
645

    
646
  def MigrateInstance(self, instance, target, live):
647
    """Migrate an instance to a target node.
648

649
    The migration will not be attempted if the instance is not
650
    currently running.
651

652
    @type instance: L{objects.Instance}
653
    @param instance: the instance to be migrated
654
    @type target: string
655
    @param target: ip address of the target node
656
    @type live: boolean
657
    @param live: perform a live migration
658

659
    """
660
    if self.GetInstanceInfo(instance.name) is None:
661
      raise errors.HypervisorError("Instance not running, cannot migrate")
662

    
663
    port = instance.hvparams[constants.HV_MIGRATION_PORT]
664

    
665
    if (self._cmd == constants.XEN_CMD_XM and
666
        not netutils.TcpPing(target, port, live_port_needed=True)):
667
      raise errors.HypervisorError("Remote host %s not listening on port"
668
                                   " %s, cannot migrate" % (target, port))
669

    
670
    args = ["migrate"]
671

    
672
    if self._cmd == constants.XEN_CMD_XM:
673
      args.extend(["-p", "%d" % port])
674
      if live:
675
        args.append("-l")
676

    
677
    elif self._cmd == constants.XEN_CMD_XL:
678
      cluster_name = ssconf.SimpleStore().GetClusterName()
679
      args.extend(["-s", constants.XL_SSH_CMD % cluster_name])
680
      args.extend(["-C", self._ConfigFileName(instance.name)])
681

    
682
    else:
683
      raise errors.HypervisorError("Unsupported xen command: %s" % self._cmd)
684

    
685
    args.extend([instance.name, target])
686

    
687
    result = self._RunXen(args)
688
    if result.failed:
689
      raise errors.HypervisorError("Failed to migrate instance %s: %s" %
690
                                   (instance.name, result.output))
691

    
692
  def FinalizeMigrationSource(self, instance, success, live):
693
    """Finalize the instance migration on the source node.
694

695
    @type instance: L{objects.Instance}
696
    @param instance: the instance that was migrated
697
    @type success: bool
698
    @param success: whether the migration succeeded or not
699
    @type live: bool
700
    @param live: whether the user requested a live migration or not
701

702
    """
703
    # pylint: disable=W0613
704
    if success:
705
      # remove old xen file after migration succeeded
706
      try:
707
        self._RemoveConfigFile(instance.name)
708
      except EnvironmentError:
709
        logging.exception("Failure while removing instance config file")
710

    
711
  def GetMigrationStatus(self, instance):
712
    """Get the migration status
713

714
    As MigrateInstance for Xen is still blocking, if this method is called it
715
    means that MigrateInstance has completed successfully. So we can safely
716
    assume that the migration was successful and notify this fact to the client.
717

718
    @type instance: L{objects.Instance}
719
    @param instance: the instance that is being migrated
720
    @rtype: L{objects.MigrationStatus}
721
    @return: the status of the current migration (one of
722
             L{constants.HV_MIGRATION_VALID_STATUSES}), plus any additional
723
             progress info that can be retrieved from the hypervisor
724

725
    """
726
    return objects.MigrationStatus(status=constants.HV_MIGRATION_COMPLETED)
727

    
728
  @classmethod
729
  def PowercycleNode(cls):
730
    """Xen-specific powercycle.
731

732
    This first does a Linux reboot (which triggers automatically a Xen
733
    reboot), and if that fails it tries to do a Xen reboot. The reason
734
    we don't try a Xen reboot first is that the xen reboot launches an
735
    external command which connects to the Xen hypervisor, and that
736
    won't work in case the root filesystem is broken and/or the xend
737
    daemon is not working.
738

739
    """
740
    try:
741
      cls.LinuxPowercycle()
742
    finally:
743
      utils.RunCmd([constants.XEN_CMD, "debug", "R"])
744

    
745

    
746
class XenPvmHypervisor(XenHypervisor):
747
  """Xen PVM hypervisor interface"""
748

    
749
  PARAMETERS = {
750
    constants.HV_USE_BOOTLOADER: hv_base.NO_CHECK,
751
    constants.HV_BOOTLOADER_PATH: hv_base.OPT_FILE_CHECK,
752
    constants.HV_BOOTLOADER_ARGS: hv_base.NO_CHECK,
753
    constants.HV_KERNEL_PATH: hv_base.REQ_FILE_CHECK,
754
    constants.HV_INITRD_PATH: hv_base.OPT_FILE_CHECK,
755
    constants.HV_ROOT_PATH: hv_base.NO_CHECK,
756
    constants.HV_KERNEL_ARGS: hv_base.NO_CHECK,
757
    constants.HV_MIGRATION_PORT: hv_base.REQ_NET_PORT_CHECK,
758
    constants.HV_MIGRATION_MODE: hv_base.MIGRATION_MODE_CHECK,
759
    # TODO: Add a check for the blockdev prefix (matching [a-z:] or similar).
760
    constants.HV_BLOCKDEV_PREFIX: hv_base.NO_CHECK,
761
    constants.HV_REBOOT_BEHAVIOR:
762
      hv_base.ParamInSet(True, constants.REBOOT_BEHAVIORS),
763
    constants.HV_CPU_MASK: hv_base.OPT_MULTI_CPU_MASK_CHECK,
764
    constants.HV_CPU_CAP: hv_base.OPT_NONNEGATIVE_INT_CHECK,
765
    constants.HV_CPU_WEIGHT:
766
      (False, lambda x: 0 < x < 65536, "invalid weight", None, None),
767
    }
768

    
769
  def _GetConfig(self, instance, startup_memory, block_devices):
770
    """Write the Xen config file for the instance.
771

772
    """
773
    hvp = instance.hvparams
774
    config = StringIO()
775
    config.write("# this is autogenerated by Ganeti, please do not edit\n#\n")
776

    
777
    # if bootloader is True, use bootloader instead of kernel and ramdisk
778
    # parameters.
779
    if hvp[constants.HV_USE_BOOTLOADER]:
780
      # bootloader handling
781
      bootloader_path = hvp[constants.HV_BOOTLOADER_PATH]
782
      if bootloader_path:
783
        config.write("bootloader = '%s'\n" % bootloader_path)
784
      else:
785
        raise errors.HypervisorError("Bootloader enabled, but missing"
786
                                     " bootloader path")
787

    
788
      bootloader_args = hvp[constants.HV_BOOTLOADER_ARGS]
789
      if bootloader_args:
790
        config.write("bootargs = '%s'\n" % bootloader_args)
791
    else:
792
      # kernel handling
793
      kpath = hvp[constants.HV_KERNEL_PATH]
794
      config.write("kernel = '%s'\n" % kpath)
795

    
796
      # initrd handling
797
      initrd_path = hvp[constants.HV_INITRD_PATH]
798
      if initrd_path:
799
        config.write("ramdisk = '%s'\n" % initrd_path)
800

    
801
    # rest of the settings
802
    config.write("memory = %d\n" % startup_memory)
803
    config.write("maxmem = %d\n" % instance.beparams[constants.BE_MAXMEM])
804
    config.write("vcpus = %d\n" % instance.beparams[constants.BE_VCPUS])
805
    cpu_pinning = _CreateConfigCpus(hvp[constants.HV_CPU_MASK])
806
    if cpu_pinning:
807
      config.write("%s\n" % cpu_pinning)
808
    cpu_cap = hvp[constants.HV_CPU_CAP]
809
    if cpu_cap:
810
      config.write("cpu_cap=%d\n" % cpu_cap)
811
    cpu_weight = hvp[constants.HV_CPU_WEIGHT]
812
    if cpu_weight:
813
      config.write("cpu_weight=%d\n" % cpu_weight)
814

    
815
    config.write("name = '%s'\n" % instance.name)
816

    
817
    vif_data = []
818
    for nic in instance.nics:
819
      nic_str = "mac=%s" % (nic.mac)
820
      ip = getattr(nic, "ip", None)
821
      if ip is not None:
822
        nic_str += ", ip=%s" % ip
823
      if nic.nicparams[constants.NIC_MODE] == constants.NIC_MODE_BRIDGED:
824
        nic_str += ", bridge=%s" % nic.nicparams[constants.NIC_LINK]
825
      vif_data.append("'%s'" % nic_str)
826

    
827
    disk_data = \
828
      _GetConfigFileDiskData(block_devices, hvp[constants.HV_BLOCKDEV_PREFIX])
829

    
830
    config.write("vif = [%s]\n" % ",".join(vif_data))
831
    config.write("disk = [%s]\n" % ",".join(disk_data))
832

    
833
    if hvp[constants.HV_ROOT_PATH]:
834
      config.write("root = '%s'\n" % hvp[constants.HV_ROOT_PATH])
835
    config.write("on_poweroff = 'destroy'\n")
836
    if hvp[constants.HV_REBOOT_BEHAVIOR] == constants.INSTANCE_REBOOT_ALLOWED:
837
      config.write("on_reboot = 'restart'\n")
838
    else:
839
      config.write("on_reboot = 'destroy'\n")
840
    config.write("on_crash = 'restart'\n")
841
    config.write("extra = '%s'\n" % hvp[constants.HV_KERNEL_ARGS])
842

    
843
    return config.getvalue()
844

    
845

    
846
class XenHvmHypervisor(XenHypervisor):
847
  """Xen HVM hypervisor interface"""
848

    
849
  ANCILLARY_FILES = XenHypervisor.ANCILLARY_FILES + [
850
    pathutils.VNC_PASSWORD_FILE,
851
    ]
852
  ANCILLARY_FILES_OPT = XenHypervisor.ANCILLARY_FILES_OPT + [
853
    pathutils.VNC_PASSWORD_FILE,
854
    ]
855

    
856
  PARAMETERS = {
857
    constants.HV_ACPI: hv_base.NO_CHECK,
858
    constants.HV_BOOT_ORDER: (True, ) +
859
      (lambda x: x and len(x.strip("acdn")) == 0,
860
       "Invalid boot order specified, must be one or more of [acdn]",
861
       None, None),
862
    constants.HV_CDROM_IMAGE_PATH: hv_base.OPT_FILE_CHECK,
863
    constants.HV_DISK_TYPE:
864
      hv_base.ParamInSet(True, constants.HT_HVM_VALID_DISK_TYPES),
865
    constants.HV_NIC_TYPE:
866
      hv_base.ParamInSet(True, constants.HT_HVM_VALID_NIC_TYPES),
867
    constants.HV_PAE: hv_base.NO_CHECK,
868
    constants.HV_VNC_BIND_ADDRESS:
869
      (False, netutils.IP4Address.IsValid,
870
       "VNC bind address is not a valid IP address", None, None),
871
    constants.HV_KERNEL_PATH: hv_base.REQ_FILE_CHECK,
872
    constants.HV_DEVICE_MODEL: hv_base.REQ_FILE_CHECK,
873
    constants.HV_VNC_PASSWORD_FILE: hv_base.REQ_FILE_CHECK,
874
    constants.HV_MIGRATION_PORT: hv_base.REQ_NET_PORT_CHECK,
875
    constants.HV_MIGRATION_MODE: hv_base.MIGRATION_MODE_CHECK,
876
    constants.HV_USE_LOCALTIME: hv_base.NO_CHECK,
877
    # TODO: Add a check for the blockdev prefix (matching [a-z:] or similar).
878
    constants.HV_BLOCKDEV_PREFIX: hv_base.NO_CHECK,
879
    # Add PCI passthrough
880
    constants.HV_PASSTHROUGH: hv_base.NO_CHECK,
881
    constants.HV_REBOOT_BEHAVIOR:
882
      hv_base.ParamInSet(True, constants.REBOOT_BEHAVIORS),
883
    constants.HV_CPU_MASK: hv_base.OPT_MULTI_CPU_MASK_CHECK,
884
    constants.HV_CPU_CAP: hv_base.NO_CHECK,
885
    constants.HV_CPU_WEIGHT:
886
      (False, lambda x: 0 < x < 65535, "invalid weight", None, None),
887
    }
888

    
889
  def _GetConfig(self, instance, startup_memory, block_devices):
890
    """Create a Xen 3.1 HVM config file.
891

892
    """
893
    hvp = instance.hvparams
894

    
895
    config = StringIO()
896

    
897
    # kernel handling
898
    kpath = hvp[constants.HV_KERNEL_PATH]
899
    config.write("kernel = '%s'\n" % kpath)
900

    
901
    config.write("builder = 'hvm'\n")
902
    config.write("memory = %d\n" % startup_memory)
903
    config.write("maxmem = %d\n" % instance.beparams[constants.BE_MAXMEM])
904
    config.write("vcpus = %d\n" % instance.beparams[constants.BE_VCPUS])
905
    cpu_pinning = _CreateConfigCpus(hvp[constants.HV_CPU_MASK])
906
    if cpu_pinning:
907
      config.write("%s\n" % cpu_pinning)
908
    cpu_cap = hvp[constants.HV_CPU_CAP]
909
    if cpu_cap:
910
      config.write("cpu_cap=%d\n" % cpu_cap)
911
    cpu_weight = hvp[constants.HV_CPU_WEIGHT]
912
    if cpu_weight:
913
      config.write("cpu_weight=%d\n" % cpu_weight)
914

    
915
    config.write("name = '%s'\n" % instance.name)
916
    if hvp[constants.HV_PAE]:
917
      config.write("pae = 1\n")
918
    else:
919
      config.write("pae = 0\n")
920
    if hvp[constants.HV_ACPI]:
921
      config.write("acpi = 1\n")
922
    else:
923
      config.write("acpi = 0\n")
924
    config.write("apic = 1\n")
925
    config.write("device_model = '%s'\n" % hvp[constants.HV_DEVICE_MODEL])
926
    config.write("boot = '%s'\n" % hvp[constants.HV_BOOT_ORDER])
927
    config.write("sdl = 0\n")
928
    config.write("usb = 1\n")
929
    config.write("usbdevice = 'tablet'\n")
930
    config.write("vnc = 1\n")
931
    if hvp[constants.HV_VNC_BIND_ADDRESS] is None:
932
      config.write("vnclisten = '%s'\n" % constants.VNC_DEFAULT_BIND_ADDRESS)
933
    else:
934
      config.write("vnclisten = '%s'\n" % hvp[constants.HV_VNC_BIND_ADDRESS])
935

    
936
    if instance.network_port > constants.VNC_BASE_PORT:
937
      display = instance.network_port - constants.VNC_BASE_PORT
938
      config.write("vncdisplay = %s\n" % display)
939
      config.write("vncunused = 0\n")
940
    else:
941
      config.write("# vncdisplay = 1\n")
942
      config.write("vncunused = 1\n")
943

    
944
    vnc_pwd_file = hvp[constants.HV_VNC_PASSWORD_FILE]
945
    try:
946
      password = utils.ReadFile(vnc_pwd_file)
947
    except EnvironmentError, err:
948
      raise errors.HypervisorError("Failed to open VNC password file %s: %s" %
949
                                   (vnc_pwd_file, err))
950

    
951
    config.write("vncpasswd = '%s'\n" % password.rstrip())
952

    
953
    config.write("serial = 'pty'\n")
954
    if hvp[constants.HV_USE_LOCALTIME]:
955
      config.write("localtime = 1\n")
956

    
957
    vif_data = []
958
    nic_type = hvp[constants.HV_NIC_TYPE]
959
    if nic_type is None:
960
      # ensure old instances don't change
961
      nic_type_str = ", type=ioemu"
962
    elif nic_type == constants.HT_NIC_PARAVIRTUAL:
963
      nic_type_str = ", type=paravirtualized"
964
    else:
965
      nic_type_str = ", model=%s, type=ioemu" % nic_type
966
    for nic in instance.nics:
967
      nic_str = "mac=%s%s" % (nic.mac, nic_type_str)
968
      ip = getattr(nic, "ip", None)
969
      if ip is not None:
970
        nic_str += ", ip=%s" % ip
971
      if nic.nicparams[constants.NIC_MODE] == constants.NIC_MODE_BRIDGED:
972
        nic_str += ", bridge=%s" % nic.nicparams[constants.NIC_LINK]
973
      vif_data.append("'%s'" % nic_str)
974

    
975
    config.write("vif = [%s]\n" % ",".join(vif_data))
976

    
977
    disk_data = \
978
      _GetConfigFileDiskData(block_devices, hvp[constants.HV_BLOCKDEV_PREFIX])
979

    
980
    iso_path = hvp[constants.HV_CDROM_IMAGE_PATH]
981
    if iso_path:
982
      iso = "'file:%s,hdc:cdrom,r'" % iso_path
983
      disk_data.append(iso)
984

    
985
    config.write("disk = [%s]\n" % (",".join(disk_data)))
986
    # Add PCI passthrough
987
    pci_pass_arr = []
988
    pci_pass = hvp[constants.HV_PASSTHROUGH]
989
    if pci_pass:
990
      pci_pass_arr = pci_pass.split(";")
991
      config.write("pci = %s\n" % pci_pass_arr)
992
    config.write("on_poweroff = 'destroy'\n")
993
    if hvp[constants.HV_REBOOT_BEHAVIOR] == constants.INSTANCE_REBOOT_ALLOWED:
994
      config.write("on_reboot = 'restart'\n")
995
    else:
996
      config.write("on_reboot = 'destroy'\n")
997
    config.write("on_crash = 'restart'\n")
998

    
999
    return config.getvalue()