split-secret/scripts/generate-hashtable.py

60 lines
3.0 KiB
Python

import argparse
import random
import string
def getPassword():
characters = string.ascii_letters + string.digits
return ''.join(random.choice(characters) for i in range(int(64*quota_factor))).upper()
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('-a', '--amount',type=int, dest='amount_of_secret_holders',required=True)
parser.add_argument('-q', '--quota', type=int, dest='decryption_quota', choices=range(1,101),required=True)
args = parser.parse_args()
amount_of_secret_holders = args.amount_of_secret_holders
decryption_quota = args.decryption_quota
quota_factor=decryption_quota/100
password_group_members_amount=amount_of_secret_holders * quota_factor
amount_of_partner_secrets=(amount_of_secret_holders * password_group_members_amount)
#required_passwords=amount_of_partner_secrets * ( amount_of_secret_holders -1) ;
required_passwords=amount_of_partner_secrets * amount_of_secret_holders
print(quota_factor);
print(amount_of_secret_holders);
print(decryption_quota);
print(required_passwords);
mathematical_formular_verification=[]
# Create Passwords
password_groups = {}
password_group_index = 0
secret_holder_index = 0
while password_group_index < required_passwords :
password_groups[password_group_index] = {};
password_groups[password_group_index]['members'] = {}
password_groups[password_group_index]['password'] = ''
group_members_count = 0
while group_members_count < password_group_members_amount :
if secret_holder_index == amount_of_secret_holders:
secret_holder_index = 0
password_groups[password_group_index]['members'][secret_holder_index]={}
password_groups[password_group_index]['members'][secret_holder_index]['password_part'] = getPassword()
password_groups[password_group_index]['members'][secret_holder_index]['password_index'] = group_members_count
password_groups[password_group_index]['password'] += ''.join(password_groups[password_group_index]['members'][secret_holder_index]['password_part'])
secret_holder_index += 1
group_members_count += 1
mathematical_formular_verification.append(sorted(password_groups[password_group_index]['members']))
password_group_index += 1
print(password_groups)
# Create User Mapping
user_splitted_passwords = {}
for password_group_index in password_groups:
for member_id in password_groups[password_group_index]['members']:
if not member_id in user_splitted_passwords:
user_splitted_passwords[member_id] = []
user_splitted_passwords[member_id].append({"password_information" : password_groups[password_group_index]['members'][member_id], "members": list(password_groups[password_group_index]['members'].keys())});
#print(user_splitted_passwords)
print(sorted(mathematical_formular_verification));