main.py 77 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075
  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. import os
  24. import signal
  25. import fcntl
  26. import errno
  27. import dbus
  28. import dbus.service
  29. from dbus.mainloop.qt import DBusQtMainLoop
  30. from PyQt4.QtCore import *
  31. from PyQt4.QtGui import *
  32. from qubes.qubes import QubesVmCollection
  33. from qubes.qubes import QubesException
  34. from qubes.qubes import qubes_store_filename
  35. from qubes.qubes import QubesVmLabels
  36. from qubes.qubes import dry_run
  37. from qubes.qubes import qubes_guid_path
  38. from qubes.qubes import QubesDaemonPidfile
  39. from qubes.qubes import QubesHost
  40. from qubes import qubes
  41. from qubes import qubesutils
  42. import qubesmanager.resources_rc
  43. import ui_newappvmdlg
  44. from ui_mainwindow import *
  45. from create_new_vm import NewVmDlg
  46. from settings import VMSettingsWindow
  47. from restore import RestoreVMsWindow
  48. from backup import BackupVMsWindow
  49. from global_settings import GlobalSettingsWindow
  50. from log_dialog import LogDialog
  51. from thread_monitor import *
  52. from pyinotify import WatchManager, Notifier, ThreadedNotifier, EventsCodes, ProcessEvent
  53. import subprocess
  54. import time
  55. from datetime import datetime,timedelta
  56. from qubes.qubes import updates_stat_file
  57. qubes_dom0_updates_stat_file = '/var/lib/qubes/updates/dom0-updates-available'
  58. qubes_guid_path = '/usr/bin/qubes_guid'
  59. update_suggestion_interval = 14 # 14 days
  60. dbus_object_path = '/org/qubesos/QubesManager'
  61. dbus_interface = 'org.qubesos.QubesManager'
  62. system_bus = None
  63. power_order = Qt.DescendingOrder
  64. update_order = Qt.AscendingOrder
  65. class QubesConfigFileWatcher(ProcessEvent):
  66. def __init__ (self, update_func):
  67. self.update_func = update_func
  68. def process_IN_MODIFY (self, event):
  69. self.update_func()
  70. class VmIconWidget (QWidget):
  71. def __init__(self, icon_path, enabled=True, size_multiplier=0.7, tooltip = None, parent=None):
  72. super(VmIconWidget, self).__init__(parent)
  73. self.label_icon = QLabel()
  74. icon = QIcon (icon_path)
  75. icon_sz = QSize (VmManagerWindow.row_height * size_multiplier, VmManagerWindow.row_height * size_multiplier)
  76. icon_pixmap = icon.pixmap(icon_sz, QIcon.Disabled if not enabled else QIcon.Normal)
  77. self.label_icon.setPixmap (icon_pixmap)
  78. self.label_icon.setFixedSize (icon_sz)
  79. if tooltip != None:
  80. self.label_icon.setToolTip(tooltip)
  81. layout = QHBoxLayout()
  82. layout.addWidget(self.label_icon)
  83. layout.setContentsMargins(0,0,0,0)
  84. self.setLayout(layout)
  85. def setToolTip(self, tooltip):
  86. if tooltip is not None:
  87. self.label_icon.setToolTip(tooltip)
  88. else:
  89. self.label_icon.setToolTip('')
  90. class VmTypeWidget(VmIconWidget):
  91. class VmTypeItem(QTableWidgetItem):
  92. def __init__(self, value):
  93. super(VmTypeWidget.VmTypeItem, self).__init__()
  94. self.value = value
  95. def set_value(self, value):
  96. self.value = value
  97. def __lt__(self, other):
  98. return self.value < other.value
  99. def __init__(self, vm, parent=None):
  100. (icon_path, tooltip) = self.get_vm_icon(vm)
  101. super (VmTypeWidget, self).__init__(icon_path, True, 0.8, tooltip, parent)
  102. self.vm = vm
  103. self.tableItem = self.VmTypeItem(self.value)
  104. def get_vm_icon(self, vm):
  105. if vm.qid == 0:
  106. self.value = 0
  107. return (":/dom0.png", "Dom0")
  108. elif vm.is_netvm() and not vm.is_proxyvm():
  109. self.value = 1
  110. return (":/netvm.png", "NetVM")
  111. elif vm.is_proxyvm():
  112. self.value = 2
  113. return (":/proxyvm.png", "ProxyVM")
  114. elif vm.is_template():
  115. self.value = 3
  116. return (":/templatevm.png", "TemplateVM")
  117. elif vm.is_appvm() and vm.template is None:
  118. self.value = 4
  119. return (":/standalonevm.png", "StandaloneVM")
  120. elif vm.type == "HVM":
  121. self.value = 5
  122. return (":/hvm.png", "HVM")
  123. elif vm.is_appvm() or vm.is_disposablevm():
  124. self.value = 5 + vm.label.index
  125. return (":/appvm.png", "AppVM")
  126. class VmLabelWidget(VmIconWidget):
  127. class VmLabelItem(QTableWidgetItem):
  128. def __init__(self, value):
  129. super(VmLabelWidget.VmLabelItem, self).__init__()
  130. self.value = value
  131. def set_value(self, value):
  132. self.value = value
  133. def __lt__(self, other):
  134. return self.value < other.value
  135. def __init__(self, vm, parent=None):
  136. icon_path = self.get_vm_icon_path(vm)
  137. super (VmLabelWidget, self).__init__(icon_path, True, 0.8, None, parent)
  138. self.vm = vm
  139. self.tableItem = self.VmLabelItem(self.value)
  140. def get_vm_icon_path(self, vm):
  141. if vm.qid == 0:
  142. self.value = 100
  143. return ":/off.png"
  144. else:
  145. self.value = vm.label.index
  146. return vm.label.icon_path
  147. class VmNameItem (QTableWidgetItem):
  148. def __init__(self, vm):
  149. super(VmNameItem, self).__init__()
  150. self.setFlags(Qt.ItemIsSelectable|Qt.ItemIsEnabled)
  151. self.setText(vm.name)
  152. self.setTextAlignment(Qt.AlignVCenter)
  153. self.qid = vm.qid
  154. class VmStatusIcon(QLabel):
  155. def __init__(self, vm, parent=None):
  156. super (VmStatusIcon, self).__init__(parent)
  157. self.vm = vm
  158. self.set_on_icon()
  159. self.previous_power_state = vm.last_power_state
  160. def update(self):
  161. if self.previous_power_state != self.vm.last_power_state:
  162. self.set_on_icon()
  163. self.previous_power_state = self.vm.last_power_state
  164. def set_on_icon(self):
  165. if self.vm.last_power_state == "Running":
  166. icon = QIcon (":/on.png")
  167. elif self.vm.last_power_state in ["Transient", "Halting", "Dying"]:
  168. icon = QIcon (":/transient.png")
  169. else:
  170. icon = QIcon (":/off.png")
  171. icon_sz = QSize (VmManagerWindow.row_height * 0.5, VmManagerWindow.row_height *0.5)
  172. icon_pixmap = icon.pixmap(icon_sz)
  173. self.setPixmap (icon_pixmap)
  174. self.setFixedSize (icon_sz)
  175. class VmInfoWidget (QWidget):
  176. class VmInfoItem (QTableWidgetItem):
  177. def __init__(self, upd_info_item, vm):
  178. super(VmInfoWidget.VmInfoItem, self).__init__()
  179. self.upd_info_item = upd_info_item
  180. self.vm = vm
  181. def __lt__(self, other):
  182. self_val = self.upd_info_item.value
  183. other_val = other.upd_info_item.value
  184. if self.tableWidget().horizontalHeader().sortIndicatorOrder() == update_order:
  185. # the result will be sorted by upd, sorting order: Ascending
  186. self_val += 1 if self.vm.is_running() else 0
  187. other_val += 1 if other.vm.is_running() else 0
  188. return (self_val) > (other_val)
  189. elif self.tableWidget().horizontalHeader().sortIndicatorOrder() == power_order:
  190. #the result will be sorted by power state, sorting order: Descending
  191. self_val = -(self_val/10 + 10*(1 if self.vm.is_running() else 0))
  192. other_val = -(other_val/10 + 10*(1 if other.vm.is_running() else 0))
  193. return (self_val) > (other_val)
  194. else:
  195. #it would be strange if this happened
  196. return
  197. def __init__(self, vm, parent = None):
  198. super (VmInfoWidget, self).__init__(parent)
  199. self.vm = vm
  200. layout = QHBoxLayout ()
  201. self.on_icon = VmStatusIcon(vm)
  202. self.upd_info = VmUpdateInfoWidget(vm, show_text=False)
  203. self.error_icon = VmIconWidget(":/warning.png")
  204. self.blk_icon = VmIconWidget(":/mount.png")
  205. layout.addWidget(self.on_icon)
  206. layout.addWidget(self.upd_info)
  207. layout.addWidget(self.error_icon)
  208. layout.addItem(QSpacerItem(0, 10, QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding))
  209. layout.addWidget(self.blk_icon)
  210. layout.setContentsMargins(5,0,5,0)
  211. self.setLayout(layout)
  212. self.blk_icon.setVisible(False)
  213. self.error_icon.setVisible(False)
  214. self.tableItem = self.VmInfoItem(self.upd_info.tableItem, vm)
  215. def update_vm_state(self, vm, blk_visible):
  216. self.on_icon.update()
  217. self.upd_info.update_outdated(vm)
  218. if blk_visible != None:
  219. self.blk_icon.setVisible(blk_visible)
  220. self.error_icon.setToolTip(vm.error_msg)
  221. self.error_icon.setVisible(vm.error_msg is not None)
  222. class VmTemplateItem (QTableWidgetItem):
  223. def __init__(self, vm):
  224. super(VmTemplateItem, self).__init__()
  225. self.setFlags(Qt.ItemIsSelectable|Qt.ItemIsEnabled)
  226. if vm.template is not None:
  227. self.setText(vm.template.name)
  228. else:
  229. font = QFont()
  230. font.setStyle(QFont.StyleItalic)
  231. self.setFont(font)
  232. self.setTextColor(QColor("gray"))
  233. if vm.is_appvm(): # and vm.template is None
  234. self.setText("StandaloneVM")
  235. elif vm.is_template():
  236. self.setText("TemplateVM")
  237. elif vm.qid == 0:
  238. self.setText("AdminVM")
  239. elif vm.is_netvm():
  240. self.setText("NetVM")
  241. else:
  242. self.setText("---")
  243. self.setTextAlignment(Qt.AlignVCenter)
  244. class VmNetvmItem (QTableWidgetItem):
  245. def __init__(self, vm):
  246. super(VmNetvmItem, self).__init__()
  247. self.setFlags(Qt.ItemIsSelectable|Qt.ItemIsEnabled)
  248. if vm.is_netvm() and not vm.is_proxyvm():
  249. self.setText("n/a")
  250. elif vm.netvm is not None:
  251. self.setText(vm.netvm.name)
  252. else:
  253. self.setText("---")
  254. self.setTextAlignment(Qt.AlignVCenter)
  255. class VmUsageBarWidget (QWidget):
  256. class VmUsageBarItem (QTableWidgetItem):
  257. def __init__(self, value):
  258. super(VmUsageBarWidget.VmUsageBarItem, self).__init__()
  259. self.value = value
  260. def set_value(self, value):
  261. self.value = value
  262. def __lt__(self, other):
  263. return int(self.value) < int(other.value)
  264. def __init__(self, min, max, format, update_func, vm, load, hue=210, parent = None):
  265. super (VmUsageBarWidget, self).__init__(parent)
  266. self.min = min
  267. self.max = max
  268. self.update_func = update_func
  269. self.value = min
  270. self.widget = QProgressBar()
  271. self.widget.setMinimum(min)
  272. self.widget.setMaximum(max)
  273. self.widget.setFormat(format);
  274. self.widget.setStyleSheet(
  275. "QProgressBar:horizontal{" +\
  276. "border: 1px solid hsv({0}, 100, 250);".format(hue) +\
  277. "border-radius: 4px;\
  278. background: white;\
  279. text-align: center;\
  280. }\
  281. QProgressBar::chunk:horizontal {\
  282. background: qlineargradient(x1: 0, y1: 0.5, x2: 1, y2: 0.5, " +\
  283. "stop: 0 hsv({0}, 170, 207),".format(hue) +
  284. " stop: 1 white); \
  285. }"
  286. )
  287. layout = QHBoxLayout()
  288. layout.addWidget(self.widget)
  289. self.setLayout(layout)
  290. self.tableItem = self.VmUsageBarItem(min)
  291. self.update_load(vm, load)
  292. def update_load(self, vm, load):
  293. self.value = self.update_func(vm, load)
  294. self.widget.setValue(self.value)
  295. self.tableItem.set_value(self.value)
  296. class ChartWidget (QWidget):
  297. class ChartItem (QTableWidgetItem):
  298. def __init__(self, value):
  299. super(ChartWidget.ChartItem, self).__init__()
  300. self.value = value
  301. def set_value(self, value):
  302. self.value = value
  303. def __lt__(self, other):
  304. return self.value < other.value
  305. def __init__(self, vm, update_func, hue, load = 0, parent = None):
  306. super (ChartWidget, self).__init__(parent)
  307. self.update_func = update_func
  308. self.hue = hue
  309. if hue < 0 or hue > 255:
  310. self.hue = 255
  311. self.load = load
  312. assert self.load >= 0 and self.load <= 100, "load = {0}".format(self.load)
  313. self.load_history = [self.load]
  314. self.tableItem = ChartWidget.ChartItem(self.load)
  315. def update_load (self, vm, load):
  316. self.load = self.update_func(vm, load)
  317. assert self.load >= 0, "load = {0}".format(self.load)
  318. # assert self.load >= 0 and self.load <= 100, "load = {0}".format(self.load)
  319. if self.load > 100:
  320. # FIXME: This is an ugly workaround for cpu_load:/
  321. self.load = 100
  322. self.load_history.append (self.load)
  323. self.tableItem.set_value(self.load)
  324. self.repaint()
  325. def paintEvent (self, Event = None):
  326. p = QPainter (self)
  327. dx = 4
  328. W = self.width()
  329. H = self.height() - 5
  330. N = len(self.load_history)
  331. if N > W/dx:
  332. tail = N - W/dx
  333. N = W/dx
  334. self.load_history = self.load_history[tail:]
  335. assert len(self.load_history) == N
  336. for i in range (0, N-1):
  337. val = self.load_history[N- i - 1]
  338. sat = 70 + val*(255-70)/100
  339. color = QColor.fromHsv (self.hue, sat, 255)
  340. pen = QPen (color)
  341. pen.setWidth(dx-1)
  342. p.setPen(pen)
  343. if val > 0:
  344. p.drawLine (W - i*dx - dx, H , W - i*dx - dx, H - (H - 5) * val/100)
  345. class VmUpdateInfoWidget(QWidget):
  346. class VmUpdateInfoItem (QTableWidgetItem):
  347. def __init__(self, value):
  348. super(VmUpdateInfoWidget.VmUpdateInfoItem, self).__init__()
  349. self.value = 0
  350. self.set_value(value)
  351. def set_value(self, value):
  352. if value == "outdated":
  353. self.value = 30
  354. elif value == "update":
  355. self.value = 20
  356. else:
  357. self.value = 0
  358. def __lt__(self, other):
  359. return self.value < other.value
  360. def __init__(self, vm, show_text=True, parent = None):
  361. super (VmUpdateInfoWidget, self).__init__(parent)
  362. layout = QHBoxLayout ()
  363. self.show_text = show_text
  364. if self.show_text:
  365. self.label=QLabel("")
  366. layout.addWidget(self.label, alignment=Qt.AlignCenter)
  367. else:
  368. self.icon = QLabel("")
  369. layout.addWidget(self.icon, alignment=Qt.AlignCenter)
  370. self.setLayout(layout)
  371. self.previous_outdated = False
  372. self.previous_update_recommended = None
  373. self.value = None
  374. self.tableItem = VmUpdateInfoWidget.VmUpdateInfoItem(self.value)
  375. def update_outdated(self, vm):
  376. if vm.type == "HVM":
  377. return
  378. outdated = vm.is_outdated()
  379. if outdated and not self.previous_outdated:
  380. self.update_status_widget("outdated")
  381. elif not outdated and self.previous_outdated:
  382. self.update_status_widget(None)
  383. self.previous_outdated = outdated
  384. if not vm.is_updateable():
  385. return
  386. if vm.qid == 0:
  387. update_recommended = self.previous_update_recommended
  388. if os.path.exists(qubes_dom0_updates_stat_file):
  389. update_recommended = True
  390. else:
  391. update_recommended = False
  392. else:
  393. update_recommended = self.previous_update_recommended
  394. stat_file_path = vm.dir_path + '/' + updates_stat_file
  395. if not os.path.exists(stat_file_path):
  396. update_recommended = False
  397. else:
  398. if (not hasattr(vm, "updates_stat_file_read_time")) or vm.updates_stat_file_read_time <= os.path.getmtime(stat_file_path):
  399. stat_file = open(stat_file_path, "r")
  400. updates = stat_file.read().strip()
  401. stat_file.close()
  402. if updates.isdigit():
  403. updates = int(updates)
  404. else:
  405. updates = 0
  406. if updates == 0:
  407. update_recommended = False
  408. else:
  409. update_recommended = True
  410. vm.updates_stat_file_read_time = time.time()
  411. if update_recommended and not self.previous_update_recommended:
  412. self.update_status_widget("update")
  413. elif self.previous_update_recommended and not update_recommended:
  414. self.update_status_widget(None)
  415. self.previous_update_recommended = update_recommended
  416. def update_status_widget(self, state):
  417. self.value = state
  418. self.tableItem.set_value(state)
  419. if state == "update":
  420. label_text = "<font color=\"#CCCC00\">Check updates</font>"
  421. icon_path = ":/update-recommended.png"
  422. tooltip_text = "Updates pending!"
  423. elif state == "outdated":
  424. label_text = "<font color=\"red\">VM outdated</font>"
  425. icon_path = ":/outdated.png"
  426. tooltip_text = "The VM must be restarted for its filesystem to reflect the template's recent changes."
  427. elif state == None:
  428. label_text = ""
  429. icon_path = None
  430. tooltip_text = None
  431. if self.show_text:
  432. self.label.setText(label_text)
  433. else:
  434. self.layout().removeWidget(self.icon)
  435. self.icon.deleteLater()
  436. if icon_path != None:
  437. self.icon = VmIconWidget(icon_path, True, 0.7)
  438. self.icon.setToolTip(tooltip_text)
  439. else:
  440. self.icon = QLabel(label_text)
  441. self.layout().addWidget(self.icon, alignment=Qt.AlignCenter)
  442. class VmSizeOnDiskItem (QTableWidgetItem):
  443. def __init__(self, vm):
  444. super(VmSizeOnDiskItem, self).__init__()
  445. self.setFlags(Qt.ItemIsSelectable|Qt.ItemIsEnabled)
  446. self.vm = vm
  447. self.value = 0
  448. self.update()
  449. self.setTextAlignment(Qt.AlignVCenter)
  450. def update(self):
  451. if self.vm.qid == 0:
  452. self.setText("n/a")
  453. else:
  454. self.value = self.vm.get_disk_utilization()/(1024*1024)
  455. self.setText( str(self.value) + " MiB")
  456. def __lt__(self, other):
  457. return self.value < other.value
  458. class VmRowInTable(object):
  459. cpu_graph_hue = 210
  460. mem_graph_hue = 120
  461. def __init__(self, vm, row_no, table, block_manager):
  462. self.vm = vm
  463. self.row_no = row_no
  464. table.setRowHeight (row_no, VmManagerWindow.row_height)
  465. self.type_widget = VmTypeWidget(vm)
  466. table.setCellWidget(row_no, VmManagerWindow.columns_indices['Type'], self.type_widget)
  467. table.setItem(row_no, VmManagerWindow.columns_indices['Type'], self.type_widget.tableItem)
  468. self.label_widget = VmLabelWidget(vm)
  469. table.setCellWidget(row_no, VmManagerWindow.columns_indices['Label'], self.label_widget)
  470. table.setItem(row_no, VmManagerWindow.columns_indices['Label'], self.label_widget.tableItem)
  471. self.name_widget = VmNameItem(vm)
  472. table.setItem(row_no, VmManagerWindow.columns_indices['Name'], self.name_widget)
  473. self.info_widget = VmInfoWidget(vm)
  474. table.setCellWidget(row_no, VmManagerWindow.columns_indices['State'], self.info_widget)
  475. table.setItem(row_no, VmManagerWindow.columns_indices['State'], self.info_widget.tableItem)
  476. self.template_widget = VmTemplateItem(vm)
  477. table.setItem(row_no, VmManagerWindow.columns_indices['Template'], self.template_widget)
  478. self.netvm_widget = VmNetvmItem(vm)
  479. table.setItem(row_no, VmManagerWindow.columns_indices['NetVM'], self.netvm_widget)
  480. self.cpu_usage_widget = VmUsageBarWidget(0, 100, "%v %",
  481. lambda vm, val: val if vm.last_running else 0, vm, 0, self.cpu_graph_hue)
  482. table.setCellWidget(row_no, VmManagerWindow.columns_indices['CPU'], self.cpu_usage_widget)
  483. table.setItem(row_no, VmManagerWindow.columns_indices['CPU'], self.cpu_usage_widget.tableItem)
  484. self.load_widget = ChartWidget(vm, lambda vm, val: val if vm.last_running else 0, self.cpu_graph_hue, 0 )
  485. table.setCellWidget(row_no, VmManagerWindow.columns_indices['CPU Graph'], self.load_widget)
  486. table.setItem(row_no, VmManagerWindow.columns_indices['CPU Graph'], self.load_widget.tableItem)
  487. self.mem_usage_widget = VmUsageBarWidget(0, qubes_host.memory_total/1024, "%v MB",
  488. lambda vm, val: vm.get_mem()/1024 if vm.last_running else 0, vm, 0, self.mem_graph_hue)
  489. table.setCellWidget(row_no, VmManagerWindow.columns_indices['MEM'], self.mem_usage_widget)
  490. table.setItem(row_no, VmManagerWindow.columns_indices['MEM'], self.mem_usage_widget.tableItem)
  491. self.mem_widget = ChartWidget(vm, lambda vm, val: vm.get_mem()*100/qubes_host.memory_total if vm.last_running else 0, self.mem_graph_hue, 0)
  492. table.setCellWidget(row_no, VmManagerWindow.columns_indices['MEM Graph'], self.mem_widget)
  493. table.setItem(row_no, VmManagerWindow.columns_indices['MEM Graph'], self.mem_widget.tableItem)
  494. self.size_widget = VmSizeOnDiskItem(vm)
  495. table.setItem(row_no, VmManagerWindow.columns_indices['Size'], self.size_widget)
  496. def update(self, blk_visible = None, cpu_load = None, update_size_on_disk = False):
  497. self.info_widget.update_vm_state(self.vm, blk_visible)
  498. if cpu_load is not None:
  499. self.cpu_usage_widget.update_load(self.vm, cpu_load)
  500. self.mem_usage_widget.update_load(self.vm, None)
  501. self.load_widget.update_load(self.vm, cpu_load)
  502. self.mem_widget.update_load(self.vm, None)
  503. if update_size_on_disk == True:
  504. self.size_widget.update()
  505. vm_shutdown_timeout = 20000 # in msec
  506. class VmShutdownMonitor(QObject):
  507. def __init__(self, vm, shutdown_time = vm_shutdown_timeout):
  508. self.vm = vm
  509. self.shutdown_time = shutdown_time
  510. def check_if_vm_has_shutdown(self):
  511. vm = self.vm
  512. vm_start_time = vm.get_start_time()
  513. if not vm.is_running() or (vm_start_time and vm_start_time >= datetime.utcnow() - timedelta(0,self.shutdown_time/1000)):
  514. return
  515. reply = QMessageBox.question(None, "VM Shutdown",
  516. "The VM <b>'{0}'</b> hasn't shutdown within the last {1} seconds, do you want to kill it?<br>".format(vm.name, self.shutdown_time/1000),
  517. "Kill it!", "Wait another {0} seconds...".format(self.shutdown_time/1000))
  518. if reply == 0:
  519. vm.force_shutdown()
  520. else:
  521. QTimer.singleShot (self.shutdown_time, self.check_if_vm_has_shutdown)
  522. class VmManagerWindow(Ui_VmManagerWindow, QMainWindow):
  523. row_height = 30
  524. column_width = 200
  525. min_visible_rows = 10
  526. update_interval = 1000 # in msec
  527. show_inactive_vms = True
  528. columns_indices = { "Type": 0,
  529. "Label": 1,
  530. "Name": 2,
  531. "State": 3,
  532. "Template": 4,
  533. "NetVM": 5,
  534. "CPU": 6,
  535. "CPU Graph": 7,
  536. "MEM": 8,
  537. "MEM Graph": 9,
  538. "Size": 10,}
  539. def __init__(self, parent=None):
  540. super(VmManagerWindow, self).__init__()
  541. self.setupUi(self)
  542. self.toolbar = self.toolBar
  543. self.qubes_watch = qubesutils.QubesWatch()
  544. self.qvm_collection = QubesVmCollection()
  545. self.blk_manager = QubesBlockDevicesManager(self.qvm_collection)
  546. self.qubes_watch.setup_block_watch(self.blk_manager.block_devs_event)
  547. self.blk_watch_thread = threading.Thread(target=self.qubes_watch.watch_loop)
  548. self.blk_watch_thread.daemon = True
  549. self.blk_watch_thread.start()
  550. self.connect(self.table, SIGNAL("itemSelectionChanged()"), self.table_selection_changed)
  551. self.table.setColumnWidth(0, self.column_width)
  552. self.setSizeIncrement(QtCore.QSize(200, 30))
  553. self.centralwidget.setSizeIncrement(QtCore.QSize(200, 30))
  554. self.table.setSizeIncrement(QtCore.QSize(200, 30))
  555. self.sort_by_mem = None
  556. self.sort_by_cpu = None
  557. self.sort_by_state = None
  558. self.sort_by_size_on_disk = None
  559. self.screen_number = -1
  560. self.screen_changed = False
  561. self.running_vms_count = 0
  562. self.vm_errors = {}
  563. self.frame_width = 0
  564. self.frame_height = 0
  565. self.fill_table()
  566. self.move(self.x(), 0)
  567. self.columns_actions = {}
  568. self.columns_actions[ self.columns_indices["Type"] ] = self.action_vm_type
  569. self.columns_actions[ self.columns_indices["Label"] ] = self.action_label
  570. self.columns_actions[ self.columns_indices["Name"] ] = self.action_name
  571. self.columns_actions[ self.columns_indices["State"] ] = self.action_state
  572. self.columns_actions[ self.columns_indices["Template"] ] = self.action_template
  573. self.columns_actions[ self.columns_indices["NetVM"] ] = self.action_netvm
  574. self.columns_actions[ self.columns_indices["CPU"] ] = self.action_cpu
  575. self.columns_actions[ self.columns_indices["CPU Graph"] ] = self.action_cpu_graph
  576. self.columns_actions[ self.columns_indices["MEM"] ] = self.action_mem
  577. self.columns_actions[ self.columns_indices["MEM Graph"] ] = self.action_mem_graph
  578. self.columns_actions[ self.columns_indices["Size"] ] = self.action_size_on_disk
  579. self.visible_columns_count = len(self.columns_indices);
  580. self.table.setColumnHidden( self.columns_indices["NetVM"], True)
  581. self.action_netvm.setChecked(False)
  582. self.table.setColumnHidden( self.columns_indices["CPU Graph"], True)
  583. self.action_cpu_graph.setChecked(False)
  584. self.table.setColumnHidden( self.columns_indices["MEM Graph"], True)
  585. self.action_mem_graph.setChecked(False)
  586. self.table.setColumnHidden( self.columns_indices["Size"], True)
  587. self.action_size_on_disk.setChecked(False)
  588. self.table.setColumnWidth(self.columns_indices["State"], 80)
  589. self.table.setColumnWidth(self.columns_indices["Name"], 150)
  590. self.table.setColumnWidth(self.columns_indices["Label"], 40)
  591. self.table.setColumnWidth(self.columns_indices["Type"], 40)
  592. self.table.setColumnWidth(self.columns_indices["Size"], 100)
  593. self.action_showallvms.setChecked(True)
  594. self.table.horizontalHeader().setResizeMode(QHeaderView.Fixed)
  595. self.table.sortItems(self.columns_indices["Type"], Qt.AscendingOrder)
  596. self.context_menu = QMenu(self)
  597. self.context_menu.addAction(self.action_settings)
  598. self.context_menu.addAction(self.action_editfwrules)
  599. self.context_menu.addAction(self.action_appmenus)
  600. self.context_menu.addAction(self.action_set_keyboard_layout)
  601. self.context_menu.addMenu(self.blk_menu)
  602. self.context_menu.addSeparator()
  603. self.context_menu.addAction(self.action_updatevm)
  604. self.context_menu.addAction(self.action_run_command_in_vm)
  605. self.context_menu.addAction(self.action_resumevm)
  606. self.context_menu.addAction(self.action_pausevm)
  607. self.context_menu.addAction(self.action_shutdownvm)
  608. self.context_menu.addAction(self.action_killvm)
  609. self.context_menu.addSeparator()
  610. self.context_menu.addAction(self.action_clonevm)
  611. self.context_menu.addAction(self.action_removevm)
  612. self.context_menu.addSeparator()
  613. self.context_menu.addMenu(self.logs_menu)
  614. self.context_menu.addSeparator()
  615. self.table_selection_changed()
  616. self.connect(self.table.horizontalHeader(), SIGNAL("sortIndicatorChanged(int, Qt::SortOrder)"), self.sortIndicatorChanged)
  617. self.connect(self.table, SIGNAL("customContextMenuRequested(const QPoint&)"), self.open_context_menu)
  618. self.connect(self.blk_menu, SIGNAL("triggered(QAction *)"), self.attach_dettach_device_triggered)
  619. self.connect(self.logs_menu, SIGNAL("triggered(QAction *)"), self.show_log)
  620. self.table.setContentsMargins(0,0,0,0)
  621. self.centralwidget.layout().setContentsMargins(0,0,0,0)
  622. self.layout().setContentsMargins(0,0,0,0)
  623. self.connect(self.action_menubar, SIGNAL("toggled(bool)"), self.showhide_menubar)
  624. self.connect(self.action_toolbar, SIGNAL("toggled(bool)"), self.showhide_toolbar)
  625. self.counter = 0
  626. self.update_size_on_disk = False
  627. self.shutdown_monitor = {}
  628. self.last_measure_results = {}
  629. self.last_measure_time = time.time()
  630. QTimer.singleShot (self.update_interval, self.update_table)
  631. def show(self):
  632. super(VmManagerWindow, self).show()
  633. self.screen_number = app.desktop().screenNumber(self)
  634. def set_table_geom_size(self):
  635. desktop_width = app.desktop().availableGeometry(self).width() - self.frame_width # might be wrong...
  636. desktop_height = app.desktop().availableGeometry(self).height() - self.frame_height # might be wrong...
  637. desktop_height -= self.row_height #UGLY! to somehow ommit taskbar...
  638. W = self.table.horizontalHeader().length() +\
  639. self.table.verticalScrollBar().width() +\
  640. 2*self.table.frameWidth() +1
  641. H = self.table.horizontalHeader().height() +\
  642. 2*self.table.frameWidth()
  643. mainwindow_to_add = 0
  644. available_space = desktop_height
  645. if self.menubar.isVisible():
  646. menubar_height = self.menubar.sizeHint().height() + self.menubar.contentsMargins().top() + self.menubar.contentsMargins().bottom()
  647. available_space -= menubar_height
  648. mainwindow_to_add += menubar_height
  649. if self.toolbar.isVisible():
  650. toolbar_height = self.toolbar.sizeHint().height() + self.toolbar.contentsMargins().top() + self.toolbar.contentsMargins().bottom()
  651. available_space -= toolbar_height
  652. mainwindow_to_add += toolbar_height
  653. if W >= desktop_width:
  654. available_space -= self.table.horizontalScrollBar().height()
  655. H += self.table.horizontalScrollBar().height()
  656. default_rows = int(available_space/self.row_height)
  657. if self.show_inactive_vms:
  658. n = self.table.rowCount()
  659. else:
  660. n = self.running_vms_count
  661. if n > default_rows:
  662. H += default_rows*self.row_height
  663. self.table.verticalScrollBar().show()
  664. else:
  665. H += n*self.row_height
  666. self.table.verticalScrollBar().hide()
  667. W -= self.table.verticalScrollBar().width()
  668. W = min(desktop_width, W)
  669. self.centralwidget.setFixedHeight(H)
  670. H += mainwindow_to_add
  671. self.setMaximumHeight(H)
  672. self.setMinimumHeight(H)
  673. self.table.setFixedWidth(W)
  674. self.centralwidget.setFixedWidth(W)
  675. # don't change the following two lines to setFixedWidth!
  676. self.setMaximumWidth(W)
  677. self.setMinimumWidth(W)
  678. def moveEvent(self, event):
  679. super(VmManagerWindow, self).moveEvent(event)
  680. screen_number = app.desktop().screenNumber(self)
  681. if self.screen_number != screen_number:
  682. self.screen_changed = True
  683. self.screen_number = screen_number
  684. def get_vms_list(self):
  685. self.qvm_collection.lock_db_for_reading()
  686. self.qvm_collection.load()
  687. self.qvm_collection.unlock_db()
  688. running_count = 0
  689. vms_list = [vm for vm in self.qvm_collection.values()]
  690. for vm in vms_list:
  691. vm.last_power_state = vm.get_power_state()
  692. vm.last_running = vm.last_power_state in ["Running", "Transient"]
  693. if vm.last_running:
  694. running_count += 1
  695. vm.error_msg = self.vm_errors[vm.qid] if vm.qid in self.vm_errors else None
  696. self.running_vms_count = running_count
  697. return vms_list
  698. def fill_table(self):
  699. self.table.setSortingEnabled(False)
  700. self.table.clearContents()
  701. vms_list = self.get_vms_list()
  702. self.table.setRowCount(len(vms_list))
  703. vms_in_table = {}
  704. row_no = 0
  705. for vm in vms_list:
  706. if vm.internal:
  707. continue
  708. vm_row = VmRowInTable (vm, row_no, self.table, self.blk_manager)
  709. vms_in_table[vm.qid] = vm_row
  710. row_no += 1
  711. self.table.setRowCount(row_no)
  712. self.vms_list = vms_list
  713. self.vms_in_table = vms_in_table
  714. self.reload_table = False
  715. self.table.setSortingEnabled(True)
  716. self.showhide_inactive_vms(True)
  717. self.set_table_geom_size()
  718. if not self.show_inactive_vms:
  719. self.showhide_inactive_vms(self.show_inactive_vms)
  720. self.set_table_geom_size()
  721. def showhide_inactive_vms(self, show_inactive):
  722. if show_inactive:
  723. row_no = 0
  724. while row_no < self.table.rowCount():
  725. self.table.setRowHidden(row_no, False)
  726. row_no += 1
  727. else:
  728. row_no = 0
  729. while row_no < self.table.rowCount():
  730. widget = self.table.cellWidget(row_no, self.columns_indices["State"])
  731. running = widget.vm.last_running
  732. if not running:
  733. self.table.setRowHidden(row_no, True)
  734. row_no += 1
  735. def mark_table_for_update(self):
  736. self.reload_table = True
  737. # When calling update_table() directly, always use out_of_schedule=True!
  738. def update_table(self, out_of_schedule=False):
  739. update_devs = self.update_block_devices() or out_of_schedule
  740. reload_table = self.reload_table
  741. if manager_window.isVisible():
  742. some_vms_have_changed_power_state = False
  743. for vm in self.vms_list:
  744. state = vm.get_power_state()
  745. if vm.last_power_state != state:
  746. prev_running = vm.last_running
  747. vm.last_power_state = state
  748. vm.last_running = (state in ["Running", "Transient"])
  749. if not prev_running and vm.last_running:
  750. self.running_vms_count += 1
  751. some_vms_have_changed_power_state = True
  752. # Clear error state when VM just started
  753. self.clear_error(vm.qid)
  754. elif prev_running and not vm.last_running:
  755. self.running_vms_count -= 1
  756. some_vms_have_changed_power_state = True
  757. if self.screen_changed == True:
  758. reload_table = True
  759. self.screen_changed = False
  760. if reload_table:
  761. self.fill_table()
  762. update_devs=True
  763. if (not self.show_inactive_vms) and some_vms_have_changed_power_state:
  764. self.showhide_inactive_vms(True)
  765. self.showhide_inactive_vms(False)
  766. self.set_table_geom_size()
  767. if self.sort_by_state != None and some_vms_have_changed_power_state:
  768. self.table.sortItems(self.columns_indices["State"], self.sort_by_state)
  769. blk_visible = None
  770. rows_with_blk = None
  771. if update_devs == True:
  772. rows_with_blk = []
  773. self.blk_manager.blk_lock.acquire()
  774. for d in self.blk_manager.attached_devs:
  775. rows_with_blk.append( self.blk_manager.attached_devs[d]['attached_to']['vm'])
  776. self.blk_manager.blk_lock.release()
  777. if (not self.table.isColumnHidden(self.columns_indices['Size']) ) and self.counter % 60 == 0 or out_of_schedule:
  778. self.update_size_on_disk = True
  779. if self.counter % 3 == 0 or out_of_schedule:
  780. (self.last_measure_time, self.last_measure_results) = \
  781. qubes_host.measure_cpu_usage(self.last_measure_results,
  782. self.last_measure_time)
  783. for vm_row in self.vms_in_table.values():
  784. cur_cpu_load = None
  785. if vm_row.vm.get_xid() in self.last_measure_results:
  786. cur_cpu_load = self.last_measure_results[vm_row.vm.xid]['cpu_usage']
  787. else:
  788. cur_cpu_load = 0
  789. if rows_with_blk != None:
  790. if vm_row.vm.name in rows_with_blk:
  791. blk_visible = True
  792. else:
  793. blk_visible = False
  794. vm_row.update(blk_visible=blk_visible, cpu_load = cur_cpu_load, update_size_on_disk = self.update_size_on_disk)
  795. else:
  796. for vm_row in self.vms_in_table.values():
  797. if rows_with_blk != None:
  798. if vm_row.vm.name in rows_with_blk:
  799. blk_visible = True
  800. else:
  801. blk_visible = False
  802. vm_row.update(blk_visible=blk_visible, update_size_on_disk = self.update_size_on_disk)
  803. if self.sort_by_cpu != None:
  804. self.table.sortItems(self.columns_indices["CPU"], self.sort_by_cpu)
  805. elif self.sort_by_mem != None:
  806. self.table.sortItems(self.columns_indices["MEM"], self.sort_by_mem)
  807. elif self.sort_by_state != None and reload_table:
  808. #needed to sort after reload (fill_table sorts items with setSortingEnabled, but by that time the widgets values are not correct yet).
  809. self.table.sortItems(self.columns_indices["State"], self.sort_by_state)
  810. elif self.sort_by_size_on_disk != None and self.update_size_on_disk == True:
  811. self.table.sortItems(self.columns_indices["Size"], self.sort_by_size_on_disk)
  812. self.table_selection_changed()
  813. self.update_size_on_disk = False
  814. if not out_of_schedule:
  815. self.counter += 1
  816. QTimer.singleShot (self.update_interval, self.update_table)
  817. def update_block_devices(self):
  818. res, msg = self.blk_manager.check_for_updates()
  819. if msg != None and len(msg) > 0:
  820. str = "\n".join(msg)
  821. trayIcon.showMessage (str, msecs=5000)
  822. return res
  823. def sortIndicatorChanged(self, column, order):
  824. if column == self.columns_indices["CPU"] or column == self.columns_indices["CPU Graph"]:
  825. self.sort_by_mem = None
  826. self.sort_by_state = None
  827. self.sort_by_size_on_disk = None
  828. self.sort_by_cpu = order
  829. return
  830. elif column == self.columns_indices["MEM"] or column == self.columns_indices["MEM Graph"]:
  831. self.sort_by_cpu = None
  832. self.sort_by_state = None
  833. self.sort_by_size_on_disk = None
  834. self.sort_by_mem = order
  835. return
  836. elif column == self.columns_indices["State"]:
  837. self.sort_by_cpu = None
  838. self.sort_by_mem = None
  839. self.sort_by_size_on_disk = None
  840. self.sort_by_state = order
  841. return
  842. elif column == self.columns_indices["Size"]:
  843. self.sort_by_cpu = None
  844. self.sort_by_mem = None
  845. self.sort_by_state = None
  846. self.sort_by_size_on_disk = order
  847. return
  848. else:
  849. self.sort_by_cpu = None
  850. self.sort_by_mem = None
  851. self.sort_by_state = None
  852. def table_selection_changed (self):
  853. vm = self.get_selected_vm()
  854. if vm != None:
  855. # Update available actions:
  856. self.action_settings.setEnabled(vm.qid != 0)
  857. self.action_removevm.setEnabled(not vm.installed_by_rpm and not (vm.last_running))
  858. self.action_clonevm.setEnabled(not vm.installed_by_rpm and not (vm.last_running) and not vm.is_netvm())
  859. self.action_resumevm.setEnabled(not vm.last_running)
  860. self.action_pausevm.setEnabled(vm.last_running and vm.qid != 0)
  861. self.action_shutdownvm.setEnabled(vm.last_running and vm.qid != 0)
  862. self.action_killvm.setEnabled(vm.last_running and vm.qid != 0)
  863. self.action_appmenus.setEnabled(not vm.is_netvm())
  864. self.action_editfwrules.setEnabled(vm.is_networked() and not (vm.is_netvm() and not vm.is_proxyvm()))
  865. self.action_updatevm.setEnabled(vm.is_updateable() or vm.qid == 0)
  866. self.action_run_command_in_vm.setEnabled(vm.qid != 0)
  867. self.action_set_keyboard_layout.setEnabled(vm.qid != 0 and vm.last_running)
  868. else:
  869. self.action_settings.setEnabled(False)
  870. self.action_removevm.setEnabled(False)
  871. self.action_clonevm.setEnabled(False)
  872. self.action_resumevm.setEnabled(False)
  873. self.action_pausevm.setEnabled(False)
  874. self.action_shutdownvm.setEnabled(False)
  875. self.action_killvm.setEnabled(False)
  876. self.action_appmenus.setEnabled(False)
  877. self.action_editfwrules.setEnabled(False)
  878. self.action_updatevm.setEnabled(False)
  879. self.action_run_command_in_vm.setEnabled(False)
  880. self.action_set_keyboard_layout.setEnabled(False)
  881. def closeEvent (self, event):
  882. if event.spontaneous(): # There is something borked in Qt, as the logic here is inverted on X11
  883. self.hide()
  884. event.ignore()
  885. def set_error(self, qid, message):
  886. for vm in self.vms_list:
  887. if vm.qid == qid:
  888. vm.error_msg = message
  889. # Store error in separate dict to make it immune to VM list reload
  890. self.vm_errors[qid] = message
  891. def clear_error(self, qid):
  892. self.vm_errors.pop(qid, None)
  893. for vm in self.vms_list:
  894. if vm.qid == qid:
  895. vm.error_msg = None
  896. def clear_error_exact(self, qid, message):
  897. for vm in self.vms_list:
  898. if vm.qid == qid:
  899. if vm.error_msg == message:
  900. vm.error_msg = None
  901. self.vm_errors.pop(qid, None)
  902. @pyqtSlot(name='on_action_createvm_triggered')
  903. def action_createvm_triggered(self):
  904. dialog = NewVmDlg(app, self.qvm_collection, trayIcon)
  905. dialog.exec_()
  906. def get_selected_vm(self):
  907. #vm selection relies on the VmInfo widget's value used for sorting by VM name
  908. row_index = self.table.currentRow()
  909. if row_index != -1:
  910. qid = self.table.item(row_index, self.columns_indices["Name"]).qid
  911. assert self.vms_in_table[qid] is not None
  912. vm = self.vms_in_table[qid].vm
  913. return vm
  914. else:
  915. return None
  916. @pyqtSlot(name='on_action_removevm_triggered')
  917. def action_removevm_triggered(self):
  918. vm = self.get_selected_vm()
  919. assert not vm.is_running()
  920. assert not vm.installed_by_rpm
  921. self.qvm_collection.lock_db_for_reading()
  922. self.qvm_collection.load()
  923. self.qvm_collection.unlock_db()
  924. vm = self.qvm_collection[vm.qid]
  925. if vm.is_template():
  926. dependent_vms = self.qvm_collection.get_vms_based_on(vm.qid)
  927. if len(dependent_vms) > 0:
  928. QMessageBox.warning (None, "Warning!",
  929. "This Template VM cannot be removed, because there is at least one AppVM that is based on it.<br>"
  930. "<small>If you want to remove this Template VM and all the AppVMs based on it,"
  931. "you should first remove each individual AppVM that uses this template.</small>")
  932. return
  933. reply = QMessageBox.question(None, "VM Removal Confirmation",
  934. "Are you sure you want to remove the VM <b>'{0}'</b>?<br>"
  935. "<small>All data on this VM's private storage will be lost!</small>".format(vm.name),
  936. QMessageBox.Yes | QMessageBox.Cancel)
  937. if reply == QMessageBox.Yes:
  938. thread_monitor = ThreadMonitor()
  939. thread = threading.Thread (target=self.do_remove_vm, args=(vm, thread_monitor))
  940. thread.daemon = True
  941. thread.start()
  942. progress = QProgressDialog ("Removing VM: <b>{0}</b>...".format(vm.name), "", 0, 0)
  943. progress.setCancelButton(None)
  944. progress.setModal(True)
  945. progress.show()
  946. while not thread_monitor.is_finished():
  947. app.processEvents()
  948. time.sleep (0.1)
  949. progress.hide()
  950. if thread_monitor.success:
  951. trayIcon.showMessage ("VM '{0}' has been removed.".format(vm.name), msecs=3000)
  952. else:
  953. QMessageBox.warning (None, "Error removing VM!", "ERROR: {0}".format(thread_monitor.error_msg))
  954. def do_remove_vm (self, vm, thread_monitor):
  955. try:
  956. self.qvm_collection.lock_db_for_writing()
  957. self.qvm_collection.load()
  958. vm = self.qvm_collection[vm.qid]
  959. #TODO: the following two conditions should really be checked by qvm_collection.pop() overload...
  960. if vm.is_template() and self.qvm_collection.default_template_qid == vm.qid:
  961. self.qvm_collection.default_template_qid = None
  962. if vm.is_netvm() and self.qvm_collection.default_netvm_qid == vm.qid:
  963. self.qvm_collection.default_netvm_qid = None
  964. vm.remove_from_disk()
  965. self.qvm_collection.pop(vm.qid)
  966. self.qvm_collection.save()
  967. except Exception as ex:
  968. thread_monitor.set_error_msg (str(ex))
  969. finally:
  970. self.qvm_collection.unlock_db()
  971. thread_monitor.set_finished()
  972. @pyqtSlot(name='on_action_clonevm_triggered')
  973. def action_clonevm_triggered(self):
  974. vm = self.get_selected_vm()
  975. name_number = 1
  976. name_format = vm.name + '-clone-%d'
  977. while self.qvm_collection.get_vm_by_name(name_format % name_number):
  978. name_number += 1
  979. cmd = ['kdialog', '--title', 'Qubes clone VM', '--inputbox', 'Enter name for VM <b>'+vm.name+'</b> clone:', name_format % name_number]
  980. kdialog = subprocess.Popen(cmd, stdout = subprocess.PIPE)
  981. clone_name = kdialog.stdout.read().strip()
  982. if clone_name == "":
  983. return
  984. thread_monitor = ThreadMonitor()
  985. thread = threading.Thread (target=self.do_clone_vm, args=(vm, clone_name, thread_monitor))
  986. thread.daemon = True
  987. thread.start()
  988. progress = QProgressDialog ("Cloning VM <b>{0}</b> to <b>{1}</b>...".format(vm.name, clone_name), "", 0, 0)
  989. progress.setCancelButton(None)
  990. progress.setModal(True)
  991. progress.show()
  992. while not thread_monitor.is_finished():
  993. app.processEvents()
  994. time.sleep (0.2)
  995. progress.hide()
  996. if not thread_monitor.success:
  997. QMessageBox.warning (None, "Error while cloning VM", "Exception while cloning:<br>{0}".format(thread_monitor.error_msg))
  998. def do_clone_vm(self, vm, dst_name, thread_monitor):
  999. try:
  1000. self.qvm_collection.lock_db_for_writing()
  1001. self.qvm_collection.load()
  1002. src_vm = self.qvm_collection[vm.qid]
  1003. dst_vm = None
  1004. if isinstance(src_vm, qubes.QubesTemplateVm):
  1005. dst_vm = self.qvm_collection.add_new_templatevm(name=dst_name,
  1006. installed_by_rpm=False)
  1007. elif isinstance(src_vm, qubes.QubesAppVm):
  1008. dst_vm = self.qvm_collection.add_new_appvm(name=dst_name, template=src_vm.template,
  1009. label=src_vm.label)
  1010. elif hasattr(qubes, 'QubesHVm') and isinstance(src_vm, qubes.QubesHVm):
  1011. dst_vm = self.qvm_collection.add_new_hvm(name=dst_name, label=src_vm.label)
  1012. dst_vm.clone_attrs(src_vm)
  1013. dst_vm.clone_disk_files (src_vm=src_vm, verbose=False)
  1014. self.qvm_collection.save()
  1015. self.qvm_collection.unlock_db()
  1016. except Exception as ex:
  1017. if dst_vm:
  1018. self.qvm_collection.pop(dst_vm.qid)
  1019. self.qvm_collection.unlock_db()
  1020. thread_monitor.set_error_msg(str(ex))
  1021. thread_monitor.set_finished()
  1022. @pyqtSlot(name='on_action_resumevm_triggered')
  1023. def action_resumevm_triggered(self):
  1024. vm = self.get_selected_vm()
  1025. assert not vm.is_running()
  1026. if vm.is_paused():
  1027. try:
  1028. subprocess.check_call (["/usr/sbin/xl", "unpause", vm.name])
  1029. except Exception as ex:
  1030. QMessageBox.warning (None, "Error unpausing VM!", "ERROR: {0}".format(ex))
  1031. return
  1032. thread_monitor = ThreadMonitor()
  1033. thread = threading.Thread (target=self.do_start_vm, args=(vm, thread_monitor))
  1034. thread.daemon = True
  1035. thread.start()
  1036. trayIcon.showMessage ("Starting '{0}'...".format(vm.name), msecs=3000)
  1037. while not thread_monitor.is_finished():
  1038. app.processEvents()
  1039. time.sleep (0.1)
  1040. if thread_monitor.success:
  1041. trayIcon.showMessage ("VM '{0}' has been started.".format(vm.name), msecs=3000)
  1042. else:
  1043. trayIcon.showMessage ("Error starting VM <b>'{0}'</b>: {1}".format(vm.name, thread_monitor.error_msg ), msecs=3000)
  1044. self.set_error(vm.qid, "Error starting VM: %s" % thread_monitor.error_msg)
  1045. def do_start_vm(self, vm, thread_monitor):
  1046. try:
  1047. vm.verify_files()
  1048. xid = vm.start()
  1049. except Exception as ex:
  1050. thread_monitor.set_error_msg(str(ex))
  1051. thread_monitor.set_finished()
  1052. return
  1053. thread_monitor.set_finished()
  1054. @pyqtSlot(name='on_action_pausevm_triggered')
  1055. def action_pausevm_triggered(self):
  1056. vm = self.get_selected_vm()
  1057. assert vm.is_running()
  1058. try:
  1059. subprocess.check_call (["/usr/sbin/xl", "pause", vm.name])
  1060. except Exception as ex:
  1061. QMessageBox.warning (None, "Error pausing VM!", "ERROR: {0}".format(ex))
  1062. return
  1063. @pyqtSlot(name='on_action_shutdownvm_triggered')
  1064. def action_shutdownvm_triggered(self):
  1065. vm = self.get_selected_vm()
  1066. assert vm.is_running()
  1067. self.blk_manager.check_if_serves_as_backend(vm)
  1068. reply = QMessageBox.question(None, "VM Shutdown Confirmation",
  1069. "Are you sure you want to power down the VM <b>'{0}'</b>?<br>"
  1070. "<small>This will shutdown all the running applications within this VM.</small>".format(vm.name),
  1071. QMessageBox.Yes | QMessageBox.Cancel)
  1072. app.processEvents()
  1073. if reply == QMessageBox.Yes:
  1074. self.shutdown_vm(vm)
  1075. def shutdown_vm(self, vm, shutdown_time = vm_shutdown_timeout):
  1076. try:
  1077. subprocess.check_call (["/usr/sbin/xl", "shutdown", vm.name])
  1078. except Exception as ex:
  1079. QMessageBox.warning (None, "Error shutting down VM!", "ERROR: {0}".format(ex))
  1080. return
  1081. trayIcon.showMessage ("VM '{0}' is shutting down...".format(vm.name), msecs=3000)
  1082. self.shutdown_monitor[vm.qid] = VmShutdownMonitor (vm, shutdown_time)
  1083. QTimer.singleShot (shutdown_time, self.shutdown_monitor[vm.qid].check_if_vm_has_shutdown)
  1084. @pyqtSlot(name='on_action_killvm_triggered')
  1085. def action_killvm_triggered(self):
  1086. vm = self.get_selected_vm()
  1087. assert vm.is_running()
  1088. reply = QMessageBox.question(None, "VM Kill Confirmation",
  1089. "Are you sure you want to kill the VM <b>'{0}'</b>?<br>"
  1090. "<small>This will end <b>(not shutdown!)</b> all the running applications within this VM.</small>".format(vm.name),
  1091. QMessageBox.Yes | QMessageBox.Cancel, QMessageBox.Cancel)
  1092. app.processEvents()
  1093. if reply == QMessageBox.Yes:
  1094. try:
  1095. vm.force_shutdown()
  1096. except Exception as ex:
  1097. QMessageBox.critical (None, "Error while killing VM!", "<b>An exception ocurred while killing {0}.</b><br>ERROR: {1}".format(vm.name, ex))
  1098. return
  1099. trayIcon.showMessage ("VM '{0}' killed!".format(vm.name), msecs=3000)
  1100. @pyqtSlot(name='on_action_settings_triggered')
  1101. def action_settings_triggered(self):
  1102. vm = self.get_selected_vm()
  1103. settings_window = VMSettingsWindow(vm, app, self.qvm_collection, "basic")
  1104. settings_window.exec_()
  1105. @pyqtSlot(name='on_action_appmenus_triggered')
  1106. def action_appmenus_triggered(self):
  1107. vm = self.get_selected_vm()
  1108. settings_window = VMSettingsWindow(vm, app, self.qvm_collection, "applications")
  1109. settings_window.exec_()
  1110. @pyqtSlot(name='on_action_updatevm_triggered')
  1111. def action_updatevm_triggered(self):
  1112. vm = self.get_selected_vm()
  1113. if not vm.is_running():
  1114. reply = QMessageBox.question(None, "VM Update Confirmation",
  1115. "<b>{0}</b><br>The VM has to be running to be updated.<br>Do you want to start it?<br>".format(vm.name),
  1116. QMessageBox.Yes | QMessageBox.Cancel)
  1117. if reply != QMessageBox.Yes:
  1118. return
  1119. trayIcon.showMessage ("Starting '{0}'...".format(vm.name), msecs=3000)
  1120. app.processEvents()
  1121. thread_monitor = ThreadMonitor()
  1122. thread = threading.Thread (target=self.do_update_vm, args=(vm, thread_monitor))
  1123. thread.daemon = True
  1124. thread.start()
  1125. progress = QProgressDialog ("<b>{0}</b><br>Please wait for the updater to launch...".format(vm.name), "", 0, 0)
  1126. progress.setCancelButton(None)
  1127. progress.setModal(True)
  1128. progress.show()
  1129. while not thread_monitor.is_finished():
  1130. app.processEvents()
  1131. time.sleep (0.2)
  1132. progress.hide()
  1133. if vm.qid != 0:
  1134. if not thread_monitor.success:
  1135. QMessageBox.warning (None, "Error VM update!", "ERROR: {0}".format(thread_monitor.error_msg))
  1136. def do_update_vm(self, vm, thread_monitor):
  1137. try:
  1138. if vm.qid == 0:
  1139. subprocess.check_call (["/usr/bin/qubes-dom0-update", "--clean", "--gui"])
  1140. else:
  1141. vm.run("user:gpk-update-viewer;sudo service qubes-update-check start", verbose=False, autostart=True,
  1142. notify_function=lambda lvl, msg: trayIcon.showMessage(msg, msecs=3000) )
  1143. except Exception as ex:
  1144. thread_monitor.set_error_msg(str(ex))
  1145. thread_monitor.set_finished()
  1146. return
  1147. thread_monitor.set_finished()
  1148. @pyqtSlot(name='on_action_run_command_in_vm_triggered')
  1149. def action_run_command_in_vm_triggered(self):
  1150. vm = self.get_selected_vm()
  1151. thread_monitor = ThreadMonitor()
  1152. thread = threading.Thread (target=self.do_run_command_in_vm, args=(vm, thread_monitor))
  1153. thread.daemon = True
  1154. thread.start()
  1155. while not thread_monitor.is_finished():
  1156. app.processEvents()
  1157. time.sleep (0.2)
  1158. if not thread_monitor.success:
  1159. QMessageBox.warning (None, "Error while running command", "Exception while running command:<br>{0}".format(thread_monitor.error_msg))
  1160. def do_run_command_in_vm(self, vm, thread_monitor):
  1161. cmd = ['kdialog', '--title', 'Qubes command entry', '--inputbox', 'Run command in <b>'+vm.name+'</b>:']
  1162. kdialog = subprocess.Popen(cmd, stdout = subprocess.PIPE)
  1163. command_to_run = kdialog.stdout.read()
  1164. command_to_run = command_to_run.strip()
  1165. if command_to_run != "":
  1166. try:
  1167. vm.run("user:" + command_to_run, verbose=False, autostart=True,
  1168. notify_function=lambda lvl, msg: trayIcon.showMessage(msg, msecs=3000) )
  1169. except Exception as ex:
  1170. thread_monitor.set_error_msg(str(ex))
  1171. thread_monitor.set_finished()
  1172. @pyqtSlot(name='on_action_set_keyboard_layout_triggered')
  1173. def action_set_keyboard_layout_triggered(self):
  1174. vm = self.get_selected_vm()
  1175. vm.run('user:qubes-change-keyboard-layout', verbose = False)
  1176. @pyqtSlot(name='on_action_showallvms_triggered')
  1177. def action_showallvms_triggered(self):
  1178. self.show_inactive_vms = self.action_showallvms.isChecked()
  1179. self.showhide_inactive_vms(self.show_inactive_vms)
  1180. self.set_table_geom_size()
  1181. @pyqtSlot(name='on_action_editfwrules_triggered')
  1182. def action_editfwrules_triggered(self):
  1183. vm = self.get_selected_vm()
  1184. settings_window = VMSettingsWindow(vm, app, self.qvm_collection, "firewall")
  1185. settings_window.exec_()
  1186. @pyqtSlot(name='on_action_global_settings_triggered')
  1187. def action_global_settings_triggered(self):
  1188. global_settings_window = GlobalSettingsWindow(app, self.qvm_collection)
  1189. global_settings_window.exec_()
  1190. @pyqtSlot(name='on_action_restore_triggered')
  1191. def action_restore_triggered(self):
  1192. restore_window = RestoreVMsWindow(app, self.qvm_collection, self.blk_manager)
  1193. restore_window.exec_()
  1194. @pyqtSlot(name='on_action_backup_triggered')
  1195. def action_backup_triggered(self):
  1196. backup_window = BackupVMsWindow(app, self.qvm_collection, self.blk_manager, self.shutdown_vm)
  1197. backup_window.exec_()
  1198. def showhide_menubar(self, checked):
  1199. self.menuWidget().setVisible(checked)
  1200. self.set_table_geom_size()
  1201. if not checked:
  1202. self.context_menu.addAction(self.action_menubar)
  1203. else:
  1204. self.context_menu.removeAction(self.action_menubar)
  1205. def showhide_toolbar(self, checked):
  1206. self.toolbar.setVisible(checked)
  1207. self.set_table_geom_size()
  1208. if not checked:
  1209. self.context_menu.addAction(self.action_toolbar)
  1210. else:
  1211. self.context_menu.removeAction(self.action_toolbar)
  1212. def showhide_column(self, col_num, show):
  1213. self.table.setColumnHidden( col_num, not show)
  1214. self.set_table_geom_size()
  1215. val = 1 if show else -1
  1216. self.visible_columns_count += val
  1217. if self.visible_columns_count == 1:
  1218. #disable hiding the last one
  1219. for c in self.columns_actions:
  1220. if self.columns_actions[c].isChecked():
  1221. self.columns_actions[c].setEnabled(False)
  1222. break
  1223. elif self.visible_columns_count == 2 and val == 1:
  1224. #enable hiding previously disabled column
  1225. for c in self.columns_actions:
  1226. if not self.columns_actions[c].isEnabled():
  1227. self.columns_actions[c].setEnabled(True)
  1228. break;
  1229. def on_action_vm_type_toggled(self, checked):
  1230. self.showhide_column( self.columns_indices['Type'], checked)
  1231. def on_action_label_toggled(self, checked):
  1232. self.showhide_column( self.columns_indices['Label'], checked)
  1233. def on_action_name_toggled(self, checked):
  1234. self.showhide_column( self.columns_indices['Name'], checked)
  1235. def on_action_state_toggled(self, checked):
  1236. self.showhide_column( self.columns_indices['State'], checked)
  1237. def on_action_template_toggled(self, checked):
  1238. self.showhide_column( self.columns_indices['Template'], checked)
  1239. def on_action_netvm_toggled(self, checked):
  1240. self.showhide_column( self.columns_indices['NetVM'], checked)
  1241. def on_action_cpu_toggled(self, checked):
  1242. self.showhide_column( self.columns_indices['CPU'], checked)
  1243. def on_action_cpu_graph_toggled(self, checked):
  1244. self.showhide_column( self.columns_indices['CPU Graph'], checked)
  1245. def on_action_mem_toggled(self, checked):
  1246. self.showhide_column( self.columns_indices['MEM'], checked)
  1247. def on_action_mem_graph_toggled(self, checked):
  1248. self.showhide_column( self.columns_indices['MEM Graph'], checked)
  1249. def on_action_size_on_disk_toggled(self, checked):
  1250. self.showhide_column( self.columns_indices['Size'], checked)
  1251. @pyqtSlot(name='on_action_about_qubes_triggered')
  1252. def action_about_qubes_triggered(self):
  1253. QMessageBox.about(self, "About...", "<b>Qubes OS</b><br><br>Release 1.0")
  1254. def createPopupMenu(self):
  1255. menu = QMenu()
  1256. menu.addAction(self.action_toolbar)
  1257. menu.addAction(self.action_menubar)
  1258. return menu
  1259. @pyqtSlot('const QPoint&')
  1260. def open_context_menu(self, point):
  1261. vm = self.get_selected_vm()
  1262. running = vm.is_running()
  1263. #logs menu
  1264. self.logs_menu.clear()
  1265. if vm.qid == 0:
  1266. text = "/var/log/xen/console/hypervisor.log"
  1267. action = self.logs_menu.addAction(QIcon(":/log.png"), text)
  1268. action.setData(QVariant(text))
  1269. self.logs_menu.setEnabled(True)
  1270. else:
  1271. menu_empty = True
  1272. text = "/var/log/xen/console/guest-"+vm.name+".log"
  1273. if os.path.exists(text):
  1274. action = self.logs_menu.addAction(QIcon(":/log.png"), text)
  1275. action.setData(QVariant(text))
  1276. menu_empty = False
  1277. text = "/var/log/xen/console/guest-"+vm.name+"-dm.log"
  1278. if os.path.exists(text):
  1279. action = self.logs_menu.addAction(QIcon(":/log.png"), text)
  1280. action.setData(QVariant(text))
  1281. menu_empty = False
  1282. if running:
  1283. xid = vm.xid
  1284. if xid != None:
  1285. text = "/var/log/qubes/guid."+str(xid)+".log"
  1286. action = self.logs_menu.addAction(QIcon(":/log.png"), text)
  1287. action.setData(QVariant(text))
  1288. text = "/var/log/qubes/qrexec."+str(xid)+".log"
  1289. action = self.logs_menu.addAction(QIcon(":/log.png"), text)
  1290. action.setData(QVariant(text))
  1291. menu_empty = False
  1292. self.logs_menu.setEnabled(not menu_empty)
  1293. # blk menu
  1294. if not running:
  1295. self.blk_menu.setEnabled(False)
  1296. else:
  1297. self.blk_menu.clear()
  1298. self.blk_menu.setEnabled(True)
  1299. self.blk_manager.blk_lock.acquire()
  1300. if len(self.blk_manager.attached_devs) > 0 :
  1301. for d in self.blk_manager.attached_devs:
  1302. if self.blk_manager.attached_devs[d]['attached_to']['vm'] == vm.name:
  1303. text = "Detach " + d + " " + unicode(self.blk_manager.attached_devs[d]['size']) + " " + self.blk_manager.attached_devs[d]['desc']
  1304. action = self.blk_menu.addAction(QIcon(":/remove.png"), text)
  1305. action.setData(QVariant(d))
  1306. if len(self.blk_manager.free_devs) > 0:
  1307. for d in self.blk_manager.free_devs:
  1308. if d.startswith(vm.name):
  1309. continue
  1310. # skip partitions heuristic
  1311. if d[-1].isdigit() and d[0:-1] in self.blk_manager.current_blk:
  1312. continue
  1313. text = "Attach " + d + " " + unicode(self.blk_manager.free_devs[d]['size']) + " " + self.blk_manager.free_devs[d]['desc']
  1314. action = self.blk_menu.addAction(QIcon(":/add.png"), text)
  1315. action.setData(QVariant(d))
  1316. self.blk_manager.blk_lock.release()
  1317. if self.blk_menu.isEmpty():
  1318. self.blk_menu.setEnabled(False)
  1319. self.context_menu.exec_(self.table.mapToGlobal(point))
  1320. @pyqtSlot('QAction *')
  1321. def show_log(self, action):
  1322. log = str(action.data().toString())
  1323. log_dialog = LogDialog(app, log)
  1324. log_dialog.exec_()
  1325. @pyqtSlot('QAction *')
  1326. def attach_dettach_device_triggered(self, action):
  1327. dev = str(action.data().toString())
  1328. vm = self.get_selected_vm()
  1329. self.blk_manager.blk_lock.acquire()
  1330. if dev in self.blk_manager.attached_devs:
  1331. self.blk_manager.detach_device(vm, dev)
  1332. else:
  1333. self.blk_manager.attach_device(vm, dev)
  1334. self.blk_manager.blk_lock.release()
  1335. class QubesBlockDevicesManager():
  1336. def __init__(self, qvm_collection):
  1337. self.qvm_collection = qvm_collection
  1338. self.attached_devs = {}
  1339. self.free_devs = {}
  1340. self.current_blk = {}
  1341. self.current_attached = {}
  1342. self.devs_changed = False
  1343. self.last_update_time = time.time()
  1344. self.blk_state_changed = True
  1345. self.msg = []
  1346. self.check_counter = 0
  1347. self.blk_lock = threading.Lock()
  1348. self.update()
  1349. def block_devs_event(self, xid):
  1350. now = time.time()
  1351. #don't update more often than 1/10 s
  1352. if now - self.last_update_time >= 0.1:
  1353. self.last_update_time = now
  1354. self.blk_lock.acquire()
  1355. self.blk_state_changed = True
  1356. self.blk_lock.release()
  1357. def check_for_updates(self):
  1358. self.blk_lock.acquire()
  1359. ret = (self.blk_state_changed, self.msg)
  1360. if self.blk_state_changed == True:
  1361. self.check_counter += 1
  1362. self.update()
  1363. ret = (self.blk_state_changed, self.msg)
  1364. #let the update last for 3 manager-update cycles
  1365. if self.check_counter == 3:
  1366. self.check_counter = 0
  1367. self.blk_state_changed = False
  1368. self.msg = []
  1369. self.blk_lock.release()
  1370. return ret
  1371. def update(self):
  1372. blk = qubesutils.block_list()
  1373. for b in blk:
  1374. att = qubesutils.block_check_attached(None, blk[b]['device'], backend_xid = blk[b]['xid'])
  1375. if b in self.current_blk:
  1376. if blk[b] == self.current_blk[b]:
  1377. if self.current_attached[b] != att: #devices the same, sth with attaching changed
  1378. self.current_attached[b] = att
  1379. else: #device changed ?!
  1380. self.current_blk[b] = blk[b]
  1381. self.current_attached[b] = att
  1382. else: #new device
  1383. self.current_blk[b] = blk[b]
  1384. self.current_attached[b] = att
  1385. self.msg.append("Attached new device: {0}".format(blk[b]['device']))
  1386. to_delete = []
  1387. for b in self.current_blk: #remove devices that are not there anymore
  1388. if b not in blk:
  1389. to_delete.append(b)
  1390. self.msg.append("Detached device: {0}".format(self.current_blk[b]['device']))
  1391. for d in to_delete:
  1392. del self.current_blk[d]
  1393. del self.current_attached[d]
  1394. self.__update_blk_entries__()
  1395. def __update_blk_entries__(self):
  1396. self.free_devs.clear()
  1397. self.attached_devs.clear()
  1398. for b in self.current_attached:
  1399. if self.current_attached[b]:
  1400. self.attached_devs[b] = self.__make_entry__(b, self.current_blk[b], self.current_attached[b])
  1401. else:
  1402. self.free_devs[b] = self.__make_entry__(b, self.current_blk[b], None)
  1403. def __make_entry__(self, k, dev, att):
  1404. size_str = qubesutils.bytes_to_kmg(dev['size'])
  1405. entry = { 'dev': dev['device'],
  1406. 'backend_name': dev['vm'],
  1407. 'desc': dev['desc'],
  1408. 'size': size_str,
  1409. 'attached_to': att, }
  1410. return entry
  1411. def attach_device(self, vm, dev):
  1412. backend_vm_name = self.free_devs[dev]['backend_name']
  1413. dev_id = self.free_devs[dev]['dev']
  1414. backend_vm = self.qvm_collection.get_vm_by_name(backend_vm_name)
  1415. trayIcon.showMessage ("{0} - attaching {1}".format(vm.name, dev), msecs=3000)
  1416. qubesutils.block_attach(vm, backend_vm, dev_id)
  1417. def detach_device(self, vm, dev_name):
  1418. dev_id = self.attached_devs[dev_name]['attached_to']['devid']
  1419. vm_xid = self.attached_devs[dev_name]['attached_to']['xid']
  1420. trayIcon.showMessage ("{0} - detaching {1}".format(vm.name, dev_name), msecs=3000)
  1421. qubesutils.block_detach(None, dev_id, vm_xid)
  1422. def check_if_serves_as_backend(self, vm):
  1423. serves_for = []
  1424. for d in self.attached_devs:
  1425. if self.attached_devs[d]['backend_name'] == vm.name:
  1426. serves_for.append((self.attached_devs[d]['dev'], self.attached_devs[d]['attached_to']['vm']))
  1427. if len(serves_for) > 0:
  1428. msg = "VM <b>" + vm.name + "</b> attaches block devices to other VMs: "
  1429. msg += ', '.join(["<b>"+v+"</b>("+d+")" for (d,v) in serves_for ])
  1430. msg += ".<br><br> Shutting the VM down will dettach the devices from them."
  1431. QMessageBox.warning (None, "Warning!", msg)
  1432. class QubesTrayIcon(QSystemTrayIcon):
  1433. def __init__(self, icon):
  1434. QSystemTrayIcon.__init__(self, icon)
  1435. self.menu = QMenu()
  1436. action_showmanager = self.createAction ("Open VM Manager", slot=show_manager, icon="qubes")
  1437. action_backup = self.createAction ("Make backup")
  1438. action_preferences = self.createAction ("Preferences")
  1439. action_set_netvm = self.createAction ("Set default NetVM", icon="networking")
  1440. action_sys_info = self.createAction ("System Info", icon="dom0")
  1441. action_exit = self.createAction ("Exit", slot=exit_app)
  1442. action_backup.setDisabled(True)
  1443. action_preferences.setDisabled(True)
  1444. action_set_netvm.setDisabled(True)
  1445. action_sys_info.setDisabled(True)
  1446. self.addActions (self.menu, (action_showmanager, action_backup, action_sys_info, None, action_preferences, action_set_netvm, None, action_exit))
  1447. self.setContextMenu(self.menu)
  1448. self.connect (self, SIGNAL("activated (QSystemTrayIcon::ActivationReason)"), self.icon_clicked)
  1449. self.tray_object = dbus.SessionBus().get_object("org.freedesktop.Notifications", "/org/freedesktop/Notifications")
  1450. self.tray_notifier = dbus.Interface(self.tray_object, "org.freedesktop.Notifications" )
  1451. def icon_clicked(self, reason):
  1452. if reason == QSystemTrayIcon.Context:
  1453. # Handle the right click normally, i.e. display the context menu
  1454. return
  1455. else:
  1456. bring_manager_to_front()
  1457. def addActions(self, target, actions):
  1458. for action in actions:
  1459. if action is None:
  1460. target.addSeparator()
  1461. else:
  1462. target.addAction(action)
  1463. def showMessage(self, message, msecs):
  1464. self.tray_notifier.Notify("Qubes", 0, "/usr/share/qubes/icons/qubes.png", "Qubes VM Manager", message, [], [], msecs)
  1465. def createAction(self, text, slot=None, shortcut=None, icon=None,
  1466. tip=None, checkable=False, signal="triggered()"):
  1467. action = QAction(text, self)
  1468. if icon is not None:
  1469. action.setIcon(QIcon(":/%s.png" % icon))
  1470. if shortcut is not None:
  1471. action.setShortcut(shortcut)
  1472. if tip is not None:
  1473. action.setToolTip(tip)
  1474. action.setStatusTip(tip)
  1475. if slot is not None:
  1476. self.connect(action, SIGNAL(signal), slot)
  1477. if checkable:
  1478. action.setCheckable(True)
  1479. return action
  1480. class QubesDbusNotify(dbus.service.Object):
  1481. def __init__(self, bus, manager_window):
  1482. # export to system bus (instead of session) to make possible
  1483. # communication from outside of session (eg. from qmemman)
  1484. dbus.service.Object.__init__(self, bus, dbus_object_path)
  1485. self.manager_window = manager_window
  1486. @dbus.service.method(dbus_interface=dbus_interface,
  1487. in_signature='ss', out_signature='')
  1488. def notify_error(self, vmname, message):
  1489. vm = self.manager_window.qvm_collection.get_vm_by_name(vmname)
  1490. if vm:
  1491. self.manager_window.set_error(vm.qid, message)
  1492. else:
  1493. # ignore VM-not-found error
  1494. pass
  1495. @dbus.service.method(dbus_interface=dbus_interface,
  1496. in_signature='ss', out_signature='')
  1497. def clear_error_exact(self, vmname, message):
  1498. vm = self.manager_window.qvm_collection.get_vm_by_name(vmname)
  1499. if vm:
  1500. self.manager_window.clear_error_exact(vm.qid, message)
  1501. else:
  1502. # ignore VM-not-found error
  1503. pass
  1504. @dbus.service.method(dbus_interface=dbus_interface,
  1505. in_signature='s', out_signature='')
  1506. def clear_error(self, vmname):
  1507. vm = self.manager_window.qvm_collection.get_vm_by_name(vmname)
  1508. if vm:
  1509. self.manager_window.clear_error(vm.qid, message)
  1510. else:
  1511. # ignore VM-not-found error
  1512. pass
  1513. @dbus.service.method(dbus_interface=dbus_interface,
  1514. in_signature='', out_signature='')
  1515. def show_manager(self):
  1516. bring_manager_to_front()
  1517. def get_frame_size():
  1518. w = 0
  1519. h = 0
  1520. cmd = ['xprop', '-name', 'Qubes VM Manager', '|', 'grep', '_NET_FRAME_EXTENTS']
  1521. xprop = subprocess.Popen(cmd, stdout = subprocess.PIPE)
  1522. for l in xprop.stdout:
  1523. line = l.split('=')
  1524. if len(line) == 2:
  1525. line = line[1].strip().split(',')
  1526. if len(line) == 4:
  1527. w = int(line[0].strip())+ int(line[1].strip())
  1528. h = int(line[2].strip())+ int(line[3].strip())
  1529. break;
  1530. #in case of some weird window managers we have to assume sth...
  1531. if w<= 0:
  1532. w = 10
  1533. if h <= 0:
  1534. h = 30
  1535. manager_window.frame_width = w
  1536. manager_window.frame_height = h
  1537. return
  1538. def show_manager():
  1539. manager_window.show()
  1540. manager_window.set_table_geom_size()
  1541. manager_window.repaint()
  1542. app.processEvents()
  1543. get_frame_size()
  1544. #print manager_window.frame_width, " x ", manager_window.frame_height
  1545. manager_window.set_table_geom_size()
  1546. def bring_manager_to_front():
  1547. if manager_window.isVisible():
  1548. subprocess.check_call(['wmctrl', '-R', str(manager_window.windowTitle())])
  1549. else:
  1550. show_manager()
  1551. def show_running_manager_via_dbus():
  1552. global system_bus
  1553. if system_bus is None:
  1554. system_bus = dbus.SystemBus()
  1555. try:
  1556. qubes_manager = system_bus.get_object('org.qubesos.QubesManager',
  1557. '/org/qubesos/QubesManager')
  1558. qubes_manager.show_manager(dbus_interface='org.qubesos.QubesManager')
  1559. except dbus.DBusException:
  1560. # ignore the case when no qubes-manager is running
  1561. pass
  1562. def exit_app():
  1563. notifier.stop()
  1564. app.exit()
  1565. # Bases on the original code by:
  1566. # Copyright (c) 2002-2007 Pascal Varet <p.varet@gmail.com>
  1567. def handle_exception( exc_type, exc_value, exc_traceback ):
  1568. import sys
  1569. import os.path
  1570. import traceback
  1571. filename, line, dummy, dummy = traceback.extract_tb( exc_traceback ).pop()
  1572. filename = os.path.basename( filename )
  1573. error = "%s: %s" % ( exc_type.__name__, exc_value )
  1574. strace = ""
  1575. stacktrace = traceback.extract_tb( exc_traceback )
  1576. while len(stacktrace) > 0:
  1577. (filename, line, func, txt) = stacktrace.pop()
  1578. strace += "----\n"
  1579. strace += "line: %s\n" %txt
  1580. strace += "func: %s\n" %func
  1581. strace += "line no.: %d\n" %line
  1582. strace += "file: %s\n" %filename
  1583. msg_box = QMessageBox()
  1584. msg_box.setDetailedText(strace)
  1585. msg_box.setIcon(QMessageBox.Critical)
  1586. msg_box.setWindowTitle( "Houston, we have a problem...")
  1587. msg_box.setText("Whoops. A critical error has occured. This is most likely a bug "
  1588. "in Qubes Manager.<br><br>"
  1589. "<b><i>%s</i></b>" % error +
  1590. "<br/>at line <b>%d</b><br/>of file %s.<br/><br/>"
  1591. % ( line, filename ))
  1592. msg_box.exec_()
  1593. def sighup_handler(signum, frame):
  1594. os.execl("/usr/bin/qubes-manager")
  1595. def main():
  1596. # Avoid starting more than one instance of the app
  1597. lock = QubesDaemonPidfile ("qubes-manager")
  1598. if lock.pidfile_exists():
  1599. if lock.read_pid() == os.getpid():
  1600. pass
  1601. elif lock.pidfile_is_stale():
  1602. lock.remove_pidfile()
  1603. print "Removed stale pidfile (has the previous daemon instance crashed?)."
  1604. else:
  1605. show_running_manager_via_dbus()
  1606. exit (0)
  1607. lock.create_pidfile()
  1608. signal.signal(signal.SIGHUP, sighup_handler)
  1609. global qubes_host
  1610. qubes_host = QubesHost()
  1611. global app
  1612. app = QApplication(sys.argv)
  1613. app.setOrganizationName("The Qubes Project")
  1614. app.setOrganizationDomain("http://qubes-os.org")
  1615. app.setApplicationName("Qubes VM Manager")
  1616. app.setWindowIcon(QIcon(":/qubes.png"))
  1617. sys.excepthook = handle_exception
  1618. # setup dbus connection
  1619. DBusQtMainLoop(set_as_default=True)
  1620. global manager_window
  1621. manager_window = VmManagerWindow()
  1622. wm = WatchManager()
  1623. mask = EventsCodes.OP_FLAGS.get('IN_MODIFY')
  1624. global notifier
  1625. notifier = ThreadedNotifier(wm, QubesConfigFileWatcher(manager_window.mark_table_for_update))
  1626. notifier.start()
  1627. wdd = wm.add_watch(qubes_store_filename, mask)
  1628. global system_bus
  1629. system_bus = dbus.SystemBus()
  1630. name = dbus.service.BusName('org.qubesos.QubesManager', system_bus)
  1631. dbus_notifier = QubesDbusNotify(system_bus, manager_window)
  1632. global trayIcon
  1633. trayIcon = QubesTrayIcon(QIcon(":/qubes.png"))
  1634. trayIcon.show()
  1635. show_manager()
  1636. app.exec_()
  1637. trayIcon = None