# # The Qubes OS Project, https://www.qubes-os.org/ # # Copyright (C) 2010-2015 Joanna Rutkowska # Copyright (C) 2013-2015 Marek Marczykowski-Górecki # # Copyright (C) 2014-2015 Wojtek Porczyk # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License along # with this program; if not, write to the Free Software Foundation, Inc., # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. # from __future__ import absolute_import import asyncio import copy import base64 import datetime import os import os.path import shutil import string import subprocess import uuid import warnings import grp import errno import lxml import libvirt # pylint: disable=import-error import qubes import qubes.config import qubes.exc import qubes.storage import qubes.storage.domain import qubes.storage.file import qubes.utils import qubes.vm import qubes.vm.mix.net qmemman_present = False try: import qubes.qmemman.client # pylint: disable=wrong-import-position qmemman_present = True except ImportError: pass MEM_OVERHEAD_BASE = (3 + 1) * 1024 * 1024 MEM_OVERHEAD_PER_VCPU = 3 * 1024 * 1024 / 2 def _setter_qid(self, prop, value): ''' Helper for setting the domain qid ''' # pylint: disable=unused-argument value = int(value) if not 0 <= value <= qubes.config.max_qid: raise ValueError( '{} value must be between 0 and qubes.config.max_qid'.format( prop.__name__)) return value def _setter_name(self, prop, value): ''' Helper for setting the domain name ''' qubes.vm.validate_name(self, prop, value) if self.is_running(): raise qubes.exc.QubesVMNotHaltedError( self, 'Cannot change name of running VM') try: if self.installed_by_rpm: raise qubes.exc.QubesException('Cannot rename VM installed by RPM ' '-- first clone VM and then use yum to remove package.') except AttributeError: pass if value in self.app.domains: raise qubes.exc.QubesPropertyValueError(self, prop, value, 'VM named {} alread exists'.format(value)) return value def _setter_kernel(self, prop, value): ''' Helper for setting the domain kernel and running sanity checks on it. ''' # pylint: disable=unused-argument if value is None: return value value = str(value) if '/' in value: raise qubes.exc.QubesPropertyValueError(self, prop, value, 'Kernel name cannot contain \'/\'') dirname = os.path.join( qubes.config.system_path['qubes_base_dir'], qubes.config.system_path['qubes_kernels_base_dir'], value) if not os.path.exists(dirname): raise qubes.exc.QubesPropertyValueError(self, prop, value, 'Kernel {!r} not installed'.format(value)) for filename in ('vmlinuz', 'initramfs'): if not os.path.exists(os.path.join(dirname, filename)): raise qubes.exc.QubesPropertyValueError(self, prop, value, 'Kernel {!r} not properly installed: missing {!r} file'.format( value, filename)) return value def _setter_label(self, prop, value): ''' Helper for setting the domain label ''' # pylint: disable=unused-argument if isinstance(value, qubes.Label): return value if isinstance(value, str) and value.startswith('label-'): return self.app.labels[int(value.split('-', 1)[1])] return self.app.get_label(value) def _setter_positive_int(self, prop, value): ''' Helper for setting a positive int. Checks that the int is >= 0 ''' # pylint: disable=unused-argument value = int(value) if value <= 0: raise ValueError('Value must be positive') return value def _setter_default_user(self, prop, value): ''' Helper for setting default user ''' value = str(value) # specifically forbid: ':', ' ', ''', '"' allowed_chars = string.ascii_letters + string.digits + '_-+,.' if not all(c in allowed_chars for c in value): raise qubes.exc.QubesPropertyValueError(self, prop, value, 'Username can contain only those characters: ' + allowed_chars) return value class QubesVM(qubes.vm.mix.net.NetVMMixin, qubes.vm.BaseVM): '''Base functionality of Qubes VM shared between all VMs. The following events are raised on this class or its subclasses: .. event:: domain-init (subject, event) Fired at the end of class' constructor. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-init'``) .. event:: domain-load (subject, event) Fired after the qube was loaded from :file:`qubes.xml` :param subject: Event emitter (the qube object) :param event: Event name (``'domain-loaded'``) .. event:: domain-pre-start \ (subject, event, preparing_dvm, start_guid, mem_required) Fired at the beginning of :py:meth:`start` method. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-pre-start'``) *other arguments are as in :py:meth:`start`* .. event:: domain-spawn (subject, event, preparing_dvm, start_guid) Fired after creating libvirt domain. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-spawn'``) *other arguments are as in :py:meth:`start`* .. event:: domain-start (subject, event, preparing_dvm, start_guid) Fired at the end of :py:meth:`start` method. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-start'``) *other arguments are as in :py:meth:`start`* .. event:: domain-shutdown (subject, event) Fired when domain has been shut down. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-shutdown'``) .. event:: domain-pre-shutdown (subject, event, force) Fired at the beginning of :py:meth:`shutdown` method. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-pre-shutdown'``) :param force: If the shutdown is to be forceful .. event:: domain-cmd-pre-run (subject, event, start_guid) Fired at the beginning of :py:meth:`run_service` method. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-cmd-pre-run'``) :param start_guid: If the gui daemon can be started .. event:: domain-create-on-disk (subject, event) Fired at the end of :py:meth:`create_on_disk` method. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-create-on-disk'``) .. event:: domain-remove-from-disk (subject, event) Fired at the beginning of :py:meth:`remove_from_disk` method, before the qube directory is removed. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-remove-from-disk'``) .. event:: domain-clone-files (subject, event, src) Fired at the end of :py:meth:`clone_disk_files` method. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-clone-files'``) :param src: source qube .. event:: domain-verify-files (subject, event) Fired at the end of :py:meth:`clone_disk_files` method. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-verify-files'``) If you think some files are missing or damaged, raise an exception. .. event:: domain-is-fully-usable (subject, event) Fired at the end of :py:meth:`clone_disk_files` method. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-is-fully-usable'``) You may ``yield False`` from the handler if you think the qube is not fully usable. This will cause the domain to be in "transient" state in the domain lifecycle. .. event:: domain-qtb-create (subject, event) Fired at the end of :py:meth:`create_qdb_entries` method. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-qtb-create'``) This event is a good place to add your custom entries to the qdb. .. event:: backup-get-files (subject, event) Collects additional file to be included in a backup. :param subject: Event emitter (the qube object) :param event: Event name (``'backup-get-files'``) Handlers should yield paths of the files. .. event:: domain-restore (subject, event) Domain was just restored from backup, although the storage was not yet verified and the app object was not yet saved. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-restore'``) .. event:: domain-feature-set (subject, event, key, value) A feature was changed. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-feature-set'``) :param key: feature name :param value: new value .. event:: domain-feature-delete (subject, event, key) A feature was removed. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-feature-delete'``) :param key: feature name .. event:: domain-tag-add (subject, event, tag) A tag was added. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-tag-add'``) :param tag: tag name .. event:: domain-tag-delete (subject, event, tag) A feature was removed. :param subject: Event emitter (the qube object) :param event: Event name (``'domain-tag-delete'``) :param tag: tag name .. event:: feature-request (subject, event, *, untrusted_features) The domain is performing a feature request. :param subject: Event emitter (the qube object) :param event: Event name (``'feature-request'``) :param untrusted_features: :py:class:`dict` containing the feature \ request The content of the `untrusted_features` variable is, as the name implies, **UNTRUSTED**. The remind this to programmer, the variable name has to be exactly as provided. It is up to the extensions to decide, what to do with request, ranging from plainly ignoring the request to verbatim copy into :py:attr:`features` with only minimal sanitisation. .. event:: monitor-layout-change (subject, event, monitor_layout) Desktop layout was changed, probably because a display was plugged in or out. :param subject: Event emitter (the qube object) :param event: Event name (``'monitor-layout-change'``) :param monitor_layout: The new layout .. event:: firewall-changed (subject, event) Firewall was changed. :param subject: Event emitter (the qube object) :param event: Event name (``'firewall-changed'``) .. event:: net-domain-connect (subject, event, vm) Fired after connecting a domiain to this vm. :param subject: Event emitter (the qube object) :param event: Event name (``'net-domain-connect'``) :param vm: The domain that was just connected. On the `vm` object there was probably ``property-set:netvm`` fired earlier. ''' # # per-class properties # #: directory in which domains of this class will reside dir_path_prefix = qubes.config.system_path['qubes_appvms_dir'] # # properties loaded from XML # label = qubes.property('label', setter=_setter_label, saver=(lambda self, prop, value: 'label-{}'.format(value.index)), doc='''Colourful label assigned to VM. This is where the colour of the padlock is set.''') # provides_network = qubes.property('provides_network', # type=bool, setter=qubes.property.bool, # doc='`True` if it is NetVM or ProxyVM, false otherwise.') qid = qubes.property('qid', type=int, write_once=True, setter=_setter_qid, clone=False, doc='''Internal, persistent identificator of particular domain. Note this is different from Xen domid.''') name = qubes.property('name', type=str, clone=False, doc='User-specified name of the domain.') uuid = qubes.property('uuid', type=uuid.UUID, write_once=True, clone=False, doc='UUID from libvirt.') hvm = qubes.property('hvm', type=bool, setter=qubes.property.bool, default=False, doc='''Use full virtualisation (HVM) for this qube, instead of paravirtualisation (PV)''') installed_by_rpm = qubes.property('installed_by_rpm', type=bool, setter=qubes.property.bool, default=False, doc='''If this domain's image was installed from package tracked by package manager.''') memory = qubes.property('memory', type=int, setter=_setter_positive_int, default=(lambda self: qubes.config.defaults['hvm_memory' if self.hvm else 'memory']), doc='Memory currently available for this VM.') maxmem = qubes.property('maxmem', type=int, setter=_setter_positive_int, default=(lambda self: int(min(self.app.host.memory_total / 1024 / 2, 4000))), doc='''Maximum amount of memory available for this VM (for the purpose of the memory balancer).''') stubdom_mem = qubes.property('stubdom_mem', type=int, setter=_setter_positive_int, default=None, doc='Memory ammount allocated for the stubdom') internal = qubes.property('internal', default=False, type=bool, setter=qubes.property.bool, doc='''Internal VM (not shown in qubes-manager, don't create appmenus entries.''') vcpus = qubes.property('vcpus', type=int, setter=_setter_positive_int, default=(lambda self: self.app.host.no_cpus), doc='FIXME') # CORE2: swallowed uses_default_kernel kernel = qubes.property('kernel', type=str, setter=_setter_kernel, default=(lambda self: self.app.default_kernel), doc='Kernel used by this domain.') # CORE2: swallowed uses_default_kernelopts # pylint: disable=no-member kernelopts = qubes.property('kernelopts', type=str, load_stage=4, default=(lambda self: qubes.config.defaults['kernelopts_pcidevs'] # pylint: disable=no-member if list(self.devices['pci'].persistent()) else self.template.kernelopts if hasattr(self, 'template') else qubes.config.defaults['kernelopts']), doc='Kernel command line passed to domain.') debug = qubes.property('debug', type=bool, default=False, setter=qubes.property.bool, doc='Turns on debugging features.') # XXX what this exactly does? # XXX shouldn't this go to standalone VM and TemplateVM, and leave here # only plain property? default_user = qubes.property('default_user', type=str, # pylint: disable=no-member default=(lambda self: self.template.default_user if hasattr(self, 'template') else 'user'), setter=_setter_default_user, doc='FIXME') # pylint: enable=no-member # @property # def default_user(self): # if self.template is not None: # return self.template.default_user # else: # return self._default_user qrexec_timeout = qubes.property('qrexec_timeout', type=int, default=60, setter=_setter_positive_int, doc='''Time in seconds after which qrexec connection attempt is deemed failed. Operating system inside VM should be able to boot in this time.''') autostart = qubes.property('autostart', default=False, type=bool, setter=qubes.property.bool, doc='''Setting this to `True` means that VM should be autostarted on dom0 boot.''') include_in_backups = qubes.property('include_in_backups', default=(lambda self: not self.internal), type=bool, setter=qubes.property.bool, doc='If this domain is to be included in default backup.') # format got changed from %s to str(datetime.datetime) backup_timestamp = qubes.property('backup_timestamp', default=None, setter=(lambda self, prop, value: value if isinstance(value, datetime.datetime) else datetime.datetime.fromtimestamp(int(value))), saver=(lambda self, prop, value: value.strftime('%s')), doc='FIXME') default_dispvm = qubes.VMProperty('default_dispvm', load_stage=4, allow_none=True, default=(lambda self: self.app.default_dispvm), doc='Default VM to be used as Disposable VM for service calls.') updateable = qubes.property('updateable', default=(lambda self: not hasattr(self, 'template')), type=bool, setter=qubes.property.forbidden, doc='True if this machine may be updated on its own.') # # static, class-wide properties # # # properties not loaded from XML, calculated at run-time # def __str__(self): return self.name # VMM-related @qubes.stateless_property def xid(self): '''Xen ID. Or not Xen, but ID. ''' if self.libvirt_domain is None: return -1 try: return self.libvirt_domain.ID() except libvirt.libvirtError as e: if e.get_error_code() == libvirt.VIR_ERR_NO_DOMAIN: return -1 else: self.log.exception('libvirt error code: {!r}'.format( e.get_error_code())) raise @qubes.stateless_property def stubdom_xid(self): if not self.is_running(): return -1 if self.app.vmm.xs is None: return -1 stubdom_xid_str = self.app.vmm.xs.read('', '/local/domain/{}/image/device-model-domid'.format(self.xid)) if stubdom_xid_str is None or not stubdom_xid_str.isdigit(): return -1 return int(stubdom_xid_str) @property def attached_volumes(self): result = [] xml_desc = self.libvirt_domain.XMLDesc() xml = lxml.etree.fromstring(xml_desc) for disk in xml.xpath("//domain/devices/disk"): if disk.find('backenddomain') is not None: pool_name = 'p_%s' % disk.find('backenddomain').get('name') pool = self.app.pools[pool_name] vid = disk.find('source').get('dev').split('/dev/')[1] for volume in pool.volumes: if volume.vid == vid: result += [volume] break return result + list(self.volumes.values()) @property def libvirt_domain(self): '''Libvirt domain object from libvirt. May be :py:obj:`None`, if libvirt knows nothing about this domain. ''' if self._libvirt_domain is not None: return self._libvirt_domain # XXX _update_libvirt_domain? try: self._libvirt_domain = self.app.vmm.libvirt_conn.lookupByUUID( self.uuid.bytes) except libvirt.libvirtError as e: if e.get_error_code() == libvirt.VIR_ERR_NO_DOMAIN: self._update_libvirt_domain() else: raise return self._libvirt_domain @property def block_devices(self): ''' Return all :py:class:`qubes.storage.BlockDevice`s for current domain for serialization in the libvirt XML template as . ''' return [v.block_device() for v in self.volumes.values()] @property def qdb(self): '''QubesDB handle for this domain.''' if self._qdb_connection is None: if self.is_running(): import qubesdb # pylint: disable=import-error self._qdb_connection = qubesdb.QubesDB(self.name) return self._qdb_connection @property def dir_path(self): '''Root directory for files related to this domain''' return os.path.join( qubes.config.system_path['qubes_base_dir'], self.dir_path_prefix, self.name) @property def icon_path(self): return os.path.join(self.dir_path, 'icon.png') @property def conf_file(self): return os.path.join(self.dir_path, 'libvirt.xml') # network-related # # constructor # def __init__(self, app, xml, volume_config=None, **kwargs): super(QubesVM, self).__init__(app, xml, **kwargs) self.volumes = {} self.storage = None if volume_config is None: volume_config = {} if hasattr(self, 'volume_config'): if xml is not None: for node in xml.xpath('volume-config/volume'): name = node.get('name') assert name for key, value in node.items(): # pylint: disable=no-member if value == 'True': value = True try: self.volume_config[name][key] = value except KeyError: self.volume_config[name] = {key: value} for name, conf in volume_config.items(): for key, value in conf.items(): # pylint: disable=no-member try: self.volume_config[name][key] = value except KeyError: self.volume_config[name] = {key: value} elif volume_config: raise TypeError( 'volume_config specified, but {} did not expect that.'.format( self.__class__.__name__)) # Init private attrs self._libvirt_domain = None self._qdb_connection = None #: this :py:class:`asyncio.Event` will fire when session is obtained self.have_session = asyncio.Event() if xml is None: # we are creating new VM and attributes came through kwargs assert hasattr(self, 'qid') assert hasattr(self, 'name') # Linux specific cap: max memory can't scale beyond 10.79*init_mem # see https://groups.google.com/forum/#!topic/qubes-devel/VRqkFj1IOtA if self.maxmem > self.memory * 10: self.maxmem = self.memory * 10 # By default allow use all VCPUs # if not hasattr(self, 'vcpus') and not self.app.vmm.offline_mode: # self.vcpus = self.app.host.no_cpus if xml is None: # new qube, disable updates check if requested for new qubes # SEE: 1637 when features are done, migrate to plugin if not self.app.check_updates_vm: self.features['check-updates'] = None # will be initialized after loading all the properties # fire hooks if xml is None: self.events_enabled = True self.fire_event('domain-init') def __hash__(self): return self.qid def __lt__(self, other): return self.name < other.name def __xml__(self): element = super(QubesVM, self).__xml__() if hasattr(self, 'volumes'): volume_config_node = lxml.etree.Element('volume-config') for volume in self.volumes.values(): volume_config_node.append(volume.__xml__()) element.append(volume_config_node) return element # # event handlers # @qubes.events.handler('domain-init', 'domain-load') def on_domain_init_loaded(self, event): # pylint: disable=unused-argument if not hasattr(self, 'uuid'): self.uuid = uuid.uuid4() # Initialize VM image storage class self.storage = qubes.storage.Storage(self) vm_pool = qubes.storage.domain.DomainPool(self) self.app.pools[vm_pool.name] = vm_pool @qubes.events.handler('property-set:label') def on_property_set_label(self, event, name, newvalue, oldvalue=None): # pylint: disable=unused-argument if self.icon_path: try: os.remove(self.icon_path) except OSError: pass if hasattr(os, "symlink"): os.symlink(newvalue.icon_path, self.icon_path) subprocess.call(['sudo', 'xdg-icon-resource', 'forceupdate']) else: shutil.copy(newvalue.icon_path, self.icon_path) @qubes.events.handler('property-pre-set:name') def on_property_pre_set_name(self, event, name, newvalue, oldvalue=None): # pylint: disable=unused-argument try: self.app.domains[newvalue] except KeyError: pass else: raise qubes.exc.QubesValueError( 'VM named {!r} already exists'.format(newvalue)) # TODO not self.is_stopped() would be more appropriate if self.is_running(): raise qubes.exc.QubesVMNotHaltedError( 'Cannot change name of running domain {!r}'.format(oldvalue)) if self.autostart: subprocess.check_call(['sudo', 'systemctl', '-q', 'disable', 'qubes-vm@{}.service'.format(oldvalue)]) @qubes.events.handler('property-set:name') def on_property_set_name(self, event, name, newvalue, oldvalue=None): # pylint: disable=unused-argument self.init_log() old_dir_path = os.path.join(os.path.dirname(self.dir_path), oldvalue) new_dir_path = os.path.join(os.path.dirname(self.dir_path), newvalue) os.rename(old_dir_path, new_dir_path) self.storage.rename(oldvalue, newvalue) if self._libvirt_domain is not None: self.libvirt_domain.undefine() self._libvirt_domain = None if self._qdb_connection is not None: self._qdb_connection.close() self._qdb_connection = None self._update_libvirt_domain() if self.autostart: self.autostart = self.autostart @qubes.events.handler('property-pre-set:autostart') def on_property_pre_set_autostart(self, event, prop, newvalue, oldvalue=None): # pylint: disable=unused-argument # workaround https://bugzilla.redhat.com/show_bug.cgi?id=1181922 if newvalue: retcode = subprocess.call( ["sudo", "ln", "-sf", "/usr/lib/systemd/system/qubes-vm@.service", "/etc/systemd/system/multi-user.target.wants/qubes-vm@" "{}.service".format(self.name)]) else: retcode = subprocess.call( ['sudo', 'systemctl', 'disable', 'qubes-vm@{}.service'.format(self.name)]) if retcode: raise qubes.exc.QubesException( 'Failed to set autostart for VM in systemd') @qubes.events.handler('property-pre-del:autostart') def on_property_pre_del_autostart(self, event, prop, oldvalue=None): # pylint: disable=unused-argument if oldvalue: retcode = subprocess.call( ['sudo', 'systemctl', 'disable', 'qubes-vm@{}.service'.format(self.name)]) if retcode: raise qubes.exc.QubesException( 'Failed to reset autostart for VM in systemd') # # methods for changing domain state # @asyncio.coroutine def start(self, preparing_dvm=False, start_guid=True, notify_function=None, mem_required=None): '''Start domain :param bool preparing_dvm: FIXME :param bool start_guid: FIXME :param collections.Callable notify_function: FIXME :param int mem_required: FIXME ''' # Intentionally not used is_running(): eliminate also "Paused", # "Crashed", "Halting" if self.get_power_state() != 'Halted': raise qubes.exc.QubesVMNotHaltedError(self) self.log.info('Starting {}'.format(self.name)) self.fire_event_pre('domain-pre-start', preparing_dvm=preparing_dvm, start_guid=start_guid, mem_required=mem_required) yield from self.storage.verify() if self.netvm is not None: # pylint: disable = no-member if self.netvm.qid != 0: if not self.netvm.is_running(): yield from self.netvm.start(start_guid=start_guid, notify_function=notify_function) # TODO: lock qmemman_client = yield from asyncio.get_event_loop().run_in_executor( None, self.request_memory, mem_required) yield from self.storage.start() self._update_libvirt_domain() try: self.libvirt_domain.createWithFlags(libvirt.VIR_DOMAIN_START_PAUSED) except: if qmemman_client: qmemman_client.close() raise try: self.fire_event('domain-spawn', preparing_dvm=preparing_dvm, start_guid=start_guid) self.log.info('Setting Qubes DB info for the VM') yield from self.start_qubesdb() self.create_qdb_entries() if preparing_dvm: self.qdb.write('/dvm', '1') self.log.warning('Activating the {} VM'.format(self.name)) self.libvirt_domain.resume() # close() is not really needed, because the descriptor is # close-on-exec anyway, the reason to postpone close() is that # possibly xl is not done constructing the domain after its main # process exits so we close() when we know the domain is up the # successful unpause is some indicator of it if qmemman_client: qmemman_client.close() # if self._start_guid_first and start_guid and not preparing_dvm \ # and os.path.exists('/var/run/shm.id'): # self.start_guid() if not preparing_dvm: yield from self.start_qrexec_daemon() self.fire_event('domain-start', preparing_dvm=preparing_dvm, start_guid=start_guid) except: # pylint: disable=bare-except if self.is_running() or self.is_paused(): # This avoids losing the exception if an exception is raised in # self.force_shutdown(), because the vm is not running or paused yield from self.kill() # pylint: disable=not-an-iterable raise asyncio.ensure_future(self._wait_for_session()) return self @qubes.events.handler('domain-shutdown') def on_domain_shutdown(self, _event, **_kwargs): '''Cleanup after domain shutdown''' asyncio.ensure_future(self.storage.stop()) @asyncio.coroutine def shutdown(self, force=False, wait=False): '''Shutdown domain. :raises qubes.exc.QubesVMNotStartedError: \ when domain is already shut down. ''' if self.is_halted(): raise qubes.exc.QubesVMNotStartedError(self) self.fire_event_pre('domain-pre-shutdown', force=force) self.libvirt_domain.shutdown() while wait and not self.is_halted(): yield from asyncio.sleep(0.25) return self @asyncio.coroutine def kill(self): '''Forcefuly shutdown (destroy) domain. :raises qubes.exc.QubesVMNotStartedError: \ when domain is already shut down. ''' if not self.is_running() and not self.is_paused(): raise qubes.exc.QubesVMNotStartedError(self) self.libvirt_domain.destroy() return self def force_shutdown(self, *args, **kwargs): '''Deprecated alias for :py:meth:`kill`''' warnings.warn( 'Call to deprecated function force_shutdown(), use kill() instead', DeprecationWarning, stacklevel=2) return self.kill(*args, **kwargs) @asyncio.coroutine def suspend(self): '''Suspend (pause) domain. :raises qubes.exc.QubesVMNotRunnignError: \ when domain is already shut down. :raises qubes.exc.QubesNotImplemetedError: \ when domain has PCI devices attached. ''' if not self.is_running() and not self.is_paused(): raise qubes.exc.QubesVMNotRunningError(self) if list(self.devices['pci'].attached()): raise qubes.exc.QubesNotImplementedError( 'Cannot suspend domain {!r} which has PCI devices attached' .format(self.name)) else: self.libvirt_domain.suspend() return self @asyncio.coroutine def pause(self): '''Pause (suspend) domain. This currently delegates to \ :py:meth:`suspend`.''' if not self.is_running(): raise qubes.exc.QubesVMNotRunningError(self) self.suspend() return self @asyncio.coroutine def resume(self): '''Resume suspended domain. :raises qubes.exc.QubesVMNotSuspendedError: when machine is not paused :raises qubes.exc.QubesVMError: when machine is suspended ''' if self.get_power_state() == "Suspended": raise qubes.exc.QubesVMError(self, 'Cannot resume suspended domain {!r}'.format(self.name)) else: self.unpause() return self @asyncio.coroutine def unpause(self): '''Resume (unpause) a domain''' if not self.is_paused(): raise qubes.exc.QubesVMNotPausedError(self) self.libvirt_domain.resume() return self @asyncio.coroutine def run_service(self, service, source=None, user=None, filter_esc=False, autostart=False, gui=False, **kwargs): '''Run service on this VM :param str service: service name :param qubes.vm.qubesvm.QubesVM source: source domain as presented to this VM :param str user: username to run service as :param bool filter_esc: filter escape sequences to protect terminal \ emulator :param bool autostart: if :py:obj:`True`, machine will be started if \ it is not running :param bool gui: when autostarting, also start gui daemon :rtype: asyncio.subprocess.Process .. note:: User ``root`` is redefined to ``SYSTEM`` in the Windows agent code ''' # UNSUPPORTED from previous incarnation: # localcmd, wait, passio*, notify_function, `-e` switch # # - passio* and friends depend on params to command (like in stdlib) # - the filter_esc is orthogonal to passio* # - input: see run_service_for_stdio # - wait has no purpose since this is asynchronous # - notify_function is gone source = 'dom0' if source is None else self.app.domains[source].name if user is None: user = self.default_user if self.is_paused(): # XXX what about autostart? raise qubes.exc.QubesVMNotRunningError( self, 'Domain {!r} is paused'.format(self.name)) elif not self.is_running(): if not autostart: raise qubes.exc.QubesVMNotRunningError(self) yield from self.start(start_guid=gui) if not self.is_qrexec_running(): raise qubes.exc.QubesVMError( self, 'Domain {!r}: qrexec not connected'.format(self.name)) if gui and not self.have_session.is_set(): raise qubes.exc.QubesVMError(self, 'don\'t have session yet') self.fire_event_pre('domain-cmd-pre-run', start_guid=gui) return (yield from asyncio.create_subprocess_exec( qubes.config.system_path['qrexec_client_path'], '-d', str(self.name), *(('-t', '-T') if filter_esc else ()), '{}:QUBESRPC {} {}'.format(user, service, source), **kwargs)) @asyncio.coroutine def run_service_for_stdio(self, *args, input=None, **kwargs): '''Run a service, pass an optional input and return (stdout, stderr). Raises an exception if return code != 0. *args* and *kwargs* are passed verbatim to :py:meth:`run_service`. .. warning:: There are some combinations if stdio-related *kwargs*, which are not filtered for problems originating between the keyboard and the chair. ''' # pylint: disable=redefined-builtin kwargs.setdefault('stdin', subprocess.PIPE) kwargs.setdefault('stdout', subprocess.PIPE) kwargs.setdefault('stderr', subprocess.PIPE) p = yield from self.run_service(*args, **kwargs) # this one is actually a tuple, but there is no need to unpack it stdouterr = yield from p.communicate(input=input) if p.returncode: raise qubes.exc.QubesVMError(self, 'service {!r} failed with retcode {!r}; ' 'stdout={!r} stderr={!r}'.format( args, p.returncode, *stdouterr)) return stdouterr @staticmethod def _prepare_input_for_vmshell(command, input): '''Prepare shell input for the given command and optional (real) input ''' # pylint: disable=redefined-builtin if input is None: input = b'' return b''.join((command.rstrip('\n').encode('utf-8'), b'\n', input)) def run(self, command, input=None, **kwargs): '''Run a shell command inside the domain using qubes.VMShell qrexec. This method is a coroutine. *kwargs* are passed verbatim to :py:meth:`run_service`. ''' # pylint: disable=redefined-builtin return self.run_service('qubes.VMShell', input=self._prepare_input_for_vmshell(command, input), **kwargs) def run_for_stdio(self, command, input=None, **kwargs): '''Run a shell command inside the domain using qubes.VMShell qrexec. This method is a coroutine. *kwargs* are passed verbatim to :py:meth:`run_service_for_stdio`. See disclaimer there. ''' # pylint: disable=redefined-builtin return self.run_service_for_stdio('qubes.VMShell', input=self._prepare_input_for_vmshell(command, input), **kwargs) def request_memory(self, mem_required=None): # overhead of per-qube/per-vcpu Xen structures, # taken from OpenStack nova/virt/xenapi/driver.py # see https://wiki.openstack.org/wiki/XenServer/Overhead # add an extra MB because Nova rounds up to MBs if not qmemman_present: return if mem_required is None: if self.hvm: if self.stubdom_mem: stubdom_mem = self.stubdom_mem else: if self.features.check_with_template('linux-stubdom', True): stubdom_mem = 128 # from libxl_create.c else: stubdom_mem = 28 # from libxl_create.c stubdom_mem += 16 # video ram else: stubdom_mem = 0 mem_required = int(self.memory + stubdom_mem) * 1024 * 1024 qmemman_client = qubes.qmemman.client.QMemmanClient() try: mem_required_with_overhead = mem_required + MEM_OVERHEAD_BASE \ + self.vcpus * MEM_OVERHEAD_PER_VCPU got_memory = qmemman_client.request_memory( mem_required_with_overhead) except IOError as e: raise IOError('Failed to connect to qmemman: {!s}'.format(e)) if not got_memory: qmemman_client.close() raise qubes.exc.QubesMemoryError(self) return qmemman_client @staticmethod @asyncio.coroutine def start_daemon(*command, input=None, **kwargs): '''Start a daemon for the VM This function take care to run it as appropriate user. :param command: command to run (array for :py:meth:`subprocess.check_call`) :param kwargs: args for :py:meth:`subprocess.check_call` :return: None ''' # pylint: disable=redefined-builtin if os.getuid() == 0: # try to always have VM daemons running as normal user, otherwise # some files (like clipboard) may be created as root and cause # permission problems qubes_group = grp.getgrnam('qubes') command = ['runuser', '-u', qubes_group.gr_mem[0], '--'] + \ list(command) p = yield from asyncio.create_subprocess_exec(*command, **kwargs) stdout, stderr = yield from p.communicate(input=input) if p.returncode: raise subprocess.CalledProcessError(p.returncode, command, output=stdout, stderr=stderr) @asyncio.coroutine def start_qrexec_daemon(self): '''Start qrexec daemon. :raises OSError: when starting fails. ''' self.log.debug('Starting the qrexec daemon') qrexec_args = [str(self.xid), self.name, self.default_user] if not self.debug: qrexec_args.insert(0, "-q") qrexec_env = os.environ.copy() if not self.features.check_with_template('qrexec', not self.hvm): self.log.debug( 'Starting the qrexec daemon in background, because of features') qrexec_env['QREXEC_STARTUP_NOWAIT'] = '1' else: qrexec_env['QREXEC_STARTUP_TIMEOUT'] = str(self.qrexec_timeout) try: yield from self.start_daemon( qubes.config.system_path['qrexec_daemon_path'], *qrexec_args, env=qrexec_env) except subprocess.CalledProcessError: raise qubes.exc.QubesVMError(self, 'Cannot execute qrexec-daemon!') @asyncio.coroutine def start_qubesdb(self): '''Start QubesDB daemon. :raises OSError: when starting fails. ''' # drop old connection to QubesDB, if any self._qdb_connection = None self.log.info('Starting Qubes DB') try: yield from self.start_daemon( qubes.config.system_path['qubesdb_daemon_path'], str(self.xid), self.name) except subprocess.CalledProcessError: raise qubes.exc.QubesException('Cannot execute qubesdb-daemon') @asyncio.coroutine def _wait_for_session(self): '''Wait until machine finished boot sequence. This is done by executing qubes RPC call that checks if dummy system service (which is started late in standard runlevel) is active. ''' self.log.info('Waiting for qubes-session') yield from self.run_service_for_stdio('qubes.WaitForSession', user='root', gui=False, input=self.default_user.encode()) self.log.info('qubes-session acquired') self.have_session.set() self.fire_event('domain-has-session') @asyncio.coroutine def create_on_disk(self, pool=None, pools=None): '''Create files needed for VM. ''' self.log.info('Creating directory: {0}'.format(self.dir_path)) os.makedirs(self.dir_path, mode=0o775) if pool or pools: # pylint: disable=attribute-defined-outside-init self.volume_config = _patch_volume_config(self.volume_config, pool, pools) self.storage = qubes.storage.Storage(self) try: yield from self.storage.create() except: try: yield from self.storage.remove() os.rmdir(self.dir_path) except: # pylint: disable=bare-except self.log.exception('failed to cleanup {} after failed VM ' 'creation'.format(self.dir_path)) raise self.log.info('Creating icon symlink: {} -> {}'.format( self.icon_path, self.label.icon_path)) if hasattr(os, "symlink"): os.symlink(self.label.icon_path, self.icon_path) else: shutil.copy(self.label.icon_path, self.icon_path) # fire hooks self.fire_event('domain-create-on-disk') @asyncio.coroutine def remove_from_disk(self): '''Remove domain remnants from disk.''' if not self.is_halted(): raise qubes.exc.QubesVMNotHaltedError( "Can't remove VM {!s}, beacuse it's in state {!r}.".format( self, self.get_power_state())) self.fire_event('domain-remove-from-disk') try: # TODO: make it async? shutil.rmtree(self.dir_path) except OSError as e: if e.errno == errno.ENOENT: pass else: raise yield from self.storage.remove() @asyncio.coroutine def clone_disk_files(self, src, pool=None, pools=None, ): '''Clone files from other vm. :param qubes.vm.qubesvm.QubesVM src: source VM ''' # If the current vm name is not a part of `self.app.domains.keys()`, # then the current vm is in creation process. Calling # `self.is_halted()` at this point, would instantiate libvirt, we want # avoid that. if self.name in self.app.domains.keys() and not self.is_halted(): raise qubes.exc.QubesVMNotHaltedError( self, 'Cannot clone a running domain {!r}'.format(self.name)) msg = "Destination {!s} already exists".format(self.dir_path) assert not os.path.exists(self.dir_path), msg self.log.info('Creating directory: {0}'.format(self.dir_path)) os.makedirs(self.dir_path, mode=0o775) if pool or pools: # pylint: disable=attribute-defined-outside-init self.volume_config = _patch_volume_config(self.volume_config, pool, pools) self.storage = qubes.storage.Storage(self) yield from self.storage.clone(src) self.storage.verify() assert self.volumes != {} if src.icon_path is not None \ and os.path.exists(src.dir_path) \ and self.icon_path is not None: if os.path.islink(src.icon_path): icon_path = os.readlink(src.icon_path) self.log.info( 'Creating icon symlink {} -> {}'.format( self.icon_path, icon_path)) os.symlink(icon_path, self.icon_path) else: self.log.info( 'Copying icon {} -> {}'.format( src.icon_path, self.icon_path)) shutil.copy(src.icon_path, self.icon_path) # fire hooks self.fire_event('domain-clone-files', src=src) # # methods for querying domain state # # state of the machine def get_power_state(self): '''Return power state description string. Return value may be one of those: =============== ======================================================== return value meaning =============== ======================================================== ``'Halted'`` Machine is not active. ``'Transient'`` Machine is running, but does not have :program:`guid` or :program:`qrexec` available. ``'Running'`` Machine is ready and running. ``'Paused'`` Machine is paused (currently not available, see below). ``'Suspended'`` Machine is S3-suspended. ``'Halting'`` Machine is in process of shutting down. ``'Dying'`` Machine crashed and is unusable. ``'Crashed'`` Machine crashed and is unusable, probably because of bug in dom0. ``'NA'`` Machine is in unknown state (most likely libvirt domain is undefined). =============== ======================================================== ``Paused`` state is currently unavailable because of missing code in libvirt/xen glue. FIXME: graph below may be incomplete and wrong. Click on method name to see its documentation. .. graphviz:: digraph { node [fontname="sans-serif"]; edge [fontname="mono"]; Halted; NA; Dying; Crashed; Transient; Halting; Running; Paused [color=gray75 fontcolor=gray75]; Suspended; NA -> Halted; Halted -> NA [constraint=false]; Halted -> Transient [xlabel="start()" URL="#qubes.vm.qubesvm.QubesVM.start"]; Transient -> Running; Running -> Halting [xlabel="shutdown()" URL="#qubes.vm.qubesvm.QubesVM.shutdown" constraint=false]; Halting -> Dying -> Halted [constraint=false]; /* cosmetic, invisible edges to put rank constraint */ Dying -> Halting [style="invis"]; Halting -> Transient [style="invis"]; Running -> Halted [label="force_shutdown()" URL="#qubes.vm.qubesvm.QubesVM.force_shutdown" constraint=false]; Running -> Crashed [constraint=false]; Crashed -> Halted [constraint=false]; Running -> Paused [label="pause()" URL="#qubes.vm.qubesvm.QubesVM.pause" color=gray75 fontcolor=gray75]; Running -> Suspended [label="pause()" URL="#qubes.vm.qubesvm.QubesVM.pause" color=gray50 fontcolor=gray50]; Paused -> Running [label="unpause()" URL="#qubes.vm.qubesvm.QubesVM.unpause" color=gray75 fontcolor=gray75]; Suspended -> Running [label="unpause()" URL="#qubes.vm.qubesvm.QubesVM.unpause" color=gray50 fontcolor=gray50]; Running -> Suspended [label="suspend()" URL="#qubes.vm.qubesvm.QubesVM.suspend"]; Suspended -> Running [label="resume()" URL="#qubes.vm.qubesvm.QubesVM.resume"]; { rank=source; Halted NA }; { rank=same; Transient Halting }; { rank=same; Crashed Dying }; { rank=sink; Paused Suspended }; } .. seealso:: http://wiki.libvirt.org/page/VM_lifecycle Description of VM life cycle from the point of view of libvirt. https://libvirt.org/html/libvirt-libvirt-domain.html#virDomainState Libvirt's enum describing precise state of a domain. ''' # pylint: disable=too-many-return-statements libvirt_domain = self.libvirt_domain if libvirt_domain is None: return 'Halted' try: if libvirt_domain.isActive(): # pylint: disable=line-too-long if libvirt_domain.state()[0] == libvirt.VIR_DOMAIN_PAUSED: return "Paused" elif libvirt_domain.state()[0] == libvirt.VIR_DOMAIN_CRASHED: return "Crashed" elif libvirt_domain.state()[0] == libvirt.VIR_DOMAIN_SHUTDOWN: return "Halting" elif libvirt_domain.state()[0] == libvirt.VIR_DOMAIN_SHUTOFF: return "Dying" elif libvirt_domain.state()[0] == libvirt.VIR_DOMAIN_PMSUSPENDED: # nopep8 return "Suspended" else: if not self.is_fully_usable(): return "Transient" return "Running" return 'Halted' except libvirt.libvirtError as e: if e.get_error_code() == libvirt.VIR_ERR_NO_DOMAIN: return 'Halted' raise assert False def is_halted(self): ''' Check whether this domain's state is 'Halted' :returns: :py:obj:`True` if this domain is halted, \ :py:obj:`False` otherwise. :rtype: bool ''' return self.get_power_state() == 'Halted' def is_running(self): '''Check whether this domain is running. :returns: :py:obj:`True` if this domain is started, \ :py:obj:`False` otherwise. :rtype: bool ''' if self.app.vmm.offline_mode: return False # TODO context manager #1693 return self.libvirt_domain and self.libvirt_domain.isActive() def is_paused(self): '''Check whether this domain is paused. :returns: :py:obj:`True` if this domain is paused, \ :py:obj:`False` otherwise. :rtype: bool ''' return self.libvirt_domain \ and self.libvirt_domain.state()[0] == libvirt.VIR_DOMAIN_PAUSED def is_qrexec_running(self): '''Check whether qrexec for this domain is available. :returns: :py:obj:`True` if qrexec is running, \ :py:obj:`False` otherwise. :rtype: bool ''' if self.xid < 0: return False return os.path.exists('/var/run/qubes/qrexec.%s' % self.name) def is_fully_usable(self): return all(self.fire_event('domain-is-fully-usable')) @qubes.events.handler('domain-is-fully-usable') def on_domain_is_fully_usable(self, event): '''Check whether domain is running and sane. Currently this checks for running qrexec. ''' # pylint: disable=unused-argument # Running gui-daemon implies also VM running if not self.is_qrexec_running(): yield False # memory and disk def get_mem(self): '''Get current memory usage from VM. :returns: Memory usage [FIXME unit]. :rtype: FIXME ''' if self.libvirt_domain is None: return 0 try: if not self.libvirt_domain.isActive(): return 0 return self.libvirt_domain.info()[1] except libvirt.libvirtError as e: if e.get_error_code() in ( # qube no longer exists libvirt.VIR_ERR_NO_DOMAIN, # libxl_domain_info failed (race condition from isActive) libvirt.VIR_ERR_INTERNAL_ERROR): return 0 else: self.log.exception( 'libvirt error code: {!r}'.format(e.get_error_code())) raise def get_mem_static_max(self): '''Get maximum memory available to VM. :returns: Memory limit [FIXME unit]. :rtype: FIXME ''' if self.libvirt_domain is None: return 0 try: return self.libvirt_domain.maxMemory() except libvirt.libvirtError as e: if e.get_error_code() in ( # qube no longer exists libvirt.VIR_ERR_NO_DOMAIN, # libxl_domain_info failed (race condition from isActive) libvirt.VIR_ERR_INTERNAL_ERROR): return 0 else: self.log.exception( 'libvirt error code: {!r}'.format(e.get_error_code())) raise def get_cputime(self): '''Get total CPU time burned by this domain since start. :returns: CPU time usage [FIXME unit]. :rtype: FIXME ''' if self.libvirt_domain is None: return 0 if self.libvirt_domain is None: return 0 if not self.libvirt_domain.isActive(): return 0 try: if not self.libvirt_domain.isActive(): return 0 # this does not work, because libvirt # return self.libvirt_domain.getCPUStats( # libvirt.VIR_NODE_CPU_STATS_ALL_CPUS, 0)[0]['cpu_time']/10**9 return self.libvirt_domain.info()[4] except libvirt.libvirtError as e: if e.get_error_code() in ( # qube no longer exists libvirt.VIR_ERR_NO_DOMAIN, # libxl_domain_info failed (race condition from isActive) libvirt.VIR_ERR_INTERNAL_ERROR): return 0 else: self.log.exception( 'libvirt error code: {!r}'.format(e.get_error_code())) raise # miscellanous def get_start_time(self): '''Tell when machine was started. :rtype: datetime.datetime ''' if not self.is_running(): return None # TODO shouldn't this be qubesdb? start_time = self.app.vmm.xs.read('', '/vm/{}/start_time'.format(self.uuid)) if start_time != '': return datetime.datetime.fromtimestamp(float(start_time)) return None def is_outdated(self): '''Check whether domain needs restart to update root image from \ template. :returns: :py:obj:`True` if is outdated, :py:obj:`False` otherwise. :rtype: bool ''' # pylint: disable=no-member # Makes sense only on VM based on template if self.template is None: return False if not self.is_running(): return False if not hasattr(self.template, 'rootcow_img'): return False rootimg_inode = os.stat(self.template.root_img) try: rootcow_inode = os.stat(self.template.rootcow_img) except OSError: # The only case when rootcow_img doesn't exists is in the middle of # commit_changes, so VM is outdated right now return True current_dmdev = "/dev/mapper/snapshot-{0:x}:{1}-{2:x}:{3}".format( rootimg_inode[2], rootimg_inode[1], rootcow_inode[2], rootcow_inode[1]) # FIXME # 51712 (0xCA00) is xvda # backend node name not available through xenapi :( used_dmdev = self.app.vmm.xs.read('', '/local/domain/0/backend/vbd/{}/51712/node'.format(self.xid)) return used_dmdev != current_dmdev # # helper methods # def relative_path(self, path): '''Return path relative to py:attr:`dir_path`. :param str path: Path in question. :returns: Relative path. ''' return os.path.relpath(path, self.dir_path) def create_qdb_entries(self): '''Create entries in Qubes DB. ''' # pylint: disable=no-member self.qdb.write('/name', self.name) self.qdb.write('/type', self.__class__.__name__) self.qdb.write('/qubes-vm-updateable', str(self.updateable)) self.qdb.write('/qubes-vm-persistence', 'full' if self.updateable else 'rw-only') self.qdb.write('/qubes-debug-mode', str(int(self.debug))) try: self.qdb.write('/qubes-base-template', self.template.name) except AttributeError: self.qdb.write('/qubes-base-template', '') self.qdb.write('/qubes-random-seed', base64.b64encode(qubes.utils.urandom(64))) if self.provides_network: # '/qubes-netvm-network' value is only checked for being non empty self.qdb.write('/qubes-netvm-network', self.gateway) self.qdb.write('/qubes-netvm-gateway', self.gateway) self.qdb.write('/qubes-netvm-netmask', self.netmask) for i, addr in zip(('primary', 'secondary'), self.dns): self.qdb.write('/qubes-netvm-{}-dns'.format(i), addr) if self.netvm is not None: self.qdb.write('/qubes-ip', self.visible_ip) self.qdb.write('/qubes-netmask', self.visible_netmask) self.qdb.write('/qubes-gateway', self.visible_gateway) for i, addr in zip(('primary', 'secondary'), self.dns): self.qdb.write('/qubes-{}-dns'.format(i), addr) tzname = qubes.utils.get_timezone() if tzname: self.qdb.write('/qubes-timezone', tzname) for feature, value in self.features.items(): self.qdb.write('/features/{0}'.format(feature), str(value) if value else '') self.qdb.write('/qubes-block-devices', '') self.qdb.write('/qubes-usb-devices', '') # TODO: Currently the whole qmemman is quite Xen-specific, so stay with # xenstore for it until decided otherwise if qmemman_present: self.app.vmm.xs.set_permissions('', '/local/domain/{}/memory'.format(self.xid), [{'dom': self.xid}]) self.fire_event('domain-qdb-create') # TODO async; update this in constructor def _update_libvirt_domain(self): '''Re-initialise :py:attr:`libvirt_domain`.''' domain_config = self.create_config_file() try: self._libvirt_domain = self.app.vmm.libvirt_conn.defineXML( domain_config) except libvirt.libvirtError as e: if e.get_error_code() == libvirt.VIR_ERR_OS_TYPE \ and e.get_str2() == 'hvm': raise qubes.exc.QubesVMError(self, 'HVM qubes are not supported on this machine. ' 'Check BIOS settings for VT-x/AMD-V extensions.') else: raise # # workshop -- those are to be reworked later # def get_prefmem(self): # TODO: qmemman is still xen specific untrusted_meminfo_key = self.app.vmm.xs.read('', '/local/domain/{}/memory/meminfo'.format(self.xid)) if untrusted_meminfo_key is None or untrusted_meminfo_key == '': return 0 domain = qubes.qmemman.DomainState(self.xid) qubes.qmemman.algo.refresh_meminfo_for_domain( domain, untrusted_meminfo_key) if domain.mem_used is None: # apparently invalid xenstore content return 0 domain.memory_maximum = self.get_mem_static_max() * 1024 return qubes.qmemman.algo.prefmem(domain) / 1024 def _clean_volume_config(config): common_attributes = ['name', 'pool', 'size', 'internal', 'removable', 'revisions_to_keep', 'rw', 'snap_on_start', 'save_on_stop', 'source'] config_copy = copy.deepcopy(config) return {k: v for k, v in config_copy.items() if k in common_attributes} def _patch_pool_config(config, pool=None, pools=None): assert pool is not None or pools is not None is_saveable = 'save_on_stop' in config and config['save_on_stop'] is_resetable = not ('snap_on_start' in config and # volatile config['snap_on_start'] and not is_saveable) is_exportable = is_saveable or is_resetable name = config['name'] if pool and is_exportable and config['pool'] == 'default': config['pool'] = str(pool) elif pool and not is_exportable: pass elif pools and name in pools.keys(): if is_exportable: config['pool'] = str(pools[name]) else: msg = "Can't clone a snapshot volume {!s} to pool {!s} " \ .format(name, pools[name]) raise qubes.exc.QubesException(msg) return config def _patch_volume_config(volume_config, pool=None, pools=None): assert not (pool and pools), \ 'You can not pass pool & pools parameter at same time' assert pool or pools result = {} for name, config in volume_config.items(): # copy only the subset of volume_config key/values dst_config = _clean_volume_config(config) if pool is not None or pools is not None: dst_config = _patch_pool_config(dst_config, pool, pools) result[name] = dst_config return result