Implemented further decription steps

Readme.md

@@ -22,8 +22,9 @@ python scripts/main.py --mode decrypt --master-password "ewrwerwerew"  &&
 python scripts/main.py --mode decrypt --user "1" 
 
 
-python scripts/main.py --mode cleanup --user "1"  && python scripts/main.py --mode decrypt --user "1" --user-password "O3ITMWXZED9FKYQ0PB2WNVRWSCSCYVXCD00PJ6GQ4MFPIUWBVDCYSSSX9ZDBW5QU"
-
+python scripts/main.py --mode cleanup --user "1"  && python scripts/main.py --mode decrypt --user "1" --user-password "O3ITMWXZED9FKYQ0PB2WNVRWSCSCYVXCD00PJ6GQ4MFPIUWBVDCYSSSX9ZDBW5QU" << END_OF_INPUTS
+2
+END_OF_INPUTS
 ```
 # Requirements to know
 - Amount of People
@@ -42,8 +43,9 @@ python scripts/main.py --mode cleanup --user "1"  && python scripts/main.py --mo
 ```
 
 ## todo 
-- Implement cleanup
+- implement tails setup script
 - implement relativ call
+- implement tmp mount for decrypted files
 
 ## Further Information
 - https://www.tutorialspoint.com/python/python_command_line_arguments.htm

scripts/classes/Decryption.py

@@ -33,6 +33,19 @@ class Decryption(AbstractSplittedSecret):
         self.decrypter_ids = []
         self.addDecrypterId(self.user_id)
         
+    def resetPasswordShare(self):
+        self.password_parts = {}
+        self.addPasswordShare(self.user_id,self.getPasswordShare())
+
+    def addPasswordShare(self,user_id,password_share):
+        self.password_parts[str(user_id)] = password_share
+        
+    def getSharedPassword(self):
+        shared_password = ''
+        for password_share_index in sorted(self.password_parts):
+            shared_password += str(self.password_parts[password_share_index])
+        return shared_password
+    
     def addDecrypterId(self,decrypter_id):
         decrypter_id = int(decrypter_id)
         if decrypter_id not in self.valid_decrypter_ids:
@@ -43,9 +56,24 @@ class Decryption(AbstractSplittedSecret):
             raise Exception("The decrypter is already in the list.")
         self.decrypter_ids.append(decrypter_id)
     
-    def getDecryptersIds(self):
+    def getUserId(self):
+        return self.user_id
+    
+    def getCoDecrypterIds(self):
+        co_decrypter_ids = self.decrypter_ids[:]
+        co_decrypter_ids.remove(int(self.user_id))
+        return co_decrypter_ids
+    
+    def getDecrypterIds(self):
         return self.decrypter_ids
     
+    def getDecryptersGroupName(self):
+        self.decrypter_ids.sort()
+        return ''.join(str(x) for x in self.decrypter_ids)
+    
+    def getPasswordShare(self):
+        return self.user_data['groups'][str(self.getDecryptersGroupName())]
+        
     def getNeededCoDecryptersAmount(self):
         return self.needed_decrypters_amount -1
     

scripts/classes/Encryption.py

@@ -51,11 +51,11 @@ class Encryption(AbstractSplittedSecret):
         characters = string.ascii_letters + string.digits
         return (''.join(random.choice(characters) for i in range(length)).upper())
     
-    def isGroupValid(self,password_group_index_str):
+    def isGroupValid(self,password_group_name):
         secret_stakeholders_range=range(1,(self.amount_of_secret_holders+1))
         valid_numbers = re.compile("([" + ','.join([str(x) for x in secret_stakeholders_range]) + "]{" + str(self.group_members_amount) + "})")
         unvalid_sequenz = re.compile("(.)\\1+")
-        return re.search(valid_numbers, password_group_index_str) and not re.search(unvalid_sequenz, password_group_index_str)
+        return re.search(valid_numbers, password_group_name) and not re.search(unvalid_sequenz, password_group_name)
     
     def compileContacts(self):
         contacts = {}
@@ -71,21 +71,21 @@ class Encryption(AbstractSplittedSecret):
         self.compileContacts()
         index = self.getStartnumber()
         while index < self.getEndnumber():
-            password_group_index_str = ''.join(sorted(str(index)))
-            if self.isGroupValid(password_group_index_str):
-                password_group_index_int = int(password_group_index_str)
+            password_group_name = ''.join(sorted(str(index)))
+            if self.isGroupValid(password_group_name):
+                password_group_index_int = int(password_group_name)
                 if not password_group_index_int in self.group_mapped_data:
                     self.group_mapped_data[password_group_index_int] = {}
                     self.group_mapped_data[password_group_index_int]['members'] = {}
                     self.group_mapped_data[password_group_index_int]['password'] = '' 
                     password = ''
-                    for secret_holder_index in password_group_index_str:
+                    for secret_holder_index in password_group_name:
                         self.group_mapped_data[password_group_index_int]['members'][secret_holder_index]={}
                         particial_password_length= int(self.OVERALL_PASSWORD_LENGTHS*self.quota_factor); 
                         password_part = self.createPassword(particial_password_length)
                         self.group_mapped_data[password_group_index_int]['members'][secret_holder_index] = password_part
                         password += password_part
-                        self.user_mapped_data[secret_holder_index]['groups'][password_group_index_str] = password_part
+                        self.user_mapped_data[secret_holder_index]['groups'][password_group_name] = password_part
                     self.group_mapped_data[password_group_index_int]['password'] += password
             index += 1
             

scripts/main.py

@@ -4,6 +4,7 @@ from classes.Cleanup import Cleanup
 from classes.Decryption import Decryption
 from getpass import getpass
 from classes.AbstractSplittedSecret import AbstractSplittedSecret
+import traceback
 
 cleanup = Cleanup()
 
@@ -13,8 +14,8 @@ def clean_exit():
     print("Leaving program. Goodby :)")
     exit();
     pass
-
-if __name__ == '__main__':
+try:
+    if __name__ == '__main__':
         parser = argparse.ArgumentParser()
         parser.add_argument('--mode',type=str, dest='mode',required=True,choices=['cleanup','encrypt','decrypt'])
         parser.add_argument('--amount',type=int, dest='amount_of_secret_holders',required=False,choices=AbstractSplittedSecret.getCoSecretHoldersRange())
@@ -75,7 +76,7 @@ if __name__ == '__main__':
                     try:
                         person_counter = 1
                         while person_counter <= decrypt.getNeededCoDecryptersAmount():
-                        print("The following user id's are in the decryption list: " + str(decrypt.getDecryptersIds()))
+                            print("The following user id's are in the decryption list: " + str(decrypt.getDecrypterIds()))
                             print("You need at least <<" + str(decrypt.getNeededCoDecryptersAmount()) +">> other person to decrypt the secret.")
                             print("Type in the user id of another encrypter:")
                             decrypt.addDecrypterId(int(input()))
@@ -83,6 +84,20 @@ if __name__ == '__main__':
                         break
                     except Exception as error:
                         print("The following error occured <<" + str(error) + ">> :( \n Please try again :)")
+                print("\nFOR PASSWORD GROUP: " + decrypt.getDecryptersGroupName()) 
+                print("FOR USER ID: "  + decrypt.getUserId())
+                print("PASSWORD SHARE IS: " + decrypt.getPasswordShare() + "\n")
+                while True:
+                    decrypt.resetPasswordShare()
+                    co_decrypter_ids = decrypt.getCoDecrypterIds()
+                    print("Please execute this script at the users " + str(co_decrypter_ids) + ".")
+                    for co_decrypter_id in decrypt.getCoDecrypterIds():
+                        print("\nFOR PASSWORD GROUP: " + decrypt.getDecryptersGroupName()) 
+                        print("FOR USER: " + str(co_decrypter_id)) 
+                        print("PASSWORD SHARE IS: ")
+                        decrypt.addPasswordShare(co_decrypter_id, input())
+                    print("\nTHE SHARED PASSWORD IS: " + decrypt.getSharedPassword())
+                    break;
                         
                 clean_exit()  
             print("Decrypting accumulated file...")
@@ -105,4 +120,6 @@ if __name__ == '__main__':
             encrypt.compileData()
             encrypt.encrypt()
             clean_exit()
+except Exception:
+    print(traceback.format_exc())
 clean_exit()