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
|
/**
* Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
* SPDX-License-Identifier: Apache-2.0.
*/
#include <aws/cal/hash.h>
#include <CommonCrypto/CommonDigest.h>
static void s_destroy(struct aws_hash *hash);
static int s_update(struct aws_hash *hash, const struct aws_byte_cursor *to_hash);
static int s_finalize(struct aws_hash *hash, struct aws_byte_buf *output);
static struct aws_hash_vtable s_vtable = {
.destroy = s_destroy,
.update = s_update,
.finalize = s_finalize,
.alg_name = "SHA256",
.provider = "CommonCrypto",
};
struct cc_sha256_hash {
struct aws_hash hash;
CC_SHA256_CTX cc_hash;
};
struct aws_hash *aws_sha256_default_new(struct aws_allocator *allocator) {
struct cc_sha256_hash *sha256_hash = aws_mem_acquire(allocator, sizeof(struct cc_sha256_hash));
if (!sha256_hash) {
return NULL;
}
sha256_hash->hash.allocator = allocator;
sha256_hash->hash.vtable = &s_vtable;
sha256_hash->hash.impl = sha256_hash;
sha256_hash->hash.digest_size = AWS_SHA256_LEN;
sha256_hash->hash.good = true;
CC_SHA256_Init(&sha256_hash->cc_hash);
return &sha256_hash->hash;
}
static void s_destroy(struct aws_hash *hash) {
struct cc_sha256_hash *ctx = hash->impl;
aws_mem_release(hash->allocator, ctx);
}
static int s_update(struct aws_hash *hash, const struct aws_byte_cursor *to_hash) {
if (!hash->good) {
return aws_raise_error(AWS_ERROR_INVALID_STATE);
}
struct cc_sha256_hash *ctx = hash->impl;
CC_SHA256_Update(&ctx->cc_hash, to_hash->ptr, (CC_LONG)to_hash->len);
return AWS_OP_SUCCESS;
}
static int s_finalize(struct aws_hash *hash, struct aws_byte_buf *output) {
if (!hash->good) {
return aws_raise_error(AWS_ERROR_INVALID_STATE);
}
struct cc_sha256_hash *ctx = hash->impl;
size_t buffer_len = output->capacity - output->len;
if (buffer_len < hash->digest_size) {
return aws_raise_error(AWS_ERROR_SHORT_BUFFER);
}
CC_SHA256_Final(output->buffer + output->len, &ctx->cc_hash);
hash->good = false;
output->len += hash->digest_size;
return AWS_OP_SUCCESS;
}
|
