Fix #385: Add a benchmark for generating random bytes

wultra · Nov 30, 2023 · abe22cd · abe22cd
1 parent 8db4fda
commit abe22cd
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/...ava-cmd-lib/src/main/java/io/getlime/security/powerauth/lib/cmd/consts/PowerAuthStep.java b/...ava-cmd-lib/src/main/java/io/getlime/security/powerauth/lib/cmd/consts/PowerAuthStep.java
@@ -63,6 +63,11 @@ public enum PowerAuthStep {
      */
     ENCRYPT("encrypt", "Encrypt Request", "encrypt"),
 
+    /**
+     * Run benchmark for generating random bytes
+     */
+    GENERATE_BENCHMARK("generate-benchmark", "Benchmark for generating random bytes", "generate-benchmark"),
+
     /**
      * Confirming an activation recovery
      */

diff --git a/powerauth-java-cmd/src/main/java/io/getlime/security/powerauth/app/cmd/Application.java b/powerauth-java-cmd/src/main/java/io/getlime/security/powerauth/app/cmd/Application.java
@@ -18,6 +18,7 @@
 
 import io.getlime.security.powerauth.app.cmd.exception.ExecutionException;
 import io.getlime.security.powerauth.crypto.lib.enums.PowerAuthSignatureTypes;
+import io.getlime.security.powerauth.crypto.lib.generator.KeyGenerator;
 import io.getlime.security.powerauth.lib.cmd.CmdLibApplication;
 import io.getlime.security.powerauth.lib.cmd.consts.PowerAuthStep;
 import io.getlime.security.powerauth.lib.cmd.consts.PowerAuthVersion;
@@ -57,6 +58,9 @@
  */
 public class Application {
 
+    private static final long GENERATE_RANDOM_BENCHMARK_ROUNDS = 10_000L;
+    private static final KeyGenerator KEY_GENERATOR = new KeyGenerator();
+
     /**
      * Application main
      * @param args Arguments, use --help to print expected arguments
@@ -107,6 +111,7 @@ public static void main(String[] args) {
             options.addOption("P", "platform", true, "User device platform.");
             options.addOption("D", "device-info", true, "Information about user device.");
             options.addOption("q", "qr-code-data", true, "Data for offline signature encoded in QR code.");
+            options.addOption("gr", "generate-rounds", true, "Number of rounds for generate random bytes benchmark.");
             options.addOption("v", "version", true, "PowerAuth protocol version.");
 
             Option httpHeaderOption = Option.builder("H")
@@ -547,6 +552,24 @@ public static void main(String[] args) {
 
                     stepExecutionService.execute(powerAuthStep, version, model);
                 }
+
+                case GENERATE_BENCHMARK -> {
+                    final String roundsParam = cmd.getOptionValue("gr");
+                    final long rounds;
+                    if (roundsParam != null && roundsParam.matches("[0-9]+")) {
+                        rounds = Long.parseLong(roundsParam);
+                    } else {
+                        rounds = GENERATE_RANDOM_BENCHMARK_ROUNDS;
+                    }
+                    System.out.println("Generating " + rounds + " rounds of 32-byte random bytes started.");
+                    long timeStart = System.currentTimeMillis();
+                    for (int i = 0; i < rounds; i++) {
+                        KEY_GENERATOR.generateRandomBytes(32);
+                    }
+                    long totalTime = System.currentTimeMillis() - timeStart;
+                    System.out.println("Generating " + rounds + " rounds of 32-byte random bytes finished, total time: " + totalTime + "ms.");
+                }
+
                 default -> {
                     System.err.println("Not recognized PowerAuth step: " + powerAuthStep);
                     printPowerAuthStepsHelp(stepProvider);