1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495 |
- #!/usr/bin/python3
- #
- # The Qubes OS Project, http://www.qubes-os.org
- #
- # Copyright (C) 2012 Agnieszka Kostrzewa <agnieszka.kostrzewa@gmail.com>
- # Copyright (C) 2012 Marek Marczykowski-Górecki
- # <marmarek@invisiblethingslab.com>
- # Copyright (C) 2017 Wojtek Porczyk <woju@invisiblethingslab.com>
- #
- # 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 Lesser General Public License along
- # with this program; if not, see <http://www.gnu.org/licenses/>.
- #
- #
- import collections
- import functools
- import re
- import subprocess
- import traceback
- from qubesadmin.tools import QubesArgumentParser
- from qubesadmin import devices
- from qubesadmin import utils as admin_utils
- from qubesadmin.tools import qvm_start
- import qubesadmin.exc
- from . import bootfromdevice
- from . import utils
- from . import multiselectwidget
- from . import common_threads
- from . import device_list
- from . import clone_vm
- from .appmenu_select import AppmenuSelectManager
- from . import firewall
- from PyQt5 import QtCore, QtWidgets, QtGui # pylint: disable=import-error
- from . import ui_settingsdlg # pylint: disable=no-name-in-module
- # pylint: disable=too-few-public-methods
- class RenameVMThread(common_threads.QubesThread):
- def __init__(self, vm, new_vm_name, dependencies):
- super().__init__(vm)
- self.new_vm_name = new_vm_name
- self.dependencies = dependencies
- def run(self):
- try:
- new_vm = self.vm.app.clone_vm(self.vm, self.new_vm_name)
- failed_props = []
- for (holder, prop) in self.dependencies:
- try:
- if holder is None:
- setattr(self.vm.app, prop, new_vm)
- else:
- setattr(holder, prop, new_vm)
- except qubesadmin.exc.QubesException:
- failed_props += [(holder, prop)]
- if not failed_props:
- del self.vm.app.domains[self.vm.name]
- else:
- list_text = utils.format_dependencies_list(failed_props)
- self.msg = (self.tr("Warning: rename partially unsuccessful!"),
- self.tr("Some properties could not be changed to "
- "the new name. The system has now both {} "
- "and {} qubes. To resolve this, please "
- "check and change the following properties "
- "and remove the qube {} manually.<br>"
- ).format(self.vm.name, self.vm.name,
- self.vm.name) + list_text)
- except qubesadmin.exc.QubesException as ex:
- self.msg = (self.tr("Rename error!"), str(ex))
- except Exception as ex: # pylint: disable=broad-except
- self.msg = (self.tr("Rename error!"), repr(ex))
- # pylint: disable=too-few-public-methods
- class RefreshAppsVMThread(common_threads.QubesThread):
- def __init__(self, vm, button):
- super().__init__(vm)
- self.button = button
- def run(self):
- vms_to_refresh = [self.vm]
- template = getattr(self.vm, 'template', None)
- if template:
- vms_to_refresh.append(template)
- for vm in vms_to_refresh:
- self.button.setText(
- self.tr('Refresh in progress (refreshing applications '
- 'from {})').format(vm.name))
- try:
- if not utils.is_running(vm, True):
- not_running = True
- vm.start()
- else:
- not_running = False
- subprocess.check_call(['qvm-sync-appmenus', vm.name])
- if not_running:
- vm.shutdown()
- except Exception as ex: # pylint: disable=broad-except
- self.msg = (self.tr("Refresh failed!"), str(ex))
- # pylint: disable=too-many-instance-attributes
- class VMSettingsWindow(ui_settingsdlg.Ui_SettingsDialog, QtWidgets.QDialog):
- tabs_indices = collections.OrderedDict((
- ('basic', 0),
- ('advanced', 1),
- ('firewall', 2),
- ('devices', 3),
- ('applications', 4),
- ('services', 5),
- ))
- def __init__(self, vm, init_page="basic", qapp=None, qubesapp=None,
- parent=None):
- super().__init__(parent)
- self.vm = vm
- self.qapp = qapp
- self.qubesapp = qubesapp
- self.threads_list = []
- self.progress = None
- self.thread_closes = False
- self.setupUi(self)
- self.setWindowTitle(self.tr("Settings: {vm}").format(vm=self.vm.name))
- if init_page in self.tabs_indices:
- idx = self.tabs_indices[init_page]
- assert idx in range(self.tabWidget.count())
- self.tabWidget.setCurrentIndex(idx)
- self.buttonBox.button(QtWidgets.QDialogButtonBox.Apply).clicked.connect(
- self.apply)
- self.tabWidget.currentChanged.connect(self.current_tab_changed)
- # Initialize several auxillary variables for pylint's sake
- self.root_img_size = None
- self.priv_img_size = None
- ###### basic tab
- self.__init_basic_tab__()
- self.rename_vm_button.clicked.connect(self.rename_vm)
- self.delete_vm_button.clicked.connect(self.remove_vm)
- self.clone_vm_button.clicked.connect(self.clone_vm)
- ###### advanced tab
- self.__init_advanced_tab__()
- self.include_in_balancing.stateChanged.connect(
- self.include_in_balancing_changed)
- self.init_mem.editingFinished.connect(self.check_mem_changes)
- self.max_mem_size.editingFinished.connect(self.check_mem_changes)
- self.check_mem_changes()
- self.boot_from_device_button.clicked.connect(
- self.boot_from_cdrom_button_pressed)
- ###### firewall tab
- if self.tabWidget.isTabEnabled(self.tabs_indices['firewall']):
- model = firewall.QubesFirewallRulesModel()
- try:
- model.set_vm(vm)
- self.set_fw_model(model)
- self.firewall_modified_outside_label.setVisible(False)
- except firewall.FirewallModifiedOutsideError:
- self.disable_all_fw_conf()
- except qubesadmin.exc.QubesException:
- self.tabWidget.setTabEnabled(
- self.tabs_indices['firewall'], False)
- self.new_rule_button.clicked.connect(self.new_rule_button_pressed)
- self.edit_rule_button.clicked.connect(self.edit_rule_button_pressed)
- self.delete_rule_button.clicked.connect(
- self.delete_rule_button_pressed)
- self.policy_deny_radio_button.clicked.connect(self.policy_changed)
- self.policy_allow_radio_button.clicked.connect(self.policy_changed)
- if init_page == 'firewall':
- self.check_network_availability()
- ####### devices tab
- self.__init_devices_tab__()
- self.dev_list.selectedChanged.connect(self.devices_selection_changed)
- self.no_strict_reset_button.clicked.connect(
- self.strict_reset_button_pressed)
- self.current_strict_reset_list = []
- self.new_strict_reset_list = []
- self.define_strict_reset_devices()
- ####### services tab
- self.__init_services_tab__()
- self.add_srv_button.clicked.connect(self.__add_service__)
- self.remove_srv_button.clicked.connect(self.__remove_service__)
- ####### apps tab
- if self.tabWidget.isTabEnabled(self.tabs_indices["applications"]):
- self.app_list = multiselectwidget.MultiSelectWidget(self)
- self.apps_layout.addWidget(self.app_list)
- self.app_list_manager = AppmenuSelectManager(self.vm, self.app_list)
- self.refresh_apps_button.clicked.connect(
- self.refresh_apps_button_pressed)
- # template change
- if self.template_name.isEnabled():
- self.template_name.currentIndexChanged.connect(
- self.template_apps_change)
- self.warn_template_missing_apps.setVisible(
- self.app_list_manager.has_missing)
- def setup_application(self):
- self.qapp.setApplicationName(self.tr("Qube Settings"))
- self.qapp.setWindowIcon(QtGui.QIcon.fromTheme("qubes-manager"))
- def clear_threads(self):
- for thread in self.threads_list:
- if thread.isFinished():
- if self.progress:
- self.progress.hide()
- self.progress = None
- if thread.msg:
- (title, msg) = thread.msg
- QtWidgets.QMessageBox.warning(
- self,
- title,
- msg)
- self.threads_list.remove(thread)
- if self.thread_closes:
- self.done(0)
- return
- raise RuntimeError(self.tr('No finished thread found'))
- def keyPressEvent(self, event): # pylint: disable=invalid-name
- if event.key() == QtCore.Qt.Key_Enter \
- or event.key() == QtCore.Qt.Key_Return:
- return
- super().keyPressEvent(event)
- def accept(self):
- self.save_and_apply()
- def save_changes(self):
- with common_threads.busy_cursor():
- error = self.__save_changes__()
- if error:
- QtWidgets.QMessageBox.warning(
- self,
- self.tr("Error while changing settings for {0}!"
- ).format(self.vm.name),
- self.tr("ERROR: {0}").format('\n'.join(error)))
- def apply(self):
- self.save_changes()
- # these signals must be disconnected to avoid unintended behavior
- # on refreshing the drop-downs
- self.netVM.currentIndexChanged.disconnect()
- self.kernel.currentIndexChanged.disconnect()
- self.default_dispvm.currentIndexChanged.disconnect()
- self.__init_basic_tab__()
- self.__init_advanced_tab__()
- def save_and_apply(self):
- self.save_changes()
- self.done(0)
- def __save_changes__(self):
- ret = []
- try:
- ret_tmp = self.__apply_basic_tab__()
- if ret_tmp:
- ret += [self.tr("Basic tab:")] + ret_tmp
- ret_tmp = self.__apply_advanced_tab__()
- if ret_tmp:
- ret += [self.tr("Advanced tab:")] + ret_tmp
- ret_tmp = self.__apply_devices_tab__()
- if ret_tmp:
- ret += [self.tr("Devices tab:")] + ret_tmp
- ret_tmp = self.__apply_services_tab__()
- if ret_tmp:
- ret += [self.tr("Sevices tab:")] + ret_tmp
- except qubesadmin.exc.QubesException as qex:
- ret.append(self.tr('Error while saving changes: ') + str(qex))
- except Exception as ex: # pylint: disable=broad-except
- ret.append(repr(ex))
- try:
- if self.tabWidget.isTabEnabled(self.tabs_indices['firewall']) and \
- self.policy_allow_radio_button.isEnabled():
- self.fw_model.apply_rules(
- self.policy_allow_radio_button.isChecked(),
- self.temp_full_access.isChecked(),
- self.temp_full_access_time.value())
- except qubesadmin.exc.QubesException as qex:
- ret += [self.tr("Firewall tab:"), str(qex)]
- except Exception as ex: # pylint: disable=broad-except
- ret += [self.tr("Firewall tab:"), repr(ex)]
- try:
- if self.tabWidget.isTabEnabled(self.tabs_indices["applications"]):
- self.app_list_manager.save_appmenu_select_changes()
- except qubesadmin.exc.QubesException as qex:
- ret += [self.tr("Applications tab:"), str(qex)]
- except Exception as ex: # pylint: disable=broad-except
- ret += [self.tr("Applications tab:"), repr(ex)]
- utils.debug('\n'.join(ret))
- return ret
- def check_network_availability(self):
- netvm = getattr(self.vm, 'netvm', None)
- provides_network = getattr(self.vm, 'provides_network', False)
- self.no_netvm_label.setVisible(netvm is None and not provides_network)
- try:
- no_firewall_state = \
- netvm is not None and \
- not netvm.features.check_with_template('qubes-firewall', False)
- except qubesadmin.exc.QubesDaemonAccessError:
- no_firewall_state = False
- self.netvm_no_firewall_label.setVisible(no_firewall_state)
- self.sysnet_warning_label.setVisible(netvm is None and provides_network)
- def current_tab_changed(self, idx):
- if idx == self.tabs_indices["firewall"]:
- self.check_network_availability()
- ######### basic tab
- # TODO LISTENERS
- # - vm start/shutdown -> setEnabled on fields: template labels
- # - vm create/delete -> choices lists, whole window deactiv (if self.vm)
- # - property-set -> individual fields
- # TODO INTERACTIONS
- # netvm -> networking_groupbox
- # hvm -> include_in_balancing
- def __init_basic_tab__(self):
- self.vmname.setText(self.vm.name)
- self.vmname.setValidator(
- QtGui.QRegExpValidator(
- QtCore.QRegExp("[a-zA-Z0-9_-]*",
- QtCore.Qt.CaseInsensitive), None))
- self.vmname.setEnabled(False)
- self.rename_vm_button.setEnabled(not self.vm.is_running())
- self.delete_vm_button.setEnabled(not self.vm.is_running())
- if utils.is_running(self.vm, False):
- self.delete_vm_button.setText(
- self.tr('Delete qube (cannot delete a running qube)'))
- if self.vm.klass == 'AdminVM':
- self.vmlabel.setVisible(False)
- else:
- try:
- utils.initialize_widget_with_labels(
- widget=self.vmlabel,
- qubes_app=self.qubesapp,
- holder=self.vm)
- self.vmlabel.setVisible(True)
- self.vmlabel.setEnabled(not utils.is_running(self.vm, False))
- except qubesadmin.exc.QubesDaemonAccessError:
- self.vmlabel.setEnabled(False)
- if self.vm.klass == 'AppVM':
- try:
- utils.initialize_widget_with_vms(
- widget=self.template_name,
- qubes_app=self.qubesapp,
- filter_function=(lambda vm: vm.klass == 'TemplateVM'),
- holder=self.vm,
- property_name='template')
- except qubesadmin.exc.QubesDaemonAccessError:
- self.template_name.setCurrentIndex(-1)
- self.template_name.setEnabled(False)
- elif self.vm.klass == 'DispVM':
- try:
- utils.initialize_widget_with_vms(
- widget=self.template_name,
- qubes_app=self.qubesapp,
- filter_function=(
- lambda vm: getattr(vm, 'template_for_dispvms', False)),
- holder=self.vm,
- property_name='template')
- except qubesadmin.exc.QubesDaemonAccessError:
- self.template_name.setCurrentIndex(-1)
- self.template_name.setEnabled(False)
- else:
- self.template_name.setEnabled(False)
- if utils.is_running(self.vm, False):
- self.template_name.setEnabled(False)
- try:
- utils.initialize_widget_with_vms(
- widget=self.netVM,
- qubes_app=self.qubesapp,
- filter_function=(lambda vm:
- getattr(vm, 'provides_network', False)),
- holder=self.vm,
- property_name='netvm',
- allow_default=True,
- allow_none=True)
- except qubesadmin.exc.QubesDaemonAccessError:
- self.netVM.setEnabled(False)
- self.netVM.setCurrentIndex(-1)
- self.netVM.currentIndexChanged.connect(self.check_warn_dispvmnetvm)
- self.netVM.currentIndexChanged.connect(self.check_warn_templatenetvm)
- try:
- self.include_in_backups.setChecked(self.vm.include_in_backups)
- except qubesadmin.exc.QubesDaemonAccessError:
- self.include_in_backups.setEnabled(False)
- try:
- has_shutdown_idle = self.vm.features.get(
- "supported-service.shutdown-idle", False)
- if has_shutdown_idle:
- self.idle_shutdown_checkbox.setChecked(self.vm.features.get(
- "shutdown-idle", False))
- else:
- text = "Shut down when idle "
- if getattr(self.vm, 'template', None):
- additional_text = "(unavailable: package " \
- "qubes-app-shutdown-idle missing " \
- "in the template)"
- else:
- additional_text = "(unavailable: package " \
- "qubes-app-shutdown-idle missing " \
- "in the qube)"
- self.idle_shutdown_checkbox.setText(
- text + additional_text)
- self.idle_shutdown_checkbox.setEnabled(False)
- except qubesadmin.exc.QubesDaemonCommunicationError:
- self.idle_shutdown_checkbox.setText(
- self.idle_shutdown_checkbox.text() +
- " (unavailable: permission denied)")
- self.idle_shutdown_checkbox.setEnabled(False)
- try:
- self.autostart_vm.setChecked(self.vm.autostart)
- self.autostart_vm.setVisible(True)
- except qubesadmin.exc.QubesDaemonAccessError:
- self.autostart_vm.setEnabled(False)
- except AttributeError:
- self.autostart_vm.setVisible(False)
- # type
- self.type_label.setText(self.vm.klass)
- # installed by rpm
- self.rpm_label.setText(
- 'Yes' if getattr(self.vm, 'installed_by_rpm', False) else 'No')
- # networking info
- if getattr(self.vm, 'netvm', None):
- self.networking_groupbox.setEnabled(True)
- self.ip_label.setText(getattr(self.vm, 'ip', None) or "none")
- self.netmask_label.setText(
- getattr(self.vm, 'visible_netmask', None) or "none")
- self.gateway_label.setText(
- getattr(self.vm, 'visible_gateway', None) or "none")
- dns_list = getattr(self.vm, 'dns', ['10.139.1.1', '10.139.1.2'])
- self.dns_label.setText(", ".join(dns_list))
- else:
- self.networking_groupbox.setEnabled(False)
- # max priv storage
- try:
- self.priv_img_size = self.vm.volumes['private'].size // 1024**2
- self.max_priv_storage.setMinimum(self.priv_img_size)
- self.max_priv_storage.setValue(self.priv_img_size)
- self.max_priv_storage.setMaximum(
- max(self.priv_img_size,
- self.qubesapp.pools[self.vm.volumes['private'].pool].size
- // 1024**2))
- except qubesadmin.exc.QubesException:
- self.max_priv_storage.setEnabled(False)
- try:
- self.root_img_size = self.vm.volumes['root'].size // 1024**2
- self.root_resize.setValue(self.root_img_size)
- self.root_resize.setMinimum(self.root_img_size)
- self.root_resize.setMaximum(
- max(self.root_img_size,
- self.qubesapp.pools[self.vm.volumes['root'].pool].size
- // 1024**2))
- self.root_resize.setEnabled(self.vm.volumes['root'].save_on_stop)
- if not self.root_resize.isEnabled():
- self.root_resize.setToolTip(
- self.tr("To change system storage size, change properties "
- "of the underlying template."))
- self.root_resize_label.setEnabled(self.root_resize.isEnabled())
- except qubesadmin.exc.QubesException:
- self.root_resize.setEnabled(False)
- self.warn_template_missing_apps.setVisible(False)
- def __apply_basic_tab__(self):
- msg = []
- # vm label changed
- try:
- if self.vmlabel.isVisible():
- if utils.did_widget_selection_change(self.vmlabel):
- self.vm.label = self.vmlabel.currentData()
- except qubesadmin.exc.QubesException as ex:
- msg.append(str(ex))
- # vm template changed
- try:
- if utils.did_widget_selection_change(self.template_name):
- self.vm.template = self.template_name.currentData()
- except qubesadmin.exc.QubesException as ex:
- msg.append(str(ex))
- # vm netvm changed
- try:
- if utils.did_widget_selection_change(self.netVM):
- self.vm.netvm = self.netVM.currentData()
- except qubesadmin.exc.QubesException as ex:
- msg.append(str(ex))
- # include in backups
- try:
- if self.include_in_backups.isEnabled() and\
- self.vm.include_in_backups != \
- self.include_in_backups.isChecked():
- self.vm.include_in_backups = self.include_in_backups.isChecked()
- except qubesadmin.exc.QubesException as ex:
- msg.append(str(ex))
- # autostart_vm
- try:
- if self.autostart_vm.isVisible():
- if self.vm.autostart != self.autostart_vm.isChecked():
- self.vm.autostart = self.autostart_vm.isChecked()
- except qubesadmin.exc.QubesException as ex:
- msg.append(str(ex))
- # shutdown-idle
- try:
- current_idle = self.vm.features.get("shutdown-idle", False)
- if self.idle_shutdown_checkbox.isEnabled() and \
- self.idle_shutdown_checkbox.isChecked() != current_idle:
- self.vm.features["shutdown-idle"] = \
- self.idle_shutdown_checkbox.isChecked()
- except qubesadmin.exc.QubesException as ex:
- msg.append(str(ex))
- # max priv storage
- if self.max_priv_storage.isEnabled():
- priv_size = self.max_priv_storage.value()
- if self.priv_img_size != priv_size:
- try:
- self.vm.volumes['private'].resize(priv_size * 1024**2)
- self.priv_img_size = priv_size
- except qubesadmin.exc.QubesException as ex:
- msg.append(str(ex))
- # max sys storage
- if self.root_resize.isEnabled():
- sys_size = self.root_resize.value()
- if self.root_img_size != sys_size:
- try:
- self.vm.volumes['root'].resize(sys_size * 1024**2)
- self.root_img_size = sys_size
- except qubesadmin.exc.QubesException as ex:
- msg.append(str(ex))
- return msg
- def check_mem_changes(self):
- self.warn_too_much_mem_label.setVisible(False)
- if not self.include_in_balancing.isChecked():
- # do not interfere with settings if the VM is not included in memory
- # balancing
- return
- if not self.max_mem_size.isEnabled() or not self.init_mem.isEnabled():
- # do not interfere with settings if they are unavailable
- return
- if self.max_mem_size.value() < self.init_mem.value():
- QtWidgets.QMessageBox.warning(
- self,
- self.tr("Warning!"),
- self.tr("Max memory can not be less than initial memory.<br>"
- "Setting max memory to equal initial memory."))
- self.max_mem_size.setValue(self.init_mem.value())
- # Linux specific limit: init memory must not be below
- # max_mem_size/10.79 in order to allow scaling up to
- # max_mem_size (or else "add_memory() failed: -17" problem)
- try:
- is_linux = self.vm.features.check_with_template('os', None) == \
- 'Linux'
- except qubesadmin.exc.QubesException:
- is_linux = False
- if is_linux and \
- self.init_mem.value() * 10 < self.max_mem_size.value():
- self.warn_too_much_mem_label.setVisible(True)
- def check_warn_templatenetvm(self):
- if self.vm.klass != 'TemplateVM':
- return
- current_netvm = self.netVM.currentData()
- if current_netvm is None:
- return
- if current_netvm != qubesadmin.DEFAULT:
- QtWidgets.QMessageBox.warning(
- self,
- self.tr("Warning!"),
- self.tr(
- "Connecting a TemplateVM directly to a network is highly"
- " discouraged! <br> <small>You are breaking a basic part "
- "of Qubes security and there is probably no real need"
- " to do so. Continue at your own risk.</small>"))
- def check_warn_dispvmnetvm(self):
- if not hasattr(self.vm, 'default_dispvm'):
- self.warn_netvm_dispvm.setVisible(False)
- return
- dispvm = self.default_dispvm.currentData()
- own_netvm = self.netVM.currentData()
- if dispvm == qubesadmin.DEFAULT:
- try:
- dispvm = self.vm.property_get_default('default_dispvm')
- except qubesadmin.exc.QubesDaemonAccessError:
- pass
- if dispvm == self.vm:
- self.warn_netvm_dispvm.setVisible(False)
- return
- dispvm_netvm = getattr(dispvm, 'netvm', None)
- if own_netvm == qubesadmin.DEFAULT:
- try:
- own_netvm = self.vm.property_get_default('netvm')
- except qubesadmin.exc.QubesDaemonAccessError:
- # no point in warning if we don't know what we're warning about
- self.warn_netvm_dispvm.setVisible(False)
- return
- if dispvm_netvm and dispvm_netvm != own_netvm:
- self.warn_netvm_dispvm.setVisible(True)
- else:
- self.warn_netvm_dispvm.setVisible(False)
- def rename_vm(self):
- dependencies = admin_utils.vm_dependencies(self.vm.app, self.vm)
- running_dependencies = [vm.name for (vm, prop) in dependencies
- if vm and prop == 'template'
- and utils.is_running(vm, False)]
- if running_dependencies:
- QtWidgets.QMessageBox.warning(
- self,
- self.tr("Qube cannot be renamed!"),
- self.tr(
- "The following qubes using this qube as a template are "
- "running: <br> {}. <br> In order to rename this qube, you "
- "must first shut them down.").format(
- ", ".join(running_dependencies)))
- return
- new_vm_name, ok = QtWidgets.QInputDialog.getText(
- self,
- self.tr('Rename qube'),
- self.tr('New name: (WARNING: all other changes will be discarded)'),
- text=self.vm.name)
- if ok:
- thread = RenameVMThread(self.vm, new_vm_name, dependencies)
- self.threads_list.append(thread)
- thread.finished.connect(self.clear_threads)
- self.progress = QtWidgets.QProgressDialog(
- self.tr("Renaming Qube..."), "", 0, 0)
- self.progress.setCancelButton(None)
- self.progress.setModal(True)
- self.thread_closes = True
- self.progress.show()
- thread.start()
- def remove_vm(self):
- dependencies = admin_utils.vm_dependencies(self.vm.app, self.vm)
- if dependencies:
- list_text = utils.format_dependencies_list(dependencies)
- QtWidgets.QMessageBox.warning(
- self,
- self.tr("Qube cannot be removed!"),
- self.tr("This qube cannot be removed. It is used as:"
- " <br> {} <small>If you want to remove this qube, "
- "you should remove or change settings of each qube "
- "or setting that uses it.</small>").format(list_text))
- return
- answer, ok = QtWidgets.QInputDialog.getText(
- self,
- self.tr('Delete qube'),
- self.tr('Are you absolutely sure you want to delete this qube? '
- '<br/> All qube settings and data will be irrevocably'
- ' deleted. <br/> If you are sure, please enter this '
- 'qube\'s name below.'))
- if ok and answer == self.vm.name:
- thread = common_threads.RemoveVMThread(self.vm)
- thread.finished.connect(self.clear_threads)
- self.threads_list.append(thread)
- self.progress = QtWidgets.QProgressDialog(
- self.tr("Deleting Qube..."), "", 0, 0)
- self.progress.setCancelButton(None)
- self.progress.setModal(True)
- self.thread_closes = True
- self.progress.show()
- thread.start()
- elif ok:
- QtWidgets.QMessageBox.warning(
- self,
- self.tr("Removal cancelled"),
- self.tr("The qube will not be removed."))
- def clone_vm(self):
- with common_threads.busy_cursor():
- clone_window = clone_vm.CloneVMDlg(
- self.qapp, self.qubesapp, src_vm=self.vm)
- clone_window.exec_()
- ######### advanced tab
- def __init_advanced_tab__(self):
- vm_memory = getattr(self.vm, 'memory', None)
- vm_maxmem = getattr(self.vm, 'maxmem', None)
- if vm_memory is None:
- self.init_mem.setEnabled(False)
- else:
- self.init_mem.setValue(int(vm_memory))
- if vm_maxmem is None:
- self.max_mem_size.setEnabled(False)
- elif vm_maxmem > 0:
- self.max_mem_size.setValue(int(vm_maxmem))
- else:
- try:
- maxmem = self.vm.property_get_default('maxmem')
- except qubesadmin.exc.QubesDaemonAccessError:
- maxmem = 0
- if maxmem == 0:
- maxmem = vm_memory
- self.max_mem_size.setValue(
- int(utils.get_feature(
- self.vm, 'qubesmanager.maxmem_value', maxmem)))
- self.vcpus.setMinimum(1)
- self.vcpus.setValue(int(getattr(self.vm, 'vcpus', 1)))
- self.include_in_balancing.setEnabled(True)
- self.include_in_balancing.setChecked(
- int(getattr(self.vm, 'maxmem', 0)) > 0)
- self.max_mem_size.setEnabled(self.include_in_balancing.isChecked())
- # in case VM is HVM
- if hasattr(self.vm, "kernel"):
- self.kernel_groupbox.setVisible(True)
- try:
- utils.initialize_widget_with_kernels(
- widget=self.kernel,
- qubes_app=self.qubesapp,
- allow_none=True,
- allow_default=True,
- holder=self.vm,
- property_name='kernel')
- self.kernel.currentIndexChanged.connect(self.kernel_changed)
- self.kernel_opts.setText(getattr(self.vm, 'kernelopts', '-'))
- except qubesadmin.exc.QubesDaemonAccessError:
- self.kernel_groupbox.setVisible(False)
- else:
- self.kernel_groupbox.setVisible(False)
- if not hasattr(self.vm, 'default_dispvm'):
- self.other_groupbox.setVisible(False)
- else:
- try:
- self.other_groupbox.setVisible(True)
- utils.initialize_widget_with_vms(
- widget=self.default_dispvm,
- qubes_app=self.qubesapp,
- filter_function=(lambda vm:
- getattr(
- vm, 'template_for_dispvms', False)),
- allow_none=True,
- allow_default=True,
- holder=self.vm,
- property_name='default_dispvm'
- )
- self.default_dispvm.currentIndexChanged.connect(
- self.check_warn_dispvmnetvm)
- except qubesadmin.exc.QubesDaemonAccessError:
- self.other_groupbox.setVisible(False)
- self.check_warn_dispvmnetvm()
- self.update_virt_mode_list()
- try:
- windows_running = \
- self.vm.features.check_with_template('os', None) == 'Windows' \
- and self.vm.is_running()
- except qubesadmin.exc.QubesException:
- windows_running = False
- self.seamless_on_button.setEnabled(windows_running)
- self.seamless_off_button.setEnabled(windows_running)
- self.seamless_on_button.clicked.connect(self.enable_seamless)
- self.seamless_off_button.clicked.connect(self.disable_seamless)
- self.dvm_template_checkbox.setChecked(
- getattr(self.vm, 'template_for_dispvms', False))
- self.provides_network_checkbox.setChecked(
- getattr(self.vm, 'provides_network', False))
- if self.provides_network_checkbox.isChecked():
- domains_using = [vm.name for vm
- in getattr(self.vm, 'connected_vms', [])]
- if domains_using:
- self.provides_network_checkbox.setEnabled(False)
- self.provides_network_checkbox.setToolTip(self.tr(
- "Cannot change this setting while this qube is used as a "
- "NetVM by the following qubes:\n") +
- "\n".join(domains_using))
- try:
- self.run_in_debug_mode.setChecked(self.vm.debug)
- self.run_in_debug_mode.setVisible(True)
- except AttributeError:
- self.run_in_debug_mode.setVisible(False)
- utils.initialize_widget(
- widget=self.allow_fullscreen,
- choices=[
- ('(use system default)', None),
- ('allow', True),
- ('disallow', False)
- ],
- selected_value=utils.get_boolean_feature(self.vm,
- 'gui-allow-fullscreen'))
- self.allow_fullscreen_initial = self.allow_fullscreen.currentIndex()
- utils.initialize_widget(
- widget=self.allow_utf8,
- choices=[
- ('(use system default)', None),
- ('allow', True),
- ('disallow', False)
- ],
- selected_value=utils.get_boolean_feature(self.vm,
- 'gui-allow-utf8-titles'))
- self.allow_utf8_initial = self.allow_utf8.currentIndex()
- def enable_seamless(self):
- try:
- self.vm.run_service_for_stdio("qubes.SetGuiMode", input=b'SEAMLESS')
- except qubesadmin.exc.QubesException as ex:
- QtWidgets.QMessageBox.warning(
- self,
- self.tr("Failed to set seamless mode"),
- self.tr("Error occured: {}".format(str(ex))))
- def disable_seamless(self):
- try:
- self.vm.run_service_for_stdio("qubes.SetGuiMode",
- input=b'FULLSCREEN')
- except qubesadmin.exc.QubesException as ex:
- QtWidgets.QMessageBox.warning(
- self,
- self.tr("Failed to set fullscreen mode"),
- self.tr("Error occured: {}".format(str(ex))))
- def __apply_advanced_tab__(self):
- msg = []
- # mem/cpu
- try:
- if self.init_mem.isEnabled() and \
- self.init_mem.value() != int(self.vm.memory):
- self.vm.memory = self.init_mem.value()
- curr_maxmem = int(getattr(self.vm, 'maxmem', 0))
- if not self.include_in_balancing.isChecked():
- maxmem = 0
- else:
- maxmem = self.max_mem_size.value()
- if maxmem != curr_maxmem:
- if curr_maxmem > 0:
- self.vm.features['qubesmanager.maxmem_value'] = curr_maxmem
- if maxmem == 0 or self.max_mem_size.isEnabled():
- self.vm.maxmem = maxmem
- if self.vcpus.isEnabled() and \
- self.vcpus.value() != int(self.vm.vcpus):
- self.vm.vcpus = self.vcpus.value()
- except qubesadmin.exc.QubesException as ex:
- msg.append(str(ex))
- # in case VM is not Linux
- if hasattr(self.vm, "kernel") and self.kernel_groupbox.isVisible():
- try:
- if utils.did_widget_selection_change(self.kernel):
- self.vm.kernel = self.kernel.currentData()
- except qubesadmin.exc.QubesException as ex:
- msg.append(str(ex))
- # vm default_dispvm changed
- try:
- if utils.did_widget_selection_change(self.default_dispvm):
- self.vm.default_dispvm = self.default_dispvm.currentData()
- except qubesadmin.exc.QubesException as ex:
- msg.append(str(ex))
- try:
- if utils.did_widget_selection_change(self.virt_mode):
- self.vm.virt_mode = self.virt_mode.currentData()
- except Exception as ex: # pylint: disable=broad-except
- msg.append(str(ex))
- if getattr(self.vm, "template_for_dispvms", False) != \
- self.dvm_template_checkbox.isChecked():
- try:
- self.vm.template_for_dispvms = \
- self.dvm_template_checkbox.isChecked()
- if self.dvm_template_checkbox.isChecked():
- self.vm.features["appmenus-dispvm"] = True
- else:
- del self.vm.features["appmenus-dispvm"]
- except Exception as ex: # pylint: disable=broad-except
- msg.append(str(ex))
- if getattr(self.vm, 'provides_network', False) != \
- self.provides_network_checkbox.isChecked():
- try:
- self.vm.provides_network = \
- self.provides_network_checkbox.isChecked()
- except Exception as ex: # pylint: disable=broad-except
- msg.append(str(ex))
- # run_in_debug_mode
- try:
- if self.run_in_debug_mode.isVisible():
- if self.vm.debug != self.run_in_debug_mode.isChecked():
- self.vm.debug = self.run_in_debug_mode.isChecked()
- except qubesadmin.exc.QubesException as ex:
- msg.append(str(ex))
- if self.allow_fullscreen_initial !=\
- self.allow_fullscreen.currentIndex():
- try:
- if self.allow_fullscreen.currentData() is None:
- del self.vm.features['gui-allow-fullscreen']
- else:
- self.vm.features['gui-allow-fullscreen'] = \
- self.allow_fullscreen.currentData()
- except qubesadmin.exc.QubesException as ex:
- msg.append(str(ex))
- if self.allow_utf8_initial !=\
- self.allow_utf8.currentIndex():
- try:
- if self.allow_utf8.currentData() is None:
- del self.vm.features['gui-allow-utf8-titles']
- else:
- self.vm.features['gui-allow-utf8-titles'] = \
- self.allow_utf8.currentData()
- except qubesadmin.exc.QubesException as ex:
- msg.append(str(ex))
- return msg
- def include_in_balancing_changed(self, state):
- if self.dev_list.selected_list.count() > 0:
- if state == ui_settingsdlg.QtCore.Qt.Checked:
- self.dmm_warning_adv.show()
- self.dmm_warning_dev.show()
- else:
- self.dmm_warning_adv.hide()
- self.dmm_warning_dev.hide()
- self.max_mem_size.setEnabled(self.include_in_balancing.isChecked())
- if self.include_in_balancing.isChecked():
- self.check_mem_changes()
- def boot_from_cdrom_button_pressed(self):
- boot_dialog = bootfromdevice.VMBootFromDeviceWindow(
- self.vm.name, self.qapp, self.qubesapp, self)
- if boot_dialog.exec_():
- self.save_and_apply()
- qvm_start.main(
- ['--cdrom', boot_dialog.cdrom_location, self.vm.name])
- def virt_mode_changed(self, new_idx): # pylint: disable=unused-argument
- self.update_pv_warning()
- self.update_pvh_dont_support_devs()
- self.update_pvh_kernel_ver_warning()
- def update_pv_warning(self):
- if self.virt_mode.currentData() == 'pv':
- self.pv_warning.show()
- else:
- self.pv_warning.hide()
- def update_virt_mode_list(self):
- choices = [('HVM', 'hvm'),
- ('PV', 'pv')]
- if hasattr(self, "dev_list"):
- devs_attached = self.dev_list.selected_list.count() != 0
- else:
- try:
- devs_attached = bool(list(self.vm.devices['pci'].persistent()))
- except qubesadmin.exc.QubesException:
- devs_attached = False
- if devs_attached:
- self.pvh_mode_hidden.show()
- else:
- choices.insert(0, ('PVH', 'pvh'))
- self.pvh_mode_hidden.hide()
- old_mode = self.virt_mode.currentData()
- if old_mode:
- self.virt_mode.currentIndexChanged.disconnect()
- # due to how virtualization mode has uniquely different displayed and
- # actual name of the default value, I will add it manually
- try:
- choices.insert(0, (
- "default ({})".format(
- self.vm.property_get_default('virt_mode').upper()),
- qubesadmin.DEFAULT))
- except qubesadmin.exc.QubesException:
- choices.insert(0,
- ("default ({SYSTEM DEFAULT})", qubesadmin.DEFAULT))
- try:
- utils.initialize_widget_for_property(
- widget=self.virt_mode,
- choices=choices,
- holder=self.vm,
- property_name='virt_mode')
- except qubesadmin.exc.QubesDaemonAccessError:
- self.virt_mode.setEnabled(False)
- if self.virt_mode.isEnabled() and old_mode is not None:
- self.virt_mode.setCurrentIndex(self.virt_mode.findData(old_mode))
- self.virt_mode.currentIndexChanged.connect(self.virt_mode_changed)
- self.update_pv_warning()
- self.update_pvh_kernel_ver_warning()
- def update_pvh_kernel_ver_warning(self):
- if self.virt_mode.currentData() != 'pvh':
- self.pvh_kernel_version_warning.hide()
- return
- kernel = self.kernel.currentData()
- if self.pvh_kernel_version_ok(kernel):
- self.pvh_kernel_version_warning.hide()
- else:
- self.pvh_kernel_version_warning.show()
- def kernel_changed(self):
- self.update_pvh_kernel_ver_warning()
- def pvh_kernel_version_ok(self, name):
- # There are nearly no limitaions on kernel names (only file system and
- # general qvm-prefs rules). So we just look if we see something which
- # looks like a version number. It's just a warning to help the user
- # anyways.
- if name is None:
- return False
- if name is qubesadmin.DEFAULT:
- name = getattr(self.vm.app, 'default_kernel', None)
- m = re.search(r'(\d+)\.(\d+)', name)
- if m is None:
- return False
- return (int(m.group(1)), int(m.group(2))) >= (4, 11)
- ######## devices tab
- def __init_devices_tab__(self):
- self.dev_list = multiselectwidget.MultiSelectWidget(self)
- self.dev_list.add_all_button.setVisible(False)
- self.devices_layout.addWidget(self.dev_list)
- try:
- dom0_devs = \
- list(self.vm.app.domains['dom0'].devices['pci'].available())
- attached_devs = list(self.vm.devices['pci'].persistent())
- except qubesadmin.exc.QubesException:
- # no permission to access devices
- self.tabWidget.setTabEnabled(self.tabs_indices['devices'], False)
- return
- # pylint: disable=too-few-public-methods
- class DevListWidgetItem(QtWidgets.QListWidgetItem):
- def __init__(self, dev, unknown=False, parent=None):
- super().__init__(parent)
- name = dev.ident.replace('_', ":") + ' ' + dev.description
- if unknown:
- name += ' (unknown)'
- self.setText(name)
- self.dev = dev
- for dev in dom0_devs:
- if dev in attached_devs:
- self.dev_list.selected_list.addItem(DevListWidgetItem(dev))
- else:
- self.dev_list.available_list.addItem(DevListWidgetItem(dev))
- for dev in attached_devs:
- if dev not in dom0_devs:
- self.dev_list.selected_list.addItem(
- DevListWidgetItem(dev, unknown=True))
- if self.dev_list.selected_list.count() > 0\
- and self.include_in_balancing.isChecked():
- self.dmm_warning_adv.show()
- self.dmm_warning_dev.show()
- else:
- self.dmm_warning_adv.hide()
- self.dmm_warning_dev.hide()
- if utils.is_running(self.vm, False):
- self.dev_list.setEnabled(False)
- self.turn_off_vm_to_modify_devs.setVisible(True)
- self.no_strict_reset_button.setEnabled(False)
- else:
- self.dev_list.setEnabled(True)
- self.turn_off_vm_to_modify_devs.setVisible(False)
- self.update_pvh_dont_support_devs()
- self.dev_list.setEnabled(not utils.is_running(self.vm, False))
- def __apply_devices_tab__(self):
- msg = []
- if not self.tabWidget.isTabEnabled(self.tabs_indices['devices']):
- return msg
- try:
- old_devs = list(self.vm.devices['pci'].persistent())
- new_devs = [self.dev_list.selected_list.item(i).dev
- for i in range(self.dev_list.selected_list.count())]
- for dev in new_devs:
- if dev not in old_devs:
- options = {}
- if dev.ident in self.new_strict_reset_list:
- options['no-strict-reset'] = True
- ass = devices.DeviceAssignment(
- self.vm.app.domains['dom0'],
- dev.ident, persistent=True, options=options)
- self.vm.devices['pci'].attach(ass)
- elif (dev.ident in self.current_strict_reset_list) != \
- (dev.ident in self.new_strict_reset_list):
- current_assignment = None
- for assignment in self.vm.devices['pci'].assignments(
- persistent=True):
- if assignment.ident == dev.ident:
- current_assignment = assignment
- break
- if current_assignment is None:
- # it would be very weird if this happened
- msg.append(self.tr("Error re-assigning device ") +
- dev.ident)
- continue
- self.vm.devices['pci'].detach(current_assignment)
- current_assignment.options['no-strict-reset'] = \
- (dev.ident in self.new_strict_reset_list)
- self.vm.devices['pci'].attach(current_assignment)
- for ass in self.vm.devices['pci'].assignments(persistent=True):
- if ass.device not in new_devs:
- self.vm.devices['pci'].detach(ass)
- except qubesadmin.exc.QubesException as ex:
- if utils.is_debug():
- traceback.print_exc()
- msg.append(str(ex))
- return msg
- def devices_selection_changed(self):
- if self.include_in_balancing.isChecked():
- if self.dev_list.selected_list.count() > 0:
- self.dmm_warning_adv.show()
- self.dmm_warning_dev.show()
- else:
- self.dmm_warning_adv.hide()
- self.dmm_warning_dev.hide()
- self.update_virt_mode_list()
- def update_pvh_dont_support_devs(self):
- # this is the easiest way to check for both normal 'PVH' and
- # default (PVH) options
- if 'PVH' in self.virt_mode.currentText().upper():
- self.dev_list.setEnabled(False)
- self.pvh_dont_support_devs.setVisible(True)
- else:
- self.dev_list.setEnabled(True)
- self.pvh_dont_support_devs.setVisible(False)
- def define_strict_reset_devices(self):
- for assignment in self.vm.devices['pci'].assignments():
- if assignment.options.get('no-strict-reset', False):
- self.current_strict_reset_list.append(
- assignment.ident.replace('_', ':'))
- self.new_strict_reset_list = self.current_strict_reset_list.copy()
- def strict_reset_button_pressed(self):
- device_list_window = device_list.PCIDeviceListWindow(
- self.vm, self.qapp, self.dev_list, self.new_strict_reset_list, self)
- device_list_window.exec()
- ######## applications tab
- def refresh_apps_button_pressed(self):
- self.refresh_apps_button.setEnabled(False)
- self.refresh_apps_button.setText(self.tr('Refresh in progress...'))
- thread = RefreshAppsVMThread(self.vm, self.refresh_apps_button)
- thread.finished.connect(self.clear_threads)
- thread.finished.connect(self.refresh_finished)
- self.threads_list.append(thread)
- thread.start()
- def refresh_finished(self):
- self.app_list_manager = AppmenuSelectManager(self.vm, self.app_list)
- self.refresh_apps_button.setEnabled(True)
- self.refresh_apps_button.setText(self.tr('Refresh Applications'))
- def template_apps_change(self):
- if self.tabWidget.isTabEnabled(self.tabs_indices["applications"]):
- self.app_list_manager.fill_apps_list(
- template=self.template_name.currentData())
- # add a label to show
- self.warn_template_missing_apps.setVisible(
- self.app_list_manager.has_missing)
- ######## services tab
- def __init_services_tab__(self):
- self.new_srv_dict = {}
- try:
- for feature in self.vm.features:
- if not feature.startswith('service.'):
- continue
- service = feature[len('service.'):]
- item = QtWidgets.QListWidgetItem(service)
- item.setCheckState(ui_settingsdlg.QtCore.Qt.Checked
- if self.vm.features[feature]
- else ui_settingsdlg.QtCore.Qt.Unchecked)
- self.services_list.addItem(item)
- self.new_srv_dict[service] = self.vm.features[feature]
- except qubesadmin.exc.QubesDaemonAccessError:
- self.tabWidget.setTabEnabled(self.tabs_indices["services"], False)
- return
- self.service_line_edit.addItem("")
- supported_services = set()
- service_prefix = "supported-service."
- for feature in self.vm.features:
- if feature.startswith(service_prefix):
- supported_services.add(feature[len(service_prefix):])
- if getattr(self.vm, "template", None):
- try:
- for feature in self.vm.template.features:
- if feature.startswith(service_prefix):
- supported_services.add(feature[len(service_prefix):])
- except qubesadmin.exc.QubesDaemonAccessError:
- pass
- for service in sorted(supported_services):
- self.service_line_edit.addItem(service)
- self.service_line_edit.addItem(self.tr('(custom...)'))
- self.service_line_edit.setEditText("")
- def __add_service__(self):
- srv = str(self.service_line_edit.currentText()).strip()
- if srv != "":
- if self.service_line_edit.currentIndex() == \
- len(self.service_line_edit) - 1:
- (custom_name, ok) = QtWidgets.QInputDialog.getText(
- self, self.tr("Custom service name"),
- self.tr(
- "Name of the service:"))
- if ok:
- srv = custom_name.strip()
- else:
- return
- if srv in self.new_srv_dict:
- QtWidgets.QMessageBox.information(
- self,
- '',
- self.tr('Service already on the list!'))
- return
- item = QtWidgets.QListWidgetItem(srv)
- item.setCheckState(ui_settingsdlg.QtCore.Qt.Checked)
- self.services_list.addItem(item)
- self.new_srv_dict[srv] = True
- def __remove_service__(self):
- item = self.services_list.currentItem()
- if not item:
- return
- row = self.services_list.currentRow()
- item = self.services_list.takeItem(row)
- del self.new_srv_dict[str(item.text())]
- def __apply_services_tab__(self):
- msg = []
- if not self.tabWidget.isTabEnabled(self.tabs_indices['services']):
- return msg
- try:
- for i in range(self.services_list.count()):
- item = self.services_list.item(i)
- self.new_srv_dict[str(item.text())] = \
- (item.checkState() == ui_settingsdlg.QtCore.Qt.Checked)
- for service, v in self.new_srv_dict.items():
- feature = 'service.' + service
- if v != self.vm.features.get(feature, object()):
- self.vm.features[feature] = v
- for feature in self.vm.features:
- if not feature.startswith('service.'):
- continue
- service = feature[len('service.'):]
- if service not in self.new_srv_dict:
- del self.vm.features[feature]
- except qubesadmin.exc.QubesException as ex:
- msg.append(str(ex))
- return msg
- ######### firewall tab related
- def set_fw_model(self, model):
- self.fw_model = model
- self.rulesTreeView.setModel(model)
- self.rulesTreeView.header().setSectionResizeMode(
- QtWidgets.QHeaderView.ResizeToContents)
- self.rulesTreeView.header().setSectionResizeMode(
- 0, QtWidgets.QHeaderView.Stretch)
- self.set_allow(model.allow)
- if model.temp_full_access_expire_time:
- self.temp_full_access.setChecked(True)
- self.temp_full_access_time.setValue(
- (model.temp_full_access_expire_time -
- int(firewall.datetime.datetime.now().strftime("%s"))) / 60)
- def disable_all_fw_conf(self):
- self.firewall_modified_outside_label.setVisible(True)
- self.policy_allow_radio_button.setEnabled(False)
- self.policy_deny_radio_button.setEnabled(False)
- self.rulesTreeView.setEnabled(False)
- self.new_rule_button.setEnabled(False)
- self.edit_rule_button.setEnabled(False)
- self.delete_rule_button.setEnabled(False)
- self.firewal_rules_label.setEnabled(False)
- self.tempFullAccessWidget.setEnabled(False)
- def set_allow(self, allow):
- self.policy_allow_radio_button.setChecked(allow)
- self.policy_deny_radio_button.setChecked(not allow)
- self.policy_changed()
- def policy_changed(self):
- self.rulesTreeView.setEnabled(
- self.policy_deny_radio_button.isChecked())
- self.new_rule_button.setEnabled(
- self.policy_deny_radio_button.isChecked())
- self.edit_rule_button.setEnabled(
- self.policy_deny_radio_button.isChecked())
- self.delete_rule_button.setEnabled(
- self.policy_deny_radio_button.isChecked())
- self.firewal_rules_label.setEnabled(
- self.policy_deny_radio_button.isChecked())
- self.tempFullAccessWidget.setEnabled(
- self.policy_deny_radio_button.isChecked())
- def new_rule_button_pressed(self):
- dialog = firewall.NewFwRuleDlg()
- self.fw_model.run_rule_dialog(dialog)
- def edit_rule_button_pressed(self):
- selected = self.rulesTreeView.selectedIndexes()
- if selected:
- dialog = firewall.NewFwRuleDlg()
- dialog.set_ok_state(True)
- row = self.rulesTreeView.selectedIndexes().pop().row()
- self.fw_model.populate_edit_dialog(dialog, row)
- self.fw_model.run_rule_dialog(dialog, row)
- def delete_rule_button_pressed(self):
- for i in {index.row() for index
- in self.rulesTreeView.selectedIndexes()}:
- self.fw_model.remove_child(i)
- parser = QubesArgumentParser(vmname_nargs=1)
- parser.add_argument('--tab', metavar='TAB',
- action='store',
- choices=VMSettingsWindow.tabs_indices.keys())
- parser.set_defaults(
- tab='basic',
- )
- def main(args=None):
- args = parser.parse_args(args)
- vm = args.domains.pop()
- utils.run_synchronous(functools.partial(VMSettingsWindow, vm, args.tab))
- if __name__ == "__main__":
- main()
- # vim:sw=4:et:
|