ml-dsa 0.0.4

Pure Rust implementation of ML-DSA (formerly known as CRYSTALS-Dilithium) as described in FIPS-204 (final)
Documentation 
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
56

use ml_dsa::{
    KeyGen, KeyPair, MlDsa44, MlDsa65, MlDsa87, Signature,
    signature::{Signer, Verifier},
};
use proptest::prelude::*;

/// Example message
const MSG: &[u8] = b"Hello world";

// Keypairs
prop_compose! {
    fn mldsa44_keypair()(seed_bytes in any::<[u8; 32]>()) -> KeyPair<MlDsa44> {
       MlDsa44::key_gen_internal(seed_bytes.as_ref())
    }
}
prop_compose! {
    fn mldsa65_keypair()(seed_bytes in any::<[u8; 32]>()) -> KeyPair<MlDsa65> {
       MlDsa65::key_gen_internal(seed_bytes.as_ref())
    }
}
prop_compose! {
    fn mldsa87_keypair()(seed_bytes in any::<[u8; 32]>()) -> KeyPair<MlDsa87> {
        MlDsa87::key_gen_internal(seed_bytes.as_ref())
    }
}

macro_rules! round_trip_test {
    ($params:path, $keypair:expr) => {
        let sig = $keypair.signing_key().sign(MSG);

        // Check signature verification
        let verify_result = $keypair.verifying_key().verify(MSG, &sig);
        prop_assert!(verify_result.is_ok());

        // Check signature encoding round trip
        let sig_decoded = Signature::<$params>::decode(&sig.encode());
        prop_assert_eq!(Some(sig), sig_decoded);
    };
}

proptest! {
    #[test]
    fn mldsa44_round_trip(keypair in mldsa44_keypair()) {
        round_trip_test!(MlDsa44, keypair);
    }

    #[test]
    fn mldsa65_round_trip(keypair in mldsa65_keypair()) {
        round_trip_test!(MlDsa65, keypair);
    }

    #[test]
    fn mldsa87_round_trip(keypair in mldsa87_keypair()) {
        round_trip_test!(MlDsa87, keypair);
    }
}