utils.py 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164
  1. #
  2. # The Qubes OS Project, https://www.qubes-os.org
  3. #
  4. # Copyright (C) 2012 Agnieszka Kostrzewa <agnieszka.kostrzewa@gmail.com>
  5. # Copyright (C) 2012 Marek Marczykowski-Górecki
  6. # <marmarek@invisiblethingslab.com>
  7. # Copyright (C) 2017 Wojtek Porczyk <woju@invisiblethingslab.com>
  8. #
  9. # This program is free software: you can redistribute it and/or modify
  10. # it under the terms of the GNU General Public License as published by
  11. # the Free Software Foundation, either version 2 of the License, or
  12. # (at your option) any later version.
  13. #
  14. # This program is distributed in the hope that it will be useful,
  15. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  16. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  17. # GNU General Public License for more details.
  18. #
  19. # You should have received a copy of the GNU General Public License
  20. # along with this program. If not, see <http://www.gnu.org/licenses/>.
  21. #
  22. import os
  23. import re
  24. import qubesadmin
  25. from PyQt4.QtGui import QIcon # pylint: disable=import-error
  26. def _filter_internal(vm):
  27. return (not vm.klass == 'AdminVM'
  28. and not vm.features.get('internal', False))
  29. def prepare_choice(widget, holder, propname, choice, default,
  30. filter_function=None, *,
  31. icon_getter=None, allow_internal=None, allow_default=False,
  32. allow_none=False):
  33. # for newly created vms, set propname to None
  34. debug(
  35. 'prepare_choice(widget={widget!r}, '
  36. 'holder={holder!r}, '
  37. 'propname={propname!r}, '
  38. 'choice={choice!r}, '
  39. 'default={default!r}, '
  40. 'filter_function={filter_function!r}, '
  41. 'icon_getter={icon_getter!r}, '
  42. 'allow_internal={allow_internal!r}, '
  43. 'allow_default={allow_default!r}, '
  44. 'allow_none={allow_none!r})'.format(**locals()))
  45. if allow_internal is None:
  46. allow_internal = propname is None or not propname.endswith('vm')
  47. if propname is not None:
  48. oldvalue = getattr(holder, propname)
  49. is_default = holder.property_is_default(propname)
  50. else:
  51. oldvalue = object() # won't match for identity
  52. is_default = False
  53. idx = 0
  54. choice_list = list(choice)[:]
  55. if not allow_internal:
  56. choice_list = filter(_filter_internal, choice_list)
  57. if filter_function is not None:
  58. choice_list = filter(filter_function, choice_list)
  59. choice_list = list(choice_list)
  60. if allow_default:
  61. choice_list.insert(0, qubesadmin.DEFAULT)
  62. if allow_none:
  63. choice_list.append(None)
  64. for i, item in enumerate(choice_list):
  65. debug('i={} item={}'.format(i, item))
  66. # 0: default (unset)
  67. if item is qubesadmin.DEFAULT:
  68. text = 'default ({})'.format(
  69. str(default) if default is not None else 'none')
  70. # N+1: explicit None
  71. elif item is None:
  72. text = '(none)'
  73. # 1..N: choices
  74. else:
  75. text = str(item)
  76. if item is qubesadmin.DEFAULT and is_default \
  77. or item is not qubesadmin.DEFAULT and item == oldvalue:
  78. text += ' (current)'
  79. idx = i
  80. widget.insertItem(i, text)
  81. if icon_getter is not None:
  82. icon = icon_getter(item)
  83. if icon is not None:
  84. widget.setItemIcon(i, icon)
  85. widget.setCurrentIndex(idx)
  86. return choice_list, idx
  87. def prepare_kernel_choice(widget, holder, propname, default, *args, **kwargs):
  88. # TODO get from storage API (pool 'linux-kernel') (suggested by @marmarta)
  89. return prepare_choice(widget, holder, propname,
  90. os.listdir('/var/lib/qubes/vm-kernels'), default, *args, **kwargs)
  91. def prepare_label_choice(widget, holder, propname, default, *args, **kwargs):
  92. try:
  93. app = holder.app
  94. except AttributeError:
  95. app = holder
  96. return prepare_choice(widget, holder, propname,
  97. sorted(app.labels, key=lambda l: l.index),
  98. default, *args,
  99. icon_getter=(lambda label: QIcon.fromTheme(label.icon)),
  100. **kwargs)
  101. def prepare_vm_choice(widget, holder, propname, default, *args, **kwargs):
  102. try:
  103. app = holder.app
  104. except AttributeError:
  105. app = holder
  106. return prepare_choice(widget, holder, propname, app.domains, default,
  107. *args, **kwargs)
  108. def is_debug():
  109. return os.getenv('QUBES_MANAGER_DEBUG', '') not in ('', '0')
  110. def debug(*args, **kwargs):
  111. if not is_debug():
  112. return
  113. print(*args, **kwargs)
  114. def get_path_from_vm(vm, service_name):
  115. """
  116. Displays a file/directory selection window for the given VM.
  117. :param vm: vm from which to select path
  118. :param service_name: qubes.SelectFile or qubes.SelectDirectory
  119. :return: path to file, checked for validity
  120. """
  121. path_re = re.compile(r"[a-zA-Z0-9/:.,_+=() -]*")
  122. path_max_len = 512
  123. if not vm:
  124. return None
  125. stdout, _stderr = vm.run_service_for_stdio(service_name)
  126. untrusted_path = stdout.decode(encoding='ascii')[:path_max_len]
  127. if not untrusted_path:
  128. return None
  129. if path_re.match(untrusted_path):
  130. assert '../' not in untrusted_path
  131. assert '\0' not in untrusted_path
  132. return untrusted_path.strip()
  133. else:
  134. raise ValueError('Unexpected characters in path.')