Merge remote-tracking branch 'qubesos/pr/42'

* qubesos/pr/42:
  verify-only avoid dom0 handler
  Fix verify-only
  Issue #3303

qubesadmin/backup/restore.py

@@ -70,6 +70,7 @@ _re_alphanum = re.compile(r'^[A-Za-z0-9-]*$')
 _tar_msg_re = re.compile(r".*#[0-9].*restore_pipe")
 _tar_file_size_re = re.compile(r"^[^ ]+ [^ ]+/[^ ]+ *([0-9]+) .*")
 
+
 class BackupCanceledError(QubesException):
     '''Exception raised when backup/restore was cancelled'''
     def __init__(self, msg, tmpdir=None):
@@ -1356,7 +1357,10 @@ class BackupRestore(object):
                     # This all means that if the file was correctly verified
                     # + decrypted, we will surely access the right file
                     filename = self._verify_and_decrypt(filename)
+                if not self.options.verify_only:
                     to_extract.put(os.path.join(self.tmpdir, filename))
+                else:
+                    os.unlink(os.path.join(self.tmpdir, filename))
 
             if self.canceled:
                 raise BackupCanceledError("Restore canceled",
@@ -1749,6 +1753,9 @@ class BackupRestore(object):
             if vm and vm_info.subdir:
                 vms_size += int(vm_info.size)
                 vms_dirs.append(vm_info.subdir)
+
+                if self.options.verify_only:
+                    continue
                 for name, volume in vm.volumes.items():
                     if not volume.save_on_stop:
                         continue
@@ -1768,6 +1775,7 @@ class BackupRestore(object):
                 restore_info['dom0'].good_to_go:
             vms_dirs.append(os.path.dirname(restore_info['dom0'].subdir))
             vms_size += restore_info['dom0'].size
+            if not self.options.verify_only:
                 handlers[restore_info['dom0'].subdir] = (self._handle_dom0, None)
         try:
             self._restore_vm_data(vms_dirs=vms_dirs, vms_size=vms_size,
@@ -1777,19 +1785,17 @@ class BackupRestore(object):
                 raise
             else:
                 self.log.error('Error extracting data: ' + str(err))
-                self.log.warning(
+        finally:
-                    "Continuing anyway to restore at least some VMs")
+            if self.log.getEffectiveLevel() > logging.DEBUG:
-
-        if self.options.verify_only:
                 shutil.rmtree(self.tmpdir)
-            return
 
         if self.canceled:
             raise BackupCanceledError("Restore canceled",
                                       tmpdir=self.tmpdir)
 
-        shutil.rmtree(self.tmpdir)
+        self.log.info("-> Done.")
-        self.log.info("-> Done. Please install updates for all the restored "
+        if not self.options.verify_only:
+            self.log.info("-> Please install updates for all the restored "
                           "templates.")
 
     def _restore_vms_metadata(self, restore_info):
@@ -1811,6 +1817,9 @@ class BackupRestore(object):
         for vm in self._templates_first(vms.values()):
             if self.canceled:
                 return
+            if self.options.verify_only:
+                self.log.info("-> Verifying %s...", vm.name)
+            else:
                 self.log.info("-> Restoring %s...", vm.name)
             kwargs = {}
             if vm.template:
@@ -1824,6 +1833,9 @@ class BackupRestore(object):
             new_vm = None
             vm_name = restore_info[vm.name].name
 
+            if self.options.verify_only:
+                new_vm = self.backup_app.domains[vm_name]
+            else:
                 try:
                     # first only create VMs, later setting may require other VMs
                     # be already created

qubesadmin/tools/qvm_backup_restore.py

@@ -134,6 +134,10 @@ def handle_broken(app, args, restore_info):
                 "Or use --rename-conflicting to restore those VMs under "
                 "modified names (with numbers at the end).")
 
+    if args.verify_only:
+        app.log.info("The above VM archive(s) will be verified.")
+        app.log.info("Existing VMs will NOT be removed or altered.")
+    else:
         app.log.info("The above VMs will be copied and added to your system.")
         app.log.info("Exisiting VMs will NOT be removed.")