1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
/**
* Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
* SPDX-License-Identifier: Apache-2.0.
*/
#include "aws/s3/private/s3_checksum_context.h"
#include "aws/s3/private/s3_checksums.h"
#include <aws/common/byte_buf.h>
#include <aws/testing/aws_test_harness.h>
static int s_test_upload_request_checksum_context_get_checksum_cursor(struct aws_allocator *allocator, void *ctx) {
(void)ctx;
struct aws_s3_meta_request_checksum_config_storage config = {
.allocator = allocator,
.checksum_algorithm = AWS_SCA_CRC32,
.location = AWS_SCL_HEADER,
.has_full_object_checksum = false,
};
AWS_ZERO_STRUCT(config.full_object_checksum);
/* Test get checksum cursor with context that has no calculated checksum */
struct aws_s3_upload_request_checksum_context *context =
aws_s3_upload_request_checksum_context_new(allocator, &config);
ASSERT_NOT_NULL(context);
struct aws_byte_cursor cursor = aws_s3_upload_request_checksum_context_get_checksum_cursor(context);
ASSERT_TRUE(cursor.len == 0);
ASSERT_NULL(cursor.ptr);
aws_s3_upload_request_checksum_context_release(context);
/* Test get checksum cursor with context that has calculated checksum */
struct aws_byte_cursor existing_checksum = aws_byte_cursor_from_c_str("dGVzdA==");
context =
aws_s3_upload_request_checksum_context_new_with_existing_base64_checksum(allocator, &config, existing_checksum);
ASSERT_NOT_NULL(context);
cursor = aws_s3_upload_request_checksum_context_get_checksum_cursor(context);
ASSERT_TRUE(cursor.len == existing_checksum.len);
ASSERT_TRUE(aws_byte_cursor_eq(&cursor, &existing_checksum));
aws_s3_upload_request_checksum_context_release(context);
/* Test get checksum cursor with NULL context */
cursor = aws_s3_upload_request_checksum_context_get_checksum_cursor(NULL);
ASSERT_TRUE(cursor.len == 0);
ASSERT_NULL(cursor.ptr);
return AWS_OP_SUCCESS;
}
AWS_TEST_CASE(
test_upload_request_checksum_context_get_checksum_cursor,
s_test_upload_request_checksum_context_get_checksum_cursor)
static int s_test_upload_request_checksum_context_error_cases(struct aws_allocator *allocator, void *ctx) {
(void)ctx;
struct aws_s3_meta_request_checksum_config_storage config = {
.allocator = allocator,
.checksum_algorithm = AWS_SCA_CRC32,
.location = AWS_SCL_HEADER,
.has_full_object_checksum = false,
};
AWS_ZERO_STRUCT(config.full_object_checksum);
/* Test creation with mismatched checksum size */
struct aws_byte_cursor wrong_size_checksum = aws_byte_cursor_from_c_str("short");
struct aws_s3_upload_request_checksum_context *context =
aws_s3_upload_request_checksum_context_new_with_existing_base64_checksum(
allocator, &config, wrong_size_checksum);
ASSERT_NULL(context);
/* Test helper functions with NULL context */
ASSERT_FALSE(aws_s3_upload_request_checksum_context_should_calculate(NULL));
ASSERT_FALSE(aws_s3_upload_request_checksum_context_should_add_header(NULL));
ASSERT_FALSE(aws_s3_upload_request_checksum_context_should_add_trailer(NULL));
/* Test acquire/release with NULL context */
ASSERT_NULL(aws_s3_upload_request_checksum_context_acquire(NULL));
ASSERT_NULL(aws_s3_upload_request_checksum_context_release(NULL));
return AWS_OP_SUCCESS;
}
AWS_TEST_CASE(test_upload_request_checksum_context_error_cases, s_test_upload_request_checksum_context_error_cases)
static int s_test_upload_request_checksum_context_different_algorithms(struct aws_allocator *allocator, void *ctx) {
(void)ctx;
/* Test different checksum algorithms */
enum aws_s3_checksum_algorithm algorithms[] = {
AWS_SCA_CRC32, AWS_SCA_CRC32C, AWS_SCA_SHA1, AWS_SCA_SHA256, AWS_SCA_CRC64NVME};
for (size_t i = 0; i < AWS_ARRAY_SIZE(algorithms); ++i) {
struct aws_s3_meta_request_checksum_config_storage config = {
.allocator = allocator,
.checksum_algorithm = algorithms[i],
.location = AWS_SCL_HEADER,
.has_full_object_checksum = false,
};
AWS_ZERO_STRUCT(config.full_object_checksum);
struct aws_s3_upload_request_checksum_context *context =
aws_s3_upload_request_checksum_context_new(allocator, &config);
ASSERT_NOT_NULL(context);
ASSERT_INT_EQUALS(algorithms[i], context->algorithm);
ASSERT_INT_EQUALS(AWS_SCL_HEADER, context->location);
ASSERT_TRUE(context->encoded_checksum_size > 0);
ASSERT_TRUE(aws_s3_upload_request_checksum_context_should_calculate(context));
ASSERT_TRUE(aws_s3_upload_request_checksum_context_should_add_header(context));
ASSERT_FALSE(aws_s3_upload_request_checksum_context_should_add_trailer(context));
aws_s3_upload_request_checksum_context_release(context);
}
return AWS_OP_SUCCESS;
}
AWS_TEST_CASE(
test_upload_request_checksum_context_different_algorithms,
s_test_upload_request_checksum_context_different_algorithms)
|
