main.py 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759
  1. #!/usr/bin/python2.6
  2. #
  3. # The Qubes OS Project, http://www.qubes-os.org
  4. #
  5. # Copyright (C) 2010 Joanna Rutkowska <joanna@invisiblethingslab.com>
  6. #
  7. # This program is free software; you can redistribute it and/or
  8. # modify it under the terms of the GNU General Public License
  9. # as published by the Free Software Foundation; either version 2
  10. # of the License, or (at your option) any later version.
  11. #
  12. # This program is distributed in the hope that it will be useful,
  13. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15. # GNU General Public License for more details.
  16. #
  17. # You should have received a copy of the GNU General Public License
  18. # along with this program; if not, write to the Free Software
  19. # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  20. #
  21. #
  22. import sys
  23. from PyQt4.QtCore import *
  24. from PyQt4.QtGui import *
  25. from qubes.qubes import QubesVmCollection
  26. from qubes.qubes import QubesException
  27. from qubes.qubes import qubes_store_filename
  28. from qubes.qubes import QubesVmLabels
  29. from qubes.qubes import dry_run
  30. from qubes.qubes import qubes_guid_path
  31. from qubes.qubes import QubesDaemonPidfile
  32. import qubesmanager.qrc_resources
  33. import ui_newappvmdlg
  34. from pyinotify import WatchManager, Notifier, ThreadedNotifier, EventsCodes, ProcessEvent
  35. import subprocess
  36. import time
  37. import threading
  38. class QubesConfigFileWatcher(ProcessEvent):
  39. def __init__ (self, update_func):
  40. self.update_func = update_func
  41. pass
  42. def process_IN_CLOSE_WRITE (self, event):
  43. self.update_func()
  44. class VmStatusIcon(QLabel):
  45. def __init__(self, vm, parent=None):
  46. super (VmStatusIcon, self).__init__(parent)
  47. self.vm = vm
  48. (icon_pixmap, icon_sz) = self.set_vm_icon(self.vm)
  49. self.setPixmap (icon_pixmap)
  50. self.setFixedSize (icon_sz)
  51. self.previous_power_state = vm.is_running()
  52. def update(self):
  53. if self.previous_power_state != self.vm.is_running():
  54. (icon_pixmap, icon_sz) = self.set_vm_icon(self.vm)
  55. self.setPixmap (icon_pixmap)
  56. self.setFixedSize (icon_sz)
  57. self.previous_power_state = self.vm.is_running()
  58. def set_vm_icon(self, vm):
  59. if vm.qid == 0:
  60. icon = QIcon (":/dom0.png")
  61. elif vm.is_appvm():
  62. icon = QIcon (vm.label.icon_path)
  63. elif vm.is_templete():
  64. icon = QIcon (":/templatevm.png")
  65. elif vm.is_netvm():
  66. icon = QIcon (":/netvm.png")
  67. else:
  68. icon = QIcon()
  69. icon_sz = QSize (VmManagerWindow.row_height * 0.8, VmManagerWindow.row_height * 0.8)
  70. if vm.is_running():
  71. icon_pixmap = icon.pixmap(icon_sz)
  72. else:
  73. icon_pixmap = icon.pixmap(icon_sz, QIcon.Disabled)
  74. return (icon_pixmap, icon_sz)
  75. class VmInfoWidget (QWidget):
  76. def __init__(self, vm, parent = None):
  77. super (VmInfoWidget, self).__init__(parent)
  78. layout0 = QHBoxLayout()
  79. label_name = QLabel (vm.name)
  80. self.vm_running = vm.is_running()
  81. layout0.addWidget(label_name, alignment=Qt.AlignLeft)
  82. layout1 = QHBoxLayout()
  83. if vm.is_appvm():
  84. label_tmpl = QLabel ("<i><font color=\"gray\">" + vm.template_vm.name + "</i></font>")
  85. elif vm.is_templete():
  86. label_tmpl = QLabel ("<i><font color=\"gray\">TemplateVM</i></font>")
  87. elif vm.qid == 0:
  88. label_tmpl = QLabel ("<i><font color=\"gray\">AdminVM</i></font>")
  89. elif vm.is_netvm():
  90. label_tmpl = QLabel ("<i><font color=\"gray\">NetVM</i></font>")
  91. else:
  92. label_tmpl = QLabel ("")
  93. label_icon_networked = self.set_icon(":/networking.png", vm.is_networked())
  94. layout1.addWidget(label_icon_networked, alignment=Qt.AlignLeft)
  95. if vm.is_updateable():
  96. label_icon_updtbl = self.set_icon(":/updateable.png", True)
  97. layout1.addWidget(label_icon_updtbl, alignment=Qt.AlignLeft)
  98. layout1.addWidget(label_tmpl, alignment=Qt.AlignLeft)
  99. layout1.addStretch()
  100. layout2 = QVBoxLayout ()
  101. layout2.addLayout(layout0)
  102. layout2.addLayout(layout1)
  103. layout3 = QHBoxLayout ()
  104. self.vm_icon = VmStatusIcon(vm)
  105. layout3.addWidget(self.vm_icon)
  106. layout3.addSpacing (10)
  107. layout3.addLayout(layout2)
  108. self.setLayout(layout3)
  109. def set_icon(self, icon_path, enabled = True):
  110. label_icon = QLabel()
  111. icon = QIcon (icon_path)
  112. icon_sz = QSize (VmManagerWindow.row_height * 0.3, VmManagerWindow.row_height * 0.3)
  113. icon_pixmap = icon.pixmap(icon_sz, QIcon.Disabled if not enabled else QIcon.Normal)
  114. label_icon.setPixmap (icon_pixmap)
  115. label_icon.setFixedSize (icon_sz)
  116. return label_icon
  117. def update_vm_state (self, vm):
  118. self.vm_icon.update()
  119. class LoadChartWidget (QWidget):
  120. def __init__(self, vm, parent = None):
  121. super (LoadChartWidget, self).__init__(parent)
  122. self.load = vm.get_cpu_total_load() if vm.is_running() else 0
  123. assert self.load >= 0 and self.load <= 100, "load = {0}".format(self.load)
  124. self.load_history = [self.load]
  125. def update_load (self, vm):
  126. self.load = vm.get_cpu_total_load() if vm.is_running() else 0
  127. assert self.load >= 0 and self.load <= 100, "load = {0}".format(self.load)
  128. self.load_history.append (self.load)
  129. self.repaint()
  130. def paintEvent (self, Event = None):
  131. p = QPainter (self)
  132. dx = 4
  133. W = self.width()
  134. H = self.height() - 5
  135. N = len(self.load_history)
  136. if N > W/dx:
  137. N = W/dx
  138. self.load_history.pop(0)
  139. for i in range (0, N-1):
  140. val = self.load_history[N- i - 1]
  141. hue = 200
  142. sat = 70 + val*(255-70)/100
  143. color = QColor.fromHsv (hue, sat, 255)
  144. pen = QPen (color)
  145. pen.setWidth(dx-1)
  146. p.setPen(pen)
  147. if val > 0:
  148. p.drawLine (W - i*dx - dx, H , W - i*dx - dx, H - (H - 5) * val/100)
  149. class VmRowInTable(object):
  150. def __init__(self, vm, row_no, table):
  151. self.vm = vm
  152. self.row_no = row_no
  153. table.setRowHeight (row_no, VmManagerWindow.row_height)
  154. self.info_widget = VmInfoWidget(vm)
  155. table.setCellWidget(row_no, 0, self.info_widget)
  156. self.load_widget = LoadChartWidget(vm)
  157. table.setCellWidget(row_no, 1, self.load_widget)
  158. def update(self, counter):
  159. self.info_widget.update_vm_state(self.vm)
  160. if counter % 3 == 0:
  161. self.load_widget.update_load(self.vm)
  162. class NewAppVmDlg (QDialog, ui_newappvmdlg.Ui_NewAppVMDlg):
  163. def __init__(self, parent = None):
  164. super (NewAppVmDlg, self).__init__(parent)
  165. self.setupUi(self)
  166. vm_shutdown_timeout = 15000 # in msec
  167. class VmShutdownMonitor(QObject):
  168. def __init__(self, vm):
  169. self.vm = vm
  170. def check_if_vm_has_shutdown(self):
  171. vm = self.vm
  172. if not vm.is_running():
  173. return
  174. reply = QMessageBox.question(None, "VM Shutdown",
  175. "The VM <b>'{0}'</b> hasn't shutdown within the last {1} seconds, do you want to kill it?<br>".format(vm_name, vm_shutdown_timeout/1000),
  176. "Kill it!", "Wait another {0} seconds...".format(vm_shutdown_timeout/1000))
  177. if reply == 0:
  178. vm.force_shutdown()
  179. else:
  180. QTimer.singleShot (vm_shutdown_timeout, self.check_if_vm_has_shutdown)
  181. class ThreadMonitor(QObject):
  182. def __init__(self):
  183. self.success = True
  184. self.error_msg = None
  185. self.event_finished = threading.Event()
  186. def set_error_msg(self, error_msg):
  187. self.success = False
  188. self.error_msg = error_msg
  189. self.set_finished()
  190. def is_finished(self):
  191. return self.event_finished.is_set()
  192. def set_finished(self):
  193. self.event_finished.set()
  194. class VmManagerWindow(QMainWindow):
  195. columns_widths = [200, 150]
  196. row_height = 50
  197. max_visible_rows = 14
  198. update_interval = 1000 # in msec
  199. show_inactive_vms = True
  200. def __init__(self, parent=None):
  201. super(VmManagerWindow, self).__init__(parent)
  202. self.action_createvm = self.createAction ("Create AppVM", slot=self.create_appvm,
  203. icon="createvm", tip="Create a new AppVM")
  204. self.action_removevm = self.createAction ("Remove AppVM", slot=self.remove_appvm,
  205. icon="removevm", tip="Remove an existing AppVM (must be stopped first)")
  206. self.action_resumevm = self.createAction ("Start/Resume VM", slot=self.resume_vm,
  207. icon="resumevm", tip="Start/Resusme a VM")
  208. self.action_pausevm = self.createAction ("Pause VM", slot=self.pause_vm,
  209. icon="pausevm", tip="Pause a running VM")
  210. self.action_shutdownvm = self.createAction ("Shutdown VM", slot=self.shutdown_vm,
  211. icon="shutdownvm", tip="Shutdown a running VM")
  212. self.action_updatevm = self.createAction ("Update VM", slot=None,
  213. icon="updateable", tip="Update VM (only for 'updateable' VMs, e.g. templates)")
  214. self.action_showallvms = self.createAction ("Show/Hide Inactive VMs", slot=None, checkable=True,
  215. icon="showallvms", tip="Show/Hide Inactive VMs")
  216. self.action_showcpuload = self.createAction ("Show/Hide CPU Load chart", slot=self.showcpuload, checkable=True,
  217. icon="showcpuload", tip="Show/Hide CPU Load chart")
  218. self.action_removevm.setDisabled(True)
  219. self.action_resumevm.setDisabled(True)
  220. self.action_pausevm.setDisabled(True)
  221. self.action_shutdownvm.setDisabled(True)
  222. self.action_updatevm.setDisabled(True)
  223. self.action_showcpuload.setDisabled(True)
  224. self.toolbar = self.addToolBar ("Toolbar")
  225. self.toolbar.setFloatable(False)
  226. self.addActions (self.toolbar, (self.action_createvm, self.action_removevm,
  227. None,
  228. self.action_resumevm, self.action_pausevm, self.action_shutdownvm, self.action_updatevm,
  229. None,
  230. self.action_showcpuload,
  231. ))
  232. self.table = QTableWidget()
  233. self.setCentralWidget(self.table)
  234. self.table.clear()
  235. self.table.setColumnCount(len(VmManagerWindow.columns_widths))
  236. for (col, width) in enumerate (VmManagerWindow.columns_widths):
  237. self.table.setColumnWidth (col, width)
  238. self.table.horizontalHeader().setStretchLastSection(True)
  239. self.table.setAlternatingRowColors(True)
  240. self.table.verticalHeader().hide()
  241. self.table.horizontalHeader().hide()
  242. self.table.setGridStyle(Qt.NoPen)
  243. self.table.setSortingEnabled(False)
  244. self.table.setSelectionBehavior(QTableWidget.SelectRows)
  245. self.table.setSelectionMode(QTableWidget.SingleSelection)
  246. self.qvm_collection = QubesVmCollection()
  247. self.setWindowTitle("Qubes VM Manager")
  248. self.connect(self.table, SIGNAL("itemSelectionChanged()"), self.table_selection_changed)
  249. self.fill_table()
  250. tbl_W = 0
  251. for (i, w) in enumerate(VmManagerWindow.columns_widths):
  252. tbl_W += w
  253. # TODO: '6' -- WTF?!
  254. tbl_H = self.toolbar.height() + 6 + \
  255. self.table.horizontalHeader().height() + 6
  256. n = self.table.rowCount();
  257. if n > VmManagerWindow.max_visible_rows:
  258. n = VmManagerWindow.max_visible_rows
  259. for i in range (0, n):
  260. tbl_H += self.table.rowHeight(i)
  261. self.setMinimumWidth(tbl_W)
  262. self.setGeometry(self.x(), self.y(), self.x() + tbl_W, self.y() + tbl_H)
  263. self.counter = 0
  264. self.shutdown_monitor = {}
  265. QTimer.singleShot (self.update_interval, self.update_table)
  266. def addActions(self, target, actions):
  267. for action in actions:
  268. if action is None:
  269. target.addSeparator()
  270. else:
  271. target.addAction(action)
  272. def createAction(self, text, slot=None, shortcut=None, icon=None,
  273. tip=None, checkable=False, signal="triggered()"):
  274. action = QAction(text, self)
  275. if icon is not None:
  276. action.setIcon(QIcon(":/%s.png" % icon))
  277. if shortcut is not None:
  278. action.setShortcut(shortcut)
  279. if tip is not None:
  280. action.setToolTip(tip)
  281. action.setStatusTip(tip)
  282. if slot is not None:
  283. self.connect(action, SIGNAL(signal), slot)
  284. if checkable:
  285. action.setCheckable(True)
  286. return action
  287. def get_vms_list(self):
  288. self.qvm_collection.lock_db_for_reading()
  289. self.qvm_collection.load()
  290. self.qvm_collection.unlock_db()
  291. if self.show_inactive_vms:
  292. vms_list = [vm for vm in self.qvm_collection.values()]
  293. else:
  294. vms_list = [vm for vm in self.qvm_collection.values() if vm.is_running()]
  295. no_vms = len (vms_list)
  296. vms_to_display = []
  297. # First, the NetVMs...
  298. for netvm in vms_list:
  299. if netvm.is_netvm():
  300. vms_to_display.append (netvm)
  301. # Now, the templates...
  302. for tvm in vms_list:
  303. if tvm.is_templete():
  304. vms_to_display.append (tvm)
  305. label_list = QubesVmLabels.values()
  306. label_list.sort(key=lambda l: l.index)
  307. for label in [label.name for label in label_list]:
  308. for appvm in [vm for vm in vms_list if (vm.is_appvm() and vm.label.name == label)]:
  309. vms_to_display.append(appvm)
  310. assert len(vms_to_display) == no_vms
  311. return vms_to_display
  312. def fill_table(self):
  313. self.table.clear()
  314. vms_list = self.get_vms_list()
  315. self.table.setRowCount(len(vms_list))
  316. vms_in_table = []
  317. for (row_no, vm) in enumerate(vms_list):
  318. vm_row = VmRowInTable (vm, row_no, self.table)
  319. vms_in_table.append (vm_row)
  320. self.vms_list = vms_list
  321. self.vms_in_table = vms_in_table
  322. self.reload_table = False
  323. def mark_table_for_update(self):
  324. self.reload_table = True
  325. # When calling update_table() directly, always use out_of_schedule=True!
  326. def update_table(self, out_of_schedule=False):
  327. if self.reload_table:
  328. self.fill_table()
  329. for vm_row in self.vms_in_table:
  330. vm_row.update(self.counter)
  331. self.table_selection_changed()
  332. if not out_of_schedule:
  333. self.counter += 1
  334. QTimer.singleShot (self.update_interval, self.update_table)
  335. def table_selection_changed (self):
  336. vm = self.get_selected_vm()
  337. # Update available actions:
  338. self.action_removevm.setEnabled(not vm.installed_by_rpm and not vm.is_running())
  339. #self.action_resumevm.setEnabled(not vm.is_running())
  340. #self.action_pausevm.setEnabled(vm.is_running() and vm.qid != 0)
  341. self.action_shutdownvm.setEnabled(vm.is_running() and vm.qid != 0)
  342. def closeEvent (self, event):
  343. self.hide()
  344. event.ignore()
  345. def create_appvm(self):
  346. dialog = NewAppVmDlg()
  347. # Theoretically we should be locking for writing here and unlock
  348. # only after the VM creation finished. But the code would be more messy...
  349. # Instead we lock for writing in the actual worker thread
  350. self.qvm_collection.lock_db_for_reading()
  351. self.qvm_collection.load()
  352. self.qvm_collection.unlock_db()
  353. label_list = QubesVmLabels.values()
  354. label_list.sort(key=lambda l: l.index)
  355. for (i, label) in enumerate(label_list):
  356. dialog.vmlabel.insertItem(i, label.name)
  357. dialog.vmlabel.setItemIcon (i, QIcon(label.icon_path))
  358. template_vm_list = [vm for vm in self.qvm_collection.values() if vm.is_templete()]
  359. default_index = 0
  360. for (i, vm) in enumerate(template_vm_list):
  361. if vm is self.qvm_collection.get_default_template_vm():
  362. default_index = i
  363. dialog.template_name.insertItem(i, vm.name + " (default)")
  364. else:
  365. dialog.template_name.insertItem(i, vm.name)
  366. dialog.template_name.setCurrentIndex(default_index)
  367. dialog.vmname.selectAll()
  368. dialog.vmname.setFocus()
  369. if dialog.exec_():
  370. vmname = str(dialog.vmname.text())
  371. if self.qvm_collection.get_vm_by_name(vmname) is not None:
  372. QMessageBox.warning (None, "Incorrect AppVM Name!", "A VM with the name <b>{0}</b> already exists in the system!".format(vmname))
  373. return
  374. label = label_list[dialog.vmlabel.currentIndex()]
  375. template_vm = template_vm_list[dialog.template_name.currentIndex()]
  376. thread_monitor = ThreadMonitor()
  377. thread = threading.Thread (target=self.do_create_appvm, args=(vmname, label, template_vm, thread_monitor))
  378. thread.daemon = True
  379. thread.start()
  380. progress = QProgressDialog ("Creating new AppVM <b>{0}</b>...".format(vmname), "", 0, 0)
  381. progress.setCancelButton(None)
  382. progress.setModal(True)
  383. progress.show()
  384. while not thread_monitor.is_finished():
  385. app.processEvents()
  386. time.sleep (0.1)
  387. progress.hide()
  388. if thread_monitor.success:
  389. trayIcon.showMessage ("Qubes Manager", "VM '{0}' has been created.".format(vmname), msecs=3000)
  390. else:
  391. QMessageBox.warning (None, "Error creating AppVM!", "ERROR: {0}".format(thread_monitor.error_msg))
  392. def do_create_appvm (self, vmname, label, template_vm, thread_monitor):
  393. try:
  394. self.qvm_collection.lock_db_for_writing()
  395. self.qvm_collection.load()
  396. vm = self.qvm_collection.add_new_appvm(vmname, template_vm, label = label)
  397. vm.create_on_disk(verbose=False)
  398. vm.add_to_xen_storage()
  399. self.qvm_collection.save()
  400. except Exception as ex:
  401. thread_monitor.set_error_msg (str(ex))
  402. vm.remove_from_disk()
  403. finally:
  404. self.qvm_collection.unlock_db()
  405. thread_monitor.set_finished()
  406. def get_selected_vm(self):
  407. row_index = self.table.currentRow()
  408. assert self.vms_in_table[row_index] is not None
  409. vm = self.vms_in_table[row_index].vm
  410. return vm
  411. def remove_appvm(self):
  412. vm = self.get_selected_vm()
  413. assert not vm.is_running()
  414. assert not vm.installed_by_rpm
  415. self.qvm_collection.lock_db_for_reading()
  416. self.qvm_collection.load()
  417. self.qvm_collection.unlock_db()
  418. if vm.is_templete():
  419. dependent_vms = self.qvm_collection.get_vms_based_on(vm.qid)
  420. if len(dependent_vms) > 0:
  421. QMessageBox.warning (None, "Warning!",
  422. "This Template VM cannot be removed, because there is at least one AppVM that is based on it.<br>"
  423. "<small>If you want to remove this Template VM and all the AppVMs based on it,"
  424. "you should first remove each individual AppVM that uses this template.</small>")
  425. return
  426. reply = QMessageBox.question(None, "VM Removal Confirmation",
  427. "Are you sure you want to remove the VM <b>'{0}'</b>?<br>"
  428. "<small>All data on this VM's private storage will be lost!</small>".format(vm.name),
  429. QMessageBox.Yes | QMessageBox.Cancel)
  430. if reply == QMessageBox.Yes:
  431. thread_monitor = ThreadMonitor()
  432. thread = threading.Thread (target=self.do_remove_vm, args=(vm, thread_monitor))
  433. thread.daemon = True
  434. thread.start()
  435. progress = QProgressDialog ("Removing VM: <b>{0}</b>...".format(vm.name), "", 0, 0)
  436. progress.setCancelButton(None)
  437. progress.setModal(True)
  438. progress.show()
  439. while not thread_monitor.is_finished():
  440. app.processEvents()
  441. time.sleep (0.1)
  442. progress.hide()
  443. if thread_monitor.success:
  444. trayIcon.showMessage ("Qubes Manager", "VM '{0}' has been removed.".format(vm.name), msecs=3000)
  445. else:
  446. QMessageBox.warning (None, "Error removing M!", "ERROR: {0}".format(thread_monitor.error_msg))
  447. def do_remove_vm (self, vm, thread_monitor):
  448. try:
  449. self.qvm_collection.lock_db_for_writing()
  450. self.qvm_collection.load()
  451. #TODO: the following two conditions should really be checked by qvm_collection.pop() overload...
  452. if vm.is_templete() and qvm_collection.default_template_qid == vm.qid:
  453. qvm_collection.default_template_qid = None
  454. if vm.is_netvm() and qvm_collection.default_netvm_qid == vm.qid:
  455. qvm_collection.default_netvm_qid = None
  456. vm.remove_from_xen_storage()
  457. vm.remove_from_disk()
  458. self.qvm_collection.pop(vm.qid)
  459. self.qvm_collection.save()
  460. except Exception as ex:
  461. thread_monitor.set_error_msg (str(ex))
  462. finally:
  463. self.qvm_collection.unlock_db()
  464. thread_monitor.set_finished()
  465. def resume_vm(self):
  466. pass
  467. def pause_vm(self):
  468. pass
  469. def shutdown_vm(self):
  470. vm = self.get_selected_vm()
  471. assert vm.is_running()
  472. reply = QMessageBox.question(None, "VM Shutdown Confirmation",
  473. "Are you sure you want to power down the VM <b>'{0}'</b>?<br>"
  474. "<small>This will shutdown all the running applications within this VM.</small>".format(vm.name),
  475. QMessageBox.Yes | QMessageBox.Cancel)
  476. if reply == QMessageBox.Yes:
  477. try:
  478. subprocess.check_call (["/usr/sbin/xm", "shutdown", vm.name])
  479. except Exception as ex:
  480. QMessageBox.warning (None, "Error shutting down VM!", "ERROR: {0}".format(ex))
  481. return
  482. trayIcon.showMessage ("Qubes Manager", "VM '{0}' is shutting down...".format(vm.name), msecs=3000)
  483. self.shutdown_monitor[vm.qid] = VmShutdownMonitor (vm)
  484. QTimer.singleShot (vm_shutdown_timeout, self.shutdown_monitor[vm.qid].check_if_vm_has_shutdown)
  485. def showcpuload(self):
  486. pass
  487. class QubesTrayIcon(QSystemTrayIcon):
  488. def __init__(self, icon):
  489. QSystemTrayIcon.__init__(self, icon)
  490. self.menu = QMenu()
  491. action_showmanager = self.createAction ("Open VM Manager", slot=show_manager, icon="qubes")
  492. action_backup = self.createAction ("Make backup")
  493. action_preferences = self.createAction ("Preferences")
  494. action_set_netvm = self.createAction ("Set default NetVM", icon="networking")
  495. action_sys_info = self.createAction ("System Info", icon="dom0")
  496. action_exit = self.createAction ("Exit", slot=exit_app)
  497. action_backup.setDisabled(True)
  498. action_preferences.setDisabled(True)
  499. action_set_netvm.setDisabled(True)
  500. action_sys_info.setDisabled(True)
  501. self.addActions (self.menu, (action_showmanager, action_backup, action_sys_info, None, action_preferences, action_set_netvm, None, action_exit))
  502. self.setContextMenu(self.menu)
  503. self.connect (self, SIGNAL("activated (QSystemTrayIcon::ActivationReason)"), self.icon_clicked)
  504. def icon_clicked(self, reason):
  505. if reason == QSystemTrayIcon.Context:
  506. # Handle the right click normally, i.e. display the context menu
  507. return
  508. else:
  509. show_manager()
  510. def addActions(self, target, actions):
  511. for action in actions:
  512. if action is None:
  513. target.addSeparator()
  514. else:
  515. target.addAction(action)
  516. def createAction(self, text, slot=None, shortcut=None, icon=None,
  517. tip=None, checkable=False, signal="triggered()"):
  518. action = QAction(text, self)
  519. if icon is not None:
  520. action.setIcon(QIcon(":/%s.png" % icon))
  521. if shortcut is not None:
  522. action.setShortcut(shortcut)
  523. if tip is not None:
  524. action.setToolTip(tip)
  525. action.setStatusTip(tip)
  526. if slot is not None:
  527. self.connect(action, SIGNAL(signal), slot)
  528. if checkable:
  529. action.setCheckable(True)
  530. return action
  531. def show_manager():
  532. manager_window.show()
  533. def exit_app():
  534. notifier.stop()
  535. app.exit()
  536. # Bases on the original code by:
  537. # Copyright (c) 2002-2007 Pascal Varet <p.varet@gmail.com>
  538. def handle_exception( exc_type, exc_value, exc_traceback ):
  539. import sys
  540. import os.path
  541. import traceback
  542. filename, line, dummy, dummy = traceback.extract_tb( exc_traceback ).pop()
  543. filename = os.path.basename( filename )
  544. error = "%s: %s" % ( exc_type.__name__, exc_value )
  545. QMessageBox.critical(None, "Houston, we have a problem...",
  546. "Whoops. A critical error has occured. This is most likely a bug "
  547. "in Qubes Manager.<br><br>"
  548. "<b><i>%s</i></b>" % error +
  549. "at <b>line %d</b> of file <b>%s</b>.<br/><br/>"
  550. % ( line, filename ))
  551. #sys.exit(1)
  552. def main():
  553. # Avoid starting more than one instance of the app
  554. lock = QubesDaemonPidfile ("qubes-manager")
  555. if lock.pidfile_exists():
  556. if lock.pidfile_is_stale():
  557. lock.remove_pidfile()
  558. print "Removed stale pidfile (has the previous daemon instance crashed?)."
  559. else:
  560. exit (0)
  561. lock.create_pidfile()
  562. global app
  563. app = QApplication(sys.argv)
  564. app.setOrganizationName("The Qubes Project")
  565. app.setOrganizationDomain("http://qubes-os.org")
  566. app.setApplicationName("Qubes VM Manager")
  567. app.setWindowIcon(QIcon(":/qubes.png"))
  568. sys.excepthook = handle_exception
  569. global manager_window
  570. manager_window = VmManagerWindow()
  571. wm = WatchManager()
  572. mask = EventsCodes.OP_FLAGS.get('IN_CLOSE_WRITE')
  573. global notifier
  574. notifier = ThreadedNotifier(wm, QubesConfigFileWatcher(manager_window.mark_table_for_update))
  575. notifier.start()
  576. wdd = wm.add_watch(qubes_store_filename, mask)
  577. global trayIcon
  578. trayIcon = QubesTrayIcon(QIcon(":/qubes.png"))
  579. trayIcon.show()
  580. app.exec_()
  581. trayIcon = None