// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) Meta Platforms, Inc. and affiliates. */

#include <linux/unaligned.h>
#include <linux/pci.h>
#include <linux/pldmfw.h>
#include <linux/types.h>
#include <net/devlink.h>

#include "fbnic.h"
#include "fbnic_fw.h"
#include "fbnic_tlv.h"

#define FBNIC_SN_STR_LEN	24

static int fbnic_version_running_put(struct devlink_info_req *req,
				     struct fbnic_fw_ver *fw_ver,
				     char *ver_name)
{
	char running_ver[FBNIC_FW_VER_MAX_SIZE];
	int err;

	fbnic_mk_fw_ver_str(fw_ver->version, running_ver);
	err = devlink_info_version_running_put(req, ver_name, running_ver);
	if (err)
		return err;

	if (strlen(fw_ver->commit) > 0) {
		char commit_name[FBNIC_SN_STR_LEN];

		snprintf(commit_name, FBNIC_SN_STR_LEN, "%s.commit", ver_name);
		err = devlink_info_version_running_put(req, commit_name,
						       fw_ver->commit);
		if (err)
			return err;
	}

	return 0;
}

static int fbnic_version_stored_put(struct devlink_info_req *req,
				    struct fbnic_fw_ver *fw_ver,
				    char *ver_name)
{
	char stored_ver[FBNIC_FW_VER_MAX_SIZE];
	int err;

	fbnic_mk_fw_ver_str(fw_ver->version, stored_ver);
	err = devlink_info_version_stored_put(req, ver_name, stored_ver);
	if (err)
		return err;

	if (strlen(fw_ver->commit) > 0) {
		char commit_name[FBNIC_SN_STR_LEN];

		snprintf(commit_name, FBNIC_SN_STR_LEN, "%s.commit", ver_name);
		err = devlink_info_version_stored_put(req, commit_name,
						      fw_ver->commit);
		if (err)
			return err;
	}

	return 0;
}

static int fbnic_devlink_info_get(struct devlink *devlink,
				  struct devlink_info_req *req,
				  struct netlink_ext_ack *extack)
{
	struct fbnic_dev *fbd = devlink_priv(devlink);
	int err;

	err = fbnic_version_running_put(req, &fbd->fw_cap.running.mgmt,
					DEVLINK_INFO_VERSION_GENERIC_FW);
	if (err)
		return err;

	err = fbnic_version_running_put(req, &fbd->fw_cap.running.bootloader,
					DEVLINK_INFO_VERSION_GENERIC_FW_BOOTLOADER);
	if (err)
		return err;

	err = fbnic_version_stored_put(req, &fbd->fw_cap.stored.mgmt,
				       DEVLINK_INFO_VERSION_GENERIC_FW);
	if (err)
		return err;

	err = fbnic_version_stored_put(req, &fbd->fw_cap.stored.bootloader,
				       DEVLINK_INFO_VERSION_GENERIC_FW_BOOTLOADER);
	if (err)
		return err;

	err = fbnic_version_stored_put(req, &fbd->fw_cap.stored.undi,
				       DEVLINK_INFO_VERSION_GENERIC_FW_UNDI);
	if (err)
		return err;

	if (fbd->dsn) {
		unsigned char serial[FBNIC_SN_STR_LEN];
		u8 dsn[8];

		put_unaligned_be64(fbd->dsn, dsn);
		err = snprintf(serial, FBNIC_SN_STR_LEN, "%8phD", dsn);
		if (err < 0)
			return err;

		err = devlink_info_serial_number_put(req, serial);
		if (err)
			return err;
	}

	return 0;
}

static bool
fbnic_pldm_match_record(struct pldmfw *context, struct pldmfw_record *record)
{
	struct pldmfw_desc_tlv *desc;
	u32 anti_rollback_ver = 0;
	struct devlink *devlink;
	struct fbnic_dev *fbd;
	struct pci_dev *pdev;

	/* First, use the standard PCI matching function */
	if (!pldmfw_op_pci_match_record(context, record))
		return false;

	pdev = to_pci_dev(context->dev);
	fbd = pci_get_drvdata(pdev);
	devlink = priv_to_devlink(fbd);

	/* If PCI match is successful, check for vendor-specific descriptors */
	list_for_each_entry(desc, &record->descs, entry) {
		if (desc->type != PLDM_DESC_ID_VENDOR_DEFINED)
			continue;

		if (desc->size < 21 || desc->data[0] != 1 ||
		    desc->data[1] != 15)
			continue;

		if (memcmp(desc->data + 2, "AntiRollbackVer", 15) != 0)
			continue;

		anti_rollback_ver = get_unaligned_le32(desc->data + 17);
		break;
	}

	/* Compare versions and return error if they do not match */
	if (anti_rollback_ver < fbd->fw_cap.anti_rollback_version) {
		char buf[128];

		snprintf(buf, sizeof(buf),
			 "New firmware anti-rollback version (0x%x) is older than device version (0x%x)!",
			 anti_rollback_ver, fbd->fw_cap.anti_rollback_version);
		devlink_flash_update_status_notify(devlink, buf,
						   "Anti-Rollback", 0, 0);

		return false;
	}

	return true;
}

static int
fbnic_flash_start(struct fbnic_dev *fbd, struct pldmfw_component *component)
{
	struct fbnic_fw_completion *cmpl;
	int err;

	cmpl = fbnic_fw_alloc_cmpl(FBNIC_TLV_MSG_ID_FW_START_UPGRADE_REQ);
	if (!cmpl)
		return -ENOMEM;

	err = fbnic_fw_xmit_fw_start_upgrade(fbd, cmpl,
					     component->identifier,
					     component->component_size);
	if (err)
		goto cmpl_free;

	/* Wait for firmware to ack firmware upgrade start */
	if (wait_for_completion_timeout(&cmpl->done, 10 * HZ))
		err = cmpl->result;
	else
		err = -ETIMEDOUT;

	fbnic_mbx_clear_cmpl(fbd, cmpl);
cmpl_free:
	fbnic_fw_put_cmpl(cmpl);

	return err;
}

static int
fbnic_flash_component(struct pldmfw *context,
		      struct pldmfw_component *component)
{
	const u8 *data = component->component_data;
	const u32 size = component->component_size;
	struct fbnic_fw_completion *cmpl;
	const char *component_name;
	struct devlink *devlink;
	struct fbnic_dev *fbd;
	struct pci_dev *pdev;
	u32 offset = 0;
	u32 length = 0;
	char buf[32];
	int err;

	pdev = to_pci_dev(context->dev);
	fbd = pci_get_drvdata(pdev);
	devlink = priv_to_devlink(fbd);

	switch (component->identifier) {
	case QSPI_SECTION_CMRT:
		component_name = "boot1";
		break;
	case QSPI_SECTION_CONTROL_FW:
		component_name = "boot2";
		break;
	case QSPI_SECTION_OPTION_ROM:
		component_name = "option-rom";
		break;
	default:
		snprintf(buf, sizeof(buf), "Unknown component ID %u!",
			 component->identifier);
		devlink_flash_update_status_notify(devlink, buf, NULL, 0,
						   size);
		return -EINVAL;
	}

	/* Once firmware receives the request to start upgrading it responds
	 * with two messages:
	 * 1. An ACK that it received the message and possible error code
	 *    indicating that an upgrade is not currently possible.
	 * 2. A request for the first chunk of data
	 *
	 * Setup completions for write before issuing the start message so
	 * the driver can catch both messages.
	 */
	cmpl = fbnic_fw_alloc_cmpl(FBNIC_TLV_MSG_ID_FW_WRITE_CHUNK_REQ);
	if (!cmpl)
		return -ENOMEM;

	err = fbnic_mbx_set_cmpl(fbd, cmpl);
	if (err)
		goto cmpl_free;

	devlink_flash_update_timeout_notify(devlink, "Initializing",
					    component_name, 15);
	err = fbnic_flash_start(fbd, component);
	if (err)
		goto err_no_msg;

	while (offset < size) {
		if (!wait_for_completion_timeout(&cmpl->done, 15 * HZ)) {
			err = -ETIMEDOUT;
			break;
		}

		err = cmpl->result;
		if (err)
			break;

		/* Verify firmware is requesting the next chunk in the seq. */
		if (cmpl->u.fw_update.offset != offset + length) {
			err = -EFAULT;
			break;
		}

		offset = cmpl->u.fw_update.offset;
		length = cmpl->u.fw_update.length;

		if (length > TLV_MAX_DATA || offset + length > size) {
			err = -EFAULT;
			break;
		}

		devlink_flash_update_status_notify(devlink, "Flashing",
						   component_name,
						   offset, size);

		/* Mailbox will set length to 0 once it receives the finish
		 * message.
		 */
		if (!length)
			continue;

		reinit_completion(&cmpl->done);
		err = fbnic_fw_xmit_fw_write_chunk(fbd, data, offset, length,
						   0);
		if (err)
			break;
	}

	if (err) {
		fbnic_fw_xmit_fw_write_chunk(fbd, NULL, 0, 0, err);
err_no_msg:
		snprintf(buf, sizeof(buf), "Mailbox encountered error %d!",
			 err);
		devlink_flash_update_status_notify(devlink, buf,
						   component_name, 0, 0);
	}

	fbnic_mbx_clear_cmpl(fbd, cmpl);
cmpl_free:
	fbnic_fw_put_cmpl(cmpl);

	return err;
}

static const struct pldmfw_ops fbnic_pldmfw_ops = {
	.match_record = fbnic_pldm_match_record,
	.flash_component = fbnic_flash_component,
};

static int
fbnic_devlink_flash_update(struct devlink *devlink,
			   struct devlink_flash_update_params *params,
			   struct netlink_ext_ack *extack)
{
	struct fbnic_dev *fbd = devlink_priv(devlink);
	const struct firmware *fw = params->fw;
	struct device *dev = fbd->dev;
	struct pldmfw context;
	char *err_msg;
	int err;

	context.ops = &fbnic_pldmfw_ops;
	context.dev = dev;

	err = pldmfw_flash_image(&context, fw);
	if (err) {
		switch (err) {
		case -EINVAL:
			err_msg = "Invalid image";
			break;
		case -EOPNOTSUPP:
			err_msg = "Unsupported image";
			break;
		case -ENOMEM:
			err_msg = "Out of memory";
			break;
		case -EFAULT:
			err_msg = "Invalid header";
			break;
		case -ENOENT:
			err_msg = "No matching record";
			break;
		case -ENODEV:
			err_msg = "No matching device";
			break;
		case -ETIMEDOUT:
			err_msg = "Timed out waiting for reply";
			break;
		default:
			err_msg = "Unknown error";
			break;
		}

		NL_SET_ERR_MSG_FMT_MOD(extack,
				       "Failed to flash PLDM Image: %s (error: %d)",
				       err_msg, err);
	}

	return err;
}

static const struct devlink_ops fbnic_devlink_ops = {
	.info_get	= fbnic_devlink_info_get,
	.flash_update	= fbnic_devlink_flash_update,
};

static int fbnic_fw_reporter_dump(struct devlink_health_reporter *reporter,
				  struct devlink_fmsg *fmsg, void *priv_ctx,
				  struct netlink_ext_ack *extack)
{
	struct fbnic_dev *fbd = devlink_health_reporter_priv(reporter);
	u32 offset, index, index_count, length, size;
	struct fbnic_fw_completion *fw_cmpl;
	u8 *dump_data, **data;
	int err;

	fw_cmpl = fbnic_fw_alloc_cmpl(FBNIC_TLV_MSG_ID_COREDUMP_GET_INFO_RESP);
	if (!fw_cmpl)
		return -ENOMEM;

	err = fbnic_fw_xmit_coredump_info_msg(fbd, fw_cmpl, true);
	if (err) {
		NL_SET_ERR_MSG_MOD(extack,
				   "Failed to transmit core dump info msg");
		goto cmpl_free;
	}
	if (!wait_for_completion_timeout(&fw_cmpl->done, 2 * HZ)) {
		NL_SET_ERR_MSG_MOD(extack,
				   "Timed out waiting on core dump info");
		err = -ETIMEDOUT;
		goto cmpl_cleanup;
	}

	size = fw_cmpl->u.coredump_info.size;
	err = fw_cmpl->result;

	fbnic_mbx_clear_cmpl(fbd, fw_cmpl);
	fbnic_fw_put_cmpl(fw_cmpl);

	/* Handle error returned by firmware */
	if (err) {
		NL_SET_ERR_MSG_MOD(extack, "Firmware core dump returned error");
		return err;
	}
	if (!size) {
		NL_SET_ERR_MSG_MOD(extack,
				   "Firmware core dump returned size 0");
		return -EIO;
	}

	/* Read the dump, we can only transfer TLV_MAX_DATA at a time */
	index_count = DIV_ROUND_UP(size, TLV_MAX_DATA);

	fw_cmpl = __fbnic_fw_alloc_cmpl(FBNIC_TLV_MSG_ID_COREDUMP_READ_RESP,
					sizeof(void *) * index_count + size);
	if (!fw_cmpl)
		return -ENOMEM;

	/* Populate pointer table w/ pointer offsets */
	dump_data = (void *)&fw_cmpl->u.coredump.data[index_count];
	data = fw_cmpl->u.coredump.data;
	fw_cmpl->u.coredump.size = size;
	fw_cmpl->u.coredump.stride = TLV_MAX_DATA;

	for (index = 0; index < index_count; index++) {
		/* First iteration installs completion */
		struct fbnic_fw_completion *cmpl_arg = index ? NULL : fw_cmpl;

		offset = index * TLV_MAX_DATA;
		length = min(size - offset, TLV_MAX_DATA);

		data[index] = dump_data + offset;
		err = fbnic_fw_xmit_coredump_read_msg(fbd, cmpl_arg,
						      offset, length);
		if (err) {
			NL_SET_ERR_MSG_MOD(extack,
					   "Failed to transmit core dump msg");
			if (cmpl_arg)
				goto cmpl_free;
			else
				goto cmpl_cleanup;
		}

		if (wait_for_completion_timeout(&fw_cmpl->done, 2 * HZ)) {
			reinit_completion(&fw_cmpl->done);
		} else {
			NL_SET_ERR_MSG_FMT_MOD(extack,
					       "Timed out waiting on core dump (%d/%d)",
					       index + 1, index_count);
			err = -ETIMEDOUT;
			goto cmpl_cleanup;
		}

		/* If we didn't see the reply record as incomplete */
		if (fw_cmpl->u.coredump.data[index]) {
			NL_SET_ERR_MSG_FMT_MOD(extack,
					       "No data for core dump chunk (%d/%d)",
					       index + 1, index_count);
			err = -EIO;
			goto cmpl_cleanup;
		}
	}

	devlink_fmsg_binary_pair_nest_start(fmsg, "FW coredump");

	for (offset = 0; offset < size; offset += length) {
		length = min_t(u32, size - offset, TLV_MAX_DATA);

		devlink_fmsg_binary_put(fmsg, dump_data + offset, length);
	}

	devlink_fmsg_binary_pair_nest_end(fmsg);

cmpl_cleanup:
	fbnic_mbx_clear_cmpl(fbd, fw_cmpl);
cmpl_free:
	fbnic_fw_put_cmpl(fw_cmpl);

	return err;
}

static int
fbnic_fw_reporter_diagnose(struct devlink_health_reporter *reporter,
			   struct devlink_fmsg *fmsg,
			   struct netlink_ext_ack *extack)
{
	struct fbnic_dev *fbd = devlink_health_reporter_priv(reporter);
	u32 sec, msec;

	/* Device is most likely down, we're not exchanging heartbeats */
	if (!fbd->prev_firmware_time)
		return 0;

	sec = div_u64_rem(fbd->firmware_time, MSEC_PER_SEC, &msec);

	devlink_fmsg_pair_nest_start(fmsg, "last_heartbeat");
	devlink_fmsg_obj_nest_start(fmsg);
	devlink_fmsg_pair_nest_start(fmsg, "fw_uptime");
	devlink_fmsg_obj_nest_start(fmsg);
	devlink_fmsg_u32_pair_put(fmsg, "sec", sec);
	devlink_fmsg_u32_pair_put(fmsg, "msec", msec);
	devlink_fmsg_obj_nest_end(fmsg);
	devlink_fmsg_pair_nest_end(fmsg);
	devlink_fmsg_obj_nest_end(fmsg);
	devlink_fmsg_pair_nest_end(fmsg);

	return 0;
}

void __printf(2, 3)
fbnic_devlink_fw_report(struct fbnic_dev *fbd, const char *format, ...)
{
	char msg[FBNIC_FW_LOG_MAX_SIZE];
	va_list args;

	va_start(args, format);
	vsnprintf(msg, FBNIC_FW_LOG_MAX_SIZE, format, args);
	va_end(args);

	devlink_health_report(fbd->fw_reporter, msg, fbd);
	if (fbnic_fw_log_ready(fbd))
		fbnic_fw_log_write(fbd, 0, fbd->firmware_time, msg);
}

static const struct devlink_health_reporter_ops fbnic_fw_ops = {
	.name = "fw",
	.dump = fbnic_fw_reporter_dump,
	.diagnose = fbnic_fw_reporter_diagnose,
};

static u32 fbnic_read_otp_status(struct fbnic_dev *fbd)
{
	return fbnic_fw_rd32(fbd, FBNIC_NS_OTP_STATUS);
}

static int
fbnic_otp_reporter_dump(struct devlink_health_reporter *reporter,
			struct devlink_fmsg *fmsg, void *priv_ctx,
			struct netlink_ext_ack *extack)
{
	struct fbnic_dev *fbd = devlink_health_reporter_priv(reporter);
	u32 otp_status, otp_write_status, m;

	otp_status = fbnic_read_otp_status(fbd);
	otp_write_status = fbnic_fw_rd32(fbd, FBNIC_NS_OTP_WRITE_STATUS);

	/* Dump OTP status */
	devlink_fmsg_pair_nest_start(fmsg, "OTP");
	devlink_fmsg_obj_nest_start(fmsg);

	devlink_fmsg_u32_pair_put(fmsg, "Status", otp_status);

	/* Extract OTP Write Data status */
	m = FBNIC_NS_OTP_WRITE_DATA_STATUS_MASK;
	devlink_fmsg_u32_pair_put(fmsg, "Data",
				  FIELD_GET(m, otp_write_status));

	/* Extract OTP Write ECC status */
	m = FBNIC_NS_OTP_WRITE_ECC_STATUS_MASK;
	devlink_fmsg_u32_pair_put(fmsg, "ECC",
				  FIELD_GET(m, otp_write_status));

	devlink_fmsg_obj_nest_end(fmsg);
	devlink_fmsg_pair_nest_end(fmsg);

	return 0;
}

void fbnic_devlink_otp_check(struct fbnic_dev *fbd, const char *msg)
{
	/* Check if there is anything to report */
	if (!fbnic_read_otp_status(fbd))
		return;

	devlink_health_report(fbd->otp_reporter, msg, fbd);
	if (fbnic_fw_log_ready(fbd))
		fbnic_fw_log_write(fbd, 0, fbd->firmware_time, msg);
}

static const struct devlink_health_reporter_ops fbnic_otp_ops = {
	.name = "otp",
	.dump = fbnic_otp_reporter_dump,
};

int fbnic_devlink_health_create(struct fbnic_dev *fbd)
{
	fbd->fw_reporter = devlink_health_reporter_create(priv_to_devlink(fbd),
							  &fbnic_fw_ops, fbd);
	if (IS_ERR(fbd->fw_reporter)) {
		dev_warn(fbd->dev,
			 "Failed to create FW fault reporter: %pe\n",
			 fbd->fw_reporter);
		return PTR_ERR(fbd->fw_reporter);
	}

	fbd->otp_reporter = devlink_health_reporter_create(priv_to_devlink(fbd),
							   &fbnic_otp_ops, fbd);
	if (IS_ERR(fbd->otp_reporter)) {
		devlink_health_reporter_destroy(fbd->fw_reporter);
		dev_warn(fbd->dev,
			 "Failed to create OTP fault reporter: %pe\n",
			 fbd->otp_reporter);
		return PTR_ERR(fbd->otp_reporter);
	}

	return 0;
}

void fbnic_devlink_health_destroy(struct fbnic_dev *fbd)
{
	devlink_health_reporter_destroy(fbd->otp_reporter);
	devlink_health_reporter_destroy(fbd->fw_reporter);
}

void fbnic_devlink_free(struct fbnic_dev *fbd)
{
	struct devlink *devlink = priv_to_devlink(fbd);

	devlink_free(devlink);
}

struct fbnic_dev *fbnic_devlink_alloc(struct pci_dev *pdev)
{
	void __iomem * const *iomap_table;
	struct devlink *devlink;
	struct fbnic_dev *fbd;

	devlink = devlink_alloc(&fbnic_devlink_ops, sizeof(struct fbnic_dev),
				&pdev->dev);
	if (!devlink)
		return NULL;

	fbd = devlink_priv(devlink);
	pci_set_drvdata(pdev, fbd);
	fbd->dev = &pdev->dev;

	iomap_table = pcim_iomap_table(pdev);
	fbd->uc_addr0 = iomap_table[0];
	fbd->uc_addr4 = iomap_table[4];

	fbd->dsn = pci_get_dsn(pdev);
	fbd->mps = pcie_get_mps(pdev);
	fbd->readrq = pcie_get_readrq(pdev);

	fbd->mac_addr_boundary = FBNIC_RPC_TCAM_MACDA_DEFAULT_BOUNDARY;

	return fbd;
}

void fbnic_devlink_register(struct fbnic_dev *fbd)
{
	struct devlink *devlink = priv_to_devlink(fbd);

	devlink_register(devlink);
}

void fbnic_devlink_unregister(struct fbnic_dev *fbd)
{
	struct devlink *devlink = priv_to_devlink(fbd);

	devlink_unregister(devlink);
}