commonMain.aws.sdk.kotlin.services.glacier.internal.GlacierBodyChecksum.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of glacier-jvm Show documentation

The AWS Kotlin client for Glacier

There is a newer version: 1.3.99

/*
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 * SPDX-License-Identifier: Apache-2.0
 */

package aws.sdk.kotlin.services.glacier.internal

import aws.sdk.kotlin.services.glacier.model.GlacierException
import aws.smithy.kotlin.runtime.client.operationName
import aws.smithy.kotlin.runtime.hashing.Sha256
import aws.smithy.kotlin.runtime.http.operation.ModifyRequestMiddleware
import aws.smithy.kotlin.runtime.http.operation.SdkHttpOperation
import aws.smithy.kotlin.runtime.http.operation.SdkHttpRequest
import aws.smithy.kotlin.runtime.http.request.headers
import aws.smithy.kotlin.runtime.util.encodeToHex

private const val defaultChunkSizeBytes = 1024 * 1024 // 1MB

internal class GlacierBodyChecksum(
    private val treeHasher: TreeHasher = TreeHasherImpl(defaultChunkSizeBytes, ::Sha256),
) : ModifyRequestMiddleware {
    override fun install(op: SdkHttpOperation<*, *>) {
        // after signing
        op.execution.receive.register(this)
    }

    override suspend fun modifyRequest(req: SdkHttpRequest): SdkHttpRequest {
        val body = req.subject.body
        if (body.isOneShot) {
            val opName = req.context.operationName ?: "This operation"
            throw GlacierException("$opName requires a byte array or replayable stream")
        }
        val hashes = treeHasher.calculateHashes(body)
        req.subject.headers {
            set("X-Amz-Content-Sha256", hashes.fullHash.encodeToHex())
            set("X-Amz-Sha256-Tree-Hash", hashes.treeHash.encodeToHex())
        }

        return req
    }
}