Home Explore Help
Sign In
Qubes
/
core-admin
2
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Tree: 64755b2d98
Branches Tags
core-admin
/
tests
/
backupcompatibility.py

backupcompatibility.py 26 KB
History Raw

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286 
  1. #!/usr/bin/python
    2. 

  2. # -*- coding: utf-8 -*-
    3. 

  3. #
    4. 

  4. # The Qubes OS Project, http://www.qubes-os.org
    5. 

  5. #
    6. 

  6. # Copyright (C) 2014 Marek Marczykowski-Górecki <marmarek@invisiblethingslab.com>
    7. 

  7. #
    8. 

  8. # This program is free software; you can redistribute it and/or
    9. 

  9. # modify it under the terms of the GNU General Public License
    10. 

  10. # as published by the Free Software Foundation; either version 2
    11. 

  11. # of the License, or (at your option) any later version.
    12. 

  12. #
    13. 

  13. # This program is distributed in the hope that it will be useful,
    14. 

  14. # but WITHOUT ANY WARRANTY; without even the implied warranty of
    15. 

  15. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    16. 

  16. # GNU General Public License for more details.
    17. 

  17. #
    18. 

  18. # You should have received a copy of the GNU General Public License
    19. 

  19. # along with this program; if not, write to the Free Software
    20. 

  20. # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
    21. 

  21. #
    22. 

  22. #
    23. 

  23. from multiprocessing import Queue
    24. 

  24. 

  25. import os
    26. 

  26. import shutil
    27. 

  27. import subprocess
    28. 

  28. 

  29. import unittest
    30. 

  30. import sys
    31. 

  31. from qubes.qubes import QubesVmCollection, QubesException
    32. 

  32. from qubes import backup
    33. 

  33. 

  34. import qubes.tests
    35. 

  35. 

  36. QUBESXML_R2B2 = '''
    37. 

  37. <QubesVmCollection updatevm="3" default_kernel="3.7.6-2" default_netvm="3" default_fw_netvm="2" default_template="1" clockvm="2">
    38. 

  38.   <QubesTemplateVm installed_by_rpm="True" kernel="3.7.6-2" uses_default_kernelopts="True" qid="1" include_in_backups="True" uses_default_kernel="True" qrexec_timeout="60" internal="False" conf_file="fedora-18-x64.conf" label="black" template_qid="none" kernelopts="" memory="400" default_user="user" netvm_qid="3" uses_default_netvm="True" volatile_img="volatile.img" services="{'meminfo-writer': True}" maxmem="1535" pcidevs="[]" name="fedora-18-x64" private_img="private.img" vcpus="2" root_img="root.img" debug="False" dir_path="/var/lib/qubes/vm-templates/fedora-18-x64"/>
    39. 

  39.   <QubesNetVm installed_by_rpm="False" kernel="3.7.6-2" uses_default_kernelopts="True" qid="2" include_in_backups="True" uses_default_kernel="True" qrexec_timeout="60" internal="False" conf_file="netvm.conf" label="red" template_qid="1" kernelopts="iommu=soft swiotlb=4096" memory="200" default_user="user" volatile_img="volatile.img" services="{'ntpd': False, 'meminfo-writer': False}" maxmem="1535" pcidevs="['02:00.0', '03:00.0']" name="netvm" netid="1" private_img="private.img" vcpus="2" root_img="root.img" debug="False" dir_path="/var/lib/qubes/servicevms/netvm"/>
    40. 

  40.   <QubesProxyVm installed_by_rpm="False" kernel="3.7.6-2" uses_default_kernelopts="True" qid="3" include_in_backups="True" uses_default_kernel="True" qrexec_timeout="60" internal="False" conf_file="firewallvm.conf" label="green" template_qid="1" kernelopts="" memory="200" default_user="user" netvm_qid="2" volatile_img="volatile.img" services="{'meminfo-writer': True}" maxmem="1535" pcidevs="[]" name="firewallvm" netid="2" private_img="private.img" vcpus="2" root_img="root.img" debug="False" dir_path="/var/lib/qubes/servicevms/firewallvm"/>
    41. 

  41.   <QubesAppVm installed_by_rpm="False" kernel="3.7.6-2" uses_default_kernelopts="True" qid="4" include_in_backups="True" uses_default_kernel="True" qrexec_timeout="60" internal="True" conf_file="fedora-18-x64-dvm.conf" label="gray" template_qid="1" kernelopts="" memory="400" default_user="user" netvm_qid="3" uses_default_netvm="True" volatile_img="volatile.img" services="{'meminfo-writer': True}" maxmem="1535" pcidevs="[]" name="fedora-18-x64-dvm" private_img="private.img" vcpus="1" root_img="root.img" debug="False" dir_path="/var/lib/qubes/appvms/fedora-18-x64-dvm"/>
    42. 

  42.   <QubesAppVm installed_by_rpm="False" kernel="3.7.6-2" uses_default_kernelopts="True" qid="5" include_in_backups="True" uses_default_kernel="True" qrexec_timeout="60" internal="False" conf_file="test-work.conf" label="green" template_qid="1" kernelopts="" memory="400" default_user="user" netvm_qid="3" uses_default_netvm="True" volatile_img="volatile.img" services="{'meminfo-writer': True}" maxmem="1535" pcidevs="[]" name="test-work" private_img="private.img" vcpus="2" root_img="root.img" debug="False" dir_path="/var/lib/qubes/appvms/test-work"/>
    43. 

  43.   <QubesAppVm installed_by_rpm="False" kernel="3.7.6-2" uses_default_kernelopts="True" qid="6" include_in_backups="True" uses_default_kernel="True" qrexec_timeout="60" internal="False" conf_file="banking.conf" label="green" template_qid="1" kernelopts="" memory="400" default_user="user" netvm_qid="3" uses_default_netvm="True" volatile_img="volatile.img" services="{'meminfo-writer': True}" maxmem="1535" pcidevs="[]" name="banking" private_img="private.img" vcpus="2" root_img="root.img" debug="False" dir_path="/var/lib/qubes/appvms/banking"/>
    44. 

  44.   <QubesAppVm installed_by_rpm="False" kernel="3.7.6-2" uses_default_kernelopts="True" qid="7" include_in_backups="True" uses_default_kernel="True" qrexec_timeout="60" internal="False" conf_file="personal.conf" label="yellow" template_qid="1" kernelopts="" memory="400" default_user="user" netvm_qid="3" uses_default_netvm="True" volatile_img="volatile.img" services="{'meminfo-writer': True}" maxmem="1535" pcidevs="[]" name="personal" private_img="private.img" vcpus="2" root_img="root.img" debug="False" dir_path="/var/lib/qubes/appvms/personal"/>
    45. 

  45.   <QubesAppVm installed_by_rpm="False" kernel="3.7.6-2" uses_default_kernelopts="True" qid="8" include_in_backups="True" uses_default_kernel="True" qrexec_timeout="60" internal="False" conf_file="untrusted.conf" label="red" template_qid="1" kernelopts="" memory="400" default_user="user" netvm_qid="12" uses_default_netvm="False" volatile_img="volatile.img" services="{'meminfo-writer': True}" maxmem="1535" pcidevs="[]" name="untrusted" private_img="private.img" vcpus="2" root_img="root.img" debug="False" dir_path="/var/lib/qubes/appvms/untrusted"/>
    46. 

  46.   <QubesTemplateVm installed_by_rpm="False" kernel="3.7.6-2" uses_default_kernelopts="True" qid="9" include_in_backups="True" uses_default_kernel="True" qrexec_timeout="60" internal="False" conf_file="test-template-clone.conf" label="green" template_qid="none" kernelopts="" memory="400" default_user="user" netvm_qid="3" uses_default_netvm="True" volatile_img="volatile.img" services="{'meminfo-writer': True}" maxmem="1535" pcidevs="[]" name="test-template-clone" private_img="private.img" vcpus="2" root_img="root.img" debug="False" dir_path="/var/lib/qubes/vm-templates/test-template-clone"/>
    47. 

  47.   <QubesAppVm installed_by_rpm="False" kernel="3.7.6-2" uses_default_kernelopts="True" qid="10" include_in_backups="True" uses_default_kernel="True" qrexec_timeout="60" internal="False" conf_file="test-custom-template-appvm.conf" label="yellow" template_qid="9" kernelopts="" memory="400" default_user="user" netvm_qid="3" uses_default_netvm="True" volatile_img="volatile.img" services="{'meminfo-writer': True}" maxmem="1535" pcidevs="[]" name="test-custom-template-appvm" private_img="private.img" vcpus="2" root_img="root.img" debug="False" dir_path="/var/lib/qubes/appvms/test-custom-template-appvm"/>
    48. 

  48.   <QubesAppVm installed_by_rpm="False" kernel="3.7.6-2" uses_default_kernelopts="True" qid="11" include_in_backups="True" uses_default_kernel="True" qrexec_timeout="60" internal="False" conf_file="test-standalonevm.conf" label="blue" template_qid="none" kernelopts="" memory="400" default_user="user" netvm_qid="3" uses_default_netvm="True" volatile_img="volatile.img" services="{'meminfo-writer': True}" maxmem="1535" pcidevs="[]" name="test-standalonevm" private_img="private.img" vcpus="2" root_img="root.img" debug="False" dir_path="/var/lib/qubes/appvms/test-standalonevm"/>
    49. 

  49.   <QubesProxyVm installed_by_rpm="False" kernel="3.7.6-2" uses_default_kernelopts="True" qid="12" include_in_backups="True" uses_default_kernel="True" qrexec_timeout="60" internal="False" conf_file="test-testproxy.conf" label="red" template_qid="1" kernelopts="" memory="200" default_user="user" netvm_qid="3" volatile_img="volatile.img" services="{'meminfo-writer': True}" maxmem="1535" pcidevs="[]" name="test-testproxy" netid="3" private_img="private.img" vcpus="2" root_img="root.img" debug="False" dir_path="/var/lib/qubes/servicevms/test-testproxy"/>
    50. 

  50.   <QubesProxyVm installed_by_rpm="False" kernel="3.7.6-2" uses_default_kernelopts="True" qid="13" include_in_backups="True" uses_default_kernel="True" qrexec_timeout="60" internal="False" conf_file="testproxy2.conf" label="red" template_qid="9" kernelopts="" memory="200" default_user="user" netvm_qid="2" volatile_img="volatile.img" services="{'meminfo-writer': True}" maxmem="1535" pcidevs="[]" name="testproxy2" netid="4" private_img="private.img" vcpus="2" root_img="root.img" debug="False" dir_path="/var/lib/qubes/servicevms/testproxy2"/>
    51. 

  51.   <QubesHVm installed_by_rpm="False" netvm_qid="none" qid="14" include_in_backups="True" timezone="localtime" qrexec_timeout="60" conf_file="test-testhvm.conf" label="purple" template_qid="none" internal="False" memory="512" uses_default_netvm="True" services="{'meminfo-writer': False}" default_user="user" pcidevs="[]" name="test-testhvm" qrexec_installed="False" private_img="private.img" drive="None" vcpus="2" root_img="root.img" guiagent_installed="False" debug="False" dir_path="/var/lib/qubes/appvms/test-testhvm"/>
    52. 

  52. </QubesVmCollection>
    53. 

  53. '''
    54. 

  54. 

  55. APPTEMPLATE_R2B2 = '''
    56. 

  56. [Desktop Entry]
    57. 

  57. Name=%VMNAME%: {name}
    58. 

  58. GenericName=%VMNAME%: {name}
    59. 

  59. GenericName[ca]=%VMNAME%: Navegador web
    60. 

  60. GenericName[cs]=%VMNAME%: Webový prohlížeč
    61. 

  61. GenericName[es]=%VMNAME%: Navegador web
    62. 

  62. GenericName[fa]=%VMNAME%: مرورر اینترنتی
    63. 

  63. GenericName[fi]=%VMNAME%: WWW-selain
    64. 

  64. GenericName[fr]=%VMNAME%: Navigateur Web
    65. 

  65. GenericName[hu]=%VMNAME%: Webböngésző
    66. 

  66. GenericName[it]=%VMNAME%: Browser Web
    67. 

  67. GenericName[ja]=%VMNAME%: ウェブ・ブラウザ
    68. 

  68. GenericName[ko]=%VMNAME%: 웹 브라우저
    69. 

  69. GenericName[nb]=%VMNAME%: Nettleser
    70. 

  70. GenericName[nl]=%VMNAME%: Webbrowser
    71. 

  71. GenericName[nn]=%VMNAME%: Nettlesar
    72. 

  72. GenericName[no]=%VMNAME%: Nettleser
    73. 

  73. GenericName[pl]=%VMNAME%: Przeglądarka WWW
    74. 

  74. GenericName[pt]=%VMNAME%: Navegador Web
    75. 

  75. GenericName[pt_BR]=%VMNAME%: Navegador Web
    76. 

  76. GenericName[sk]=%VMNAME%: Internetový prehliadač
    77. 

  77. GenericName[sv]=%VMNAME%: Webbläsare
    78. 

  78. Comment={comment}
    79. 

  79. Comment[ca]=Navegueu per el web
    80. 

  80. Comment[cs]=Prohlížení stránek World Wide Webu
    81. 

  81. Comment[de]=Im Internet surfen
    82. 

  82. Comment[es]=Navegue por la web
    83. 

  83. Comment[fa]=صفحات شبه جهانی اینترنت را مرور نمایید
    84. 

  84. Comment[fi]=Selaa Internetin WWW-sivuja
    85. 

  85. Comment[fr]=Navigue sur Internet
    86. 

  86. Comment[hu]=A világháló böngészése
    87. 

  87. Comment[it]=Esplora il web
    88. 

  88. Comment[ja]=ウェブを閲覧します
    89. 

  89. Comment[ko]=웹을 돌아 다닙니다
    90. 

  90. Comment[nb]=Surf på nettet
    91. 

  91. Comment[nl]=Verken het internet
    92. 

  92. Comment[nn]=Surf på nettet
    93. 

  93. Comment[no]=Surf på nettet
    94. 

  94. Comment[pl]=Przeglądanie stron WWW
    95. 

  95. Comment[pt]=Navegue na Internet
    96. 

  96. Comment[pt_BR]=Navegue na Internet
    97. 

  97. Comment[sk]=Prehliadanie internetu
    98. 

  98. Comment[sv]=Surfa på webben
    99. 

  99. Exec=qvm-run -q --tray -a %VMNAME% '{command} %u'
    100. 

  100. Categories=Network;WebBrowser;
    101. 

  101. X-Qubes-VmName=%VMNAME%
    102. 

  102. Icon=%VMDIR%/icon.png
    103. 

  103. '''
    104. 

  104. 

  105. QUBESXML_R1 = '''<?xml version='1.0' encoding='UTF-8'?>
    106. 

  106. <QubesVmCollection clockvm="2" default_fw_netvm="2" default_kernel="3.2.7-10" default_netvm="3" default_template="1" updatevm="3"><QubesTemplateVm conf_file="fedora-17-x64.conf" debug="False" default_user="user" dir_path="/var/lib/qubes/vm-templates/fedora-17-x64" include_in_backups="True" installed_by_rpm="True" internal="False" kernel="3.2.7-10" kernelopts="" label="gray" maxmem="4063" memory="400" name="fedora-17-x64" netvm_qid="3" pcidevs="[]" private_img="private.img" qid="1" root_img="root.img" services="{&apos;meminfo-writer&apos;: True}" template_qid="none" uses_default_kernel="True" uses_default_kernelopts="True" uses_default_netvm="True" vcpus="2" volatile_img="volatile.img" /><QubesNetVm conf_file="netvm.conf" debug="False" default_user="user" dir_path="/var/lib/qubes/servicevms/netvm" include_in_backups="True" installed_by_rpm="False" internal="False" kernel="3.2.7-10" kernelopts="iommu=soft swiotlb=2048" label="red" maxmem="4063" memory="200" name="netvm" netid="1" pcidevs="[&apos;00:19.0&apos;, &apos;03:00.0&apos;]" private_img="private.img" qid="2" root_img="root.img" services="{&apos;ntpd&apos;: False, &apos;meminfo-writer&apos;: False}" template_qid="1" uses_default_kernel="True" uses_default_kernelopts="True" vcpus="2" volatile_img="volatile.img" /><QubesProxyVm conf_file="firewallvm.conf" debug="False" default_user="user" dir_path="/var/lib/qubes/servicevms/firewallvm" include_in_backups="True" installed_by_rpm="False" internal="False" kernel="3.2.7-10" kernelopts="" label="green" maxmem="4063" memory="200" name="firewallvm" netid="2" netvm_qid="2" pcidevs="[]" private_img="private.img" qid="3" root_img="root.img" services="{&apos;meminfo-writer&apos;: True}" template_qid="1" uses_default_kernel="True" uses_default_kernelopts="True" vcpus="2" volatile_img="volatile.img" /><QubesAppVm conf_file="fedora-17-x64-dvm.conf" debug="False" default_user="user" dir_path="/var/lib/qubes/appvms/fedora-17-x64-dvm" include_in_backups="True" installed_by_rpm="False" internal="True" kernel="3.2.7-10" kernelopts="" label="gray" maxmem="4063" memory="400" name="fedora-17-x64-dvm" netvm_qid="3" pcidevs="[]" private_img="private.img" qid="4" root_img="root.img" services="{&apos;meminfo-writer&apos;: True}" template_qid="1" uses_default_kernel="True" uses_default_kernelopts="True" uses_default_netvm="True" vcpus="1" volatile_img="volatile.img" /><QubesAppVm conf_file="test-work.conf" debug="False" default_user="user" dir_path="/var/lib/qubes/appvms/test-work" include_in_backups="True" installed_by_rpm="False" internal="False" kernel="3.2.7-10" kernelopts="" label="green" maxmem="4063" memory="400" name="test-work" netvm_qid="3" pcidevs="[]" private_img="private.img" qid="5" root_img="root.img" services="{&apos;meminfo-writer&apos;: True}" template_qid="1" uses_default_kernel="True" uses_default_kernelopts="True" uses_default_netvm="True" vcpus="2" volatile_img="volatile.img" /><QubesAppVm conf_file="personal.conf" debug="False" default_user="user" dir_path="/var/lib/qubes/appvms/personal" include_in_backups="True" installed_by_rpm="False" internal="False" kernel="3.2.7-10" kernelopts="" label="yellow" maxmem="4063" memory="400" name="personal" netvm_qid="3" pcidevs="[]" private_img="private.img" qid="6" root_img="root.img" services="{&apos;meminfo-writer&apos;: True}" template_qid="1" uses_default_kernel="True" uses_default_kernelopts="True" uses_default_netvm="True" vcpus="2" volatile_img="volatile.img" /><QubesAppVm conf_file="banking.conf" debug="False" default_user="user" dir_path="/var/lib/qubes/appvms/banking" include_in_backups="True" installed_by_rpm="False" internal="False" kernel="3.2.7-10" kernelopts="" label="green" maxmem="4063" memory="400" name="banking" netvm_qid="3" pcidevs="[]" private_img="private.img" qid="7" root_img="root.img" services="{&apos;meminfo-writer&apos;: True}" template_qid="1" uses_default_kernel="True" uses_default_kernelopts="True" uses_default_netvm="True" vcpus="2" volatile_img="volatile.img" /><QubesAppVm conf_file="untrusted.conf" debug="False" default_user="user" dir_path="/var/lib/qubes/appvms/untrusted" include_in_backups="True" installed_by_rpm="False" internal="False" kernel="3.2.7-10" kernelopts="" label="red" maxmem="4063" memory="400" name="untrusted" netvm_qid="3" pcidevs="[]" private_img="private.img" qid="8" root_img="root.img" services="{&apos;meminfo-writer&apos;: True}" template_qid="1" uses_default_kernel="True" uses_default_kernelopts="True" uses_default_netvm="True" vcpus="2" volatile_img="volatile.img" /><QubesAppVm conf_file="test-standalonevm.conf" debug="False" default_user="user" dir_path="/var/lib/qubes/appvms/test-standalonevm" include_in_backups="True" installed_by_rpm="False" internal="False" kernel="None" kernelopts="" label="red" maxmem="4063" memory="400" name="test-standalonevm" netvm_qid="3" pcidevs="[]" private_img="private.img" qid="9" root_img="root.img" services="{&apos;meminfo-writer&apos;: True}" template_qid="none" uses_default_kernel="False" uses_default_kernelopts="True" uses_default_netvm="True" vcpus="2" volatile_img="volatile.img" /><QubesAppVm conf_file="test-testvm.conf" debug="False" default_user="user" dir_path="/var/lib/qubes/appvms/test-testvm" include_in_backups="True" installed_by_rpm="False" internal="False" kernel="3.2.7-10" kernelopts="" label="red" mac="00:16:3E:5E:6C:55" maxmem="4063" memory="400" name="test-testvm" netvm_qid="3" pcidevs="[]" private_img="private.img" qid="10" root_img="root.img" services="{&apos;meminfo-writer&apos;: True}" template_qid="1" uses_default_kernel="True" uses_default_kernelopts="True" uses_default_netvm="True" vcpus="2" volatile_img="volatile.img" /><QubesTemplateVm conf_file="test-template-clone.conf" debug="False" default_user="user" dir_path="/var/lib/qubes/vm-templates/test-template-clone" include_in_backups="True" installed_by_rpm="False" internal="False" kernel="3.2.7-10" kernelopts="" label="gray" maxmem="4063" memory="400" name="test-template-clone" netvm_qid="3" pcidevs="[]" private_img="private.img" qid="11" root_img="root.img" services="{&apos;meminfo-writer&apos;: True}" template_qid="none" uses_default_kernel="True" uses_default_kernelopts="True" uses_default_netvm="True" vcpus="2" volatile_img="volatile.img" /><QubesAppVm conf_file="test-custom-template-appvm.conf" debug="False" default_user="user" dir_path="/var/lib/qubes/appvms/test-custom-template-appvm" include_in_backups="True" installed_by_rpm="False" internal="False" kernel="3.2.7-10" kernelopts="" label="yellow" maxmem="4063" memory="400" name="test-custom-template-appvm" netvm_qid="3" pcidevs="[]" private_img="private.img" qid="12" root_img="root.img" services="{&apos;meminfo-writer&apos;: True}" template_qid="11" uses_default_kernel="True" uses_default_kernelopts="True" uses_default_netvm="True" vcpus="2" volatile_img="volatile.img" /><QubesProxyVm conf_file="test-testproxy.conf" debug="False" default_user="user" dir_path="/var/lib/qubes/servicevms/test-testproxy" include_in_backups="True" installed_by_rpm="False" internal="False" kernel="3.2.7-10" kernelopts="" label="yellow" maxmem="4063" memory="200" name="test-testproxy" netid="3" netvm_qid="2" pcidevs="[]" private_img="private.img" qid="13" root_img="root.img" services="{&apos;meminfo-writer&apos;: True}" template_qid="1" uses_default_kernel="True" uses_default_kernelopts="True" vcpus="2" volatile_img="volatile.img" /></QubesVmCollection>
    107. 

  107. '''
    108. 

  108. 

  109. 

  110. class TC_00_BackupCompatibility(qubes.tests.BackupTestsMixin, qubes.tests.QubesTestCase):
    111. 

  111.     def create_whitelisted_appmenus(self, filename):
    112. 

  112.         f = open(filename, "w")
    113. 

  113.         f.write("gnome-terminal.desktop\n")
    114. 

  114.         f.write("nautilus.desktop\n")
    115. 

  115.         f.write("firefox.desktop\n")
    116. 

  116.         f.write("mozilla-thunderbird.desktop\n")
    117. 

  117.         f.write("libreoffice-startcenter.desktop\n")
    118. 

  118.         f.close()
    119. 

  119. 

  120.     def create_appmenus(self, dir, template, list):
    121. 

  121.         for name in list:
    122. 

  122.             f = open(os.path.join(dir, name + ".desktop"), "w")
    123. 

  123.             f.write(template.format(name=name, comment=name, command=name))
    124. 

  124.             f.close()
    125. 

  125. 

  126.     def create_private_img(self, filename):
    127. 

  127.         self.create_sparse(filename, 2*2**30)
    128. 

  128.         subprocess.check_call(["/usr/sbin/mkfs.ext4", "-q", "-F", filename])
    129. 

  129. 

  130.     def create_volatile_img(self, filename):
    131. 

  131.         self.create_sparse(filename, 11.5*2**30)
    132. 

  132.         sfdisk_input="0,1024,S\n,10240,L\n"
    133. 

  133.         p = subprocess.Popen(["/usr/sbin/sfdisk", "--no-reread", "-u",
    134. 

  134.                                    "M",
    135. 

  135.                                filename], stdout=open("/dev/null","w"),
    136. 

  136.                               stderr=subprocess.STDOUT, stdin=subprocess.PIPE)
    137. 

  137.         p.communicate(input=sfdisk_input)
    138. 

  138.         self.assertEqual(p.returncode, 0, "sfdisk failed with code %d" % p
    139. 

  139.                          .returncode)
    140. 

  140.         # TODO: mkswap
    141. 

  141. 

  142.     def fullpath(self, name):
    143. 

  143.         return os.path.join(self.backupdir, name)
    144. 

  144. 

  145.     def create_v1_files(self, r2b2=False):
    146. 

  146.         appmenus_list = [
    147. 

  147.             "firefox", "gnome-terminal", "evince", "evolution",
    148. 

  148.             "mozilla-thunderbird", "libreoffice-startcenter", "nautilus",
    149. 

  149.             "gedit", "gpk-update-viewer", "gpk-application"
    150. 

  150.         ]
    151. 

  151. 

  152.         os.mkdir(self.fullpath("appvms"))
    153. 

  153.         os.mkdir(self.fullpath("servicevms"))
    154. 

  154.         os.mkdir(self.fullpath("vm-templates"))
    155. 

  155. 

  156.         # normal AppVM
    157. 

  157.         os.mkdir(self.fullpath("appvms/test-work"))
    158. 

  158.         self.create_whitelisted_appmenus(self.fullpath(
    159. 

  159.             "appvms/test-work/whitelisted-appmenus.list"))
    160. 

  160.         os.symlink("/usr/share/qubes/icons/green.png",
    161. 

  161.                    self.fullpath("appvms/test-work/icon.png"))
    162. 

  162.         self.create_private_img(self.fullpath("appvms/test-work/private.img"))
    163. 

  163. 

  164.         # StandaloneVM
    165. 

  165.         os.mkdir(self.fullpath("appvms/test-standalonevm"))
    166. 

  166.         self.create_whitelisted_appmenus(self.fullpath(
    167. 

  167.             "appvms/test-standalonevm/whitelisted-appmenus.list"))
    168. 

  168.         os.symlink("/usr/share/qubes/icons/blue.png",
    169. 

  169.                    self.fullpath("appvms/test-standalonevm/icon.png"))
    170. 

  170.         self.create_private_img(self.fullpath(
    171. 

  171.             "appvms/test-standalonevm/private.img"))
    172. 

  172.         self.create_sparse(
    173. 

  173.             self.fullpath("appvms/test-standalonevm/root.img"), 10*2**30)
    174. 

  174.         self.fill_image(self.fullpath("appvms/test-standalonevm/root.img"),
    175. 

  175.                         100*1024*1024, True)
    176. 

  176.         os.mkdir(self.fullpath("appvms/test-standalonevm/apps.templates"))
    177. 

  177.         self.create_appmenus(self.fullpath("appvms/test-standalonevm/apps"
    178. 

  178.                                            ".templates"),
    179. 

  179.                              APPTEMPLATE_R2B2,
    180. 

  180.                              appmenus_list)
    181. 

  181.         os.mkdir(self.fullpath("appvms/test-standalonevm/kernels"))
    182. 

  182.         for k_file in ["initramfs", "vmlinuz", "modules.img"]:
    183. 

  183.             self.fill_image(self.fullpath("appvms/test-standalonevm/kernels/"
    184. 

  184.             + k_file), 10*1024*1024)
    185. 

  185. 

  186.         # VM based on custom template
    187. 

  187.         subprocess.check_call(
    188. 

  188.             ["/bin/cp", "-a", self.fullpath("appvms/test-work"),
    189. 

  189.                         self.fullpath("appvms/test-custom-template-appvm")])
    190. 

  190. 

  191.         # HVM
    192. 

  192.         if r2b2:
    193. 

  193.             subprocess.check_call(
    194. 

  194.                 ["/bin/cp", "-a", self.fullpath("appvms/test-standalonevm"),
    195. 

  195.                             self.fullpath("appvms/test-testhvm")])
    196. 

  196. 

  197.         # ProxyVM
    198. 

  198.         os.mkdir(self.fullpath("servicevms/test-testproxy"))
    199. 

  199.         self.create_whitelisted_appmenus(self.fullpath(
    200. 

  200.             "servicevms/test-testproxy/whitelisted-appmenus.list"))
    201. 

  201.         self.create_private_img(
    202. 

  202.             self.fullpath("servicevms/test-testproxy/private.img"))
    203. 

  203. 

  204.         # Custom template
    205. 

  205.         os.mkdir(self.fullpath("vm-templates/test-template-clone"))
    206. 

  206.         self.create_private_img(
    207. 

  207.             self.fullpath("vm-templates/test-template-clone/private.img"))
    208. 

  208.         self.create_sparse(self.fullpath(
    209. 

  209.             "vm-templates/test-template-clone/root-cow.img"), 10*2**30)
    210. 

  210.         self.create_sparse(self.fullpath(
    211. 

  211.             "vm-templates/test-template-clone/root.img"), 10*2**30)
    212. 

  212.         self.fill_image(self.fullpath(
    213. 

  213.             "vm-templates/test-template-clone/root.img"), 1*2**30, True)
    214. 

  214.         self.create_volatile_img(self.fullpath(
    215. 

  215.             "vm-templates/test-template-clone/volatile.img"))
    216. 

  216.         subprocess.check_call([
    217. 

  217.             "/bin/tar", "cS",
    218. 

  218.             "-f", self.fullpath(
    219. 

  219.                 "vm-templates/test-template-clone/clean-volatile.img.tar"),
    220. 

  220.             "-C", self.fullpath("vm-templates/test-template-clone"),
    221. 

  221.             "volatile.img"])
    222. 

  222.         self.create_whitelisted_appmenus(self.fullpath(
    223. 

  223.             "vm-templates/test-template-clone/whitelisted-appmenus.list"))
    224. 

  224.         self.create_whitelisted_appmenus(self.fullpath(
    225. 

  225.             "vm-templates/test-template-clone/vm-whitelisted-appmenus.list"))
    226. 

  226.         if r2b2:
    227. 

  227.             self.create_whitelisted_appmenus(self.fullpath(
    228. 

  228.                 "vm-templates/test-template-clone/netvm-whitelisted-appmenus"
    229. 

  229.                 ".list"))
    230. 

  230.         os.symlink("/usr/share/qubes/icons/green.png",
    231. 

  231.                    self.fullpath("vm-templates/test-template-clone/icon.png"))
    232. 

  232.         os.mkdir(
    233. 

  233.             self.fullpath("vm-templates/test-template-clone/apps.templates"))
    234. 

  234.         self.create_appmenus(
    235. 

  235.             self.fullpath("vm-templates/test-template-clone/apps.templates"),
    236. 

  236.             APPTEMPLATE_R2B2,
    237. 

  237.             appmenus_list)
    238. 

  238.         os.mkdir(self.fullpath("vm-templates/test-template-clone/apps"))
    239. 

  239.         self.create_appmenus(
    240. 

  240.             self.fullpath("vm-templates/test-template-clone/apps"),
    241. 

  241.             APPTEMPLATE_R2B2.replace("%VMNAME%", "test-template-clone")
    242. 

  242.             .replace("%VMDIR%", self.fullpath(
    243. 

  243.                 "vm-templates/test-template-clone")),
    244. 

  244.             appmenus_list)
    245. 

  245. 

  246. 

  247.     def test_100_r1(self):
    248. 

  248.         self.create_v1_files(r2b2=False)
    249. 

  249. 

  250.         f = open(self.fullpath("qubes.xml"), "w")
    251. 

  251.         f.write(QUBESXML_R1)
    252. 

  252.         f.close()
    253. 

  253. 

  254.         self.restore_backup(self.backupdir, options={
    255. 

  255.             'use-default-template': True,
    256. 

  256.         })
    257. 

  257.         self.assertIsNotNone(self.qc.get_vm_by_name("test-template-clone"))
    258. 

  258.         self.assertIsNotNone(self.qc.get_vm_by_name("test-testproxy"))
    259. 

  259.         self.assertIsNotNone(self.qc.get_vm_by_name("test-work"))
    260. 

  260.         self.assertIsNotNone(self.qc.get_vm_by_name("test-standalonevm"))
    261. 

  261.         self.assertIsNotNone(self.qc.get_vm_by_name(
    262. 

  262.             "test-custom-template-appvm"))
    263. 

  263.         self.assertEqual(self.qc.get_vm_by_name("test-custom-template-appvm")
    264. 

  264.                          .template,
    265. 

  265.                          self.qc.get_vm_by_name("test-template-clone"))
    266. 

  266. 

  267.     def test_200_r2b2(self):
    268. 

  268.         self.create_v1_files(r2b2=True)
    269. 

  269. 

  270.         f = open(self.fullpath("qubes.xml"), "w")
    271. 

  271.         f.write(QUBESXML_R2B2)
    272. 

  272.         f.close()
    273. 

  273. 

  274.         self.restore_backup(self.backupdir, options={
    275. 

  275.             'use-default-template': True,
    276. 

  276.         })
    277. 

  277.         self.assertIsNotNone(self.qc.get_vm_by_name("test-template-clone"))
    278. 

  278.         self.assertIsNotNone(self.qc.get_vm_by_name("test-testproxy"))
    279. 

  279.         self.assertIsNotNone(self.qc.get_vm_by_name("test-work"))
    280. 

  280.         self.assertIsNotNone(self.qc.get_vm_by_name("test-testhvm"))
    281. 

  281.         self.assertIsNotNone(self.qc.get_vm_by_name("test-standalonevm"))
    282. 

  282.         self.assertIsNotNone(self.qc.get_vm_by_name(
    283. 

  283.             "test-custom-template-appvm"))
    284. 

  284.         self.assertEqual(self.qc.get_vm_by_name("test-custom-template-appvm")
    285. 

  285.                          .template,
    286. 

  286.                          self.qc.get_vm_by_name("test-template-clone"))
    287.