feat: expose chunked signing (#47)

awslabs · Mar 7, 2022 · 7408d64 · 7408d64
1 parent 01972f5
commit 7408d64
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/src/common/src/aws/sdk/kotlin/crt/auth/signing/AwsSigner.kt b/src/common/src/aws/sdk/kotlin/crt/auth/signing/AwsSigner.kt
@@ -11,4 +11,6 @@ public expect object AwsSigner {
     public suspend fun signRequest(request: HttpRequest, config: AwsSigningConfig): HttpRequest
 
     public suspend fun sign(request: HttpRequest, config: AwsSigningConfig): AwsSigningResult
+
+    public suspend fun signChunk(chunkBody: ByteArray, prevSignature: ByteArray, config: AwsSigningConfig): AwsSigningResult
 }
diff --git a/src/jvm/src/aws/sdk/kotlin/crt/auth/signing/AwsSignerJVM.kt b/src/jvm/src/aws/sdk/kotlin/crt/auth/signing/AwsSignerJVM.kt
@@ -7,7 +7,7 @@ package aws.sdk.kotlin.crt.auth.signing
 
 import aws.sdk.kotlin.crt.asyncCrtJniCall
 import aws.sdk.kotlin.crt.auth.credentials.JniCredentialsProvider
-import aws.sdk.kotlin.crt.http.HttpRequest
+import aws.sdk.kotlin.crt.http.*
 import aws.sdk.kotlin.crt.http.from
 import aws.sdk.kotlin.crt.http.into
 import kotlinx.coroutines.future.await
@@ -50,6 +50,24 @@ public actual object AwsSigner {
             AwsSigningResult(signedRequest, jniResult.signature)
         }
     }
+
+    /**
+     * Signs a body chunk according to the supplied signing configuration
+     * @param chunkBody the chunk payload to sign
+     * @param prevSignature the signature of the previous component of the request (either the initial request itself
+     * for the first chunk or the previous chunk otherwise)
+     * @param config signing configuration
+     * @return signing result, which provides access to all signing-related result properties
+     */
+    public actual suspend fun signChunk(chunkBody: ByteArray, prevSignature: ByteArray, config: AwsSigningConfig): AwsSigningResult {
+        val ktStream = HttpRequestBodyStream.fromByteArray(chunkBody)
+        val bodyStream = JniRequestBodyStream(ktStream)
+        return asyncCrtJniCall {
+            val signFuture = AwsSignerJni.sign(bodyStream, prevSignature, config.into())
+            val jniResult = signFuture.await()
+            AwsSigningResult(null, jniResult.signature)
+        }
+    }
 }
 
 private fun AwsSigningConfig.into(): AwsSigningConfigJni {