settings.py 49 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320
  1. #!/usr/bin/python3
  2. #
  3. # The Qubes OS Project, http://www.qubes-os.org
  4. #
  5. # Copyright (C) 2012 Agnieszka Kostrzewa <agnieszka.kostrzewa@gmail.com>
  6. # Copyright (C) 2012 Marek Marczykowski-Górecki
  7. # <marmarek@invisiblethingslab.com>
  8. # Copyright (C) 2017 Wojtek Porczyk <woju@invisiblethingslab.com>
  9. #
  10. # This program is free software; you can redistribute it and/or
  11. # modify it under the terms of the GNU General Public License
  12. # as published by the Free Software Foundation; either version 2
  13. # of the License, or (at your option) any later version.
  14. #
  15. # This program is distributed in the hope that it will be useful,
  16. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  17. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  18. # GNU General Public License for more details.
  19. #
  20. # You should have received a copy of the GNU Lesser General Public License along
  21. # with this program; if not, see <http://www.gnu.org/licenses/>.
  22. #
  23. #
  24. import collections
  25. import functools
  26. import re
  27. import subprocess
  28. import traceback
  29. from qubesadmin.tools import QubesArgumentParser
  30. from qubesadmin import devices
  31. from qubesadmin import utils as admin_utils
  32. import qubesadmin.exc
  33. from . import utils
  34. from . import multiselectwidget
  35. from . import common_threads
  36. from . import device_list
  37. from .appmenu_select import AppmenuSelectManager
  38. from . import firewall
  39. from PyQt5 import QtCore, QtWidgets, QtGui # pylint: disable=import-error
  40. from . import ui_settingsdlg # pylint: disable=no-name-in-module
  41. # pylint: disable=too-few-public-methods
  42. class RenameVMThread(common_threads.QubesThread):
  43. def __init__(self, vm, new_vm_name, dependencies):
  44. super(RenameVMThread, self).__init__(vm)
  45. self.new_vm_name = new_vm_name
  46. self.dependencies = dependencies
  47. def run(self):
  48. try:
  49. new_vm = self.vm.app.clone_vm(self.vm, self.new_vm_name)
  50. failed_props = []
  51. for (holder, prop) in self.dependencies:
  52. try:
  53. if holder is None:
  54. setattr(self.vm.app, prop, new_vm)
  55. else:
  56. setattr(holder, prop, new_vm)
  57. except qubesadmin.exc.QubesException:
  58. failed_props += [(holder, prop)]
  59. if not failed_props:
  60. del self.vm.app.domains[self.vm.name]
  61. else:
  62. list_text = utils.format_dependencies_list(failed_props)
  63. self.msg = (self.tr("Warning: rename partially unsuccessful!"),
  64. self.tr("Some properties could not be changed to "
  65. "the new name. The system has now both {} "
  66. "and {} qubes. To resolve this, please "
  67. "check and change the following properties "
  68. "and remove the qube {} manually.<br>"
  69. ).format(self.vm.name, self.vm.name,
  70. self.vm.name) + list_text)
  71. except qubesadmin.exc.QubesException as ex:
  72. self.msg = (self.tr("Rename error!"), str(ex))
  73. except Exception as ex: # pylint: disable=broad-except
  74. self.msg = (self.tr("Rename error!"), repr(ex))
  75. # pylint: disable=too-few-public-methods
  76. class RefreshAppsVMThread(common_threads.QubesThread):
  77. def __init__(self, vm, button):
  78. super(RefreshAppsVMThread, self).__init__(vm)
  79. self.button = button
  80. def run(self):
  81. vms_to_refresh = [self.vm]
  82. template = getattr(self.vm, 'template', None)
  83. if template:
  84. vms_to_refresh.append(template)
  85. for vm in vms_to_refresh:
  86. self.button.setText(
  87. self.tr('Refresh in progress (refreshing applications '
  88. 'from {})').format(vm.name))
  89. try:
  90. if not vm.is_running():
  91. not_running = True
  92. vm.start()
  93. else:
  94. not_running = False
  95. subprocess.check_call(['qvm-sync-appmenus', vm.name])
  96. if not_running:
  97. vm.shutdown()
  98. except Exception as ex: # pylint: disable=broad-except
  99. self.msg = (self.tr("Refresh failed!"), str(ex))
  100. # pylint: disable=too-many-instance-attributes
  101. class VMSettingsWindow(ui_settingsdlg.Ui_SettingsDialog, QtWidgets.QDialog):
  102. tabs_indices = collections.OrderedDict((
  103. ('basic', 0),
  104. ('advanced', 1),
  105. ('firewall', 2),
  106. ('devices', 3),
  107. ('applications', 4),
  108. ('services', 5),
  109. ))
  110. def __init__(self, vm, init_page="basic", qapp=None, qubesapp=None,
  111. parent=None):
  112. super(VMSettingsWindow, self).__init__(parent)
  113. self.vm = vm
  114. self.qapp = qapp
  115. self.qubesapp = qubesapp
  116. self.threads_list = []
  117. self.progress = None
  118. self.thread_closes = False
  119. try:
  120. self.source_vm = self.vm.template
  121. except AttributeError:
  122. self.source_vm = self.vm
  123. self.setupUi(self)
  124. self.setWindowTitle(self.tr("Settings: {vm}").format(vm=self.vm.name))
  125. if init_page in self.tabs_indices:
  126. idx = self.tabs_indices[init_page]
  127. assert idx in range(self.tabWidget.count())
  128. self.tabWidget.setCurrentIndex(idx)
  129. self.buttonBox.button(QtWidgets.QDialogButtonBox.Apply).clicked.connect(
  130. self.apply)
  131. self.tabWidget.currentChanged.connect(self.current_tab_changed)
  132. ###### basic tab
  133. self.__init_basic_tab__()
  134. self.rename_vm_button.clicked.connect(self.rename_vm)
  135. self.delete_vm_button.clicked.connect(self.remove_vm)
  136. self.clone_vm_button.clicked.connect(self.clone_vm)
  137. ###### advanced tab
  138. self.__init_advanced_tab__()
  139. self.include_in_balancing.stateChanged.connect(
  140. self.include_in_balancing_changed)
  141. self.init_mem.editingFinished.connect(self.check_mem_changes)
  142. self.max_mem_size.editingFinished.connect(self.check_mem_changes)
  143. self.boot_from_device_button.clicked.connect(
  144. self.boot_from_cdrom_button_pressed)
  145. ###### firewall tab
  146. if self.tabWidget.isTabEnabled(self.tabs_indices['firewall']):
  147. model = firewall.QubesFirewallRulesModel()
  148. try:
  149. model.set_vm(vm)
  150. self.set_fw_model(model)
  151. self.firewall_modified_outside_label.setVisible(False)
  152. except firewall.FirewallModifiedOutsideError:
  153. self.disable_all_fw_conf()
  154. self.new_rule_button.clicked.connect(self.new_rule_button_pressed)
  155. self.edit_rule_button.clicked.connect(self.edit_rule_button_pressed)
  156. self.delete_rule_button.clicked.connect(
  157. self.delete_rule_button_pressed)
  158. self.policy_deny_radio_button.clicked.connect(self.policy_changed)
  159. self.policy_allow_radio_button.clicked.connect(self.policy_changed)
  160. if init_page == 'firewall':
  161. self.check_network_availability()
  162. ####### devices tab
  163. self.__init_devices_tab__()
  164. self.dev_list.selectedChanged.connect(self.devices_selection_changed)
  165. self.no_strict_reset_button.clicked.connect(
  166. self.strict_reset_button_pressed)
  167. self.current_strict_reset_list = []
  168. self.new_strict_reset_list = []
  169. self.define_strict_reset_devices()
  170. ####### services tab
  171. self.__init_services_tab__()
  172. self.add_srv_button.clicked.connect(self.__add_service__)
  173. self.remove_srv_button.clicked.connect(self.__remove_service__)
  174. ####### apps tab
  175. if self.tabWidget.isTabEnabled(self.tabs_indices["applications"]):
  176. self.app_list = multiselectwidget.MultiSelectWidget(self)
  177. self.apps_layout.addWidget(self.app_list)
  178. self.app_list_manager = AppmenuSelectManager(self.vm, self.app_list)
  179. self.refresh_apps_button.clicked.connect(
  180. self.refresh_apps_button_pressed)
  181. # template change
  182. if self.template_name.isEnabled():
  183. self.template_name.currentIndexChanged.connect(
  184. self.template_apps_change)
  185. self.warn_template_missing_apps.setVisible(
  186. self.app_list_manager.has_missing)
  187. def setup_application(self):
  188. self.qapp.setApplicationName(self.tr("Qube Settings"))
  189. self.qapp.setWindowIcon(QtGui.QIcon.fromTheme("qubes-manager"))
  190. def clear_threads(self):
  191. for thread in self.threads_list:
  192. if thread.isFinished():
  193. if self.progress:
  194. self.progress.hide()
  195. self.progress = None
  196. if thread.msg:
  197. (title, msg) = thread.msg
  198. QtWidgets.QMessageBox.warning(
  199. self,
  200. title,
  201. msg)
  202. self.threads_list.remove(thread)
  203. if self.thread_closes:
  204. self.done(0)
  205. return
  206. raise RuntimeError(self.tr('No finished thread found'))
  207. def keyPressEvent(self, event): # pylint: disable=invalid-name
  208. if event.key() == QtCore.Qt.Key_Enter \
  209. or event.key() == QtCore.Qt.Key_Return:
  210. return
  211. super(VMSettingsWindow, self).keyPressEvent(event)
  212. def accept(self):
  213. self.save_and_apply()
  214. def save_changes(self):
  215. with common_threads.busy_cursor():
  216. error = self.__save_changes__()
  217. if error:
  218. QtWidgets.QMessageBox.warning(
  219. self,
  220. self.tr("Error while changing settings for {0}!"
  221. ).format(self.vm.name),
  222. self.tr("ERROR: {0}").format('\n'.join(error)))
  223. def apply(self):
  224. self.save_changes()
  225. # these signals must be disconnected to avoid unintended behavior
  226. # on refreshing the drop-downs
  227. self.netVM.currentIndexChanged.disconnect()
  228. self.kernel.currentIndexChanged.disconnect()
  229. self.default_dispvm.currentIndexChanged.disconnect()
  230. self.__init_basic_tab__()
  231. self.__init_advanced_tab__()
  232. def save_and_apply(self):
  233. self.save_changes()
  234. self.done(0)
  235. def __save_changes__(self):
  236. ret = []
  237. try:
  238. ret_tmp = self.__apply_basic_tab__()
  239. if ret_tmp:
  240. ret += [self.tr("Basic tab:")] + ret_tmp
  241. ret_tmp = self.__apply_advanced_tab__()
  242. if ret_tmp:
  243. ret += [self.tr("Advanced tab:")] + ret_tmp
  244. ret_tmp = self.__apply_devices_tab__()
  245. if ret_tmp:
  246. ret += [self.tr("Devices tab:")] + ret_tmp
  247. ret_tmp = self.__apply_services_tab__()
  248. if ret_tmp:
  249. ret += [self.tr("Sevices tab:")] + ret_tmp
  250. except qubesadmin.exc.QubesException as qex:
  251. ret.append(self.tr('Error while saving changes: ') + str(qex))
  252. except Exception as ex: # pylint: disable=broad-except
  253. ret.append(repr(ex))
  254. try:
  255. if self.policy_allow_radio_button.isEnabled():
  256. self.fw_model.apply_rules(
  257. self.policy_allow_radio_button.isChecked(),
  258. self.temp_full_access.isChecked(),
  259. self.temp_full_access_time.value())
  260. except qubesadmin.exc.QubesException as qex:
  261. ret += [self.tr("Firewall tab:"), str(qex)]
  262. except Exception as ex: # pylint: disable=broad-except
  263. ret += [self.tr("Firewall tab:"), repr(ex)]
  264. try:
  265. if self.tabWidget.isTabEnabled(self.tabs_indices["applications"]):
  266. self.app_list_manager.save_appmenu_select_changes()
  267. except qubesadmin.exc.QubesException as qex:
  268. ret += [self.tr("Applications tab:"), str(qex)]
  269. except Exception as ex: # pylint: disable=broad-except
  270. ret += [self.tr("Applications tab:"), repr(ex)]
  271. utils.debug('\n'.join(ret))
  272. return ret
  273. def check_network_availability(self):
  274. netvm = self.vm.netvm
  275. try:
  276. provides_network = self.vm.provides_network
  277. except AttributeError:
  278. provides_network = False
  279. self.no_netvm_label.setVisible(netvm is None and not provides_network)
  280. self.netvm_no_firewall_label.setVisible(
  281. netvm is not None and
  282. not netvm.features.check_with_template('qubes-firewall', False))
  283. self.sysnet_warning_label.setVisible(netvm is None and provides_network)
  284. def current_tab_changed(self, idx):
  285. if idx == self.tabs_indices["firewall"]:
  286. self.check_network_availability()
  287. ######### basic tab
  288. # TODO LISTENERS
  289. # - vm start/shutdown -> setEnabled on fields: template labels
  290. # - vm create/delete -> choices lists, whole window deactiv (if self.vm)
  291. # - property-set -> individual fields
  292. # TODO INTERACTIONS
  293. # netvm -> networking_groupbox
  294. # hvm -> include_in_balancing
  295. def __init_basic_tab__(self):
  296. self.vmname.setText(self.vm.name)
  297. self.vmname.setValidator(
  298. QtGui.QRegExpValidator(
  299. QtCore.QRegExp("[a-zA-Z0-9_-]*",
  300. QtCore.Qt.CaseInsensitive), None))
  301. self.vmname.setEnabled(False)
  302. self.rename_vm_button.setEnabled(not self.vm.is_running())
  303. self.delete_vm_button.setEnabled(not self.vm.is_running())
  304. if self.vm.is_running():
  305. self.delete_vm_button.setText(
  306. self.tr('Delete qube (cannot delete a running qube)'))
  307. if self.vm.qid == 0:
  308. self.vmlabel.setVisible(False)
  309. else:
  310. utils.initialize_widget_with_labels(
  311. widget=self.vmlabel,
  312. qubes_app=self.qubesapp,
  313. holder=self.vm)
  314. self.vmlabel.setVisible(True)
  315. self.vmlabel.setEnabled(not self.vm.is_running())
  316. if self.vm.klass == 'AppVM':
  317. utils.initialize_widget_with_vms(
  318. widget=self.template_name,
  319. qubes_app=self.qubesapp,
  320. filter_function=(lambda vm: vm.klass == 'TemplateVM'),
  321. holder=self.vm,
  322. property_name='template')
  323. elif self.vm.klass == 'DispVM':
  324. utils.initialize_widget_with_vms(
  325. widget=self.template_name,
  326. qubes_app=self.qubesapp,
  327. filter_function=(lambda vm:
  328. getattr(vm, 'template_for_dispvms', False)),
  329. holder=self.vm,
  330. property_name='template')
  331. else:
  332. self.template_name.setEnabled(False)
  333. if self.vm.is_running():
  334. self.template_name.setEnabled(False)
  335. utils.initialize_widget_with_vms(
  336. widget=self.netVM,
  337. qubes_app=self.qubesapp,
  338. filter_function=(lambda vm: vm.provides_network),
  339. holder=self.vm,
  340. property_name='netvm',
  341. allow_default=True)
  342. self.netVM.currentIndexChanged.connect(self.check_warn_dispvmnetvm)
  343. self.include_in_backups.setChecked(self.vm.include_in_backups)
  344. try:
  345. self.autostart_vm.setChecked(self.vm.autostart)
  346. self.autostart_vm.setVisible(True)
  347. except AttributeError:
  348. self.autostart_vm.setVisible(False)
  349. # type
  350. self.type_label.setText(self.vm.klass)
  351. # installed by rpm
  352. self.rpm_label.setText('Yes' if self.vm.installed_by_rpm else 'No')
  353. # networking info
  354. if self.vm.netvm:
  355. self.networking_groupbox.setEnabled(True)
  356. self.ip_label.setText(self.vm.ip or "none")
  357. self.netmask_label.setText(self.vm.visible_netmask or "none")
  358. self.gateway_label.setText(self.vm.visible_gateway or "none")
  359. dns_list = getattr(self.vm, 'dns', ['10.139.1.1', '10.139.1.2'])
  360. self.dns_label.setText(", ".join(dns_list))
  361. else:
  362. self.networking_groupbox.setEnabled(False)
  363. # max priv storage
  364. self.priv_img_size = self.vm.volumes['private'].size // 1024**2
  365. self.max_priv_storage.setMinimum(self.priv_img_size)
  366. self.max_priv_storage.setValue(self.priv_img_size)
  367. self.max_priv_storage.setMaximum(
  368. max(self.priv_img_size,
  369. self.qubesapp.pools[self.vm.volumes['private'].pool].size
  370. // 1024**2))
  371. self.root_img_size = self.vm.volumes['root'].size // 1024**2
  372. self.root_resize.setValue(self.root_img_size)
  373. self.root_resize.setMinimum(self.root_img_size)
  374. self.root_resize.setMaximum(
  375. max(self.root_img_size,
  376. self.qubesapp.pools[self.vm.volumes['root'].pool].size
  377. // 1024**2))
  378. self.root_resize.setEnabled(self.vm.volumes['root'].save_on_stop)
  379. if not self.root_resize.isEnabled():
  380. self.root_resize.setToolTip(
  381. self.tr("To change system storage size, change properties "
  382. "of the underlying template."))
  383. self.root_resize_label.setEnabled(self.root_resize.isEnabled())
  384. self.warn_template_missing_apps.setVisible(False)
  385. def __apply_basic_tab__(self):
  386. msg = []
  387. # vm label changed
  388. try:
  389. if self.vmlabel.isVisible():
  390. if utils.did_widget_selection_change(self.vmlabel):
  391. self.vm.label = self.vmlabel.currentData()
  392. except qubesadmin.exc.QubesException as ex:
  393. msg.append(str(ex))
  394. # vm template changed
  395. try:
  396. if utils.did_widget_selection_change(self.template_name):
  397. self.vm.template = self.template_name.currentData()
  398. except qubesadmin.exc.QubesException as ex:
  399. msg.append(str(ex))
  400. # vm netvm changed
  401. try:
  402. if utils.did_widget_selection_change(self.netVM):
  403. self.vm.netvm = self.netVM.currentData()
  404. except qubesadmin.exc.QubesException as ex:
  405. msg.append(str(ex))
  406. # include in backups
  407. try:
  408. if self.vm.include_in_backups != \
  409. self.include_in_backups.isChecked():
  410. self.vm.include_in_backups = self.include_in_backups.isChecked()
  411. except qubesadmin.exc.QubesException as ex:
  412. msg.append(str(ex))
  413. # autostart_vm
  414. try:
  415. if self.autostart_vm.isVisible():
  416. if self.vm.autostart != self.autostart_vm.isChecked():
  417. self.vm.autostart = self.autostart_vm.isChecked()
  418. except qubesadmin.exc.QubesException as ex:
  419. msg.append(str(ex))
  420. # max priv storage
  421. priv_size = self.max_priv_storage.value()
  422. if self.priv_img_size != priv_size:
  423. try:
  424. self.vm.volumes['private'].resize(priv_size * 1024**2)
  425. self.priv_img_size = priv_size
  426. except qubesadmin.exc.QubesException as ex:
  427. msg.append(str(ex))
  428. # max sys storage
  429. sys_size = self.root_resize.value()
  430. if self.root_img_size != sys_size:
  431. try:
  432. self.vm.volumes['root'].resize(sys_size * 1024**2)
  433. self.root_img_size = sys_size
  434. except qubesadmin.exc.QubesException as ex:
  435. msg.append(str(ex))
  436. return msg
  437. def check_mem_changes(self):
  438. if not self.include_in_balancing.isChecked():
  439. # do not interfere with settings if the VM is not included in memory
  440. # balancing
  441. return
  442. if self.max_mem_size.value() < self.init_mem.value():
  443. QtWidgets.QMessageBox.warning(
  444. self,
  445. self.tr("Warning!"),
  446. self.tr("Max memory can not be less than initial memory.<br>"
  447. "Setting max memory to equal initial memory."))
  448. self.max_mem_size.setValue(self.init_mem.value())
  449. # Linux specific limit: init memory must not be below
  450. # max_mem_size/10.79 in order to allow scaling up to
  451. # max_mem_size (or else "add_memory() failed: -17" problem)
  452. if self.vm.features.check_with_template('os', None) == 'Linux' and \
  453. self.init_mem.value() * 10 < self.max_mem_size.value():
  454. self.init_mem.setValue((self.max_mem_size.value() + 9) // 10)
  455. QtWidgets.QMessageBox.warning(
  456. self,
  457. self.tr("Warning!"),
  458. self.tr("For Linux qubes, Initial memory can not be less than "
  459. "one tenth Max memory.<br>Setting initial memory "
  460. "to the minimum allowed value."))
  461. def check_warn_dispvmnetvm(self):
  462. if not hasattr(self.vm, 'default_dispvm'):
  463. self.warn_netvm_dispvm.setVisible(False)
  464. return
  465. dispvm = self.default_dispvm.currentData()
  466. own_netvm = self.netVM.currentData()
  467. if dispvm == qubesadmin.DEFAULT:
  468. dispvm = self.vm.property_get_default('default_dispvm')
  469. if dispvm == self.vm:
  470. self.warn_netvm_dispvm.setVisible(False)
  471. return
  472. dispvm_netvm = getattr(dispvm, 'netvm', None)
  473. if own_netvm == qubesadmin.DEFAULT:
  474. own_netvm = self.vm.property_get_default('netvm')
  475. if dispvm_netvm and dispvm_netvm != own_netvm:
  476. self.warn_netvm_dispvm.setVisible(True)
  477. else:
  478. self.warn_netvm_dispvm.setVisible(False)
  479. def rename_vm(self):
  480. dependencies = admin_utils.vm_dependencies(self.vm.app, self.vm)
  481. running_dependencies = [vm.name for (vm, prop) in dependencies
  482. if vm and prop == 'template'
  483. and vm.is_running()]
  484. if running_dependencies:
  485. QtWidgets.QMessageBox.warning(
  486. self,
  487. self.tr("Qube cannot be renamed!"),
  488. self.tr(
  489. "The following qubes using this qube as a template are "
  490. "running: <br> {}. <br> In order to rename this qube, you "
  491. "must first shut them down.").format(
  492. ", ".join(running_dependencies)))
  493. return
  494. new_vm_name, ok = QtWidgets.QInputDialog.getText(
  495. self,
  496. self.tr('Rename qube'),
  497. self.tr('New name: (WARNING: all other changes will be discarded)'),
  498. text=self.vm.name)
  499. if ok:
  500. thread = RenameVMThread(self.vm, new_vm_name, dependencies)
  501. self.threads_list.append(thread)
  502. thread.finished.connect(self.clear_threads)
  503. self.progress = QtWidgets.QProgressDialog(
  504. self.tr("Renaming Qube..."), "", 0, 0)
  505. self.progress.setCancelButton(None)
  506. self.progress.setModal(True)
  507. self.thread_closes = True
  508. self.progress.show()
  509. thread.start()
  510. def remove_vm(self):
  511. dependencies = admin_utils.vm_dependencies(self.vm.app, self.vm)
  512. if dependencies:
  513. list_text = utils.format_dependencies_list(dependencies)
  514. QtWidgets.QMessageBox.warning(
  515. self,
  516. self.tr("Qube cannot be removed!"),
  517. self.tr("This qube cannot be removed. It is used as:"
  518. " <br> {} <small>If you want to remove this qube, "
  519. "you should remove or change settings of each qube "
  520. "or setting that uses it.</small>").format(list_text))
  521. return
  522. answer, ok = QtWidgets.QInputDialog.getText(
  523. self,
  524. self.tr('Delete qube'),
  525. self.tr('Are you absolutely sure you want to delete this qube? '
  526. '<br/> All qube settings and data will be irrevocably'
  527. ' deleted. <br/> If you are sure, please enter this '
  528. 'qube\'s name below.'))
  529. if ok and answer == self.vm.name:
  530. thread = common_threads.RemoveVMThread(self.vm)
  531. thread.finished.connect(self.clear_threads)
  532. self.threads_list.append(thread)
  533. self.progress = QtWidgets.QProgressDialog(
  534. self.tr("Deleting Qube..."), "", 0, 0)
  535. self.progress.setCancelButton(None)
  536. self.progress.setModal(True)
  537. self.thread_closes = True
  538. self.progress.show()
  539. thread.start()
  540. elif ok:
  541. QtWidgets.QMessageBox.warning(
  542. self,
  543. self.tr("Removal cancelled"),
  544. self.tr("The qube will not be removed."))
  545. def clone_vm(self):
  546. cloned_vm_name, ok = QtWidgets.QInputDialog.getText(
  547. self,
  548. self.tr('Clone qube'),
  549. self.tr('Name for the cloned qube:'))
  550. if ok:
  551. thread = common_threads.CloneVMThread(self.vm, cloned_vm_name)
  552. thread.finished.connect(self.clear_threads)
  553. self.threads_list.append(thread)
  554. self.progress = QtWidgets.QProgressDialog(
  555. self.tr("Cloning Qube..."), "", 0, 0)
  556. self.progress.setCancelButton(None)
  557. self.progress.setModal(True)
  558. self.thread_closes = True
  559. self.progress.show()
  560. # TODO: maybe this can not be repeated all the time?
  561. thread.start()
  562. ######### advanced tab
  563. def __init_advanced_tab__(self):
  564. self.init_mem.setValue(int(self.vm.memory))
  565. if self.vm.maxmem > 0:
  566. self.max_mem_size.setValue(int(self.vm.maxmem))
  567. else:
  568. maxmem = self.vm.property_get_default('maxmem')
  569. if maxmem == 0:
  570. maxmem = self.vm.memory
  571. self.max_mem_size.setValue(int(
  572. self.vm.features.get('qubesmanager.maxmem_value', maxmem)))
  573. self.vcpus.setMinimum(1)
  574. self.vcpus.setValue(int(self.vm.vcpus))
  575. self.include_in_balancing.setEnabled(True)
  576. self.include_in_balancing.setChecked(int(self.vm.maxmem) > 0)
  577. self.max_mem_size.setEnabled(self.include_in_balancing.isChecked())
  578. # in case VM is HVM
  579. if hasattr(self.vm, "kernel"):
  580. self.kernel_groupbox.setVisible(True)
  581. utils.initialize_widget_with_kernels(
  582. widget=self.kernel,
  583. qubes_app=self.qubesapp,
  584. allow_none=True,
  585. allow_default=True,
  586. holder=self.vm,
  587. property_name='kernel')
  588. self.kernel.currentIndexChanged.connect(self.kernel_changed)
  589. self.kernel_opts.setText(getattr(self.vm, 'kernelopts', '-'))
  590. else:
  591. self.kernel_groupbox.setVisible(False)
  592. self.other_groupbox.setVisible(False)
  593. if not hasattr(self.vm, 'default_dispvm'):
  594. self.other_groupbox.setVisible(False)
  595. else:
  596. self.other_groupbox.setVisible(True)
  597. utils.initialize_widget_with_vms(
  598. widget=self.default_dispvm,
  599. qubes_app=self.qubesapp,
  600. filter_function=(lambda vm:
  601. getattr(vm, 'template_for_dispvms', False)),
  602. allow_none=True,
  603. allow_default=True,
  604. holder=self.vm,
  605. property_name='default_dispvm'
  606. )
  607. self.default_dispvm.currentIndexChanged.connect(
  608. self.check_warn_dispvmnetvm)
  609. self.check_warn_dispvmnetvm()
  610. self.update_virt_mode_list()
  611. windows_running = \
  612. self.vm.features.check_with_template('os', None) == 'Windows' \
  613. and self.vm.is_running()
  614. self.seamless_on_button.setEnabled(windows_running)
  615. self.seamless_off_button.setEnabled(windows_running)
  616. self.seamless_on_button.clicked.connect(self.enable_seamless)
  617. self.seamless_off_button.clicked.connect(self.disable_seamless)
  618. if hasattr(self.vm, "template_for_dispvms"):
  619. self.dvm_template_checkbox.setChecked(self.vm.template_for_dispvms)
  620. else:
  621. self.dvm_template_checkbox.setVisible(False)
  622. self.provides_network_checkbox.setChecked(
  623. getattr(self.vm, 'provides_network', False))
  624. if self.provides_network_checkbox.isChecked():
  625. domains_using = [vm.name for vm in self.vm.connected_vms]
  626. if domains_using:
  627. self.provides_network_checkbox.setEnabled(False)
  628. self.provides_network_checkbox.setToolTip(self.tr(
  629. "Cannot change this setting while this qube is used as a "
  630. "NetVM by the following qubes:\n") +
  631. "\n".join(domains_using))
  632. try:
  633. self.run_in_debug_mode.setChecked(self.vm.debug)
  634. self.run_in_debug_mode.setVisible(True)
  635. except AttributeError:
  636. self.run_in_debug_mode.setVisible(False)
  637. utils.initialize_widget(
  638. widget=self.allow_fullscreen,
  639. choices=[
  640. ('(use system default)', None),
  641. ('allow', True),
  642. ('disallow', False)
  643. ],
  644. selected_value=utils.get_boolean_feature(self.vm,
  645. 'gui-allow-fullscreen'))
  646. self.allow_fullscreen_initial = self.allow_fullscreen.currentIndex()
  647. utils.initialize_widget(
  648. widget=self.allow_utf8,
  649. choices=[
  650. ('(use system default)', None),
  651. ('allow', True),
  652. ('disallow', False)
  653. ],
  654. selected_value=utils.get_boolean_feature(self.vm,
  655. 'gui-allow-utf8-titles'))
  656. self.allow_utf8_initial = self.allow_utf8.currentIndex()
  657. def enable_seamless(self):
  658. self.vm.run_service_for_stdio("qubes.SetGuiMode", input=b'SEAMLESS')
  659. def disable_seamless(self):
  660. self.vm.run_service_for_stdio("qubes.SetGuiMode", input=b'FULLSCREEN')
  661. def __apply_advanced_tab__(self):
  662. msg = []
  663. # mem/cpu
  664. try:
  665. if self.init_mem.value() != int(self.vm.memory):
  666. self.vm.memory = self.init_mem.value()
  667. curr_maxmem = int(self.vm.maxmem)
  668. if not self.include_in_balancing.isChecked():
  669. maxmem = 0
  670. else:
  671. maxmem = self.max_mem_size.value()
  672. if maxmem != curr_maxmem:
  673. if curr_maxmem > 0:
  674. self.vm.features['qubesmanager.maxmem_value'] = curr_maxmem
  675. self.vm.maxmem = maxmem
  676. if self.vcpus.value() != int(self.vm.vcpus):
  677. self.vm.vcpus = self.vcpus.value()
  678. except qubesadmin.exc.QubesException as ex:
  679. msg.append(str(ex))
  680. # in case VM is not Linux
  681. if hasattr(self.vm, "kernel") and self.kernel_groupbox.isVisible():
  682. try:
  683. if utils.did_widget_selection_change(self.kernel):
  684. self.vm.kernel = self.kernel.currentData()
  685. except qubesadmin.exc.QubesException as ex:
  686. msg.append(str(ex))
  687. # vm default_dispvm changed
  688. try:
  689. if utils.did_widget_selection_change(self.default_dispvm):
  690. self.vm.default_dispvm = self.default_dispvm.currentData()
  691. except qubesadmin.exc.QubesException as ex:
  692. msg.append(str(ex))
  693. try:
  694. if utils.did_widget_selection_change(self.virt_mode):
  695. self.vm.virt_mode = self.virt_mode.currentData()
  696. except Exception as ex: # pylint: disable=broad-except
  697. msg.append(str(ex))
  698. if getattr(self.vm, "template_for_dispvms", False) != \
  699. self.dvm_template_checkbox.isChecked():
  700. try:
  701. self.vm.template_for_dispvms = \
  702. self.dvm_template_checkbox.isChecked()
  703. if self.dvm_template_checkbox.isChecked():
  704. self.vm.features["appmenus-dispvm"] = True
  705. else:
  706. del self.vm.features["appmenus-dispvm"]
  707. except Exception as ex: # pylint: disable=broad-except
  708. msg.append(str(ex))
  709. if getattr(self.vm, 'provides_network', False) != \
  710. self.provides_network_checkbox.isChecked():
  711. try:
  712. self.vm.provides_network = \
  713. self.provides_network_checkbox.isChecked()
  714. except Exception as ex: # pylint: disable=broad-except
  715. msg.append(str(ex))
  716. # run_in_debug_mode
  717. try:
  718. if self.run_in_debug_mode.isVisible():
  719. if self.vm.debug != self.run_in_debug_mode.isChecked():
  720. self.vm.debug = self.run_in_debug_mode.isChecked()
  721. except qubesadmin.exc.QubesException as ex:
  722. msg.append(str(ex))
  723. if self.allow_fullscreen_initial !=\
  724. self.allow_fullscreen.currentIndex():
  725. try:
  726. if self.allow_fullscreen.currentData() is None:
  727. del self.vm.features['gui-allow-fullscreen']
  728. else:
  729. self.vm.features['gui-allow-fullscreen'] = \
  730. self.allow_fullscreen.currentData()
  731. except qubesadmin.exc.QubesException as ex:
  732. msg.append(str(ex))
  733. if self.allow_utf8_initial !=\
  734. self.allow_utf8.currentIndex():
  735. try:
  736. if self.allow_utf8.currentData() is None:
  737. del self.vm.features['gui-allow-utf8-titles']
  738. else:
  739. self.vm.features['gui-allow-utf8-titles'] = \
  740. self.allow_utf8.currentData()
  741. except qubesadmin.exc.QubesException as ex:
  742. msg.append(str(ex))
  743. return msg
  744. def include_in_balancing_changed(self, state):
  745. if self.dev_list.selected_list.count() > 0:
  746. if state == ui_settingsdlg.QtCore.Qt.Checked:
  747. self.dmm_warning_adv.show()
  748. self.dmm_warning_dev.show()
  749. else:
  750. self.dmm_warning_adv.hide()
  751. self.dmm_warning_dev.hide()
  752. self.max_mem_size.setEnabled(self.include_in_balancing.isChecked())
  753. if self.include_in_balancing.isChecked():
  754. self.check_mem_changes()
  755. def boot_from_cdrom_button_pressed(self):
  756. self.save_and_apply()
  757. subprocess.check_call(['qubes-vm-boot-from-device', self.vm.name])
  758. def virt_mode_changed(self, new_idx): # pylint: disable=unused-argument
  759. self.update_pv_warning()
  760. self.update_pvh_dont_support_devs()
  761. self.update_pvh_kernel_ver_warning()
  762. def update_pv_warning(self):
  763. if self.virt_mode.currentData() == 'pv':
  764. self.pv_warning.show()
  765. else:
  766. self.pv_warning.hide()
  767. def update_virt_mode_list(self):
  768. choices = [('HVM', 'hvm'),
  769. ('PV', 'pv')]
  770. if hasattr(self, "dev_list"):
  771. devs_attached = self.dev_list.selected_list.count() != 0
  772. else:
  773. devs_attached = bool(list(self.vm.devices['pci'].persistent()))
  774. if devs_attached:
  775. self.pvh_mode_hidden.show()
  776. else:
  777. choices.insert(0, ('PVH', 'pvh'))
  778. self.pvh_mode_hidden.hide()
  779. old_mode = self.virt_mode.currentData()
  780. if old_mode:
  781. self.virt_mode.currentIndexChanged.disconnect()
  782. utils.initialize_widget_for_property(
  783. widget=self.virt_mode,
  784. choices=choices,
  785. holder=self.vm,
  786. property_name='virt_mode',
  787. allow_default=True)
  788. if old_mode is not None:
  789. self.virt_mode.setCurrentIndex(self.virt_mode.findData(old_mode))
  790. self.virt_mode.currentIndexChanged.connect(self.virt_mode_changed)
  791. self.update_pv_warning()
  792. self.update_pvh_kernel_ver_warning()
  793. def update_pvh_kernel_ver_warning(self):
  794. if self.virt_mode.currentData() != 'pvh':
  795. self.pvh_kernel_version_warning.hide()
  796. return
  797. kernel = self.kernel.currentData()
  798. if self.pvh_kernel_version_ok(kernel):
  799. self.pvh_kernel_version_warning.hide()
  800. else:
  801. self.pvh_kernel_version_warning.show()
  802. def kernel_changed(self):
  803. self.update_pvh_kernel_ver_warning()
  804. def pvh_kernel_version_ok(self, name):
  805. # There are nearly no limitaions on kernel names (only file system and
  806. # general qvm-prefs rules). So we just look if we see something which
  807. # looks like a version number. It's just a warning to help the user
  808. # anyways.
  809. if name is None:
  810. return False
  811. if name is qubesadmin.DEFAULT:
  812. name = self.vm.app.default_kernel
  813. m = re.search(r'(\d+)\.(\d+)', name)
  814. if m is None:
  815. return False
  816. return (int(m.group(1)), int(m.group(2))) >= (4, 11)
  817. ######## devices tab
  818. def __init_devices_tab__(self):
  819. self.dev_list = multiselectwidget.MultiSelectWidget(self)
  820. self.dev_list.add_all_button.setVisible(False)
  821. self.devices_layout.addWidget(self.dev_list)
  822. dom0_devs = list(self.vm.app.domains['dom0'].devices['pci'].available())
  823. attached_devs = list(self.vm.devices['pci'].persistent())
  824. # pylint: disable=too-few-public-methods
  825. class DevListWidgetItem(QtWidgets.QListWidgetItem):
  826. def __init__(self, dev, unknown=False, parent=None):
  827. super(DevListWidgetItem, self).__init__(parent)
  828. name = dev.ident.replace('_', ":") + ' ' + dev.description
  829. if unknown:
  830. name += ' (unknown)'
  831. self.setText(name)
  832. self.dev = dev
  833. for dev in dom0_devs:
  834. if dev in attached_devs:
  835. self.dev_list.selected_list.addItem(DevListWidgetItem(dev))
  836. else:
  837. self.dev_list.available_list.addItem(DevListWidgetItem(dev))
  838. for dev in attached_devs:
  839. if dev not in dom0_devs:
  840. self.dev_list.selected_list.addItem(
  841. DevListWidgetItem(dev, unknown=True))
  842. if self.dev_list.selected_list.count() > 0\
  843. and self.include_in_balancing.isChecked():
  844. self.dmm_warning_adv.show()
  845. self.dmm_warning_dev.show()
  846. else:
  847. self.dmm_warning_adv.hide()
  848. self.dmm_warning_dev.hide()
  849. if self.vm.is_running():
  850. self.dev_list.setEnabled(False)
  851. self.turn_off_vm_to_modify_devs.setVisible(True)
  852. self.no_strict_reset_button.setEnabled(False)
  853. else:
  854. self.dev_list.setEnabled(True)
  855. self.turn_off_vm_to_modify_devs.setVisible(False)
  856. self.update_pvh_dont_support_devs()
  857. self.dev_list.setEnabled(not self.vm.is_running())
  858. def __apply_devices_tab__(self):
  859. msg = []
  860. try:
  861. old_devs = list(self.vm.devices['pci'].persistent())
  862. new_devs = [self.dev_list.selected_list.item(i).dev
  863. for i in range(self.dev_list.selected_list.count())]
  864. for dev in new_devs:
  865. if dev not in old_devs:
  866. options = {}
  867. if dev.ident in self.new_strict_reset_list:
  868. options['no-strict-reset'] = True
  869. ass = devices.DeviceAssignment(
  870. self.vm.app.domains['dom0'],
  871. dev.ident, persistent=True, options=options)
  872. self.vm.devices['pci'].attach(ass)
  873. elif (dev.ident in self.current_strict_reset_list) != \
  874. (dev.ident in self.new_strict_reset_list):
  875. current_assignment = None
  876. for assignment in self.vm.devices['pci'].assignments(
  877. persistent=True):
  878. if assignment.ident == dev.ident:
  879. current_assignment = assignment
  880. break
  881. if current_assignment is None:
  882. # it would be very weird if this happened
  883. msg.append(self.tr("Error re-assigning device ") +
  884. dev.ident)
  885. continue
  886. self.vm.devices['pci'].detach(current_assignment)
  887. current_assignment.options['no-strict-reset'] = \
  888. (dev.ident in self.new_strict_reset_list)
  889. self.vm.devices['pci'].attach(current_assignment)
  890. for ass in self.vm.devices['pci'].assignments(persistent=True):
  891. if ass.device not in new_devs:
  892. self.vm.devices['pci'].detach(ass)
  893. except qubesadmin.exc.QubesException as ex:
  894. if utils.is_debug():
  895. traceback.print_exc()
  896. msg.append(str(ex))
  897. return msg
  898. def devices_selection_changed(self):
  899. if self.include_in_balancing.isChecked():
  900. if self.dev_list.selected_list.count() > 0:
  901. self.dmm_warning_adv.show()
  902. self.dmm_warning_dev.show()
  903. else:
  904. self.dmm_warning_adv.hide()
  905. self.dmm_warning_dev.hide()
  906. self.update_virt_mode_list()
  907. def update_pvh_dont_support_devs(self):
  908. if self.virt_mode.currentData() == 'pvh':
  909. self.dev_list.setEnabled(False)
  910. self.pvh_dont_support_devs.setVisible(True)
  911. else:
  912. self.dev_list.setEnabled(True)
  913. self.pvh_dont_support_devs.setVisible(False)
  914. def define_strict_reset_devices(self):
  915. for assignment in self.vm.devices['pci'].assignments():
  916. if assignment.options.get('no-strict-reset', False):
  917. self.current_strict_reset_list.append(
  918. assignment.ident.replace('_', ':'))
  919. self.new_strict_reset_list = self.current_strict_reset_list.copy()
  920. def strict_reset_button_pressed(self):
  921. device_list_window = device_list.PCIDeviceListWindow(
  922. self.vm, self.qapp, self.dev_list, self.new_strict_reset_list, self)
  923. device_list_window.exec()
  924. ######## applications tab
  925. def refresh_apps_button_pressed(self):
  926. self.refresh_apps_button.setEnabled(False)
  927. self.refresh_apps_button.setText(self.tr('Refresh in progress...'))
  928. thread = RefreshAppsVMThread(self.vm, self.refresh_apps_button)
  929. thread.finished.connect(self.clear_threads)
  930. thread.finished.connect(self.refresh_finished)
  931. self.threads_list.append(thread)
  932. thread.start()
  933. def refresh_finished(self):
  934. self.app_list_manager = AppmenuSelectManager(self.vm, self.app_list)
  935. self.refresh_apps_button.setEnabled(True)
  936. self.refresh_apps_button.setText(self.tr('Refresh Applications'))
  937. def template_apps_change(self):
  938. if self.tabWidget.isTabEnabled(self.tabs_indices["applications"]):
  939. self.app_list_manager.fill_apps_list(
  940. template=self.template_name.currentData())
  941. # add a label to show
  942. self.warn_template_missing_apps.setVisible(
  943. self.app_list_manager.has_missing)
  944. ######## services tab
  945. def __init_services_tab__(self):
  946. self.new_srv_dict = {}
  947. for feature in self.vm.features:
  948. if not feature.startswith('service.'):
  949. continue
  950. service = feature[len('service.'):]
  951. item = QtWidgets.QListWidgetItem(service)
  952. item.setCheckState(ui_settingsdlg.QtCore.Qt.Checked
  953. if self.vm.features[feature]
  954. else ui_settingsdlg.QtCore.Qt.Unchecked)
  955. self.services_list.addItem(item)
  956. self.new_srv_dict[service] = self.vm.features[feature]
  957. self.service_line_edit.addItem("")
  958. supported_services = set()
  959. service_prefix = "supported-service."
  960. for feature in self.vm.features:
  961. if feature.startswith(service_prefix):
  962. supported_services.add(feature[len(service_prefix):])
  963. if getattr(self.vm, "template", None):
  964. for feature in self.vm.template.features:
  965. if feature.startswith(service_prefix):
  966. supported_services.add(feature[len(service_prefix):])
  967. for service in sorted(supported_services):
  968. self.service_line_edit.addItem(service)
  969. self.service_line_edit.addItem(self.tr('(custom...)'))
  970. self.service_line_edit.setEditText("")
  971. def __add_service__(self):
  972. srv = str(self.service_line_edit.currentText()).strip()
  973. if srv != "":
  974. if self.service_line_edit.currentIndex() == \
  975. len(self.service_line_edit) - 1:
  976. (custom_name, ok) = QtWidgets.QInputDialog.getText(
  977. self, self.tr("Custom service name"),
  978. self.tr(
  979. "Name of the service:"))
  980. if ok:
  981. srv = custom_name.strip()
  982. else:
  983. return
  984. if srv in self.new_srv_dict:
  985. QtWidgets.QMessageBox.information(
  986. self,
  987. '',
  988. self.tr('Service already on the list!'))
  989. return
  990. item = QtWidgets.QListWidgetItem(srv)
  991. item.setCheckState(ui_settingsdlg.QtCore.Qt.Checked)
  992. self.services_list.addItem(item)
  993. self.new_srv_dict[srv] = True
  994. def __remove_service__(self):
  995. item = self.services_list.currentItem()
  996. if not item:
  997. return
  998. row = self.services_list.currentRow()
  999. item = self.services_list.takeItem(row)
  1000. del self.new_srv_dict[str(item.text())]
  1001. def __apply_services_tab__(self):
  1002. msg = []
  1003. try:
  1004. for i in range(self.services_list.count()):
  1005. item = self.services_list.item(i)
  1006. self.new_srv_dict[str(item.text())] = \
  1007. (item.checkState() == ui_settingsdlg.QtCore.Qt.Checked)
  1008. for service, v in self.new_srv_dict.items():
  1009. feature = 'service.' + service
  1010. if v != self.vm.features.get(feature, object()):
  1011. self.vm.features[feature] = v
  1012. for feature in self.vm.features:
  1013. if not feature.startswith('service.'):
  1014. continue
  1015. service = feature[len('service.'):]
  1016. if service not in self.new_srv_dict:
  1017. del self.vm.features[feature]
  1018. except qubesadmin.exc.QubesException as ex:
  1019. msg.append(str(ex))
  1020. return msg
  1021. ######### firewall tab related
  1022. def set_fw_model(self, model):
  1023. self.fw_model = model
  1024. self.rulesTreeView.setModel(model)
  1025. self.rulesTreeView.header().setSectionResizeMode(
  1026. QtWidgets.QHeaderView.ResizeToContents)
  1027. self.rulesTreeView.header().setSectionResizeMode(
  1028. 0, QtWidgets.QHeaderView.Stretch)
  1029. self.set_allow(model.allow)
  1030. if model.temp_full_access_expire_time:
  1031. self.temp_full_access.setChecked(True)
  1032. self.temp_full_access_time.setValue(
  1033. (model.temp_full_access_expire_time -
  1034. int(firewall.datetime.datetime.now().strftime("%s"))) / 60)
  1035. def disable_all_fw_conf(self):
  1036. self.firewall_modified_outside_label.setVisible(True)
  1037. self.policy_allow_radio_button.setEnabled(False)
  1038. self.policy_deny_radio_button.setEnabled(False)
  1039. self.rulesTreeView.setEnabled(False)
  1040. self.new_rule_button.setEnabled(False)
  1041. self.edit_rule_button.setEnabled(False)
  1042. self.delete_rule_button.setEnabled(False)
  1043. self.firewal_rules_label.setEnabled(False)
  1044. self.tempFullAccessWidget.setEnabled(False)
  1045. def set_allow(self, allow):
  1046. self.policy_allow_radio_button.setChecked(allow)
  1047. self.policy_deny_radio_button.setChecked(not allow)
  1048. self.policy_changed()
  1049. def policy_changed(self):
  1050. self.rulesTreeView.setEnabled(
  1051. self.policy_deny_radio_button.isChecked())
  1052. self.new_rule_button.setEnabled(
  1053. self.policy_deny_radio_button.isChecked())
  1054. self.edit_rule_button.setEnabled(
  1055. self.policy_deny_radio_button.isChecked())
  1056. self.delete_rule_button.setEnabled(
  1057. self.policy_deny_radio_button.isChecked())
  1058. self.firewal_rules_label.setEnabled(
  1059. self.policy_deny_radio_button.isChecked())
  1060. self.tempFullAccessWidget.setEnabled(
  1061. self.policy_deny_radio_button.isChecked())
  1062. def new_rule_button_pressed(self):
  1063. dialog = firewall.NewFwRuleDlg()
  1064. self.fw_model.run_rule_dialog(dialog)
  1065. def edit_rule_button_pressed(self):
  1066. selected = self.rulesTreeView.selectedIndexes()
  1067. if selected:
  1068. dialog = firewall.NewFwRuleDlg()
  1069. dialog.set_ok_state(True)
  1070. row = self.rulesTreeView.selectedIndexes().pop().row()
  1071. self.fw_model.populate_edit_dialog(dialog, row)
  1072. self.fw_model.run_rule_dialog(dialog, row)
  1073. def delete_rule_button_pressed(self):
  1074. for i in {index.row() for index
  1075. in self.rulesTreeView.selectedIndexes()}:
  1076. self.fw_model.remove_child(i)
  1077. parser = QubesArgumentParser(vmname_nargs=1)
  1078. parser.add_argument('--tab', metavar='TAB',
  1079. action='store',
  1080. choices=VMSettingsWindow.tabs_indices.keys())
  1081. parser.set_defaults(
  1082. tab='basic',
  1083. )
  1084. def main(args=None):
  1085. args = parser.parse_args(args)
  1086. vm = args.domains.pop()
  1087. utils.run_synchronous(functools.partial(VMSettingsWindow, vm, args.tab))
  1088. if __name__ == "__main__":
  1089. main()
  1090. # vim:sw=4:et: