125 lines
4.0 KiB
C
125 lines
4.0 KiB
C
|
|
/**
|
|
* Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
|
|
* SPDX-License-Identifier: Apache-2.0.
|
|
*/
|
|
#include "checksums.h"
|
|
|
|
#include <aws/checksums/crc.h>
|
|
#include <aws/common/encoding.h>
|
|
|
|
napi_value crc32_common(
|
|
napi_env env,
|
|
napi_callback_info info,
|
|
uint32_t (*checksum_fn)(const uint8_t *, size_t, uint32_t)) {
|
|
napi_value node_args[2];
|
|
size_t num_args = AWS_ARRAY_SIZE(node_args);
|
|
struct aws_byte_buf to_hash;
|
|
AWS_ZERO_STRUCT(to_hash);
|
|
|
|
napi_value node_val = NULL;
|
|
|
|
if (napi_get_cb_info(env, info, &num_args, node_args, NULL, NULL)) {
|
|
napi_throw_error(env, NULL, "Failed to retrieve callback information");
|
|
goto done;
|
|
}
|
|
if (num_args != AWS_ARRAY_SIZE(node_args)) {
|
|
napi_throw_error(env, NULL, "aws_napi_checksums_crc needs exactly 2 arguments");
|
|
goto done;
|
|
}
|
|
|
|
if (aws_byte_buf_init_from_napi(&to_hash, env, node_args[0])) {
|
|
napi_throw_type_error(env, NULL, "to_hash argument must be a string or array");
|
|
goto done;
|
|
}
|
|
uint8_t *buffer = to_hash.buffer;
|
|
size_t length = to_hash.len;
|
|
uint32_t previous_crc = 0;
|
|
|
|
if (!aws_napi_is_null_or_undefined(env, node_args[1])) {
|
|
if (napi_get_value_uint32(env, node_args[1], &previous_crc)) {
|
|
napi_throw_type_error(env, NULL, "previous_crc argument must be undefined or a positive number");
|
|
goto done;
|
|
}
|
|
}
|
|
|
|
uint32_t val = checksum_fn(buffer, length, previous_crc);
|
|
AWS_NAPI_CALL(env, napi_create_uint32(env, val, &node_val), { goto done; });
|
|
|
|
done:
|
|
aws_byte_buf_clean_up(&to_hash);
|
|
|
|
return node_val;
|
|
}
|
|
|
|
napi_value aws_napi_checksums_crc32(napi_env env, napi_callback_info info) {
|
|
return crc32_common(env, info, aws_checksums_crc32_ex);
|
|
}
|
|
|
|
napi_value aws_napi_checksums_crc32c(napi_env env, napi_callback_info info) {
|
|
return crc32_common(env, info, aws_checksums_crc32c_ex);
|
|
}
|
|
|
|
napi_value aws_napi_checksums_crc64nvme(napi_env env, napi_callback_info info) {
|
|
napi_value node_args[2];
|
|
size_t num_args = AWS_ARRAY_SIZE(node_args);
|
|
struct aws_byte_buf to_hash;
|
|
AWS_ZERO_STRUCT(to_hash);
|
|
struct aws_byte_buf previous_crc_buf;
|
|
AWS_ZERO_STRUCT(previous_crc_buf);
|
|
|
|
napi_value node_val = NULL;
|
|
|
|
if (napi_get_cb_info(env, info, &num_args, node_args, NULL, NULL)) {
|
|
napi_throw_error(env, NULL, "Failed to retrieve callback information");
|
|
goto done;
|
|
}
|
|
if (num_args != AWS_ARRAY_SIZE(node_args)) {
|
|
napi_throw_error(env, NULL, "aws_napi_checksums_crc64 needs exactly 2 arguments");
|
|
goto done;
|
|
}
|
|
|
|
if (aws_byte_buf_init_from_napi(&to_hash, env, node_args[0])) {
|
|
napi_throw_type_error(env, NULL, "to_hash argument must be a string or array");
|
|
goto done;
|
|
}
|
|
uint8_t *buffer = to_hash.buffer;
|
|
size_t length = to_hash.len;
|
|
|
|
uint64_t previous_crc = 0;
|
|
if (!aws_napi_is_null_or_undefined(env, node_args[1])) {
|
|
if (aws_byte_buf_init_from_napi(&previous_crc_buf, env, node_args[1])) {
|
|
napi_throw_type_error(env, NULL, "previous_crc argument must be undefined or a DataView over 8 bytes");
|
|
goto done;
|
|
}
|
|
if (previous_crc_buf.len != 8) {
|
|
napi_throw_type_error(env, NULL, "previous_crc argument must be a DataView over 8 bytes");
|
|
goto done;
|
|
}
|
|
previous_crc = aws_read_u64(previous_crc_buf.buffer);
|
|
}
|
|
|
|
uint64_t val = aws_checksums_crc64nvme_ex(buffer, length, previous_crc);
|
|
|
|
napi_value arraybuffer;
|
|
void *data = NULL;
|
|
if (napi_create_arraybuffer(env, 8, &data, &arraybuffer)) {
|
|
napi_throw_error(env, NULL, "Failed to create output arraybuffer");
|
|
goto done;
|
|
}
|
|
|
|
struct aws_byte_buf out_buf = aws_byte_buf_from_empty_array(data, 8);
|
|
aws_byte_buf_write_be64(&out_buf, val);
|
|
|
|
if (napi_create_dataview(env, 8, arraybuffer, 0, &node_val)) {
|
|
napi_throw_error(env, NULL, "Failed to create output dataview");
|
|
goto done;
|
|
}
|
|
|
|
done:
|
|
aws_byte_buf_clean_up(&to_hash);
|
|
aws_byte_buf_clean_up(&previous_crc_buf);
|
|
|
|
return node_val;
|
|
}
|