/*
 * Copyright (c) 2016, Fuzhou Rockchip Electronics Co., Ltd
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 */

#include <asm/cacheflush.h>
#include <linux/clk.h>
#include <linux/debugfs.h>
#include <linux/dma-mapping.h>
#include <linux/irq.h>
#include <linux/interrupt.h>
#include <linux/kernel.h>
#include <linux/miscdevice.h>
#include <linux/module.h>
#ifdef CONFIG_OF
#include <linux/of.h>
#endif
#include <linux/platform_device.h>
#include <linux/pm_runtime.h>
#include <linux/slab.h>
#include <linux/timer.h>
#include <linux/uaccess.h>

#include "rk_nand_blk.h"
#include "rk_ftl_api.h"
#include "rk_nand_base.h"

#define RKNAND_VERSION_AND_DATE  "rknandbase v1.2 2021-01-07"

static struct rk_nandc_info g_nandc_info[2];
struct device *g_nand_device;
static char nand_idb_data[2048];
static int rk_nand_wait_busy_schedule;
static int rk_nand_suspend_state;
static int rk_nand_shutdown_state;
/*1:flash 2:emmc 4:sdcard0 8:sdcard1*/
static int rknand_boot_media = 2;
static DECLARE_WAIT_QUEUE_HEAD(rk29_nandc_wait);
static void rk_nand_iqr_timeout_hack(struct timer_list *unused);
static DEFINE_TIMER(rk_nand_iqr_timeout, rk_nand_iqr_timeout_hack);
static int nandc0_xfer_completed_flag;
static int nandc0_ready_completed_flag;
static int nandc1_xfer_completed_flag;
static int nandc1_ready_completed_flag;
static int rk_timer_add;

void *ftl_malloc(int size)
{
	return kmalloc(size, GFP_KERNEL | GFP_DMA);
}

void ftl_free(void *buf)
{
	kfree(buf);
}

int rknand_get_clk_rate(int nandc_id)
{
	return g_nandc_info[nandc_id].clk_rate;
}
EXPORT_SYMBOL(rknand_get_clk_rate);

unsigned long rknand_dma_map_single(unsigned long ptr, int size, int dir)
{
	return dma_map_single(g_nand_device, (void *)ptr, size
		, dir ? DMA_TO_DEVICE : DMA_FROM_DEVICE);
}
EXPORT_SYMBOL(rknand_dma_map_single);

void rknand_dma_unmap_single(unsigned long ptr, int size, int dir)
{
	dma_unmap_single(g_nand_device, (dma_addr_t)ptr, size
		, dir ? DMA_TO_DEVICE : DMA_FROM_DEVICE);
}
EXPORT_SYMBOL(rknand_dma_unmap_single);

int rknand_flash_cs_init(int id)
{
	return 0;
}
EXPORT_SYMBOL(rknand_flash_cs_init);

int rknand_get_reg_addr(unsigned long *p_nandc0, unsigned long *p_nandc1)
{
	*p_nandc0 = (unsigned long)g_nandc_info[0].reg_base;
	*p_nandc1 = (unsigned long)g_nandc_info[1].reg_base;
	return 0;
}
EXPORT_SYMBOL(rknand_get_reg_addr);

int rknand_get_boot_media(void)
{
	return rknand_boot_media;
}
EXPORT_SYMBOL(rknand_get_boot_media);

unsigned long rk_copy_from_user(void *to, const void __user *from,
				unsigned long n)
{
	return copy_from_user(to, from, n);
}

unsigned long rk_copy_to_user(void __user *to, const void *from,
			      unsigned long n)
{
	return copy_to_user(to, from, n);
}

static const struct file_operations rknand_sys_storage_fops = {
	.compat_ioctl = rknand_sys_storage_ioctl,
	.unlocked_ioctl = rknand_sys_storage_ioctl,
};

static struct miscdevice rknand_sys_storage_dev = {
	.minor = MISC_DYNAMIC_MINOR,
	.name  = "rknand_sys_storage",
	.fops  = &rknand_sys_storage_fops,
};

int rknand_sys_storage_init(void)
{
	return misc_register(&rknand_sys_storage_dev);
}

static const struct file_operations rknand_vendor_storage_fops = {
	.compat_ioctl	= rk_ftl_vendor_storage_ioctl,
	.unlocked_ioctl = rk_ftl_vendor_storage_ioctl,
};

static struct miscdevice rknand_vender_storage_dev = {
	.minor = MISC_DYNAMIC_MINOR,
	.name  = "vendor_storage",
	.fops  = &rknand_vendor_storage_fops,
};

int rknand_vendor_storage_init(void)
{
	return misc_register(&rknand_vender_storage_dev);
}

int rk_nand_schedule_enable_config(int en)
{
	int tmp = rk_nand_wait_busy_schedule;

	rk_nand_wait_busy_schedule = en;
	return tmp;
}

static void rk_nand_iqr_timeout_hack(struct timer_list *unused)
{
	del_timer(&rk_nand_iqr_timeout);
	rk_timer_add = 0;
	nandc0_xfer_completed_flag = 1;
	nandc0_ready_completed_flag = 1;
	nandc1_xfer_completed_flag = 1;
	nandc1_ready_completed_flag = 1;
	wake_up(&rk29_nandc_wait);
}

static void rk_add_timer(void)
{
	if (rk_timer_add == 0) {
		rk_timer_add = 1;
		rk_nand_iqr_timeout.expires = jiffies + HZ / 50;
		add_timer(&rk_nand_iqr_timeout);
	}
}

static void rk_del_timer(void)
{
	if (rk_timer_add)
		del_timer(&rk_nand_iqr_timeout);
	rk_timer_add = 0;
}

static irqreturn_t rk_nandc_interrupt(int irq, void *dev_id)
{
	unsigned int irq_status = rk_nandc_get_irq_status(dev_id);

	if (irq_status & (1 << 0)) {
		rk_nandc_flash_xfer_completed(dev_id);
		if (dev_id == g_nandc_info[0].reg_base)
			nandc0_xfer_completed_flag = 1;
		else
			nandc1_xfer_completed_flag = 1;
	}

	if (irq_status & (1 << 1)) {
		rk_nandc_flash_ready(dev_id);
		if (dev_id == g_nandc_info[0].reg_base)
			nandc0_ready_completed_flag = 1;
		else
			nandc1_ready_completed_flag = 1;
	}

	wake_up(&rk29_nandc_wait);
	return IRQ_HANDLED;
}

void rk_nandc_xfer_irq_flag_init(void *nandc_reg)
{
	if (nandc_reg == g_nandc_info[0].reg_base)
		nandc0_xfer_completed_flag = 0;
	else
		nandc1_xfer_completed_flag = 0;
}

void rk_nandc_rb_irq_flag_init(void *nandc_reg)
{
	if (nandc_reg == g_nandc_info[0].reg_base)
		nandc0_ready_completed_flag = 0;
	else
		nandc1_ready_completed_flag = 0;
}

void wait_for_nandc_xfer_completed(void *nandc_reg)
{
	if (rk_nand_wait_busy_schedule)	{
		rk_add_timer();
		if (nandc_reg == g_nandc_info[0].reg_base)
			wait_event(rk29_nandc_wait, nandc0_xfer_completed_flag);
		else
			wait_event(rk29_nandc_wait, nandc1_xfer_completed_flag);
		rk_del_timer();
	}
	if (nandc_reg == g_nandc_info[0].reg_base)
		nandc0_xfer_completed_flag = 0;
	else
		nandc1_xfer_completed_flag = 0;
}

void wait_for_nand_flash_ready(void *nandc_reg)
{
	if (rk_nand_wait_busy_schedule)	{
		rk_add_timer();
		if (nandc_reg == g_nandc_info[0].reg_base)
			wait_event(rk29_nandc_wait
				, nandc0_ready_completed_flag);
		else
			wait_event(rk29_nandc_wait
				, nandc1_ready_completed_flag);
		rk_del_timer();
	}
	if (nandc_reg == g_nandc_info[0].reg_base)
		nandc0_ready_completed_flag = 0;
	else
		nandc1_ready_completed_flag = 0;
}

static int rk_nandc_irq_config(int id, int mode, void *pfun)
{
	int ret = 0;
	int irq = g_nandc_info[id].irq;

	if (mode)
		ret = request_irq(irq, pfun, 0, "nandc"
			, g_nandc_info[id].reg_base);
	else
		free_irq(irq,  NULL);
	return ret;
}

int rk_nandc_irq_init(void)
{
	int ret = 0;

	rk_timer_add = 0;
	nandc0_ready_completed_flag = 0;
	nandc0_xfer_completed_flag = 0;
	ret = rk_nandc_irq_config(0, 1, rk_nandc_interrupt);

	if (!g_nandc_info[1].reg_base) {
		nandc1_ready_completed_flag = 0;
		nandc1_xfer_completed_flag = 0;
		rk_nandc_irq_config(1, 1, rk_nandc_interrupt);
	}
	return ret;
}

int rk_nandc_irq_deinit(void)
{
	rk_nandc_irq_config(0, 0, rk_nandc_interrupt);
	if (!g_nandc_info[1].reg_base)
		rk_nandc_irq_config(1, 0, rk_nandc_interrupt);
	return 0;
}

static int rknand_probe(struct platform_device *pdev)
{
	unsigned int id = 0;
	int irq;
	struct resource	*mem;
	void __iomem	*membase;

	g_nand_device = &pdev->dev;
	mem = platform_get_resource(pdev, IORESOURCE_MEM, 0);
	membase = devm_ioremap_resource(&pdev->dev, mem);
	if (!membase) {
		dev_err(&pdev->dev, "no reg resource?\n");
		return -1;
	}

	#ifdef CONFIG_OF
	of_property_read_u32(pdev->dev.of_node, "nandc_id", &id);
	pdev->id = id;
	#endif

	if (id == 0) {
		memcpy(nand_idb_data, membase + 0x1000, 0x800);
		if (*(int *)(&nand_idb_data[0]) == 0x44535953) {
			rknand_boot_media = *(int *)(&nand_idb_data[8]);
			if (rknand_boot_media == 2) /*boot from emmc*/
				return -1;
		}
	}

	irq = platform_get_irq(pdev, 0);
	if (irq < 0) {
		dev_err(&pdev->dev, "no irq resource?\n");
		return irq;
	}

	g_nandc_info[id].id = id;
	g_nandc_info[id].irq = irq;
	g_nandc_info[id].reg_base = membase;

	g_nandc_info[id].hclk = devm_clk_get(&pdev->dev, "hclk_nandc");
	g_nandc_info[id].clk = devm_clk_get(&pdev->dev, "clk_nandc");
	g_nandc_info[id].gclk = devm_clk_get(&pdev->dev, "g_clk_nandc");

	if (unlikely(IS_ERR(g_nandc_info[id].hclk))) {
		dev_err(&pdev->dev, "rknand_probe get hclk error\n");
		return PTR_ERR(g_nandc_info[id].hclk);
	}

	if (!(IS_ERR(g_nandc_info[id].clk))) {
		clk_set_rate(g_nandc_info[id].clk, 150 * 1000 * 1000);
		g_nandc_info[id].clk_rate = clk_get_rate(g_nandc_info[id].clk);
		clk_prepare_enable(g_nandc_info[id].clk);
		dev_info(&pdev->dev,
			 "rknand_probe clk rate = %d\n",
			 g_nandc_info[id].clk_rate);
	}

	clk_prepare_enable(g_nandc_info[id].hclk);
	if (!(IS_ERR(g_nandc_info[id].gclk)))
		clk_prepare_enable(g_nandc_info[id].gclk);

	pm_runtime_enable(&pdev->dev);
	pm_runtime_get_sync(&pdev->dev);

	return dma_set_mask(g_nand_device, DMA_BIT_MASK(32));
}

static int rknand_suspend(struct platform_device *pdev, pm_message_t state)
{
	if (rk_nand_suspend_state == 0) {
		rk_nand_suspend_state = 1;
		rknand_dev_suspend();
	}
	return 0;
}

static int rknand_resume(struct platform_device *pdev)
{
	if (rk_nand_suspend_state == 1) {
		rk_nand_suspend_state = 0;
		rknand_dev_resume();
	}
	return 0;
}

static void rknand_shutdown(struct platform_device *pdev)
{
	if (rk_nand_shutdown_state == 0) {
		rk_nand_shutdown_state = 1;
		rknand_dev_shutdown();
	}
}

void rknand_dev_cache_flush(void)
{
	rknand_dev_flush();
}

static int rknand_pm_suspend(struct device *dev)
{
	if (rk_nand_suspend_state == 0) {
		rk_nand_suspend_state = 1;
		rknand_dev_suspend();
		pm_runtime_put(dev);
	}
	return 0;
}

static int rknand_pm_resume(struct device *dev)
{
	if (rk_nand_suspend_state == 1) {
		rk_nand_suspend_state = 0;
		pm_runtime_get_sync(dev);
		rknand_dev_resume();
	}
	return 0;
}

static const struct dev_pm_ops rknand_dev_pm_ops = {
	SET_SYSTEM_SLEEP_PM_OPS(rknand_pm_suspend, rknand_pm_resume)
};

#ifdef CONFIG_OF
static const struct of_device_id of_rk_nandc_match[] = {
	{.compatible = "rockchip,rk-nandc"},
	{.compatible = "rockchip,rk-nandc-v9"},
	{}
};
#endif

static struct platform_driver rknand_driver = {
	.probe		= rknand_probe,
	.suspend	= rknand_suspend,
	.resume		= rknand_resume,
	.shutdown	= rknand_shutdown,
	.driver		= {
		.name	= "rknand",
#ifdef CONFIG_OF
		.of_match_table	= of_rk_nandc_match,
#endif
		.pm = &rknand_dev_pm_ops,
	},
};

static void __exit rknand_driver_exit(void)
{
	rknand_dev_exit();
	platform_driver_unregister(&rknand_driver);
}

static int __init rknand_driver_init(void)
{
	int ret = 0;

	pr_err("%s\n", RKNAND_VERSION_AND_DATE);
	ret = platform_driver_register(&rknand_driver);
	if (ret == 0)
		ret = rknand_dev_init();
	return ret;
}

module_init(rknand_driver_init);
module_exit(rknand_driver_exit);
MODULE_ALIAS("rknand");
MODULE_LICENSE("GPL v2");