1 files changed, 10 insertions, 3 deletions

diff --git a/arch/um/kernel/reboot.c b/arch/um/kernel/reboot.c
index 4d93dff6b371..3d15243ce692 100644
--- a/arch/um/kernel/reboot.c
+++ b/arch/um/kernel/reboot.c
@@ -4,7 +4,9 @@
  */
 
 #include "linux/sched.h"
+#include "linux/spinlock.h"
 #include "linux/slab.h"
+#include "linux/oom.h"
 #include "kern_util.h"
 #include "os.h"
 #include "skas.h"
@@ -22,13 +24,18 @@ static void kill_off_processes(void)
                 struct task_struct *p;
                 int pid;
 
+                read_lock(&tasklist_lock);
                 for_each_process(p) {
-                        if (p->mm == NULL)
-                                continue;
+                        struct task_struct *t;
 
-                        pid = p->mm->context.id.u.pid;
+                        t = find_lock_task_mm(p);
+                        if (!t)
+                                continue;
+                        pid = t->mm->context.id.u.pid;
+                        task_unlock(t);
                         os_kill_ptraced_process(pid, 1);
                 }
+                read_unlock(&tasklist_lock);
         }
 }