def test_static_512_key_parameters(self):
kp = ThresholdCrypto.static_512_key_parameters()
self.assertEqual(kp.p, 2 * kp.q + 1) # safe prime
self.assertEqual(pow(kp.g, kp.q, kp.p),
1) # g generates q order subgroup
self.assertNotEqual(pow(kp.g, 2, kp.p), 1)
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)