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

#include <linux/spinlock.h>
#include <linux/vmalloc.h>

#include "fbnic.h"
#include "fbnic_fw.h"
#include "fbnic_fw_log.h"

void fbnic_fw_log_enable(struct fbnic_dev *fbd, bool send_hist)
{
	int err;

	if (!fbnic_fw_log_ready(fbd))
		return;

	if (fbd->fw_cap.running.mgmt.version < MIN_FW_VER_CODE_HIST)
		send_hist = false;

	err = fbnic_fw_xmit_send_logs(fbd, true, send_hist);
	if (err && err != -EOPNOTSUPP)
		dev_warn(fbd->dev, "Unable to enable firmware logs: %d\n", err);
}

void fbnic_fw_log_disable(struct fbnic_dev *fbd)
{
	int err;

	err = fbnic_fw_xmit_send_logs(fbd, false, false);
	if (err && err != -EOPNOTSUPP)
		dev_warn(fbd->dev, "Unable to disable firmware logs: %d\n",
			 err);
}

int fbnic_fw_log_init(struct fbnic_dev *fbd)
{
	struct fbnic_fw_log *log = &fbd->fw_log;
	void *data;

	if (WARN_ON_ONCE(fbnic_fw_log_ready(fbd)))
		return -EEXIST;

	data = vmalloc(FBNIC_FW_LOG_SIZE);
	if (!data)
		return -ENOMEM;

	spin_lock_init(&fbd->fw_log.lock);
	INIT_LIST_HEAD(&log->entries);
	log->size = FBNIC_FW_LOG_SIZE;
	log->data_start = data;
	log->data_end = data + FBNIC_FW_LOG_SIZE;

	fbnic_fw_log_enable(fbd, true);

	return 0;
}

void fbnic_fw_log_free(struct fbnic_dev *fbd)
{
	struct fbnic_fw_log *log = &fbd->fw_log;

	if (!fbnic_fw_log_ready(fbd))
		return;

	fbnic_fw_log_disable(fbd);
	INIT_LIST_HEAD(&log->entries);
	log->size = 0;
	vfree(log->data_start);
	log->data_start = NULL;
	log->data_end = NULL;
}

int fbnic_fw_log_write(struct fbnic_dev *fbd, u64 index, u32 timestamp,
		       char *msg)
{
	struct fbnic_fw_log_entry *entry, *head, *tail, *next;
	struct fbnic_fw_log *log = &fbd->fw_log;
	size_t msg_len = strlen(msg) + 1;
	unsigned long flags;
	void *entry_end;

	if (!fbnic_fw_log_ready(fbd)) {
		dev_err(fbd->dev, "Firmware sent log entry without being requested!\n");
		return -ENOSPC;
	}

	spin_lock_irqsave(&log->lock, flags);

	if (list_empty(&log->entries)) {
		entry = log->data_start;
	} else {
		head = list_first_entry(&log->entries, typeof(*head), list);
		entry_end = head->msg + head->len + 1;
		entry = PTR_ALIGN(entry_end, 8);
	}

	entry_end = entry->msg + msg_len + 1;

	/* We've reached the end of the buffer, wrap around */
	if (entry_end > log->data_end) {
		entry = log->data_start;
		entry_end = entry->msg + msg_len + 1;
	}

	/* Make room for entry by removing from tail. */
	list_for_each_entry_safe_reverse(tail, next, &log->entries, list) {
		if (entry <= tail && entry_end > (void *)tail)
			list_del(&tail->list);
		else
			break;
	}

	entry->index = index;
	entry->timestamp = timestamp;
	entry->len = msg_len;
	strscpy(entry->msg, msg, entry->len);
	list_add(&entry->list, &log->entries);

	spin_unlock_irqrestore(&log->lock, flags);

	return 0;
}