/*-
 * Broadcom NetXtreme-C/E network driver.
 *
 * Copyright (c) 2026 Broadcom Inc.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS'
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
 * THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <sys/param.h>
#include <sys/endian.h>
#include <sys/lock.h>
#include <sys/malloc.h>
#include <sys/mutex.h>
#include <sys/systm.h>

#include "bnxt.h"
#include "bnxt_log.h"

MALLOC_DEFINE(M_BNXT_LOG, "bnxt_log_buffer", "buffer for bnxt logging module");

void
bnxt_fill_coredump_seg_hdr(struct bnxt_softc *bp,
			   struct bnxt_coredump_segment_hdr *seg_hdr,
			   struct coredump_segment_record *seg_rec,
			   uint32_t seg_len, int status, uint32_t duration,
			   uint32_t instance, uint32_t comp_id, uint32_t seg_id)
{
	memset(seg_hdr, 0, sizeof(*seg_hdr));
	memcpy(seg_hdr->signature, "sEgM", 4);
	if (seg_rec) {
		seg_hdr->component_id =
		    htole32((uint32_t)seg_rec->component_id);
		seg_hdr->segment_id = htole32((uint32_t)seg_rec->segment_id);
		seg_hdr->low_version = seg_rec->version_low;
		seg_hdr->high_version = seg_rec->version_hi;
		seg_hdr->flags = seg_rec->compress_flags;
	} else {
		seg_hdr->component_id = htole32(comp_id);
		seg_hdr->segment_id = htole32(seg_id);
	}
	seg_hdr->function_id =  htole16(bp->func.fw_fid);
	seg_hdr->length = htole32(seg_len);
	seg_hdr->status = htole32((uint32_t)status);
	seg_hdr->duration = htole32(duration);
	seg_hdr->data_offset = htole32(sizeof(*seg_hdr));
	seg_hdr->instance = htole32(instance);
}

int
bnxt_register_logger(struct bnxt_softc *bp, uint16_t logger_id,
		     uint32_t num_buffs, void (*log_live)(void *),
		     uint32_t live_max_size)
{
	struct bnxt_logger *logger;
	void *data;

	if (!log_live || !live_max_size)
		return (EINVAL);

	if (num_buffs == 0 || (num_buffs & (num_buffs - 1)) != 0)
		return (EINVAL);

	logger = malloc(sizeof(*logger), M_BNXT_LOG, M_WAITOK | M_ZERO);

	logger->logger_id = logger_id;
	logger->buffer_size = num_buffs * BNXT_LOG_MSG_SIZE;
	logger->log_live_op = log_live;
	logger->max_live_buff_size = live_max_size;

	data = malloc(logger->buffer_size, M_BNXT_LOG, M_WAITOK);
	logger->msgs = data;

	mtx_lock(&bp->log_lock);
	TAILQ_INSERT_TAIL(&bp->loggers_list, logger, list);
	mtx_unlock(&bp->log_lock);
	return (0);
}

void
bnxt_unregister_logger(struct bnxt_softc *bp, int logger_id)
{
	struct bnxt_logger *l = NULL, *tmp;

	mtx_lock(&bp->log_lock);
	TAILQ_FOREACH_SAFE(l, &bp->loggers_list, list, tmp) {
		if (l->logger_id == logger_id) {
			TAILQ_REMOVE(&bp->loggers_list, l, list);
			break;
		}
	}
	mtx_unlock(&bp->log_lock);

	if (!l) {
		device_printf(bp->dev, "logger id %d not registered\n",
			      logger_id);
		return;
	}

	free(l->msgs, M_BNXT_LOG);
	free(l, M_BNXT_LOG);
}

static int
bnxt_log_info(char *buf, size_t max_len, const char *format, va_list args)
{
	static char textbuf[BNXT_LOG_MSG_SIZE];
	char *text = textbuf;
	size_t text_len;
	char *next;

	text_len = vsnprintf(text, sizeof(textbuf), format, args);
	if (text_len >= sizeof(textbuf))
		text_len = sizeof(textbuf) - 1;

	next = memchr(text, '\n', text_len);
	if (next)
		text_len = next - text;
	else if (text[text_len] == '\0')
		text[text_len] = '\n';

	if (text_len > max_len) {
		/* Truncate */
		text_len = max_len;
		text[text_len] = '\n';
	}

	memcpy(buf, text, text_len + 1);

	return (text_len + 1);
}

void
bnxt_log_add_msg(struct bnxt_softc *bp, uint16_t logger_id,
    const char *format, ...)
{
	struct bnxt_logger *logger = NULL, *tmp;
	uint16_t start, tail;
	va_list args;
	void *buf;
	uint32_t mask;

	mtx_lock(&bp->log_lock);
	TAILQ_FOREACH_SAFE(logger, &bp->loggers_list, list, tmp) {
		if (logger->logger_id == logger_id)
			break;
	}

	if (!logger) {
		mtx_unlock(&bp->log_lock);
		return;
	}

	mask = BNXT_LOG_NUM_BUFFERS(logger->buffer_size) - 1;
	tail = logger->tail;
	start = logger->head;

	if (logger->valid && start == tail)
		logger->head = ++start & mask;

	buf = (uint8_t *)logger->msgs + BNXT_LOG_MSG_SIZE * logger->tail;
	logger->tail = ++tail & mask;

	if (!logger->valid)
		logger->valid = true;

	va_start(args, format);
	bnxt_log_info(buf, BNXT_LOG_MSG_SIZE, format, args);
	va_end(args);
	mtx_unlock(&bp->log_lock);
}

void
bnxt_log_live(struct bnxt_softc *bp, uint16_t logger_id,
    const char *format, ...)
{
	struct bnxt_logger *logger = NULL, *tmp;
	va_list args;
	int len;

	TAILQ_FOREACH_SAFE(logger, &bp->loggers_list, list, tmp) {
		if (logger->logger_id == logger_id)
			break;
	}

	if (!logger || !logger->live_msgs)
		return;

	va_start(args, format);
	len = bnxt_log_info(
	    (uint8_t *)logger->live_msgs + logger->live_msgs_len,
	    logger->max_live_buff_size - logger->live_msgs_len,
	    format, args);
	va_end(args);

	logger->live_msgs_len += len;
}

static size_t
bnxt_get_data_len(char *buf)
{
	size_t count = 0;

	while (*buf++ != '\n')
		count++;
	return (count + 1);
}

static size_t
bnxt_collect_logs_buffer(struct bnxt_logger *logger, char *dest)
{
	uint32_t mask = BNXT_LOG_NUM_BUFFERS(logger->buffer_size) - 1;
	uint16_t head = logger->head;
	uint16_t tail = logger->tail;
	size_t total_len = 0;
	int count;

	if (!logger->valid)
		return (0);

	count = (tail > head) ? (tail - head) : (tail - head + mask + 1);
	while (count--) {
		void *src = (uint8_t *)logger->msgs +
		    BNXT_LOG_MSG_SIZE * (head & mask);
		size_t len;

		len = bnxt_get_data_len(src);
		memcpy(dest + total_len, src, len);
		total_len += len;
		head++;
	}

	return (total_len);
}

size_t
bnxt_get_loggers_coredump_size(struct bnxt_softc *bp)
{
	struct bnxt_logger *logger, *tmp;
	size_t len = 0;

	mtx_lock(&bp->log_lock);
	TAILQ_FOREACH_SAFE(logger, &bp->loggers_list, list, tmp) {
		len += sizeof(struct bnxt_coredump_segment_hdr) +
		       logger->max_live_buff_size + logger->buffer_size;
	}
	mtx_unlock(&bp->log_lock);
	return (len);
}

int
bnxt_start_logging_driver_coredump(struct bnxt_softc *bp, char *dest_buf)
{
	struct bnxt_logger *logger, *tmp;
	size_t offset = 0;
	uint32_t seg_id = 0;

	if (!dest_buf)
		return (0);

	mtx_lock(&bp->log_lock);
	TAILQ_FOREACH_SAFE(logger, &bp->loggers_list, list, tmp) {
		struct bnxt_coredump_segment_hdr seg_hdr;
		void *seg_hdr_dest = dest_buf + offset;
		size_t len;

		offset += sizeof(seg_hdr);
		/* First collect logs from buffer */
		len = bnxt_collect_logs_buffer(logger, dest_buf + offset);
		offset += len;
		/* Let logger to collect live messages */
		logger->live_msgs = dest_buf + offset;
		logger->live_msgs_len = 0;
		logger->log_live_op(bp);

		len += logger->buffer_size;
		offset += logger->buffer_size;

		bnxt_fill_coredump_seg_hdr(bp, &seg_hdr, NULL, len,
					   0, 0, 0, 13, seg_id);
		memcpy(seg_hdr_dest, &seg_hdr, sizeof(seg_hdr));
		seg_id++;
	}
	mtx_unlock(&bp->log_lock);
	return (offset);
}

void
bnxt_reset_loggers(struct bnxt_softc *bp)
{
	struct bnxt_logger *logger, *tmp;

	mtx_lock(&bp->log_lock);
	TAILQ_FOREACH_SAFE(logger, &bp->loggers_list, list, tmp) {
		logger->head = 0;
		logger->tail = 0;
		logger->valid = false;
	}
	mtx_unlock(&bp->log_lock);
}