Fix return state in raw_write when signature is false

1 files changed, 3 insertions, 4 deletions

diff --git a/backend/raw_write.py b/backend/raw_write.py
index cdac7e2..508a9ae 100755
--- a/backend/raw_write.py
+++ b/backend/raw_write.py
@@ -119,10 +119,10 @@ class Dumper(object):
                    return
             ifc.close()
             self._progress = 100
-            self.finished.set()
             self.return_state = True
             self.return_message = _("Writing terminated")
             logging.debug(self.return_message)
+            self.finished.set()
             return
 
     def _get_sum(self, source):
@@ -175,7 +175,6 @@ class Dumper(object):
 
     def _check_write(self, target, source):
         logging.debug("Start checking")
-        self.return_state = False
         self.return_message = ""
         b = os.path.getsize(source)
         # Compute the sum from the written device
@@ -203,11 +202,12 @@ class Dumper(object):
         if self.signature_found and not self.signature_checked:
                 #,  keep the pourcent, this is the place for source file name
             self.return_message = _('Invalid signature for %s')%self.source_file
+            self.return_state = False
         if (self.sum_check == "") :  # Can't get stored sum
                 self.return_message += _('SHA3 sum: {}').format(sha512sumcalc)
         #  compare the sums
         elif (sha512sumcalc == self.sum_check) :
-            if self.signature_checked:
+            if self.signature_checked and self.signature_found:
                 #,  keep the bracket, this is the place for sum type
                 self.return_message +="\n" + _("The {} sum check is OK and the sum is signed").format(self.sum_type)
             else :
@@ -223,7 +223,6 @@ class Dumper(object):
         #except:
         	#pass
         self._progress = 100
-
         logging.info(self.return_message)
         self.finished.set()