Update KES

Author: iquerejeta

Cargo.lock

@@ -27,7 +27,7 @@ glob = "0.3"
 hex = "0.4.3"
 http = "0.2.6"
 jsonschema = "0.16.0"
-kes-summed-ed25519 = { version = "0.1.1", features = ["serde_enabled"] }
+kes-summed-ed25519 = { version = "0.2.0", features = ["serde_enabled", "sk_clone_enabled"]}
 lazy_static = "1.4.0"
 mockall = "0.11.0"
 nom = "7.1"
@@ -39,6 +39,7 @@ serde = { version = "1.0", features = ["derive"] }
 serde_bytes = "0.11.7"
 serde_cbor = "0.11.2"
 serde_json = "1.0"
+serde_with = "2.2.0"
 serde_yaml = "0.9.10"
 sha2 = "0.10.2"
 slog = { version = "2.7.0", features = ["max_level_trace", "release_max_level_debug"] }

mithril-common/Cargo.toml

@@ -635,7 +635,9 @@ pool1qz2vzszautc2c8mljnqre2857dpmheq7kgt6vav0s38tvvhxm6w   1.051e-6
     #[tokio::test]
     async fn test_get_current_kes_period() {
         let keypair = ColdKeyGenerator::create_deterministic_keypair([0u8; 32]);
-        let (_, kes_verification_key) = Sum6Kes::keygen(&mut [0u8; 32]);
+        let mut dummy_key_buffer = [0u8; Sum6Kes::SIZE + 4];
+        let mut dummy_seed = [0u8; 32];
+        let (_, kes_verification_key) = Sum6Kes::keygen(&mut dummy_key_buffer, &mut dummy_seed);
         let operational_certificate = OpCert::new(kes_verification_key, 0, 0, keypair);
         let observer = CardanoCliChainObserver::new(Box::new(TestCliRunner {}));
         let kes_period = observer

mithril-common/src/chain_observer/cli_observer.rs

@@ -13,13 +13,24 @@
 
 use hex::FromHex;
 use kes_summed_ed25519::kes::Sum6Kes;
+use kes_summed_ed25519::traits::KesSk;
 use serde::de::DeserializeOwned;
 use serde::{Deserialize, Serialize};
+use serde_with::{As, Bytes};
 use std::fs;
 use std::io::Write;
 use std::path::Path;
 use thiserror::Error;
 
+/// We need to create this struct because the design of Sum6Kes takes
+/// a reference to a mutable pointer. It is therefore not possible to
+/// implement Ser/Deser using serde.
+// We need this helper structure, because we are currently getting the key
+// from a file, instead of directly consuming a buffer.
+// todo: create the KES key directly from a buffer instead of deserialising from disk
+#[derive(Clone, Serialize, Deserialize)]
+pub struct Sum6KesBytes(#[serde(with = "As::<Bytes>")] pub [u8; 612]);
+
 /// Parse error
 #[derive(Error, Debug)]
 pub enum ParseError {
@@ -85,7 +96,7 @@ pub trait SerDeShelleyFileFormat: Serialize + DeserializeOwned {
     }
 }
 
-impl SerDeShelleyFileFormat for Sum6Kes {
+impl SerDeShelleyFileFormat for Sum6KesBytes {
     const TYPE: &'static str = "KesSigningKey_ed25519_kes_2^6";
     const DESCRIPTION: &'static str = "KES Signing Key";
 
@@ -110,6 +121,12 @@ impl SerDeShelleyFileFormat for Sum6Kes {
     }
 }
 
+impl<'a> From<&'a mut Sum6KesBytes> for Sum6Kes<'a> {
+    fn from(value: &'a mut Sum6KesBytes) -> Self {
+        Self::from_bytes(&mut value.0).expect("Invalid data format")
+    }
+}
+
 #[cfg(all(test))]
 mod test {
     use super::*;
@@ -122,8 +139,8 @@ mod test {
         let cbor_string = "590260fe77acdfa56281e4b05198f5136018057a65f425411f0990cac4aca0f2917aa00a3d51e191f6f425d870aca3c6a2a41833621f5729d7bc0e3dfc3ae77d057e5e1253b71def7a54157b9f98973ca3c49edd9f311e5f4b23ac268b56a6ac040c14c6d2217925492e42f00dc89a2a01ff363571df0ca0db5ba37001cee56790cc01cd69c6aa760fca55a65a110305ea3c11da0a27be345a589329a584ebfc499c43c55e8c6db5d9c0b014692533ee78abd7ac1e79f7ec9335c7551d31668369b4d5111db78072f010043e35e5ca7f11acc3c05b26b9c7fe56f02aa41544f00cb7685e87f34c73b617260ade3c7b8d8c4df46693694998f85ad80d2cbab0b575b6ccd65d90574e84368169578bff57f751bc94f7eec5c0d7055ec88891a69545eedbfbd3c5f1b1c1fe09c14099f6b052aa215efdc5cb6cdc84aa810db41dbe8cb7d28f7c4beb75cc53915d3ac75fc9d0bf1c734a46e401e15150c147d013a938b7e07cc4f25a582b914e94783d15896530409b8acbe31ef471de8a1988ac78dfb7510729eff008084885f07df870b65e4f382ca15908e1dcda77384b5c724350de90cec22b1dcbb1cdaed88da08bb4772a82266ec154f5887f89860d0920dba705c45957ef6d93e42f6c9509c966277d368dd0eefa67c8147aa15d40a222f7953a4f34616500b310d00aa1b5b73eb237dc4f76c0c16813d321b2fc5ac97039be25b22509d1201d61f4ccc11cd4ff40fffe39f0e937b4722074d8e073a775d7283b715d46f79ce128e3f1362f35615fa72364d20b6db841193d96e58d9d8e86b516bbd1f05e45b39823a93f6e9f29d9e01acf2c12c072d1c64e0afbbabf6903ef542e".to_string();
 
         let file_format = ShelleyFileFormat {
-            file_type: Sum6Kes::TYPE.to_string(),
-            description: Sum6Kes::DESCRIPTION.to_string(),
+            file_type: Sum6KesBytes::TYPE.to_string(),
+            description: Sum6KesBytes::DESCRIPTION.to_string(),
             cbor_hex: cbor_string,
         };
 
@@ -134,8 +151,9 @@ mod test {
 
         write!(file, "{json_str}").expect("Unexpected error writing to file.");
 
-        let kes_sk = Sum6Kes::from_file(&sk_dir);
+        let mut kes_sk_bytes =
+            Sum6KesBytes::from_file(&sk_dir).expect("Failure parsing Shelley file format.");
 
-        assert!(kes_sk.is_ok(), "Failure parsing Shelley file format.");
+        let _kes_sk = Sum6Kes::from(&mut kes_sk_bytes); // Panics if data is incorrect
     }
 }

mithril-common/src/crypto_helper/cardano/codec.rs

@@ -15,6 +15,7 @@ use mithril_stm::key_reg::{ClosedKeyReg, KeyReg};
 use mithril_stm::stm::{Stake, StmInitializer, StmParameters, StmSigner, StmVerificationKeyPoP};
 use mithril_stm::RegisterError;
 
+use crate::crypto_helper::cardano::Sum6KesBytes;
 use blake2::{
     digest::{consts::U32, FixedOutput},
     Blake2b, Digest,
@@ -123,8 +124,8 @@ impl StmInitializerWrapper {
     ) -> Result<Self, ProtocolInitializerErrorWrapper> {
         let stm_initializer = StmInitializer::setup(params, stake, rng);
         let kes_signature = if let Some(kes_sk_path) = kes_sk_path {
-            let mut kes_sk: Sum6Kes = Sum6Kes::from_file(kes_sk_path)?;
-
+            let mut kes_sk_bytes = Sum6KesBytes::from_file(kes_sk_path)?;
+            let mut kes_sk = Sum6Kes::from(&mut kes_sk_bytes);
             let kes_sk_period = kes_sk.get_period();
             let provided_period = kes_period.unwrap_or_default();
             if kes_sk_period > provided_period {
@@ -308,10 +309,15 @@ mod test {
     fn create_cryptographic_material(party_idx: u64) -> (ProtocolPartyId, PathBuf, PathBuf) {
         let temp_dir = setup_temp_directory();
         let keypair = ColdKeyGenerator::create_deterministic_keypair([party_idx as u8; 32]);
-        let (kes_secret_key, kes_verification_key) = Sum6Kes::keygen(&mut [party_idx as u8; 32]);
+        let mut dummy_buffer = [0u8; Sum6Kes::SIZE + 4];
+        let mut dummy_seed = [party_idx as u8; 32];
+        let (kes_secret_key, kes_verification_key) =
+            Sum6Kes::keygen(&mut dummy_buffer, &mut dummy_seed);
+        let mut kes_bytes = Sum6KesBytes([0u8; Sum6Kes::SIZE + 4]);
+        kes_bytes.0.copy_from_slice(&kes_secret_key.clone_sk());
         let operational_certificate = OpCert::new(kes_verification_key, 0, 0, keypair);
         let kes_secret_key_file = temp_dir.join(format!("kes{party_idx}.skey"));
-        kes_secret_key
+        kes_bytes
             .to_file(&kes_secret_key_file)
             .expect("KES secret key file export should not fail");
         let operational_certificate_file = temp_dir.join(format!("pool{party_idx}.cert"));

mithril-common/src/crypto_helper/cardano/key_certification.rs

@@ -187,7 +187,9 @@ mod tests {
     fn test_vector_opcert() {
         let temp_dir = setup_temp_directory();
         let keypair = ColdKeyGenerator::create_deterministic_keypair([0u8; 32]);
-        let (_, kes_verification_key) = Sum6Kes::keygen(&mut [0u8; 32]);
+        let mut dummy_key_buffer = [0u8; Sum6Kes::SIZE + 4];
+        let mut dummy_seed = [0u8; 32];
+        let (_, kes_verification_key) = Sum6Kes::keygen(&mut dummy_key_buffer, &mut dummy_seed);
         let operational_certificate = OpCert::new(kes_verification_key, 0, 0, keypair);
         assert!(operational_certificate.validate().is_ok());
 

mithril-common/src/crypto_helper/cardano/opcert.rs

@@ -11,7 +11,7 @@ mod types;
 
 #[cfg(any(test, feature = "test_only"))]
 pub use cardano::ColdKeyGenerator;
-pub use cardano::{KESPeriod, OpCert, SerDeShelleyFileFormat};
+pub use cardano::{KESPeriod, OpCert, SerDeShelleyFileFormat, Sum6KesBytes};
 pub use codec::*;
 pub use era::{
     EraMarkersSigner, EraMarkersVerifier, EraMarkersVerifierError, EraMarkersVerifierSecretKey,

mithril-common/src/crypto_helper/mod.rs

@@ -5,7 +5,7 @@ use rand_core::{RngCore, SeedableRng};
 use crate::{
     crypto_helper::{
         tests_setup, tests_setup::setup_temp_directory_for_signer, ColdKeyGenerator, OpCert,
-        ProtocolStakeDistribution, SerDeShelleyFileFormat,
+        ProtocolStakeDistribution, SerDeShelleyFileFormat, Sum6KesBytes,
     },
     entities::{PartyId, ProtocolParameters, StakeDistribution},
     test_utils::{fake_data, mithril_fixture::MithrilFixture},
@@ -121,15 +121,18 @@ fn build_party_with_operational_certificate(
     kes_key_seed: &mut [u8],
 ) -> PartyId {
     let keypair = ColdKeyGenerator::create_deterministic_keypair([party_index as u8; 32]);
-    let (kes_secret_key, kes_verification_key) = Sum6Kes::keygen(kes_key_seed);
+    let mut dummy_buffer = [0u8; Sum6Kes::SIZE + 4];
+    let (kes_secret_key, kes_verification_key) = Sum6Kes::keygen(&mut dummy_buffer, kes_key_seed);
+    let mut kes_bytes = Sum6KesBytes([0u8; Sum6Kes::SIZE + 4]);
+    kes_bytes.0.copy_from_slice(&kes_secret_key.clone_sk());
     let operational_certificate = OpCert::new(kes_verification_key, 0, 0, keypair);
     let party_id = operational_certificate
         .compute_protocol_party_id()
         .expect("compute protocol party id should not fail");
     let temp_dir = setup_temp_directory_for_signer(&party_id, true)
         .expect("setup temp directory should return a value");
     if !temp_dir.join("kes.sk").exists() {
-        kes_secret_key
+        kes_bytes
             .to_file(temp_dir.join("kes.sk"))
             .expect("KES secret key file export should not fail");
     }