--- zzzz-none-000/linux-4.1.52/arch/arm/kernel/psci.c	2018-05-28 02:26:45.000000000 +0000
+++ bcm63-7530ax-731/linux-4.1.52/arch/arm/kernel/psci.c	2022-03-02 11:37:12.000000000 +0000
@@ -80,6 +80,17 @@
 	return err;
 }
 
+#if defined(CONFIG_BCM_KF_OPTEE) && defined (CONFIG_BCM_KERNEL_CODE_PROTECTION)
+static int psci_mem_info(void)
+{
+	int err;
+	extern char _stext[], _etext[], __init_begin[];
+
+	err = invoke_psci_fn(PSCI_0_2_FN_MEM_INFO, (u32)_stext, (u32)_etext, (u32)__init_begin);
+	return err;
+}
+#endif
+
 static int psci_cpu_suspend(struct psci_power_state state,
 			    unsigned long entry_point)
 {
@@ -232,7 +243,9 @@
 	arm_pm_restart = psci_sys_reset;
 
 	pm_power_off = psci_sys_poweroff;
-
+#if defined(CONFIG_BCM_KF_OPTEE) && defined (CONFIG_BCM_KERNEL_CODE_PROTECTION)
+	psci_mem_info();
+#endif
 out_put_node:
 	of_node_put(np);
 	return err;