bootfromdevice.py 6.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176
  1. #!/usr/bin/python3
  2. #
  3. # The Qubes OS Project, http://www.qubes-os.org
  4. #
  5. # This program is free software; you can redistribute it and/or
  6. # modify it under the terms of the GNU General Public License
  7. # as published by the Free Software Foundation; either version 2
  8. # of the License, or (at your option) any later version.
  9. #
  10. # This program is distributed in the hope that it will be useful,
  11. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13. # GNU General Public License for more details.
  14. #
  15. # You should have received a copy of the GNU Lesser General Public License along
  16. # with this program; if not, see <http://www.gnu.org/licenses/>.
  17. #
  18. #
  19. import functools
  20. import subprocess
  21. from . import utils
  22. from . import ui_bootfromdevice # pylint: disable=no-name-in-module
  23. from PyQt5 import QtWidgets, QtGui # pylint: disable=import-error
  24. from qubesadmin import tools
  25. from qubesadmin.tools import qvm_start
  26. from qubesadmin import exc
  27. class VMBootFromDeviceWindow(ui_bootfromdevice.Ui_BootDialog,
  28. QtWidgets.QDialog):
  29. def __init__(self, vm, qapp, qubesapp=None, parent=None):
  30. super().__init__(parent)
  31. self.vm = vm
  32. self.qapp = qapp
  33. self.qubesapp = qubesapp
  34. self.setupUi(self)
  35. self.setWindowTitle(
  36. self.tr("Boot {vm} from device").format(vm=self.vm.name))
  37. self.buttonBox.accepted.connect(self.save_and_apply)
  38. self.buttonBox.rejected.connect(self.reject)
  39. # populate buttons and such
  40. self.__init_buttons__()
  41. # warn user if the VM is currently running
  42. self.__warn_if_running__()
  43. def setup_application(self):
  44. self.qapp.setApplicationName(self.tr("Boot Qube From Device"))
  45. self.qapp.setWindowIcon(QtGui.QIcon.fromTheme("qubes-manager"))
  46. def reject(self):
  47. self.done(0)
  48. def save_and_apply(self):
  49. if self.blockDeviceRadioButton.isChecked():
  50. cdrom_location = self.blockDeviceComboBox.currentText()
  51. elif self.fileRadioButton.isChecked():
  52. cdrom_location = str(self.fileVM.currentData()) + \
  53. ":" + self.pathText.text()
  54. else:
  55. QtWidgets.QMessageBox.warning(
  56. self,
  57. self.tr("ERROR!"),
  58. self.tr("No file or block device selected; please select one."))
  59. return
  60. # warn user if the VM is currently running
  61. self.__warn_if_running__()
  62. qvm_start.main(['--cdrom', cdrom_location, self.vm.name])
  63. self.done(0)
  64. def __warn_if_running__(self):
  65. try:
  66. if self.vm.is_running():
  67. QtWidgets.QMessageBox.warning(
  68. self,
  69. self.tr("Warning!"),
  70. self.tr("Qube must be turned off before booting it from "
  71. "device. Please turn off the qube."))
  72. except exc.QubesDaemonAccessError:
  73. QtWidgets.QMessageBox.warning(
  74. self,
  75. self.tr("Warning!"),
  76. self.tr("Insufficient permissions to determine if qube is "
  77. "running. It must be turned off before booting it from "
  78. "device."))
  79. def __init_buttons__(self):
  80. self.fileVM.setEnabled(False)
  81. self.selectFileButton.setEnabled(False)
  82. self.blockDeviceComboBox.setEnabled(False)
  83. self.blockDeviceRadioButton.clicked.connect(self.radio_button_clicked)
  84. self.fileRadioButton.clicked.connect(self.radio_button_clicked)
  85. self.selectFileButton.clicked.connect(self.select_file_dialog)
  86. utils.initialize_widget_with_vms(
  87. widget=self.fileVM,
  88. qubes_app=self.qubesapp,
  89. allow_internal=True
  90. )
  91. device_choice = []
  92. for domain in self.vm.app.domains:
  93. try:
  94. for device in domain.devices["block"]:
  95. device_choice.append((str(device), device))
  96. except exc.QubesDaemonAccessError:
  97. # insufficient permissions
  98. pass
  99. if device_choice:
  100. utils.initialize_widget(
  101. widget=self.blockDeviceComboBox,
  102. choices=device_choice,
  103. selected_value=device_choice[0][1],
  104. add_current_label=False
  105. )
  106. else:
  107. self.blockDeviceRadioButton.setEnabled(False)
  108. self.blockDeviceComboBox.setEnabled(False)
  109. self.blockDeviceComboBox.addItem("no block devices found!")
  110. self.blockDeviceComboBox.setCurrentIndex(0)
  111. def radio_button_clicked(self):
  112. self.blockDeviceComboBox.setEnabled(
  113. self.blockDeviceRadioButton.isChecked())
  114. self.fileVM.setEnabled(self.fileRadioButton.isChecked())
  115. self.selectFileButton.setEnabled(self.fileRadioButton.isChecked())
  116. self.pathText.setEnabled(self.fileRadioButton.isChecked())
  117. def select_file_dialog(self):
  118. backend_vm = self.fileVM.currentData()
  119. error_occurred = False
  120. try:
  121. new_path = utils.get_path_from_vm(backend_vm, "qubes.SelectFile")
  122. except subprocess.CalledProcessError as ex:
  123. if ex.returncode != 1:
  124. # Error other than 'user did not select a file'
  125. error_occurred = True
  126. new_path = None
  127. except Exception: # pylint: disable=broad-except
  128. error_occurred = True
  129. new_path = None
  130. if error_occurred:
  131. QtWidgets.QMessageBox.warning(
  132. None,
  133. self.tr("Failed to display file selection dialog"),
  134. self.tr("Check if the qube {0} can be started and has a file"
  135. " manager installed.").format(backend_vm)
  136. )
  137. if new_path:
  138. self.pathText.setText(new_path)
  139. parser = tools.QubesArgumentParser(vmname_nargs=1)
  140. def main(args=None):
  141. args = parser.parse_args(args)
  142. vm = args.domains.pop()
  143. utils.run_synchronous(functools.partial(VMBootFromDeviceWindow, vm))
  144. if __name__ == "__main__":
  145. main()