Add SunRsaSign to RSA benchmark test

wolfSSL · Jan 31, 2025 · 5020c12 · 5020c12
1 parent 8fd2674
commit 5020c12
Showing 1 changed file with 13 additions and 12 deletions.
diff --git a/examples/provider/CryptoBenchmark.java b/examples/provider/CryptoBenchmark.java
@@ -289,8 +289,14 @@ private static void runRSABenchmark(String providerName, int keySize) throws Exc
         String cipherMode = "RSA/ECB/PKCS1Padding";
 
         /* Initialize key generator and cipher */
-        keyGen = KeyPairGenerator.getInstance("RSA", providerName);
-        cipher = Cipher.getInstance(cipherMode, providerName);
+        if (providerName.equals("SunJCE")) {
+            keyGen = KeyPairGenerator.getInstance("RSA", "SunRsaSign");
+            cipher = Cipher.getInstance(cipherMode, "SunJCE");
+            providerName = "JCE/Sign";
+        } else {
+            keyGen = KeyPairGenerator.getInstance("RSA", providerName);
+            cipher = Cipher.getInstance(cipherMode, providerName);
+        }
         testData = generateTestData(SMALL_MESSAGE_SIZE);
 
         /* Key Generation benchmark */
@@ -407,18 +413,13 @@ public static void main(String[] args) {
             System.out.println("-----------------------------------------------------------------------------");
 
             for (Provider provider : providers) {
-                if (!provider.getName().equals("SunJCE")) {
-                    Security.insertProviderAt(provider, 1);
-                    System.out.println("\n" + provider.getName() + ":");
-
-                    for (int keySize : RSA_KEY_SIZES) {
-                        runRSABenchmark(provider.getName(), keySize);
-                    }
-
-                    Security.removeProvider(provider.getName());
+                Security.insertProviderAt(provider, 1);
+                System.out.println("\n" + (provider.getName().equals("SunJCE") ? "SunJCE / SunRsaSign" : provider.getName()) + ":");
+                for (int keySize : RSA_KEY_SIZES) {
+                    runRSABenchmark(provider.getName(), keySize);
                 }
+                Security.removeProvider(provider.getName());
             }
-
             System.out.println("-----------------------------------------------------------------------------");
 
             /* Print delta table */