Statistics
| Branch: | Tag: | Revision:

root / lib / backend.py @ b5b8309d

History | View | Annotate | Download (94.8 kB)

1
#
2
#
3

    
4
# Copyright (C) 2006, 2007 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
"""Functions used by the node daemon
23

24
@var _ALLOWED_UPLOAD_FILES: denotes which files are accepted in
25
     the L{UploadFile} function
26
@var _ALLOWED_CLEAN_DIRS: denotes which directories are accepted
27
     in the L{_CleanDirectory} function
28

29
"""
30

    
31
# pylint: disable-msg=E1103
32

    
33
# E1103: %s %r has no %r member (but some types could not be
34
# inferred), because the _TryOSFromDisk returns either (True, os_obj)
35
# or (False, "string") which confuses pylint
36

    
37

    
38
import os
39
import os.path
40
import shutil
41
import time
42
import stat
43
import errno
44
import re
45
import random
46
import logging
47
import tempfile
48
import zlib
49
import base64
50
import signal
51

    
52
from ganeti import errors
53
from ganeti import utils
54
from ganeti import ssh
55
from ganeti import hypervisor
56
from ganeti import constants
57
from ganeti import bdev
58
from ganeti import objects
59
from ganeti import ssconf
60
from ganeti import serializer
61

    
62

    
63
_BOOT_ID_PATH = "/proc/sys/kernel/random/boot_id"
64
_ALLOWED_CLEAN_DIRS = frozenset([
65
  constants.DATA_DIR,
66
  constants.JOB_QUEUE_ARCHIVE_DIR,
67
  constants.QUEUE_DIR,
68
  constants.CRYPTO_KEYS_DIR,
69
  ])
70
_MAX_SSL_CERT_VALIDITY = 7 * 24 * 60 * 60
71
_X509_KEY_FILE = "key"
72
_X509_CERT_FILE = "cert"
73
_IES_STATUS_FILE = "status"
74
_IES_PID_FILE = "pid"
75
_IES_CA_FILE = "ca"
76

    
77

    
78
class RPCFail(Exception):
79
  """Class denoting RPC failure.
80

81
  Its argument is the error message.
82

83
  """
84

    
85

    
86
def _Fail(msg, *args, **kwargs):
87
  """Log an error and the raise an RPCFail exception.
88

89
  This exception is then handled specially in the ganeti daemon and
90
  turned into a 'failed' return type. As such, this function is a
91
  useful shortcut for logging the error and returning it to the master
92
  daemon.
93

94
  @type msg: string
95
  @param msg: the text of the exception
96
  @raise RPCFail
97

98
  """
99
  if args:
100
    msg = msg % args
101
  if "log" not in kwargs or kwargs["log"]: # if we should log this error
102
    if "exc" in kwargs and kwargs["exc"]:
103
      logging.exception(msg)
104
    else:
105
      logging.error(msg)
106
  raise RPCFail(msg)
107

    
108

    
109
def _GetConfig():
110
  """Simple wrapper to return a SimpleStore.
111

112
  @rtype: L{ssconf.SimpleStore}
113
  @return: a SimpleStore instance
114

115
  """
116
  return ssconf.SimpleStore()
117

    
118

    
119
def _GetSshRunner(cluster_name):
120
  """Simple wrapper to return an SshRunner.
121

122
  @type cluster_name: str
123
  @param cluster_name: the cluster name, which is needed
124
      by the SshRunner constructor
125
  @rtype: L{ssh.SshRunner}
126
  @return: an SshRunner instance
127

128
  """
129
  return ssh.SshRunner(cluster_name)
130

    
131

    
132
def _Decompress(data):
133
  """Unpacks data compressed by the RPC client.
134

135
  @type data: list or tuple
136
  @param data: Data sent by RPC client
137
  @rtype: str
138
  @return: Decompressed data
139

140
  """
141
  assert isinstance(data, (list, tuple))
142
  assert len(data) == 2
143
  (encoding, content) = data
144
  if encoding == constants.RPC_ENCODING_NONE:
145
    return content
146
  elif encoding == constants.RPC_ENCODING_ZLIB_BASE64:
147
    return zlib.decompress(base64.b64decode(content))
148
  else:
149
    raise AssertionError("Unknown data encoding")
150

    
151

    
152
def _CleanDirectory(path, exclude=None):
153
  """Removes all regular files in a directory.
154

155
  @type path: str
156
  @param path: the directory to clean
157
  @type exclude: list
158
  @param exclude: list of files to be excluded, defaults
159
      to the empty list
160

161
  """
162
  if path not in _ALLOWED_CLEAN_DIRS:
163
    _Fail("Path passed to _CleanDirectory not in allowed clean targets: '%s'",
164
          path)
165

    
166
  if not os.path.isdir(path):
167
    return
168
  if exclude is None:
169
    exclude = []
170
  else:
171
    # Normalize excluded paths
172
    exclude = [os.path.normpath(i) for i in exclude]
173

    
174
  for rel_name in utils.ListVisibleFiles(path):
175
    full_name = utils.PathJoin(path, rel_name)
176
    if full_name in exclude:
177
      continue
178
    if os.path.isfile(full_name) and not os.path.islink(full_name):
179
      utils.RemoveFile(full_name)
180

    
181

    
182
def _BuildUploadFileList():
183
  """Build the list of allowed upload files.
184

185
  This is abstracted so that it's built only once at module import time.
186

187
  """
188
  allowed_files = set([
189
    constants.CLUSTER_CONF_FILE,
190
    constants.ETC_HOSTS,
191
    constants.SSH_KNOWN_HOSTS_FILE,
192
    constants.VNC_PASSWORD_FILE,
193
    constants.RAPI_CERT_FILE,
194
    constants.RAPI_USERS_FILE,
195
    constants.CONFD_HMAC_KEY,
196
    constants.CLUSTER_DOMAIN_SECRET_FILE,
197
    ])
198

    
199
  for hv_name in constants.HYPER_TYPES:
200
    hv_class = hypervisor.GetHypervisorClass(hv_name)
201
    allowed_files.update(hv_class.GetAncillaryFiles())
202

    
203
  return frozenset(allowed_files)
204

    
205

    
206
_ALLOWED_UPLOAD_FILES = _BuildUploadFileList()
207

    
208

    
209
def JobQueuePurge():
210
  """Removes job queue files and archived jobs.
211

212
  @rtype: tuple
213
  @return: True, None
214

215
  """
216
  _CleanDirectory(constants.QUEUE_DIR, exclude=[constants.JOB_QUEUE_LOCK_FILE])
217
  _CleanDirectory(constants.JOB_QUEUE_ARCHIVE_DIR)
218

    
219

    
220
def GetMasterInfo():
221
  """Returns master information.
222

223
  This is an utility function to compute master information, either
224
  for consumption here or from the node daemon.
225

226
  @rtype: tuple
227
  @return: master_netdev, master_ip, master_name
228
  @raise RPCFail: in case of errors
229

230
  """
231
  try:
232
    cfg = _GetConfig()
233
    master_netdev = cfg.GetMasterNetdev()
234
    master_ip = cfg.GetMasterIP()
235
    master_node = cfg.GetMasterNode()
236
  except errors.ConfigurationError, err:
237
    _Fail("Cluster configuration incomplete: %s", err, exc=True)
238
  return (master_netdev, master_ip, master_node)
239

    
240

    
241
def StartMaster(start_daemons, no_voting):
242
  """Activate local node as master node.
243

244
  The function will always try activate the IP address of the master
245
  (unless someone else has it). It will also start the master daemons,
246
  based on the start_daemons parameter.
247

248
  @type start_daemons: boolean
249
  @param start_daemons: whether to also start the master
250
      daemons (ganeti-masterd and ganeti-rapi)
251
  @type no_voting: boolean
252
  @param no_voting: whether to start ganeti-masterd without a node vote
253
      (if start_daemons is True), but still non-interactively
254
  @rtype: None
255

256
  """
257
  # GetMasterInfo will raise an exception if not able to return data
258
  master_netdev, master_ip, _ = GetMasterInfo()
259

    
260
  err_msgs = []
261
  if utils.TcpPing(master_ip, constants.DEFAULT_NODED_PORT):
262
    if utils.OwnIpAddress(master_ip):
263
      # we already have the ip:
264
      logging.debug("Master IP already configured, doing nothing")
265
    else:
266
      msg = "Someone else has the master ip, not activating"
267
      logging.error(msg)
268
      err_msgs.append(msg)
269
  else:
270
    result = utils.RunCmd(["ip", "address", "add", "%s/32" % master_ip,
271
                           "dev", master_netdev, "label",
272
                           "%s:0" % master_netdev])
273
    if result.failed:
274
      msg = "Can't activate master IP: %s" % result.output
275
      logging.error(msg)
276
      err_msgs.append(msg)
277

    
278
    result = utils.RunCmd(["arping", "-q", "-U", "-c 3", "-I", master_netdev,
279
                           "-s", master_ip, master_ip])
280
    # we'll ignore the exit code of arping
281

    
282
  # and now start the master and rapi daemons
283
  if start_daemons:
284
    if no_voting:
285
      masterd_args = "--no-voting --yes-do-it"
286
    else:
287
      masterd_args = ""
288

    
289
    env = {
290
      "EXTRA_MASTERD_ARGS": masterd_args,
291
      }
292

    
293
    result = utils.RunCmd([constants.DAEMON_UTIL, "start-master"], env=env)
294
    if result.failed:
295
      msg = "Can't start Ganeti master: %s" % result.output
296
      logging.error(msg)
297
      err_msgs.append(msg)
298

    
299
  if err_msgs:
300
    _Fail("; ".join(err_msgs))
301

    
302

    
303
def StopMaster(stop_daemons):
304
  """Deactivate this node as master.
305

306
  The function will always try to deactivate the IP address of the
307
  master. It will also stop the master daemons depending on the
308
  stop_daemons parameter.
309

310
  @type stop_daemons: boolean
311
  @param stop_daemons: whether to also stop the master daemons
312
      (ganeti-masterd and ganeti-rapi)
313
  @rtype: None
314

315
  """
316
  # TODO: log and report back to the caller the error failures; we
317
  # need to decide in which case we fail the RPC for this
318

    
319
  # GetMasterInfo will raise an exception if not able to return data
320
  master_netdev, master_ip, _ = GetMasterInfo()
321

    
322
  result = utils.RunCmd(["ip", "address", "del", "%s/32" % master_ip,
323
                         "dev", master_netdev])
324
  if result.failed:
325
    logging.error("Can't remove the master IP, error: %s", result.output)
326
    # but otherwise ignore the failure
327

    
328
  if stop_daemons:
329
    result = utils.RunCmd([constants.DAEMON_UTIL, "stop-master"])
330
    if result.failed:
331
      logging.error("Could not stop Ganeti master, command %s had exitcode %s"
332
                    " and error %s",
333
                    result.cmd, result.exit_code, result.output)
334

    
335

    
336
def AddNode(dsa, dsapub, rsa, rsapub, sshkey, sshpub):
337
  """Joins this node to the cluster.
338

339
  This does the following:
340
      - updates the hostkeys of the machine (rsa and dsa)
341
      - adds the ssh private key to the user
342
      - adds the ssh public key to the users' authorized_keys file
343

344
  @type dsa: str
345
  @param dsa: the DSA private key to write
346
  @type dsapub: str
347
  @param dsapub: the DSA public key to write
348
  @type rsa: str
349
  @param rsa: the RSA private key to write
350
  @type rsapub: str
351
  @param rsapub: the RSA public key to write
352
  @type sshkey: str
353
  @param sshkey: the SSH private key to write
354
  @type sshpub: str
355
  @param sshpub: the SSH public key to write
356
  @rtype: boolean
357
  @return: the success of the operation
358

359
  """
360
  sshd_keys =  [(constants.SSH_HOST_RSA_PRIV, rsa, 0600),
361
                (constants.SSH_HOST_RSA_PUB, rsapub, 0644),
362
                (constants.SSH_HOST_DSA_PRIV, dsa, 0600),
363
                (constants.SSH_HOST_DSA_PUB, dsapub, 0644)]
364
  for name, content, mode in sshd_keys:
365
    utils.WriteFile(name, data=content, mode=mode)
366

    
367
  try:
368
    priv_key, pub_key, auth_keys = ssh.GetUserFiles(constants.GANETI_RUNAS,
369
                                                    mkdir=True)
370
  except errors.OpExecError, err:
371
    _Fail("Error while processing user ssh files: %s", err, exc=True)
372

    
373
  for name, content in [(priv_key, sshkey), (pub_key, sshpub)]:
374
    utils.WriteFile(name, data=content, mode=0600)
375

    
376
  utils.AddAuthorizedKey(auth_keys, sshpub)
377

    
378
  result = utils.RunCmd([constants.DAEMON_UTIL, "reload-ssh-keys"])
379
  if result.failed:
380
    _Fail("Unable to reload SSH keys (command %r, exit code %s, output %r)",
381
          result.cmd, result.exit_code, result.output)
382

    
383

    
384
def LeaveCluster(modify_ssh_setup):
385
  """Cleans up and remove the current node.
386

387
  This function cleans up and prepares the current node to be removed
388
  from the cluster.
389

390
  If processing is successful, then it raises an
391
  L{errors.QuitGanetiException} which is used as a special case to
392
  shutdown the node daemon.
393

394
  @param modify_ssh_setup: boolean
395

396
  """
397
  _CleanDirectory(constants.DATA_DIR)
398
  _CleanDirectory(constants.CRYPTO_KEYS_DIR)
399
  JobQueuePurge()
400

    
401
  if modify_ssh_setup:
402
    try:
403
      priv_key, pub_key, auth_keys = ssh.GetUserFiles(constants.GANETI_RUNAS)
404

    
405
      utils.RemoveAuthorizedKey(auth_keys, utils.ReadFile(pub_key))
406

    
407
      utils.RemoveFile(priv_key)
408
      utils.RemoveFile(pub_key)
409
    except errors.OpExecError:
410
      logging.exception("Error while processing ssh files")
411

    
412
  try:
413
    utils.RemoveFile(constants.CONFD_HMAC_KEY)
414
    utils.RemoveFile(constants.RAPI_CERT_FILE)
415
    utils.RemoveFile(constants.NODED_CERT_FILE)
416
  except: # pylint: disable-msg=W0702
417
    logging.exception("Error while removing cluster secrets")
418

    
419
  result = utils.RunCmd([constants.DAEMON_UTIL, "stop", constants.CONFD])
420
  if result.failed:
421
    logging.error("Command %s failed with exitcode %s and error %s",
422
                  result.cmd, result.exit_code, result.output)
423

    
424
  # Raise a custom exception (handled in ganeti-noded)
425
  raise errors.QuitGanetiException(True, 'Shutdown scheduled')
426

    
427

    
428
def GetNodeInfo(vgname, hypervisor_type):
429
  """Gives back a hash with different information about the node.
430

431
  @type vgname: C{string}
432
  @param vgname: the name of the volume group to ask for disk space information
433
  @type hypervisor_type: C{str}
434
  @param hypervisor_type: the name of the hypervisor to ask for
435
      memory information
436
  @rtype: C{dict}
437
  @return: dictionary with the following keys:
438
      - vg_size is the size of the configured volume group in MiB
439
      - vg_free is the free size of the volume group in MiB
440
      - memory_dom0 is the memory allocated for domain0 in MiB
441
      - memory_free is the currently available (free) ram in MiB
442
      - memory_total is the total number of ram in MiB
443

444
  """
445
  outputarray = {}
446
  vginfo = _GetVGInfo(vgname)
447
  outputarray['vg_size'] = vginfo['vg_size']
448
  outputarray['vg_free'] = vginfo['vg_free']
449

    
450
  hyper = hypervisor.GetHypervisor(hypervisor_type)
451
  hyp_info = hyper.GetNodeInfo()
452
  if hyp_info is not None:
453
    outputarray.update(hyp_info)
454

    
455
  outputarray["bootid"] = utils.ReadFile(_BOOT_ID_PATH, size=128).rstrip("\n")
456

    
457
  return outputarray
458

    
459

    
460
def VerifyNode(what, cluster_name):
461
  """Verify the status of the local node.
462

463
  Based on the input L{what} parameter, various checks are done on the
464
  local node.
465

466
  If the I{filelist} key is present, this list of
467
  files is checksummed and the file/checksum pairs are returned.
468

469
  If the I{nodelist} key is present, we check that we have
470
  connectivity via ssh with the target nodes (and check the hostname
471
  report).
472

473
  If the I{node-net-test} key is present, we check that we have
474
  connectivity to the given nodes via both primary IP and, if
475
  applicable, secondary IPs.
476

477
  @type what: C{dict}
478
  @param what: a dictionary of things to check:
479
      - filelist: list of files for which to compute checksums
480
      - nodelist: list of nodes we should check ssh communication with
481
      - node-net-test: list of nodes we should check node daemon port
482
        connectivity with
483
      - hypervisor: list with hypervisors to run the verify for
484
  @rtype: dict
485
  @return: a dictionary with the same keys as the input dict, and
486
      values representing the result of the checks
487

488
  """
489
  result = {}
490
  my_name = utils.HostInfo().name
491
  port = utils.GetDaemonPort(constants.NODED)
492

    
493
  if constants.NV_HYPERVISOR in what:
494
    result[constants.NV_HYPERVISOR] = tmp = {}
495
    for hv_name in what[constants.NV_HYPERVISOR]:
496
      try:
497
        val = hypervisor.GetHypervisor(hv_name).Verify()
498
      except errors.HypervisorError, err:
499
        val = "Error while checking hypervisor: %s" % str(err)
500
      tmp[hv_name] = val
501

    
502
  if constants.NV_FILELIST in what:
503
    result[constants.NV_FILELIST] = utils.FingerprintFiles(
504
      what[constants.NV_FILELIST])
505

    
506
  if constants.NV_NODELIST in what:
507
    result[constants.NV_NODELIST] = tmp = {}
508
    random.shuffle(what[constants.NV_NODELIST])
509
    for node in what[constants.NV_NODELIST]:
510
      success, message = _GetSshRunner(cluster_name).VerifyNodeHostname(node)
511
      if not success:
512
        tmp[node] = message
513

    
514
  if constants.NV_NODENETTEST in what:
515
    result[constants.NV_NODENETTEST] = tmp = {}
516
    my_pip = my_sip = None
517
    for name, pip, sip in what[constants.NV_NODENETTEST]:
518
      if name == my_name:
519
        my_pip = pip
520
        my_sip = sip
521
        break
522
    if not my_pip:
523
      tmp[my_name] = ("Can't find my own primary/secondary IP"
524
                      " in the node list")
525
    else:
526
      for name, pip, sip in what[constants.NV_NODENETTEST]:
527
        fail = []
528
        if not utils.TcpPing(pip, port, source=my_pip):
529
          fail.append("primary")
530
        if sip != pip:
531
          if not utils.TcpPing(sip, port, source=my_sip):
532
            fail.append("secondary")
533
        if fail:
534
          tmp[name] = ("failure using the %s interface(s)" %
535
                       " and ".join(fail))
536

    
537
  if constants.NV_MASTERIP in what:
538
    # FIXME: add checks on incoming data structures (here and in the
539
    # rest of the function)
540
    master_name, master_ip = what[constants.NV_MASTERIP]
541
    if master_name == my_name:
542
      source = constants.LOCALHOST_IP_ADDRESS
543
    else:
544
      source = None
545
    result[constants.NV_MASTERIP] = utils.TcpPing(master_ip, port,
546
                                                  source=source)
547

    
548
  if constants.NV_LVLIST in what:
549
    try:
550
      val = GetVolumeList(what[constants.NV_LVLIST])
551
    except RPCFail, err:
552
      val = str(err)
553
    result[constants.NV_LVLIST] = val
554

    
555
  if constants.NV_INSTANCELIST in what:
556
    # GetInstanceList can fail
557
    try:
558
      val = GetInstanceList(what[constants.NV_INSTANCELIST])
559
    except RPCFail, err:
560
      val = str(err)
561
    result[constants.NV_INSTANCELIST] = val
562

    
563
  if constants.NV_VGLIST in what:
564
    result[constants.NV_VGLIST] = utils.ListVolumeGroups()
565

    
566
  if constants.NV_PVLIST in what:
567
    result[constants.NV_PVLIST] = \
568
      bdev.LogicalVolume.GetPVInfo(what[constants.NV_PVLIST],
569
                                   filter_allocatable=False)
570

    
571
  if constants.NV_VERSION in what:
572
    result[constants.NV_VERSION] = (constants.PROTOCOL_VERSION,
573
                                    constants.RELEASE_VERSION)
574

    
575
  if constants.NV_HVINFO in what:
576
    hyper = hypervisor.GetHypervisor(what[constants.NV_HVINFO])
577
    result[constants.NV_HVINFO] = hyper.GetNodeInfo()
578

    
579
  if constants.NV_DRBDLIST in what:
580
    try:
581
      used_minors = bdev.DRBD8.GetUsedDevs().keys()
582
    except errors.BlockDeviceError, err:
583
      logging.warning("Can't get used minors list", exc_info=True)
584
      used_minors = str(err)
585
    result[constants.NV_DRBDLIST] = used_minors
586

    
587
  if constants.NV_NODESETUP in what:
588
    result[constants.NV_NODESETUP] = tmpr = []
589
    if not os.path.isdir("/sys/block") or not os.path.isdir("/sys/class/net"):
590
      tmpr.append("The sysfs filesytem doesn't seem to be mounted"
591
                  " under /sys, missing required directories /sys/block"
592
                  " and /sys/class/net")
593
    if (not os.path.isdir("/proc/sys") or
594
        not os.path.isfile("/proc/sysrq-trigger")):
595
      tmpr.append("The procfs filesystem doesn't seem to be mounted"
596
                  " under /proc, missing required directory /proc/sys and"
597
                  " the file /proc/sysrq-trigger")
598

    
599
  if constants.NV_TIME in what:
600
    result[constants.NV_TIME] = utils.SplitTime(time.time())
601

    
602
  return result
603

    
604

    
605
def GetVolumeList(vg_name):
606
  """Compute list of logical volumes and their size.
607

608
  @type vg_name: str
609
  @param vg_name: the volume group whose LVs we should list
610
  @rtype: dict
611
  @return:
612
      dictionary of all partions (key) with value being a tuple of
613
      their size (in MiB), inactive and online status::
614

615
        {'test1': ('20.06', True, True)}
616

617
      in case of errors, a string is returned with the error
618
      details.
619

620
  """
621
  lvs = {}
622
  sep = '|'
623
  result = utils.RunCmd(["lvs", "--noheadings", "--units=m", "--nosuffix",
624
                         "--separator=%s" % sep,
625
                         "-olv_name,lv_size,lv_attr", vg_name])
626
  if result.failed:
627
    _Fail("Failed to list logical volumes, lvs output: %s", result.output)
628

    
629
  valid_line_re = re.compile("^ *([^|]+)\|([0-9.]+)\|([^|]{6})\|?$")
630
  for line in result.stdout.splitlines():
631
    line = line.strip()
632
    match = valid_line_re.match(line)
633
    if not match:
634
      logging.error("Invalid line returned from lvs output: '%s'", line)
635
      continue
636
    name, size, attr = match.groups()
637
    inactive = attr[4] == '-'
638
    online = attr[5] == 'o'
639
    virtual = attr[0] == 'v'
640
    if virtual:
641
      # we don't want to report such volumes as existing, since they
642
      # don't really hold data
643
      continue
644
    lvs[name] = (size, inactive, online)
645

    
646
  return lvs
647

    
648

    
649
def ListVolumeGroups():
650
  """List the volume groups and their size.
651

652
  @rtype: dict
653
  @return: dictionary with keys volume name and values the
654
      size of the volume
655

656
  """
657
  return utils.ListVolumeGroups()
658

    
659

    
660
def NodeVolumes():
661
  """List all volumes on this node.
662

663
  @rtype: list
664
  @return:
665
    A list of dictionaries, each having four keys:
666
      - name: the logical volume name,
667
      - size: the size of the logical volume
668
      - dev: the physical device on which the LV lives
669
      - vg: the volume group to which it belongs
670

671
    In case of errors, we return an empty list and log the
672
    error.
673

674
    Note that since a logical volume can live on multiple physical
675
    volumes, the resulting list might include a logical volume
676
    multiple times.
677

678
  """
679
  result = utils.RunCmd(["lvs", "--noheadings", "--units=m", "--nosuffix",
680
                         "--separator=|",
681
                         "--options=lv_name,lv_size,devices,vg_name"])
682
  if result.failed:
683
    _Fail("Failed to list logical volumes, lvs output: %s",
684
          result.output)
685

    
686
  def parse_dev(dev):
687
    return dev.split('(')[0]
688

    
689
  def handle_dev(dev):
690
    return [parse_dev(x) for x in dev.split(",")]
691

    
692
  def map_line(line):
693
    line = [v.strip() for v in line]
694
    return [{'name': line[0], 'size': line[1],
695
             'dev': dev, 'vg': line[3]} for dev in handle_dev(line[2])]
696

    
697
  all_devs = []
698
  for line in result.stdout.splitlines():
699
    if line.count('|') >= 3:
700
      all_devs.extend(map_line(line.split('|')))
701
    else:
702
      logging.warning("Strange line in the output from lvs: '%s'", line)
703
  return all_devs
704

    
705

    
706
def BridgesExist(bridges_list):
707
  """Check if a list of bridges exist on the current node.
708

709
  @rtype: boolean
710
  @return: C{True} if all of them exist, C{False} otherwise
711

712
  """
713
  missing = []
714
  for bridge in bridges_list:
715
    if not utils.BridgeExists(bridge):
716
      missing.append(bridge)
717

    
718
  if missing:
719
    _Fail("Missing bridges %s", utils.CommaJoin(missing))
720

    
721

    
722
def GetInstanceList(hypervisor_list):
723
  """Provides a list of instances.
724

725
  @type hypervisor_list: list
726
  @param hypervisor_list: the list of hypervisors to query information
727

728
  @rtype: list
729
  @return: a list of all running instances on the current node
730
    - instance1.example.com
731
    - instance2.example.com
732

733
  """
734
  results = []
735
  for hname in hypervisor_list:
736
    try:
737
      names = hypervisor.GetHypervisor(hname).ListInstances()
738
      results.extend(names)
739
    except errors.HypervisorError, err:
740
      _Fail("Error enumerating instances (hypervisor %s): %s",
741
            hname, err, exc=True)
742

    
743
  return results
744

    
745

    
746
def GetInstanceInfo(instance, hname):
747
  """Gives back the information about an instance as a dictionary.
748

749
  @type instance: string
750
  @param instance: the instance name
751
  @type hname: string
752
  @param hname: the hypervisor type of the instance
753

754
  @rtype: dict
755
  @return: dictionary with the following keys:
756
      - memory: memory size of instance (int)
757
      - state: xen state of instance (string)
758
      - time: cpu time of instance (float)
759

760
  """
761
  output = {}
762

    
763
  iinfo = hypervisor.GetHypervisor(hname).GetInstanceInfo(instance)
764
  if iinfo is not None:
765
    output['memory'] = iinfo[2]
766
    output['state'] = iinfo[4]
767
    output['time'] = iinfo[5]
768

    
769
  return output
770

    
771

    
772
def GetInstanceMigratable(instance):
773
  """Gives whether an instance can be migrated.
774

775
  @type instance: L{objects.Instance}
776
  @param instance: object representing the instance to be checked.
777

778
  @rtype: tuple
779
  @return: tuple of (result, description) where:
780
      - result: whether the instance can be migrated or not
781
      - description: a description of the issue, if relevant
782

783
  """
784
  hyper = hypervisor.GetHypervisor(instance.hypervisor)
785
  iname = instance.name
786
  if iname not in hyper.ListInstances():
787
    _Fail("Instance %s is not running", iname)
788

    
789
  for idx in range(len(instance.disks)):
790
    link_name = _GetBlockDevSymlinkPath(iname, idx)
791
    if not os.path.islink(link_name):
792
      _Fail("Instance %s was not restarted since ganeti 1.2.5", iname)
793

    
794

    
795
def GetAllInstancesInfo(hypervisor_list):
796
  """Gather data about all instances.
797

798
  This is the equivalent of L{GetInstanceInfo}, except that it
799
  computes data for all instances at once, thus being faster if one
800
  needs data about more than one instance.
801

802
  @type hypervisor_list: list
803
  @param hypervisor_list: list of hypervisors to query for instance data
804

805
  @rtype: dict
806
  @return: dictionary of instance: data, with data having the following keys:
807
      - memory: memory size of instance (int)
808
      - state: xen state of instance (string)
809
      - time: cpu time of instance (float)
810
      - vcpus: the number of vcpus
811

812
  """
813
  output = {}
814

    
815
  for hname in hypervisor_list:
816
    iinfo = hypervisor.GetHypervisor(hname).GetAllInstancesInfo()
817
    if iinfo:
818
      for name, _, memory, vcpus, state, times in iinfo:
819
        value = {
820
          'memory': memory,
821
          'vcpus': vcpus,
822
          'state': state,
823
          'time': times,
824
          }
825
        if name in output:
826
          # we only check static parameters, like memory and vcpus,
827
          # and not state and time which can change between the
828
          # invocations of the different hypervisors
829
          for key in 'memory', 'vcpus':
830
            if value[key] != output[name][key]:
831
              _Fail("Instance %s is running twice"
832
                    " with different parameters", name)
833
        output[name] = value
834

    
835
  return output
836

    
837

    
838
def _InstanceLogName(kind, os_name, instance):
839
  """Compute the OS log filename for a given instance and operation.
840

841
  The instance name and os name are passed in as strings since not all
842
  operations have these as part of an instance object.
843

844
  @type kind: string
845
  @param kind: the operation type (e.g. add, import, etc.)
846
  @type os_name: string
847
  @param os_name: the os name
848
  @type instance: string
849
  @param instance: the name of the instance being imported/added/etc.
850

851
  """
852
  # TODO: Use tempfile.mkstemp to create unique filename
853
  base = ("%s-%s-%s-%s.log" %
854
          (kind, os_name, instance, utils.TimestampForFilename()))
855
  return utils.PathJoin(constants.LOG_OS_DIR, base)
856

    
857

    
858
def InstanceOsAdd(instance, reinstall, debug):
859
  """Add an OS to an instance.
860

861
  @type instance: L{objects.Instance}
862
  @param instance: Instance whose OS is to be installed
863
  @type reinstall: boolean
864
  @param reinstall: whether this is an instance reinstall
865
  @type debug: integer
866
  @param debug: debug level, passed to the OS scripts
867
  @rtype: None
868

869
  """
870
  inst_os = OSFromDisk(instance.os)
871

    
872
  create_env = OSEnvironment(instance, inst_os, debug)
873
  if reinstall:
874
    create_env['INSTANCE_REINSTALL'] = "1"
875

    
876
  logfile = _InstanceLogName("add", instance.os, instance.name)
877

    
878
  result = utils.RunCmd([inst_os.create_script], env=create_env,
879
                        cwd=inst_os.path, output=logfile,)
880
  if result.failed:
881
    logging.error("os create command '%s' returned error: %s, logfile: %s,"
882
                  " output: %s", result.cmd, result.fail_reason, logfile,
883
                  result.output)
884
    lines = [utils.SafeEncode(val)
885
             for val in utils.TailFile(logfile, lines=20)]
886
    _Fail("OS create script failed (%s), last lines in the"
887
          " log file:\n%s", result.fail_reason, "\n".join(lines), log=False)
888

    
889

    
890
def RunRenameInstance(instance, old_name, debug):
891
  """Run the OS rename script for an instance.
892

893
  @type instance: L{objects.Instance}
894
  @param instance: Instance whose OS is to be installed
895
  @type old_name: string
896
  @param old_name: previous instance name
897
  @type debug: integer
898
  @param debug: debug level, passed to the OS scripts
899
  @rtype: boolean
900
  @return: the success of the operation
901

902
  """
903
  inst_os = OSFromDisk(instance.os)
904

    
905
  rename_env = OSEnvironment(instance, inst_os, debug)
906
  rename_env['OLD_INSTANCE_NAME'] = old_name
907

    
908
  logfile = _InstanceLogName("rename", instance.os,
909
                             "%s-%s" % (old_name, instance.name))
910

    
911
  result = utils.RunCmd([inst_os.rename_script], env=rename_env,
912
                        cwd=inst_os.path, output=logfile)
913

    
914
  if result.failed:
915
    logging.error("os create command '%s' returned error: %s output: %s",
916
                  result.cmd, result.fail_reason, result.output)
917
    lines = [utils.SafeEncode(val)
918
             for val in utils.TailFile(logfile, lines=20)]
919
    _Fail("OS rename script failed (%s), last lines in the"
920
          " log file:\n%s", result.fail_reason, "\n".join(lines), log=False)
921

    
922

    
923
def _GetVGInfo(vg_name):
924
  """Get information about the volume group.
925

926
  @type vg_name: str
927
  @param vg_name: the volume group which we query
928
  @rtype: dict
929
  @return:
930
    A dictionary with the following keys:
931
      - C{vg_size} is the total size of the volume group in MiB
932
      - C{vg_free} is the free size of the volume group in MiB
933
      - C{pv_count} are the number of physical disks in that VG
934

935
    If an error occurs during gathering of data, we return the same dict
936
    with keys all set to None.
937

938
  """
939
  retdic = dict.fromkeys(["vg_size", "vg_free", "pv_count"])
940

    
941
  retval = utils.RunCmd(["vgs", "-ovg_size,vg_free,pv_count", "--noheadings",
942
                         "--nosuffix", "--units=m", "--separator=:", vg_name])
943

    
944
  if retval.failed:
945
    logging.error("volume group %s not present", vg_name)
946
    return retdic
947
  valarr = retval.stdout.strip().rstrip(':').split(':')
948
  if len(valarr) == 3:
949
    try:
950
      retdic = {
951
        "vg_size": int(round(float(valarr[0]), 0)),
952
        "vg_free": int(round(float(valarr[1]), 0)),
953
        "pv_count": int(valarr[2]),
954
        }
955
    except (TypeError, ValueError), err:
956
      logging.exception("Fail to parse vgs output: %s", err)
957
  else:
958
    logging.error("vgs output has the wrong number of fields (expected"
959
                  " three): %s", str(valarr))
960
  return retdic
961

    
962

    
963
def _GetBlockDevSymlinkPath(instance_name, idx):
964
  return utils.PathJoin(constants.DISK_LINKS_DIR,
965
                        "%s:%d" % (instance_name, idx))
966

    
967

    
968
def _SymlinkBlockDev(instance_name, device_path, idx):
969
  """Set up symlinks to a instance's block device.
970

971
  This is an auxiliary function run when an instance is start (on the primary
972
  node) or when an instance is migrated (on the target node).
973

974

975
  @param instance_name: the name of the target instance
976
  @param device_path: path of the physical block device, on the node
977
  @param idx: the disk index
978
  @return: absolute path to the disk's symlink
979

980
  """
981
  link_name = _GetBlockDevSymlinkPath(instance_name, idx)
982
  try:
983
    os.symlink(device_path, link_name)
984
  except OSError, err:
985
    if err.errno == errno.EEXIST:
986
      if (not os.path.islink(link_name) or
987
          os.readlink(link_name) != device_path):
988
        os.remove(link_name)
989
        os.symlink(device_path, link_name)
990
    else:
991
      raise
992

    
993
  return link_name
994

    
995

    
996
def _RemoveBlockDevLinks(instance_name, disks):
997
  """Remove the block device symlinks belonging to the given instance.
998

999
  """
1000
  for idx, _ in enumerate(disks):
1001
    link_name = _GetBlockDevSymlinkPath(instance_name, idx)
1002
    if os.path.islink(link_name):
1003
      try:
1004
        os.remove(link_name)
1005
      except OSError:
1006
        logging.exception("Can't remove symlink '%s'", link_name)
1007

    
1008

    
1009
def _GatherAndLinkBlockDevs(instance):
1010
  """Set up an instance's block device(s).
1011

1012
  This is run on the primary node at instance startup. The block
1013
  devices must be already assembled.
1014

1015
  @type instance: L{objects.Instance}
1016
  @param instance: the instance whose disks we shoul assemble
1017
  @rtype: list
1018
  @return: list of (disk_object, device_path)
1019

1020
  """
1021
  block_devices = []
1022
  for idx, disk in enumerate(instance.disks):
1023
    device = _RecursiveFindBD(disk)
1024
    if device is None:
1025
      raise errors.BlockDeviceError("Block device '%s' is not set up." %
1026
                                    str(disk))
1027
    device.Open()
1028
    try:
1029
      link_name = _SymlinkBlockDev(instance.name, device.dev_path, idx)
1030
    except OSError, e:
1031
      raise errors.BlockDeviceError("Cannot create block device symlink: %s" %
1032
                                    e.strerror)
1033

    
1034
    block_devices.append((disk, link_name))
1035

    
1036
  return block_devices
1037

    
1038

    
1039
def StartInstance(instance):
1040
  """Start an instance.
1041

1042
  @type instance: L{objects.Instance}
1043
  @param instance: the instance object
1044
  @rtype: None
1045

1046
  """
1047
  running_instances = GetInstanceList([instance.hypervisor])
1048

    
1049
  if instance.name in running_instances:
1050
    logging.info("Instance %s already running, not starting", instance.name)
1051
    return
1052

    
1053
  try:
1054
    block_devices = _GatherAndLinkBlockDevs(instance)
1055
    hyper = hypervisor.GetHypervisor(instance.hypervisor)
1056
    hyper.StartInstance(instance, block_devices)
1057
  except errors.BlockDeviceError, err:
1058
    _Fail("Block device error: %s", err, exc=True)
1059
  except errors.HypervisorError, err:
1060
    _RemoveBlockDevLinks(instance.name, instance.disks)
1061
    _Fail("Hypervisor error: %s", err, exc=True)
1062

    
1063

    
1064
def InstanceShutdown(instance, timeout):
1065
  """Shut an instance down.
1066

1067
  @note: this functions uses polling with a hardcoded timeout.
1068

1069
  @type instance: L{objects.Instance}
1070
  @param instance: the instance object
1071
  @type timeout: integer
1072
  @param timeout: maximum timeout for soft shutdown
1073
  @rtype: None
1074

1075
  """
1076
  hv_name = instance.hypervisor
1077
  hyper = hypervisor.GetHypervisor(hv_name)
1078
  iname = instance.name
1079

    
1080
  if instance.name not in hyper.ListInstances():
1081
    logging.info("Instance %s not running, doing nothing", iname)
1082
    return
1083

    
1084
  class _TryShutdown:
1085
    def __init__(self):
1086
      self.tried_once = False
1087

    
1088
    def __call__(self):
1089
      if iname not in hyper.ListInstances():
1090
        return
1091

    
1092
      try:
1093
        hyper.StopInstance(instance, retry=self.tried_once)
1094
      except errors.HypervisorError, err:
1095
        if iname not in hyper.ListInstances():
1096
          # if the instance is no longer existing, consider this a
1097
          # success and go to cleanup
1098
          return
1099

    
1100
        _Fail("Failed to stop instance %s: %s", iname, err)
1101

    
1102
      self.tried_once = True
1103

    
1104
      raise utils.RetryAgain()
1105

    
1106
  try:
1107
    utils.Retry(_TryShutdown(), 5, timeout)
1108
  except utils.RetryTimeout:
1109
    # the shutdown did not succeed
1110
    logging.error("Shutdown of '%s' unsuccessful, forcing", iname)
1111

    
1112
    try:
1113
      hyper.StopInstance(instance, force=True)
1114
    except errors.HypervisorError, err:
1115
      if iname in hyper.ListInstances():
1116
        # only raise an error if the instance still exists, otherwise
1117
        # the error could simply be "instance ... unknown"!
1118
        _Fail("Failed to force stop instance %s: %s", iname, err)
1119

    
1120
    time.sleep(1)
1121

    
1122
    if iname in hyper.ListInstances():
1123
      _Fail("Could not shutdown instance %s even by destroy", iname)
1124

    
1125
  try:
1126
    hyper.CleanupInstance(instance.name)
1127
  except errors.HypervisorError, err:
1128
    logging.warning("Failed to execute post-shutdown cleanup step: %s", err)
1129

    
1130
  _RemoveBlockDevLinks(iname, instance.disks)
1131

    
1132

    
1133
def InstanceReboot(instance, reboot_type, shutdown_timeout):
1134
  """Reboot an instance.
1135

1136
  @type instance: L{objects.Instance}
1137
  @param instance: the instance object to reboot
1138
  @type reboot_type: str
1139
  @param reboot_type: the type of reboot, one the following
1140
    constants:
1141
      - L{constants.INSTANCE_REBOOT_SOFT}: only reboot the
1142
        instance OS, do not recreate the VM
1143
      - L{constants.INSTANCE_REBOOT_HARD}: tear down and
1144
        restart the VM (at the hypervisor level)
1145
      - the other reboot type (L{constants.INSTANCE_REBOOT_FULL}) is
1146
        not accepted here, since that mode is handled differently, in
1147
        cmdlib, and translates into full stop and start of the
1148
        instance (instead of a call_instance_reboot RPC)
1149
  @type shutdown_timeout: integer
1150
  @param shutdown_timeout: maximum timeout for soft shutdown
1151
  @rtype: None
1152

1153
  """
1154
  running_instances = GetInstanceList([instance.hypervisor])
1155

    
1156
  if instance.name not in running_instances:
1157
    _Fail("Cannot reboot instance %s that is not running", instance.name)
1158

    
1159
  hyper = hypervisor.GetHypervisor(instance.hypervisor)
1160
  if reboot_type == constants.INSTANCE_REBOOT_SOFT:
1161
    try:
1162
      hyper.RebootInstance(instance)
1163
    except errors.HypervisorError, err:
1164
      _Fail("Failed to soft reboot instance %s: %s", instance.name, err)
1165
  elif reboot_type == constants.INSTANCE_REBOOT_HARD:
1166
    try:
1167
      InstanceShutdown(instance, shutdown_timeout)
1168
      return StartInstance(instance)
1169
    except errors.HypervisorError, err:
1170
      _Fail("Failed to hard reboot instance %s: %s", instance.name, err)
1171
  else:
1172
    _Fail("Invalid reboot_type received: %s", reboot_type)
1173

    
1174

    
1175
def MigrationInfo(instance):
1176
  """Gather information about an instance to be migrated.
1177

1178
  @type instance: L{objects.Instance}
1179
  @param instance: the instance definition
1180

1181
  """
1182
  hyper = hypervisor.GetHypervisor(instance.hypervisor)
1183
  try:
1184
    info = hyper.MigrationInfo(instance)
1185
  except errors.HypervisorError, err:
1186
    _Fail("Failed to fetch migration information: %s", err, exc=True)
1187
  return info
1188

    
1189

    
1190
def AcceptInstance(instance, info, target):
1191
  """Prepare the node to accept an instance.
1192

1193
  @type instance: L{objects.Instance}
1194
  @param instance: the instance definition
1195
  @type info: string/data (opaque)
1196
  @param info: migration information, from the source node
1197
  @type target: string
1198
  @param target: target host (usually ip), on this node
1199

1200
  """
1201
  hyper = hypervisor.GetHypervisor(instance.hypervisor)
1202
  try:
1203
    hyper.AcceptInstance(instance, info, target)
1204
  except errors.HypervisorError, err:
1205
    _Fail("Failed to accept instance: %s", err, exc=True)
1206

    
1207

    
1208
def FinalizeMigration(instance, info, success):
1209
  """Finalize any preparation to accept an instance.
1210

1211
  @type instance: L{objects.Instance}
1212
  @param instance: the instance definition
1213
  @type info: string/data (opaque)
1214
  @param info: migration information, from the source node
1215
  @type success: boolean
1216
  @param success: whether the migration was a success or a failure
1217

1218
  """
1219
  hyper = hypervisor.GetHypervisor(instance.hypervisor)
1220
  try:
1221
    hyper.FinalizeMigration(instance, info, success)
1222
  except errors.HypervisorError, err:
1223
    _Fail("Failed to finalize migration: %s", err, exc=True)
1224

    
1225

    
1226
def MigrateInstance(instance, target, live):
1227
  """Migrates an instance to another node.
1228

1229
  @type instance: L{objects.Instance}
1230
  @param instance: the instance definition
1231
  @type target: string
1232
  @param target: the target node name
1233
  @type live: boolean
1234
  @param live: whether the migration should be done live or not (the
1235
      interpretation of this parameter is left to the hypervisor)
1236
  @rtype: tuple
1237
  @return: a tuple of (success, msg) where:
1238
      - succes is a boolean denoting the success/failure of the operation
1239
      - msg is a string with details in case of failure
1240

1241
  """
1242
  hyper = hypervisor.GetHypervisor(instance.hypervisor)
1243

    
1244
  try:
1245
    hyper.MigrateInstance(instance, target, live)
1246
  except errors.HypervisorError, err:
1247
    _Fail("Failed to migrate instance: %s", err, exc=True)
1248

    
1249

    
1250
def BlockdevCreate(disk, size, owner, on_primary, info):
1251
  """Creates a block device for an instance.
1252

1253
  @type disk: L{objects.Disk}
1254
  @param disk: the object describing the disk we should create
1255
  @type size: int
1256
  @param size: the size of the physical underlying device, in MiB
1257
  @type owner: str
1258
  @param owner: the name of the instance for which disk is created,
1259
      used for device cache data
1260
  @type on_primary: boolean
1261
  @param on_primary:  indicates if it is the primary node or not
1262
  @type info: string
1263
  @param info: string that will be sent to the physical device
1264
      creation, used for example to set (LVM) tags on LVs
1265

1266
  @return: the new unique_id of the device (this can sometime be
1267
      computed only after creation), or None. On secondary nodes,
1268
      it's not required to return anything.
1269

1270
  """
1271
  # TODO: remove the obsolete 'size' argument
1272
  # pylint: disable-msg=W0613
1273
  clist = []
1274
  if disk.children:
1275
    for child in disk.children:
1276
      try:
1277
        crdev = _RecursiveAssembleBD(child, owner, on_primary)
1278
      except errors.BlockDeviceError, err:
1279
        _Fail("Can't assemble device %s: %s", child, err)
1280
      if on_primary or disk.AssembleOnSecondary():
1281
        # we need the children open in case the device itself has to
1282
        # be assembled
1283
        try:
1284
          # pylint: disable-msg=E1103
1285
          crdev.Open()
1286
        except errors.BlockDeviceError, err:
1287
          _Fail("Can't make child '%s' read-write: %s", child, err)
1288
      clist.append(crdev)
1289

    
1290
  try:
1291
    device = bdev.Create(disk.dev_type, disk.physical_id, clist, disk.size)
1292
  except errors.BlockDeviceError, err:
1293
    _Fail("Can't create block device: %s", err)
1294

    
1295
  if on_primary or disk.AssembleOnSecondary():
1296
    try:
1297
      device.Assemble()
1298
    except errors.BlockDeviceError, err:
1299
      _Fail("Can't assemble device after creation, unusual event: %s", err)
1300
    device.SetSyncSpeed(constants.SYNC_SPEED)
1301
    if on_primary or disk.OpenOnSecondary():
1302
      try:
1303
        device.Open(force=True)
1304
      except errors.BlockDeviceError, err:
1305
        _Fail("Can't make device r/w after creation, unusual event: %s", err)
1306
    DevCacheManager.UpdateCache(device.dev_path, owner,
1307
                                on_primary, disk.iv_name)
1308

    
1309
  device.SetInfo(info)
1310

    
1311
  return device.unique_id
1312

    
1313

    
1314
def BlockdevRemove(disk):
1315
  """Remove a block device.
1316

1317
  @note: This is intended to be called recursively.
1318

1319
  @type disk: L{objects.Disk}
1320
  @param disk: the disk object we should remove
1321
  @rtype: boolean
1322
  @return: the success of the operation
1323

1324
  """
1325
  msgs = []
1326
  try:
1327
    rdev = _RecursiveFindBD(disk)
1328
  except errors.BlockDeviceError, err:
1329
    # probably can't attach
1330
    logging.info("Can't attach to device %s in remove", disk)
1331
    rdev = None
1332
  if rdev is not None:
1333
    r_path = rdev.dev_path
1334
    try:
1335
      rdev.Remove()
1336
    except errors.BlockDeviceError, err:
1337
      msgs.append(str(err))
1338
    if not msgs:
1339
      DevCacheManager.RemoveCache(r_path)
1340

    
1341
  if disk.children:
1342
    for child in disk.children:
1343
      try:
1344
        BlockdevRemove(child)
1345
      except RPCFail, err:
1346
        msgs.append(str(err))
1347

    
1348
  if msgs:
1349
    _Fail("; ".join(msgs))
1350

    
1351

    
1352
def _RecursiveAssembleBD(disk, owner, as_primary):
1353
  """Activate a block device for an instance.
1354

1355
  This is run on the primary and secondary nodes for an instance.
1356

1357
  @note: this function is called recursively.
1358

1359
  @type disk: L{objects.Disk}
1360
  @param disk: the disk we try to assemble
1361
  @type owner: str
1362
  @param owner: the name of the instance which owns the disk
1363
  @type as_primary: boolean
1364
  @param as_primary: if we should make the block device
1365
      read/write
1366

1367
  @return: the assembled device or None (in case no device
1368
      was assembled)
1369
  @raise errors.BlockDeviceError: in case there is an error
1370
      during the activation of the children or the device
1371
      itself
1372

1373
  """
1374
  children = []
1375
  if disk.children:
1376
    mcn = disk.ChildrenNeeded()
1377
    if mcn == -1:
1378
      mcn = 0 # max number of Nones allowed
1379
    else:
1380
      mcn = len(disk.children) - mcn # max number of Nones
1381
    for chld_disk in disk.children:
1382
      try:
1383
        cdev = _RecursiveAssembleBD(chld_disk, owner, as_primary)
1384
      except errors.BlockDeviceError, err:
1385
        if children.count(None) >= mcn:
1386
          raise
1387
        cdev = None
1388
        logging.error("Error in child activation (but continuing): %s",
1389
                      str(err))
1390
      children.append(cdev)
1391

    
1392
  if as_primary or disk.AssembleOnSecondary():
1393
    r_dev = bdev.Assemble(disk.dev_type, disk.physical_id, children, disk.size)
1394
    r_dev.SetSyncSpeed(constants.SYNC_SPEED)
1395
    result = r_dev
1396
    if as_primary or disk.OpenOnSecondary():
1397
      r_dev.Open()
1398
    DevCacheManager.UpdateCache(r_dev.dev_path, owner,
1399
                                as_primary, disk.iv_name)
1400

    
1401
  else:
1402
    result = True
1403
  return result
1404

    
1405

    
1406
def BlockdevAssemble(disk, owner, as_primary):
1407
  """Activate a block device for an instance.
1408

1409
  This is a wrapper over _RecursiveAssembleBD.
1410

1411
  @rtype: str or boolean
1412
  @return: a C{/dev/...} path for primary nodes, and
1413
      C{True} for secondary nodes
1414

1415
  """
1416
  try:
1417
    result = _RecursiveAssembleBD(disk, owner, as_primary)
1418
    if isinstance(result, bdev.BlockDev):
1419
      # pylint: disable-msg=E1103
1420
      result = result.dev_path
1421
  except errors.BlockDeviceError, err:
1422
    _Fail("Error while assembling disk: %s", err, exc=True)
1423

    
1424
  return result
1425

    
1426

    
1427
def BlockdevShutdown(disk):
1428
  """Shut down a block device.
1429

1430
  First, if the device is assembled (Attach() is successful), then
1431
  the device is shutdown. Then the children of the device are
1432
  shutdown.
1433

1434
  This function is called recursively. Note that we don't cache the
1435
  children or such, as oppossed to assemble, shutdown of different
1436
  devices doesn't require that the upper device was active.
1437

1438
  @type disk: L{objects.Disk}
1439
  @param disk: the description of the disk we should
1440
      shutdown
1441
  @rtype: None
1442

1443
  """
1444
  msgs = []
1445
  r_dev = _RecursiveFindBD(disk)
1446
  if r_dev is not None:
1447
    r_path = r_dev.dev_path
1448
    try:
1449
      r_dev.Shutdown()
1450
      DevCacheManager.RemoveCache(r_path)
1451
    except errors.BlockDeviceError, err:
1452
      msgs.append(str(err))
1453

    
1454
  if disk.children:
1455
    for child in disk.children:
1456
      try:
1457
        BlockdevShutdown(child)
1458
      except RPCFail, err:
1459
        msgs.append(str(err))
1460

    
1461
  if msgs:
1462
    _Fail("; ".join(msgs))
1463

    
1464

    
1465
def BlockdevAddchildren(parent_cdev, new_cdevs):
1466
  """Extend a mirrored block device.
1467

1468
  @type parent_cdev: L{objects.Disk}
1469
  @param parent_cdev: the disk to which we should add children
1470
  @type new_cdevs: list of L{objects.Disk}
1471
  @param new_cdevs: the list of children which we should add
1472
  @rtype: None
1473

1474
  """
1475
  parent_bdev = _RecursiveFindBD(parent_cdev)
1476
  if parent_bdev is None:
1477
    _Fail("Can't find parent device '%s' in add children", parent_cdev)
1478
  new_bdevs = [_RecursiveFindBD(disk) for disk in new_cdevs]
1479
  if new_bdevs.count(None) > 0:
1480
    _Fail("Can't find new device(s) to add: %s:%s", new_bdevs, new_cdevs)
1481
  parent_bdev.AddChildren(new_bdevs)
1482

    
1483

    
1484
def BlockdevRemovechildren(parent_cdev, new_cdevs):
1485
  """Shrink a mirrored block device.
1486

1487
  @type parent_cdev: L{objects.Disk}
1488
  @param parent_cdev: the disk from which we should remove children
1489
  @type new_cdevs: list of L{objects.Disk}
1490
  @param new_cdevs: the list of children which we should remove
1491
  @rtype: None
1492

1493
  """
1494
  parent_bdev = _RecursiveFindBD(parent_cdev)
1495
  if parent_bdev is None:
1496
    _Fail("Can't find parent device '%s' in remove children", parent_cdev)
1497
  devs = []
1498
  for disk in new_cdevs:
1499
    rpath = disk.StaticDevPath()
1500
    if rpath is None:
1501
      bd = _RecursiveFindBD(disk)
1502
      if bd is None:
1503
        _Fail("Can't find device %s while removing children", disk)
1504
      else:
1505
        devs.append(bd.dev_path)
1506
    else:
1507
      if not utils.IsNormAbsPath(rpath):
1508
        _Fail("Strange path returned from StaticDevPath: '%s'", rpath)
1509
      devs.append(rpath)
1510
  parent_bdev.RemoveChildren(devs)
1511

    
1512

    
1513
def BlockdevGetmirrorstatus(disks):
1514
  """Get the mirroring status of a list of devices.
1515

1516
  @type disks: list of L{objects.Disk}
1517
  @param disks: the list of disks which we should query
1518
  @rtype: disk
1519
  @return:
1520
      a list of (mirror_done, estimated_time) tuples, which
1521
      are the result of L{bdev.BlockDev.CombinedSyncStatus}
1522
  @raise errors.BlockDeviceError: if any of the disks cannot be
1523
      found
1524

1525
  """
1526
  stats = []
1527
  for dsk in disks:
1528
    rbd = _RecursiveFindBD(dsk)
1529
    if rbd is None:
1530
      _Fail("Can't find device %s", dsk)
1531

    
1532
    stats.append(rbd.CombinedSyncStatus())
1533

    
1534
  return stats
1535

    
1536

    
1537
def _RecursiveFindBD(disk):
1538
  """Check if a device is activated.
1539

1540
  If so, return information about the real device.
1541

1542
  @type disk: L{objects.Disk}
1543
  @param disk: the disk object we need to find
1544

1545
  @return: None if the device can't be found,
1546
      otherwise the device instance
1547

1548
  """
1549
  children = []
1550
  if disk.children:
1551
    for chdisk in disk.children:
1552
      children.append(_RecursiveFindBD(chdisk))
1553

    
1554
  return bdev.FindDevice(disk.dev_type, disk.physical_id, children, disk.size)
1555

    
1556

    
1557
def _OpenRealBD(disk):
1558
  """Opens the underlying block device of a disk.
1559

1560
  @type disk: L{objects.Disk}
1561
  @param disk: the disk object we want to open
1562

1563
  """
1564
  real_disk = _RecursiveFindBD(disk)
1565
  if real_disk is None:
1566
    _Fail("Block device '%s' is not set up", disk)
1567

    
1568
  real_disk.Open()
1569

    
1570
  return real_disk
1571

    
1572

    
1573
def BlockdevFind(disk):
1574
  """Check if a device is activated.
1575

1576
  If it is, return information about the real device.
1577

1578
  @type disk: L{objects.Disk}
1579
  @param disk: the disk to find
1580
  @rtype: None or objects.BlockDevStatus
1581
  @return: None if the disk cannot be found, otherwise a the current
1582
           information
1583

1584
  """
1585
  try:
1586
    rbd = _RecursiveFindBD(disk)
1587
  except errors.BlockDeviceError, err:
1588
    _Fail("Failed to find device: %s", err, exc=True)
1589

    
1590
  if rbd is None:
1591
    return None
1592

    
1593
  return rbd.GetSyncStatus()
1594

    
1595

    
1596
def BlockdevGetsize(disks):
1597
  """Computes the size of the given disks.
1598

1599
  If a disk is not found, returns None instead.
1600

1601
  @type disks: list of L{objects.Disk}
1602
  @param disks: the list of disk to compute the size for
1603
  @rtype: list
1604
  @return: list with elements None if the disk cannot be found,
1605
      otherwise the size
1606

1607
  """
1608
  result = []
1609
  for cf in disks:
1610
    try:
1611
      rbd = _RecursiveFindBD(cf)
1612
    except errors.BlockDeviceError:
1613
      result.append(None)
1614
      continue
1615
    if rbd is None:
1616
      result.append(None)
1617
    else:
1618
      result.append(rbd.GetActualSize())
1619
  return result
1620

    
1621

    
1622
def BlockdevExport(disk, dest_node, dest_path, cluster_name):
1623
  """Export a block device to a remote node.
1624

1625
  @type disk: L{objects.Disk}
1626
  @param disk: the description of the disk to export
1627
  @type dest_node: str
1628
  @param dest_node: the destination node to export to
1629
  @type dest_path: str
1630
  @param dest_path: the destination path on the target node
1631
  @type cluster_name: str
1632
  @param cluster_name: the cluster name, needed for SSH hostalias
1633
  @rtype: None
1634

1635
  """
1636
  real_disk = _OpenRealBD(disk)
1637

    
1638
  # the block size on the read dd is 1MiB to match our units
1639
  expcmd = utils.BuildShellCmd("set -e; set -o pipefail; "
1640
                               "dd if=%s bs=1048576 count=%s",
1641
                               real_disk.dev_path, str(disk.size))
1642

    
1643
  # we set here a smaller block size as, due to ssh buffering, more
1644
  # than 64-128k will mostly ignored; we use nocreat to fail if the
1645
  # device is not already there or we pass a wrong path; we use
1646
  # notrunc to no attempt truncate on an LV device; we use oflag=dsync
1647
  # to not buffer too much memory; this means that at best, we flush
1648
  # every 64k, which will not be very fast
1649
  destcmd = utils.BuildShellCmd("dd of=%s conv=nocreat,notrunc bs=65536"
1650
                                " oflag=dsync", dest_path)
1651

    
1652
  remotecmd = _GetSshRunner(cluster_name).BuildCmd(dest_node,
1653
                                                   constants.GANETI_RUNAS,
1654
                                                   destcmd)
1655

    
1656
  # all commands have been checked, so we're safe to combine them
1657
  command = '|'.join([expcmd, utils.ShellQuoteArgs(remotecmd)])
1658

    
1659
  result = utils.RunCmd(["bash", "-c", command])
1660

    
1661
  if result.failed:
1662
    _Fail("Disk copy command '%s' returned error: %s"
1663
          " output: %s", command, result.fail_reason, result.output)
1664

    
1665

    
1666
def UploadFile(file_name, data, mode, uid, gid, atime, mtime):
1667
  """Write a file to the filesystem.
1668

1669
  This allows the master to overwrite(!) a file. It will only perform
1670
  the operation if the file belongs to a list of configuration files.
1671

1672
  @type file_name: str
1673
  @param file_name: the target file name
1674
  @type data: str
1675
  @param data: the new contents of the file
1676
  @type mode: int
1677
  @param mode: the mode to give the file (can be None)
1678
  @type uid: int
1679
  @param uid: the owner of the file (can be -1 for default)
1680
  @type gid: int
1681
  @param gid: the group of the file (can be -1 for default)
1682
  @type atime: float
1683
  @param atime: the atime to set on the file (can be None)
1684
  @type mtime: float
1685
  @param mtime: the mtime to set on the file (can be None)
1686
  @rtype: None
1687

1688
  """
1689
  if not os.path.isabs(file_name):
1690
    _Fail("Filename passed to UploadFile is not absolute: '%s'", file_name)
1691

    
1692
  if file_name not in _ALLOWED_UPLOAD_FILES:
1693
    _Fail("Filename passed to UploadFile not in allowed upload targets: '%s'",
1694
          file_name)
1695

    
1696
  raw_data = _Decompress(data)
1697

    
1698
  utils.WriteFile(file_name, data=raw_data, mode=mode, uid=uid, gid=gid,
1699
                  atime=atime, mtime=mtime)
1700

    
1701

    
1702
def WriteSsconfFiles(values):
1703
  """Update all ssconf files.
1704

1705
  Wrapper around the SimpleStore.WriteFiles.
1706

1707
  """
1708
  ssconf.SimpleStore().WriteFiles(values)
1709

    
1710

    
1711
def _ErrnoOrStr(err):
1712
  """Format an EnvironmentError exception.
1713

1714
  If the L{err} argument has an errno attribute, it will be looked up
1715
  and converted into a textual C{E...} description. Otherwise the
1716
  string representation of the error will be returned.
1717

1718
  @type err: L{EnvironmentError}
1719
  @param err: the exception to format
1720

1721
  """
1722
  if hasattr(err, 'errno'):
1723
    detail = errno.errorcode[err.errno]
1724
  else:
1725
    detail = str(err)
1726
  return detail
1727

    
1728

    
1729
def _OSOndiskAPIVersion(os_dir):
1730
  """Compute and return the API version of a given OS.
1731

1732
  This function will try to read the API version of the OS residing in
1733
  the 'os_dir' directory.
1734

1735
  @type os_dir: str
1736
  @param os_dir: the directory in which we should look for the OS
1737
  @rtype: tuple
1738
  @return: tuple (status, data) with status denoting the validity and
1739
      data holding either the vaid versions or an error message
1740

1741
  """
1742
  api_file = utils.PathJoin(os_dir, constants.OS_API_FILE)
1743

    
1744
  try:
1745
    st = os.stat(api_file)
1746
  except EnvironmentError, err:
1747
    return False, ("Required file '%s' not found under path %s: %s" %
1748
                   (constants.OS_API_FILE, os_dir, _ErrnoOrStr(err)))
1749

    
1750
  if not stat.S_ISREG(stat.S_IFMT(st.st_mode)):
1751
    return False, ("File '%s' in %s is not a regular file" %
1752
                   (constants.OS_API_FILE, os_dir))
1753

    
1754
  try:
1755
    api_versions = utils.ReadFile(api_file).splitlines()
1756
  except EnvironmentError, err:
1757
    return False, ("Error while reading the API version file at %s: %s" %
1758
                   (api_file, _ErrnoOrStr(err)))
1759

    
1760
  try:
1761
    api_versions = [int(version.strip()) for version in api_versions]
1762
  except (TypeError, ValueError), err:
1763
    return False, ("API version(s) can't be converted to integer: %s" %
1764
                   str(err))
1765

    
1766
  return True, api_versions
1767

    
1768

    
1769
def DiagnoseOS(top_dirs=None):
1770
  """Compute the validity for all OSes.
1771

1772
  @type top_dirs: list
1773
  @param top_dirs: the list of directories in which to
1774
      search (if not given defaults to
1775
      L{constants.OS_SEARCH_PATH})
1776
  @rtype: list of L{objects.OS}
1777
  @return: a list of tuples (name, path, status, diagnose, variants)
1778
      for all (potential) OSes under all search paths, where:
1779
          - name is the (potential) OS name
1780
          - path is the full path to the OS
1781
          - status True/False is the validity of the OS
1782
          - diagnose is the error message for an invalid OS, otherwise empty
1783
          - variants is a list of supported OS variants, if any
1784

1785
  """
1786
  if top_dirs is None:
1787
    top_dirs = constants.OS_SEARCH_PATH
1788

    
1789
  result = []
1790
  for dir_name in top_dirs:
1791
    if os.path.isdir(dir_name):
1792
      try:
1793
        f_names = utils.ListVisibleFiles(dir_name)
1794
      except EnvironmentError, err:
1795
        logging.exception("Can't list the OS directory %s: %s", dir_name, err)
1796
        break
1797
      for name in f_names:
1798
        os_path = utils.PathJoin(dir_name, name)
1799
        status, os_inst = _TryOSFromDisk(name, base_dir=dir_name)
1800
        if status:
1801
          diagnose = ""
1802
          variants = os_inst.supported_variants
1803
        else:
1804
          diagnose = os_inst
1805
          variants = []
1806
        result.append((name, os_path, status, diagnose, variants))
1807

    
1808
  return result
1809

    
1810

    
1811
def _TryOSFromDisk(name, base_dir=None):
1812
  """Create an OS instance from disk.
1813

1814
  This function will return an OS instance if the given name is a
1815
  valid OS name.
1816

1817
  @type base_dir: string
1818
  @keyword base_dir: Base directory containing OS installations.
1819
                     Defaults to a search in all the OS_SEARCH_PATH dirs.
1820
  @rtype: tuple
1821
  @return: success and either the OS instance if we find a valid one,
1822
      or error message
1823

1824
  """
1825
  if base_dir is None:
1826
    os_dir = utils.FindFile(name, constants.OS_SEARCH_PATH, os.path.isdir)
1827
  else:
1828
    os_dir = utils.FindFile(name, [base_dir], os.path.isdir)
1829

    
1830
  if os_dir is None:
1831
    return False, "Directory for OS %s not found in search path" % name
1832

    
1833
  status, api_versions = _OSOndiskAPIVersion(os_dir)
1834
  if not status:
1835
    # push the error up
1836
    return status, api_versions
1837

    
1838
  if not constants.OS_API_VERSIONS.intersection(api_versions):
1839
    return False, ("API version mismatch for path '%s': found %s, want %s." %
1840
                   (os_dir, api_versions, constants.OS_API_VERSIONS))
1841

    
1842
  # OS Files dictionary, we will populate it with the absolute path names
1843
  os_files = dict.fromkeys(constants.OS_SCRIPTS)
1844

    
1845
  if max(api_versions) >= constants.OS_API_V15:
1846
    os_files[constants.OS_VARIANTS_FILE] = ''
1847

    
1848
  for filename in os_files:
1849
    os_files[filename] = utils.PathJoin(os_dir, filename)
1850

    
1851
    try:
1852
      st = os.stat(os_files[filename])
1853
    except EnvironmentError, err:
1854
      return False, ("File '%s' under path '%s' is missing (%s)" %
1855
                     (filename, os_dir, _ErrnoOrStr(err)))
1856

    
1857
    if not stat.S_ISREG(stat.S_IFMT(st.st_mode)):
1858
      return False, ("File '%s' under path '%s' is not a regular file" %
1859
                     (filename, os_dir))
1860

    
1861
    if filename in constants.OS_SCRIPTS:
1862
      if stat.S_IMODE(st.st_mode) & stat.S_IXUSR != stat.S_IXUSR:
1863
        return False, ("File '%s' under path '%s' is not executable" %
1864
                       (filename, os_dir))
1865

    
1866
  variants = []
1867
  if constants.OS_VARIANTS_FILE in os_files:
1868
    variants_file = os_files[constants.OS_VARIANTS_FILE]
1869
    try:
1870
      variants = utils.ReadFile(variants_file).splitlines()
1871
    except EnvironmentError, err:
1872
      return False, ("Error while reading the OS variants file at %s: %s" %
1873
                     (variants_file, _ErrnoOrStr(err)))
1874
    if not variants:
1875
      return False, ("No supported os variant found")
1876

    
1877
  os_obj = objects.OS(name=name, path=os_dir,
1878
                      create_script=os_files[constants.OS_SCRIPT_CREATE],
1879
                      export_script=os_files[constants.OS_SCRIPT_EXPORT],
1880
                      import_script=os_files[constants.OS_SCRIPT_IMPORT],
1881
                      rename_script=os_files[constants.OS_SCRIPT_RENAME],
1882
                      supported_variants=variants,
1883
                      api_versions=api_versions)
1884
  return True, os_obj
1885

    
1886

    
1887
def OSFromDisk(name, base_dir=None):
1888
  """Create an OS instance from disk.
1889

1890
  This function will return an OS instance if the given name is a
1891
  valid OS name. Otherwise, it will raise an appropriate
1892
  L{RPCFail} exception, detailing why this is not a valid OS.
1893

1894
  This is just a wrapper over L{_TryOSFromDisk}, which doesn't raise
1895
  an exception but returns true/false status data.
1896

1897
  @type base_dir: string
1898
  @keyword base_dir: Base directory containing OS installations.
1899
                     Defaults to a search in all the OS_SEARCH_PATH dirs.
1900
  @rtype: L{objects.OS}
1901
  @return: the OS instance if we find a valid one
1902
  @raise RPCFail: if we don't find a valid OS
1903

1904
  """
1905
  name_only = name.split("+", 1)[0]
1906
  status, payload = _TryOSFromDisk(name_only, base_dir)
1907

    
1908
  if not status:
1909
    _Fail(payload)
1910

    
1911
  return payload
1912

    
1913

    
1914
def OSCoreEnv(inst_os, debug=0):
1915
  """Calculate the basic environment for an os script.
1916

1917
  @type inst_os: L{objects.OS}
1918
  @param inst_os: operating system for which the environment is being built
1919
  @type debug: integer
1920
  @param debug: debug level (0 or 1, for OS Api 10)
1921
  @rtype: dict
1922
  @return: dict of environment variables
1923
  @raise errors.BlockDeviceError: if the block device
1924
      cannot be found
1925

1926
  """
1927
  result = {}
1928
  api_version = \
1929
    max(constants.OS_API_VERSIONS.intersection(inst_os.api_versions))
1930
  result['OS_API_VERSION'] = '%d' % api_version
1931
  result['OS_NAME'] = inst_os.name
1932
  result['DEBUG_LEVEL'] = '%d' % debug
1933

    
1934
  # OS variants
1935
  if api_version >= constants.OS_API_V15:
1936
    try:
1937
      variant = inst_os.name.split('+', 1)[1]
1938
    except IndexError:
1939
      variant = inst_os.supported_variants[0]
1940
    result['OS_VARIANT'] = variant
1941

    
1942
  return result
1943

    
1944

    
1945
def OSEnvironment(instance, inst_os, debug=0):
1946
  """Calculate the environment for an os script.
1947

1948
  @type instance: L{objects.Instance}
1949
  @param instance: target instance for the os script run
1950
  @type inst_os: L{objects.OS}
1951
  @param inst_os: operating system for which the environment is being built
1952
  @type debug: integer
1953
  @param debug: debug level (0 or 1, for OS Api 10)
1954
  @rtype: dict
1955
  @return: dict of environment variables
1956
  @raise errors.BlockDeviceError: if the block device
1957
      cannot be found
1958

1959
  """
1960
  result = OSCoreEnv(inst_os, debug)
1961

    
1962
  result['INSTANCE_NAME'] = instance.name
1963
  result['INSTANCE_OS'] = instance.os
1964
  result['HYPERVISOR'] = instance.hypervisor
1965
  result['DISK_COUNT'] = '%d' % len(instance.disks)
1966
  result['NIC_COUNT'] = '%d' % len(instance.nics)
1967

    
1968
  # Disks
1969
  for idx, disk in enumerate(instance.disks):
1970
    real_disk = _OpenRealBD(disk)
1971
    result['DISK_%d_PATH' % idx] = real_disk.dev_path
1972
    result['DISK_%d_ACCESS' % idx] = disk.mode
1973
    if constants.HV_DISK_TYPE in instance.hvparams:
1974
      result['DISK_%d_FRONTEND_TYPE' % idx] = \
1975
        instance.hvparams[constants.HV_DISK_TYPE]
1976
    if disk.dev_type in constants.LDS_BLOCK:
1977
      result['DISK_%d_BACKEND_TYPE' % idx] = 'block'
1978
    elif disk.dev_type == constants.LD_FILE:
1979
      result['DISK_%d_BACKEND_TYPE' % idx] = \
1980
        'file:%s' % disk.physical_id[0]
1981

    
1982
  # NICs
1983
  for idx, nic in enumerate(instance.nics):
1984
    result['NIC_%d_MAC' % idx] = nic.mac
1985
    if nic.ip:
1986
      result['NIC_%d_IP' % idx] = nic.ip
1987
    result['NIC_%d_MODE' % idx] = nic.nicparams[constants.NIC_MODE]
1988
    if nic.nicparams[constants.NIC_MODE] == constants.NIC_MODE_BRIDGED:
1989
      result['NIC_%d_BRIDGE' % idx] = nic.nicparams[constants.NIC_LINK]
1990
    if nic.nicparams[constants.NIC_LINK]:
1991
      result['NIC_%d_LINK' % idx] = nic.nicparams[constants.NIC_LINK]
1992
    if constants.HV_NIC_TYPE in instance.hvparams:
1993
      result['NIC_%d_FRONTEND_TYPE' % idx] = \
1994
        instance.hvparams[constants.HV_NIC_TYPE]
1995

    
1996
  # HV/BE params
1997
  for source, kind in [(instance.beparams, "BE"), (instance.hvparams, "HV")]:
1998
    for key, value in source.items():
1999
      result["INSTANCE_%s_%s" % (kind, key)] = str(value)
2000

    
2001
  return result
2002

    
2003

    
2004
def BlockdevGrow(disk, amount):
2005
  """Grow a stack of block devices.
2006

2007
  This function is called recursively, with the childrens being the
2008
  first ones to resize.
2009

2010
  @type disk: L{objects.Disk}
2011
  @param disk: the disk to be grown
2012
  @rtype: (status, result)
2013
  @return: a tuple with the status of the operation
2014
      (True/False), and the errors message if status
2015
      is False
2016

2017
  """
2018
  r_dev = _RecursiveFindBD(disk)
2019
  if r_dev is None:
2020
    _Fail("Cannot find block device %s", disk)
2021

    
2022
  try:
2023
    r_dev.Grow(amount)
2024
  except errors.BlockDeviceError, err:
2025
    _Fail("Failed to grow block device: %s", err, exc=True)
2026

    
2027

    
2028
def BlockdevSnapshot(disk):
2029
  """Create a snapshot copy of a block device.
2030

2031
  This function is called recursively, and the snapshot is actually created
2032
  just for the leaf lvm backend device.
2033

2034
  @type disk: L{objects.Disk}
2035
  @param disk: the disk to be snapshotted
2036
  @rtype: string
2037
  @return: snapshot disk path
2038

2039
  """
2040
  if disk.dev_type == constants.LD_DRBD8:
2041
    if not disk.children:
2042
      _Fail("DRBD device '%s' without backing storage cannot be snapshotted",
2043
            disk.unique_id)
2044
    return BlockdevSnapshot(disk.children[0])
2045
  elif disk.dev_type == constants.LD_LV:
2046
    r_dev = _RecursiveFindBD(disk)
2047
    if r_dev is not None:
2048
      # FIXME: choose a saner value for the snapshot size
2049
      # let's stay on the safe side and ask for the full size, for now
2050
      return r_dev.Snapshot(disk.size)
2051
    else:
2052
      _Fail("Cannot find block device %s", disk)
2053
  else:
2054
    _Fail("Cannot snapshot non-lvm block device '%s' of type '%s'",
2055
          disk.unique_id, disk.dev_type)
2056

    
2057

    
2058
def FinalizeExport(instance, snap_disks):
2059
  """Write out the export configuration information.
2060

2061
  @type instance: L{objects.Instance}
2062
  @param instance: the instance which we export, used for
2063
      saving configuration
2064
  @type snap_disks: list of L{objects.Disk}
2065
  @param snap_disks: list of snapshot block devices, which
2066
      will be used to get the actual name of the dump file
2067

2068
  @rtype: None
2069

2070
  """
2071
  destdir = utils.PathJoin(constants.EXPORT_DIR, instance.name + ".new")
2072
  finaldestdir = utils.PathJoin(constants.EXPORT_DIR, instance.name)
2073

    
2074
  config = objects.SerializableConfigParser()
2075

    
2076
  config.add_section(constants.INISECT_EXP)
2077
  config.set(constants.INISECT_EXP, 'version', '0')
2078
  config.set(constants.INISECT_EXP, 'timestamp', '%d' % int(time.time()))
2079
  config.set(constants.INISECT_EXP, 'source', instance.primary_node)
2080
  config.set(constants.INISECT_EXP, 'os', instance.os)
2081
  config.set(constants.INISECT_EXP, 'compression', 'gzip')
2082

    
2083
  config.add_section(constants.INISECT_INS)
2084
  config.set(constants.INISECT_INS, 'name', instance.name)
2085
  config.set(constants.INISECT_INS, 'memory', '%d' %
2086
             instance.beparams[constants.BE_MEMORY])
2087
  config.set(constants.INISECT_INS, 'vcpus', '%d' %
2088
             instance.beparams[constants.BE_VCPUS])
2089
  config.set(constants.INISECT_INS, 'disk_template', instance.disk_template)
2090
  config.set(constants.INISECT_INS, 'hypervisor', instance.hypervisor)
2091

    
2092
  nic_total = 0
2093
  for nic_count, nic in enumerate(instance.nics):
2094
    nic_total += 1
2095
    config.set(constants.INISECT_INS, 'nic%d_mac' %
2096
               nic_count, '%s' % nic.mac)
2097
    config.set(constants.INISECT_INS, 'nic%d_ip' % nic_count, '%s' % nic.ip)
2098
    for param in constants.NICS_PARAMETER_TYPES:
2099
      config.set(constants.INISECT_INS, 'nic%d_%s' % (nic_count, param),
2100
                 '%s' % nic.nicparams.get(param, None))
2101
  # TODO: redundant: on load can read nics until it doesn't exist
2102
  config.set(constants.INISECT_INS, 'nic_count' , '%d' % nic_total)
2103

    
2104
  disk_total = 0
2105
  for disk_count, disk in enumerate(snap_disks):
2106
    if disk:
2107
      disk_total += 1
2108
      config.set(constants.INISECT_INS, 'disk%d_ivname' % disk_count,
2109
                 ('%s' % disk.iv_name))
2110
      config.set(constants.INISECT_INS, 'disk%d_dump' % disk_count,
2111
                 ('%s' % disk.physical_id[1]))
2112
      config.set(constants.INISECT_INS, 'disk%d_size' % disk_count,
2113
                 ('%d' % disk.size))
2114

    
2115
  config.set(constants.INISECT_INS, 'disk_count' , '%d' % disk_total)
2116

    
2117
  # New-style hypervisor/backend parameters
2118

    
2119
  config.add_section(constants.INISECT_HYP)
2120
  for name, value in instance.hvparams.items():
2121
    if name not in constants.HVC_GLOBALS:
2122
      config.set(constants.INISECT_HYP, name, str(value))
2123

    
2124
  config.add_section(constants.INISECT_BEP)
2125
  for name, value in instance.beparams.items():
2126
    config.set(constants.INISECT_BEP, name, str(value))
2127

    
2128
  utils.WriteFile(utils.PathJoin(destdir, constants.EXPORT_CONF_FILE),
2129
                  data=config.Dumps())
2130
  shutil.rmtree(finaldestdir, ignore_errors=True)
2131
  shutil.move(destdir, finaldestdir)
2132

    
2133

    
2134
def ExportInfo(dest):
2135
  """Get export configuration information.
2136

2137
  @type dest: str
2138
  @param dest: directory containing the export
2139

2140
  @rtype: L{objects.SerializableConfigParser}
2141
  @return: a serializable config file containing the
2142
      export info
2143

2144
  """
2145
  cff = utils.PathJoin(dest, constants.EXPORT_CONF_FILE)
2146

    
2147
  config = objects.SerializableConfigParser()
2148
  config.read(cff)
2149

    
2150
  if (not config.has_section(constants.INISECT_EXP) or
2151
      not config.has_section(constants.INISECT_INS)):
2152
    _Fail("Export info file doesn't have the required fields")
2153

    
2154
  return config.Dumps()
2155

    
2156

    
2157
def ListExports():
2158
  """Return a list of exports currently available on this machine.
2159

2160
  @rtype: list
2161
  @return: list of the exports
2162

2163
  """
2164
  if os.path.isdir(constants.EXPORT_DIR):
2165
    return sorted(utils.ListVisibleFiles(constants.EXPORT_DIR))
2166
  else:
2167
    _Fail("No exports directory")
2168

    
2169

    
2170
def RemoveExport(export):
2171
  """Remove an existing export from the node.
2172

2173
  @type export: str
2174
  @param export: the name of the export to remove
2175
  @rtype: None
2176

2177
  """
2178
  target = utils.PathJoin(constants.EXPORT_DIR, export)
2179

    
2180
  try:
2181
    shutil.rmtree(target)
2182
  except EnvironmentError, err:
2183
    _Fail("Error while removing the export: %s", err, exc=True)
2184

    
2185

    
2186
def BlockdevRename(devlist):
2187
  """Rename a list of block devices.
2188

2189
  @type devlist: list of tuples
2190
  @param devlist: list of tuples of the form  (disk,
2191
      new_logical_id, new_physical_id); disk is an
2192
      L{objects.Disk} object describing the current disk,
2193
      and new logical_id/physical_id is the name we
2194
      rename it to
2195
  @rtype: boolean
2196
  @return: True if all renames succeeded, False otherwise
2197

2198
  """
2199
  msgs = []
2200
  result = True
2201
  for disk, unique_id in devlist:
2202
    dev = _RecursiveFindBD(disk)
2203
    if dev is None:
2204
      msgs.append("Can't find device %s in rename" % str(disk))
2205
      result = False
2206
      continue
2207
    try:
2208
      old_rpath = dev.dev_path
2209
      dev.Rename(unique_id)
2210
      new_rpath = dev.dev_path
2211
      if old_rpath != new_rpath:
2212
        DevCacheManager.RemoveCache(old_rpath)
2213
        # FIXME: we should add the new cache information here, like:
2214
        # DevCacheManager.UpdateCache(new_rpath, owner, ...)
2215
        # but we don't have the owner here - maybe parse from existing
2216
        # cache? for now, we only lose lvm data when we rename, which
2217
        # is less critical than DRBD or MD
2218
    except errors.BlockDeviceError, err:
2219
      msgs.append("Can't rename device '%s' to '%s': %s" %
2220
                  (dev, unique_id, err))
2221
      logging.exception("Can't rename device '%s' to '%s'", dev, unique_id)
2222
      result = False
2223
  if not result:
2224
    _Fail("; ".join(msgs))
2225

    
2226

    
2227
def _TransformFileStorageDir(file_storage_dir):
2228
  """Checks whether given file_storage_dir is valid.
2229

2230
  Checks wheter the given file_storage_dir is within the cluster-wide
2231
  default file_storage_dir stored in SimpleStore. Only paths under that
2232
  directory are allowed.
2233

2234
  @type file_storage_dir: str
2235
  @param file_storage_dir: the path to check
2236

2237
  @return: the normalized path if valid, None otherwise
2238

2239
  """
2240
  if not constants.ENABLE_FILE_STORAGE:
2241
    _Fail("File storage disabled at configure time")
2242
  cfg = _GetConfig()
2243
  file_storage_dir = os.path.normpath(file_storage_dir)
2244
  base_file_storage_dir = cfg.GetFileStorageDir()
2245
  if (os.path.commonprefix([file_storage_dir, base_file_storage_dir]) !=
2246
      base_file_storage_dir):
2247
    _Fail("File storage directory '%s' is not under base file"
2248
          " storage directory '%s'", file_storage_dir, base_file_storage_dir)
2249
  return file_storage_dir
2250

    
2251

    
2252
def CreateFileStorageDir(file_storage_dir):
2253
  """Create file storage directory.
2254

2255
  @type file_storage_dir: str
2256
  @param file_storage_dir: directory to create
2257

2258
  @rtype: tuple
2259
  @return: tuple with first element a boolean indicating wheter dir
2260
      creation was successful or not
2261

2262
  """
2263
  file_storage_dir = _TransformFileStorageDir(file_storage_dir)
2264
  if os.path.exists(file_storage_dir):
2265
    if not os.path.isdir(file_storage_dir):
2266
      _Fail("Specified storage dir '%s' is not a directory",
2267
            file_storage_dir)
2268
  else:
2269
    try:
2270
      os.makedirs(file_storage_dir, 0750)
2271
    except OSError, err:
2272
      _Fail("Cannot create file storage directory '%s': %s",
2273
            file_storage_dir, err, exc=True)
2274

    
2275

    
2276
def RemoveFileStorageDir(file_storage_dir):
2277
  """Remove file storage directory.
2278

2279
  Remove it only if it's empty. If not log an error and return.
2280

2281
  @type file_storage_dir: str
2282
  @param file_storage_dir: the directory we should cleanup
2283
  @rtype: tuple (success,)
2284
  @return: tuple of one element, C{success}, denoting
2285
      whether the operation was successful
2286

2287
  """
2288
  file_storage_dir = _TransformFileStorageDir(file_storage_dir)
2289
  if os.path.exists(file_storage_dir):
2290
    if not os.path.isdir(file_storage_dir):
2291
      _Fail("Specified Storage directory '%s' is not a directory",
2292
            file_storage_dir)
2293
    # deletes dir only if empty, otherwise we want to fail the rpc call
2294
    try:
2295
      os.rmdir(file_storage_dir)
2296
    except OSError, err:
2297
      _Fail("Cannot remove file storage directory '%s': %s",
2298
            file_storage_dir, err)
2299

    
2300

    
2301
def RenameFileStorageDir(old_file_storage_dir, new_file_storage_dir):
2302
  """Rename the file storage directory.
2303

2304
  @type old_file_storage_dir: str
2305
  @param old_file_storage_dir: the current path
2306
  @type new_file_storage_dir: str
2307
  @param new_file_storage_dir: the name we should rename to
2308
  @rtype: tuple (success,)
2309
  @return: tuple of one element, C{success}, denoting
2310
      whether the operation was successful
2311

2312
  """
2313
  old_file_storage_dir = _TransformFileStorageDir(old_file_storage_dir)
2314
  new_file_storage_dir = _TransformFileStorageDir(new_file_storage_dir)
2315
  if not os.path.exists(new_file_storage_dir):
2316
    if os.path.isdir(old_file_storage_dir):
2317
      try:
2318
        os.rename(old_file_storage_dir, new_file_storage_dir)
2319
      except OSError, err:
2320
        _Fail("Cannot rename '%s' to '%s': %s",
2321
              old_file_storage_dir, new_file_storage_dir, err)
2322
    else:
2323
      _Fail("Specified storage dir '%s' is not a directory",
2324
            old_file_storage_dir)
2325
  else:
2326
    if os.path.exists(old_file_storage_dir):
2327
      _Fail("Cannot rename '%s' to '%s': both locations exist",
2328
            old_file_storage_dir, new_file_storage_dir)
2329

    
2330

    
2331
def _EnsureJobQueueFile(file_name):
2332
  """Checks whether the given filename is in the queue directory.
2333

2334
  @type file_name: str
2335
  @param file_name: the file name we should check
2336
  @rtype: None
2337
  @raises RPCFail: if the file is not valid
2338

2339
  """
2340
  queue_dir = os.path.normpath(constants.QUEUE_DIR)
2341
  result = (os.path.commonprefix([queue_dir, file_name]) == queue_dir)
2342

    
2343
  if not result:
2344
    _Fail("Passed job queue file '%s' does not belong to"
2345
          " the queue directory '%s'", file_name, queue_dir)
2346

    
2347

    
2348
def JobQueueUpdate(file_name, content):
2349
  """Updates a file in the queue directory.
2350

2351
  This is just a wrapper over L{utils.WriteFile}, with proper
2352
  checking.
2353

2354
  @type file_name: str
2355
  @param file_name: the job file name
2356
  @type content: str
2357
  @param content: the new job contents
2358
  @rtype: boolean
2359
  @return: the success of the operation
2360

2361
  """
2362
  _EnsureJobQueueFile(file_name)
2363

    
2364
  # Write and replace the file atomically
2365
  utils.WriteFile(file_name, data=_Decompress(content))
2366

    
2367

    
2368
def JobQueueRename(old, new):
2369
  """Renames a job queue file.
2370

2371
  This is just a wrapper over os.rename with proper checking.
2372

2373
  @type old: str
2374
  @param old: the old (actual) file name
2375
  @type new: str
2376
  @param new: the desired file name
2377
  @rtype: tuple
2378
  @return: the success of the operation and payload
2379

2380
  """
2381
  _EnsureJobQueueFile(old)
2382
  _EnsureJobQueueFile(new)
2383

    
2384
  utils.RenameFile(old, new, mkdir=True)
2385

    
2386

    
2387
def BlockdevClose(instance_name, disks):
2388
  """Closes the given block devices.
2389

2390
  This means they will be switched to secondary mode (in case of
2391
  DRBD).
2392

2393
  @param instance_name: if the argument is not empty, the symlinks
2394
      of this instance will be removed
2395
  @type disks: list of L{objects.Disk}
2396
  @param disks: the list of disks to be closed
2397
  @rtype: tuple (success, message)
2398
  @return: a tuple of success and message, where success
2399
      indicates the succes of the operation, and message
2400
      which will contain the error details in case we
2401
      failed
2402

2403
  """
2404
  bdevs = []
2405
  for cf in disks:
2406
    rd = _RecursiveFindBD(cf)
2407
    if rd is None:
2408
      _Fail("Can't find device %s", cf)
2409
    bdevs.append(rd)
2410

    
2411
  msg = []
2412
  for rd in bdevs:
2413
    try:
2414
      rd.Close()
2415
    except errors.BlockDeviceError, err:
2416
      msg.append(str(err))
2417
  if msg:
2418
    _Fail("Can't make devices secondary: %s", ",".join(msg))
2419
  else:
2420
    if instance_name:
2421
      _RemoveBlockDevLinks(instance_name, disks)
2422

    
2423

    
2424
def ValidateHVParams(hvname, hvparams):
2425
  """Validates the given hypervisor parameters.
2426

2427
  @type hvname: string
2428
  @param hvname: the hypervisor name
2429
  @type hvparams: dict
2430
  @param hvparams: the hypervisor parameters to be validated
2431
  @rtype: None
2432

2433
  """
2434
  try:
2435
    hv_type = hypervisor.GetHypervisor(hvname)
2436
    hv_type.ValidateParameters(hvparams)
2437
  except errors.HypervisorError, err:
2438
    _Fail(str(err), log=False)
2439

    
2440

    
2441
def DemoteFromMC():
2442
  """Demotes the current node from master candidate role.
2443

2444
  """
2445
  # try to ensure we're not the master by mistake
2446
  master, myself = ssconf.GetMasterAndMyself()
2447
  if master == myself:
2448
    _Fail("ssconf status shows I'm the master node, will not demote")
2449

    
2450
  result = utils.RunCmd([constants.DAEMON_UTIL, "check", constants.MASTERD])
2451
  if not result.failed:
2452
    _Fail("The master daemon is running, will not demote")
2453

    
2454
  try:
2455
    if os.path.isfile(constants.CLUSTER_CONF_FILE):
2456
      utils.CreateBackup(constants.CLUSTER_CONF_FILE)
2457
  except EnvironmentError, err:
2458
    if err.errno != errno.ENOENT:
2459
      _Fail("Error while backing up cluster file: %s", err, exc=True)
2460

    
2461
  utils.RemoveFile(constants.CLUSTER_CONF_FILE)
2462

    
2463

    
2464
def _GetX509Filenames(cryptodir, name):
2465
  """Returns the full paths for the private key and certificate.
2466

2467
  """
2468
  return (utils.PathJoin(cryptodir, name),
2469
          utils.PathJoin(cryptodir, name, _X509_KEY_FILE),
2470
          utils.PathJoin(cryptodir, name, _X509_CERT_FILE))
2471

    
2472

    
2473
def CreateX509Certificate(validity, cryptodir=constants.CRYPTO_KEYS_DIR):
2474
  """Creates a new X509 certificate for SSL/TLS.
2475

2476
  @type validity: int
2477
  @param validity: Validity in seconds
2478
  @rtype: tuple; (string, string)
2479
  @return: Certificate name and public part
2480

2481
  """
2482
  (key_pem, cert_pem) = \
2483
    utils.GenerateSelfSignedX509Cert(utils.HostInfo.SysName(),
2484
                                     min(validity, _MAX_SSL_CERT_VALIDITY))
2485

    
2486
  cert_dir = tempfile.mkdtemp(dir=cryptodir,
2487
                              prefix="x509-%s-" % utils.TimestampForFilename())
2488
  try:
2489
    name = os.path.basename(cert_dir)
2490
    assert len(name) > 5
2491

    
2492
    (_, key_file, cert_file) = _GetX509Filenames(cryptodir, name)
2493

    
2494
    utils.WriteFile(key_file, mode=0400, data=key_pem)
2495
    utils.WriteFile(cert_file, mode=0400, data=cert_pem)
2496

    
2497
    # Never return private key as it shouldn't leave the node
2498
    return (name, cert_pem)
2499
  except Exception:
2500
    shutil.rmtree(cert_dir, ignore_errors=True)
2501
    raise
2502

    
2503

    
2504
def RemoveX509Certificate(name, cryptodir=constants.CRYPTO_KEYS_DIR):
2505
  """Removes a X509 certificate.
2506

2507
  @type name: string
2508
  @param name: Certificate name
2509

2510
  """
2511
  (cert_dir, key_file, cert_file) = _GetX509Filenames(cryptodir, name)
2512

    
2513
  utils.RemoveFile(key_file)
2514
  utils.RemoveFile(cert_file)
2515

    
2516
  try:
2517
    os.rmdir(cert_dir)
2518
  except EnvironmentError, err:
2519
    _Fail("Cannot remove certificate directory '%s': %s",
2520
          cert_dir, err)
2521

    
2522

    
2523
def _GetImportExportIoCommand(instance, mode, ieio, ieargs):
2524
  """Returns the command for the requested input/output.
2525

2526
  @type instance: L{objects.Instance}
2527
  @param instance: The instance object
2528
  @param mode: Import/export mode
2529
  @param ieio: Input/output type
2530
  @param ieargs: Input/output arguments
2531

2532
  """
2533
  assert mode in (constants.IEM_IMPORT, constants.IEM_EXPORT)
2534

    
2535
  env = None
2536
  prefix = None
2537
  suffix = None
2538
  exp_size = None
2539

    
2540
  if ieio == constants.IEIO_FILE:
2541
    (filename, ) = ieargs
2542

    
2543
    if not utils.IsNormAbsPath(filename):
2544
      _Fail("Path '%s' is not normalized or absolute", filename)
2545

    
2546
    directory = os.path.normpath(os.path.dirname(filename))
2547

    
2548
    if (os.path.commonprefix([constants.EXPORT_DIR, directory]) !=
2549
        constants.EXPORT_DIR):
2550
      _Fail("File '%s' is not under exports directory '%s'",
2551
            filename, constants.EXPORT_DIR)
2552

    
2553
    # Create directory
2554
    utils.Makedirs(directory, mode=0750)
2555

    
2556
    quoted_filename = utils.ShellQuote(filename)
2557

    
2558
    if mode == constants.IEM_IMPORT:
2559
      suffix = "> %s" % quoted_filename
2560
    elif mode == constants.IEM_EXPORT:
2561
      suffix = "< %s" % quoted_filename
2562

    
2563
      # Retrieve file size
2564
      try:
2565
        st = os.stat(filename)
2566
      except EnvironmentError, err:
2567
        logging.error("Can't stat(2) %s: %s", filename, err)
2568
      else:
2569
        exp_size = utils.BytesToMebibyte(st.st_size)
2570

    
2571
  elif ieio == constants.IEIO_RAW_DISK:
2572
    (disk, ) = ieargs
2573

    
2574
    real_disk = _OpenRealBD(disk)
2575

    
2576
    if mode == constants.IEM_IMPORT:
2577
      # we set here a smaller block size as, due to transport buffering, more
2578
      # than 64-128k will mostly ignored; we use nocreat to fail if the device
2579
      # is not already there or we pass a wrong path; we use notrunc to no
2580
      # attempt truncate on an LV device; we use oflag=dsync to not buffer too
2581
      # much memory; this means that at best, we flush every 64k, which will
2582
      # not be very fast
2583
      suffix = utils.BuildShellCmd(("| dd of=%s conv=nocreat,notrunc"
2584
                                    " bs=%s oflag=dsync"),
2585
                                    real_disk.dev_path,
2586
                                    str(64 * 1024))
2587

    
2588
    elif mode == constants.IEM_EXPORT:
2589
      # the block size on the read dd is 1MiB to match our units
2590
      prefix = utils.BuildShellCmd("dd if=%s bs=%s count=%s |",
2591
                                   real_disk.dev_path,
2592
                                   str(1024 * 1024), # 1 MB
2593
                                   str(disk.size))
2594
      exp_size = disk.size
2595

    
2596
  elif ieio == constants.IEIO_SCRIPT:
2597
    (disk, disk_index, ) = ieargs
2598

    
2599
    assert isinstance(disk_index, (int, long))
2600

    
2601
    real_disk = _OpenRealBD(disk)
2602

    
2603
    inst_os = OSFromDisk(instance.os)
2604
    env = OSEnvironment(instance, inst_os)
2605

    
2606
    if mode == constants.IEM_IMPORT:
2607
      env["IMPORT_DEVICE"] = env["DISK_%d_PATH" % disk_index]
2608
      env["IMPORT_INDEX"] = str(disk_index)
2609
      script = inst_os.import_script
2610

    
2611
    elif mode == constants.IEM_EXPORT:
2612
      env["EXPORT_DEVICE"] = real_disk.dev_path
2613
      env["EXPORT_INDEX"] = str(disk_index)
2614
      script = inst_os.export_script
2615

    
2616
    # TODO: Pass special environment only to script
2617
    script_cmd = utils.BuildShellCmd("( cd %s && %s; )", inst_os.path, script)
2618

    
2619
    if mode == constants.IEM_IMPORT:
2620
      suffix = "| %s" % script_cmd
2621

    
2622
    elif mode == constants.IEM_EXPORT:
2623
      prefix = "%s |" % script_cmd
2624

    
2625
    # Let script predict size
2626
    exp_size = constants.IE_CUSTOM_SIZE
2627

    
2628
  else:
2629
    _Fail("Invalid %s I/O mode %r", mode, ieio)
2630

    
2631
  return (env, prefix, suffix, exp_size)
2632

    
2633

    
2634
def _CreateImportExportStatusDir(prefix):
2635
  """Creates status directory for import/export.
2636

2637
  """
2638
  return tempfile.mkdtemp(dir=constants.IMPORT_EXPORT_DIR,
2639
                          prefix=("%s-%s-" %
2640
                                  (prefix, utils.TimestampForFilename())))
2641

    
2642

    
2643
def StartImportExportDaemon(mode, opts, host, port, instance, ieio, ieioargs):
2644
  """Starts an import or export daemon.
2645

2646
  @param mode: Import/output mode
2647
  @type opts: L{objects.ImportExportOptions}
2648
  @param opts: Daemon options
2649
  @type host: string
2650
  @param host: Remote host for export (None for import)
2651
  @type port: int
2652
  @param port: Remote port for export (None for import)
2653
  @type instance: L{objects.Instance}
2654
  @param instance: Instance object
2655
  @param ieio: Input/output type
2656
  @param ieioargs: Input/output arguments
2657

2658
  """
2659
  if mode == constants.IEM_IMPORT:
2660
    prefix = "import"
2661

    
2662
    if not (host is None and port is None):
2663
      _Fail("Can not specify host or port on import")
2664

    
2665
  elif mode == constants.IEM_EXPORT:
2666
    prefix = "export"
2667

    
2668
    if host is None or port is None:
2669
      _Fail("Host and port must be specified for an export")
2670

    
2671
  else:
2672
    _Fail("Invalid mode %r", mode)
2673

    
2674
  if (opts.key_name is None) ^ (opts.ca_pem is None):
2675
    _Fail("Cluster certificate can only be used for both key and CA")
2676

    
2677
  (cmd_env, cmd_prefix, cmd_suffix, exp_size) = \
2678
    _GetImportExportIoCommand(instance, mode, ieio, ieioargs)
2679

    
2680
  if opts.key_name is None:
2681
    # Use server.pem
2682
    key_path = constants.NODED_CERT_FILE
2683
    cert_path = constants.NODED_CERT_FILE
2684
    assert opts.ca_pem is None
2685
  else:
2686
    (_, key_path, cert_path) = _GetX509Filenames(constants.CRYPTO_KEYS_DIR,
2687
                                                 opts.key_name)
2688
    assert opts.ca_pem is not None
2689

    
2690
  for i in [key_path, cert_path]:
2691
    if not os.path.exists(i):
2692
      _Fail("File '%s' does not exist" % i)
2693

    
2694
  status_dir = _CreateImportExportStatusDir(prefix)
2695
  try:
2696
    status_file = utils.PathJoin(status_dir, _IES_STATUS_FILE)
2697
    pid_file = utils.PathJoin(status_dir, _IES_PID_FILE)
2698
    ca_file = utils.PathJoin(status_dir, _IES_CA_FILE)
2699

    
2700
    if opts.ca_pem is None:
2701
      # Use server.pem
2702
      ca = utils.ReadFile(constants.NODED_CERT_FILE)
2703
    else:
2704
      ca = opts.ca_pem
2705

    
2706
    # Write CA file
2707
    utils.WriteFile(ca_file, data=ca, mode=0400)
2708

    
2709
    cmd = [
2710
      constants.IMPORT_EXPORT_DAEMON,
2711
      status_file, mode,
2712
      "--key=%s" % key_path,
2713
      "--cert=%s" % cert_path,
2714
      "--ca=%s" % ca_file,
2715
      ]
2716

    
2717
    if host:
2718
      cmd.append("--host=%s" % host)
2719

    
2720
    if port:
2721
      cmd.append("--port=%s" % port)
2722

    
2723
    if opts.compress:
2724
      cmd.append("--compress=%s" % opts.compress)
2725

    
2726
    if opts.magic:
2727
      cmd.append("--magic=%s" % opts.magic)
2728

    
2729
    if exp_size is not None:
2730
      cmd.append("--expected-size=%s" % exp_size)
2731

    
2732
    if cmd_prefix:
2733
      cmd.append("--cmd-prefix=%s" % cmd_prefix)
2734

    
2735
    if cmd_suffix:
2736
      cmd.append("--cmd-suffix=%s" % cmd_suffix)
2737

    
2738
    logfile = _InstanceLogName(prefix, instance.os, instance.name)
2739

    
2740
    # TODO: Once _InstanceLogName uses tempfile.mkstemp, StartDaemon has
2741
    # support for receiving a file descriptor for output
2742
    utils.StartDaemon(cmd, env=cmd_env, pidfile=pid_file,
2743
                      output=logfile)
2744

    
2745
    # The import/export name is simply the status directory name
2746
    return os.path.basename(status_dir)
2747

    
2748
  except Exception:
2749
    shutil.rmtree(status_dir, ignore_errors=True)
2750
    raise
2751

    
2752

    
2753
def GetImportExportStatus(names):
2754
  """Returns import/export daemon status.
2755

2756
  @type names: sequence
2757
  @param names: List of names
2758
  @rtype: List of dicts
2759
  @return: Returns a list of the state of each named import/export or None if a
2760
           status couldn't be read
2761

2762
  """
2763
  result = []
2764

    
2765
  for name in names:
2766
    status_file = utils.PathJoin(constants.IMPORT_EXPORT_DIR, name,
2767
                                 _IES_STATUS_FILE)
2768

    
2769
    try:
2770
      data = utils.ReadFile(status_file)
2771
    except EnvironmentError, err:
2772
      if err.errno != errno.ENOENT:
2773
        raise
2774
      data = None
2775

    
2776
    if not data:
2777
      result.append(None)
2778
      continue
2779

    
2780
    result.append(serializer.LoadJson(data))
2781

    
2782
  return result
2783

    
2784

    
2785
def AbortImportExport(name):
2786
  """Sends SIGTERM to a running import/export daemon.
2787

2788
  """
2789
  logging.info("Abort import/export %s", name)
2790

    
2791
  status_dir = utils.PathJoin(constants.IMPORT_EXPORT_DIR, name)
2792
  pid = utils.ReadLockedPidFile(utils.PathJoin(status_dir, _IES_PID_FILE))
2793

    
2794
  if pid:
2795
    logging.info("Import/export %s is running with PID %s, sending SIGTERM",
2796
                 name, pid)
2797
    utils.IgnoreProcessNotFound(os.kill, pid, signal.SIGTERM)
2798

    
2799

    
2800
def CleanupImportExport(name):
2801
  """Cleanup after an import or export.
2802

2803
  If the import/export daemon is still running it's killed. Afterwards the
2804
  whole status directory is removed.
2805

2806
  """
2807
  logging.info("Finalizing import/export %s", name)
2808

    
2809
  status_dir = utils.PathJoin(constants.IMPORT_EXPORT_DIR, name)
2810

    
2811
  pid = utils.ReadLockedPidFile(utils.PathJoin(status_dir, _IES_PID_FILE))
2812

    
2813
  if pid:
2814
    logging.info("Import/export %s is still running with PID %s",
2815
                 name, pid)
2816
    utils.KillProcess(pid, waitpid=False)
2817

    
2818
  shutil.rmtree(status_dir, ignore_errors=True)
2819

    
2820

    
2821
def _FindDisks(nodes_ip, disks):
2822
  """Sets the physical ID on disks and returns the block devices.
2823

2824
  """
2825
  # set the correct physical ID
2826
  my_name = utils.HostInfo().name
2827
  for cf in disks:
2828
    cf.SetPhysicalID(my_name, nodes_ip)
2829

    
2830
  bdevs = []
2831

    
2832
  for cf in disks:
2833
    rd = _RecursiveFindBD(cf)
2834
    if rd is None:
2835
      _Fail("Can't find device %s", cf)
2836
    bdevs.append(rd)
2837
  return bdevs
2838

    
2839

    
2840
def DrbdDisconnectNet(nodes_ip, disks):
2841
  """Disconnects the network on a list of drbd devices.
2842

2843
  """
2844
  bdevs = _FindDisks(nodes_ip, disks)
2845

    
2846
  # disconnect disks
2847
  for rd in bdevs:
2848
    try:
2849
      rd.DisconnectNet()
2850
    except errors.BlockDeviceError, err:
2851
      _Fail("Can't change network configuration to standalone mode: %s",
2852
            err, exc=True)
2853

    
2854

    
2855
def DrbdAttachNet(nodes_ip, disks, instance_name, multimaster):
2856
  """Attaches the network on a list of drbd devices.
2857

2858
  """
2859
  bdevs = _FindDisks(nodes_ip, disks)
2860

    
2861
  if multimaster:
2862
    for idx, rd in enumerate(bdevs):
2863
      try:
2864
        _SymlinkBlockDev(instance_name, rd.dev_path, idx)
2865
      except EnvironmentError, err:
2866
        _Fail("Can't create symlink: %s", err)
2867
  # reconnect disks, switch to new master configuration and if
2868
  # needed primary mode
2869
  for rd in bdevs:
2870
    try:
2871
      rd.AttachNet(multimaster)
2872
    except errors.BlockDeviceError, err:
2873
      _Fail("Can't change network configuration: %s", err)
2874

    
2875
  # wait until the disks are connected; we need to retry the re-attach
2876
  # if the device becomes standalone, as this might happen if the one
2877
  # node disconnects and reconnects in a different mode before the
2878
  # other node reconnects; in this case, one or both of the nodes will
2879
  # decide it has wrong configuration and switch to standalone
2880

    
2881
  def _Attach():
2882
    all_connected = True
2883

    
2884
    for rd in bdevs:
2885
      stats = rd.GetProcStatus()
2886

    
2887
      all_connected = (all_connected and
2888
                       (stats.is_connected or stats.is_in_resync))
2889

    
2890
      if stats.is_standalone:
2891
        # peer had different config info and this node became
2892
        # standalone, even though this should not happen with the
2893
        # new staged way of changing disk configs
2894
        try:
2895
          rd.AttachNet(multimaster)
2896
        except errors.BlockDeviceError, err:
2897
          _Fail("Can't change network configuration: %s", err)
2898

    
2899
    if not all_connected:
2900
      raise utils.RetryAgain()
2901

    
2902
  try:
2903
    # Start with a delay of 100 miliseconds and go up to 5 seconds
2904
    utils.Retry(_Attach, (0.1, 1.5, 5.0), 2 * 60)
2905
  except utils.RetryTimeout:
2906
    _Fail("Timeout in disk reconnecting")
2907

    
2908
  if multimaster:
2909
    # change to primary mode
2910
    for rd in bdevs:
2911
      try:
2912
        rd.Open()
2913
      except errors.BlockDeviceError, err:
2914
        _Fail("Can't change to primary mode: %s", err)
2915

    
2916

    
2917
def DrbdWaitSync(nodes_ip, disks):
2918
  """Wait until DRBDs have synchronized.
2919

2920
  """
2921
  def _helper(rd):
2922
    stats = rd.GetProcStatus()
2923
    if not (stats.is_connected or stats.is_in_resync):
2924
      raise utils.RetryAgain()
2925
    return stats
2926

    
2927
  bdevs = _FindDisks(nodes_ip, disks)
2928

    
2929
  min_resync = 100
2930
  alldone = True
2931
  for rd in bdevs:
2932
    try:
2933
      # poll each second for 15 seconds
2934
      stats = utils.Retry(_helper, 1, 15, args=[rd])
2935
    except utils.RetryTimeout:
2936
      stats = rd.GetProcStatus()
2937
      # last check
2938
      if not (stats.is_connected or stats.is_in_resync):
2939
        _Fail("DRBD device %s is not in sync: stats=%s", rd, stats)
2940
    alldone = alldone and (not stats.is_in_resync)
2941
    if stats.sync_percent is not None:
2942
      min_resync = min(min_resync, stats.sync_percent)
2943

    
2944
  return (alldone, min_resync)
2945

    
2946

    
2947
def PowercycleNode(hypervisor_type):
2948
  """Hard-powercycle the node.
2949

2950
  Because we need to return first, and schedule the powercycle in the
2951
  background, we won't be able to report failures nicely.
2952

2953
  """
2954
  hyper = hypervisor.GetHypervisor(hypervisor_type)
2955
  try:
2956
    pid = os.fork()
2957
  except OSError:
2958
    # if we can't fork, we'll pretend that we're in the child process
2959
    pid = 0
2960
  if pid > 0:
2961
    return "Reboot scheduled in 5 seconds"
2962
  # ensure the child is running on ram
2963
  try:
2964
    utils.Mlockall()
2965
  except Exception: # pylint: disable-msg=W0703
2966
    pass
2967
  time.sleep(5)
2968
  hyper.PowercycleNode()
2969

    
2970

    
2971
class HooksRunner(object):
2972
  """Hook runner.
2973

2974
  This class is instantiated on the node side (ganeti-noded) and not
2975
  on the master side.
2976

2977
  """
2978
  def __init__(self, hooks_base_dir=None):
2979
    """Constructor for hooks runner.
2980

2981
    @type hooks_base_dir: str or None
2982
    @param hooks_base_dir: if not None, this overrides the
2983
        L{constants.HOOKS_BASE_DIR} (useful for unittests)
2984

2985
    """
2986
    if hooks_base_dir is None:
2987
      hooks_base_dir = constants.HOOKS_BASE_DIR
2988
    # yeah, _BASE_DIR is not valid for attributes, we use it like a
2989
    # constant
2990
    self._BASE_DIR = hooks_base_dir # pylint: disable-msg=C0103
2991

    
2992
  def RunHooks(self, hpath, phase, env):
2993
    """Run the scripts in the hooks directory.
2994

2995
    @type hpath: str
2996
    @param hpath: the path to the hooks directory which
2997
        holds the scripts
2998
    @type phase: str
2999
    @param phase: either L{constants.HOOKS_PHASE_PRE} or
3000
        L{constants.HOOKS_PHASE_POST}
3001
    @type env: dict
3002
    @param env: dictionary with the environment for the hook
3003
    @rtype: list
3004
    @return: list of 3-element tuples:
3005
      - script path
3006
      - script result, either L{constants.HKR_SUCCESS} or
3007
        L{constants.HKR_FAIL}
3008
      - output of the script
3009

3010
    @raise errors.ProgrammerError: for invalid input
3011
        parameters
3012

3013
    """
3014
    if phase == constants.HOOKS_PHASE_PRE:
3015
      suffix = "pre"
3016
    elif phase == constants.HOOKS_PHASE_POST:
3017
      suffix = "post"
3018
    else:
3019
      _Fail("Unknown hooks phase '%s'", phase)
3020

    
3021

    
3022
    subdir = "%s-%s.d" % (hpath, suffix)
3023
    dir_name = utils.PathJoin(self._BASE_DIR, subdir)
3024

    
3025
    results = []
3026

    
3027
    if not os.path.isdir(dir_name):
3028
      # for non-existing/non-dirs, we simply exit instead of logging a
3029
      # warning at every operation
3030
      return results
3031

    
3032
    runparts_results = utils.RunParts(dir_name, env=env, reset_env=True)
3033

    
3034
    for (relname, relstatus, runresult)  in runparts_results:
3035
      if relstatus == constants.RUNPARTS_SKIP:
3036
        rrval = constants.HKR_SKIP
3037
        output = ""
3038
      elif relstatus == constants.RUNPARTS_ERR:
3039
        rrval = constants.HKR_FAIL
3040
        output = "Hook script execution error: %s" % runresult
3041
      elif relstatus == constants.RUNPARTS_RUN:
3042
        if runresult.failed:
3043
          rrval = constants.HKR_FAIL
3044
        else:
3045
          rrval = constants.HKR_SUCCESS
3046
        output = utils.SafeEncode(runresult.output.strip())
3047
      results.append(("%s/%s" % (subdir, relname), rrval, output))
3048

    
3049
    return results
3050

    
3051

    
3052
class IAllocatorRunner(object):
3053
  """IAllocator runner.
3054

3055
  This class is instantiated on the node side (ganeti-noded) and not on
3056
  the master side.
3057

3058
  """
3059
  @staticmethod
3060
  def Run(name, idata):
3061
    """Run an iallocator script.
3062

3063
    @type name: str
3064
    @param name: the iallocator script name
3065
    @type idata: str
3066
    @param idata: the allocator input data
3067

3068
    @rtype: tuple
3069
    @return: two element tuple of:
3070
       - status
3071
       - either error message or stdout of allocator (for success)
3072

3073
    """
3074
    alloc_script = utils.FindFile(name, constants.IALLOCATOR_SEARCH_PATH,
3075
                                  os.path.isfile)
3076
    if alloc_script is None:
3077
      _Fail("iallocator module '%s' not found in the search path", name)
3078

    
3079
    fd, fin_name = tempfile.mkstemp(prefix="ganeti-iallocator.")
3080
    try:
3081
      os.write(fd, idata)
3082
      os.close(fd)
3083
      result = utils.RunCmd([alloc_script, fin_name])
3084
      if result.failed:
3085
        _Fail("iallocator module '%s' failed: %s, output '%s'",
3086
              name, result.fail_reason, result.output)
3087
    finally:
3088
      os.unlink(fin_name)
3089

    
3090
    return result.stdout
3091

    
3092

    
3093
class DevCacheManager(object):
3094
  """Simple class for managing a cache of block device information.
3095

3096
  """
3097
  _DEV_PREFIX = "/dev/"
3098
  _ROOT_DIR = constants.BDEV_CACHE_DIR
3099

    
3100
  @classmethod
3101
  def _ConvertPath(cls, dev_path):
3102
    """Converts a /dev/name path to the cache file name.
3103

3104
    This replaces slashes with underscores and strips the /dev
3105
    prefix. It then returns the full path to the cache file.
3106

3107
    @type dev_path: str
3108
    @param dev_path: the C{/dev/} path name
3109
    @rtype: str
3110
    @return: the converted path name
3111

3112
    """
3113
    if dev_path.startswith(cls._DEV_PREFIX):
3114
      dev_path = dev_path[len(cls._DEV_PREFIX):]
3115
    dev_path = dev_path.replace("/", "_")
3116
    fpath = utils.PathJoin(cls._ROOT_DIR, "bdev_%s" % dev_path)
3117
    return fpath
3118

    
3119
  @classmethod
3120
  def UpdateCache(cls, dev_path, owner, on_primary, iv_name):
3121
    """Updates the cache information for a given device.
3122

3123
    @type dev_path: str
3124
    @param dev_path: the pathname of the device
3125
    @type owner: str
3126
    @param owner: the owner (instance name) of the device
3127
    @type on_primary: bool
3128
    @param on_primary: whether this is the primary
3129
        node nor not
3130
    @type iv_name: str
3131
    @param iv_name: the instance-visible name of the
3132
        device, as in objects.Disk.iv_name
3133

3134
    @rtype: None
3135

3136
    """
3137
    if dev_path is None:
3138
      logging.error("DevCacheManager.UpdateCache got a None dev_path")
3139
      return
3140
    fpath = cls._ConvertPath(dev_path)
3141
    if on_primary:
3142
      state = "primary"
3143
    else:
3144
      state = "secondary"
3145
    if iv_name is None:
3146
      iv_name = "not_visible"
3147
    fdata = "%s %s %s\n" % (str(owner), state, iv_name)
3148
    try:
3149
      utils.WriteFile(fpath, data=fdata)
3150
    except EnvironmentError, err:
3151
      logging.exception("Can't update bdev cache for %s: %s", dev_path, err)
3152

    
3153
  @classmethod
3154
  def RemoveCache(cls, dev_path):
3155
    """Remove data for a dev_path.
3156

3157
    This is just a wrapper over L{utils.RemoveFile} with a converted
3158
    path name and logging.
3159

3160
    @type dev_path: str
3161
    @param dev_path: the pathname of the device
3162

3163
    @rtype: None
3164

3165
    """
3166
    if dev_path is None:
3167
      logging.error("DevCacheManager.RemoveCache got a None dev_path")
3168
      return
3169
    fpath = cls._ConvertPath(dev_path)
3170
    try:
3171
      utils.RemoveFile(fpath)
3172
    except EnvironmentError, err:
3173
      logging.exception("Can't update bdev cache for %s: %s", dev_path, err)