// SPDX-License-Identifier: GPL-2.0-only

/*
 * Copyright (c) 2021, The Linux Foundation. All rights reserved.
 * Copyright (c) 2022 Qualcomm Innovation Center, Inc. All rights reserved.
 */

#define pr_fmt(fmt) "logbuf_vh: " fmt

#include <linux/init.h>
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/of.h>
#include <linux/platform_device.h>
#include <linux/atomic.h>
#include <linux/slab.h>
#include <soc/qcom/minidump.h>
#include <linux/kallsyms.h>

#include <trace/hooks/logbuf.h>
#include "../../../kernel/printk/printk_ringbuffer.h"
#include "debug_symbol.h"
#include "qcom_logbuf_boot_log.h"

void register_log_minidump(struct printk_ringbuffer *prb)
{
	struct prb_desc_ring descring;
	struct prb_data_ring textdata_ring;
	struct prb_desc *descaddr;
	struct printk_info *p_infos;
	struct md_region md_entry;
	int ret;
	static bool log_buf_registered;

	if (log_buf_registered || !prb)
		return;

	log_buf_registered = true;
	descring = prb->desc_ring;
	textdata_ring = prb->text_data_ring;
	descaddr = descring.descs;
	p_infos = descring.infos;

	strscpy(md_entry.name, "KLOGBUF", sizeof(md_entry.name));
	md_entry.virt_addr = (uintptr_t)textdata_ring.data;
	md_entry.phys_addr = virt_to_phys(textdata_ring.data);
	md_entry.size = _DATA_SIZE(textdata_ring.size_bits);
	ret = msm_minidump_add_region(&md_entry);
	if (ret < 0)
		pr_err("Failed to add log_text entry in minidump table\n");

	strscpy(md_entry.name, "LOG_DESC", sizeof(md_entry.name));
	md_entry.virt_addr = (uintptr_t)descaddr;
	md_entry.phys_addr = virt_to_phys(descaddr);
	md_entry.size = sizeof(struct prb_desc) * _DESCS_COUNT(descring.count_bits);
	ret = msm_minidump_add_region(&md_entry);
	if (ret < 0)
		pr_err("Failed to add log_desc entry in minidump table\n");

	strscpy(md_entry.name, "LOG_INFO", sizeof(md_entry.name));
	md_entry.virt_addr = (uintptr_t)p_infos;
	md_entry.phys_addr = virt_to_phys(p_infos);
	md_entry.size = sizeof(struct printk_info) * _DESCS_COUNT(descring.count_bits);
	ret = msm_minidump_add_region(&md_entry);
	if (ret < 0)
		pr_err("Failed to add log_info entry in minidump table\n");
}

static int logbuf_vh_driver_probe(struct platform_device *pdev)
{
	int ret = 0;
	struct printk_ringbuffer *prb = NULL;

	if (IS_BUILTIN(CONFIG_QCOM_LOGBUF_VENDOR_HOOKS)) {
		prb = *(struct printk_ringbuffer **)kallsyms_lookup_name("prb");
	} else {
		ret = debug_symbol_avaliable();
		if (ret == 0) {
			prb = *(struct printk_ringbuffer **)
				      debug_symbol_lookup_name("prb");
		} else if (ret == -EPROBE_DEFER) {
			ret = -EPROBE_DEFER;
			return ret;
		}
	}
	register_log_minidump(prb);
	ret = boot_log_register(&pdev->dev);

	return ret;
}

static int logbuf_vh_driver_remove(struct platform_device *pdev)
{
	boot_log_unregister();
	return 0;
}

static const struct of_device_id logbuf_vh_of_match[] = {
	{ .compatible = "qcom,logbuf-vendor-hooks" },
	{ }
};
MODULE_DEVICE_TABLE(of, logbuf_vh_of_match);

static struct platform_driver logbuf_vh_driver = {
	.driver = {
		.name = "qcom-logbuf-vh",
		.of_match_table = logbuf_vh_of_match,
	},
	.probe = logbuf_vh_driver_probe,
	.remove = logbuf_vh_driver_remove,
};
module_platform_driver(logbuf_vh_driver);

MODULE_DESCRIPTION("QCOM Logbuf Vendor Hooks Driver");
MODULE_LICENSE("GPL v2");