// 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;

        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;

        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,
                       const 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;
}