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

Author: jpraynaud

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

@@ -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/model/builder.rs

@@ -48,6 +48,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()
         }
@@ -68,13 +69,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);
 
@@ -136,6 +144,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![],