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)