solana_zk_token_sdk/zk_token_elgamal/
decryption.rs

1#[cfg(not(target_os = "solana"))]
2use crate::{
3    encryption::elgamal::{ElGamalCiphertext, ElGamalSecretKey},
4    zk_token_elgamal::pod,
5};
6
7#[cfg(not(target_os = "solana"))]
8impl pod::ElGamalCiphertext {
9    pub fn decrypt(self, secret_key: &ElGamalSecretKey) -> Option<u64> {
10        let deserialized_ciphertext: Option<ElGamalCiphertext> = self.try_into().ok();
11        if let Some(ciphertext) = deserialized_ciphertext {
12            ciphertext.decrypt_u32(secret_key)
13        } else {
14            None
15        }
16    }
17}
18
19#[cfg(test)]
20mod tests {
21    use {super::*, crate::encryption::elgamal::ElGamalKeypair};
22
23    #[test]
24    fn test_pod_decryption() {
25        let keypair = ElGamalKeypair::new_rand();
26
27        let pod_ciphertext = pod::ElGamalCiphertext([0u8; 64]);
28        assert_eq!(pod_ciphertext.decrypt(keypair.secret()).unwrap(), 0);
29
30        let amount = 55_u64;
31        let ciphertext = keypair.pubkey().encrypt(amount);
32        let pod_ciphertext: pod::ElGamalCiphertext = ciphertext.into();
33        assert_eq!(pod_ciphertext.decrypt(keypair.secret()).unwrap(), 55);
34    }
35}