1 files changed, 11 insertions, 6 deletions

diff --git a/drivers/base/power/wakeup.c b/drivers/base/power/wakeup.c
index 8a0a9ca6ad65..e6ee5e80e546 100644
--- a/drivers/base/power/wakeup.c
+++ b/drivers/base/power/wakeup.c
@@ -127,6 +127,8 @@ EXPORT_SYMBOL_GPL(wakeup_source_destroy);
  */
 void wakeup_source_add(struct wakeup_source *ws)
 {
+        unsigned long flags;
+
         if (WARN_ON(!ws))
                 return;
 
@@ -135,9 +137,9 @@ void wakeup_source_add(struct wakeup_source *ws)
         ws->active = false;
         ws->last_time = ktime_get();
 
-        spin_lock_irq(&events_lock);
+        spin_lock_irqsave(&events_lock, flags);
         list_add_rcu(&ws->entry, &wakeup_sources);
-        spin_unlock_irq(&events_lock);
+        spin_unlock_irqrestore(&events_lock, flags);
 }
 EXPORT_SYMBOL_GPL(wakeup_source_add);
 
@@ -147,12 +149,14 @@ EXPORT_SYMBOL_GPL(wakeup_source_add);
  */
 void wakeup_source_remove(struct wakeup_source *ws)
 {
+        unsigned long flags;
+
         if (WARN_ON(!ws))
                 return;
 
-        spin_lock_irq(&events_lock);
+        spin_lock_irqsave(&events_lock, flags);
         list_del_rcu(&ws->entry);
-        spin_unlock_irq(&events_lock);
+        spin_unlock_irqrestore(&events_lock, flags);
         synchronize_rcu();
 }
 EXPORT_SYMBOL_GPL(wakeup_source_remove);
@@ -752,15 +756,16 @@ bool pm_get_wakeup_count(unsigned int *count, bool block)
 bool pm_save_wakeup_count(unsigned int count)
 {
         unsigned int cnt, inpr;
+        unsigned long flags;
 
         events_check_enabled = false;
-        spin_lock_irq(&events_lock);
+        spin_lock_irqsave(&events_lock, flags);
         split_counters(&cnt, &inpr);
         if (cnt == count && inpr == 0) {
                 saved_count = count;
                 events_check_enabled = true;
         }
-        spin_unlock_irq(&events_lock);
+        spin_unlock_irqrestore(&events_lock, flags);
         return events_check_enabled;
 }