From 187a20276c12e36b5e1aa925f8f4cf7a5757eef3 Mon Sep 17 00:00:00 2001
From: XiaoDong Huang <derrick.huang@rock-chips.com>
Date: Mon, 9 May 2022 15:52:07 +0800
Subject: [PATCH] firmware: rockchip_sip: support wdt config

Signed-off-by: XiaoDong Huang <derrick.huang@rock-chips.com>
Change-Id: I49bceb0dfe0c6bed797d34e98a48ab793f91dbce
---
 drivers/firmware/rockchip_sip.c       | 12 ++++++++++++
 include/linux/rockchip/rockchip_sip.h | 17 +++++++++++++++++
 2 files changed, 29 insertions(+)

diff --git a/drivers/firmware/rockchip_sip.c b/drivers/firmware/rockchip_sip.c
index d955c17b0872..46ab17e7603c 100644
--- a/drivers/firmware/rockchip_sip.c
+++ b/drivers/firmware/rockchip_sip.c
@@ -584,6 +584,18 @@ int sip_fiq_control(u32 sub_func, u32 irq, unsigned long data)
 	return res.a0;
 }
 EXPORT_SYMBOL_GPL(sip_fiq_control);
+
+int sip_wdt_config(u32 sub_func, u32 arg1, u32 arg2, u32 arg3)
+{
+	struct arm_smccc_res res;
+
+	arm_smccc_smc(SIP_WDT_CFG, sub_func, arg1, arg2, arg3,
+		      0, 0, 0, &res);
+
+	return res.a0;
+}
+EXPORT_SYMBOL_GPL(sip_wdt_config);
+
 /******************************************************************************/
 #ifdef CONFIG_ARM
 static __init int sip_firmware_init(void)
diff --git a/include/linux/rockchip/rockchip_sip.h b/include/linux/rockchip/rockchip_sip.h
index 6f089f01a58b..b9572da69053 100644
--- a/include/linux/rockchip/rockchip_sip.h
+++ b/include/linux/rockchip/rockchip_sip.h
@@ -54,6 +54,7 @@
 #define RK_SIP_AMP_CFG			0x82000022
 #define RK_SIP_FIQ_CTRL			0x82000024
 #define SIP_HDCP_CONFIG			0x82000025
+#define SIP_WDT_CFG			0x82000026
 
 /* Rockchip Sip version */
 #define SIP_IMPLEMENT_V1                (1)
@@ -182,6 +183,13 @@ enum {
 	MAX_DEVICE,
 };
 
+/* SIP_WDT_CONFIG call types  */
+enum {
+	WDT_START = 0,
+	WDT_STOP = 1,
+	WDT_PING = 2,
+};
+
 /*
  * Rules: struct arm_smccc_res contains result and data, details:
  *
@@ -224,6 +232,7 @@ int sip_fiq_debugger_sdei_switch_cpu(u32 cur_cpu, u32 target_cpu, u32 flag);
 int sip_fiq_debugger_is_enabled(void);
 int sip_fiq_debugger_sdei_get_event_id(u32 *fiq, u32 *sw_cpu, u32 *flag);
 int sip_fiq_control(u32 sub_func, u32 irq, unsigned long data);
+int sip_wdt_config(u32 sub_func, u32 arg1, u32 arg2, u32 arg3);
 #else
 static inline struct arm_smccc_res sip_smc_get_atf_version(void)
 {
@@ -348,6 +357,14 @@ static inline int sip_fiq_control(u32 sub_func, u32 irq, unsigned long data)
 {
 	return 0;
 }
+
+static inline int sip_wdt_config(u32 sub_func,
+				 u32 arg1,
+				 u32 arg2,
+				 u32 arg3)
+{
+	return 0;
+}
 #endif
 
 /* 32-bit OP-TEE context, never change order of members! */