Add method to measure KDF transform rounds performance

keemobile · May 5, 2024 · 434b919 · 434b919
1 parent a3cdc93
commit 434b919
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/kotpass/src/main/kotlin/app/keemobile/kotpass/database/KeePassDatabase.kt b/kotpass/src/main/kotlin/app/keemobile/kotpass/database/KeePassDatabase.kt
@@ -3,15 +3,18 @@
 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
 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()
+}