/* SPDX-License-Identifier: BSD-2-Clause */
/*
 * Copyright (c) 2015 - 2018 Intel Corporation
 * All rights reserved.
 */

#ifdef HAVE_CONFIG_H
#include "config.h" // IWYU pragma: keep
#endif

#include <inttypes.h> // for PRIxPTR

#include "tcti-common.h"
#include "tss2_mu.h"         // for Tss2_MU_UINT32_Marshal, Tss2_MU_UINT32_...
#include "tss2_tpm2_types.h" // for TPM2_HANDLE
#include "util/aux_util.h"   // for UNUSED

#define LOGMODULE tcti
#include "util/log.h" // for LOG_ERROR, LOG_TRACE

TSS2_TCTI_COMMON_CONTEXT *
tcti_common_context_cast(TSS2_TCTI_CONTEXT *ctx) {
    return (TSS2_TCTI_COMMON_CONTEXT *)ctx;
}

TSS2_TCTI_CONTEXT *
tcti_common_down_cast(TSS2_TCTI_COMMON_CONTEXT *ctx) {
    return (TSS2_TCTI_CONTEXT *)&ctx->v2;
}

TSS2_RC
tcti_common_cancel_checks(TSS2_TCTI_COMMON_CONTEXT *tcti_common, uint64_t magic) {
    if (tcti_common == NULL) {
        return TSS2_TCTI_RC_BAD_REFERENCE;
    }

    if (TSS2_TCTI_MAGIC(tcti_common) != magic) {
        return TSS2_TCTI_RC_BAD_CONTEXT;
    }

    if (tcti_common->state != TCTI_STATE_RECEIVE) {
        return TSS2_TCTI_RC_BAD_SEQUENCE;
    }
    return TSS2_RC_SUCCESS;
}

TSS2_RC
tcti_common_transmit_checks(TSS2_TCTI_COMMON_CONTEXT *tcti_common,
                            const uint8_t            *command_buffer,
                            uint64_t                  magic) {
    if (command_buffer == NULL || tcti_common == NULL) {
        return TSS2_TCTI_RC_BAD_REFERENCE;
    }

    if (TSS2_TCTI_MAGIC(tcti_common) != magic) {
        return TSS2_TCTI_RC_BAD_CONTEXT;
    }

    if (tcti_common->state != TCTI_STATE_TRANSMIT) {
        return TSS2_TCTI_RC_BAD_SEQUENCE;
    }

    return TSS2_RC_SUCCESS;
}

TSS2_RC
tcti_common_receive_checks(TSS2_TCTI_COMMON_CONTEXT *tcti_common,
                           size_t                   *response_size,
                           uint64_t                  magic) {
    if (response_size == NULL || tcti_common == NULL) {
        return TSS2_TCTI_RC_BAD_REFERENCE;
    }

    if (TSS2_TCTI_MAGIC(tcti_common) != magic) {
        return TSS2_TCTI_RC_BAD_CONTEXT;
    }

    if (tcti_common->state != TCTI_STATE_RECEIVE) {
        return TSS2_TCTI_RC_BAD_SEQUENCE;
    }

    return TSS2_RC_SUCCESS;
}

TSS2_RC
tcti_common_set_locality_checks(TSS2_TCTI_COMMON_CONTEXT *tcti_common, uint64_t magic) {
    if (tcti_common == NULL) {
        return TSS2_TCTI_RC_BAD_REFERENCE;
    }

    if (TSS2_TCTI_MAGIC(tcti_common) != magic) {
        return TSS2_TCTI_RC_BAD_CONTEXT;
    }

    if (tcti_common->state != TCTI_STATE_TRANSMIT) {
        return TSS2_TCTI_RC_BAD_SEQUENCE;
    }
    return TSS2_RC_SUCCESS;
}

TSS2_RC
tcti_make_sticky_not_implemented(TSS2_TCTI_CONTEXT *tctiContext,
                                 TPM2_HANDLE       *handle,
                                 uint8_t            sticky) {
    UNUSED(tctiContext);
    UNUSED(handle);
    UNUSED(sticky);
    return TSS2_TCTI_RC_NOT_IMPLEMENTED;
}

TSS2_RC
header_unmarshal(const uint8_t *buf, tpm_header_t *header) {
    TSS2_RC rc;
    size_t  offset = 0;

    LOG_TRACE("Parsing header from buffer: 0x%" PRIxPTR, (uintptr_t)buf);
    rc = Tss2_MU_TPM2_ST_Unmarshal(buf, TPM_HEADER_SIZE, &offset, &header->tag);
    if (rc != TSS2_RC_SUCCESS) {
        LOG_ERROR("Failed to unmarshal tag.");
        return rc;
    }
    rc = Tss2_MU_UINT32_Unmarshal(buf, TPM_HEADER_SIZE, &offset, &header->size);
    if (rc != TSS2_RC_SUCCESS) {
        LOG_ERROR("Failed to unmarshal command size.");
        return rc;
    }
    rc = Tss2_MU_UINT32_Unmarshal(buf, TPM_HEADER_SIZE, &offset, &header->code);
    if (rc != TSS2_RC_SUCCESS) {
        LOG_ERROR("Failed to unmarshal command code.");
    }
    return rc;
}

TSS2_RC
header_marshal(const tpm_header_t *header, uint8_t *buf) {
    TSS2_RC rc;
    size_t  offset = 0;

    LOG_TRACE("Parsing header from buffer: 0x%" PRIxPTR, (uintptr_t)buf);
    rc = Tss2_MU_TPM2_ST_Marshal(header->tag, buf, TPM_HEADER_SIZE, &offset);
    if (rc != TSS2_RC_SUCCESS) {
        LOG_ERROR("Failed to marshal tag.");
        return rc;
    }
    rc = Tss2_MU_UINT32_Marshal(header->size, buf, TPM_HEADER_SIZE, &offset);
    if (rc != TSS2_RC_SUCCESS) {
        LOG_ERROR("Failed to marshal command size.");
        return rc;
    }
    rc = Tss2_MU_UINT32_Marshal(header->code, buf, TPM_HEADER_SIZE, &offset);
    if (rc != TSS2_RC_SUCCESS) {
        LOG_ERROR("Failed to marshal command code.");
    }
    return rc;
}
