qube_manager.py 62 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594
  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 subprocess
  25. from datetime import datetime, timedelta
  26. from functools import partial
  27. from os import path
  28. from qubesadmin import exc
  29. from qubesadmin import utils
  30. # pylint: disable=import-error
  31. from PyQt5.QtCore import (Qt, QAbstractTableModel, QObject, pyqtSlot, QEvent,
  32. QSettings, QRegExp, QSortFilterProxyModel, QSize, QPoint, QTimer)
  33. # pylint: disable=import-error
  34. from PyQt5.QtWidgets import (QLineEdit, QStyledItemDelegate, QToolTip,
  35. QMenu, QInputDialog, QMainWindow, QProgressDialog, QStyleOptionViewItem,
  36. QMessageBox)
  37. # pylint: disable=import-error
  38. from PyQt5.QtGui import (QIcon, QPixmap, QRegExpValidator, QFont, QColor)
  39. from qubesmanager.about import AboutDialog
  40. from . import ui_qubemanager # pylint: disable=no-name-in-module
  41. from . import settings
  42. from . import global_settings
  43. from . import restore
  44. from . import backup
  45. from . import create_new_vm
  46. from . import log_dialog
  47. from . import utils as manager_utils
  48. from . import common_threads
  49. from . import clone_vm
  50. class SearchBox(QLineEdit):
  51. def __init__(self, parent=None):
  52. super().__init__(parent)
  53. self.focusing = False
  54. def focusInEvent(self, e): # pylint: disable=invalid-name
  55. super().focusInEvent(e)
  56. self.selectAll()
  57. self.focusing = True
  58. def mousePressEvent(self, e): # pylint: disable=invalid-name
  59. super().mousePressEvent(e)
  60. if self.focusing:
  61. self.selectAll()
  62. self.focusing = False
  63. icon_size = QSize(22, 22)
  64. # pylint: disable=invalid-name
  65. class StateIconDelegate(QStyledItemDelegate):
  66. lastIndex = None
  67. def __init__(self):
  68. super().__init__()
  69. self.stateIcons = {
  70. "Running" : QIcon(":/on.png"),
  71. "Paused" : QIcon(":/paused.png"),
  72. "Suspended" : QIcon(":/paused.png"),
  73. "Transient" : QIcon(":/transient.png"),
  74. "Halting" : QIcon(":/transient.png"),
  75. "Dying" : QIcon(":/transient.png"),
  76. "Halted" : QIcon(":/off.png")
  77. }
  78. self.outdatedIcons = {
  79. "update" : QIcon(":/update-recommended.png"),
  80. "outdated" : QIcon(":/outdated.png"),
  81. "to-be-outdated" : QIcon(":/to-be-outdated.png"),
  82. }
  83. self.outdatedTooltips = {
  84. "update" : self.tr("Updates pending!"),
  85. "outdated" : self.tr(
  86. "The qube must be restarted for its filesystem to reflect"
  87. " the template's recent committed changes."),
  88. "to-be-outdated" : self.tr(
  89. "The Template must be stopped before changes from its "
  90. "current session can be picked up by this qube."),
  91. }
  92. def sizeHint(self, option, index):
  93. hint = super().sizeHint(option, index)
  94. option = QStyleOptionViewItem(option)
  95. option.features |= option.HasDecoration
  96. widget = option.widget
  97. style = widget.style()
  98. iconRect = style.subElementRect(style.SE_ItemViewItemDecoration,
  99. option, widget)
  100. width = iconRect.width() * 3 # Nº of possible icons
  101. hint.setWidth(width)
  102. return hint
  103. def paint(self, qp, option, index):
  104. # create a new QStyleOption (*never* use the one given in arguments)
  105. option = QStyleOptionViewItem(option)
  106. widget = option.widget
  107. style = widget.style()
  108. # paint the base item (borders, gradients, selection colors, etc)
  109. style.drawControl(style.CE_ItemViewItem, option, qp, widget)
  110. # "lie" about the decoration, to get a valid icon rectangle (even if we
  111. # don't have any "real" icon set for the item)
  112. option.features |= option.HasDecoration
  113. iconRect = style.subElementRect(style.SE_ItemViewItemDecoration,
  114. option, widget)
  115. iconSize = iconRect.size()
  116. margin = iconRect.left() - option.rect.left()
  117. qp.save()
  118. # ensure that we do not draw outside the item rectangle (and add some
  119. # fancy margin on the right
  120. qp.setClipRect(option.rect.adjusted(0, 0, -margin, 0))
  121. # draw the main state icon, assuming all items have one
  122. qp.drawPixmap(iconRect,
  123. self.stateIcons[index.data()['power']].pixmap(iconSize))
  124. left = delta = margin + iconRect.width()
  125. if index.data()['outdated']:
  126. qp.drawPixmap(iconRect.translated(left, 0),
  127. self.outdatedIcons[index.data()['outdated']]\
  128. .pixmap(iconSize))
  129. left += delta
  130. qp.restore()
  131. def helpEvent(self, event, view, option, index):
  132. if event.type() != QEvent.ToolTip:
  133. return super().helpEvent(event, view,
  134. option, index)
  135. option = QStyleOptionViewItem(option)
  136. widget = option.widget
  137. style = widget.style()
  138. option.features |= option.HasDecoration
  139. iconRect = style.subElementRect(style.SE_ItemViewItemDecoration,
  140. option, widget)
  141. iconRect.setTop(option.rect.y())
  142. iconRect.setHeight(option.rect.height())
  143. # similar to what we do in the paint() method
  144. if event.pos() in iconRect:
  145. # (*) clear any existing tooltip; a single space is better , as
  146. # sometimes it's not enough to use an empty string
  147. if index != self.lastIndex:
  148. QToolTip.showText(QPoint(), ' ')
  149. QToolTip.showText(event.globalPos(),
  150. index.data()['power'], view)
  151. else:
  152. margin = iconRect.left() - option.rect.left()
  153. left = delta = margin + iconRect.width()
  154. if index.data()['outdated']:
  155. if event.pos() in iconRect.translated(left, 0):
  156. # see above (*)
  157. if index != self.lastIndex:
  158. QToolTip.showText(QPoint(), ' ')
  159. QToolTip.showText(event.globalPos(),
  160. self.outdatedTooltips[index.data()['outdated']],
  161. view)
  162. # shift the left *only* if the role is True, otherwise we
  163. # can assume that that icon doesn't exist at all
  164. left += delta
  165. self.lastIndex = index
  166. return True
  167. # pylint: disable=too-many-instance-attributes
  168. # pylint: disable=too-few-public-methods
  169. class VmInfo():
  170. def __init__(self, vm):
  171. self.vm = vm
  172. self.qid = vm.qid
  173. self.name = self.vm.name
  174. self.label = getattr(self.vm, 'label', None)
  175. self.klass = getattr(self.vm, 'klass', None)
  176. self.icon = getattr(vm, 'icon', 'appvm-black')
  177. self.state = {'power': "", 'outdated': ""}
  178. self.updateable = getattr(vm, 'updateable', False)
  179. self.update(True)
  180. def update_power_state(self):
  181. try:
  182. self.state['power'] = self.vm.get_power_state()
  183. except exc.QubesDaemonAccessError:
  184. self.state['power'] = ""
  185. self.state['outdated'] = ""
  186. try:
  187. if manager_utils.is_running(self.vm, False):
  188. if hasattr(self.vm, 'template') and \
  189. manager_utils.is_running(self.vm.template, False):
  190. self.state['outdated'] = "to-be-outdated"
  191. else:
  192. try:
  193. if any(vol.is_outdated()
  194. for vol in self.vm.volumes.values()):
  195. self.state['outdated'] = "outdated"
  196. except exc.QubesDaemonAccessError:
  197. pass
  198. if self.vm.klass in {'TemplateVM', 'StandaloneVM'} and \
  199. manager_utils.get_feature(
  200. self.vm, 'updates-available', False):
  201. self.state['outdated'] = 'update'
  202. except exc.QubesDaemonAccessError:
  203. pass
  204. def update(self, update_size_on_disk=False, event=None):
  205. """
  206. Update VmInfo
  207. :param update_size_on_disk: should disk utilization be updated?
  208. :param event: name of the event that caused the update, to avoid
  209. updating unnecessary properties; if event is none, update everything
  210. :return: None
  211. """
  212. self.update_power_state()
  213. if not event or event.endswith(':label'):
  214. self.label = getattr(self.vm, 'label', None)
  215. self.icon = getattr(self.vm, 'icon', 'appvm-black')
  216. if not event or event.endswith(':template'):
  217. try:
  218. self.template = self.vm.template.name
  219. except AttributeError:
  220. self.template = None
  221. if not event or event.endswith(':netvm'):
  222. self.netvm = getattr(self.vm, 'netvm', None)
  223. if self.netvm:
  224. self.netvm = str(self.netvm)
  225. else:
  226. self.netvm = "n/a"
  227. try:
  228. if hasattr(self.vm, 'netvm') \
  229. and self.vm.property_is_default("netvm"):
  230. self.netvm = "default (" + self.netvm + ")"
  231. except exc.QubesDaemonAccessError:
  232. pass
  233. if not event or event.endswith(':internal'):
  234. self.internal = manager_utils.get_boolean_feature(
  235. self.vm, 'internal')
  236. if not event or event.endswith(':ip'):
  237. self.ip = getattr(self.vm, 'ip', "n/a")
  238. if not event or event.endswith(':include_in_backups'):
  239. self.inc_backup = getattr(self.vm, 'include_in_backups', None)
  240. if not event or event.endswith(':backup_timestamp'):
  241. self.last_backup = getattr(self.vm, 'backup_timestamp', None)
  242. if self.last_backup:
  243. self.last_backup = str(datetime.fromtimestamp(self.last_backup))
  244. if not event or event.endswith(':default_dispvm'):
  245. self.dvm = getattr(self.vm, 'default_dispvm', None)
  246. try:
  247. if self.vm.property_is_default("default_dispvm"):
  248. self.dvm = "default (" + str(self.dvm) + ")"
  249. elif self.dvm is not None:
  250. self.dvm = str(self.dvm)
  251. except exc.QubesDaemonAccessError:
  252. if self.dvm is not None:
  253. self.dvm = str(self.dvm)
  254. if not event or event.endswith(':template_for_dispvms'):
  255. self.dvm_template = getattr(self.vm, 'template_for_dispvms', None)
  256. if self.vm.klass != 'AdminVM' and update_size_on_disk:
  257. try:
  258. self.disk_float = float(self.vm.get_disk_utilization())
  259. self.disk = str(round(self.disk_float/(1024*1024), 2)) + " MiB"
  260. except exc.QubesDaemonAccessError:
  261. self.disk_float = None
  262. self.disk = None
  263. if self.vm.klass != 'AdminVM':
  264. self.virt_mode = getattr(self.vm, 'virt_mode', None)
  265. else:
  266. self.virt_mode = None
  267. self.disk = "n/a"
  268. class QubesCache(QAbstractTableModel):
  269. def __init__(self, qubes_app):
  270. QAbstractTableModel.__init__(self)
  271. self._qubes_app = qubes_app
  272. self._info_list = []
  273. self._info_by_id = {}
  274. def add_vm(self, vm):
  275. vm_info = VmInfo(vm)
  276. self._info_list.append(vm_info)
  277. self._info_by_id[vm.qid] = vm_info
  278. def remove_vm(self, name):
  279. vm_info = self.get_vm(name=name)
  280. self._info_list.remove(vm_info)
  281. del self._info_by_id[vm_info.qid]
  282. def get_vm(self, row=None, qid=None, name=None):
  283. if row is not None:
  284. return self._info_list[row]
  285. if qid is not None:
  286. return self._info_by_id[qid]
  287. return next(x for x in self._info_list if x.name == name)
  288. def __len__(self):
  289. return len(self._info_list)
  290. def __iter__(self):
  291. return iter(self._info_list)
  292. class QubesTableModel(QAbstractTableModel):
  293. def __init__(self, qubes_cache):
  294. QAbstractTableModel.__init__(self)
  295. self.qubes_cache = qubes_cache
  296. self.template = {}
  297. self.klass_pixmap = {}
  298. self.label_pixmap = {}
  299. self.columns_indices = [
  300. "Type",
  301. "Label",
  302. "Name",
  303. "State",
  304. "Template",
  305. "NetVM",
  306. "Disk Usage",
  307. "Internal",
  308. "IP",
  309. "Backup",
  310. "Last backup",
  311. "Default DispVM",
  312. "Is DVM Template",
  313. "Virt Mode"
  314. ]
  315. # pylint: disable=invalid-name
  316. def rowCount(self, _):
  317. return len(self.qubes_cache)
  318. # pylint: disable=invalid-name
  319. def columnCount(self, _):
  320. return len(self.columns_indices)
  321. # pylint: disable=too-many-return-statements
  322. def data(self, index, role):
  323. if not index.isValid():
  324. return None
  325. col = index.column()
  326. row = index.row()
  327. col_name = self.columns_indices[col]
  328. vm = self.qubes_cache.get_vm(row)
  329. if role == Qt.DisplayRole:
  330. if col in [0, 1]:
  331. return None
  332. if col_name == "Name":
  333. return vm.name
  334. if col_name == "State":
  335. return vm.state
  336. if col_name == "Template":
  337. if vm.template is None:
  338. return vm.klass
  339. return vm.template
  340. if col_name == "NetVM":
  341. return vm.netvm
  342. if col_name == "Disk Usage":
  343. return vm.disk
  344. if col_name == "Internal":
  345. return "Yes" if vm.internal else ""
  346. if col_name == "IP":
  347. return vm.ip
  348. if col_name == "Last backup":
  349. return vm.last_backup
  350. if col_name == "Default DispVM":
  351. return vm.dvm
  352. if col_name == "Is DVM Template":
  353. return "Yes" if vm.dvm_template else ""
  354. if col_name == "Virt Mode":
  355. return vm.virt_mode
  356. if role == Qt.DecorationRole:
  357. if col_name == "Type":
  358. try:
  359. return self.klass_pixmap[vm.klass]
  360. except KeyError:
  361. pixmap = QPixmap()
  362. icon_name = ":/"+vm.klass.lower()+".png"
  363. icon_name = icon_name.replace("adminvm", "dom0")
  364. icon_name = icon_name.replace("dispvm", "appvm")
  365. pixmap.load(icon_name)
  366. self.klass_pixmap[vm.klass] = pixmap.scaled(icon_size)
  367. return self.klass_pixmap[vm.klass]
  368. except exc.QubesDaemonAccessError:
  369. return None
  370. if col_name == "Label":
  371. try:
  372. return self.label_pixmap[vm.icon]
  373. except (KeyError, AttributeError):
  374. icon = QIcon.fromTheme(vm.icon)
  375. self.label_pixmap[vm.icon] = icon.pixmap(icon_size)
  376. return self.label_pixmap[vm.icon]
  377. if role == Qt.CheckStateRole:
  378. if col_name == "Backup":
  379. return Qt.Checked if vm.inc_backup else Qt.Unchecked
  380. if role == Qt.FontRole:
  381. if col_name == "Template":
  382. if vm.template is None:
  383. font = QFont()
  384. font.setItalic(True)
  385. return font
  386. if role == Qt.ForegroundRole:
  387. if col_name == "Template":
  388. if vm.template is None:
  389. return QColor("gray")
  390. # Used for get VM Object
  391. if role == Qt.UserRole:
  392. return vm
  393. # Used for sorting
  394. if role == Qt.UserRole + 1:
  395. if vm.klass == 'AdminVM':
  396. return ""
  397. if col_name == "Type":
  398. return vm.klass
  399. if col_name == "Label":
  400. return str(vm.label)
  401. if col_name == "State":
  402. return str(vm.state)
  403. if col_name == "Disk Usage":
  404. return vm.disk_float
  405. return self.data(index, Qt.DisplayRole)
  406. # pylint: disable=invalid-name
  407. def headerData(self, col, orientation, role):
  408. if col < 2:
  409. return None
  410. if orientation == Qt.Horizontal and role == Qt.DisplayRole:
  411. return self.columns_indices[col]
  412. return None
  413. def setData(self, index, value, role=Qt.EditRole):
  414. if not index.isValid():
  415. return False
  416. if role == Qt.CheckStateRole:
  417. col_name = self.columns_indices[index.column()]
  418. if col_name == "Backup":
  419. vm = self.qubes_cache.get_vm(index.row())
  420. vm.vm.include_in_backups = (value == Qt.Checked)
  421. vm.inc_backup = (value == Qt.Checked)
  422. return True
  423. return False
  424. def flags(self, index):
  425. if not index.isValid():
  426. return False
  427. def_flags = QAbstractTableModel.flags(self, index)
  428. if self.columns_indices[index.column()] == "Backup":
  429. return def_flags | Qt.ItemIsUserCheckable
  430. return def_flags
  431. vm_shutdown_timeout = 20000 # in msec
  432. vm_restart_check_timeout = 1000 # in msec
  433. class VmShutdownMonitor(QObject):
  434. def __init__(self, vm, shutdown_time=vm_shutdown_timeout,
  435. check_time=vm_restart_check_timeout,
  436. and_restart=False, caller=None):
  437. QObject.__init__(self)
  438. self.vm = vm
  439. self.shutdown_time = shutdown_time
  440. self.check_time = check_time
  441. self.and_restart = and_restart
  442. self.shutdown_started = datetime.now()
  443. self.caller = caller
  444. def restart_vm_if_needed(self):
  445. if self.and_restart and self.caller:
  446. self.caller.start_vm(self.vm)
  447. def check_again_later(self):
  448. # noinspection PyTypeChecker,PyCallByClass
  449. QTimer.singleShot(self.check_time, self.check_if_vm_has_shutdown)
  450. def timeout_reached(self):
  451. actual = datetime.now() - self.shutdown_started
  452. allowed = timedelta(milliseconds=self.shutdown_time)
  453. return actual > allowed
  454. def check_if_vm_has_shutdown(self):
  455. vm = self.vm
  456. vm_is_running = manager_utils.is_running(vm, False)
  457. try:
  458. vm_start_time = datetime.fromtimestamp(float(vm.start_time))
  459. except (AttributeError, TypeError, ValueError):
  460. vm_start_time = None
  461. if vm_is_running and vm_start_time \
  462. and vm_start_time < self.shutdown_started:
  463. if self.timeout_reached():
  464. msgbox = QMessageBox(self.caller)
  465. msgbox.setIcon(QMessageBox.Question)
  466. msgbox.setWindowTitle(self.tr("Qube Shutdown"))
  467. msgbox.setText(self.tr(
  468. "The Qube <b>'{0}'</b> hasn't shutdown within the last "
  469. "{1} seconds, do you want to kill it?<br>").format(
  470. vm.name, self.shutdown_time / 1000))
  471. kill_button = msgbox.addButton(
  472. self.tr("Kill it!"), QMessageBox.YesRole)
  473. wait_button = msgbox.addButton(
  474. self.tr("Wait another {0} seconds...").format(
  475. self.shutdown_time / 1000),
  476. QMessageBox.NoRole)
  477. ignore_button = msgbox.addButton(self.tr("Don't ask again"),
  478. QMessageBox.RejectRole)
  479. msgbox.setDefaultButton(wait_button)
  480. msgbox.setEscapeButton(ignore_button)
  481. msgbox.setWindowFlags(
  482. msgbox.windowFlags() | Qt.CustomizeWindowHint)
  483. msgbox.setWindowFlags(
  484. msgbox.windowFlags() & ~Qt.WindowCloseButtonHint)
  485. msgbox.exec_()
  486. msgbox.deleteLater()
  487. if msgbox.clickedButton() is kill_button:
  488. try:
  489. vm.kill()
  490. except exc.QubesVMNotStartedError:
  491. # the VM shut down while the user was thinking about
  492. # shutting it down
  493. pass
  494. self.restart_vm_if_needed()
  495. elif msgbox.clickedButton() is ignore_button:
  496. return
  497. else:
  498. self.shutdown_started = datetime.now()
  499. self.check_again_later()
  500. else:
  501. self.check_again_later()
  502. else:
  503. if vm_is_running:
  504. # Due to unknown reasons, Xen sometimes reports that a domain
  505. # is running even though its start-up timestamp is not valid.
  506. # Make sure that "restart_vm_if_needed" is not called until
  507. # the domain has been completely shut down according to Xen.
  508. self.check_again_later()
  509. return
  510. self.restart_vm_if_needed()
  511. # pylint: disable=too-few-public-methods
  512. class StartVMThread(common_threads.QubesThread):
  513. def run(self):
  514. try:
  515. self.vm.start()
  516. except exc.QubesException as ex:
  517. self.msg = ("Error starting Qube!", str(ex))
  518. # pylint: disable=too-few-public-methods
  519. class UpdateVMThread(common_threads.QubesThread):
  520. def run(self):
  521. try:
  522. if self.vm.klass == 'AdminVM':
  523. subprocess.check_call(
  524. ["/usr/bin/qubes-dom0-update", "--clean", "--gui"])
  525. else:
  526. if not manager_utils.is_running(self.vm, False):
  527. try:
  528. self.vm.start()
  529. except exc.QubesDaemonAccessError:
  530. # permission denied, let us hope for the best
  531. pass
  532. # apply DSA-4371
  533. with open('/usr/libexec/qubes-manager/dsa-4371-update', 'rb') \
  534. as dsa4371update:
  535. stdout, stderr = self.vm.run_service_for_stdio(
  536. "qubes.VMShell",
  537. user="root",
  538. input=dsa4371update.read())
  539. if stdout == b'changed=yes\n':
  540. subprocess.call(
  541. ['notify-send', '-i', 'dialog-information',
  542. self.tr('Debian DSA-4371 fix installed in {}').format(
  543. self.vm.name)])
  544. elif stdout == b'changed=no\n':
  545. pass
  546. else:
  547. raise exc.QubesException(
  548. self.tr("Failed to apply DSA-4371 fix: {}").format(
  549. stderr.decode('ascii')))
  550. self.vm.run_service("qubes.InstallUpdatesGUI",
  551. user="root", wait=False)
  552. except (ChildProcessError, exc.QubesException) as ex:
  553. self.msg = (self.tr("Error on qube update!"), str(ex))
  554. # pylint: disable=too-few-public-methods
  555. class RunCommandThread(common_threads.QubesThread):
  556. def __init__(self, vm, command_to_run):
  557. super().__init__(vm)
  558. self.command_to_run = command_to_run
  559. def run(self):
  560. try:
  561. self.vm.run(self.command_to_run)
  562. except (ChildProcessError, exc.QubesException) as ex:
  563. self.msg = (self.tr("Error while running command!"), str(ex))
  564. class QubesProxyModel(QSortFilterProxyModel):
  565. def __init__(self, window):
  566. super().__init__()
  567. self.window = window
  568. def lessThan(self, left, right):
  569. if left.data(self.sortRole()) != right.data(self.sortRole()):
  570. return super().lessThan(left, right)
  571. left_vm = left.data(Qt.UserRole)
  572. right_vm = right.data(Qt.UserRole)
  573. return left_vm.name.lower() < right_vm.name.lower()
  574. # pylint: disable=too-many-return-statements
  575. def filterAcceptsRow(self, sourceRow, sourceParent):
  576. if self.window.show_all.isChecked():
  577. return super().filterAcceptsRow(sourceRow, sourceParent)
  578. index = self.sourceModel().index(sourceRow, 0, sourceParent)
  579. vm = self.sourceModel().data(index, Qt.UserRole)
  580. if self.window.show_running.isChecked() and \
  581. vm.state['power'] == 'Running':
  582. return super().filterAcceptsRow(sourceRow, sourceParent)
  583. if self.window.show_halted.isChecked() and \
  584. vm.state['power'] == 'Halted':
  585. return super().filterAcceptsRow(sourceRow, sourceParent)
  586. if self.window.show_network.isChecked() and \
  587. getattr(vm.vm, 'provides_network', False):
  588. return super().filterAcceptsRow(sourceRow, sourceParent)
  589. if self.window.show_templates.isChecked() and vm.klass == 'TemplateVM':
  590. return super().filterAcceptsRow(sourceRow, sourceParent)
  591. if self.window.show_standalone.isChecked() \
  592. and vm.klass == 'StandaloneVM':
  593. return super().filterAcceptsRow(sourceRow, sourceParent)
  594. return False
  595. class VmManagerWindow(ui_qubemanager.Ui_VmManagerWindow, QMainWindow):
  596. # suppress saving settings while initializing widgets
  597. settings_loaded = False
  598. def __init__(self, qt_app, qubes_app, dispatcher, _parent=None):
  599. super().__init__()
  600. self.setupUi(self)
  601. self.manager_settings = QSettings(self)
  602. self.qubes_app = qubes_app
  603. self.qt_app = qt_app
  604. self.searchbox = SearchBox()
  605. self.searchbox.setValidator(QRegExpValidator(
  606. QRegExp("[a-zA-Z0-9_-]*", Qt.CaseInsensitive), None))
  607. self.searchbox.textChanged.connect(self.do_search)
  608. self.searchContainer.insertWidget(1, self.searchbox)
  609. self.settings_windows = {}
  610. self.frame_width = 0
  611. self.frame_height = 0
  612. self.__init_context_menu()
  613. self.tools_context_menu = QMenu(self)
  614. self.tools_context_menu.addAction(self.action_toolbar)
  615. self.tools_context_menu.addAction(self.action_menubar)
  616. self.menubar.customContextMenuRequested.connect(
  617. lambda pos: self.open_tools_context_menu(self.menubar, pos))
  618. self.toolbar.customContextMenuRequested.connect(
  619. lambda pos: self.open_tools_context_menu(self.toolbar, pos))
  620. self.action_menubar.toggled.connect(self.showhide_menubar)
  621. self.action_toolbar.toggled.connect(self.showhide_toolbar)
  622. self.action_show_logs.triggered.connect(self.show_log)
  623. self.action_compact_view.toggled.connect(self.set_compactview)
  624. self.table.resizeColumnsToContents()
  625. self.update_size_on_disk = False
  626. self.shutdown_monitor = {}
  627. self.qubes_cache = QubesCache(qubes_app)
  628. self.fill_cache()
  629. self.qubes_model = QubesTableModel(self.qubes_cache)
  630. self.proxy = QubesProxyModel(self)
  631. self.proxy.setSourceModel(self.qubes_model)
  632. self.proxy.setSortRole(Qt.UserRole + 1)
  633. self.proxy.setSortCaseSensitivity(Qt.CaseInsensitive)
  634. self.proxy.setFilterKeyColumn(2)
  635. self.proxy.setFilterCaseSensitivity(Qt.CaseInsensitive)
  636. self.proxy.layoutChanged.connect(self.save_sorting)
  637. self.show_running.stateChanged.connect(self.invalidate)
  638. self.show_halted.stateChanged.connect(self.invalidate)
  639. self.show_network.stateChanged.connect(self.invalidate)
  640. self.show_templates.stateChanged.connect(self.invalidate)
  641. self.show_standalone.stateChanged.connect(self.invalidate)
  642. self.show_all.stateChanged.connect(self.invalidate)
  643. self.table.setModel(self.proxy)
  644. self.table.setItemDelegateForColumn(3, StateIconDelegate())
  645. self.table.resizeColumnsToContents()
  646. selection_model = self.table.selectionModel()
  647. selection_model.selectionChanged.connect(self.table_selection_changed)
  648. self.table.setContextMenuPolicy(Qt.CustomContextMenu)
  649. self.table.customContextMenuRequested.connect(self.open_context_menu)
  650. # Create view menu
  651. for col_no in range(len(self.qubes_model.columns_indices)):
  652. column = self.qubes_model.columns_indices[col_no]
  653. action = self.menu_view.addAction(column)
  654. action.setData(column)
  655. action.setCheckable(True)
  656. action.toggled.connect(partial(self.showhide_column, col_no))
  657. self.menu_view.addSeparator()
  658. self.menu_view.addAction(self.action_toolbar)
  659. self.menu_view.addAction(self.action_menubar)
  660. self.menu_view.addSeparator()
  661. self.menu_view.addAction(self.action_compact_view)
  662. try:
  663. self.load_manager_settings()
  664. except Exception as ex: # pylint: disable=broad-except
  665. QMessageBox.warning(
  666. self,
  667. self.tr("Manager settings unreadable"),
  668. self.tr("Qube Manager settings cannot be parsed. Previously "
  669. "saved display settings may not be restored "
  670. "correctly.\nError: {}".format(str(ex))))
  671. self.settings_loaded = True
  672. # Connect events
  673. self.dispatcher = dispatcher
  674. dispatcher.add_handler('domain-pre-start',
  675. self.on_domain_status_changed)
  676. dispatcher.add_handler('domain-start', self.on_domain_status_changed)
  677. dispatcher.add_handler('domain-start-failed',
  678. self.on_domain_status_changed)
  679. dispatcher.add_handler('domain-stopped', self.on_domain_status_changed)
  680. dispatcher.add_handler('domain-pre-shutdown',
  681. self.on_domain_status_changed)
  682. dispatcher.add_handler('domain-shutdown', self.on_domain_status_changed)
  683. dispatcher.add_handler('domain-paused', self.on_domain_status_changed)
  684. dispatcher.add_handler('domain-unpaused', self.on_domain_status_changed)
  685. dispatcher.add_handler('domain-add', self.on_domain_added)
  686. dispatcher.add_handler('domain-delete', self.on_domain_removed)
  687. dispatcher.add_handler('property-set:*',
  688. self.on_domain_changed)
  689. dispatcher.add_handler('property-del:*',
  690. self.on_domain_changed)
  691. dispatcher.add_handler('property-load',
  692. self.on_domain_changed)
  693. dispatcher.add_handler('domain-feature-set:internal',
  694. self.on_domain_changed)
  695. dispatcher.add_handler('domain-feature-delete:internal',
  696. self.on_domain_changed)
  697. dispatcher.add_handler('domain-feature-set:updates-available',
  698. self.on_domain_updates_available)
  699. dispatcher.add_handler('domain-feature-delete:updates-available',
  700. self.on_domain_updates_available)
  701. # It needs to store threads until they finish
  702. self.threads_list = []
  703. self.progress = None
  704. self.check_updates()
  705. def __init_context_menu(self):
  706. self.context_menu = QMenu(self)
  707. self.context_menu.addAction(self.action_settings)
  708. self.context_menu.addAction(self.action_editfwrules)
  709. self.context_menu.addAction(self.action_appmenus)
  710. self.context_menu.addAction(self.action_set_keyboard_layout)
  711. self.context_menu.addSeparator()
  712. self.context_menu.addAction(self.action_updatevm)
  713. self.context_menu.addAction(self.action_run_command_in_vm)
  714. self.context_menu.addAction(self.action_open_console)
  715. self.context_menu.addAction(self.action_resumevm)
  716. self.context_menu.addAction(self.action_startvm_tools_install)
  717. self.context_menu.addAction(self.action_pausevm)
  718. self.context_menu.addAction(self.action_shutdownvm)
  719. self.context_menu.addAction(self.action_restartvm)
  720. self.context_menu.addAction(self.action_killvm)
  721. self.context_menu.addSeparator()
  722. self.context_menu.addAction(self.action_clonevm)
  723. self.context_menu.addAction(self.action_removevm)
  724. self.context_menu.addSeparator()
  725. self.context_menu.addAction(self.action_show_logs)
  726. def save_showing(self):
  727. self.manager_settings.setValue('show/running',
  728. self.show_running.isChecked())
  729. self.manager_settings.setValue('show/halted',
  730. self.show_halted.isChecked())
  731. self.manager_settings.setValue('show/network',
  732. self.show_network.isChecked())
  733. self.manager_settings.setValue('show/templates',
  734. self.show_templates.isChecked())
  735. self.manager_settings.setValue('show/standalone',
  736. self.show_standalone.isChecked())
  737. self.manager_settings.setValue('show/all', self.show_all.isChecked())
  738. def save_sorting(self):
  739. self.manager_settings.setValue('view/sort_column',
  740. self.proxy.sortColumn())
  741. self.manager_settings.setValue('view/sort_order',
  742. self.proxy.sortOrder())
  743. def invalidate(self):
  744. self.proxy.invalidate()
  745. self.table.resizeColumnsToContents()
  746. def fill_cache(self):
  747. progress = QProgressDialog(
  748. self.tr(
  749. "Loading Qube Manager..."), "", 0,
  750. len(self.qubes_app.domains.keys()))
  751. progress.setWindowTitle(self.tr("Qube Manager"))
  752. progress.setMinimumDuration(1000)
  753. progress.setWindowModality(Qt.WindowModal)
  754. progress.setCancelButton(None)
  755. row_no = 0
  756. for vm in self.qubes_app.domains:
  757. progress.setValue(row_no)
  758. self.qubes_cache.add_vm(vm)
  759. row_no += 1
  760. progress.setValue(row_no)
  761. def setup_application(self):
  762. self.qt_app.setApplicationName(self.tr("Qube Manager"))
  763. self.qt_app.setWindowIcon(QIcon.fromTheme("qubes-manager"))
  764. def keyPressEvent(self, event): # pylint: disable=invalid-name
  765. if event.key() == Qt.Key_Escape:
  766. self.searchbox.clear()
  767. super().keyPressEvent(event)
  768. def clear_threads(self):
  769. for thread in self.threads_list:
  770. if thread.isFinished():
  771. if self.progress:
  772. self.progress.hide()
  773. self.progress = None
  774. if thread.msg:
  775. (title, msg) = thread.msg
  776. if thread.msg_is_success:
  777. QMessageBox.information(
  778. self,
  779. title,
  780. msg)
  781. else:
  782. QMessageBox.warning(
  783. self,
  784. title,
  785. msg)
  786. self.threads_list.remove(thread)
  787. return
  788. raise RuntimeError(self.tr('No finished thread found'))
  789. # pylint: disable=invalid-name
  790. def resizeEvent(self, event):
  791. self.manager_settings.setValue("window_size", event.size())
  792. def check_updates(self, info=None):
  793. if info is None:
  794. for info_iter in self.qubes_cache:
  795. self.check_updates(info_iter)
  796. return
  797. try:
  798. if info.vm.klass in {'TemplateVM', 'StandaloneVM'} and \
  799. manager_utils.get_feature(
  800. info.vm, 'updates-available', False):
  801. info.state['outdated'] = 'update'
  802. except exc.QubesDaemonAccessError:
  803. return
  804. def on_domain_added(self, _submitter, _event, vm, **_kwargs):
  805. try:
  806. domain = self.qubes_app.domains[vm]
  807. self.qubes_cache.add_vm(domain)
  808. self.proxy.invalidate()
  809. except (exc.QubesException, KeyError):
  810. pass
  811. def on_domain_removed(self, _submitter, _event, **kwargs):
  812. self.qubes_cache.remove_vm(name=kwargs['vm'])
  813. self.proxy.invalidate()
  814. def on_domain_status_changed(self, vm, event, **_kwargs):
  815. try:
  816. self.qubes_cache.get_vm(qid=vm.qid).update(event=event)
  817. if vm.klass in {'TemplateVM'}:
  818. for appvm in vm.appvms:
  819. self.qubes_cache.get_vm(qid=appvm.qid).\
  820. update(event="outdated")
  821. self.proxy.invalidate()
  822. self.table_selection_changed()
  823. except exc.QubesDaemonAccessError:
  824. return # the VM was deleted before its status could be updated
  825. except KeyError: # adding the VM failed for some reason
  826. self.on_domain_added(None, None, vm)
  827. def on_domain_updates_available(self, vm, _event, **_kwargs):
  828. self.check_updates(self.qubes_cache.get_vm(qid=vm.qid))
  829. def on_domain_changed(self, vm, event, **_kwargs):
  830. if not vm: # change of global properties occured
  831. if event.endswith(':default_netvm'):
  832. for vm_info in self.qubes_cache:
  833. vm_info.update(event='property-set:netvm')
  834. if event.endswith(':default_dispvm'):
  835. for vm_info in self.qubes_cache:
  836. vm_info.update(event='property-set:default_dispvm')
  837. return
  838. try:
  839. self.qubes_cache.get_vm(qid=vm.qid).update(event=event)
  840. self.proxy.invalidate()
  841. except exc.QubesDaemonAccessError:
  842. return # the VM was deleted before its status could be updated
  843. def load_manager_settings(self):
  844. # Load view menu settings
  845. for action in self.menu_view.actions():
  846. column = action.data()
  847. if column is not None:
  848. col_no = self.qubes_model.columns_indices.index(column)
  849. if column == 'Name':
  850. # 'Name' column should be always visible
  851. action.setChecked(True)
  852. else:
  853. visible = self.manager_settings.value('columns/%s' % column,
  854. defaultValue="true")
  855. action.setChecked(visible == "true")
  856. self.showhide_column(col_no, visible == "true")
  857. # Restore sorting
  858. sort_column = int(self.manager_settings.value("view/sort_column",
  859. defaultValue=2))
  860. order = Qt.SortOrder(self.manager_settings.value("view/sort_order",
  861. defaultValue=Qt.AscendingOrder))
  862. if not sort_column: # Default sort by name
  863. self.table.sortByColumn(2, Qt.AscendingOrder)
  864. else:
  865. self.table.sortByColumn(sort_column, order)
  866. if not self.manager_settings.value("view/menubar_visible",
  867. defaultValue=True):
  868. self.action_menubar.setChecked(False)
  869. if not self.manager_settings.value("view/toolbar_visible",
  870. defaultValue=True):
  871. self.action_toolbar.setChecked(False)
  872. if self.manager_settings.value("view/compactview",
  873. defaultValue="false") != "false":
  874. self.action_compact_view.setChecked(True)
  875. # Restore show checkboxes
  876. self.show_running.setChecked(self.manager_settings.value(
  877. 'show/running', "true") == "true")
  878. self.show_halted.setChecked(self.manager_settings.value(
  879. 'show/halted', "true") == "true")
  880. self.show_network.setChecked(self.manager_settings.value(
  881. 'show/network', "true") == "true")
  882. self.show_templates.setChecked(self.manager_settings.value(
  883. 'show/templates', "true") == "true")
  884. self.show_standalone.setChecked(self.manager_settings.value(
  885. 'show/standalone', "true") == "true")
  886. self.show_all.setChecked(self.manager_settings.value(
  887. 'show/all', "true") == "true")
  888. # load last window size
  889. self.resize(self.manager_settings.value("window_size",
  890. QSize(1100, 600)))
  891. @pyqtSlot(str)
  892. def do_search(self, search):
  893. self.proxy.setFilterFixedString(search)
  894. # noinspection PyArgumentList
  895. @pyqtSlot(name='on_action_search_triggered')
  896. def action_search_triggered(self):
  897. self.searchbox.setFocus()
  898. def get_selected_vms(self):
  899. vms = []
  900. selection = self.table.selectionModel().selection()
  901. indexes = self.proxy.mapSelectionToSource(selection).indexes()
  902. for index in indexes:
  903. if index.column() != 0:
  904. continue
  905. vms.append(index.data(Qt.UserRole))
  906. return vms
  907. def table_selection_changed(self):
  908. # Since selection could have multiple domains
  909. # enable all first and then filter them
  910. for action in self.toolbar.actions() + self.context_menu.actions():
  911. action.setEnabled(True)
  912. for vm in self.get_selected_vms():
  913. # TODO: add boot from device to menu and add windows tools there
  914. # Update available actions:
  915. if vm.state['power'] in \
  916. ['Running', 'Transient', 'Halting', 'Dying']:
  917. self.action_resumevm.setEnabled(False)
  918. self.action_removevm.setEnabled(False)
  919. elif vm.state['power'] == 'Paused':
  920. self.action_removevm.setEnabled(False)
  921. self.action_pausevm.setEnabled(False)
  922. self.action_set_keyboard_layout.setEnabled(False)
  923. self.action_restartvm.setEnabled(False)
  924. self.action_open_console.setEnabled(False)
  925. elif vm.state['power'] == 'Suspend':
  926. self.action_set_keyboard_layout.setEnabled(False)
  927. self.action_removevm.setEnabled(False)
  928. self.action_pausevm.setEnabled(False)
  929. self.action_open_console.setEnabled(False)
  930. elif vm.state['power'] == 'Halted':
  931. self.action_set_keyboard_layout.setEnabled(False)
  932. self.action_pausevm.setEnabled(False)
  933. self.action_shutdownvm.setEnabled(False)
  934. self.action_restartvm.setEnabled(False)
  935. self.action_killvm.setEnabled(False)
  936. self.action_open_console.setEnabled(False)
  937. if vm.klass == 'AdminVM':
  938. self.action_open_console.setEnabled(False)
  939. self.action_settings.setEnabled(False)
  940. self.action_resumevm.setEnabled(False)
  941. self.action_removevm.setEnabled(False)
  942. self.action_clonevm.setEnabled(False)
  943. self.action_pausevm.setEnabled(False)
  944. self.action_restartvm.setEnabled(False)
  945. self.action_killvm.setEnabled(False)
  946. self.action_shutdownvm.setEnabled(False)
  947. self.action_appmenus.setEnabled(False)
  948. self.action_editfwrules.setEnabled(False)
  949. self.action_set_keyboard_layout.setEnabled(False)
  950. self.action_run_command_in_vm.setEnabled(False)
  951. elif vm.klass == 'DispVM':
  952. self.action_appmenus.setEnabled(False)
  953. self.action_restartvm.setEnabled(False)
  954. if vm.vm.features.get('internal', False):
  955. self.action_appmenus.setEnabled(False)
  956. if not vm.updateable and vm.klass != 'AdminVM':
  957. self.action_updatevm.setEnabled(False)
  958. # noinspection PyArgumentList
  959. @pyqtSlot(name='on_action_createvm_triggered')
  960. def action_createvm_triggered(self):
  961. with common_threads.busy_cursor():
  962. create_window = create_new_vm.NewVmDlg(self.qt_app, self.qubes_app)
  963. create_window.exec_()
  964. # noinspection PyArgumentList
  965. @pyqtSlot(name='on_action_removevm_triggered')
  966. def action_removevm_triggered(self):
  967. remove_vms = []
  968. for vm_info in self.get_selected_vms():
  969. vm = vm_info.vm
  970. dependencies = utils.vm_dependencies(self.qubes_app, vm)
  971. if dependencies:
  972. list_deps = manager_utils.format_dependencies_list(dependencies)
  973. list_text = "<br>" + list_deps + "<br>"
  974. info_dialog = QMessageBox(self)
  975. info_dialog.setWindowTitle(self.tr("Warning!"))
  976. info_dialog.setText(
  977. self.tr("This qube cannot be removed. It is used as: <br> "
  978. "{} <small>If you want to remove this qube, you "
  979. "should remove or change settings of each qube or "
  980. "setting that uses it.</small>").format(list_text))
  981. info_dialog.setModal(False)
  982. info_dialog.show()
  983. return
  984. (requested_name, ok) = QInputDialog.getText(
  985. self, self.tr("Qube Removal Confirmation"),
  986. self.tr("Are you sure you want to remove the Qube <b>'{0}'</b>"
  987. "?<br> All data on this Qube's private storage will be "
  988. "lost!<br><br>Type the name of the Qube (<b>{1}</b>) be"
  989. "low to confirm:").format(vm.name, vm.name))
  990. if not ok:
  991. # user clicked cancel
  992. continue
  993. if requested_name == vm.name:
  994. remove_vms.append(vm)
  995. else:
  996. # name did not match
  997. QMessageBox.warning(
  998. self,
  999. self.tr("Qube removal confirmation failed"),
  1000. self.tr(
  1001. "Entered name did not match! Not removing "
  1002. "{0}.").format(vm.name))
  1003. # remove the VMs
  1004. for vm in remove_vms:
  1005. thread = common_threads.RemoveVMThread(vm)
  1006. self.threads_list.append(thread)
  1007. thread.finished.connect(self.clear_threads)
  1008. thread.start()
  1009. # noinspection PyArgumentList
  1010. @pyqtSlot(name='on_action_clonevm_triggered')
  1011. def action_clonevm_triggered(self):
  1012. for vm_info in self.get_selected_vms():
  1013. vm = vm_info.vm
  1014. with common_threads.busy_cursor():
  1015. clone_window = clone_vm.CloneVMDlg(
  1016. self.qt_app, self.qubes_app, src_vm=vm)
  1017. clone_window.exec_()
  1018. # noinspection PyArgumentList
  1019. @pyqtSlot(name='on_action_resumevm_triggered')
  1020. def action_resumevm_triggered(self):
  1021. for vm_info in self.get_selected_vms():
  1022. vm = vm_info.vm
  1023. try:
  1024. if vm.get_power_state() in ["Paused", "Suspended"]:
  1025. vm.unpause()
  1026. except exc.QubesException as ex:
  1027. QMessageBox.warning(
  1028. self, self.tr("Error unpausing Qube!"),
  1029. self.tr("ERROR: {0}").format(ex))
  1030. return
  1031. self.start_vm(vm)
  1032. def start_vm(self, vm):
  1033. if manager_utils.is_running(vm, False):
  1034. return
  1035. thread = StartVMThread(vm)
  1036. self.threads_list.append(thread)
  1037. thread.finished.connect(self.clear_threads)
  1038. thread.start()
  1039. # noinspection PyArgumentList
  1040. @pyqtSlot(name='on_action_startvm_tools_install_triggered')
  1041. # TODO: replace with boot from device
  1042. def action_startvm_tools_install_triggered(self):
  1043. # pylint: disable=invalid-name
  1044. pass
  1045. @pyqtSlot(name='on_action_pausevm_triggered')
  1046. def action_pausevm_triggered(self):
  1047. for vm_info in self.get_selected_vms():
  1048. try:
  1049. vm_info.vm.pause()
  1050. except exc.QubesException as ex:
  1051. QMessageBox.warning(
  1052. self,
  1053. self.tr("Error pausing Qube!"),
  1054. self.tr("ERROR: {0}").format(ex))
  1055. return
  1056. # noinspection PyArgumentList
  1057. @pyqtSlot(name='on_action_shutdownvm_triggered')
  1058. def action_shutdownvm_triggered(self):
  1059. for vm_info in self.get_selected_vms():
  1060. vm = vm_info.vm
  1061. reply = QMessageBox.question(
  1062. self, self.tr("Qube Shutdown Confirmation"),
  1063. self.tr("Are you sure you want to power down the Qube <b>'{0}'"
  1064. "</b>?<br><small>This will shutdown all the running"
  1065. " applications within this Qube.</small>").format(
  1066. vm.name),
  1067. QMessageBox.Yes | QMessageBox.Cancel)
  1068. if reply == QMessageBox.Yes:
  1069. self.shutdown_vm(vm)
  1070. def get_connected_vms(self, vm, connected_vms):
  1071. for connected_vm in vm.connected_vms:
  1072. if connected_vm.is_running():
  1073. connected_vms.append(connected_vm)
  1074. self.get_connected_vms(connected_vm, connected_vms)
  1075. def shutdown_vm(self, vm, shutdown_time=vm_shutdown_timeout,
  1076. check_time=vm_restart_check_timeout, and_restart=False):
  1077. try:
  1078. connected_vms = []
  1079. self.get_connected_vms(vm, connected_vms)
  1080. if len(connected_vms) > 0:
  1081. reply = QMessageBox.question(
  1082. self, self.tr("Qube Shutdown Confirmation"),
  1083. self.tr("There are some qubes connected to <b>'{0}'</b>!"
  1084. "<br><small>Do you want to shutdown: </small>"
  1085. "<b>'{1}'</b>?").format(vm.name,
  1086. ", ".join([x.name for x in connected_vms])),
  1087. QMessageBox.Yes | QMessageBox.Cancel)
  1088. if reply != QMessageBox.Yes:
  1089. return False
  1090. for connected_vm in connected_vms:
  1091. connected_vm.shutdown(force=True)
  1092. vm.shutdown(force=True)
  1093. else:
  1094. vm.shutdown()
  1095. except exc.QubesException as ex:
  1096. QMessageBox.warning(
  1097. self,
  1098. self.tr("Error shutting down Qube!"),
  1099. self.tr("ERROR: {0}").format(ex))
  1100. return False
  1101. self.shutdown_monitor[vm.qid] = VmShutdownMonitor(vm, shutdown_time,
  1102. check_time,
  1103. and_restart, self)
  1104. # noinspection PyCallByClass,PyTypeChecker
  1105. QTimer.singleShot(check_time, self.shutdown_monitor[
  1106. vm.qid].check_if_vm_has_shutdown)
  1107. return True
  1108. # noinspection PyArgumentList
  1109. @pyqtSlot(name='on_action_restartvm_triggered')
  1110. def action_restartvm_triggered(self):
  1111. for vm_info in self.get_selected_vms():
  1112. vm = vm_info.vm
  1113. reply = QMessageBox.question(
  1114. self, self.tr("Qube Restart Confirmation"),
  1115. self.tr("Are you sure you want to restart the Qube <b>'{0}'</b>"
  1116. "?<br><small>This will shutdown all the running applica"
  1117. "tions within this Qube.</small>").format(vm.name),
  1118. QMessageBox.Yes | QMessageBox.Cancel)
  1119. if reply == QMessageBox.Yes:
  1120. # in case the user shut down the VM in the meantime
  1121. try:
  1122. if manager_utils.is_running(vm, False):
  1123. self.shutdown_vm(vm, and_restart=True)
  1124. else:
  1125. self.start_vm(vm)
  1126. except exc.QubesException as ex:
  1127. QMessageBox.warning(
  1128. self,
  1129. self.tr("Error restarting Qube!"),
  1130. self.tr("ERROR: {0}").format(ex))
  1131. # noinspection PyArgumentList
  1132. @pyqtSlot(name='on_action_killvm_triggered')
  1133. def action_killvm_triggered(self):
  1134. for vm_info in self.get_selected_vms():
  1135. vm = vm_info.vm
  1136. try:
  1137. vm_not_running = not (vm.is_running() or vm.is_paused())
  1138. except exc.QubesDaemonAccessError:
  1139. vm_not_running = False
  1140. if vm_not_running:
  1141. info = self.tr("Qube <b>'{0}'</b> is not running. Are you "
  1142. "absolutely sure you want to try to kill it?<br>"
  1143. "<small>This will end <b>(not shutdown!)</b> "
  1144. "all the running applications within this "
  1145. "Qube.</small>").format(vm.name)
  1146. else:
  1147. info = self.tr("Are you sure you want to kill the Qube "
  1148. "<b>'{0}'</b>?<br><small>This will end <b>(not "
  1149. "shutdown!)</b> all the running applications "
  1150. "within this Qube.</small>").format(vm.name)
  1151. reply = QMessageBox.question(
  1152. self, self.tr("Qube Kill Confirmation"), info,
  1153. QMessageBox.Yes | QMessageBox.Cancel,
  1154. QMessageBox.Cancel)
  1155. if reply == QMessageBox.Yes:
  1156. try:
  1157. vm.kill()
  1158. except exc.QubesException as ex:
  1159. QMessageBox.critical(
  1160. self, self.tr("Error while killing Qube!"),
  1161. self.tr(
  1162. "<b>An exception occurred while killing {0}.</b>"
  1163. "<br>ERROR: {1}").format(vm.name, ex))
  1164. return
  1165. def open_settings(self, vm, tab='basic'):
  1166. try:
  1167. with common_threads.busy_cursor():
  1168. settings_window = settings.VMSettingsWindow(
  1169. vm, init_page=tab, qapp=self.qt_app,
  1170. qubesapp=self.qubes_app)
  1171. settings_window.show()
  1172. self.settings_windows[vm.name] = settings_window
  1173. except exc.QubesException as ex:
  1174. QMessageBox.warning(
  1175. self,
  1176. self.tr("Qube settings unavailable"),
  1177. self.tr(
  1178. "Qube settings cannot be opened. The qube might have "
  1179. "been removed or unavailable due to policy settings."
  1180. "\nError: {}".format(str(ex))))
  1181. return
  1182. def closeEvent(self, _):
  1183. self.save_showing()
  1184. # noinspection PyArgumentList
  1185. @pyqtSlot(name='on_action_settings_triggered')
  1186. def action_settings_triggered(self):
  1187. for vm_info in self.get_selected_vms():
  1188. self.open_settings(vm_info.vm, "basic")
  1189. # noinspection PyArgumentList
  1190. @pyqtSlot(name='on_action_appmenus_triggered')
  1191. def action_appmenus_triggered(self):
  1192. for vm_info in self.get_selected_vms():
  1193. self.open_settings(vm_info.vm, "applications")
  1194. # noinspection PyArgumentList
  1195. @pyqtSlot(name='on_action_updatevm_triggered')
  1196. def action_updatevm_triggered(self):
  1197. for vm_info in self.get_selected_vms():
  1198. vm = vm_info.vm
  1199. if not manager_utils.is_running(vm, True):
  1200. reply = QMessageBox.question(
  1201. self, self.tr("Qube Update Confirmation"),
  1202. self.tr(
  1203. "<b>{0}</b>"
  1204. "<br>The Qube has to be running to be updated."
  1205. "<br>Do you want to start it?<br>").format(vm.name),
  1206. QMessageBox.Yes | QMessageBox.Cancel)
  1207. if reply != QMessageBox.Yes:
  1208. return
  1209. thread = UpdateVMThread(vm)
  1210. self.threads_list.append(thread)
  1211. thread.finished.connect(self.clear_threads)
  1212. thread.start()
  1213. # noinspection PyArgumentList
  1214. @pyqtSlot(name='on_action_run_command_in_vm_triggered')
  1215. def action_run_command_in_vm_triggered(self):
  1216. # pylint: disable=invalid-name
  1217. for vm_info in self.get_selected_vms():
  1218. (command_to_run, ok) = QInputDialog.getText(
  1219. self, self.tr('Qubes command entry'),
  1220. self.tr('Run command in <b>{}</b>:').format(vm_info.name))
  1221. if not ok or command_to_run == "":
  1222. return
  1223. thread = RunCommandThread(vm_info.vm, command_to_run)
  1224. self.threads_list.append(thread)
  1225. thread.finished.connect(self.clear_threads)
  1226. thread.start()
  1227. # noinspection PyArgumentList
  1228. @pyqtSlot(name='on_action_open_console_triggered')
  1229. def action_open_console_triggered(self):
  1230. # pylint: disable=invalid-name
  1231. for vm in self.get_selected_vms():
  1232. subprocess.Popen(['qvm-console-dispvm', vm.name],
  1233. stdout=subprocess.DEVNULL,
  1234. stderr=subprocess.DEVNULL)
  1235. # noinspection PyArgumentList
  1236. @pyqtSlot(name='on_action_set_keyboard_layout_triggered')
  1237. def action_set_keyboard_layout_triggered(self):
  1238. # pylint: disable=invalid-name
  1239. for vm_info in self.get_selected_vms():
  1240. if vm_info.vm.features.check_with_template(
  1241. "supported-feature.keyboard-layout", False):
  1242. vm_info.vm.run('qubes-change-keyboard-layout')
  1243. else:
  1244. QMessageBox.warning(
  1245. self,
  1246. self.tr("Keyboard layout change unsupported"),
  1247. self.tr(
  1248. "Please update the qube {} or its template to the "
  1249. "newest version of Qubes tools.").format(
  1250. str(vm_info.vm)))
  1251. # noinspection PyArgumentList
  1252. @pyqtSlot(name='on_action_editfwrules_triggered')
  1253. def action_editfwrules_triggered(self):
  1254. for vm_info in self.get_selected_vms():
  1255. self.open_settings(vm_info.vm, "firewall")
  1256. # noinspection PyArgumentList
  1257. @pyqtSlot(name='on_action_global_settings_triggered')
  1258. def action_global_settings_triggered(self): # pylint: disable=invalid-name
  1259. with common_threads.busy_cursor():
  1260. global_settings_window = global_settings.GlobalSettingsWindow(
  1261. self.qt_app,
  1262. self.qubes_app)
  1263. global_settings_window.show()
  1264. self.settings_windows['global_settings_window'] = global_settings_window
  1265. # noinspection PyArgumentList
  1266. @pyqtSlot(name='on_action_manage_templates_triggered')
  1267. def action_manage_templates_triggered(self):
  1268. # pylint: disable=no-self-use
  1269. subprocess.check_call('qubes-template-manager')
  1270. # noinspection PyArgumentList
  1271. @pyqtSlot(name='on_action_show_network_triggered')
  1272. def action_show_network_triggered(self):
  1273. pass
  1274. # TODO: revive for 4.1
  1275. # network_notes_dialog = NetworkNotesDialog()
  1276. # network_notes_dialog.exec_()
  1277. # noinspection PyArgumentList
  1278. @pyqtSlot(name='on_action_restore_triggered')
  1279. def action_restore_triggered(self):
  1280. with common_threads.busy_cursor():
  1281. restore_window = restore.RestoreVMsWindow(self.qt_app,
  1282. self.qubes_app)
  1283. restore_window.exec_()
  1284. # noinspection PyArgumentList
  1285. @pyqtSlot(name='on_action_backup_triggered')
  1286. def action_backup_triggered(self):
  1287. with common_threads.busy_cursor():
  1288. backup_window = backup.BackupVMsWindow(
  1289. self.qt_app, self.qubes_app, self.dispatcher, self)
  1290. backup_window.show()
  1291. # noinspection PyArgumentList
  1292. @pyqtSlot(name='on_action_exit_triggered')
  1293. def action_exit_triggered(self):
  1294. self.close()
  1295. def set_compactview(self, checked):
  1296. if checked:
  1297. self.toolbar.setToolButtonStyle(Qt.ToolButtonIconOnly)
  1298. else:
  1299. self.toolbar.setToolButtonStyle(Qt.ToolButtonTextUnderIcon)
  1300. if self.settings_loaded:
  1301. self.manager_settings.setValue('view/compactview', checked)
  1302. def showhide_menubar(self, checked):
  1303. self.menubar.setVisible(checked)
  1304. if not checked:
  1305. self.context_menu.addAction(self.action_menubar)
  1306. else:
  1307. self.context_menu.removeAction(self.action_menubar)
  1308. if self.settings_loaded:
  1309. self.manager_settings.setValue('view/menubar_visible', checked)
  1310. def showhide_toolbar(self, checked):
  1311. self.toolbar.setVisible(checked)
  1312. if not checked:
  1313. self.context_menu.addAction(self.action_toolbar)
  1314. else:
  1315. self.context_menu.removeAction(self.action_toolbar)
  1316. if self.settings_loaded:
  1317. self.manager_settings.setValue('view/toolbar_visible', checked)
  1318. def showhide_column(self, col_num, show):
  1319. self.table.setColumnHidden(col_num, not show)
  1320. col_name = self.qubes_model.columns_indices[col_num]
  1321. self.manager_settings.setValue('columns/%s' % col_name, show)
  1322. # noinspection PyArgumentList
  1323. @pyqtSlot(name='on_action_about_qubes_triggered')
  1324. def action_about_qubes_triggered(self): # pylint: disable=no-self-use
  1325. about = AboutDialog()
  1326. about.exec_()
  1327. def createPopupMenu(self): # pylint: disable=invalid-name
  1328. menu = QMenu()
  1329. menu.addAction(self.action_toolbar)
  1330. menu.addAction(self.action_menubar)
  1331. return menu
  1332. def open_tools_context_menu(self, widget, point):
  1333. self.tools_context_menu.exec_(widget.mapToGlobal(point))
  1334. @pyqtSlot('const QPoint&')
  1335. def open_context_menu(self, point):
  1336. self.context_menu.exec_(self.table.mapToGlobal(
  1337. point + QPoint(10, 0)))
  1338. def show_log(self):
  1339. logfiles = []
  1340. try:
  1341. for vm_info in self.get_selected_vms():
  1342. vm = vm_info.vm
  1343. if vm.klass == 'AdminVM':
  1344. logfiles.append("/var/log/xen/console/hypervisor.log")
  1345. else:
  1346. logfiles.extend([
  1347. "/var/log/xen/console/guest-" + vm.name + ".log",
  1348. "/var/log/xen/console/guest-" + vm.name + "-dm.log",
  1349. "/var/log/qubes/guid." + vm.name + ".log",
  1350. "/var/log/qubes/qrexec." + vm.name + ".log",
  1351. ])
  1352. logfiles = [x for x in logfiles if path.exists(x)]
  1353. if len(logfiles) > 0:
  1354. log_dlg = log_dialog.LogDialog(self.qt_app, logfiles)
  1355. log_dlg.exec_()
  1356. else:
  1357. QMessageBox.warning(
  1358. self,
  1359. self.tr("Error"),
  1360. self.tr(
  1361. "No log files where found for the current selection."))
  1362. except exc.QubesDaemonAccessError:
  1363. pass
  1364. def main():
  1365. manager_utils.run_asynchronous(VmManagerWindow)
  1366. if __name__ == "__main__":
  1367. main()