mirror of
https://github.com/kevinveenbirkenbach/splitted-secret.git
synced 2024-11-01 00:53:11 +01:00
78 lines
3.9 KiB
Python
78 lines
3.9 KiB
Python
import random
|
|
import string
|
|
import math
|
|
import numpy
|
|
import re
|
|
import classes.Cli as Cli
|
|
|
|
class Generate:
|
|
|
|
def __init__(self, amount_of_secret_holders, decryption_quota):
|
|
self.amount_of_secret_holders = amount_of_secret_holders
|
|
self.decryption_quota = decryption_quota
|
|
self.decrypted_master_password_file_path="data/decrypted/password_files/master-password.txt"
|
|
self.quota_factor=self.decryption_quota/100
|
|
self.group_members_amount=math.ceil(self.amount_of_secret_holders * self.quota_factor)
|
|
self.cli = Cli.Cli()
|
|
|
|
def getStartnumber(self):
|
|
index = 0
|
|
start_number = ''
|
|
while index < self.group_members_amount:
|
|
start_number += '1'
|
|
index += 1
|
|
return int(start_number)
|
|
|
|
def getEndnumber(self):
|
|
index = 0
|
|
start_number = ''
|
|
while index < self.group_members_amount:
|
|
start_number += str(self.amount_of_secret_holders)
|
|
index += 1
|
|
return int(start_number)
|
|
|
|
def savePassword(self,password,password_file_path):
|
|
print("Saving password to: " + password_file_path)
|
|
master_password_file = open(password_file_path, "a")
|
|
master_password_file.seek(0)
|
|
master_password_file.truncate()
|
|
master_password_file.write(password)
|
|
master_password_file.close()
|
|
|
|
def getPassword(self):
|
|
characters = string.ascii_letters + string.digits
|
|
return ''.join(random.choice(characters) for i in range(int(64*self.quota_factor))).upper()
|
|
|
|
def isGroupValid(self,password_group_index_str):
|
|
width= range(1,(self.amount_of_secret_holders+1))
|
|
regex="([" + ','.join([str(x) for x in width]) + "]{" + str(self.group_members_amount) + "})"
|
|
valid_numbers = re.compile(regex)
|
|
unvalid_sequenz = re.compile("(.)\\1+")
|
|
return re.search(valid_numbers, password_group_index_str) and not re.search(unvalid_sequenz, password_group_index_str)
|
|
|
|
def execute(self):
|
|
index = self.getStartnumber()
|
|
password_groups = {}
|
|
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)
|
|
if not password_group_index_int in password_groups:
|
|
password_index = 1
|
|
password_groups[password_group_index_int] = {}
|
|
password_groups[password_group_index_int]['members'] = {}
|
|
password_groups[password_group_index_int]['password'] = ''
|
|
password = ''
|
|
for secret_holder_index in password_group_index_str:
|
|
password_groups[password_group_index_int]['members'][secret_holder_index]={}
|
|
password_part = self.getPassword()
|
|
password_groups[password_group_index_int]['members'][secret_holder_index]['password_part'] = password_part
|
|
password_groups[password_group_index_int]['members'][secret_holder_index]['password_index'] = password_index
|
|
password += password_part
|
|
password_index += 1
|
|
password_groups[password_group_index_int]['password'] += password
|
|
decrypted_splitted_password_file = "data/decrypted/splitted_password_files/" + password_group_index_str + ".txt"
|
|
encrypted_splitted_password_file = "data/encrypted/splitted_password_files/" + password_group_index_str + ".txt.gpg"
|
|
self.cli.executeCommand('cp -v "' + self.decrypted_master_password_file_path + '" "' + decrypted_splitted_password_file + '" && gpg --batch --passphrase "' + password + '" -c "' + decrypted_splitted_password_file + '"')
|
|
print(self.cli.getCommandString())
|
|
index += 1 |