/*
* Copyright (c) 2025 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 "dm_auth_attest_common.h"

#include "dm_error_type.h"
#include "dm_log.h"

namespace OHOS {
namespace DistributedHardware {

constexpr int32_t HEX_TO_UINT8 = 2;

AuthAttestCommon::AuthAttestCommon()
{
    LOGD("AuthAttestCommon constructor");
}

AuthAttestCommon::~AuthAttestCommon()
{
    LOGD("AuthAttestCommon destructor");
}

std::string AuthAttestCommon::SerializeDmCertChain(const DmCertChain *chain)
{
    if (chain == nullptr || chain->cert == nullptr || chain->certCount != DM_CERTS_COUNT) {
        LOGE("input param is invalid.");
        return "{}";
    }
    JsonObject jsonObject;
    jsonObject[TAG_CERT_COUNT] = chain->certCount;
    JsonObject jsonArrayObj(JsonCreateType::JSON_CREATE_TYPE_ARRAY);
    for (uint32_t i = 0; i < chain->certCount; ++i) {
        const DmBlob &blob = chain->cert[i];
        if (blob.data == nullptr || blob.size == 0 || blob.size > DM_CERTIFICATE_SIZE) {
            LOGE("Invalid blob: null data or invalid size.");
            return "{}";
        }
        const uint32_t hexLen = blob.size * HEX_TO_UINT8 + 1; // 2*blob.size + 1
        char *hexBuffer = new char[hexLen]{0};
        if (hexBuffer == nullptr) {
            LOGE("hexBuffer malloc failed.");
            return "{}";
        }
        int32_t ret = Crypto::ConvertBytesToHexString(hexBuffer, hexLen, blob.data, blob.size);
        if (ret != DM_OK) {
            LOGE("ConvertBytesToHexString failed.");
            delete[] hexBuffer;
            return "{}";
        }
        jsonArrayObj.PushBack(std::string(hexBuffer));
        delete[] hexBuffer;
    }
    jsonObject.Insert(TAG_CERT, jsonArrayObj);
    return jsonObject.Dump();
}

//LCOV_EXCL_START
bool ValidateInputJson(const std::string &data)
{
    JsonObject jsonObject;
    jsonObject.Parse(data);
    if (!IsUint32(jsonObject, TAG_CERT_COUNT) || !jsonObject.Contains(TAG_CERT)) {
        LOGE("Missing required fields 'certCount' or 'cert'");
        return false;
    }
    const uint32_t certCount = jsonObject[TAG_CERT_COUNT].Get<uint32_t>();
    if (certCount != DM_CERTS_COUNT) {
        LOGE("Invalid certCount value %{public}u", certCount);
        return false;
    }
    JsonObject jsonArrayObj(JsonCreateType::JSON_CREATE_TYPE_ARRAY);
    jsonArrayObj.Parse(jsonObject[TAG_CERT].Dump());
    const uint32_t certSize = jsonArrayObj.Items().size();
    if (certSize != certCount) {
        LOGE("certSize = %{public}u is invalid.", certSize);
        return false;
    }
    return true;
}

 bool ProcessCertItem(const JsonItemObject &item, DmBlob &cert, uint32_t processedIndex)
 {
    if (!item.IsString()) {
        LOGE("input param is invalid.");
        return false;
    }
    std::string hexStr = item.Get<std::string>();
    const size_t hexLen = hexStr.length();
    if (hexLen == 0 || hexLen % HEX_TO_UINT8 != 0) {
        LOGE("Invalid HEX length %{public}zu at index %{public}u", hexLen, processedIndex);
        return false;
    }
    const uint32_t binSize = hexLen / HEX_TO_UINT8;
    if (binSize > DM_CERTIFICATE_SIZE) {
        LOGE("binSize = %{public}u is invalid.", binSize);
        return false;
    }
    cert.data = new uint8_t[binSize]{0};
    if (cert.data == nullptr) {
        LOGE("Data allocation failed at index %{public}u", processedIndex);
        return false;
    }
    int32_t ret = Crypto::ConvertHexStringToBytes(cert.data, binSize, hexStr.c_str(), hexLen);
    if (ret != DM_OK) {
        LOGE("HEX conversion failed at index %{public}u, ret = %{public}d",
            processedIndex, ret);
        delete[] cert.data;
        cert.data = nullptr;
        return false;
    }
    cert.size = binSize;
    return true;
}

bool AuthAttestCommon::DeserializeDmCertChain(const std::string &data, DmCertChain *outChain)
{
    if (outChain == nullptr || data.empty() || !ValidateInputJson(data)) {
        LOGE("Invalid input");
        return false;
    }
    JsonObject jsonObject;
    jsonObject.Parse(data);
    const uint32_t certCount = jsonObject[TAG_CERT_COUNT].Get<uint32_t>();
    JsonObject jsonArrayObj(JsonCreateType::JSON_CREATE_TYPE_ARRAY);
    jsonArrayObj.Parse(jsonObject[TAG_CERT].Dump());
    DmBlob *certs = new DmBlob[certCount];
    if (certs == nullptr) {
        LOGE("Memory allocation failed for certs array");
        return false;
    }
    if (memset_s(certs, sizeof(DmBlob) * certCount, 0, sizeof(DmBlob) * certCount) != DM_OK) {
        LOGE("memset_s failed.");
        delete[] certs;
        certs = nullptr;
        return false;
    }
    bool success = true;
    uint32_t processedIndex = 0;
    for (const auto &item : jsonArrayObj.Items()) {
        if (!ProcessCertItem(item, certs[processedIndex], processedIndex)) {
            success = false;
            break;
        }
        processedIndex++;
    }
    if (!success) {
        for (uint32_t i = 0; i < processedIndex; ++i) {
            delete[] certs[i].data;
        }
        delete[] certs;
        return false;
    }
    FreeDmCertChain(*outChain);
    outChain->cert = certs;
    outChain->certCount = certCount;
    return true;
}

void AuthAttestCommon::FreeDmCertChain(DmCertChain &chain)
{
    if (chain.cert == nullptr) {
        return;
    }
    for (uint32_t i = 0; i < chain.certCount; ++i) {
        if (chain.cert[i].data != nullptr) {
            delete[] chain.cert[i].data;
            chain.cert[i].data = nullptr;
        }
        chain.cert[i].size = 0;
    }
    delete[] chain.cert;
    chain.cert = nullptr;
    chain.certCount = 0;
}
//LCOV_EXCL_STOP
} // namespace DistributedHardware
} // namespace OHOS