From e7270900205b93af338cabe68f59d8ccfb0ed020 Mon Sep 17 00:00:00 2001
From: Alexander Bondarenko <486682+dpwiz@users.noreply.github.com>
Date: Thu, 29 Feb 2024 16:23:10 +0200
Subject: [PATCH] add sntrup761 benchmark

---
 benchmarks/Bench.hs           | 11 +++++++----
 benchmarks/Bench/SNTRUP761.hs | 15 +++++++++++++++
 simplexmq.cabal               |  1 +
 3 files changed, 23 insertions(+), 4 deletions(-)
 create mode 100644 benchmarks/Bench/SNTRUP761.hs

diff --git a/benchmarks/Bench.hs b/benchmarks/Bench.hs
index 67094ce94..abfc9da9c 100644
--- a/benchmarks/Bench.hs
+++ b/benchmarks/Bench.hs
@@ -8,10 +8,13 @@ Pick one or group: cabal bench -O2 simplexmq-bench --benchmark-options "-p TRcvQ
 
 module Main where
 
-import Test.Tasty.Bench
+import Bench.SNTRUP761
 import Bench.TRcvQueues
+import Test.Tasty.Bench
 
 main :: IO ()
-main = defaultMain
-  [ bgroup "TRcvQueues" benchTRcvQueues
-  ]
+main =
+  defaultMain
+    [ bgroup "TRcvQueues" benchTRcvQueues,
+      bgroup "SNTRUP761" benchSNTRUP761
+    ]
diff --git a/benchmarks/Bench/SNTRUP761.hs b/benchmarks/Bench/SNTRUP761.hs
new file mode 100644
index 000000000..6b02e0bf5
--- /dev/null
+++ b/benchmarks/Bench/SNTRUP761.hs
@@ -0,0 +1,15 @@
+module Bench.SNTRUP761 where
+
+import qualified Simplex.Messaging.Crypto as C
+import Simplex.Messaging.Crypto.SNTRUP761.Bindings
+import Test.Tasty.Bench
+
+import Test.Tasty (withResource)
+
+benchSNTRUP761 :: [Benchmark]
+benchSNTRUP761 =
+  [ bgroup
+      "sntrup761Keypair"
+      [ withResource C.newRandom (\_ -> pure ()) $ bench "current" . whnfAppIO (>>= sntrup761Keypair)
+      ]
+  ]
diff --git a/simplexmq.cabal b/simplexmq.cabal
index e87150a36..66b671139 100644
--- a/simplexmq.cabal
+++ b/simplexmq.cabal
@@ -720,6 +720,7 @@ benchmark simplexmq-bench
   type: exitcode-stdio-1.0
   main-is: Bench.hs
   other-modules:
+      Bench.SNTRUP761
       Bench.TRcvQueues
       Paths_simplexmq
   hs-source-dirs: