// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2011 MediaTek Inc.
 */

#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/init.h>
#include <linux/moduleparam.h>
#include <linux/slab.h>
#include <linux/unistd.h>
#include <linux/sched.h>
#include <linux/fs.h>
#include <linux/uaccess.h>
#include <linux/version.h>
#include <linux/spinlock.h>
#include <linux/semaphore.h>
#include <linux/delay.h>
#include <linux/kthread.h>
#include <linux/errno.h>
#include <linux/cdev.h>
#include <linux/kdev_t.h>
#include <linux/platform_device.h>
#include <linux/device.h>
#include <linux/mutex.h>
#include <linux/vmalloc.h>
#include <linux/poll.h>
#include <linux/proc_fs.h>
#include <linux/string.h>
#include <linux/reboot.h>
#include <linux/reboot-mode.h>
#include <linux/io.h>
#include <linux/device.h>
#include <linux/cdev.h>
#include <linux/proc_fs.h>
#include <linux/seq_file.h>
#include <linux/interrupt.h>
#include <linux/irq.h>
#include <linux/of.h>
#include <linux/of_platform.h>
#include <linux/of_irq.h>
#include <linux/of_address.h>
#include <linux/of_fdt.h>

#include "sec_hal.h"
#include "sec_osal.h"
#include "sec_mod.h"
#include "sec_boot_lib.h"
#include "sec_version.h"

#define MOD                         "MASP"

#define SEC_DEV_NAME                "sec"
#define SEC_MAJOR                   182

#define TRACE_FUNC()                MSG_FUNC(SEC_DEV_NAME)

/*************************************************************************
 *  GLOBAL VARIABLE
 **************************************************************************/
static struct sec_mod sec = { 0 };
static struct cdev sec_dev;
static struct class *sec_class;
static struct device *sec_device;
static const struct of_device_id masp_of_ids[] = {
	{.compatible = "mediatek,masp",},
	{}
};

/**************************************************************************
 *  SEC DRIVER OPEN
 **************************************************************************/
static int sec_open(struct inode *inode, struct file *file)
{
	return 0;
}

/**************************************************************************
 *  SEC DRIVER RELEASE
 **************************************************************************/
static int sec_release(struct inode *inode, struct file *file)
{
	return 0;
}

/**************************************************************************
 *  SEC DRIVER IOCTL
 **************************************************************************/
static long sec_ioctl(struct file *file, unsigned int cmd, unsigned long arg)
{
	return sec_core_ioctl(file, cmd, arg);
}

static const struct file_operations sec_fops = {
	.owner = THIS_MODULE,
	.open = sec_open,
	.release = sec_release,
	.write = NULL,
	.read = NULL,
	.unlocked_ioctl = sec_ioctl
};

/**************************************************************************
 *  SEC RID PROC FUNCTION
 **************************************************************************/
static int sec_proc_rid_show(struct seq_file *m, void *v)
{
	unsigned int rid[4] = { 0 };
	unsigned int i = 0;

	sec_get_random_id((unsigned int *)rid);

	for (i = 0; i < 16; i++)
		seq_putc(m, *((char *)rid + i));

	return 0;
}

static int sec_proc_rid_open(struct inode *inode, struct file *file)
{
	return single_open(file, sec_proc_rid_show, NULL);
}

static const struct proc_ops sec_proc_rid_fops = {
	.proc_open = sec_proc_rid_open,
	.proc_read = seq_read,
	.proc_lseek = seq_lseek,
	.proc_release = seq_release,
};

/**************************************************************************
 *  set_dmverity_reboot eio flag
 **************************************************************************/
// notify_call function
static int reboot_handler_set_eio_flag(struct notifier_block *reboot,
						unsigned long mode,
					  void *cmd)
{	int ret = 0;
	const char *dm_error_cmd = "dm-verity device corrupted";

	if (cmd && !strcmp(cmd, dm_error_cmd))
		ret = masp_hal_set_dm_verity_error();
	return ret;
}

static struct notifier_block reboot_handler_notifier = {
	.notifier_call = reboot_handler_set_eio_flag,
};

/**************************************************************************
 *  SEC MODULE PARAMETER
 **************************************************************************/
static uint recovery_done;
module_param(recovery_done, uint, 0644); /* rw-r--r-- */

MODULE_PARM_DESC(recovery_done,
		 "recovery_done status(0 = complete, 1 = on-going, 2 = error)");

/* SEC DRIVER INIT */
static int sec_init(struct platform_device *dev)
{
	int ret = 0;
	dev_t id = {0};
	struct proc_dir_entry *entry = NULL;

	pr_debug("[%s] %s (%d)\n", SEC_DEV_NAME, __func__, ret);

	id = MKDEV(SEC_MAJOR, 0);
	ret = register_chrdev_region(id, 1, SEC_DEV_NAME);

	if (ret) {
		pr_notice("[%s] Regist Failed (%d)\n", SEC_DEV_NAME, ret);
		return ret;
	}

	sec_class = class_create(THIS_MODULE, SEC_DEV_NAME);
	if (IS_ERR(sec_class)) {
		ret = PTR_ERR(sec_class);
		pr_notice("[%s] Create class failed(0x%x)\n",
			  SEC_DEV_NAME,
			  ret);
		goto err_chedev;
	}

	cdev_init(&sec_dev, &sec_fops);
	sec_dev.owner = THIS_MODULE;

	ret = cdev_add(&sec_dev, id, 1);
	if (ret) {
		pr_notice("[%s] Cdev_add Failed (0x%x)\n",
			SEC_DEV_NAME,
			ret);
		goto err_class;
	}

	sec_device = device_create(sec_class, NULL, id, NULL, SEC_DEV_NAME);
	if (IS_ERR(sec_device)) {
		ret = PTR_ERR(sec_device);
		pr_notice("[%s] Create device failed(0x%x)\n",
			  SEC_DEV_NAME,
			  ret);
		goto err_cdev;
	}

	sec.id = id;
	sec.init = 1;
	spin_lock_init(&sec.lock);

	entry = proc_create("rid", 0444, NULL, &sec_proc_rid_fops);

	if (!entry) {
		ret = -ENOMEM;
		pr_notice("[%s] Create /proc/rid failed(0x%x)\n",
			  SEC_DEV_NAME,
			  ret);
		goto err_device;
	}

	return ret;

err_device:
	device_destroy(sec_class, id);
err_cdev:
	cdev_del(&sec_dev);
err_class:
	class_destroy(sec_class);
err_chedev:
	unregister_chrdev_region(id, 1);
	memset(&sec, 0, sizeof(sec));

	return ret;
}


/**************************************************************************
 *  SEC DRIVER EXIT
 **************************************************************************/
static void sec_exit(void)
{
	remove_proc_entry("rid", NULL);
	device_destroy(sec_class, sec.id);
	cdev_del(&sec_dev);
	class_destroy(sec_class);
	unregister_chrdev_region(sec.id, 1);
	memset(&sec, 0, sizeof(sec));

	sec_core_exit();
}

/**************************************************************************
 *  MASP PLATFORM DRIVER WRAPPER, FOR BUILD-IN SEQUENCE
 **************************************************************************/
int masp_probe(struct platform_device *dev)
{
	int ret = 0;

	ret = sec_init(dev);
	return ret;
}


int masp_remove(struct platform_device *dev)
{
	sec_exit();
	return 0;
}

static struct platform_driver masp_driver = {
	.driver = {
		.name = "masp",
		.owner = THIS_MODULE,
		.of_match_table = masp_of_ids,
	},
	.probe = masp_probe,
	.remove = masp_remove,
};

static int __init masp_get_from_dts(void)
{
	struct masp_tag *tags;
	int i;
	struct device_node *np_chosen = NULL;

	np_chosen = of_find_node_by_path("/chosen");
	if (!np_chosen) {

		np_chosen = of_find_node_by_path("/chosen@0");
		if (!np_chosen)
			return 1;
	}

	tags = (struct masp_tag *)
			of_get_property(np_chosen, "atag,masp", NULL);


	if (!tags)
		return 1;

	g_rom_info_sbc_attr = tags->rom_info_sbc_attr;
	g_rom_info_sdl_attr = tags->rom_info_sdl_attr;
	g_hw_sbcen = tags->hw_sbcen;
	g_lock_state = tags->lock_state;
	lks = tags->lock_state;
	for (i = 0; i < NUM_RID; i++)
		g_random_id[i] = tags->rid[i];
	for (i = 0; i < NUM_CRYPTO_SEED; i++)
		g_crypto_seed[i] = tags->crypto_seed[i];
	for (i = 0; i < NUM_SBC_PUBK_HASH; i++)
		g_sbc_pubk_hash[i] = tags->sbc_pubk_hash[i];

	return 0;
}


static int __init masp_init(void)
{
	int ret;

	masp_get_from_dts();
	ret = platform_driver_register(&masp_driver);
	if (ret) {
		pr_notice("[%s] Reg platform driver failed (%d)\n",
			  SEC_DEV_NAME,
			  ret);
		return ret;
	}
	register_reboot_notifier(&reboot_handler_notifier);
	return ret;
}

static void __exit masp_exit(void)
{
	platform_driver_unregister(&masp_driver);
	unregister_reboot_notifier(&reboot_handler_notifier);
}


module_init(masp_init);
module_exit(masp_exit);

/**************************************************************************
 *  EXPORT FUNCTION
 **************************************************************************/
EXPORT_SYMBOL(sec_get_random_id);

MODULE_LICENSE("GPL v2");
MODULE_AUTHOR("MediaTek Inc.");
MODULE_DESCRIPTION("MediaTek Kernel Security Module");