exit.c 47.0 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/mnt_namespace.h>
16
#include <linux/iocontext.h>
L
Linus Torvalds 已提交
17 18 19 20
#include <linux/key.h>
#include <linux/security.h>
#include <linux/cpu.h>
#include <linux/acct.h>
21
#include <linux/tsacct_kern.h>
L
Linus Torvalds 已提交
22
#include <linux/file.h>
A
Al Viro 已提交
23
#include <linux/fdtable.h>
L
Linus Torvalds 已提交
24
#include <linux/binfmts.h>
S
Serge E. Hallyn 已提交
25
#include <linux/nsproxy.h>
26
#include <linux/pid_namespace.h>
L
Linus Torvalds 已提交
27 28 29 30
#include <linux/ptrace.h>
#include <linux/profile.h>
#include <linux/mount.h>
#include <linux/proc_fs.h>
31
#include <linux/kthread.h>
L
Linus Torvalds 已提交
32
#include <linux/mempolicy.h>
33
#include <linux/taskstats_kern.h>
34
#include <linux/delayacct.h>
35
#include <linux/freezer.h>
36
#include <linux/cgroup.h>
L
Linus Torvalds 已提交
37
#include <linux/syscalls.h>
38
#include <linux/signal.h>
39
#include <linux/posix-timers.h>
M
Matt Helsley 已提交
40
#include <linux/cn_proc.h>
41
#include <linux/mutex.h>
42
#include <linux/futex.h>
43
#include <linux/compat.h>
44
#include <linux/pipe_fs_i.h>
45
#include <linux/audit.h> /* for audit_free() */
A
Adrian Bunk 已提交
46
#include <linux/resource.h>
47
#include <linux/blkdev.h>
48
#include <linux/task_io_accounting_ops.h>
L
Linus Torvalds 已提交
49 50 51 52 53 54

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

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

57 58 59 60 61
static inline int task_detached(struct task_struct *p)
{
	return p->exit_signal == -1;
}

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

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

77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98
/*
 * This function expects the tasklist_lock write-locked.
 */
static void __exit_signal(struct task_struct *tsk)
{
	struct signal_struct *sig = tsk->signal;
	struct sighand_struct *sighand;

	BUG_ON(!sig);
	BUG_ON(!atomic_read(&sig->count));

	sighand = rcu_dereference(tsk->sighand);
	spin_lock(&sighand->siglock);

	posix_cpu_timers_exit(tsk);
	if (atomic_dec_and_test(&sig->count))
		posix_cpu_timers_exit_group(tsk);
	else {
		/*
		 * If there is any task waiting for the group exit
		 * then notify it:
		 */
99
		if (sig->group_exit_task && atomic_read(&sig->count) == sig->notify_count)
100
			wake_up_process(sig->group_exit_task);
101

102 103 104 105 106 107 108 109 110 111 112 113 114 115
		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.
		 */
		sig->utime = cputime_add(sig->utime, tsk->utime);
		sig->stime = cputime_add(sig->stime, tsk->stime);
116
		sig->gtime = cputime_add(sig->gtime, tsk->gtime);
117 118 119 120
		sig->min_flt += tsk->min_flt;
		sig->maj_flt += tsk->maj_flt;
		sig->nvcsw += tsk->nvcsw;
		sig->nivcsw += tsk->nivcsw;
121 122
		sig->inblock += task_io_get_inblock(tsk);
		sig->oublock += task_io_get_oublock(tsk);
123 124 125 126 127 128 129 130 131 132 133 134
#ifdef CONFIG_TASK_XACCT
		sig->rchar += tsk->rchar;
		sig->wchar += tsk->wchar;
		sig->syscr += tsk->syscr;
		sig->syscw += tsk->syscw;
#endif /* CONFIG_TASK_XACCT */
#ifdef CONFIG_TASK_IO_ACCOUNTING
		sig->ioac.read_bytes += tsk->ioac.read_bytes;
		sig->ioac.write_bytes += tsk->ioac.write_bytes;
		sig->ioac.cancelled_write_bytes +=
					tsk->ioac.cancelled_write_bytes;
#endif /* CONFIG_TASK_IO_ACCOUNTING */
135
		sig->sum_sched_runtime += tsk->se.sum_exec_runtime;
136 137 138
		sig = NULL; /* Marker for below. */
	}

139 140
	__unhash_process(tsk);

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

147
	tsk->signal = NULL;
148
	tsk->sighand = NULL;
149 150
	spin_unlock(&sighand->siglock);

151
	__cleanup_sighand(sighand);
152 153 154
	clear_tsk_thread_flag(tsk,TIF_SIGPENDING);
	if (sig) {
		flush_sigqueue(&sig->shared_pending);
155
		taskstats_tgid_free(sig);
156 157 158 159
		__cleanup_signal(sig);
	}
}

160 161 162 163 164
static void delayed_put_task_struct(struct rcu_head *rhp)
{
	put_task_struct(container_of(rhp, struct task_struct, rcu));
}

R
Roland McGrath 已提交
165 166 167 168 169 170 171 172 173 174 175 176
/*
 * Do final ptrace-related cleanup of a zombie being reaped.
 *
 * Called with write_lock(&tasklist_lock) held.
 */
static void ptrace_release_task(struct task_struct *p)
{
	BUG_ON(!list_empty(&p->ptraced));
	ptrace_unlink(p);
	BUG_ON(!list_empty(&p->ptrace_entry));
}

L
Linus Torvalds 已提交
177 178
void release_task(struct task_struct * p)
{
179
	struct task_struct *leader;
L
Linus Torvalds 已提交
180
	int zap_leader;
181
repeat:
L
Linus Torvalds 已提交
182
	atomic_dec(&p->user->processes);
183
	proc_flush_task(p);
L
Linus Torvalds 已提交
184
	write_lock_irq(&tasklist_lock);
R
Roland McGrath 已提交
185
	ptrace_release_task(p);
L
Linus Torvalds 已提交
186
	__exit_signal(p);
187

L
Linus Torvalds 已提交
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) {
196
		BUG_ON(task_detached(leader));
L
Linus Torvalds 已提交
197 198 199 200 201 202 203 204 205
		do_notify_parent(leader, leader->exit_signal);
		/*
		 * 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.
		 *
		 * do_notify_parent() will have marked it self-reaping in
		 * that case.
		 */
206
		zap_leader = task_detached(leader);
L
Linus Torvalds 已提交
207 208 209 210
	}

	write_unlock_irq(&tasklist_lock);
	release_thread(p);
211
	call_rcu(&p->rcu, delayed_put_task_struct);
L
Linus Torvalds 已提交
212 213 214 215 216 217 218 219 220 221

	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...
222 223
 *
 * The caller must hold rcu lock or the tasklist lock.
L
Linus Torvalds 已提交
224
 */
225
struct pid *session_of_pgrp(struct pid *pgrp)
L
Linus Torvalds 已提交
226 227
{
	struct task_struct *p;
228
	struct pid *sid = NULL;
229

230
	p = pid_task(pgrp, PIDTYPE_PGID);
231
	if (p == NULL)
232
		p = pid_task(pgrp, PIDTYPE_PID);
233
	if (p != NULL)
234
		sid = task_session(p);
235

L
Linus Torvalds 已提交
236 237 238 239 240 241 242 243 244 245 246
	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?"
 */
247
static int will_become_orphaned_pgrp(struct pid *pgrp, struct task_struct *ignored_task)
L
Linus Torvalds 已提交
248 249 250
{
	struct task_struct *p;

251
	do_each_pid_task(pgrp, PIDTYPE_PGID, p) {
252 253 254
		if ((p == ignored_task) ||
		    (p->exit_state && thread_group_empty(p)) ||
		    is_global_init(p->real_parent))
L
Linus Torvalds 已提交
255
			continue;
256

257
		if (task_pgrp(p->real_parent) != pgrp &&
258 259
		    task_session(p->real_parent) == task_session(p))
			return 0;
260
	} while_each_pid_task(pgrp, PIDTYPE_PGID, p);
261 262

	return 1;
L
Linus Torvalds 已提交
263 264
}

265
int is_current_pgrp_orphaned(void)
L
Linus Torvalds 已提交
266 267 268 269
{
	int retval;

	read_lock(&tasklist_lock);
270
	retval = will_become_orphaned_pgrp(task_pgrp(current), NULL);
L
Linus Torvalds 已提交
271 272 273 274 275
	read_unlock(&tasklist_lock);

	return retval;
}

276
static int has_stopped_jobs(struct pid *pgrp)
L
Linus Torvalds 已提交
277 278 279 280
{
	int retval = 0;
	struct task_struct *p;

281
	do_each_pid_task(pgrp, PIDTYPE_PGID, p) {
M
Matthew Wilcox 已提交
282
		if (!task_is_stopped(p))
L
Linus Torvalds 已提交
283 284 285
			continue;
		retval = 1;
		break;
286
	} while_each_pid_task(pgrp, PIDTYPE_PGID, p);
L
Linus Torvalds 已提交
287 288 289
	return retval;
}

290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320
/*
 * 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 已提交
321
/**
322
 * reparent_to_kthreadd - Reparent the calling kernel thread to kthreadd
L
Linus Torvalds 已提交
323 324
 *
 * If a kernel thread is launched as a result of a system call, or if
325 326
 * 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 已提交
327 328 329 330
 *
 * 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.
 *
331
 * NOTE that reparent_to_kthreadd() gives the caller full capabilities.
L
Linus Torvalds 已提交
332
 */
333
static void reparent_to_kthreadd(void)
L
Linus Torvalds 已提交
334 335 336 337 338
{
	write_lock_irq(&tasklist_lock);

	ptrace_unlink(current);
	/* Reparent to init */
339
	current->real_parent = current->parent = kthreadd_task;
R
Roland McGrath 已提交
340
	list_move_tail(&current->sibling, &current->real_parent->children);
L
Linus Torvalds 已提交
341 342 343 344

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

345
	if (task_nice(current) < 0)
L
Linus Torvalds 已提交
346 347 348 349 350 351 352 353 354 355 356 357
		set_user_nice(current, 0);
	/* cpus_allowed? */
	/* rt_priority? */
	/* signals? */
	security_task_reparent_to_init(current);
	memcpy(current->signal->rlim, init_task.signal->rlim,
	       sizeof(current->signal->rlim));
	atomic_inc(&(INIT_USER->__count));
	write_unlock_irq(&tasklist_lock);
	switch_uid(INIT_USER);
}

358
void __set_special_pids(struct pid *pid)
L
Linus Torvalds 已提交
359
{
360
	struct task_struct *curr = current->group_leader;
361
	pid_t nr = pid_nr(pid);
L
Linus Torvalds 已提交
362

363
	if (task_session(curr) != pid) {
364
		change_pid(curr, PIDTYPE_SID, pid);
365
		set_task_session(curr, nr);
L
Linus Torvalds 已提交
366
	}
367
	if (task_pgrp(curr) != pid) {
368
		change_pid(curr, PIDTYPE_PGID, pid);
369
		set_task_pgrp(curr, nr);
L
Linus Torvalds 已提交
370 371 372
	}
}

373
static void set_special_pids(struct pid *pid)
L
Linus Torvalds 已提交
374 375
{
	write_lock_irq(&tasklist_lock);
376
	__set_special_pids(pid);
L
Linus Torvalds 已提交
377 378 379 380 381 382 383 384 385 386
	write_unlock_irq(&tasklist_lock);
}

/*
 * Let kernel threads use this to say that they
 * allow a certain signal (since daemonize() will
 * have disabled all of them by default).
 */
int allow_signal(int sig)
{
387
	if (!valid_signal(sig) || sig < 1)
L
Linus Torvalds 已提交
388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407
		return -EINVAL;

	spin_lock_irq(&current->sighand->siglock);
	sigdelset(&current->blocked, sig);
	if (!current->mm) {
		/* 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;
	}
	recalc_sigpending();
	spin_unlock_irq(&current->sighand->siglock);
	return 0;
}

EXPORT_SYMBOL(allow_signal);

int disallow_signal(int sig)
{
408
	if (!valid_signal(sig) || sig < 1)
L
Linus Torvalds 已提交
409 410 411
		return -EINVAL;

	spin_lock_irq(&current->sighand->siglock);
412
	current->sighand->action[(sig)-1].sa.sa_handler = SIG_IGN;
L
Linus Torvalds 已提交
413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440
	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;
	struct fs_struct *fs;
	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);
441 442 443 444
	/*
	 * We don't want to have TIF_FREEZE set if the system-wide hibernation
	 * or suspend transition begins right now.
	 */
O
Oleg Nesterov 已提交
445
	current->flags |= (PF_NOFREEZE | PF_KTHREAD);
L
Linus Torvalds 已提交
446

447 448 449 450
	if (current->nsproxy != &init_nsproxy) {
		get_nsproxy(&init_nsproxy);
		switch_task_namespaces(current, &init_nsproxy);
	}
451
	set_special_pids(&init_struct_pid);
452
	proc_clear_tty(current);
L
Linus Torvalds 已提交
453 454 455 456 457 458 459 460 461 462 463 464

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

	/* Become as one with the init task */

	exit_fs(current);	/* current->fs->count--; */
	fs = init_task.fs;
	current->fs = fs;
	atomic_inc(&fs->count);
S
Serge E. Hallyn 已提交
465

466
	exit_files(current);
L
Linus Torvalds 已提交
467 468 469
	current->files = init_task.files;
	atomic_inc(&current->files->count);

470
	reparent_to_kthreadd();
L
Linus Torvalds 已提交
471 472 473 474
}

EXPORT_SYMBOL(daemonize);

475
static void close_files(struct files_struct * files)
L
Linus Torvalds 已提交
476 477
{
	int i, j;
478
	struct fdtable *fdt;
L
Linus Torvalds 已提交
479 480

	j = 0;
481 482 483 484 485 486

	/*
	 * It is safe to dereference the fd table without RCU or
	 * ->file_lock because this is the last reference to the
	 * files structure.
	 */
487
	fdt = files_fdtable(files);
L
Linus Torvalds 已提交
488 489 490
	for (;;) {
		unsigned long set;
		i = j * __NFDBITS;
491
		if (i >= fdt->max_fds)
L
Linus Torvalds 已提交
492
			break;
493
		set = fdt->open_fds->fds_bits[j++];
L
Linus Torvalds 已提交
494 495
		while (set) {
			if (set & 1) {
496
				struct file * file = xchg(&fdt->fd[i], NULL);
497
				if (file) {
L
Linus Torvalds 已提交
498
					filp_close(file, files);
499 500
					cond_resched();
				}
L
Linus Torvalds 已提交
501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520
			}
			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;
}

521
void put_files_struct(struct files_struct *files)
L
Linus Torvalds 已提交
522
{
523 524
	struct fdtable *fdt;

L
Linus Torvalds 已提交
525 526 527 528
	if (atomic_dec_and_test(&files->count)) {
		close_files(files);
		/*
		 * Free the fd and fdset arrays if we expanded them.
529 530 531
		 * 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 已提交
532
		 */
533
		fdt = files_fdtable(files);
534
		if (fdt != &files->fdtab)
535
			kmem_cache_free(files_cachep, files);
536
		free_fdtable(fdt);
L
Linus Torvalds 已提交
537 538 539
	}
}

540
void reset_files_struct(struct files_struct *files)
541
{
542
	struct task_struct *tsk = current;
543 544 545 546 547 548 549 550 551
	struct files_struct *old;

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

552
void exit_files(struct task_struct *tsk)
L
Linus Torvalds 已提交
553 554 555 556 557 558 559 560 561 562 563
{
	struct files_struct * files = tsk->files;

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

564
void put_fs_struct(struct fs_struct *fs)
L
Linus Torvalds 已提交
565 566 567
{
	/* No need to hold fs->lock if we are killing it */
	if (atomic_dec_and_test(&fs->count)) {
J
Jan Blunck 已提交
568 569 570 571
		path_put(&fs->root);
		path_put(&fs->pwd);
		if (fs->altroot.dentry)
			path_put(&fs->altroot);
L
Linus Torvalds 已提交
572 573 574 575
		kmem_cache_free(fs_cachep, fs);
	}
}

576
void exit_fs(struct task_struct *tsk)
L
Linus Torvalds 已提交
577 578 579 580 581 582 583
{
	struct fs_struct * fs = tsk->fs;

	if (fs) {
		task_lock(tsk);
		tsk->fs = NULL;
		task_unlock(tsk);
584
		put_fs_struct(fs);
L
Linus Torvalds 已提交
585 586 587 588 589
	}
}

EXPORT_SYMBOL_GPL(exit_fs);

590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671
#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 (!mm)
		return 0;
	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
	 */
	list_for_each_entry(c, &p->parent->children, sibling) {
		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);
	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);
	/*
	 * Delay read_unlock() till we have the task_lock()
	 * to ensure that c does not slip away underneath us
	 */
	read_unlock(&tasklist_lock);
	if (c->mm != mm) {
		task_unlock(c);
		put_task_struct(c);
		goto retry;
	}
	cgroup_mm_owner_callbacks(mm->owner, c);
	mm->owner = c;
	task_unlock(c);
	put_task_struct(c);
}
#endif /* CONFIG_MM_OWNER */

L
Linus Torvalds 已提交
672 673 674 675
/*
 * Turn us into a lazy TLB process if we
 * aren't already..
 */
A
Adrian Bunk 已提交
676
static void exit_mm(struct task_struct * tsk)
L
Linus Torvalds 已提交
677 678
{
	struct mm_struct *mm = tsk->mm;
679
	struct core_state *core_state;
L
Linus Torvalds 已提交
680 681 682 683 684 685

	mm_release(tsk, mm);
	if (!mm)
		return;
	/*
	 * Serialize with any possible pending coredump.
686
	 * We must hold mmap_sem around checking core_state
L
Linus Torvalds 已提交
687
	 * and clearing tsk->mm.  The core-inducing thread
688
	 * will increment ->nr_threads for each thread in the
L
Linus Torvalds 已提交
689 690 691
	 * group with ->mm != NULL.
	 */
	down_read(&mm->mmap_sem);
692 693 694
	core_state = mm->core_state;
	if (core_state) {
		struct core_thread self;
L
Linus Torvalds 已提交
695
		up_read(&mm->mmap_sem);
696

697 698 699 700 701 702 703 704
		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 已提交
705

O
Oleg Nesterov 已提交
706 707 708 709 710 711 712
		for (;;) {
			set_task_state(tsk, TASK_UNINTERRUPTIBLE);
			if (!self.task) /* see coredump_finish() */
				break;
			schedule();
		}
		__set_task_state(tsk, TASK_RUNNING);
L
Linus Torvalds 已提交
713 714 715
		down_read(&mm->mmap_sem);
	}
	atomic_inc(&mm->mm_count);
E
Eric Sesterhenn 已提交
716
	BUG_ON(mm != tsk->active_mm);
L
Linus Torvalds 已提交
717 718 719 720 721
	/* more a memory barrier than a real lock */
	task_lock(tsk);
	tsk->mm = NULL;
	up_read(&mm->mmap_sem);
	enter_lazy_tlb(mm, current);
722 723
	/* We don't want this task to be frozen prematurely */
	clear_freeze_flag(tsk);
L
Linus Torvalds 已提交
724
	task_unlock(tsk);
725
	mm_update_next_owner(mm);
L
Linus Torvalds 已提交
726 727 728
	mmput(mm);
}

729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745
/*
 * Return nonzero if @parent's children should reap themselves.
 *
 * Called with write_lock_irq(&tasklist_lock) held.
 */
static int ignoring_children(struct task_struct *parent)
{
	int ret;
	struct sighand_struct *psig = parent->sighand;
	unsigned long flags;
	spin_lock_irqsave(&psig->siglock, flags);
	ret = (psig->action[SIGCHLD-1].sa.sa_handler == SIG_IGN ||
	       (psig->action[SIGCHLD-1].sa.sa_flags & SA_NOCLDWAIT));
	spin_unlock_irqrestore(&psig->siglock, flags);
	return ret;
}

R
Roland McGrath 已提交
746 747 748 749 750 751 752
/*
 * Detach all tasks we were using ptrace on.
 * Any that need to be release_task'd are put on the @dead list.
 *
 * Called with write_lock(&tasklist_lock) held.
 */
static void ptrace_exit(struct task_struct *parent, struct list_head *dead)
L
Linus Torvalds 已提交
753
{
R
Roland McGrath 已提交
754
	struct task_struct *p, *n;
755
	int ign = -1;
756

R
Roland McGrath 已提交
757 758 759 760 761 762 763 764 765 766 767 768 769 770
	list_for_each_entry_safe(p, n, &parent->ptraced, ptrace_entry) {
		__ptrace_unlink(p);

		if (p->exit_state != EXIT_ZOMBIE)
			continue;

		/*
		 * If it's a zombie, our attachedness prevented normal
		 * parent notification or self-reaping.  Do notification
		 * now if it would have happened earlier.  If it should
		 * reap itself, add it to the @dead list.  We can't call
		 * release_task() here because we already hold tasklist_lock.
		 *
		 * If it's our own child, there is no notification to do.
771 772
		 * But if our normal children self-reap, then this child
		 * was prevented by ptrace and we must reap it now.
L
Linus Torvalds 已提交
773
		 */
R
Roland McGrath 已提交
774 775 776
		if (!task_detached(p) && thread_group_empty(p)) {
			if (!same_thread_group(p->real_parent, parent))
				do_notify_parent(p, p->exit_signal);
777 778 779 780 781 782
			else {
				if (ign < 0)
					ign = ignoring_children(parent);
				if (ign)
					p->exit_signal = -1;
			}
R
Roland McGrath 已提交
783
		}
L
Linus Torvalds 已提交
784

R
Roland McGrath 已提交
785
		if (task_detached(p)) {
L
Linus Torvalds 已提交
786
			/*
R
Roland McGrath 已提交
787
			 * Mark it as in the process of being reaped.
L
Linus Torvalds 已提交
788
			 */
R
Roland McGrath 已提交
789 790
			p->exit_state = EXIT_DEAD;
			list_add(&p->ptrace_entry, dead);
L
Linus Torvalds 已提交
791 792
		}
	}
R
Roland McGrath 已提交
793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819
}

/*
 * Finish up exit-time ptrace cleanup.
 *
 * Called without locks.
 */
static void ptrace_exit_finish(struct task_struct *parent,
			       struct list_head *dead)
{
	struct task_struct *p, *n;

	BUG_ON(!list_empty(&parent->ptraced));

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

static void reparent_thread(struct task_struct *p, struct task_struct *father)
{
	if (p->pdeath_signal)
		/* We already hold the tasklist_lock here.  */
		group_send_sig_info(p->pdeath_signal, SEND_SIG_NOINFO, p);

	list_move_tail(&p->sibling, &p->real_parent->children);
L
Linus Torvalds 已提交
820

821 822 823
	/* If this is a threaded reparent there is no need to
	 * notify anyone anything has happened.
	 */
824
	if (same_thread_group(p->real_parent, father))
825 826 827
		return;

	/* We don't want people slaying init.  */
828
	if (!task_detached(p))
829 830 831 832 833
		p->exit_signal = SIGCHLD;

	/* If we'd notified the old parent about this child's death,
	 * also notify the new parent.
	 */
R
Roland McGrath 已提交
834 835
	if (!ptrace_reparented(p) &&
	    p->exit_state == EXIT_ZOMBIE &&
836
	    !task_detached(p) && thread_group_empty(p))
837 838
		do_notify_parent(p, p->exit_signal);

839
	kill_orphaned_pgrp(p, father);
L
Linus Torvalds 已提交
840 841 842 843 844 845
}

/*
 * 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
846 847
 * the child reaper process (ie "init") in our pid
 * space.
L
Linus Torvalds 已提交
848
 */
849
static void forget_original_parent(struct task_struct *father)
L
Linus Torvalds 已提交
850
{
851
	struct task_struct *p, *n, *reaper = father;
R
Roland McGrath 已提交
852
	LIST_HEAD(ptrace_dead);
853 854

	write_lock_irq(&tasklist_lock);
L
Linus Torvalds 已提交
855

R
Roland McGrath 已提交
856 857 858 859 860
	/*
	 * First clean up ptrace if we were using it.
	 */
	ptrace_exit(father, &ptrace_dead);

L
Linus Torvalds 已提交
861 862 863
	do {
		reaper = next_thread(reaper);
		if (reaper == father) {
864
			reaper = task_child_reaper(father);
L
Linus Torvalds 已提交
865 866
			break;
		}
867
	} while (reaper->flags & PF_EXITING);
L
Linus Torvalds 已提交
868

869
	list_for_each_entry_safe(p, n, &father->children, sibling) {
870
		p->real_parent = reaper;
R
Roland McGrath 已提交
871 872 873 874 875
		if (p->parent == father) {
			BUG_ON(p->ptrace);
			p->parent = p->real_parent;
		}
		reparent_thread(p, father);
L
Linus Torvalds 已提交
876
	}
877 878 879 880

	write_unlock_irq(&tasklist_lock);
	BUG_ON(!list_empty(&father->children));

R
Roland McGrath 已提交
881
	ptrace_exit_finish(father, &ptrace_dead);
L
Linus Torvalds 已提交
882 883 884 885 886 887
}

/*
 * Send signals to all our closest relatives so that they know
 * to properly mourn us..
 */
888
static void exit_notify(struct task_struct *tsk, int group_dead)
L
Linus Torvalds 已提交
889 890 891 892 893 894 895 896 897 898 899
{
	int state;

	/*
	 * 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)
	 */
900
	forget_original_parent(tsk);
901
	exit_task_namespaces(tsk);
L
Linus Torvalds 已提交
902

903
	write_lock_irq(&tasklist_lock);
904 905
	if (group_dead)
		kill_orphaned_pgrp(tsk->group_leader, NULL);
L
Linus Torvalds 已提交
906

907
	/* Let father know we died
L
Linus Torvalds 已提交
908 909
	 *
	 * Thread signals are configurable, but you aren't going to use
910
	 * that to send signals to arbitary processes.
L
Linus Torvalds 已提交
911 912 913 914 915 916 917 918 919 920
	 * 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.
	 */
921
	if (tsk->exit_signal != SIGCHLD && !task_detached(tsk) &&
922
	    (tsk->parent_exec_id != tsk->real_parent->self_exec_id ||
923 924
	     tsk->self_exec_id != tsk->parent_exec_id) &&
	    !capable(CAP_KILL))
L
Linus Torvalds 已提交
925 926 927 928 929 930
		tsk->exit_signal = SIGCHLD;

	/* If something other than our normal parent is ptracing us, then
	 * send it a SIGCHLD instead of honoring exit_signal.  exit_signal
	 * only has special meaning to our real parent.
	 */
931
	if (!task_detached(tsk) && thread_group_empty(tsk)) {
932 933
		int signal = ptrace_reparented(tsk) ?
				SIGCHLD : tsk->exit_signal;
L
Linus Torvalds 已提交
934 935 936 937 938 939
		do_notify_parent(tsk, signal);
	} else if (tsk->ptrace) {
		do_notify_parent(tsk, SIGCHLD);
	}

	state = EXIT_ZOMBIE;
940
	if (task_detached(tsk) && likely(!tsk->ptrace))
L
Linus Torvalds 已提交
941 942 943
		state = EXIT_DEAD;
	tsk->exit_state = state;

944
	/* mt-exec, de_thread() is waiting for us */
945 946 947 948 949
	if (thread_group_leader(tsk) &&
	    tsk->signal->notify_count < 0 &&
	    tsk->signal->group_exit_task)
		wake_up_process(tsk->signal->group_exit_task);

L
Linus Torvalds 已提交
950 951 952 953 954 955 956
	write_unlock_irq(&tasklist_lock);

	/* If the process is dead, release it - nobody will wait for it */
	if (state == EXIT_DEAD)
		release_task(tsk);
}

957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984
#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 *n = end_of_stack(current);
	unsigned long free;

	while (*n == 0)
		n++;
	free = (unsigned long)n - (unsigned long)end_of_stack(current);

	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

985 986
static inline void exit_child_reaper(struct task_struct *tsk)
{
987
	if (likely(tsk->group_leader != task_child_reaper(tsk)))
988 989
		return;

990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015
	if (tsk->nsproxy->pid_ns == &init_pid_ns)
		panic("Attempted to kill init!");

	/*
	 * @tsk is the last thread in the 'cgroup-init' and is exiting.
	 * Terminate all remaining processes in the namespace and reap them
	 * before exiting @tsk.
	 *
	 * Note that @tsk (last thread of cgroup-init) may not necessarily
	 * be the child-reaper (i.e main thread of cgroup-init) of the
	 * namespace i.e the child_reaper may have already exited.
	 *
	 * Even after a child_reaper exits, we let it inherit orphaned children,
	 * because, pid_ns->child_reaper remains valid as long as there is
	 * at least one living sub-thread in the cgroup init.

	 * This living sub-thread of the cgroup-init will be notified when
	 * a child inherited by the 'child-reaper' exits (do_notify_parent()
	 * uses __group_send_sig_info()). Further, when reaping child processes,
	 * do_wait() iterates over children of all living sub threads.

	 * i.e even though 'child_reaper' thread is listed as the parent of the
	 * orphaned children, any living sub-thread in the cgroup-init can
	 * perform the role of the child_reaper.
	 */
	zap_pid_ns_processes(tsk->nsproxy->pid_ns);
1016 1017
}

1018
NORET_TYPE void do_exit(long code)
L
Linus Torvalds 已提交
1019 1020 1021 1022 1023 1024
{
	struct task_struct *tsk = current;
	int group_dead;

	profile_task_exit(tsk);

1025 1026
	WARN_ON(atomic_read(&tsk->fs_excl));

L
Linus Torvalds 已提交
1027 1028 1029 1030 1031 1032 1033 1034 1035 1036
	if (unlikely(in_interrupt()))
		panic("Aiee, killing interrupt handler!");
	if (unlikely(!tsk->pid))
		panic("Attempted to kill the idle task!");

	if (unlikely(current->ptrace & PT_TRACE_EXIT)) {
		current->ptrace_message = code;
		ptrace_notify((PTRACE_EVENT_EXIT << 8) | SIGTRAP);
	}

1037 1038 1039 1040 1041 1042 1043
	/*
	 * 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");
1044 1045 1046 1047 1048 1049 1050 1051 1052 1053
		/*
		 * 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;
1054 1055
		if (tsk->io_context)
			exit_io_context();
1056 1057 1058 1059
		set_current_state(TASK_UNINTERRUPTIBLE);
		schedule();
	}

O
Oleg Nesterov 已提交
1060
	exit_signals(tsk);  /* sets PF_EXITING */
1061 1062 1063 1064
	/*
	 * tsk->flags are checked in the futex code to protect against
	 * an exiting task cleaning up the robust pi futexes.
	 */
1065 1066
	smp_mb();
	spin_unlock_wait(&tsk->pi_lock);
L
Linus Torvalds 已提交
1067 1068 1069

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

	acct_update_integrals(tsk);
1074 1075 1076 1077
	if (tsk->mm) {
		update_hiwater_rss(tsk->mm);
		update_hiwater_vm(tsk->mm);
	}
L
Linus Torvalds 已提交
1078
	group_dead = atomic_dec_and_test(&tsk->signal->live);
1079
	if (group_dead) {
1080
		exit_child_reaper(tsk);
1081
		hrtimer_cancel(&tsk->signal->real_timer);
1082
		exit_itimers(tsk->signal);
1083
	}
1084
	acct_collect(code, group_dead);
1085
#ifdef CONFIG_FUTEX
1086 1087
	if (unlikely(tsk->robust_list))
		exit_robust_list(tsk);
1088
#ifdef CONFIG_COMPAT
1089 1090
	if (unlikely(tsk->compat_robust_list))
		compat_exit_robust_list(tsk);
1091
#endif
1092
#endif
M
Miloslav Trmac 已提交
1093 1094
	if (group_dead)
		tty_audit_exit();
1095 1096
	if (unlikely(tsk->audit_context))
		audit_free(tsk);
1097

1098
	tsk->exit_code = code;
1099
	taskstats_exit(tsk, group_dead);
1100

L
Linus Torvalds 已提交
1101 1102
	exit_mm(tsk);

1103
	if (group_dead)
1104
		acct_process();
L
Linus Torvalds 已提交
1105
	exit_sem(tsk);
1106 1107
	exit_files(tsk);
	exit_fs(tsk);
1108
	check_stack_usage();
L
Linus Torvalds 已提交
1109
	exit_thread();
1110
	cgroup_exit(tsk, 1);
L
Linus Torvalds 已提交
1111 1112 1113 1114 1115
	exit_keys(tsk);

	if (group_dead && tsk->signal->leader)
		disassociate_ctty(1);

A
Al Viro 已提交
1116
	module_put(task_thread_info(tsk)->exec_domain->module);
L
Linus Torvalds 已提交
1117 1118 1119
	if (tsk->binfmt)
		module_put(tsk->binfmt->module);

M
Matt Helsley 已提交
1120
	proc_exit_connector(tsk);
1121
	exit_notify(tsk, group_dead);
L
Linus Torvalds 已提交
1122
#ifdef CONFIG_NUMA
1123
	mpol_put(tsk->mempolicy);
L
Linus Torvalds 已提交
1124 1125
	tsk->mempolicy = NULL;
#endif
1126
#ifdef CONFIG_FUTEX
1127 1128 1129 1130 1131 1132 1133 1134
	/*
	 * This must happen late, after the PID is not
	 * hashed anymore:
	 */
	if (unlikely(!list_empty(&tsk->pi_state_list)))
		exit_pi_state_list(tsk);
	if (unlikely(current->pi_state_cache))
		kfree(current->pi_state_cache);
1135
#endif
1136
	/*
1137
	 * Make sure we are holding no locks:
1138
	 */
1139
	debug_check_no_locks_held(tsk);
1140 1141 1142 1143 1144 1145
	/*
	 * 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 已提交
1146

1147 1148 1149
	if (tsk->io_context)
		exit_io_context();

1150 1151 1152
	if (tsk->splice_pipe)
		__free_pipe_info(tsk->splice_pipe);

C
Coywolf Qi Hunt 已提交
1153
	preempt_disable();
O
Oleg Nesterov 已提交
1154
	/* causes final put_task_struct in finish_task_switch(). */
1155
	tsk->state = TASK_DEAD;
C
Coywolf Qi Hunt 已提交
1156

L
Linus Torvalds 已提交
1157 1158 1159
	schedule();
	BUG();
	/* Avoid "noreturn function does return".  */
A
Alan Cox 已提交
1160 1161
	for (;;)
		cpu_relax();	/* For when BUG is null */
L
Linus Torvalds 已提交
1162 1163
}

1164 1165
EXPORT_SYMBOL_GPL(do_exit);

L
Linus Torvalds 已提交
1166 1167 1168 1169
NORET_TYPE void complete_and_exit(struct completion *comp, long code)
{
	if (comp)
		complete(comp);
O
Oleg Nesterov 已提交
1170

L
Linus Torvalds 已提交
1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187
	do_exit(code);
}

EXPORT_SYMBOL(complete_and_exit);

asmlinkage long sys_exit(int error_code)
{
	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)
{
1188 1189
	struct signal_struct *sig = current->signal;

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

1192 1193
	if (signal_group_exit(sig))
		exit_code = sig->group_exit_code;
L
Linus Torvalds 已提交
1194 1195 1196
	else if (!thread_group_empty(current)) {
		struct sighand_struct *const sighand = current->sighand;
		spin_lock_irq(&sighand->siglock);
1197
		if (signal_group_exit(sig))
L
Linus Torvalds 已提交
1198 1199 1200 1201
			/* Another thread got here before we took the lock.  */
			exit_code = sig->group_exit_code;
		else {
			sig->group_exit_code = exit_code;
1202
			sig->flags = SIGNAL_GROUP_EXIT;
L
Linus Torvalds 已提交
1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221
			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.
 */
asmlinkage void sys_exit_group(int error_code)
{
	do_group_exit((error_code & 0xff) << 8);
}

E
Eric W. Biederman 已提交
1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233
static struct pid *task_pid_type(struct task_struct *task, enum pid_type type)
{
	struct pid *pid = NULL;
	if (type == PIDTYPE_PID)
		pid = task->pids[type].pid;
	else if (type < PIDTYPE_MAX)
		pid = task->group_leader->pids[type].pid;
	return pid;
}

static int eligible_child(enum pid_type type, struct pid *pid, int options,
			  struct task_struct *p)
L
Linus Torvalds 已提交
1234
{
1235 1236
	int err;

E
Eric W. Biederman 已提交
1237 1238
	if (type < PIDTYPE_MAX) {
		if (task_pid_type(p, type) != pid)
L
Linus Torvalds 已提交
1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250
			return 0;
	}

	/* 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.) */
	if (((p->exit_signal != SIGCHLD) ^ ((options & __WCLONE) != 0))
	    && !(options & __WALL))
		return 0;

1251
	err = security_task_wait(p);
1252 1253
	if (err)
		return err;
L
Linus Torvalds 已提交
1254

1255
	return 1;
L
Linus Torvalds 已提交
1256 1257
}

1258
static int wait_noreap_copyout(struct task_struct *p, pid_t pid, uid_t uid,
L
Linus Torvalds 已提交
1259 1260 1261 1262 1263
			       int why, int status,
			       struct siginfo __user *infop,
			       struct rusage __user *rusagep)
{
	int retval = rusagep ? getrusage(p, RUSAGE_BOTH, rusagep) : 0;
1264

L
Linus Torvalds 已提交
1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288
	put_task_struct(p);
	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);
	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.
 */
R
Roland McGrath 已提交
1289
static int wait_task_zombie(struct task_struct *p, int options,
L
Linus Torvalds 已提交
1290 1291 1292 1293
			    struct siginfo __user *infop,
			    int __user *stat_addr, struct rusage __user *ru)
{
	unsigned long state;
1294
	int retval, status, traced;
1295
	pid_t pid = task_pid_vnr(p);
L
Linus Torvalds 已提交
1296

R
Roland McGrath 已提交
1297 1298 1299 1300
	if (!likely(options & WEXITED))
		return 0;

	if (unlikely(options & WNOWAIT)) {
L
Linus Torvalds 已提交
1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327
		uid_t uid = p->uid;
		int exit_code = p->exit_code;
		int why, status;

		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;
		}
		return wait_noreap_copyout(p, pid, uid, why,
					   status, infop, ru);
	}

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

1328
	traced = ptrace_reparented(p);
1329 1330

	if (likely(!traced)) {
1331 1332 1333
		struct signal_struct *psig;
		struct signal_struct *sig;

L
Linus Torvalds 已提交
1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349
		/*
		 * 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
		 * need to protect the access to p->parent->signal fields,
		 * as other threads in the parent group can be right
		 * here reaping other children at the same time.
		 */
		spin_lock_irq(&p->parent->sighand->siglock);
1350 1351 1352 1353
		psig = p->parent->signal;
		sig = p->signal;
		psig->cutime =
			cputime_add(psig->cutime,
L
Linus Torvalds 已提交
1354
			cputime_add(p->utime,
1355 1356 1357 1358
			cputime_add(sig->utime,
				    sig->cutime)));
		psig->cstime =
			cputime_add(psig->cstime,
L
Linus Torvalds 已提交
1359
			cputime_add(p->stime,
1360 1361
			cputime_add(sig->stime,
				    sig->cstime)));
1362 1363 1364 1365 1366
		psig->cgtime =
			cputime_add(psig->cgtime,
			cputime_add(p->gtime,
			cputime_add(sig->gtime,
				    sig->cgtime)));
1367 1368 1369 1370 1371 1372 1373 1374
		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;
1375 1376 1377 1378 1379 1380
		psig->cinblock +=
			task_io_get_inblock(p) +
			sig->inblock + sig->cinblock;
		psig->coublock +=
			task_io_get_oublock(p) +
			sig->oublock + sig->coublock;
1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395
#ifdef CONFIG_TASK_XACCT
		psig->rchar += p->rchar + sig->rchar;
		psig->wchar += p->wchar + sig->wchar;
		psig->syscr += p->syscr + sig->syscr;
		psig->syscw += p->syscw + sig->syscw;
#endif /* CONFIG_TASK_XACCT */
#ifdef CONFIG_TASK_IO_ACCOUNTING
		psig->ioac.read_bytes +=
			p->ioac.read_bytes + sig->ioac.read_bytes;
		psig->ioac.write_bytes +=
			p->ioac.write_bytes + sig->ioac.write_bytes;
		psig->ioac.cancelled_write_bytes +=
				p->ioac.cancelled_write_bytes +
				sig->ioac.cancelled_write_bytes;
#endif /* CONFIG_TASK_IO_ACCOUNTING */
L
Linus Torvalds 已提交
1396 1397 1398 1399 1400 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 1427 1428
		spin_unlock_irq(&p->parent->sighand->siglock);
	}

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

	retval = ru ? getrusage(p, RUSAGE_BOTH, ru) : 0;
	status = (p->signal->flags & SIGNAL_GROUP_EXIT)
		? p->signal->group_exit_code : p->exit_code;
	if (!retval && stat_addr)
		retval = put_user(status, stat_addr);
	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)
1429
		retval = put_user(pid, &infop->si_pid);
L
Linus Torvalds 已提交
1430 1431
	if (!retval && infop)
		retval = put_user(p->uid, &infop->si_uid);
1432
	if (!retval)
1433
		retval = pid;
1434 1435

	if (traced) {
L
Linus Torvalds 已提交
1436
		write_lock_irq(&tasklist_lock);
1437 1438 1439 1440 1441 1442 1443
		/* We dropped tasklist, ptracer could die and untrace */
		ptrace_unlink(p);
		/*
		 * If this is not a detached task, notify the parent.
		 * If it's still not detached after that, don't release
		 * it now.
		 */
1444
		if (!task_detached(p)) {
1445
			do_notify_parent(p, p->exit_signal);
1446
			if (!task_detached(p)) {
1447 1448
				p->exit_state = EXIT_ZOMBIE;
				p = NULL;
L
Linus Torvalds 已提交
1449 1450 1451 1452 1453 1454
			}
		}
		write_unlock_irq(&tasklist_lock);
	}
	if (p != NULL)
		release_task(p);
1455

L
Linus Torvalds 已提交
1456 1457 1458 1459 1460 1461 1462 1463 1464
	return retval;
}

/*
 * Handle sys_wait4 work for one task in state TASK_STOPPED.  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.
 */
R
Roland McGrath 已提交
1465
static int wait_task_stopped(int ptrace, struct task_struct *p,
R
Roland McGrath 已提交
1466
			     int options, struct siginfo __user *infop,
L
Linus Torvalds 已提交
1467 1468
			     int __user *stat_addr, struct rusage __user *ru)
{
1469 1470
	int retval, exit_code, why;
	uid_t uid = 0; /* unneeded, required by compiler */
1471
	pid_t pid;
L
Linus Torvalds 已提交
1472

R
Roland McGrath 已提交
1473
	if (!(options & WUNTRACED))
R
Roland McGrath 已提交
1474 1475
		return 0;

1476 1477 1478 1479 1480 1481
	exit_code = 0;
	spin_lock_irq(&p->sighand->siglock);

	if (unlikely(!task_is_stopped_or_traced(p)))
		goto unlock_sig;

R
Roland McGrath 已提交
1482
	if (!ptrace && p->signal->group_stop_count > 0)
L
Linus Torvalds 已提交
1483 1484 1485 1486
		/*
		 * A group stop is in progress and this is the group leader.
		 * We won't report until all threads have stopped.
		 */
1487 1488 1489 1490 1491 1492
		goto unlock_sig;

	exit_code = p->exit_code;
	if (!exit_code)
		goto unlock_sig;

R
Roland McGrath 已提交
1493
	if (!unlikely(options & WNOWAIT))
1494 1495 1496 1497 1498 1499
		p->exit_code = 0;

	uid = p->uid;
unlock_sig:
	spin_unlock_irq(&p->sighand->siglock);
	if (!exit_code)
L
Linus Torvalds 已提交
1500 1501 1502 1503 1504 1505 1506 1507 1508 1509
		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);
1510
	pid = task_pid_vnr(p);
R
Roland McGrath 已提交
1511
	why = ptrace ? CLD_TRAPPED : CLD_STOPPED;
L
Linus Torvalds 已提交
1512 1513
	read_unlock(&tasklist_lock);

R
Roland McGrath 已提交
1514
	if (unlikely(options & WNOWAIT))
L
Linus Torvalds 已提交
1515
		return wait_noreap_copyout(p, pid, uid,
1516
					   why, exit_code,
L
Linus Torvalds 已提交
1517 1518 1519 1520 1521 1522 1523 1524 1525 1526
					   infop, ru);

	retval = ru ? getrusage(p, RUSAGE_BOTH, ru) : 0;
	if (!retval && stat_addr)
		retval = put_user((exit_code << 8) | 0x7f, stat_addr);
	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 已提交
1527
		retval = put_user((short)why, &infop->si_code);
L
Linus Torvalds 已提交
1528 1529 1530
	if (!retval && infop)
		retval = put_user(exit_code, &infop->si_status);
	if (!retval && infop)
1531
		retval = put_user(pid, &infop->si_pid);
L
Linus Torvalds 已提交
1532
	if (!retval && infop)
1533
		retval = put_user(uid, &infop->si_uid);
L
Linus Torvalds 已提交
1534
	if (!retval)
1535
		retval = pid;
L
Linus Torvalds 已提交
1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547
	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.
 */
R
Roland McGrath 已提交
1548
static int wait_task_continued(struct task_struct *p, int options,
L
Linus Torvalds 已提交
1549 1550 1551 1552 1553 1554 1555
			       struct siginfo __user *infop,
			       int __user *stat_addr, struct rusage __user *ru)
{
	int retval;
	pid_t pid;
	uid_t uid;

R
Roland McGrath 已提交
1556 1557 1558
	if (!unlikely(options & WCONTINUED))
		return 0;

L
Linus Torvalds 已提交
1559 1560 1561 1562 1563 1564 1565 1566 1567
	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;
	}
R
Roland McGrath 已提交
1568
	if (!unlikely(options & WNOWAIT))
L
Linus Torvalds 已提交
1569 1570 1571
		p->signal->flags &= ~SIGNAL_STOP_CONTINUED;
	spin_unlock_irq(&p->sighand->siglock);

1572
	pid = task_pid_vnr(p);
L
Linus Torvalds 已提交
1573 1574 1575 1576 1577 1578 1579 1580 1581 1582
	uid = p->uid;
	get_task_struct(p);
	read_unlock(&tasklist_lock);

	if (!infop) {
		retval = ru ? getrusage(p, RUSAGE_BOTH, ru) : 0;
		put_task_struct(p);
		if (!retval && stat_addr)
			retval = put_user(0xffff, stat_addr);
		if (!retval)
1583
			retval = pid;
L
Linus Torvalds 已提交
1584 1585 1586 1587 1588 1589 1590 1591 1592 1593
	} else {
		retval = wait_noreap_copyout(p, pid, uid,
					     CLD_CONTINUED, SIGCONT,
					     infop, ru);
		BUG_ON(retval == 0);
	}

	return retval;
}

R
Roland McGrath 已提交
1594 1595 1596 1597 1598 1599
/*
 * Consider @p for a wait by @parent.
 *
 * -ECHILD should be in *@notask_error before the first call.
 * Returns nonzero for a final return, when we have unlocked tasklist_lock.
 * Returns zero if the search for a child should continue;
1600 1601
 * then *@notask_error is 0 if @p is an eligible child,
 * or another error from security_task_wait(), or still -ECHILD.
R
Roland McGrath 已提交
1602
 */
R
Roland McGrath 已提交
1603
static int wait_consider_task(struct task_struct *parent, int ptrace,
R
Roland McGrath 已提交
1604 1605 1606 1607 1608 1609
			      struct task_struct *p, int *notask_error,
			      enum pid_type type, struct pid *pid, int options,
			      struct siginfo __user *infop,
			      int __user *stat_addr, struct rusage __user *ru)
{
	int ret = eligible_child(type, pid, options, p);
1610
	if (!ret)
R
Roland McGrath 已提交
1611 1612
		return ret;

1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624
	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.
		 */
		if (*notask_error)
			*notask_error = ret;
	}

R
Roland McGrath 已提交
1625 1626 1627 1628 1629 1630 1631 1632 1633
	if (likely(!ptrace) && unlikely(p->ptrace)) {
		/*
		 * This child is hidden by ptrace.
		 * We aren't allowed to see it now, but eventually we will.
		 */
		*notask_error = 0;
		return 0;
	}

R
Roland McGrath 已提交
1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649
	if (p->exit_state == EXIT_DEAD)
		return 0;

	/*
	 * We don't reap group leaders with subthreads.
	 */
	if (p->exit_state == EXIT_ZOMBIE && !delay_group_leader(p))
		return wait_task_zombie(p, options, infop, stat_addr, ru);

	/*
	 * It's stopped or running now, so it might
	 * later continue, exit, or stop again.
	 */
	*notask_error = 0;

	if (task_is_stopped_or_traced(p))
R
Roland McGrath 已提交
1650 1651
		return wait_task_stopped(ptrace, p, options,
					 infop, stat_addr, ru);
R
Roland McGrath 已提交
1652 1653 1654 1655 1656 1657 1658 1659 1660 1661

	return wait_task_continued(p, options, infop, stat_addr, ru);
}

/*
 * Do the work of do_wait() for one thread in the group, @tsk.
 *
 * -ECHILD should be in *@notask_error before the first call.
 * Returns nonzero for a final return, when we have unlocked tasklist_lock.
 * Returns zero if the search for a child should continue; then
1662 1663
 * *@notask_error is 0 if there were any eligible children,
 * or another error from security_task_wait(), or still -ECHILD.
R
Roland McGrath 已提交
1664 1665 1666 1667 1668 1669 1670 1671 1672
 */
static int do_wait_thread(struct task_struct *tsk, int *notask_error,
			  enum pid_type type, struct pid *pid, int options,
			  struct siginfo __user *infop, int __user *stat_addr,
			  struct rusage __user *ru)
{
	struct task_struct *p;

	list_for_each_entry(p, &tsk->children, sibling) {
R
Roland McGrath 已提交
1673 1674 1675 1676 1677 1678 1679 1680 1681 1682
		/*
		 * Do not consider detached threads.
		 */
		if (!task_detached(p)) {
			int ret = wait_consider_task(tsk, 0, p, notask_error,
						     type, pid, options,
						     infop, stat_addr, ru);
			if (ret)
				return ret;
		}
R
Roland McGrath 已提交
1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695
	}

	return 0;
}

static int ptrace_do_wait(struct task_struct *tsk, int *notask_error,
			  enum pid_type type, struct pid *pid, int options,
			  struct siginfo __user *infop, int __user *stat_addr,
			  struct rusage __user *ru)
{
	struct task_struct *p;

	/*
R
Roland McGrath 已提交
1696
	 * Traditionally we see ptrace'd stopped tasks regardless of options.
R
Roland McGrath 已提交
1697
	 */
R
Roland McGrath 已提交
1698
	options |= WUNTRACED;
R
Roland McGrath 已提交
1699

R
Roland McGrath 已提交
1700 1701 1702 1703 1704
	list_for_each_entry(p, &tsk->ptraced, ptrace_entry) {
		int ret = wait_consider_task(tsk, 1, p, notask_error,
					     type, pid, options,
					     infop, stat_addr, ru);
		if (ret)
R
Roland McGrath 已提交
1705 1706 1707 1708 1709 1710
			return ret;
	}

	return 0;
}

E
Eric W. Biederman 已提交
1711 1712 1713
static long do_wait(enum pid_type type, struct pid *pid, int options,
		    struct siginfo __user *infop, int __user *stat_addr,
		    struct rusage __user *ru)
L
Linus Torvalds 已提交
1714 1715 1716
{
	DECLARE_WAITQUEUE(wait, current);
	struct task_struct *tsk;
R
Roland McGrath 已提交
1717
	int retval;
L
Linus Torvalds 已提交
1718 1719 1720

	add_wait_queue(&current->signal->wait_chldexit,&wait);
repeat:
R
Roland McGrath 已提交
1721 1722 1723 1724 1725
	/*
	 * If there is nothing that can match our critiera just get out.
	 * We will clear @retval to zero if we see any child that might later
	 * match our criteria, even if we are not able to reap it yet.
	 */
E
Eric W. Biederman 已提交
1726 1727 1728 1729
	retval = -ECHILD;
	if ((type < PIDTYPE_MAX) && (!pid || hlist_empty(&pid->tasks[type])))
		goto end;

L
Linus Torvalds 已提交
1730 1731 1732 1733
	current->state = TASK_INTERRUPTIBLE;
	read_lock(&tasklist_lock);
	tsk = current;
	do {
R
Roland McGrath 已提交
1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746
		int tsk_result = do_wait_thread(tsk, &retval,
						type, pid, options,
						infop, stat_addr, ru);
		if (!tsk_result)
			tsk_result = ptrace_do_wait(tsk, &retval,
						    type, pid, options,
						    infop, stat_addr, ru);
		if (tsk_result) {
			/*
			 * tasklist_lock is unlocked and we have a final result.
			 */
			retval = tsk_result;
			goto end;
L
Linus Torvalds 已提交
1747
		}
R
Roland McGrath 已提交
1748

L
Linus Torvalds 已提交
1749 1750 1751
		if (options & __WNOTHREAD)
			break;
		tsk = next_thread(tsk);
E
Eric Sesterhenn 已提交
1752
		BUG_ON(tsk->signal != current->signal);
L
Linus Torvalds 已提交
1753 1754
	} while (tsk != current);
	read_unlock(&tasklist_lock);
O
Oleg Nesterov 已提交
1755

R
Roland McGrath 已提交
1756
	if (!retval && !(options & WNOHANG)) {
L
Linus Torvalds 已提交
1757
		retval = -ERESTARTSYS;
R
Roland McGrath 已提交
1758 1759 1760 1761
		if (!signal_pending(current)) {
			schedule();
			goto repeat;
		}
L
Linus Torvalds 已提交
1762
	}
R
Roland McGrath 已提交
1763

L
Linus Torvalds 已提交
1764 1765 1766 1767 1768
end:
	current->state = TASK_RUNNING;
	remove_wait_queue(&current->signal->wait_chldexit,&wait);
	if (infop) {
		if (retval > 0)
1769
			retval = 0;
L
Linus Torvalds 已提交
1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792
		else {
			/*
			 * For a WNOHANG return, clear out all the fields
			 * we would set so the user can easily tell the
			 * difference.
			 */
			if (!retval)
				retval = put_user(0, &infop->si_signo);
			if (!retval)
				retval = put_user(0, &infop->si_errno);
			if (!retval)
				retval = put_user(0, &infop->si_code);
			if (!retval)
				retval = put_user(0, &infop->si_pid);
			if (!retval)
				retval = put_user(0, &infop->si_uid);
			if (!retval)
				retval = put_user(0, &infop->si_status);
		}
	}
	return retval;
}

E
Eric W. Biederman 已提交
1793
asmlinkage long sys_waitid(int which, pid_t upid,
L
Linus Torvalds 已提交
1794 1795 1796
			   struct siginfo __user *infop, int options,
			   struct rusage __user *ru)
{
E
Eric W. Biederman 已提交
1797 1798
	struct pid *pid = NULL;
	enum pid_type type;
L
Linus Torvalds 已提交
1799 1800 1801 1802 1803 1804 1805 1806 1807
	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 已提交
1808
		type = PIDTYPE_MAX;
L
Linus Torvalds 已提交
1809 1810
		break;
	case P_PID:
E
Eric W. Biederman 已提交
1811 1812
		type = PIDTYPE_PID;
		if (upid <= 0)
L
Linus Torvalds 已提交
1813 1814 1815
			return -EINVAL;
		break;
	case P_PGID:
E
Eric W. Biederman 已提交
1816 1817
		type = PIDTYPE_PGID;
		if (upid <= 0)
L
Linus Torvalds 已提交
1818 1819 1820 1821 1822 1823
			return -EINVAL;
		break;
	default:
		return -EINVAL;
	}

E
Eric W. Biederman 已提交
1824 1825 1826 1827
	if (type < PIDTYPE_MAX)
		pid = find_get_pid(upid);
	ret = do_wait(type, pid, options, infop, NULL, ru);
	put_pid(pid);
L
Linus Torvalds 已提交
1828 1829

	/* avoid REGPARM breakage on x86: */
1830
	asmlinkage_protect(5, ret, which, upid, infop, options, ru);
L
Linus Torvalds 已提交
1831 1832 1833
	return ret;
}

E
Eric W. Biederman 已提交
1834
asmlinkage long sys_wait4(pid_t upid, int __user *stat_addr,
L
Linus Torvalds 已提交
1835 1836
			  int options, struct rusage __user *ru)
{
E
Eric W. Biederman 已提交
1837 1838
	struct pid *pid = NULL;
	enum pid_type type;
L
Linus Torvalds 已提交
1839 1840 1841 1842 1843
	long ret;

	if (options & ~(WNOHANG|WUNTRACED|WCONTINUED|
			__WNOTHREAD|__WCLONE|__WALL))
		return -EINVAL;
E
Eric W. Biederman 已提交
1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859

	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;
		pid = get_pid(task_pgrp(current));
	} else /* upid > 0 */ {
		type = PIDTYPE_PID;
		pid = find_get_pid(upid);
	}

	ret = do_wait(type, pid, options | WEXITED, NULL, stat_addr, ru);
	put_pid(pid);
L
Linus Torvalds 已提交
1860 1861

	/* avoid REGPARM breakage on x86: */
1862
	asmlinkage_protect(4, ret, upid, stat_addr, options, ru);
L
Linus Torvalds 已提交
1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877
	return ret;
}

#ifdef __ARCH_WANT_SYS_WAITPID

/*
 * sys_waitpid() remains for compatibility. waitpid() should be
 * implemented by calling sys_wait4() from libc.a.
 */
asmlinkage long sys_waitpid(pid_t pid, int __user *stat_addr, int options)
{
	return sys_wait4(pid, stat_addr, options, NULL);
}

#endif