fork.c 44.5 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
/*
 *  linux/kernel/fork.c
 *
 *  Copyright (C) 1991, 1992  Linus Torvalds
 */

/*
 *  'fork.c' contains the help-routines for the 'fork' system call
 * (see also entry.S and others).
 * Fork is rather simple, once you get the hang of it, but the memory
 * management can be a bitch. See 'mm/memory.c': 'copy_page_range()'
 */

#include <linux/slab.h>
#include <linux/init.h>
#include <linux/unistd.h>
#include <linux/module.h>
#include <linux/vmalloc.h>
#include <linux/completion.h>
#include <linux/personality.h>
#include <linux/mempolicy.h>
#include <linux/sem.h>
#include <linux/file.h>
A
Al Viro 已提交
24
#include <linux/fdtable.h>
25
#include <linux/iocontext.h>
L
Linus Torvalds 已提交
26 27 28
#include <linux/key.h>
#include <linux/binfmts.h>
#include <linux/mman.h>
A
Andrea Arcangeli 已提交
29
#include <linux/mmu_notifier.h>
L
Linus Torvalds 已提交
30
#include <linux/fs.h>
S
Serge E. Hallyn 已提交
31
#include <linux/nsproxy.h>
32
#include <linux/capability.h>
L
Linus Torvalds 已提交
33
#include <linux/cpu.h>
34
#include <linux/cgroup.h>
L
Linus Torvalds 已提交
35
#include <linux/security.h>
36
#include <linux/hugetlb.h>
37
#include <linux/seccomp.h>
L
Linus Torvalds 已提交
38 39 40 41
#include <linux/swap.h>
#include <linux/syscalls.h>
#include <linux/jiffies.h>
#include <linux/futex.h>
42
#include <linux/compat.h>
43
#include <linux/kthread.h>
44
#include <linux/task_io_accounting_ops.h>
45
#include <linux/rcupdate.h>
L
Linus Torvalds 已提交
46 47 48
#include <linux/ptrace.h>
#include <linux/mount.h>
#include <linux/audit.h>
49
#include <linux/memcontrol.h>
50
#include <linux/ftrace.h>
51
#include <linux/proc_fs.h>
L
Linus Torvalds 已提交
52 53
#include <linux/profile.h>
#include <linux/rmap.h>
H
Hugh Dickins 已提交
54
#include <linux/ksm.h>
L
Linus Torvalds 已提交
55
#include <linux/acct.h>
56
#include <linux/tsacct_kern.h>
M
Matt Helsley 已提交
57
#include <linux/cn_proc.h>
R
Rafael J. Wysocki 已提交
58
#include <linux/freezer.h>
59
#include <linux/delayacct.h>
60
#include <linux/taskstats_kern.h>
61
#include <linux/random.h>
M
Miloslav Trmac 已提交
62
#include <linux/tty.h>
63
#include <linux/blkdev.h>
64
#include <linux/fs_struct.h>
65
#include <linux/magic.h>
66
#include <linux/perf_event.h>
67
#include <linux/posix-timers.h>
68
#include <linux/user-return-notifier.h>
Y
Ying Han 已提交
69
#include <linux/oom.h>
A
Andrea Arcangeli 已提交
70
#include <linux/khugepaged.h>
71
#include <linux/signalfd.h>
72
#include <linux/uprobes.h>
L
Linus Torvalds 已提交
73 74 75 76 77 78 79 80

#include <asm/pgtable.h>
#include <asm/pgalloc.h>
#include <asm/uaccess.h>
#include <asm/mmu_context.h>
#include <asm/cacheflush.h>
#include <asm/tlbflush.h>

81 82
#include <trace/events/sched.h>

83 84 85
#define CREATE_TRACE_POINTS
#include <trace/events/task.h>

L
Linus Torvalds 已提交
86 87 88 89
/*
 * Protected counters by write_lock_irq(&tasklist_lock)
 */
unsigned long total_forks;	/* Handle normal Linux uptimes. */
90
int nr_threads;			/* The idle threads do not count.. */
L
Linus Torvalds 已提交
91 92 93 94 95

int max_threads;		/* tunable limit on nr_threads */

DEFINE_PER_CPU(unsigned long, process_counts) = 0;

96
__cacheline_aligned DEFINE_RWLOCK(tasklist_lock);  /* outer */
97 98 99 100 101 102 103 104

#ifdef CONFIG_PROVE_RCU
int lockdep_tasklist_lock_is_held(void)
{
	return lockdep_is_held(&tasklist_lock);
}
EXPORT_SYMBOL_GPL(lockdep_tasklist_lock_is_held);
#endif /* #ifdef CONFIG_PROVE_RCU */
L
Linus Torvalds 已提交
105 106 107 108 109 110

int nr_processes(void)
{
	int cpu;
	int total = 0;

111
	for_each_possible_cpu(cpu)
L
Linus Torvalds 已提交
112 113 114 115 116
		total += per_cpu(process_counts, cpu);

	return total;
}

117
#ifndef CONFIG_ARCH_TASK_STRUCT_ALLOCATOR
118
static struct kmem_cache *task_struct_cachep;
119 120 121 122 123 124 125 126 127 128 129 130 131

static inline struct task_struct *alloc_task_struct_node(int node)
{
	return kmem_cache_alloc_node(task_struct_cachep, GFP_KERNEL, node);
}

void __weak arch_release_task_struct(struct task_struct *tsk) { }

static inline void free_task_struct(struct task_struct *tsk)
{
	arch_release_task_struct(tsk);
	kmem_cache_free(task_struct_cachep, tsk);
}
L
Linus Torvalds 已提交
132 133
#endif

134
#ifndef CONFIG_ARCH_THREAD_INFO_ALLOCATOR
135 136
void __weak arch_release_thread_info(struct thread_info *ti) { }

137 138 139 140 141
/*
 * Allocate pages if THREAD_SIZE is >= PAGE_SIZE, otherwise use a
 * kmemcache based allocator.
 */
# if THREAD_SIZE >= PAGE_SIZE
142 143
static struct thread_info *alloc_thread_info_node(struct task_struct *tsk,
						  int node)
144
{
145 146
	struct page *page = alloc_pages_node(node, THREADINFO_GFP,
					     THREAD_SIZE_ORDER);
147 148

	return page ? page_address(page) : NULL;
149 150 151 152
}

static inline void free_thread_info(struct thread_info *ti)
{
153
	arch_release_thread_info(ti);
154 155
	free_pages((unsigned long)ti, THREAD_SIZE_ORDER);
}
156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177
# else
static struct kmem_cache *thread_info_cache;

static struct thread_info *alloc_thread_info_node(struct task_struct *tsk,
						  int node)
{
	return kmem_cache_alloc_node(thread_info_cache, THREADINFO_GFP, node);
}

static void free_thread_info(struct thread_info *ti)
{
	arch_release_thread_info(ti);
	kmem_cache_free(thread_info_cache, ti);
}

void thread_info_cache_init(void)
{
	thread_info_cache = kmem_cache_create("thread_info", THREAD_SIZE,
					      THREAD_SIZE, 0, NULL);
	BUG_ON(thread_info_cache == NULL);
}
# endif
178 179
#endif

L
Linus Torvalds 已提交
180
/* SLAB cache for signal_struct structures (tsk->signal) */
181
static struct kmem_cache *signal_cachep;
L
Linus Torvalds 已提交
182 183

/* SLAB cache for sighand_struct structures (tsk->sighand) */
184
struct kmem_cache *sighand_cachep;
L
Linus Torvalds 已提交
185 186

/* SLAB cache for files_struct structures (tsk->files) */
187
struct kmem_cache *files_cachep;
L
Linus Torvalds 已提交
188 189

/* SLAB cache for fs_struct structures (tsk->fs) */
190
struct kmem_cache *fs_cachep;
L
Linus Torvalds 已提交
191 192

/* SLAB cache for vm_area_struct structures */
193
struct kmem_cache *vm_area_cachep;
L
Linus Torvalds 已提交
194 195

/* SLAB cache for mm_struct structures (tsk->mm) */
196
static struct kmem_cache *mm_cachep;
L
Linus Torvalds 已提交
197

198 199 200 201 202 203 204
static void account_kernel_stack(struct thread_info *ti, int account)
{
	struct zone *zone = page_zone(virt_to_page(ti));

	mod_zone_page_state(zone, NR_KERNEL_STACK, account);
}

L
Linus Torvalds 已提交
205 206
void free_task(struct task_struct *tsk)
{
207
	account_kernel_stack(tsk->stack, -1);
R
Roman Zippel 已提交
208
	free_thread_info(tsk->stack);
I
Ingo Molnar 已提交
209
	rt_mutex_debug_task_free(tsk);
210
	ftrace_graph_exit_task(tsk);
211
	put_seccomp_filter(tsk);
L
Linus Torvalds 已提交
212 213 214 215
	free_task_struct(tsk);
}
EXPORT_SYMBOL(free_task);

216 217
static inline void free_signal_struct(struct signal_struct *sig)
{
218
	taskstats_tgid_free(sig);
219
	sched_autogroup_exit(sig);
220 221 222 223 224
	kmem_cache_free(signal_cachep, sig);
}

static inline void put_signal_struct(struct signal_struct *sig)
{
225
	if (atomic_dec_and_test(&sig->sigcnt))
226 227 228
		free_signal_struct(sig);
}

229
void __put_task_struct(struct task_struct *tsk)
L
Linus Torvalds 已提交
230
{
E
Eugene Teo 已提交
231
	WARN_ON(!tsk->exit_state);
L
Linus Torvalds 已提交
232 233 234
	WARN_ON(atomic_read(&tsk->usage));
	WARN_ON(tsk == current);

235
	security_task_free(tsk);
236
	exit_creds(tsk);
237
	delayacct_tsk_free(tsk);
238
	put_signal_struct(tsk->signal);
L
Linus Torvalds 已提交
239 240 241 242

	if (!profile_handoff_task(tsk))
		free_task(tsk);
}
243
EXPORT_SYMBOL_GPL(__put_task_struct);
L
Linus Torvalds 已提交
244

T
Thomas Gleixner 已提交
245
void __init __weak arch_task_cache_init(void) { }
246

L
Linus Torvalds 已提交
247 248
void __init fork_init(unsigned long mempages)
{
249
#ifndef CONFIG_ARCH_TASK_STRUCT_ALLOCATOR
L
Linus Torvalds 已提交
250 251 252 253 254 255
#ifndef ARCH_MIN_TASKALIGN
#define ARCH_MIN_TASKALIGN	L1_CACHE_BYTES
#endif
	/* create a slab on which task_structs can be allocated */
	task_struct_cachep =
		kmem_cache_create("task_struct", sizeof(struct task_struct),
V
Vegard Nossum 已提交
256
			ARCH_MIN_TASKALIGN, SLAB_PANIC | SLAB_NOTRACK, NULL);
L
Linus Torvalds 已提交
257 258
#endif

259 260 261
	/* do the arch specific task caches init */
	arch_task_cache_init();

L
Linus Torvalds 已提交
262 263 264 265 266 267 268 269 270 271
	/*
	 * The default maximum number of threads is set to a safe
	 * value: the thread structures can take up at most half
	 * of memory.
	 */
	max_threads = mempages / (8 * THREAD_SIZE / PAGE_SIZE);

	/*
	 * we need to allow at least 20 threads to boot a system
	 */
272
	if (max_threads < 20)
L
Linus Torvalds 已提交
273 274 275 276 277 278 279 280
		max_threads = 20;

	init_task.signal->rlim[RLIMIT_NPROC].rlim_cur = max_threads/2;
	init_task.signal->rlim[RLIMIT_NPROC].rlim_max = max_threads/2;
	init_task.signal->rlim[RLIMIT_SIGPENDING] =
		init_task.signal->rlim[RLIMIT_NPROC];
}

281 282 283 284 285 286 287
int __attribute__((weak)) arch_dup_task_struct(struct task_struct *dst,
					       struct task_struct *src)
{
	*dst = *src;
	return 0;
}

L
Linus Torvalds 已提交
288 289 290 291
static struct task_struct *dup_task_struct(struct task_struct *orig)
{
	struct task_struct *tsk;
	struct thread_info *ti;
292
	unsigned long *stackend;
293
	int node = tsk_fork_get_node(orig);
P
Peter Zijlstra 已提交
294
	int err;
L
Linus Torvalds 已提交
295

296
	tsk = alloc_task_struct_node(node);
L
Linus Torvalds 已提交
297 298 299
	if (!tsk)
		return NULL;

300
	ti = alloc_thread_info_node(tsk, node);
L
Linus Torvalds 已提交
301 302 303 304 305
	if (!ti) {
		free_task_struct(tsk);
		return NULL;
	}

306
	err = arch_dup_task_struct(tsk, orig);
307

308 309 310 311
	/*
	 * We defer looking at err, because we will need this setup
	 * for the clean up path to work correctly.
	 */
R
Roman Zippel 已提交
312
	tsk->stack = ti;
313
	setup_thread_stack(tsk, orig);
314 315 316 317

	if (err)
		goto out;

318
	clear_user_return_notifier(tsk);
319
	clear_tsk_need_resched(tsk);
320 321
	stackend = end_of_stack(tsk);
	*stackend = STACK_END_MAGIC;	/* for overflow detection */
L
Linus Torvalds 已提交
322

323 324 325 326
#ifdef CONFIG_CC_STACKPROTECTOR
	tsk->stack_canary = get_random_int();
#endif

327 328 329 330 331
	/*
	 * One for us, one for whoever does the "release_task()" (usually
	 * parent)
	 */
	atomic_set(&tsk->usage, 2);
332
#ifdef CONFIG_BLK_DEV_IO_TRACE
333
	tsk->btrace_seq = 0;
334
#endif
335
	tsk->splice_pipe = NULL;
336 337 338

	account_kernel_stack(ti, 1);

L
Linus Torvalds 已提交
339
	return tsk;
340 341 342 343 344

out:
	free_thread_info(ti);
	free_task_struct(tsk);
	return NULL;
L
Linus Torvalds 已提交
345 346 347
}

#ifdef CONFIG_MMU
A
Alexey Dobriyan 已提交
348
static int dup_mmap(struct mm_struct *mm, struct mm_struct *oldmm)
L
Linus Torvalds 已提交
349
{
350
	struct vm_area_struct *mpnt, *tmp, *prev, **pprev;
L
Linus Torvalds 已提交
351 352 353 354 355 356
	struct rb_node **rb_link, *rb_parent;
	int retval;
	unsigned long charge;
	struct mempolicy *pol;

	down_write(&oldmm->mmap_sem);
357
	flush_cache_dup_mm(oldmm);
358 359 360 361
	/*
	 * Not linked in yet - no deadlock potential:
	 */
	down_write_nested(&mm->mmap_sem, SINGLE_DEPTH_NESTING);
362

L
Linus Torvalds 已提交
363 364 365 366
	mm->locked_vm = 0;
	mm->mmap = NULL;
	mm->mmap_cache = NULL;
	mm->free_area_cache = oldmm->mmap_base;
367
	mm->cached_hole_size = ~0UL;
L
Linus Torvalds 已提交
368
	mm->map_count = 0;
369
	cpumask_clear(mm_cpumask(mm));
L
Linus Torvalds 已提交
370 371 372 373
	mm->mm_rb = RB_ROOT;
	rb_link = &mm->mm_rb.rb_node;
	rb_parent = NULL;
	pprev = &mm->mmap;
H
Hugh Dickins 已提交
374
	retval = ksm_fork(mm, oldmm);
A
Andrea Arcangeli 已提交
375 376 377
	if (retval)
		goto out;
	retval = khugepaged_fork(mm, oldmm);
H
Hugh Dickins 已提交
378 379
	if (retval)
		goto out;
L
Linus Torvalds 已提交
380

381
	prev = NULL;
H
Hugh Dickins 已提交
382
	for (mpnt = oldmm->mmap; mpnt; mpnt = mpnt->vm_next) {
L
Linus Torvalds 已提交
383 384 385
		struct file *file;

		if (mpnt->vm_flags & VM_DONTCOPY) {
386 387
			long pages = vma_pages(mpnt);
			mm->total_vm -= pages;
388
			vm_stat_account(mm, mpnt->vm_flags, mpnt->vm_file,
389
								-pages);
L
Linus Torvalds 已提交
390 391 392 393
			continue;
		}
		charge = 0;
		if (mpnt->vm_flags & VM_ACCOUNT) {
394 395
			unsigned long len = vma_pages(mpnt);

396
			if (security_vm_enough_memory_mm(oldmm, len)) /* sic */
L
Linus Torvalds 已提交
397 398 399
				goto fail_nomem;
			charge = len;
		}
400
		tmp = kmem_cache_alloc(vm_area_cachep, GFP_KERNEL);
L
Linus Torvalds 已提交
401 402 403
		if (!tmp)
			goto fail_nomem;
		*tmp = *mpnt;
404
		INIT_LIST_HEAD(&tmp->anon_vma_chain);
405
		pol = mpol_dup(vma_policy(mpnt));
L
Linus Torvalds 已提交
406 407 408 409
		retval = PTR_ERR(pol);
		if (IS_ERR(pol))
			goto fail_nomem_policy;
		vma_set_policy(tmp, pol);
A
Andrea Arcangeli 已提交
410
		tmp->vm_mm = mm;
411 412
		if (anon_vma_fork(tmp, mpnt))
			goto fail_nomem_anon_vma_fork;
L
Linus Torvalds 已提交
413
		tmp->vm_flags &= ~VM_LOCKED;
414
		tmp->vm_next = tmp->vm_prev = NULL;
L
Linus Torvalds 已提交
415 416
		file = tmp->vm_file;
		if (file) {
417
			struct inode *inode = file->f_path.dentry->d_inode;
H
Hugh Dickins 已提交
418 419
			struct address_space *mapping = file->f_mapping;

L
Linus Torvalds 已提交
420 421 422
			get_file(file);
			if (tmp->vm_flags & VM_DENYWRITE)
				atomic_dec(&inode->i_writecount);
423
			mutex_lock(&mapping->i_mmap_mutex);
H
Hugh Dickins 已提交
424 425 426 427
			if (tmp->vm_flags & VM_SHARED)
				mapping->i_mmap_writable++;
			flush_dcache_mmap_lock(mapping);
			/* insert tmp into the share list, just after mpnt */
L
Linus Torvalds 已提交
428
			vma_prio_tree_add(tmp, mpnt);
H
Hugh Dickins 已提交
429
			flush_dcache_mmap_unlock(mapping);
430
			mutex_unlock(&mapping->i_mmap_mutex);
L
Linus Torvalds 已提交
431 432
		}

433 434 435 436 437 438 439 440
		/*
		 * Clear hugetlb-related page reserves for children. This only
		 * affects MAP_PRIVATE mappings. Faults generated by the child
		 * are not guaranteed to succeed, even if read-only
		 */
		if (is_vm_hugetlb_page(tmp))
			reset_vma_resv_huge_pages(tmp);

L
Linus Torvalds 已提交
441
		/*
442
		 * Link in the new vma and copy the page table entries.
L
Linus Torvalds 已提交
443 444 445
		 */
		*pprev = tmp;
		pprev = &tmp->vm_next;
446 447
		tmp->vm_prev = prev;
		prev = tmp;
L
Linus Torvalds 已提交
448 449 450 451 452 453

		__vma_link_rb(mm, tmp, rb_link, rb_parent);
		rb_link = &tmp->vm_rb.rb_right;
		rb_parent = &tmp->vm_rb;

		mm->map_count++;
454
		retval = copy_page_range(mm, oldmm, mpnt);
L
Linus Torvalds 已提交
455 456 457 458 459 460

		if (tmp->vm_ops && tmp->vm_ops->open)
			tmp->vm_ops->open(tmp);

		if (retval)
			goto out;
461 462 463

		if (file && uprobe_mmap(tmp))
			goto out;
L
Linus Torvalds 已提交
464
	}
465 466
	/* a new mm has just been created */
	arch_dup_mmap(oldmm, mm);
L
Linus Torvalds 已提交
467 468
	retval = 0;
out:
469
	up_write(&mm->mmap_sem);
H
Hugh Dickins 已提交
470
	flush_tlb_mm(oldmm);
L
Linus Torvalds 已提交
471 472
	up_write(&oldmm->mmap_sem);
	return retval;
473 474
fail_nomem_anon_vma_fork:
	mpol_put(pol);
L
Linus Torvalds 已提交
475 476 477 478 479 480 481 482
fail_nomem_policy:
	kmem_cache_free(vm_area_cachep, tmp);
fail_nomem:
	retval = -ENOMEM;
	vm_unacct_memory(charge);
	goto out;
}

483
static inline int mm_alloc_pgd(struct mm_struct *mm)
L
Linus Torvalds 已提交
484 485 486 487 488 489 490
{
	mm->pgd = pgd_alloc(mm);
	if (unlikely(!mm->pgd))
		return -ENOMEM;
	return 0;
}

491
static inline void mm_free_pgd(struct mm_struct *mm)
L
Linus Torvalds 已提交
492
{
493
	pgd_free(mm, mm->pgd);
L
Linus Torvalds 已提交
494 495 496 497 498 499 500
}
#else
#define dup_mmap(mm, oldmm)	(0)
#define mm_alloc_pgd(mm)	(0)
#define mm_free_pgd(mm)
#endif /* CONFIG_MMU */

D
Daniel Walker 已提交
501
__cacheline_aligned_in_smp DEFINE_SPINLOCK(mmlist_lock);
L
Linus Torvalds 已提交
502

503
#define allocate_mm()	(kmem_cache_alloc(mm_cachep, GFP_KERNEL))
L
Linus Torvalds 已提交
504 505
#define free_mm(mm)	(kmem_cache_free(mm_cachep, (mm)))

506 507 508 509 510 511 512 513 514 515 516 517
static unsigned long default_dump_filter = MMF_DUMP_FILTER_DEFAULT;

static int __init coredump_filter_setup(char *s)
{
	default_dump_filter =
		(simple_strtoul(s, NULL, 0) << MMF_DUMP_FILTER_SHIFT) &
		MMF_DUMP_FILTER_MASK;
	return 1;
}

__setup("coredump_filter=", coredump_filter_setup);

L
Linus Torvalds 已提交
518 519
#include <linux/init_task.h>

A
Alexey Dobriyan 已提交
520 521 522 523 524 525 526 527
static void mm_init_aio(struct mm_struct *mm)
{
#ifdef CONFIG_AIO
	spin_lock_init(&mm->ioctx_lock);
	INIT_HLIST_HEAD(&mm->ioctx_list);
#endif
}

528
static struct mm_struct *mm_init(struct mm_struct *mm, struct task_struct *p)
L
Linus Torvalds 已提交
529 530 531 532 533
{
	atomic_set(&mm->mm_users, 1);
	atomic_set(&mm->mm_count, 1);
	init_rwsem(&mm->mmap_sem);
	INIT_LIST_HEAD(&mm->mmlist);
H
Hugh Dickins 已提交
534 535
	mm->flags = (current->mm) ?
		(current->mm->flags & MMF_INIT_MASK) : default_dump_filter;
536
	mm->core_state = NULL;
L
Linus Torvalds 已提交
537
	mm->nr_ptes = 0;
K
KAMEZAWA Hiroyuki 已提交
538
	memset(&mm->rss_stat, 0, sizeof(mm->rss_stat));
L
Linus Torvalds 已提交
539 540
	spin_lock_init(&mm->page_table_lock);
	mm->free_area_cache = TASK_UNMAPPED_BASE;
541
	mm->cached_hole_size = ~0UL;
A
Alexey Dobriyan 已提交
542
	mm_init_aio(mm);
543
	mm_init_owner(mm, p);
L
Linus Torvalds 已提交
544 545 546

	if (likely(!mm_alloc_pgd(mm))) {
		mm->def_flags = 0;
A
Andrea Arcangeli 已提交
547
		mmu_notifier_mm_init(mm);
L
Linus Torvalds 已提交
548 549
		return mm;
	}
550

L
Linus Torvalds 已提交
551 552 553 554
	free_mm(mm);
	return NULL;
}

555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571
static void check_mm(struct mm_struct *mm)
{
	int i;

	for (i = 0; i < NR_MM_COUNTERS; i++) {
		long x = atomic_long_read(&mm->rss_stat.count[i]);

		if (unlikely(x))
			printk(KERN_ALERT "BUG: Bad rss-counter state "
					  "mm:%p idx:%d val:%ld\n", mm, i, x);
	}

#ifdef CONFIG_TRANSPARENT_HUGEPAGE
	VM_BUG_ON(mm->pmd_huge_pte);
#endif
}

L
Linus Torvalds 已提交
572 573 574
/*
 * Allocate and initialize an mm_struct.
 */
575
struct mm_struct *mm_alloc(void)
L
Linus Torvalds 已提交
576
{
577
	struct mm_struct *mm;
L
Linus Torvalds 已提交
578 579

	mm = allocate_mm();
580 581 582 583
	if (!mm)
		return NULL;

	memset(mm, 0, sizeof(*mm));
584 585
	mm_init_cpumask(mm);
	return mm_init(mm, current);
L
Linus Torvalds 已提交
586 587 588 589 590 591 592
}

/*
 * Called when the last reference to the mm
 * is dropped: either by a lazy thread or by
 * mmput. Free the page directory and the mm.
 */
593
void __mmdrop(struct mm_struct *mm)
L
Linus Torvalds 已提交
594 595 596 597
{
	BUG_ON(mm == &init_mm);
	mm_free_pgd(mm);
	destroy_context(mm);
A
Andrea Arcangeli 已提交
598
	mmu_notifier_mm_destroy(mm);
599
	check_mm(mm);
L
Linus Torvalds 已提交
600 601
	free_mm(mm);
}
602
EXPORT_SYMBOL_GPL(__mmdrop);
L
Linus Torvalds 已提交
603 604 605 606 607 608

/*
 * Decrement the use count and release all resources for an mm.
 */
void mmput(struct mm_struct *mm)
{
A
Andrew Morton 已提交
609 610
	might_sleep();

L
Linus Torvalds 已提交
611
	if (atomic_dec_and_test(&mm->mm_users)) {
612
		uprobe_clear_state(mm);
L
Linus Torvalds 已提交
613
		exit_aio(mm);
614
		ksm_exit(mm);
A
Andrea Arcangeli 已提交
615
		khugepaged_exit(mm); /* must run before exit_mmap */
L
Linus Torvalds 已提交
616
		exit_mmap(mm);
M
Matt Helsley 已提交
617
		set_mm_exe_file(mm, NULL);
L
Linus Torvalds 已提交
618 619 620 621 622
		if (!list_empty(&mm->mmlist)) {
			spin_lock(&mmlist_lock);
			list_del(&mm->mmlist);
			spin_unlock(&mmlist_lock);
		}
623 624
		if (mm->binfmt)
			module_put(mm->binfmt->module);
L
Linus Torvalds 已提交
625 626 627 628 629
		mmdrop(mm);
	}
}
EXPORT_SYMBOL_GPL(mmput);

630 631 632 633 634 635 636 637 638 639 640 641 642
/*
 * We added or removed a vma mapping the executable. The vmas are only mapped
 * during exec and are not mapped with the mmap system call.
 * Callers must hold down_write() on the mm's mmap_sem for these
 */
void added_exe_file_vma(struct mm_struct *mm)
{
	mm->num_exe_file_vmas++;
}

void removed_exe_file_vma(struct mm_struct *mm)
{
	mm->num_exe_file_vmas--;
643
	if ((mm->num_exe_file_vmas == 0) && mm->exe_file) {
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 672 673 674 675 676 677 678 679 680
		fput(mm->exe_file);
		mm->exe_file = NULL;
	}

}

void set_mm_exe_file(struct mm_struct *mm, struct file *new_exe_file)
{
	if (new_exe_file)
		get_file(new_exe_file);
	if (mm->exe_file)
		fput(mm->exe_file);
	mm->exe_file = new_exe_file;
	mm->num_exe_file_vmas = 0;
}

struct file *get_mm_exe_file(struct mm_struct *mm)
{
	struct file *exe_file;

	/* We need mmap_sem to protect against races with removal of
	 * VM_EXECUTABLE vmas */
	down_read(&mm->mmap_sem);
	exe_file = mm->exe_file;
	if (exe_file)
		get_file(exe_file);
	up_read(&mm->mmap_sem);
	return exe_file;
}

static void dup_mm_exe_file(struct mm_struct *oldmm, struct mm_struct *newmm)
{
	/* It's safe to write the exe_file pointer without exe_file_lock because
	 * this is called during fork when the task is not yet in /proc */
	newmm->exe_file = get_mm_exe_file(oldmm);
}

L
Linus Torvalds 已提交
681 682 683
/**
 * get_task_mm - acquire a reference to the task's mm
 *
684
 * Returns %NULL if the task has no mm.  Checks PF_KTHREAD (meaning
L
Linus Torvalds 已提交
685 686 687 688 689 690 691 692 693 694 695 696
 * this kernel workthread has transiently adopted a user mm with use_mm,
 * to do its AIO) is not set and if so returns a reference to it, after
 * bumping up the use count.  User must release the mm via mmput()
 * after use.  Typically used by /proc and ptrace.
 */
struct mm_struct *get_task_mm(struct task_struct *task)
{
	struct mm_struct *mm;

	task_lock(task);
	mm = task->mm;
	if (mm) {
697
		if (task->flags & PF_KTHREAD)
L
Linus Torvalds 已提交
698 699 700 701 702 703 704 705 706
			mm = NULL;
		else
			atomic_inc(&mm->mm_users);
	}
	task_unlock(task);
	return mm;
}
EXPORT_SYMBOL_GPL(get_task_mm);

707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726
struct mm_struct *mm_access(struct task_struct *task, unsigned int mode)
{
	struct mm_struct *mm;
	int err;

	err =  mutex_lock_killable(&task->signal->cred_guard_mutex);
	if (err)
		return ERR_PTR(err);

	mm = get_task_mm(task);
	if (mm && mm != current->mm &&
			!ptrace_may_access(task, mode)) {
		mmput(mm);
		mm = ERR_PTR(-EACCES);
	}
	mutex_unlock(&task->signal->cred_guard_mutex);

	return mm;
}

727
static void complete_vfork_done(struct task_struct *tsk)
728
{
O
Oleg Nesterov 已提交
729
	struct completion *vfork;
730

O
Oleg Nesterov 已提交
731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756
	task_lock(tsk);
	vfork = tsk->vfork_done;
	if (likely(vfork)) {
		tsk->vfork_done = NULL;
		complete(vfork);
	}
	task_unlock(tsk);
}

static int wait_for_vfork_done(struct task_struct *child,
				struct completion *vfork)
{
	int killed;

	freezer_do_not_count();
	killed = wait_for_completion_killable(vfork);
	freezer_count();

	if (killed) {
		task_lock(child);
		child->vfork_done = NULL;
		task_unlock(child);
	}

	put_task_struct(child);
	return killed;
757 758
}

L
Linus Torvalds 已提交
759 760 761 762 763 764 765 766 767 768 769 770 771 772 773
/* Please note the differences between mmput and mm_release.
 * mmput is called whenever we stop holding onto a mm_struct,
 * error success whatever.
 *
 * mm_release is called after a mm_struct has been removed
 * from the current process.
 *
 * This difference is important for error handling, when we
 * only half set up a mm_struct for a new process and need to restore
 * the old one.  Because we mmput the new mm_struct before
 * restoring the old one. . .
 * Eric Biederman 10 January 1998
 */
void mm_release(struct task_struct *tsk, struct mm_struct *mm)
{
774 775
	/* Get rid of any futexes when releasing the mm */
#ifdef CONFIG_FUTEX
776
	if (unlikely(tsk->robust_list)) {
777
		exit_robust_list(tsk);
778 779
		tsk->robust_list = NULL;
	}
780
#ifdef CONFIG_COMPAT
781
	if (unlikely(tsk->compat_robust_list)) {
782
		compat_exit_robust_list(tsk);
783 784
		tsk->compat_robust_list = NULL;
	}
785
#endif
786 787
	if (unlikely(!list_empty(&tsk->pi_state_list)))
		exit_pi_state_list(tsk);
788 789
#endif

790 791
	uprobe_free_utask(tsk);

L
Linus Torvalds 已提交
792 793 794
	/* Get rid of any cached register state */
	deactivate_mm(tsk, mm);

795 796 797 798
	/*
	 * If we're exiting normally, clear a user-space tid field if
	 * requested.  We leave this alone when dying by signal, to leave
	 * the value intact in a core dump, and to save the unnecessary
O
Oleg Nesterov 已提交
799 800
	 * trouble, say, a killed vfork parent shouldn't touch this mm.
	 * Userland only wants this done for a sys_exit.
801
	 */
802 803 804 805 806 807 808 809 810 811 812
	if (tsk->clear_child_tid) {
		if (!(tsk->flags & PF_SIGNALED) &&
		    atomic_read(&mm->mm_users) > 1) {
			/*
			 * We don't check the error code - if userspace has
			 * not set up a proper pointer then tough luck.
			 */
			put_user(0, tsk->clear_child_tid);
			sys_futex(tsk->clear_child_tid, FUTEX_WAKE,
					1, NULL, NULL, 0);
		}
L
Linus Torvalds 已提交
813 814
		tsk->clear_child_tid = NULL;
	}
815 816 817 818 819 820 821

	/*
	 * All done, finally we can wake up parent and return this mm to him.
	 * Also kthread_stop() uses this completion for synchronization.
	 */
	if (tsk->vfork_done)
		complete_vfork_done(tsk);
L
Linus Torvalds 已提交
822 823
}

824 825 826 827
/*
 * Allocate a new mm structure and copy contents from the
 * mm structure of the passed in task structure.
 */
828
struct mm_struct *dup_mm(struct task_struct *tsk)
829 830 831 832 833 834 835 836 837 838 839 840
{
	struct mm_struct *mm, *oldmm = current->mm;
	int err;

	if (!oldmm)
		return NULL;

	mm = allocate_mm();
	if (!mm)
		goto fail_nomem;

	memcpy(mm, oldmm, sizeof(*mm));
841
	mm_init_cpumask(mm);
842

843 844 845
#ifdef CONFIG_TRANSPARENT_HUGEPAGE
	mm->pmd_huge_pte = NULL;
#endif
846
	uprobe_reset_state(mm);
847

848
	if (!mm_init(mm, tsk))
849 850 851 852 853
		goto fail_nomem;

	if (init_new_context(tsk, mm))
		goto fail_nocontext;

M
Matt Helsley 已提交
854 855
	dup_mm_exe_file(oldmm, mm);

856 857 858 859 860 861 862
	err = dup_mmap(mm, oldmm);
	if (err)
		goto free_pt;

	mm->hiwater_rss = get_mm_rss(mm);
	mm->hiwater_vm = mm->total_vm;

863 864 865
	if (mm->binfmt && !try_module_get(mm->binfmt->module))
		goto free_pt;

866 867 868
	return mm;

free_pt:
869 870
	/* don't put binfmt in mmput, we haven't got module yet */
	mm->binfmt = NULL;
871 872 873 874 875 876 877 878 879 880 881 882 883 884 885
	mmput(mm);

fail_nomem:
	return NULL;

fail_nocontext:
	/*
	 * If init_new_context() failed, we cannot use mmput() to free the mm
	 * because it calls destroy_context()
	 */
	mm_free_pgd(mm);
	free_mm(mm);
	return NULL;
}

886
static int copy_mm(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
887
{
888
	struct mm_struct *mm, *oldmm;
L
Linus Torvalds 已提交
889 890 891 892
	int retval;

	tsk->min_flt = tsk->maj_flt = 0;
	tsk->nvcsw = tsk->nivcsw = 0;
893 894 895
#ifdef CONFIG_DETECT_HUNG_TASK
	tsk->last_switch_count = tsk->nvcsw + tsk->nivcsw;
#endif
L
Linus Torvalds 已提交
896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915

	tsk->mm = NULL;
	tsk->active_mm = NULL;

	/*
	 * Are we cloning a kernel thread?
	 *
	 * We need to steal a active VM for that..
	 */
	oldmm = current->mm;
	if (!oldmm)
		return 0;

	if (clone_flags & CLONE_VM) {
		atomic_inc(&oldmm->mm_users);
		mm = oldmm;
		goto good_mm;
	}

	retval = -ENOMEM;
916
	mm = dup_mm(tsk);
L
Linus Torvalds 已提交
917 918 919 920 921 922 923 924 925 926 927 928
	if (!mm)
		goto fail_nomem;

good_mm:
	tsk->mm = mm;
	tsk->active_mm = mm;
	return 0;

fail_nomem:
	return retval;
}

A
Alexey Dobriyan 已提交
929
static int copy_fs(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
930
{
A
Al Viro 已提交
931
	struct fs_struct *fs = current->fs;
L
Linus Torvalds 已提交
932
	if (clone_flags & CLONE_FS) {
A
Al Viro 已提交
933
		/* tsk->fs is already what we want */
N
Nick Piggin 已提交
934
		spin_lock(&fs->lock);
A
Al Viro 已提交
935
		if (fs->in_exec) {
N
Nick Piggin 已提交
936
			spin_unlock(&fs->lock);
A
Al Viro 已提交
937 938 939
			return -EAGAIN;
		}
		fs->users++;
N
Nick Piggin 已提交
940
		spin_unlock(&fs->lock);
L
Linus Torvalds 已提交
941 942
		return 0;
	}
A
Al Viro 已提交
943
	tsk->fs = copy_fs_struct(fs);
L
Linus Torvalds 已提交
944 945 946 947 948
	if (!tsk->fs)
		return -ENOMEM;
	return 0;
}

949
static int copy_files(unsigned long clone_flags, struct task_struct *tsk)
950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975
{
	struct files_struct *oldf, *newf;
	int error = 0;

	/*
	 * A background process may not have any files ...
	 */
	oldf = current->files;
	if (!oldf)
		goto out;

	if (clone_flags & CLONE_FILES) {
		atomic_inc(&oldf->count);
		goto out;
	}

	newf = dup_fd(oldf, &error);
	if (!newf)
		goto out;

	tsk->files = newf;
	error = 0;
out:
	return error;
}

976
static int copy_io(unsigned long clone_flags, struct task_struct *tsk)
977 978 979
{
#ifdef CONFIG_BLOCK
	struct io_context *ioc = current->io_context;
980
	struct io_context *new_ioc;
981 982 983

	if (!ioc)
		return 0;
984 985 986 987
	/*
	 * Share io context with parent, if CLONE_IO is set
	 */
	if (clone_flags & CLONE_IO) {
T
Tejun Heo 已提交
988 989
		ioc_task_link(ioc);
		tsk->io_context = ioc;
990
	} else if (ioprio_valid(ioc->ioprio)) {
991 992
		new_ioc = get_task_io_context(tsk, GFP_KERNEL, NUMA_NO_NODE);
		if (unlikely(!new_ioc))
993 994
			return -ENOMEM;

995
		new_ioc->ioprio = ioc->ioprio;
996
		put_io_context(new_ioc);
997 998 999 1000 1001
	}
#endif
	return 0;
}

A
Alexey Dobriyan 已提交
1002
static int copy_sighand(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
1003 1004 1005
{
	struct sighand_struct *sig;

Z
Zhaolei 已提交
1006
	if (clone_flags & CLONE_SIGHAND) {
L
Linus Torvalds 已提交
1007 1008 1009 1010
		atomic_inc(&current->sighand->count);
		return 0;
	}
	sig = kmem_cache_alloc(sighand_cachep, GFP_KERNEL);
I
Ingo Molnar 已提交
1011
	rcu_assign_pointer(tsk->sighand, sig);
L
Linus Torvalds 已提交
1012 1013 1014 1015 1016 1017 1018
	if (!sig)
		return -ENOMEM;
	atomic_set(&sig->count, 1);
	memcpy(sig->action, current->sighand->action, sizeof(sig->action));
	return 0;
}

1019
void __cleanup_sighand(struct sighand_struct *sighand)
1020
{
1021 1022
	if (atomic_dec_and_test(&sighand->count)) {
		signalfd_cleanup(sighand);
1023
		kmem_cache_free(sighand_cachep, sighand);
1024
	}
1025 1026
}

1027 1028 1029 1030 1031 1032

/*
 * Initialize POSIX timer handling for a thread group.
 */
static void posix_cpu_timers_init_group(struct signal_struct *sig)
{
1033 1034
	unsigned long cpu_limit;

1035 1036 1037
	/* Thread group counters. */
	thread_group_cputime_init(sig);

1038 1039 1040
	cpu_limit = ACCESS_ONCE(sig->rlim[RLIMIT_CPU].rlim_cur);
	if (cpu_limit != RLIM_INFINITY) {
		sig->cputime_expires.prof_exp = secs_to_cputime(cpu_limit);
1041 1042 1043
		sig->cputimer.running = 1;
	}

1044 1045 1046 1047 1048 1049
	/* The timer lists. */
	INIT_LIST_HEAD(&sig->cpu_timers[0]);
	INIT_LIST_HEAD(&sig->cpu_timers[1]);
	INIT_LIST_HEAD(&sig->cpu_timers[2]);
}

A
Alexey Dobriyan 已提交
1050
static int copy_signal(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
1051 1052 1053
{
	struct signal_struct *sig;

1054
	if (clone_flags & CLONE_THREAD)
1055 1056
		return 0;

1057
	sig = kmem_cache_zalloc(signal_cachep, GFP_KERNEL);
L
Linus Torvalds 已提交
1058 1059 1060 1061
	tsk->signal = sig;
	if (!sig)
		return -ENOMEM;

1062
	sig->nr_threads = 1;
L
Linus Torvalds 已提交
1063
	atomic_set(&sig->live, 1);
1064
	atomic_set(&sig->sigcnt, 1);
L
Linus Torvalds 已提交
1065
	init_waitqueue_head(&sig->wait_chldexit);
1066 1067
	if (clone_flags & CLONE_NEWPID)
		sig->flags |= SIGNAL_UNKILLABLE;
1068
	sig->curr_target = tsk;
L
Linus Torvalds 已提交
1069 1070 1071
	init_sigpending(&sig->shared_pending);
	INIT_LIST_HEAD(&sig->posix_timers);

1072
	hrtimer_init(&sig->real_timer, CLOCK_MONOTONIC, HRTIMER_MODE_REL);
L
Linus Torvalds 已提交
1073 1074 1075 1076 1077 1078
	sig->real_timer.function = it_real_fn;

	task_lock(current->group_leader);
	memcpy(sig->rlim, current->signal->rlim, sizeof sig->rlim);
	task_unlock(current->group_leader);

1079 1080
	posix_cpu_timers_init_group(sig);

M
Miloslav Trmac 已提交
1081
	tty_audit_fork(sig);
1082
	sched_autogroup_fork(sig);
M
Miloslav Trmac 已提交
1083

1084
#ifdef CONFIG_CGROUPS
1085
	init_rwsem(&sig->group_rwsem);
1086 1087
#endif

1088
	sig->oom_adj = current->signal->oom_adj;
D
David Rientjes 已提交
1089
	sig->oom_score_adj = current->signal->oom_score_adj;
1090
	sig->oom_score_adj_min = current->signal->oom_score_adj_min;
1091

1092 1093 1094
	sig->has_child_subreaper = current->signal->has_child_subreaper ||
				   current->signal->is_child_subreaper;

1095 1096
	mutex_init(&sig->cred_guard_mutex);

L
Linus Torvalds 已提交
1097 1098 1099
	return 0;
}

A
Alexey Dobriyan 已提交
1100
static void copy_flags(unsigned long clone_flags, struct task_struct *p)
L
Linus Torvalds 已提交
1101 1102 1103
{
	unsigned long new_flags = p->flags;

T
Tejun Heo 已提交
1104
	new_flags &= ~(PF_SUPERPRIV | PF_WQ_WORKER);
L
Linus Torvalds 已提交
1105 1106 1107 1108
	new_flags |= PF_FORKNOEXEC;
	p->flags = new_flags;
}

1109
SYSCALL_DEFINE1(set_tid_address, int __user *, tidptr)
L
Linus Torvalds 已提交
1110 1111 1112
{
	current->clear_child_tid = tidptr;

1113
	return task_pid_vnr(current);
L
Linus Torvalds 已提交
1114 1115
}

A
Alexey Dobriyan 已提交
1116
static void rt_mutex_init_task(struct task_struct *p)
I
Ingo Molnar 已提交
1117
{
1118
	raw_spin_lock_init(&p->pi_lock);
1119
#ifdef CONFIG_RT_MUTEXES
1120
	plist_head_init(&p->pi_waiters);
I
Ingo Molnar 已提交
1121 1122 1123 1124
	p->pi_blocked_on = NULL;
#endif
}

1125 1126 1127 1128 1129 1130 1131
#ifdef CONFIG_MM_OWNER
void mm_init_owner(struct mm_struct *mm, struct task_struct *p)
{
	mm->owner = p;
}
#endif /* CONFIG_MM_OWNER */

1132 1133 1134 1135 1136
/*
 * Initialize POSIX timer handling for a single task.
 */
static void posix_cpu_timers_init(struct task_struct *tsk)
{
1137 1138
	tsk->cputime_expires.prof_exp = 0;
	tsk->cputime_expires.virt_exp = 0;
1139 1140 1141 1142 1143 1144
	tsk->cputime_expires.sched_exp = 0;
	INIT_LIST_HEAD(&tsk->cpu_timers[0]);
	INIT_LIST_HEAD(&tsk->cpu_timers[1]);
	INIT_LIST_HEAD(&tsk->cpu_timers[2]);
}

L
Linus Torvalds 已提交
1145 1146 1147 1148 1149 1150 1151 1152
/*
 * This creates a new process as a copy of the old one,
 * but does not actually start it yet.
 *
 * It copies the registers, and all the appropriate
 * parts of the process environment (as per the clone
 * flags). The actual kick-off is left to the caller.
 */
1153 1154 1155 1156 1157
static struct task_struct *copy_process(unsigned long clone_flags,
					unsigned long stack_start,
					struct pt_regs *regs,
					unsigned long stack_size,
					int __user *child_tidptr,
R
Roland McGrath 已提交
1158 1159
					struct pid *pid,
					int trace)
L
Linus Torvalds 已提交
1160 1161
{
	int retval;
1162
	struct task_struct *p;
1163
	int cgroup_callbacks_done = 0;
L
Linus Torvalds 已提交
1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182

	if ((clone_flags & (CLONE_NEWNS|CLONE_FS)) == (CLONE_NEWNS|CLONE_FS))
		return ERR_PTR(-EINVAL);

	/*
	 * Thread groups must share signals as well, and detached threads
	 * can only be started up within the thread group.
	 */
	if ((clone_flags & CLONE_THREAD) && !(clone_flags & CLONE_SIGHAND))
		return ERR_PTR(-EINVAL);

	/*
	 * Shared signal handlers imply shared VM. By way of the above,
	 * thread groups also imply shared VM. Blocking this case allows
	 * for various simplifications in other code.
	 */
	if ((clone_flags & CLONE_SIGHAND) && !(clone_flags & CLONE_VM))
		return ERR_PTR(-EINVAL);

1183 1184 1185 1186 1187 1188 1189 1190 1191 1192
	/*
	 * Siblings of global init remain as zombies on exit since they are
	 * not reaped by their parent (swapper). To solve this and to avoid
	 * multi-rooted process trees, prevent global and container-inits
	 * from creating siblings.
	 */
	if ((clone_flags & CLONE_PARENT) &&
				current->signal->flags & SIGNAL_UNKILLABLE)
		return ERR_PTR(-EINVAL);

L
Linus Torvalds 已提交
1193 1194 1195 1196 1197 1198 1199 1200 1201
	retval = security_task_create(clone_flags);
	if (retval)
		goto fork_out;

	retval = -ENOMEM;
	p = dup_task_struct(current);
	if (!p)
		goto fork_out;

1202
	ftrace_graph_init_task(p);
1203
	get_seccomp_filter(p);
1204

1205 1206
	rt_mutex_init_task(p);

I
Ingo Molnar 已提交
1207
#ifdef CONFIG_PROVE_LOCKING
1208 1209 1210
	DEBUG_LOCKS_WARN_ON(!p->hardirqs_enabled);
	DEBUG_LOCKS_WARN_ON(!p->softirqs_enabled);
#endif
L
Linus Torvalds 已提交
1211
	retval = -EAGAIN;
1212
	if (atomic_read(&p->real_cred->user->processes) >=
1213
			task_rlimit(p, RLIMIT_NPROC)) {
L
Linus Torvalds 已提交
1214
		if (!capable(CAP_SYS_ADMIN) && !capable(CAP_SYS_RESOURCE) &&
1215
		    p->real_cred->user != INIT_USER)
L
Linus Torvalds 已提交
1216 1217
			goto bad_fork_free;
	}
1218
	current->flags &= ~PF_NPROC_EXCEEDED;
L
Linus Torvalds 已提交
1219

1220 1221 1222
	retval = copy_creds(p, clone_flags);
	if (retval < 0)
		goto bad_fork_free;
L
Linus Torvalds 已提交
1223 1224 1225 1226 1227 1228

	/*
	 * If multiple threads are within copy_process(), then this check
	 * triggers too late. This doesn't hurt, the check is only there
	 * to stop root fork bombs.
	 */
1229
	retval = -EAGAIN;
L
Linus Torvalds 已提交
1230 1231 1232
	if (nr_threads >= max_threads)
		goto bad_fork_cleanup_count;

A
Al Viro 已提交
1233
	if (!try_module_get(task_thread_info(p)->exec_domain->module))
L
Linus Torvalds 已提交
1234 1235 1236
		goto bad_fork_cleanup_count;

	p->did_exec = 0;
1237
	delayacct_tsk_init(p);	/* Must remain after dup_task_struct() */
L
Linus Torvalds 已提交
1238 1239 1240
	copy_flags(clone_flags, p);
	INIT_LIST_HEAD(&p->children);
	INIT_LIST_HEAD(&p->sibling);
1241
	rcu_copy_process(p);
L
Linus Torvalds 已提交
1242 1243 1244 1245 1246
	p->vfork_done = NULL;
	spin_lock_init(&p->alloc_lock);

	init_sigpending(&p->pending);

1247 1248
	p->utime = p->stime = p->gtime = 0;
	p->utimescaled = p->stimescaled = 0;
1249
#ifndef CONFIG_VIRT_CPU_ACCOUNTING
1250
	p->prev_utime = p->prev_stime = 0;
1251
#endif
1252 1253 1254
#if defined(SPLIT_RSS_COUNTING)
	memset(&p->rss_stat, 0, sizeof(p->rss_stat));
#endif
1255

1256 1257
	p->default_timer_slack_ns = current->timer_slack_ns;

1258
	task_io_accounting_init(&p->ioac);
L
Linus Torvalds 已提交
1259 1260
	acct_clear_integrals(p);

1261
	posix_cpu_timers_init(p);
L
Linus Torvalds 已提交
1262 1263

	do_posix_clock_monotonic_gettime(&p->start_time);
1264 1265
	p->real_start_time = p->start_time;
	monotonic_to_bootbased(&p->real_start_time);
L
Linus Torvalds 已提交
1266 1267
	p->io_context = NULL;
	p->audit_context = NULL;
1268
	if (clone_flags & CLONE_THREAD)
1269
		threadgroup_change_begin(current);
1270
	cgroup_fork(p);
L
Linus Torvalds 已提交
1271
#ifdef CONFIG_NUMA
1272
	p->mempolicy = mpol_dup(p->mempolicy);
1273 1274 1275 1276 1277
	if (IS_ERR(p->mempolicy)) {
		retval = PTR_ERR(p->mempolicy);
		p->mempolicy = NULL;
		goto bad_fork_cleanup_cgroup;
	}
1278
	mpol_fix_fork_child_flag(p);
L
Linus Torvalds 已提交
1279
#endif
1280 1281 1282
#ifdef CONFIG_CPUSETS
	p->cpuset_mem_spread_rotor = NUMA_NO_NODE;
	p->cpuset_slab_spread_rotor = NUMA_NO_NODE;
1283
	seqcount_init(&p->mems_allowed_seq);
1284
#endif
1285 1286
#ifdef CONFIG_TRACE_IRQFLAGS
	p->irq_events = 0;
1287 1288 1289
#ifdef __ARCH_WANT_INTERRUPTS_ON_CTXSW
	p->hardirqs_enabled = 1;
#else
1290
	p->hardirqs_enabled = 0;
1291
#endif
1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303
	p->hardirq_enable_ip = 0;
	p->hardirq_enable_event = 0;
	p->hardirq_disable_ip = _THIS_IP_;
	p->hardirq_disable_event = 0;
	p->softirqs_enabled = 1;
	p->softirq_enable_ip = _THIS_IP_;
	p->softirq_enable_event = 0;
	p->softirq_disable_ip = 0;
	p->softirq_disable_event = 0;
	p->hardirq_context = 0;
	p->softirq_context = 0;
#endif
I
Ingo Molnar 已提交
1304 1305 1306 1307 1308
#ifdef CONFIG_LOCKDEP
	p->lockdep_depth = 0; /* no locks held yet */
	p->curr_chain_key = 0;
	p->lockdep_recursion = 0;
#endif
L
Linus Torvalds 已提交
1309

1310 1311 1312
#ifdef CONFIG_DEBUG_MUTEXES
	p->blocked_on = NULL; /* not blocked yet */
#endif
1313 1314 1315 1316
#ifdef CONFIG_CGROUP_MEM_RES_CTLR
	p->memcg_batch.do_batch = 0;
	p->memcg_batch.memcg = NULL;
#endif
1317

1318
	/* Perform scheduler related setup. Assign this task to a CPU. */
1319
	sched_fork(p);
1320

1321
	retval = perf_event_init_task(p);
1322 1323
	if (retval)
		goto bad_fork_cleanup_policy;
1324 1325
	retval = audit_alloc(p);
	if (retval)
1326
		goto bad_fork_cleanup_policy;
L
Linus Torvalds 已提交
1327
	/* copy all the process information */
1328 1329
	retval = copy_semundo(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1330
		goto bad_fork_cleanup_audit;
1331 1332
	retval = copy_files(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1333
		goto bad_fork_cleanup_semundo;
1334 1335
	retval = copy_fs(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1336
		goto bad_fork_cleanup_files;
1337 1338
	retval = copy_sighand(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1339
		goto bad_fork_cleanup_fs;
1340 1341
	retval = copy_signal(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1342
		goto bad_fork_cleanup_sighand;
1343 1344
	retval = copy_mm(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1345
		goto bad_fork_cleanup_signal;
1346 1347
	retval = copy_namespaces(clone_flags, p);
	if (retval)
D
David Howells 已提交
1348
		goto bad_fork_cleanup_mm;
1349 1350
	retval = copy_io(clone_flags, p);
	if (retval)
1351
		goto bad_fork_cleanup_namespaces;
A
Alexey Dobriyan 已提交
1352
	retval = copy_thread(clone_flags, stack_start, stack_size, p, regs);
L
Linus Torvalds 已提交
1353
	if (retval)
1354
		goto bad_fork_cleanup_io;
L
Linus Torvalds 已提交
1355

1356 1357
	if (pid != &init_struct_pid) {
		retval = -ENOMEM;
1358
		pid = alloc_pid(p->nsproxy->pid_ns);
1359
		if (!pid)
1360
			goto bad_fork_cleanup_io;
1361 1362 1363 1364 1365 1366 1367
	}

	p->pid = pid_nr(pid);
	p->tgid = p->pid;
	if (clone_flags & CLONE_THREAD)
		p->tgid = current->tgid;

L
Linus Torvalds 已提交
1368 1369 1370 1371
	p->set_child_tid = (clone_flags & CLONE_CHILD_SETTID) ? child_tidptr : NULL;
	/*
	 * Clear TID on mm_release()?
	 */
1372
	p->clear_child_tid = (clone_flags & CLONE_CHILD_CLEARTID) ? child_tidptr : NULL;
1373 1374 1375
#ifdef CONFIG_BLOCK
	p->plug = NULL;
#endif
1376
#ifdef CONFIG_FUTEX
1377 1378 1379 1380
	p->robust_list = NULL;
#ifdef CONFIG_COMPAT
	p->compat_robust_list = NULL;
#endif
1381 1382
	INIT_LIST_HEAD(&p->pi_state_list);
	p->pi_state_cache = NULL;
1383
#endif
1384
	uprobe_copy_process(p);
1385 1386 1387 1388 1389 1390
	/*
	 * sigaltstack should be cleared when sharing the same VM
	 */
	if ((clone_flags & (CLONE_VM|CLONE_VFORK)) == CLONE_VM)
		p->sas_ss_sp = p->sas_ss_size = 0;

L
Linus Torvalds 已提交
1391
	/*
1392 1393
	 * Syscall tracing and stepping should be turned off in the
	 * child regardless of CLONE_PTRACE.
L
Linus Torvalds 已提交
1394
	 */
1395
	user_disable_single_step(p);
L
Linus Torvalds 已提交
1396
	clear_tsk_thread_flag(p, TIF_SYSCALL_TRACE);
1397 1398 1399
#ifdef TIF_SYSCALL_EMU
	clear_tsk_thread_flag(p, TIF_SYSCALL_EMU);
#endif
A
Arjan van de Ven 已提交
1400
	clear_all_latency_tracing(p);
L
Linus Torvalds 已提交
1401 1402

	/* ok, now we should be set up.. */
1403 1404 1405 1406 1407 1408 1409
	if (clone_flags & CLONE_THREAD)
		p->exit_signal = -1;
	else if (clone_flags & CLONE_PARENT)
		p->exit_signal = current->group_leader->exit_signal;
	else
		p->exit_signal = (clone_flags & CSIGNAL);

L
Linus Torvalds 已提交
1410 1411 1412
	p->pdeath_signal = 0;
	p->exit_state = 0;

1413 1414
	p->nr_dirtied = 0;
	p->nr_dirtied_pause = 128 >> (PAGE_SHIFT - 10);
1415
	p->dirty_paused_when = 0;
1416

L
Linus Torvalds 已提交
1417 1418 1419 1420 1421
	/*
	 * Ok, make it visible to the rest of the system.
	 * We dont wake it up yet.
	 */
	p->group_leader = p;
O
Oleg Nesterov 已提交
1422
	INIT_LIST_HEAD(&p->thread_group);
A
Al Viro 已提交
1423
	p->task_works = NULL;
L
Linus Torvalds 已提交
1424

1425 1426 1427 1428 1429 1430
	/* Now that the task is set up, run cgroup callbacks if
	 * necessary. We need to run them before the task is visible
	 * on the tasklist. */
	cgroup_fork_callbacks(p);
	cgroup_callbacks_done = 1;

L
Linus Torvalds 已提交
1431 1432 1433 1434
	/* Need tasklist lock for parent etc handling! */
	write_lock_irq(&tasklist_lock);

	/* CLONE_PARENT re-uses the old parent */
1435
	if (clone_flags & (CLONE_PARENT|CLONE_THREAD)) {
L
Linus Torvalds 已提交
1436
		p->real_parent = current->real_parent;
1437 1438
		p->parent_exec_id = current->parent_exec_id;
	} else {
L
Linus Torvalds 已提交
1439
		p->real_parent = current;
1440 1441
		p->parent_exec_id = current->self_exec_id;
	}
L
Linus Torvalds 已提交
1442

1443
	spin_lock(&current->sighand->siglock);
1444 1445 1446 1447 1448 1449 1450 1451

	/*
	 * Process group and session signals need to be delivered to just the
	 * parent before the fork or both the parent and the child after the
	 * fork. Restart if a signal comes in before we add the new process to
	 * it's process group.
	 * A fatal signal pending means that current will exit, so the new
	 * thread can't slip out of an OOM kill (or normal SIGKILL).
1452
	*/
D
Daniel Walker 已提交
1453
	recalc_sigpending();
1454 1455 1456 1457
	if (signal_pending(current)) {
		spin_unlock(&current->sighand->siglock);
		write_unlock_irq(&tasklist_lock);
		retval = -ERESTARTNOINTR;
1458
		goto bad_fork_free_pid;
1459 1460
	}

L
Linus Torvalds 已提交
1461
	if (clone_flags & CLONE_THREAD) {
1462
		current->signal->nr_threads++;
1463
		atomic_inc(&current->signal->live);
1464
		atomic_inc(&current->signal->sigcnt);
L
Linus Torvalds 已提交
1465
		p->group_leader = current->group_leader;
O
Oleg Nesterov 已提交
1466
		list_add_tail_rcu(&p->thread_group, &p->group_leader->thread_group);
L
Linus Torvalds 已提交
1467 1468
	}

1469
	if (likely(p->pid)) {
T
Tejun Heo 已提交
1470
		ptrace_init_task(p, (clone_flags & CLONE_PTRACE) || trace);
1471 1472

		if (thread_group_leader(p)) {
1473
			if (is_child_reaper(pid))
1474
				p->nsproxy->pid_ns->child_reaper = p;
1475

1476
			p->signal->leader_pid = pid;
A
Alan Cox 已提交
1477
			p->signal->tty = tty_kref_get(current->signal->tty);
E
Eric W. Biederman 已提交
1478 1479
			attach_pid(p, PIDTYPE_PGID, task_pgrp(current));
			attach_pid(p, PIDTYPE_SID, task_session(current));
1480
			list_add_tail(&p->sibling, &p->real_parent->children);
1481
			list_add_tail_rcu(&p->tasks, &init_task.tasks);
1482
			__this_cpu_inc(process_counts);
1483
		}
1484
		attach_pid(p, PIDTYPE_PID, pid);
1485
		nr_threads++;
L
Linus Torvalds 已提交
1486 1487 1488
	}

	total_forks++;
1489
	spin_unlock(&current->sighand->siglock);
L
Linus Torvalds 已提交
1490
	write_unlock_irq(&tasklist_lock);
1491
	proc_fork_connector(p);
1492
	cgroup_post_fork(p);
1493
	if (clone_flags & CLONE_THREAD)
1494
		threadgroup_change_end(current);
1495
	perf_event_fork(p);
1496 1497 1498

	trace_task_newtask(p, clone_flags);

L
Linus Torvalds 已提交
1499 1500
	return p;

1501 1502 1503
bad_fork_free_pid:
	if (pid != &init_struct_pid)
		free_pid(pid);
1504
bad_fork_cleanup_io:
1505 1506
	if (p->io_context)
		exit_io_context(p);
S
Serge E. Hallyn 已提交
1507
bad_fork_cleanup_namespaces:
1508 1509
	if (unlikely(clone_flags & CLONE_NEWPID))
		pid_ns_release_proc(p->nsproxy->pid_ns);
1510
	exit_task_namespaces(p);
L
Linus Torvalds 已提交
1511
bad_fork_cleanup_mm:
D
David Rientjes 已提交
1512
	if (p->mm)
L
Linus Torvalds 已提交
1513 1514
		mmput(p->mm);
bad_fork_cleanup_signal:
1515
	if (!(clone_flags & CLONE_THREAD))
1516
		free_signal_struct(p->signal);
L
Linus Torvalds 已提交
1517
bad_fork_cleanup_sighand:
1518
	__cleanup_sighand(p->sighand);
L
Linus Torvalds 已提交
1519 1520 1521 1522 1523 1524 1525 1526 1527
bad_fork_cleanup_fs:
	exit_fs(p); /* blocking */
bad_fork_cleanup_files:
	exit_files(p); /* blocking */
bad_fork_cleanup_semundo:
	exit_sem(p);
bad_fork_cleanup_audit:
	audit_free(p);
bad_fork_cleanup_policy:
1528
	perf_event_free_task(p);
L
Linus Torvalds 已提交
1529
#ifdef CONFIG_NUMA
1530
	mpol_put(p->mempolicy);
1531
bad_fork_cleanup_cgroup:
L
Linus Torvalds 已提交
1532
#endif
1533
	if (clone_flags & CLONE_THREAD)
1534
		threadgroup_change_end(current);
1535
	cgroup_exit(p, cgroup_callbacks_done);
1536
	delayacct_tsk_free(p);
A
Al Viro 已提交
1537
	module_put(task_thread_info(p)->exec_domain->module);
L
Linus Torvalds 已提交
1538
bad_fork_cleanup_count:
D
David Howells 已提交
1539
	atomic_dec(&p->cred->user->processes);
1540
	exit_creds(p);
L
Linus Torvalds 已提交
1541 1542
bad_fork_free:
	free_task(p);
1543 1544
fork_out:
	return ERR_PTR(retval);
L
Linus Torvalds 已提交
1545 1546
}

A
Adrian Bunk 已提交
1547
noinline struct pt_regs * __cpuinit __attribute__((weak)) idle_regs(struct pt_regs *regs)
L
Linus Torvalds 已提交
1548 1549 1550 1551 1552
{
	memset(regs, 0, sizeof(struct pt_regs));
	return regs;
}

1553 1554 1555 1556 1557 1558 1559 1560 1561 1562
static inline void init_idle_pids(struct pid_link *links)
{
	enum pid_type type;

	for (type = PIDTYPE_PID; type < PIDTYPE_MAX; ++type) {
		INIT_HLIST_NODE(&links[type].node); /* not really needed */
		links[type].pid = &init_struct_pid;
	}
}

1563
struct task_struct * __cpuinit fork_idle(int cpu)
L
Linus Torvalds 已提交
1564
{
1565
	struct task_struct *task;
L
Linus Torvalds 已提交
1566 1567
	struct pt_regs regs;

1568
	task = copy_process(CLONE_VM, 0, idle_regs(&regs), 0, NULL,
R
Roland McGrath 已提交
1569
			    &init_struct_pid, 0);
1570 1571
	if (!IS_ERR(task)) {
		init_idle_pids(task->pids);
1572
		init_idle(task, cpu);
1573
	}
1574

L
Linus Torvalds 已提交
1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592
	return task;
}

/*
 *  Ok, this is the main fork-routine.
 *
 * It copies the process, and if successful kick-starts
 * it and waits for it to finish using the VM if required.
 */
long do_fork(unsigned long clone_flags,
	      unsigned long stack_start,
	      struct pt_regs *regs,
	      unsigned long stack_size,
	      int __user *parent_tidptr,
	      int __user *child_tidptr)
{
	struct task_struct *p;
	int trace = 0;
1593
	long nr;
L
Linus Torvalds 已提交
1594

1595 1596 1597 1598 1599 1600 1601 1602 1603 1604
	/*
	 * Do some preliminary argument and permissions checking before we
	 * actually start allocating stuff
	 */
	if (clone_flags & CLONE_NEWUSER) {
		if (clone_flags & CLONE_THREAD)
			return -EINVAL;
		/* hopefully this check will go away when userns support is
		 * complete
		 */
1605 1606
		if (!capable(CAP_SYS_ADMIN) || !capable(CAP_SETUID) ||
				!capable(CAP_SETGID))
1607 1608 1609
			return -EPERM;
	}

R
Roland McGrath 已提交
1610
	/*
T
Tejun Heo 已提交
1611 1612 1613 1614
	 * Determine whether and which event to report to ptracer.  When
	 * called from kernel_thread or CLONE_UNTRACED is explicitly
	 * requested, no event is reported; otherwise, report if the event
	 * for the type of forking is enabled.
R
Roland McGrath 已提交
1615
	 */
T
Tejun Heo 已提交
1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626
	if (likely(user_mode(regs)) && !(clone_flags & CLONE_UNTRACED)) {
		if (clone_flags & CLONE_VFORK)
			trace = PTRACE_EVENT_VFORK;
		else if ((clone_flags & CSIGNAL) != SIGCHLD)
			trace = PTRACE_EVENT_CLONE;
		else
			trace = PTRACE_EVENT_FORK;

		if (likely(!ptrace_event_enabled(current, trace)))
			trace = 0;
	}
L
Linus Torvalds 已提交
1627

1628
	p = copy_process(clone_flags, stack_start, regs, stack_size,
R
Roland McGrath 已提交
1629
			 child_tidptr, NULL, trace);
L
Linus Torvalds 已提交
1630 1631 1632 1633 1634 1635 1636
	/*
	 * Do this prior waking up the new thread - the thread pointer
	 * might get invalid after that point, if the thread exits quickly.
	 */
	if (!IS_ERR(p)) {
		struct completion vfork;

1637 1638
		trace_sched_process_fork(current, p);

1639
		nr = task_pid_vnr(p);
1640 1641 1642

		if (clone_flags & CLONE_PARENT_SETTID)
			put_user(nr, parent_tidptr);
1643

L
Linus Torvalds 已提交
1644 1645 1646
		if (clone_flags & CLONE_VFORK) {
			p->vfork_done = &vfork;
			init_completion(&vfork);
O
Oleg Nesterov 已提交
1647
			get_task_struct(p);
L
Linus Torvalds 已提交
1648 1649
		}

1650
		wake_up_new_task(p);
L
Linus Torvalds 已提交
1651

T
Tejun Heo 已提交
1652 1653 1654
		/* forking complete and child started to run, tell ptracer */
		if (unlikely(trace))
			ptrace_event(trace, nr);
R
Roland McGrath 已提交
1655

L
Linus Torvalds 已提交
1656
		if (clone_flags & CLONE_VFORK) {
O
Oleg Nesterov 已提交
1657 1658
			if (!wait_for_vfork_done(p, &vfork))
				ptrace_event(PTRACE_EVENT_VFORK_DONE, nr);
L
Linus Torvalds 已提交
1659 1660
		}
	} else {
1661
		nr = PTR_ERR(p);
L
Linus Torvalds 已提交
1662
	}
1663
	return nr;
L
Linus Torvalds 已提交
1664 1665
}

1666 1667 1668 1669
#ifndef ARCH_MIN_MMSTRUCT_ALIGN
#define ARCH_MIN_MMSTRUCT_ALIGN 0
#endif

1670
static void sighand_ctor(void *data)
1671 1672 1673
{
	struct sighand_struct *sighand = data;

C
Christoph Lameter 已提交
1674
	spin_lock_init(&sighand->siglock);
D
Davide Libenzi 已提交
1675
	init_waitqueue_head(&sighand->signalfd_wqh);
1676 1677
}

L
Linus Torvalds 已提交
1678 1679 1680 1681
void __init proc_caches_init(void)
{
	sighand_cachep = kmem_cache_create("sighand_cache",
			sizeof(struct sighand_struct), 0,
V
Vegard Nossum 已提交
1682 1683
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_DESTROY_BY_RCU|
			SLAB_NOTRACK, sighand_ctor);
L
Linus Torvalds 已提交
1684 1685
	signal_cachep = kmem_cache_create("signal_cache",
			sizeof(struct signal_struct), 0,
V
Vegard Nossum 已提交
1686
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1687
	files_cachep = kmem_cache_create("files_cache",
L
Linus Torvalds 已提交
1688
			sizeof(struct files_struct), 0,
V
Vegard Nossum 已提交
1689
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1690
	fs_cachep = kmem_cache_create("fs_cache",
L
Linus Torvalds 已提交
1691
			sizeof(struct fs_struct), 0,
V
Vegard Nossum 已提交
1692
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1693 1694 1695 1696 1697 1698 1699
	/*
	 * FIXME! The "sizeof(struct mm_struct)" currently includes the
	 * whole struct cpumask for the OFFSTACK case. We could change
	 * this to *only* allocate as much of it as required by the
	 * maximum number of CPU's we can ever have.  The cpumask_allocation
	 * is at the end of the structure, exactly for that reason.
	 */
L
Linus Torvalds 已提交
1700
	mm_cachep = kmem_cache_create("mm_struct",
1701
			sizeof(struct mm_struct), ARCH_MIN_MMSTRUCT_ALIGN,
V
Vegard Nossum 已提交
1702
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1703
	vm_area_cachep = KMEM_CACHE(vm_area_struct, SLAB_PANIC);
1704
	mmap_init();
1705
	nsproxy_cache_init();
L
Linus Torvalds 已提交
1706
}
1707 1708

/*
1709
 * Check constraints on flags passed to the unshare system call.
1710
 */
1711
static int check_unshare_flags(unsigned long unshare_flags)
1712
{
1713 1714 1715 1716
	if (unshare_flags & ~(CLONE_THREAD|CLONE_FS|CLONE_NEWNS|CLONE_SIGHAND|
				CLONE_VM|CLONE_FILES|CLONE_SYSVSEM|
				CLONE_NEWUTS|CLONE_NEWIPC|CLONE_NEWNET))
		return -EINVAL;
1717
	/*
1718 1719 1720
	 * Not implemented, but pretend it works if there is nothing to
	 * unshare. Note that unsharing CLONE_THREAD or CLONE_SIGHAND
	 * needs to unshare vm.
1721
	 */
1722 1723 1724 1725 1726
	if (unshare_flags & (CLONE_THREAD | CLONE_SIGHAND | CLONE_VM)) {
		/* FIXME: get_task_mm() increments ->mm_users */
		if (atomic_read(&current->mm->mm_users) > 1)
			return -EINVAL;
	}
1727 1728 1729 1730 1731

	return 0;
}

/*
1732
 * Unshare the filesystem structure if it is being shared
1733 1734 1735 1736 1737
 */
static int unshare_fs(unsigned long unshare_flags, struct fs_struct **new_fsp)
{
	struct fs_struct *fs = current->fs;

A
Al Viro 已提交
1738 1739 1740 1741 1742 1743 1744 1745 1746 1747
	if (!(unshare_flags & CLONE_FS) || !fs)
		return 0;

	/* don't need lock here; in the worst case we'll do useless copy */
	if (fs->users == 1)
		return 0;

	*new_fsp = copy_fs_struct(fs);
	if (!*new_fsp)
		return -ENOMEM;
1748 1749 1750 1751 1752

	return 0;
}

/*
1753
 * Unshare file descriptor table if it is being shared
1754 1755 1756 1757
 */
static int unshare_fd(unsigned long unshare_flags, struct files_struct **new_fdp)
{
	struct files_struct *fd = current->files;
1758
	int error = 0;
1759 1760

	if ((unshare_flags & CLONE_FILES) &&
1761 1762 1763 1764 1765
	    (fd && atomic_read(&fd->count) > 1)) {
		*new_fdp = dup_fd(fd, &error);
		if (!*new_fdp)
			return error;
	}
1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777

	return 0;
}

/*
 * unshare allows a process to 'unshare' part of the process
 * context which was originally shared using clone.  copy_*
 * functions used by do_fork() cannot be used here directly
 * because they modify an inactive task_struct that is being
 * constructed. Here we are modifying the current, active,
 * task_struct.
 */
1778
SYSCALL_DEFINE1(unshare, unsigned long, unshare_flags)
1779 1780 1781
{
	struct fs_struct *fs, *new_fs = NULL;
	struct files_struct *fd, *new_fd = NULL;
1782
	struct nsproxy *new_nsproxy = NULL;
1783
	int do_sysvsem = 0;
1784
	int err;
1785

1786 1787
	err = check_unshare_flags(unshare_flags);
	if (err)
1788 1789
		goto bad_unshare_out;

1790 1791 1792 1793 1794
	/*
	 * If unsharing namespace, must also unshare filesystem information.
	 */
	if (unshare_flags & CLONE_NEWNS)
		unshare_flags |= CLONE_FS;
1795 1796 1797 1798 1799 1800
	/*
	 * CLONE_NEWIPC must also detach from the undolist: after switching
	 * to a new ipc namespace, the semaphore arrays from the old
	 * namespace are unreachable.
	 */
	if (unshare_flags & (CLONE_NEWIPC|CLONE_SYSVSEM))
1801
		do_sysvsem = 1;
1802 1803
	err = unshare_fs(unshare_flags, &new_fs);
	if (err)
1804
		goto bad_unshare_out;
1805 1806
	err = unshare_fd(unshare_flags, &new_fd);
	if (err)
1807
		goto bad_unshare_cleanup_fs;
1808 1809
	err = unshare_nsproxy_namespaces(unshare_flags, &new_nsproxy, new_fs);
	if (err)
1810
		goto bad_unshare_cleanup_fd;
1811

1812
	if (new_fs || new_fd || do_sysvsem || new_nsproxy) {
1813 1814 1815 1816 1817 1818
		if (do_sysvsem) {
			/*
			 * CLONE_SYSVSEM is equivalent to sys_exit().
			 */
			exit_sem(current);
		}
S
Serge E. Hallyn 已提交
1819

1820
		if (new_nsproxy) {
1821 1822
			switch_task_namespaces(current, new_nsproxy);
			new_nsproxy = NULL;
1823
		}
1824

1825 1826
		task_lock(current);

1827 1828
		if (new_fs) {
			fs = current->fs;
N
Nick Piggin 已提交
1829
			spin_lock(&fs->lock);
1830
			current->fs = new_fs;
A
Al Viro 已提交
1831 1832 1833 1834
			if (--fs->users)
				new_fs = NULL;
			else
				new_fs = fs;
N
Nick Piggin 已提交
1835
			spin_unlock(&fs->lock);
1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846
		}

		if (new_fd) {
			fd = current->files;
			current->files = new_fd;
			new_fd = fd;
		}

		task_unlock(current);
	}

1847
	if (new_nsproxy)
1848
		put_nsproxy(new_nsproxy);
1849

1850 1851 1852 1853 1854 1855
bad_unshare_cleanup_fd:
	if (new_fd)
		put_files_struct(new_fd);

bad_unshare_cleanup_fs:
	if (new_fs)
A
Al Viro 已提交
1856
		free_fs_struct(new_fs);
1857 1858 1859 1860

bad_unshare_out:
	return err;
}
1861 1862 1863 1864 1865 1866 1867 1868 1869 1870

/*
 *	Helper to unshare the files of the current task.
 *	We don't want to expose copy_files internals to
 *	the exec layer of the kernel.
 */

int unshare_files(struct files_struct **displaced)
{
	struct task_struct *task = current;
1871
	struct files_struct *copy = NULL;
1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884
	int error;

	error = unshare_fd(CLONE_FILES, &copy);
	if (error || !copy) {
		*displaced = NULL;
		return error;
	}
	*displaced = task->files;
	task_lock(task);
	task->files = copy;
	task_unlock(task);
	return 0;
}