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

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

#include <asm/uaccess.h>
#include <asm/unistd.h>
#include <asm/pgtable.h>
#include <asm/mmu_context.h>

A
Adrian Bunk 已提交
60 61
static void exit_mm(struct task_struct * tsk);

62
static void __unhash_process(struct task_struct *p, bool group_dead)
L
Linus Torvalds 已提交
63 64 65
{
	nr_threads--;
	detach_pid(p, PIDTYPE_PID);
66
	if (group_dead) {
L
Linus Torvalds 已提交
67 68
		detach_pid(p, PIDTYPE_PGID);
		detach_pid(p, PIDTYPE_SID);
69

70
		list_del_rcu(&p->tasks);
71
		list_del_init(&p->sibling);
72
		__this_cpu_dec(process_counts);
L
Linus Torvalds 已提交
73
	}
O
Oleg Nesterov 已提交
74
	list_del_rcu(&p->thread_group);
L
Linus Torvalds 已提交
75 76
}

77 78 79 80 81 82
/*
 * This function expects the tasklist_lock write-locked.
 */
static void __exit_signal(struct task_struct *tsk)
{
	struct signal_struct *sig = tsk->signal;
83
	bool group_dead = thread_group_leader(tsk);
84
	struct sighand_struct *sighand;
85
	struct tty_struct *uninitialized_var(tty);
86

87 88
	sighand = rcu_dereference_check(tsk->sighand,
					rcu_read_lock_held() ||
89
					lockdep_tasklist_lock_is_held());
90 91 92
	spin_lock(&sighand->siglock);

	posix_cpu_timers_exit(tsk);
93
	if (group_dead) {
94
		posix_cpu_timers_exit_group(tsk);
95 96
		tty = sig->tty;
		sig->tty = NULL;
97
	} else {
98 99 100 101 102 103 104 105
		/*
		 * 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);

106 107 108 109
		/*
		 * If there is any task waiting for the group exit
		 * then notify it:
		 */
110
		if (sig->notify_count > 0 && !--sig->notify_count)
111
			wake_up_process(sig->group_exit_task);
112

113 114 115 116 117 118 119 120 121 122 123 124
		if (tsk == sig->curr_target)
			sig->curr_target = next_thread(tsk);
		/*
		 * Accumulate here the counters for all threads but the
		 * group leader as they die, so they can be added into
		 * the process-wide totals when those are taken.
		 * The group leader stays around as a zombie as long
		 * as there are other threads.  When it gets reaped,
		 * the exit.c code will add its counts into these totals.
		 * We won't ever get here for the group leader, since it
		 * will have been the last reference on the signal_struct.
		 */
125 126
		sig->utime = cputime_add(sig->utime, tsk->utime);
		sig->stime = cputime_add(sig->stime, tsk->stime);
H
Hidetoshi Seto 已提交
127
		sig->gtime = cputime_add(sig->gtime, tsk->gtime);
128 129 130 131
		sig->min_flt += tsk->min_flt;
		sig->maj_flt += tsk->maj_flt;
		sig->nvcsw += tsk->nvcsw;
		sig->nivcsw += tsk->nivcsw;
132 133
		sig->inblock += task_io_get_inblock(tsk);
		sig->oublock += task_io_get_oublock(tsk);
134
		task_io_accounting_add(&sig->ioac, &tsk->ioac);
135
		sig->sum_sched_runtime += tsk->se.sum_exec_runtime;
136 137
	}

138
	sig->nr_threads--;
139
	__unhash_process(tsk, group_dead);
140

141 142 143 144 145
	/*
	 * Do this under ->siglock, we can race with another thread
	 * doing sigqueue_free() if we have SIGQUEUE_PREALLOC signals.
	 */
	flush_sigqueue(&tsk->pending);
146
	tsk->sighand = NULL;
147 148
	spin_unlock(&sighand->siglock);

149
	__cleanup_sighand(sighand);
150
	clear_tsk_thread_flag(tsk,TIF_SIGPENDING);
151
	if (group_dead) {
152
		flush_sigqueue(&sig->shared_pending);
153
		tty_kref_put(tty);
154 155 156
	}
}

157 158
static void delayed_put_task_struct(struct rcu_head *rhp)
{
159 160
	struct task_struct *tsk = container_of(rhp, struct task_struct, rcu);

161
	perf_event_delayed_put(tsk);
162 163
	trace_sched_process_free(tsk);
	put_task_struct(tsk);
164 165
}

R
Roland McGrath 已提交
166

L
Linus Torvalds 已提交
167 168
void release_task(struct task_struct * p)
{
169
	struct task_struct *leader;
L
Linus Torvalds 已提交
170
	int zap_leader;
171
repeat:
172
	/* don't need to get the RCU readlock here - the process is dead and
173 174
	 * can't be modifying its own credentials. But shut RCU-lockdep up */
	rcu_read_lock();
175
	atomic_dec(&__task_cred(p)->user->processes);
176
	rcu_read_unlock();
177

178
	proc_flush_task(p);
179

L
Linus Torvalds 已提交
180
	write_lock_irq(&tasklist_lock);
T
Tejun Heo 已提交
181
	ptrace_release_task(p);
L
Linus Torvalds 已提交
182
	__exit_signal(p);
183

L
Linus Torvalds 已提交
184 185 186 187 188 189 190 191 192 193 194 195
	/*
	 * 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;
	if (leader != p && thread_group_empty(leader) && leader->exit_state == EXIT_ZOMBIE) {
		/*
		 * 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 已提交
196
		 */
197
		zap_leader = do_notify_parent(leader, leader->exit_signal);
R
Roland McGrath 已提交
198 199
		if (zap_leader)
			leader->exit_state = EXIT_DEAD;
L
Linus Torvalds 已提交
200 201 202 203
	}

	write_unlock_irq(&tasklist_lock);
	release_thread(p);
204
	call_rcu(&p->rcu, delayed_put_task_struct);
L
Linus Torvalds 已提交
205 206 207 208 209 210 211 212 213 214

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

/*
 * This checks not only the pgrp, but falls back on the pid if no
 * satisfactory pgrp is found. I dunno - gdb doesn't work correctly
 * without this...
215 216
 *
 * The caller must hold rcu lock or the tasklist lock.
L
Linus Torvalds 已提交
217
 */
218
struct pid *session_of_pgrp(struct pid *pgrp)
L
Linus Torvalds 已提交
219 220
{
	struct task_struct *p;
221
	struct pid *sid = NULL;
222

223
	p = pid_task(pgrp, PIDTYPE_PGID);
224
	if (p == NULL)
225
		p = pid_task(pgrp, PIDTYPE_PID);
226
	if (p != NULL)
227
		sid = task_session(p);
228

L
Linus Torvalds 已提交
229 230 231 232 233 234 235 236 237 238 239
	return sid;
}

/*
 * 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?"
 */
240
static int will_become_orphaned_pgrp(struct pid *pgrp, struct task_struct *ignored_task)
L
Linus Torvalds 已提交
241 242 243
{
	struct task_struct *p;

244
	do_each_pid_task(pgrp, PIDTYPE_PGID, p) {
245 246 247
		if ((p == ignored_task) ||
		    (p->exit_state && thread_group_empty(p)) ||
		    is_global_init(p->real_parent))
L
Linus Torvalds 已提交
248
			continue;
249

250
		if (task_pgrp(p->real_parent) != pgrp &&
251 252
		    task_session(p->real_parent) == task_session(p))
			return 0;
253
	} while_each_pid_task(pgrp, PIDTYPE_PGID, p);
254 255

	return 1;
L
Linus Torvalds 已提交
256 257
}

258
int is_current_pgrp_orphaned(void)
L
Linus Torvalds 已提交
259 260 261 262
{
	int retval;

	read_lock(&tasklist_lock);
263
	retval = will_become_orphaned_pgrp(task_pgrp(current), NULL);
L
Linus Torvalds 已提交
264 265 266 267 268
	read_unlock(&tasklist_lock);

	return retval;
}

269
static int has_stopped_jobs(struct pid *pgrp)
L
Linus Torvalds 已提交
270 271 272 273
{
	int retval = 0;
	struct task_struct *p;

274
	do_each_pid_task(pgrp, PIDTYPE_PGID, p) {
M
Matthew Wilcox 已提交
275
		if (!task_is_stopped(p))
L
Linus Torvalds 已提交
276 277 278
			continue;
		retval = 1;
		break;
279
	} while_each_pid_task(pgrp, PIDTYPE_PGID, p);
L
Linus Torvalds 已提交
280 281 282
	return retval;
}

283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313
/*
 * 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)
		 /* exit: our father is in a different pgrp than
		  * we are and we were the only connection outside.
		  */
		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);
	}
}

L
Linus Torvalds 已提交
314
/**
315
 * reparent_to_kthreadd - Reparent the calling kernel thread to kthreadd
L
Linus Torvalds 已提交
316 317
 *
 * If a kernel thread is launched as a result of a system call, or if
318 319
 * it ever exits, it should generally reparent itself to kthreadd so it
 * isn't in the way of other processes and is correctly cleaned up on exit.
L
Linus Torvalds 已提交
320 321 322 323
 *
 * The various task state such as scheduling policy and priority may have
 * been inherited from a user process, so we reset them to sane values here.
 *
324
 * NOTE that reparent_to_kthreadd() gives the caller full capabilities.
L
Linus Torvalds 已提交
325
 */
326
static void reparent_to_kthreadd(void)
L
Linus Torvalds 已提交
327 328 329 330 331
{
	write_lock_irq(&tasklist_lock);

	ptrace_unlink(current);
	/* Reparent to init */
332
	current->real_parent = current->parent = kthreadd_task;
R
Roland McGrath 已提交
333
	list_move_tail(&current->sibling, &current->real_parent->children);
L
Linus Torvalds 已提交
334 335 336 337

	/* Set the exit signal to SIGCHLD so we signal init on exit */
	current->exit_signal = SIGCHLD;

338
	if (task_nice(current) < 0)
L
Linus Torvalds 已提交
339 340 341 342 343 344
		set_user_nice(current, 0);
	/* cpus_allowed? */
	/* rt_priority? */
	/* signals? */
	memcpy(current->signal->rlim, init_task.signal->rlim,
	       sizeof(current->signal->rlim));
D
David Howells 已提交
345 346 347

	atomic_inc(&init_cred.usage);
	commit_creds(&init_cred);
L
Linus Torvalds 已提交
348 349 350
	write_unlock_irq(&tasklist_lock);
}

351
void __set_special_pids(struct pid *pid)
L
Linus Torvalds 已提交
352
{
353
	struct task_struct *curr = current->group_leader;
L
Linus Torvalds 已提交
354

355
	if (task_session(curr) != pid)
356
		change_pid(curr, PIDTYPE_SID, pid);
357 358

	if (task_pgrp(curr) != pid)
359
		change_pid(curr, PIDTYPE_PGID, pid);
L
Linus Torvalds 已提交
360 361
}

362
static void set_special_pids(struct pid *pid)
L
Linus Torvalds 已提交
363 364
{
	write_lock_irq(&tasklist_lock);
365
	__set_special_pids(pid);
L
Linus Torvalds 已提交
366 367 368 369
	write_unlock_irq(&tasklist_lock);
}

/*
370 371
 * Let kernel threads use this to say that they allow a certain signal.
 * Must not be used if kthread was cloned with CLONE_SIGHAND.
L
Linus Torvalds 已提交
372 373 374
 */
int allow_signal(int sig)
{
375
	if (!valid_signal(sig) || sig < 1)
L
Linus Torvalds 已提交
376 377 378
		return -EINVAL;

	spin_lock_irq(&current->sighand->siglock);
379
	/* This is only needed for daemonize()'ed kthreads */
L
Linus Torvalds 已提交
380
	sigdelset(&current->blocked, sig);
381 382 383 384 385 386
	/*
	 * Kernel threads handle their own signals. Let the signal code
	 * know it'll be handled, so that they don't get converted to
	 * SIGKILL or just silently dropped.
	 */
	current->sighand->action[(sig)-1].sa.sa_handler = (void __user *)2;
L
Linus Torvalds 已提交
387 388 389 390 391 392 393 394 395
	recalc_sigpending();
	spin_unlock_irq(&current->sighand->siglock);
	return 0;
}

EXPORT_SYMBOL(allow_signal);

int disallow_signal(int sig)
{
396
	if (!valid_signal(sig) || sig < 1)
L
Linus Torvalds 已提交
397 398 399
		return -EINVAL;

	spin_lock_irq(&current->sighand->siglock);
400
	current->sighand->action[(sig)-1].sa.sa_handler = SIG_IGN;
L
Linus Torvalds 已提交
401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427
	recalc_sigpending();
	spin_unlock_irq(&current->sighand->siglock);
	return 0;
}

EXPORT_SYMBOL(disallow_signal);

/*
 *	Put all the gunge required to become a kernel thread without
 *	attached user resources in one place where it belongs.
 */

void daemonize(const char *name, ...)
{
	va_list args;
	sigset_t blocked;

	va_start(args, name);
	vsnprintf(current->comm, sizeof(current->comm), name, args);
	va_end(args);

	/*
	 * If we were started as result of loading a module, close all of the
	 * user space pages.  We don't need them, and if we didn't close them
	 * they would be locked into memory.
	 */
	exit_mm(current);
428 429 430 431
	/*
	 * We don't want to have TIF_FREEZE set if the system-wide hibernation
	 * or suspend transition begins right now.
	 */
O
Oleg Nesterov 已提交
432
	current->flags |= (PF_NOFREEZE | PF_KTHREAD);
L
Linus Torvalds 已提交
433

434 435 436 437
	if (current->nsproxy != &init_nsproxy) {
		get_nsproxy(&init_nsproxy);
		switch_task_namespaces(current, &init_nsproxy);
	}
438
	set_special_pids(&init_struct_pid);
439
	proc_clear_tty(current);
L
Linus Torvalds 已提交
440 441 442 443 444 445 446 447

	/* Block and flush all signals */
	sigfillset(&blocked);
	sigprocmask(SIG_BLOCK, &blocked, NULL);
	flush_signals(current);

	/* Become as one with the init task */

448
	daemonize_fs_struct();
449
	exit_files(current);
L
Linus Torvalds 已提交
450 451 452
	current->files = init_task.files;
	atomic_inc(&current->files->count);

453
	reparent_to_kthreadd();
L
Linus Torvalds 已提交
454 455 456 457
}

EXPORT_SYMBOL(daemonize);

458
static void close_files(struct files_struct * files)
L
Linus Torvalds 已提交
459 460
{
	int i, j;
461
	struct fdtable *fdt;
L
Linus Torvalds 已提交
462 463

	j = 0;
464 465 466 467

	/*
	 * It is safe to dereference the fd table without RCU or
	 * ->file_lock because this is the last reference to the
468
	 * files structure.  But use RCU to shut RCU-lockdep up.
469
	 */
470
	rcu_read_lock();
471
	fdt = files_fdtable(files);
472
	rcu_read_unlock();
L
Linus Torvalds 已提交
473 474 475
	for (;;) {
		unsigned long set;
		i = j * __NFDBITS;
476
		if (i >= fdt->max_fds)
L
Linus Torvalds 已提交
477
			break;
478
		set = fdt->open_fds->fds_bits[j++];
L
Linus Torvalds 已提交
479 480
		while (set) {
			if (set & 1) {
481
				struct file * file = xchg(&fdt->fd[i], NULL);
482
				if (file) {
L
Linus Torvalds 已提交
483
					filp_close(file, files);
484 485
					cond_resched();
				}
L
Linus Torvalds 已提交
486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505
			}
			i++;
			set >>= 1;
		}
	}
}

struct files_struct *get_files_struct(struct task_struct *task)
{
	struct files_struct *files;

	task_lock(task);
	files = task->files;
	if (files)
		atomic_inc(&files->count);
	task_unlock(task);

	return files;
}

506
void put_files_struct(struct files_struct *files)
L
Linus Torvalds 已提交
507
{
508 509
	struct fdtable *fdt;

L
Linus Torvalds 已提交
510 511 512 513
	if (atomic_dec_and_test(&files->count)) {
		close_files(files);
		/*
		 * Free the fd and fdset arrays if we expanded them.
514 515 516
		 * If the fdtable was embedded, pass files for freeing
		 * at the end of the RCU grace period. Otherwise,
		 * you can free files immediately.
L
Linus Torvalds 已提交
517
		 */
518
		rcu_read_lock();
519
		fdt = files_fdtable(files);
520
		if (fdt != &files->fdtab)
521
			kmem_cache_free(files_cachep, files);
522
		free_fdtable(fdt);
523
		rcu_read_unlock();
L
Linus Torvalds 已提交
524 525 526
	}
}

527
void reset_files_struct(struct files_struct *files)
528
{
529
	struct task_struct *tsk = current;
530 531 532 533 534 535 536 537 538
	struct files_struct *old;

	old = tsk->files;
	task_lock(tsk);
	tsk->files = files;
	task_unlock(tsk);
	put_files_struct(old);
}

539
void exit_files(struct task_struct *tsk)
L
Linus Torvalds 已提交
540 541 542 543 544 545 546 547 548 549 550
{
	struct files_struct * files = tsk->files;

	if (files) {
		task_lock(tsk);
		tsk->files = NULL;
		task_unlock(tsk);
		put_files_struct(files);
	}
}

551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588
#ifdef CONFIG_MM_OWNER
/*
 * Task p is exiting and it owned mm, lets find a new owner for it
 */
static inline int
mm_need_new_owner(struct mm_struct *mm, struct task_struct *p)
{
	/*
	 * If there are other users of the mm and the owner (us) is exiting
	 * we need to find a new owner to take on the responsibility.
	 */
	if (atomic_read(&mm->mm_users) <= 1)
		return 0;
	if (mm->owner != p)
		return 0;
	return 1;
}

void mm_update_next_owner(struct mm_struct *mm)
{
	struct task_struct *c, *g, *p = current;

retry:
	if (!mm_need_new_owner(mm, p))
		return;

	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
	 */
589
	list_for_each_entry(c, &p->real_parent->children, sibling) {
590 591 592 593 594 595 596 597 598 599 600 601 602 603
		if (c->mm == mm)
			goto assign_new_owner;
	}

	/*
	 * Search through everything else. We should not get
	 * here often
	 */
	do_each_thread(g, c) {
		if (c->mm == mm)
			goto assign_new_owner;
	} while_each_thread(g, c);

	read_unlock(&tasklist_lock);
604 605 606
	/*
	 * We found no owner yet mm_users > 1: this implies that we are
	 * most likely racing with swapoff (try_to_unuse()) or /proc or
607
	 * ptrace or page migration (get_task_mm()).  Mark owner as NULL.
608 609
	 */
	mm->owner = NULL;
610 611 612 613 614 615 616 617 618 619
	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);
620 621 622 623 624
	/*
	 * Delay read_unlock() till we have the task_lock()
	 * to ensure that c does not slip away underneath us
	 */
	read_unlock(&tasklist_lock);
625 626 627 628 629 630 631 632 633 634 635
	if (c->mm != mm) {
		task_unlock(c);
		put_task_struct(c);
		goto retry;
	}
	mm->owner = c;
	task_unlock(c);
	put_task_struct(c);
}
#endif /* CONFIG_MM_OWNER */

L
Linus Torvalds 已提交
636 637 638 639
/*
 * Turn us into a lazy TLB process if we
 * aren't already..
 */
A
Adrian Bunk 已提交
640
static void exit_mm(struct task_struct * tsk)
L
Linus Torvalds 已提交
641 642
{
	struct mm_struct *mm = tsk->mm;
643
	struct core_state *core_state;
L
Linus Torvalds 已提交
644 645 646 647 648 649

	mm_release(tsk, mm);
	if (!mm)
		return;
	/*
	 * Serialize with any possible pending coredump.
650
	 * We must hold mmap_sem around checking core_state
L
Linus Torvalds 已提交
651
	 * and clearing tsk->mm.  The core-inducing thread
652
	 * will increment ->nr_threads for each thread in the
L
Linus Torvalds 已提交
653 654 655
	 * group with ->mm != NULL.
	 */
	down_read(&mm->mmap_sem);
656 657 658
	core_state = mm->core_state;
	if (core_state) {
		struct core_thread self;
L
Linus Torvalds 已提交
659 660
		up_read(&mm->mmap_sem);

661 662 663 664 665 666 667 668
		self.task = tsk;
		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 已提交
669

O
Oleg Nesterov 已提交
670 671 672 673 674 675 676
		for (;;) {
			set_task_state(tsk, TASK_UNINTERRUPTIBLE);
			if (!self.task) /* see coredump_finish() */
				break;
			schedule();
		}
		__set_task_state(tsk, TASK_RUNNING);
L
Linus Torvalds 已提交
677 678 679
		down_read(&mm->mmap_sem);
	}
	atomic_inc(&mm->mm_count);
E
Eric Sesterhenn 已提交
680
	BUG_ON(mm != tsk->active_mm);
L
Linus Torvalds 已提交
681 682 683 684 685
	/* more a memory barrier than a real lock */
	task_lock(tsk);
	tsk->mm = NULL;
	up_read(&mm->mmap_sem);
	enter_lazy_tlb(mm, current);
686 687
	/* We don't want this task to be frozen prematurely */
	clear_freeze_flag(tsk);
Y
Ying Han 已提交
688 689
	if (tsk->signal->oom_score_adj == OOM_SCORE_ADJ_MIN)
		atomic_dec(&mm->oom_disable_count);
L
Linus Torvalds 已提交
690
	task_unlock(tsk);
691
	mm_update_next_owner(mm);
L
Linus Torvalds 已提交
692 693 694 695 696 697 698
	mmput(mm);
}

/*
 * When we die, we re-parent all our children.
 * Try to give them to another thread in our thread
 * group, and if no such member exists, give it to
699 700
 * the child reaper process (ie "init") in our pid
 * space.
L
Linus Torvalds 已提交
701
 */
702
static struct task_struct *find_new_reaper(struct task_struct *father)
703 704
	__releases(&tasklist_lock)
	__acquires(&tasklist_lock)
L
Linus Torvalds 已提交
705
{
706 707
	struct pid_namespace *pid_ns = task_active_pid_ns(father);
	struct task_struct *thread;
L
Linus Torvalds 已提交
708

709 710 711 712 713 714 715 716
	thread = father;
	while_each_thread(father, thread) {
		if (thread->flags & PF_EXITING)
			continue;
		if (unlikely(pid_ns->child_reaper == father))
			pid_ns->child_reaper = thread;
		return thread;
	}
L
Linus Torvalds 已提交
717

718 719 720 721
	if (unlikely(pid_ns->child_reaper == father)) {
		write_unlock_irq(&tasklist_lock);
		if (unlikely(pid_ns == &init_pid_ns))
			panic("Attempted to kill init!");
L
Linus Torvalds 已提交
722

723 724
		zap_pid_ns_processes(pid_ns);
		write_lock_irq(&tasklist_lock);
L
Linus Torvalds 已提交
725
		/*
726 727 728
		 * We can not clear ->child_reaper or leave it alone.
		 * There may by stealth EXIT_DEAD tasks on ->children,
		 * forget_original_parent() must move them somewhere.
L
Linus Torvalds 已提交
729
		 */
730
		pid_ns->child_reaper = init_pid_ns.child_reaper;
L
Linus Torvalds 已提交
731
	}
732

733 734 735
	return pid_ns->child_reaper;
}

736 737 738
/*
* Any that need to be release_task'd are put on the @dead list.
 */
739
static void reparent_leader(struct task_struct *father, struct task_struct *p,
740 741 742 743
				struct list_head *dead)
{
	list_move_tail(&p->sibling, &p->real_parent->children);

744
	if (p->exit_state == EXIT_DEAD)
745 746 747 748 749 750 751 752 753 754 755 756
		return;
	/*
	 * If this is a threaded reparent there is no need to
	 * notify anyone anything has happened.
	 */
	if (same_thread_group(p->real_parent, father))
		return;

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

	/* If it has exited notify the new parent about this child's death. */
T
Tejun Heo 已提交
757
	if (!p->ptrace &&
758
	    p->exit_state == EXIT_ZOMBIE && thread_group_empty(p)) {
759
		if (do_notify_parent(p, p->exit_signal)) {
760 761 762 763 764 765 766 767
			p->exit_state = EXIT_DEAD;
			list_move_tail(&p->sibling, dead);
		}
	}

	kill_orphaned_pgrp(p, father);
}

768
static void forget_original_parent(struct task_struct *father)
L
Linus Torvalds 已提交
769
{
770
	struct task_struct *p, *n, *reaper;
771
	LIST_HEAD(dead_children);
772 773

	write_lock_irq(&tasklist_lock);
774 775 776 777 778
	/*
	 * Note that exit_ptrace() and find_new_reaper() might
	 * drop tasklist_lock and reacquire it.
	 */
	exit_ptrace(father);
779
	reaper = find_new_reaper(father);
R
Roland McGrath 已提交
780

781
	list_for_each_entry_safe(p, n, &father->children, sibling) {
782 783 784 785
		struct task_struct *t = p;
		do {
			t->real_parent = reaper;
			if (t->parent == father) {
T
Tejun Heo 已提交
786
				BUG_ON(t->ptrace);
787 788 789 790 791 792 793
				t->parent = t->real_parent;
			}
			if (t->pdeath_signal)
				group_send_sig_info(t->pdeath_signal,
						    SEND_SIG_NOINFO, t);
		} while_each_thread(p, t);
		reparent_leader(father, p, &dead_children);
L
Linus Torvalds 已提交
794
	}
795
	write_unlock_irq(&tasklist_lock);
796

797 798
	BUG_ON(!list_empty(&father->children));

799 800
	list_for_each_entry_safe(p, n, &dead_children, sibling) {
		list_del_init(&p->sibling);
801 802
		release_task(p);
	}
L
Linus Torvalds 已提交
803 804 805 806 807 808
}

/*
 * Send signals to all our closest relatives so that they know
 * to properly mourn us..
 */
809
static void exit_notify(struct task_struct *tsk, int group_dead)
L
Linus Torvalds 已提交
810
{
811
	bool autoreap;
L
Linus Torvalds 已提交
812 813 814 815 816 817 818 819 820

	/*
	 * 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)
	 */
821
	forget_original_parent(tsk);
822
	exit_task_namespaces(tsk);
L
Linus Torvalds 已提交
823

824
	write_lock_irq(&tasklist_lock);
825 826
	if (group_dead)
		kill_orphaned_pgrp(tsk->group_leader, NULL);
L
Linus Torvalds 已提交
827

828
	/* Let father know we died
L
Linus Torvalds 已提交
829 830
	 *
	 * Thread signals are configurable, but you aren't going to use
L
Lucas De Marchi 已提交
831
	 * that to send signals to arbitrary processes.
L
Linus Torvalds 已提交
832 833 834 835 836 837 838 839 840 841
	 * That stops right now.
	 *
	 * If the parent exec id doesn't match the exec id we saved
	 * when we started then we know the parent has changed security
	 * domain.
	 *
	 * If our self_exec id doesn't match our parent_exec_id then
	 * we have changed execution domain as these two values started
	 * the same after a fork.
	 */
O
Oleg Nesterov 已提交
842
	if (thread_group_leader(tsk) && tsk->exit_signal != SIGCHLD &&
843
	    (tsk->parent_exec_id != tsk->real_parent->self_exec_id ||
844
	     tsk->self_exec_id != tsk->parent_exec_id))
L
Linus Torvalds 已提交
845 846
		tsk->exit_signal = SIGCHLD;

O
Oleg Nesterov 已提交
847 848 849 850 851 852 853 854 855 856 857 858
	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 已提交
859

860
	tsk->exit_state = autoreap ? EXIT_DEAD : EXIT_ZOMBIE;
L
Linus Torvalds 已提交
861

862 863
	/* mt-exec, de_thread() is waiting for group leader */
	if (unlikely(tsk->signal->notify_count < 0))
864
		wake_up_process(tsk->signal->group_exit_task);
L
Linus Torvalds 已提交
865 866 867
	write_unlock_irq(&tasklist_lock);

	/* If the process is dead, release it - nobody will wait for it */
868
	if (autoreap)
L
Linus Torvalds 已提交
869 870 871
		release_task(tsk);
}

872 873 874 875 876 877 878
#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;

879
	free = stack_not_used(current);
880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896

	if (free >= lowest_to_date)
		return;

	spin_lock(&low_water_lock);
	if (free < lowest_to_date) {
		printk(KERN_WARNING "%s used greatest stack depth: %lu bytes "
				"left\n",
				current->comm, free);
		lowest_to_date = free;
	}
	spin_unlock(&low_water_lock);
}
#else
static inline void check_stack_usage(void) {}
#endif

897
NORET_TYPE void do_exit(long code)
L
Linus Torvalds 已提交
898 899 900 901 902 903
{
	struct task_struct *tsk = current;
	int group_dead;

	profile_task_exit(tsk);

904
	WARN_ON(atomic_read(&tsk->fs_excl));
905
	WARN_ON(blk_needs_flush_plug(tsk));
906

L
Linus Torvalds 已提交
907 908 909 910 911
	if (unlikely(in_interrupt()))
		panic("Aiee, killing interrupt handler!");
	if (unlikely(!tsk->pid))
		panic("Attempted to kill the idle task!");

912 913 914 915 916 917 918 919 920
	/*
	 * 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 已提交
921
	ptrace_event(PTRACE_EVENT_EXIT, code);
L
Linus Torvalds 已提交
922

923 924
	validate_creds_for_do_exit(tsk);

925 926 927 928 929 930 931
	/*
	 * 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)) {
		printk(KERN_ALERT
			"Fixing recursive fault but reboot is needed!\n");
932 933 934 935 936 937 938 939 940 941
		/*
		 * 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;
942 943 944 945
		set_current_state(TASK_UNINTERRUPTIBLE);
		schedule();
	}

946 947
	exit_irq_thread();

O
Oleg Nesterov 已提交
948
	exit_signals(tsk);  /* sets PF_EXITING */
949 950 951 952
	/*
	 * tsk->flags are checked in the futex code to protect against
	 * an exiting task cleaning up the robust pi futexes.
	 */
953
	smp_mb();
954
	raw_spin_unlock_wait(&tsk->pi_lock);
L
Linus Torvalds 已提交
955 956 957

	if (unlikely(in_atomic()))
		printk(KERN_INFO "note: %s[%d] exited with preempt_count %d\n",
958
				current->comm, task_pid_nr(current),
L
Linus Torvalds 已提交
959 960 961
				preempt_count());

	acct_update_integrals(tsk);
962
	/* sync mm's RSS info before statistics gathering */
963 964
	if (tsk->mm)
		sync_mm_rss(tsk, tsk->mm);
L
Linus Torvalds 已提交
965
	group_dead = atomic_dec_and_test(&tsk->signal->live);
966
	if (group_dead) {
967
		hrtimer_cancel(&tsk->signal->real_timer);
968
		exit_itimers(tsk->signal);
J
Jiri Pirko 已提交
969 970
		if (tsk->mm)
			setmax_mm_hiwater_rss(&tsk->signal->maxrss, tsk->mm);
971
	}
972
	acct_collect(code, group_dead);
M
Miloslav Trmac 已提交
973 974
	if (group_dead)
		tty_audit_exit();
975 976
	if (unlikely(tsk->audit_context))
		audit_free(tsk);
977

978
	tsk->exit_code = code;
979
	taskstats_exit(tsk, group_dead);
980

L
Linus Torvalds 已提交
981 982
	exit_mm(tsk);

983
	if (group_dead)
984
		acct_process();
985 986
	trace_sched_process_exit(tsk);

L
Linus Torvalds 已提交
987
	exit_sem(tsk);
988 989
	exit_files(tsk);
	exit_fs(tsk);
990
	check_stack_usage();
L
Linus Torvalds 已提交
991
	exit_thread();
992 993 994 995 996 997 998 999 1000

	/*
	 * 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);

1001
	cgroup_exit(tsk, 1);
L
Linus Torvalds 已提交
1002

1003
	if (group_dead)
L
Linus Torvalds 已提交
1004 1005
		disassociate_ctty(1);

A
Al Viro 已提交
1006
	module_put(task_thread_info(tsk)->exec_domain->module);
L
Linus Torvalds 已提交
1007

M
Matt Helsley 已提交
1008
	proc_exit_connector(tsk);
1009

1010 1011 1012
	/*
	 * FIXME: do that only when needed, using sched_exit tracepoint
	 */
1013
	ptrace_put_breakpoints(tsk);
1014

1015
	exit_notify(tsk, group_dead);
L
Linus Torvalds 已提交
1016
#ifdef CONFIG_NUMA
1017
	task_lock(tsk);
1018
	mpol_put(tsk->mempolicy);
L
Linus Torvalds 已提交
1019
	tsk->mempolicy = NULL;
1020
	task_unlock(tsk);
L
Linus Torvalds 已提交
1021
#endif
1022
#ifdef CONFIG_FUTEX
1023 1024
	if (unlikely(current->pi_state_cache))
		kfree(current->pi_state_cache);
1025
#endif
1026
	/*
1027
	 * Make sure we are holding no locks:
1028
	 */
1029
	debug_check_no_locks_held(tsk);
1030 1031 1032 1033 1034 1035
	/*
	 * 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 已提交
1036

1037
	if (tsk->io_context)
1038
		exit_io_context(tsk);
1039

1040 1041 1042
	if (tsk->splice_pipe)
		__free_pipe_info(tsk->splice_pipe);

1043 1044
	validate_creds_for_do_exit(tsk);

C
Coywolf Qi Hunt 已提交
1045
	preempt_disable();
1046
	exit_rcu();
O
Oleg Nesterov 已提交
1047
	/* causes final put_task_struct in finish_task_switch(). */
1048
	tsk->state = TASK_DEAD;
L
Linus Torvalds 已提交
1049 1050 1051
	schedule();
	BUG();
	/* Avoid "noreturn function does return".  */
A
Alan Cox 已提交
1052 1053
	for (;;)
		cpu_relax();	/* For when BUG is null */
L
Linus Torvalds 已提交
1054 1055
}

1056 1057
EXPORT_SYMBOL_GPL(do_exit);

L
Linus Torvalds 已提交
1058 1059 1060 1061
NORET_TYPE void complete_and_exit(struct completion *comp, long code)
{
	if (comp)
		complete(comp);
O
Oleg Nesterov 已提交
1062

L
Linus Torvalds 已提交
1063 1064 1065 1066 1067
	do_exit(code);
}

EXPORT_SYMBOL(complete_and_exit);

1068
SYSCALL_DEFINE1(exit, int, error_code)
L
Linus Torvalds 已提交
1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079
{
	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).
 */
NORET_TYPE void
do_group_exit(int exit_code)
{
1080 1081
	struct signal_struct *sig = current->signal;

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

1084 1085
	if (signal_group_exit(sig))
		exit_code = sig->group_exit_code;
L
Linus Torvalds 已提交
1086 1087 1088
	else if (!thread_group_empty(current)) {
		struct sighand_struct *const sighand = current->sighand;
		spin_lock_irq(&sighand->siglock);
1089
		if (signal_group_exit(sig))
L
Linus Torvalds 已提交
1090 1091 1092 1093
			/* Another thread got here before we took the lock.  */
			exit_code = sig->group_exit_code;
		else {
			sig->group_exit_code = exit_code;
1094
			sig->flags = SIGNAL_GROUP_EXIT;
L
Linus Torvalds 已提交
1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108
			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.
 */
1109
SYSCALL_DEFINE1(exit_group, int, error_code)
L
Linus Torvalds 已提交
1110 1111
{
	do_group_exit((error_code & 0xff) << 8);
1112 1113
	/* NOTREACHED */
	return 0;
L
Linus Torvalds 已提交
1114 1115
}

1116 1117 1118
struct wait_opts {
	enum pid_type		wo_type;
	int			wo_flags;
1119
	struct pid		*wo_pid;
1120 1121 1122 1123 1124

	struct siginfo __user	*wo_info;
	int __user		*wo_stat;
	struct rusage __user	*wo_rusage;

1125
	wait_queue_t		child_wait;
1126 1127 1128
	int			notask_error;
};

1129 1130
static inline
struct pid *task_pid_type(struct task_struct *task, enum pid_type type)
E
Eric W. Biederman 已提交
1131
{
1132 1133 1134
	if (type != PIDTYPE_PID)
		task = task->group_leader;
	return task->pids[type].pid;
E
Eric W. Biederman 已提交
1135 1136
}

1137
static int eligible_pid(struct wait_opts *wo, struct task_struct *p)
L
Linus Torvalds 已提交
1138
{
1139 1140 1141
	return	wo->wo_type == PIDTYPE_MAX ||
		task_pid_type(p, wo->wo_type) == wo->wo_pid;
}
L
Linus Torvalds 已提交
1142

1143 1144 1145 1146
static int eligible_child(struct wait_opts *wo, struct task_struct *p)
{
	if (!eligible_pid(wo, p))
		return 0;
L
Linus Torvalds 已提交
1147 1148 1149 1150 1151
	/* Wait for all children (clone and not) if __WALL is set;
	 * 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.) */
1152 1153
	if (((p->exit_signal != SIGCHLD) ^ !!(wo->wo_flags & __WCLONE))
	    && !(wo->wo_flags & __WALL))
L
Linus Torvalds 已提交
1154 1155
		return 0;

1156
	return 1;
L
Linus Torvalds 已提交
1157 1158
}

1159 1160
static int wait_noreap_copyout(struct wait_opts *wo, struct task_struct *p,
				pid_t pid, uid_t uid, int why, int status)
L
Linus Torvalds 已提交
1161
{
1162 1163 1164
	struct siginfo __user *infop;
	int retval = wo->wo_rusage
		? getrusage(p, RUSAGE_BOTH, wo->wo_rusage) : 0;
1165

L
Linus Torvalds 已提交
1166
	put_task_struct(p);
1167
	infop = wo->wo_info;
1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181
	if (infop) {
		if (!retval)
			retval = put_user(SIGCHLD, &infop->si_signo);
		if (!retval)
			retval = put_user(0, &infop->si_errno);
		if (!retval)
			retval = put_user((short)why, &infop->si_code);
		if (!retval)
			retval = put_user(pid, &infop->si_pid);
		if (!retval)
			retval = put_user(uid, &infop->si_uid);
		if (!retval)
			retval = put_user(status, &infop->si_status);
	}
L
Linus Torvalds 已提交
1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192
	if (!retval)
		retval = pid;
	return retval;
}

/*
 * 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.
 */
1193
static int wait_task_zombie(struct wait_opts *wo, struct task_struct *p)
L
Linus Torvalds 已提交
1194 1195
{
	unsigned long state;
1196
	int retval, status, traced;
1197
	pid_t pid = task_pid_vnr(p);
1198
	uid_t uid = __task_cred(p)->uid;
1199
	struct siginfo __user *infop;
L
Linus Torvalds 已提交
1200

1201
	if (!likely(wo->wo_flags & WEXITED))
R
Roland McGrath 已提交
1202 1203
		return 0;

1204
	if (unlikely(wo->wo_flags & WNOWAIT)) {
L
Linus Torvalds 已提交
1205
		int exit_code = p->exit_code;
1206
		int why;
L
Linus Torvalds 已提交
1207 1208 1209 1210 1211 1212 1213 1214 1215 1216

		get_task_struct(p);
		read_unlock(&tasklist_lock);
		if ((exit_code & 0x7f) == 0) {
			why = CLD_EXITED;
			status = exit_code >> 8;
		} else {
			why = (exit_code & 0x80) ? CLD_DUMPED : CLD_KILLED;
			status = exit_code & 0x7f;
		}
1217
		return wait_noreap_copyout(wo, p, pid, uid, why, status);
L
Linus Torvalds 已提交
1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229
	}

	/*
	 * Try to move the task's state to DEAD
	 * only one thread is allowed to do this:
	 */
	state = xchg(&p->exit_state, EXIT_DEAD);
	if (state != EXIT_ZOMBIE) {
		BUG_ON(state != EXIT_DEAD);
		return 0;
	}

1230
	traced = ptrace_reparented(p);
1231 1232
	/*
	 * It can be ptraced but not reparented, check
O
Oleg Nesterov 已提交
1233
	 * thread_group_leader() to filter out sub-threads.
1234
	 */
O
Oleg Nesterov 已提交
1235
	if (likely(!traced) && thread_group_leader(p)) {
1236 1237
		struct signal_struct *psig;
		struct signal_struct *sig;
J
Jiri Pirko 已提交
1238
		unsigned long maxrss;
1239
		cputime_t tgutime, tgstime;
1240

L
Linus Torvalds 已提交
1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251
		/*
		 * 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
		 * p->signal fields, because they are only touched by
		 * __exit_signal, which runs with tasklist_lock
		 * write-locked anyway, and so is excluded here.  We do
1252
		 * need to protect the access to parent->signal fields,
L
Linus Torvalds 已提交
1253 1254
		 * as other threads in the parent group can be right
		 * here reaping other children at the same time.
1255 1256 1257 1258
		 *
		 * We use thread_group_times() to get times for the thread
		 * group, which consolidates times for all threads in the
		 * group including the group leader.
L
Linus Torvalds 已提交
1259
		 */
1260
		thread_group_times(p, &tgutime, &tgstime);
1261 1262
		spin_lock_irq(&p->real_parent->sighand->siglock);
		psig = p->real_parent->signal;
1263 1264 1265
		sig = p->signal;
		psig->cutime =
			cputime_add(psig->cutime,
1266 1267
			cputime_add(tgutime,
				    sig->cutime));
1268 1269
		psig->cstime =
			cputime_add(psig->cstime,
1270 1271
			cputime_add(tgstime,
				    sig->cstime));
1272 1273 1274 1275 1276
		psig->cgtime =
			cputime_add(psig->cgtime,
			cputime_add(p->gtime,
			cputime_add(sig->gtime,
				    sig->cgtime)));
1277 1278 1279 1280 1281 1282 1283 1284
		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;
1285 1286 1287 1288 1289 1290
		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 已提交
1291 1292 1293
		maxrss = max(sig->maxrss, sig->cmaxrss);
		if (psig->cmaxrss < maxrss)
			psig->cmaxrss = maxrss;
1294 1295
		task_io_accounting_add(&psig->ioac, &p->ioac);
		task_io_accounting_add(&psig->ioac, &sig->ioac);
1296
		spin_unlock_irq(&p->real_parent->sighand->siglock);
L
Linus Torvalds 已提交
1297 1298 1299 1300 1301 1302 1303 1304
	}

	/*
	 * Now we are sure this task is interesting, and no other
	 * thread can reap it because we set its state to EXIT_DEAD.
	 */
	read_unlock(&tasklist_lock);

1305 1306
	retval = wo->wo_rusage
		? getrusage(p, RUSAGE_BOTH, wo->wo_rusage) : 0;
L
Linus Torvalds 已提交
1307 1308
	status = (p->signal->flags & SIGNAL_GROUP_EXIT)
		? p->signal->group_exit_code : p->exit_code;
1309 1310 1311 1312
	if (!retval && wo->wo_stat)
		retval = put_user(status, wo->wo_stat);

	infop = wo->wo_info;
L
Linus Torvalds 已提交
1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331
	if (!retval && infop)
		retval = put_user(SIGCHLD, &infop->si_signo);
	if (!retval && infop)
		retval = put_user(0, &infop->si_errno);
	if (!retval && infop) {
		int why;

		if ((status & 0x7f) == 0) {
			why = CLD_EXITED;
			status >>= 8;
		} else {
			why = (status & 0x80) ? CLD_DUMPED : CLD_KILLED;
			status &= 0x7f;
		}
		retval = put_user((short)why, &infop->si_code);
		if (!retval)
			retval = put_user(status, &infop->si_status);
	}
	if (!retval && infop)
1332
		retval = put_user(pid, &infop->si_pid);
L
Linus Torvalds 已提交
1333
	if (!retval && infop)
1334
		retval = put_user(uid, &infop->si_uid);
1335
	if (!retval)
1336
		retval = pid;
1337 1338

	if (traced) {
L
Linus Torvalds 已提交
1339
		write_lock_irq(&tasklist_lock);
1340 1341 1342
		/* We dropped tasklist, ptracer could die and untrace */
		ptrace_unlink(p);
		/*
1343 1344
		 * If this is not a sub-thread, notify the parent.
		 * If parent wants a zombie, don't release it now.
1345
		 */
1346 1347 1348 1349
		if (thread_group_leader(p) &&
		    !do_notify_parent(p, p->exit_signal)) {
			p->exit_state = EXIT_ZOMBIE;
			p = NULL;
L
Linus Torvalds 已提交
1350 1351 1352 1353 1354
		}
		write_unlock_irq(&tasklist_lock);
	}
	if (p != NULL)
		release_task(p);
1355

L
Linus Torvalds 已提交
1356 1357 1358
	return retval;
}

1359 1360 1361
static int *task_stopped_code(struct task_struct *p, bool ptrace)
{
	if (ptrace) {
T
Tejun Heo 已提交
1362 1363
		if (task_is_stopped_or_traced(p) &&
		    !(p->jobctl & JOBCTL_LISTENING))
1364 1365 1366 1367 1368 1369 1370 1371
			return &p->exit_code;
	} else {
		if (p->signal->flags & SIGNAL_STOP_STOPPED)
			return &p->signal->group_exit_code;
	}
	return NULL;
}

1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388
/**
 * 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 已提交
1389
 */
1390 1391
static int wait_task_stopped(struct wait_opts *wo,
				int ptrace, struct task_struct *p)
L
Linus Torvalds 已提交
1392
{
1393
	struct siginfo __user *infop;
1394
	int retval, exit_code, *p_code, why;
1395
	uid_t uid = 0; /* unneeded, required by compiler */
1396
	pid_t pid;
L
Linus Torvalds 已提交
1397

1398 1399 1400
	/*
	 * Traditionally we see ptrace'd stopped tasks regardless of options.
	 */
1401
	if (!ptrace && !(wo->wo_flags & WUNTRACED))
R
Roland McGrath 已提交
1402 1403
		return 0;

1404 1405 1406
	if (!task_stopped_code(p, ptrace))
		return 0;

1407 1408 1409
	exit_code = 0;
	spin_lock_irq(&p->sighand->siglock);

1410 1411
	p_code = task_stopped_code(p, ptrace);
	if (unlikely(!p_code))
1412 1413
		goto unlock_sig;

1414
	exit_code = *p_code;
1415 1416 1417
	if (!exit_code)
		goto unlock_sig;

1418
	if (!unlikely(wo->wo_flags & WNOWAIT))
1419
		*p_code = 0;
1420

1421
	uid = task_uid(p);
1422 1423 1424
unlock_sig:
	spin_unlock_irq(&p->sighand->siglock);
	if (!exit_code)
L
Linus Torvalds 已提交
1425 1426 1427 1428 1429 1430 1431 1432 1433 1434
		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);
1435
	pid = task_pid_vnr(p);
R
Roland McGrath 已提交
1436
	why = ptrace ? CLD_TRAPPED : CLD_STOPPED;
L
Linus Torvalds 已提交
1437 1438
	read_unlock(&tasklist_lock);

1439 1440 1441 1442 1443 1444 1445
	if (unlikely(wo->wo_flags & WNOWAIT))
		return wait_noreap_copyout(wo, p, pid, uid, why, exit_code);

	retval = wo->wo_rusage
		? getrusage(p, RUSAGE_BOTH, wo->wo_rusage) : 0;
	if (!retval && wo->wo_stat)
		retval = put_user((exit_code << 8) | 0x7f, wo->wo_stat);
L
Linus Torvalds 已提交
1446

1447
	infop = wo->wo_info;
L
Linus Torvalds 已提交
1448 1449 1450 1451 1452
	if (!retval && infop)
		retval = put_user(SIGCHLD, &infop->si_signo);
	if (!retval && infop)
		retval = put_user(0, &infop->si_errno);
	if (!retval && infop)
R
Roland McGrath 已提交
1453
		retval = put_user((short)why, &infop->si_code);
L
Linus Torvalds 已提交
1454 1455 1456
	if (!retval && infop)
		retval = put_user(exit_code, &infop->si_status);
	if (!retval && infop)
1457
		retval = put_user(pid, &infop->si_pid);
L
Linus Torvalds 已提交
1458
	if (!retval && infop)
1459
		retval = put_user(uid, &infop->si_uid);
L
Linus Torvalds 已提交
1460
	if (!retval)
1461
		retval = pid;
L
Linus Torvalds 已提交
1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473
	put_task_struct(p);

	BUG_ON(!retval);
	return retval;
}

/*
 * 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.
 */
1474
static int wait_task_continued(struct wait_opts *wo, struct task_struct *p)
L
Linus Torvalds 已提交
1475 1476 1477 1478 1479
{
	int retval;
	pid_t pid;
	uid_t uid;

1480
	if (!unlikely(wo->wo_flags & WCONTINUED))
R
Roland McGrath 已提交
1481 1482
		return 0;

L
Linus Torvalds 已提交
1483 1484 1485 1486 1487 1488 1489 1490 1491
	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;
	}
1492
	if (!unlikely(wo->wo_flags & WNOWAIT))
L
Linus Torvalds 已提交
1493
		p->signal->flags &= ~SIGNAL_STOP_CONTINUED;
1494
	uid = task_uid(p);
L
Linus Torvalds 已提交
1495 1496
	spin_unlock_irq(&p->sighand->siglock);

1497
	pid = task_pid_vnr(p);
L
Linus Torvalds 已提交
1498 1499 1500
	get_task_struct(p);
	read_unlock(&tasklist_lock);

1501 1502 1503
	if (!wo->wo_info) {
		retval = wo->wo_rusage
			? getrusage(p, RUSAGE_BOTH, wo->wo_rusage) : 0;
L
Linus Torvalds 已提交
1504
		put_task_struct(p);
1505 1506
		if (!retval && wo->wo_stat)
			retval = put_user(0xffff, wo->wo_stat);
L
Linus Torvalds 已提交
1507
		if (!retval)
1508
			retval = pid;
L
Linus Torvalds 已提交
1509
	} else {
1510 1511
		retval = wait_noreap_copyout(wo, p, pid, uid,
					     CLD_CONTINUED, SIGCONT);
L
Linus Torvalds 已提交
1512 1513 1514 1515 1516 1517
		BUG_ON(retval == 0);
	}

	return retval;
}

R
Roland McGrath 已提交
1518 1519 1520
/*
 * Consider @p for a wait by @parent.
 *
1521
 * -ECHILD should be in ->notask_error before the first call.
R
Roland McGrath 已提交
1522 1523
 * Returns nonzero for a final return, when we have unlocked tasklist_lock.
 * Returns zero if the search for a child should continue;
1524
 * then ->notask_error is 0 if @p is an eligible child,
1525
 * or another error from security_task_wait(), or still -ECHILD.
R
Roland McGrath 已提交
1526
 */
1527 1528
static int wait_consider_task(struct wait_opts *wo, int ptrace,
				struct task_struct *p)
R
Roland McGrath 已提交
1529
{
1530
	int ret = eligible_child(wo, p);
1531
	if (!ret)
R
Roland McGrath 已提交
1532 1533
		return ret;

1534
	ret = security_task_wait(p);
1535 1536 1537 1538 1539 1540 1541 1542
	if (unlikely(ret < 0)) {
		/*
		 * If we have not yet seen any eligible child,
		 * then let this error code replace -ECHILD.
		 * A permission error will give the user a clue
		 * to look for security policy problems, rather
		 * than for mysterious wait bugs.
		 */
1543 1544
		if (wo->notask_error)
			wo->notask_error = ret;
1545
		return 0;
1546 1547
	}

1548 1549 1550 1551
	/* dead body doesn't have much to contribute */
	if (p->exit_state == EXIT_DEAD)
		return 0;

1552 1553
	/* slay zombie? */
	if (p->exit_state == EXIT_ZOMBIE) {
R
Roland McGrath 已提交
1554
		/*
1555 1556 1557
		 * A zombie ptracee is only visible to its ptracer.
		 * Notification and reaping will be cascaded to the real
		 * parent when the ptracer detaches.
R
Roland McGrath 已提交
1558
		 */
T
Tejun Heo 已提交
1559
		if (likely(!ptrace) && unlikely(p->ptrace)) {
1560 1561 1562 1563
			/* it will become visible, clear notask_error */
			wo->notask_error = 0;
			return 0;
		}
R
Roland McGrath 已提交
1564

1565 1566 1567
		/* we don't reap group leaders with subthreads */
		if (!delay_group_leader(p))
			return wait_task_zombie(wo, p);
R
Roland McGrath 已提交
1568

R
Roland McGrath 已提交
1569
		/*
1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591
		 * 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 {
1592 1593 1594 1595 1596 1597 1598 1599 1600 1601
		/*
		 * If @p is ptraced by a task in its real parent's group,
		 * hide group stop/continued state when looking at @p as
		 * the real parent; otherwise, a single stop can be
		 * reported twice as group and ptrace stops.
		 *
		 * If a ptracer wants to distinguish the two events for its
		 * own children, it should create a separate process which
		 * takes the role of real parent.
		 */
T
Tejun Heo 已提交
1602
		if (likely(!ptrace) && p->ptrace &&
1603 1604 1605
		    same_thread_group(p->parent, p->real_parent))
			return 0;

1606 1607 1608
		/*
		 * @p is alive and it's gonna stop, continue or exit, so
		 * there always is something to wait for.
R
Roland McGrath 已提交
1609
		 */
1610
		wo->notask_error = 0;
R
Roland McGrath 已提交
1611 1612
	}

R
Roland McGrath 已提交
1613
	/*
1614 1615
	 * Wait for stopped.  Depending on @ptrace, different stopped state
	 * is used and the two don't interact with each other.
R
Roland McGrath 已提交
1616
	 */
1617 1618 1619
	ret = wait_task_stopped(wo, ptrace, p);
	if (ret)
		return ret;
R
Roland McGrath 已提交
1620 1621

	/*
1622 1623 1624
	 * 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 已提交
1625
	 */
1626
	return wait_task_continued(wo, p);
R
Roland McGrath 已提交
1627 1628 1629 1630 1631
}

/*
 * Do the work of do_wait() for one thread in the group, @tsk.
 *
1632
 * -ECHILD should be in ->notask_error before the first call.
R
Roland McGrath 已提交
1633 1634
 * Returns nonzero for a final return, when we have unlocked tasklist_lock.
 * Returns zero if the search for a child should continue; then
1635
 * ->notask_error is 0 if there were any eligible children,
1636
 * or another error from security_task_wait(), or still -ECHILD.
R
Roland McGrath 已提交
1637
 */
1638
static int do_wait_thread(struct wait_opts *wo, struct task_struct *tsk)
R
Roland McGrath 已提交
1639 1640 1641 1642
{
	struct task_struct *p;

	list_for_each_entry(p, &tsk->children, sibling) {
1643 1644 1645
		int ret = wait_consider_task(wo, 0, p);
		if (ret)
			return ret;
R
Roland McGrath 已提交
1646 1647 1648 1649 1650
	}

	return 0;
}

1651
static int ptrace_do_wait(struct wait_opts *wo, struct task_struct *tsk)
R
Roland McGrath 已提交
1652 1653 1654
{
	struct task_struct *p;

R
Roland McGrath 已提交
1655
	list_for_each_entry(p, &tsk->ptraced, ptrace_entry) {
1656
		int ret = wait_consider_task(wo, 1, p);
R
Roland McGrath 已提交
1657
		if (ret)
R
Roland McGrath 已提交
1658 1659 1660 1661 1662 1663
			return ret;
	}

	return 0;
}

1664 1665 1666 1667 1668 1669 1670
static int child_wait_callback(wait_queue_t *wait, unsigned mode,
				int sync, void *key)
{
	struct wait_opts *wo = container_of(wait, struct wait_opts,
						child_wait);
	struct task_struct *p = key;

1671
	if (!eligible_pid(wo, p))
1672 1673
		return 0;

1674 1675 1676
	if ((wo->wo_flags & __WNOTHREAD) && wait->private != p->parent)
		return 0;

1677 1678 1679
	return default_wake_function(wait, mode, sync, key);
}

1680 1681
void __wake_up_parent(struct task_struct *p, struct task_struct *parent)
{
1682 1683
	__wake_up_sync_key(&parent->signal->wait_chldexit,
				TASK_INTERRUPTIBLE, 1, p);
1684 1685
}

1686
static long do_wait(struct wait_opts *wo)
L
Linus Torvalds 已提交
1687 1688
{
	struct task_struct *tsk;
R
Roland McGrath 已提交
1689
	int retval;
L
Linus Torvalds 已提交
1690

1691
	trace_sched_process_wait(wo->wo_pid);
1692

1693 1694 1695
	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 已提交
1696
repeat:
R
Roland McGrath 已提交
1697 1698
	/*
	 * If there is nothing that can match our critiera just get out.
1699 1700 1701
	 * 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 已提交
1702
	 */
1703
	wo->notask_error = -ECHILD;
1704 1705
	if ((wo->wo_type < PIDTYPE_MAX) &&
	   (!wo->wo_pid || hlist_empty(&wo->wo_pid->tasks[wo->wo_type])))
1706
		goto notask;
E
Eric W. Biederman 已提交
1707

1708
	set_current_state(TASK_INTERRUPTIBLE);
L
Linus Torvalds 已提交
1709 1710 1711
	read_lock(&tasklist_lock);
	tsk = current;
	do {
1712 1713 1714
		retval = do_wait_thread(wo, tsk);
		if (retval)
			goto end;
1715

1716 1717
		retval = ptrace_do_wait(wo, tsk);
		if (retval)
R
Roland McGrath 已提交
1718 1719
			goto end;

1720
		if (wo->wo_flags & __WNOTHREAD)
L
Linus Torvalds 已提交
1721
			break;
1722
	} while_each_thread(current, tsk);
L
Linus Torvalds 已提交
1723
	read_unlock(&tasklist_lock);
O
Oleg Nesterov 已提交
1724

1725
notask:
1726 1727
	retval = wo->notask_error;
	if (!retval && !(wo->wo_flags & WNOHANG)) {
L
Linus Torvalds 已提交
1728
		retval = -ERESTARTSYS;
R
Roland McGrath 已提交
1729 1730 1731 1732
		if (!signal_pending(current)) {
			schedule();
			goto repeat;
		}
L
Linus Torvalds 已提交
1733 1734
	}
end:
1735
	__set_current_state(TASK_RUNNING);
1736
	remove_wait_queue(&current->signal->wait_chldexit, &wo->child_wait);
L
Linus Torvalds 已提交
1737 1738 1739
	return retval;
}

1740 1741
SYSCALL_DEFINE5(waitid, int, which, pid_t, upid, struct siginfo __user *,
		infop, int, options, struct rusage __user *, ru)
L
Linus Torvalds 已提交
1742
{
1743
	struct wait_opts wo;
E
Eric W. Biederman 已提交
1744 1745
	struct pid *pid = NULL;
	enum pid_type type;
L
Linus Torvalds 已提交
1746 1747 1748 1749 1750 1751 1752 1753 1754
	long ret;

	if (options & ~(WNOHANG|WNOWAIT|WEXITED|WSTOPPED|WCONTINUED))
		return -EINVAL;
	if (!(options & (WEXITED|WSTOPPED|WCONTINUED)))
		return -EINVAL;

	switch (which) {
	case P_ALL:
E
Eric W. Biederman 已提交
1755
		type = PIDTYPE_MAX;
L
Linus Torvalds 已提交
1756 1757
		break;
	case P_PID:
E
Eric W. Biederman 已提交
1758 1759
		type = PIDTYPE_PID;
		if (upid <= 0)
L
Linus Torvalds 已提交
1760 1761 1762
			return -EINVAL;
		break;
	case P_PGID:
E
Eric W. Biederman 已提交
1763 1764
		type = PIDTYPE_PGID;
		if (upid <= 0)
L
Linus Torvalds 已提交
1765 1766 1767 1768 1769 1770
			return -EINVAL;
		break;
	default:
		return -EINVAL;
	}

E
Eric W. Biederman 已提交
1771 1772
	if (type < PIDTYPE_MAX)
		pid = find_get_pid(upid);
1773 1774 1775 1776 1777 1778 1779 1780

	wo.wo_type	= type;
	wo.wo_pid	= pid;
	wo.wo_flags	= options;
	wo.wo_info	= infop;
	wo.wo_stat	= NULL;
	wo.wo_rusage	= ru;
	ret = do_wait(&wo);
1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803

	if (ret > 0) {
		ret = 0;
	} else if (infop) {
		/*
		 * For a WNOHANG return, clear out all the fields
		 * we would set so the user can easily tell the
		 * difference.
		 */
		if (!ret)
			ret = put_user(0, &infop->si_signo);
		if (!ret)
			ret = put_user(0, &infop->si_errno);
		if (!ret)
			ret = put_user(0, &infop->si_code);
		if (!ret)
			ret = put_user(0, &infop->si_pid);
		if (!ret)
			ret = put_user(0, &infop->si_uid);
		if (!ret)
			ret = put_user(0, &infop->si_status);
	}

E
Eric W. Biederman 已提交
1804
	put_pid(pid);
L
Linus Torvalds 已提交
1805 1806

	/* avoid REGPARM breakage on x86: */
1807
	asmlinkage_protect(5, ret, which, upid, infop, options, ru);
L
Linus Torvalds 已提交
1808 1809 1810
	return ret;
}

1811 1812
SYSCALL_DEFINE4(wait4, pid_t, upid, int __user *, stat_addr,
		int, options, struct rusage __user *, ru)
L
Linus Torvalds 已提交
1813
{
1814
	struct wait_opts wo;
E
Eric W. Biederman 已提交
1815 1816
	struct pid *pid = NULL;
	enum pid_type type;
L
Linus Torvalds 已提交
1817 1818 1819 1820 1821
	long ret;

	if (options & ~(WNOHANG|WUNTRACED|WCONTINUED|
			__WNOTHREAD|__WCLONE|__WALL))
		return -EINVAL;
E
Eric W. Biederman 已提交
1822 1823 1824 1825 1826 1827 1828 1829

	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;
1830
		pid = get_task_pid(current, PIDTYPE_PGID);
E
Eric W. Biederman 已提交
1831 1832 1833 1834 1835
	} else /* upid > 0 */ {
		type = PIDTYPE_PID;
		pid = find_get_pid(upid);
	}

1836 1837 1838 1839 1840 1841 1842
	wo.wo_type	= type;
	wo.wo_pid	= pid;
	wo.wo_flags	= options | WEXITED;
	wo.wo_info	= NULL;
	wo.wo_stat	= stat_addr;
	wo.wo_rusage	= ru;
	ret = do_wait(&wo);
E
Eric W. Biederman 已提交
1843
	put_pid(pid);
L
Linus Torvalds 已提交
1844 1845

	/* avoid REGPARM breakage on x86: */
1846
	asmlinkage_protect(4, ret, upid, stat_addr, options, ru);
L
Linus Torvalds 已提交
1847 1848 1849 1850 1851 1852 1853 1854 1855
	return ret;
}

#ifdef __ARCH_WANT_SYS_WAITPID

/*
 * sys_waitpid() remains for compatibility. waitpid() should be
 * implemented by calling sys_wait4() from libc.a.
 */
1856
SYSCALL_DEFINE3(waitpid, pid_t, pid, int __user *, stat_addr, int, options)
L
Linus Torvalds 已提交
1857 1858 1859 1860 1861
{
	return sys_wait4(pid, stat_addr, options, NULL);
}

#endif