OpenHarmony-v4.0-Release/s

/*
 * Copyright (c) 2022 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "huks_napi_common_item.h"

#include <string>
#include <vector>

#include "hks_errcode_adapter.h"
#include "hks_log.h"
#include "hks_param.h"
#include "hks_type.h"
#include "securec.h"

namespace HuksNapiItem {
namespace {
constexpr int HKS_MAX_DATA_LEN = 0x6400000; // The maximum length is 100M
constexpr size_t ASYNCCALLBACK_ARGC = 2;
}  // namespace

napi_value ParseKeyAlias(napi_env env, napi_value object, HksBlob *&alias)
{
    napi_valuetype valueType = napi_valuetype::napi_undefined;
    napi_status status = napi_typeof(env, object, &valueType);
    if (status != napi_ok) {
        HKS_LOG_E("could not get object type");
        return nullptr;
    }

    if (valueType != napi_valuetype::napi_string) {
        napi_throw_error(env, std::to_string(HUKS_ERR_CODE_ILLEGAL_ARGUMENT).c_str(), "the type of alias isn't string");
        HKS_LOG_E("no string type");
        return nullptr;
    }

    size_t length = 0;
    status = napi_get_value_string_utf8(env, object, nullptr, 0, &length);
    if (status != napi_ok) {
        HKS_LOG_E("could not get string length");
        return nullptr;
    }

    if (length > HKS_MAX_DATA_LEN) {
        napi_throw_error(env, std::to_string(HUKS_ERR_CODE_ILLEGAL_ARGUMENT).c_str(),
            "the length of alias is too long");
        HKS_LOG_E("input key alias length too large");
        return nullptr;
    }

    char *data = static_cast<char *>(HksMalloc(length + 1));
    if (data == nullptr) {
        HKS_LOG_E("could not alloc memory");
        return nullptr;
    }
    (void)memset_s(data, length + 1, 0, length + 1);

    size_t result = 0;
    status = napi_get_value_string_utf8(env, object, data, length + 1, &result);
    if (status != napi_ok) {
        HksFree(data);
        HKS_LOG_E("could not get string");
        return nullptr;
    }

    alias = static_cast<HksBlob *>(HksMalloc(sizeof(HksBlob)));
    if (alias == nullptr) {
        HksFree(data);
        HKS_LOG_E("could not alloc memory");
        return nullptr;
    }
    alias->data = reinterpret_cast<uint8_t *>(data);
    alias->size = static_cast<uint32_t>(length & UINT32_MAX);

    return GetInt32(env, 0);
}

napi_value GetUint8Array(napi_env env, napi_value object, HksBlob &arrayBlob)
{
    napi_typedarray_type arrayType;
    napi_value arrayBuffer = nullptr;
    size_t length = 0;
    size_t offset = 0;
    void *rawData = nullptr;

    NAPI_CALL(
        env, napi_get_typedarray_info(env, object, &arrayType, &length, &rawData, &arrayBuffer, &offset));
    if (arrayType != napi_uint8_array) {
        napi_throw_error(env, std::to_string(HUKS_ERR_CODE_ILLEGAL_ARGUMENT).c_str(),
            "the type of data is not Uint8Array");
        HKS_LOG_E("the type of  data is not Uint8Array");
        return nullptr;
    }

    if (length > HKS_MAX_DATA_LEN) {
        napi_throw_error(env, std::to_string(HUKS_ERR_CODE_ILLEGAL_ARGUMENT).c_str(),
            "the length of data is too long");
        HKS_LOG_E("data len is too large, len = %" LOG_PUBLIC "zx", length);
        return nullptr;
    }
    if (length == 0) {
        HKS_LOG_I("the created memory length just 1 Byte");
        // the created memory length just 1 Byte
        arrayBlob.data = static_cast<uint8_t *>(HksMalloc(1));
    } else {
        arrayBlob.data = static_cast<uint8_t *>(HksMalloc(length));
    }
    if (arrayBlob.data == nullptr) {
        return nullptr;
    }

    (void)memcpy_s(arrayBlob.data, length, rawData, length);
    arrayBlob.size = static_cast<uint32_t>(length);

    return GetInt32(env, 0);
}

static napi_value CheckParamValueType(napi_env env, uint32_t tag, napi_value value)
{
    napi_value result = nullptr;
    napi_valuetype valueType = napi_valuetype::napi_undefined;
    napi_status status = napi_typeof(env, value, &valueType);
    if (status != napi_ok) {
        HKS_LOG_E("could not get object type");
        return nullptr;
    }

    switch (tag & HKS_TAG_TYPE_MASK) {
        case HKS_TAG_TYPE_INT:
            if (valueType == napi_valuetype::napi_number) {
                result = GetInt32(env, 0);
            }
            break;
        case HKS_TAG_TYPE_UINT:
            if (valueType == napi_valuetype::napi_number) {
                result = GetInt32(env, 0);
            }
            break;
        case HKS_TAG_TYPE_ULONG:
            if (valueType == napi_valuetype::napi_number) {
                result = GetInt32(env, 0);
            }
            break;
        case HKS_TAG_TYPE_BOOL:
            if (valueType == napi_valuetype::napi_boolean) {
                result = GetInt32(env, 0);
            }
            break;
        case HKS_TAG_TYPE_BYTES:
            if (valueType == napi_valuetype::napi_object) {
                result = GetInt32(env, 0);
            }
            break;
        default:
            HKS_LOG_E("invalid tag value 0x%" LOG_PUBLIC "x", tag);
            break;
    }

    if (result == nullptr) {
        napi_throw_error(env, std::to_string(HUKS_ERR_CODE_ILLEGAL_ARGUMENT).c_str(),
            "the value of the tag is of an incorrect type");
        HKS_LOG_E("invalid tag or the type of value, tag = 0x%" LOG_PUBLIC "x, type = %" LOG_PUBLIC "u",
            tag, valueType);
    }
    return result;
}

static napi_value GetHksParam(napi_env env, napi_value object, HksParam &param)
{
    napi_value tag = GetPropertyFromOptions(env, object, HKS_PARAM_PROPERTY_TAG);
    if (tag == nullptr) {
        HKS_LOG_E("get param tag failed");
        return nullptr;
    }
    NAPI_CALL(env, napi_get_value_uint32(env, tag, &param.tag));

    napi_value value = GetPropertyFromOptions(env, object, HKS_PARAM_PROPERTY_VALUE);
    if (value == nullptr) {
        HKS_LOG_E("get param value failed");
        return nullptr;
    }

    if (CheckParamValueType(env, param.tag, value) == nullptr) {
        return nullptr;
    }

    napi_value result = nullptr;
    switch (param.tag & HKS_TAG_TYPE_MASK) {
        case HKS_TAG_TYPE_INT:
            NAPI_CALL(env, napi_get_value_int32(env, value, &param.int32Param));
            result = GetInt32(env, 0);
            break;
        case HKS_TAG_TYPE_UINT:
            NAPI_CALL(env, napi_get_value_uint32(env, value, &param.uint32Param));
            result = GetInt32(env, 0);
            break;
        case HKS_TAG_TYPE_ULONG:
            NAPI_CALL(env, napi_get_value_int64(env, value, reinterpret_cast<int64_t *>(&param.uint64Param)));
            result = GetInt32(env, 0);
            break;
        case HKS_TAG_TYPE_BOOL:
            NAPI_CALL(env, napi_get_value_bool(env, value, &param.boolParam));
            result = GetInt32(env, 0);
            break;
        case HKS_TAG_TYPE_BYTES:
            result = GetUint8Array(env, value, param.blob);
            if (result == nullptr) {
                HKS_LOG_E("get uint8 array fail.");
            } else {
                HKS_LOG_D("tag 0x%" LOG_PUBLIC "x, len 0x%" LOG_PUBLIC "x", param.tag, param.blob.size);
            }
            break;
        default:
            HKS_LOG_E("invalid tag value 0x%" LOG_PUBLIC "x", param.tag);
            break;
    }

    return result;
}

void FreeParsedParams(std::vector<HksParam> &params)
{
    HksParam *param = params.data();
    size_t paramCount = params.size();
    if (param == nullptr) {
        return;
    }
    while (paramCount > 0) {
        paramCount--;
        if ((param->tag & HKS_TAG_TYPE_MASK) == HKS_TAG_TYPE_BYTES) {
            HKS_FREE_PTR(param->blob.data);
            param->blob.size = 0;
        }
        ++param;
    }
}

napi_value ParseParams(napi_env env, napi_value object, std::vector<HksParam> &params)
{
    bool hasNextElement = false;
    napi_value result = nullptr;
    size_t index = 0;
    while ((napi_has_element(env, object, index, &hasNextElement) == napi_ok) && hasNextElement) {
        napi_value element = nullptr;
        NAPI_CALL(env, napi_get_element(env, object, index, &element));

        HksParam param = { 0 };
        result = GetHksParam(env, element, param);
        if (result == nullptr) {
            HKS_LOG_E("get param failed when parse input params.");
            return nullptr;
        }

        params.push_back(param);
        index++;
    }
    return GetInt32(env, 0);
}

static int32_t AddParams(const std::vector<HksParam> &params, struct HksParamSet *&paramSet)
{
    for (auto &param : params) {
        int32_t ret = HksAddParams(paramSet, &param, 1);
        if (ret != HKS_SUCCESS) {
            HKS_LOG_E("add param.tag[%" LOG_PUBLIC "x] failed", param.tag);
            return ret;
        }
    }
    return HKS_SUCCESS;
}

static napi_value ParseHksParamSetOrAddParam(napi_env env, napi_value object, HksParamSet *&paramSet,
    HksParam *addParam)
{
    if (paramSet != nullptr) {
        HKS_LOG_E("param input invalid");
        return nullptr;
    }

    std::vector<HksParam> params;
    HksParamSet *outParamSet = nullptr;
    do {
        if (HksInitParamSet(&outParamSet) != HKS_SUCCESS) {
            HKS_LOG_E("paramset init failed");
            break;
        }

        if (ParseParams(env, object, params) == nullptr) {
            HKS_LOG_E("parse params failed");
            break;
        }

        if (addParam != nullptr) {
            params.push_back(*addParam);
        }

        if (AddParams(params, outParamSet) != HKS_SUCCESS) {
            HKS_LOG_E("add params failed");
            break;
        }

        if (HksBuildParamSet(&outParamSet) != HKS_SUCCESS) {
            HKS_LOG_E("HksBuildParamSet failed");
            break;
        }

        FreeParsedParams(params);
        paramSet = outParamSet;
        return GetInt32(env, 0);
    } while (0);

    HksFreeParamSet(&outParamSet);
    FreeParsedParams(params);
    return nullptr;
}

static napi_value ParseHksParamSet(napi_env env, napi_value object, HksParamSet *&paramSet)
{
    return ParseHksParamSetOrAddParam(env, object, paramSet, nullptr);
}

napi_value ParseHksParamSetAndAddParam(napi_env env, napi_value object, HksParamSet *&paramSet, HksParam *addParam)
{
    return ParseHksParamSetOrAddParam(env, object, paramSet, addParam);
}

napi_ref GetCallback(napi_env env, napi_value object)
{
    napi_valuetype valueType = napi_valuetype::napi_undefined;
    napi_status status = napi_typeof(env, object, &valueType);
    if (status != napi_ok) {
        HKS_LOG_E("could not get object type");
        return nullptr;
    }

    if (valueType != napi_valuetype::napi_function) {
        HKS_LOG_I("no callback fun, process as promise func");
        return nullptr;
    }

    napi_ref ref = nullptr;
    status = napi_create_reference(env, object, 1, &ref);
    if (status != napi_ok) {
        HKS_LOG_E("could not create reference");
        return nullptr;
    }
    return ref;
}

static napi_value GenerateAarrayBuffer(napi_env env, uint8_t *data, uint32_t size)
{
    uint8_t *buffer = static_cast<uint8_t *>(HksMalloc(size));
    if (buffer == nullptr) {
        return nullptr;
    }

    napi_value outBuffer = nullptr;
    (void)memcpy_s(buffer, size, data, size);

    napi_status status = napi_create_external_arraybuffer(
        env, buffer, size, [](napi_env env, void *data, void *hint) { HksFree(data); }, nullptr, &outBuffer);
    if (status == napi_ok) {
        // free by finalize callback
        buffer = nullptr;
    } else {
        HksFree(buffer);
    }

    return outBuffer;
}

static napi_value GenerateHksParam(napi_env env, const HksParam &param)
{
    napi_value hksParam = nullptr;
    NAPI_CALL(env, napi_create_object(env, &hksParam));

    napi_value tag = nullptr;
    NAPI_CALL(env, napi_create_uint32(env, param.tag, &tag));
    NAPI_CALL(env, napi_set_named_property(env, hksParam, HKS_PARAM_PROPERTY_TAG.c_str(), tag));

    napi_value value = nullptr;
    switch (param.tag & HKS_TAG_TYPE_MASK) {
        case HKS_TAG_TYPE_INT:
            NAPI_CALL(env, napi_create_int32(env, param.int32Param, &value));
            break;
        case HKS_TAG_TYPE_UINT:
            NAPI_CALL(env, napi_create_uint32(env, param.uint32Param, &value));
            break;
        case HKS_TAG_TYPE_ULONG:
            NAPI_CALL(env, napi_create_int64(env, param.uint64Param, &value));
            break;
        case HKS_TAG_TYPE_BOOL:
            NAPI_CALL(env, napi_get_boolean(env, param.boolParam, &value));
            break;
        case HKS_TAG_TYPE_BYTES:
            value = GenerateAarrayBuffer(env, param.blob.data, param.blob.size);
            break;
        default:
            value = GetNull(env);
            break;
    }
    NAPI_CALL(env, napi_set_named_property(env, hksParam, HKS_PARAM_PROPERTY_VALUE.c_str(), value));

    return hksParam;
}

static napi_value GenerateHksParamArray(napi_env env, const HksParamSet &paramSet)
{
    napi_value paramArray = nullptr;
    NAPI_CALL(env, napi_create_array(env, &paramArray));

    for (uint32_t i = 0; i < paramSet.paramsCnt; i++) {
        napi_value element = nullptr;
        element = GenerateHksParam(env, paramSet.params[i]);
        napi_set_element(env, paramArray, i, element);
    }

    return paramArray;
}

static napi_value GenerateStringArray(napi_env env, const struct HksBlob *blob, uint32_t blobCount)
{
    if (blobCount == 0 || blob == nullptr) {
        return nullptr;
    }
    napi_value array = nullptr;
    NAPI_CALL(env, napi_create_array(env, &array));
    for (uint32_t i = 0; i < blobCount; i++) {
        napi_value element = nullptr;
        napi_create_string_latin1(env, (const char *)blob[i].data, blob[i].size, &element);
        napi_set_element(env, array, i, element);
    }
    return array;
}

void FreeHksCertChain(HksCertChain *&certChain)
{
    if (certChain == nullptr) {
        return;
    }

    if (certChain->certsCount > 0 && certChain->certs != nullptr) {
        for (uint32_t i = 0; i < certChain->certsCount; i++) {
            if (certChain->certs[i].data != nullptr) {
                HksFree(certChain->certs[i].data);
                certChain->certs[i].data = nullptr;
            }
        }
    }

    HksFree(certChain);
    certChain = nullptr;
}

napi_value GetHandleValue(napi_env env, napi_value object, struct HksBlob *&handleBlob)
{
    if (handleBlob != nullptr) {
        HKS_LOG_E("param input invalid");
        return nullptr;
    }

    napi_valuetype valueType = napi_valuetype::napi_undefined;
    napi_status status = napi_typeof(env, object, &valueType);
    if (status != napi_ok) {
        HKS_LOG_E("could not get object type");
        return nullptr;
    }

    if (valueType != napi_valuetype::napi_number) {
        napi_throw_error(env, std::to_string(HUKS_ERR_CODE_ILLEGAL_ARGUMENT).c_str(),
            "the type of handle isn't number");
        return nullptr;
    }

    uint32_t handleTmp = 0;
    status = napi_get_value_uint32(env, object, &handleTmp);
    if (status != napi_ok) {
        HKS_LOG_E("Retrieve field failed");
        return nullptr;
    }

    uint64_t handle = static_cast<uint64_t>(handleTmp);

    handleBlob = static_cast<struct HksBlob *>(HksMalloc(sizeof(struct HksBlob)));
    if (handleBlob == nullptr) {
        HKS_LOG_E("could not alloc memory");
        return nullptr;
    }

    handleBlob->data = static_cast<uint8_t *>(HksMalloc(sizeof(uint64_t)));
    if (handleBlob->data == nullptr) {
        HKS_FREE_PTR(handleBlob);
        HKS_LOG_E("could not alloc memory");
        return nullptr;
    }
    handleBlob->size = sizeof(uint64_t);

    if (memcpy_s(handleBlob->data, sizeof(uint64_t), &handle, sizeof(uint64_t)) != EOK) {
        // the memory of handleBlob free by finalize callback
        return nullptr;
    }

    return GetInt32(env, 0);
}

void DeleteCommonAsyncContext(napi_env env, napi_async_work &asyncWork, napi_ref &callback,
    struct HksBlob *&blob, struct HksParamSet *&paramSet)
{
    if (asyncWork != nullptr) {
        napi_delete_async_work(env, asyncWork);
        asyncWork = nullptr;
    }

    if (callback != nullptr) {
        napi_delete_reference(env, callback);
        callback = nullptr;
    }

    if (blob != nullptr) {
        FreeHksBlob(blob);
    }

    if (paramSet != nullptr) {
        HksFreeParamSet(&paramSet);
    }
}

napi_value GetPropertyFromOptions(napi_env env, napi_value value, const std::string propertyStr)
{
    napi_valuetype valueType = napi_valuetype::napi_undefined;
    napi_status status = napi_typeof(env, value, &valueType);
    if (status != napi_ok) {
        HKS_LOG_E("could not get object type");
        return nullptr;
    }

    if (valueType != napi_valuetype::napi_object) {
        napi_throw_error(env, std::to_string(HUKS_ERR_CODE_ILLEGAL_ARGUMENT).c_str(),
            "the type is not object");
        HKS_LOG_E("no object type");
        return nullptr;
    }

    napi_value property = nullptr;
    status = napi_get_named_property(env, value,
        propertyStr.c_str(), &property);
    if (status != napi_ok || property == nullptr) {
        napi_throw_error(env, std::to_string(HUKS_ERR_CODE_ILLEGAL_ARGUMENT).c_str(),
            "get value failed, maybe the target does not exist");
        HKS_LOG_E("could not get property %" LOG_PUBLIC "s", propertyStr.c_str());
        return nullptr;
    }

    return property;
}

static napi_value ParseGetHksParamSet(napi_env env, napi_value value, HksParamSet *&paramSet)
{
    napi_value property = GetPropertyFromOptions(env, value, HKS_OPTIONS_PROPERTY_PROPERTIES);
    if (property == nullptr) {
        return nullptr;
    }

    napi_value result = ParseHksParamSet(env, property, paramSet);
    if (result == nullptr) {
        HKS_LOG_E("could not get paramset");
        return nullptr;
    }

    return GetInt32(env, 0);
}

napi_value ParseHandleAndHksParamSet(napi_env env, napi_value *argv, size_t &index,
    HksBlob *&handleBlob, HksParamSet *&paramSet)
{
    // the index is controlled by the caller and needs to ensure that it does not overflow
    if (argv == nullptr || handleBlob != nullptr || paramSet != nullptr) {
        HKS_LOG_E("param input invalid");
        return nullptr;
    }

    napi_value result = GetHandleValue(env, argv[index], handleBlob);
    if (result == nullptr) {
        HKS_LOG_E("could not get handle value");
        return nullptr;
    }

    index++;
    result = ParseGetHksParamSet(env, argv[index], paramSet);
    if (result == nullptr) {
        HKS_LOG_E("could not get hksParamSet");
        return nullptr;
    }

    return GetInt32(env, 0);
}

napi_value ParseKeyAliasAndHksParamSet(napi_env env, napi_value *argv, size_t &index,
    HksBlob *&keyAliasBlob, HksParamSet *&paramSet)
{
    // the index is controlled by the caller and needs to ensure that it does not overflow
    if (argv == nullptr || keyAliasBlob != nullptr || paramSet != nullptr) {
        HKS_LOG_E("param input invalid");
        return nullptr;
    }

    napi_value result = ParseKeyAlias(env, argv[index], keyAliasBlob);
    if (result == nullptr) {
        HKS_LOG_E("could not get keyAlias");
        return nullptr;
    }

    index++;
    result = ParseGetHksParamSet(env, argv[index], paramSet);
    if (result == nullptr) {
        HKS_LOG_E("get hksParamSet failed");
        return nullptr;
    }

    return GetInt32(env, 0);
}

napi_value ParseKeyData(napi_env env, napi_value value, HksBlob *&keyDataBlob)
{
    if (keyDataBlob != nullptr) {
        HKS_LOG_E("param input invalid");
        return nullptr;
    }

    napi_value inData = GetPropertyFromOptions(env, value, HKS_OPTIONS_PROPERTY_INDATA);
    if (inData == nullptr) {
        HKS_LOG_E("get indata property failed");
        return nullptr;
    }

    keyDataBlob = static_cast<struct HksBlob *>(HksMalloc(sizeof(HksBlob)));
    if (keyDataBlob == nullptr) {
        HKS_LOG_E("could not alloc memory");
        return nullptr;
    }
    (void)memset_s(keyDataBlob, sizeof(HksBlob), 0, sizeof(HksBlob));

    if (GetUint8Array(env, inData, *keyDataBlob) == nullptr) {
        FreeHksBlob(keyDataBlob);
        HKS_LOG_E("could not get indata");
        return nullptr;
    }

    return GetInt32(env, 0);
}

static napi_value AddHandleOrChallenge(napi_env env, napi_value &object,
    const struct HksBlob *handle, const struct HksBlob *challenge)
{
    napi_value addResult = nullptr;

    // add handle
    if ((handle != nullptr) && (handle->data != nullptr) && (handle->size == sizeof(uint64_t))) {
        void *handleData = static_cast<void *>(handle->data);
        uint64_t tempHandle = *(static_cast<uint64_t *>(handleData));
        uint32_t handleValue = static_cast<uint32_t>(tempHandle); /* Temporarily only use 32 bit handle */
        napi_value handlejs = nullptr;
        NAPI_CALL(env, napi_create_uint32(env, handleValue, &handlejs));
        NAPI_CALL(env, napi_set_named_property(env, object, HKS_HANDLE_PROPERTY_HANDLE.c_str(), handlejs));
        addResult = GetInt32(env, 0);
    }

    // add challenge
    if ((challenge != nullptr) && (challenge->size != 0) && (challenge->data != nullptr)) {
        napi_value challengejs = nullptr;
        napi_value outBuffer = GenerateAarrayBuffer(env, challenge->data, challenge->size);
        if (outBuffer == nullptr) {
            HKS_LOG_E("add token failed");
            return nullptr;
        }

        NAPI_CALL(env, napi_create_typedarray(env, napi_uint8_array, challenge->size, outBuffer, 0, &challengejs));
        NAPI_CALL(env, napi_set_named_property(env, object, HKS_HANDLE_PROPERTY_CHALLENGE.c_str(), challengejs));
        addResult = GetInt32(env, 0);
    }

    return addResult;
}

static napi_value AddOutDataParamSetOrCertChain(napi_env env, napi_value &object,
    const struct HksBlob *outData, const HksParamSet *paramSet, const struct HksCertChain *certChain)
{
    napi_value addResult = nullptr;

    // add outData
    if ((outData != nullptr) && (outData->data != nullptr) && (outData->size != 0)) {
        napi_value outDataJs = nullptr;
        napi_value outBuffer = GenerateAarrayBuffer(env, outData->data, outData->size);
        if (outBuffer == nullptr) {
            HKS_LOG_E("add outData failed");
            return nullptr;
        }
        NAPI_CALL(env, napi_create_typedarray(env, napi_uint8_array, outData->size, outBuffer, 0, &outDataJs));
        NAPI_CALL(env, napi_set_named_property(env, object, HKS_RESULT_PROPERTY_OUTDATA.c_str(), outDataJs));
        addResult = GetInt32(env, 0);
    }

    // add paramSet
    if (paramSet != nullptr) {
        napi_value properties = nullptr;
        properties = GenerateHksParamArray(env, *paramSet);
        if (properties == nullptr) {
            HKS_LOG_E("add paramSet failed");
            return nullptr;
        }
        NAPI_CALL(env, napi_set_named_property(env, object, HKS_RESULT_PRPPERTY_PROPERTIES.c_str(), properties));
        addResult = GetInt32(env, 0);
    }

    // add certChain
    if ((certChain != nullptr) && (certChain->certs != nullptr) && (certChain->certsCount != 0)) {
        napi_value certChainJs = GenerateStringArray(env, certChain->certs, certChain->certsCount);
        if (certChainJs == nullptr) {
            HKS_LOG_E("add certChain failed");
            return nullptr;
        }
        NAPI_CALL(env, napi_set_named_property(env, object, HKS_RESULT_PRPPERTY_CERTCHAINS.c_str(), certChainJs));
        addResult = GetInt32(env, 0);
    }

    return addResult;
}

static napi_value GenerateResult(napi_env env, const struct HksSuccessReturnResult resultData)
{
    napi_value result = nullptr;

    if (resultData.isOnlyReturnBoolResult) {
        if (napi_get_boolean(env, true, &result) != napi_ok) {
            return GetNull(env);
        }
        return result;
    }

    if (napi_create_object(env, &result) != napi_ok) {
        return GetNull(env);
    }

    napi_value status1 = AddHandleOrChallenge(env, result, resultData.handle, resultData.challenge);
    napi_value status2 = AddOutDataParamSetOrCertChain(env, result,
        resultData.outData, resultData.paramSet, resultData.certChain);
    if (status1 == nullptr && status2 == nullptr) {
        return GetNull(env);
    }

    return result;
}

static napi_value GenerateBusinessError(napi_env env, int32_t errorCode)
{
    napi_value businessError = nullptr;
    napi_status status = napi_create_object(env, &businessError);
    if (status != napi_ok) {
        HKS_LOG_E("create object failed");
        return GetNull(env);
    }

    struct HksResult errInfo = HksConvertErrCode(errorCode);
    // add errorCode
    napi_value code = nullptr;
    status = napi_create_int32(env, errInfo.errorCode, &code);
    if (status != napi_ok || code == nullptr) {
        code = GetNull(env);
    }
    status = napi_set_named_property(env, businessError, BUSINESS_ERROR_PROPERTY_CODE.c_str(), code);
    if (status != napi_ok) {
        HKS_LOG_E("set errorCode failed");
        return GetNull(env);
    }

    // add errorMessage
    napi_value msg = GetNull(env);
    if (errInfo.errorMsg != nullptr) {
#ifdef HUKS_NAPI_ERRORCODE_WITH_MESSAGE
        uint32_t errorMsgLen = strlen(errInfo.errorMsg);
        uint8_t errorMsgBuf[errorMsgLen];
        (void)memcpy_s(errorMsgBuf, errorMsgLen, errInfo.errorMsg, errorMsgLen);
        struct HksBlob msgBlob = { errorMsgLen, errorMsgBuf };
        msg = GenerateStringArray(env, &msgBlob, sizeof(msgBlob) / sizeof(struct HksBlob));
        msg = ((msg == nullptr) ? GetNull(env) : msg);
#endif
    }
    status = napi_set_named_property(env, businessError, BUSINESS_ERROR_PROPERTY_MESSAGE.c_str(), msg);
    if (status != napi_ok) {
        HKS_LOG_E("set errorMsg failed");
        return GetNull(env);
    }

    // add errorData
    napi_value data = GetNull(env);
    status = napi_set_named_property(env, businessError, BUSINESS_ERROR_PROPERTY_DATA.c_str(), data);
    if (status != napi_ok) {
        HKS_LOG_E("set errorData failed");
        return GetNull(env);
    }

    return businessError;
}

static void CallbackResultFailure(napi_env env, napi_ref callback, int32_t error)
{
    if (error == HKS_SUCCESS) {
        HKS_LOG_E("callback: failure cannot be executed");
        return;
    }

    napi_value func = nullptr;
    NAPI_CALL_RETURN_VOID(env, napi_get_reference_value(env, callback, &func));

    napi_value recv = nullptr;
    napi_value result = nullptr;
    NAPI_CALL_RETURN_VOID(env, napi_get_undefined(env, &recv));

    napi_value businessError = GenerateBusinessError(env, error);
    NAPI_CALL_RETURN_VOID(env, napi_call_function(env, recv, func, 1, &businessError, &result));
}

static void CallbackResultSuccess(napi_env env, napi_ref callback, const struct HksSuccessReturnResult resultData)
{
    napi_value params[ASYNCCALLBACK_ARGC] = { GetNull(env), GenerateResult(env, resultData) };
    napi_value func = nullptr;
    NAPI_CALL_RETURN_VOID(env, napi_get_reference_value(env, callback, &func));

    napi_value recv = nullptr;
    napi_value result = nullptr;
    NAPI_CALL_RETURN_VOID(env, napi_get_undefined(env, &recv));
    NAPI_CALL_RETURN_VOID(env, napi_call_function(env, recv, func, ASYNCCALLBACK_ARGC, params, &result));
}

static void PromiseResultFailure(napi_env env, napi_deferred deferred, int32_t error)
{
    if (error == HKS_SUCCESS) {
        HKS_LOG_E("promise: failure cannot be executed");
        return;
    }
    napi_value result = nullptr;
    result = GenerateBusinessError(env, error);
    napi_reject_deferred(env, deferred, result);
}

static void PromiseResultSuccess(napi_env env, napi_deferred deferred,
    const struct HksSuccessReturnResult resultData)
{
    napi_value result = nullptr;
    result = GenerateResult(env, resultData);
    napi_resolve_deferred(env, deferred, result);
}

void SuccessReturnResultInit(struct HksSuccessReturnResult &resultData)
{
    resultData.isOnlyReturnBoolResult = false;
    resultData.handle = nullptr;
    resultData.challenge = nullptr;
    resultData.outData = nullptr;
    resultData.paramSet = nullptr;
    resultData.certChain = nullptr;
}

void HksReturnNapiResult(napi_env env, napi_ref callback, napi_deferred deferred, int32_t errorCode,
    const struct HksSuccessReturnResult resultData)
{
    if (callback == nullptr) {
        if (errorCode == HKS_SUCCESS) {
            PromiseResultSuccess(env, deferred, resultData);
        } else {
            PromiseResultFailure(env, deferred, errorCode);
        }
    } else {
        if (errorCode == HKS_SUCCESS) {
            CallbackResultSuccess(env, callback, resultData);
        } else {
            CallbackResultFailure(env, callback, errorCode);
        }
    }
}
}  // namespace HuksNapiItem