Add purpose validation on initialization of key generator

Cach30verfl0w · Jun 12, 2024 · 198cfbf · 198cfbf
1 parent 5768515
commit 198cfbf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/...advcrypto/src/androidMain/kotlin/io/karma/advcrypto/android/providers/PQCryptoProvider.kt b/...advcrypto/src/androidMain/kotlin/io/karma/advcrypto/android/providers/PQCryptoProvider.kt
@@ -58,7 +58,7 @@ class PQCryptoProvider: AbstractProvider(
 
         algorithm("Kyber") {
             keyGenerator(
-                Key.PURPOSE_ENCRYPT or Key.PURPOSE_ENCRYPT,
+                Key.PURPOSE_ENCRYPT or Key.PURPOSE_DECRYPT,
                 arrayOf(512, 768, 1024),
                 1024
             ) {

diff --git a/...ypto/src/commonMain/kotlin/io/karma/advcrypto/algorithm/delegates/KeyGeneratorDelegate.kt b/...ypto/src/commonMain/kotlin/io/karma/advcrypto/algorithm/delegates/KeyGeneratorDelegate.kt
@@ -120,7 +120,9 @@ class KeyGeneratorDelegate<C: Any>(
      */
     fun initializer(closure: (KeyGeneratorSpec) -> C) {
         this.initializer = { keyGenSpec ->
-            // TODO: Add validation for purposes
+            if ((keyGenSpec.purposes and this.keyPurposes) != keyGenSpec.purposes) {
+                throw IllegalArgumentException("Key purposes ${keyGenSpec.purposes} not supported")
+            }
             KeyGenContext(keyGenSpec, closure(keyGenSpec))
         }
     }