PM / Sleep: Add "prevent autosleep time" statistics to wakeup sources

Android uses one wakelock statistics that is only necessary for
opportunistic sleep.  Namely, the prevent_suspend_time field
accumulates the total time the given wakelock has been locked
while "automatic suspend" was enabled.  Add an analogous field,
prevent_sleep_time, to wakeup sources and make it behave in a similar
way.

Signed-off-by: Rafael J. Wysocki <rjw@sisk.pl>
Acked-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>

2 files changed, 81 insertions, 4 deletions

diff --git a/drivers/base/power/sysfs.c b/drivers/base/power/sysfs.c
index 13e40b9021b9..48be2ad4dd2c 100644
--- a/drivers/base/power/sysfs.c
+++ b/drivers/base/power/sysfs.c
@@ -417,6 +417,27 @@ static ssize_t wakeup_last_time_show(struct device *dev,
 }
 
 static DEVICE_ATTR(wakeup_last_time_ms, 0444, wakeup_last_time_show, NULL);
+
+#ifdef CONFIG_PM_AUTOSLEEP
+static ssize_t wakeup_prevent_sleep_time_show(struct device *dev,
+                                              struct device_attribute *attr,
+                                              char *buf)
+{
+        s64 msec = 0;
+        bool enabled = false;
+
+        spin_lock_irq(&dev->power.lock);
+        if (dev->power.wakeup) {
+                msec = ktime_to_ms(dev->power.wakeup->prevent_sleep_time);
+                enabled = true;
+        }
+        spin_unlock_irq(&dev->power.lock);
+        return enabled ? sprintf(buf, "%lld\n", msec) : sprintf(buf, "\n");
+}
+
+static DEVICE_ATTR(wakeup_prevent_sleep_time_ms, 0444,
+                   wakeup_prevent_sleep_time_show, NULL);
+#endif /* CONFIG_PM_AUTOSLEEP */
 #endif /* CONFIG_PM_SLEEP */
 
 #ifdef CONFIG_PM_ADVANCED_DEBUG
@@ -511,6 +532,9 @@ static struct attribute *wakeup_attrs[] = {
         &dev_attr_wakeup_total_time_ms.attr,
         &dev_attr_wakeup_max_time_ms.attr,
         &dev_attr_wakeup_last_time_ms.attr,
+#ifdef CONFIG_PM_AUTOSLEEP
+        &dev_attr_wakeup_prevent_sleep_time_ms.attr,
+#endif
 #endif
         NULL,
 };
diff --git a/drivers/base/power/wakeup.c b/drivers/base/power/wakeup.c
index cf1706df7610..2595b8d8fe1f 100644
--- a/drivers/base/power/wakeup.c
+++ b/drivers/base/power/wakeup.c
@@ -380,6 +380,8 @@ static void wakeup_source_activate(struct wakeup_source *ws)
         ws->active = true;
         ws->active_count++;
         ws->last_time = ktime_get();
+        if (ws->autosleep_enabled)
+                ws->start_prevent_time = ws->last_time;
 
         /* Increment the counter of events in progress. */
         cec = atomic_inc_return(&combined_event_count);
@@ -449,6 +451,17 @@ void pm_stay_awake(struct device *dev)
 }
 EXPORT_SYMBOL_GPL(pm_stay_awake);
 
+#ifdef CONFIG_PM_AUTOSLEEP
+static void update_prevent_sleep_time(struct wakeup_source *ws, ktime_t now)
+{
+        ktime_t delta = ktime_sub(now, ws->start_prevent_time);
+        ws->prevent_sleep_time = ktime_add(ws->prevent_sleep_time, delta);
+}
+#else
+static inline void update_prevent_sleep_time(struct wakeup_source *ws,
+                                             ktime_t now) {}
+#endif
+
 /**
  * wakup_source_deactivate - Mark given wakeup source as inactive.
  * @ws: Wakeup source to handle.
@@ -490,6 +503,9 @@ static void wakeup_source_deactivate(struct wakeup_source *ws)
         del_timer(&ws->timer);
         ws->timer_expires = 0;
 
+        if (ws->autosleep_enabled)
+                update_prevent_sleep_time(ws, now);
+
         /*
          * Increment the counter of registered wakeup events and decrement the
          * couter of wakeup events in progress simultaneously.
@@ -718,6 +734,34 @@ bool pm_save_wakeup_count(unsigned int count)
         return events_check_enabled;
 }
 
+#ifdef CONFIG_PM_AUTOSLEEP
+/**
+ * pm_wakep_autosleep_enabled - Modify autosleep_enabled for all wakeup sources.
+ * @enabled: Whether to set or to clear the autosleep_enabled flags.
+ */
+void pm_wakep_autosleep_enabled(bool set)
+{
+        struct wakeup_source *ws;
+        ktime_t now = ktime_get();
+
+        rcu_read_lock();
+        list_for_each_entry_rcu(ws, &wakeup_sources, entry) {
+                spin_lock_irq(&ws->lock);
+                if (ws->autosleep_enabled != set) {
+                        ws->autosleep_enabled = set;
+                        if (ws->active) {
+                                if (set)
+                                        ws->start_prevent_time = now;
+                                else
+                                        update_prevent_sleep_time(ws, now);
+                        }
+                }
+                spin_unlock_irq(&ws->lock);
+        }
+        rcu_read_unlock();
+}
+#endif /* CONFIG_PM_AUTOSLEEP */
+
 static struct dentry *wakeup_sources_stats_dentry;
 
 /**
@@ -733,28 +777,37 @@ static int print_wakeup_source_stats(struct seq_file *m,
         ktime_t max_time;
         unsigned long active_count;
         ktime_t active_time;
+        ktime_t prevent_sleep_time;
         int ret;
 
         spin_lock_irqsave(&ws->lock, flags);
 
         total_time = ws->total_time;
         max_time = ws->max_time;
+        prevent_sleep_time = ws->prevent_sleep_time;
         active_count = ws->active_count;
         if (ws->active) {
-                active_time = ktime_sub(ktime_get(), ws->last_time);
+                ktime_t now = ktime_get();
+
+                active_time = ktime_sub(now, ws->last_time);
                 total_time = ktime_add(total_time, active_time);
                 if (active_time.tv64 > max_time.tv64)
                         max_time = active_time;
+
+                if (ws->autosleep_enabled)
+                        prevent_sleep_time = ktime_add(prevent_sleep_time,
+                                ktime_sub(now, ws->start_prevent_time));
         } else {
                 active_time = ktime_set(0, 0);
         }
 
         ret = seq_printf(m, "%-12s\t%lu\t\t%lu\t\t%lu\t\t%lu\t\t"
-                        "%lld\t\t%lld\t\t%lld\t\t%lld\n",
+                        "%lld\t\t%lld\t\t%lld\t\t%lld\t\t%lld\n",
                         ws->name, active_count, ws->event_count,
                         ws->wakeup_count, ws->expire_count,
                         ktime_to_ms(active_time), ktime_to_ms(total_time),
-                        ktime_to_ms(max_time), ktime_to_ms(ws->last_time));
+                        ktime_to_ms(max_time), ktime_to_ms(ws->last_time),
+                        ktime_to_ms(prevent_sleep_time));
 
         spin_unlock_irqrestore(&ws->lock, flags);
 
@@ -771,7 +824,7 @@ static int wakeup_sources_stats_show(struct seq_file *m, void *unused)
 
         seq_puts(m, "name\t\tactive_count\tevent_count\twakeup_count\t"
                 "expire_count\tactive_since\ttotal_time\tmax_time\t"
-                "last_change\n");
+                "last_change\tprevent_suspend_time\n");
 
         rcu_read_lock();
         list_for_each_entry_rcu(ws, &wakeup_sources, entry)