chore(kms-connector): update kms_grcp + config

Author: eudelins-zama

kms-connector/Cargo.lock

@@ -19,8 +19,8 @@ kms-worker.path = "crates/kms-worker"
 tx-sender.path = "crates/tx-sender"
 connector-utils.path = "crates/utils"
 fhevm_gateway_bindings = { git = "https://github.com/zama-ai/fhevm.git", rev = "47cdcf2fe54b684616abe3204b794f478b485c77", default-features = false }
-kms-grpc = { git = "https://github.com/zama-ai/kms.git", tag = "v0.11.1", default-features = true }
-bc2wrap = { git = "https://github.com/zama-ai/kms.git", tag = "v0.11.1", default-features = true }
+kms-grpc = { git = "https://github.com/zama-ai/kms.git", rev = "bc02d569cc0c346c60e4c11678b7708cb94c84be", default-features = true }
+bc2wrap = { git = "https://github.com/zama-ai/kms.git", rev = "bc02d569cc0c346c60e4c11678b7708cb94c84be", default-features = true }
 tfhe = "=1.3.3"
 
 #####################################################################

kms-connector/Cargo.toml

@@ -10,13 +10,13 @@ use connector_utils::{
     },
 };
 use kms_grpc::kms::v1::{
-    CrsGenResult, KeyGenPreprocResult, KeyGenResult, PublicDecryptionResponse,
+    CrsGenResult, KeyDigest, KeyGenPreprocResult, KeyGenResult, PublicDecryptionResponse,
     PublicDecryptionResponsePayload, RequestId, UserDecryptionResponse,
     UserDecryptionResponsePayload,
 };
 use kms_worker::core::{DbKmsResponsePublisher, KmsResponsePublisher};
 use sqlx::Row;
-use std::collections::HashMap;
+use std::str::FromStr;
 use tracing::info;
 
 #[tokio::test]
@@ -134,9 +134,16 @@ async fn test_publish_keygen_response() -> anyhow::Result<()> {
     let rand_key_id = rand_u256();
     let rand_prep_keygen_id = rand_u256();
     let rand_signature = rand_signature();
-    let mut rand_key_digests = HashMap::new();
-    rand_key_digests.insert(String::from("ServerKey"), rand_digest().to_vec());
-    rand_key_digests.insert(String::from("PublicKey"), rand_digest().to_vec());
+    let rand_key_digests = vec![
+        KeyDigest {
+            key_type: String::from("ServerKey"),
+            digest: rand_digest().to_vec(),
+        },
+        KeyDigest {
+            key_type: String::from("PublicKey"),
+            digest: rand_digest().to_vec(),
+        },
+    ];
 
     let grpc_response = KmsGrpcResponse::Keygen(KeyGenResult {
         request_id: Some(RequestId {
@@ -162,12 +169,12 @@ async fn test_publish_keygen_response() -> anyhow::Result<()> {
     let key_digests = row.try_get::<Vec<KeyDigestDbItem>, _>("key_digests")?;
     let signature = row.try_get::<Vec<u8>, _>("signature")?;
     assert_eq!(key_id, rand_key_id);
-    for kd in key_digests {
-        let key_type_str = match kd.key_type {
-            KeyType::Public => "PublicKey",
-            KeyType::Server => "ServerKey",
-        };
-        assert_eq!(Some(&kd.digest), rand_key_digests.get(key_type_str));
+    for (i, kd) in key_digests.iter().enumerate() {
+        assert_eq!(
+            kd.key_type,
+            KeyType::from_str(&rand_key_digests[i].key_type)?
+        );
+        assert_eq!(kd.digest, rand_key_digests[i].digest);
     }
     assert_eq!(signature, rand_signature);
     info!("Response successfully stored!");

kms-connector/crates/kms-worker/tests/response_publisher.rs

@@ -99,7 +99,7 @@ impl KmsInstance {
             .with_copy_to(
                 "/app/kms-core-client/config.toml".to_string(),
                 PathBuf::from_str(&format!(
-                    "{}/../../../test-suite/fhevm/config/core-client/config.toml",
+                    "{}/../../tests/data/core-client-config.toml",
                     env!("CARGO_MANIFEST_DIR"),
                 ))
                 .unwrap(),

kms-connector/crates/utils/src/tests/setup/kms.rs

@@ -237,10 +237,10 @@ impl KeygenResponse {
         let key_digests = grpc_response
             .key_digests
             .into_iter()
-            .map(|(k, d)| {
+            .map(|kd| {
                 Ok(KeyDigestDbItem {
-                    key_type: k.parse()?,
-                    digest: d,
+                    key_type: kd.key_type.parse()?,
+                    digest: kd.digest,
                 })
             })
             .collect::<anyhow::Result<Vec<KeyDigestDbItem>>>()?;

kms-connector/crates/utils/src/types/kms_response.rs

@@ -0,0 +1,12 @@
+kms_type = "centralized"
+fhe_params = "Default"
+decryption_mode = "NoiseFloodSmall"
+num_majority = 1
+num_reconstruct = 1
+
+[[cores]]
+party_id = 1
+address = "kms-core:50051"
+s3_endpoint = "http://minio:9000/kms-public"
+object_folder = "PUB"
+