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_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)
def test_message_encryption(self):
em = ThresholdCrypto.encrypt_message('Some secret message', self.pk)
self.assertTrue(em.c >= 0)
self.assertTrue(em.v >= 0)
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)
def elgamal_enc_threshold(pk, message):
encrypted = ThresholdCrypto.encrypt_message(pk, message)
cipher = ThresholdCipher(encrypted.v, encrypted.c, encrypted.enc)
return cipher