File Explorer

1 2/**3 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.4 * SPDX-License-Identifier: Apache-2.0.5 */6#include "checksums.h"7 8#include <aws/checksums/crc.h>9#include <aws/common/encoding.h>10 11napi_value crc32_common(12    napi_env env,13    napi_callback_info info,14    uint32_t (*checksum_fn)(const uint8_t *, size_t, uint32_t)) {15    napi_value node_args[2];16    size_t num_args = AWS_ARRAY_SIZE(node_args);17    struct aws_byte_buf to_hash;18    AWS_ZERO_STRUCT(to_hash);19 20    napi_value node_val = NULL;21 22    if (napi_get_cb_info(env, info, &num_args, node_args, NULL, NULL)) {23        napi_throw_error(env, NULL, "Failed to retrieve callback information");24        goto done;25    }26    if (num_args != AWS_ARRAY_SIZE(node_args)) {27        napi_throw_error(env, NULL, "aws_napi_checksums_crc needs exactly 2 arguments");28        goto done;29    }30 31    if (aws_byte_buf_init_from_napi(&to_hash, env, node_args[0])) {32        napi_throw_type_error(env, NULL, "to_hash argument must be a string or array");33        goto done;34    }35    uint8_t *buffer = to_hash.buffer;36    size_t length = to_hash.len;37    uint32_t previous_crc = 0;38 39    if (!aws_napi_is_null_or_undefined(env, node_args[1])) {40        if (napi_get_value_uint32(env, node_args[1], &previous_crc)) {41            napi_throw_type_error(env, NULL, "previous_crc argument must be undefined or a positive number");42            goto done;43        }44    }45 46    uint32_t val = checksum_fn(buffer, length, previous_crc);47    AWS_NAPI_CALL(env, napi_create_uint32(env, val, &node_val), { goto done; });48 49done:50    aws_byte_buf_clean_up(&to_hash);51 52    return node_val;53}54 55napi_value aws_napi_checksums_crc32(napi_env env, napi_callback_info info) {56    return crc32_common(env, info, aws_checksums_crc32_ex);57}58 59napi_value aws_napi_checksums_crc32c(napi_env env, napi_callback_info info) {60    return crc32_common(env, info, aws_checksums_crc32c_ex);61}62 63napi_value aws_napi_checksums_crc64nvme(napi_env env, napi_callback_info info) {64    napi_value node_args[2];65    size_t num_args = AWS_ARRAY_SIZE(node_args);66    struct aws_byte_buf to_hash;67    AWS_ZERO_STRUCT(to_hash);68    struct aws_byte_buf previous_crc_buf;69    AWS_ZERO_STRUCT(previous_crc_buf);70 71    napi_value node_val = NULL;72 73    if (napi_get_cb_info(env, info, &num_args, node_args, NULL, NULL)) {74        napi_throw_error(env, NULL, "Failed to retrieve callback information");75        goto done;76    }77    if (num_args != AWS_ARRAY_SIZE(node_args)) {78        napi_throw_error(env, NULL, "aws_napi_checksums_crc64 needs exactly 2 arguments");79        goto done;80    }81 82    if (aws_byte_buf_init_from_napi(&to_hash, env, node_args[0])) {83        napi_throw_type_error(env, NULL, "to_hash argument must be a string or array");84        goto done;85    }86    uint8_t *buffer = to_hash.buffer;87    size_t length = to_hash.len;88 89    uint64_t previous_crc = 0;90    if (!aws_napi_is_null_or_undefined(env, node_args[1])) {91        if (aws_byte_buf_init_from_napi(&previous_crc_buf, env, node_args[1])) {92            napi_throw_type_error(env, NULL, "previous_crc argument must be undefined or a DataView over 8 bytes");93            goto done;94        }95        if (previous_crc_buf.len != 8) {96            napi_throw_type_error(env, NULL, "previous_crc argument must be a DataView over 8 bytes");97            goto done;98        }99        previous_crc = aws_read_u64(previous_crc_buf.buffer);100    }101 102    uint64_t val = aws_checksums_crc64nvme_ex(buffer, length, previous_crc);103 104    napi_value arraybuffer;105    void *data = NULL;106    if (napi_create_arraybuffer(env, 8, &data, &arraybuffer)) {107        napi_throw_error(env, NULL, "Failed to create output arraybuffer");108        goto done;109    }110 111    struct aws_byte_buf out_buf = aws_byte_buf_from_empty_array(data, 8);112    aws_byte_buf_write_be64(&out_buf, val);113 114    if (napi_create_dataview(env, 8, arraybuffer, 0, &node_val)) {115        napi_throw_error(env, NULL, "Failed to create output dataview");116        goto done;117    }118 119done:120    aws_byte_buf_clean_up(&to_hash);121    aws_byte_buf_clean_up(&previous_crc_buf);122 123    return node_val;124}125