From: Arve Hjønnevåg <arve@android.com>
Date: Thu, 11 Dec 2008 04:06:28 +0000 (-0800)
Subject: sched: Enable might_sleep before initializing drivers.
X-Git-Tag: firefly_0821_release~2958^2~319
X-Git-Url: http://demsky.eecs.uci.edu/git/?a=commitdiff_plain;h=b3bb0e5bbf24780300621d69e6ffbdcc11c72358;p=firefly-linux-kernel-4.4.55.git

sched: Enable might_sleep before initializing drivers.

This allows detection of init bugs in built-in drivers.

Signed-off-by: Arve Hjønnevåg <arve@android.com>
---

diff --git a/kernel/sched/core.c b/kernel/sched/core.c
index 732e993b564b..67359c7f35db 100644
--- a/kernel/sched/core.c
+++ b/kernel/sched/core.c
@@ -7542,6 +7542,14 @@ static inline int preempt_count_equals(int preempt_offset)
 	return (nested == preempt_offset);
 }
 
+static int __might_sleep_init_called;
+int __init __might_sleep_init(void)
+{
+	__might_sleep_init_called = 1;
+	return 0;
+}
+early_initcall(__might_sleep_init);
+
 void __might_sleep(const char *file, int line, int preempt_offset)
 {
 	/*
@@ -7566,8 +7574,10 @@ void ___might_sleep(const char *file, int line, int preempt_offset)
 
 	rcu_sleep_check(); /* WARN_ON_ONCE() by default, no rate limit reqd. */
 	if ((preempt_count_equals(preempt_offset) && !irqs_disabled() &&
-	     !is_idle_task(current)) ||
-	    system_state != SYSTEM_RUNNING || oops_in_progress)
+	     !is_idle_task(current)) || oops_in_progress)
+		return;
+	if (system_state != SYSTEM_RUNNING &&
+	    (!__might_sleep_init_called || system_state != SYSTEM_BOOTING))
 		return;
 	if (time_before(jiffies, prev_jiffy + HZ) && prev_jiffy)
 		return;