qube_manager.py 50 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301
  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 sys
  25. import os
  26. import os.path
  27. import subprocess
  28. from datetime import datetime, timedelta
  29. import traceback
  30. from contextlib import suppress
  31. import quamash
  32. import asyncio
  33. from qubesadmin import Qubes
  34. from qubesadmin import exc
  35. from qubesadmin import utils
  36. from qubesadmin import events
  37. from PyQt4 import QtGui # pylint: disable=import-error
  38. from PyQt4 import QtCore # pylint: disable=import-error
  39. from qubesmanager.about import AboutDialog
  40. from . import ui_qubemanager # pylint: disable=no-name-in-module
  41. from . import table_widgets
  42. from . import settings
  43. from . import global_settings
  44. from . import restore
  45. from . import backup
  46. from . import create_new_vm
  47. from . import log_dialog
  48. from . import utils as manager_utils
  49. from . import common_threads
  50. class SearchBox(QtGui.QLineEdit):
  51. def __init__(self, parent=None):
  52. super(SearchBox, self).__init__(parent)
  53. self.focusing = False
  54. def focusInEvent(self, e): # pylint: disable=invalid-name
  55. super(SearchBox, self).focusInEvent(e)
  56. self.selectAll()
  57. self.focusing = True
  58. def mousePressEvent(self, e): # pylint: disable=invalid-name
  59. super(SearchBox, self).mousePressEvent(e)
  60. if self.focusing:
  61. self.selectAll()
  62. self.focusing = False
  63. class VmRowInTable:
  64. # pylint: disable=too-few-public-methods
  65. def __init__(self, vm, row_no, table):
  66. self.vm = vm
  67. # TODO: replace a various different widgets with a more generic
  68. # VmFeatureWidget or VMPropertyWidget
  69. table_widgets.row_height = VmManagerWindow.row_height
  70. table.setRowHeight(row_no, VmManagerWindow.row_height)
  71. self.type_widget = table_widgets.VmTypeWidget(vm)
  72. table.setCellWidget(row_no, VmManagerWindow.columns_indices['Type'],
  73. self.type_widget)
  74. table.setItem(row_no, VmManagerWindow.columns_indices['Type'],
  75. self.type_widget.table_item)
  76. self.label_widget = table_widgets.VmLabelWidget(vm)
  77. table.setCellWidget(row_no, VmManagerWindow.columns_indices['Label'],
  78. self.label_widget)
  79. table.setItem(row_no, VmManagerWindow.columns_indices['Label'],
  80. self.label_widget.table_item)
  81. self.name_widget = table_widgets.VmNameItem(vm)
  82. table.setItem(row_no, VmManagerWindow.columns_indices['Name'],
  83. self.name_widget)
  84. self.info_widget = table_widgets.VmInfoWidget(vm)
  85. table.setCellWidget(row_no, VmManagerWindow.columns_indices['State'],
  86. self.info_widget)
  87. table.setItem(row_no, VmManagerWindow.columns_indices['State'],
  88. self.info_widget.table_item)
  89. self.template_widget = table_widgets.VmTemplateItem(vm)
  90. table.setItem(row_no, VmManagerWindow.columns_indices['Template'],
  91. self.template_widget)
  92. self.netvm_widget = table_widgets.VmNetvmItem(vm)
  93. table.setItem(row_no, VmManagerWindow.columns_indices['NetVM'],
  94. self.netvm_widget)
  95. self.size_widget = table_widgets.VmSizeOnDiskItem(vm)
  96. table.setItem(row_no, VmManagerWindow.columns_indices['Size'],
  97. self.size_widget)
  98. self.internal_widget = table_widgets.VmInternalItem(vm)
  99. table.setItem(row_no, VmManagerWindow.columns_indices['Internal'],
  100. self.internal_widget)
  101. self.ip_widget = table_widgets.VmIPItem(vm)
  102. table.setItem(row_no, VmManagerWindow.columns_indices['IP'],
  103. self.ip_widget)
  104. self.include_in_backups_widget = \
  105. table_widgets.VmIncludeInBackupsItem(vm)
  106. table.setItem(row_no, VmManagerWindow.columns_indices[
  107. 'Backups'], self.include_in_backups_widget)
  108. self.last_backup_widget = table_widgets.VmLastBackupItem(vm)
  109. table.setItem(row_no, VmManagerWindow.columns_indices[
  110. 'Last backup'], self.last_backup_widget)
  111. self.table = table
  112. def update(self, update_size_on_disk=False, event=None):
  113. """
  114. Update info in a single VM row
  115. :param update_size_on_disk: should disk utilization be updated? the
  116. widget will extract the data from VM object
  117. :param event: name of the event that caused the update, to avoid
  118. updating unnecessary properties; if event is none, update everything
  119. :return: None
  120. """
  121. try:
  122. self.info_widget.update_vm_state()
  123. if not event or event.endswith(':label'):
  124. self.label_widget.update()
  125. if not event or event.endswith(':template'):
  126. self.template_widget.update()
  127. if not event or event.endswith(':netvm'):
  128. self.netvm_widget.update()
  129. if not event or event.endswith(':internal'):
  130. self.internal_widget.update()
  131. if not event or event.endswith(':ip'):
  132. self.ip_widget.update()
  133. if not event or event.endswith(':include_in_backups'):
  134. self.include_in_backups_widget.update()
  135. if not event or event.endswith(':backup_timestamp'):
  136. self.last_backup_widget.update()
  137. if update_size_on_disk:
  138. self.size_widget.update()
  139. except exc.QubesPropertyAccessError:
  140. pass
  141. except exc.QubesDaemonNoResponseError:
  142. # TODO: this will be fixed by a rewrite moving the event system to
  143. # AdminAPI
  144. pass
  145. #force re-sorting
  146. self.table.setSortingEnabled(True)
  147. vm_shutdown_timeout = 20000 # in msec
  148. vm_restart_check_timeout = 1000 # in msec
  149. class VmShutdownMonitor(QtCore.QObject):
  150. def __init__(self, vm, shutdown_time=vm_shutdown_timeout,
  151. check_time=vm_restart_check_timeout,
  152. and_restart=False, caller=None):
  153. QtCore.QObject.__init__(self)
  154. self.vm = vm
  155. self.shutdown_time = shutdown_time
  156. self.check_time = check_time
  157. self.and_restart = and_restart
  158. self.shutdown_started = datetime.now()
  159. self.caller = caller
  160. def restart_vm_if_needed(self):
  161. if self.and_restart and self.caller:
  162. self.caller.start_vm(self.vm)
  163. def check_again_later(self):
  164. # noinspection PyTypeChecker,PyCallByClass
  165. QtCore.QTimer.singleShot(self.check_time, self.check_if_vm_has_shutdown)
  166. def timeout_reached(self):
  167. actual = datetime.now() - self.shutdown_started
  168. allowed = timedelta(milliseconds=self.shutdown_time)
  169. return actual > allowed
  170. def check_if_vm_has_shutdown(self):
  171. vm = self.vm
  172. vm_is_running = vm.is_running()
  173. try:
  174. vm_start_time = datetime.fromtimestamp(float(vm.start_time))
  175. except (AttributeError, TypeError, ValueError):
  176. vm_start_time = None
  177. if vm_is_running and vm_start_time \
  178. and vm_start_time < self.shutdown_started:
  179. if self.timeout_reached():
  180. reply = QtGui.QMessageBox.question(
  181. None, self.tr("Qube Shutdown"),
  182. self.tr(
  183. "The Qube <b>'{0}'</b> hasn't shutdown within the last "
  184. "{1} seconds, do you want to kill it?<br>").format(
  185. vm.name, self.shutdown_time / 1000),
  186. self.tr("Kill it!"),
  187. self.tr("Wait another {0} seconds...").format(
  188. self.shutdown_time / 1000))
  189. if reply == 0:
  190. try:
  191. vm.kill()
  192. except exc.QubesVMNotStartedError:
  193. # the VM shut down while the user was thinking about
  194. # shutting it down
  195. pass
  196. self.restart_vm_if_needed()
  197. else:
  198. self.shutdown_started = datetime.now()
  199. self.check_again_later()
  200. else:
  201. self.check_again_later()
  202. else:
  203. if vm_is_running:
  204. # Due to unknown reasons, Xen sometimes reports that a domain
  205. # is running even though its start-up timestamp is not valid.
  206. # Make sure that "restart_vm_if_needed" is not called until
  207. # the domain has been completely shut down according to Xen.
  208. self.check_again_later()
  209. return
  210. self.restart_vm_if_needed()
  211. # pylint: disable=too-few-public-methods
  212. class StartVMThread(common_threads.QubesThread):
  213. def run(self):
  214. try:
  215. self.vm.start()
  216. except exc.QubesException as ex:
  217. self.msg = ("Error starting Qube!", str(ex))
  218. # pylint: disable=too-few-public-methods
  219. class UpdateVMThread(common_threads.QubesThread):
  220. def run(self):
  221. try:
  222. if self.vm.qid == 0:
  223. subprocess.check_call(
  224. ["/usr/bin/qubes-dom0-update", "--clean", "--gui"])
  225. else:
  226. if not self.vm.is_running():
  227. self.vm.start()
  228. # apply DSA-4371
  229. with open('/usr/libexec/qubes-manager/dsa-4371-update', 'rb') \
  230. as dsa4371update:
  231. stdout, stderr = self.vm.run_service_for_stdio(
  232. "qubes.VMShell",
  233. user="root",
  234. input=dsa4371update.read())
  235. if stdout == b'changed=yes\n':
  236. subprocess.call(['notify-send', '-i', 'dialog-information',
  237. 'Debian DSA-4371 fix installed in {}'.format(
  238. self.vm.name)])
  239. elif stdout == b'changed=no\n':
  240. pass
  241. else:
  242. raise exc.QubesException(
  243. "Failed to apply DSA-4371 fix: {}".format(
  244. stderr.decode('ascii')))
  245. self.vm.run_service("qubes.InstallUpdatesGUI",\
  246. user="root", wait=False)
  247. except (ChildProcessError, exc.QubesException) as ex:
  248. self.msg = ("Error on qube update!", str(ex))
  249. # pylint: disable=too-few-public-methods
  250. class RunCommandThread(common_threads.QubesThread):
  251. def __init__(self, vm, command_to_run):
  252. super(RunCommandThread, self).__init__(vm)
  253. self.command_to_run = command_to_run
  254. def run(self):
  255. try:
  256. self.vm.run(self.command_to_run)
  257. except (ChildProcessError, exc.QubesException) as ex:
  258. self.msg = ("Error while running command!", str(ex))
  259. class VmManagerWindow(ui_qubemanager.Ui_VmManagerWindow, QtGui.QMainWindow):
  260. # pylint: disable=too-many-instance-attributes
  261. row_height = 30
  262. column_width = 200
  263. search = ""
  264. # suppress saving settings while initializing widgets
  265. settings_loaded = False
  266. columns_indices = {"Type": 0,
  267. "Label": 1,
  268. "Name": 2,
  269. "State": 3,
  270. "Template": 4,
  271. "NetVM": 5,
  272. "Size": 6,
  273. "Internal": 7,
  274. "IP": 8,
  275. "Backups": 9,
  276. "Last backup": 10,
  277. }
  278. def __init__(self, qt_app, qubes_app, dispatcher, parent=None):
  279. # pylint: disable=unused-argument
  280. super(VmManagerWindow, self).__init__()
  281. self.setupUi(self)
  282. self.manager_settings = QtCore.QSettings(self)
  283. self.qubes_app = qubes_app
  284. self.qt_app = qt_app
  285. self.searchbox = SearchBox()
  286. self.searchbox.setValidator(QtGui.QRegExpValidator(
  287. QtCore.QRegExp("[a-zA-Z0-9_-]*", QtCore.Qt.CaseInsensitive), None))
  288. self.searchContainer.addWidget(self.searchbox)
  289. self.connect(self.table, QtCore.SIGNAL("itemSelectionChanged()"),
  290. self.table_selection_changed)
  291. self.table.setColumnWidth(0, self.column_width)
  292. self.sort_by_column = "Type"
  293. self.sort_order = QtCore.Qt.AscendingOrder
  294. self.vms_list = []
  295. self.vms_in_table = {}
  296. self.frame_width = 0
  297. self.frame_height = 0
  298. self.columns_actions = {
  299. self.columns_indices["Type"]: self.action_vm_type,
  300. self.columns_indices["Label"]: self.action_label,
  301. self.columns_indices["Name"]: self.action_name,
  302. self.columns_indices["State"]: self.action_state,
  303. self.columns_indices["Template"]: self.action_template,
  304. self.columns_indices["NetVM"]: self.action_netvm,
  305. self.columns_indices["Size"]: self.action_size_on_disk,
  306. self.columns_indices["Internal"]: self.action_internal,
  307. self.columns_indices["IP"]: self
  308. .action_ip, self.columns_indices["Backups"]: self
  309. .action_backups, self.columns_indices["Last backup"]: self
  310. .action_last_backup
  311. }
  312. self.visible_columns_count = len(self.columns_indices)
  313. # Other columns get sensible default sizes, but those have only
  314. # icon content, and thus PyQt makes them too wide
  315. self.table.setColumnWidth(self.columns_indices["State"], 80)
  316. self.table.setColumnWidth(self.columns_indices["Label"], 40)
  317. self.table.setColumnWidth(self.columns_indices["Type"], 40)
  318. self.table.horizontalHeader().setResizeMode(
  319. QtGui.QHeaderView.Interactive)
  320. self.table.horizontalHeader().setStretchLastSection(True)
  321. self.table.horizontalHeader().setMinimumSectionSize(40)
  322. self.context_menu = QtGui.QMenu(self)
  323. self.context_menu.addAction(self.action_settings)
  324. self.context_menu.addAction(self.action_editfwrules)
  325. self.context_menu.addAction(self.action_appmenus)
  326. self.context_menu.addAction(self.action_set_keyboard_layout)
  327. self.context_menu.addSeparator()
  328. self.context_menu.addAction(self.action_updatevm)
  329. self.context_menu.addAction(self.action_run_command_in_vm)
  330. self.context_menu.addAction(self.action_resumevm)
  331. self.context_menu.addAction(self.action_startvm_tools_install)
  332. self.context_menu.addAction(self.action_pausevm)
  333. self.context_menu.addAction(self.action_shutdownvm)
  334. self.context_menu.addAction(self.action_restartvm)
  335. self.context_menu.addAction(self.action_killvm)
  336. self.context_menu.addSeparator()
  337. self.context_menu.addAction(self.action_clonevm)
  338. self.context_menu.addAction(self.action_removevm)
  339. self.context_menu.addSeparator()
  340. self.context_menu.addMenu(self.logs_menu)
  341. self.context_menu.addSeparator()
  342. self.tools_context_menu = QtGui.QMenu(self)
  343. self.tools_context_menu.addAction(self.action_toolbar)
  344. self.tools_context_menu.addAction(self.action_menubar)
  345. self.dom0_context_menu = QtGui.QMenu(self)
  346. self.dom0_context_menu.addAction(self.action_global_settings)
  347. self.dom0_context_menu.addAction(self.action_updatevm)
  348. self.dom0_context_menu.addSeparator()
  349. self.dom0_context_menu.addMenu(self.logs_menu)
  350. self.dom0_context_menu.addSeparator()
  351. self.connect(
  352. self.table.horizontalHeader(),
  353. QtCore.SIGNAL("sortIndicatorChanged(int, Qt::SortOrder)"),
  354. self.sort_indicator_changed)
  355. self.connect(self.table,
  356. QtCore.SIGNAL("customContextMenuRequested(const QPoint&)"),
  357. self.open_context_menu)
  358. self.connect(self.menubar,
  359. QtCore.SIGNAL("customContextMenuRequested(const QPoint&)"),
  360. lambda pos: self.open_tools_context_menu(self.menubar,
  361. pos))
  362. self.connect(self.toolbar,
  363. QtCore.SIGNAL("customContextMenuRequested(const QPoint&)"),
  364. lambda pos: self.open_tools_context_menu(self.toolbar,
  365. pos))
  366. self.connect(self.logs_menu, QtCore.SIGNAL("triggered(QAction *)"),
  367. self.show_log)
  368. self.connect(self.searchbox,
  369. QtCore.SIGNAL("textChanged(const QString&)"),
  370. self.do_search)
  371. self.table.setContentsMargins(0, 0, 0, 0)
  372. self.centralwidget.layout().setContentsMargins(0, 0, 0, 0)
  373. self.layout().setContentsMargins(0, 0, 0, 0)
  374. self.connect(self.action_menubar, QtCore.SIGNAL("toggled(bool)"),
  375. self.showhide_menubar)
  376. self.connect(self.action_toolbar, QtCore.SIGNAL("toggled(bool)"),
  377. self.showhide_toolbar)
  378. self.load_manager_settings()
  379. self.fill_table()
  380. self.update_size_on_disk = False
  381. self.shutdown_monitor = {}
  382. # Connect events
  383. self.dispatcher = dispatcher
  384. dispatcher.add_handler('domain-pre-start',
  385. self.on_domain_status_changed)
  386. dispatcher.add_handler('domain-start', self.on_domain_status_changed)
  387. dispatcher.add_handler('domain-start-failed',
  388. self.on_domain_status_changed)
  389. dispatcher.add_handler('domain-stopped', self.on_domain_status_changed)
  390. dispatcher.add_handler('domain-pre-shutdown',
  391. self.on_domain_status_changed)
  392. dispatcher.add_handler('domain-shutdown', self.on_domain_status_changed)
  393. dispatcher.add_handler('domain-paused', self.on_domain_status_changed)
  394. dispatcher.add_handler('domain-unpaused', self.on_domain_status_changed)
  395. dispatcher.add_handler('domain-add', self.on_domain_added)
  396. dispatcher.add_handler('domain-delete', self.on_domain_removed)
  397. dispatcher.add_handler('property-set:*',
  398. self.on_domain_changed)
  399. dispatcher.add_handler('property-del:*',
  400. self.on_domain_changed)
  401. dispatcher.add_handler('property-load',
  402. self.on_domain_changed)
  403. # It needs to store threads until they finish
  404. self.threads_list = []
  405. self.progress = None
  406. # Check Updates Timer
  407. timer = QtCore.QTimer(self)
  408. timer.timeout.connect(self.check_updates)
  409. timer.start(1000 * 30) # 30s
  410. self.check_updates()
  411. def keyPressEvent(self, event): # pylint: disable=invalid-name
  412. if event.key() == QtCore.Qt.Key_Escape:
  413. self.searchbox.clear()
  414. super(VmManagerWindow, self).keyPressEvent(event)
  415. def clear_threads(self):
  416. for thread in self.threads_list:
  417. if thread.isFinished():
  418. if self.progress:
  419. self.progress.hide()
  420. self.progress = None
  421. if thread.msg:
  422. (title, msg) = thread.msg
  423. if thread.msg_is_success:
  424. QtGui.QMessageBox.information(
  425. None,
  426. self.tr(title),
  427. self.tr(msg))
  428. else:
  429. QtGui.QMessageBox.warning(
  430. None,
  431. self.tr(title),
  432. self.tr(msg))
  433. self.threads_list.remove(thread)
  434. return
  435. raise RuntimeError('No finished thread found')
  436. def closeEvent(self, event):
  437. # pylint: disable=invalid-name
  438. # save window size at close
  439. self.manager_settings.setValue("window_size", self.size())
  440. event.accept()
  441. def check_updates(self):
  442. for vm in self.qubes_app.domains:
  443. if vm.klass in {'TemplateVM', 'StandaloneVM'}:
  444. try:
  445. self.vms_in_table[vm.qid].info_widget.update_vm_state()
  446. except (exc.QubesException, KeyError):
  447. # the VM might have vanished in the meantime or
  448. # the signal might have been handled in the wrong order
  449. pass
  450. def on_domain_added(self, _submitter, _event, vm, **_kwargs):
  451. row_no = 0
  452. self.table.setSortingEnabled(False)
  453. try:
  454. domain = self.qubes_app.domains[vm]
  455. row_no = self.table.rowCount()
  456. self.table.setRowCount(row_no + 1)
  457. vm_row = VmRowInTable(domain, row_no, self.table)
  458. self.vms_in_table[domain.qid] = vm_row
  459. except (exc.QubesException, KeyError):
  460. if row_no != 0:
  461. self.table.removeRow(row_no)
  462. self.table.setSortingEnabled(True)
  463. self.showhide_vms()
  464. def on_domain_removed(self, _submitter, _event, **kwargs):
  465. row_to_delete = None
  466. qid_to_delete = None
  467. for qid, row in self.vms_in_table.items():
  468. if row.vm.name == kwargs['vm']:
  469. row_to_delete = row
  470. qid_to_delete = qid
  471. if not row_to_delete:
  472. return # for some reason, the VM was removed in some other way
  473. del self.vms_in_table[qid_to_delete]
  474. self.table.removeRow(row_to_delete.name_widget.row())
  475. def on_domain_status_changed(self, vm, _event, **_kwargs):
  476. try:
  477. self.vms_in_table[vm.qid].info_widget.update_vm_state()
  478. except exc.QubesPropertyAccessError:
  479. return # the VM was deleted before its status could be updated
  480. if vm == self.get_selected_vm():
  481. self.table_selection_changed()
  482. if vm.klass == 'TemplateVM':
  483. for row in self.vms_in_table.values():
  484. if getattr(row.vm, 'template', None) == vm:
  485. row.info_widget.update_vm_state()
  486. def on_domain_changed(self, vm, event, **_kwargs):
  487. if not vm: # change of global properties occured
  488. return
  489. try:
  490. self.vms_in_table[vm.qid].update(event=event)
  491. except exc.QubesPropertyAccessError:
  492. return # the VM was deleted before its status could be updated
  493. def load_manager_settings(self):
  494. # visible columns
  495. self.visible_columns_count = 0
  496. for col in self.columns_indices:
  497. col_no = self.columns_indices[col]
  498. visible = self.manager_settings.value(
  499. 'columns/%s' % col,
  500. defaultValue="true")
  501. self.columns_actions[col_no].setChecked(visible == "true")
  502. self.visible_columns_count += 1
  503. self.sort_by_column = str(
  504. self.manager_settings.value("view/sort_column",
  505. defaultValue=self.sort_by_column))
  506. self.sort_order = QtCore.Qt.SortOrder(
  507. self.manager_settings.value("view/sort_order",
  508. defaultValue=self.sort_order))
  509. self.table.sortItems(self.columns_indices[self.sort_by_column],
  510. self.sort_order)
  511. if not self.manager_settings.value("view/menubar_visible",
  512. defaultValue=True):
  513. self.action_menubar.setChecked(False)
  514. if not self.manager_settings.value("view/toolbar_visible",
  515. defaultValue=True):
  516. self.action_toolbar.setChecked(False)
  517. # load last window size
  518. self.resize(self.manager_settings.value("window_size",
  519. QtCore.QSize(1100, 600)))
  520. self.settings_loaded = True
  521. def get_vms_list(self):
  522. return [vm for vm in self.qubes_app.domains]
  523. def fill_table(self):
  524. self.table.setSortingEnabled(False)
  525. vms_list = self.get_vms_list()
  526. vms_in_table = {}
  527. self.table.setRowCount(len(vms_list))
  528. progress = QtGui.QProgressDialog(
  529. self.tr(
  530. "Loading Qube Manager..."), "", 0, len(vms_list))
  531. progress.setWindowTitle(self.tr("Qube Manager"))
  532. progress.setMinimumDuration(1000)
  533. progress.setCancelButton(None)
  534. row_no = 0
  535. for vm in vms_list:
  536. progress.setValue(row_no)
  537. vm_row = VmRowInTable(vm, row_no, self.table)
  538. vms_in_table[vm.qid] = vm_row
  539. row_no += 1
  540. progress.setValue(row_no)
  541. self.vms_list = vms_list
  542. self.vms_in_table = vms_in_table
  543. self.table.setSortingEnabled(True)
  544. def showhide_vms(self):
  545. if not self.search:
  546. for row_no in range(self.table.rowCount()):
  547. self.table.setRowHidden(row_no, False)
  548. else:
  549. for row_no in range(self.table.rowCount()):
  550. widget = self.table.cellWidget(row_no,
  551. self.columns_indices["State"])
  552. show = (self.search in widget.vm.name)
  553. self.table.setRowHidden(row_no, not show)
  554. @QtCore.pyqtSlot(str)
  555. def do_search(self, search):
  556. self.search = str(search)
  557. self.showhide_vms()
  558. # noinspection PyArgumentList
  559. @QtCore.pyqtSlot(name='on_action_search_triggered')
  560. def action_search_triggered(self):
  561. self.searchbox.setFocus()
  562. # noinspection PyPep8Naming
  563. def sort_indicator_changed(self, column, order):
  564. self.sort_by_column = [name for name in self.columns_indices if
  565. self.columns_indices[name] == column][0]
  566. self.sort_order = order
  567. if self.settings_loaded:
  568. self.manager_settings.setValue('view/sort_column',
  569. self.sort_by_column)
  570. self.manager_settings.setValue('view/sort_order', self.sort_order)
  571. self.manager_settings.sync()
  572. def table_selection_changed(self):
  573. vm = self.get_selected_vm()
  574. if vm is not None and vm in self.qubes_app.domains:
  575. # TODO: add boot from device to menu and add windows tools there
  576. # Update available actions:
  577. self.action_settings.setEnabled(vm.klass != 'AdminVM')
  578. self.action_removevm.setEnabled(
  579. vm.klass != 'AdminVM' and not vm.is_running())
  580. self.action_clonevm.setEnabled(vm.klass != 'AdminVM')
  581. self.action_resumevm.setEnabled(
  582. not vm.is_running() or vm.get_power_state() == "Paused")
  583. self.action_pausevm.setEnabled(
  584. vm.is_running() and vm.get_power_state() != "Paused"
  585. and vm.klass != 'AdminVM')
  586. self.action_shutdownvm.setEnabled(
  587. vm.is_running() and vm.get_power_state() != "Paused"
  588. and vm.klass != 'AdminVM')
  589. self.action_restartvm.setEnabled(
  590. vm.is_running() and vm.get_power_state() != "Paused"
  591. and vm.klass != 'AdminVM'
  592. and (vm.klass != 'DispVM' or not vm.auto_cleanup))
  593. self.action_killvm.setEnabled(
  594. (vm.get_power_state() == "Paused" or vm.is_running())
  595. and vm.klass != 'AdminVM')
  596. self.action_appmenus.setEnabled(
  597. vm.klass != 'AdminVM' and vm.klass != 'DispVM'
  598. and not vm.features.get('internal', False))
  599. self.action_editfwrules.setEnabled(vm.klass != 'AdminVM')
  600. self.action_updatevm.setEnabled(getattr(vm, 'updateable', False)
  601. or vm.qid == 0)
  602. self.action_run_command_in_vm.setEnabled(
  603. not vm.get_power_state() == "Paused" and vm.qid != 0)
  604. self.action_set_keyboard_layout.setEnabled(
  605. vm.qid != 0 and
  606. vm.get_power_state() != "Paused" and vm.is_running())
  607. else:
  608. self.action_settings.setEnabled(False)
  609. self.action_removevm.setEnabled(False)
  610. self.action_clonevm.setEnabled(False)
  611. self.action_resumevm.setEnabled(False)
  612. self.action_pausevm.setEnabled(False)
  613. self.action_shutdownvm.setEnabled(False)
  614. self.action_restartvm.setEnabled(False)
  615. self.action_killvm.setEnabled(False)
  616. self.action_appmenus.setEnabled(False)
  617. self.action_editfwrules.setEnabled(False)
  618. self.action_updatevm.setEnabled(False)
  619. self.action_run_command_in_vm.setEnabled(False)
  620. self.action_set_keyboard_layout.setEnabled(False)
  621. # noinspection PyArgumentList
  622. @QtCore.pyqtSlot(name='on_action_createvm_triggered')
  623. def action_createvm_triggered(self): # pylint: disable=no-self-use
  624. create_window = create_new_vm.NewVmDlg(self.qt_app, self.qubes_app)
  625. create_window.exec_()
  626. def get_selected_vm(self):
  627. # vm selection relies on the VmInfo widget's value used
  628. # for sorting by VM name
  629. row_index = self.table.currentRow()
  630. if row_index != -1:
  631. vm_item = self.table.item(row_index, self.columns_indices["Name"])
  632. # here is possible race with update_table timer so check
  633. # if really got the item
  634. if vm_item is None:
  635. return None
  636. qid = vm_item.qid
  637. assert self.vms_in_table[qid] is not None
  638. vm = self.vms_in_table[qid].vm
  639. return vm
  640. return None
  641. # noinspection PyArgumentList
  642. @QtCore.pyqtSlot(name='on_action_removevm_triggered')
  643. def action_removevm_triggered(self):
  644. # pylint: disable=no-else-return
  645. vm = self.get_selected_vm()
  646. dependencies = utils.vm_dependencies(self.qubes_app, vm)
  647. if dependencies:
  648. list_text = "<br>" + \
  649. manager_utils.format_dependencies_list(dependencies) + \
  650. "<br>"
  651. info_dialog = QtGui.QMessageBox(self)
  652. info_dialog.setWindowTitle(self.tr("Warning!"))
  653. info_dialog.setText(
  654. self.tr("This qube cannot be removed. It is used as:"
  655. " <br> {} <small>If you want to remove this qube, "
  656. "you should remove or change settings of each qube "
  657. "or setting that uses it.</small>").format(list_text))
  658. info_dialog.setModal(False)
  659. info_dialog.show()
  660. return
  661. (requested_name, ok) = QtGui.QInputDialog.getText(
  662. None, self.tr("Qube Removal Confirmation"),
  663. self.tr("Are you sure you want to remove the Qube <b>'{0}'</b>"
  664. "?<br> All data on this Qube's private storage will be "
  665. "lost!<br><br>Type the name of the Qube (<b>{1}</b>) below "
  666. "to confirm:").format(vm.name, vm.name))
  667. if not ok:
  668. # user clicked cancel
  669. return
  670. if requested_name != vm.name:
  671. # name did not match
  672. QtGui.QMessageBox.warning(
  673. None,
  674. self.tr("Qube removal confirmation failed"),
  675. self.tr(
  676. "Entered name did not match! Not removing "
  677. "{0}.").format(vm.name))
  678. return
  679. else:
  680. # remove the VM
  681. thread = common_threads.RemoveVMThread(vm)
  682. self.threads_list.append(thread)
  683. thread.finished.connect(self.clear_threads)
  684. thread.start()
  685. # noinspection PyArgumentList
  686. @QtCore.pyqtSlot(name='on_action_clonevm_triggered')
  687. def action_clonevm_triggered(self):
  688. vm = self.get_selected_vm()
  689. name_number = 1
  690. name_format = vm.name + '-clone-%d'
  691. while name_format % name_number in self.qubes_app.domains.keys():
  692. name_number += 1
  693. (clone_name, ok) = QtGui.QInputDialog.getText(
  694. self, self.tr('Qubes clone Qube'),
  695. self.tr('Enter name for Qube <b>{}</b> clone:').format(vm.name),
  696. text=(name_format % name_number))
  697. if not ok or clone_name == "":
  698. return
  699. self.progress = QtGui.QProgressDialog(
  700. self.tr(
  701. "Cloning Qube..."), "", 0, 0)
  702. self.progress.setCancelButton(None)
  703. self.progress.setModal(True)
  704. self.progress.show()
  705. thread = common_threads.CloneVMThread(vm, clone_name)
  706. thread.finished.connect(self.clear_threads)
  707. self.threads_list.append(thread)
  708. thread.start()
  709. # noinspection PyArgumentList
  710. @QtCore.pyqtSlot(name='on_action_resumevm_triggered')
  711. def action_resumevm_triggered(self):
  712. vm = self.get_selected_vm()
  713. if vm.get_power_state() in ["Paused", "Suspended"]:
  714. try:
  715. vm.unpause()
  716. except exc.QubesException as ex:
  717. QtGui.QMessageBox.warning(
  718. None, self.tr("Error unpausing Qube!"),
  719. self.tr("ERROR: {0}").format(ex))
  720. return
  721. self.start_vm(vm)
  722. def start_vm(self, vm):
  723. if vm.is_running():
  724. return
  725. thread = StartVMThread(vm)
  726. self.threads_list.append(thread)
  727. thread.finished.connect(self.clear_threads)
  728. thread.start()
  729. # noinspection PyArgumentList
  730. @QtCore.pyqtSlot(name='on_action_startvm_tools_install_triggered')
  731. # TODO: replace with boot from device
  732. def action_startvm_tools_install_triggered(self):
  733. # pylint: disable=invalid-name
  734. pass
  735. @QtCore.pyqtSlot(name='on_action_pausevm_triggered')
  736. def action_pausevm_triggered(self):
  737. vm = self.get_selected_vm()
  738. try:
  739. vm.pause()
  740. except exc.QubesException as ex:
  741. QtGui.QMessageBox.warning(
  742. None,
  743. self.tr("Error pausing Qube!"),
  744. self.tr("ERROR: {0}").format(ex))
  745. return
  746. # noinspection PyArgumentList
  747. @QtCore.pyqtSlot(name='on_action_shutdownvm_triggered')
  748. def action_shutdownvm_triggered(self):
  749. vm = self.get_selected_vm()
  750. reply = QtGui.QMessageBox.question(
  751. None, self.tr("Qube Shutdown Confirmation"),
  752. self.tr("Are you sure you want to power down the Qube"
  753. " <b>'{0}'</b>?<br><small>This will shutdown all the "
  754. "running applications within this Qube.</small>").format(
  755. vm.name), QtGui.QMessageBox.Yes | QtGui.QMessageBox.Cancel)
  756. if reply == QtGui.QMessageBox.Yes:
  757. self.shutdown_vm(vm)
  758. def shutdown_vm(self, vm, shutdown_time=vm_shutdown_timeout,
  759. check_time=vm_restart_check_timeout, and_restart=False):
  760. try:
  761. vm.shutdown()
  762. except exc.QubesException as ex:
  763. QtGui.QMessageBox.warning(
  764. None,
  765. self.tr("Error shutting down Qube!"),
  766. self.tr("ERROR: {0}").format(ex))
  767. return
  768. self.shutdown_monitor[vm.qid] = VmShutdownMonitor(vm, shutdown_time,
  769. check_time,
  770. and_restart, self)
  771. # noinspection PyCallByClass,PyTypeChecker
  772. QtCore.QTimer.singleShot(check_time, self.shutdown_monitor[
  773. vm.qid].check_if_vm_has_shutdown)
  774. # noinspection PyArgumentList
  775. @QtCore.pyqtSlot(name='on_action_restartvm_triggered')
  776. def action_restartvm_triggered(self):
  777. vm = self.get_selected_vm()
  778. reply = QtGui.QMessageBox.question(
  779. None, self.tr("Qube Restart Confirmation"),
  780. self.tr("Are you sure you want to restart the Qube <b>'{0}'</b>?"
  781. "<br><small>This will shutdown all the running "
  782. "applications within this Qube.</small>").format(vm.name),
  783. QtGui.QMessageBox.Yes | QtGui.QMessageBox.Cancel)
  784. if reply == QtGui.QMessageBox.Yes:
  785. # in case the user shut down the VM in the meantime
  786. if vm.is_running():
  787. self.shutdown_vm(vm, and_restart=True)
  788. else:
  789. self.start_vm(vm)
  790. # noinspection PyArgumentList
  791. @QtCore.pyqtSlot(name='on_action_killvm_triggered')
  792. def action_killvm_triggered(self):
  793. vm = self.get_selected_vm()
  794. if not (vm.is_running() or vm.is_paused()):
  795. info = self.tr("Qube <b>'{0}'</b> is not running. Are you "
  796. "absolutely sure you want to try to kill it?<br>"
  797. "<small>This will end <b>(not shutdown!)</b> all "
  798. "the running applications within this "
  799. "Qube.</small>").format(vm.name)
  800. else:
  801. info = self.tr("Are you sure you want to kill the Qube "
  802. "<b>'{0}'</b>?<br><small>This will end <b>(not "
  803. "shutdown!)</b> all the running applications within "
  804. "this Qube.</small>").format(vm.name)
  805. reply = QtGui.QMessageBox.question(
  806. None, self.tr("Qube Kill Confirmation"), info,
  807. QtGui.QMessageBox.Yes | QtGui.QMessageBox.Cancel,
  808. QtGui.QMessageBox.Cancel)
  809. if reply == QtGui.QMessageBox.Yes:
  810. try:
  811. vm.kill()
  812. except exc.QubesException as ex:
  813. QtGui.QMessageBox.critical(
  814. None, self.tr("Error while killing Qube!"),
  815. self.tr(
  816. "<b>An exception ocurred while killing {0}.</b><br>"
  817. "ERROR: {1}").format(vm.name, ex))
  818. return
  819. # noinspection PyArgumentList
  820. @QtCore.pyqtSlot(name='on_action_settings_triggered')
  821. def action_settings_triggered(self):
  822. vm = self.get_selected_vm()
  823. if vm:
  824. settings_window = settings.VMSettingsWindow(
  825. vm, self.qt_app, "basic")
  826. settings_window.exec_()
  827. vm_deleted = False
  828. try:
  829. # the VM might not exist after running Settings - it might
  830. # have been cloned or removed
  831. self.vms_in_table[vm.qid].update()
  832. except exc.QubesException:
  833. # TODO: this will be replaced by proper signal handling once
  834. # settings are migrated to AdminAPI
  835. vm_deleted = True
  836. if vm_deleted:
  837. for row in self.vms_in_table:
  838. try:
  839. self.vms_in_table[row].update()
  840. except exc.QubesException:
  841. pass
  842. # noinspection PyArgumentList
  843. @QtCore.pyqtSlot(name='on_action_appmenus_triggered')
  844. def action_appmenus_triggered(self):
  845. vm = self.get_selected_vm()
  846. if vm:
  847. settings_window = settings.VMSettingsWindow(
  848. vm, self.qt_app, "applications")
  849. settings_window.exec_()
  850. # noinspection PyArgumentList
  851. @QtCore.pyqtSlot(name='on_action_updatevm_triggered')
  852. def action_updatevm_triggered(self):
  853. vm = self.get_selected_vm()
  854. if not vm.is_running():
  855. reply = QtGui.QMessageBox.question(
  856. None, self.tr("Qube Update Confirmation"),
  857. self.tr(
  858. "<b>{0}</b><br>The Qube has to be running to be updated."
  859. "<br>Do you want to start it?<br>").format(vm.name),
  860. QtGui.QMessageBox.Yes | QtGui.QMessageBox.Cancel)
  861. if reply != QtGui.QMessageBox.Yes:
  862. return
  863. thread = UpdateVMThread(vm)
  864. self.threads_list.append(thread)
  865. thread.finished.connect(self.clear_threads)
  866. thread.start()
  867. # noinspection PyArgumentList
  868. @QtCore.pyqtSlot(name='on_action_run_command_in_vm_triggered')
  869. def action_run_command_in_vm_triggered(self):
  870. # pylint: disable=invalid-name
  871. vm = self.get_selected_vm()
  872. (command_to_run, ok) = QtGui.QInputDialog.getText(
  873. self, self.tr('Qubes command entry'),
  874. self.tr('Run command in <b>{}</b>:').format(vm.name))
  875. if not ok or command_to_run == "":
  876. return
  877. thread = RunCommandThread(vm, command_to_run)
  878. self.threads_list.append(thread)
  879. thread.finished.connect(self.clear_threads)
  880. thread.start()
  881. # noinspection PyArgumentList
  882. @QtCore.pyqtSlot(name='on_action_set_keyboard_layout_triggered')
  883. def action_set_keyboard_layout_triggered(self):
  884. # pylint: disable=invalid-name
  885. vm = self.get_selected_vm()
  886. vm.run('qubes-change-keyboard-layout')
  887. # noinspection PyArgumentList
  888. @QtCore.pyqtSlot(name='on_action_editfwrules_triggered')
  889. def action_editfwrules_triggered(self):
  890. vm = self.get_selected_vm()
  891. settings_window = settings.VMSettingsWindow(vm, self.qt_app, "firewall")
  892. settings_window.exec_()
  893. # noinspection PyArgumentList
  894. @QtCore.pyqtSlot(name='on_action_global_settings_triggered')
  895. def action_global_settings_triggered(self): # pylint: disable=invalid-name
  896. global_settings_window = global_settings.GlobalSettingsWindow(
  897. self.qt_app,
  898. self.qubes_app)
  899. global_settings_window.exec_()
  900. # noinspection PyArgumentList
  901. @QtCore.pyqtSlot(name='on_action_manage_templates_triggered')
  902. def action_manage_templates_triggered(self):
  903. # pylint: disable=invalid-name, no-self-use
  904. subprocess.check_call('qubes-template-manager')
  905. # noinspection PyArgumentList
  906. @QtCore.pyqtSlot(name='on_action_show_network_triggered')
  907. def action_show_network_triggered(self):
  908. pass
  909. # TODO: revive for 4.1
  910. # network_notes_dialog = NetworkNotesDialog()
  911. # network_notes_dialog.exec_()
  912. # noinspection PyArgumentList
  913. @QtCore.pyqtSlot(name='on_action_restore_triggered')
  914. def action_restore_triggered(self):
  915. restore_window = restore.RestoreVMsWindow(self.qt_app, self.qubes_app)
  916. restore_window.exec_()
  917. # noinspection PyArgumentList
  918. @QtCore.pyqtSlot(name='on_action_backup_triggered')
  919. def action_backup_triggered(self):
  920. backup_window = backup.BackupVMsWindow(self.qt_app, self.qubes_app,
  921. self.dispatcher, self)
  922. backup_window.show()
  923. # noinspection PyArgumentList
  924. @QtCore.pyqtSlot(name='on_action_exit_triggered')
  925. def action_exit_triggered(self):
  926. self.close()
  927. def showhide_menubar(self, checked):
  928. self.menubar.setVisible(checked)
  929. if not checked:
  930. self.context_menu.addAction(self.action_menubar)
  931. else:
  932. self.context_menu.removeAction(self.action_menubar)
  933. if self.settings_loaded:
  934. self.manager_settings.setValue('view/menubar_visible', checked)
  935. self.manager_settings.sync()
  936. def showhide_toolbar(self, checked):
  937. self.toolbar.setVisible(checked)
  938. if not checked:
  939. self.context_menu.addAction(self.action_toolbar)
  940. else:
  941. self.context_menu.removeAction(self.action_toolbar)
  942. if self.settings_loaded:
  943. self.manager_settings.setValue('view/toolbar_visible', checked)
  944. self.manager_settings.sync()
  945. def showhide_column(self, col_num, show):
  946. self.table.setColumnHidden(col_num, not show)
  947. val = 1 if show else -1
  948. self.visible_columns_count += val
  949. if self.visible_columns_count == 1:
  950. # disable hiding the last one
  951. for col in self.columns_actions:
  952. if self.columns_actions[col].isChecked():
  953. self.columns_actions[col].setEnabled(False)
  954. break
  955. elif self.visible_columns_count == 2 and val == 1:
  956. # enable hiding previously disabled column
  957. for col in self.columns_actions:
  958. if not self.columns_actions[col].isEnabled():
  959. self.columns_actions[col].setEnabled(True)
  960. break
  961. if self.settings_loaded:
  962. col_name = [name for name in self.columns_indices if
  963. self.columns_indices[name] == col_num][0]
  964. self.manager_settings.setValue('columns/%s' % col_name, show)
  965. self.manager_settings.sync()
  966. def on_action_vm_type_toggled(self, checked):
  967. self.showhide_column(self.columns_indices['Type'], checked)
  968. def on_action_label_toggled(self, checked):
  969. self.showhide_column(self.columns_indices['Label'], checked)
  970. def on_action_name_toggled(self, checked):
  971. self.showhide_column(self.columns_indices['Name'], checked)
  972. def on_action_state_toggled(self, checked):
  973. self.showhide_column(self.columns_indices['State'], checked)
  974. def on_action_internal_toggled(self, checked):
  975. self.showhide_column(self.columns_indices['Internal'], checked)
  976. def on_action_ip_toggled(self, checked):
  977. self.showhide_column(self.columns_indices['IP'], checked)
  978. def on_action_backups_toggled(self, checked):
  979. self.showhide_column(self.columns_indices['Backups'], checked)
  980. def on_action_last_backup_toggled(self, checked):
  981. self.showhide_column(self.columns_indices['Last backup'], checked)
  982. def on_action_template_toggled(self, checked):
  983. self.showhide_column(self.columns_indices['Template'], checked)
  984. def on_action_netvm_toggled(self, checked):
  985. self.showhide_column(self.columns_indices['NetVM'], checked)
  986. def on_action_size_on_disk_toggled(self, checked):
  987. self.showhide_column(self.columns_indices['Size'], checked)
  988. # noinspection PyArgumentList
  989. @QtCore.pyqtSlot(name='on_action_about_qubes_triggered')
  990. def action_about_qubes_triggered(self): # pylint: disable=no-self-use
  991. about = AboutDialog()
  992. about.exec_()
  993. def createPopupMenu(self): # pylint: disable=invalid-name
  994. menu = QtGui.QMenu()
  995. menu.addAction(self.action_toolbar)
  996. menu.addAction(self.action_menubar)
  997. return menu
  998. def open_tools_context_menu(self, widget, point):
  999. self.tools_context_menu.exec_(widget.mapToGlobal(point))
  1000. @QtCore.pyqtSlot('const QPoint&')
  1001. def open_context_menu(self, point):
  1002. try:
  1003. vm = self.get_selected_vm()
  1004. # logs menu
  1005. self.logs_menu.clear()
  1006. if vm.qid == 0:
  1007. logfiles = ["/var/log/xen/console/hypervisor.log"]
  1008. else:
  1009. logfiles = [
  1010. "/var/log/xen/console/guest-" + vm.name + ".log",
  1011. "/var/log/xen/console/guest-" + vm.name + "-dm.log",
  1012. "/var/log/qubes/guid." + vm.name + ".log",
  1013. "/var/log/qubes/qrexec." + vm.name + ".log",
  1014. ]
  1015. menu_empty = True
  1016. for logfile in logfiles:
  1017. if os.path.exists(logfile):
  1018. action = self.logs_menu.addAction(QtGui.QIcon(":/log.png"),
  1019. logfile)
  1020. action.setData(logfile)
  1021. menu_empty = False
  1022. self.logs_menu.setEnabled(not menu_empty)
  1023. if vm.qid == 0:
  1024. self.dom0_context_menu.exec_(self.table.mapToGlobal(
  1025. point + QtCore.QPoint(10, 0)))
  1026. else:
  1027. self.context_menu.exec_(self.table.mapToGlobal(
  1028. point + QtCore.QPoint(10, 0)))
  1029. except exc.QubesPropertyAccessError:
  1030. pass
  1031. @QtCore.pyqtSlot('QAction *')
  1032. def show_log(self, action):
  1033. log = str(action.data())
  1034. log_dlg = log_dialog.LogDialog(self.qt_app, log)
  1035. log_dlg.exec_()
  1036. # Bases on the original code by:
  1037. # Copyright (c) 2002-2007 Pascal Varet <p.varet@gmail.com>
  1038. def handle_exception(exc_type, exc_value, exc_traceback):
  1039. filename, line, dummy, dummy = traceback.extract_tb(exc_traceback).pop()
  1040. filename = os.path.basename(filename)
  1041. error = "%s: %s" % (exc_type.__name__, exc_value)
  1042. strace = ""
  1043. stacktrace = traceback.extract_tb(exc_traceback)
  1044. while stacktrace:
  1045. (filename, line, func, txt) = stacktrace.pop()
  1046. strace += "----\n"
  1047. strace += "line: %s\n" % txt
  1048. strace += "func: %s\n" % func
  1049. strace += "line no.: %d\n" % line
  1050. strace += "file: %s\n" % filename
  1051. msg_box = QtGui.QMessageBox()
  1052. msg_box.setDetailedText(strace)
  1053. msg_box.setIcon(QtGui.QMessageBox.Critical)
  1054. msg_box.setWindowTitle("Houston, we have a problem...")
  1055. msg_box.setText("Whoops. A critical error has occured. "
  1056. "This is most likely a bug in Qubes Manager.<br><br>"
  1057. "<b><i>%s</i></b>" % error +
  1058. "<br/>at line <b>%d</b><br/>of file %s.<br/><br/>"
  1059. % (line, filename))
  1060. msg_box.exec_()
  1061. def loop_shutdown():
  1062. pending = asyncio.Task.all_tasks()
  1063. for task in pending:
  1064. with suppress(asyncio.CancelledError):
  1065. task.cancel()
  1066. def main():
  1067. qt_app = QtGui.QApplication(sys.argv)
  1068. qt_app.setOrganizationName("The Qubes Project")
  1069. qt_app.setOrganizationDomain("http://qubes-os.org")
  1070. qt_app.setApplicationName("Qube Manager")
  1071. qt_app.setWindowIcon(QtGui.QIcon.fromTheme("qubes-manager"))
  1072. qt_app.lastWindowClosed.connect(loop_shutdown)
  1073. qubes_app = Qubes()
  1074. loop = quamash.QEventLoop(qt_app)
  1075. asyncio.set_event_loop(loop)
  1076. dispatcher = events.EventsDispatcher(qubes_app)
  1077. manager_window = VmManagerWindow(qt_app, qubes_app, dispatcher)
  1078. manager_window.show()
  1079. try:
  1080. loop.run_until_complete(
  1081. asyncio.ensure_future(dispatcher.listen_for_events()))
  1082. except asyncio.CancelledError:
  1083. pass
  1084. except Exception: # pylint: disable=broad-except
  1085. loop_shutdown()
  1086. exc_type, exc_value, exc_traceback = sys.exc_info()[:3]
  1087. handle_exception(exc_type, exc_value, exc_traceback)
  1088. if __name__ == "__main__":
  1089. main()