From 434b91930b29d22a04b1d4f3c47e1f2a6c491b6f Mon Sep 17 00:00:00 2001
From: Denis Trotsenko <denis.tnko@gmail.com>
Date: Sun, 5 May 2024 18:51:18 +0200
Subject: [PATCH] Add method to measure KDF transform rounds performance

---
 .../keemobile/kotpass/database/KeePassDatabase.kt   | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/kotpass/src/main/kotlin/app/keemobile/kotpass/database/KeePassDatabase.kt b/kotpass/src/main/kotlin/app/keemobile/kotpass/database/KeePassDatabase.kt
index e570c6b..82d2ce5 100644
--- a/kotpass/src/main/kotlin/app/keemobile/kotpass/database/KeePassDatabase.kt
+++ b/kotpass/src/main/kotlin/app/keemobile/kotpass/database/KeePassDatabase.kt
@@ -3,8 +3,10 @@
 package app.keemobile.kotpass.database
 
 import app.keemobile.kotpass.constants.GroupOverride
+import app.keemobile.kotpass.cryptography.KeyTransform
 import app.keemobile.kotpass.database.header.DatabaseHeader
 import app.keemobile.kotpass.database.header.DatabaseInnerHeader
+import app.keemobile.kotpass.extensions.clear
 import app.keemobile.kotpass.models.DatabaseContent
 import app.keemobile.kotpass.models.DatabaseElement
 import app.keemobile.kotpass.models.Entry
@@ -12,6 +14,7 @@ import app.keemobile.kotpass.models.Group
 import app.keemobile.kotpass.models.Meta
 import java.security.SecureRandom
 import java.util.UUID
+import kotlin.system.measureTimeMillis
 
 /**
  * Main class which describes Keepass database.
@@ -216,3 +219,13 @@ fun KeePassDatabase.findEntries(
         .group
         .findChildEntries(true, content.meta.recycleBinUuid, predicate)
 }
+
+/**
+ * Measures KDF transform rounds performance based on
+ * [header][KeePassDatabase.header] parameters.
+ */
+fun KeePassDatabase.measureKeyTransformMillis() = measureTimeMillis {
+    KeyTransform
+        .transformedKey(header, credentials)
+        .clear()
+}