def setUp(self):
self.tp = ThresholdParameters(3, 5)
self.kp = ThresholdCrypto.static_512_key_parameters()
self.pk, self.shares = ThresholdCrypto.create_public_key_and_shares_centralized(
self.kp, self.tp)
self.em = ThresholdCrypto.encrypt_message('Some secret message',
self.pk)
self.reconstruct_shares = [self.shares[i] for i in [0, 2, 4]
] # choose 3 of 5 key shares
self.partial_decryptions = [
ThresholdCrypto.compute_partial_decryption(self.em, share)
for share in self.reconstruct_shares
]
def test_key_encryption_decryption_without_enough_shares(self):
r = number.getRandomRange(2, self.kp.q)
testkey_element = pow(self.kp.g, r, self.kp.p)
g_k, c = ThresholdCrypto._encrypt_key_element(testkey_element, self.pk)
em = EncryptedMessage(g_k, c, '')
reconstruct_shares = [self.shares[i]
for i in [0, 4]] # choose 2 of 5 key shares
partial_decryptions = [
ThresholdCrypto.compute_partial_decryption(em, share)
for share in reconstruct_shares
]
rec_testkey_element = ThresholdCrypto._combine_shares(
partial_decryptions, em, self.tp, self.kp)
self.assertNotEqual(testkey_element, rec_testkey_element)
def test_complete_process_with_enough_shares(self):
key_params = ThresholdCrypto.static_512_key_parameters()
thresh_params = ThresholdParameters(3, 5)
pub_key, key_shares = ThresholdCrypto.create_public_key_and_shares_centralized(
key_params, thresh_params)
message = 'Some secret message to be encrypted!'
encrypted_message = ThresholdCrypto.encrypt_message(message, pub_key)
reconstruct_shares = [key_shares[i]
for i in [0, 2, 4]] # choose 3 of 5 key shares
partial_decryptions = [
ThresholdCrypto.compute_partial_decryption(encrypted_message,
share)
for share in reconstruct_shares
]
decrypted_message = ThresholdCrypto.decrypt_message(
partial_decryptions, encrypted_message, thresh_params, key_params)
self.assertEqual(message, decrypted_message)
from threshold_crypto import (ThresholdCrypto, ThresholdParameters)
# Generate parameters, public key and shares
key_params = ThresholdCrypto.static_2048_key_parameters()
thresh_params = ThresholdParameters(3, 5)
pub_key, key_shares = ThresholdCrypto.create_public_key_and_shares_centralized(
key_params, thresh_params)
# encrypt message using the public key
message = 'Some secret message to be encrypted!'
encrypted_message = ThresholdCrypto.encrypt_message(message, pub_key)
# build partial decryptions of three share owners using their shares
reconstruct_shares = [key_shares[i] for i in [0, 2, 4]]
partial_decryptions = [
ThresholdCrypto.compute_partial_decryption(encrypted_message, share)
for share in reconstruct_shares
]
# combine these partial decryptions to recover the message
decrypted_message = ThresholdCrypto.decrypt_message(partial_decryptions,
encrypted_message,
thresh_params, key_params)
print(decrypted_message)