fix: have a better error thrown if key format not supported

beatt83 · Dec 7, 2024 · 9a78a8a · 9a78a8a
1 parent afb3128
commit 9a78a8a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Sources/JSONWebAlgorithms/KeyManagement/KeyEncryption/AES/AES256GCM+KeyEncryption.swift b/Sources/JSONWebAlgorithms/KeyManagement/KeyEncryption/AES/AES256GCM+KeyEncryption.swift
@@ -33,8 +33,11 @@ extension AES256GCM: KeyWrapping {
         using: JWK,
         arguments: [KeyEncryptionArguments]
     ) throws -> KeyEncriptionResultMetadata {
+        guard using.keyType == .octetSequence else {
+            throw CryptoError.keyFormatNotSupported(format: using.keyType.rawValue, supportedFormats: [JWK.KeyType.octetSequence.rawValue])
+        }
         guard let key = using.key else {
-            throw CryptoError.missingArguments([])
+            throw CryptoError.missingArguments(["JWK.Key"])
         }
 
         let initializationVector = try arguments.initializationVector ?? generateInitializationVector()