// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (c) 2013-2023, Intel Corporation. All rights reserved.
 * Intel Management Engine Interface (Intel MEI) Linux driver
 */

#include <linux/kernel.h>
#include <linux/sched.h>
#include <linux/module.h>
#include <linux/device.h>
#include <linux/slab.h>

#include <linux/mei.h>
#include <linux/mei_cl_bus.h>

#include "mei_dev.h"
#include "client.h"
#include "mkhi.h"

#define MEI_UUID_NFC_INFO UUID_LE(0xd2de1625, 0x382d, 0x417d, \
                        0x48, 0xa4, 0xef, 0xab, 0xba, 0x8a, 0x12, 0x06)

static const uuid_le mei_nfc_info_guid = MEI_UUID_NFC_INFO;

#define MEI_UUID_NFC_HCI UUID_LE(0x0bb17a78, 0x2a8e, 0x4c50, \
                        0x94, 0xd4, 0x50, 0x26, 0x67, 0x23, 0x77, 0x5c)

#define MEI_UUID_WD UUID_LE(0x05B79A6F, 0x4628, 0x4D7F, \
                            0x89, 0x9D, 0xA9, 0x15, 0x14, 0xCB, 0x32, 0xAB)

#define MEI_UUID_MKHIF_FIX UUID_LE(0x55213584, 0x9a29, 0x4916, \
                        0xba, 0xdf, 0xf, 0xb7, 0xed, 0x68, 0x2a, 0xeb)

#define MEI_UUID_IGSC_MKHI UUID_LE(0xE2C2AFA2, 0x3817, 0x4D19, \
                        0x9D, 0x95, 0x06, 0xB1, 0x6B, 0x58, 0x8A, 0x5D)

#define MEI_UUID_IGSC_MKHI_FIX UUID_LE(0x46E0C1FB, 0xA546, 0x414F, \
                        0x91, 0x70, 0xB7, 0xF4, 0x6D, 0x57, 0xB4, 0xAD)

#define MEI_UUID_HDCP UUID_LE(0xB638AB7E, 0x94E2, 0x4EA2, \
                              0xA5, 0x52, 0xD1, 0xC5, 0x4B, 0x62, 0x7F, 0x04)

#define MEI_UUID_PAVP UUID_LE(0xfbf6fcf1, 0x96cf, 0x4e2e, 0xA6, \
                              0xa6, 0x1b, 0xab, 0x8c, 0xbe, 0x36, 0xb1)

#define MEI_UUID_ANY NULL_UUID_LE

/**
 * number_of_connections - determine whether an client be on the bus
 *    according number of connections
 *    We support only clients:
 *       1. with single connection
 *       2. and fixed clients (max_number_of_connections == 0)
 *
 * @cldev: me clients device
 */
static void number_of_connections(struct mei_cl_device *cldev)
{
        if (cldev->me_cl->props.max_number_of_connections > 1)
                cldev->do_match = 0;
}

/**
 * blacklist - blacklist a client from the bus
 *
 * @cldev: me clients device
 */
static void blacklist(struct mei_cl_device *cldev)
{
        cldev->do_match = 0;
}

/**
 * whitelist - forcefully whitelist client
 *
 * @cldev: me clients device
 */
static void whitelist(struct mei_cl_device *cldev)
{
        cldev->do_match = 1;
}

#define MKHI_SEND_MAX_TIMEOUT_MSEC 4000

#define OSTYPE_LINUX    2
struct mei_os_ver {
        __le16 build;
        __le16 reserved1;
        u8  os_type;
        u8  major;
        u8  minor;
        u8  reserved2;
} __packed;

struct mkhi_fw_ver_block {
        u16 minor;
        u8 major;
        u8 platform;
        u16 buildno;
        u16 hotfix;
} __packed;

struct mkhi_fw_ver {
        struct mkhi_fw_ver_block ver[MEI_MAX_FW_VER_BLOCKS];
} __packed;

#define MKHI_OSVER_BUF_LEN (sizeof(struct mkhi_msg_hdr) + \
                            sizeof(struct mkhi_fwcaps) + \
                            sizeof(struct mei_os_ver))
static int mei_osver(struct mei_cl_device *cldev)
{
        const size_t size = MKHI_OSVER_BUF_LEN;
        u8 buf[MKHI_OSVER_BUF_LEN];
        struct mkhi_msg *req;
        struct mkhi_fwcaps *fwcaps;
        struct mei_os_ver *os_ver;
        unsigned int mode = MEI_CL_IO_TX_BLOCKING | MEI_CL_IO_TX_INTERNAL;

        memset(buf, 0, size);

        req = (struct mkhi_msg *)buf;
        req->hdr.group_id = MKHI_FWCAPS_GROUP_ID;
        req->hdr.command = MKHI_FWCAPS_SET_OS_VER_APP_RULE_CMD;

        fwcaps = (struct mkhi_fwcaps *)req->data;

        fwcaps->id.rule_type = 0x0;
        fwcaps->id.feature_id = MKHI_FEATURE_PTT;
        fwcaps->len = sizeof(*os_ver);
        os_ver = (struct mei_os_ver *)fwcaps->data;
        os_ver->os_type = OSTYPE_LINUX;

        return __mei_cl_send_timeout(cldev->cl, buf, size, 0, mode, MKHI_SEND_MAX_TIMEOUT_MSEC);
}

#define MKHI_FWVER_BUF_LEN (sizeof(struct mkhi_msg_hdr) + \
                            sizeof(struct mkhi_fw_ver))
#define MKHI_FWVER_LEN(__num) (sizeof(struct mkhi_msg_hdr) + \
                               sizeof(struct mkhi_fw_ver_block) * (__num))
static int mei_fwver(struct mei_cl_device *cldev)
{
        u8 buf[MKHI_FWVER_BUF_LEN];
        struct mkhi_msg req;
        struct mkhi_msg *rsp;
        struct mkhi_fw_ver *fwver;
        int bytes_recv, ret, i;

        memset(buf, 0, sizeof(buf));

        req.hdr.group_id = MKHI_GEN_GROUP_ID;
        req.hdr.command = MKHI_GEN_GET_FW_VERSION_CMD;

        ret = __mei_cl_send_timeout(cldev->cl, (u8 *)&req, sizeof(req), 0,
                                    MEI_CL_IO_TX_BLOCKING, MKHI_SEND_MAX_TIMEOUT_MSEC);
        if (ret < 0) {
                dev_info(&cldev->dev, "Could not send ReqFWVersion cmd ret = %d\n", ret);
                return ret;
        }

        ret = 0;
        bytes_recv = __mei_cl_recv(cldev->cl, buf, sizeof(buf), NULL, 0,
                                   cldev->bus->timeouts.mkhi_recv);
        if (bytes_recv < 0 || (size_t)bytes_recv < MKHI_FWVER_LEN(1)) {
                /*
                 * Should be at least one version block,
                 * error out if nothing found
                 */
                dev_info(&cldev->dev, "Could not read FW version ret = %d\n", bytes_recv);
                return -EIO;
        }

        rsp = (struct mkhi_msg *)buf;
        fwver = (struct mkhi_fw_ver *)rsp->data;
        memset(cldev->bus->fw_ver, 0, sizeof(cldev->bus->fw_ver));
        for (i = 0; i < MEI_MAX_FW_VER_BLOCKS; i++) {
                if ((size_t)bytes_recv < MKHI_FWVER_LEN(i + 1))
                        break;
                dev_dbg(&cldev->dev, "FW version%d %d:%d.%d.%d.%d\n",
                        i, fwver->ver[i].platform,
                        fwver->ver[i].major, fwver->ver[i].minor,
                        fwver->ver[i].hotfix, fwver->ver[i].buildno);

                cldev->bus->fw_ver[i].platform = fwver->ver[i].platform;
                cldev->bus->fw_ver[i].major = fwver->ver[i].major;
                cldev->bus->fw_ver[i].minor = fwver->ver[i].minor;
                cldev->bus->fw_ver[i].hotfix = fwver->ver[i].hotfix;
                cldev->bus->fw_ver[i].buildno = fwver->ver[i].buildno;
        }
        cldev->bus->fw_ver_received = 1;

        return ret;
}

#define GFX_MEMORY_READY_TIMEOUT 200 /* timeout in milliseconds */

static int mei_gfx_memory_ready(struct mei_cl_device *cldev)
{
        struct mkhi_gfx_mem_ready req = {0};
        unsigned int mode = MEI_CL_IO_TX_INTERNAL | MEI_CL_IO_TX_BLOCKING;

        req.hdr.group_id = MKHI_GROUP_ID_GFX;
        req.hdr.command = MKHI_GFX_MEMORY_READY_CMD_REQ;
        req.flags = MKHI_GFX_MEM_READY_PXP_ALLOWED;

        dev_dbg(&cldev->dev, "Sending memory ready command\n");
        return __mei_cl_send_timeout(cldev->cl, (u8 *)&req, sizeof(req), 0,
                                     mode, GFX_MEMORY_READY_TIMEOUT);
}

static void mei_mkhi_fix(struct mei_cl_device *cldev)
{
        int ret;

        /* No need to enable the client if nothing is needed from it */
        if (!cldev->bus->fw_f_fw_ver_supported &&
            !cldev->bus->hbm_f_os_supported)
                return;

        ret = mei_cldev_enable(cldev);
        if (ret)
                return;

        if (cldev->bus->fw_f_fw_ver_supported) {
                ret = mei_fwver(cldev);
                if (ret < 0)
                        dev_info(&cldev->dev, "FW version command failed %d\n",
                                 ret);
        }

        if (cldev->bus->hbm_f_os_supported) {
                ret = mei_osver(cldev);
                if (ret < 0)
                        dev_info(&cldev->dev, "OS version command failed %d\n",
                                 ret);
        }
        mei_cldev_disable(cldev);
}

static void mei_gsc_mkhi_ver(struct mei_cl_device *cldev)
{
        int ret;

        /*
         * No need to enable the client if nothing is needed from it.
         * No need to fill in version if it is already filled in by the fix address client.
         */
        if (!cldev->bus->fw_f_fw_ver_supported || cldev->bus->fw_ver_received)
                return;

        ret = mei_cldev_enable(cldev);
        if (ret)
                return;

        ret = mei_fwver(cldev);
        if (ret < 0)
                dev_info(&cldev->dev, "FW version command failed %d\n", ret);
        mei_cldev_disable(cldev);
}

static void mei_gsc_mkhi_fix_ver(struct mei_cl_device *cldev)
{
        int ret;

        /* No need to enable the client if nothing is needed from it */
        if (!cldev->bus->fw_f_fw_ver_supported &&
            cldev->bus->pxp_mode != MEI_DEV_PXP_INIT)
                return;

        ret = mei_cldev_enable(cldev);
        if (ret)
                return;

        if (cldev->bus->pxp_mode == MEI_DEV_PXP_INIT) {
                ret = mei_gfx_memory_ready(cldev);
                if (ret < 0) {
                        dev_err(&cldev->dev, "memory ready command failed %d\n", ret);
                } else {
                        dev_dbg(&cldev->dev, "memory ready command sent\n");
                        cldev->bus->pxp_mode = MEI_DEV_PXP_SETUP;
                }
                /* we go to reset after that */
                goto out;
        }

        ret = mei_fwver(cldev);
        if (ret < 0)
                dev_info(&cldev->dev, "FW version command failed %d\n",
                         ret);
out:
        mei_cldev_disable(cldev);
}

/**
 * mei_wd - wd client on the bus, change protocol version
 *   as the API has changed.
 *
 * @cldev: me clients device
 */
#if IS_ENABLED(CONFIG_INTEL_MEI_ME)
#include <linux/pci.h>
#include "hw-me-regs.h"
static void mei_wd(struct mei_cl_device *cldev)
{
        struct pci_dev *pdev = to_pci_dev(cldev->dev.parent);

        if (pdev->device == MEI_DEV_ID_WPT_LP ||
            pdev->device == MEI_DEV_ID_SPT ||
            pdev->device == MEI_DEV_ID_SPT_H)
                cldev->me_cl->props.protocol_version = 0x2;

        cldev->do_match = 1;
}
#else
static inline void mei_wd(struct mei_cl_device *cldev) {}
#endif /* CONFIG_INTEL_MEI_ME */

struct mei_nfc_cmd {
        u8 command;
        u8 status;
        u16 req_id;
        u32 reserved;
        u16 data_size;
        u8 sub_command;
        u8 data[];
} __packed;

struct mei_nfc_reply {
        u8 command;
        u8 status;
        u16 req_id;
        u32 reserved;
        u16 data_size;
        u8 sub_command;
        u8 reply_status;
        u8 data[];
} __packed;

struct mei_nfc_if_version {
        u8 radio_version_sw[3];
        u8 reserved[3];
        u8 radio_version_hw[3];
        u8 i2c_addr;
        u8 fw_ivn;
        u8 vendor_id;
        u8 radio_type;
} __packed;


#define MEI_NFC_CMD_MAINTENANCE 0x00
#define MEI_NFC_SUBCMD_IF_VERSION 0x01

/* Vendors */
#define MEI_NFC_VENDOR_INSIDE 0x00
#define MEI_NFC_VENDOR_NXP    0x01

/* Radio types */
#define MEI_NFC_VENDOR_INSIDE_UREAD 0x00
#define MEI_NFC_VENDOR_NXP_PN544    0x01

/**
 * mei_nfc_if_version - get NFC interface version
 *
 * @cl: host client (nfc info)
 * @ver: NFC interface version to be filled in
 *
 * Return: 0 on success; < 0 otherwise
 */
static int mei_nfc_if_version(struct mei_cl *cl,
                              struct mei_nfc_if_version *ver)
{
        struct mei_device *bus;
        struct mei_nfc_cmd cmd = {
                .command = MEI_NFC_CMD_MAINTENANCE,
                .data_size = 1,
                .sub_command = MEI_NFC_SUBCMD_IF_VERSION,
        };
        struct mei_nfc_reply *reply = NULL;
        size_t if_version_length;
        u8 vtag;
        int bytes_recv, ret;

        bus = cl->dev;

        WARN_ON(mutex_is_locked(&bus->device_lock));

        ret = __mei_cl_send(cl, (u8 *)&cmd, sizeof(cmd), 0,
                            MEI_CL_IO_TX_BLOCKING);
        if (ret < 0) {
                dev_err(&bus->dev, "Could not send IF version cmd ret = %d\n", ret);
                return ret;
        }

        /* to be sure on the stack we alloc memory */
        if_version_length = sizeof(*reply) + sizeof(*ver);

        reply = kzalloc(if_version_length, GFP_KERNEL);
        if (!reply)
                return -ENOMEM;

        ret = 0;
        bytes_recv = __mei_cl_recv(cl, (u8 *)reply, if_version_length, &vtag,
                                   0, 0);
        if (bytes_recv < 0 || (size_t)bytes_recv < if_version_length) {
                dev_err(&bus->dev, "Could not read IF version ret = %d\n", bytes_recv);
                ret = -EIO;
                goto err;
        }

        memcpy(ver, reply->data, sizeof(*ver));

        dev_info(&bus->dev, "NFC MEI VERSION: IVN 0x%x Vendor ID 0x%x Type 0x%x\n",
                 ver->fw_ivn, ver->vendor_id, ver->radio_type);

err:
        kfree(reply);
        return ret;
}

/**
 * mei_nfc_radio_name - derive nfc radio name from the interface version
 *
 * @ver: NFC radio version
 *
 * Return: radio name string
 */
static const char *mei_nfc_radio_name(struct mei_nfc_if_version *ver)
{

        if (ver->vendor_id == MEI_NFC_VENDOR_INSIDE) {
                if (ver->radio_type == MEI_NFC_VENDOR_INSIDE_UREAD)
                        return "microread";
        }

        if (ver->vendor_id == MEI_NFC_VENDOR_NXP) {
                if (ver->radio_type == MEI_NFC_VENDOR_NXP_PN544)
                        return "pn544";
        }

        return NULL;
}

/**
 * mei_nfc - The nfc fixup function. The function retrieves nfc radio
 *    name and set is as device attribute so we can load
 *    the proper device driver for it
 *
 * @cldev: me client device (nfc)
 */
static void mei_nfc(struct mei_cl_device *cldev)
{
        struct mei_device *bus;
        struct mei_cl *cl;
        struct mei_me_client *me_cl = NULL;
        struct mei_nfc_if_version ver;
        const char *radio_name = NULL;
        int ret;

        bus = cldev->bus;

        mutex_lock(&bus->device_lock);
        /* we need to connect to INFO GUID */
        cl = mei_cl_alloc_linked(bus);
        if (IS_ERR(cl)) {
                ret = PTR_ERR(cl);
                cl = NULL;
                dev_err(&cldev->dev, "nfc hook alloc failed %d\n", ret);
                goto out;
        }

        me_cl = mei_me_cl_by_uuid(bus, &mei_nfc_info_guid);
        if (!me_cl) {
                ret = -ENOTTY;
                dev_err(&cldev->dev, "Cannot find nfc info %d\n", ret);
                goto out;
        }

        ret = mei_cl_connect(cl, me_cl, NULL);
        if (ret < 0) {
                dev_err(&cldev->dev, "Can't connect to the NFC INFO ME ret = %d\n",
                        ret);
                goto out;
        }

        mutex_unlock(&bus->device_lock);

        ret = mei_nfc_if_version(cl, &ver);
        if (ret)
                goto disconnect;

        radio_name = mei_nfc_radio_name(&ver);

        if (!radio_name) {
                ret = -ENOENT;
                dev_err(&cldev->dev, "Can't get the NFC interface version ret = %d\n",
                        ret);
                goto disconnect;
        }

        dev_dbg(&cldev->dev, "nfc radio %s\n", radio_name);
        strscpy(cldev->name, radio_name, sizeof(cldev->name));

disconnect:
        mutex_lock(&bus->device_lock);
        if (mei_cl_disconnect(cl) < 0)
                dev_err(&cldev->dev, "Can't disconnect the NFC INFO ME\n");

        mei_cl_flush_queues(cl, NULL);

out:
        mei_cl_unlink(cl);
        mutex_unlock(&bus->device_lock);
        mei_me_cl_put(me_cl);
        kfree(cl);

        if (ret)
                cldev->do_match = 0;

        dev_dbg(&cldev->dev, "end of fixup match = %d\n", cldev->do_match);
}

/**
 * vt_support - enable on bus clients with vtag support
 *
 * @cldev: me clients device
 */
static void vt_support(struct mei_cl_device *cldev)
{
        if (cldev->me_cl->props.vt_supported == 1)
                cldev->do_match = 1;
}

/**
 * pxp_is_ready - enable bus client if pxp is ready
 *
 * @cldev: me clients device
 */
static void pxp_is_ready(struct mei_cl_device *cldev)
{
        struct mei_device *bus = cldev->bus;

        switch (bus->pxp_mode) {
        case MEI_DEV_PXP_READY:
        case MEI_DEV_PXP_DEFAULT:
                cldev->do_match = 1;
        break;
        default:
                cldev->do_match = 0;
        break;
        }
}

#define MEI_FIXUP(_uuid, _hook) { _uuid, _hook }

static struct mei_fixup {

        const uuid_le uuid;
        void (*hook)(struct mei_cl_device *cldev);
} mei_fixups[] = {
        MEI_FIXUP(MEI_UUID_ANY, number_of_connections),
        MEI_FIXUP(MEI_UUID_NFC_INFO, blacklist),
        MEI_FIXUP(MEI_UUID_NFC_HCI, mei_nfc),
        MEI_FIXUP(MEI_UUID_WD, mei_wd),
        MEI_FIXUP(MEI_UUID_MKHIF_FIX, mei_mkhi_fix),
        MEI_FIXUP(MEI_UUID_IGSC_MKHI_FIX, mei_gsc_mkhi_fix_ver),
        MEI_FIXUP(MEI_UUID_IGSC_MKHI, mei_gsc_mkhi_ver),
        MEI_FIXUP(MEI_UUID_HDCP, whitelist),
        MEI_FIXUP(MEI_UUID_ANY, vt_support),
        MEI_FIXUP(MEI_UUID_PAVP, pxp_is_ready),
};

/**
 * mei_cl_bus_dev_fixup - run fixup handlers
 *
 * @cldev: me client device
 */
void mei_cl_bus_dev_fixup(struct mei_cl_device *cldev)
{
        struct mei_fixup *f;
        const uuid_le *uuid = mei_me_cl_uuid(cldev->me_cl);
        size_t i;

        for (i = 0; i < ARRAY_SIZE(mei_fixups); i++) {

                f = &mei_fixups[i];
                if (uuid_le_cmp(f->uuid, MEI_UUID_ANY) == 0 ||
                    uuid_le_cmp(f->uuid, *uuid) == 0)
                        f->hook(cldev);
        }
}