Merge pull request #2636 from input-output-hk/jpraynaud/2622-fix-kes-pallas

fix(dmq): add missing KES period in 'DmqMsg'

Author: jpraynaud

Cargo.lock

@@ -1,7 +1,7 @@
 [package]
 name = "mithril-dmq"
 description = "Mechanisms to publish and consume messages of a 'Decentralized Message Queue network' through a DMQ node"
-version = "0.1.2"
+version = "0.1.3"
 authors.workspace = true
 documentation.workspace = true
 edition.workspace = true

internal/mithril-dmq/Cargo.toml

@@ -177,6 +177,7 @@ mod tests {
                     32, 32, 253, 186, 201, 177, 11, 117, 135, 187, 167, 181, 188, 22, 59, 206, 105,
                     231, 150, 215, 30, 78, 212, 76, 16, 252, 180, 72, 134, 137, 247, 161, 68,
                 ],
+                kes_period: 10,
             },
             DmqMsg {
                 msg_id: vec![1, 2],
@@ -195,6 +196,7 @@ mod tests {
                     240, 103, 245, 159, 147, 177, 110, 58, 248, 115, 58, 152, 138, 220, 35, 65,
                     245, 200,
                 ],
+                kes_period: 11,
             },
         ]
     }

internal/mithril-dmq/src/consumer/pallas.rs

@@ -46,6 +46,7 @@ impl DmqMessageBuilder {
             hasher.update(dmq_message.ttl.to_be_bytes());
             hasher.update(&dmq_message.kes_signature);
             hasher.update(&dmq_message.operational_certificate);
+            hasher.update(dmq_message.kes_period.to_be_bytes());
 
             hasher.finalize().to_vec()
         }
@@ -66,13 +67,20 @@ impl DmqMessageBuilder {
             .kes_signer
             .sign(message_bytes, block_number)
             .with_context(|| "Failed to KES sign message while building DMQ message")?;
+        let kes_period = self
+            .chain_observer
+            .get_current_kes_period(&operational_certificate)
+            .await
+            .with_context(|| "Failed to get KES period while building DMQ message")?
+            .unwrap_or_default();
         let mut dmq_message = DmqMsg {
             msg_id: vec![],
             msg_body: message_bytes.to_vec(),
             block_number,
             ttl: self.ttl_blocks,
             kes_signature: kes_signature.to_bytes_vec()?,
             operational_certificate: operational_certificate.to_bytes_vec()?,
+            kes_period,
         };
         dmq_message.msg_id = compute_msg_id(&dmq_message);
 
@@ -134,6 +142,7 @@ mod tests {
                 ttl: 100,
                 kes_signature: kes_signature.to_bytes_vec().unwrap(),
                 operational_certificate: operational_certificate.to_bytes_vec().unwrap(),
+                kes_period: 0,
             },
             DmqMsg {
                 msg_id: vec![],