1 // SPDX-License-Identifier: GPL-2.0-or-later 1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /* Function to determine if a thread group is 2 /* Function to determine if a thread group is single threaded or not
3 * 3 *
4 * Copyright (C) 2008 Red Hat, Inc. All Rights 4 * Copyright (C) 2008 Red Hat, Inc. All Rights Reserved.
5 * Written by David Howells (dhowells@redhat.c 5 * Written by David Howells (dhowells@redhat.com)
6 * - Derived from security/selinux/hooks.c 6 * - Derived from security/selinux/hooks.c
7 */ 7 */
8 #include <linux/sched/signal.h> 8 #include <linux/sched/signal.h>
9 #include <linux/sched/task.h> 9 #include <linux/sched/task.h>
10 #include <linux/sched/mm.h> 10 #include <linux/sched/mm.h>
11 11
12 /* 12 /*
13 * Returns true if the task does not share ->m 13 * Returns true if the task does not share ->mm with another thread/process.
14 */ 14 */
15 bool current_is_single_threaded(void) 15 bool current_is_single_threaded(void)
16 { 16 {
17 struct task_struct *task = current; 17 struct task_struct *task = current;
18 struct mm_struct *mm = task->mm; 18 struct mm_struct *mm = task->mm;
19 struct task_struct *p, *t; 19 struct task_struct *p, *t;
20 bool ret; 20 bool ret;
21 21
22 if (atomic_read(&task->signal->live) ! 22 if (atomic_read(&task->signal->live) != 1)
23 return false; 23 return false;
24 24
25 if (atomic_read(&mm->mm_users) == 1) 25 if (atomic_read(&mm->mm_users) == 1)
26 return true; 26 return true;
27 27
28 ret = false; 28 ret = false;
29 rcu_read_lock(); 29 rcu_read_lock();
30 for_each_process(p) { 30 for_each_process(p) {
31 if (unlikely(p->flags & PF_KTH 31 if (unlikely(p->flags & PF_KTHREAD))
32 continue; 32 continue;
33 if (unlikely(p == task->group_ 33 if (unlikely(p == task->group_leader))
34 continue; 34 continue;
35 35
36 for_each_thread(p, t) { 36 for_each_thread(p, t) {
37 if (unlikely(t->mm == 37 if (unlikely(t->mm == mm))
38 goto found; 38 goto found;
39 if (likely(t->mm)) 39 if (likely(t->mm))
40 break; 40 break;
41 /* 41 /*
42 * t->mm == NULL. Make 42 * t->mm == NULL. Make sure next_thread/next_task
43 * will see other CLON 43 * will see other CLONE_VM tasks which might be
44 * forked before exiti 44 * forked before exiting.
45 */ 45 */
46 smp_rmb(); 46 smp_rmb();
47 } 47 }
48 } 48 }
49 ret = true; 49 ret = true;
50 found: 50 found:
51 rcu_read_unlock(); 51 rcu_read_unlock();
52 52
53 return ret; 53 return ret;
54 } 54 }
55 55
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.