.gitignore

@@ -1,2 +1,2 @@
 __pycache__
-INSTRUCTIONS.md
+testcases.txt

Readme.md

@@ -40,23 +40,15 @@ To delete all encrypted data execute:
 python scripts/main.py --mode cleanup --file-types encrypted
 ```
 
-## decrypt 
+## decrypt
 
 ### decrypt automatic
 To decrypt the data execute:
 
 ```bash 
-python scripts/main.py --mode decrypt-data
+python scripts/main.py --mode decrypt
 ```
 
-### decrypt accumulated file
-To decrypt the accumulated datafile execute:
-
-```bash 
-python scripts/main.py --mode decrypt --meta
-```
-
-
 ### decrypt defined user
 To decrypt the data for a defined user execute:
 
@@ -64,19 +56,18 @@ To decrypt the data for a defined user execute:
 python scripts/main.py --mode decrypt --user "<<user_id>>"
 ```
 
-### addtional instructions
-In the [INSTRUCTIONS.md](./Instruction.md) file the master encrypter can leave additional instructions.
-
 ## encrypt
 
 ### encrypt main data
+
 ```bash 
 python scripts/main.py --secret-holders-amount "<<amount>>" --quota "<<quota>>" --mode encrypt --master-password "<<master_password>>" --input-directory "<<input_directory>>"
 ```
 
 ### encrypt master password
+
 To encrypt the master-password file and to create the neccessary encrypted meta data execute: 
 
 ```bash 
-python scripts/main.py --secret-holders-amount "<<amount>>" --quota "<<quota>>" --mode encrypt --add-user-information --master-password "<<master_password>>" --meta
+python scripts/main.py --secret-holders-amount "<<amount>>" --quota "<<quota>>" --mode encrypt --add-user-information --master-password "<<master_password>>" --create-meta-data
 ```

scripts/classes/Decryption.py

@@ -1,7 +1,6 @@
 import json
 import os
 from pathlib import Path
-import shlex
 
 class AutomaticIdentificationImpossibleException(Exception):
     pass
@@ -36,18 +35,12 @@ class Decryption():
         self.initializeNeededDecryptersAmount()
         self.initializeValidDecrypterIds()
     
-    def getEscapedMasterPassword(self):
-        return shlex.quote(self.master_password)
-    
     def initializeGroupDataEncryption(self):
         self.group_name = self.getDecryptersGroupName()
         self.encrypted_group_file_path = self.paths.getGroupFilePath(self.group_name, self.paths.TYPE_DECRYPTED)
         self.decryptGroupFile()
         self.master_password = self.loadTxtFile(self.encrypted_group_file_path).strip()
 
-    def getMasterPassword(self):
-        return self.master_password
-
     def initializeNeededDecryptersAmount(self):
         self.needed_decrypters_amount = len(str(list(self.user_data['groups'].keys())[0]))
     
@@ -77,6 +70,9 @@ class Decryption():
             shared_password += str(self.password_parts[password_share_index])
         return shared_password
     
+    def getMasterPassword(self):
+        return self.master_password
+    
     def addDecrypterId(self,decrypter_id):
         decrypter_id = int(decrypter_id)
         if decrypter_id not in self.valid_decrypter_ids:
@@ -118,7 +114,7 @@ class Decryption():
         return data
     
     def decryptFile(self,password,input_file_path,output_file_path):
-        self.cli.executeCommand('gpg --batch --passphrase '+ shlex.quote(password)  + ' -o "' + output_file_path +'" "'+ input_file_path+'"')
+        self.cli.executeCommand('gpg --batch --passphrase "'+ password + '" -o "' + output_file_path +'" "'+ input_file_path+'"')
     
     def decryptUserFile(self):
         input_file_path = self.paths.getUserFilePath(self.user_id,self.paths.TYPE_ENCRYPTED)
@@ -134,4 +130,4 @@ class Decryption():
         self.decryptFile(self.user_password, input_file_path, output_file_path)
     
     def decryptMainData(self):
-        self.cli.executeCommand('gpg --batch --passphrase ' + shlex.quote(self.getMasterPassword()) + ' -d "' + self.paths.getEncryptedMainDataFile() + '" | tar --one-top-level="' + self.paths.getDecryptedMainDataStandartFolder() + '" -xvzf -')
+        self.cli.executeCommand('gpg --batch --passphrase "' + self.getMasterPassword() + '" -d "' + self.paths.getEncryptedMainDataFile() + '" | tar --one-top-level="' + self.paths.getDecryptedMainDataStandartFolder() + '" -xvzf -')

scripts/classes/Encryption.py

@@ -5,7 +5,6 @@ import numpy
 import re
 import json
 from .Paths import Paths
-import shlex
 
 class Encryption():
     
@@ -105,7 +104,7 @@ class Encryption():
             index += 1
             
     def encryptStringToFile(self,text,output_file,password):
-        self.cli.executeCommand('echo ' + shlex.quote(text) + ' | gpg --symmetric --armor --batch --passphrase ' + shlex.quote(password) + ' -o "' + output_file + '"')
+        self.cli.executeCommand('echo \'' + text + '\' | gpg --symmetric --armor --batch --passphrase "' + password + '" -o "' + output_file + '"')
     
     def encryptGroupFiles(self):
         for password_group_index_int in self.group_mapped_data:
@@ -126,7 +125,7 @@ class Encryption():
     See: https://stackoverflow.com/questions/30650841/why-am-i-getting-errno-7-argument-list-too-long-and-oserror-errno-24-too-ma
     '''
     def encryptFileToFile(self,input_file,output_file,password):
-        self.cli.executeCommand('cat \'' + input_file + '\' | gpg --symmetric --armor --batch --passphrase ' + shlex.quote(password) + ' -o "' + output_file + '"')
+        self.cli.executeCommand('cat \'' + input_file + '\' | gpg --symmetric --armor --batch --passphrase "' + password + '" -o "' + output_file + '"')
     
     def deleteDecryptedAccumulatedFile(self):
         self.cli.executeCommand('rm ' + self.paths.getAccumulatedFilePath(Paths.TYPE_DECRYPTED))
@@ -145,7 +144,7 @@ class Encryption():
         self.deleteDecryptedAccumulatedFile()
         
     def encryptMainData(self,input_directory):
-        self.cli.executeCommand('tar -C"' + input_directory + '" -cvzf - ./ | gpg -c --batch --passphrase ' + shlex.quote(self.master_password) + ' > "' + self.paths.getEncryptedMainDataFile() + '"')
+        self.cli.executeCommand('tar -C"' + input_directory + '" -cvzf - ./ | gpg -c --batch --passphrase "' + self.master_password +'" > "' + self.paths.getEncryptedMainDataFile() + '"')
     
     def encryptMetaData(self):
         self.encryptUserFile()

scripts/main.py

@@ -40,13 +40,11 @@ try:
         parser.add_argument('--user',type=int, dest='user',choices=Encryption.getSecretHoldersRange(),required=False)
         parser.add_argument('--add-user-information',type=bool, dest='add_user_information', default=False, required=False, action=argparse.BooleanOptionalAction, help="Add additional information to users.")
         parser.add_argument('--input-directory',type=str,dest='input_directory',required=False, help="The directory from which the data should be encrypted.")
-        parser.add_argument('--meta-data',type=bool, dest='meta_data', default=False, required=False, action=argparse.BooleanOptionalAction, help="When mode is encrypt and this flag is set, the encrypted meta data is created. When mode is decrypt and this flag is set the accumulated file will be decrypted.")
+        parser.add_argument('--create-meta-data',type=bool, dest='create_meta_data', default=False, required=False, action=argparse.BooleanOptionalAction, help="When mode is encrypt and this flag is set, the encrypted meta data is created.")
         args = parser.parse_args()
 
         print("Application started.")
         print("To leave the appplication use the key kombination: <<Ctr>> + <<Alt>> + <<C>>")
-        print("Cleaning up all decrypted files.")
-        cleanup.cleanupFiles(Paths.TYPE_DECRYPTED)
         print("Selected Mode: " + args.mode)
         
         if args.mode == 'cleanup':   
@@ -65,91 +63,85 @@ try:
             
         if args.mode == 'decrypt':
             decrypt = Decryption(cli,paths)
-            if args.meta_data is True:
-                if args.master_password is None:
-                    print("Enter the master password:")
-                    master_password = getpass()
+            if args.master_password is None:
+                if args.user is None:
+                    try:
+                        print("Attempt to identify user.") 
+                        user_id = decrypt.identifyUser()
+                        print("The user id is: " + user_id)
+                    except:
+                        print("A automatic user id identification wasn't possible.")
+                        print("Type in the user id:")
+                        user_id = input()
+                    decrypt.initializeUser(user_id)
                 else:
-                    master_password = args.master_password
-                decrypt = Decryption(cli,paths)
-                print("Decrypting accumulated data.")
-                decrypt.setUserPassword(master_password)
-                decrypt.decryptAccumulatedFile()
-                dirty_exit()
-            if args.user is None:
-                try:
-                    print("Attempt to identify user.") 
-                    user_id = decrypt.identifyUser()
-                    print("The user id is: " + user_id)
-                except:
-                    print("A automatic user id identification wasn't possible.")
-                    print("Type in the user id:")
-                    user_id = input()
-                decrypt.initializeUser(user_id)
-            else:
-                decrypt.initializeUser(args.user)
-            if args.user_password is None:
-                while True:
-                    print("Enter the user password:")
-                    decrypt.setUserPassword(getpass())
+                    decrypt.initializeUser(args.user)
+                if args.user_password is None:
+                    while True:
+                        print("Enter the user password:")
+                        decrypt.setUserPassword(getpass())
+                        print("Decrypting User File...")
+                        try:
+                            decrypt.initializeUserDataDecryption();
+                            break;
+                        except Exception as error:
+                            print("An error occured. Propably you typed in a wrong password :( The error is: " + str(error))
+                else:
+                    decrypt.setUserPassword(args.user_password)
                     print("Decrypting User File...")
                     try:
                         decrypt.initializeUserDataDecryption();
-                        break;
                     except Exception as error:
-                        print("An error occured. Propably you typed in a wrong password :( The error is: " + str(error))
-            else:
-                decrypt.setUserPassword(args.user_password)
-                print("Decrypting User File...")
-                try:
-                    decrypt.initializeUserDataDecryption();
-                except Exception as error:
-                    print("An error occured. Propably you passed a wrong password :( The error is: " + str(error))
-                    clean_exit()
-            print("\nContact the following persons and request their password share: \n")
-            for contact_id in decrypt.user_data['contacts']:
-                print("user_id: " + contact_id)
-                for label in decrypt.user_data['contacts'][contact_id]:
-                    print(label + ": " + decrypt.user_data['contacts'][contact_id][label])
-            while True:
-                print("\nReset password shares.\n")
-                decrypt.resetDecrypterIds()
-                try:
-                    password_shares_count = 1
-                    while password_shares_count < decrypt.getNeededDecryptersAmount():
-                        print(str(password_shares_count) + " password shares had been added.")
-                        print("Password shares for the the users " + str(decrypt.getDecrypterIds()) + " been added. ")
-                        print("You need to add " + str((decrypt.getNeededDecryptersAmount()-password_shares_count)) +" more password shares.")
-                        print("\nType in the user id of another decrypter:")
-                        decrypt.addDecrypterId(int(input()))
-                        password_shares_count += 1
-                    break
-                except Exception as error:
-                    print("The following error occured <<" + str(error) + ">> :( \n Try again :)")
-            print("\nYour data is:\n")
-            print("FOR PASSWORD GROUP: " + decrypt.getDecryptersGroupName()) 
-            print("FOR USER ID: "  + decrypt.getUserId())
-            print("PASSWORD SHARE IS: " + decrypt.getPasswordShare() + "\n")
-            while True:
-                try:
-                    decrypt.resetPasswordShare()
-                    co_decrypter_ids = decrypt.getCoDecrypterIds()
-                    for co_decrypter_id in decrypt.getCoDecrypterIds():
-                        print("Type in the password share for: \n")
-                        print("FOR PASSWORD GROUP: " + decrypt.getDecryptersGroupName()) 
-                        print("FOR USER: " + str(co_decrypter_id)) 
-                        print("PASSWORD SHARE IS: ")
-                        decrypt.addPasswordShare(co_decrypter_id, input())
-                    print("\nTHE GROUP PASSWORD IS: " + decrypt.getGroupPassword())
-                    print("\nDecrypting group password file.\n")
-                    decrypt.initializeGroupDataEncryption()
-                    print("THE MASTER PASSWORD IS: " + decrypt.getMasterPassword())
-                    break;
-                except:
-                    print("An unexpected error occured: \n" + traceback.format_exc())
-            print("Decrypting main data.")
-            decrypt.decryptMainData()
-            print("The data was decrypted to: " + paths.getDecryptedMainDataStandartFolder())
+                        print("An error occured. Propably you passed a wrong password :( The error is: " + str(error))
+                        clean_exit()
+                print("\nContact the following persons and request their password share: \n")
+                for contact_id in decrypt.user_data['contacts']:
+                    print("user_id: " + contact_id)
+                    for label in decrypt.user_data['contacts'][contact_id]:
+                        print(label + ": " + decrypt.user_data['contacts'][contact_id][label])
+                while True:
+                    print("\nReset password shares.\n")
+                    decrypt.resetDecrypterIds()
+                    try:
+                        password_shares_count = 1
+                        while password_shares_count < decrypt.getNeededDecryptersAmount():
+                            print(str(password_shares_count) + " password shares had been added.")
+                            print("Password shares for the the users " + str(decrypt.getDecrypterIds()) + " been added. ")
+                            print("You need to add " + str((decrypt.getNeededDecryptersAmount()-password_shares_count)) +" more password shares.")
+                            print("\nType in the user id of another decrypter:")
+                            decrypt.addDecrypterId(int(input()))
+                            password_shares_count += 1
+                        break
+                    except Exception as error:
+                        print("The following error occured <<" + str(error) + ">> :( \n Try again :)")
+                print("\nYour data is:\n")
+                print("FOR PASSWORD GROUP: " + decrypt.getDecryptersGroupName()) 
+                print("FOR USER ID: "  + decrypt.getUserId())
+                print("PASSWORD SHARE IS: " + decrypt.getPasswordShare() + "\n")
+                while True:
+                    try:
+                        decrypt.resetPasswordShare()
+                        co_decrypter_ids = decrypt.getCoDecrypterIds()
+                        for co_decrypter_id in decrypt.getCoDecrypterIds():
+                            print("Type in the password share for: \n")
+                            print("FOR PASSWORD GROUP: " + decrypt.getDecryptersGroupName()) 
+                            print("FOR USER: " + str(co_decrypter_id)) 
+                            print("PASSWORD SHARE IS: ")
+                            decrypt.addPasswordShare(co_decrypter_id, input())
+                        print("\nTHE GROUP PASSWORD IS: " + decrypt.getGroupPassword())
+                        print("\nDecrypting group password file.\n")
+                        decrypt.initializeGroupDataEncryption()
+                        print("THE MASTER PASSWORD IS: " + decrypt.getMasterPassword())
+                        break;
+                    except:
+                        print("An unexpected error occured: \n" + traceback.format_exc())
+                print("Decrypting main data.")
+                decrypt.decryptMainData()
+                print("The data was decrypted to: " + paths.getDecryptedMainDataStandartFolder())
+                dirty_exit()
+            print("Decrypting accumulated data.")
+            decrypt.setUserPassword(args.master_password)
+            decrypt.decryptAccumulatedFile()
             dirty_exit()
         
         if args.mode == 'encrypt':
@@ -165,7 +157,7 @@ try:
                         print("Enter attribut <<" + label + ">> for user <<" + user_id+ ">>:" )
                         encrypt.addInformationToUser(user_id, label, str(input()))
             encrypt.compileData()
-            if args.meta_data is True:
+            if args.create_meta_data is True:
                 print("Create and encrypt meta data.")
                 encrypt.encryptMetaData()
             if args.input_directory is not None: