--- zzzz-none-000/linux-4.1.52/arch/arm64/kernel/psci.c	2018-05-28 02:26:45.000000000 +0000
+++ bcm63-7530ax-731/linux-4.1.52/arch/arm64/kernel/psci.c	2022-03-02 11:37:12.000000000 +0000
@@ -122,6 +122,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)(long)_stext, (u32)(long)_etext, (u32)(long)__init_begin);
+	return err;
+}
+#endif
+
 static int psci_cpu_suspend(struct psci_power_state state,
 			    unsigned long entry_point)
 {
@@ -329,7 +340,9 @@
 	}
 
 	psci_0_2_set_functions();
-
+#if defined(CONFIG_BCM_KF_OPTEE) && defined (CONFIG_BCM_KERNEL_CODE_PROTECTION)
+	psci_mem_info();
+#endif
 	return 0;
 }