Statistics
| Branch: | Tag: | Revision:

root / lib / backend.py @ 6c626518

History | View | Annotate | Download (75.3 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

    
25
import os
26
import os.path
27
import shutil
28
import time
29
import stat
30
import errno
31
import re
32
import subprocess
33
import random
34
import logging
35
import tempfile
36
import zlib
37
import base64
38

    
39
from ganeti import errors
40
from ganeti import utils
41
from ganeti import ssh
42
from ganeti import hypervisor
43
from ganeti import constants
44
from ganeti import bdev
45
from ganeti import objects
46
from ganeti import ssconf
47

    
48

    
49
def _GetConfig():
50
  """Simple wrapper to return a SimpleStore.
51

52
  @rtype: L{ssconf.SimpleStore}
53
  @return: a SimpleStore instance
54

55
  """
56
  return ssconf.SimpleStore()
57

    
58

    
59
def _GetSshRunner(cluster_name):
60
  """Simple wrapper to return an SshRunner.
61

62
  @type cluster_name: str
63
  @param cluster_name: the cluster name, which is needed
64
      by the SshRunner constructor
65
  @rtype: L{ssh.SshRunner}
66
  @return: an SshRunner instance
67

68
  """
69
  return ssh.SshRunner(cluster_name)
70

    
71

    
72
def _Decompress(data):
73
  """Unpacks data compressed by the RPC client.
74

75
  @type data: list or tuple
76
  @param data: Data sent by RPC client
77
  @rtype: str
78
  @return: Decompressed data
79

80
  """
81
  assert isinstance(data, (list, tuple))
82
  assert len(data) == 2
83
  (encoding, content) = data
84
  if encoding == constants.RPC_ENCODING_NONE:
85
    return content
86
  elif encoding == constants.RPC_ENCODING_ZLIB_BASE64:
87
    return zlib.decompress(base64.b64decode(content))
88
  else:
89
    raise AssertionError("Unknown data encoding")
90

    
91

    
92
def _CleanDirectory(path, exclude=None):
93
  """Removes all regular files in a directory.
94

95
  @type path: str
96
  @param path: the directory to clean
97
  @type exclude: list
98
  @param exclude: list of files to be excluded, defaults
99
      to the empty list
100

101
  """
102
  if not os.path.isdir(path):
103
    return
104
  if exclude is None:
105
    exclude = []
106
  else:
107
    # Normalize excluded paths
108
    exclude = [os.path.normpath(i) for i in exclude]
109

    
110
  for rel_name in utils.ListVisibleFiles(path):
111
    full_name = os.path.normpath(os.path.join(path, rel_name))
112
    if full_name in exclude:
113
      continue
114
    if os.path.isfile(full_name) and not os.path.islink(full_name):
115
      utils.RemoveFile(full_name)
116

    
117

    
118
def JobQueuePurge():
119
  """Removes job queue files and archived jobs.
120

121
  @rtype: None
122

123
  """
124
  _CleanDirectory(constants.QUEUE_DIR, exclude=[constants.JOB_QUEUE_LOCK_FILE])
125
  _CleanDirectory(constants.JOB_QUEUE_ARCHIVE_DIR)
126

    
127

    
128
def GetMasterInfo():
129
  """Returns master information.
130

131
  This is an utility function to compute master information, either
132
  for consumption here or from the node daemon.
133

134
  @rtype: tuple
135
  @return: (master_netdev, master_ip, master_name) if we have a good
136
      configuration, otherwise (None, None, None)
137

138
  """
139
  try:
140
    cfg = _GetConfig()
141
    master_netdev = cfg.GetMasterNetdev()
142
    master_ip = cfg.GetMasterIP()
143
    master_node = cfg.GetMasterNode()
144
  except errors.ConfigurationError, err:
145
    logging.exception("Cluster configuration incomplete")
146
    return (None, None, None)
147
  return (master_netdev, master_ip, master_node)
148

    
149

    
150
def StartMaster(start_daemons):
151
  """Activate local node as master node.
152

153
  The function will always try activate the IP address of the master
154
  (unless someone else has it). It will also start the master daemons,
155
  based on the start_daemons parameter.
156

157
  @type start_daemons: boolean
158
  @param start_daemons: whther to also start the master
159
      daemons (ganeti-masterd and ganeti-rapi)
160
  @rtype: None
161

162
  """
163
  ok = True
164
  master_netdev, master_ip, _ = GetMasterInfo()
165
  if not master_netdev:
166
    return False
167

    
168
  if utils.TcpPing(master_ip, constants.DEFAULT_NODED_PORT):
169
    if utils.OwnIpAddress(master_ip):
170
      # we already have the ip:
171
      logging.debug("Already started")
172
    else:
173
      logging.error("Someone else has the master ip, not activating")
174
      ok = False
175
  else:
176
    result = utils.RunCmd(["ip", "address", "add", "%s/32" % master_ip,
177
                           "dev", master_netdev, "label",
178
                           "%s:0" % master_netdev])
179
    if result.failed:
180
      logging.error("Can't activate master IP: %s", result.output)
181
      ok = False
182

    
183
    result = utils.RunCmd(["arping", "-q", "-U", "-c 3", "-I", master_netdev,
184
                           "-s", master_ip, master_ip])
185
    # we'll ignore the exit code of arping
186

    
187
  # and now start the master and rapi daemons
188
  if start_daemons:
189
    for daemon in 'ganeti-masterd', 'ganeti-rapi':
190
      result = utils.RunCmd([daemon])
191
      if result.failed:
192
        logging.error("Can't start daemon %s: %s", daemon, result.output)
193
        ok = False
194
  return ok
195

    
196

    
197
def StopMaster(stop_daemons):
198
  """Deactivate this node as master.
199

200
  The function will always try to deactivate the IP address of the
201
  master. It will also stop the master daemons depending on the
202
  stop_daemons parameter.
203

204
  @type stop_daemons: boolean
205
  @param stop_daemons: whether to also stop the master daemons
206
      (ganeti-masterd and ganeti-rapi)
207
  @rtype: None
208

209
  """
210
  master_netdev, master_ip, _ = GetMasterInfo()
211
  if not master_netdev:
212
    return False
213

    
214
  result = utils.RunCmd(["ip", "address", "del", "%s/32" % master_ip,
215
                         "dev", master_netdev])
216
  if result.failed:
217
    logging.error("Can't remove the master IP, error: %s", result.output)
218
    # but otherwise ignore the failure
219

    
220
  if stop_daemons:
221
    # stop/kill the rapi and the master daemon
222
    for daemon in constants.RAPI_PID, constants.MASTERD_PID:
223
      utils.KillProcess(utils.ReadPidFile(utils.DaemonPidFileName(daemon)))
224

    
225
  return True
226

    
227

    
228
def AddNode(dsa, dsapub, rsa, rsapub, sshkey, sshpub):
229
  """Joins this node to the cluster.
230

231
  This does the following:
232
      - updates the hostkeys of the machine (rsa and dsa)
233
      - adds the ssh private key to the user
234
      - adds the ssh public key to the users' authorized_keys file
235

236
  @type dsa: str
237
  @param dsa: the DSA private key to write
238
  @type dsapub: str
239
  @param dsapub: the DSA public key to write
240
  @type rsa: str
241
  @param rsa: the RSA private key to write
242
  @type rsapub: str
243
  @param rsapub: the RSA public key to write
244
  @type sshkey: str
245
  @param sshkey: the SSH private key to write
246
  @type sshpub: str
247
  @param sshpub: the SSH public key to write
248
  @rtype: boolean
249
  @return: the success of the operation
250

251
  """
252
  sshd_keys =  [(constants.SSH_HOST_RSA_PRIV, rsa, 0600),
253
                (constants.SSH_HOST_RSA_PUB, rsapub, 0644),
254
                (constants.SSH_HOST_DSA_PRIV, dsa, 0600),
255
                (constants.SSH_HOST_DSA_PUB, dsapub, 0644)]
256
  for name, content, mode in sshd_keys:
257
    utils.WriteFile(name, data=content, mode=mode)
258

    
259
  try:
260
    priv_key, pub_key, auth_keys = ssh.GetUserFiles(constants.GANETI_RUNAS,
261
                                                    mkdir=True)
262
  except errors.OpExecError, err:
263
    logging.exception("Error while processing user ssh files")
264
    return False
265

    
266
  for name, content in [(priv_key, sshkey), (pub_key, sshpub)]:
267
    utils.WriteFile(name, data=content, mode=0600)
268

    
269
  utils.AddAuthorizedKey(auth_keys, sshpub)
270

    
271
  utils.RunCmd([constants.SSH_INITD_SCRIPT, "restart"])
272

    
273
  return True
274

    
275

    
276
def LeaveCluster():
277
  """Cleans up and remove the current node.
278

279
  This function cleans up and prepares the current node to be removed
280
  from the cluster.
281

282
  If processing is successful, then it raises an
283
  L{errors.QuitGanetiException} which is used as a special case to
284
  shutdown the node daemon.
285

286
  """
287
  _CleanDirectory(constants.DATA_DIR)
288
  JobQueuePurge()
289

    
290
  try:
291
    priv_key, pub_key, auth_keys = ssh.GetUserFiles(constants.GANETI_RUNAS)
292
  except errors.OpExecError:
293
    logging.exception("Error while processing ssh files")
294
    return
295

    
296
  f = open(pub_key, 'r')
297
  try:
298
    utils.RemoveAuthorizedKey(auth_keys, f.read(8192))
299
  finally:
300
    f.close()
301

    
302
  utils.RemoveFile(priv_key)
303
  utils.RemoveFile(pub_key)
304

    
305
  # Return a reassuring string to the caller, and quit
306
  raise errors.QuitGanetiException(False, 'Shutdown scheduled')
307

    
308

    
309
def GetNodeInfo(vgname, hypervisor_type):
310
  """Gives back a hash with different informations about the node.
311

312
  @type vgname: C{string}
313
  @param vgname: the name of the volume group to ask for disk space information
314
  @type hypervisor_type: C{str}
315
  @param hypervisor_type: the name of the hypervisor to ask for
316
      memory information
317
  @rtype: C{dict}
318
  @return: dictionary with the following keys:
319
      - vg_size is the size of the configured volume group in MiB
320
      - vg_free is the free size of the volume group in MiB
321
      - memory_dom0 is the memory allocated for domain0 in MiB
322
      - memory_free is the currently available (free) ram in MiB
323
      - memory_total is the total number of ram in MiB
324

325
  """
326
  outputarray = {}
327
  vginfo = _GetVGInfo(vgname)
328
  outputarray['vg_size'] = vginfo['vg_size']
329
  outputarray['vg_free'] = vginfo['vg_free']
330

    
331
  hyper = hypervisor.GetHypervisor(hypervisor_type)
332
  hyp_info = hyper.GetNodeInfo()
333
  if hyp_info is not None:
334
    outputarray.update(hyp_info)
335

    
336
  f = open("/proc/sys/kernel/random/boot_id", 'r')
337
  try:
338
    outputarray["bootid"] = f.read(128).rstrip("\n")
339
  finally:
340
    f.close()
341

    
342
  return outputarray
343

    
344

    
345
def VerifyNode(what, cluster_name):
346
  """Verify the status of the local node.
347

348
  Based on the input L{what} parameter, various checks are done on the
349
  local node.
350

351
  If the I{filelist} key is present, this list of
352
  files is checksummed and the file/checksum pairs are returned.
353

354
  If the I{nodelist} key is present, we check that we have
355
  connectivity via ssh with the target nodes (and check the hostname
356
  report).
357

358
  If the I{node-net-test} key is present, we check that we have
359
  connectivity to the given nodes via both primary IP and, if
360
  applicable, secondary IPs.
361

362
  @type what: C{dict}
363
  @param what: a dictionary of things to check:
364
      - filelist: list of files for which to compute checksums
365
      - nodelist: list of nodes we should check ssh communication with
366
      - node-net-test: list of nodes we should check node daemon port
367
        connectivity with
368
      - hypervisor: list with hypervisors to run the verify for
369
  @rtype: dict
370
  @return: a dictionary with the same keys as the input dict, and
371
      values representing the result of the checks
372

373
  """
374
  result = {}
375

    
376
  if constants.NV_HYPERVISOR in what:
377
    result[constants.NV_HYPERVISOR] = tmp = {}
378
    for hv_name in what[constants.NV_HYPERVISOR]:
379
      tmp[hv_name] = hypervisor.GetHypervisor(hv_name).Verify()
380

    
381
  if constants.NV_FILELIST in what:
382
    result[constants.NV_FILELIST] = utils.FingerprintFiles(
383
      what[constants.NV_FILELIST])
384

    
385
  if constants.NV_NODELIST in what:
386
    result[constants.NV_NODELIST] = tmp = {}
387
    random.shuffle(what[constants.NV_NODELIST])
388
    for node in what[constants.NV_NODELIST]:
389
      success, message = _GetSshRunner(cluster_name).VerifyNodeHostname(node)
390
      if not success:
391
        tmp[node] = message
392

    
393
  if constants.NV_NODENETTEST in what:
394
    result[constants.NV_NODENETTEST] = tmp = {}
395
    my_name = utils.HostInfo().name
396
    my_pip = my_sip = None
397
    for name, pip, sip in what[constants.NV_NODENETTEST]:
398
      if name == my_name:
399
        my_pip = pip
400
        my_sip = sip
401
        break
402
    if not my_pip:
403
      tmp[my_name] = ("Can't find my own primary/secondary IP"
404
                      " in the node list")
405
    else:
406
      port = utils.GetNodeDaemonPort()
407
      for name, pip, sip in what[constants.NV_NODENETTEST]:
408
        fail = []
409
        if not utils.TcpPing(pip, port, source=my_pip):
410
          fail.append("primary")
411
        if sip != pip:
412
          if not utils.TcpPing(sip, port, source=my_sip):
413
            fail.append("secondary")
414
        if fail:
415
          tmp[name] = ("failure using the %s interface(s)" %
416
                       " and ".join(fail))
417

    
418
  if constants.NV_LVLIST in what:
419
    result[constants.NV_LVLIST] = GetVolumeList(what[constants.NV_LVLIST])
420

    
421
  if constants.NV_INSTANCELIST in what:
422
    result[constants.NV_INSTANCELIST] = GetInstanceList(
423
      what[constants.NV_INSTANCELIST])
424

    
425
  if constants.NV_VGLIST in what:
426
    result[constants.NV_VGLIST] = ListVolumeGroups()
427

    
428
  if constants.NV_VERSION in what:
429
    result[constants.NV_VERSION] = constants.PROTOCOL_VERSION
430

    
431
  if constants.NV_HVINFO in what:
432
    hyper = hypervisor.GetHypervisor(what[constants.NV_HVINFO])
433
    result[constants.NV_HVINFO] = hyper.GetNodeInfo()
434

    
435
  return result
436

    
437

    
438
def GetVolumeList(vg_name):
439
  """Compute list of logical volumes and their size.
440

441
  @type vg_name: str
442
  @param vg_name: the volume group whose LVs we should list
443
  @rtype: dict
444
  @return:
445
      dictionary of all partions (key) with value being a tuple of
446
      their size (in MiB), inactive and online status::
447

448
        {'test1': ('20.06', True, True)}
449

450
      in case of errors, a string is returned with the error
451
      details.
452

453
  """
454
  lvs = {}
455
  sep = '|'
456
  result = utils.RunCmd(["lvs", "--noheadings", "--units=m", "--nosuffix",
457
                         "--separator=%s" % sep,
458
                         "-olv_name,lv_size,lv_attr", vg_name])
459
  if result.failed:
460
    logging.error("Failed to list logical volumes, lvs output: %s",
461
                  result.output)
462
    return result.output
463

    
464
  valid_line_re = re.compile("^ *([^|]+)\|([0-9.]+)\|([^|]{6})\|?$")
465
  for line in result.stdout.splitlines():
466
    line = line.strip()
467
    match = valid_line_re.match(line)
468
    if not match:
469
      logging.error("Invalid line returned from lvs output: '%s'", line)
470
      continue
471
    name, size, attr = match.groups()
472
    inactive = attr[4] == '-'
473
    online = attr[5] == 'o'
474
    lvs[name] = (size, inactive, online)
475

    
476
  return lvs
477

    
478

    
479
def ListVolumeGroups():
480
  """List the volume groups and their size.
481

482
  @rtype: dict
483
  @return: dictionary with keys volume name and values the
484
      size of the volume
485

486
  """
487
  return utils.ListVolumeGroups()
488

    
489

    
490
def NodeVolumes():
491
  """List all volumes on this node.
492

493
  @rtype: list
494
  @return:
495
    A list of dictionaries, each having four keys:
496
      - name: the logical volume name,
497
      - size: the size of the logical volume
498
      - dev: the physical device on which the LV lives
499
      - vg: the volume group to which it belongs
500

501
    In case of errors, we return an empty list and log the
502
    error.
503

504
    Note that since a logical volume can live on multiple physical
505
    volumes, the resulting list might include a logical volume
506
    multiple times.
507

508
  """
509
  result = utils.RunCmd(["lvs", "--noheadings", "--units=m", "--nosuffix",
510
                         "--separator=|",
511
                         "--options=lv_name,lv_size,devices,vg_name"])
512
  if result.failed:
513
    logging.error("Failed to list logical volumes, lvs output: %s",
514
                  result.output)
515
    return []
516

    
517
  def parse_dev(dev):
518
    if '(' in dev:
519
      return dev.split('(')[0]
520
    else:
521
      return dev
522

    
523
  def map_line(line):
524
    return {
525
      'name': line[0].strip(),
526
      'size': line[1].strip(),
527
      'dev': parse_dev(line[2].strip()),
528
      'vg': line[3].strip(),
529
    }
530

    
531
  return [map_line(line.split('|')) for line in result.stdout.splitlines()
532
          if line.count('|') >= 3]
533

    
534

    
535
def BridgesExist(bridges_list):
536
  """Check if a list of bridges exist on the current node.
537

538
  @rtype: boolean
539
  @return: C{True} if all of them exist, C{False} otherwise
540

541
  """
542
  for bridge in bridges_list:
543
    if not utils.BridgeExists(bridge):
544
      return False
545

    
546
  return True
547

    
548

    
549
def GetInstanceList(hypervisor_list):
550
  """Provides a list of instances.
551

552
  @type hypervisor_list: list
553
  @param hypervisor_list: the list of hypervisors to query information
554

555
  @rtype: list
556
  @return: a list of all running instances on the current node
557
    - instance1.example.com
558
    - instance2.example.com
559

560
  """
561
  results = []
562
  for hname in hypervisor_list:
563
    try:
564
      names = hypervisor.GetHypervisor(hname).ListInstances()
565
      results.extend(names)
566
    except errors.HypervisorError, err:
567
      logging.exception("Error enumerating instances for hypevisor %s", hname)
568
      # FIXME: should we somehow not propagate this to the master?
569
      raise
570

    
571
  return results
572

    
573

    
574
def GetInstanceInfo(instance, hname):
575
  """Gives back the informations about an instance as a dictionary.
576

577
  @type instance: string
578
  @param instance: the instance name
579
  @type hname: string
580
  @param hname: the hypervisor type of the instance
581

582
  @rtype: dict
583
  @return: dictionary with the following keys:
584
      - memory: memory size of instance (int)
585
      - state: xen state of instance (string)
586
      - time: cpu time of instance (float)
587

588
  """
589
  output = {}
590

    
591
  iinfo = hypervisor.GetHypervisor(hname).GetInstanceInfo(instance)
592
  if iinfo is not None:
593
    output['memory'] = iinfo[2]
594
    output['state'] = iinfo[4]
595
    output['time'] = iinfo[5]
596

    
597
  return output
598

    
599

    
600
def GetInstanceMigratable(instance):
601
  """Gives whether an instance can be migrated.
602

603
  @type instance: L{objects.Instance}
604
  @param instance: object representing the instance to be checked.
605

606
  @rtype: tuple
607
  @return: tuple of (result, description) where:
608
      - result: whether the instance can be migrated or not
609
      - description: a description of the issue, if relevant
610

611
  """
612
  hyper = hypervisor.GetHypervisor(instance.hypervisor)
613
  if instance.name not in hyper.ListInstances():
614
    return (False, 'not running')
615

    
616
  for idx in range(len(instance.disks)):
617
    link_name = _GetBlockDevSymlinkPath(instance.name, idx)
618
    if not os.path.islink(link_name):
619
      return (False, 'not restarted since ganeti 1.2.5')
620

    
621
  return (True, '')
622

    
623

    
624
def GetAllInstancesInfo(hypervisor_list):
625
  """Gather data about all instances.
626

627
  This is the equivalent of L{GetInstanceInfo}, except that it
628
  computes data for all instances at once, thus being faster if one
629
  needs data about more than one instance.
630

631
  @type hypervisor_list: list
632
  @param hypervisor_list: list of hypervisors to query for instance data
633

634
  @rtype: dict
635
  @return: dictionary of instance: data, with data having the following keys:
636
      - memory: memory size of instance (int)
637
      - state: xen state of instance (string)
638
      - time: cpu time of instance (float)
639
      - vcpus: the number of vcpus
640

641
  """
642
  output = {}
643

    
644
  for hname in hypervisor_list:
645
    iinfo = hypervisor.GetHypervisor(hname).GetAllInstancesInfo()
646
    if iinfo:
647
      for name, inst_id, memory, vcpus, state, times in iinfo:
648
        value = {
649
          'memory': memory,
650
          'vcpus': vcpus,
651
          'state': state,
652
          'time': times,
653
          }
654
        if name in output and output[name] != value:
655
          raise errors.HypervisorError("Instance %s running duplicate"
656
                                       " with different parameters" % name)
657
        output[name] = value
658

    
659
  return output
660

    
661

    
662
def AddOSToInstance(instance):
663
  """Add an OS to an instance.
664

665
  @type instance: L{objects.Instance}
666
  @param instance: Instance whose OS is to be installed
667
  @rtype: boolean
668
  @return: the success of the operation
669

670
  """
671
  inst_os = OSFromDisk(instance.os)
672

    
673
  create_env = OSEnvironment(instance)
674

    
675
  logfile = "%s/add-%s-%s-%d.log" % (constants.LOG_OS_DIR, instance.os,
676
                                     instance.name, int(time.time()))
677

    
678
  result = utils.RunCmd([inst_os.create_script], env=create_env,
679
                        cwd=inst_os.path, output=logfile,)
680
  if result.failed:
681
    logging.error("os create command '%s' returned error: %s, logfile: %s,"
682
                  " output: %s", result.cmd, result.fail_reason, logfile,
683
                  result.output)
684
    return False
685

    
686
  return True
687

    
688

    
689
def RunRenameInstance(instance, old_name):
690
  """Run the OS rename script for an instance.
691

692
  @type instance: L{objects.Instance}
693
  @param instance: Instance whose OS is to be installed
694
  @type old_name: string
695
  @param old_name: previous instance name
696
  @rtype: boolean
697
  @return: the success of the operation
698

699
  """
700
  inst_os = OSFromDisk(instance.os)
701

    
702
  rename_env = OSEnvironment(instance)
703
  rename_env['OLD_INSTANCE_NAME'] = old_name
704

    
705
  logfile = "%s/rename-%s-%s-%s-%d.log" % (constants.LOG_OS_DIR, instance.os,
706
                                           old_name,
707
                                           instance.name, int(time.time()))
708

    
709
  result = utils.RunCmd([inst_os.rename_script], env=rename_env,
710
                        cwd=inst_os.path, output=logfile)
711

    
712
  if result.failed:
713
    logging.error("os create command '%s' returned error: %s output: %s",
714
                  result.cmd, result.fail_reason, result.output)
715
    return False
716

    
717
  return True
718

    
719

    
720
def _GetVGInfo(vg_name):
721
  """Get informations about the volume group.
722

723
  @type vg_name: str
724
  @param vg_name: the volume group which we query
725
  @rtype: dict
726
  @return:
727
    A dictionary with the following keys:
728
      - C{vg_size} is the total size of the volume group in MiB
729
      - C{vg_free} is the free size of the volume group in MiB
730
      - C{pv_count} are the number of physical disks in that VG
731

732
    If an error occurs during gathering of data, we return the same dict
733
    with keys all set to None.
734

735
  """
736
  retdic = dict.fromkeys(["vg_size", "vg_free", "pv_count"])
737

    
738
  retval = utils.RunCmd(["vgs", "-ovg_size,vg_free,pv_count", "--noheadings",
739
                         "--nosuffix", "--units=m", "--separator=:", vg_name])
740

    
741
  if retval.failed:
742
    logging.error("volume group %s not present", vg_name)
743
    return retdic
744
  valarr = retval.stdout.strip().rstrip(':').split(':')
745
  if len(valarr) == 3:
746
    try:
747
      retdic = {
748
        "vg_size": int(round(float(valarr[0]), 0)),
749
        "vg_free": int(round(float(valarr[1]), 0)),
750
        "pv_count": int(valarr[2]),
751
        }
752
    except ValueError, err:
753
      logging.exception("Fail to parse vgs output")
754
  else:
755
    logging.error("vgs output has the wrong number of fields (expected"
756
                  " three): %s", str(valarr))
757
  return retdic
758

    
759

    
760
def _GetBlockDevSymlinkPath(instance_name, idx):
761
  return os.path.join(constants.DISK_LINKS_DIR,
762
                      "%s:%d" % (instance_name, idx))
763

    
764

    
765
def _SymlinkBlockDev(instance_name, device_path, idx):
766
  """Set up symlinks to a instance's block device.
767

768
  This is an auxiliary function run when an instance is start (on the primary
769
  node) or when an instance is migrated (on the target node).
770

771

772
  @param instance_name: the name of the target instance
773
  @param device_path: path of the physical block device, on the node
774
  @param idx: the disk index
775
  @return: absolute path to the disk's symlink
776

777
  """
778
  link_name = _GetBlockDevSymlinkPath(instance_name, idx)
779
  try:
780
    os.symlink(device_path, link_name)
781
  except OSError, err:
782
    if err.errno == errno.EEXIST:
783
      if (not os.path.islink(link_name) or
784
          os.readlink(link_name) != device_path):
785
        os.remove(link_name)
786
        os.symlink(device_path, link_name)
787
    else:
788
      raise
789

    
790
  return link_name
791

    
792

    
793
def _RemoveBlockDevLinks(instance_name, disks):
794
  """Remove the block device symlinks belonging to the given instance.
795

796
  """
797
  for idx, disk in enumerate(disks):
798
    link_name = _GetBlockDevSymlinkPath(instance_name, idx)
799
    if os.path.islink(link_name):
800
      try:
801
        os.remove(link_name)
802
      except OSError:
803
        logging.exception("Can't remove symlink '%s'", link_name)
804

    
805

    
806
def _GatherAndLinkBlockDevs(instance):
807
  """Set up an instance's block device(s).
808

809
  This is run on the primary node at instance startup. The block
810
  devices must be already assembled.
811

812
  @type instance: L{objects.Instance}
813
  @param instance: the instance whose disks we shoul assemble
814
  @rtype: list
815
  @return: list of (disk_object, device_path)
816

817
  """
818
  block_devices = []
819
  for idx, disk in enumerate(instance.disks):
820
    device = _RecursiveFindBD(disk)
821
    if device is None:
822
      raise errors.BlockDeviceError("Block device '%s' is not set up." %
823
                                    str(disk))
824
    device.Open()
825
    try:
826
      link_name = _SymlinkBlockDev(instance.name, device.dev_path, idx)
827
    except OSError, e:
828
      raise errors.BlockDeviceError("Cannot create block device symlink: %s" %
829
                                    e.strerror)
830

    
831
    block_devices.append((disk, link_name))
832

    
833
  return block_devices
834

    
835

    
836
def StartInstance(instance, extra_args):
837
  """Start an instance.
838

839
  @type instance: L{objects.Instance}
840
  @param instance: the instance object
841
  @rtype: boolean
842
  @return: whether the startup was successful or not
843

844
  """
845
  running_instances = GetInstanceList([instance.hypervisor])
846

    
847
  if instance.name in running_instances:
848
    return True
849

    
850
  try:
851
    block_devices = _GatherAndLinkBlockDevs(instance)
852
    hyper = hypervisor.GetHypervisor(instance.hypervisor)
853
    hyper.StartInstance(instance, block_devices, extra_args)
854
  except errors.BlockDeviceError, err:
855
    logging.exception("Failed to start instance")
856
    return False
857
  except errors.HypervisorError, err:
858
    logging.exception("Failed to start instance")
859
    _RemoveBlockDevLinks(instance.name, instance.disks)
860
    return False
861

    
862
  return True
863

    
864

    
865
def ShutdownInstance(instance):
866
  """Shut an instance down.
867

868
  @note: this functions uses polling with a hardcoded timeout.
869

870
  @type instance: L{objects.Instance}
871
  @param instance: the instance object
872
  @rtype: boolean
873
  @return: whether the startup was successful or not
874

875
  """
876
  hv_name = instance.hypervisor
877
  running_instances = GetInstanceList([hv_name])
878

    
879
  if instance.name not in running_instances:
880
    return True
881

    
882
  hyper = hypervisor.GetHypervisor(hv_name)
883
  try:
884
    hyper.StopInstance(instance)
885
  except errors.HypervisorError, err:
886
    logging.error("Failed to stop instance")
887
    return False
888

    
889
  # test every 10secs for 2min
890

    
891
  time.sleep(1)
892
  for dummy in range(11):
893
    if instance.name not in GetInstanceList([hv_name]):
894
      break
895
    time.sleep(10)
896
  else:
897
    # the shutdown did not succeed
898
    logging.error("shutdown of '%s' unsuccessful, using destroy", instance)
899

    
900
    try:
901
      hyper.StopInstance(instance, force=True)
902
    except errors.HypervisorError, err:
903
      logging.exception("Failed to stop instance")
904
      return False
905

    
906
    time.sleep(1)
907
    if instance.name in GetInstanceList([hv_name]):
908
      logging.error("could not shutdown instance '%s' even by destroy",
909
                    instance.name)
910
      return False
911

    
912
  _RemoveBlockDevLinks(instance.name, instance.disks)
913

    
914
  return True
915

    
916

    
917
def RebootInstance(instance, reboot_type, extra_args):
918
  """Reboot an instance.
919

920
  @type instance: L{objects.Instance}
921
  @param instance: the instance object to reboot
922
  @type reboot_type: str
923
  @param reboot_type: the type of reboot, one the following
924
    constants:
925
      - L{constants.INSTANCE_REBOOT_SOFT}: only reboot the
926
        instance OS, do not recreate the VM
927
      - L{constants.INSTANCE_REBOOT_HARD}: tear down and
928
        restart the VM (at the hypervisor level)
929
      - the other reboot type (L{constants.INSTANCE_REBOOT_HARD})
930
        is not accepted here, since that mode is handled
931
        differently
932
  @rtype: boolean
933
  @return: the success of the operation
934

935
  """
936
  running_instances = GetInstanceList([instance.hypervisor])
937

    
938
  if instance.name not in running_instances:
939
    logging.error("Cannot reboot instance that is not running")
940
    return False
941

    
942
  hyper = hypervisor.GetHypervisor(instance.hypervisor)
943
  if reboot_type == constants.INSTANCE_REBOOT_SOFT:
944
    try:
945
      hyper.RebootInstance(instance)
946
    except errors.HypervisorError, err:
947
      logging.exception("Failed to soft reboot instance")
948
      return False
949
  elif reboot_type == constants.INSTANCE_REBOOT_HARD:
950
    try:
951
      ShutdownInstance(instance)
952
      StartInstance(instance, extra_args)
953
    except errors.HypervisorError, err:
954
      logging.exception("Failed to hard reboot instance")
955
      return False
956
  else:
957
    raise errors.ParameterError("reboot_type invalid")
958

    
959
  return True
960

    
961

    
962
def MigrateInstance(instance, target, live):
963
  """Migrates an instance to another node.
964

965
  @type instance: L{objects.Instance}
966
  @param instance: the instance definition
967
  @type target: string
968
  @param target: the target node name
969
  @type live: boolean
970
  @param live: whether the migration should be done live or not (the
971
      interpretation of this parameter is left to the hypervisor)
972
  @rtype: tuple
973
  @return: a tuple of (success, msg) where:
974
      - succes is a boolean denoting the success/failure of the operation
975
      - msg is a string with details in case of failure
976

977
  """
978
  hyper = hypervisor.GetHypervisor(instance.hypervisor)
979

    
980
  try:
981
    hyper.MigrateInstance(instance.name, target, live)
982
  except errors.HypervisorError, err:
983
    msg = "Failed to migrate instance"
984
    logging.exception(msg)
985
    return (False, "%s: %s" % (msg, err))
986
  return (True, "Migration successfull")
987

    
988

    
989
def CreateBlockDevice(disk, size, owner, on_primary, info):
990
  """Creates a block device for an instance.
991

992
  @type disk: L{objects.Disk}
993
  @param disk: the object describing the disk we should create
994
  @type size: int
995
  @param size: the size of the physical underlying device, in MiB
996
  @type owner: str
997
  @param owner: the name of the instance for which disk is created,
998
      used for device cache data
999
  @type on_primary: boolean
1000
  @param on_primary:  indicates if it is the primary node or not
1001
  @type info: string
1002
  @param info: string that will be sent to the physical device
1003
      creation, used for example to set (LVM) tags on LVs
1004

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

1009
  """
1010
  clist = []
1011
  if disk.children:
1012
    for child in disk.children:
1013
      crdev = _RecursiveAssembleBD(child, owner, on_primary)
1014
      if on_primary or disk.AssembleOnSecondary():
1015
        # we need the children open in case the device itself has to
1016
        # be assembled
1017
        crdev.Open()
1018
      clist.append(crdev)
1019

    
1020
  device = bdev.Create(disk.dev_type, disk.physical_id, clist, size)
1021

    
1022
  if on_primary or disk.AssembleOnSecondary():
1023
    if not device.Assemble():
1024
      errorstring = "Can't assemble device after creation"
1025
      logging.error(errorstring)
1026
      raise errors.BlockDeviceError("%s, very unusual event - check the node"
1027
                                    " daemon logs" % errorstring)
1028
    device.SetSyncSpeed(constants.SYNC_SPEED)
1029
    if on_primary or disk.OpenOnSecondary():
1030
      device.Open(force=True)
1031
    DevCacheManager.UpdateCache(device.dev_path, owner,
1032
                                on_primary, disk.iv_name)
1033

    
1034
  device.SetInfo(info)
1035

    
1036
  physical_id = device.unique_id
1037
  return physical_id
1038

    
1039

    
1040
def RemoveBlockDevice(disk):
1041
  """Remove a block device.
1042

1043
  @note: This is intended to be called recursively.
1044

1045
  @type disk: L{objects.Disk}
1046
  @param disk: the disk object we should remove
1047
  @rtype: boolean
1048
  @return: the success of the operation
1049

1050
  """
1051
  try:
1052
    rdev = _RecursiveFindBD(disk)
1053
  except errors.BlockDeviceError, err:
1054
    # probably can't attach
1055
    logging.info("Can't attach to device %s in remove", disk)
1056
    rdev = None
1057
  if rdev is not None:
1058
    r_path = rdev.dev_path
1059
    result = rdev.Remove()
1060
    if result:
1061
      DevCacheManager.RemoveCache(r_path)
1062
  else:
1063
    result = True
1064
  if disk.children:
1065
    for child in disk.children:
1066
      result = result and RemoveBlockDevice(child)
1067
  return result
1068

    
1069

    
1070
def _RecursiveAssembleBD(disk, owner, as_primary):
1071
  """Activate a block device for an instance.
1072

1073
  This is run on the primary and secondary nodes for an instance.
1074

1075
  @note: this function is called recursively.
1076

1077
  @type disk: L{objects.Disk}
1078
  @param disk: the disk we try to assemble
1079
  @type owner: str
1080
  @param owner: the name of the instance which owns the disk
1081
  @type as_primary: boolean
1082
  @param as_primary: if we should make the block device
1083
      read/write
1084

1085
  @return: the assembled device or None (in case no device
1086
      was assembled)
1087
  @raise errors.BlockDeviceError: in case there is an error
1088
      during the activation of the children or the device
1089
      itself
1090

1091
  """
1092
  children = []
1093
  if disk.children:
1094
    mcn = disk.ChildrenNeeded()
1095
    if mcn == -1:
1096
      mcn = 0 # max number of Nones allowed
1097
    else:
1098
      mcn = len(disk.children) - mcn # max number of Nones
1099
    for chld_disk in disk.children:
1100
      try:
1101
        cdev = _RecursiveAssembleBD(chld_disk, owner, as_primary)
1102
      except errors.BlockDeviceError, err:
1103
        if children.count(None) >= mcn:
1104
          raise
1105
        cdev = None
1106
        logging.debug("Error in child activation: %s", str(err))
1107
      children.append(cdev)
1108

    
1109
  if as_primary or disk.AssembleOnSecondary():
1110
    r_dev = bdev.Assemble(disk.dev_type, disk.physical_id, children)
1111
    r_dev.SetSyncSpeed(constants.SYNC_SPEED)
1112
    result = r_dev
1113
    if as_primary or disk.OpenOnSecondary():
1114
      r_dev.Open()
1115
    DevCacheManager.UpdateCache(r_dev.dev_path, owner,
1116
                                as_primary, disk.iv_name)
1117

    
1118
  else:
1119
    result = True
1120
  return result
1121

    
1122

    
1123
def AssembleBlockDevice(disk, owner, as_primary):
1124
  """Activate a block device for an instance.
1125

1126
  This is a wrapper over _RecursiveAssembleBD.
1127

1128
  @rtype: str or boolean
1129
  @return: a C{/dev/...} path for primary nodes, and
1130
      C{True} for secondary nodes
1131

1132
  """
1133
  result = _RecursiveAssembleBD(disk, owner, as_primary)
1134
  if isinstance(result, bdev.BlockDev):
1135
    result = result.dev_path
1136
  return result
1137

    
1138

    
1139
def ShutdownBlockDevice(disk):
1140
  """Shut down a block device.
1141

1142
  First, if the device is assembled (Attach() is successfull), then
1143
  the device is shutdown. Then the children of the device are
1144
  shutdown.
1145

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

1150
  @type disk: L{objects.Disk}
1151
  @param disk: the description of the disk we should
1152
      shutdown
1153
  @rtype: boolean
1154
  @return: the success of the operation
1155

1156
  """
1157
  r_dev = _RecursiveFindBD(disk)
1158
  if r_dev is not None:
1159
    r_path = r_dev.dev_path
1160
    result = r_dev.Shutdown()
1161
    if result:
1162
      DevCacheManager.RemoveCache(r_path)
1163
  else:
1164
    result = True
1165
  if disk.children:
1166
    for child in disk.children:
1167
      result = result and ShutdownBlockDevice(child)
1168
  return result
1169

    
1170

    
1171
def MirrorAddChildren(parent_cdev, new_cdevs):
1172
  """Extend a mirrored block device.
1173

1174
  @type parent_cdev: L{objects.Disk}
1175
  @param parent_cdev: the disk to which we should add children
1176
  @type new_cdevs: list of L{objects.Disk}
1177
  @param new_cdevs: the list of children which we should add
1178
  @rtype: boolean
1179
  @return: the success of the operation
1180

1181
  """
1182
  parent_bdev = _RecursiveFindBD(parent_cdev)
1183
  if parent_bdev is None:
1184
    logging.error("Can't find parent device")
1185
    return False
1186
  new_bdevs = [_RecursiveFindBD(disk) for disk in new_cdevs]
1187
  if new_bdevs.count(None) > 0:
1188
    logging.error("Can't find new device(s) to add: %s:%s",
1189
                  new_bdevs, new_cdevs)
1190
    return False
1191
  parent_bdev.AddChildren(new_bdevs)
1192
  return True
1193

    
1194

    
1195
def MirrorRemoveChildren(parent_cdev, new_cdevs):
1196
  """Shrink a mirrored block device.
1197

1198
  @type parent_cdev: L{objects.Disk}
1199
  @param parent_cdev: the disk from which we should remove children
1200
  @type new_cdevs: list of L{objects.Disk}
1201
  @param new_cdevs: the list of children which we should remove
1202
  @rtype: boolean
1203
  @return: the success of the operation
1204

1205
  """
1206
  parent_bdev = _RecursiveFindBD(parent_cdev)
1207
  if parent_bdev is None:
1208
    logging.error("Can't find parent in remove children: %s", parent_cdev)
1209
    return False
1210
  devs = []
1211
  for disk in new_cdevs:
1212
    rpath = disk.StaticDevPath()
1213
    if rpath is None:
1214
      bd = _RecursiveFindBD(disk)
1215
      if bd is None:
1216
        logging.error("Can't find dynamic device %s while removing children",
1217
                      disk)
1218
        return False
1219
      else:
1220
        devs.append(bd.dev_path)
1221
    else:
1222
      devs.append(rpath)
1223
  parent_bdev.RemoveChildren(devs)
1224
  return True
1225

    
1226

    
1227
def GetMirrorStatus(disks):
1228
  """Get the mirroring status of a list of devices.
1229

1230
  @type disks: list of L{objects.Disk}
1231
  @param disks: the list of disks which we should query
1232
  @rtype: disk
1233
  @return:
1234
      a list of (mirror_done, estimated_time) tuples, which
1235
      are the result of L{bdev.BlockDev.CombinedSyncStatus}
1236
  @raise errors.BlockDeviceError: if any of the disks cannot be
1237
      found
1238

1239
  """
1240
  stats = []
1241
  for dsk in disks:
1242
    rbd = _RecursiveFindBD(dsk)
1243
    if rbd is None:
1244
      raise errors.BlockDeviceError("Can't find device %s" % str(dsk))
1245
    stats.append(rbd.CombinedSyncStatus())
1246
  return stats
1247

    
1248

    
1249
def _RecursiveFindBD(disk):
1250
  """Check if a device is activated.
1251

1252
  If so, return informations about the real device.
1253

1254
  @type disk: L{objects.Disk}
1255
  @param disk: the disk object we need to find
1256

1257
  @return: None if the device can't be found,
1258
      otherwise the device instance
1259

1260
  """
1261
  children = []
1262
  if disk.children:
1263
    for chdisk in disk.children:
1264
      children.append(_RecursiveFindBD(chdisk))
1265

    
1266
  return bdev.FindDevice(disk.dev_type, disk.physical_id, children)
1267

    
1268

    
1269
def FindBlockDevice(disk):
1270
  """Check if a device is activated.
1271

1272
  If it is, return informations about the real device.
1273

1274
  @type disk: L{objects.Disk}
1275
  @param disk: the disk to find
1276
  @rtype: None or tuple
1277
  @return: None if the disk cannot be found, otherwise a
1278
      tuple (device_path, major, minor, sync_percent,
1279
      estimated_time, is_degraded)
1280

1281
  """
1282
  rbd = _RecursiveFindBD(disk)
1283
  if rbd is None:
1284
    return rbd
1285
  return (rbd.dev_path, rbd.major, rbd.minor) + rbd.GetSyncStatus()
1286

    
1287

    
1288
def UploadFile(file_name, data, mode, uid, gid, atime, mtime):
1289
  """Write a file to the filesystem.
1290

1291
  This allows the master to overwrite(!) a file. It will only perform
1292
  the operation if the file belongs to a list of configuration files.
1293

1294
  @type file_name: str
1295
  @param file_name: the target file name
1296
  @type data: str
1297
  @param data: the new contents of the file
1298
  @type mode: int
1299
  @param mode: the mode to give the file (can be None)
1300
  @type uid: int
1301
  @param uid: the owner of the file (can be -1 for default)
1302
  @type gid: int
1303
  @param gid: the group of the file (can be -1 for default)
1304
  @type atime: float
1305
  @param atime: the atime to set on the file (can be None)
1306
  @type mtime: float
1307
  @param mtime: the mtime to set on the file (can be None)
1308
  @rtype: boolean
1309
  @return: the success of the operation; errors are logged
1310
      in the node daemon log
1311

1312
  """
1313
  if not os.path.isabs(file_name):
1314
    logging.error("Filename passed to UploadFile is not absolute: '%s'",
1315
                  file_name)
1316
    return False
1317

    
1318
  allowed_files = [
1319
    constants.CLUSTER_CONF_FILE,
1320
    constants.ETC_HOSTS,
1321
    constants.SSH_KNOWN_HOSTS_FILE,
1322
    constants.VNC_PASSWORD_FILE,
1323
    ]
1324

    
1325
  if file_name not in allowed_files:
1326
    logging.error("Filename passed to UploadFile not in allowed"
1327
                 " upload targets: '%s'", file_name)
1328
    return False
1329

    
1330
  raw_data = _Decompress(data)
1331

    
1332
  utils.WriteFile(file_name, data=raw_data, mode=mode, uid=uid, gid=gid,
1333
                  atime=atime, mtime=mtime)
1334
  return True
1335

    
1336

    
1337
def WriteSsconfFiles(values):
1338
  """Update all ssconf files.
1339

1340
  Wrapper around the SimpleStore.WriteFiles.
1341

1342
  """
1343
  ssconf.SimpleStore().WriteFiles(values)
1344

    
1345

    
1346
def _ErrnoOrStr(err):
1347
  """Format an EnvironmentError exception.
1348

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

1353
  @type err: L{EnvironmentError}
1354
  @param err: the exception to format
1355

1356
  """
1357
  if hasattr(err, 'errno'):
1358
    detail = errno.errorcode[err.errno]
1359
  else:
1360
    detail = str(err)
1361
  return detail
1362

    
1363

    
1364
def _OSOndiskVersion(name, os_dir):
1365
  """Compute and return the API version of a given OS.
1366

1367
  This function will try to read the API version of the OS given by
1368
  the 'name' parameter and residing in the 'os_dir' directory.
1369

1370
  @type name: str
1371
  @param name: the OS name we should look for
1372
  @type os_dir: str
1373
  @param os_dir: the directory inwhich we should look for the OS
1374
  @rtype: int or None
1375
  @return:
1376
      Either an integer denoting the version or None in the
1377
      case when this is not a valid OS name.
1378
  @raise errors.InvalidOS: if the OS cannot be found
1379

1380
  """
1381
  api_file = os.path.sep.join([os_dir, "ganeti_api_version"])
1382

    
1383
  try:
1384
    st = os.stat(api_file)
1385
  except EnvironmentError, err:
1386
    raise errors.InvalidOS(name, os_dir, "'ganeti_api_version' file not"
1387
                           " found (%s)" % _ErrnoOrStr(err))
1388

    
1389
  if not stat.S_ISREG(stat.S_IFMT(st.st_mode)):
1390
    raise errors.InvalidOS(name, os_dir, "'ganeti_api_version' file is not"
1391
                           " a regular file")
1392

    
1393
  try:
1394
    f = open(api_file)
1395
    try:
1396
      api_versions = f.readlines()
1397
    finally:
1398
      f.close()
1399
  except EnvironmentError, err:
1400
    raise errors.InvalidOS(name, os_dir, "error while reading the"
1401
                           " API version (%s)" % _ErrnoOrStr(err))
1402

    
1403
  api_versions = [version.strip() for version in api_versions]
1404
  try:
1405
    api_versions = [int(version) for version in api_versions]
1406
  except (TypeError, ValueError), err:
1407
    raise errors.InvalidOS(name, os_dir,
1408
                           "API version is not integer (%s)" % str(err))
1409

    
1410
  return api_versions
1411

    
1412

    
1413
def DiagnoseOS(top_dirs=None):
1414
  """Compute the validity for all OSes.
1415

1416
  @type top_dirs: list
1417
  @param top_dirs: the list of directories in which to
1418
      search (if not given defaults to
1419
      L{constants.OS_SEARCH_PATH})
1420
  @rtype: list of L{objects.OS}
1421
  @return: an OS object for each name in all the given
1422
      directories
1423

1424
  """
1425
  if top_dirs is None:
1426
    top_dirs = constants.OS_SEARCH_PATH
1427

    
1428
  result = []
1429
  for dir_name in top_dirs:
1430
    if os.path.isdir(dir_name):
1431
      try:
1432
        f_names = utils.ListVisibleFiles(dir_name)
1433
      except EnvironmentError, err:
1434
        logging.exception("Can't list the OS directory %s", dir_name)
1435
        break
1436
      for name in f_names:
1437
        try:
1438
          os_inst = OSFromDisk(name, base_dir=dir_name)
1439
          result.append(os_inst)
1440
        except errors.InvalidOS, err:
1441
          result.append(objects.OS.FromInvalidOS(err))
1442

    
1443
  return result
1444

    
1445

    
1446
def OSFromDisk(name, base_dir=None):
1447
  """Create an OS instance from disk.
1448

1449
  This function will return an OS instance if the given name is a
1450
  valid OS name. Otherwise, it will raise an appropriate
1451
  L{errors.InvalidOS} exception, detailing why this is not a valid OS.
1452

1453
  @type base_dir: string
1454
  @keyword base_dir: Base directory containing OS installations.
1455
                     Defaults to a search in all the OS_SEARCH_PATH dirs.
1456
  @rtype: L{objects.OS}
1457
  @return: the OS instance if we find a valid one
1458
  @raise errors.InvalidOS: if we don't find a valid OS
1459

1460
  """
1461
  if base_dir is None:
1462
    os_dir = utils.FindFile(name, constants.OS_SEARCH_PATH, os.path.isdir)
1463
    if os_dir is None:
1464
      raise errors.InvalidOS(name, None, "OS dir not found in search path")
1465
  else:
1466
    os_dir = os.path.sep.join([base_dir, name])
1467

    
1468
  api_versions = _OSOndiskVersion(name, os_dir)
1469

    
1470
  if constants.OS_API_VERSION not in api_versions:
1471
    raise errors.InvalidOS(name, os_dir, "API version mismatch"
1472
                           " (found %s want %s)"
1473
                           % (api_versions, constants.OS_API_VERSION))
1474

    
1475
  # OS Scripts dictionary, we will populate it with the actual script names
1476
  os_scripts = dict.fromkeys(constants.OS_SCRIPTS)
1477

    
1478
  for script in os_scripts:
1479
    os_scripts[script] = os.path.sep.join([os_dir, script])
1480

    
1481
    try:
1482
      st = os.stat(os_scripts[script])
1483
    except EnvironmentError, err:
1484
      raise errors.InvalidOS(name, os_dir, "'%s' script missing (%s)" %
1485
                             (script, _ErrnoOrStr(err)))
1486

    
1487
    if stat.S_IMODE(st.st_mode) & stat.S_IXUSR != stat.S_IXUSR:
1488
      raise errors.InvalidOS(name, os_dir, "'%s' script not executable" %
1489
                             script)
1490

    
1491
    if not stat.S_ISREG(stat.S_IFMT(st.st_mode)):
1492
      raise errors.InvalidOS(name, os_dir, "'%s' is not a regular file" %
1493
                             script)
1494

    
1495

    
1496
  return objects.OS(name=name, path=os_dir, status=constants.OS_VALID_STATUS,
1497
                    create_script=os_scripts[constants.OS_SCRIPT_CREATE],
1498
                    export_script=os_scripts[constants.OS_SCRIPT_EXPORT],
1499
                    import_script=os_scripts[constants.OS_SCRIPT_IMPORT],
1500
                    rename_script=os_scripts[constants.OS_SCRIPT_RENAME],
1501
                    api_versions=api_versions)
1502

    
1503
def OSEnvironment(instance, debug=0):
1504
  """Calculate the environment for an os script.
1505

1506
  @type instance: L{objects.Instance}
1507
  @param instance: target instance for the os script run
1508
  @type debug: integer
1509
  @param debug: debug level (0 or 1, for OS Api 10)
1510
  @rtype: dict
1511
  @return: dict of environment variables
1512
  @raise errors.BlockDeviceError: if the block device
1513
      cannot be found
1514

1515
  """
1516
  result = {}
1517
  result['OS_API_VERSION'] = '%d' % constants.OS_API_VERSION
1518
  result['INSTANCE_NAME'] = instance.name
1519
  result['HYPERVISOR'] = instance.hypervisor
1520
  result['DISK_COUNT'] = '%d' % len(instance.disks)
1521
  result['NIC_COUNT'] = '%d' % len(instance.nics)
1522
  result['DEBUG_LEVEL'] = '%d' % debug
1523
  for idx, disk in enumerate(instance.disks):
1524
    real_disk = _RecursiveFindBD(disk)
1525
    if real_disk is None:
1526
      raise errors.BlockDeviceError("Block device '%s' is not set up" %
1527
                                    str(disk))
1528
    real_disk.Open()
1529
    result['DISK_%d_PATH' % idx] = real_disk.dev_path
1530
    # FIXME: When disks will have read-only mode, populate this
1531
    result['DISK_%d_ACCESS' % idx] = 'W'
1532
    if constants.HV_DISK_TYPE in instance.hvparams:
1533
      result['DISK_%d_FRONTEND_TYPE' % idx] = \
1534
        instance.hvparams[constants.HV_DISK_TYPE]
1535
    if disk.dev_type in constants.LDS_BLOCK:
1536
      result['DISK_%d_BACKEND_TYPE' % idx] = 'block'
1537
    elif disk.dev_type == constants.LD_FILE:
1538
      result['DISK_%d_BACKEND_TYPE' % idx] = \
1539
        'file:%s' % disk.physical_id[0]
1540
  for idx, nic in enumerate(instance.nics):
1541
    result['NIC_%d_MAC' % idx] = nic.mac
1542
    if nic.ip:
1543
      result['NIC_%d_IP' % idx] = nic.ip
1544
    result['NIC_%d_BRIDGE' % idx] = nic.bridge
1545
    if constants.HV_NIC_TYPE in instance.hvparams:
1546
      result['NIC_%d_FRONTEND_TYPE' % idx] = \
1547
        instance.hvparams[constants.HV_NIC_TYPE]
1548

    
1549
  return result
1550

    
1551
def GrowBlockDevice(disk, amount):
1552
  """Grow a stack of block devices.
1553

1554
  This function is called recursively, with the childrens being the
1555
  first ones to resize.
1556

1557
  @type disk: L{objects.Disk}
1558
  @param disk: the disk to be grown
1559
  @rtype: (status, result)
1560
  @return: a tuple with the status of the operation
1561
      (True/False), and the errors message if status
1562
      is False
1563

1564
  """
1565
  r_dev = _RecursiveFindBD(disk)
1566
  if r_dev is None:
1567
    return False, "Cannot find block device %s" % (disk,)
1568

    
1569
  try:
1570
    r_dev.Grow(amount)
1571
  except errors.BlockDeviceError, err:
1572
    return False, str(err)
1573

    
1574
  return True, None
1575

    
1576

    
1577
def SnapshotBlockDevice(disk):
1578
  """Create a snapshot copy of a block device.
1579

1580
  This function is called recursively, and the snapshot is actually created
1581
  just for the leaf lvm backend device.
1582

1583
  @type disk: L{objects.Disk}
1584
  @param disk: the disk to be snapshotted
1585
  @rtype: string
1586
  @return: snapshot disk path
1587

1588
  """
1589
  if disk.children:
1590
    if len(disk.children) == 1:
1591
      # only one child, let's recurse on it
1592
      return SnapshotBlockDevice(disk.children[0])
1593
    else:
1594
      # more than one child, choose one that matches
1595
      for child in disk.children:
1596
        if child.size == disk.size:
1597
          # return implies breaking the loop
1598
          return SnapshotBlockDevice(child)
1599
  elif disk.dev_type == constants.LD_LV:
1600
    r_dev = _RecursiveFindBD(disk)
1601
    if r_dev is not None:
1602
      # let's stay on the safe side and ask for the full size, for now
1603
      return r_dev.Snapshot(disk.size)
1604
    else:
1605
      return None
1606
  else:
1607
    raise errors.ProgrammerError("Cannot snapshot non-lvm block device"
1608
                                 " '%s' of type '%s'" %
1609
                                 (disk.unique_id, disk.dev_type))
1610

    
1611

    
1612
def ExportSnapshot(disk, dest_node, instance, cluster_name, idx):
1613
  """Export a block device snapshot to a remote node.
1614

1615
  @type disk: L{objects.Disk}
1616
  @param disk: the description of the disk to export
1617
  @type dest_node: str
1618
  @param dest_node: the destination node to export to
1619
  @type instance: L{objects.Instance}
1620
  @param instance: the instance object to whom the disk belongs
1621
  @type cluster_name: str
1622
  @param cluster_name: the cluster name, needed for SSH hostalias
1623
  @type idx: int
1624
  @param idx: the index of the disk in the instance's disk list,
1625
      used to export to the OS scripts environment
1626
  @rtype: boolean
1627
  @return: the success of the operation
1628

1629
  """
1630
  export_env = OSEnvironment(instance)
1631

    
1632
  inst_os = OSFromDisk(instance.os)
1633
  export_script = inst_os.export_script
1634

    
1635
  logfile = "%s/exp-%s-%s-%s.log" % (constants.LOG_OS_DIR, inst_os.name,
1636
                                     instance.name, int(time.time()))
1637
  if not os.path.exists(constants.LOG_OS_DIR):
1638
    os.mkdir(constants.LOG_OS_DIR, 0750)
1639
  real_disk = _RecursiveFindBD(disk)
1640
  if real_disk is None:
1641
    raise errors.BlockDeviceError("Block device '%s' is not set up" %
1642
                                  str(disk))
1643
  real_disk.Open()
1644

    
1645
  export_env['EXPORT_DEVICE'] = real_disk.dev_path
1646
  export_env['EXPORT_INDEX'] = str(idx)
1647

    
1648
  destdir = os.path.join(constants.EXPORT_DIR, instance.name + ".new")
1649
  destfile = disk.physical_id[1]
1650

    
1651
  # the target command is built out of three individual commands,
1652
  # which are joined by pipes; we check each individual command for
1653
  # valid parameters
1654
  expcmd = utils.BuildShellCmd("cd %s; %s 2>%s", inst_os.path,
1655
                               export_script, logfile)
1656

    
1657
  comprcmd = "gzip"
1658

    
1659
  destcmd = utils.BuildShellCmd("mkdir -p %s && cat > %s/%s",
1660
                                destdir, destdir, destfile)
1661
  remotecmd = _GetSshRunner(cluster_name).BuildCmd(dest_node,
1662
                                                   constants.GANETI_RUNAS,
1663
                                                   destcmd)
1664

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

    
1668
  result = utils.RunCmd(command, env=export_env)
1669

    
1670
  if result.failed:
1671
    logging.error("os snapshot export command '%s' returned error: %s"
1672
                  " output: %s", command, result.fail_reason, result.output)
1673
    return False
1674

    
1675
  return True
1676

    
1677

    
1678
def FinalizeExport(instance, snap_disks):
1679
  """Write out the export configuration information.
1680

1681
  @type instance: L{objects.Instance}
1682
  @param instance: the instance which we export, used for
1683
      saving configuration
1684
  @type snap_disks: list of L{objects.Disk}
1685
  @param snap_disks: list of snapshot block devices, which
1686
      will be used to get the actual name of the dump file
1687

1688
  @rtype: boolean
1689
  @return: the success of the operation
1690

1691
  """
1692
  destdir = os.path.join(constants.EXPORT_DIR, instance.name + ".new")
1693
  finaldestdir = os.path.join(constants.EXPORT_DIR, instance.name)
1694

    
1695
  config = objects.SerializableConfigParser()
1696

    
1697
  config.add_section(constants.INISECT_EXP)
1698
  config.set(constants.INISECT_EXP, 'version', '0')
1699
  config.set(constants.INISECT_EXP, 'timestamp', '%d' % int(time.time()))
1700
  config.set(constants.INISECT_EXP, 'source', instance.primary_node)
1701
  config.set(constants.INISECT_EXP, 'os', instance.os)
1702
  config.set(constants.INISECT_EXP, 'compression', 'gzip')
1703

    
1704
  config.add_section(constants.INISECT_INS)
1705
  config.set(constants.INISECT_INS, 'name', instance.name)
1706
  config.set(constants.INISECT_INS, 'memory', '%d' %
1707
             instance.beparams[constants.BE_MEMORY])
1708
  config.set(constants.INISECT_INS, 'vcpus', '%d' %
1709
             instance.beparams[constants.BE_VCPUS])
1710
  config.set(constants.INISECT_INS, 'disk_template', instance.disk_template)
1711

    
1712
  nic_total = 0
1713
  for nic_count, nic in enumerate(instance.nics):
1714
    nic_total += 1
1715
    config.set(constants.INISECT_INS, 'nic%d_mac' %
1716
               nic_count, '%s' % nic.mac)
1717
    config.set(constants.INISECT_INS, 'nic%d_ip' % nic_count, '%s' % nic.ip)
1718
    config.set(constants.INISECT_INS, 'nic%d_bridge' % nic_count,
1719
               '%s' % nic.bridge)
1720
  # TODO: redundant: on load can read nics until it doesn't exist
1721
  config.set(constants.INISECT_INS, 'nic_count' , '%d' % nic_total)
1722

    
1723
  disk_total = 0
1724
  for disk_count, disk in enumerate(snap_disks):
1725
    if disk:
1726
      disk_total += 1
1727
      config.set(constants.INISECT_INS, 'disk%d_ivname' % disk_count,
1728
                 ('%s' % disk.iv_name))
1729
      config.set(constants.INISECT_INS, 'disk%d_dump' % disk_count,
1730
                 ('%s' % disk.physical_id[1]))
1731
      config.set(constants.INISECT_INS, 'disk%d_size' % disk_count,
1732
                 ('%d' % disk.size))
1733

    
1734
  config.set(constants.INISECT_INS, 'disk_count' , '%d' % disk_total)
1735

    
1736
  utils.WriteFile(os.path.join(destdir, constants.EXPORT_CONF_FILE),
1737
                  data=config.Dumps())
1738
  shutil.rmtree(finaldestdir, True)
1739
  shutil.move(destdir, finaldestdir)
1740

    
1741
  return True
1742

    
1743

    
1744
def ExportInfo(dest):
1745
  """Get export configuration information.
1746

1747
  @type dest: str
1748
  @param dest: directory containing the export
1749

1750
  @rtype: L{objects.SerializableConfigParser}
1751
  @return: a serializable config file containing the
1752
      export info
1753

1754
  """
1755
  cff = os.path.join(dest, constants.EXPORT_CONF_FILE)
1756

    
1757
  config = objects.SerializableConfigParser()
1758
  config.read(cff)
1759

    
1760
  if (not config.has_section(constants.INISECT_EXP) or
1761
      not config.has_section(constants.INISECT_INS)):
1762
    return None
1763

    
1764
  return config
1765

    
1766

    
1767
def ImportOSIntoInstance(instance, src_node, src_images, cluster_name):
1768
  """Import an os image into an instance.
1769

1770
  @type instance: L{objects.Instance}
1771
  @param instance: instance to import the disks into
1772
  @type src_node: string
1773
  @param src_node: source node for the disk images
1774
  @type src_images: list of string
1775
  @param src_images: absolute paths of the disk images
1776
  @rtype: list of boolean
1777
  @return: each boolean represent the success of importing the n-th disk
1778

1779
  """
1780
  import_env = OSEnvironment(instance)
1781
  inst_os = OSFromDisk(instance.os)
1782
  import_script = inst_os.import_script
1783

    
1784
  logfile = "%s/import-%s-%s-%s.log" % (constants.LOG_OS_DIR, instance.os,
1785
                                        instance.name, int(time.time()))
1786
  if not os.path.exists(constants.LOG_OS_DIR):
1787
    os.mkdir(constants.LOG_OS_DIR, 0750)
1788

    
1789
  comprcmd = "gunzip"
1790
  impcmd = utils.BuildShellCmd("(cd %s; %s >%s 2>&1)", inst_os.path,
1791
                               import_script, logfile)
1792

    
1793
  final_result = []
1794
  for idx, image in enumerate(src_images):
1795
    if image:
1796
      destcmd = utils.BuildShellCmd('cat %s', image)
1797
      remotecmd = _GetSshRunner(cluster_name).BuildCmd(src_node,
1798
                                                       constants.GANETI_RUNAS,
1799
                                                       destcmd)
1800
      command = '|'.join([utils.ShellQuoteArgs(remotecmd), comprcmd, impcmd])
1801
      import_env['IMPORT_DEVICE'] = import_env['DISK_%d_PATH' % idx]
1802
      import_env['IMPORT_INDEX'] = str(idx)
1803
      result = utils.RunCmd(command, env=import_env)
1804
      if result.failed:
1805
        logging.error("Disk import command '%s' returned error: %s"
1806
                      " output: %s", command, result.fail_reason,
1807
                      result.output)
1808
        final_result.append(False)
1809
      else:
1810
        final_result.append(True)
1811
    else:
1812
      final_result.append(True)
1813

    
1814
  return final_result
1815

    
1816

    
1817
def ListExports():
1818
  """Return a list of exports currently available on this machine.
1819

1820
  @rtype: list
1821
  @return: list of the exports
1822

1823
  """
1824
  if os.path.isdir(constants.EXPORT_DIR):
1825
    return utils.ListVisibleFiles(constants.EXPORT_DIR)
1826
  else:
1827
    return []
1828

    
1829

    
1830
def RemoveExport(export):
1831
  """Remove an existing export from the node.
1832

1833
  @type export: str
1834
  @param export: the name of the export to remove
1835
  @rtype: boolean
1836
  @return: the success of the operation
1837

1838
  """
1839
  target = os.path.join(constants.EXPORT_DIR, export)
1840

    
1841
  shutil.rmtree(target)
1842
  # TODO: catch some of the relevant exceptions and provide a pretty
1843
  # error message if rmtree fails.
1844

    
1845
  return True
1846

    
1847

    
1848
def RenameBlockDevices(devlist):
1849
  """Rename a list of block devices.
1850

1851
  @type devlist: list of tuples
1852
  @param devlist: list of tuples of the form  (disk,
1853
      new_logical_id, new_physical_id); disk is an
1854
      L{objects.Disk} object describing the current disk,
1855
      and new logical_id/physical_id is the name we
1856
      rename it to
1857
  @rtype: boolean
1858
  @return: True if all renames succeeded, False otherwise
1859

1860
  """
1861
  result = True
1862
  for disk, unique_id in devlist:
1863
    dev = _RecursiveFindBD(disk)
1864
    if dev is None:
1865
      result = False
1866
      continue
1867
    try:
1868
      old_rpath = dev.dev_path
1869
      dev.Rename(unique_id)
1870
      new_rpath = dev.dev_path
1871
      if old_rpath != new_rpath:
1872
        DevCacheManager.RemoveCache(old_rpath)
1873
        # FIXME: we should add the new cache information here, like:
1874
        # DevCacheManager.UpdateCache(new_rpath, owner, ...)
1875
        # but we don't have the owner here - maybe parse from existing
1876
        # cache? for now, we only lose lvm data when we rename, which
1877
        # is less critical than DRBD or MD
1878
    except errors.BlockDeviceError, err:
1879
      logging.exception("Can't rename device '%s' to '%s'", dev, unique_id)
1880
      result = False
1881
  return result
1882

    
1883

    
1884
def _TransformFileStorageDir(file_storage_dir):
1885
  """Checks whether given file_storage_dir is valid.
1886

1887
  Checks wheter the given file_storage_dir is within the cluster-wide
1888
  default file_storage_dir stored in SimpleStore. Only paths under that
1889
  directory are allowed.
1890

1891
  @type file_storage_dir: str
1892
  @param file_storage_dir: the path to check
1893

1894
  @return: the normalized path if valid, None otherwise
1895

1896
  """
1897
  cfg = _GetConfig()
1898
  file_storage_dir = os.path.normpath(file_storage_dir)
1899
  base_file_storage_dir = cfg.GetFileStorageDir()
1900
  if (not os.path.commonprefix([file_storage_dir, base_file_storage_dir]) ==
1901
      base_file_storage_dir):
1902
    logging.error("file storage directory '%s' is not under base file"
1903
                  " storage directory '%s'",
1904
                  file_storage_dir, base_file_storage_dir)
1905
    return None
1906
  return file_storage_dir
1907

    
1908

    
1909
def CreateFileStorageDir(file_storage_dir):
1910
  """Create file storage directory.
1911

1912
  @type file_storage_dir: str
1913
  @param file_storage_dir: directory to create
1914

1915
  @rtype: tuple
1916
  @return: tuple with first element a boolean indicating wheter dir
1917
      creation was successful or not
1918

1919
  """
1920
  file_storage_dir = _TransformFileStorageDir(file_storage_dir)
1921
  result = True,
1922
  if not file_storage_dir:
1923
    result = False,
1924
  else:
1925
    if os.path.exists(file_storage_dir):
1926
      if not os.path.isdir(file_storage_dir):
1927
        logging.error("'%s' is not a directory", file_storage_dir)
1928
        result = False,
1929
    else:
1930
      try:
1931
        os.makedirs(file_storage_dir, 0750)
1932
      except OSError, err:
1933
        logging.error("Cannot create file storage directory '%s': %s",
1934
                      file_storage_dir, err)
1935
        result = False,
1936
  return result
1937

    
1938

    
1939
def RemoveFileStorageDir(file_storage_dir):
1940
  """Remove file storage directory.
1941

1942
  Remove it only if it's empty. If not log an error and return.
1943

1944
  @type file_storage_dir: str
1945
  @param file_storage_dir: the directory we should cleanup
1946
  @rtype: tuple (success,)
1947
  @return: tuple of one element, C{success}, denoting
1948
      whether the operation was successfull
1949

1950
  """
1951
  file_storage_dir = _TransformFileStorageDir(file_storage_dir)
1952
  result = True,
1953
  if not file_storage_dir:
1954
    result = False,
1955
  else:
1956
    if os.path.exists(file_storage_dir):
1957
      if not os.path.isdir(file_storage_dir):
1958
        logging.error("'%s' is not a directory", file_storage_dir)
1959
        result = False,
1960
      # deletes dir only if empty, otherwise we want to return False
1961
      try:
1962
        os.rmdir(file_storage_dir)
1963
      except OSError, err:
1964
        logging.exception("Cannot remove file storage directory '%s'",
1965
                          file_storage_dir)
1966
        result = False,
1967
  return result
1968

    
1969

    
1970
def RenameFileStorageDir(old_file_storage_dir, new_file_storage_dir):
1971
  """Rename the file storage directory.
1972

1973
  @type old_file_storage_dir: str
1974
  @param old_file_storage_dir: the current path
1975
  @type new_file_storage_dir: str
1976
  @param new_file_storage_dir: the name we should rename to
1977
  @rtype: tuple (success,)
1978
  @return: tuple of one element, C{success}, denoting
1979
      whether the operation was successful
1980

1981
  """
1982
  old_file_storage_dir = _TransformFileStorageDir(old_file_storage_dir)
1983
  new_file_storage_dir = _TransformFileStorageDir(new_file_storage_dir)
1984
  result = True,
1985
  if not old_file_storage_dir or not new_file_storage_dir:
1986
    result = False,
1987
  else:
1988
    if not os.path.exists(new_file_storage_dir):
1989
      if os.path.isdir(old_file_storage_dir):
1990
        try:
1991
          os.rename(old_file_storage_dir, new_file_storage_dir)
1992
        except OSError, err:
1993
          logging.exception("Cannot rename '%s' to '%s'",
1994
                            old_file_storage_dir, new_file_storage_dir)
1995
          result =  False,
1996
      else:
1997
        logging.error("'%s' is not a directory", old_file_storage_dir)
1998
        result = False,
1999
    else:
2000
      if os.path.exists(old_file_storage_dir):
2001
        logging.error("Cannot rename '%s' to '%s'. Both locations exist.",
2002
                      old_file_storage_dir, new_file_storage_dir)
2003
        result = False,
2004
  return result
2005

    
2006

    
2007
def _IsJobQueueFile(file_name):
2008
  """Checks whether the given filename is in the queue directory.
2009

2010
  @type file_name: str
2011
  @param file_name: the file name we should check
2012
  @rtype: boolean
2013
  @return: whether the file is under the queue directory
2014

2015
  """
2016
  queue_dir = os.path.normpath(constants.QUEUE_DIR)
2017
  result = (os.path.commonprefix([queue_dir, file_name]) == queue_dir)
2018

    
2019
  if not result:
2020
    logging.error("'%s' is not a file in the queue directory",
2021
                  file_name)
2022

    
2023
  return result
2024

    
2025

    
2026
def JobQueueUpdate(file_name, content):
2027
  """Updates a file in the queue directory.
2028

2029
  This is just a wrapper over L{utils.WriteFile}, with proper
2030
  checking.
2031

2032
  @type file_name: str
2033
  @param file_name: the job file name
2034
  @type content: str
2035
  @param content: the new job contents
2036
  @rtype: boolean
2037
  @return: the success of the operation
2038

2039
  """
2040
  if not _IsJobQueueFile(file_name):
2041
    return False
2042

    
2043
  # Write and replace the file atomically
2044
  utils.WriteFile(file_name, data=_Decompress(content))
2045

    
2046
  return True
2047

    
2048

    
2049
def JobQueueRename(old, new):
2050
  """Renames a job queue file.
2051

2052
  This is just a wrapper over os.rename with proper checking.
2053

2054
  @type old: str
2055
  @param old: the old (actual) file name
2056
  @type new: str
2057
  @param new: the desired file name
2058
  @rtype: boolean
2059
  @return: the success of the operation
2060

2061
  """
2062
  if not (_IsJobQueueFile(old) and _IsJobQueueFile(new)):
2063
    return False
2064

    
2065
  utils.RenameFile(old, new, mkdir=True)
2066

    
2067
  return True
2068

    
2069

    
2070
def JobQueueSetDrainFlag(drain_flag):
2071
  """Set the drain flag for the queue.
2072

2073
  This will set or unset the queue drain flag.
2074

2075
  @type drain_flag: boolean
2076
  @param drain_flag: if True, will set the drain flag, otherwise reset it.
2077
  @rtype: boolean
2078
  @return: always True
2079
  @warning: the function always returns True
2080

2081
  """
2082
  if drain_flag:
2083
    utils.WriteFile(constants.JOB_QUEUE_DRAIN_FILE, data="", close=True)
2084
  else:
2085
    utils.RemoveFile(constants.JOB_QUEUE_DRAIN_FILE)
2086

    
2087
  return True
2088

    
2089

    
2090
def CloseBlockDevices(instance_name, disks):
2091
  """Closes the given block devices.
2092

2093
  This means they will be switched to secondary mode (in case of
2094
  DRBD).
2095

2096
  @param instance_name: if the argument is not empty, the symlinks
2097
      of this instance will be removed
2098
  @type disks: list of L{objects.Disk}
2099
  @param disks: the list of disks to be closed
2100
  @rtype: tuple (success, message)
2101
  @return: a tuple of success and message, where success
2102
      indicates the succes of the operation, and message
2103
      which will contain the error details in case we
2104
      failed
2105

2106
  """
2107
  bdevs = []
2108
  for cf in disks:
2109
    rd = _RecursiveFindBD(cf)
2110
    if rd is None:
2111
      return (False, "Can't find device %s" % cf)
2112
    bdevs.append(rd)
2113

    
2114
  msg = []
2115
  for rd in bdevs:
2116
    try:
2117
      rd.Close()
2118
    except errors.BlockDeviceError, err:
2119
      msg.append(str(err))
2120
  if msg:
2121
    return (False, "Can't make devices secondary: %s" % ",".join(msg))
2122
  else:
2123
    if instance_name:
2124
      _RemoveBlockDevLinks(instance_name, disks)
2125
    return (True, "All devices secondary")
2126

    
2127

    
2128
def ValidateHVParams(hvname, hvparams):
2129
  """Validates the given hypervisor parameters.
2130

2131
  @type hvname: string
2132
  @param hvname: the hypervisor name
2133
  @type hvparams: dict
2134
  @param hvparams: the hypervisor parameters to be validated
2135
  @rtype: tuple (success, message)
2136
  @return: a tuple of success and message, where success
2137
      indicates the succes of the operation, and message
2138
      which will contain the error details in case we
2139
      failed
2140

2141
  """
2142
  try:
2143
    hv_type = hypervisor.GetHypervisor(hvname)
2144
    hv_type.ValidateParameters(hvparams)
2145
    return (True, "Validation passed")
2146
  except errors.HypervisorError, err:
2147
    return (False, str(err))
2148

    
2149

    
2150
def DemoteFromMC():
2151
  """Demotes the current node from master candidate role.
2152

2153
  """
2154
  # try to ensure we're not the master by mistake
2155
  master, myself = ssconf.GetMasterAndMyself()
2156
  if master == myself:
2157
    return (False, "ssconf status shows I'm the master node, will not demote")
2158
  pid_file = utils.DaemonPidFileName(constants.MASTERD_PID)
2159
  if utils.IsProcessAlive(utils.ReadPidFile(pid_file)):
2160
    return (False, "The master daemon is running, will not demote")
2161
  try:
2162
    utils.CreateBackup(constants.CLUSTER_CONF_FILE)
2163
  except EnvironmentError, err:
2164
    if err.errno != errno.ENOENT:
2165
      return (False, "Error while backing up cluster file: %s" % str(err))
2166
  utils.RemoveFile(constants.CLUSTER_CONF_FILE)
2167
  return (True, "Done")
2168

    
2169

    
2170
def _FindDisks(nodes_ip, disks):
2171
  """Sets the physical ID on disks and returns the block devices.
2172

2173
  """
2174
  # set the correct physical ID
2175
  my_name = utils.HostInfo().name
2176
  for cf in disks:
2177
    cf.SetPhysicalID(my_name, nodes_ip)
2178

    
2179
  bdevs = []
2180

    
2181
  for cf in disks:
2182
    rd = _RecursiveFindBD(cf)
2183
    if rd is None:
2184
      return (False, "Can't find device %s" % cf)
2185
    bdevs.append(rd)
2186
  return (True, bdevs)
2187

    
2188

    
2189
def DrbdDisconnectNet(nodes_ip, disks):
2190
  """Disconnects the network on a list of drbd devices.
2191

2192
  """
2193
  status, bdevs = _FindDisks(nodes_ip, disks)
2194
  if not status:
2195
    return status, bdevs
2196

    
2197
  # disconnect disks
2198
  for rd in bdevs:
2199
    try:
2200
      rd.DisconnectNet()
2201
    except errors.BlockDeviceError, err:
2202
      logging.exception("Failed to go into standalone mode")
2203
      return (False, "Can't change network configuration: %s" % str(err))
2204
  return (True, "All disks are now disconnected")
2205

    
2206

    
2207
def DrbdAttachNet(nodes_ip, disks, instance_name, multimaster):
2208
  """Attaches the network on a list of drbd devices.
2209

2210
  """
2211
  status, bdevs = _FindDisks(nodes_ip, disks)
2212
  if not status:
2213
    return status, bdevs
2214

    
2215
  if multimaster:
2216
    for idx, rd in enumerate(bdevs):
2217
      try:
2218
        _SymlinkBlockDev(instance_name, rd.dev_path, idx)
2219
      except EnvironmentError, err:
2220
        return (False, "Can't create symlink: %s" % str(err))
2221
  # reconnect disks, switch to new master configuration and if
2222
  # needed primary mode
2223
  for rd in bdevs:
2224
    try:
2225
      rd.AttachNet(multimaster)
2226
    except errors.BlockDeviceError, err:
2227
      return (False, "Can't change network configuration: %s" % str(err))
2228
  # wait until the disks are connected; we need to retry the re-attach
2229
  # if the device becomes standalone, as this might happen if the one
2230
  # node disconnects and reconnects in a different mode before the
2231
  # other node reconnects; in this case, one or both of the nodes will
2232
  # decide it has wrong configuration and switch to standalone
2233
  RECONNECT_TIMEOUT = 2 * 60
2234
  sleep_time = 0.100 # start with 100 miliseconds
2235
  timeout_limit = time.time() + RECONNECT_TIMEOUT
2236
  while time.time() < timeout_limit:
2237
    all_connected = True
2238
    for rd in bdevs:
2239
      stats = rd.GetProcStatus()
2240
      if not (stats.is_connected or stats.is_in_resync):
2241
        all_connected = False
2242
      if stats.is_standalone:
2243
        # peer had different config info and this node became
2244
        # standalone, even though this should not happen with the
2245
        # new staged way of changing disk configs
2246
        try:
2247
          rd.ReAttachNet(multimaster)
2248
        except errors.BlockDeviceError, err:
2249
          return (False, "Can't change network configuration: %s" % str(err))
2250
    if all_connected:
2251
      break
2252
    time.sleep(sleep_time)
2253
    sleep_time = min(5, sleep_time * 1.5)
2254
  if not all_connected:
2255
    return (False, "Timeout in disk reconnecting")
2256
  if multimaster:
2257
    # change to primary mode
2258
    for rd in bdevs:
2259
      rd.Open()
2260
  if multimaster:
2261
    msg = "multi-master and primary"
2262
  else:
2263
    msg = "single-master"
2264
  return (True, "Disks are now configured as %s" % msg)
2265

    
2266

    
2267
def DrbdWaitSync(nodes_ip, disks):
2268
  """Wait until DRBDs have synchronized.
2269

2270
  """
2271
  status, bdevs = _FindDisks(nodes_ip, disks)
2272
  if not status:
2273
    return status, bdevs
2274

    
2275
  min_resync = 100
2276
  alldone = True
2277
  failure = False
2278
  for rd in bdevs:
2279
    stats = rd.GetProcStatus()
2280
    if not (stats.is_connected or stats.is_in_resync):
2281
      failure = True
2282
      break
2283
    alldone = alldone and (not stats.is_in_resync)
2284
    if stats.sync_percent is not None:
2285
      min_resync = min(min_resync, stats.sync_percent)
2286
  return (not failure, (alldone, min_resync))
2287

    
2288

    
2289
class HooksRunner(object):
2290
  """Hook runner.
2291

2292
  This class is instantiated on the node side (ganeti-noded) and not
2293
  on the master side.
2294

2295
  """
2296
  RE_MASK = re.compile("^[a-zA-Z0-9_-]+$")
2297

    
2298
  def __init__(self, hooks_base_dir=None):
2299
    """Constructor for hooks runner.
2300

2301
    @type hooks_base_dir: str or None
2302
    @param hooks_base_dir: if not None, this overrides the
2303
        L{constants.HOOKS_BASE_DIR} (useful for unittests)
2304

2305
    """
2306
    if hooks_base_dir is None:
2307
      hooks_base_dir = constants.HOOKS_BASE_DIR
2308
    self._BASE_DIR = hooks_base_dir
2309

    
2310
  @staticmethod
2311
  def ExecHook(script, env):
2312
    """Exec one hook script.
2313

2314
    @type script: str
2315
    @param script: the full path to the script
2316
    @type env: dict
2317
    @param env: the environment with which to exec the script
2318
    @rtype: tuple (success, message)
2319
    @return: a tuple of success and message, where success
2320
        indicates the succes of the operation, and message
2321
        which will contain the error details in case we
2322
        failed
2323

2324
    """
2325
    # exec the process using subprocess and log the output
2326
    fdstdin = None
2327
    try:
2328
      fdstdin = open("/dev/null", "r")
2329
      child = subprocess.Popen([script], stdin=fdstdin, stdout=subprocess.PIPE,
2330
                               stderr=subprocess.STDOUT, close_fds=True,
2331
                               shell=False, cwd="/", env=env)
2332
      output = ""
2333
      try:
2334
        output = child.stdout.read(4096)
2335
        child.stdout.close()
2336
      except EnvironmentError, err:
2337
        output += "Hook script error: %s" % str(err)
2338

    
2339
      while True:
2340
        try:
2341
          result = child.wait()
2342
          break
2343
        except EnvironmentError, err:
2344
          if err.errno == errno.EINTR:
2345
            continue
2346
          raise
2347
    finally:
2348
      # try not to leak fds
2349
      for fd in (fdstdin, ):
2350
        if fd is not None:
2351
          try:
2352
            fd.close()
2353
          except EnvironmentError, err:
2354
            # just log the error
2355
            #logging.exception("Error while closing fd %s", fd)
2356
            pass
2357

    
2358
    return result == 0, output
2359

    
2360
  def RunHooks(self, hpath, phase, env):
2361
    """Run the scripts in the hooks directory.
2362

2363
    @type hpath: str
2364
    @param hpath: the path to the hooks directory which
2365
        holds the scripts
2366
    @type phase: str
2367
    @param phase: either L{constants.HOOKS_PHASE_PRE} or
2368
        L{constants.HOOKS_PHASE_POST}
2369
    @type env: dict
2370
    @param env: dictionary with the environment for the hook
2371
    @rtype: list
2372
    @return: list of 3-element tuples:
2373
      - script path
2374
      - script result, either L{constants.HKR_SUCCESS} or
2375
        L{constants.HKR_FAIL}
2376
      - output of the script
2377

2378
    @raise errors.ProgrammerError: for invalid input
2379
        parameters
2380

2381
    """
2382
    if phase == constants.HOOKS_PHASE_PRE:
2383
      suffix = "pre"
2384
    elif phase == constants.HOOKS_PHASE_POST:
2385
      suffix = "post"
2386
    else:
2387
      raise errors.ProgrammerError("Unknown hooks phase: '%s'" % phase)
2388
    rr = []
2389

    
2390
    subdir = "%s-%s.d" % (hpath, suffix)
2391
    dir_name = "%s/%s" % (self._BASE_DIR, subdir)
2392
    try:
2393
      dir_contents = utils.ListVisibleFiles(dir_name)
2394
    except OSError, err:
2395
      # FIXME: must log output in case of failures
2396
      return rr
2397

    
2398
    # we use the standard python sort order,
2399
    # so 00name is the recommended naming scheme
2400
    dir_contents.sort()
2401
    for relname in dir_contents:
2402
      fname = os.path.join(dir_name, relname)
2403
      if not (os.path.isfile(fname) and os.access(fname, os.X_OK) and
2404
          self.RE_MASK.match(relname) is not None):
2405
        rrval = constants.HKR_SKIP
2406
        output = ""
2407
      else:
2408
        result, output = self.ExecHook(fname, env)
2409
        if not result:
2410
          rrval = constants.HKR_FAIL
2411
        else:
2412
          rrval = constants.HKR_SUCCESS
2413
      rr.append(("%s/%s" % (subdir, relname), rrval, output))
2414

    
2415
    return rr
2416

    
2417

    
2418
class IAllocatorRunner(object):
2419
  """IAllocator runner.
2420

2421
  This class is instantiated on the node side (ganeti-noded) and not on
2422
  the master side.
2423

2424
  """
2425
  def Run(self, name, idata):
2426
    """Run an iallocator script.
2427

2428
    @type name: str
2429
    @param name: the iallocator script name
2430
    @type idata: str
2431
    @param idata: the allocator input data
2432

2433
    @rtype: tuple
2434
    @return: four element tuple of:
2435
       - run status (one of the IARUN_ constants)
2436
       - stdout
2437
       - stderr
2438
       - fail reason (as from L{utils.RunResult})
2439

2440
    """
2441
    alloc_script = utils.FindFile(name, constants.IALLOCATOR_SEARCH_PATH,
2442
                                  os.path.isfile)
2443
    if alloc_script is None:
2444
      return (constants.IARUN_NOTFOUND, None, None, None)
2445

    
2446
    fd, fin_name = tempfile.mkstemp(prefix="ganeti-iallocator.")
2447
    try:
2448
      os.write(fd, idata)
2449
      os.close(fd)
2450
      result = utils.RunCmd([alloc_script, fin_name])
2451
      if result.failed:
2452
        return (constants.IARUN_FAILURE, result.stdout, result.stderr,
2453
                result.fail_reason)
2454
    finally:
2455
      os.unlink(fin_name)
2456

    
2457
    return (constants.IARUN_SUCCESS, result.stdout, result.stderr, None)
2458

    
2459

    
2460
class DevCacheManager(object):
2461
  """Simple class for managing a cache of block device information.
2462

2463
  """
2464
  _DEV_PREFIX = "/dev/"
2465
  _ROOT_DIR = constants.BDEV_CACHE_DIR
2466

    
2467
  @classmethod
2468
  def _ConvertPath(cls, dev_path):
2469
    """Converts a /dev/name path to the cache file name.
2470

2471
    This replaces slashes with underscores and strips the /dev
2472
    prefix. It then returns the full path to the cache file.
2473

2474
    @type dev_path: str
2475
    @param dev_path: the C{/dev/} path name
2476
    @rtype: str
2477
    @return: the converted path name
2478

2479
    """
2480
    if dev_path.startswith(cls._DEV_PREFIX):
2481
      dev_path = dev_path[len(cls._DEV_PREFIX):]
2482
    dev_path = dev_path.replace("/", "_")
2483
    fpath = "%s/bdev_%s" % (cls._ROOT_DIR, dev_path)
2484
    return fpath
2485

    
2486
  @classmethod
2487
  def UpdateCache(cls, dev_path, owner, on_primary, iv_name):
2488
    """Updates the cache information for a given device.
2489

2490
    @type dev_path: str
2491
    @param dev_path: the pathname of the device
2492
    @type owner: str
2493
    @param owner: the owner (instance name) of the device
2494
    @type on_primary: bool
2495
    @param on_primary: whether this is the primary
2496
        node nor not
2497
    @type iv_name: str
2498
    @param iv_name: the instance-visible name of the
2499
        device, as in objects.Disk.iv_name
2500

2501
    @rtype: None
2502

2503
    """
2504
    if dev_path is None:
2505
      logging.error("DevCacheManager.UpdateCache got a None dev_path")
2506
      return
2507
    fpath = cls._ConvertPath(dev_path)
2508
    if on_primary:
2509
      state = "primary"
2510
    else:
2511
      state = "secondary"
2512
    if iv_name is None:
2513
      iv_name = "not_visible"
2514
    fdata = "%s %s %s\n" % (str(owner), state, iv_name)
2515
    try:
2516
      utils.WriteFile(fpath, data=fdata)
2517
    except EnvironmentError, err:
2518
      logging.exception("Can't update bdev cache for %s", dev_path)
2519

    
2520
  @classmethod
2521
  def RemoveCache(cls, dev_path):
2522
    """Remove data for a dev_path.
2523

2524
    This is just a wrapper over L{utils.RemoveFile} with a converted
2525
    path name and logging.
2526

2527
    @type dev_path: str
2528
    @param dev_path: the pathname of the device
2529

2530
    @rtype: None
2531

2532
    """
2533
    if dev_path is None:
2534
      logging.error("DevCacheManager.RemoveCache got a None dev_path")
2535
      return
2536
    fpath = cls._ConvertPath(dev_path)
2537
    try:
2538
      utils.RemoveFile(fpath)
2539
    except EnvironmentError, err:
2540
      logging.exception("Can't update bdev cache for %s", dev_path)