Statistics
| Branch: | Tag: | Revision:

root / lib / backend.py @ c3f0a12f

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

    
36
from ganeti import errors
37
from ganeti import utils
38
from ganeti import ssh
39
from ganeti import hypervisor
40
from ganeti import constants
41
from ganeti import bdev
42
from ganeti import objects
43
from ganeti import ssconf
44

    
45

    
46
def _GetSshRunner():
47
  return ssh.SshRunner()
48

    
49

    
50
def StartMaster():
51
  """Activate local node as master node.
52

53
  There are two needed steps for this:
54
    - run the master script
55
    - register the cron script
56

57
  """
58
  result = utils.RunCmd([constants.MASTER_SCRIPT, "-d", "start"])
59

    
60
  if result.failed:
61
    logging.error("could not activate cluster interface with command %s,"
62
                  " error: '%s'", result.cmd, result.output)
63
    return False
64

    
65
  return True
66

    
67

    
68
def StopMaster():
69
  """Deactivate this node as master.
70

71
  This runs the master stop script.
72

73
  """
74
  result = utils.RunCmd([constants.MASTER_SCRIPT, "-d", "stop"])
75

    
76
  if result.failed:
77
    logging.error("could not deactivate cluster interface with command %s,"
78
                  " error: '%s'", result.cmd, result.output)
79
    return False
80

    
81
  return True
82

    
83

    
84
def AddNode(dsa, dsapub, rsa, rsapub, sshkey, sshpub):
85
  """Joins this node to the cluster.
86

87
  This does the following:
88
      - updates the hostkeys of the machine (rsa and dsa)
89
      - adds the ssh private key to the user
90
      - adds the ssh public key to the users' authorized_keys file
91

92
  """
93
  sshd_keys =  [(constants.SSH_HOST_RSA_PRIV, rsa, 0600),
94
                (constants.SSH_HOST_RSA_PUB, rsapub, 0644),
95
                (constants.SSH_HOST_DSA_PRIV, dsa, 0600),
96
                (constants.SSH_HOST_DSA_PUB, dsapub, 0644)]
97
  for name, content, mode in sshd_keys:
98
    utils.WriteFile(name, data=content, mode=mode)
99

    
100
  try:
101
    priv_key, pub_key, auth_keys = ssh.GetUserFiles(constants.GANETI_RUNAS,
102
                                                    mkdir=True)
103
  except errors.OpExecError, err:
104
    logging.exception("Error while processing user ssh files")
105
    return False
106

    
107
  for name, content in [(priv_key, sshkey), (pub_key, sshpub)]:
108
    utils.WriteFile(name, data=content, mode=0600)
109

    
110
  utils.AddAuthorizedKey(auth_keys, sshpub)
111

    
112
  utils.RunCmd([constants.SSH_INITD_SCRIPT, "restart"])
113

    
114
  return True
115

    
116

    
117
def LeaveCluster():
118
  """Cleans up the current node and prepares it to be removed from the cluster.
119

120
  """
121
  if os.path.isdir(constants.DATA_DIR):
122
    for rel_name in utils.ListVisibleFiles(constants.DATA_DIR):
123
      full_name = os.path.join(constants.DATA_DIR, rel_name)
124
      if os.path.isfile(full_name) and not os.path.islink(full_name):
125
        utils.RemoveFile(full_name)
126

    
127
  try:
128
    priv_key, pub_key, auth_keys = ssh.GetUserFiles(constants.GANETI_RUNAS)
129
  except errors.OpExecError:
130
    logging.exception("Error while processing ssh files")
131
    return
132

    
133
  f = open(pub_key, 'r')
134
  try:
135
    utils.RemoveAuthorizedKey(auth_keys, f.read(8192))
136
  finally:
137
    f.close()
138

    
139
  utils.RemoveFile(priv_key)
140
  utils.RemoveFile(pub_key)
141

    
142
  # Return a reassuring string to the caller, and quit
143
  raise errors.QuitGanetiException(False, 'Shutdown scheduled')
144

    
145

    
146
def GetNodeInfo(vgname):
147
  """Gives back a hash with different informations about the node.
148

149
  Returns:
150
    { 'vg_size' : xxx,  'vg_free' : xxx, 'memory_domain0': xxx,
151
      'memory_free' : xxx, 'memory_total' : xxx }
152
    where
153
    vg_size is the size of the configured volume group in MiB
154
    vg_free is the free size of the volume group in MiB
155
    memory_dom0 is the memory allocated for domain0 in MiB
156
    memory_free is the currently available (free) ram in MiB
157
    memory_total is the total number of ram in MiB
158

159
  """
160
  outputarray = {}
161
  vginfo = _GetVGInfo(vgname)
162
  outputarray['vg_size'] = vginfo['vg_size']
163
  outputarray['vg_free'] = vginfo['vg_free']
164

    
165
  hyper = hypervisor.GetHypervisor()
166
  hyp_info = hyper.GetNodeInfo()
167
  if hyp_info is not None:
168
    outputarray.update(hyp_info)
169

    
170
  f = open("/proc/sys/kernel/random/boot_id", 'r')
171
  try:
172
    outputarray["bootid"] = f.read(128).rstrip("\n")
173
  finally:
174
    f.close()
175

    
176
  return outputarray
177

    
178

    
179
def VerifyNode(what):
180
  """Verify the status of the local node.
181

182
  Args:
183
    what - a dictionary of things to check:
184
      'filelist' : list of files for which to compute checksums
185
      'nodelist' : list of nodes we should check communication with
186
      'hypervisor': run the hypervisor-specific verify
187

188
  Requested files on local node are checksummed and the result returned.
189

190
  The nodelist is traversed, with the following checks being made
191
  for each node:
192
  - known_hosts key correct
193
  - correct resolving of node name (target node returns its own hostname
194
    by ssh-execution of 'hostname', result compared against name in list.
195

196
  """
197
  result = {}
198

    
199
  if 'hypervisor' in what:
200
    result['hypervisor'] = hypervisor.GetHypervisor().Verify()
201

    
202
  if 'filelist' in what:
203
    result['filelist'] = utils.FingerprintFiles(what['filelist'])
204

    
205
  if 'nodelist' in what:
206
    result['nodelist'] = {}
207
    random.shuffle(what['nodelist'])
208
    for node in what['nodelist']:
209
      success, message = _GetSshRunner().VerifyNodeHostname(node)
210
      if not success:
211
        result['nodelist'][node] = message
212
  if 'node-net-test' in what:
213
    result['node-net-test'] = {}
214
    my_name = utils.HostInfo().name
215
    my_pip = my_sip = None
216
    for name, pip, sip in what['node-net-test']:
217
      if name == my_name:
218
        my_pip = pip
219
        my_sip = sip
220
        break
221
    if not my_pip:
222
      result['node-net-test'][my_name] = ("Can't find my own"
223
                                          " primary/secondary IP"
224
                                          " in the node list")
225
    else:
226
      port = ssconf.SimpleStore().GetNodeDaemonPort()
227
      for name, pip, sip in what['node-net-test']:
228
        fail = []
229
        if not utils.TcpPing(pip, port, source=my_pip):
230
          fail.append("primary")
231
        if sip != pip:
232
          if not utils.TcpPing(sip, port, source=my_sip):
233
            fail.append("secondary")
234
        if fail:
235
          result['node-net-test'][name] = ("failure using the %s"
236
                                           " interface(s)" %
237
                                           " and ".join(fail))
238

    
239
  return result
240

    
241

    
242
def GetVolumeList(vg_name):
243
  """Compute list of logical volumes and their size.
244

245
  Returns:
246
    dictionary of all partions (key) with their size (in MiB), inactive
247
    and online status:
248
    {'test1': ('20.06', True, True)}
249

250
  """
251
  lvs = {}
252
  sep = '|'
253
  result = utils.RunCmd(["lvs", "--noheadings", "--units=m", "--nosuffix",
254
                         "--separator=%s" % sep,
255
                         "-olv_name,lv_size,lv_attr", vg_name])
256
  if result.failed:
257
    logging.error("Failed to list logical volumes, lvs output: %s",
258
                  result.output)
259
    return result.output
260

    
261
  valid_line_re = re.compile("^ *([^|]+)\|([0-9.]+)\|([^|]{6})\|?$")
262
  for line in result.stdout.splitlines():
263
    line = line.strip()
264
    match = valid_line_re.match(line)
265
    if not match:
266
      logging.error("Invalid line returned from lvs output: '%s'", line)
267
      continue
268
    name, size, attr = match.groups()
269
    inactive = attr[4] == '-'
270
    online = attr[5] == 'o'
271
    lvs[name] = (size, inactive, online)
272

    
273
  return lvs
274

    
275

    
276
def ListVolumeGroups():
277
  """List the volume groups and their size.
278

279
  Returns:
280
    Dictionary with keys volume name and values the size of the volume
281

282
  """
283
  return utils.ListVolumeGroups()
284

    
285

    
286
def NodeVolumes():
287
  """List all volumes on this node.
288

289
  """
290
  result = utils.RunCmd(["lvs", "--noheadings", "--units=m", "--nosuffix",
291
                         "--separator=|",
292
                         "--options=lv_name,lv_size,devices,vg_name"])
293
  if result.failed:
294
    logging.error("Failed to list logical volumes, lvs output: %s",
295
                  result.output)
296
    return {}
297

    
298
  def parse_dev(dev):
299
    if '(' in dev:
300
      return dev.split('(')[0]
301
    else:
302
      return dev
303

    
304
  def map_line(line):
305
    return {
306
      'name': line[0].strip(),
307
      'size': line[1].strip(),
308
      'dev': parse_dev(line[2].strip()),
309
      'vg': line[3].strip(),
310
    }
311

    
312
  return [map_line(line.split('|')) for line in result.stdout.splitlines()
313
          if line.count('|') >= 3]
314

    
315

    
316
def BridgesExist(bridges_list):
317
  """Check if a list of bridges exist on the current node.
318

319
  Returns:
320
    True if all of them exist, false otherwise
321

322
  """
323
  for bridge in bridges_list:
324
    if not utils.BridgeExists(bridge):
325
      return False
326

    
327
  return True
328

    
329

    
330
def GetInstanceList():
331
  """Provides a list of instances.
332

333
  Returns:
334
    A list of all running instances on the current node
335
    - instance1.example.com
336
    - instance2.example.com
337

338
  """
339
  try:
340
    names = hypervisor.GetHypervisor().ListInstances()
341
  except errors.HypervisorError, err:
342
    logging.exception("Error enumerating instances")
343
    raise
344

    
345
  return names
346

    
347

    
348
def GetInstanceInfo(instance):
349
  """Gives back the informations about an instance as a dictionary.
350

351
  Args:
352
    instance: name of the instance (ex. instance1.example.com)
353

354
  Returns:
355
    { 'memory' : 511, 'state' : '-b---', 'time' : 3188.8, }
356
    where
357
    memory: memory size of instance (int)
358
    state: xen state of instance (string)
359
    time: cpu time of instance (float)
360

361
  """
362
  output = {}
363

    
364
  iinfo = hypervisor.GetHypervisor().GetInstanceInfo(instance)
365
  if iinfo is not None:
366
    output['memory'] = iinfo[2]
367
    output['state'] = iinfo[4]
368
    output['time'] = iinfo[5]
369

    
370
  return output
371

    
372

    
373
def GetAllInstancesInfo():
374
  """Gather data about all instances.
375

376
  This is the equivalent of `GetInstanceInfo()`, except that it
377
  computes data for all instances at once, thus being faster if one
378
  needs data about more than one instance.
379

380
  Returns: a dictionary of dictionaries, keys being the instance name,
381
    and with values:
382
    { 'memory' : 511, 'state' : '-b---', 'time' : 3188.8, }
383
    where
384
    memory: memory size of instance (int)
385
    state: xen state of instance (string)
386
    time: cpu time of instance (float)
387
    vcpus: the number of cpus
388

389
  """
390
  output = {}
391

    
392
  iinfo = hypervisor.GetHypervisor().GetAllInstancesInfo()
393
  if iinfo:
394
    for name, inst_id, memory, vcpus, state, times in iinfo:
395
      output[name] = {
396
        'memory': memory,
397
        'vcpus': vcpus,
398
        'state': state,
399
        'time': times,
400
        }
401

    
402
  return output
403

    
404

    
405
def AddOSToInstance(instance, os_disk, swap_disk):
406
  """Add an OS to an instance.
407

408
  Args:
409
    instance: the instance object
410
    os_disk: the instance-visible name of the os device
411
    swap_disk: the instance-visible name of the swap device
412

413
  """
414
  inst_os = OSFromDisk(instance.os)
415

    
416
  create_script = inst_os.create_script
417

    
418
  os_device = instance.FindDisk(os_disk)
419
  if os_device is None:
420
    logging.error("Can't find this device-visible name '%s'", os_disk)
421
    return False
422

    
423
  swap_device = instance.FindDisk(swap_disk)
424
  if swap_device is None:
425
    logging.error("Can't find this device-visible name '%s'", swap_disk)
426
    return False
427

    
428
  real_os_dev = _RecursiveFindBD(os_device)
429
  if real_os_dev is None:
430
    raise errors.BlockDeviceError("Block device '%s' is not set up" %
431
                                  str(os_device))
432
  real_os_dev.Open()
433

    
434
  real_swap_dev = _RecursiveFindBD(swap_device)
435
  if real_swap_dev is None:
436
    raise errors.BlockDeviceError("Block device '%s' is not set up" %
437
                                  str(swap_device))
438
  real_swap_dev.Open()
439

    
440
  logfile = "%s/add-%s-%s-%d.log" % (constants.LOG_OS_DIR, instance.os,
441
                                     instance.name, int(time.time()))
442
  if not os.path.exists(constants.LOG_OS_DIR):
443
    os.mkdir(constants.LOG_OS_DIR, 0750)
444

    
445
  command = utils.BuildShellCmd("cd %s && %s -i %s -b %s -s %s &>%s",
446
                                inst_os.path, create_script, instance.name,
447
                                real_os_dev.dev_path, real_swap_dev.dev_path,
448
                                logfile)
449

    
450
  result = utils.RunCmd(command)
451
  if result.failed:
452
    logging.error("os create command '%s' returned error: %s, logfile: %s,"
453
                  " output: %s", command, result.fail_reason, logfile,
454
                  result.output)
455
    return False
456

    
457
  return True
458

    
459

    
460
def RunRenameInstance(instance, old_name, os_disk, swap_disk):
461
  """Run the OS rename script for an instance.
462

463
  Args:
464
    instance: the instance object
465
    old_name: the old name of the instance
466
    os_disk: the instance-visible name of the os device
467
    swap_disk: the instance-visible name of the swap device
468

469
  """
470
  inst_os = OSFromDisk(instance.os)
471

    
472
  script = inst_os.rename_script
473

    
474
  os_device = instance.FindDisk(os_disk)
475
  if os_device is None:
476
    logging.error("Can't find this device-visible name '%s'", os_disk)
477
    return False
478

    
479
  swap_device = instance.FindDisk(swap_disk)
480
  if swap_device is None:
481
    logging.error("Can't find this device-visible name '%s'", swap_disk)
482
    return False
483

    
484
  real_os_dev = _RecursiveFindBD(os_device)
485
  if real_os_dev is None:
486
    raise errors.BlockDeviceError("Block device '%s' is not set up" %
487
                                  str(os_device))
488
  real_os_dev.Open()
489

    
490
  real_swap_dev = _RecursiveFindBD(swap_device)
491
  if real_swap_dev is None:
492
    raise errors.BlockDeviceError("Block device '%s' is not set up" %
493
                                  str(swap_device))
494
  real_swap_dev.Open()
495

    
496
  logfile = "%s/rename-%s-%s-%s-%d.log" % (constants.LOG_OS_DIR, instance.os,
497
                                           old_name,
498
                                           instance.name, int(time.time()))
499
  if not os.path.exists(constants.LOG_OS_DIR):
500
    os.mkdir(constants.LOG_OS_DIR, 0750)
501

    
502
  command = utils.BuildShellCmd("cd %s && %s -o %s -n %s -b %s -s %s &>%s",
503
                                inst_os.path, script, old_name, instance.name,
504
                                real_os_dev.dev_path, real_swap_dev.dev_path,
505
                                logfile)
506

    
507
  result = utils.RunCmd(command)
508

    
509
  if result.failed:
510
    logging.error("os create command '%s' returned error: %s output: %s",
511
                  command, result.fail_reason, result.output)
512
    return False
513

    
514
  return True
515

    
516

    
517
def _GetVGInfo(vg_name):
518
  """Get informations about the volume group.
519

520
  Args:
521
    vg_name: the volume group
522

523
  Returns:
524
    { 'vg_size' : xxx, 'vg_free' : xxx, 'pv_count' : xxx }
525
    where
526
    vg_size is the total size of the volume group in MiB
527
    vg_free is the free size of the volume group in MiB
528
    pv_count are the number of physical disks in that vg
529

530
  If an error occurs during gathering of data, we return the same dict
531
  with keys all set to None.
532

533
  """
534
  retdic = dict.fromkeys(["vg_size", "vg_free", "pv_count"])
535

    
536
  retval = utils.RunCmd(["vgs", "-ovg_size,vg_free,pv_count", "--noheadings",
537
                         "--nosuffix", "--units=m", "--separator=:", vg_name])
538

    
539
  if retval.failed:
540
    logging.error("volume group %s not present", vg_name)
541
    return retdic
542
  valarr = retval.stdout.strip().rstrip(':').split(':')
543
  if len(valarr) == 3:
544
    try:
545
      retdic = {
546
        "vg_size": int(round(float(valarr[0]), 0)),
547
        "vg_free": int(round(float(valarr[1]), 0)),
548
        "pv_count": int(valarr[2]),
549
        }
550
    except ValueError, err:
551
      logging.exception("Fail to parse vgs output")
552
  else:
553
    logging.error("vgs output has the wrong number of fields (expected"
554
                  " three): %s", str(valarr))
555
  return retdic
556

    
557

    
558
def _GatherBlockDevs(instance):
559
  """Set up an instance's block device(s).
560

561
  This is run on the primary node at instance startup. The block
562
  devices must be already assembled.
563

564
  """
565
  block_devices = []
566
  for disk in instance.disks:
567
    device = _RecursiveFindBD(disk)
568
    if device is None:
569
      raise errors.BlockDeviceError("Block device '%s' is not set up." %
570
                                    str(disk))
571
    device.Open()
572
    block_devices.append((disk, device))
573
  return block_devices
574

    
575

    
576
def StartInstance(instance, extra_args):
577
  """Start an instance.
578

579
  Args:
580
    instance - name of instance to start.
581

582
  """
583
  running_instances = GetInstanceList()
584

    
585
  if instance.name in running_instances:
586
    return True
587

    
588
  block_devices = _GatherBlockDevs(instance)
589
  hyper = hypervisor.GetHypervisor()
590

    
591
  try:
592
    hyper.StartInstance(instance, block_devices, extra_args)
593
  except errors.HypervisorError, err:
594
    logging.exception("Failed to start instance")
595
    return False
596

    
597
  return True
598

    
599

    
600
def ShutdownInstance(instance):
601
  """Shut an instance down.
602

603
  Args:
604
    instance - name of instance to shutdown.
605

606
  """
607
  running_instances = GetInstanceList()
608

    
609
  if instance.name not in running_instances:
610
    return True
611

    
612
  hyper = hypervisor.GetHypervisor()
613
  try:
614
    hyper.StopInstance(instance)
615
  except errors.HypervisorError, err:
616
    logging.error("Failed to stop instance")
617
    return False
618

    
619
  # test every 10secs for 2min
620
  shutdown_ok = False
621

    
622
  time.sleep(1)
623
  for dummy in range(11):
624
    if instance.name not in GetInstanceList():
625
      break
626
    time.sleep(10)
627
  else:
628
    # the shutdown did not succeed
629
    logging.error("shutdown of '%s' unsuccessful, using destroy", instance)
630

    
631
    try:
632
      hyper.StopInstance(instance, force=True)
633
    except errors.HypervisorError, err:
634
      logging.exception("Failed to stop instance")
635
      return False
636

    
637
    time.sleep(1)
638
    if instance.name in GetInstanceList():
639
      logging.error("could not shutdown instance '%s' even by destroy",
640
                    instance.name)
641
      return False
642

    
643
  return True
644

    
645

    
646
def RebootInstance(instance, reboot_type, extra_args):
647
  """Reboot an instance.
648

649
  Args:
650
    instance    - name of instance to reboot
651
    reboot_type - how to reboot [soft,hard,full]
652

653
  """
654
  running_instances = GetInstanceList()
655

    
656
  if instance.name not in running_instances:
657
    logging.error("Cannot reboot instance that is not running")
658
    return False
659

    
660
  hyper = hypervisor.GetHypervisor()
661
  if reboot_type == constants.INSTANCE_REBOOT_SOFT:
662
    try:
663
      hyper.RebootInstance(instance)
664
    except errors.HypervisorError, err:
665
      logging.exception("Failed to soft reboot instance")
666
      return False
667
  elif reboot_type == constants.INSTANCE_REBOOT_HARD:
668
    try:
669
      ShutdownInstance(instance)
670
      StartInstance(instance, extra_args)
671
    except errors.HypervisorError, err:
672
      logging.exception("Failed to hard reboot instance")
673
      return False
674
  else:
675
    raise errors.ParameterError("reboot_type invalid")
676

    
677

    
678
  return True
679

    
680

    
681
def MigrateInstance(instance, target, live):
682
  """Migrates an instance to another node.
683

684
  """
685
  hyper = hypervisor.GetHypervisor()
686

    
687
  try:
688
    hyper.MigrateInstance(instance, target, live)
689
  except errors.HypervisorError, err:
690
    msg = "Failed to migrate instance: %s" % str(err)
691
    logging.error(msg)
692
    return (False, msg)
693
  return (True, "Migration successfull")
694

    
695

    
696
def CreateBlockDevice(disk, size, owner, on_primary, info):
697
  """Creates a block device for an instance.
698

699
  Args:
700
   disk: a ganeti.objects.Disk object
701
   size: the size of the physical underlying device
702
   owner: a string with the name of the instance
703
   on_primary: a boolean indicating if it is the primary node or not
704
   info: string that will be sent to the physical device creation
705

706
  Returns:
707
    the new unique_id of the device (this can sometime be
708
    computed only after creation), or None. On secondary nodes,
709
    it's not required to return anything.
710

711
  """
712
  clist = []
713
  if disk.children:
714
    for child in disk.children:
715
      crdev = _RecursiveAssembleBD(child, owner, on_primary)
716
      if on_primary or disk.AssembleOnSecondary():
717
        # we need the children open in case the device itself has to
718
        # be assembled
719
        crdev.Open()
720
      clist.append(crdev)
721
  try:
722
    device = bdev.FindDevice(disk.dev_type, disk.physical_id, clist)
723
    if device is not None:
724
      logging.info("removing existing device %s", disk)
725
      device.Remove()
726
  except errors.BlockDeviceError, err:
727
    pass
728

    
729
  device = bdev.Create(disk.dev_type, disk.physical_id,
730
                       clist, size)
731
  if device is None:
732
    raise ValueError("Can't create child device for %s, %s" %
733
                     (disk, size))
734
  if on_primary or disk.AssembleOnSecondary():
735
    if not device.Assemble():
736
      errorstring = "Can't assemble device after creation"
737
      logging.error(errorstring)
738
      raise errors.BlockDeviceError("%s, very unusual event - check the node"
739
                                    " daemon logs" % errorstring)
740
    device.SetSyncSpeed(constants.SYNC_SPEED)
741
    if on_primary or disk.OpenOnSecondary():
742
      device.Open(force=True)
743
    DevCacheManager.UpdateCache(device.dev_path, owner,
744
                                on_primary, disk.iv_name)
745

    
746
  device.SetInfo(info)
747

    
748
  physical_id = device.unique_id
749
  return physical_id
750

    
751

    
752
def RemoveBlockDevice(disk):
753
  """Remove a block device.
754

755
  This is intended to be called recursively.
756

757
  """
758
  try:
759
    # since we are removing the device, allow a partial match
760
    # this allows removal of broken mirrors
761
    rdev = _RecursiveFindBD(disk, allow_partial=True)
762
  except errors.BlockDeviceError, err:
763
    # probably can't attach
764
    logging.info("Can't attach to device %s in remove", disk)
765
    rdev = None
766
  if rdev is not None:
767
    r_path = rdev.dev_path
768
    result = rdev.Remove()
769
    if result:
770
      DevCacheManager.RemoveCache(r_path)
771
  else:
772
    result = True
773
  if disk.children:
774
    for child in disk.children:
775
      result = result and RemoveBlockDevice(child)
776
  return result
777

    
778

    
779
def _RecursiveAssembleBD(disk, owner, as_primary):
780
  """Activate a block device for an instance.
781

782
  This is run on the primary and secondary nodes for an instance.
783

784
  This function is called recursively.
785

786
  Args:
787
    disk: a objects.Disk object
788
    as_primary: if we should make the block device read/write
789

790
  Returns:
791
    the assembled device or None (in case no device was assembled)
792

793
  If the assembly is not successful, an exception is raised.
794

795
  """
796
  children = []
797
  if disk.children:
798
    mcn = disk.ChildrenNeeded()
799
    if mcn == -1:
800
      mcn = 0 # max number of Nones allowed
801
    else:
802
      mcn = len(disk.children) - mcn # max number of Nones
803
    for chld_disk in disk.children:
804
      try:
805
        cdev = _RecursiveAssembleBD(chld_disk, owner, as_primary)
806
      except errors.BlockDeviceError, err:
807
        if children.count(None) >= mcn:
808
          raise
809
        cdev = None
810
        logging.debug("Error in child activation: %s", str(err))
811
      children.append(cdev)
812

    
813
  if as_primary or disk.AssembleOnSecondary():
814
    r_dev = bdev.AttachOrAssemble(disk.dev_type, disk.physical_id, children)
815
    r_dev.SetSyncSpeed(constants.SYNC_SPEED)
816
    result = r_dev
817
    if as_primary or disk.OpenOnSecondary():
818
      r_dev.Open()
819
    DevCacheManager.UpdateCache(r_dev.dev_path, owner,
820
                                as_primary, disk.iv_name)
821

    
822
  else:
823
    result = True
824
  return result
825

    
826

    
827
def AssembleBlockDevice(disk, owner, as_primary):
828
  """Activate a block device for an instance.
829

830
  This is a wrapper over _RecursiveAssembleBD.
831

832
  Returns:
833
    a /dev path for primary nodes
834
    True for secondary nodes
835

836
  """
837
  result = _RecursiveAssembleBD(disk, owner, as_primary)
838
  if isinstance(result, bdev.BlockDev):
839
    result = result.dev_path
840
  return result
841

    
842

    
843
def ShutdownBlockDevice(disk):
844
  """Shut down a block device.
845

846
  First, if the device is assembled (can `Attach()`), then the device
847
  is shutdown. Then the children of the device are shutdown.
848

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

853
  """
854
  r_dev = _RecursiveFindBD(disk)
855
  if r_dev is not None:
856
    r_path = r_dev.dev_path
857
    result = r_dev.Shutdown()
858
    if result:
859
      DevCacheManager.RemoveCache(r_path)
860
  else:
861
    result = True
862
  if disk.children:
863
    for child in disk.children:
864
      result = result and ShutdownBlockDevice(child)
865
  return result
866

    
867

    
868
def MirrorAddChildren(parent_cdev, new_cdevs):
869
  """Extend a mirrored block device.
870

871
  """
872
  parent_bdev = _RecursiveFindBD(parent_cdev, allow_partial=True)
873
  if parent_bdev is None:
874
    logging.error("Can't find parent device")
875
    return False
876
  new_bdevs = [_RecursiveFindBD(disk) for disk in new_cdevs]
877
  if new_bdevs.count(None) > 0:
878
    logging.error("Can't find new device(s) to add: %s:%s",
879
                  new_bdevs, new_cdevs)
880
    return False
881
  parent_bdev.AddChildren(new_bdevs)
882
  return True
883

    
884

    
885
def MirrorRemoveChildren(parent_cdev, new_cdevs):
886
  """Shrink a mirrored block device.
887

888
  """
889
  parent_bdev = _RecursiveFindBD(parent_cdev)
890
  if parent_bdev is None:
891
    logging.error("Can't find parent in remove children: %s", parent_cdev)
892
    return False
893
  devs = []
894
  for disk in new_cdevs:
895
    rpath = disk.StaticDevPath()
896
    if rpath is None:
897
      bd = _RecursiveFindBD(disk)
898
      if bd is None:
899
        logging.error("Can't find dynamic device %s while removing children",
900
                      disk)
901
        return False
902
      else:
903
        devs.append(bd.dev_path)
904
    else:
905
      devs.append(rpath)
906
  parent_bdev.RemoveChildren(devs)
907
  return True
908

    
909

    
910
def GetMirrorStatus(disks):
911
  """Get the mirroring status of a list of devices.
912

913
  Args:
914
    disks: list of `objects.Disk`
915

916
  Returns:
917
    list of (mirror_done, estimated_time) tuples, which
918
    are the result of bdev.BlockDevice.CombinedSyncStatus()
919

920
  """
921
  stats = []
922
  for dsk in disks:
923
    rbd = _RecursiveFindBD(dsk)
924
    if rbd is None:
925
      raise errors.BlockDeviceError("Can't find device %s" % str(dsk))
926
    stats.append(rbd.CombinedSyncStatus())
927
  return stats
928

    
929

    
930
def _RecursiveFindBD(disk, allow_partial=False):
931
  """Check if a device is activated.
932

933
  If so, return informations about the real device.
934

935
  Args:
936
    disk: the objects.Disk instance
937
    allow_partial: don't abort the find if a child of the
938
                   device can't be found; this is intended to be
939
                   used when repairing mirrors
940

941
  Returns:
942
    None if the device can't be found
943
    otherwise the device instance
944

945
  """
946
  children = []
947
  if disk.children:
948
    for chdisk in disk.children:
949
      children.append(_RecursiveFindBD(chdisk))
950

    
951
  return bdev.FindDevice(disk.dev_type, disk.physical_id, children)
952

    
953

    
954
def FindBlockDevice(disk):
955
  """Check if a device is activated.
956

957
  If so, return informations about the real device.
958

959
  Args:
960
    disk: the objects.Disk instance
961
  Returns:
962
    None if the device can't be found
963
    (device_path, major, minor, sync_percent, estimated_time, is_degraded)
964

965
  """
966
  rbd = _RecursiveFindBD(disk)
967
  if rbd is None:
968
    return rbd
969
  return (rbd.dev_path, rbd.major, rbd.minor) + rbd.GetSyncStatus()
970

    
971

    
972
def UploadFile(file_name, data, mode, uid, gid, atime, mtime):
973
  """Write a file to the filesystem.
974

975
  This allows the master to overwrite(!) a file. It will only perform
976
  the operation if the file belongs to a list of configuration files.
977

978
  """
979
  if not os.path.isabs(file_name):
980
    logging.error("Filename passed to UploadFile is not absolute: '%s'",
981
                  file_name)
982
    return False
983

    
984
  allowed_files = [
985
    constants.CLUSTER_CONF_FILE,
986
    constants.ETC_HOSTS,
987
    constants.SSH_KNOWN_HOSTS_FILE,
988
    constants.VNC_PASSWORD_FILE,
989
    constants.JOB_QUEUE_SERIAL_FILE,
990
    ]
991
  allowed_files.extend(ssconf.SimpleStore().GetFileList())
992
  if file_name not in allowed_files:
993
    logging.error("Filename passed to UploadFile not in allowed"
994
                 " upload targets: '%s'", file_name)
995
    return False
996

    
997
  utils.WriteFile(file_name, data=data, mode=mode, uid=uid, gid=gid,
998
                  atime=atime, mtime=mtime)
999
  return True
1000

    
1001

    
1002
def _ErrnoOrStr(err):
1003
  """Format an EnvironmentError exception.
1004

1005
  If the `err` argument has an errno attribute, it will be looked up
1006
  and converted into a textual EXXXX description. Otherwise the string
1007
  representation of the error will be returned.
1008

1009
  """
1010
  if hasattr(err, 'errno'):
1011
    detail = errno.errorcode[err.errno]
1012
  else:
1013
    detail = str(err)
1014
  return detail
1015

    
1016

    
1017
def _OSOndiskVersion(name, os_dir):
1018
  """Compute and return the API version of a given OS.
1019

1020
  This function will try to read the API version of the os given by
1021
  the 'name' parameter and residing in the 'os_dir' directory.
1022

1023
  Return value will be either an integer denoting the version or None in the
1024
  case when this is not a valid OS name.
1025

1026
  """
1027
  api_file = os.path.sep.join([os_dir, "ganeti_api_version"])
1028

    
1029
  try:
1030
    st = os.stat(api_file)
1031
  except EnvironmentError, err:
1032
    raise errors.InvalidOS(name, os_dir, "'ganeti_api_version' file not"
1033
                           " found (%s)" % _ErrnoOrStr(err))
1034

    
1035
  if not stat.S_ISREG(stat.S_IFMT(st.st_mode)):
1036
    raise errors.InvalidOS(name, os_dir, "'ganeti_api_version' file is not"
1037
                           " a regular file")
1038

    
1039
  try:
1040
    f = open(api_file)
1041
    try:
1042
      api_version = f.read(256)
1043
    finally:
1044
      f.close()
1045
  except EnvironmentError, err:
1046
    raise errors.InvalidOS(name, os_dir, "error while reading the"
1047
                           " API version (%s)" % _ErrnoOrStr(err))
1048

    
1049
  api_version = api_version.strip()
1050
  try:
1051
    api_version = int(api_version)
1052
  except (TypeError, ValueError), err:
1053
    raise errors.InvalidOS(name, os_dir,
1054
                           "API version is not integer (%s)" % str(err))
1055

    
1056
  return api_version
1057

    
1058

    
1059
def DiagnoseOS(top_dirs=None):
1060
  """Compute the validity for all OSes.
1061

1062
  Returns an OS object for each name in all the given top directories
1063
  (if not given defaults to constants.OS_SEARCH_PATH)
1064

1065
  Returns:
1066
    list of OS objects
1067

1068
  """
1069
  if top_dirs is None:
1070
    top_dirs = constants.OS_SEARCH_PATH
1071

    
1072
  result = []
1073
  for dir_name in top_dirs:
1074
    if os.path.isdir(dir_name):
1075
      try:
1076
        f_names = utils.ListVisibleFiles(dir_name)
1077
      except EnvironmentError, err:
1078
        logging.exception("Can't list the OS directory %s", dir_name)
1079
        break
1080
      for name in f_names:
1081
        try:
1082
          os_inst = OSFromDisk(name, base_dir=dir_name)
1083
          result.append(os_inst)
1084
        except errors.InvalidOS, err:
1085
          result.append(objects.OS.FromInvalidOS(err))
1086

    
1087
  return result
1088

    
1089

    
1090
def OSFromDisk(name, base_dir=None):
1091
  """Create an OS instance from disk.
1092

1093
  This function will return an OS instance if the given name is a
1094
  valid OS name. Otherwise, it will raise an appropriate
1095
  `errors.InvalidOS` exception, detailing why this is not a valid
1096
  OS.
1097

1098
  Args:
1099
    os_dir: Directory containing the OS scripts. Defaults to a search
1100
            in all the OS_SEARCH_PATH directories.
1101

1102
  """
1103

    
1104
  if base_dir is None:
1105
    os_dir = utils.FindFile(name, constants.OS_SEARCH_PATH, os.path.isdir)
1106
    if os_dir is None:
1107
      raise errors.InvalidOS(name, None, "OS dir not found in search path")
1108
  else:
1109
    os_dir = os.path.sep.join([base_dir, name])
1110

    
1111
  api_version = _OSOndiskVersion(name, os_dir)
1112

    
1113
  if api_version != constants.OS_API_VERSION:
1114
    raise errors.InvalidOS(name, os_dir, "API version mismatch"
1115
                           " (found %s want %s)"
1116
                           % (api_version, constants.OS_API_VERSION))
1117

    
1118
  # OS Scripts dictionary, we will populate it with the actual script names
1119
  os_scripts = {'create': '', 'export': '', 'import': '', 'rename': ''}
1120

    
1121
  for script in os_scripts:
1122
    os_scripts[script] = os.path.sep.join([os_dir, script])
1123

    
1124
    try:
1125
      st = os.stat(os_scripts[script])
1126
    except EnvironmentError, err:
1127
      raise errors.InvalidOS(name, os_dir, "'%s' script missing (%s)" %
1128
                             (script, _ErrnoOrStr(err)))
1129

    
1130
    if stat.S_IMODE(st.st_mode) & stat.S_IXUSR != stat.S_IXUSR:
1131
      raise errors.InvalidOS(name, os_dir, "'%s' script not executable" %
1132
                             script)
1133

    
1134
    if not stat.S_ISREG(stat.S_IFMT(st.st_mode)):
1135
      raise errors.InvalidOS(name, os_dir, "'%s' is not a regular file" %
1136
                             script)
1137

    
1138

    
1139
  return objects.OS(name=name, path=os_dir, status=constants.OS_VALID_STATUS,
1140
                    create_script=os_scripts['create'],
1141
                    export_script=os_scripts['export'],
1142
                    import_script=os_scripts['import'],
1143
                    rename_script=os_scripts['rename'],
1144
                    api_version=api_version)
1145

    
1146

    
1147
def GrowBlockDevice(disk, amount):
1148
  """Grow a stack of block devices.
1149

1150
  This function is called recursively, with the childrens being the
1151
  first one resize.
1152

1153
  Args:
1154
    disk: the disk to be grown
1155

1156
  Returns: a tuple of (status, result), with:
1157
    status: the result (true/false) of the operation
1158
    result: the error message if the operation failed, otherwise not used
1159

1160
  """
1161
  r_dev = _RecursiveFindBD(disk)
1162
  if r_dev is None:
1163
    return False, "Cannot find block device %s" % (disk,)
1164

    
1165
  try:
1166
    r_dev.Grow(amount)
1167
  except errors.BlockDeviceError, err:
1168
    return False, str(err)
1169

    
1170
  return True, None
1171

    
1172

    
1173
def SnapshotBlockDevice(disk):
1174
  """Create a snapshot copy of a block device.
1175

1176
  This function is called recursively, and the snapshot is actually created
1177
  just for the leaf lvm backend device.
1178

1179
  Args:
1180
    disk: the disk to be snapshotted
1181

1182
  Returns:
1183
    a config entry for the actual lvm device snapshotted.
1184

1185
  """
1186
  if disk.children:
1187
    if len(disk.children) == 1:
1188
      # only one child, let's recurse on it
1189
      return SnapshotBlockDevice(disk.children[0])
1190
    else:
1191
      # more than one child, choose one that matches
1192
      for child in disk.children:
1193
        if child.size == disk.size:
1194
          # return implies breaking the loop
1195
          return SnapshotBlockDevice(child)
1196
  elif disk.dev_type == constants.LD_LV:
1197
    r_dev = _RecursiveFindBD(disk)
1198
    if r_dev is not None:
1199
      # let's stay on the safe side and ask for the full size, for now
1200
      return r_dev.Snapshot(disk.size)
1201
    else:
1202
      return None
1203
  else:
1204
    raise errors.ProgrammerError("Cannot snapshot non-lvm block device"
1205
                                 " '%s' of type '%s'" %
1206
                                 (disk.unique_id, disk.dev_type))
1207

    
1208

    
1209
def ExportSnapshot(disk, dest_node, instance):
1210
  """Export a block device snapshot to a remote node.
1211

1212
  Args:
1213
    disk: the snapshot block device
1214
    dest_node: the node to send the image to
1215
    instance: instance being exported
1216

1217
  Returns:
1218
    True if successful, False otherwise.
1219

1220
  """
1221
  inst_os = OSFromDisk(instance.os)
1222
  export_script = inst_os.export_script
1223

    
1224
  logfile = "%s/exp-%s-%s-%s.log" % (constants.LOG_OS_DIR, inst_os.name,
1225
                                     instance.name, int(time.time()))
1226
  if not os.path.exists(constants.LOG_OS_DIR):
1227
    os.mkdir(constants.LOG_OS_DIR, 0750)
1228

    
1229
  real_os_dev = _RecursiveFindBD(disk)
1230
  if real_os_dev is None:
1231
    raise errors.BlockDeviceError("Block device '%s' is not set up" %
1232
                                  str(disk))
1233
  real_os_dev.Open()
1234

    
1235
  destdir = os.path.join(constants.EXPORT_DIR, instance.name + ".new")
1236
  destfile = disk.physical_id[1]
1237

    
1238
  # the target command is built out of three individual commands,
1239
  # which are joined by pipes; we check each individual command for
1240
  # valid parameters
1241

    
1242
  expcmd = utils.BuildShellCmd("cd %s; %s -i %s -b %s 2>%s", inst_os.path,
1243
                               export_script, instance.name,
1244
                               real_os_dev.dev_path, logfile)
1245

    
1246
  comprcmd = "gzip"
1247

    
1248
  destcmd = utils.BuildShellCmd("mkdir -p %s && cat > %s/%s",
1249
                                destdir, destdir, destfile)
1250
  remotecmd = _GetSshRunner().BuildCmd(dest_node, constants.GANETI_RUNAS,
1251
                                       destcmd)
1252

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

    
1256
  result = utils.RunCmd(command)
1257

    
1258
  if result.failed:
1259
    logging.error("os snapshot export command '%s' returned error: %s"
1260
                  " output: %s", command, result.fail_reason, result.output)
1261
    return False
1262

    
1263
  return True
1264

    
1265

    
1266
def FinalizeExport(instance, snap_disks):
1267
  """Write out the export configuration information.
1268

1269
  Args:
1270
    instance: instance configuration
1271
    snap_disks: snapshot block devices
1272

1273
  Returns:
1274
    False in case of error, True otherwise.
1275

1276
  """
1277
  destdir = os.path.join(constants.EXPORT_DIR, instance.name + ".new")
1278
  finaldestdir = os.path.join(constants.EXPORT_DIR, instance.name)
1279

    
1280
  config = objects.SerializableConfigParser()
1281

    
1282
  config.add_section(constants.INISECT_EXP)
1283
  config.set(constants.INISECT_EXP, 'version', '0')
1284
  config.set(constants.INISECT_EXP, 'timestamp', '%d' % int(time.time()))
1285
  config.set(constants.INISECT_EXP, 'source', instance.primary_node)
1286
  config.set(constants.INISECT_EXP, 'os', instance.os)
1287
  config.set(constants.INISECT_EXP, 'compression', 'gzip')
1288

    
1289
  config.add_section(constants.INISECT_INS)
1290
  config.set(constants.INISECT_INS, 'name', instance.name)
1291
  config.set(constants.INISECT_INS, 'memory', '%d' % instance.memory)
1292
  config.set(constants.INISECT_INS, 'vcpus', '%d' % instance.vcpus)
1293
  config.set(constants.INISECT_INS, 'disk_template', instance.disk_template)
1294

    
1295
  nic_count = 0
1296
  for nic_count, nic in enumerate(instance.nics):
1297
    config.set(constants.INISECT_INS, 'nic%d_mac' %
1298
               nic_count, '%s' % nic.mac)
1299
    config.set(constants.INISECT_INS, 'nic%d_ip' % nic_count, '%s' % nic.ip)
1300
    config.set(constants.INISECT_INS, 'nic%d_bridge' % nic_count, '%s' % nic.bridge)
1301
  # TODO: redundant: on load can read nics until it doesn't exist
1302
  config.set(constants.INISECT_INS, 'nic_count' , '%d' % nic_count)
1303

    
1304
  disk_count = 0
1305
  for disk_count, disk in enumerate(snap_disks):
1306
    config.set(constants.INISECT_INS, 'disk%d_ivname' % disk_count,
1307
               ('%s' % disk.iv_name))
1308
    config.set(constants.INISECT_INS, 'disk%d_dump' % disk_count,
1309
               ('%s' % disk.physical_id[1]))
1310
    config.set(constants.INISECT_INS, 'disk%d_size' % disk_count,
1311
               ('%d' % disk.size))
1312
  config.set(constants.INISECT_INS, 'disk_count' , '%d' % disk_count)
1313

    
1314
  cff = os.path.join(destdir, constants.EXPORT_CONF_FILE)
1315
  cfo = open(cff, 'w')
1316
  try:
1317
    config.write(cfo)
1318
  finally:
1319
    cfo.close()
1320

    
1321
  shutil.rmtree(finaldestdir, True)
1322
  shutil.move(destdir, finaldestdir)
1323

    
1324
  return True
1325

    
1326

    
1327
def ExportInfo(dest):
1328
  """Get export configuration information.
1329

1330
  Args:
1331
    dest: directory containing the export
1332

1333
  Returns:
1334
    A serializable config file containing the export info.
1335

1336
  """
1337
  cff = os.path.join(dest, constants.EXPORT_CONF_FILE)
1338

    
1339
  config = objects.SerializableConfigParser()
1340
  config.read(cff)
1341

    
1342
  if (not config.has_section(constants.INISECT_EXP) or
1343
      not config.has_section(constants.INISECT_INS)):
1344
    return None
1345

    
1346
  return config
1347

    
1348

    
1349
def ImportOSIntoInstance(instance, os_disk, swap_disk, src_node, src_image):
1350
  """Import an os image into an instance.
1351

1352
  Args:
1353
    instance: the instance object
1354
    os_disk: the instance-visible name of the os device
1355
    swap_disk: the instance-visible name of the swap device
1356
    src_node: node holding the source image
1357
    src_image: path to the source image on src_node
1358

1359
  Returns:
1360
    False in case of error, True otherwise.
1361

1362
  """
1363
  inst_os = OSFromDisk(instance.os)
1364
  import_script = inst_os.import_script
1365

    
1366
  os_device = instance.FindDisk(os_disk)
1367
  if os_device is None:
1368
    logging.error("Can't find this device-visible name '%s'", os_disk)
1369
    return False
1370

    
1371
  swap_device = instance.FindDisk(swap_disk)
1372
  if swap_device is None:
1373
    logging.error("Can't find this device-visible name '%s'", swap_disk)
1374
    return False
1375

    
1376
  real_os_dev = _RecursiveFindBD(os_device)
1377
  if real_os_dev is None:
1378
    raise errors.BlockDeviceError("Block device '%s' is not set up" %
1379
                                  str(os_device))
1380
  real_os_dev.Open()
1381

    
1382
  real_swap_dev = _RecursiveFindBD(swap_device)
1383
  if real_swap_dev is None:
1384
    raise errors.BlockDeviceError("Block device '%s' is not set up" %
1385
                                  str(swap_device))
1386
  real_swap_dev.Open()
1387

    
1388
  logfile = "%s/import-%s-%s-%s.log" % (constants.LOG_OS_DIR, instance.os,
1389
                                        instance.name, int(time.time()))
1390
  if not os.path.exists(constants.LOG_OS_DIR):
1391
    os.mkdir(constants.LOG_OS_DIR, 0750)
1392

    
1393
  destcmd = utils.BuildShellCmd('cat %s', src_image)
1394
  remotecmd = _GetSshRunner().BuildCmd(src_node, constants.GANETI_RUNAS,
1395
                                       destcmd)
1396

    
1397
  comprcmd = "gunzip"
1398
  impcmd = utils.BuildShellCmd("(cd %s; %s -i %s -b %s -s %s &>%s)",
1399
                               inst_os.path, import_script, instance.name,
1400
                               real_os_dev.dev_path, real_swap_dev.dev_path,
1401
                               logfile)
1402

    
1403
  command = '|'.join([utils.ShellQuoteArgs(remotecmd), comprcmd, impcmd])
1404

    
1405
  result = utils.RunCmd(command)
1406

    
1407
  if result.failed:
1408
    logging.error("os import command '%s' returned error: %s"
1409
                  " output: %s", command, result.fail_reason, result.output)
1410
    return False
1411

    
1412
  return True
1413

    
1414

    
1415
def ListExports():
1416
  """Return a list of exports currently available on this machine.
1417

1418
  """
1419
  if os.path.isdir(constants.EXPORT_DIR):
1420
    return utils.ListVisibleFiles(constants.EXPORT_DIR)
1421
  else:
1422
    return []
1423

    
1424

    
1425
def RemoveExport(export):
1426
  """Remove an existing export from the node.
1427

1428
  Args:
1429
    export: the name of the export to remove
1430

1431
  Returns:
1432
    False in case of error, True otherwise.
1433

1434
  """
1435
  target = os.path.join(constants.EXPORT_DIR, export)
1436

    
1437
  shutil.rmtree(target)
1438
  # TODO: catch some of the relevant exceptions and provide a pretty
1439
  # error message if rmtree fails.
1440

    
1441
  return True
1442

    
1443

    
1444
def RenameBlockDevices(devlist):
1445
  """Rename a list of block devices.
1446

1447
  The devlist argument is a list of tuples (disk, new_logical,
1448
  new_physical). The return value will be a combined boolean result
1449
  (True only if all renames succeeded).
1450

1451
  """
1452
  result = True
1453
  for disk, unique_id in devlist:
1454
    dev = _RecursiveFindBD(disk)
1455
    if dev is None:
1456
      result = False
1457
      continue
1458
    try:
1459
      old_rpath = dev.dev_path
1460
      dev.Rename(unique_id)
1461
      new_rpath = dev.dev_path
1462
      if old_rpath != new_rpath:
1463
        DevCacheManager.RemoveCache(old_rpath)
1464
        # FIXME: we should add the new cache information here, like:
1465
        # DevCacheManager.UpdateCache(new_rpath, owner, ...)
1466
        # but we don't have the owner here - maybe parse from existing
1467
        # cache? for now, we only lose lvm data when we rename, which
1468
        # is less critical than DRBD or MD
1469
    except errors.BlockDeviceError, err:
1470
      logging.exception("Can't rename device '%s' to '%s'", dev, unique_id)
1471
      result = False
1472
  return result
1473

    
1474

    
1475
def _TransformFileStorageDir(file_storage_dir):
1476
  """Checks whether given file_storage_dir is valid.
1477

1478
  Checks wheter the given file_storage_dir is within the cluster-wide
1479
  default file_storage_dir stored in SimpleStore. Only paths under that
1480
  directory are allowed.
1481

1482
  Args:
1483
    file_storage_dir: string with path
1484

1485
  Returns:
1486
    normalized file_storage_dir (string) if valid, None otherwise
1487

1488
  """
1489
  file_storage_dir = os.path.normpath(file_storage_dir)
1490
  base_file_storage_dir = ssconf.SimpleStore().GetFileStorageDir()
1491
  if (not os.path.commonprefix([file_storage_dir, base_file_storage_dir]) ==
1492
      base_file_storage_dir):
1493
    logging.error("file storage directory '%s' is not under base file"
1494
                  " storage directory '%s'",
1495
                  file_storage_dir, base_file_storage_dir)
1496
    return None
1497
  return file_storage_dir
1498

    
1499

    
1500
def CreateFileStorageDir(file_storage_dir):
1501
  """Create file storage directory.
1502

1503
  Args:
1504
    file_storage_dir: string containing the path
1505

1506
  Returns:
1507
    tuple with first element a boolean indicating wheter dir
1508
    creation was successful or not
1509

1510
  """
1511
  file_storage_dir = _TransformFileStorageDir(file_storage_dir)
1512
  result = True,
1513
  if not file_storage_dir:
1514
    result = False,
1515
  else:
1516
    if os.path.exists(file_storage_dir):
1517
      if not os.path.isdir(file_storage_dir):
1518
        logging.error("'%s' is not a directory", file_storage_dir)
1519
        result = False,
1520
    else:
1521
      try:
1522
        os.makedirs(file_storage_dir, 0750)
1523
      except OSError, err:
1524
        logging.error("Cannot create file storage directory '%s': %s",
1525
                      file_storage_dir, err)
1526
        result = False,
1527
  return result
1528

    
1529

    
1530
def RemoveFileStorageDir(file_storage_dir):
1531
  """Remove file storage directory.
1532

1533
  Remove it only if it's empty. If not log an error and return.
1534

1535
  Args:
1536
    file_storage_dir: string containing the path
1537

1538
  Returns:
1539
    tuple with first element a boolean indicating wheter dir
1540
    removal was successful or not
1541

1542
  """
1543
  file_storage_dir = _TransformFileStorageDir(file_storage_dir)
1544
  result = True,
1545
  if not file_storage_dir:
1546
    result = False,
1547
  else:
1548
    if os.path.exists(file_storage_dir):
1549
      if not os.path.isdir(file_storage_dir):
1550
        logging.error("'%s' is not a directory", file_storage_dir)
1551
        result = False,
1552
      # deletes dir only if empty, otherwise we want to return False
1553
      try:
1554
        os.rmdir(file_storage_dir)
1555
      except OSError, err:
1556
        logging.exception("Cannot remove file storage directory '%s'",
1557
                          file_storage_dir)
1558
        result = False,
1559
  return result
1560

    
1561

    
1562
def RenameFileStorageDir(old_file_storage_dir, new_file_storage_dir):
1563
  """Rename the file storage directory.
1564

1565
  Args:
1566
    old_file_storage_dir: string containing the old path
1567
    new_file_storage_dir: string containing the new path
1568

1569
  Returns:
1570
    tuple with first element a boolean indicating wheter dir
1571
    rename was successful or not
1572

1573
  """
1574
  old_file_storage_dir = _TransformFileStorageDir(old_file_storage_dir)
1575
  new_file_storage_dir = _TransformFileStorageDir(new_file_storage_dir)
1576
  result = True,
1577
  if not old_file_storage_dir or not new_file_storage_dir:
1578
    result = False,
1579
  else:
1580
    if not os.path.exists(new_file_storage_dir):
1581
      if os.path.isdir(old_file_storage_dir):
1582
        try:
1583
          os.rename(old_file_storage_dir, new_file_storage_dir)
1584
        except OSError, err:
1585
          logging.exception("Cannot rename '%s' to '%s'",
1586
                            old_file_storage_dir, new_file_storage_dir)
1587
          result =  False,
1588
      else:
1589
        logging.error("'%s' is not a directory", old_file_storage_dir)
1590
        result = False,
1591
    else:
1592
      if os.path.exists(old_file_storage_dir):
1593
        logging.error("Cannot rename '%s' to '%s'. Both locations exist.",
1594
                      old_file_storage_dir, new_file_storage_dir)
1595
        result = False,
1596
  return result
1597

    
1598

    
1599
def CloseBlockDevices(disks):
1600
  """Closes the given block devices.
1601

1602
  This means they will be switched to secondary mode (in case of DRBD).
1603

1604
  """
1605
  bdevs = []
1606
  for cf in disks:
1607
    rd = _RecursiveFindBD(cf)
1608
    if rd is None:
1609
      return (False, "Can't find device %s" % cf)
1610
    bdevs.append(rd)
1611

    
1612
  msg = []
1613
  for rd in bdevs:
1614
    try:
1615
      rd.Close()
1616
    except errors.BlockDeviceError, err:
1617
      msg.append(str(err))
1618
  if msg:
1619
    return (False, "Can't make devices secondary: %s" % ",".join(msg))
1620
  else:
1621
    return (True, "All devices secondary")
1622

    
1623

    
1624
class HooksRunner(object):
1625
  """Hook runner.
1626

1627
  This class is instantiated on the node side (ganeti-noded) and not on
1628
  the master side.
1629

1630
  """
1631
  RE_MASK = re.compile("^[a-zA-Z0-9_-]+$")
1632

    
1633
  def __init__(self, hooks_base_dir=None):
1634
    """Constructor for hooks runner.
1635

1636
    Args:
1637
      - hooks_base_dir: if not None, this overrides the
1638
        constants.HOOKS_BASE_DIR (useful for unittests)
1639

1640
    """
1641
    if hooks_base_dir is None:
1642
      hooks_base_dir = constants.HOOKS_BASE_DIR
1643
    self._BASE_DIR = hooks_base_dir
1644

    
1645
  @staticmethod
1646
  def ExecHook(script, env):
1647
    """Exec one hook script.
1648

1649
    Args:
1650
     - script: the full path to the script
1651
     - env: the environment with which to exec the script
1652

1653
    """
1654
    # exec the process using subprocess and log the output
1655
    fdstdin = None
1656
    try:
1657
      fdstdin = open("/dev/null", "r")
1658
      child = subprocess.Popen([script], stdin=fdstdin, stdout=subprocess.PIPE,
1659
                               stderr=subprocess.STDOUT, close_fds=True,
1660
                               shell=False, cwd="/", env=env)
1661
      output = ""
1662
      try:
1663
        output = child.stdout.read(4096)
1664
        child.stdout.close()
1665
      except EnvironmentError, err:
1666
        output += "Hook script error: %s" % str(err)
1667

    
1668
      while True:
1669
        try:
1670
          result = child.wait()
1671
          break
1672
        except EnvironmentError, err:
1673
          if err.errno == errno.EINTR:
1674
            continue
1675
          raise
1676
    finally:
1677
      # try not to leak fds
1678
      for fd in (fdstdin, ):
1679
        if fd is not None:
1680
          try:
1681
            fd.close()
1682
          except EnvironmentError, err:
1683
            # just log the error
1684
            #logging.exception("Error while closing fd %s", fd)
1685
            pass
1686

    
1687
    return result == 0, output
1688

    
1689
  def RunHooks(self, hpath, phase, env):
1690
    """Run the scripts in the hooks directory.
1691

1692
    This method will not be usually overriden by child opcodes.
1693

1694
    """
1695
    if phase == constants.HOOKS_PHASE_PRE:
1696
      suffix = "pre"
1697
    elif phase == constants.HOOKS_PHASE_POST:
1698
      suffix = "post"
1699
    else:
1700
      raise errors.ProgrammerError("Unknown hooks phase: '%s'" % phase)
1701
    rr = []
1702

    
1703
    subdir = "%s-%s.d" % (hpath, suffix)
1704
    dir_name = "%s/%s" % (self._BASE_DIR, subdir)
1705
    try:
1706
      dir_contents = utils.ListVisibleFiles(dir_name)
1707
    except OSError, err:
1708
      # must log
1709
      return rr
1710

    
1711
    # we use the standard python sort order,
1712
    # so 00name is the recommended naming scheme
1713
    dir_contents.sort()
1714
    for relname in dir_contents:
1715
      fname = os.path.join(dir_name, relname)
1716
      if not (os.path.isfile(fname) and os.access(fname, os.X_OK) and
1717
          self.RE_MASK.match(relname) is not None):
1718
        rrval = constants.HKR_SKIP
1719
        output = ""
1720
      else:
1721
        result, output = self.ExecHook(fname, env)
1722
        if not result:
1723
          rrval = constants.HKR_FAIL
1724
        else:
1725
          rrval = constants.HKR_SUCCESS
1726
      rr.append(("%s/%s" % (subdir, relname), rrval, output))
1727

    
1728
    return rr
1729

    
1730

    
1731
class IAllocatorRunner(object):
1732
  """IAllocator runner.
1733

1734
  This class is instantiated on the node side (ganeti-noded) and not on
1735
  the master side.
1736

1737
  """
1738
  def Run(self, name, idata):
1739
    """Run an iallocator script.
1740

1741
    Return value: tuple of:
1742
       - run status (one of the IARUN_ constants)
1743
       - stdout
1744
       - stderr
1745
       - fail reason (as from utils.RunResult)
1746

1747
    """
1748
    alloc_script = utils.FindFile(name, constants.IALLOCATOR_SEARCH_PATH,
1749
                                  os.path.isfile)
1750
    if alloc_script is None:
1751
      return (constants.IARUN_NOTFOUND, None, None, None)
1752

    
1753
    fd, fin_name = tempfile.mkstemp(prefix="ganeti-iallocator.")
1754
    try:
1755
      os.write(fd, idata)
1756
      os.close(fd)
1757
      result = utils.RunCmd([alloc_script, fin_name])
1758
      if result.failed:
1759
        return (constants.IARUN_FAILURE, result.stdout, result.stderr,
1760
                result.fail_reason)
1761
    finally:
1762
      os.unlink(fin_name)
1763

    
1764
    return (constants.IARUN_SUCCESS, result.stdout, result.stderr, None)
1765

    
1766

    
1767
class DevCacheManager(object):
1768
  """Simple class for managing a cache of block device information.
1769

1770
  """
1771
  _DEV_PREFIX = "/dev/"
1772
  _ROOT_DIR = constants.BDEV_CACHE_DIR
1773

    
1774
  @classmethod
1775
  def _ConvertPath(cls, dev_path):
1776
    """Converts a /dev/name path to the cache file name.
1777

1778
    This replaces slashes with underscores and strips the /dev
1779
    prefix. It then returns the full path to the cache file
1780

1781
    """
1782
    if dev_path.startswith(cls._DEV_PREFIX):
1783
      dev_path = dev_path[len(cls._DEV_PREFIX):]
1784
    dev_path = dev_path.replace("/", "_")
1785
    fpath = "%s/bdev_%s" % (cls._ROOT_DIR, dev_path)
1786
    return fpath
1787

    
1788
  @classmethod
1789
  def UpdateCache(cls, dev_path, owner, on_primary, iv_name):
1790
    """Updates the cache information for a given device.
1791

1792
    """
1793
    if dev_path is None:
1794
      logging.error("DevCacheManager.UpdateCache got a None dev_path")
1795
      return
1796
    fpath = cls._ConvertPath(dev_path)
1797
    if on_primary:
1798
      state = "primary"
1799
    else:
1800
      state = "secondary"
1801
    if iv_name is None:
1802
      iv_name = "not_visible"
1803
    fdata = "%s %s %s\n" % (str(owner), state, iv_name)
1804
    try:
1805
      utils.WriteFile(fpath, data=fdata)
1806
    except EnvironmentError, err:
1807
      logging.exception("Can't update bdev cache for %s", dev_path)
1808

    
1809
  @classmethod
1810
  def RemoveCache(cls, dev_path):
1811
    """Remove data for a dev_path.
1812

1813
    """
1814
    if dev_path is None:
1815
      logging.error("DevCacheManager.RemoveCache got a None dev_path")
1816
      return
1817
    fpath = cls._ConvertPath(dev_path)
1818
    try:
1819
      utils.RemoveFile(fpath)
1820
    except EnvironmentError, err:
1821
      logging.exception("Can't update bdev cache for %s", dev_path)