mithril-common 0.7.14

Common types, interfaces, and utilities for Mithril nodes.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

use std::{fs, path::PathBuf};

use kes_summed_ed25519::kes::Sum6Kes;
use kes_summed_ed25519::traits::KesSk;

use crate::crypto_helper::{ColdKeyGenerator, KesPeriod, OpCert, ProtocolPartyId, Sum6KesBytes};
use crate::test::crypto_helper::SerDeShelleyFileFormatTestExtension;

/// A type alias for the party index used in KES cryptographic material.
pub type KesPartyIndexForTest = u64;

/// A struct to hold KES cryptographic material for testing purposes.
pub(crate) struct KesCryptographicMaterialForTest {
    #[allow(dead_code)]
    pub party_id: ProtocolPartyId,
    pub operational_certificate_file: PathBuf,
    pub kes_secret_key_file: PathBuf,
}

/// Create KES cryptographic material for testing purposes.
pub fn create_kes_cryptographic_material(
    party_idx: KesPartyIndexForTest,
    start_kes_period: KesPeriod,
    test_directory: &str,
) -> KesCryptographicMaterialForTest {
    let temp_dir = std::env::temp_dir()
        .join("create_kes_cryptographic_material")
        .join(format!("{test_directory}_{party_idx}"));
    fs::create_dir_all(&temp_dir).expect("temp dir creation should not fail");
    let keypair = ColdKeyGenerator::create_deterministic_keypair([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, start_kes_period, keypair);
    let kes_secret_key_file = temp_dir.join(format!("kes{party_idx}.skey"));
    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"));
    operational_certificate
        .to_file(&operational_certificate_file)
        .expect("operational certificate file export should not fail");
    let party_id = operational_certificate
        .compute_protocol_party_id()
        .expect("compute protocol party id should not fail");

    KesCryptographicMaterialForTest {
        party_id,
        operational_certificate_file,
        kes_secret_key_file,
    }
}