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)