[PATCH] Fix obvious stupid race in do_stop

We don't set the task state to TASK_INTERRUPTIBLE _before_ checking for
kthread_should_stop in do_stop.

kernel/stop_machine.c

@@ -149,10 +149,12 @@ static int do_stop(void *_smdata)
 	complete(&smdata->done);
 
 	/* Wait for kthread_stop */
-	while (!kthread_should_stop()) {
 	__set_current_state(TASK_INTERRUPTIBLE);
+	while (!kthread_should_stop()) {
 		schedule();
+		__set_current_state(TASK_INTERRUPTIBLE);
 	}
+	__set_current_state(TASK_RUNNING);
 	return ret;
 }