From f0cfd2d0dca7f876717883318a1e9b5df638524c Mon Sep 17 00:00:00 2001
From: Maulik Shah <mkshah@codeaurora.org>
Date: Mon, 15 Mar 2021 19:25:23 +0530
Subject: [PATCH] sched: idle: Add support to print wake irqs during s2idle

syscore ops prints wake irq during "deep" suspend exit however
syscore ops do not execute for "s2idle" suspend.

Add changes to print wakeup irq during "s2idle" mode exit.

The change prints wakeup irq from the first cpu waking up from s2idle
during which local irqs on the cpu is disabled.

Change-Id: Idd05d42f30b97cc5f9d62627339e21731f131673
Signed-off-by: Maulik Shah <mkshah@codeaurora.org>
---
 drivers/irqchip/Kconfig            |  8 ++++++++
 drivers/irqchip/irq-gic-v3.c       |  8 ++++++++
 include/linux/irqchip/arm-gic-v3.h |  4 ++++
 kernel/sched/idle.c                | 25 +++++++++++++++++++++++++
 4 files changed, 45 insertions(+)

diff --git a/drivers/irqchip/Kconfig b/drivers/irqchip/Kconfig
index d039126411be..6c4d8619c4e1 100644
--- a/drivers/irqchip/Kconfig
+++ b/drivers/irqchip/Kconfig
@@ -37,6 +37,7 @@ config ARM_GIC_V3
 	select IRQ_DOMAIN_HIERARCHY
 	select PARTITION_PERCPU
 	select GENERIC_IRQ_EFFECTIVE_AFF_MASK
+	select QGKI_SHOW_S2IDLE_WAKE_IRQ if QGKI
 
 config ARM_GIC_V3_ITS
 	bool
@@ -516,4 +517,11 @@ config SIFIVE_PLIC
 
 	   If you don't know what to do here, say Y.
 
+config QGKI_SHOW_S2IDLE_WAKE_IRQ
+	bool "Qualcomm Technologies, Inc. (QTI) SHOW S2IDLE WAKE IRQ"
+	depends on ARM_GIC_V3
+	help
+	   When this option is selected, first cpu waking up from s2idle
+	   prints the interrupts pending and enabled at GIC.
+
 endmenu
diff --git a/drivers/irqchip/irq-gic-v3.c b/drivers/irqchip/irq-gic-v3.c
index bf7be1b538d1..b9c71e36afda 100644
--- a/drivers/irqchip/irq-gic-v3.c
+++ b/drivers/irqchip/irq-gic-v3.c
@@ -18,6 +18,7 @@
 #include <linux/percpu.h>
 #include <linux/refcount.h>
 #include <linux/slab.h>
+#include <linux/suspend.h>
 #include <linux/msm_rtb.h>
 #include <linux/wakeup_reason.h>
 
@@ -1244,6 +1245,13 @@ static int gic_set_affinity(struct irq_data *d, const struct cpumask *mask_val,
 #define gic_smp_init()		do { } while(0)
 #endif
 
+#ifdef CONFIG_QGKI_SHOW_S2IDLE_WAKE_IRQ
+void gic_s2idle_wake(void)
+{
+	gic_resume_one(&gic_data);
+}
+#endif /* CONFIG_QGKI_SHOW_S2IDLE_WAKE_IRQ */
+
 #ifdef CONFIG_CPU_PM
 static int gic_cpu_pm_notifier(struct notifier_block *self,
 			       unsigned long cmd, void *v)
diff --git a/include/linux/irqchip/arm-gic-v3.h b/include/linux/irqchip/arm-gic-v3.h
index 5cc10cf7cb3e..2f349fb5f196 100644
--- a/include/linux/irqchip/arm-gic-v3.h
+++ b/include/linux/irqchip/arm-gic-v3.h
@@ -638,6 +638,10 @@ static inline bool gic_enable_sre(void)
 	return !!(val & ICC_SRE_EL1_SRE);
 }
 
+#ifdef CONFIG_QGKI_SHOW_S2IDLE_WAKE_IRQ
+void gic_s2idle_wake(void);
+#endif /* CONFIG_QGKI_SHOW_S2IDLE_WAKE_IRQ */
+
 #endif
 
 #endif
diff --git a/kernel/sched/idle.c b/kernel/sched/idle.c
index 251a7df055d6..93b3195bfd97 100644
--- a/kernel/sched/idle.c
+++ b/kernel/sched/idle.c
@@ -8,6 +8,9 @@
  */
 #include "sched.h"
 
+#ifdef CONFIG_QGKI_SHOW_S2IDLE_WAKE_IRQ
+#include <linux/irqchip/arm-gic-v3.h>
+#endif /* CONFIG_QGKI_SHOW_S2IDLE_WAKE_IRQ */
 #include <trace/events/power.h>
 
 /* Linker adds these: start and end of __cpuidle functions */
@@ -128,6 +131,11 @@ static int call_cpuidle(struct cpuidle_driver *drv, struct cpuidle_device *dev,
  * set, and it returns with polling set.  If it ever stops polling, it
  * must clear the polling bit.
  */
+
+#ifdef CONFIG_QGKI_SHOW_S2IDLE_WAKE_IRQ
+static cpumask_t cpu_state;
+#endif /* CONFIG_QGKI_SHOW_S2IDLE_WAKE_IRQ */
+
 static void cpuidle_idle_call(void)
 {
 	struct cpuidle_device *dev = cpuidle_get_device();
@@ -169,9 +177,26 @@ static void cpuidle_idle_call(void)
 
 	if (idle_should_enter_s2idle() || dev->use_deepest_state) {
 		if (idle_should_enter_s2idle()) {
+
+#ifdef CONFIG_QGKI_SHOW_S2IDLE_WAKE_IRQ
+			bool print_wake_irq;
+
+			cpumask_set_cpu(dev->cpu, &cpu_state);
+#endif /* CONFIG_QGKI_SHOW_S2IDLE_WAKE_IRQ */
+
 			rcu_idle_enter();
 
 			entered_state = cpuidle_enter_s2idle(drv, dev);
+
+#ifdef CONFIG_QGKI_SHOW_S2IDLE_WAKE_IRQ
+			print_wake_irq = cpumask_weight(&cpu_state) ==
+				    cpumask_weight(cpu_online_mask) ?
+				    true : false;
+			cpumask_clear_cpu(dev->cpu, &cpu_state);
+			if (print_wake_irq)
+				gic_s2idle_wake();
+#endif /* CONFIG_QGKI_SHOW_S2IDLE_WAKE_IRQ */
+
 			if (entered_state > 0) {
 				local_irq_enable();
 				goto exit_idle;