Copy SSE-C parameters when performing multipart copy (#4219)

This commit fixes a bug where the SSE-C parameters are not copied to the
CompleteMultipartUpload request when transforming to a multipart copy.

Author: dagnir

.changes/next-release/bugfix-S3TransferManager-d9c09d4.json

@@ -0,0 +1,6 @@
+{
+    "type": "bugfix",
+    "category": "S3 Transfer Manager",
+    "contributor": "",
+    "description": "Fix a bug where the SSE-C parameters are not copied to the CompleteMultipartUpload request when transforming to a multipart copy."
+}

services/s3/src/main/java/software/amazon/awssdk/services/s3/internal/crt/CopyObjectHelper.java

@@ -189,6 +189,9 @@ private CompletableFuture<CompleteMultipartUploadResponse> completeMultipartUplo
                                           .multipartUpload(CompletedMultipartUpload.builder()
                                                                                    .parts(parts)
                                                                                    .build())
+                                          .sseCustomerAlgorithm(copyObjectRequest.sseCustomerAlgorithm())
+                                          .sseCustomerKey(copyObjectRequest.sseCustomerKey())
+                                          .sseCustomerKeyMD5(copyObjectRequest.sseCustomerKeyMD5())
                                           .build();
 
         return s3AsyncClient.completeMultipartUpload(completeMultipartUploadRequest);

services/s3/src/test/java/software/amazon/awssdk/services/s3/internal/crt/CopyObjectHelperTest.java

@@ -24,6 +24,7 @@
 import static org.mockito.Mockito.when;
 
 import java.util.List;
+import java.util.Random;
 import java.util.concurrent.CompletableFuture;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
@@ -47,7 +48,9 @@
 import software.amazon.awssdk.services.s3.model.NoSuchBucketException;
 import software.amazon.awssdk.services.s3.model.UploadPartCopyRequest;
 import software.amazon.awssdk.services.s3.model.UploadPartCopyResponse;
+import software.amazon.awssdk.utils.BinaryUtils;
 import software.amazon.awssdk.utils.CompletableFutureUtils;
+import software.amazon.awssdk.utils.Md5Utils;
 
 class CopyObjectHelperTest {
 
@@ -56,13 +59,15 @@ class CopyObjectHelperTest {
     private static final String DESTINATION_BUCKET = "destination";
     private static final String DESTINATION_KEY = "destinationKey";
     private static final String MULTIPART_ID = "multipartId";
+
+    private static final Long PART_SIZE_BYTES = 1024L;
     private S3AsyncClient s3AsyncClient;
     private CopyObjectHelper copyHelper;
 
     @BeforeEach
     public void setUp() {
         s3AsyncClient = Mockito.mock(S3AsyncClient.class);
-        copyHelper = new CopyObjectHelper(s3AsyncClient, 1024L);
+        copyHelper = new CopyObjectHelper(s3AsyncClient, PART_SIZE_BYTES);
     }
 
     @Test
@@ -256,6 +261,35 @@ void multiPartCopy_contentSizeExceeds10000Parts_shouldAdjustPartSize() {
         }
     }
 
+    @Test
+    public void multiPartCopy_sseCHeadersSetInOriginalRequest_includedInCompleteMultipart() {
+        String customerAlgorithm = "algorithm";
+        String customerKey = "key";
+        String customerKeyMd5 = "keyMd5";
+
+        CopyObjectRequest copyRequest = copyObjectRequest().copy(r -> r.sseCustomerAlgorithm(customerAlgorithm)
+                                                                       .sseCustomerKey(customerKey)
+                                                                       .sseCustomerKeyMD5(customerKeyMd5));
+
+        stubSuccessfulHeadObjectCall(2 * PART_SIZE_BYTES);
+        stubSuccessfulCreateMulipartCall();
+        stubSuccessfulUploadPartCopyCalls();
+        stubSuccessfulCompleteMultipartCall();
+
+        copyHelper.copyObject(copyRequest).join();
+
+        ArgumentCaptor<CompleteMultipartUploadRequest> completeMultipartCaptor =
+            ArgumentCaptor.forClass(CompleteMultipartUploadRequest.class);
+
+        verify(s3AsyncClient).completeMultipartUpload(completeMultipartCaptor.capture());
+
+        CompleteMultipartUploadRequest completeRequest = completeMultipartCaptor.getValue();
+
+        assertThat(completeRequest.sseCustomerAlgorithm()).isEqualTo(customerAlgorithm);
+        assertThat(completeRequest.sseCustomerKey()).isEqualTo(customerKey);
+        assertThat(completeRequest.sseCustomerKeyMD5()).isEqualTo(customerKeyMd5);
+    }
+
     @Test
     void copy_cancelResponseFuture_shouldPropagate() {
         CopyObjectRequest copyObjectRequest = copyObjectRequest();