New StmInitializer setup

input-output-hk · Sep 23, 2022 · a6467c7 · a6467c7
1 parent d9d10bd
commit a6467c7
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 5 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/mithril-core/Cargo.toml b/mithril-core/Cargo.toml
@@ -6,15 +6,16 @@ edition = "2018"
 [dependencies]
 blake2      = "0.10.4"
 blst = { version = "0.3.10" }
-num-bigint  = { version = "0.4.0", optional = true }
-num-rational = { version = "0.4.0", optional = true }
-num-traits  = { version = "0.2.14", optional = true }
+digest      = { version = "0.9.0", features = ["alloc"] }
+ed25519-dalek = "1.0.1"
+kes-summed-ed25519 = { git = "https://github.com/input-output-hk/kes" }
+num-bigint  = {version = "0.4.0", optional = true }
+num-rational = {version = "0.4.0", optional = true }
+num-traits  = {version = "0.2.14", optional = true }
 rand_core   = "0.6.3"
 rug = { version = "1.14", optional = true }
 serde = { version = "1", features = ["rc", "derive"] }
 thiserror = "1.0"
-ed25519-dalek = "1.0.1"
-kes-summed-ed25519 = { git = "https://github.com/input-output-hk/kes" }
 
 [dev-dependencies]
 bincode = "1.3.3"

diff --git a/mithril-core/src/stm.rs b/mithril-core/src/stm.rs
@@ -117,6 +117,8 @@ use serde::{Deserialize, Serialize};
 use std::collections::{BTreeMap, HashMap, HashSet};
 use std::convert::{From, TryFrom, TryInto};
 use std::hash::{Hash, Hasher};
+use kes_summed_ed25519::kes::{Sum6Kes, Sum6KesSig};
+use kes_summed_ed25519::traits::KesSk;
 
 /// The quantity of stake held by a party, represented as a `u64`.
 pub type Stake = u64;
@@ -161,6 +163,8 @@ pub struct StmInitializer {
     pub(crate) sk: SigningKey,
     /// Verification (public) key + proof of possession.
     pub(crate) pk: StmVerificationKeyPoP,
+    /// KES signature of `pk`
+    pub(crate) kes_sig: Option<Sum6KesSig>,
 }
 
 /// Participant in the protocol can sign messages.
@@ -261,6 +265,23 @@ impl StmParameters {
 }
 
 impl StmInitializer {
+    /// Builds an `StmInitializer` that is ready to register with the key registration service.
+    /// This function generates the signing and verification key with a PoP, signs the verification
+    /// key with a provided KES signing key, and initialises the structure.
+    pub fn setup_new<R: RngCore + CryptoRng>(params: StmParameters, kes_sk: &[u8], kes_period: usize, stake: Stake, rng: &mut R) -> Self {
+        let sk = SigningKey::gen(rng);
+        let pk = StmVerificationKeyPoP::from(&sk);
+        let kes_sk = Sum6Kes::from_bytes(kes_sk).expect("Invalid KES key provided"); // todo: handle this
+        let kes_sig = Some(kes_sk.sign(kes_period, &pk.to_bytes()));
+
+        Self {
+            stake,
+            params,
+            sk,
+            pk,
+            kes_sig
+        }
+    }
     /// Builds an `StmInitializer` that is ready to register with the key registration service.
     /// This function generates the signing and verification key with a PoP, and initialises the structure.
     pub fn setup<R: RngCore + CryptoRng>(params: StmParameters, stake: Stake, rng: &mut R) -> Self {
@@ -271,6 +292,7 @@ impl StmInitializer {
             params,
             sk,
             pk,
+            kes_sig: None
         }
     }
 
@@ -347,6 +369,7 @@ impl StmInitializer {
             params,
             sk,
             pk,
+            kes_sig: None
         })
     }
 }
@@ -523,6 +546,7 @@ impl<D: Clone + Digest> StmSigner<D> {
             params: self.params,
             pk: StmVerificationKeyPoP::from(&self.sk),
             sk: self.sk,
+            kes_sig: None
         }
     }