// SPDX-License-Identifier: (GPL-2.0+ OR MIT)
/*
 * Copyright (c) 2019 Amlogic, Inc. All rights reserved.
 */

#include <common.h>
#include <stdlib.h>

#define __asmeq(x, y) ".ifnc " x "," y " ; .err ; .endif\n\t"

#define OPTEE_SMC_CHECK_WM_STATUS        0xB200E003
#define OPTEE_SMC_INIT_WM                0xB200E080

#define CMD_RET_SUCCESS                  0x00000000

static uint32_t check_wm_sts(void)
{
	register uint32_t x0 asm("x0") = OPTEE_SMC_CHECK_WM_STATUS;

	do {
		asm volatile(__asmeq("%0", "x0")
			__asmeq("%1", "x0")
			"smc    #0\n"
			: "=r"(x0)
			: "r"(x0));
	} while (0);

	return x0;
}

static uint32_t init_wm(void)
{
	register uint32_t x0 asm("x0") = OPTEE_SMC_INIT_WM;

	do {
		asm volatile(__asmeq("%0", "x0")
			__asmeq("%1", "x0")
			"smc    #0\n"
			: "=r"(x0)
			: "r"(x0));
	} while (0);

	return x0;
}

static int do_wm_init(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	if (check_wm_sts() == 0)
		printf("WATERMARK: initialize Watermark return 0x%08X\n", init_wm());
	else
		printf("WATERMARK: Watermark is disabled\n");

	return CMD_RET_SUCCESS;
}

/* -------------------------------------------------------------------- */
U_BOOT_CMD(watermark_init, CONFIG_SYS_MAXARGS, 0, do_wm_init,
	"initialize watermark\n",
	"watermark_init");