fork.c 60.9 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14
/*
 *  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>
15
#include <linux/sched/autogroup.h>
16
#include <linux/sched/mm.h>
17
#include <linux/sched/coredump.h>
18
#include <linux/sched/user.h>
19
#include <linux/sched/numa_balancing.h>
20
#include <linux/sched/stat.h>
21
#include <linux/sched/task.h>
22
#include <linux/sched/task_stack.h>
23
#include <linux/sched/cputime.h>
24
#include <linux/rtmutex.h>
L
Linus Torvalds 已提交
25 26 27 28 29 30 31 32 33
#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 已提交
34
#include <linux/fdtable.h>
35
#include <linux/iocontext.h>
L
Linus Torvalds 已提交
36 37 38
#include <linux/key.h>
#include <linux/binfmts.h>
#include <linux/mman.h>
A
Andrea Arcangeli 已提交
39
#include <linux/mmu_notifier.h>
40
#include <linux/hmm.h>
L
Linus Torvalds 已提交
41
#include <linux/fs.h>
D
Davidlohr Bueso 已提交
42 43
#include <linux/mm.h>
#include <linux/vmacache.h>
S
Serge E. Hallyn 已提交
44
#include <linux/nsproxy.h>
45
#include <linux/capability.h>
L
Linus Torvalds 已提交
46
#include <linux/cpu.h>
47
#include <linux/cgroup.h>
L
Linus Torvalds 已提交
48
#include <linux/security.h>
49
#include <linux/hugetlb.h>
50
#include <linux/seccomp.h>
L
Linus Torvalds 已提交
51 52 53 54
#include <linux/swap.h>
#include <linux/syscalls.h>
#include <linux/jiffies.h>
#include <linux/futex.h>
55
#include <linux/compat.h>
56
#include <linux/kthread.h>
57
#include <linux/task_io_accounting_ops.h>
58
#include <linux/rcupdate.h>
L
Linus Torvalds 已提交
59 60 61
#include <linux/ptrace.h>
#include <linux/mount.h>
#include <linux/audit.h>
62
#include <linux/memcontrol.h>
63
#include <linux/ftrace.h>
64
#include <linux/proc_fs.h>
L
Linus Torvalds 已提交
65 66
#include <linux/profile.h>
#include <linux/rmap.h>
H
Hugh Dickins 已提交
67
#include <linux/ksm.h>
L
Linus Torvalds 已提交
68
#include <linux/acct.h>
69
#include <linux/userfaultfd_k.h>
70
#include <linux/tsacct_kern.h>
M
Matt Helsley 已提交
71
#include <linux/cn_proc.h>
R
Rafael J. Wysocki 已提交
72
#include <linux/freezer.h>
73
#include <linux/delayacct.h>
74
#include <linux/taskstats_kern.h>
75
#include <linux/random.h>
M
Miloslav Trmac 已提交
76
#include <linux/tty.h>
77
#include <linux/blkdev.h>
78
#include <linux/fs_struct.h>
79
#include <linux/magic.h>
80
#include <linux/sched/mm.h>
81
#include <linux/perf_event.h>
82
#include <linux/posix-timers.h>
83
#include <linux/user-return-notifier.h>
Y
Ying Han 已提交
84
#include <linux/oom.h>
A
Andrea Arcangeli 已提交
85
#include <linux/khugepaged.h>
86
#include <linux/signalfd.h>
87
#include <linux/uprobes.h>
88
#include <linux/aio.h>
89
#include <linux/compiler.h>
90
#include <linux/sysctl.h>
D
Dmitry Vyukov 已提交
91
#include <linux/kcov.h>
92
#include <linux/livepatch.h>
93
#include <linux/thread_info.h>
L
Linus Torvalds 已提交
94 95 96

#include <asm/pgtable.h>
#include <asm/pgalloc.h>
97
#include <linux/uaccess.h>
L
Linus Torvalds 已提交
98 99 100 101
#include <asm/mmu_context.h>
#include <asm/cacheflush.h>
#include <asm/tlbflush.h>

102 103
#include <trace/events/sched.h>

104 105 106
#define CREATE_TRACE_POINTS
#include <trace/events/task.h>

107 108 109 110 111 112 113 114 115 116
/*
 * Minimum number of threads to boot the kernel
 */
#define MIN_THREADS 20

/*
 * Maximum number of threads
 */
#define MAX_THREADS FUTEX_TID_MASK

L
Linus Torvalds 已提交
117 118 119 120
/*
 * Protected counters by write_lock_irq(&tasklist_lock)
 */
unsigned long total_forks;	/* Handle normal Linux uptimes. */
121
int nr_threads;			/* The idle threads do not count.. */
L
Linus Torvalds 已提交
122 123 124 125 126

int max_threads;		/* tunable limit on nr_threads */

DEFINE_PER_CPU(unsigned long, process_counts) = 0;

127
__cacheline_aligned DEFINE_RWLOCK(tasklist_lock);  /* outer */
128 129 130 131 132 133 134 135

#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 已提交
136 137 138 139 140 141

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

142
	for_each_possible_cpu(cpu)
L
Linus Torvalds 已提交
143 144 145 146 147
		total += per_cpu(process_counts, cpu);

	return total;
}

148 149 150 151
void __weak arch_release_task_struct(struct task_struct *tsk)
{
}

152
#ifndef CONFIG_ARCH_TASK_STRUCT_ALLOCATOR
153
static struct kmem_cache *task_struct_cachep;
154 155 156 157 158 159 160 161 162 163

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

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

166
void __weak arch_release_thread_stack(unsigned long *stack)
167 168 169
{
}

170
#ifndef CONFIG_ARCH_THREAD_STACK_ALLOCATOR
171

172 173 174 175
/*
 * Allocate pages if THREAD_SIZE is >= PAGE_SIZE, otherwise use a
 * kmemcache based allocator.
 */
176
# if THREAD_SIZE >= PAGE_SIZE || defined(CONFIG_VMAP_STACK)
177 178 179 180 181 182 183 184

#ifdef CONFIG_VMAP_STACK
/*
 * vmalloc() is a bit slow, and calling vfree() enough times will force a TLB
 * flush.  Try to minimize the number of calls by caching stacks.
 */
#define NR_CACHED_STACKS 2
static DEFINE_PER_CPU(struct vm_struct *, cached_stacks[NR_CACHED_STACKS]);
185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202

static int free_vm_stack_cache(unsigned int cpu)
{
	struct vm_struct **cached_vm_stacks = per_cpu_ptr(cached_stacks, cpu);
	int i;

	for (i = 0; i < NR_CACHED_STACKS; i++) {
		struct vm_struct *vm_stack = cached_vm_stacks[i];

		if (!vm_stack)
			continue;

		vfree(vm_stack->addr);
		cached_vm_stacks[i] = NULL;
	}

	return 0;
}
203 204
#endif

205
static unsigned long *alloc_thread_stack_node(struct task_struct *tsk, int node)
206
{
207
#ifdef CONFIG_VMAP_STACK
208 209 210 211
	void *stack;
	int i;

	for (i = 0; i < NR_CACHED_STACKS; i++) {
212 213 214
		struct vm_struct *s;

		s = this_cpu_xchg(cached_stacks[i], NULL);
215 216 217 218

		if (!s)
			continue;

219 220
		/* Clear stale pointers from reused stack. */
		memset(s->addr, 0, THREAD_SIZE);
221

222 223 224 225
		tsk->stack_vm_area = s;
		return s->addr;
	}

226
	stack = __vmalloc_node_range(THREAD_SIZE, THREAD_ALIGN,
227
				     VMALLOC_START, VMALLOC_END,
228
				     THREADINFO_GFP,
229 230
				     PAGE_KERNEL,
				     0, node, __builtin_return_address(0));
231 232 233 234 235 236 237 238 239 240

	/*
	 * We can't call find_vm_area() in interrupt context, and
	 * free_thread_stack() can be called in interrupt context,
	 * so cache the vm_struct.
	 */
	if (stack)
		tsk->stack_vm_area = find_vm_area(stack);
	return stack;
#else
241 242
	struct page *page = alloc_pages_node(node, THREADINFO_GFP,
					     THREAD_SIZE_ORDER);
243 244

	return page ? page_address(page) : NULL;
245
#endif
246 247
}

248
static inline void free_thread_stack(struct task_struct *tsk)
249
{
250 251 252 253 254
#ifdef CONFIG_VMAP_STACK
	if (task_stack_vm_area(tsk)) {
		int i;

		for (i = 0; i < NR_CACHED_STACKS; i++) {
255 256
			if (this_cpu_cmpxchg(cached_stacks[i],
					NULL, tsk->stack_vm_area) != NULL)
257 258 259 260 261
				continue;

			return;
		}

262
		vfree_atomic(tsk->stack);
263 264 265 266 267
		return;
	}
#endif

	__free_pages(virt_to_page(tsk->stack), THREAD_SIZE_ORDER);
268
}
269
# else
270
static struct kmem_cache *thread_stack_cache;
271

272
static unsigned long *alloc_thread_stack_node(struct task_struct *tsk,
273 274
						  int node)
{
275
	return kmem_cache_alloc_node(thread_stack_cache, THREADINFO_GFP, node);
276 277
}

278
static void free_thread_stack(struct task_struct *tsk)
279
{
280
	kmem_cache_free(thread_stack_cache, tsk->stack);
281 282
}

283
void thread_stack_cache_init(void)
284
{
285 286 287
	thread_stack_cache = kmem_cache_create_usercopy("thread_stack",
					THREAD_SIZE, THREAD_SIZE, 0, 0,
					THREAD_SIZE, NULL);
288
	BUG_ON(thread_stack_cache == NULL);
289 290
}
# endif
291 292
#endif

L
Linus Torvalds 已提交
293
/* SLAB cache for signal_struct structures (tsk->signal) */
294
static struct kmem_cache *signal_cachep;
L
Linus Torvalds 已提交
295 296

/* SLAB cache for sighand_struct structures (tsk->sighand) */
297
struct kmem_cache *sighand_cachep;
L
Linus Torvalds 已提交
298 299

/* SLAB cache for files_struct structures (tsk->files) */
300
struct kmem_cache *files_cachep;
L
Linus Torvalds 已提交
301 302

/* SLAB cache for fs_struct structures (tsk->fs) */
303
struct kmem_cache *fs_cachep;
L
Linus Torvalds 已提交
304 305

/* SLAB cache for vm_area_struct structures */
306
static struct kmem_cache *vm_area_cachep;
L
Linus Torvalds 已提交
307 308

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

311
struct vm_area_struct *vm_area_alloc(struct mm_struct *mm)
312
{
313 314
	struct vm_area_struct *vma = kmem_cache_zalloc(vm_area_cachep, GFP_KERNEL);

K
Kirill A. Shutemov 已提交
315 316
	if (vma)
		vma_init(vma, mm);
317
	return vma;
318 319 320 321
}

struct vm_area_struct *vm_area_dup(struct vm_area_struct *orig)
{
322 323 324 325 326 327 328
	struct vm_area_struct *new = kmem_cache_alloc(vm_area_cachep, GFP_KERNEL);

	if (new) {
		*new = *orig;
		INIT_LIST_HEAD(&new->anon_vma_chain);
	}
	return new;
329 330 331 332 333 334 335
}

void vm_area_free(struct vm_area_struct *vma)
{
	kmem_cache_free(vm_area_cachep, vma);
}

336
static void account_kernel_stack(struct task_struct *tsk, int account)
337
{
338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354
	void *stack = task_stack_page(tsk);
	struct vm_struct *vm = task_stack_vm_area(tsk);

	BUILD_BUG_ON(IS_ENABLED(CONFIG_VMAP_STACK) && PAGE_SIZE % 1024 != 0);

	if (vm) {
		int i;

		BUG_ON(vm->nr_pages != THREAD_SIZE / PAGE_SIZE);

		for (i = 0; i < THREAD_SIZE / PAGE_SIZE; i++) {
			mod_zone_page_state(page_zone(vm->pages[i]),
					    NR_KERNEL_STACK_KB,
					    PAGE_SIZE / 1024 * account);
		}

		/* All stack pages belong to the same memcg. */
355 356
		mod_memcg_page_state(vm->pages[0], MEMCG_KERNEL_STACK_KB,
				     account * (THREAD_SIZE / 1024));
357 358 359 360 361 362 363 364 365 366
	} else {
		/*
		 * All stack pages are in the same zone and belong to the
		 * same memcg.
		 */
		struct page *first_page = virt_to_page(stack);

		mod_zone_page_state(page_zone(first_page), NR_KERNEL_STACK_KB,
				    THREAD_SIZE / 1024 * account);

367 368
		mod_memcg_page_state(first_page, MEMCG_KERNEL_STACK_KB,
				     account * (THREAD_SIZE / 1024));
369
	}
370 371
}

372
static void release_task_stack(struct task_struct *tsk)
L
Linus Torvalds 已提交
373
{
374 375 376
	if (WARN_ON(tsk->state != TASK_DEAD))
		return;  /* Better to leak the stack than to free prematurely */

377
	account_kernel_stack(tsk, -1);
378
	arch_release_thread_stack(tsk->stack);
379
	free_thread_stack(tsk);
380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408
	tsk->stack = NULL;
#ifdef CONFIG_VMAP_STACK
	tsk->stack_vm_area = NULL;
#endif
}

#ifdef CONFIG_THREAD_INFO_IN_TASK
void put_task_stack(struct task_struct *tsk)
{
	if (atomic_dec_and_test(&tsk->stack_refcount))
		release_task_stack(tsk);
}
#endif

void free_task(struct task_struct *tsk)
{
#ifndef CONFIG_THREAD_INFO_IN_TASK
	/*
	 * The task is finally done with both the stack and thread_info,
	 * so free both.
	 */
	release_task_stack(tsk);
#else
	/*
	 * If the task had a separate stack allocation, it should be gone
	 * by now.
	 */
	WARN_ON_ONCE(atomic_read(&tsk->stack_refcount) != 0);
#endif
I
Ingo Molnar 已提交
409
	rt_mutex_debug_task_free(tsk);
410
	ftrace_graph_exit_task(tsk);
411
	put_seccomp_filter(tsk);
412
	arch_release_task_struct(tsk);
413 414
	if (tsk->flags & PF_KTHREAD)
		free_kthread_struct(tsk);
L
Linus Torvalds 已提交
415 416 417 418
	free_task_struct(tsk);
}
EXPORT_SYMBOL(free_task);

419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467
#ifdef CONFIG_MMU
static __latent_entropy int dup_mmap(struct mm_struct *mm,
					struct mm_struct *oldmm)
{
	struct vm_area_struct *mpnt, *tmp, *prev, **pprev;
	struct rb_node **rb_link, *rb_parent;
	int retval;
	unsigned long charge;
	LIST_HEAD(uf);

	uprobe_start_dup_mmap();
	if (down_write_killable(&oldmm->mmap_sem)) {
		retval = -EINTR;
		goto fail_uprobe_end;
	}
	flush_cache_dup_mm(oldmm);
	uprobe_dup_mmap(oldmm, mm);
	/*
	 * Not linked in yet - no deadlock potential:
	 */
	down_write_nested(&mm->mmap_sem, SINGLE_DEPTH_NESTING);

	/* No ordering required: file already has been exposed. */
	RCU_INIT_POINTER(mm->exe_file, get_mm_exe_file(oldmm));

	mm->total_vm = oldmm->total_vm;
	mm->data_vm = oldmm->data_vm;
	mm->exec_vm = oldmm->exec_vm;
	mm->stack_vm = oldmm->stack_vm;

	rb_link = &mm->mm_rb.rb_node;
	rb_parent = NULL;
	pprev = &mm->mmap;
	retval = ksm_fork(mm, oldmm);
	if (retval)
		goto out;
	retval = khugepaged_fork(mm, oldmm);
	if (retval)
		goto out;

	prev = NULL;
	for (mpnt = oldmm->mmap; mpnt; mpnt = mpnt->vm_next) {
		struct file *file;

		if (mpnt->vm_flags & VM_DONTCOPY) {
			vm_stat_account(mm, mpnt->vm_flags, -vma_pages(mpnt));
			continue;
		}
		charge = 0;
468 469 470 471 472 473 474 475
		/*
		 * Don't duplicate many vmas if we've been oom-killed (for
		 * example)
		 */
		if (fatal_signal_pending(current)) {
			retval = -EINTR;
			goto out;
		}
476 477 478 479 480 481 482
		if (mpnt->vm_flags & VM_ACCOUNT) {
			unsigned long len = vma_pages(mpnt);

			if (security_vm_enough_memory_mm(oldmm, len)) /* sic */
				goto fail_nomem;
			charge = len;
		}
483
		tmp = vm_area_dup(mpnt);
484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564
		if (!tmp)
			goto fail_nomem;
		retval = vma_dup_policy(mpnt, tmp);
		if (retval)
			goto fail_nomem_policy;
		tmp->vm_mm = mm;
		retval = dup_userfaultfd(tmp, &uf);
		if (retval)
			goto fail_nomem_anon_vma_fork;
		if (tmp->vm_flags & VM_WIPEONFORK) {
			/* VM_WIPEONFORK gets a clean slate in the child. */
			tmp->anon_vma = NULL;
			if (anon_vma_prepare(tmp))
				goto fail_nomem_anon_vma_fork;
		} else if (anon_vma_fork(tmp, mpnt))
			goto fail_nomem_anon_vma_fork;
		tmp->vm_flags &= ~(VM_LOCKED | VM_LOCKONFAULT);
		tmp->vm_next = tmp->vm_prev = NULL;
		file = tmp->vm_file;
		if (file) {
			struct inode *inode = file_inode(file);
			struct address_space *mapping = file->f_mapping;

			get_file(file);
			if (tmp->vm_flags & VM_DENYWRITE)
				atomic_dec(&inode->i_writecount);
			i_mmap_lock_write(mapping);
			if (tmp->vm_flags & VM_SHARED)
				atomic_inc(&mapping->i_mmap_writable);
			flush_dcache_mmap_lock(mapping);
			/* insert tmp into the share list, just after mpnt */
			vma_interval_tree_insert_after(tmp, mpnt,
					&mapping->i_mmap);
			flush_dcache_mmap_unlock(mapping);
			i_mmap_unlock_write(mapping);
		}

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

		/*
		 * Link in the new vma and copy the page table entries.
		 */
		*pprev = tmp;
		pprev = &tmp->vm_next;
		tmp->vm_prev = prev;
		prev = tmp;

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

		mm->map_count++;
		if (!(tmp->vm_flags & VM_WIPEONFORK))
			retval = copy_page_range(mm, oldmm, mpnt);

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

		if (retval)
			goto out;
	}
	/* a new mm has just been created */
	arch_dup_mmap(oldmm, mm);
	retval = 0;
out:
	up_write(&mm->mmap_sem);
	flush_tlb_mm(oldmm);
	up_write(&oldmm->mmap_sem);
	dup_userfaultfd_complete(&uf);
fail_uprobe_end:
	uprobe_end_dup_mmap();
	return retval;
fail_nomem_anon_vma_fork:
	mpol_put(vma_policy(tmp));
fail_nomem_policy:
565
	vm_area_free(tmp);
566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 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
fail_nomem:
	retval = -ENOMEM;
	vm_unacct_memory(charge);
	goto out;
}

static inline int mm_alloc_pgd(struct mm_struct *mm)
{
	mm->pgd = pgd_alloc(mm);
	if (unlikely(!mm->pgd))
		return -ENOMEM;
	return 0;
}

static inline void mm_free_pgd(struct mm_struct *mm)
{
	pgd_free(mm, mm->pgd);
}
#else
static int dup_mmap(struct mm_struct *mm, struct mm_struct *oldmm)
{
	down_write(&oldmm->mmap_sem);
	RCU_INIT_POINTER(mm->exe_file, get_mm_exe_file(oldmm));
	up_write(&oldmm->mmap_sem);
	return 0;
}
#define mm_alloc_pgd(mm)	(0)
#define mm_free_pgd(mm)
#endif /* CONFIG_MMU */

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);
	}

	if (mm_pgtables_bytes(mm))
		pr_alert("BUG: non-zero pgtables_bytes on freeing mm: %ld\n",
				mm_pgtables_bytes(mm));

#if defined(CONFIG_TRANSPARENT_HUGEPAGE) && !USE_SPLIT_PMD_PTLOCKS
	VM_BUG_ON_MM(mm->pmd_huge_pte, mm);
#endif
}

#define allocate_mm()	(kmem_cache_alloc(mm_cachep, GFP_KERNEL))
#define free_mm(mm)	(kmem_cache_free(mm_cachep, (mm)))

/*
 * 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.
 */
625
void __mmdrop(struct mm_struct *mm)
626 627
{
	BUG_ON(mm == &init_mm);
628 629
	WARN_ON_ONCE(mm == current->mm);
	WARN_ON_ONCE(mm == current->active_mm);
630 631 632 633 634 635 636 637
	mm_free_pgd(mm);
	destroy_context(mm);
	hmm_mm_destroy(mm);
	mmu_notifier_mm_destroy(mm);
	check_mm(mm);
	put_user_ns(mm->user_ns);
	free_mm(mm);
}
638
EXPORT_SYMBOL_GPL(__mmdrop);
639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655

static void mmdrop_async_fn(struct work_struct *work)
{
	struct mm_struct *mm;

	mm = container_of(work, struct mm_struct, async_put_work);
	__mmdrop(mm);
}

static void mmdrop_async(struct mm_struct *mm)
{
	if (unlikely(atomic_dec_and_test(&mm->mm_count))) {
		INIT_WORK(&mm->async_put_work, mmdrop_async_fn);
		schedule_work(&mm->async_put_work);
	}
}

656 657
static inline void free_signal_struct(struct signal_struct *sig)
{
658
	taskstats_tgid_free(sig);
659
	sched_autogroup_exit(sig);
660 661 662 663
	/*
	 * __mmdrop is not safe to call from softirq context on x86 due to
	 * pgd_dtor so postpone it to the async context
	 */
664
	if (sig->oom_mm)
665
		mmdrop_async(sig->oom_mm);
666 667 668 669 670
	kmem_cache_free(signal_cachep, sig);
}

static inline void put_signal_struct(struct signal_struct *sig)
{
671
	if (atomic_dec_and_test(&sig->sigcnt))
672 673 674
		free_signal_struct(sig);
}

675
void __put_task_struct(struct task_struct *tsk)
L
Linus Torvalds 已提交
676
{
E
Eugene Teo 已提交
677
	WARN_ON(!tsk->exit_state);
L
Linus Torvalds 已提交
678 679 680
	WARN_ON(atomic_read(&tsk->usage));
	WARN_ON(tsk == current);

681
	cgroup_free(tsk);
682
	task_numa_free(tsk);
683
	security_task_free(tsk);
684
	exit_creds(tsk);
685
	delayacct_tsk_free(tsk);
686
	put_signal_struct(tsk->signal);
L
Linus Torvalds 已提交
687 688 689 690

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

T
Thomas Gleixner 已提交
693
void __init __weak arch_task_cache_init(void) { }
694

695 696 697
/*
 * set_max_threads
 */
698
static void set_max_threads(unsigned int max_threads_suggested)
699
{
700
	u64 threads;
701 702

	/*
703 704
	 * The number of threads shall be limited such that the thread
	 * structures may only consume a small part of the available memory.
705
	 */
706 707 708 709 710 711
	if (fls64(totalram_pages) + fls64(PAGE_SIZE) > 64)
		threads = MAX_THREADS;
	else
		threads = div64_u64((u64) totalram_pages * (u64) PAGE_SIZE,
				    (u64) THREAD_SIZE * 8UL);

712 713 714
	if (threads > max_threads_suggested)
		threads = max_threads_suggested;

715
	max_threads = clamp_t(u64, threads, MIN_THREADS, MAX_THREADS);
716 717
}

718 719 720 721
#ifdef CONFIG_ARCH_WANTS_DYNAMIC_TASK_STRUCT
/* Initialized by the architecture: */
int arch_task_struct_size __read_mostly;
#endif
722

723 724 725 726 727 728 729 730 731 732 733 734 735 736 737
static void task_struct_whitelist(unsigned long *offset, unsigned long *size)
{
	/* Fetch thread_struct whitelist for the architecture. */
	arch_thread_struct_whitelist(offset, size);

	/*
	 * Handle zero-sized whitelist or empty thread_struct, otherwise
	 * adjust offset to position of thread_struct in task_struct.
	 */
	if (unlikely(*size == 0))
		*offset = 0;
	else
		*offset += offsetof(struct task_struct, thread);
}

738
void __init fork_init(void)
L
Linus Torvalds 已提交
739
{
740
	int i;
741
#ifndef CONFIG_ARCH_TASK_STRUCT_ALLOCATOR
L
Linus Torvalds 已提交
742
#ifndef ARCH_MIN_TASKALIGN
743
#define ARCH_MIN_TASKALIGN	0
L
Linus Torvalds 已提交
744
#endif
P
Peter Zijlstra 已提交
745
	int align = max_t(int, L1_CACHE_BYTES, ARCH_MIN_TASKALIGN);
746
	unsigned long useroffset, usersize;
747

L
Linus Torvalds 已提交
748
	/* create a slab on which task_structs can be allocated */
749 750
	task_struct_whitelist(&useroffset, &usersize);
	task_struct_cachep = kmem_cache_create_usercopy("task_struct",
751
			arch_task_struct_size, align,
752 753
			SLAB_PANIC|SLAB_ACCOUNT,
			useroffset, usersize, NULL);
L
Linus Torvalds 已提交
754 755
#endif

756 757 758
	/* do the arch specific task caches init */
	arch_task_cache_init();

759
	set_max_threads(MAX_THREADS);
L
Linus Torvalds 已提交
760 761 762 763 764

	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];
765

766 767 768
	for (i = 0; i < UCOUNT_COUNTS; i++) {
		init_user_ns.ucount_max[i] = max_threads/2;
	}
769 770 771 772 773

#ifdef CONFIG_VMAP_STACK
	cpuhp_setup_state(CPUHP_BP_PREPARE_DYN, "fork:vm_stack_cache",
			  NULL, free_vm_stack_cache);
#endif
774 775

	lockdep_init_task(&init_task);
L
Linus Torvalds 已提交
776 777
}

778
int __weak arch_dup_task_struct(struct task_struct *dst,
779 780 781 782 783 784
					       struct task_struct *src)
{
	*dst = *src;
	return 0;
}

785 786 787 788 789 790 791 792
void set_task_stack_end_magic(struct task_struct *tsk)
{
	unsigned long *stackend;

	stackend = end_of_stack(tsk);
	*stackend = STACK_END_MAGIC;	/* for overflow detection */
}

793
static struct task_struct *dup_task_struct(struct task_struct *orig, int node)
L
Linus Torvalds 已提交
794 795
{
	struct task_struct *tsk;
796
	unsigned long *stack;
797
	struct vm_struct *stack_vm_area;
P
Peter Zijlstra 已提交
798
	int err;
L
Linus Torvalds 已提交
799

800 801
	if (node == NUMA_NO_NODE)
		node = tsk_fork_get_node(orig);
802
	tsk = alloc_task_struct_node(node);
L
Linus Torvalds 已提交
803 804 805
	if (!tsk)
		return NULL;

806 807
	stack = alloc_thread_stack_node(tsk, node);
	if (!stack)
808
		goto free_tsk;
L
Linus Torvalds 已提交
809

810 811
	stack_vm_area = task_stack_vm_area(tsk);

812
	err = arch_dup_task_struct(tsk, orig);
813 814 815 816 817 818 819 820 821 822

	/*
	 * arch_dup_task_struct() clobbers the stack-related fields.  Make
	 * sure they're properly initialized before using any stack-related
	 * functions again.
	 */
	tsk->stack = stack;
#ifdef CONFIG_VMAP_STACK
	tsk->stack_vm_area = stack_vm_area;
#endif
823 824 825
#ifdef CONFIG_THREAD_INFO_IN_TASK
	atomic_set(&tsk->stack_refcount, 1);
#endif
826

827
	if (err)
828
		goto free_stack;
829

K
Kees Cook 已提交
830 831 832 833 834 835 836 837 838
#ifdef CONFIG_SECCOMP
	/*
	 * We must handle setting up seccomp filters once we're under
	 * the sighand lock in case orig has changed between now and
	 * then. Until then, filter must be NULL to avoid messing up
	 * the usage counts on the error path calling free_task.
	 */
	tsk->seccomp.filter = NULL;
#endif
839 840

	setup_thread_stack(tsk, orig);
841
	clear_user_return_notifier(tsk);
842
	clear_tsk_need_resched(tsk);
843
	set_task_stack_end_magic(tsk);
L
Linus Torvalds 已提交
844

845
#ifdef CONFIG_STACKPROTECTOR
846
	tsk->stack_canary = get_random_canary();
847 848
#endif

849 850 851 852 853
	/*
	 * One for us, one for whoever does the "release_task()" (usually
	 * parent)
	 */
	atomic_set(&tsk->usage, 2);
854
#ifdef CONFIG_BLK_DEV_IO_TRACE
855
	tsk->btrace_seq = 0;
856
#endif
857
	tsk->splice_pipe = NULL;
858
	tsk->task_frag.page = NULL;
859
	tsk->wake_q.next = NULL;
860

861
	account_kernel_stack(tsk, 1);
862

D
Dmitry Vyukov 已提交
863 864
	kcov_task_init(tsk);

865 866 867 868
#ifdef CONFIG_FAULT_INJECTION
	tsk->fail_nth = 0;
#endif

869 870 871 872 873
#ifdef CONFIG_BLK_CGROUP
	tsk->throttle_queue = NULL;
	tsk->use_memdelay = 0;
#endif

874 875 876
#ifdef CONFIG_MEMCG
	tsk->active_memcg = NULL;
#endif
L
Linus Torvalds 已提交
877
	return tsk;
878

879
free_stack:
880
	free_thread_stack(tsk);
881
free_tsk:
882 883
	free_task_struct(tsk);
	return NULL;
L
Linus Torvalds 已提交
884 885
}

D
Daniel Walker 已提交
886
__cacheline_aligned_in_smp DEFINE_SPINLOCK(mmlist_lock);
L
Linus Torvalds 已提交
887

888 889 890 891 892 893 894 895 896 897 898 899
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 已提交
900 901
#include <linux/init_task.h>

A
Alexey Dobriyan 已提交
902 903 904 905
static void mm_init_aio(struct mm_struct *mm)
{
#ifdef CONFIG_AIO
	spin_lock_init(&mm->ioctx_lock);
906
	mm->ioctx_table = NULL;
A
Alexey Dobriyan 已提交
907 908 909
#endif
}

910 911 912 913 914 915 916
static void mm_init_owner(struct mm_struct *mm, struct task_struct *p)
{
#ifdef CONFIG_MEMCG
	mm->owner = p;
#endif
}

917 918 919 920 921 922 923
static void mm_init_uprobes_state(struct mm_struct *mm)
{
#ifdef CONFIG_UPROBES
	mm->uprobes_state.xol_area = NULL;
#endif
}

924 925
static struct mm_struct *mm_init(struct mm_struct *mm, struct task_struct *p,
	struct user_namespace *user_ns)
L
Linus Torvalds 已提交
926
{
927 928 929
	mm->mmap = NULL;
	mm->mm_rb = RB_ROOT;
	mm->vmacache_seqnum = 0;
L
Linus Torvalds 已提交
930 931 932 933
	atomic_set(&mm->mm_users, 1);
	atomic_set(&mm->mm_count, 1);
	init_rwsem(&mm->mmap_sem);
	INIT_LIST_HEAD(&mm->mmlist);
934
	mm->core_state = NULL;
935
	mm_pgtables_bytes_init(mm);
936 937
	mm->map_count = 0;
	mm->locked_vm = 0;
V
Vladimir Davydov 已提交
938
	mm->pinned_vm = 0;
K
KAMEZAWA Hiroyuki 已提交
939
	memset(&mm->rss_stat, 0, sizeof(mm->rss_stat));
L
Linus Torvalds 已提交
940
	spin_lock_init(&mm->page_table_lock);
941
	spin_lock_init(&mm->arg_lock);
942
	mm_init_cpumask(mm);
A
Alexey Dobriyan 已提交
943
	mm_init_aio(mm);
944
	mm_init_owner(mm, p);
945
	RCU_INIT_POINTER(mm->exe_file, NULL);
946
	mmu_notifier_mm_init(mm);
947
	hmm_mm_init(mm);
948
	init_tlb_flush_pending(mm);
949 950 951
#if defined(CONFIG_TRANSPARENT_HUGEPAGE) && !USE_SPLIT_PMD_PTLOCKS
	mm->pmd_huge_pte = NULL;
#endif
952
	mm_init_uprobes_state(mm);
L
Linus Torvalds 已提交
953

954 955 956 957 958
	if (current->mm) {
		mm->flags = current->mm->flags & MMF_INIT_MASK;
		mm->def_flags = current->mm->def_flags & VM_INIT_DEF_MASK;
	} else {
		mm->flags = default_dump_filter;
L
Linus Torvalds 已提交
959
		mm->def_flags = 0;
960 961
	}

962 963 964 965 966
	if (mm_alloc_pgd(mm))
		goto fail_nopgd;

	if (init_new_context(p, mm))
		goto fail_nocontext;
967

968
	mm->user_ns = get_user_ns(user_ns);
969 970 971 972 973
	return mm;

fail_nocontext:
	mm_free_pgd(mm);
fail_nopgd:
L
Linus Torvalds 已提交
974 975 976 977 978 979 980
	free_mm(mm);
	return NULL;
}

/*
 * Allocate and initialize an mm_struct.
 */
981
struct mm_struct *mm_alloc(void)
L
Linus Torvalds 已提交
982
{
983
	struct mm_struct *mm;
L
Linus Torvalds 已提交
984 985

	mm = allocate_mm();
986 987 988 989
	if (!mm)
		return NULL;

	memset(mm, 0, sizeof(*mm));
990
	return mm_init(mm, current, current_user_ns());
L
Linus Torvalds 已提交
991 992
}

993 994 995 996 997 998 999 1000 1001
static inline void __mmput(struct mm_struct *mm)
{
	VM_BUG_ON(atomic_read(&mm->mm_users));

	uprobe_clear_state(mm);
	exit_aio(mm);
	ksm_exit(mm);
	khugepaged_exit(mm); /* must run before exit_mmap */
	exit_mmap(mm);
1002
	mm_put_huge_zero_page(mm);
1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013
	set_mm_exe_file(mm, NULL);
	if (!list_empty(&mm->mmlist)) {
		spin_lock(&mmlist_lock);
		list_del(&mm->mmlist);
		spin_unlock(&mmlist_lock);
	}
	if (mm->binfmt)
		module_put(mm->binfmt->module);
	mmdrop(mm);
}

L
Linus Torvalds 已提交
1014 1015 1016 1017 1018
/*
 * Decrement the use count and release all resources for an mm.
 */
void mmput(struct mm_struct *mm)
{
A
Andrew Morton 已提交
1019 1020
	might_sleep();

1021 1022 1023 1024 1025
	if (atomic_dec_and_test(&mm->mm_users))
		__mmput(mm);
}
EXPORT_SYMBOL_GPL(mmput);

1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043
#ifdef CONFIG_MMU
static void mmput_async_fn(struct work_struct *work)
{
	struct mm_struct *mm = container_of(work, struct mm_struct,
					    async_put_work);

	__mmput(mm);
}

void mmput_async(struct mm_struct *mm)
{
	if (atomic_dec_and_test(&mm->mm_users)) {
		INIT_WORK(&mm->async_put_work, mmput_async_fn);
		schedule_work(&mm->async_put_work);
	}
}
#endif

1044 1045 1046 1047 1048
/**
 * set_mm_exe_file - change a reference to the mm's executable file
 *
 * This changes mm's executable file (shown as symlink /proc/[pid]/exe).
 *
1049 1050 1051 1052 1053
 * Main users are mmput() and sys_execve(). Callers prevent concurrent
 * invocations: in mmput() nobody alive left, in execve task is single
 * threaded. sys_prctl(PR_SET_MM_MAP/EXE_FILE) also needs to set the
 * mm->exe_file, but does so without using set_mm_exe_file() in order
 * to do avoid the need for any locks.
1054
 */
1055 1056
void set_mm_exe_file(struct mm_struct *mm, struct file *new_exe_file)
{
1057 1058 1059 1060 1061 1062 1063 1064
	struct file *old_exe_file;

	/*
	 * It is safe to dereference the exe_file without RCU as
	 * this function is only called if nobody else can access
	 * this mm -- see comment above for justification.
	 */
	old_exe_file = rcu_dereference_raw(mm->exe_file);
1065

1066 1067
	if (new_exe_file)
		get_file(new_exe_file);
1068 1069 1070
	rcu_assign_pointer(mm->exe_file, new_exe_file);
	if (old_exe_file)
		fput(old_exe_file);
1071 1072
}

1073 1074 1075 1076 1077 1078
/**
 * get_mm_exe_file - acquire a reference to the mm's executable file
 *
 * Returns %NULL if mm has no associated executable file.
 * User must release file via fput().
 */
1079 1080 1081 1082
struct file *get_mm_exe_file(struct mm_struct *mm)
{
	struct file *exe_file;

1083 1084 1085 1086 1087
	rcu_read_lock();
	exe_file = rcu_dereference(mm->exe_file);
	if (exe_file && !get_file_rcu(exe_file))
		exe_file = NULL;
	rcu_read_unlock();
1088 1089
	return exe_file;
}
1090
EXPORT_SYMBOL(get_mm_exe_file);
1091

M
Mateusz Guzik 已提交
1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113
/**
 * get_task_exe_file - acquire a reference to the task's executable file
 *
 * Returns %NULL if task's mm (if any) has no associated executable file or
 * this is a kernel thread with borrowed mm (see the comment above get_task_mm).
 * User must release file via fput().
 */
struct file *get_task_exe_file(struct task_struct *task)
{
	struct file *exe_file = NULL;
	struct mm_struct *mm;

	task_lock(task);
	mm = task->mm;
	if (mm) {
		if (!(task->flags & PF_KTHREAD))
			exe_file = get_mm_exe_file(mm);
	}
	task_unlock(task);
	return exe_file;
}
EXPORT_SYMBOL(get_task_exe_file);
1114

L
Linus Torvalds 已提交
1115 1116 1117
/**
 * get_task_mm - acquire a reference to the task's mm
 *
1118
 * Returns %NULL if the task has no mm.  Checks PF_KTHREAD (meaning
L
Linus Torvalds 已提交
1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130
 * 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) {
1131
		if (task->flags & PF_KTHREAD)
L
Linus Torvalds 已提交
1132 1133
			mm = NULL;
		else
V
Vegard Nossum 已提交
1134
			mmget(mm);
L
Linus Torvalds 已提交
1135 1136 1137 1138 1139 1140
	}
	task_unlock(task);
	return mm;
}
EXPORT_SYMBOL_GPL(get_task_mm);

1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160
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;
}

1161
static void complete_vfork_done(struct task_struct *tsk)
1162
{
O
Oleg Nesterov 已提交
1163
	struct completion *vfork;
1164

O
Oleg Nesterov 已提交
1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190
	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;
1191 1192
}

L
Linus Torvalds 已提交
1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207
/* 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)
{
1208 1209
	/* Get rid of any futexes when releasing the mm */
#ifdef CONFIG_FUTEX
1210
	if (unlikely(tsk->robust_list)) {
1211
		exit_robust_list(tsk);
1212 1213
		tsk->robust_list = NULL;
	}
1214
#ifdef CONFIG_COMPAT
1215
	if (unlikely(tsk->compat_robust_list)) {
1216
		compat_exit_robust_list(tsk);
1217 1218
		tsk->compat_robust_list = NULL;
	}
1219
#endif
1220 1221
	if (unlikely(!list_empty(&tsk->pi_state_list)))
		exit_pi_state_list(tsk);
1222 1223
#endif

1224 1225
	uprobe_free_utask(tsk);

L
Linus Torvalds 已提交
1226 1227 1228
	/* Get rid of any cached register state */
	deactivate_mm(tsk, mm);

1229
	/*
1230 1231 1232
	 * Signal userspace if we're not exiting with a core dump
	 * because we want to leave the value intact for debugging
	 * purposes.
1233
	 */
1234
	if (tsk->clear_child_tid) {
1235
		if (!(tsk->signal->flags & SIGNAL_GROUP_COREDUMP) &&
1236 1237 1238 1239 1240 1241
		    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);
1242 1243
			do_futex(tsk->clear_child_tid, FUTEX_WAKE,
					1, NULL, NULL, 0, 0);
1244
		}
L
Linus Torvalds 已提交
1245 1246
		tsk->clear_child_tid = NULL;
	}
1247 1248 1249 1250 1251 1252 1253

	/*
	 * 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 已提交
1254 1255
}

1256 1257 1258 1259
/*
 * Allocate a new mm structure and copy contents from the
 * mm structure of the passed in task structure.
 */
1260
static struct mm_struct *dup_mm(struct task_struct *tsk)
1261 1262 1263 1264 1265 1266 1267 1268 1269 1270
{
	struct mm_struct *mm, *oldmm = current->mm;
	int err;

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

	memcpy(mm, oldmm, sizeof(*mm));

1271
	if (!mm_init(mm, tsk, mm->user_ns))
1272 1273 1274 1275 1276 1277 1278 1279 1280
		goto fail_nomem;

	err = dup_mmap(mm, oldmm);
	if (err)
		goto free_pt;

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

1281 1282 1283
	if (mm->binfmt && !try_module_get(mm->binfmt->module))
		goto free_pt;

1284 1285 1286
	return mm;

free_pt:
1287 1288
	/* don't put binfmt in mmput, we haven't got module yet */
	mm->binfmt = NULL;
1289 1290 1291 1292 1293 1294
	mmput(mm);

fail_nomem:
	return NULL;
}

1295
static int copy_mm(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
1296
{
1297
	struct mm_struct *mm, *oldmm;
L
Linus Torvalds 已提交
1298 1299 1300 1301
	int retval;

	tsk->min_flt = tsk->maj_flt = 0;
	tsk->nvcsw = tsk->nivcsw = 0;
1302 1303 1304
#ifdef CONFIG_DETECT_HUNG_TASK
	tsk->last_switch_count = tsk->nvcsw + tsk->nivcsw;
#endif
L
Linus Torvalds 已提交
1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317

	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;

D
Davidlohr Bueso 已提交
1318 1319 1320
	/* initialize the new vmacache entries */
	vmacache_flush(tsk);

L
Linus Torvalds 已提交
1321
	if (clone_flags & CLONE_VM) {
V
Vegard Nossum 已提交
1322
		mmget(oldmm);
L
Linus Torvalds 已提交
1323 1324 1325 1326 1327
		mm = oldmm;
		goto good_mm;
	}

	retval = -ENOMEM;
1328
	mm = dup_mm(tsk);
L
Linus Torvalds 已提交
1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340
	if (!mm)
		goto fail_nomem;

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

fail_nomem:
	return retval;
}

A
Alexey Dobriyan 已提交
1341
static int copy_fs(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
1342
{
A
Al Viro 已提交
1343
	struct fs_struct *fs = current->fs;
L
Linus Torvalds 已提交
1344
	if (clone_flags & CLONE_FS) {
A
Al Viro 已提交
1345
		/* tsk->fs is already what we want */
N
Nick Piggin 已提交
1346
		spin_lock(&fs->lock);
A
Al Viro 已提交
1347
		if (fs->in_exec) {
N
Nick Piggin 已提交
1348
			spin_unlock(&fs->lock);
A
Al Viro 已提交
1349 1350 1351
			return -EAGAIN;
		}
		fs->users++;
N
Nick Piggin 已提交
1352
		spin_unlock(&fs->lock);
L
Linus Torvalds 已提交
1353 1354
		return 0;
	}
A
Al Viro 已提交
1355
	tsk->fs = copy_fs_struct(fs);
L
Linus Torvalds 已提交
1356 1357 1358 1359 1360
	if (!tsk->fs)
		return -ENOMEM;
	return 0;
}

1361
static int copy_files(unsigned long clone_flags, struct task_struct *tsk)
1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387
{
	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;
}

1388
static int copy_io(unsigned long clone_flags, struct task_struct *tsk)
1389 1390 1391
{
#ifdef CONFIG_BLOCK
	struct io_context *ioc = current->io_context;
1392
	struct io_context *new_ioc;
1393 1394 1395

	if (!ioc)
		return 0;
1396 1397 1398 1399
	/*
	 * Share io context with parent, if CLONE_IO is set
	 */
	if (clone_flags & CLONE_IO) {
T
Tejun Heo 已提交
1400 1401
		ioc_task_link(ioc);
		tsk->io_context = ioc;
1402
	} else if (ioprio_valid(ioc->ioprio)) {
1403 1404
		new_ioc = get_task_io_context(tsk, GFP_KERNEL, NUMA_NO_NODE);
		if (unlikely(!new_ioc))
1405 1406
			return -ENOMEM;

1407
		new_ioc->ioprio = ioc->ioprio;
1408
		put_io_context(new_ioc);
1409 1410 1411 1412 1413
	}
#endif
	return 0;
}

A
Alexey Dobriyan 已提交
1414
static int copy_sighand(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
1415 1416 1417
{
	struct sighand_struct *sig;

Z
Zhaolei 已提交
1418
	if (clone_flags & CLONE_SIGHAND) {
L
Linus Torvalds 已提交
1419 1420 1421 1422
		atomic_inc(&current->sighand->count);
		return 0;
	}
	sig = kmem_cache_alloc(sighand_cachep, GFP_KERNEL);
I
Ingo Molnar 已提交
1423
	rcu_assign_pointer(tsk->sighand, sig);
L
Linus Torvalds 已提交
1424 1425
	if (!sig)
		return -ENOMEM;
1426

L
Linus Torvalds 已提交
1427 1428 1429 1430 1431
	atomic_set(&sig->count, 1);
	memcpy(sig->action, current->sighand->action, sizeof(sig->action));
	return 0;
}

1432
void __cleanup_sighand(struct sighand_struct *sighand)
1433
{
1434 1435
	if (atomic_dec_and_test(&sighand->count)) {
		signalfd_cleanup(sighand);
1436
		/*
1437
		 * sighand_cachep is SLAB_TYPESAFE_BY_RCU so we can free it
1438 1439
		 * without an RCU grace period, see __lock_task_sighand().
		 */
1440
		kmem_cache_free(sighand_cachep, sighand);
1441
	}
1442 1443
}

1444
#ifdef CONFIG_POSIX_TIMERS
1445 1446 1447 1448 1449
/*
 * Initialize POSIX timer handling for a thread group.
 */
static void posix_cpu_timers_init_group(struct signal_struct *sig)
{
1450 1451
	unsigned long cpu_limit;

1452
	cpu_limit = READ_ONCE(sig->rlim[RLIMIT_CPU].rlim_cur);
1453
	if (cpu_limit != RLIM_INFINITY) {
1454
		sig->cputime_expires.prof_exp = cpu_limit * NSEC_PER_SEC;
1455
		sig->cputimer.running = true;
1456 1457
	}

1458 1459 1460 1461 1462
	/* The timer lists. */
	INIT_LIST_HEAD(&sig->cpu_timers[0]);
	INIT_LIST_HEAD(&sig->cpu_timers[1]);
	INIT_LIST_HEAD(&sig->cpu_timers[2]);
}
1463 1464 1465
#else
static inline void posix_cpu_timers_init_group(struct signal_struct *sig) { }
#endif
1466

A
Alexey Dobriyan 已提交
1467
static int copy_signal(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
1468 1469 1470
{
	struct signal_struct *sig;

1471
	if (clone_flags & CLONE_THREAD)
1472 1473
		return 0;

1474
	sig = kmem_cache_zalloc(signal_cachep, GFP_KERNEL);
L
Linus Torvalds 已提交
1475 1476 1477 1478
	tsk->signal = sig;
	if (!sig)
		return -ENOMEM;

1479
	sig->nr_threads = 1;
L
Linus Torvalds 已提交
1480
	atomic_set(&sig->live, 1);
1481
	atomic_set(&sig->sigcnt, 1);
1482 1483 1484 1485 1486

	/* list_add(thread_node, thread_head) without INIT_LIST_HEAD() */
	sig->thread_head = (struct list_head)LIST_HEAD_INIT(tsk->thread_node);
	tsk->thread_node = (struct list_head)LIST_HEAD_INIT(sig->thread_head);

L
Linus Torvalds 已提交
1487
	init_waitqueue_head(&sig->wait_chldexit);
1488
	sig->curr_target = tsk;
L
Linus Torvalds 已提交
1489
	init_sigpending(&sig->shared_pending);
1490
	seqlock_init(&sig->stats_lock);
1491
	prev_cputime_init(&sig->prev_cputime);
L
Linus Torvalds 已提交
1492

1493
#ifdef CONFIG_POSIX_TIMERS
1494
	INIT_LIST_HEAD(&sig->posix_timers);
1495
	hrtimer_init(&sig->real_timer, CLOCK_MONOTONIC, HRTIMER_MODE_REL);
L
Linus Torvalds 已提交
1496
	sig->real_timer.function = it_real_fn;
1497
#endif
L
Linus Torvalds 已提交
1498 1499 1500 1501 1502

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

1503 1504
	posix_cpu_timers_init_group(sig);

M
Miloslav Trmac 已提交
1505
	tty_audit_fork(sig);
1506
	sched_autogroup_fork(sig);
M
Miloslav Trmac 已提交
1507

D
David Rientjes 已提交
1508
	sig->oom_score_adj = current->signal->oom_score_adj;
1509
	sig->oom_score_adj_min = current->signal->oom_score_adj_min;
1510

1511 1512
	mutex_init(&sig->cred_guard_mutex);

L
Linus Torvalds 已提交
1513 1514 1515
	return 0;
}

K
Kees Cook 已提交
1516 1517 1518 1519 1520 1521 1522 1523 1524
static void copy_seccomp(struct task_struct *p)
{
#ifdef CONFIG_SECCOMP
	/*
	 * Must be called with sighand->lock held, which is common to
	 * all threads in the group. Holding cred_guard_mutex is not
	 * needed because this new task is not yet running and cannot
	 * be racing exec.
	 */
1525
	assert_spin_locked(&current->sighand->siglock);
K
Kees Cook 已提交
1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548

	/* Ref-count the new filter user, and assign it. */
	get_seccomp_filter(current);
	p->seccomp = current->seccomp;

	/*
	 * Explicitly enable no_new_privs here in case it got set
	 * between the task_struct being duplicated and holding the
	 * sighand lock. The seccomp state and nnp must be in sync.
	 */
	if (task_no_new_privs(current))
		task_set_no_new_privs(p);

	/*
	 * If the parent gained a seccomp mode after copying thread
	 * flags and between before we held the sighand lock, we have
	 * to manually enable the seccomp thread flag here.
	 */
	if (p->seccomp.mode != SECCOMP_MODE_DISABLED)
		set_tsk_thread_flag(p, TIF_SECCOMP);
#endif
}

1549
SYSCALL_DEFINE1(set_tid_address, int __user *, tidptr)
L
Linus Torvalds 已提交
1550 1551 1552
{
	current->clear_child_tid = tidptr;

1553
	return task_pid_vnr(current);
L
Linus Torvalds 已提交
1554 1555
}

A
Alexey Dobriyan 已提交
1556
static void rt_mutex_init_task(struct task_struct *p)
I
Ingo Molnar 已提交
1557
{
1558
	raw_spin_lock_init(&p->pi_lock);
1559
#ifdef CONFIG_RT_MUTEXES
1560
	p->pi_waiters = RB_ROOT_CACHED;
1561
	p->pi_top_task = NULL;
I
Ingo Molnar 已提交
1562 1563 1564 1565
	p->pi_blocked_on = NULL;
#endif
}

1566
#ifdef CONFIG_POSIX_TIMERS
1567 1568 1569 1570 1571
/*
 * Initialize POSIX timer handling for a single task.
 */
static void posix_cpu_timers_init(struct task_struct *tsk)
{
1572 1573
	tsk->cputime_expires.prof_exp = 0;
	tsk->cputime_expires.virt_exp = 0;
1574 1575 1576 1577 1578
	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]);
}
1579 1580 1581
#else
static inline void posix_cpu_timers_init(struct task_struct *tsk) { }
#endif
1582

1583 1584 1585 1586 1587 1588
static inline void
init_task_pid(struct task_struct *task, enum pid_type type, struct pid *pid)
{
	 task->pids[type].pid = pid;
}

1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603
static inline void rcu_copy_process(struct task_struct *p)
{
#ifdef CONFIG_PREEMPT_RCU
	p->rcu_read_lock_nesting = 0;
	p->rcu_read_unlock_special.s = 0;
	p->rcu_blocked_node = NULL;
	INIT_LIST_HEAD(&p->rcu_node_entry);
#endif /* #ifdef CONFIG_PREEMPT_RCU */
#ifdef CONFIG_TASKS_RCU
	p->rcu_tasks_holdout = false;
	INIT_LIST_HEAD(&p->rcu_tasks_holdout_list);
	p->rcu_tasks_idle_cpu = -1;
#endif /* #ifdef CONFIG_TASKS_RCU */
}

L
Linus Torvalds 已提交
1604 1605 1606 1607 1608 1609 1610 1611
/*
 * 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.
 */
1612 1613
static __latent_entropy struct task_struct *copy_process(
					unsigned long clone_flags,
1614 1615 1616
					unsigned long stack_start,
					unsigned long stack_size,
					int __user *child_tidptr,
R
Roland McGrath 已提交
1617
					struct pid *pid,
1618
					int trace,
1619 1620
					unsigned long tls,
					int node)
L
Linus Torvalds 已提交
1621 1622
{
	int retval;
1623
	struct task_struct *p;
L
Linus Torvalds 已提交
1624

1625 1626 1627 1628
	/*
	 * Don't allow sharing the root directory with processes in a different
	 * namespace
	 */
L
Linus Torvalds 已提交
1629 1630 1631
	if ((clone_flags & (CLONE_NEWNS|CLONE_FS)) == (CLONE_NEWNS|CLONE_FS))
		return ERR_PTR(-EINVAL);

1632 1633 1634
	if ((clone_flags & (CLONE_NEWUSER|CLONE_FS)) == (CLONE_NEWUSER|CLONE_FS))
		return ERR_PTR(-EINVAL);

L
Linus Torvalds 已提交
1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649
	/*
	 * 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);

1650 1651 1652 1653 1654 1655 1656 1657 1658 1659
	/*
	 * 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);

1660
	/*
1661
	 * If the new process will be in a different pid or user namespace
1662
	 * do not allow it to share a thread group with the forking task.
1663
	 */
1664
	if (clone_flags & CLONE_THREAD) {
1665 1666 1667 1668 1669
		if ((clone_flags & (CLONE_NEWUSER | CLONE_NEWPID)) ||
		    (task_active_pid_ns(current) !=
				current->nsproxy->pid_ns_for_children))
			return ERR_PTR(-EINVAL);
	}
1670

L
Linus Torvalds 已提交
1671
	retval = -ENOMEM;
1672
	p = dup_task_struct(current, node);
L
Linus Torvalds 已提交
1673 1674 1675
	if (!p)
		goto fork_out;

1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687
	/*
	 * This _must_ happen before we call free_task(), i.e. before we jump
	 * to any of the bad_fork_* labels. This is to avoid freeing
	 * p->set_child_tid which is (ab)used as a kthread's data pointer for
	 * kernel threads (PF_KTHREAD).
	 */
	p->set_child_tid = (clone_flags & CLONE_CHILD_SETTID) ? child_tidptr : NULL;
	/*
	 * Clear TID on mm_release()?
	 */
	p->clear_child_tid = (clone_flags & CLONE_CHILD_CLEARTID) ? child_tidptr : NULL;

1688 1689
	ftrace_graph_init_task(p);

1690 1691
	rt_mutex_init_task(p);

I
Ingo Molnar 已提交
1692
#ifdef CONFIG_PROVE_LOCKING
1693 1694 1695
	DEBUG_LOCKS_WARN_ON(!p->hardirqs_enabled);
	DEBUG_LOCKS_WARN_ON(!p->softirqs_enabled);
#endif
L
Linus Torvalds 已提交
1696
	retval = -EAGAIN;
1697
	if (atomic_read(&p->real_cred->user->processes) >=
1698
			task_rlimit(p, RLIMIT_NPROC)) {
1699 1700
		if (p->real_cred->user != INIT_USER &&
		    !capable(CAP_SYS_RESOURCE) && !capable(CAP_SYS_ADMIN))
L
Linus Torvalds 已提交
1701 1702
			goto bad_fork_free;
	}
1703
	current->flags &= ~PF_NPROC_EXCEEDED;
L
Linus Torvalds 已提交
1704

1705 1706 1707
	retval = copy_creds(p, clone_flags);
	if (retval < 0)
		goto bad_fork_free;
L
Linus Torvalds 已提交
1708 1709 1710 1711 1712 1713

	/*
	 * 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.
	 */
1714
	retval = -EAGAIN;
L
Linus Torvalds 已提交
1715 1716 1717
	if (nr_threads >= max_threads)
		goto bad_fork_cleanup_count;

1718
	delayacct_tsk_init(p);	/* Must remain after dup_task_struct() */
1719
	p->flags &= ~(PF_SUPERPRIV | PF_WQ_WORKER | PF_IDLE);
1720
	p->flags |= PF_FORKNOEXEC;
L
Linus Torvalds 已提交
1721 1722
	INIT_LIST_HEAD(&p->children);
	INIT_LIST_HEAD(&p->sibling);
1723
	rcu_copy_process(p);
L
Linus Torvalds 已提交
1724 1725 1726 1727 1728
	p->vfork_done = NULL;
	spin_lock_init(&p->alloc_lock);

	init_sigpending(&p->pending);

1729
	p->utime = p->stime = p->gtime = 0;
1730
#ifdef CONFIG_ARCH_HAS_SCALED_CPUTIME
1731
	p->utimescaled = p->stimescaled = 0;
1732
#endif
1733 1734
	prev_cputime_init(&p->prev_cputime);

1735
#ifdef CONFIG_VIRT_CPU_ACCOUNTING_GEN
1736 1737 1738
	seqcount_init(&p->vtime.seqcount);
	p->vtime.starttime = 0;
	p->vtime.state = VTIME_INACTIVE;
1739 1740
#endif

1741 1742 1743
#if defined(SPLIT_RSS_COUNTING)
	memset(&p->rss_stat, 0, sizeof(p->rss_stat));
#endif
1744

1745 1746
	p->default_timer_slack_ns = current->timer_slack_ns;

1747
	task_io_accounting_init(&p->ioac);
L
Linus Torvalds 已提交
1748 1749
	acct_clear_integrals(p);

1750
	posix_cpu_timers_init(p);
L
Linus Torvalds 已提交
1751

1752
	p->start_time = ktime_get_ns();
1753
	p->real_start_time = ktime_get_boot_ns();
L
Linus Torvalds 已提交
1754
	p->io_context = NULL;
1755
	audit_set_context(p, NULL);
1756
	cgroup_fork(p);
L
Linus Torvalds 已提交
1757
#ifdef CONFIG_NUMA
1758
	p->mempolicy = mpol_dup(p->mempolicy);
1759 1760 1761
	if (IS_ERR(p->mempolicy)) {
		retval = PTR_ERR(p->mempolicy);
		p->mempolicy = NULL;
1762
		goto bad_fork_cleanup_threadgroup_lock;
1763
	}
L
Linus Torvalds 已提交
1764
#endif
1765 1766 1767
#ifdef CONFIG_CPUSETS
	p->cpuset_mem_spread_rotor = NUMA_NO_NODE;
	p->cpuset_slab_spread_rotor = NUMA_NO_NODE;
1768
	seqcount_init(&p->mems_allowed_seq);
1769
#endif
1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784
#ifdef CONFIG_TRACE_IRQFLAGS
	p->irq_events = 0;
	p->hardirqs_enabled = 0;
	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
1785 1786 1787

	p->pagefault_disabled = 0;

I
Ingo Molnar 已提交
1788 1789 1790 1791
#ifdef CONFIG_LOCKDEP
	p->lockdep_depth = 0; /* no locks held yet */
	p->curr_chain_key = 0;
	p->lockdep_recursion = 0;
1792
	lockdep_init_task(p);
I
Ingo Molnar 已提交
1793
#endif
L
Linus Torvalds 已提交
1794

1795 1796 1797
#ifdef CONFIG_DEBUG_MUTEXES
	p->blocked_on = NULL; /* not blocked yet */
#endif
K
Kent Overstreet 已提交
1798 1799 1800 1801
#ifdef CONFIG_BCACHE
	p->sequential_io	= 0;
	p->sequential_io_avg	= 0;
#endif
1802

1803
	/* Perform scheduler related setup. Assign this task to a CPU. */
1804 1805 1806
	retval = sched_fork(clone_flags, p);
	if (retval)
		goto bad_fork_cleanup_policy;
1807

1808
	retval = perf_event_init_task(p);
1809 1810
	if (retval)
		goto bad_fork_cleanup_policy;
1811 1812
	retval = audit_alloc(p);
	if (retval)
P
Peter Zijlstra 已提交
1813
		goto bad_fork_cleanup_perf;
L
Linus Torvalds 已提交
1814
	/* copy all the process information */
1815
	shm_init_task(p);
1816
	retval = security_task_alloc(p, clone_flags);
1817
	if (retval)
L
Linus Torvalds 已提交
1818
		goto bad_fork_cleanup_audit;
1819 1820 1821
	retval = copy_semundo(clone_flags, p);
	if (retval)
		goto bad_fork_cleanup_security;
1822 1823
	retval = copy_files(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1824
		goto bad_fork_cleanup_semundo;
1825 1826
	retval = copy_fs(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1827
		goto bad_fork_cleanup_files;
1828 1829
	retval = copy_sighand(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1830
		goto bad_fork_cleanup_fs;
1831 1832
	retval = copy_signal(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1833
		goto bad_fork_cleanup_sighand;
1834 1835
	retval = copy_mm(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1836
		goto bad_fork_cleanup_signal;
1837 1838
	retval = copy_namespaces(clone_flags, p);
	if (retval)
D
David Howells 已提交
1839
		goto bad_fork_cleanup_mm;
1840 1841
	retval = copy_io(clone_flags, p);
	if (retval)
1842
		goto bad_fork_cleanup_namespaces;
1843
	retval = copy_thread_tls(clone_flags, stack_start, stack_size, p, tls);
L
Linus Torvalds 已提交
1844
	if (retval)
1845
		goto bad_fork_cleanup_io;
L
Linus Torvalds 已提交
1846

1847
	if (pid != &init_struct_pid) {
1848
		pid = alloc_pid(p->nsproxy->pid_ns_for_children);
1849 1850
		if (IS_ERR(pid)) {
			retval = PTR_ERR(pid);
1851
			goto bad_fork_cleanup_thread;
1852
		}
1853 1854
	}

1855 1856 1857
#ifdef CONFIG_BLOCK
	p->plug = NULL;
#endif
1858
#ifdef CONFIG_FUTEX
1859 1860 1861 1862
	p->robust_list = NULL;
#ifdef CONFIG_COMPAT
	p->compat_robust_list = NULL;
#endif
1863 1864
	INIT_LIST_HEAD(&p->pi_state_list);
	p->pi_state_cache = NULL;
1865
#endif
1866 1867 1868 1869
	/*
	 * sigaltstack should be cleared when sharing the same VM
	 */
	if ((clone_flags & (CLONE_VM|CLONE_VFORK)) == CLONE_VM)
1870
		sas_ss_reset(p);
1871

L
Linus Torvalds 已提交
1872
	/*
1873 1874
	 * Syscall tracing and stepping should be turned off in the
	 * child regardless of CLONE_PTRACE.
L
Linus Torvalds 已提交
1875
	 */
1876
	user_disable_single_step(p);
L
Linus Torvalds 已提交
1877
	clear_tsk_thread_flag(p, TIF_SYSCALL_TRACE);
1878 1879 1880
#ifdef TIF_SYSCALL_EMU
	clear_tsk_thread_flag(p, TIF_SYSCALL_EMU);
#endif
A
Arjan van de Ven 已提交
1881
	clear_all_latency_tracing(p);
L
Linus Torvalds 已提交
1882 1883

	/* ok, now we should be set up.. */
1884 1885
	p->pid = pid_nr(pid);
	if (clone_flags & CLONE_THREAD) {
1886
		p->exit_signal = -1;
1887 1888 1889 1890 1891 1892 1893 1894 1895 1896
		p->group_leader = current->group_leader;
		p->tgid = current->tgid;
	} else {
		if (clone_flags & CLONE_PARENT)
			p->exit_signal = current->group_leader->exit_signal;
		else
			p->exit_signal = (clone_flags & CSIGNAL);
		p->group_leader = p;
		p->tgid = p->pid;
	}
1897

1898 1899
	p->nr_dirtied = 0;
	p->nr_dirtied_pause = 128 >> (PAGE_SHIFT - 10);
1900
	p->dirty_paused_when = 0;
1901

1902
	p->pdeath_signal = 0;
O
Oleg Nesterov 已提交
1903
	INIT_LIST_HEAD(&p->thread_group);
A
Al Viro 已提交
1904
	p->task_works = NULL;
L
Linus Torvalds 已提交
1905

1906
	cgroup_threadgroup_change_begin(current);
1907 1908 1909 1910 1911 1912
	/*
	 * Ensure that the cgroup subsystem policies allow the new process to be
	 * forked. It should be noted the the new process's css_set can be changed
	 * between here and cgroup_post_fork() if an organisation operation is in
	 * progress.
	 */
1913
	retval = cgroup_can_fork(p);
1914 1915 1916
	if (retval)
		goto bad_fork_free_pid;

1917 1918 1919 1920
	/*
	 * Make it visible to the rest of the system, but dont wake it up yet.
	 * Need tasklist lock for parent etc handling!
	 */
L
Linus Torvalds 已提交
1921 1922 1923
	write_lock_irq(&tasklist_lock);

	/* CLONE_PARENT re-uses the old parent */
1924
	if (clone_flags & (CLONE_PARENT|CLONE_THREAD)) {
L
Linus Torvalds 已提交
1925
		p->real_parent = current->real_parent;
1926 1927
		p->parent_exec_id = current->parent_exec_id;
	} else {
L
Linus Torvalds 已提交
1928
		p->real_parent = current;
1929 1930
		p->parent_exec_id = current->self_exec_id;
	}
L
Linus Torvalds 已提交
1931

1932 1933
	klp_copy_process(p);

1934
	spin_lock(&current->sighand->siglock);
1935

K
Kees Cook 已提交
1936 1937 1938 1939 1940 1941
	/*
	 * Copy seccomp details explicitly here, in case they were changed
	 * before holding sighand lock.
	 */
	copy_seccomp(p);

1942 1943
	rseq_fork(p, clone_flags);

1944 1945 1946 1947 1948 1949 1950
	/*
	 * 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).
1951
	*/
D
Daniel Walker 已提交
1952
	recalc_sigpending();
1953 1954
	if (signal_pending(current)) {
		retval = -ERESTARTNOINTR;
1955
		goto bad_fork_cancel_cgroup;
1956
	}
G
Gargi Sharma 已提交
1957
	if (unlikely(!(ns_of_pid(pid)->pid_allocated & PIDNS_ADDING))) {
1958 1959 1960
		retval = -ENOMEM;
		goto bad_fork_cancel_cgroup;
	}
1961

1962
	if (likely(p->pid)) {
T
Tejun Heo 已提交
1963
		ptrace_init_task(p, (clone_flags & CLONE_PTRACE) || trace);
1964

1965
		init_task_pid(p, PIDTYPE_PID, pid);
1966
		if (thread_group_leader(p)) {
1967 1968 1969
			init_task_pid(p, PIDTYPE_PGID, task_pgrp(current));
			init_task_pid(p, PIDTYPE_SID, task_session(current));

1970
			if (is_child_reaper(pid)) {
1971
				ns_of_pid(pid)->child_reaper = p;
1972 1973
				p->signal->flags |= SIGNAL_UNKILLABLE;
			}
1974

1975
			p->signal->leader_pid = pid;
A
Alan Cox 已提交
1976
			p->signal->tty = tty_kref_get(current->signal->tty);
1977 1978 1979 1980 1981 1982 1983
			/*
			 * Inherit has_child_subreaper flag under the same
			 * tasklist_lock with adding child to the process tree
			 * for propagate_has_child_subreaper optimization.
			 */
			p->signal->has_child_subreaper = p->real_parent->signal->has_child_subreaper ||
							 p->real_parent->signal->is_child_subreaper;
1984
			list_add_tail(&p->sibling, &p->real_parent->children);
1985
			list_add_tail_rcu(&p->tasks, &init_task.tasks);
1986 1987
			attach_pid(p, PIDTYPE_PGID);
			attach_pid(p, PIDTYPE_SID);
1988
			__this_cpu_inc(process_counts);
1989 1990 1991 1992 1993 1994
		} else {
			current->signal->nr_threads++;
			atomic_inc(&current->signal->live);
			atomic_inc(&current->signal->sigcnt);
			list_add_tail_rcu(&p->thread_group,
					  &p->group_leader->thread_group);
1995 1996
			list_add_tail_rcu(&p->thread_node,
					  &p->signal->thread_head);
1997
		}
1998
		attach_pid(p, PIDTYPE_PID);
1999
		nr_threads++;
L
Linus Torvalds 已提交
2000 2001 2002
	}

	total_forks++;
2003
	spin_unlock(&current->sighand->siglock);
2004
	syscall_tracepoint_update(p);
L
Linus Torvalds 已提交
2005
	write_unlock_irq(&tasklist_lock);
2006

2007
	proc_fork_connector(p);
2008
	cgroup_post_fork(p);
2009
	cgroup_threadgroup_change_end(current);
2010
	perf_event_fork(p);
2011 2012

	trace_task_newtask(p, clone_flags);
2013
	uprobe_copy_process(p, clone_flags);
2014

L
Linus Torvalds 已提交
2015 2016
	return p;

2017
bad_fork_cancel_cgroup:
2018 2019
	spin_unlock(&current->sighand->siglock);
	write_unlock_irq(&tasklist_lock);
2020
	cgroup_cancel_fork(p);
2021
bad_fork_free_pid:
2022
	cgroup_threadgroup_change_end(current);
2023 2024
	if (pid != &init_struct_pid)
		free_pid(pid);
2025 2026
bad_fork_cleanup_thread:
	exit_thread(p);
2027
bad_fork_cleanup_io:
2028 2029
	if (p->io_context)
		exit_io_context(p);
S
Serge E. Hallyn 已提交
2030
bad_fork_cleanup_namespaces:
2031
	exit_task_namespaces(p);
L
Linus Torvalds 已提交
2032
bad_fork_cleanup_mm:
D
David Rientjes 已提交
2033
	if (p->mm)
L
Linus Torvalds 已提交
2034 2035
		mmput(p->mm);
bad_fork_cleanup_signal:
2036
	if (!(clone_flags & CLONE_THREAD))
2037
		free_signal_struct(p->signal);
L
Linus Torvalds 已提交
2038
bad_fork_cleanup_sighand:
2039
	__cleanup_sighand(p->sighand);
L
Linus Torvalds 已提交
2040 2041 2042 2043 2044 2045
bad_fork_cleanup_fs:
	exit_fs(p); /* blocking */
bad_fork_cleanup_files:
	exit_files(p); /* blocking */
bad_fork_cleanup_semundo:
	exit_sem(p);
2046 2047
bad_fork_cleanup_security:
	security_task_free(p);
L
Linus Torvalds 已提交
2048 2049
bad_fork_cleanup_audit:
	audit_free(p);
P
Peter Zijlstra 已提交
2050
bad_fork_cleanup_perf:
2051
	perf_event_free_task(p);
P
Peter Zijlstra 已提交
2052
bad_fork_cleanup_policy:
2053
	lockdep_free_task(p);
L
Linus Torvalds 已提交
2054
#ifdef CONFIG_NUMA
2055
	mpol_put(p->mempolicy);
2056
bad_fork_cleanup_threadgroup_lock:
L
Linus Torvalds 已提交
2057
#endif
2058
	delayacct_tsk_free(p);
L
Linus Torvalds 已提交
2059
bad_fork_cleanup_count:
D
David Howells 已提交
2060
	atomic_dec(&p->cred->user->processes);
2061
	exit_creds(p);
L
Linus Torvalds 已提交
2062
bad_fork_free:
2063
	p->state = TASK_DEAD;
2064
	put_task_stack(p);
L
Linus Torvalds 已提交
2065
	free_task(p);
2066 2067
fork_out:
	return ERR_PTR(retval);
L
Linus Torvalds 已提交
2068 2069
}

2070 2071 2072 2073 2074 2075 2076 2077 2078 2079
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;
	}
}

2080
struct task_struct *fork_idle(int cpu)
L
Linus Torvalds 已提交
2081
{
2082
	struct task_struct *task;
2083 2084
	task = copy_process(CLONE_VM, 0, 0, NULL, &init_struct_pid, 0, 0,
			    cpu_to_node(cpu));
2085 2086
	if (!IS_ERR(task)) {
		init_idle_pids(task->pids);
2087
		init_idle(task, cpu);
2088
	}
2089

L
Linus Torvalds 已提交
2090 2091 2092 2093 2094 2095 2096 2097 2098
	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.
 */
2099
long _do_fork(unsigned long clone_flags,
L
Linus Torvalds 已提交
2100 2101 2102
	      unsigned long stack_start,
	      unsigned long stack_size,
	      int __user *parent_tidptr,
2103 2104
	      int __user *child_tidptr,
	      unsigned long tls)
L
Linus Torvalds 已提交
2105
{
2106 2107
	struct completion vfork;
	struct pid *pid;
L
Linus Torvalds 已提交
2108 2109
	struct task_struct *p;
	int trace = 0;
2110
	long nr;
L
Linus Torvalds 已提交
2111

R
Roland McGrath 已提交
2112
	/*
T
Tejun Heo 已提交
2113 2114 2115 2116
	 * 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 已提交
2117
	 */
2118
	if (!(clone_flags & CLONE_UNTRACED)) {
T
Tejun Heo 已提交
2119 2120 2121 2122 2123 2124 2125 2126 2127 2128
		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 已提交
2129

A
Al Viro 已提交
2130
	p = copy_process(clone_flags, stack_start, stack_size,
2131
			 child_tidptr, NULL, trace, tls, NUMA_NO_NODE);
2132
	add_latent_entropy();
2133 2134 2135 2136

	if (IS_ERR(p))
		return PTR_ERR(p);

L
Linus Torvalds 已提交
2137 2138 2139 2140
	/*
	 * Do this prior waking up the new thread - the thread pointer
	 * might get invalid after that point, if the thread exits quickly.
	 */
2141
	trace_sched_process_fork(current, p);
2142

2143 2144
	pid = get_task_pid(p, PIDTYPE_PID);
	nr = pid_vnr(pid);
2145

2146 2147
	if (clone_flags & CLONE_PARENT_SETTID)
		put_user(nr, parent_tidptr);
2148

2149 2150 2151 2152 2153
	if (clone_flags & CLONE_VFORK) {
		p->vfork_done = &vfork;
		init_completion(&vfork);
		get_task_struct(p);
	}
L
Linus Torvalds 已提交
2154

2155
	wake_up_new_task(p);
R
Roland McGrath 已提交
2156

2157 2158 2159
	/* forking complete and child started to run, tell ptracer */
	if (unlikely(trace))
		ptrace_event_pid(trace, pid);
2160

2161 2162 2163
	if (clone_flags & CLONE_VFORK) {
		if (!wait_for_vfork_done(p, &vfork))
			ptrace_event_pid(PTRACE_EVENT_VFORK_DONE, pid);
L
Linus Torvalds 已提交
2164
	}
2165 2166

	put_pid(pid);
2167
	return nr;
L
Linus Torvalds 已提交
2168 2169
}

2170 2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183
#ifndef CONFIG_HAVE_COPY_THREAD_TLS
/* For compatibility with architectures that call do_fork directly rather than
 * using the syscall entry points below. */
long do_fork(unsigned long clone_flags,
	      unsigned long stack_start,
	      unsigned long stack_size,
	      int __user *parent_tidptr,
	      int __user *child_tidptr)
{
	return _do_fork(clone_flags, stack_start, stack_size,
			parent_tidptr, child_tidptr, 0);
}
#endif

2184 2185 2186 2187 2188
/*
 * Create a kernel thread.
 */
pid_t kernel_thread(int (*fn)(void *), void *arg, unsigned long flags)
{
2189 2190
	return _do_fork(flags|CLONE_VM|CLONE_UNTRACED, (unsigned long)fn,
		(unsigned long)arg, NULL, NULL, 0);
2191 2192
}

A
Al Viro 已提交
2193 2194 2195 2196
#ifdef __ARCH_WANT_SYS_FORK
SYSCALL_DEFINE0(fork)
{
#ifdef CONFIG_MMU
2197
	return _do_fork(SIGCHLD, 0, 0, NULL, NULL, 0);
A
Al Viro 已提交
2198 2199
#else
	/* can not support in nommu mode */
2200
	return -EINVAL;
A
Al Viro 已提交
2201 2202 2203 2204 2205 2206 2207
#endif
}
#endif

#ifdef __ARCH_WANT_SYS_VFORK
SYSCALL_DEFINE0(vfork)
{
2208 2209
	return _do_fork(CLONE_VFORK | CLONE_VM | SIGCHLD, 0,
			0, NULL, NULL, 0);
A
Al Viro 已提交
2210 2211 2212 2213 2214 2215 2216
}
#endif

#ifdef __ARCH_WANT_SYS_CLONE
#ifdef CONFIG_CLONE_BACKWARDS
SYSCALL_DEFINE5(clone, unsigned long, clone_flags, unsigned long, newsp,
		 int __user *, parent_tidptr,
2217
		 unsigned long, tls,
A
Al Viro 已提交
2218 2219 2220 2221 2222
		 int __user *, child_tidptr)
#elif defined(CONFIG_CLONE_BACKWARDS2)
SYSCALL_DEFINE5(clone, unsigned long, newsp, unsigned long, clone_flags,
		 int __user *, parent_tidptr,
		 int __user *, child_tidptr,
2223
		 unsigned long, tls)
M
Michal Simek 已提交
2224 2225 2226 2227 2228
#elif defined(CONFIG_CLONE_BACKWARDS3)
SYSCALL_DEFINE6(clone, unsigned long, clone_flags, unsigned long, newsp,
		int, stack_size,
		int __user *, parent_tidptr,
		int __user *, child_tidptr,
2229
		unsigned long, tls)
A
Al Viro 已提交
2230 2231 2232 2233
#else
SYSCALL_DEFINE5(clone, unsigned long, clone_flags, unsigned long, newsp,
		 int __user *, parent_tidptr,
		 int __user *, child_tidptr,
2234
		 unsigned long, tls)
A
Al Viro 已提交
2235 2236
#endif
{
2237
	return _do_fork(clone_flags, newsp, 0, parent_tidptr, child_tidptr, tls);
A
Al Viro 已提交
2238 2239 2240
}
#endif

2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272
void walk_process_tree(struct task_struct *top, proc_visitor visitor, void *data)
{
	struct task_struct *leader, *parent, *child;
	int res;

	read_lock(&tasklist_lock);
	leader = top = top->group_leader;
down:
	for_each_thread(leader, parent) {
		list_for_each_entry(child, &parent->children, sibling) {
			res = visitor(child, data);
			if (res) {
				if (res < 0)
					goto out;
				leader = child;
				goto down;
			}
up:
			;
		}
	}

	if (leader != top) {
		child = leader;
		parent = child->real_parent;
		leader = parent->group_leader;
		goto up;
	}
out:
	read_unlock(&tasklist_lock);
}

2273 2274 2275 2276
#ifndef ARCH_MIN_MMSTRUCT_ALIGN
#define ARCH_MIN_MMSTRUCT_ALIGN 0
#endif

2277
static void sighand_ctor(void *data)
2278 2279 2280
{
	struct sighand_struct *sighand = data;

C
Christoph Lameter 已提交
2281
	spin_lock_init(&sighand->siglock);
D
Davide Libenzi 已提交
2282
	init_waitqueue_head(&sighand->signalfd_wqh);
2283 2284
}

L
Linus Torvalds 已提交
2285 2286
void __init proc_caches_init(void)
{
2287 2288
	unsigned int mm_size;

L
Linus Torvalds 已提交
2289 2290
	sighand_cachep = kmem_cache_create("sighand_cache",
			sizeof(struct sighand_struct), 0,
2291
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_TYPESAFE_BY_RCU|
2292
			SLAB_ACCOUNT, sighand_ctor);
L
Linus Torvalds 已提交
2293 2294
	signal_cachep = kmem_cache_create("signal_cache",
			sizeof(struct signal_struct), 0,
2295
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_ACCOUNT,
2296
			NULL);
2297
	files_cachep = kmem_cache_create("files_cache",
L
Linus Torvalds 已提交
2298
			sizeof(struct files_struct), 0,
2299
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_ACCOUNT,
2300
			NULL);
2301
	fs_cachep = kmem_cache_create("fs_cache",
L
Linus Torvalds 已提交
2302
			sizeof(struct fs_struct), 0,
2303
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_ACCOUNT,
2304
			NULL);
2305

2306
	/*
2307 2308 2309
	 * The mm_cpumask is located at the end of mm_struct, and is
	 * dynamically sized based on the maximum CPU number this system
	 * can have, taking hotplug into account (nr_cpu_ids).
2310
	 */
2311 2312
	mm_size = sizeof(struct mm_struct) + cpumask_size();

2313
	mm_cachep = kmem_cache_create_usercopy("mm_struct",
2314
			mm_size, ARCH_MIN_MMSTRUCT_ALIGN,
2315
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_ACCOUNT,
2316 2317
			offsetof(struct mm_struct, saved_auxv),
			sizeof_field(struct mm_struct, saved_auxv),
2318 2319
			NULL);
	vm_area_cachep = KMEM_CACHE(vm_area_struct, SLAB_PANIC|SLAB_ACCOUNT);
2320
	mmap_init();
2321
	nsproxy_cache_init();
L
Linus Torvalds 已提交
2322
}
2323 2324

/*
2325
 * Check constraints on flags passed to the unshare system call.
2326
 */
2327
static int check_unshare_flags(unsigned long unshare_flags)
2328
{
2329 2330
	if (unshare_flags & ~(CLONE_THREAD|CLONE_FS|CLONE_NEWNS|CLONE_SIGHAND|
				CLONE_VM|CLONE_FILES|CLONE_SYSVSEM|
2331
				CLONE_NEWUTS|CLONE_NEWIPC|CLONE_NEWNET|
2332
				CLONE_NEWUSER|CLONE_NEWPID|CLONE_NEWCGROUP))
2333
		return -EINVAL;
2334
	/*
2335 2336 2337 2338
	 * Not implemented, but pretend it works if there is nothing
	 * to unshare.  Note that unsharing the address space or the
	 * signal handlers also need to unshare the signal queues (aka
	 * CLONE_THREAD).
2339
	 */
2340
	if (unshare_flags & (CLONE_THREAD | CLONE_SIGHAND | CLONE_VM)) {
2341 2342 2343 2344 2345 2346 2347 2348 2349
		if (!thread_group_empty(current))
			return -EINVAL;
	}
	if (unshare_flags & (CLONE_SIGHAND | CLONE_VM)) {
		if (atomic_read(&current->sighand->count) > 1)
			return -EINVAL;
	}
	if (unshare_flags & CLONE_VM) {
		if (!current_is_single_threaded())
2350 2351
			return -EINVAL;
	}
2352 2353 2354 2355 2356

	return 0;
}

/*
2357
 * Unshare the filesystem structure if it is being shared
2358 2359 2360 2361 2362
 */
static int unshare_fs(unsigned long unshare_flags, struct fs_struct **new_fsp)
{
	struct fs_struct *fs = current->fs;

A
Al Viro 已提交
2363 2364 2365 2366 2367 2368 2369 2370 2371 2372
	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;
2373 2374 2375 2376 2377

	return 0;
}

/*
2378
 * Unshare file descriptor table if it is being shared
2379 2380 2381 2382
 */
static int unshare_fd(unsigned long unshare_flags, struct files_struct **new_fdp)
{
	struct files_struct *fd = current->files;
2383
	int error = 0;
2384 2385

	if ((unshare_flags & CLONE_FILES) &&
2386 2387 2388 2389 2390
	    (fd && atomic_read(&fd->count) > 1)) {
		*new_fdp = dup_fd(fd, &error);
		if (!*new_fdp)
			return error;
	}
2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402

	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.
 */
2403
int ksys_unshare(unsigned long unshare_flags)
2404 2405 2406
{
	struct fs_struct *fs, *new_fs = NULL;
	struct files_struct *fd, *new_fd = NULL;
2407
	struct cred *new_cred = NULL;
2408
	struct nsproxy *new_nsproxy = NULL;
2409
	int do_sysvsem = 0;
2410
	int err;
2411

2412
	/*
2413 2414
	 * If unsharing a user namespace must also unshare the thread group
	 * and unshare the filesystem root and working directories.
2415 2416
	 */
	if (unshare_flags & CLONE_NEWUSER)
2417
		unshare_flags |= CLONE_THREAD | CLONE_FS;
2418 2419 2420 2421 2422
	/*
	 * If unsharing vm, must also unshare signal handlers.
	 */
	if (unshare_flags & CLONE_VM)
		unshare_flags |= CLONE_SIGHAND;
2423 2424 2425 2426 2427
	/*
	 * If unsharing a signal handlers, must also unshare the signal queues.
	 */
	if (unshare_flags & CLONE_SIGHAND)
		unshare_flags |= CLONE_THREAD;
2428 2429 2430 2431 2432
	/*
	 * If unsharing namespace, must also unshare filesystem information.
	 */
	if (unshare_flags & CLONE_NEWNS)
		unshare_flags |= CLONE_FS;
2433 2434 2435 2436

	err = check_unshare_flags(unshare_flags);
	if (err)
		goto bad_unshare_out;
2437 2438 2439 2440 2441 2442
	/*
	 * 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))
2443
		do_sysvsem = 1;
2444 2445
	err = unshare_fs(unshare_flags, &new_fs);
	if (err)
2446
		goto bad_unshare_out;
2447 2448
	err = unshare_fd(unshare_flags, &new_fd);
	if (err)
2449
		goto bad_unshare_cleanup_fs;
2450
	err = unshare_userns(unshare_flags, &new_cred);
2451
	if (err)
2452
		goto bad_unshare_cleanup_fd;
2453 2454 2455 2456
	err = unshare_nsproxy_namespaces(unshare_flags, &new_nsproxy,
					 new_cred, new_fs);
	if (err)
		goto bad_unshare_cleanup_cred;
2457

2458
	if (new_fs || new_fd || do_sysvsem || new_cred || new_nsproxy) {
2459 2460 2461 2462 2463 2464
		if (do_sysvsem) {
			/*
			 * CLONE_SYSVSEM is equivalent to sys_exit().
			 */
			exit_sem(current);
		}
2465 2466 2467 2468 2469
		if (unshare_flags & CLONE_NEWIPC) {
			/* Orphan segments in old ns (see sem above). */
			exit_shm(current);
			shm_init_task(current);
		}
S
Serge E. Hallyn 已提交
2470

A
Alan Cox 已提交
2471
		if (new_nsproxy)
2472
			switch_task_namespaces(current, new_nsproxy);
2473

2474 2475
		task_lock(current);

2476 2477
		if (new_fs) {
			fs = current->fs;
N
Nick Piggin 已提交
2478
			spin_lock(&fs->lock);
2479
			current->fs = new_fs;
A
Al Viro 已提交
2480 2481 2482 2483
			if (--fs->users)
				new_fs = NULL;
			else
				new_fs = fs;
N
Nick Piggin 已提交
2484
			spin_unlock(&fs->lock);
2485 2486 2487 2488 2489 2490 2491 2492 2493
		}

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

		task_unlock(current);
2494 2495 2496 2497 2498 2499

		if (new_cred) {
			/* Install the new user namespace */
			commit_creds(new_cred);
			new_cred = NULL;
		}
2500 2501
	}

2502 2503
	perf_event_namespaces(current);

2504 2505 2506
bad_unshare_cleanup_cred:
	if (new_cred)
		put_cred(new_cred);
2507 2508 2509 2510 2511 2512
bad_unshare_cleanup_fd:
	if (new_fd)
		put_files_struct(new_fd);

bad_unshare_cleanup_fs:
	if (new_fs)
A
Al Viro 已提交
2513
		free_fs_struct(new_fs);
2514 2515 2516 2517

bad_unshare_out:
	return err;
}
2518

2519 2520 2521 2522 2523
SYSCALL_DEFINE1(unshare, unsigned long, unshare_flags)
{
	return ksys_unshare(unshare_flags);
}

2524 2525 2526 2527 2528 2529 2530 2531 2532
/*
 *	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;
2533
	struct files_struct *copy = NULL;
2534 2535 2536 2537 2538 2539 2540 2541 2542 2543 2544 2545 2546
	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;
}
2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569

int sysctl_max_threads(struct ctl_table *table, int write,
		       void __user *buffer, size_t *lenp, loff_t *ppos)
{
	struct ctl_table t;
	int ret;
	int threads = max_threads;
	int min = MIN_THREADS;
	int max = MAX_THREADS;

	t = *table;
	t.data = &threads;
	t.extra1 = &min;
	t.extra2 = &max;

	ret = proc_dointvec_minmax(&t, write, buffer, lenp, ppos);
	if (ret || !write)
		return ret;

	set_max_threads(threads);

	return 0;
}