Convert call_blockdev_removechildren to new result
[ganeti-local] / lib / backend.py
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     msg = "Error while processing user ssh files"
264     logging.exception(msg)
265     return (False, "%s: %s" % (msg, err))
266
267   for name, content in [(priv_key, sshkey), (pub_key, sshpub)]:
268     utils.WriteFile(name, data=content, mode=0600)
269
270   utils.AddAuthorizedKey(auth_keys, sshpub)
271
272   utils.RunCmd([constants.SSH_INITD_SCRIPT, "restart"])
273
274   return (True, "Node added successfully")
275
276
277 def LeaveCluster():
278   """Cleans up and remove the current node.
279
280   This function cleans up and prepares the current node to be removed
281   from the cluster.
282
283   If processing is successful, then it raises an
284   L{errors.QuitGanetiException} which is used as a special case to
285   shutdown the node daemon.
286
287   """
288   _CleanDirectory(constants.DATA_DIR)
289   JobQueuePurge()
290
291   try:
292     priv_key, pub_key, auth_keys = ssh.GetUserFiles(constants.GANETI_RUNAS)
293   except errors.OpExecError:
294     logging.exception("Error while processing ssh files")
295     return
296
297   f = open(pub_key, 'r')
298   try:
299     utils.RemoveAuthorizedKey(auth_keys, f.read(8192))
300   finally:
301     f.close()
302
303   utils.RemoveFile(priv_key)
304   utils.RemoveFile(pub_key)
305
306   # Return a reassuring string to the caller, and quit
307   raise errors.QuitGanetiException(False, 'Shutdown scheduled')
308
309
310 def GetNodeInfo(vgname, hypervisor_type):
311   """Gives back a hash with different informations about the node.
312
313   @type vgname: C{string}
314   @param vgname: the name of the volume group to ask for disk space information
315   @type hypervisor_type: C{str}
316   @param hypervisor_type: the name of the hypervisor to ask for
317       memory information
318   @rtype: C{dict}
319   @return: dictionary with the following keys:
320       - vg_size is the size of the configured volume group in MiB
321       - vg_free is the free size of the volume group in MiB
322       - memory_dom0 is the memory allocated for domain0 in MiB
323       - memory_free is the currently available (free) ram in MiB
324       - memory_total is the total number of ram in MiB
325
326   """
327   outputarray = {}
328   vginfo = _GetVGInfo(vgname)
329   outputarray['vg_size'] = vginfo['vg_size']
330   outputarray['vg_free'] = vginfo['vg_free']
331
332   hyper = hypervisor.GetHypervisor(hypervisor_type)
333   hyp_info = hyper.GetNodeInfo()
334   if hyp_info is not None:
335     outputarray.update(hyp_info)
336
337   f = open("/proc/sys/kernel/random/boot_id", 'r')
338   try:
339     outputarray["bootid"] = f.read(128).rstrip("\n")
340   finally:
341     f.close()
342
343   return outputarray
344
345
346 def VerifyNode(what, cluster_name):
347   """Verify the status of the local node.
348
349   Based on the input L{what} parameter, various checks are done on the
350   local node.
351
352   If the I{filelist} key is present, this list of
353   files is checksummed and the file/checksum pairs are returned.
354
355   If the I{nodelist} key is present, we check that we have
356   connectivity via ssh with the target nodes (and check the hostname
357   report).
358
359   If the I{node-net-test} key is present, we check that we have
360   connectivity to the given nodes via both primary IP and, if
361   applicable, secondary IPs.
362
363   @type what: C{dict}
364   @param what: a dictionary of things to check:
365       - filelist: list of files for which to compute checksums
366       - nodelist: list of nodes we should check ssh communication with
367       - node-net-test: list of nodes we should check node daemon port
368         connectivity with
369       - hypervisor: list with hypervisors to run the verify for
370   @rtype: dict
371   @return: a dictionary with the same keys as the input dict, and
372       values representing the result of the checks
373
374   """
375   result = {}
376
377   if constants.NV_HYPERVISOR in what:
378     result[constants.NV_HYPERVISOR] = tmp = {}
379     for hv_name in what[constants.NV_HYPERVISOR]:
380       tmp[hv_name] = hypervisor.GetHypervisor(hv_name).Verify()
381
382   if constants.NV_FILELIST in what:
383     result[constants.NV_FILELIST] = utils.FingerprintFiles(
384       what[constants.NV_FILELIST])
385
386   if constants.NV_NODELIST in what:
387     result[constants.NV_NODELIST] = tmp = {}
388     random.shuffle(what[constants.NV_NODELIST])
389     for node in what[constants.NV_NODELIST]:
390       success, message = _GetSshRunner(cluster_name).VerifyNodeHostname(node)
391       if not success:
392         tmp[node] = message
393
394   if constants.NV_NODENETTEST in what:
395     result[constants.NV_NODENETTEST] = tmp = {}
396     my_name = utils.HostInfo().name
397     my_pip = my_sip = None
398     for name, pip, sip in what[constants.NV_NODENETTEST]:
399       if name == my_name:
400         my_pip = pip
401         my_sip = sip
402         break
403     if not my_pip:
404       tmp[my_name] = ("Can't find my own primary/secondary IP"
405                       " in the node list")
406     else:
407       port = utils.GetNodeDaemonPort()
408       for name, pip, sip in what[constants.NV_NODENETTEST]:
409         fail = []
410         if not utils.TcpPing(pip, port, source=my_pip):
411           fail.append("primary")
412         if sip != pip:
413           if not utils.TcpPing(sip, port, source=my_sip):
414             fail.append("secondary")
415         if fail:
416           tmp[name] = ("failure using the %s interface(s)" %
417                        " and ".join(fail))
418
419   if constants.NV_LVLIST in what:
420     result[constants.NV_LVLIST] = GetVolumeList(what[constants.NV_LVLIST])
421
422   if constants.NV_INSTANCELIST in what:
423     result[constants.NV_INSTANCELIST] = GetInstanceList(
424       what[constants.NV_INSTANCELIST])
425
426   if constants.NV_VGLIST in what:
427     result[constants.NV_VGLIST] = ListVolumeGroups()
428
429   if constants.NV_VERSION in what:
430     result[constants.NV_VERSION] = (constants.PROTOCOL_VERSION,
431                                     constants.RELEASE_VERSION)
432
433   if constants.NV_HVINFO in what:
434     hyper = hypervisor.GetHypervisor(what[constants.NV_HVINFO])
435     result[constants.NV_HVINFO] = hyper.GetNodeInfo()
436
437   if constants.NV_DRBDLIST in what:
438     try:
439       used_minors = bdev.DRBD8.GetUsedDevs().keys()
440     except errors.BlockDeviceError, err:
441       logging.warning("Can't get used minors list", exc_info=True)
442       used_minors = str(err)
443     result[constants.NV_DRBDLIST] = used_minors
444
445   return result
446
447
448 def GetVolumeList(vg_name):
449   """Compute list of logical volumes and their size.
450
451   @type vg_name: str
452   @param vg_name: the volume group whose LVs we should list
453   @rtype: dict
454   @return:
455       dictionary of all partions (key) with value being a tuple of
456       their size (in MiB), inactive and online status::
457
458         {'test1': ('20.06', True, True)}
459
460       in case of errors, a string is returned with the error
461       details.
462
463   """
464   lvs = {}
465   sep = '|'
466   result = utils.RunCmd(["lvs", "--noheadings", "--units=m", "--nosuffix",
467                          "--separator=%s" % sep,
468                          "-olv_name,lv_size,lv_attr", vg_name])
469   if result.failed:
470     logging.error("Failed to list logical volumes, lvs output: %s",
471                   result.output)
472     return result.output
473
474   valid_line_re = re.compile("^ *([^|]+)\|([0-9.]+)\|([^|]{6})\|?$")
475   for line in result.stdout.splitlines():
476     line = line.strip()
477     match = valid_line_re.match(line)
478     if not match:
479       logging.error("Invalid line returned from lvs output: '%s'", line)
480       continue
481     name, size, attr = match.groups()
482     inactive = attr[4] == '-'
483     online = attr[5] == 'o'
484     lvs[name] = (size, inactive, online)
485
486   return lvs
487
488
489 def ListVolumeGroups():
490   """List the volume groups and their size.
491
492   @rtype: dict
493   @return: dictionary with keys volume name and values the
494       size of the volume
495
496   """
497   return utils.ListVolumeGroups()
498
499
500 def NodeVolumes():
501   """List all volumes on this node.
502
503   @rtype: list
504   @return:
505     A list of dictionaries, each having four keys:
506       - name: the logical volume name,
507       - size: the size of the logical volume
508       - dev: the physical device on which the LV lives
509       - vg: the volume group to which it belongs
510
511     In case of errors, we return an empty list and log the
512     error.
513
514     Note that since a logical volume can live on multiple physical
515     volumes, the resulting list might include a logical volume
516     multiple times.
517
518   """
519   result = utils.RunCmd(["lvs", "--noheadings", "--units=m", "--nosuffix",
520                          "--separator=|",
521                          "--options=lv_name,lv_size,devices,vg_name"])
522   if result.failed:
523     logging.error("Failed to list logical volumes, lvs output: %s",
524                   result.output)
525     return []
526
527   def parse_dev(dev):
528     if '(' in dev:
529       return dev.split('(')[0]
530     else:
531       return dev
532
533   def map_line(line):
534     return {
535       'name': line[0].strip(),
536       'size': line[1].strip(),
537       'dev': parse_dev(line[2].strip()),
538       'vg': line[3].strip(),
539     }
540
541   return [map_line(line.split('|')) for line in result.stdout.splitlines()
542           if line.count('|') >= 3]
543
544
545 def BridgesExist(bridges_list):
546   """Check if a list of bridges exist on the current node.
547
548   @rtype: boolean
549   @return: C{True} if all of them exist, C{False} otherwise
550
551   """
552   for bridge in bridges_list:
553     if not utils.BridgeExists(bridge):
554       return False
555
556   return True
557
558
559 def GetInstanceList(hypervisor_list):
560   """Provides a list of instances.
561
562   @type hypervisor_list: list
563   @param hypervisor_list: the list of hypervisors to query information
564
565   @rtype: list
566   @return: a list of all running instances on the current node
567     - instance1.example.com
568     - instance2.example.com
569
570   """
571   results = []
572   for hname in hypervisor_list:
573     try:
574       names = hypervisor.GetHypervisor(hname).ListInstances()
575       results.extend(names)
576     except errors.HypervisorError, err:
577       logging.exception("Error enumerating instances for hypevisor %s", hname)
578       raise
579
580   return results
581
582
583 def GetInstanceInfo(instance, hname):
584   """Gives back the informations about an instance as a dictionary.
585
586   @type instance: string
587   @param instance: the instance name
588   @type hname: string
589   @param hname: the hypervisor type of the instance
590
591   @rtype: dict
592   @return: dictionary with the following keys:
593       - memory: memory size of instance (int)
594       - state: xen state of instance (string)
595       - time: cpu time of instance (float)
596
597   """
598   output = {}
599
600   iinfo = hypervisor.GetHypervisor(hname).GetInstanceInfo(instance)
601   if iinfo is not None:
602     output['memory'] = iinfo[2]
603     output['state'] = iinfo[4]
604     output['time'] = iinfo[5]
605
606   return output
607
608
609 def GetInstanceMigratable(instance):
610   """Gives whether an instance can be migrated.
611
612   @type instance: L{objects.Instance}
613   @param instance: object representing the instance to be checked.
614
615   @rtype: tuple
616   @return: tuple of (result, description) where:
617       - result: whether the instance can be migrated or not
618       - description: a description of the issue, if relevant
619
620   """
621   hyper = hypervisor.GetHypervisor(instance.hypervisor)
622   if instance.name not in hyper.ListInstances():
623     return (False, 'not running')
624
625   for idx in range(len(instance.disks)):
626     link_name = _GetBlockDevSymlinkPath(instance.name, idx)
627     if not os.path.islink(link_name):
628       return (False, 'not restarted since ganeti 1.2.5')
629
630   return (True, '')
631
632
633 def GetAllInstancesInfo(hypervisor_list):
634   """Gather data about all instances.
635
636   This is the equivalent of L{GetInstanceInfo}, except that it
637   computes data for all instances at once, thus being faster if one
638   needs data about more than one instance.
639
640   @type hypervisor_list: list
641   @param hypervisor_list: list of hypervisors to query for instance data
642
643   @rtype: dict
644   @return: dictionary of instance: data, with data having the following keys:
645       - memory: memory size of instance (int)
646       - state: xen state of instance (string)
647       - time: cpu time of instance (float)
648       - vcpus: the number of vcpus
649
650   """
651   output = {}
652
653   for hname in hypervisor_list:
654     iinfo = hypervisor.GetHypervisor(hname).GetAllInstancesInfo()
655     if iinfo:
656       for name, inst_id, memory, vcpus, state, times in iinfo:
657         value = {
658           'memory': memory,
659           'vcpus': vcpus,
660           'state': state,
661           'time': times,
662           }
663         if name in output:
664           # we only check static parameters, like memory and vcpus,
665           # and not state and time which can change between the
666           # invocations of the different hypervisors
667           for key in 'memory', 'vcpus':
668             if value[key] != output[name][key]:
669               raise errors.HypervisorError("Instance %s is running twice"
670                                            " with different parameters" % name)
671         output[name] = value
672
673   return output
674
675
676 def InstanceOsAdd(instance, reinstall):
677   """Add an OS to an instance.
678
679   @type instance: L{objects.Instance}
680   @param instance: Instance whose OS is to be installed
681   @type reinstall: boolean
682   @param reinstall: whether this is an instance reinstall
683   @rtype: boolean
684   @return: the success of the operation
685
686   """
687   try:
688     inst_os = OSFromDisk(instance.os)
689   except errors.InvalidOS, err:
690     os_name, os_dir, os_err = err.args
691     if os_dir is None:
692       return (False, "Can't find OS '%s': %s" % (os_name, os_err))
693     else:
694       return (False, "Error parsing OS '%s' in directory %s: %s" %
695               (os_name, os_dir, os_err))
696
697   create_env = OSEnvironment(instance)
698   if reinstall:
699     create_env['INSTANCE_REINSTALL'] = "1"
700
701   logfile = "%s/add-%s-%s-%d.log" % (constants.LOG_OS_DIR, instance.os,
702                                      instance.name, int(time.time()))
703
704   result = utils.RunCmd([inst_os.create_script], env=create_env,
705                         cwd=inst_os.path, output=logfile,)
706   if result.failed:
707     logging.error("os create command '%s' returned error: %s, logfile: %s,"
708                   " output: %s", result.cmd, result.fail_reason, logfile,
709                   result.output)
710     lines = [utils.SafeEncode(val)
711              for val in utils.TailFile(logfile, lines=20)]
712     return (False, "OS create script failed (%s), last lines in the"
713             " log file:\n%s" % (result.fail_reason, "\n".join(lines)))
714
715   return (True, "Successfully installed")
716
717
718 def RunRenameInstance(instance, old_name):
719   """Run the OS rename script for an instance.
720
721   @type instance: L{objects.Instance}
722   @param instance: Instance whose OS is to be installed
723   @type old_name: string
724   @param old_name: previous instance name
725   @rtype: boolean
726   @return: the success of the operation
727
728   """
729   inst_os = OSFromDisk(instance.os)
730
731   rename_env = OSEnvironment(instance)
732   rename_env['OLD_INSTANCE_NAME'] = old_name
733
734   logfile = "%s/rename-%s-%s-%s-%d.log" % (constants.LOG_OS_DIR, instance.os,
735                                            old_name,
736                                            instance.name, int(time.time()))
737
738   result = utils.RunCmd([inst_os.rename_script], env=rename_env,
739                         cwd=inst_os.path, output=logfile)
740
741   if result.failed:
742     logging.error("os create command '%s' returned error: %s output: %s",
743                   result.cmd, result.fail_reason, result.output)
744     lines = [utils.SafeEncode(val)
745              for val in utils.TailFile(logfile, lines=20)]
746     return (False, "OS rename script failed (%s), last lines in the"
747             " log file:\n%s" % (result.fail_reason, "\n".join(lines)))
748
749   return (True, "Rename successful")
750
751
752 def _GetVGInfo(vg_name):
753   """Get informations about the volume group.
754
755   @type vg_name: str
756   @param vg_name: the volume group which we query
757   @rtype: dict
758   @return:
759     A dictionary with the following keys:
760       - C{vg_size} is the total size of the volume group in MiB
761       - C{vg_free} is the free size of the volume group in MiB
762       - C{pv_count} are the number of physical disks in that VG
763
764     If an error occurs during gathering of data, we return the same dict
765     with keys all set to None.
766
767   """
768   retdic = dict.fromkeys(["vg_size", "vg_free", "pv_count"])
769
770   retval = utils.RunCmd(["vgs", "-ovg_size,vg_free,pv_count", "--noheadings",
771                          "--nosuffix", "--units=m", "--separator=:", vg_name])
772
773   if retval.failed:
774     logging.error("volume group %s not present", vg_name)
775     return retdic
776   valarr = retval.stdout.strip().rstrip(':').split(':')
777   if len(valarr) == 3:
778     try:
779       retdic = {
780         "vg_size": int(round(float(valarr[0]), 0)),
781         "vg_free": int(round(float(valarr[1]), 0)),
782         "pv_count": int(valarr[2]),
783         }
784     except ValueError, err:
785       logging.exception("Fail to parse vgs output")
786   else:
787     logging.error("vgs output has the wrong number of fields (expected"
788                   " three): %s", str(valarr))
789   return retdic
790
791
792 def _GetBlockDevSymlinkPath(instance_name, idx):
793   return os.path.join(constants.DISK_LINKS_DIR,
794                       "%s:%d" % (instance_name, idx))
795
796
797 def _SymlinkBlockDev(instance_name, device_path, idx):
798   """Set up symlinks to a instance's block device.
799
800   This is an auxiliary function run when an instance is start (on the primary
801   node) or when an instance is migrated (on the target node).
802
803
804   @param instance_name: the name of the target instance
805   @param device_path: path of the physical block device, on the node
806   @param idx: the disk index
807   @return: absolute path to the disk's symlink
808
809   """
810   link_name = _GetBlockDevSymlinkPath(instance_name, idx)
811   try:
812     os.symlink(device_path, link_name)
813   except OSError, err:
814     if err.errno == errno.EEXIST:
815       if (not os.path.islink(link_name) or
816           os.readlink(link_name) != device_path):
817         os.remove(link_name)
818         os.symlink(device_path, link_name)
819     else:
820       raise
821
822   return link_name
823
824
825 def _RemoveBlockDevLinks(instance_name, disks):
826   """Remove the block device symlinks belonging to the given instance.
827
828   """
829   for idx, disk in enumerate(disks):
830     link_name = _GetBlockDevSymlinkPath(instance_name, idx)
831     if os.path.islink(link_name):
832       try:
833         os.remove(link_name)
834       except OSError:
835         logging.exception("Can't remove symlink '%s'", link_name)
836
837
838 def _GatherAndLinkBlockDevs(instance):
839   """Set up an instance's block device(s).
840
841   This is run on the primary node at instance startup. The block
842   devices must be already assembled.
843
844   @type instance: L{objects.Instance}
845   @param instance: the instance whose disks we shoul assemble
846   @rtype: list
847   @return: list of (disk_object, device_path)
848
849   """
850   block_devices = []
851   for idx, disk in enumerate(instance.disks):
852     device = _RecursiveFindBD(disk)
853     if device is None:
854       raise errors.BlockDeviceError("Block device '%s' is not set up." %
855                                     str(disk))
856     device.Open()
857     try:
858       link_name = _SymlinkBlockDev(instance.name, device.dev_path, idx)
859     except OSError, e:
860       raise errors.BlockDeviceError("Cannot create block device symlink: %s" %
861                                     e.strerror)
862
863     block_devices.append((disk, link_name))
864
865   return block_devices
866
867
868 def StartInstance(instance):
869   """Start an instance.
870
871   @type instance: L{objects.Instance}
872   @param instance: the instance object
873   @rtype: boolean
874   @return: whether the startup was successful or not
875
876   """
877   running_instances = GetInstanceList([instance.hypervisor])
878
879   if instance.name in running_instances:
880     return (True, "Already running")
881
882   try:
883     block_devices = _GatherAndLinkBlockDevs(instance)
884     hyper = hypervisor.GetHypervisor(instance.hypervisor)
885     hyper.StartInstance(instance, block_devices)
886   except errors.BlockDeviceError, err:
887     logging.exception("Failed to start instance")
888     return (False, "Block device error: %s" % str(err))
889   except errors.HypervisorError, err:
890     logging.exception("Failed to start instance")
891     _RemoveBlockDevLinks(instance.name, instance.disks)
892     return (False, "Hypervisor error: %s" % str(err))
893
894   return (True, "Instance started successfully")
895
896
897 def InstanceShutdown(instance):
898   """Shut an instance down.
899
900   @note: this functions uses polling with a hardcoded timeout.
901
902   @type instance: L{objects.Instance}
903   @param instance: the instance object
904   @rtype: boolean
905   @return: whether the startup was successful or not
906
907   """
908   hv_name = instance.hypervisor
909   running_instances = GetInstanceList([hv_name])
910
911   if instance.name not in running_instances:
912     return (True, "Instance already stopped")
913
914   hyper = hypervisor.GetHypervisor(hv_name)
915   try:
916     hyper.StopInstance(instance)
917   except errors.HypervisorError, err:
918     msg = "Failed to stop instance %s: %s" % (instance.name, err)
919     logging.error(msg)
920     return (False, msg)
921
922   # test every 10secs for 2min
923
924   time.sleep(1)
925   for dummy in range(11):
926     if instance.name not in GetInstanceList([hv_name]):
927       break
928     time.sleep(10)
929   else:
930     # the shutdown did not succeed
931     logging.error("Shutdown of '%s' unsuccessful, using destroy",
932                   instance.name)
933
934     try:
935       hyper.StopInstance(instance, force=True)
936     except errors.HypervisorError, err:
937       msg = "Failed to force stop instance %s: %s" % (instance.name, err)
938       logging.error(msg)
939       return (False, msg)
940
941     time.sleep(1)
942     if instance.name in GetInstanceList([hv_name]):
943       msg = ("Could not shutdown instance %s even by destroy" %
944              instance.name)
945       logging.error(msg)
946       return (False, msg)
947
948   _RemoveBlockDevLinks(instance.name, instance.disks)
949
950   return (True, "Instance has been shutdown successfully")
951
952
953 def InstanceReboot(instance, reboot_type):
954   """Reboot an instance.
955
956   @type instance: L{objects.Instance}
957   @param instance: the instance object to reboot
958   @type reboot_type: str
959   @param reboot_type: the type of reboot, one the following
960     constants:
961       - L{constants.INSTANCE_REBOOT_SOFT}: only reboot the
962         instance OS, do not recreate the VM
963       - L{constants.INSTANCE_REBOOT_HARD}: tear down and
964         restart the VM (at the hypervisor level)
965       - the other reboot type (L{constants.INSTANCE_REBOOT_HARD})
966         is not accepted here, since that mode is handled
967         differently
968   @rtype: boolean
969   @return: the success of the operation
970
971   """
972   running_instances = GetInstanceList([instance.hypervisor])
973
974   if instance.name not in running_instances:
975     msg = "Cannot reboot instance %s that is not running" % instance.name
976     logging.error(msg)
977     return (False, msg)
978
979   hyper = hypervisor.GetHypervisor(instance.hypervisor)
980   if reboot_type == constants.INSTANCE_REBOOT_SOFT:
981     try:
982       hyper.RebootInstance(instance)
983     except errors.HypervisorError, err:
984       msg = "Failed to soft reboot instance %s: %s" % (instance.name, err)
985       logging.error(msg)
986       return (False, msg)
987   elif reboot_type == constants.INSTANCE_REBOOT_HARD:
988     try:
989       stop_result = InstanceShutdown(instance)
990       if not stop_result[0]:
991         return stop_result
992       return StartInstance(instance)
993     except errors.HypervisorError, err:
994       msg = "Failed to hard reboot instance %s: %s" % (instance.name, err)
995       logging.error(msg)
996       return (False, msg)
997   else:
998     return (False, "Invalid reboot_type received: %s" % (reboot_type,))
999
1000   return (True, "Reboot successful")
1001
1002
1003 def MigrationInfo(instance):
1004   """Gather information about an instance to be migrated.
1005
1006   @type instance: L{objects.Instance}
1007   @param instance: the instance definition
1008
1009   """
1010   hyper = hypervisor.GetHypervisor(instance.hypervisor)
1011   try:
1012     info = hyper.MigrationInfo(instance)
1013   except errors.HypervisorError, err:
1014     msg = "Failed to fetch migration information"
1015     logging.exception(msg)
1016     return (False, '%s: %s' % (msg, err))
1017   return (True, info)
1018
1019
1020 def AcceptInstance(instance, info, target):
1021   """Prepare the node to accept an instance.
1022
1023   @type instance: L{objects.Instance}
1024   @param instance: the instance definition
1025   @type info: string/data (opaque)
1026   @param info: migration information, from the source node
1027   @type target: string
1028   @param target: target host (usually ip), on this node
1029
1030   """
1031   hyper = hypervisor.GetHypervisor(instance.hypervisor)
1032   try:
1033     hyper.AcceptInstance(instance, info, target)
1034   except errors.HypervisorError, err:
1035     msg = "Failed to accept instance"
1036     logging.exception(msg)
1037     return (False, '%s: %s' % (msg, err))
1038   return (True, "Accept successfull")
1039
1040
1041 def FinalizeMigration(instance, info, success):
1042   """Finalize any preparation to accept an instance.
1043
1044   @type instance: L{objects.Instance}
1045   @param instance: the instance definition
1046   @type info: string/data (opaque)
1047   @param info: migration information, from the source node
1048   @type success: boolean
1049   @param success: whether the migration was a success or a failure
1050
1051   """
1052   hyper = hypervisor.GetHypervisor(instance.hypervisor)
1053   try:
1054     hyper.FinalizeMigration(instance, info, success)
1055   except errors.HypervisorError, err:
1056     msg = "Failed to finalize migration"
1057     logging.exception(msg)
1058     return (False, '%s: %s' % (msg, err))
1059   return (True, "Migration Finalized")
1060
1061
1062 def MigrateInstance(instance, target, live):
1063   """Migrates an instance to another node.
1064
1065   @type instance: L{objects.Instance}
1066   @param instance: the instance definition
1067   @type target: string
1068   @param target: the target node name
1069   @type live: boolean
1070   @param live: whether the migration should be done live or not (the
1071       interpretation of this parameter is left to the hypervisor)
1072   @rtype: tuple
1073   @return: a tuple of (success, msg) where:
1074       - succes is a boolean denoting the success/failure of the operation
1075       - msg is a string with details in case of failure
1076
1077   """
1078   hyper = hypervisor.GetHypervisor(instance.hypervisor)
1079
1080   try:
1081     hyper.MigrateInstance(instance.name, target, live)
1082   except errors.HypervisorError, err:
1083     msg = "Failed to migrate instance"
1084     logging.exception(msg)
1085     return (False, "%s: %s" % (msg, err))
1086   return (True, "Migration successfull")
1087
1088
1089 def BlockdevCreate(disk, size, owner, on_primary, info):
1090   """Creates a block device for an instance.
1091
1092   @type disk: L{objects.Disk}
1093   @param disk: the object describing the disk we should create
1094   @type size: int
1095   @param size: the size of the physical underlying device, in MiB
1096   @type owner: str
1097   @param owner: the name of the instance for which disk is created,
1098       used for device cache data
1099   @type on_primary: boolean
1100   @param on_primary:  indicates if it is the primary node or not
1101   @type info: string
1102   @param info: string that will be sent to the physical device
1103       creation, used for example to set (LVM) tags on LVs
1104
1105   @return: the new unique_id of the device (this can sometime be
1106       computed only after creation), or None. On secondary nodes,
1107       it's not required to return anything.
1108
1109   """
1110   clist = []
1111   if disk.children:
1112     for child in disk.children:
1113       try:
1114         crdev = _RecursiveAssembleBD(child, owner, on_primary)
1115       except errors.BlockDeviceError, err:
1116         errmsg = "Can't assemble device %s: %s" % (child, err)
1117         logging.error(errmsg)
1118         return False, errmsg
1119       if on_primary or disk.AssembleOnSecondary():
1120         # we need the children open in case the device itself has to
1121         # be assembled
1122         try:
1123           crdev.Open()
1124         except errors.BlockDeviceError, err:
1125           errmsg = "Can't make child '%s' read-write: %s" % (child, err)
1126           logging.error(errmsg)
1127           return False, errmsg
1128       clist.append(crdev)
1129
1130   try:
1131     device = bdev.Create(disk.dev_type, disk.physical_id, clist, size)
1132   except errors.BlockDeviceError, err:
1133     return False, "Can't create block device: %s" % str(err)
1134
1135   if on_primary or disk.AssembleOnSecondary():
1136     try:
1137       device.Assemble()
1138     except errors.BlockDeviceError, err:
1139       errmsg = ("Can't assemble device after creation, very"
1140                 " unusual event: %s" % str(err))
1141       logging.error(errmsg)
1142       return False, errmsg
1143     device.SetSyncSpeed(constants.SYNC_SPEED)
1144     if on_primary or disk.OpenOnSecondary():
1145       try:
1146         device.Open(force=True)
1147       except errors.BlockDeviceError, err:
1148         errmsg = ("Can't make device r/w after creation, very"
1149                   " unusual event: %s" % str(err))
1150         logging.error(errmsg)
1151         return False, errmsg
1152     DevCacheManager.UpdateCache(device.dev_path, owner,
1153                                 on_primary, disk.iv_name)
1154
1155   device.SetInfo(info)
1156
1157   physical_id = device.unique_id
1158   return True, physical_id
1159
1160
1161 def BlockdevRemove(disk):
1162   """Remove a block device.
1163
1164   @note: This is intended to be called recursively.
1165
1166   @type disk: L{objects.Disk}
1167   @param disk: the disk object we should remove
1168   @rtype: boolean
1169   @return: the success of the operation
1170
1171   """
1172   msgs = []
1173   result = True
1174   try:
1175     rdev = _RecursiveFindBD(disk)
1176   except errors.BlockDeviceError, err:
1177     # probably can't attach
1178     logging.info("Can't attach to device %s in remove", disk)
1179     rdev = None
1180   if rdev is not None:
1181     r_path = rdev.dev_path
1182     try:
1183       rdev.Remove()
1184     except errors.BlockDeviceError, err:
1185       msgs.append(str(err))
1186       result = False
1187     if result:
1188       DevCacheManager.RemoveCache(r_path)
1189
1190   if disk.children:
1191     for child in disk.children:
1192       c_status, c_msg = BlockdevRemove(child)
1193       result = result and c_status
1194       if c_msg: # not an empty message
1195         msgs.append(c_msg)
1196
1197   return (result, "; ".join(msgs))
1198
1199
1200 def _RecursiveAssembleBD(disk, owner, as_primary):
1201   """Activate a block device for an instance.
1202
1203   This is run on the primary and secondary nodes for an instance.
1204
1205   @note: this function is called recursively.
1206
1207   @type disk: L{objects.Disk}
1208   @param disk: the disk we try to assemble
1209   @type owner: str
1210   @param owner: the name of the instance which owns the disk
1211   @type as_primary: boolean
1212   @param as_primary: if we should make the block device
1213       read/write
1214
1215   @return: the assembled device or None (in case no device
1216       was assembled)
1217   @raise errors.BlockDeviceError: in case there is an error
1218       during the activation of the children or the device
1219       itself
1220
1221   """
1222   children = []
1223   if disk.children:
1224     mcn = disk.ChildrenNeeded()
1225     if mcn == -1:
1226       mcn = 0 # max number of Nones allowed
1227     else:
1228       mcn = len(disk.children) - mcn # max number of Nones
1229     for chld_disk in disk.children:
1230       try:
1231         cdev = _RecursiveAssembleBD(chld_disk, owner, as_primary)
1232       except errors.BlockDeviceError, err:
1233         if children.count(None) >= mcn:
1234           raise
1235         cdev = None
1236         logging.error("Error in child activation (but continuing): %s",
1237                       str(err))
1238       children.append(cdev)
1239
1240   if as_primary or disk.AssembleOnSecondary():
1241     r_dev = bdev.Assemble(disk.dev_type, disk.physical_id, children)
1242     r_dev.SetSyncSpeed(constants.SYNC_SPEED)
1243     result = r_dev
1244     if as_primary or disk.OpenOnSecondary():
1245       r_dev.Open()
1246     DevCacheManager.UpdateCache(r_dev.dev_path, owner,
1247                                 as_primary, disk.iv_name)
1248
1249   else:
1250     result = True
1251   return result
1252
1253
1254 def BlockdevAssemble(disk, owner, as_primary):
1255   """Activate a block device for an instance.
1256
1257   This is a wrapper over _RecursiveAssembleBD.
1258
1259   @rtype: str or boolean
1260   @return: a C{/dev/...} path for primary nodes, and
1261       C{True} for secondary nodes
1262
1263   """
1264   status = True
1265   result = "no error information"
1266   try:
1267     result = _RecursiveAssembleBD(disk, owner, as_primary)
1268     if isinstance(result, bdev.BlockDev):
1269       result = result.dev_path
1270   except errors.BlockDeviceError, err:
1271     result = "Error while assembling disk: %s" % str(err)
1272     status = False
1273   return (status, result)
1274
1275
1276 def BlockdevShutdown(disk):
1277   """Shut down a block device.
1278
1279   First, if the device is assembled (Attach() is successfull), then
1280   the device is shutdown. Then the children of the device are
1281   shutdown.
1282
1283   This function is called recursively. Note that we don't cache the
1284   children or such, as oppossed to assemble, shutdown of different
1285   devices doesn't require that the upper device was active.
1286
1287   @type disk: L{objects.Disk}
1288   @param disk: the description of the disk we should
1289       shutdown
1290   @rtype: boolean
1291   @return: the success of the operation
1292
1293   """
1294   msgs = []
1295   result = True
1296   r_dev = _RecursiveFindBD(disk)
1297   if r_dev is not None:
1298     r_path = r_dev.dev_path
1299     try:
1300       r_dev.Shutdown()
1301       DevCacheManager.RemoveCache(r_path)
1302     except errors.BlockDeviceError, err:
1303       msgs.append(str(err))
1304       result = False
1305
1306   if disk.children:
1307     for child in disk.children:
1308       c_status, c_msg = BlockdevShutdown(child)
1309       result = result and c_status
1310       if c_msg: # not an empty message
1311         msgs.append(c_msg)
1312
1313   return (result, "; ".join(msgs))
1314
1315
1316 def BlockdevAddchildren(parent_cdev, new_cdevs):
1317   """Extend a mirrored block device.
1318
1319   @type parent_cdev: L{objects.Disk}
1320   @param parent_cdev: the disk to which we should add children
1321   @type new_cdevs: list of L{objects.Disk}
1322   @param new_cdevs: the list of children which we should add
1323   @rtype: boolean
1324   @return: the success of the operation
1325
1326   """
1327   parent_bdev = _RecursiveFindBD(parent_cdev)
1328   if parent_bdev is None:
1329     msg = "Can't find parent device '%s' in add children" % str(parent_cdev)
1330     logging.error("BlockdevAddchildren: %s", msg)
1331     return (False, msg)
1332   new_bdevs = [_RecursiveFindBD(disk) for disk in new_cdevs]
1333   if new_bdevs.count(None) > 0:
1334     msg = "Can't find new device(s) to add: %s:%s" % (new_bdevs, new_cdevs)
1335     logging.error(msg)
1336     return (False, msg)
1337   parent_bdev.AddChildren(new_bdevs)
1338   return (True, None)
1339
1340
1341 def BlockdevRemovechildren(parent_cdev, new_cdevs):
1342   """Shrink a mirrored block device.
1343
1344   @type parent_cdev: L{objects.Disk}
1345   @param parent_cdev: the disk from which we should remove children
1346   @type new_cdevs: list of L{objects.Disk}
1347   @param new_cdevs: the list of children which we should remove
1348   @rtype: boolean
1349   @return: the success of the operation
1350
1351   """
1352   parent_bdev = _RecursiveFindBD(parent_cdev)
1353   if parent_bdev is None:
1354     msg = "Can't find parent device '%s' in remove children" % str(parent_cdev)
1355     logging.error(msg)
1356     return (False, msg)
1357   devs = []
1358   for disk in new_cdevs:
1359     rpath = disk.StaticDevPath()
1360     if rpath is None:
1361       bd = _RecursiveFindBD(disk)
1362       if bd is None:
1363         msg = "Can't find device %s while removing children" % (disk,)
1364         logging.error(msg)
1365         return (False, msg)
1366       else:
1367         devs.append(bd.dev_path)
1368     else:
1369       devs.append(rpath)
1370   parent_bdev.RemoveChildren(devs)
1371   return (True, None)
1372
1373
1374 def BlockdevGetmirrorstatus(disks):
1375   """Get the mirroring status of a list of devices.
1376
1377   @type disks: list of L{objects.Disk}
1378   @param disks: the list of disks which we should query
1379   @rtype: disk
1380   @return:
1381       a list of (mirror_done, estimated_time) tuples, which
1382       are the result of L{bdev.BlockDev.CombinedSyncStatus}
1383   @raise errors.BlockDeviceError: if any of the disks cannot be
1384       found
1385
1386   """
1387   stats = []
1388   for dsk in disks:
1389     rbd = _RecursiveFindBD(dsk)
1390     if rbd is None:
1391       raise errors.BlockDeviceError("Can't find device %s" % str(dsk))
1392     stats.append(rbd.CombinedSyncStatus())
1393   return stats
1394
1395
1396 def _RecursiveFindBD(disk):
1397   """Check if a device is activated.
1398
1399   If so, return informations about the real device.
1400
1401   @type disk: L{objects.Disk}
1402   @param disk: the disk object we need to find
1403
1404   @return: None if the device can't be found,
1405       otherwise the device instance
1406
1407   """
1408   children = []
1409   if disk.children:
1410     for chdisk in disk.children:
1411       children.append(_RecursiveFindBD(chdisk))
1412
1413   return bdev.FindDevice(disk.dev_type, disk.physical_id, children)
1414
1415
1416 def BlockdevFind(disk):
1417   """Check if a device is activated.
1418
1419   If it is, return informations about the real device.
1420
1421   @type disk: L{objects.Disk}
1422   @param disk: the disk to find
1423   @rtype: None or tuple
1424   @return: None if the disk cannot be found, otherwise a
1425       tuple (device_path, major, minor, sync_percent,
1426       estimated_time, is_degraded)
1427
1428   """
1429   try:
1430     rbd = _RecursiveFindBD(disk)
1431   except errors.BlockDeviceError, err:
1432     return (False, str(err))
1433   if rbd is None:
1434     return (True, None)
1435   return (True, (rbd.dev_path, rbd.major, rbd.minor) + rbd.GetSyncStatus())
1436
1437
1438 def UploadFile(file_name, data, mode, uid, gid, atime, mtime):
1439   """Write a file to the filesystem.
1440
1441   This allows the master to overwrite(!) a file. It will only perform
1442   the operation if the file belongs to a list of configuration files.
1443
1444   @type file_name: str
1445   @param file_name: the target file name
1446   @type data: str
1447   @param data: the new contents of the file
1448   @type mode: int
1449   @param mode: the mode to give the file (can be None)
1450   @type uid: int
1451   @param uid: the owner of the file (can be -1 for default)
1452   @type gid: int
1453   @param gid: the group of the file (can be -1 for default)
1454   @type atime: float
1455   @param atime: the atime to set on the file (can be None)
1456   @type mtime: float
1457   @param mtime: the mtime to set on the file (can be None)
1458   @rtype: boolean
1459   @return: the success of the operation; errors are logged
1460       in the node daemon log
1461
1462   """
1463   if not os.path.isabs(file_name):
1464     err = "Filename passed to UploadFile is not absolute: '%s'" % file_name
1465     logging.error(err)
1466     return (False, err)
1467
1468   allowed_files = set([
1469     constants.CLUSTER_CONF_FILE,
1470     constants.ETC_HOSTS,
1471     constants.SSH_KNOWN_HOSTS_FILE,
1472     constants.VNC_PASSWORD_FILE,
1473     constants.RAPI_CERT_FILE,
1474     constants.RAPI_USERS_FILE,
1475     ])
1476
1477   for hv_name in constants.HYPER_TYPES:
1478     hv_class = hypervisor.GetHypervisor(hv_name)
1479     allowed_files.update(hv_class.GetAncillaryFiles())
1480
1481   if file_name not in allowed_files:
1482     err = "Filename passed to UploadFile not in allowed upload targets: '%s'" \
1483           % file_name
1484     logging.error(err)
1485     return (False, err)
1486
1487   raw_data = _Decompress(data)
1488
1489   utils.WriteFile(file_name, data=raw_data, mode=mode, uid=uid, gid=gid,
1490                   atime=atime, mtime=mtime)
1491   return (True, "success")
1492
1493
1494 def WriteSsconfFiles(values):
1495   """Update all ssconf files.
1496
1497   Wrapper around the SimpleStore.WriteFiles.
1498
1499   """
1500   ssconf.SimpleStore().WriteFiles(values)
1501
1502
1503 def _ErrnoOrStr(err):
1504   """Format an EnvironmentError exception.
1505
1506   If the L{err} argument has an errno attribute, it will be looked up
1507   and converted into a textual C{E...} description. Otherwise the
1508   string representation of the error will be returned.
1509
1510   @type err: L{EnvironmentError}
1511   @param err: the exception to format
1512
1513   """
1514   if hasattr(err, 'errno'):
1515     detail = errno.errorcode[err.errno]
1516   else:
1517     detail = str(err)
1518   return detail
1519
1520
1521 def _OSOndiskVersion(name, os_dir):
1522   """Compute and return the API version of a given OS.
1523
1524   This function will try to read the API version of the OS given by
1525   the 'name' parameter and residing in the 'os_dir' directory.
1526
1527   @type name: str
1528   @param name: the OS name we should look for
1529   @type os_dir: str
1530   @param os_dir: the directory inwhich we should look for the OS
1531   @rtype: int or None
1532   @return:
1533       Either an integer denoting the version or None in the
1534       case when this is not a valid OS name.
1535   @raise errors.InvalidOS: if the OS cannot be found
1536
1537   """
1538   api_file = os.path.sep.join([os_dir, "ganeti_api_version"])
1539
1540   try:
1541     st = os.stat(api_file)
1542   except EnvironmentError, err:
1543     raise errors.InvalidOS(name, os_dir, "'ganeti_api_version' file not"
1544                            " found (%s)" % _ErrnoOrStr(err))
1545
1546   if not stat.S_ISREG(stat.S_IFMT(st.st_mode)):
1547     raise errors.InvalidOS(name, os_dir, "'ganeti_api_version' file is not"
1548                            " a regular file")
1549
1550   try:
1551     f = open(api_file)
1552     try:
1553       api_versions = f.readlines()
1554     finally:
1555       f.close()
1556   except EnvironmentError, err:
1557     raise errors.InvalidOS(name, os_dir, "error while reading the"
1558                            " API version (%s)" % _ErrnoOrStr(err))
1559
1560   api_versions = [version.strip() for version in api_versions]
1561   try:
1562     api_versions = [int(version) for version in api_versions]
1563   except (TypeError, ValueError), err:
1564     raise errors.InvalidOS(name, os_dir,
1565                            "API version is not integer (%s)" % str(err))
1566
1567   return api_versions
1568
1569
1570 def DiagnoseOS(top_dirs=None):
1571   """Compute the validity for all OSes.
1572
1573   @type top_dirs: list
1574   @param top_dirs: the list of directories in which to
1575       search (if not given defaults to
1576       L{constants.OS_SEARCH_PATH})
1577   @rtype: list of L{objects.OS}
1578   @return: an OS object for each name in all the given
1579       directories
1580
1581   """
1582   if top_dirs is None:
1583     top_dirs = constants.OS_SEARCH_PATH
1584
1585   result = []
1586   for dir_name in top_dirs:
1587     if os.path.isdir(dir_name):
1588       try:
1589         f_names = utils.ListVisibleFiles(dir_name)
1590       except EnvironmentError, err:
1591         logging.exception("Can't list the OS directory %s", dir_name)
1592         break
1593       for name in f_names:
1594         try:
1595           os_inst = OSFromDisk(name, base_dir=dir_name)
1596           result.append(os_inst)
1597         except errors.InvalidOS, err:
1598           result.append(objects.OS.FromInvalidOS(err))
1599
1600   return result
1601
1602
1603 def OSFromDisk(name, base_dir=None):
1604   """Create an OS instance from disk.
1605
1606   This function will return an OS instance if the given name is a
1607   valid OS name. Otherwise, it will raise an appropriate
1608   L{errors.InvalidOS} exception, detailing why this is not a valid OS.
1609
1610   @type base_dir: string
1611   @keyword base_dir: Base directory containing OS installations.
1612                      Defaults to a search in all the OS_SEARCH_PATH dirs.
1613   @rtype: L{objects.OS}
1614   @return: the OS instance if we find a valid one
1615   @raise errors.InvalidOS: if we don't find a valid OS
1616
1617   """
1618   if base_dir is None:
1619     os_dir = utils.FindFile(name, constants.OS_SEARCH_PATH, os.path.isdir)
1620     if os_dir is None:
1621       raise errors.InvalidOS(name, None, "OS dir not found in search path")
1622   else:
1623     os_dir = os.path.sep.join([base_dir, name])
1624
1625   api_versions = _OSOndiskVersion(name, os_dir)
1626
1627   if constants.OS_API_VERSION not in api_versions:
1628     raise errors.InvalidOS(name, os_dir, "API version mismatch"
1629                            " (found %s want %s)"
1630                            % (api_versions, constants.OS_API_VERSION))
1631
1632   # OS Scripts dictionary, we will populate it with the actual script names
1633   os_scripts = dict.fromkeys(constants.OS_SCRIPTS)
1634
1635   for script in os_scripts:
1636     os_scripts[script] = os.path.sep.join([os_dir, script])
1637
1638     try:
1639       st = os.stat(os_scripts[script])
1640     except EnvironmentError, err:
1641       raise errors.InvalidOS(name, os_dir, "'%s' script missing (%s)" %
1642                              (script, _ErrnoOrStr(err)))
1643
1644     if stat.S_IMODE(st.st_mode) & stat.S_IXUSR != stat.S_IXUSR:
1645       raise errors.InvalidOS(name, os_dir, "'%s' script not executable" %
1646                              script)
1647
1648     if not stat.S_ISREG(stat.S_IFMT(st.st_mode)):
1649       raise errors.InvalidOS(name, os_dir, "'%s' is not a regular file" %
1650                              script)
1651
1652
1653   return objects.OS(name=name, path=os_dir, status=constants.OS_VALID_STATUS,
1654                     create_script=os_scripts[constants.OS_SCRIPT_CREATE],
1655                     export_script=os_scripts[constants.OS_SCRIPT_EXPORT],
1656                     import_script=os_scripts[constants.OS_SCRIPT_IMPORT],
1657                     rename_script=os_scripts[constants.OS_SCRIPT_RENAME],
1658                     api_versions=api_versions)
1659
1660 def OSEnvironment(instance, debug=0):
1661   """Calculate the environment for an os script.
1662
1663   @type instance: L{objects.Instance}
1664   @param instance: target instance for the os script run
1665   @type debug: integer
1666   @param debug: debug level (0 or 1, for OS Api 10)
1667   @rtype: dict
1668   @return: dict of environment variables
1669   @raise errors.BlockDeviceError: if the block device
1670       cannot be found
1671
1672   """
1673   result = {}
1674   result['OS_API_VERSION'] = '%d' % constants.OS_API_VERSION
1675   result['INSTANCE_NAME'] = instance.name
1676   result['INSTANCE_OS'] = instance.os
1677   result['HYPERVISOR'] = instance.hypervisor
1678   result['DISK_COUNT'] = '%d' % len(instance.disks)
1679   result['NIC_COUNT'] = '%d' % len(instance.nics)
1680   result['DEBUG_LEVEL'] = '%d' % debug
1681   for idx, disk in enumerate(instance.disks):
1682     real_disk = _RecursiveFindBD(disk)
1683     if real_disk is None:
1684       raise errors.BlockDeviceError("Block device '%s' is not set up" %
1685                                     str(disk))
1686     real_disk.Open()
1687     result['DISK_%d_PATH' % idx] = real_disk.dev_path
1688     result['DISK_%d_ACCESS' % idx] = disk.mode
1689     if constants.HV_DISK_TYPE in instance.hvparams:
1690       result['DISK_%d_FRONTEND_TYPE' % idx] = \
1691         instance.hvparams[constants.HV_DISK_TYPE]
1692     if disk.dev_type in constants.LDS_BLOCK:
1693       result['DISK_%d_BACKEND_TYPE' % idx] = 'block'
1694     elif disk.dev_type == constants.LD_FILE:
1695       result['DISK_%d_BACKEND_TYPE' % idx] = \
1696         'file:%s' % disk.physical_id[0]
1697   for idx, nic in enumerate(instance.nics):
1698     result['NIC_%d_MAC' % idx] = nic.mac
1699     if nic.ip:
1700       result['NIC_%d_IP' % idx] = nic.ip
1701     result['NIC_%d_BRIDGE' % idx] = nic.bridge
1702     if constants.HV_NIC_TYPE in instance.hvparams:
1703       result['NIC_%d_FRONTEND_TYPE' % idx] = \
1704         instance.hvparams[constants.HV_NIC_TYPE]
1705
1706   return result
1707
1708 def BlockdevGrow(disk, amount):
1709   """Grow a stack of block devices.
1710
1711   This function is called recursively, with the childrens being the
1712   first ones to resize.
1713
1714   @type disk: L{objects.Disk}
1715   @param disk: the disk to be grown
1716   @rtype: (status, result)
1717   @return: a tuple with the status of the operation
1718       (True/False), and the errors message if status
1719       is False
1720
1721   """
1722   r_dev = _RecursiveFindBD(disk)
1723   if r_dev is None:
1724     return False, "Cannot find block device %s" % (disk,)
1725
1726   try:
1727     r_dev.Grow(amount)
1728   except errors.BlockDeviceError, err:
1729     return False, str(err)
1730
1731   return True, None
1732
1733
1734 def BlockdevSnapshot(disk):
1735   """Create a snapshot copy of a block device.
1736
1737   This function is called recursively, and the snapshot is actually created
1738   just for the leaf lvm backend device.
1739
1740   @type disk: L{objects.Disk}
1741   @param disk: the disk to be snapshotted
1742   @rtype: string
1743   @return: snapshot disk path
1744
1745   """
1746   if disk.children:
1747     if len(disk.children) == 1:
1748       # only one child, let's recurse on it
1749       return BlockdevSnapshot(disk.children[0])
1750     else:
1751       # more than one child, choose one that matches
1752       for child in disk.children:
1753         if child.size == disk.size:
1754           # return implies breaking the loop
1755           return BlockdevSnapshot(child)
1756   elif disk.dev_type == constants.LD_LV:
1757     r_dev = _RecursiveFindBD(disk)
1758     if r_dev is not None:
1759       # let's stay on the safe side and ask for the full size, for now
1760       return r_dev.Snapshot(disk.size)
1761     else:
1762       return None
1763   else:
1764     raise errors.ProgrammerError("Cannot snapshot non-lvm block device"
1765                                  " '%s' of type '%s'" %
1766                                  (disk.unique_id, disk.dev_type))
1767
1768
1769 def ExportSnapshot(disk, dest_node, instance, cluster_name, idx):
1770   """Export a block device snapshot to a remote node.
1771
1772   @type disk: L{objects.Disk}
1773   @param disk: the description of the disk to export
1774   @type dest_node: str
1775   @param dest_node: the destination node to export to
1776   @type instance: L{objects.Instance}
1777   @param instance: the instance object to whom the disk belongs
1778   @type cluster_name: str
1779   @param cluster_name: the cluster name, needed for SSH hostalias
1780   @type idx: int
1781   @param idx: the index of the disk in the instance's disk list,
1782       used to export to the OS scripts environment
1783   @rtype: boolean
1784   @return: the success of the operation
1785
1786   """
1787   export_env = OSEnvironment(instance)
1788
1789   inst_os = OSFromDisk(instance.os)
1790   export_script = inst_os.export_script
1791
1792   logfile = "%s/exp-%s-%s-%s.log" % (constants.LOG_OS_DIR, inst_os.name,
1793                                      instance.name, int(time.time()))
1794   if not os.path.exists(constants.LOG_OS_DIR):
1795     os.mkdir(constants.LOG_OS_DIR, 0750)
1796   real_disk = _RecursiveFindBD(disk)
1797   if real_disk is None:
1798     raise errors.BlockDeviceError("Block device '%s' is not set up" %
1799                                   str(disk))
1800   real_disk.Open()
1801
1802   export_env['EXPORT_DEVICE'] = real_disk.dev_path
1803   export_env['EXPORT_INDEX'] = str(idx)
1804
1805   destdir = os.path.join(constants.EXPORT_DIR, instance.name + ".new")
1806   destfile = disk.physical_id[1]
1807
1808   # the target command is built out of three individual commands,
1809   # which are joined by pipes; we check each individual command for
1810   # valid parameters
1811   expcmd = utils.BuildShellCmd("cd %s; %s 2>%s", inst_os.path,
1812                                export_script, logfile)
1813
1814   comprcmd = "gzip"
1815
1816   destcmd = utils.BuildShellCmd("mkdir -p %s && cat > %s/%s",
1817                                 destdir, destdir, destfile)
1818   remotecmd = _GetSshRunner(cluster_name).BuildCmd(dest_node,
1819                                                    constants.GANETI_RUNAS,
1820                                                    destcmd)
1821
1822   # all commands have been checked, so we're safe to combine them
1823   command = '|'.join([expcmd, comprcmd, utils.ShellQuoteArgs(remotecmd)])
1824
1825   result = utils.RunCmd(command, env=export_env)
1826
1827   if result.failed:
1828     logging.error("os snapshot export command '%s' returned error: %s"
1829                   " output: %s", command, result.fail_reason, result.output)
1830     return False
1831
1832   return True
1833
1834
1835 def FinalizeExport(instance, snap_disks):
1836   """Write out the export configuration information.
1837
1838   @type instance: L{objects.Instance}
1839   @param instance: the instance which we export, used for
1840       saving configuration
1841   @type snap_disks: list of L{objects.Disk}
1842   @param snap_disks: list of snapshot block devices, which
1843       will be used to get the actual name of the dump file
1844
1845   @rtype: boolean
1846   @return: the success of the operation
1847
1848   """
1849   destdir = os.path.join(constants.EXPORT_DIR, instance.name + ".new")
1850   finaldestdir = os.path.join(constants.EXPORT_DIR, instance.name)
1851
1852   config = objects.SerializableConfigParser()
1853
1854   config.add_section(constants.INISECT_EXP)
1855   config.set(constants.INISECT_EXP, 'version', '0')
1856   config.set(constants.INISECT_EXP, 'timestamp', '%d' % int(time.time()))
1857   config.set(constants.INISECT_EXP, 'source', instance.primary_node)
1858   config.set(constants.INISECT_EXP, 'os', instance.os)
1859   config.set(constants.INISECT_EXP, 'compression', 'gzip')
1860
1861   config.add_section(constants.INISECT_INS)
1862   config.set(constants.INISECT_INS, 'name', instance.name)
1863   config.set(constants.INISECT_INS, 'memory', '%d' %
1864              instance.beparams[constants.BE_MEMORY])
1865   config.set(constants.INISECT_INS, 'vcpus', '%d' %
1866              instance.beparams[constants.BE_VCPUS])
1867   config.set(constants.INISECT_INS, 'disk_template', instance.disk_template)
1868
1869   nic_total = 0
1870   for nic_count, nic in enumerate(instance.nics):
1871     nic_total += 1
1872     config.set(constants.INISECT_INS, 'nic%d_mac' %
1873                nic_count, '%s' % nic.mac)
1874     config.set(constants.INISECT_INS, 'nic%d_ip' % nic_count, '%s' % nic.ip)
1875     config.set(constants.INISECT_INS, 'nic%d_bridge' % nic_count,
1876                '%s' % nic.bridge)
1877   # TODO: redundant: on load can read nics until it doesn't exist
1878   config.set(constants.INISECT_INS, 'nic_count' , '%d' % nic_total)
1879
1880   disk_total = 0
1881   for disk_count, disk in enumerate(snap_disks):
1882     if disk:
1883       disk_total += 1
1884       config.set(constants.INISECT_INS, 'disk%d_ivname' % disk_count,
1885                  ('%s' % disk.iv_name))
1886       config.set(constants.INISECT_INS, 'disk%d_dump' % disk_count,
1887                  ('%s' % disk.physical_id[1]))
1888       config.set(constants.INISECT_INS, 'disk%d_size' % disk_count,
1889                  ('%d' % disk.size))
1890
1891   config.set(constants.INISECT_INS, 'disk_count' , '%d' % disk_total)
1892
1893   utils.WriteFile(os.path.join(destdir, constants.EXPORT_CONF_FILE),
1894                   data=config.Dumps())
1895   shutil.rmtree(finaldestdir, True)
1896   shutil.move(destdir, finaldestdir)
1897
1898   return True
1899
1900
1901 def ExportInfo(dest):
1902   """Get export configuration information.
1903
1904   @type dest: str
1905   @param dest: directory containing the export
1906
1907   @rtype: L{objects.SerializableConfigParser}
1908   @return: a serializable config file containing the
1909       export info
1910
1911   """
1912   cff = os.path.join(dest, constants.EXPORT_CONF_FILE)
1913
1914   config = objects.SerializableConfigParser()
1915   config.read(cff)
1916
1917   if (not config.has_section(constants.INISECT_EXP) or
1918       not config.has_section(constants.INISECT_INS)):
1919     return None
1920
1921   return config
1922
1923
1924 def ImportOSIntoInstance(instance, src_node, src_images, cluster_name):
1925   """Import an os image into an instance.
1926
1927   @type instance: L{objects.Instance}
1928   @param instance: instance to import the disks into
1929   @type src_node: string
1930   @param src_node: source node for the disk images
1931   @type src_images: list of string
1932   @param src_images: absolute paths of the disk images
1933   @rtype: list of boolean
1934   @return: each boolean represent the success of importing the n-th disk
1935
1936   """
1937   import_env = OSEnvironment(instance)
1938   inst_os = OSFromDisk(instance.os)
1939   import_script = inst_os.import_script
1940
1941   logfile = "%s/import-%s-%s-%s.log" % (constants.LOG_OS_DIR, instance.os,
1942                                         instance.name, int(time.time()))
1943   if not os.path.exists(constants.LOG_OS_DIR):
1944     os.mkdir(constants.LOG_OS_DIR, 0750)
1945
1946   comprcmd = "gunzip"
1947   impcmd = utils.BuildShellCmd("(cd %s; %s >%s 2>&1)", inst_os.path,
1948                                import_script, logfile)
1949
1950   final_result = []
1951   for idx, image in enumerate(src_images):
1952     if image:
1953       destcmd = utils.BuildShellCmd('cat %s', image)
1954       remotecmd = _GetSshRunner(cluster_name).BuildCmd(src_node,
1955                                                        constants.GANETI_RUNAS,
1956                                                        destcmd)
1957       command = '|'.join([utils.ShellQuoteArgs(remotecmd), comprcmd, impcmd])
1958       import_env['IMPORT_DEVICE'] = import_env['DISK_%d_PATH' % idx]
1959       import_env['IMPORT_INDEX'] = str(idx)
1960       result = utils.RunCmd(command, env=import_env)
1961       if result.failed:
1962         logging.error("Disk import command '%s' returned error: %s"
1963                       " output: %s", command, result.fail_reason,
1964                       result.output)
1965         final_result.append(False)
1966       else:
1967         final_result.append(True)
1968     else:
1969       final_result.append(True)
1970
1971   return final_result
1972
1973
1974 def ListExports():
1975   """Return a list of exports currently available on this machine.
1976
1977   @rtype: list
1978   @return: list of the exports
1979
1980   """
1981   if os.path.isdir(constants.EXPORT_DIR):
1982     return utils.ListVisibleFiles(constants.EXPORT_DIR)
1983   else:
1984     return []
1985
1986
1987 def RemoveExport(export):
1988   """Remove an existing export from the node.
1989
1990   @type export: str
1991   @param export: the name of the export to remove
1992   @rtype: boolean
1993   @return: the success of the operation
1994
1995   """
1996   target = os.path.join(constants.EXPORT_DIR, export)
1997
1998   shutil.rmtree(target)
1999   # TODO: catch some of the relevant exceptions and provide a pretty
2000   # error message if rmtree fails.
2001
2002   return True
2003
2004
2005 def BlockdevRename(devlist):
2006   """Rename a list of block devices.
2007
2008   @type devlist: list of tuples
2009   @param devlist: list of tuples of the form  (disk,
2010       new_logical_id, new_physical_id); disk is an
2011       L{objects.Disk} object describing the current disk,
2012       and new logical_id/physical_id is the name we
2013       rename it to
2014   @rtype: boolean
2015   @return: True if all renames succeeded, False otherwise
2016
2017   """
2018   msgs = []
2019   result = True
2020   for disk, unique_id in devlist:
2021     dev = _RecursiveFindBD(disk)
2022     if dev is None:
2023       msgs.append("Can't find device %s in rename" % str(disk))
2024       result = False
2025       continue
2026     try:
2027       old_rpath = dev.dev_path
2028       dev.Rename(unique_id)
2029       new_rpath = dev.dev_path
2030       if old_rpath != new_rpath:
2031         DevCacheManager.RemoveCache(old_rpath)
2032         # FIXME: we should add the new cache information here, like:
2033         # DevCacheManager.UpdateCache(new_rpath, owner, ...)
2034         # but we don't have the owner here - maybe parse from existing
2035         # cache? for now, we only lose lvm data when we rename, which
2036         # is less critical than DRBD or MD
2037     except errors.BlockDeviceError, err:
2038       msgs.append("Can't rename device '%s' to '%s': %s" %
2039                   (dev, unique_id, err))
2040       logging.exception("Can't rename device '%s' to '%s'", dev, unique_id)
2041       result = False
2042   return (result, "; ".join(msgs))
2043
2044
2045 def _TransformFileStorageDir(file_storage_dir):
2046   """Checks whether given file_storage_dir is valid.
2047
2048   Checks wheter the given file_storage_dir is within the cluster-wide
2049   default file_storage_dir stored in SimpleStore. Only paths under that
2050   directory are allowed.
2051
2052   @type file_storage_dir: str
2053   @param file_storage_dir: the path to check
2054
2055   @return: the normalized path if valid, None otherwise
2056
2057   """
2058   cfg = _GetConfig()
2059   file_storage_dir = os.path.normpath(file_storage_dir)
2060   base_file_storage_dir = cfg.GetFileStorageDir()
2061   if (not os.path.commonprefix([file_storage_dir, base_file_storage_dir]) ==
2062       base_file_storage_dir):
2063     logging.error("file storage directory '%s' is not under base file"
2064                   " storage directory '%s'",
2065                   file_storage_dir, base_file_storage_dir)
2066     return None
2067   return file_storage_dir
2068
2069
2070 def CreateFileStorageDir(file_storage_dir):
2071   """Create file storage directory.
2072
2073   @type file_storage_dir: str
2074   @param file_storage_dir: directory to create
2075
2076   @rtype: tuple
2077   @return: tuple with first element a boolean indicating wheter dir
2078       creation was successful or not
2079
2080   """
2081   file_storage_dir = _TransformFileStorageDir(file_storage_dir)
2082   result = True,
2083   if not file_storage_dir:
2084     result = False,
2085   else:
2086     if os.path.exists(file_storage_dir):
2087       if not os.path.isdir(file_storage_dir):
2088         logging.error("'%s' is not a directory", file_storage_dir)
2089         result = False,
2090     else:
2091       try:
2092         os.makedirs(file_storage_dir, 0750)
2093       except OSError, err:
2094         logging.error("Cannot create file storage directory '%s': %s",
2095                       file_storage_dir, err)
2096         result = False,
2097   return result
2098
2099
2100 def RemoveFileStorageDir(file_storage_dir):
2101   """Remove file storage directory.
2102
2103   Remove it only if it's empty. If not log an error and return.
2104
2105   @type file_storage_dir: str
2106   @param file_storage_dir: the directory we should cleanup
2107   @rtype: tuple (success,)
2108   @return: tuple of one element, C{success}, denoting
2109       whether the operation was successfull
2110
2111   """
2112   file_storage_dir = _TransformFileStorageDir(file_storage_dir)
2113   result = True,
2114   if not file_storage_dir:
2115     result = False,
2116   else:
2117     if os.path.exists(file_storage_dir):
2118       if not os.path.isdir(file_storage_dir):
2119         logging.error("'%s' is not a directory", file_storage_dir)
2120         result = False,
2121       # deletes dir only if empty, otherwise we want to return False
2122       try:
2123         os.rmdir(file_storage_dir)
2124       except OSError, err:
2125         logging.exception("Cannot remove file storage directory '%s'",
2126                           file_storage_dir)
2127         result = False,
2128   return result
2129
2130
2131 def RenameFileStorageDir(old_file_storage_dir, new_file_storage_dir):
2132   """Rename the file storage directory.
2133
2134   @type old_file_storage_dir: str
2135   @param old_file_storage_dir: the current path
2136   @type new_file_storage_dir: str
2137   @param new_file_storage_dir: the name we should rename to
2138   @rtype: tuple (success,)
2139   @return: tuple of one element, C{success}, denoting
2140       whether the operation was successful
2141
2142   """
2143   old_file_storage_dir = _TransformFileStorageDir(old_file_storage_dir)
2144   new_file_storage_dir = _TransformFileStorageDir(new_file_storage_dir)
2145   result = True,
2146   if not old_file_storage_dir or not new_file_storage_dir:
2147     result = False,
2148   else:
2149     if not os.path.exists(new_file_storage_dir):
2150       if os.path.isdir(old_file_storage_dir):
2151         try:
2152           os.rename(old_file_storage_dir, new_file_storage_dir)
2153         except OSError, err:
2154           logging.exception("Cannot rename '%s' to '%s'",
2155                             old_file_storage_dir, new_file_storage_dir)
2156           result =  False,
2157       else:
2158         logging.error("'%s' is not a directory", old_file_storage_dir)
2159         result = False,
2160     else:
2161       if os.path.exists(old_file_storage_dir):
2162         logging.error("Cannot rename '%s' to '%s'. Both locations exist.",
2163                       old_file_storage_dir, new_file_storage_dir)
2164         result = False,
2165   return result
2166
2167
2168 def _IsJobQueueFile(file_name):
2169   """Checks whether the given filename is in the queue directory.
2170
2171   @type file_name: str
2172   @param file_name: the file name we should check
2173   @rtype: boolean
2174   @return: whether the file is under the queue directory
2175
2176   """
2177   queue_dir = os.path.normpath(constants.QUEUE_DIR)
2178   result = (os.path.commonprefix([queue_dir, file_name]) == queue_dir)
2179
2180   if not result:
2181     logging.error("'%s' is not a file in the queue directory",
2182                   file_name)
2183
2184   return result
2185
2186
2187 def JobQueueUpdate(file_name, content):
2188   """Updates a file in the queue directory.
2189
2190   This is just a wrapper over L{utils.WriteFile}, with proper
2191   checking.
2192
2193   @type file_name: str
2194   @param file_name: the job file name
2195   @type content: str
2196   @param content: the new job contents
2197   @rtype: boolean
2198   @return: the success of the operation
2199
2200   """
2201   if not _IsJobQueueFile(file_name):
2202     return False
2203
2204   # Write and replace the file atomically
2205   utils.WriteFile(file_name, data=_Decompress(content))
2206
2207   return True
2208
2209
2210 def JobQueueRename(old, new):
2211   """Renames a job queue file.
2212
2213   This is just a wrapper over os.rename with proper checking.
2214
2215   @type old: str
2216   @param old: the old (actual) file name
2217   @type new: str
2218   @param new: the desired file name
2219   @rtype: boolean
2220   @return: the success of the operation
2221
2222   """
2223   if not (_IsJobQueueFile(old) and _IsJobQueueFile(new)):
2224     return False
2225
2226   utils.RenameFile(old, new, mkdir=True)
2227
2228   return True
2229
2230
2231 def JobQueueSetDrainFlag(drain_flag):
2232   """Set the drain flag for the queue.
2233
2234   This will set or unset the queue drain flag.
2235
2236   @type drain_flag: boolean
2237   @param drain_flag: if True, will set the drain flag, otherwise reset it.
2238   @rtype: boolean
2239   @return: always True
2240   @warning: the function always returns True
2241
2242   """
2243   if drain_flag:
2244     utils.WriteFile(constants.JOB_QUEUE_DRAIN_FILE, data="", close=True)
2245   else:
2246     utils.RemoveFile(constants.JOB_QUEUE_DRAIN_FILE)
2247
2248   return True
2249
2250
2251 def BlockdevClose(instance_name, disks):
2252   """Closes the given block devices.
2253
2254   This means they will be switched to secondary mode (in case of
2255   DRBD).
2256
2257   @param instance_name: if the argument is not empty, the symlinks
2258       of this instance will be removed
2259   @type disks: list of L{objects.Disk}
2260   @param disks: the list of disks to be closed
2261   @rtype: tuple (success, message)
2262   @return: a tuple of success and message, where success
2263       indicates the succes of the operation, and message
2264       which will contain the error details in case we
2265       failed
2266
2267   """
2268   bdevs = []
2269   for cf in disks:
2270     rd = _RecursiveFindBD(cf)
2271     if rd is None:
2272       return (False, "Can't find device %s" % cf)
2273     bdevs.append(rd)
2274
2275   msg = []
2276   for rd in bdevs:
2277     try:
2278       rd.Close()
2279     except errors.BlockDeviceError, err:
2280       msg.append(str(err))
2281   if msg:
2282     return (False, "Can't make devices secondary: %s" % ",".join(msg))
2283   else:
2284     if instance_name:
2285       _RemoveBlockDevLinks(instance_name, disks)
2286     return (True, "All devices secondary")
2287
2288
2289 def ValidateHVParams(hvname, hvparams):
2290   """Validates the given hypervisor parameters.
2291
2292   @type hvname: string
2293   @param hvname: the hypervisor name
2294   @type hvparams: dict
2295   @param hvparams: the hypervisor parameters to be validated
2296   @rtype: tuple (success, message)
2297   @return: a tuple of success and message, where success
2298       indicates the succes of the operation, and message
2299       which will contain the error details in case we
2300       failed
2301
2302   """
2303   try:
2304     hv_type = hypervisor.GetHypervisor(hvname)
2305     hv_type.ValidateParameters(hvparams)
2306     return (True, "Validation passed")
2307   except errors.HypervisorError, err:
2308     return (False, str(err))
2309
2310
2311 def DemoteFromMC():
2312   """Demotes the current node from master candidate role.
2313
2314   """
2315   # try to ensure we're not the master by mistake
2316   master, myself = ssconf.GetMasterAndMyself()
2317   if master == myself:
2318     return (False, "ssconf status shows I'm the master node, will not demote")
2319   pid_file = utils.DaemonPidFileName(constants.MASTERD_PID)
2320   if utils.IsProcessAlive(utils.ReadPidFile(pid_file)):
2321     return (False, "The master daemon is running, will not demote")
2322   try:
2323     utils.CreateBackup(constants.CLUSTER_CONF_FILE)
2324   except EnvironmentError, err:
2325     if err.errno != errno.ENOENT:
2326       return (False, "Error while backing up cluster file: %s" % str(err))
2327   utils.RemoveFile(constants.CLUSTER_CONF_FILE)
2328   return (True, "Done")
2329
2330
2331 def _FindDisks(nodes_ip, disks):
2332   """Sets the physical ID on disks and returns the block devices.
2333
2334   """
2335   # set the correct physical ID
2336   my_name = utils.HostInfo().name
2337   for cf in disks:
2338     cf.SetPhysicalID(my_name, nodes_ip)
2339
2340   bdevs = []
2341
2342   for cf in disks:
2343     rd = _RecursiveFindBD(cf)
2344     if rd is None:
2345       return (False, "Can't find device %s" % cf)
2346     bdevs.append(rd)
2347   return (True, bdevs)
2348
2349
2350 def DrbdDisconnectNet(nodes_ip, disks):
2351   """Disconnects the network on a list of drbd devices.
2352
2353   """
2354   status, bdevs = _FindDisks(nodes_ip, disks)
2355   if not status:
2356     return status, bdevs
2357
2358   # disconnect disks
2359   for rd in bdevs:
2360     try:
2361       rd.DisconnectNet()
2362     except errors.BlockDeviceError, err:
2363       logging.exception("Failed to go into standalone mode")
2364       return (False, "Can't change network configuration: %s" % str(err))
2365   return (True, "All disks are now disconnected")
2366
2367
2368 def DrbdAttachNet(nodes_ip, disks, instance_name, multimaster):
2369   """Attaches the network on a list of drbd devices.
2370
2371   """
2372   status, bdevs = _FindDisks(nodes_ip, disks)
2373   if not status:
2374     return status, bdevs
2375
2376   if multimaster:
2377     for idx, rd in enumerate(bdevs):
2378       try:
2379         _SymlinkBlockDev(instance_name, rd.dev_path, idx)
2380       except EnvironmentError, err:
2381         return (False, "Can't create symlink: %s" % str(err))
2382   # reconnect disks, switch to new master configuration and if
2383   # needed primary mode
2384   for rd in bdevs:
2385     try:
2386       rd.AttachNet(multimaster)
2387     except errors.BlockDeviceError, err:
2388       return (False, "Can't change network configuration: %s" % str(err))
2389   # wait until the disks are connected; we need to retry the re-attach
2390   # if the device becomes standalone, as this might happen if the one
2391   # node disconnects and reconnects in a different mode before the
2392   # other node reconnects; in this case, one or both of the nodes will
2393   # decide it has wrong configuration and switch to standalone
2394   RECONNECT_TIMEOUT = 2 * 60
2395   sleep_time = 0.100 # start with 100 miliseconds
2396   timeout_limit = time.time() + RECONNECT_TIMEOUT
2397   while time.time() < timeout_limit:
2398     all_connected = True
2399     for rd in bdevs:
2400       stats = rd.GetProcStatus()
2401       if not (stats.is_connected or stats.is_in_resync):
2402         all_connected = False
2403       if stats.is_standalone:
2404         # peer had different config info and this node became
2405         # standalone, even though this should not happen with the
2406         # new staged way of changing disk configs
2407         try:
2408           rd.ReAttachNet(multimaster)
2409         except errors.BlockDeviceError, err:
2410           return (False, "Can't change network configuration: %s" % str(err))
2411     if all_connected:
2412       break
2413     time.sleep(sleep_time)
2414     sleep_time = min(5, sleep_time * 1.5)
2415   if not all_connected:
2416     return (False, "Timeout in disk reconnecting")
2417   if multimaster:
2418     # change to primary mode
2419     for rd in bdevs:
2420       try:
2421         rd.Open()
2422       except errors.BlockDeviceError, err:
2423         return (False, "Can't change to primary mode: %s" % str(err))
2424   if multimaster:
2425     msg = "multi-master and primary"
2426   else:
2427     msg = "single-master"
2428   return (True, "Disks are now configured as %s" % msg)
2429
2430
2431 def DrbdWaitSync(nodes_ip, disks):
2432   """Wait until DRBDs have synchronized.
2433
2434   """
2435   status, bdevs = _FindDisks(nodes_ip, disks)
2436   if not status:
2437     return status, bdevs
2438
2439   min_resync = 100
2440   alldone = True
2441   failure = False
2442   for rd in bdevs:
2443     stats = rd.GetProcStatus()
2444     if not (stats.is_connected or stats.is_in_resync):
2445       failure = True
2446       break
2447     alldone = alldone and (not stats.is_in_resync)
2448     if stats.sync_percent is not None:
2449       min_resync = min(min_resync, stats.sync_percent)
2450   return (not failure, (alldone, min_resync))
2451
2452
2453 def PowercycleNode(hypervisor_type):
2454   """Hard-powercycle the node.
2455
2456   Because we need to return first, and schedule the powercycle in the
2457   background, we won't be able to report failures nicely.
2458
2459   """
2460   hyper = hypervisor.GetHypervisor(hypervisor_type)
2461   try:
2462     pid = os.fork()
2463   except OSError, err:
2464     # if we can't fork, we'll pretend that we're in the child process
2465     pid = 0
2466   if pid > 0:
2467     return (True, "Reboot scheduled in 5 seconds")
2468   time.sleep(5)
2469   hyper.PowercycleNode()
2470
2471
2472 class HooksRunner(object):
2473   """Hook runner.
2474
2475   This class is instantiated on the node side (ganeti-noded) and not
2476   on the master side.
2477
2478   """
2479   RE_MASK = re.compile("^[a-zA-Z0-9_-]+$")
2480
2481   def __init__(self, hooks_base_dir=None):
2482     """Constructor for hooks runner.
2483
2484     @type hooks_base_dir: str or None
2485     @param hooks_base_dir: if not None, this overrides the
2486         L{constants.HOOKS_BASE_DIR} (useful for unittests)
2487
2488     """
2489     if hooks_base_dir is None:
2490       hooks_base_dir = constants.HOOKS_BASE_DIR
2491     self._BASE_DIR = hooks_base_dir
2492
2493   @staticmethod
2494   def ExecHook(script, env):
2495     """Exec one hook script.
2496
2497     @type script: str
2498     @param script: the full path to the script
2499     @type env: dict
2500     @param env: the environment with which to exec the script
2501     @rtype: tuple (success, message)
2502     @return: a tuple of success and message, where success
2503         indicates the succes of the operation, and message
2504         which will contain the error details in case we
2505         failed
2506
2507     """
2508     # exec the process using subprocess and log the output
2509     fdstdin = None
2510     try:
2511       fdstdin = open("/dev/null", "r")
2512       child = subprocess.Popen([script], stdin=fdstdin, stdout=subprocess.PIPE,
2513                                stderr=subprocess.STDOUT, close_fds=True,
2514                                shell=False, cwd="/", env=env)
2515       output = ""
2516       try:
2517         output = child.stdout.read(4096)
2518         child.stdout.close()
2519       except EnvironmentError, err:
2520         output += "Hook script error: %s" % str(err)
2521
2522       while True:
2523         try:
2524           result = child.wait()
2525           break
2526         except EnvironmentError, err:
2527           if err.errno == errno.EINTR:
2528             continue
2529           raise
2530     finally:
2531       # try not to leak fds
2532       for fd in (fdstdin, ):
2533         if fd is not None:
2534           try:
2535             fd.close()
2536           except EnvironmentError, err:
2537             # just log the error
2538             #logging.exception("Error while closing fd %s", fd)
2539             pass
2540
2541     return result == 0, utils.SafeEncode(output.strip())
2542
2543   def RunHooks(self, hpath, phase, env):
2544     """Run the scripts in the hooks directory.
2545
2546     @type hpath: str
2547     @param hpath: the path to the hooks directory which
2548         holds the scripts
2549     @type phase: str
2550     @param phase: either L{constants.HOOKS_PHASE_PRE} or
2551         L{constants.HOOKS_PHASE_POST}
2552     @type env: dict
2553     @param env: dictionary with the environment for the hook
2554     @rtype: list
2555     @return: list of 3-element tuples:
2556       - script path
2557       - script result, either L{constants.HKR_SUCCESS} or
2558         L{constants.HKR_FAIL}
2559       - output of the script
2560
2561     @raise errors.ProgrammerError: for invalid input
2562         parameters
2563
2564     """
2565     if phase == constants.HOOKS_PHASE_PRE:
2566       suffix = "pre"
2567     elif phase == constants.HOOKS_PHASE_POST:
2568       suffix = "post"
2569     else:
2570       raise errors.ProgrammerError("Unknown hooks phase: '%s'" % phase)
2571     rr = []
2572
2573     subdir = "%s-%s.d" % (hpath, suffix)
2574     dir_name = "%s/%s" % (self._BASE_DIR, subdir)
2575     try:
2576       dir_contents = utils.ListVisibleFiles(dir_name)
2577     except OSError, err:
2578       # FIXME: must log output in case of failures
2579       return rr
2580
2581     # we use the standard python sort order,
2582     # so 00name is the recommended naming scheme
2583     dir_contents.sort()
2584     for relname in dir_contents:
2585       fname = os.path.join(dir_name, relname)
2586       if not (os.path.isfile(fname) and os.access(fname, os.X_OK) and
2587           self.RE_MASK.match(relname) is not None):
2588         rrval = constants.HKR_SKIP
2589         output = ""
2590       else:
2591         result, output = self.ExecHook(fname, env)
2592         if not result:
2593           rrval = constants.HKR_FAIL
2594         else:
2595           rrval = constants.HKR_SUCCESS
2596       rr.append(("%s/%s" % (subdir, relname), rrval, output))
2597
2598     return rr
2599
2600
2601 class IAllocatorRunner(object):
2602   """IAllocator runner.
2603
2604   This class is instantiated on the node side (ganeti-noded) and not on
2605   the master side.
2606
2607   """
2608   def Run(self, name, idata):
2609     """Run an iallocator script.
2610
2611     @type name: str
2612     @param name: the iallocator script name
2613     @type idata: str
2614     @param idata: the allocator input data
2615
2616     @rtype: tuple
2617     @return: four element tuple of:
2618        - run status (one of the IARUN_ constants)
2619        - stdout
2620        - stderr
2621        - fail reason (as from L{utils.RunResult})
2622
2623     """
2624     alloc_script = utils.FindFile(name, constants.IALLOCATOR_SEARCH_PATH,
2625                                   os.path.isfile)
2626     if alloc_script is None:
2627       return (constants.IARUN_NOTFOUND, None, None, None)
2628
2629     fd, fin_name = tempfile.mkstemp(prefix="ganeti-iallocator.")
2630     try:
2631       os.write(fd, idata)
2632       os.close(fd)
2633       result = utils.RunCmd([alloc_script, fin_name])
2634       if result.failed:
2635         return (constants.IARUN_FAILURE, result.stdout, result.stderr,
2636                 result.fail_reason)
2637     finally:
2638       os.unlink(fin_name)
2639
2640     return (constants.IARUN_SUCCESS, result.stdout, result.stderr, None)
2641
2642
2643 class DevCacheManager(object):
2644   """Simple class for managing a cache of block device information.
2645
2646   """
2647   _DEV_PREFIX = "/dev/"
2648   _ROOT_DIR = constants.BDEV_CACHE_DIR
2649
2650   @classmethod
2651   def _ConvertPath(cls, dev_path):
2652     """Converts a /dev/name path to the cache file name.
2653
2654     This replaces slashes with underscores and strips the /dev
2655     prefix. It then returns the full path to the cache file.
2656
2657     @type dev_path: str
2658     @param dev_path: the C{/dev/} path name
2659     @rtype: str
2660     @return: the converted path name
2661
2662     """
2663     if dev_path.startswith(cls._DEV_PREFIX):
2664       dev_path = dev_path[len(cls._DEV_PREFIX):]
2665     dev_path = dev_path.replace("/", "_")
2666     fpath = "%s/bdev_%s" % (cls._ROOT_DIR, dev_path)
2667     return fpath
2668
2669   @classmethod
2670   def UpdateCache(cls, dev_path, owner, on_primary, iv_name):
2671     """Updates the cache information for a given device.
2672
2673     @type dev_path: str
2674     @param dev_path: the pathname of the device
2675     @type owner: str
2676     @param owner: the owner (instance name) of the device
2677     @type on_primary: bool
2678     @param on_primary: whether this is the primary
2679         node nor not
2680     @type iv_name: str
2681     @param iv_name: the instance-visible name of the
2682         device, as in objects.Disk.iv_name
2683
2684     @rtype: None
2685
2686     """
2687     if dev_path is None:
2688       logging.error("DevCacheManager.UpdateCache got a None dev_path")
2689       return
2690     fpath = cls._ConvertPath(dev_path)
2691     if on_primary:
2692       state = "primary"
2693     else:
2694       state = "secondary"
2695     if iv_name is None:
2696       iv_name = "not_visible"
2697     fdata = "%s %s %s\n" % (str(owner), state, iv_name)
2698     try:
2699       utils.WriteFile(fpath, data=fdata)
2700     except EnvironmentError, err:
2701       logging.exception("Can't update bdev cache for %s", dev_path)
2702
2703   @classmethod
2704   def RemoveCache(cls, dev_path):
2705     """Remove data for a dev_path.
2706
2707     This is just a wrapper over L{utils.RemoveFile} with a converted
2708     path name and logging.
2709
2710     @type dev_path: str
2711     @param dev_path: the pathname of the device
2712
2713     @rtype: None
2714
2715     """
2716     if dev_path is None:
2717       logging.error("DevCacheManager.RemoveCache got a None dev_path")
2718       return
2719     fpath = cls._ConvertPath(dev_path)
2720     try:
2721       utils.RemoveFile(fpath)
2722     except EnvironmentError, err:
2723       logging.exception("Can't update bdev cache for %s", dev_path)