unit test

Author: xinsong-cui

runtime/runtime-core/jvm/test/aws/smithy/kotlin/runtime/hashing/EcdsaJVMTest.kt

@@ -6,6 +6,8 @@ package aws.smithy.kotlin.runtime.hashing
 
 import org.junit.jupiter.api.Assertions.assertFalse
 import org.junit.jupiter.api.Assertions.assertTrue
+import org.junit.jupiter.api.condition.EnabledForJreRange
+import org.junit.jupiter.api.condition.JRE
 import java.security.*
 import java.security.interfaces.*
 import java.security.spec.*
@@ -30,6 +32,10 @@ class EcdsaJVMTest {
 
         assertTrue(signature.isNotEmpty())
         assertTrue(signature.size >= 64) // ECDSA signatures are typically 70-72 bytes in DER format
+
+        val rawSignature = ecdsaSecp256r1Rs(privateKey, message)
+        assertTrue(signature.isNotEmpty())
+        assertTrue(rawSignature.size == 64)
     }
 
     @Test
@@ -44,6 +50,13 @@ class EcdsaJVMTest {
         assertTrue(signature1.isNotEmpty())
         assertTrue(signature2.isNotEmpty())
         assertFalse(signature1.contentEquals(signature2))
+
+        val rawSignature1 = ecdsaSecp256r1Rs(privateKey, message1)
+        val rawSignature2 = ecdsaSecp256r1Rs(privateKey, message2)
+
+        assertTrue(rawSignature1.isNotEmpty())
+        assertTrue(rawSignature2.isNotEmpty())
+        assertFalse(rawSignature1.contentEquals(rawSignature2))
     }
 
     @Test
@@ -53,6 +66,9 @@ class EcdsaJVMTest {
 
         val signature = ecdsaSecp256r1(privateKey, message)
         assertTrue(signature.isNotEmpty())
+
+        val rawSignature = ecdsaSecp256r1Rs(privateKey, message)
+        assertTrue(rawSignature.isNotEmpty())
     }
 
     @Test
@@ -62,6 +78,9 @@ class EcdsaJVMTest {
 
         val signature = ecdsaSecp256r1(privateKey, largeMessage)
         assertTrue(signature.isNotEmpty())
+
+        val rawSignature = ecdsaSecp256r1Rs(privateKey, largeMessage)
+        assertTrue(rawSignature.isNotEmpty())
     }
 
     @Test
@@ -78,6 +97,23 @@ class EcdsaJVMTest {
         val verifier = Signature.getInstance("SHA256withECDSA")
         verifier.initVerify(publicKey)
         verifier.update(message)
+    }
+
+    @Test
+    @EnabledForJreRange(min = JRE.JAVA_11)
+    fun testVerifyRawSignature() {
+        val keyGen = KeyPairGenerator.getInstance("EC")
+        keyGen.initialize(ECGenParameterSpec("secp256r1"))
+        val keyPair = keyGen.generateKeyPair()
+        val privateKey = (keyPair.private as ECPrivateKey).s.toByteArray()
+        val publicKey = keyPair.public
+
+        val message = "Hello, World!".toByteArray()
+        val signature = ecdsaSecp256r1Rs(privateKey, message)
+
+        val verifier = Signature.getInstance("SHA256withECDSAinP1363Format")
+        verifier.initVerify(publicKey)
+        verifier.update(message)
 
         assertTrue(verifier.verify(signature))
     }