/* SPDX-License-Identifier: BSD-3-Clause */
/* Copyright(c) 2007-2026 Intel Corporation */
#include "qat_freebsd.h"
#include "adf_cfg.h"
#include "adf_common_drv.h"
#include "adf_accel_devices.h"
#include "adf_4xxx_hw_data.h"
#include "adf_gen4_hw_data.h"
#include "adf_fw_counters.h"
#include "adf_cfg_device.h"
#include "adf_dbgfs.h"
#include <sys/types.h>
#include <sys/kernel.h>
#include <sys/malloc.h>
#include <machine/bus_dma.h>
#include <dev/pci/pcireg.h>

static MALLOC_DEFINE(M_QAT_4XXX, "qat_4xxx", "qat_4xxx");

#define ADF_SYSTEM_DEVICE(device_id)                                           \
        {                                                                      \
                PCI_VENDOR_ID_INTEL, device_id                                 \
        }

static const struct pci_device_id adf_pci_tbl[] = {
        ADF_SYSTEM_DEVICE(ADF_4XXX_PCI_DEVICE_ID),
        ADF_SYSTEM_DEVICE(ADF_401XX_PCI_DEVICE_ID),
        ADF_SYSTEM_DEVICE(ADF_402XX_PCI_DEVICE_ID),
        {
            0,
        }
};

static int
adf_probe(device_t dev)
{
        const struct pci_device_id *id;

        for (id = adf_pci_tbl; id->vendor != 0; id++) {
                if (pci_get_vendor(dev) == id->vendor &&
                    pci_get_device(dev) == id->device) {
                        device_set_desc(dev,
                                        "Intel " ADF_4XXX_DEVICE_NAME
                                        " QuickAssist");
                        return BUS_PROBE_GENERIC;
                }
        }
        return ENXIO;
}

#ifdef QAT_DISABLE_SAFE_DC_MODE
static int adf_4xxx_sysctl_disable_safe_dc_mode(SYSCTL_HANDLER_ARGS)
{
        struct adf_accel_dev *accel_dev = arg1;
        int error, value = accel_dev->disable_safe_dc_mode;

        error = sysctl_handle_int(oidp, &value, 0, req);
        if (error || !req->newptr)
                return error;

        if (value != 1 && value != 0)
                return EINVAL;

        if (adf_dev_started(accel_dev)) {
                device_printf(
                    GET_DEV(accel_dev),
                    "QAT: configuration can only be changed in \"down\" device state\n");
                return EBUSY;
        }

        accel_dev->disable_safe_dc_mode = (u8)value;

        return 0;
}

static void
adf_4xxx_disable_safe_dc_sysctl_add(struct adf_accel_dev *accel_dev)
{
        struct sysctl_ctx_list *qat_sysctl_ctx;
        struct sysctl_oid *qat_sysctl_tree;

        qat_sysctl_ctx =
            device_get_sysctl_ctx(accel_dev->accel_pci_dev.pci_dev);
        qat_sysctl_tree =
            device_get_sysctl_tree(accel_dev->accel_pci_dev.pci_dev);
        accel_dev->safe_dc_mode =
            SYSCTL_ADD_OID(qat_sysctl_ctx,
                           SYSCTL_CHILDREN(qat_sysctl_tree),
                           OID_AUTO,
                           "disable_safe_dc_mode",
                           CTLTYPE_INT | CTLFLAG_WR | CTLFLAG_TUN |
                               CTLFLAG_SKIP,
                           accel_dev,
                           0,
                           adf_4xxx_sysctl_disable_safe_dc_mode,
                           "LU",
                           "Disable QAT safe data compression mode");
}

static void
adf_4xxx_disable_safe_dc_sysctl_remove(struct adf_accel_dev *accel_dev)
{
        int ret;
        struct sysctl_ctx_list *qat_sysctl_ctx =
            device_get_sysctl_ctx(accel_dev->accel_pci_dev.pci_dev);

        ret = sysctl_ctx_entry_del(qat_sysctl_ctx, accel_dev->safe_dc_mode);
        if (ret) {
                device_printf(GET_DEV(accel_dev), "Failed to delete entry\n");
        } else {
                ret = sysctl_remove_oid(accel_dev->safe_dc_mode, 1, 1);
                if (ret)
                        device_printf(GET_DEV(accel_dev),
                                      "Failed to delete oid\n");
        }
}
#endif /* QAT_DISABLE_SAFE_DC_MODE */

static void
adf_cleanup_accel(struct adf_accel_dev *accel_dev)
{
        struct adf_accel_pci *accel_pci_dev = &accel_dev->accel_pci_dev;
        int i;

        if (accel_dev->dma_tag)
                bus_dma_tag_destroy(accel_dev->dma_tag);
        for (i = 0; i < ADF_PCI_MAX_BARS; i++) {
                struct adf_bar *bar = &accel_pci_dev->pci_bars[i];

                if (bar->virt_addr)
                        bus_free_resource(accel_pci_dev->pci_dev,
                                          SYS_RES_MEMORY,
                                          bar->virt_addr);
        }

        if (accel_dev->hw_device) {
                switch (pci_get_device(accel_pci_dev->pci_dev)) {
                case ADF_4XXX_PCI_DEVICE_ID:
                case ADF_401XX_PCI_DEVICE_ID:
                case ADF_402XX_PCI_DEVICE_ID:
                        adf_clean_hw_data_4xxx(accel_dev->hw_device);
                        break;
                default:
                        break;
                }
                free(accel_dev->hw_device, M_QAT_4XXX);
                accel_dev->hw_device = NULL;
        }
#ifdef QAT_DISABLE_SAFE_DC_MODE
        adf_4xxx_disable_safe_dc_sysctl_remove(accel_dev);
#endif /* QAT_DISABLE_SAFE_DC_MODE */
        adf_dbgfs_exit(accel_dev);
        adf_cfg_dev_remove(accel_dev);
        adf_devmgr_rm_dev(accel_dev, NULL);
}

static int
adf_attach(device_t dev)
{
        struct adf_accel_dev *accel_dev;
        struct adf_accel_pci *accel_pci_dev;
        struct adf_hw_device_data *hw_data;
        unsigned int bar_nr;
        int ret = 0, rid;
        struct adf_cfg_device *cfg_dev = NULL;

        /* Set pci MaxPayLoad to 512. Implemented to avoid the issue of
         * Pci-passthrough causing Maxpayload to be reset to 128 bytes
         * when the device is reset.
         */
        if (pci_get_max_payload(dev) != 512)
                pci_set_max_payload(dev, 512);

        accel_dev = device_get_softc(dev);

        mutex_init(&accel_dev->lock);
        INIT_LIST_HEAD(&accel_dev->crypto_list);
        accel_pci_dev = &accel_dev->accel_pci_dev;
        accel_pci_dev->pci_dev = dev;

        if (bus_get_domain(dev, &accel_pci_dev->node) != 0)
                accel_pci_dev->node = 0;

        /* Add accel device to accel table.
         * This should be called before adf_cleanup_accel is called
         */
        ret = adf_devmgr_add_dev(accel_dev, NULL);
        if (ret) {
                device_printf(dev, "Failed to add new accelerator device.\n");
                goto out_err_lock;
        }

        /* Allocate and configure device configuration structure */
        hw_data = malloc(sizeof(*hw_data), M_QAT_4XXX, M_WAITOK | M_ZERO);

        accel_dev->hw_device = hw_data;
        adf_init_hw_data_4xxx(accel_dev->hw_device, pci_get_device(dev));
        accel_pci_dev->revid = pci_get_revid(dev);
        hw_data->fuses = pci_read_config(dev, ADF_4XXX_FUSECTL4_OFFSET, 4);

        /* Get PPAERUCM values and store */
        ret = adf_aer_store_ppaerucm_reg(dev, hw_data);
        if (ret)
                goto out_err;

        /* Get Accelerators and Accelerators Engines masks */
        hw_data->accel_mask = hw_data->get_accel_mask(accel_dev);
        hw_data->ae_mask = hw_data->get_ae_mask(accel_dev);

        accel_pci_dev->sku = hw_data->get_sku(hw_data);
        /* If the device has no acceleration engines then ignore it. */
        if (!hw_data->accel_mask || !hw_data->ae_mask ||
            (~hw_data->ae_mask & 0x01ULL)) {
                device_printf(dev, "No acceleration units found\n");
                ret = ENXIO;
                goto out_err;
        }

        /* Create device configuration table */
        ret = adf_cfg_dev_add(accel_dev);
        if (ret)
                goto out_err;
        ret = adf_clock_debugfs_add(accel_dev);
        if (ret)
                goto out_err;

#ifdef QAT_DISABLE_SAFE_DC_MODE
        adf_4xxx_disable_safe_dc_sysctl_add(accel_dev);
#endif /* QAT_DISABLE_SAFE_DC_MODE */

        pci_set_max_read_req(dev, 4096);

        ret = bus_dma_tag_create(bus_get_dma_tag(dev),
                                 1,
                                 0,
                                 BUS_SPACE_MAXADDR,
                                 BUS_SPACE_MAXADDR,
                                 NULL,
                                 NULL,
                                 BUS_SPACE_MAXSIZE,
                                 /* BUS_SPACE_UNRESTRICTED */ 1,
                                 BUS_SPACE_MAXSIZE,
                                 0,
                                 NULL,
                                 NULL,
                                 &accel_dev->dma_tag);
        if (ret)
                goto out_err;

        if (hw_data->get_accel_cap) {
                hw_data->accel_capabilities_mask =
                    hw_data->get_accel_cap(accel_dev);
        }

        /* Find and map all the device's BARS */
        /* Logical BARs configuration for 64bit BARs:
             bar 0 and 1 - logical BAR0
             bar 2 and 3 - logical BAR1
             bar 4 and 5 - logical BAR3
        */
        for (bar_nr = 0;
             bar_nr < (ADF_PCI_MAX_BARS * 2) && bar_nr < PCIR_MAX_BAR_0;
             bar_nr += 2) {
                struct adf_bar *bar;

                rid = PCIR_BAR(bar_nr);
                bar = &accel_pci_dev->pci_bars[bar_nr / 2];

                bar->virt_addr = bus_alloc_resource_any(dev,
                                                        SYS_RES_MEMORY,
                                                        &rid,
                                                        RF_ACTIVE);
                if (!bar->virt_addr) {
                        device_printf(dev, "Failed to map BAR %d\n", bar_nr);
                        ret = ENXIO;
                        goto out_err;
                }
                bar->base_addr = rman_get_start(bar->virt_addr);
                bar->size = rman_get_size(bar->virt_addr);
        }
        ret = pci_enable_busmaster(dev);
        if (ret)
                goto out_err;

        adf_dbgfs_init(accel_dev);

        if (!accel_dev->hw_device->config_device) {
                ret = EFAULT;
                goto out_err_disable;
        }

        ret = accel_dev->hw_device->config_device(accel_dev);
        if (ret)
                goto out_err_disable;

        ret = adf_dev_init(accel_dev);
        if (ret)
                goto out_dev_shutdown;

        ret = adf_dev_start(accel_dev);
        if (ret)
                goto out_dev_stop;

        cfg_dev = accel_dev->cfg->dev;
        adf_cfg_device_clear(cfg_dev, accel_dev);
        free(cfg_dev, M_QAT);
        accel_dev->cfg->dev = NULL;
        return ret;
out_dev_stop:
        adf_dev_stop(accel_dev);
out_dev_shutdown:
        adf_dev_shutdown(accel_dev);
out_err_disable:
        pci_disable_busmaster(dev);
out_err:
        adf_cleanup_accel(accel_dev);
out_err_lock:
        mutex_destroy(&accel_dev->lock);

        return ret;
}

static int
adf_detach(device_t dev)
{
        struct adf_accel_dev *accel_dev = device_get_softc(dev);

        if (adf_dev_stop(accel_dev)) {
                device_printf(dev, "Failed to stop QAT accel dev\n");
                return EBUSY;
        }

        adf_dev_shutdown(accel_dev);

        pci_disable_busmaster(dev);
        adf_cleanup_accel(accel_dev);
        mutex_destroy(&accel_dev->lock);

        return 0;
}

static device_method_t adf_methods[] = { DEVMETHOD(device_probe, adf_probe),
                                         DEVMETHOD(device_attach, adf_attach),
                                         DEVMETHOD(device_detach, adf_detach),

                                         DEVMETHOD_END };

static driver_t adf_driver = { "qat",
                               adf_methods,
                               sizeof(struct adf_accel_dev) };

DRIVER_MODULE_ORDERED(qat_4xxx, pci, adf_driver, NULL, NULL, SI_ORDER_THIRD);
MODULE_VERSION(qat_4xxx, 1);
MODULE_DEPEND(qat_4xxx, qat_common, 1, 1, 1);
MODULE_DEPEND(qat_4xxx, qat_api, 1, 1, 1);
MODULE_DEPEND(qat_4xxx, linuxkpi, 1, 1, 1);