zk-token-sdk: implement Display for ciphertexts (solana-labs#26755)

CriesofCarrots · Jul 23, 2022 · 700da0f · 700da0f
1 parent 3396d4e
commit 700da0f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/zk-token-sdk/src/encryption/auth_encryption.rs b/zk-token-sdk/src/encryption/auth_encryption.rs
@@ -15,7 +15,7 @@ use {
         signature::Signature,
         signer::{Signer, SignerError},
     },
-    std::convert::TryInto,
+    std::{convert::TryInto, fmt},
     zeroize::Zeroize,
 };
 
@@ -129,6 +129,12 @@ impl AeCiphertext {
     }
 }
 
+impl fmt::Display for AeCiphertext {
+    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+        write!(f, "{}", base64::encode(self.to_bytes()))
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use {

diff --git a/zk-token-sdk/src/encryption/elgamal.rs b/zk-token-sdk/src/encryption/elgamal.rs
@@ -457,6 +457,12 @@ impl ElGamalCiphertext {
     }
 }
 
+impl fmt::Display for ElGamalCiphertext {
+    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+        write!(f, "{}", base64::encode(self.to_bytes()))
+    }
+}
+
 impl<'a, 'b> Add<&'b ElGamalCiphertext> for &'a ElGamalCiphertext {
     type Output = ElGamalCiphertext;