bootfromdevice.py 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170
  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(VMBootFromDeviceWindow, self).__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. utils.initialize_widget(
  100. widget=self.blockDeviceComboBox,
  101. choices=device_choice,
  102. selected_value=device_choice[0][1],
  103. add_current_label=False
  104. )
  105. def radio_button_clicked(self):
  106. self.blockDeviceComboBox.setEnabled(
  107. self.blockDeviceRadioButton.isChecked())
  108. self.fileVM.setEnabled(self.fileRadioButton.isChecked())
  109. self.selectFileButton.setEnabled(self.fileRadioButton.isChecked())
  110. self.pathText.setEnabled(self.fileRadioButton.isChecked())
  111. def select_file_dialog(self):
  112. backend_vm = self.fileVM.currentData()
  113. error_occurred = False
  114. try:
  115. new_path = utils.get_path_from_vm(backend_vm, "qubes.SelectFile")
  116. except subprocess.CalledProcessError as ex:
  117. if ex.returncode != 1:
  118. # Error other than 'user did not select a file'
  119. error_occurred = True
  120. new_path = None
  121. except Exception: # pylint: disable=broad-except
  122. error_occurred = True
  123. new_path = None
  124. if error_occurred:
  125. QtWidgets.QMessageBox.warning(
  126. None,
  127. self.tr("Failed to display file selection dialog"),
  128. self.tr("Check if the qube {0} can be started and has a file"
  129. " manager installed.").format(backend_vm)
  130. )
  131. if new_path:
  132. self.pathText.setText(new_path)
  133. parser = tools.QubesArgumentParser(vmname_nargs=1)
  134. def main(args=None):
  135. args = parser.parse_args(args)
  136. vm = args.domains.pop()
  137. utils.run_synchronous(functools.partial(VMBootFromDeviceWindow, vm))
  138. if __name__ == "__main__":
  139. main()