public void TestGetObjectAPMKMS()
{
PutObjectRequest putObjectRequest = new PutObjectRequest()
{
BucketName = bucketName,
Key = $"key-{Guid.NewGuid()}",
ContentBody = SampleContent
};
s3EncryptionClientMetadataModeKMSV1N.PutObject(putObjectRequest);
GetObjectRequest getObjectRequest = new GetObjectRequest
{
BucketName = bucketName,
Key = putObjectRequest.Key
};
AssertExtensions.ExpectException(() =>
{
s3EncryptionClientMetadataModeKMSV1N.EndGetObject(
s3EncryptionClientMetadataModeKMSV1N.BeginGetObject(getObjectRequest, null, null));
}, typeof(NotSupportedException), APMKMSErrorRegex);
AssertExtensions.ExpectException(() =>
{
s3EncryptionClientMetadataModeKMSV2.EndGetObject(
s3EncryptionClientMetadataModeKMSV2.BeginGetObject(getObjectRequest, null, null));
}, typeof(NotSupportedException), APMKMSErrorRegex);
}
private static void TestGetAPM(string key, string uploadedData, AmazonS3EncryptionClient s3EncryptionClient)
{
GetObjectRequest getObjectRequest = new GetObjectRequest
{
BucketName = bucketName,
Key = key
};
var asyncResult = s3EncryptionClient.BeginGetObject(getObjectRequest, null, null);
using (GetObjectResponse getObjectResponse = s3EncryptionClient.EndGetObject(asyncResult))
{
using (var stream = getObjectResponse.ResponseStream)
using (var reader = new StreamReader(stream))
{
string data = reader.ReadToEnd();
Assert.AreEqual(uploadedData, data);
}
}
}
public void TestGetObjectAPMKMS()
{
PutObjectRequest putObjectRequest = new PutObjectRequest()
{
BucketName = bucketName,
Key = "key" + random.Next(),
ContentBody = sampleContent,
CannedACL = S3CannedACL.AuthenticatedRead
};
s3EncryptionClientMetadataModeKMS.PutObject(putObjectRequest);
GetObjectRequest getObjectRequest = new GetObjectRequest
{
BucketName = bucketName,
Key = putObjectRequest.Key
};
AssertExtensions.ExpectException(() =>
{
s3EncryptionClientMetadataModeKMS.EndGetObject(
s3EncryptionClientMetadataModeKMS.BeginGetObject(getObjectRequest, null, null));
}, typeof(NotSupportedException), APMKMSErrorRegex);
}
public void MultipartEncryptionTestAPM(AmazonS3EncryptionClient s3EncryptionClient)
{
var nextRandom = random.Next();
var filePath = @"C:\temp\multi-" + nextRandom + ".txt";
var retrievedFilepath = @"C:\temp\retreived-" + nextRandom + ".txt";
var totalSize = MegSize * 15;
UtilityMethods.GenerateFile(filePath, totalSize);
string key = "key-" + random.Next();
Stream inputStream = File.OpenRead(filePath);
try
{
InitiateMultipartUploadRequest initRequest = new InitiateMultipartUploadRequest()
{
BucketName = bucketName,
Key = key,
StorageClass = S3StorageClass.ReducedRedundancy,
ContentType = "text/html",
CannedACL = S3CannedACL.PublicRead
};
InitiateMultipartUploadResponse initResponse = null;
if (IsKMSEncryptionClient(s3EncryptionClient))
{
initResponse = s3EncryptionClient.InitiateMultipartUpload(initRequest);
}
else
{
initResponse = s3EncryptionClient.EndInitiateMultipartUpload(
s3EncryptionClient.BeginInitiateMultipartUpload(initRequest, null, null));
}
// Upload part 1
UploadPartRequest uploadRequest = new UploadPartRequest()
{
BucketName = bucketName,
Key = key,
UploadId = initResponse.UploadId,
PartNumber = 1,
PartSize = 5 * MegSize,
InputStream = inputStream,
};
UploadPartResponse up1Response = s3EncryptionClient.EndUploadPart(
s3EncryptionClient.BeginUploadPart(uploadRequest, null, null));
// Upload part 2
uploadRequest = new UploadPartRequest()
{
BucketName = bucketName,
Key = key,
UploadId = initResponse.UploadId,
PartNumber = 2,
PartSize = 5 * MegSize,
InputStream = inputStream,
};
UploadPartResponse up2Response = s3EncryptionClient.EndUploadPart(
s3EncryptionClient.BeginUploadPart(uploadRequest, null, null));
// Upload part 3
uploadRequest = new UploadPartRequest()
{
BucketName = bucketName,
Key = key,
UploadId = initResponse.UploadId,
PartNumber = 3,
InputStream = inputStream,
IsLastPart = true
};
UploadPartResponse up3Response = s3EncryptionClient.EndUploadPart(
s3EncryptionClient.BeginUploadPart(uploadRequest, null, null));
ListPartsRequest listPartRequest = new ListPartsRequest()
{
BucketName = bucketName,
Key = key,
UploadId = initResponse.UploadId
};
ListPartsResponse listPartResponse = s3EncryptionClient.EndListParts(
s3EncryptionClient.BeginListParts(listPartRequest, null, null));
Assert.AreEqual(3, listPartResponse.Parts.Count);
Assert.AreEqual(up1Response.PartNumber, listPartResponse.Parts[0].PartNumber);
Assert.AreEqual(up1Response.ETag, listPartResponse.Parts[0].ETag);
Assert.AreEqual(up2Response.PartNumber, listPartResponse.Parts[1].PartNumber);
Assert.AreEqual(up2Response.ETag, listPartResponse.Parts[1].ETag);
Assert.AreEqual(up3Response.PartNumber, listPartResponse.Parts[2].PartNumber);
Assert.AreEqual(up3Response.ETag, listPartResponse.Parts[2].ETag);
listPartRequest.MaxParts = 1;
listPartResponse = s3EncryptionClient.EndListParts(
s3EncryptionClient.BeginListParts(listPartRequest, null, null));
Assert.AreEqual(1, listPartResponse.Parts.Count);
// Complete the response
CompleteMultipartUploadRequest compRequest = new CompleteMultipartUploadRequest()
{
BucketName = bucketName,
Key = key,
UploadId = initResponse.UploadId
};
compRequest.AddPartETags(up1Response, up2Response, up3Response);
CompleteMultipartUploadResponse compResponse = s3EncryptionClient.EndCompleteMultipartUpload(
s3EncryptionClient.BeginCompleteMultipartUpload(compRequest, null, null));
Assert.AreEqual(bucketName, compResponse.BucketName);
Assert.IsNotNull(compResponse.ETag);
Assert.AreEqual(key, compResponse.Key);
Assert.IsNotNull(compResponse.Location);
// Get the file back from S3 and make sure it is still the same.
GetObjectRequest getRequest = new GetObjectRequest()
{
BucketName = bucketName,
Key = key
};
GetObjectResponse getResponse = null;
if (IsKMSEncryptionClient(s3EncryptionClient))
{
getResponse = s3EncryptionClient.GetObject(getRequest);
}
else
{
getResponse = s3EncryptionClient.EndGetObject(
s3EncryptionClient.BeginGetObject(getRequest, null, null));
}
getResponse.WriteResponseStreamToFile(retrievedFilepath);
UtilityMethods.CompareFiles(filePath, retrievedFilepath);
GetObjectMetadataRequest metaDataRequest = new GetObjectMetadataRequest()
{
BucketName = bucketName,
Key = key
};
GetObjectMetadataResponse metaDataResponse = s3EncryptionClient.EndGetObjectMetadata(
s3EncryptionClient.BeginGetObjectMetadata(metaDataRequest, null, null));
Assert.AreEqual("text/html", metaDataResponse.Headers.ContentType);
}
finally
{
inputStream.Close();
if (File.Exists(filePath))
{
File.Delete(filePath);
}
if (File.Exists(retrievedFilepath))
{
File.Delete(retrievedFilepath);
}
}
}