fork.c 44.4 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
	if (err)
		goto out;

R
Roman Zippel 已提交
310
	tsk->stack = ti;
P
Peter Zijlstra 已提交
311

312
	setup_thread_stack(tsk, orig);
313
	clear_user_return_notifier(tsk);
314
	clear_tsk_need_resched(tsk);
315 316
	stackend = end_of_stack(tsk);
	*stackend = STACK_END_MAGIC;	/* for overflow detection */
L
Linus Torvalds 已提交
317

318 319 320 321
#ifdef CONFIG_CC_STACKPROTECTOR
	tsk->stack_canary = get_random_int();
#endif

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

	account_kernel_stack(ti, 1);

L
Linus Torvalds 已提交
334
	return tsk;
335 336 337 338 339

out:
	free_thread_info(ti);
	free_task_struct(tsk);
	return NULL;
L
Linus Torvalds 已提交
340 341 342
}

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

	down_write(&oldmm->mmap_sem);
352
	flush_cache_dup_mm(oldmm);
353 354 355 356
	/*
	 * Not linked in yet - no deadlock potential:
	 */
	down_write_nested(&mm->mmap_sem, SINGLE_DEPTH_NESTING);
357

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

376
	prev = NULL;
H
Hugh Dickins 已提交
377
	for (mpnt = oldmm->mmap; mpnt; mpnt = mpnt->vm_next) {
L
Linus Torvalds 已提交
378 379 380
		struct file *file;

		if (mpnt->vm_flags & VM_DONTCOPY) {
381 382
			long pages = vma_pages(mpnt);
			mm->total_vm -= pages;
383
			vm_stat_account(mm, mpnt->vm_flags, mpnt->vm_file,
384
								-pages);
L
Linus Torvalds 已提交
385 386 387 388
			continue;
		}
		charge = 0;
		if (mpnt->vm_flags & VM_ACCOUNT) {
389 390
			unsigned long len;
			len = (mpnt->vm_end - mpnt->vm_start) >> PAGE_SHIFT;
391
			if (security_vm_enough_memory_mm(oldmm, len)) /* sic */
L
Linus Torvalds 已提交
392 393 394
				goto fail_nomem;
			charge = len;
		}
395
		tmp = kmem_cache_alloc(vm_area_cachep, GFP_KERNEL);
L
Linus Torvalds 已提交
396 397 398
		if (!tmp)
			goto fail_nomem;
		*tmp = *mpnt;
399
		INIT_LIST_HEAD(&tmp->anon_vma_chain);
400
		pol = mpol_dup(vma_policy(mpnt));
L
Linus Torvalds 已提交
401 402 403 404
		retval = PTR_ERR(pol);
		if (IS_ERR(pol))
			goto fail_nomem_policy;
		vma_set_policy(tmp, pol);
A
Andrea Arcangeli 已提交
405
		tmp->vm_mm = mm;
406 407
		if (anon_vma_fork(tmp, mpnt))
			goto fail_nomem_anon_vma_fork;
L
Linus Torvalds 已提交
408
		tmp->vm_flags &= ~VM_LOCKED;
409
		tmp->vm_next = tmp->vm_prev = NULL;
L
Linus Torvalds 已提交
410 411
		file = tmp->vm_file;
		if (file) {
412
			struct inode *inode = file->f_path.dentry->d_inode;
H
Hugh Dickins 已提交
413 414
			struct address_space *mapping = file->f_mapping;

L
Linus Torvalds 已提交
415 416 417
			get_file(file);
			if (tmp->vm_flags & VM_DENYWRITE)
				atomic_dec(&inode->i_writecount);
418
			mutex_lock(&mapping->i_mmap_mutex);
H
Hugh Dickins 已提交
419 420 421 422
			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 已提交
423
			vma_prio_tree_add(tmp, mpnt);
H
Hugh Dickins 已提交
424
			flush_dcache_mmap_unlock(mapping);
425
			mutex_unlock(&mapping->i_mmap_mutex);
L
Linus Torvalds 已提交
426 427
		}

428 429 430 431 432 433 434 435
		/*
		 * 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 已提交
436
		/*
437
		 * Link in the new vma and copy the page table entries.
L
Linus Torvalds 已提交
438 439 440
		 */
		*pprev = tmp;
		pprev = &tmp->vm_next;
441 442
		tmp->vm_prev = prev;
		prev = tmp;
L
Linus Torvalds 已提交
443 444 445 446 447 448

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

		mm->map_count++;
449
		retval = copy_page_range(mm, oldmm, mpnt);
L
Linus Torvalds 已提交
450 451 452 453 454 455

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

		if (retval)
			goto out;
456 457 458

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

478
static inline int mm_alloc_pgd(struct mm_struct *mm)
L
Linus Torvalds 已提交
479 480 481 482 483 484 485
{
	mm->pgd = pgd_alloc(mm);
	if (unlikely(!mm->pgd))
		return -ENOMEM;
	return 0;
}

486
static inline void mm_free_pgd(struct mm_struct *mm)
L
Linus Torvalds 已提交
487
{
488
	pgd_free(mm, mm->pgd);
L
Linus Torvalds 已提交
489 490 491 492 493 494 495
}
#else
#define dup_mmap(mm, oldmm)	(0)
#define mm_alloc_pgd(mm)	(0)
#define mm_free_pgd(mm)
#endif /* CONFIG_MMU */

D
Daniel Walker 已提交
496
__cacheline_aligned_in_smp DEFINE_SPINLOCK(mmlist_lock);
L
Linus Torvalds 已提交
497

498
#define allocate_mm()	(kmem_cache_alloc(mm_cachep, GFP_KERNEL))
L
Linus Torvalds 已提交
499 500
#define free_mm(mm)	(kmem_cache_free(mm_cachep, (mm)))

501 502 503 504 505 506 507 508 509 510 511 512
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 已提交
513 514
#include <linux/init_task.h>

A
Alexey Dobriyan 已提交
515 516 517 518 519 520 521 522
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
}

523
static struct mm_struct *mm_init(struct mm_struct *mm, struct task_struct *p)
L
Linus Torvalds 已提交
524 525 526 527 528
{
	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 已提交
529 530
	mm->flags = (current->mm) ?
		(current->mm->flags & MMF_INIT_MASK) : default_dump_filter;
531
	mm->core_state = NULL;
L
Linus Torvalds 已提交
532
	mm->nr_ptes = 0;
K
KAMEZAWA Hiroyuki 已提交
533
	memset(&mm->rss_stat, 0, sizeof(mm->rss_stat));
L
Linus Torvalds 已提交
534 535
	spin_lock_init(&mm->page_table_lock);
	mm->free_area_cache = TASK_UNMAPPED_BASE;
536
	mm->cached_hole_size = ~0UL;
A
Alexey Dobriyan 已提交
537
	mm_init_aio(mm);
538
	mm_init_owner(mm, p);
L
Linus Torvalds 已提交
539 540 541

	if (likely(!mm_alloc_pgd(mm))) {
		mm->def_flags = 0;
A
Andrea Arcangeli 已提交
542
		mmu_notifier_mm_init(mm);
L
Linus Torvalds 已提交
543 544
		return mm;
	}
545

L
Linus Torvalds 已提交
546 547 548 549
	free_mm(mm);
	return NULL;
}

550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566
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 已提交
567 568 569
/*
 * Allocate and initialize an mm_struct.
 */
570
struct mm_struct *mm_alloc(void)
L
Linus Torvalds 已提交
571
{
572
	struct mm_struct *mm;
L
Linus Torvalds 已提交
573 574

	mm = allocate_mm();
575 576 577 578
	if (!mm)
		return NULL;

	memset(mm, 0, sizeof(*mm));
579 580
	mm_init_cpumask(mm);
	return mm_init(mm, current);
L
Linus Torvalds 已提交
581 582 583 584 585 586 587
}

/*
 * 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.
 */
588
void __mmdrop(struct mm_struct *mm)
L
Linus Torvalds 已提交
589 590 591 592
{
	BUG_ON(mm == &init_mm);
	mm_free_pgd(mm);
	destroy_context(mm);
A
Andrea Arcangeli 已提交
593
	mmu_notifier_mm_destroy(mm);
594
	check_mm(mm);
L
Linus Torvalds 已提交
595 596
	free_mm(mm);
}
597
EXPORT_SYMBOL_GPL(__mmdrop);
L
Linus Torvalds 已提交
598 599 600 601 602 603

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

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

625 626 627 628 629 630 631 632 633 634 635 636 637
/*
 * 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--;
638
	if ((mm->num_exe_file_vmas == 0) && mm->exe_file) {
639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675
		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 已提交
676 677 678
/**
 * get_task_mm - acquire a reference to the task's mm
 *
679
 * Returns %NULL if the task has no mm.  Checks PF_KTHREAD (meaning
L
Linus Torvalds 已提交
680 681 682 683 684 685 686 687 688 689 690 691
 * 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) {
692
		if (task->flags & PF_KTHREAD)
L
Linus Torvalds 已提交
693 694 695 696 697 698 699 700 701
			mm = NULL;
		else
			atomic_inc(&mm->mm_users);
	}
	task_unlock(task);
	return mm;
}
EXPORT_SYMBOL_GPL(get_task_mm);

702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721
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;
}

722
static void complete_vfork_done(struct task_struct *tsk)
723
{
O
Oleg Nesterov 已提交
724
	struct completion *vfork;
725

O
Oleg Nesterov 已提交
726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751
	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;
752 753
}

L
Linus Torvalds 已提交
754 755 756 757 758 759 760 761 762 763 764 765 766 767 768
/* 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)
{
769 770
	/* Get rid of any futexes when releasing the mm */
#ifdef CONFIG_FUTEX
771
	if (unlikely(tsk->robust_list)) {
772
		exit_robust_list(tsk);
773 774
		tsk->robust_list = NULL;
	}
775
#ifdef CONFIG_COMPAT
776
	if (unlikely(tsk->compat_robust_list)) {
777
		compat_exit_robust_list(tsk);
778 779
		tsk->compat_robust_list = NULL;
	}
780
#endif
781 782
	if (unlikely(!list_empty(&tsk->pi_state_list)))
		exit_pi_state_list(tsk);
783 784
#endif

785 786
	uprobe_free_utask(tsk);

L
Linus Torvalds 已提交
787 788 789
	/* Get rid of any cached register state */
	deactivate_mm(tsk, mm);

790 791 792 793
	/*
	 * 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 已提交
794 795
	 * trouble, say, a killed vfork parent shouldn't touch this mm.
	 * Userland only wants this done for a sys_exit.
796
	 */
797 798 799 800 801 802 803 804 805 806 807
	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 已提交
808 809
		tsk->clear_child_tid = NULL;
	}
810 811 812 813 814 815 816

	/*
	 * 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 已提交
817 818
}

819 820 821 822
/*
 * Allocate a new mm structure and copy contents from the
 * mm structure of the passed in task structure.
 */
823
struct mm_struct *dup_mm(struct task_struct *tsk)
824 825 826 827 828 829 830 831 832 833 834 835
{
	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));
836
	mm_init_cpumask(mm);
837

838 839 840
#ifdef CONFIG_TRANSPARENT_HUGEPAGE
	mm->pmd_huge_pte = NULL;
#endif
841
	uprobe_reset_state(mm);
842

843
	if (!mm_init(mm, tsk))
844 845 846 847 848
		goto fail_nomem;

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

M
Matt Helsley 已提交
849 850
	dup_mm_exe_file(oldmm, mm);

851 852 853 854 855 856 857
	err = dup_mmap(mm, oldmm);
	if (err)
		goto free_pt;

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

858 859 860
	if (mm->binfmt && !try_module_get(mm->binfmt->module))
		goto free_pt;

861 862 863
	return mm;

free_pt:
864 865
	/* don't put binfmt in mmput, we haven't got module yet */
	mm->binfmt = NULL;
866 867 868 869 870 871 872 873 874 875 876 877 878 879 880
	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;
}

881
static int copy_mm(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
882
{
883
	struct mm_struct *mm, *oldmm;
L
Linus Torvalds 已提交
884 885 886 887
	int retval;

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

	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;
911
	mm = dup_mm(tsk);
L
Linus Torvalds 已提交
912 913 914 915 916 917 918 919 920 921 922 923
	if (!mm)
		goto fail_nomem;

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

fail_nomem:
	return retval;
}

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

944
static int copy_files(unsigned long clone_flags, struct task_struct *tsk)
945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970
{
	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;
}

971
static int copy_io(unsigned long clone_flags, struct task_struct *tsk)
972 973 974
{
#ifdef CONFIG_BLOCK
	struct io_context *ioc = current->io_context;
975
	struct io_context *new_ioc;
976 977 978

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

990
		new_ioc->ioprio = ioc->ioprio;
991
		put_io_context(new_ioc);
992 993 994 995 996
	}
#endif
	return 0;
}

A
Alexey Dobriyan 已提交
997
static int copy_sighand(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
998 999 1000
{
	struct sighand_struct *sig;

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

1014
void __cleanup_sighand(struct sighand_struct *sighand)
1015
{
1016 1017
	if (atomic_dec_and_test(&sighand->count)) {
		signalfd_cleanup(sighand);
1018
		kmem_cache_free(sighand_cachep, sighand);
1019
	}
1020 1021
}

1022 1023 1024 1025 1026 1027

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

1030 1031 1032
	/* Thread group counters. */
	thread_group_cputime_init(sig);

1033 1034 1035
	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);
1036 1037 1038
		sig->cputimer.running = 1;
	}

1039 1040 1041 1042 1043 1044
	/* 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 已提交
1045
static int copy_signal(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
1046 1047 1048
{
	struct signal_struct *sig;

1049
	if (clone_flags & CLONE_THREAD)
1050 1051
		return 0;

1052
	sig = kmem_cache_zalloc(signal_cachep, GFP_KERNEL);
L
Linus Torvalds 已提交
1053 1054 1055 1056
	tsk->signal = sig;
	if (!sig)
		return -ENOMEM;

1057
	sig->nr_threads = 1;
L
Linus Torvalds 已提交
1058
	atomic_set(&sig->live, 1);
1059
	atomic_set(&sig->sigcnt, 1);
L
Linus Torvalds 已提交
1060
	init_waitqueue_head(&sig->wait_chldexit);
1061 1062
	if (clone_flags & CLONE_NEWPID)
		sig->flags |= SIGNAL_UNKILLABLE;
1063
	sig->curr_target = tsk;
L
Linus Torvalds 已提交
1064 1065 1066
	init_sigpending(&sig->shared_pending);
	INIT_LIST_HEAD(&sig->posix_timers);

1067
	hrtimer_init(&sig->real_timer, CLOCK_MONOTONIC, HRTIMER_MODE_REL);
L
Linus Torvalds 已提交
1068 1069 1070 1071 1072 1073
	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);

1074 1075
	posix_cpu_timers_init_group(sig);

M
Miloslav Trmac 已提交
1076
	tty_audit_fork(sig);
1077
	sched_autogroup_fork(sig);
M
Miloslav Trmac 已提交
1078

1079
#ifdef CONFIG_CGROUPS
1080
	init_rwsem(&sig->group_rwsem);
1081 1082
#endif

1083
	sig->oom_adj = current->signal->oom_adj;
D
David Rientjes 已提交
1084
	sig->oom_score_adj = current->signal->oom_score_adj;
1085
	sig->oom_score_adj_min = current->signal->oom_score_adj_min;
1086

1087 1088 1089
	sig->has_child_subreaper = current->signal->has_child_subreaper ||
				   current->signal->is_child_subreaper;

1090 1091
	mutex_init(&sig->cred_guard_mutex);

L
Linus Torvalds 已提交
1092 1093 1094
	return 0;
}

A
Alexey Dobriyan 已提交
1095
static void copy_flags(unsigned long clone_flags, struct task_struct *p)
L
Linus Torvalds 已提交
1096 1097 1098
{
	unsigned long new_flags = p->flags;

T
Tejun Heo 已提交
1099
	new_flags &= ~(PF_SUPERPRIV | PF_WQ_WORKER);
L
Linus Torvalds 已提交
1100 1101 1102 1103
	new_flags |= PF_FORKNOEXEC;
	p->flags = new_flags;
}

1104
SYSCALL_DEFINE1(set_tid_address, int __user *, tidptr)
L
Linus Torvalds 已提交
1105 1106 1107
{
	current->clear_child_tid = tidptr;

1108
	return task_pid_vnr(current);
L
Linus Torvalds 已提交
1109 1110
}

A
Alexey Dobriyan 已提交
1111
static void rt_mutex_init_task(struct task_struct *p)
I
Ingo Molnar 已提交
1112
{
1113
	raw_spin_lock_init(&p->pi_lock);
1114
#ifdef CONFIG_RT_MUTEXES
1115
	plist_head_init(&p->pi_waiters);
I
Ingo Molnar 已提交
1116 1117 1118 1119
	p->pi_blocked_on = NULL;
#endif
}

1120 1121 1122 1123 1124 1125 1126
#ifdef CONFIG_MM_OWNER
void mm_init_owner(struct mm_struct *mm, struct task_struct *p)
{
	mm->owner = p;
}
#endif /* CONFIG_MM_OWNER */

1127 1128 1129 1130 1131
/*
 * Initialize POSIX timer handling for a single task.
 */
static void posix_cpu_timers_init(struct task_struct *tsk)
{
1132 1133
	tsk->cputime_expires.prof_exp = 0;
	tsk->cputime_expires.virt_exp = 0;
1134 1135 1136 1137 1138 1139
	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 已提交
1140 1141 1142 1143 1144 1145 1146 1147
/*
 * 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.
 */
1148 1149 1150 1151 1152
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 已提交
1153 1154
					struct pid *pid,
					int trace)
L
Linus Torvalds 已提交
1155 1156
{
	int retval;
1157
	struct task_struct *p;
1158
	int cgroup_callbacks_done = 0;
L
Linus Torvalds 已提交
1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177

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

1178 1179 1180 1181 1182 1183 1184 1185 1186 1187
	/*
	 * 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 已提交
1188 1189 1190 1191 1192 1193 1194 1195 1196
	retval = security_task_create(clone_flags);
	if (retval)
		goto fork_out;

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

1197
	ftrace_graph_init_task(p);
1198
	get_seccomp_filter(p);
1199

1200 1201
	rt_mutex_init_task(p);

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

1215 1216 1217
	retval = copy_creds(p, clone_flags);
	if (retval < 0)
		goto bad_fork_free;
L
Linus Torvalds 已提交
1218 1219 1220 1221 1222 1223

	/*
	 * 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.
	 */
1224
	retval = -EAGAIN;
L
Linus Torvalds 已提交
1225 1226 1227
	if (nr_threads >= max_threads)
		goto bad_fork_cleanup_count;

A
Al Viro 已提交
1228
	if (!try_module_get(task_thread_info(p)->exec_domain->module))
L
Linus Torvalds 已提交
1229 1230 1231
		goto bad_fork_cleanup_count;

	p->did_exec = 0;
1232
	delayacct_tsk_init(p);	/* Must remain after dup_task_struct() */
L
Linus Torvalds 已提交
1233 1234 1235
	copy_flags(clone_flags, p);
	INIT_LIST_HEAD(&p->children);
	INIT_LIST_HEAD(&p->sibling);
1236
	rcu_copy_process(p);
L
Linus Torvalds 已提交
1237 1238 1239 1240 1241
	p->vfork_done = NULL;
	spin_lock_init(&p->alloc_lock);

	init_sigpending(&p->pending);

1242 1243
	p->utime = p->stime = p->gtime = 0;
	p->utimescaled = p->stimescaled = 0;
1244
#ifndef CONFIG_VIRT_CPU_ACCOUNTING
1245
	p->prev_utime = p->prev_stime = 0;
1246
#endif
1247 1248 1249
#if defined(SPLIT_RSS_COUNTING)
	memset(&p->rss_stat, 0, sizeof(p->rss_stat));
#endif
1250

1251 1252
	p->default_timer_slack_ns = current->timer_slack_ns;

1253
	task_io_accounting_init(&p->ioac);
L
Linus Torvalds 已提交
1254 1255
	acct_clear_integrals(p);

1256
	posix_cpu_timers_init(p);
L
Linus Torvalds 已提交
1257 1258

	do_posix_clock_monotonic_gettime(&p->start_time);
1259 1260
	p->real_start_time = p->start_time;
	monotonic_to_bootbased(&p->real_start_time);
L
Linus Torvalds 已提交
1261 1262
	p->io_context = NULL;
	p->audit_context = NULL;
1263
	if (clone_flags & CLONE_THREAD)
1264
		threadgroup_change_begin(current);
1265
	cgroup_fork(p);
L
Linus Torvalds 已提交
1266
#ifdef CONFIG_NUMA
1267
	p->mempolicy = mpol_dup(p->mempolicy);
1268 1269 1270 1271 1272
	if (IS_ERR(p->mempolicy)) {
		retval = PTR_ERR(p->mempolicy);
		p->mempolicy = NULL;
		goto bad_fork_cleanup_cgroup;
	}
1273
	mpol_fix_fork_child_flag(p);
L
Linus Torvalds 已提交
1274
#endif
1275 1276 1277
#ifdef CONFIG_CPUSETS
	p->cpuset_mem_spread_rotor = NUMA_NO_NODE;
	p->cpuset_slab_spread_rotor = NUMA_NO_NODE;
1278
	seqcount_init(&p->mems_allowed_seq);
1279
#endif
1280 1281
#ifdef CONFIG_TRACE_IRQFLAGS
	p->irq_events = 0;
1282 1283 1284
#ifdef __ARCH_WANT_INTERRUPTS_ON_CTXSW
	p->hardirqs_enabled = 1;
#else
1285
	p->hardirqs_enabled = 0;
1286
#endif
1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298
	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 已提交
1299 1300 1301 1302 1303
#ifdef CONFIG_LOCKDEP
	p->lockdep_depth = 0; /* no locks held yet */
	p->curr_chain_key = 0;
	p->lockdep_recursion = 0;
#endif
L
Linus Torvalds 已提交
1304

1305 1306 1307
#ifdef CONFIG_DEBUG_MUTEXES
	p->blocked_on = NULL; /* not blocked yet */
#endif
1308 1309 1310 1311
#ifdef CONFIG_CGROUP_MEM_RES_CTLR
	p->memcg_batch.do_batch = 0;
	p->memcg_batch.memcg = NULL;
#endif
1312

1313
	/* Perform scheduler related setup. Assign this task to a CPU. */
1314
	sched_fork(p);
1315

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

1351 1352
	if (pid != &init_struct_pid) {
		retval = -ENOMEM;
1353
		pid = alloc_pid(p->nsproxy->pid_ns);
1354
		if (!pid)
1355
			goto bad_fork_cleanup_io;
1356 1357 1358 1359 1360 1361 1362
	}

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

L
Linus Torvalds 已提交
1363 1364 1365 1366
	p->set_child_tid = (clone_flags & CLONE_CHILD_SETTID) ? child_tidptr : NULL;
	/*
	 * Clear TID on mm_release()?
	 */
1367
	p->clear_child_tid = (clone_flags & CLONE_CHILD_CLEARTID) ? child_tidptr : NULL;
1368 1369 1370
#ifdef CONFIG_BLOCK
	p->plug = NULL;
#endif
1371
#ifdef CONFIG_FUTEX
1372 1373 1374 1375
	p->robust_list = NULL;
#ifdef CONFIG_COMPAT
	p->compat_robust_list = NULL;
#endif
1376 1377
	INIT_LIST_HEAD(&p->pi_state_list);
	p->pi_state_cache = NULL;
1378
#endif
1379
	uprobe_copy_process(p);
1380 1381 1382 1383 1384 1385
	/*
	 * 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 已提交
1386
	/*
1387 1388
	 * Syscall tracing and stepping should be turned off in the
	 * child regardless of CLONE_PTRACE.
L
Linus Torvalds 已提交
1389
	 */
1390
	user_disable_single_step(p);
L
Linus Torvalds 已提交
1391
	clear_tsk_thread_flag(p, TIF_SYSCALL_TRACE);
1392 1393 1394
#ifdef TIF_SYSCALL_EMU
	clear_tsk_thread_flag(p, TIF_SYSCALL_EMU);
#endif
A
Arjan van de Ven 已提交
1395
	clear_all_latency_tracing(p);
L
Linus Torvalds 已提交
1396 1397

	/* ok, now we should be set up.. */
1398 1399 1400 1401 1402 1403 1404
	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 已提交
1405 1406 1407
	p->pdeath_signal = 0;
	p->exit_state = 0;

1408 1409
	p->nr_dirtied = 0;
	p->nr_dirtied_pause = 128 >> (PAGE_SHIFT - 10);
1410
	p->dirty_paused_when = 0;
1411

L
Linus Torvalds 已提交
1412 1413 1414 1415 1416
	/*
	 * Ok, make it visible to the rest of the system.
	 * We dont wake it up yet.
	 */
	p->group_leader = p;
O
Oleg Nesterov 已提交
1417
	INIT_LIST_HEAD(&p->thread_group);
1418
	INIT_HLIST_HEAD(&p->task_works);
L
Linus Torvalds 已提交
1419

1420 1421 1422 1423 1424 1425
	/* 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 已提交
1426 1427 1428 1429
	/* Need tasklist lock for parent etc handling! */
	write_lock_irq(&tasklist_lock);

	/* CLONE_PARENT re-uses the old parent */
1430
	if (clone_flags & (CLONE_PARENT|CLONE_THREAD)) {
L
Linus Torvalds 已提交
1431
		p->real_parent = current->real_parent;
1432 1433
		p->parent_exec_id = current->parent_exec_id;
	} else {
L
Linus Torvalds 已提交
1434
		p->real_parent = current;
1435 1436
		p->parent_exec_id = current->self_exec_id;
	}
L
Linus Torvalds 已提交
1437

1438
	spin_lock(&current->sighand->siglock);
1439 1440 1441 1442 1443 1444 1445 1446

	/*
	 * 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).
1447
	*/
D
Daniel Walker 已提交
1448
	recalc_sigpending();
1449 1450 1451 1452
	if (signal_pending(current)) {
		spin_unlock(&current->sighand->siglock);
		write_unlock_irq(&tasklist_lock);
		retval = -ERESTARTNOINTR;
1453
		goto bad_fork_free_pid;
1454 1455
	}

L
Linus Torvalds 已提交
1456
	if (clone_flags & CLONE_THREAD) {
1457
		current->signal->nr_threads++;
1458
		atomic_inc(&current->signal->live);
1459
		atomic_inc(&current->signal->sigcnt);
L
Linus Torvalds 已提交
1460
		p->group_leader = current->group_leader;
O
Oleg Nesterov 已提交
1461
		list_add_tail_rcu(&p->thread_group, &p->group_leader->thread_group);
L
Linus Torvalds 已提交
1462 1463
	}

1464
	if (likely(p->pid)) {
T
Tejun Heo 已提交
1465
		ptrace_init_task(p, (clone_flags & CLONE_PTRACE) || trace);
1466 1467

		if (thread_group_leader(p)) {
1468
			if (is_child_reaper(pid))
1469
				p->nsproxy->pid_ns->child_reaper = p;
1470

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

	total_forks++;
1484
	spin_unlock(&current->sighand->siglock);
L
Linus Torvalds 已提交
1485
	write_unlock_irq(&tasklist_lock);
1486
	proc_fork_connector(p);
1487
	cgroup_post_fork(p);
1488
	if (clone_flags & CLONE_THREAD)
1489
		threadgroup_change_end(current);
1490
	perf_event_fork(p);
1491 1492 1493

	trace_task_newtask(p, clone_flags);

L
Linus Torvalds 已提交
1494 1495
	return p;

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

A
Adrian Bunk 已提交
1542
noinline struct pt_regs * __cpuinit __attribute__((weak)) idle_regs(struct pt_regs *regs)
L
Linus Torvalds 已提交
1543 1544 1545 1546 1547
{
	memset(regs, 0, sizeof(struct pt_regs));
	return regs;
}

1548 1549 1550 1551 1552 1553 1554 1555 1556 1557
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;
	}
}

1558
struct task_struct * __cpuinit fork_idle(int cpu)
L
Linus Torvalds 已提交
1559
{
1560
	struct task_struct *task;
L
Linus Torvalds 已提交
1561 1562
	struct pt_regs regs;

1563
	task = copy_process(CLONE_VM, 0, idle_regs(&regs), 0, NULL,
R
Roland McGrath 已提交
1564
			    &init_struct_pid, 0);
1565 1566
	if (!IS_ERR(task)) {
		init_idle_pids(task->pids);
1567
		init_idle(task, cpu);
1568
	}
1569

L
Linus Torvalds 已提交
1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587
	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;
1588
	long nr;
L
Linus Torvalds 已提交
1589

1590 1591 1592 1593 1594 1595 1596 1597 1598 1599
	/*
	 * 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
		 */
1600 1601
		if (!capable(CAP_SYS_ADMIN) || !capable(CAP_SETUID) ||
				!capable(CAP_SETGID))
1602 1603 1604
			return -EPERM;
	}

R
Roland McGrath 已提交
1605
	/*
T
Tejun Heo 已提交
1606 1607 1608 1609
	 * 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 已提交
1610
	 */
T
Tejun Heo 已提交
1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621
	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 已提交
1622

1623
	p = copy_process(clone_flags, stack_start, regs, stack_size,
R
Roland McGrath 已提交
1624
			 child_tidptr, NULL, trace);
L
Linus Torvalds 已提交
1625 1626 1627 1628 1629 1630 1631
	/*
	 * 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;

1632 1633
		trace_sched_process_fork(current, p);

1634
		nr = task_pid_vnr(p);
1635 1636 1637

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

L
Linus Torvalds 已提交
1639 1640 1641
		if (clone_flags & CLONE_VFORK) {
			p->vfork_done = &vfork;
			init_completion(&vfork);
O
Oleg Nesterov 已提交
1642
			get_task_struct(p);
L
Linus Torvalds 已提交
1643 1644
		}

1645
		wake_up_new_task(p);
L
Linus Torvalds 已提交
1646

T
Tejun Heo 已提交
1647 1648 1649
		/* forking complete and child started to run, tell ptracer */
		if (unlikely(trace))
			ptrace_event(trace, nr);
R
Roland McGrath 已提交
1650

L
Linus Torvalds 已提交
1651
		if (clone_flags & CLONE_VFORK) {
O
Oleg Nesterov 已提交
1652 1653
			if (!wait_for_vfork_done(p, &vfork))
				ptrace_event(PTRACE_EVENT_VFORK_DONE, nr);
L
Linus Torvalds 已提交
1654 1655
		}
	} else {
1656
		nr = PTR_ERR(p);
L
Linus Torvalds 已提交
1657
	}
1658
	return nr;
L
Linus Torvalds 已提交
1659 1660
}

1661 1662 1663 1664
#ifndef ARCH_MIN_MMSTRUCT_ALIGN
#define ARCH_MIN_MMSTRUCT_ALIGN 0
#endif

1665
static void sighand_ctor(void *data)
1666 1667 1668
{
	struct sighand_struct *sighand = data;

C
Christoph Lameter 已提交
1669
	spin_lock_init(&sighand->siglock);
D
Davide Libenzi 已提交
1670
	init_waitqueue_head(&sighand->signalfd_wqh);
1671 1672
}

L
Linus Torvalds 已提交
1673 1674 1675 1676
void __init proc_caches_init(void)
{
	sighand_cachep = kmem_cache_create("sighand_cache",
			sizeof(struct sighand_struct), 0,
V
Vegard Nossum 已提交
1677 1678
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_DESTROY_BY_RCU|
			SLAB_NOTRACK, sighand_ctor);
L
Linus Torvalds 已提交
1679 1680
	signal_cachep = kmem_cache_create("signal_cache",
			sizeof(struct signal_struct), 0,
V
Vegard Nossum 已提交
1681
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1682
	files_cachep = kmem_cache_create("files_cache",
L
Linus Torvalds 已提交
1683
			sizeof(struct files_struct), 0,
V
Vegard Nossum 已提交
1684
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1685
	fs_cachep = kmem_cache_create("fs_cache",
L
Linus Torvalds 已提交
1686
			sizeof(struct fs_struct), 0,
V
Vegard Nossum 已提交
1687
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1688 1689 1690 1691 1692 1693 1694
	/*
	 * 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 已提交
1695
	mm_cachep = kmem_cache_create("mm_struct",
1696
			sizeof(struct mm_struct), ARCH_MIN_MMSTRUCT_ALIGN,
V
Vegard Nossum 已提交
1697
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1698
	vm_area_cachep = KMEM_CACHE(vm_area_struct, SLAB_PANIC);
1699
	mmap_init();
1700
	nsproxy_cache_init();
L
Linus Torvalds 已提交
1701
}
1702 1703

/*
1704
 * Check constraints on flags passed to the unshare system call.
1705
 */
1706
static int check_unshare_flags(unsigned long unshare_flags)
1707
{
1708 1709 1710 1711
	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;
1712
	/*
1713 1714 1715
	 * Not implemented, but pretend it works if there is nothing to
	 * unshare. Note that unsharing CLONE_THREAD or CLONE_SIGHAND
	 * needs to unshare vm.
1716
	 */
1717 1718 1719 1720 1721
	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;
	}
1722 1723 1724 1725 1726

	return 0;
}

/*
1727
 * Unshare the filesystem structure if it is being shared
1728 1729 1730 1731 1732
 */
static int unshare_fs(unsigned long unshare_flags, struct fs_struct **new_fsp)
{
	struct fs_struct *fs = current->fs;

A
Al Viro 已提交
1733 1734 1735 1736 1737 1738 1739 1740 1741 1742
	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;
1743 1744 1745 1746 1747

	return 0;
}

/*
1748
 * Unshare file descriptor table if it is being shared
1749 1750 1751 1752
 */
static int unshare_fd(unsigned long unshare_flags, struct files_struct **new_fdp)
{
	struct files_struct *fd = current->files;
1753
	int error = 0;
1754 1755

	if ((unshare_flags & CLONE_FILES) &&
1756 1757 1758 1759 1760
	    (fd && atomic_read(&fd->count) > 1)) {
		*new_fdp = dup_fd(fd, &error);
		if (!*new_fdp)
			return error;
	}
1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772

	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.
 */
1773
SYSCALL_DEFINE1(unshare, unsigned long, unshare_flags)
1774 1775 1776
{
	struct fs_struct *fs, *new_fs = NULL;
	struct files_struct *fd, *new_fd = NULL;
1777
	struct nsproxy *new_nsproxy = NULL;
1778
	int do_sysvsem = 0;
1779
	int err;
1780

1781 1782
	err = check_unshare_flags(unshare_flags);
	if (err)
1783 1784
		goto bad_unshare_out;

1785 1786 1787 1788 1789
	/*
	 * If unsharing namespace, must also unshare filesystem information.
	 */
	if (unshare_flags & CLONE_NEWNS)
		unshare_flags |= CLONE_FS;
1790 1791 1792 1793 1794 1795
	/*
	 * 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))
1796
		do_sysvsem = 1;
1797 1798
	err = unshare_fs(unshare_flags, &new_fs);
	if (err)
1799
		goto bad_unshare_out;
1800 1801
	err = unshare_fd(unshare_flags, &new_fd);
	if (err)
1802
		goto bad_unshare_cleanup_fs;
1803 1804
	err = unshare_nsproxy_namespaces(unshare_flags, &new_nsproxy, new_fs);
	if (err)
1805
		goto bad_unshare_cleanup_fd;
1806

1807
	if (new_fs || new_fd || do_sysvsem || new_nsproxy) {
1808 1809 1810 1811 1812 1813
		if (do_sysvsem) {
			/*
			 * CLONE_SYSVSEM is equivalent to sys_exit().
			 */
			exit_sem(current);
		}
S
Serge E. Hallyn 已提交
1814

1815
		if (new_nsproxy) {
1816 1817
			switch_task_namespaces(current, new_nsproxy);
			new_nsproxy = NULL;
1818
		}
1819

1820 1821
		task_lock(current);

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

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

		task_unlock(current);
	}

1842
	if (new_nsproxy)
1843
		put_nsproxy(new_nsproxy);
1844

1845 1846 1847 1848 1849 1850
bad_unshare_cleanup_fd:
	if (new_fd)
		put_files_struct(new_fd);

bad_unshare_cleanup_fs:
	if (new_fs)
A
Al Viro 已提交
1851
		free_fs_struct(new_fs);
1852 1853 1854 1855

bad_unshare_out:
	return err;
}
1856 1857 1858 1859 1860 1861 1862 1863 1864 1865

/*
 *	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;
1866
	struct files_struct *copy = NULL;
1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879
	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;
}