diff --git a/crypto/test/src/pqc/crypto/test/CrystalsDilithiumTest.cs b/crypto/test/src/pqc/crypto/test/CrystalsDilithiumTest.cs
new file mode 100644
index 000000000..1fddfb565
--- /dev/null
+++ b/crypto/test/src/pqc/crypto/test/CrystalsDilithiumTest.cs
@@ -0,0 +1,144 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+
+using NUnit.Framework;
+
+using Org.BouncyCastle.Crypto;
+using Org.BouncyCastle.Pqc.Crypto.Crystals.Dilithium;
+using Org.BouncyCastle.Pqc.Crypto.Utilities;
+using Org.BouncyCastle.Utilities;
+using Org.BouncyCastle.Utilities.Encoders;
+using Org.BouncyCastle.Utilities.Test;
+
+namespace Org.BouncyCastle.Pqc.Crypto.Tests
+{
+ [TestFixture]
+ public class CrystalsDilithiumTest
+ {
+ private static readonly Dictionary<string, DilithiumParameters> parameters = new Dictionary<string, DilithiumParameters>()
+ {
+ { "PQCsignKAT_Dilithium2.rsp", DilithiumParameters.Dilithium2 },
+ { "PQCsignKAT_Dilithium3.rsp", DilithiumParameters.Dilithium3 },
+ { "PQCsignKAT_Dilithium5.rsp", DilithiumParameters.Dilithium5 }
+ };
+
+ private static readonly string[] TestVectorFilesBasic =
+ {
+ "PQCsignKAT_Dilithium2.rsp",
+ "PQCsignKAT_Dilithium3.rsp",
+ "PQCsignKAT_Dilithium5.rsp"
+ };
+
+ [TestCaseSource(nameof(TestVectorFilesBasic))]
+ [Parallelizable(ParallelScope.All)]
+ public void TestVectorsBasic(string testVectorFile)
+ {
+ RunTestVectorFile(testVectorFile);
+ }
+
+ private static void TestVectors(string name, IDictionary<string, string> buf)
+ {
+ string count = buf["count"];
+ byte[] seed = Hex.Decode(buf["seed"]); // seed for Dilithium secure random
+ int mlen = int.Parse(buf["mlen"]); // message length
+ byte[] msg = Hex.Decode(buf["msg"]); // message
+ byte[] pk = Hex.Decode(buf["pk"]); // public key
+ byte[] sk = Hex.Decode(buf["sk"]); // private key
+ int smlen = int.Parse(buf["smlen"]); // signature length
+ byte[] sigExpected = Hex.Decode(buf["sm"]); // signature
+
+ NistSecureRandom random = new NistSecureRandom(seed, null);
+ DilithiumParameters dilithiumparameters = parameters[name];
+
+ DilithiumKeyPairGenerator kpGen = new DilithiumKeyPairGenerator();
+ DilithiumKeyGenerationParameters genParams = new DilithiumKeyGenerationParameters(random, dilithiumparameters);
+
+ //
+ // Generate keys and test.
+ //
+ kpGen.Init(genParams);
+ AsymmetricCipherKeyPair kp = kpGen.GenerateKeyPair();
+
+
+ DilithiumPublicKeyParameters pubParams = (DilithiumPublicKeyParameters) kp.Public;
+ DilithiumPrivateKeyParameters privParams = (DilithiumPrivateKeyParameters) kp.Private;
+
+ //Console.WriteLine(string.Format("{0} Expected pk = {1}", pk.Length, Convert.ToHexString(pk)));
+ //Console.WriteLine(String.Format("{0} Actual Public key = {1}", pubParams.GetEncoded().Length, Convert.ToHexString(pubParams.GetEncoded())));
+
+ Assert.True(Arrays.AreEqual(pk, pubParams.GetEncoded()), name + " " + count + ": public key");
+ Assert.True(Arrays.AreEqual(sk, privParams.GetEncoded()), name + " " + count + ": secret key");
+
+ //
+ // Signature test
+ //
+ DilithiumSigner signer = new DilithiumSigner(random);
+
+ signer.Init(true, privParams);
+ byte[] sigGenerated = signer.GenerateSignature(msg);
+ byte[] attachedSig = Arrays.ConcatenateAll(sigGenerated, msg);
+
+
+ //Console.WriteLine(string.Format("{0} Expected sig = {1}", sigExpected.Length, Convert.ToHexString(sigExpected)));
+ //Console.WriteLine(String.Format("{0} Actual Signature = {1}", attachedSig.Length, Convert.ToHexString(attachedSig)));
+
+ Assert.True(smlen == attachedSig.Length, name + " " + count + ": signature length");
+
+ byte[] msg1 = new byte[msg.Length];
+
+ signer.Init(false, pubParams);
+ Assert.True(signer.VerifySignature(msg1, attachedSig), (name + " " + count + ": signature verify"));
+
+ Assert.True(Arrays.AreEqual(msg, msg1), name + " " + count + ": signature message verify");
+
+ Assert.True(Arrays.AreEqual(sigExpected, attachedSig), name + " " + count + ": signature gen match");
+ }
+
+ private static byte[] UInt32_To_LE(uint n)
+ {
+ byte[] bs = new byte[4];
+ bs[0] = (byte)(n);
+ bs[1] = (byte)(n >> 8);
+ bs[2] = (byte)(n >> 16);
+ bs[3] = (byte)(n >> 24);
+ return bs;
+ }
+ public static void RunTestVectorFile(string name)
+ {
+ var buf = new Dictionary<string, string>();
+
+ using (var src = new StreamReader(SimpleTest.GetTestDataAsStream("pqc.crystals.dilithium." + name)))
+ {
+ string line;
+ while ((line = src.ReadLine()) != null)
+ {
+ line = line.Trim();
+ if (line.StartsWith("#"))
+ continue;
+
+ if (line.Length > 0)
+ {
+ int a = line.IndexOf('=');
+ if (a > -1)
+ {
+ buf[line.Substring(0, a).Trim()] = line.Substring(a + 1).Trim();
+ }
+ continue;
+ }
+
+ if (buf.Count > 0)
+ {
+ TestVectors(name, buf);
+ buf.Clear();
+ }
+ }
+
+ if (buf.Count > 0)
+ {
+ TestVectors(name, buf);
+ }
+ }
+ }
+ }
+}
|
