diff --git a/kms/keys/src/keys/mod.rs b/kms/keys/src/keys/mod.rs index 4f2c56e33..0c53aab22 100644 --- a/kms/keys/src/keys/mod.rs +++ b/kms/keys/src/keys/mod.rs @@ -16,7 +16,7 @@ pub use crate::keys::{ }, zk::{ MAX_ZK_SIGNING_KEYS, PublicKey as ZkPublicKey, PublicKeys as ZkPublicKeys, - Signature as ZkSignature, UnsecuredZkKey, ZkKey, + Signature as ZkSignature, UnsecuredZkKey, ZkKey, public_inputs_from_pks, }, }; diff --git a/kms/keys/src/keys/zk/mod.rs b/kms/keys/src/keys/zk/mod.rs index d19dd88b1..be5f65cb6 100644 --- a/kms/keys/src/keys/zk/mod.rs +++ b/kms/keys/src/keys/zk/mod.rs @@ -13,7 +13,9 @@ use crate::keys::{errors::KeyError, secured_key::SecuredKey}; mod private; pub use self::private::SecretKey as UnsecuredZkKey; mod public; -pub use self::public::{MAX_ZK_SIGNING_KEYS, PublicKey, PublicKeys}; +pub use self::public::{ + MAX_ZK_SIGNING_KEYS, PublicKey, PublicKeys, inputs_from_pks as public_inputs_from_pks, +}; mod signature; pub use self::signature::Signature; diff --git a/kms/keys/src/keys/zk/public.rs b/kms/keys/src/keys/zk/public.rs index 46131c435..dfa15d88a 100644 --- a/kms/keys/src/keys/zk/public.rs +++ b/kms/keys/src/keys/zk/public.rs @@ -41,7 +41,7 @@ impl PublicKey { #[must_use] pub fn verify_multi(pks: &[Self], data: &Fr, signature: &Signature) -> bool { - let inputs = match try_from_pks((*data).into(), pks) { + let inputs = match inputs_from_pks((*data).into(), pks) { Ok(inputs) => inputs, Err(e) => { error!(target: LOG_TARGET, "Error building verifier inputs: {e:?}"); @@ -74,7 +74,10 @@ impl From for PublicKey { } } -fn try_from_pks(msg: Groth16Input, pks: &[PublicKey]) -> Result { +pub fn inputs_from_pks( + msg: Groth16Input, + pks: &[PublicKey], +) -> Result { if pks.len() > MAX_ZK_SIGNING_KEYS { return Err(ZkSignError::TooManyKeys(pks.len())); }