OpenHarmony-v3.2-Release/s

/*
 * Copyright (C) 2021 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 "iso_server_bind_exchange_task.h"
#include "das_task_common.h"
#include "das_module_defines.h"
#include "hc_log.h"
#include "hc_types.h"
#include "iso_protocol_common.h"
#include "iso_task_common.h"
#include "protocol_common.h"

enum {
    TASK_TYPE_BEGIN = 1,
    TASK_TYPE_FINAL,
};

static CurTaskType GetTaskType(void)
{
    return TASK_TYPE_BIND_LITE_EXCHANGE;
}

void DestroyServerBindExchangeTask(struct SymBaseCurTaskT *task)
{
    HcFree(task);
}

static int Process(struct SymBaseCurTaskT *task, IsoParams *params, const CJson *in, CJson *out, int32_t *status)
{
    IsoServerBindExchangeTask *realTask = (IsoServerBindExchangeTask *)task;
    if (realTask->taskBase.taskStatus < TASK_TYPE_BEGIN) {
        LOGE("task status failed");
        return HC_ERR_BAD_MESSAGE;
    }

    if (realTask->taskBase.taskStatus > TASK_TYPE_BEGIN) {
        LOGI("The message is repeated, ignore it, status: %d", realTask->taskBase.taskStatus);
        *status = IGNORE_MSG;
        return HC_SUCCESS;
    }
    int res;
    int32_t message = 0;
    res = GetIntFromJson(in, FIELD_MESSAGE, &message);
    if (res != 0 || message != ISO_CLIENT_BIND_EXCHANGE_CONFIRM) {
        return HC_ERR_BAD_MESSAGE;
    }
    res = CheckEncResult(params, in, RESULT_AAD);
    if (res != 0) {
        LOGE("CheckEncResult failed, res:%d", res);
        DeleteAuthCode(params);
        return res;
    }
    res = SendResultToFinalSelf(params, out, false);
    if (res != 0) {
        LOGE("SendResultToFinalSelf failed, res:%d", res);
        DeleteAuthCode(params);
        return res;
    }
    realTask->taskBase.taskStatus = TASK_TYPE_FINAL;
    *status = FINISH;
    return res;
}

static int DecryptChallenge(const IsoParams *params, const CJson *in, uint8_t *challenge, uint32_t challengeLen)
{
    (void)challengeLen;
    int res;
    uint8_t *encData = NULL;
    uint8_t *nonce = NULL;

    encData = (uint8_t *)HcMalloc(ENC_CHALLENGE_LEN, 0);
    if (encData == NULL) {
        res = HC_ERR_ALLOC_MEMORY;
        goto ERR;
    }
    GOTO_ERR_AND_SET_RET(GetByteFromJson(in, FIELD_ENC_DATA, encData, ENC_CHALLENGE_LEN), res);
    nonce = (uint8_t *)HcMalloc(NONCE_SIZE, 0);
    if (nonce == NULL) {
        res = HC_ERR_ALLOC_MEMORY;
        goto ERR;
    }
    GOTO_ERR_AND_SET_RET(GetByteFromJson(in, FIELD_NONCE, nonce, NONCE_SIZE), res);
    Uint8Buff encDataBuf = { encData, ENC_CHALLENGE_LEN };
    GcmParam gcmParam;
    gcmParam.aad = (uint8_t *)EXCHANGE_AAD;
    gcmParam.aadLen = (uint32_t)strlen(EXCHANGE_AAD);
    gcmParam.nonce = nonce;
    gcmParam.nonceLen = NONCE_SIZE;
    Uint8Buff challengeBuf = { challenge, CHALLENGE_SIZE };
    res = params->baseParams.loader->aesGcmDecrypt(&params->baseParams.sessionKey, &encDataBuf, &gcmParam, false,
        &challengeBuf);
    if (res != 0) {
        LOGE("decrypt challenge failed, res:%d", res);
        goto ERR;
    }
ERR:
    HcFree(encData);
    HcFree(nonce);
    return res;
}

static int GenAndEncAuthCode(const IsoParams *params, Uint8Buff *nonceBuf, const Uint8Buff *challengeBuf,
    Uint8Buff *encAuthCodeBuf)
{
    int res;
    uint8_t *keyAlias = NULL;
    uint8_t *authCode = (uint8_t *)HcMalloc(AUTH_CODE_LEN, 0);
    if (authCode == NULL) {
        res = HC_ERR_ALLOC_MEMORY;
        goto ERR;
    }
    Uint8Buff authCodeBuf = { authCode, AUTH_CODE_LEN };
    res = params->baseParams.loader->generateRandom(&authCodeBuf);
    if (res != 0) {
        LOGE("generate auth code failed, res:%d", res);
        goto ERR;
    }

    res = params->baseParams.loader->generateRandom(nonceBuf);
    if (res != 0) {
        LOGE("generate nonce failed, res:%d", res);
        goto ERR;
    }
    GcmParam gcmParam = { nonceBuf->val, nonceBuf->length, challengeBuf->val, challengeBuf->length };
    res = params->baseParams.loader->aesGcmEncrypt(&params->baseParams.sessionKey, &authCodeBuf, &gcmParam, false,
        encAuthCodeBuf);
    if (res != 0) {
        LOGE("encrypt auth code failed, res:%d", res);
        goto ERR;
    }

    keyAlias = (uint8_t *)HcMalloc(ISO_KEY_ALIAS_LEN, 0);
    if (keyAlias == NULL) {
        res = HC_ERR_ALLOC_MEMORY;
        goto ERR;
    }
    res = GenerateKeyAliasInIso(params, keyAlias, ISO_KEY_ALIAS_LEN, true);
    if (res != 0) {
        LOGE("GenerateKeyAliasInIso failed, res:%d", res);
        goto ERR;
    }

    LOGI("AuthCode alias: %x%x%x%x****.", keyAlias[0], keyAlias[1], keyAlias[2], keyAlias[3]);
    Uint8Buff keyAliasBuf = { keyAlias, ISO_KEY_ALIAS_LEN };
    ExtraInfo exInfo = { { params->baseParams.authIdPeer.val, params->baseParams.authIdPeer.length },
        params->peerUserType, PAIR_TYPE_BIND };
    res = params->baseParams.loader->importSymmetricKey(&keyAliasBuf, &authCodeBuf, KEY_PURPOSE_MAC, &exInfo);
    if (res != 0) {
        LOGE("ImportSymmetricKey failed, res: %x.", res);
        goto ERR;
    }
ERR:
    HcFree(keyAlias);
    FreeAndCleanKey(&authCodeBuf);
    return res;
}

static int GenerateAuthCodeAndImport(const IsoParams *params, CJson *out, uint8_t *challenge, uint32_t challengeLen)
{
    int res;
    uint8_t *nonce = NULL;
    uint8_t *encAuthCode = NULL;
    CJson *payload = NULL;
    CJson *sendToPeer = NULL;

    nonce = (uint8_t *)HcMalloc(NONCE_SIZE, 0);
    if (nonce == NULL) {
        res = HC_ERR_ALLOC_MEMORY;
        goto ERR;
    }
    encAuthCode = (uint8_t *)HcMalloc(AUTH_CODE_LEN + TAG_LEN, 0);
    if (encAuthCode == NULL) {
        res = HC_ERR_ALLOC_MEMORY;
        goto ERR;
    }
    Uint8Buff encAuthCodeBuf = { encAuthCode, AUTH_CODE_LEN + TAG_LEN };
    Uint8Buff nonceBuf = { nonce, NONCE_SIZE };
    Uint8Buff challengeBuf = { challenge, challengeLen };

    res = GenAndEncAuthCode(params, &nonceBuf, &challengeBuf, &encAuthCodeBuf);
    if (res != 0) {
        LOGE("GenAndEncAuthCode failed, res:%d", res);
        goto ERR;
    }
    payload = CreateJson();
    if (payload == NULL) {
        res = HC_ERR_ALLOC_MEMORY;
        goto ERR;
    }
    sendToPeer = CreateJson();
    if (sendToPeer == NULL) {
        res = HC_ERR_ALLOC_MEMORY;
        goto ERR;
    }
    GOTO_ERR_AND_SET_RET(AddIntToJson(sendToPeer, FIELD_MESSAGE, ISO_SERVER_BIND_EXCHANGE_RET), res);
    GOTO_ERR_AND_SET_RET(AddByteToJson(payload, FIELD_ENC_AUTH_TOKEN, encAuthCodeBuf.val, encAuthCodeBuf.length), res);
    GOTO_ERR_AND_SET_RET(AddByteToJson(payload, FIELD_NONCE, nonceBuf.val, nonceBuf.length), res);
    GOTO_ERR_AND_SET_RET(AddObjToJson(sendToPeer, FIELD_PAYLOAD, payload), res);
    GOTO_ERR_AND_SET_RET(AddObjToJson(out, FIELD_SEND_TO_PEER, sendToPeer), res);

ERR:
    HcFree(nonce);
    HcFree(encAuthCode);
    FreeJson(payload);
    FreeJson(sendToPeer);
    return res;
}

static int ServerBindExchangeStart(const IsoParams *params, IsoServerBindExchangeTask *task,
    const CJson *in, CJson *out, int32_t *status)
{
    int res;
    uint8_t *challenge = (uint8_t *)HcMalloc(CHALLENGE_SIZE, 0);
    if (challenge == NULL) {
        res = HC_ERR_ALLOC_MEMORY;
        goto ERR;
    }
    res = DecryptChallenge(params, in, challenge, CHALLENGE_SIZE);
    if (res != 0) {
        LOGE("decrypt challenge failed, res:%d", res);
        goto ERR;
    }
    res = GenerateAuthCodeAndImport(params, out, challenge, CHALLENGE_SIZE);
    if (res != 0) {
        LOGE("GenerateAuthCodeAndImport failed, res:%d", res);
        goto ERR;
    }
    task->taskBase.taskStatus = TASK_TYPE_BEGIN;
    *status = CONTINUE;
ERR:
    HcFree(challenge);
    return res;
}

SymBaseCurTask *CreateServerBindExchangeTask(IsoParams *params, const CJson *in, CJson *out, int32_t *status)
{
    IsoServerBindExchangeTask *task = (IsoServerBindExchangeTask *)HcMalloc(sizeof(IsoServerBindExchangeTask), 0);
    if (task == NULL) {
        return NULL;
    }
    task->taskBase.destroyTask = DestroyServerBindExchangeTask;
    task->taskBase.process = Process;
    task->taskBase.getCurTaskType = GetTaskType;
    int res = ServerBindExchangeStart(params, task, in, out, status);
    if (res != 0) {
        DestroyServerBindExchangeTask((struct SymBaseCurTaskT *)task);
        return NULL;
    }
    return (SymBaseCurTask *)task;
}