// SPDX-License-Identifier: GPL-2.0-only
/* Copyright(c) 2026 Intel Corporation */
#include <linux/dma-mapping.h>

#include "adf_admin.h"
#include "adf_cfg_services.h"
#include "adf_common_drv.h"
#include "adf_kpt.h"

static bool adf_kpt_supported(struct adf_accel_dev *accel_dev)
{
        struct adf_hw_device_data *hw_data = GET_HW_DATA(accel_dev);

        return hw_data->accel_capabilities_mask & ICP_ACCEL_CAPABILITIES_KPT;
}

int adf_enable_kpt(struct adf_accel_dev *accel_dev)
{
        struct adf_kpt_interface_data *user_data = GET_KPT_USER_DATA(accel_dev);
        struct adf_hw_device_data *hw_data = GET_HW_DATA(accel_dev);
        dma_addr_t paddr;
        void *vaddr;
        int ret;
        int svc;

        /* Return 0 if KPT is not supported by the hardware */
        if (!adf_kpt_supported(accel_dev))
                return 0;

        if (!user_data->enable) {
                /* Disable KPT capability if user has not enabled it */
                hw_data->accel_capabilities_mask &= ~ICP_ACCEL_CAPABILITIES_KPT;
                return 0;
        }

        svc = adf_get_service_enabled(accel_dev);
        if (svc < 0)
                return svc;

        if (svc != SVC_ASYM) {
                dev_err(&GET_DEV(accel_dev),
                        "KPT can only be enabled when service is configured as 'asym'\n");
                return -EINVAL;
        }

        vaddr = dma_alloc_coherent(&GET_DEV(accel_dev), PAGE_SIZE, &paddr,
                                   GFP_KERNEL);
        if (!vaddr)
                return -ENOMEM;

        ret = adf_send_admin_kpt_init(accel_dev, vaddr, PAGE_SIZE, paddr);

        dma_free_coherent(&GET_DEV(accel_dev), PAGE_SIZE, vaddr, paddr);

        return ret;
}