Merge branch 'main' of gitlab.cryptoworkshop.com:root/bc-java

Author: dghgit

core/src/test/java/org/bouncycastle/pqc/crypto/test/FalconTest.java

@@ -141,34 +141,42 @@ public void testVectors()
             // System.out.println("testing successful!");
         }
     }
-    
-    public void testFalconRandom()
-    {
-        byte[] msg = Strings.toByteArray("Hello World!");
-        FalconKeyPairGenerator keyGen = new FalconKeyPairGenerator();
 
+    public void testRandom()
+        throws Exception
+    {
         SecureRandom random = new SecureRandom();
+        byte[] msg = Strings.toByteArray("Hello World!");
 
+        FalconKeyPairGenerator keyGen = new FalconKeyPairGenerator();
         keyGen.init(new FalconKeyGenerationParameters(random, FalconParameters.falcon_512));
 
-        for (int i = 0; i != 100; i++)
+        for (int i = 0; i < 10; ++i)
         {
             AsymmetricCipherKeyPair keyPair = keyGen.generateKeyPair();
 
-            // sign
-            FalconSigner signer = new FalconSigner();
-            FalconPrivateKeyParameters skparam = (FalconPrivateKeyParameters)keyPair.getPrivate();
-            ParametersWithRandom skwrand = new ParametersWithRandom(skparam, random);
-            signer.init(true, skwrand);
-
-            byte[] sigGenerated = signer.generateSignature(msg);
+            FalconPrivateKeyParameters privParams = (FalconPrivateKeyParameters)keyPair.getPrivate();
+            FalconPublicKeyParameters pubParams = (FalconPublicKeyParameters)keyPair.getPublic();
 
-            // verify
-            FalconSigner verifier = new FalconSigner();
-            FalconPublicKeyParameters pkparam = (FalconPublicKeyParameters)keyPair.getPublic();
-            verifier.init(false, pkparam);
+            privParams = (FalconPrivateKeyParameters)PrivateKeyFactory.createKey(
+                PrivateKeyInfoFactory.createPrivateKeyInfo(privParams));
+            pubParams = (FalconPublicKeyParameters)PublicKeyFactory.createKey(
+                SubjectPublicKeyInfoFactory.createSubjectPublicKeyInfo(pubParams));
+            
+            for (int j = 0; j < 10; ++j)
+            {
+                // sign
+                FalconSigner signer = new FalconSigner();
+                signer.init(true, new ParametersWithRandom(privParams, random));
+                byte[] signature = signer.generateSignature(msg);
+    
+                // verify
+                FalconSigner verifier = new FalconSigner();
+                verifier.init(false, pubParams);
+                boolean verified = verifier.verifySignature(msg, signature);
 
-            assertTrue("count = " + i, verifier.verifySignature(msg, sigGenerated));
+                assertTrue("count = " + i, verified);
+            }
         }
     }
 }