From 689066fd44f256a1556e1841e6cbc2175053c225 Mon Sep 17 00:00:00 2001
From: Vitor Pamplona <vitor@vitorpamplona.com>
Date: Mon, 15 Sep 2025 12:29:26 -0400
Subject: [PATCH] Benchmark for NIP-44 operations

---
 .../quartz/benchmark/EncryptDecrypt.kt        | 86 +++++++++++++++++++
 1 file changed, 86 insertions(+)
 create mode 100644 benchmark/src/androidTest/java/com/vitorpamplona/quartz/benchmark/EncryptDecrypt.kt

diff --git a/benchmark/src/androidTest/java/com/vitorpamplona/quartz/benchmark/EncryptDecrypt.kt b/benchmark/src/androidTest/java/com/vitorpamplona/quartz/benchmark/EncryptDecrypt.kt
new file mode 100644
index 000000000..c315d0d4c
--- /dev/null
+++ b/benchmark/src/androidTest/java/com/vitorpamplona/quartz/benchmark/EncryptDecrypt.kt
@@ -0,0 +1,86 @@
+/**
+ * Copyright (c) 2025 Vitor Pamplona
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy of
+ * this software and associated documentation files (the "Software"), to deal in
+ * the Software without restriction, including without limitation the rights to use,
+ * copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the
+ * Software, and to permit persons to whom the Software is furnished to do so,
+ * subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+ * FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+ * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN
+ * AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+ * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+package com.vitorpamplona.quartz.benchmark
+
+import androidx.benchmark.junit4.BenchmarkRule
+import androidx.benchmark.junit4.measureRepeated
+import androidx.test.ext.junit.runners.AndroidJUnit4
+import com.vitorpamplona.quartz.nip01Core.crypto.Nip01
+import com.vitorpamplona.quartz.nip44Encryption.Nip44v2
+import com.vitorpamplona.quartz.nip44Encryption.crypto.Hkdf
+import org.junit.Rule
+import org.junit.Test
+import org.junit.runner.RunWith
+
+@RunWith(AndroidJUnit4::class)
+class EncryptDecrypt {
+    @get:Rule val benchmarkRule = BenchmarkRule()
+
+    companion object {
+        val nip44v2 = Nip44v2()
+        val msg = "Hi, how are you? this is supposed to be representative of an average message on Nostr"
+
+        val privateKey = Nip01.privKeyCreate()
+        val publicKey = Nip01.pubKeyCreate(privateKey)
+
+        val sharedKey = nip44v2.getConversationKey(privateKey, publicKey)
+        val encrypted = nip44v2.encrypt(msg, sharedKey)
+    }
+
+    @Test
+    fun encrypt() {
+        benchmarkRule.measureRepeated {
+            nip44v2.encrypt(msg, sharedKey)
+        }
+    }
+
+    @Test
+    fun decrypt() {
+        benchmarkRule.measureRepeated {
+            nip44v2.decrypt(encrypted, sharedKey)
+        }
+    }
+
+    @Test
+    fun messageKeys() {
+        benchmarkRule.measureRepeated {
+            nip44v2.getMessageKeys(sharedKey, encrypted.nonce)
+        }
+    }
+
+    @Test
+    fun checkMac() {
+        val messageKey = nip44v2.getMessageKeys(sharedKey, encrypted.nonce)
+
+        benchmarkRule.measureRepeated {
+            nip44v2.checkHMacAad(messageKey, encrypted)
+        }
+    }
+
+    @Test
+    fun hkdfExtract() {
+        val messageKey = nip44v2.getMessageKeys(sharedKey, encrypted.nonce)
+        val hkdf = Hkdf()
+        benchmarkRule.measureRepeated {
+            hkdf.extract(encrypted.nonce, encrypted.ciphertext, messageKey.hmacKey)
+        }
+    }
+}