/*
 * drivers/staging/android/fiq_debugger.c
 *
 * Serial Debugger Interface accessed through an FIQ interrupt.
 *
 * Copyright (C) 2008 Google, Inc.
 *
 * This software is licensed under the terms of the GNU General Public
 * License version 2, as published by the Free Software Foundation, and
 * may be copied, distributed, and modified under those terms.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 */

#include <linux/stdarg.h>
#include <linux/module.h>
#include <linux/io.h>
#include <linux/console.h>
#include <linux/interrupt.h>
#include <linux/clk.h>
#include <linux/platform_device.h>
#include <linux/kernel_stat.h>
#include <linux/kmsg_dump.h>
#include <linux/irq.h>
#include <linux/delay.h>
#include <linux/reboot.h>
#include <linux/sched/signal.h>
#include <linux/slab.h>
#include <linux/smp.h>
#include <linux/timer.h>
#include <linux/tty.h>
#include <linux/tty_flip.h>
#include <linux/ptrace.h>
#include <linux/proc_fs.h>

#ifdef CONFIG_FIQ_DEBUGGER_TRUST_ZONE
#include <linux/rockchip/rockchip_sip.h>
#include "rk_fiq_debugger.h"
#endif

#ifdef CONFIG_FIQ_GLUE
#include <asm/fiq_glue.h>
#endif

#ifdef CONFIG_FIQ_DEBUGGER_UART_OVERLAY
#include <linux/of.h>
#endif

#include <linux/uaccess.h>
#include <linux/cpuhotplug.h>

#include "fiq_debugger.h"
#include "fiq_debugger_priv.h"
#include "fiq_debugger_ringbuf.h"

#ifdef CONFIG_ROCKCHIP_DEBUG
#include "../rockchip_debug.h"
#endif

#define DEBUG_MAX 64
#define CMD_COUNT 0x0f
#define MAX_UNHANDLED_FIQ_COUNT 1000000

#ifdef CONFIG_ARCH_ROCKCHIP
#define MAX_FIQ_DEBUGGER_PORTS 1
#else
#define MAX_FIQ_DEBUGGER_PORTS 4
#endif

struct fiq_debugger_state {
#ifdef CONFIG_FIQ_GLUE
	struct fiq_glue_handler handler;
#endif
	struct fiq_debugger_output output;

	int fiq;
	int uart_irq;
	int signal_irq;
	int wakeup_irq;
	bool wakeup_irq_no_set_wake;
	struct clk *clk;
	struct fiq_debugger_pdata *pdata;
	struct platform_device *pdev;

	char debug_cmd[DEBUG_MAX];
	int debug_busy;
	int debug_abort;

	char debug_buf[DEBUG_MAX];
	int debug_count;

#ifdef CONFIG_ARCH_ROCKCHIP
	char cmd_buf[CMD_COUNT + 1][DEBUG_MAX];
	int back_pointer;
	int current_pointer;
#endif

	bool no_sleep;
	bool debug_enable;
	bool ignore_next_wakeup_irq;
	struct timer_list sleep_timer;
	spinlock_t sleep_timer_lock;
	bool uart_enabled;
	struct wakeup_source debugger_wake_src;
	bool console_enable;
	int current_cpu;
	atomic_t unhandled_fiq_count;
	bool in_fiq;

	struct work_struct work;
	spinlock_t work_lock;
	char work_cmd[DEBUG_MAX];

#ifdef CONFIG_FIQ_DEBUGGER_CONSOLE
	spinlock_t console_lock;
	struct console console;
	struct tty_port tty_port;
	struct fiq_debugger_ringbuf *tty_rbuf;
	bool syslog_dumping;
#endif

#ifdef CONFIG_ARCH_ROCKCHIP
	unsigned int last_irqs[1024];
	unsigned int last_local_irqs[NR_CPUS][32];
#else
	unsigned int last_irqs[NR_IRQS];
	unsigned int last_local_timer_irqs[NR_CPUS];
#endif
};

#ifdef CONFIG_FIQ_DEBUGGER_CONSOLE
static struct tty_driver *fiq_tty_driver;
#endif

#ifdef CONFIG_FIQ_DEBUGGER_NO_SLEEP
static bool initial_no_sleep = true;
#else
static bool initial_no_sleep;
#endif

#ifdef CONFIG_FIQ_DEBUGGER_CONSOLE_DEFAULT_ENABLE
static bool initial_debug_enable = true;
static bool initial_console_enable = true;
#else
static bool initial_debug_enable;
static bool initial_console_enable;
#endif

static struct fiq_debugger_state *g_state;
static bool fiq_kgdb_enable;
static bool fiq_debugger_disable;

module_param_named(no_sleep, initial_no_sleep, bool, 0644);
module_param_named(debug_enable, initial_debug_enable, bool, 0644);
module_param_named(console_enable, initial_console_enable, bool, 0644);
module_param_named(kgdb_enable, fiq_kgdb_enable, bool, 0644);
module_param_named(disable, fiq_debugger_disable, bool, 0644);

#ifdef CONFIG_ARM64
#include "fiq_debugger_arm64.c"
#else
#include "fiq_debugger_arm.c"
#endif

#ifdef CONFIG_FIQ_DEBUGGER_WAKEUP_IRQ_ALWAYS_ON
static inline
void fiq_debugger_enable_wakeup_irq(struct fiq_debugger_state *state) {}
static inline
void fiq_debugger_disable_wakeup_irq(struct fiq_debugger_state *state) {}
#else
static inline
void fiq_debugger_enable_wakeup_irq(struct fiq_debugger_state *state)
{
	if (state->wakeup_irq < 0)
		return;
	enable_irq(state->wakeup_irq);
	if (!state->wakeup_irq_no_set_wake)
		enable_irq_wake(state->wakeup_irq);
}
static inline
void fiq_debugger_disable_wakeup_irq(struct fiq_debugger_state *state)
{
	if (state->wakeup_irq < 0)
		return;
	disable_irq_nosync(state->wakeup_irq);
	if (!state->wakeup_irq_no_set_wake)
		disable_irq_wake(state->wakeup_irq);
}
#endif

static inline bool fiq_debugger_have_fiq(struct fiq_debugger_state *state)
{
	return (state->fiq >= 0);
}

static void fiq_debugger_force_irq(struct fiq_debugger_state *state)
{
	unsigned int irq = state->signal_irq;

	if (irq < 0)
		return;

	if (state->pdata->force_irq) {
		state->pdata->force_irq(state->pdev, irq);
	} else {
		struct irq_chip *chip = irq_get_chip(irq);

		if (chip && chip->irq_retrigger)
			chip->irq_retrigger(irq_get_irq_data(irq));
	}
}

static void fiq_debugger_uart_enable(struct fiq_debugger_state *state)
{
	if (state->clk)
		clk_enable(state->clk);
	if (state->pdata->uart_enable)
		state->pdata->uart_enable(state->pdev);
}

static void fiq_debugger_uart_disable(struct fiq_debugger_state *state)
{
	if (state->pdata->uart_disable)
		state->pdata->uart_disable(state->pdev);
	if (state->clk)
		clk_disable(state->clk);
}

static void fiq_debugger_uart_flush(struct fiq_debugger_state *state)
{
	if (state->pdata->uart_flush)
		state->pdata->uart_flush(state->pdev);
}

static void fiq_debugger_putc(struct fiq_debugger_state *state, char c)
{
	if (state->pdata->uart_putc)
		state->pdata->uart_putc(state->pdev, c);
}

static void fiq_debugger_puts(struct fiq_debugger_state *state, char *s)
{
	unsigned c;
	while ((c = *s++)) {
		if (c == '\n')
			fiq_debugger_putc(state, '\r');
		fiq_debugger_putc(state, c);
	}
}

static void fiq_debugger_prompt(struct fiq_debugger_state *state)
{
	fiq_debugger_puts(state, "debug> ");
}

#ifndef CONFIG_FIQ_DEBUGGER_MODULE
static void fiq_debugger_dump_kernel_log(struct fiq_debugger_state *state)
{
	char buf[512];
	size_t len;
	struct kmsg_dump_iter iter;

	kmsg_dump_rewind(&iter);
	while (kmsg_dump_get_line(&iter, true, buf,
					 sizeof(buf) - 1, &len)) {
		buf[len] = 0;
		fiq_debugger_puts(state, buf);
	}
}
#endif

__printf(2, 3)
static void fiq_debugger_printf(struct fiq_debugger_output *output,
			       const char *fmt, ...)
{
	struct fiq_debugger_state *state;
	char buf[256];
	va_list ap;

	state = container_of(output, struct fiq_debugger_state, output);
	va_start(ap, fmt);
	vsnprintf(buf, sizeof(buf), fmt, ap);
	va_end(ap);

	fiq_debugger_puts(state, buf);
}

/* Safe outside fiq context */
__printf(2, 3)
static int fiq_debugger_printf_nfiq(void *cookie, const char *fmt, ...)
{
	struct fiq_debugger_state *state = cookie;
	char buf[256];
	va_list ap;
	unsigned long irq_flags;

	va_start(ap, fmt);
	vsnprintf(buf, 128, fmt, ap);
	va_end(ap);

	local_irq_save(irq_flags);
	fiq_debugger_puts(state, buf);
	fiq_debugger_uart_flush(state);
	local_irq_restore(irq_flags);
	return state->debug_abort;
}

#ifndef CONFIG_FIQ_DEBUGGER_MODULE
static void fiq_debugger_dump_irqs(struct fiq_debugger_state *state)
{
	int n;
	struct irq_desc *desc;

	fiq_debugger_printf(&state->output,
			"irqnr       total  since-last   status  name\n");
	for_each_irq_desc(n, desc) {
		struct irqaction *act = desc->action;
		if (!act && !kstat_irqs_usr(n))
			continue;
		fiq_debugger_printf(&state->output, "%5d: %10u %11u %8x  %s\n", n,
			kstat_irqs_usr(n),
			kstat_irqs_usr(n) - state->last_irqs[n],
			desc->status_use_accessors,
			(act && act->name) ? act->name : "???");
		state->last_irqs[n] = kstat_irqs_usr(n);
	}
}
#endif

#ifndef CONFIG_FIQ_DEBUGGER_MODULE
static void fiq_debugger_do_ps(struct fiq_debugger_state *state)
{
	struct task_struct *g;
	struct task_struct *p;
	unsigned task_state;
	static const char stat_nam[] = "RSDTtZX";

	fiq_debugger_printf(&state->output, "pid   ppid  prio task            pc\n");
	read_lock(&tasklist_lock);
	do_each_thread(g, p) {
		task_state = p->__state ? __ffs(p->__state) + 1 : 0;
		fiq_debugger_printf(&state->output,
			     "%5d %5d %4d ", p->pid, p->parent->pid, p->prio);
		fiq_debugger_printf(&state->output, "%-13.13s %c", p->comm,
			     task_state >= sizeof(stat_nam) ? '?' : stat_nam[task_state]);
		if (task_state == TASK_RUNNING)
			fiq_debugger_printf(&state->output, " running\n");
		else
			fiq_debugger_printf(&state->output, " %08lx\n",
					thread_saved_pc(p));
	} while_each_thread(g, p);
	read_unlock(&tasklist_lock);
}
#endif

#ifndef CONFIG_FIQ_DEBUGGER_MODULE
#ifdef CONFIG_FIQ_DEBUGGER_CONSOLE
static void fiq_debugger_begin_syslog_dump(struct fiq_debugger_state *state)
{
	state->syslog_dumping = true;
}

static void fiq_debugger_end_syslog_dump(struct fiq_debugger_state *state)
{
	state->syslog_dumping = false;
}
#else
extern int do_syslog(int type, char __user *bug, int count);
static void fiq_debugger_begin_syslog_dump(struct fiq_debugger_state *state)
{
	do_syslog(5 /* clear */, NULL, 0);
}

static void fiq_debugger_end_syslog_dump(struct fiq_debugger_state *state)
{
	fiq_debugger_dump_kernel_log(state);
}
#endif
#endif

#ifndef CONFIG_FIQ_DEBUGGER_MODULE
static void fiq_debugger_do_sysrq(struct fiq_debugger_state *state, char rq)
{
	if ((rq == 'g' || rq == 'G') && !fiq_kgdb_enable) {
		fiq_debugger_printf(&state->output, "sysrq-g blocked\n");
		return;
	}
	fiq_debugger_begin_syslog_dump(state);
	__handle_sysrq(rq, false);
	fiq_debugger_end_syslog_dump(state);
}
#endif

#ifdef CONFIG_KGDB
static void fiq_debugger_do_kgdb(struct fiq_debugger_state *state)
{
	if (!fiq_kgdb_enable) {
		fiq_debugger_printf(&state->output, "kgdb through fiq debugger not enabled\n");
		return;
	}

	fiq_debugger_printf(&state->output, "enabling console and triggering kgdb\n");
	state->console_enable = true;
	handle_sysrq('g');
}
#endif

static void fiq_debugger_schedule_work(struct fiq_debugger_state *state,
		char *cmd)
{
	unsigned long flags;

	spin_lock_irqsave(&state->work_lock, flags);
	if (state->work_cmd[0] != '\0') {
		fiq_debugger_printf(&state->output, "work command processor busy\n");
		spin_unlock_irqrestore(&state->work_lock, flags);
		return;
	}

	strlcpy(state->work_cmd, cmd, sizeof(state->work_cmd));
	spin_unlock_irqrestore(&state->work_lock, flags);

	schedule_work(&state->work);
}

static void fiq_debugger_work(struct work_struct *work)
{
	struct fiq_debugger_state *state;
	char work_cmd[DEBUG_MAX];
	char *cmd;
	unsigned long flags;

	state = container_of(work, struct fiq_debugger_state, work);

	spin_lock_irqsave(&state->work_lock, flags);

	strlcpy(work_cmd, state->work_cmd, sizeof(work_cmd));
	state->work_cmd[0] = '\0';

	spin_unlock_irqrestore(&state->work_lock, flags);

	cmd = work_cmd;
	if (!strncmp(cmd, "reboot", 6)) {
		cmd += 6;
		while (*cmd == ' ')
			cmd++;
		if (*cmd != '\0')
			kernel_restart(cmd);
		else
			kernel_restart(NULL);
	} else {
		fiq_debugger_printf(&state->output, "unknown work command '%s'\n",
				work_cmd);
	}
}

/* This function CANNOT be called in FIQ context */
static void fiq_debugger_irq_exec(struct fiq_debugger_state *state, char *cmd)
{
	int invalid_cmd = 0;

#ifndef CONFIG_FIQ_DEBUGGER_MODULE
	if (!strcmp(cmd, "ps"))
		fiq_debugger_do_ps(state);
	if (!strcmp(cmd, "sysrq"))
		fiq_debugger_do_sysrq(state, 'h');
	if (!strncmp(cmd, "sysrq ", 6))
		fiq_debugger_do_sysrq(state, cmd[6]);
#endif
#ifdef CONFIG_KGDB
	if (!strcmp(cmd, "kgdb"))
		fiq_debugger_do_kgdb(state);
#endif
	if (!strncmp(cmd, "reboot", 6))
		fiq_debugger_schedule_work(state, cmd);
#ifdef CONFIG_ARCH_ROCKCHIP
	else {
		invalid_cmd = 1;
		memset(state->debug_buf, 0, DEBUG_MAX);
	}

	if (invalid_cmd == 0) {
		state->current_pointer =
				(state->current_pointer - 1) & CMD_COUNT;
		if (strcmp(state->cmd_buf[state->current_pointer], state->debug_buf)) {
			state->current_pointer =
				(state->current_pointer + 1) & CMD_COUNT;
			memset(state->cmd_buf[state->current_pointer], 0, DEBUG_MAX);
			strcpy(state->cmd_buf[state->current_pointer], state->debug_buf);
		}
		memset(state->debug_buf, 0, DEBUG_MAX);
		state->current_pointer = (state->current_pointer + 1) & CMD_COUNT;
		state->back_pointer = state->current_pointer;
	}
#endif
}

#ifdef CONFIG_ARCH_ROCKCHIP
static char cmd_buf[][16] = {
		{"pc"},
		{"regs"},
		{"allregs"},
		{"bt"},
		{"reboot"},
#ifdef CONFIG_ROCKCHIP_DEBUG
		{"pcsr"},
#endif
		{"sleep"},
		{"nosleep"},
		{"console"},
		{"cpu"},
#ifndef CONFIG_FIQ_DEBUGGER_MODULE
		{"reset"},
		{"irqs"},
		{"kmsg"},
		{"version"},
		{"ps"},
		{"sysrq"},
#endif

#ifdef CONFIG_KGDB
		{"kgdb"},
#endif
};
#endif

static void fiq_debugger_help(struct fiq_debugger_state *state)
{
	fiq_debugger_printf(&state->output,
				"FIQ Debugger commands:\n"
				" pc            PC status\n"
				" regs          Register dump\n"
				" allregs       Extended Register dump\n"
				" bt            Stack trace\n"
#ifdef CONFIG_ROCKCHIP_DEBUG
				" pcsr          Dump all cpus pc by DBGPCSR\n"
#endif
				" cpu           Current CPU\n"
				" cpu <number>  Switch to CPU<number>\n"
				" reboot [<c>]  Reboot with command <c>\n");

	fiq_debugger_printf(&state->output,
				" sleep         Allow sleep while in FIQ\n"
				" nosleep       Disable sleep while in FIQ\n"
				" console       Switch terminal to console\n");

#ifndef CONFIG_FIQ_DEBUGGER_MODULE
	fiq_debugger_printf(&state->output,
				" reset [<c>]   Hard reset with command <c>\n"
				" irqs          Interrupt status\n"
				" kmsg          Kernel log\n"
				" version       Kernel version\n");

	fiq_debugger_printf(&state->output,
				" ps            Process list\n"
				" sysrq         sysrq options\n"
				" sysrq <param> Execute sysrq with <param>\n");
#endif

#ifdef CONFIG_KGDB
	fiq_debugger_printf(&state->output,
				" kgdb          Enter kernel debugger\n");
#endif
}

#ifndef CONFIG_FIQ_DEBUGGER_MODULE
static void fiq_debugger_take_affinity(struct fiq_debugger_state *state, int cpu)
{
	struct cpumask cpumask;

	cpumask_clear(&cpumask);
	cpumask_set_cpu(cpu, &cpumask);
	irq_set_affinity(state->uart_irq, &cpumask);
}
#else
static void fiq_debugger_take_affinity(struct fiq_debugger_state *state, int cpu)
{
}
#endif

static void fiq_debugger_switch_cpu(struct fiq_debugger_state *state, int cpu)
{
	if (!cpu_online(cpu)) {
		fiq_debugger_printf(&state->output, "cpu %d offline\n", cpu);
		return;
	}

	if (!fiq_debugger_have_fiq(state)) {
		fiq_debugger_take_affinity(state, cpu);
	}
#ifdef CONFIG_ARCH_ROCKCHIP
	else {
#ifdef CONFIG_FIQ_DEBUGGER_TRUST_ZONE
		if (sip_fiq_debugger_is_enabled() ||
		    sdei_fiq_debugger_is_enabled()) {
			if (state->pdata->switch_cpu) {
				state->pdata->switch_cpu(state->pdev, cpu);
				state->current_cpu = cpu;
			}
			return;
		}
#else
		struct cpumask cpumask;

		cpumask_clear(&cpumask);
		cpumask_set_cpu(cpu, &cpumask);
#ifndef CONFIG_FIQ_DEBUGGER_MODULE
		irq_set_affinity(state->fiq, &cpumask);
		irq_set_affinity(state->uart_irq, &cpumask);
#endif
#endif
	}
#endif
	state->current_cpu = cpu;
}

static bool fiq_debugger_fiq_exec(struct fiq_debugger_state *state,
			const char *cmd, const struct pt_regs *regs,
			void *svc_sp)
{
	bool signal_helper = false;

	if (!strcmp(cmd, "help") || !strcmp(cmd, "?")) {
		fiq_debugger_help(state);
	} else if (!strcmp(cmd, "pc")) {
		fiq_debugger_dump_pc(&state->output, regs);
	} else if (!strcmp(cmd, "regs")) {
		fiq_debugger_dump_regs(&state->output, regs);
	} else if (!strcmp(cmd, "allregs")) {
		fiq_debugger_dump_allregs(&state->output, regs);
#ifndef CONFIG_FIQ_DEBUGGER_MODULE
	} else if (!strcmp(cmd, "bt")) {
		/*
		 * ARM64:
		 * Cpu is at ELx(1 or 2), but EL0_SP(svc_sp) may be user space.
		 * If EL0_SP.63 is 0, use TTBR0.
		 */
		if (user_mode((struct pt_regs *)regs) ||
		    (IS_ENABLED(CONFIG_ARM64) && (((unsigned long)svc_sp & 0x8000000000000000) == 0)) ||
		    (IS_ENABLED(CONFIG_ARM) && (((unsigned long)svc_sp < PAGE_OFFSET) || ((unsigned long)svc_sp > -256UL))))
			fiq_debugger_printf(&state->output, "User mode\n");
		else
			fiq_debugger_dump_stacktrace(&state->output, regs,
						     100, svc_sp);
	} else if (!strncmp(cmd, "reset", 5)) {
		cmd += 5;
		while (*cmd == ' ')
			cmd++;
		if (*cmd) {
			char tmp_cmd[32];
			strlcpy(tmp_cmd, cmd, sizeof(tmp_cmd));
			machine_restart(tmp_cmd);
		} else {
			machine_restart(NULL);
		}
	} else if (!strcmp(cmd, "irqs")) {
		fiq_debugger_dump_irqs(state);
	} else if (!strcmp(cmd, "kmsg")) {
		fiq_debugger_dump_kernel_log(state);
#endif
#ifdef CONFIG_ROCKCHIP_DEBUG
	} else if (!strcmp(cmd, "pcsr")) {
		rockchip_debug_dump_pcsr(&state->output);
#endif
#ifndef CONFIG_FIQ_DEBUGGER_MODULE
	} else if (!strcmp(cmd, "version")) {
		fiq_debugger_printf(&state->output, "%s\n", linux_banner);
#endif
	} else if (!strcmp(cmd, "sleep")) {
		state->no_sleep = false;
		fiq_debugger_printf(&state->output, "enabling sleep\n");
	} else if (!strcmp(cmd, "nosleep")) {
		state->no_sleep = true;
		fiq_debugger_printf(&state->output, "disabling sleep\n");
	} else if (!strcmp(cmd, "console")) {
		fiq_debugger_printf(&state->output, "console mode\n");
		fiq_debugger_uart_flush(state);
		state->console_enable = true;
#ifdef CONFIG_FIQ_DEBUGGER_TRUST_ZONE
		if (sip_fiq_debugger_is_enabled()) {
			if (state->pdata->enable_debug)
				state->pdata->enable_debug(state->pdev, false);
		}
#endif
	} else if (!strcmp(cmd, "cpu")) {
		fiq_debugger_printf(&state->output, "cpu %d\n", state->current_cpu);
	} else if (!strncmp(cmd, "cpu ", 4)) {
		unsigned long cpu = 0;
		if (kstrtoul(cmd + 4, 10, &cpu) == 0)
			fiq_debugger_switch_cpu(state, cpu);
		else
			fiq_debugger_printf(&state->output, "invalid cpu\n");

		fiq_debugger_printf(&state->output, "cpu %d\n", state->current_cpu);
	} else {
		if (state->debug_busy) {
			fiq_debugger_printf(&state->output,
				"command processor busy. trying to abort.\n");
			state->debug_abort = -1;
		} else {
			strcpy(state->debug_cmd, cmd);
			state->debug_busy = 1;
		}

		return true;
	}
	if (!state->console_enable)
		fiq_debugger_prompt(state);

	return signal_helper;
}

static void fiq_debugger_sleep_timer_expired(struct timer_list *t)
{
	struct fiq_debugger_state *state = from_timer(state, t, sleep_timer);
	unsigned long flags;

	spin_lock_irqsave(&state->sleep_timer_lock, flags);
	if (state->uart_enabled && !state->no_sleep) {
		if (state->debug_enable && !state->console_enable) {
			state->debug_enable = false;
			fiq_debugger_printf_nfiq(state,
					"suspending fiq debugger\n");
		}
		state->ignore_next_wakeup_irq = true;
		fiq_debugger_uart_disable(state);
		state->uart_enabled = false;
		fiq_debugger_enable_wakeup_irq(state);
	}
	__pm_relax(&state->debugger_wake_src);
	spin_unlock_irqrestore(&state->sleep_timer_lock, flags);
}

static void fiq_debugger_handle_wakeup(struct fiq_debugger_state *state)
{
	unsigned long flags;

	spin_lock_irqsave(&state->sleep_timer_lock, flags);
	if (state->wakeup_irq >= 0 && state->ignore_next_wakeup_irq) {
		state->ignore_next_wakeup_irq = false;
	} else if (!state->uart_enabled) {
		__pm_stay_awake(&state->debugger_wake_src);
		fiq_debugger_uart_enable(state);
		state->uart_enabled = true;
		fiq_debugger_disable_wakeup_irq(state);
		mod_timer(&state->sleep_timer, jiffies + HZ / 2);
	}
	spin_unlock_irqrestore(&state->sleep_timer_lock, flags);
}

static irqreturn_t fiq_debugger_wakeup_irq_handler(int irq, void *dev)
{
	struct fiq_debugger_state *state = dev;

	if (!state->no_sleep)
		fiq_debugger_puts(state, "WAKEUP\n");
	fiq_debugger_handle_wakeup(state);

	return IRQ_HANDLED;
}

static
void fiq_debugger_handle_console_irq_context(struct fiq_debugger_state *state)
{
#if defined(CONFIG_FIQ_DEBUGGER_CONSOLE)
	if (state->tty_port.ops) {
		int i;
		int count = fiq_debugger_ringbuf_level(state->tty_rbuf);
		for (i = 0; i < count; i++) {
			int c = fiq_debugger_ringbuf_peek(state->tty_rbuf, 0);
			tty_insert_flip_char(&state->tty_port, c, TTY_NORMAL);
			if (!fiq_debugger_ringbuf_consume(state->tty_rbuf, 1))
				pr_warn("fiq tty failed to consume byte\n");
		}
		tty_flip_buffer_push(&state->tty_port);
	}
#endif
}

static void fiq_debugger_handle_irq_context(struct fiq_debugger_state *state)
{
	if (!state->no_sleep) {
		unsigned long flags;

		spin_lock_irqsave(&state->sleep_timer_lock, flags);
		__pm_stay_awake(&state->debugger_wake_src);
		mod_timer(&state->sleep_timer, jiffies + HZ * 5);
		spin_unlock_irqrestore(&state->sleep_timer_lock, flags);
	}
	fiq_debugger_handle_console_irq_context(state);
	if (state->debug_busy) {
		fiq_debugger_irq_exec(state, state->debug_cmd);
		if (!state->console_enable)
			fiq_debugger_prompt(state);
		state->debug_busy = 0;
	}
}

static int fiq_debugger_getc(struct fiq_debugger_state *state)
{
	if (state->pdata->uart_getc)
		return state->pdata->uart_getc(state->pdev);
	else
		return FIQ_DEBUGGER_NO_CHAR;
}

static int fiq_debugger_cmd_check_back(struct fiq_debugger_state *state, char c)
{
	char *s;
	int i = 0;

	if (c == 'A') {
		state->back_pointer = (state->back_pointer - 1) & CMD_COUNT;
		if (state->back_pointer != state->current_pointer) {
			s = state->cmd_buf[state->back_pointer];
			if (*s != 0) {
				for (i = 0; i < strlen(state->debug_buf) - 1; i++) {
					fiq_debugger_putc(state, 8);
					fiq_debugger_putc(state, ' ');
					fiq_debugger_putc(state, 8);
				}
				memset(state->debug_buf, 0, DEBUG_MAX);
				strcpy(state->debug_buf, s);
				state->debug_count = strlen(state->debug_buf);
				fiq_debugger_printf(&state->output, state->debug_buf);
			} else {
				state->back_pointer = (state->back_pointer + 1) & CMD_COUNT;
			}

		} else {
			state->back_pointer = (state->back_pointer + 1) & CMD_COUNT;
		}
	} else if (c == 'B') {
		if (state->back_pointer != state->current_pointer) {
			state->back_pointer = (state->back_pointer + 1) & CMD_COUNT;
			if (state->back_pointer == state->current_pointer) {
				goto cmd_clear;
			} else {
				s = state->cmd_buf[state->back_pointer];
				if (*s != 0) {
					for (i = 0; i < strlen(state->debug_buf) - 1; i++) {
						fiq_debugger_putc(state, 8);
						fiq_debugger_putc(state, ' ');
						fiq_debugger_putc(state, 8);
					}
					memset(state->debug_buf, 0, DEBUG_MAX);
					strcpy(state->debug_buf, s);
					state->debug_count = strlen(state->debug_buf);
					fiq_debugger_printf(&state->output, state->debug_buf);
				}
			}
		} else {
cmd_clear:
			for (i = 0; i < strlen(state->debug_buf) - 1; i++) {
				fiq_debugger_putc(state, 8);
				fiq_debugger_putc(state, ' ');
				fiq_debugger_putc(state, 8);
			}
			memset(state->debug_buf, 0, DEBUG_MAX);
			state->debug_count = 0;
		}
	}
	return 0;
}

static void fiq_debugger_cmd_tab(struct fiq_debugger_state *state)
{
	int i, j;
	int count = 0;

	for (i = 0; i < ARRAY_SIZE(cmd_buf); i++)
		cmd_buf[i][15] = 1;

	for (j = 1; j <= strlen(state->debug_buf); j++) {
		count = 0;
		for (i = 0; i < ARRAY_SIZE(cmd_buf); i++) {
			if (cmd_buf[i][15] == 1) {
				if (strncmp(state->debug_buf, cmd_buf[i], j))
					cmd_buf[i][15] = 0;
				else
					count++;
			}
		}
		if (count == 0)
			break;
	}

	if (count == 1) {
		for (i = 0; i < ARRAY_SIZE(cmd_buf); i++) {
			if (cmd_buf[i][15] == 1)
				break;
		}

		for (j = 0; j < strlen(state->debug_buf); j++) {
			fiq_debugger_putc(state, 8);
			fiq_debugger_putc(state, ' ');
			fiq_debugger_putc(state, 8);
		}
		memset(state->debug_buf, 0, DEBUG_MAX);
		strcpy(state->debug_buf, cmd_buf[i]);
		state->debug_count = strlen(state->debug_buf);
		fiq_debugger_printf(&state->output, state->debug_buf);
	}
}

static bool fiq_debugger_handle_uart_interrupt(struct fiq_debugger_state *state,
			int this_cpu, const struct pt_regs *regs, void *svc_sp)
{
	int c;
	static int last_c;
	int count = 0;
	bool signal_helper = false;

	if (state->current_cpu == -1)
		state->current_cpu = this_cpu;

	if (this_cpu != state->current_cpu) {
		if (state->in_fiq)
			return false;

		if (atomic_inc_return(&state->unhandled_fiq_count) !=
					MAX_UNHANDLED_FIQ_COUNT)
			return false;

		fiq_debugger_printf(&state->output,
			"fiq_debugger: cpu %d not responding, "
			"reverting to cpu %d\n", state->current_cpu,
			this_cpu);

		atomic_set(&state->unhandled_fiq_count, 0);
		state->current_cpu = this_cpu;
		return false;
	}

	state->in_fiq = true;

	while ((c = fiq_debugger_getc(state)) != FIQ_DEBUGGER_NO_CHAR) {
		count++;
		if (!state->debug_enable) {
			if ((c == 13) || (c == 10)) {
				state->debug_enable = true;
				state->debug_count = 0;
				fiq_debugger_prompt(state);
			}
		} else if (c == FIQ_DEBUGGER_BREAK) {
			state->console_enable = false;
#ifdef CONFIG_ARCH_ROCKCHIP
			fiq_debugger_puts(state, "\nWelcome to ");
#endif
			if (fiq_debugger_have_fiq(state))
				fiq_debugger_puts(state,
						  "fiq debugger mode\n");
			else
				fiq_debugger_puts(state,
						  "irq debugger mode\n");
			state->debug_count = 0;
#ifdef CONFIG_ARCH_ROCKCHIP
			fiq_debugger_puts(state, "Enter ? to get command help\n");
			state->back_pointer = CMD_COUNT;
			state->current_pointer = CMD_COUNT;
			memset(state->cmd_buf, 0, (CMD_COUNT + 1) * DEBUG_MAX);
#endif

#ifdef CONFIG_FIQ_DEBUGGER_TRUST_ZONE
			if (sip_fiq_debugger_is_enabled()) {
				if (state->pdata->enable_debug)
					state->pdata->enable_debug(state->pdev,
								   true);
			}
#endif
			fiq_debugger_prompt(state);
#ifdef CONFIG_FIQ_DEBUGGER_CONSOLE
			fiq_debugger_ringbuf_push(state->tty_rbuf, 8);
			fiq_debugger_ringbuf_push(state->tty_rbuf, 8);
#endif
#ifdef CONFIG_FIQ_DEBUGGER_CONSOLE
		} else if (state->console_enable && state->tty_rbuf) {
			fiq_debugger_ringbuf_push(state->tty_rbuf, c);
			signal_helper = true;
#endif
#ifdef CONFIG_ARCH_ROCKCHIP
		} else if (last_c == '[' && (c == 'A' || c == 'B' || c == 'C' || c == 'D')) {
			if (state->debug_count > 0) {
				state->debug_count--;
				fiq_debugger_putc(state, 8);
				fiq_debugger_putc(state, ' ');
				fiq_debugger_putc(state, 8);
			}
			fiq_debugger_cmd_check_back(state, c);
		} else if (c == 9) {
			fiq_debugger_cmd_tab(state);
#endif
		} else if ((c >= ' ') && (c < 127)) {
			if (state->debug_count < (DEBUG_MAX - 1)) {
				state->debug_buf[state->debug_count++] = c;
				fiq_debugger_putc(state, c);
			}
		} else if ((c == 8) || (c == 127)) {
			if (state->debug_count > 0) {
				state->debug_count--;
				fiq_debugger_putc(state, 8);
				fiq_debugger_putc(state, ' ');
				fiq_debugger_putc(state, 8);
			}
		} else if ((c == 13) || (c == 10)) {
			if (c == '\r' || (c == '\n' && last_c != '\r')) {
				fiq_debugger_putc(state, '\r');
				fiq_debugger_putc(state, '\n');
			}
			if (state->debug_count) {
				state->debug_buf[state->debug_count] = 0;
				state->debug_count = 0;
				signal_helper |=
					fiq_debugger_fiq_exec(state,
							state->debug_buf,
							regs, svc_sp);
#ifdef CONFIG_ARCH_ROCKCHIP
				if (signal_helper == false) {
					state->current_pointer =
							(state->current_pointer - 1) & CMD_COUNT;
					if (strcmp(state->cmd_buf[state->current_pointer], state->debug_buf)) {
						state->current_pointer =
							(state->current_pointer + 1) & CMD_COUNT;
						memset(state->cmd_buf[state->current_pointer], 0, DEBUG_MAX);
						strcpy(state->cmd_buf[state->current_pointer], state->debug_buf);
					}
					memset(state->debug_buf, 0, DEBUG_MAX);
					state->current_pointer =
						(state->current_pointer + 1) & CMD_COUNT;
					state->back_pointer =
						state->current_pointer;
				}
#endif
			} else {
				fiq_debugger_prompt(state);
			}
		}
		last_c = c;
	}
	if (!state->console_enable)
		fiq_debugger_uart_flush(state);
	if (state->pdata->fiq_ack)
		state->pdata->fiq_ack(state->pdev, state->fiq);

	/* poke sleep timer if necessary */
	if (state->debug_enable && !state->no_sleep)
		signal_helper = true;

	atomic_set(&state->unhandled_fiq_count, 0);
	state->in_fiq = false;

	return signal_helper;
}

#ifdef CONFIG_FIQ_GLUE
static void fiq_debugger_fiq(struct fiq_glue_handler *h,
		void *regs, void *svc_sp)
{
	struct fiq_debugger_state *state =
		container_of(h, struct fiq_debugger_state, handler);
	unsigned int this_cpu = THREAD_INFO(svc_sp)->cpu;
	bool need_irq;

	need_irq = fiq_debugger_handle_uart_interrupt(state, this_cpu, regs,
			svc_sp);
	if (need_irq)
		fiq_debugger_force_irq(state);
}
#endif

#ifdef CONFIG_FIQ_DEBUGGER_TRUST_ZONE
void fiq_debugger_fiq(void *regs, u32 cpu)
{
	struct fiq_debugger_state *state = g_state;
	bool need_irq;

	if (!state)
		return;

	need_irq = fiq_debugger_handle_uart_interrupt(state, cpu, regs,
						      current_thread_info());
	if (need_irq)
		fiq_debugger_force_irq(state);
}
EXPORT_SYMBOL(fiq_debugger_fiq);
#endif

/*
 * When not using FIQs, we only use this single interrupt as an entry point.
 * This just effectively takes over the UART interrupt and does all the work
 * in this context.
 */
static irqreturn_t fiq_debugger_uart_irq(int irq, void *dev)
{
	struct fiq_debugger_state *state = dev;
	bool not_done;

	fiq_debugger_handle_wakeup(state);

	/* handle the debugger irq in regular context */
	not_done = fiq_debugger_handle_uart_interrupt(state, smp_processor_id(),
#ifdef CONFIG_NO_GKI
					      get_irq_regs(),
#else
					      NULL,
#endif
					      current_thread_info());
	if (not_done)
		fiq_debugger_force_irq(state);

	return IRQ_HANDLED;
}

/*
 * If FIQs are used, not everything can happen in fiq context.
 * FIQ handler does what it can and then signals this interrupt to finish the
 * job in irq context.
 */
static irqreturn_t fiq_debugger_signal_irq(int irq, void *dev)
{
	struct fiq_debugger_state *state = dev;

	if (state->pdata->force_irq_ack)
		state->pdata->force_irq_ack(state->pdev, state->signal_irq);

	fiq_debugger_handle_irq_context(state);

	return IRQ_HANDLED;
}

#ifdef CONFIG_FIQ_GLUE
static void fiq_debugger_resume(struct fiq_glue_handler *h)
{
	struct fiq_debugger_state *state =
		container_of(h, struct fiq_debugger_state, handler);
	if (state->pdata->uart_resume)
		state->pdata->uart_resume(state->pdev);
}
#endif

#if defined(CONFIG_FIQ_DEBUGGER_CONSOLE)
static struct tty_driver *fiq_debugger_console_device(struct console *co, int *index)
{
	*index = co->index;
	return fiq_tty_driver;
}

static void fiq_debugger_console_write(struct console *co,
				const char *s, unsigned int count)
{
	struct fiq_debugger_state *state;
	unsigned long flags;

	state = container_of(co, struct fiq_debugger_state, console);

	if (!state->console_enable && !state->syslog_dumping)
		return;

#ifdef CONFIG_RK_CONSOLE_THREAD
	if (state->pdata->console_write) {
		state->pdata->console_write(state->pdev, s, count);
		return;
	}
#endif

	fiq_debugger_uart_enable(state);
	spin_lock_irqsave(&state->console_lock, flags);
	while (count--) {
		if (*s == '\n')
			fiq_debugger_putc(state, '\r');
		fiq_debugger_putc(state, *s++);
	}
	fiq_debugger_uart_flush(state);
	spin_unlock_irqrestore(&state->console_lock, flags);
	fiq_debugger_uart_disable(state);
}

static struct console fiq_debugger_console = {
	.name = "ttyFIQ",
	.device = fiq_debugger_console_device,
	.write = fiq_debugger_console_write,
	.flags = CON_PRINTBUFFER | CON_ANYTIME | CON_ENABLED,
};

static int fiq_tty_open(struct tty_struct *tty, struct file *filp)
{
	int line = tty->index;
	struct fiq_debugger_state **states = tty->driver->driver_state;
	struct fiq_debugger_state *state = states[line];

	return tty_port_open(&state->tty_port, tty, filp);
}

static void fiq_tty_close(struct tty_struct *tty, struct file *filp)
{
	tty_port_close(tty->port, tty, filp);
}

void fiq_tty_wake_up(struct platform_device *pdev)
{
	struct fiq_debugger_state *state = platform_get_drvdata(pdev);

	if (tty_port_initialized(&state->tty_port))
		tty_port_tty_wakeup(&state->tty_port);
}
EXPORT_SYMBOL_GPL(fiq_tty_wake_up);

static int fiq_tty_write(struct tty_struct *tty, const unsigned char *buf, int count)
{
	int i;
	int line = tty->index;
	struct fiq_debugger_state **states = tty->driver->driver_state;
	struct fiq_debugger_state *state = states[line];

	if (!state->console_enable)
		return count;

#ifdef CONFIG_RK_CONSOLE_THREAD
	if (state->pdata->tty_write)
		return state->pdata->tty_write(state->pdev, buf, count);
#endif

	fiq_debugger_uart_enable(state);
#ifndef CONFIG_RK_CONSOLE_THREAD
	spin_lock_irq(&state->console_lock);
#endif
	for (i = 0; i < count; i++)
		fiq_debugger_putc(state, *buf++);
#ifndef CONFIG_RK_CONSOLE_THREAD
	spin_unlock_irq(&state->console_lock);
#endif
	fiq_debugger_uart_disable(state);

	return count;
}

static unsigned int fiq_tty_write_room(struct tty_struct *tty)
{
#ifdef CONFIG_RK_CONSOLE_THREAD
	int line = tty->index;
	struct fiq_debugger_state **states = tty->driver->driver_state;
	struct fiq_debugger_state *state = states[line];

	if (state->pdata->write_room)
		return state->pdata->write_room(state->pdev);
#endif
	return 2048;
}

#ifdef CONFIG_CONSOLE_POLL
static int fiq_tty_poll_init(struct tty_driver *driver, int line, char *options)
{
	return 0;
}

static int fiq_tty_poll_get_char(struct tty_driver *driver, int line)
{
	struct fiq_debugger_state **states = driver->driver_state;
	struct fiq_debugger_state *state = states[line];
	int c = NO_POLL_CHAR;

	fiq_debugger_uart_enable(state);
	if (fiq_debugger_have_fiq(state)) {
#ifdef CONFIG_FIQ_DEBUGGER_CONSOLE
		int count = fiq_debugger_ringbuf_level(state->tty_rbuf);
		if (count > 0) {
			c = fiq_debugger_ringbuf_peek(state->tty_rbuf, 0);
			fiq_debugger_ringbuf_consume(state->tty_rbuf, 1);
		}
#endif
	} else {
		c = fiq_debugger_getc(state);
		if (c == FIQ_DEBUGGER_NO_CHAR)
			c = NO_POLL_CHAR;
	}
	fiq_debugger_uart_disable(state);

	return c;
}

static void fiq_tty_poll_put_char(struct tty_driver *driver, int line, char ch)
{
	struct fiq_debugger_state **states = driver->driver_state;
	struct fiq_debugger_state *state = states[line];
	fiq_debugger_uart_enable(state);
	fiq_debugger_putc(state, ch);
	fiq_debugger_uart_disable(state);
}
#endif

#ifdef CONFIG_PROC_FS
static int fiq_tty_proc_show(struct seq_file *m, void *v)
{
	struct tty_driver *driver = m->private;
	struct fiq_debugger_state **states = driver->driver_state;
	struct fiq_debugger_state *state;
	int i;

	seq_puts(m, "fiq-debugger driver\n");
	for (i = 0; i < MAX_FIQ_DEBUGGER_PORTS; i++) {
		state = states[i];
		if (!state)
			continue;

		seq_printf(m, "%d:", i);
		seq_printf(m, " state:%d", state->console_enable);
		seq_putc(m, '\n');
	}
	return 0;
}
#endif

static const struct tty_port_operations fiq_tty_port_ops;

static const struct tty_operations fiq_tty_driver_ops = {
	.write = fiq_tty_write,
	.write_room = fiq_tty_write_room,
	.open = fiq_tty_open,
	.close = fiq_tty_close,
#ifdef CONFIG_CONSOLE_POLL
	.poll_init = fiq_tty_poll_init,
	.poll_get_char = fiq_tty_poll_get_char,
	.poll_put_char = fiq_tty_poll_put_char,
#endif
#ifdef CONFIG_PROC_FS
	.proc_show = fiq_tty_proc_show,
#endif
};

static int fiq_debugger_tty_init(void)
{
	int ret;
	struct fiq_debugger_state **states = NULL;
	struct tty_driver *drv;

	states = kzalloc(sizeof(*states) * MAX_FIQ_DEBUGGER_PORTS, GFP_KERNEL);
	if (!states) {
		pr_err("Failed to allocate fiq debugger state structres\n");
		return -ENOMEM;
	}

	drv = tty_alloc_driver(MAX_FIQ_DEBUGGER_PORTS, TTY_DRIVER_REAL_RAW | TTY_DRIVER_DYNAMIC_DEV);
	if (IS_ERR(drv)) {
		pr_err("Failed to allocate fiq debugger tty\n");
		ret = -ENOMEM;
		goto err_free_state;
	}
	fiq_tty_driver = drv;

	fiq_tty_driver->owner		= THIS_MODULE;
	fiq_tty_driver->driver_name	= "fiq-debugger";
	fiq_tty_driver->name		= "ttyFIQ";
	fiq_tty_driver->type		= TTY_DRIVER_TYPE_SERIAL;
	fiq_tty_driver->subtype		= SERIAL_TYPE_NORMAL;
	fiq_tty_driver->init_termios	= tty_std_termios;
	fiq_tty_driver->driver_state	= states;

	fiq_tty_driver->init_termios.c_cflag =
					B115200 | CS8 | CREAD | HUPCL | CLOCAL;
	fiq_tty_driver->init_termios.c_ispeed = 115200;
	fiq_tty_driver->init_termios.c_ospeed = 115200;

	tty_set_operations(fiq_tty_driver, &fiq_tty_driver_ops);

	ret = tty_register_driver(fiq_tty_driver);
	if (ret) {
		pr_err("Failed to register fiq tty: %d\n", ret);
		goto err_free_tty;
	}

	pr_info("Registered FIQ tty driver\n");
	return 0;

err_free_tty:
	tty_driver_kref_put(fiq_tty_driver);
	fiq_tty_driver = NULL;
err_free_state:
	kfree(states);
	return ret;
}

static int fiq_debugger_tty_init_one(struct fiq_debugger_state *state)
{
	int ret;
	struct device *tty_dev;
	struct fiq_debugger_state **states = fiq_tty_driver->driver_state;

	states[state->pdev->id] = state;

	state->tty_rbuf = fiq_debugger_ringbuf_alloc(1024);
	if (!state->tty_rbuf) {
		pr_err("Failed to allocate fiq debugger ringbuf\n");
		ret = -ENOMEM;
		goto err;
	}

	tty_port_init(&state->tty_port);
	state->tty_port.ops = &fiq_tty_port_ops;

	tty_dev = tty_port_register_device(&state->tty_port, fiq_tty_driver,
					   state->pdev->id, &state->pdev->dev);
	if (IS_ERR(tty_dev)) {
		pr_err("Failed to register fiq debugger tty device\n");
		ret = PTR_ERR(tty_dev);
		goto err;
	}

	device_set_wakeup_capable(tty_dev, 1);

	pr_info("Registered fiq debugger ttyFIQ%d\n", state->pdev->id);

	return 0;

err:
	fiq_debugger_ringbuf_free(state->tty_rbuf);
	state->tty_rbuf = NULL;
	return ret;
}
#endif

static int fiq_debugger_dev_suspend(struct device *dev)
{
	struct platform_device *pdev = to_platform_device(dev);
	struct fiq_debugger_state *state = platform_get_drvdata(pdev);

	if (state->pdata->uart_dev_suspend)
		return state->pdata->uart_dev_suspend(pdev);
	return 0;
}

static int fiq_debugger_dev_resume(struct device *dev)
{
	struct platform_device *pdev = to_platform_device(dev);
	struct fiq_debugger_state *state = platform_get_drvdata(pdev);

	if (state->pdata->uart_dev_resume)
		return state->pdata->uart_dev_resume(pdev);
	return 0;
}

static int fiq_debugger_cpu_offine_migrate_irq(unsigned int cpu)
{
	if (g_state && cpu == g_state->current_cpu) {
		unsigned int new_cpu = cpumask_any_but(cpu_online_mask, cpu);

		if (new_cpu < nr_cpu_ids)
			g_state->current_cpu = new_cpu;
	}

	return 0;
}

static int fiq_debugger_probe(struct platform_device *pdev)
{
	int ret;
	struct fiq_debugger_pdata *pdata = dev_get_platdata(&pdev->dev);
	struct fiq_debugger_state *state;
	int fiq;
	int uart_irq;
	enum cpuhp_state cs = -1;

	if (pdev->id >= MAX_FIQ_DEBUGGER_PORTS)
		return -EINVAL;

	if (!pdata->uart_getc || !pdata->uart_putc)
		return -EINVAL;
	if ((pdata->uart_enable && !pdata->uart_disable) ||
	    (!pdata->uart_enable && pdata->uart_disable))
		return -EINVAL;

	fiq = platform_get_irq_byname(pdev, "fiq");
	uart_irq = platform_get_irq_byname(pdev, "uart_irq");

#ifndef CONFIG_ARCH_ROCKCHIP
	/* uart_irq mode and fiq mode are mutually exclusive, but one of them
	 * is required */
	if ((uart_irq < 0 && fiq < 0) || (uart_irq >= 0 && fiq >= 0))
		return -EINVAL;
	if (fiq >= 0 && !pdata->fiq_enable)
		return -EINVAL;
#endif
	state = kzalloc(sizeof(*state), GFP_KERNEL);
	state->output.printf = fiq_debugger_printf;
	timer_setup(&state->sleep_timer, fiq_debugger_sleep_timer_expired, 0);
	state->pdata = pdata;
	state->pdev = pdev;
	state->no_sleep = initial_no_sleep;
	state->debug_enable = initial_debug_enable;
	state->console_enable = initial_console_enable;
	state->current_cpu = -1;

	state->fiq = fiq;
	state->uart_irq = uart_irq;
	state->signal_irq = platform_get_irq_byname(pdev, "signal");
	state->wakeup_irq = platform_get_irq_byname(pdev, "wakeup");

	INIT_WORK(&state->work, fiq_debugger_work);
	spin_lock_init(&state->work_lock);

	platform_set_drvdata(pdev, state);

	spin_lock_init(&state->sleep_timer_lock);

	if (state->wakeup_irq < 0 && fiq_debugger_have_fiq(state))
		state->no_sleep = true;
	state->ignore_next_wakeup_irq = !state->no_sleep;

	state->debugger_wake_src.name = "serial-debug";
	wakeup_source_add(&state->debugger_wake_src);

#ifdef CONFIG_ARCH_ROCKCHIP
	if (uart_irq < 0 && fiq < 0)
		goto console_out;
#endif

	state->clk = clk_get(&pdev->dev, NULL);
	if (IS_ERR(state->clk))
		state->clk = NULL;

	/* do not call pdata->uart_enable here since uart_init may still
	 * need to do some initialization before uart_enable can work.
	 * So, only try to manage the clock during init.
	 */
	if (state->clk)
		clk_enable(state->clk);

	if (fiq_debugger_have_fiq(state)) {
#ifdef CONFIG_FIQ_GLUE
#ifdef CONFIG_FIQ_DEBUGGER_TRUST_ZONE
		if (sip_fiq_debugger_is_enabled() ||
		    sdei_fiq_debugger_is_enabled()) {
		} else
#endif
		{
		state->handler.fiq = fiq_debugger_fiq;
		state->handler.resume = fiq_debugger_resume;
		ret = fiq_glue_register_handler(&state->handler);
		if (ret) {
			pr_err("%s: could not install fiq handler\n", __func__);
			goto err_register_irq;
		}
#ifdef CONFIG_ARCH_ROCKCHIP
		/* set state->fiq to secure state, so fiq is available */
		gic_set_irq_secure(irq_get_irq_data(state->fiq));
		/*
		* set state->fiq priority a little higher than other
		* interrupts (normal is 0xa0)
		*/
		gic_set_irq_priority(irq_get_irq_data(state->fiq), 0x90);
#endif
		pdata->fiq_enable(pdev, state->fiq, 1);
		}
#endif
	} else {
		irq_set_status_flags(state->uart_irq, IRQ_NOAUTOEN);

		if (IS_ENABLED(CONFIG_NO_GKI))
			ret = request_nmi(state->uart_irq, fiq_debugger_uart_irq,
					  IRQF_PERCPU, "debug", state);
		else
			ret = -EINVAL;
		if (ret) {
			pr_err("%s: could not install nmi irq handler\n", __func__);
			irq_clear_status_flags(state->uart_irq, IRQ_NOAUTOEN);
			ret = request_irq(state->uart_irq, fiq_debugger_uart_irq,
					  IRQF_NO_SUSPEND | IRQF_NOBALANCING, "debug", state);
		} else {
			enable_nmi(state->uart_irq);
		}

		if (ret) {
			pr_err("%s: could not install irq handler\n", __func__);
			goto err_register_irq;
		}

		/* for irq-only mode, we want this irq to wake us up, if it
		 * can.
		 */
		enable_irq_wake(state->uart_irq);

		ret = cpuhp_setup_state_nocalls(CPUHP_AP_ONLINE_DYN,
						"soc/fiq_debugger",
						NULL,
						fiq_debugger_cpu_offine_migrate_irq);
		if (ret < 0)
			pr_err("%s: could not setup cpu offine handler\n", __func__);
		else
			cs = ret;
	}

	if (state->signal_irq >= 0) {
		ret = request_irq(state->signal_irq, fiq_debugger_signal_irq,
			  IRQF_TRIGGER_RISING, "debug-signal", state);
		if (ret)
			pr_err("serial_debugger: could not install signal_irq");
	}

	if (state->wakeup_irq >= 0) {
		ret = request_irq(state->wakeup_irq,
				  fiq_debugger_wakeup_irq_handler,
				  IRQF_TRIGGER_FALLING,
				  "debug-wakeup", state);
		if (ret) {
			pr_err("serial_debugger: "
				"could not install wakeup irq\n");
			state->wakeup_irq = -1;
		} else {
			ret = enable_irq_wake(state->wakeup_irq);
			if (ret) {
				pr_err("serial_debugger: "
					"could not enable wakeup\n");
				state->wakeup_irq_no_set_wake = true;
			}
		}
	}
	if (state->no_sleep)
		fiq_debugger_handle_wakeup(state);

	if (pdata->uart_init) {
		ret = pdata->uart_init(pdev);
		if (ret)
			goto err_uart_init;
	}

	if (state->clk)
		clk_disable(state->clk);
#ifdef CONFIG_ARCH_ROCKCHIP
console_out:
#endif
#if defined(CONFIG_FIQ_DEBUGGER_CONSOLE)
	spin_lock_init(&state->console_lock);
	state->console = fiq_debugger_console;
	state->console.index = pdev->id;
#ifndef CONFIG_FIQ_DEBUGGER_MODULE
	if (!console_set_on_cmdline)
		add_preferred_console(state->console.name,
			state->console.index, NULL);
#endif
	register_console(&state->console);
	fiq_debugger_tty_init_one(state);
#endif

	/* switch to cpu0 default */
	fiq_debugger_switch_cpu(state, 0);
	g_state = state;
	return 0;

err_register_irq:
	if (pdata->uart_free)
		pdata->uart_free(pdev);
err_uart_init:
	if (state->clk)
		clk_disable(state->clk);
	if (state->clk)
		clk_put(state->clk);
	if (cs >= 0)
		cpuhp_remove_state_nocalls(cs);
	wakeup_source_remove(&state->debugger_wake_src);
	__pm_relax(&state->debugger_wake_src);
	platform_set_drvdata(pdev, NULL);
	kfree(state);
	return ret;
}

static const struct dev_pm_ops fiq_debugger_dev_pm_ops = {
	.suspend	= fiq_debugger_dev_suspend,
	.resume		= fiq_debugger_dev_resume,
};

static struct platform_driver fiq_debugger_driver = {
	.probe	= fiq_debugger_probe,
	.driver	= {
		.name	= "fiq_debugger",
		.pm	= &fiq_debugger_dev_pm_ops,
	},
};

#if defined(CONFIG_FIQ_DEBUGGER_UART_OVERLAY)
int fiq_debugger_uart_overlay(void)
{
	struct device_node *onp = of_find_node_by_path("/uart_overlay@0");
	int ret;

	if (!onp) {
		pr_err("serial_debugger: uart overlay not found\n");
		return -ENODEV;
	}

	ret = of_overlay_create(onp);
	if (ret < 0) {
		pr_err("serial_debugger: fail to create overlay: %d\n", ret);
		of_node_put(onp);
		return ret;
	}

	pr_info("serial_debugger: uart overlay applied\n");
	return 0;
}
#endif

static int __init fiq_debugger_init(void)
{
	if (fiq_debugger_disable) {
		pr_err("serial_debugger: disabled\n");
		return -ENODEV;
	}
#if defined(CONFIG_FIQ_DEBUGGER_CONSOLE)
	fiq_debugger_tty_init();
#endif
#if defined(CONFIG_FIQ_DEBUGGER_UART_OVERLAY)
	fiq_debugger_uart_overlay();
#endif
	return platform_driver_register(&fiq_debugger_driver);
}

postcore_initcall(fiq_debugger_init);

static void __exit fiq_debugger_exit(void)
{
	platform_driver_unregister(&fiq_debugger_driver);
}
module_exit(fiq_debugger_exit);

MODULE_DESCRIPTION("FIQ Debugger");
MODULE_LICENSE("GPL");
MODULE_ALIAS("platform:fiq-debugger");