/* Copyright Statement:
*
* This software/firmware and related documentation ("MediaTek Software") are
* protected under relevant copyright laws. The information contained herein
* is confidential and proprietary to MediaTek Inc. and/or its licensors.
* Without the prior written permission of MediaTek inc. and/or its licensors,
* any reproduction, modification, use or disclosure of MediaTek Software,
* and information contained herein, in whole or in part, shall be strictly prohibited.
*/
/* MediaTek Inc. (C) 2016. All rights reserved.
*
* BY OPENING THIS FILE, RECEIVER HEREBY UNEQUIVOCALLY ACKNOWLEDGES AND AGREES
* THAT THE SOFTWARE/FIRMWARE AND ITS DOCUMENTATIONS ("MEDIATEK SOFTWARE")
* RECEIVED FROM MEDIATEK AND/OR ITS REPRESENTATIVES ARE PROVIDED TO RECEIVER ON
* AN "AS-IS" BASIS ONLY. MEDIATEK EXPRESSLY DISCLAIMS ANY AND ALL WARRANTIES,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE IMPLIED WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE OR NONINFRINGEMENT.
* NEITHER DOES MEDIATEK PROVIDE ANY WARRANTY WHATSOEVER WITH RESPECT TO THE
* SOFTWARE OF ANY THIRD PARTY WHICH MAY BE USED BY, INCORPORATED IN, OR
* SUPPLIED WITH THE MEDIATEK SOFTWARE, AND RECEIVER AGREES TO LOOK ONLY TO SUCH
* THIRD PARTY FOR ANY WARRANTY CLAIM RELATING THERETO. RECEIVER EXPRESSLY ACKNOWLEDGES
* THAT IT IS RECEIVER'S SOLE RESPONSIBILITY TO OBTAIN FROM ANY THIRD PARTY ALL PROPER LICENSES
* CONTAINED IN MEDIATEK SOFTWARE. MEDIATEK SHALL ALSO NOT BE RESPONSIBLE FOR ANY MEDIATEK
* SOFTWARE RELEASES MADE TO RECEIVER'S SPECIFICATION OR TO CONFORM TO A PARTICULAR
* STANDARD OR OPEN FORUM. RECEIVER'S SOLE AND EXCLUSIVE REMEDY AND MEDIATEK'S ENTIRE AND
* CUMULATIVE LIABILITY WITH RESPECT TO THE MEDIATEK SOFTWARE RELEASED HEREUNDER WILL BE,
* AT MEDIATEK'S OPTION, TO REVISE OR REPLACE THE MEDIATEK SOFTWARE AT ISSUE,
* OR REFUND ANY SOFTWARE LICENSE FEES OR SERVICE CHARGE PAID BY RECEIVER TO
* MEDIATEK FOR SUCH MEDIATEK SOFTWARE AT ISSUE.
*/

#include <stdio.h>
#include <stdlib.h>		// for ALIGN()
#include <string.h>
#include <arch/arm/mmu.h>
#include <platform/boot_mode.h>
#include <platform/ram_console.h>
#include <libfdt.h>

#define RAM_CONSOLE_SIG (0x43474244) /* DBGC */
#define MOD "RAM_CONSOLE"

struct ram_console_buffer {
	u32 sig;
	/* for size comptible */
	u32 off_pl;
	u32 off_lpl; /* last preloader */
	u32 sz_pl;
	u32 off_lk;
	u32 off_llk; /* last lk */
	u32 sz_lk;
	u32 padding[2]; /* size = 2 * 16 = 32 byte */
	u32 dump_step;
	u32 sz_buffer;
	u32 off_linux;
	u32 off_console;
	u32 padding2[3];
};

#define RAM_CONSOLE_LK_SIZE 16
struct reboot_reason_lk {
	u32 last_func[RAM_CONSOLE_LK_SIZE];
};

struct reboot_reason_pl {
	u32 wdt_status;
	u32 last_func[0];
};

struct reboot_reason_kernel {
	u32 fiq_step;
	/* 0xaeedeadX: X=1 (HWT), X=2 (KE), X=3 (nested panic) */
	/* details see enum AEE_EXP_TYPE_NUM in ram_console_common.h */
	u32 exp_type;
	u32 others[0];
};

enum CUR_BOOT_TYPE {
	BOOT_NORMAL = 0,
	BOOT_ABNORMAL,
	BOOT_INVALID
};

static int cur_boot = BOOT_INVALID;

extern int sLOG(char *fmt, ...);
#define LOG(fmt, ...)           \
    do { \
        sLOG(fmt, ##__VA_ARGS__);       \
        printf(fmt, ##__VA_ARGS__); \
    } while(0)

static struct ram_console_buffer *ram_console = NULL;
static uint32_t ram_console_size;
extern BOOT_ARGUMENT *g_boot_arg;

static void ram_console_fatal(const char *str)
{
	LOG("%s. FATAL:%s\n", MOD, str);
	while(1)
		;
}

static void ram_console_memory_info_valid(struct ram_console_memory_info *memory_info)
{
	if (memory_info->magic1 != MEM_MAGIC1 || memory_info->magic2 != MEM_MAGIC2) {
		LOG("%s. passed arg magic:[0x%x, 0x%x] mismatch\n", MOD, memory_info->magic1, memory_info->magic2);
		ram_console_fatal("illegal magic number");
	}
}

static void ram_console_atag_valid(void)
{
	u32 sram_addr;
	u32 sram_size;
	u32 def_type;
	u32 memory_info_offset;

	if (g_boot_arg != NULL) {
		sram_addr = g_boot_arg->ram_console_sram_addr;
		sram_size = g_boot_arg->ram_console_sram_size;
		def_type = g_boot_arg->ram_console_def_type;
		memory_info_offset = g_boot_arg->ram_console_memory_info_offset;
		if (def_type != RAM_CONSOLE_DEF_SRAM && def_type != RAM_CONSOLE_DEF_DRAM)
			ram_console_fatal("unknown def type");
		if (sram_addr == 0 || sram_size == 0)
			ram_console_fatal("sram addr size zero");
		if (memory_info_offset <= sram_size)
			ram_console_fatal("memory offset invalid value");
	} else {
		ram_console_fatal("unexpected g_boot_arg null pointer");
	}
}

static struct ram_console_memory_info *ram_console_atag_memory_info(void)
{
	u32 sram_addr;
	u32 memory_info_offset;
	struct ram_console_memory_info *memory_info = NULL;

	ram_console_atag_valid();

	sram_addr = g_boot_arg->ram_console_sram_addr;
	memory_info_offset = g_boot_arg->ram_console_memory_info_offset;
	memory_info = (struct ram_console_memory_info *)(sram_addr + memory_info_offset);
	ram_console_memory_info_valid(memory_info);
	return memory_info;
}

extern bool cmdline_append(const char *append_string);
static void ram_console_ptr_init_by_atag(void)
{
	struct ram_console_memory_info *memory_info = NULL;
	u32 sram_addr;
	u32 sram_size;
	u32 def_type;
	u32 memory_info_offset;
	char cmd_buf[256];

	if (g_boot_arg == NULL)
		ram_console_fatal("unexpected g_boot_arg null pointer");

	sram_addr = g_boot_arg->ram_console_sram_addr;
	sram_size = g_boot_arg->ram_console_sram_size;
	def_type = g_boot_arg->ram_console_def_type;
	memory_info_offset = g_boot_arg->ram_console_memory_info_offset;
	LOG("%s. boot_arg(PL2LK): sram_addr:0x%x, sram_size:0x%x, def_type:0x%x, memory_info_offset:0x%x\n", MOD, sram_addr, sram_size, def_type, memory_info_offset);

	memory_info = ram_console_atag_memory_info();

	if (def_type == RAM_CONSOLE_DEF_SRAM) {
		ram_console = (struct ram_console_buffer *)sram_addr;
		ram_console_size = sram_size;
	} else if (def_type == RAM_CONSOLE_DEF_DRAM) {
		ram_console = (struct ram_console_buffer *)memory_info->dram_addr;
		ram_console_size = memory_info->dram_size;
#ifdef MTK_3LEVEL_PAGETABLE
	{
		uint32_t start = (uint32_t)ram_console;
		arch_mmu_map(ROUNDDOWN((uint64_t)start, PAGE_SIZE), ROUNDDOWN((uint32_t)start, PAGE_SIZE),
		             MMU_MEMORY_TYPE_NORMAL_WRITE_BACK | MMU_MEMORY_AP_P_RW_U_NA, ROUNDUP(ram_console_size, PAGE_SIZE));
	}
#endif
	} else {
		ram_console_fatal("unknown def type");
	}

	if (ram_console->sig != RAM_CONSOLE_SIG) {
		LOG("%s. %s(0x%x) sig %x mismatch\n", MOD, (def_type == RAM_CONSOLE_DEF_SRAM) ? "sram" : "dram", (unsigned int)ram_console, ram_console->sig);
		ram_console_fatal("sig mismatch");
	}

	memset(cmd_buf, 0x0, sizeof(cmd_buf));
	snprintf(cmd_buf, 256,
		"ramoops.mem_address=0x%x ramoops.mem_size=0x%x ramoops.pmsg_size=0x%x ramoops.console_size=0x%x",
		memory_info->pstore_addr, memory_info->pstore_size,
		memory_info->pstore_pmsg_size, memory_info->pstore_console_size);
	cmdline_append(cmd_buf);
}

void ram_console_init(void)
{
	int i;
	struct reboot_reason_lk *rr_lk;
	ram_console_ptr_init_by_atag();

	if (ram_console) {
		LOG("%s. start: %p, size: 0x%x\n", MOD, ram_console, ram_console_size);
	} else {
		LOG("%s. sig not match\n", MOD);
		return;
	}

	ram_console->off_lk = ram_console->off_lpl + ALIGN(ram_console->sz_pl, 64);
	ram_console->off_llk = ram_console->off_lk + ALIGN(sizeof(struct reboot_reason_lk), 64);
	ram_console->sz_lk = sizeof(struct reboot_reason_lk);
	if (ram_console->sz_lk == sizeof(struct reboot_reason_lk) && (ram_console->off_lk + ram_console->sz_lk == ram_console->off_llk)) {
		LOG("%s. lk last status: ", MOD);
		rr_lk = (void*)ram_console + ram_console->off_lk;
		for (i = 0; i < RAM_CONSOLE_LK_SIZE; i++) {
			LOG("0x%x ", rr_lk->last_func[i]);
		}
		LOG("\n");
		memcpy((void*)ram_console + ram_console->off_llk, (void*)ram_console + ram_console->off_lk, ram_console->sz_lk);
	} else {
		LOG("%s. lk size mismatch %x + %x != %x\n", MOD, ram_console->sz_lk, ram_console->off_lk, ram_console->off_llk);
		ram_console->sz_lk = sizeof(struct reboot_reason_lk);
	}
	ram_console->off_linux = ram_console->off_llk + ALIGN(ram_console->sz_lk, 64);
}


#define RE_BOOT_BY_WDT_SW 2
#define RE_BOOT_NORMAL_BOOT 0
#define RE_BOOT_BY_EINT 256/*we can find the definition from preloader ,this value should  sync with preloader incase issue happened*/
#define RE_BOOT_BY_SYSRST 512/*we can find the definition from preloader ,this value should  sync with preloader incase issue happened*/

#ifdef MTK_PMIC_FULL_RESET
#define RE_BOOT_FULL_PMIC 0x800
#endif

#define U_VAR(type) rc_##type
static unsigned int U_VAR(wdt_status), U_VAR(fiq_step), U_VAR(exp_type);
static bool U_VAR(set_flag) = false;

bool ram_console_reboot_by_mrdump_key(void)
{
	unsigned int wdt_status;
	wdt_status = ((struct reboot_reason_pl*)((void*)ram_console + ram_console->off_pl))->wdt_status;
	return ((wdt_status & RE_BOOT_BY_EINT)|(wdt_status & RE_BOOT_BY_SYSRST))?true:false;
}

#ifdef MTK_PMIC_FULL_RESET
bool ram_console_reboot_by_cold_reset(void)
{
	unsigned int wdt_status;
	wdt_status = ((struct reboot_reason_pl*)((void*)ram_console + ram_console->off_pl))->wdt_status;
	return (wdt_status & RE_BOOT_FULL_PMIC) ? true : false;
}
#endif

bool ram_console_get_wdt_status(unsigned int *wdt_status)
{
	if (wdt_status && U_VAR(set_flag)) {
		*wdt_status = U_VAR(wdt_status);
		return true;
	}
	return false;
}

bool ram_console_get_fiq_step(unsigned int *fiq_step)
{
	if (fiq_step && U_VAR(set_flag)) {
		*fiq_step = U_VAR(fiq_step);
		return true;
	}
	return false;
}

bool ram_console_get_exp_type(unsigned int *exp_type)
{
	if (exp_type && U_VAR(set_flag)) {
		*exp_type = U_VAR(exp_type);
		return true;
	}
	return false;
}

bool ram_console_set_exp_type(unsigned int exp_type)
{
	bool ret = false;
	if (ram_console && ram_console->off_linux &&
		(ram_console->off_linux == (ram_console->off_llk + ALIGN(ram_console->sz_lk, 64))) &&
		(ram_console->off_pl == sizeof(struct ram_console_buffer))) {
		if (exp_type < 16) {
			exp_type = exp_type ^ RAM_CONSOLE_EXP_TYPE_MAGIC;
			((struct reboot_reason_kernel*)((void*)ram_console + ram_console->off_linux))->exp_type = exp_type;
			ret = true;
		} else {
			LOG("%s. set exp type failed: off_linux:0x%x, off_llk:0x%x, off_pl:0x%x, exp type:%d\n", MOD, ram_console->off_linux, ram_console->off_llk, ram_console->off_pl, exp_type);
		}
	}
	return ret;
}

/*
 * struct last_reboot_reason is defined at [kernel/drivers/misc/mediatek/ram_console/mtk_ram_console.c]
 * if you need to access struct members in LK, this struct must be sync-ed with kernel code.
 */
struct last_reboot_reason {
	uint32_t fiq_step;
	/* 0xaeedeadX: X=1 (HWT), X=2 (KE), X=3 (nested panic) */
	uint32_t exp_type;
	uint64_t kaslr_offset;
	uint64_t ram_console_buffer_addr;
};
static struct last_reboot_reason* krbr;

bool ram_console_get_kaslr_offset(uint64_t* kaslr_offset)
{
	if (krbr && U_VAR(set_flag)) {
		if (krbr->kaslr_offset == 0xecab1e) {
			*kaslr_offset = 0;
			return false;
		}

		if (krbr->kaslr_offset == 0xd15ab1e) {
			*kaslr_offset = 0;
		} else {
			*kaslr_offset = krbr->kaslr_offset;
		}
		return true;
	}
	return false;
}

bool ram_console_is_abnormal_boot(void)
{
	unsigned int fiq_step, wdt_status, exp_type;
	int reinit_flag = 0;

	if (!ram_console) {
		ram_console_ptr_init_by_atag();
		reinit_flag = 1;
	}
	if (ram_console && ram_console->off_linux &&
		(ram_console->off_linux == (ram_console->off_llk + ALIGN(ram_console->sz_lk, 64))) &&
		(ram_console->off_pl == sizeof(struct ram_console_buffer))) {
		wdt_status = ((struct reboot_reason_pl*)((void*)ram_console + ram_console->off_pl))->wdt_status;
		krbr = ((struct last_reboot_reason*)((void*)ram_console + ram_console->off_linux));
		fiq_step = ((struct reboot_reason_kernel*)((void*)ram_console + ram_console->off_linux))->fiq_step;
		exp_type = ((struct reboot_reason_kernel*)((void*)ram_console + ram_console->off_linux))->exp_type;
		LOG("%s. wdt_status 0x%x, fiq_step 0x%x, exp_type 0x%x\n", MOD, wdt_status, fiq_step, RAM_CONSOLE_EXP_TYPE_DEC(exp_type));
		if (fiq_step != 0 && (exp_type ^ RAM_CONSOLE_EXP_TYPE_MAGIC) >= 16) {
			fiq_step = 0;
			((struct reboot_reason_kernel*)((void*)ram_console + ram_console->off_linux))->fiq_step = fiq_step;
		}
		U_VAR(wdt_status) = wdt_status;
		U_VAR(fiq_step) = fiq_step;
		U_VAR(exp_type) = RAM_CONSOLE_EXP_TYPE_DEC(exp_type);
		U_VAR(set_flag) = true;
		LOG("%s. set reboot reason info done\n", MOD);
#ifdef MTK_PMIC_FULL_RESET
		if ((fiq_step == 0 && (wdt_status == RE_BOOT_BY_WDT_SW || wdt_status == RE_BOOT_FULL_PMIC)) || (wdt_status == RE_BOOT_NORMAL_BOOT))
#else
		if ((fiq_step == 0 && wdt_status == RE_BOOT_BY_WDT_SW) || (wdt_status == RE_BOOT_NORMAL_BOOT))
#endif
			return false;
		else
			return true;
	} else {
		if (ram_console) {
			LOG("%s. set reboot reason info failed: off_linux:0x%x, off_llk:0x%x, off_pl:0x%x, reinit flag:%d\n", MOD, ram_console->off_linux, ram_console->off_llk, ram_console->off_pl, reinit_flag);
		} else {
			LOG("%s. ram console buffer NULL\n", MOD);
		}
	}
	return false;
}

/*
 * The return value of ram_console_is_abnoraml_boot() maybe changed if
 * ram_console was restored by kedump_restore_mem(). So we recored the value
 * in cur_boot before ram_console restored.
 * The current_is_abnormal_boot() would return the value of cur_boot.
 */
int current_is_abnormal_boot(void)
{
	if (cur_boot == BOOT_NORMAL || cur_boot == BOOT_ABNORMAL)
		return cur_boot;

	cur_boot = ram_console_is_abnormal_boot() ? BOOT_ABNORMAL : BOOT_NORMAL;

	return cur_boot;
}

bool ram_console_should_restore(struct ram_console_buffer *tmp_ram_console)
{
	unsigned int fiq_step, wdt_status, exp_type;

	if (cur_boot != BOOT_NORMAL && cur_boot != BOOT_ABNORMAL)
		cur_boot = ram_console_is_abnormal_boot() ? BOOT_ABNORMAL : BOOT_NORMAL;

	if (U_VAR(set_flag) && tmp_ram_console && tmp_ram_console->off_linux &&
		(tmp_ram_console->off_linux == (tmp_ram_console->off_llk + ALIGN(tmp_ram_console->sz_lk, 64))) &&
		(tmp_ram_console->off_pl == sizeof(struct ram_console_buffer))) {
		wdt_status = ((struct reboot_reason_pl*)((void*)tmp_ram_console + tmp_ram_console->off_pl))->wdt_status;
		fiq_step = ((struct reboot_reason_kernel*)((void*)tmp_ram_console + tmp_ram_console->off_linux))->fiq_step;
		exp_type = RAM_CONSOLE_EXP_TYPE_DEC(((struct reboot_reason_kernel *)((void *)tmp_ram_console + tmp_ram_console->off_linux))->exp_type);

		if ((wdt_status != 0 && wdt_status != U_VAR(wdt_status)) ||
			(fiq_step != 0 && fiq_step != U_VAR(fiq_step)) ||
			(exp_type != 0 && exp_type != U_VAR(exp_type))) {
			return true;
		}
	}
	return false;
}

void ram_console_lk_save(unsigned int val, int index)
{
	struct reboot_reason_lk *rr_lk;
	if (ram_console && ram_console->off_lk < ram_console_size) {
		rr_lk = (void*)ram_console + ram_console->off_lk;
		if (index < RAM_CONSOLE_LK_SIZE)
			rr_lk->last_func[index] = val;
	}
}

/* default can be either sram or dram */
void ram_console_addr_size(unsigned int *addr, unsigned int *size)
{
	if (addr == NULL || size == NULL) {
		LOG("%s. get ram console default info args invalid\n", MOD);
		return;
	}

	*addr = (unsigned int)ram_console;
	*size = ram_console_size;
}

void ram_console_sram_addr_size(unsigned int *addr, unsigned int *size)
{
	if (addr == NULL || size == NULL) {
		LOG("%s. get ram console sram info args invalid\n", MOD);
		return;
	}

	ram_console_atag_valid();
	*addr = g_boot_arg->ram_console_sram_addr;
	*size = g_boot_arg->ram_console_sram_size;
}

u32 ram_console_def_memory(void)
{
	ram_console_atag_valid();
	return g_boot_arg->ram_console_def_type;
}

u32 ram_console_sram_offset(void)
{
	ram_console_atag_valid();
	return g_boot_arg->ram_console_memory_info_offset;
}

void sram_plat_dbg_info_addr_size(unsigned int *addr, unsigned int *size)
{
	struct ram_console_memory_info *memory_info = NULL;

	if (addr == NULL || size == NULL) {
		LOG("%s. get plat dbg info args invalid\n", MOD);
		return;
	}

	memory_info = ram_console_atag_memory_info();
	*addr = memory_info->sram_plat_dbg_info_addr;
	*size = memory_info->sram_plat_dbg_info_size;
}

void sram_log_store_addr_size(unsigned int *addr, unsigned int *size)
{
	struct ram_console_memory_info *memory_info = NULL;

	if (addr == NULL || size == NULL) {
		LOG("%s. get log_store info args invalid\n", MOD);
		return;
	}

	memory_info = ram_console_atag_memory_info();
	*addr = memory_info->sram_log_store_addr;
	*size = memory_info->sram_log_store_size;
}

void sram_mrdump_addr_size(unsigned int *addr, unsigned int *size)
{
	struct ram_console_memory_info *memory_info = NULL;

	if (addr == NULL || size == NULL) {
		LOG("%s. get mrdump info args invalid\n", MOD);
		return;
	}

	memory_info = ram_console_atag_memory_info();
	*addr = memory_info->mrdump_addr;
	*size = memory_info->mrdump_size;
}

void pstore_addr_size(unsigned int *addr, unsigned int *size)
{
	struct ram_console_memory_info *memory_info = NULL;

	if (addr == NULL || size == NULL) {
		LOG("%s. get pstore info args invalid\n", MOD);
		return;
	}

	memory_info = ram_console_atag_memory_info();
	*addr = memory_info->pstore_addr;
	*size = memory_info->pstore_size;
}

void mrdump_mini_header_addr_size(unsigned int *addr, unsigned int *size)
{
	struct ram_console_memory_info *memory_info = NULL;

	if (addr == NULL || size == NULL) {
		LOG("%s. get minirdump header info args invalid\n", MOD);
		return;
	}

	memory_info = ram_console_atag_memory_info();
	*addr = memory_info->mrdump_mini_header_addr;
	*size = memory_info->mrdump_mini_header_size;
}

void ram_console_dts_written(void *fdt)
{
	unsigned int addr = 0;
	unsigned int size = 0;
	unsigned int def_type = 0;
	unsigned int memory_info_offset = 0;
	unsigned int *ptemp;
	unsigned int buf[4];
	int offset;
	int ret;

	offset = fdt_path_offset(fdt, "/chosen");
	if (offset < 0) {
		LOG("%s. get /chosen offset failed(%d)\n", MOD, offset);
		ram_console_fatal("get /chosen offset failed");
		return;
	}
	ptemp = (unsigned int *)buf;

	ram_console_sram_addr_size(&addr, &size);
	*ptemp = addr;
	*(ptemp + 1) = size;
	def_type = ram_console_def_memory();
	*(ptemp + 2) = def_type;
	memory_info_offset = ram_console_sram_offset();
	*(ptemp + 3) = memory_info_offset;
	LOG("%s. [0x%x,0x%x,0x%x,0x%x]\n", MOD, *ptemp, *(ptemp + 1),
		*(ptemp + 2), *(ptemp + 3));
	ret = fdt_setprop(fdt, offset, "ram_console", buf, (4 * sizeof(*ptemp)));
	if (ret)
		ram_console_fatal("dts written failed");

	sram_log_store_addr_size(&addr, &size);
	*ptemp = addr;
	*(ptemp + 1) = size;
	LOG("%s. log_store [0x%x,0x%x]\n", MOD, *ptemp, *(ptemp + 1));
	ret = fdt_setprop(fdt, offset, "log_store", buf, (2 * sizeof(*ptemp)));
	if (ret)
		ram_console_fatal("log_store dts written failed");

}

void ram_console_set_dump_step(unsigned int step)
{
	if (ram_console) {
		ram_console->dump_step = step;
	}
}

int ram_console_get_dump_step()
{
	if (ram_console)
		return ram_console->dump_step;
	else {
		LOG("%s. ram_console not ready\n", MOD);
		return 0;
	}
}