exit.c 46.5 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8
/*
 *  linux/kernel/exit.c
 *
 *  Copyright (C) 1991, 1992  Linus Torvalds
 */

#include <linux/mm.h>
#include <linux/slab.h>
9
#include <linux/sched/autogroup.h>
10
#include <linux/sched/mm.h>
11
#include <linux/sched/stat.h>
12
#include <linux/sched/task.h>
13
#include <linux/sched/task_stack.h>
14
#include <linux/sched/cputime.h>
L
Linus Torvalds 已提交
15 16
#include <linux/interrupt.h>
#include <linux/module.h>
17
#include <linux/capability.h>
L
Linus Torvalds 已提交
18 19 20
#include <linux/completion.h>
#include <linux/personality.h>
#include <linux/tty.h>
21
#include <linux/iocontext.h>
L
Linus Torvalds 已提交
22 23 24
#include <linux/key.h>
#include <linux/cpu.h>
#include <linux/acct.h>
25
#include <linux/tsacct_kern.h>
L
Linus Torvalds 已提交
26
#include <linux/file.h>
A
Al Viro 已提交
27
#include <linux/fdtable.h>
28
#include <linux/freezer.h>
L
Linus Torvalds 已提交
29
#include <linux/binfmts.h>
S
Serge E. Hallyn 已提交
30
#include <linux/nsproxy.h>
31
#include <linux/pid_namespace.h>
L
Linus Torvalds 已提交
32 33 34 35
#include <linux/ptrace.h>
#include <linux/profile.h>
#include <linux/mount.h>
#include <linux/proc_fs.h>
36
#include <linux/kthread.h>
L
Linus Torvalds 已提交
37
#include <linux/mempolicy.h>
38
#include <linux/taskstats_kern.h>
39
#include <linux/delayacct.h>
40
#include <linux/cgroup.h>
L
Linus Torvalds 已提交
41
#include <linux/syscalls.h>
42
#include <linux/signal.h>
43
#include <linux/posix-timers.h>
M
Matt Helsley 已提交
44
#include <linux/cn_proc.h>
45
#include <linux/mutex.h>
46
#include <linux/futex.h>
47
#include <linux/pipe_fs_i.h>
48
#include <linux/audit.h> /* for audit_free() */
A
Adrian Bunk 已提交
49
#include <linux/resource.h>
50
#include <linux/blkdev.h>
51
#include <linux/task_io_accounting_ops.h>
R
Roland McGrath 已提交
52
#include <linux/tracehook.h>
53
#include <linux/fs_struct.h>
D
David Howells 已提交
54
#include <linux/init_task.h>
55
#include <linux/perf_event.h>
56
#include <trace/events/sched.h>
57
#include <linux/hw_breakpoint.h>
Y
Ying Han 已提交
58
#include <linux/oom.h>
59
#include <linux/writeback.h>
A
Al Viro 已提交
60
#include <linux/shm.h>
D
Dmitry Vyukov 已提交
61
#include <linux/kcov.h>
62
#include <linux/random.h>
63
#include <linux/rcuwait.h>
64
#include <linux/compat.h>
L
Linus Torvalds 已提交
65

66
#include <linux/uaccess.h>
L
Linus Torvalds 已提交
67 68 69 70
#include <asm/unistd.h>
#include <asm/pgtable.h>
#include <asm/mmu_context.h>

71
static void __unhash_process(struct task_struct *p, bool group_dead)
L
Linus Torvalds 已提交
72 73
{
	nr_threads--;
74
	detach_pid(p, PIDTYPE_PID);
75
	if (group_dead) {
E
Eric W. Biederman 已提交
76
		detach_pid(p, PIDTYPE_TGID);
L
Linus Torvalds 已提交
77 78
		detach_pid(p, PIDTYPE_PGID);
		detach_pid(p, PIDTYPE_SID);
79

80
		list_del_rcu(&p->tasks);
81
		list_del_init(&p->sibling);
82
		__this_cpu_dec(process_counts);
L
Linus Torvalds 已提交
83
	}
O
Oleg Nesterov 已提交
84
	list_del_rcu(&p->thread_group);
85
	list_del_rcu(&p->thread_node);
L
Linus Torvalds 已提交
86 87
}

88 89 90 91 92 93
/*
 * This function expects the tasklist_lock write-locked.
 */
static void __exit_signal(struct task_struct *tsk)
{
	struct signal_struct *sig = tsk->signal;
94
	bool group_dead = thread_group_leader(tsk);
95
	struct sighand_struct *sighand;
96
	struct tty_struct *uninitialized_var(tty);
97
	u64 utime, stime;
98

99
	sighand = rcu_dereference_check(tsk->sighand,
100
					lockdep_tasklist_lock_is_held());
101 102
	spin_lock(&sighand->siglock);

103
#ifdef CONFIG_POSIX_TIMERS
104
	posix_cpu_timers_exit(tsk);
105
	if (group_dead) {
106
		posix_cpu_timers_exit_group(tsk);
107
	} else {
108 109 110 111 112 113 114
		/*
		 * This can only happen if the caller is de_thread().
		 * FIXME: this is the temporary hack, we should teach
		 * posix-cpu-timers to handle this case correctly.
		 */
		if (unlikely(has_group_leader_pid(tsk)))
			posix_cpu_timers_exit_group(tsk);
115 116
	}
#endif
117

118 119 120 121
	if (group_dead) {
		tty = sig->tty;
		sig->tty = NULL;
	} else {
122 123 124 125
		/*
		 * If there is any task waiting for the group exit
		 * then notify it:
		 */
126
		if (sig->notify_count > 0 && !--sig->notify_count)
127
			wake_up_process(sig->group_exit_task);
128

129 130 131 132
		if (tsk == sig->curr_target)
			sig->curr_target = next_thread(tsk);
	}

133 134 135
	add_device_randomness((const void*) &tsk->se.sum_exec_runtime,
			      sizeof(unsigned long long));

136
	/*
137 138 139 140
	 * Accumulate here the counters for all threads as they die. We could
	 * skip the group leader because it is the last user of signal_struct,
	 * but we want to avoid the race with thread_group_cputime() which can
	 * see the empty ->thread_head list.
141 142
	 */
	task_cputime(tsk, &utime, &stime);
143
	write_seqlock(&sig->stats_lock);
144 145 146 147 148 149 150 151 152 153 154
	sig->utime += utime;
	sig->stime += stime;
	sig->gtime += task_gtime(tsk);
	sig->min_flt += tsk->min_flt;
	sig->maj_flt += tsk->maj_flt;
	sig->nvcsw += tsk->nvcsw;
	sig->nivcsw += tsk->nivcsw;
	sig->inblock += task_io_get_inblock(tsk);
	sig->oublock += task_io_get_oublock(tsk);
	task_io_accounting_add(&sig->ioac, &tsk->ioac);
	sig->sum_sched_runtime += tsk->se.sum_exec_runtime;
155
	sig->nr_threads--;
156
	__unhash_process(tsk, group_dead);
157
	write_sequnlock(&sig->stats_lock);
158

159 160 161 162 163
	/*
	 * Do this under ->siglock, we can race with another thread
	 * doing sigqueue_free() if we have SIGQUEUE_PREALLOC signals.
	 */
	flush_sigqueue(&tsk->pending);
164
	tsk->sighand = NULL;
165 166
	spin_unlock(&sighand->siglock);

167
	__cleanup_sighand(sighand);
168
	clear_tsk_thread_flag(tsk, TIF_SIGPENDING);
169
	if (group_dead) {
170
		flush_sigqueue(&sig->shared_pending);
171
		tty_kref_put(tty);
172 173 174
	}
}

175 176
static void delayed_put_task_struct(struct rcu_head *rhp)
{
177 178
	struct task_struct *tsk = container_of(rhp, struct task_struct, rcu);

179
	perf_event_delayed_put(tsk);
180 181
	trace_sched_process_free(tsk);
	put_task_struct(tsk);
182 183
}

R
Roland McGrath 已提交
184

185
void release_task(struct task_struct *p)
L
Linus Torvalds 已提交
186
{
187
	struct task_struct *leader;
L
Linus Torvalds 已提交
188
	int zap_leader;
189
repeat:
190
	/* don't need to get the RCU readlock here - the process is dead and
191 192
	 * can't be modifying its own credentials. But shut RCU-lockdep up */
	rcu_read_lock();
193
	atomic_dec(&__task_cred(p)->user->processes);
194
	rcu_read_unlock();
195

196
	proc_flush_task(p);
197
	cgroup_release(p);
198

L
Linus Torvalds 已提交
199
	write_lock_irq(&tasklist_lock);
T
Tejun Heo 已提交
200
	ptrace_release_task(p);
L
Linus Torvalds 已提交
201
	__exit_signal(p);
202

L
Linus Torvalds 已提交
203 204 205 206 207 208 209
	/*
	 * If we are the last non-leader member of the thread
	 * group, and the leader is zombie, then notify the
	 * group leader's parent process. (if it wants notification.)
	 */
	zap_leader = 0;
	leader = p->group_leader;
210 211
	if (leader != p && thread_group_empty(leader)
			&& leader->exit_state == EXIT_ZOMBIE) {
L
Linus Torvalds 已提交
212 213 214 215
		/*
		 * If we were the last child thread and the leader has
		 * exited already, and the leader's parent ignores SIGCHLD,
		 * then we are the one who should release the leader.
R
Roland McGrath 已提交
216
		 */
217
		zap_leader = do_notify_parent(leader, leader->exit_signal);
R
Roland McGrath 已提交
218 219
		if (zap_leader)
			leader->exit_state = EXIT_DEAD;
L
Linus Torvalds 已提交
220 221 222 223
	}

	write_unlock_irq(&tasklist_lock);
	release_thread(p);
224
	call_rcu(&p->rcu, delayed_put_task_struct);
L
Linus Torvalds 已提交
225 226 227 228 229 230

	p = leader;
	if (unlikely(zap_leader))
		goto repeat;
}

231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293
/*
 * Note that if this function returns a valid task_struct pointer (!NULL)
 * task->usage must remain >0 for the duration of the RCU critical section.
 */
struct task_struct *task_rcu_dereference(struct task_struct **ptask)
{
	struct sighand_struct *sighand;
	struct task_struct *task;

	/*
	 * We need to verify that release_task() was not called and thus
	 * delayed_put_task_struct() can't run and drop the last reference
	 * before rcu_read_unlock(). We check task->sighand != NULL,
	 * but we can read the already freed and reused memory.
	 */
retry:
	task = rcu_dereference(*ptask);
	if (!task)
		return NULL;

	probe_kernel_address(&task->sighand, sighand);

	/*
	 * Pairs with atomic_dec_and_test() in put_task_struct(). If this task
	 * was already freed we can not miss the preceding update of this
	 * pointer.
	 */
	smp_rmb();
	if (unlikely(task != READ_ONCE(*ptask)))
		goto retry;

	/*
	 * We've re-checked that "task == *ptask", now we have two different
	 * cases:
	 *
	 * 1. This is actually the same task/task_struct. In this case
	 *    sighand != NULL tells us it is still alive.
	 *
	 * 2. This is another task which got the same memory for task_struct.
	 *    We can't know this of course, and we can not trust
	 *    sighand != NULL.
	 *
	 *    In this case we actually return a random value, but this is
	 *    correct.
	 *
	 *    If we return NULL - we can pretend that we actually noticed that
	 *    *ptask was updated when the previous task has exited. Or pretend
	 *    that probe_slab_address(&sighand) reads NULL.
	 *
	 *    If we return the new task (because sighand is not NULL for any
	 *    reason) - this is fine too. This (new) task can't go away before
	 *    another gp pass.
	 *
	 *    And note: We could even eliminate the false positive if re-read
	 *    task->sighand once again to avoid the falsely NULL. But this case
	 *    is very unlikely so we don't care.
	 */
	if (!sighand)
		return NULL;

	return task;
}

294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310
void rcuwait_wake_up(struct rcuwait *w)
{
	struct task_struct *task;

	rcu_read_lock();

	/*
	 * Order condition vs @task, such that everything prior to the load
	 * of @task is visible. This is the condition as to why the user called
	 * rcuwait_trywake() in the first place. Pairs with set_current_state()
	 * barrier (A) in rcuwait_wait_event().
	 *
	 *    WAIT                WAKE
	 *    [S] tsk = current	  [S] cond = true
	 *        MB (A)	      MB (B)
	 *    [L] cond		  [L] tsk
	 */
311
	smp_mb(); /* (B) */
312 313 314 315 316 317 318 319 320 321 322

	/*
	 * Avoid using task_rcu_dereference() magic as long as we are careful,
	 * see comment in rcuwait_wait_event() regarding ->exit_state.
	 */
	task = rcu_dereference(w->task);
	if (task)
		wake_up_process(task);
	rcu_read_unlock();
}

L
Linus Torvalds 已提交
323 324 325 326 327 328 329 330
/*
 * Determine if a process group is "orphaned", according to the POSIX
 * definition in 2.2.2.52.  Orphaned process groups are not to be affected
 * by terminal-generated stop signals.  Newly orphaned process groups are
 * to receive a SIGHUP and a SIGCONT.
 *
 * "I ask you, have you ever known what it is to be an orphan?"
 */
331 332
static int will_become_orphaned_pgrp(struct pid *pgrp,
					struct task_struct *ignored_task)
L
Linus Torvalds 已提交
333 334 335
{
	struct task_struct *p;

336
	do_each_pid_task(pgrp, PIDTYPE_PGID, p) {
337 338 339
		if ((p == ignored_task) ||
		    (p->exit_state && thread_group_empty(p)) ||
		    is_global_init(p->real_parent))
L
Linus Torvalds 已提交
340
			continue;
341

342
		if (task_pgrp(p->real_parent) != pgrp &&
343 344
		    task_session(p->real_parent) == task_session(p))
			return 0;
345
	} while_each_pid_task(pgrp, PIDTYPE_PGID, p);
346 347

	return 1;
L
Linus Torvalds 已提交
348 349
}

350
int is_current_pgrp_orphaned(void)
L
Linus Torvalds 已提交
351 352 353 354
{
	int retval;

	read_lock(&tasklist_lock);
355
	retval = will_become_orphaned_pgrp(task_pgrp(current), NULL);
L
Linus Torvalds 已提交
356 357 358 359 360
	read_unlock(&tasklist_lock);

	return retval;
}

361
static bool has_stopped_jobs(struct pid *pgrp)
L
Linus Torvalds 已提交
362 363 364
{
	struct task_struct *p;

365
	do_each_pid_task(pgrp, PIDTYPE_PGID, p) {
366 367
		if (p->signal->flags & SIGNAL_STOP_STOPPED)
			return true;
368
	} while_each_pid_task(pgrp, PIDTYPE_PGID, p);
369 370

	return false;
L
Linus Torvalds 已提交
371 372
}

373 374 375 376 377 378 379 380 381 382 383 384
/*
 * Check to see if any process groups have become orphaned as
 * a result of our exiting, and if they have any stopped jobs,
 * send them a SIGHUP and then a SIGCONT. (POSIX 3.2.2.2)
 */
static void
kill_orphaned_pgrp(struct task_struct *tsk, struct task_struct *parent)
{
	struct pid *pgrp = task_pgrp(tsk);
	struct task_struct *ignored_task = tsk;

	if (!parent)
385 386 387
		/* exit: our father is in a different pgrp than
		 * we are and we were the only connection outside.
		 */
388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403
		parent = tsk->real_parent;
	else
		/* reparent: our child is in a different pgrp than
		 * we are, and it was the only connection outside.
		 */
		ignored_task = NULL;

	if (task_pgrp(parent) != pgrp &&
	    task_session(parent) == task_session(tsk) &&
	    will_become_orphaned_pgrp(pgrp, ignored_task) &&
	    has_stopped_jobs(pgrp)) {
		__kill_pgrp_info(SIGHUP, SEND_SIG_PRIV, pgrp);
		__kill_pgrp_info(SIGCONT, SEND_SIG_PRIV, pgrp);
	}
}

O
Oleg Nesterov 已提交
404
#ifdef CONFIG_MEMCG
405
/*
406
 * A task is exiting.   If it owned this mm, find a new owner for the mm.
407 408 409 410 411 412
 */
void mm_update_next_owner(struct mm_struct *mm)
{
	struct task_struct *c, *g, *p = current;

retry:
413 414 415 416 417
	/*
	 * If the exiting or execing task is not the owner, it's
	 * someone else's problem.
	 */
	if (mm->owner != p)
418
		return;
419 420 421 422 423 424 425 426 427
	/*
	 * The current owner is exiting/execing and there are no other
	 * candidates.  Do not leave the mm pointing to a possibly
	 * freed task structure.
	 */
	if (atomic_read(&mm->mm_users) <= 1) {
		mm->owner = NULL;
		return;
	}
428 429 430 431 432 433 434 435 436 437 438 439 440

	read_lock(&tasklist_lock);
	/*
	 * Search in the children
	 */
	list_for_each_entry(c, &p->children, sibling) {
		if (c->mm == mm)
			goto assign_new_owner;
	}

	/*
	 * Search in the siblings
	 */
441
	list_for_each_entry(c, &p->real_parent->children, sibling) {
442 443 444 445 446
		if (c->mm == mm)
			goto assign_new_owner;
	}

	/*
447
	 * Search through everything else, we should not get here often.
448
	 */
449 450 451 452 453 454 455 456 457
	for_each_process(g) {
		if (g->flags & PF_KTHREAD)
			continue;
		for_each_thread(g, c) {
			if (c->mm == mm)
				goto assign_new_owner;
			if (c->mm)
				break;
		}
458
	}
459
	read_unlock(&tasklist_lock);
460 461 462
	/*
	 * We found no owner yet mm_users > 1: this implies that we are
	 * most likely racing with swapoff (try_to_unuse()) or /proc or
463
	 * ptrace or page migration (get_task_mm()).  Mark owner as NULL.
464 465
	 */
	mm->owner = NULL;
466 467 468 469 470 471 472 473 474 475
	return;

assign_new_owner:
	BUG_ON(c == p);
	get_task_struct(c);
	/*
	 * The task_lock protects c->mm from changing.
	 * We always want mm->owner->mm == mm
	 */
	task_lock(c);
476 477 478 479 480
	/*
	 * Delay read_unlock() till we have the task_lock()
	 * to ensure that c does not slip away underneath us
	 */
	read_unlock(&tasklist_lock);
481 482 483 484 485 486 487 488 489
	if (c->mm != mm) {
		task_unlock(c);
		put_task_struct(c);
		goto retry;
	}
	mm->owner = c;
	task_unlock(c);
	put_task_struct(c);
}
O
Oleg Nesterov 已提交
490
#endif /* CONFIG_MEMCG */
491

L
Linus Torvalds 已提交
492 493 494 495
/*
 * Turn us into a lazy TLB process if we
 * aren't already..
 */
496
static void exit_mm(void)
L
Linus Torvalds 已提交
497
{
498
	struct mm_struct *mm = current->mm;
499
	struct core_state *core_state;
L
Linus Torvalds 已提交
500

501
	mm_release(current, mm);
L
Linus Torvalds 已提交
502 503
	if (!mm)
		return;
504
	sync_mm_rss(mm);
L
Linus Torvalds 已提交
505 506
	/*
	 * Serialize with any possible pending coredump.
507
	 * We must hold mmap_sem around checking core_state
L
Linus Torvalds 已提交
508
	 * and clearing tsk->mm.  The core-inducing thread
509
	 * will increment ->nr_threads for each thread in the
L
Linus Torvalds 已提交
510 511 512
	 * group with ->mm != NULL.
	 */
	down_read(&mm->mmap_sem);
513 514 515
	core_state = mm->core_state;
	if (core_state) {
		struct core_thread self;
516

L
Linus Torvalds 已提交
517 518
		up_read(&mm->mmap_sem);

519
		self.task = current;
520 521 522 523 524 525 526
		self.next = xchg(&core_state->dumper.next, &self);
		/*
		 * Implies mb(), the result of xchg() must be visible
		 * to core_state->dumper.
		 */
		if (atomic_dec_and_test(&core_state->nr_threads))
			complete(&core_state->startup);
L
Linus Torvalds 已提交
527

O
Oleg Nesterov 已提交
528
		for (;;) {
529
			set_current_state(TASK_UNINTERRUPTIBLE);
O
Oleg Nesterov 已提交
530 531
			if (!self.task) /* see coredump_finish() */
				break;
532
			freezable_schedule();
O
Oleg Nesterov 已提交
533
		}
534
		__set_current_state(TASK_RUNNING);
L
Linus Torvalds 已提交
535 536
		down_read(&mm->mmap_sem);
	}
V
Vegard Nossum 已提交
537
	mmgrab(mm);
538
	BUG_ON(mm != current->active_mm);
L
Linus Torvalds 已提交
539
	/* more a memory barrier than a real lock */
540 541
	task_lock(current);
	current->mm = NULL;
L
Linus Torvalds 已提交
542 543
	up_read(&mm->mmap_sem);
	enter_lazy_tlb(mm, current);
544
	task_unlock(current);
545
	mm_update_next_owner(mm);
L
Linus Torvalds 已提交
546
	mmput(mm);
547
	if (test_thread_flag(TIF_MEMDIE))
548
		exit_oom_victim();
L
Linus Torvalds 已提交
549 550
}

551 552 553 554 555 556 557 558 559 560 561
static struct task_struct *find_alive_thread(struct task_struct *p)
{
	struct task_struct *t;

	for_each_thread(p, t) {
		if (!(t->flags & PF_EXITING))
			return t;
	}
	return NULL;
}

562 563
static struct task_struct *find_child_reaper(struct task_struct *father,
						struct list_head *dead)
564 565 566 567 568
	__releases(&tasklist_lock)
	__acquires(&tasklist_lock)
{
	struct pid_namespace *pid_ns = task_active_pid_ns(father);
	struct task_struct *reaper = pid_ns->child_reaper;
569
	struct task_struct *p, *n;
570 571 572 573

	if (likely(reaper != father))
		return reaper;

574 575
	reaper = find_alive_thread(father);
	if (reaper) {
576 577 578 579 580 581 582 583 584
		pid_ns->child_reaper = reaper;
		return reaper;
	}

	write_unlock_irq(&tasklist_lock);
	if (unlikely(pid_ns == &init_pid_ns)) {
		panic("Attempted to kill init! exitcode=0x%08x\n",
			father->signal->group_exit_code ?: father->exit_code);
	}
585 586 587 588 589 590

	list_for_each_entry_safe(p, n, dead, ptrace_entry) {
		list_del_init(&p->ptrace_entry);
		release_task(p);
	}

591 592 593 594 595 596
	zap_pid_ns_processes(pid_ns);
	write_lock_irq(&tasklist_lock);

	return father;
}

L
Linus Torvalds 已提交
597
/*
598 599 600 601 602
 * When we die, we re-parent all our children, and try to:
 * 1. give them to another thread in our thread group, if such a member exists
 * 2. give it to the first ancestor process which prctl'd itself as a
 *    child_subreaper for its children (like a service manager)
 * 3. give it to the init process (PID 1) in our pid namespace
L
Linus Torvalds 已提交
603
 */
604 605
static struct task_struct *find_new_reaper(struct task_struct *father,
					   struct task_struct *child_reaper)
L
Linus Torvalds 已提交
606
{
607
	struct task_struct *thread, *reaper;
L
Linus Torvalds 已提交
608

609 610
	thread = find_alive_thread(father);
	if (thread)
611
		return thread;
L
Linus Torvalds 已提交
612

613
	if (father->signal->has_child_subreaper) {
614
		unsigned int ns_level = task_pid(father)->level;
615
		/*
616
		 * Find the first ->is_child_subreaper ancestor in our pid_ns.
617 618 619 620 621
		 * We can't check reaper != child_reaper to ensure we do not
		 * cross the namespaces, the exiting parent could be injected
		 * by setns() + fork().
		 * We check pid->level, this is slightly more efficient than
		 * task_active_pid_ns(reaper) != task_active_pid_ns(father).
622
		 */
623 624
		for (reaper = father->real_parent;
		     task_pid(reaper)->level == ns_level;
625
		     reaper = reaper->real_parent) {
626
			if (reaper == &init_task)
627 628 629
				break;
			if (!reaper->signal->is_child_subreaper)
				continue;
630 631 632
			thread = find_alive_thread(reaper);
			if (thread)
				return thread;
633
		}
L
Linus Torvalds 已提交
634
	}
635

636
	return child_reaper;
637 638
}

639 640 641
/*
* Any that need to be release_task'd are put on the @dead list.
 */
642
static void reparent_leader(struct task_struct *father, struct task_struct *p,
643 644
				struct list_head *dead)
{
645
	if (unlikely(p->exit_state == EXIT_DEAD))
646 647
		return;

648
	/* We don't want people slaying init. */
649 650 651
	p->exit_signal = SIGCHLD;

	/* If it has exited notify the new parent about this child's death. */
T
Tejun Heo 已提交
652
	if (!p->ptrace &&
653
	    p->exit_state == EXIT_ZOMBIE && thread_group_empty(p)) {
654
		if (do_notify_parent(p, p->exit_signal)) {
655
			p->exit_state = EXIT_DEAD;
656
			list_add(&p->ptrace_entry, dead);
657 658 659 660 661 662
		}
	}

	kill_orphaned_pgrp(p, father);
}

663 664 665 666 667 668 669 670 671 672
/*
 * This does two things:
 *
 * A.  Make init inherit all the child processes
 * B.  Check to see if any process groups have become orphaned
 *	as a result of our exiting, and if they have any stopped
 *	jobs, send them a SIGHUP and then a SIGCONT.  (POSIX 3.2.2.2)
 */
static void forget_original_parent(struct task_struct *father,
					struct list_head *dead)
L
Linus Torvalds 已提交
673
{
674
	struct task_struct *p, *t, *reaper;
675

676
	if (unlikely(!list_empty(&father->ptraced)))
677
		exit_ptrace(father, dead);
R
Roland McGrath 已提交
678

679
	/* Can drop and reacquire tasklist_lock */
680
	reaper = find_child_reaper(father, dead);
681
	if (list_empty(&father->children))
682
		return;
683 684

	reaper = find_new_reaper(father, reaper);
685
	list_for_each_entry(p, &father->children, sibling) {
686
		for_each_thread(p, t) {
687
			t->real_parent = reaper;
688 689
			BUG_ON((!t->ptrace) != (t->parent == father));
			if (likely(!t->ptrace))
690 691 692
				t->parent = t->real_parent;
			if (t->pdeath_signal)
				group_send_sig_info(t->pdeath_signal,
693 694
						    SEND_SIG_NOINFO, t,
						    PIDTYPE_TGID);
695
		}
696 697 698 699 700
		/*
		 * If this is a threaded reparent there is no need to
		 * notify anyone anything has happened.
		 */
		if (!same_thread_group(reaper, father))
701
			reparent_leader(father, p, dead);
L
Linus Torvalds 已提交
702
	}
703
	list_splice_tail_init(&father->children, &reaper->children);
L
Linus Torvalds 已提交
704 705 706 707 708 709
}

/*
 * Send signals to all our closest relatives so that they know
 * to properly mourn us..
 */
710
static void exit_notify(struct task_struct *tsk, int group_dead)
L
Linus Torvalds 已提交
711
{
712
	bool autoreap;
713
	unsigned long flags;
714 715
	struct task_struct *p, *n;
	LIST_HEAD(dead);
L
Linus Torvalds 已提交
716

717 718 719 720 721 722 723
retry:
	if (!write_trylock_irqsave(&tasklist_lock, flags)) {
		if (!in_atomic() && !irqs_disabled())
			cond_resched();
		goto retry;
	}

724 725
	forget_original_parent(tsk, &dead);

726 727
	if (group_dead)
		kill_orphaned_pgrp(tsk->group_leader, NULL);
L
Linus Torvalds 已提交
728

O
Oleg Nesterov 已提交
729 730 731 732 733 734 735 736 737 738 739 740
	if (unlikely(tsk->ptrace)) {
		int sig = thread_group_leader(tsk) &&
				thread_group_empty(tsk) &&
				!ptrace_reparented(tsk) ?
			tsk->exit_signal : SIGCHLD;
		autoreap = do_notify_parent(tsk, sig);
	} else if (thread_group_leader(tsk)) {
		autoreap = thread_group_empty(tsk) &&
			do_notify_parent(tsk, tsk->exit_signal);
	} else {
		autoreap = true;
	}
L
Linus Torvalds 已提交
741

742
	tsk->exit_state = autoreap ? EXIT_DEAD : EXIT_ZOMBIE;
743 744
	if (tsk->exit_state == EXIT_DEAD)
		list_add(&tsk->ptrace_entry, &dead);
L
Linus Torvalds 已提交
745

746 747
	/* mt-exec, de_thread() is waiting for group leader */
	if (unlikely(tsk->signal->notify_count < 0))
748
		wake_up_process(tsk->signal->group_exit_task);
L
Linus Torvalds 已提交
749 750
	write_unlock_irq(&tasklist_lock);

751 752 753 754
	list_for_each_entry_safe(p, n, &dead, ptrace_entry) {
		list_del_init(&p->ptrace_entry);
		release_task(p);
	}
L
Linus Torvalds 已提交
755 756
}

757 758 759 760 761 762 763
#ifdef CONFIG_DEBUG_STACK_USAGE
static void check_stack_usage(void)
{
	static DEFINE_SPINLOCK(low_water_lock);
	static int lowest_to_date = THREAD_SIZE;
	unsigned long free;

764
	free = stack_not_used(current);
765 766 767 768 769 770

	if (free >= lowest_to_date)
		return;

	spin_lock(&low_water_lock);
	if (free < lowest_to_date) {
771
		pr_info("%s (%d) used greatest stack depth: %lu bytes left\n",
772
			current->comm, task_pid_nr(current), free);
773 774 775 776 777 778 779 780
		lowest_to_date = free;
	}
	spin_unlock(&low_water_lock);
}
#else
static inline void check_stack_usage(void) {}
#endif

781
void __noreturn do_exit(long code)
L
Linus Torvalds 已提交
782 783 784 785 786
{
	struct task_struct *tsk = current;
	int group_dead;

	profile_task_exit(tsk);
D
Dmitry Vyukov 已提交
787
	kcov_task_exit(tsk);
L
Linus Torvalds 已提交
788

789
	WARN_ON(blk_needs_flush_plug(tsk));
790

L
Linus Torvalds 已提交
791 792 793 794 795
	if (unlikely(in_interrupt()))
		panic("Aiee, killing interrupt handler!");
	if (unlikely(!tsk->pid))
		panic("Attempted to kill the idle task!");

796 797 798 799 800 801 802 803 804
	/*
	 * If do_exit is called because this processes oopsed, it's possible
	 * that get_fs() was left as KERNEL_DS, so reset it to USER_DS before
	 * continuing. Amongst other possible reasons, this is to prevent
	 * mm_release()->clear_child_tid() from writing to a user-controlled
	 * kernel address.
	 */
	set_fs(USER_DS);

T
Tejun Heo 已提交
805
	ptrace_event(PTRACE_EVENT_EXIT, code);
L
Linus Torvalds 已提交
806

807 808
	validate_creds_for_do_exit(tsk);

809 810 811 812 813
	/*
	 * We're taking recursive faults here in do_exit. Safest is to just
	 * leave this task alone and wait for reboot.
	 */
	if (unlikely(tsk->flags & PF_EXITING)) {
814
		pr_alert("Fixing recursive fault but reboot is needed!\n");
815 816 817 818 819 820 821 822 823 824
		/*
		 * We can do this unlocked here. The futex code uses
		 * this flag just to verify whether the pi state
		 * cleanup has been done or not. In the worst case it
		 * loops once more. We pretend that the cleanup was
		 * done as there is no way to return. Either the
		 * OWNER_DIED bit is set by now or we push the blocked
		 * task into the wait for ever nirwana as well.
		 */
		tsk->flags |= PF_EXITPIDONE;
825 826 827 828
		set_current_state(TASK_UNINTERRUPTIBLE);
		schedule();
	}

O
Oleg Nesterov 已提交
829
	exit_signals(tsk);  /* sets PF_EXITING */
830
	/*
831 832
	 * Ensure that all new tsk->pi_lock acquisitions must observe
	 * PF_EXITING. Serializes against futex.c:attach_to_pi_owner().
833
	 */
834
	smp_mb();
835 836 837 838
	/*
	 * Ensure that we must observe the pi_state in exit_mm() ->
	 * mm_release() -> exit_pi_state_list().
	 */
839 840
	raw_spin_lock_irq(&tsk->pi_lock);
	raw_spin_unlock_irq(&tsk->pi_lock);
L
Linus Torvalds 已提交
841

842
	if (unlikely(in_atomic())) {
843 844 845
		pr_info("note: %s[%d] exited with preempt_count %d\n",
			current->comm, task_pid_nr(current),
			preempt_count());
846 847
		preempt_count_set(PREEMPT_ENABLED);
	}
L
Linus Torvalds 已提交
848

849 850 851
	/* sync mm's RSS info before statistics gathering */
	if (tsk->mm)
		sync_mm_rss(tsk->mm);
852
	acct_update_integrals(tsk);
L
Linus Torvalds 已提交
853
	group_dead = atomic_dec_and_test(&tsk->signal->live);
854
	if (group_dead) {
855
#ifdef CONFIG_POSIX_TIMERS
856
		hrtimer_cancel(&tsk->signal->real_timer);
857
		exit_itimers(tsk->signal);
858
#endif
J
Jiri Pirko 已提交
859 860
		if (tsk->mm)
			setmax_mm_hiwater_rss(&tsk->signal->maxrss, tsk->mm);
861
	}
862
	acct_collect(code, group_dead);
M
Miloslav Trmac 已提交
863 864
	if (group_dead)
		tty_audit_exit();
865
	audit_free(tsk);
866

867
	tsk->exit_code = code;
868
	taskstats_exit(tsk, group_dead);
869

870
	exit_mm();
L
Linus Torvalds 已提交
871

872
	if (group_dead)
873
		acct_process();
874 875
	trace_sched_process_exit(tsk);

L
Linus Torvalds 已提交
876
	exit_sem(tsk);
877
	exit_shm(tsk);
878 879
	exit_files(tsk);
	exit_fs(tsk);
880 881
	if (group_dead)
		disassociate_ctty(1);
882
	exit_task_namespaces(tsk);
883
	exit_task_work(tsk);
884
	exit_thread(tsk);
885 886 887 888 889 890 891 892 893

	/*
	 * Flush inherited counters to the parent - before the parent
	 * gets woken up by child-exit notifications.
	 *
	 * because of cgroup mode, must be called before cgroup_exit()
	 */
	perf_event_exit_task(tsk);

894
	sched_autogroup_exit_task(tsk);
895
	cgroup_exit(tsk);
L
Linus Torvalds 已提交
896

897 898 899
	/*
	 * FIXME: do that only when needed, using sched_exit tracepoint
	 */
900
	flush_ptrace_hw_breakpoint(tsk);
901

902
	exit_tasks_rcu_start();
903
	exit_notify(tsk, group_dead);
904
	proc_exit_connector(tsk);
905
	mpol_put_task_policy(tsk);
906
#ifdef CONFIG_FUTEX
907 908
	if (unlikely(current->pi_state_cache))
		kfree(current->pi_state_cache);
909
#endif
910
	/*
911
	 * Make sure we are holding no locks:
912
	 */
913
	debug_check_no_locks_held();
914 915 916 917 918 919
	/*
	 * We can do this unlocked here. The futex code uses this flag
	 * just to verify whether the pi state cleanup has been done
	 * or not. In the worst case it loops once more.
	 */
	tsk->flags |= PF_EXITPIDONE;
L
Linus Torvalds 已提交
920

921
	if (tsk->io_context)
922
		exit_io_context(tsk);
923

924
	if (tsk->splice_pipe)
925
		free_pipe_info(tsk->splice_pipe);
926

927 928 929
	if (tsk->task_frag.page)
		put_page(tsk->task_frag.page);

930 931
	validate_creds_for_do_exit(tsk);

932
	check_stack_usage();
C
Coywolf Qi Hunt 已提交
933
	preempt_disable();
934 935
	if (tsk->nr_dirtied)
		__this_cpu_add(dirty_throttle_leaks, tsk->nr_dirtied);
936
	exit_rcu();
937
	exit_tasks_rcu_finish();
938

939
	lockdep_free_task(tsk);
940
	do_task_dead();
L
Linus Torvalds 已提交
941
}
942 943
EXPORT_SYMBOL_GPL(do_exit);

944
void complete_and_exit(struct completion *comp, long code)
L
Linus Torvalds 已提交
945 946 947
{
	if (comp)
		complete(comp);
O
Oleg Nesterov 已提交
948

L
Linus Torvalds 已提交
949 950 951 952
	do_exit(code);
}
EXPORT_SYMBOL(complete_and_exit);

953
SYSCALL_DEFINE1(exit, int, error_code)
L
Linus Torvalds 已提交
954 955 956 957 958 959 960 961
{
	do_exit((error_code&0xff)<<8);
}

/*
 * Take down every thread in the group.  This is called by fatal signals
 * as well as by sys_exit_group (below).
 */
962
void
L
Linus Torvalds 已提交
963 964
do_group_exit(int exit_code)
{
965 966
	struct signal_struct *sig = current->signal;

L
Linus Torvalds 已提交
967 968
	BUG_ON(exit_code & 0x80); /* core dumps don't get here */

969 970
	if (signal_group_exit(sig))
		exit_code = sig->group_exit_code;
L
Linus Torvalds 已提交
971 972
	else if (!thread_group_empty(current)) {
		struct sighand_struct *const sighand = current->sighand;
973

L
Linus Torvalds 已提交
974
		spin_lock_irq(&sighand->siglock);
975
		if (signal_group_exit(sig))
L
Linus Torvalds 已提交
976 977 978 979
			/* Another thread got here before we took the lock.  */
			exit_code = sig->group_exit_code;
		else {
			sig->group_exit_code = exit_code;
980
			sig->flags = SIGNAL_GROUP_EXIT;
L
Linus Torvalds 已提交
981 982 983 984 985 986 987 988 989 990 991 992 993 994
			zap_other_threads(current);
		}
		spin_unlock_irq(&sighand->siglock);
	}

	do_exit(exit_code);
	/* NOTREACHED */
}

/*
 * this kills every thread in the thread group. Note that any externally
 * wait4()-ing process will get the correct exit code - even if this
 * thread is not the thread group leader.
 */
995
SYSCALL_DEFINE1(exit_group, int, error_code)
L
Linus Torvalds 已提交
996 997
{
	do_group_exit((error_code & 0xff) << 8);
998 999
	/* NOTREACHED */
	return 0;
L
Linus Torvalds 已提交
1000 1001
}

1002 1003 1004 1005 1006 1007 1008
struct waitid_info {
	pid_t pid;
	uid_t uid;
	int status;
	int cause;
};

1009 1010 1011
struct wait_opts {
	enum pid_type		wo_type;
	int			wo_flags;
1012
	struct pid		*wo_pid;
1013

1014
	struct waitid_info	*wo_info;
1015
	int			wo_stat;
1016
	struct rusage		*wo_rusage;
1017

1018
	wait_queue_entry_t		child_wait;
1019 1020 1021
	int			notask_error;
};

1022
static int eligible_pid(struct wait_opts *wo, struct task_struct *p)
L
Linus Torvalds 已提交
1023
{
1024 1025 1026
	return	wo->wo_type == PIDTYPE_MAX ||
		task_pid_type(p, wo->wo_type) == wo->wo_pid;
}
L
Linus Torvalds 已提交
1027

1028 1029
static int
eligible_child(struct wait_opts *wo, bool ptrace, struct task_struct *p)
1030 1031 1032
{
	if (!eligible_pid(wo, p))
		return 0;
1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049

	/*
	 * Wait for all children (clone and not) if __WALL is set or
	 * if it is traced by us.
	 */
	if (ptrace || (wo->wo_flags & __WALL))
		return 1;

	/*
	 * Otherwise, wait for clone children *only* if __WCLONE is set;
	 * otherwise, wait for non-clone children *only*.
	 *
	 * Note: a "clone" child here is one that reports to its parent
	 * using a signal other than SIGCHLD, or a non-leader thread which
	 * we can only see if it is traced by us.
	 */
	if ((p->exit_signal != SIGCHLD) ^ !!(wo->wo_flags & __WCLONE))
L
Linus Torvalds 已提交
1050 1051
		return 0;

1052
	return 1;
L
Linus Torvalds 已提交
1053 1054 1055 1056 1057 1058 1059 1060
}

/*
 * Handle sys_wait4 work for one task in state EXIT_ZOMBIE.  We hold
 * read_lock(&tasklist_lock) on entry.  If we return zero, we still hold
 * the lock and this task is uninteresting.  If we return nonzero, we have
 * released the lock and the system call should return.
 */
1061
static int wait_task_zombie(struct wait_opts *wo, struct task_struct *p)
L
Linus Torvalds 已提交
1062
{
1063
	int state, status;
1064
	pid_t pid = task_pid_vnr(p);
1065
	uid_t uid = from_kuid_munged(current_user_ns(), task_uid(p));
1066
	struct waitid_info *infop;
L
Linus Torvalds 已提交
1067

1068
	if (!likely(wo->wo_flags & WEXITED))
R
Roland McGrath 已提交
1069 1070
		return 0;

1071
	if (unlikely(wo->wo_flags & WNOWAIT)) {
1072
		status = p->exit_code;
L
Linus Torvalds 已提交
1073 1074
		get_task_struct(p);
		read_unlock(&tasklist_lock);
1075
		sched_annotate_sleep();
1076 1077
		if (wo->wo_rusage)
			getrusage(p, RUSAGE_BOTH, wo->wo_rusage);
A
Al Viro 已提交
1078
		put_task_struct(p);
1079
		goto out_info;
L
Linus Torvalds 已提交
1080 1081
	}
	/*
1082
	 * Move the task's state to DEAD/TRACE, only one thread can do this.
L
Linus Torvalds 已提交
1083
	 */
1084 1085
	state = (ptrace_reparented(p) && thread_group_leader(p)) ?
		EXIT_TRACE : EXIT_DEAD;
1086
	if (cmpxchg(&p->exit_state, EXIT_ZOMBIE, state) != EXIT_ZOMBIE)
L
Linus Torvalds 已提交
1087
		return 0;
1088 1089 1090 1091 1092
	/*
	 * We own this thread, nobody else can reap it.
	 */
	read_unlock(&tasklist_lock);
	sched_annotate_sleep();
1093

1094
	/*
1095
	 * Check thread_group_leader() to exclude the traced sub-threads.
1096
	 */
1097
	if (state == EXIT_DEAD && thread_group_leader(p)) {
1098 1099
		struct signal_struct *sig = p->signal;
		struct signal_struct *psig = current->signal;
J
Jiri Pirko 已提交
1100
		unsigned long maxrss;
1101
		u64 tgutime, tgstime;
1102

L
Linus Torvalds 已提交
1103 1104 1105 1106 1107 1108 1109 1110
		/*
		 * The resource counters for the group leader are in its
		 * own task_struct.  Those for dead threads in the group
		 * are in its signal_struct, as are those for the child
		 * processes it has previously reaped.  All these
		 * accumulate in the parent's signal_struct c* fields.
		 *
		 * We don't bother to take a lock here to protect these
1111 1112 1113 1114 1115 1116 1117
		 * p->signal fields because the whole thread group is dead
		 * and nobody can change them.
		 *
		 * psig->stats_lock also protects us from our sub-theads
		 * which can reap other children at the same time. Until
		 * we change k_getrusage()-like users to rely on this lock
		 * we have to take ->siglock as well.
1118
		 *
1119 1120 1121
		 * We use thread_group_cputime_adjusted() to get times for
		 * the thread group, which consolidates times for all threads
		 * in the group including the group leader.
L
Linus Torvalds 已提交
1122
		 */
1123
		thread_group_cputime_adjusted(p, &tgutime, &tgstime);
1124
		spin_lock_irq(&current->sighand->siglock);
1125
		write_seqlock(&psig->stats_lock);
1126 1127
		psig->cutime += tgutime + sig->cutime;
		psig->cstime += tgstime + sig->cstime;
1128
		psig->cgtime += task_gtime(p) + sig->gtime + sig->cgtime;
1129 1130 1131 1132 1133 1134 1135 1136
		psig->cmin_flt +=
			p->min_flt + sig->min_flt + sig->cmin_flt;
		psig->cmaj_flt +=
			p->maj_flt + sig->maj_flt + sig->cmaj_flt;
		psig->cnvcsw +=
			p->nvcsw + sig->nvcsw + sig->cnvcsw;
		psig->cnivcsw +=
			p->nivcsw + sig->nivcsw + sig->cnivcsw;
1137 1138 1139 1140 1141 1142
		psig->cinblock +=
			task_io_get_inblock(p) +
			sig->inblock + sig->cinblock;
		psig->coublock +=
			task_io_get_oublock(p) +
			sig->oublock + sig->coublock;
J
Jiri Pirko 已提交
1143 1144 1145
		maxrss = max(sig->maxrss, sig->cmaxrss);
		if (psig->cmaxrss < maxrss)
			psig->cmaxrss = maxrss;
1146 1147
		task_io_accounting_add(&psig->ioac, &p->ioac);
		task_io_accounting_add(&psig->ioac, &sig->ioac);
1148
		write_sequnlock(&psig->stats_lock);
1149
		spin_unlock_irq(&current->sighand->siglock);
L
Linus Torvalds 已提交
1150 1151
	}

1152 1153
	if (wo->wo_rusage)
		getrusage(p, RUSAGE_BOTH, wo->wo_rusage);
L
Linus Torvalds 已提交
1154 1155
	status = (p->signal->flags & SIGNAL_GROUP_EXIT)
		? p->signal->group_exit_code : p->exit_code;
1156
	wo->wo_stat = status;
1157

1158
	if (state == EXIT_TRACE) {
L
Linus Torvalds 已提交
1159
		write_lock_irq(&tasklist_lock);
1160 1161
		/* We dropped tasklist, ptracer could die and untrace */
		ptrace_unlink(p);
1162 1163 1164 1165 1166

		/* If parent wants a zombie, don't release it now */
		state = EXIT_ZOMBIE;
		if (do_notify_parent(p, p->exit_signal))
			state = EXIT_DEAD;
1167
		p->exit_state = state;
L
Linus Torvalds 已提交
1168 1169
		write_unlock_irq(&tasklist_lock);
	}
1170
	if (state == EXIT_DEAD)
L
Linus Torvalds 已提交
1171
		release_task(p);
1172

1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186
out_info:
	infop = wo->wo_info;
	if (infop) {
		if ((status & 0x7f) == 0) {
			infop->cause = CLD_EXITED;
			infop->status = status >> 8;
		} else {
			infop->cause = (status & 0x80) ? CLD_DUMPED : CLD_KILLED;
			infop->status = status & 0x7f;
		}
		infop->pid = pid;
		infop->uid = uid;
	}

1187
	return pid;
L
Linus Torvalds 已提交
1188 1189
}

1190 1191 1192
static int *task_stopped_code(struct task_struct *p, bool ptrace)
{
	if (ptrace) {
1193
		if (task_is_traced(p) && !(p->jobctl & JOBCTL_LISTENING))
1194 1195 1196 1197 1198 1199 1200 1201
			return &p->exit_code;
	} else {
		if (p->signal->flags & SIGNAL_STOP_STOPPED)
			return &p->signal->group_exit_code;
	}
	return NULL;
}

1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218
/**
 * wait_task_stopped - Wait for %TASK_STOPPED or %TASK_TRACED
 * @wo: wait options
 * @ptrace: is the wait for ptrace
 * @p: task to wait for
 *
 * Handle sys_wait4() work for %p in state %TASK_STOPPED or %TASK_TRACED.
 *
 * CONTEXT:
 * read_lock(&tasklist_lock), which is released if return value is
 * non-zero.  Also, grabs and releases @p->sighand->siglock.
 *
 * RETURNS:
 * 0 if wait condition didn't exist and search for other wait conditions
 * should continue.  Non-zero return, -errno on failure and @p's pid on
 * success, implies that tasklist_lock is released and wait condition
 * search should terminate.
L
Linus Torvalds 已提交
1219
 */
1220 1221
static int wait_task_stopped(struct wait_opts *wo,
				int ptrace, struct task_struct *p)
L
Linus Torvalds 已提交
1222
{
1223 1224
	struct waitid_info *infop;
	int exit_code, *p_code, why;
1225
	uid_t uid = 0; /* unneeded, required by compiler */
1226
	pid_t pid;
L
Linus Torvalds 已提交
1227

1228 1229 1230
	/*
	 * Traditionally we see ptrace'd stopped tasks regardless of options.
	 */
1231
	if (!ptrace && !(wo->wo_flags & WUNTRACED))
R
Roland McGrath 已提交
1232 1233
		return 0;

1234 1235 1236
	if (!task_stopped_code(p, ptrace))
		return 0;

1237 1238 1239
	exit_code = 0;
	spin_lock_irq(&p->sighand->siglock);

1240 1241
	p_code = task_stopped_code(p, ptrace);
	if (unlikely(!p_code))
1242 1243
		goto unlock_sig;

1244
	exit_code = *p_code;
1245 1246 1247
	if (!exit_code)
		goto unlock_sig;

1248
	if (!unlikely(wo->wo_flags & WNOWAIT))
1249
		*p_code = 0;
1250

1251
	uid = from_kuid_munged(current_user_ns(), task_uid(p));
1252 1253 1254
unlock_sig:
	spin_unlock_irq(&p->sighand->siglock);
	if (!exit_code)
L
Linus Torvalds 已提交
1255 1256 1257 1258 1259 1260 1261 1262 1263 1264
		return 0;

	/*
	 * Now we are pretty sure this task is interesting.
	 * Make sure it doesn't get reaped out from under us while we
	 * give up the lock and then examine it below.  We don't want to
	 * keep holding onto the tasklist_lock while we call getrusage and
	 * possibly take page faults for user memory.
	 */
	get_task_struct(p);
1265
	pid = task_pid_vnr(p);
R
Roland McGrath 已提交
1266
	why = ptrace ? CLD_TRAPPED : CLD_STOPPED;
L
Linus Torvalds 已提交
1267
	read_unlock(&tasklist_lock);
1268
	sched_annotate_sleep();
1269 1270
	if (wo->wo_rusage)
		getrusage(p, RUSAGE_BOTH, wo->wo_rusage);
A
Al Viro 已提交
1271
	put_task_struct(p);
L
Linus Torvalds 已提交
1272

A
Al Viro 已提交
1273 1274
	if (likely(!(wo->wo_flags & WNOWAIT)))
		wo->wo_stat = (exit_code << 8) | 0x7f;
L
Linus Torvalds 已提交
1275

1276
	infop = wo->wo_info;
1277 1278 1279 1280 1281 1282 1283
	if (infop) {
		infop->cause = why;
		infop->status = exit_code;
		infop->pid = pid;
		infop->uid = uid;
	}
	return pid;
L
Linus Torvalds 已提交
1284 1285 1286 1287 1288 1289 1290 1291
}

/*
 * Handle do_wait work for one task in a live, non-stopped state.
 * read_lock(&tasklist_lock) on entry.  If we return zero, we still hold
 * the lock and this task is uninteresting.  If we return nonzero, we have
 * released the lock and the system call should return.
 */
1292
static int wait_task_continued(struct wait_opts *wo, struct task_struct *p)
L
Linus Torvalds 已提交
1293
{
A
Al Viro 已提交
1294
	struct waitid_info *infop;
L
Linus Torvalds 已提交
1295 1296 1297
	pid_t pid;
	uid_t uid;

1298
	if (!unlikely(wo->wo_flags & WCONTINUED))
R
Roland McGrath 已提交
1299 1300
		return 0;

L
Linus Torvalds 已提交
1301 1302 1303 1304 1305 1306 1307 1308 1309
	if (!(p->signal->flags & SIGNAL_STOP_CONTINUED))
		return 0;

	spin_lock_irq(&p->sighand->siglock);
	/* Re-check with the lock held.  */
	if (!(p->signal->flags & SIGNAL_STOP_CONTINUED)) {
		spin_unlock_irq(&p->sighand->siglock);
		return 0;
	}
1310
	if (!unlikely(wo->wo_flags & WNOWAIT))
L
Linus Torvalds 已提交
1311
		p->signal->flags &= ~SIGNAL_STOP_CONTINUED;
1312
	uid = from_kuid_munged(current_user_ns(), task_uid(p));
L
Linus Torvalds 已提交
1313 1314
	spin_unlock_irq(&p->sighand->siglock);

1315
	pid = task_pid_vnr(p);
L
Linus Torvalds 已提交
1316 1317
	get_task_struct(p);
	read_unlock(&tasklist_lock);
1318
	sched_annotate_sleep();
1319 1320
	if (wo->wo_rusage)
		getrusage(p, RUSAGE_BOTH, wo->wo_rusage);
A
Al Viro 已提交
1321
	put_task_struct(p);
L
Linus Torvalds 已提交
1322

A
Al Viro 已提交
1323 1324
	infop = wo->wo_info;
	if (!infop) {
1325
		wo->wo_stat = 0xffff;
L
Linus Torvalds 已提交
1326
	} else {
A
Al Viro 已提交
1327 1328 1329 1330
		infop->cause = CLD_CONTINUED;
		infop->pid = pid;
		infop->uid = uid;
		infop->status = SIGCONT;
L
Linus Torvalds 已提交
1331
	}
A
Al Viro 已提交
1332
	return pid;
L
Linus Torvalds 已提交
1333 1334
}

R
Roland McGrath 已提交
1335 1336 1337
/*
 * Consider @p for a wait by @parent.
 *
1338
 * -ECHILD should be in ->notask_error before the first call.
R
Roland McGrath 已提交
1339 1340
 * Returns nonzero for a final return, when we have unlocked tasklist_lock.
 * Returns zero if the search for a child should continue;
1341
 * then ->notask_error is 0 if @p is an eligible child,
1342
 * or still -ECHILD.
R
Roland McGrath 已提交
1343
 */
1344 1345
static int wait_consider_task(struct wait_opts *wo, int ptrace,
				struct task_struct *p)
R
Roland McGrath 已提交
1346
{
1347 1348 1349 1350 1351
	/*
	 * We can race with wait_task_zombie() from another thread.
	 * Ensure that EXIT_ZOMBIE -> EXIT_DEAD/EXIT_TRACE transition
	 * can't confuse the checks below.
	 */
1352
	int exit_state = READ_ONCE(p->exit_state);
1353 1354
	int ret;

1355
	if (unlikely(exit_state == EXIT_DEAD))
1356 1357
		return 0;

1358
	ret = eligible_child(wo, ptrace, p);
1359
	if (!ret)
R
Roland McGrath 已提交
1360 1361
		return ret;

1362
	if (unlikely(exit_state == EXIT_TRACE)) {
1363
		/*
1364 1365
		 * ptrace == 0 means we are the natural parent. In this case
		 * we should clear notask_error, debugger will notify us.
1366
		 */
1367
		if (likely(!ptrace))
1368
			wo->notask_error = 0;
1369
		return 0;
1370
	}
1371

1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387
	if (likely(!ptrace) && unlikely(p->ptrace)) {
		/*
		 * If it is traced by its real parent's group, just pretend
		 * the caller is ptrace_do_wait() and reap this child if it
		 * is zombie.
		 *
		 * This also hides group stop state from real parent; otherwise
		 * a single stop can be reported twice as group and ptrace stop.
		 * If a ptracer wants to distinguish these two events for its
		 * own children it should create a separate process which takes
		 * the role of real parent.
		 */
		if (!ptrace_reparented(p))
			ptrace = 1;
	}

1388
	/* slay zombie? */
1389
	if (exit_state == EXIT_ZOMBIE) {
1390
		/* we don't reap group leaders with subthreads */
1391 1392 1393 1394 1395 1396 1397 1398 1399
		if (!delay_group_leader(p)) {
			/*
			 * A zombie ptracee is only visible to its ptracer.
			 * Notification and reaping will be cascaded to the
			 * real parent when the ptracer detaches.
			 */
			if (unlikely(ptrace) || likely(!p->ptrace))
				return wait_task_zombie(wo, p);
		}
R
Roland McGrath 已提交
1400

R
Roland McGrath 已提交
1401
		/*
1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426
		 * Allow access to stopped/continued state via zombie by
		 * falling through.  Clearing of notask_error is complex.
		 *
		 * When !@ptrace:
		 *
		 * If WEXITED is set, notask_error should naturally be
		 * cleared.  If not, subset of WSTOPPED|WCONTINUED is set,
		 * so, if there are live subthreads, there are events to
		 * wait for.  If all subthreads are dead, it's still safe
		 * to clear - this function will be called again in finite
		 * amount time once all the subthreads are released and
		 * will then return without clearing.
		 *
		 * When @ptrace:
		 *
		 * Stopped state is per-task and thus can't change once the
		 * target task dies.  Only continued and exited can happen.
		 * Clear notask_error if WCONTINUED | WEXITED.
		 */
		if (likely(!ptrace) || (wo->wo_flags & (WCONTINUED | WEXITED)))
			wo->notask_error = 0;
	} else {
		/*
		 * @p is alive and it's gonna stop, continue or exit, so
		 * there always is something to wait for.
R
Roland McGrath 已提交
1427
		 */
1428
		wo->notask_error = 0;
R
Roland McGrath 已提交
1429 1430
	}

R
Roland McGrath 已提交
1431
	/*
1432 1433
	 * Wait for stopped.  Depending on @ptrace, different stopped state
	 * is used and the two don't interact with each other.
R
Roland McGrath 已提交
1434
	 */
1435 1436 1437
	ret = wait_task_stopped(wo, ptrace, p);
	if (ret)
		return ret;
R
Roland McGrath 已提交
1438 1439

	/*
1440 1441 1442
	 * Wait for continued.  There's only one continued state and the
	 * ptracer can consume it which can confuse the real parent.  Don't
	 * use WCONTINUED from ptracer.  You don't need or want it.
R
Roland McGrath 已提交
1443
	 */
1444
	return wait_task_continued(wo, p);
R
Roland McGrath 已提交
1445 1446 1447 1448 1449
}

/*
 * Do the work of do_wait() for one thread in the group, @tsk.
 *
1450
 * -ECHILD should be in ->notask_error before the first call.
R
Roland McGrath 已提交
1451 1452
 * Returns nonzero for a final return, when we have unlocked tasklist_lock.
 * Returns zero if the search for a child should continue; then
1453
 * ->notask_error is 0 if there were any eligible children,
1454
 * or still -ECHILD.
R
Roland McGrath 已提交
1455
 */
1456
static int do_wait_thread(struct wait_opts *wo, struct task_struct *tsk)
R
Roland McGrath 已提交
1457 1458 1459 1460
{
	struct task_struct *p;

	list_for_each_entry(p, &tsk->children, sibling) {
1461
		int ret = wait_consider_task(wo, 0, p);
1462

1463 1464
		if (ret)
			return ret;
R
Roland McGrath 已提交
1465 1466 1467 1468 1469
	}

	return 0;
}

1470
static int ptrace_do_wait(struct wait_opts *wo, struct task_struct *tsk)
R
Roland McGrath 已提交
1471 1472 1473
{
	struct task_struct *p;

R
Roland McGrath 已提交
1474
	list_for_each_entry(p, &tsk->ptraced, ptrace_entry) {
1475
		int ret = wait_consider_task(wo, 1, p);
1476

R
Roland McGrath 已提交
1477
		if (ret)
R
Roland McGrath 已提交
1478 1479 1480 1481 1482 1483
			return ret;
	}

	return 0;
}

1484
static int child_wait_callback(wait_queue_entry_t *wait, unsigned mode,
1485 1486 1487 1488 1489 1490
				int sync, void *key)
{
	struct wait_opts *wo = container_of(wait, struct wait_opts,
						child_wait);
	struct task_struct *p = key;

1491
	if (!eligible_pid(wo, p))
1492 1493
		return 0;

1494 1495 1496
	if ((wo->wo_flags & __WNOTHREAD) && wait->private != p->parent)
		return 0;

1497 1498 1499
	return default_wake_function(wait, mode, sync, key);
}

1500 1501
void __wake_up_parent(struct task_struct *p, struct task_struct *parent)
{
1502 1503
	__wake_up_sync_key(&parent->signal->wait_chldexit,
				TASK_INTERRUPTIBLE, 1, p);
1504 1505
}

1506
static long do_wait(struct wait_opts *wo)
L
Linus Torvalds 已提交
1507 1508
{
	struct task_struct *tsk;
R
Roland McGrath 已提交
1509
	int retval;
L
Linus Torvalds 已提交
1510

1511
	trace_sched_process_wait(wo->wo_pid);
1512

1513 1514 1515
	init_waitqueue_func_entry(&wo->child_wait, child_wait_callback);
	wo->child_wait.private = current;
	add_wait_queue(&current->signal->wait_chldexit, &wo->child_wait);
L
Linus Torvalds 已提交
1516
repeat:
R
Roland McGrath 已提交
1517
	/*
F
Frans Klaver 已提交
1518
	 * If there is nothing that can match our criteria, just get out.
1519 1520 1521
	 * We will clear ->notask_error to zero if we see any child that
	 * might later match our criteria, even if we are not able to reap
	 * it yet.
R
Roland McGrath 已提交
1522
	 */
1523
	wo->notask_error = -ECHILD;
1524 1525
	if ((wo->wo_type < PIDTYPE_MAX) &&
	   (!wo->wo_pid || hlist_empty(&wo->wo_pid->tasks[wo->wo_type])))
1526
		goto notask;
E
Eric W. Biederman 已提交
1527

1528
	set_current_state(TASK_INTERRUPTIBLE);
L
Linus Torvalds 已提交
1529 1530 1531
	read_lock(&tasklist_lock);
	tsk = current;
	do {
1532 1533 1534
		retval = do_wait_thread(wo, tsk);
		if (retval)
			goto end;
1535

1536 1537
		retval = ptrace_do_wait(wo, tsk);
		if (retval)
R
Roland McGrath 已提交
1538 1539
			goto end;

1540
		if (wo->wo_flags & __WNOTHREAD)
L
Linus Torvalds 已提交
1541
			break;
1542
	} while_each_thread(current, tsk);
L
Linus Torvalds 已提交
1543
	read_unlock(&tasklist_lock);
O
Oleg Nesterov 已提交
1544

1545
notask:
1546 1547
	retval = wo->notask_error;
	if (!retval && !(wo->wo_flags & WNOHANG)) {
L
Linus Torvalds 已提交
1548
		retval = -ERESTARTSYS;
R
Roland McGrath 已提交
1549 1550 1551 1552
		if (!signal_pending(current)) {
			schedule();
			goto repeat;
		}
L
Linus Torvalds 已提交
1553 1554
	}
end:
1555
	__set_current_state(TASK_RUNNING);
1556
	remove_wait_queue(&current->signal->wait_chldexit, &wo->child_wait);
L
Linus Torvalds 已提交
1557 1558 1559
	return retval;
}

1560
static long kernel_waitid(int which, pid_t upid, struct waitid_info *infop,
1561
			  int options, struct rusage *ru)
L
Linus Torvalds 已提交
1562
{
1563
	struct wait_opts wo;
E
Eric W. Biederman 已提交
1564 1565
	struct pid *pid = NULL;
	enum pid_type type;
L
Linus Torvalds 已提交
1566 1567
	long ret;

1568 1569
	if (options & ~(WNOHANG|WNOWAIT|WEXITED|WSTOPPED|WCONTINUED|
			__WNOTHREAD|__WCLONE|__WALL))
L
Linus Torvalds 已提交
1570 1571 1572 1573 1574 1575
		return -EINVAL;
	if (!(options & (WEXITED|WSTOPPED|WCONTINUED)))
		return -EINVAL;

	switch (which) {
	case P_ALL:
E
Eric W. Biederman 已提交
1576
		type = PIDTYPE_MAX;
L
Linus Torvalds 已提交
1577 1578
		break;
	case P_PID:
E
Eric W. Biederman 已提交
1579 1580
		type = PIDTYPE_PID;
		if (upid <= 0)
L
Linus Torvalds 已提交
1581 1582 1583
			return -EINVAL;
		break;
	case P_PGID:
E
Eric W. Biederman 已提交
1584 1585
		type = PIDTYPE_PGID;
		if (upid <= 0)
L
Linus Torvalds 已提交
1586 1587 1588 1589 1590 1591
			return -EINVAL;
		break;
	default:
		return -EINVAL;
	}

E
Eric W. Biederman 已提交
1592 1593
	if (type < PIDTYPE_MAX)
		pid = find_get_pid(upid);
1594 1595 1596 1597 1598 1599 1600

	wo.wo_type	= type;
	wo.wo_pid	= pid;
	wo.wo_flags	= options;
	wo.wo_info	= infop;
	wo.wo_rusage	= ru;
	ret = do_wait(&wo);
1601

E
Eric W. Biederman 已提交
1602
	put_pid(pid);
L
Linus Torvalds 已提交
1603 1604 1605
	return ret;
}

1606 1607 1608 1609
SYSCALL_DEFINE5(waitid, int, which, pid_t, upid, struct siginfo __user *,
		infop, int, options, struct rusage __user *, ru)
{
	struct rusage r;
1610 1611
	struct waitid_info info = {.status = 0};
	long err = kernel_waitid(which, upid, &info, options, ru ? &r : NULL);
A
Al Viro 已提交
1612
	int signo = 0;
A
Al Viro 已提交
1613

A
Al Viro 已提交
1614 1615 1616
	if (err > 0) {
		signo = SIGCHLD;
		err = 0;
1617 1618 1619
		if (ru && copy_to_user(ru, &r, sizeof(struct rusage)))
			return -EFAULT;
	}
1620 1621 1622
	if (!infop)
		return err;

1623
	if (!user_access_begin(infop, sizeof(*infop)))
1624
		return -EFAULT;
1625

A
Al Viro 已提交
1626
	unsafe_put_user(signo, &infop->si_signo, Efault);
1627
	unsafe_put_user(0, &infop->si_errno, Efault);
1628
	unsafe_put_user(info.cause, &infop->si_code, Efault);
1629 1630 1631 1632
	unsafe_put_user(info.pid, &infop->si_pid, Efault);
	unsafe_put_user(info.uid, &infop->si_uid, Efault);
	unsafe_put_user(info.status, &infop->si_status, Efault);
	user_access_end();
1633
	return err;
1634 1635 1636
Efault:
	user_access_end();
	return -EFAULT;
1637 1638
}

A
Al Viro 已提交
1639 1640
long kernel_wait4(pid_t upid, int __user *stat_addr, int options,
		  struct rusage *ru)
L
Linus Torvalds 已提交
1641
{
1642
	struct wait_opts wo;
E
Eric W. Biederman 已提交
1643 1644
	struct pid *pid = NULL;
	enum pid_type type;
L
Linus Torvalds 已提交
1645 1646 1647 1648 1649
	long ret;

	if (options & ~(WNOHANG|WUNTRACED|WCONTINUED|
			__WNOTHREAD|__WCLONE|__WALL))
		return -EINVAL;
E
Eric W. Biederman 已提交
1650

1651 1652 1653 1654
	/* -INT_MIN is not defined */
	if (upid == INT_MIN)
		return -ESRCH;

E
Eric W. Biederman 已提交
1655 1656 1657 1658 1659 1660 1661
	if (upid == -1)
		type = PIDTYPE_MAX;
	else if (upid < 0) {
		type = PIDTYPE_PGID;
		pid = find_get_pid(-upid);
	} else if (upid == 0) {
		type = PIDTYPE_PGID;
1662
		pid = get_task_pid(current, PIDTYPE_PGID);
E
Eric W. Biederman 已提交
1663 1664 1665 1666 1667
	} else /* upid > 0 */ {
		type = PIDTYPE_PID;
		pid = find_get_pid(upid);
	}

1668 1669 1670 1671
	wo.wo_type	= type;
	wo.wo_pid	= pid;
	wo.wo_flags	= options | WEXITED;
	wo.wo_info	= NULL;
1672
	wo.wo_stat	= 0;
1673 1674
	wo.wo_rusage	= ru;
	ret = do_wait(&wo);
E
Eric W. Biederman 已提交
1675
	put_pid(pid);
1676 1677
	if (ret > 0 && stat_addr && put_user(wo.wo_stat, stat_addr))
		ret = -EFAULT;
L
Linus Torvalds 已提交
1678 1679 1680 1681

	return ret;
}

1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694
SYSCALL_DEFINE4(wait4, pid_t, upid, int __user *, stat_addr,
		int, options, struct rusage __user *, ru)
{
	struct rusage r;
	long err = kernel_wait4(upid, stat_addr, options, ru ? &r : NULL);

	if (err > 0) {
		if (ru && copy_to_user(ru, &r, sizeof(struct rusage)))
			return -EFAULT;
	}
	return err;
}

L
Linus Torvalds 已提交
1695 1696 1697 1698 1699 1700
#ifdef __ARCH_WANT_SYS_WAITPID

/*
 * sys_waitpid() remains for compatibility. waitpid() should be
 * implemented by calling sys_wait4() from libc.a.
 */
1701
SYSCALL_DEFINE3(waitpid, pid_t, pid, int __user *, stat_addr, int, options)
L
Linus Torvalds 已提交
1702
{
1703
	return kernel_wait4(pid, stat_addr, options, NULL);
L
Linus Torvalds 已提交
1704 1705 1706
}

#endif
1707 1708 1709 1710 1711 1712 1713 1714

#ifdef CONFIG_COMPAT
COMPAT_SYSCALL_DEFINE4(wait4,
	compat_pid_t, pid,
	compat_uint_t __user *, stat_addr,
	int, options,
	struct compat_rusage __user *, ru)
{
1715 1716 1717 1718 1719
	struct rusage r;
	long err = kernel_wait4(pid, stat_addr, options, ru ? &r : NULL);
	if (err > 0) {
		if (ru && put_compat_rusage(&r, ru))
			return -EFAULT;
1720
	}
1721
	return err;
1722 1723 1724 1725 1726 1727 1728 1729
}

COMPAT_SYSCALL_DEFINE5(waitid,
		int, which, compat_pid_t, pid,
		struct compat_siginfo __user *, infop, int, options,
		struct compat_rusage __user *, uru)
{
	struct rusage ru;
1730 1731
	struct waitid_info info = {.status = 0};
	long err = kernel_waitid(which, pid, &info, options, uru ? &ru : NULL);
A
Al Viro 已提交
1732 1733 1734 1735
	int signo = 0;
	if (err > 0) {
		signo = SIGCHLD;
		err = 0;
A
Al Viro 已提交
1736 1737 1738 1739 1740 1741 1742 1743 1744
		if (uru) {
			/* kernel_waitid() overwrites everything in ru */
			if (COMPAT_USE_64BIT_TIME)
				err = copy_to_user(uru, &ru, sizeof(ru));
			else
				err = put_compat_rusage(&ru, uru);
			if (err)
				return -EFAULT;
		}
1745 1746
	}

1747 1748 1749
	if (!infop)
		return err;

1750
	if (!user_access_begin(infop, sizeof(*infop)))
1751
		return -EFAULT;
1752

A
Al Viro 已提交
1753
	unsafe_put_user(signo, &infop->si_signo, Efault);
1754
	unsafe_put_user(0, &infop->si_errno, Efault);
1755
	unsafe_put_user(info.cause, &infop->si_code, Efault);
1756 1757 1758 1759
	unsafe_put_user(info.pid, &infop->si_pid, Efault);
	unsafe_put_user(info.uid, &infop->si_uid, Efault);
	unsafe_put_user(info.status, &infop->si_status, Efault);
	user_access_end();
1760
	return err;
1761 1762 1763
Efault:
	user_access_end();
	return -EFAULT;
1764 1765
}
#endif
S
Sudip Mukherjee 已提交
1766 1767 1768 1769 1770 1771 1772 1773

__weak void abort(void)
{
	BUG();

	/* if that doesn't kill us, halt */
	panic("Oops failed to kill thread");
}
1774
EXPORT_SYMBOL(abort);
1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884

static int force_reap_process(int ptrace, struct task_struct *tsk)
{
	struct wait_opts wo;
	struct pid *pid = get_task_pid(tsk, PIDTYPE_PID);
	int ret;

	memset(&wo, 0, sizeof(wo));
	wo.wo_type = PIDTYPE_PID;
	wo.wo_pid = pid;
	wo.wo_flags = WEXITED;

	ret = wait_consider_task(&wo, ptrace, tsk);
	put_pid(pid);

	return ret;
}


static void exit_zombie_process(struct task_struct *tsk)
{
	int ret;

	read_lock(&tasklist_lock);
	ret = force_reap_process(0, tsk);
	if (ret)
		goto out;
	read_unlock(&tasklist_lock);
out:
	return;
}

extern int pid_max;
#define LEN	8
static ssize_t reclaim_zombie_process_write(struct file *file,
					const char __user *ubuf,
					size_t count, loff_t *ppos)
{
	int ret;
	unsigned int pid;
	struct task_struct *tsk;
	char buf[LEN];

	memset(buf, 0, LEN);
	if (count > LEN - 1) {
		ret = -EINVAL;
		goto out;
	}

	if (copy_from_user(buf, ubuf, count)) {
		ret = -EFAULT;
		goto out;
	}

	ret = kstrtoint(strstrip(buf), 10, &pid);
	if (ret)
		goto out;

	if (pid > pid_max) {
		ret = -EINVAL;
		goto out;
	}

	/* make sure the task status is zombie. */
	rcu_read_lock();
	tsk = find_task_by_pid_ns(pid, task_active_pid_ns(current));
	if (tsk) {
		get_task_struct(tsk);
		if (READ_ONCE(tsk->exit_state) != EXIT_ZOMBIE) {
			rcu_read_unlock();
			put_task_struct(tsk);
			ret = -EINVAL;
			goto out;
		}
	} else {
		rcu_read_unlock();
		ret = -EINVAL;
		goto out;
	}
	rcu_read_unlock();
	exit_zombie_process(tsk);
	put_task_struct(tsk);
out:
	return ret == 0 ? count : ret;
}

const struct file_operations  reclaim_zombie_process_fops = {
	.owner           = THIS_MODULE,
	.llseek		 = noop_llseek,
	.write		 = reclaim_zombie_process_write,
};

static int __init reclaim_zombie_process_init(void)
{
	struct proc_dir_entry *rzp_dir, *rzp_file;

	rzp_dir = proc_mkdir("toolkit", NULL);
	if (!rzp_dir)
		return -ENOMEM;

	rzp_file = proc_create("reclaim_zombie_process", 0200, rzp_dir,
						&reclaim_zombie_process_fops);
	if (!rzp_file) {
		remove_proc_entry("toolkit", NULL);
		return -ENOMEM;
	}

	return 0;
}
late_initcall(reclaim_zombie_process_init);